From 37fe86063d52b582e162d5cff92edbdc282a1991 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 12:38:51 +0530
Subject: [PATCH] add model testing

---
 enhancer/cli/train.py | 1 +
 1 file changed, 1 insertion(+)

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"