File size: 8,896 Bytes
de6b535
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9984,
  "eval_steps": 100,
  "global_step": 156,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 3.125e-07,
      "logits/chosen": -2.783127546310425,
      "logits/rejected": -2.8651208877563477,
      "logps/chosen": -134.2432861328125,
      "logps/rejected": -141.6387939453125,
      "loss": 0.6931,
      "rewards/accuracies": 0.0,
      "rewards/chosen": 0.0,
      "rewards/margins": 0.0,
      "rewards/rejected": 0.0,
      "step": 1
    },
    {
      "epoch": 0.06,
      "learning_rate": 3.125e-06,
      "logits/chosen": -2.801163911819458,
      "logits/rejected": -2.7942428588867188,
      "logps/chosen": -112.86317443847656,
      "logps/rejected": -110.90413665771484,
      "loss": 0.6931,
      "rewards/accuracies": 0.4444444477558136,
      "rewards/chosen": 0.0007365171913988888,
      "rewards/margins": -3.1015442800708115e-05,
      "rewards/rejected": 0.0007675326196476817,
      "step": 10
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.989935734988098e-06,
      "logits/chosen": -2.7939412593841553,
      "logits/rejected": -2.761488437652588,
      "logps/chosen": -111.9773178100586,
      "logps/rejected": -112.6773681640625,
      "loss": 0.693,
      "rewards/accuracies": 0.5531250238418579,
      "rewards/chosen": 0.007041769567877054,
      "rewards/margins": 0.0005797661142423749,
      "rewards/rejected": 0.006462003104388714,
      "step": 20
    },
    {
      "epoch": 0.19,
      "learning_rate": 4.8776412907378845e-06,
      "logits/chosen": -2.7452473640441895,
      "logits/rejected": -2.71763277053833,
      "logps/chosen": -110.77873229980469,
      "logps/rejected": -116.26164245605469,
      "loss": 0.6919,
      "rewards/accuracies": 0.590624988079071,
      "rewards/chosen": 0.027953902259469032,
      "rewards/margins": 0.002175498753786087,
      "rewards/rejected": 0.025778403505682945,
      "step": 30
    },
    {
      "epoch": 0.26,
      "learning_rate": 4.646121984004666e-06,
      "logits/chosen": -2.7981982231140137,
      "logits/rejected": -2.760676860809326,
      "logps/chosen": -110.84574890136719,
      "logps/rejected": -111.5237808227539,
      "loss": 0.6908,
      "rewards/accuracies": 0.606249988079071,
      "rewards/chosen": 0.05981600284576416,
      "rewards/margins": 0.0045239622704684734,
      "rewards/rejected": 0.055292047560214996,
      "step": 40
    },
    {
      "epoch": 0.32,
      "learning_rate": 4.3069871595684795e-06,
      "logits/chosen": -2.7610268592834473,
      "logits/rejected": -2.7524187564849854,
      "logps/chosen": -109.18692779541016,
      "logps/rejected": -110.45135498046875,
      "loss": 0.6898,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": 0.07008253037929535,
      "rewards/margins": 0.003140285611152649,
      "rewards/rejected": 0.0669422522187233,
      "step": 50
    },
    {
      "epoch": 0.38,
      "learning_rate": 3.8772424536302565e-06,
      "logits/chosen": -2.7594542503356934,
      "logits/rejected": -2.7281312942504883,
      "logps/chosen": -105.79144287109375,
      "logps/rejected": -108.7798080444336,
      "loss": 0.6899,
      "rewards/accuracies": 0.606249988079071,
      "rewards/chosen": 0.08012694865465164,
      "rewards/margins": 0.008532116189599037,
      "rewards/rejected": 0.07159483432769775,
      "step": 60
    },
    {
      "epoch": 0.45,
      "learning_rate": 3.3784370602033572e-06,
      "logits/chosen": -2.7187533378601074,
      "logits/rejected": -2.6961326599121094,
      "logps/chosen": -120.80352783203125,
      "logps/rejected": -117.37217712402344,
      "loss": 0.6883,
      "rewards/accuracies": 0.5375000238418579,
      "rewards/chosen": 0.05428874492645264,
      "rewards/margins": 0.007891224697232246,
      "rewards/rejected": 0.04639752581715584,
      "step": 70
    },
    {
      "epoch": 0.51,
      "learning_rate": 2.835583164544139e-06,
      "logits/chosen": -2.7139596939086914,
      "logits/rejected": -2.6886355876922607,
      "logps/chosen": -117.95060729980469,
      "logps/rejected": -119.08524322509766,
      "loss": 0.6878,
      "rewards/accuracies": 0.5843750238418579,
      "rewards/chosen": 0.04534279555082321,
      "rewards/margins": 0.010814065113663673,
      "rewards/rejected": 0.03452872857451439,
      "step": 80
    },
    {
      "epoch": 0.58,
      "learning_rate": 2.2759017277414165e-06,
      "logits/chosen": -2.7270092964172363,
      "logits/rejected": -2.677152633666992,
      "logps/chosen": -109.81172943115234,
      "logps/rejected": -110.4306640625,
      "loss": 0.6872,
      "rewards/accuracies": 0.550000011920929,
      "rewards/chosen": 0.026867974549531937,
      "rewards/margins": 0.010544367134571075,
      "rewards/rejected": 0.01632360741496086,
      "step": 90
    },
    {
      "epoch": 0.64,
      "learning_rate": 1.7274575140626318e-06,
      "logits/chosen": -2.720876932144165,
      "logits/rejected": -2.7096195220947266,
      "logps/chosen": -115.74837493896484,
      "logps/rejected": -114.7901382446289,
      "loss": 0.6844,
      "rewards/accuracies": 0.5531250238418579,
      "rewards/chosen": 0.042589299380779266,
      "rewards/margins": 0.017415037378668785,
      "rewards/rejected": 0.02517426386475563,
      "step": 100
    },
    {
      "epoch": 0.64,
      "eval_logits/chosen": -2.722978353500366,
      "eval_logits/rejected": -2.648787021636963,
      "eval_logps/chosen": -280.98046875,
      "eval_logps/rejected": -257.7253723144531,
      "eval_loss": 0.677879810333252,
      "eval_rewards/accuracies": 0.6480000019073486,
      "eval_rewards/chosen": 0.03059276193380356,
      "eval_rewards/margins": 0.032687921077013016,
      "eval_rewards/rejected": -0.0020951556507498026,
      "eval_runtime": 383.9393,
      "eval_samples_per_second": 5.209,
      "eval_steps_per_second": 0.651,
      "step": 100
    },
    {
      "epoch": 0.7,
      "learning_rate": 1.217751806485235e-06,
      "logits/chosen": -2.6989266872406006,
      "logits/rejected": -2.669929265975952,
      "logps/chosen": -106.67137145996094,
      "logps/rejected": -105.73466491699219,
      "loss": 0.6849,
      "rewards/accuracies": 0.640625,
      "rewards/chosen": 0.05706842616200447,
      "rewards/margins": 0.01945202611386776,
      "rewards/rejected": 0.03761639818549156,
      "step": 110
    },
    {
      "epoch": 0.77,
      "learning_rate": 7.723433775328385e-07,
      "logits/chosen": -2.6370861530303955,
      "logits/rejected": -2.6202707290649414,
      "logps/chosen": -107.36185455322266,
      "logps/rejected": -108.76318359375,
      "loss": 0.6858,
      "rewards/accuracies": 0.5843750238418579,
      "rewards/chosen": 0.06415753811597824,
      "rewards/margins": 0.012921139597892761,
      "rewards/rejected": 0.05123639851808548,
      "step": 120
    },
    {
      "epoch": 0.83,
      "learning_rate": 4.1356686569674344e-07,
      "logits/chosen": -2.7071995735168457,
      "logits/rejected": -2.6700949668884277,
      "logps/chosen": -110.8349609375,
      "logps/rejected": -116.52140045166016,
      "loss": 0.6859,
      "rewards/accuracies": 0.581250011920929,
      "rewards/chosen": 0.062352847307920456,
      "rewards/margins": 0.018708955496549606,
      "rewards/rejected": 0.04364389181137085,
      "step": 130
    },
    {
      "epoch": 0.9,
      "learning_rate": 1.59412823400657e-07,
      "logits/chosen": -2.7012362480163574,
      "logits/rejected": -2.673867702484131,
      "logps/chosen": -114.33103942871094,
      "logps/rejected": -117.1007308959961,
      "loss": 0.6849,
      "rewards/accuracies": 0.612500011920929,
      "rewards/chosen": 0.055680979043245316,
      "rewards/margins": 0.020022045820951462,
      "rewards/rejected": 0.03565892577171326,
      "step": 140
    },
    {
      "epoch": 0.96,
      "learning_rate": 2.262559558016325e-08,
      "logits/chosen": -2.6588995456695557,
      "logits/rejected": -2.643526077270508,
      "logps/chosen": -107.59385681152344,
      "logps/rejected": -118.29096984863281,
      "loss": 0.6837,
      "rewards/accuracies": 0.609375,
      "rewards/chosen": 0.05322548747062683,
      "rewards/margins": 0.02210088260471821,
      "rewards/rejected": 0.031124606728553772,
      "step": 150
    },
    {
      "epoch": 1.0,
      "step": 156,
      "total_flos": 0.0,
      "train_loss": 0.6878455548714368,
      "train_runtime": 6320.6894,
      "train_samples_per_second": 3.164,
      "train_steps_per_second": 0.025
    }
  ],
  "logging_steps": 10,
  "max_steps": 156,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 100,
  "total_flos": 0.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}