Update README.md

This commit is contained in:
William Falcon
2019-03-30 21:48:04 -04:00
committed by GitHub
parent ded703fb4d
commit dc17445c77
+2 -2
View File
@@ -8,7 +8,7 @@ To use lightning do 2 things:
### Example:
### Define the trainer
#### Define the trainer
```python
# trainer.py
@@ -60,7 +60,7 @@ model = My_Model()
trainer.fit(model)
```
### Define the model
#### Define the model
```python
import torch.nn as nn