Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename loss

This commit is contained in:
shahules786 2022-10-12 18:45:07 +05:30
parent cb0040b508
commit 2e58091543
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/cli/train.py
View File

@ -39,7 +39,7 @@ def main(config: DictConfig):
checkpoint = ModelCheckpoint( checkpoint = ModelCheckpoint(
dirpath="./model", dirpath="./model",
filename=f"model_{JOB_ID}", filename=f"model_{JOB_ID}",
monitor="val_loss", monitor="valid_loss",
verbose=False, verbose=False,
mode=direction, mode=direction,
every_n_epochs=1, every_n_epochs=1,