From 71113ca770318547eed4e0d720950ba16f085662 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 31 Mar 2019 16:46:00 -0400 Subject: [PATCH] Update README.md --- README.md | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 6a4dcab54d..ec9ede3d24 100644 --- a/README.md +++ b/README.md @@ -46,9 +46,7 @@ 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). -### Example: - -#### Define the trainer +#### Basic trainer example ```python import os @@ -112,7 +110,8 @@ if __name__ == '__main__': ``` -#### Define the model +#### Basic model example +Here we only show the method signatures. It's up to you to define the content. ```python from torch import nn