Update model links

RF5 · Jul 8, 2019 · c8518d9 · c8518d9
1 parent 9a8d5cb
commit c8518d9
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/danbooru_resnet.py b/danbooru_resnet.py
@@ -85,9 +85,9 @@ def resnet18(pretrained=True, progress=True, top_n=100, **kwargs):
     model = _resnet(models.resnet18, top_n, **kwargs)
     if pretrained:
         if top_n == 100: 
-            #state = torch.hub.load_state_dict_from_url("GITHUB RELEASE URL!!!!", 
-            #                                        progress=progress)
-            state = torch.load('weights/resnet18.pth')
+            state = torch.hub.load_state_dict_from_url("https://github.com/RF5/danbooru-pretrained/releases/download/untagged-d7dd93226bc56cbf2fe8/resnet18.pth", 
+                                                   progress=progress)
+            # state = torch.load('weights/resnet18.pth')
             model.load_state_dict(state)
         else:
             raise ValueError("Sorry, the resnet18 model only supports the top-100 tags \
@@ -108,9 +108,9 @@ def resnet34(pretrained=True, progress=True, top_n=500, **kwargs):
     model = _resnet(models.resnet34, top_n, **kwargs)
     if pretrained:
         if top_n == 500: 
-            #state = torch.hub.load_state_dict_from_url("GITHUB RELEASE URL!!!!", 
-            #                                        progress=progress)
-            state = torch.load('weights/resnet34.pth')
+            state = torch.hub.load_state_dict_from_url("https://github.com/RF5/danbooru-pretrained/releases/download/untagged-d7dd93226bc56cbf2fe8/resnet34.pth", 
+                                                   progress=progress)
+            # state = torch.load('weights/resnet34.pth')
             model.load_state_dict(state)
         else:
             raise ValueError("Sorry, the resnet34 model only supports the top-500 tags \
@@ -130,9 +130,9 @@ def resnet50(pretrained=True, progress=True, top_n=6000, **kwargs):
     model = _resnet(models.resnet50, top_n, **kwargs)
     if pretrained:
         if top_n == 6000: 
-            #state = torch.hub.load_state_dict_from_url("GITHUB RELEASE URL!!!!", 
-            #                                        progress=progress)
-            state = torch.load('weights/resnet50.pth')
+            state = torch.hub.load_state_dict_from_url("https://github.com/RF5/danbooru-pretrained/releases/download/untagged-d7dd93226bc56cbf2fe8/resnet50.pth", 
+                                                   progress=progress)
+            # state = torch.load('weights/resnet50.pth')
             model.load_state_dict(state)
         else:
             raise ValueError("Sorry, the resnet50 model only supports the top-6000 tags \