Update: This is a method to display the results of each model during …

…training and can be commented out.
chairc · Sep 22, 2023 · 3443662 · 3443662
1 parent 4b43d32
commit 3443662
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/train.py b/tools/train.py
@@ -281,7 +281,8 @@ def train(rank=None, args=None):
                     sampled_images = diffusion.sample(model=model, n=n, labels=labels, cfg_scale=cfg_scale)
                     ema_sampled_images = diffusion.sample(model=ema_model, n=n, labels=labels,
                                                           cfg_scale=cfg_scale)
-                    plot_images(images=sampled_images)
+                    # This is a method to display the results of each model during training and can be commented out
+                    # plot_images(images=sampled_images)
                     save_images(images=sampled_images, path=os.path.join(results_vis_dir, f"{save_name}.jpg"))
                     save_images(images=ema_sampled_images, path=os.path.join(results_vis_dir, f"{save_name}_ema.jpg"))
                 if save_model_interval and epoch > start_model_interval: