diff options
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 |