youssefkhalil320 commited on
Commit
98bad5c
·
verified ·
1 Parent(s): ce569bc

Upload folder using huggingface_hub

Browse files
checkpoint-12500/1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
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-12500/README.md ADDED
@@ -0,0 +1,460 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: sentence-transformers/all-mpnet-base-v2
3
+ language:
4
+ - en
5
+ library_name: sentence-transformers
6
+ license: apache-2.0
7
+ pipeline_tag: sentence-similarity
8
+ tags:
9
+ - sentence-transformers
10
+ - sentence-similarity
11
+ - feature-extraction
12
+ - generated_from_trainer
13
+ - dataset_size:1363306
14
+ - loss:CoSENTLoss
15
+ widget:
16
+ - source_sentence: labneh
17
+ sentences:
18
+ - iftar
19
+ - bathing suit
20
+ - coffee cup
21
+ - source_sentence: Velvet flock Veil
22
+ sentences:
23
+ - mermaid purse
24
+ - veil
25
+ - mobile bag
26
+ - source_sentence: Red lipstick
27
+ sentences:
28
+ - chemise dress
29
+ - tote
30
+ - rouge
31
+ - source_sentence: Unisex Travel bag
32
+ sentences:
33
+ - spf
34
+ - basic vega ring
35
+ - travel backpack
36
+ - source_sentence: jeremy hush book
37
+ sentences:
38
+ - chinese jumper
39
+ - perfume
40
+ - home automation device
41
+ ---
42
+
43
+ # all-mpnet-base-v2-pair_score
44
+
45
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2). It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
46
+
47
+ ## Model Details
48
+
49
+ ### Model Description
50
+ - **Model Type:** Sentence Transformer
51
+ - **Base model:** [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) <!-- at revision 9a3225965996d404b775526de6dbfe85d3368642 -->
52
+ - **Maximum Sequence Length:** 384 tokens
53
+ - **Output Dimensionality:** 768 tokens
54
+ - **Similarity Function:** Cosine Similarity
55
+ <!-- - **Training Dataset:** Unknown -->
56
+ - **Language:** en
57
+ - **License:** apache-2.0
58
+
59
+ ### Model Sources
60
+
61
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
62
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
63
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
64
+
65
+ ### Full Model Architecture
66
+
67
+ ```
68
+ SentenceTransformer(
69
+ (0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel
70
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
71
+ (2): Normalize()
72
+ )
73
+ ```
74
+
75
+ ## Usage
76
+
77
+ ### Direct Usage (Sentence Transformers)
78
+
79
+ First install the Sentence Transformers library:
80
+
81
+ ```bash
82
+ pip install -U sentence-transformers
83
+ ```
84
+
85
+ Then you can load this model and run inference.
86
+ ```python
87
+ from sentence_transformers import SentenceTransformer
88
+
89
+ # Download from the 🤗 Hub
90
+ model = SentenceTransformer("sentence_transformers_model_id")
91
+ # Run inference
92
+ sentences = [
93
+ 'jeremy hush book',
94
+ 'chinese jumper',
95
+ 'perfume',
96
+ ]
97
+ embeddings = model.encode(sentences)
98
+ print(embeddings.shape)
99
+ # [3, 768]
100
+
101
+ # Get the similarity scores for the embeddings
102
+ similarities = model.similarity(embeddings, embeddings)
103
+ print(similarities.shape)
104
+ # [3, 3]
105
+ ```
106
+
107
+ <!--
108
+ ### Direct Usage (Transformers)
109
+
110
+ <details><summary>Click to see the direct usage in Transformers</summary>
111
+
112
+ </details>
113
+ -->
114
+
115
+ <!--
116
+ ### Downstream Usage (Sentence Transformers)
117
+
118
+ You can finetune this model on your own dataset.
119
+
120
+ <details><summary>Click to expand</summary>
121
+
122
+ </details>
123
+ -->
124
+
125
+ <!--
126
+ ### Out-of-Scope Use
127
+
128
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
129
+ -->
130
+
131
+ <!--
132
+ ## Bias, Risks and Limitations
133
+
134
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
135
+ -->
136
+
137
+ <!--
138
+ ### Recommendations
139
+
140
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
141
+ -->
142
+
143
+ ## Training Details
144
+
145
+ ### Training Hyperparameters
146
+ #### Non-Default Hyperparameters
147
+
148
+ - `eval_strategy`: steps
149
+ - `per_device_train_batch_size`: 128
150
+ - `per_device_eval_batch_size`: 128
151
+ - `learning_rate`: 2e-05
152
+ - `num_train_epochs`: 2
153
+ - `warmup_ratio`: 0.1
154
+ - `fp16`: True
155
+
156
+ #### All Hyperparameters
157
+ <details><summary>Click to expand</summary>
158
+
159
+ - `overwrite_output_dir`: False
160
+ - `do_predict`: False
161
+ - `eval_strategy`: steps
162
+ - `prediction_loss_only`: True
163
+ - `per_device_train_batch_size`: 128
164
+ - `per_device_eval_batch_size`: 128
165
+ - `per_gpu_train_batch_size`: None
166
+ - `per_gpu_eval_batch_size`: None
167
+ - `gradient_accumulation_steps`: 1
168
+ - `eval_accumulation_steps`: None
169
+ - `torch_empty_cache_steps`: None
170
+ - `learning_rate`: 2e-05
171
+ - `weight_decay`: 0.0
172
+ - `adam_beta1`: 0.9
173
+ - `adam_beta2`: 0.999
174
+ - `adam_epsilon`: 1e-08
175
+ - `max_grad_norm`: 1.0
176
+ - `num_train_epochs`: 2
177
+ - `max_steps`: -1
178
+ - `lr_scheduler_type`: linear
179
+ - `lr_scheduler_kwargs`: {}
180
+ - `warmup_ratio`: 0.1
181
+ - `warmup_steps`: 0
182
+ - `log_level`: passive
183
+ - `log_level_replica`: warning
184
+ - `log_on_each_node`: True
185
+ - `logging_nan_inf_filter`: True
186
+ - `save_safetensors`: True
187
+ - `save_on_each_node`: False
188
+ - `save_only_model`: False
189
+ - `restore_callback_states_from_checkpoint`: False
190
+ - `no_cuda`: False
191
+ - `use_cpu`: False
192
+ - `use_mps_device`: False
193
+ - `seed`: 42
194
+ - `data_seed`: None
195
+ - `jit_mode_eval`: False
196
+ - `use_ipex`: False
197
+ - `bf16`: False
198
+ - `fp16`: True
199
+ - `fp16_opt_level`: O1
200
+ - `half_precision_backend`: auto
201
+ - `bf16_full_eval`: False
202
+ - `fp16_full_eval`: False
203
+ - `tf32`: None
204
+ - `local_rank`: 0
205
+ - `ddp_backend`: None
206
+ - `tpu_num_cores`: None
207
+ - `tpu_metrics_debug`: False
208
+ - `debug`: []
209
+ - `dataloader_drop_last`: False
210
+ - `dataloader_num_workers`: 0
211
+ - `dataloader_prefetch_factor`: None
212
+ - `past_index`: -1
213
+ - `disable_tqdm`: False
214
+ - `remove_unused_columns`: True
215
+ - `label_names`: None
216
+ - `load_best_model_at_end`: False
217
+ - `ignore_data_skip`: False
218
+ - `fsdp`: []
219
+ - `fsdp_min_num_params`: 0
220
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
221
+ - `fsdp_transformer_layer_cls_to_wrap`: None
222
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
223
+ - `deepspeed`: None
224
+ - `label_smoothing_factor`: 0.0
225
+ - `optim`: adamw_torch
226
+ - `optim_args`: None
227
+ - `adafactor`: False
228
+ - `group_by_length`: False
229
+ - `length_column_name`: length
230
+ - `ddp_find_unused_parameters`: None
231
+ - `ddp_bucket_cap_mb`: None
232
+ - `ddp_broadcast_buffers`: False
233
+ - `dataloader_pin_memory`: True
234
+ - `dataloader_persistent_workers`: False
235
+ - `skip_memory_metrics`: True
236
+ - `use_legacy_prediction_loop`: False
237
+ - `push_to_hub`: False
238
+ - `resume_from_checkpoint`: None
239
+ - `hub_model_id`: None
240
+ - `hub_strategy`: every_save
241
+ - `hub_private_repo`: False
242
+ - `hub_always_push`: False
243
+ - `gradient_checkpointing`: False
244
+ - `gradient_checkpointing_kwargs`: None
245
+ - `include_inputs_for_metrics`: False
246
+ - `eval_do_concat_batches`: True
247
+ - `fp16_backend`: auto
248
+ - `push_to_hub_model_id`: None
249
+ - `push_to_hub_organization`: None
250
+ - `mp_parameters`:
251
+ - `auto_find_batch_size`: False
252
+ - `full_determinism`: False
253
+ - `torchdynamo`: None
254
+ - `ray_scope`: last
255
+ - `ddp_timeout`: 1800
256
+ - `torch_compile`: False
257
+ - `torch_compile_backend`: None
258
+ - `torch_compile_mode`: None
259
+ - `dispatch_batches`: None
260
+ - `split_batches`: None
261
+ - `include_tokens_per_second`: False
262
+ - `include_num_input_tokens_seen`: False
263
+ - `neftune_noise_alpha`: None
264
+ - `optim_target_modules`: None
265
+ - `batch_eval_metrics`: False
266
+ - `eval_on_start`: False
267
+ - `use_liger_kernel`: False
268
+ - `eval_use_gather_object`: False
269
+ - `batch_sampler`: batch_sampler
270
+ - `multi_dataset_batch_sampler`: proportional
271
+
272
+ </details>
273
+
274
+ ### Training Logs
275
+ <details><summary>Click to expand</summary>
276
+
277
+ | Epoch | Step | Training Loss | loss |
278
+ |:------:|:-----:|:-------------:|:------:|
279
+ | 0.0094 | 100 | 16.0755 | - |
280
+ | 0.0188 | 200 | 13.0643 | - |
281
+ | 0.0282 | 300 | 9.3474 | - |
282
+ | 0.0376 | 400 | 8.2606 | - |
283
+ | 0.0469 | 500 | 8.084 | - |
284
+ | 0.0563 | 600 | 8.0581 | - |
285
+ | 0.0657 | 700 | 8.0175 | - |
286
+ | 0.0751 | 800 | 8.0285 | - |
287
+ | 0.0845 | 900 | 8.0024 | - |
288
+ | 0.0939 | 1000 | 8.0161 | - |
289
+ | 0.1033 | 1100 | 7.9941 | - |
290
+ | 0.1127 | 1200 | 8.0233 | - |
291
+ | 0.1221 | 1300 | 8.0141 | - |
292
+ | 0.1314 | 1400 | 7.9644 | - |
293
+ | 0.1408 | 1500 | 8.0311 | - |
294
+ | 0.1502 | 1600 | 8.0306 | - |
295
+ | 0.1596 | 1700 | 7.989 | - |
296
+ | 0.1690 | 1800 | 8.0034 | - |
297
+ | 0.1784 | 1900 | 8.0107 | - |
298
+ | 0.1878 | 2000 | 7.9737 | - |
299
+ | 0.1972 | 2100 | 7.9827 | - |
300
+ | 0.2066 | 2200 | 8.0389 | - |
301
+ | 0.2159 | 2300 | 7.973 | - |
302
+ | 0.2253 | 2400 | 7.9669 | - |
303
+ | 0.2347 | 2500 | 8.0296 | - |
304
+ | 0.2441 | 2600 | 7.9984 | - |
305
+ | 0.2535 | 2700 | 7.9772 | - |
306
+ | 0.2629 | 2800 | 7.9838 | - |
307
+ | 0.2723 | 2900 | 7.9816 | - |
308
+ | 0.2817 | 3000 | 8.0021 | - |
309
+ | 0.2911 | 3100 | 7.9715 | - |
310
+ | 0.3004 | 3200 | 7.9809 | - |
311
+ | 0.3098 | 3300 | 7.9849 | - |
312
+ | 0.3192 | 3400 | 7.9463 | - |
313
+ | 0.3286 | 3500 | 8.0067 | - |
314
+ | 0.3380 | 3600 | 7.9431 | - |
315
+ | 0.3474 | 3700 | 7.9877 | - |
316
+ | 0.3568 | 3800 | 7.9494 | - |
317
+ | 0.3662 | 3900 | 7.9466 | - |
318
+ | 0.3756 | 4000 | 7.9708 | - |
319
+ | 0.3849 | 4100 | 7.9525 | - |
320
+ | 0.3943 | 4200 | 7.9322 | - |
321
+ | 0.4037 | 4300 | 7.9415 | - |
322
+ | 0.4131 | 4400 | 7.9932 | - |
323
+ | 0.4225 | 4500 | 7.9481 | - |
324
+ | 0.4319 | 4600 | 7.976 | - |
325
+ | 0.4413 | 4700 | 7.971 | - |
326
+ | 0.4507 | 4800 | 7.9647 | - |
327
+ | 0.4601 | 4900 | 7.9217 | - |
328
+ | 0.4694 | 5000 | 7.9374 | 7.9518 |
329
+ | 0.4788 | 5100 | 7.9026 | - |
330
+ | 0.4882 | 5200 | 7.9304 | - |
331
+ | 0.4976 | 5300 | 7.9148 | - |
332
+ | 0.5070 | 5400 | 7.9538 | - |
333
+ | 0.5164 | 5500 | 8.0002 | - |
334
+ | 0.5258 | 5600 | 7.9571 | - |
335
+ | 0.5352 | 5700 | 7.932 | - |
336
+ | 0.5445 | 5800 | 7.9047 | - |
337
+ | 0.5539 | 5900 | 7.9353 | - |
338
+ | 0.5633 | 6000 | 7.9203 | - |
339
+ | 0.5727 | 6100 | 7.8967 | - |
340
+ | 0.5821 | 6200 | 7.9414 | - |
341
+ | 0.5915 | 6300 | 7.9631 | - |
342
+ | 0.6009 | 6400 | 7.9606 | - |
343
+ | 0.6103 | 6500 | 7.9377 | - |
344
+ | 0.6197 | 6600 | 7.9108 | - |
345
+ | 0.6290 | 6700 | 7.9225 | - |
346
+ | 0.6384 | 6800 | 7.9154 | - |
347
+ | 0.6478 | 6900 | 7.9191 | - |
348
+ | 0.6572 | 7000 | 7.8903 | - |
349
+ | 0.6666 | 7100 | 7.9213 | - |
350
+ | 0.6760 | 7200 | 7.9202 | - |
351
+ | 0.6854 | 7300 | 7.8998 | - |
352
+ | 0.6948 | 7400 | 7.9153 | - |
353
+ | 0.7042 | 7500 | 7.9037 | - |
354
+ | 0.7135 | 7600 | 7.9146 | - |
355
+ | 0.7229 | 7700 | 7.8972 | - |
356
+ | 0.7323 | 7800 | 7.9374 | - |
357
+ | 0.7417 | 7900 | 7.8647 | - |
358
+ | 0.7511 | 8000 | 7.8915 | - |
359
+ | 0.7605 | 8100 | 7.8846 | - |
360
+ | 0.7699 | 8200 | 7.8988 | - |
361
+ | 0.7793 | 8300 | 7.8702 | - |
362
+ | 0.7887 | 8400 | 7.923 | - |
363
+ | 0.7980 | 8500 | 7.891 | - |
364
+ | 0.8074 | 8600 | 7.8832 | - |
365
+ | 0.8168 | 8700 | 7.8726 | - |
366
+ | 0.8262 | 8800 | 7.8813 | - |
367
+ | 0.8356 | 8900 | 7.8986 | - |
368
+ | 0.8450 | 9000 | 7.8743 | - |
369
+ | 0.8544 | 9100 | 7.8791 | - |
370
+ | 0.8638 | 9200 | 7.8783 | - |
371
+ | 0.8732 | 9300 | 7.8528 | - |
372
+ | 0.8825 | 9400 | 7.8864 | - |
373
+ | 0.8919 | 9500 | 7.8989 | - |
374
+ | 0.9013 | 9600 | 7.8617 | - |
375
+ | 0.9107 | 9700 | 7.8371 | - |
376
+ | 0.9201 | 9800 | 7.8566 | - |
377
+ | 0.9295 | 9900 | 7.8776 | - |
378
+ | 0.9389 | 10000 | 7.8558 | 7.8492 |
379
+ | 0.9483 | 10100 | 7.848 | - |
380
+ | 0.9577 | 10200 | 7.8227 | - |
381
+ | 0.9670 | 10300 | 7.8311 | - |
382
+ | 0.9764 | 10400 | 7.8437 | - |
383
+ | 0.9858 | 10500 | 7.8454 | - |
384
+ | 0.9952 | 10600 | 7.8362 | - |
385
+ | 1.0046 | 10700 | 7.8681 | - |
386
+ | 1.0140 | 10800 | 7.8745 | - |
387
+ | 1.0234 | 10900 | 7.8339 | - |
388
+ | 1.0328 | 11000 | 7.8458 | - |
389
+ | 1.0422 | 11100 | 7.8493 | - |
390
+ | 1.0515 | 11200 | 7.8317 | - |
391
+ | 1.0609 | 11300 | 7.841 | - |
392
+ | 1.0703 | 11400 | 7.8292 | - |
393
+ | 1.0797 | 11500 | 7.8121 | - |
394
+ | 1.0891 | 11600 | 7.8165 | - |
395
+ | 1.0985 | 11700 | 7.8259 | - |
396
+ | 1.1079 | 11800 | 7.8303 | - |
397
+ | 1.1173 | 11900 | 7.809 | - |
398
+ | 1.1267 | 12000 | 7.818 | - |
399
+ | 1.1360 | 12100 | 7.8071 | - |
400
+ | 1.1454 | 12200 | 7.801 | - |
401
+ | 1.1548 | 12300 | 7.8123 | - |
402
+ | 1.1642 | 12400 | 7.8203 | - |
403
+ | 1.1736 | 12500 | 7.8609 | - |
404
+
405
+ </details>
406
+
407
+ ### Framework Versions
408
+ - Python: 3.8.10
409
+ - Sentence Transformers: 3.1.1
410
+ - Transformers: 4.45.2
411
+ - PyTorch: 2.4.1+cu118
412
+ - Accelerate: 1.0.1
413
+ - Datasets: 3.0.1
414
+ - Tokenizers: 0.20.3
415
+
416
+ ## Citation
417
+
418
+ ### BibTeX
419
+
420
+ #### Sentence Transformers
421
+ ```bibtex
422
+ @inproceedings{reimers-2019-sentence-bert,
423
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
424
+ author = "Reimers, Nils and Gurevych, Iryna",
425
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
426
+ month = "11",
427
+ year = "2019",
428
+ publisher = "Association for Computational Linguistics",
429
+ url = "https://arxiv.org/abs/1908.10084",
430
+ }
431
+ ```
432
+
433
+ #### CoSENTLoss
434
+ ```bibtex
435
+ @online{kexuefm-8847,
436
+ title={CoSENT: A more efficient sentence vector scheme than Sentence-BERT},
437
+ author={Su Jianlin},
438
+ year={2022},
439
+ month={Jan},
440
+ url={https://kexue.fm/archives/8847},
441
+ }
442
+ ```
443
+
444
+ <!--
445
+ ## Glossary
446
+
447
+ *Clearly define terms in order to be accessible across audiences.*
448
+ -->
449
+
450
+ <!--
451
+ ## Model Card Authors
452
+
453
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
454
+ -->
455
+
456
+ <!--
457
+ ## Model Card Contact
458
+
459
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
460
+ -->
checkpoint-12500/config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "sentence-transformers/all-mpnet-base-v2",
3
+ "architectures": [
4
+ "MPNetModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "eos_token_id": 2,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-05,
15
+ "max_position_embeddings": 514,
16
+ "model_type": "mpnet",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 12,
19
+ "pad_token_id": 1,
20
+ "relative_attention_num_buckets": 32,
21
+ "torch_dtype": "float32",
22
+ "transformers_version": "4.45.2",
23
+ "vocab_size": 30527
24
+ }
checkpoint-12500/config_sentence_transformers.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "sentence_transformers": "3.1.1",
4
+ "transformers": "4.45.2",
5
+ "pytorch": "2.4.1+cu118"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null,
9
+ "similarity_fn_name": null
10
+ }
checkpoint-12500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdc1d64e7d32dc5eb973469c43c8ec586c89d8ffe02604942f4a03b14ab1c0ff
3
+ size 437967672
checkpoint-12500/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-12500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d36ce949f20ef8c90fcdd060fa0b64dff246db3aceb978b5110b9c700d2748de
3
+ size 871331770
checkpoint-12500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4e518be47a55c93dae6a5973321ecf007f6933e9b74e8893c75352d61eb351f
3
+ size 14244
checkpoint-12500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdf044c829f706dde98889ca9528498e7683377a1ca11017846fdcafef082e9d
3
+ size 1064
checkpoint-12500/sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 384,
3
+ "do_lower_case": false
4
+ }
checkpoint-12500/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "[UNK]",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
checkpoint-12500/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-12500/tokenizer_config.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": true,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "104": {
36
+ "content": "[UNK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "30526": {
44
+ "content": "<mask>",
45
+ "lstrip": true,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ }
51
+ },
52
+ "bos_token": "<s>",
53
+ "clean_up_tokenization_spaces": false,
54
+ "cls_token": "<s>",
55
+ "do_lower_case": true,
56
+ "eos_token": "</s>",
57
+ "mask_token": "<mask>",
58
+ "max_length": 128,
59
+ "model_max_length": 384,
60
+ "pad_to_multiple_of": null,
61
+ "pad_token": "<pad>",
62
+ "pad_token_type_id": 0,
63
+ "padding_side": "right",
64
+ "sep_token": "</s>",
65
+ "stride": 0,
66
+ "strip_accents": null,
67
+ "tokenize_chinese_chars": true,
68
+ "tokenizer_class": "MPNetTokenizer",
69
+ "truncation_side": "right",
70
+ "truncation_strategy": "longest_first",
71
+ "unk_token": "[UNK]"
72
+ }
checkpoint-12500/trainer_state.json ADDED
@@ -0,0 +1,924 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.1735987231245892,
5
+ "eval_steps": 5000,
6
+ "global_step": 12500,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.009388789784996713,
13
+ "grad_norm": 72.80598449707031,
14
+ "learning_rate": 9.103707179727828e-07,
15
+ "loss": 16.0755,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.018777579569993427,
20
+ "grad_norm": 72.60921478271484,
21
+ "learning_rate": 1.8301267010793056e-06,
22
+ "loss": 13.0643,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.02816636935499014,
27
+ "grad_norm": 10.017908096313477,
28
+ "learning_rate": 2.7592679493195683e-06,
29
+ "loss": 9.3474,
30
+ "step": 300
31
+ },
32
+ {
33
+ "epoch": 0.03755515913998685,
34
+ "grad_norm": 5.694988250732422,
35
+ "learning_rate": 3.6977944626935713e-06,
36
+ "loss": 8.2606,
37
+ "step": 400
38
+ },
39
+ {
40
+ "epoch": 0.04694394892498357,
41
+ "grad_norm": 4.844100475311279,
42
+ "learning_rate": 4.6363209760675744e-06,
43
+ "loss": 8.084,
44
+ "step": 500
45
+ },
46
+ {
47
+ "epoch": 0.05633273870998028,
48
+ "grad_norm": 3.6125142574310303,
49
+ "learning_rate": 5.574847489441577e-06,
50
+ "loss": 8.0581,
51
+ "step": 600
52
+ },
53
+ {
54
+ "epoch": 0.06572152849497699,
55
+ "grad_norm": 3.166046380996704,
56
+ "learning_rate": 6.51337400281558e-06,
57
+ "loss": 8.0175,
58
+ "step": 700
59
+ },
60
+ {
61
+ "epoch": 0.0751103182799737,
62
+ "grad_norm": 2.600433111190796,
63
+ "learning_rate": 7.451900516189583e-06,
64
+ "loss": 8.0285,
65
+ "step": 800
66
+ },
67
+ {
68
+ "epoch": 0.08449910806497042,
69
+ "grad_norm": 2.3805315494537354,
70
+ "learning_rate": 8.390427029563585e-06,
71
+ "loss": 8.0024,
72
+ "step": 900
73
+ },
74
+ {
75
+ "epoch": 0.09388789784996714,
76
+ "grad_norm": 12.382240295410156,
77
+ "learning_rate": 9.328953542937589e-06,
78
+ "loss": 8.0161,
79
+ "step": 1000
80
+ },
81
+ {
82
+ "epoch": 0.10327668763496385,
83
+ "grad_norm": 2.7355728149414062,
84
+ "learning_rate": 1.0267480056311592e-05,
85
+ "loss": 7.9941,
86
+ "step": 1100
87
+ },
88
+ {
89
+ "epoch": 0.11266547741996057,
90
+ "grad_norm": 2.0243470668792725,
91
+ "learning_rate": 1.1206006569685594e-05,
92
+ "loss": 8.0233,
93
+ "step": 1200
94
+ },
95
+ {
96
+ "epoch": 0.12205426720495728,
97
+ "grad_norm": 1.9162158966064453,
98
+ "learning_rate": 1.2144533083059597e-05,
99
+ "loss": 8.0141,
100
+ "step": 1300
101
+ },
102
+ {
103
+ "epoch": 0.13144305698995398,
104
+ "grad_norm": 11.409939765930176,
105
+ "learning_rate": 1.3083059596433601e-05,
106
+ "loss": 7.9644,
107
+ "step": 1400
108
+ },
109
+ {
110
+ "epoch": 0.1408318467749507,
111
+ "grad_norm": 1.712424635887146,
112
+ "learning_rate": 1.4021586109807603e-05,
113
+ "loss": 8.0311,
114
+ "step": 1500
115
+ },
116
+ {
117
+ "epoch": 0.1502206365599474,
118
+ "grad_norm": 2.4589834213256836,
119
+ "learning_rate": 1.4960112623181606e-05,
120
+ "loss": 8.0306,
121
+ "step": 1600
122
+ },
123
+ {
124
+ "epoch": 0.15960942634494413,
125
+ "grad_norm": 1.7343533039093018,
126
+ "learning_rate": 1.589863913655561e-05,
127
+ "loss": 7.989,
128
+ "step": 1700
129
+ },
130
+ {
131
+ "epoch": 0.16899821612994084,
132
+ "grad_norm": 2.0726826190948486,
133
+ "learning_rate": 1.6837165649929613e-05,
134
+ "loss": 8.0034,
135
+ "step": 1800
136
+ },
137
+ {
138
+ "epoch": 0.17838700591493756,
139
+ "grad_norm": 1.7758458852767944,
140
+ "learning_rate": 1.7775692163303613e-05,
141
+ "loss": 8.0107,
142
+ "step": 1900
143
+ },
144
+ {
145
+ "epoch": 0.18777579569993427,
146
+ "grad_norm": 3.3816475868225098,
147
+ "learning_rate": 1.8714218676677617e-05,
148
+ "loss": 7.9737,
149
+ "step": 2000
150
+ },
151
+ {
152
+ "epoch": 0.197164585484931,
153
+ "grad_norm": 1.8136950731277466,
154
+ "learning_rate": 1.965274519005162e-05,
155
+ "loss": 7.9827,
156
+ "step": 2100
157
+ },
158
+ {
159
+ "epoch": 0.2065533752699277,
160
+ "grad_norm": 1.7819303274154663,
161
+ "learning_rate": 1.9934275728965626e-05,
162
+ "loss": 8.0389,
163
+ "step": 2200
164
+ },
165
+ {
166
+ "epoch": 0.21594216505492442,
167
+ "grad_norm": 2.269160509109497,
168
+ "learning_rate": 1.9829951489228525e-05,
169
+ "loss": 7.973,
170
+ "step": 2300
171
+ },
172
+ {
173
+ "epoch": 0.22533095483992113,
174
+ "grad_norm": 3.3508036136627197,
175
+ "learning_rate": 1.972562724949142e-05,
176
+ "loss": 7.9669,
177
+ "step": 2400
178
+ },
179
+ {
180
+ "epoch": 0.23471974462491785,
181
+ "grad_norm": 1.674142599105835,
182
+ "learning_rate": 1.962130300975432e-05,
183
+ "loss": 8.0296,
184
+ "step": 2500
185
+ },
186
+ {
187
+ "epoch": 0.24410853440991456,
188
+ "grad_norm": 1.454300880432129,
189
+ "learning_rate": 1.9516978770017215e-05,
190
+ "loss": 7.9984,
191
+ "step": 2600
192
+ },
193
+ {
194
+ "epoch": 0.2534973241949113,
195
+ "grad_norm": 2.2951695919036865,
196
+ "learning_rate": 1.9412654530280113e-05,
197
+ "loss": 7.9772,
198
+ "step": 2700
199
+ },
200
+ {
201
+ "epoch": 0.26288611397990796,
202
+ "grad_norm": 6.295051574707031,
203
+ "learning_rate": 1.930833029054301e-05,
204
+ "loss": 7.9838,
205
+ "step": 2800
206
+ },
207
+ {
208
+ "epoch": 0.2722749037649047,
209
+ "grad_norm": 1.8874555826187134,
210
+ "learning_rate": 1.9204006050805904e-05,
211
+ "loss": 7.9816,
212
+ "step": 2900
213
+ },
214
+ {
215
+ "epoch": 0.2816636935499014,
216
+ "grad_norm": 20.835277557373047,
217
+ "learning_rate": 1.9099681811068803e-05,
218
+ "loss": 8.0021,
219
+ "step": 3000
220
+ },
221
+ {
222
+ "epoch": 0.29105248333489814,
223
+ "grad_norm": 2.1683876514434814,
224
+ "learning_rate": 1.8995357571331702e-05,
225
+ "loss": 7.9715,
226
+ "step": 3100
227
+ },
228
+ {
229
+ "epoch": 0.3004412731198948,
230
+ "grad_norm": 1.6533387899398804,
231
+ "learning_rate": 1.8891033331594598e-05,
232
+ "loss": 7.9809,
233
+ "step": 3200
234
+ },
235
+ {
236
+ "epoch": 0.30983006290489157,
237
+ "grad_norm": 4.595189094543457,
238
+ "learning_rate": 1.8786709091857496e-05,
239
+ "loss": 7.9849,
240
+ "step": 3300
241
+ },
242
+ {
243
+ "epoch": 0.31921885268988826,
244
+ "grad_norm": 1.994147539138794,
245
+ "learning_rate": 1.8682384852120392e-05,
246
+ "loss": 7.9463,
247
+ "step": 3400
248
+ },
249
+ {
250
+ "epoch": 0.328607642474885,
251
+ "grad_norm": 1.961474895477295,
252
+ "learning_rate": 1.8578060612383287e-05,
253
+ "loss": 8.0067,
254
+ "step": 3500
255
+ },
256
+ {
257
+ "epoch": 0.3379964322598817,
258
+ "grad_norm": 24.005535125732422,
259
+ "learning_rate": 1.8473736372646186e-05,
260
+ "loss": 7.9431,
261
+ "step": 3600
262
+ },
263
+ {
264
+ "epoch": 0.34738522204487843,
265
+ "grad_norm": 1.9433845281600952,
266
+ "learning_rate": 1.8369412132909085e-05,
267
+ "loss": 7.9877,
268
+ "step": 3700
269
+ },
270
+ {
271
+ "epoch": 0.3567740118298751,
272
+ "grad_norm": 10.296500205993652,
273
+ "learning_rate": 1.826508789317198e-05,
274
+ "loss": 7.9494,
275
+ "step": 3800
276
+ },
277
+ {
278
+ "epoch": 0.36616280161487186,
279
+ "grad_norm": 2.194976568222046,
280
+ "learning_rate": 1.8160763653434876e-05,
281
+ "loss": 7.9466,
282
+ "step": 3900
283
+ },
284
+ {
285
+ "epoch": 0.37555159139986855,
286
+ "grad_norm": 1.5201098918914795,
287
+ "learning_rate": 1.8056439413697775e-05,
288
+ "loss": 7.9708,
289
+ "step": 4000
290
+ },
291
+ {
292
+ "epoch": 0.3849403811848653,
293
+ "grad_norm": 2.9077212810516357,
294
+ "learning_rate": 1.795211517396067e-05,
295
+ "loss": 7.9525,
296
+ "step": 4100
297
+ },
298
+ {
299
+ "epoch": 0.394329170969862,
300
+ "grad_norm": 2.041530132293701,
301
+ "learning_rate": 1.784779093422357e-05,
302
+ "loss": 7.9322,
303
+ "step": 4200
304
+ },
305
+ {
306
+ "epoch": 0.4037179607548587,
307
+ "grad_norm": 2.0275838375091553,
308
+ "learning_rate": 1.7743466694486468e-05,
309
+ "loss": 7.9415,
310
+ "step": 4300
311
+ },
312
+ {
313
+ "epoch": 0.4131067505398554,
314
+ "grad_norm": 2.000778913497925,
315
+ "learning_rate": 1.7639142454749364e-05,
316
+ "loss": 7.9932,
317
+ "step": 4400
318
+ },
319
+ {
320
+ "epoch": 0.42249554032485215,
321
+ "grad_norm": 2.2984609603881836,
322
+ "learning_rate": 1.753481821501226e-05,
323
+ "loss": 7.9481,
324
+ "step": 4500
325
+ },
326
+ {
327
+ "epoch": 0.43188433010984884,
328
+ "grad_norm": 1.0885875225067139,
329
+ "learning_rate": 1.7430493975275155e-05,
330
+ "loss": 7.976,
331
+ "step": 4600
332
+ },
333
+ {
334
+ "epoch": 0.4412731198948456,
335
+ "grad_norm": 4.441020488739014,
336
+ "learning_rate": 1.7326169735538053e-05,
337
+ "loss": 7.971,
338
+ "step": 4700
339
+ },
340
+ {
341
+ "epoch": 0.45066190967984227,
342
+ "grad_norm": 2.2056221961975098,
343
+ "learning_rate": 1.7221845495800952e-05,
344
+ "loss": 7.9647,
345
+ "step": 4800
346
+ },
347
+ {
348
+ "epoch": 0.460050699464839,
349
+ "grad_norm": 2.1192028522491455,
350
+ "learning_rate": 1.7117521256063848e-05,
351
+ "loss": 7.9217,
352
+ "step": 4900
353
+ },
354
+ {
355
+ "epoch": 0.4694394892498357,
356
+ "grad_norm": 14.011516571044922,
357
+ "learning_rate": 1.7013197016326747e-05,
358
+ "loss": 7.9374,
359
+ "step": 5000
360
+ },
361
+ {
362
+ "epoch": 0.4694394892498357,
363
+ "eval_loss": 7.951793193817139,
364
+ "eval_runtime": 900.8057,
365
+ "eval_samples_per_second": 378.358,
366
+ "eval_steps_per_second": 2.956,
367
+ "step": 5000
368
+ },
369
+ {
370
+ "epoch": 0.4788282790348324,
371
+ "grad_norm": 39.19038009643555,
372
+ "learning_rate": 1.6908872776589642e-05,
373
+ "loss": 7.9026,
374
+ "step": 5100
375
+ },
376
+ {
377
+ "epoch": 0.4882170688198291,
378
+ "grad_norm": 1.812458872795105,
379
+ "learning_rate": 1.6804548536852537e-05,
380
+ "loss": 7.9304,
381
+ "step": 5200
382
+ },
383
+ {
384
+ "epoch": 0.4976058586048258,
385
+ "grad_norm": 70.96247863769531,
386
+ "learning_rate": 1.6700224297115436e-05,
387
+ "loss": 7.9148,
388
+ "step": 5300
389
+ },
390
+ {
391
+ "epoch": 0.5069946483898226,
392
+ "grad_norm": 1.6605011224746704,
393
+ "learning_rate": 1.6595900057378335e-05,
394
+ "loss": 7.9538,
395
+ "step": 5400
396
+ },
397
+ {
398
+ "epoch": 0.5163834381748192,
399
+ "grad_norm": 2.0463483333587646,
400
+ "learning_rate": 1.649157581764123e-05,
401
+ "loss": 8.0002,
402
+ "step": 5500
403
+ },
404
+ {
405
+ "epoch": 0.5257722279598159,
406
+ "grad_norm": 3.8601722717285156,
407
+ "learning_rate": 1.6387251577904126e-05,
408
+ "loss": 7.9571,
409
+ "step": 5600
410
+ },
411
+ {
412
+ "epoch": 0.5351610177448127,
413
+ "grad_norm": 2.184122323989868,
414
+ "learning_rate": 1.6282927338167025e-05,
415
+ "loss": 7.932,
416
+ "step": 5700
417
+ },
418
+ {
419
+ "epoch": 0.5445498075298094,
420
+ "grad_norm": 2.165367603302002,
421
+ "learning_rate": 1.617860309842992e-05,
422
+ "loss": 7.9047,
423
+ "step": 5800
424
+ },
425
+ {
426
+ "epoch": 0.5539385973148061,
427
+ "grad_norm": 1.5312166213989258,
428
+ "learning_rate": 1.607427885869282e-05,
429
+ "loss": 7.9353,
430
+ "step": 5900
431
+ },
432
+ {
433
+ "epoch": 0.5633273870998028,
434
+ "grad_norm": 21.75490379333496,
435
+ "learning_rate": 1.5969954618955715e-05,
436
+ "loss": 7.9203,
437
+ "step": 6000
438
+ },
439
+ {
440
+ "epoch": 0.5727161768847996,
441
+ "grad_norm": 1.8674250841140747,
442
+ "learning_rate": 1.5865630379218614e-05,
443
+ "loss": 7.8967,
444
+ "step": 6100
445
+ },
446
+ {
447
+ "epoch": 0.5821049666697963,
448
+ "grad_norm": 49.87809371948242,
449
+ "learning_rate": 1.576130613948151e-05,
450
+ "loss": 7.9414,
451
+ "step": 6200
452
+ },
453
+ {
454
+ "epoch": 0.591493756454793,
455
+ "grad_norm": 54.42366409301758,
456
+ "learning_rate": 1.5658025142141778e-05,
457
+ "loss": 7.9631,
458
+ "step": 6300
459
+ },
460
+ {
461
+ "epoch": 0.6008825462397896,
462
+ "grad_norm": 37.58320236206055,
463
+ "learning_rate": 1.5554744144802047e-05,
464
+ "loss": 7.9606,
465
+ "step": 6400
466
+ },
467
+ {
468
+ "epoch": 0.6102713360247864,
469
+ "grad_norm": 3.1502482891082764,
470
+ "learning_rate": 1.5450419905064945e-05,
471
+ "loss": 7.9377,
472
+ "step": 6500
473
+ },
474
+ {
475
+ "epoch": 0.6196601258097831,
476
+ "grad_norm": 2.5369224548339844,
477
+ "learning_rate": 1.534609566532784e-05,
478
+ "loss": 7.9108,
479
+ "step": 6600
480
+ },
481
+ {
482
+ "epoch": 0.6290489155947798,
483
+ "grad_norm": 2.5891005992889404,
484
+ "learning_rate": 1.5241771425590736e-05,
485
+ "loss": 7.9225,
486
+ "step": 6700
487
+ },
488
+ {
489
+ "epoch": 0.6384377053797765,
490
+ "grad_norm": 1.7794080972671509,
491
+ "learning_rate": 1.5137447185853635e-05,
492
+ "loss": 7.9154,
493
+ "step": 6800
494
+ },
495
+ {
496
+ "epoch": 0.6478264951647733,
497
+ "grad_norm": 2.324805974960327,
498
+ "learning_rate": 1.5033122946116532e-05,
499
+ "loss": 7.9191,
500
+ "step": 6900
501
+ },
502
+ {
503
+ "epoch": 0.65721528494977,
504
+ "grad_norm": 2.601715564727783,
505
+ "learning_rate": 1.4928798706379428e-05,
506
+ "loss": 7.8903,
507
+ "step": 7000
508
+ },
509
+ {
510
+ "epoch": 0.6666040747347667,
511
+ "grad_norm": 2.4438092708587646,
512
+ "learning_rate": 1.4824474466642325e-05,
513
+ "loss": 7.9213,
514
+ "step": 7100
515
+ },
516
+ {
517
+ "epoch": 0.6759928645197634,
518
+ "grad_norm": 8.118125915527344,
519
+ "learning_rate": 1.4720150226905224e-05,
520
+ "loss": 7.9202,
521
+ "step": 7200
522
+ },
523
+ {
524
+ "epoch": 0.6853816543047602,
525
+ "grad_norm": 3.908555746078491,
526
+ "learning_rate": 1.461582598716812e-05,
527
+ "loss": 7.8998,
528
+ "step": 7300
529
+ },
530
+ {
531
+ "epoch": 0.6947704440897569,
532
+ "grad_norm": 2.72293758392334,
533
+ "learning_rate": 1.4511501747431017e-05,
534
+ "loss": 7.9153,
535
+ "step": 7400
536
+ },
537
+ {
538
+ "epoch": 0.7041592338747535,
539
+ "grad_norm": 3.108797073364258,
540
+ "learning_rate": 1.4407177507693915e-05,
541
+ "loss": 7.9037,
542
+ "step": 7500
543
+ },
544
+ {
545
+ "epoch": 0.7135480236597502,
546
+ "grad_norm": 2.6256439685821533,
547
+ "learning_rate": 1.430285326795681e-05,
548
+ "loss": 7.9146,
549
+ "step": 7600
550
+ },
551
+ {
552
+ "epoch": 0.7229368134447469,
553
+ "grad_norm": 3.5525624752044678,
554
+ "learning_rate": 1.4198529028219708e-05,
555
+ "loss": 7.8972,
556
+ "step": 7700
557
+ },
558
+ {
559
+ "epoch": 0.7323256032297437,
560
+ "grad_norm": 2.6983673572540283,
561
+ "learning_rate": 1.4094204788482607e-05,
562
+ "loss": 7.9374,
563
+ "step": 7800
564
+ },
565
+ {
566
+ "epoch": 0.7417143930147404,
567
+ "grad_norm": 1.545486569404602,
568
+ "learning_rate": 1.3989880548745502e-05,
569
+ "loss": 7.8647,
570
+ "step": 7900
571
+ },
572
+ {
573
+ "epoch": 0.7511031827997371,
574
+ "grad_norm": 2.5116941928863525,
575
+ "learning_rate": 1.38855563090084e-05,
576
+ "loss": 7.8915,
577
+ "step": 8000
578
+ },
579
+ {
580
+ "epoch": 0.7604919725847338,
581
+ "grad_norm": 1.8576518297195435,
582
+ "learning_rate": 1.3781232069271295e-05,
583
+ "loss": 7.8846,
584
+ "step": 8100
585
+ },
586
+ {
587
+ "epoch": 0.7698807623697306,
588
+ "grad_norm": 3.3226571083068848,
589
+ "learning_rate": 1.3676907829534194e-05,
590
+ "loss": 7.8988,
591
+ "step": 8200
592
+ },
593
+ {
594
+ "epoch": 0.7792695521547273,
595
+ "grad_norm": 2.946324586868286,
596
+ "learning_rate": 1.3572583589797091e-05,
597
+ "loss": 7.8702,
598
+ "step": 8300
599
+ },
600
+ {
601
+ "epoch": 0.788658341939724,
602
+ "grad_norm": 2.5089969635009766,
603
+ "learning_rate": 1.3468259350059986e-05,
604
+ "loss": 7.923,
605
+ "step": 8400
606
+ },
607
+ {
608
+ "epoch": 0.7980471317247206,
609
+ "grad_norm": 2.2807912826538086,
610
+ "learning_rate": 1.3363935110322885e-05,
611
+ "loss": 7.891,
612
+ "step": 8500
613
+ },
614
+ {
615
+ "epoch": 0.8074359215097174,
616
+ "grad_norm": 2.5889735221862793,
617
+ "learning_rate": 1.3259610870585782e-05,
618
+ "loss": 7.8832,
619
+ "step": 8600
620
+ },
621
+ {
622
+ "epoch": 0.8168247112947141,
623
+ "grad_norm": 2.8306784629821777,
624
+ "learning_rate": 1.3155286630848678e-05,
625
+ "loss": 7.8726,
626
+ "step": 8700
627
+ },
628
+ {
629
+ "epoch": 0.8262135010797108,
630
+ "grad_norm": 2.626786231994629,
631
+ "learning_rate": 1.3050962391111577e-05,
632
+ "loss": 7.8813,
633
+ "step": 8800
634
+ },
635
+ {
636
+ "epoch": 0.8356022908647075,
637
+ "grad_norm": 3.195319414138794,
638
+ "learning_rate": 1.2946638151374474e-05,
639
+ "loss": 7.8986,
640
+ "step": 8900
641
+ },
642
+ {
643
+ "epoch": 0.8449910806497043,
644
+ "grad_norm": 5.254043102264404,
645
+ "learning_rate": 1.284231391163737e-05,
646
+ "loss": 7.8743,
647
+ "step": 9000
648
+ },
649
+ {
650
+ "epoch": 0.854379870434701,
651
+ "grad_norm": 2.9493279457092285,
652
+ "learning_rate": 1.2737989671900267e-05,
653
+ "loss": 7.8791,
654
+ "step": 9100
655
+ },
656
+ {
657
+ "epoch": 0.8637686602196977,
658
+ "grad_norm": 3.130415439605713,
659
+ "learning_rate": 1.2633665432163165e-05,
660
+ "loss": 7.8783,
661
+ "step": 9200
662
+ },
663
+ {
664
+ "epoch": 0.8731574500046944,
665
+ "grad_norm": 4.030152797698975,
666
+ "learning_rate": 1.2529341192426061e-05,
667
+ "loss": 7.8528,
668
+ "step": 9300
669
+ },
670
+ {
671
+ "epoch": 0.8825462397896912,
672
+ "grad_norm": 2.9882099628448486,
673
+ "learning_rate": 1.2425016952688958e-05,
674
+ "loss": 7.8864,
675
+ "step": 9400
676
+ },
677
+ {
678
+ "epoch": 0.8919350295746878,
679
+ "grad_norm": 3.802172899246216,
680
+ "learning_rate": 1.2320692712951855e-05,
681
+ "loss": 7.8989,
682
+ "step": 9500
683
+ },
684
+ {
685
+ "epoch": 0.9013238193596845,
686
+ "grad_norm": 2.724433183670044,
687
+ "learning_rate": 1.2216368473214752e-05,
688
+ "loss": 7.8617,
689
+ "step": 9600
690
+ },
691
+ {
692
+ "epoch": 0.9107126091446812,
693
+ "grad_norm": 2.459376573562622,
694
+ "learning_rate": 1.211204423347765e-05,
695
+ "loss": 7.8371,
696
+ "step": 9700
697
+ },
698
+ {
699
+ "epoch": 0.920101398929678,
700
+ "grad_norm": 4.715926647186279,
701
+ "learning_rate": 1.2007719993740547e-05,
702
+ "loss": 7.8566,
703
+ "step": 9800
704
+ },
705
+ {
706
+ "epoch": 0.9294901887146747,
707
+ "grad_norm": 2.6845057010650635,
708
+ "learning_rate": 1.1903395754003444e-05,
709
+ "loss": 7.8776,
710
+ "step": 9900
711
+ },
712
+ {
713
+ "epoch": 0.9388789784996714,
714
+ "grad_norm": 2.62907075881958,
715
+ "learning_rate": 1.1799071514266341e-05,
716
+ "loss": 7.8558,
717
+ "step": 10000
718
+ },
719
+ {
720
+ "epoch": 0.9388789784996714,
721
+ "eval_loss": 7.849188327789307,
722
+ "eval_runtime": 1155.7489,
723
+ "eval_samples_per_second": 294.897,
724
+ "eval_steps_per_second": 2.304,
725
+ "step": 10000
726
+ },
727
+ {
728
+ "epoch": 0.9482677682846681,
729
+ "grad_norm": 4.570381164550781,
730
+ "learning_rate": 1.1694747274529237e-05,
731
+ "loss": 7.848,
732
+ "step": 10100
733
+ },
734
+ {
735
+ "epoch": 0.9576565580696648,
736
+ "grad_norm": 21.764062881469727,
737
+ "learning_rate": 1.1590423034792135e-05,
738
+ "loss": 7.8227,
739
+ "step": 10200
740
+ },
741
+ {
742
+ "epoch": 0.9670453478546616,
743
+ "grad_norm": 18.442140579223633,
744
+ "learning_rate": 1.1486098795055033e-05,
745
+ "loss": 7.8311,
746
+ "step": 10300
747
+ },
748
+ {
749
+ "epoch": 0.9764341376396583,
750
+ "grad_norm": 4.737902641296387,
751
+ "learning_rate": 1.1381774555317928e-05,
752
+ "loss": 7.8437,
753
+ "step": 10400
754
+ },
755
+ {
756
+ "epoch": 0.9858229274246549,
757
+ "grad_norm": 3.0295650959014893,
758
+ "learning_rate": 1.1277450315580827e-05,
759
+ "loss": 7.8454,
760
+ "step": 10500
761
+ },
762
+ {
763
+ "epoch": 0.9952117172096516,
764
+ "grad_norm": 3.0269651412963867,
765
+ "learning_rate": 1.1173126075843724e-05,
766
+ "loss": 7.8362,
767
+ "step": 10600
768
+ },
769
+ {
770
+ "epoch": 1.0046005069946484,
771
+ "grad_norm": 4.033662796020508,
772
+ "learning_rate": 1.1069845078503991e-05,
773
+ "loss": 7.8681,
774
+ "step": 10700
775
+ },
776
+ {
777
+ "epoch": 1.013989296779645,
778
+ "grad_norm": 3.5319488048553467,
779
+ "learning_rate": 1.0965520838766888e-05,
780
+ "loss": 7.8745,
781
+ "step": 10800
782
+ },
783
+ {
784
+ "epoch": 1.0233780865646418,
785
+ "grad_norm": 2.7731130123138428,
786
+ "learning_rate": 1.0861196599029787e-05,
787
+ "loss": 7.8339,
788
+ "step": 10900
789
+ },
790
+ {
791
+ "epoch": 1.0327668763496385,
792
+ "grad_norm": 4.000971794128418,
793
+ "learning_rate": 1.0756872359292683e-05,
794
+ "loss": 7.8458,
795
+ "step": 11000
796
+ },
797
+ {
798
+ "epoch": 1.0421556661346352,
799
+ "grad_norm": 15.05604362487793,
800
+ "learning_rate": 1.065254811955558e-05,
801
+ "loss": 7.8493,
802
+ "step": 11100
803
+ },
804
+ {
805
+ "epoch": 1.0515444559196319,
806
+ "grad_norm": 4.498584747314453,
807
+ "learning_rate": 1.0548223879818477e-05,
808
+ "loss": 7.8317,
809
+ "step": 11200
810
+ },
811
+ {
812
+ "epoch": 1.0609332457046288,
813
+ "grad_norm": 2.8218085765838623,
814
+ "learning_rate": 1.0443899640081374e-05,
815
+ "loss": 7.841,
816
+ "step": 11300
817
+ },
818
+ {
819
+ "epoch": 1.0703220354896255,
820
+ "grad_norm": 3.627685785293579,
821
+ "learning_rate": 1.0339575400344271e-05,
822
+ "loss": 7.8292,
823
+ "step": 11400
824
+ },
825
+ {
826
+ "epoch": 1.0797108252746221,
827
+ "grad_norm": 4.804520606994629,
828
+ "learning_rate": 1.0235251160607167e-05,
829
+ "loss": 7.8121,
830
+ "step": 11500
831
+ },
832
+ {
833
+ "epoch": 1.0890996150596188,
834
+ "grad_norm": 15.256156921386719,
835
+ "learning_rate": 1.0130926920870066e-05,
836
+ "loss": 7.8165,
837
+ "step": 11600
838
+ },
839
+ {
840
+ "epoch": 1.0984884048446155,
841
+ "grad_norm": 3.684401273727417,
842
+ "learning_rate": 1.0026602681132963e-05,
843
+ "loss": 7.8259,
844
+ "step": 11700
845
+ },
846
+ {
847
+ "epoch": 1.1078771946296122,
848
+ "grad_norm": 3.7146763801574707,
849
+ "learning_rate": 9.92227844139586e-06,
850
+ "loss": 7.8303,
851
+ "step": 11800
852
+ },
853
+ {
854
+ "epoch": 1.117265984414609,
855
+ "grad_norm": 3.4437708854675293,
856
+ "learning_rate": 9.817954201658755e-06,
857
+ "loss": 7.809,
858
+ "step": 11900
859
+ },
860
+ {
861
+ "epoch": 1.1266547741996056,
862
+ "grad_norm": 4.232120990753174,
863
+ "learning_rate": 9.713629961921654e-06,
864
+ "loss": 7.818,
865
+ "step": 12000
866
+ },
867
+ {
868
+ "epoch": 1.1360435639846025,
869
+ "grad_norm": 3.4967739582061768,
870
+ "learning_rate": 9.60930572218455e-06,
871
+ "loss": 7.8071,
872
+ "step": 12100
873
+ },
874
+ {
875
+ "epoch": 1.1454323537695992,
876
+ "grad_norm": 10.542444229125977,
877
+ "learning_rate": 9.504981482447447e-06,
878
+ "loss": 7.801,
879
+ "step": 12200
880
+ },
881
+ {
882
+ "epoch": 1.1548211435545959,
883
+ "grad_norm": 3.744981527328491,
884
+ "learning_rate": 9.400657242710344e-06,
885
+ "loss": 7.8123,
886
+ "step": 12300
887
+ },
888
+ {
889
+ "epoch": 1.1642099333395926,
890
+ "grad_norm": 3.3549323081970215,
891
+ "learning_rate": 9.296333002973241e-06,
892
+ "loss": 7.8203,
893
+ "step": 12400
894
+ },
895
+ {
896
+ "epoch": 1.1735987231245892,
897
+ "grad_norm": 5.337845325469971,
898
+ "learning_rate": 9.192008763236138e-06,
899
+ "loss": 7.8609,
900
+ "step": 12500
901
+ }
902
+ ],
903
+ "logging_steps": 100,
904
+ "max_steps": 21302,
905
+ "num_input_tokens_seen": 0,
906
+ "num_train_epochs": 2,
907
+ "save_steps": 500,
908
+ "stateful_callbacks": {
909
+ "TrainerControl": {
910
+ "args": {
911
+ "should_epoch_stop": false,
912
+ "should_evaluate": false,
913
+ "should_log": false,
914
+ "should_save": true,
915
+ "should_training_stop": false
916
+ },
917
+ "attributes": {}
918
+ }
919
+ },
920
+ "total_flos": 0.0,
921
+ "train_batch_size": 128,
922
+ "trial_name": null,
923
+ "trial_params": null
924
+ }
checkpoint-12500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ea049d2d832069bea07040be4041d4f978baa54eb55f016301b0dad6a593f47
3
+ size 5496
checkpoint-12500/vocab.txt ADDED
The diff for this file is too large to render. See raw diff