diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-29 21:56:59 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-29 21:56:59 +0800 |
| commit | 6cf6f7fb0e43b437c80f134c3eb1a1c5afffdba9 (patch) | |
| tree | 95748afaa51a49c2e774e48afc55d048c92c75eb /models/auto_encoder.py | |
| parent | 92f455bd31231675797798f2e77e1e1a588528f3 (diff) | |
Return canonical features at condition 1 for later aggregation
Diffstat (limited to 'models/auto_encoder.py')
| -rw-r--r-- | models/auto_encoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
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) |