adamhao123 commited on
Commit
b5b78c1
·
verified ·
1 Parent(s): 2ab0ee6

Model save

Browse files
README.md CHANGED
@@ -1,11 +1,9 @@
1
  ---
2
  base_model: deepseek-ai/DeepSeek-R1-Distill-Qwen-32B
3
- datasets: jdy_analysis
4
  library_name: transformers
5
  model_name: Qwen2.5-32B-Open-R1-Distill-jdy-ft
6
  tags:
7
  - generated_from_trainer
8
- - open-r1
9
  - trl
10
  - sft
11
  licence: license
@@ -13,7 +11,7 @@ licence: license
13
 
14
  # Model Card for Qwen2.5-32B-Open-R1-Distill-jdy-ft
15
 
16
- This model is a fine-tuned version of [deepseek-ai/DeepSeek-R1-Distill-Qwen-32B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B) on the [jdy_analysis](https://huggingface.co/datasets/jdy_analysis) dataset.
17
  It has been trained using [TRL](https://github.com/huggingface/trl).
18
 
19
  ## Quick start
@@ -29,7 +27,7 @@ print(output["generated_text"])
29
 
30
  ## Training procedure
31
 
32
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/songhao9021-9uest/huggingface/runs/b9jksdu6)
33
 
34
 
35
  This model was trained with SFT.
 
1
  ---
2
  base_model: deepseek-ai/DeepSeek-R1-Distill-Qwen-32B
 
3
  library_name: transformers
4
  model_name: Qwen2.5-32B-Open-R1-Distill-jdy-ft
5
  tags:
6
  - generated_from_trainer
 
7
  - trl
8
  - sft
9
  licence: license
 
11
 
12
  # Model Card for Qwen2.5-32B-Open-R1-Distill-jdy-ft
13
 
14
+ This model is a fine-tuned version of [deepseek-ai/DeepSeek-R1-Distill-Qwen-32B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
27
 
28
  ## Training procedure
29
 
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/songhao9021-9uest/huggingface/runs/mvb67nds)
31
 
32
 
33
  This model was trained with SFT.
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "total_flos": 978754951168.0,
3
- "train_loss": 1.4972400459749946,
4
- "train_runtime": 1197.8134,
5
  "train_samples": 114,
6
- "train_samples_per_second": 0.095,
7
- "train_steps_per_second": 0.024
8
  }
 
1
  {
2
+ "total_flos": 955996483584.0,
3
+ "train_loss": 1.542164264054134,
4
+ "train_runtime": 1339.7897,
5
  "train_samples": 114,
6
+ "train_samples_per_second": 0.085,
7
+ "train_steps_per_second": 0.022
8
  }
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.49.0",
26
- "use_cache": true,
27
  "use_sliding_window": false,
28
  "vocab_size": 152064
29
  }
 
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.49.0",
26
+ "use_cache": false,
27
  "use_sliding_window": false,
28
  "vocab_size": 152064
29
  }
model-00001-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c98ad299bfb0ad21d3f4f30c9b46bd079c5abffddb796045f20d9a6235745778
3
  size 4891730992
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:723e9cc765fe6d7e1298eb0b8a8a36aafb71c64b3afc05efb71b329ca7e3ca03
3
  size 4891730992
model-00002-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8c239207fe1daed298851ed1e20c6cfe0b8ac4e8370db4ef2735ed8f492ecb06
3
  size 4876059352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d9f0f73d44a40e5d87ca8db4efefeb94d54215c5e098537b2cff4b175b98e76
3
  size 4876059352
model-00003-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8913b0145e91f057b5d9a5590ac42336272ac82a3d03662704bb49db5e056e7c
3
  size 4876059384
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2d339ee1815d9b130f674be93c8e5e1a1e9f3b316cd87176a2474b2baf8ec18
3
  size 4876059384
model-00004-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57d337ac26eb0a1f348b5b7b609c150bcf86ea24e68edcc5cf9cbffe80236bf6
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cdf440c8fdcfc19be30d1512774c9a57208c7b15fe1a39bb2e185a22f9677cfa
3
  size 4876059416
model-00005-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b5e7da71a26e0279b0a8114371a973654ee48b29ef5f75a4f3b908522d991740
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:752f257f75f379c6e9701a6fe26d150be02099356f874dcf509b891c8846ff63
3
  size 4876059416
model-00006-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3caf9484d748eabf2cb6ef903ab2ed3613bfbe5b0f0980cb6a3618b5bca5d590
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e013b32e86bb103c9e995121a2295bd64190b7fdd49e2deda46c66006558049
3
  size 4876059416
model-00007-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0c1caa1accd0b3f811e76c114728d1a9b689c0073654880521902697ae800463
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7d868db5bb5231bafa3eff67d81fc8cec1a0789c4a874f850b2bb02282b185d
3
  size 4876059416
model-00008-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:46eae515c6642f537372a2d531b8a8bec79fc9a97eec73fde60c176bf4bc7eef
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dcac482f774c4ed4528af65f31b2c4a70882c0446b9bcfbbfbd5e38dd213823
3
  size 4876059416
model-00009-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ef6c7494e61f9cb5d591e64303874956a5b5825bc3195fc315d0e6aec4e0ffab
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1459f8167fa1376bc6245f54a8db86f17fcf360cfc533b9890b6da451984a15d
3
  size 4876059416
model-00010-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7a1f5fada1ea8b081533df536b6a7f226105c1df01d8296381671c1dea312218
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bbce57bb600be813a27649838b5f1397e7de9d003328c8a2b6162e75624d8f9
3
  size 4876059416
model-00011-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea50b5b5edce1b55fb34d465e1d701b8d2a3242319b1e275cdd6140c78b1b358
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7c2a839fcd50146834b27e6356b8df535409f0fe5df5fd9761a1d6e813ce325
3
  size 4876059416
model-00012-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ecfcdd48c9481edbd8826c679280745e6b87074bf97e85b1a596e2e548b02477
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:278cbfc415104daa0bdd51d7dca602ce8987200636a19b03f4a191a72f8d87fb
3
  size 4876059416
model-00013-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8cbc5dfea8eb28ba59749282884883bc6f1805b63366cf92bfbe54160bc52309
3
  size 4876059416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f30cb4f146ffc55d56f89c3ee3b8db64c3c42b2b74710d47dbde71a90b89071
3
  size 4876059416
model-00014-of-00014.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f447e36f64421c710826847bff3ae27e7e28cc5768f0db0b58700d3a7fd62ed1
3
  size 2123397800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8db71c7aa728cd9ffff7032ee0b7afa1a17db979e4ceec6be6734fd1f1a4f791
3
  size 2123397800
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "total_flos": 978754951168.0,
3
- "train_loss": 1.4972400459749946,
4
- "train_runtime": 1197.8134,
5
  "train_samples": 114,
6
- "train_samples_per_second": 0.095,
7
- "train_steps_per_second": 0.024
8
  }
 
1
  {
2
+ "total_flos": 955996483584.0,
3
+ "train_loss": 1.542164264054134,
4
+ "train_runtime": 1339.7897,
5
  "train_samples": 114,
6
+ "train_samples_per_second": 0.085,
7
+ "train_steps_per_second": 0.022
8
  }
trainer_state.json CHANGED
@@ -10,244 +10,244 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.034482758620689655,
13
- "grad_norm": 3.9996883869171143,
14
  "learning_rate": 3.333333333333333e-07,
15
- "loss": 1.4892,
16
- "mean_token_accuracy": 0.6603646278381348,
17
  "step": 1
18
  },
19
  {
20
  "epoch": 0.06896551724137931,
21
- "grad_norm": 4.237244606018066,
22
  "learning_rate": 6.666666666666666e-07,
23
- "loss": 1.4666,
24
- "mean_token_accuracy": 0.7045964002609253,
25
  "step": 2
26
  },
27
  {
28
  "epoch": 0.10344827586206896,
29
- "grad_norm": 4.874926567077637,
30
  "learning_rate": 1e-06,
31
- "loss": 1.7228,
32
- "mean_token_accuracy": 0.6302095651626587,
33
  "step": 3
34
  },
35
  {
36
  "epoch": 0.13793103448275862,
37
- "grad_norm": 4.235203266143799,
38
  "learning_rate": 9.963544370490268e-07,
39
- "loss": 1.4493,
40
- "mean_token_accuracy": 0.6814345717430115,
41
  "step": 4
42
  },
43
  {
44
  "epoch": 0.1724137931034483,
45
- "grad_norm": 7.380453109741211,
46
  "learning_rate": 9.85470908713026e-07,
47
- "loss": 2.0416,
48
- "mean_token_accuracy": 0.6577650904655457,
49
  "step": 5
50
  },
51
  {
52
  "epoch": 0.20689655172413793,
53
- "grad_norm": 3.541438341140747,
54
  "learning_rate": 9.675081213427074e-07,
55
- "loss": 1.2546,
56
- "mean_token_accuracy": 0.7483809590339661,
57
  "step": 6
58
  },
59
  {
60
  "epoch": 0.2413793103448276,
61
- "grad_norm": 3.448702812194824,
62
  "learning_rate": 9.427280128266049e-07,
63
- "loss": 1.3896,
64
- "mean_token_accuracy": 0.6593328714370728,
65
  "step": 7
66
  },
67
  {
68
  "epoch": 0.27586206896551724,
69
- "grad_norm": 3.706294298171997,
70
  "learning_rate": 9.114919329468282e-07,
71
- "loss": 1.4015,
72
- "mean_token_accuracy": 0.6922000646591187,
73
  "step": 8
74
  },
75
  {
76
  "epoch": 0.3103448275862069,
77
- "grad_norm": 4.539882659912109,
78
  "learning_rate": 8.742553740855505e-07,
79
- "loss": 1.7425,
80
- "mean_token_accuracy": 0.5878053307533264,
81
  "step": 9
82
  },
83
  {
84
  "epoch": 0.3448275862068966,
85
- "grad_norm": 3.8082709312438965,
86
  "learning_rate": 8.315613291203976e-07,
87
- "loss": 1.4157,
88
- "mean_token_accuracy": 0.6789916157722473,
89
  "step": 10
90
  },
91
  {
92
  "epoch": 0.3793103448275862,
93
- "grad_norm": 5.028677940368652,
94
  "learning_rate": 7.840323733655778e-07,
95
- "loss": 1.755,
96
- "mean_token_accuracy": 0.6366782188415527,
97
  "step": 11
98
  },
99
  {
100
  "epoch": 0.41379310344827586,
101
- "grad_norm": 5.152957439422607,
102
  "learning_rate": 7.323615860218842e-07,
103
- "loss": 1.7695,
104
- "mean_token_accuracy": 0.6314002275466919,
105
  "step": 12
106
  },
107
  {
108
  "epoch": 0.4482758620689655,
109
- "grad_norm": 3.268697738647461,
110
  "learning_rate": 6.773024435212677e-07,
111
- "loss": 1.4895,
112
- "mean_token_accuracy": 0.6149526834487915,
113
  "step": 13
114
  },
115
  {
116
  "epoch": 0.4827586206896552,
117
- "grad_norm": 3.3466408252716064,
118
  "learning_rate": 6.196578321437789e-07,
119
- "loss": 1.1299,
120
- "mean_token_accuracy": 0.7559174299240112,
121
  "step": 14
122
  },
123
  {
124
  "epoch": 0.5172413793103449,
125
- "grad_norm": 4.494246482849121,
126
  "learning_rate": 5.602683401276614e-07,
127
- "loss": 1.5723,
128
- "mean_token_accuracy": 0.6703522801399231,
129
  "step": 15
130
  },
131
  {
132
  "epoch": 0.5517241379310345,
133
- "grad_norm": 2.2075376510620117,
134
  "learning_rate": 5e-07,
135
- "loss": 1.0356,
136
- "mean_token_accuracy": 0.7093867063522339,
137
  "step": 16
138
  },
139
  {
140
  "epoch": 0.5862068965517241,
141
- "grad_norm": 2.7901251316070557,
142
  "learning_rate": 4.397316598723385e-07,
143
- "loss": 1.1132,
144
- "mean_token_accuracy": 0.7466140985488892,
145
  "step": 17
146
  },
147
  {
148
  "epoch": 0.6206896551724138,
149
- "grad_norm": 3.604022979736328,
150
  "learning_rate": 3.8034216785622125e-07,
151
- "loss": 1.3046,
152
- "mean_token_accuracy": 0.7072184681892395,
153
  "step": 18
154
  },
155
  {
156
  "epoch": 0.6551724137931034,
157
- "grad_norm": 3.637402057647705,
158
  "learning_rate": 3.2269755647873214e-07,
159
- "loss": 1.5471,
160
- "mean_token_accuracy": 0.6090646386146545,
161
  "step": 19
162
  },
163
  {
164
  "epoch": 0.6896551724137931,
165
- "grad_norm": 3.484041452407837,
166
  "learning_rate": 2.676384139781157e-07,
167
- "loss": 1.4808,
168
- "mean_token_accuracy": 0.6458947658538818,
169
  "step": 20
170
  },
171
  {
172
  "epoch": 0.7241379310344828,
173
- "grad_norm": 4.93519926071167,
174
  "learning_rate": 2.1596762663442213e-07,
175
- "loss": 1.853,
176
- "mean_token_accuracy": 0.6095913648605347,
177
  "step": 21
178
  },
179
  {
180
  "epoch": 0.7586206896551724,
181
- "grad_norm": 6.128385543823242,
182
  "learning_rate": 1.6843867087960251e-07,
183
- "loss": 1.552,
184
- "mean_token_accuracy": 0.6161980032920837,
185
  "step": 22
186
  },
187
  {
188
  "epoch": 0.7931034482758621,
189
- "grad_norm": 5.635701656341553,
190
  "learning_rate": 1.257446259144494e-07,
191
- "loss": 1.5498,
192
- "mean_token_accuracy": 0.7546396851539612,
193
  "step": 23
194
  },
195
  {
196
  "epoch": 0.8275862068965517,
197
- "grad_norm": 3.2936410903930664,
198
  "learning_rate": 8.850806705317182e-08,
199
- "loss": 1.542,
200
- "mean_token_accuracy": 0.5884397029876709,
201
  "step": 24
202
  },
203
  {
204
  "epoch": 0.8620689655172413,
205
- "grad_norm": 3.3056509494781494,
206
  "learning_rate": 5.72719871733951e-08,
207
- "loss": 1.4571,
208
- "mean_token_accuracy": 0.677709698677063,
209
  "step": 25
210
  },
211
  {
212
  "epoch": 0.896551724137931,
213
- "grad_norm": 3.8348419666290283,
214
  "learning_rate": 3.2491878657292635e-08,
215
- "loss": 1.5325,
216
- "mean_token_accuracy": 0.6370223760604858,
217
  "step": 26
218
  },
219
  {
220
  "epoch": 0.9310344827586207,
221
- "grad_norm": 3.4010775089263916,
222
  "learning_rate": 1.4529091286973993e-08,
223
- "loss": 1.3759,
224
- "mean_token_accuracy": 0.6779050230979919,
225
  "step": 27
226
  },
227
  {
228
  "epoch": 0.9655172413793104,
229
- "grad_norm": 3.990252733230591,
230
  "learning_rate": 3.6455629509730135e-09,
231
- "loss": 1.4335,
232
- "mean_token_accuracy": 0.688022255897522,
233
  "step": 28
234
  },
235
  {
236
  "epoch": 1.0,
237
- "grad_norm": 4.286991596221924,
238
  "learning_rate": 0.0,
239
- "loss": 1.5531,
240
- "mean_token_accuracy": 0.6274685859680176,
241
  "step": 29
242
  },
243
  {
244
  "epoch": 1.0,
245
  "step": 29,
246
- "total_flos": 978754951168.0,
247
- "train_loss": 1.4972400459749946,
248
- "train_runtime": 1197.8134,
249
- "train_samples_per_second": 0.095,
250
- "train_steps_per_second": 0.024
251
  }
252
  ],
253
  "logging_steps": 1,
@@ -267,7 +267,7 @@
267
  "attributes": {}
268
  }
269
  },
270
- "total_flos": 978754951168.0,
271
  "train_batch_size": 1,
272
  "trial_name": null,
273
  "trial_params": null
 
10
  "log_history": [
11
  {
12
  "epoch": 0.034482758620689655,
13
+ "grad_norm": 4.296210765838623,
14
  "learning_rate": 3.333333333333333e-07,
15
+ "loss": 1.5533,
16
+ "mean_token_accuracy": 0.6920064091682434,
17
  "step": 1
18
  },
19
  {
20
  "epoch": 0.06896551724137931,
21
+ "grad_norm": 5.163259029388428,
22
  "learning_rate": 6.666666666666666e-07,
23
+ "loss": 1.7853,
24
+ "mean_token_accuracy": 0.6263327598571777,
25
  "step": 2
26
  },
27
  {
28
  "epoch": 0.10344827586206896,
29
+ "grad_norm": 4.452831745147705,
30
  "learning_rate": 1e-06,
31
+ "loss": 1.5075,
32
+ "mean_token_accuracy": 0.7126692533493042,
33
  "step": 3
34
  },
35
  {
36
  "epoch": 0.13793103448275862,
37
+ "grad_norm": 6.790956497192383,
38
  "learning_rate": 9.963544370490268e-07,
39
+ "loss": 1.9253,
40
+ "mean_token_accuracy": 0.6712393760681152,
41
  "step": 4
42
  },
43
  {
44
  "epoch": 0.1724137931034483,
45
+ "grad_norm": 3.9182674884796143,
46
  "learning_rate": 9.85470908713026e-07,
47
+ "loss": 1.6858,
48
+ "mean_token_accuracy": 0.559170126914978,
49
  "step": 5
50
  },
51
  {
52
  "epoch": 0.20689655172413793,
53
+ "grad_norm": 3.839620351791382,
54
  "learning_rate": 9.675081213427074e-07,
55
+ "loss": 1.5418,
56
+ "mean_token_accuracy": 0.6354226469993591,
57
  "step": 6
58
  },
59
  {
60
  "epoch": 0.2413793103448276,
61
+ "grad_norm": 4.464385986328125,
62
  "learning_rate": 9.427280128266049e-07,
63
+ "loss": 1.5432,
64
+ "mean_token_accuracy": 0.6771755814552307,
65
  "step": 7
66
  },
67
  {
68
  "epoch": 0.27586206896551724,
69
+ "grad_norm": 3.300748348236084,
70
  "learning_rate": 9.114919329468282e-07,
71
+ "loss": 1.3202,
72
+ "mean_token_accuracy": 0.6744688153266907,
73
  "step": 8
74
  },
75
  {
76
  "epoch": 0.3103448275862069,
77
+ "grad_norm": 4.529242515563965,
78
  "learning_rate": 8.742553740855505e-07,
79
+ "loss": 1.5632,
80
+ "mean_token_accuracy": 0.7091418504714966,
81
  "step": 9
82
  },
83
  {
84
  "epoch": 0.3448275862068966,
85
+ "grad_norm": 4.9364800453186035,
86
  "learning_rate": 8.315613291203976e-07,
87
+ "loss": 1.3939,
88
+ "mean_token_accuracy": 0.676058828830719,
89
  "step": 10
90
  },
91
  {
92
  "epoch": 0.3793103448275862,
93
+ "grad_norm": 4.520140171051025,
94
  "learning_rate": 7.840323733655778e-07,
95
+ "loss": 1.6434,
96
+ "mean_token_accuracy": 0.6085848808288574,
97
  "step": 11
98
  },
99
  {
100
  "epoch": 0.41379310344827586,
101
+ "grad_norm": 3.786013126373291,
102
  "learning_rate": 7.323615860218842e-07,
103
+ "loss": 1.4203,
104
+ "mean_token_accuracy": 0.6495858430862427,
105
  "step": 12
106
  },
107
  {
108
  "epoch": 0.4482758620689655,
109
+ "grad_norm": 4.0102105140686035,
110
  "learning_rate": 6.773024435212677e-07,
111
+ "loss": 1.4744,
112
+ "mean_token_accuracy": 0.715476393699646,
113
  "step": 13
114
  },
115
  {
116
  "epoch": 0.4827586206896552,
117
+ "grad_norm": 6.1422576904296875,
118
  "learning_rate": 6.196578321437789e-07,
119
+ "loss": 1.7096,
120
+ "mean_token_accuracy": 0.6734651327133179,
121
  "step": 14
122
  },
123
  {
124
  "epoch": 0.5172413793103449,
125
+ "grad_norm": 4.225828647613525,
126
  "learning_rate": 5.602683401276614e-07,
127
+ "loss": 1.6184,
128
+ "mean_token_accuracy": 0.6401326656341553,
129
  "step": 15
130
  },
131
  {
132
  "epoch": 0.5517241379310345,
133
+ "grad_norm": 2.499864101409912,
134
  "learning_rate": 5e-07,
135
+ "loss": 1.0916,
136
+ "mean_token_accuracy": 0.712737500667572,
137
  "step": 16
138
  },
139
  {
140
  "epoch": 0.5862068965517241,
141
+ "grad_norm": 3.797278881072998,
142
  "learning_rate": 4.397316598723385e-07,
143
+ "loss": 1.5532,
144
+ "mean_token_accuracy": 0.6137330532073975,
145
  "step": 17
146
  },
147
  {
148
  "epoch": 0.6206896551724138,
149
+ "grad_norm": 3.9196736812591553,
150
  "learning_rate": 3.8034216785622125e-07,
151
+ "loss": 1.5172,
152
+ "mean_token_accuracy": 0.6513392925262451,
153
  "step": 18
154
  },
155
  {
156
  "epoch": 0.6551724137931034,
157
+ "grad_norm": 3.312458038330078,
158
  "learning_rate": 3.2269755647873214e-07,
159
+ "loss": 1.4259,
160
+ "mean_token_accuracy": 0.6606206893920898,
161
  "step": 19
162
  },
163
  {
164
  "epoch": 0.6896551724137931,
165
+ "grad_norm": 4.863149642944336,
166
  "learning_rate": 2.676384139781157e-07,
167
+ "loss": 1.801,
168
+ "mean_token_accuracy": 0.6028177738189697,
169
  "step": 20
170
  },
171
  {
172
  "epoch": 0.7241379310344828,
173
+ "grad_norm": 6.340369701385498,
174
  "learning_rate": 2.1596762663442213e-07,
175
+ "loss": 1.8937,
176
+ "mean_token_accuracy": 0.6635770797729492,
177
  "step": 21
178
  },
179
  {
180
  "epoch": 0.7586206896551724,
181
+ "grad_norm": 2.400047540664673,
182
  "learning_rate": 1.6843867087960251e-07,
183
+ "loss": 1.1017,
184
+ "mean_token_accuracy": 0.7170366048812866,
185
  "step": 22
186
  },
187
  {
188
  "epoch": 0.7931034482758621,
189
+ "grad_norm": 6.618093490600586,
190
  "learning_rate": 1.257446259144494e-07,
191
+ "loss": 1.8901,
192
+ "mean_token_accuracy": 0.6319121718406677,
193
  "step": 23
194
  },
195
  {
196
  "epoch": 0.8275862068965517,
197
+ "grad_norm": 3.391517400741577,
198
  "learning_rate": 8.850806705317182e-08,
199
+ "loss": 1.3663,
200
+ "mean_token_accuracy": 0.6898515224456787,
201
  "step": 24
202
  },
203
  {
204
  "epoch": 0.8620689655172413,
205
+ "grad_norm": 4.413179874420166,
206
  "learning_rate": 5.72719871733951e-08,
207
+ "loss": 1.5641,
208
+ "mean_token_accuracy": 0.6674157381057739,
209
  "step": 25
210
  },
211
  {
212
  "epoch": 0.896551724137931,
213
+ "grad_norm": 4.75150728225708,
214
  "learning_rate": 3.2491878657292635e-08,
215
+ "loss": 1.6575,
216
+ "mean_token_accuracy": 0.6414722204208374,
217
  "step": 26
218
  },
219
  {
220
  "epoch": 0.9310344827586207,
221
+ "grad_norm": 3.682435989379883,
222
  "learning_rate": 1.4529091286973993e-08,
223
+ "loss": 1.3724,
224
+ "mean_token_accuracy": 0.674655020236969,
225
  "step": 27
226
  },
227
  {
228
  "epoch": 0.9655172413793104,
229
+ "grad_norm": 3.129405975341797,
230
  "learning_rate": 3.6455629509730135e-09,
231
+ "loss": 1.5536,
232
+ "mean_token_accuracy": 0.5809841752052307,
233
  "step": 28
234
  },
235
  {
236
  "epoch": 1.0,
237
+ "grad_norm": 2.9102606773376465,
238
  "learning_rate": 0.0,
239
+ "loss": 1.2491,
240
+ "mean_token_accuracy": 0.6780526041984558,
241
  "step": 29
242
  },
243
  {
244
  "epoch": 1.0,
245
  "step": 29,
246
+ "total_flos": 955996483584.0,
247
+ "train_loss": 1.542164264054134,
248
+ "train_runtime": 1339.7897,
249
+ "train_samples_per_second": 0.085,
250
+ "train_steps_per_second": 0.022
251
  }
252
  ],
253
  "logging_steps": 1,
 
267
  "attributes": {}
268
  }
269
  },
270
+ "total_flos": 955996483584.0,
271
  "train_batch_size": 1,
272
  "trial_name": null,
273
  "trial_params": null
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d7e8bf88eeb802e31b36a5851e691b0655ff8116d2b7229b7ff93ac1200c5af3
3
  size 7864
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d32fdd6e3debf45b5ea78119281481411bda5d400189c4d03a7bc24c2d0bc3d
3
  size 7864