cc

baselines/cyclegan-cut/train.py

@@ -34,7 +34,9 @@ if __name__ == "__main__":
         visualizer.reset()  # reset the visualizer: make sure it saves the results to HTML at least once every epoch
 
         dataset.set_epoch(epoch)
-        for i, data in tqdm(enumerate(dataset)):  # inner loop within one epoch
+        for i, data in tqdm(
+            enumerate(dataset), total=len(dataset)
+        ):  # inner loop within one epoch
             iter_start_time = time.time()  # timer for computation per iteration
             if total_iters % opt.print_freq == 0:
                 t_data = iter_start_time - iter_data_time