From 81c5f13ff6c9fa5d33017b880522b0e30d87f52e Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 15:32:37 +0530
Subject: [PATCH] log metric

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

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 07564cd..4f1d75b 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -181,11 +181,13 @@ class Model(pl.LightningModule):
             value = metric(target, prediction)
             metric_dict[metric.name] = value
 
-        self.logger.experiment.log_metrics(
-            run_id=self.logger.run_id,
-            metrics=metric_dict,
-            step=self.global_step,
-        )
+        for k, v in metric_dict.items():
+            self.logger.experiment.log_metric(
+                run_id=self.logger.run_id,
+                key=k,
+                value=v,
+                step=self.global_step,
+            )
 
         return metric_dict