From ded703fb4d59d16ee59310e3c19cdadeaadb9c27 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 30 Mar 2019 21:47:51 -0400 Subject: [PATCH] Update README.md --- README.md | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a4ec14a5..0cfb42d7 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,11 @@ The Keras for ML-researchers in PyTorch. ## Usage 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). +2. [Define a model](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/sample_model_template/model_template.py). + +### Example: + +### Define the trainer ```python # trainer.py @@ -56,7 +60,8 @@ model = My_Model() trainer.fit(model) ``` -next define a model that implements these 10 functions: +### Define the model + ```python import torch.nn as nn