summaryrefslogtreecommitdiff
path: root/models/auto_encoder.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-04-08 12:58:18 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-04-08 12:58:18 +0800
commit4fda9dfcb55cf78dda1e942992354037c3e08456 (patch)
treef2c5f3565fdb21e7b86084e2a23704fd1a15c114 /models/auto_encoder.py
parentb8892d3a3838fe6f5e18c9d76f16ea6368d715f2 (diff)
parent543b163c1e8b3914be5c69fa746033da27952449 (diff)
Merge branch 'python3.8' into python3.7
# Conflicts: # utils/configuration.py
Diffstat (limited to 'models/auto_encoder.py')
-rw-r--r--models/auto_encoder.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/auto_encoder.py b/models/auto_encoder.py
index 1028767..d11ec99 100644
--- a/models/auto_encoder.py
+++ b/models/auto_encoder.py
@@ -154,7 +154,7 @@ class AutoEncoder(nn.Module):
return (
(f_a_c1_t2, f_c_c1_t2, f_p_c1_t2),
- (xrecon_loss / 10, cano_cons_loss, pose_sim_loss * 10)
+ (xrecon_loss, cano_cons_loss, pose_sim_loss * 10)
)
else: # evaluating
return f_a_c1_t2, f_c_c1_t2, f_p_c1_t2