Update README.md

This commit is contained in:
William Falcon 2019-11-23 11:12:45 -05:00 committed by GitHub
parent 48b797fdb0
commit bdebe18df6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -31,6 +31,8 @@ Simple installation from PyPI
pip install pytorch-lightning
```
[LIVE COLAB DEMO](https://colab.research.google.com/drive/1F_RNcHzTfFuQf-LeKvSlud6x7jXYkG31#scrollTo=HOk9c4_35FKg)
## Docs
**[View the docs here](https://williamfalcon.github.io/pytorch-lightning/)**
@ -68,8 +70,6 @@ Lightning sets up all the boilerplate state-of-the-art training for you so you c
## How do I do use it?
Think about Lightning as refactoring your research code instead of using a new framework. The research code goes into a [LightningModule](https://williamfalcon.github.io/pytorch-lightning/LightningModule/RequiredTrainerInterface/) which you fit using a Trainer.
[Try this live demo on colab!](https://colab.research.google.com/drive/1F_RNcHzTfFuQf-LeKvSlud6x7jXYkG31#scrollTo=HOk9c4_35FKg)
The LightningModule defines a *system* such as seq-2-seq, GAN, etc... It can ALSO define a simple classifier such as the example below.
To use lightning do 2 things: