antonpolishko commited on
Commit
a9a6912
·
verified ·
1 Parent(s): 99aa32f

Model save

Browse files
README.md CHANGED
@@ -27,7 +27,7 @@ print(output["generated_text"])
27
 
28
  ## Training procedure
29
 
30
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/polyagent/MamayLM-ORPO/runs/a0fqttv2)
31
 
32
 
33
  This model was trained with ORPO, a method introduced in [ORPO: Monolithic Preference Optimization without Reference Model](https://huggingface.co/papers/2403.07691).
 
27
 
28
  ## Training procedure
29
 
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/polyagent/MamayLM-ORPO/runs/3to80ym3)
31
 
32
 
33
  This model was trained with ORPO, a method introduced in [ORPO: Monolithic Preference Optimization without Reference Model](https://huggingface.co/papers/2403.07691).
adapter_config.json CHANGED
@@ -24,8 +24,8 @@
24
  "rank_pattern": {},
25
  "revision": null,
26
  "target_modules": [
27
- "q_proj",
28
- "v_proj"
29
  ],
30
  "task_type": "CAUSAL_LM",
31
  "trainable_token_indices": null,
 
24
  "rank_pattern": {},
25
  "revision": null,
26
  "target_modules": [
27
+ "v_proj",
28
+ "q_proj"
29
  ],
30
  "task_type": "CAUSAL_LM",
31
  "trainable_token_indices": null,
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:839616525392f19abfa121fe7eb94ae1ad5b6e0486123990f132c758bc966a4e
3
  size 71588104
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b40ac4fb9ca6db92eb1c8f97fccebeedd404d9b4ea366bea25cde654b788caff
3
  size 71588104
all_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "epoch": 1.0,
3
  "total_flos": 0.0,
4
- "train_loss": 1.1825333921370969,
5
- "train_runtime": 1561.4009,
6
  "train_samples": 12400,
7
- "train_samples_per_second": 7.942,
8
- "train_steps_per_second": 0.496
9
  }
 
1
  {
2
  "epoch": 1.0,
3
  "total_flos": 0.0,
4
+ "train_loss": 1.0746982106854839,
5
+ "train_runtime": 1520.1443,
6
  "train_samples": 12400,
7
+ "train_samples_per_second": 8.157,
8
+ "train_steps_per_second": 0.51
9
  }
runs/May09_01-29-54_sky-3542-mykola-5e9d8a2f-head-4qgcm7zi-compute/events.out.tfevents.1746754269.sky-3542-mykola-5e9d8a2f-head-4qgcm7zi-compute.103671.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f53a2af8f14c03425388f7ce02892f00fda987681a5e9f2a2b1b5da737699304
3
+ size 72147
train_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "epoch": 1.0,
3
  "total_flos": 0.0,
4
- "train_loss": 1.1825333921370969,
5
- "train_runtime": 1561.4009,
6
  "train_samples": 12400,
7
- "train_samples_per_second": 7.942,
8
- "train_steps_per_second": 0.496
9
  }
 
1
  {
2
  "epoch": 1.0,
3
  "total_flos": 0.0,
4
+ "train_loss": 1.0746982106854839,
5
+ "train_runtime": 1520.1443,
6
  "train_samples": 12400,
7
+ "train_samples_per_second": 8.157,
8
+ "train_steps_per_second": 0.51
9
  }
trainer_state.json CHANGED
@@ -11,1398 +11,1398 @@
11
  "log_history": [
12
  {
13
  "epoch": 0.012903225806451613,
14
- "grad_norm": 25.644566750661518,
15
- "learning_rate": 9e-08,
16
- "log_odds_chosen": 1.0859375,
17
- "log_odds_ratio": -0.3345703184604645,
18
- "logits/chosen": -1.0812499523162842,
19
- "logits/rejected": 15.675000190734863,
20
- "logps/chosen": -1.661718726158142,
21
- "logps/rejected": -2.6156249046325684,
22
- "loss": 2.0143,
23
- "nll_loss": 1.997656226158142,
24
  "rewards/accuracies": 0.949999988079071,
25
- "rewards/chosen": -0.083251953125,
26
- "rewards/margins": 0.04765624925494194,
27
- "rewards/rejected": -0.13076171278953552,
28
  "step": 10
29
  },
30
  {
31
  "epoch": 0.025806451612903226,
32
- "grad_norm": 20.851994834851144,
33
- "learning_rate": 1.8999999999999998e-07,
34
- "log_odds_chosen": 1.193359375,
35
- "log_odds_ratio": -0.3104492127895355,
36
- "logits/chosen": -1.044531226158142,
37
- "logits/rejected": 16.793750762939453,
38
- "logps/chosen": -1.6945312023162842,
39
- "logps/rejected": -2.75,
40
- "loss": 2.0119,
41
- "nll_loss": 1.9953124523162842,
42
  "rewards/accuracies": 0.9750000238418579,
43
- "rewards/chosen": -0.084716796875,
44
- "rewards/margins": 0.05290527269244194,
45
- "rewards/rejected": -0.13750000298023224,
46
  "step": 20
47
  },
48
  {
49
  "epoch": 0.03870967741935484,
50
- "grad_norm": 20.360177564962466,
51
- "learning_rate": 2.9e-07,
52
- "log_odds_chosen": 1.03125,
53
- "log_odds_ratio": -0.3462890684604645,
54
- "logits/chosen": -1.400537133216858,
55
- "logits/rejected": 16.28125,
56
- "logps/chosen": -1.689062476158142,
57
- "logps/rejected": -2.5843749046325684,
58
- "loss": 2.0161,
59
- "nll_loss": 1.9992187023162842,
60
- "rewards/accuracies": 0.9375,
61
- "rewards/chosen": -0.08442382514476776,
62
- "rewards/margins": 0.04475097730755806,
63
- "rewards/rejected": -0.12924805283546448,
64
  "step": 30
65
  },
66
  {
67
  "epoch": 0.05161290322580645,
68
- "grad_norm": 16.05099297622858,
69
- "learning_rate": 3.8999999999999997e-07,
70
- "log_odds_chosen": 0.9496093988418579,
71
- "log_odds_ratio": -0.3568359315395355,
72
- "logits/chosen": -0.643359363079071,
73
- "logits/rejected": 16.575000762939453,
74
- "logps/chosen": -1.596093773841858,
75
- "logps/rejected": -2.409374952316284,
76
- "loss": 1.9323,
77
- "nll_loss": 1.916406273841858,
78
- "rewards/accuracies": 0.9624999761581421,
79
- "rewards/chosen": -0.07978515326976776,
80
- "rewards/margins": 0.04074706882238388,
81
- "rewards/rejected": -0.12055663764476776,
82
  "step": 40
83
  },
84
  {
85
  "epoch": 0.06451612903225806,
86
- "grad_norm": 12.104063691736703,
87
- "learning_rate": 4.9e-07,
88
- "log_odds_chosen": 0.8785156011581421,
89
- "log_odds_ratio": -0.3833984434604645,
90
- "logits/chosen": 0.05205078050494194,
91
- "logits/rejected": 16.825000762939453,
92
- "logps/chosen": -1.5359375476837158,
93
- "logps/rejected": -2.280468702316284,
94
- "loss": 1.8297,
95
- "nll_loss": 1.810156226158142,
96
- "rewards/accuracies": 0.956250011920929,
97
- "rewards/chosen": -0.07680664211511612,
98
- "rewards/margins": 0.03731689602136612,
99
- "rewards/rejected": -0.11411132663488388,
100
  "step": 50
101
  },
102
  {
103
  "epoch": 0.07741935483870968,
104
- "grad_norm": 12.913754892711125,
105
- "learning_rate": 5.9e-07,
106
- "log_odds_chosen": 0.667187511920929,
107
- "log_odds_ratio": -0.44609373807907104,
108
- "logits/chosen": 0.5088135004043579,
109
- "logits/rejected": 16.625,
110
- "logps/chosen": -1.4953124523162842,
111
- "logps/rejected": -2.0453124046325684,
112
- "loss": 1.7744,
113
- "nll_loss": 1.752343773841858,
114
- "rewards/accuracies": 0.90625,
115
- "rewards/chosen": -0.07480468600988388,
116
- "rewards/margins": 0.02764892578125,
117
- "rewards/rejected": -0.10234375298023224,
118
  "step": 60
119
  },
120
  {
121
  "epoch": 0.09032258064516129,
122
- "grad_norm": 9.142494182776774,
123
- "learning_rate": 6.9e-07,
124
- "log_odds_chosen": 0.633984386920929,
125
- "log_odds_ratio": -0.4501953125,
126
- "logits/chosen": 0.6404052972793579,
127
- "logits/rejected": 17.112499237060547,
128
- "logps/chosen": -1.3546874523162842,
129
- "logps/rejected": -1.8671875,
130
- "loss": 1.6087,
131
- "nll_loss": 1.5859375,
132
- "rewards/accuracies": 0.9437500238418579,
133
- "rewards/chosen": -0.067626953125,
134
- "rewards/margins": 0.02565917931497097,
135
- "rewards/rejected": -0.09335937350988388,
136
  "step": 70
137
  },
138
  {
139
  "epoch": 0.1032258064516129,
140
- "grad_norm": 7.642065491860574,
141
- "learning_rate": 7.9e-07,
142
- "log_odds_chosen": 0.4232421815395355,
143
- "log_odds_ratio": -0.5289062261581421,
144
- "logits/chosen": 0.902783215045929,
145
- "logits/rejected": 17.225000381469727,
146
- "logps/chosen": -1.3156249523162842,
147
- "logps/rejected": -1.6398437023162842,
148
- "loss": 1.5167,
149
- "nll_loss": 1.489843726158142,
150
- "rewards/accuracies": 0.800000011920929,
151
- "rewards/chosen": -0.06577148288488388,
152
- "rewards/margins": 0.016265869140625,
153
- "rewards/rejected": -0.08198241889476776,
154
  "step": 80
155
  },
156
  {
157
  "epoch": 0.11612903225806452,
158
- "grad_norm": 6.3505930263621835,
159
- "learning_rate": 8.9e-07,
160
- "log_odds_chosen": 0.38359373807907104,
161
- "log_odds_ratio": -0.5501953363418579,
162
- "logits/chosen": 0.916333019733429,
163
- "logits/rejected": 17.206249237060547,
164
- "logps/chosen": -1.25,
165
- "logps/rejected": -1.540624976158142,
166
- "loss": 1.439,
167
- "nll_loss": 1.411718726158142,
168
- "rewards/accuracies": 0.800000011920929,
169
- "rewards/chosen": -0.06254883110523224,
170
- "rewards/margins": 0.0145416259765625,
171
- "rewards/rejected": -0.07700195163488388,
172
  "step": 90
173
  },
174
  {
175
  "epoch": 0.12903225806451613,
176
- "grad_norm": 6.45766719390769,
177
- "learning_rate": 9.9e-07,
178
- "log_odds_chosen": 0.2682128846645355,
179
- "log_odds_ratio": -0.596484363079071,
180
- "logits/chosen": 0.8069213628768921,
181
- "logits/rejected": 16.768749237060547,
182
- "logps/chosen": -1.243749976158142,
183
- "logps/rejected": -1.455468773841858,
184
- "loss": 1.4104,
185
- "nll_loss": 1.3820312023162842,
186
- "rewards/accuracies": 0.675000011920929,
187
- "rewards/chosen": -0.06218261644244194,
188
- "rewards/margins": 0.01055297814309597,
189
- "rewards/rejected": -0.07265625149011612,
190
  "step": 100
191
  },
192
  {
193
  "epoch": 0.14193548387096774,
194
- "grad_norm": 5.146838422076048,
195
- "learning_rate": 9.578262852211513e-07,
196
- "log_odds_chosen": 0.22499999403953552,
197
- "log_odds_ratio": -0.6167968511581421,
198
- "logits/chosen": 1.126953125,
199
- "logits/rejected": 17.337499618530273,
200
- "logps/chosen": -1.2062499523162842,
201
- "logps/rejected": -1.380468726158142,
202
- "loss": 1.352,
203
- "nll_loss": 1.3203125,
204
- "rewards/accuracies": 0.6937500238418579,
205
- "rewards/chosen": -0.06035156175494194,
206
- "rewards/margins": 0.00867309607565403,
207
- "rewards/rejected": -0.0689697265625,
208
  "step": 110
209
  },
210
  {
211
  "epoch": 0.15483870967741936,
212
- "grad_norm": 5.171528624192429,
213
- "learning_rate": 9.166984970282113e-07,
214
- "log_odds_chosen": 0.156982421875,
215
- "log_odds_ratio": -0.6578124761581421,
216
- "logits/chosen": 0.30644530057907104,
217
- "logits/rejected": 17.549999237060547,
218
- "logps/chosen": -1.1554687023162842,
219
- "logps/rejected": -1.2781250476837158,
220
- "loss": 1.2863,
221
- "nll_loss": 1.25390625,
222
- "rewards/accuracies": 0.5687500238418579,
223
- "rewards/chosen": -0.05783691257238388,
224
- "rewards/margins": 0.0060440064407885075,
225
- "rewards/rejected": -0.06389160454273224,
226
  "step": 120
227
  },
228
  {
229
  "epoch": 0.16774193548387098,
230
- "grad_norm": 5.593424920766334,
231
- "learning_rate": 8.804509063256238e-07,
232
- "log_odds_chosen": 0.23623046278953552,
233
- "log_odds_ratio": -0.622265636920929,
234
- "logits/chosen": 0.842944324016571,
235
- "logits/rejected": 17.143749237060547,
236
- "logps/chosen": -1.16796875,
237
- "logps/rejected": -1.3429687023162842,
238
- "loss": 1.2996,
239
- "nll_loss": 1.26953125,
240
- "rewards/accuracies": 0.706250011920929,
241
- "rewards/chosen": -0.05837402492761612,
242
- "rewards/margins": 0.00878295861184597,
243
- "rewards/rejected": -0.06723632663488388,
244
  "step": 130
245
  },
246
  {
247
  "epoch": 0.18064516129032257,
248
- "grad_norm": 6.260522888118125,
249
- "learning_rate": 8.481889296799708e-07,
250
- "log_odds_chosen": 0.28535157442092896,
251
- "log_odds_ratio": -0.596875011920929,
252
- "logits/chosen": -0.04296875,
253
- "logits/rejected": 16.424999237060547,
254
- "logps/chosen": -1.12890625,
255
- "logps/rejected": -1.3406250476837158,
256
- "loss": 1.2356,
257
- "nll_loss": 1.2062499523162842,
258
- "rewards/accuracies": 0.6937500238418579,
259
- "rewards/chosen": -0.05642089992761612,
260
- "rewards/margins": 0.01062927208840847,
261
- "rewards/rejected": -0.06704101711511612,
262
  "step": 140
263
  },
264
  {
265
  "epoch": 0.1935483870967742,
266
- "grad_norm": 5.345226504031526,
267
- "learning_rate": 8.192319205190404e-07,
268
- "log_odds_chosen": 0.4795898497104645,
269
- "log_odds_ratio": -0.5205078125,
270
- "logits/chosen": -0.7708984613418579,
271
- "logits/rejected": 16.268749237060547,
272
- "logps/chosen": -1.1124999523162842,
273
- "logps/rejected": -1.47265625,
274
- "loss": 1.2153,
275
- "nll_loss": 1.189062476158142,
276
- "rewards/accuracies": 0.793749988079071,
277
- "rewards/chosen": -0.0556640625,
278
- "rewards/margins": 0.01801147498190403,
279
- "rewards/rejected": -0.07373046875,
280
  "step": 150
281
  },
282
  {
283
  "epoch": 0.2064516129032258,
284
- "grad_norm": 6.282793059430496,
285
- "learning_rate": 7.93051585718144e-07,
286
- "log_odds_chosen": 0.657031238079071,
287
- "log_odds_ratio": -0.458984375,
288
- "logits/chosen": -1.400390625,
289
- "logits/rejected": 15.518750190734863,
290
- "logps/chosen": -1.078125,
291
- "logps/rejected": -1.568750023841858,
292
- "loss": 1.1709,
293
- "nll_loss": 1.1484375,
294
- "rewards/accuracies": 0.887499988079071,
295
- "rewards/chosen": -0.05393066257238388,
296
- "rewards/margins": 0.02449340745806694,
297
- "rewards/rejected": -0.07846679538488388,
298
  "step": 160
299
  },
300
  {
301
  "epoch": 0.21935483870967742,
302
- "grad_norm": 4.627417393099247,
303
- "learning_rate": 7.692307692307691e-07,
304
- "log_odds_chosen": 1.0222656726837158,
305
- "log_odds_ratio": -0.34492188692092896,
306
- "logits/chosen": -1.593359351158142,
307
- "logits/rejected": 16.537500381469727,
308
- "logps/chosen": -1.0378906726837158,
309
- "logps/rejected": -1.810156226158142,
310
- "loss": 1.1228,
311
- "nll_loss": 1.105859398841858,
312
- "rewards/accuracies": 0.949999988079071,
313
- "rewards/chosen": -0.05180663987994194,
314
- "rewards/margins": 0.03864746168255806,
315
- "rewards/rejected": -0.09047851711511612,
316
  "step": 170
317
  },
318
  {
319
  "epoch": 0.23225806451612904,
320
- "grad_norm": 5.521117755627069,
321
- "learning_rate": 7.474350927519359e-07,
322
- "log_odds_chosen": 0.9964843988418579,
323
- "log_odds_ratio": -0.3558593690395355,
324
- "logits/chosen": -1.365020751953125,
325
- "logits/rejected": 15.856249809265137,
326
- "logps/chosen": -1.075781226158142,
327
- "logps/rejected": -1.837499976158142,
328
- "loss": 1.1376,
329
- "nll_loss": 1.120703101158142,
330
- "rewards/accuracies": 0.949999988079071,
331
- "rewards/chosen": -0.0538330078125,
332
- "rewards/margins": 0.03806152194738388,
333
- "rewards/rejected": -0.091796875,
334
  "step": 180
335
  },
336
  {
337
  "epoch": 0.24516129032258063,
338
- "grad_norm": 4.749176898602799,
339
- "learning_rate": 7.273929674533079e-07,
340
- "log_odds_chosen": 1.082421898841858,
341
- "log_odds_ratio": -0.3404296934604645,
342
- "logits/chosen": -0.965893566608429,
343
- "logits/rejected": 16.287500381469727,
344
- "logps/chosen": -1.0546875,
345
- "logps/rejected": -1.890625,
346
- "loss": 1.13,
347
- "nll_loss": 1.1121094226837158,
348
- "rewards/accuracies": 0.918749988079071,
349
- "rewards/chosen": -0.052734375,
350
- "rewards/margins": 0.04179687425494194,
351
- "rewards/rejected": -0.094482421875,
352
  "step": 190
353
  },
354
  {
355
  "epoch": 0.25806451612903225,
356
- "grad_norm": 4.695266267620632,
357
- "learning_rate": 7.088812050083358e-07,
358
- "log_odds_chosen": 1.3390624523162842,
359
- "log_odds_ratio": -0.2783203125,
360
- "logits/chosen": -1.335351586341858,
361
- "logits/rejected": 16.443750381469727,
362
- "logps/chosen": -1.094140648841858,
363
- "logps/rejected": -2.1539063453674316,
364
- "loss": 1.1532,
365
- "nll_loss": 1.139062523841858,
366
- "rewards/accuracies": 0.981249988079071,
367
- "rewards/chosen": -0.0546875,
368
- "rewards/margins": 0.05307617038488388,
369
- "rewards/rejected": -0.10776367038488388,
370
  "step": 200
371
  },
372
  {
373
  "epoch": 0.2709677419354839,
374
- "grad_norm": 4.984030253178343,
375
- "learning_rate": 6.917144638660746e-07,
376
- "log_odds_chosen": 1.2507812976837158,
377
- "log_odds_ratio": -0.3287109434604645,
378
- "logits/chosen": -1.6774413585662842,
379
- "logits/rejected": 15.637499809265137,
380
- "logps/chosen": -1.0691406726837158,
381
- "logps/rejected": -2.0609374046325684,
382
- "loss": 1.1217,
383
- "nll_loss": 1.1062500476837158,
384
- "rewards/accuracies": 0.90625,
385
- "rewards/chosen": -0.05351562425494194,
386
- "rewards/margins": 0.04941406100988388,
387
- "rewards/rejected": -0.10297851264476776,
388
  "step": 210
389
  },
390
  {
391
  "epoch": 0.2838709677419355,
392
- "grad_norm": 5.319599464728118,
393
- "learning_rate": 6.757373783994859e-07,
394
- "log_odds_chosen": 1.2843749523162842,
395
- "log_odds_ratio": -0.33281248807907104,
396
- "logits/chosen": -0.9686523675918579,
397
- "logits/rejected": 16.100000381469727,
398
- "logps/chosen": -1.119531273841858,
399
- "logps/rejected": -2.167187452316284,
400
- "loss": 1.1792,
401
- "nll_loss": 1.162500023841858,
402
- "rewards/accuracies": 0.862500011920929,
403
- "rewards/chosen": -0.0560302734375,
404
- "rewards/margins": 0.05231933668255806,
405
- "rewards/rejected": -0.10844726860523224,
406
  "step": 220
407
  },
408
  {
409
  "epoch": 0.2967741935483871,
410
- "grad_norm": 5.150777264477966,
411
- "learning_rate": 6.608186004550897e-07,
412
- "log_odds_chosen": 1.2585937976837158,
413
- "log_odds_ratio": -0.32441407442092896,
414
- "logits/chosen": -0.751171886920929,
415
- "logits/rejected": 15.818750381469727,
416
- "logps/chosen": -1.129296898841858,
417
- "logps/rejected": -2.1468749046325684,
418
- "loss": 1.1738,
419
- "nll_loss": 1.1570312976837158,
420
- "rewards/accuracies": 0.9312499761581421,
421
- "rewards/chosen": -0.05646972730755806,
422
- "rewards/margins": 0.05085449293255806,
423
- "rewards/rejected": -0.107421875,
424
  "step": 230
425
  },
426
  {
427
  "epoch": 0.3096774193548387,
428
- "grad_norm": 4.4743402953186475,
429
- "learning_rate": 6.468462273531509e-07,
430
- "log_odds_chosen": 1.442968726158142,
431
- "log_odds_ratio": -0.28154295682907104,
432
- "logits/chosen": -1.921606421470642,
433
- "logits/rejected": 14.418749809265137,
434
- "logps/chosen": -1.079687476158142,
435
- "logps/rejected": -2.249218702316284,
436
- "loss": 1.1332,
437
- "nll_loss": 1.119531273841858,
438
- "rewards/accuracies": 0.9624999761581421,
439
- "rewards/chosen": -0.05400390550494194,
440
- "rewards/margins": 0.05839843675494194,
441
- "rewards/rejected": -0.11249999701976776,
442
  "step": 240
443
  },
444
  {
445
  "epoch": 0.3225806451612903,
446
- "grad_norm": 4.613865271523531,
447
- "learning_rate": 6.337242505244779e-07,
448
- "log_odds_chosen": 1.81640625,
449
- "log_odds_ratio": -0.2099609375,
450
- "logits/chosen": -2.2573485374450684,
451
- "logits/rejected": 15.393750190734863,
452
- "logps/chosen": -1.0378906726837158,
453
- "logps/rejected": -2.515625,
454
- "loss": 1.0852,
455
- "nll_loss": 1.0734374523162842,
456
  "rewards/accuracies": 0.981249988079071,
457
- "rewards/chosen": -0.05190429836511612,
458
- "rewards/margins": 0.07392577826976776,
459
- "rewards/rejected": -0.12568359076976776,
460
  "step": 250
461
  },
462
  {
463
  "epoch": 0.33548387096774196,
464
- "grad_norm": 4.8096386553052275,
465
- "learning_rate": 6.213697660011999e-07,
466
- "log_odds_chosen": 1.71875,
467
- "log_odds_ratio": -0.23369140923023224,
468
- "logits/chosen": -2.139453172683716,
469
- "logits/rejected": 15.368749618530273,
470
- "logps/chosen": -1.0593750476837158,
471
- "logps/rejected": -2.456249952316284,
472
- "loss": 1.1047,
473
- "nll_loss": 1.0929687023162842,
474
- "rewards/accuracies": 0.9624999761581421,
475
- "rewards/chosen": -0.052978515625,
476
- "rewards/margins": 0.06979980319738388,
477
- "rewards/rejected": -0.12275390326976776,
478
  "step": 260
479
  },
480
  {
481
  "epoch": 0.34838709677419355,
482
- "grad_norm": 4.766482993858386,
483
- "learning_rate": 6.097107608496924e-07,
484
- "log_odds_chosen": 1.78125,
485
- "log_odds_ratio": -0.24179688096046448,
486
- "logits/chosen": -1.9189453125,
487
- "logits/rejected": 15.231249809265137,
488
- "logps/chosen": -1.023828148841858,
489
- "logps/rejected": -2.484375,
490
- "loss": 1.0711,
491
- "nll_loss": 1.058984398841858,
492
- "rewards/accuracies": 0.956250011920929,
493
- "rewards/chosen": -0.0511474609375,
494
- "rewards/margins": 0.07294921576976776,
495
- "rewards/rejected": -0.12416992336511612,
496
  "step": 270
497
  },
498
  {
499
  "epoch": 0.36129032258064514,
500
- "grad_norm": 5.318754743128315,
501
- "learning_rate": 5.986843400892497e-07,
502
- "log_odds_chosen": 1.9093749523162842,
503
- "log_odds_ratio": -0.21914061903953552,
504
- "logits/chosen": -1.540991187095642,
505
- "logits/rejected": 15.493749618530273,
506
- "logps/chosen": -1.0964844226837158,
507
- "logps/rejected": -2.684375047683716,
508
- "loss": 1.1231,
509
- "nll_loss": 1.1124999523162842,
510
- "rewards/accuracies": 0.9375,
511
- "rewards/chosen": -0.05488281324505806,
512
- "rewards/margins": 0.07929687201976776,
513
- "rewards/rejected": -0.13408203423023224,
514
  "step": 280
515
  },
516
  {
517
  "epoch": 0.3741935483870968,
518
- "grad_norm": 6.666659671454164,
519
- "learning_rate": 5.88235294117647e-07,
520
- "log_odds_chosen": 1.927343726158142,
521
- "log_odds_ratio": -0.19658203423023224,
522
- "logits/chosen": -1.6470215320587158,
523
- "logits/rejected": 15.8125,
524
- "logps/chosen": -1.060937523841858,
525
- "logps/rejected": -2.635937452316284,
526
- "loss": 1.0834,
527
- "nll_loss": 1.073828101158142,
528
  "rewards/accuracies": 0.981249988079071,
529
- "rewards/chosen": -0.05300293117761612,
530
- "rewards/margins": 0.078857421875,
531
- "rewards/rejected": -0.13203124701976776,
532
  "step": 290
533
  },
534
  {
535
  "epoch": 0.3870967741935484,
536
- "grad_norm": 15.408328095211239,
537
- "learning_rate": 5.783149319662401e-07,
538
- "log_odds_chosen": 2.1585936546325684,
539
- "log_odds_ratio": -0.19511719048023224,
540
- "logits/chosen": -1.6232421398162842,
541
- "logits/rejected": 15.637499809265137,
542
- "logps/chosen": -1.088281273841858,
543
- "logps/rejected": -2.9000000953674316,
544
- "loss": 1.116,
545
- "nll_loss": 1.1062500476837158,
546
- "rewards/accuracies": 0.9437500238418579,
547
- "rewards/chosen": -0.05439453199505806,
548
- "rewards/margins": 0.09067382663488388,
549
- "rewards/rejected": -0.14501953125,
550
  "step": 300
551
  },
552
  {
553
  "epoch": 0.4,
554
- "grad_norm": 5.802623061365325,
555
- "learning_rate": 5.688801239885743e-07,
556
- "log_odds_chosen": 2.0640625953674316,
557
- "log_odds_ratio": -0.20878906548023224,
558
- "logits/chosen": -1.6769530773162842,
559
- "logits/rejected": 15.118749618530273,
560
- "logps/chosen": -1.041015625,
561
- "logps/rejected": -2.7632813453674316,
562
- "loss": 1.0771,
563
- "nll_loss": 1.0652344226837158,
564
- "rewards/accuracies": 0.9624999761581421,
565
- "rewards/chosen": -0.05205078050494194,
566
- "rewards/margins": 0.086181640625,
567
- "rewards/rejected": -0.13813476264476776,
568
  "step": 310
569
  },
570
  {
571
  "epoch": 0.4129032258064516,
572
- "grad_norm": 4.9433941112320126,
573
- "learning_rate": 5.598925109558543e-07,
574
- "log_odds_chosen": 2.0234375,
575
- "log_odds_ratio": -0.21616211533546448,
576
- "logits/chosen": -1.5070312023162842,
577
- "logits/rejected": 15.681249618530273,
578
- "logps/chosen": -1.0535156726837158,
579
- "logps/rejected": -2.739062547683716,
580
- "loss": 1.0773,
581
- "nll_loss": 1.066796898841858,
582
- "rewards/accuracies": 0.9375,
583
- "rewards/chosen": -0.05266113206744194,
584
- "rewards/margins": 0.08437500149011612,
585
- "rewards/rejected": -0.13715820014476776,
586
  "step": 320
587
  },
588
  {
589
  "epoch": 0.4258064516129032,
590
- "grad_norm": 6.776871203605618,
591
- "learning_rate": 5.513178464199713e-07,
592
- "log_odds_chosen": 2.2484374046325684,
593
- "log_odds_ratio": -0.18437500298023224,
594
- "logits/chosen": -1.753808617591858,
595
- "logits/rejected": 15.706250190734863,
596
- "logps/chosen": -1.079687476158142,
597
- "logps/rejected": -2.9859375953674316,
598
- "loss": 1.1246,
599
- "nll_loss": 1.115625023841858,
600
- "rewards/accuracies": 0.9624999761581421,
601
- "rewards/chosen": -0.053955078125,
602
- "rewards/margins": 0.09511718899011612,
603
- "rewards/rejected": -0.149169921875,
604
  "step": 330
605
  },
606
  {
607
  "epoch": 0.43870967741935485,
608
- "grad_norm": 6.2530553748320274,
609
- "learning_rate": 5.431254465935684e-07,
610
- "log_odds_chosen": 2.2367186546325684,
611
- "log_odds_ratio": -0.17998047173023224,
612
- "logits/chosen": -1.1892578601837158,
613
- "logits/rejected": 15.59375,
614
- "logps/chosen": -1.080468773841858,
615
- "logps/rejected": -2.971874952316284,
616
- "loss": 1.0915,
617
- "nll_loss": 1.082421898841858,
618
- "rewards/accuracies": 0.956250011920929,
619
- "rewards/chosen": -0.05400390550494194,
620
- "rewards/margins": 0.09462890774011612,
621
- "rewards/rejected": -0.14873047173023224,
622
  "step": 340
623
  },
624
  {
625
  "epoch": 0.45161290322580644,
626
- "grad_norm": 4.957504153968922,
627
- "learning_rate": 5.352877275721892e-07,
628
- "log_odds_chosen": 2.3226561546325684,
629
- "log_odds_ratio": -0.18168945610523224,
630
- "logits/chosen": -1.8606445789337158,
631
- "logits/rejected": 15.431249618530273,
632
- "logps/chosen": -1.0343749523162842,
633
- "logps/rejected": -2.9906249046325684,
634
- "loss": 1.0724,
635
- "nll_loss": 1.064062476158142,
636
  "rewards/accuracies": 0.9624999761581421,
637
- "rewards/chosen": -0.05168456956744194,
638
- "rewards/margins": 0.097900390625,
639
- "rewards/rejected": -0.14970703423023224,
640
  "step": 350
641
  },
642
  {
643
  "epoch": 0.4645161290322581,
644
- "grad_norm": 5.393454505901281,
645
- "learning_rate": 5.277798139692595e-07,
646
- "log_odds_chosen": 2.4281249046325684,
647
- "log_odds_ratio": -0.16191406548023224,
648
- "logits/chosen": -1.86328125,
649
- "logits/rejected": 15.481249809265137,
650
- "logps/chosen": -1.060156226158142,
651
- "logps/rejected": -3.121875047683716,
652
- "loss": 1.0862,
653
- "nll_loss": 1.078125,
654
- "rewards/accuracies": 0.9624999761581421,
655
- "rewards/chosen": -0.05302734300494194,
656
- "rewards/margins": 0.10307617485523224,
657
- "rewards/rejected": -0.15615233778953552,
658
  "step": 360
659
  },
660
  {
661
  "epoch": 0.4774193548387097,
662
- "grad_norm": 26.086626231454886,
663
- "learning_rate": 5.205792062953535e-07,
664
- "log_odds_chosen": 2.4546875953674316,
665
- "log_odds_ratio": -0.16103515028953552,
666
- "logits/chosen": -1.950781226158142,
667
- "logits/rejected": 15.84375,
668
- "logps/chosen": -1.0734374523162842,
669
- "logps/rejected": -3.1624999046325684,
670
- "loss": 1.0914,
671
- "nll_loss": 1.0832030773162842,
672
- "rewards/accuracies": 0.949999988079071,
673
- "rewards/chosen": -0.0537109375,
674
- "rewards/margins": 0.1044921875,
675
- "rewards/rejected": -0.15810546278953552,
676
  "step": 370
677
  },
678
  {
679
  "epoch": 0.49032258064516127,
680
- "grad_norm": 4.527756676582892,
681
- "learning_rate": 5.136654969375141e-07,
682
- "log_odds_chosen": 2.241406202316284,
683
- "log_odds_ratio": -0.18183593451976776,
684
- "logits/chosen": -1.874609351158142,
685
- "logits/rejected": 15.537500381469727,
686
- "logps/chosen": -1.04296875,
687
- "logps/rejected": -2.9140625,
688
- "loss": 1.0563,
689
- "nll_loss": 1.047265648841858,
690
- "rewards/accuracies": 0.9624999761581421,
691
- "rewards/chosen": -0.0521240234375,
692
- "rewards/margins": 0.09355469048023224,
693
- "rewards/rejected": -0.14560547471046448,
694
  "step": 380
695
  },
696
  {
697
  "epoch": 0.5032258064516129,
698
- "grad_norm": 5.084767531042406,
699
- "learning_rate": 5.070201265633937e-07,
700
- "log_odds_chosen": 2.249218702316284,
701
- "log_odds_ratio": -0.19853515923023224,
702
- "logits/chosen": -1.7765624523162842,
703
- "logits/rejected": 15.793749809265137,
704
- "logps/chosen": -1.060156226158142,
705
- "logps/rejected": -2.96875,
706
- "loss": 1.0876,
707
- "nll_loss": 1.0773437023162842,
708
- "rewards/accuracies": 0.956250011920929,
709
- "rewards/chosen": -0.05300293117761612,
710
- "rewards/margins": 0.09536132961511612,
711
- "rewards/rejected": -0.14814452826976776,
712
  "step": 390
713
  },
714
  {
715
  "epoch": 0.5161290322580645,
716
- "grad_norm": 4.84944127844018,
717
- "learning_rate": 5.006261743217588e-07,
718
- "log_odds_chosen": 2.28125,
719
- "log_odds_ratio": -0.17817382514476776,
720
- "logits/chosen": -2.0472655296325684,
721
- "logits/rejected": 15.443750381469727,
722
- "logps/chosen": -1.044531226158142,
723
- "logps/rejected": -2.9593749046325684,
724
- "loss": 1.069,
725
- "nll_loss": 1.060937523841858,
726
- "rewards/accuracies": 0.956250011920929,
727
- "rewards/chosen": -0.05222167819738388,
728
- "rewards/margins": 0.095703125,
729
- "rewards/rejected": -0.14790038764476776,
730
  "step": 400
731
  },
732
  {
733
  "epoch": 0.5290322580645161,
734
- "grad_norm": 5.638415085859226,
735
- "learning_rate": 4.944681764341487e-07,
736
- "log_odds_chosen": 2.3343749046325684,
737
- "log_odds_ratio": -0.17128905653953552,
738
- "logits/chosen": -2.0162110328674316,
739
- "logits/rejected": 15.5,
740
- "logps/chosen": -1.034765601158142,
741
- "logps/rejected": -3.012500047683716,
742
- "loss": 1.0596,
743
- "nll_loss": 1.05078125,
744
- "rewards/accuracies": 0.9750000238418579,
745
- "rewards/chosen": -0.05173339694738388,
746
- "rewards/margins": 0.09880371391773224,
747
- "rewards/rejected": -0.15048828721046448,
748
  "step": 410
749
  },
750
  {
751
  "epoch": 0.5419354838709678,
752
- "grad_norm": 5.7392251910086305,
753
- "learning_rate": 4.885319687460315e-07,
754
- "log_odds_chosen": 2.4296875,
755
- "log_odds_ratio": -0.16596679389476776,
756
- "logits/chosen": -1.506250023841858,
757
- "logits/rejected": 15.356249809265137,
758
- "logps/chosen": -1.0859375,
759
- "logps/rejected": -3.1703124046325684,
760
- "loss": 1.1023,
761
- "nll_loss": 1.0929687023162842,
762
- "rewards/accuracies": 0.956250011920929,
763
- "rewards/chosen": -0.05429687350988388,
764
- "rewards/margins": 0.104248046875,
765
- "rewards/rejected": -0.15859374403953552,
766
  "step": 420
767
  },
768
  {
769
  "epoch": 0.5548387096774193,
770
- "grad_norm": 5.166878698732603,
771
- "learning_rate": 4.828045495852675e-07,
772
- "log_odds_chosen": 2.5101561546325684,
773
- "log_odds_ratio": -0.1593017578125,
774
- "logits/chosen": -0.912109375,
775
- "logits/rejected": 16.268749237060547,
776
- "logps/chosen": -1.096093773841858,
777
- "logps/rejected": -3.245312452316284,
778
- "loss": 1.083,
779
- "nll_loss": 1.075781226158142,
780
- "rewards/accuracies": 0.9750000238418579,
781
- "rewards/chosen": -0.05485839769244194,
782
- "rewards/margins": 0.10732422024011612,
783
- "rewards/rejected": -0.162109375,
784
  "step": 430
785
  },
786
  {
787
  "epoch": 0.567741935483871,
788
- "grad_norm": 6.098476119626343,
789
- "learning_rate": 4.772739599033477e-07,
790
- "log_odds_chosen": 2.315624952316284,
791
- "log_odds_ratio": -0.18708495795726776,
792
- "logits/chosen": -1.7859375476837158,
793
- "logits/rejected": 15.631250381469727,
794
- "logps/chosen": -1.068750023841858,
795
- "logps/rejected": -3.0484375953674316,
796
- "loss": 1.0989,
797
- "nll_loss": 1.08984375,
798
- "rewards/accuracies": 0.9624999761581421,
799
- "rewards/chosen": -0.05339355394244194,
800
- "rewards/margins": 0.09897460788488388,
801
- "rewards/rejected": -0.15234375,
802
  "step": 440
803
  },
804
  {
805
  "epoch": 0.5806451612903226,
806
- "grad_norm": 5.0584668135526005,
807
- "learning_rate": 4.719291781830087e-07,
808
- "log_odds_chosen": 2.1578125953674316,
809
- "log_odds_ratio": -0.21845702826976776,
810
- "logits/chosen": -1.546179175376892,
811
- "logits/rejected": 15.356249809265137,
812
- "logps/chosen": -1.0906250476837158,
813
- "logps/rejected": -2.9203124046325684,
814
- "loss": 1.1175,
815
- "nll_loss": 1.107812523841858,
816
- "rewards/accuracies": 0.925000011920929,
817
- "rewards/chosen": -0.05454101413488388,
818
- "rewards/margins": 0.09147949516773224,
819
- "rewards/rejected": -0.14609375596046448,
820
  "step": 450
821
  },
822
  {
823
  "epoch": 0.5935483870967742,
824
- "grad_norm": 5.481001924793037,
825
- "learning_rate": 4.6676002800933655e-07,
826
- "log_odds_chosen": 2.4593749046325684,
827
- "log_odds_ratio": -0.18085937201976776,
828
- "logits/chosen": -1.66748046875,
829
- "logits/rejected": 15.725000381469727,
830
- "logps/chosen": -1.047265648841858,
831
- "logps/rejected": -3.137500047683716,
832
- "loss": 1.0722,
833
- "nll_loss": 1.062890648841858,
834
- "rewards/accuracies": 0.9312499761581421,
835
- "rewards/chosen": -0.0523681640625,
836
- "rewards/margins": 0.10444335639476776,
837
- "rewards/rejected": -0.15683594346046448,
838
  "step": 460
839
  },
840
  {
841
  "epoch": 0.6064516129032258,
842
- "grad_norm": 6.720988957315027,
843
- "learning_rate": 4.617570965396101e-07,
844
- "log_odds_chosen": 2.2718749046325684,
845
- "log_odds_ratio": -0.20751953125,
846
- "logits/chosen": -1.7820312976837158,
847
- "logits/rejected": 15.100000381469727,
848
- "logps/chosen": -1.09375,
849
- "logps/rejected": -3.043750047683716,
850
- "loss": 1.1066,
851
- "nll_loss": 1.095312476158142,
852
- "rewards/accuracies": 0.918749988079071,
853
- "rewards/chosen": -0.05463867262005806,
854
- "rewards/margins": 0.09743652492761612,
855
- "rewards/rejected": -0.15205077826976776,
856
  "step": 470
857
  },
858
  {
859
  "epoch": 0.6193548387096774,
860
- "grad_norm": 5.422677434794996,
861
- "learning_rate": 4.5691166238495384e-07,
862
- "log_odds_chosen": 2.639843702316284,
863
- "log_odds_ratio": -0.15380859375,
864
- "logits/chosen": -1.9500000476837158,
865
- "logits/rejected": 15.90625,
866
- "logps/chosen": -0.991015613079071,
867
- "logps/rejected": -3.237499952316284,
868
- "loss": 1.0207,
869
- "nll_loss": 1.0128905773162842,
870
- "rewards/accuracies": 0.949999988079071,
871
- "rewards/chosen": -0.049560546875,
872
- "rewards/margins": 0.1123046875,
873
- "rewards/rejected": -0.16181640326976776,
874
  "step": 480
875
  },
876
  {
877
  "epoch": 0.632258064516129,
878
- "grad_norm": 5.792363122862243,
879
- "learning_rate": 4.5221563164613456e-07,
880
- "log_odds_chosen": 2.366406202316284,
881
- "log_odds_ratio": -0.17709961533546448,
882
- "logits/chosen": -1.802148461341858,
883
- "logits/rejected": 16.075000762939453,
884
- "logps/chosen": -1.0851562023162842,
885
- "logps/rejected": -3.104687452316284,
886
- "loss": 1.1088,
887
- "nll_loss": 1.1003906726837158,
888
- "rewards/accuracies": 0.9624999761581421,
889
- "rewards/chosen": -0.05417480319738388,
890
- "rewards/margins": 0.10112304985523224,
891
- "rewards/rejected": -0.1552734375,
892
  "step": 490
893
  },
894
  {
895
  "epoch": 0.6451612903225806,
896
- "grad_norm": 4.989147526438258,
897
- "learning_rate": 4.4766148103584516e-07,
898
- "log_odds_chosen": 2.2945313453674316,
899
- "log_odds_ratio": -0.21923828125,
900
- "logits/chosen": -1.5978515148162842,
901
- "logits/rejected": 15.574999809265137,
902
- "logps/chosen": -1.1085937023162842,
903
- "logps/rejected": -3.0796875953674316,
904
- "loss": 1.1167,
905
- "nll_loss": 1.107031226158142,
906
- "rewards/accuracies": 0.925000011920929,
907
- "rewards/chosen": -0.05534667894244194,
908
- "rewards/margins": 0.09858398139476776,
909
- "rewards/rejected": -0.15380859375,
910
  "step": 500
911
  },
912
  {
913
  "epoch": 0.6580645161290323,
914
- "grad_norm": 5.380815151192619,
915
- "learning_rate": 4.4324220717793627e-07,
916
- "log_odds_chosen": 2.4789061546325684,
917
- "log_odds_ratio": -0.17275390028953552,
918
- "logits/chosen": -1.827392578125,
919
- "logits/rejected": 15.425000190734863,
920
- "logps/chosen": -1.019921898841858,
921
- "logps/rejected": -3.1312499046325684,
922
- "loss": 1.0581,
923
- "nll_loss": 1.048828125,
924
- "rewards/accuracies": 0.96875,
925
- "rewards/chosen": -0.051025390625,
926
- "rewards/margins": 0.10556640475988388,
927
- "rewards/rejected": -0.15668945014476776,
928
  "step": 510
929
  },
930
  {
931
  "epoch": 0.6709677419354839,
932
- "grad_norm": 27.261099049123168,
933
- "learning_rate": 4.38951281306147e-07,
934
- "log_odds_chosen": 2.5054688453674316,
935
- "log_odds_ratio": -0.161865234375,
936
- "logits/chosen": -2.1246094703674316,
937
- "logits/rejected": 15.712499618530273,
938
- "logps/chosen": -1.0304687023162842,
939
- "logps/rejected": -3.160937547683716,
940
- "loss": 1.0544,
941
- "nll_loss": 1.0457031726837158,
942
- "rewards/accuracies": 0.9750000238418579,
943
- "rewards/chosen": -0.05156249925494194,
944
- "rewards/margins": 0.10654296725988388,
945
- "rewards/rejected": -0.158203125,
946
  "step": 520
947
  },
948
  {
949
  "epoch": 0.6838709677419355,
950
- "grad_norm": 5.332717429106649,
951
- "learning_rate": 4.347826086956522e-07,
952
- "log_odds_chosen": 2.4390625953674316,
953
- "log_odds_ratio": -0.16909179091453552,
954
- "logits/chosen": -0.8804687261581421,
955
- "logits/rejected": 15.324999809265137,
956
- "logps/chosen": -1.0148437023162842,
957
- "logps/rejected": -3.073437452316284,
958
- "loss": 1.0413,
959
- "nll_loss": 1.032812476158142,
960
- "rewards/accuracies": 0.9624999761581421,
961
- "rewards/chosen": -0.05073242262005806,
962
- "rewards/margins": 0.10288085788488388,
963
- "rewards/rejected": -0.15371093153953552,
964
  "step": 530
965
  },
966
  {
967
  "epoch": 0.6967741935483871,
968
- "grad_norm": 4.8798362635733,
969
- "learning_rate": 4.3073049225394797e-07,
970
- "log_odds_chosen": 2.442187547683716,
971
- "log_odds_ratio": -0.21738281846046448,
972
- "logits/chosen": -0.97314453125,
973
- "logits/rejected": 16.256250381469727,
974
- "logps/chosen": -1.0964844226837158,
975
- "logps/rejected": -3.1656250953674316,
976
- "loss": 1.062,
977
- "nll_loss": 1.051171898841858,
978
- "rewards/accuracies": 0.96875,
979
- "rewards/chosen": -0.05483398586511612,
980
- "rewards/margins": 0.10336913913488388,
981
- "rewards/rejected": -0.15805664658546448,
982
  "step": 540
983
  },
984
  {
985
  "epoch": 0.7096774193548387,
986
- "grad_norm": 5.915531309666701,
987
- "learning_rate": 4.267895997763199e-07,
988
- "log_odds_chosen": 2.432812452316284,
989
- "log_odds_ratio": -0.16074219346046448,
990
- "logits/chosen": -1.596093773841858,
991
- "logits/rejected": 15.6875,
992
- "logps/chosen": -1.058203101158142,
993
- "logps/rejected": -3.1328125,
994
- "loss": 1.0698,
995
- "nll_loss": 1.0617187023162842,
996
- "rewards/accuracies": 0.981249988079071,
997
- "rewards/chosen": -0.05290527269244194,
998
- "rewards/margins": 0.10356445610523224,
999
- "rewards/rejected": -0.15644530951976776,
1000
  "step": 550
1001
  },
1002
  {
1003
  "epoch": 0.7225806451612903,
1004
- "grad_norm": 5.599089020060382,
1005
- "learning_rate": 4.2295493443781356e-07,
1006
- "log_odds_chosen": 2.4468750953674316,
1007
- "log_odds_ratio": -0.17919921875,
1008
- "logits/chosen": -1.71875,
1009
- "logits/rejected": 15.606249809265137,
1010
- "logps/chosen": -1.052343726158142,
1011
- "logps/rejected": -3.151562452316284,
1012
- "loss": 1.0818,
1013
- "nll_loss": 1.074609398841858,
1014
- "rewards/accuracies": 0.9437500238418579,
1015
- "rewards/chosen": -0.05263672024011612,
1016
- "rewards/margins": 0.10502929985523224,
1017
- "rewards/rejected": -0.15742187201976776,
1018
  "step": 560
1019
  },
1020
  {
1021
  "epoch": 0.7354838709677419,
1022
- "grad_norm": 5.159448709809527,
1023
- "learning_rate": 4.192218081503185e-07,
1024
- "log_odds_chosen": 2.532031297683716,
1025
- "log_odds_ratio": -0.16254882514476776,
1026
- "logits/chosen": -1.670800805091858,
1027
- "logits/rejected": 15.712499618530273,
1028
- "logps/chosen": -1.015234351158142,
1029
- "logps/rejected": -3.174999952316284,
1030
- "loss": 1.0451,
1031
- "nll_loss": 1.037500023841858,
1032
- "rewards/accuracies": 0.96875,
1033
- "rewards/chosen": -0.05075683444738388,
1034
- "rewards/margins": 0.10781250149011612,
1035
- "rewards/rejected": -0.15849609673023224,
1036
  "step": 570
1037
  },
1038
  {
1039
  "epoch": 0.7483870967741936,
1040
- "grad_norm": 5.112624827446688,
1041
- "learning_rate": 4.1558581746169135e-07,
1042
- "log_odds_chosen": 2.356250047683716,
1043
- "log_odds_ratio": -0.16840820014476776,
1044
- "logits/chosen": -1.854101538658142,
1045
- "logits/rejected": 15.09375,
1046
- "logps/chosen": -0.995312511920929,
1047
- "logps/rejected": -2.965625047683716,
1048
- "loss": 1.0322,
1049
- "nll_loss": 1.0242187976837158,
1050
- "rewards/accuracies": 0.9624999761581421,
1051
- "rewards/chosen": -0.04978027194738388,
1052
- "rewards/margins": 0.09848632663488388,
1053
- "rewards/rejected": -0.1484375,
1054
  "step": 580
1055
  },
1056
  {
1057
  "epoch": 0.7612903225806451,
1058
- "grad_norm": 5.281398784435252,
1059
- "learning_rate": 4.1204282171516457e-07,
1060
- "log_odds_chosen": 2.559375047683716,
1061
- "log_odds_ratio": -0.14653320610523224,
1062
- "logits/chosen": -2.137988328933716,
1063
- "logits/rejected": 15.412500381469727,
1064
- "logps/chosen": -0.9964843988418579,
1065
- "logps/rejected": -3.1624999046325684,
1066
- "loss": 1.0347,
1067
- "nll_loss": 1.02734375,
1068
- "rewards/accuracies": 0.96875,
1069
- "rewards/chosen": -0.0498046875,
1070
- "rewards/margins": 0.108154296875,
1071
- "rewards/rejected": -0.157958984375,
1072
  "step": 590
1073
  },
1074
  {
1075
  "epoch": 0.7741935483870968,
1076
- "grad_norm": 5.043808760019304,
1077
- "learning_rate": 4.085889232227184e-07,
1078
- "log_odds_chosen": 2.6171875,
1079
- "log_odds_ratio": -0.14313964545726776,
1080
- "logits/chosen": -1.745507836341858,
1081
- "logits/rejected": 15.662500381469727,
1082
- "logps/chosen": -1.0437500476837158,
1083
- "logps/rejected": -3.284374952316284,
1084
- "loss": 1.0715,
1085
- "nll_loss": 1.064062476158142,
1086
- "rewards/accuracies": 0.9750000238418579,
1087
- "rewards/chosen": -0.05219726637005806,
1088
- "rewards/margins": 0.11201171576976776,
1089
- "rewards/rejected": -0.16416016221046448,
1090
  "step": 600
1091
  },
1092
  {
1093
  "epoch": 0.7870967741935484,
1094
- "grad_norm": 4.910512899103639,
1095
- "learning_rate": 4.052204492365539e-07,
1096
- "log_odds_chosen": 2.828125,
1097
- "log_odds_ratio": -0.13234862685203552,
1098
- "logits/chosen": -2.5374999046325684,
1099
- "logits/rejected": 15.556249618530273,
1100
- "logps/chosen": -1.056249976158142,
1101
- "logps/rejected": -3.503124952316284,
1102
- "loss": 1.0816,
1103
- "nll_loss": 1.07421875,
1104
- "rewards/accuracies": 0.9624999761581421,
1105
- "rewards/chosen": -0.05283202975988388,
1106
- "rewards/margins": 0.12236328423023224,
1107
- "rewards/rejected": -0.17509765923023224,
1108
  "step": 610
1109
  },
1110
  {
1111
  "epoch": 0.8,
1112
- "grad_norm": 5.830451823751554,
1113
- "learning_rate": 4.0193393552907036e-07,
1114
- "log_odds_chosen": 2.551562547683716,
1115
- "log_odds_ratio": -0.16445311903953552,
1116
- "logits/chosen": -1.998046875,
1117
- "logits/rejected": 15.949999809265137,
1118
- "logps/chosen": -1.038671851158142,
1119
- "logps/rejected": -3.2109375,
1120
- "loss": 1.0646,
1121
- "nll_loss": 1.056249976158142,
1122
- "rewards/accuracies": 0.9437500238418579,
1123
- "rewards/chosen": -0.05195312574505806,
1124
- "rewards/margins": 0.10849609225988388,
1125
- "rewards/rejected": -0.16025391221046448,
1126
  "step": 620
1127
  },
1128
  {
1129
  "epoch": 0.8129032258064516,
1130
- "grad_norm": 4.869329829957484,
1131
- "learning_rate": 3.9872611141445e-07,
1132
- "log_odds_chosen": 2.4749999046325684,
1133
- "log_odds_ratio": -0.16875000298023224,
1134
- "logits/chosen": -1.8136718273162842,
1135
- "logits/rejected": 15.84375,
1136
- "logps/chosen": -1.00390625,
1137
- "logps/rejected": -3.1015625,
1138
- "loss": 1.0204,
1139
- "nll_loss": 1.0125000476837158,
1140
- "rewards/accuracies": 0.96875,
1141
- "rewards/chosen": -0.05026855319738388,
1142
- "rewards/margins": 0.10495605319738388,
1143
- "rewards/rejected": -0.15517577528953552,
1144
  "step": 630
1145
  },
1146
  {
1147
  "epoch": 0.8258064516129032,
1148
- "grad_norm": 7.19876508368735,
1149
- "learning_rate": 3.9559388606461773e-07,
1150
- "log_odds_chosen": 2.518749952316284,
1151
- "log_odds_ratio": -0.178466796875,
1152
- "logits/chosen": -1.6452147960662842,
1153
- "logits/rejected": 15.887499809265137,
1154
- "logps/chosen": -1.057031273841858,
1155
- "logps/rejected": -3.203125,
1156
- "loss": 1.0684,
1157
- "nll_loss": 1.0593750476837158,
1158
- "rewards/accuracies": 0.956250011920929,
1159
- "rewards/chosen": -0.05280761793255806,
1160
- "rewards/margins": 0.10737304389476776,
1161
- "rewards/rejected": -0.16015625,
1162
  "step": 640
1163
  },
1164
  {
1165
  "epoch": 0.8387096774193549,
1166
- "grad_norm": 5.176818621993991,
1167
- "learning_rate": 3.9253433598942976e-07,
1168
- "log_odds_chosen": 2.8984375,
1169
- "log_odds_ratio": -0.11740722507238388,
1170
- "logits/chosen": -1.798437476158142,
1171
- "logits/rejected": 16.40625,
1172
- "logps/chosen": -1.0539062023162842,
1173
- "logps/rejected": -3.559375047683716,
1174
- "loss": 1.0807,
1175
- "nll_loss": 1.076171875,
1176
- "rewards/accuracies": 0.9750000238418579,
1177
- "rewards/chosen": -0.0526123046875,
1178
- "rewards/margins": 0.12519530951976776,
1179
- "rewards/rejected": -0.17792968451976776,
1180
  "step": 650
1181
  },
1182
  {
1183
  "epoch": 0.8516129032258064,
1184
- "grad_norm": 5.815910493259651,
1185
- "learning_rate": 3.8954469356580987e-07,
1186
- "log_odds_chosen": 2.557812452316284,
1187
- "log_odds_ratio": -0.17656250298023224,
1188
- "logits/chosen": -1.216943383216858,
1189
- "logits/rejected": 15.793749809265137,
1190
- "logps/chosen": -1.0695312023162842,
1191
- "logps/rejected": -3.2718749046325684,
1192
- "loss": 1.0767,
1193
- "nll_loss": 1.06640625,
1194
- "rewards/accuracies": 0.949999988079071,
1195
- "rewards/chosen": -0.05354003980755806,
1196
- "rewards/margins": 0.11015625298023224,
1197
- "rewards/rejected": -0.16357421875,
1198
  "step": 660
1199
  },
1200
  {
1201
  "epoch": 0.864516129032258,
1202
- "grad_norm": 4.608527017371815,
1203
- "learning_rate": 3.866223365135295e-07,
1204
- "log_odds_chosen": 2.7125000953674316,
1205
- "log_odds_ratio": -0.12763671576976776,
1206
- "logits/chosen": -1.838769555091858,
1207
- "logits/rejected": 15.675000190734863,
1208
- "logps/chosen": -1.041406273841858,
1209
- "logps/rejected": -3.356250047683716,
1210
- "loss": 1.0641,
1211
- "nll_loss": 1.0578124523162842,
1212
  "rewards/accuracies": 0.987500011920929,
1213
- "rewards/chosen": -0.05202636867761612,
1214
- "rewards/margins": 0.11591796576976776,
1215
- "rewards/rejected": -0.16787108778953552,
1216
  "step": 670
1217
  },
1218
  {
1219
  "epoch": 0.8774193548387097,
1220
- "grad_norm": 5.515685176320282,
1221
- "learning_rate": 3.837647782266681e-07,
1222
- "log_odds_chosen": 2.7484374046325684,
1223
- "log_odds_ratio": -0.13393554091453552,
1224
- "logits/chosen": -1.3933837413787842,
1225
- "logits/rejected": 15.78125,
1226
- "logps/chosen": -1.103906273841858,
1227
- "logps/rejected": -3.487499952316284,
1228
- "loss": 1.1,
1229
- "nll_loss": 1.094140648841858,
1230
  "rewards/accuracies": 0.987500011920929,
1231
- "rewards/chosen": -0.05517578125,
1232
- "rewards/margins": 0.11904297024011612,
1233
- "rewards/rejected": -0.17431640625,
1234
  "step": 680
1235
  },
1236
  {
1237
  "epoch": 0.8903225806451613,
1238
- "grad_norm": 5.325597984478911,
1239
- "learning_rate": 3.809696588797297e-07,
1240
- "log_odds_chosen": 2.688281297683716,
1241
- "log_odds_ratio": -0.19277343153953552,
1242
- "logits/chosen": -1.479223608970642,
1243
- "logits/rejected": 15.399999618530273,
1244
- "logps/chosen": -1.103124976158142,
1245
- "logps/rejected": -3.395312547683716,
1246
- "loss": 1.0544,
1247
- "nll_loss": 1.0460937023162842,
1248
  "rewards/accuracies": 0.9750000238418579,
1249
- "rewards/chosen": -0.05517578125,
1250
- "rewards/margins": 0.11457519233226776,
1251
- "rewards/rejected": -0.16972656548023224,
1252
  "step": 690
1253
  },
1254
  {
1255
  "epoch": 0.9032258064516129,
1256
- "grad_norm": 5.301776773779751,
1257
- "learning_rate": 3.782347372361169e-07,
1258
- "log_odds_chosen": 2.557812452316284,
1259
- "log_odds_ratio": -0.14824219048023224,
1260
- "logits/chosen": -1.569726586341858,
1261
- "logits/rejected": 15.568750381469727,
1262
- "logps/chosen": -1.013671875,
1263
- "logps/rejected": -3.184375047683716,
1264
- "loss": 1.0306,
1265
- "nll_loss": 1.023046851158142,
1266
- "rewards/accuracies": 0.981249988079071,
1267
- "rewards/chosen": -0.05073242262005806,
1268
- "rewards/margins": 0.1083984375,
1269
- "rewards/rejected": -0.1591796875,
1270
  "step": 700
1271
  },
1272
  {
1273
  "epoch": 0.9161290322580645,
1274
- "grad_norm": 5.552358875749227,
1275
- "learning_rate": 3.7555788309433984e-07,
1276
- "log_odds_chosen": 2.6640625,
1277
- "log_odds_ratio": -0.16386719048023224,
1278
- "logits/chosen": -1.9296875,
1279
- "logits/rejected": 15.518750190734863,
1280
- "logps/chosen": -1.0304687023162842,
1281
- "logps/rejected": -3.3265624046325684,
1282
- "loss": 1.0624,
1283
- "nll_loss": 1.0546875,
1284
- "rewards/accuracies": 0.949999988079071,
1285
- "rewards/chosen": -0.05148925632238388,
1286
- "rewards/margins": 0.11484374850988388,
1287
- "rewards/rejected": -0.16640624403953552,
1288
  "step": 710
1289
  },
1290
  {
1291
  "epoch": 0.9290322580645162,
1292
- "grad_norm": 4.910437189800529,
1293
- "learning_rate": 3.7293707031410074e-07,
1294
- "log_odds_chosen": 2.84375,
1295
- "log_odds_ratio": -0.1279296875,
1296
- "logits/chosen": -1.5740234851837158,
1297
- "logits/rejected": 15.693750381469727,
1298
- "logps/chosen": -1.0148437023162842,
1299
- "logps/rejected": -3.4671874046325684,
1300
- "loss": 1.044,
1301
- "nll_loss": 1.0378906726837158,
1302
- "rewards/accuracies": 0.987500011920929,
1303
- "rewards/chosen": -0.05075683444738388,
1304
- "rewards/margins": 0.12255859375,
1305
- "rewards/rejected": -0.17324218153953552,
1306
  "step": 720
1307
  },
1308
  {
1309
  "epoch": 0.9419354838709677,
1310
- "grad_norm": 5.736860098919117,
1311
- "learning_rate": 3.703703703703703e-07,
1312
- "log_odds_chosen": 2.5562500953674316,
1313
- "log_odds_ratio": -0.157470703125,
1314
- "logits/chosen": -1.914648413658142,
1315
- "logits/rejected": 15.518750190734863,
1316
- "logps/chosen": -1.027734398841858,
1317
- "logps/rejected": -3.207812547683716,
1318
- "loss": 1.0585,
1319
- "nll_loss": 1.051171898841858,
1320
- "rewards/accuracies": 0.987500011920929,
1321
- "rewards/chosen": -0.0513916015625,
1322
- "rewards/margins": 0.10898437350988388,
1323
- "rewards/rejected": -0.16035155951976776,
1324
  "step": 730
1325
  },
1326
  {
1327
  "epoch": 0.9548387096774194,
1328
- "grad_norm": 5.692342253961986,
1329
- "learning_rate": 3.678559463888548e-07,
1330
- "log_odds_chosen": 2.5078125,
1331
- "log_odds_ratio": -0.15683594346046448,
1332
- "logits/chosen": -2.1890625953674316,
1333
- "logits/rejected": 15.087499618530273,
1334
- "logps/chosen": -1.0089843273162842,
1335
- "logps/rejected": -3.129687547683716,
1336
- "loss": 1.0155,
1337
- "nll_loss": 1.007421851158142,
1338
- "rewards/accuracies": 0.96875,
1339
- "rewards/chosen": -0.05043945461511612,
1340
- "rewards/margins": 0.10600586235523224,
1341
- "rewards/rejected": -0.15634766221046448,
1342
  "step": 740
1343
  },
1344
  {
1345
  "epoch": 0.967741935483871,
1346
- "grad_norm": 5.3505954948967105,
1347
- "learning_rate": 3.653920476209348e-07,
1348
- "log_odds_chosen": 2.768749952316284,
1349
- "log_odds_ratio": -0.12919922173023224,
1350
- "logits/chosen": -1.8494141101837158,
1351
- "logits/rejected": 15.756250381469727,
1352
- "logps/chosen": -1.056249976158142,
1353
- "logps/rejected": -3.4312500953674316,
1354
- "loss": 1.0808,
1355
- "nll_loss": 1.07421875,
1356
- "rewards/accuracies": 0.9624999761581421,
1357
- "rewards/chosen": -0.05275879055261612,
1358
- "rewards/margins": 0.118896484375,
1359
- "rewards/rejected": -0.17167969048023224,
1360
  "step": 750
1361
  },
1362
  {
1363
  "epoch": 0.9806451612903225,
1364
- "grad_norm": 5.3186211134335935,
1365
- "learning_rate": 3.629770043203138e-07,
1366
- "log_odds_chosen": 2.6187500953674316,
1367
- "log_odds_ratio": -0.16599121689796448,
1368
- "logits/chosen": -1.660253882408142,
1369
- "logits/rejected": 15.524999618530273,
1370
- "logps/chosen": -1.033203125,
1371
- "logps/rejected": -3.2734375,
1372
- "loss": 1.0639,
1373
- "nll_loss": 1.0558593273162842,
1374
- "rewards/accuracies": 0.949999988079071,
1375
- "rewards/chosen": -0.0516357421875,
1376
- "rewards/margins": 0.11201171576976776,
1377
- "rewards/rejected": -0.16367188096046448,
1378
  "step": 760
1379
  },
1380
  {
1381
  "epoch": 0.9935483870967742,
1382
- "grad_norm": 6.551111857556524,
1383
- "learning_rate": 3.6060922298730937e-07,
1384
- "log_odds_chosen": 2.6328125,
1385
- "log_odds_ratio": -0.16083984076976776,
1386
- "logits/chosen": -2.123828172683716,
1387
- "logits/rejected": 15.381250381469727,
1388
- "logps/chosen": -1.0390625,
1389
- "logps/rejected": -3.2890625,
1390
- "loss": 1.0635,
1391
- "nll_loss": 1.056249976158142,
1392
- "rewards/accuracies": 0.9624999761581421,
1393
- "rewards/chosen": -0.05195312574505806,
1394
- "rewards/margins": 0.11245117336511612,
1395
- "rewards/rejected": -0.16435547173023224,
1396
  "step": 770
1397
  },
1398
  {
1399
  "epoch": 1.0,
1400
  "step": 775,
1401
  "total_flos": 0.0,
1402
- "train_loss": 1.1825333921370969,
1403
- "train_runtime": 1561.4009,
1404
- "train_samples_per_second": 7.942,
1405
- "train_steps_per_second": 0.496
1406
  }
1407
  ],
1408
  "logging_steps": 10,
 
11
  "log_history": [
12
  {
13
  "epoch": 0.012903225806451613,
14
+ "grad_norm": 24.415898031026043,
15
+ "learning_rate": 4.5000000000000003e-07,
16
+ "log_odds_chosen": 1.075781226158142,
17
+ "log_odds_ratio": -0.3363281190395355,
18
+ "logits/chosen": -1.048828125,
19
+ "logits/rejected": 15.6875,
20
+ "logps/chosen": -1.6554687023162842,
21
+ "logps/rejected": -2.598437547683716,
22
+ "loss": 2.0046,
23
+ "nll_loss": 1.986718773841858,
24
  "rewards/accuracies": 0.949999988079071,
25
+ "rewards/chosen": -0.082763671875,
26
+ "rewards/margins": 0.04719238355755806,
27
+ "rewards/rejected": -0.12983398139476776,
28
  "step": 10
29
  },
30
  {
31
  "epoch": 0.025806451612903226,
32
+ "grad_norm": 13.677769862646576,
33
+ "learning_rate": 9.500000000000001e-07,
34
+ "log_odds_chosen": 1.037109375,
35
+ "log_odds_ratio": -0.34501951932907104,
36
+ "logits/chosen": -0.6742187738418579,
37
+ "logits/rejected": 16.96875,
38
+ "logps/chosen": -1.6023437976837158,
39
+ "logps/rejected": -2.5,
40
+ "loss": 1.8988,
41
+ "nll_loss": 1.8820312023162842,
42
  "rewards/accuracies": 0.9750000238418579,
43
+ "rewards/chosen": -0.08017577975988388,
44
+ "rewards/margins": 0.044921875,
45
+ "rewards/rejected": -0.12504883110523224,
46
  "step": 20
47
  },
48
  {
49
  "epoch": 0.03870967741935484,
50
+ "grad_norm": 10.696630985039008,
51
+ "learning_rate": 1.45e-06,
52
+ "log_odds_chosen": 0.707226574420929,
53
+ "log_odds_ratio": -0.4332031309604645,
54
+ "logits/chosen": -0.3716796934604645,
55
+ "logits/rejected": 16.725000381469727,
56
+ "logps/chosen": -1.485937476158142,
57
+ "logps/rejected": -2.063281297683716,
58
+ "loss": 1.7568,
59
+ "nll_loss": 1.7351562976837158,
60
+ "rewards/accuracies": 0.90625,
61
+ "rewards/chosen": -0.07431640475988388,
62
+ "rewards/margins": 0.02889404259622097,
63
+ "rewards/rejected": -0.10317382961511612,
64
  "step": 30
65
  },
66
  {
67
  "epoch": 0.05161290322580645,
68
+ "grad_norm": 6.801175564639496,
69
+ "learning_rate": 1.9500000000000004e-06,
70
+ "log_odds_chosen": 0.49785155057907104,
71
+ "log_odds_ratio": -0.498046875,
72
+ "logits/chosen": 0.8692382574081421,
73
+ "logits/rejected": 17.21875,
74
+ "logps/chosen": -1.31640625,
75
+ "logps/rejected": -1.7062499523162842,
76
+ "loss": 1.5594,
77
+ "nll_loss": 1.533593773841858,
78
+ "rewards/accuracies": 0.862500011920929,
79
+ "rewards/chosen": -0.06594238430261612,
80
+ "rewards/margins": 0.019378662109375,
81
+ "rewards/rejected": -0.08525390923023224,
82
  "step": 40
83
  },
84
  {
85
  "epoch": 0.06451612903225806,
86
+ "grad_norm": 5.069050171950685,
87
+ "learning_rate": 2.4500000000000003e-06,
88
+ "log_odds_chosen": 0.35029298067092896,
89
+ "log_odds_ratio": -0.5658203363418579,
90
+ "logits/chosen": 1.32763671875,
91
+ "logits/rejected": 17.381250381469727,
92
+ "logps/chosen": -1.22265625,
93
+ "logps/rejected": -1.489843726158142,
94
+ "loss": 1.3957,
95
+ "nll_loss": 1.3671875,
96
+ "rewards/accuracies": 0.7437499761581421,
97
+ "rewards/chosen": -0.06108398362994194,
98
+ "rewards/margins": 0.013356017880141735,
99
+ "rewards/rejected": -0.07441405951976776,
100
  "step": 50
101
  },
102
  {
103
  "epoch": 0.07741935483870968,
104
+ "grad_norm": 5.83545698860985,
105
+ "learning_rate": 2.95e-06,
106
+ "log_odds_chosen": 0.19368895888328552,
107
+ "log_odds_ratio": -0.637890636920929,
108
+ "logits/chosen": 0.9898437261581421,
109
+ "logits/rejected": 16.862499237060547,
110
+ "logps/chosen": -1.1828124523162842,
111
+ "logps/rejected": -1.33203125,
112
+ "loss": 1.3299,
113
+ "nll_loss": 1.296875,
114
+ "rewards/accuracies": 0.625,
115
+ "rewards/chosen": -0.05915527418255806,
116
+ "rewards/margins": 0.007449341006577015,
117
+ "rewards/rejected": -0.06660155951976776,
118
  "step": 60
119
  },
120
  {
121
  "epoch": 0.09032258064516129,
122
+ "grad_norm": 5.270396473856655,
123
+ "learning_rate": 3.45e-06,
124
+ "log_odds_chosen": 0.28056639432907104,
125
+ "log_odds_ratio": -0.590624988079071,
126
+ "logits/chosen": -0.01455078087747097,
127
+ "logits/rejected": 16.762500762939453,
128
+ "logps/chosen": -1.068750023841858,
129
+ "logps/rejected": -1.275781273841858,
130
+ "loss": 1.1896,
131
+ "nll_loss": 1.1593749523162842,
132
+ "rewards/accuracies": 0.699999988079071,
133
+ "rewards/chosen": -0.053466796875,
134
+ "rewards/margins": 0.010417175479233265,
135
+ "rewards/rejected": -0.06386718899011612,
136
  "step": 70
137
  },
138
  {
139
  "epoch": 0.1032258064516129,
140
+ "grad_norm": 5.648914075324417,
141
+ "learning_rate": 3.95e-06,
142
+ "log_odds_chosen": 0.39033204317092896,
143
+ "log_odds_ratio": -0.55078125,
144
+ "logits/chosen": -0.808667004108429,
145
+ "logits/rejected": 16.25,
146
+ "logps/chosen": -1.0945312976837158,
147
+ "logps/rejected": -1.372656226158142,
148
+ "loss": 1.1625,
149
+ "nll_loss": 1.13671875,
150
+ "rewards/accuracies": 0.7875000238418579,
151
+ "rewards/chosen": -0.0546875,
152
+ "rewards/margins": 0.01395568810403347,
153
+ "rewards/rejected": -0.06857910007238388,
154
  "step": 80
155
  },
156
  {
157
  "epoch": 0.11612903225806452,
158
+ "grad_norm": 4.665506563337419,
159
+ "learning_rate": 4.450000000000001e-06,
160
+ "log_odds_chosen": 1.002343773841858,
161
+ "log_odds_ratio": -0.35888671875,
162
+ "logits/chosen": -1.157812476158142,
163
+ "logits/rejected": 15.862500190734863,
164
+ "logps/chosen": -1.06640625,
165
+ "logps/rejected": -1.828125,
166
+ "loss": 1.1173,
167
+ "nll_loss": 1.099609375,
168
+ "rewards/accuracies": 0.9437500238418579,
169
+ "rewards/chosen": -0.05329589918255806,
170
+ "rewards/margins": 0.03801269456744194,
171
+ "rewards/rejected": -0.09135742485523224,
172
  "step": 90
173
  },
174
  {
175
  "epoch": 0.12903225806451613,
176
+ "grad_norm": 4.9194277923750285,
177
+ "learning_rate": 4.95e-06,
178
+ "log_odds_chosen": 1.3742187023162842,
179
+ "log_odds_ratio": -0.29130858182907104,
180
+ "logits/chosen": -1.532812476158142,
181
+ "logits/rejected": 15.25,
182
+ "logps/chosen": -1.092187523841858,
183
+ "logps/rejected": -2.2007813453674316,
184
+ "loss": 1.1352,
185
+ "nll_loss": 1.1218750476837158,
186
+ "rewards/accuracies": 0.8999999761581421,
187
+ "rewards/chosen": -0.05461425706744194,
188
+ "rewards/margins": 0.0555419921875,
189
+ "rewards/rejected": -0.110107421875,
190
  "step": 100
191
  },
192
  {
193
  "epoch": 0.14193548387096774,
194
+ "grad_norm": 4.9620026922411204,
195
+ "learning_rate": 4.789131426105757e-06,
196
+ "log_odds_chosen": 1.75390625,
197
+ "log_odds_ratio": -0.212890625,
198
+ "logits/chosen": -0.905517578125,
199
+ "logits/rejected": 16.043750762939453,
200
+ "logps/chosen": -1.085546851158142,
201
+ "logps/rejected": -2.515625,
202
+ "loss": 1.1138,
203
+ "nll_loss": 1.103124976158142,
204
+ "rewards/accuracies": 0.981249988079071,
205
+ "rewards/chosen": -0.05424804612994194,
206
+ "rewards/margins": 0.07148437201976776,
207
+ "rewards/rejected": -0.125732421875,
208
  "step": 110
209
  },
210
  {
211
  "epoch": 0.15483870967741936,
212
+ "grad_norm": 4.526815051594843,
213
+ "learning_rate": 4.583492485141056e-06,
214
+ "log_odds_chosen": 1.5597655773162842,
215
+ "log_odds_ratio": -0.2994140684604645,
216
+ "logits/chosen": -1.553466796875,
217
+ "logits/rejected": 16.456249237060547,
218
+ "logps/chosen": -1.057031273841858,
219
+ "logps/rejected": -2.3335938453674316,
220
+ "loss": 1.0869,
221
+ "nll_loss": 1.071874976158142,
222
+ "rewards/accuracies": 0.8812500238418579,
223
+ "rewards/chosen": -0.05280761793255806,
224
+ "rewards/margins": 0.06374511867761612,
225
+ "rewards/rejected": -0.11655273288488388,
226
  "step": 120
227
  },
228
  {
229
  "epoch": 0.16774193548387098,
230
+ "grad_norm": 4.9259847932667595,
231
+ "learning_rate": 4.4022545316281195e-06,
232
+ "log_odds_chosen": 1.9718749523162842,
233
+ "log_odds_ratio": -0.19765624403953552,
234
+ "logits/chosen": -0.6851562261581421,
235
+ "logits/rejected": 16.143749237060547,
236
+ "logps/chosen": -1.0867187976837158,
237
+ "logps/rejected": -2.7265625,
238
+ "loss": 1.1121,
239
+ "nll_loss": 1.103124976158142,
240
+ "rewards/accuracies": 0.956250011920929,
241
+ "rewards/chosen": -0.0543212890625,
242
+ "rewards/margins": 0.08188476413488388,
243
+ "rewards/rejected": -0.13618163764476776,
244
  "step": 130
245
  },
246
  {
247
  "epoch": 0.18064516129032257,
248
+ "grad_norm": 5.039653421145649,
249
+ "learning_rate": 4.240944648399854e-06,
250
+ "log_odds_chosen": 1.8359375,
251
+ "log_odds_ratio": -0.20595702528953552,
252
+ "logits/chosen": -1.1907227039337158,
253
+ "logits/rejected": 15.668749809265137,
254
+ "logps/chosen": -1.072656273841858,
255
+ "logps/rejected": -2.571093797683716,
256
+ "loss": 1.0928,
257
+ "nll_loss": 1.08203125,
258
+ "rewards/accuracies": 0.9750000238418579,
259
+ "rewards/chosen": -0.05368652194738388,
260
+ "rewards/margins": 0.07490234076976776,
261
+ "rewards/rejected": -0.128662109375,
262
  "step": 140
263
  },
264
  {
265
  "epoch": 0.1935483870967742,
266
+ "grad_norm": 4.8956379810361375,
267
+ "learning_rate": 4.096159602595203e-06,
268
+ "log_odds_chosen": 1.8679687976837158,
269
+ "log_odds_ratio": -0.22802734375,
270
+ "logits/chosen": -1.8488280773162842,
271
+ "logits/rejected": 15.550000190734863,
272
+ "logps/chosen": -1.05078125,
273
+ "logps/rejected": -2.5999999046325684,
274
+ "loss": 1.0779,
275
+ "nll_loss": 1.0656249523162842,
276
+ "rewards/accuracies": 0.96875,
277
+ "rewards/chosen": -0.0526123046875,
278
+ "rewards/margins": 0.07744140923023224,
279
+ "rewards/rejected": -0.12998047471046448,
280
  "step": 150
281
  },
282
  {
283
  "epoch": 0.2064516129032258,
284
+ "grad_norm": 5.648955197242222,
285
+ "learning_rate": 3.965257928590721e-06,
286
+ "log_odds_chosen": 2.026562452316284,
287
+ "log_odds_ratio": -0.21791991591453552,
288
+ "logits/chosen": -2.2445311546325684,
289
+ "logits/rejected": 14.962499618530273,
290
+ "logps/chosen": -1.0164062976837158,
291
+ "logps/rejected": -2.700000047683716,
292
+ "loss": 1.0501,
293
+ "nll_loss": 1.0402343273162842,
294
+ "rewards/accuracies": 0.96875,
295
+ "rewards/chosen": -0.05083007737994194,
296
+ "rewards/margins": 0.084228515625,
297
+ "rewards/rejected": -0.135009765625,
298
  "step": 160
299
  },
300
  {
301
  "epoch": 0.21935483870967742,
302
+ "grad_norm": 4.50921729284181,
303
+ "learning_rate": 3.846153846153846e-06,
304
+ "log_odds_chosen": 2.225781202316284,
305
+ "log_odds_ratio": -0.15439453721046448,
306
+ "logits/chosen": -2.3375000953674316,
307
+ "logits/rejected": 16.0625,
308
+ "logps/chosen": -0.977734386920929,
309
+ "logps/rejected": -2.809375047683716,
310
+ "loss": 1.0144,
311
+ "nll_loss": 1.006250023841858,
312
+ "rewards/accuracies": 0.9937499761581421,
313
+ "rewards/chosen": -0.04885254055261612,
314
+ "rewards/margins": 0.09160156548023224,
315
+ "rewards/rejected": -0.14057616889476776,
316
  "step": 170
317
  },
318
  {
319
  "epoch": 0.23225806451612904,
320
+ "grad_norm": 5.153444368200689,
321
+ "learning_rate": 3.7371754637596797e-06,
322
+ "log_odds_chosen": 2.354687452316284,
323
+ "log_odds_ratio": -0.14643554389476776,
324
+ "logits/chosen": -1.9169921875,
325
+ "logits/rejected": 15.456250190734863,
326
+ "logps/chosen": -1.01953125,
327
+ "logps/rejected": -2.981250047683716,
328
+ "loss": 1.0307,
329
+ "nll_loss": 1.0234375,
330
+ "rewards/accuracies": 0.981249988079071,
331
+ "rewards/chosen": -0.05097656324505806,
332
+ "rewards/margins": 0.09814453125,
333
+ "rewards/rejected": -0.14912109076976776,
334
  "step": 180
335
  },
336
  {
337
  "epoch": 0.24516129032258063,
338
+ "grad_norm": 5.596223310698167,
339
+ "learning_rate": 3.63696483726654e-06,
340
+ "log_odds_chosen": 2.479687452316284,
341
+ "log_odds_ratio": -0.14628906548023224,
342
+ "logits/chosen": -1.4436156749725342,
343
+ "logits/rejected": 15.90625,
344
+ "logps/chosen": -1.0,
345
+ "logps/rejected": -3.0859375,
346
+ "loss": 1.0302,
347
+ "nll_loss": 1.023046851158142,
348
+ "rewards/accuracies": 0.96875,
349
+ "rewards/chosen": -0.04997558519244194,
350
+ "rewards/margins": 0.10419921576976776,
351
+ "rewards/rejected": -0.15419921278953552,
352
  "step": 190
353
  },
354
  {
355
  "epoch": 0.25806451612903225,
356
+ "grad_norm": 4.6505717562701765,
357
+ "learning_rate": 3.5444060250416796e-06,
358
+ "log_odds_chosen": 2.6781249046325684,
359
+ "log_odds_ratio": -0.11782226711511612,
360
+ "logits/chosen": -1.956640601158142,
361
+ "logits/rejected": 16.037500381469727,
362
+ "logps/chosen": -1.033203125,
363
+ "logps/rejected": -3.3046875,
364
+ "loss": 1.0531,
365
+ "nll_loss": 1.0480468273162842,
366
+ "rewards/accuracies": 0.9937499761581421,
367
+ "rewards/chosen": -0.0516357421875,
368
+ "rewards/margins": 0.11372070014476776,
369
+ "rewards/rejected": -0.16542968153953552,
370
  "step": 200
371
  },
372
  {
373
  "epoch": 0.2709677419354839,
374
+ "grad_norm": 5.056481299672191,
375
+ "learning_rate": 3.458572319330373e-06,
376
+ "log_odds_chosen": 2.6109375953674316,
377
+ "log_odds_ratio": -0.16538086533546448,
378
+ "logits/chosen": -2.354687452316284,
379
+ "logits/rejected": 15.25,
380
+ "logps/chosen": -1.021875023841858,
381
+ "logps/rejected": -3.239062547683716,
382
+ "loss": 1.0347,
383
+ "nll_loss": 1.0265624523162842,
384
+ "rewards/accuracies": 0.9624999761581421,
385
+ "rewards/chosen": -0.05104980617761612,
386
+ "rewards/margins": 0.11093749850988388,
387
+ "rewards/rejected": -0.162109375,
388
  "step": 210
389
  },
390
  {
391
  "epoch": 0.2838709677419355,
392
+ "grad_norm": 4.888314219389848,
393
+ "learning_rate": 3.3786868919974304e-06,
394
+ "log_odds_chosen": 2.6578125953674316,
395
+ "log_odds_ratio": -0.17612305283546448,
396
+ "logits/chosen": -1.467187523841858,
397
+ "logits/rejected": 15.787500381469727,
398
+ "logps/chosen": -1.0832030773162842,
399
+ "logps/rejected": -3.378124952316284,
400
+ "loss": 1.095,
401
+ "nll_loss": 1.0867187976837158,
402
+ "rewards/accuracies": 0.9312499761581421,
403
+ "rewards/chosen": -0.05415039137005806,
404
+ "rewards/margins": 0.11494140326976776,
405
+ "rewards/rejected": -0.16904297471046448,
406
  "step": 220
407
  },
408
  {
409
  "epoch": 0.2967741935483871,
410
+ "grad_norm": 4.697252316410647,
411
+ "learning_rate": 3.3040930022754493e-06,
412
+ "log_odds_chosen": 2.167187452316284,
413
+ "log_odds_ratio": -0.19511719048023224,
414
+ "logits/chosen": -1.1691405773162842,
415
+ "logits/rejected": 15.512499809265137,
416
+ "logps/chosen": -1.071874976158142,
417
+ "logps/rejected": -2.8968749046325684,
418
+ "loss": 1.0854,
419
+ "nll_loss": 1.0773437023162842,
420
+ "rewards/accuracies": 0.96875,
421
+ "rewards/chosen": -0.05363769456744194,
422
+ "rewards/margins": 0.09116210788488388,
423
+ "rewards/rejected": -0.14492186903953552,
424
  "step": 230
425
  },
426
  {
427
  "epoch": 0.3096774193548387,
428
+ "grad_norm": 4.577631356182271,
429
+ "learning_rate": 3.234231136765755e-06,
430
+ "log_odds_chosen": 2.2632813453674316,
431
+ "log_odds_ratio": -0.19497069716453552,
432
+ "logits/chosen": -2.502148389816284,
433
+ "logits/rejected": 14.100000381469727,
434
+ "logps/chosen": -1.017968773841858,
435
+ "logps/rejected": -2.9296875,
436
+ "loss": 1.048,
437
+ "nll_loss": 1.0378906726837158,
438
+ "rewards/accuracies": 0.987500011920929,
439
+ "rewards/chosen": -0.05097656324505806,
440
+ "rewards/margins": 0.09550781548023224,
441
+ "rewards/rejected": -0.14628906548023224,
442
  "step": 240
443
  },
444
  {
445
  "epoch": 0.3225806451612903,
446
+ "grad_norm": 4.858199223746265,
447
+ "learning_rate": 3.1686212526223897e-06,
448
+ "log_odds_chosen": 2.8984375,
449
+ "log_odds_ratio": -0.1181640625,
450
+ "logits/chosen": -2.688671827316284,
451
+ "logits/rejected": 15.199999809265137,
452
+ "logps/chosen": -0.9878906011581421,
453
+ "logps/rejected": -3.464062452316284,
454
+ "loss": 1.0084,
455
+ "nll_loss": 1.0031249523162842,
456
  "rewards/accuracies": 0.981249988079071,
457
+ "rewards/chosen": -0.0494384765625,
458
+ "rewards/margins": 0.1240234375,
459
+ "rewards/rejected": -0.17333984375,
460
  "step": 250
461
  },
462
  {
463
  "epoch": 0.33548387096774196,
464
+ "grad_norm": 4.519869758016946,
465
+ "learning_rate": 3.1068488300060003e-06,
466
+ "log_odds_chosen": 2.4437499046325684,
467
+ "log_odds_ratio": -0.14501953125,
468
+ "logits/chosen": -2.409374952316284,
469
+ "logits/rejected": 15.206250190734863,
470
+ "logps/chosen": -1.005859375,
471
+ "logps/rejected": -3.0453124046325684,
472
+ "loss": 1.03,
473
+ "nll_loss": 1.0222656726837158,
474
+ "rewards/accuracies": 0.987500011920929,
475
+ "rewards/chosen": -0.05021972581744194,
476
+ "rewards/margins": 0.10209961235523224,
477
+ "rewards/rejected": -0.15224608778953552,
478
  "step": 260
479
  },
480
  {
481
  "epoch": 0.34838709677419355,
482
+ "grad_norm": 5.239684216445077,
483
+ "learning_rate": 3.0485538042484623e-06,
484
+ "log_odds_chosen": 2.598437547683716,
485
+ "log_odds_ratio": -0.14907225966453552,
486
+ "logits/chosen": -2.1636719703674316,
487
+ "logits/rejected": 15.050000190734863,
488
+ "logps/chosen": -0.981640636920929,
489
+ "logps/rejected": -3.1781249046325684,
490
+ "loss": 1.0049,
491
+ "nll_loss": 0.996874988079071,
492
+ "rewards/accuracies": 0.987500011920929,
493
+ "rewards/chosen": -0.049072265625,
494
+ "rewards/margins": 0.10976562649011612,
495
+ "rewards/rejected": -0.15888671576976776,
496
  "step": 270
497
  },
498
  {
499
  "epoch": 0.36129032258064514,
500
+ "grad_norm": 5.276617193754623,
501
+ "learning_rate": 2.9934217004462487e-06,
502
+ "log_odds_chosen": 2.9078125953674316,
503
+ "log_odds_ratio": -0.12812499701976776,
504
+ "logits/chosen": -2.070117235183716,
505
+ "logits/rejected": 15.25,
506
+ "logps/chosen": -1.036718726158142,
507
+ "logps/rejected": -3.535937547683716,
508
+ "loss": 1.0451,
509
+ "nll_loss": 1.0390625,
510
+ "rewards/accuracies": 0.9750000238418579,
511
+ "rewards/chosen": -0.05183105543255806,
512
+ "rewards/margins": 0.12495116889476776,
513
+ "rewards/rejected": -0.1767578125,
514
  "step": 280
515
  },
516
  {
517
  "epoch": 0.3741935483870968,
518
+ "grad_norm": 4.901202785480476,
519
+ "learning_rate": 2.9411764705882355e-06,
520
+ "log_odds_chosen": 2.957812547683716,
521
+ "log_odds_ratio": -0.11459960788488388,
522
+ "logits/chosen": -2.172070264816284,
523
+ "logits/rejected": 15.556249618530273,
524
+ "logps/chosen": -0.998046875,
525
+ "logps/rejected": -3.5250000953674316,
526
+ "loss": 1.0108,
527
+ "nll_loss": 1.0050780773162842,
528
  "rewards/accuracies": 0.981249988079071,
529
+ "rewards/chosen": -0.04987793043255806,
530
+ "rewards/margins": 0.12641601264476776,
531
+ "rewards/rejected": -0.17607422173023224,
532
  "step": 290
533
  },
534
  {
535
  "epoch": 0.3870967741935484,
536
+ "grad_norm": 6.3489235644192314,
537
+ "learning_rate": 2.891574659831201e-06,
538
+ "log_odds_chosen": 3.4140625,
539
+ "log_odds_ratio": -0.12001953274011612,
540
+ "logits/chosen": -2.2777342796325684,
541
+ "logits/rejected": 15.387499809265137,
542
+ "logps/chosen": -1.029687523841858,
543
+ "logps/rejected": -4.029687404632568,
544
+ "loss": 1.0416,
545
+ "nll_loss": 1.037109375,
546
+ "rewards/accuracies": 0.949999988079071,
547
+ "rewards/chosen": -0.05146484449505806,
548
+ "rewards/margins": 0.15004882216453552,
549
+ "rewards/rejected": -0.20146484673023224,
550
  "step": 300
551
  },
552
  {
553
  "epoch": 0.4,
554
+ "grad_norm": 5.3538840183503815,
555
+ "learning_rate": 2.844400619942872e-06,
556
+ "log_odds_chosen": 2.953125,
557
+ "log_odds_ratio": -0.14028319716453552,
558
+ "logits/chosen": -2.112109422683716,
559
+ "logits/rejected": 14.90625,
560
+ "logps/chosen": -0.994921863079071,
561
+ "logps/rejected": -3.535937547683716,
562
+ "loss": 1.0155,
563
+ "nll_loss": 1.008203148841858,
564
+ "rewards/accuracies": 0.9750000238418579,
565
+ "rewards/chosen": -0.0496826171875,
566
+ "rewards/margins": 0.12700195610523224,
567
+ "rewards/rejected": -0.17685547471046448,
568
  "step": 310
569
  },
570
  {
571
  "epoch": 0.4129032258064516,
572
+ "grad_norm": 5.370323496962624,
573
+ "learning_rate": 2.799462554779272e-06,
574
+ "log_odds_chosen": 3.078125,
575
+ "log_odds_ratio": -0.1260986328125,
576
+ "logits/chosen": -2.033203125,
577
+ "logits/rejected": 15.4375,
578
+ "logps/chosen": -0.99609375,
579
+ "logps/rejected": -3.65625,
580
+ "loss": 1.0071,
581
+ "nll_loss": 1.0007812976837158,
582
+ "rewards/accuracies": 0.96875,
583
+ "rewards/chosen": -0.04982910305261612,
584
+ "rewards/margins": 0.13276366889476776,
585
+ "rewards/rejected": -0.18251952528953552,
586
  "step": 320
587
  },
588
  {
589
  "epoch": 0.4258064516129032,
590
+ "grad_norm": 6.23268929266231,
591
+ "learning_rate": 2.7565892320998566e-06,
592
+ "log_odds_chosen": 3.582812547683716,
593
+ "log_odds_ratio": -0.09384765475988388,
594
+ "logits/chosen": -2.2328124046325684,
595
+ "logits/rejected": 15.524999618530273,
596
+ "logps/chosen": -1.01953125,
597
+ "logps/rejected": -4.173437595367432,
598
+ "loss": 1.0513,
599
+ "nll_loss": 1.046484351158142,
600
+ "rewards/accuracies": 0.987500011920929,
601
+ "rewards/chosen": -0.051025390625,
602
+ "rewards/margins": 0.15756836533546448,
603
+ "rewards/rejected": -0.20859375596046448,
604
  "step": 330
605
  },
606
  {
607
  "epoch": 0.43870967741935485,
608
+ "grad_norm": 5.462484402984074,
609
+ "learning_rate": 2.7156272329678423e-06,
610
+ "log_odds_chosen": 3.59375,
611
+ "log_odds_ratio": -0.0960693359375,
612
+ "logits/chosen": -1.870507836341858,
613
+ "logits/rejected": 15.387499809265137,
614
+ "logps/chosen": -1.039453148841858,
615
+ "logps/rejected": -4.217187404632568,
616
+ "loss": 1.0323,
617
+ "nll_loss": 1.0281250476837158,
618
+ "rewards/accuracies": 0.9937499761581421,
619
+ "rewards/chosen": -0.05192871019244194,
620
+ "rewards/margins": 0.15878906846046448,
621
+ "rewards/rejected": -0.21064452826976776,
622
  "step": 340
623
  },
624
  {
625
  "epoch": 0.45161290322580644,
626
+ "grad_norm": 5.006492348614002,
627
+ "learning_rate": 2.6764386378609465e-06,
628
+ "log_odds_chosen": 3.403125047683716,
629
+ "log_odds_ratio": -0.13632813096046448,
630
+ "logits/chosen": -2.469531297683716,
631
+ "logits/rejected": 15.212499618530273,
632
+ "logps/chosen": -0.9750000238418579,
633
+ "logps/rejected": -3.9609375,
634
+ "loss": 1.0064,
635
+ "nll_loss": 1.000390648841858,
636
  "rewards/accuracies": 0.9624999761581421,
637
+ "rewards/chosen": -0.04877929762005806,
638
+ "rewards/margins": 0.149169921875,
639
+ "rewards/rejected": -0.19794921576976776,
640
  "step": 350
641
  },
642
  {
643
  "epoch": 0.4645161290322581,
644
+ "grad_norm": 5.708462747115479,
645
+ "learning_rate": 2.638899069846298e-06,
646
+ "log_odds_chosen": 3.543750047683716,
647
+ "log_odds_ratio": -0.11757812649011612,
648
+ "logits/chosen": -2.37890625,
649
+ "logits/rejected": 15.28125,
650
+ "logps/chosen": -1.0078125,
651
+ "logps/rejected": -4.129687309265137,
652
+ "loss": 1.0219,
653
+ "nll_loss": 1.0167968273162842,
654
+ "rewards/accuracies": 0.96875,
655
+ "rewards/chosen": -0.05039062350988388,
656
+ "rewards/margins": 0.15629883110523224,
657
+ "rewards/rejected": -0.20654296875,
658
  "step": 360
659
  },
660
  {
661
  "epoch": 0.4774193548387097,
662
+ "grad_norm": 15.819166923351714,
663
+ "learning_rate": 2.6028960314767675e-06,
664
+ "log_odds_chosen": 3.362499952316284,
665
+ "log_odds_ratio": -0.13581542670726776,
666
+ "logits/chosen": -2.4253907203674316,
667
+ "logits/rejected": 15.675000190734863,
668
+ "logps/chosen": -1.008203148841858,
669
+ "logps/rejected": -3.965625047683716,
670
+ "loss": 1.0221,
671
+ "nll_loss": 1.0148437023162842,
672
+ "rewards/accuracies": 0.956250011920929,
673
+ "rewards/chosen": -0.05039062350988388,
674
+ "rewards/margins": 0.14765624701976776,
675
+ "rewards/rejected": -0.1982421875,
676
  "step": 370
677
  },
678
  {
679
  "epoch": 0.49032258064516127,
680
+ "grad_norm": 4.605973549061064,
681
+ "learning_rate": 2.568327484687571e-06,
682
+ "log_odds_chosen": 2.30859375,
683
+ "log_odds_ratio": -0.17836913466453552,
684
+ "logits/chosen": -2.288281202316284,
685
+ "logits/rejected": 15.518750190734863,
686
+ "logps/chosen": -0.975390613079071,
687
+ "logps/rejected": -2.8828125,
688
+ "loss": 0.9912,
689
+ "nll_loss": 0.981640636920929,
690
+ "rewards/accuracies": 0.956250011920929,
691
+ "rewards/chosen": -0.0487060546875,
692
+ "rewards/margins": 0.09526367485523224,
693
+ "rewards/rejected": -0.14399413764476776,
694
  "step": 380
695
  },
696
  {
697
  "epoch": 0.5032258064516129,
698
+ "grad_norm": 5.353525659101831,
699
+ "learning_rate": 2.535100632816969e-06,
700
+ "log_odds_chosen": 2.34765625,
701
+ "log_odds_ratio": -0.158203125,
702
+ "logits/chosen": -2.428906202316284,
703
+ "logits/rejected": 15.625,
704
+ "logps/chosen": -0.9789062738418579,
705
+ "logps/rejected": -2.9312500953674316,
706
+ "loss": 1.0115,
707
+ "nll_loss": 1.003515601158142,
708
+ "rewards/accuracies": 0.96875,
709
+ "rewards/chosen": -0.04892577975988388,
710
+ "rewards/margins": 0.09760741889476776,
711
+ "rewards/rejected": -0.14653320610523224,
712
  "step": 390
713
  },
714
  {
715
  "epoch": 0.5161290322580645,
716
+ "grad_norm": 4.669342637045956,
717
+ "learning_rate": 2.5031308716087944e-06,
718
+ "log_odds_chosen": 2.7562499046325684,
719
+ "log_odds_ratio": -0.12431640923023224,
720
+ "logits/chosen": -2.8046875,
721
+ "logits/rejected": 15.206250190734863,
722
+ "logps/chosen": -0.9898437261581421,
723
+ "logps/rejected": -3.317187547683716,
724
+ "loss": 1.0071,
725
+ "nll_loss": 1.0011718273162842,
726
+ "rewards/accuracies": 0.981249988079071,
727
+ "rewards/chosen": -0.04946289211511612,
728
+ "rewards/margins": 0.11630859225988388,
729
+ "rewards/rejected": -0.16591796278953552,
730
  "step": 400
731
  },
732
  {
733
  "epoch": 0.5290322580645161,
734
+ "grad_norm": 5.425862615574437,
735
+ "learning_rate": 2.472340882170744e-06,
736
+ "log_odds_chosen": 3.4671874046325684,
737
+ "log_odds_ratio": -0.08171387016773224,
738
+ "logits/chosen": -2.4320311546325684,
739
+ "logits/rejected": 15.337499618530273,
740
+ "logps/chosen": -0.987500011920929,
741
+ "logps/rejected": -4.004687309265137,
742
+ "loss": 0.9981,
743
+ "nll_loss": 0.993359386920929,
744
+ "rewards/accuracies": 1.0,
745
+ "rewards/chosen": -0.04938964918255806,
746
+ "rewards/margins": 0.15092773735523224,
747
+ "rewards/rejected": -0.20039062201976776,
748
  "step": 410
749
  },
750
  {
751
  "epoch": 0.5419354838709678,
752
+ "grad_norm": 5.859653969362379,
753
+ "learning_rate": 2.4426598437301577e-06,
754
+ "log_odds_chosen": 3.825000047683716,
755
+ "log_odds_ratio": -0.10820312798023224,
756
+ "logits/chosen": -2.335156202316284,
757
+ "logits/rejected": 15.056249618530273,
758
+ "logps/chosen": -1.012109398841858,
759
+ "logps/rejected": -4.428124904632568,
760
+ "loss": 1.0235,
761
+ "nll_loss": 1.019140601158142,
762
+ "rewards/accuracies": 0.9624999761581421,
763
+ "rewards/chosen": -0.05058593675494194,
764
+ "rewards/margins": 0.17080077528953552,
765
+ "rewards/rejected": -0.22119140625,
766
  "step": 420
767
  },
768
  {
769
  "epoch": 0.5548387096774193,
770
+ "grad_norm": 14.090172688528586,
771
+ "learning_rate": 2.4140227479263382e-06,
772
+ "log_odds_chosen": 3.9375,
773
+ "log_odds_ratio": -0.09453125298023224,
774
+ "logits/chosen": -1.5808594226837158,
775
+ "logits/rejected": 16.018749237060547,
776
+ "logps/chosen": -1.0070312023162842,
777
+ "logps/rejected": -4.506249904632568,
778
+ "loss": 1.0058,
779
+ "nll_loss": 1.002343773841858,
780
+ "rewards/accuracies": 0.987500011920929,
781
+ "rewards/chosen": -0.05031738430261612,
782
+ "rewards/margins": 0.1748046875,
783
+ "rewards/rejected": -0.22519531846046448,
784
  "step": 430
785
  },
786
  {
787
  "epoch": 0.567741935483871,
788
+ "grad_norm": 6.569549660660238,
789
+ "learning_rate": 2.3863697995167388e-06,
790
+ "log_odds_chosen": 3.371875047683716,
791
+ "log_odds_ratio": -0.11165771633386612,
792
+ "logits/chosen": -2.5687499046325684,
793
+ "logits/rejected": 15.324999809265137,
794
+ "logps/chosen": -1.009765625,
795
+ "logps/rejected": -3.965625047683716,
796
+ "loss": 1.0316,
797
+ "nll_loss": 1.0261719226837158,
798
+ "rewards/accuracies": 0.987500011920929,
799
+ "rewards/chosen": -0.05051269382238388,
800
+ "rewards/margins": 0.14790038764476776,
801
+ "rewards/rejected": -0.19843749701976776,
802
  "step": 440
803
  },
804
  {
805
  "epoch": 0.5806451612903226,
806
+ "grad_norm": 5.065013160531643,
807
+ "learning_rate": 2.359645890915044e-06,
808
+ "log_odds_chosen": 3.518749952316284,
809
+ "log_odds_ratio": -0.11590576171875,
810
+ "logits/chosen": -2.403515577316284,
811
+ "logits/rejected": 15.050000190734863,
812
+ "logps/chosen": -1.0281250476837158,
813
+ "logps/rejected": -4.128125190734863,
814
+ "loss": 1.0486,
815
+ "nll_loss": 1.043359398841858,
816
+ "rewards/accuracies": 0.9624999761581421,
817
+ "rewards/chosen": -0.0513916015625,
818
+ "rewards/margins": 0.15498046576976776,
819
+ "rewards/rejected": -0.20644530653953552,
820
  "step": 450
821
  },
822
  {
823
  "epoch": 0.5935483870967742,
824
+ "grad_norm": 5.920025683401155,
825
+ "learning_rate": 2.333800140046683e-06,
826
+ "log_odds_chosen": 4.048437595367432,
827
+ "log_odds_ratio": -0.09355469048023224,
828
+ "logits/chosen": -2.242138624191284,
829
+ "logits/rejected": 15.487500190734863,
830
+ "logps/chosen": -0.994921863079071,
831
+ "logps/rejected": -4.592187404632568,
832
+ "loss": 1.0104,
833
+ "nll_loss": 1.0050780773162842,
834
+ "rewards/accuracies": 0.9624999761581421,
835
+ "rewards/chosen": -0.04975586012005806,
836
+ "rewards/margins": 0.17998047173023224,
837
+ "rewards/rejected": -0.22968749701976776,
838
  "step": 460
839
  },
840
  {
841
  "epoch": 0.6064516129032258,
842
+ "grad_norm": 6.138559547910098,
843
+ "learning_rate": 2.3087854826980507e-06,
844
+ "log_odds_chosen": 3.6640625,
845
+ "log_odds_ratio": -0.1455078125,
846
+ "logits/chosen": -2.6703124046325684,
847
+ "logits/rejected": 14.762499809265137,
848
+ "logps/chosen": -1.03125,
849
+ "logps/rejected": -4.307812690734863,
850
+ "loss": 1.0394,
851
+ "nll_loss": 1.0320312976837158,
852
+ "rewards/accuracies": 0.925000011920929,
853
+ "rewards/chosen": -0.05156249925494194,
854
+ "rewards/margins": 0.16391602158546448,
855
+ "rewards/rejected": -0.21542969346046448,
856
  "step": 470
857
  },
858
  {
859
  "epoch": 0.6193548387096774,
860
+ "grad_norm": 5.567487993441089,
861
+ "learning_rate": 2.2845583119247696e-06,
862
+ "log_odds_chosen": 4.364062309265137,
863
+ "log_odds_ratio": -0.078857421875,
864
+ "logits/chosen": -2.549999952316284,
865
+ "logits/rejected": 15.662500381469727,
866
+ "logps/chosen": -0.9398437738418579,
867
+ "logps/rejected": -4.828125,
868
+ "loss": 0.9611,
869
+ "nll_loss": 0.9574218988418579,
870
+ "rewards/accuracies": 0.9750000238418579,
871
+ "rewards/chosen": -0.04694824293255806,
872
+ "rewards/margins": 0.19462890923023224,
873
+ "rewards/rejected": -0.24169921875,
874
  "step": 480
875
  },
876
  {
877
  "epoch": 0.632258064516129,
878
+ "grad_norm": 5.862239199019293,
879
+ "learning_rate": 2.2610781582306733e-06,
880
+ "log_odds_chosen": 4.1484375,
881
+ "log_odds_ratio": -0.08124999701976776,
882
+ "logits/chosen": -2.7220215797424316,
883
+ "logits/rejected": 15.737500190734863,
884
+ "logps/chosen": -1.0242187976837158,
885
+ "logps/rejected": -4.734375,
886
+ "loss": 1.0391,
887
+ "nll_loss": 1.0343749523162842,
888
+ "rewards/accuracies": 0.9750000238418579,
889
+ "rewards/chosen": -0.05124511569738388,
890
+ "rewards/margins": 0.18544921278953552,
891
+ "rewards/rejected": -0.23681640625,
892
  "step": 490
893
  },
894
  {
895
  "epoch": 0.6451612903225806,
896
+ "grad_norm": 5.118213222200493,
897
+ "learning_rate": 2.238307405179226e-06,
898
+ "log_odds_chosen": 3.965625047683716,
899
+ "log_odds_ratio": -0.10859374701976776,
900
+ "logits/chosen": -2.469921827316284,
901
+ "logits/rejected": 15.243749618530273,
902
+ "logps/chosen": -1.066796898841858,
903
+ "logps/rejected": -4.618750095367432,
904
+ "loss": 1.057,
905
+ "nll_loss": 1.051171898841858,
906
+ "rewards/accuracies": 0.96875,
907
+ "rewards/chosen": -0.05332031100988388,
908
+ "rewards/margins": 0.17753906548023224,
909
+ "rewards/rejected": -0.23095703125,
910
  "step": 500
911
  },
912
  {
913
  "epoch": 0.6580645161290323,
914
+ "grad_norm": 5.512232200800921,
915
+ "learning_rate": 2.2162110358896814e-06,
916
+ "log_odds_chosen": 3.957812547683716,
917
+ "log_odds_ratio": -0.08552245795726776,
918
+ "logits/chosen": -2.5843749046325684,
919
+ "logits/rejected": 15.137499809265137,
920
+ "logps/chosen": -0.9624999761581421,
921
+ "logps/rejected": -4.465624809265137,
922
+ "loss": 0.9945,
923
+ "nll_loss": 0.9906250238418579,
924
+ "rewards/accuracies": 0.981249988079071,
925
+ "rewards/chosen": -0.04819335788488388,
926
+ "rewards/margins": 0.175048828125,
927
+ "rewards/rejected": -0.22314453125,
928
  "step": 510
929
  },
930
  {
931
  "epoch": 0.6709677419354839,
932
+ "grad_norm": 5.353147766002135,
933
+ "learning_rate": 2.1947564065307354e-06,
934
+ "log_odds_chosen": 3.9203124046325684,
935
+ "log_odds_ratio": -0.08530273288488388,
936
+ "logits/chosen": -2.832812547683716,
937
+ "logits/rejected": 15.462499618530273,
938
+ "logps/chosen": -0.9613281488418579,
939
+ "logps/rejected": -4.428124904632568,
940
+ "loss": 0.9806,
941
+ "nll_loss": 0.9765625,
942
+ "rewards/accuracies": 0.9937499761581421,
943
+ "rewards/chosen": -0.04807128757238388,
944
+ "rewards/margins": 0.17333984375,
945
+ "rewards/rejected": -0.22138671576976776,
946
  "step": 520
947
  },
948
  {
949
  "epoch": 0.6838709677419355,
950
+ "grad_norm": 5.53934453000343,
951
+ "learning_rate": 2.173913043478261e-06,
952
+ "log_odds_chosen": 3.6078124046325684,
953
+ "log_odds_ratio": -0.093505859375,
954
+ "logits/chosen": -1.6005859375,
955
+ "logits/rejected": 15.037500381469727,
956
+ "logps/chosen": -0.955859363079071,
957
+ "logps/rejected": -4.104687690734863,
958
+ "loss": 0.9783,
959
+ "nll_loss": 0.973828136920929,
960
+ "rewards/accuracies": 0.9750000238418579,
961
+ "rewards/chosen": -0.04775390774011612,
962
+ "rewards/margins": 0.15732422471046448,
963
+ "rewards/rejected": -0.20488281548023224,
964
  "step": 530
965
  },
966
  {
967
  "epoch": 0.6967741935483871,
968
+ "grad_norm": 5.046849897934945,
969
+ "learning_rate": 2.1536524612697402e-06,
970
+ "log_odds_chosen": 3.75,
971
+ "log_odds_ratio": -0.094482421875,
972
+ "logits/chosen": -1.859375,
973
+ "logits/rejected": 15.962499618530273,
974
+ "logps/chosen": -1.030859351158142,
975
+ "logps/rejected": -4.309374809265137,
976
+ "loss": 0.9959,
977
+ "nll_loss": 0.9917968511581421,
978
+ "rewards/accuracies": 0.9937499761581421,
979
+ "rewards/chosen": -0.05153808742761612,
980
+ "rewards/margins": 0.16396483778953552,
981
+ "rewards/rejected": -0.21562500298023224,
982
  "step": 540
983
  },
984
  {
985
  "epoch": 0.7096774193548387,
986
+ "grad_norm": 5.947499132612451,
987
+ "learning_rate": 2.1339479988816e-06,
988
+ "log_odds_chosen": 3.6875,
989
+ "log_odds_ratio": -0.06818847358226776,
990
+ "logits/chosen": -2.174999952316284,
991
+ "logits/rejected": 15.493749618530273,
992
+ "logps/chosen": -0.999218761920929,
993
+ "logps/rejected": -4.228125095367432,
994
+ "loss": 1.003,
995
+ "nll_loss": 1.0,
996
+ "rewards/accuracies": 0.987500011920929,
997
+ "rewards/chosen": -0.0499267578125,
998
+ "rewards/margins": 0.16171875596046448,
999
+ "rewards/rejected": -0.21142578125,
1000
  "step": 550
1001
  },
1002
  {
1003
  "epoch": 0.7225806451612903,
1004
+ "grad_norm": 5.351752547147555,
1005
+ "learning_rate": 2.114774672189068e-06,
1006
+ "log_odds_chosen": 3.862499952316284,
1007
+ "log_odds_ratio": -0.08652343600988388,
1008
+ "logits/chosen": -2.469531297683716,
1009
+ "logits/rejected": 15.331250190734863,
1010
+ "logps/chosen": -0.9957031011581421,
1011
+ "logps/rejected": -4.415625095367432,
1012
+ "loss": 1.0154,
1013
+ "nll_loss": 1.011328101158142,
1014
+ "rewards/accuracies": 0.9624999761581421,
1015
+ "rewards/chosen": -0.04978027194738388,
1016
+ "rewards/margins": 0.17119140923023224,
1017
+ "rewards/rejected": -0.22089843451976776,
1018
  "step": 560
1019
  },
1020
  {
1021
  "epoch": 0.7354838709677419,
1022
+ "grad_norm": 4.936722562656586,
1023
+ "learning_rate": 2.0961090407515926e-06,
1024
+ "log_odds_chosen": 4.0703125,
1025
+ "log_odds_ratio": -0.06992187350988388,
1026
+ "logits/chosen": -2.4644532203674316,
1027
+ "logits/rejected": 15.46875,
1028
+ "logps/chosen": -0.9624999761581421,
1029
+ "logps/rejected": -4.567187309265137,
1030
+ "loss": 0.9827,
1031
+ "nll_loss": 0.979687511920929,
1032
+ "rewards/accuracies": 0.9937499761581421,
1033
+ "rewards/chosen": -0.04814453050494194,
1034
+ "rewards/margins": 0.18017578125,
1035
+ "rewards/rejected": -0.22832031548023224,
1036
  "step": 570
1037
  },
1038
  {
1039
  "epoch": 0.7483870967741936,
1040
+ "grad_norm": 4.997513662861683,
1041
+ "learning_rate": 2.077929087308457e-06,
1042
+ "log_odds_chosen": 3.856250047683716,
1043
+ "log_odds_ratio": -0.07062987983226776,
1044
+ "logits/chosen": -2.723437547683716,
1045
+ "logits/rejected": 14.800000190734863,
1046
+ "logps/chosen": -0.94140625,
1047
+ "logps/rejected": -4.317187309265137,
1048
+ "loss": 0.9656,
1049
+ "nll_loss": 0.9624999761581421,
1050
+ "rewards/accuracies": 0.987500011920929,
1051
+ "rewards/chosen": -0.047119140625,
1052
+ "rewards/margins": 0.16884765028953552,
1053
+ "rewards/rejected": -0.21562500298023224,
1054
  "step": 580
1055
  },
1056
  {
1057
  "epoch": 0.7612903225806451,
1058
+ "grad_norm": 5.381405087912754,
1059
+ "learning_rate": 2.0602141085758233e-06,
1060
+ "log_odds_chosen": 4.1015625,
1061
+ "log_odds_ratio": -0.06821288913488388,
1062
+ "logits/chosen": -2.957812547683716,
1063
+ "logits/rejected": 15.106249809265137,
1064
+ "logps/chosen": -0.9488281011581421,
1065
+ "logps/rejected": -4.574999809265137,
1066
+ "loss": 0.976,
1067
+ "nll_loss": 0.973437488079071,
1068
+ "rewards/accuracies": 0.981249988079071,
1069
+ "rewards/chosen": -0.04743652418255806,
1070
+ "rewards/margins": 0.18144531548023224,
1071
+ "rewards/rejected": -0.22861328721046448,
1072
  "step": 590
1073
  },
1074
  {
1075
  "epoch": 0.7741935483870968,
1076
+ "grad_norm": 5.488636413342299,
1077
+ "learning_rate": 2.0429446161135922e-06,
1078
+ "log_odds_chosen": 4.181250095367432,
1079
+ "log_odds_ratio": -0.06918945163488388,
1080
+ "logits/chosen": -2.314160108566284,
1081
+ "logits/rejected": 15.449999809265137,
1082
+ "logps/chosen": -0.9921875,
1083
+ "logps/rejected": -4.721875190734863,
1084
+ "loss": 1.0089,
1085
+ "nll_loss": 1.0050780773162842,
1086
+ "rewards/accuracies": 0.987500011920929,
1087
+ "rewards/chosen": -0.04958496242761612,
1088
+ "rewards/margins": 0.18642577528953552,
1089
+ "rewards/rejected": -0.23593750596046448,
1090
  "step": 600
1091
  },
1092
  {
1093
  "epoch": 0.7870967741935484,
1094
+ "grad_norm": 5.427887234676386,
1095
+ "learning_rate": 2.0261022461827697e-06,
1096
+ "log_odds_chosen": 4.599999904632568,
1097
+ "log_odds_ratio": -0.06470946967601776,
1098
+ "logits/chosen": -3.4375,
1099
+ "logits/rejected": 15.268750190734863,
1100
+ "logps/chosen": -0.998046875,
1101
+ "logps/rejected": -5.140625,
1102
+ "loss": 1.0151,
1103
+ "nll_loss": 1.0128905773162842,
1104
+ "rewards/accuracies": 0.981249988079071,
1105
+ "rewards/chosen": -0.04990234225988388,
1106
+ "rewards/margins": 0.20703125,
1107
+ "rewards/rejected": -0.25712889432907104,
1108
  "step": 610
1109
  },
1110
  {
1111
  "epoch": 0.8,
1112
+ "grad_norm": 6.166954138248167,
1113
+ "learning_rate": 2.009669677645352e-06,
1114
+ "log_odds_chosen": 4.235937595367432,
1115
+ "log_odds_ratio": -0.08128662407398224,
1116
+ "logits/chosen": -2.9867186546325684,
1117
+ "logits/rejected": 15.600000381469727,
1118
+ "logps/chosen": -0.962890625,
1119
+ "logps/rejected": -4.724999904632568,
1120
+ "loss": 0.9886,
1121
+ "nll_loss": 0.985156238079071,
1122
+ "rewards/accuracies": 0.9750000238418579,
1123
+ "rewards/chosen": -0.04814453050494194,
1124
+ "rewards/margins": 0.18828125298023224,
1125
+ "rewards/rejected": -0.23642578721046448,
1126
  "step": 620
1127
  },
1128
  {
1129
  "epoch": 0.8129032258064516,
1130
+ "grad_norm": 5.497842991013724,
1131
+ "learning_rate": 1.9936305570722503e-06,
1132
+ "log_odds_chosen": 4.198437690734863,
1133
+ "log_odds_ratio": -0.05922241136431694,
1134
+ "logits/chosen": -2.833984375,
1135
+ "logits/rejected": 15.46875,
1136
+ "logps/chosen": -0.94921875,
1137
+ "logps/rejected": -4.667187690734863,
1138
+ "loss": 0.9591,
1139
+ "nll_loss": 0.956250011920929,
1140
+ "rewards/accuracies": 1.0,
1141
+ "rewards/chosen": -0.04746093600988388,
1142
+ "rewards/margins": 0.18569335341453552,
1143
+ "rewards/rejected": -0.23310546576976776,
1144
  "step": 630
1145
  },
1146
  {
1147
  "epoch": 0.8258064516129032,
1148
+ "grad_norm": 8.242142638367945,
1149
+ "learning_rate": 1.977969430323089e-06,
1150
+ "log_odds_chosen": 4.364062309265137,
1151
+ "log_odds_ratio": -0.06558837741613388,
1152
+ "logits/chosen": -2.439453125,
1153
+ "logits/rejected": 15.581250190734863,
1154
+ "logps/chosen": -1.0246093273162842,
1155
+ "logps/rejected": -4.918749809265137,
1156
+ "loss": 1.0151,
1157
+ "nll_loss": 1.011328101158142,
1158
+ "rewards/accuracies": 0.9937499761581421,
1159
+ "rewards/chosen": -0.05124511569738388,
1160
+ "rewards/margins": 0.19462890923023224,
1161
+ "rewards/rejected": -0.24580077826976776,
1162
  "step": 640
1163
  },
1164
  {
1165
  "epoch": 0.8387096774193549,
1166
+ "grad_norm": 5.407957699348334,
1167
+ "learning_rate": 1.962671679947149e-06,
1168
+ "log_odds_chosen": 4.728125095367432,
1169
+ "log_odds_ratio": -0.04775390774011612,
1170
+ "logits/chosen": -2.6484375,
1171
+ "logits/rejected": 16.125,
1172
+ "logps/chosen": -1.0050780773162842,
1173
+ "logps/rejected": -5.262499809265137,
1174
+ "loss": 1.0211,
1175
+ "nll_loss": 1.019140601158142,
1176
+ "rewards/accuracies": 0.987500011920929,
1177
+ "rewards/chosen": -0.0501708984375,
1178
+ "rewards/margins": 0.21298828721046448,
1179
+ "rewards/rejected": -0.2630859315395355,
1180
  "step": 650
1181
  },
1182
  {
1183
  "epoch": 0.8516129032258064,
1184
+ "grad_norm": 5.429871193266642,
1185
+ "learning_rate": 1.9477234678290497e-06,
1186
+ "log_odds_chosen": 4.28125,
1187
+ "log_odds_ratio": -0.06833495944738388,
1188
+ "logits/chosen": -2.073046922683716,
1189
+ "logits/rejected": 15.493749618530273,
1190
+ "logps/chosen": -1.019921898841858,
1191
+ "logps/rejected": -4.854687690734863,
1192
+ "loss": 1.0178,
1193
+ "nll_loss": 1.0148437023162842,
1194
+ "rewards/accuracies": 0.987500011920929,
1195
+ "rewards/chosen": -0.05095214769244194,
1196
+ "rewards/margins": 0.19204100966453552,
1197
+ "rewards/rejected": -0.24287109076976776,
1198
  "step": 660
1199
  },
1200
  {
1201
  "epoch": 0.864516129032258,
1202
+ "grad_norm": 5.38838761477697,
1203
+ "learning_rate": 1.9331116825676476e-06,
1204
+ "log_odds_chosen": 4.264062404632568,
1205
+ "log_odds_ratio": -0.07292480766773224,
1206
+ "logits/chosen": -2.7894530296325684,
1207
+ "logits/rejected": 15.3125,
1208
+ "logps/chosen": -0.9847656488418579,
1209
+ "logps/rejected": -4.784375190734863,
1210
+ "loss": 1.0014,
1211
+ "nll_loss": 0.998046875,
1212
  "rewards/accuracies": 0.987500011920929,
1213
+ "rewards/chosen": -0.04924316331744194,
1214
+ "rewards/margins": 0.18974609673023224,
1215
+ "rewards/rejected": -0.2392578125,
1216
  "step": 670
1217
  },
1218
  {
1219
  "epoch": 0.8774193548387097,
1220
+ "grad_norm": 6.231144536945813,
1221
+ "learning_rate": 1.9188238911333405e-06,
1222
+ "log_odds_chosen": 4.495312690734863,
1223
+ "log_odds_ratio": -0.0684814453125,
1224
+ "logits/chosen": -2.408984422683716,
1225
+ "logits/rejected": 15.449999809265137,
1226
+ "logps/chosen": -1.029687523841858,
1227
+ "logps/rejected": -5.090624809265137,
1228
+ "loss": 1.0266,
1229
+ "nll_loss": 1.023828148841858,
1230
  "rewards/accuracies": 0.987500011920929,
1231
+ "rewards/chosen": -0.05156249925494194,
1232
+ "rewards/margins": 0.20302733778953552,
1233
+ "rewards/rejected": -0.25439453125,
1234
  "step": 680
1235
  },
1236
  {
1237
  "epoch": 0.8903225806451613,
1238
+ "grad_norm": 5.900000733774859,
1239
+ "learning_rate": 1.9048482943986487e-06,
1240
+ "log_odds_chosen": 4.173437595367432,
1241
+ "log_odds_ratio": -0.10228271782398224,
1242
+ "logits/chosen": -2.411328077316284,
1243
+ "logits/rejected": 15.0625,
1244
+ "logps/chosen": -0.987109363079071,
1245
+ "logps/rejected": -4.704687595367432,
1246
+ "loss": 0.9854,
1247
+ "nll_loss": 0.981249988079071,
1248
  "rewards/accuracies": 0.9750000238418579,
1249
+ "rewards/chosen": -0.04938964918255806,
1250
+ "rewards/margins": 0.18564453721046448,
1251
+ "rewards/rejected": -0.23476561903953552,
1252
  "step": 690
1253
  },
1254
  {
1255
  "epoch": 0.9032258064516129,
1256
+ "grad_norm": 5.5507604309289045,
1257
+ "learning_rate": 1.8911736861805847e-06,
1258
+ "log_odds_chosen": 3.984375,
1259
+ "log_odds_ratio": -0.08645019680261612,
1260
+ "logits/chosen": -2.534374952316284,
1261
+ "logits/rejected": 15.199999809265137,
1262
+ "logps/chosen": -0.946484386920929,
1263
+ "logps/rejected": -4.471875190734863,
1264
+ "loss": 0.9628,
1265
+ "nll_loss": 0.958984375,
1266
+ "rewards/accuracies": 0.9750000238418579,
1267
+ "rewards/chosen": -0.04731445387005806,
1268
+ "rewards/margins": 0.17587891221046448,
1269
+ "rewards/rejected": -0.22343750298023224,
1270
  "step": 700
1271
  },
1272
  {
1273
  "epoch": 0.9161290322580645,
1274
+ "grad_norm": 5.72722978193488,
1275
+ "learning_rate": 1.8777894154716994e-06,
1276
+ "log_odds_chosen": 4.260937690734863,
1277
+ "log_odds_ratio": -0.08770751953125,
1278
+ "logits/chosen": -2.785937547683716,
1279
+ "logits/rejected": 15.212499618530273,
1280
+ "logps/chosen": -0.975781261920929,
1281
+ "logps/rejected": -4.790625095367432,
1282
+ "loss": 0.9996,
1283
+ "nll_loss": 0.9957031011581421,
1284
+ "rewards/accuracies": 0.96875,
1285
+ "rewards/chosen": -0.048828125,
1286
+ "rewards/margins": 0.19072265923023224,
1287
+ "rewards/rejected": -0.23945312201976776,
1288
  "step": 710
1289
  },
1290
  {
1291
  "epoch": 0.9290322580645162,
1292
+ "grad_norm": 5.182886168475235,
1293
+ "learning_rate": 1.864685351570504e-06,
1294
+ "log_odds_chosen": 4.417187690734863,
1295
+ "log_odds_ratio": -0.05722656100988388,
1296
+ "logits/chosen": -2.477343797683716,
1297
+ "logits/rejected": 15.387499809265137,
1298
+ "logps/chosen": -0.953906238079071,
1299
+ "logps/rejected": -4.900000095367432,
1300
+ "loss": 0.9764,
1301
+ "nll_loss": 0.973437488079071,
1302
+ "rewards/accuracies": 1.0,
1303
+ "rewards/chosen": -0.04770507663488388,
1304
+ "rewards/margins": 0.19716796278953552,
1305
+ "rewards/rejected": -0.24492187798023224,
1306
  "step": 720
1307
  },
1308
  {
1309
  "epoch": 0.9419354838709677,
1310
+ "grad_norm": 6.373900642995302,
1311
+ "learning_rate": 1.8518518518518519e-06,
1312
+ "log_odds_chosen": 3.9078125953674316,
1313
+ "log_odds_ratio": -0.07728271186351776,
1314
+ "logits/chosen": -2.785937547683716,
1315
+ "logits/rejected": 15.181249618530273,
1316
+ "logps/chosen": -0.979687511920929,
1317
+ "logps/rejected": -4.4296875,
1318
+ "loss": 0.9999,
1319
+ "nll_loss": 0.9957031011581421,
1320
+ "rewards/accuracies": 0.981249988079071,
1321
+ "rewards/chosen": -0.049072265625,
1322
+ "rewards/margins": 0.17246094346046448,
1323
+ "rewards/rejected": -0.22158202528953552,
1324
  "step": 730
1325
  },
1326
  {
1327
  "epoch": 0.9548387096774194,
1328
+ "grad_norm": 5.301970324773416,
1329
+ "learning_rate": 1.8392797319442742e-06,
1330
+ "log_odds_chosen": 3.7484374046325684,
1331
+ "log_odds_ratio": -0.07939453423023224,
1332
+ "logits/chosen": -3.1468749046325684,
1333
+ "logits/rejected": 14.743749618530273,
1334
+ "logps/chosen": -0.958203136920929,
1335
+ "logps/rejected": -4.229687690734863,
1336
+ "loss": 0.9549,
1337
+ "nll_loss": 0.9527343511581421,
1338
+ "rewards/accuracies": 1.0,
1339
+ "rewards/chosen": -0.04790038987994194,
1340
+ "rewards/margins": 0.16396483778953552,
1341
+ "rewards/rejected": -0.21162109076976776,
1342
  "step": 740
1343
  },
1344
  {
1345
  "epoch": 0.967741935483871,
1346
+ "grad_norm": 5.607236715482459,
1347
+ "learning_rate": 1.8269602381046742e-06,
1348
+ "log_odds_chosen": 4.028124809265137,
1349
+ "log_odds_ratio": -0.06224365159869194,
1350
+ "logits/chosen": -2.905468702316284,
1351
+ "logits/rejected": 15.387499809265137,
1352
+ "logps/chosen": -0.992968738079071,
1353
+ "logps/rejected": -4.550000190734863,
1354
+ "loss": 1.0119,
1355
+ "nll_loss": 1.010156273841858,
1356
+ "rewards/accuracies": 0.9937499761581421,
1357
+ "rewards/chosen": -0.04965820163488388,
1358
+ "rewards/margins": 0.17753906548023224,
1359
+ "rewards/rejected": -0.2275390625,
1360
  "step": 750
1361
  },
1362
  {
1363
  "epoch": 0.9806451612903225,
1364
+ "grad_norm": 5.46826560416884,
1365
+ "learning_rate": 1.8148850216015693e-06,
1366
+ "log_odds_chosen": 3.9203124046325684,
1367
+ "log_odds_ratio": -0.08592529594898224,
1368
+ "logits/chosen": -2.5835938453674316,
1369
+ "logits/rejected": 15.125,
1370
+ "logps/chosen": -0.974609375,
1371
+ "logps/rejected": -4.432812690734863,
1372
+ "loss": 0.9957,
1373
+ "nll_loss": 0.9925781488418579,
1374
+ "rewards/accuracies": 0.9750000238418579,
1375
+ "rewards/chosen": -0.04873047024011612,
1376
+ "rewards/margins": 0.17304687201976776,
1377
+ "rewards/rejected": -0.22158202528953552,
1378
  "step": 760
1379
  },
1380
  {
1381
  "epoch": 0.9935483870967742,
1382
+ "grad_norm": 6.701663744929856,
1383
+ "learning_rate": 1.803046114936547e-06,
1384
+ "log_odds_chosen": 4.103125095367432,
1385
+ "log_odds_ratio": -0.0823974609375,
1386
+ "logits/chosen": -3.176562547683716,
1387
+ "logits/rejected": 14.993749618530273,
1388
+ "logps/chosen": -0.97265625,
1389
+ "logps/rejected": -4.615624904632568,
1390
+ "loss": 0.9939,
1391
+ "nll_loss": 0.9906250238418579,
1392
+ "rewards/accuracies": 0.9750000238418579,
1393
+ "rewards/chosen": -0.048583984375,
1394
+ "rewards/margins": 0.18212890625,
1395
+ "rewards/rejected": -0.23056641221046448,
1396
  "step": 770
1397
  },
1398
  {
1399
  "epoch": 1.0,
1400
  "step": 775,
1401
  "total_flos": 0.0,
1402
+ "train_loss": 1.0746982106854839,
1403
+ "train_runtime": 1520.1443,
1404
+ "train_samples_per_second": 8.157,
1405
+ "train_steps_per_second": 0.51
1406
  }
1407
  ],
1408
  "logging_steps": 10,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f588a362e0e6fab1d890ecdbe8b375eb3ebbbad23e60db21dbd7ae74d645294c
3
  size 7032
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c9f48d2605a16695bdd6a65f12d736b4e7c6a0fb9d67dcff3355744a75a6822
3
  size 7032