(self, x, relative_pos_enc=None)
| 496 | return x |
| 497 | |
| 498 | def forward(self, x, relative_pos_enc=None): |
| 499 | if self.grad_checkpoint and x.requires_grad: |
| 500 | x = checkpoint.checkpoint(self._forward_impl, x, relative_pos_enc) |
| 501 | else: |
| 502 | x = self._forward_impl(x, relative_pos_enc) |
| 503 | return x |
| 504 | |
| 505 | class RetBlock(nn.Module): |
| 506 |
nothing calls this directly
no test coverage detected