diff --git a/model.py b/model.py index 9fafc562..efe3971c 100644 --- a/model.py +++ b/model.py @@ -15,8 +15,8 @@ def __init__(self, load_weights=False): if not load_weights: mod = models.vgg16(pretrained = True) self._initialize_weights() - for i in xrange(len(self.frontend.state_dict().items())): - self.frontend.state_dict().items()[i][1].data[:] = mod.state_dict().items()[i][1].data[:] + for i in range(len(self.frontend.state_dict().items())): + list(self.frontend.state_dict().items())[i][1].data[:] = list(mod.state_dict().items())[i][1].data[:] def forward(self,x): x = self.frontend(x) x = self.backend(x) @@ -49,4 +49,4 @@ def make_layers(cfg, in_channels = 3,batch_norm=False,dilation = False): else: layers += [conv2d, nn.ReLU(inplace=True)] in_channels = v - return nn.Sequential(*layers) \ No newline at end of file + return nn.Sequential(*layers)