wzhouad commited on
Commit
10b4efc
1 Parent(s): 0d4f8c7

Model save

Browse files
README.md CHANGED
@@ -32,7 +32,7 @@ More information needed
32
  ### Training hyperparameters
33
 
34
  The following hyperparameters were used during training:
35
- - learning_rate: 1e-06
36
  - train_batch_size: 2
37
  - eval_batch_size: 8
38
  - seed: 1
 
32
  ### Training hyperparameters
33
 
34
  The following hyperparameters were used during training:
35
+ - learning_rate: 5e-06
36
  - train_batch_size: 2
37
  - eval_batch_size: 8
38
  - seed: 1
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 2.0,
3
- "train_loss": 0.25192113166124047,
4
- "train_runtime": 8113.4503,
5
  "train_samples": 45548,
6
- "train_samples_per_second": 11.228,
7
- "train_steps_per_second": 0.088
8
  }
 
1
  {
2
  "epoch": 2.0,
3
+ "train_loss": 0.1250429530667377,
4
+ "train_runtime": 7954.4917,
5
  "train_samples": 45548,
6
+ "train_samples_per_second": 11.452,
7
+ "train_steps_per_second": 0.089
8
  }
model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3402aa3938e1c31d822d0e375cf01dcc5f8377b7a31a3b6cf0575f1fb7953d4a
3
  size 4976698672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb89283c64a9aa24f68c2e6ff899db2a4eff0089ec7f651d36a630d06ead65c6
3
  size 4976698672
model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:655f95ca04b0961ed4b515a7348c05cf72428837d0d1656bb20de7415dd6d23b
3
  size 4999802720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e4e4da391be8ea6eb88812a2d53d80df1fe445d950f4dab4b0ab10af79c538e
3
  size 4999802720
model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c8c376df770dfdaedca5de952735c527acff55b46dffecb9de5c18634541d6e2
3
  size 4915916176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91ff76793cb8b35d103a71c2a1031a7a34debec12bd3585181646e51a457a68e
3
  size 4915916176
model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a996e33d039658928f6a70a590d6caec1f6957e1db80bebabe1c6f9186fd855d
3
  size 1168138808
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d07159171f547d0196276801d699317fda916d2e68e7f02ad52368f2eca2348e
3
  size 1168138808
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 2.0,
3
- "train_loss": 0.25192113166124047,
4
- "train_runtime": 8113.4503,
5
  "train_samples": 45548,
6
- "train_samples_per_second": 11.228,
7
- "train_steps_per_second": 0.088
8
  }
 
1
  {
2
  "epoch": 2.0,
3
+ "train_loss": 0.1250429530667377,
4
+ "train_runtime": 7954.4917,
5
  "train_samples": 45548,
6
+ "train_samples_per_second": 11.452,
7
+ "train_steps_per_second": 0.089
8
  }
trainer_state.json CHANGED
@@ -10,7 +10,7 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
- "learning_rate": 1.408450704225352e-08,
14
  "logits/chosen": -0.20257116854190826,
15
  "logits/rejected": -0.07568661123514175,
16
  "logps/chosen": -521.0269775390625,
@@ -24,1006 +24,1006 @@
24
  },
25
  {
26
  "epoch": 0.03,
27
- "learning_rate": 1.4084507042253522e-07,
28
- "logits/chosen": -0.13786724209785461,
29
- "logits/rejected": -0.08816312253475189,
30
- "logps/chosen": -345.8097229003906,
31
- "logps/rejected": -280.6612854003906,
32
- "loss": 0.3323,
33
- "rewards/accuracies": 0.4444444477558136,
34
- "rewards/chosen": 0.0006422583828680217,
35
- "rewards/margins": 0.0004346017958596349,
36
- "rewards/rejected": 0.00020765644148923457,
37
  "step": 10
38
  },
39
  {
40
  "epoch": 0.06,
41
- "learning_rate": 2.8169014084507043e-07,
42
- "logits/chosen": -0.09473811089992523,
43
- "logits/rejected": -0.056791383773088455,
44
- "logps/chosen": -331.81280517578125,
45
- "logps/rejected": -241.23910522460938,
46
- "loss": 0.3272,
47
- "rewards/accuracies": 0.5562499761581421,
48
- "rewards/chosen": 0.0002513823565095663,
49
- "rewards/margins": 0.0021114428527653217,
50
- "rewards/rejected": -0.001860060729086399,
51
  "step": 20
52
  },
53
  {
54
  "epoch": 0.08,
55
- "learning_rate": 4.225352112676056e-07,
56
- "logits/chosen": -0.05887429043650627,
57
- "logits/rejected": -0.0021190904080867767,
58
- "logps/chosen": -387.2514953613281,
59
- "logps/rejected": -252.19833374023438,
60
- "loss": 0.3291,
61
- "rewards/accuracies": 0.612500011920929,
62
- "rewards/chosen": 0.004256925545632839,
63
- "rewards/margins": 0.018127940595149994,
64
- "rewards/rejected": -0.01387101411819458,
65
  "step": 30
66
  },
67
  {
68
  "epoch": 0.11,
69
- "learning_rate": 5.633802816901409e-07,
70
- "logits/chosen": -0.07777061313390732,
71
- "logits/rejected": -0.023820143193006516,
72
- "logps/chosen": -335.6570129394531,
73
- "logps/rejected": -263.23907470703125,
74
- "loss": 0.3421,
75
- "rewards/accuracies": 0.5375000238418579,
76
- "rewards/chosen": -0.004789065103977919,
77
- "rewards/margins": 0.021946147084236145,
78
- "rewards/rejected": -0.02673521265387535,
79
  "step": 40
80
  },
81
  {
82
  "epoch": 0.14,
83
- "learning_rate": 7.04225352112676e-07,
84
- "logits/chosen": -0.03277156129479408,
85
- "logits/rejected": 0.012101896107196808,
86
- "logps/chosen": -336.37176513671875,
87
- "logps/rejected": -272.0014953613281,
88
- "loss": 0.358,
89
- "rewards/accuracies": 0.59375,
90
- "rewards/chosen": -0.026565855368971825,
91
- "rewards/margins": 0.06764300167560577,
92
- "rewards/rejected": -0.09420885145664215,
93
  "step": 50
94
  },
95
  {
96
  "epoch": 0.17,
97
- "learning_rate": 8.450704225352112e-07,
98
- "logits/chosen": -0.11417659372091293,
99
- "logits/rejected": -0.05113930255174637,
100
- "logps/chosen": -427.59600830078125,
101
- "logps/rejected": -268.9815673828125,
102
- "loss": 0.3784,
103
- "rewards/accuracies": 0.6875,
104
- "rewards/chosen": -0.052729975432157516,
105
- "rewards/margins": 0.18380871415138245,
106
- "rewards/rejected": -0.23653869330883026,
107
  "step": 60
108
  },
109
  {
110
  "epoch": 0.2,
111
- "learning_rate": 9.859154929577465e-07,
112
- "logits/chosen": -0.12808646261692047,
113
- "logits/rejected": -0.009413952939212322,
114
- "logps/chosen": -417.78680419921875,
115
- "logps/rejected": -287.1832580566406,
116
- "loss": 0.3848,
117
- "rewards/accuracies": 0.5874999761581421,
118
- "rewards/chosen": -0.10650447756052017,
119
- "rewards/margins": 0.15690955519676208,
120
- "rewards/rejected": -0.26341402530670166,
121
  "step": 70
122
  },
123
  {
124
  "epoch": 0.22,
125
- "learning_rate": 9.995106132599868e-07,
126
- "logits/chosen": -0.0769750103354454,
127
- "logits/rejected": -0.06951722502708435,
128
- "logps/chosen": -294.3128662109375,
129
- "logps/rejected": -227.47463989257812,
130
- "loss": 0.3948,
131
- "rewards/accuracies": 0.606249988079071,
132
- "rewards/chosen": -0.06992264091968536,
133
- "rewards/margins": 0.16703224182128906,
134
- "rewards/rejected": -0.23695488274097443,
135
  "step": 80
136
  },
137
  {
138
  "epoch": 0.25,
139
- "learning_rate": 9.978201358980644e-07,
140
- "logits/chosen": -0.2559957504272461,
141
- "logits/rejected": -0.20053008198738098,
142
- "logps/chosen": -389.33929443359375,
143
- "logps/rejected": -247.1188201904297,
144
- "loss": 0.4103,
145
- "rewards/accuracies": 0.7124999761581421,
146
- "rewards/chosen": 0.11513955891132355,
147
- "rewards/margins": 0.3396330177783966,
148
- "rewards/rejected": -0.22449345886707306,
149
  "step": 90
150
  },
151
  {
152
  "epoch": 0.28,
153
- "learning_rate": 9.949266103908894e-07,
154
- "logits/chosen": -0.18818387389183044,
155
- "logits/rejected": -0.16776612401008606,
156
- "logps/chosen": -360.2618103027344,
157
- "logps/rejected": -283.3966979980469,
158
- "loss": 0.4011,
159
- "rewards/accuracies": 0.6312500238418579,
160
- "rewards/chosen": 0.07560204714536667,
161
- "rewards/margins": 0.13260915875434875,
162
- "rewards/rejected": -0.05700710415840149,
163
  "step": 100
164
  },
165
  {
166
  "epoch": 0.31,
167
- "learning_rate": 9.908370293252287e-07,
168
- "logits/chosen": -0.2396635115146637,
169
- "logits/rejected": -0.16480541229248047,
170
- "logps/chosen": -319.79925537109375,
171
- "logps/rejected": -279.1387023925781,
172
- "loss": 0.3716,
173
- "rewards/accuracies": 0.5375000238418579,
174
- "rewards/chosen": 0.08118756860494614,
175
- "rewards/margins": 0.0795254111289978,
176
- "rewards/rejected": 0.0016621515387669206,
177
  "step": 110
178
  },
179
  {
180
  "epoch": 0.34,
181
- "learning_rate": 9.855612757141654e-07,
182
- "logits/chosen": -0.2065432071685791,
183
- "logits/rejected": -0.17942771315574646,
184
- "logps/chosen": -320.2702941894531,
185
- "logps/rejected": -222.2958221435547,
186
- "loss": 0.3645,
187
- "rewards/accuracies": 0.6312500238418579,
188
- "rewards/chosen": 0.12114707380533218,
189
- "rewards/margins": 0.17006908357143402,
190
- "rewards/rejected": -0.04892200976610184,
191
  "step": 120
192
  },
193
  {
194
  "epoch": 0.37,
195
- "learning_rate": 9.791120991134902e-07,
196
- "logits/chosen": -0.162322998046875,
197
- "logits/rejected": -0.08893263339996338,
198
- "logps/chosen": -376.00360107421875,
199
- "logps/rejected": -244.6248016357422,
200
- "loss": 0.3721,
201
- "rewards/accuracies": 0.637499988079071,
202
- "rewards/chosen": 0.10144136846065521,
203
- "rewards/margins": 0.25357523560523987,
204
- "rewards/rejected": -0.15213388204574585,
205
  "step": 130
206
  },
207
  {
208
  "epoch": 0.39,
209
- "learning_rate": 9.715050848107168e-07,
210
- "logits/chosen": -0.14301994442939758,
211
- "logits/rejected": -0.13107290863990784,
212
- "logps/chosen": -337.5934753417969,
213
- "logps/rejected": -264.13153076171875,
214
- "loss": 0.3693,
215
- "rewards/accuracies": 0.550000011920929,
216
- "rewards/chosen": -0.008726155385375023,
217
- "rewards/margins": 0.13979385793209076,
218
- "rewards/rejected": -0.14851999282836914,
219
  "step": 140
220
  },
221
  {
222
  "epoch": 0.42,
223
- "learning_rate": 9.627586161611731e-07,
224
- "logits/chosen": -0.25995808839797974,
225
- "logits/rejected": -0.22309823334217072,
226
- "logps/chosen": -305.5233459472656,
227
- "logps/rejected": -264.06097412109375,
228
- "loss": 0.387,
229
- "rewards/accuracies": 0.612500011920929,
230
- "rewards/chosen": -0.004295418970286846,
231
- "rewards/margins": 0.1561904400587082,
232
- "rewards/rejected": -0.16048584878444672,
233
  "step": 150
234
  },
235
  {
236
  "epoch": 0.45,
237
- "learning_rate": 9.528938301621955e-07,
238
- "logits/chosen": -0.17277738451957703,
239
- "logits/rejected": -0.054037339985370636,
240
- "logps/chosen": -388.6319885253906,
241
- "logps/rejected": -281.8338317871094,
242
- "loss": 0.3624,
243
- "rewards/accuracies": 0.581250011920929,
244
- "rewards/chosen": -0.027996420860290527,
245
- "rewards/margins": 0.2142985761165619,
246
- "rewards/rejected": -0.2422950267791748,
247
  "step": 160
248
  },
249
  {
250
  "epoch": 0.48,
251
- "learning_rate": 9.419345663727804e-07,
252
- "logits/chosen": 0.1668136566877365,
253
- "logits/rejected": 0.24267610907554626,
254
- "logps/chosen": -348.5519104003906,
255
- "logps/rejected": -298.996826171875,
256
- "loss": 0.3338,
257
- "rewards/accuracies": 0.59375,
258
- "rewards/chosen": -0.16783149540424347,
259
- "rewards/margins": 0.18178348243236542,
260
- "rewards/rejected": -0.3496149778366089,
261
  "step": 170
262
  },
263
  {
264
  "epoch": 0.51,
265
- "learning_rate": 9.299073093021404e-07,
266
- "logits/chosen": 0.047697070986032486,
267
- "logits/rejected": 0.1468953937292099,
268
- "logps/chosen": -363.13934326171875,
269
- "logps/rejected": -292.49822998046875,
270
- "loss": 0.3368,
271
- "rewards/accuracies": 0.606249988079071,
272
- "rewards/chosen": -0.21875329315662384,
273
- "rewards/margins": 0.1854187548160553,
274
- "rewards/rejected": -0.40417203307151794,
275
  "step": 180
276
  },
277
  {
278
  "epoch": 0.53,
279
- "learning_rate": 9.168411244063861e-07,
280
- "logits/chosen": 0.016937877982854843,
281
- "logits/rejected": 0.10573382675647736,
282
- "logps/chosen": -369.8836975097656,
283
- "logps/rejected": -270.32574462890625,
284
- "loss": 0.3274,
285
- "rewards/accuracies": 0.643750011920929,
286
- "rewards/chosen": -0.134426087141037,
287
- "rewards/margins": 0.2750268578529358,
288
- "rewards/rejected": -0.40945297479629517,
289
  "step": 190
290
  },
291
  {
292
  "epoch": 0.56,
293
- "learning_rate": 9.02767587848013e-07,
294
- "logits/chosen": -0.031068842858076096,
295
- "logits/rejected": 0.08885425329208374,
296
- "logps/chosen": -358.65374755859375,
297
- "logps/rejected": -290.309814453125,
298
- "loss": 0.3447,
299
- "rewards/accuracies": 0.5625,
300
- "rewards/chosen": -0.28612247109413147,
301
- "rewards/margins": 0.17394599318504333,
302
- "rewards/rejected": -0.4600685238838196,
303
  "step": 200
304
  },
305
  {
306
  "epoch": 0.59,
307
- "learning_rate": 8.877207101879301e-07,
308
- "logits/chosen": 0.1502365916967392,
309
- "logits/rejected": 0.24661250412464142,
310
- "logps/chosen": -428.3617248535156,
311
- "logps/rejected": -313.82257080078125,
312
- "loss": 0.3299,
313
- "rewards/accuracies": 0.5687500238418579,
314
- "rewards/chosen": -0.25656384229660034,
315
- "rewards/margins": 0.19923393428325653,
316
- "rewards/rejected": -0.4557977616786957,
317
  "step": 210
318
  },
319
  {
320
  "epoch": 0.62,
321
- "learning_rate": 8.717368541944452e-07,
322
- "logits/chosen": 0.08803655207157135,
323
- "logits/rejected": 0.1639718860387802,
324
- "logps/chosen": -423.03228759765625,
325
- "logps/rejected": -313.6676025390625,
326
- "loss": 0.3218,
327
- "rewards/accuracies": 0.637499988079071,
328
- "rewards/chosen": -0.3232230246067047,
329
- "rewards/margins": 0.1633337289094925,
330
- "rewards/rejected": -0.4865567088127136,
331
  "step": 220
332
  },
333
  {
334
  "epoch": 0.65,
335
- "learning_rate": 8.54854646967831e-07,
336
- "logits/chosen": -0.0030194534920156,
337
- "logits/rejected": 0.14094045758247375,
338
- "logps/chosen": -369.0701904296875,
339
- "logps/rejected": -286.33990478515625,
340
- "loss": 0.3382,
341
- "rewards/accuracies": 0.612500011920929,
342
- "rewards/chosen": -0.18699464201927185,
343
- "rewards/margins": 0.23964881896972656,
344
- "rewards/rejected": -0.4266434609889984,
345
  "step": 230
346
  },
347
  {
348
  "epoch": 0.67,
349
- "learning_rate": 8.371148865928318e-07,
350
- "logits/chosen": -0.2055165320634842,
351
- "logits/rejected": -0.08643672615289688,
352
- "logps/chosen": -377.8648986816406,
353
- "logps/rejected": -279.52703857421875,
354
- "loss": 0.3443,
355
- "rewards/accuracies": 0.6812499761581421,
356
- "rewards/chosen": -0.19047299027442932,
357
- "rewards/margins": 0.21315133571624756,
358
- "rewards/rejected": -0.40362435579299927,
359
  "step": 240
360
  },
361
  {
362
  "epoch": 0.7,
363
- "learning_rate": 8.185604435447001e-07,
364
- "logits/chosen": -0.21048621833324432,
365
- "logits/rejected": -0.12320221960544586,
366
- "logps/chosen": -396.4952392578125,
367
- "logps/rejected": -284.85693359375,
368
- "loss": 0.3476,
369
- "rewards/accuracies": 0.625,
370
- "rewards/chosen": -0.14602257311344147,
371
- "rewards/margins": 0.29507723450660706,
372
- "rewards/rejected": -0.4410998225212097,
373
  "step": 250
374
  },
375
  {
376
  "epoch": 0.73,
377
- "learning_rate": 7.992361570870287e-07,
378
- "logits/chosen": -0.09357413649559021,
379
- "logits/rejected": -0.02147941291332245,
380
- "logps/chosen": -391.3370666503906,
381
- "logps/rejected": -304.73248291015625,
382
- "loss": 0.3457,
383
- "rewards/accuracies": 0.6625000238418579,
384
- "rewards/chosen": -0.30016764998435974,
385
- "rewards/margins": 0.2906726002693176,
386
- "rewards/rejected": -0.590840220451355,
387
  "step": 260
388
  },
389
  {
390
  "epoch": 0.76,
391
- "learning_rate": 7.791887269117441e-07,
392
- "logits/chosen": -0.01912299357354641,
393
- "logits/rejected": 0.015499010682106018,
394
- "logps/chosen": -385.39788818359375,
395
- "logps/rejected": -318.03045654296875,
396
- "loss": 0.3332,
397
- "rewards/accuracies": 0.6000000238418579,
398
- "rewards/chosen": -0.46580857038497925,
399
- "rewards/margins": 0.2723786234855652,
400
- "rewards/rejected": -0.7381871938705444,
401
  "step": 270
402
  },
403
  {
404
  "epoch": 0.79,
405
- "learning_rate": 7.584666002831294e-07,
406
- "logits/chosen": -0.1838226318359375,
407
- "logits/rejected": -0.11540188640356064,
408
- "logps/chosen": -431.819580078125,
409
- "logps/rejected": -314.17315673828125,
410
- "loss": 0.3741,
411
- "rewards/accuracies": 0.65625,
412
- "rewards/chosen": -0.38983967900276184,
413
- "rewards/margins": 0.3283058702945709,
414
- "rewards/rejected": -0.718145489692688,
415
  "step": 280
416
  },
417
  {
418
  "epoch": 0.81,
419
- "learning_rate": 7.37119854958609e-07,
420
- "logits/chosen": -0.16166582703590393,
421
- "logits/rejected": -0.06856708228588104,
422
- "logps/chosen": -389.5946960449219,
423
- "logps/rejected": -316.12677001953125,
424
- "loss": 0.3638,
425
- "rewards/accuracies": 0.6000000238418579,
426
- "rewards/chosen": -0.3782878816127777,
427
- "rewards/margins": 0.20532508194446564,
428
- "rewards/rejected": -0.5836129188537598,
429
  "step": 290
430
  },
431
  {
432
  "epoch": 0.84,
433
- "learning_rate": 7.152000781692285e-07,
434
- "logits/chosen": -0.15225966274738312,
435
- "logits/rejected": 0.023759862408041954,
436
- "logps/chosen": -398.35040283203125,
437
- "logps/rejected": -296.66046142578125,
438
- "loss": 0.3393,
439
- "rewards/accuracies": 0.606249988079071,
440
- "rewards/chosen": -0.21167917549610138,
441
- "rewards/margins": 0.28109148144721985,
442
- "rewards/rejected": -0.4927706718444824,
443
  "step": 300
444
  },
445
  {
446
  "epoch": 0.87,
447
- "learning_rate": 6.927602419522946e-07,
448
- "logits/chosen": -0.1469275802373886,
449
- "logits/rejected": 0.009778399951756,
450
- "logps/chosen": -370.21051025390625,
451
- "logps/rejected": -298.3064880371094,
452
- "loss": 0.3295,
453
- "rewards/accuracies": 0.6499999761581421,
454
- "rewards/chosen": -0.29429182410240173,
455
- "rewards/margins": 0.266606867313385,
456
- "rewards/rejected": -0.5608987212181091,
457
  "step": 310
458
  },
459
  {
460
  "epoch": 0.9,
461
- "learning_rate": 6.698545751374463e-07,
462
- "logits/chosen": -0.008808034472167492,
463
- "logits/rejected": 0.06853709369897842,
464
- "logps/chosen": -371.8343200683594,
465
- "logps/rejected": -314.5874328613281,
466
- "loss": 0.3437,
467
- "rewards/accuracies": 0.59375,
468
- "rewards/chosen": -0.3911176323890686,
469
- "rewards/margins": 0.1818263977766037,
470
- "rewards/rejected": -0.5729440450668335,
471
  "step": 320
472
  },
473
  {
474
  "epoch": 0.93,
475
- "learning_rate": 6.465384322955224e-07,
476
- "logits/chosen": -0.17208532989025116,
477
- "logits/rejected": -0.05994819477200508,
478
- "logps/chosen": -450.02850341796875,
479
- "logps/rejected": -345.80419921875,
480
- "loss": 0.3342,
481
- "rewards/accuracies": 0.59375,
482
- "rewards/chosen": -0.3475356698036194,
483
- "rewards/margins": 0.32027798891067505,
484
- "rewards/rejected": -0.6678135991096497,
485
  "step": 330
486
  },
487
  {
488
  "epoch": 0.96,
489
- "learning_rate": 6.228681599669248e-07,
490
- "logits/chosen": 0.0015781521797180176,
491
- "logits/rejected": 0.043049950152635574,
492
- "logps/chosen": -387.4454345703125,
493
- "logps/rejected": -302.407958984375,
494
- "loss": 0.3188,
495
- "rewards/accuracies": 0.6000000238418579,
496
- "rewards/chosen": -0.3666529059410095,
497
- "rewards/margins": 0.25499850511550903,
498
- "rewards/rejected": -0.6216514110565186,
499
  "step": 340
500
  },
501
  {
502
  "epoch": 0.98,
503
- "learning_rate": 5.989009604927586e-07,
504
- "logits/chosen": -0.03641257435083389,
505
- "logits/rejected": 0.037856753915548325,
506
- "logps/chosen": -401.0362854003906,
507
- "logps/rejected": -317.9659729003906,
508
- "loss": 0.3231,
509
- "rewards/accuracies": 0.6875,
510
- "rewards/chosen": -0.3948978781700134,
511
- "rewards/margins": 0.2911498546600342,
512
- "rewards/rejected": -0.6860476732254028,
513
  "step": 350
514
  },
515
  {
516
  "epoch": 1.01,
517
- "learning_rate": 5.74694753777815e-07,
518
- "logits/chosen": -0.12373703718185425,
519
- "logits/rejected": -0.014805403538048267,
520
- "logps/chosen": -356.97100830078125,
521
- "logps/rejected": -344.86102294921875,
522
- "loss": 0.3052,
523
- "rewards/accuracies": 0.643750011920929,
524
- "rewards/chosen": -0.41758236289024353,
525
- "rewards/margins": 0.3036467730998993,
526
- "rewards/rejected": -0.721229076385498,
527
  "step": 360
528
  },
529
  {
530
  "epoch": 1.04,
531
- "learning_rate": 5.503080373194666e-07,
532
- "logits/chosen": -0.13851913809776306,
533
- "logits/rejected": -0.07055823504924774,
534
- "logps/chosen": -396.57769775390625,
535
- "logps/rejected": -335.1228942871094,
536
- "loss": 0.2391,
537
- "rewards/accuracies": 0.7749999761581421,
538
- "rewards/chosen": -0.39428186416625977,
539
- "rewards/margins": 0.44222506880760193,
540
- "rewards/rejected": -0.8365070223808289,
541
  "step": 370
542
  },
543
  {
544
  "epoch": 1.07,
545
- "learning_rate": 5.257997448407366e-07,
546
- "logits/chosen": -0.13537368178367615,
547
- "logits/rejected": 0.009506871923804283,
548
- "logps/chosen": -375.4892578125,
549
- "logps/rejected": -355.96136474609375,
550
- "loss": 0.232,
551
- "rewards/accuracies": 0.7562500238418579,
552
- "rewards/chosen": -0.5489501357078552,
553
- "rewards/margins": 0.44577375054359436,
554
- "rewards/rejected": -0.9947239756584167,
555
  "step": 380
556
  },
557
  {
558
  "epoch": 1.1,
559
- "learning_rate": 5.012291038691665e-07,
560
- "logits/chosen": -0.292557954788208,
561
- "logits/rejected": -0.12162590026855469,
562
- "logps/chosen": -359.124755859375,
563
- "logps/rejected": -338.5267333984375,
564
- "loss": 0.2175,
565
- "rewards/accuracies": 0.71875,
566
- "rewards/chosen": -0.7096319794654846,
567
- "rewards/margins": 0.47353801131248474,
568
- "rewards/rejected": -1.1831700801849365,
569
  "step": 390
570
  },
571
  {
572
  "epoch": 1.12,
573
- "learning_rate": 4.7665549260567063e-07,
574
- "logits/chosen": -0.24239158630371094,
575
- "logits/rejected": -0.13934944570064545,
576
- "logps/chosen": -426.60369873046875,
577
- "logps/rejected": -403.0608215332031,
578
- "loss": 0.1717,
579
- "rewards/accuracies": 0.78125,
580
- "rewards/chosen": -0.8361096382141113,
581
- "rewards/margins": 0.5920412540435791,
582
- "rewards/rejected": -1.4281507730484009,
583
  "step": 400
584
  },
585
  {
586
  "epoch": 1.15,
587
- "learning_rate": 4.521382964292663e-07,
588
- "logits/chosen": -0.040043365210294724,
589
- "logits/rejected": -0.01921779289841652,
590
- "logps/chosen": -417.27215576171875,
591
- "logps/rejected": -388.7860412597656,
592
- "loss": 0.1755,
593
- "rewards/accuracies": 0.7749999761581421,
594
- "rewards/chosen": -0.9553691744804382,
595
- "rewards/margins": 0.6799647212028503,
596
- "rewards/rejected": -1.6353340148925781,
597
  "step": 410
598
  },
599
  {
600
  "epoch": 1.18,
601
- "learning_rate": 4.277367643844574e-07,
602
- "logits/chosen": -0.097591333091259,
603
- "logits/rejected": 0.017194192856550217,
604
- "logps/chosen": -437.2616271972656,
605
- "logps/rejected": -411.5907287597656,
606
- "loss": 0.1823,
607
- "rewards/accuracies": 0.7250000238418579,
608
- "rewards/chosen": -1.0141613483428955,
609
- "rewards/margins": 0.5529431700706482,
610
- "rewards/rejected": -1.567104458808899,
611
  "step": 420
612
  },
613
  {
614
  "epoch": 1.21,
615
- "learning_rate": 4.035098659980891e-07,
616
- "logits/chosen": 0.12445954233407974,
617
- "logits/rejected": 0.16896803677082062,
618
- "logps/chosen": -484.2261657714844,
619
- "logps/rejected": -403.1665344238281,
620
- "loss": 0.1577,
621
- "rewards/accuracies": 0.8125,
622
- "rewards/chosen": -0.9584755897521973,
623
- "rewards/margins": 0.7136095762252808,
624
- "rewards/rejected": -1.672085165977478,
625
  "step": 430
626
  },
627
  {
628
  "epoch": 1.24,
629
- "learning_rate": 3.795161487716928e-07,
630
- "logits/chosen": 0.10622860491275787,
631
- "logits/rejected": 0.20460453629493713,
632
- "logps/chosen": -416.31878662109375,
633
- "logps/rejected": -390.43353271484375,
634
- "loss": 0.1601,
635
- "rewards/accuracies": 0.731249988079071,
636
- "rewards/chosen": -1.0357643365859985,
637
- "rewards/margins": 0.6402472257614136,
638
- "rewards/rejected": -1.6760118007659912,
639
  "step": 440
640
  },
641
  {
642
  "epoch": 1.26,
643
- "learning_rate": 3.5581359669371223e-07,
644
- "logits/chosen": 0.04869859293103218,
645
- "logits/rejected": 0.1988603174686432,
646
- "logps/chosen": -459.97161865234375,
647
- "logps/rejected": -406.51129150390625,
648
- "loss": 0.1485,
649
- "rewards/accuracies": 0.8062499761581421,
650
- "rewards/chosen": -0.9077569246292114,
651
- "rewards/margins": 0.6647528409957886,
652
- "rewards/rejected": -1.572509527206421,
653
  "step": 450
654
  },
655
  {
656
  "epoch": 1.29,
657
- "learning_rate": 3.324594901135326e-07,
658
- "logits/chosen": 0.09600436687469482,
659
- "logits/rejected": 0.2769695520401001,
660
- "logps/chosen": -412.62481689453125,
661
- "logps/rejected": -427.94573974609375,
662
- "loss": 0.1594,
663
- "rewards/accuracies": 0.762499988079071,
664
- "rewards/chosen": -0.8976230621337891,
665
- "rewards/margins": 0.6307999491691589,
666
- "rewards/rejected": -1.5284230709075928,
667
  "step": 460
668
  },
669
  {
670
  "epoch": 1.32,
671
- "learning_rate": 3.095102673159463e-07,
672
- "logits/chosen": -0.04745341092348099,
673
- "logits/rejected": 0.04035920277237892,
674
- "logps/chosen": -448.31304931640625,
675
- "logps/rejected": -434.83026123046875,
676
- "loss": 0.162,
677
- "rewards/accuracies": 0.800000011920929,
678
- "rewards/chosen": -1.009493112564087,
679
- "rewards/margins": 0.8154042363166809,
680
- "rewards/rejected": -1.8248971700668335,
681
  "step": 470
682
  },
683
  {
684
  "epoch": 1.35,
685
- "learning_rate": 2.870213881305802e-07,
686
- "logits/chosen": 0.003581273602321744,
687
- "logits/rejected": 0.15165765583515167,
688
- "logps/chosen": -442.5594177246094,
689
- "logps/rejected": -411.17144775390625,
690
- "loss": 0.1555,
691
  "rewards/accuracies": 0.800000011920929,
692
- "rewards/chosen": -0.9550750851631165,
693
- "rewards/margins": 0.7856963872909546,
694
- "rewards/rejected": -1.7407715320587158,
695
  "step": 480
696
  },
697
  {
698
  "epoch": 1.38,
699
- "learning_rate": 2.6504719990588745e-07,
700
- "logits/chosen": -0.00950574316084385,
701
- "logits/rejected": 0.15859754383563995,
702
- "logps/chosen": -484.21954345703125,
703
- "logps/rejected": -441.0923767089844,
704
- "loss": 0.1608,
705
- "rewards/accuracies": 0.78125,
706
- "rewards/chosen": -1.0694897174835205,
707
- "rewards/margins": 0.7761608362197876,
708
- "rewards/rejected": -1.8456504344940186,
709
  "step": 490
710
  },
711
  {
712
  "epoch": 1.4,
713
- "learning_rate": 2.436408061715988e-07,
714
- "logits/chosen": -0.03855917602777481,
715
- "logits/rejected": 0.09556344896554947,
716
- "logps/chosen": -491.30523681640625,
717
- "logps/rejected": -452.349609375,
718
- "loss": 0.1511,
719
- "rewards/accuracies": 0.8125,
720
- "rewards/chosen": -1.0780155658721924,
721
- "rewards/margins": 0.823270320892334,
722
- "rewards/rejected": -1.9012857675552368,
723
  "step": 500
724
  },
725
  {
726
  "epoch": 1.43,
727
- "learning_rate": 2.22853938307025e-07,
728
- "logits/chosen": 0.043701499700546265,
729
- "logits/rejected": 0.22214682400226593,
730
- "logps/chosen": -456.72662353515625,
731
- "logps/rejected": -426.0232849121094,
732
- "loss": 0.1501,
733
- "rewards/accuracies": 0.8374999761581421,
734
- "rewards/chosen": -0.9664266705513,
735
- "rewards/margins": 0.6386443376541138,
736
- "rewards/rejected": -1.6050708293914795,
737
  "step": 510
738
  },
739
  {
740
  "epoch": 1.46,
741
- "learning_rate": 2.0273683052534173e-07,
742
- "logits/chosen": 0.0689583569765091,
743
- "logits/rejected": 0.1506797820329666,
744
- "logps/chosen": -445.80133056640625,
745
- "logps/rejected": -414.3836975097656,
746
- "loss": 0.1396,
747
- "rewards/accuracies": 0.824999988079071,
748
- "rewards/chosen": -1.0923160314559937,
749
- "rewards/margins": 0.7477865219116211,
750
- "rewards/rejected": -1.8401025533676147,
751
  "step": 520
752
  },
753
  {
754
  "epoch": 1.49,
755
- "learning_rate": 1.833380984759764e-07,
756
- "logits/chosen": 0.10293974727392197,
757
- "logits/rejected": 0.26486995816230774,
758
- "logps/chosen": -436.48907470703125,
759
- "logps/rejected": -422.7872009277344,
760
- "loss": 0.1299,
761
  "rewards/accuracies": 0.762499988079071,
762
- "rewards/chosen": -1.0291588306427002,
763
- "rewards/margins": 0.720792829990387,
764
- "rewards/rejected": -1.749951720237732,
765
  "step": 530
766
  },
767
  {
768
  "epoch": 1.52,
769
- "learning_rate": 1.6470462175846606e-07,
770
- "logits/chosen": 0.07386808097362518,
771
- "logits/rejected": 0.2429957389831543,
772
- "logps/chosen": -517.3760986328125,
773
- "logps/rejected": -460.2138671875,
774
- "loss": 0.1413,
775
- "rewards/accuracies": 0.8187500238418579,
776
- "rewards/chosen": -1.3018550872802734,
777
- "rewards/margins": 0.7603974342346191,
778
- "rewards/rejected": -2.0622525215148926,
779
  "step": 540
780
  },
781
  {
782
  "epoch": 1.55,
783
- "learning_rate": 1.468814306317092e-07,
784
- "logits/chosen": 0.11437557637691498,
785
- "logits/rejected": 0.2697022557258606,
786
- "logps/chosen": -451.03326416015625,
787
- "logps/rejected": -420.84100341796875,
788
- "loss": 0.1376,
789
- "rewards/accuracies": 0.793749988079071,
790
- "rewards/chosen": -1.2259981632232666,
791
- "rewards/margins": 0.9535650014877319,
792
- "rewards/rejected": -2.179563045501709,
793
  "step": 550
794
  },
795
  {
796
  "epoch": 1.57,
797
- "learning_rate": 1.299115971923958e-07,
798
- "logits/chosen": 0.07330118119716644,
799
- "logits/rejected": 0.24497468769550323,
800
- "logps/chosen": -482.2550354003906,
801
- "logps/rejected": -460.70367431640625,
802
- "loss": 0.1354,
803
- "rewards/accuracies": 0.762499988079071,
804
- "rewards/chosen": -1.2959102392196655,
805
- "rewards/margins": 0.8086190223693848,
806
- "rewards/rejected": -2.10452938079834,
807
  "step": 560
808
  },
809
  {
810
  "epoch": 1.6,
811
- "learning_rate": 1.1383613128559305e-07,
812
- "logits/chosen": 0.15279309451580048,
813
- "logits/rejected": 0.3158939480781555,
814
- "logps/chosen": -485.0721740722656,
815
- "logps/rejected": -450.25799560546875,
816
- "loss": 0.1225,
817
- "rewards/accuracies": 0.8062499761581421,
818
- "rewards/chosen": -1.0827850103378296,
819
- "rewards/margins": 0.839277446269989,
820
- "rewards/rejected": -1.9220625162124634,
821
  "step": 570
822
  },
823
  {
824
  "epoch": 1.63,
825
- "learning_rate": 9.869388139903495e-08,
826
- "logits/chosen": 0.09564773738384247,
827
- "logits/rejected": 0.150350421667099,
828
- "logps/chosen": -427.63616943359375,
829
- "logps/rejected": -459.97064208984375,
830
- "loss": 0.1297,
831
- "rewards/accuracies": 0.793749988079071,
832
- "rewards/chosen": -1.125893473625183,
833
- "rewards/margins": 0.7602401971817017,
834
- "rewards/rejected": -1.8861335515975952,
835
  "step": 580
836
  },
837
  {
838
  "epoch": 1.66,
839
- "learning_rate": 8.452144078061818e-08,
840
- "logits/chosen": 0.08417952060699463,
841
- "logits/rejected": 0.249791219830513,
842
- "logps/chosen": -476.9566345214844,
843
- "logps/rejected": -474.86944580078125,
844
- "loss": 0.1333,
845
- "rewards/accuracies": 0.8125,
846
- "rewards/chosen": -1.174066185951233,
847
- "rewards/margins": 0.9380697011947632,
848
- "rewards/rejected": -2.112135648727417,
849
  "step": 590
850
  },
851
  {
852
  "epoch": 1.69,
853
- "learning_rate": 7.135305900598321e-08,
854
- "logits/chosen": 0.11366138607263565,
855
- "logits/rejected": 0.24674567580223083,
856
- "logps/chosen": -484.5184631347656,
857
- "logps/rejected": -416.90863037109375,
858
- "loss": 0.1273,
859
- "rewards/accuracies": 0.75,
860
- "rewards/chosen": -1.1246851682662964,
861
- "rewards/margins": 0.781345546245575,
862
- "rewards/rejected": -1.9060306549072266,
863
  "step": 600
864
  },
865
  {
866
  "epoch": 1.71,
867
- "learning_rate": 5.9220559209888166e-08,
868
- "logits/chosen": 0.10064254701137543,
869
- "logits/rejected": 0.16818474233150482,
870
- "logps/chosen": -491.97662353515625,
871
- "logps/rejected": -483.4784240722656,
872
- "loss": 0.1357,
873
- "rewards/accuracies": 0.7562500238418579,
874
- "rewards/chosen": -1.2799218893051147,
875
- "rewards/margins": 0.9080696105957031,
876
- "rewards/rejected": -2.1879916191101074,
877
  "step": 610
878
  },
879
  {
880
  "epoch": 1.74,
881
- "learning_rate": 4.815326118139812e-08,
882
- "logits/chosen": 0.08698529005050659,
883
- "logits/rejected": 0.18599075078964233,
884
- "logps/chosen": -458.9814453125,
885
- "logps/rejected": -416.03424072265625,
886
- "loss": 0.1329,
887
- "rewards/accuracies": 0.8125,
888
- "rewards/chosen": -1.0884288549423218,
889
- "rewards/margins": 0.8256447911262512,
890
- "rewards/rejected": -1.9140735864639282,
891
  "step": 620
892
  },
893
  {
894
  "epoch": 1.77,
895
- "learning_rate": 3.81779105087407e-08,
896
- "logits/chosen": 0.028252029791474342,
897
- "logits/rejected": 0.24228385090827942,
898
- "logps/chosen": -447.22283935546875,
899
- "logps/rejected": -428.1376953125,
900
- "loss": 0.1349,
901
- "rewards/accuracies": 0.7749999761581421,
902
- "rewards/chosen": -1.2487438917160034,
903
- "rewards/margins": 0.7152423858642578,
904
- "rewards/rejected": -1.9639861583709717,
905
  "step": 630
906
  },
907
  {
908
  "epoch": 1.8,
909
- "learning_rate": 2.9318613945057637e-08,
910
- "logits/chosen": 0.03168308362364769,
911
- "logits/rejected": 0.14438685774803162,
912
- "logps/chosen": -530.6989135742188,
913
- "logps/rejected": -481.19317626953125,
914
- "loss": 0.1283,
915
- "rewards/accuracies": 0.8374999761581421,
916
- "rewards/chosen": -1.2953087091445923,
917
- "rewards/margins": 0.9182836413383484,
918
- "rewards/rejected": -2.213592529296875,
919
  "step": 640
920
  },
921
  {
922
  "epoch": 1.83,
923
- "learning_rate": 2.1596781151249523e-08,
924
- "logits/chosen": 0.15784808993339539,
925
- "logits/rejected": 0.28222331404685974,
926
- "logps/chosen": -461.800048828125,
927
- "logps/rejected": -412.47320556640625,
928
- "loss": 0.136,
929
- "rewards/accuracies": 0.824999988079071,
930
- "rewards/chosen": -1.0977789163589478,
931
- "rewards/margins": 0.8101053237915039,
932
- "rewards/rejected": -1.9078842401504517,
933
  "step": 650
934
  },
935
  {
936
  "epoch": 1.85,
937
- "learning_rate": 1.5031072956701695e-08,
938
- "logits/chosen": 0.04679325968027115,
939
- "logits/rejected": 0.1709231436252594,
940
- "logps/chosen": -501.66656494140625,
941
- "logps/rejected": -491.91925048828125,
942
- "loss": 0.1335,
943
- "rewards/accuracies": 0.824999988079071,
944
- "rewards/chosen": -1.1601709127426147,
945
- "rewards/margins": 1.024567723274231,
946
- "rewards/rejected": -2.1847386360168457,
947
  "step": 660
948
  },
949
  {
950
  "epoch": 1.88,
951
- "learning_rate": 9.637356262923723e-09,
952
- "logits/chosen": 0.12124764919281006,
953
- "logits/rejected": 0.2632105350494385,
954
- "logps/chosen": -449.41064453125,
955
- "logps/rejected": -440.3700256347656,
956
- "loss": 0.125,
957
- "rewards/accuracies": 0.7875000238418579,
958
- "rewards/chosen": -1.137796401977539,
959
- "rewards/margins": 0.832270622253418,
960
- "rewards/rejected": -1.970067024230957,
961
  "step": 670
962
  },
963
  {
964
  "epoch": 1.91,
965
- "learning_rate": 5.428665699084789e-09,
966
- "logits/chosen": -0.006133219692856073,
967
- "logits/rejected": 0.22374041378498077,
968
- "logps/chosen": -471.9480895996094,
969
- "logps/rejected": -433.7080078125,
970
- "loss": 0.1286,
971
- "rewards/accuracies": 0.8187500238418579,
972
- "rewards/chosen": -1.1709232330322266,
973
- "rewards/margins": 0.926952064037323,
974
- "rewards/rejected": -2.0978751182556152,
975
  "step": 680
976
  },
977
  {
978
  "epoch": 1.94,
979
- "learning_rate": 2.415172122110343e-09,
980
- "logits/chosen": -0.02816038951277733,
981
- "logits/rejected": 0.07123323529958725,
982
- "logps/chosen": -522.4472045898438,
983
- "logps/rejected": -474.63934326171875,
984
- "loss": 0.1306,
985
- "rewards/accuracies": 0.793749988079071,
986
- "rewards/chosen": -1.2836827039718628,
987
- "rewards/margins": 0.8452218174934387,
988
- "rewards/rejected": -2.1289045810699463,
989
  "step": 690
990
  },
991
  {
992
  "epoch": 1.97,
993
- "learning_rate": 6.041580374618327e-10,
994
- "logits/chosen": 0.06725062429904938,
995
- "logits/rejected": 0.20876435935497284,
996
- "logps/chosen": -496.45098876953125,
997
- "logps/rejected": -499.33856201171875,
998
- "loss": 0.1292,
999
- "rewards/accuracies": 0.75,
1000
- "rewards/chosen": -1.3299496173858643,
1001
- "rewards/margins": 0.8479114770889282,
1002
- "rewards/rejected": -2.177860975265503,
1003
  "step": 700
1004
  },
1005
  {
1006
  "epoch": 2.0,
1007
  "learning_rate": 0.0,
1008
- "logits/chosen": 0.11434557288885117,
1009
- "logits/rejected": 0.21849770843982697,
1010
- "logps/chosen": -448.5828552246094,
1011
- "logps/rejected": -464.2832946777344,
1012
- "loss": 0.1326,
1013
- "rewards/accuracies": 0.831250011920929,
1014
- "rewards/chosen": -1.2470009326934814,
1015
- "rewards/margins": 0.9275137782096863,
1016
- "rewards/rejected": -2.1745147705078125,
1017
  "step": 710
1018
  },
1019
  {
1020
  "epoch": 2.0,
1021
  "step": 710,
1022
  "total_flos": 0.0,
1023
- "train_loss": 0.25192113166124047,
1024
- "train_runtime": 8113.4503,
1025
- "train_samples_per_second": 11.228,
1026
- "train_steps_per_second": 0.088
1027
  }
1028
  ],
1029
  "logging_steps": 10,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
+ "learning_rate": 7.042253521126761e-08,
14
  "logits/chosen": -0.20257116854190826,
15
  "logits/rejected": -0.07568661123514175,
16
  "logps/chosen": -521.0269775390625,
 
24
  },
25
  {
26
  "epoch": 0.03,
27
+ "learning_rate": 7.042253521126762e-07,
28
+ "logits/chosen": -0.13681186735630035,
29
+ "logits/rejected": -0.0872005894780159,
30
+ "logps/chosen": -345.9254150390625,
31
+ "logps/rejected": -280.7078857421875,
32
+ "loss": 0.3327,
33
+ "rewards/accuracies": 0.4791666567325592,
34
+ "rewards/chosen": -0.000514834770001471,
35
+ "rewards/margins": -0.00025620695669203997,
36
+ "rewards/rejected": -0.00025862775510177016,
37
  "step": 10
38
  },
39
  {
40
  "epoch": 0.06,
41
+ "learning_rate": 1.4084507042253523e-06,
42
+ "logits/chosen": -0.09434644877910614,
43
+ "logits/rejected": -0.054363738745450974,
44
+ "logps/chosen": -332.7166442871094,
45
+ "logps/rejected": -246.1228790283203,
46
+ "loss": 0.3405,
47
+ "rewards/accuracies": 0.574999988079071,
48
+ "rewards/chosen": -0.008787451311945915,
49
+ "rewards/margins": 0.04191022366285324,
50
+ "rewards/rejected": -0.05069767311215401,
51
  "step": 20
52
  },
53
  {
54
  "epoch": 0.08,
55
+ "learning_rate": 2.1126760563380285e-06,
56
+ "logits/chosen": -0.06431654840707779,
57
+ "logits/rejected": -0.008967995643615723,
58
+ "logps/chosen": -398.9429931640625,
59
+ "logps/rejected": -280.65277099609375,
60
+ "loss": 0.3833,
61
+ "rewards/accuracies": 0.637499988079071,
62
+ "rewards/chosen": -0.11265812814235687,
63
+ "rewards/margins": 0.18575695157051086,
64
+ "rewards/rejected": -0.29841506481170654,
65
  "step": 30
66
  },
67
  {
68
  "epoch": 0.11,
69
+ "learning_rate": 2.8169014084507046e-06,
70
+ "logits/chosen": -0.10810466855764389,
71
+ "logits/rejected": -0.04957962408661842,
72
+ "logps/chosen": -347.4810485839844,
73
+ "logps/rejected": -286.8487854003906,
74
+ "loss": 0.4038,
75
+ "rewards/accuracies": 0.5687500238418579,
76
+ "rewards/chosen": -0.12302961200475693,
77
+ "rewards/margins": 0.13980238139629364,
78
+ "rewards/rejected": -0.26283198595046997,
79
  "step": 40
80
  },
81
  {
82
  "epoch": 0.14,
83
+ "learning_rate": 3.5211267605633804e-06,
84
+ "logits/chosen": -0.09582972526550293,
85
+ "logits/rejected": -0.03566538169980049,
86
+ "logps/chosen": -331.12353515625,
87
+ "logps/rejected": -272.82281494140625,
88
+ "loss": 0.3735,
89
+ "rewards/accuracies": 0.5625,
90
+ "rewards/chosen": 0.02591649256646633,
91
+ "rewards/margins": 0.12833817303180695,
92
+ "rewards/rejected": -0.10242166370153427,
93
  "step": 50
94
  },
95
  {
96
  "epoch": 0.17,
97
+ "learning_rate": 4.225352112676057e-06,
98
+ "logits/chosen": -0.28920671343803406,
99
+ "logits/rejected": -0.21953177452087402,
100
+ "logps/chosen": -414.81640625,
101
+ "logps/rejected": -273.1141052246094,
102
+ "loss": 0.3825,
103
+ "rewards/accuracies": 0.7250000238418579,
104
+ "rewards/chosen": 0.07506568729877472,
105
+ "rewards/margins": 0.3529297113418579,
106
+ "rewards/rejected": -0.2778640389442444,
107
  "step": 60
108
  },
109
  {
110
  "epoch": 0.2,
111
+ "learning_rate": 4.929577464788733e-06,
112
+ "logits/chosen": -0.1399366408586502,
113
+ "logits/rejected": -0.017617041245102882,
114
+ "logps/chosen": -428.28375244140625,
115
+ "logps/rejected": -310.7579650878906,
116
+ "loss": 0.3466,
117
+ "rewards/accuracies": 0.625,
118
+ "rewards/chosen": -0.21147418022155762,
119
+ "rewards/margins": 0.28768691420555115,
120
+ "rewards/rejected": -0.49916109442710876,
121
  "step": 70
122
  },
123
  {
124
  "epoch": 0.22,
125
+ "learning_rate": 4.997553066299934e-06,
126
+ "logits/chosen": 0.08324304968118668,
127
+ "logits/rejected": 0.10611347109079361,
128
+ "logps/chosen": -343.00885009765625,
129
+ "logps/rejected": -283.32366943359375,
130
+ "loss": 0.3241,
131
+ "rewards/accuracies": 0.637499988079071,
132
+ "rewards/chosen": -0.5568825006484985,
133
+ "rewards/margins": 0.2385629415512085,
134
+ "rewards/rejected": -0.795445442199707,
135
  "step": 80
136
  },
137
  {
138
  "epoch": 0.25,
139
+ "learning_rate": 4.989100679490323e-06,
140
+ "logits/chosen": -0.06007497385144234,
141
+ "logits/rejected": 0.01979098841547966,
142
+ "logps/chosen": -443.93310546875,
143
+ "logps/rejected": -310.85113525390625,
144
+ "loss": 0.3178,
145
+ "rewards/accuracies": 0.71875,
146
+ "rewards/chosen": -0.4307987093925476,
147
+ "rewards/margins": 0.43101730942726135,
148
+ "rewards/rejected": -0.8618160486221313,
149
  "step": 90
150
  },
151
  {
152
  "epoch": 0.28,
153
+ "learning_rate": 4.974633051954447e-06,
154
+ "logits/chosen": 0.1131998747587204,
155
+ "logits/rejected": 0.13339917361736298,
156
+ "logps/chosen": -433.76629638671875,
157
+ "logps/rejected": -361.7824401855469,
158
+ "loss": 0.2628,
159
+ "rewards/accuracies": 0.5874999761581421,
160
+ "rewards/chosen": -0.6594427824020386,
161
+ "rewards/margins": 0.18142201006412506,
162
+ "rewards/rejected": -0.8408648371696472,
163
  "step": 100
164
  },
165
  {
166
  "epoch": 0.31,
167
+ "learning_rate": 4.954185146626145e-06,
168
+ "logits/chosen": -0.03576885908842087,
169
+ "logits/rejected": 0.0655001848936081,
170
+ "logps/chosen": -414.16192626953125,
171
+ "logps/rejected": -375.30816650390625,
172
+ "loss": 0.2445,
173
+ "rewards/accuracies": 0.5625,
174
+ "rewards/chosen": -0.8624384999275208,
175
+ "rewards/margins": 0.09759419411420822,
176
+ "rewards/rejected": -0.9600327610969543,
177
  "step": 110
178
  },
179
  {
180
  "epoch": 0.34,
181
+ "learning_rate": 4.927806378570827e-06,
182
+ "logits/chosen": -0.13169847428798676,
183
+ "logits/rejected": -0.09171108901500702,
184
+ "logps/chosen": -411.78973388671875,
185
+ "logps/rejected": -328.10693359375,
186
+ "loss": 0.2521,
187
+ "rewards/accuracies": 0.637499988079071,
188
+ "rewards/chosen": -0.7940471768379211,
189
+ "rewards/margins": 0.3129858374595642,
190
+ "rewards/rejected": -1.1070330142974854,
191
  "step": 120
192
  },
193
  {
194
  "epoch": 0.37,
195
+ "learning_rate": 4.895560495567452e-06,
196
+ "logits/chosen": 0.02038062922656536,
197
+ "logits/rejected": 0.09609419107437134,
198
+ "logps/chosen": -509.9686584472656,
199
+ "logps/rejected": -392.2916564941406,
200
+ "loss": 0.2267,
201
+ "rewards/accuracies": 0.731249988079071,
202
+ "rewards/chosen": -1.2382087707519531,
203
+ "rewards/margins": 0.39059334993362427,
204
+ "rewards/rejected": -1.6288020610809326,
205
  "step": 130
206
  },
207
  {
208
  "epoch": 0.39,
209
+ "learning_rate": 4.857525424053584e-06,
210
+ "logits/chosen": 0.11588418483734131,
211
+ "logits/rejected": 0.14068534970283508,
212
+ "logps/chosen": -441.15704345703125,
213
+ "logps/rejected": -364.00799560546875,
214
+ "loss": 0.2176,
215
+ "rewards/accuracies": 0.5375000238418579,
216
+ "rewards/chosen": -1.0443618297576904,
217
+ "rewards/margins": 0.1029224619269371,
218
+ "rewards/rejected": -1.1472842693328857,
219
  "step": 140
220
  },
221
  {
222
  "epoch": 0.42,
223
+ "learning_rate": 4.813793080805866e-06,
224
+ "logits/chosen": 0.1739293485879898,
225
+ "logits/rejected": 0.2377185821533203,
226
+ "logps/chosen": -406.53631591796875,
227
+ "logps/rejected": -369.4870910644531,
228
+ "loss": 0.217,
229
+ "rewards/accuracies": 0.5625,
230
+ "rewards/chosen": -1.0144250392913818,
231
+ "rewards/margins": 0.20032191276550293,
232
+ "rewards/rejected": -1.2147470712661743,
233
  "step": 150
234
  },
235
  {
236
  "epoch": 0.45,
237
+ "learning_rate": 4.764469150810978e-06,
238
+ "logits/chosen": 0.1326332688331604,
239
+ "logits/rejected": 0.22836916148662567,
240
+ "logps/chosen": -519.8853759765625,
241
+ "logps/rejected": -416.75482177734375,
242
+ "loss": 0.1838,
243
+ "rewards/accuracies": 0.59375,
244
+ "rewards/chosen": -1.3405306339263916,
245
+ "rewards/margins": 0.2509741485118866,
246
+ "rewards/rejected": -1.5915048122406006,
247
  "step": 160
248
  },
249
  {
250
  "epoch": 0.48,
251
+ "learning_rate": 4.709672831863903e-06,
252
+ "logits/chosen": 0.29795151948928833,
253
+ "logits/rejected": 0.3732362687587738,
254
+ "logps/chosen": -471.41204833984375,
255
+ "logps/rejected": -430.14678955078125,
256
+ "loss": 0.173,
257
+ "rewards/accuracies": 0.581250011920929,
258
+ "rewards/chosen": -1.3964329957962036,
259
+ "rewards/margins": 0.2646811604499817,
260
+ "rewards/rejected": -1.6611143350601196,
261
  "step": 170
262
  },
263
  {
264
  "epoch": 0.51,
265
+ "learning_rate": 4.649536546510702e-06,
266
+ "logits/chosen": 0.10790077596902847,
267
+ "logits/rejected": 0.2056044638156891,
268
+ "logps/chosen": -483.7994689941406,
269
+ "logps/rejected": -419.7510681152344,
270
+ "loss": 0.1935,
271
+ "rewards/accuracies": 0.612500011920929,
272
+ "rewards/chosen": -1.4253547191619873,
273
+ "rewards/margins": 0.25134578347206116,
274
+ "rewards/rejected": -1.6767005920410156,
275
  "step": 180
276
  },
277
  {
278
  "epoch": 0.53,
279
+ "learning_rate": 4.584205622031932e-06,
280
+ "logits/chosen": 0.11415908485651016,
281
+ "logits/rejected": 0.17238232493400574,
282
+ "logps/chosen": -478.956298828125,
283
+ "logps/rejected": -395.60382080078125,
284
+ "loss": 0.1794,
285
+ "rewards/accuracies": 0.668749988079071,
286
+ "rewards/chosen": -1.2251521348953247,
287
+ "rewards/margins": 0.4370814859867096,
288
+ "rewards/rejected": -1.6622337102890015,
289
  "step": 190
290
  },
291
  {
292
  "epoch": 0.56,
293
+ "learning_rate": 4.513837939240066e-06,
294
+ "logits/chosen": -0.009004750289022923,
295
+ "logits/rejected": 0.11390223354101181,
296
+ "logps/chosen": -502.33441162109375,
297
+ "logps/rejected": -436.2867736816406,
298
+ "loss": 0.1964,
299
+ "rewards/accuracies": 0.581250011920929,
300
+ "rewards/chosen": -1.7229286432266235,
301
+ "rewards/margins": 0.19690918922424316,
302
+ "rewards/rejected": -1.9198377132415771,
303
  "step": 200
304
  },
305
  {
306
  "epoch": 0.59,
307
+ "learning_rate": 4.438603550939651e-06,
308
+ "logits/chosen": 0.11682312190532684,
309
+ "logits/rejected": 0.21160845458507538,
310
+ "logps/chosen": -531.32421875,
311
+ "logps/rejected": -420.9443359375,
312
+ "loss": 0.2037,
313
+ "rewards/accuracies": 0.5874999761581421,
314
+ "rewards/chosen": -1.2861896753311157,
315
+ "rewards/margins": 0.2408260852098465,
316
+ "rewards/rejected": -1.5270155668258667,
317
  "step": 210
318
  },
319
  {
320
  "epoch": 0.62,
321
+ "learning_rate": 4.358684270972226e-06,
322
+ "logits/chosen": -0.03151511400938034,
323
+ "logits/rejected": 0.05730857327580452,
324
+ "logps/chosen": -499.53924560546875,
325
+ "logps/rejected": -399.32147216796875,
326
+ "loss": 0.222,
327
+ "rewards/accuracies": 0.6312500238418579,
328
+ "rewards/chosen": -1.0882928371429443,
329
+ "rewards/margins": 0.2548028826713562,
330
+ "rewards/rejected": -1.3430957794189453,
331
  "step": 220
332
  },
333
  {
334
  "epoch": 0.65,
335
+ "learning_rate": 4.2742732348391555e-06,
336
+ "logits/chosen": 0.026254689320921898,
337
+ "logits/rejected": 0.15401390194892883,
338
+ "logps/chosen": -465.4871520996094,
339
+ "logps/rejected": -386.40081787109375,
340
+ "loss": 0.2377,
341
+ "rewards/accuracies": 0.6187499761581421,
342
+ "rewards/chosen": -1.151163935661316,
343
+ "rewards/margins": 0.27608826756477356,
344
+ "rewards/rejected": -1.427252173423767,
345
  "step": 230
346
  },
347
  {
348
  "epoch": 0.67,
349
+ "learning_rate": 4.18557443296416e-06,
350
+ "logits/chosen": -0.057366080582141876,
351
+ "logits/rejected": 0.04457206651568413,
352
+ "logps/chosen": -472.771728515625,
353
+ "logps/rejected": -376.9158020019531,
354
+ "loss": 0.1975,
355
+ "rewards/accuracies": 0.574999988079071,
356
+ "rewards/chosen": -1.139541745185852,
357
+ "rewards/margins": 0.23796971142292023,
358
+ "rewards/rejected": -1.3775115013122559,
359
  "step": 240
360
  },
361
  {
362
  "epoch": 0.7,
363
+ "learning_rate": 4.092802217723501e-06,
364
+ "logits/chosen": -0.03958164528012276,
365
+ "logits/rejected": 0.07425127923488617,
366
+ "logps/chosen": -530.5653076171875,
367
+ "logps/rejected": -422.4510192871094,
368
+ "loss": 0.1985,
369
+ "rewards/accuracies": 0.6625000238418579,
370
+ "rewards/chosen": -1.486722707748413,
371
+ "rewards/margins": 0.33031830191612244,
372
+ "rewards/rejected": -1.817041039466858,
373
  "step": 250
374
  },
375
  {
376
  "epoch": 0.73,
377
+ "learning_rate": 3.9961807854351446e-06,
378
+ "logits/chosen": 0.07857771217823029,
379
+ "logits/rejected": 0.13072842359542847,
380
+ "logps/chosen": -527.210693359375,
381
+ "logps/rejected": -451.3309020996094,
382
+ "loss": 0.2017,
383
+ "rewards/accuracies": 0.625,
384
+ "rewards/chosen": -1.6589043140411377,
385
+ "rewards/margins": 0.3979204297065735,
386
+ "rewards/rejected": -2.0568246841430664,
387
  "step": 260
388
  },
389
  {
390
  "epoch": 0.76,
391
+ "learning_rate": 3.895943634558721e-06,
392
+ "logits/chosen": 0.26124829053878784,
393
+ "logits/rejected": 0.3160572052001953,
394
+ "logps/chosen": -498.911376953125,
395
+ "logps/rejected": -438.7411193847656,
396
+ "loss": 0.1763,
397
+ "rewards/accuracies": 0.643750011920929,
398
+ "rewards/chosen": -1.6009438037872314,
399
+ "rewards/margins": 0.3443499505519867,
400
+ "rewards/rejected": -1.9452937841415405,
401
  "step": 270
402
  },
403
  {
404
  "epoch": 0.79,
405
+ "learning_rate": 3.792333001415648e-06,
406
+ "logits/chosen": 0.2600861191749573,
407
+ "logits/rejected": 0.38268446922302246,
408
+ "logps/chosen": -588.9376220703125,
409
+ "logps/rejected": -477.60101318359375,
410
+ "loss": 0.1838,
411
+ "rewards/accuracies": 0.6499999761581421,
412
+ "rewards/chosen": -1.961019515991211,
413
+ "rewards/margins": 0.3914053440093994,
414
+ "rewards/rejected": -2.3524250984191895,
415
  "step": 280
416
  },
417
  {
418
  "epoch": 0.81,
419
+ "learning_rate": 3.6855992747930456e-06,
420
+ "logits/chosen": 0.34141284227371216,
421
+ "logits/rejected": 0.470449835062027,
422
+ "logps/chosen": -519.5052490234375,
423
+ "logps/rejected": -442.36871337890625,
424
+ "loss": 0.1849,
425
+ "rewards/accuracies": 0.550000011920929,
426
+ "rewards/chosen": -1.677393913269043,
427
+ "rewards/margins": 0.16863858699798584,
428
+ "rewards/rejected": -1.8460325002670288,
429
  "step": 290
430
  },
431
  {
432
  "epoch": 0.84,
433
+ "learning_rate": 3.5760003908461432e-06,
434
+ "logits/chosen": 0.2685919404029846,
435
+ "logits/rejected": 0.3938666582107544,
436
+ "logps/chosen": -494.66009521484375,
437
+ "logps/rejected": -392.2904052734375,
438
+ "loss": 0.2002,
439
+ "rewards/accuracies": 0.6000000238418579,
440
+ "rewards/chosen": -1.1747759580612183,
441
+ "rewards/margins": 0.2742941677570343,
442
+ "rewards/rejected": -1.4490702152252197,
443
  "step": 300
444
  },
445
  {
446
  "epoch": 0.87,
447
+ "learning_rate": 3.4638012097614735e-06,
448
+ "logits/chosen": 0.2535479664802551,
449
+ "logits/rejected": 0.37561532855033875,
450
+ "logps/chosen": -460.380859375,
451
+ "logps/rejected": -389.4805603027344,
452
+ "loss": 0.2132,
453
+ "rewards/accuracies": 0.574999988079071,
454
+ "rewards/chosen": -1.1959959268569946,
455
+ "rewards/margins": 0.27664345502853394,
456
+ "rewards/rejected": -1.4726393222808838,
457
  "step": 310
458
  },
459
  {
460
  "epoch": 0.9,
461
+ "learning_rate": 3.3492728756872324e-06,
462
+ "logits/chosen": 0.29194751381874084,
463
+ "logits/rejected": 0.34319376945495605,
464
+ "logps/chosen": -467.84783935546875,
465
+ "logps/rejected": -414.1121520996094,
466
+ "loss": 0.2141,
467
+ "rewards/accuracies": 0.581250011920929,
468
+ "rewards/chosen": -1.3512526750564575,
469
+ "rewards/margins": 0.21693870425224304,
470
+ "rewards/rejected": -1.5681912899017334,
471
  "step": 320
472
  },
473
  {
474
  "epoch": 0.93,
475
+ "learning_rate": 3.2326921614776125e-06,
476
+ "logits/chosen": 0.12696528434753418,
477
+ "logits/rejected": 0.2001853734254837,
478
+ "logps/chosen": -572.5255126953125,
479
+ "logps/rejected": -473.86932373046875,
480
+ "loss": 0.1812,
481
+ "rewards/accuracies": 0.612500011920929,
482
+ "rewards/chosen": -1.5725057125091553,
483
+ "rewards/margins": 0.37595897912979126,
484
+ "rewards/rejected": -1.9484646320343018,
485
  "step": 330
486
  },
487
  {
488
  "epoch": 0.96,
489
+ "learning_rate": 3.114340799834624e-06,
490
+ "logits/chosen": 0.24421611428260803,
491
+ "logits/rejected": 0.309607595205307,
492
+ "logps/chosen": -510.90948486328125,
493
+ "logps/rejected": -431.14642333984375,
494
+ "loss": 0.1856,
495
+ "rewards/accuracies": 0.59375,
496
+ "rewards/chosen": -1.6012935638427734,
497
+ "rewards/margins": 0.30774253606796265,
498
+ "rewards/rejected": -1.9090360403060913,
499
  "step": 340
500
  },
501
  {
502
  "epoch": 0.98,
503
+ "learning_rate": 2.9945048024637935e-06,
504
+ "logits/chosen": 0.3243614733219147,
505
+ "logits/rejected": 0.42463597655296326,
506
+ "logps/chosen": -533.7662353515625,
507
+ "logps/rejected": -459.022705078125,
508
+ "loss": 0.1946,
509
+ "rewards/accuracies": 0.6812499761581421,
510
+ "rewards/chosen": -1.722198247909546,
511
+ "rewards/margins": 0.3744170069694519,
512
+ "rewards/rejected": -2.0966150760650635,
513
  "step": 350
514
  },
515
  {
516
  "epoch": 1.01,
517
+ "learning_rate": 2.8734737688890752e-06,
518
+ "logits/chosen": 0.3470906615257263,
519
+ "logits/rejected": 0.42189717292785645,
520
+ "logps/chosen": -492.981689453125,
521
+ "logps/rejected": -519.1780395507812,
522
+ "loss": 0.1261,
523
+ "rewards/accuracies": 0.6625000238418579,
524
+ "rewards/chosen": -1.7776886224746704,
525
+ "rewards/margins": 0.6867099404335022,
526
+ "rewards/rejected": -2.4643986225128174,
527
  "step": 360
528
  },
529
  {
530
  "epoch": 1.04,
531
+ "learning_rate": 2.7515401865973333e-06,
532
+ "logits/chosen": 0.29653865098953247,
533
+ "logits/rejected": 0.31189900636672974,
534
+ "logps/chosen": -541.0857543945312,
535
+ "logps/rejected": -558.9527587890625,
536
+ "loss": 0.0325,
537
+ "rewards/accuracies": 0.8187500238418579,
538
+ "rewards/chosen": -1.839362382888794,
539
+ "rewards/margins": 1.2354429960250854,
540
+ "rewards/rejected": -3.074805498123169,
541
  "step": 370
542
  },
543
  {
544
  "epoch": 1.07,
545
+ "learning_rate": 2.628998724203683e-06,
546
+ "logits/chosen": 0.24393782019615173,
547
+ "logits/rejected": 0.3541429042816162,
548
+ "logps/chosen": -601.4793090820312,
549
+ "logps/rejected": -695.1011352539062,
550
+ "loss": 0.0162,
551
+ "rewards/accuracies": 0.793749988079071,
552
+ "rewards/chosen": -2.808850049972534,
553
+ "rewards/margins": 1.5772719383239746,
554
+ "rewards/rejected": -4.386122226715088,
555
  "step": 380
556
  },
557
  {
558
  "epoch": 1.1,
559
+ "learning_rate": 2.5061455193458326e-06,
560
+ "logits/chosen": 0.2903648018836975,
561
+ "logits/rejected": 0.4108067452907562,
562
+ "logps/chosen": -613.734619140625,
563
+ "logps/rejected": -712.7540893554688,
564
+ "loss": 0.0109,
565
+ "rewards/accuracies": 0.7749999761581421,
566
+ "rewards/chosen": -3.2557308673858643,
567
+ "rewards/margins": 1.669713020324707,
568
+ "rewards/rejected": -4.92544412612915,
569
  "step": 390
570
  },
571
  {
572
  "epoch": 1.12,
573
+ "learning_rate": 2.3832774630283534e-06,
574
+ "logits/chosen": 0.3819536864757538,
575
+ "logits/rejected": 0.4252726137638092,
576
+ "logps/chosen": -615.9334716796875,
577
+ "logps/rejected": -717.3001098632812,
578
+ "loss": 0.0112,
579
+ "rewards/accuracies": 0.8125,
580
+ "rewards/chosen": -2.7294070720672607,
581
+ "rewards/margins": 1.8411362171173096,
582
+ "rewards/rejected": -4.57054328918457,
583
  "step": 400
584
  },
585
  {
586
  "epoch": 1.15,
587
+ "learning_rate": 2.2606914821463317e-06,
588
+ "logits/chosen": 0.5641288161277771,
589
+ "logits/rejected": 0.5737628936767578,
590
+ "logps/chosen": -588.9856567382812,
591
+ "logps/rejected": -711.607421875,
592
+ "loss": 0.0108,
593
+ "rewards/accuracies": 0.793749988079071,
594
+ "rewards/chosen": -2.6725032329559326,
595
+ "rewards/margins": 2.191044330596924,
596
+ "rewards/rejected": -4.8635478019714355,
597
  "step": 410
598
  },
599
  {
600
  "epoch": 1.18,
601
+ "learning_rate": 2.138683821922287e-06,
602
+ "logits/chosen": 0.44424986839294434,
603
+ "logits/rejected": 0.5702834129333496,
604
+ "logps/chosen": -649.6632080078125,
605
+ "logps/rejected": -763.0523681640625,
606
+ "loss": 0.0084,
607
+ "rewards/accuracies": 0.8187500238418579,
608
+ "rewards/chosen": -3.1381773948669434,
609
+ "rewards/margins": 1.943543791770935,
610
+ "rewards/rejected": -5.081721305847168,
611
  "step": 420
612
  },
613
  {
614
  "epoch": 1.21,
615
+ "learning_rate": 2.0175493299904456e-06,
616
+ "logits/chosen": 0.5278526544570923,
617
+ "logits/rejected": 0.5699671506881714,
618
+ "logps/chosen": -760.1556396484375,
619
+ "logps/rejected": -837.56298828125,
620
+ "loss": 0.0057,
621
+ "rewards/accuracies": 0.8500000238418579,
622
+ "rewards/chosen": -3.7177700996398926,
623
+ "rewards/margins": 2.2982795238494873,
624
+ "rewards/rejected": -6.016049385070801,
625
  "step": 430
626
  },
627
  {
628
  "epoch": 1.24,
629
+ "learning_rate": 1.8975807438584643e-06,
630
+ "logits/chosen": 0.4808652400970459,
631
+ "logits/rejected": 0.5634757280349731,
632
+ "logps/chosen": -693.7796020507812,
633
+ "logps/rejected": -791.9578247070312,
634
+ "loss": 0.0054,
635
+ "rewards/accuracies": 0.7562500238418579,
636
+ "rewards/chosen": -3.8103713989257812,
637
+ "rewards/margins": 1.8808832168579102,
638
+ "rewards/rejected": -5.69125509262085,
639
  "step": 440
640
  },
641
  {
642
  "epoch": 1.26,
643
+ "learning_rate": 1.7790679834685615e-06,
644
+ "logits/chosen": 0.48697033524513245,
645
+ "logits/rejected": 0.6213686466217041,
646
+ "logps/chosen": -726.90185546875,
647
+ "logps/rejected": -848.7301635742188,
648
+ "loss": 0.0053,
649
+ "rewards/accuracies": 0.8125,
650
+ "rewards/chosen": -3.577058792114258,
651
+ "rewards/margins": 2.4176392555236816,
652
+ "rewards/rejected": -5.9946980476379395,
653
  "step": 450
654
  },
655
  {
656
  "epoch": 1.29,
657
+ "learning_rate": 1.662297450567663e-06,
658
+ "logits/chosen": 0.6364110708236694,
659
+ "logits/rejected": 0.7476580739021301,
660
+ "logps/chosen": -679.8997802734375,
661
+ "logps/rejected": -860.2080078125,
662
+ "loss": 0.0055,
663
+ "rewards/accuracies": 0.8187500238418579,
664
+ "rewards/chosen": -3.570373058319092,
665
+ "rewards/margins": 2.280672073364258,
666
+ "rewards/rejected": -5.85104513168335,
667
  "step": 460
668
  },
669
  {
670
  "epoch": 1.32,
671
+ "learning_rate": 1.5475513365797317e-06,
672
+ "logits/chosen": 0.47412365674972534,
673
+ "logits/rejected": 0.47869664430618286,
674
+ "logps/chosen": -706.4371337890625,
675
+ "logps/rejected": -842.859375,
676
+ "loss": 0.0054,
677
+ "rewards/accuracies": 0.8062499761581421,
678
+ "rewards/chosen": -3.590733766555786,
679
+ "rewards/margins": 2.3144545555114746,
680
+ "rewards/rejected": -5.90518856048584,
681
  "step": 470
682
  },
683
  {
684
  "epoch": 1.35,
685
+ "learning_rate": 1.435106940652901e-06,
686
+ "logits/chosen": 0.5485504865646362,
687
+ "logits/rejected": 0.681022584438324,
688
+ "logps/chosen": -682.8555908203125,
689
+ "logps/rejected": -810.1474609375,
690
+ "loss": 0.0055,
691
  "rewards/accuracies": 0.800000011920929,
692
+ "rewards/chosen": -3.3580360412597656,
693
+ "rewards/margins": 2.372495174407959,
694
+ "rewards/rejected": -5.730531215667725,
695
  "step": 480
696
  },
697
  {
698
  "epoch": 1.38,
699
+ "learning_rate": 1.3252359995294376e-06,
700
+ "logits/chosen": 0.5862436890602112,
701
+ "logits/rejected": 0.6748989820480347,
702
+ "logps/chosen": -704.3018188476562,
703
+ "logps/rejected": -828.2868041992188,
704
+ "loss": 0.0047,
705
+ "rewards/accuracies": 0.8500000238418579,
706
+ "rewards/chosen": -3.270312786102295,
707
+ "rewards/margins": 2.447282075881958,
708
+ "rewards/rejected": -5.717595100402832,
709
  "step": 490
710
  },
711
  {
712
  "epoch": 1.4,
713
+ "learning_rate": 1.2182040308579942e-06,
714
+ "logits/chosen": 0.5599916577339172,
715
+ "logits/rejected": 0.6381617784500122,
716
+ "logps/chosen": -757.0291748046875,
717
+ "logps/rejected": -892.974609375,
718
+ "loss": 0.0041,
719
+ "rewards/accuracies": 0.8062499761581421,
720
+ "rewards/chosen": -3.7352547645568848,
721
+ "rewards/margins": 2.5722806453704834,
722
+ "rewards/rejected": -6.307535171508789,
723
  "step": 500
724
  },
725
  {
726
  "epoch": 1.43,
727
+ "learning_rate": 1.114269691535125e-06,
728
+ "logits/chosen": 0.6536546945571899,
729
+ "logits/rejected": 0.7505389451980591,
730
+ "logps/chosen": -729.1384887695312,
731
+ "logps/rejected": -864.0009765625,
732
+ "loss": 0.004,
733
+ "rewards/accuracies": 0.78125,
734
+ "rewards/chosen": -3.6905455589294434,
735
+ "rewards/margins": 2.2943015098571777,
736
+ "rewards/rejected": -5.984846591949463,
737
  "step": 510
738
  },
739
  {
740
  "epoch": 1.46,
741
+ "learning_rate": 1.013684152626709e-06,
742
+ "logits/chosen": 0.5734939575195312,
743
+ "logits/rejected": 0.6782180070877075,
744
+ "logps/chosen": -710.2030029296875,
745
+ "logps/rejected": -866.6712646484375,
746
+ "loss": 0.0042,
747
+ "rewards/accuracies": 0.8374999761581421,
748
+ "rewards/chosen": -3.7363319396972656,
749
+ "rewards/margins": 2.6266465187072754,
750
+ "rewards/rejected": -6.362977981567383,
751
  "step": 520
752
  },
753
  {
754
  "epoch": 1.49,
755
+ "learning_rate": 9.166904923798822e-07,
756
+ "logits/chosen": 0.6636993885040283,
757
+ "logits/rejected": 0.7554783225059509,
758
+ "logps/chosen": -705.7659912109375,
759
+ "logps/rejected": -842.1114501953125,
760
+ "loss": 0.0035,
761
  "rewards/accuracies": 0.762499988079071,
762
+ "rewards/chosen": -3.721928358078003,
763
+ "rewards/margins": 2.2212657928466797,
764
+ "rewards/rejected": -5.943194389343262,
765
  "step": 530
766
  },
767
  {
768
  "epoch": 1.52,
769
+ "learning_rate": 8.235231087923304e-07,
770
+ "logits/chosen": 0.5900656580924988,
771
+ "logits/rejected": 0.7390304803848267,
772
+ "logps/chosen": -762.967041015625,
773
+ "logps/rejected": -907.9431762695312,
774
+ "loss": 0.0039,
775
+ "rewards/accuracies": 0.862500011920929,
776
+ "rewards/chosen": -3.7577643394470215,
777
+ "rewards/margins": 2.781780242919922,
778
+ "rewards/rejected": -6.539544582366943,
779
  "step": 540
780
  },
781
  {
782
  "epoch": 1.55,
783
+ "learning_rate": 7.344071531585462e-07,
784
+ "logits/chosen": 0.6735460162162781,
785
+ "logits/rejected": 0.7701358795166016,
786
+ "logps/chosen": -721.6229248046875,
787
+ "logps/rejected": -840.623046875,
788
+ "loss": 0.0035,
789
+ "rewards/accuracies": 0.800000011920929,
790
+ "rewards/chosen": -3.9318947792053223,
791
+ "rewards/margins": 2.4454891681671143,
792
+ "rewards/rejected": -6.377383232116699,
793
  "step": 550
794
  },
795
  {
796
  "epoch": 1.57,
797
+ "learning_rate": 6.495579859619791e-07,
798
+ "logits/chosen": 0.6034525632858276,
799
+ "logits/rejected": 0.7575315237045288,
800
+ "logps/chosen": -774.0142211914062,
801
+ "logps/rejected": -933.1998291015625,
802
+ "loss": 0.0038,
803
+ "rewards/accuracies": 0.8062499761581421,
804
+ "rewards/chosen": -4.213502407073975,
805
+ "rewards/margins": 2.61598801612854,
806
+ "rewards/rejected": -6.829490661621094,
807
  "step": 560
808
  },
809
  {
810
  "epoch": 1.6,
811
+ "learning_rate": 5.691806564279653e-07,
812
+ "logits/chosen": 0.6900167465209961,
813
+ "logits/rejected": 0.8468635678291321,
814
+ "logps/chosen": -746.2769775390625,
815
+ "logps/rejected": -889.0632934570312,
816
+ "loss": 0.0034,
817
+ "rewards/accuracies": 0.831250011920929,
818
+ "rewards/chosen": -3.694833278656006,
819
+ "rewards/margins": 2.6152822971343994,
820
+ "rewards/rejected": -6.310114860534668,
821
  "step": 570
822
  },
823
  {
824
  "epoch": 1.63,
825
+ "learning_rate": 4.934694069951748e-07,
826
+ "logits/chosen": 0.6488009691238403,
827
+ "logits/rejected": 0.6377809643745422,
828
+ "logps/chosen": -676.0171508789062,
829
+ "logps/rejected": -875.9207763671875,
830
+ "loss": 0.0033,
831
+ "rewards/accuracies": 0.8125,
832
+ "rewards/chosen": -3.6097030639648438,
833
+ "rewards/margins": 2.43593168258667,
834
+ "rewards/rejected": -6.04563570022583,
835
  "step": 580
836
  },
837
  {
838
  "epoch": 1.66,
839
+ "learning_rate": 4.22607203903091e-07,
840
+ "logits/chosen": 0.6412608623504639,
841
+ "logits/rejected": 0.7388615608215332,
842
+ "logps/chosen": -764.0328369140625,
843
+ "logps/rejected": -968.5056762695312,
844
+ "loss": 0.0035,
845
+ "rewards/accuracies": 0.856249988079071,
846
+ "rewards/chosen": -4.044827461242676,
847
+ "rewards/margins": 3.0036702156066895,
848
+ "rewards/rejected": -7.048497676849365,
849
  "step": 590
850
  },
851
  {
852
  "epoch": 1.69,
853
+ "learning_rate": 3.5676529502991606e-07,
854
+ "logits/chosen": 0.6487208604812622,
855
+ "logits/rejected": 0.7878293395042419,
856
+ "logps/chosen": -745.95556640625,
857
+ "logps/rejected": -838.0792236328125,
858
+ "loss": 0.0039,
859
+ "rewards/accuracies": 0.8187500238418579,
860
+ "rewards/chosen": -3.73905611038208,
861
+ "rewards/margins": 2.3786799907684326,
862
+ "rewards/rejected": -6.117735862731934,
863
  "step": 600
864
  },
865
  {
866
  "epoch": 1.71,
867
+ "learning_rate": 2.961027960494409e-07,
868
+ "logits/chosen": 0.6219756603240967,
869
+ "logits/rejected": 0.6644929647445679,
870
+ "logps/chosen": -760.4959106445312,
871
+ "logps/rejected": -936.97265625,
872
+ "loss": 0.0036,
873
+ "rewards/accuracies": 0.824999988079071,
874
+ "rewards/chosen": -3.9651145935058594,
875
+ "rewards/margins": 2.757819652557373,
876
+ "rewards/rejected": -6.722934722900391,
877
  "step": 610
878
  },
879
  {
880
  "epoch": 1.74,
881
+ "learning_rate": 2.4076630590699065e-07,
882
+ "logits/chosen": 0.6320921182632446,
883
+ "logits/rejected": 0.7059046626091003,
884
+ "logps/chosen": -700.6550903320312,
885
+ "logps/rejected": -826.9192504882812,
886
+ "loss": 0.0037,
887
+ "rewards/accuracies": 0.8500000238418579,
888
+ "rewards/chosen": -3.5051655769348145,
889
+ "rewards/margins": 2.5177574157714844,
890
+ "rewards/rejected": -6.022922515869141,
891
  "step": 620
892
  },
893
  {
894
  "epoch": 1.77,
895
+ "learning_rate": 1.9088955254370351e-07,
896
+ "logits/chosen": 0.5824363827705383,
897
+ "logits/rejected": 0.765023946762085,
898
+ "logps/chosen": -715.6721801757812,
899
+ "logps/rejected": -855.0099487304688,
900
+ "loss": 0.0042,
901
+ "rewards/accuracies": 0.793749988079071,
902
+ "rewards/chosen": -3.933236598968506,
903
+ "rewards/margins": 2.2994723320007324,
904
+ "rewards/rejected": -6.232708930969238,
905
  "step": 630
906
  },
907
  {
908
  "epoch": 1.8,
909
+ "learning_rate": 1.465930697252882e-07,
910
+ "logits/chosen": 0.5748051404953003,
911
+ "logits/rejected": 0.668735146522522,
912
+ "logps/chosen": -811.7765502929688,
913
+ "logps/rejected": -966.9187622070312,
914
+ "loss": 0.0032,
915
+ "rewards/accuracies": 0.862500011920929,
916
+ "rewards/chosen": -4.106084823608398,
917
+ "rewards/margins": 2.9647626876831055,
918
+ "rewards/rejected": -7.070847988128662,
919
  "step": 640
920
  },
921
  {
922
  "epoch": 1.83,
923
+ "learning_rate": 1.0798390575624762e-07,
924
+ "logits/chosen": 0.7111840844154358,
925
+ "logits/rejected": 0.8005073666572571,
926
+ "logps/chosen": -721.6022338867188,
927
+ "logps/rejected": -854.2897338867188,
928
+ "loss": 0.0033,
929
+ "rewards/accuracies": 0.8374999761581421,
930
+ "rewards/chosen": -3.6958014965057373,
931
+ "rewards/margins": 2.6302475929260254,
932
+ "rewards/rejected": -6.326048851013184,
933
  "step": 650
934
  },
935
  {
936
  "epoch": 1.85,
937
+ "learning_rate": 7.515536478350849e-08,
938
+ "logits/chosen": 0.6084551215171814,
939
+ "logits/rejected": 0.6986690163612366,
940
+ "logps/chosen": -774.9620361328125,
941
+ "logps/rejected": -953.1961669921875,
942
+ "loss": 0.0036,
943
+ "rewards/accuracies": 0.875,
944
+ "rewards/chosen": -3.89312481880188,
945
+ "rewards/margins": 2.904383420944214,
946
+ "rewards/rejected": -6.79750919342041,
947
  "step": 660
948
  },
949
  {
950
  "epoch": 1.88,
951
+ "learning_rate": 4.818678131461862e-08,
952
+ "logits/chosen": 0.7103031277656555,
953
+ "logits/rejected": 0.8091765642166138,
954
+ "logps/chosen": -720.9351806640625,
955
+ "logps/rejected": -879.3292846679688,
956
+ "loss": 0.0028,
957
+ "rewards/accuracies": 0.8374999761581421,
958
+ "rewards/chosen": -3.853041887283325,
959
+ "rewards/margins": 2.506618022918701,
960
+ "rewards/rejected": -6.359659671783447,
961
  "step": 670
962
  },
963
  {
964
  "epoch": 1.91,
965
+ "learning_rate": 2.7143328495423948e-08,
966
+ "logits/chosen": 0.5255075693130493,
967
+ "logits/rejected": 0.7452009320259094,
968
+ "logps/chosen": -735.2042236328125,
969
+ "logps/rejected": -863.3488159179688,
970
+ "loss": 0.0034,
971
+ "rewards/accuracies": 0.824999988079071,
972
+ "rewards/chosen": -3.803483247756958,
973
+ "rewards/margins": 2.5907998085021973,
974
+ "rewards/rejected": -6.394282817840576,
975
  "step": 680
976
  },
977
  {
978
  "epoch": 1.94,
979
+ "learning_rate": 1.2075860610551715e-08,
980
+ "logits/chosen": 0.4812963604927063,
981
+ "logits/rejected": 0.5726746320724487,
982
+ "logps/chosen": -796.3458251953125,
983
+ "logps/rejected": -932.7238159179688,
984
+ "loss": 0.0033,
985
+ "rewards/accuracies": 0.8374999761581421,
986
+ "rewards/chosen": -4.022668361663818,
987
+ "rewards/margins": 2.6870803833007812,
988
+ "rewards/rejected": -6.7097487449646,
989
  "step": 690
990
  },
991
  {
992
  "epoch": 1.97,
993
+ "learning_rate": 3.0207901873091638e-09,
994
+ "logits/chosen": 0.6011410355567932,
995
+ "logits/rejected": 0.7141892313957214,
996
+ "logps/chosen": -788.8591918945312,
997
+ "logps/rejected": -992.2294921875,
998
+ "loss": 0.0025,
999
+ "rewards/accuracies": 0.831250011920929,
1000
+ "rewards/chosen": -4.254032611846924,
1001
+ "rewards/margins": 2.852738380432129,
1002
+ "rewards/rejected": -7.1067705154418945,
1003
  "step": 700
1004
  },
1005
  {
1006
  "epoch": 2.0,
1007
  "learning_rate": 0.0,
1008
+ "logits/chosen": 0.6989771127700806,
1009
+ "logits/rejected": 0.749651312828064,
1010
+ "logps/chosen": -713.0648803710938,
1011
+ "logps/rejected": -953.052734375,
1012
+ "loss": 0.0041,
1013
+ "rewards/accuracies": 0.8500000238418579,
1014
+ "rewards/chosen": -3.8918204307556152,
1015
+ "rewards/margins": 3.1703882217407227,
1016
+ "rewards/rejected": -7.062208652496338,
1017
  "step": 710
1018
  },
1019
  {
1020
  "epoch": 2.0,
1021
  "step": 710,
1022
  "total_flos": 0.0,
1023
+ "train_loss": 0.1250429530667377,
1024
+ "train_runtime": 7954.4917,
1025
+ "train_samples_per_second": 11.452,
1026
+ "train_steps_per_second": 0.089
1027
  }
1028
  ],
1029
  "logging_steps": 10,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c481ba85df8845bff76e10943520a1179a74d1197923c5d013b878850f576711
3
  size 6008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ca9dc28a8826421823e18d437752c8a513c4321a73f387682f187d9117b4213
3
  size 6008