diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-18 15:17:32 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-18 15:17:32 +0200 |
| commit | a52d2c4cb29a8549e0a63fe06192151ac9212e62 (patch) | |
| tree | a225274ed03f9005903d75574bfbcd61057a1d9f | |
| parent | 83ae51c717f1c7f3c357ab8ac18bd5aa90c9b263 (diff) | |
mogrifico!!!!
| -rw-r--r-- | live-mogrify.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index 360cf2e..cdd9a72 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -112,8 +112,12 @@ def process_live_input(opt, data_opt, rpc_client): break if data_opt.load_checkpoint is True: - 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) + checkpoint_fn = "%s_net_%s.pth".format(data_opt.epoch, 'G') + checkpoint_path = os.path.join(opt.checkpoints_dir, opt.module_name, data_opt.checkpoint_name) + checkpoint_fn_path = os.path.join(checkpoint_path, checkpoint_fn) + if os.path.exists(checkpoint_fn_path): + model.save_dir = checkpoint_path + model.load_network(model.netG, 'G', data_opt.epoch) data_opt.load_checkpoint = False if data_opt.load_sequence is True: data_opt.load_sequence = False |