From a57a7ff0fcb853eb9072c5f775f888be262b3d76 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Tue, 11 Oct 2022 21:30:13 +0530
Subject: [PATCH] print metrics report

---
 enhancer/models/model.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 04f79a8..3f74a74 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -230,13 +230,10 @@ class Model(pl.LightningModule):
         for metric in test_mean_metrics.keys():
             test_mean_metrics[metric] /= len(outputs)
 
-        for k, v in test_mean_metrics.items():
-            self.logger.experiment.log_metric(
-                run_id=self.logger.run_id,
-                key=k,
-                value=v,
-                step=self.current_epoch,
-            )
+        print("----------TEST REPORT----------\n")
+        for metric in test_mean_metrics.keys():
+            print(f"|{metric.upper()} | {test_mean_metrics[metric]} |")
+        print("--------------------------------")
 
     def on_save_checkpoint(self, checkpoint):