From 6bbaccd8d8774fc171629479db5ff575e01aa308 Mon Sep 17 00:00:00 2001 From: Pablo Olivares Date: Mon, 22 Apr 2024 12:55:18 +0200 Subject: [PATCH] Config files adjusted advances #2 --- config/config.yaml | 4 ++-- config/config_test.yaml | 2 +- config/fine_tuning_config.yaml | 14 +++++++------- 3 files changed, 10 insertions(+), 10 deletions(-) diff --git a/config/config.yaml b/config/config.yaml index 8751b6d..2f88169 100644 --- a/config/config.yaml +++ b/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 diff --git a/config/config_test.yaml b/config/config_test.yaml index 26d82c1..92528a1 100644 --- a/config/config_test.yaml +++ b/config/config_test.yaml @@ -2,7 +2,7 @@ trainer: "BasicTrainer" data: name: "FakeData" - dataset_path: ./data + dataset_path: "./data" transforms: - name: ToTensor diff --git a/config/fine_tuning_config.yaml b/config/fine_tuning_config.yaml index bf39df5..c5e9a00 100644 --- a/config/fine_tuning_config.yaml +++ b/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