Method forward

monai/networks/nets/unet.py:297–299

(self, x: torch.Tensor)

        return conv

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        x = self.model(x)
        return x


class CheckpointUNet(UNet):