Upload folder using huggingface_hub
Browse files- .gitattributes +1 -0
- README.md +61 -0
- added_tokens.json +24 -0
- all_results.json +8 -0
- config.json +29 -0
- generation_config.json +14 -0
- merges.txt +0 -0
- model.safetensors +3 -0
- special_tokens_map.json +31 -0
- tokenizer.json +3 -0
- tokenizer_config.json +209 -0
- train_results.json +8 -0
- trainer_log.jsonl +51 -0
- trainer_state.json +392 -0
- training_args.bin +3 -0
- training_loss.png +0 -0
- vocab.json +0 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: transformers
|
3 |
+
license: other
|
4 |
+
base_model: jdqqjr/Qwen2.5-0.5B-Open-R1-Distill
|
5 |
+
tags:
|
6 |
+
- llama-factory
|
7 |
+
- full
|
8 |
+
- generated_from_trainer
|
9 |
+
model-index:
|
10 |
+
- name: Qwen2.5-0.5B-Open-R1-Distill-FactThink-SFT
|
11 |
+
results: []
|
12 |
+
---
|
13 |
+
|
14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
15 |
+
should probably proofread and complete it, then remove this comment. -->
|
16 |
+
|
17 |
+
# Qwen2.5-0.5B-Open-R1-Distill-FactThink-SFT
|
18 |
+
|
19 |
+
This model is a fine-tuned version of [jdqqjr/Qwen2.5-0.5B-Open-R1-Distill](https://huggingface.co/jdqqjr/Qwen2.5-0.5B-Open-R1-Distill) on the cot_fact_think_sft_train dataset.
|
20 |
+
|
21 |
+
## Model description
|
22 |
+
|
23 |
+
More information needed
|
24 |
+
|
25 |
+
## Intended uses & limitations
|
26 |
+
|
27 |
+
More information needed
|
28 |
+
|
29 |
+
## Training and evaluation data
|
30 |
+
|
31 |
+
More information needed
|
32 |
+
|
33 |
+
## Training procedure
|
34 |
+
|
35 |
+
### Training hyperparameters
|
36 |
+
|
37 |
+
The following hyperparameters were used during training:
|
38 |
+
- learning_rate: 0.0001
|
39 |
+
- train_batch_size: 2
|
40 |
+
- eval_batch_size: 8
|
41 |
+
- seed: 42
|
42 |
+
- distributed_type: multi-GPU
|
43 |
+
- num_devices: 4
|
44 |
+
- gradient_accumulation_steps: 6
|
45 |
+
- total_train_batch_size: 48
|
46 |
+
- total_eval_batch_size: 32
|
47 |
+
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
48 |
+
- lr_scheduler_type: cosine
|
49 |
+
- num_epochs: 4.0
|
50 |
+
- mixed_precision_training: Native AMP
|
51 |
+
|
52 |
+
### Training results
|
53 |
+
|
54 |
+
|
55 |
+
|
56 |
+
### Framework versions
|
57 |
+
|
58 |
+
- Transformers 4.47.0
|
59 |
+
- Pytorch 2.3.1+cu121
|
60 |
+
- Datasets 2.20.0
|
61 |
+
- Tokenizers 0.21.0
|
added_tokens.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"</tool_call>": 151658,
|
3 |
+
"<tool_call>": 151657,
|
4 |
+
"<|box_end|>": 151649,
|
5 |
+
"<|box_start|>": 151648,
|
6 |
+
"<|endoftext|>": 151643,
|
7 |
+
"<|file_sep|>": 151664,
|
8 |
+
"<|fim_middle|>": 151660,
|
9 |
+
"<|fim_pad|>": 151662,
|
10 |
+
"<|fim_prefix|>": 151659,
|
11 |
+
"<|fim_suffix|>": 151661,
|
12 |
+
"<|im_end|>": 151645,
|
13 |
+
"<|im_start|>": 151644,
|
14 |
+
"<|image_pad|>": 151655,
|
15 |
+
"<|object_ref_end|>": 151647,
|
16 |
+
"<|object_ref_start|>": 151646,
|
17 |
+
"<|quad_end|>": 151651,
|
18 |
+
"<|quad_start|>": 151650,
|
19 |
+
"<|repo_name|>": 151663,
|
20 |
+
"<|video_pad|>": 151656,
|
21 |
+
"<|vision_end|>": 151653,
|
22 |
+
"<|vision_pad|>": 151654,
|
23 |
+
"<|vision_start|>": 151652
|
24 |
+
}
|
all_results.json
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 4.0,
|
3 |
+
"total_flos": 4.036279852125389e+16,
|
4 |
+
"train_loss": 7.3980213584899905,
|
5 |
+
"train_runtime": 1227.9067,
|
6 |
+
"train_samples_per_second": 19.542,
|
7 |
+
"train_steps_per_second": 0.407
|
8 |
+
}
|
config.json
ADDED
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "jdqqjr/Qwen2.5-0.5B-Open-R1-Distill",
|
3 |
+
"architectures": [
|
4 |
+
"Qwen2ForCausalLM"
|
5 |
+
],
|
6 |
+
"attention_dropout": 0.0,
|
7 |
+
"bos_token_id": 151643,
|
8 |
+
"eos_token_id": 151645,
|
9 |
+
"hidden_act": "silu",
|
10 |
+
"hidden_size": 896,
|
11 |
+
"initializer_range": 0.02,
|
12 |
+
"intermediate_size": 4864,
|
13 |
+
"max_position_embeddings": 32768,
|
14 |
+
"max_window_layers": 21,
|
15 |
+
"model_type": "qwen2",
|
16 |
+
"num_attention_heads": 14,
|
17 |
+
"num_hidden_layers": 24,
|
18 |
+
"num_key_value_heads": 2,
|
19 |
+
"rms_norm_eps": 1e-06,
|
20 |
+
"rope_scaling": null,
|
21 |
+
"rope_theta": 1000000.0,
|
22 |
+
"sliding_window": null,
|
23 |
+
"tie_word_embeddings": true,
|
24 |
+
"torch_dtype": "float32",
|
25 |
+
"transformers_version": "4.47.0",
|
26 |
+
"use_cache": false,
|
27 |
+
"use_sliding_window": false,
|
28 |
+
"vocab_size": 151936
|
29 |
+
}
|
generation_config.json
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token_id": 151643,
|
3 |
+
"do_sample": true,
|
4 |
+
"eos_token_id": [
|
5 |
+
151645,
|
6 |
+
151643
|
7 |
+
],
|
8 |
+
"pad_token_id": 151643,
|
9 |
+
"repetition_penalty": 1.1,
|
10 |
+
"temperature": 0.7,
|
11 |
+
"top_k": 20,
|
12 |
+
"top_p": 0.8,
|
13 |
+
"transformers_version": "4.47.0"
|
14 |
+
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc2713383ea3d47c46b4bca480cd6c813a9808286819fd333cdf0df2ef94b722
|
3 |
+
size 1976163472
|
special_tokens_map.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|im_start|>",
|
4 |
+
"<|im_end|>",
|
5 |
+
"<|object_ref_start|>",
|
6 |
+
"<|object_ref_end|>",
|
7 |
+
"<|box_start|>",
|
8 |
+
"<|box_end|>",
|
9 |
+
"<|quad_start|>",
|
10 |
+
"<|quad_end|>",
|
11 |
+
"<|vision_start|>",
|
12 |
+
"<|vision_end|>",
|
13 |
+
"<|vision_pad|>",
|
14 |
+
"<|image_pad|>",
|
15 |
+
"<|video_pad|>"
|
16 |
+
],
|
17 |
+
"eos_token": {
|
18 |
+
"content": "<|im_end|>",
|
19 |
+
"lstrip": false,
|
20 |
+
"normalized": false,
|
21 |
+
"rstrip": false,
|
22 |
+
"single_word": false
|
23 |
+
},
|
24 |
+
"pad_token": {
|
25 |
+
"content": "<|im_end|>",
|
26 |
+
"lstrip": false,
|
27 |
+
"normalized": false,
|
28 |
+
"rstrip": false,
|
29 |
+
"single_word": false
|
30 |
+
}
|
31 |
+
}
|
tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9c5ae00e602b8860cbd784ba82a8aa14e8feecec692e7076590d014d7b7fdafa
|
3 |
+
size 11421896
|
tokenizer_config.json
ADDED
@@ -0,0 +1,209 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_prefix_space": false,
|
4 |
+
"added_tokens_decoder": {
|
5 |
+
"151643": {
|
6 |
+
"content": "<|endoftext|>",
|
7 |
+
"lstrip": false,
|
8 |
+
"normalized": false,
|
9 |
+
"rstrip": false,
|
10 |
+
"single_word": false,
|
11 |
+
"special": true
|
12 |
+
},
|
13 |
+
"151644": {
|
14 |
+
"content": "<|im_start|>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false,
|
19 |
+
"special": true
|
20 |
+
},
|
21 |
+
"151645": {
|
22 |
+
"content": "<|im_end|>",
|
23 |
+
"lstrip": false,
|
24 |
+
"normalized": false,
|
25 |
+
"rstrip": false,
|
26 |
+
"single_word": false,
|
27 |
+
"special": true
|
28 |
+
},
|
29 |
+
"151646": {
|
30 |
+
"content": "<|object_ref_start|>",
|
31 |
+
"lstrip": false,
|
32 |
+
"normalized": false,
|
33 |
+
"rstrip": false,
|
34 |
+
"single_word": false,
|
35 |
+
"special": true
|
36 |
+
},
|
37 |
+
"151647": {
|
38 |
+
"content": "<|object_ref_end|>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false,
|
43 |
+
"special": true
|
44 |
+
},
|
45 |
+
"151648": {
|
46 |
+
"content": "<|box_start|>",
|
47 |
+
"lstrip": false,
|
48 |
+
"normalized": false,
|
49 |
+
"rstrip": false,
|
50 |
+
"single_word": false,
|
51 |
+
"special": true
|
52 |
+
},
|
53 |
+
"151649": {
|
54 |
+
"content": "<|box_end|>",
|
55 |
+
"lstrip": false,
|
56 |
+
"normalized": false,
|
57 |
+
"rstrip": false,
|
58 |
+
"single_word": false,
|
59 |
+
"special": true
|
60 |
+
},
|
61 |
+
"151650": {
|
62 |
+
"content": "<|quad_start|>",
|
63 |
+
"lstrip": false,
|
64 |
+
"normalized": false,
|
65 |
+
"rstrip": false,
|
66 |
+
"single_word": false,
|
67 |
+
"special": true
|
68 |
+
},
|
69 |
+
"151651": {
|
70 |
+
"content": "<|quad_end|>",
|
71 |
+
"lstrip": false,
|
72 |
+
"normalized": false,
|
73 |
+
"rstrip": false,
|
74 |
+
"single_word": false,
|
75 |
+
"special": true
|
76 |
+
},
|
77 |
+
"151652": {
|
78 |
+
"content": "<|vision_start|>",
|
79 |
+
"lstrip": false,
|
80 |
+
"normalized": false,
|
81 |
+
"rstrip": false,
|
82 |
+
"single_word": false,
|
83 |
+
"special": true
|
84 |
+
},
|
85 |
+
"151653": {
|
86 |
+
"content": "<|vision_end|>",
|
87 |
+
"lstrip": false,
|
88 |
+
"normalized": false,
|
89 |
+
"rstrip": false,
|
90 |
+
"single_word": false,
|
91 |
+
"special": true
|
92 |
+
},
|
93 |
+
"151654": {
|
94 |
+
"content": "<|vision_pad|>",
|
95 |
+
"lstrip": false,
|
96 |
+
"normalized": false,
|
97 |
+
"rstrip": false,
|
98 |
+
"single_word": false,
|
99 |
+
"special": true
|
100 |
+
},
|
101 |
+
"151655": {
|
102 |
+
"content": "<|image_pad|>",
|
103 |
+
"lstrip": false,
|
104 |
+
"normalized": false,
|
105 |
+
"rstrip": false,
|
106 |
+
"single_word": false,
|
107 |
+
"special": true
|
108 |
+
},
|
109 |
+
"151656": {
|
110 |
+
"content": "<|video_pad|>",
|
111 |
+
"lstrip": false,
|
112 |
+
"normalized": false,
|
113 |
+
"rstrip": false,
|
114 |
+
"single_word": false,
|
115 |
+
"special": true
|
116 |
+
},
|
117 |
+
"151657": {
|
118 |
+
"content": "<tool_call>",
|
119 |
+
"lstrip": false,
|
120 |
+
"normalized": false,
|
121 |
+
"rstrip": false,
|
122 |
+
"single_word": false,
|
123 |
+
"special": false
|
124 |
+
},
|
125 |
+
"151658": {
|
126 |
+
"content": "</tool_call>",
|
127 |
+
"lstrip": false,
|
128 |
+
"normalized": false,
|
129 |
+
"rstrip": false,
|
130 |
+
"single_word": false,
|
131 |
+
"special": false
|
132 |
+
},
|
133 |
+
"151659": {
|
134 |
+
"content": "<|fim_prefix|>",
|
135 |
+
"lstrip": false,
|
136 |
+
"normalized": false,
|
137 |
+
"rstrip": false,
|
138 |
+
"single_word": false,
|
139 |
+
"special": false
|
140 |
+
},
|
141 |
+
"151660": {
|
142 |
+
"content": "<|fim_middle|>",
|
143 |
+
"lstrip": false,
|
144 |
+
"normalized": false,
|
145 |
+
"rstrip": false,
|
146 |
+
"single_word": false,
|
147 |
+
"special": false
|
148 |
+
},
|
149 |
+
"151661": {
|
150 |
+
"content": "<|fim_suffix|>",
|
151 |
+
"lstrip": false,
|
152 |
+
"normalized": false,
|
153 |
+
"rstrip": false,
|
154 |
+
"single_word": false,
|
155 |
+
"special": false
|
156 |
+
},
|
157 |
+
"151662": {
|
158 |
+
"content": "<|fim_pad|>",
|
159 |
+
"lstrip": false,
|
160 |
+
"normalized": false,
|
161 |
+
"rstrip": false,
|
162 |
+
"single_word": false,
|
163 |
+
"special": false
|
164 |
+
},
|
165 |
+
"151663": {
|
166 |
+
"content": "<|repo_name|>",
|
167 |
+
"lstrip": false,
|
168 |
+
"normalized": false,
|
169 |
+
"rstrip": false,
|
170 |
+
"single_word": false,
|
171 |
+
"special": false
|
172 |
+
},
|
173 |
+
"151664": {
|
174 |
+
"content": "<|file_sep|>",
|
175 |
+
"lstrip": false,
|
176 |
+
"normalized": false,
|
177 |
+
"rstrip": false,
|
178 |
+
"single_word": false,
|
179 |
+
"special": false
|
180 |
+
}
|
181 |
+
},
|
182 |
+
"additional_special_tokens": [
|
183 |
+
"<|im_start|>",
|
184 |
+
"<|im_end|>",
|
185 |
+
"<|object_ref_start|>",
|
186 |
+
"<|object_ref_end|>",
|
187 |
+
"<|box_start|>",
|
188 |
+
"<|box_end|>",
|
189 |
+
"<|quad_start|>",
|
190 |
+
"<|quad_end|>",
|
191 |
+
"<|vision_start|>",
|
192 |
+
"<|vision_end|>",
|
193 |
+
"<|vision_pad|>",
|
194 |
+
"<|image_pad|>",
|
195 |
+
"<|video_pad|>"
|
196 |
+
],
|
197 |
+
"bos_token": null,
|
198 |
+
"chat_template": "{% set system_message = 'Your role as an assistant involves thoroughly exploring questions through a systematic long thinking process before providing the final precise and accurate solutions. This requires engaging in a comprehensive cycle of analysis, summarizing, exploration, reassessment, reflection, backtracing, and iteration to develop well-considered thinking process. Please structure your response into two main sections: Thought and Solution. In the Thought section, detail your reasoning process using the specified format: <think> {thought with steps separated with \\'\\n\\n\\'} </think> Each step should include detailed considerations such as analysing questions, summarizing relevant findings, brainstorming new ideas, verifying the accuracy of the current steps, refining any errors, and revisiting previous steps. In the Solution section, based on various attempts, explorations, and reflections from the Thought section, systematically present the final solution that you deem correct. The solution should remain a logical, accurate, concise expression style and detail necessary step needed to reach the conclusion, formatted as follows: <answer> {final formatted, precise, and clear solution} </answer> Now, try to solve the following question through the above guidelines:' %}{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% endif %}{% if system_message is defined %}{{ '<|im_start|>system\\n' + system_message + '<|im_end|>\\n' }}{% endif %}{% for message in messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ '<|im_start|>user\\n' + content + '<|im_end|>\\n<|im_start|>assistant\\n' }}{% elif message['role'] == 'assistant' %}{{ content + '<|im_end|>\\n' + '\\n' }}{% endif %}{% endfor %}",
|
199 |
+
"clean_up_tokenization_spaces": false,
|
200 |
+
"eos_token": "<|im_end|>",
|
201 |
+
"errors": "replace",
|
202 |
+
"extra_special_tokens": {},
|
203 |
+
"model_max_length": 131072,
|
204 |
+
"pad_token": "<|im_end|>",
|
205 |
+
"padding_side": "right",
|
206 |
+
"split_special_tokens": false,
|
207 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
208 |
+
"unk_token": null
|
209 |
+
}
|
train_results.json
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 4.0,
|
3 |
+
"total_flos": 4.036279852125389e+16,
|
4 |
+
"train_loss": 7.3980213584899905,
|
5 |
+
"train_runtime": 1227.9067,
|
6 |
+
"train_samples_per_second": 19.542,
|
7 |
+
"train_steps_per_second": 0.407
|
8 |
+
}
|
trainer_log.jsonl
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{"current_steps": 10, "total_steps": 500, "loss": 16.6271, "learning_rate": 9.996447363202946e-05, "epoch": 0.08, "percentage": 2.0, "elapsed_time": "0:00:24", "remaining_time": "0:19:39"}
|
2 |
+
{"current_steps": 20, "total_steps": 500, "loss": 13.6217, "learning_rate": 9.977809823015401e-05, "epoch": 0.16, "percentage": 4.0, "elapsed_time": "0:00:47", "remaining_time": "0:19:02"}
|
3 |
+
{"current_steps": 30, "total_steps": 500, "loss": 12.3656, "learning_rate": 9.938441702975689e-05, "epoch": 0.24, "percentage": 6.0, "elapsed_time": "0:01:11", "remaining_time": "0:18:33"}
|
4 |
+
{"current_steps": 40, "total_steps": 500, "loss": 12.2646, "learning_rate": 9.879583809693738e-05, "epoch": 0.32, "percentage": 8.0, "elapsed_time": "0:01:35", "remaining_time": "0:18:12"}
|
5 |
+
{"current_steps": 50, "total_steps": 500, "loss": 12.1808, "learning_rate": 9.801468428384716e-05, "epoch": 0.4, "percentage": 10.0, "elapsed_time": "0:01:58", "remaining_time": "0:17:49"}
|
6 |
+
{"current_steps": 60, "total_steps": 500, "loss": 12.1502, "learning_rate": 9.704403844771128e-05, "epoch": 0.48, "percentage": 12.0, "elapsed_time": "0:02:22", "remaining_time": "0:17:26"}
|
7 |
+
{"current_steps": 70, "total_steps": 500, "loss": 11.9683, "learning_rate": 9.588773128419906e-05, "epoch": 0.56, "percentage": 14.0, "elapsed_time": "0:02:46", "remaining_time": "0:17:05"}
|
8 |
+
{"current_steps": 80, "total_steps": 500, "loss": 11.8564, "learning_rate": 9.45503262094184e-05, "epoch": 0.64, "percentage": 16.0, "elapsed_time": "0:03:10", "remaining_time": "0:16:41"}
|
9 |
+
{"current_steps": 90, "total_steps": 500, "loss": 12.0212, "learning_rate": 9.30371013501972e-05, "epoch": 0.72, "percentage": 18.0, "elapsed_time": "0:03:35", "remaining_time": "0:16:20"}
|
10 |
+
{"current_steps": 100, "total_steps": 500, "loss": 11.8631, "learning_rate": 9.135402871372808e-05, "epoch": 0.8, "percentage": 20.0, "elapsed_time": "0:03:59", "remaining_time": "0:15:56"}
|
11 |
+
{"current_steps": 110, "total_steps": 500, "loss": 11.6954, "learning_rate": 8.950775061878453e-05, "epoch": 0.88, "percentage": 22.0, "elapsed_time": "0:04:23", "remaining_time": "0:15:33"}
|
12 |
+
{"current_steps": 120, "total_steps": 500, "loss": 11.7669, "learning_rate": 8.750555348152298e-05, "epoch": 0.96, "percentage": 24.0, "elapsed_time": "0:04:47", "remaining_time": "0:15:10"}
|
13 |
+
{"current_steps": 130, "total_steps": 500, "loss": 10.3003, "learning_rate": 8.535533905932738e-05, "epoch": 1.04, "percentage": 26.0, "elapsed_time": "0:05:11", "remaining_time": "0:14:46"}
|
14 |
+
{"current_steps": 140, "total_steps": 500, "loss": 8.5731, "learning_rate": 8.306559326618259e-05, "epoch": 1.12, "percentage": 28.0, "elapsed_time": "0:05:35", "remaining_time": "0:14:23"}
|
15 |
+
{"current_steps": 150, "total_steps": 500, "loss": 8.5948, "learning_rate": 8.064535268264883e-05, "epoch": 1.2, "percentage": 30.0, "elapsed_time": "0:06:00", "remaining_time": "0:14:00"}
|
16 |
+
{"current_steps": 160, "total_steps": 500, "loss": 8.656, "learning_rate": 7.810416889260653e-05, "epoch": 1.28, "percentage": 32.0, "elapsed_time": "0:06:24", "remaining_time": "0:13:37"}
|
17 |
+
{"current_steps": 170, "total_steps": 500, "loss": 8.7269, "learning_rate": 7.545207078751857e-05, "epoch": 1.3599999999999999, "percentage": 34.0, "elapsed_time": "0:06:49", "remaining_time": "0:13:14"}
|
18 |
+
{"current_steps": 180, "total_steps": 500, "loss": 8.6868, "learning_rate": 7.269952498697734e-05, "epoch": 1.44, "percentage": 36.0, "elapsed_time": "0:07:13", "remaining_time": "0:12:50"}
|
19 |
+
{"current_steps": 190, "total_steps": 500, "loss": 8.7702, "learning_rate": 6.985739453173903e-05, "epoch": 1.52, "percentage": 38.0, "elapsed_time": "0:07:37", "remaining_time": "0:12:26"}
|
20 |
+
{"current_steps": 200, "total_steps": 500, "loss": 8.7521, "learning_rate": 6.693689601226458e-05, "epoch": 1.6, "percentage": 40.0, "elapsed_time": "0:08:01", "remaining_time": "0:12:02"}
|
21 |
+
{"current_steps": 210, "total_steps": 500, "loss": 8.8742, "learning_rate": 6.394955530196147e-05, "epoch": 1.6800000000000002, "percentage": 42.0, "elapsed_time": "0:08:25", "remaining_time": "0:11:38"}
|
22 |
+
{"current_steps": 220, "total_steps": 500, "loss": 8.7707, "learning_rate": 6.090716206982714e-05, "epoch": 1.76, "percentage": 44.0, "elapsed_time": "0:08:49", "remaining_time": "0:11:14"}
|
23 |
+
{"current_steps": 230, "total_steps": 500, "loss": 8.7433, "learning_rate": 5.782172325201155e-05, "epoch": 1.8399999999999999, "percentage": 46.0, "elapsed_time": "0:09:13", "remaining_time": "0:10:50"}
|
24 |
+
{"current_steps": 240, "total_steps": 500, "loss": 8.7179, "learning_rate": 5.470541566592573e-05, "epoch": 1.92, "percentage": 48.0, "elapsed_time": "0:09:38", "remaining_time": "0:10:26"}
|
25 |
+
{"current_steps": 250, "total_steps": 500, "loss": 8.6486, "learning_rate": 5.157053795390642e-05, "epoch": 2.0, "percentage": 50.0, "elapsed_time": "0:10:02", "remaining_time": "0:10:02"}
|
26 |
+
{"current_steps": 260, "total_steps": 500, "loss": 5.591, "learning_rate": 4.8429462046093585e-05, "epoch": 2.08, "percentage": 52.0, "elapsed_time": "0:10:26", "remaining_time": "0:09:38"}
|
27 |
+
{"current_steps": 270, "total_steps": 500, "loss": 5.4084, "learning_rate": 4.529458433407429e-05, "epoch": 2.16, "percentage": 54.0, "elapsed_time": "0:10:50", "remaining_time": "0:09:14"}
|
28 |
+
{"current_steps": 280, "total_steps": 500, "loss": 5.3605, "learning_rate": 4.2178276747988446e-05, "epoch": 2.24, "percentage": 56.0, "elapsed_time": "0:11:14", "remaining_time": "0:08:50"}
|
29 |
+
{"current_steps": 290, "total_steps": 500, "loss": 5.3401, "learning_rate": 3.9092837930172884e-05, "epoch": 2.32, "percentage": 58.0, "elapsed_time": "0:11:39", "remaining_time": "0:08:26"}
|
30 |
+
{"current_steps": 300, "total_steps": 500, "loss": 5.3064, "learning_rate": 3.605044469803854e-05, "epoch": 2.4, "percentage": 60.0, "elapsed_time": "0:12:03", "remaining_time": "0:08:02"}
|
31 |
+
{"current_steps": 310, "total_steps": 500, "loss": 5.3187, "learning_rate": 3.3063103987735433e-05, "epoch": 2.48, "percentage": 62.0, "elapsed_time": "0:12:27", "remaining_time": "0:07:38"}
|
32 |
+
{"current_steps": 320, "total_steps": 500, "loss": 5.413, "learning_rate": 3.0142605468260978e-05, "epoch": 2.56, "percentage": 64.0, "elapsed_time": "0:12:52", "remaining_time": "0:07:14"}
|
33 |
+
{"current_steps": 330, "total_steps": 500, "loss": 5.2867, "learning_rate": 2.7300475013022663e-05, "epoch": 2.64, "percentage": 66.0, "elapsed_time": "0:13:16", "remaining_time": "0:06:50"}
|
34 |
+
{"current_steps": 340, "total_steps": 500, "loss": 5.3684, "learning_rate": 2.4547929212481435e-05, "epoch": 2.7199999999999998, "percentage": 68.0, "elapsed_time": "0:13:40", "remaining_time": "0:06:26"}
|
35 |
+
{"current_steps": 350, "total_steps": 500, "loss": 5.3484, "learning_rate": 2.1895831107393484e-05, "epoch": 2.8, "percentage": 70.0, "elapsed_time": "0:14:05", "remaining_time": "0:06:02"}
|
36 |
+
{"current_steps": 360, "total_steps": 500, "loss": 5.2971, "learning_rate": 1.9354647317351188e-05, "epoch": 2.88, "percentage": 72.0, "elapsed_time": "0:14:29", "remaining_time": "0:05:38"}
|
37 |
+
{"current_steps": 370, "total_steps": 500, "loss": 5.2595, "learning_rate": 1.6934406733817414e-05, "epoch": 2.96, "percentage": 74.0, "elapsed_time": "0:14:53", "remaining_time": "0:05:14"}
|
38 |
+
{"current_steps": 380, "total_steps": 500, "loss": 4.2182, "learning_rate": 1.4644660940672627e-05, "epoch": 3.04, "percentage": 76.0, "elapsed_time": "0:15:18", "remaining_time": "0:04:50"}
|
39 |
+
{"current_steps": 390, "total_steps": 500, "loss": 3.1071, "learning_rate": 1.2494446518477022e-05, "epoch": 3.12, "percentage": 78.0, "elapsed_time": "0:15:43", "remaining_time": "0:04:25"}
|
40 |
+
{"current_steps": 400, "total_steps": 500, "loss": 3.068, "learning_rate": 1.049224938121548e-05, "epoch": 3.2, "percentage": 80.0, "elapsed_time": "0:16:07", "remaining_time": "0:04:01"}
|
41 |
+
{"current_steps": 410, "total_steps": 500, "loss": 2.9727, "learning_rate": 8.645971286271904e-06, "epoch": 3.2800000000000002, "percentage": 82.0, "elapsed_time": "0:16:31", "remaining_time": "0:03:37"}
|
42 |
+
{"current_steps": 420, "total_steps": 500, "loss": 2.9938, "learning_rate": 6.962898649802823e-06, "epoch": 3.36, "percentage": 84.0, "elapsed_time": "0:16:55", "remaining_time": "0:03:13"}
|
43 |
+
{"current_steps": 430, "total_steps": 500, "loss": 3.0744, "learning_rate": 5.449673790581611e-06, "epoch": 3.44, "percentage": 86.0, "elapsed_time": "0:17:20", "remaining_time": "0:02:49"}
|
44 |
+
{"current_steps": 440, "total_steps": 500, "loss": 2.9339, "learning_rate": 4.112268715800943e-06, "epoch": 3.52, "percentage": 88.0, "elapsed_time": "0:17:44", "remaining_time": "0:02:25"}
|
45 |
+
{"current_steps": 450, "total_steps": 500, "loss": 2.9912, "learning_rate": 2.9559615522887273e-06, "epoch": 3.6, "percentage": 90.0, "elapsed_time": "0:18:09", "remaining_time": "0:02:01"}
|
46 |
+
{"current_steps": 460, "total_steps": 500, "loss": 2.9679, "learning_rate": 1.985315716152847e-06, "epoch": 3.68, "percentage": 92.0, "elapsed_time": "0:18:33", "remaining_time": "0:01:36"}
|
47 |
+
{"current_steps": 470, "total_steps": 500, "loss": 3.0297, "learning_rate": 1.2041619030626284e-06, "epoch": 3.76, "percentage": 94.0, "elapsed_time": "0:18:57", "remaining_time": "0:01:12"}
|
48 |
+
{"current_steps": 480, "total_steps": 500, "loss": 2.9479, "learning_rate": 6.15582970243117e-07, "epoch": 3.84, "percentage": 96.0, "elapsed_time": "0:19:22", "remaining_time": "0:00:48"}
|
49 |
+
{"current_steps": 490, "total_steps": 500, "loss": 3.0567, "learning_rate": 2.219017698460002e-07, "epoch": 3.92, "percentage": 98.0, "elapsed_time": "0:19:46", "remaining_time": "0:00:24"}
|
50 |
+
{"current_steps": 500, "total_steps": 500, "loss": 3.0452, "learning_rate": 2.467198171342e-08, "epoch": 4.0, "percentage": 100.0, "elapsed_time": "0:20:10", "remaining_time": "0:00:00"}
|
51 |
+
{"current_steps": 500, "total_steps": 500, "epoch": 4.0, "percentage": 100.0, "elapsed_time": "0:20:20", "remaining_time": "0:00:00"}
|
trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 4.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 500,
|
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.08,
|
13 |
+
"grad_norm": 114.14533233642578,
|
14 |
+
"learning_rate": 9.996447363202946e-05,
|
15 |
+
"loss": 16.6271,
|
16 |
+
"step": 10
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.16,
|
20 |
+
"grad_norm": 19.26224136352539,
|
21 |
+
"learning_rate": 9.977809823015401e-05,
|
22 |
+
"loss": 13.6217,
|
23 |
+
"step": 20
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.24,
|
27 |
+
"grad_norm": 16.628232955932617,
|
28 |
+
"learning_rate": 9.938441702975689e-05,
|
29 |
+
"loss": 12.3656,
|
30 |
+
"step": 30
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.32,
|
34 |
+
"grad_norm": 13.531852722167969,
|
35 |
+
"learning_rate": 9.879583809693738e-05,
|
36 |
+
"loss": 12.2646,
|
37 |
+
"step": 40
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.4,
|
41 |
+
"grad_norm": 14.354416847229004,
|
42 |
+
"learning_rate": 9.801468428384716e-05,
|
43 |
+
"loss": 12.1808,
|
44 |
+
"step": 50
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.48,
|
48 |
+
"grad_norm": 13.167389869689941,
|
49 |
+
"learning_rate": 9.704403844771128e-05,
|
50 |
+
"loss": 12.1502,
|
51 |
+
"step": 60
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.56,
|
55 |
+
"grad_norm": 12.674458503723145,
|
56 |
+
"learning_rate": 9.588773128419906e-05,
|
57 |
+
"loss": 11.9683,
|
58 |
+
"step": 70
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.64,
|
62 |
+
"grad_norm": 13.783093452453613,
|
63 |
+
"learning_rate": 9.45503262094184e-05,
|
64 |
+
"loss": 11.8564,
|
65 |
+
"step": 80
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.72,
|
69 |
+
"grad_norm": 17.22150993347168,
|
70 |
+
"learning_rate": 9.30371013501972e-05,
|
71 |
+
"loss": 12.0212,
|
72 |
+
"step": 90
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.8,
|
76 |
+
"grad_norm": 13.618257522583008,
|
77 |
+
"learning_rate": 9.135402871372808e-05,
|
78 |
+
"loss": 11.8631,
|
79 |
+
"step": 100
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.88,
|
83 |
+
"grad_norm": 13.423781394958496,
|
84 |
+
"learning_rate": 8.950775061878453e-05,
|
85 |
+
"loss": 11.6954,
|
86 |
+
"step": 110
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.96,
|
90 |
+
"grad_norm": 12.201665878295898,
|
91 |
+
"learning_rate": 8.750555348152298e-05,
|
92 |
+
"loss": 11.7669,
|
93 |
+
"step": 120
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 1.04,
|
97 |
+
"grad_norm": 14.449649810791016,
|
98 |
+
"learning_rate": 8.535533905932738e-05,
|
99 |
+
"loss": 10.3003,
|
100 |
+
"step": 130
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 1.12,
|
104 |
+
"grad_norm": 14.11440658569336,
|
105 |
+
"learning_rate": 8.306559326618259e-05,
|
106 |
+
"loss": 8.5731,
|
107 |
+
"step": 140
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 1.2,
|
111 |
+
"grad_norm": 12.817862510681152,
|
112 |
+
"learning_rate": 8.064535268264883e-05,
|
113 |
+
"loss": 8.5948,
|
114 |
+
"step": 150
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 1.28,
|
118 |
+
"grad_norm": 12.568441390991211,
|
119 |
+
"learning_rate": 7.810416889260653e-05,
|
120 |
+
"loss": 8.656,
|
121 |
+
"step": 160
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 1.3599999999999999,
|
125 |
+
"grad_norm": 13.24677848815918,
|
126 |
+
"learning_rate": 7.545207078751857e-05,
|
127 |
+
"loss": 8.7269,
|
128 |
+
"step": 170
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 1.44,
|
132 |
+
"grad_norm": 12.547300338745117,
|
133 |
+
"learning_rate": 7.269952498697734e-05,
|
134 |
+
"loss": 8.6868,
|
135 |
+
"step": 180
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 1.52,
|
139 |
+
"grad_norm": 12.896221160888672,
|
140 |
+
"learning_rate": 6.985739453173903e-05,
|
141 |
+
"loss": 8.7702,
|
142 |
+
"step": 190
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 1.6,
|
146 |
+
"grad_norm": 12.103998184204102,
|
147 |
+
"learning_rate": 6.693689601226458e-05,
|
148 |
+
"loss": 8.7521,
|
149 |
+
"step": 200
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 1.6800000000000002,
|
153 |
+
"grad_norm": 12.651740074157715,
|
154 |
+
"learning_rate": 6.394955530196147e-05,
|
155 |
+
"loss": 8.8742,
|
156 |
+
"step": 210
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 1.76,
|
160 |
+
"grad_norm": 12.18869400024414,
|
161 |
+
"learning_rate": 6.090716206982714e-05,
|
162 |
+
"loss": 8.7707,
|
163 |
+
"step": 220
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 1.8399999999999999,
|
167 |
+
"grad_norm": 12.143824577331543,
|
168 |
+
"learning_rate": 5.782172325201155e-05,
|
169 |
+
"loss": 8.7433,
|
170 |
+
"step": 230
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 1.92,
|
174 |
+
"grad_norm": 11.785919189453125,
|
175 |
+
"learning_rate": 5.470541566592573e-05,
|
176 |
+
"loss": 8.7179,
|
177 |
+
"step": 240
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 2.0,
|
181 |
+
"grad_norm": 12.037582397460938,
|
182 |
+
"learning_rate": 5.157053795390642e-05,
|
183 |
+
"loss": 8.6486,
|
184 |
+
"step": 250
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 2.08,
|
188 |
+
"grad_norm": 12.652670860290527,
|
189 |
+
"learning_rate": 4.8429462046093585e-05,
|
190 |
+
"loss": 5.591,
|
191 |
+
"step": 260
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 2.16,
|
195 |
+
"grad_norm": 14.214653015136719,
|
196 |
+
"learning_rate": 4.529458433407429e-05,
|
197 |
+
"loss": 5.4084,
|
198 |
+
"step": 270
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 2.24,
|
202 |
+
"grad_norm": 12.59438419342041,
|
203 |
+
"learning_rate": 4.2178276747988446e-05,
|
204 |
+
"loss": 5.3605,
|
205 |
+
"step": 280
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 2.32,
|
209 |
+
"grad_norm": 12.09847354888916,
|
210 |
+
"learning_rate": 3.9092837930172884e-05,
|
211 |
+
"loss": 5.3401,
|
212 |
+
"step": 290
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 2.4,
|
216 |
+
"grad_norm": 13.492688179016113,
|
217 |
+
"learning_rate": 3.605044469803854e-05,
|
218 |
+
"loss": 5.3064,
|
219 |
+
"step": 300
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 2.48,
|
223 |
+
"grad_norm": 12.142609596252441,
|
224 |
+
"learning_rate": 3.3063103987735433e-05,
|
225 |
+
"loss": 5.3187,
|
226 |
+
"step": 310
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 2.56,
|
230 |
+
"grad_norm": 12.719853401184082,
|
231 |
+
"learning_rate": 3.0142605468260978e-05,
|
232 |
+
"loss": 5.413,
|
233 |
+
"step": 320
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 2.64,
|
237 |
+
"grad_norm": 12.052017211914062,
|
238 |
+
"learning_rate": 2.7300475013022663e-05,
|
239 |
+
"loss": 5.2867,
|
240 |
+
"step": 330
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 2.7199999999999998,
|
244 |
+
"grad_norm": 12.2785062789917,
|
245 |
+
"learning_rate": 2.4547929212481435e-05,
|
246 |
+
"loss": 5.3684,
|
247 |
+
"step": 340
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 2.8,
|
251 |
+
"grad_norm": 11.52890396118164,
|
252 |
+
"learning_rate": 2.1895831107393484e-05,
|
253 |
+
"loss": 5.3484,
|
254 |
+
"step": 350
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 2.88,
|
258 |
+
"grad_norm": 12.041983604431152,
|
259 |
+
"learning_rate": 1.9354647317351188e-05,
|
260 |
+
"loss": 5.2971,
|
261 |
+
"step": 360
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 2.96,
|
265 |
+
"grad_norm": 12.474526405334473,
|
266 |
+
"learning_rate": 1.6934406733817414e-05,
|
267 |
+
"loss": 5.2595,
|
268 |
+
"step": 370
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 3.04,
|
272 |
+
"grad_norm": 10.717528343200684,
|
273 |
+
"learning_rate": 1.4644660940672627e-05,
|
274 |
+
"loss": 4.2182,
|
275 |
+
"step": 380
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 3.12,
|
279 |
+
"grad_norm": 11.499459266662598,
|
280 |
+
"learning_rate": 1.2494446518477022e-05,
|
281 |
+
"loss": 3.1071,
|
282 |
+
"step": 390
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 3.2,
|
286 |
+
"grad_norm": 11.054661750793457,
|
287 |
+
"learning_rate": 1.049224938121548e-05,
|
288 |
+
"loss": 3.068,
|
289 |
+
"step": 400
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 3.2800000000000002,
|
293 |
+
"grad_norm": 11.394171714782715,
|
294 |
+
"learning_rate": 8.645971286271904e-06,
|
295 |
+
"loss": 2.9727,
|
296 |
+
"step": 410
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 3.36,
|
300 |
+
"grad_norm": 10.573503494262695,
|
301 |
+
"learning_rate": 6.962898649802823e-06,
|
302 |
+
"loss": 2.9938,
|
303 |
+
"step": 420
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 3.44,
|
307 |
+
"grad_norm": 11.186671257019043,
|
308 |
+
"learning_rate": 5.449673790581611e-06,
|
309 |
+
"loss": 3.0744,
|
310 |
+
"step": 430
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 3.52,
|
314 |
+
"grad_norm": 10.377039909362793,
|
315 |
+
"learning_rate": 4.112268715800943e-06,
|
316 |
+
"loss": 2.9339,
|
317 |
+
"step": 440
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 3.6,
|
321 |
+
"grad_norm": 10.85782527923584,
|
322 |
+
"learning_rate": 2.9559615522887273e-06,
|
323 |
+
"loss": 2.9912,
|
324 |
+
"step": 450
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 3.68,
|
328 |
+
"grad_norm": 11.210704803466797,
|
329 |
+
"learning_rate": 1.985315716152847e-06,
|
330 |
+
"loss": 2.9679,
|
331 |
+
"step": 460
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 3.76,
|
335 |
+
"grad_norm": 11.051790237426758,
|
336 |
+
"learning_rate": 1.2041619030626284e-06,
|
337 |
+
"loss": 3.0297,
|
338 |
+
"step": 470
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 3.84,
|
342 |
+
"grad_norm": 10.918357849121094,
|
343 |
+
"learning_rate": 6.15582970243117e-07,
|
344 |
+
"loss": 2.9479,
|
345 |
+
"step": 480
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 3.92,
|
349 |
+
"grad_norm": 10.786859512329102,
|
350 |
+
"learning_rate": 2.219017698460002e-07,
|
351 |
+
"loss": 3.0567,
|
352 |
+
"step": 490
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 4.0,
|
356 |
+
"grad_norm": 10.881739616394043,
|
357 |
+
"learning_rate": 2.467198171342e-08,
|
358 |
+
"loss": 3.0452,
|
359 |
+
"step": 500
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 4.0,
|
363 |
+
"step": 500,
|
364 |
+
"total_flos": 4.036279852125389e+16,
|
365 |
+
"train_loss": 7.3980213584899905,
|
366 |
+
"train_runtime": 1227.9067,
|
367 |
+
"train_samples_per_second": 19.542,
|
368 |
+
"train_steps_per_second": 0.407
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 10,
|
372 |
+
"max_steps": 500,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 4,
|
375 |
+
"save_steps": 2000,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": true,
|
383 |
+
"should_training_stop": true
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 4.036279852125389e+16,
|
389 |
+
"train_batch_size": 2,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:be8e15e7ee5c54fefa3a3ec8ddf285112e954bc3f80498cf81f01aa3c0e48ec1
|
3 |
+
size 5560
|
training_loss.png
ADDED
![]() |
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|