cutout examples

examples/new_project_templates/trainer_cpu_template.py

@@ -3,7 +3,7 @@ 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.utilities.arg_parse import add_default_args
 from pytorch_lightning.callbacks.pt_callbacks import EarlyStopping, ModelCheckpoint
 from lightning_module_template import LightningTemplateModel
 

pytorch_lightning/models/trainer.py

@@ -18,7 +18,7 @@ import tqdm
 from pytorch_lightning.root_module.memory import get_gpu_memory_map
 from pytorch_lightning.root_module.model_saving import TrainerIO
 from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel, LightningDataParallel
-from pytorch_lightning.utils.debugging import MisconfigurationException
+from pytorch_lightning.utilities.debugging import MisconfigurationException
 
 try:
     from apex import amp

tests/debug.py

@@ -1,15 +1,10 @@
-import pytest
 from pytorch_lightning import Trainer
-from pytorch_lightning.examples.new_project_templates.lightning_module_template import LightningTemplateModel
+from examples import LightningTemplateModel
 from argparse import Namespace
 from test_tube import Experiment
 from pytorch_lightning.callbacks import ModelCheckpoint
-import numpy as np
-import warnings
-import torch
 import os
 import shutil
-import pdb
 
 import pytorch_lightning as ptl
 import torch

tests/test_models.py

@@ -1,20 +1,20 @@
+import os
+import shutil
+import warnings
+
 import pytest
+import numpy as np
+import torch
 from pytorch_lightning import Trainer
-from pytorch_lightning.examples.new_project_templates.lightning_module_template import LightningTemplateModel
-from pytorch_lightning.testing_models.lm_test_module import LightningTestModel
+from examples import LightningTemplateModel
+from pytorch_lightning.testing.lm_test_module import LightningTestModel
 from argparse import Namespace
 from test_tube import Experiment, SlurmCluster
 from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
-from pytorch_lightning.utils.debugging import MisconfigurationException
+from pytorch_lightning.utilities.debugging import MisconfigurationException
 from pytorch_lightning.root_module import memory
 from pytorch_lightning.models.trainer import reduce_distributed_output
 from pytorch_lightning.root_module import model_saving
-import numpy as np
-import warnings
-import torch
-import os
-import shutil
-import pdb
 
 SEED = 2334
 torch.manual_seed(SEED)
@@ -232,8 +232,6 @@ def test_model_saving_loading():
     clear_save_dir()
 
 
-
-
 def test_model_freeze_unfreeze():
     hparams = get_hparams()
     model = LightningTestModel(hparams)