Forward function.
(self, x)
| 77 | self.convs = nn.Sequential(*convs) |
| 78 | |
| 79 | def forward(self, x): |
| 80 | """Forward function.""" |
| 81 | |
| 82 | if self.with_cp and x.requires_grad: |
| 83 | out = cp.checkpoint(self.convs, x) |
| 84 | else: |
| 85 | out = self.convs(x) |
| 86 | return out |
| 87 | |
| 88 | |
| 89 | @UPSAMPLE_LAYERS.register_module() |
nothing calls this directly
no outgoing calls
no test coverage detected