From 6cf6f7fb0e43b437c80f134c3eb1a1c5afffdba9 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 29 Dec 2020 21:56:59 +0800 Subject: Return canonical features at condition 1 for later aggregation --- models/auto_encoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'models') diff --git a/models/auto_encoder.py b/models/auto_encoder.py index feec5e2..de38572 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -142,4 +142,5 @@ class AutoEncoder(nn.Module): + self.mse_loss(f_c_c1_t2, f_c_c2_t2) + self.xent_loss(y, y_)) - return xrecon_loss_t2, (f_p_c1_t2, f_p_c2_t2), cano_cons_loss_t2 + return (f_c_c1_t2, f_p_c1_t2, f_p_c2_t2, + xrecon_loss_t2, cano_cons_loss_t2) -- cgit v1.2.3