leafxyz commited on
Commit
30cdd74
·
verified ·
1 Parent(s): 531e891

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -38,3 +38,4 @@ checkpoint-400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
  checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
  checkpoint-800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
40
  checkpoint-1000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
38
  checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
  checkpoint-800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
40
  checkpoint-1000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
41
+ checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-1200/1_Pooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "embedding_dimension": 1024,
3
+ "pooling_mode": "lasttoken",
4
+ "include_prompt": true
5
+ }
checkpoint-1200/README.md ADDED
@@ -0,0 +1,515 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - generated_from_trainer
7
+ - dataset_size:342061
8
+ - loss:CachedMultipleNegativesRankingLoss
9
+ base_model: prestoai/qwen3-embedding-0.6b-arabic-ecom
10
+ widget:
11
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
12
+ product that best matches it
13
+
14
+ Query: نسونكس Spray'
15
+ sentences:
16
+ - Nasonex - Nasal Spray
17
+ - كابل شحن مايكرو XKIN - 2.4A
18
+ - حلوى الشوكولاتة - Choco Lapki
19
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
20
+ product that best matches it
21
+
22
+ Query: مكرونة رقم 42'
23
+ sentences:
24
+ - بنطلون رجالي - 0112
25
+ - حقيبة حزام خصر - 4862
26
+ - مكرونة الجيد معكوفة رقم 42 - 500 غ
27
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
28
+ product that best matches it
29
+
30
+ Query: سباغيتي'
31
+ sentences:
32
+ - ملعب كرة قدم - DD18
33
+ - مكرونة معكوفة - Favelli
34
+ - مكرونة سباغيتي - Favelli
35
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
36
+ product that best matches it
37
+
38
+ Query: جبنة هواء'
39
+ sentences:
40
+ - جبنة - ابو الولد
41
+ - جبنة - Hawaa
42
+ - كاني طعام كلاب البالغين دجاج - 3 ك
43
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
44
+ product that best matches it
45
+
46
+ Query: شاحن تايب سي للسيارة'
47
+ sentences:
48
+ - شاحن سيارة قرين ليون مدخلين 36 وات مع كابل تايب سي - CBK
49
+ - صوص المكرونة هاينز - 365 غ
50
+ - بسكويت جولون بدون سكر شكلاتة ساندوتش
51
+ pipeline_tag: sentence-similarity
52
+ library_name: sentence-transformers
53
+ ---
54
+
55
+ # SentenceTransformer based on prestoai/qwen3-embedding-0.6b-arabic-ecom
56
+
57
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [prestoai/qwen3-embedding-0.6b-arabic-ecom](https://huggingface.co/prestoai/qwen3-embedding-0.6b-arabic-ecom) on the pairs_with_negatives and positives datasets. It maps sentences & paragraphs to a 1024-dimensional dense vector space and can be used for retrieval.
58
+
59
+ ## Model Details
60
+
61
+ ### Model Description
62
+ - **Model Type:** Sentence Transformer
63
+ - **Base model:** [prestoai/qwen3-embedding-0.6b-arabic-ecom](https://huggingface.co/prestoai/qwen3-embedding-0.6b-arabic-ecom) <!-- at revision 80f273fd53c6644d65e14a2ac1fbf74b8c924097 -->
64
+ - **Maximum Sequence Length:** 128 tokens
65
+ - **Output Dimensionality:** 1024 dimensions
66
+ - **Similarity Function:** Cosine Similarity
67
+ - **Supported Modality:** Text
68
+ - **Training Datasets:**
69
+ - pairs_with_negatives
70
+ - positives
71
+ <!-- - **Language:** Unknown -->
72
+ <!-- - **License:** Unknown -->
73
+
74
+ ### Model Sources
75
+
76
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
77
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
78
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
79
+
80
+ ### Full Model Architecture
81
+
82
+ ```
83
+ SentenceTransformer(
84
+ (0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'last_hidden_state'}}, 'module_output_name': 'token_embeddings', 'architecture': 'Qwen3Model'})
85
+ (1): Pooling({'embedding_dimension': 1024, 'pooling_mode': 'lasttoken', 'include_prompt': True})
86
+ (2): Normalize({})
87
+ )
88
+ ```
89
+
90
+ ## Usage
91
+
92
+ ### Direct Usage (Sentence Transformers)
93
+
94
+ First install the Sentence Transformers library:
95
+
96
+ ```bash
97
+ pip install -U sentence-transformers
98
+ ```
99
+ Then you can load this model and run inference.
100
+ ```python
101
+ from sentence_transformers import SentenceTransformer
102
+
103
+ # Download from the 🤗 Hub
104
+ model = SentenceTransformer("sentence_transformers_model_id")
105
+ # Run inference
106
+ queries = [
107
+ 'Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it\nQuery: شاحن تايب سي للسيارة',
108
+ ]
109
+ documents = [
110
+ 'شاحن سيارة قرين ليون مدخلين 36 وات مع كابل تايب سي - CBK',
111
+ 'بسكويت جولون بدون سكر شكلاتة ساندوتش',
112
+ 'صوص المكرونة هاينز - 365 غ',
113
+ ]
114
+ query_embeddings = model.encode_query(queries)
115
+ document_embeddings = model.encode_document(documents)
116
+ print(query_embeddings.shape, document_embeddings.shape)
117
+ # [1, 1024] [3, 1024]
118
+
119
+ # Get the similarity scores for the embeddings
120
+ similarities = model.similarity(query_embeddings, document_embeddings)
121
+ print(similarities)
122
+ # tensor([[ 0.6392, -0.0423, -0.0569]])
123
+ ```
124
+ <!--
125
+ ### Direct Usage (Transformers)
126
+
127
+ <details><summary>Click to see the direct usage in Transformers</summary>
128
+
129
+ </details>
130
+ -->
131
+
132
+ <!--
133
+ ### Downstream Usage (Sentence Transformers)
134
+
135
+ You can finetune this model on your own dataset.
136
+
137
+ <details><summary>Click to expand</summary>
138
+
139
+ </details>
140
+ -->
141
+
142
+ <!--
143
+ ### Out-of-Scope Use
144
+
145
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
146
+ -->
147
+
148
+ <!--
149
+ ## Bias, Risks and Limitations
150
+
151
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
152
+ -->
153
+
154
+ <!--
155
+ ### Recommendations
156
+
157
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
158
+ -->
159
+
160
+ ## Training Details
161
+
162
+ ### Training Datasets
163
+
164
+ #### pairs_with_negatives
165
+
166
+ * Dataset: pairs_with_negatives
167
+ * Size: 124,261 training samples
168
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
169
+ * Approximate statistics based on the first 1000 samples:
170
+ | | anchor | positive | negative |
171
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
172
+ | type | string | string | string |
173
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.44 tokens</li><li>max: 42 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 15.95 tokens</li><li>max: 49 tokens</li></ul> | <ul><li>min: 2 tokens</li><li>mean: 14.87 tokens</li><li>max: 40 tokens</li></ul> |
174
+ * Samples:
175
+ | anchor | positive | negative |
176
+ |:--------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------|:-------------------------------------------------------------------|
177
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: واقي للوجه دهني</code> | <code>Anthelios Oil Control (Dry Touch) - La Roche Posay</code> | <code>Anthelios Invisible Mist (Dry Touch) - La Roche Posay</code> |
178
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: تن منارة زليتن</code> | <code>تن منارة زليتن بزيت دوار الشمس - 160 غ</code> | <code>تن فاني بزيت دوار الشمس - 160 غ</code> |
179
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: فانتا زجاجة صغيرة</code> | <code>مشروب فانتا برتقال زجاجة - 330 مل</code> | <code>مشروب فانتا - 1 ل (برتقال)</code> |
180
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
181
+ ```json
182
+ {
183
+ "scale": 20.0,
184
+ "similarity_fct": "cos_sim",
185
+ "mini_batch_size": 8,
186
+ "gather_across_devices": false,
187
+ "directions": [
188
+ "query_to_doc"
189
+ ],
190
+ "partition_mode": "joint",
191
+ "hardness_mode": null,
192
+ "hardness_strength": 0.0
193
+ }
194
+ ```
195
+
196
+ #### positives
197
+
198
+ * Dataset: positives
199
+ * Size: 217,800 training samples
200
+ * Columns: <code>anchor</code> and <code>positive</code>
201
+ * Approximate statistics based on the first 1000 samples:
202
+ | | anchor | positive |
203
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
204
+ | type | string | string |
205
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.61 tokens</li><li>max: 52 tokens</li></ul> | <ul><li>min: 2 tokens</li><li>mean: 13.81 tokens</li><li>max: 41 tokens</li></ul> |
206
+ * Samples:
207
+ | anchor | positive |
208
+ |:-----------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------|
209
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: نحب جبنة القرية</code> | <code>ميرسين جبنة القرية 200 جم</code> |
210
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كابل شحن مايكرو Moxom A2.4</code> | <code>كابل شحن مايكرو Moxom - A2.4</code> |
211
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: dry idea</code> | <code>Dry idea (powder fresh)</code> |
212
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
213
+ ```json
214
+ {
215
+ "scale": 20.0,
216
+ "similarity_fct": "cos_sim",
217
+ "mini_batch_size": 8,
218
+ "gather_across_devices": false,
219
+ "directions": [
220
+ "query_to_doc"
221
+ ],
222
+ "partition_mode": "joint",
223
+ "hardness_mode": null,
224
+ "hardness_strength": 0.0
225
+ }
226
+ ```
227
+
228
+ ### Evaluation Datasets
229
+
230
+ #### pairs_with_negatives
231
+
232
+ * Dataset: pairs_with_negatives
233
+ * Size: 1,256 evaluation samples
234
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
235
+ * Approximate statistics based on the first 1000 samples:
236
+ | | anchor | positive | negative |
237
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
238
+ | type | string | string | string |
239
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.56 tokens</li><li>max: 40 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 16.04 tokens</li><li>max: 43 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 15.05 tokens</li><li>max: 33 tokens</li></ul> |
240
+ * Samples:
241
+ | anchor | positive | negative |
242
+ |:-------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------|:--------------------------------------|
243
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: هيبوتك بوزن</code> | <code>عطر Hypnotic Poison - PERFECTO COLLECTION</code> | <code>عطر Poison Girl - Dior</code> |
244
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: شاحن مايكرو 2.4A</code> | <code>شحن مايكرو Smila - 2.4A</code> | <code>شحن تايب سي Smila - 2.4A</code> |
245
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: تمر صعيدي</code> | <code>تمر صعيدي مشفوط</code> | <code>تمر قصيم مشفوط</code> |
246
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
247
+ ```json
248
+ {
249
+ "scale": 20.0,
250
+ "similarity_fct": "cos_sim",
251
+ "mini_batch_size": 8,
252
+ "gather_across_devices": false,
253
+ "directions": [
254
+ "query_to_doc"
255
+ ],
256
+ "partition_mode": "joint",
257
+ "hardness_mode": null,
258
+ "hardness_strength": 0.0
259
+ }
260
+ ```
261
+
262
+ #### positives
263
+
264
+ * Dataset: positives
265
+ * Size: 2,200 evaluation samples
266
+ * Columns: <code>anchor</code> and <code>positive</code>
267
+ * Approximate statistics based on the first 1000 samples:
268
+ | | anchor | positive |
269
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
270
+ | type | string | string |
271
+ | details | <ul><li>min: 24 tokens</li><li>mean: 29.27 tokens</li><li>max: 42 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 13.61 tokens</li><li>max: 44 tokens</li></ul> |
272
+ * Samples:
273
+ | anchor | positive |
274
+ |:-------------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------|
275
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: حاملة أدوات القطط</code> | <code>حاملة أدوات القطة - AA04</code> |
276
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كريم شمس أطفال</code> | <code>واقي شمس كريمي - Chicco</code> |
277
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كانديسارتان هيدروكلوروثيازيد</code> | <code>Candesartan and Hydrochlorothiazide 16mg/12.5mg</code> |
278
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
279
+ ```json
280
+ {
281
+ "scale": 20.0,
282
+ "similarity_fct": "cos_sim",
283
+ "mini_batch_size": 8,
284
+ "gather_across_devices": false,
285
+ "directions": [
286
+ "query_to_doc"
287
+ ],
288
+ "partition_mode": "joint",
289
+ "hardness_mode": null,
290
+ "hardness_strength": 0.0
291
+ }
292
+ ```
293
+
294
+ ### Training Hyperparameters
295
+ #### Non-Default Hyperparameters
296
+
297
+ - `per_device_train_batch_size`: 32
298
+ - `learning_rate`: 0.0001
299
+ - `num_train_epochs`: 1
300
+ - `warmup_steps`: 0.05
301
+ - `fp16`: True
302
+
303
+ #### All Hyperparameters
304
+ <details><summary>Click to expand</summary>
305
+
306
+ - `do_predict`: False
307
+ - `prediction_loss_only`: True
308
+ - `per_device_train_batch_size`: 32
309
+ - `per_device_eval_batch_size`: 8
310
+ - `gradient_accumulation_steps`: 1
311
+ - `eval_accumulation_steps`: None
312
+ - `torch_empty_cache_steps`: None
313
+ - `learning_rate`: 0.0001
314
+ - `weight_decay`: 0.0
315
+ - `adam_beta1`: 0.9
316
+ - `adam_beta2`: 0.999
317
+ - `adam_epsilon`: 1e-08
318
+ - `max_grad_norm`: 1.0
319
+ - `num_train_epochs`: 1
320
+ - `max_steps`: -1
321
+ - `lr_scheduler_type`: linear
322
+ - `lr_scheduler_kwargs`: None
323
+ - `warmup_ratio`: None
324
+ - `warmup_steps`: 0.05
325
+ - `log_level`: passive
326
+ - `log_level_replica`: warning
327
+ - `log_on_each_node`: True
328
+ - `logging_nan_inf_filter`: True
329
+ - `enable_jit_checkpoint`: False
330
+ - `save_on_each_node`: False
331
+ - `save_only_model`: False
332
+ - `restore_callback_states_from_checkpoint`: False
333
+ - `use_cpu`: False
334
+ - `seed`: 42
335
+ - `data_seed`: None
336
+ - `bf16`: False
337
+ - `fp16`: True
338
+ - `bf16_full_eval`: False
339
+ - `fp16_full_eval`: False
340
+ - `tf32`: None
341
+ - `local_rank`: -1
342
+ - `ddp_backend`: None
343
+ - `debug`: []
344
+ - `dataloader_drop_last`: False
345
+ - `dataloader_num_workers`: 0
346
+ - `dataloader_prefetch_factor`: None
347
+ - `disable_tqdm`: False
348
+ - `remove_unused_columns`: True
349
+ - `label_names`: None
350
+ - `load_best_model_at_end`: False
351
+ - `ignore_data_skip`: False
352
+ - `fsdp`: []
353
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
354
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
355
+ - `parallelism_config`: None
356
+ - `deepspeed`: None
357
+ - `label_smoothing_factor`: 0.0
358
+ - `optim`: adamw_torch_fused
359
+ - `optim_args`: None
360
+ - `group_by_length`: False
361
+ - `length_column_name`: length
362
+ - `project`: huggingface
363
+ - `trackio_space_id`: trackio
364
+ - `ddp_find_unused_parameters`: None
365
+ - `ddp_bucket_cap_mb`: None
366
+ - `ddp_broadcast_buffers`: False
367
+ - `dataloader_pin_memory`: True
368
+ - `dataloader_persistent_workers`: False
369
+ - `skip_memory_metrics`: True
370
+ - `push_to_hub`: False
371
+ - `resume_from_checkpoint`: None
372
+ - `hub_model_id`: None
373
+ - `hub_strategy`: every_save
374
+ - `hub_private_repo`: None
375
+ - `hub_always_push`: False
376
+ - `hub_revision`: None
377
+ - `gradient_checkpointing`: False
378
+ - `gradient_checkpointing_kwargs`: None
379
+ - `include_for_metrics`: []
380
+ - `eval_do_concat_batches`: True
381
+ - `auto_find_batch_size`: False
382
+ - `full_determinism`: False
383
+ - `ddp_timeout`: 1800
384
+ - `torch_compile`: False
385
+ - `torch_compile_backend`: None
386
+ - `torch_compile_mode`: None
387
+ - `include_num_input_tokens_seen`: no
388
+ - `neftune_noise_alpha`: None
389
+ - `optim_target_modules`: None
390
+ - `batch_eval_metrics`: False
391
+ - `eval_on_start`: False
392
+ - `use_liger_kernel`: False
393
+ - `liger_kernel_config`: None
394
+ - `eval_use_gather_object`: False
395
+ - `average_tokens_across_devices`: True
396
+ - `use_cache`: False
397
+ - `prompts`: None
398
+ - `batch_sampler`: batch_sampler
399
+ - `multi_dataset_batch_sampler`: proportional
400
+ - `router_mapping`: {}
401
+ - `learning_rate_mapping`: {}
402
+
403
+ </details>
404
+
405
+ ### Training Logs
406
+ | Epoch | Step | Training Loss | pairs with negatives loss | positives loss |
407
+ |:------:|:----:|:-------------:|:-------------------------:|:--------------:|
408
+ | 0.0023 | 25 | 0.4772 | - | - |
409
+ | 0.0047 | 50 | 0.4593 | - | - |
410
+ | 0.0070 | 75 | 0.3945 | - | - |
411
+ | 0.0094 | 100 | 0.3752 | - | - |
412
+ | 0.0117 | 125 | 0.4048 | - | - |
413
+ | 0.0140 | 150 | 0.4563 | - | - |
414
+ | 0.0164 | 175 | 0.3492 | - | - |
415
+ | 0.0187 | 200 | 0.4171 | 0.3915 | 0.1481 |
416
+ | 0.0210 | 225 | 0.4297 | - | - |
417
+ | 0.0234 | 250 | 0.4365 | - | - |
418
+ | 0.0257 | 275 | 0.4344 | - | - |
419
+ | 0.0281 | 300 | 0.4184 | - | - |
420
+ | 0.0304 | 325 | 0.4198 | - | - |
421
+ | 0.0327 | 350 | 0.4293 | - | - |
422
+ | 0.0351 | 375 | 0.4759 | - | - |
423
+ | 0.0374 | 400 | 0.3312 | 0.3695 | 0.1180 |
424
+ | 0.0398 | 425 | 0.3887 | - | - |
425
+ | 0.0421 | 450 | 0.4402 | - | - |
426
+ | 0.0444 | 475 | 0.4105 | - | - |
427
+ | 0.0468 | 500 | 0.3923 | - | - |
428
+ | 0.0491 | 525 | 0.3163 | - | - |
429
+ | 0.0514 | 550 | 0.3565 | - | - |
430
+ | 0.0538 | 575 | 0.3707 | - | - |
431
+ | 0.0561 | 600 | 0.3008 | 0.3388 | 0.1086 |
432
+ | 0.0585 | 625 | 0.3594 | - | - |
433
+ | 0.0608 | 650 | 0.3936 | - | - |
434
+ | 0.0631 | 675 | 0.3207 | - | - |
435
+ | 0.0655 | 700 | 0.3371 | - | - |
436
+ | 0.0678 | 725 | 0.3385 | - | - |
437
+ | 0.0702 | 750 | 0.2718 | - | - |
438
+ | 0.0725 | 775 | 0.4429 | - | - |
439
+ | 0.0748 | 800 | 0.2684 | 0.3453 | 0.1043 |
440
+ | 0.0772 | 825 | 0.2539 | - | - |
441
+ | 0.0795 | 850 | 0.3239 | - | - |
442
+ | 0.0818 | 875 | 0.2944 | - | - |
443
+ | 0.0842 | 900 | 0.3067 | - | - |
444
+ | 0.0865 | 925 | 0.3113 | - | - |
445
+ | 0.0889 | 950 | 0.3387 | - | - |
446
+ | 0.0912 | 975 | 0.2735 | - | - |
447
+ | 0.0935 | 1000 | 0.2985 | 0.3211 | 0.0891 |
448
+ | 0.0959 | 1025 | 0.3553 | - | - |
449
+ | 0.0982 | 1050 | 0.2568 | - | - |
450
+ | 0.1006 | 1075 | 0.3447 | - | - |
451
+ | 0.1029 | 1100 | 0.3239 | - | - |
452
+ | 0.1052 | 1125 | 0.3015 | - | - |
453
+ | 0.1076 | 1150 | 0.3865 | - | - |
454
+ | 0.1099 | 1175 | 0.2982 | - | - |
455
+ | 0.1122 | 1200 | 0.3105 | 0.3232 | 0.0829 |
456
+
457
+
458
+ ### Training Time
459
+ - **Training**: 1.3 hours
460
+
461
+ ### Framework Versions
462
+ - Python: 3.12.13
463
+ - Sentence Transformers: 5.4.1
464
+ - Transformers: 5.0.0
465
+ - PyTorch: 2.10.0+cu128
466
+ - Accelerate: 1.13.0
467
+ - Datasets: 5.0.0
468
+ - Tokenizers: 0.22.2
469
+
470
+ ## Citation
471
+
472
+ ### BibTeX
473
+
474
+ #### Sentence Transformers
475
+ ```bibtex
476
+ @inproceedings{reimers-2019-sentence-bert,
477
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
478
+ author = "Reimers, Nils and Gurevych, Iryna",
479
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
480
+ month = "11",
481
+ year = "2019",
482
+ publisher = "Association for Computational Linguistics",
483
+ url = "https://arxiv.org/abs/1908.10084",
484
+ }
485
+ ```
486
+
487
+ #### CachedMultipleNegativesRankingLoss
488
+ ```bibtex
489
+ @misc{gao2021scaling,
490
+ title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
491
+ author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
492
+ year={2021},
493
+ eprint={2101.06983},
494
+ archivePrefix={arXiv},
495
+ primaryClass={cs.LG}
496
+ }
497
+ ```
498
+
499
+ <!--
500
+ ## Glossary
501
+
502
+ *Clearly define terms in order to be accessible across audiences.*
503
+ -->
504
+
505
+ <!--
506
+ ## Model Card Authors
507
+
508
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
509
+ -->
510
+
511
+ <!--
512
+ ## Model Card Contact
513
+
514
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
515
+ -->
checkpoint-1200/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "prestoai/qwen3-embedding-0.6b-arabic-ecom",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": false,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 16,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "v_proj",
34
+ "gate_proj",
35
+ "down_proj",
36
+ "o_proj",
37
+ "k_proj",
38
+ "up_proj",
39
+ "q_proj"
40
+ ],
41
+ "target_parameters": null,
42
+ "task_type": "FEATURE_EXTRACTION",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
checkpoint-1200/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6efe62fabf6894020e446a8cad7ad5f9111cbc8c8dd56f9a0254c04313896801
3
+ size 40419816
checkpoint-1200/chat_template.jinja ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {{- messages[0].content + '\n\n' }}
5
+ {%- endif %}
6
+ {{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
7
+ {%- for tool in tools %}
8
+ {{- "\n" }}
9
+ {{- tool | tojson }}
10
+ {%- endfor %}
11
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
12
+ {%- else %}
13
+ {%- if messages[0].role == 'system' %}
14
+ {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
15
+ {%- endif %}
16
+ {%- endif %}
17
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
18
+ {%- for message in messages[::-1] %}
19
+ {%- set index = (messages|length - 1) - loop.index0 %}
20
+ {%- if ns.multi_step_tool and message.role == "user" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
21
+ {%- set ns.multi_step_tool = false %}
22
+ {%- set ns.last_query_index = index %}
23
+ {%- endif %}
24
+ {%- endfor %}
25
+ {%- for message in messages %}
26
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
27
+ {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
28
+ {%- elif message.role == "assistant" %}
29
+ {%- set content = message.content %}
30
+ {%- set reasoning_content = '' %}
31
+ {%- if message.reasoning_content is defined and message.reasoning_content is not none %}
32
+ {%- set reasoning_content = message.reasoning_content %}
33
+ {%- else %}
34
+ {%- if '</think>' in message.content %}
35
+ {%- set content = message.content.split('</think>')[-1].lstrip('\n') %}
36
+ {%- set reasoning_content = message.content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
37
+ {%- endif %}
38
+ {%- endif %}
39
+ {%- if loop.index0 > ns.last_query_index %}
40
+ {%- if loop.last or (not loop.last and reasoning_content) %}
41
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
42
+ {%- else %}
43
+ {{- '<|im_start|>' + message.role + '\n' + content }}
44
+ {%- endif %}
45
+ {%- else %}
46
+ {{- '<|im_start|>' + message.role + '\n' + content }}
47
+ {%- endif %}
48
+ {%- if message.tool_calls %}
49
+ {%- for tool_call in message.tool_calls %}
50
+ {%- if (loop.first and content) or (not loop.first) %}
51
+ {{- '\n' }}
52
+ {%- endif %}
53
+ {%- if tool_call.function %}
54
+ {%- set tool_call = tool_call.function %}
55
+ {%- endif %}
56
+ {{- '<tool_call>\n{"name": "' }}
57
+ {{- tool_call.name }}
58
+ {{- '", "arguments": ' }}
59
+ {%- if tool_call.arguments is string %}
60
+ {{- tool_call.arguments }}
61
+ {%- else %}
62
+ {{- tool_call.arguments | tojson }}
63
+ {%- endif %}
64
+ {{- '}\n</tool_call>' }}
65
+ {%- endfor %}
66
+ {%- endif %}
67
+ {{- '<|im_end|>\n' }}
68
+ {%- elif message.role == "tool" %}
69
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
70
+ {{- '<|im_start|>user' }}
71
+ {%- endif %}
72
+ {{- '\n<tool_response>\n' }}
73
+ {{- message.content }}
74
+ {{- '\n</tool_response>' }}
75
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
76
+ {{- '<|im_end|>\n' }}
77
+ {%- endif %}
78
+ {%- endif %}
79
+ {%- endfor %}
80
+ {%- if add_generation_prompt %}
81
+ {{- '<|im_start|>assistant\n' }}
82
+ {%- if enable_thinking is defined and enable_thinking is false %}
83
+ {{- '<think>\n\n</think>\n\n' }}
84
+ {%- endif %}
85
+ {%- endif %}
checkpoint-1200/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "pytorch": "2.10.0+cu128",
4
+ "sentence_transformers": "5.4.1",
5
+ "transformers": "5.0.0"
6
+ },
7
+ "default_prompt_name": null,
8
+ "model_type": "SentenceTransformer",
9
+ "prompts": {
10
+ "document": "",
11
+ "query": "Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it\nQuery: "
12
+ },
13
+ "similarity_fn_name": "cosine"
14
+ }
checkpoint-1200/modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.base.modules.transformer.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.sentence_transformer.modules.pooling.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.sentence_transformer.modules.normalize.Normalize"
19
+ }
20
+ ]
checkpoint-1200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2afe6b1b5cc9a43113c29b2cd657d087d1a85b4f1004b90bf09dddfdee3c9d0
3
+ size 81075515
checkpoint-1200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:760effd946150b7e654cdceb9e7130624823ff6d1d74405c7cfba1dcdc97eda4
3
+ size 14581
checkpoint-1200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18783150ac09b6b81cea5af47876a10bfe5f36c3d76aca4ffce5382bdfaf7b28
3
+ size 1383
checkpoint-1200/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ce8ce5c416c8018250428a2a831b80a0fe5fdf8eba39370afd9e563a74d41ac
3
+ size 1465
checkpoint-1200/sentence_bert_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer_task": "feature-extraction",
3
+ "modality_config": {
4
+ "text": {
5
+ "method": "forward",
6
+ "method_output_name": "last_hidden_state"
7
+ }
8
+ },
9
+ "module_output_name": "token_embeddings"
10
+ }
checkpoint-1200/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67b23115c7cace3b6700c333da5038e66e95a9c2c71531b6953563a8c8bb1dee
3
+ size 11423967
checkpoint-1200/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "is_local": false,
9
+ "local_files_only": false,
10
+ "model_max_length": 128,
11
+ "pad_token": "<|endoftext|>",
12
+ "split_special_tokens": false,
13
+ "tokenizer_class": "Qwen2Tokenizer",
14
+ "unk_token": null
15
+ }
checkpoint-1200/trainer_state.json ADDED
@@ -0,0 +1,466 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.11224394350388177,
6
+ "eval_steps": 200,
7
+ "global_step": 1200,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.0023384154896642037,
14
+ "grad_norm": 1.9981695413589478,
15
+ "learning_rate": 4.4859813084112145e-06,
16
+ "loss": 0.47716766357421875,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.004676830979328407,
21
+ "grad_norm": 1.9483280181884766,
22
+ "learning_rate": 9.158878504672897e-06,
23
+ "loss": 0.4592530059814453,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.0070152464689926105,
28
+ "grad_norm": 2.1290855407714844,
29
+ "learning_rate": 1.3831775700934579e-05,
30
+ "loss": 0.394501838684082,
31
+ "step": 75
32
+ },
33
+ {
34
+ "epoch": 0.009353661958656815,
35
+ "grad_norm": 2.8394205570220947,
36
+ "learning_rate": 1.8504672897196264e-05,
37
+ "loss": 0.3752219009399414,
38
+ "step": 100
39
+ },
40
+ {
41
+ "epoch": 0.011692077448321018,
42
+ "grad_norm": 2.5833706855773926,
43
+ "learning_rate": 2.3177570093457944e-05,
44
+ "loss": 0.40475303649902344,
45
+ "step": 125
46
+ },
47
+ {
48
+ "epoch": 0.014030492937985221,
49
+ "grad_norm": 2.622011661529541,
50
+ "learning_rate": 2.7850467289719627e-05,
51
+ "loss": 0.45625301361083986,
52
+ "step": 150
53
+ },
54
+ {
55
+ "epoch": 0.016368908427649424,
56
+ "grad_norm": 2.9406535625457764,
57
+ "learning_rate": 3.2523364485981314e-05,
58
+ "loss": 0.34918937683105467,
59
+ "step": 175
60
+ },
61
+ {
62
+ "epoch": 0.01870732391731363,
63
+ "grad_norm": 3.8391785621643066,
64
+ "learning_rate": 3.7196261682242994e-05,
65
+ "loss": 0.41711517333984377,
66
+ "step": 200
67
+ },
68
+ {
69
+ "epoch": 0.01870732391731363,
70
+ "eval_pairs_with_negatives_loss": 0.3914637565612793,
71
+ "eval_pairs_with_negatives_runtime": 44.0353,
72
+ "eval_pairs_with_negatives_samples_per_second": 28.523,
73
+ "eval_pairs_with_negatives_steps_per_second": 3.565,
74
+ "step": 200
75
+ },
76
+ {
77
+ "epoch": 0.01870732391731363,
78
+ "eval_positives_loss": 0.1480681300163269,
79
+ "eval_positives_runtime": 51.7193,
80
+ "eval_positives_samples_per_second": 42.537,
81
+ "eval_positives_steps_per_second": 5.317,
82
+ "step": 200
83
+ },
84
+ {
85
+ "epoch": 0.02104573940697783,
86
+ "grad_norm": 2.3628170490264893,
87
+ "learning_rate": 4.186915887850468e-05,
88
+ "loss": 0.4296648406982422,
89
+ "step": 225
90
+ },
91
+ {
92
+ "epoch": 0.023384154896642036,
93
+ "grad_norm": 2.698835611343384,
94
+ "learning_rate": 4.6542056074766354e-05,
95
+ "loss": 0.43652393341064455,
96
+ "step": 250
97
+ },
98
+ {
99
+ "epoch": 0.02572257038630624,
100
+ "grad_norm": 4.1166558265686035,
101
+ "learning_rate": 5.121495327102804e-05,
102
+ "loss": 0.4344004440307617,
103
+ "step": 275
104
+ },
105
+ {
106
+ "epoch": 0.028060985875970442,
107
+ "grad_norm": 4.551556587219238,
108
+ "learning_rate": 5.588785046728973e-05,
109
+ "loss": 0.41842262268066405,
110
+ "step": 300
111
+ },
112
+ {
113
+ "epoch": 0.030399401365634647,
114
+ "grad_norm": 2.3925657272338867,
115
+ "learning_rate": 6.056074766355141e-05,
116
+ "loss": 0.4197782897949219,
117
+ "step": 325
118
+ },
119
+ {
120
+ "epoch": 0.03273781685529885,
121
+ "grad_norm": 3.1023190021514893,
122
+ "learning_rate": 6.52336448598131e-05,
123
+ "loss": 0.4292975616455078,
124
+ "step": 350
125
+ },
126
+ {
127
+ "epoch": 0.03507623234496305,
128
+ "grad_norm": 4.1481032371521,
129
+ "learning_rate": 6.990654205607478e-05,
130
+ "loss": 0.4758753204345703,
131
+ "step": 375
132
+ },
133
+ {
134
+ "epoch": 0.03741464783462726,
135
+ "grad_norm": 3.8799962997436523,
136
+ "learning_rate": 7.457943925233644e-05,
137
+ "loss": 0.3311753845214844,
138
+ "step": 400
139
+ },
140
+ {
141
+ "epoch": 0.03741464783462726,
142
+ "eval_pairs_with_negatives_loss": 0.3694910407066345,
143
+ "eval_pairs_with_negatives_runtime": 44.2166,
144
+ "eval_pairs_with_negatives_samples_per_second": 28.406,
145
+ "eval_pairs_with_negatives_steps_per_second": 3.551,
146
+ "step": 400
147
+ },
148
+ {
149
+ "epoch": 0.03741464783462726,
150
+ "eval_positives_loss": 0.11797073483467102,
151
+ "eval_positives_runtime": 51.3355,
152
+ "eval_positives_samples_per_second": 42.855,
153
+ "eval_positives_steps_per_second": 5.357,
154
+ "step": 400
155
+ },
156
+ {
157
+ "epoch": 0.03975306332429146,
158
+ "grad_norm": 3.7068564891815186,
159
+ "learning_rate": 7.925233644859813e-05,
160
+ "loss": 0.388726806640625,
161
+ "step": 425
162
+ },
163
+ {
164
+ "epoch": 0.04209147881395566,
165
+ "grad_norm": 2.3407328128814697,
166
+ "learning_rate": 8.392523364485981e-05,
167
+ "loss": 0.4401784133911133,
168
+ "step": 450
169
+ },
170
+ {
171
+ "epoch": 0.04442989430361987,
172
+ "grad_norm": 3.0923948287963867,
173
+ "learning_rate": 8.85981308411215e-05,
174
+ "loss": 0.41050430297851564,
175
+ "step": 475
176
+ },
177
+ {
178
+ "epoch": 0.04676830979328407,
179
+ "grad_norm": 2.626753568649292,
180
+ "learning_rate": 9.327102803738317e-05,
181
+ "loss": 0.3923164749145508,
182
+ "step": 500
183
+ },
184
+ {
185
+ "epoch": 0.04910672528294827,
186
+ "grad_norm": 2.729557991027832,
187
+ "learning_rate": 9.794392523364486e-05,
188
+ "loss": 0.3163204765319824,
189
+ "step": 525
190
+ },
191
+ {
192
+ "epoch": 0.05144514077261248,
193
+ "grad_norm": 3.1612608432769775,
194
+ "learning_rate": 9.986215045293424e-05,
195
+ "loss": 0.35652976989746094,
196
+ "step": 550
197
+ },
198
+ {
199
+ "epoch": 0.05378355626227668,
200
+ "grad_norm": 2.224350690841675,
201
+ "learning_rate": 9.961599054745964e-05,
202
+ "loss": 0.37072269439697264,
203
+ "step": 575
204
+ },
205
+ {
206
+ "epoch": 0.056121971751940884,
207
+ "grad_norm": 3.207382917404175,
208
+ "learning_rate": 9.936983064198503e-05,
209
+ "loss": 0.3007513427734375,
210
+ "step": 600
211
+ },
212
+ {
213
+ "epoch": 0.056121971751940884,
214
+ "eval_pairs_with_negatives_loss": 0.3388254940509796,
215
+ "eval_pairs_with_negatives_runtime": 43.9558,
216
+ "eval_pairs_with_negatives_samples_per_second": 28.574,
217
+ "eval_pairs_with_negatives_steps_per_second": 3.572,
218
+ "step": 600
219
+ },
220
+ {
221
+ "epoch": 0.056121971751940884,
222
+ "eval_positives_loss": 0.10862970352172852,
223
+ "eval_positives_runtime": 51.7355,
224
+ "eval_positives_samples_per_second": 42.524,
225
+ "eval_positives_steps_per_second": 5.315,
226
+ "step": 600
227
+ },
228
+ {
229
+ "epoch": 0.058460387241605086,
230
+ "grad_norm": 4.494985103607178,
231
+ "learning_rate": 9.912367073651045e-05,
232
+ "loss": 0.35943916320800784,
233
+ "step": 625
234
+ },
235
+ {
236
+ "epoch": 0.060798802731269294,
237
+ "grad_norm": 3.7057838439941406,
238
+ "learning_rate": 9.887751083103585e-05,
239
+ "loss": 0.39355968475341796,
240
+ "step": 650
241
+ },
242
+ {
243
+ "epoch": 0.06313721822093349,
244
+ "grad_norm": 1.6454180479049683,
245
+ "learning_rate": 9.863135092556125e-05,
246
+ "loss": 0.32073760986328126,
247
+ "step": 675
248
+ },
249
+ {
250
+ "epoch": 0.0654756337105977,
251
+ "grad_norm": 3.901522636413574,
252
+ "learning_rate": 9.838519102008666e-05,
253
+ "loss": 0.3371485900878906,
254
+ "step": 700
255
+ },
256
+ {
257
+ "epoch": 0.0678140492002619,
258
+ "grad_norm": 2.2362253665924072,
259
+ "learning_rate": 9.813903111461206e-05,
260
+ "loss": 0.3385154724121094,
261
+ "step": 725
262
+ },
263
+ {
264
+ "epoch": 0.0701524646899261,
265
+ "grad_norm": 2.490204334259033,
266
+ "learning_rate": 9.789287120913746e-05,
267
+ "loss": 0.2718297004699707,
268
+ "step": 750
269
+ },
270
+ {
271
+ "epoch": 0.07249088017959031,
272
+ "grad_norm": 2.2875030040740967,
273
+ "learning_rate": 9.764671130366287e-05,
274
+ "loss": 0.44294830322265627,
275
+ "step": 775
276
+ },
277
+ {
278
+ "epoch": 0.07482929566925452,
279
+ "grad_norm": 1.40684175491333,
280
+ "learning_rate": 9.740055139818827e-05,
281
+ "loss": 0.2684394645690918,
282
+ "step": 800
283
+ },
284
+ {
285
+ "epoch": 0.07482929566925452,
286
+ "eval_pairs_with_negatives_loss": 0.34527480602264404,
287
+ "eval_pairs_with_negatives_runtime": 44.0724,
288
+ "eval_pairs_with_negatives_samples_per_second": 28.499,
289
+ "eval_pairs_with_negatives_steps_per_second": 3.562,
290
+ "step": 800
291
+ },
292
+ {
293
+ "epoch": 0.07482929566925452,
294
+ "eval_positives_loss": 0.10431129485368729,
295
+ "eval_positives_runtime": 51.6255,
296
+ "eval_positives_samples_per_second": 42.615,
297
+ "eval_positives_steps_per_second": 5.327,
298
+ "step": 800
299
+ },
300
+ {
301
+ "epoch": 0.07716771115891871,
302
+ "grad_norm": 2.8712947368621826,
303
+ "learning_rate": 9.715439149271367e-05,
304
+ "loss": 0.25394817352294924,
305
+ "step": 825
306
+ },
307
+ {
308
+ "epoch": 0.07950612664858292,
309
+ "grad_norm": 1.8017457723617554,
310
+ "learning_rate": 9.690823158723908e-05,
311
+ "loss": 0.32388198852539063,
312
+ "step": 850
313
+ },
314
+ {
315
+ "epoch": 0.08184454213824713,
316
+ "grad_norm": 2.945971727371216,
317
+ "learning_rate": 9.666207168176448e-05,
318
+ "loss": 0.29435272216796876,
319
+ "step": 875
320
+ },
321
+ {
322
+ "epoch": 0.08418295762791132,
323
+ "grad_norm": 2.1370229721069336,
324
+ "learning_rate": 9.641591177628988e-05,
325
+ "loss": 0.30667646408081056,
326
+ "step": 900
327
+ },
328
+ {
329
+ "epoch": 0.08652137311757553,
330
+ "grad_norm": 3.4880053997039795,
331
+ "learning_rate": 9.616975187081529e-05,
332
+ "loss": 0.3113353729248047,
333
+ "step": 925
334
+ },
335
+ {
336
+ "epoch": 0.08885978860723974,
337
+ "grad_norm": 1.221985101699829,
338
+ "learning_rate": 9.592359196534069e-05,
339
+ "loss": 0.33869354248046873,
340
+ "step": 950
341
+ },
342
+ {
343
+ "epoch": 0.09119820409690393,
344
+ "grad_norm": 2.5686190128326416,
345
+ "learning_rate": 9.567743205986609e-05,
346
+ "loss": 0.2734596633911133,
347
+ "step": 975
348
+ },
349
+ {
350
+ "epoch": 0.09353661958656814,
351
+ "grad_norm": 1.7126892805099487,
352
+ "learning_rate": 9.54312721543915e-05,
353
+ "loss": 0.2984904098510742,
354
+ "step": 1000
355
+ },
356
+ {
357
+ "epoch": 0.09353661958656814,
358
+ "eval_pairs_with_negatives_loss": 0.3210560381412506,
359
+ "eval_pairs_with_negatives_runtime": 44.9784,
360
+ "eval_pairs_with_negatives_samples_per_second": 27.925,
361
+ "eval_pairs_with_negatives_steps_per_second": 3.491,
362
+ "step": 1000
363
+ },
364
+ {
365
+ "epoch": 0.09353661958656814,
366
+ "eval_positives_loss": 0.08913413435220718,
367
+ "eval_positives_runtime": 52.3889,
368
+ "eval_positives_samples_per_second": 41.994,
369
+ "eval_positives_steps_per_second": 5.249,
370
+ "step": 1000
371
+ },
372
+ {
373
+ "epoch": 0.09587503507623235,
374
+ "grad_norm": 1.6900842189788818,
375
+ "learning_rate": 9.51851122489169e-05,
376
+ "loss": 0.355332145690918,
377
+ "step": 1025
378
+ },
379
+ {
380
+ "epoch": 0.09821345056589655,
381
+ "grad_norm": 0.7989633083343506,
382
+ "learning_rate": 9.49389523434423e-05,
383
+ "loss": 0.25679166793823244,
384
+ "step": 1050
385
+ },
386
+ {
387
+ "epoch": 0.10055186605556075,
388
+ "grad_norm": 1.7311954498291016,
389
+ "learning_rate": 9.469279243796771e-05,
390
+ "loss": 0.344718017578125,
391
+ "step": 1075
392
+ },
393
+ {
394
+ "epoch": 0.10289028154522496,
395
+ "grad_norm": 1.9694781303405762,
396
+ "learning_rate": 9.444663253249311e-05,
397
+ "loss": 0.3239378356933594,
398
+ "step": 1100
399
+ },
400
+ {
401
+ "epoch": 0.10522869703488916,
402
+ "grad_norm": 2.0283656120300293,
403
+ "learning_rate": 9.420047262701852e-05,
404
+ "loss": 0.3015079879760742,
405
+ "step": 1125
406
+ },
407
+ {
408
+ "epoch": 0.10756711252455337,
409
+ "grad_norm": 2.0422067642211914,
410
+ "learning_rate": 9.395431272154392e-05,
411
+ "loss": 0.3864572906494141,
412
+ "step": 1150
413
+ },
414
+ {
415
+ "epoch": 0.10990552801421756,
416
+ "grad_norm": 2.2227938175201416,
417
+ "learning_rate": 9.370815281606933e-05,
418
+ "loss": 0.2981968116760254,
419
+ "step": 1175
420
+ },
421
+ {
422
+ "epoch": 0.11224394350388177,
423
+ "grad_norm": 1.9632631540298462,
424
+ "learning_rate": 9.346199291059473e-05,
425
+ "loss": 0.3105415725708008,
426
+ "step": 1200
427
+ },
428
+ {
429
+ "epoch": 0.11224394350388177,
430
+ "eval_pairs_with_negatives_loss": 0.3232487738132477,
431
+ "eval_pairs_with_negatives_runtime": 44.5297,
432
+ "eval_pairs_with_negatives_samples_per_second": 28.206,
433
+ "eval_pairs_with_negatives_steps_per_second": 3.526,
434
+ "step": 1200
435
+ },
436
+ {
437
+ "epoch": 0.11224394350388177,
438
+ "eval_positives_loss": 0.08286190032958984,
439
+ "eval_positives_runtime": 52.0911,
440
+ "eval_positives_samples_per_second": 42.234,
441
+ "eval_positives_steps_per_second": 5.279,
442
+ "step": 1200
443
+ }
444
+ ],
445
+ "logging_steps": 25,
446
+ "max_steps": 10691,
447
+ "num_input_tokens_seen": 0,
448
+ "num_train_epochs": 1,
449
+ "save_steps": 200,
450
+ "stateful_callbacks": {
451
+ "TrainerControl": {
452
+ "args": {
453
+ "should_epoch_stop": false,
454
+ "should_evaluate": false,
455
+ "should_log": false,
456
+ "should_save": true,
457
+ "should_training_stop": false
458
+ },
459
+ "attributes": {}
460
+ }
461
+ },
462
+ "total_flos": 0.0,
463
+ "train_batch_size": 32,
464
+ "trial_name": null,
465
+ "trial_params": null
466
+ }
checkpoint-1200/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f64c13625404e7e8b2e3825b7c2b388aa17cbed32c8e2be6731de0c81a3f24a0
3
+ size 5457