From 6fb0aeae7fe76aa2b81a21738c348196d35012c3 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 12 Oct 2022 17:56:50 +0530 Subject: [PATCH] name loss wraper --- enhancer/loss.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/enhancer/loss.py b/enhancer/loss.py index fd1a609..cdd15a5 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -151,9 +151,11 @@ class LossWrapper(nn.Module): ) self.higher_better = direction[0] + self.name = "" for loss in losses: loss = self.validate_loss(loss) self.valid_losses.append(loss()) + self.name += f"{loss().name}_" def validate_loss(self, loss: str): if loss not in LOSS_MAP.keys():