File size: 8,938 Bytes
e20d002
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.62762188911438,
      "logits/rejected": -2.586808681488037,
      "logps/chosen": -233.10275268554688,
      "logps/rejected": -220.01724243164062,
      "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.640223741531372,
      "logits/rejected": -2.6421992778778076,
      "logps/chosen": -172.71710205078125,
      "logps/rejected": -174.91177368164062,
      "loss": 0.6932,
      "rewards/accuracies": 0.4270833432674408,
      "rewards/chosen": -0.0005670329555869102,
      "rewards/margins": -0.00034490120015107095,
      "rewards/rejected": -0.00022213184274733067,
      "step": 10
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.989935734988098e-06,
      "logits/chosen": -2.6506519317626953,
      "logits/rejected": -2.6606290340423584,
      "logps/chosen": -163.25784301757812,
      "logps/rejected": -166.31251525878906,
      "loss": 0.6923,
      "rewards/accuracies": 0.546875,
      "rewards/chosen": -0.005096795037388802,
      "rewards/margins": 0.002602009102702141,
      "rewards/rejected": -0.007698804140090942,
      "step": 20
    },
    {
      "epoch": 0.19,
      "learning_rate": 4.8776412907378845e-06,
      "logits/chosen": -2.6732003688812256,
      "logits/rejected": -2.6337249279022217,
      "logps/chosen": -179.1455535888672,
      "logps/rejected": -177.36630249023438,
      "loss": 0.6928,
      "rewards/accuracies": 0.5218750238418579,
      "rewards/chosen": -0.07040824741125107,
      "rewards/margins": 0.0008022676920518279,
      "rewards/rejected": -0.07121052592992783,
      "step": 30
    },
    {
      "epoch": 0.26,
      "learning_rate": 4.646121984004666e-06,
      "logits/chosen": -2.592510223388672,
      "logits/rejected": -2.569596767425537,
      "logps/chosen": -176.0773162841797,
      "logps/rejected": -177.98220825195312,
      "loss": 0.6939,
      "rewards/accuracies": 0.5249999761581421,
      "rewards/chosen": -0.06681038439273834,
      "rewards/margins": -0.002174162771552801,
      "rewards/rejected": -0.0646362155675888,
      "step": 40
    },
    {
      "epoch": 0.32,
      "learning_rate": 4.3069871595684795e-06,
      "logits/chosen": -2.6304595470428467,
      "logits/rejected": -2.6132376194000244,
      "logps/chosen": -170.5052947998047,
      "logps/rejected": -170.62728881835938,
      "loss": 0.6925,
      "rewards/accuracies": 0.5218750238418579,
      "rewards/chosen": -0.03540392965078354,
      "rewards/margins": 0.004562810063362122,
      "rewards/rejected": -0.03996673971414566,
      "step": 50
    },
    {
      "epoch": 0.38,
      "learning_rate": 3.8772424536302565e-06,
      "logits/chosen": -2.5987067222595215,
      "logits/rejected": -2.573859691619873,
      "logps/chosen": -180.9357452392578,
      "logps/rejected": -177.1747589111328,
      "loss": 0.6923,
      "rewards/accuracies": 0.48750001192092896,
      "rewards/chosen": -0.09346936643123627,
      "rewards/margins": 0.0016549427527934313,
      "rewards/rejected": -0.09512430429458618,
      "step": 60
    },
    {
      "epoch": 0.45,
      "learning_rate": 3.3784370602033572e-06,
      "logits/chosen": -2.5544121265411377,
      "logits/rejected": -2.570075750350952,
      "logps/chosen": -172.70663452148438,
      "logps/rejected": -173.13766479492188,
      "loss": 0.692,
      "rewards/accuracies": 0.49687498807907104,
      "rewards/chosen": -0.15676796436309814,
      "rewards/margins": -0.002442942699417472,
      "rewards/rejected": -0.15432503819465637,
      "step": 70
    },
    {
      "epoch": 0.51,
      "learning_rate": 2.835583164544139e-06,
      "logits/chosen": -2.604128122329712,
      "logits/rejected": -2.5767688751220703,
      "logps/chosen": -184.4633331298828,
      "logps/rejected": -183.540283203125,
      "loss": 0.6936,
      "rewards/accuracies": 0.5562499761581421,
      "rewards/chosen": -0.18969281017780304,
      "rewards/margins": -0.0006604200461879373,
      "rewards/rejected": -0.18903236091136932,
      "step": 80
    },
    {
      "epoch": 0.58,
      "learning_rate": 2.2759017277414165e-06,
      "logits/chosen": -2.661609172821045,
      "logits/rejected": -2.5941507816314697,
      "logps/chosen": -191.8031463623047,
      "logps/rejected": -189.98135375976562,
      "loss": 0.6914,
      "rewards/accuracies": 0.565625011920929,
      "rewards/chosen": -0.08525966852903366,
      "rewards/margins": 0.013879667036235332,
      "rewards/rejected": -0.09913934022188187,
      "step": 90
    },
    {
      "epoch": 0.64,
      "learning_rate": 1.7274575140626318e-06,
      "logits/chosen": -2.6250574588775635,
      "logits/rejected": -2.6189353466033936,
      "logps/chosen": -178.07730102539062,
      "logps/rejected": -179.85862731933594,
      "loss": 0.6902,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.038252077996730804,
      "rewards/margins": 0.017853444442152977,
      "rewards/rejected": -0.05610552430152893,
      "step": 100
    },
    {
      "epoch": 0.64,
      "eval_logits/chosen": -2.555354595184326,
      "eval_logits/rejected": -2.4580423831939697,
      "eval_logps/chosen": -310.5243835449219,
      "eval_logps/rejected": -302.46649169921875,
      "eval_loss": 0.6929731369018555,
      "eval_rewards/accuracies": 0.5139999985694885,
      "eval_rewards/chosen": -0.12465862929821014,
      "eval_rewards/margins": 0.00494543369859457,
      "eval_rewards/rejected": -0.12960407137870789,
      "eval_runtime": 383.981,
      "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.651780128479004,
      "logits/rejected": -2.581507921218872,
      "logps/chosen": -194.07461547851562,
      "logps/rejected": -183.43362426757812,
      "loss": 0.6898,
      "rewards/accuracies": 0.4781250059604645,
      "rewards/chosen": -0.11336056143045425,
      "rewards/margins": 0.0036790375597774982,
      "rewards/rejected": -0.11703959852457047,
      "step": 110
    },
    {
      "epoch": 0.77,
      "learning_rate": 7.723433775328385e-07,
      "logits/chosen": -2.6290206909179688,
      "logits/rejected": -2.6049842834472656,
      "logps/chosen": -186.02259826660156,
      "logps/rejected": -193.3793487548828,
      "loss": 0.6902,
      "rewards/accuracies": 0.534375011920929,
      "rewards/chosen": -0.11839202791452408,
      "rewards/margins": 0.004132995847612619,
      "rewards/rejected": -0.12252502143383026,
      "step": 120
    },
    {
      "epoch": 0.83,
      "learning_rate": 4.1356686569674344e-07,
      "logits/chosen": -2.5702383518218994,
      "logits/rejected": -2.5472493171691895,
      "logps/chosen": -169.70834350585938,
      "logps/rejected": -174.14761352539062,
      "loss": 0.6891,
      "rewards/accuracies": 0.5562499761581421,
      "rewards/chosen": -0.09281644970178604,
      "rewards/margins": 0.012840485200285912,
      "rewards/rejected": -0.1056569367647171,
      "step": 130
    },
    {
      "epoch": 0.9,
      "learning_rate": 1.59412823400657e-07,
      "logits/chosen": -2.5276107788085938,
      "logits/rejected": -2.5228161811828613,
      "logps/chosen": -174.78802490234375,
      "logps/rejected": -176.2909698486328,
      "loss": 0.6878,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.05636407807469368,
      "rewards/margins": 0.011297663673758507,
      "rewards/rejected": -0.06766173988580704,
      "step": 140
    },
    {
      "epoch": 0.96,
      "learning_rate": 2.262559558016325e-08,
      "logits/chosen": -2.568333864212036,
      "logits/rejected": -2.586660623550415,
      "logps/chosen": -166.882080078125,
      "logps/rejected": -171.17140197753906,
      "loss": 0.6892,
      "rewards/accuracies": 0.565625011920929,
      "rewards/chosen": -0.061885736882686615,
      "rewards/margins": 0.011388251557946205,
      "rewards/rejected": -0.07327400147914886,
      "step": 150
    },
    {
      "epoch": 1.0,
      "step": 156,
      "total_flos": 0.0,
      "train_loss": 0.6914212252849188,
      "train_runtime": 7065.3773,
      "train_samples_per_second": 2.831,
      "train_steps_per_second": 0.022
    }
  ],
  "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
}