diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-05-17 17:19:20 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-05-17 17:19:20 +0800 |
| commit | 2b5bc2350df8ebb8e4e5722f5563f2e44823ac34 (patch) | |
| tree | 94816e528ed7ac1732c44877181540fd5a8ad810 /models/model.py | |
| parent | fde8d8fc0e4d986be37b26b468b0ac9d53d7bcd9 (diff) | |
Reduce channels in the auto-encoder and add more layers
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py index 6118bdf..bf23428 100644 --- a/models/model.py +++ b/models/model.py @@ -366,7 +366,7 @@ class Model: for _f_p_c1_t2, _f_p_c2_t2 in zip(*f_loss[2]) ]).sum() - return xrecon_loss, cano_cons_loss * 10, pose_sim_loss * 100 + return xrecon_loss / 10, cano_cons_loss, pose_sim_loss * 10 def _classification_loss(self, embedding, y): # Duplicate labels for each part |