Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix log metric

This commit is contained in:
shahules786 2022-09-29 17:52:16 +05:30
parent 79525df76e
commit 0bd5d3f994
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/models/model.py
View File

@ -83,7 +83,7 @@ class Model(pl.LightningModule):
loss = self.loss(prediction, target) loss = self.loss(prediction, target)
if self.logger: 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} return {"loss":loss}
@ -95,7 +95,7 @@ class Model(pl.LightningModule):
loss = self.metric(prediction, target) loss = self.metric(prediction, target)
if self.logger: 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} return {"loss":loss}