Hugging Face's logo

lombardata
/
drone-DinoVdeau-large-2024_09_16-batch-size64_epochs100_freeze

TensorBoard
Safetensors
English
dinov2
multilabel-image-classification
multilabel
Generated from Trainer
Model card Files Metrics Community
lombardata commited on
Commit
9ef78d8
•
1 Parent(s): 464d5f7

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +117 -94
config.json CHANGED
@@ -1,95 +1,118 @@
1
  {
2
- "_name_or_path": "facebook/dinov2-large",
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": 1024,
12
- "id2label": {
13
- "0": "Acropore_branched",
14
- "1": "Acropore_digitised",
15
- "2": "Acropore_tabular",
16
- "3": "Algae",
17
- "4": "Dead_coral",
18
- "5": "Fish",
19
- "6": "Millepore",
20
- "7": "No_acropore_encrusting",
21
- "8": "No_acropore_massive",
22
- "9": "No_acropore_sub_massive",
23
- "10": "Rock",
24
- "11": "Rubble",
25
- "12": "Sand",
26
- "13": "Sea_cucumber",
27
- "14": "Sea_urchins"
28
- },
29
- "image_size": 64,
30
- "initializer_range": 0.02,
31
- "label2id": {
32
- "Acropore_branched": 0,
33
- "Acropore_digitised": 1,
34
- "Acropore_tabular": 2,
35
- "Algae": 3,
36
- "Dead_coral": 4,
37
- "Fish": 5,
38
- "Millepore": 6,
39
- "No_acropore_encrusting": 7,
40
- "No_acropore_massive": 8,
41
- "No_acropore_sub_massive": 9,
42
- "Rock": 10,
43
- "Rubble": 11,
44
- "Sand": 12,
45
- "Sea_cucumber": 13,
46
- "Sea_urchins": 14
47
- },
48
- "layer_norm_eps": 1e-06,
49
- "layerscale_value": 1.0,
50
- "mlp_ratio": 4,
51
- "model_type": "dinov2",
52
- "num_attention_heads": 16,
53
- "num_channels": 3,
54
- "num_hidden_layers": 24,
55
- "out_features": [
56
- "stage24"
57
- ],
58
- "out_indices": [
59
- 24
60
- ],
61
- "patch_size": 14,
62
- "problem_type": "multi_label_classification",
63
- "qkv_bias": true,
64
- "reshape_hidden_states": true,
65
- "stage_names": [
66
- "stem",
67
- "stage1",
68
- "stage2",
69
- "stage3",
70
- "stage4",
71
- "stage5",
72
- "stage6",
73
- "stage7",
74
- "stage8",
75
- "stage9",
76
- "stage10",
77
- "stage11",
78
- "stage12",
79
- "stage13",
80
- "stage14",
81
- "stage15",
82
- "stage16",
83
- "stage17",
84
- "stage18",
85
- "stage19",
86
- "stage20",
87
- "stage21",
88
- "stage22",
89
- "stage23",
90
- "stage24"
91
- ],
92
- "torch_dtype": "float32",
93
- "transformers_version": "4.41.1",
94
- "use_swiglu_ffn": false
95
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "facebook/dinov2-large",
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": 1024,
12
+ "id2label": {
13
+ "0": "Acropore_branched",
14
+ "1": "Acropore_digitised",
15
+ "2": "Acropore_tabular",
16
+ "3": "Algae",
17
+ "4": "Dead_coral",
18
+ "5": "Fish",
19
+ "6": "Millepore",
20
+ "7": "No_acropore_encrusting",
21
+ "8": "No_acropore_massive",
22
+ "9": "No_acropore_sub_massive",
23
+ "10": "Rock",
24
+ "11": "Rubble",
25
+ "12": "Sand",
26
+ "13": "Sea_cucumber",
27
+ "14": "Sea_urchins"
28
+ },
29
+ "image_size": 64,
30
+ "initializer_range": 0.02,
31
+ "label2id": {
32
+ "Acropore_branched": 0,
33
+ "Acropore_digitised": 1,
34
+ "Acropore_tabular": 2,
35
+ "Algae": 3,
36
+ "Dead_coral": 4,
37
+ "Fish": 5,
38
+ "Millepore": 6,
39
+ "No_acropore_encrusting": 7,
40
+ "No_acropore_massive": 8,
41
+ "No_acropore_sub_massive": 9,
42
+ "Rock": 10,
43
+ "Rubble": 11,
44
+ "Sand": 12,
45
+ "Sea_cucumber": 13,
46
+ "Sea_urchins": 14
47
+ },
48
+ "layer_norm_eps": 1e-06,
49
+ "layerscale_value": 1.0,
50
+ "mlp_ratio": 4,
51
+ "model_type": "dinov2",
52
+ "num_attention_heads": 16,
53
+ "num_channels": 3,
54
+ "num_hidden_layers": 24,
55
+ "out_features": [
56
+ "stage24"
57
+ ],
58
+ "out_indices": [
59
+ 24
60
+ ],
61
+ "patch_size": 14,
62
+ "problem_type": "multi_label_classification",
63
+ "qkv_bias": true,
64
+ "reshape_hidden_states": true,
65
+ "stage_names": [
66
+ "stem",
67
+ "stage1",
68
+ "stage2",
69
+ "stage3",
70
+ "stage4",
71
+ "stage5",
72
+ "stage6",
73
+ "stage7",
74
+ "stage8",
75
+ "stage9",
76
+ "stage10",
77
+ "stage11",
78
+ "stage12",
79
+ "stage13",
80
+ "stage14",
81
+ "stage15",
82
+ "stage16",
83
+ "stage17",
84
+ "stage18",
85
+ "stage19",
86
+ "stage20",
87
+ "stage21",
88
+ "stage22",
89
+ "stage23",
90
+ "stage24"
91
+ ],
92
+ "torch_dtype": "float32",
93
+ "transformers_version": "4.41.1",
94
+ "use_swiglu_ffn": false,
95
+ "initial_learning_rate": 0.001,
96
+ "train_batch_size": 64,
97
+ "eval_batch_size": 64,
98
+ "optimizer": {
99
+ "type": "Adam"
100
+ },
101
+ "lr_scheduler_type": {
102
+ "type": "ReduceLROnPlateau"
103
+ },
104
+ "patience_lr_scheduler": 5,
105
+ "factor_lr_scheduler": 0.1,
106
+ "weight_decay": 0.0001,
107
+ "early_stopping_patience": 10,
108
+ "freeze_encoder": true,
109
+ "data_augmentation": true,
110
+ "num_epochs": 100,
111
+ "emissions_data": {
112
+ "emissions": 0.13788314685965944,
113
+ "source": "Code Carbon",
114
+ "training_type": "fine-tuning",
115
+ "geographical_location": "Brest, France",
116
+ "hardware_used": "NVIDIA Tesla V100 PCIe 32 Go"
117
+ }
118
+ }