Fix #21 — old artifact in hubconf.py

kdexd · Apr 27, 2021 · 341aef7 · 341aef7
1 parent 2948a3b
commit 341aef7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -30,6 +30,6 @@ def resnet50(pretrained: bool = False, **kwargs):
             torch.hub.load_state_dict_from_url(
                 "https://umich.box.com/shared/static/gsjqm4i4fm1wpzi947h27wweljd8gcpy.pth",
                 progress=False,
-            )["model"]
+            )
         )
     return model