From ed91ad6407378257e5bfbd18d50179991cf34713 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 10:41:05 -0500 Subject: [PATCH] working on trainer docs --- pytorch_lightning/trainer/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 7bbada8c..ed8b935d 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -839,7 +839,7 @@ class Trainer(TrainerIOMixin, Example:: - # Option 1: + # Option 1 # run test after fitting trainer = Trainer() model = LightningModule() @@ -847,8 +847,8 @@ class Trainer(TrainerIOMixin, trainer.fit() trainer.test() - # Option 2: - # run test from a model loaded + # Option 2 + # run test from a loaded model model = LightningModule.load_from_checkpoint('path/to/checkpoint.ckpt') trainer = Trainer()