zclfe commited on
Commit
c050228
·
verified ·
1 Parent(s): 9ceaf43

Add files using upload-large-folder tool

Browse files
Files changed (6) hide show
  1. README.md +200 -0
  2. adapter_config.json +31 -0
  3. merges.txt +0 -0
  4. tokenizer_config.json +236 -0
  5. trainer_state.json +384 -0
  6. vocab.json +0 -0
README.md ADDED
@@ -0,0 +1,200 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ ---
3
+
4
+ # Model Card for Model ID
5
+
6
+ <!-- Provide a quick summary of what the model is/does. -->
7
+
8
+
9
+
10
+ ## Model Details
11
+
12
+ ### Model Description
13
+
14
+ <!-- Provide a longer summary of what this model is. -->
15
+
16
+
17
+
18
+ - **Developed by:** [More Information Needed]
19
+ - **Funded by [optional]:** [More Information Needed]
20
+ - **Shared by [optional]:** [More Information Needed]
21
+ - **Model type:** [More Information Needed]
22
+ - **Language(s) (NLP):** [More Information Needed]
23
+ - **License:** [More Information Needed]
24
+ - **Finetuned from model [optional]:** [More Information Needed]
25
+
26
+ ### Model Sources [optional]
27
+
28
+ <!-- Provide the basic links for the model. -->
29
+
30
+ - **Repository:** [More Information Needed]
31
+ - **Paper [optional]:** [More Information Needed]
32
+ - **Demo [optional]:** [More Information Needed]
33
+
34
+ ## Uses
35
+
36
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
37
+
38
+ ### Direct Use
39
+
40
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
41
+
42
+ [More Information Needed]
43
+
44
+ ### Downstream Use [optional]
45
+
46
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
47
+
48
+ [More Information Needed]
49
+
50
+ ### Out-of-Scope Use
51
+
52
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
53
+
54
+ [More Information Needed]
55
+
56
+ ## Bias, Risks, and Limitations
57
+
58
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
59
+
60
+ [More Information Needed]
61
+
62
+ ### Recommendations
63
+
64
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
65
+
66
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
67
+
68
+ ## How to Get Started with the Model
69
+
70
+ Use the code below to get started with the model.
71
+
72
+ [More Information Needed]
73
+
74
+ ## Training Details
75
+
76
+ ### Training Data
77
+
78
+ <!-- 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. -->
79
+
80
+ [More Information Needed]
81
+
82
+ ### Training Procedure
83
+
84
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
85
+
86
+ #### Preprocessing [optional]
87
+
88
+ [More Information Needed]
89
+
90
+
91
+ #### Training Hyperparameters
92
+
93
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
94
+
95
+ #### Speeds, Sizes, Times [optional]
96
+
97
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
98
+
99
+ [More Information Needed]
100
+
101
+ ## Evaluation
102
+
103
+ <!-- This section describes the evaluation protocols and provides the results. -->
104
+
105
+ ### Testing Data, Factors & Metrics
106
+
107
+ #### Testing Data
108
+
109
+ <!-- This should link to a Dataset Card if possible. -->
110
+
111
+ [More Information Needed]
112
+
113
+ #### Factors
114
+
115
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
116
+
117
+ [More Information Needed]
118
+
119
+ #### Metrics
120
+
121
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
122
+
123
+ [More Information Needed]
124
+
125
+ ### Results
126
+
127
+ [More Information Needed]
128
+
129
+ #### Summary
130
+
131
+
132
+
133
+ ## Model Examination [optional]
134
+
135
+ <!-- Relevant interpretability work for the model goes here -->
136
+
137
+ [More Information Needed]
138
+
139
+ ## Environmental Impact
140
+
141
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
142
+
143
+ 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).
144
+
145
+ - **Hardware Type:** [More Information Needed]
146
+ - **Hours used:** [More Information Needed]
147
+ - **Cloud Provider:** [More Information Needed]
148
+ - **Compute Region:** [More Information Needed]
149
+ - **Carbon Emitted:** [More Information Needed]
150
+
151
+ ## Technical Specifications [optional]
152
+
153
+ ### Model Architecture and Objective
154
+
155
+ [More Information Needed]
156
+
157
+ ### Compute Infrastructure
158
+
159
+ [More Information Needed]
160
+
161
+ #### Hardware
162
+
163
+ [More Information Needed]
164
+
165
+ #### Software
166
+
167
+ [More Information Needed]
168
+
169
+ ## Citation [optional]
170
+
171
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
172
+
173
+ **BibTeX:**
174
+
175
+ [More Information Needed]
176
+
177
+ **APA:**
178
+
179
+ [More Information Needed]
180
+
181
+ ## Glossary [optional]
182
+
183
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
184
+
185
+ [More Information Needed]
186
+
187
+ ## More Information [optional]
188
+
189
+ [More Information Needed]
190
+
191
+ ## Model Card Authors [optional]
192
+
193
+ [More Information Needed]
194
+
195
+ ## Model Card Contact
196
+
197
+ [More Information Needed]
198
+ ### Framework versions
199
+
200
+ - PEFT 0.11.1
adapter_config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/mnt/tidal-alsh01/dataset/DQARanker/NoteRAG/train_log/ppr_dpo_qwen_process_step2_dp0.1_epoch2_beta0.5_lr1e-4/vllm_ready_model",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.1,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 8,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "q_proj",
24
+ "k_proj",
25
+ "v_proj",
26
+ "o_proj"
27
+ ],
28
+ "task_type": "CAUSAL_LM",
29
+ "use_dora": false,
30
+ "use_rslora": false
31
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,236 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "151665": {
182
+ "content": "<Reference>",
183
+ "lstrip": false,
184
+ "normalized": true,
185
+ "rstrip": false,
186
+ "single_word": false,
187
+ "special": false
188
+ },
189
+ "151666": {
190
+ "content": "</Reference>",
191
+ "lstrip": false,
192
+ "normalized": true,
193
+ "rstrip": false,
194
+ "single_word": false,
195
+ "special": false
196
+ },
197
+ "151667": {
198
+ "content": "<|source_end|>",
199
+ "lstrip": false,
200
+ "normalized": true,
201
+ "rstrip": false,
202
+ "single_word": false,
203
+ "special": false
204
+ }
205
+ },
206
+ "additional_special_tokens": [
207
+ "<|im_start|>",
208
+ "<|im_end|>",
209
+ "<|object_ref_start|>",
210
+ "<|object_ref_end|>",
211
+ "<|box_start|>",
212
+ "<|box_end|>",
213
+ "<|quad_start|>",
214
+ "<|quad_end|>",
215
+ "<|vision_start|>",
216
+ "<|vision_end|>",
217
+ "<|vision_pad|>",
218
+ "<|image_pad|>",
219
+ "<|video_pad|>"
220
+ ],
221
+ "bos_token": null,
222
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\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>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\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\" }}\n{%- else %}\n {%- if messages[0]['role'] == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n {%- else %}\n {{- '<|im_start|>system\\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
223
+ "clean_up_tokenization_spaces": false,
224
+ "eos_token": "<|im_end|>",
225
+ "errors": "replace",
226
+ "extra_special_tokens": {},
227
+ "max_length": null,
228
+ "model_max_length": 40000,
229
+ "pad_to_multiple_of": null,
230
+ "pad_token": "<|endoftext|>",
231
+ "pad_token_type_id": 0,
232
+ "padding_side": "left",
233
+ "split_special_tokens": false,
234
+ "tokenizer_class": "Qwen2Tokenizer",
235
+ "unk_token": null
236
+ }
trainer_state.json ADDED
@@ -0,0 +1,384 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 33,
7
+ "global_step": 167,
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.059880239520958084,
14
+ "grad_norm": 12.36841869354248,
15
+ "learning_rate": 9.936708860759493e-05,
16
+ "logits/chosen": -0.735156238079071,
17
+ "logits/rejected": -0.6986328363418579,
18
+ "logps/chosen": -19.024999618530273,
19
+ "logps/rejected": -28.543750762939453,
20
+ "loss": 0.5971,
21
+ "nll_loss": 0.20131835341453552,
22
+ "rewards/accuracies": 0.4937500059604645,
23
+ "rewards/chosen": 0.01967773400247097,
24
+ "rewards/margins": 0.14492186903953552,
25
+ "rewards/rejected": -0.12529297173023224,
26
+ "step": 10
27
+ },
28
+ {
29
+ "epoch": 0.11976047904191617,
30
+ "grad_norm": 8.02683162689209,
31
+ "learning_rate": 9.303797468354431e-05,
32
+ "logits/chosen": -0.865039050579071,
33
+ "logits/rejected": -0.849804699420929,
34
+ "logps/chosen": -17.296875,
35
+ "logps/rejected": -28.184375762939453,
36
+ "loss": 0.547,
37
+ "nll_loss": 0.18623046576976776,
38
+ "rewards/accuracies": 0.6187499761581421,
39
+ "rewards/chosen": -0.09682617336511612,
40
+ "rewards/margins": 0.814404308795929,
41
+ "rewards/rejected": -0.91064453125,
42
+ "step": 20
43
+ },
44
+ {
45
+ "epoch": 0.17964071856287425,
46
+ "grad_norm": 10.350639343261719,
47
+ "learning_rate": 8.670886075949367e-05,
48
+ "logits/chosen": -0.9947265386581421,
49
+ "logits/rejected": -0.942187488079071,
50
+ "logps/chosen": -18.146875381469727,
51
+ "logps/rejected": -30.168750762939453,
52
+ "loss": 0.5183,
53
+ "nll_loss": 0.19204100966453552,
54
+ "rewards/accuracies": 0.6812499761581421,
55
+ "rewards/chosen": -0.14462891221046448,
56
+ "rewards/margins": 1.4348633289337158,
57
+ "rewards/rejected": -1.5797851085662842,
58
+ "step": 30
59
+ },
60
+ {
61
+ "epoch": 0.19760479041916168,
62
+ "eval_logits/chosen": -1.0199424028396606,
63
+ "eval_logits/rejected": -0.9895148277282715,
64
+ "eval_logps/chosen": -18.636512756347656,
65
+ "eval_logps/rejected": -31.595394134521484,
66
+ "eval_loss": 0.5152081847190857,
67
+ "eval_nll_loss": 0.19370631873607635,
68
+ "eval_rewards/accuracies": 0.6973684430122375,
69
+ "eval_rewards/chosen": -0.248046875,
70
+ "eval_rewards/margins": 1.5133634805679321,
71
+ "eval_rewards/rejected": -1.7625411748886108,
72
+ "eval_runtime": 37.7687,
73
+ "eval_samples_per_second": 7.89,
74
+ "eval_steps_per_second": 1.006,
75
+ "step": 33
76
+ },
77
+ {
78
+ "epoch": 0.23952095808383234,
79
+ "grad_norm": 12.673577308654785,
80
+ "learning_rate": 8.037974683544304e-05,
81
+ "logits/chosen": -1.006250023841858,
82
+ "logits/rejected": -0.9486328363418579,
83
+ "logps/chosen": -18.325000762939453,
84
+ "logps/rejected": -30.96875,
85
+ "loss": 0.5276,
86
+ "nll_loss": 0.19785156846046448,
87
+ "rewards/accuracies": 0.675000011920929,
88
+ "rewards/chosen": -0.36396485567092896,
89
+ "rewards/margins": 1.570214867591858,
90
+ "rewards/rejected": -1.9357421398162842,
91
+ "step": 40
92
+ },
93
+ {
94
+ "epoch": 0.2994011976047904,
95
+ "grad_norm": 14.858013153076172,
96
+ "learning_rate": 7.40506329113924e-05,
97
+ "logits/chosen": -0.971484363079071,
98
+ "logits/rejected": -0.9488281011581421,
99
+ "logps/chosen": -19.975000381469727,
100
+ "logps/rejected": -30.5,
101
+ "loss": 0.5554,
102
+ "nll_loss": 0.21669921278953552,
103
+ "rewards/accuracies": 0.637499988079071,
104
+ "rewards/chosen": -0.32304686307907104,
105
+ "rewards/margins": 1.359960913658142,
106
+ "rewards/rejected": -1.684472680091858,
107
+ "step": 50
108
+ },
109
+ {
110
+ "epoch": 0.3592814371257485,
111
+ "grad_norm": 13.94198226928711,
112
+ "learning_rate": 6.772151898734177e-05,
113
+ "logits/chosen": -0.9164062738418579,
114
+ "logits/rejected": -0.895703136920929,
115
+ "logps/chosen": -19.456249237060547,
116
+ "logps/rejected": -31.3125,
117
+ "loss": 0.5578,
118
+ "nll_loss": 0.20249024033546448,
119
+ "rewards/accuracies": 0.6625000238418579,
120
+ "rewards/chosen": -0.30244141817092896,
121
+ "rewards/margins": 1.196874976158142,
122
+ "rewards/rejected": -1.50048828125,
123
+ "step": 60
124
+ },
125
+ {
126
+ "epoch": 0.39520958083832336,
127
+ "eval_logits/chosen": -0.8888774514198303,
128
+ "eval_logits/rejected": -0.8612253069877625,
129
+ "eval_logps/chosen": -18.394737243652344,
130
+ "eval_logps/rejected": -30.789474487304688,
131
+ "eval_loss": 0.46207094192504883,
132
+ "eval_nll_loss": 0.1904296875,
133
+ "eval_rewards/accuracies": 0.7532894611358643,
134
+ "eval_rewards/chosen": -0.1289319545030594,
135
+ "eval_rewards/margins": 1.2322677373886108,
136
+ "eval_rewards/rejected": -1.3603515625,
137
+ "eval_runtime": 37.6185,
138
+ "eval_samples_per_second": 7.922,
139
+ "eval_steps_per_second": 1.01,
140
+ "step": 66
141
+ },
142
+ {
143
+ "epoch": 0.41916167664670656,
144
+ "grad_norm": 10.616917610168457,
145
+ "learning_rate": 6.139240506329115e-05,
146
+ "logits/chosen": -0.826953113079071,
147
+ "logits/rejected": -0.830273449420929,
148
+ "logps/chosen": -20.596874237060547,
149
+ "logps/rejected": -29.740625381469727,
150
+ "loss": 0.5122,
151
+ "nll_loss": 0.20893554389476776,
152
+ "rewards/accuracies": 0.612500011920929,
153
+ "rewards/chosen": -0.11264648288488388,
154
+ "rewards/margins": 1.2311522960662842,
155
+ "rewards/rejected": -1.3449218273162842,
156
+ "step": 70
157
+ },
158
+ {
159
+ "epoch": 0.47904191616766467,
160
+ "grad_norm": 11.696365356445312,
161
+ "learning_rate": 5.5063291139240514e-05,
162
+ "logits/chosen": -0.82421875,
163
+ "logits/rejected": -0.8033202886581421,
164
+ "logps/chosen": -19.581249237060547,
165
+ "logps/rejected": -32.271873474121094,
166
+ "loss": 0.5047,
167
+ "nll_loss": 0.19770507514476776,
168
+ "rewards/accuracies": 0.675000011920929,
169
+ "rewards/chosen": -0.20634765923023224,
170
+ "rewards/margins": 1.2736327648162842,
171
+ "rewards/rejected": -1.4802734851837158,
172
+ "step": 80
173
+ },
174
+ {
175
+ "epoch": 0.5389221556886228,
176
+ "grad_norm": 11.490265846252441,
177
+ "learning_rate": 4.8734177215189874e-05,
178
+ "logits/chosen": -0.863476574420929,
179
+ "logits/rejected": -0.841015636920929,
180
+ "logps/chosen": -19.484375,
181
+ "logps/rejected": -33.95624923706055,
182
+ "loss": 0.4647,
183
+ "nll_loss": 0.20292969048023224,
184
+ "rewards/accuracies": 0.668749988079071,
185
+ "rewards/chosen": -0.24423828721046448,
186
+ "rewards/margins": 1.6508300304412842,
187
+ "rewards/rejected": -1.8943359851837158,
188
+ "step": 90
189
+ },
190
+ {
191
+ "epoch": 0.592814371257485,
192
+ "eval_logits/chosen": -0.9116981625556946,
193
+ "eval_logits/rejected": -0.8838404417037964,
194
+ "eval_logps/chosen": -18.429275512695312,
195
+ "eval_logps/rejected": -31.63157844543457,
196
+ "eval_loss": 0.43091967701911926,
197
+ "eval_nll_loss": 0.19037829339504242,
198
+ "eval_rewards/accuracies": 0.7532894611358643,
199
+ "eval_rewards/chosen": -0.15003083646297455,
200
+ "eval_rewards/margins": 1.6260793209075928,
201
+ "eval_rewards/rejected": -1.7764699459075928,
202
+ "eval_runtime": 37.7297,
203
+ "eval_samples_per_second": 7.898,
204
+ "eval_steps_per_second": 1.007,
205
+ "step": 99
206
+ },
207
+ {
208
+ "epoch": 0.5988023952095808,
209
+ "grad_norm": 8.535696983337402,
210
+ "learning_rate": 4.240506329113924e-05,
211
+ "logits/chosen": -0.848828136920929,
212
+ "logits/rejected": -0.8291015625,
213
+ "logps/chosen": -18.943750381469727,
214
+ "logps/rejected": -29.356250762939453,
215
+ "loss": 0.4623,
216
+ "nll_loss": 0.19814452528953552,
217
+ "rewards/accuracies": 0.699999988079071,
218
+ "rewards/chosen": -0.10761718451976776,
219
+ "rewards/margins": 1.6233398914337158,
220
+ "rewards/rejected": -1.7314453125,
221
+ "step": 100
222
+ },
223
+ {
224
+ "epoch": 0.6586826347305389,
225
+ "grad_norm": 5.726072311401367,
226
+ "learning_rate": 3.607594936708861e-05,
227
+ "logits/chosen": -0.8636718988418579,
228
+ "logits/rejected": -0.866992175579071,
229
+ "logps/chosen": -20.371875762939453,
230
+ "logps/rejected": -31.131250381469727,
231
+ "loss": 0.477,
232
+ "nll_loss": 0.212890625,
233
+ "rewards/accuracies": 0.668749988079071,
234
+ "rewards/chosen": -0.22114257514476776,
235
+ "rewards/margins": 1.5869140625,
236
+ "rewards/rejected": -1.808203101158142,
237
+ "step": 110
238
+ },
239
+ {
240
+ "epoch": 0.718562874251497,
241
+ "grad_norm": 6.890474796295166,
242
+ "learning_rate": 2.9746835443037974e-05,
243
+ "logits/chosen": -0.9125000238418579,
244
+ "logits/rejected": -0.9126952886581421,
245
+ "logps/chosen": -17.600000381469727,
246
+ "logps/rejected": -33.837501525878906,
247
+ "loss": 0.4372,
248
+ "nll_loss": 0.185546875,
249
+ "rewards/accuracies": 0.7562500238418579,
250
+ "rewards/chosen": -0.20854492485523224,
251
+ "rewards/margins": 2.422656297683716,
252
+ "rewards/rejected": -2.6328125,
253
+ "step": 120
254
+ },
255
+ {
256
+ "epoch": 0.7784431137724551,
257
+ "grad_norm": 7.191044807434082,
258
+ "learning_rate": 2.341772151898734e-05,
259
+ "logits/chosen": -0.93359375,
260
+ "logits/rejected": -0.8980468511581421,
261
+ "logps/chosen": -16.762500762939453,
262
+ "logps/rejected": -30.174999237060547,
263
+ "loss": 0.4398,
264
+ "nll_loss": 0.18398436903953552,
265
+ "rewards/accuracies": 0.7562500238418579,
266
+ "rewards/chosen": -0.18193359673023224,
267
+ "rewards/margins": 1.81787109375,
268
+ "rewards/rejected": -2.0015625953674316,
269
+ "step": 130
270
+ },
271
+ {
272
+ "epoch": 0.7904191616766467,
273
+ "eval_logits/chosen": -0.9958881735801697,
274
+ "eval_logits/rejected": -0.9672080874443054,
275
+ "eval_logps/chosen": -18.569900512695312,
276
+ "eval_logps/rejected": -32.644737243652344,
277
+ "eval_loss": 0.41646456718444824,
278
+ "eval_nll_loss": 0.19179172813892365,
279
+ "eval_rewards/accuracies": 0.75,
280
+ "eval_rewards/chosen": -0.2185187041759491,
281
+ "eval_rewards/margins": 2.0729339122772217,
282
+ "eval_rewards/rejected": -2.2922492027282715,
283
+ "eval_runtime": 37.9044,
284
+ "eval_samples_per_second": 7.862,
285
+ "eval_steps_per_second": 1.003,
286
+ "step": 132
287
+ },
288
+ {
289
+ "epoch": 0.8383233532934131,
290
+ "grad_norm": 10.231584548950195,
291
+ "learning_rate": 1.7088607594936708e-05,
292
+ "logits/chosen": -0.9898437261581421,
293
+ "logits/rejected": -0.976757824420929,
294
+ "logps/chosen": -17.987499237060547,
295
+ "logps/rejected": -31.737499237060547,
296
+ "loss": 0.3967,
297
+ "nll_loss": 0.19194336235523224,
298
+ "rewards/accuracies": 0.7875000238418579,
299
+ "rewards/chosen": -0.28300780057907104,
300
+ "rewards/margins": 1.9914062023162842,
301
+ "rewards/rejected": -2.274218797683716,
302
+ "step": 140
303
+ },
304
+ {
305
+ "epoch": 0.8982035928143712,
306
+ "grad_norm": 8.393524169921875,
307
+ "learning_rate": 1.0759493670886076e-05,
308
+ "logits/chosen": -0.975390613079071,
309
+ "logits/rejected": -0.953906238079071,
310
+ "logps/chosen": -19.209375381469727,
311
+ "logps/rejected": -33.400001525878906,
312
+ "loss": 0.5022,
313
+ "nll_loss": 0.20380859076976776,
314
+ "rewards/accuracies": 0.71875,
315
+ "rewards/chosen": -0.4892578125,
316
+ "rewards/margins": 1.9375,
317
+ "rewards/rejected": -2.4291014671325684,
318
+ "step": 150
319
+ },
320
+ {
321
+ "epoch": 0.9580838323353293,
322
+ "grad_norm": 10.486552238464355,
323
+ "learning_rate": 4.430379746835443e-06,
324
+ "logits/chosen": -0.991992175579071,
325
+ "logits/rejected": -0.948046863079071,
326
+ "logps/chosen": -19.268749237060547,
327
+ "logps/rejected": -33.556251525878906,
328
+ "loss": 0.5564,
329
+ "nll_loss": 0.206787109375,
330
+ "rewards/accuracies": 0.675000011920929,
331
+ "rewards/chosen": -0.40800780057907104,
332
+ "rewards/margins": 1.840429663658142,
333
+ "rewards/rejected": -2.244873046875,
334
+ "step": 160
335
+ },
336
+ {
337
+ "epoch": 0.9880239520958084,
338
+ "eval_logits/chosen": -1.0091488361358643,
339
+ "eval_logits/rejected": -0.98046875,
340
+ "eval_logps/chosen": -18.596216201782227,
341
+ "eval_logps/rejected": -32.71381759643555,
342
+ "eval_loss": 0.41747432947158813,
343
+ "eval_nll_loss": 0.19197162985801697,
344
+ "eval_rewards/accuracies": 0.7730262875556946,
345
+ "eval_rewards/chosen": -0.22913239896297455,
346
+ "eval_rewards/margins": 2.0992496013641357,
347
+ "eval_rewards/rejected": -2.329050064086914,
348
+ "eval_runtime": 37.8726,
349
+ "eval_samples_per_second": 7.868,
350
+ "eval_steps_per_second": 1.003,
351
+ "step": 165
352
+ },
353
+ {
354
+ "epoch": 1.0,
355
+ "step": 167,
356
+ "total_flos": 0.0,
357
+ "train_loss": 0.5023956755678097,
358
+ "train_runtime": 947.2222,
359
+ "train_samples_per_second": 2.816,
360
+ "train_steps_per_second": 0.176
361
+ }
362
+ ],
363
+ "logging_steps": 10,
364
+ "max_steps": 167,
365
+ "num_input_tokens_seen": 0,
366
+ "num_train_epochs": 1,
367
+ "save_steps": 33,
368
+ "stateful_callbacks": {
369
+ "TrainerControl": {
370
+ "args": {
371
+ "should_epoch_stop": false,
372
+ "should_evaluate": false,
373
+ "should_log": false,
374
+ "should_save": true,
375
+ "should_training_stop": true
376
+ },
377
+ "attributes": {}
378
+ }
379
+ },
380
+ "total_flos": 0.0,
381
+ "train_batch_size": 1,
382
+ "trial_name": null,
383
+ "trial_params": null
384
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff