Update README.md

2026-09-09 11:32:07 +08:00 · 2019-08-13 12:56:12 -04:00
parent 1cd5dde164
commit bb75cec076
1 changed files with 1 additions and 1 deletions
@@ -77,7 +77,7 @@ import pytorch_lightning as pl
 class CoolSystem(pl.LightningModule):

    def __init__(self):
-        super(CoolModel, self).__init__()
+        super(CoolSystem, self).__init__()
        # not the best model...
        self.l1 = torch.nn.Linear(28 * 28, 10)