fix checkpoint carpedm20#383

ashvithashetty · May 16, 2021 · 3a9b021 · 3a9b021
1 parent 69deb14
commit 3a9b021
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -52,12 +52,19 @@ def main(_):
   if FLAGS.output_height is None: FLAGS.output_height = FLAGS.input_height
   if FLAGS.input_width is None: FLAGS.input_width = FLAGS.input_height
   if FLAGS.output_width is None: FLAGS.output_width = FLAGS.output_height
-
+    
   # output folders
   if FLAGS.out_name == "":
-      FLAGS.out_name = '{} - {} - {}'.format(timestamp(), FLAGS.data_dir.split('/')[-1], FLAGS.dataset) # penultimate folder of path
-      if FLAGS.train:
-        FLAGS.out_name += ' - x{}.z{}.{}.y{}.b{}'.format(FLAGS.input_width, FLAGS.z_dim, FLAGS.z_dist, FLAGS.output_width, FLAGS.batch_size)
+      #FLAGS.out_name = '{} - {} - {}'.format(timestamp(), FLAGS.data_dir.split('/')[-1], FLAGS.dataset) # penultimate folder of path
+      FLAGS.out_name = '{} - {}'.format( FLAGS.data_dir.split('/')[-1],FLAGS.dataset)  # penultimate folder of path
+#     if FLAGS.train:
+#        FLAGS.out_name += ' - x{}.z{}.{}.y{}.b{}'.format(FLAGS.input_width, FLAGS.z_dim, FLAGS.z_dist, FLAGS.output_width, FLAGS.batch_size)
+      FLAGS.out_name += ' - x{}.z{}.{}.y{}.b{}'.format(FLAGS.input_width, FLAGS.z_dim, FLAGS.z_dist, FLAGS.output_width, FLAGS.batch_size)
+#   # output folders
+#   if FLAGS.out_name == "":
+#       FLAGS.out_name = '{} - {} - {}'.format(timestamp(), FLAGS.data_dir.split('/')[-1], FLAGS.dataset) # penultimate folder of path
+#       if FLAGS.train:
+#         FLAGS.out_name += ' - x{}.z{}.{}.y{}.b{}'.format(FLAGS.input_width, FLAGS.z_dim, FLAGS.z_dist, FLAGS.output_width, FLAGS.batch_size)
 
   FLAGS.out_dir = os.path.join(FLAGS.out_dir, FLAGS.out_name)
   FLAGS.checkpoint_dir = os.path.join(FLAGS.out_dir, FLAGS.checkpoint_dir)