RikkiXu commited on
Commit
72a9f09
1 Parent(s): 44a6132

Model save

Browse files
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.5566485422347293,
4
- "train_runtime": 6841.2124,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 8.22,
7
  "train_steps_per_second": 0.064
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 6.969558794025951,
4
+ "train_runtime": 6844.0218,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.217,
7
  "train_steps_per_second": 0.064
8
  }
config.json CHANGED
@@ -20,7 +20,7 @@
20
  "sliding_window": 4096,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
23
- "transformers_version": "4.41.1",
24
  "use_cache": false,
25
  "vocab_size": 32000
26
  }
 
20
  "sliding_window": 4096,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
23
+ "transformers_version": "4.39.3",
24
  "use_cache": false,
25
  "vocab_size": 32000
26
  }
model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea0d09d1476d8a49fe285bd946375e1cbdcb6b821865850265935b691a3e4643
3
  size 4943162336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e640ffc82529b75d69ac2fef13cf2d117f16eae6ce4736d73db1f1e86487b78a
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:3cb1982972159f5dbb8063c84e14c867f92ca23eda59ef58fee5ea9fcd73336c
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:297a6b580e9752eb5586014e574122b6431ca9c78e7aca1abc6337824516760d
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:d179a041fb683849f3965d5a376a01e933a9c09cf0615f02ac2d59791e92b180
3
  size 4540516344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe3700b7100ceae0088757ea4c673b8e67e5727f459a539a1dd6471ce337ee14
3
  size 4540516344
runs/Jul02_16-21-36_n136-129-074/events.out.tfevents.1719908518.n136-129-074.2331461.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ac24fcce5a4056e645486d804188b2a9cfa8d8adb16182fe4432538ebf41644d
3
- size 33009
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48799e24e4812fdb83269c2038d2ff01500a150f5083c1b5910eb65ae5df979e
3
+ size 35427
tokenizer.json CHANGED
@@ -134,7 +134,6 @@
134
  "end_of_word_suffix": null,
135
  "fuse_unk": true,
136
  "byte_fallback": true,
137
- "ignore_merges": false,
138
  "vocab": {
139
  "<unk>": 0,
140
  "<s>": 1,
 
134
  "end_of_word_suffix": null,
135
  "fuse_unk": true,
136
  "byte_fallback": true,
 
137
  "vocab": {
138
  "<unk>": 0,
139
  "<s>": 1,
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.5566485422347293,
4
- "train_runtime": 6841.2124,
5
  "train_samples": 56236,
6
- "train_samples_per_second": 8.22,
7
  "train_steps_per_second": 0.064
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 6.969558794025951,
4
+ "train_runtime": 6844.0218,
5
  "train_samples": 56236,
6
+ "train_samples_per_second": 8.217,
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": 113.90219393876704,
14
  "learning_rate": 2.2727272727272727e-09,
15
  "logits/chosen": -1.6768856048583984,
16
  "logits/rejected": -1.7259055376052856,
17
  "logps/chosen": -1.2793102264404297,
18
  "logps/rejected": -1.2162058353424072,
19
- "loss": 0.6931,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
@@ -25,656 +25,656 @@
25
  },
26
  {
27
  "epoch": 0.02,
28
- "grad_norm": 129.29173344209107,
29
  "learning_rate": 2.2727272727272725e-08,
30
- "logits/chosen": -1.702974796295166,
31
- "logits/rejected": -1.6685913801193237,
32
- "logps/chosen": -1.213309645652771,
33
- "logps/rejected": -1.2203179597854614,
34
- "loss": 0.6934,
35
- "rewards/accuracies": 0.4375,
36
- "rewards/chosen": -4.0181395888794214e-05,
37
- "rewards/margins": -0.004325871355831623,
38
- "rewards/rejected": 0.0042856899090111256,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
- "grad_norm": 129.013817774111,
44
  "learning_rate": 4.545454545454545e-08,
45
- "logits/chosen": -1.7797605991363525,
46
- "logits/rejected": -1.7345327138900757,
47
- "logps/chosen": -1.144714593887329,
48
- "logps/rejected": -1.1853687763214111,
49
- "loss": 0.6906,
50
- "rewards/accuracies": 0.5687500238418579,
51
- "rewards/chosen": -0.014920851215720177,
52
- "rewards/margins": 0.0041695646941661835,
53
- "rewards/rejected": -0.01909041777253151,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
- "grad_norm": 125.39732746183036,
59
  "learning_rate": 6.818181818181817e-08,
60
- "logits/chosen": -1.7443883419036865,
61
- "logits/rejected": -1.6757313013076782,
62
- "logps/chosen": -1.1939805746078491,
63
- "logps/rejected": -1.2470417022705078,
64
- "loss": 0.6755,
65
- "rewards/accuracies": 0.6812499761581421,
66
- "rewards/chosen": -0.11079101264476776,
67
- "rewards/margins": 0.058313049376010895,
68
- "rewards/rejected": -0.16910405457019806,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
- "grad_norm": 99.57108978782348,
74
  "learning_rate": 9.09090909090909e-08,
75
- "logits/chosen": -1.731239914894104,
76
- "logits/rejected": -1.6651241779327393,
77
- "logps/chosen": -1.2352855205535889,
78
- "logps/rejected": -1.3083709478378296,
79
- "loss": 0.6603,
80
- "rewards/accuracies": 0.637499988079071,
81
- "rewards/chosen": -0.2833693027496338,
82
- "rewards/margins": 0.20761942863464355,
83
- "rewards/rejected": -0.49098867177963257,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
- "grad_norm": 94.95534987876165,
89
  "learning_rate": 9.994307990108962e-08,
90
- "logits/chosen": -1.6967604160308838,
91
- "logits/rejected": -1.63302743434906,
92
- "logps/chosen": -1.2550745010375977,
93
- "logps/rejected": -1.304324984550476,
94
- "loss": 0.6261,
95
- "rewards/accuracies": 0.668749988079071,
96
- "rewards/chosen": -0.4698738157749176,
97
- "rewards/margins": 0.29158082604408264,
98
- "rewards/rejected": -0.7614546418190002,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
- "grad_norm": 134.17306631598268,
104
  "learning_rate": 9.959570405988094e-08,
105
- "logits/chosen": -1.7350832223892212,
106
- "logits/rejected": -1.6556373834609985,
107
- "logps/chosen": -1.1744005680084229,
108
- "logps/rejected": -1.2519078254699707,
109
- "loss": 0.6292,
110
- "rewards/accuracies": 0.706250011920929,
111
- "rewards/chosen": -0.6530232429504395,
112
- "rewards/margins": 0.25578898191452026,
113
- "rewards/rejected": -0.9088121652603149,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
- "grad_norm": 82.25189789646025,
119
  "learning_rate": 9.893476820924666e-08,
120
- "logits/chosen": -1.8206403255462646,
121
- "logits/rejected": -1.7381073236465454,
122
- "logps/chosen": -1.2853193283081055,
123
- "logps/rejected": -1.3663524389266968,
124
- "loss": 0.5893,
125
- "rewards/accuracies": 0.6937500238418579,
126
- "rewards/chosen": -0.7247735261917114,
127
- "rewards/margins": 0.34368234872817993,
128
- "rewards/rejected": -1.0684559345245361,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
- "grad_norm": 111.68081612756365,
134
  "learning_rate": 9.796445099843647e-08,
135
- "logits/chosen": -1.8098180294036865,
136
- "logits/rejected": -1.7271239757537842,
137
- "logps/chosen": -1.2911165952682495,
138
- "logps/rejected": -1.3880374431610107,
139
- "loss": 0.6284,
140
- "rewards/accuracies": 0.6625000238418579,
141
- "rewards/chosen": -0.8414822816848755,
142
- "rewards/margins": 0.3639078140258789,
143
- "rewards/rejected": -1.2053899765014648,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
- "grad_norm": 98.75254486198716,
149
  "learning_rate": 9.669088708527066e-08,
150
- "logits/chosen": -1.7565476894378662,
151
- "logits/rejected": -1.6922813653945923,
152
- "logps/chosen": -1.3374165296554565,
153
- "logps/rejected": -1.3871676921844482,
154
- "loss": 0.5903,
155
- "rewards/accuracies": 0.699999988079071,
156
- "rewards/chosen": -1.0171922445297241,
157
- "rewards/margins": 0.42706432938575745,
158
- "rewards/rejected": -1.4442565441131592,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
- "grad_norm": 110.42472659036405,
164
  "learning_rate": 9.512212835085849e-08,
165
- "logits/chosen": -1.7919187545776367,
166
- "logits/rejected": -1.7078930139541626,
167
- "logps/chosen": -1.284099817276001,
168
- "logps/rejected": -1.3786065578460693,
169
- "loss": 0.5835,
170
- "rewards/accuracies": 0.7124999761581421,
171
- "rewards/chosen": -1.1564443111419678,
172
- "rewards/margins": 0.5179702639579773,
173
- "rewards/rejected": -1.6744146347045898,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
- "grad_norm": 94.2164442304763,
179
  "learning_rate": 9.326809299301306e-08,
180
- "logits/chosen": -1.7935024499893188,
181
- "logits/rejected": -1.6928462982177734,
182
- "logps/chosen": -1.3194783926010132,
183
- "logps/rejected": -1.4348446130752563,
184
- "loss": 0.5774,
185
- "rewards/accuracies": 0.71875,
186
- "rewards/chosen": -1.188363790512085,
187
- "rewards/margins": 0.63763028383255,
188
- "rewards/rejected": -1.8259938955307007,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
- "grad_norm": 124.23898564185464,
194
  "learning_rate": 9.114050282021158e-08,
195
- "logits/chosen": -1.7868391275405884,
196
- "logits/rejected": -1.7275841236114502,
197
- "logps/chosen": -1.271259069442749,
198
- "logps/rejected": -1.3735918998718262,
199
- "loss": 0.563,
200
- "rewards/accuracies": 0.75,
201
- "rewards/chosen": -1.2369645833969116,
202
- "rewards/margins": 0.5494047403335571,
203
- "rewards/rejected": -1.7863690853118896,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
- "grad_norm": 94.58029537839892,
209
  "learning_rate": 8.875280914254802e-08,
210
- "logits/chosen": -1.7857173681259155,
211
- "logits/rejected": -1.6955511569976807,
212
- "logps/chosen": -1.3196834325790405,
213
- "logps/rejected": -1.4175035953521729,
214
- "loss": 0.5654,
215
- "rewards/accuracies": 0.71875,
216
- "rewards/chosen": -1.2685333490371704,
217
- "rewards/margins": 0.6939533948898315,
218
- "rewards/rejected": -1.9624868631362915,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
- "grad_norm": 78.26072920533862,
224
  "learning_rate": 8.612010772821971e-08,
225
- "logits/chosen": -1.8125206232070923,
226
- "logits/rejected": -1.767704725265503,
227
- "logps/chosen": -1.3238108158111572,
228
- "logps/rejected": -1.362679123878479,
229
- "loss": 0.5665,
230
- "rewards/accuracies": 0.762499988079071,
231
- "rewards/chosen": -1.0993187427520752,
232
- "rewards/margins": 0.6377816200256348,
233
- "rewards/rejected": -1.73710036277771,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
- "grad_norm": 79.54962530262117,
239
  "learning_rate": 8.325904336322055e-08,
240
- "logits/chosen": -1.784720778465271,
241
- "logits/rejected": -1.728867530822754,
242
- "logps/chosen": -1.2866287231445312,
243
- "logps/rejected": -1.4028968811035156,
244
- "loss": 0.565,
245
- "rewards/accuracies": 0.731249988079071,
246
- "rewards/chosen": -1.192639708518982,
247
- "rewards/margins": 0.6909436583518982,
248
- "rewards/rejected": -1.883583426475525,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
- "grad_norm": 81.2500983198601,
254
  "learning_rate": 8.01877046176447e-08,
255
- "logits/chosen": -1.7332093715667725,
256
- "logits/rejected": -1.6599019765853882,
257
- "logps/chosen": -1.3141100406646729,
258
- "logps/rejected": -1.4209729433059692,
259
- "loss": 0.5662,
260
- "rewards/accuracies": 0.706250011920929,
261
- "rewards/chosen": -1.358100175857544,
262
- "rewards/margins": 0.5547691583633423,
263
- "rewards/rejected": -1.9128694534301758,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
- "grad_norm": 82.5330575937106,
269
  "learning_rate": 7.692550948392249e-08,
270
- "logits/chosen": -1.7886979579925537,
271
- "logits/rejected": -1.7225795984268188,
272
- "logps/chosen": -1.324988603591919,
273
- "logps/rejected": -1.400198221206665,
274
- "loss": 0.5593,
275
- "rewards/accuracies": 0.731249988079071,
276
- "rewards/chosen": -1.1473289728164673,
277
- "rewards/margins": 0.647649884223938,
278
- "rewards/rejected": -1.7949787378311157,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
- "grad_norm": 90.45090476516394,
284
  "learning_rate": 7.349308261002021e-08,
285
- "logits/chosen": -1.7442381381988525,
286
- "logits/rejected": -1.6850671768188477,
287
- "logps/chosen": -1.318302869796753,
288
- "logps/rejected": -1.4245436191558838,
289
- "loss": 0.5533,
290
- "rewards/accuracies": 0.7437499761581421,
291
- "rewards/chosen": -1.2121667861938477,
292
- "rewards/margins": 0.5659047961235046,
293
- "rewards/rejected": -1.778071641921997,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
- "grad_norm": 84.71027965029923,
299
  "learning_rate": 6.991212490377531e-08,
300
- "logits/chosen": -1.7995620965957642,
301
- "logits/rejected": -1.745843529701233,
302
- "logps/chosen": -1.3425184488296509,
303
- "logps/rejected": -1.4334449768066406,
304
- "loss": 0.5291,
305
- "rewards/accuracies": 0.7562500238418579,
306
- "rewards/chosen": -1.1241543292999268,
307
- "rewards/margins": 0.6938456296920776,
308
- "rewards/rejected": -1.818000078201294,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
- "grad_norm": 104.41708218825802,
314
  "learning_rate": 6.620527633276978e-08,
315
- "logits/chosen": -1.7381435632705688,
316
- "logits/rejected": -1.662602424621582,
317
- "logps/chosen": -1.3073325157165527,
318
- "logps/rejected": -1.4739927053451538,
319
- "loss": 0.5404,
320
- "rewards/accuracies": 0.762499988079071,
321
- "rewards/chosen": -1.1287935972213745,
322
- "rewards/margins": 0.7987143993377686,
323
- "rewards/rejected": -1.927507758140564,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
- "grad_norm": 98.12458997212138,
329
  "learning_rate": 6.239597278716581e-08,
330
- "logits/chosen": -1.808566689491272,
331
- "logits/rejected": -1.7453174591064453,
332
- "logps/chosen": -1.3598778247833252,
333
- "logps/rejected": -1.4206109046936035,
334
- "loss": 0.5355,
335
- "rewards/accuracies": 0.762499988079071,
336
- "rewards/chosen": -1.1454766988754272,
337
- "rewards/margins": 0.813967227935791,
338
- "rewards/rejected": -1.9594440460205078,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
- "grad_norm": 73.95327688888813,
344
  "learning_rate": 5.8508297910462456e-08,
345
- "logits/chosen": -1.7581428289413452,
346
- "logits/rejected": -1.6758205890655518,
347
- "logps/chosen": -1.2785987854003906,
348
- "logps/rejected": -1.4212281703948975,
349
- "loss": 0.5267,
350
- "rewards/accuracies": 0.793749988079071,
351
- "rewards/chosen": -1.1508338451385498,
352
- "rewards/margins": 0.8485604524612427,
353
- "rewards/rejected": -1.999394178390503,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
- "grad_norm": 101.92396024664481,
359
  "learning_rate": 5.456683083494731e-08,
360
- "logits/chosen": -1.7508561611175537,
361
- "logits/rejected": -1.7111085653305054,
362
- "logps/chosen": -1.2999995946884155,
363
- "logps/rejected": -1.3745485544204712,
364
- "loss": 0.554,
365
- "rewards/accuracies": 0.6937500238418579,
366
- "rewards/chosen": -1.0619090795516968,
367
- "rewards/margins": 0.5080693364143372,
368
- "rewards/rejected": -1.5699783563613892,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
- "grad_norm": 110.90195766775575,
374
  "learning_rate": 5.059649078450834e-08,
375
- "logits/chosen": -1.745862603187561,
376
- "logits/rejected": -1.69782292842865,
377
- "logps/chosen": -1.2861733436584473,
378
- "logps/rejected": -1.4080326557159424,
379
- "loss": 0.5271,
380
- "rewards/accuracies": 0.7437499761581421,
381
- "rewards/chosen": -1.1781983375549316,
382
- "rewards/margins": 0.5982070565223694,
383
- "rewards/rejected": -1.7764055728912354,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
- "grad_norm": 90.78412484714156,
389
  "learning_rate": 4.6622379527277186e-08,
390
- "logits/chosen": -1.7498047351837158,
391
- "logits/rejected": -1.6966612339019775,
392
- "logps/chosen": -1.263024091720581,
393
- "logps/rejected": -1.3622833490371704,
394
- "loss": 0.5322,
395
- "rewards/accuracies": 0.699999988079071,
396
- "rewards/chosen": -1.241353988647461,
397
- "rewards/margins": 0.6617964506149292,
398
- "rewards/rejected": -1.9031503200531006,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
- "grad_norm": 80.47587413810997,
404
  "learning_rate": 4.26696226741691e-08,
405
- "logits/chosen": -1.766274094581604,
406
- "logits/rejected": -1.6926915645599365,
407
- "logps/chosen": -1.3750711679458618,
408
- "logps/rejected": -1.4678473472595215,
409
- "loss": 0.5309,
410
- "rewards/accuracies": 0.699999988079071,
411
- "rewards/chosen": -1.4224761724472046,
412
- "rewards/margins": 0.7280942797660828,
413
- "rewards/rejected": -2.1505703926086426,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
- "grad_norm": 81.2536856430445,
419
  "learning_rate": 3.876321082668098e-08,
420
- "logits/chosen": -1.8292741775512695,
421
- "logits/rejected": -1.7543790340423584,
422
- "logps/chosen": -1.3468070030212402,
423
- "logps/rejected": -1.4356333017349243,
424
- "loss": 0.5228,
425
  "rewards/accuracies": 0.78125,
426
- "rewards/chosen": -1.0831630229949951,
427
- "rewards/margins": 0.7951737642288208,
428
- "rewards/rejected": -1.8783369064331055,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
- "grad_norm": 87.95396768106268,
434
  "learning_rate": 3.492784157826244e-08,
435
- "logits/chosen": -1.754434585571289,
436
- "logits/rejected": -1.654488205909729,
437
- "logps/chosen": -1.331969141960144,
438
- "logps/rejected": -1.4491215944290161,
439
- "loss": 0.5257,
440
- "rewards/accuracies": 0.7124999761581421,
441
- "rewards/chosen": -1.0909301042556763,
442
- "rewards/margins": 0.8176229596138,
443
- "rewards/rejected": -1.908552885055542,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
- "grad_norm": 75.75936548706386,
449
  "learning_rate": 3.118776336817812e-08,
450
- "logits/chosen": -1.800217628479004,
451
- "logits/rejected": -1.7326066493988037,
452
- "logps/chosen": -1.3206775188446045,
453
- "logps/rejected": -1.4107261896133423,
454
- "loss": 0.5236,
455
  "rewards/accuracies": 0.7562500238418579,
456
- "rewards/chosen": -1.1742886304855347,
457
- "rewards/margins": 0.8283244967460632,
458
- "rewards/rejected": -2.002613067626953,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
- "grad_norm": 86.61923660491236,
464
  "learning_rate": 2.7566622175067443e-08,
465
- "logits/chosen": -1.781106948852539,
466
- "logits/rejected": -1.7118265628814697,
467
- "logps/chosen": -1.3446478843688965,
468
- "logps/rejected": -1.4952938556671143,
469
- "loss": 0.5304,
470
- "rewards/accuracies": 0.7250000238418579,
471
- "rewards/chosen": -1.3606449365615845,
472
- "rewards/margins": 0.7265914082527161,
473
- "rewards/rejected": -2.0872364044189453,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
- "grad_norm": 104.45230542444779,
479
  "learning_rate": 2.408731201945432e-08,
480
- "logits/chosen": -1.7766481637954712,
481
- "logits/rejected": -1.7281579971313477,
482
- "logps/chosen": -1.3146270513534546,
483
- "logps/rejected": -1.4018762111663818,
484
- "loss": 0.5176,
485
  "rewards/accuracies": 0.768750011920929,
486
- "rewards/chosen": -1.2073173522949219,
487
- "rewards/margins": 0.670944333076477,
488
- "rewards/rejected": -1.8782615661621094,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
- "grad_norm": 97.89401056240122,
494
  "learning_rate": 2.0771830220378112e-08,
495
- "logits/chosen": -1.7322553396224976,
496
- "logits/rejected": -1.6741504669189453,
497
- "logps/chosen": -1.347646951675415,
498
- "logps/rejected": -1.4063564538955688,
499
- "loss": 0.5268,
500
- "rewards/accuracies": 0.737500011920929,
501
- "rewards/chosen": -1.2476258277893066,
502
- "rewards/margins": 0.6655985713005066,
503
- "rewards/rejected": -1.913224458694458,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
- "grad_norm": 79.38110491324964,
509
  "learning_rate": 1.7641138321260257e-08,
510
- "logits/chosen": -1.7678320407867432,
511
- "logits/rejected": -1.692814826965332,
512
- "logps/chosen": -1.3020561933517456,
513
- "logps/rejected": -1.4196479320526123,
514
- "loss": 0.5181,
515
- "rewards/accuracies": 0.8187500238418579,
516
- "rewards/chosen": -1.1820148229599,
517
- "rewards/margins": 0.900565505027771,
518
- "rewards/rejected": -2.082580327987671,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
- "grad_norm": 79.242696260741,
524
  "learning_rate": 1.4715029564277793e-08,
525
- "logits/chosen": -1.8388326168060303,
526
- "logits/rejected": -1.7862517833709717,
527
- "logps/chosen": -1.306144118309021,
528
- "logps/rejected": -1.4227768182754517,
529
- "loss": 0.5342,
530
  "rewards/accuracies": 0.75,
531
- "rewards/chosen": -1.0148370265960693,
532
- "rewards/margins": 0.7956900596618652,
533
- "rewards/rejected": -1.8105270862579346,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
- "grad_norm": 86.49780334887537,
539
  "learning_rate": 1.2012003751113343e-08,
540
- "logits/chosen": -1.822198510169983,
541
- "logits/rejected": -1.756326675415039,
542
- "logps/chosen": -1.2951385974884033,
543
- "logps/rejected": -1.4268760681152344,
544
- "loss": 0.5045,
545
  "rewards/accuracies": 0.75,
546
- "rewards/chosen": -1.329688310623169,
547
- "rewards/margins": 0.815467357635498,
548
- "rewards/rejected": -2.145155668258667,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
- "grad_norm": 82.9744571713248,
554
  "learning_rate": 9.549150281252633e-09,
555
- "logits/chosen": -1.7854712009429932,
556
- "logits/rejected": -1.7307994365692139,
557
- "logps/chosen": -1.2910486459732056,
558
- "logps/rejected": -1.4215586185455322,
559
- "loss": 0.508,
560
- "rewards/accuracies": 0.737500011920929,
561
- "rewards/chosen": -1.1891343593597412,
562
- "rewards/margins": 0.7877954244613647,
563
- "rewards/rejected": -1.9769299030303955,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
- "grad_norm": 77.94995731245406,
569
  "learning_rate": 7.3420401072985306e-09,
570
- "logits/chosen": -1.8163235187530518,
571
- "logits/rejected": -1.7626298666000366,
572
- "logps/chosen": -1.3424322605133057,
573
- "logps/rejected": -1.483293890953064,
574
- "loss": 0.5133,
575
- "rewards/accuracies": 0.737500011920929,
576
- "rewards/chosen": -1.2781879901885986,
577
- "rewards/margins": 0.8339790105819702,
578
- "rewards/rejected": -2.1121671199798584,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
- "grad_norm": 99.08582811249858,
584
  "learning_rate": 5.404627290395369e-09,
585
- "logits/chosen": -1.781881332397461,
586
- "logits/rejected": -1.712882399559021,
587
- "logps/chosen": -1.3003751039505005,
588
- "logps/rejected": -1.4186222553253174,
589
- "loss": 0.5136,
590
- "rewards/accuracies": 0.737500011920929,
591
- "rewards/chosen": -1.1076927185058594,
592
- "rewards/margins": 0.7825103998184204,
593
- "rewards/rejected": -1.8902031183242798,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
- "grad_norm": 142.606766968544,
599
  "learning_rate": 3.74916077816162e-09,
600
- "logits/chosen": -1.7914397716522217,
601
- "logits/rejected": -1.7269777059555054,
602
- "logps/chosen": -1.2829267978668213,
603
- "logps/rejected": -1.3663567304611206,
604
- "loss": 0.5283,
605
  "rewards/accuracies": 0.706250011920929,
606
- "rewards/chosen": -1.3313188552856445,
607
- "rewards/margins": 0.6730116605758667,
608
- "rewards/rejected": -2.0043303966522217,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
- "grad_norm": 87.75942228209004,
614
  "learning_rate": 2.386106962899165e-09,
615
- "logits/chosen": -1.7152767181396484,
616
- "logits/rejected": -1.6405121088027954,
617
- "logps/chosen": -1.3587360382080078,
618
- "logps/rejected": -1.4870920181274414,
619
- "loss": 0.523,
620
- "rewards/accuracies": 0.7124999761581421,
621
- "rewards/chosen": -1.3231509923934937,
622
- "rewards/margins": 0.7201642394065857,
623
- "rewards/rejected": -2.0433154106140137,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
- "grad_norm": 95.93932206444032,
629
  "learning_rate": 1.3240835096913706e-09,
630
- "logits/chosen": -1.7500959634780884,
631
- "logits/rejected": -1.6428474187850952,
632
- "logps/chosen": -1.280632734298706,
633
- "logps/rejected": -1.440298080444336,
634
- "loss": 0.5334,
635
- "rewards/accuracies": 0.7124999761581421,
636
- "rewards/chosen": -1.0908193588256836,
637
- "rewards/margins": 0.9025093913078308,
638
- "rewards/rejected": -1.9933288097381592,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
- "grad_norm": 185.3215699605587,
644
  "learning_rate": 5.698048727497462e-10,
645
- "logits/chosen": -1.7732937335968018,
646
- "logits/rejected": -1.6997044086456299,
647
- "logps/chosen": -1.3186805248260498,
648
- "logps/rejected": -1.473042607307434,
649
- "loss": 0.5207,
650
- "rewards/accuracies": 0.7875000238418579,
651
- "rewards/chosen": -1.2738254070281982,
652
- "rewards/margins": 0.8740528225898743,
653
- "rewards/rejected": -2.1478781700134277,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
- "grad_norm": 85.90784633584286,
659
  "learning_rate": 1.2803984447259387e-10,
660
- "logits/chosen": -1.7828824520111084,
661
- "logits/rejected": -1.7222639322280884,
662
- "logps/chosen": -1.310931921005249,
663
- "logps/rejected": -1.4457513093948364,
664
- "loss": 0.5116,
665
- "rewards/accuracies": 0.71875,
666
- "rewards/chosen": -1.2859268188476562,
667
- "rewards/margins": 0.8716639280319214,
668
- "rewards/rejected": -2.157590866088867,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
- "train_loss": 0.5566485422347293,
676
- "train_runtime": 6841.2124,
677
- "train_samples_per_second": 8.22,
678
  "train_steps_per_second": 0.064
679
  }
680
  ],
 
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
+ "grad_norm": 2276.7198889454276,
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": 10.0,
20
  "rewards/accuracies": 0.0,
21
  "rewards/chosen": 0.0,
22
  "rewards/margins": 0.0,
 
25
  },
26
  {
27
  "epoch": 0.02,
28
+ "grad_norm": 2572.843914099358,
29
  "learning_rate": 2.2727272727272725e-08,
30
+ "logits/chosen": -1.7036548852920532,
31
+ "logits/rejected": -1.668819785118103,
32
+ "logps/chosen": -1.2131812572479248,
33
+ "logps/rejected": -1.2204509973526,
34
+ "loss": 9.9973,
35
+ "rewards/accuracies": 0.4583333432674408,
36
+ "rewards/chosen": 0.01242614071816206,
37
+ "rewards/margins": -0.017121784389019012,
38
+ "rewards/rejected": 0.029547926038503647,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.05,
43
+ "grad_norm": 2655.4615274085163,
44
  "learning_rate": 4.545454545454545e-08,
45
+ "logits/chosen": -1.7792682647705078,
46
+ "logits/rejected": -1.7343353033065796,
47
+ "logps/chosen": -1.144930124282837,
48
+ "logps/rejected": -1.1850289106369019,
49
+ "loss": 9.9319,
50
+ "rewards/accuracies": 0.5062500238418579,
51
+ "rewards/chosen": -0.1707584410905838,
52
+ "rewards/margins": -0.013824567198753357,
53
+ "rewards/rejected": -0.15693387389183044,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.07,
58
+ "grad_norm": 2519.456913863188,
59
  "learning_rate": 6.818181818181817e-08,
60
+ "logits/chosen": -1.7445284128189087,
61
+ "logits/rejected": -1.6756689548492432,
62
+ "logps/chosen": -1.195391297340393,
63
+ "logps/rejected": -1.2486639022827148,
64
+ "loss": 9.495,
65
+ "rewards/accuracies": 0.6312500238418579,
66
+ "rewards/chosen": -1.2489749193191528,
67
+ "rewards/margins": 0.6042808890342712,
68
+ "rewards/rejected": -1.8532556295394897,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.09,
73
+ "grad_norm": 1627.869026136585,
74
  "learning_rate": 9.09090909090909e-08,
75
+ "logits/chosen": -1.7305688858032227,
76
+ "logits/rejected": -1.664052963256836,
77
+ "logps/chosen": -1.247615098953247,
78
+ "logps/rejected": -1.3294904232025146,
79
+ "loss": 9.009,
80
+ "rewards/accuracies": 0.643750011920929,
81
+ "rewards/chosen": -4.066654205322266,
82
+ "rewards/margins": 2.955174446105957,
83
+ "rewards/rejected": -7.021829128265381,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.11,
88
+ "grad_norm": 1748.050155291869,
89
  "learning_rate": 9.994307990108962e-08,
90
+ "logits/chosen": -1.6962864398956299,
91
+ "logits/rejected": -1.6318918466567993,
92
+ "logps/chosen": -1.2720497846603394,
93
+ "logps/rejected": -1.3243694305419922,
94
+ "loss": 8.2497,
95
+ "rewards/accuracies": 0.6937500238418579,
96
+ "rewards/chosen": -6.396273612976074,
97
+ "rewards/margins": 3.222705125808716,
98
+ "rewards/rejected": -9.618978500366211,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.14,
103
+ "grad_norm": 2112.072704134999,
104
  "learning_rate": 9.959570405988094e-08,
105
+ "logits/chosen": -1.7340772151947021,
106
+ "logits/rejected": -1.6545569896697998,
107
+ "logps/chosen": -1.1886461973190308,
108
+ "logps/rejected": -1.2660636901855469,
109
+ "loss": 8.3676,
110
+ "rewards/accuracies": 0.7124999761581421,
111
+ "rewards/chosen": -7.954795837402344,
112
+ "rewards/margins": 2.5489163398742676,
113
+ "rewards/rejected": -10.50371265411377,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.16,
118
+ "grad_norm": 1596.1626784723217,
119
  "learning_rate": 9.893476820924666e-08,
120
+ "logits/chosen": -1.8238868713378906,
121
+ "logits/rejected": -1.740943193435669,
122
+ "logps/chosen": -1.2993048429489136,
123
+ "logps/rejected": -1.3821274042129517,
124
+ "loss": 7.5786,
125
+ "rewards/accuracies": 0.6875,
126
+ "rewards/chosen": -8.646288871765137,
127
+ "rewards/margins": 3.6157679557800293,
128
+ "rewards/rejected": -12.262056350708008,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.18,
133
+ "grad_norm": 1725.7540930385703,
134
  "learning_rate": 9.796445099843647e-08,
135
+ "logits/chosen": -1.8160178661346436,
136
+ "logits/rejected": -1.7339365482330322,
137
+ "logps/chosen": -1.3012406826019287,
138
+ "logps/rejected": -1.4006680250167847,
139
+ "loss": 8.0893,
140
+ "rewards/accuracies": 0.643750011920929,
141
+ "rewards/chosen": -9.427220344543457,
142
+ "rewards/margins": 3.889753818511963,
143
+ "rewards/rejected": -13.316973686218262,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.2,
148
+ "grad_norm": 1643.8851213987334,
149
  "learning_rate": 9.669088708527066e-08,
150
+ "logits/chosen": -1.7639058828353882,
151
+ "logits/rejected": -1.7009646892547607,
152
+ "logps/chosen": -1.3530588150024414,
153
+ "logps/rejected": -1.4040849208831787,
154
+ "loss": 7.3762,
155
+ "rewards/accuracies": 0.6937500238418579,
156
+ "rewards/chosen": -11.736152648925781,
157
+ "rewards/margins": 4.398137092590332,
158
+ "rewards/rejected": -16.13429069519043,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.23,
163
+ "grad_norm": 1928.7061081884092,
164
  "learning_rate": 9.512212835085849e-08,
165
+ "logits/chosen": -1.799574613571167,
166
+ "logits/rejected": -1.715815782546997,
167
+ "logps/chosen": -1.2974745035171509,
168
+ "logps/rejected": -1.3930069208145142,
169
+ "loss": 7.3866,
170
+ "rewards/accuracies": 0.6937500238418579,
171
+ "rewards/chosen": -12.9019136428833,
172
+ "rewards/margins": 5.282271862030029,
173
+ "rewards/rejected": -18.184185028076172,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.25,
178
+ "grad_norm": 1572.512514314558,
179
  "learning_rate": 9.326809299301306e-08,
180
+ "logits/chosen": -1.7981303930282593,
181
+ "logits/rejected": -1.6978185176849365,
182
+ "logps/chosen": -1.3314235210418701,
183
+ "logps/rejected": -1.4466179609298706,
184
+ "loss": 7.2883,
185
+ "rewards/accuracies": 0.6937500238418579,
186
+ "rewards/chosen": -13.078150749206543,
187
+ "rewards/margins": 6.359132289886475,
188
+ "rewards/rejected": -19.43728256225586,
189
  "step": 110
190
  },
191
  {
192
  "epoch": 0.27,
193
+ "grad_norm": 1543.802209188709,
194
  "learning_rate": 9.114050282021158e-08,
195
+ "logits/chosen": -1.7882074117660522,
196
+ "logits/rejected": -1.7293879985809326,
197
+ "logps/chosen": -1.286062479019165,
198
+ "logps/rejected": -1.391157865524292,
199
+ "loss": 7.1006,
200
+ "rewards/accuracies": 0.768750011920929,
201
+ "rewards/chosen": -13.849987983703613,
202
+ "rewards/margins": 5.770316123962402,
203
+ "rewards/rejected": -19.62030601501465,
204
  "step": 120
205
  },
206
  {
207
  "epoch": 0.3,
208
+ "grad_norm": 1776.0112703946616,
209
  "learning_rate": 8.875280914254802e-08,
210
+ "logits/chosen": -1.7862787246704102,
211
+ "logits/rejected": -1.6958926916122437,
212
+ "logps/chosen": -1.3420274257659912,
213
+ "logps/rejected": -1.442373514175415,
214
+ "loss": 7.0019,
215
+ "rewards/accuracies": 0.7250000238418579,
216
+ "rewards/chosen": -14.9197416305542,
217
+ "rewards/margins": 7.192111015319824,
218
+ "rewards/rejected": -22.111852645874023,
219
  "step": 130
220
  },
221
  {
222
  "epoch": 0.32,
223
+ "grad_norm": 1386.1262451501352,
224
  "learning_rate": 8.612010772821971e-08,
225
+ "logits/chosen": -1.8119525909423828,
226
+ "logits/rejected": -1.7669906616210938,
227
+ "logps/chosen": -1.3379199504852295,
228
+ "logps/rejected": -1.380608320236206,
229
+ "loss": 6.9963,
230
+ "rewards/accuracies": 0.7562500238418579,
231
+ "rewards/chosen": -12.404088020324707,
232
+ "rewards/margins": 6.759835243225098,
233
+ "rewards/rejected": -19.163921356201172,
234
  "step": 140
235
  },
236
  {
237
  "epoch": 0.34,
238
+ "grad_norm": 1421.2461006997341,
239
  "learning_rate": 8.325904336322055e-08,
240
+ "logits/chosen": -1.784933090209961,
241
+ "logits/rejected": -1.7292178869247437,
242
+ "logps/chosen": -1.3000556230545044,
243
+ "logps/rejected": -1.41317617893219,
244
+ "loss": 7.1117,
245
+ "rewards/accuracies": 0.7124999761581421,
246
+ "rewards/chosen": -13.269081115722656,
247
+ "rewards/margins": 6.594681739807129,
248
+ "rewards/rejected": -19.8637638092041,
249
  "step": 150
250
  },
251
  {
252
  "epoch": 0.36,
253
+ "grad_norm": 1409.35455155076,
254
  "learning_rate": 8.01877046176447e-08,
255
+ "logits/chosen": -1.733028769493103,
256
+ "logits/rejected": -1.659033179283142,
257
+ "logps/chosen": -1.3279526233673096,
258
+ "logps/rejected": -1.4333665370941162,
259
+ "loss": 7.0435,
260
+ "rewards/accuracies": 0.6812499761581421,
261
+ "rewards/chosen": -14.965250968933105,
262
+ "rewards/margins": 5.402812957763672,
263
+ "rewards/rejected": -20.368064880371094,
264
  "step": 160
265
  },
266
  {
267
  "epoch": 0.39,
268
+ "grad_norm": 1409.9938243352115,
269
  "learning_rate": 7.692550948392249e-08,
270
+ "logits/chosen": -1.7891366481781006,
271
+ "logits/rejected": -1.7234008312225342,
272
+ "logps/chosen": -1.3368794918060303,
273
+ "logps/rejected": -1.410075068473816,
274
+ "loss": 7.0875,
275
+ "rewards/accuracies": 0.7124999761581421,
276
+ "rewards/chosen": -12.662399291992188,
277
+ "rewards/margins": 6.27509880065918,
278
+ "rewards/rejected": -18.937496185302734,
279
  "step": 170
280
  },
281
  {
282
  "epoch": 0.41,
283
+ "grad_norm": 1576.9612574780165,
284
  "learning_rate": 7.349308261002021e-08,
285
+ "logits/chosen": -1.7433929443359375,
286
+ "logits/rejected": -1.6840171813964844,
287
+ "logps/chosen": -1.3272547721862793,
288
+ "logps/rejected": -1.4322535991668701,
289
+ "loss": 6.8292,
290
+ "rewards/accuracies": 0.75,
291
+ "rewards/chosen": -13.016866683959961,
292
+ "rewards/margins": 5.534828186035156,
293
+ "rewards/rejected": -18.551694869995117,
294
  "step": 180
295
  },
296
  {
297
  "epoch": 0.43,
298
+ "grad_norm": 1468.7451103929668,
299
  "learning_rate": 6.991212490377531e-08,
300
+ "logits/chosen": -1.7984473705291748,
301
+ "logits/rejected": -1.745666742324829,
302
+ "logps/chosen": -1.3522603511810303,
303
+ "logps/rejected": -1.4409388303756714,
304
+ "loss": 6.3646,
305
+ "rewards/accuracies": 0.7437499761581421,
306
+ "rewards/chosen": -12.215718269348145,
307
+ "rewards/margins": 6.713670253753662,
308
+ "rewards/rejected": -18.929386138916016,
309
  "step": 190
310
  },
311
  {
312
  "epoch": 0.46,
313
+ "grad_norm": 1702.6463731612896,
314
  "learning_rate": 6.620527633276978e-08,
315
+ "logits/chosen": -1.7361818552017212,
316
+ "logits/rejected": -1.6605857610702515,
317
+ "logps/chosen": -1.3193273544311523,
318
+ "logps/rejected": -1.4827220439910889,
319
+ "loss": 6.6324,
320
+ "rewards/accuracies": 0.75,
321
+ "rewards/chosen": -12.487411499023438,
322
+ "rewards/margins": 7.660594940185547,
323
+ "rewards/rejected": -20.148006439208984,
324
  "step": 200
325
  },
326
  {
327
  "epoch": 0.48,
328
+ "grad_norm": 1518.6969019476235,
329
  "learning_rate": 6.239597278716581e-08,
330
+ "logits/chosen": -1.8051319122314453,
331
+ "logits/rejected": -1.7425997257232666,
332
+ "logps/chosen": -1.3682256937026978,
333
+ "logps/rejected": -1.4255290031433105,
334
+ "loss": 6.5008,
335
+ "rewards/accuracies": 0.768750011920929,
336
+ "rewards/chosen": -12.289575576782227,
337
+ "rewards/margins": 7.796680450439453,
338
+ "rewards/rejected": -20.08625602722168,
339
  "step": 210
340
  },
341
  {
342
  "epoch": 0.5,
343
+ "grad_norm": 1375.8319371273024,
344
  "learning_rate": 5.8508297910462456e-08,
345
+ "logits/chosen": -1.7532581090927124,
346
+ "logits/rejected": -1.6711580753326416,
347
+ "logps/chosen": -1.285611629486084,
348
+ "logps/rejected": -1.4231507778167725,
349
+ "loss": 6.3973,
350
+ "rewards/accuracies": 0.768750011920929,
351
+ "rewards/chosen": -12.209630012512207,
352
+ "rewards/margins": 7.976588249206543,
353
+ "rewards/rejected": -20.186220169067383,
354
  "step": 220
355
  },
356
  {
357
  "epoch": 0.52,
358
+ "grad_norm": 1612.3674111913235,
359
  "learning_rate": 5.456683083494731e-08,
360
+ "logits/chosen": -1.745671033859253,
361
+ "logits/rejected": -1.7057764530181885,
362
+ "logps/chosen": -1.3081130981445312,
363
+ "logps/rejected": -1.3810980319976807,
364
+ "loss": 6.8051,
365
+ "rewards/accuracies": 0.6875,
366
+ "rewards/chosen": -11.430437088012695,
367
+ "rewards/margins": 4.924307823181152,
368
+ "rewards/rejected": -16.3547420501709,
369
  "step": 230
370
  },
371
  {
372
  "epoch": 0.55,
373
+ "grad_norm": 1819.5547604545825,
374
  "learning_rate": 5.059649078450834e-08,
375
+ "logits/chosen": -1.7406162023544312,
376
+ "logits/rejected": -1.6928155422210693,
377
+ "logps/chosen": -1.2918732166290283,
378
+ "logps/rejected": -1.4103202819824219,
379
+ "loss": 6.3818,
380
+ "rewards/accuracies": 0.7250000238418579,
381
+ "rewards/chosen": -12.351980209350586,
382
+ "rewards/margins": 5.640812873840332,
383
+ "rewards/rejected": -17.992794036865234,
384
  "step": 240
385
  },
386
  {
387
  "epoch": 0.57,
388
+ "grad_norm": 1555.9014610436766,
389
  "learning_rate": 4.6622379527277186e-08,
390
+ "logits/chosen": -1.7455848455429077,
391
+ "logits/rejected": -1.6927179098129272,
392
+ "logps/chosen": -1.266522765159607,
393
+ "logps/rejected": -1.3599469661712646,
394
+ "loss": 6.4606,
395
+ "rewards/accuracies": 0.6875,
396
+ "rewards/chosen": -12.763422012329102,
397
+ "rewards/margins": 6.03444766998291,
398
+ "rewards/rejected": -18.797870635986328,
399
  "step": 250
400
  },
401
  {
402
  "epoch": 0.59,
403
+ "grad_norm": 1440.8481572009775,
404
  "learning_rate": 4.26696226741691e-08,
405
+ "logits/chosen": -1.7621545791625977,
406
+ "logits/rejected": -1.6889396905899048,
407
+ "logps/chosen": -1.3751699924468994,
408
+ "logps/rejected": -1.4615355730056763,
409
+ "loss": 6.5489,
410
+ "rewards/accuracies": 0.6937500238418579,
411
+ "rewards/chosen": -14.234657287597656,
412
+ "rewards/margins": 6.639866828918457,
413
+ "rewards/rejected": -20.874523162841797,
414
  "step": 260
415
  },
416
  {
417
  "epoch": 0.61,
418
+ "grad_norm": 1565.6866426892864,
419
  "learning_rate": 3.876321082668098e-08,
420
+ "logits/chosen": -1.8229167461395264,
421
+ "logits/rejected": -1.7482731342315674,
422
+ "logps/chosen": -1.3493683338165283,
423
+ "logps/rejected": -1.4336580038070679,
424
+ "loss": 6.2912,
425
  "rewards/accuracies": 0.78125,
426
+ "rewards/chosen": -11.087759017944336,
427
+ "rewards/margins": 7.498082637786865,
428
+ "rewards/rejected": -18.585840225219727,
429
  "step": 270
430
  },
431
  {
432
  "epoch": 0.64,
433
+ "grad_norm": 1644.1893084856104,
434
  "learning_rate": 3.492784157826244e-08,
435
+ "logits/chosen": -1.7463124990463257,
436
+ "logits/rejected": -1.6467139720916748,
437
+ "logps/chosen": -1.3356127738952637,
438
+ "logps/rejected": -1.4482067823410034,
439
+ "loss": 6.3638,
440
+ "rewards/accuracies": 0.71875,
441
+ "rewards/chosen": -11.273643493652344,
442
+ "rewards/margins": 7.72042179107666,
443
+ "rewards/rejected": -18.994068145751953,
444
  "step": 280
445
  },
446
  {
447
  "epoch": 0.66,
448
+ "grad_norm": 1246.6661658945,
449
  "learning_rate": 3.118776336817812e-08,
450
+ "logits/chosen": -1.7918853759765625,
451
+ "logits/rejected": -1.7234630584716797,
452
+ "logps/chosen": -1.3221752643585205,
453
+ "logps/rejected": -1.4063078165054321,
454
+ "loss": 6.2091,
455
  "rewards/accuracies": 0.7562500238418579,
456
+ "rewards/chosen": -11.892666816711426,
457
+ "rewards/margins": 7.691611289978027,
458
+ "rewards/rejected": -19.584278106689453,
459
  "step": 290
460
  },
461
  {
462
  "epoch": 0.68,
463
+ "grad_norm": 1502.285401325102,
464
  "learning_rate": 2.7566622175067443e-08,
465
+ "logits/chosen": -1.7708361148834229,
466
+ "logits/rejected": -1.7010380029678345,
467
+ "logps/chosen": -1.349440336227417,
468
+ "logps/rejected": -1.4947826862335205,
469
+ "loss": 6.3971,
470
+ "rewards/accuracies": 0.731249988079071,
471
+ "rewards/chosen": -14.085695266723633,
472
+ "rewards/margins": 6.735556125640869,
473
+ "rewards/rejected": -20.821250915527344,
474
  "step": 300
475
  },
476
  {
477
  "epoch": 0.71,
478
+ "grad_norm": 2172.757810414492,
479
  "learning_rate": 2.408731201945432e-08,
480
+ "logits/chosen": -1.7669150829315186,
481
+ "logits/rejected": -1.7179067134857178,
482
+ "logps/chosen": -1.3212790489196777,
483
+ "logps/rejected": -1.4049441814422607,
484
+ "loss": 6.286,
485
  "rewards/accuracies": 0.768750011920929,
486
+ "rewards/chosen": -12.738364219665527,
487
+ "rewards/margins": 6.351044178009033,
488
+ "rewards/rejected": -19.08941078186035,
489
  "step": 310
490
  },
491
  {
492
  "epoch": 0.73,
493
+ "grad_norm": 1676.044674890287,
494
  "learning_rate": 2.0771830220378112e-08,
495
+ "logits/chosen": -1.7216575145721436,
496
+ "logits/rejected": -1.6640005111694336,
497
+ "logps/chosen": -1.3518606424331665,
498
+ "logps/rejected": -1.4053070545196533,
499
+ "loss": 6.3543,
500
+ "rewards/accuracies": 0.731249988079071,
501
+ "rewards/chosen": -12.897618293762207,
502
+ "rewards/margins": 6.129672050476074,
503
+ "rewards/rejected": -19.02729034423828,
504
  "step": 320
505
  },
506
  {
507
  "epoch": 0.75,
508
+ "grad_norm": 1454.0668443582122,
509
  "learning_rate": 1.7641138321260257e-08,
510
+ "logits/chosen": -1.7589213848114014,
511
+ "logits/rejected": -1.682925820350647,
512
+ "logps/chosen": -1.30251145362854,
513
+ "logps/rejected": -1.4143444299697876,
514
+ "loss": 6.1555,
515
+ "rewards/accuracies": 0.8374999761581421,
516
+ "rewards/chosen": -11.865683555603027,
517
+ "rewards/margins": 8.429769515991211,
518
+ "rewards/rejected": -20.295452117919922,
519
  "step": 330
520
  },
521
  {
522
  "epoch": 0.77,
523
+ "grad_norm": 1332.7467990696193,
524
  "learning_rate": 1.4715029564277793e-08,
525
+ "logits/chosen": -1.8296537399291992,
526
+ "logits/rejected": -1.776614785194397,
527
+ "logps/chosen": -1.3082842826843262,
528
+ "logps/rejected": -1.4200184345245361,
529
+ "loss": 6.5417,
530
  "rewards/accuracies": 0.75,
531
+ "rewards/chosen": -10.362393379211426,
532
+ "rewards/margins": 7.467054843902588,
533
+ "rewards/rejected": -17.82944679260254,
534
  "step": 340
535
  },
536
  {
537
  "epoch": 0.8,
538
+ "grad_norm": 1523.4091955188806,
539
  "learning_rate": 1.2012003751113343e-08,
540
+ "logits/chosen": -1.8123096227645874,
541
+ "logits/rejected": -1.7457765340805054,
542
+ "logps/chosen": -1.2949508428573608,
543
+ "logps/rejected": -1.42155921459198,
544
+ "loss": 5.9983,
545
  "rewards/accuracies": 0.75,
546
+ "rewards/chosen": -13.278091430664062,
547
+ "rewards/margins": 7.641787528991699,
548
+ "rewards/rejected": -20.919878005981445,
549
  "step": 350
550
  },
551
  {
552
  "epoch": 0.82,
553
+ "grad_norm": 1479.8993498955824,
554
  "learning_rate": 9.549150281252633e-09,
555
+ "logits/chosen": -1.7768176794052124,
556
+ "logits/rejected": -1.7211551666259766,
557
+ "logps/chosen": -1.2912156581878662,
558
+ "logps/rejected": -1.4147446155548096,
559
+ "loss": 6.1375,
560
+ "rewards/accuracies": 0.71875,
561
+ "rewards/chosen": -11.908048629760742,
562
+ "rewards/margins": 7.179844856262207,
563
+ "rewards/rejected": -19.0878963470459,
564
  "step": 360
565
  },
566
  {
567
  "epoch": 0.84,
568
+ "grad_norm": 1484.9108288363668,
569
  "learning_rate": 7.3420401072985306e-09,
570
+ "logits/chosen": -1.8079217672348022,
571
+ "logits/rejected": -1.7539104223251343,
572
+ "logps/chosen": -1.3423258066177368,
573
+ "logps/rejected": -1.47573983669281,
574
+ "loss": 6.2086,
575
+ "rewards/accuracies": 0.762499988079071,
576
+ "rewards/chosen": -12.771210670471191,
577
+ "rewards/margins": 7.595045566558838,
578
+ "rewards/rejected": -20.366256713867188,
579
  "step": 370
580
  },
581
  {
582
  "epoch": 0.86,
583
+ "grad_norm": 1618.353367766913,
584
  "learning_rate": 5.404627290395369e-09,
585
+ "logits/chosen": -1.7728559970855713,
586
+ "logits/rejected": -1.7030681371688843,
587
+ "logps/chosen": -1.3002992868423462,
588
+ "logps/rejected": -1.4108442068099976,
589
+ "loss": 6.298,
590
+ "rewards/accuracies": 0.75,
591
+ "rewards/chosen": -11.069345474243164,
592
+ "rewards/margins": 7.0548906326293945,
593
+ "rewards/rejected": -18.124235153198242,
594
  "step": 380
595
  },
596
  {
597
  "epoch": 0.89,
598
+ "grad_norm": 2196.213456928754,
599
  "learning_rate": 3.74916077816162e-09,
600
+ "logits/chosen": -1.7825977802276611,
601
+ "logits/rejected": -1.7169339656829834,
602
+ "logps/chosen": -1.2813310623168945,
603
+ "logps/rejected": -1.3594629764556885,
604
+ "loss": 6.404,
605
  "rewards/accuracies": 0.706250011920929,
606
+ "rewards/chosen": -13.153631210327148,
607
+ "rewards/margins": 6.200293064117432,
608
+ "rewards/rejected": -19.353923797607422,
609
  "step": 390
610
  },
611
  {
612
  "epoch": 0.91,
613
+ "grad_norm": 1485.6207798922674,
614
  "learning_rate": 2.386106962899165e-09,
615
+ "logits/chosen": -1.7052252292633057,
616
+ "logits/rejected": -1.6294324398040771,
617
+ "logps/chosen": -1.357877492904663,
618
+ "logps/rejected": -1.4793249368667603,
619
+ "loss": 6.3409,
620
+ "rewards/accuracies": 0.731249988079071,
621
+ "rewards/chosen": -13.145660400390625,
622
+ "rewards/margins": 6.510775089263916,
623
+ "rewards/rejected": -19.656435012817383,
624
  "step": 400
625
  },
626
  {
627
  "epoch": 0.93,
628
+ "grad_norm": 1593.3048999278337,
629
  "learning_rate": 1.3240835096913706e-09,
630
+ "logits/chosen": -1.7411197423934937,
631
+ "logits/rejected": -1.6328284740447998,
632
+ "logps/chosen": -1.2782161235809326,
633
+ "logps/rejected": -1.4311167001724243,
634
+ "loss": 6.5502,
635
+ "rewards/accuracies": 0.71875,
636
+ "rewards/chosen": -10.666521072387695,
637
+ "rewards/margins": 8.348638534545898,
638
+ "rewards/rejected": -19.015161514282227,
639
  "step": 410
640
  },
641
  {
642
  "epoch": 0.96,
643
+ "grad_norm": 1557.0443174679092,
644
  "learning_rate": 5.698048727497462e-10,
645
+ "logits/chosen": -1.763695478439331,
646
+ "logits/rejected": -1.6889617443084717,
647
+ "logps/chosen": -1.3193089962005615,
648
+ "logps/rejected": -1.4645164012908936,
649
+ "loss": 6.3476,
650
+ "rewards/accuracies": 0.768750011920929,
651
+ "rewards/chosen": -12.801119804382324,
652
+ "rewards/margins": 7.825028419494629,
653
+ "rewards/rejected": -20.626148223876953,
654
  "step": 420
655
  },
656
  {
657
  "epoch": 0.98,
658
+ "grad_norm": 1384.5036857203058,
659
  "learning_rate": 1.2803984447259387e-10,
660
+ "logits/chosen": -1.772894263267517,
661
+ "logits/rejected": -1.7122442722320557,
662
+ "logps/chosen": -1.3083751201629639,
663
+ "logps/rejected": -1.4367603063583374,
664
+ "loss": 6.1773,
665
+ "rewards/accuracies": 0.7124999761581421,
666
+ "rewards/chosen": -12.603564262390137,
667
+ "rewards/margins": 8.073251724243164,
668
+ "rewards/rejected": -20.67681884765625,
669
  "step": 430
670
  },
671
  {
672
  "epoch": 1.0,
673
  "step": 439,
674
  "total_flos": 0.0,
675
+ "train_loss": 6.969558794025951,
676
+ "train_runtime": 6844.0218,
677
+ "train_samples_per_second": 8.217,
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:1987847d5ecd496557addb7e51b8b1d07a504e606fde07b6cd2c8c2d0088225b
3
- size 6520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e588dab50a63217715e4ea48b897019c21921d9010f656a0ac29dbc3749cf6d4
3
+ size 6328