Upload config.json

config.json

@@ -1,107 +1,130 @@
 {
-  "_name_or_path": "microsoft/resnet-50",
-  "architectures": [
-    "ResNetForImageClassification"
-  ],
-  "depths": [
-    3,
-    4,
-    6,
-    3
-  ],
-  "downsample_in_bottleneck": false,
-  "downsample_in_first_stage": false,
-  "embedding_size": 64,
-  "hidden_act": "relu",
-  "hidden_sizes": [
-    256,
-    512,
-    1024,
-    2048
-  ],
-  "id2label": {
-    "0": "Acropore_branched",
-    "1": "Acropore_digitised",
-    "2": "Acropore_sub_massive",
-    "3": "Acropore_tabular",
-    "4": "Algae_assembly",
-    "5": "Algae_drawn_up",
-    "6": "Algae_limestone",
-    "7": "Algae_sodding",
-    "8": "Atra/Leucospilota",
-    "9": "Bleached_coral",
-    "10": "Blurred",
-    "11": "Dead_coral",
-    "12": "Fish",
-    "13": "Homo_sapiens",
-    "14": "Human_object",
-    "15": "Living_coral",
-    "16": "Millepore",
-    "17": "No_acropore_encrusting",
-    "18": "No_acropore_foliaceous",
-    "19": "No_acropore_massive",
-    "20": "No_acropore_solitary",
-    "21": "No_acropore_sub_massive",
-    "22": "Rock",
-    "23": "Rubble",
-    "24": "Sand",
-    "25": "Sea_cucumber",
-    "26": "Sea_urchins",
-    "27": "Sponge",
-    "28": "Syringodium_isoetifolium",
-    "29": "Thalassodendron_ciliatum",
-    "30": "Useless"
-  },
-  "label2id": {
-    "Acropore_branched": 0,
-    "Acropore_digitised": 1,
-    "Acropore_sub_massive": 2,
-    "Acropore_tabular": 3,
-    "Algae_assembly": 4,
-    "Algae_drawn_up": 5,
-    "Algae_limestone": 6,
-    "Algae_sodding": 7,
-    "Atra/Leucospilota": 8,
-    "Bleached_coral": 9,
-    "Blurred": 10,
-    "Dead_coral": 11,
-    "Fish": 12,
-    "Homo_sapiens": 13,
-    "Human_object": 14,
-    "Living_coral": 15,
-    "Millepore": 16,
-    "No_acropore_encrusting": 17,
-    "No_acropore_foliaceous": 18,
-    "No_acropore_massive": 19,
-    "No_acropore_solitary": 20,
-    "No_acropore_sub_massive": 21,
-    "Rock": 22,
-    "Rubble": 23,
-    "Sand": 24,
-    "Sea_cucumber": 25,
-    "Sea_urchins": 26,
-    "Sponge": 27,
-    "Syringodium_isoetifolium": 28,
-    "Thalassodendron_ciliatum": 29,
-    "Useless": 30
-  },
-  "layer_type": "bottleneck",
-  "model_type": "resnet",
-  "num_channels": 3,
-  "out_features": [
-    "stage4"
-  ],
-  "out_indices": [
-    4
-  ],
-  "problem_type": "multi_label_classification",
-  "stage_names": [
-    "stem",
-    "stage1",
-    "stage2",
-    "stage3",
-    "stage4"
-  ],
-  "torch_dtype": "float32",
-  "transformers_version": "4.44.2"
-}
+    "_name_or_path": "microsoft/resnet-50",
+    "architectures": [
+        "ResNetForImageClassification"
+    ],
+    "depths": [
+        3,
+        4,
+        6,
+        3
+    ],
+    "downsample_in_bottleneck": false,
+    "downsample_in_first_stage": false,
+    "embedding_size": 64,
+    "hidden_act": "relu",
+    "hidden_sizes": [
+        256,
+        512,
+        1024,
+        2048
+    ],
+    "id2label": {
+        "0": "Acropore_branched",
+        "1": "Acropore_digitised",
+        "2": "Acropore_sub_massive",
+        "3": "Acropore_tabular",
+        "4": "Algae_assembly",
+        "5": "Algae_drawn_up",
+        "6": "Algae_limestone",
+        "7": "Algae_sodding",
+        "8": "Atra/Leucospilota",
+        "9": "Bleached_coral",
+        "10": "Blurred",
+        "11": "Dead_coral",
+        "12": "Fish",
+        "13": "Homo_sapiens",
+        "14": "Human_object",
+        "15": "Living_coral",
+        "16": "Millepore",
+        "17": "No_acropore_encrusting",
+        "18": "No_acropore_foliaceous",
+        "19": "No_acropore_massive",
+        "20": "No_acropore_solitary",
+        "21": "No_acropore_sub_massive",
+        "22": "Rock",
+        "23": "Rubble",
+        "24": "Sand",
+        "25": "Sea_cucumber",
+        "26": "Sea_urchins",
+        "27": "Sponge",
+        "28": "Syringodium_isoetifolium",
+        "29": "Thalassodendron_ciliatum",
+        "30": "Useless"
+    },
+    "label2id": {
+        "Acropore_branched": 0,
+        "Acropore_digitised": 1,
+        "Acropore_sub_massive": 2,
+        "Acropore_tabular": 3,
+        "Algae_assembly": 4,
+        "Algae_drawn_up": 5,
+        "Algae_limestone": 6,
+        "Algae_sodding": 7,
+        "Atra/Leucospilota": 8,
+        "Bleached_coral": 9,
+        "Blurred": 10,
+        "Dead_coral": 11,
+        "Fish": 12,
+        "Homo_sapiens": 13,
+        "Human_object": 14,
+        "Living_coral": 15,
+        "Millepore": 16,
+        "No_acropore_encrusting": 17,
+        "No_acropore_foliaceous": 18,
+        "No_acropore_massive": 19,
+        "No_acropore_solitary": 20,
+        "No_acropore_sub_massive": 21,
+        "Rock": 22,
+        "Rubble": 23,
+        "Sand": 24,
+        "Sea_cucumber": 25,
+        "Sea_urchins": 26,
+        "Sponge": 27,
+        "Syringodium_isoetifolium": 28,
+        "Thalassodendron_ciliatum": 29,
+        "Useless": 30
+    },
+    "layer_type": "bottleneck",
+    "model_type": "resnet",
+    "num_channels": 3,
+    "out_features": [
+        "stage4"
+    ],
+    "out_indices": [
+        4
+    ],
+    "problem_type": "multi_label_classification",
+    "stage_names": [
+        "stem",
+        "stage1",
+        "stage2",
+        "stage3",
+        "stage4"
+    ],
+    "torch_dtype": "float32",
+    "transformers_version": "4.44.2",
+    "initial_learning_rate": 0.001,
+    "train_batch_size": 32,
+    "eval_batch_size": 32,
+    "optimizer": {
+        "type": "Adam"
+    },
+    "lr_scheduler_type": {
+        "type": "ReduceLROnPlateau"
+    },
+    "patience_lr_scheduler": 5,
+    "factor_lr_scheduler": 0.1,
+    "weight_decay": 0.0001,
+    "early_stopping_patience": 10,
+    "freeze_encoder": true,
+    "data_augmentation": true,
+    "num_epochs": 400,
+    "emissions_data": {
+        "emissions": 0.1871415951855612,
+        "source": "Code Carbon",
+        "training_type": "fine-tuning",
+        "geographical_location": "Brest, France",
+        "hardware_used": "NVIDIA Tesla V100 PCIe 32 Go"
+    }
+}