summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-04-03 19:45:00 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-04-03 19:45:00 +0800
commitb9f35fbe7d78b3c478086ea26c2a76f72ce35687 (patch)
tree4528443563fd6f22407a430b88a396e690214180
parent99d1b18ac380bcf7d7579d80e09c9ddaecde44b3 (diff)
Revert "Normalize triplet losses"
This reverts commit 99d1b18a
-rw-r--r--models/model.py4
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