diff options
Diffstat (limited to 'live-mogrify.py')
| -rw-r--r-- | live-mogrify.py | 2 |
1 files changed, 1 insertions, 1 deletions
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: |