leafxyz commited on
Commit
44f2ec9
·
verified ·
1 Parent(s): 7f9f883

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  checkpoint-200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
  checkpoint-400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  checkpoint-200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
  checkpoint-400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-600/1_Pooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "embedding_dimension": 1024,
3
+ "pooling_mode": "lasttoken",
4
+ "include_prompt": true
5
+ }
checkpoint-600/README.md ADDED
@@ -0,0 +1,491 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.5870, -0.0146, -0.0091]])
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
+
433
+
434
+ ### Training Time
435
+ - **Training**: 38.9 minutes
436
+
437
+ ### Framework Versions
438
+ - Python: 3.12.13
439
+ - Sentence Transformers: 5.4.1
440
+ - Transformers: 5.0.0
441
+ - PyTorch: 2.10.0+cu128
442
+ - Accelerate: 1.13.0
443
+ - Datasets: 5.0.0
444
+ - Tokenizers: 0.22.2
445
+
446
+ ## Citation
447
+
448
+ ### BibTeX
449
+
450
+ #### Sentence Transformers
451
+ ```bibtex
452
+ @inproceedings{reimers-2019-sentence-bert,
453
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
454
+ author = "Reimers, Nils and Gurevych, Iryna",
455
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
456
+ month = "11",
457
+ year = "2019",
458
+ publisher = "Association for Computational Linguistics",
459
+ url = "https://arxiv.org/abs/1908.10084",
460
+ }
461
+ ```
462
+
463
+ #### CachedMultipleNegativesRankingLoss
464
+ ```bibtex
465
+ @misc{gao2021scaling,
466
+ title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
467
+ author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
468
+ year={2021},
469
+ eprint={2101.06983},
470
+ archivePrefix={arXiv},
471
+ primaryClass={cs.LG}
472
+ }
473
+ ```
474
+
475
+ <!--
476
+ ## Glossary
477
+
478
+ *Clearly define terms in order to be accessible across audiences.*
479
+ -->
480
+
481
+ <!--
482
+ ## Model Card Authors
483
+
484
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
485
+ -->
486
+
487
+ <!--
488
+ ## Model Card Contact
489
+
490
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
491
+ -->
checkpoint-600/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-600/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fa40fd4c5aabed69771831d311c347fa24999fdb375a028e6999d683f5052b9
3
+ size 40419816
checkpoint-600/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-600/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-600/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-600/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ce17b3f0fafb8a0cdb00dd4caa459fc614bdaf30ff973b179098f2dd9a57d4f
3
+ size 81075515
checkpoint-600/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:733a2445e7b6bdaf5c99555cf91d2d5ec505741a1faf553e075890f8cc101633
3
+ size 14581
checkpoint-600/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb7fde5111803012042c93a73aa191336bb6e10b3ad44f6bd1d94fc7008a22b6
3
+ size 1383
checkpoint-600/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dede232c9e117374ca020d5aa7476810921c48adee2bbb5be1a24dc3227031dd
3
+ size 1465
checkpoint-600/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-600/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67b23115c7cace3b6700c333da5038e66e95a9c2c71531b6953563a8c8bb1dee
3
+ size 11423967
checkpoint-600/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-600/trainer_state.json ADDED
@@ -0,0 +1,250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.056121971751940884,
6
+ "eval_steps": 200,
7
+ "global_step": 600,
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
+ "logging_steps": 25,
230
+ "max_steps": 10691,
231
+ "num_input_tokens_seen": 0,
232
+ "num_train_epochs": 1,
233
+ "save_steps": 200,
234
+ "stateful_callbacks": {
235
+ "TrainerControl": {
236
+ "args": {
237
+ "should_epoch_stop": false,
238
+ "should_evaluate": false,
239
+ "should_log": false,
240
+ "should_save": true,
241
+ "should_training_stop": false
242
+ },
243
+ "attributes": {}
244
+ }
245
+ },
246
+ "total_flos": 0.0,
247
+ "train_batch_size": 32,
248
+ "trial_name": null,
249
+ "trial_params": null
250
+ }
checkpoint-600/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f64c13625404e7e8b2e3825b7c2b388aa17cbed32c8e2be6731de0c81a3f24a0
3
+ size 5457