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

load best model to test

This commit is contained in:
shahules786 2022-10-16 12:22:46 +05:30
parent dab68de260
commit 3014a41501
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

@ -76,7 +76,7 @@ def main(config: DictConfig):
trainer = instantiate(config.trainer, logger=logger, callbacks=callbacks) trainer = instantiate(config.trainer, logger=logger, callbacks=callbacks)
trainer.fit(model) trainer.fit(model)
trainer.test(model) trainer.test(ckpt_path="best")
logger.experiment.log_artifact( logger.experiment.log_artifact(
logger.run_id, f"{trainer.default_root_dir}/config_log.yaml" logger.run_id, f"{trainer.default_root_dir}/config_log.yaml"