Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -8,7 +8,7 @@ To use lightning do 2 things:
### Example: ### Example:
### Define the trainer #### Define the trainer
```python ```python
# trainer.py # trainer.py
@ -60,7 +60,7 @@ model = My_Model()
trainer.fit(model) trainer.fit(model)
``` ```
### Define the model #### Define the model
```python ```python
import torch.nn as nn import torch.nn as nn