From a3b91f12d247b4470f63546f15181789b03a2a58 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 03:37:15 +0200 Subject: augment paths --- live-mogrify.py | 2 +- models/base_model.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/live-mogrify.py b/live-mogrify.py index 5b6adc7..2d8989a 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -263,7 +263,7 @@ def process_live_input(opt, data_opt, rpc_client, model): if i >= opt.how_many: break if data_opt.load_checkpoint is True: - model.save_dir = os.path.join(opt.checkpoints_dir, data_opt.checkpoint_name) + model.save_dir = os.path.join(opt.checkpoints_dir, opt.module_name, data_opt.checkpoint_name) model.load_network(model.netG, 'G', data_opt.epoch) data_opt.load_checkpoint = False if data_opt.load_sequence is True: diff --git a/models/base_model.py b/models/base_model.py index d3d07d4..4e7122a 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -12,6 +12,7 @@ class BaseModel(): self.isTrain = opt.isTrain self.Tensor = torch.cuda.FloatTensor if self.gpu_ids else torch.Tensor self.save_dir = os.path.join(opt.checkpoints_dir, opt.module_name, opt.name) + print(self.save_dir) def set_input(self, input): self.input = input -- cgit v1.2.3-70-g09d2