(self, x, data=None)
| 213 | return {'vis_pos_embed', 'pre_token', 'next_token', 'char_embed'} |
| 214 | |
| 215 | def forward(self, x, data=None): |
| 216 | if self.training: |
| 217 | return self.forward_train(x, data) |
| 218 | else: |
| 219 | if self.infer_aug: |
| 220 | if self.bi_attn: |
| 221 | return self.forward_test_bi_attn(x) |
| 222 | return self.forward_test_bi(x) |
| 223 | return self.forward_test(x) |
| 224 | |
| 225 | def forward_test_bi(self, x): |
| 226 | # self.attn_maps = [] |
nothing calls this directly
no test coverage detected