Nataliia767567
commited on
phi_tuned_test_2
Browse files- README.md +57 -0
- adapter_config.json +34 -0
- adapter_model.safetensors +3 -0
- added_tokens.json +13 -0
- all_results.json +8 -0
- special_tokens_map.json +24 -0
- tokenizer.json +0 -0
- tokenizer.model +3 -0
- tokenizer_config.json +132 -0
- train_results.json +8 -0
- trainer_state.json +560 -0
- training_args.bin +3 -0
README.md
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: microsoft/Phi-3-mini-4k-instruct
|
3 |
+
library_name: transformers
|
4 |
+
model_name: checkpoint_dir
|
5 |
+
tags:
|
6 |
+
- generated_from_trainer
|
7 |
+
- trl
|
8 |
+
- sft
|
9 |
+
licence: license
|
10 |
+
---
|
11 |
+
|
12 |
+
# Model Card for checkpoint_dir
|
13 |
+
|
14 |
+
This model is a fine-tuned version of [microsoft/Phi-3-mini-4k-instruct](https://huggingface.co/microsoft/Phi-3-mini-4k-instruct).
|
15 |
+
It has been trained using [TRL](https://github.com/huggingface/trl).
|
16 |
+
|
17 |
+
## Quick start
|
18 |
+
|
19 |
+
```python
|
20 |
+
from transformers import pipeline
|
21 |
+
|
22 |
+
question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
|
23 |
+
generator = pipeline("text-generation", model="Nataliia767567/checkpoint_dir", device="cuda")
|
24 |
+
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
|
25 |
+
print(output["generated_text"])
|
26 |
+
```
|
27 |
+
|
28 |
+
## Training procedure
|
29 |
+
|
30 |
+
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/natalipolyakovska-softserve/huggingface/runs/yvb3ur2f)
|
31 |
+
|
32 |
+
This model was trained with SFT.
|
33 |
+
|
34 |
+
### Framework versions
|
35 |
+
|
36 |
+
- TRL: 0.12.0
|
37 |
+
- Transformers: 4.47.1
|
38 |
+
- Pytorch: 2.2.2
|
39 |
+
- Datasets: 3.2.0
|
40 |
+
- Tokenizers: 0.21.0
|
41 |
+
|
42 |
+
## Citations
|
43 |
+
|
44 |
+
|
45 |
+
|
46 |
+
Cite TRL as:
|
47 |
+
|
48 |
+
```bibtex
|
49 |
+
@misc{vonwerra2022trl,
|
50 |
+
title = {{TRL: Transformer Reinforcement Learning}},
|
51 |
+
author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
|
52 |
+
year = 2020,
|
53 |
+
journal = {GitHub repository},
|
54 |
+
publisher = {GitHub},
|
55 |
+
howpublished = {\url{https://github.com/huggingface/trl}}
|
56 |
+
}
|
57 |
+
```
|
adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "microsoft/Phi-3-mini-4k-instruct",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.05,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 16,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"gate_up_proj",
|
27 |
+
"qkv_proj",
|
28 |
+
"o_proj",
|
29 |
+
"down_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fdd631e2d13eb4b8bcea80269dd0e81600583dc80141ddbf0b01b0403f159a31
|
3 |
+
size 100697728
|
added_tokens.json
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<|assistant|>": 32001,
|
3 |
+
"<|endoftext|>": 32000,
|
4 |
+
"<|end|>": 32007,
|
5 |
+
"<|placeholder1|>": 32002,
|
6 |
+
"<|placeholder2|>": 32003,
|
7 |
+
"<|placeholder3|>": 32004,
|
8 |
+
"<|placeholder4|>": 32005,
|
9 |
+
"<|placeholder5|>": 32008,
|
10 |
+
"<|placeholder6|>": 32009,
|
11 |
+
"<|system|>": 32006,
|
12 |
+
"<|user|>": 32010
|
13 |
+
}
|
all_results.json
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 3.0,
|
3 |
+
"total_flos": 6.836552056302797e+16,
|
4 |
+
"train_loss": 0.8828093713381475,
|
5 |
+
"train_runtime": 1774.0294,
|
6 |
+
"train_samples_per_second": 3.348,
|
7 |
+
"train_steps_per_second": 0.837
|
8 |
+
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<s>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"eos_token": {
|
10 |
+
"content": "<|endoftext|>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": "<unk>",
|
17 |
+
"unk_token": {
|
18 |
+
"content": "<unk>",
|
19 |
+
"lstrip": false,
|
20 |
+
"normalized": false,
|
21 |
+
"rstrip": false,
|
22 |
+
"single_word": false
|
23 |
+
}
|
24 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
|
3 |
+
size 499723
|
tokenizer_config.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_eos_token": false,
|
4 |
+
"add_prefix_space": null,
|
5 |
+
"added_tokens_decoder": {
|
6 |
+
"0": {
|
7 |
+
"content": "<unk>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false,
|
12 |
+
"special": true
|
13 |
+
},
|
14 |
+
"1": {
|
15 |
+
"content": "<s>",
|
16 |
+
"lstrip": false,
|
17 |
+
"normalized": false,
|
18 |
+
"rstrip": false,
|
19 |
+
"single_word": false,
|
20 |
+
"special": true
|
21 |
+
},
|
22 |
+
"2": {
|
23 |
+
"content": "</s>",
|
24 |
+
"lstrip": false,
|
25 |
+
"normalized": false,
|
26 |
+
"rstrip": true,
|
27 |
+
"single_word": false,
|
28 |
+
"special": false
|
29 |
+
},
|
30 |
+
"32000": {
|
31 |
+
"content": "<|endoftext|>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false,
|
36 |
+
"special": true
|
37 |
+
},
|
38 |
+
"32001": {
|
39 |
+
"content": "<|assistant|>",
|
40 |
+
"lstrip": false,
|
41 |
+
"normalized": false,
|
42 |
+
"rstrip": true,
|
43 |
+
"single_word": false,
|
44 |
+
"special": true
|
45 |
+
},
|
46 |
+
"32002": {
|
47 |
+
"content": "<|placeholder1|>",
|
48 |
+
"lstrip": false,
|
49 |
+
"normalized": false,
|
50 |
+
"rstrip": true,
|
51 |
+
"single_word": false,
|
52 |
+
"special": true
|
53 |
+
},
|
54 |
+
"32003": {
|
55 |
+
"content": "<|placeholder2|>",
|
56 |
+
"lstrip": false,
|
57 |
+
"normalized": false,
|
58 |
+
"rstrip": true,
|
59 |
+
"single_word": false,
|
60 |
+
"special": true
|
61 |
+
},
|
62 |
+
"32004": {
|
63 |
+
"content": "<|placeholder3|>",
|
64 |
+
"lstrip": false,
|
65 |
+
"normalized": false,
|
66 |
+
"rstrip": true,
|
67 |
+
"single_word": false,
|
68 |
+
"special": true
|
69 |
+
},
|
70 |
+
"32005": {
|
71 |
+
"content": "<|placeholder4|>",
|
72 |
+
"lstrip": false,
|
73 |
+
"normalized": false,
|
74 |
+
"rstrip": true,
|
75 |
+
"single_word": false,
|
76 |
+
"special": true
|
77 |
+
},
|
78 |
+
"32006": {
|
79 |
+
"content": "<|system|>",
|
80 |
+
"lstrip": false,
|
81 |
+
"normalized": false,
|
82 |
+
"rstrip": true,
|
83 |
+
"single_word": false,
|
84 |
+
"special": true
|
85 |
+
},
|
86 |
+
"32007": {
|
87 |
+
"content": "<|end|>",
|
88 |
+
"lstrip": false,
|
89 |
+
"normalized": false,
|
90 |
+
"rstrip": true,
|
91 |
+
"single_word": false,
|
92 |
+
"special": true
|
93 |
+
},
|
94 |
+
"32008": {
|
95 |
+
"content": "<|placeholder5|>",
|
96 |
+
"lstrip": false,
|
97 |
+
"normalized": false,
|
98 |
+
"rstrip": true,
|
99 |
+
"single_word": false,
|
100 |
+
"special": true
|
101 |
+
},
|
102 |
+
"32009": {
|
103 |
+
"content": "<|placeholder6|>",
|
104 |
+
"lstrip": false,
|
105 |
+
"normalized": false,
|
106 |
+
"rstrip": true,
|
107 |
+
"single_word": false,
|
108 |
+
"special": true
|
109 |
+
},
|
110 |
+
"32010": {
|
111 |
+
"content": "<|user|>",
|
112 |
+
"lstrip": false,
|
113 |
+
"normalized": false,
|
114 |
+
"rstrip": true,
|
115 |
+
"single_word": false,
|
116 |
+
"special": true
|
117 |
+
}
|
118 |
+
},
|
119 |
+
"bos_token": "<s>",
|
120 |
+
"chat_template": "{% for message in messages %}{% if message['role'] == 'system' %}{{'<|system|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'user' %}{{'<|user|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'assistant' %}{{'<|assistant|>\n' + message['content'] + '<|end|>\n'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>\n' }}{% else %}{{ eos_token }}{% endif %}",
|
121 |
+
"clean_up_tokenization_spaces": false,
|
122 |
+
"eos_token": "<|endoftext|>",
|
123 |
+
"extra_special_tokens": {},
|
124 |
+
"legacy": false,
|
125 |
+
"model_max_length": 2048,
|
126 |
+
"pad_token": "<unk>",
|
127 |
+
"padding_side": "left",
|
128 |
+
"sp_model_kwargs": {},
|
129 |
+
"tokenizer_class": "LlamaTokenizer",
|
130 |
+
"unk_token": "<unk>",
|
131 |
+
"use_default_system_prompt": false
|
132 |
+
}
|
train_results.json
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 3.0,
|
3 |
+
"total_flos": 6.836552056302797e+16,
|
4 |
+
"train_loss": 0.8828093713381475,
|
5 |
+
"train_runtime": 1774.0294,
|
6 |
+
"train_samples_per_second": 3.348,
|
7 |
+
"train_steps_per_second": 0.837
|
8 |
+
}
|
trainer_state.json
ADDED
@@ -0,0 +1,560 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 3.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 1485,
|
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.04040404040404041,
|
13 |
+
"grad_norm": 0.41251227259635925,
|
14 |
+
"learning_rate": 3.3670033670033673e-07,
|
15 |
+
"loss": 1.6394,
|
16 |
+
"step": 20
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.08080808080808081,
|
20 |
+
"grad_norm": 0.4119182229042053,
|
21 |
+
"learning_rate": 6.734006734006735e-07,
|
22 |
+
"loss": 1.6705,
|
23 |
+
"step": 40
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.12121212121212122,
|
27 |
+
"grad_norm": 0.4101647734642029,
|
28 |
+
"learning_rate": 1.01010101010101e-06,
|
29 |
+
"loss": 1.6682,
|
30 |
+
"step": 60
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.16161616161616163,
|
34 |
+
"grad_norm": 0.4267776310443878,
|
35 |
+
"learning_rate": 1.346801346801347e-06,
|
36 |
+
"loss": 1.55,
|
37 |
+
"step": 80
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.20202020202020202,
|
41 |
+
"grad_norm": 0.444989413022995,
|
42 |
+
"learning_rate": 1.6835016835016838e-06,
|
43 |
+
"loss": 1.6651,
|
44 |
+
"step": 100
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.24242424242424243,
|
48 |
+
"grad_norm": 0.49537068605422974,
|
49 |
+
"learning_rate": 2.02020202020202e-06,
|
50 |
+
"loss": 1.5103,
|
51 |
+
"step": 120
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.2828282828282828,
|
55 |
+
"grad_norm": 0.4137594699859619,
|
56 |
+
"learning_rate": 2.3569023569023572e-06,
|
57 |
+
"loss": 1.5489,
|
58 |
+
"step": 140
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.32323232323232326,
|
62 |
+
"grad_norm": 0.4259663224220276,
|
63 |
+
"learning_rate": 2.693602693602694e-06,
|
64 |
+
"loss": 1.6416,
|
65 |
+
"step": 160
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.36363636363636365,
|
69 |
+
"grad_norm": 0.39349138736724854,
|
70 |
+
"learning_rate": 3.0303030303030305e-06,
|
71 |
+
"loss": 1.554,
|
72 |
+
"step": 180
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.40404040404040403,
|
76 |
+
"grad_norm": 0.4005858302116394,
|
77 |
+
"learning_rate": 3.3670033670033675e-06,
|
78 |
+
"loss": 1.523,
|
79 |
+
"step": 200
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.4444444444444444,
|
83 |
+
"grad_norm": 0.4026505947113037,
|
84 |
+
"learning_rate": 3.7037037037037037e-06,
|
85 |
+
"loss": 1.4523,
|
86 |
+
"step": 220
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.48484848484848486,
|
90 |
+
"grad_norm": 0.48675334453582764,
|
91 |
+
"learning_rate": 4.04040404040404e-06,
|
92 |
+
"loss": 1.3475,
|
93 |
+
"step": 240
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.5252525252525253,
|
97 |
+
"grad_norm": 0.7367889881134033,
|
98 |
+
"learning_rate": 4.377104377104377e-06,
|
99 |
+
"loss": 1.2313,
|
100 |
+
"step": 260
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.5656565656565656,
|
104 |
+
"grad_norm": 0.47958847880363464,
|
105 |
+
"learning_rate": 4.7138047138047145e-06,
|
106 |
+
"loss": 0.9929,
|
107 |
+
"step": 280
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.6060606060606061,
|
111 |
+
"grad_norm": 0.5394216179847717,
|
112 |
+
"learning_rate": 4.999921328558333e-06,
|
113 |
+
"loss": 0.974,
|
114 |
+
"step": 300
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.6464646464646465,
|
118 |
+
"grad_norm": 0.6015712022781372,
|
119 |
+
"learning_rate": 4.995377268577495e-06,
|
120 |
+
"loss": 0.8598,
|
121 |
+
"step": 320
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.6868686868686869,
|
125 |
+
"grad_norm": 0.45018038153648376,
|
126 |
+
"learning_rate": 4.983854712613647e-06,
|
127 |
+
"loss": 0.8335,
|
128 |
+
"step": 340
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.7272727272727273,
|
132 |
+
"grad_norm": 0.5190030336380005,
|
133 |
+
"learning_rate": 4.965385884295467e-06,
|
134 |
+
"loss": 0.8485,
|
135 |
+
"step": 360
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.7676767676767676,
|
139 |
+
"grad_norm": 0.504775881767273,
|
140 |
+
"learning_rate": 4.941453335558682e-06,
|
141 |
+
"loss": 0.8446,
|
142 |
+
"step": 380
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.8080808080808081,
|
146 |
+
"grad_norm": 0.5671610832214355,
|
147 |
+
"learning_rate": 4.9096053963998555e-06,
|
148 |
+
"loss": 0.796,
|
149 |
+
"step": 400
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.8484848484848485,
|
153 |
+
"grad_norm": 0.6207985281944275,
|
154 |
+
"learning_rate": 4.871018828260492e-06,
|
155 |
+
"loss": 0.8135,
|
156 |
+
"step": 420
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.8888888888888888,
|
160 |
+
"grad_norm": 0.6469517350196838,
|
161 |
+
"learning_rate": 4.825801541160509e-06,
|
162 |
+
"loss": 0.8143,
|
163 |
+
"step": 440
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.9292929292929293,
|
167 |
+
"grad_norm": 0.5718827247619629,
|
168 |
+
"learning_rate": 4.7740799883862966e-06,
|
169 |
+
"loss": 0.79,
|
170 |
+
"step": 460
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.9696969696969697,
|
174 |
+
"grad_norm": 0.9612074494361877,
|
175 |
+
"learning_rate": 4.715998812855305e-06,
|
176 |
+
"loss": 0.7133,
|
177 |
+
"step": 480
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 1.0101010101010102,
|
181 |
+
"grad_norm": 0.6769607067108154,
|
182 |
+
"learning_rate": 4.651720442612076e-06,
|
183 |
+
"loss": 0.765,
|
184 |
+
"step": 500
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 1.0505050505050506,
|
188 |
+
"grad_norm": 0.6824682354927063,
|
189 |
+
"learning_rate": 4.5814246365869285e-06,
|
190 |
+
"loss": 0.7051,
|
191 |
+
"step": 520
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 1.0909090909090908,
|
195 |
+
"grad_norm": 0.4644620716571808,
|
196 |
+
"learning_rate": 4.50530798188761e-06,
|
197 |
+
"loss": 0.7534,
|
198 |
+
"step": 540
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 1.1313131313131313,
|
202 |
+
"grad_norm": 0.8265301585197449,
|
203 |
+
"learning_rate": 4.423583344029786e-06,
|
204 |
+
"loss": 0.7873,
|
205 |
+
"step": 560
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 1.1717171717171717,
|
209 |
+
"grad_norm": 0.6619002819061279,
|
210 |
+
"learning_rate": 4.336479271643833e-06,
|
211 |
+
"loss": 0.7264,
|
212 |
+
"step": 580
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 1.2121212121212122,
|
216 |
+
"grad_norm": 0.5203114151954651,
|
217 |
+
"learning_rate": 4.244239357322705e-06,
|
218 |
+
"loss": 0.7754,
|
219 |
+
"step": 600
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 1.2525252525252526,
|
223 |
+
"grad_norm": 0.5513920783996582,
|
224 |
+
"learning_rate": 4.1471215563983125e-06,
|
225 |
+
"loss": 0.694,
|
226 |
+
"step": 620
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 1.2929292929292928,
|
230 |
+
"grad_norm": 0.6095554232597351,
|
231 |
+
"learning_rate": 4.045397465551513e-06,
|
232 |
+
"loss": 0.7402,
|
233 |
+
"step": 640
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 1.3333333333333333,
|
237 |
+
"grad_norm": 0.46472665667533875,
|
238 |
+
"learning_rate": 3.93935156327311e-06,
|
239 |
+
"loss": 0.741,
|
240 |
+
"step": 660
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 1.3737373737373737,
|
244 |
+
"grad_norm": 0.6970565915107727,
|
245 |
+
"learning_rate": 3.82928041429998e-06,
|
246 |
+
"loss": 0.7332,
|
247 |
+
"step": 680
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 1.4141414141414141,
|
251 |
+
"grad_norm": 0.46008697152137756,
|
252 |
+
"learning_rate": 3.715491840251172e-06,
|
253 |
+
"loss": 0.7401,
|
254 |
+
"step": 700
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 1.4545454545454546,
|
258 |
+
"grad_norm": 0.5285646319389343,
|
259 |
+
"learning_rate": 3.598304058783357e-06,
|
260 |
+
"loss": 0.7809,
|
261 |
+
"step": 720
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 1.494949494949495,
|
265 |
+
"grad_norm": 0.46231013536453247,
|
266 |
+
"learning_rate": 3.478044793673025e-06,
|
267 |
+
"loss": 0.7506,
|
268 |
+
"step": 740
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 1.5353535353535355,
|
272 |
+
"grad_norm": 0.5669680833816528,
|
273 |
+
"learning_rate": 3.3550503583141726e-06,
|
274 |
+
"loss": 0.732,
|
275 |
+
"step": 760
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 1.5757575757575757,
|
279 |
+
"grad_norm": 0.6222017407417297,
|
280 |
+
"learning_rate": 3.2296647151945116e-06,
|
281 |
+
"loss": 0.7184,
|
282 |
+
"step": 780
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 1.6161616161616161,
|
286 |
+
"grad_norm": 0.5148311853408813,
|
287 |
+
"learning_rate": 3.102238513980471e-06,
|
288 |
+
"loss": 0.7674,
|
289 |
+
"step": 800
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 1.6565656565656566,
|
293 |
+
"grad_norm": 0.5587109327316284,
|
294 |
+
"learning_rate": 2.973128110901026e-06,
|
295 |
+
"loss": 0.7514,
|
296 |
+
"step": 820
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 1.696969696969697,
|
300 |
+
"grad_norm": 0.6635349988937378,
|
301 |
+
"learning_rate": 2.842694572172737e-06,
|
302 |
+
"loss": 0.7388,
|
303 |
+
"step": 840
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 1.7373737373737375,
|
307 |
+
"grad_norm": 0.5663965344429016,
|
308 |
+
"learning_rate": 2.7113026642529733e-06,
|
309 |
+
"loss": 0.7061,
|
310 |
+
"step": 860
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 1.7777777777777777,
|
314 |
+
"grad_norm": 0.5994731783866882,
|
315 |
+
"learning_rate": 2.57931983374517e-06,
|
316 |
+
"loss": 0.726,
|
317 |
+
"step": 880
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 1.8181818181818183,
|
321 |
+
"grad_norm": 0.5646547079086304,
|
322 |
+
"learning_rate": 2.4471151798088465e-06,
|
323 |
+
"loss": 0.7363,
|
324 |
+
"step": 900
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 1.8585858585858586,
|
328 |
+
"grad_norm": 0.575791597366333,
|
329 |
+
"learning_rate": 2.3150584219481644e-06,
|
330 |
+
"loss": 0.7105,
|
331 |
+
"step": 920
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 1.898989898989899,
|
335 |
+
"grad_norm": 0.43835175037384033,
|
336 |
+
"learning_rate": 2.183518866065627e-06,
|
337 |
+
"loss": 0.688,
|
338 |
+
"step": 940
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 1.9393939393939394,
|
342 |
+
"grad_norm": 0.6634537577629089,
|
343 |
+
"learning_rate": 2.0528643716724572e-06,
|
344 |
+
"loss": 0.62,
|
345 |
+
"step": 960
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 1.9797979797979797,
|
349 |
+
"grad_norm": 0.45726099610328674,
|
350 |
+
"learning_rate": 1.9234603231439e-06,
|
351 |
+
"loss": 0.7451,
|
352 |
+
"step": 980
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 2.0202020202020203,
|
356 |
+
"grad_norm": 0.564659833908081,
|
357 |
+
"learning_rate": 1.7956686078964257e-06,
|
358 |
+
"loss": 0.6724,
|
359 |
+
"step": 1000
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 2.0606060606060606,
|
363 |
+
"grad_norm": 0.6070213317871094,
|
364 |
+
"learning_rate": 1.6698466043444122e-06,
|
365 |
+
"loss": 0.7313,
|
366 |
+
"step": 1020
|
367 |
+
},
|
368 |
+
{
|
369 |
+
"epoch": 2.101010101010101,
|
370 |
+
"grad_norm": 0.5348501801490784,
|
371 |
+
"learning_rate": 1.546346182466566e-06,
|
372 |
+
"loss": 0.7431,
|
373 |
+
"step": 1040
|
374 |
+
},
|
375 |
+
{
|
376 |
+
"epoch": 2.1414141414141414,
|
377 |
+
"grad_norm": 0.5485573410987854,
|
378 |
+
"learning_rate": 1.425512719777071e-06,
|
379 |
+
"loss": 0.7365,
|
380 |
+
"step": 1060
|
381 |
+
},
|
382 |
+
{
|
383 |
+
"epoch": 2.1818181818181817,
|
384 |
+
"grad_norm": 0.5143328905105591,
|
385 |
+
"learning_rate": 1.3076841354533658e-06,
|
386 |
+
"loss": 0.7516,
|
387 |
+
"step": 1080
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"epoch": 2.2222222222222223,
|
391 |
+
"grad_norm": 0.4357577860355377,
|
392 |
+
"learning_rate": 1.1931899453216698e-06,
|
393 |
+
"loss": 0.6963,
|
394 |
+
"step": 1100
|
395 |
+
},
|
396 |
+
{
|
397 |
+
"epoch": 2.2626262626262625,
|
398 |
+
"grad_norm": 0.5489456057548523,
|
399 |
+
"learning_rate": 1.0823503403430736e-06,
|
400 |
+
"loss": 0.6867,
|
401 |
+
"step": 1120
|
402 |
+
},
|
403 |
+
{
|
404 |
+
"epoch": 2.303030303030303,
|
405 |
+
"grad_norm": 0.5727762579917908,
|
406 |
+
"learning_rate": 9.754752911772616e-07,
|
407 |
+
"loss": 0.6873,
|
408 |
+
"step": 1140
|
409 |
+
},
|
410 |
+
{
|
411 |
+
"epoch": 2.3434343434343434,
|
412 |
+
"grad_norm": 0.469957172870636,
|
413 |
+
"learning_rate": 8.728636813280164e-07,
|
414 |
+
"loss": 0.691,
|
415 |
+
"step": 1160
|
416 |
+
},
|
417 |
+
{
|
418 |
+
"epoch": 2.3838383838383836,
|
419 |
+
"grad_norm": 0.7846599817276001,
|
420 |
+
"learning_rate": 7.748024712947205e-07,
|
421 |
+
"loss": 0.7064,
|
422 |
+
"step": 1180
|
423 |
+
},
|
424 |
+
{
|
425 |
+
"epoch": 2.4242424242424243,
|
426 |
+
"grad_norm": 0.5549547076225281,
|
427 |
+
"learning_rate": 6.815658960673782e-07,
|
428 |
+
"loss": 0.7707,
|
429 |
+
"step": 1200
|
430 |
+
},
|
431 |
+
{
|
432 |
+
"epoch": 2.4646464646464645,
|
433 |
+
"grad_norm": 0.4853402078151703,
|
434 |
+
"learning_rate": 5.934146982094049e-07,
|
435 |
+
"loss": 0.7368,
|
436 |
+
"step": 1220
|
437 |
+
},
|
438 |
+
{
|
439 |
+
"epoch": 2.505050505050505,
|
440 |
+
"grad_norm": 0.6494936943054199,
|
441 |
+
"learning_rate": 5.105953986729196e-07,
|
442 |
+
"loss": 0.7046,
|
443 |
+
"step": 1240
|
444 |
+
},
|
445 |
+
{
|
446 |
+
"epoch": 2.5454545454545454,
|
447 |
+
"grad_norm": 0.48534753918647766,
|
448 |
+
"learning_rate": 4.3333960738577236e-07,
|
449 |
+
"loss": 0.6873,
|
450 |
+
"step": 1260
|
451 |
+
},
|
452 |
+
{
|
453 |
+
"epoch": 2.5858585858585856,
|
454 |
+
"grad_norm": 0.6183688640594482,
|
455 |
+
"learning_rate": 3.6186337553827747e-07,
|
456 |
+
"loss": 0.7095,
|
457 |
+
"step": 1280
|
458 |
+
},
|
459 |
+
{
|
460 |
+
"epoch": 2.6262626262626263,
|
461 |
+
"grad_norm": 0.5592140555381775,
|
462 |
+
"learning_rate": 2.963665913810451e-07,
|
463 |
+
"loss": 0.6618,
|
464 |
+
"step": 1300
|
465 |
+
},
|
466 |
+
{
|
467 |
+
"epoch": 2.6666666666666665,
|
468 |
+
"grad_norm": 0.5510610938072205,
|
469 |
+
"learning_rate": 2.370324212235936e-07,
|
470 |
+
"loss": 0.7339,
|
471 |
+
"step": 1320
|
472 |
+
},
|
473 |
+
{
|
474 |
+
"epoch": 2.707070707070707,
|
475 |
+
"grad_norm": 0.6054675579071045,
|
476 |
+
"learning_rate": 1.840267971970344e-07,
|
477 |
+
"loss": 0.7237,
|
478 |
+
"step": 1340
|
479 |
+
},
|
480 |
+
{
|
481 |
+
"epoch": 2.7474747474747474,
|
482 |
+
"grad_norm": 0.6097862720489502,
|
483 |
+
"learning_rate": 1.3749795321332887e-07,
|
484 |
+
"loss": 0.7374,
|
485 |
+
"step": 1360
|
486 |
+
},
|
487 |
+
{
|
488 |
+
"epoch": 2.787878787878788,
|
489 |
+
"grad_norm": 0.4065544605255127,
|
490 |
+
"learning_rate": 9.757601041885694e-08,
|
491 |
+
"loss": 0.7664,
|
492 |
+
"step": 1380
|
493 |
+
},
|
494 |
+
{
|
495 |
+
"epoch": 2.8282828282828283,
|
496 |
+
"grad_norm": 0.573664665222168,
|
497 |
+
"learning_rate": 6.437261330158206e-08,
|
498 |
+
"loss": 0.7463,
|
499 |
+
"step": 1400
|
500 |
+
},
|
501 |
+
{
|
502 |
+
"epoch": 2.8686868686868685,
|
503 |
+
"grad_norm": 0.47344183921813965,
|
504 |
+
"learning_rate": 3.798061746947995e-08,
|
505 |
+
"loss": 0.6859,
|
506 |
+
"step": 1420
|
507 |
+
},
|
508 |
+
{
|
509 |
+
"epoch": 2.909090909090909,
|
510 |
+
"grad_norm": 0.48462924361228943,
|
511 |
+
"learning_rate": 1.847382997337943e-08,
|
512 |
+
"loss": 0.659,
|
513 |
+
"step": 1440
|
514 |
+
},
|
515 |
+
{
|
516 |
+
"epoch": 2.9494949494949494,
|
517 |
+
"grad_norm": 0.6021179556846619,
|
518 |
+
"learning_rate": 5.906802900412789e-09,
|
519 |
+
"loss": 0.6971,
|
520 |
+
"step": 1460
|
521 |
+
},
|
522 |
+
{
|
523 |
+
"epoch": 2.98989898989899,
|
524 |
+
"grad_norm": 0.5301674008369446,
|
525 |
+
"learning_rate": 3.146808153123293e-10,
|
526 |
+
"loss": 0.6606,
|
527 |
+
"step": 1480
|
528 |
+
},
|
529 |
+
{
|
530 |
+
"epoch": 3.0,
|
531 |
+
"step": 1485,
|
532 |
+
"total_flos": 6.836552056302797e+16,
|
533 |
+
"train_loss": 0.8828093713381475,
|
534 |
+
"train_runtime": 1774.0294,
|
535 |
+
"train_samples_per_second": 3.348,
|
536 |
+
"train_steps_per_second": 0.837
|
537 |
+
}
|
538 |
+
],
|
539 |
+
"logging_steps": 20,
|
540 |
+
"max_steps": 1485,
|
541 |
+
"num_input_tokens_seen": 0,
|
542 |
+
"num_train_epochs": 3,
|
543 |
+
"save_steps": 100,
|
544 |
+
"stateful_callbacks": {
|
545 |
+
"TrainerControl": {
|
546 |
+
"args": {
|
547 |
+
"should_epoch_stop": false,
|
548 |
+
"should_evaluate": false,
|
549 |
+
"should_log": false,
|
550 |
+
"should_save": true,
|
551 |
+
"should_training_stop": true
|
552 |
+
},
|
553 |
+
"attributes": {}
|
554 |
+
}
|
555 |
+
},
|
556 |
+
"total_flos": 6.836552056302797e+16,
|
557 |
+
"train_batch_size": 4,
|
558 |
+
"trial_name": null,
|
559 |
+
"trial_params": null
|
560 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bb5c68933c56cd908223dee0b16725da2ff2723941dc413276a144aaca8594d2
|
3 |
+
size 5624
|