groderg commited on
Commit
83f9cb9
·
verified ·
1 Parent(s): b8a04e3

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +126 -110
config.json CHANGED
@@ -1,111 +1,127 @@
1
  {
2
- "_name_or_path": "facebook/dinov2-large",
3
- "apply_layernorm": true,
4
- "architectures": [
5
- "Dinov2ForImageClassification"
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": "ALGAE",
14
- "1": "Acr",
15
- "2": "Acr_Br",
16
- "3": "Anem",
17
- "4": "CCA",
18
- "5": "Ech",
19
- "6": "Fts",
20
- "7": "Gal",
21
- "8": "Gon",
22
- "9": "Mtp",
23
- "10": "P",
24
- "11": "Poc",
25
- "12": "Por",
26
- "13": "R",
27
- "14": "RDC",
28
- "15": "S",
29
- "16": "SG",
30
- "17": "Sarg",
31
- "18": "Ser",
32
- "19": "Slt",
33
- "20": "Sp",
34
- "21": "Turf",
35
- "22": "UNK"
36
- },
37
- "image_size": 128,
38
- "initializer_range": 0.02,
39
- "label2id": {
40
- "ALGAE": 0,
41
- "Acr": 1,
42
- "Acr_Br": 2,
43
- "Anem": 3,
44
- "CCA": 4,
45
- "Ech": 5,
46
- "Fts": 6,
47
- "Gal": 7,
48
- "Gon": 8,
49
- "Mtp": 9,
50
- "P": 10,
51
- "Poc": 11,
52
- "Por": 12,
53
- "R": 13,
54
- "RDC": 14,
55
- "S": 15,
56
- "SG": 16,
57
- "Sarg": 17,
58
- "Ser": 18,
59
- "Slt": 19,
60
- "Sp": 20,
61
- "Turf": 21,
62
- "UNK": 22
63
- },
64
- "layer_norm_eps": 1e-06,
65
- "layerscale_value": 1.0,
66
- "mlp_ratio": 4,
67
- "model_type": "dinov2",
68
- "num_attention_heads": 16,
69
- "num_channels": 3,
70
- "num_hidden_layers": 24,
71
- "out_features": [
72
- "stage24"
73
- ],
74
- "out_indices": [
75
- 24
76
- ],
77
- "patch_size": 14,
78
- "problem_type": "multi_label_classification",
79
- "qkv_bias": true,
80
- "reshape_hidden_states": true,
81
- "stage_names": [
82
- "stem",
83
- "stage1",
84
- "stage2",
85
- "stage3",
86
- "stage4",
87
- "stage5",
88
- "stage6",
89
- "stage7",
90
- "stage8",
91
- "stage9",
92
- "stage10",
93
- "stage11",
94
- "stage12",
95
- "stage13",
96
- "stage14",
97
- "stage15",
98
- "stage16",
99
- "stage17",
100
- "stage18",
101
- "stage19",
102
- "stage20",
103
- "stage21",
104
- "stage22",
105
- "stage23",
106
- "stage24"
107
- ],
108
- "torch_dtype": "float32",
109
- "transformers_version": "4.44.2",
110
- "use_swiglu_ffn": false
111
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "facebook/dinov2-large",
3
+ "apply_layernorm": true,
4
+ "architectures": [
5
+ "Dinov2ForImageClassification"
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": "ALGAE",
14
+ "1": "Acr",
15
+ "2": "Acr_Br",
16
+ "3": "Anem",
17
+ "4": "CCA",
18
+ "5": "Ech",
19
+ "6": "Fts",
20
+ "7": "Gal",
21
+ "8": "Gon",
22
+ "9": "Mtp",
23
+ "10": "P",
24
+ "11": "Poc",
25
+ "12": "Por",
26
+ "13": "R",
27
+ "14": "RDC",
28
+ "15": "S",
29
+ "16": "SG",
30
+ "17": "Sarg",
31
+ "18": "Ser",
32
+ "19": "Slt",
33
+ "20": "Sp",
34
+ "21": "Turf",
35
+ "22": "UNK"
36
+ },
37
+ "image_size": 128,
38
+ "initializer_range": 0.02,
39
+ "label2id": {
40
+ "ALGAE": 0,
41
+ "Acr": 1,
42
+ "Acr_Br": 2,
43
+ "Anem": 3,
44
+ "CCA": 4,
45
+ "Ech": 5,
46
+ "Fts": 6,
47
+ "Gal": 7,
48
+ "Gon": 8,
49
+ "Mtp": 9,
50
+ "P": 10,
51
+ "Poc": 11,
52
+ "Por": 12,
53
+ "R": 13,
54
+ "RDC": 14,
55
+ "S": 15,
56
+ "SG": 16,
57
+ "Sarg": 17,
58
+ "Ser": 18,
59
+ "Slt": 19,
60
+ "Sp": 20,
61
+ "Turf": 21,
62
+ "UNK": 22
63
+ },
64
+ "layer_norm_eps": 1e-06,
65
+ "layerscale_value": 1.0,
66
+ "mlp_ratio": 4,
67
+ "model_type": "dinov2",
68
+ "num_attention_heads": 16,
69
+ "num_channels": 3,
70
+ "num_hidden_layers": 24,
71
+ "out_features": [
72
+ "stage24"
73
+ ],
74
+ "out_indices": [
75
+ 24
76
+ ],
77
+ "patch_size": 14,
78
+ "problem_type": "multi_label_classification",
79
+ "qkv_bias": true,
80
+ "reshape_hidden_states": true,
81
+ "stage_names": [
82
+ "stem",
83
+ "stage1",
84
+ "stage2",
85
+ "stage3",
86
+ "stage4",
87
+ "stage5",
88
+ "stage6",
89
+ "stage7",
90
+ "stage8",
91
+ "stage9",
92
+ "stage10",
93
+ "stage11",
94
+ "stage12",
95
+ "stage13",
96
+ "stage14",
97
+ "stage15",
98
+ "stage16",
99
+ "stage17",
100
+ "stage18",
101
+ "stage19",
102
+ "stage20",
103
+ "stage21",
104
+ "stage22",
105
+ "stage23",
106
+ "stage24"
107
+ ],
108
+ "torch_dtype": "float32",
109
+ "transformers_version": "4.44.2",
110
+ "use_swiglu_ffn": false,
111
+ "initial_learning_rate": 0.001,
112
+ "train_batch_size": 64,
113
+ "eval_batch_size": 64,
114
+ "optimizer": {
115
+ "type": "Adam"
116
+ },
117
+ "lr_scheduler_type": {
118
+ "type": "ReduceLROnPlateau"
119
+ },
120
+ "patience_lr_scheduler": 5,
121
+ "factor_lr_scheduler": 0.1,
122
+ "weight_decay": 0.0001,
123
+ "early_stopping_patience": 10,
124
+ "freeze_encoder": true,
125
+ "data_augmentation": true,
126
+ "num_epochs": 150
127
+ }