extend ResNet pretrained info (Project-MONAI#2854)

Signed-off-by: Jirka <jirka.borovec@seznam.cz>
Borda · Aug 27, 2021 · 0725bce · 0725bce
1 parent e8db3dc
commit 0725bce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/monai/networks/nets/resnet.py b/monai/networks/nets/resnet.py
@@ -315,7 +315,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