diff --git a/main_lightning_SDKN.py b/main_lightning_SDKN.py index 2f1ecc28ca728abfb0e2b29b4356aed235c976e2..84f9d86e17b08965272eb99ef71b85c05c249ef7 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 e17677aaaa6cb7d9bb3483df82858d40640ff302..4d4bb9c0e416dfdd77221d83a6679958feb65dd9 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 960462fe8e0555cc4795ea7c4370507002dfd66d..742cbf978175de33e759fb3e497db9ece7d666b4 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):