add model testing

2022-10-10 12:38:51 +05:30 · 2022-10-10 12:38:51 +05:30 · 37fe86063d
parent 43a22d2432
commit 37fe86063d
1 changed files with 1 additions and 0 deletions
--- a/enhancer/cli/train.py
+++ b/enhancer/cli/train.py
@ -75,6 +75,7 @@ def main(config: DictConfig):

    trainer = instantiate(config.trainer, logger=logger, callbacks=callbacks)
    trainer.fit(model)
+    trainer.test(model)

    logger.experiment.log_artifact(
        logger.run_id, f"{trainer.default_root_dir}/config_log.yaml"