(self, x, **kwargs)
| 102 | return out |
| 103 | |
| 104 | def forward(self, x, **kwargs): |
| 105 | x_inp, x_labels = x[:, :-1], x[:, 1:] |
| 106 | logits = self.net(x_inp, **kwargs) |
| 107 | return F.cross_entropy(rearrange(logits, "b c n -> b n c"), x_labels) |
nothing calls this directly
no outgoing calls
no test coverage detected