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