From 3ed7742370fd9b297c89a3a3fe4324b3687b69c4 Mon Sep 17 00:00:00 2001 From: Tizian Wenzel <wenzeltn@nbanm02.mathematik.uni-stuttgart.de> Date: Mon, 26 Dec 2022 22:43:46 +0100 Subject: [PATCH] sdkn: Some files added (required for P41 computations). --- main_lightning_SDKN.py | 2 +- utils/lightning_models.py | 6 +++--- utils/utilities.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/main_lightning_SDKN.py b/main_lightning_SDKN.py index 84f9d86..2f1ecc2 100644 --- a/main_lightning_SDKN.py +++ b/main_lightning_SDKN.py @@ -4,7 +4,7 @@ import pytorch_lightning as pl from pytorch_lightning.loggers import TensorBoardLogger -from utils import utilities +from sdkn.utils import utilities from utils import lightning_models from utils import settings from datetime import date diff --git a/utils/lightning_models.py b/utils/lightning_models.py index 4d4bb9c..e17677a 100644 --- a/utils/lightning_models.py +++ b/utils/lightning_models.py @@ -3,10 +3,10 @@ import pytorch_lightning as pl import torch from torch import nn -from utils.utilities import ActivFunc -from utils import kernels +from sdkn.utils.utilities import ActivFunc +from sdkn.utils import kernels from torch.nn import functional as F -from utils import settings +from sdkn.utils import settings class Network(pl.LightningModule): diff --git a/utils/utilities.py b/utils/utilities.py index 742cbf9..960462f 100644 --- a/utils/utilities.py +++ b/utils/utilities.py @@ -5,8 +5,8 @@ import torch from torch.utils.data import DataLoader from torch.utils.data import Dataset -from utils import kernels -from utils import settings +from sdkn.utils import kernels +from sdkn.utils import settings class TorchDataset(Dataset): -- GitLab