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 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) |