From 8e2e95e55dfcde78ca6c04375131d0075f14f156 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 31 Mar 2019 16:47:15 -0400 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index ec9ede3d24..620d8a7bb0 100644 --- a/README.md +++ b/README.md @@ -47,6 +47,7 @@ To use lightning do 2 things: 2. [Define a model](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/sample_model_template/model_template.py). #### Basic trainer example +See [this demo](https://github.com/williamFalcon/pytorch-lightning/blob/master/demo/fully_featured_trainer.py) for a more robust trainer example. ```python import os @@ -186,6 +187,8 @@ class My_Model(RootModule): parser.add_argument('--out_features', default=20) return parser ``` + + ### Details #### Model definition