[models.config] Make preset configurations location independent

src/ptbench/configs/models/alexnet.py → src/ptbench/models/config/alexnet.py

@@ -6,8 +6,8 @@
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import SGD
 
-from ...data.augmentations import ElasticDeformation
-from ...models.alexnet import Alexnet
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.alexnet import Alexnet
 
 model = Alexnet(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/alexnet_pretrained.py → src/ptbench/models/config/alexnet_pretrained.py

@@ -6,8 +6,8 @@
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import SGD
 
-from ...data.augmentations import ElasticDeformation
-from ...models.alexnet import Alexnet
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.alexnet import Alexnet
 
 model = Alexnet(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/densenet.py → src/ptbench/models/config/densenet.py

@@ -6,8 +6,8 @@
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import Adam
 
-from ...data.augmentations import ElasticDeformation
-from ...models.densenet import Densenet
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.densenet import Densenet
 
 model = Densenet(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/densenet_pretrained.py → src/ptbench/models/config/densenet_pretrained.py

@@ -6,8 +6,8 @@
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import Adam
 
-from ...data.augmentations import ElasticDeformation
-from ...models.densenet import Densenet
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.densenet import Densenet
 
 model = Densenet(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/densenet_rs.py → src/ptbench/models/config/densenet_rs.py

@@ -10,8 +10,8 @@ CXR-14 label format (ie. 14 outputs).
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import Adam
 
-from ...data.augmentations import ElasticDeformation
-from ...models.densenet import Densenet
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.densenet import Densenet
 
 model = Densenet(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/logistic_regression.py → src/ptbench/models/config/logistic_regression.py

@@ -9,7 +9,7 @@ predicting tuberculosis presence in output.
 from torch import empty
 from torch.nn import BCEWithLogitsLoss
 
-from ...models.logistic_regression import LogisticRegression
+from ptbench.models.logistic_regression import LogisticRegression
 
 # config
 optimizer_configs = {"lr": 1e-2}

src/ptbench/configs/models/pasa.py → src/ptbench/models/config/pasa.py

@@ -13,8 +13,8 @@ Reference: [PASA-2019]_
 from torch.nn import BCEWithLogitsLoss
 from torch.optim import Adam
 
-from ...data.augmentations import ElasticDeformation
-from ...models.pasa import Pasa
+from ptbench.data.augmentations import ElasticDeformation
+from ptbench.models.pasa import Pasa
 
 model = Pasa(
     train_loss=BCEWithLogitsLoss(),

src/ptbench/configs/models/signs_to_tb.py → src/ptbench/models/config/signs_to_tb.py

@@ -10,7 +10,7 @@ predicting tuberculosis presence in output.
 from torch import empty
 from torch.nn import BCEWithLogitsLoss
 
-from ...models.signs_to_tb import SignsToTB
+from ptbench.models.signs_to_tb import SignsToTB
 
 # config
 optimizer_configs = {"lr": 1e-2}