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

train log rename

This commit is contained in:
shahules786 2022-10-12 17:57:42 +05:30
parent 226758e91f
commit ad56a160e4
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

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