diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:00 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:00 +0800 |
| commit | b9f35fbe7d78b3c478086ea26c2a76f72ce35687 (patch) | |
| tree | 4528443563fd6f22407a430b88a396e690214180 /models/model.py | |
| parent | 99d1b18ac380bcf7d7579d80e09c9ddaecde44b3 (diff) | |
Revert "Normalize triplet losses"
This reverts commit 99d1b18a
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 4 |
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 |