r1char9 commited on
Commit
f38c6cf
·
verified ·
1 Parent(s): 729e98c

Upload checkpoint-4870 for Oblivion 1 version

Browse files
.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,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen2.5-1.5B
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.15.2
adapter_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "Qwen/Qwen2.5-1.5B",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "r": 16,
24
+ "rank_pattern": {},
25
+ "revision": null,
26
+ "target_modules": [
27
+ "v_proj",
28
+ "q_proj"
29
+ ],
30
+ "task_type": "CAUSAL_LM",
31
+ "trainable_token_indices": null,
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:08c81702046a58b7b121e94e3e7952979c09aa2b0a7b91b6a7d4f47658cd7517
3
+ size 8731128
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
+ }
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 %}
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88ec65124bf0f855ebe5d311ed1557155fc1a55104b8bf89e666572d1ab2b6c4
3
+ size 17526842
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9ca9e72990c5af0d35217a61fb66b302ae217d96db755bdc05fc88d8c5df110
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8413b6d48044851b42e8fda63c5e0d63780704e6bf229337e97665bed519cafa
3
+ size 1064
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
+ }
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,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
+ }
trainer_state.json ADDED
@@ -0,0 +1,793 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 4870,
3
+ "best_metric": 1.0858850479125977,
4
+ "best_model_checkpoint": "/mnt/tank/scratch/rgurtsiev/workflow/STAGE1-DEMO/checkpoint-4870",
5
+ "epoch": 10.0,
6
+ "eval_steps": 500,
7
+ "global_step": 4870,
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.1026694045174538,
14
+ "grad_norm": 1.1697688102722168,
15
+ "learning_rate": 1.3415468856947298e-06,
16
+ "loss": 3.6536,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.2053388090349076,
21
+ "grad_norm": 1.3012523651123047,
22
+ "learning_rate": 2.7104722792607808e-06,
23
+ "loss": 3.6247,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.3080082135523614,
28
+ "grad_norm": 1.3252615928649902,
29
+ "learning_rate": 4.0793976728268315e-06,
30
+ "loss": 3.5833,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 0.4106776180698152,
35
+ "grad_norm": 1.2821773290634155,
36
+ "learning_rate": 5.448323066392882e-06,
37
+ "loss": 3.471,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 0.5133470225872689,
42
+ "grad_norm": 1.5595322847366333,
43
+ "learning_rate": 6.817248459958933e-06,
44
+ "loss": 3.2428,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 0.6160164271047228,
49
+ "grad_norm": 1.5284234285354614,
50
+ "learning_rate": 8.186173853524983e-06,
51
+ "loss": 2.9622,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 0.7186858316221766,
56
+ "grad_norm": 1.7251927852630615,
57
+ "learning_rate": 9.555099247091034e-06,
58
+ "loss": 2.536,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 0.8213552361396304,
63
+ "grad_norm": 1.3935151100158691,
64
+ "learning_rate": 1.0924024640657084e-05,
65
+ "loss": 2.0349,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 0.9240246406570842,
70
+ "grad_norm": 0.7713799476623535,
71
+ "learning_rate": 1.2292950034223137e-05,
72
+ "loss": 1.7696,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_loss": 1.657446265220642,
78
+ "eval_runtime": 396.4486,
79
+ "eval_samples_per_second": 4.911,
80
+ "eval_steps_per_second": 0.615,
81
+ "step": 487
82
+ },
83
+ {
84
+ "epoch": 1.0266940451745379,
85
+ "grad_norm": 0.8125890493392944,
86
+ "learning_rate": 1.3661875427789186e-05,
87
+ "loss": 1.6644,
88
+ "step": 500
89
+ },
90
+ {
91
+ "epoch": 1.1293634496919918,
92
+ "grad_norm": 0.7177293300628662,
93
+ "learning_rate": 1.5030800821355238e-05,
94
+ "loss": 1.5654,
95
+ "step": 550
96
+ },
97
+ {
98
+ "epoch": 1.2320328542094456,
99
+ "grad_norm": 0.9869002103805542,
100
+ "learning_rate": 1.6399726214921288e-05,
101
+ "loss": 1.543,
102
+ "step": 600
103
+ },
104
+ {
105
+ "epoch": 1.3347022587268995,
106
+ "grad_norm": 1.156468152999878,
107
+ "learning_rate": 1.776865160848734e-05,
108
+ "loss": 1.5008,
109
+ "step": 650
110
+ },
111
+ {
112
+ "epoch": 1.4373716632443532,
113
+ "grad_norm": 0.9758607149124146,
114
+ "learning_rate": 1.913757700205339e-05,
115
+ "loss": 1.4734,
116
+ "step": 700
117
+ },
118
+ {
119
+ "epoch": 1.5400410677618068,
120
+ "grad_norm": 1.0511637926101685,
121
+ "learning_rate": 2.050650239561944e-05,
122
+ "loss": 1.4343,
123
+ "step": 750
124
+ },
125
+ {
126
+ "epoch": 1.6427104722792607,
127
+ "grad_norm": 1.1518610715866089,
128
+ "learning_rate": 2.187542778918549e-05,
129
+ "loss": 1.4289,
130
+ "step": 800
131
+ },
132
+ {
133
+ "epoch": 1.7453798767967146,
134
+ "grad_norm": 1.1441558599472046,
135
+ "learning_rate": 2.324435318275154e-05,
136
+ "loss": 1.3902,
137
+ "step": 850
138
+ },
139
+ {
140
+ "epoch": 1.8480492813141685,
141
+ "grad_norm": 1.3006062507629395,
142
+ "learning_rate": 2.4613278576317596e-05,
143
+ "loss": 1.3834,
144
+ "step": 900
145
+ },
146
+ {
147
+ "epoch": 1.9507186858316223,
148
+ "grad_norm": 1.1244959831237793,
149
+ "learning_rate": 2.5982203969883643e-05,
150
+ "loss": 1.3556,
151
+ "step": 950
152
+ },
153
+ {
154
+ "epoch": 2.0,
155
+ "eval_loss": 1.3616195917129517,
156
+ "eval_runtime": 396.64,
157
+ "eval_samples_per_second": 4.909,
158
+ "eval_steps_per_second": 0.615,
159
+ "step": 974
160
+ },
161
+ {
162
+ "epoch": 2.0533880903490758,
163
+ "grad_norm": 1.4575847387313843,
164
+ "learning_rate": 2.7351129363449694e-05,
165
+ "loss": 1.3388,
166
+ "step": 1000
167
+ },
168
+ {
169
+ "epoch": 2.1560574948665296,
170
+ "grad_norm": 1.3578325510025024,
171
+ "learning_rate": 2.8720054757015744e-05,
172
+ "loss": 1.3067,
173
+ "step": 1050
174
+ },
175
+ {
176
+ "epoch": 2.2587268993839835,
177
+ "grad_norm": 1.9210494756698608,
178
+ "learning_rate": 3.0088980150581795e-05,
179
+ "loss": 1.3183,
180
+ "step": 1100
181
+ },
182
+ {
183
+ "epoch": 2.3613963039014374,
184
+ "grad_norm": 2.051852226257324,
185
+ "learning_rate": 3.1457905544147846e-05,
186
+ "loss": 1.274,
187
+ "step": 1150
188
+ },
189
+ {
190
+ "epoch": 2.4640657084188913,
191
+ "grad_norm": 1.660125970840454,
192
+ "learning_rate": 3.282683093771389e-05,
193
+ "loss": 1.2661,
194
+ "step": 1200
195
+ },
196
+ {
197
+ "epoch": 2.566735112936345,
198
+ "grad_norm": 4.411022663116455,
199
+ "learning_rate": 3.419575633127995e-05,
200
+ "loss": 1.2339,
201
+ "step": 1250
202
+ },
203
+ {
204
+ "epoch": 2.669404517453799,
205
+ "grad_norm": 1.519783854484558,
206
+ "learning_rate": 3.5564681724846e-05,
207
+ "loss": 1.2286,
208
+ "step": 1300
209
+ },
210
+ {
211
+ "epoch": 2.7720739219712525,
212
+ "grad_norm": 1.7709407806396484,
213
+ "learning_rate": 3.693360711841205e-05,
214
+ "loss": 1.219,
215
+ "step": 1350
216
+ },
217
+ {
218
+ "epoch": 2.8747433264887063,
219
+ "grad_norm": 1.8909534215927124,
220
+ "learning_rate": 3.8302532511978096e-05,
221
+ "loss": 1.2072,
222
+ "step": 1400
223
+ },
224
+ {
225
+ "epoch": 2.97741273100616,
226
+ "grad_norm": 1.4890103340148926,
227
+ "learning_rate": 3.967145790554415e-05,
228
+ "loss": 1.1951,
229
+ "step": 1450
230
+ },
231
+ {
232
+ "epoch": 3.0,
233
+ "eval_loss": 1.2139294147491455,
234
+ "eval_runtime": 396.2906,
235
+ "eval_samples_per_second": 4.913,
236
+ "eval_steps_per_second": 0.616,
237
+ "step": 1461
238
+ },
239
+ {
240
+ "epoch": 3.080082135523614,
241
+ "grad_norm": 2.104808807373047,
242
+ "learning_rate": 3.998773779555414e-05,
243
+ "loss": 1.2078,
244
+ "step": 1500
245
+ },
246
+ {
247
+ "epoch": 3.1827515400410675,
248
+ "grad_norm": 1.3491119146347046,
249
+ "learning_rate": 3.993426857404775e-05,
250
+ "loss": 1.1717,
251
+ "step": 1550
252
+ },
253
+ {
254
+ "epoch": 3.2854209445585214,
255
+ "grad_norm": 1.7482995986938477,
256
+ "learning_rate": 3.983848290367661e-05,
257
+ "loss": 1.1757,
258
+ "step": 1600
259
+ },
260
+ {
261
+ "epoch": 3.3880903490759753,
262
+ "grad_norm": 1.8251243829727173,
263
+ "learning_rate": 3.9700584118182686e-05,
264
+ "loss": 1.1688,
265
+ "step": 1650
266
+ },
267
+ {
268
+ "epoch": 3.490759753593429,
269
+ "grad_norm": 1.9155806303024292,
270
+ "learning_rate": 3.9520864948993926e-05,
271
+ "loss": 1.1742,
272
+ "step": 1700
273
+ },
274
+ {
275
+ "epoch": 3.593429158110883,
276
+ "grad_norm": 1.8360294103622437,
277
+ "learning_rate": 3.929970690381425e-05,
278
+ "loss": 1.1474,
279
+ "step": 1750
280
+ },
281
+ {
282
+ "epoch": 3.696098562628337,
283
+ "grad_norm": 1.893100380897522,
284
+ "learning_rate": 3.903757945675931e-05,
285
+ "loss": 1.1596,
286
+ "step": 1800
287
+ },
288
+ {
289
+ "epoch": 3.798767967145791,
290
+ "grad_norm": 1.7409580945968628,
291
+ "learning_rate": 3.8735039051757204e-05,
292
+ "loss": 1.1496,
293
+ "step": 1850
294
+ },
295
+ {
296
+ "epoch": 3.9014373716632442,
297
+ "grad_norm": 1.4829061031341553,
298
+ "learning_rate": 3.83927279213298e-05,
299
+ "loss": 1.1262,
300
+ "step": 1900
301
+ },
302
+ {
303
+ "epoch": 4.0,
304
+ "eval_loss": 1.1556336879730225,
305
+ "eval_runtime": 396.6647,
306
+ "eval_samples_per_second": 4.908,
307
+ "eval_steps_per_second": 0.615,
308
+ "step": 1948
309
+ },
310
+ {
311
+ "epoch": 4.0041067761806985,
312
+ "grad_norm": 1.6514705419540405,
313
+ "learning_rate": 3.801137272326205e-05,
314
+ "loss": 1.1419,
315
+ "step": 1950
316
+ },
317
+ {
318
+ "epoch": 4.1067761806981515,
319
+ "grad_norm": 1.7151414155960083,
320
+ "learning_rate": 3.759178299805344e-05,
321
+ "loss": 1.1441,
322
+ "step": 2000
323
+ },
324
+ {
325
+ "epoch": 4.209445585215605,
326
+ "grad_norm": 1.5632063150405884,
327
+ "learning_rate": 3.713484945042608e-05,
328
+ "loss": 1.1271,
329
+ "step": 2050
330
+ },
331
+ {
332
+ "epoch": 4.312114989733059,
333
+ "grad_norm": 1.619882583618164,
334
+ "learning_rate": 3.664154205853749e-05,
335
+ "loss": 1.1016,
336
+ "step": 2100
337
+ },
338
+ {
339
+ "epoch": 4.414784394250513,
340
+ "grad_norm": 1.6811403036117554,
341
+ "learning_rate": 3.6112908014911744e-05,
342
+ "loss": 1.1203,
343
+ "step": 2150
344
+ },
345
+ {
346
+ "epoch": 4.517453798767967,
347
+ "grad_norm": 1.52841055393219,
348
+ "learning_rate": 3.555006950346002e-05,
349
+ "loss": 1.1152,
350
+ "step": 2200
351
+ },
352
+ {
353
+ "epoch": 4.620123203285421,
354
+ "grad_norm": 1.482682704925537,
355
+ "learning_rate": 3.4954221317309576e-05,
356
+ "loss": 1.1063,
357
+ "step": 2250
358
+ },
359
+ {
360
+ "epoch": 4.722792607802875,
361
+ "grad_norm": 1.508597731590271,
362
+ "learning_rate": 3.4326628322497834e-05,
363
+ "loss": 1.1095,
364
+ "step": 2300
365
+ },
366
+ {
367
+ "epoch": 4.825462012320329,
368
+ "grad_norm": 1.5862213373184204,
369
+ "learning_rate": 3.366862277291585e-05,
370
+ "loss": 1.1062,
371
+ "step": 2350
372
+ },
373
+ {
374
+ "epoch": 4.9281314168377826,
375
+ "grad_norm": 1.6203546524047852,
376
+ "learning_rate": 3.298160148220079e-05,
377
+ "loss": 1.1189,
378
+ "step": 2400
379
+ },
380
+ {
381
+ "epoch": 5.0,
382
+ "eval_loss": 1.1254490613937378,
383
+ "eval_runtime": 395.8276,
384
+ "eval_samples_per_second": 4.919,
385
+ "eval_steps_per_second": 0.616,
386
+ "step": 2435
387
+ },
388
+ {
389
+ "epoch": 5.030800821355236,
390
+ "grad_norm": 1.5772205591201782,
391
+ "learning_rate": 3.2267022858581186e-05,
392
+ "loss": 1.114,
393
+ "step": 2450
394
+ },
395
+ {
396
+ "epoch": 5.13347022587269,
397
+ "grad_norm": 1.4345210790634155,
398
+ "learning_rate": 3.152640380896904e-05,
399
+ "loss": 1.0856,
400
+ "step": 2500
401
+ },
402
+ {
403
+ "epoch": 5.236139630390143,
404
+ "grad_norm": 1.4504789113998413,
405
+ "learning_rate": 3.0761316518871184e-05,
406
+ "loss": 1.0866,
407
+ "step": 2550
408
+ },
409
+ {
410
+ "epoch": 5.338809034907597,
411
+ "grad_norm": 1.841271996498108,
412
+ "learning_rate": 2.997338511495513e-05,
413
+ "loss": 1.0819,
414
+ "step": 2600
415
+ },
416
+ {
417
+ "epoch": 5.441478439425051,
418
+ "grad_norm": 1.3801902532577515,
419
+ "learning_rate": 2.916428221735433e-05,
420
+ "loss": 1.1059,
421
+ "step": 2650
422
+ },
423
+ {
424
+ "epoch": 5.544147843942505,
425
+ "grad_norm": 1.5078575611114502,
426
+ "learning_rate": 2.8335725389031572e-05,
427
+ "loss": 1.0974,
428
+ "step": 2700
429
+ },
430
+ {
431
+ "epoch": 5.646817248459959,
432
+ "grad_norm": 1.6412862539291382,
433
+ "learning_rate": 2.7489473489737663e-05,
434
+ "loss": 1.1025,
435
+ "step": 2750
436
+ },
437
+ {
438
+ "epoch": 5.749486652977413,
439
+ "grad_norm": 1.6640971899032593,
440
+ "learning_rate": 2.6627322942305504e-05,
441
+ "loss": 1.0769,
442
+ "step": 2800
443
+ },
444
+ {
445
+ "epoch": 5.852156057494867,
446
+ "grad_norm": 1.8713879585266113,
447
+ "learning_rate": 2.57511039192051e-05,
448
+ "loss": 1.0739,
449
+ "step": 2850
450
+ },
451
+ {
452
+ "epoch": 5.95482546201232,
453
+ "grad_norm": 1.477726936340332,
454
+ "learning_rate": 2.4862676457455003e-05,
455
+ "loss": 1.0914,
456
+ "step": 2900
457
+ },
458
+ {
459
+ "epoch": 6.0,
460
+ "eval_loss": 1.1069694757461548,
461
+ "eval_runtime": 396.2551,
462
+ "eval_samples_per_second": 4.914,
463
+ "eval_steps_per_second": 0.616,
464
+ "step": 2922
465
+ },
466
+ {
467
+ "epoch": 6.057494866529774,
468
+ "grad_norm": 1.5570802688598633,
469
+ "learning_rate": 2.396392651013723e-05,
470
+ "loss": 1.0646,
471
+ "step": 2950
472
+ },
473
+ {
474
+ "epoch": 6.160164271047228,
475
+ "grad_norm": 1.5152437686920166,
476
+ "learning_rate": 2.3056761942897655e-05,
477
+ "loss": 1.0817,
478
+ "step": 3000
479
+ },
480
+ {
481
+ "epoch": 6.262833675564682,
482
+ "grad_norm": 1.4843331575393677,
483
+ "learning_rate": 2.2143108483930357e-05,
484
+ "loss": 1.0762,
485
+ "step": 3050
486
+ },
487
+ {
488
+ "epoch": 6.365503080082135,
489
+ "grad_norm": 1.504683256149292,
490
+ "learning_rate": 2.122490563604342e-05,
491
+ "loss": 1.0669,
492
+ "step": 3100
493
+ },
494
+ {
495
+ "epoch": 6.468172484599589,
496
+ "grad_norm": 1.5838227272033691,
497
+ "learning_rate": 2.0304102559483984e-05,
498
+ "loss": 1.0597,
499
+ "step": 3150
500
+ },
501
+ {
502
+ "epoch": 6.570841889117043,
503
+ "grad_norm": 1.61528480052948,
504
+ "learning_rate": 1.9382653934262424e-05,
505
+ "loss": 1.0839,
506
+ "step": 3200
507
+ },
508
+ {
509
+ "epoch": 6.673511293634497,
510
+ "grad_norm": 1.4133933782577515,
511
+ "learning_rate": 1.8462515810759347e-05,
512
+ "loss": 1.0798,
513
+ "step": 3250
514
+ },
515
+ {
516
+ "epoch": 6.776180698151951,
517
+ "grad_norm": 1.4809049367904663,
518
+ "learning_rate": 1.7545641457423334e-05,
519
+ "loss": 1.0666,
520
+ "step": 3300
521
+ },
522
+ {
523
+ "epoch": 6.878850102669404,
524
+ "grad_norm": 1.6875197887420654,
525
+ "learning_rate": 1.6633977214374384e-05,
526
+ "loss": 1.0671,
527
+ "step": 3350
528
+ },
529
+ {
530
+ "epoch": 6.981519507186858,
531
+ "grad_norm": 1.5088533163070679,
532
+ "learning_rate": 1.5729458361714657e-05,
533
+ "loss": 1.0756,
534
+ "step": 3400
535
+ },
536
+ {
537
+ "epoch": 7.0,
538
+ "eval_loss": 1.0947245359420776,
539
+ "eval_runtime": 396.3527,
540
+ "eval_samples_per_second": 4.912,
541
+ "eval_steps_per_second": 0.616,
542
+ "step": 3409
543
+ },
544
+ {
545
+ "epoch": 7.084188911704312,
546
+ "grad_norm": 1.9185519218444824,
547
+ "learning_rate": 1.4834005011317545e-05,
548
+ "loss": 1.055,
549
+ "step": 3450
550
+ },
551
+ {
552
+ "epoch": 7.186858316221766,
553
+ "grad_norm": 1.6549229621887207,
554
+ "learning_rate": 1.3949518030815667e-05,
555
+ "loss": 1.0565,
556
+ "step": 3500
557
+ },
558
+ {
559
+ "epoch": 7.28952772073922,
560
+ "grad_norm": 1.5964972972869873,
561
+ "learning_rate": 1.3077875008440714e-05,
562
+ "loss": 1.0696,
563
+ "step": 3550
564
+ },
565
+ {
566
+ "epoch": 7.392197125256674,
567
+ "grad_norm": 1.5329166650772095,
568
+ "learning_rate": 1.2220926267280677e-05,
569
+ "loss": 1.0657,
570
+ "step": 3600
571
+ },
572
+ {
573
+ "epoch": 7.494866529774128,
574
+ "grad_norm": 1.6698694229125977,
575
+ "learning_rate": 1.1380490937415516e-05,
576
+ "loss": 1.0596,
577
+ "step": 3650
578
+ },
579
+ {
580
+ "epoch": 7.597535934291582,
581
+ "grad_norm": 1.5411231517791748,
582
+ "learning_rate": 1.0558353094269314e-05,
583
+ "loss": 1.059,
584
+ "step": 3700
585
+ },
586
+ {
587
+ "epoch": 7.700205338809035,
588
+ "grad_norm": 1.7882490158081055,
589
+ "learning_rate": 9.756257971376489e-06,
590
+ "loss": 1.0731,
591
+ "step": 3750
592
+ },
593
+ {
594
+ "epoch": 7.8028747433264884,
595
+ "grad_norm": 1.5504204034805298,
596
+ "learning_rate": 8.975908255601452e-06,
597
+ "loss": 1.0565,
598
+ "step": 3800
599
+ },
600
+ {
601
+ "epoch": 7.905544147843942,
602
+ "grad_norm": 1.4786473512649536,
603
+ "learning_rate": 8.21896047267632e-06,
604
+ "loss": 1.0502,
605
+ "step": 3850
606
+ },
607
+ {
608
+ "epoch": 8.0,
609
+ "eval_loss": 1.0888762474060059,
610
+ "eval_runtime": 396.8423,
611
+ "eval_samples_per_second": 4.906,
612
+ "eval_steps_per_second": 0.615,
613
+ "step": 3896
614
+ },
615
+ {
616
+ "epoch": 8.008213552361397,
617
+ "grad_norm": 1.6214848756790161,
618
+ "learning_rate": 7.487021470729523e-06,
619
+ "loss": 1.0634,
620
+ "step": 3900
621
+ },
622
+ {
623
+ "epoch": 8.11088295687885,
624
+ "grad_norm": 1.712120771408081,
625
+ "learning_rate": 6.781645009269846e-06,
626
+ "loss": 1.062,
627
+ "step": 3950
628
+ },
629
+ {
630
+ "epoch": 8.213552361396303,
631
+ "grad_norm": 1.5750735998153687,
632
+ "learning_rate": 6.1043284608671e-06,
633
+ "loss": 1.0387,
634
+ "step": 4000
635
+ },
636
+ {
637
+ "epoch": 8.316221765913758,
638
+ "grad_norm": 1.5793296098709106,
639
+ "learning_rate": 5.45650963253082e-06,
640
+ "loss": 1.0594,
641
+ "step": 4050
642
+ },
643
+ {
644
+ "epoch": 8.41889117043121,
645
+ "grad_norm": 1.445952296257019,
646
+ "learning_rate": 4.839563713534743e-06,
647
+ "loss": 1.0614,
648
+ "step": 4100
649
+ },
650
+ {
651
+ "epoch": 8.521560574948666,
652
+ "grad_norm": 1.731099009513855,
653
+ "learning_rate": 4.254800356166155e-06,
654
+ "loss": 1.0599,
655
+ "step": 4150
656
+ },
657
+ {
658
+ "epoch": 8.624229979466119,
659
+ "grad_norm": 1.7094162702560425,
660
+ "learning_rate": 3.7034608955971175e-06,
661
+ "loss": 1.0437,
662
+ "step": 4200
663
+ },
664
+ {
665
+ "epoch": 8.726899383983573,
666
+ "grad_norm": 1.6651116609573364,
667
+ "learning_rate": 3.1867157147791585e-06,
668
+ "loss": 1.0667,
669
+ "step": 4250
670
+ },
671
+ {
672
+ "epoch": 8.829568788501026,
673
+ "grad_norm": 1.5388877391815186,
674
+ "learning_rate": 2.705661759955318e-06,
675
+ "loss": 1.0529,
676
+ "step": 4300
677
+ },
678
+ {
679
+ "epoch": 8.932238193018481,
680
+ "grad_norm": 1.7173577547073364,
681
+ "learning_rate": 2.261320212063589e-06,
682
+ "loss": 1.0619,
683
+ "step": 4350
684
+ },
685
+ {
686
+ "epoch": 9.0,
687
+ "eval_loss": 1.0863490104675293,
688
+ "eval_runtime": 395.7285,
689
+ "eval_samples_per_second": 4.92,
690
+ "eval_steps_per_second": 0.617,
691
+ "step": 4383
692
+ },
693
+ {
694
+ "epoch": 9.034907597535934,
695
+ "grad_norm": 1.79407799243927,
696
+ "learning_rate": 1.854634318974835e-06,
697
+ "loss": 1.0494,
698
+ "step": 4400
699
+ },
700
+ {
701
+ "epoch": 9.137577002053389,
702
+ "grad_norm": 1.6543513536453247,
703
+ "learning_rate": 1.486467393167017e-06,
704
+ "loss": 1.0497,
705
+ "step": 4450
706
+ },
707
+ {
708
+ "epoch": 9.240246406570842,
709
+ "grad_norm": 1.5759410858154297,
710
+ "learning_rate": 1.1576009790861598e-06,
711
+ "loss": 1.0503,
712
+ "step": 4500
713
+ },
714
+ {
715
+ "epoch": 9.342915811088295,
716
+ "grad_norm": 1.4627271890640259,
717
+ "learning_rate": 8.687331940844545e-07,
718
+ "loss": 1.0585,
719
+ "step": 4550
720
+ },
721
+ {
722
+ "epoch": 9.44558521560575,
723
+ "grad_norm": 1.7757667303085327,
724
+ "learning_rate": 6.204772464573161e-07,
725
+ "loss": 1.0468,
726
+ "step": 4600
727
+ },
728
+ {
729
+ "epoch": 9.548254620123203,
730
+ "grad_norm": 1.7230888605117798,
731
+ "learning_rate": 4.1336013372533125e-07,
732
+ "loss": 1.0485,
733
+ "step": 4650
734
+ },
735
+ {
736
+ "epoch": 9.650924024640657,
737
+ "grad_norm": 1.7910829782485962,
738
+ "learning_rate": 2.478215239243409e-07,
739
+ "loss": 1.0498,
740
+ "step": 4700
741
+ },
742
+ {
743
+ "epoch": 9.75359342915811,
744
+ "grad_norm": 1.7358592748641968,
745
+ "learning_rate": 1.2421282227850662e-07,
746
+ "loss": 1.0548,
747
+ "step": 4750
748
+ },
749
+ {
750
+ "epoch": 9.856262833675565,
751
+ "grad_norm": 1.603044867515564,
752
+ "learning_rate": 4.2796425237592306e-08,
753
+ "loss": 1.0589,
754
+ "step": 4800
755
+ },
756
+ {
757
+ "epoch": 9.958932238193018,
758
+ "grad_norm": 1.4900743961334229,
759
+ "learning_rate": 3.745163461994494e-09,
760
+ "loss": 1.056,
761
+ "step": 4850
762
+ },
763
+ {
764
+ "epoch": 10.0,
765
+ "eval_loss": 1.0858850479125977,
766
+ "eval_runtime": 396.092,
767
+ "eval_samples_per_second": 4.916,
768
+ "eval_steps_per_second": 0.616,
769
+ "step": 4870
770
+ }
771
+ ],
772
+ "logging_steps": 50,
773
+ "max_steps": 4870,
774
+ "num_input_tokens_seen": 0,
775
+ "num_train_epochs": 10,
776
+ "save_steps": 500,
777
+ "stateful_callbacks": {
778
+ "TrainerControl": {
779
+ "args": {
780
+ "should_epoch_stop": false,
781
+ "should_evaluate": false,
782
+ "should_log": false,
783
+ "should_save": true,
784
+ "should_training_stop": true
785
+ },
786
+ "attributes": {}
787
+ }
788
+ },
789
+ "total_flos": 6.27791813148672e+17,
790
+ "train_batch_size": 8,
791
+ "trial_name": null,
792
+ "trial_params": null
793
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:265759dbde48001f5c535a55f68b501c70f86b72f86eefab85437e26517a9473
3
+ size 5560
vocab.json ADDED
The diff for this file is too large to render. See raw diff