Trevor Johnson commited on
Commit
4c0b9cd
·
1 Parent(s): 4fe23cb
config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "sshleifer/distilbart-xsum-6-6",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": false,
8
+ "architectures": [
9
+ "BartForConditionalGeneration"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bos_token_id": 0,
13
+ "classif_dropout": 0.0,
14
+ "classifier_dropout": 0.0,
15
+ "d_model": 1024,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 6,
20
+ "decoder_start_token_id": 2,
21
+ "dropout": 0.1,
22
+ "early_stopping": true,
23
+ "encoder_attention_heads": 16,
24
+ "encoder_ffn_dim": 4096,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 6,
27
+ "eos_token_id": 2,
28
+ "eos_token_ids": [
29
+ 2
30
+ ],
31
+ "extra_pos_embeddings": 2,
32
+ "forced_eos_token_id": 2,
33
+ "gradient_checkpointing": false,
34
+ "id2label": {
35
+ "0": "LABEL_0",
36
+ "1": "LABEL_1",
37
+ "2": "LABEL_2"
38
+ },
39
+ "init_std": 0.02,
40
+ "is_encoder_decoder": true,
41
+ "label2id": {
42
+ "LABEL_0": 0,
43
+ "LABEL_1": 1,
44
+ "LABEL_2": 2
45
+ },
46
+ "length_penalty": 0.5,
47
+ "max_length": 62,
48
+ "max_position_embeddings": 1024,
49
+ "min_length": 11,
50
+ "model_type": "bart",
51
+ "no_repeat_ngram_size": 3,
52
+ "normalize_before": false,
53
+ "normalize_embedding": true,
54
+ "num_beams": 6,
55
+ "num_hidden_layers": 6,
56
+ "output_past": true,
57
+ "pad_token_id": 1,
58
+ "prefix": " ",
59
+ "replacing_rate": 0,
60
+ "save_step": 38,
61
+ "scale_embedding": false,
62
+ "static_position_embeddings": false,
63
+ "student_decoder_layers": null,
64
+ "student_encoder_layers": null,
65
+ "torch_dtype": "float32",
66
+ "transformers_version": "4.20.1",
67
+ "use_cache": true,
68
+ "vocab_size": 50264
69
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a397898d81e22857171dacbdb9b6d1e32d402bdfc2af818f48c3453aafbe44c
3
+ size 920019705
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aac94f20286386763cd0607c70c493ccc282a8e35a572ac6f8afc537005d3215
3
+ size 14503
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1da8ec0372377c998ad6584c307b7a13226637731a2fab52f87d03b089e5ff21
3
+ size 623
special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": {
4
+ "__type": "AddedToken",
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false
10
+ },
11
+ "cls_token": {
12
+ "__type": "AddedToken",
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "eos_token": {
20
+ "__type": "AddedToken",
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ },
27
+ "errors": "replace",
28
+ "mask_token": {
29
+ "__type": "AddedToken",
30
+ "content": "<mask>",
31
+ "lstrip": true,
32
+ "normalized": true,
33
+ "rstrip": false,
34
+ "single_word": false
35
+ },
36
+ "model_max_length": 1024,
37
+ "name_or_path": "sshleifer/distilbart-xsum-6-6",
38
+ "pad_token": {
39
+ "__type": "AddedToken",
40
+ "content": "<pad>",
41
+ "lstrip": false,
42
+ "normalized": true,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "sep_token": {
47
+ "__type": "AddedToken",
48
+ "content": "</s>",
49
+ "lstrip": false,
50
+ "normalized": true,
51
+ "rstrip": false,
52
+ "single_word": false
53
+ },
54
+ "special_tokens_map_file": null,
55
+ "tokenizer_class": "BartTokenizer",
56
+ "trim_offsets": true,
57
+ "unk_token": {
58
+ "__type": "AddedToken",
59
+ "content": "<unk>",
60
+ "lstrip": false,
61
+ "normalized": true,
62
+ "rstrip": false,
63
+ "single_word": false
64
+ }
65
+ }
trainer_state.json ADDED
@@ -0,0 +1,312 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "global_step": 22500,
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.07,
12
+ "learning_rate": 1.9555555555555557e-05,
13
+ "loss": 2.9226,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.13,
18
+ "learning_rate": 1.9111111111111113e-05,
19
+ "loss": 2.9051,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.2,
24
+ "learning_rate": 1.866666666666667e-05,
25
+ "loss": 2.9092,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.27,
30
+ "learning_rate": 1.8222222222222224e-05,
31
+ "loss": 2.915,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.33,
36
+ "learning_rate": 1.7777777777777777e-05,
37
+ "loss": 2.9285,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.4,
42
+ "learning_rate": 1.7333333333333336e-05,
43
+ "loss": 2.8844,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.47,
48
+ "learning_rate": 1.688888888888889e-05,
49
+ "loss": 2.9206,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.53,
54
+ "learning_rate": 1.6444444444444444e-05,
55
+ "loss": 2.9069,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.6,
60
+ "learning_rate": 1.6000000000000003e-05,
61
+ "loss": 2.9354,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 0.67,
66
+ "learning_rate": 1.555555555555556e-05,
67
+ "loss": 2.9054,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 0.73,
72
+ "learning_rate": 1.5111111111111112e-05,
73
+ "loss": 2.9195,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 0.8,
78
+ "learning_rate": 1.4666666666666666e-05,
79
+ "loss": 2.9421,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 0.87,
84
+ "learning_rate": 1.4222222222222224e-05,
85
+ "loss": 2.9313,
86
+ "step": 6500
87
+ },
88
+ {
89
+ "epoch": 0.93,
90
+ "learning_rate": 1.377777777777778e-05,
91
+ "loss": 2.9654,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 1.0,
96
+ "learning_rate": 1.3333333333333333e-05,
97
+ "loss": 2.9579,
98
+ "step": 7500
99
+ },
100
+ {
101
+ "epoch": 1.0,
102
+ "eval_gen_len": 19.442,
103
+ "eval_loss": 3.445647954940796,
104
+ "eval_rouge1": 18.8989,
105
+ "eval_rouge2": 5.0689,
106
+ "eval_rougeL": 15.4851,
107
+ "eval_rougeLsum": 16.3155,
108
+ "eval_runtime": 673.7677,
109
+ "eval_samples_per_second": 5.937,
110
+ "eval_steps_per_second": 0.742,
111
+ "step": 7500
112
+ },
113
+ {
114
+ "epoch": 1.07,
115
+ "learning_rate": 1.288888888888889e-05,
116
+ "loss": 2.5931,
117
+ "step": 8000
118
+ },
119
+ {
120
+ "epoch": 1.13,
121
+ "learning_rate": 1.2444444444444446e-05,
122
+ "loss": 2.5813,
123
+ "step": 8500
124
+ },
125
+ {
126
+ "epoch": 1.2,
127
+ "learning_rate": 1.2e-05,
128
+ "loss": 2.6225,
129
+ "step": 9000
130
+ },
131
+ {
132
+ "epoch": 1.27,
133
+ "learning_rate": 1.1555555555555556e-05,
134
+ "loss": 2.6269,
135
+ "step": 9500
136
+ },
137
+ {
138
+ "epoch": 1.33,
139
+ "learning_rate": 1.1111111111111113e-05,
140
+ "loss": 2.6208,
141
+ "step": 10000
142
+ },
143
+ {
144
+ "epoch": 1.4,
145
+ "learning_rate": 1.0666666666666667e-05,
146
+ "loss": 2.6287,
147
+ "step": 10500
148
+ },
149
+ {
150
+ "epoch": 1.47,
151
+ "learning_rate": 1.0222222222222223e-05,
152
+ "loss": 2.6569,
153
+ "step": 11000
154
+ },
155
+ {
156
+ "epoch": 1.53,
157
+ "learning_rate": 9.777777777777779e-06,
158
+ "loss": 2.6644,
159
+ "step": 11500
160
+ },
161
+ {
162
+ "epoch": 1.6,
163
+ "learning_rate": 9.333333333333334e-06,
164
+ "loss": 2.6725,
165
+ "step": 12000
166
+ },
167
+ {
168
+ "epoch": 1.67,
169
+ "learning_rate": 8.888888888888888e-06,
170
+ "loss": 2.6747,
171
+ "step": 12500
172
+ },
173
+ {
174
+ "epoch": 1.73,
175
+ "learning_rate": 8.444444444444446e-06,
176
+ "loss": 2.7015,
177
+ "step": 13000
178
+ },
179
+ {
180
+ "epoch": 1.8,
181
+ "learning_rate": 8.000000000000001e-06,
182
+ "loss": 2.701,
183
+ "step": 13500
184
+ },
185
+ {
186
+ "epoch": 1.87,
187
+ "learning_rate": 7.555555555555556e-06,
188
+ "loss": 2.7293,
189
+ "step": 14000
190
+ },
191
+ {
192
+ "epoch": 1.93,
193
+ "learning_rate": 7.111111111111112e-06,
194
+ "loss": 2.7086,
195
+ "step": 14500
196
+ },
197
+ {
198
+ "epoch": 2.0,
199
+ "learning_rate": 6.666666666666667e-06,
200
+ "loss": 2.7437,
201
+ "step": 15000
202
+ },
203
+ {
204
+ "epoch": 2.0,
205
+ "eval_gen_len": 19.113,
206
+ "eval_loss": 3.4790942668914795,
207
+ "eval_rouge1": 18.8818,
208
+ "eval_rouge2": 4.9775,
209
+ "eval_rougeL": 15.4084,
210
+ "eval_rougeLsum": 16.2892,
211
+ "eval_runtime": 647.6595,
212
+ "eval_samples_per_second": 6.176,
213
+ "eval_steps_per_second": 0.772,
214
+ "step": 15000
215
+ },
216
+ {
217
+ "epoch": 2.07,
218
+ "learning_rate": 6.222222222222223e-06,
219
+ "loss": 2.4254,
220
+ "step": 15500
221
+ },
222
+ {
223
+ "epoch": 2.13,
224
+ "learning_rate": 5.777777777777778e-06,
225
+ "loss": 2.5738,
226
+ "step": 16000
227
+ },
228
+ {
229
+ "epoch": 2.2,
230
+ "learning_rate": 5.333333333333334e-06,
231
+ "loss": 2.5708,
232
+ "step": 16500
233
+ },
234
+ {
235
+ "epoch": 2.27,
236
+ "learning_rate": 4.888888888888889e-06,
237
+ "loss": 2.6077,
238
+ "step": 17000
239
+ },
240
+ {
241
+ "epoch": 2.33,
242
+ "learning_rate": 4.444444444444444e-06,
243
+ "loss": 2.5844,
244
+ "step": 17500
245
+ },
246
+ {
247
+ "epoch": 2.4,
248
+ "learning_rate": 4.000000000000001e-06,
249
+ "loss": 2.6059,
250
+ "step": 18000
251
+ },
252
+ {
253
+ "epoch": 2.47,
254
+ "learning_rate": 3.555555555555556e-06,
255
+ "loss": 2.5876,
256
+ "step": 18500
257
+ },
258
+ {
259
+ "epoch": 2.53,
260
+ "learning_rate": 3.1111111111111116e-06,
261
+ "loss": 2.6129,
262
+ "step": 19000
263
+ },
264
+ {
265
+ "epoch": 2.6,
266
+ "learning_rate": 2.666666666666667e-06,
267
+ "loss": 2.5966,
268
+ "step": 19500
269
+ },
270
+ {
271
+ "epoch": 2.67,
272
+ "learning_rate": 2.222222222222222e-06,
273
+ "loss": 2.6078,
274
+ "step": 20000
275
+ },
276
+ {
277
+ "epoch": 2.73,
278
+ "learning_rate": 1.777777777777778e-06,
279
+ "loss": 2.6067,
280
+ "step": 20500
281
+ },
282
+ {
283
+ "epoch": 2.8,
284
+ "learning_rate": 1.3333333333333334e-06,
285
+ "loss": 2.5888,
286
+ "step": 21000
287
+ },
288
+ {
289
+ "epoch": 2.87,
290
+ "learning_rate": 8.88888888888889e-07,
291
+ "loss": 2.5656,
292
+ "step": 21500
293
+ },
294
+ {
295
+ "epoch": 2.93,
296
+ "learning_rate": 4.444444444444445e-07,
297
+ "loss": 2.5685,
298
+ "step": 22000
299
+ },
300
+ {
301
+ "epoch": 3.0,
302
+ "learning_rate": 0.0,
303
+ "loss": 2.5666,
304
+ "step": 22500
305
+ }
306
+ ],
307
+ "max_steps": 22500,
308
+ "num_train_epochs": 3,
309
+ "total_flos": 1.2837568824213504e+17,
310
+ "trial_name": null,
311
+ "trial_params": null
312
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f1696d9af70779663d3fc1a812f0d62a82d457ad81548d3c0c8c2dc6be431fa
3
+ size 3375
vocab.json ADDED
The diff for this file is too large to render. See raw diff