BurnyCoder commited on
Commit
bce1339
·
verified ·
1 Parent(s): ec67be8

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
36
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
  checkpoint-84/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
  checkpoint-126/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
36
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
  checkpoint-84/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
  checkpoint-126/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
+ checkpoint-42/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-42/README.md ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen2.5-0.5B
4
+ datasets:
5
+ - trl-lib/Capybara
6
+ language:
7
+ - en
8
+ tags:
9
+ - conversational
10
+ - multi-turn
11
+ - instruction-following
12
+ - reasoning
13
+ - sft
14
+ - trl
15
+ pipeline_tag: text-generation
16
+ library_name: transformers
17
+ ---
18
+
19
+ # Qwen2.5-0.5B-Capybara
20
+
21
+ This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B](https://huggingface.co/Qwen/Qwen2.5-0.5B) on the [trl-lib/Capybara](https://huggingface.co/datasets/trl-lib/Capybara) dataset using Supervised Fine-Tuning (SFT).
22
+
23
+ ## Model Description
24
+
25
+ **Qwen2.5-0.5B-Capybara** is trained on the Capybara dataset, which is a high-quality multi-turn conversational dataset emphasizing:
26
+
27
+ - **Reasoning and Logic**: Strong focus on extrapolation and logical thinking
28
+ - **Information Diversity**: Wide range of domains including STEM, pop-culture, and general knowledge
29
+ - **Multi-turn Conversations**: Average of 3+ turns per conversation with 1,000+ tokens context
30
+ - **Natural Prose**: Maintains conversational flow while exploring complex topics
31
+
32
+ The Capybara dataset was created using the Amplify-Instruct synthesis method, combining techniques from Airoboros, Evol-Instruct (WizardLM), Orca, and other high-performing datasets.
33
+
34
+ ## Training Details
35
+
36
+ ### Training Configuration
37
+
38
+ | Parameter | Value |
39
+ |-----------|-------|
40
+ | Base Model | Qwen/Qwen2.5-0.5B |
41
+ | Training Method | Supervised Fine-Tuning (SFT) |
42
+ | Dataset | trl-lib/Capybara (15,806 samples) |
43
+ | Epochs | 3 |
44
+ | Batch Size | 8 |
45
+ | Gradient Accumulation | 4 |
46
+ | Effective Batch Size | 32 |
47
+ | Learning Rate | 2e-5 |
48
+ | LR Scheduler | Linear |
49
+ | Precision | BF16 |
50
+ | Max Sequence Length | 1024 |
51
+ | Optimizer | AdamW (fused) |
52
+
53
+ ### Memory Optimizations
54
+
55
+ - **Liger Kernel**: Enabled for ~60% VRAM reduction
56
+ - **Gradient Checkpointing**: Enabled
57
+ - **BF16 Mixed Precision**: Enabled
58
+
59
+ ### Training Infrastructure
60
+
61
+ - Framework: [TRL](https://github.com/huggingface/trl) (Transformer Reinforcement Learning)
62
+ - Hardware: Single GPU (8GB VRAM)
63
+ - Training Time: ~5 minutes per step
64
+
65
+ ### Training Progress
66
+
67
+ > **Note**: This is checkpoint at step 42 of 1,482 total steps (~3% of training). This is an early checkpoint from an ongoing training run.
68
+
69
+ | Metric | Value |
70
+ |--------|-------|
71
+ | Steps Completed | 42 / 1,482 |
72
+ | Training Progress | ~3% |
73
+ | Final Loss (step 42) | 1.3058 |
74
+ | Initial Loss (step 1) | 1.9030 |
75
+
76
+ ## Dataset Information
77
+
78
+ The [trl-lib/Capybara](https://huggingface.co/datasets/trl-lib/Capybara) dataset contains:
79
+
80
+ - **15,806 training samples** of multi-turn conversations
81
+ - **Sources**: GPT4LLM, GOAT, EverythingLM, Know-Logic, SuperCOT, Airoboros, Dove, TheoremQA, TaskSource, General-Instruct
82
+ - **Format**: Conversation messages with user/assistant roles
83
+ - **Quality**: Aggressively filtered to remove alignment artifacts and common undesirable behaviors
84
+
85
+ ## Usage
86
+
87
+ ```python
88
+ from transformers import AutoModelForCausalLM, AutoTokenizer
89
+
90
+ model = AutoModelForCausalLM.from_pretrained("BurnyCoder/Qwen2.5-0.5B-Capybara")
91
+ tokenizer = AutoTokenizer.from_pretrained("BurnyCoder/Qwen2.5-0.5B-Capybara")
92
+
93
+ messages = [
94
+ {"role": "user", "content": "Explain the concept of recursion in programming."}
95
+ ]
96
+
97
+ text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
98
+ inputs = tokenizer(text, return_tensors="pt")
99
+ outputs = model.generate(**inputs, max_new_tokens=512)
100
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
101
+ ```
102
+
103
+ ## Limitations
104
+
105
+ - This is an early checkpoint (~3% trained) and may not reflect full model capabilities
106
+ - Model size is 494M parameters - suitable for edge deployment but limited compared to larger models
107
+ - Training was conducted on a single GPU with memory optimizations
108
+
109
+ ## Citation
110
+
111
+ If you use this model, please cite:
112
+
113
+ ```bibtex
114
+ @misc{qwen2.5-0.5b-capybara,
115
+ author = {BurnyCoder},
116
+ title = {Qwen2.5-0.5B-Capybara: Multi-turn Conversational Fine-tuning},
117
+ year = {2026},
118
+ publisher = {Hugging Face},
119
+ url = {https://huggingface.co/BurnyCoder/Qwen2.5-0.5B-Capybara}
120
+ }
121
+ ```
122
+
123
+ ## Acknowledgments
124
+
125
+ - [Qwen Team](https://huggingface.co/Qwen) for the base model
126
+ - [TRL Library](https://github.com/huggingface/trl) for the training framework
127
+ - [Capybara Dataset](https://huggingface.co/datasets/trl-lib/Capybara) creators
checkpoint-42/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
+ }
checkpoint-42/chat_template.jinja ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0]['role'] == 'system' %}
4
+ {{- messages[0]['content'] }}
5
+ {%- else %}
6
+ {{- 'You are a helpful assistant.' }}
7
+ {%- endif %}
8
+ {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
9
+ {%- for tool in tools %}
10
+ {{- "\n" }}
11
+ {{- tool | tojson }}
12
+ {%- endfor %}
13
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
14
+ {%- else %}
15
+ {%- if messages[0]['role'] == 'system' %}
16
+ {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
17
+ {%- else %}
18
+ {{- '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n' }}
19
+ {%- endif %}
20
+ {%- endif %}
21
+ {%- for message in messages %}
22
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
23
+ {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
24
+ {%- elif message.role == "assistant" %}
25
+ {{- '<|im_start|>' + message.role }}
26
+ {%- if message.content %}
27
+ {{- '\n' + message.content }}
28
+ {%- endif %}
29
+ {%- for tool_call in message.tool_calls %}
30
+ {%- if tool_call.function is defined %}
31
+ {%- set tool_call = tool_call.function %}
32
+ {%- endif %}
33
+ {{- '\n<tool_call>\n{"name": "' }}
34
+ {{- tool_call.name }}
35
+ {{- '", "arguments": ' }}
36
+ {{- tool_call.arguments | tojson }}
37
+ {{- '}\n</tool_call>' }}
38
+ {%- endfor %}
39
+ {{- '<|im_end|>\n' }}
40
+ {%- elif message.role == "tool" %}
41
+ {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
42
+ {{- '<|im_start|>user' }}
43
+ {%- endif %}
44
+ {{- '\n<tool_response>\n' }}
45
+ {{- message.content }}
46
+ {{- '\n</tool_response>' }}
47
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
48
+ {{- '<|im_end|>\n' }}
49
+ {%- endif %}
50
+ {%- endif %}
51
+ {%- endfor %}
52
+ {%- if add_generation_prompt %}
53
+ {{- '<|im_start|>assistant\n' }}
54
+ {%- endif %}
checkpoint-42/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "dtype": "float32",
7
+ "eos_token_id": 151643,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 896,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 4864,
12
+ "layer_types": [
13
+ "full_attention",
14
+ "full_attention",
15
+ "full_attention",
16
+ "full_attention",
17
+ "full_attention",
18
+ "full_attention",
19
+ "full_attention",
20
+ "full_attention",
21
+ "full_attention",
22
+ "full_attention",
23
+ "full_attention",
24
+ "full_attention",
25
+ "full_attention",
26
+ "full_attention",
27
+ "full_attention",
28
+ "full_attention",
29
+ "full_attention",
30
+ "full_attention",
31
+ "full_attention",
32
+ "full_attention",
33
+ "full_attention",
34
+ "full_attention",
35
+ "full_attention",
36
+ "full_attention"
37
+ ],
38
+ "max_position_embeddings": 32768,
39
+ "max_window_layers": 24,
40
+ "model_type": "qwen2",
41
+ "num_attention_heads": 14,
42
+ "num_hidden_layers": 24,
43
+ "num_key_value_heads": 2,
44
+ "pad_token_id": 151643,
45
+ "rms_norm_eps": 1e-06,
46
+ "rope_scaling": null,
47
+ "rope_theta": 1000000.0,
48
+ "sliding_window": null,
49
+ "tie_word_embeddings": true,
50
+ "transformers_version": "4.57.3",
51
+ "use_cache": true,
52
+ "use_mrope": false,
53
+ "use_sliding_window": false,
54
+ "vocab_size": 151936
55
+ }
checkpoint-42/generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eos_token_id": [
3
+ 151643
4
+ ],
5
+ "max_new_tokens": 2048,
6
+ "pad_token_id": 151643,
7
+ "transformers_version": "4.57.3"
8
+ }
checkpoint-42/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-42/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b476748dfa600e9a136149064439f11c9c01b50d5f7cb27c27addd3da0633954
3
+ size 1976163472
checkpoint-42/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e71fbbeb0c4989bb565ed6176492a866eb3a58a7aaa7a9ccc2fcf98723ca4aa9
3
+ size 3952509771
checkpoint-42/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26776f79c2b16e7edf3d8ed6b77bbe9f1d991ec3a50a0e4e9f6affcbbd7ec955
3
+ size 14645
checkpoint-42/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88bb743c5f089b081e33593697c9871df45b5e1c616e88dfcecbc1f6d9fa1b1b
3
+ size 1465
checkpoint-42/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": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }
checkpoint-42/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c5ae00e602b8860cbd784ba82a8aa14e8feecec692e7076590d014d7b7fdafa
3
+ size 11421896
checkpoint-42/tokenizer_config.json ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "clean_up_tokenization_spaces": false,
199
+ "eos_token": "<|endoftext|>",
200
+ "errors": "replace",
201
+ "extra_special_tokens": {},
202
+ "model_max_length": 131072,
203
+ "pad_token": "<|endoftext|>",
204
+ "split_special_tokens": false,
205
+ "tokenizer_class": "Qwen2Tokenizer",
206
+ "unk_token": null
207
+ }
checkpoint-42/trainer_state.json ADDED
@@ -0,0 +1,412 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08502024291497975,
6
+ "eval_steps": 500,
7
+ "global_step": 42,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.0020242914979757085,
14
+ "grad_norm": 13.587785720825195,
15
+ "learning_rate": 2e-05,
16
+ "loss": 1.903,
17
+ "mean_token_accuracy": 0.5894419699907303,
18
+ "num_tokens": 23998.0,
19
+ "step": 1
20
+ },
21
+ {
22
+ "epoch": 0.004048582995951417,
23
+ "grad_norm": 10.837627410888672,
24
+ "learning_rate": 1.998650472334683e-05,
25
+ "loss": 1.7773,
26
+ "mean_token_accuracy": 0.6223782598972321,
27
+ "num_tokens": 46596.0,
28
+ "step": 2
29
+ },
30
+ {
31
+ "epoch": 0.006072874493927126,
32
+ "grad_norm": 5.157018184661865,
33
+ "learning_rate": 1.9973009446693658e-05,
34
+ "loss": 1.9274,
35
+ "mean_token_accuracy": 0.5814870595932007,
36
+ "num_tokens": 71542.0,
37
+ "step": 3
38
+ },
39
+ {
40
+ "epoch": 0.008097165991902834,
41
+ "grad_norm": 5.115257263183594,
42
+ "learning_rate": 1.9959514170040488e-05,
43
+ "loss": 1.6163,
44
+ "mean_token_accuracy": 0.6391739100217819,
45
+ "num_tokens": 90870.0,
46
+ "step": 4
47
+ },
48
+ {
49
+ "epoch": 0.010121457489878543,
50
+ "grad_norm": 4.290038585662842,
51
+ "learning_rate": 1.9946018893387314e-05,
52
+ "loss": 1.5241,
53
+ "mean_token_accuracy": 0.6456017643213272,
54
+ "num_tokens": 115219.0,
55
+ "step": 5
56
+ },
57
+ {
58
+ "epoch": 0.012145748987854251,
59
+ "grad_norm": 3.599583864212036,
60
+ "learning_rate": 1.9932523616734144e-05,
61
+ "loss": 1.363,
62
+ "mean_token_accuracy": 0.6773215681314468,
63
+ "num_tokens": 136447.0,
64
+ "step": 6
65
+ },
66
+ {
67
+ "epoch": 0.01417004048582996,
68
+ "grad_norm": 3.737048625946045,
69
+ "learning_rate": 1.9919028340080974e-05,
70
+ "loss": 1.6063,
71
+ "mean_token_accuracy": 0.6296929270029068,
72
+ "num_tokens": 159775.0,
73
+ "step": 7
74
+ },
75
+ {
76
+ "epoch": 0.016194331983805668,
77
+ "grad_norm": 3.6092047691345215,
78
+ "learning_rate": 1.9905533063427804e-05,
79
+ "loss": 1.5649,
80
+ "mean_token_accuracy": 0.639984205365181,
81
+ "num_tokens": 184127.0,
82
+ "step": 8
83
+ },
84
+ {
85
+ "epoch": 0.018218623481781375,
86
+ "grad_norm": 3.934776782989502,
87
+ "learning_rate": 1.989203778677463e-05,
88
+ "loss": 1.8535,
89
+ "mean_token_accuracy": 0.594915583729744,
90
+ "num_tokens": 206169.0,
91
+ "step": 9
92
+ },
93
+ {
94
+ "epoch": 0.020242914979757085,
95
+ "grad_norm": 3.7700464725494385,
96
+ "learning_rate": 1.987854251012146e-05,
97
+ "loss": 1.4936,
98
+ "mean_token_accuracy": 0.6473288685083389,
99
+ "num_tokens": 226067.0,
100
+ "step": 10
101
+ },
102
+ {
103
+ "epoch": 0.022267206477732792,
104
+ "grad_norm": 3.7429916858673096,
105
+ "learning_rate": 1.986504723346829e-05,
106
+ "loss": 1.5662,
107
+ "mean_token_accuracy": 0.6233761757612228,
108
+ "num_tokens": 248812.0,
109
+ "step": 11
110
+ },
111
+ {
112
+ "epoch": 0.024291497975708502,
113
+ "grad_norm": 3.451914072036743,
114
+ "learning_rate": 1.9851551956815116e-05,
115
+ "loss": 1.7484,
116
+ "mean_token_accuracy": 0.6058520972728729,
117
+ "num_tokens": 272961.0,
118
+ "step": 12
119
+ },
120
+ {
121
+ "epoch": 0.02631578947368421,
122
+ "grad_norm": 3.706216335296631,
123
+ "learning_rate": 1.9838056680161946e-05,
124
+ "loss": 1.4828,
125
+ "mean_token_accuracy": 0.6424361318349838,
126
+ "num_tokens": 293964.0,
127
+ "step": 13
128
+ },
129
+ {
130
+ "epoch": 0.02834008097165992,
131
+ "grad_norm": 3.402036428451538,
132
+ "learning_rate": 1.9824561403508773e-05,
133
+ "loss": 1.6059,
134
+ "mean_token_accuracy": 0.6380074322223663,
135
+ "num_tokens": 314414.0,
136
+ "step": 14
137
+ },
138
+ {
139
+ "epoch": 0.030364372469635626,
140
+ "grad_norm": 3.5934112071990967,
141
+ "learning_rate": 1.9811066126855602e-05,
142
+ "loss": 1.6123,
143
+ "mean_token_accuracy": 0.6258052587509155,
144
+ "num_tokens": 338761.0,
145
+ "step": 15
146
+ },
147
+ {
148
+ "epoch": 0.032388663967611336,
149
+ "grad_norm": 3.2826409339904785,
150
+ "learning_rate": 1.979757085020243e-05,
151
+ "loss": 1.5093,
152
+ "mean_token_accuracy": 0.647676095366478,
153
+ "num_tokens": 364154.0,
154
+ "step": 16
155
+ },
156
+ {
157
+ "epoch": 0.03441295546558704,
158
+ "grad_norm": 3.411837339401245,
159
+ "learning_rate": 1.978407557354926e-05,
160
+ "loss": 1.5875,
161
+ "mean_token_accuracy": 0.6337466537952423,
162
+ "num_tokens": 387522.0,
163
+ "step": 17
164
+ },
165
+ {
166
+ "epoch": 0.03643724696356275,
167
+ "grad_norm": 3.537415027618408,
168
+ "learning_rate": 1.977058029689609e-05,
169
+ "loss": 1.6839,
170
+ "mean_token_accuracy": 0.6119341999292374,
171
+ "num_tokens": 410636.0,
172
+ "step": 18
173
+ },
174
+ {
175
+ "epoch": 0.038461538461538464,
176
+ "grad_norm": 3.6170666217803955,
177
+ "learning_rate": 1.9757085020242915e-05,
178
+ "loss": 1.3929,
179
+ "mean_token_accuracy": 0.6720004975795746,
180
+ "num_tokens": 430858.0,
181
+ "step": 19
182
+ },
183
+ {
184
+ "epoch": 0.04048582995951417,
185
+ "grad_norm": 3.725717782974243,
186
+ "learning_rate": 1.9743589743589745e-05,
187
+ "loss": 1.3478,
188
+ "mean_token_accuracy": 0.6779870688915253,
189
+ "num_tokens": 451526.0,
190
+ "step": 20
191
+ },
192
+ {
193
+ "epoch": 0.04251012145748988,
194
+ "grad_norm": 3.7410740852355957,
195
+ "learning_rate": 1.9730094466936575e-05,
196
+ "loss": 1.5027,
197
+ "mean_token_accuracy": 0.6415430754423141,
198
+ "num_tokens": 472555.0,
199
+ "step": 21
200
+ },
201
+ {
202
+ "epoch": 0.044534412955465584,
203
+ "grad_norm": 3.8375744819641113,
204
+ "learning_rate": 1.9716599190283405e-05,
205
+ "loss": 1.3805,
206
+ "mean_token_accuracy": 0.6554747521877289,
207
+ "num_tokens": 492412.0,
208
+ "step": 22
209
+ },
210
+ {
211
+ "epoch": 0.0465587044534413,
212
+ "grad_norm": 3.4310216903686523,
213
+ "learning_rate": 1.970310391363023e-05,
214
+ "loss": 1.3993,
215
+ "mean_token_accuracy": 0.6516353040933609,
216
+ "num_tokens": 517688.0,
217
+ "step": 23
218
+ },
219
+ {
220
+ "epoch": 0.048582995951417005,
221
+ "grad_norm": 3.4065134525299072,
222
+ "learning_rate": 1.968960863697706e-05,
223
+ "loss": 1.4144,
224
+ "mean_token_accuracy": 0.6639417558908463,
225
+ "num_tokens": 539605.0,
226
+ "step": 24
227
+ },
228
+ {
229
+ "epoch": 0.05060728744939271,
230
+ "grad_norm": 3.4423940181732178,
231
+ "learning_rate": 1.9676113360323887e-05,
232
+ "loss": 1.6237,
233
+ "mean_token_accuracy": 0.6199875771999359,
234
+ "num_tokens": 563276.0,
235
+ "step": 25
236
+ },
237
+ {
238
+ "epoch": 0.05263157894736842,
239
+ "grad_norm": 3.211747407913208,
240
+ "learning_rate": 1.9662618083670717e-05,
241
+ "loss": 1.4059,
242
+ "mean_token_accuracy": 0.6552923172712326,
243
+ "num_tokens": 586603.0,
244
+ "step": 26
245
+ },
246
+ {
247
+ "epoch": 0.05465587044534413,
248
+ "grad_norm": 3.1153526306152344,
249
+ "learning_rate": 1.9649122807017544e-05,
250
+ "loss": 1.2644,
251
+ "mean_token_accuracy": 0.6816990375518799,
252
+ "num_tokens": 612691.0,
253
+ "step": 27
254
+ },
255
+ {
256
+ "epoch": 0.05668016194331984,
257
+ "grad_norm": 3.2474708557128906,
258
+ "learning_rate": 1.9635627530364373e-05,
259
+ "loss": 1.4524,
260
+ "mean_token_accuracy": 0.6650048345327377,
261
+ "num_tokens": 636325.0,
262
+ "step": 28
263
+ },
264
+ {
265
+ "epoch": 0.058704453441295545,
266
+ "grad_norm": 3.521009683609009,
267
+ "learning_rate": 1.9622132253711203e-05,
268
+ "loss": 1.3588,
269
+ "mean_token_accuracy": 0.6608386486768723,
270
+ "num_tokens": 657410.0,
271
+ "step": 29
272
+ },
273
+ {
274
+ "epoch": 0.06072874493927125,
275
+ "grad_norm": 3.240419387817383,
276
+ "learning_rate": 1.960863697705803e-05,
277
+ "loss": 1.6196,
278
+ "mean_token_accuracy": 0.634381040930748,
279
+ "num_tokens": 678587.0,
280
+ "step": 30
281
+ },
282
+ {
283
+ "epoch": 0.06275303643724696,
284
+ "grad_norm": 3.0680091381073,
285
+ "learning_rate": 1.959514170040486e-05,
286
+ "loss": 1.537,
287
+ "mean_token_accuracy": 0.6444396674633026,
288
+ "num_tokens": 700383.0,
289
+ "step": 31
290
+ },
291
+ {
292
+ "epoch": 0.06477732793522267,
293
+ "grad_norm": 3.087522029876709,
294
+ "learning_rate": 1.958164642375169e-05,
295
+ "loss": 1.6414,
296
+ "mean_token_accuracy": 0.6257035434246063,
297
+ "num_tokens": 723769.0,
298
+ "step": 32
299
+ },
300
+ {
301
+ "epoch": 0.06680161943319839,
302
+ "grad_norm": 3.2430222034454346,
303
+ "learning_rate": 1.9568151147098516e-05,
304
+ "loss": 1.5166,
305
+ "mean_token_accuracy": 0.6486149281263351,
306
+ "num_tokens": 747387.0,
307
+ "step": 33
308
+ },
309
+ {
310
+ "epoch": 0.06882591093117409,
311
+ "grad_norm": 3.1888442039489746,
312
+ "learning_rate": 1.9554655870445346e-05,
313
+ "loss": 1.4071,
314
+ "mean_token_accuracy": 0.6582628488540649,
315
+ "num_tokens": 771537.0,
316
+ "step": 34
317
+ },
318
+ {
319
+ "epoch": 0.0708502024291498,
320
+ "grad_norm": 2.9818553924560547,
321
+ "learning_rate": 1.9541160593792176e-05,
322
+ "loss": 1.4723,
323
+ "mean_token_accuracy": 0.6373147964477539,
324
+ "num_tokens": 794954.0,
325
+ "step": 35
326
+ },
327
+ {
328
+ "epoch": 0.0728744939271255,
329
+ "grad_norm": 2.8076112270355225,
330
+ "learning_rate": 1.9527665317139005e-05,
331
+ "loss": 1.5494,
332
+ "mean_token_accuracy": 0.6327401697635651,
333
+ "num_tokens": 820451.0,
334
+ "step": 36
335
+ },
336
+ {
337
+ "epoch": 0.07489878542510121,
338
+ "grad_norm": 3.305832862854004,
339
+ "learning_rate": 1.9514170040485832e-05,
340
+ "loss": 1.5983,
341
+ "mean_token_accuracy": 0.6266501545906067,
342
+ "num_tokens": 840241.0,
343
+ "step": 37
344
+ },
345
+ {
346
+ "epoch": 0.07692307692307693,
347
+ "grad_norm": 2.9532933235168457,
348
+ "learning_rate": 1.9500674763832662e-05,
349
+ "loss": 1.4127,
350
+ "mean_token_accuracy": 0.6481295526027679,
351
+ "num_tokens": 862831.0,
352
+ "step": 38
353
+ },
354
+ {
355
+ "epoch": 0.07894736842105263,
356
+ "grad_norm": 2.7358744144439697,
357
+ "learning_rate": 1.9487179487179488e-05,
358
+ "loss": 1.4085,
359
+ "mean_token_accuracy": 0.6563303023576736,
360
+ "num_tokens": 887709.0,
361
+ "step": 39
362
+ },
363
+ {
364
+ "epoch": 0.08097165991902834,
365
+ "grad_norm": 2.550145149230957,
366
+ "learning_rate": 1.9473684210526318e-05,
367
+ "loss": 1.398,
368
+ "mean_token_accuracy": 0.6520788222551346,
369
+ "num_tokens": 914680.0,
370
+ "step": 40
371
+ },
372
+ {
373
+ "epoch": 0.08299595141700405,
374
+ "grad_norm": 2.6927826404571533,
375
+ "learning_rate": 1.9460188933873144e-05,
376
+ "loss": 1.618,
377
+ "mean_token_accuracy": 0.624066486954689,
378
+ "num_tokens": 939210.0,
379
+ "step": 41
380
+ },
381
+ {
382
+ "epoch": 0.08502024291497975,
383
+ "grad_norm": 2.7264392375946045,
384
+ "learning_rate": 1.9446693657219974e-05,
385
+ "loss": 1.3058,
386
+ "mean_token_accuracy": 0.6675033718347549,
387
+ "num_tokens": 960279.0,
388
+ "step": 42
389
+ }
390
+ ],
391
+ "logging_steps": 1,
392
+ "max_steps": 1482,
393
+ "num_input_tokens_seen": 0,
394
+ "num_train_epochs": 3,
395
+ "save_steps": 42,
396
+ "stateful_callbacks": {
397
+ "TrainerControl": {
398
+ "args": {
399
+ "should_epoch_stop": false,
400
+ "should_evaluate": false,
401
+ "should_log": false,
402
+ "should_save": true,
403
+ "should_training_stop": false
404
+ },
405
+ "attributes": {}
406
+ }
407
+ },
408
+ "total_flos": 2947625944952832.0,
409
+ "train_batch_size": 8,
410
+ "trial_name": null,
411
+ "trial_params": null
412
+ }
checkpoint-42/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ff6ea40a4d93ee5f566100c9ce5a240747fc169e395973755b75186069ac74d
3
+ size 6289
checkpoint-42/vocab.json ADDED
The diff for this file is too large to render. See raw diff