chizhik commited on
Commit
6be84d1
1 Parent(s): 76322cc

update readme and add model weights

Browse files
README.md CHANGED
@@ -1,3 +1,4 @@
1
  ---
2
  license: afl-3.0
 
3
  ---
 
1
  ---
2
  license: afl-3.0
3
+ inference: false
4
  ---
config.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/mnt/beegfs/mc000051/CERPLES/Models/bio-bert-base-spanish-wwm-uncased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "LABEL_0",
14
+ "1": "LABEL_1",
15
+ "2": "LABEL_2",
16
+ "3": "LABEL_3",
17
+ "4": "LABEL_4",
18
+ "5": "LABEL_5",
19
+ "6": "LABEL_6",
20
+ "7": "LABEL_7",
21
+ "8": "LABEL_8",
22
+ "9": "LABEL_9",
23
+ "10": "LABEL_10",
24
+ "11": "LABEL_11",
25
+ "12": "LABEL_12",
26
+ "13": "LABEL_13",
27
+ "14": "LABEL_14",
28
+ "15": "LABEL_15"
29
+ },
30
+ "initializer_range": 0.02,
31
+ "intermediate_size": 3072,
32
+ "label2id": {
33
+ "LABEL_0": 0,
34
+ "LABEL_1": 1,
35
+ "LABEL_10": 10,
36
+ "LABEL_11": 11,
37
+ "LABEL_12": 12,
38
+ "LABEL_13": 13,
39
+ "LABEL_14": 14,
40
+ "LABEL_15": 15,
41
+ "LABEL_2": 2,
42
+ "LABEL_3": 3,
43
+ "LABEL_4": 4,
44
+ "LABEL_5": 5,
45
+ "LABEL_6": 6,
46
+ "LABEL_7": 7,
47
+ "LABEL_8": 8,
48
+ "LABEL_9": 9
49
+ },
50
+ "layer_norm_eps": 1e-12,
51
+ "max_position_embeddings": 512,
52
+ "model_type": "bert",
53
+ "num_attention_heads": 12,
54
+ "num_hidden_layers": 12,
55
+ "output_past": true,
56
+ "pad_token_id": 1,
57
+ "position_embedding_type": "absolute",
58
+ "problem_type": "multi_label_classification",
59
+ "torch_dtype": "float32",
60
+ "transformers_version": "4.19.2",
61
+ "type_vocab_size": 2,
62
+ "use_cache": true,
63
+ "vocab_size": 38468
64
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9f7ed6306ce2558c67241d43e75108ef5504d67223240c7cdc3c037e22af4dd
3
+ size 924893085
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13199df00899bc172e4a71c8c8055064f5b667553461e2bafa89578e1e0bc393
3
+ size 462459309
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3661d92c3c0d07b200452f1fd844a838ddd196d6a9c84c47e571efdde5b87954
3
+ size 14567
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03a4078fa2defe0274b0261f1ba356b0aa8e2ec3604b66663f1a29d151167f2a
3
+ size 623
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"do_lower_case": true, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "tokenize_chinese_chars": true, "strip_accents": false, "do_basic_tokenize": true, "never_split": null, "model_max_length": 512, "special_tokens_map_file": "/mnt/beegfs/mc000051/CERPLES/Models/bio-bert-base-spanish-wwm-uncased/special_tokens_map.json", "name_or_path": "/mnt/beegfs/mc000051/CERPLES/Models/bio-bert-base-spanish-wwm-uncased", "tokenizer_class": "BertTokenizer"}
trainer_state.json ADDED
@@ -0,0 +1,336 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.5951067347267447,
3
+ "best_model_checkpoint": "./CARES/checkpoints/bio-bert/checkpoint-8000",
4
+ "epoch": 56.737588652482266,
5
+ "global_step": 8000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 3.55,
12
+ "learning_rate": 4.822695035460993e-05,
13
+ "loss": 0.2147,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 3.55,
18
+ "eval_loss": 0.18118813633918762,
19
+ "eval_macro_f1": 0.30207207042893824,
20
+ "eval_macro_precision": 0.5201644497890242,
21
+ "eval_macro_recall": 0.2643836699737847,
22
+ "eval_micro_f1": 0.6030504497457959,
23
+ "eval_micro_precision": 0.7529296875,
24
+ "eval_micro_recall": 0.50293542074364,
25
+ "eval_runtime": 2.8226,
26
+ "eval_samples_per_second": 342.243,
27
+ "eval_steps_per_second": 21.612,
28
+ "step": 500
29
+ },
30
+ {
31
+ "epoch": 7.09,
32
+ "learning_rate": 4.645390070921986e-05,
33
+ "loss": 0.1141,
34
+ "step": 1000
35
+ },
36
+ {
37
+ "epoch": 7.09,
38
+ "eval_loss": 0.17683139443397522,
39
+ "eval_macro_f1": 0.4546518122185415,
40
+ "eval_macro_precision": 0.5858554608554609,
41
+ "eval_macro_recall": 0.3972024243991636,
42
+ "eval_micro_f1": 0.6614564831261102,
43
+ "eval_micro_precision": 0.7262090483619345,
44
+ "eval_micro_recall": 0.6073059360730594,
45
+ "eval_runtime": 8.8815,
46
+ "eval_samples_per_second": 108.765,
47
+ "eval_steps_per_second": 6.868,
48
+ "step": 1000
49
+ },
50
+ {
51
+ "epoch": 10.64,
52
+ "learning_rate": 4.468085106382979e-05,
53
+ "loss": 0.0644,
54
+ "step": 1500
55
+ },
56
+ {
57
+ "epoch": 10.64,
58
+ "eval_loss": 0.19797927141189575,
59
+ "eval_macro_f1": 0.49327845224918243,
60
+ "eval_macro_precision": 0.5767520501746984,
61
+ "eval_macro_recall": 0.45487076049782793,
62
+ "eval_micro_f1": 0.6739427012278308,
63
+ "eval_micro_precision": 0.7062187276626162,
64
+ "eval_micro_recall": 0.644487932159165,
65
+ "eval_runtime": 2.8151,
66
+ "eval_samples_per_second": 343.151,
67
+ "eval_steps_per_second": 21.669,
68
+ "step": 1500
69
+ },
70
+ {
71
+ "epoch": 14.18,
72
+ "learning_rate": 4.2907801418439716e-05,
73
+ "loss": 0.0352,
74
+ "step": 2000
75
+ },
76
+ {
77
+ "epoch": 14.18,
78
+ "eval_loss": 0.20916847884655,
79
+ "eval_macro_f1": 0.4970581008166056,
80
+ "eval_macro_precision": 0.5794598382601246,
81
+ "eval_macro_recall": 0.4507358136783414,
82
+ "eval_micro_f1": 0.6827225130890052,
83
+ "eval_micro_precision": 0.7342342342342343,
84
+ "eval_micro_recall": 0.6379647749510763,
85
+ "eval_runtime": 2.8186,
86
+ "eval_samples_per_second": 342.721,
87
+ "eval_steps_per_second": 21.642,
88
+ "step": 2000
89
+ },
90
+ {
91
+ "epoch": 17.73,
92
+ "learning_rate": 4.1134751773049644e-05,
93
+ "loss": 0.0206,
94
+ "step": 2500
95
+ },
96
+ {
97
+ "epoch": 17.73,
98
+ "eval_loss": 0.2218163162469864,
99
+ "eval_macro_f1": 0.5259843138534941,
100
+ "eval_macro_precision": 0.6617788311276105,
101
+ "eval_macro_recall": 0.47650769040572616,
102
+ "eval_micro_f1": 0.7059634608755602,
103
+ "eval_micro_precision": 0.7485380116959064,
104
+ "eval_micro_recall": 0.6679712981082844,
105
+ "eval_runtime": 2.81,
106
+ "eval_samples_per_second": 343.773,
107
+ "eval_steps_per_second": 21.708,
108
+ "step": 2500
109
+ },
110
+ {
111
+ "epoch": 21.28,
112
+ "learning_rate": 3.936170212765958e-05,
113
+ "loss": 0.0122,
114
+ "step": 3000
115
+ },
116
+ {
117
+ "epoch": 21.28,
118
+ "eval_loss": 0.23995181918144226,
119
+ "eval_macro_f1": 0.5252080208777923,
120
+ "eval_macro_precision": 0.6447606443742053,
121
+ "eval_macro_recall": 0.4820232253204122,
122
+ "eval_micro_f1": 0.7052341597796142,
123
+ "eval_micro_precision": 0.7469000729394603,
124
+ "eval_micro_recall": 0.6679712981082844,
125
+ "eval_runtime": 2.8113,
126
+ "eval_samples_per_second": 343.611,
127
+ "eval_steps_per_second": 21.698,
128
+ "step": 3000
129
+ },
130
+ {
131
+ "epoch": 24.82,
132
+ "learning_rate": 3.75886524822695e-05,
133
+ "loss": 0.008,
134
+ "step": 3500
135
+ },
136
+ {
137
+ "epoch": 24.82,
138
+ "eval_loss": 0.25718826055526733,
139
+ "eval_macro_f1": 0.5539441017425312,
140
+ "eval_macro_precision": 0.7083242996515385,
141
+ "eval_macro_recall": 0.4999427418977689,
142
+ "eval_micro_f1": 0.7067209775967414,
143
+ "eval_micro_precision": 0.7367303609341825,
144
+ "eval_micro_recall": 0.6790606653620352,
145
+ "eval_runtime": 2.8104,
146
+ "eval_samples_per_second": 343.718,
147
+ "eval_steps_per_second": 21.705,
148
+ "step": 3500
149
+ },
150
+ {
151
+ "epoch": 28.37,
152
+ "learning_rate": 3.5815602836879437e-05,
153
+ "loss": 0.0057,
154
+ "step": 4000
155
+ },
156
+ {
157
+ "epoch": 28.37,
158
+ "eval_loss": 0.26836735010147095,
159
+ "eval_macro_f1": 0.5549375909054308,
160
+ "eval_macro_precision": 0.6742213004313782,
161
+ "eval_macro_recall": 0.517984206223687,
162
+ "eval_micro_f1": 0.6922293858160842,
163
+ "eval_micro_precision": 0.7213578500707214,
164
+ "eval_micro_recall": 0.6653620352250489,
165
+ "eval_runtime": 2.8131,
166
+ "eval_samples_per_second": 343.392,
167
+ "eval_steps_per_second": 21.684,
168
+ "step": 4000
169
+ },
170
+ {
171
+ "epoch": 31.91,
172
+ "learning_rate": 3.4042553191489365e-05,
173
+ "loss": 0.0044,
174
+ "step": 4500
175
+ },
176
+ {
177
+ "epoch": 31.91,
178
+ "eval_loss": 0.2778957784175873,
179
+ "eval_macro_f1": 0.5565319075579724,
180
+ "eval_macro_precision": 0.7103639870539683,
181
+ "eval_macro_recall": 0.4991815941846965,
182
+ "eval_micro_f1": 0.7065663474692202,
183
+ "eval_micro_precision": 0.7426312005751258,
184
+ "eval_micro_recall": 0.6738421395955643,
185
+ "eval_runtime": 2.8149,
186
+ "eval_samples_per_second": 343.169,
187
+ "eval_steps_per_second": 21.67,
188
+ "step": 4500
189
+ },
190
+ {
191
+ "epoch": 35.46,
192
+ "learning_rate": 3.226950354609929e-05,
193
+ "loss": 0.0031,
194
+ "step": 5000
195
+ },
196
+ {
197
+ "epoch": 35.46,
198
+ "eval_loss": 0.29447928071022034,
199
+ "eval_macro_f1": 0.5603530484048236,
200
+ "eval_macro_precision": 0.7069918743571822,
201
+ "eval_macro_recall": 0.5094178224183427,
202
+ "eval_micro_f1": 0.7012113055181696,
203
+ "eval_micro_precision": 0.7241139680333565,
204
+ "eval_micro_recall": 0.6797129810828441,
205
+ "eval_runtime": 2.8093,
206
+ "eval_samples_per_second": 343.853,
207
+ "eval_steps_per_second": 21.713,
208
+ "step": 5000
209
+ },
210
+ {
211
+ "epoch": 39.01,
212
+ "learning_rate": 3.0496453900709222e-05,
213
+ "loss": 0.0031,
214
+ "step": 5500
215
+ },
216
+ {
217
+ "epoch": 39.01,
218
+ "eval_loss": 0.30334585905075073,
219
+ "eval_macro_f1": 0.5559562958714916,
220
+ "eval_macro_precision": 0.6802693513173442,
221
+ "eval_macro_recall": 0.5132993305341865,
222
+ "eval_micro_f1": 0.7078189300411522,
223
+ "eval_micro_precision": 0.7462039045553145,
224
+ "eval_micro_recall": 0.6731898238747553,
225
+ "eval_runtime": 2.814,
226
+ "eval_samples_per_second": 343.285,
227
+ "eval_steps_per_second": 21.677,
228
+ "step": 5500
229
+ },
230
+ {
231
+ "epoch": 42.55,
232
+ "learning_rate": 2.8723404255319154e-05,
233
+ "loss": 0.0023,
234
+ "step": 6000
235
+ },
236
+ {
237
+ "epoch": 42.55,
238
+ "eval_loss": 0.3123699724674225,
239
+ "eval_macro_f1": 0.5623581591291773,
240
+ "eval_macro_precision": 0.7164115784901167,
241
+ "eval_macro_recall": 0.5082951108653608,
242
+ "eval_micro_f1": 0.701129750085587,
243
+ "eval_micro_precision": 0.7377521613832853,
244
+ "eval_micro_recall": 0.6679712981082844,
245
+ "eval_runtime": 2.8076,
246
+ "eval_samples_per_second": 344.062,
247
+ "eval_steps_per_second": 21.726,
248
+ "step": 6000
249
+ },
250
+ {
251
+ "epoch": 46.1,
252
+ "learning_rate": 2.695035460992908e-05,
253
+ "loss": 0.0024,
254
+ "step": 6500
255
+ },
256
+ {
257
+ "epoch": 46.1,
258
+ "eval_loss": 0.315276563167572,
259
+ "eval_macro_f1": 0.5844872930041309,
260
+ "eval_macro_precision": 0.6789210155524211,
261
+ "eval_macro_recall": 0.5412028544350216,
262
+ "eval_micro_f1": 0.7022849462365591,
263
+ "eval_micro_precision": 0.7241857241857241,
264
+ "eval_micro_recall": 0.6816699282452707,
265
+ "eval_runtime": 2.8361,
266
+ "eval_samples_per_second": 340.607,
267
+ "eval_steps_per_second": 21.508,
268
+ "step": 6500
269
+ },
270
+ {
271
+ "epoch": 49.65,
272
+ "learning_rate": 2.5177304964539007e-05,
273
+ "loss": 0.002,
274
+ "step": 7000
275
+ },
276
+ {
277
+ "epoch": 49.65,
278
+ "eval_loss": 0.3173142075538635,
279
+ "eval_macro_f1": 0.5797973374509848,
280
+ "eval_macro_precision": 0.7081281214068704,
281
+ "eval_macro_recall": 0.5234036689357117,
282
+ "eval_micro_f1": 0.7121682178559119,
283
+ "eval_micro_precision": 0.7551169590643275,
284
+ "eval_micro_recall": 0.6738421395955643,
285
+ "eval_runtime": 2.818,
286
+ "eval_samples_per_second": 342.802,
287
+ "eval_steps_per_second": 21.647,
288
+ "step": 7000
289
+ },
290
+ {
291
+ "epoch": 53.19,
292
+ "learning_rate": 2.340425531914894e-05,
293
+ "loss": 0.0016,
294
+ "step": 7500
295
+ },
296
+ {
297
+ "epoch": 53.19,
298
+ "eval_loss": 0.3481275737285614,
299
+ "eval_macro_f1": 0.5715812887434382,
300
+ "eval_macro_precision": 0.7237539680214637,
301
+ "eval_macro_recall": 0.5104648237002123,
302
+ "eval_micro_f1": 0.703448275862069,
303
+ "eval_micro_precision": 0.7461594732991953,
304
+ "eval_micro_recall": 0.6653620352250489,
305
+ "eval_runtime": 2.8127,
306
+ "eval_samples_per_second": 343.438,
307
+ "eval_steps_per_second": 21.687,
308
+ "step": 7500
309
+ },
310
+ {
311
+ "epoch": 56.74,
312
+ "learning_rate": 2.1631205673758867e-05,
313
+ "loss": 0.0012,
314
+ "step": 8000
315
+ },
316
+ {
317
+ "epoch": 56.74,
318
+ "eval_loss": 0.34702157974243164,
319
+ "eval_macro_f1": 0.5951067347267447,
320
+ "eval_macro_precision": 0.7516773928626448,
321
+ "eval_macro_recall": 0.5430089368236234,
322
+ "eval_micro_f1": 0.6986486486486487,
323
+ "eval_micro_precision": 0.7245970567624387,
324
+ "eval_micro_recall": 0.6744944553163731,
325
+ "eval_runtime": 2.8072,
326
+ "eval_samples_per_second": 344.113,
327
+ "eval_steps_per_second": 21.73,
328
+ "step": 8000
329
+ }
330
+ ],
331
+ "max_steps": 14100,
332
+ "num_train_epochs": 100,
333
+ "total_flos": 3.1977769947832704e+16,
334
+ "trial_name": null,
335
+ "trial_params": null
336
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a28a2c5a1f7f8d2b224f6418522014d5e9f8cb50c87f0b69e1be2aaaa0e7c9d7
3
+ size 3247
vocab.txt ADDED
The diff for this file is too large to render. See raw diff