From 62f14a6ef0d902b9ffd4e57427a40663e2e5c2ad Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sat, 9 Jan 2021 21:02:34 +0800 Subject: Change auto-encoder input in evaluation --- 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 7c1f7ef..5e7558b 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -128,7 +128,7 @@ class AutoEncoder(nn.Module): BasicLinear(f_c_dim, num_class) ) - def forward(self, x_c1_t1, x_c1_t2, x_c2_t2, y=None): + def forward(self, x_c1_t2, x_c1_t1=None, x_c2_t2=None, y=None): # x_c1_t2 is the frame for later module (f_a_c1_t2, f_c_c1_t2, f_p_c1_t2) = self.encoder(x_c1_t2) -- cgit v1.2.3