comment out validation_plots

jpata · Jan 15, 2025 · f04675e · f04675e
1 parent 1a5d245
commit f04675e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlpf/model/training.py b/mlpf/model/training.py
@@ -257,8 +257,8 @@ def eval_epoch(
         )
 
         # Save validation plots for first batch
-        if (rank == 0 or rank == "cpu") and ival == 0:
-            validation_plots(batch, ypred_raw, ytarget, ypred, tensorboard_writer, epoch, outdir)
+        # if (rank == 0 or rank == "cpu") and ival == 0:
+        #     validation_plots(batch, ypred_raw, ytarget, ypred, tensorboard_writer, epoch, outdir)
 
         # Accumulate losses
         for loss_name in loss: