summaryrefslogtreecommitdiff
path: root/live-mogrify.py
diff options
context:
space:
mode:
Diffstat (limited to 'live-mogrify.py')
-rw-r--r--live-mogrify.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/live-mogrify.py b/live-mogrify.py
index 480c35d..bf3975a 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -160,6 +160,7 @@ class Listener():
self.data_opt = data_opt
self.data_opt_parser = data_opt_parser.parser
self.model = create_model(opt)
+ self.data_opt.load_checkpoint = True
self.working = False
def _set_fn(self, key, value):
if hasattr(self.data_opt, key):