Fixed bugs in training loop

Cryaaa · Sep 1, 2024 · ca44740 · ca44740
1 parent d049454
commit ca44740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/time-series/train_second_model_unet_encdec.py b/scripts/time-series/train_second_model_unet_encdec.py
@@ -106,7 +106,7 @@ def validate(
         checkpoint_dir, 
         metadata=None,
         tb_logger=None,
-        log_image_interval = 300,
+        log_image_interval = 10,
     ):
     model.eval()
     validation_loss = 0