Fixed typos in README (#1219)

This commit is contained in:
Rishabh Anand
2020-03-24 14:48:31 -04:00
committed by GitHub
parent 4c2026bf9a
commit e880e29f2b
+2 -2
View File
@@ -137,7 +137,7 @@ In summary, you:
class LitSystem(pl.LightningModule):
def __init__(self):
super(CoolSystem, self).__init__()
super(LitSystem, self).__init__()
# not the best model...
self.l1 = torch.nn.Linear(28 * 28, 10)
@@ -152,7 +152,7 @@ In summary, you:
```python
from pytorch_lightning import Trainer
model = CoolSystem()
model = LitSystem()
# most basic trainer, uses good defaults
trainer = Trainer()