Upload folder using huggingface_hub
Browse files- checkpoint-20500/1_Pooling/config.json +10 -0
- checkpoint-20500/README.md +540 -0
- checkpoint-20500/config.json +24 -0
- checkpoint-20500/config_sentence_transformers.json +10 -0
- checkpoint-20500/model.safetensors +3 -0
- checkpoint-20500/modules.json +20 -0
- checkpoint-20500/optimizer.pt +3 -0
- checkpoint-20500/rng_state.pth +3 -0
- checkpoint-20500/scheduler.pt +3 -0
- checkpoint-20500/sentence_bert_config.json +4 -0
- checkpoint-20500/special_tokens_map.json +51 -0
- checkpoint-20500/tokenizer.json +0 -0
- checkpoint-20500/tokenizer_config.json +72 -0
- checkpoint-20500/trainer_state.json +1500 -0
- checkpoint-20500/training_args.bin +3 -0
- checkpoint-20500/vocab.txt +0 -0
checkpoint-20500/1_Pooling/config.json
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"word_embedding_dimension": 768,
|
3 |
+
"pooling_mode_cls_token": false,
|
4 |
+
"pooling_mode_mean_tokens": true,
|
5 |
+
"pooling_mode_max_tokens": false,
|
6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
7 |
+
"pooling_mode_weightedmean_tokens": false,
|
8 |
+
"pooling_mode_lasttoken": false,
|
9 |
+
"include_prompt": true
|
10 |
+
}
|
checkpoint-20500/README.md
ADDED
@@ -0,0 +1,540 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: sentence-transformers/all-mpnet-base-v2
|
3 |
+
language:
|
4 |
+
- en
|
5 |
+
library_name: sentence-transformers
|
6 |
+
license: apache-2.0
|
7 |
+
pipeline_tag: sentence-similarity
|
8 |
+
tags:
|
9 |
+
- sentence-transformers
|
10 |
+
- sentence-similarity
|
11 |
+
- feature-extraction
|
12 |
+
- generated_from_trainer
|
13 |
+
- dataset_size:1363306
|
14 |
+
- loss:CoSENTLoss
|
15 |
+
widget:
|
16 |
+
- source_sentence: labneh
|
17 |
+
sentences:
|
18 |
+
- iftar
|
19 |
+
- bathing suit
|
20 |
+
- coffee cup
|
21 |
+
- source_sentence: Velvet flock Veil
|
22 |
+
sentences:
|
23 |
+
- mermaid purse
|
24 |
+
- veil
|
25 |
+
- mobile bag
|
26 |
+
- source_sentence: Red lipstick
|
27 |
+
sentences:
|
28 |
+
- chemise dress
|
29 |
+
- tote
|
30 |
+
- rouge
|
31 |
+
- source_sentence: Unisex Travel bag
|
32 |
+
sentences:
|
33 |
+
- spf
|
34 |
+
- basic vega ring
|
35 |
+
- travel backpack
|
36 |
+
- source_sentence: jeremy hush book
|
37 |
+
sentences:
|
38 |
+
- chinese jumper
|
39 |
+
- perfume
|
40 |
+
- home automation device
|
41 |
+
---
|
42 |
+
|
43 |
+
# all-mpnet-base-v2-pair_score
|
44 |
+
|
45 |
+
This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2). It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
46 |
+
|
47 |
+
## Model Details
|
48 |
+
|
49 |
+
### Model Description
|
50 |
+
- **Model Type:** Sentence Transformer
|
51 |
+
- **Base model:** [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) <!-- at revision 9a3225965996d404b775526de6dbfe85d3368642 -->
|
52 |
+
- **Maximum Sequence Length:** 384 tokens
|
53 |
+
- **Output Dimensionality:** 768 tokens
|
54 |
+
- **Similarity Function:** Cosine Similarity
|
55 |
+
<!-- - **Training Dataset:** Unknown -->
|
56 |
+
- **Language:** en
|
57 |
+
- **License:** apache-2.0
|
58 |
+
|
59 |
+
### Model Sources
|
60 |
+
|
61 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
62 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
63 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
64 |
+
|
65 |
+
### Full Model Architecture
|
66 |
+
|
67 |
+
```
|
68 |
+
SentenceTransformer(
|
69 |
+
(0): Transformer({'max_seq_length': 384, 'do_lower_case': False}) with Transformer model: MPNetModel
|
70 |
+
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
|
71 |
+
(2): Normalize()
|
72 |
+
)
|
73 |
+
```
|
74 |
+
|
75 |
+
## Usage
|
76 |
+
|
77 |
+
### Direct Usage (Sentence Transformers)
|
78 |
+
|
79 |
+
First install the Sentence Transformers library:
|
80 |
+
|
81 |
+
```bash
|
82 |
+
pip install -U sentence-transformers
|
83 |
+
```
|
84 |
+
|
85 |
+
Then you can load this model and run inference.
|
86 |
+
```python
|
87 |
+
from sentence_transformers import SentenceTransformer
|
88 |
+
|
89 |
+
# Download from the 🤗 Hub
|
90 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
91 |
+
# Run inference
|
92 |
+
sentences = [
|
93 |
+
'jeremy hush book',
|
94 |
+
'chinese jumper',
|
95 |
+
'perfume',
|
96 |
+
]
|
97 |
+
embeddings = model.encode(sentences)
|
98 |
+
print(embeddings.shape)
|
99 |
+
# [3, 768]
|
100 |
+
|
101 |
+
# Get the similarity scores for the embeddings
|
102 |
+
similarities = model.similarity(embeddings, embeddings)
|
103 |
+
print(similarities.shape)
|
104 |
+
# [3, 3]
|
105 |
+
```
|
106 |
+
|
107 |
+
<!--
|
108 |
+
### Direct Usage (Transformers)
|
109 |
+
|
110 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
111 |
+
|
112 |
+
</details>
|
113 |
+
-->
|
114 |
+
|
115 |
+
<!--
|
116 |
+
### Downstream Usage (Sentence Transformers)
|
117 |
+
|
118 |
+
You can finetune this model on your own dataset.
|
119 |
+
|
120 |
+
<details><summary>Click to expand</summary>
|
121 |
+
|
122 |
+
</details>
|
123 |
+
-->
|
124 |
+
|
125 |
+
<!--
|
126 |
+
### Out-of-Scope Use
|
127 |
+
|
128 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
129 |
+
-->
|
130 |
+
|
131 |
+
<!--
|
132 |
+
## Bias, Risks and Limitations
|
133 |
+
|
134 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
135 |
+
-->
|
136 |
+
|
137 |
+
<!--
|
138 |
+
### Recommendations
|
139 |
+
|
140 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
141 |
+
-->
|
142 |
+
|
143 |
+
## Training Details
|
144 |
+
|
145 |
+
### Training Hyperparameters
|
146 |
+
#### Non-Default Hyperparameters
|
147 |
+
|
148 |
+
- `eval_strategy`: steps
|
149 |
+
- `per_device_train_batch_size`: 128
|
150 |
+
- `per_device_eval_batch_size`: 128
|
151 |
+
- `learning_rate`: 2e-05
|
152 |
+
- `num_train_epochs`: 2
|
153 |
+
- `warmup_ratio`: 0.1
|
154 |
+
- `fp16`: True
|
155 |
+
|
156 |
+
#### All Hyperparameters
|
157 |
+
<details><summary>Click to expand</summary>
|
158 |
+
|
159 |
+
- `overwrite_output_dir`: False
|
160 |
+
- `do_predict`: False
|
161 |
+
- `eval_strategy`: steps
|
162 |
+
- `prediction_loss_only`: True
|
163 |
+
- `per_device_train_batch_size`: 128
|
164 |
+
- `per_device_eval_batch_size`: 128
|
165 |
+
- `per_gpu_train_batch_size`: None
|
166 |
+
- `per_gpu_eval_batch_size`: None
|
167 |
+
- `gradient_accumulation_steps`: 1
|
168 |
+
- `eval_accumulation_steps`: None
|
169 |
+
- `torch_empty_cache_steps`: None
|
170 |
+
- `learning_rate`: 2e-05
|
171 |
+
- `weight_decay`: 0.0
|
172 |
+
- `adam_beta1`: 0.9
|
173 |
+
- `adam_beta2`: 0.999
|
174 |
+
- `adam_epsilon`: 1e-08
|
175 |
+
- `max_grad_norm`: 1.0
|
176 |
+
- `num_train_epochs`: 2
|
177 |
+
- `max_steps`: -1
|
178 |
+
- `lr_scheduler_type`: linear
|
179 |
+
- `lr_scheduler_kwargs`: {}
|
180 |
+
- `warmup_ratio`: 0.1
|
181 |
+
- `warmup_steps`: 0
|
182 |
+
- `log_level`: passive
|
183 |
+
- `log_level_replica`: warning
|
184 |
+
- `log_on_each_node`: True
|
185 |
+
- `logging_nan_inf_filter`: True
|
186 |
+
- `save_safetensors`: True
|
187 |
+
- `save_on_each_node`: False
|
188 |
+
- `save_only_model`: False
|
189 |
+
- `restore_callback_states_from_checkpoint`: False
|
190 |
+
- `no_cuda`: False
|
191 |
+
- `use_cpu`: False
|
192 |
+
- `use_mps_device`: False
|
193 |
+
- `seed`: 42
|
194 |
+
- `data_seed`: None
|
195 |
+
- `jit_mode_eval`: False
|
196 |
+
- `use_ipex`: False
|
197 |
+
- `bf16`: False
|
198 |
+
- `fp16`: True
|
199 |
+
- `fp16_opt_level`: O1
|
200 |
+
- `half_precision_backend`: auto
|
201 |
+
- `bf16_full_eval`: False
|
202 |
+
- `fp16_full_eval`: False
|
203 |
+
- `tf32`: None
|
204 |
+
- `local_rank`: 0
|
205 |
+
- `ddp_backend`: None
|
206 |
+
- `tpu_num_cores`: None
|
207 |
+
- `tpu_metrics_debug`: False
|
208 |
+
- `debug`: []
|
209 |
+
- `dataloader_drop_last`: False
|
210 |
+
- `dataloader_num_workers`: 0
|
211 |
+
- `dataloader_prefetch_factor`: None
|
212 |
+
- `past_index`: -1
|
213 |
+
- `disable_tqdm`: False
|
214 |
+
- `remove_unused_columns`: True
|
215 |
+
- `label_names`: None
|
216 |
+
- `load_best_model_at_end`: False
|
217 |
+
- `ignore_data_skip`: False
|
218 |
+
- `fsdp`: []
|
219 |
+
- `fsdp_min_num_params`: 0
|
220 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
221 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
222 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
223 |
+
- `deepspeed`: None
|
224 |
+
- `label_smoothing_factor`: 0.0
|
225 |
+
- `optim`: adamw_torch
|
226 |
+
- `optim_args`: None
|
227 |
+
- `adafactor`: False
|
228 |
+
- `group_by_length`: False
|
229 |
+
- `length_column_name`: length
|
230 |
+
- `ddp_find_unused_parameters`: None
|
231 |
+
- `ddp_bucket_cap_mb`: None
|
232 |
+
- `ddp_broadcast_buffers`: False
|
233 |
+
- `dataloader_pin_memory`: True
|
234 |
+
- `dataloader_persistent_workers`: False
|
235 |
+
- `skip_memory_metrics`: True
|
236 |
+
- `use_legacy_prediction_loop`: False
|
237 |
+
- `push_to_hub`: False
|
238 |
+
- `resume_from_checkpoint`: None
|
239 |
+
- `hub_model_id`: None
|
240 |
+
- `hub_strategy`: every_save
|
241 |
+
- `hub_private_repo`: False
|
242 |
+
- `hub_always_push`: False
|
243 |
+
- `gradient_checkpointing`: False
|
244 |
+
- `gradient_checkpointing_kwargs`: None
|
245 |
+
- `include_inputs_for_metrics`: False
|
246 |
+
- `eval_do_concat_batches`: True
|
247 |
+
- `fp16_backend`: auto
|
248 |
+
- `push_to_hub_model_id`: None
|
249 |
+
- `push_to_hub_organization`: None
|
250 |
+
- `mp_parameters`:
|
251 |
+
- `auto_find_batch_size`: False
|
252 |
+
- `full_determinism`: False
|
253 |
+
- `torchdynamo`: None
|
254 |
+
- `ray_scope`: last
|
255 |
+
- `ddp_timeout`: 1800
|
256 |
+
- `torch_compile`: False
|
257 |
+
- `torch_compile_backend`: None
|
258 |
+
- `torch_compile_mode`: None
|
259 |
+
- `dispatch_batches`: None
|
260 |
+
- `split_batches`: None
|
261 |
+
- `include_tokens_per_second`: False
|
262 |
+
- `include_num_input_tokens_seen`: False
|
263 |
+
- `neftune_noise_alpha`: None
|
264 |
+
- `optim_target_modules`: None
|
265 |
+
- `batch_eval_metrics`: False
|
266 |
+
- `eval_on_start`: False
|
267 |
+
- `use_liger_kernel`: False
|
268 |
+
- `eval_use_gather_object`: False
|
269 |
+
- `batch_sampler`: batch_sampler
|
270 |
+
- `multi_dataset_batch_sampler`: proportional
|
271 |
+
|
272 |
+
</details>
|
273 |
+
|
274 |
+
### Training Logs
|
275 |
+
<details><summary>Click to expand</summary>
|
276 |
+
|
277 |
+
| Epoch | Step | Training Loss | loss |
|
278 |
+
|:------:|:-----:|:-------------:|:------:|
|
279 |
+
| 0.0094 | 100 | 16.0755 | - |
|
280 |
+
| 0.0188 | 200 | 13.0643 | - |
|
281 |
+
| 0.0282 | 300 | 9.3474 | - |
|
282 |
+
| 0.0376 | 400 | 8.2606 | - |
|
283 |
+
| 0.0469 | 500 | 8.084 | - |
|
284 |
+
| 0.0563 | 600 | 8.0581 | - |
|
285 |
+
| 0.0657 | 700 | 8.0175 | - |
|
286 |
+
| 0.0751 | 800 | 8.0285 | - |
|
287 |
+
| 0.0845 | 900 | 8.0024 | - |
|
288 |
+
| 0.0939 | 1000 | 8.0161 | - |
|
289 |
+
| 0.1033 | 1100 | 7.9941 | - |
|
290 |
+
| 0.1127 | 1200 | 8.0233 | - |
|
291 |
+
| 0.1221 | 1300 | 8.0141 | - |
|
292 |
+
| 0.1314 | 1400 | 7.9644 | - |
|
293 |
+
| 0.1408 | 1500 | 8.0311 | - |
|
294 |
+
| 0.1502 | 1600 | 8.0306 | - |
|
295 |
+
| 0.1596 | 1700 | 7.989 | - |
|
296 |
+
| 0.1690 | 1800 | 8.0034 | - |
|
297 |
+
| 0.1784 | 1900 | 8.0107 | - |
|
298 |
+
| 0.1878 | 2000 | 7.9737 | - |
|
299 |
+
| 0.1972 | 2100 | 7.9827 | - |
|
300 |
+
| 0.2066 | 2200 | 8.0389 | - |
|
301 |
+
| 0.2159 | 2300 | 7.973 | - |
|
302 |
+
| 0.2253 | 2400 | 7.9669 | - |
|
303 |
+
| 0.2347 | 2500 | 8.0296 | - |
|
304 |
+
| 0.2441 | 2600 | 7.9984 | - |
|
305 |
+
| 0.2535 | 2700 | 7.9772 | - |
|
306 |
+
| 0.2629 | 2800 | 7.9838 | - |
|
307 |
+
| 0.2723 | 2900 | 7.9816 | - |
|
308 |
+
| 0.2817 | 3000 | 8.0021 | - |
|
309 |
+
| 0.2911 | 3100 | 7.9715 | - |
|
310 |
+
| 0.3004 | 3200 | 7.9809 | - |
|
311 |
+
| 0.3098 | 3300 | 7.9849 | - |
|
312 |
+
| 0.3192 | 3400 | 7.9463 | - |
|
313 |
+
| 0.3286 | 3500 | 8.0067 | - |
|
314 |
+
| 0.3380 | 3600 | 7.9431 | - |
|
315 |
+
| 0.3474 | 3700 | 7.9877 | - |
|
316 |
+
| 0.3568 | 3800 | 7.9494 | - |
|
317 |
+
| 0.3662 | 3900 | 7.9466 | - |
|
318 |
+
| 0.3756 | 4000 | 7.9708 | - |
|
319 |
+
| 0.3849 | 4100 | 7.9525 | - |
|
320 |
+
| 0.3943 | 4200 | 7.9322 | - |
|
321 |
+
| 0.4037 | 4300 | 7.9415 | - |
|
322 |
+
| 0.4131 | 4400 | 7.9932 | - |
|
323 |
+
| 0.4225 | 4500 | 7.9481 | - |
|
324 |
+
| 0.4319 | 4600 | 7.976 | - |
|
325 |
+
| 0.4413 | 4700 | 7.971 | - |
|
326 |
+
| 0.4507 | 4800 | 7.9647 | - |
|
327 |
+
| 0.4601 | 4900 | 7.9217 | - |
|
328 |
+
| 0.4694 | 5000 | 7.9374 | 7.9518 |
|
329 |
+
| 0.4788 | 5100 | 7.9026 | - |
|
330 |
+
| 0.4882 | 5200 | 7.9304 | - |
|
331 |
+
| 0.4976 | 5300 | 7.9148 | - |
|
332 |
+
| 0.5070 | 5400 | 7.9538 | - |
|
333 |
+
| 0.5164 | 5500 | 8.0002 | - |
|
334 |
+
| 0.5258 | 5600 | 7.9571 | - |
|
335 |
+
| 0.5352 | 5700 | 7.932 | - |
|
336 |
+
| 0.5445 | 5800 | 7.9047 | - |
|
337 |
+
| 0.5539 | 5900 | 7.9353 | - |
|
338 |
+
| 0.5633 | 6000 | 7.9203 | - |
|
339 |
+
| 0.5727 | 6100 | 7.8967 | - |
|
340 |
+
| 0.5821 | 6200 | 7.9414 | - |
|
341 |
+
| 0.5915 | 6300 | 7.9631 | - |
|
342 |
+
| 0.6009 | 6400 | 7.9606 | - |
|
343 |
+
| 0.6103 | 6500 | 7.9377 | - |
|
344 |
+
| 0.6197 | 6600 | 7.9108 | - |
|
345 |
+
| 0.6290 | 6700 | 7.9225 | - |
|
346 |
+
| 0.6384 | 6800 | 7.9154 | - |
|
347 |
+
| 0.6478 | 6900 | 7.9191 | - |
|
348 |
+
| 0.6572 | 7000 | 7.8903 | - |
|
349 |
+
| 0.6666 | 7100 | 7.9213 | - |
|
350 |
+
| 0.6760 | 7200 | 7.9202 | - |
|
351 |
+
| 0.6854 | 7300 | 7.8998 | - |
|
352 |
+
| 0.6948 | 7400 | 7.9153 | - |
|
353 |
+
| 0.7042 | 7500 | 7.9037 | - |
|
354 |
+
| 0.7135 | 7600 | 7.9146 | - |
|
355 |
+
| 0.7229 | 7700 | 7.8972 | - |
|
356 |
+
| 0.7323 | 7800 | 7.9374 | - |
|
357 |
+
| 0.7417 | 7900 | 7.8647 | - |
|
358 |
+
| 0.7511 | 8000 | 7.8915 | - |
|
359 |
+
| 0.7605 | 8100 | 7.8846 | - |
|
360 |
+
| 0.7699 | 8200 | 7.8988 | - |
|
361 |
+
| 0.7793 | 8300 | 7.8702 | - |
|
362 |
+
| 0.7887 | 8400 | 7.923 | - |
|
363 |
+
| 0.7980 | 8500 | 7.891 | - |
|
364 |
+
| 0.8074 | 8600 | 7.8832 | - |
|
365 |
+
| 0.8168 | 8700 | 7.8726 | - |
|
366 |
+
| 0.8262 | 8800 | 7.8813 | - |
|
367 |
+
| 0.8356 | 8900 | 7.8986 | - |
|
368 |
+
| 0.8450 | 9000 | 7.8743 | - |
|
369 |
+
| 0.8544 | 9100 | 7.8791 | - |
|
370 |
+
| 0.8638 | 9200 | 7.8783 | - |
|
371 |
+
| 0.8732 | 9300 | 7.8528 | - |
|
372 |
+
| 0.8825 | 9400 | 7.8864 | - |
|
373 |
+
| 0.8919 | 9500 | 7.8989 | - |
|
374 |
+
| 0.9013 | 9600 | 7.8617 | - |
|
375 |
+
| 0.9107 | 9700 | 7.8371 | - |
|
376 |
+
| 0.9201 | 9800 | 7.8566 | - |
|
377 |
+
| 0.9295 | 9900 | 7.8776 | - |
|
378 |
+
| 0.9389 | 10000 | 7.8558 | 7.8492 |
|
379 |
+
| 0.9483 | 10100 | 7.848 | - |
|
380 |
+
| 0.9577 | 10200 | 7.8227 | - |
|
381 |
+
| 0.9670 | 10300 | 7.8311 | - |
|
382 |
+
| 0.9764 | 10400 | 7.8437 | - |
|
383 |
+
| 0.9858 | 10500 | 7.8454 | - |
|
384 |
+
| 0.9952 | 10600 | 7.8362 | - |
|
385 |
+
| 1.0046 | 10700 | 7.8681 | - |
|
386 |
+
| 1.0140 | 10800 | 7.8745 | - |
|
387 |
+
| 1.0234 | 10900 | 7.8339 | - |
|
388 |
+
| 1.0328 | 11000 | 7.8458 | - |
|
389 |
+
| 1.0422 | 11100 | 7.8493 | - |
|
390 |
+
| 1.0515 | 11200 | 7.8317 | - |
|
391 |
+
| 1.0609 | 11300 | 7.841 | - |
|
392 |
+
| 1.0703 | 11400 | 7.8292 | - |
|
393 |
+
| 1.0797 | 11500 | 7.8121 | - |
|
394 |
+
| 1.0891 | 11600 | 7.8165 | - |
|
395 |
+
| 1.0985 | 11700 | 7.8259 | - |
|
396 |
+
| 1.1079 | 11800 | 7.8303 | - |
|
397 |
+
| 1.1173 | 11900 | 7.809 | - |
|
398 |
+
| 1.1267 | 12000 | 7.818 | - |
|
399 |
+
| 1.1360 | 12100 | 7.8071 | - |
|
400 |
+
| 1.1454 | 12200 | 7.801 | - |
|
401 |
+
| 1.1548 | 12300 | 7.8123 | - |
|
402 |
+
| 1.1642 | 12400 | 7.8203 | - |
|
403 |
+
| 1.1736 | 12500 | 7.8609 | - |
|
404 |
+
| 1.1830 | 12600 | 7.7782 | - |
|
405 |
+
| 1.1924 | 12700 | 7.8092 | - |
|
406 |
+
| 1.2018 | 12800 | 7.815 | - |
|
407 |
+
| 1.2112 | 12900 | 7.8196 | - |
|
408 |
+
| 1.2205 | 13000 | 7.8206 | - |
|
409 |
+
| 1.2299 | 13100 | 7.8022 | - |
|
410 |
+
| 1.2393 | 13200 | 7.8043 | - |
|
411 |
+
| 1.2487 | 13300 | 7.7823 | - |
|
412 |
+
| 1.2581 | 13400 | 7.8061 | - |
|
413 |
+
| 1.2675 | 13500 | 7.8016 | - |
|
414 |
+
| 1.2769 | 13600 | 7.8076 | - |
|
415 |
+
| 1.2863 | 13700 | 7.7996 | - |
|
416 |
+
| 1.2957 | 13800 | 7.8035 | - |
|
417 |
+
| 1.3050 | 13900 | 7.8092 | - |
|
418 |
+
| 1.3144 | 14000 | 7.7902 | - |
|
419 |
+
| 1.3238 | 14100 | 7.8114 | - |
|
420 |
+
| 1.3332 | 14200 | 7.8112 | - |
|
421 |
+
| 1.3426 | 14300 | 7.8036 | - |
|
422 |
+
| 1.3520 | 14400 | 7.8178 | - |
|
423 |
+
| 1.3614 | 14500 | 7.8391 | - |
|
424 |
+
| 1.3708 | 14600 | 7.8151 | - |
|
425 |
+
| 1.3802 | 14700 | 7.7957 | - |
|
426 |
+
| 1.3895 | 14800 | 7.7833 | - |
|
427 |
+
| 1.3989 | 14900 | 7.8049 | - |
|
428 |
+
| 1.4083 | 15000 | 7.8163 | 7.8078 |
|
429 |
+
| 1.4177 | 15100 | 7.7864 | - |
|
430 |
+
| 1.4271 | 15200 | 7.8241 | - |
|
431 |
+
| 1.4365 | 15300 | 7.7694 | - |
|
432 |
+
| 1.4459 | 15400 | 7.7784 | - |
|
433 |
+
| 1.4553 | 15500 | 7.7628 | - |
|
434 |
+
| 1.4647 | 15600 | 7.8044 | - |
|
435 |
+
| 1.4740 | 15700 | 7.7871 | - |
|
436 |
+
| 1.4834 | 15800 | 7.809 | - |
|
437 |
+
| 1.4928 | 15900 | 7.7955 | - |
|
438 |
+
| 1.5022 | 16000 | 7.8056 | - |
|
439 |
+
| 1.5116 | 16100 | 7.774 | - |
|
440 |
+
| 1.5210 | 16200 | 7.7874 | - |
|
441 |
+
| 1.5304 | 16300 | 7.7918 | - |
|
442 |
+
| 1.5398 | 16400 | 7.7787 | - |
|
443 |
+
| 1.5492 | 16500 | 7.7881 | - |
|
444 |
+
| 1.5585 | 16600 | 7.7723 | - |
|
445 |
+
| 1.5679 | 16700 | 7.7809 | - |
|
446 |
+
| 1.5773 | 16800 | 7.8096 | - |
|
447 |
+
| 1.5867 | 16900 | 7.7559 | - |
|
448 |
+
| 1.5961 | 17000 | 7.8063 | - |
|
449 |
+
| 1.6055 | 17100 | 7.8137 | - |
|
450 |
+
| 1.6149 | 17200 | 7.761 | - |
|
451 |
+
| 1.6243 | 17300 | 7.7672 | - |
|
452 |
+
| 1.6336 | 17400 | 7.7939 | - |
|
453 |
+
| 1.6430 | 17500 | 7.8052 | - |
|
454 |
+
| 1.6524 | 17600 | 7.7519 | - |
|
455 |
+
| 1.6618 | 17700 | 7.7643 | - |
|
456 |
+
| 1.6712 | 17800 | 7.7823 | - |
|
457 |
+
| 1.6806 | 17900 | 7.7507 | - |
|
458 |
+
| 1.6900 | 18000 | 7.777 | - |
|
459 |
+
| 1.6994 | 18100 | 7.786 | - |
|
460 |
+
| 1.7088 | 18200 | 7.8097 | - |
|
461 |
+
| 1.7181 | 18300 | 7.7749 | - |
|
462 |
+
| 1.7275 | 18400 | 7.7626 | - |
|
463 |
+
| 1.7369 | 18500 | 7.7783 | - |
|
464 |
+
| 1.7463 | 18600 | 7.7552 | - |
|
465 |
+
| 1.7557 | 18700 | 7.7837 | - |
|
466 |
+
| 1.7651 | 18800 | 7.7583 | - |
|
467 |
+
| 1.7745 | 18900 | 7.7617 | - |
|
468 |
+
| 1.7839 | 19000 | 7.7649 | - |
|
469 |
+
| 1.7933 | 19100 | 7.7767 | - |
|
470 |
+
| 1.8026 | 19200 | 7.7565 | - |
|
471 |
+
| 1.8120 | 19300 | 7.7702 | - |
|
472 |
+
| 1.8214 | 19400 | 7.7552 | - |
|
473 |
+
| 1.8308 | 19500 | 7.7511 | - |
|
474 |
+
| 1.8402 | 19600 | 7.7818 | - |
|
475 |
+
| 1.8496 | 19700 | 7.7704 | - |
|
476 |
+
| 1.8590 | 19800 | 7.7824 | - |
|
477 |
+
| 1.8684 | 19900 | 7.751 | - |
|
478 |
+
| 1.8778 | 20000 | 7.7868 | 7.7942 |
|
479 |
+
| 1.8871 | 20100 | 7.7981 | - |
|
480 |
+
| 1.8965 | 20200 | 7.7673 | - |
|
481 |
+
| 1.9059 | 20300 | 7.7695 | - |
|
482 |
+
| 1.9153 | 20400 | 7.7587 | - |
|
483 |
+
| 1.9247 | 20500 | 7.7444 | - |
|
484 |
+
|
485 |
+
</details>
|
486 |
+
|
487 |
+
### Framework Versions
|
488 |
+
- Python: 3.8.10
|
489 |
+
- Sentence Transformers: 3.1.1
|
490 |
+
- Transformers: 4.45.2
|
491 |
+
- PyTorch: 2.4.1+cu118
|
492 |
+
- Accelerate: 1.0.1
|
493 |
+
- Datasets: 3.0.1
|
494 |
+
- Tokenizers: 0.20.3
|
495 |
+
|
496 |
+
## Citation
|
497 |
+
|
498 |
+
### BibTeX
|
499 |
+
|
500 |
+
#### Sentence Transformers
|
501 |
+
```bibtex
|
502 |
+
@inproceedings{reimers-2019-sentence-bert,
|
503 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
504 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
505 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
506 |
+
month = "11",
|
507 |
+
year = "2019",
|
508 |
+
publisher = "Association for Computational Linguistics",
|
509 |
+
url = "https://arxiv.org/abs/1908.10084",
|
510 |
+
}
|
511 |
+
```
|
512 |
+
|
513 |
+
#### CoSENTLoss
|
514 |
+
```bibtex
|
515 |
+
@online{kexuefm-8847,
|
516 |
+
title={CoSENT: A more efficient sentence vector scheme than Sentence-BERT},
|
517 |
+
author={Su Jianlin},
|
518 |
+
year={2022},
|
519 |
+
month={Jan},
|
520 |
+
url={https://kexue.fm/archives/8847},
|
521 |
+
}
|
522 |
+
```
|
523 |
+
|
524 |
+
<!--
|
525 |
+
## Glossary
|
526 |
+
|
527 |
+
*Clearly define terms in order to be accessible across audiences.*
|
528 |
+
-->
|
529 |
+
|
530 |
+
<!--
|
531 |
+
## Model Card Authors
|
532 |
+
|
533 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
534 |
+
-->
|
535 |
+
|
536 |
+
<!--
|
537 |
+
## Model Card Contact
|
538 |
+
|
539 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
540 |
+
-->
|
checkpoint-20500/config.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "sentence-transformers/all-mpnet-base-v2",
|
3 |
+
"architectures": [
|
4 |
+
"MPNetModel"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.1,
|
7 |
+
"bos_token_id": 0,
|
8 |
+
"eos_token_id": 2,
|
9 |
+
"hidden_act": "gelu",
|
10 |
+
"hidden_dropout_prob": 0.1,
|
11 |
+
"hidden_size": 768,
|
12 |
+
"initializer_range": 0.02,
|
13 |
+
"intermediate_size": 3072,
|
14 |
+
"layer_norm_eps": 1e-05,
|
15 |
+
"max_position_embeddings": 514,
|
16 |
+
"model_type": "mpnet",
|
17 |
+
"num_attention_heads": 12,
|
18 |
+
"num_hidden_layers": 12,
|
19 |
+
"pad_token_id": 1,
|
20 |
+
"relative_attention_num_buckets": 32,
|
21 |
+
"torch_dtype": "float32",
|
22 |
+
"transformers_version": "4.45.2",
|
23 |
+
"vocab_size": 30527
|
24 |
+
}
|
checkpoint-20500/config_sentence_transformers.json
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"__version__": {
|
3 |
+
"sentence_transformers": "3.1.1",
|
4 |
+
"transformers": "4.45.2",
|
5 |
+
"pytorch": "2.4.1+cu118"
|
6 |
+
},
|
7 |
+
"prompts": {},
|
8 |
+
"default_prompt_name": null,
|
9 |
+
"similarity_fn_name": null
|
10 |
+
}
|
checkpoint-20500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9f586cd9fe24b449fc6e0a37e01ec338fc20fe1870cdb3aadd3d4cb052a9f7ab
|
3 |
+
size 437967672
|
checkpoint-20500/modules.json
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[
|
2 |
+
{
|
3 |
+
"idx": 0,
|
4 |
+
"name": "0",
|
5 |
+
"path": "",
|
6 |
+
"type": "sentence_transformers.models.Transformer"
|
7 |
+
},
|
8 |
+
{
|
9 |
+
"idx": 1,
|
10 |
+
"name": "1",
|
11 |
+
"path": "1_Pooling",
|
12 |
+
"type": "sentence_transformers.models.Pooling"
|
13 |
+
},
|
14 |
+
{
|
15 |
+
"idx": 2,
|
16 |
+
"name": "2",
|
17 |
+
"path": "2_Normalize",
|
18 |
+
"type": "sentence_transformers.models.Normalize"
|
19 |
+
}
|
20 |
+
]
|
checkpoint-20500/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:04dc687aae5d16cff3539d7bfd96d9587980395df419a39fec5856e76e877e21
|
3 |
+
size 871331770
|
checkpoint-20500/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a2b651375116ed5fd7c7d85be3cb1ab5f8aeff75b36a07a9bf069d5aca63c564
|
3 |
+
size 14244
|
checkpoint-20500/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b50bd604e3bbb5dd315f90fbd8f494e24597cb4aef78abdb38d02e24e6a9f7ef
|
3 |
+
size 1064
|
checkpoint-20500/sentence_bert_config.json
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"max_seq_length": 384,
|
3 |
+
"do_lower_case": false
|
4 |
+
}
|
checkpoint-20500/special_tokens_map.json
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<s>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"cls_token": {
|
10 |
+
"content": "<s>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"eos_token": {
|
17 |
+
"content": "</s>",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"mask_token": {
|
24 |
+
"content": "<mask>",
|
25 |
+
"lstrip": true,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
},
|
30 |
+
"pad_token": {
|
31 |
+
"content": "<pad>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false
|
36 |
+
},
|
37 |
+
"sep_token": {
|
38 |
+
"content": "</s>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false
|
43 |
+
},
|
44 |
+
"unk_token": {
|
45 |
+
"content": "[UNK]",
|
46 |
+
"lstrip": false,
|
47 |
+
"normalized": false,
|
48 |
+
"rstrip": false,
|
49 |
+
"single_word": false
|
50 |
+
}
|
51 |
+
}
|
checkpoint-20500/tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
checkpoint-20500/tokenizer_config.json
ADDED
@@ -0,0 +1,72 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"added_tokens_decoder": {
|
3 |
+
"0": {
|
4 |
+
"content": "<s>",
|
5 |
+
"lstrip": false,
|
6 |
+
"normalized": false,
|
7 |
+
"rstrip": false,
|
8 |
+
"single_word": false,
|
9 |
+
"special": true
|
10 |
+
},
|
11 |
+
"1": {
|
12 |
+
"content": "<pad>",
|
13 |
+
"lstrip": false,
|
14 |
+
"normalized": false,
|
15 |
+
"rstrip": false,
|
16 |
+
"single_word": false,
|
17 |
+
"special": true
|
18 |
+
},
|
19 |
+
"2": {
|
20 |
+
"content": "</s>",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false,
|
25 |
+
"special": true
|
26 |
+
},
|
27 |
+
"3": {
|
28 |
+
"content": "<unk>",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": true,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false,
|
33 |
+
"special": true
|
34 |
+
},
|
35 |
+
"104": {
|
36 |
+
"content": "[UNK]",
|
37 |
+
"lstrip": false,
|
38 |
+
"normalized": false,
|
39 |
+
"rstrip": false,
|
40 |
+
"single_word": false,
|
41 |
+
"special": true
|
42 |
+
},
|
43 |
+
"30526": {
|
44 |
+
"content": "<mask>",
|
45 |
+
"lstrip": true,
|
46 |
+
"normalized": false,
|
47 |
+
"rstrip": false,
|
48 |
+
"single_word": false,
|
49 |
+
"special": true
|
50 |
+
}
|
51 |
+
},
|
52 |
+
"bos_token": "<s>",
|
53 |
+
"clean_up_tokenization_spaces": false,
|
54 |
+
"cls_token": "<s>",
|
55 |
+
"do_lower_case": true,
|
56 |
+
"eos_token": "</s>",
|
57 |
+
"mask_token": "<mask>",
|
58 |
+
"max_length": 128,
|
59 |
+
"model_max_length": 384,
|
60 |
+
"pad_to_multiple_of": null,
|
61 |
+
"pad_token": "<pad>",
|
62 |
+
"pad_token_type_id": 0,
|
63 |
+
"padding_side": "right",
|
64 |
+
"sep_token": "</s>",
|
65 |
+
"stride": 0,
|
66 |
+
"strip_accents": null,
|
67 |
+
"tokenize_chinese_chars": true,
|
68 |
+
"tokenizer_class": "MPNetTokenizer",
|
69 |
+
"truncation_side": "right",
|
70 |
+
"truncation_strategy": "longest_first",
|
71 |
+
"unk_token": "[UNK]"
|
72 |
+
}
|
checkpoint-20500/trainer_state.json
ADDED
@@ -0,0 +1,1500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.9247019059243264,
|
5 |
+
"eval_steps": 5000,
|
6 |
+
"global_step": 20500,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.009388789784996713,
|
13 |
+
"grad_norm": 72.80598449707031,
|
14 |
+
"learning_rate": 9.103707179727828e-07,
|
15 |
+
"loss": 16.0755,
|
16 |
+
"step": 100
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.018777579569993427,
|
20 |
+
"grad_norm": 72.60921478271484,
|
21 |
+
"learning_rate": 1.8301267010793056e-06,
|
22 |
+
"loss": 13.0643,
|
23 |
+
"step": 200
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.02816636935499014,
|
27 |
+
"grad_norm": 10.017908096313477,
|
28 |
+
"learning_rate": 2.7592679493195683e-06,
|
29 |
+
"loss": 9.3474,
|
30 |
+
"step": 300
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.03755515913998685,
|
34 |
+
"grad_norm": 5.694988250732422,
|
35 |
+
"learning_rate": 3.6977944626935713e-06,
|
36 |
+
"loss": 8.2606,
|
37 |
+
"step": 400
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.04694394892498357,
|
41 |
+
"grad_norm": 4.844100475311279,
|
42 |
+
"learning_rate": 4.6363209760675744e-06,
|
43 |
+
"loss": 8.084,
|
44 |
+
"step": 500
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.05633273870998028,
|
48 |
+
"grad_norm": 3.6125142574310303,
|
49 |
+
"learning_rate": 5.574847489441577e-06,
|
50 |
+
"loss": 8.0581,
|
51 |
+
"step": 600
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.06572152849497699,
|
55 |
+
"grad_norm": 3.166046380996704,
|
56 |
+
"learning_rate": 6.51337400281558e-06,
|
57 |
+
"loss": 8.0175,
|
58 |
+
"step": 700
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.0751103182799737,
|
62 |
+
"grad_norm": 2.600433111190796,
|
63 |
+
"learning_rate": 7.451900516189583e-06,
|
64 |
+
"loss": 8.0285,
|
65 |
+
"step": 800
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.08449910806497042,
|
69 |
+
"grad_norm": 2.3805315494537354,
|
70 |
+
"learning_rate": 8.390427029563585e-06,
|
71 |
+
"loss": 8.0024,
|
72 |
+
"step": 900
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.09388789784996714,
|
76 |
+
"grad_norm": 12.382240295410156,
|
77 |
+
"learning_rate": 9.328953542937589e-06,
|
78 |
+
"loss": 8.0161,
|
79 |
+
"step": 1000
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.10327668763496385,
|
83 |
+
"grad_norm": 2.7355728149414062,
|
84 |
+
"learning_rate": 1.0267480056311592e-05,
|
85 |
+
"loss": 7.9941,
|
86 |
+
"step": 1100
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.11266547741996057,
|
90 |
+
"grad_norm": 2.0243470668792725,
|
91 |
+
"learning_rate": 1.1206006569685594e-05,
|
92 |
+
"loss": 8.0233,
|
93 |
+
"step": 1200
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.12205426720495728,
|
97 |
+
"grad_norm": 1.9162158966064453,
|
98 |
+
"learning_rate": 1.2144533083059597e-05,
|
99 |
+
"loss": 8.0141,
|
100 |
+
"step": 1300
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.13144305698995398,
|
104 |
+
"grad_norm": 11.409939765930176,
|
105 |
+
"learning_rate": 1.3083059596433601e-05,
|
106 |
+
"loss": 7.9644,
|
107 |
+
"step": 1400
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.1408318467749507,
|
111 |
+
"grad_norm": 1.712424635887146,
|
112 |
+
"learning_rate": 1.4021586109807603e-05,
|
113 |
+
"loss": 8.0311,
|
114 |
+
"step": 1500
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.1502206365599474,
|
118 |
+
"grad_norm": 2.4589834213256836,
|
119 |
+
"learning_rate": 1.4960112623181606e-05,
|
120 |
+
"loss": 8.0306,
|
121 |
+
"step": 1600
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.15960942634494413,
|
125 |
+
"grad_norm": 1.7343533039093018,
|
126 |
+
"learning_rate": 1.589863913655561e-05,
|
127 |
+
"loss": 7.989,
|
128 |
+
"step": 1700
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.16899821612994084,
|
132 |
+
"grad_norm": 2.0726826190948486,
|
133 |
+
"learning_rate": 1.6837165649929613e-05,
|
134 |
+
"loss": 8.0034,
|
135 |
+
"step": 1800
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.17838700591493756,
|
139 |
+
"grad_norm": 1.7758458852767944,
|
140 |
+
"learning_rate": 1.7775692163303613e-05,
|
141 |
+
"loss": 8.0107,
|
142 |
+
"step": 1900
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.18777579569993427,
|
146 |
+
"grad_norm": 3.3816475868225098,
|
147 |
+
"learning_rate": 1.8714218676677617e-05,
|
148 |
+
"loss": 7.9737,
|
149 |
+
"step": 2000
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.197164585484931,
|
153 |
+
"grad_norm": 1.8136950731277466,
|
154 |
+
"learning_rate": 1.965274519005162e-05,
|
155 |
+
"loss": 7.9827,
|
156 |
+
"step": 2100
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.2065533752699277,
|
160 |
+
"grad_norm": 1.7819303274154663,
|
161 |
+
"learning_rate": 1.9934275728965626e-05,
|
162 |
+
"loss": 8.0389,
|
163 |
+
"step": 2200
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.21594216505492442,
|
167 |
+
"grad_norm": 2.269160509109497,
|
168 |
+
"learning_rate": 1.9829951489228525e-05,
|
169 |
+
"loss": 7.973,
|
170 |
+
"step": 2300
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.22533095483992113,
|
174 |
+
"grad_norm": 3.3508036136627197,
|
175 |
+
"learning_rate": 1.972562724949142e-05,
|
176 |
+
"loss": 7.9669,
|
177 |
+
"step": 2400
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.23471974462491785,
|
181 |
+
"grad_norm": 1.674142599105835,
|
182 |
+
"learning_rate": 1.962130300975432e-05,
|
183 |
+
"loss": 8.0296,
|
184 |
+
"step": 2500
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.24410853440991456,
|
188 |
+
"grad_norm": 1.454300880432129,
|
189 |
+
"learning_rate": 1.9516978770017215e-05,
|
190 |
+
"loss": 7.9984,
|
191 |
+
"step": 2600
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.2534973241949113,
|
195 |
+
"grad_norm": 2.2951695919036865,
|
196 |
+
"learning_rate": 1.9412654530280113e-05,
|
197 |
+
"loss": 7.9772,
|
198 |
+
"step": 2700
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.26288611397990796,
|
202 |
+
"grad_norm": 6.295051574707031,
|
203 |
+
"learning_rate": 1.930833029054301e-05,
|
204 |
+
"loss": 7.9838,
|
205 |
+
"step": 2800
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.2722749037649047,
|
209 |
+
"grad_norm": 1.8874555826187134,
|
210 |
+
"learning_rate": 1.9204006050805904e-05,
|
211 |
+
"loss": 7.9816,
|
212 |
+
"step": 2900
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.2816636935499014,
|
216 |
+
"grad_norm": 20.835277557373047,
|
217 |
+
"learning_rate": 1.9099681811068803e-05,
|
218 |
+
"loss": 8.0021,
|
219 |
+
"step": 3000
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.29105248333489814,
|
223 |
+
"grad_norm": 2.1683876514434814,
|
224 |
+
"learning_rate": 1.8995357571331702e-05,
|
225 |
+
"loss": 7.9715,
|
226 |
+
"step": 3100
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.3004412731198948,
|
230 |
+
"grad_norm": 1.6533387899398804,
|
231 |
+
"learning_rate": 1.8891033331594598e-05,
|
232 |
+
"loss": 7.9809,
|
233 |
+
"step": 3200
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.30983006290489157,
|
237 |
+
"grad_norm": 4.595189094543457,
|
238 |
+
"learning_rate": 1.8786709091857496e-05,
|
239 |
+
"loss": 7.9849,
|
240 |
+
"step": 3300
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.31921885268988826,
|
244 |
+
"grad_norm": 1.994147539138794,
|
245 |
+
"learning_rate": 1.8682384852120392e-05,
|
246 |
+
"loss": 7.9463,
|
247 |
+
"step": 3400
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.328607642474885,
|
251 |
+
"grad_norm": 1.961474895477295,
|
252 |
+
"learning_rate": 1.8578060612383287e-05,
|
253 |
+
"loss": 8.0067,
|
254 |
+
"step": 3500
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.3379964322598817,
|
258 |
+
"grad_norm": 24.005535125732422,
|
259 |
+
"learning_rate": 1.8473736372646186e-05,
|
260 |
+
"loss": 7.9431,
|
261 |
+
"step": 3600
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.34738522204487843,
|
265 |
+
"grad_norm": 1.9433845281600952,
|
266 |
+
"learning_rate": 1.8369412132909085e-05,
|
267 |
+
"loss": 7.9877,
|
268 |
+
"step": 3700
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.3567740118298751,
|
272 |
+
"grad_norm": 10.296500205993652,
|
273 |
+
"learning_rate": 1.826508789317198e-05,
|
274 |
+
"loss": 7.9494,
|
275 |
+
"step": 3800
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.36616280161487186,
|
279 |
+
"grad_norm": 2.194976568222046,
|
280 |
+
"learning_rate": 1.8160763653434876e-05,
|
281 |
+
"loss": 7.9466,
|
282 |
+
"step": 3900
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.37555159139986855,
|
286 |
+
"grad_norm": 1.5201098918914795,
|
287 |
+
"learning_rate": 1.8056439413697775e-05,
|
288 |
+
"loss": 7.9708,
|
289 |
+
"step": 4000
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.3849403811848653,
|
293 |
+
"grad_norm": 2.9077212810516357,
|
294 |
+
"learning_rate": 1.795211517396067e-05,
|
295 |
+
"loss": 7.9525,
|
296 |
+
"step": 4100
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.394329170969862,
|
300 |
+
"grad_norm": 2.041530132293701,
|
301 |
+
"learning_rate": 1.784779093422357e-05,
|
302 |
+
"loss": 7.9322,
|
303 |
+
"step": 4200
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.4037179607548587,
|
307 |
+
"grad_norm": 2.0275838375091553,
|
308 |
+
"learning_rate": 1.7743466694486468e-05,
|
309 |
+
"loss": 7.9415,
|
310 |
+
"step": 4300
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.4131067505398554,
|
314 |
+
"grad_norm": 2.000778913497925,
|
315 |
+
"learning_rate": 1.7639142454749364e-05,
|
316 |
+
"loss": 7.9932,
|
317 |
+
"step": 4400
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.42249554032485215,
|
321 |
+
"grad_norm": 2.2984609603881836,
|
322 |
+
"learning_rate": 1.753481821501226e-05,
|
323 |
+
"loss": 7.9481,
|
324 |
+
"step": 4500
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.43188433010984884,
|
328 |
+
"grad_norm": 1.0885875225067139,
|
329 |
+
"learning_rate": 1.7430493975275155e-05,
|
330 |
+
"loss": 7.976,
|
331 |
+
"step": 4600
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.4412731198948456,
|
335 |
+
"grad_norm": 4.441020488739014,
|
336 |
+
"learning_rate": 1.7326169735538053e-05,
|
337 |
+
"loss": 7.971,
|
338 |
+
"step": 4700
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.45066190967984227,
|
342 |
+
"grad_norm": 2.2056221961975098,
|
343 |
+
"learning_rate": 1.7221845495800952e-05,
|
344 |
+
"loss": 7.9647,
|
345 |
+
"step": 4800
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.460050699464839,
|
349 |
+
"grad_norm": 2.1192028522491455,
|
350 |
+
"learning_rate": 1.7117521256063848e-05,
|
351 |
+
"loss": 7.9217,
|
352 |
+
"step": 4900
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 0.4694394892498357,
|
356 |
+
"grad_norm": 14.011516571044922,
|
357 |
+
"learning_rate": 1.7013197016326747e-05,
|
358 |
+
"loss": 7.9374,
|
359 |
+
"step": 5000
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 0.4694394892498357,
|
363 |
+
"eval_loss": 7.951793193817139,
|
364 |
+
"eval_runtime": 900.8057,
|
365 |
+
"eval_samples_per_second": 378.358,
|
366 |
+
"eval_steps_per_second": 2.956,
|
367 |
+
"step": 5000
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"epoch": 0.4788282790348324,
|
371 |
+
"grad_norm": 39.19038009643555,
|
372 |
+
"learning_rate": 1.6908872776589642e-05,
|
373 |
+
"loss": 7.9026,
|
374 |
+
"step": 5100
|
375 |
+
},
|
376 |
+
{
|
377 |
+
"epoch": 0.4882170688198291,
|
378 |
+
"grad_norm": 1.812458872795105,
|
379 |
+
"learning_rate": 1.6804548536852537e-05,
|
380 |
+
"loss": 7.9304,
|
381 |
+
"step": 5200
|
382 |
+
},
|
383 |
+
{
|
384 |
+
"epoch": 0.4976058586048258,
|
385 |
+
"grad_norm": 70.96247863769531,
|
386 |
+
"learning_rate": 1.6700224297115436e-05,
|
387 |
+
"loss": 7.9148,
|
388 |
+
"step": 5300
|
389 |
+
},
|
390 |
+
{
|
391 |
+
"epoch": 0.5069946483898226,
|
392 |
+
"grad_norm": 1.6605011224746704,
|
393 |
+
"learning_rate": 1.6595900057378335e-05,
|
394 |
+
"loss": 7.9538,
|
395 |
+
"step": 5400
|
396 |
+
},
|
397 |
+
{
|
398 |
+
"epoch": 0.5163834381748192,
|
399 |
+
"grad_norm": 2.0463483333587646,
|
400 |
+
"learning_rate": 1.649157581764123e-05,
|
401 |
+
"loss": 8.0002,
|
402 |
+
"step": 5500
|
403 |
+
},
|
404 |
+
{
|
405 |
+
"epoch": 0.5257722279598159,
|
406 |
+
"grad_norm": 3.8601722717285156,
|
407 |
+
"learning_rate": 1.6387251577904126e-05,
|
408 |
+
"loss": 7.9571,
|
409 |
+
"step": 5600
|
410 |
+
},
|
411 |
+
{
|
412 |
+
"epoch": 0.5351610177448127,
|
413 |
+
"grad_norm": 2.184122323989868,
|
414 |
+
"learning_rate": 1.6282927338167025e-05,
|
415 |
+
"loss": 7.932,
|
416 |
+
"step": 5700
|
417 |
+
},
|
418 |
+
{
|
419 |
+
"epoch": 0.5445498075298094,
|
420 |
+
"grad_norm": 2.165367603302002,
|
421 |
+
"learning_rate": 1.617860309842992e-05,
|
422 |
+
"loss": 7.9047,
|
423 |
+
"step": 5800
|
424 |
+
},
|
425 |
+
{
|
426 |
+
"epoch": 0.5539385973148061,
|
427 |
+
"grad_norm": 1.5312166213989258,
|
428 |
+
"learning_rate": 1.607427885869282e-05,
|
429 |
+
"loss": 7.9353,
|
430 |
+
"step": 5900
|
431 |
+
},
|
432 |
+
{
|
433 |
+
"epoch": 0.5633273870998028,
|
434 |
+
"grad_norm": 21.75490379333496,
|
435 |
+
"learning_rate": 1.5969954618955715e-05,
|
436 |
+
"loss": 7.9203,
|
437 |
+
"step": 6000
|
438 |
+
},
|
439 |
+
{
|
440 |
+
"epoch": 0.5727161768847996,
|
441 |
+
"grad_norm": 1.8674250841140747,
|
442 |
+
"learning_rate": 1.5865630379218614e-05,
|
443 |
+
"loss": 7.8967,
|
444 |
+
"step": 6100
|
445 |
+
},
|
446 |
+
{
|
447 |
+
"epoch": 0.5821049666697963,
|
448 |
+
"grad_norm": 49.87809371948242,
|
449 |
+
"learning_rate": 1.576130613948151e-05,
|
450 |
+
"loss": 7.9414,
|
451 |
+
"step": 6200
|
452 |
+
},
|
453 |
+
{
|
454 |
+
"epoch": 0.591493756454793,
|
455 |
+
"grad_norm": 54.42366409301758,
|
456 |
+
"learning_rate": 1.5658025142141778e-05,
|
457 |
+
"loss": 7.9631,
|
458 |
+
"step": 6300
|
459 |
+
},
|
460 |
+
{
|
461 |
+
"epoch": 0.6008825462397896,
|
462 |
+
"grad_norm": 37.58320236206055,
|
463 |
+
"learning_rate": 1.5554744144802047e-05,
|
464 |
+
"loss": 7.9606,
|
465 |
+
"step": 6400
|
466 |
+
},
|
467 |
+
{
|
468 |
+
"epoch": 0.6102713360247864,
|
469 |
+
"grad_norm": 3.1502482891082764,
|
470 |
+
"learning_rate": 1.5450419905064945e-05,
|
471 |
+
"loss": 7.9377,
|
472 |
+
"step": 6500
|
473 |
+
},
|
474 |
+
{
|
475 |
+
"epoch": 0.6196601258097831,
|
476 |
+
"grad_norm": 2.5369224548339844,
|
477 |
+
"learning_rate": 1.534609566532784e-05,
|
478 |
+
"loss": 7.9108,
|
479 |
+
"step": 6600
|
480 |
+
},
|
481 |
+
{
|
482 |
+
"epoch": 0.6290489155947798,
|
483 |
+
"grad_norm": 2.5891005992889404,
|
484 |
+
"learning_rate": 1.5241771425590736e-05,
|
485 |
+
"loss": 7.9225,
|
486 |
+
"step": 6700
|
487 |
+
},
|
488 |
+
{
|
489 |
+
"epoch": 0.6384377053797765,
|
490 |
+
"grad_norm": 1.7794080972671509,
|
491 |
+
"learning_rate": 1.5137447185853635e-05,
|
492 |
+
"loss": 7.9154,
|
493 |
+
"step": 6800
|
494 |
+
},
|
495 |
+
{
|
496 |
+
"epoch": 0.6478264951647733,
|
497 |
+
"grad_norm": 2.324805974960327,
|
498 |
+
"learning_rate": 1.5033122946116532e-05,
|
499 |
+
"loss": 7.9191,
|
500 |
+
"step": 6900
|
501 |
+
},
|
502 |
+
{
|
503 |
+
"epoch": 0.65721528494977,
|
504 |
+
"grad_norm": 2.601715564727783,
|
505 |
+
"learning_rate": 1.4928798706379428e-05,
|
506 |
+
"loss": 7.8903,
|
507 |
+
"step": 7000
|
508 |
+
},
|
509 |
+
{
|
510 |
+
"epoch": 0.6666040747347667,
|
511 |
+
"grad_norm": 2.4438092708587646,
|
512 |
+
"learning_rate": 1.4824474466642325e-05,
|
513 |
+
"loss": 7.9213,
|
514 |
+
"step": 7100
|
515 |
+
},
|
516 |
+
{
|
517 |
+
"epoch": 0.6759928645197634,
|
518 |
+
"grad_norm": 8.118125915527344,
|
519 |
+
"learning_rate": 1.4720150226905224e-05,
|
520 |
+
"loss": 7.9202,
|
521 |
+
"step": 7200
|
522 |
+
},
|
523 |
+
{
|
524 |
+
"epoch": 0.6853816543047602,
|
525 |
+
"grad_norm": 3.908555746078491,
|
526 |
+
"learning_rate": 1.461582598716812e-05,
|
527 |
+
"loss": 7.8998,
|
528 |
+
"step": 7300
|
529 |
+
},
|
530 |
+
{
|
531 |
+
"epoch": 0.6947704440897569,
|
532 |
+
"grad_norm": 2.72293758392334,
|
533 |
+
"learning_rate": 1.4511501747431017e-05,
|
534 |
+
"loss": 7.9153,
|
535 |
+
"step": 7400
|
536 |
+
},
|
537 |
+
{
|
538 |
+
"epoch": 0.7041592338747535,
|
539 |
+
"grad_norm": 3.108797073364258,
|
540 |
+
"learning_rate": 1.4407177507693915e-05,
|
541 |
+
"loss": 7.9037,
|
542 |
+
"step": 7500
|
543 |
+
},
|
544 |
+
{
|
545 |
+
"epoch": 0.7135480236597502,
|
546 |
+
"grad_norm": 2.6256439685821533,
|
547 |
+
"learning_rate": 1.430285326795681e-05,
|
548 |
+
"loss": 7.9146,
|
549 |
+
"step": 7600
|
550 |
+
},
|
551 |
+
{
|
552 |
+
"epoch": 0.7229368134447469,
|
553 |
+
"grad_norm": 3.5525624752044678,
|
554 |
+
"learning_rate": 1.4198529028219708e-05,
|
555 |
+
"loss": 7.8972,
|
556 |
+
"step": 7700
|
557 |
+
},
|
558 |
+
{
|
559 |
+
"epoch": 0.7323256032297437,
|
560 |
+
"grad_norm": 2.6983673572540283,
|
561 |
+
"learning_rate": 1.4094204788482607e-05,
|
562 |
+
"loss": 7.9374,
|
563 |
+
"step": 7800
|
564 |
+
},
|
565 |
+
{
|
566 |
+
"epoch": 0.7417143930147404,
|
567 |
+
"grad_norm": 1.545486569404602,
|
568 |
+
"learning_rate": 1.3989880548745502e-05,
|
569 |
+
"loss": 7.8647,
|
570 |
+
"step": 7900
|
571 |
+
},
|
572 |
+
{
|
573 |
+
"epoch": 0.7511031827997371,
|
574 |
+
"grad_norm": 2.5116941928863525,
|
575 |
+
"learning_rate": 1.38855563090084e-05,
|
576 |
+
"loss": 7.8915,
|
577 |
+
"step": 8000
|
578 |
+
},
|
579 |
+
{
|
580 |
+
"epoch": 0.7604919725847338,
|
581 |
+
"grad_norm": 1.8576518297195435,
|
582 |
+
"learning_rate": 1.3781232069271295e-05,
|
583 |
+
"loss": 7.8846,
|
584 |
+
"step": 8100
|
585 |
+
},
|
586 |
+
{
|
587 |
+
"epoch": 0.7698807623697306,
|
588 |
+
"grad_norm": 3.3226571083068848,
|
589 |
+
"learning_rate": 1.3676907829534194e-05,
|
590 |
+
"loss": 7.8988,
|
591 |
+
"step": 8200
|
592 |
+
},
|
593 |
+
{
|
594 |
+
"epoch": 0.7792695521547273,
|
595 |
+
"grad_norm": 2.946324586868286,
|
596 |
+
"learning_rate": 1.3572583589797091e-05,
|
597 |
+
"loss": 7.8702,
|
598 |
+
"step": 8300
|
599 |
+
},
|
600 |
+
{
|
601 |
+
"epoch": 0.788658341939724,
|
602 |
+
"grad_norm": 2.5089969635009766,
|
603 |
+
"learning_rate": 1.3468259350059986e-05,
|
604 |
+
"loss": 7.923,
|
605 |
+
"step": 8400
|
606 |
+
},
|
607 |
+
{
|
608 |
+
"epoch": 0.7980471317247206,
|
609 |
+
"grad_norm": 2.2807912826538086,
|
610 |
+
"learning_rate": 1.3363935110322885e-05,
|
611 |
+
"loss": 7.891,
|
612 |
+
"step": 8500
|
613 |
+
},
|
614 |
+
{
|
615 |
+
"epoch": 0.8074359215097174,
|
616 |
+
"grad_norm": 2.5889735221862793,
|
617 |
+
"learning_rate": 1.3259610870585782e-05,
|
618 |
+
"loss": 7.8832,
|
619 |
+
"step": 8600
|
620 |
+
},
|
621 |
+
{
|
622 |
+
"epoch": 0.8168247112947141,
|
623 |
+
"grad_norm": 2.8306784629821777,
|
624 |
+
"learning_rate": 1.3155286630848678e-05,
|
625 |
+
"loss": 7.8726,
|
626 |
+
"step": 8700
|
627 |
+
},
|
628 |
+
{
|
629 |
+
"epoch": 0.8262135010797108,
|
630 |
+
"grad_norm": 2.626786231994629,
|
631 |
+
"learning_rate": 1.3050962391111577e-05,
|
632 |
+
"loss": 7.8813,
|
633 |
+
"step": 8800
|
634 |
+
},
|
635 |
+
{
|
636 |
+
"epoch": 0.8356022908647075,
|
637 |
+
"grad_norm": 3.195319414138794,
|
638 |
+
"learning_rate": 1.2946638151374474e-05,
|
639 |
+
"loss": 7.8986,
|
640 |
+
"step": 8900
|
641 |
+
},
|
642 |
+
{
|
643 |
+
"epoch": 0.8449910806497043,
|
644 |
+
"grad_norm": 5.254043102264404,
|
645 |
+
"learning_rate": 1.284231391163737e-05,
|
646 |
+
"loss": 7.8743,
|
647 |
+
"step": 9000
|
648 |
+
},
|
649 |
+
{
|
650 |
+
"epoch": 0.854379870434701,
|
651 |
+
"grad_norm": 2.9493279457092285,
|
652 |
+
"learning_rate": 1.2737989671900267e-05,
|
653 |
+
"loss": 7.8791,
|
654 |
+
"step": 9100
|
655 |
+
},
|
656 |
+
{
|
657 |
+
"epoch": 0.8637686602196977,
|
658 |
+
"grad_norm": 3.130415439605713,
|
659 |
+
"learning_rate": 1.2633665432163165e-05,
|
660 |
+
"loss": 7.8783,
|
661 |
+
"step": 9200
|
662 |
+
},
|
663 |
+
{
|
664 |
+
"epoch": 0.8731574500046944,
|
665 |
+
"grad_norm": 4.030152797698975,
|
666 |
+
"learning_rate": 1.2529341192426061e-05,
|
667 |
+
"loss": 7.8528,
|
668 |
+
"step": 9300
|
669 |
+
},
|
670 |
+
{
|
671 |
+
"epoch": 0.8825462397896912,
|
672 |
+
"grad_norm": 2.9882099628448486,
|
673 |
+
"learning_rate": 1.2425016952688958e-05,
|
674 |
+
"loss": 7.8864,
|
675 |
+
"step": 9400
|
676 |
+
},
|
677 |
+
{
|
678 |
+
"epoch": 0.8919350295746878,
|
679 |
+
"grad_norm": 3.802172899246216,
|
680 |
+
"learning_rate": 1.2320692712951855e-05,
|
681 |
+
"loss": 7.8989,
|
682 |
+
"step": 9500
|
683 |
+
},
|
684 |
+
{
|
685 |
+
"epoch": 0.9013238193596845,
|
686 |
+
"grad_norm": 2.724433183670044,
|
687 |
+
"learning_rate": 1.2216368473214752e-05,
|
688 |
+
"loss": 7.8617,
|
689 |
+
"step": 9600
|
690 |
+
},
|
691 |
+
{
|
692 |
+
"epoch": 0.9107126091446812,
|
693 |
+
"grad_norm": 2.459376573562622,
|
694 |
+
"learning_rate": 1.211204423347765e-05,
|
695 |
+
"loss": 7.8371,
|
696 |
+
"step": 9700
|
697 |
+
},
|
698 |
+
{
|
699 |
+
"epoch": 0.920101398929678,
|
700 |
+
"grad_norm": 4.715926647186279,
|
701 |
+
"learning_rate": 1.2007719993740547e-05,
|
702 |
+
"loss": 7.8566,
|
703 |
+
"step": 9800
|
704 |
+
},
|
705 |
+
{
|
706 |
+
"epoch": 0.9294901887146747,
|
707 |
+
"grad_norm": 2.6845057010650635,
|
708 |
+
"learning_rate": 1.1903395754003444e-05,
|
709 |
+
"loss": 7.8776,
|
710 |
+
"step": 9900
|
711 |
+
},
|
712 |
+
{
|
713 |
+
"epoch": 0.9388789784996714,
|
714 |
+
"grad_norm": 2.62907075881958,
|
715 |
+
"learning_rate": 1.1799071514266341e-05,
|
716 |
+
"loss": 7.8558,
|
717 |
+
"step": 10000
|
718 |
+
},
|
719 |
+
{
|
720 |
+
"epoch": 0.9388789784996714,
|
721 |
+
"eval_loss": 7.849188327789307,
|
722 |
+
"eval_runtime": 1155.7489,
|
723 |
+
"eval_samples_per_second": 294.897,
|
724 |
+
"eval_steps_per_second": 2.304,
|
725 |
+
"step": 10000
|
726 |
+
},
|
727 |
+
{
|
728 |
+
"epoch": 0.9482677682846681,
|
729 |
+
"grad_norm": 4.570381164550781,
|
730 |
+
"learning_rate": 1.1694747274529237e-05,
|
731 |
+
"loss": 7.848,
|
732 |
+
"step": 10100
|
733 |
+
},
|
734 |
+
{
|
735 |
+
"epoch": 0.9576565580696648,
|
736 |
+
"grad_norm": 21.764062881469727,
|
737 |
+
"learning_rate": 1.1590423034792135e-05,
|
738 |
+
"loss": 7.8227,
|
739 |
+
"step": 10200
|
740 |
+
},
|
741 |
+
{
|
742 |
+
"epoch": 0.9670453478546616,
|
743 |
+
"grad_norm": 18.442140579223633,
|
744 |
+
"learning_rate": 1.1486098795055033e-05,
|
745 |
+
"loss": 7.8311,
|
746 |
+
"step": 10300
|
747 |
+
},
|
748 |
+
{
|
749 |
+
"epoch": 0.9764341376396583,
|
750 |
+
"grad_norm": 4.737902641296387,
|
751 |
+
"learning_rate": 1.1381774555317928e-05,
|
752 |
+
"loss": 7.8437,
|
753 |
+
"step": 10400
|
754 |
+
},
|
755 |
+
{
|
756 |
+
"epoch": 0.9858229274246549,
|
757 |
+
"grad_norm": 3.0295650959014893,
|
758 |
+
"learning_rate": 1.1277450315580827e-05,
|
759 |
+
"loss": 7.8454,
|
760 |
+
"step": 10500
|
761 |
+
},
|
762 |
+
{
|
763 |
+
"epoch": 0.9952117172096516,
|
764 |
+
"grad_norm": 3.0269651412963867,
|
765 |
+
"learning_rate": 1.1173126075843724e-05,
|
766 |
+
"loss": 7.8362,
|
767 |
+
"step": 10600
|
768 |
+
},
|
769 |
+
{
|
770 |
+
"epoch": 1.0046005069946484,
|
771 |
+
"grad_norm": 4.033662796020508,
|
772 |
+
"learning_rate": 1.1069845078503991e-05,
|
773 |
+
"loss": 7.8681,
|
774 |
+
"step": 10700
|
775 |
+
},
|
776 |
+
{
|
777 |
+
"epoch": 1.013989296779645,
|
778 |
+
"grad_norm": 3.5319488048553467,
|
779 |
+
"learning_rate": 1.0965520838766888e-05,
|
780 |
+
"loss": 7.8745,
|
781 |
+
"step": 10800
|
782 |
+
},
|
783 |
+
{
|
784 |
+
"epoch": 1.0233780865646418,
|
785 |
+
"grad_norm": 2.7731130123138428,
|
786 |
+
"learning_rate": 1.0861196599029787e-05,
|
787 |
+
"loss": 7.8339,
|
788 |
+
"step": 10900
|
789 |
+
},
|
790 |
+
{
|
791 |
+
"epoch": 1.0327668763496385,
|
792 |
+
"grad_norm": 4.000971794128418,
|
793 |
+
"learning_rate": 1.0756872359292683e-05,
|
794 |
+
"loss": 7.8458,
|
795 |
+
"step": 11000
|
796 |
+
},
|
797 |
+
{
|
798 |
+
"epoch": 1.0421556661346352,
|
799 |
+
"grad_norm": 15.05604362487793,
|
800 |
+
"learning_rate": 1.065254811955558e-05,
|
801 |
+
"loss": 7.8493,
|
802 |
+
"step": 11100
|
803 |
+
},
|
804 |
+
{
|
805 |
+
"epoch": 1.0515444559196319,
|
806 |
+
"grad_norm": 4.498584747314453,
|
807 |
+
"learning_rate": 1.0548223879818477e-05,
|
808 |
+
"loss": 7.8317,
|
809 |
+
"step": 11200
|
810 |
+
},
|
811 |
+
{
|
812 |
+
"epoch": 1.0609332457046288,
|
813 |
+
"grad_norm": 2.8218085765838623,
|
814 |
+
"learning_rate": 1.0443899640081374e-05,
|
815 |
+
"loss": 7.841,
|
816 |
+
"step": 11300
|
817 |
+
},
|
818 |
+
{
|
819 |
+
"epoch": 1.0703220354896255,
|
820 |
+
"grad_norm": 3.627685785293579,
|
821 |
+
"learning_rate": 1.0339575400344271e-05,
|
822 |
+
"loss": 7.8292,
|
823 |
+
"step": 11400
|
824 |
+
},
|
825 |
+
{
|
826 |
+
"epoch": 1.0797108252746221,
|
827 |
+
"grad_norm": 4.804520606994629,
|
828 |
+
"learning_rate": 1.0235251160607167e-05,
|
829 |
+
"loss": 7.8121,
|
830 |
+
"step": 11500
|
831 |
+
},
|
832 |
+
{
|
833 |
+
"epoch": 1.0890996150596188,
|
834 |
+
"grad_norm": 15.256156921386719,
|
835 |
+
"learning_rate": 1.0130926920870066e-05,
|
836 |
+
"loss": 7.8165,
|
837 |
+
"step": 11600
|
838 |
+
},
|
839 |
+
{
|
840 |
+
"epoch": 1.0984884048446155,
|
841 |
+
"grad_norm": 3.684401273727417,
|
842 |
+
"learning_rate": 1.0026602681132963e-05,
|
843 |
+
"loss": 7.8259,
|
844 |
+
"step": 11700
|
845 |
+
},
|
846 |
+
{
|
847 |
+
"epoch": 1.1078771946296122,
|
848 |
+
"grad_norm": 3.7146763801574707,
|
849 |
+
"learning_rate": 9.92227844139586e-06,
|
850 |
+
"loss": 7.8303,
|
851 |
+
"step": 11800
|
852 |
+
},
|
853 |
+
{
|
854 |
+
"epoch": 1.117265984414609,
|
855 |
+
"grad_norm": 3.4437708854675293,
|
856 |
+
"learning_rate": 9.817954201658755e-06,
|
857 |
+
"loss": 7.809,
|
858 |
+
"step": 11900
|
859 |
+
},
|
860 |
+
{
|
861 |
+
"epoch": 1.1266547741996056,
|
862 |
+
"grad_norm": 4.232120990753174,
|
863 |
+
"learning_rate": 9.713629961921654e-06,
|
864 |
+
"loss": 7.818,
|
865 |
+
"step": 12000
|
866 |
+
},
|
867 |
+
{
|
868 |
+
"epoch": 1.1360435639846025,
|
869 |
+
"grad_norm": 3.4967739582061768,
|
870 |
+
"learning_rate": 9.60930572218455e-06,
|
871 |
+
"loss": 7.8071,
|
872 |
+
"step": 12100
|
873 |
+
},
|
874 |
+
{
|
875 |
+
"epoch": 1.1454323537695992,
|
876 |
+
"grad_norm": 10.542444229125977,
|
877 |
+
"learning_rate": 9.504981482447447e-06,
|
878 |
+
"loss": 7.801,
|
879 |
+
"step": 12200
|
880 |
+
},
|
881 |
+
{
|
882 |
+
"epoch": 1.1548211435545959,
|
883 |
+
"grad_norm": 3.744981527328491,
|
884 |
+
"learning_rate": 9.400657242710344e-06,
|
885 |
+
"loss": 7.8123,
|
886 |
+
"step": 12300
|
887 |
+
},
|
888 |
+
{
|
889 |
+
"epoch": 1.1642099333395926,
|
890 |
+
"grad_norm": 3.3549323081970215,
|
891 |
+
"learning_rate": 9.296333002973241e-06,
|
892 |
+
"loss": 7.8203,
|
893 |
+
"step": 12400
|
894 |
+
},
|
895 |
+
{
|
896 |
+
"epoch": 1.1735987231245892,
|
897 |
+
"grad_norm": 5.337845325469971,
|
898 |
+
"learning_rate": 9.192008763236138e-06,
|
899 |
+
"loss": 7.8609,
|
900 |
+
"step": 12500
|
901 |
+
},
|
902 |
+
{
|
903 |
+
"epoch": 1.182987512909586,
|
904 |
+
"grad_norm": 3.204465627670288,
|
905 |
+
"learning_rate": 9.087684523499036e-06,
|
906 |
+
"loss": 7.7782,
|
907 |
+
"step": 12600
|
908 |
+
},
|
909 |
+
{
|
910 |
+
"epoch": 1.1923763026945826,
|
911 |
+
"grad_norm": 4.669897079467773,
|
912 |
+
"learning_rate": 8.983360283761933e-06,
|
913 |
+
"loss": 7.8092,
|
914 |
+
"step": 12700
|
915 |
+
},
|
916 |
+
{
|
917 |
+
"epoch": 1.2017650924795793,
|
918 |
+
"grad_norm": 3.1824800968170166,
|
919 |
+
"learning_rate": 8.87903604402483e-06,
|
920 |
+
"loss": 7.815,
|
921 |
+
"step": 12800
|
922 |
+
},
|
923 |
+
{
|
924 |
+
"epoch": 1.211153882264576,
|
925 |
+
"grad_norm": 3.6459527015686035,
|
926 |
+
"learning_rate": 8.774711804287727e-06,
|
927 |
+
"loss": 7.8196,
|
928 |
+
"step": 12900
|
929 |
+
},
|
930 |
+
{
|
931 |
+
"epoch": 1.220542672049573,
|
932 |
+
"grad_norm": 3.732983112335205,
|
933 |
+
"learning_rate": 8.670387564550624e-06,
|
934 |
+
"loss": 7.8206,
|
935 |
+
"step": 13000
|
936 |
+
},
|
937 |
+
{
|
938 |
+
"epoch": 1.2299314618345696,
|
939 |
+
"grad_norm": 4.478656768798828,
|
940 |
+
"learning_rate": 8.566063324813521e-06,
|
941 |
+
"loss": 7.8022,
|
942 |
+
"step": 13100
|
943 |
+
},
|
944 |
+
{
|
945 |
+
"epoch": 1.2393202516195663,
|
946 |
+
"grad_norm": 3.7781801223754883,
|
947 |
+
"learning_rate": 8.461739085076418e-06,
|
948 |
+
"loss": 7.8043,
|
949 |
+
"step": 13200
|
950 |
+
},
|
951 |
+
{
|
952 |
+
"epoch": 1.248709041404563,
|
953 |
+
"grad_norm": 5.932605743408203,
|
954 |
+
"learning_rate": 8.357414845339316e-06,
|
955 |
+
"loss": 7.7823,
|
956 |
+
"step": 13300
|
957 |
+
},
|
958 |
+
{
|
959 |
+
"epoch": 1.2580978311895596,
|
960 |
+
"grad_norm": 3.8288068771362305,
|
961 |
+
"learning_rate": 8.254133847999584e-06,
|
962 |
+
"loss": 7.8061,
|
963 |
+
"step": 13400
|
964 |
+
},
|
965 |
+
{
|
966 |
+
"epoch": 1.2674866209745563,
|
967 |
+
"grad_norm": 4.60470724105835,
|
968 |
+
"learning_rate": 8.14980960826248e-06,
|
969 |
+
"loss": 7.8016,
|
970 |
+
"step": 13500
|
971 |
+
},
|
972 |
+
{
|
973 |
+
"epoch": 1.276875410759553,
|
974 |
+
"grad_norm": 5.450839996337891,
|
975 |
+
"learning_rate": 8.045485368525377e-06,
|
976 |
+
"loss": 7.8076,
|
977 |
+
"step": 13600
|
978 |
+
},
|
979 |
+
{
|
980 |
+
"epoch": 1.28626420054455,
|
981 |
+
"grad_norm": 7.866298198699951,
|
982 |
+
"learning_rate": 7.941161128788276e-06,
|
983 |
+
"loss": 7.7996,
|
984 |
+
"step": 13700
|
985 |
+
},
|
986 |
+
{
|
987 |
+
"epoch": 1.2956529903295464,
|
988 |
+
"grad_norm": 3.059967041015625,
|
989 |
+
"learning_rate": 7.836836889051171e-06,
|
990 |
+
"loss": 7.8035,
|
991 |
+
"step": 13800
|
992 |
+
},
|
993 |
+
{
|
994 |
+
"epoch": 1.3050417801145433,
|
995 |
+
"grad_norm": 3.5380911827087402,
|
996 |
+
"learning_rate": 7.732512649314069e-06,
|
997 |
+
"loss": 7.8092,
|
998 |
+
"step": 13900
|
999 |
+
},
|
1000 |
+
{
|
1001 |
+
"epoch": 1.31443056989954,
|
1002 |
+
"grad_norm": 4.589097499847412,
|
1003 |
+
"learning_rate": 7.628188409576966e-06,
|
1004 |
+
"loss": 7.7902,
|
1005 |
+
"step": 14000
|
1006 |
+
},
|
1007 |
+
{
|
1008 |
+
"epoch": 1.3238193596845367,
|
1009 |
+
"grad_norm": 6.932407855987549,
|
1010 |
+
"learning_rate": 7.523864169839863e-06,
|
1011 |
+
"loss": 7.8114,
|
1012 |
+
"step": 14100
|
1013 |
+
},
|
1014 |
+
{
|
1015 |
+
"epoch": 1.3332081494695334,
|
1016 |
+
"grad_norm": 3.5786869525909424,
|
1017 |
+
"learning_rate": 7.41953993010276e-06,
|
1018 |
+
"loss": 7.8112,
|
1019 |
+
"step": 14200
|
1020 |
+
},
|
1021 |
+
{
|
1022 |
+
"epoch": 1.34259693925453,
|
1023 |
+
"grad_norm": 4.283187389373779,
|
1024 |
+
"learning_rate": 7.315215690365657e-06,
|
1025 |
+
"loss": 7.8036,
|
1026 |
+
"step": 14300
|
1027 |
+
},
|
1028 |
+
{
|
1029 |
+
"epoch": 1.3519857290395267,
|
1030 |
+
"grad_norm": 14.625285148620605,
|
1031 |
+
"learning_rate": 7.210891450628554e-06,
|
1032 |
+
"loss": 7.8178,
|
1033 |
+
"step": 14400
|
1034 |
+
},
|
1035 |
+
{
|
1036 |
+
"epoch": 1.3613745188245234,
|
1037 |
+
"grad_norm": 3.5072567462921143,
|
1038 |
+
"learning_rate": 7.106567210891451e-06,
|
1039 |
+
"loss": 7.8391,
|
1040 |
+
"step": 14500
|
1041 |
+
},
|
1042 |
+
{
|
1043 |
+
"epoch": 1.3707633086095203,
|
1044 |
+
"grad_norm": 4.140475749969482,
|
1045 |
+
"learning_rate": 7.002242971154349e-06,
|
1046 |
+
"loss": 7.8151,
|
1047 |
+
"step": 14600
|
1048 |
+
},
|
1049 |
+
{
|
1050 |
+
"epoch": 1.380152098394517,
|
1051 |
+
"grad_norm": 6.985396385192871,
|
1052 |
+
"learning_rate": 6.897918731417246e-06,
|
1053 |
+
"loss": 7.7957,
|
1054 |
+
"step": 14700
|
1055 |
+
},
|
1056 |
+
{
|
1057 |
+
"epoch": 1.3895408881795137,
|
1058 |
+
"grad_norm": 3.8024065494537354,
|
1059 |
+
"learning_rate": 6.793594491680142e-06,
|
1060 |
+
"loss": 7.7833,
|
1061 |
+
"step": 14800
|
1062 |
+
},
|
1063 |
+
{
|
1064 |
+
"epoch": 1.3989296779645104,
|
1065 |
+
"grad_norm": 4.183823585510254,
|
1066 |
+
"learning_rate": 6.689270251943039e-06,
|
1067 |
+
"loss": 7.8049,
|
1068 |
+
"step": 14900
|
1069 |
+
},
|
1070 |
+
{
|
1071 |
+
"epoch": 1.408318467749507,
|
1072 |
+
"grad_norm": 3.431105375289917,
|
1073 |
+
"learning_rate": 6.5849460122059365e-06,
|
1074 |
+
"loss": 7.8163,
|
1075 |
+
"step": 15000
|
1076 |
+
},
|
1077 |
+
{
|
1078 |
+
"epoch": 1.408318467749507,
|
1079 |
+
"eval_loss": 7.807833671569824,
|
1080 |
+
"eval_runtime": 335.7694,
|
1081 |
+
"eval_samples_per_second": 1015.063,
|
1082 |
+
"eval_steps_per_second": 7.931,
|
1083 |
+
"step": 15000
|
1084 |
+
},
|
1085 |
+
{
|
1086 |
+
"epoch": 1.4177072575345038,
|
1087 |
+
"grad_norm": 8.183846473693848,
|
1088 |
+
"learning_rate": 6.480621772468834e-06,
|
1089 |
+
"loss": 7.7864,
|
1090 |
+
"step": 15100
|
1091 |
+
},
|
1092 |
+
{
|
1093 |
+
"epoch": 1.4270960473195005,
|
1094 |
+
"grad_norm": 11.66592788696289,
|
1095 |
+
"learning_rate": 6.376297532731731e-06,
|
1096 |
+
"loss": 7.8241,
|
1097 |
+
"step": 15200
|
1098 |
+
},
|
1099 |
+
{
|
1100 |
+
"epoch": 1.4364848371044971,
|
1101 |
+
"grad_norm": 9.620813369750977,
|
1102 |
+
"learning_rate": 6.271973292994628e-06,
|
1103 |
+
"loss": 7.7694,
|
1104 |
+
"step": 15300
|
1105 |
+
},
|
1106 |
+
{
|
1107 |
+
"epoch": 1.4458736268894938,
|
1108 |
+
"grad_norm": 4.250575065612793,
|
1109 |
+
"learning_rate": 6.167649053257525e-06,
|
1110 |
+
"loss": 7.7784,
|
1111 |
+
"step": 15400
|
1112 |
+
},
|
1113 |
+
{
|
1114 |
+
"epoch": 1.4552624166744907,
|
1115 |
+
"grad_norm": 3.8679049015045166,
|
1116 |
+
"learning_rate": 6.0633248135204214e-06,
|
1117 |
+
"loss": 7.7628,
|
1118 |
+
"step": 15500
|
1119 |
+
},
|
1120 |
+
{
|
1121 |
+
"epoch": 1.4646512064594874,
|
1122 |
+
"grad_norm": 4.648382186889648,
|
1123 |
+
"learning_rate": 5.959000573783319e-06,
|
1124 |
+
"loss": 7.8044,
|
1125 |
+
"step": 15600
|
1126 |
+
},
|
1127 |
+
{
|
1128 |
+
"epoch": 1.4740399962444841,
|
1129 |
+
"grad_norm": 4.5424113273620605,
|
1130 |
+
"learning_rate": 5.854676334046217e-06,
|
1131 |
+
"loss": 7.7871,
|
1132 |
+
"step": 15700
|
1133 |
+
},
|
1134 |
+
{
|
1135 |
+
"epoch": 1.4834287860294808,
|
1136 |
+
"grad_norm": 4.026553630828857,
|
1137 |
+
"learning_rate": 5.750352094309113e-06,
|
1138 |
+
"loss": 7.809,
|
1139 |
+
"step": 15800
|
1140 |
+
},
|
1141 |
+
{
|
1142 |
+
"epoch": 1.4928175758144775,
|
1143 |
+
"grad_norm": 6.175102233886719,
|
1144 |
+
"learning_rate": 5.647071096969381e-06,
|
1145 |
+
"loss": 7.7955,
|
1146 |
+
"step": 15900
|
1147 |
+
},
|
1148 |
+
{
|
1149 |
+
"epoch": 1.5022063655994742,
|
1150 |
+
"grad_norm": 4.672608375549316,
|
1151 |
+
"learning_rate": 5.542746857232279e-06,
|
1152 |
+
"loss": 7.8056,
|
1153 |
+
"step": 16000
|
1154 |
+
},
|
1155 |
+
{
|
1156 |
+
"epoch": 1.5115951553844709,
|
1157 |
+
"grad_norm": 7.012312412261963,
|
1158 |
+
"learning_rate": 5.438422617495176e-06,
|
1159 |
+
"loss": 7.774,
|
1160 |
+
"step": 16100
|
1161 |
+
},
|
1162 |
+
{
|
1163 |
+
"epoch": 1.5209839451694678,
|
1164 |
+
"grad_norm": 5.2042131423950195,
|
1165 |
+
"learning_rate": 5.334098377758072e-06,
|
1166 |
+
"loss": 7.7874,
|
1167 |
+
"step": 16200
|
1168 |
+
},
|
1169 |
+
{
|
1170 |
+
"epoch": 1.5303727349544642,
|
1171 |
+
"grad_norm": 3.745805501937866,
|
1172 |
+
"learning_rate": 5.22977413802097e-06,
|
1173 |
+
"loss": 7.7918,
|
1174 |
+
"step": 16300
|
1175 |
+
},
|
1176 |
+
{
|
1177 |
+
"epoch": 1.5397615247394612,
|
1178 |
+
"grad_norm": 4.060446262359619,
|
1179 |
+
"learning_rate": 5.125449898283867e-06,
|
1180 |
+
"loss": 7.7787,
|
1181 |
+
"step": 16400
|
1182 |
+
},
|
1183 |
+
{
|
1184 |
+
"epoch": 1.5491503145244578,
|
1185 |
+
"grad_norm": 21.851919174194336,
|
1186 |
+
"learning_rate": 5.021125658546764e-06,
|
1187 |
+
"loss": 7.7881,
|
1188 |
+
"step": 16500
|
1189 |
+
},
|
1190 |
+
{
|
1191 |
+
"epoch": 1.5585391043094545,
|
1192 |
+
"grad_norm": 4.261013507843018,
|
1193 |
+
"learning_rate": 4.916801418809661e-06,
|
1194 |
+
"loss": 7.7723,
|
1195 |
+
"step": 16600
|
1196 |
+
},
|
1197 |
+
{
|
1198 |
+
"epoch": 1.5679278940944512,
|
1199 |
+
"grad_norm": 3.9473931789398193,
|
1200 |
+
"learning_rate": 4.812477179072558e-06,
|
1201 |
+
"loss": 7.7809,
|
1202 |
+
"step": 16700
|
1203 |
+
},
|
1204 |
+
{
|
1205 |
+
"epoch": 1.577316683879448,
|
1206 |
+
"grad_norm": 6.088964939117432,
|
1207 |
+
"learning_rate": 4.709196181732826e-06,
|
1208 |
+
"loss": 7.8096,
|
1209 |
+
"step": 16800
|
1210 |
+
},
|
1211 |
+
{
|
1212 |
+
"epoch": 1.5867054736644448,
|
1213 |
+
"grad_norm": 7.912614822387695,
|
1214 |
+
"learning_rate": 4.604871941995723e-06,
|
1215 |
+
"loss": 7.7559,
|
1216 |
+
"step": 16900
|
1217 |
+
},
|
1218 |
+
{
|
1219 |
+
"epoch": 1.5960942634494413,
|
1220 |
+
"grad_norm": 7.268245697021484,
|
1221 |
+
"learning_rate": 4.50054770225862e-06,
|
1222 |
+
"loss": 7.8063,
|
1223 |
+
"step": 17000
|
1224 |
+
},
|
1225 |
+
{
|
1226 |
+
"epoch": 1.6054830532344382,
|
1227 |
+
"grad_norm": 15.331354141235352,
|
1228 |
+
"learning_rate": 4.3962234625215175e-06,
|
1229 |
+
"loss": 7.8137,
|
1230 |
+
"step": 17100
|
1231 |
+
},
|
1232 |
+
{
|
1233 |
+
"epoch": 1.6148718430194346,
|
1234 |
+
"grad_norm": 5.976458549499512,
|
1235 |
+
"learning_rate": 4.291899222784414e-06,
|
1236 |
+
"loss": 7.761,
|
1237 |
+
"step": 17200
|
1238 |
+
},
|
1239 |
+
{
|
1240 |
+
"epoch": 1.6242606328044316,
|
1241 |
+
"grad_norm": 4.332185745239258,
|
1242 |
+
"learning_rate": 4.187574983047311e-06,
|
1243 |
+
"loss": 7.7672,
|
1244 |
+
"step": 17300
|
1245 |
+
},
|
1246 |
+
{
|
1247 |
+
"epoch": 1.6336494225894282,
|
1248 |
+
"grad_norm": 5.834400653839111,
|
1249 |
+
"learning_rate": 4.083250743310209e-06,
|
1250 |
+
"loss": 7.7939,
|
1251 |
+
"step": 17400
|
1252 |
+
},
|
1253 |
+
{
|
1254 |
+
"epoch": 1.643038212374425,
|
1255 |
+
"grad_norm": 17.61608123779297,
|
1256 |
+
"learning_rate": 3.978926503573105e-06,
|
1257 |
+
"loss": 7.8052,
|
1258 |
+
"step": 17500
|
1259 |
+
},
|
1260 |
+
{
|
1261 |
+
"epoch": 1.6524270021594216,
|
1262 |
+
"grad_norm": 4.811455249786377,
|
1263 |
+
"learning_rate": 3.8746022638360024e-06,
|
1264 |
+
"loss": 7.7519,
|
1265 |
+
"step": 17600
|
1266 |
+
},
|
1267 |
+
{
|
1268 |
+
"epoch": 1.6618157919444183,
|
1269 |
+
"grad_norm": 4.777581691741943,
|
1270 |
+
"learning_rate": 3.7702780240988996e-06,
|
1271 |
+
"loss": 7.7643,
|
1272 |
+
"step": 17700
|
1273 |
+
},
|
1274 |
+
{
|
1275 |
+
"epoch": 1.6712045817294152,
|
1276 |
+
"grad_norm": 3.996382474899292,
|
1277 |
+
"learning_rate": 3.6659537843617963e-06,
|
1278 |
+
"loss": 7.7823,
|
1279 |
+
"step": 17800
|
1280 |
+
},
|
1281 |
+
{
|
1282 |
+
"epoch": 1.6805933715144117,
|
1283 |
+
"grad_norm": 13.98630142211914,
|
1284 |
+
"learning_rate": 3.561629544624694e-06,
|
1285 |
+
"loss": 7.7507,
|
1286 |
+
"step": 17900
|
1287 |
+
},
|
1288 |
+
{
|
1289 |
+
"epoch": 1.6899821612994086,
|
1290 |
+
"grad_norm": 5.8219499588012695,
|
1291 |
+
"learning_rate": 3.457305304887591e-06,
|
1292 |
+
"loss": 7.777,
|
1293 |
+
"step": 18000
|
1294 |
+
},
|
1295 |
+
{
|
1296 |
+
"epoch": 1.699370951084405,
|
1297 |
+
"grad_norm": 6.051421165466309,
|
1298 |
+
"learning_rate": 3.352981065150488e-06,
|
1299 |
+
"loss": 7.786,
|
1300 |
+
"step": 18100
|
1301 |
+
},
|
1302 |
+
{
|
1303 |
+
"epoch": 1.708759740869402,
|
1304 |
+
"grad_norm": 7.037374019622803,
|
1305 |
+
"learning_rate": 3.248656825413385e-06,
|
1306 |
+
"loss": 7.8097,
|
1307 |
+
"step": 18200
|
1308 |
+
},
|
1309 |
+
{
|
1310 |
+
"epoch": 1.7181485306543987,
|
1311 |
+
"grad_norm": 4.905298709869385,
|
1312 |
+
"learning_rate": 3.1443325856762826e-06,
|
1313 |
+
"loss": 7.7749,
|
1314 |
+
"step": 18300
|
1315 |
+
},
|
1316 |
+
{
|
1317 |
+
"epoch": 1.7275373204393953,
|
1318 |
+
"grad_norm": 10.381073951721191,
|
1319 |
+
"learning_rate": 3.0400083459391793e-06,
|
1320 |
+
"loss": 7.7626,
|
1321 |
+
"step": 18400
|
1322 |
+
},
|
1323 |
+
{
|
1324 |
+
"epoch": 1.736926110224392,
|
1325 |
+
"grad_norm": 4.788754463195801,
|
1326 |
+
"learning_rate": 2.9356841062020765e-06,
|
1327 |
+
"loss": 7.7783,
|
1328 |
+
"step": 18500
|
1329 |
+
},
|
1330 |
+
{
|
1331 |
+
"epoch": 1.7463149000093887,
|
1332 |
+
"grad_norm": 7.451583385467529,
|
1333 |
+
"learning_rate": 2.8313598664649732e-06,
|
1334 |
+
"loss": 7.7552,
|
1335 |
+
"step": 18600
|
1336 |
+
},
|
1337 |
+
{
|
1338 |
+
"epoch": 1.7557036897943856,
|
1339 |
+
"grad_norm": 4.68840217590332,
|
1340 |
+
"learning_rate": 2.7270356267278704e-06,
|
1341 |
+
"loss": 7.7837,
|
1342 |
+
"step": 18700
|
1343 |
+
},
|
1344 |
+
{
|
1345 |
+
"epoch": 1.765092479579382,
|
1346 |
+
"grad_norm": 5.026941776275635,
|
1347 |
+
"learning_rate": 2.6227113869907676e-06,
|
1348 |
+
"loss": 7.7583,
|
1349 |
+
"step": 18800
|
1350 |
+
},
|
1351 |
+
{
|
1352 |
+
"epoch": 1.774481269364379,
|
1353 |
+
"grad_norm": 5.614253520965576,
|
1354 |
+
"learning_rate": 2.5183871472536647e-06,
|
1355 |
+
"loss": 7.7617,
|
1356 |
+
"step": 18900
|
1357 |
+
},
|
1358 |
+
{
|
1359 |
+
"epoch": 1.7838700591493757,
|
1360 |
+
"grad_norm": 67.81696319580078,
|
1361 |
+
"learning_rate": 2.414062907516562e-06,
|
1362 |
+
"loss": 7.7649,
|
1363 |
+
"step": 19000
|
1364 |
+
},
|
1365 |
+
{
|
1366 |
+
"epoch": 1.7932588489343724,
|
1367 |
+
"grad_norm": 5.342656135559082,
|
1368 |
+
"learning_rate": 2.3097386677794586e-06,
|
1369 |
+
"loss": 7.7767,
|
1370 |
+
"step": 19100
|
1371 |
+
},
|
1372 |
+
{
|
1373 |
+
"epoch": 1.802647638719369,
|
1374 |
+
"grad_norm": 5.355633735656738,
|
1375 |
+
"learning_rate": 2.205414428042356e-06,
|
1376 |
+
"loss": 7.7565,
|
1377 |
+
"step": 19200
|
1378 |
+
},
|
1379 |
+
{
|
1380 |
+
"epoch": 1.8120364285043657,
|
1381 |
+
"grad_norm": 4.396353244781494,
|
1382 |
+
"learning_rate": 2.101090188305253e-06,
|
1383 |
+
"loss": 7.7702,
|
1384 |
+
"step": 19300
|
1385 |
+
},
|
1386 |
+
{
|
1387 |
+
"epoch": 1.8214252182893627,
|
1388 |
+
"grad_norm": 7.156164646148682,
|
1389 |
+
"learning_rate": 1.99676594856815e-06,
|
1390 |
+
"loss": 7.7552,
|
1391 |
+
"step": 19400
|
1392 |
+
},
|
1393 |
+
{
|
1394 |
+
"epoch": 1.8308140080743591,
|
1395 |
+
"grad_norm": 6.207972049713135,
|
1396 |
+
"learning_rate": 1.8934849512284182e-06,
|
1397 |
+
"loss": 7.7511,
|
1398 |
+
"step": 19500
|
1399 |
+
},
|
1400 |
+
{
|
1401 |
+
"epoch": 1.840202797859356,
|
1402 |
+
"grad_norm": 19.932350158691406,
|
1403 |
+
"learning_rate": 1.7891607114913152e-06,
|
1404 |
+
"loss": 7.7818,
|
1405 |
+
"step": 19600
|
1406 |
+
},
|
1407 |
+
{
|
1408 |
+
"epoch": 1.8495915876443525,
|
1409 |
+
"grad_norm": 5.5889811515808105,
|
1410 |
+
"learning_rate": 1.6848364717542123e-06,
|
1411 |
+
"loss": 7.7704,
|
1412 |
+
"step": 19700
|
1413 |
+
},
|
1414 |
+
{
|
1415 |
+
"epoch": 1.8589803774293494,
|
1416 |
+
"grad_norm": 4.45137357711792,
|
1417 |
+
"learning_rate": 1.5805122320171093e-06,
|
1418 |
+
"loss": 7.7824,
|
1419 |
+
"step": 19800
|
1420 |
+
},
|
1421 |
+
{
|
1422 |
+
"epoch": 1.868369167214346,
|
1423 |
+
"grad_norm": 4.196966648101807,
|
1424 |
+
"learning_rate": 1.4761879922800062e-06,
|
1425 |
+
"loss": 7.751,
|
1426 |
+
"step": 19900
|
1427 |
+
},
|
1428 |
+
{
|
1429 |
+
"epoch": 1.8777579569993428,
|
1430 |
+
"grad_norm": 4.43785285949707,
|
1431 |
+
"learning_rate": 1.3718637525429034e-06,
|
1432 |
+
"loss": 7.7868,
|
1433 |
+
"step": 20000
|
1434 |
+
},
|
1435 |
+
{
|
1436 |
+
"epoch": 1.8777579569993428,
|
1437 |
+
"eval_loss": 7.7942399978637695,
|
1438 |
+
"eval_runtime": 335.6395,
|
1439 |
+
"eval_samples_per_second": 1015.456,
|
1440 |
+
"eval_steps_per_second": 7.934,
|
1441 |
+
"step": 20000
|
1442 |
+
},
|
1443 |
+
{
|
1444 |
+
"epoch": 1.8871467467843395,
|
1445 |
+
"grad_norm": 4.186898231506348,
|
1446 |
+
"learning_rate": 1.2675395128058006e-06,
|
1447 |
+
"loss": 7.7981,
|
1448 |
+
"step": 20100
|
1449 |
+
},
|
1450 |
+
{
|
1451 |
+
"epoch": 1.8965355365693362,
|
1452 |
+
"grad_norm": 3.898251533508301,
|
1453 |
+
"learning_rate": 1.1632152730686975e-06,
|
1454 |
+
"loss": 7.7673,
|
1455 |
+
"step": 20200
|
1456 |
+
},
|
1457 |
+
{
|
1458 |
+
"epoch": 1.905924326354333,
|
1459 |
+
"grad_norm": 3.6025681495666504,
|
1460 |
+
"learning_rate": 1.0588910333315947e-06,
|
1461 |
+
"loss": 7.7695,
|
1462 |
+
"step": 20300
|
1463 |
+
},
|
1464 |
+
{
|
1465 |
+
"epoch": 1.9153131161393295,
|
1466 |
+
"grad_norm": 4.628914833068848,
|
1467 |
+
"learning_rate": 9.545667935944918e-07,
|
1468 |
+
"loss": 7.7587,
|
1469 |
+
"step": 20400
|
1470 |
+
},
|
1471 |
+
{
|
1472 |
+
"epoch": 1.9247019059243264,
|
1473 |
+
"grad_norm": 4.496853351593018,
|
1474 |
+
"learning_rate": 8.502425538573889e-07,
|
1475 |
+
"loss": 7.7444,
|
1476 |
+
"step": 20500
|
1477 |
+
}
|
1478 |
+
],
|
1479 |
+
"logging_steps": 100,
|
1480 |
+
"max_steps": 21302,
|
1481 |
+
"num_input_tokens_seen": 0,
|
1482 |
+
"num_train_epochs": 2,
|
1483 |
+
"save_steps": 500,
|
1484 |
+
"stateful_callbacks": {
|
1485 |
+
"TrainerControl": {
|
1486 |
+
"args": {
|
1487 |
+
"should_epoch_stop": false,
|
1488 |
+
"should_evaluate": false,
|
1489 |
+
"should_log": false,
|
1490 |
+
"should_save": true,
|
1491 |
+
"should_training_stop": false
|
1492 |
+
},
|
1493 |
+
"attributes": {}
|
1494 |
+
}
|
1495 |
+
},
|
1496 |
+
"total_flos": 0.0,
|
1497 |
+
"train_batch_size": 128,
|
1498 |
+
"trial_name": null,
|
1499 |
+
"trial_params": null
|
1500 |
+
}
|
checkpoint-20500/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ea049d2d832069bea07040be4041d4f978baa54eb55f016301b0dad6a593f47
|
3 |
+
size 5496
|
checkpoint-20500/vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|