From ad56a160e4074153602c81697a5061a4c3c09d40 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 12 Oct 2022 17:57:42 +0530
Subject: [PATCH] train log rename

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

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 8eb19a8..3ad5fa7 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -135,7 +135,7 @@ class Model(pl.LightningModule):
         loss = self.loss(prediction, target)
 
         self.log(
-            "train_loss",
+            f"train_{self.loss.name}",
             loss.item(),
             on_epoch=True,
             on_step=True,
@@ -175,7 +175,7 @@ class Model(pl.LightningModule):
 
         for metric in self.metric:
             value = metric(target, prediction)
-            metric_dict[metric.name] = value
+            metric_dict[f"test_{metric.name}"] = value
 
         self.log_dict(
             metric_dict,