ZhangYunchenY commited on
Commit
aa152f2
1 Parent(s): 8d80afb

[Model] roberta-base-squad

Browse files
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/mnt/lustre/zhangyunchen/transformers/roberta-base",
3
+ "architectures": [
4
+ "RobertaForQuestionAnswering"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "torch_dtype": "float32",
23
+ "transformers_version": "4.16.2",
24
+ "type_vocab_size": 1,
25
+ "use_cache": true,
26
+ "vocab_size": 50265
27
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a19302120e85cc2a15a7cd14e7448cd959fd37a4a08663247cbbadab94bbf14
3
+ size 992578299
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:337e9be487ab59776f867f5d26abc99d452bc79654b0c0fa3ea536805e2c6829
3
+ size 496316479
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e231ee29a21d397cfa238d58290590d66400b70e77fc6adba92f5cc29f26067d
3
+ size 14659
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69783cee06c8112fea4525a0a2039742bffb21a0e3fa020517f5e6f66fcb72a0
3
+ size 623
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "pad_token": "<pad>", "cls_token": "<s>", "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false}}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "add_prefix_space": false, "errors": "replace", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": "<mask>", "trim_offsets": true, "use_fase": true, "special_tokens_map_file": null, "name_or_path": "/mnt/lustre/zhangyunchen/transformers/roberta-base", "tokenizer_class": "RobertaTokenizer"}
trainer_state.json ADDED
@@ -0,0 +1,328 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.696531791907514,
5
+ "global_step": 26000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.18,
12
+ "learning_rate": 6.020469596628538e-06,
13
+ "loss": 3.3245,
14
+ "step": 1000
15
+ },
16
+ {
17
+ "epoch": 0.18,
18
+ "eval_exact_match": 69.29990539262063,
19
+ "eval_f1": 79.5326236755824,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.36,
24
+ "learning_rate": 9.869710596102849e-06,
25
+ "loss": 1.3078,
26
+ "step": 2000
27
+ },
28
+ {
29
+ "epoch": 0.36,
30
+ "eval_exact_match": 79.11069063386944,
31
+ "eval_f1": 87.17731552605335,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.54,
36
+ "learning_rate": 9.485376071332488e-06,
37
+ "loss": 1.0887,
38
+ "step": 3000
39
+ },
40
+ {
41
+ "epoch": 0.54,
42
+ "eval_exact_match": 81.40964995269631,
43
+ "eval_f1": 88.70193797570302,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.72,
48
+ "learning_rate": 9.101041546562128e-06,
49
+ "loss": 1.0038,
50
+ "step": 4000
51
+ },
52
+ {
53
+ "epoch": 0.72,
54
+ "eval_exact_match": 82.98959318826869,
55
+ "eval_f1": 90.04823015326417,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.9,
60
+ "learning_rate": 8.716707021791769e-06,
61
+ "loss": 0.9564,
62
+ "step": 5000
63
+ },
64
+ {
65
+ "epoch": 0.9,
66
+ "eval_exact_match": 83.6802270577105,
67
+ "eval_f1": 90.58225811526646,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 1.08,
72
+ "learning_rate": 8.332372497021408e-06,
73
+ "loss": 0.8923,
74
+ "step": 6000
75
+ },
76
+ {
77
+ "epoch": 1.08,
78
+ "eval_exact_match": 83.91674550614948,
79
+ "eval_f1": 90.87384438135182,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 1.26,
84
+ "learning_rate": 7.948037972251048e-06,
85
+ "loss": 0.8145,
86
+ "step": 7000
87
+ },
88
+ {
89
+ "epoch": 1.26,
90
+ "eval_exact_match": 84.65468306527909,
91
+ "eval_f1": 91.29700250894122,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 1.45,
96
+ "learning_rate": 7.563703447480688e-06,
97
+ "loss": 0.816,
98
+ "step": 8000
99
+ },
100
+ {
101
+ "epoch": 1.45,
102
+ "eval_exact_match": 84.75875118259225,
103
+ "eval_f1": 91.47394559853937,
104
+ "step": 8000
105
+ },
106
+ {
107
+ "epoch": 1.63,
108
+ "learning_rate": 7.179368922710328e-06,
109
+ "loss": 0.7832,
110
+ "step": 9000
111
+ },
112
+ {
113
+ "epoch": 1.63,
114
+ "eval_exact_match": 85.36423841059603,
115
+ "eval_f1": 91.79656306794526,
116
+ "step": 9000
117
+ },
118
+ {
119
+ "epoch": 1.81,
120
+ "learning_rate": 6.795034397939968e-06,
121
+ "loss": 0.8032,
122
+ "step": 10000
123
+ },
124
+ {
125
+ "epoch": 1.81,
126
+ "eval_exact_match": 84.95742667928099,
127
+ "eval_f1": 91.6847055956044,
128
+ "step": 10000
129
+ },
130
+ {
131
+ "epoch": 1.99,
132
+ "learning_rate": 6.410699873169607e-06,
133
+ "loss": 0.7782,
134
+ "step": 11000
135
+ },
136
+ {
137
+ "epoch": 1.99,
138
+ "eval_exact_match": 85.51561021759697,
139
+ "eval_f1": 91.86846688872524,
140
+ "step": 11000
141
+ },
142
+ {
143
+ "epoch": 2.17,
144
+ "learning_rate": 6.026365348399247e-06,
145
+ "loss": 0.6575,
146
+ "step": 12000
147
+ },
148
+ {
149
+ "epoch": 2.17,
150
+ "eval_exact_match": 85.4399243140965,
151
+ "eval_f1": 91.69142033782958,
152
+ "step": 12000
153
+ },
154
+ {
155
+ "epoch": 2.35,
156
+ "learning_rate": 5.6420308236288864e-06,
157
+ "loss": 0.6717,
158
+ "step": 13000
159
+ },
160
+ {
161
+ "epoch": 2.35,
162
+ "eval_exact_match": 85.26963103122044,
163
+ "eval_f1": 91.80508749286186,
164
+ "step": 13000
165
+ },
166
+ {
167
+ "epoch": 2.53,
168
+ "learning_rate": 5.257696298858527e-06,
169
+ "loss": 0.6647,
170
+ "step": 14000
171
+ },
172
+ {
173
+ "epoch": 2.53,
174
+ "eval_exact_match": 85.42100283822138,
175
+ "eval_f1": 91.80456676699409,
176
+ "step": 14000
177
+ },
178
+ {
179
+ "epoch": 2.71,
180
+ "learning_rate": 4.873361774088167e-06,
181
+ "loss": 0.6583,
182
+ "step": 15000
183
+ },
184
+ {
185
+ "epoch": 2.71,
186
+ "eval_exact_match": 85.72374645222327,
187
+ "eval_f1": 92.1390421209737,
188
+ "step": 15000
189
+ },
190
+ {
191
+ "epoch": 2.89,
192
+ "learning_rate": 4.4890272493178065e-06,
193
+ "loss": 0.652,
194
+ "step": 16000
195
+ },
196
+ {
197
+ "epoch": 2.89,
198
+ "eval_exact_match": 85.73320719016083,
199
+ "eval_f1": 92.03736167459341,
200
+ "step": 16000
201
+ },
202
+ {
203
+ "epoch": 3.07,
204
+ "learning_rate": 4.104692724547447e-06,
205
+ "loss": 0.6102,
206
+ "step": 17000
207
+ },
208
+ {
209
+ "epoch": 3.07,
210
+ "eval_exact_match": 85.26017029328288,
211
+ "eval_f1": 91.89783353916026,
212
+ "step": 17000
213
+ },
214
+ {
215
+ "epoch": 3.25,
216
+ "learning_rate": 3.720358199777086e-06,
217
+ "loss": 0.5709,
218
+ "step": 18000
219
+ },
220
+ {
221
+ "epoch": 3.25,
222
+ "eval_exact_match": 85.47776726584674,
223
+ "eval_f1": 91.91999508342143,
224
+ "step": 18000
225
+ },
226
+ {
227
+ "epoch": 3.43,
228
+ "learning_rate": 3.3360236750067258e-06,
229
+ "loss": 0.5745,
230
+ "step": 19000
231
+ },
232
+ {
233
+ "epoch": 3.43,
234
+ "eval_exact_match": 85.3926206244087,
235
+ "eval_f1": 91.91182535908425,
236
+ "step": 19000
237
+ },
238
+ {
239
+ "epoch": 3.61,
240
+ "learning_rate": 2.951689150236366e-06,
241
+ "loss": 0.564,
242
+ "step": 20000
243
+ },
244
+ {
245
+ "epoch": 3.61,
246
+ "eval_exact_match": 85.45884578997162,
247
+ "eval_f1": 91.93228761137887,
248
+ "step": 20000
249
+ },
250
+ {
251
+ "epoch": 3.79,
252
+ "learning_rate": 2.567354625466006e-06,
253
+ "loss": 0.5774,
254
+ "step": 21000
255
+ },
256
+ {
257
+ "epoch": 3.79,
258
+ "eval_exact_match": 85.81835383159887,
259
+ "eval_f1": 92.14088561726169,
260
+ "step": 21000
261
+ },
262
+ {
263
+ "epoch": 3.97,
264
+ "learning_rate": 2.1830201006956455e-06,
265
+ "loss": 0.5667,
266
+ "step": 22000
267
+ },
268
+ {
269
+ "epoch": 3.97,
270
+ "eval_exact_match": 85.71428571428571,
271
+ "eval_f1": 92.11242630306818,
272
+ "step": 22000
273
+ },
274
+ {
275
+ "epoch": 4.15,
276
+ "learning_rate": 1.7986855759252856e-06,
277
+ "loss": 0.514,
278
+ "step": 23000
279
+ },
280
+ {
281
+ "epoch": 4.15,
282
+ "eval_exact_match": 85.88457899716178,
283
+ "eval_f1": 92.24928398942093,
284
+ "step": 23000
285
+ },
286
+ {
287
+ "epoch": 4.34,
288
+ "learning_rate": 1.4143510511549255e-06,
289
+ "loss": 0.5074,
290
+ "step": 24000
291
+ },
292
+ {
293
+ "epoch": 4.34,
294
+ "eval_exact_match": 85.88457899716178,
295
+ "eval_f1": 92.20790215772705,
296
+ "step": 24000
297
+ },
298
+ {
299
+ "epoch": 4.52,
300
+ "learning_rate": 1.0300165263845651e-06,
301
+ "loss": 0.5195,
302
+ "step": 25000
303
+ },
304
+ {
305
+ "epoch": 4.52,
306
+ "eval_exact_match": 85.72374645222327,
307
+ "eval_f1": 92.19634199090316,
308
+ "step": 25000
309
+ },
310
+ {
311
+ "epoch": 4.7,
312
+ "learning_rate": 6.45682001614205e-07,
313
+ "loss": 0.5049,
314
+ "step": 26000
315
+ },
316
+ {
317
+ "epoch": 4.7,
318
+ "eval_exact_match": 85.82781456953643,
319
+ "eval_f1": 92.25386967488927,
320
+ "step": 26000
321
+ }
322
+ ],
323
+ "max_steps": 27680,
324
+ "num_train_epochs": 5,
325
+ "total_flos": 8.151831697947034e+16,
326
+ "trial_name": null,
327
+ "trial_params": null
328
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a3ef0f04653add10e4bb045d6d4d8dcec03ede13353f66ef1e4fb296658950c
3
+ size 3055
vocab.json ADDED
The diff for this file is too large to render. See raw diff