Fix ModelCheckpoint monitor bug

r-pad · May 24, 2024 · 3fca11d · 3fca11d
1 parent 2ba0dce
commit 3fca11d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train_residual_flow_multimodal.py b/scripts/train_residual_flow_multimodal.py
@@ -77,7 +77,7 @@ def setup_main(cfg):
                 ModelCheckpoint(
                     dirpath=cfg.lightning.checkpoint_dir,
                     filename="{epoch}-{step}-{train_loss:.2f}-weights-only",
-                    monitor="val_loss",
+                    monitor="val_loss/val_0",
                     mode="min",
                     save_weights_only=True,
                 ),