Normalize triplet losses

1 files changed, 2 insertions, 2 deletions

diff --git a/models/model.py b/models/model.py
index ceadb92..cc5887e 100644
--- a/models/model.py
+++ b/models/model.py
@@ -344,8 +344,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