Update README.md

This commit is contained in:
William Falcon 2019-03-31 16:47:15 -04:00 committed by GitHub
parent 71113ca770
commit 8e2e95e55d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -47,6 +47,7 @@ To use lightning do 2 things:
2. [Define a model](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/sample_model_template/model_template.py).
#### Basic trainer example
See [this demo](https://github.com/williamFalcon/pytorch-lightning/blob/master/demo/fully_featured_trainer.py) for a more robust trainer example.
```python
import os
@ -186,6 +187,8 @@ class My_Model(RootModule):
parser.add_argument('--out_features', default=20)
return parser
```
### Details
#### Model definition