fix imports

Author: Borda

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'

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)

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)

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)

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)

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):