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

rename loss

This commit is contained in:
shahules786 2022-10-12 18:44:22 +05:30
parent 00ef644179
commit cb0040b508
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
enhancer/models/model.py
View File

@ -135,7 +135,7 @@ class Model(pl.LightningModule):
loss = self.loss(prediction, target) loss = self.loss(prediction, target)
self.log( self.log(
f"train_{self.loss.name}", "train_loss",
loss.item(), loss.item(),
on_epoch=True, on_epoch=True,
on_step=True, on_step=True,
@ -152,6 +152,7 @@ class Model(pl.LightningModule):
target = batch["clean"] target = batch["clean"]
prediction = self(mixed_waveform) prediction = self(mixed_waveform)
metric_dict["valid_loss"] = self.loss(target, prediction).item()
for metric in self.metric: for metric in self.metric:
value = metric(target, prediction) value = metric(target, prediction)
metric_dict[f"valid_{metric.name}"] = value.item() metric_dict[f"valid_{metric.name}"] = value.item()