diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:59 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-03 19:45:59 +0800 |
| commit | fbc13e2e0a9138771ea41a5600753aa33e34eec0 (patch) | |
| tree | 1769491b1f54a7509663a7793dea99db2cb748bf /models | |
| parent | 30425d5fa8bdb58460adaaaaa2aae02b0d488812 (diff) | |
| parent | 8630dbfe3b2f720ee7908899d649ba37333a2ac9 (diff) | |
Merge branch 'python3.8' into python3.7
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 dbbc557..78a9c0f 100644 --- a/models/model.py +++ b/models/model.py @@ -341,8 +341,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 |