(self,
feat,
out_enc,
label=None,
valid_ratios=None,
word_positions=None,
train_mode=True)
| 86 | raise NotImplementedError |
| 87 | |
| 88 | def forward(self, |
| 89 | feat, |
| 90 | out_enc, |
| 91 | label=None, |
| 92 | valid_ratios=None, |
| 93 | word_positions=None, |
| 94 | train_mode=True): |
| 95 | self.train_mode = train_mode |
| 96 | |
| 97 | if train_mode: |
| 98 | return self.forward_train(feat, out_enc, label, valid_ratios, |
| 99 | word_positions) |
| 100 | return self.forward_test(feat, out_enc, valid_ratios, word_positions) |
| 101 | |
| 102 | |
| 103 | class ChannelReductionEncoder(nn.Module): |
nothing calls this directly
no test coverage detected