Update README.md

2026-09-09 11:32:07 +08:00 · 2020-05-12 08:59:23 -04:00
parent 087bb34c68
commit 98cb7c2ce2
1 changed files with 1 additions and 1 deletions
@@ -62,7 +62,7 @@ Here's a minimal example without a validation or test loop.
 class LitClassifier(pl.LightningModule):

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

    def forward(self, x):