From 5983b094701f2858c65f9d96ff6f0517b3c2fe14 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 29 Sep 2022 22:38:59 +0530 Subject: [PATCH] fix logger --- enhancer/models/model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/enhancer/models/model.py b/enhancer/models/model.py index b030b23..20c8196 100644 --- a/enhancer/models/model.py +++ b/enhancer/models/model.py @@ -83,7 +83,9 @@ class Model(pl.LightningModule): loss = self.loss(prediction, target) if self.logger: - self.logger.experiment.log_metric(key="train_loss",value=loss.item(), step=self.global_step) + self.logger.experiment.log_metric(run_id=self.logger.run_id, + key="train_loss", value=loss.item(), + step=self.global_step) return {"loss":loss} @@ -95,7 +97,9 @@ class Model(pl.LightningModule): loss = self.metric(prediction, target) if self.logger: - self.logger.experiment.log_metric(key="val_loss",value=loss.item(), step=self.global_step) + self.logger.experiment.log_metric(run_id=self.logger.run_id, + key="val_loss",value=loss.item(), + step=self.global_step) return {"loss":loss}