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

name loss wraper

This commit is contained in:
shahules786 2022-10-12 17:56:50 +05:30
parent 1831dc2013
commit 6fb0aeae7f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/loss.py
View File

@ -151,9 +151,11 @@ class LossWrapper(nn.Module):
) )
self.higher_better = direction[0] self.higher_better = direction[0]
self.name = ""
for loss in losses: for loss in losses:
loss = self.validate_loss(loss) loss = self.validate_loss(loss)
self.valid_losses.append(loss()) self.valid_losses.append(loss())
self.name += f"{loss().name}_"
def validate_loss(self, loss: str): def validate_loss(self, loss: str):
if loss not in LOSS_MAP.keys(): if loss not in LOSS_MAP.keys():