Method __init__

samples/common.py:321–323

(self, model)

    class SegmentationEnd2End(torch.nn.Module):
        def __init__(self, model):
            super().__init__()
            self.model = model

        def forward(self, x: torch.Tensor) -> torch.Tensor:
            softmax = torch.nn.functional.softmax(self.model(x)["out"], dim=1)