Method __init__

tests/test_pytorch_lightning.py:36–38

(self)

class LitClassifier(pl.LightningModule):

    def __init__(self):
        super().__init__()
        self.l1 = torch.nn.Linear(28 * 28, 10)

    def forward(self, x):
        return F.relu(self.l1(x.view(x.size(0), -1)))