diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-08 12:57:54 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-08 12:57:54 +0800 |
| commit | 543b163c1e8b3914be5c69fa746033da27952449 (patch) | |
| tree | 0daac884baf8fc70f49d4adce0fdec21660151a5 /models/auto_encoder.py | |
| parent | b54596ab5ce41110100214cd76ff50acc65b589a (diff) | |
| parent | a66e5a456a2b036b2b787371da3160efc559031f (diff) | |
Merge branch 'master' into python3.8
Diffstat (limited to 'models/auto_encoder.py')
| -rw-r--r-- | models/auto_encoder.py | 2 |
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 |