diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:45 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:45 +0800 |
| commit | 8630dbfe3b2f720ee7908899d649ba37333a2ac9 (patch) | |
| tree | 9ad619fcf1205b3ab438840f4ded1d1455e39ff3 /models | |
| parent | cc9055ed68f4839cf953ce9a7e74e4c87d2ed279 (diff) | |
| parent | b9f35fbe7d78b3c478086ea26c2a76f72ce35687 (diff) | |
Merge branch 'master' into python3.8
Diffstat (limited to 'models')
| -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 8e4398b..b2947a0 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 |