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