From 3014a415011bef9985a7867da54e920c442e80ef Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sun, 16 Oct 2022 12:22:46 +0530 Subject: [PATCH] load best model to test --- 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 08f4d3e..398fa2b 100644 --- a/enhancer/cli/train.py +++ b/enhancer/cli/train.py @@ -76,7 +76,7 @@ def main(config: DictConfig): trainer = instantiate(config.trainer, logger=logger, callbacks=callbacks) trainer.fit(model) - trainer.test(model) + trainer.test(ckpt_path="best") logger.experiment.log_artifact( logger.run_id, f"{trainer.default_root_dir}/config_log.yaml"