Method __init__

samples/common.py:273–275

(self, model)

    class ClassifierEnd2End(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), dim=1)