cleaned up imports for examples

pytorch_lightning/examples/new_project_templates/multi_node_cluster_template.py

@@ -1,13 +1,10 @@
 import os
-import sys
 import numpy as np
 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.callbacks import EarlyStopping, ModelCheckpoint
 
 SEED = 2334

pytorch_lightning/examples/new_project_templates/single_cpu_template.py

@@ -2,15 +2,11 @@
 Runs a model on a single node across N-gpus.
 """
 import os
-import sys
 import numpy as np
-from time import sleep
 import torch
 
-from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster
+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.callbacks import EarlyStopping, ModelCheckpoint
 
 SEED = 2334

pytorch_lightning/examples/new_project_templates/single_gpu_node_16bit_template.py

@@ -2,15 +2,11 @@
 Runs a model on a single node across N-gpus.
 """
 import os
-import sys
 import numpy as np
-from time import sleep
 import torch
 
-from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster
+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.callbacks import EarlyStopping, ModelCheckpoint
 
 SEED = 2334

pytorch_lightning/examples/new_project_templates/single_gpu_node_dp_template.py

@@ -2,15 +2,11 @@
 Runs a model on a single node across N-gpus.
 """
 import os
-import sys
 import numpy as np
-from time import sleep
 import torch
 
-from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster
+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.callbacks import EarlyStopping, ModelCheckpoint
 
 SEED = 2334

pytorch_lightning/examples/new_project_templates/single_gpu_node_template.py

@@ -2,14 +2,11 @@
 Runs a model on a single node across N-gpus.
 """
 import os
-import sys
 import numpy as np
-from time import sleep
 import torch
 
-from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster
+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.callbacks import EarlyStopping, ModelCheckpoint
 

pytorch_lightning/examples/new_project_templates/single_gpu_template.py

@@ -2,14 +2,11 @@
 Runs a model on a single node across N-gpus.
 """
 import os
-import sys
 import numpy as np
-from time import sleep
 import torch
 
-from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster
+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.callbacks import EarlyStopping, ModelCheckpoint
 

pytorch_lightning/examples/new_project_templates/trainer_cpu_template.py

@@ -5,7 +5,7 @@ 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.callbacks.pt_callbacks import EarlyStopping, ModelCheckpoint
-from docs.source.examples.example_model import ExampleModel
+from lightning_module_template import LightningTemplateModel
 
 
 def main(hparams):
@@ -28,7 +28,7 @@ def main(hparams):
     exp.save()
 
     # build model
-    model = ExampleModel(hparams)
+    model = LightningTemplateModel(hparams)
 
     # callbacks
     early_stop = EarlyStopping(