Upload folder using huggingface_hub

checkpoint-9000/1_Pooling/config.json

@@ -0,0 +1,10 @@
+{
+  "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
+}

checkpoint-9000/README.md

@@ -0,0 +1,423 @@
+---
+base_model: sentence-transformers/all-mpnet-base-v2
+language:
+- en
+library_name: sentence-transformers
+license: apache-2.0
+pipeline_tag: sentence-similarity
+tags:
+- sentence-transformers
+- sentence-similarity
+- feature-extraction
+- generated_from_trainer
+- dataset_size:1363306
+- loss:AnglELoss
+widget:
+- source_sentence: labneh
+  sentences:
+  - iftar
+  - bathing suit
+  - coffee cup
+- source_sentence: Velvet flock Veil
+  sentences:
+  - mermaid purse
+  - veil
+  - mobile bag
+- source_sentence: Red lipstick
+  sentences:
+  - chemise dress
+  - tote
+  - rouge
+- source_sentence: Unisex Travel bag
+  sentences:
+  - spf
+  - basic vega ring
+  - travel backpack
+- source_sentence: jeremy hush book
+  sentences:
+  - chinese jumper
+  - perfume
+  - home automation device
+---
+
+# all-mpnet-base-v3-pair_score
+
+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.
+
+## Model Details
+
+### Model Description
+- **Model Type:** Sentence Transformer
+- **Base model:** [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) <!-- at revision 9a3225965996d404b775526de6dbfe85d3368642 -->
+- **Maximum Sequence Length:** 384 tokens
+- **Output Dimensionality:** 768 tokens
+- **Similarity Function:** Cosine Similarity
+<!-- - **Training Dataset:** Unknown -->
+- **Language:** en
+- **License:** apache-2.0
+
+### Model Sources
+
+- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
+- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
+- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
+
+### Full Model Architecture
+
+```
+SentenceTransformer(
+  (0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel 
+  (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})
+  (2): Normalize()
+)
+```
+
+## Usage
+
+### Direct Usage (Sentence Transformers)
+
+First install the Sentence Transformers library:
+
+```bash
+pip install -U sentence-transformers
+```
+
+Then you can load this model and run inference.
+```python
+from sentence_transformers import SentenceTransformer
+
+# Download from the 🤗 Hub
+model = SentenceTransformer("sentence_transformers_model_id")
+# Run inference
+sentences = [
+    'jeremy hush book',
+    'chinese jumper',
+    'perfume',
+]
+embeddings = model.encode(sentences)
+print(embeddings.shape)
+# [3, 768]
+
+# Get the similarity scores for the embeddings
+similarities = model.similarity(embeddings, embeddings)
+print(similarities.shape)
+# [3, 3]
+```
+
+<!--
+### Direct Usage (Transformers)
+
+<details><summary>Click to see the direct usage in Transformers</summary>
+
+</details>
+-->
+
+<!--
+### Downstream Usage (Sentence Transformers)
+
+You can finetune this model on your own dataset.
+
+<details><summary>Click to expand</summary>
+
+</details>
+-->
+
+<!--
+### Out-of-Scope Use
+
+*List how the model may foreseeably be misused and address what users ought not to do with the model.*
+-->
+
+<!--
+## Bias, Risks and Limitations
+
+*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
+-->
+
+<!--
+### Recommendations
+
+*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
+-->
+
+## Training Details
+
+### Training Hyperparameters
+#### Non-Default Hyperparameters
+
+- `eval_strategy`: steps
+- `per_device_train_batch_size`: 128
+- `per_device_eval_batch_size`: 128
+- `learning_rate`: 2e-05
+- `num_train_epochs`: 2
+- `warmup_ratio`: 0.1
+- `fp16`: True
+
+#### All Hyperparameters
+<details><summary>Click to expand</summary>
+
+- `overwrite_output_dir`: False
+- `do_predict`: False
+- `eval_strategy`: steps
+- `prediction_loss_only`: True
+- `per_device_train_batch_size`: 128
+- `per_device_eval_batch_size`: 128
+- `per_gpu_train_batch_size`: None
+- `per_gpu_eval_batch_size`: None
+- `gradient_accumulation_steps`: 1
+- `eval_accumulation_steps`: None
+- `torch_empty_cache_steps`: None
+- `learning_rate`: 2e-05
+- `weight_decay`: 0.0
+- `adam_beta1`: 0.9
+- `adam_beta2`: 0.999
+- `adam_epsilon`: 1e-08
+- `max_grad_norm`: 1.0
+- `num_train_epochs`: 2
+- `max_steps`: -1
+- `lr_scheduler_type`: linear
+- `lr_scheduler_kwargs`: {}
+- `warmup_ratio`: 0.1
+- `warmup_steps`: 0
+- `log_level`: passive
+- `log_level_replica`: warning
+- `log_on_each_node`: True
+- `logging_nan_inf_filter`: True
+- `save_safetensors`: True
+- `save_on_each_node`: False
+- `save_only_model`: False
+- `restore_callback_states_from_checkpoint`: False
+- `no_cuda`: False
+- `use_cpu`: False
+- `use_mps_device`: False
+- `seed`: 42
+- `data_seed`: None
+- `jit_mode_eval`: False
+- `use_ipex`: False
+- `bf16`: False
+- `fp16`: True
+- `fp16_opt_level`: O1
+- `half_precision_backend`: auto
+- `bf16_full_eval`: False
+- `fp16_full_eval`: False
+- `tf32`: None
+- `local_rank`: 0
+- `ddp_backend`: None
+- `tpu_num_cores`: None
+- `tpu_metrics_debug`: False
+- `debug`: []
+- `dataloader_drop_last`: False
+- `dataloader_num_workers`: 0
+- `dataloader_prefetch_factor`: None
+- `past_index`: -1
+- `disable_tqdm`: False
+- `remove_unused_columns`: True
+- `label_names`: None
+- `load_best_model_at_end`: False
+- `ignore_data_skip`: False
+- `fsdp`: []
+- `fsdp_min_num_params`: 0
+- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
+- `fsdp_transformer_layer_cls_to_wrap`: None
+- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
+- `deepspeed`: None
+- `label_smoothing_factor`: 0.0
+- `optim`: adamw_torch
+- `optim_args`: None
+- `adafactor`: False
+- `group_by_length`: False
+- `length_column_name`: length
+- `ddp_find_unused_parameters`: None
+- `ddp_bucket_cap_mb`: None
+- `ddp_broadcast_buffers`: False
+- `dataloader_pin_memory`: True
+- `dataloader_persistent_workers`: False
+- `skip_memory_metrics`: True
+- `use_legacy_prediction_loop`: False
+- `push_to_hub`: False
+- `resume_from_checkpoint`: None
+- `hub_model_id`: None
+- `hub_strategy`: every_save
+- `hub_private_repo`: False
+- `hub_always_push`: False
+- `gradient_checkpointing`: False
+- `gradient_checkpointing_kwargs`: None
+- `include_inputs_for_metrics`: False
+- `eval_do_concat_batches`: True
+- `fp16_backend`: auto
+- `push_to_hub_model_id`: None
+- `push_to_hub_organization`: None
+- `mp_parameters`: 
+- `auto_find_batch_size`: False
+- `full_determinism`: False
+- `torchdynamo`: None
+- `ray_scope`: last
+- `ddp_timeout`: 1800
+- `torch_compile`: False
+- `torch_compile_backend`: None
+- `torch_compile_mode`: None
+- `dispatch_batches`: None
+- `split_batches`: None
+- `include_tokens_per_second`: False
+- `include_num_input_tokens_seen`: False
+- `neftune_noise_alpha`: None
+- `optim_target_modules`: None
+- `batch_eval_metrics`: False
+- `eval_on_start`: False
+- `use_liger_kernel`: False
+- `eval_use_gather_object`: False
+- `batch_sampler`: batch_sampler
+- `multi_dataset_batch_sampler`: proportional
+
+</details>
+
+### Training Logs
+| Epoch  | Step | Training Loss | loss   |
+|:------:|:----:|:-------------:|:------:|
+| 0.0094 | 100  | 16.2337       | -      |
+| 0.0188 | 200  | 13.5901       | -      |
+| 0.0282 | 300  | 9.8565        | -      |
+| 0.0376 | 400  | 8.3332        | -      |
+| 0.0469 | 500  | 8.1261        | -      |
+| 0.0563 | 600  | 8.0697        | -      |
+| 0.0657 | 700  | 8.0298        | -      |
+| 0.0751 | 800  | 8.033         | -      |
+| 0.0845 | 900  | 7.9858        | -      |
+| 0.0939 | 1000 | 8.012         | -      |
+| 0.1033 | 1100 | 7.9745        | -      |
+| 0.1127 | 1200 | 8.0091        | -      |
+| 0.1221 | 1300 | 8.0221        | -      |
+| 0.1314 | 1400 | 7.9583        | -      |
+| 0.1408 | 1500 | 8.0031        | -      |
+| 0.1502 | 1600 | 7.9985        | -      |
+| 0.1596 | 1700 | 7.9647        | -      |
+| 0.1690 | 1800 | 7.9857        | -      |
+| 0.1784 | 1900 | 7.9806        | -      |
+| 0.1878 | 2000 | 7.9761        | -      |
+| 0.1972 | 2100 | 7.9696        | -      |
+| 0.2066 | 2200 | 8.0014        | -      |
+| 0.2159 | 2300 | 7.9546        | -      |
+| 0.2253 | 2400 | 7.9874        | -      |
+| 0.2347 | 2500 | 7.9846        | -      |
+| 0.2441 | 2600 | 7.9664        | -      |
+| 0.2535 | 2700 | 7.9725        | -      |
+| 0.2629 | 2800 | 7.9419        | -      |
+| 0.2723 | 2900 | 7.9786        | -      |
+| 0.2817 | 3000 | 7.9479        | -      |
+| 0.2911 | 3100 | 7.9526        | -      |
+| 0.3004 | 3200 | 7.9613        | -      |
+| 0.3098 | 3300 | 7.9994        | -      |
+| 0.3192 | 3400 | 7.9464        | -      |
+| 0.3286 | 3500 | 7.9429        | -      |
+| 0.3380 | 3600 | 7.9539        | -      |
+| 0.3474 | 3700 | 7.9699        | -      |
+| 0.3568 | 3800 | 7.9144        | -      |
+| 0.3662 | 3900 | 7.9424        | -      |
+| 0.3756 | 4000 | 7.9361        | -      |
+| 0.3849 | 4100 | 7.9144        | -      |
+| 0.3943 | 4200 | 7.907         | -      |
+| 0.4037 | 4300 | 7.9049        | -      |
+| 0.4131 | 4400 | 7.939         | -      |
+| 0.4225 | 4500 | 7.9067        | -      |
+| 0.4319 | 4600 | 7.9149        | -      |
+| 0.4413 | 4700 | 7.9705        | -      |
+| 0.4507 | 4800 | 7.8992        | -      |
+| 0.4601 | 4900 | 7.9077        | -      |
+| 0.4694 | 5000 | 7.8992        | 7.9167 |
+| 0.4788 | 5100 | 7.914         | -      |
+| 0.4882 | 5200 | 7.8913        | -      |
+| 0.4976 | 5300 | 7.8999        | -      |
+| 0.5070 | 5400 | 7.8818        | -      |
+| 0.5164 | 5500 | 7.9383        | -      |
+| 0.5258 | 5600 | 7.9094        | -      |
+| 0.5352 | 5700 | 7.8986        | -      |
+| 0.5445 | 5800 | 7.9015        | -      |
+| 0.5539 | 5900 | 7.9059        | -      |
+| 0.5633 | 6000 | 7.8524        | -      |
+| 0.5727 | 6100 | 7.8788        | -      |
+| 0.5821 | 6200 | 7.8712        | -      |
+| 0.5915 | 6300 | 7.8967        | -      |
+| 0.6009 | 6400 | 7.8677        | -      |
+| 0.6103 | 6500 | 7.9132        | -      |
+| 0.6197 | 6600 | 7.853         | -      |
+| 0.6290 | 6700 | 7.8968        | -      |
+| 0.6384 | 6800 | 7.8656        | -      |
+| 0.6478 | 6900 | 7.8801        | -      |
+| 0.6572 | 7000 | 7.8378        | -      |
+| 0.6666 | 7100 | 7.8554        | -      |
+| 0.6760 | 7200 | 7.8305        | -      |
+| 0.6854 | 7300 | 7.8613        | -      |
+| 0.6948 | 7400 | 7.8554        | -      |
+| 0.7042 | 7500 | 7.8653        | -      |
+| 0.7135 | 7600 | 7.8387        | -      |
+| 0.7229 | 7700 | 7.8513        | -      |
+| 0.7323 | 7800 | 7.8496        | -      |
+| 0.7417 | 7900 | 7.8276        | -      |
+| 0.7511 | 8000 | 7.8353        | -      |
+| 0.7605 | 8100 | 7.8103        | -      |
+| 0.7699 | 8200 | 7.8622        | -      |
+| 0.7793 | 8300 | 7.832         | -      |
+| 0.7887 | 8400 | 7.8349        | -      |
+| 0.7980 | 8500 | 7.855         | -      |
+| 0.8074 | 8600 | 7.8316        | -      |
+| 0.8168 | 8700 | 7.8066        | -      |
+| 0.8262 | 8800 | 7.8166        | -      |
+| 0.8356 | 8900 | 7.8588        | -      |
+| 0.8450 | 9000 | 7.8042        | -      |
+
+
+### Framework Versions
+- Python: 3.8.10
+- Sentence Transformers: 3.1.1
+- Transformers: 4.45.2
+- PyTorch: 2.4.1+cu118
+- Accelerate: 1.0.1
+- Datasets: 3.0.1
+- Tokenizers: 0.20.3
+
+## Citation
+
+### BibTeX
+
+#### Sentence Transformers
+```bibtex
+@inproceedings{reimers-2019-sentence-bert,
+    title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
+    author = "Reimers, Nils and Gurevych, Iryna",
+    booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
+    month = "11",
+    year = "2019",
+    publisher = "Association for Computational Linguistics",
+    url = "https://arxiv.org/abs/1908.10084",
+}
+```
+
+#### AnglELoss
+```bibtex
+@misc{li2023angleoptimized,
+    title={AnglE-optimized Text Embeddings},
+    author={Xianming Li and Jing Li},
+    year={2023},
+    eprint={2309.12871},
+    archivePrefix={arXiv},
+    primaryClass={cs.CL}
+}
+```
+
+<!--
+## Glossary
+
+*Clearly define terms in order to be accessible across audiences.*
+-->
+
+<!--
+## Model Card Authors
+
+*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
+-->
+
+<!--
+## Model Card Contact
+
+*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
+-->

checkpoint-9000/config.json

@@ -0,0 +1,24 @@
+{
+  "_name_or_path": "sentence-transformers/all-mpnet-base-v2",
+  "architectures": [
+    "MPNetModel"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "eos_token_id": 2,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 514,
+  "model_type": "mpnet",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "relative_attention_num_buckets": 32,
+  "torch_dtype": "float32",
+  "transformers_version": "4.45.2",
+  "vocab_size": 30527
+}

checkpoint-9000/config_sentence_transformers.json

@@ -0,0 +1,10 @@
+{
+  "__version__": {
+    "sentence_transformers": "3.1.1",
+    "transformers": "4.45.2",
+    "pytorch": "2.4.1+cu118"
+  },
+  "prompts": {},
+  "default_prompt_name": null,
+  "similarity_fn_name": null
+}

checkpoint-9000/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a09d57903d3b5848f6bf502200d1725609d92622f107ddd7e18d061a18cad82
+size 437967672

checkpoint-9000/modules.json

@@ -0,0 +1,20 @@
+[
+  {
+    "idx": 0,
+    "name": "0",
+    "path": "",
+    "type": "sentence_transformers.models.Transformer"
+  },
+  {
+    "idx": 1,
+    "name": "1",
+    "path": "1_Pooling",
+    "type": "sentence_transformers.models.Pooling"
+  },
+  {
+    "idx": 2,
+    "name": "2",
+    "path": "2_Normalize",
+    "type": "sentence_transformers.models.Normalize"
+  }
+]

checkpoint-9000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c9c7aadc973b006b88f3418fe79a6ba92378950427b7c9f8c228ea1718827de
+size 871331770

checkpoint-9000/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b27ac6bc3c0ac5c0de3b4b8ce04f227a4b4c3244e1abcee33a1c819c91a367fb
+size 14244

checkpoint-9000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d21106fa3a0fc4755c7872577921e5f529b66c78840687c2db03621467806511
+size 1064

checkpoint-9000/sentence_bert_config.json

@@ -0,0 +1,4 @@
+{
+  "max_seq_length": 384,
+  "do_lower_case": false
+}

checkpoint-9000/special_tokens_map.json

@@ -0,0 +1,51 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "cls_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-9000/tokenizer_config.json

@@ -0,0 +1,72 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "104": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "30526": {
+      "content": "<mask>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "cls_token": "<s>",
+  "do_lower_case": true,
+  "eos_token": "</s>",
+  "mask_token": "<mask>",
+  "max_length": 128,
+  "model_max_length": 384,
+  "pad_to_multiple_of": null,
+  "pad_token": "<pad>",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "</s>",
+  "stride": 0,
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "MPNetTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "[UNK]"
+}

checkpoint-9000/trainer_state.json

@@ -0,0 +1,671 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.8449910806497043,
+  "eval_steps": 5000,
+  "global_step": 9000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.009388789784996713,
+      "grad_norm": 87.77811431884766,
+      "learning_rate": 9.009854528390429e-07,
+      "loss": 16.2337,
+      "step": 100
+    },
+    {
+      "epoch": 0.018777579569993427,
+      "grad_norm": 85.86430358886719,
+      "learning_rate": 1.8395119662130456e-06,
+      "loss": 13.5901,
+      "step": 200
+    },
+    {
+      "epoch": 0.02816636935499014,
+      "grad_norm": 14.885255813598633,
+      "learning_rate": 2.7592679493195683e-06,
+      "loss": 9.8565,
+      "step": 300
+    },
+    {
+      "epoch": 0.03755515913998685,
+      "grad_norm": 6.9691972732543945,
+      "learning_rate": 3.6977944626935713e-06,
+      "loss": 8.3332,
+      "step": 400
+    },
+    {
+      "epoch": 0.04694394892498357,
+      "grad_norm": 5.612818241119385,
+      "learning_rate": 4.6363209760675744e-06,
+      "loss": 8.1261,
+      "step": 500
+    },
+    {
+      "epoch": 0.05633273870998028,
+      "grad_norm": 4.705409526824951,
+      "learning_rate": 5.574847489441577e-06,
+      "loss": 8.0697,
+      "step": 600
+    },
+    {
+      "epoch": 0.06572152849497699,
+      "grad_norm": 4.337332725524902,
+      "learning_rate": 6.51337400281558e-06,
+      "loss": 8.0298,
+      "step": 700
+    },
+    {
+      "epoch": 0.0751103182799737,
+      "grad_norm": 3.6314213275909424,
+      "learning_rate": 7.451900516189583e-06,
+      "loss": 8.033,
+      "step": 800
+    },
+    {
+      "epoch": 0.08449910806497042,
+      "grad_norm": 3.4845075607299805,
+      "learning_rate": 8.390427029563585e-06,
+      "loss": 7.9858,
+      "step": 900
+    },
+    {
+      "epoch": 0.09388789784996714,
+      "grad_norm": 5.188210487365723,
+      "learning_rate": 9.328953542937589e-06,
+      "loss": 8.012,
+      "step": 1000
+    },
+    {
+      "epoch": 0.10327668763496385,
+      "grad_norm": 3.0830442905426025,
+      "learning_rate": 1.0267480056311592e-05,
+      "loss": 7.9745,
+      "step": 1100
+    },
+    {
+      "epoch": 0.11266547741996057,
+      "grad_norm": 3.4729278087615967,
+      "learning_rate": 1.1206006569685594e-05,
+      "loss": 8.0091,
+      "step": 1200
+    },
+    {
+      "epoch": 0.12205426720495728,
+      "grad_norm": 2.329235076904297,
+      "learning_rate": 1.2144533083059597e-05,
+      "loss": 8.0221,
+      "step": 1300
+    },
+    {
+      "epoch": 0.13144305698995398,
+      "grad_norm": 2.7225279808044434,
+      "learning_rate": 1.3083059596433601e-05,
+      "loss": 7.9583,
+      "step": 1400
+    },
+    {
+      "epoch": 0.1408318467749507,
+      "grad_norm": 2.012805938720703,
+      "learning_rate": 1.4021586109807603e-05,
+      "loss": 8.0031,
+      "step": 1500
+    },
+    {
+      "epoch": 0.1502206365599474,
+      "grad_norm": 2.9397523403167725,
+      "learning_rate": 1.4960112623181606e-05,
+      "loss": 7.9985,
+      "step": 1600
+    },
+    {
+      "epoch": 0.15960942634494413,
+      "grad_norm": 2.356337308883667,
+      "learning_rate": 1.589863913655561e-05,
+      "loss": 7.9647,
+      "step": 1700
+    },
+    {
+      "epoch": 0.16899821612994084,
+      "grad_norm": 2.6846818923950195,
+      "learning_rate": 1.6837165649929613e-05,
+      "loss": 7.9857,
+      "step": 1800
+    },
+    {
+      "epoch": 0.17838700591493756,
+      "grad_norm": 2.0188565254211426,
+      "learning_rate": 1.7775692163303613e-05,
+      "loss": 7.9806,
+      "step": 1900
+    },
+    {
+      "epoch": 0.18777579569993427,
+      "grad_norm": 4.030488014221191,
+      "learning_rate": 1.8714218676677617e-05,
+      "loss": 7.9761,
+      "step": 2000
+    },
+    {
+      "epoch": 0.197164585484931,
+      "grad_norm": 4.183101654052734,
+      "learning_rate": 1.965274519005162e-05,
+      "loss": 7.9696,
+      "step": 2100
+    },
+    {
+      "epoch": 0.2065533752699277,
+      "grad_norm": 1.4769889116287231,
+      "learning_rate": 1.9934275728965626e-05,
+      "loss": 8.0014,
+      "step": 2200
+    },
+    {
+      "epoch": 0.21594216505492442,
+      "grad_norm": 2.1914358139038086,
+      "learning_rate": 1.9829951489228525e-05,
+      "loss": 7.9546,
+      "step": 2300
+    },
+    {
+      "epoch": 0.22533095483992113,
+      "grad_norm": 22.55516815185547,
+      "learning_rate": 1.972562724949142e-05,
+      "loss": 7.9874,
+      "step": 2400
+    },
+    {
+      "epoch": 0.23471974462491785,
+      "grad_norm": 1.635116457939148,
+      "learning_rate": 1.962130300975432e-05,
+      "loss": 7.9846,
+      "step": 2500
+    },
+    {
+      "epoch": 0.24410853440991456,
+      "grad_norm": 5.707275390625,
+      "learning_rate": 1.9516978770017215e-05,
+      "loss": 7.9664,
+      "step": 2600
+    },
+    {
+      "epoch": 0.2534973241949113,
+      "grad_norm": 4.194604396820068,
+      "learning_rate": 1.9412654530280113e-05,
+      "loss": 7.9725,
+      "step": 2700
+    },
+    {
+      "epoch": 0.26288611397990796,
+      "grad_norm": 2.0074055194854736,
+      "learning_rate": 1.930833029054301e-05,
+      "loss": 7.9419,
+      "step": 2800
+    },
+    {
+      "epoch": 0.2722749037649047,
+      "grad_norm": 26.4300479888916,
+      "learning_rate": 1.9204006050805904e-05,
+      "loss": 7.9786,
+      "step": 2900
+    },
+    {
+      "epoch": 0.2816636935499014,
+      "grad_norm": 2.5870931148529053,
+      "learning_rate": 1.9099681811068803e-05,
+      "loss": 7.9479,
+      "step": 3000
+    },
+    {
+      "epoch": 0.29105248333489814,
+      "grad_norm": 1.6209933757781982,
+      "learning_rate": 1.8995357571331702e-05,
+      "loss": 7.9526,
+      "step": 3100
+    },
+    {
+      "epoch": 0.3004412731198948,
+      "grad_norm": 19.398080825805664,
+      "learning_rate": 1.8891033331594598e-05,
+      "loss": 7.9613,
+      "step": 3200
+    },
+    {
+      "epoch": 0.30983006290489157,
+      "grad_norm": 2.124729871749878,
+      "learning_rate": 1.8786709091857496e-05,
+      "loss": 7.9994,
+      "step": 3300
+    },
+    {
+      "epoch": 0.31921885268988826,
+      "grad_norm": 2.701019763946533,
+      "learning_rate": 1.8682384852120392e-05,
+      "loss": 7.9464,
+      "step": 3400
+    },
+    {
+      "epoch": 0.328607642474885,
+      "grad_norm": 1.944600224494934,
+      "learning_rate": 1.8578060612383287e-05,
+      "loss": 7.9429,
+      "step": 3500
+    },
+    {
+      "epoch": 0.3379964322598817,
+      "grad_norm": 41.85493087768555,
+      "learning_rate": 1.8473736372646186e-05,
+      "loss": 7.9539,
+      "step": 3600
+    },
+    {
+      "epoch": 0.34738522204487843,
+      "grad_norm": 2.024019479751587,
+      "learning_rate": 1.8369412132909085e-05,
+      "loss": 7.9699,
+      "step": 3700
+    },
+    {
+      "epoch": 0.3567740118298751,
+      "grad_norm": 2.2833781242370605,
+      "learning_rate": 1.826508789317198e-05,
+      "loss": 7.9144,
+      "step": 3800
+    },
+    {
+      "epoch": 0.36616280161487186,
+      "grad_norm": 3.1666574478149414,
+      "learning_rate": 1.8160763653434876e-05,
+      "loss": 7.9424,
+      "step": 3900
+    },
+    {
+      "epoch": 0.37555159139986855,
+      "grad_norm": 1.7259443998336792,
+      "learning_rate": 1.8056439413697775e-05,
+      "loss": 7.9361,
+      "step": 4000
+    },
+    {
+      "epoch": 0.3849403811848653,
+      "grad_norm": 2.5588905811309814,
+      "learning_rate": 1.795211517396067e-05,
+      "loss": 7.9144,
+      "step": 4100
+    },
+    {
+      "epoch": 0.394329170969862,
+      "grad_norm": 2.542963743209839,
+      "learning_rate": 1.784779093422357e-05,
+      "loss": 7.907,
+      "step": 4200
+    },
+    {
+      "epoch": 0.4037179607548587,
+      "grad_norm": 2.755725622177124,
+      "learning_rate": 1.7743466694486468e-05,
+      "loss": 7.9049,
+      "step": 4300
+    },
+    {
+      "epoch": 0.4131067505398554,
+      "grad_norm": 2.6067683696746826,
+      "learning_rate": 1.7639142454749364e-05,
+      "loss": 7.939,
+      "step": 4400
+    },
+    {
+      "epoch": 0.42249554032485215,
+      "grad_norm": 2.614475965499878,
+      "learning_rate": 1.753481821501226e-05,
+      "loss": 7.9067,
+      "step": 4500
+    },
+    {
+      "epoch": 0.43188433010984884,
+      "grad_norm": 2.172943353652954,
+      "learning_rate": 1.7430493975275155e-05,
+      "loss": 7.9149,
+      "step": 4600
+    },
+    {
+      "epoch": 0.4412731198948456,
+      "grad_norm": 6.292716979980469,
+      "learning_rate": 1.7326169735538053e-05,
+      "loss": 7.9705,
+      "step": 4700
+    },
+    {
+      "epoch": 0.45066190967984227,
+      "grad_norm": 2.595337152481079,
+      "learning_rate": 1.7221845495800952e-05,
+      "loss": 7.8992,
+      "step": 4800
+    },
+    {
+      "epoch": 0.460050699464839,
+      "grad_norm": 5.529814720153809,
+      "learning_rate": 1.7117521256063848e-05,
+      "loss": 7.9077,
+      "step": 4900
+    },
+    {
+      "epoch": 0.4694394892498357,
+      "grad_norm": 6.696155548095703,
+      "learning_rate": 1.7013197016326747e-05,
+      "loss": 7.8992,
+      "step": 5000
+    },
+    {
+      "epoch": 0.4694394892498357,
+      "eval_loss": 7.916718006134033,
+      "eval_runtime": 334.3618,
+      "eval_samples_per_second": 1019.336,
+      "eval_steps_per_second": 7.964,
+      "step": 5000
+    },
+    {
+      "epoch": 0.4788282790348324,
+      "grad_norm": 26.500988006591797,
+      "learning_rate": 1.6908872776589642e-05,
+      "loss": 7.914,
+      "step": 5100
+    },
+    {
+      "epoch": 0.4882170688198291,
+      "grad_norm": 2.454939126968384,
+      "learning_rate": 1.6804548536852537e-05,
+      "loss": 7.8913,
+      "step": 5200
+    },
+    {
+      "epoch": 0.4976058586048258,
+      "grad_norm": 1.8935959339141846,
+      "learning_rate": 1.6700224297115436e-05,
+      "loss": 7.8999,
+      "step": 5300
+    },
+    {
+      "epoch": 0.5069946483898226,
+      "grad_norm": 2.2066667079925537,
+      "learning_rate": 1.6595900057378335e-05,
+      "loss": 7.8818,
+      "step": 5400
+    },
+    {
+      "epoch": 0.5163834381748192,
+      "grad_norm": 2.7672908306121826,
+      "learning_rate": 1.649157581764123e-05,
+      "loss": 7.9383,
+      "step": 5500
+    },
+    {
+      "epoch": 0.5257722279598159,
+      "grad_norm": 4.411477088928223,
+      "learning_rate": 1.6387251577904126e-05,
+      "loss": 7.9094,
+      "step": 5600
+    },
+    {
+      "epoch": 0.5351610177448127,
+      "grad_norm": 50.536991119384766,
+      "learning_rate": 1.6282927338167025e-05,
+      "loss": 7.8986,
+      "step": 5700
+    },
+    {
+      "epoch": 0.5445498075298094,
+      "grad_norm": 2.9657583236694336,
+      "learning_rate": 1.617860309842992e-05,
+      "loss": 7.9015,
+      "step": 5800
+    },
+    {
+      "epoch": 0.5539385973148061,
+      "grad_norm": 1.910575270652771,
+      "learning_rate": 1.607427885869282e-05,
+      "loss": 7.9059,
+      "step": 5900
+    },
+    {
+      "epoch": 0.5633273870998028,
+      "grad_norm": 3.9665298461914062,
+      "learning_rate": 1.5969954618955715e-05,
+      "loss": 7.8524,
+      "step": 6000
+    },
+    {
+      "epoch": 0.5727161768847996,
+      "grad_norm": 2.0286688804626465,
+      "learning_rate": 1.5865630379218614e-05,
+      "loss": 7.8788,
+      "step": 6100
+    },
+    {
+      "epoch": 0.5821049666697963,
+      "grad_norm": 3.200676202774048,
+      "learning_rate": 1.576130613948151e-05,
+      "loss": 7.8712,
+      "step": 6200
+    },
+    {
+      "epoch": 0.591493756454793,
+      "grad_norm": 2.0278351306915283,
+      "learning_rate": 1.5656981899744408e-05,
+      "loss": 7.8967,
+      "step": 6300
+    },
+    {
+      "epoch": 0.6008825462397896,
+      "grad_norm": 3.4500420093536377,
+      "learning_rate": 1.5553700902404677e-05,
+      "loss": 7.8677,
+      "step": 6400
+    },
+    {
+      "epoch": 0.6102713360247864,
+      "grad_norm": 2.8257575035095215,
+      "learning_rate": 1.5450419905064945e-05,
+      "loss": 7.9132,
+      "step": 6500
+    },
+    {
+      "epoch": 0.6196601258097831,
+      "grad_norm": 2.537797212600708,
+      "learning_rate": 1.534609566532784e-05,
+      "loss": 7.853,
+      "step": 6600
+    },
+    {
+      "epoch": 0.6290489155947798,
+      "grad_norm": 3.010673999786377,
+      "learning_rate": 1.5241771425590736e-05,
+      "loss": 7.8968,
+      "step": 6700
+    },
+    {
+      "epoch": 0.6384377053797765,
+      "grad_norm": 2.7398829460144043,
+      "learning_rate": 1.5137447185853635e-05,
+      "loss": 7.8656,
+      "step": 6800
+    },
+    {
+      "epoch": 0.6478264951647733,
+      "grad_norm": 4.402348518371582,
+      "learning_rate": 1.5033122946116532e-05,
+      "loss": 7.8801,
+      "step": 6900
+    },
+    {
+      "epoch": 0.65721528494977,
+      "grad_norm": 2.982344388961792,
+      "learning_rate": 1.4928798706379428e-05,
+      "loss": 7.8378,
+      "step": 7000
+    },
+    {
+      "epoch": 0.6666040747347667,
+      "grad_norm": 12.262134552001953,
+      "learning_rate": 1.4824474466642325e-05,
+      "loss": 7.8554,
+      "step": 7100
+    },
+    {
+      "epoch": 0.6759928645197634,
+      "grad_norm": 3.8007514476776123,
+      "learning_rate": 1.4720150226905224e-05,
+      "loss": 7.8305,
+      "step": 7200
+    },
+    {
+      "epoch": 0.6853816543047602,
+      "grad_norm": 3.8391048908233643,
+      "learning_rate": 1.461582598716812e-05,
+      "loss": 7.8613,
+      "step": 7300
+    },
+    {
+      "epoch": 0.6947704440897569,
+      "grad_norm": 3.107123374938965,
+      "learning_rate": 1.4511501747431017e-05,
+      "loss": 7.8554,
+      "step": 7400
+    },
+    {
+      "epoch": 0.7041592338747535,
+      "grad_norm": 2.5167956352233887,
+      "learning_rate": 1.4407177507693915e-05,
+      "loss": 7.8653,
+      "step": 7500
+    },
+    {
+      "epoch": 0.7135480236597502,
+      "grad_norm": 5.4338765144348145,
+      "learning_rate": 1.430285326795681e-05,
+      "loss": 7.8387,
+      "step": 7600
+    },
+    {
+      "epoch": 0.7229368134447469,
+      "grad_norm": 3.295238733291626,
+      "learning_rate": 1.4198529028219708e-05,
+      "loss": 7.8513,
+      "step": 7700
+    },
+    {
+      "epoch": 0.7323256032297437,
+      "grad_norm": 3.02062726020813,
+      "learning_rate": 1.4094204788482607e-05,
+      "loss": 7.8496,
+      "step": 7800
+    },
+    {
+      "epoch": 0.7417143930147404,
+      "grad_norm": 2.2482168674468994,
+      "learning_rate": 1.3989880548745502e-05,
+      "loss": 7.8276,
+      "step": 7900
+    },
+    {
+      "epoch": 0.7511031827997371,
+      "grad_norm": 3.394895315170288,
+      "learning_rate": 1.38855563090084e-05,
+      "loss": 7.8353,
+      "step": 8000
+    },
+    {
+      "epoch": 0.7604919725847338,
+      "grad_norm": 5.990811824798584,
+      "learning_rate": 1.3781232069271295e-05,
+      "loss": 7.8103,
+      "step": 8100
+    },
+    {
+      "epoch": 0.7698807623697306,
+      "grad_norm": 6.237292289733887,
+      "learning_rate": 1.3676907829534194e-05,
+      "loss": 7.8622,
+      "step": 8200
+    },
+    {
+      "epoch": 0.7792695521547273,
+      "grad_norm": 3.848353147506714,
+      "learning_rate": 1.3572583589797091e-05,
+      "loss": 7.832,
+      "step": 8300
+    },
+    {
+      "epoch": 0.788658341939724,
+      "grad_norm": 3.6994402408599854,
+      "learning_rate": 1.3468259350059986e-05,
+      "loss": 7.8349,
+      "step": 8400
+    },
+    {
+      "epoch": 0.7980471317247206,
+      "grad_norm": 2.945556163787842,
+      "learning_rate": 1.3363935110322885e-05,
+      "loss": 7.855,
+      "step": 8500
+    },
+    {
+      "epoch": 0.8074359215097174,
+      "grad_norm": 3.0218355655670166,
+      "learning_rate": 1.3259610870585782e-05,
+      "loss": 7.8316,
+      "step": 8600
+    },
+    {
+      "epoch": 0.8168247112947141,
+      "grad_norm": 3.8162949085235596,
+      "learning_rate": 1.3155286630848678e-05,
+      "loss": 7.8066,
+      "step": 8700
+    },
+    {
+      "epoch": 0.8262135010797108,
+      "grad_norm": 4.187964916229248,
+      "learning_rate": 1.3050962391111577e-05,
+      "loss": 7.8166,
+      "step": 8800
+    },
+    {
+      "epoch": 0.8356022908647075,
+      "grad_norm": 3.577139139175415,
+      "learning_rate": 1.2946638151374474e-05,
+      "loss": 7.8588,
+      "step": 8900
+    },
+    {
+      "epoch": 0.8449910806497043,
+      "grad_norm": 2.923499345779419,
+      "learning_rate": 1.284231391163737e-05,
+      "loss": 7.8042,
+      "step": 9000
+    }
+  ],
+  "logging_steps": 100,
+  "max_steps": 21302,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 128,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-9000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:006196a6760605807ae7aab04b3ff5762b98610dc83070d5ff9eb10989362210
+size 5496