diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:29:02 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:29:02 +0800 |
| commit | 966d4431c037b0c4641aa2a5fc22f05be064b331 (patch) | |
| tree | 0239ba89d31857a7f086acf627fc1bbf167855a9 /models/auto_encoder.py | |
| parent | 7825f978f198e56958703f0d08f7ccbd8cef49ca (diff) | |
| parent | 36cf502afe9b93efe31c244030270b0a62e644b8 (diff) | |
Merge branch 'master' into python3.8
# Conflicts:
# models/model.py
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 1e7c323..64c52e3 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -130,7 +130,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) |