From 71af821b8572cbce5e8a3d4d94db79a520100f58 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 26 Jun 2019 19:48:50 -0400 Subject: [PATCH] added docs page --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 4bab2881a6..cfca75bf7d 100644 --- a/README.md +++ b/README.md @@ -41,8 +41,10 @@ To use lightning do 2 things: from pytorch_lightning import Trainer from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint +# 1 - look at the this page for the interface (https://williamfalcon.github.io/pytorch-lightning/) model = MyModel() +# 2 - feed to trainer trainer = Trainer( checkpoint_callback=ModelCheckpoint(...), early_stop_callback=EarlyStopping(...),