rename loss

2022-10-12 18:45:07 +05:30 · 2022-10-12 18:45:07 +05:30 · 2e58091543
parent cb0040b508
commit 2e58091543
1 changed files with 1 additions and 1 deletions
--- a/enhancer/cli/train.py
+++ b/enhancer/cli/train.py
@ -39,7 +39,7 @@ def main(config: DictConfig):
    checkpoint = ModelCheckpoint(
        dirpath="./model",
        filename=f"model_{JOB_ID}",
-        monitor="val_loss",
+        monitor="valid_loss",
        verbose=False,
        mode=direction,
        every_n_epochs=1,