testy

1 files changed, 4 insertions, 2 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index f7c1e50..2c636d9 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -29,13 +29,15 @@ if __name__ == '__main__':
   opt.no_flip = True  # no flip
   # opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0]
 
-  if opt.tag == '':
+  if data_opt.tag == '':
     d = datetime.now()
-    tag = opt.tag = "{}_{}_{}".format(
+    tag = data_opt.tag = "{}_{}_{}".format(
       opt.name,
       opt.experiment,
       d.strftime('%Y%m%d%H%M')
     )
+  else:
+    tag = data_opt.tag
 
   opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + tag + "/"
   A_offset = 0