fix: fix logging message bug #16

twndus · May 23, 2024 · 56975f7 · 56975f7
1 parent c65e77a
commit 56975f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trainers/cdae_trainer.py b/trainers/cdae_trainer.py
@@ -29,8 +29,8 @@ def _loss(self):
         elif self.cfg.loss_name.lower() == 'bce' and not self.cfg.negative_sampling:
             return nn.BCELoss()
         else:
-            logger.error(f"Loss Not Exists: {loss_name} when negative_sampling == {self.cfg.negative_sampling}")
-            raise NotImplementedError(f"Loss Not Exists: {loss_name}")
+            logger.error(f"Loss Not Exists: {self.cfg.loss_name} when negative_sampling == {self.cfg.negative_sampling}")
+            raise NotImplementedError(f"Loss Not Exists: {self.cfg.loss_name}")
 
 
     def train(self, train_dataloader: DataLoader) -> float: