Final cleaning closes #2

config/config_efficientnet.yaml

@@ -8,15 +8,15 @@ model:
     pretrained: true
 
 training:
-  batch_size: 32
+  batch_size: 64
   epochs:
-    initial: 200
-    fine_tuning: 200
+    initial: 300
+    fine_tuning: 300
   loss_function:
     type: "CrossEntropyLoss"
     parameters: {}
   optimizer:
-    type: "Adam"
+    type: "SGD"
     parameters:
       learning_rate: 0.005
   learning_rates:
@@ -62,10 +62,24 @@ data:
       parameters:
         mean: [0.485, 0.456, 0.406]
         std: [0.229, 0.224, 0.225]
+    - type: "ColorJitter"
+      parameters:
+        brightness: 0.3
+        contrast: 0.3
+        saturation: 0.3
+        hue: 0.1
+    - type: "RandomResizedCrop"
+      parameters:
+        size: [224, 224]
+        scale: [0.8, 1.0]
+        ratio: [0.75, 1.33]
+    - type: "RandomRotation"
+      parameters:
+        degrees: [-10, 10]
   eval_transforms:
     - type: "Resize"
       parameters:
-        size: [240, 240]
+        size: [224, 224]
     - type: "ToTensor"
       parameters: {}
     - type: "Normalize"

config/mm_config_densenet.yaml

@@ -10,13 +10,13 @@ model:
 training:
   batch_size: 32
   epochs:
-    initial: 200
-    fine_tuning: 200
+    initial: 300
+    fine_tuning: 300
   loss_function:
     type: "CrossEntropyLoss"
     parameters: {}
   optimizer:
-    type: "Adam"
+    type: "SGD"
     parameters:
       learning_rate: 0.005
   learning_rates:
@@ -41,7 +41,7 @@ callbacks:
   EarlyStopping:
     parameters:
       monitor: "val_loss"
-      patience: 5
+      patience: 15
       delta: 0
       verbose: true
 
@@ -53,7 +53,7 @@ data:
   transforms:
     - type: "Resize"
       parameters:
-        size: [240, 240]
+        size: [224, 224]
     #- type: "TrivialAugmentWide"
     #  parameters: {}
     - type: "ToTensor"
@@ -62,10 +62,31 @@ data:
       parameters:
         mean: [0.485, 0.456, 0.406]
         std: [0.229, 0.224, 0.225]
+    - type: "ColorJitter"
+      parameters:
+        brightness: 0.3
+        contrast: 0.3
+        saturation: 0.3
+        hue: 0.1
+    - type: "RandomResizedCrop"
+      parameters:
+        size: [224, 224]
+        scale: [0.8, 1.0]
+        ratio: [0.75, 1.33]
+    - type: "RandomRotation"
+      parameters:
+        degrees: [-10, 10]
+    - type: "GaussianBlur"
+      parameters:
+        kernel_size: [5, 5]
+        sigma: [0.1, 2.0]
+    - type: "RandomHorizontalFlip"
+      parameters:
+        p: 0.5
   eval_transforms:
     - type: "Resize"
       parameters:
-        size: [240, 240]
+        size: [224, 224]
     - type: "ToTensor"
       parameters: {}
     - type: "Normalize"

environment.yaml

@@ -328,4 +328,4 @@ dependencies:
       - torch-topological==0.1.7
       - wheel==0.43.0
       - widgetsnbextension==4.0.11
-prefix: /home/pab1s/.conda/envs/tda-nn-separability
+prefix: /home/pab1s/.conda/envs/tda-nn-analysis