diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index a32c41f..7b245d8 100644 --- 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"