diff --git a/examples/new_project_templates/multi_node_cluster_template.py b/examples/new_project_templates/multi_node_cluster_template.py index cdbda003..c4af6d41 100644 --- a/examples/new_project_templates/multi_node_cluster_template.py +++ b/examples/new_project_templates/multi_node_cluster_template.py @@ -10,7 +10,7 @@ from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel SEED = 2334 torch.manual_seed(SEED) diff --git a/examples/new_project_templates/single_cpu_template.py b/examples/new_project_templates/single_cpu_template.py index 9822f216..c0f4826f 100644 --- a/examples/new_project_templates/single_cpu_template.py +++ b/examples/new_project_templates/single_cpu_template.py @@ -9,7 +9,7 @@ from test_tube import HyperOptArgumentParser, Experiment from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel SEED = 2334 torch.manual_seed(SEED) diff --git a/examples/new_project_templates/single_gpu_node_16bit_template.py b/examples/new_project_templates/single_gpu_node_16bit_template.py index 137f0e48..babf18e7 100644 --- a/examples/new_project_templates/single_gpu_node_16bit_template.py +++ b/examples/new_project_templates/single_gpu_node_16bit_template.py @@ -9,7 +9,7 @@ from test_tube import HyperOptArgumentParser, Experiment from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel SEED = 2334 torch.manual_seed(SEED) diff --git a/examples/new_project_templates/single_gpu_node_ddp_template.py b/examples/new_project_templates/single_gpu_node_ddp_template.py index e8f46012..68a332ae 100644 --- a/examples/new_project_templates/single_gpu_node_ddp_template.py +++ b/examples/new_project_templates/single_gpu_node_ddp_template.py @@ -9,7 +9,7 @@ from test_tube import HyperOptArgumentParser, Experiment from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel SEED = 2334 torch.manual_seed(SEED) diff --git a/examples/new_project_templates/single_gpu_node_dp_template.py b/examples/new_project_templates/single_gpu_node_dp_template.py index f48df5ca..d752713e 100644 --- a/examples/new_project_templates/single_gpu_node_dp_template.py +++ b/examples/new_project_templates/single_gpu_node_dp_template.py @@ -9,7 +9,7 @@ from test_tube import HyperOptArgumentParser, Experiment from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel SEED = 2334 torch.manual_seed(SEED) diff --git a/examples/new_project_templates/trainer_cpu_template.py b/examples/new_project_templates/trainer_cpu_template.py index 21d705c6..84a29a9b 100644 --- a/examples/new_project_templates/trainer_cpu_template.py +++ b/examples/new_project_templates/trainer_cpu_template.py @@ -6,7 +6,7 @@ from pytorch_lightning.models.trainer import Trainer from pytorch_lightning.utilities.arg_parse import add_default_args from pytorch_lightning.callbacks.pt_callbacks import EarlyStopping, ModelCheckpoint -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel def main(hparams):