From bc82188377a8dc3068cd7b2ba5f02ed8bd4f2a54 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 7 Apr 2021 15:46:34 +0800 Subject: Revert cross-reconstruction loss factor and make image log steps adjustable --- models/auto_encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/auto_encoder.py') 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 -- cgit v1.2.3