diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-09 21:02:34 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-09 21:02:34 +0800 |
| commit | 62f14a6ef0d902b9ffd4e57427a40663e2e5c2ad (patch) | |
| tree | adaa0151474197f21e6f53bf9fed7a0ea149f311 /models/auto_encoder.py | |
| parent | 6f278a962d70e90ac530f5723e198c7c356e8297 (diff) | |
Change auto-encoder input in evaluation
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) |