Skip to content

Commit

Permalink
extend ResNet pretrained info (#2854)
Browse files Browse the repository at this point in the history
Signed-off-by: Jirka <jirka.borovec@seznam.cz>
  • Loading branch information
Borda authored Aug 26, 2021
1 parent 3eb71c7 commit dd56514
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion monai/networks/nets/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -308,7 +308,10 @@ def _resnet(
# Author of paper zipped the state_dict on googledrive,
# so would need to download, unzip and read (2.8gb file for a ~150mb state dict).
# Would like to load dict from url but need somewhere to save the state dicts.
raise NotImplementedError("Currently not implemented, see comments in source code")
raise NotImplementedError(
"Currently not implemented. You need to manually download weights provided by the paper's author"
" and load then to the model with `state_dict`. See https://github.com/Tencent/MedicalNet"
)
return model


Expand Down

0 comments on commit dd56514

Please sign in to comment.