(self, q, x, mask=None)
| 69 | self.cls = nn.Linear(d_model, out_channels) |
| 70 | |
| 71 | def forward_decoder(self, q, x, mask=None): |
| 72 | for decoder_layer in self.decoder: |
| 73 | q = decoder_layer(q, x, cross_mask=mask) |
| 74 | output = q # (N, T, E) |
| 75 | logits = self.cls(output) # (N, T, C) |
| 76 | return logits |
| 77 | |
| 78 | def forward(self, img_feat, data=None): |
| 79 | """ |