From d62ca3e5f2ed5d2c03425007cc2ab91824a23bcc Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 30 Mar 2019 21:47:20 -0400 Subject: [PATCH] Update README.md --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 9d6e01eb..a4ec14a5 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,9 @@ The Keras for ML-researchers in PyTorch. ## Usage -To use lightning, first define a trainer function: +To use lightning do 2 things: +1. [Define a trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/trainer_main.py) (which will run ALL your models). +2. [Define a model](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/sample_model_template/model_template.py). ```python # trainer.py