From 0ae8be853d7744b1b8b85d9366f4a1e0c0493738 Mon Sep 17 00:00:00 2001 From: emizzz Date: Thu, 8 Oct 2020 19:26:32 +0200 Subject: [PATCH] fix hardcoded output path --- lib/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/model.py b/lib/model.py index 5c0b9dd..08e6a29 100644 --- a/lib/model.py +++ b/lib/model.py @@ -184,7 +184,7 @@ def test(self): with torch.no_grad(): # Load the weights of netg and netd. if self.opt.load_weights: - path = "./output/{}/{}/train/weights/netG.pth".format(self.name.lower(), self.opt.dataset) + path = "{}/{}/{}/train/weights/netG.pth".format(self.opt.outf, self.name.lower(), self.opt.dataset) pretrained_dict = torch.load(path)['state_dict'] try: