From 9037c386e7217b07abc049d584155a494dd1ab38 Mon Sep 17 00:00:00 2001 From: Tizian Wenzel <wenzeltn@nbanm02.mathematik.uni-stuttgart.de> Date: Mon, 26 Dec 2022 22:55:06 +0100 Subject: [PATCH] Revert "sdkn: Some files added (required for P41 computations)." This reverts commit 3ed7742370fd9b297c89a3a3fe4324b3687b69c4. --- 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 2f1ecc2..84f9d86 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 sdkn.utils import utilities +from 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 e17677a..4d4bb9c 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 sdkn.utils.utilities import ActivFunc -from sdkn.utils import kernels +from utils.utilities import ActivFunc +from utils import kernels from torch.nn import functional as F -from sdkn.utils import settings +from utils import settings class Network(pl.LightningModule): diff --git a/utils/utilities.py b/utils/utilities.py index 960462f..742cbf9 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 sdkn.utils import kernels -from sdkn.utils import settings +from utils import kernels +from utils import settings class TorchDataset(Dataset): -- GitLab