From b79b011d5e7fd9e13ff376c80b495232a8de87de Mon Sep 17 00:00:00 2001 From: Shreyas Bapat Date: Wed, 3 Apr 2019 22:31:22 +0530 Subject: [PATCH] Some more fixes --- demo/__init__.py | 0 docs/source/examples/basic_trainer.py | 6 +++--- docs/source/examples/example_model.py | 2 +- docs/source/examples/fully_featured_trainer.py | 9 ++++----- 4 files changed, 8 insertions(+), 9 deletions(-) delete mode 100644 demo/__init__.py 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) ) -