RikkiXu commited on
Commit
6849bf5
1 Parent(s): 3502f31

Model save

Browse files
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.4350618153878388,
4
- "train_runtime": 5287.1692,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 10.636,
7
- "train_steps_per_second": 0.083
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.5807303016441016,
4
+ "train_runtime": 6860.7908,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.197,
7
+ "train_steps_per_second": 0.064
8
  }
model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1aad48cb8aa7b2903f93b8cfd7cb4737c66ec3d00b09189ab8e7eb0e36e7f63d
3
  size 4943162336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82d958e991ee8e8b81e29d04fcc8041bd5590fa21a2b4b16d78ac83600e78932
3
  size 4943162336
model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:df0b2b647126bc8b703c31af35bd8e2148134aa0e7a73730fa9484c71e1715ba
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ded0292a572c378c561e70cb2c28e9ce4fd32ea16fbfbec8c2b36978e671b88f
3
  size 4999819336
model-00003-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2c4087aac41dc68561e56069eabd2d1e6f707422d96ba377ee692b1d30a1ec69
3
  size 4540516344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23081b1de882d06c5f50e034e626fcca1f28e623615e921b1870ad9b5ff57021
3
  size 4540516344
runs/Jul03_01-53-04_n136-100-194/events.out.tfevents.1719942806.n136-100-194.1930716.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e5f8eb1dd7e34790c767b8227852e30128a8a582ad5c501bc88020d16318e17b
3
- size 32997
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29dfb50f1ac3f975f3605a500c6d2377cafa56d1723892e0483456ad85387b5a
3
+ size 35415
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.4350618153878388,
4
- "train_runtime": 5287.1692,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 10.636,
7
- "train_steps_per_second": 0.083
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.5807303016441016,
4
+ "train_runtime": 6860.7908,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.197,
7
+ "train_steps_per_second": 0.064
8
  }
trainer_state.json CHANGED
@@ -10,13 +10,13 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
- "grad_norm": 22.786040256127208,
14
  "learning_rate": 2.2727272727272727e-09,
15
  "logits/chosen": -1.6768856048583984,
16
  "logits/rejected": -1.7259055376052856,
17
  "logps/chosen": -1.2793102264404297,
18
  "logps/rejected": -1.2162058353424072,
19
- "loss": 0.6931,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
@@ -25,657 +25,657 @@
25
  },
26
  {
27
  "epoch": 0.02,
28
- "grad_norm": 25.6836964844408,
29
  "learning_rate": 2.2727272727272725e-08,
30
- "logits/chosen": -1.7025811672210693,
31
- "logits/rejected": -1.6681201457977295,
32
- "logps/chosen": -1.2131770849227905,
33
- "logps/rejected": -1.2202790975570679,
34
- "loss": 0.6931,
35
- "rewards/accuracies": 0.4166666567325592,
36
- "rewards/chosen": 0.00025717043899931014,
37
- "rewards/margins": -0.0006778165698051453,
38
- "rewards/rejected": 0.0009349870379082859,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
- "grad_norm": 26.405806920127038,
44
  "learning_rate": 4.545454545454545e-08,
45
- "logits/chosen": -1.7794504165649414,
46
- "logits/rejected": -1.734635591506958,
47
- "logps/chosen": -1.144683599472046,
48
- "logps/rejected": -1.1852140426635742,
49
- "loss": 0.6925,
50
- "rewards/accuracies": 0.5375000238418579,
51
- "rewards/chosen": -0.0029223989695310593,
52
- "rewards/margins": 0.0005865652346983552,
53
- "rewards/rejected": -0.0035089640878140926,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
- "grad_norm": 28.068883928310647,
59
  "learning_rate": 6.818181818181817e-08,
60
- "logits/chosen": -1.7446922063827515,
61
- "logits/rejected": -1.676011085510254,
62
- "logps/chosen": -1.1949479579925537,
63
- "logps/rejected": -1.2484194040298462,
64
- "loss": 0.6882,
65
- "rewards/accuracies": 0.6625000238418579,
66
- "rewards/chosen": -0.024093259125947952,
67
- "rewards/margins": 0.012483139522373676,
68
- "rewards/rejected": -0.036576397716999054,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
- "grad_norm": 19.593533440172358,
74
  "learning_rate": 9.09090909090909e-08,
75
- "logits/chosen": -1.7312889099121094,
76
- "logits/rejected": -1.6649024486541748,
77
- "logps/chosen": -1.2515684366226196,
78
- "logps/rejected": -1.3352997303009033,
79
- "loss": 0.6797,
80
- "rewards/accuracies": 0.6625000238418579,
81
- "rewards/chosen": -0.08923971652984619,
82
- "rewards/margins": 0.06281541287899017,
83
- "rewards/rejected": -0.15205512940883636,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
- "grad_norm": 24.263100694250465,
89
  "learning_rate": 9.994307990108962e-08,
90
- "logits/chosen": -1.6921930313110352,
91
- "logits/rejected": -1.6271032094955444,
92
- "logps/chosen": -1.298782467842102,
93
- "logps/rejected": -1.3565475940704346,
94
- "loss": 0.6637,
95
- "rewards/accuracies": 0.7250000238418579,
96
- "rewards/chosen": -0.1813907027244568,
97
- "rewards/margins": 0.07534531503915787,
98
- "rewards/rejected": -0.25673604011535645,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
- "grad_norm": 28.402662229181594,
104
  "learning_rate": 9.959570405988094e-08,
105
- "logits/chosen": -1.7207958698272705,
106
- "logits/rejected": -1.6402089595794678,
107
- "logps/chosen": -1.2883288860321045,
108
- "logps/rejected": -1.378244161605835,
109
- "loss": 0.6599,
110
- "rewards/accuracies": 0.6499999761581421,
111
- "rewards/chosen": -0.35846152901649475,
112
- "rewards/margins": 0.07597381621599197,
113
- "rewards/rejected": -0.4344353675842285,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
- "grad_norm": 21.201003636828116,
119
  "learning_rate": 9.893476820924666e-08,
120
- "logits/chosen": -1.801526665687561,
121
- "logits/rejected": -1.7130231857299805,
122
- "logps/chosen": -1.4759318828582764,
123
- "logps/rejected": -1.5954515933990479,
124
- "loss": 0.6498,
125
- "rewards/accuracies": 0.699999988079071,
126
- "rewards/chosen": -0.5261799097061157,
127
- "rewards/margins": 0.14570936560630798,
128
- "rewards/rejected": -0.6718892455101013,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
- "grad_norm": 26.86998789465821,
134
  "learning_rate": 9.796445099843647e-08,
135
- "logits/chosen": -1.7849477529525757,
136
- "logits/rejected": -1.6990400552749634,
137
- "logps/chosen": -1.5336757898330688,
138
- "logps/rejected": -1.6886417865753174,
139
- "loss": 0.654,
140
- "rewards/accuracies": 0.668749988079071,
141
- "rewards/chosen": -0.6534146070480347,
142
- "rewards/margins": 0.18887227773666382,
143
- "rewards/rejected": -0.8422868847846985,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
- "grad_norm": 25.145260129960253,
149
  "learning_rate": 9.669088708527066e-08,
150
- "logits/chosen": -1.7331278324127197,
151
- "logits/rejected": -1.6643707752227783,
152
- "logps/chosen": -1.604683518409729,
153
- "logps/rejected": -1.6745525598526,
154
- "loss": 0.6397,
155
- "rewards/accuracies": 0.643750011920929,
156
- "rewards/chosen": -0.7379721999168396,
157
- "rewards/margins": 0.12564878165721893,
158
- "rewards/rejected": -0.8636210560798645,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
- "grad_norm": 28.208412906348087,
164
  "learning_rate": 9.512212835085849e-08,
165
- "logits/chosen": -1.770500898361206,
166
- "logits/rejected": -1.6813795566558838,
167
- "logps/chosen": -1.557369351387024,
168
- "logps/rejected": -1.7003123760223389,
169
- "loss": 0.623,
170
  "rewards/accuracies": 0.6937500238418579,
171
- "rewards/chosen": -0.7778279781341553,
172
- "rewards/margins": 0.200466588139534,
173
- "rewards/rejected": -0.9782945513725281,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
- "grad_norm": 25.857658698377822,
179
  "learning_rate": 9.326809299301306e-08,
180
- "logits/chosen": -1.7706959247589111,
181
- "logits/rejected": -1.666680932044983,
182
- "logps/chosen": -1.6019655466079712,
183
- "logps/rejected": -1.7736393213272095,
184
- "loss": 0.6212,
185
- "rewards/accuracies": 0.7124999761581421,
186
- "rewards/chosen": -0.8026469945907593,
187
- "rewards/margins": 0.240141361951828,
188
- "rewards/rejected": -1.0427882671356201,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
- "grad_norm": 27.113925141570313,
194
  "learning_rate": 9.114050282021158e-08,
195
- "logits/chosen": -1.7661209106445312,
196
- "logits/rejected": -1.7044270038604736,
197
- "logps/chosen": -1.5668163299560547,
198
- "logps/rejected": -1.7313486337661743,
199
- "loss": 0.6144,
200
- "rewards/accuracies": 0.731249988079071,
201
- "rewards/chosen": -0.8385075330734253,
202
- "rewards/margins": 0.2342797964811325,
203
- "rewards/rejected": -1.0727872848510742,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
- "grad_norm": 25.192318558632277,
209
  "learning_rate": 8.875280914254802e-08,
210
- "logits/chosen": -1.7651408910751343,
211
- "logits/rejected": -1.6746151447296143,
212
- "logps/chosen": -1.7618343830108643,
213
- "logps/rejected": -1.9648230075836182,
214
- "loss": 0.5995,
215
- "rewards/accuracies": 0.668749988079071,
216
- "rewards/chosen": -1.1380090713500977,
217
- "rewards/margins": 0.34912705421447754,
218
- "rewards/rejected": -1.4871362447738647,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
- "grad_norm": 24.555074276571457,
224
  "learning_rate": 8.612010772821971e-08,
225
- "logits/chosen": -1.7977707386016846,
226
- "logits/rejected": -1.7540991306304932,
227
- "logps/chosen": -1.8272440433502197,
228
- "logps/rejected": -1.9743133783340454,
229
- "loss": 0.5993,
230
  "rewards/accuracies": 0.737500011920929,
231
- "rewards/chosen": -1.2267296314239502,
232
- "rewards/margins": 0.34395888447761536,
233
- "rewards/rejected": -1.5706886053085327,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
- "grad_norm": 29.798327176387467,
239
  "learning_rate": 8.325904336322055e-08,
240
- "logits/chosen": -1.7762537002563477,
241
- "logits/rejected": -1.722198247909546,
242
- "logps/chosen": -1.9330543279647827,
243
- "logps/rejected": -2.1491074562072754,
244
- "loss": 0.6088,
245
- "rewards/accuracies": 0.6812499761581421,
246
- "rewards/chosen": -1.5313791036605835,
247
- "rewards/margins": 0.3377588093280792,
248
- "rewards/rejected": -1.8691380023956299,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
- "grad_norm": 22.379243172505653,
254
  "learning_rate": 8.01877046176447e-08,
255
- "logits/chosen": -1.7237749099731445,
256
- "logits/rejected": -1.6540390253067017,
257
- "logps/chosen": -1.9900233745574951,
258
- "logps/rejected": -2.1875462532043457,
259
- "loss": 0.5903,
260
- "rewards/accuracies": 0.675000011920929,
261
- "rewards/chosen": -1.6234468221664429,
262
- "rewards/margins": 0.29227370023727417,
263
- "rewards/rejected": -1.9157205820083618,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
- "grad_norm": 19.935363442551864,
269
  "learning_rate": 7.692550948392249e-08,
270
- "logits/chosen": -1.7721624374389648,
271
- "logits/rejected": -1.7196706533432007,
272
- "logps/chosen": -1.960934042930603,
273
- "logps/rejected": -2.164457321166992,
274
- "loss": 0.5921,
275
- "rewards/accuracies": 0.706250011920929,
276
- "rewards/chosen": -1.501357078552246,
277
- "rewards/margins": 0.38615694642066956,
278
- "rewards/rejected": -1.887514352798462,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
- "grad_norm": 27.99675949527233,
284
  "learning_rate": 7.349308261002021e-08,
285
- "logits/chosen": -1.7341270446777344,
286
- "logits/rejected": -1.6832844018936157,
287
- "logps/chosen": -1.9921613931655884,
288
- "logps/rejected": -2.2205045223236084,
289
- "loss": 0.5854,
290
- "rewards/accuracies": 0.699999988079071,
291
- "rewards/chosen": -1.5901505947113037,
292
- "rewards/margins": 0.35738545656204224,
293
- "rewards/rejected": -1.9475358724594116,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
- "grad_norm": 24.529175323127856,
299
  "learning_rate": 6.991212490377531e-08,
300
- "logits/chosen": -1.7890892028808594,
301
- "logits/rejected": -1.746469259262085,
302
- "logps/chosen": -2.087700843811035,
303
- "logps/rejected": -2.3541736602783203,
304
- "loss": 0.5608,
305
- "rewards/accuracies": 0.75,
306
- "rewards/chosen": -1.7151954174041748,
307
- "rewards/margins": 0.4898623526096344,
308
- "rewards/rejected": -2.2050578594207764,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
- "grad_norm": 30.651962115004057,
314
  "learning_rate": 6.620527633276978e-08,
315
- "logits/chosen": -1.7287298440933228,
316
- "logits/rejected": -1.6635551452636719,
317
- "logps/chosen": -2.149373769760132,
318
- "logps/rejected": -2.5112643241882324,
319
- "loss": 0.5803,
320
- "rewards/accuracies": 0.71875,
321
- "rewards/chosen": -1.909841537475586,
322
- "rewards/margins": 0.5502029657363892,
323
- "rewards/rejected": -2.4600441455841064,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
- "grad_norm": 27.408811645376396,
329
  "learning_rate": 6.239597278716581e-08,
330
- "logits/chosen": -1.7830358743667603,
331
- "logits/rejected": -1.7272449731826782,
332
- "logps/chosen": -2.303990125656128,
333
- "logps/rejected": -2.549621105194092,
334
- "loss": 0.5627,
335
- "rewards/accuracies": 0.7250000238418579,
336
- "rewards/chosen": -2.1173198223114014,
337
- "rewards/margins": 0.5325894355773926,
338
- "rewards/rejected": -2.649909496307373,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
- "grad_norm": 25.141892376540234,
344
  "learning_rate": 5.8508297910462456e-08,
345
- "logits/chosen": -1.729365587234497,
346
- "logits/rejected": -1.6552003622055054,
347
- "logps/chosen": -2.227386713027954,
348
- "logps/rejected": -2.558889627456665,
349
- "loss": 0.5666,
350
- "rewards/accuracies": 0.6937500238418579,
351
- "rewards/chosen": -2.1277427673339844,
352
- "rewards/margins": 0.5474593043327332,
353
- "rewards/rejected": -2.675201892852783,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
- "grad_norm": 25.203206141016988,
359
  "learning_rate": 5.456683083494731e-08,
360
- "logits/chosen": -1.7191989421844482,
361
- "logits/rejected": -1.6832917928695679,
362
- "logps/chosen": -2.193159580230713,
363
- "logps/rejected": -2.4023752212524414,
364
- "loss": 0.5847,
365
- "rewards/accuracies": 0.65625,
366
- "rewards/chosen": -1.9987010955810547,
367
- "rewards/margins": 0.3709481358528137,
368
- "rewards/rejected": -2.3696494102478027,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
- "grad_norm": 31.30257665339357,
374
  "learning_rate": 5.059649078450834e-08,
375
- "logits/chosen": -1.7015504837036133,
376
- "logits/rejected": -1.662376046180725,
377
- "logps/chosen": -2.2371976375579834,
378
- "logps/rejected": -2.5139238834381104,
379
- "loss": 0.5533,
380
- "rewards/accuracies": 0.6625000238418579,
381
- "rewards/chosen": -2.137688398361206,
382
- "rewards/margins": 0.4293746054172516,
383
- "rewards/rejected": -2.567063570022583,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
- "grad_norm": 24.832978144124045,
389
  "learning_rate": 4.6622379527277186e-08,
390
- "logits/chosen": -1.7127622365951538,
391
- "logits/rejected": -1.6652097702026367,
392
- "logps/chosen": -2.328359603881836,
393
- "logps/rejected": -2.558023452758789,
394
- "loss": 0.5645,
395
- "rewards/accuracies": 0.6187499761581421,
396
- "rewards/chosen": -2.378941535949707,
397
- "rewards/margins": 0.3931688666343689,
398
- "rewards/rejected": -2.7721104621887207,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
- "grad_norm": 28.702758237373775,
404
  "learning_rate": 4.26696226741691e-08,
405
- "logits/chosen": -1.725927710533142,
406
- "logits/rejected": -1.6632360219955444,
407
- "logps/chosen": -2.4254117012023926,
408
- "logps/rejected": -2.7066590785980225,
409
- "loss": 0.5648,
410
- "rewards/accuracies": 0.65625,
411
- "rewards/chosen": -2.385176181793213,
412
- "rewards/margins": 0.522560715675354,
413
- "rewards/rejected": -2.9077370166778564,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
- "grad_norm": 24.57643832571065,
419
  "learning_rate": 3.876321082668098e-08,
420
- "logits/chosen": -1.7822297811508179,
421
- "logits/rejected": -1.7207348346710205,
422
- "logps/chosen": -2.3284859657287598,
423
- "logps/rejected": -2.6171371936798096,
424
- "loss": 0.5581,
425
- "rewards/accuracies": 0.7250000238418579,
426
- "rewards/chosen": -2.179990530014038,
427
- "rewards/margins": 0.5586847066879272,
428
- "rewards/rejected": -2.7386748790740967,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
- "grad_norm": 26.182300162855046,
434
  "learning_rate": 3.492784157826244e-08,
435
- "logits/chosen": -1.721687912940979,
436
- "logits/rejected": -1.6329882144927979,
437
- "logps/chosen": -2.33394193649292,
438
- "logps/rejected": -2.6748275756835938,
439
- "loss": 0.5564,
440
- "rewards/accuracies": 0.7124999761581421,
441
- "rewards/chosen": -2.2221310138702393,
442
- "rewards/margins": 0.610991895198822,
443
- "rewards/rejected": -2.833122730255127,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
- "grad_norm": 22.815010055443274,
449
  "learning_rate": 3.118776336817812e-08,
450
- "logits/chosen": -1.755319595336914,
451
- "logits/rejected": -1.7038185596466064,
452
- "logps/chosen": -2.277468204498291,
453
- "logps/rejected": -2.616783618927002,
454
- "loss": 0.5473,
455
- "rewards/accuracies": 0.731249988079071,
456
- "rewards/chosen": -2.1484389305114746,
457
- "rewards/margins": 0.6641989946365356,
458
- "rewards/rejected": -2.8126378059387207,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
- "grad_norm": 23.65537068910306,
464
  "learning_rate": 2.7566622175067443e-08,
465
- "logits/chosen": -1.7356256246566772,
466
- "logits/rejected": -1.6781995296478271,
467
- "logps/chosen": -2.3750412464141846,
468
- "logps/rejected": -2.7629873752593994,
469
- "loss": 0.5497,
470
- "rewards/accuracies": 0.7562500238418579,
471
- "rewards/chosen": -2.3329153060913086,
472
- "rewards/margins": 0.6199190616607666,
473
- "rewards/rejected": -2.9528346061706543,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
- "grad_norm": 27.623729083981935,
479
  "learning_rate": 2.408731201945432e-08,
480
- "logits/chosen": -1.7345046997070312,
481
- "logits/rejected": -1.6898084878921509,
482
- "logps/chosen": -2.304429531097412,
483
- "logps/rejected": -2.555560827255249,
484
- "loss": 0.5555,
485
- "rewards/accuracies": 0.675000011920929,
486
- "rewards/chosen": -2.2210681438446045,
487
- "rewards/margins": 0.46195346117019653,
488
- "rewards/rejected": -2.6830217838287354,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
- "grad_norm": 26.649151417142534,
494
  "learning_rate": 2.0771830220378112e-08,
495
- "logits/chosen": -1.68793523311615,
496
- "logits/rejected": -1.6301701068878174,
497
- "logps/chosen": -2.3206005096435547,
498
- "logps/rejected": -2.559023380279541,
499
- "loss": 0.5553,
500
- "rewards/accuracies": 0.625,
501
- "rewards/chosen": -2.195432424545288,
502
- "rewards/margins": 0.4925463795661926,
503
- "rewards/rejected": -2.687978744506836,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
- "grad_norm": 24.875498436093228,
509
  "learning_rate": 1.7641138321260257e-08,
510
- "logits/chosen": -1.722652792930603,
511
- "logits/rejected": -1.6583467721939087,
512
- "logps/chosen": -2.2422232627868652,
513
- "logps/rejected": -2.6411871910095215,
514
- "loss": 0.5479,
515
- "rewards/accuracies": 0.831250011920929,
516
- "rewards/chosen": -2.116737127304077,
517
- "rewards/margins": 0.7428576946258545,
518
- "rewards/rejected": -2.8595950603485107,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
- "grad_norm": 26.307918213457313,
524
  "learning_rate": 1.4715029564277793e-08,
525
- "logits/chosen": -1.7855126857757568,
526
- "logits/rejected": -1.7428696155548096,
527
- "logps/chosen": -2.2222747802734375,
528
- "logps/rejected": -2.578981399536133,
529
- "loss": 0.5581,
530
- "rewards/accuracies": 0.75,
531
- "rewards/chosen": -2.0352282524108887,
532
- "rewards/margins": 0.6392861604690552,
533
- "rewards/rejected": -2.6745142936706543,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
- "grad_norm": 25.341211676404754,
539
  "learning_rate": 1.2012003751113343e-08,
540
- "logits/chosen": -1.7763115167617798,
541
- "logits/rejected": -1.7206395864486694,
542
- "logps/chosen": -2.328850269317627,
543
- "logps/rejected": -2.70664119720459,
544
- "loss": 0.5366,
545
- "rewards/accuracies": 0.71875,
546
- "rewards/chosen": -2.3333606719970703,
547
- "rewards/margins": 0.6552005410194397,
548
- "rewards/rejected": -2.9885611534118652,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
- "grad_norm": 31.147010925293188,
554
  "learning_rate": 9.549150281252633e-09,
555
- "logits/chosen": -1.725211501121521,
556
- "logits/rejected": -1.6820595264434814,
557
- "logps/chosen": -2.2657017707824707,
558
- "logps/rejected": -2.6082816123962402,
559
- "loss": 0.5395,
560
- "rewards/accuracies": 0.6937500238418579,
561
- "rewards/chosen": -2.1871337890625,
562
- "rewards/margins": 0.5816982388496399,
563
- "rewards/rejected": -2.768831729888916,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
- "grad_norm": 24.868596217446985,
569
  "learning_rate": 7.3420401072985306e-09,
570
- "logits/chosen": -1.7670882940292358,
571
- "logits/rejected": -1.7162415981292725,
572
- "logps/chosen": -2.3288838863372803,
573
- "logps/rejected": -2.694032669067383,
574
- "loss": 0.5377,
575
- "rewards/accuracies": 0.6812499761581421,
576
- "rewards/chosen": -2.2285408973693848,
577
- "rewards/margins": 0.6153696179389954,
578
- "rewards/rejected": -2.8439104557037354,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
- "grad_norm": 28.264277931849893,
584
  "learning_rate": 5.404627290395369e-09,
585
- "logits/chosen": -1.7286021709442139,
586
- "logits/rejected": -1.6685943603515625,
587
- "logps/chosen": -2.2551302909851074,
588
- "logps/rejected": -2.6050727367401123,
589
- "loss": 0.537,
590
- "rewards/accuracies": 0.737500011920929,
591
- "rewards/chosen": -2.1310482025146484,
592
- "rewards/margins": 0.619892954826355,
593
- "rewards/rejected": -2.750941514968872,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
- "grad_norm": 32.571402526944446,
599
  "learning_rate": 3.74916077816162e-09,
600
- "logits/chosen": -1.734569787979126,
601
- "logits/rejected": -1.678733468055725,
602
- "logps/chosen": -2.2776572704315186,
603
- "logps/rejected": -2.573931932449341,
604
- "loss": 0.5505,
605
- "rewards/accuracies": 0.6812499761581421,
606
- "rewards/chosen": -2.255725145339966,
607
- "rewards/margins": 0.5602914690971375,
608
- "rewards/rejected": -2.816016674041748,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
- "grad_norm": 30.76786815731744,
614
  "learning_rate": 2.386106962899165e-09,
615
- "logits/chosen": -1.6618547439575195,
616
- "logits/rejected": -1.599841594696045,
617
- "logps/chosen": -2.417301654815674,
618
- "logps/rejected": -2.7463181018829346,
619
- "loss": 0.5418,
620
- "rewards/accuracies": 0.6937500238418579,
621
- "rewards/chosen": -2.381761312484741,
622
- "rewards/margins": 0.5453529357910156,
623
- "rewards/rejected": -2.927114725112915,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
- "grad_norm": 31.25431184934112,
629
  "learning_rate": 1.3240835096913706e-09,
630
- "logits/chosen": -1.6889197826385498,
631
- "logits/rejected": -1.5975219011306763,
632
- "logps/chosen": -2.266620397567749,
633
- "logps/rejected": -2.7067856788635254,
634
- "loss": 0.5607,
635
- "rewards/accuracies": 0.7124999761581421,
636
- "rewards/chosen": -2.190139055252075,
637
- "rewards/margins": 0.7415019273757935,
638
- "rewards/rejected": -2.931641101837158,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
- "grad_norm": 23.39234165392022,
644
  "learning_rate": 5.698048727497462e-10,
645
- "logits/chosen": -1.7234703302383423,
646
- "logits/rejected": -1.65776789188385,
647
- "logps/chosen": -2.363593578338623,
648
- "logps/rejected": -2.7682502269744873,
649
- "loss": 0.5431,
650
- "rewards/accuracies": 0.7250000238418579,
651
- "rewards/chosen": -2.3445918560028076,
652
- "rewards/margins": 0.6753993034362793,
653
- "rewards/rejected": -3.019990921020508,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
- "grad_norm": 24.387678586587597,
659
  "learning_rate": 1.2803984447259387e-10,
660
- "logits/chosen": -1.7296276092529297,
661
- "logits/rejected": -1.68634033203125,
662
- "logps/chosen": -2.347764730453491,
663
- "logps/rejected": -2.731468677520752,
664
- "loss": 0.5333,
665
- "rewards/accuracies": 0.7250000238418579,
666
- "rewards/chosen": -2.330850601196289,
667
- "rewards/margins": 0.6721019744873047,
668
- "rewards/rejected": -3.002953052520752,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
- "train_loss": 0.4350618153878388,
676
- "train_runtime": 5287.1692,
677
- "train_samples_per_second": 10.636,
678
- "train_steps_per_second": 0.083
679
  }
680
  ],
681
  "logging_steps": 10,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
+ "grad_norm": 34.16262092771155,
14
  "learning_rate": 2.2727272727272727e-09,
15
  "logits/chosen": -1.6768856048583984,
16
  "logits/rejected": -1.7259055376052856,
17
  "logps/chosen": -1.2793102264404297,
18
  "logps/rejected": -1.2162058353424072,
19
+ "loss": 0.6932,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
 
25
  },
26
  {
27
  "epoch": 0.02,
28
+ "grad_norm": 38.39373462126025,
29
  "learning_rate": 2.2727272727272725e-08,
30
+ "logits/chosen": -1.703444004058838,
31
+ "logits/rejected": -1.6687698364257812,
32
+ "logps/chosen": -1.2133899927139282,
33
+ "logps/rejected": -1.2203023433685303,
34
+ "loss": 0.6932,
35
+ "rewards/accuracies": 0.4236111044883728,
36
+ "rewards/chosen": -0.00025349482893943787,
37
+ "rewards/margins": -0.0015861267456784844,
38
+ "rewards/rejected": 0.0013326319167390466,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
+ "grad_norm": 39.25857800494746,
44
  "learning_rate": 4.545454545454545e-08,
45
+ "logits/chosen": -1.7795432806015015,
46
+ "logits/rejected": -1.734358549118042,
47
+ "logps/chosen": -1.1447466611862183,
48
+ "logps/rejected": -1.185644507408142,
49
+ "loss": 0.692,
50
+ "rewards/accuracies": 0.53125,
51
+ "rewards/chosen": -0.004572628065943718,
52
+ "rewards/margins": 0.001982099376618862,
53
+ "rewards/rejected": -0.00655472744256258,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
+ "grad_norm": 42.49760342477406,
59
  "learning_rate": 6.818181818181817e-08,
60
+ "logits/chosen": -1.7439743280410767,
61
+ "logits/rejected": -1.6747764348983765,
62
+ "logps/chosen": -1.195011019706726,
63
+ "logps/rejected": -1.247870922088623,
64
+ "loss": 0.6863,
65
+ "rewards/accuracies": 0.612500011920929,
66
+ "rewards/chosen": -0.03632856160402298,
67
+ "rewards/margins": 0.01689038798213005,
68
+ "rewards/rejected": -0.05321894958615303,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
+ "grad_norm": 28.94059188031956,
74
  "learning_rate": 9.09090909090909e-08,
75
+ "logits/chosen": -1.7316467761993408,
76
+ "logits/rejected": -1.665168046951294,
77
+ "logps/chosen": -1.2489526271820068,
78
+ "logps/rejected": -1.330916166305542,
79
+ "loss": 0.6749,
80
+ "rewards/accuracies": 0.637499988079071,
81
+ "rewards/chosen": -0.12601229548454285,
82
+ "rewards/margins": 0.08891994506120682,
83
+ "rewards/rejected": -0.21493223309516907,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
+ "grad_norm": 34.449211690195526,
89
  "learning_rate": 9.994307990108962e-08,
90
+ "logits/chosen": -1.693706750869751,
91
+ "logits/rejected": -1.628753662109375,
92
+ "logps/chosen": -1.2911241054534912,
93
+ "logps/rejected": -1.3476231098175049,
94
+ "loss": 0.6547,
95
+ "rewards/accuracies": 0.71875,
96
+ "rewards/chosen": -0.2491108626127243,
97
+ "rewards/margins": 0.10921992361545563,
98
+ "rewards/rejected": -0.35833072662353516,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
+ "grad_norm": 45.18662841636004,
104
  "learning_rate": 9.959570405988094e-08,
105
+ "logits/chosen": -1.7251135110855103,
106
+ "logits/rejected": -1.644749402999878,
107
+ "logps/chosen": -1.2596534490585327,
108
+ "logps/rejected": -1.3467203378677368,
109
+ "loss": 0.6529,
110
+ "rewards/accuracies": 0.643750011920929,
111
+ "rewards/chosen": -0.4516659379005432,
112
+ "rewards/margins": 0.10541460663080215,
113
+ "rewards/rejected": -0.5570805668830872,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
+ "grad_norm": 31.773982345964285,
119
  "learning_rate": 9.893476820924666e-08,
120
+ "logits/chosen": -1.8094285726547241,
121
+ "logits/rejected": -1.723013162612915,
122
+ "logps/chosen": -1.4289662837982178,
123
+ "logps/rejected": -1.5425760746002197,
124
+ "loss": 0.6391,
125
+ "rewards/accuracies": 0.675000011920929,
126
+ "rewards/chosen": -0.648373007774353,
127
+ "rewards/margins": 0.20083490014076233,
128
+ "rewards/rejected": -0.849207878112793,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
+ "grad_norm": 34.373704859871594,
134
  "learning_rate": 9.796445099843647e-08,
135
+ "logits/chosen": -1.8001670837402344,
136
+ "logits/rejected": -1.7168426513671875,
137
+ "logps/chosen": -1.458465337753296,
138
+ "logps/rejected": -1.5951015949249268,
139
+ "loss": 0.6452,
140
+ "rewards/accuracies": 0.637499988079071,
141
+ "rewards/chosen": -0.7544907331466675,
142
+ "rewards/margins": 0.22831876575946808,
143
+ "rewards/rejected": -0.9828095436096191,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
+ "grad_norm": 33.91159834305824,
149
  "learning_rate": 9.669088708527066e-08,
150
+ "logits/chosen": -1.7510133981704712,
151
+ "logits/rejected": -1.6854064464569092,
152
+ "logps/chosen": -1.5039077997207642,
153
+ "logps/rejected": -1.5803452730178833,
154
+ "loss": 0.6218,
155
+ "rewards/accuracies": 0.65625,
156
+ "rewards/chosen": -0.8046310544013977,
157
+ "rewards/margins": 0.20817911624908447,
158
+ "rewards/rejected": -1.0128101110458374,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
+ "grad_norm": 37.79881627068083,
164
  "learning_rate": 9.512212835085849e-08,
165
+ "logits/chosen": -1.7936757802963257,
166
+ "logits/rejected": -1.7088886499404907,
167
+ "logps/chosen": -1.4624369144439697,
168
+ "logps/rejected": -1.5909606218338013,
169
+ "loss": 0.6102,
170
  "rewards/accuracies": 0.6937500238418579,
171
+ "rewards/chosen": -0.881945013999939,
172
+ "rewards/margins": 0.25744158029556274,
173
+ "rewards/rejected": -1.139386534690857,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
+ "grad_norm": 38.73339240736025,
179
  "learning_rate": 9.326809299301306e-08,
180
+ "logits/chosen": -1.7911834716796875,
181
+ "logits/rejected": -1.6914081573486328,
182
+ "logps/chosen": -1.4954617023468018,
183
+ "logps/rejected": -1.65456223487854,
184
+ "loss": 0.6056,
185
+ "rewards/accuracies": 0.706250011920929,
186
+ "rewards/chosen": -0.8844586610794067,
187
+ "rewards/margins": 0.3224923014640808,
188
+ "rewards/rejected": -1.2069510221481323,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
+ "grad_norm": 35.446910248333666,
194
  "learning_rate": 9.114050282021158e-08,
195
+ "logits/chosen": -1.786220908164978,
196
+ "logits/rejected": -1.7271583080291748,
197
+ "logps/chosen": -1.4743096828460693,
198
+ "logps/rejected": -1.6279948949813843,
199
+ "loss": 0.6015,
200
+ "rewards/accuracies": 0.7562500238418579,
201
+ "rewards/chosen": -0.9802410006523132,
202
+ "rewards/margins": 0.3188786804676056,
203
+ "rewards/rejected": -1.2991197109222412,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
+ "grad_norm": 34.33852298554088,
209
  "learning_rate": 8.875280914254802e-08,
210
+ "logits/chosen": -1.7842994928359985,
211
+ "logits/rejected": -1.6957006454467773,
212
+ "logps/chosen": -1.6290218830108643,
213
+ "logps/rejected": -1.7964956760406494,
214
+ "loss": 0.5858,
215
+ "rewards/accuracies": 0.699999988079071,
216
+ "rewards/chosen": -1.308574914932251,
217
+ "rewards/margins": 0.41714730858802795,
218
+ "rewards/rejected": -1.725722312927246,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
+ "grad_norm": 31.7434623187325,
224
  "learning_rate": 8.612010772821971e-08,
225
+ "logits/chosen": -1.8148624897003174,
226
+ "logits/rejected": -1.7723850011825562,
227
+ "logps/chosen": -1.629787802696228,
228
+ "logps/rejected": -1.7449651956558228,
229
+ "loss": 0.5865,
230
  "rewards/accuracies": 0.737500011920929,
231
+ "rewards/chosen": -1.2477266788482666,
232
+ "rewards/margins": 0.4202614724636078,
233
+ "rewards/rejected": -1.6679881811141968,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
+ "grad_norm": 30.04410305195209,
239
  "learning_rate": 8.325904336322055e-08,
240
+ "logits/chosen": -1.7936302423477173,
241
+ "logits/rejected": -1.740534782409668,
242
+ "logps/chosen": -1.6605161428451538,
243
+ "logps/rejected": -1.848489761352539,
244
+ "loss": 0.5948,
245
+ "rewards/accuracies": 0.699999988079071,
246
+ "rewards/chosen": -1.4794538021087646,
247
+ "rewards/margins": 0.4223996102809906,
248
+ "rewards/rejected": -1.901853322982788,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
+ "grad_norm": 31.6961444390199,
254
  "learning_rate": 8.01877046176447e-08,
255
+ "logits/chosen": -1.7392256259918213,
256
+ "logits/rejected": -1.6678926944732666,
257
+ "logps/chosen": -1.707990288734436,
258
+ "logps/rejected": -1.8706386089324951,
259
+ "loss": 0.5828,
260
+ "rewards/accuracies": 0.668749988079071,
261
+ "rewards/chosen": -1.5890707969665527,
262
+ "rewards/margins": 0.33378690481185913,
263
+ "rewards/rejected": -1.922857642173767,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
+ "grad_norm": 29.03821659608711,
269
  "learning_rate": 7.692550948392249e-08,
270
+ "logits/chosen": -1.7871320247650146,
271
+ "logits/rejected": -1.7301868200302124,
272
+ "logps/chosen": -1.6618130207061768,
273
+ "logps/rejected": -1.81256902217865,
274
+ "loss": 0.584,
275
+ "rewards/accuracies": 0.6937500238418579,
276
+ "rewards/chosen": -1.3546721935272217,
277
+ "rewards/margins": 0.42093437910079956,
278
+ "rewards/rejected": -1.7756065130233765,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
+ "grad_norm": 36.01706358849117,
284
  "learning_rate": 7.349308261002021e-08,
285
+ "logits/chosen": -1.743697166442871,
286
+ "logits/rejected": -1.6886732578277588,
287
+ "logps/chosen": -1.658620834350586,
288
+ "logps/rejected": -1.8341089487075806,
289
+ "loss": 0.5784,
290
+ "rewards/accuracies": 0.75,
291
+ "rewards/chosen": -1.3846042156219482,
292
+ "rewards/margins": 0.3775123953819275,
293
+ "rewards/rejected": -1.7621164321899414,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
+ "grad_norm": 31.031052349259785,
299
  "learning_rate": 6.991212490377531e-08,
300
+ "logits/chosen": -1.797126054763794,
301
+ "logits/rejected": -1.7482073307037354,
302
+ "logps/chosen": -1.731537103652954,
303
+ "logps/rejected": -1.9256778955459595,
304
+ "loss": 0.5544,
305
+ "rewards/accuracies": 0.7124999761581421,
306
+ "rewards/chosen": -1.504301905632019,
307
+ "rewards/margins": 0.517795979976654,
308
+ "rewards/rejected": -2.0220980644226074,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
+ "grad_norm": 41.499506266425385,
314
  "learning_rate": 6.620527633276978e-08,
315
+ "logits/chosen": -1.7355127334594727,
316
+ "logits/rejected": -1.6644805669784546,
317
+ "logps/chosen": -1.7263151407241821,
318
+ "logps/rejected": -2.0103392601013184,
319
+ "loss": 0.5689,
320
+ "rewards/accuracies": 0.737500011920929,
321
+ "rewards/chosen": -1.5955860614776611,
322
+ "rewards/margins": 0.591705858707428,
323
+ "rewards/rejected": -2.1872916221618652,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
+ "grad_norm": 34.74514581193027,
329
  "learning_rate": 6.239597278716581e-08,
330
+ "logits/chosen": -1.7984533309936523,
331
+ "logits/rejected": -1.7380235195159912,
332
+ "logps/chosen": -1.856172800064087,
333
+ "logps/rejected": -2.031567096710205,
334
+ "loss": 0.5577,
335
+ "rewards/accuracies": 0.75,
336
+ "rewards/chosen": -1.8325283527374268,
337
+ "rewards/margins": 0.5881733894348145,
338
+ "rewards/rejected": -2.420701742172241,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
+ "grad_norm": 30.0311431190145,
344
  "learning_rate": 5.8508297910462456e-08,
345
+ "logits/chosen": -1.7461389303207397,
346
+ "logits/rejected": -1.665144681930542,
347
+ "logps/chosen": -1.78778874874115,
348
+ "logps/rejected": -2.0407111644744873,
349
+ "loss": 0.5611,
350
+ "rewards/accuracies": 0.731249988079071,
351
+ "rewards/chosen": -1.872820258140564,
352
+ "rewards/margins": 0.5854478478431702,
353
+ "rewards/rejected": -2.4582679271698,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
+ "grad_norm": 33.05734006337458,
359
  "learning_rate": 5.456683083494731e-08,
360
+ "logits/chosen": -1.7369384765625,
361
+ "logits/rejected": -1.69870924949646,
362
+ "logps/chosen": -1.7765703201293945,
363
+ "logps/rejected": -1.9211543798446655,
364
+ "loss": 0.5821,
365
+ "rewards/accuracies": 0.643750011920929,
366
+ "rewards/chosen": -1.7482845783233643,
367
+ "rewards/margins": 0.3625262379646301,
368
+ "rewards/rejected": -2.1108105182647705,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
+ "grad_norm": 42.868772014524744,
374
  "learning_rate": 5.059649078450834e-08,
375
+ "logits/chosen": -1.7265160083770752,
376
+ "logits/rejected": -1.6816940307617188,
377
+ "logps/chosen": -1.7579532861709595,
378
+ "logps/rejected": -1.9656188488006592,
379
+ "loss": 0.5499,
380
+ "rewards/accuracies": 0.65625,
381
+ "rewards/chosen": -1.7687995433807373,
382
+ "rewards/margins": 0.43688035011291504,
383
+ "rewards/rejected": -2.2056798934936523,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
+ "grad_norm": 32.81853514392731,
389
  "learning_rate": 4.6622379527277186e-08,
390
+ "logits/chosen": -1.7367656230926514,
391
+ "logits/rejected": -1.685956358909607,
392
+ "logps/chosen": -1.7992569208145142,
393
+ "logps/rejected": -1.9660634994506836,
394
+ "loss": 0.5585,
395
+ "rewards/accuracies": 0.6625000238418579,
396
+ "rewards/chosen": -1.981104850769043,
397
+ "rewards/margins": 0.4011806845664978,
398
+ "rewards/rejected": -2.3822855949401855,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
+ "grad_norm": 33.27003433225899,
404
  "learning_rate": 4.26696226741691e-08,
405
+ "logits/chosen": -1.7512362003326416,
406
+ "logits/rejected": -1.6821905374526978,
407
+ "logps/chosen": -1.897579550743103,
408
+ "logps/rejected": -2.0977420806884766,
409
+ "loss": 0.5601,
410
+ "rewards/accuracies": 0.6499999761581421,
411
+ "rewards/chosen": -1.9942684173583984,
412
+ "rewards/margins": 0.5405869483947754,
413
+ "rewards/rejected": -2.534855365753174,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
+ "grad_norm": 28.394830302289282,
419
  "learning_rate": 3.876321082668098e-08,
420
+ "logits/chosen": -1.8098827600479126,
421
+ "logits/rejected": -1.7416326999664307,
422
+ "logps/chosen": -1.8279718160629272,
423
+ "logps/rejected": -2.0366756916046143,
424
+ "loss": 0.5516,
425
+ "rewards/accuracies": 0.7562500238418579,
426
+ "rewards/chosen": -1.7684434652328491,
427
+ "rewards/margins": 0.5981845855712891,
428
+ "rewards/rejected": -2.3666281700134277,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
+ "grad_norm": 33.9712584613191,
434
  "learning_rate": 3.492784157826244e-08,
435
+ "logits/chosen": -1.7408695220947266,
436
+ "logits/rejected": -1.6451225280761719,
437
+ "logps/chosen": -1.844485878944397,
438
+ "logps/rejected": -2.0920097827911377,
439
+ "loss": 0.5537,
440
+ "rewards/accuracies": 0.71875,
441
+ "rewards/chosen": -1.864828109741211,
442
+ "rewards/margins": 0.6364026665687561,
443
+ "rewards/rejected": -2.5012307167053223,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
+ "grad_norm": 26.97002465180278,
449
  "learning_rate": 3.118776336817812e-08,
450
+ "logits/chosen": -1.782135009765625,
451
+ "logits/rejected": -1.72174870967865,
452
+ "logps/chosen": -1.8031295537948608,
453
+ "logps/rejected": -2.0307796001434326,
454
+ "loss": 0.5443,
455
+ "rewards/accuracies": 0.71875,
456
+ "rewards/chosen": -1.7996422052383423,
457
+ "rewards/margins": 0.6613016128540039,
458
+ "rewards/rejected": -2.4609439373016357,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
+ "grad_norm": 31.17322900966105,
464
  "learning_rate": 2.7566622175067443e-08,
465
+ "logits/chosen": -1.7659027576446533,
466
+ "logits/rejected": -1.7027981281280518,
467
+ "logps/chosen": -1.8818680047988892,
468
+ "logps/rejected": -2.163147449493408,
469
+ "loss": 0.5502,
470
+ "rewards/accuracies": 0.706250011920929,
471
+ "rewards/chosen": -2.0198538303375244,
472
+ "rewards/margins": 0.6098783612251282,
473
+ "rewards/rejected": -2.629732131958008,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
+ "grad_norm": 36.262422147908055,
479
  "learning_rate": 2.408731201945432e-08,
480
+ "logits/chosen": -1.761371374130249,
481
+ "logits/rejected": -1.7142196893692017,
482
+ "logps/chosen": -1.8247020244598389,
483
+ "logps/rejected": -2.0095391273498535,
484
+ "loss": 0.5485,
485
+ "rewards/accuracies": 0.6875,
486
+ "rewards/chosen": -1.8924195766448975,
487
+ "rewards/margins": 0.49404746294021606,
488
+ "rewards/rejected": -2.386467218399048,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
+ "grad_norm": 32.89905332615021,
494
  "learning_rate": 2.0771830220378112e-08,
495
+ "logits/chosen": -1.7170169353485107,
496
+ "logits/rejected": -1.6577152013778687,
497
+ "logps/chosen": -1.8545100688934326,
498
+ "logps/rejected": -2.0157198905944824,
499
+ "loss": 0.5515,
500
+ "rewards/accuracies": 0.6625000238418579,
501
+ "rewards/chosen": -1.8948768377304077,
502
+ "rewards/margins": 0.5071803331375122,
503
+ "rewards/rejected": -2.40205717086792,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
+ "grad_norm": 29.752828533537873,
509
  "learning_rate": 1.7641138321260257e-08,
510
+ "logits/chosen": -1.7509727478027344,
511
+ "logits/rejected": -1.6813408136367798,
512
+ "logps/chosen": -1.8006902933120728,
513
+ "logps/rejected": -2.068765878677368,
514
+ "loss": 0.5463,
515
+ "rewards/accuracies": 0.8125,
516
+ "rewards/chosen": -1.8505061864852905,
517
+ "rewards/margins": 0.7216218113899231,
518
+ "rewards/rejected": -2.5721280574798584,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
+ "grad_norm": 29.865199582942665,
524
  "learning_rate": 1.4715029564277793e-08,
525
+ "logits/chosen": -1.8179073333740234,
526
+ "logits/rejected": -1.771498680114746,
527
+ "logps/chosen": -1.7910722494125366,
528
+ "logps/rejected": -2.0462353229522705,
529
+ "loss": 0.5539,
530
+ "rewards/accuracies": 0.7437499761581421,
531
+ "rewards/chosen": -1.759235143661499,
532
+ "rewards/margins": 0.6542983651161194,
533
+ "rewards/rejected": -2.4135334491729736,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
+ "grad_norm": 33.27484680109776,
539
  "learning_rate": 1.2012003751113343e-08,
540
+ "logits/chosen": -1.806565284729004,
541
+ "logits/rejected": -1.745749831199646,
542
+ "logps/chosen": -1.8502833843231201,
543
+ "logps/rejected": -2.1222567558288574,
544
+ "loss": 0.5334,
545
+ "rewards/accuracies": 0.7124999761581421,
546
+ "rewards/chosen": -2.0643410682678223,
547
+ "rewards/margins": 0.6653475761413574,
548
+ "rewards/rejected": -2.729688882827759,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
+ "grad_norm": 35.608600933807864,
554
  "learning_rate": 9.549150281252633e-09,
555
+ "logits/chosen": -1.762521505355835,
556
+ "logits/rejected": -1.7128925323486328,
557
+ "logps/chosen": -1.8086223602294922,
558
+ "logps/rejected": -2.0623652935028076,
559
+ "loss": 0.5356,
560
+ "rewards/accuracies": 0.699999988079071,
561
+ "rewards/chosen": -1.9094616174697876,
562
+ "rewards/margins": 0.6060369610786438,
563
+ "rewards/rejected": -2.515498399734497,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
+ "grad_norm": 31.739203329110374,
569
  "learning_rate": 7.3420401072985306e-09,
570
+ "logits/chosen": -1.7991466522216797,
571
+ "logits/rejected": -1.746799111366272,
572
+ "logps/chosen": -1.8659775257110596,
573
+ "logps/rejected": -2.1334667205810547,
574
+ "loss": 0.5356,
575
+ "rewards/accuracies": 0.7124999761581421,
576
+ "rewards/chosen": -1.9540916681289673,
577
+ "rewards/margins": 0.6300768256187439,
578
+ "rewards/rejected": -2.5841686725616455,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
+ "grad_norm": 37.231267463423904,
584
  "learning_rate": 5.404627290395369e-09,
585
+ "logits/chosen": -1.7623279094696045,
586
+ "logits/rejected": -1.6976515054702759,
587
+ "logps/chosen": -1.7954782247543335,
588
+ "logps/rejected": -2.0490074157714844,
589
+ "loss": 0.5377,
590
+ "rewards/accuracies": 0.7562500238418579,
591
+ "rewards/chosen": -1.8176171779632568,
592
+ "rewards/margins": 0.6405991315841675,
593
+ "rewards/rejected": -2.4582161903381348,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
+ "grad_norm": 48.31129307964873,
599
  "learning_rate": 3.74916077816162e-09,
600
+ "logits/chosen": -1.7699306011199951,
601
+ "logits/rejected": -1.7098052501678467,
602
+ "logps/chosen": -1.7985498905181885,
603
+ "logps/rejected": -1.9994351863861084,
604
+ "loss": 0.5502,
605
+ "rewards/accuracies": 0.699999988079071,
606
+ "rewards/chosen": -1.946265459060669,
607
+ "rewards/margins": 0.5542693734169006,
608
+ "rewards/rejected": -2.5005345344543457,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
+ "grad_norm": 36.09144092538641,
614
  "learning_rate": 2.386106962899165e-09,
615
+ "logits/chosen": -1.6945453882217407,
616
+ "logits/rejected": -1.62551748752594,
617
+ "logps/chosen": -1.9125568866729736,
618
+ "logps/rejected": -2.1548168659210205,
619
+ "loss": 0.5447,
620
+ "rewards/accuracies": 0.7124999761581421,
621
+ "rewards/chosen": -2.05840802192688,
622
+ "rewards/margins": 0.5577607154846191,
623
+ "rewards/rejected": -2.61616849899292,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
+ "grad_norm": 35.37496893135356,
629
  "learning_rate": 1.3240835096913706e-09,
630
+ "logits/chosen": -1.7273473739624023,
631
+ "logits/rejected": -1.6288312673568726,
632
+ "logps/chosen": -1.787335753440857,
633
+ "logps/rejected": -2.1021180152893066,
634
+ "loss": 0.5546,
635
+ "rewards/accuracies": 0.731249988079071,
636
+ "rewards/chosen": -1.8473552465438843,
637
+ "rewards/margins": 0.7361037731170654,
638
+ "rewards/rejected": -2.5834591388702393,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
+ "grad_norm": 32.802494867000924,
644
  "learning_rate": 5.698048727497462e-10,
645
+ "logits/chosen": -1.7570117712020874,
646
+ "logits/rejected": -1.6864467859268188,
647
+ "logps/chosen": -1.8803859949111938,
648
+ "logps/rejected": -2.1749351024627686,
649
+ "loss": 0.5408,
650
+ "rewards/accuracies": 0.737500011920929,
651
+ "rewards/chosen": -2.0672643184661865,
652
+ "rewards/margins": 0.6827760338783264,
653
+ "rewards/rejected": -2.7500405311584473,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
+ "grad_norm": 30.44073541133352,
659
  "learning_rate": 1.2803984447259387e-10,
660
+ "logits/chosen": -1.7648996114730835,
661
+ "logits/rejected": -1.713091254234314,
662
+ "logps/chosen": -1.8600318431854248,
663
+ "logps/rejected": -2.127103805541992,
664
+ "loss": 0.5345,
665
+ "rewards/accuracies": 0.71875,
666
+ "rewards/chosen": -2.0330777168273926,
667
+ "rewards/margins": 0.6582567095756531,
668
+ "rewards/rejected": -2.6913342475891113,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
+ "train_loss": 0.5807303016441016,
676
+ "train_runtime": 6860.7908,
677
+ "train_samples_per_second": 8.197,
678
+ "train_steps_per_second": 0.064
679
  }
680
  ],
681
  "logging_steps": 10,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a7e7e48ed484507dc4bf687f99b9afd74522a0ce37982f50ef4c64c14f4f420c
3
  size 6328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52fd958028d2a8e77c3db64f0cef52c04c727dbfa54965417b4be2b66b01aefd
3
  size 6328