LamaDiab commited on
Commit
850a867
·
verified ·
1 Parent(s): 9acb79d

Training checkpoint - Epoch 1, Step 28521

Browse files
checkpoint-28521/1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 384,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
checkpoint-28521/config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertModel"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 384,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 1536,
13
+ "layer_norm_eps": 1e-12,
14
+ "max_position_embeddings": 512,
15
+ "model_type": "bert",
16
+ "num_attention_heads": 12,
17
+ "num_hidden_layers": 6,
18
+ "pad_token_id": 0,
19
+ "position_embedding_type": "absolute",
20
+ "torch_dtype": "float32",
21
+ "transformers_version": "4.53.3",
22
+ "type_vocab_size": 2,
23
+ "use_cache": true,
24
+ "vocab_size": 30522
25
+ }
checkpoint-28521/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "5.2.0",
4
+ "transformers": "4.53.3",
5
+ "pytorch": "2.6.0+cu124"
6
+ },
7
+ "model_type": "SentenceTransformer",
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
checkpoint-28521/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a1f70654dc4ed88deb12af549aea3924a13419bcb51ffa3403925f399af5630
3
+ size 90864192
checkpoint-28521/modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.models.Normalize"
19
+ }
20
+ ]
checkpoint-28521/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e60991b876dda77c1db7ecc0e7c1085ef3a1f16294cdcf60ad5f36cb10afee71
3
+ size 180607738
checkpoint-28521/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1afa585e790a0abd97974c842e7ec70ff0bfd673f8af06d3603037e6cb09a849
3
+ size 14244
checkpoint-28521/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69b6545115ccd14f8d6eae6d9d5570039cf2edf6e315d6b5f105484412ec4c45
3
+ size 988
checkpoint-28521/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf44ecedcba524181ee6be9b35a5b3bae43faa1ab687e8e390841fcd87480b89
3
+ size 1064
checkpoint-28521/sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 256,
3
+ "do_lower_case": false
4
+ }
checkpoint-28521/special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
checkpoint-28521/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-28521/tokenizer_config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": false,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "max_length": 128,
51
+ "model_max_length": 256,
52
+ "never_split": null,
53
+ "pad_to_multiple_of": null,
54
+ "pad_token": "[PAD]",
55
+ "pad_token_type_id": 0,
56
+ "padding_side": "right",
57
+ "sep_token": "[SEP]",
58
+ "stride": 0,
59
+ "strip_accents": null,
60
+ "tokenize_chinese_chars": true,
61
+ "tokenizer_class": "BertTokenizer",
62
+ "truncation_side": "right",
63
+ "truncation_strategy": "longest_first",
64
+ "unk_token": "[UNK]"
65
+ }
checkpoint-28521/trainer_state.json ADDED
@@ -0,0 +1,953 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 28521,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 3.5061884225658285e-05,
14
+ "grad_norm": Infinity,
15
+ "learning_rate": 0.0,
16
+ "loss": 14.1655,
17
+ "step": 1
18
+ },
19
+ {
20
+ "epoch": 0.017530942112829143,
21
+ "grad_norm": 46.11793899536133,
22
+ "learning_rate": 8.695652173913044e-07,
23
+ "loss": 15.3746,
24
+ "step": 500
25
+ },
26
+ {
27
+ "epoch": 0.017530942112829143,
28
+ "eval_cosine_accuracy": 0.9304001331329346,
29
+ "eval_loss": 10.808653831481934,
30
+ "eval_runtime": 50.0101,
31
+ "eval_samples_per_second": 378.244,
32
+ "eval_steps_per_second": 1.48,
33
+ "step": 500
34
+ },
35
+ {
36
+ "epoch": 0.035061884225658285,
37
+ "grad_norm": 6.935111045837402,
38
+ "learning_rate": 1.7461430575035066e-06,
39
+ "loss": 11.7216,
40
+ "step": 1000
41
+ },
42
+ {
43
+ "epoch": 0.035061884225658285,
44
+ "eval_cosine_accuracy": 0.9219266772270203,
45
+ "eval_loss": 9.035820007324219,
46
+ "eval_runtime": 51.3273,
47
+ "eval_samples_per_second": 368.537,
48
+ "eval_steps_per_second": 1.442,
49
+ "step": 1000
50
+ },
51
+ {
52
+ "epoch": 0.05259282633848743,
53
+ "grad_norm": 4.2704758644104,
54
+ "learning_rate": 2.622720897615709e-06,
55
+ "loss": 10.0701,
56
+ "step": 1500
57
+ },
58
+ {
59
+ "epoch": 0.05259282633848743,
60
+ "eval_cosine_accuracy": 0.9366985559463501,
61
+ "eval_loss": 8.834137916564941,
62
+ "eval_runtime": 50.3538,
63
+ "eval_samples_per_second": 375.662,
64
+ "eval_steps_per_second": 1.47,
65
+ "step": 1500
66
+ },
67
+ {
68
+ "epoch": 0.07012376845131657,
69
+ "grad_norm": 3.6294240951538086,
70
+ "learning_rate": 3.49929873772791e-06,
71
+ "loss": 9.9787,
72
+ "step": 2000
73
+ },
74
+ {
75
+ "epoch": 0.07012376845131657,
76
+ "eval_cosine_accuracy": 0.9407313466072083,
77
+ "eval_loss": 8.768850326538086,
78
+ "eval_runtime": 50.9215,
79
+ "eval_samples_per_second": 371.474,
80
+ "eval_steps_per_second": 1.453,
81
+ "step": 2000
82
+ },
83
+ {
84
+ "epoch": 0.08765471056414571,
85
+ "grad_norm": 2.800403356552124,
86
+ "learning_rate": 4.375876577840113e-06,
87
+ "loss": 9.9256,
88
+ "step": 2500
89
+ },
90
+ {
91
+ "epoch": 0.08765471056414571,
92
+ "eval_cosine_accuracy": 0.9402782320976257,
93
+ "eval_loss": 8.751864433288574,
94
+ "eval_runtime": 50.75,
95
+ "eval_samples_per_second": 372.729,
96
+ "eval_steps_per_second": 1.458,
97
+ "step": 2500
98
+ },
99
+ {
100
+ "epoch": 0.10518565267697486,
101
+ "grad_norm": 2.86800217628479,
102
+ "learning_rate": 5.252454417952315e-06,
103
+ "loss": 9.8971,
104
+ "step": 3000
105
+ },
106
+ {
107
+ "epoch": 0.10518565267697486,
108
+ "eval_cosine_accuracy": 0.9425438642501831,
109
+ "eval_loss": 8.732973098754883,
110
+ "eval_runtime": 51.0941,
111
+ "eval_samples_per_second": 370.219,
112
+ "eval_steps_per_second": 1.448,
113
+ "step": 3000
114
+ },
115
+ {
116
+ "epoch": 0.122716594789804,
117
+ "grad_norm": 2.924982786178589,
118
+ "learning_rate": 6.129032258064517e-06,
119
+ "loss": 9.8627,
120
+ "step": 3500
121
+ },
122
+ {
123
+ "epoch": 0.122716594789804,
124
+ "eval_cosine_accuracy": 0.9468938112258911,
125
+ "eval_loss": 8.691949844360352,
126
+ "eval_runtime": 49.4971,
127
+ "eval_samples_per_second": 382.164,
128
+ "eval_steps_per_second": 1.495,
129
+ "step": 3500
130
+ },
131
+ {
132
+ "epoch": 0.14024753690263314,
133
+ "grad_norm": 3.102349281311035,
134
+ "learning_rate": 7.0056100981767185e-06,
135
+ "loss": 9.8413,
136
+ "step": 4000
137
+ },
138
+ {
139
+ "epoch": 0.14024753690263314,
140
+ "eval_cosine_accuracy": 0.9477094411849976,
141
+ "eval_loss": 8.692930221557617,
142
+ "eval_runtime": 50.0327,
143
+ "eval_samples_per_second": 378.073,
144
+ "eval_steps_per_second": 1.479,
145
+ "step": 4000
146
+ },
147
+ {
148
+ "epoch": 0.1577784790154623,
149
+ "grad_norm": 3.3259987831115723,
150
+ "learning_rate": 7.882187938288921e-06,
151
+ "loss": 9.8187,
152
+ "step": 4500
153
+ },
154
+ {
155
+ "epoch": 0.1577784790154623,
156
+ "eval_cosine_accuracy": 0.9502469301223755,
157
+ "eval_loss": 8.666604995727539,
158
+ "eval_runtime": 49.5796,
159
+ "eval_samples_per_second": 381.528,
160
+ "eval_steps_per_second": 1.493,
161
+ "step": 4500
162
+ },
163
+ {
164
+ "epoch": 0.17530942112829143,
165
+ "grad_norm": 3.461524248123169,
166
+ "learning_rate": 8.758765778401123e-06,
167
+ "loss": 9.797,
168
+ "step": 5000
169
+ },
170
+ {
171
+ "epoch": 0.17530942112829143,
172
+ "eval_cosine_accuracy": 0.9490235447883606,
173
+ "eval_loss": 8.687457084655762,
174
+ "eval_runtime": 50.1782,
175
+ "eval_samples_per_second": 376.977,
176
+ "eval_steps_per_second": 1.475,
177
+ "step": 5000
178
+ },
179
+ {
180
+ "epoch": 0.19284036324112058,
181
+ "grad_norm": 3.2629499435424805,
182
+ "learning_rate": 9.635343618513324e-06,
183
+ "loss": 9.7789,
184
+ "step": 5500
185
+ },
186
+ {
187
+ "epoch": 0.19284036324112058,
188
+ "eval_cosine_accuracy": 0.9515156745910645,
189
+ "eval_loss": 8.682653427124023,
190
+ "eval_runtime": 49.9304,
191
+ "eval_samples_per_second": 378.847,
192
+ "eval_steps_per_second": 1.482,
193
+ "step": 5500
194
+ },
195
+ {
196
+ "epoch": 0.2103713053539497,
197
+ "grad_norm": 2.80902361869812,
198
+ "learning_rate": 1.0511921458625528e-05,
199
+ "loss": 9.7601,
200
+ "step": 6000
201
+ },
202
+ {
203
+ "epoch": 0.2103713053539497,
204
+ "eval_cosine_accuracy": 0.9527844190597534,
205
+ "eval_loss": 8.673870086669922,
206
+ "eval_runtime": 49.5168,
207
+ "eval_samples_per_second": 382.012,
208
+ "eval_steps_per_second": 1.494,
209
+ "step": 6000
210
+ },
211
+ {
212
+ "epoch": 0.22790224746677887,
213
+ "grad_norm": 3.6036148071289062,
214
+ "learning_rate": 1.138849929873773e-05,
215
+ "loss": 9.7449,
216
+ "step": 6500
217
+ },
218
+ {
219
+ "epoch": 0.22790224746677887,
220
+ "eval_cosine_accuracy": 0.9525125622749329,
221
+ "eval_loss": 8.690710067749023,
222
+ "eval_runtime": 51.4523,
223
+ "eval_samples_per_second": 367.641,
224
+ "eval_steps_per_second": 1.438,
225
+ "step": 6500
226
+ },
227
+ {
228
+ "epoch": 0.245433189579608,
229
+ "grad_norm": 3.008776903152466,
230
+ "learning_rate": 1.226507713884993e-05,
231
+ "loss": 9.7281,
232
+ "step": 7000
233
+ },
234
+ {
235
+ "epoch": 0.245433189579608,
236
+ "eval_cosine_accuracy": 0.9523313045501709,
237
+ "eval_loss": 8.720589637756348,
238
+ "eval_runtime": 50.4169,
239
+ "eval_samples_per_second": 375.192,
240
+ "eval_steps_per_second": 1.468,
241
+ "step": 7000
242
+ },
243
+ {
244
+ "epoch": 0.26296413169243715,
245
+ "grad_norm": 3.2251088619232178,
246
+ "learning_rate": 1.3141654978962133e-05,
247
+ "loss": 9.7162,
248
+ "step": 7500
249
+ },
250
+ {
251
+ "epoch": 0.26296413169243715,
252
+ "eval_cosine_accuracy": 0.9521500468254089,
253
+ "eval_loss": 8.7124662399292,
254
+ "eval_runtime": 49.9244,
255
+ "eval_samples_per_second": 378.893,
256
+ "eval_steps_per_second": 1.482,
257
+ "step": 7500
258
+ },
259
+ {
260
+ "epoch": 0.2804950738052663,
261
+ "grad_norm": 3.34848690032959,
262
+ "learning_rate": 1.4018232819074336e-05,
263
+ "loss": 9.6994,
264
+ "step": 8000
265
+ },
266
+ {
267
+ "epoch": 0.2804950738052663,
268
+ "eval_cosine_accuracy": 0.9511532187461853,
269
+ "eval_loss": 8.740606307983398,
270
+ "eval_runtime": 51.2616,
271
+ "eval_samples_per_second": 369.009,
272
+ "eval_steps_per_second": 1.444,
273
+ "step": 8000
274
+ },
275
+ {
276
+ "epoch": 0.29802601591809547,
277
+ "grad_norm": 3.3112778663635254,
278
+ "learning_rate": 1.4894810659186536e-05,
279
+ "loss": 9.6911,
280
+ "step": 8500
281
+ },
282
+ {
283
+ "epoch": 0.29802601591809547,
284
+ "eval_cosine_accuracy": 0.9524672627449036,
285
+ "eval_loss": 8.709793090820312,
286
+ "eval_runtime": 50.3388,
287
+ "eval_samples_per_second": 375.774,
288
+ "eval_steps_per_second": 1.47,
289
+ "step": 8500
290
+ },
291
+ {
292
+ "epoch": 0.3155569580309246,
293
+ "grad_norm": 3.034113883972168,
294
+ "learning_rate": 1.577138849929874e-05,
295
+ "loss": 9.6785,
296
+ "step": 9000
297
+ },
298
+ {
299
+ "epoch": 0.3155569580309246,
300
+ "eval_cosine_accuracy": 0.9522860050201416,
301
+ "eval_loss": 8.736331939697266,
302
+ "eval_runtime": 49.8153,
303
+ "eval_samples_per_second": 379.722,
304
+ "eval_steps_per_second": 1.485,
305
+ "step": 9000
306
+ },
307
+ {
308
+ "epoch": 0.3330879001437537,
309
+ "grad_norm": 3.5567679405212402,
310
+ "learning_rate": 1.664621318373072e-05,
311
+ "loss": 9.6627,
312
+ "step": 9500
313
+ },
314
+ {
315
+ "epoch": 0.3330879001437537,
316
+ "eval_cosine_accuracy": 0.9511532187461853,
317
+ "eval_loss": 8.755115509033203,
318
+ "eval_runtime": 50.7728,
319
+ "eval_samples_per_second": 372.562,
320
+ "eval_steps_per_second": 1.457,
321
+ "step": 9500
322
+ },
323
+ {
324
+ "epoch": 0.35061884225658285,
325
+ "grad_norm": 3.8073484897613525,
326
+ "learning_rate": 1.7522791023842917e-05,
327
+ "loss": 9.6493,
328
+ "step": 10000
329
+ },
330
+ {
331
+ "epoch": 0.35061884225658285,
332
+ "eval_cosine_accuracy": 0.9495219588279724,
333
+ "eval_loss": 8.82080078125,
334
+ "eval_runtime": 49.6151,
335
+ "eval_samples_per_second": 381.255,
336
+ "eval_steps_per_second": 1.491,
337
+ "step": 10000
338
+ },
339
+ {
340
+ "epoch": 0.36814978436941204,
341
+ "grad_norm": 3.8539516925811768,
342
+ "learning_rate": 1.8399368863955122e-05,
343
+ "loss": 9.6422,
344
+ "step": 10500
345
+ },
346
+ {
347
+ "epoch": 0.36814978436941204,
348
+ "eval_cosine_accuracy": 0.9494766592979431,
349
+ "eval_loss": 8.80677318572998,
350
+ "eval_runtime": 51.2555,
351
+ "eval_samples_per_second": 369.053,
352
+ "eval_steps_per_second": 1.444,
353
+ "step": 10500
354
+ },
355
+ {
356
+ "epoch": 0.38568072648224117,
357
+ "grad_norm": 3.997718334197998,
358
+ "learning_rate": 1.9275946704067323e-05,
359
+ "loss": 9.6288,
360
+ "step": 11000
361
+ },
362
+ {
363
+ "epoch": 0.38568072648224117,
364
+ "eval_cosine_accuracy": 0.948706328868866,
365
+ "eval_loss": 8.862730026245117,
366
+ "eval_runtime": 50.5188,
367
+ "eval_samples_per_second": 374.435,
368
+ "eval_steps_per_second": 1.465,
369
+ "step": 11000
370
+ },
371
+ {
372
+ "epoch": 0.4032116685950703,
373
+ "grad_norm": 4.089541435241699,
374
+ "learning_rate": 1.9999964570084554e-05,
375
+ "loss": 9.6166,
376
+ "step": 11500
377
+ },
378
+ {
379
+ "epoch": 0.4032116685950703,
380
+ "eval_cosine_accuracy": 0.9495672583580017,
381
+ "eval_loss": 8.826420783996582,
382
+ "eval_runtime": 49.7378,
383
+ "eval_samples_per_second": 380.314,
384
+ "eval_steps_per_second": 1.488,
385
+ "step": 11500
386
+ },
387
+ {
388
+ "epoch": 0.4207426107078994,
389
+ "grad_norm": 3.3106064796447754,
390
+ "learning_rate": 1.9998387141025913e-05,
391
+ "loss": 9.6089,
392
+ "step": 12000
393
+ },
394
+ {
395
+ "epoch": 0.4207426107078994,
396
+ "eval_cosine_accuracy": 0.9481625556945801,
397
+ "eval_loss": 8.905811309814453,
398
+ "eval_runtime": 50.5979,
399
+ "eval_samples_per_second": 373.849,
400
+ "eval_steps_per_second": 1.463,
401
+ "step": 12000
402
+ },
403
+ {
404
+ "epoch": 0.4382735528207286,
405
+ "grad_norm": 3.426849603652954,
406
+ "learning_rate": 1.99944696715976e-05,
407
+ "loss": 9.5992,
408
+ "step": 12500
409
+ },
410
+ {
411
+ "epoch": 0.4382735528207286,
412
+ "eval_cosine_accuracy": 0.9498844742774963,
413
+ "eval_loss": 8.883400917053223,
414
+ "eval_runtime": 50.1249,
415
+ "eval_samples_per_second": 377.377,
416
+ "eval_steps_per_second": 1.476,
417
+ "step": 12500
418
+ },
419
+ {
420
+ "epoch": 0.45580449493355774,
421
+ "grad_norm": 2.942359209060669,
422
+ "learning_rate": 1.9988213078651134e-05,
423
+ "loss": 9.5968,
424
+ "step": 13000
425
+ },
426
+ {
427
+ "epoch": 0.45580449493355774,
428
+ "eval_cosine_accuracy": 0.9498391151428223,
429
+ "eval_loss": 8.826370239257812,
430
+ "eval_runtime": 50.4882,
431
+ "eval_samples_per_second": 374.662,
432
+ "eval_steps_per_second": 1.466,
433
+ "step": 13000
434
+ },
435
+ {
436
+ "epoch": 0.47333543704638686,
437
+ "grad_norm": 3.6883912086486816,
438
+ "learning_rate": 1.997961882649071e-05,
439
+ "loss": 9.5859,
440
+ "step": 13500
441
+ },
442
+ {
443
+ "epoch": 0.47333543704638686,
444
+ "eval_cosine_accuracy": 0.947437584400177,
445
+ "eval_loss": 8.92893123626709,
446
+ "eval_runtime": 49.8701,
447
+ "eval_samples_per_second": 379.305,
448
+ "eval_steps_per_second": 1.484,
449
+ "step": 13500
450
+ },
451
+ {
452
+ "epoch": 0.490866379159216,
453
+ "grad_norm": 3.4566166400909424,
454
+ "learning_rate": 1.996871311569756e-05,
455
+ "loss": 9.5771,
456
+ "step": 14000
457
+ },
458
+ {
459
+ "epoch": 0.490866379159216,
460
+ "eval_cosine_accuracy": 0.9474828839302063,
461
+ "eval_loss": 8.924897193908691,
462
+ "eval_runtime": 49.7604,
463
+ "eval_samples_per_second": 380.142,
464
+ "eval_steps_per_second": 1.487,
465
+ "step": 14000
466
+ },
467
+ {
468
+ "epoch": 0.5083973212720452,
469
+ "grad_norm": 4.192228317260742,
470
+ "learning_rate": 1.9955454789254234e-05,
471
+ "loss": 9.5657,
472
+ "step": 14500
473
+ },
474
+ {
475
+ "epoch": 0.5083973212720452,
476
+ "eval_cosine_accuracy": 0.9468938112258911,
477
+ "eval_loss": 8.962514877319336,
478
+ "eval_runtime": 51.4762,
479
+ "eval_samples_per_second": 367.47,
480
+ "eval_steps_per_second": 1.438,
481
+ "step": 14500
482
+ },
483
+ {
484
+ "epoch": 0.5259282633848743,
485
+ "grad_norm": 4.109299659729004,
486
+ "learning_rate": 1.9939866470405452e-05,
487
+ "loss": 9.5571,
488
+ "step": 15000
489
+ },
490
+ {
491
+ "epoch": 0.5259282633848743,
492
+ "eval_cosine_accuracy": 0.9458969831466675,
493
+ "eval_loss": 8.966425895690918,
494
+ "eval_runtime": 50.6672,
495
+ "eval_samples_per_second": 373.338,
496
+ "eval_steps_per_second": 1.461,
497
+ "step": 15000
498
+ },
499
+ {
500
+ "epoch": 0.5434592054977034,
501
+ "grad_norm": 3.5839483737945557,
502
+ "learning_rate": 1.9921951807469183e-05,
503
+ "loss": 9.5559,
504
+ "step": 15500
505
+ },
506
+ {
507
+ "epoch": 0.5434592054977034,
508
+ "eval_cosine_accuracy": 0.9465766549110413,
509
+ "eval_loss": 8.933329582214355,
510
+ "eval_runtime": 50.2462,
511
+ "eval_samples_per_second": 376.466,
512
+ "eval_steps_per_second": 1.473,
513
+ "step": 15500
514
+ },
515
+ {
516
+ "epoch": 0.5609901476105326,
517
+ "grad_norm": 3.516906499862671,
518
+ "learning_rate": 1.9901800560110894e-05,
519
+ "loss": 9.5514,
520
+ "step": 16000
521
+ },
522
+ {
523
+ "epoch": 0.5609901476105326,
524
+ "eval_cosine_accuracy": 0.9477094411849976,
525
+ "eval_loss": 8.930047988891602,
526
+ "eval_runtime": 51.0181,
527
+ "eval_samples_per_second": 370.771,
528
+ "eval_steps_per_second": 1.45,
529
+ "step": 16000
530
+ },
531
+ {
532
+ "epoch": 0.5785210897233617,
533
+ "grad_norm": 3.535010576248169,
534
+ "learning_rate": 1.9879255590325465e-05,
535
+ "loss": 9.5429,
536
+ "step": 16500
537
+ },
538
+ {
539
+ "epoch": 0.5785210897233617,
540
+ "eval_cosine_accuracy": 0.9487969279289246,
541
+ "eval_loss": 8.990011215209961,
542
+ "eval_runtime": 50.1912,
543
+ "eval_samples_per_second": 376.879,
544
+ "eval_steps_per_second": 1.474,
545
+ "step": 16500
546
+ },
547
+ {
548
+ "epoch": 0.5960520318361909,
549
+ "grad_norm": 3.881448268890381,
550
+ "learning_rate": 1.9854398461931087e-05,
551
+ "loss": 9.5319,
552
+ "step": 17000
553
+ },
554
+ {
555
+ "epoch": 0.5960520318361909,
556
+ "eval_cosine_accuracy": 0.943993866443634,
557
+ "eval_loss": 8.999488830566406,
558
+ "eval_runtime": 50.1438,
559
+ "eval_samples_per_second": 377.235,
560
+ "eval_steps_per_second": 1.476,
561
+ "step": 17000
562
+ },
563
+ {
564
+ "epoch": 0.6135829739490201,
565
+ "grad_norm": 3.682716131210327,
566
+ "learning_rate": 1.982723499253445e-05,
567
+ "loss": 9.5303,
568
+ "step": 17500
569
+ },
570
+ {
571
+ "epoch": 0.6135829739490201,
572
+ "eval_cosine_accuracy": 0.9459422826766968,
573
+ "eval_loss": 8.963994026184082,
574
+ "eval_runtime": 51.271,
575
+ "eval_samples_per_second": 368.941,
576
+ "eval_steps_per_second": 1.443,
577
+ "step": 17500
578
+ },
579
+ {
580
+ "epoch": 0.6311139160618492,
581
+ "grad_norm": 3.4688053131103516,
582
+ "learning_rate": 1.9797771539522402e-05,
583
+ "loss": 9.5292,
584
+ "step": 18000
585
+ },
586
+ {
587
+ "epoch": 0.6311139160618492,
588
+ "eval_cosine_accuracy": 0.9459422826766968,
589
+ "eval_loss": 8.978798866271973,
590
+ "eval_runtime": 51.1172,
591
+ "eval_samples_per_second": 370.051,
592
+ "eval_steps_per_second": 1.448,
593
+ "step": 18000
594
+ },
595
+ {
596
+ "epoch": 0.6486448581746783,
597
+ "grad_norm": 4.622004985809326,
598
+ "learning_rate": 1.9766014998574075e-05,
599
+ "loss": 9.5179,
600
+ "step": 18500
601
+ },
602
+ {
603
+ "epoch": 0.6486448581746783,
604
+ "eval_cosine_accuracy": 0.9446282386779785,
605
+ "eval_loss": 9.016169548034668,
606
+ "eval_runtime": 50.2418,
607
+ "eval_samples_per_second": 376.499,
608
+ "eval_steps_per_second": 1.473,
609
+ "step": 18500
610
+ },
611
+ {
612
+ "epoch": 0.6661758002875074,
613
+ "grad_norm": 3.7142796516418457,
614
+ "learning_rate": 1.973197280204698e-05,
615
+ "loss": 9.5171,
616
+ "step": 19000
617
+ },
618
+ {
619
+ "epoch": 0.6661758002875074,
620
+ "eval_cosine_accuracy": 0.9446282386779785,
621
+ "eval_loss": 9.053177833557129,
622
+ "eval_runtime": 51.5419,
623
+ "eval_samples_per_second": 367.002,
624
+ "eval_steps_per_second": 1.436,
625
+ "step": 19000
626
+ },
627
+ {
628
+ "epoch": 0.6837067424003366,
629
+ "grad_norm": 3.596008539199829,
630
+ "learning_rate": 1.969565291723755e-05,
631
+ "loss": 9.509,
632
+ "step": 19500
633
+ },
634
+ {
635
+ "epoch": 0.6837067424003366,
636
+ "eval_cosine_accuracy": 0.9444922804832458,
637
+ "eval_loss": 9.100041389465332,
638
+ "eval_runtime": 50.8331,
639
+ "eval_samples_per_second": 372.12,
640
+ "eval_steps_per_second": 1.456,
641
+ "step": 19500
642
+ },
643
+ {
644
+ "epoch": 0.7012376845131657,
645
+ "grad_norm": 3.7148385047912598,
646
+ "learning_rate": 1.965706384451643e-05,
647
+ "loss": 9.4974,
648
+ "step": 20000
649
+ },
650
+ {
651
+ "epoch": 0.7012376845131657,
652
+ "eval_cosine_accuracy": 0.9441297650337219,
653
+ "eval_loss": 9.080968856811523,
654
+ "eval_runtime": 51.7342,
655
+ "eval_samples_per_second": 365.638,
656
+ "eval_steps_per_second": 1.43,
657
+ "step": 20000
658
+ },
659
+ {
660
+ "epoch": 0.7187686266259948,
661
+ "grad_norm": 4.166687965393066,
662
+ "learning_rate": 1.9616298563166923e-05,
663
+ "loss": 9.4997,
664
+ "step": 20500
665
+ },
666
+ {
667
+ "epoch": 0.7187686266259948,
668
+ "eval_cosine_accuracy": 0.9448094367980957,
669
+ "eval_loss": 9.160306930541992,
670
+ "eval_runtime": 50.2386,
671
+ "eval_samples_per_second": 376.523,
672
+ "eval_steps_per_second": 1.473,
673
+ "step": 20500
674
+ },
675
+ {
676
+ "epoch": 0.7362995687388241,
677
+ "grad_norm": 3.890392541885376,
678
+ "learning_rate": 1.9573203229260095e-05,
679
+ "loss": 9.4949,
680
+ "step": 21000
681
+ },
682
+ {
683
+ "epoch": 0.7362995687388241,
684
+ "eval_cosine_accuracy": 0.9433141350746155,
685
+ "eval_loss": 9.170744895935059,
686
+ "eval_runtime": 50.2779,
687
+ "eval_samples_per_second": 376.229,
688
+ "eval_steps_per_second": 1.472,
689
+ "step": 21000
690
+ },
691
+ {
692
+ "epoch": 0.7538305108516532,
693
+ "grad_norm": 4.466862678527832,
694
+ "learning_rate": 1.9527867365785086e-05,
695
+ "loss": 9.4909,
696
+ "step": 21500
697
+ },
698
+ {
699
+ "epoch": 0.7538305108516532,
700
+ "eval_cosine_accuracy": 0.942679762840271,
701
+ "eval_loss": 9.167838096618652,
702
+ "eval_runtime": 51.5157,
703
+ "eval_samples_per_second": 367.189,
704
+ "eval_steps_per_second": 1.436,
705
+ "step": 21500
706
+ },
707
+ {
708
+ "epoch": 0.7713614529644823,
709
+ "grad_norm": 3.551948308944702,
710
+ "learning_rate": 1.9480301583228286e-05,
711
+ "loss": 9.4909,
712
+ "step": 22000
713
+ },
714
+ {
715
+ "epoch": 0.7713614529644823,
716
+ "eval_cosine_accuracy": 0.9448094367980957,
717
+ "eval_loss": 9.07791519165039,
718
+ "eval_runtime": 50.9951,
719
+ "eval_samples_per_second": 370.938,
720
+ "eval_steps_per_second": 1.451,
721
+ "step": 22000
722
+ },
723
+ {
724
+ "epoch": 0.7888923950773115,
725
+ "grad_norm": 4.436821460723877,
726
+ "learning_rate": 1.943061878979968e-05,
727
+ "loss": 9.4826,
728
+ "step": 22500
729
+ },
730
+ {
731
+ "epoch": 0.7888923950773115,
732
+ "eval_cosine_accuracy": 0.9433594346046448,
733
+ "eval_loss": 9.1586275100708,
734
+ "eval_runtime": 50.3731,
735
+ "eval_samples_per_second": 375.518,
736
+ "eval_steps_per_second": 1.469,
737
+ "step": 22500
738
+ },
739
+ {
740
+ "epoch": 0.8064233371901406,
741
+ "grad_norm": 4.032012939453125,
742
+ "learning_rate": 1.9378631487807903e-05,
743
+ "loss": 9.4739,
744
+ "step": 23000
745
+ },
746
+ {
747
+ "epoch": 0.8064233371901406,
748
+ "eval_cosine_accuracy": 0.9438126087188721,
749
+ "eval_loss": 9.108503341674805,
750
+ "eval_runtime": 51.2155,
751
+ "eval_samples_per_second": 369.341,
752
+ "eval_steps_per_second": 1.445,
753
+ "step": 23000
754
+ },
755
+ {
756
+ "epoch": 0.8239542793029697,
757
+ "grad_norm": 3.9402658939361572,
758
+ "learning_rate": 1.932444919416554e-05,
759
+ "loss": 9.4761,
760
+ "step": 23500
761
+ },
762
+ {
763
+ "epoch": 0.8239542793029697,
764
+ "eval_cosine_accuracy": 0.9426344633102417,
765
+ "eval_loss": 9.16297721862793,
766
+ "eval_runtime": 50.324,
767
+ "eval_samples_per_second": 375.884,
768
+ "eval_steps_per_second": 1.47,
769
+ "step": 23500
770
+ },
771
+ {
772
+ "epoch": 0.8414852214157988,
773
+ "grad_norm": 4.0512261390686035,
774
+ "learning_rate": 1.9268199488264804e-05,
775
+ "loss": 9.4678,
776
+ "step": 24000
777
+ },
778
+ {
779
+ "epoch": 0.8414852214157988,
780
+ "eval_cosine_accuracy": 0.9408219456672668,
781
+ "eval_loss": 9.143158912658691,
782
+ "eval_runtime": 50.9267,
783
+ "eval_samples_per_second": 371.436,
784
+ "eval_steps_per_second": 1.453,
785
+ "step": 24000
786
+ },
787
+ {
788
+ "epoch": 0.859016163528628,
789
+ "grad_norm": 4.608471393585205,
790
+ "learning_rate": 1.920967008957015e-05,
791
+ "loss": 9.4634,
792
+ "step": 24500
793
+ },
794
+ {
795
+ "epoch": 0.859016163528628,
796
+ "eval_cosine_accuracy": 0.9429516792297363,
797
+ "eval_loss": 9.19399642944336,
798
+ "eval_runtime": 50.1418,
799
+ "eval_samples_per_second": 377.25,
800
+ "eval_steps_per_second": 1.476,
801
+ "step": 24500
802
+ },
803
+ {
804
+ "epoch": 0.8765471056414572,
805
+ "grad_norm": 3.4046618938446045,
806
+ "learning_rate": 1.914898524325209e-05,
807
+ "loss": 9.4592,
808
+ "step": 25000
809
+ },
810
+ {
811
+ "epoch": 0.8765471056414572,
812
+ "eval_cosine_accuracy": 0.942498505115509,
813
+ "eval_loss": 9.171710014343262,
814
+ "eval_runtime": 50.5114,
815
+ "eval_samples_per_second": 374.49,
816
+ "eval_steps_per_second": 1.465,
817
+ "step": 25000
818
+ },
819
+ {
820
+ "epoch": 0.8940780477542863,
821
+ "grad_norm": 4.221813678741455,
822
+ "learning_rate": 1.908615915210035e-05,
823
+ "loss": 9.458,
824
+ "step": 25500
825
+ },
826
+ {
827
+ "epoch": 0.8940780477542863,
828
+ "eval_cosine_accuracy": 0.9407313466072083,
829
+ "eval_loss": 9.206527709960938,
830
+ "eval_runtime": 51.7106,
831
+ "eval_samples_per_second": 365.805,
832
+ "eval_steps_per_second": 1.431,
833
+ "step": 25500
834
+ },
835
+ {
836
+ "epoch": 0.9116089898671155,
837
+ "grad_norm": 4.586436748504639,
838
+ "learning_rate": 1.902120652004543e-05,
839
+ "loss": 9.449,
840
+ "step": 26000
841
+ },
842
+ {
843
+ "epoch": 0.9116089898671155,
844
+ "eval_cosine_accuracy": 0.940504789352417,
845
+ "eval_loss": 9.243682861328125,
846
+ "eval_runtime": 50.8955,
847
+ "eval_samples_per_second": 371.664,
848
+ "eval_steps_per_second": 1.454,
849
+ "step": 26000
850
+ },
851
+ {
852
+ "epoch": 0.9291399319799446,
853
+ "grad_norm": 4.309023380279541,
854
+ "learning_rate": 1.895414254871725e-05,
855
+ "loss": 9.4469,
856
+ "step": 26500
857
+ },
858
+ {
859
+ "epoch": 0.9291399319799446,
860
+ "eval_cosine_accuracy": 0.9420906901359558,
861
+ "eval_loss": 9.203714370727539,
862
+ "eval_runtime": 49.9746,
863
+ "eval_samples_per_second": 378.512,
864
+ "eval_steps_per_second": 1.481,
865
+ "step": 26500
866
+ },
867
+ {
868
+ "epoch": 0.9466708740927737,
869
+ "grad_norm": 3.725597858428955,
870
+ "learning_rate": 1.8884982933887337e-05,
871
+ "loss": 9.4501,
872
+ "step": 27000
873
+ },
874
+ {
875
+ "epoch": 0.9466708740927737,
876
+ "eval_cosine_accuracy": 0.9427704215049744,
877
+ "eval_loss": 9.19096565246582,
878
+ "eval_runtime": 51.4072,
879
+ "eval_samples_per_second": 367.964,
880
+ "eval_steps_per_second": 1.439,
881
+ "step": 27000
882
+ },
883
+ {
884
+ "epoch": 0.9642018162056029,
885
+ "grad_norm": 4.729601860046387,
886
+ "learning_rate": 1.8813743861795376e-05,
887
+ "loss": 9.4492,
888
+ "step": 27500
889
+ },
890
+ {
891
+ "epoch": 0.9642018162056029,
892
+ "eval_cosine_accuracy": 0.9398251175880432,
893
+ "eval_loss": 9.26838493347168,
894
+ "eval_runtime": 50.8114,
895
+ "eval_samples_per_second": 372.279,
896
+ "eval_steps_per_second": 1.456,
897
+ "step": 27500
898
+ },
899
+ {
900
+ "epoch": 0.981732758318432,
901
+ "grad_norm": 3.041431188583374,
902
+ "learning_rate": 1.8740590656409715e-05,
903
+ "loss": 9.445,
904
+ "step": 28000
905
+ },
906
+ {
907
+ "epoch": 0.981732758318432,
908
+ "eval_cosine_accuracy": 0.9418188333511353,
909
+ "eval_loss": 9.160036087036133,
910
+ "eval_runtime": 50.8238,
911
+ "eval_samples_per_second": 372.188,
912
+ "eval_steps_per_second": 1.456,
913
+ "step": 28000
914
+ },
915
+ {
916
+ "epoch": 0.9992637004312612,
917
+ "grad_norm": 3.983847141265869,
918
+ "learning_rate": 1.8665247245147084e-05,
919
+ "loss": 9.4356,
920
+ "step": 28500
921
+ },
922
+ {
923
+ "epoch": 0.9992637004312612,
924
+ "eval_cosine_accuracy": 0.9409579038619995,
925
+ "eval_loss": 9.164313316345215,
926
+ "eval_runtime": 50.3081,
927
+ "eval_samples_per_second": 376.003,
928
+ "eval_steps_per_second": 1.471,
929
+ "step": 28500
930
+ }
931
+ ],
932
+ "logging_steps": 500,
933
+ "max_steps": 114084,
934
+ "num_input_tokens_seen": 0,
935
+ "num_train_epochs": 4,
936
+ "save_steps": 500,
937
+ "stateful_callbacks": {
938
+ "TrainerControl": {
939
+ "args": {
940
+ "should_epoch_stop": false,
941
+ "should_evaluate": false,
942
+ "should_log": false,
943
+ "should_save": true,
944
+ "should_training_stop": false
945
+ },
946
+ "attributes": {}
947
+ }
948
+ },
949
+ "total_flos": 0.0,
950
+ "train_batch_size": 256,
951
+ "trial_name": null,
952
+ "trial_params": null
953
+ }
checkpoint-28521/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff251d0259f667a93594ea1efe197ff3b0827f9a1ffe30b0b91001733ac6ff94
3
+ size 5624
checkpoint-28521/training_metrics.json ADDED
@@ -0,0 +1,348 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metrics_history": [
3
+ {
4
+ "epoch": 0.017530942112829143,
5
+ "global_step": 500,
6
+ "eval_loss": 10.808653831481934,
7
+ "eval_cosine_accuracy": 0.9304001331329346
8
+ },
9
+ {
10
+ "epoch": 0.035061884225658285,
11
+ "global_step": 1000,
12
+ "eval_loss": 9.035820007324219,
13
+ "eval_cosine_accuracy": 0.9219266772270203
14
+ },
15
+ {
16
+ "epoch": 0.05259282633848743,
17
+ "global_step": 1500,
18
+ "eval_loss": 8.834137916564941,
19
+ "eval_cosine_accuracy": 0.9366985559463501
20
+ },
21
+ {
22
+ "epoch": 0.07012376845131657,
23
+ "global_step": 2000,
24
+ "eval_loss": 8.768850326538086,
25
+ "eval_cosine_accuracy": 0.9407313466072083
26
+ },
27
+ {
28
+ "epoch": 0.08765471056414571,
29
+ "global_step": 2500,
30
+ "eval_loss": 8.751864433288574,
31
+ "eval_cosine_accuracy": 0.9402782320976257
32
+ },
33
+ {
34
+ "epoch": 0.10518565267697486,
35
+ "global_step": 3000,
36
+ "eval_loss": 8.732973098754883,
37
+ "eval_cosine_accuracy": 0.9425438642501831
38
+ },
39
+ {
40
+ "epoch": 0.122716594789804,
41
+ "global_step": 3500,
42
+ "eval_loss": 8.691949844360352,
43
+ "eval_cosine_accuracy": 0.9468938112258911
44
+ },
45
+ {
46
+ "epoch": 0.14024753690263314,
47
+ "global_step": 4000,
48
+ "eval_loss": 8.692930221557617,
49
+ "eval_cosine_accuracy": 0.9477094411849976
50
+ },
51
+ {
52
+ "epoch": 0.1577784790154623,
53
+ "global_step": 4500,
54
+ "eval_loss": 8.666604995727539,
55
+ "eval_cosine_accuracy": 0.9502469301223755
56
+ },
57
+ {
58
+ "epoch": 0.17530942112829143,
59
+ "global_step": 5000,
60
+ "eval_loss": 8.687457084655762,
61
+ "eval_cosine_accuracy": 0.9490235447883606
62
+ },
63
+ {
64
+ "epoch": 0.19284036324112058,
65
+ "global_step": 5500,
66
+ "eval_loss": 8.682653427124023,
67
+ "eval_cosine_accuracy": 0.9515156745910645
68
+ },
69
+ {
70
+ "epoch": 0.2103713053539497,
71
+ "global_step": 6000,
72
+ "eval_loss": 8.673870086669922,
73
+ "eval_cosine_accuracy": 0.9527844190597534
74
+ },
75
+ {
76
+ "epoch": 0.22790224746677887,
77
+ "global_step": 6500,
78
+ "eval_loss": 8.690710067749023,
79
+ "eval_cosine_accuracy": 0.9525125622749329
80
+ },
81
+ {
82
+ "epoch": 0.245433189579608,
83
+ "global_step": 7000,
84
+ "eval_loss": 8.720589637756348,
85
+ "eval_cosine_accuracy": 0.9523313045501709
86
+ },
87
+ {
88
+ "epoch": 0.26296413169243715,
89
+ "global_step": 7500,
90
+ "eval_loss": 8.7124662399292,
91
+ "eval_cosine_accuracy": 0.9521500468254089
92
+ },
93
+ {
94
+ "epoch": 0.2804950738052663,
95
+ "global_step": 8000,
96
+ "eval_loss": 8.740606307983398,
97
+ "eval_cosine_accuracy": 0.9511532187461853
98
+ },
99
+ {
100
+ "epoch": 0.29802601591809547,
101
+ "global_step": 8500,
102
+ "eval_loss": 8.709793090820312,
103
+ "eval_cosine_accuracy": 0.9524672627449036
104
+ },
105
+ {
106
+ "epoch": 0.3155569580309246,
107
+ "global_step": 9000,
108
+ "eval_loss": 8.736331939697266,
109
+ "eval_cosine_accuracy": 0.9522860050201416
110
+ },
111
+ {
112
+ "epoch": 0.3330879001437537,
113
+ "global_step": 9500,
114
+ "eval_loss": 8.755115509033203,
115
+ "eval_cosine_accuracy": 0.9511532187461853
116
+ },
117
+ {
118
+ "epoch": 0.35061884225658285,
119
+ "global_step": 10000,
120
+ "eval_loss": 8.82080078125,
121
+ "eval_cosine_accuracy": 0.9495219588279724
122
+ },
123
+ {
124
+ "epoch": 0.36814978436941204,
125
+ "global_step": 10500,
126
+ "eval_loss": 8.80677318572998,
127
+ "eval_cosine_accuracy": 0.9494766592979431
128
+ },
129
+ {
130
+ "epoch": 0.38568072648224117,
131
+ "global_step": 11000,
132
+ "eval_loss": 8.862730026245117,
133
+ "eval_cosine_accuracy": 0.948706328868866
134
+ },
135
+ {
136
+ "epoch": 0.4032116685950703,
137
+ "global_step": 11500,
138
+ "eval_loss": 8.826420783996582,
139
+ "eval_cosine_accuracy": 0.9495672583580017
140
+ },
141
+ {
142
+ "epoch": 0.4207426107078994,
143
+ "global_step": 12000,
144
+ "eval_loss": 8.905811309814453,
145
+ "eval_cosine_accuracy": 0.9481625556945801
146
+ },
147
+ {
148
+ "epoch": 0.4382735528207286,
149
+ "global_step": 12500,
150
+ "eval_loss": 8.883400917053223,
151
+ "eval_cosine_accuracy": 0.9498844742774963
152
+ },
153
+ {
154
+ "epoch": 0.45580449493355774,
155
+ "global_step": 13000,
156
+ "eval_loss": 8.826370239257812,
157
+ "eval_cosine_accuracy": 0.9498391151428223
158
+ },
159
+ {
160
+ "epoch": 0.47333543704638686,
161
+ "global_step": 13500,
162
+ "eval_loss": 8.92893123626709,
163
+ "eval_cosine_accuracy": 0.947437584400177
164
+ },
165
+ {
166
+ "epoch": 0.490866379159216,
167
+ "global_step": 14000,
168
+ "eval_loss": 8.924897193908691,
169
+ "eval_cosine_accuracy": 0.9474828839302063
170
+ },
171
+ {
172
+ "epoch": 0.5083973212720452,
173
+ "global_step": 14500,
174
+ "eval_loss": 8.962514877319336,
175
+ "eval_cosine_accuracy": 0.9468938112258911
176
+ },
177
+ {
178
+ "epoch": 0.5259282633848743,
179
+ "global_step": 15000,
180
+ "eval_loss": 8.966425895690918,
181
+ "eval_cosine_accuracy": 0.9458969831466675
182
+ },
183
+ {
184
+ "epoch": 0.5434592054977034,
185
+ "global_step": 15500,
186
+ "eval_loss": 8.933329582214355,
187
+ "eval_cosine_accuracy": 0.9465766549110413
188
+ },
189
+ {
190
+ "epoch": 0.5609901476105326,
191
+ "global_step": 16000,
192
+ "eval_loss": 8.930047988891602,
193
+ "eval_cosine_accuracy": 0.9477094411849976
194
+ },
195
+ {
196
+ "epoch": 0.5785210897233617,
197
+ "global_step": 16500,
198
+ "eval_loss": 8.990011215209961,
199
+ "eval_cosine_accuracy": 0.9487969279289246
200
+ },
201
+ {
202
+ "epoch": 0.5960520318361909,
203
+ "global_step": 17000,
204
+ "eval_loss": 8.999488830566406,
205
+ "eval_cosine_accuracy": 0.943993866443634
206
+ },
207
+ {
208
+ "epoch": 0.6135829739490201,
209
+ "global_step": 17500,
210
+ "eval_loss": 8.963994026184082,
211
+ "eval_cosine_accuracy": 0.9459422826766968
212
+ },
213
+ {
214
+ "epoch": 0.6311139160618492,
215
+ "global_step": 18000,
216
+ "eval_loss": 8.978798866271973,
217
+ "eval_cosine_accuracy": 0.9459422826766968
218
+ },
219
+ {
220
+ "epoch": 0.6486448581746783,
221
+ "global_step": 18500,
222
+ "eval_loss": 9.016169548034668,
223
+ "eval_cosine_accuracy": 0.9446282386779785
224
+ },
225
+ {
226
+ "epoch": 0.6661758002875074,
227
+ "global_step": 19000,
228
+ "eval_loss": 9.053177833557129,
229
+ "eval_cosine_accuracy": 0.9446282386779785
230
+ },
231
+ {
232
+ "epoch": 0.6837067424003366,
233
+ "global_step": 19500,
234
+ "eval_loss": 9.100041389465332,
235
+ "eval_cosine_accuracy": 0.9444922804832458
236
+ },
237
+ {
238
+ "epoch": 0.7012376845131657,
239
+ "global_step": 20000,
240
+ "eval_loss": 9.080968856811523,
241
+ "eval_cosine_accuracy": 0.9441297650337219
242
+ },
243
+ {
244
+ "epoch": 0.7187686266259948,
245
+ "global_step": 20500,
246
+ "eval_loss": 9.160306930541992,
247
+ "eval_cosine_accuracy": 0.9448094367980957
248
+ },
249
+ {
250
+ "epoch": 0.7362995687388241,
251
+ "global_step": 21000,
252
+ "eval_loss": 9.170744895935059,
253
+ "eval_cosine_accuracy": 0.9433141350746155
254
+ },
255
+ {
256
+ "epoch": 0.7538305108516532,
257
+ "global_step": 21500,
258
+ "eval_loss": 9.167838096618652,
259
+ "eval_cosine_accuracy": 0.942679762840271
260
+ },
261
+ {
262
+ "epoch": 0.7713614529644823,
263
+ "global_step": 22000,
264
+ "eval_loss": 9.07791519165039,
265
+ "eval_cosine_accuracy": 0.9448094367980957
266
+ },
267
+ {
268
+ "epoch": 0.7888923950773115,
269
+ "global_step": 22500,
270
+ "eval_loss": 9.1586275100708,
271
+ "eval_cosine_accuracy": 0.9433594346046448
272
+ },
273
+ {
274
+ "epoch": 0.8064233371901406,
275
+ "global_step": 23000,
276
+ "eval_loss": 9.108503341674805,
277
+ "eval_cosine_accuracy": 0.9438126087188721
278
+ },
279
+ {
280
+ "epoch": 0.8239542793029697,
281
+ "global_step": 23500,
282
+ "eval_loss": 9.16297721862793,
283
+ "eval_cosine_accuracy": 0.9426344633102417
284
+ },
285
+ {
286
+ "epoch": 0.8414852214157988,
287
+ "global_step": 24000,
288
+ "eval_loss": 9.143158912658691,
289
+ "eval_cosine_accuracy": 0.9408219456672668
290
+ },
291
+ {
292
+ "epoch": 0.859016163528628,
293
+ "global_step": 24500,
294
+ "eval_loss": 9.19399642944336,
295
+ "eval_cosine_accuracy": 0.9429516792297363
296
+ },
297
+ {
298
+ "epoch": 0.8765471056414572,
299
+ "global_step": 25000,
300
+ "eval_loss": 9.171710014343262,
301
+ "eval_cosine_accuracy": 0.942498505115509
302
+ },
303
+ {
304
+ "epoch": 0.8940780477542863,
305
+ "global_step": 25500,
306
+ "eval_loss": 9.206527709960938,
307
+ "eval_cosine_accuracy": 0.9407313466072083
308
+ },
309
+ {
310
+ "epoch": 0.9116089898671155,
311
+ "global_step": 26000,
312
+ "eval_loss": 9.243682861328125,
313
+ "eval_cosine_accuracy": 0.940504789352417
314
+ },
315
+ {
316
+ "epoch": 0.9291399319799446,
317
+ "global_step": 26500,
318
+ "eval_loss": 9.203714370727539,
319
+ "eval_cosine_accuracy": 0.9420906901359558
320
+ },
321
+ {
322
+ "epoch": 0.9466708740927737,
323
+ "global_step": 27000,
324
+ "eval_loss": 9.19096565246582,
325
+ "eval_cosine_accuracy": 0.9427704215049744
326
+ },
327
+ {
328
+ "epoch": 0.9642018162056029,
329
+ "global_step": 27500,
330
+ "eval_loss": 9.26838493347168,
331
+ "eval_cosine_accuracy": 0.9398251175880432
332
+ },
333
+ {
334
+ "epoch": 0.981732758318432,
335
+ "global_step": 28000,
336
+ "eval_loss": 9.160036087036133,
337
+ "eval_cosine_accuracy": 0.9418188333511353
338
+ },
339
+ {
340
+ "epoch": 0.9992637004312612,
341
+ "global_step": 28500,
342
+ "eval_loss": 9.164313316345215,
343
+ "eval_cosine_accuracy": 0.9409579038619995
344
+ }
345
+ ],
346
+ "current_epoch": 1,
347
+ "global_step": 28521
348
+ }
checkpoint-28521/vocab.txt ADDED
The diff for this file is too large to render. See raw diff