Config files adjusted advances #2

config/config.yaml

@@ -14,13 +14,13 @@ training:
 
 data:
   name: "CIFAR10"
-  dataset_path: "./data/"
+  dataset_path: "./data/processed"
   test_size: 0.2
   val_size: 0.1
   transforms:
     - name: Resize
       parameters:
-        size: [64, 64]
+        size: [224, 224]
 
     - name: ToTensor
 

config/config_test.yaml

@@ -2,7 +2,7 @@ trainer: "BasicTrainer"
 
 data:
   name: "FakeData"
-  dataset_path: ./data
+  dataset_path: "./data"
   transforms:
     - name: ToTensor
 

config/fine_tuning_config.yaml

@@ -3,29 +3,29 @@ trainer: "BasicTrainer"
 random_seed: 43
 
 model:
-  name: efficientnet_b0
-  num_classes: 10
+  name: efficientnet_b1
+  num_classes: 394
   pretrained: true
 
 training:
   batch_size: 32
   num_epochs: 2   # Just maintained for compatibility
   initial_epochs: 1
-  fine_tuning_epochs: 0
+  fine_tuning_epochs: 1
   learning_rate: 0.001  # Also maintained for compatibility
   fine_tuning_learning_rate: 0.0001
-  freeze_until_layer: "classifier.1.weight"
+  freeze_until_layer: "classifier.1.weight"  # "fc.weight"
   final_fine_tuning_learning_rate: 0.00001
 
 data:
-  name: "CIFAR10"
-  dataset_path: "./data/"
+  name: "CarDataset"
+  dataset_path: "./data/processed/BD-461"
   test_size: 0.2
   val_size: 0.1
   transforms:
     - name: Resize
       parameters:
-        size: [64, 64]
+        size: [240, 240]
 
     - name: ToTensor