working on trainer docs

2026-09-09 11:32:07 +08:00 · 2020-01-16 10:41:05 -05:00
parent 62d213ab3c
commit ed91ad6407
1 changed files with 3 additions and 3 deletions
@@ -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()