From 983551653dfc0213d0210426a6d2cbf5f982e05d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 18:11:13 -0400 Subject: [PATCH] fixed basic trainer --- docs/source/examples/basic_trainer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/source/examples/basic_trainer.py b/docs/source/examples/basic_trainer.py index 8e33674d..308f24eb 100644 --- a/docs/source/examples/basic_trainer.py +++ b/docs/source/examples/basic_trainer.py @@ -2,10 +2,10 @@ import os import sys from test_tube import HyperOptArgumentParser, Experiment -from pytorch-lightning.models.trainer import Trainer -from pytorch-lightning.utils.arg_parse import add_default_args -from pytorch-lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint -from demo.example_model import ExampleModel +from pytorch_lightning.models.trainer import Trainer +from pytorch_lightning.utils.arg_parse import add_default_args +from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint +from docs.source.examples.example_model import ExampleModel def main(hparams):