diff --git a/demo/__init__.py b/demo/__init__.py deleted file mode 100644 index e69de29bb2..0000000000 diff --git a/docs/source/examples/basic_trainer.py b/docs/source/examples/basic_trainer.py index 95a47dd606..8e33674d60 100644 --- a/docs/source/examples/basic_trainer.py +++ b/docs/source/examples/basic_trainer.py @@ -2,9 +2,9 @@ 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 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 diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py index f7197a484a..16d5a47579 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -1,6 +1,6 @@ import torch.nn as nn import numpy as np -from pytorch_lightning.root_module.root_module import RootModule +from pytorch-lightning.root_module.root_module import RootModule from test_tube import HyperOptArgumentParser from torchvision.datasets import MNIST import torchvision.transforms as transforms diff --git a/docs/source/examples/fully_featured_trainer.py b/docs/source/examples/fully_featured_trainer.py index f971f28300..2412a1d0ed 100644 --- a/docs/source/examples/fully_featured_trainer.py +++ b/docs/source/examples/fully_featured_trainer.py @@ -5,10 +5,10 @@ from time import sleep import torch from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster -from pytorch_lightning.models.trainer import Trainer -from pytorch_lightning.utils.arg_parse import add_default_args +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 pytorch-lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint SEED = 2334 torch.manual_seed(SEED) @@ -27,7 +27,7 @@ AVAILABLE_MODELS = { """ Allows training by using command line arguments -Run by: +Run by: # TYPE YOUR RUN COMMAND HERE """ @@ -199,4 +199,3 @@ if __name__ == '__main__': nb_trials=hyperparams.nb_hopt_trials, nb_workers=len(gpu_ids) ) -