fenffef commited on
Commit
4fd68db
·
verified ·
1 Parent(s): 1fa6d2b

Upload 10 files

Browse files
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "train_loss": 1.461335124463566,
4
+ "train_runtime": 132050.7556,
5
+ "train_samples": 2823058,
6
+ "train_samples_per_second": 85.514,
7
+ "train_steps_per_second": 0.267
8
+ }
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./exps/rocbart-large-chinese-v2/gec/run_2024/checkpoint-52933",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "BartForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 101,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 1024,
15
+ "decoder_attention_heads": 16,
16
+ "decoder_ffn_dim": 4096,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 12,
19
+ "decoder_start_token_id": 102,
20
+ "dropout": 0.1,
21
+ "early_stopping": true,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 12,
26
+ "eos_token_id": 102,
27
+ "forced_eos_token_id": 102,
28
+ "gradient_checkpointing": false,
29
+ "id2label": {
30
+ "0": "LABEL_0",
31
+ "1": "LABEL_1",
32
+ "2": "LABEL_2"
33
+ },
34
+ "init_std": 0.02,
35
+ "is_encoder_decoder": true,
36
+ "label2id": {
37
+ "LABEL_0": 0,
38
+ "LABEL_1": 1,
39
+ "LABEL_2": 2
40
+ },
41
+ "max_length": 100,
42
+ "max_position_embeddings": 1024,
43
+ "model_type": "bart",
44
+ "no_repeat_ngram_size": 3,
45
+ "normalize_before": false,
46
+ "normalize_embedding": true,
47
+ "num_beams": 4,
48
+ "num_hidden_layers": 12,
49
+ "pad_token_id": 0,
50
+ "scale_embedding": false,
51
+ "task_specific_params": {
52
+ "summarization": {
53
+ "length_penalty": 1.0,
54
+ "max_length": 128,
55
+ "min_length": 12,
56
+ "num_beams": 4
57
+ },
58
+ "summarization_cnn": {
59
+ "length_penalty": 2.0,
60
+ "max_length": 142,
61
+ "min_length": 56,
62
+ "num_beams": 4
63
+ },
64
+ "summarization_xsum": {
65
+ "length_penalty": 1.0,
66
+ "max_length": 62,
67
+ "min_length": 11,
68
+ "num_beams": 6
69
+ }
70
+ },
71
+ "tokenizer_class": "BertTokenizer",
72
+ "torch_dtype": "float32",
73
+ "transformers_version": "4.31.0",
74
+ "use_cache": true,
75
+ "vocab_size": 51271
76
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 101,
4
+ "decoder_start_token_id": 102,
5
+ "early_stopping": true,
6
+ "eos_token_id": 102,
7
+ "forced_eos_token_id": 102,
8
+ "max_length": 100,
9
+ "no_repeat_ngram_size": 3,
10
+ "num_beams": 4,
11
+ "pad_token_id": 0,
12
+ "transformers_version": "4.31.0"
13
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e621ab7d0e8114564ade0819cc78461764fdc8d155a9814c4ae234e1d5e2d8ec
3
+ size 1629662401
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "clean_up_tokenization_spaces": true,
3
+ "cls_token": "[CLS]",
4
+ "do_basic_tokenize": true,
5
+ "do_lower_case": true,
6
+ "mask_token": "[MASK]",
7
+ "model_max_length": 1000000000000000019884624838656,
8
+ "never_split": null,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "train_loss": 1.461335124463566,
4
+ "train_runtime": 132050.7556,
5
+ "train_samples": 2823058,
6
+ "train_samples_per_second": 85.514,
7
+ "train_steps_per_second": 0.267
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,477 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.99988665023095,
5
+ "global_step": 35288,
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.06,
12
+ "learning_rate": 2.9589095443210154e-06,
13
+ "loss": 1.4651,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.11,
18
+ "learning_rate": 2.9178190886420315e-06,
19
+ "loss": 1.4652,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.17,
24
+ "learning_rate": 2.876728632963047e-06,
25
+ "loss": 1.4649,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.23,
30
+ "learning_rate": 2.8356381772840626e-06,
31
+ "loss": 1.4647,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.28,
36
+ "learning_rate": 2.7945477216050783e-06,
37
+ "loss": 1.4646,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.34,
42
+ "learning_rate": 2.753457265926094e-06,
43
+ "loss": 1.4648,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.4,
48
+ "learning_rate": 2.7123668102471093e-06,
49
+ "loss": 1.4643,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.45,
54
+ "learning_rate": 2.6712763545681255e-06,
55
+ "loss": 1.4643,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.51,
60
+ "learning_rate": 2.6301858988891408e-06,
61
+ "loss": 1.464,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 0.57,
66
+ "learning_rate": 2.589095443210156e-06,
67
+ "loss": 1.4639,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 0.62,
72
+ "learning_rate": 2.5480049875311722e-06,
73
+ "loss": 1.4642,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 0.68,
78
+ "learning_rate": 2.506914531852188e-06,
79
+ "loss": 1.4637,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 0.74,
84
+ "learning_rate": 2.4658240761732033e-06,
85
+ "loss": 1.4639,
86
+ "step": 6500
87
+ },
88
+ {
89
+ "epoch": 0.79,
90
+ "learning_rate": 2.424733620494219e-06,
91
+ "loss": 1.4639,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 0.85,
96
+ "learning_rate": 2.3836431648152347e-06,
97
+ "loss": 1.4638,
98
+ "step": 7500
99
+ },
100
+ {
101
+ "epoch": 0.91,
102
+ "learning_rate": 2.34255270913625e-06,
103
+ "loss": 1.4637,
104
+ "step": 8000
105
+ },
106
+ {
107
+ "epoch": 0.96,
108
+ "learning_rate": 2.301462253457266e-06,
109
+ "loss": 1.4629,
110
+ "step": 8500
111
+ },
112
+ {
113
+ "epoch": 1.0,
114
+ "eval_loss": 1.4708495140075684,
115
+ "eval_runtime": 265.1581,
116
+ "eval_samples_per_second": 107.54,
117
+ "eval_steps_per_second": 1.346,
118
+ "step": 8822
119
+ },
120
+ {
121
+ "epoch": 1.02,
122
+ "learning_rate": 2.2603717977782815e-06,
123
+ "loss": 1.4631,
124
+ "step": 9000
125
+ },
126
+ {
127
+ "epoch": 1.08,
128
+ "learning_rate": 2.219281342099297e-06,
129
+ "loss": 1.4621,
130
+ "step": 9500
131
+ },
132
+ {
133
+ "epoch": 1.13,
134
+ "learning_rate": 2.178190886420313e-06,
135
+ "loss": 1.4621,
136
+ "step": 10000
137
+ },
138
+ {
139
+ "epoch": 1.19,
140
+ "learning_rate": 2.1371004307413287e-06,
141
+ "loss": 1.4617,
142
+ "step": 10500
143
+ },
144
+ {
145
+ "epoch": 1.25,
146
+ "learning_rate": 2.096009975062344e-06,
147
+ "loss": 1.4618,
148
+ "step": 11000
149
+ },
150
+ {
151
+ "epoch": 1.3,
152
+ "learning_rate": 2.0549195193833597e-06,
153
+ "loss": 1.4622,
154
+ "step": 11500
155
+ },
156
+ {
157
+ "epoch": 1.36,
158
+ "learning_rate": 2.0138290637043754e-06,
159
+ "loss": 1.4617,
160
+ "step": 12000
161
+ },
162
+ {
163
+ "epoch": 1.42,
164
+ "learning_rate": 1.972738608025391e-06,
165
+ "loss": 1.4621,
166
+ "step": 12500
167
+ },
168
+ {
169
+ "epoch": 1.47,
170
+ "learning_rate": 1.931648152346407e-06,
171
+ "loss": 1.4615,
172
+ "step": 13000
173
+ },
174
+ {
175
+ "epoch": 1.53,
176
+ "learning_rate": 1.8905576966674222e-06,
177
+ "loss": 1.4615,
178
+ "step": 13500
179
+ },
180
+ {
181
+ "epoch": 1.59,
182
+ "learning_rate": 1.8494672409884385e-06,
183
+ "loss": 1.4616,
184
+ "step": 14000
185
+ },
186
+ {
187
+ "epoch": 1.64,
188
+ "learning_rate": 1.8083767853094543e-06,
189
+ "loss": 1.4615,
190
+ "step": 14500
191
+ },
192
+ {
193
+ "epoch": 1.7,
194
+ "learning_rate": 1.7672863296304691e-06,
195
+ "loss": 1.4619,
196
+ "step": 15000
197
+ },
198
+ {
199
+ "epoch": 1.76,
200
+ "learning_rate": 1.7261958739514849e-06,
201
+ "loss": 1.4613,
202
+ "step": 15500
203
+ },
204
+ {
205
+ "epoch": 1.81,
206
+ "learning_rate": 1.6851054182725004e-06,
207
+ "loss": 1.4613,
208
+ "step": 16000
209
+ },
210
+ {
211
+ "epoch": 1.87,
212
+ "learning_rate": 1.6440149625935161e-06,
213
+ "loss": 1.4615,
214
+ "step": 16500
215
+ },
216
+ {
217
+ "epoch": 1.93,
218
+ "learning_rate": 1.6029245069145325e-06,
219
+ "loss": 1.4612,
220
+ "step": 17000
221
+ },
222
+ {
223
+ "epoch": 1.98,
224
+ "learning_rate": 1.5618340512355474e-06,
225
+ "loss": 1.4613,
226
+ "step": 17500
227
+ },
228
+ {
229
+ "epoch": 2.0,
230
+ "eval_loss": 1.4701755046844482,
231
+ "eval_runtime": 263.8925,
232
+ "eval_samples_per_second": 108.055,
233
+ "eval_steps_per_second": 1.353,
234
+ "step": 17644
235
+ },
236
+ {
237
+ "epoch": 2.04,
238
+ "learning_rate": 1.520743595556563e-06,
239
+ "loss": 1.4606,
240
+ "step": 18000
241
+ },
242
+ {
243
+ "epoch": 2.1,
244
+ "learning_rate": 1.479653139877579e-06,
245
+ "loss": 1.4607,
246
+ "step": 18500
247
+ },
248
+ {
249
+ "epoch": 2.15,
250
+ "learning_rate": 1.4385626841985943e-06,
251
+ "loss": 1.4602,
252
+ "step": 19000
253
+ },
254
+ {
255
+ "epoch": 2.21,
256
+ "learning_rate": 1.39747222851961e-06,
257
+ "loss": 1.4603,
258
+ "step": 19500
259
+ },
260
+ {
261
+ "epoch": 2.27,
262
+ "learning_rate": 1.3563817728406258e-06,
263
+ "loss": 1.4602,
264
+ "step": 20000
265
+ },
266
+ {
267
+ "epoch": 2.32,
268
+ "learning_rate": 1.3152913171616415e-06,
269
+ "loss": 1.4602,
270
+ "step": 20500
271
+ },
272
+ {
273
+ "epoch": 2.38,
274
+ "learning_rate": 1.2742008614826568e-06,
275
+ "loss": 1.46,
276
+ "step": 21000
277
+ },
278
+ {
279
+ "epoch": 2.44,
280
+ "learning_rate": 1.233110405803673e-06,
281
+ "loss": 1.4599,
282
+ "step": 21500
283
+ },
284
+ {
285
+ "epoch": 2.49,
286
+ "learning_rate": 1.1920199501246883e-06,
287
+ "loss": 1.4602,
288
+ "step": 22000
289
+ },
290
+ {
291
+ "epoch": 2.55,
292
+ "learning_rate": 1.150929494445704e-06,
293
+ "loss": 1.46,
294
+ "step": 22500
295
+ },
296
+ {
297
+ "epoch": 2.61,
298
+ "learning_rate": 1.1098390387667197e-06,
299
+ "loss": 1.4601,
300
+ "step": 23000
301
+ },
302
+ {
303
+ "epoch": 2.66,
304
+ "learning_rate": 1.0687485830877354e-06,
305
+ "loss": 1.46,
306
+ "step": 23500
307
+ },
308
+ {
309
+ "epoch": 2.72,
310
+ "learning_rate": 1.0276581274087507e-06,
311
+ "loss": 1.4601,
312
+ "step": 24000
313
+ },
314
+ {
315
+ "epoch": 2.78,
316
+ "learning_rate": 9.865676717297667e-07,
317
+ "loss": 1.46,
318
+ "step": 24500
319
+ },
320
+ {
321
+ "epoch": 2.83,
322
+ "learning_rate": 9.454772160507822e-07,
323
+ "loss": 1.4601,
324
+ "step": 25000
325
+ },
326
+ {
327
+ "epoch": 2.89,
328
+ "learning_rate": 9.043867603717977e-07,
329
+ "loss": 1.4598,
330
+ "step": 25500
331
+ },
332
+ {
333
+ "epoch": 2.95,
334
+ "learning_rate": 8.632963046928134e-07,
335
+ "loss": 1.4607,
336
+ "step": 26000
337
+ },
338
+ {
339
+ "epoch": 3.0,
340
+ "eval_loss": 1.4697959423065186,
341
+ "eval_runtime": 265.5519,
342
+ "eval_samples_per_second": 107.38,
343
+ "eval_steps_per_second": 1.344,
344
+ "step": 26466
345
+ },
346
+ {
347
+ "epoch": 3.0,
348
+ "learning_rate": 8.22205849013829e-07,
349
+ "loss": 1.4597,
350
+ "step": 26500
351
+ },
352
+ {
353
+ "epoch": 3.06,
354
+ "learning_rate": 7.811153933348447e-07,
355
+ "loss": 1.4589,
356
+ "step": 27000
357
+ },
358
+ {
359
+ "epoch": 3.12,
360
+ "learning_rate": 7.400249376558603e-07,
361
+ "loss": 1.4592,
362
+ "step": 27500
363
+ },
364
+ {
365
+ "epoch": 3.17,
366
+ "learning_rate": 6.989344819768761e-07,
367
+ "loss": 1.4594,
368
+ "step": 28000
369
+ },
370
+ {
371
+ "epoch": 3.23,
372
+ "learning_rate": 6.578440262978915e-07,
373
+ "loss": 1.4595,
374
+ "step": 28500
375
+ },
376
+ {
377
+ "epoch": 3.29,
378
+ "learning_rate": 6.167535706189074e-07,
379
+ "loss": 1.4593,
380
+ "step": 29000
381
+ },
382
+ {
383
+ "epoch": 3.34,
384
+ "learning_rate": 5.756631149399229e-07,
385
+ "loss": 1.4595,
386
+ "step": 29500
387
+ },
388
+ {
389
+ "epoch": 3.4,
390
+ "learning_rate": 5.345726592609387e-07,
391
+ "loss": 1.4592,
392
+ "step": 30000
393
+ },
394
+ {
395
+ "epoch": 3.46,
396
+ "learning_rate": 4.934822035819541e-07,
397
+ "loss": 1.4595,
398
+ "step": 30500
399
+ },
400
+ {
401
+ "epoch": 3.51,
402
+ "learning_rate": 4.5239174790297e-07,
403
+ "loss": 1.4594,
404
+ "step": 31000
405
+ },
406
+ {
407
+ "epoch": 3.57,
408
+ "learning_rate": 4.113012922239855e-07,
409
+ "loss": 1.4591,
410
+ "step": 31500
411
+ },
412
+ {
413
+ "epoch": 3.63,
414
+ "learning_rate": 3.70210836545001e-07,
415
+ "loss": 1.4593,
416
+ "step": 32000
417
+ },
418
+ {
419
+ "epoch": 3.68,
420
+ "learning_rate": 3.2912038086601684e-07,
421
+ "loss": 1.4595,
422
+ "step": 32500
423
+ },
424
+ {
425
+ "epoch": 3.74,
426
+ "learning_rate": 2.880299251870323e-07,
427
+ "loss": 1.4595,
428
+ "step": 33000
429
+ },
430
+ {
431
+ "epoch": 3.8,
432
+ "learning_rate": 2.4693946950804813e-07,
433
+ "loss": 1.4598,
434
+ "step": 33500
435
+ },
436
+ {
437
+ "epoch": 3.85,
438
+ "learning_rate": 2.0584901382906362e-07,
439
+ "loss": 1.4593,
440
+ "step": 34000
441
+ },
442
+ {
443
+ "epoch": 3.91,
444
+ "learning_rate": 1.6475855815007943e-07,
445
+ "loss": 1.459,
446
+ "step": 34500
447
+ },
448
+ {
449
+ "epoch": 3.97,
450
+ "learning_rate": 1.2366810247109492e-07,
451
+ "loss": 1.4596,
452
+ "step": 35000
453
+ },
454
+ {
455
+ "epoch": 4.0,
456
+ "eval_loss": 1.4696872234344482,
457
+ "eval_runtime": 266.0179,
458
+ "eval_samples_per_second": 107.192,
459
+ "eval_steps_per_second": 1.342,
460
+ "step": 35288
461
+ },
462
+ {
463
+ "epoch": 4.0,
464
+ "step": 35288,
465
+ "total_flos": 1.100521644337791e+18,
466
+ "train_loss": 1.461335124463566,
467
+ "train_runtime": 132050.7556,
468
+ "train_samples_per_second": 85.514,
469
+ "train_steps_per_second": 0.267
470
+ }
471
+ ],
472
+ "max_steps": 35288,
473
+ "num_train_epochs": 4,
474
+ "total_flos": 1.100521644337791e+18,
475
+ "trial_name": null,
476
+ "trial_params": null
477
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76872e4782c53a6385665abd47337e525eaeb41bb91ec119407979433c657821
3
+ size 4207
vocab.txt ADDED
The diff for this file is too large to render. See raw diff