RikkiXu commited on
Commit
c2d3fa9
1 Parent(s): abe1e9b

Model save

Browse files
README.md CHANGED
@@ -15,15 +15,15 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model was trained from scratch on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 1.9828
19
- - Rewards/chosen: -3.4223
20
- - Rewards/rejected: -2.1126
21
- - Rewards/accuracies: 0.3555
22
- - Rewards/margins: -1.3097
23
- - Logps/rejected: -521.2875
24
- - Logps/chosen: -405.8879
25
- - Logits/rejected: -4.9364
26
- - Logits/chosen: -4.7068
27
 
28
  ## Model description
29
 
@@ -42,7 +42,7 @@ More information needed
42
  ### Training hyperparameters
43
 
44
  The following hyperparameters were used during training:
45
- - learning_rate: 5e-08
46
  - train_batch_size: 8
47
  - eval_batch_size: 8
48
  - seed: 42
@@ -60,9 +60,9 @@ The following hyperparameters were used during training:
60
 
61
  | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
62
  |:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
63
- | 0.3652 | 0.29 | 100 | 1.7488 | -2.2159 | -1.0097 | 0.3516 | -1.2062 | -519.0817 | -403.4752 | -4.9249 | -4.6962 |
64
- | 0.312 | 0.57 | 200 | 1.9596 | -3.1949 | -1.8164 | 0.3398 | -1.3786 | -520.6950 | -405.4332 | -4.9391 | -4.7096 |
65
- | 0.2993 | 0.86 | 300 | 1.9828 | -3.4223 | -2.1126 | 0.3555 | -1.3097 | -521.2875 | -405.8879 | -4.9364 | -4.7068 |
66
 
67
 
68
  ### Framework versions
 
15
 
16
  This model was trained from scratch on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.6931
19
+ - Rewards/chosen: -2.1301
20
+ - Rewards/rejected: -2.1301
21
+ - Rewards/accuracies: 0.0
22
+ - Rewards/margins: 0.0
23
+ - Logps/rejected: -159.8372
24
+ - Logps/chosen: -159.8372
25
+ - Logits/rejected: -3.1995
26
+ - Logits/chosen: -3.1995
27
 
28
  ## Model description
29
 
 
42
  ### Training hyperparameters
43
 
44
  The following hyperparameters were used during training:
45
+ - learning_rate: 2e-07
46
  - train_batch_size: 8
47
  - eval_batch_size: 8
48
  - seed: 42
 
60
 
61
  | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
62
  |:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
63
+ | 0.3114 | 0.29 | 100 | 0.6931 | -2.8588 | -2.8588 | 0.0 | 0.0 | -161.2947 | -161.2947 | -3.2262 | -3.2262 |
64
+ | 0.2741 | 0.57 | 200 | 0.6931 | -2.6760 | -2.6760 | 0.0 | 0.0 | -160.9289 | -160.9289 | -3.2033 | -3.2033 |
65
+ | 0.2695 | 0.86 | 300 | 0.6931 | -2.1301 | -2.1301 | 0.0 | 0.0 | -159.8372 | -159.8372 | -3.1995 | -3.1995 |
66
 
67
 
68
  ### Framework versions
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.3696309270011661,
4
- "train_runtime": 5591.2149,
5
  "train_samples": 44682,
6
- "train_samples_per_second": 7.991,
7
- "train_steps_per_second": 0.062
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.30806411576476,
4
+ "train_runtime": 5299.0292,
5
  "train_samples": 44682,
6
+ "train_samples_per_second": 8.432,
7
+ "train_steps_per_second": 0.066
8
  }
config.json CHANGED
@@ -20,7 +20,7 @@
20
  "sliding_window": 4096,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
23
- "transformers_version": "4.41.1",
24
  "use_cache": false,
25
  "vocab_size": 32002
26
  }
 
20
  "sliding_window": 4096,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
23
+ "transformers_version": "4.38.2",
24
  "use_cache": false,
25
  "vocab_size": 32002
26
  }
model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b289b436dc9787cb73fff5a2a0c5fb69ad78dc06d960a4946cb9d6b35bf920b3
3
  size 4943178720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff3d0ac00c3304811f4f56db0c3a49ea7bf825902fea45c8a9791a2e4f7431ab
3
  size 4943178720
model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:556d3f80045da81e8b580005d0676f922550a32c41b802ac16391cdb8a56eac7
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1122a41f096dacbe12e70bff78c3951321a0a6f46db2c7995a726fd8f909bf84
3
  size 4999819336
model-00003-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:06047de541b6531a598ee5f5cc63aaa960471d4a4c6da134647cc69aa23f71b0
3
  size 4540532728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c18a2acaf553956d0cddde18a5d52d992dfb5723fc496a27bcde59c9bb62529
3
  size 4540532728
runs/May30_11-21-27_n136-129-074/events.out.tfevents.1717039322.n136-129-074.2425775.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:054622cc33ba7e3d84169c49b4b7eed8365d0c64fcf5eb34d2a9a1e5f60e6a4a
3
- size 28328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8fc27d90b3ccda7d2e755a090af78f6ce2111c0b55bbb4c06e4b2693fca26d7
3
+ size 31434
tokenizer.json CHANGED
@@ -152,7 +152,6 @@
152
  "end_of_word_suffix": null,
153
  "fuse_unk": true,
154
  "byte_fallback": true,
155
- "ignore_merges": false,
156
  "vocab": {
157
  "<unk>": 0,
158
  "<s>": 1,
 
152
  "end_of_word_suffix": null,
153
  "fuse_unk": true,
154
  "byte_fallback": true,
 
155
  "vocab": {
156
  "<unk>": 0,
157
  "<s>": 1,
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 1.0,
3
- "train_loss": 0.3696309270011661,
4
- "train_runtime": 5591.2149,
5
  "train_samples": 44682,
6
- "train_samples_per_second": 7.991,
7
- "train_steps_per_second": 0.062
8
  }
 
1
  {
2
  "epoch": 1.0,
3
+ "train_loss": 0.30806411576476,
4
+ "train_runtime": 5299.0292,
5
  "train_samples": 44682,
6
+ "train_samples_per_second": 8.432,
7
+ "train_steps_per_second": 0.066
8
  }
trainer_state.json CHANGED
@@ -10,8 +10,8 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
- "grad_norm": 1482.8898632632786,
14
- "learning_rate": 1.4285714285714284e-09,
15
  "logits/chosen": -4.490396976470947,
16
  "logits/rejected": -4.787891387939453,
17
  "logps/chosen": -300.56573486328125,
@@ -25,570 +25,570 @@
25
  },
26
  {
27
  "epoch": 0.03,
28
- "grad_norm": 1626.423546399318,
29
- "learning_rate": 1.4285714285714284e-08,
30
- "logits/chosen": -4.29224967956543,
31
- "logits/rejected": -4.469963550567627,
32
- "logps/chosen": -285.7841796875,
33
- "logps/rejected": -241.37472534179688,
34
- "loss": 0.7238,
35
- "rewards/accuracies": 0.4097222089767456,
36
- "rewards/chosen": 0.00654969597235322,
37
- "rewards/margins": -0.022695984691381454,
38
- "rewards/rejected": 0.029245682060718536,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.06,
43
- "grad_norm": 1614.6738239528886,
44
- "learning_rate": 2.857142857142857e-08,
45
- "logits/chosen": -4.210062026977539,
46
- "logits/rejected": -4.502069473266602,
47
- "logps/chosen": -312.73968505859375,
48
- "logps/rejected": -247.76315307617188,
49
- "loss": 0.7059,
50
- "rewards/accuracies": 0.4625000059604645,
51
- "rewards/chosen": 0.01807965151965618,
52
- "rewards/margins": 0.015493685379624367,
53
- "rewards/rejected": 0.0025859654415398836,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.09,
58
- "grad_norm": 1444.832849324016,
59
- "learning_rate": 4.285714285714285e-08,
60
- "logits/chosen": -4.140606880187988,
61
- "logits/rejected": -4.359016418457031,
62
- "logps/chosen": -323.00701904296875,
63
- "logps/rejected": -274.5583801269531,
64
- "loss": 0.6754,
65
- "rewards/accuracies": 0.6812499761581421,
66
- "rewards/chosen": 0.26756447553634644,
67
- "rewards/margins": 0.19910377264022827,
68
- "rewards/rejected": 0.06846068799495697,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.11,
73
- "grad_norm": 1322.1199574632033,
74
- "learning_rate": 4.9968724814144246e-08,
75
- "logits/chosen": -4.296597480773926,
76
- "logits/rejected": -4.558178901672363,
77
- "logps/chosen": -277.67041015625,
78
- "logps/rejected": -227.9638214111328,
79
- "loss": 0.5887,
80
- "rewards/accuracies": 0.731249988079071,
81
- "rewards/chosen": 0.46597638726234436,
82
- "rewards/margins": 0.30959779024124146,
83
- "rewards/rejected": 0.1563786268234253,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.14,
88
- "grad_norm": 1226.7969075975607,
89
- "learning_rate": 4.971899263739325e-08,
90
- "logits/chosen": -4.339926719665527,
91
- "logits/rejected": -4.592724323272705,
92
- "logps/chosen": -270.9559020996094,
93
- "logps/rejected": -228.2990264892578,
94
- "loss": 0.5099,
95
- "rewards/accuracies": 0.7437499761581421,
96
- "rewards/chosen": 0.9494625329971313,
97
- "rewards/margins": 0.5203765034675598,
98
- "rewards/rejected": 0.4290861189365387,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.17,
103
- "grad_norm": 929.2104534934765,
104
- "learning_rate": 4.9222026055025726e-08,
105
- "logits/chosen": -4.333067893981934,
106
- "logits/rejected": -4.564365386962891,
107
- "logps/chosen": -303.43310546875,
108
- "logps/rejected": -252.1361846923828,
109
- "loss": 0.4317,
110
- "rewards/accuracies": 0.762499988079071,
111
- "rewards/chosen": 1.6878736019134521,
112
- "rewards/margins": 1.0692347288131714,
113
- "rewards/rejected": 0.6186389923095703,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.2,
118
- "grad_norm": 1085.9024365118285,
119
- "learning_rate": 4.8482795627104736e-08,
120
- "logits/chosen": -4.332296371459961,
121
- "logits/rejected": -4.567011833190918,
122
- "logps/chosen": -289.8699645996094,
123
- "logps/rejected": -241.077880859375,
124
- "loss": 0.4089,
125
- "rewards/accuracies": 0.862500011920929,
126
- "rewards/chosen": 1.8221668004989624,
127
- "rewards/margins": 1.204461693763733,
128
- "rewards/rejected": 0.6177049875259399,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.23,
133
- "grad_norm": 978.9007166471121,
134
- "learning_rate": 4.750869498807735e-08,
135
- "logits/chosen": -4.28688383102417,
136
- "logits/rejected": -4.543013572692871,
137
- "logps/chosen": -322.56048583984375,
138
- "logps/rejected": -271.0233154296875,
139
- "loss": 0.3625,
140
- "rewards/accuracies": 0.875,
141
- "rewards/chosen": 2.1792995929718018,
142
- "rewards/margins": 1.4446518421173096,
143
- "rewards/rejected": 0.7346473932266235,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.26,
148
- "grad_norm": 755.2675019751182,
149
- "learning_rate": 4.630946689712609e-08,
150
- "logits/chosen": -4.393240928649902,
151
- "logits/rejected": -4.636073112487793,
152
- "logps/chosen": -283.79290771484375,
153
- "logps/rejected": -234.5099334716797,
154
- "loss": 0.3436,
155
- "rewards/accuracies": 0.862500011920929,
156
- "rewards/chosen": 2.397665500640869,
157
- "rewards/margins": 1.638594388961792,
158
- "rewards/rejected": 0.7590711116790771,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.29,
163
- "grad_norm": 893.2972482207422,
164
- "learning_rate": 4.4897105793046024e-08,
165
- "logits/chosen": -4.302220344543457,
166
- "logits/rejected": -4.582569122314453,
167
- "logps/chosen": -270.37457275390625,
168
- "logps/rejected": -229.5946502685547,
169
- "loss": 0.3652,
170
- "rewards/accuracies": 0.8374999761581421,
171
- "rewards/chosen": 2.673680067062378,
172
- "rewards/margins": 2.020587921142578,
173
- "rewards/rejected": 0.6530919075012207,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.29,
178
- "eval_logits/chosen": -4.696198463439941,
179
- "eval_logits/rejected": -4.924900054931641,
180
- "eval_logps/chosen": -403.4751892089844,
181
- "eval_logps/rejected": -519.0816650390625,
182
- "eval_loss": 1.7488452196121216,
183
- "eval_rewards/accuracies": 0.3515625,
184
- "eval_rewards/chosen": -2.2159295082092285,
185
- "eval_rewards/margins": -1.206247329711914,
186
- "eval_rewards/rejected": -1.009682059288025,
187
- "eval_runtime": 97.4797,
188
- "eval_samples_per_second": 20.517,
189
- "eval_steps_per_second": 0.328,
190
  "step": 100
191
  },
192
  {
193
  "epoch": 0.31,
194
- "grad_norm": 872.5882367590035,
195
- "learning_rate": 4.328573782827409e-08,
196
- "logits/chosen": -4.3329758644104,
197
- "logits/rejected": -4.617272853851318,
198
- "logps/chosen": -296.02618408203125,
199
- "logps/rejected": -242.73428344726562,
200
- "loss": 0.349,
201
- "rewards/accuracies": 0.875,
202
- "rewards/chosen": 2.7564496994018555,
203
- "rewards/margins": 2.076282024383545,
204
- "rewards/rejected": 0.6801677942276001,
205
  "step": 110
206
  },
207
  {
208
  "epoch": 0.34,
209
- "grad_norm": 1043.3393138533181,
210
- "learning_rate": 4.1491479581946166e-08,
211
- "logits/chosen": -4.3465352058410645,
212
- "logits/rejected": -4.575117111206055,
213
- "logps/chosen": -300.99859619140625,
214
- "logps/rejected": -265.1387023925781,
215
- "loss": 0.3567,
216
- "rewards/accuracies": 0.8500000238418579,
217
- "rewards/chosen": 2.6209876537323,
218
- "rewards/margins": 1.951716423034668,
219
- "rewards/rejected": 0.6692714095115662,
220
  "step": 120
221
  },
222
  {
223
  "epoch": 0.37,
224
- "grad_norm": 837.2392359904699,
225
- "learning_rate": 3.953227686510564e-08,
226
- "logits/chosen": -4.319238185882568,
227
- "logits/rejected": -4.565236568450928,
228
- "logps/chosen": -288.36688232421875,
229
- "logps/rejected": -240.29177856445312,
230
- "loss": 0.3089,
231
- "rewards/accuracies": 0.887499988079071,
232
- "rewards/chosen": 2.86527419090271,
233
- "rewards/margins": 2.168524980545044,
234
- "rewards/rejected": 0.6967490911483765,
235
  "step": 130
236
  },
237
  {
238
  "epoch": 0.4,
239
- "grad_norm": 1077.8316620275032,
240
- "learning_rate": 3.7427725230301354e-08,
241
- "logits/chosen": -4.310965538024902,
242
- "logits/rejected": -4.550887584686279,
243
- "logps/chosen": -295.83868408203125,
244
- "logps/rejected": -240.91311645507812,
245
- "loss": 0.3113,
246
- "rewards/accuracies": 0.8812500238418579,
247
- "rewards/chosen": 3.000100612640381,
248
- "rewards/margins": 2.261690855026245,
249
- "rewards/rejected": 0.7384099960327148,
250
  "step": 140
251
  },
252
  {
253
  "epoch": 0.43,
254
- "grad_norm": 805.166891537404,
255
- "learning_rate": 3.5198873980801955e-08,
256
- "logits/chosen": -4.3193488121032715,
257
- "logits/rejected": -4.625092506408691,
258
- "logps/chosen": -294.2840881347656,
259
- "logps/rejected": -245.18655395507812,
260
- "loss": 0.318,
261
- "rewards/accuracies": 0.8374999761581421,
262
- "rewards/chosen": 3.1946704387664795,
263
- "rewards/margins": 2.1761879920959473,
264
- "rewards/rejected": 1.0184824466705322,
265
  "step": 150
266
  },
267
  {
268
  "epoch": 0.46,
269
- "grad_norm": 711.9582620759091,
270
- "learning_rate": 3.2868015639687205e-08,
271
- "logits/chosen": -4.259413242340088,
272
- "logits/rejected": -4.450949668884277,
273
- "logps/chosen": -311.91326904296875,
274
- "logps/rejected": -270.4508972167969,
275
- "loss": 0.3344,
276
- "rewards/accuracies": 0.8812500238418579,
277
- "rewards/chosen": 3.2649455070495605,
278
- "rewards/margins": 2.298511266708374,
279
- "rewards/rejected": 0.966434121131897,
280
  "step": 160
281
  },
282
  {
283
  "epoch": 0.49,
284
- "grad_norm": 894.588925085696,
285
- "learning_rate": 3.0458462984504134e-08,
286
- "logits/chosen": -4.303664207458496,
287
- "logits/rejected": -4.500936031341553,
288
- "logps/chosen": -274.3706359863281,
289
- "logps/rejected": -235.48678588867188,
290
- "loss": 0.2963,
291
- "rewards/accuracies": 0.856249988079071,
292
- "rewards/chosen": 3.1083619594573975,
293
- "rewards/margins": 2.440159559249878,
294
- "rewards/rejected": 0.6682023406028748,
295
  "step": 170
296
  },
297
  {
298
  "epoch": 0.52,
299
- "grad_norm": 787.4594621489149,
300
- "learning_rate": 2.7994315877542628e-08,
301
- "logits/chosen": -4.386083602905273,
302
- "logits/rejected": -4.545100212097168,
303
- "logps/chosen": -263.3323059082031,
304
- "logps/rejected": -228.91329956054688,
305
- "loss": 0.329,
306
- "rewards/accuracies": 0.8500000238418579,
307
- "rewards/chosen": 2.6448895931243896,
308
- "rewards/margins": 2.054766893386841,
309
- "rewards/rejected": 0.5901231169700623,
310
  "step": 180
311
  },
312
  {
313
  "epoch": 0.54,
314
- "grad_norm": 1024.0437070924656,
315
- "learning_rate": 2.5500220223847356e-08,
316
- "logits/chosen": -4.346238613128662,
317
- "logits/rejected": -4.559948444366455,
318
- "logps/chosen": -290.6450500488281,
319
- "logps/rejected": -252.0496063232422,
320
- "loss": 0.3385,
321
- "rewards/accuracies": 0.8374999761581421,
322
- "rewards/chosen": 3.277946949005127,
323
- "rewards/margins": 2.545849323272705,
324
- "rewards/rejected": 0.7320975661277771,
325
  "step": 190
326
  },
327
  {
328
  "epoch": 0.57,
329
- "grad_norm": 937.2977444102543,
330
- "learning_rate": 2.3001121467819626e-08,
331
- "logits/chosen": -4.290076732635498,
332
- "logits/rejected": -4.555208683013916,
333
- "logps/chosen": -317.6783447265625,
334
- "logps/rejected": -267.9818115234375,
335
- "loss": 0.312,
336
- "rewards/accuracies": 0.824999988079071,
337
- "rewards/chosen": 3.4544365406036377,
338
- "rewards/margins": 2.3504996299743652,
339
- "rewards/rejected": 1.1039369106292725,
340
  "step": 200
341
  },
342
  {
343
  "epoch": 0.57,
344
- "eval_logits/chosen": -4.7096052169799805,
345
- "eval_logits/rejected": -4.93910026550293,
346
- "eval_logps/chosen": -405.4331970214844,
347
- "eval_logps/rejected": -520.6950073242188,
348
- "eval_loss": 1.959585189819336,
349
- "eval_rewards/accuracies": 0.33984375,
350
- "eval_rewards/chosen": -3.194929599761963,
351
- "eval_rewards/margins": -1.3785794973373413,
352
- "eval_rewards/rejected": -1.8163501024246216,
353
- "eval_runtime": 97.467,
354
- "eval_samples_per_second": 20.52,
355
- "eval_steps_per_second": 0.328,
356
  "step": 200
357
  },
358
  {
359
  "epoch": 0.6,
360
- "grad_norm": 839.0813036142636,
361
- "learning_rate": 2.0522015093886612e-08,
362
- "logits/chosen": -4.317226409912109,
363
- "logits/rejected": -4.618459224700928,
364
- "logps/chosen": -294.03253173828125,
365
- "logps/rejected": -234.196533203125,
366
- "loss": 0.2938,
367
- "rewards/accuracies": 0.8500000238418579,
368
- "rewards/chosen": 3.499882459640503,
369
- "rewards/margins": 2.7310853004455566,
370
- "rewards/rejected": 0.7687975168228149,
371
  "step": 210
372
  },
373
  {
374
  "epoch": 0.63,
375
- "grad_norm": 996.0408528817461,
376
- "learning_rate": 1.808769662668035e-08,
377
- "logits/chosen": -4.358494758605957,
378
- "logits/rejected": -4.704705238342285,
379
- "logps/chosen": -299.7410888671875,
380
- "logps/rejected": -235.092041015625,
381
- "loss": 0.2943,
382
- "rewards/accuracies": 0.8999999761581421,
383
- "rewards/chosen": 3.0801563262939453,
384
- "rewards/margins": 2.7920236587524414,
385
- "rewards/rejected": 0.28813308477401733,
386
  "step": 220
387
  },
388
  {
389
  "epoch": 0.66,
390
- "grad_norm": 781.9269603792492,
391
- "learning_rate": 1.5722513631174444e-08,
392
- "logits/chosen": -4.392434597015381,
393
- "logits/rejected": -4.748046398162842,
394
- "logps/chosen": -289.19927978515625,
395
- "logps/rejected": -234.593017578125,
396
- "loss": 0.3084,
397
- "rewards/accuracies": 0.856249988079071,
398
- "rewards/chosen": 2.994288921356201,
399
- "rewards/margins": 2.508836507797241,
400
- "rewards/rejected": 0.4854525625705719,
401
  "step": 230
402
  },
403
  {
404
  "epoch": 0.69,
405
- "grad_norm": 795.1407214961006,
406
- "learning_rate": 1.345012219322345e-08,
407
- "logits/chosen": -4.238420486450195,
408
- "logits/rejected": -4.548759460449219,
409
- "logps/chosen": -285.19232177734375,
410
- "logps/rejected": -240.72509765625,
411
- "loss": 0.3269,
412
- "rewards/accuracies": 0.8687499761581421,
413
- "rewards/chosen": 2.8941564559936523,
414
- "rewards/margins": 2.3365750312805176,
415
- "rewards/rejected": 0.5575811266899109,
416
  "step": 240
417
  },
418
  {
419
  "epoch": 0.72,
420
- "grad_norm": 879.951904317175,
421
- "learning_rate": 1.1293250316137664e-08,
422
- "logits/chosen": -4.293431282043457,
423
- "logits/rejected": -4.5221476554870605,
424
- "logps/chosen": -294.2128601074219,
425
- "logps/rejected": -254.7281494140625,
426
- "loss": 0.3131,
427
- "rewards/accuracies": 0.856249988079071,
428
- "rewards/chosen": 3.089404821395874,
429
- "rewards/margins": 2.4765713214874268,
430
- "rewards/rejected": 0.6128337383270264,
431
  "step": 250
432
  },
433
  {
434
  "epoch": 0.74,
435
- "grad_norm": 1094.0422336259135,
436
- "learning_rate": 9.273470599753375e-09,
437
- "logits/chosen": -4.372658729553223,
438
- "logits/rejected": -4.525150775909424,
439
- "logps/chosen": -294.89874267578125,
440
- "logps/rejected": -263.60577392578125,
441
- "loss": 0.3409,
442
- "rewards/accuracies": 0.831250011920929,
443
- "rewards/chosen": 2.7965197563171387,
444
- "rewards/margins": 2.07285737991333,
445
- "rewards/rejected": 0.7236624956130981,
446
  "step": 260
447
  },
448
  {
449
  "epoch": 0.77,
450
- "grad_norm": 775.0184329679462,
451
- "learning_rate": 7.410984475616819e-09,
452
- "logits/chosen": -4.244694709777832,
453
- "logits/rejected": -4.5279622077941895,
454
- "logps/chosen": -282.0103454589844,
455
- "logps/rejected": -239.31399536132812,
456
- "loss": 0.2834,
457
- "rewards/accuracies": 0.856249988079071,
458
- "rewards/chosen": 3.1753017902374268,
459
- "rewards/margins": 2.7872982025146484,
460
- "rewards/rejected": 0.38800328969955444,
461
  "step": 270
462
  },
463
  {
464
  "epoch": 0.8,
465
- "grad_norm": 796.3667611732337,
466
- "learning_rate": 5.724420156318405e-09,
467
- "logits/chosen": -4.291975498199463,
468
- "logits/rejected": -4.621099472045898,
469
- "logps/chosen": -285.0769958496094,
470
- "logps/rejected": -232.0653839111328,
471
- "loss": 0.3144,
472
- "rewards/accuracies": 0.8687499761581421,
473
- "rewards/chosen": 2.847846508026123,
474
- "rewards/margins": 2.3038220405578613,
475
- "rewards/rejected": 0.5440241098403931,
476
  "step": 280
477
  },
478
  {
479
  "epoch": 0.83,
480
- "grad_norm": 1101.8200803185337,
481
- "learning_rate": 4.230646319847259e-09,
482
- "logits/chosen": -4.424475193023682,
483
- "logits/rejected": -4.705140113830566,
484
- "logps/chosen": -295.53857421875,
485
- "logps/rejected": -232.05245971679688,
486
- "loss": 0.3121,
487
- "rewards/accuracies": 0.893750011920929,
488
- "rewards/chosen": 2.7051048278808594,
489
- "rewards/margins": 2.4193003177642822,
490
- "rewards/rejected": 0.2858046591281891,
491
  "step": 290
492
  },
493
  {
494
  "epoch": 0.86,
495
- "grad_norm": 1165.1379404537536,
496
- "learning_rate": 2.944603392457931e-09,
497
- "logits/chosen": -4.335474491119385,
498
- "logits/rejected": -4.49271297454834,
499
- "logps/chosen": -292.6704406738281,
500
- "logps/rejected": -252.3407440185547,
501
- "loss": 0.2993,
502
- "rewards/accuracies": 0.831250011920929,
503
- "rewards/chosen": 3.1620662212371826,
504
- "rewards/margins": 2.2422008514404297,
505
- "rewards/rejected": 0.9198653101921082,
506
  "step": 300
507
  },
508
  {
509
  "epoch": 0.86,
510
- "eval_logits/chosen": -4.706819534301758,
511
- "eval_logits/rejected": -4.936363697052002,
512
- "eval_logps/chosen": -405.887939453125,
513
- "eval_logps/rejected": -521.2874755859375,
514
- "eval_loss": 1.9827619791030884,
515
- "eval_rewards/accuracies": 0.35546875,
516
- "eval_rewards/chosen": -3.4222922325134277,
517
- "eval_rewards/margins": -1.3097174167633057,
518
- "eval_rewards/rejected": -2.112574815750122,
519
- "eval_runtime": 97.4214,
520
- "eval_samples_per_second": 20.529,
521
- "eval_steps_per_second": 0.328,
522
  "step": 300
523
  },
524
  {
525
  "epoch": 0.89,
526
- "grad_norm": 1000.5341122694366,
527
- "learning_rate": 1.8791541175240787e-09,
528
- "logits/chosen": -4.3908562660217285,
529
- "logits/rejected": -4.607417583465576,
530
- "logps/chosen": -301.6288146972656,
531
- "logps/rejected": -246.2473907470703,
532
- "loss": 0.2942,
533
- "rewards/accuracies": 0.862500011920929,
534
- "rewards/chosen": 3.3309147357940674,
535
- "rewards/margins": 2.577249765396118,
536
- "rewards/rejected": 0.753665030002594,
537
  "step": 310
538
  },
539
  {
540
  "epoch": 0.92,
541
- "grad_norm": 942.4369602130394,
542
- "learning_rate": 1.0449549049596136e-09,
543
- "logits/chosen": -4.230744361877441,
544
- "logits/rejected": -4.579751968383789,
545
- "logps/chosen": -287.4718322753906,
546
- "logps/rejected": -232.3734893798828,
547
- "loss": 0.3078,
548
- "rewards/accuracies": 0.862500011920929,
549
- "rewards/chosen": 3.2846877574920654,
550
- "rewards/margins": 2.4689810276031494,
551
- "rewards/rejected": 0.8157066106796265,
552
  "step": 320
553
  },
554
  {
555
  "epoch": 0.94,
556
- "grad_norm": 1041.373917696444,
557
- "learning_rate": 4.5034924794443707e-10,
558
- "logits/chosen": -4.3931074142456055,
559
- "logits/rejected": -4.620477199554443,
560
- "logps/chosen": -280.0606994628906,
561
- "logps/rejected": -233.14346313476562,
562
- "loss": 0.2917,
563
- "rewards/accuracies": 0.856249988079071,
564
- "rewards/chosen": 3.2919604778289795,
565
- "rewards/margins": 2.5820116996765137,
566
- "rewards/rejected": 0.7099487781524658,
567
  "step": 330
568
  },
569
  {
570
  "epoch": 0.97,
571
- "grad_norm": 859.9709811153568,
572
- "learning_rate": 1.0128427297940723e-10,
573
- "logits/chosen": -4.328027248382568,
574
- "logits/rejected": -4.5136332511901855,
575
- "logps/chosen": -284.2748718261719,
576
- "logps/rejected": -244.03817749023438,
577
- "loss": 0.2952,
578
- "rewards/accuracies": 0.862500011920929,
579
- "rewards/chosen": 3.204352855682373,
580
- "rewards/margins": 2.4567666053771973,
581
- "rewards/rejected": 0.7475861310958862,
582
  "step": 340
583
  },
584
  {
585
  "epoch": 1.0,
586
  "step": 349,
587
  "total_flos": 0.0,
588
- "train_loss": 0.3696309270011661,
589
- "train_runtime": 5591.2149,
590
- "train_samples_per_second": 7.991,
591
- "train_steps_per_second": 0.062
592
  }
593
  ],
594
  "logging_steps": 10,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.0,
13
+ "grad_norm": 1483.079147156977,
14
+ "learning_rate": 5.7142857142857136e-09,
15
  "logits/chosen": -4.490396976470947,
16
  "logits/rejected": -4.787891387939453,
17
  "logps/chosen": -300.56573486328125,
 
25
  },
26
  {
27
  "epoch": 0.03,
28
+ "grad_norm": 1678.9009350663757,
29
+ "learning_rate": 5.714285714285714e-08,
30
+ "logits/chosen": -4.292892932891846,
31
+ "logits/rejected": -4.4704790115356445,
32
+ "logps/chosen": -285.7367858886719,
33
+ "logps/rejected": -241.40879821777344,
34
+ "loss": 0.7232,
35
+ "rewards/accuracies": 0.4375,
36
+ "rewards/chosen": 0.03021715022623539,
37
+ "rewards/margins": 0.018014244735240936,
38
+ "rewards/rejected": 0.012202905490994453,
39
  "step": 10
40
  },
41
  {
42
  "epoch": 0.06,
43
+ "grad_norm": 1307.5912774162387,
44
+ "learning_rate": 1.1428571428571427e-07,
45
+ "logits/chosen": -4.212913990020752,
46
+ "logits/rejected": -4.504936695098877,
47
+ "logps/chosen": -312.25714111328125,
48
+ "logps/rejected": -247.5605926513672,
49
+ "loss": 0.6348,
50
+ "rewards/accuracies": 0.6187499761581421,
51
+ "rewards/chosen": 0.25936827063560486,
52
+ "rewards/margins": 0.15551437437534332,
53
+ "rewards/rejected": 0.10385391861200333,
54
  "step": 20
55
  },
56
  {
57
  "epoch": 0.09,
58
+ "grad_norm": 877.5830541670848,
59
+ "learning_rate": 1.714285714285714e-07,
60
+ "logits/chosen": -4.158999919891357,
61
+ "logits/rejected": -4.378296852111816,
62
+ "logps/chosen": -320.1895446777344,
63
+ "logps/rejected": -273.6168518066406,
64
+ "loss": 0.4394,
65
+ "rewards/accuracies": 0.793749988079071,
66
+ "rewards/chosen": 1.6762861013412476,
67
+ "rewards/margins": 1.1370527744293213,
68
+ "rewards/rejected": 0.5392333269119263,
69
  "step": 30
70
  },
71
  {
72
  "epoch": 0.11,
73
+ "grad_norm": 1094.7268353859254,
74
+ "learning_rate": 1.9987489925657699e-07,
75
+ "logits/chosen": -4.327412128448486,
76
+ "logits/rejected": -4.590438365936279,
77
+ "logps/chosen": -272.52813720703125,
78
+ "logps/rejected": -226.4479522705078,
79
+ "loss": 0.3519,
80
+ "rewards/accuracies": 0.8374999761581421,
81
+ "rewards/chosen": 3.0370898246765137,
82
+ "rewards/margins": 2.1227898597717285,
83
+ "rewards/rejected": 0.9143003225326538,
84
  "step": 40
85
  },
86
  {
87
  "epoch": 0.14,
88
+ "grad_norm": 1142.545986046605,
89
+ "learning_rate": 1.98875970549573e-07,
90
+ "logits/chosen": -4.374251365661621,
91
+ "logits/rejected": -4.627996921539307,
92
+ "logps/chosen": -264.49615478515625,
93
+ "logps/rejected": -226.27197265625,
94
+ "loss": 0.3689,
95
+ "rewards/accuracies": 0.831250011920929,
96
+ "rewards/chosen": 4.1793413162231445,
97
+ "rewards/margins": 2.736725091934204,
98
+ "rewards/rejected": 1.4426158666610718,
99
  "step": 50
100
  },
101
  {
102
  "epoch": 0.17,
103
+ "grad_norm": 635.1242404757879,
104
+ "learning_rate": 1.968881042201029e-07,
105
+ "logits/chosen": -4.35978889465332,
106
+ "logits/rejected": -4.594507694244385,
107
+ "logps/chosen": -296.86907958984375,
108
+ "logps/rejected": -251.1016845703125,
109
+ "loss": 0.3246,
110
+ "rewards/accuracies": 0.8062499761581421,
111
+ "rewards/chosen": 4.9698944091796875,
112
+ "rewards/margins": 3.8340022563934326,
113
+ "rewards/rejected": 1.1358922719955444,
114
  "step": 60
115
  },
116
  {
117
  "epoch": 0.2,
118
+ "grad_norm": 871.5895225513464,
119
+ "learning_rate": 1.9393118250841894e-07,
120
+ "logits/chosen": -4.349625587463379,
121
+ "logits/rejected": -4.586968421936035,
122
+ "logps/chosen": -285.0860595703125,
123
+ "logps/rejected": -241.4550018310547,
124
+ "loss": 0.3259,
125
+ "rewards/accuracies": 0.8812500238418579,
126
+ "rewards/chosen": 4.21413516998291,
127
+ "rewards/margins": 3.7849984169006348,
128
+ "rewards/rejected": 0.42913690209388733,
129
  "step": 70
130
  },
131
  {
132
  "epoch": 0.23,
133
+ "grad_norm": 869.2800598030351,
134
+ "learning_rate": 1.900347799523094e-07,
135
+ "logits/chosen": -4.277868747711182,
136
+ "logits/rejected": -4.535857677459717,
137
+ "logps/chosen": -318.2683410644531,
138
+ "logps/rejected": -272.18206787109375,
139
+ "loss": 0.2501,
140
+ "rewards/accuracies": 0.9375,
141
+ "rewards/chosen": 4.3253655433654785,
142
+ "rewards/margins": 4.170114040374756,
143
+ "rewards/rejected": 0.15525080263614655,
144
  "step": 80
145
  },
146
  {
147
  "epoch": 0.26,
148
+ "grad_norm": 616.9171514391553,
149
+ "learning_rate": 1.8523786758850436e-07,
150
+ "logits/chosen": -4.401219844818115,
151
+ "logits/rejected": -4.6477251052856445,
152
+ "logps/chosen": -280.7421569824219,
153
+ "logps/rejected": -236.2926025390625,
154
+ "loss": 0.2859,
155
+ "rewards/accuracies": 0.856249988079071,
156
+ "rewards/chosen": 3.923023223876953,
157
+ "rewards/margins": 4.055254936218262,
158
+ "rewards/rejected": -0.13223108649253845,
159
  "step": 90
160
  },
161
  {
162
  "epoch": 0.29,
163
+ "grad_norm": 735.4670744748103,
164
+ "learning_rate": 1.795884231721841e-07,
165
+ "logits/chosen": -4.318561553955078,
166
+ "logits/rejected": -4.602439880371094,
167
+ "logps/chosen": -267.45794677734375,
168
+ "logps/rejected": -232.0417938232422,
169
+ "loss": 0.3114,
170
+ "rewards/accuracies": 0.875,
171
+ "rewards/chosen": 4.1319708824157715,
172
+ "rewards/margins": 4.702456951141357,
173
+ "rewards/rejected": -0.5704857110977173,
174
  "step": 100
175
  },
176
  {
177
  "epoch": 0.29,
178
+ "eval_logits/chosen": -3.226163148880005,
179
+ "eval_logits/rejected": -3.226163148880005,
180
+ "eval_logps/chosen": -161.294677734375,
181
+ "eval_logps/rejected": -161.294677734375,
182
+ "eval_loss": 0.6931471824645996,
183
+ "eval_rewards/accuracies": 0.0,
184
+ "eval_rewards/chosen": -2.8588294982910156,
185
+ "eval_rewards/margins": 0.0,
186
+ "eval_rewards/rejected": -2.8588294982910156,
187
+ "eval_runtime": 1.5114,
188
+ "eval_samples_per_second": 0.662,
189
+ "eval_steps_per_second": 0.662,
190
  "step": 100
191
  },
192
  {
193
  "epoch": 0.31,
194
+ "grad_norm": 817.8641381018862,
195
+ "learning_rate": 1.7314295131309637e-07,
196
+ "logits/chosen": -4.324714660644531,
197
+ "logits/rejected": -4.614729404449463,
198
+ "logps/chosen": -293.42364501953125,
199
+ "logps/rejected": -245.01620483398438,
200
+ "loss": 0.3079,
201
+ "rewards/accuracies": 0.8812500238418579,
202
+ "rewards/chosen": 4.0577192306518555,
203
+ "rewards/margins": 4.518521308898926,
204
+ "rewards/rejected": -0.4608024060726166,
205
  "step": 110
206
  },
207
  {
208
  "epoch": 0.34,
209
+ "grad_norm": 926.642960527248,
210
+ "learning_rate": 1.6596591832778466e-07,
211
+ "logits/chosen": -4.334780216217041,
212
+ "logits/rejected": -4.567566394805908,
213
+ "logps/chosen": -298.56976318359375,
214
+ "logps/rejected": -266.8869323730469,
215
+ "loss": 0.2848,
216
+ "rewards/accuracies": 0.84375,
217
+ "rewards/chosen": 3.83539080619812,
218
+ "rewards/margins": 4.040234088897705,
219
+ "rewards/rejected": -0.20484237372875214,
220
  "step": 120
221
  },
222
  {
223
  "epoch": 0.37,
224
+ "grad_norm": 675.9811999128359,
225
+ "learning_rate": 1.5812910746042256e-07,
226
+ "logits/chosen": -4.3064866065979,
227
+ "logits/rejected": -4.553826332092285,
228
+ "logps/chosen": -286.5328063964844,
229
+ "logps/rejected": -242.8630828857422,
230
+ "loss": 0.2782,
231
+ "rewards/accuracies": 0.893750011920929,
232
+ "rewards/chosen": 3.7822985649108887,
233
+ "rewards/margins": 4.371214389801025,
234
+ "rewards/rejected": -0.5889158248901367,
235
  "step": 130
236
  },
237
  {
238
  "epoch": 0.4,
239
+ "grad_norm": 1048.4484509272795,
240
+ "learning_rate": 1.4971090092120542e-07,
241
+ "logits/chosen": -4.304565906524658,
242
+ "logits/rejected": -4.545400142669678,
243
+ "logps/chosen": -294.17095947265625,
244
+ "logps/rejected": -243.83935546875,
245
+ "loss": 0.269,
246
+ "rewards/accuracies": 0.9125000238418579,
247
+ "rewards/chosen": 3.8339390754699707,
248
+ "rewards/margins": 4.558647155761719,
249
+ "rewards/rejected": -0.7247086763381958,
250
  "step": 140
251
  },
252
  {
253
  "epoch": 0.43,
254
+ "grad_norm": 844.0558457976633,
255
+ "learning_rate": 1.4079549592320782e-07,
256
+ "logits/chosen": -4.313709735870361,
257
+ "logits/rejected": -4.621119499206543,
258
+ "logps/chosen": -292.96661376953125,
259
+ "logps/rejected": -248.0418701171875,
260
+ "loss": 0.2727,
261
+ "rewards/accuracies": 0.856249988079071,
262
+ "rewards/chosen": 3.8533847332000732,
263
+ "rewards/margins": 4.262558937072754,
264
+ "rewards/rejected": -0.4091736674308777,
265
  "step": 150
266
  },
267
  {
268
  "epoch": 0.46,
269
+ "grad_norm": 684.2161612155076,
270
+ "learning_rate": 1.3147206255874882e-07,
271
+ "logits/chosen": -4.248518943786621,
272
+ "logits/rejected": -4.44361686706543,
273
+ "logps/chosen": -309.7388610839844,
274
+ "logps/rejected": -272.90252685546875,
275
+ "loss": 0.2961,
276
+ "rewards/accuracies": 0.90625,
277
+ "rewards/chosen": 4.352166652679443,
278
+ "rewards/margins": 4.611546516418457,
279
+ "rewards/rejected": -0.2593800127506256,
280
  "step": 160
281
  },
282
  {
283
  "epoch": 0.49,
284
+ "grad_norm": 795.319436926025,
285
+ "learning_rate": 1.2183385193801653e-07,
286
+ "logits/chosen": -4.289888381958008,
287
+ "logits/rejected": -4.489100933074951,
288
+ "logps/chosen": -272.3153991699219,
289
+ "logps/rejected": -237.64114379882812,
290
+ "loss": 0.2547,
291
+ "rewards/accuracies": 0.918749988079071,
292
+ "rewards/chosen": 4.135977745056152,
293
+ "rewards/margins": 4.544943809509277,
294
+ "rewards/rejected": -0.40896692872047424,
295
  "step": 170
296
  },
297
  {
298
  "epoch": 0.52,
299
+ "grad_norm": 917.2913745554154,
300
+ "learning_rate": 1.1197726351017051e-07,
301
+ "logits/chosen": -4.37355375289917,
302
+ "logits/rejected": -4.5346221923828125,
303
+ "logps/chosen": -261.6694641113281,
304
+ "logps/rejected": -230.89599609375,
305
+ "loss": 0.3104,
306
+ "rewards/accuracies": 0.862500011920929,
307
+ "rewards/chosen": 3.4763107299804688,
308
+ "rewards/margins": 3.877528429031372,
309
+ "rewards/rejected": -0.4012181758880615,
310
  "step": 180
311
  },
312
  {
313
  "epoch": 0.54,
314
+ "grad_norm": 880.6071936953809,
315
+ "learning_rate": 1.0200088089538943e-07,
316
+ "logits/chosen": -4.33815860748291,
317
+ "logits/rejected": -4.5543646812438965,
318
+ "logps/chosen": -288.26873779296875,
319
+ "logps/rejected": -253.7147216796875,
320
+ "loss": 0.3129,
321
+ "rewards/accuracies": 0.856249988079071,
322
+ "rewards/chosen": 4.4660868644714355,
323
+ "rewards/margins": 4.566544055938721,
324
+ "rewards/rejected": -0.10045762360095978,
325
  "step": 190
326
  },
327
  {
328
  "epoch": 0.57,
329
+ "grad_norm": 781.3066688232569,
330
+ "learning_rate": 9.20044858712785e-08,
331
+ "logits/chosen": -4.292983055114746,
332
+ "logits/rejected": -4.566291332244873,
333
+ "logps/chosen": -315.58251953125,
334
+ "logps/rejected": -270.0286560058594,
335
+ "loss": 0.2741,
336
+ "rewards/accuracies": 0.862500011920929,
337
+ "rewards/chosen": 4.502341270446777,
338
+ "rewards/margins": 4.4218339920043945,
339
+ "rewards/rejected": 0.08050797879695892,
340
  "step": 200
341
  },
342
  {
343
  "epoch": 0.57,
344
+ "eval_logits/chosen": -3.2033309936523438,
345
+ "eval_logits/rejected": -3.2033309936523438,
346
+ "eval_logps/chosen": -160.92892456054688,
347
+ "eval_logps/rejected": -160.92892456054688,
348
+ "eval_loss": 0.6931471824645996,
349
+ "eval_rewards/accuracies": 0.0,
350
+ "eval_rewards/chosen": -2.675952911376953,
351
+ "eval_rewards/margins": 0.0,
352
+ "eval_rewards/rejected": -2.675952911376953,
353
+ "eval_runtime": 1.48,
354
+ "eval_samples_per_second": 0.676,
355
+ "eval_steps_per_second": 0.676,
356
  "step": 200
357
  },
358
  {
359
  "epoch": 0.6,
360
+ "grad_norm": 1021.5696401893487,
361
+ "learning_rate": 8.208806037554645e-08,
362
+ "logits/chosen": -4.327147483825684,
363
+ "logits/rejected": -4.6312479972839355,
364
+ "logps/chosen": -292.85919189453125,
365
+ "logps/rejected": -236.7397918701172,
366
+ "loss": 0.2345,
367
+ "rewards/accuracies": 0.875,
368
+ "rewards/chosen": 4.086556434631348,
369
+ "rewards/margins": 4.589372634887695,
370
+ "rewards/rejected": -0.5028160810470581,
371
  "step": 210
372
  },
373
  {
374
  "epoch": 0.63,
375
+ "grad_norm": 720.6830161967047,
376
+ "learning_rate": 7.23507865067214e-08,
377
+ "logits/chosen": -4.357525825500488,
378
+ "logits/rejected": -4.7082133293151855,
379
+ "logps/chosen": -298.2110900878906,
380
+ "logps/rejected": -237.28903198242188,
381
+ "loss": 0.241,
382
+ "rewards/accuracies": 0.9125000238418579,
383
+ "rewards/chosen": 3.84515118598938,
384
+ "rewards/margins": 4.655531406402588,
385
+ "rewards/rejected": -0.8103805780410767,
386
  "step": 220
387
  },
388
  {
389
  "epoch": 0.66,
390
+ "grad_norm": 784.9312094778309,
391
+ "learning_rate": 6.289005452469778e-08,
392
+ "logits/chosen": -4.388636112213135,
393
+ "logits/rejected": -4.748753070831299,
394
+ "logps/chosen": -288.0061340332031,
395
+ "logps/rejected": -236.6077880859375,
396
+ "loss": 0.268,
397
+ "rewards/accuracies": 0.9125000238418579,
398
+ "rewards/chosen": 3.5908846855163574,
399
+ "rewards/margins": 4.1128339767456055,
400
+ "rewards/rejected": -0.5219489932060242,
401
  "step": 230
402
  },
403
  {
404
  "epoch": 0.69,
405
+ "grad_norm": 765.61870702253,
406
+ "learning_rate": 5.38004887728938e-08,
407
+ "logits/chosen": -4.234927177429199,
408
+ "logits/rejected": -4.550938606262207,
409
+ "logps/chosen": -283.8558654785156,
410
+ "logps/rejected": -242.86538696289062,
411
+ "loss": 0.2908,
412
+ "rewards/accuracies": 0.887499988079071,
413
+ "rewards/chosen": 3.562377452850342,
414
+ "rewards/margins": 4.074954032897949,
415
+ "rewards/rejected": -0.5125768780708313,
416
  "step": 240
417
  },
418
  {
419
  "epoch": 0.72,
420
+ "grad_norm": 787.5249084258048,
421
+ "learning_rate": 4.517300126455066e-08,
422
+ "logits/chosen": -4.290322303771973,
423
+ "logits/rejected": -4.522026538848877,
424
+ "logps/chosen": -292.47442626953125,
425
+ "logps/rejected": -257.0675354003906,
426
+ "loss": 0.2635,
427
+ "rewards/accuracies": 0.8812500238418579,
428
+ "rewards/chosen": 3.95862078666687,
429
+ "rewards/margins": 4.515480995178223,
430
+ "rewards/rejected": -0.5568601489067078,
431
  "step": 250
432
  },
433
  {
434
  "epoch": 0.74,
435
+ "grad_norm": 894.7487355536125,
436
+ "learning_rate": 3.70938823990135e-08,
437
+ "logits/chosen": -4.367636680603027,
438
+ "logits/rejected": -4.524864196777344,
439
+ "logps/chosen": -293.8681640625,
440
+ "logps/rejected": -266.19622802734375,
441
+ "loss": 0.2999,
442
+ "rewards/accuracies": 0.887499988079071,
443
+ "rewards/chosen": 3.311814069747925,
444
+ "rewards/margins": 3.883411407470703,
445
+ "rewards/rejected": -0.5715969800949097,
446
  "step": 260
447
  },
448
  {
449
  "epoch": 0.77,
450
+ "grad_norm": 582.5138549963486,
451
+ "learning_rate": 2.9643937902467277e-08,
452
+ "logits/chosen": -4.23429012298584,
453
+ "logits/rejected": -4.521827697753906,
454
+ "logps/chosen": -280.5379333496094,
455
+ "logps/rejected": -241.7247314453125,
456
+ "loss": 0.2388,
457
+ "rewards/accuracies": 0.862500011920929,
458
+ "rewards/chosen": 3.911522626876831,
459
+ "rewards/margins": 4.728894233703613,
460
+ "rewards/rejected": -0.8173705339431763,
461
  "step": 270
462
  },
463
  {
464
  "epoch": 0.8,
465
+ "grad_norm": 810.3954826895692,
466
+ "learning_rate": 2.289768062527362e-08,
467
+ "logits/chosen": -4.273584842681885,
468
+ "logits/rejected": -4.6050519943237305,
469
+ "logps/chosen": -283.619384765625,
470
+ "logps/rejected": -233.9824981689453,
471
+ "loss": 0.2895,
472
+ "rewards/accuracies": 0.8812500238418579,
473
+ "rewards/chosen": 3.576643466949463,
474
+ "rewards/margins": 3.991179943084717,
475
+ "rewards/rejected": -0.41453617811203003,
476
  "step": 280
477
  },
478
  {
479
  "epoch": 0.83,
480
+ "grad_norm": 730.6997584190797,
481
+ "learning_rate": 1.6922585279389035e-08,
482
+ "logits/chosen": -4.406073093414307,
483
+ "logits/rejected": -4.6901140213012695,
484
+ "logps/chosen": -294.35699462890625,
485
+ "logps/rejected": -233.74594116210938,
486
+ "loss": 0.2669,
487
+ "rewards/accuracies": 0.90625,
488
+ "rewards/chosen": 3.295900344848633,
489
+ "rewards/margins": 3.856853485107422,
490
+ "rewards/rejected": -0.5609533786773682,
491
  "step": 290
492
  },
493
  {
494
  "epoch": 0.86,
495
+ "grad_norm": 1254.3257141110294,
496
+ "learning_rate": 1.1778413569831725e-08,
497
+ "logits/chosen": -4.323153495788574,
498
+ "logits/rejected": -4.481599807739258,
499
+ "logps/chosen": -291.29742431640625,
500
+ "logps/rejected": -254.1814422607422,
501
+ "loss": 0.2695,
502
+ "rewards/accuracies": 0.856249988079071,
503
+ "rewards/chosen": 3.8485710620880127,
504
+ "rewards/margins": 3.8490569591522217,
505
+ "rewards/rejected": -0.0004864335060119629,
506
  "step": 300
507
  },
508
  {
509
  "epoch": 0.86,
510
+ "eval_logits/chosen": -3.199521064758301,
511
+ "eval_logits/rejected": -3.199521064758301,
512
+ "eval_logps/chosen": -159.83721923828125,
513
+ "eval_logps/rejected": -159.83721923828125,
514
+ "eval_loss": 0.6931471824645996,
515
+ "eval_rewards/accuracies": 0.0,
516
+ "eval_rewards/chosen": -2.1301021575927734,
517
+ "eval_rewards/margins": 0.0,
518
+ "eval_rewards/rejected": -2.1301021575927734,
519
+ "eval_runtime": 1.4681,
520
+ "eval_samples_per_second": 0.681,
521
+ "eval_steps_per_second": 0.681,
522
  "step": 300
523
  },
524
  {
525
  "epoch": 0.89,
526
+ "grad_norm": 1302.095010227008,
527
+ "learning_rate": 7.516616470096315e-09,
528
+ "logits/chosen": -4.3807525634765625,
529
+ "logits/rejected": -4.600883483886719,
530
+ "logps/chosen": -299.8600158691406,
531
+ "logps/rejected": -248.4993438720703,
532
+ "loss": 0.2545,
533
+ "rewards/accuracies": 0.875,
534
+ "rewards/chosen": 4.215323448181152,
535
+ "rewards/margins": 4.587640762329102,
536
+ "rewards/rejected": -0.3723169267177582,
537
  "step": 310
538
  },
539
  {
540
  "epoch": 0.92,
541
+ "grad_norm": 875.4939401121867,
542
+ "learning_rate": 4.179819619838454e-09,
543
+ "logits/chosen": -4.2180280685424805,
544
+ "logits/rejected": -4.570239067077637,
545
+ "logps/chosen": -286.0614013671875,
546
+ "logps/rejected": -234.3032684326172,
547
+ "loss": 0.269,
548
+ "rewards/accuracies": 0.8374999761581421,
549
+ "rewards/chosen": 3.9898929595947266,
550
+ "rewards/margins": 4.13907527923584,
551
+ "rewards/rejected": -0.1491830050945282,
552
  "step": 320
553
  },
554
  {
555
  "epoch": 0.94,
556
+ "grad_norm": 1010.1848642033616,
557
+ "learning_rate": 1.8013969917777483e-09,
558
+ "logits/chosen": -4.380679130554199,
559
+ "logits/rejected": -4.608093738555908,
560
+ "logps/chosen": -278.8153381347656,
561
+ "logps/rejected": -235.4047088623047,
562
+ "loss": 0.247,
563
+ "rewards/accuracies": 0.887499988079071,
564
+ "rewards/chosen": 3.914651870727539,
565
+ "rewards/margins": 4.33533239364624,
566
+ "rewards/rejected": -0.42068085074424744,
567
  "step": 330
568
  },
569
  {
570
  "epoch": 0.97,
571
+ "grad_norm": 702.698454631269,
572
+ "learning_rate": 4.051370919176289e-10,
573
+ "logits/chosen": -4.314477920532227,
574
+ "logits/rejected": -4.502932548522949,
575
+ "logps/chosen": -282.89666748046875,
576
+ "logps/rejected": -246.3631134033203,
577
+ "loss": 0.243,
578
+ "rewards/accuracies": 0.90625,
579
+ "rewards/chosen": 3.8934428691864014,
580
+ "rewards/margins": 4.308337688446045,
581
+ "rewards/rejected": -0.41489481925964355,
582
  "step": 340
583
  },
584
  {
585
  "epoch": 1.0,
586
  "step": 349,
587
  "total_flos": 0.0,
588
+ "train_loss": 0.30806411576476,
589
+ "train_runtime": 5299.0292,
590
+ "train_samples_per_second": 8.432,
591
+ "train_steps_per_second": 0.066
592
  }
593
  ],
594
  "logging_steps": 10,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3101c501cb9463cb757d57fc1ba9afd767800873490708bfba47496b8c2a73c9
3
- size 6520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3f943ec10c03c0b36e6f7cf63a55e819cb580b98a2a33846c2e1059e1f696c2
3
+ size 6328