From 0bd5d3f994b5e644cce848eabb9cb04cfb148d13 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 29 Sep 2022 17:52:16 +0530 Subject: [PATCH] fix log metric --- 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 64bf201..b030b23 100644 --- a/enhancer/models/model.py +++ b/enhancer/models/model.py @@ -83,7 +83,7 @@ class Model(pl.LightningModule): loss = self.loss(prediction, target) if self.logger: - self.logger.experiment.log_metric("train_loss",loss.item(), step=self.global_step) + self.logger.experiment.log_metric(key="train_loss",value=loss.item(), step=self.global_step) return {"loss":loss} @@ -95,7 +95,7 @@ class Model(pl.LightningModule): loss = self.metric(prediction, target) if self.logger: - self.logger.experiment.log_metric("val_loss",loss.item(), step=self.global_step) + self.logger.experiment.log_metric(key="val_loss",value=loss.item(), step=self.global_step) return {"loss":loss}