From 01857472f6cf5f22433f12bbd82b5e5095ee830e Mon Sep 17 00:00:00 2001 From: Ziad-Usama Date: Thu, 17 Nov 2022 01:14:09 +0200 Subject: [PATCH] fixed the hardcoded path --- models/model_adgen.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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():