From db1ad565f3b7b0e152888508e368d2936af7d254 Mon Sep 17 00:00:00 2001 From: "J. Borovec" Date: Thu, 26 Mar 2020 13:57:41 +0100 Subject: [PATCH] fix imports --- pl_examples/__init__.py | 2 +- pl_examples/basic_examples/cpu_template.py | 2 +- pl_examples/basic_examples/gpu_template.py | 2 +- pl_examples/basic_examples/multi_node_ddp2_demo.py | 2 +- pl_examples/basic_examples/multi_node_ddp_demo.py | 2 +- pl_examples/domain_templates/semseg.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/pl_examples/__init__.py b/pl_examples/__init__.py index 7d980b12c56d77..1c5908539cfdc6 100644 --- 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 index cea868e275f7bc..0db5b12e6bf6e9 100644 --- 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 index bd60a84b2ccf30..08aa9dc20e19da 100644 --- 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 index 7c4471ffa73819..1ef994f32b4ee6 100644 --- 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 index 9220d91f47f291..6539761e45c29a 100644 --- 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 index 8f25243cffb968..c5e7bb186d0776 100644 --- 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):