Fix issue clovaai#191 with torch>0.13

Victorivus · Dec 5, 2023 · f734a2d · f734a2d
1 parent a17bbff
commit f734a2d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/basenet/vgg16_bn.py b/basenet/vgg16_bn.py
@@ -4,7 +4,6 @@
 import torch.nn as nn
 import torch.nn.init as init
 from torchvision import models
-from torchvision.models.vgg import model_urls
 
 def init_weights(modules):
     for m in modules:
@@ -22,7 +21,6 @@ def init_weights(modules):
 class vgg16_bn(torch.nn.Module):
     def __init__(self, pretrained=True, freeze=True):
         super(vgg16_bn, self).__init__()
-        model_urls['vgg16_bn'] = model_urls['vgg16_bn'].replace('https://', 'http://')
         vgg_pretrained_features = models.vgg16_bn(pretrained=pretrained).features
         self.slice1 = torch.nn.Sequential()
         self.slice2 = torch.nn.Sequential()