diff --git a/README.md b/README.md index 832e42bd74..01a156c358 100644 --- a/README.md +++ b/README.md @@ -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: