summaryrefslogtreecommitdiff
path: root/models/auto_encoder.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-20 14:48:16 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-20 14:48:16 +0800
commit9b1828be1db7fd1be8731a7cec66162de9145285 (patch)
tree9efb5a37856f34e333457e9d7ab2aaa8ba811cf6 /models/auto_encoder.py
parente33c22e556ed64e1c1fdb011d78a124d1489ad15 (diff)
parentc538919cb69e35a46811aef0b23baefe6a4c499c (diff)
Merge branch 'python3.8' into data_parallel_py3.8
# Conflicts: # models/model.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 e17caed..6f388c2 100644
--- a/models/auto_encoder.py
+++ b/models/auto_encoder.py
@@ -173,7 +173,7 @@ class AutoEncoder(nn.Module):
return (
(f_a_c1_t2_, f_c_c1_t2_, f_p_c1_t2_),
- (xrecon_loss, cano_cons_loss, pose_sim_loss * 10)
+ torch.stack((xrecon_loss, cano_cons_loss, pose_sim_loss * 10))
)
else: # evaluating
return f_c_c1_t2_, f_p_c1_t2_