1 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py
index cc5887e..ceadb92 100644
--- a/models/model.py
+++ b/models/model.py
@@ -344,8 +344,8 @@ class Model:
         )
         losses = torch.stack((
             *ae_losses,
-            torch.log(hpm_result.pop('loss').mean() + 1),
-            torch.log(pn_result.pop('loss').mean() + 1)
+            hpm_result.pop('loss').mean(),
+            pn_result.pop('loss').mean()
         ))
         return losses, hpm_result, pn_result