fix imports

Lightning-AI · Mar 26, 2020 · db1ad56 · db1ad56
1 parent ed650d3
commit db1ad56
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/pl_examples/__init__.py b/pl_examples/__init__.py
@@ -140,7 +140,7 @@ def optimize_on_cluster(hyperparams):
 
 """
 
-from pl_examples.models.lightning_module_template import LightningTemplateModel
+from pl_examples.models.lightning_template import LightningTemplateModel
 
 __all__ = [
     'LightningTemplateModel'

diff --git a/pl_examples/basic_examples/cpu_template.py b/pl_examples/basic_examples/cpu_template.py
@@ -8,7 +8,7 @@
 import torch
 
 import pytorch_lightning as pl
-from pl_examples.models.lightning_module_template import LightningTemplateModel
+from pl_examples.models.lightning_template import LightningTemplateModel
 
 SEED = 2334
 torch.manual_seed(SEED)

diff --git a/pl_examples/basic_examples/gpu_template.py b/pl_examples/basic_examples/gpu_template.py
@@ -8,7 +8,7 @@
 import torch
 
 import pytorch_lightning as pl
-from pl_examples.models.lightning_module_template import LightningTemplateModel
+from pl_examples.models.lightning_template import LightningTemplateModel
 
 SEED = 2334
 torch.manual_seed(SEED)

diff --git a/pl_examples/basic_examples/multi_node_ddp2_demo.py b/pl_examples/basic_examples/multi_node_ddp2_demo.py
@@ -8,7 +8,7 @@
 import torch
 
 import pytorch_lightning as pl
-from pl_examples.models.lightning_module_template import LightningTemplateModel
+from pl_examples.models.lightning_template import LightningTemplateModel
 
 SEED = 2334
 torch.manual_seed(SEED)

diff --git a/pl_examples/basic_examples/multi_node_ddp_demo.py b/pl_examples/basic_examples/multi_node_ddp_demo.py
@@ -8,7 +8,7 @@
 import torch
 
 import pytorch_lightning as pl
-from pl_examples.models.lightning_module_template import LightningTemplateModel
+from pl_examples.models.lightning_template import LightningTemplateModel
 
 SEED = 2334
 torch.manual_seed(SEED)

diff --git a/pl_examples/domain_templates/semseg.py b/pl_examples/domain_templates/semseg.py
@@ -6,10 +6,10 @@
 import torch.nn.functional as F
 import torchvision.transforms as transforms
 from PIL import Image
-from models.unet.model import UNet
 from torch.utils.data import DataLoader, Dataset
 
 import pytorch_lightning as pl
+from pl_examples.models.unet import UNet
 
 
 class KITTI(Dataset):