mogrifico!!!!

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