diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-18 15:19:23 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-18 15:19:23 +0200 |
| commit | 430c503e3429070355c4de346347cf8b6fec4282 (patch) | |
| tree | 4c6b0ed75379306a688902f4b8ebd1c0f4e33d46 | |
| parent | a52d2c4cb29a8549e0a63fe06192151ac9212e62 (diff) | |
mogrifico!!!!
| -rw-r--r-- | live-mogrify.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index cdd9a72..a0a4015 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -116,8 +116,11 @@ def process_live_input(opt, data_opt, rpc_client): 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): + print("Load checkpoint: {}".format(checkpoint_fn_path)) model.save_dir = checkpoint_path model.load_network(model.netG, 'G', data_opt.epoch) + else: + print("Checkpoint not found: {}".format(checkpoint_fn_path)) data_opt.load_checkpoint = False if data_opt.load_sequence is True: data_opt.load_sequence = False |