José Ángel González commited on
Commit
494a76a
1 Parent(s): cfa06d4
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: barthez-deft-chimie
9
+ results:
10
+ - task:
11
+ name: Summarization
12
+ type: summarization
13
+ metrics:
14
+ - name: Rouge1
15
+ type: rouge
16
+ value: 31.8947
17
+ ---
18
+
19
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
20
+ should probably proofread and complete it, then remove this comment. -->
21
+
22
+ # barthez-deft-chimie
23
+
24
+ This model is a fine-tuned version of [moussaKam/barthez](https://huggingface.co/moussaKam/barthez) on an unknown dataset.
25
+ It achieves the following results on the evaluation set:
26
+ - Loss: 2.0710
27
+ - Rouge1: 31.8947
28
+ - Rouge2: 16.7563
29
+ - Rougel: 23.5428
30
+ - Rougelsum: 23.4918
31
+ - Gen Len: 38.5256
32
+
33
+ ## Model description
34
+
35
+ More information needed
36
+
37
+ ## Intended uses & limitations
38
+
39
+ More information needed
40
+
41
+ ## Training and evaluation data
42
+
43
+ More information needed
44
+
45
+ ## Training procedure
46
+
47
+ ### Training hyperparameters
48
+
49
+ The following hyperparameters were used during training:
50
+ - learning_rate: 3e-05
51
+ - train_batch_size: 4
52
+ - eval_batch_size: 4
53
+ - seed: 42
54
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
+ - lr_scheduler_type: linear
56
+ - num_epochs: 20.0
57
+ - mixed_precision_training: Native AMP
58
+
59
+ ### Training results
60
+
61
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
62
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
63
+ | 3.8022 | 1.0 | 118 | 2.5491 | 16.8208 | 7.0027 | 13.957 | 14.0479 | 19.1538 |
64
+ | 2.9286 | 2.0 | 236 | 2.3074 | 17.5356 | 7.8717 | 14.4874 | 14.5044 | 19.9487 |
65
+ | 2.5422 | 3.0 | 354 | 2.2322 | 19.6491 | 9.4156 | 15.9467 | 15.9433 | 19.7051 |
66
+ | 2.398 | 4.0 | 472 | 2.1500 | 18.7166 | 9.859 | 15.7535 | 15.8036 | 19.9231 |
67
+ | 2.2044 | 5.0 | 590 | 2.1372 | 19.978 | 10.6235 | 16.1348 | 16.1274 | 19.6154 |
68
+ | 1.9405 | 6.0 | 708 | 2.0992 | 20.226 | 10.551 | 16.6928 | 16.7211 | 19.9744 |
69
+ | 1.8544 | 7.0 | 826 | 2.0841 | 19.8869 | 10.8456 | 16.1072 | 16.097 | 19.8846 |
70
+ | 1.7536 | 8.0 | 944 | 2.0791 | 19.3017 | 9.4921 | 16.1541 | 16.2167 | 19.859 |
71
+ | 1.6914 | 9.0 | 1062 | 2.0710 | 21.3848 | 10.4088 | 17.1963 | 17.2254 | 19.8846 |
72
+ | 1.654 | 10.0 | 1180 | 2.1069 | 22.3811 | 10.7987 | 18.7595 | 18.761 | 19.9231 |
73
+ | 1.5899 | 11.0 | 1298 | 2.0919 | 20.8546 | 10.6958 | 16.8637 | 16.9499 | 19.8077 |
74
+ | 1.4661 | 12.0 | 1416 | 2.1065 | 22.3677 | 11.7472 | 18.262 | 18.3 | 19.9744 |
75
+ | 1.4205 | 13.0 | 1534 | 2.1164 | 20.5845 | 10.7825 | 16.9972 | 17.0216 | 19.9359 |
76
+ | 1.3797 | 14.0 | 1652 | 2.1240 | 22.2561 | 11.303 | 17.5064 | 17.5815 | 19.9744 |
77
+ | 1.3724 | 15.0 | 1770 | 2.1187 | 23.2825 | 11.912 | 18.5208 | 18.5499 | 19.9359 |
78
+ | 1.3404 | 16.0 | 1888 | 2.1394 | 22.1305 | 10.5258 | 17.772 | 17.8202 | 19.9744 |
79
+ | 1.2846 | 17.0 | 2006 | 2.1502 | 21.567 | 11.0557 | 17.2562 | 17.2974 | 20.0 |
80
+ | 1.2871 | 18.0 | 2124 | 2.1572 | 22.5871 | 11.702 | 18.2906 | 18.3826 | 19.9744 |
81
+ | 1.2422 | 19.0 | 2242 | 2.1613 | 23.0935 | 11.6824 | 18.6087 | 18.6777 | 19.9744 |
82
+ | 1.2336 | 20.0 | 2360 | 2.1581 | 22.6789 | 11.4363 | 18.1661 | 18.2346 | 19.9487 |
83
+
84
+
85
+ ### Framework versions
86
+
87
+ - Transformers 4.10.2
88
+ - Pytorch 1.7.1+cu110
89
+ - Datasets 1.11.0
90
+ - Tokenizers 0.10.3
all_results.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_gen_len": 38.5256,
4
+ "eval_loss": 2.0710394382476807,
5
+ "eval_rouge1": 31.8947,
6
+ "eval_rouge2": 16.7563,
7
+ "eval_rougeL": 23.5428,
8
+ "eval_rougeLsum": 23.4918,
9
+ "eval_runtime": 12.584,
10
+ "eval_samples": 78,
11
+ "eval_samples_per_second": 6.198,
12
+ "eval_steps_per_second": 1.589,
13
+ "train_loss": 1.7793180174746757,
14
+ "train_runtime": 373.3963,
15
+ "train_samples": 469,
16
+ "train_samples_per_second": 25.121,
17
+ "train_steps_per_second": 6.32
18
+ }
config.json ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "moussaKam/barthez",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": true,
7
+ "architectures": [
8
+ "MBartForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 0,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 768,
15
+ "decoder_attention_heads": 12,
16
+ "decoder_ffn_dim": 3072,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 6,
19
+ "decoder_start_token_id": 2,
20
+ "do_blenderbot_90_layernorm": false,
21
+ "dropout": 0.1,
22
+ "early_stopping": true,
23
+ "encoder_attention_heads": 12,
24
+ "encoder_ffn_dim": 3072,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 6,
27
+ "eos_token_id": 2,
28
+ "extra_pos_embeddings": 2,
29
+ "force_bos_token_to_be_generated": false,
30
+ "forced_eos_token_id": 2,
31
+ "gradient_checkpointing": false,
32
+ "id2label": {
33
+ "0": "LABEL_0",
34
+ "1": "LABEL_1",
35
+ "2": "LABEL_2"
36
+ },
37
+ "init_std": 0.02,
38
+ "is_encoder_decoder": true,
39
+ "label2id": {
40
+ "LABEL_0": 0,
41
+ "LABEL_1": 1,
42
+ "LABEL_2": 2
43
+ },
44
+ "max_position_embeddings": 1024,
45
+ "model_type": "mbart",
46
+ "no_repeat_ngram_size": 3,
47
+ "normalize_before": true,
48
+ "normalize_embedding": true,
49
+ "num_beams": 4,
50
+ "num_hidden_layers": 6,
51
+ "pad_token_id": 1,
52
+ "scale_embedding": false,
53
+ "static_position_embeddings": false,
54
+ "tokenizer_class": "BarthezTokenizer",
55
+ "torch_dtype": "float32",
56
+ "transformers_version": "4.10.2",
57
+ "use_cache": true,
58
+ "vocab_size": 50002
59
+ }
eval_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "eval_gen_len": 38.5256,
4
+ "eval_loss": 2.0710394382476807,
5
+ "eval_rouge1": 31.8947,
6
+ "eval_rouge2": 16.7563,
7
+ "eval_rougeL": 23.5428,
8
+ "eval_rougeLsum": 23.4918,
9
+ "eval_runtime": 12.584,
10
+ "eval_samples": 78,
11
+ "eval_samples_per_second": 6.198,
12
+ "eval_steps_per_second": 1.589
13
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3560635a9a51a7660e667116ae639ec221e0a0d7834203b2cac7eb9417e73660
3
+ size 557190519
sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:582403da7cb525c4cac90b495e553cb766b818be71755e6bd885072a1067314a
3
+ size 1115393
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
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "moussaKam/barthez", "tokenizer_class": "BarthezTokenizer"}
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.0,
3
+ "train_loss": 1.7793180174746757,
4
+ "train_runtime": 373.3963,
5
+ "train_samples": 469,
6
+ "train_samples_per_second": 25.121,
7
+ "train_steps_per_second": 6.32
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,423 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 2.0710394382476807,
3
+ "best_model_checkpoint": "./checkpoints/barthez-deft-chimie/checkpoint-1062",
4
+ "epoch": 20.0,
5
+ "global_step": 2360,
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.85,
12
+ "learning_rate": 2.8805084745762714e-05,
13
+ "loss": 3.8022,
14
+ "step": 100
15
+ },
16
+ {
17
+ "epoch": 1.0,
18
+ "eval_gen_len": 19.1538,
19
+ "eval_loss": 2.5490684509277344,
20
+ "eval_rouge1": 16.8208,
21
+ "eval_rouge2": 7.0027,
22
+ "eval_rougeL": 13.957,
23
+ "eval_rougeLsum": 14.0479,
24
+ "eval_runtime": 3.9642,
25
+ "eval_samples_per_second": 19.676,
26
+ "eval_steps_per_second": 5.045,
27
+ "step": 118
28
+ },
29
+ {
30
+ "epoch": 1.69,
31
+ "learning_rate": 2.7533898305084746e-05,
32
+ "loss": 2.9286,
33
+ "step": 200
34
+ },
35
+ {
36
+ "epoch": 2.0,
37
+ "eval_gen_len": 19.9487,
38
+ "eval_loss": 2.3073835372924805,
39
+ "eval_rouge1": 17.5356,
40
+ "eval_rouge2": 7.8717,
41
+ "eval_rougeL": 14.4874,
42
+ "eval_rougeLsum": 14.5044,
43
+ "eval_runtime": 3.9919,
44
+ "eval_samples_per_second": 19.54,
45
+ "eval_steps_per_second": 5.01,
46
+ "step": 236
47
+ },
48
+ {
49
+ "epoch": 2.54,
50
+ "learning_rate": 2.627542372881356e-05,
51
+ "loss": 2.5422,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 3.0,
56
+ "eval_gen_len": 19.7051,
57
+ "eval_loss": 2.232180118560791,
58
+ "eval_rouge1": 19.6491,
59
+ "eval_rouge2": 9.4156,
60
+ "eval_rougeL": 15.9467,
61
+ "eval_rougeLsum": 15.9433,
62
+ "eval_runtime": 3.9423,
63
+ "eval_samples_per_second": 19.785,
64
+ "eval_steps_per_second": 5.073,
65
+ "step": 354
66
+ },
67
+ {
68
+ "epoch": 3.39,
69
+ "learning_rate": 2.5004237288135596e-05,
70
+ "loss": 2.398,
71
+ "step": 400
72
+ },
73
+ {
74
+ "epoch": 4.0,
75
+ "eval_gen_len": 19.9231,
76
+ "eval_loss": 2.150047540664673,
77
+ "eval_rouge1": 18.7166,
78
+ "eval_rouge2": 9.859,
79
+ "eval_rougeL": 15.7535,
80
+ "eval_rougeLsum": 15.8036,
81
+ "eval_runtime": 3.904,
82
+ "eval_samples_per_second": 19.98,
83
+ "eval_steps_per_second": 5.123,
84
+ "step": 472
85
+ },
86
+ {
87
+ "epoch": 4.24,
88
+ "learning_rate": 2.373305084745763e-05,
89
+ "loss": 2.2044,
90
+ "step": 500
91
+ },
92
+ {
93
+ "epoch": 5.0,
94
+ "eval_gen_len": 19.6154,
95
+ "eval_loss": 2.1371994018554688,
96
+ "eval_rouge1": 19.978,
97
+ "eval_rouge2": 10.6235,
98
+ "eval_rougeL": 16.1348,
99
+ "eval_rougeLsum": 16.1274,
100
+ "eval_runtime": 3.9032,
101
+ "eval_samples_per_second": 19.984,
102
+ "eval_steps_per_second": 5.124,
103
+ "step": 590
104
+ },
105
+ {
106
+ "epoch": 5.08,
107
+ "learning_rate": 2.2461864406779662e-05,
108
+ "loss": 2.0419,
109
+ "step": 600
110
+ },
111
+ {
112
+ "epoch": 5.93,
113
+ "learning_rate": 2.1203389830508474e-05,
114
+ "loss": 1.9405,
115
+ "step": 700
116
+ },
117
+ {
118
+ "epoch": 6.0,
119
+ "eval_gen_len": 19.9744,
120
+ "eval_loss": 2.099212884902954,
121
+ "eval_rouge1": 20.226,
122
+ "eval_rouge2": 10.551,
123
+ "eval_rougeL": 16.6928,
124
+ "eval_rougeLsum": 16.7211,
125
+ "eval_runtime": 3.9012,
126
+ "eval_samples_per_second": 19.994,
127
+ "eval_steps_per_second": 5.127,
128
+ "step": 708
129
+ },
130
+ {
131
+ "epoch": 6.78,
132
+ "learning_rate": 1.993220338983051e-05,
133
+ "loss": 1.8544,
134
+ "step": 800
135
+ },
136
+ {
137
+ "epoch": 7.0,
138
+ "eval_gen_len": 19.8846,
139
+ "eval_loss": 2.0841410160064697,
140
+ "eval_rouge1": 19.8869,
141
+ "eval_rouge2": 10.8456,
142
+ "eval_rougeL": 16.1072,
143
+ "eval_rougeLsum": 16.097,
144
+ "eval_runtime": 3.8835,
145
+ "eval_samples_per_second": 20.085,
146
+ "eval_steps_per_second": 5.15,
147
+ "step": 826
148
+ },
149
+ {
150
+ "epoch": 7.63,
151
+ "learning_rate": 1.866101694915254e-05,
152
+ "loss": 1.7536,
153
+ "step": 900
154
+ },
155
+ {
156
+ "epoch": 8.0,
157
+ "eval_gen_len": 19.859,
158
+ "eval_loss": 2.0790863037109375,
159
+ "eval_rouge1": 19.3017,
160
+ "eval_rouge2": 9.4921,
161
+ "eval_rougeL": 16.1541,
162
+ "eval_rougeLsum": 16.2167,
163
+ "eval_runtime": 3.9024,
164
+ "eval_samples_per_second": 19.988,
165
+ "eval_steps_per_second": 5.125,
166
+ "step": 944
167
+ },
168
+ {
169
+ "epoch": 8.47,
170
+ "learning_rate": 1.7389830508474576e-05,
171
+ "loss": 1.6914,
172
+ "step": 1000
173
+ },
174
+ {
175
+ "epoch": 9.0,
176
+ "eval_gen_len": 19.8846,
177
+ "eval_loss": 2.0710394382476807,
178
+ "eval_rouge1": 21.3848,
179
+ "eval_rouge2": 10.4088,
180
+ "eval_rougeL": 17.1963,
181
+ "eval_rougeLsum": 17.2254,
182
+ "eval_runtime": 3.8762,
183
+ "eval_samples_per_second": 20.123,
184
+ "eval_steps_per_second": 5.16,
185
+ "step": 1062
186
+ },
187
+ {
188
+ "epoch": 9.32,
189
+ "learning_rate": 1.611864406779661e-05,
190
+ "loss": 1.654,
191
+ "step": 1100
192
+ },
193
+ {
194
+ "epoch": 10.0,
195
+ "eval_gen_len": 19.9231,
196
+ "eval_loss": 2.106861114501953,
197
+ "eval_rouge1": 22.3811,
198
+ "eval_rouge2": 10.7987,
199
+ "eval_rougeL": 18.7595,
200
+ "eval_rougeLsum": 18.761,
201
+ "eval_runtime": 3.9276,
202
+ "eval_samples_per_second": 19.859,
203
+ "eval_steps_per_second": 5.092,
204
+ "step": 1180
205
+ },
206
+ {
207
+ "epoch": 10.17,
208
+ "learning_rate": 1.4847457627118646e-05,
209
+ "loss": 1.5899,
210
+ "step": 1200
211
+ },
212
+ {
213
+ "epoch": 11.0,
214
+ "eval_gen_len": 19.8077,
215
+ "eval_loss": 2.09187650680542,
216
+ "eval_rouge1": 20.8546,
217
+ "eval_rouge2": 10.6958,
218
+ "eval_rougeL": 16.8637,
219
+ "eval_rougeLsum": 16.9499,
220
+ "eval_runtime": 3.8954,
221
+ "eval_samples_per_second": 20.024,
222
+ "eval_steps_per_second": 5.134,
223
+ "step": 1298
224
+ },
225
+ {
226
+ "epoch": 11.02,
227
+ "learning_rate": 1.357627118644068e-05,
228
+ "loss": 1.534,
229
+ "step": 1300
230
+ },
231
+ {
232
+ "epoch": 11.86,
233
+ "learning_rate": 1.2305084745762711e-05,
234
+ "loss": 1.4661,
235
+ "step": 1400
236
+ },
237
+ {
238
+ "epoch": 12.0,
239
+ "eval_gen_len": 19.9744,
240
+ "eval_loss": 2.106508255004883,
241
+ "eval_rouge1": 22.3677,
242
+ "eval_rouge2": 11.7472,
243
+ "eval_rougeL": 18.262,
244
+ "eval_rougeLsum": 18.3,
245
+ "eval_runtime": 3.8544,
246
+ "eval_samples_per_second": 20.237,
247
+ "eval_steps_per_second": 5.189,
248
+ "step": 1416
249
+ },
250
+ {
251
+ "epoch": 12.71,
252
+ "learning_rate": 1.1033898305084746e-05,
253
+ "loss": 1.4205,
254
+ "step": 1500
255
+ },
256
+ {
257
+ "epoch": 13.0,
258
+ "eval_gen_len": 19.9359,
259
+ "eval_loss": 2.116386651992798,
260
+ "eval_rouge1": 20.5845,
261
+ "eval_rouge2": 10.7825,
262
+ "eval_rougeL": 16.9972,
263
+ "eval_rougeLsum": 17.0216,
264
+ "eval_runtime": 3.9246,
265
+ "eval_samples_per_second": 19.875,
266
+ "eval_steps_per_second": 5.096,
267
+ "step": 1534
268
+ },
269
+ {
270
+ "epoch": 13.56,
271
+ "learning_rate": 9.76271186440678e-06,
272
+ "loss": 1.3797,
273
+ "step": 1600
274
+ },
275
+ {
276
+ "epoch": 14.0,
277
+ "eval_gen_len": 19.9744,
278
+ "eval_loss": 2.124025344848633,
279
+ "eval_rouge1": 22.2561,
280
+ "eval_rouge2": 11.303,
281
+ "eval_rougeL": 17.5064,
282
+ "eval_rougeLsum": 17.5815,
283
+ "eval_runtime": 3.8828,
284
+ "eval_samples_per_second": 20.089,
285
+ "eval_steps_per_second": 5.151,
286
+ "step": 1652
287
+ },
288
+ {
289
+ "epoch": 14.41,
290
+ "learning_rate": 8.491525423728813e-06,
291
+ "loss": 1.3724,
292
+ "step": 1700
293
+ },
294
+ {
295
+ "epoch": 15.0,
296
+ "eval_gen_len": 19.9359,
297
+ "eval_loss": 2.118665933609009,
298
+ "eval_rouge1": 23.2825,
299
+ "eval_rouge2": 11.912,
300
+ "eval_rougeL": 18.5208,
301
+ "eval_rougeLsum": 18.5499,
302
+ "eval_runtime": 3.9164,
303
+ "eval_samples_per_second": 19.916,
304
+ "eval_steps_per_second": 5.107,
305
+ "step": 1770
306
+ },
307
+ {
308
+ "epoch": 15.25,
309
+ "learning_rate": 7.220338983050848e-06,
310
+ "loss": 1.3404,
311
+ "step": 1800
312
+ },
313
+ {
314
+ "epoch": 16.0,
315
+ "eval_gen_len": 19.9744,
316
+ "eval_loss": 2.139371395111084,
317
+ "eval_rouge1": 22.1305,
318
+ "eval_rouge2": 10.5258,
319
+ "eval_rougeL": 17.772,
320
+ "eval_rougeLsum": 17.8202,
321
+ "eval_runtime": 3.9181,
322
+ "eval_samples_per_second": 19.908,
323
+ "eval_steps_per_second": 5.105,
324
+ "step": 1888
325
+ },
326
+ {
327
+ "epoch": 16.1,
328
+ "learning_rate": 5.949152542372881e-06,
329
+ "loss": 1.2967,
330
+ "step": 1900
331
+ },
332
+ {
333
+ "epoch": 16.95,
334
+ "learning_rate": 4.677966101694916e-06,
335
+ "loss": 1.2846,
336
+ "step": 2000
337
+ },
338
+ {
339
+ "epoch": 17.0,
340
+ "eval_gen_len": 20.0,
341
+ "eval_loss": 2.1501927375793457,
342
+ "eval_rouge1": 21.567,
343
+ "eval_rouge2": 11.0557,
344
+ "eval_rougeL": 17.2562,
345
+ "eval_rougeLsum": 17.2974,
346
+ "eval_runtime": 3.8698,
347
+ "eval_samples_per_second": 20.156,
348
+ "eval_steps_per_second": 5.168,
349
+ "step": 2006
350
+ },
351
+ {
352
+ "epoch": 17.8,
353
+ "learning_rate": 3.406779661016949e-06,
354
+ "loss": 1.2871,
355
+ "step": 2100
356
+ },
357
+ {
358
+ "epoch": 18.0,
359
+ "eval_gen_len": 19.9744,
360
+ "eval_loss": 2.1571571826934814,
361
+ "eval_rouge1": 22.5871,
362
+ "eval_rouge2": 11.702,
363
+ "eval_rougeL": 18.2906,
364
+ "eval_rougeLsum": 18.3826,
365
+ "eval_runtime": 3.8649,
366
+ "eval_samples_per_second": 20.182,
367
+ "eval_steps_per_second": 5.175,
368
+ "step": 2124
369
+ },
370
+ {
371
+ "epoch": 18.64,
372
+ "learning_rate": 2.1355932203389833e-06,
373
+ "loss": 1.2422,
374
+ "step": 2200
375
+ },
376
+ {
377
+ "epoch": 19.0,
378
+ "eval_gen_len": 19.9744,
379
+ "eval_loss": 2.161255359649658,
380
+ "eval_rouge1": 23.0935,
381
+ "eval_rouge2": 11.6824,
382
+ "eval_rougeL": 18.6087,
383
+ "eval_rougeLsum": 18.6777,
384
+ "eval_runtime": 3.8614,
385
+ "eval_samples_per_second": 20.2,
386
+ "eval_steps_per_second": 5.18,
387
+ "step": 2242
388
+ },
389
+ {
390
+ "epoch": 19.49,
391
+ "learning_rate": 8.64406779661017e-07,
392
+ "loss": 1.2336,
393
+ "step": 2300
394
+ },
395
+ {
396
+ "epoch": 20.0,
397
+ "eval_gen_len": 19.9487,
398
+ "eval_loss": 2.158088445663452,
399
+ "eval_rouge1": 22.6789,
400
+ "eval_rouge2": 11.4363,
401
+ "eval_rougeL": 18.1661,
402
+ "eval_rougeLsum": 18.2346,
403
+ "eval_runtime": 3.8624,
404
+ "eval_samples_per_second": 20.195,
405
+ "eval_steps_per_second": 5.178,
406
+ "step": 2360
407
+ },
408
+ {
409
+ "epoch": 20.0,
410
+ "step": 2360,
411
+ "total_flos": 1837511390490624.0,
412
+ "train_loss": 1.7793180174746757,
413
+ "train_runtime": 373.3963,
414
+ "train_samples_per_second": 25.121,
415
+ "train_steps_per_second": 6.32
416
+ }
417
+ ],
418
+ "max_steps": 2360,
419
+ "num_train_epochs": 20,
420
+ "total_flos": 1837511390490624.0,
421
+ "trial_name": null,
422
+ "trial_params": null
423
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b366072883e294bb66de9dd12116fe6df2dcf5fbefdd2a9e1e596150b45f5d2
3
+ size 2863