From b9f35fbe7d78b3c478086ea26c2a76f72ce35687 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sat, 3 Apr 2021 19:45:00 +0800 Subject: Revert "Normalize triplet losses" This reverts commit 99d1b18a --- models/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'models') 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 -- cgit v1.2.3