[classification.config] Add model_transforms to model configs

src/mednet/libs/classification/config/models/alexnet.py

@@ -24,4 +24,5 @@ model = Alexnet(
         torchvision.transforms.Resize(512, antialias=True),
         RGB(),
     ],
+    augmentation_transforms=[],
 )

src/mednet/libs/classification/config/models/densenet.py

@@ -18,4 +18,6 @@ model = Densenet(
     optimizer_arguments=dict(lr=0.0001),
     pretrained=False,
     dropout=0.1,
+    model_transforms=[],
+    augmentation_transforms=[],
 )

src/mednet/libs/classification/config/models/densenet_pretrained.py

@@ -31,4 +31,5 @@ model = Densenet(
         ),
         RGB(),
     ],
+    augmentation_transforms=[],
 )

src/mednet/libs/classification/config/models/densenet_rs.py

@@ -20,4 +20,6 @@ model = Densenet(
     pretrained=False,
     dropout=0.1,
     num_classes=14,  # number of classes in NIH CXR-14
+    model_transforms=[],
+    augmentation_transforms=[],
 )

src/mednet/libs/classification/config/models/pasa.py

@@ -29,4 +29,5 @@ model = Pasa(
             interpolation=torchvision.transforms.InterpolationMode.BILINEAR,
         ),
     ],
+    augmentation_transforms=[],
 )