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

fix logger

This commit is contained in:
shahules786 2022-09-29 22:38:59 +05:30
parent fccbd88ba2
commit 5983b09470
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
enhancer/models/model.py
View File

@ -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}