diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-07 15:46:34 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-04-07 15:46:34 +0800 |
| commit | bc82188377a8dc3068cd7b2ba5f02ed8bd4f2a54 (patch) | |
| tree | c3265640aa0e37fe3c7a890ac5c415e32c287c77 /models/auto_encoder.py | |
| parent | 5e8947fbc90e1d67dadae36d32330a280d057267 (diff) | |
Revert cross-reconstruction loss factor and make image log steps adjustable
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 49cb78b..96dfdb3 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -152,7 +152,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 |