RikkiXu commited on
Commit
af072c7
1 Parent(s): c7deaef

Model save

Browse files
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 1.4416759454036356,
4
- "train_runtime": 6842.4959,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 8.219,
7
  "train_steps_per_second": 0.064
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.5750866080199396,
4
+ "train_runtime": 6876.7322,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.178,
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:6093fff14ba5f70815efc9116b0a522f11f987b52b5aa7d02f24ea0352249042
3
  size 4943162336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6621176b565448451d7eae3d8bd14fd5fd1fef4e2f88cf5a9d5c854ce3c605c3
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:2468b1e3dad5ceeb033a4bb6e441d5b67b9701a6f1c187f28ec9bd44129ed13d
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c00164aa51a72eff6abefe93a27b79678c281dc71eb0d5b66cf12cdf16b78fb4
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:15a477442397d0df636b48785e2ed5386c9320212d61b7050a369d25d4b0cd06
3
  size 4540516344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee2f9d7f97bbe93fd8937d911ba8fe3705cd1aa578b01f44961a2b34c1da3105
3
  size 4540516344
runs/Jul03_03-59-22_n136-100-194/events.out.tfevents.1719950443.n136-100-194.1966183.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0cca4b39b7a92edc9c28756cede3b5c934bf9c41739fd6574b339c15178bb114
3
- size 32997
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd79ccf57834726043ae92c267517c0ab1849fa087d644d9aae74213552b7499
3
+ size 35415
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 1.4416759454036356,
4
- "train_runtime": 6842.4959,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 8.219,
7
  "train_steps_per_second": 0.064
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.5750866080199396,
4
+ "train_runtime": 6876.7322,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.178,
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": 37.89754580186611,
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": 1.7839,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
@@ -25,656 +25,656 @@
25
  },
26
  {
27
  "epoch": 0.02,
28
- "grad_norm": 43.0334370941406,
29
  "learning_rate": 2.2727272727272725e-08,
30
- "logits/chosen": -1.7035377025604248,
31
- "logits/rejected": -1.6685911417007446,
32
- "logps/chosen": -1.213752269744873,
33
- "logps/rejected": -1.2205302715301514,
34
- "loss": 1.784,
35
  "rewards/accuracies": 0.4166666567325592,
36
- "rewards/chosen": -0.0008936466183513403,
37
- "rewards/margins": -0.0013262000866234303,
38
- "rewards/rejected": 0.0004325533809605986,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
- "grad_norm": 43.872809472305185,
44
  "learning_rate": 4.545454545454545e-08,
45
- "logits/chosen": -1.7794265747070312,
46
- "logits/rejected": -1.7346296310424805,
47
- "logps/chosen": -1.1445558071136475,
48
- "logps/rejected": -1.1853783130645752,
49
- "loss": 1.7826,
50
- "rewards/accuracies": 0.581250011920929,
51
- "rewards/chosen": -0.00266675790771842,
52
- "rewards/margins": 0.0011707551311701536,
53
- "rewards/rejected": -0.0038375132717192173,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
- "grad_norm": 49.95137473484675,
59
  "learning_rate": 6.818181818181817e-08,
60
- "logits/chosen": -1.7446266412734985,
61
- "logits/rejected": -1.675819754600525,
62
- "logps/chosen": -1.1954917907714844,
63
- "logps/rejected": -1.248420000076294,
64
- "loss": 1.7757,
65
- "rewards/accuracies": 0.625,
66
- "rewards/chosen": -0.025180673226714134,
67
- "rewards/margins": 0.011397024616599083,
68
- "rewards/rejected": -0.03657769784331322,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
- "grad_norm": 33.109734069071884,
74
  "learning_rate": 9.09090909090909e-08,
75
- "logits/chosen": -1.7313644886016846,
76
- "logits/rejected": -1.664637804031372,
77
- "logps/chosen": -1.2544224262237549,
78
- "logps/rejected": -1.3396742343902588,
79
- "loss": 1.7586,
80
- "rewards/accuracies": 0.643750011920929,
81
- "rewards/chosen": -0.09494772553443909,
82
- "rewards/margins": 0.06585659831762314,
83
- "rewards/rejected": -0.16080433130264282,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
- "grad_norm": 41.50669458446675,
89
  "learning_rate": 9.994307990108962e-08,
90
- "logits/chosen": -1.6906191110610962,
91
- "logits/rejected": -1.6255552768707275,
92
- "logps/chosen": -1.307455062866211,
93
- "logps/rejected": -1.3686177730560303,
94
- "loss": 1.7249,
95
- "rewards/accuracies": 0.706250011920929,
96
- "rewards/chosen": -0.19873608648777008,
97
- "rewards/margins": 0.0821404829621315,
98
- "rewards/rejected": -0.2808765470981598,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
- "grad_norm": 50.44913139021065,
104
  "learning_rate": 9.959570405988094e-08,
105
- "logits/chosen": -1.715877890586853,
106
- "logits/rejected": -1.6342283487319946,
107
- "logps/chosen": -1.3228412866592407,
108
- "logps/rejected": -1.416560411453247,
109
- "loss": 1.7099,
110
- "rewards/accuracies": 0.643750011920929,
111
- "rewards/chosen": -0.4274860918521881,
112
- "rewards/margins": 0.08358138054609299,
113
- "rewards/rejected": -0.5110675096511841,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
- "grad_norm": 38.15307116043539,
119
  "learning_rate": 9.893476820924666e-08,
120
- "logits/chosen": -1.7885860204696655,
121
- "logits/rejected": -1.6976776123046875,
122
- "logps/chosen": -1.5145132541656494,
123
- "logps/rejected": -1.642346739768982,
124
- "loss": 1.6855,
125
- "rewards/accuracies": 0.6812499761581421,
126
- "rewards/chosen": -0.6033424735069275,
127
- "rewards/margins": 0.16233710944652557,
128
- "rewards/rejected": -0.7656795978546143,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
- "grad_norm": 41.92014391702272,
134
  "learning_rate": 9.796445099843647e-08,
135
- "logits/chosen": -1.7705227136611938,
136
- "logits/rejected": -1.6810245513916016,
137
- "logps/chosen": -1.599914789199829,
138
- "logps/rejected": -1.775125503540039,
139
- "loss": 1.6743,
140
- "rewards/accuracies": 0.6812499761581421,
141
- "rewards/chosen": -0.7858928442001343,
142
- "rewards/margins": 0.2293611466884613,
143
- "rewards/rejected": -1.015254020690918,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
- "grad_norm": 46.613355308184076,
149
  "learning_rate": 9.669088708527066e-08,
150
- "logits/chosen": -1.7133184671401978,
151
- "logits/rejected": -1.6404612064361572,
152
- "logps/chosen": -1.7784910202026367,
153
- "logps/rejected": -1.8583282232284546,
154
- "loss": 1.6455,
155
- "rewards/accuracies": 0.637499988079071,
156
- "rewards/chosen": -1.085587739944458,
157
- "rewards/margins": 0.14558473229408264,
158
- "rewards/rejected": -1.2311723232269287,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
- "grad_norm": 52.120470113937294,
164
  "learning_rate": 9.512212835085849e-08,
165
- "logits/chosen": -1.7523702383041382,
166
- "logits/rejected": -1.6582797765731812,
167
- "logps/chosen": -1.8492581844329834,
168
- "logps/rejected": -2.0074825286865234,
169
- "loss": 1.6089,
170
  "rewards/accuracies": 0.706250011920929,
171
- "rewards/chosen": -1.3616061210632324,
172
- "rewards/margins": 0.23102827370166779,
173
- "rewards/rejected": -1.5926344394683838,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
- "grad_norm": 63.63123122540048,
179
  "learning_rate": 9.326809299301306e-08,
180
- "logits/chosen": -1.755048155784607,
181
- "logits/rejected": -1.6466388702392578,
182
- "logps/chosen": -1.9677321910858154,
183
- "logps/rejected": -2.2299115657806396,
184
- "loss": 1.584,
185
- "rewards/accuracies": 0.71875,
186
- "rewards/chosen": -1.5341805219650269,
187
- "rewards/margins": 0.42115211486816406,
188
- "rewards/rejected": -1.9553325176239014,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
- "grad_norm": 56.757728243270925,
194
  "learning_rate": 9.114050282021158e-08,
195
- "logits/chosen": -1.7328088283538818,
196
- "logits/rejected": -1.6686969995498657,
197
- "logps/chosen": -1.9586963653564453,
198
- "logps/rejected": -2.1962826251983643,
199
- "loss": 1.5726,
200
- "rewards/accuracies": 0.737500011920929,
201
- "rewards/chosen": -1.6222673654556274,
202
- "rewards/margins": 0.38038817048072815,
203
- "rewards/rejected": -2.0026557445526123,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
- "grad_norm": 54.94031900178828,
209
  "learning_rate": 8.875280914254802e-08,
210
- "logits/chosen": -1.7098445892333984,
211
- "logits/rejected": -1.6136401891708374,
212
- "logps/chosen": -2.23858380317688,
213
- "logps/rejected": -2.5136303901672363,
214
- "loss": 1.5414,
215
- "rewards/accuracies": 0.668749988079071,
216
- "rewards/chosen": -2.0915074348449707,
217
- "rewards/margins": 0.4932435154914856,
218
- "rewards/rejected": -2.5847511291503906,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
- "grad_norm": 46.69719261590635,
224
  "learning_rate": 8.612010772821971e-08,
225
- "logits/chosen": -1.7307876348495483,
226
- "logits/rejected": -1.6822052001953125,
227
- "logps/chosen": -2.2878048419952393,
228
- "logps/rejected": -2.525545597076416,
229
- "loss": 1.5268,
230
- "rewards/accuracies": 0.731249988079071,
231
- "rewards/chosen": -2.1478517055511475,
232
- "rewards/margins": 0.5253016948699951,
233
- "rewards/rejected": -2.6731534004211426,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
- "grad_norm": 61.28228352616757,
239
  "learning_rate": 8.325904336322055e-08,
240
- "logits/chosen": -1.6986526250839233,
241
- "logits/rejected": -1.6437307596206665,
242
- "logps/chosen": -2.6470184326171875,
243
- "logps/rejected": -2.9312243461608887,
244
- "loss": 1.5342,
245
- "rewards/accuracies": 0.65625,
246
- "rewards/chosen": -2.9593071937561035,
247
- "rewards/margins": 0.47406500577926636,
248
- "rewards/rejected": -3.4333720207214355,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
- "grad_norm": 51.65823353682169,
254
  "learning_rate": 8.01877046176447e-08,
255
- "logits/chosen": -1.625265121459961,
256
- "logits/rejected": -1.5576308965682983,
257
- "logps/chosen": -3.098374843597412,
258
- "logps/rejected": -3.4221739768981934,
259
- "loss": 1.4524,
260
  "rewards/accuracies": 0.65625,
261
- "rewards/chosen": -3.8401496410369873,
262
- "rewards/margins": 0.5448259115219116,
263
- "rewards/rejected": -4.384975910186768,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
- "grad_norm": 49.892494463892014,
269
  "learning_rate": 7.692550948392249e-08,
270
- "logits/chosen": -1.6712944507598877,
271
- "logits/rejected": -1.6248451471328735,
272
- "logps/chosen": -3.2628607749938965,
273
- "logps/rejected": -3.629713535308838,
274
- "loss": 1.4853,
275
- "rewards/accuracies": 0.65625,
276
- "rewards/chosen": -4.10521125793457,
277
- "rewards/margins": 0.7128149271011353,
278
- "rewards/rejected": -4.818026542663574,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
- "grad_norm": 66.77572068779351,
284
  "learning_rate": 7.349308261002021e-08,
285
- "logits/chosen": -1.6318824291229248,
286
- "logits/rejected": -1.579535722732544,
287
- "logps/chosen": -3.1891632080078125,
288
- "logps/rejected": -3.5561447143554688,
289
- "loss": 1.4493,
290
- "rewards/accuracies": 0.6937500238418579,
291
- "rewards/chosen": -3.9841549396514893,
292
- "rewards/margins": 0.6346625089645386,
293
- "rewards/rejected": -4.618817329406738,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
- "grad_norm": 60.84208005393647,
299
  "learning_rate": 6.991212490377531e-08,
300
- "logits/chosen": -1.6940677165985107,
301
- "logits/rejected": -1.6511356830596924,
302
- "logps/chosen": -3.2117652893066406,
303
- "logps/rejected": -3.6691880226135254,
304
- "loss": 1.3684,
305
- "rewards/accuracies": 0.7124999761581421,
306
- "rewards/chosen": -3.963325023651123,
307
- "rewards/margins": 0.871760368347168,
308
- "rewards/rejected": -4.835085391998291,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
- "grad_norm": 69.2275656586859,
314
  "learning_rate": 6.620527633276978e-08,
315
- "logits/chosen": -1.6109001636505127,
316
- "logits/rejected": -1.5511188507080078,
317
- "logps/chosen": -3.5304808616638184,
318
- "logps/rejected": -4.158318519592285,
319
- "loss": 1.4102,
320
- "rewards/accuracies": 0.706250011920929,
321
- "rewards/chosen": -4.672055721282959,
322
- "rewards/margins": 1.0820972919464111,
323
- "rewards/rejected": -5.754153251647949,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
- "grad_norm": 63.33912483688516,
329
  "learning_rate": 6.239597278716581e-08,
330
- "logits/chosen": -1.6476631164550781,
331
- "logits/rejected": -1.596917986869812,
332
- "logps/chosen": -4.077167987823486,
333
- "logps/rejected": -4.615104675292969,
334
- "loss": 1.3553,
335
- "rewards/accuracies": 0.731249988079071,
336
- "rewards/chosen": -5.663676738739014,
337
- "rewards/margins": 1.117199420928955,
338
- "rewards/rejected": -6.780876159667969,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
- "grad_norm": 65.97263972953417,
344
  "learning_rate": 5.8508297910462456e-08,
345
- "logits/chosen": -1.5933263301849365,
346
- "logits/rejected": -1.521242380142212,
347
- "logps/chosen": -4.065090179443359,
348
- "logps/rejected": -4.714545726776123,
349
- "loss": 1.3527,
350
- "rewards/accuracies": 0.731249988079071,
351
- "rewards/chosen": -5.803150177001953,
352
- "rewards/margins": 1.1833642721176147,
353
- "rewards/rejected": -6.986514091491699,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
- "grad_norm": 66.61715813578674,
359
  "learning_rate": 5.456683083494731e-08,
360
- "logits/chosen": -1.584855079650879,
361
- "logits/rejected": -1.5489156246185303,
362
- "logps/chosen": -3.8250725269317627,
363
- "logps/rejected": -4.294682025909424,
364
- "loss": 1.4145,
365
- "rewards/accuracies": 0.6499999761581421,
366
- "rewards/chosen": -5.262526988983154,
367
- "rewards/margins": 0.8917341232299805,
368
- "rewards/rejected": -6.154262065887451,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
- "grad_norm": 71.26593821270333,
374
  "learning_rate": 5.059649078450834e-08,
375
- "logits/chosen": -1.5678187608718872,
376
- "logits/rejected": -1.5303425788879395,
377
- "logps/chosen": -3.8835272789001465,
378
- "logps/rejected": -4.416079521179199,
379
- "loss": 1.3342,
380
- "rewards/accuracies": 0.637499988079071,
381
- "rewards/chosen": -5.430348873138428,
382
- "rewards/margins": 0.9410263895988464,
383
- "rewards/rejected": -6.371375560760498,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
- "grad_norm": 59.385541974267724,
389
  "learning_rate": 4.6622379527277186e-08,
390
- "logits/chosen": -1.5805360078811646,
391
- "logits/rejected": -1.5284383296966553,
392
- "logps/chosen": -3.8606419563293457,
393
- "logps/rejected": -4.334124565124512,
394
- "loss": 1.3377,
395
- "rewards/accuracies": 0.625,
396
- "rewards/chosen": -5.443507194519043,
397
- "rewards/margins": 0.8808059692382812,
398
- "rewards/rejected": -6.324312686920166,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
- "grad_norm": 65.2459397054924,
404
  "learning_rate": 4.26696226741691e-08,
405
- "logits/chosen": -1.5909301042556763,
406
- "logits/rejected": -1.5298397541046143,
407
- "logps/chosen": -3.996027708053589,
408
- "logps/rejected": -4.560092926025391,
409
- "loss": 1.3667,
410
- "rewards/accuracies": 0.65625,
411
- "rewards/chosen": -5.5264081954956055,
412
- "rewards/margins": 1.0881966352462769,
413
- "rewards/rejected": -6.614604949951172,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
- "grad_norm": 67.8504993881358,
419
  "learning_rate": 3.876321082668098e-08,
420
- "logits/chosen": -1.6403207778930664,
421
- "logits/rejected": -1.572745442390442,
422
- "logps/chosen": -3.9399516582489014,
423
- "logps/rejected": -4.576107501983643,
424
- "loss": 1.3184,
425
- "rewards/accuracies": 0.7124999761581421,
426
- "rewards/chosen": -5.4029221534729,
427
- "rewards/margins": 1.2536938190460205,
428
- "rewards/rejected": -6.656615257263184,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
- "grad_norm": 73.6890162321455,
434
  "learning_rate": 3.492784157826244e-08,
435
- "logits/chosen": -1.57921302318573,
436
- "logits/rejected": -1.4859352111816406,
437
- "logps/chosen": -4.124494552612305,
438
- "logps/rejected": -4.830277442932129,
439
- "loss": 1.3318,
440
- "rewards/accuracies": 0.7250000238418579,
441
- "rewards/chosen": -5.803236484527588,
442
- "rewards/margins": 1.340786337852478,
443
- "rewards/rejected": -7.1440229415893555,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
- "grad_norm": 58.82124057451152,
449
  "learning_rate": 3.118776336817812e-08,
450
- "logits/chosen": -1.5940860509872437,
451
- "logits/rejected": -1.5397272109985352,
452
- "logps/chosen": -4.048806190490723,
453
- "logps/rejected": -4.8185248374938965,
454
- "loss": 1.2939,
455
  "rewards/accuracies": 0.7250000238418579,
456
- "rewards/chosen": -5.691115379333496,
457
- "rewards/margins": 1.5250051021575928,
458
- "rewards/rejected": -7.21612024307251,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
- "grad_norm": 64.00740460254956,
464
  "learning_rate": 2.7566622175067443e-08,
465
- "logits/chosen": -1.5681233406066895,
466
- "logits/rejected": -1.5011963844299316,
467
- "logps/chosen": -4.236255168914795,
468
- "logps/rejected": -5.085611820220947,
469
- "loss": 1.2905,
470
- "rewards/accuracies": 0.7437499761581421,
471
- "rewards/chosen": -6.055342674255371,
472
- "rewards/margins": 1.5427405834197998,
473
- "rewards/rejected": -7.598082065582275,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
- "grad_norm": 67.41505792418151,
479
  "learning_rate": 2.408731201945432e-08,
480
- "logits/chosen": -1.5719449520111084,
481
- "logits/rejected": -1.5258766412734985,
482
- "logps/chosen": -4.234918117523193,
483
- "logps/rejected": -4.75051736831665,
484
- "loss": 1.3279,
485
- "rewards/accuracies": 0.706250011920929,
486
- "rewards/chosen": -6.082045555114746,
487
- "rewards/margins": 0.9908892512321472,
488
- "rewards/rejected": -7.072935581207275,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
- "grad_norm": 71.34331865518075,
494
  "learning_rate": 2.0771830220378112e-08,
495
- "logits/chosen": -1.5344269275665283,
496
- "logits/rejected": -1.467549204826355,
497
- "logps/chosen": -4.098411560058594,
498
- "logps/rejected": -4.607344627380371,
499
- "loss": 1.3276,
500
- "rewards/accuracies": 0.6187499761581421,
501
- "rewards/chosen": -5.751054763793945,
502
- "rewards/margins": 1.0335659980773926,
503
- "rewards/rejected": -6.784620761871338,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
- "grad_norm": 60.225418376453746,
509
  "learning_rate": 1.7641138321260257e-08,
510
- "logits/chosen": -1.5728257894515991,
511
- "logits/rejected": -1.5012346506118774,
512
- "logps/chosen": -3.934291124343872,
513
- "logps/rejected": -4.8078932762146,
514
- "loss": 1.287,
515
- "rewards/accuracies": 0.8062499761581421,
516
- "rewards/chosen": -5.500872611999512,
517
- "rewards/margins": 1.6921342611312866,
518
- "rewards/rejected": -7.193006992340088,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
- "grad_norm": 71.67365382059906,
524
  "learning_rate": 1.4715029564277793e-08,
525
- "logits/chosen": -1.625579595565796,
526
- "logits/rejected": -1.5787419080734253,
527
- "logps/chosen": -3.8221092224121094,
528
- "logps/rejected": -4.613158226013184,
529
- "loss": 1.324,
530
- "rewards/accuracies": 0.7875000238418579,
531
- "rewards/chosen": -5.234897613525391,
532
- "rewards/margins": 1.5079706907272339,
533
- "rewards/rejected": -6.742868900299072,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
- "grad_norm": 65.09638515221782,
539
  "learning_rate": 1.2012003751113343e-08,
540
- "logits/chosen": -1.6107479333877563,
541
- "logits/rejected": -1.55300772190094,
542
- "logps/chosen": -4.135249137878418,
543
- "logps/rejected": -4.8890886306762695,
544
- "loss": 1.285,
545
- "rewards/accuracies": 0.6812499761581421,
546
- "rewards/chosen": -5.946158409118652,
547
- "rewards/margins": 1.407297968864441,
548
- "rewards/rejected": -7.353456974029541,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
- "grad_norm": 69.11132986778664,
554
  "learning_rate": 9.549150281252633e-09,
555
- "logits/chosen": -1.557506799697876,
556
- "logits/rejected": -1.5132434368133545,
557
- "logps/chosen": -3.8835136890411377,
558
- "logps/rejected": -4.530081748962402,
559
- "loss": 1.2779,
560
- "rewards/accuracies": 0.6937500238418579,
561
- "rewards/chosen": -5.422757625579834,
562
- "rewards/margins": 1.189673662185669,
563
- "rewards/rejected": -6.612431526184082,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
- "grad_norm": 67.45416999543298,
569
  "learning_rate": 7.3420401072985306e-09,
570
- "logits/chosen": -1.6095161437988281,
571
- "logits/rejected": -1.547431468963623,
572
- "logps/chosen": -3.9920966625213623,
573
- "logps/rejected": -4.789999485015869,
574
- "loss": 1.2684,
575
- "rewards/accuracies": 0.737500011920929,
576
- "rewards/chosen": -5.554965972900391,
577
- "rewards/margins": 1.4808779954910278,
578
- "rewards/rejected": -7.035844326019287,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
- "grad_norm": 67.00084975743678,
584
  "learning_rate": 5.404627290395369e-09,
585
- "logits/chosen": -1.5683467388153076,
586
- "logits/rejected": -1.5040208101272583,
587
- "logps/chosen": -3.8986880779266357,
588
- "logps/rejected": -4.654471397399902,
589
- "loss": 1.2751,
590
- "rewards/accuracies": 0.7437499761581421,
591
- "rewards/chosen": -5.4181647300720215,
592
- "rewards/margins": 1.4315736293792725,
593
- "rewards/rejected": -6.849739074707031,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
- "grad_norm": 72.22372529068039,
599
  "learning_rate": 3.74916077816162e-09,
600
- "logits/chosen": -1.5683157444000244,
601
- "logits/rejected": -1.5068817138671875,
602
- "logps/chosen": -3.9735209941864014,
603
- "logps/rejected": -4.631840229034424,
604
- "loss": 1.3125,
605
- "rewards/accuracies": 0.6937500238418579,
606
- "rewards/chosen": -5.647452354431152,
607
- "rewards/margins": 1.2843806743621826,
608
- "rewards/rejected": -6.931832790374756,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
- "grad_norm": 69.75665119516057,
614
  "learning_rate": 2.386106962899165e-09,
615
- "logits/chosen": -1.4955816268920898,
616
- "logits/rejected": -1.433675765991211,
617
- "logps/chosen": -4.157034873962402,
618
- "logps/rejected": -4.851306438446045,
619
- "loss": 1.2727,
620
- "rewards/accuracies": 0.7124999761581421,
621
- "rewards/chosen": -5.861227989196777,
622
- "rewards/margins": 1.2758640050888062,
623
- "rewards/rejected": -7.137092590332031,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
- "grad_norm": 72.88668820386522,
629
  "learning_rate": 1.3240835096913706e-09,
630
- "logits/chosen": -1.5237390995025635,
631
- "logits/rejected": -1.422570824623108,
632
- "logps/chosen": -3.9286224842071533,
633
- "logps/rejected": -4.831568241119385,
634
- "loss": 1.3583,
635
- "rewards/accuracies": 0.6937500238418579,
636
- "rewards/chosen": -5.514142990112305,
637
- "rewards/margins": 1.6670624017715454,
638
- "rewards/rejected": -7.181205749511719,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
- "grad_norm": 59.201842668489434,
644
  "learning_rate": 5.698048727497462e-10,
645
- "logits/chosen": -1.5552051067352295,
646
- "logits/rejected": -1.4861769676208496,
647
- "logps/chosen": -4.0786638259887695,
648
- "logps/rejected": -4.879299640655518,
649
- "loss": 1.2929,
650
- "rewards/accuracies": 0.699999988079071,
651
- "rewards/chosen": -5.774731159210205,
652
- "rewards/margins": 1.4673572778701782,
653
- "rewards/rejected": -7.24208927154541,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
- "grad_norm": 61.30539325165261,
659
  "learning_rate": 1.2803984447259387e-10,
660
- "logits/chosen": -1.5659511089324951,
661
- "logits/rejected": -1.5160239934921265,
662
- "logps/chosen": -4.094729423522949,
663
- "logps/rejected": -4.9491190910339355,
664
- "loss": 1.2283,
665
- "rewards/accuracies": 0.7124999761581421,
666
- "rewards/chosen": -5.824779510498047,
667
- "rewards/margins": 1.6134742498397827,
668
- "rewards/rejected": -7.438253879547119,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
- "train_loss": 1.4416759454036356,
676
- "train_runtime": 6842.4959,
677
- "train_samples_per_second": 8.219,
678
  "train_steps_per_second": 0.064
679
  }
680
  ],
 
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
+ "grad_norm": 46.148553571593695,
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.6933,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
 
25
  },
26
  {
27
  "epoch": 0.02,
28
+ "grad_norm": 51.49370478851895,
29
  "learning_rate": 2.2727272727272725e-08,
30
+ "logits/chosen": -1.7024296522140503,
31
+ "logits/rejected": -1.6681036949157715,
32
+ "logps/chosen": -1.2134602069854736,
33
+ "logps/rejected": -1.2201855182647705,
34
+ "loss": 0.6934,
35
  "rewards/accuracies": 0.4166666567325592,
36
+ "rewards/chosen": -0.0006183187360875309,
37
+ "rewards/margins": -0.0028628669679164886,
38
+ "rewards/rejected": 0.002244548173621297,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
+ "grad_norm": 51.98148395544936,
44
  "learning_rate": 4.545454545454545e-08,
45
+ "logits/chosen": -1.7797874212265015,
46
+ "logits/rejected": -1.7346786260604858,
47
+ "logps/chosen": -1.144641637802124,
48
+ "logps/rejected": -1.1857845783233643,
49
+ "loss": 0.6917,
50
+ "rewards/accuracies": 0.6000000238418579,
51
+ "rewards/chosen": -0.005677336361259222,
52
+ "rewards/margins": 0.00362207368016243,
53
+ "rewards/rejected": -0.00929941050708294,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
+ "grad_norm": 53.07532238582354,
59
  "learning_rate": 6.818181818181817e-08,
60
+ "logits/chosen": -1.7449315786361694,
61
+ "logits/rejected": -1.6763038635253906,
62
+ "logps/chosen": -1.1949758529663086,
63
+ "logps/rejected": -1.247839093208313,
64
+ "loss": 0.6842,
65
+ "rewards/accuracies": 0.606249988079071,
66
+ "rewards/chosen": -0.048297375440597534,
67
+ "rewards/margins": 0.022534046322107315,
68
+ "rewards/rejected": -0.07083141803741455,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
+ "grad_norm": 38.722113074558656,
74
  "learning_rate": 9.09090909090909e-08,
75
+ "logits/chosen": -1.7315384149551392,
76
+ "logits/rejected": -1.6650068759918213,
77
+ "logps/chosen": -1.2470494508743286,
78
+ "logps/rejected": -1.3278839588165283,
79
+ "loss": 0.6711,
80
+ "rewards/accuracies": 0.6187499761581421,
81
+ "rewards/chosen": -0.16040349006652832,
82
+ "rewards/margins": 0.11404403299093246,
83
+ "rewards/rejected": -0.274447500705719,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
+ "grad_norm": 42.974027284467326,
89
  "learning_rate": 9.994307990108962e-08,
90
+ "logits/chosen": -1.694649338722229,
91
+ "logits/rejected": -1.6296478509902954,
92
+ "logps/chosen": -1.2850488424301147,
93
+ "logps/rejected": -1.34065842628479,
94
+ "loss": 0.6472,
95
+ "rewards/accuracies": 0.699999988079071,
96
+ "rewards/chosen": -0.30784696340560913,
97
+ "rewards/margins": 0.14206835627555847,
98
+ "rewards/rejected": -0.4499153196811676,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
+ "grad_norm": 57.394151591405155,
104
  "learning_rate": 9.959570405988094e-08,
105
+ "logits/chosen": -1.7283236980438232,
106
+ "logits/rejected": -1.6481777429580688,
107
+ "logps/chosen": -1.2401286363601685,
108
+ "logps/rejected": -1.3243335485458374,
109
+ "loss": 0.6482,
110
+ "rewards/accuracies": 0.6625000238418579,
111
+ "rewards/chosen": -0.5241212248802185,
112
+ "rewards/margins": 0.12910696864128113,
113
+ "rewards/rejected": -0.6532281637191772,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
+ "grad_norm": 39.32038204769895,
119
  "learning_rate": 9.893476820924666e-08,
120
+ "logits/chosen": -1.8152574300765991,
121
+ "logits/rejected": -1.7302563190460205,
122
+ "logps/chosen": -1.3927916288375854,
123
+ "logps/rejected": -1.4982484579086304,
124
+ "loss": 0.6292,
125
+ "rewards/accuracies": 0.6625000238418579,
126
+ "rewards/chosen": -0.7197983264923096,
127
+ "rewards/margins": 0.23516786098480225,
128
+ "rewards/rejected": -0.9549661874771118,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
+ "grad_norm": 42.61872072458202,
134
  "learning_rate": 9.796445099843647e-08,
135
+ "logits/chosen": -1.8079169988632202,
136
+ "logits/rejected": -1.7261250019073486,
137
+ "logps/chosen": -1.3969148397445679,
138
+ "logps/rejected": -1.5204780101776123,
139
+ "loss": 0.6371,
140
+ "rewards/accuracies": 0.6312500238418579,
141
+ "rewards/chosen": -0.7597857713699341,
142
+ "rewards/margins": 0.25213322043418884,
143
+ "rewards/rejected": -1.0119189023971558,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
+ "grad_norm": 46.415577032465364,
149
  "learning_rate": 9.669088708527066e-08,
150
+ "logits/chosen": -1.7607409954071045,
151
+ "logits/rejected": -1.6981712579727173,
152
+ "logps/chosen": -1.4422895908355713,
153
+ "logps/rejected": -1.513820767402649,
154
+ "loss": 0.6098,
155
+ "rewards/accuracies": 0.6937500238418579,
156
+ "rewards/chosen": -0.8263689875602722,
157
+ "rewards/margins": 0.2579457759857178,
158
+ "rewards/rejected": -1.0843145847320557,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
+ "grad_norm": 48.85282914944215,
164
  "learning_rate": 9.512212835085849e-08,
165
+ "logits/chosen": -1.8021045923233032,
166
+ "logits/rejected": -1.7189390659332275,
167
+ "logps/chosen": -1.4129152297973633,
168
+ "logps/rejected": -1.5352809429168701,
169
+ "loss": 0.6021,
170
  "rewards/accuracies": 0.706250011920929,
171
+ "rewards/chosen": -0.9778400659561157,
172
+ "rewards/margins": 0.3186227083206177,
173
+ "rewards/rejected": -1.2964627742767334,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
+ "grad_norm": 46.36151439283495,
179
  "learning_rate": 9.326809299301306e-08,
180
+ "logits/chosen": -1.79812490940094,
181
+ "logits/rejected": -1.6997324228286743,
182
+ "logps/chosen": -1.4540174007415771,
183
+ "logps/rejected": -1.604739785194397,
184
+ "loss": 0.5967,
185
+ "rewards/accuracies": 0.6875,
186
+ "rewards/chosen": -1.0135016441345215,
187
+ "rewards/margins": 0.396476686000824,
188
+ "rewards/rejected": -1.4099782705307007,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
+ "grad_norm": 45.302122433536,
194
  "learning_rate": 9.114050282021158e-08,
195
+ "logits/chosen": -1.793320894241333,
196
+ "logits/rejected": -1.7357345819473267,
197
+ "logps/chosen": -1.4166992902755737,
198
+ "logps/rejected": -1.558625340461731,
199
+ "loss": 0.5941,
200
+ "rewards/accuracies": 0.7562500238418579,
201
+ "rewards/chosen": -1.0765467882156372,
202
+ "rewards/margins": 0.37813490629196167,
203
+ "rewards/rejected": -1.454681634902954,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
+ "grad_norm": 44.726794355611766,
209
  "learning_rate": 8.875280914254802e-08,
210
+ "logits/chosen": -1.7930294275283813,
211
+ "logits/rejected": -1.7056195735931396,
212
+ "logps/chosen": -1.5269113779067993,
213
+ "logps/rejected": -1.6759881973266602,
214
+ "loss": 0.5783,
215
+ "rewards/accuracies": 0.706250011920929,
216
+ "rewards/chosen": -1.3363250494003296,
217
+ "rewards/margins": 0.48260822892189026,
218
+ "rewards/rejected": -1.818933129310608,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
+ "grad_norm": 36.98397799808558,
224
  "learning_rate": 8.612010772821971e-08,
225
+ "logits/chosen": -1.8213390111923218,
226
+ "logits/rejected": -1.779136061668396,
227
+ "logps/chosen": -1.5163151025772095,
228
+ "logps/rejected": -1.6115764379501343,
229
+ "loss": 0.5802,
230
+ "rewards/accuracies": 0.7562500238418579,
231
+ "rewards/chosen": -1.2097442150115967,
232
+ "rewards/margins": 0.4806848466396332,
233
+ "rewards/rejected": -1.6904290914535522,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
+ "grad_norm": 35.445773953418566,
239
  "learning_rate": 8.325904336322055e-08,
240
+ "logits/chosen": -1.7982733249664307,
241
+ "logits/rejected": -1.7448923587799072,
242
+ "logps/chosen": -1.5137693881988525,
243
+ "logps/rejected": -1.6794990301132202,
244
+ "loss": 0.5856,
245
+ "rewards/accuracies": 0.6875,
246
+ "rewards/chosen": -1.3856182098388672,
247
+ "rewards/margins": 0.4742238521575928,
248
+ "rewards/rejected": -1.85984206199646,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
+ "grad_norm": 38.005517709582755,
254
  "learning_rate": 8.01877046176447e-08,
255
+ "logits/chosen": -1.7442073822021484,
256
+ "logits/rejected": -1.6729179620742798,
257
+ "logps/chosen": -1.5704376697540283,
258
+ "logps/rejected": -1.715036392211914,
259
+ "loss": 0.5782,
260
  "rewards/accuracies": 0.65625,
261
+ "rewards/chosen": -1.5685503482818604,
262
+ "rewards/margins": 0.37285223603248596,
263
+ "rewards/rejected": -1.9414026737213135,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
+ "grad_norm": 36.45337137576972,
269
  "learning_rate": 7.692550948392249e-08,
270
+ "logits/chosen": -1.794782042503357,
271
+ "logits/rejected": -1.7365185022354126,
272
+ "logps/chosen": -1.5552618503570557,
273
+ "logps/rejected": -1.6845000982284546,
274
+ "loss": 0.5771,
275
+ "rewards/accuracies": 0.737500011920929,
276
+ "rewards/chosen": -1.3800251483917236,
277
+ "rewards/margins": 0.47517403960227966,
278
+ "rewards/rejected": -1.8551992177963257,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
+ "grad_norm": 46.08033558224558,
284
  "learning_rate": 7.349308261002021e-08,
285
+ "logits/chosen": -1.7525560855865479,
286
+ "logits/rejected": -1.6976267099380493,
287
+ "logps/chosen": -1.5552211999893188,
288
+ "logps/rejected": -1.716825246810913,
289
+ "loss": 0.5709,
290
+ "rewards/accuracies": 0.7562500238418579,
291
+ "rewards/chosen": -1.4325402975082397,
292
+ "rewards/margins": 0.4478151798248291,
293
+ "rewards/rejected": -1.8803554773330688,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
+ "grad_norm": 38.67140094316328,
299
  "learning_rate": 6.991212490377531e-08,
300
+ "logits/chosen": -1.8065755367279053,
301
+ "logits/rejected": -1.7572141885757446,
302
+ "logps/chosen": -1.594746470451355,
303
+ "logps/rejected": -1.7574079036712646,
304
+ "loss": 0.5479,
305
+ "rewards/accuracies": 0.7250000238418579,
306
+ "rewards/chosen": -1.458573579788208,
307
+ "rewards/margins": 0.5644782185554504,
308
+ "rewards/rejected": -2.0230517387390137,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
+ "grad_norm": 50.20778300695083,
314
  "learning_rate": 6.620527633276978e-08,
315
+ "logits/chosen": -1.744382619857788,
316
+ "logits/rejected": -1.6726484298706055,
317
+ "logps/chosen": -1.568552851676941,
318
+ "logps/rejected": -1.8165289163589478,
319
+ "loss": 0.5629,
320
+ "rewards/accuracies": 0.7437499761581421,
321
+ "rewards/chosen": -1.4963985681533813,
322
+ "rewards/margins": 0.6447495222091675,
323
+ "rewards/rejected": -2.1411478519439697,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
+ "grad_norm": 40.815630226191416,
329
  "learning_rate": 6.239597278716581e-08,
330
+ "logits/chosen": -1.807816743850708,
331
+ "logits/rejected": -1.746860146522522,
332
+ "logps/chosen": -1.6599668264389038,
333
+ "logps/rejected": -1.7937523126602173,
334
+ "loss": 0.5518,
335
+ "rewards/accuracies": 0.7562500238418579,
336
+ "rewards/chosen": -1.658548355102539,
337
+ "rewards/margins": 0.6177951097488403,
338
+ "rewards/rejected": -2.276343584060669,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
+ "grad_norm": 36.81107380199821,
344
  "learning_rate": 5.8508297910462456e-08,
345
+ "logits/chosen": -1.7586944103240967,
346
+ "logits/rejected": -1.6771053075790405,
347
+ "logps/chosen": -1.590494155883789,
348
+ "logps/rejected": -1.8014583587646484,
349
+ "loss": 0.5551,
350
+ "rewards/accuracies": 0.737500011920929,
351
+ "rewards/chosen": -1.7079153060913086,
352
+ "rewards/margins": 0.6127630472183228,
353
+ "rewards/rejected": -2.320678234100342,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
+ "grad_norm": 42.81086496786825,
359
  "learning_rate": 5.456683083494731e-08,
360
+ "logits/chosen": -1.748581886291504,
361
+ "logits/rejected": -1.7098802328109741,
362
+ "logps/chosen": -1.5830055475234985,
363
+ "logps/rejected": -1.708268165588379,
364
+ "loss": 0.5748,
365
+ "rewards/accuracies": 0.637499988079071,
366
+ "rewards/chosen": -1.5567874908447266,
367
+ "rewards/margins": 0.40608254075050354,
368
+ "rewards/rejected": -1.9628700017929077,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
+ "grad_norm": 50.79898140443755,
374
  "learning_rate": 5.059649078450834e-08,
375
+ "logits/chosen": -1.7385135889053345,
376
+ "logits/rejected": -1.6935017108917236,
377
+ "logps/chosen": -1.5564693212509155,
378
+ "logps/rejected": -1.7349010705947876,
379
+ "loss": 0.5458,
380
+ "rewards/accuracies": 0.6812499761581421,
381
+ "rewards/chosen": -1.5524632930755615,
382
+ "rewards/margins": 0.46557125449180603,
383
+ "rewards/rejected": -2.0180344581604004,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
+ "grad_norm": 40.11976756667232,
389
  "learning_rate": 4.6622379527277186e-08,
390
+ "logits/chosen": -1.7455527782440186,
391
+ "logits/rejected": -1.694297432899475,
392
+ "logps/chosen": -1.564894437789917,
393
+ "logps/rejected": -1.7082773447036743,
394
+ "loss": 0.552,
395
+ "rewards/accuracies": 0.6937500238418579,
396
+ "rewards/chosen": -1.7040233612060547,
397
+ "rewards/margins": 0.44121259450912476,
398
+ "rewards/rejected": -2.1452362537384033,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
+ "grad_norm": 37.69291090254264,
404
  "learning_rate": 4.26696226741691e-08,
405
+ "logits/chosen": -1.7581895589828491,
406
+ "logits/rejected": -1.686743140220642,
407
+ "logps/chosen": -1.6672862768173218,
408
+ "logps/rejected": -1.827100396156311,
409
+ "loss": 0.5531,
410
+ "rewards/accuracies": 0.6625000238418579,
411
+ "rewards/chosen": -1.7378520965576172,
412
+ "rewards/margins": 0.559387743473053,
413
+ "rewards/rejected": -2.2972397804260254,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
+ "grad_norm": 37.31506455169964,
419
  "learning_rate": 3.876321082668098e-08,
420
+ "logits/chosen": -1.8174455165863037,
421
+ "logits/rejected": -1.7460702657699585,
422
+ "logps/chosen": -1.6105009317398071,
423
+ "logps/rejected": -1.7751582860946655,
424
+ "loss": 0.5459,
425
+ "rewards/accuracies": 0.7562500238418579,
426
+ "rewards/chosen": -1.4880399703979492,
427
+ "rewards/margins": 0.6213944554328918,
428
+ "rewards/rejected": -2.1094346046447754,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
+ "grad_norm": 41.98836134697342,
434
  "learning_rate": 3.492784157826244e-08,
435
+ "logits/chosen": -1.746079444885254,
436
+ "logits/rejected": -1.6474275588989258,
437
+ "logps/chosen": -1.5982965230941772,
438
+ "logps/rejected": -1.793271780014038,
439
+ "loss": 0.5515,
440
+ "rewards/accuracies": 0.71875,
441
+ "rewards/chosen": -1.5016809701919556,
442
+ "rewards/margins": 0.6383417844772339,
443
+ "rewards/rejected": -2.1400227546691895,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
+ "grad_norm": 33.538804743591584,
449
  "learning_rate": 3.118776336817812e-08,
450
+ "logits/chosen": -1.7901312112808228,
451
+ "logits/rejected": -1.7260141372680664,
452
+ "logps/chosen": -1.5818462371826172,
453
+ "logps/rejected": -1.7569477558135986,
454
+ "loss": 0.5398,
455
  "rewards/accuracies": 0.7250000238418579,
456
+ "rewards/chosen": -1.5143903493881226,
457
+ "rewards/margins": 0.6715410351753235,
458
+ "rewards/rejected": -2.185931444168091,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
+ "grad_norm": 37.791320621822486,
464
  "learning_rate": 2.7566622175067443e-08,
465
+ "logits/chosen": -1.774738073348999,
466
+ "logits/rejected": -1.7078733444213867,
467
+ "logps/chosen": -1.6364281177520752,
468
+ "logps/rejected": -1.8677997589111328,
469
+ "loss": 0.5504,
470
+ "rewards/accuracies": 0.699999988079071,
471
+ "rewards/chosen": -1.7113786935806274,
472
+ "rewards/margins": 0.6135402917861938,
473
+ "rewards/rejected": -2.3249192237854004,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
+ "grad_norm": 47.94773798292109,
479
  "learning_rate": 2.408731201945432e-08,
480
+ "logits/chosen": -1.7694079875946045,
481
+ "logits/rejected": -1.7203378677368164,
482
+ "logps/chosen": -1.58682119846344,
483
+ "logps/rejected": -1.7359449863433838,
484
+ "loss": 0.5402,
485
+ "rewards/accuracies": 0.71875,
486
+ "rewards/chosen": -1.5717027187347412,
487
+ "rewards/margins": 0.5158768892288208,
488
+ "rewards/rejected": -2.0875794887542725,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
+ "grad_norm": 39.31637908975089,
494
  "learning_rate": 2.0771830220378112e-08,
495
+ "logits/chosen": -1.7262191772460938,
496
+ "logits/rejected": -1.6669414043426514,
497
+ "logps/chosen": -1.6197150945663452,
498
+ "logps/rejected": -1.7450358867645264,
499
+ "loss": 0.5428,
500
+ "rewards/accuracies": 0.6812499761581421,
501
+ "rewards/chosen": -1.5873229503631592,
502
+ "rewards/margins": 0.5326842665672302,
503
+ "rewards/rejected": -2.120007038116455,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
+ "grad_norm": 35.72559300297331,
509
  "learning_rate": 1.7641138321260257e-08,
510
+ "logits/chosen": -1.7627308368682861,
511
+ "logits/rejected": -1.6903340816497803,
512
+ "logps/chosen": -1.573824167251587,
513
+ "logps/rejected": -1.7837245464324951,
514
+ "loss": 0.5391,
515
+ "rewards/accuracies": 0.824999988079071,
516
+ "rewards/chosen": -1.5598773956298828,
517
+ "rewards/margins": 0.729460597038269,
518
+ "rewards/rejected": -2.2893381118774414,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
+ "grad_norm": 35.42036199200566,
524
  "learning_rate": 1.4715029564277793e-08,
525
+ "logits/chosen": -1.8322217464447021,
526
+ "logits/rejected": -1.7827993631362915,
527
+ "logps/chosen": -1.5673482418060303,
528
+ "logps/rejected": -1.7686045169830322,
529
+ "loss": 0.5503,
530
+ "rewards/accuracies": 0.737500011920929,
531
+ "rewards/chosen": -1.4507509469985962,
532
+ "rewards/margins": 0.6567715406417847,
533
+ "rewards/rejected": -2.107522487640381,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
+ "grad_norm": 44.416038802172494,
539
  "learning_rate": 1.2012003751113343e-08,
540
+ "logits/chosen": -1.8188507556915283,
541
+ "logits/rejected": -1.7558269500732422,
542
+ "logps/chosen": -1.5964884757995605,
543
+ "logps/rejected": -1.8198480606079102,
544
+ "loss": 0.5269,
545
+ "rewards/accuracies": 0.737500011920929,
546
+ "rewards/chosen": -1.737273931503296,
547
+ "rewards/margins": 0.6926761269569397,
548
+ "rewards/rejected": -2.429949998855591,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
+ "grad_norm": 39.74205139024732,
554
  "learning_rate": 9.549150281252633e-09,
555
+ "logits/chosen": -1.779714822769165,
556
+ "logits/rejected": -1.7280519008636475,
557
+ "logps/chosen": -1.5766112804412842,
558
+ "logps/rejected": -1.7840029001235962,
559
+ "loss": 0.5288,
560
+ "rewards/accuracies": 0.6875,
561
+ "rewards/chosen": -1.6179043054580688,
562
+ "rewards/margins": 0.6226440072059631,
563
+ "rewards/rejected": -2.2405483722686768,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
+ "grad_norm": 36.101889257586436,
569
  "learning_rate": 7.3420401072985306e-09,
570
+ "logits/chosen": -1.8120644092559814,
571
+ "logits/rejected": -1.7584421634674072,
572
+ "logps/chosen": -1.6255983114242554,
573
+ "logps/rejected": -1.8503506183624268,
574
+ "loss": 0.5301,
575
+ "rewards/accuracies": 0.731249988079071,
576
+ "rewards/chosen": -1.6439390182495117,
577
+ "rewards/margins": 0.6691535711288452,
578
+ "rewards/rejected": -2.3130927085876465,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
+ "grad_norm": 43.87255026178299,
584
  "learning_rate": 5.404627290395369e-09,
585
+ "logits/chosen": -1.7784290313720703,
586
+ "logits/rejected": -1.710515022277832,
587
+ "logps/chosen": -1.5699481964111328,
588
+ "logps/rejected": -1.777539610862732,
589
+ "loss": 0.5325,
590
+ "rewards/accuracies": 0.737500011920929,
591
+ "rewards/chosen": -1.5213696956634521,
592
+ "rewards/margins": 0.6703809499740601,
593
+ "rewards/rejected": -2.1917505264282227,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
+ "grad_norm": 61.61237275525204,
599
  "learning_rate": 3.74916077816162e-09,
600
+ "logits/chosen": -1.7858253717422485,
601
+ "logits/rejected": -1.7235454320907593,
602
+ "logps/chosen": -1.5679755210876465,
603
+ "logps/rejected": -1.7219339609146118,
604
+ "loss": 0.547,
605
+ "rewards/accuracies": 0.706250011920929,
606
+ "rewards/chosen": -1.6727231740951538,
607
+ "rewards/margins": 0.5513177514076233,
608
+ "rewards/rejected": -2.2240407466888428,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
+ "grad_norm": 40.65037100090872,
614
  "learning_rate": 2.386106962899165e-09,
615
+ "logits/chosen": -1.7104368209838867,
616
+ "logits/rejected": -1.6373924016952515,
617
+ "logps/chosen": -1.6611610651016235,
618
+ "logps/rejected": -1.858628511428833,
619
+ "loss": 0.5418,
620
+ "rewards/accuracies": 0.7250000238418579,
621
+ "rewards/chosen": -1.738960862159729,
622
+ "rewards/margins": 0.5645102858543396,
623
+ "rewards/rejected": -2.303471088409424,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
+ "grad_norm": 41.948467770849064,
629
  "learning_rate": 1.3240835096913706e-09,
630
+ "logits/chosen": -1.7456386089324951,
631
+ "logits/rejected": -1.642713189125061,
632
+ "logps/chosen": -1.5583795309066772,
633
+ "logps/rejected": -1.812648057937622,
634
+ "loss": 0.5483,
635
+ "rewards/accuracies": 0.737500011920929,
636
+ "rewards/chosen": -1.5473148822784424,
637
+ "rewards/margins": 0.7394167184829712,
638
+ "rewards/rejected": -2.286731481552124,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
+ "grad_norm": 31.441343774038174,
644
  "learning_rate": 5.698048727497462e-10,
645
+ "logits/chosen": -1.7712395191192627,
646
+ "logits/rejected": -1.6984872817993164,
647
+ "logps/chosen": -1.6279726028442383,
648
+ "logps/rejected": -1.870309829711914,
649
+ "loss": 0.536,
650
+ "rewards/accuracies": 0.75,
651
+ "rewards/chosen": -1.7466990947723389,
652
+ "rewards/margins": 0.7015209794044495,
653
+ "rewards/rejected": -2.4482200145721436,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
+ "grad_norm": 37.032798704254134,
659
  "learning_rate": 1.2803984447259387e-10,
660
+ "logits/chosen": -1.781049370765686,
661
+ "logits/rejected": -1.7250728607177734,
662
+ "logps/chosen": -1.6147279739379883,
663
+ "logps/rejected": -1.8315696716308594,
664
+ "loss": 0.5305,
665
+ "rewards/accuracies": 0.7250000238418579,
666
+ "rewards/chosen": -1.7295538187026978,
667
+ "rewards/margins": 0.6767560243606567,
668
+ "rewards/rejected": -2.4063098430633545,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
+ "train_loss": 0.5750866080199396,
676
+ "train_runtime": 6876.7322,
677
+ "train_samples_per_second": 8.178,
678
  "train_steps_per_second": 0.064
679
  }
680
  ],
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:76d3ce8b0471f040e83e2648b4735a94a60a4b73d104bca0e7423ecb20433836
3
  size 6328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ea6f16cd7184efb49ed5d5f74942d37c251ac7b2aaab04f85fcfa51357639f4
3
  size 6328