From 2e58091543b6bb76d94811b208e67369d6ba21aa Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 12 Oct 2022 18:45:07 +0530 Subject: [PATCH] rename loss --- enhancer/cli/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index de48e64..49f7b3b 100644 --- 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,