(self, input, motion_mask=None, condition=None)
| 229 | self.final = nn.Linear(self.latent_dim, self.input_feats) |
| 230 | |
| 231 | def forward(self, input, motion_mask=None, condition=None): |
| 232 | B = input.shape[0] |
| 233 | T = self.max_seq_len |
| 234 | input = self.linear(input) |
| 235 | if self.use_condition: |
| 236 | if self.num_class is None: |
| 237 | condition = self.condition_bias(condition) |
| 238 | else: |
| 239 | condition = self.condition_bias[condition.long()].squeeze(1) |
| 240 | input = input + condition |
| 241 | query = self.pos_encoder.pe[:T, :].view(T, 1, -1).repeat(1, B, 1) |
| 242 | input = input.view(1, B, -1) |
| 243 | feature = self.seqTransDecoder(tgt=query, memory=input, tgt_key_padding_mask=motion_mask) |
| 244 | pose = self.final(feature).permute(1, 0, 2).contiguous() |
| 245 | return pose |
nothing calls this directly
no outgoing calls
no test coverage detected