Hugging Face's logo

lombardata
/
DinoVdeau-small-2024_08_31-batch-size32_epochs150_freeze

TensorBoard
Safetensors
English
dinov2
multilabel-image-classification
multilabel
Generated from Trainer
Model card Files Metrics Community
lombardata commited on
Commit
3228a7d
•
1 Parent(s): 206bf39

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +137 -114
config.json CHANGED
@@ -1,115 +1,138 @@
1
  {
2
- "_name_or_path": "facebook/dinov2-small",
3
- "apply_layernorm": true,
4
- "architectures": [
5
- "NewheadDinov2ForImageClassification"
6
- ],
7
- "attention_probs_dropout_prob": 0.0,
8
- "drop_path_rate": 0.0,
9
- "hidden_act": "gelu",
10
- "hidden_dropout_prob": 0.0,
11
- "hidden_size": 384,
12
- "id2label": {
13
- "0": "Acropore_branched",
14
- "1": "Acropore_digitised",
15
- "2": "Acropore_sub_massive",
16
- "3": "Acropore_tabular",
17
- "4": "Algae_assembly",
18
- "5": "Algae_drawn_up",
19
- "6": "Algae_limestone",
20
- "7": "Algae_sodding",
21
- "8": "Atra/Leucospilota",
22
- "9": "Bleached_coral",
23
- "10": "Blurred",
24
- "11": "Dead_coral",
25
- "12": "Fish",
26
- "13": "Homo_sapiens",
27
- "14": "Human_object",
28
- "15": "Living_coral",
29
- "16": "Millepore",
30
- "17": "No_acropore_encrusting",
31
- "18": "No_acropore_foliaceous",
32
- "19": "No_acropore_massive",
33
- "20": "No_acropore_solitary",
34
- "21": "No_acropore_sub_massive",
35
- "22": "Rock",
36
- "23": "Rubble",
37
- "24": "Sand",
38
- "25": "Sea_cucumber",
39
- "26": "Sea_urchins",
40
- "27": "Sponge",
41
- "28": "Syringodium_isoetifolium",
42
- "29": "Thalassodendron_ciliatum",
43
- "30": "Useless"
44
- },
45
- "image_size": 518,
46
- "initializer_range": 0.02,
47
- "label2id": {
48
- "Acropore_branched": 0,
49
- "Acropore_digitised": 1,
50
- "Acropore_sub_massive": 2,
51
- "Acropore_tabular": 3,
52
- "Algae_assembly": 4,
53
- "Algae_drawn_up": 5,
54
- "Algae_limestone": 6,
55
- "Algae_sodding": 7,
56
- "Atra/Leucospilota": 8,
57
- "Bleached_coral": 9,
58
- "Blurred": 10,
59
- "Dead_coral": 11,
60
- "Fish": 12,
61
- "Homo_sapiens": 13,
62
- "Human_object": 14,
63
- "Living_coral": 15,
64
- "Millepore": 16,
65
- "No_acropore_encrusting": 17,
66
- "No_acropore_foliaceous": 18,
67
- "No_acropore_massive": 19,
68
- "No_acropore_solitary": 20,
69
- "No_acropore_sub_massive": 21,
70
- "Rock": 22,
71
- "Rubble": 23,
72
- "Sand": 24,
73
- "Sea_cucumber": 25,
74
- "Sea_urchins": 26,
75
- "Sponge": 27,
76
- "Syringodium_isoetifolium": 28,
77
- "Thalassodendron_ciliatum": 29,
78
- "Useless": 30
79
- },
80
- "layer_norm_eps": 1e-06,
81
- "layerscale_value": 1.0,
82
- "mlp_ratio": 4,
83
- "model_type": "dinov2",
84
- "num_attention_heads": 6,
85
- "num_channels": 3,
86
- "num_hidden_layers": 12,
87
- "out_features": [
88
- "stage12"
89
- ],
90
- "out_indices": [
91
- 12
92
- ],
93
- "patch_size": 14,
94
- "problem_type": "multi_label_classification",
95
- "qkv_bias": true,
96
- "reshape_hidden_states": true,
97
- "stage_names": [
98
- "stem",
99
- "stage1",
100
- "stage2",
101
- "stage3",
102
- "stage4",
103
- "stage5",
104
- "stage6",
105
- "stage7",
106
- "stage8",
107
- "stage9",
108
- "stage10",
109
- "stage11",
110
- "stage12"
111
- ],
112
- "torch_dtype": "float32",
113
- "transformers_version": "4.41.1",
114
- "use_swiglu_ffn": false
115
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "facebook/dinov2-small",
3
+ "apply_layernorm": true,
4
+ "architectures": [
5
+ "NewheadDinov2ForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "drop_path_rate": 0.0,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 384,
12
+ "id2label": {
13
+ "0": "Acropore_branched",
14
+ "1": "Acropore_digitised",
15
+ "2": "Acropore_sub_massive",
16
+ "3": "Acropore_tabular",
17
+ "4": "Algae_assembly",
18
+ "5": "Algae_drawn_up",
19
+ "6": "Algae_limestone",
20
+ "7": "Algae_sodding",
21
+ "8": "Atra/Leucospilota",
22
+ "9": "Bleached_coral",
23
+ "10": "Blurred",
24
+ "11": "Dead_coral",
25
+ "12": "Fish",
26
+ "13": "Homo_sapiens",
27
+ "14": "Human_object",
28
+ "15": "Living_coral",
29
+ "16": "Millepore",
30
+ "17": "No_acropore_encrusting",
31
+ "18": "No_acropore_foliaceous",
32
+ "19": "No_acropore_massive",
33
+ "20": "No_acropore_solitary",
34
+ "21": "No_acropore_sub_massive",
35
+ "22": "Rock",
36
+ "23": "Rubble",
37
+ "24": "Sand",
38
+ "25": "Sea_cucumber",
39
+ "26": "Sea_urchins",
40
+ "27": "Sponge",
41
+ "28": "Syringodium_isoetifolium",
42
+ "29": "Thalassodendron_ciliatum",
43
+ "30": "Useless"
44
+ },
45
+ "image_size": 518,
46
+ "initializer_range": 0.02,
47
+ "label2id": {
48
+ "Acropore_branched": 0,
49
+ "Acropore_digitised": 1,
50
+ "Acropore_sub_massive": 2,
51
+ "Acropore_tabular": 3,
52
+ "Algae_assembly": 4,
53
+ "Algae_drawn_up": 5,
54
+ "Algae_limestone": 6,
55
+ "Algae_sodding": 7,
56
+ "Atra/Leucospilota": 8,
57
+ "Bleached_coral": 9,
58
+ "Blurred": 10,
59
+ "Dead_coral": 11,
60
+ "Fish": 12,
61
+ "Homo_sapiens": 13,
62
+ "Human_object": 14,
63
+ "Living_coral": 15,
64
+ "Millepore": 16,
65
+ "No_acropore_encrusting": 17,
66
+ "No_acropore_foliaceous": 18,
67
+ "No_acropore_massive": 19,
68
+ "No_acropore_solitary": 20,
69
+ "No_acropore_sub_massive": 21,
70
+ "Rock": 22,
71
+ "Rubble": 23,
72
+ "Sand": 24,
73
+ "Sea_cucumber": 25,
74
+ "Sea_urchins": 26,
75
+ "Sponge": 27,
76
+ "Syringodium_isoetifolium": 28,
77
+ "Thalassodendron_ciliatum": 29,
78
+ "Useless": 30
79
+ },
80
+ "layer_norm_eps": 1e-06,
81
+ "layerscale_value": 1.0,
82
+ "mlp_ratio": 4,
83
+ "model_type": "dinov2",
84
+ "num_attention_heads": 6,
85
+ "num_channels": 3,
86
+ "num_hidden_layers": 12,
87
+ "out_features": [
88
+ "stage12"
89
+ ],
90
+ "out_indices": [
91
+ 12
92
+ ],
93
+ "patch_size": 14,
94
+ "problem_type": "multi_label_classification",
95
+ "qkv_bias": true,
96
+ "reshape_hidden_states": true,
97
+ "stage_names": [
98
+ "stem",
99
+ "stage1",
100
+ "stage2",
101
+ "stage3",
102
+ "stage4",
103
+ "stage5",
104
+ "stage6",
105
+ "stage7",
106
+ "stage8",
107
+ "stage9",
108
+ "stage10",
109
+ "stage11",
110
+ "stage12"
111
+ ],
112
+ "torch_dtype": "float32",
113
+ "transformers_version": "4.41.1",
114
+ "use_swiglu_ffn": false,
115
+ "initial_learning_rate": 0.001,
116
+ "train_batch_size": 32,
117
+ "eval_batch_size": 32,
118
+ "optimizer": {
119
+ "type": "Adam"
120
+ },
121
+ "lr_scheduler_type": {
122
+ "type": "ReduceLROnPlateau"
123
+ },
124
+ "patience_lr_scheduler": 5,
125
+ "factor_lr_scheduler": 0.1,
126
+ "weight_decay": 0.0001,
127
+ "early_stopping_patience": 10,
128
+ "freeze_encoder": true,
129
+ "data_augmentation": true,
130
+ "num_epochs": 150,
131
+ "emissions_data": {
132
+ "emissions": 1.6281850650290761,
133
+ "source": "Code Carbon",
134
+ "training_type": "fine-tuning",
135
+ "geographical_location": "Brest, France",
136
+ "hardware_used": "NVIDIA Tesla V100 PCIe 32 Go"
137
+ }
138
+ }