diff --git a/models/model_adgen.py b/models/model_adgen.py index 46b1e30..602ee8f 100644 --- a/models/model_adgen.py +++ b/models/model_adgen.py @@ -75,7 +75,8 @@ def __init__(self, n_downsample, input_dim, dim, style_dim, norm, activ, pad_typ super(VggStyleEncoder, self).__init__() # self.vgg = models.vgg19(pretrained=True).features vgg19 = models.vgg19(pretrained=False) - vgg19.load_state_dict(torch.load('/home1/menyf/data/deepfashion/vgg19-dcbb9e9d.pth')) + path = os.getcwd() + vgg19.load_state_dict(torch.load('{}/data/deepfashion/vgg19-dcbb9e9d.pth'.format(path))) self.vgg = vgg19.features for param in self.vgg.parameters():