Qin Liu commited on
Commit
2fc6103
·
verified ·
1 Parent(s): 07bb642

Model save

Browse files
Files changed (4) hide show
  1. README.md +62 -0
  2. all_results.json +9 -0
  3. train_results.json +9 -0
  4. trainer_state.json +1767 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Meta-Llama-3-8B-Instruct
3
+ library_name: peft
4
+ license: llama3
5
+ tags:
6
+ - trl
7
+ - dpo
8
+ - generated_from_trainer
9
+ model-index:
10
+ - name: llama3-sudo-dpo-instruct-5epochs-jxkey
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # llama3-sudo-dpo-instruct-5epochs-jxkey
18
+
19
+ This model is a fine-tuned version of [meta-llama/Meta-Llama-3-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct) on an unknown dataset.
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 5e-06
39
+ - train_batch_size: 2
40
+ - eval_batch_size: 8
41
+ - seed: 42
42
+ - distributed_type: multi-GPU
43
+ - num_devices: 8
44
+ - gradient_accumulation_steps: 4
45
+ - total_train_batch_size: 64
46
+ - total_eval_batch_size: 64
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: cosine
49
+ - lr_scheduler_warmup_ratio: 0.1
50
+ - num_epochs: 5
51
+
52
+ ### Training results
53
+
54
+
55
+
56
+ ### Framework versions
57
+
58
+ - PEFT 0.12.0
59
+ - Transformers 4.44.0
60
+ - Pytorch 2.1.2
61
+ - Datasets 2.20.0
62
+ - Tokenizers 0.19.1
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "total_flos": 0.0,
4
+ "train_loss": 0.046525747915877304,
5
+ "train_runtime": 10909.6849,
6
+ "train_samples": 14642,
7
+ "train_samples_per_second": 6.711,
8
+ "train_steps_per_second": 0.105
9
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "total_flos": 0.0,
4
+ "train_loss": 0.046525747915877304,
5
+ "train_runtime": 10909.6849,
6
+ "train_samples": 14642,
7
+ "train_samples_per_second": 6.711,
8
+ "train_steps_per_second": 0.105
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,1767 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 5.0,
5
+ "eval_steps": 1000,
6
+ "global_step": 1145,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.004366812227074236,
13
+ "grad_norm": 0.49208763779092174,
14
+ "learning_rate": 4.347826086956522e-08,
15
+ "logits/chosen": -1.130352258682251,
16
+ "logits/rejected": -0.9433857798576355,
17
+ "logps/chosen": -272.3143005371094,
18
+ "logps/rejected": -290.848388671875,
19
+ "loss": 0.6931,
20
+ "rewards/accuracies": 0.0,
21
+ "rewards/chosen": 0.0,
22
+ "rewards/margins": 0.0,
23
+ "rewards/rejected": 0.0,
24
+ "step": 1
25
+ },
26
+ {
27
+ "epoch": 0.043668122270742356,
28
+ "grad_norm": 0.5373741627747036,
29
+ "learning_rate": 4.347826086956522e-07,
30
+ "logits/chosen": -1.0119584798812866,
31
+ "logits/rejected": -1.054663062095642,
32
+ "logps/chosen": -300.1561279296875,
33
+ "logps/rejected": -261.20513916015625,
34
+ "loss": 0.6928,
35
+ "rewards/accuracies": 0.4722222089767456,
36
+ "rewards/chosen": 0.0015893004601821303,
37
+ "rewards/margins": 0.001330445520579815,
38
+ "rewards/rejected": 0.0002588547649793327,
39
+ "step": 10
40
+ },
41
+ {
42
+ "epoch": 0.08733624454148471,
43
+ "grad_norm": 0.518120805736266,
44
+ "learning_rate": 8.695652173913044e-07,
45
+ "logits/chosen": -1.1330419778823853,
46
+ "logits/rejected": -1.0284273624420166,
47
+ "logps/chosen": -283.1369934082031,
48
+ "logps/rejected": -323.2412109375,
49
+ "loss": 0.6924,
50
+ "rewards/accuracies": 0.5874999761581421,
51
+ "rewards/chosen": -0.0005519447731785476,
52
+ "rewards/margins": 0.0006424236344173551,
53
+ "rewards/rejected": -0.0011943683493882418,
54
+ "step": 20
55
+ },
56
+ {
57
+ "epoch": 0.13100436681222707,
58
+ "grad_norm": 0.6136347394482697,
59
+ "learning_rate": 1.3043478260869566e-06,
60
+ "logits/chosen": -1.0650444030761719,
61
+ "logits/rejected": -1.107439637184143,
62
+ "logps/chosen": -290.29168701171875,
63
+ "logps/rejected": -270.15887451171875,
64
+ "loss": 0.6899,
65
+ "rewards/accuracies": 0.737500011920929,
66
+ "rewards/chosen": -0.0012172337155789137,
67
+ "rewards/margins": 0.0061761606484651566,
68
+ "rewards/rejected": -0.007393394596874714,
69
+ "step": 30
70
+ },
71
+ {
72
+ "epoch": 0.17467248908296942,
73
+ "grad_norm": 0.5452485335548993,
74
+ "learning_rate": 1.7391304347826088e-06,
75
+ "logits/chosen": -1.0981855392456055,
76
+ "logits/rejected": -1.1204359531402588,
77
+ "logps/chosen": -288.19488525390625,
78
+ "logps/rejected": -279.60968017578125,
79
+ "loss": 0.6822,
80
+ "rewards/accuracies": 0.800000011920929,
81
+ "rewards/chosen": -0.01171443797647953,
82
+ "rewards/margins": 0.023151502013206482,
83
+ "rewards/rejected": -0.03486593812704086,
84
+ "step": 40
85
+ },
86
+ {
87
+ "epoch": 0.2183406113537118,
88
+ "grad_norm": 0.5716667587131498,
89
+ "learning_rate": 2.173913043478261e-06,
90
+ "logits/chosen": -1.1297352313995361,
91
+ "logits/rejected": -1.0235356092453003,
92
+ "logps/chosen": -267.85931396484375,
93
+ "logps/rejected": -315.6715393066406,
94
+ "loss": 0.6681,
95
+ "rewards/accuracies": 0.9125000238418579,
96
+ "rewards/chosen": -0.03383685275912285,
97
+ "rewards/margins": 0.050773195922374725,
98
+ "rewards/rejected": -0.08461005985736847,
99
+ "step": 50
100
+ },
101
+ {
102
+ "epoch": 0.26200873362445415,
103
+ "grad_norm": 0.7397729774530086,
104
+ "learning_rate": 2.6086956521739132e-06,
105
+ "logits/chosen": -1.0748536586761475,
106
+ "logits/rejected": -1.096380352973938,
107
+ "logps/chosen": -298.35894775390625,
108
+ "logps/rejected": -278.65167236328125,
109
+ "loss": 0.6372,
110
+ "rewards/accuracies": 0.925000011920929,
111
+ "rewards/chosen": -0.06509696692228317,
112
+ "rewards/margins": 0.13354261219501495,
113
+ "rewards/rejected": -0.19863960146903992,
114
+ "step": 60
115
+ },
116
+ {
117
+ "epoch": 0.3056768558951965,
118
+ "grad_norm": 0.9490199807243085,
119
+ "learning_rate": 3.043478260869566e-06,
120
+ "logits/chosen": -1.1220731735229492,
121
+ "logits/rejected": -1.1358802318572998,
122
+ "logps/chosen": -287.66119384765625,
123
+ "logps/rejected": -300.89215087890625,
124
+ "loss": 0.566,
125
+ "rewards/accuracies": 0.987500011920929,
126
+ "rewards/chosen": -0.05601518228650093,
127
+ "rewards/margins": 0.2802619934082031,
128
+ "rewards/rejected": -0.33627718687057495,
129
+ "step": 70
130
+ },
131
+ {
132
+ "epoch": 0.34934497816593885,
133
+ "grad_norm": 1.1366581667233986,
134
+ "learning_rate": 3.4782608695652175e-06,
135
+ "logits/chosen": -1.1631652116775513,
136
+ "logits/rejected": -1.0547783374786377,
137
+ "logps/chosen": -273.6378173828125,
138
+ "logps/rejected": -378.0707702636719,
139
+ "loss": 0.4241,
140
+ "rewards/accuracies": 1.0,
141
+ "rewards/chosen": -0.0387813039124012,
142
+ "rewards/margins": 0.6967908143997192,
143
+ "rewards/rejected": -0.7355720400810242,
144
+ "step": 80
145
+ },
146
+ {
147
+ "epoch": 0.3930131004366812,
148
+ "grad_norm": 0.699317306849894,
149
+ "learning_rate": 3.91304347826087e-06,
150
+ "logits/chosen": -1.13710355758667,
151
+ "logits/rejected": -1.1224480867385864,
152
+ "logps/chosen": -286.89630126953125,
153
+ "logps/rejected": -425.4947204589844,
154
+ "loss": 0.1966,
155
+ "rewards/accuracies": 1.0,
156
+ "rewards/chosen": 0.010465627536177635,
157
+ "rewards/margins": 1.6259124279022217,
158
+ "rewards/rejected": -1.6154468059539795,
159
+ "step": 90
160
+ },
161
+ {
162
+ "epoch": 0.4366812227074236,
163
+ "grad_norm": 0.20225227555832068,
164
+ "learning_rate": 4.347826086956522e-06,
165
+ "logits/chosen": -1.162099003791809,
166
+ "logits/rejected": -1.0710939168930054,
167
+ "logps/chosen": -254.20108032226562,
168
+ "logps/rejected": -621.758544921875,
169
+ "loss": 0.0422,
170
+ "rewards/accuracies": 1.0,
171
+ "rewards/chosen": 0.06020994111895561,
172
+ "rewards/margins": 3.4915261268615723,
173
+ "rewards/rejected": -3.431316375732422,
174
+ "step": 100
175
+ },
176
+ {
177
+ "epoch": 0.48034934497816595,
178
+ "grad_norm": 0.09010131899726684,
179
+ "learning_rate": 4.782608695652174e-06,
180
+ "logits/chosen": -1.0987495183944702,
181
+ "logits/rejected": -1.049612283706665,
182
+ "logps/chosen": -281.94390869140625,
183
+ "logps/rejected": -799.0904541015625,
184
+ "loss": 0.01,
185
+ "rewards/accuracies": 1.0,
186
+ "rewards/chosen": 0.1645246148109436,
187
+ "rewards/margins": 5.45929479598999,
188
+ "rewards/rejected": -5.29477071762085,
189
+ "step": 110
190
+ },
191
+ {
192
+ "epoch": 0.5240174672489083,
193
+ "grad_norm": 0.030005828505106115,
194
+ "learning_rate": 4.999709285361594e-06,
195
+ "logits/chosen": -1.1363656520843506,
196
+ "logits/rejected": -1.0334757566452026,
197
+ "logps/chosen": -270.98944091796875,
198
+ "logps/rejected": -936.4185791015625,
199
+ "loss": 0.0039,
200
+ "rewards/accuracies": 1.0,
201
+ "rewards/chosen": 0.26095929741859436,
202
+ "rewards/margins": 6.916310787200928,
203
+ "rewards/rejected": -6.655351161956787,
204
+ "step": 120
205
+ },
206
+ {
207
+ "epoch": 0.5676855895196506,
208
+ "grad_norm": 0.019361577478786484,
209
+ "learning_rate": 4.997383973910631e-06,
210
+ "logits/chosen": -1.0980539321899414,
211
+ "logits/rejected": -0.9340838193893433,
212
+ "logps/chosen": -266.50250244140625,
213
+ "logps/rejected": -1020.0845947265625,
214
+ "loss": 0.002,
215
+ "rewards/accuracies": 1.0,
216
+ "rewards/chosen": 0.24470993876457214,
217
+ "rewards/margins": 7.702264308929443,
218
+ "rewards/rejected": -7.457553863525391,
219
+ "step": 130
220
+ },
221
+ {
222
+ "epoch": 0.611353711790393,
223
+ "grad_norm": 0.012282637611773778,
224
+ "learning_rate": 4.9927355140895775e-06,
225
+ "logits/chosen": -1.1392152309417725,
226
+ "logits/rejected": -0.8957275152206421,
227
+ "logps/chosen": -242.6056365966797,
228
+ "logps/rejected": -1173.87158203125,
229
+ "loss": 0.0022,
230
+ "rewards/accuracies": 1.0,
231
+ "rewards/chosen": 0.2704493999481201,
232
+ "rewards/margins": 8.908761978149414,
233
+ "rewards/rejected": -8.638312339782715,
234
+ "step": 140
235
+ },
236
+ {
237
+ "epoch": 0.6550218340611353,
238
+ "grad_norm": 0.04700706955431712,
239
+ "learning_rate": 4.985768230048011e-06,
240
+ "logits/chosen": -1.1246144771575928,
241
+ "logits/rejected": -0.9621224403381348,
242
+ "logps/chosen": -273.8687438964844,
243
+ "logps/rejected": -1090.0859375,
244
+ "loss": 0.0027,
245
+ "rewards/accuracies": 1.0,
246
+ "rewards/chosen": 0.17607423663139343,
247
+ "rewards/margins": 8.613072395324707,
248
+ "rewards/rejected": -8.43699836730957,
249
+ "step": 150
250
+ },
251
+ {
252
+ "epoch": 0.6986899563318777,
253
+ "grad_norm": 0.01943896178498091,
254
+ "learning_rate": 4.976488602981748e-06,
255
+ "logits/chosen": -1.145508885383606,
256
+ "logits/rejected": -0.9405732154846191,
257
+ "logps/chosen": -249.9766387939453,
258
+ "logps/rejected": -1222.8212890625,
259
+ "loss": 0.0009,
260
+ "rewards/accuracies": 1.0,
261
+ "rewards/chosen": 0.33843302726745605,
262
+ "rewards/margins": 9.41877269744873,
263
+ "rewards/rejected": -9.080339431762695,
264
+ "step": 160
265
+ },
266
+ {
267
+ "epoch": 0.74235807860262,
268
+ "grad_norm": 0.019746462175002255,
269
+ "learning_rate": 4.9649052651038255e-06,
270
+ "logits/chosen": -1.0897105932235718,
271
+ "logits/rejected": -0.9191669225692749,
272
+ "logps/chosen": -267.56549072265625,
273
+ "logps/rejected": -1116.246826171875,
274
+ "loss": 0.0007,
275
+ "rewards/accuracies": 1.0,
276
+ "rewards/chosen": 0.28829699754714966,
277
+ "rewards/margins": 8.88953971862793,
278
+ "rewards/rejected": -8.601243019104004,
279
+ "step": 170
280
+ },
281
+ {
282
+ "epoch": 0.7860262008733624,
283
+ "grad_norm": 0.014841671172711,
284
+ "learning_rate": 4.9510289916145295e-06,
285
+ "logits/chosen": -1.0843827724456787,
286
+ "logits/rejected": -0.8706234693527222,
287
+ "logps/chosen": -254.61972045898438,
288
+ "logps/rejected": -1236.5650634765625,
289
+ "loss": 0.0006,
290
+ "rewards/accuracies": 1.0,
291
+ "rewards/chosen": 0.22450891137123108,
292
+ "rewards/margins": 9.739435195922852,
293
+ "rewards/rejected": -9.514925956726074,
294
+ "step": 180
295
+ },
296
+ {
297
+ "epoch": 0.8296943231441049,
298
+ "grad_norm": 0.012319918167579146,
299
+ "learning_rate": 4.934872690677953e-06,
300
+ "logits/chosen": -1.0835936069488525,
301
+ "logits/rejected": -0.8566627502441406,
302
+ "logps/chosen": -267.7489013671875,
303
+ "logps/rejected": -1272.4853515625,
304
+ "loss": 0.0006,
305
+ "rewards/accuracies": 1.0,
306
+ "rewards/chosen": 0.1923988163471222,
307
+ "rewards/margins": 9.936999320983887,
308
+ "rewards/rejected": -9.744600296020508,
309
+ "step": 190
310
+ },
311
+ {
312
+ "epoch": 0.8733624454148472,
313
+ "grad_norm": 0.005430948770812638,
314
+ "learning_rate": 4.9164513914144005e-06,
315
+ "logits/chosen": -1.0162463188171387,
316
+ "logits/rejected": -0.7294031381607056,
317
+ "logps/chosen": -249.8229522705078,
318
+ "logps/rejected": -1294.401611328125,
319
+ "loss": 0.0005,
320
+ "rewards/accuracies": 1.0,
321
+ "rewards/chosen": 0.20589938759803772,
322
+ "rewards/margins": 10.359135627746582,
323
+ "rewards/rejected": -10.153237342834473,
324
+ "step": 200
325
+ },
326
+ {
327
+ "epoch": 0.9170305676855895,
328
+ "grad_norm": 0.009500725575344962,
329
+ "learning_rate": 4.8957822299198045e-06,
330
+ "logits/chosen": -1.1154823303222656,
331
+ "logits/rejected": -0.8606049418449402,
332
+ "logps/chosen": -262.8265686035156,
333
+ "logps/rejected": -1394.8551025390625,
334
+ "loss": 0.0006,
335
+ "rewards/accuracies": 1.0,
336
+ "rewards/chosen": 0.19020363688468933,
337
+ "rewards/margins": 11.131672859191895,
338
+ "rewards/rejected": -10.941469192504883,
339
+ "step": 210
340
+ },
341
+ {
342
+ "epoch": 0.9606986899563319,
343
+ "grad_norm": 0.02542499541949849,
344
+ "learning_rate": 4.872884433325169e-06,
345
+ "logits/chosen": -1.0547072887420654,
346
+ "logits/rejected": -0.8354827165603638,
347
+ "logps/chosen": -282.1262512207031,
348
+ "logps/rejected": -1277.065673828125,
349
+ "loss": 0.0003,
350
+ "rewards/accuracies": 1.0,
351
+ "rewards/chosen": 0.10952471196651459,
352
+ "rewards/margins": 10.300054550170898,
353
+ "rewards/rejected": -10.190529823303223,
354
+ "step": 220
355
+ },
356
+ {
357
+ "epoch": 1.0043668122270741,
358
+ "grad_norm": 0.019502802530769667,
359
+ "learning_rate": 4.847779301910868e-06,
360
+ "logits/chosen": -1.088416576385498,
361
+ "logits/rejected": -0.8197258114814758,
362
+ "logps/chosen": -240.4183807373047,
363
+ "logps/rejected": -1398.536376953125,
364
+ "loss": 0.0004,
365
+ "rewards/accuracies": 1.0,
366
+ "rewards/chosen": 0.14008644223213196,
367
+ "rewards/margins": 11.213750839233398,
368
+ "rewards/rejected": -11.073664665222168,
369
+ "step": 230
370
+ },
371
+ {
372
+ "epoch": 1.0480349344978166,
373
+ "grad_norm": 0.025635188925498188,
374
+ "learning_rate": 4.820490189292415e-06,
375
+ "logits/chosen": -1.019892930984497,
376
+ "logits/rejected": -0.7206239700317383,
377
+ "logps/chosen": -249.9215087890625,
378
+ "logps/rejected": -1368.528564453125,
379
+ "loss": 0.0003,
380
+ "rewards/accuracies": 1.0,
381
+ "rewards/chosen": 0.11964831501245499,
382
+ "rewards/margins": 11.160282135009766,
383
+ "rewards/rejected": -11.040634155273438,
384
+ "step": 240
385
+ },
386
+ {
387
+ "epoch": 1.091703056768559,
388
+ "grad_norm": 0.0019862259316801,
389
+ "learning_rate": 4.791042480696179e-06,
390
+ "logits/chosen": -1.1091744899749756,
391
+ "logits/rejected": -0.8607629537582397,
392
+ "logps/chosen": -280.6003112792969,
393
+ "logps/rejected": -1413.723388671875,
394
+ "loss": 0.0004,
395
+ "rewards/accuracies": 1.0,
396
+ "rewards/chosen": 0.14324969053268433,
397
+ "rewards/margins": 11.427762985229492,
398
+ "rewards/rejected": -11.284514427185059,
399
+ "step": 250
400
+ },
401
+ {
402
+ "epoch": 1.1353711790393013,
403
+ "grad_norm": 0.003461068195787667,
404
+ "learning_rate": 4.759463569345205e-06,
405
+ "logits/chosen": -1.0466573238372803,
406
+ "logits/rejected": -0.8044939041137695,
407
+ "logps/chosen": -261.8208312988281,
408
+ "logps/rejected": -1364.37255859375,
409
+ "loss": 0.0007,
410
+ "rewards/accuracies": 1.0,
411
+ "rewards/chosen": 0.16325196623802185,
412
+ "rewards/margins": 11.088302612304688,
413
+ "rewards/rejected": -10.925050735473633,
414
+ "step": 260
415
+ },
416
+ {
417
+ "epoch": 1.1790393013100438,
418
+ "grad_norm": 0.005672203724021556,
419
+ "learning_rate": 4.725782830977145e-06,
420
+ "logits/chosen": -1.0772409439086914,
421
+ "logits/rejected": -0.7674180865287781,
422
+ "logps/chosen": -244.04861450195312,
423
+ "logps/rejected": -1468.4453125,
424
+ "loss": 0.0003,
425
+ "rewards/accuracies": 1.0,
426
+ "rewards/chosen": 0.1920471489429474,
427
+ "rewards/margins": 11.930693626403809,
428
+ "rewards/rejected": -11.738645553588867,
429
+ "step": 270
430
+ },
431
+ {
432
+ "epoch": 1.222707423580786,
433
+ "grad_norm": 0.00517545244251001,
434
+ "learning_rate": 4.690031596517984e-06,
435
+ "logits/chosen": -1.0256779193878174,
436
+ "logits/rejected": -0.7417846918106079,
437
+ "logps/chosen": -246.1106414794922,
438
+ "logps/rejected": -1458.6060791015625,
439
+ "loss": 0.0003,
440
+ "rewards/accuracies": 1.0,
441
+ "rewards/chosen": 0.17811021208763123,
442
+ "rewards/margins": 11.871198654174805,
443
+ "rewards/rejected": -11.69308853149414,
444
+ "step": 280
445
+ },
446
+ {
447
+ "epoch": 1.2663755458515285,
448
+ "grad_norm": 0.0052311080547555655,
449
+ "learning_rate": 4.652243122936987e-06,
450
+ "logits/chosen": -1.0737969875335693,
451
+ "logits/rejected": -0.8021048307418823,
452
+ "logps/chosen": -262.6719665527344,
453
+ "logps/rejected": -1506.664306640625,
454
+ "loss": 0.0002,
455
+ "rewards/accuracies": 1.0,
456
+ "rewards/chosen": 0.19292786717414856,
457
+ "rewards/margins": 12.288152694702148,
458
+ "rewards/rejected": -12.09522533416748,
459
+ "step": 290
460
+ },
461
+ {
462
+ "epoch": 1.3100436681222707,
463
+ "grad_norm": 0.007845893197308343,
464
+ "learning_rate": 4.612452562309975e-06,
465
+ "logits/chosen": -1.0275319814682007,
466
+ "logits/rejected": -0.6741601228713989,
467
+ "logps/chosen": -233.83407592773438,
468
+ "logps/rejected": -1481.06640625,
469
+ "loss": 0.0001,
470
+ "rewards/accuracies": 1.0,
471
+ "rewards/chosen": 0.20325830578804016,
472
+ "rewards/margins": 12.083272933959961,
473
+ "rewards/rejected": -11.88001537322998,
474
+ "step": 300
475
+ },
476
+ {
477
+ "epoch": 1.3537117903930131,
478
+ "grad_norm": 0.001639011685908484,
479
+ "learning_rate": 4.570696929119717e-06,
480
+ "logits/chosen": -1.0229878425598145,
481
+ "logits/rejected": -0.7648278474807739,
482
+ "logps/chosen": -259.880615234375,
483
+ "logps/rejected": -1439.419189453125,
484
+ "loss": 0.0004,
485
+ "rewards/accuracies": 1.0,
486
+ "rewards/chosen": 0.13743802905082703,
487
+ "rewards/margins": 11.878623962402344,
488
+ "rewards/rejected": -11.741185188293457,
489
+ "step": 310
490
+ },
491
+ {
492
+ "epoch": 1.3973799126637554,
493
+ "grad_norm": 0.002036377670452987,
494
+ "learning_rate": 4.527015065823841e-06,
495
+ "logits/chosen": -1.013264536857605,
496
+ "logits/rejected": -0.7761918306350708,
497
+ "logps/chosen": -272.4781799316406,
498
+ "logps/rejected": -1432.2052001953125,
499
+ "loss": 0.0001,
500
+ "rewards/accuracies": 1.0,
501
+ "rewards/chosen": 0.1656205952167511,
502
+ "rewards/margins": 11.849295616149902,
503
+ "rewards/rejected": -11.683676719665527,
504
+ "step": 320
505
+ },
506
+ {
507
+ "epoch": 1.4410480349344978,
508
+ "grad_norm": 0.00825121579599894,
509
+ "learning_rate": 4.481447606722309e-06,
510
+ "logits/chosen": -1.0379178524017334,
511
+ "logits/rejected": -0.7940360903739929,
512
+ "logps/chosen": -274.08929443359375,
513
+ "logps/rejected": -1472.6800537109375,
514
+ "loss": 0.0002,
515
+ "rewards/accuracies": 1.0,
516
+ "rewards/chosen": 0.19115516543388367,
517
+ "rewards/margins": 12.22935962677002,
518
+ "rewards/rejected": -12.03820514678955,
519
+ "step": 330
520
+ },
521
+ {
522
+ "epoch": 1.48471615720524,
523
+ "grad_norm": 0.006624246041316681,
524
+ "learning_rate": 4.434036940158062e-06,
525
+ "logits/chosen": -1.0243127346038818,
526
+ "logits/rejected": -0.6224126219749451,
527
+ "logps/chosen": -222.64013671875,
528
+ "logps/rejected": -1671.427001953125,
529
+ "loss": 0.0004,
530
+ "rewards/accuracies": 1.0,
531
+ "rewards/chosen": 0.19593994319438934,
532
+ "rewards/margins": 13.684160232543945,
533
+ "rewards/rejected": -13.48822021484375,
534
+ "step": 340
535
+ },
536
+ {
537
+ "epoch": 1.5283842794759825,
538
+ "grad_norm": 0.00969023506429811,
539
+ "learning_rate": 4.384827169085993e-06,
540
+ "logits/chosen": -1.0768173933029175,
541
+ "logits/rejected": -0.9027138948440552,
542
+ "logps/chosen": -304.2896728515625,
543
+ "logps/rejected": -1465.677001953125,
544
+ "loss": 0.0002,
545
+ "rewards/accuracies": 1.0,
546
+ "rewards/chosen": 0.06724857538938522,
547
+ "rewards/margins": 12.110160827636719,
548
+ "rewards/rejected": -12.042912483215332,
549
+ "step": 350
550
+ },
551
+ {
552
+ "epoch": 1.572052401746725,
553
+ "grad_norm": 0.008449136082982325,
554
+ "learning_rate": 4.333864070046938e-06,
555
+ "logits/chosen": -0.9996398687362671,
556
+ "logits/rejected": -0.6732652187347412,
557
+ "logps/chosen": -264.0496520996094,
558
+ "logps/rejected": -1532.588134765625,
559
+ "loss": 0.0001,
560
+ "rewards/accuracies": 1.0,
561
+ "rewards/chosen": 0.09769190102815628,
562
+ "rewards/margins": 12.633995056152344,
563
+ "rewards/rejected": -12.53630256652832,
564
+ "step": 360
565
+ },
566
+ {
567
+ "epoch": 1.6157205240174672,
568
+ "grad_norm": 0.0026627154109082445,
569
+ "learning_rate": 4.28119505058483e-06,
570
+ "logits/chosen": -1.0139929056167603,
571
+ "logits/rejected": -0.7123194336891174,
572
+ "logps/chosen": -275.62481689453125,
573
+ "logps/rejected": -1458.732421875,
574
+ "loss": 0.0002,
575
+ "rewards/accuracies": 1.0,
576
+ "rewards/chosen": 0.08639346808195114,
577
+ "rewards/margins": 12.07685661315918,
578
+ "rewards/rejected": -11.990463256835938,
579
+ "step": 370
580
+ },
581
+ {
582
+ "epoch": 1.6593886462882095,
583
+ "grad_norm": 0.004704630681169572,
584
+ "learning_rate": 4.226869105146658e-06,
585
+ "logits/chosen": -1.0525470972061157,
586
+ "logits/rejected": -0.7499625086784363,
587
+ "logps/chosen": -256.189453125,
588
+ "logps/rejected": -1665.451904296875,
589
+ "loss": 0.0005,
590
+ "rewards/accuracies": 1.0,
591
+ "rewards/chosen": 0.15123505890369415,
592
+ "rewards/margins": 13.766995429992676,
593
+ "rewards/rejected": -13.615760803222656,
594
+ "step": 380
595
+ },
596
+ {
597
+ "epoch": 1.703056768558952,
598
+ "grad_norm": 0.008451726735303476,
599
+ "learning_rate": 4.170936769506222e-06,
600
+ "logits/chosen": -1.0494039058685303,
601
+ "logits/rejected": -0.7635387182235718,
602
+ "logps/chosen": -272.2103576660156,
603
+ "logps/rejected": -1517.40625,
604
+ "loss": 0.0001,
605
+ "rewards/accuracies": 1.0,
606
+ "rewards/chosen": 0.11954255402088165,
607
+ "rewards/margins": 12.597132682800293,
608
+ "rewards/rejected": -12.477590560913086,
609
+ "step": 390
610
+ },
611
+ {
612
+ "epoch": 1.7467248908296944,
613
+ "grad_norm": 0.004007124213956819,
614
+ "learning_rate": 4.1134500737541026e-06,
615
+ "logits/chosen": -1.0403097867965698,
616
+ "logits/rejected": -0.7625882029533386,
617
+ "logps/chosen": -270.08642578125,
618
+ "logps/rejected": -1559.534423828125,
619
+ "loss": 0.0001,
620
+ "rewards/accuracies": 1.0,
621
+ "rewards/chosen": 0.14756450057029724,
622
+ "rewards/margins": 12.893702507019043,
623
+ "rewards/rejected": -12.746137619018555,
624
+ "step": 400
625
+ },
626
+ {
627
+ "epoch": 1.7903930131004366,
628
+ "grad_norm": 0.008071006947580056,
629
+ "learning_rate": 4.054462493897569e-06,
630
+ "logits/chosen": -1.0538218021392822,
631
+ "logits/rejected": -0.6631166934967041,
632
+ "logps/chosen": -230.2433319091797,
633
+ "logps/rejected": -1751.548095703125,
634
+ "loss": 0.0002,
635
+ "rewards/accuracies": 1.0,
636
+ "rewards/chosen": 0.1488143503665924,
637
+ "rewards/margins": 14.362360000610352,
638
+ "rewards/rejected": -14.213546752929688,
639
+ "step": 410
640
+ },
641
+ {
642
+ "epoch": 1.8340611353711789,
643
+ "grad_norm": 0.0018837924584018762,
644
+ "learning_rate": 3.994028902115439e-06,
645
+ "logits/chosen": -1.0209208726882935,
646
+ "logits/rejected": -0.6195182204246521,
647
+ "logps/chosen": -235.20816040039062,
648
+ "logps/rejected": -1623.1568603515625,
649
+ "loss": 0.0002,
650
+ "rewards/accuracies": 1.0,
651
+ "rewards/chosen": 0.12359651178121567,
652
+ "rewards/margins": 13.363885879516602,
653
+ "rewards/rejected": -13.240289688110352,
654
+ "step": 420
655
+ },
656
+ {
657
+ "epoch": 1.8777292576419216,
658
+ "grad_norm": 0.0024505034148971385,
659
+ "learning_rate": 3.932205515714189e-06,
660
+ "logits/chosen": -1.0572965145111084,
661
+ "logits/rejected": -0.764691948890686,
662
+ "logps/chosen": -299.56732177734375,
663
+ "logps/rejected": -1628.056884765625,
664
+ "loss": 0.0001,
665
+ "rewards/accuracies": 1.0,
666
+ "rewards/chosen": -0.033301644027233124,
667
+ "rewards/margins": 13.467303276062012,
668
+ "rewards/rejected": -13.500605583190918,
669
+ "step": 430
670
+ },
671
+ {
672
+ "epoch": 1.9213973799126638,
673
+ "grad_norm": 0.006019680096008564,
674
+ "learning_rate": 3.86904984483277e-06,
675
+ "logits/chosen": -1.0619957447052002,
676
+ "logits/rejected": -0.7923606634140015,
677
+ "logps/chosen": -298.5255432128906,
678
+ "logps/rejected": -1547.7294921875,
679
+ "loss": 0.0001,
680
+ "rewards/accuracies": 1.0,
681
+ "rewards/chosen": 0.007502234075218439,
682
+ "rewards/margins": 12.943402290344238,
683
+ "rewards/rejected": -12.93589973449707,
684
+ "step": 440
685
+ },
686
+ {
687
+ "epoch": 1.965065502183406,
688
+ "grad_norm": 0.0039079828979289005,
689
+ "learning_rate": 3.8046206389447916e-06,
690
+ "logits/chosen": -1.0510538816452026,
691
+ "logits/rejected": -0.7345406413078308,
692
+ "logps/chosen": -275.7720947265625,
693
+ "logps/rejected": -1645.208251953125,
694
+ "loss": 0.0001,
695
+ "rewards/accuracies": 1.0,
696
+ "rewards/chosen": 0.0038163154385983944,
697
+ "rewards/margins": 13.638483047485352,
698
+ "rewards/rejected": -13.634668350219727,
699
+ "step": 450
700
+ },
701
+ {
702
+ "epoch": 2.0087336244541483,
703
+ "grad_norm": 0.0017708249899463602,
704
+ "learning_rate": 3.738977832207839e-06,
705
+ "logits/chosen": -1.0177559852600098,
706
+ "logits/rejected": -0.7104192972183228,
707
+ "logps/chosen": -263.3643798828125,
708
+ "logps/rejected": -1611.4786376953125,
709
+ "loss": 0.0001,
710
+ "rewards/accuracies": 1.0,
711
+ "rewards/chosen": 0.02570909820497036,
712
+ "rewards/margins": 13.336071968078613,
713
+ "rewards/rejected": -13.3103609085083,
714
+ "step": 460
715
+ },
716
+ {
717
+ "epoch": 2.052401746724891,
718
+ "grad_norm": 0.000997282058931975,
719
+ "learning_rate": 3.6721824877107588e-06,
720
+ "logits/chosen": -1.0775004625320435,
721
+ "logits/rejected": -0.8090157508850098,
722
+ "logps/chosen": -305.11627197265625,
723
+ "logps/rejected": -1616.4625244140625,
724
+ "loss": 0.0001,
725
+ "rewards/accuracies": 1.0,
726
+ "rewards/chosen": 0.022619858384132385,
727
+ "rewards/margins": 13.472871780395508,
728
+ "rewards/rejected": -13.450250625610352,
729
+ "step": 470
730
+ },
731
+ {
732
+ "epoch": 2.096069868995633,
733
+ "grad_norm": 0.0066682395247295335,
734
+ "learning_rate": 3.604296740670768e-06,
735
+ "logits/chosen": -1.0786068439483643,
736
+ "logits/rejected": -0.6820305585861206,
737
+ "logps/chosen": -257.0702819824219,
738
+ "logps/rejected": -1786.6517333984375,
739
+ "loss": 0.0001,
740
+ "rewards/accuracies": 1.0,
741
+ "rewards/chosen": 0.09799555689096451,
742
+ "rewards/margins": 14.890355110168457,
743
+ "rewards/rejected": -14.7923583984375,
744
+ "step": 480
745
+ },
746
+ {
747
+ "epoch": 2.1397379912663754,
748
+ "grad_norm": 0.004948689561737302,
749
+ "learning_rate": 3.5353837406332464e-06,
750
+ "logits/chosen": -1.0740149021148682,
751
+ "logits/rejected": -0.7413342595100403,
752
+ "logps/chosen": -247.671630859375,
753
+ "logps/rejected": -1785.2659912109375,
754
+ "loss": 0.0001,
755
+ "rewards/accuracies": 1.0,
756
+ "rewards/chosen": 0.1334933489561081,
757
+ "rewards/margins": 14.81743335723877,
758
+ "rewards/rejected": -14.683941841125488,
759
+ "step": 490
760
+ },
761
+ {
762
+ "epoch": 2.183406113537118,
763
+ "grad_norm": 0.0032297799668046237,
764
+ "learning_rate": 3.4655075927279576e-06,
765
+ "logits/chosen": -1.0715500116348267,
766
+ "logits/rejected": -0.7507299184799194,
767
+ "logps/chosen": -257.148681640625,
768
+ "logps/rejected": -1643.170654296875,
769
+ "loss": 0.0001,
770
+ "rewards/accuracies": 1.0,
771
+ "rewards/chosen": 0.04553316906094551,
772
+ "rewards/margins": 13.757342338562012,
773
+ "rewards/rejected": -13.711812019348145,
774
+ "step": 500
775
+ },
776
+ {
777
+ "epoch": 2.2270742358078603,
778
+ "grad_norm": 0.0018344305263622298,
779
+ "learning_rate": 3.3947332980363552e-06,
780
+ "logits/chosen": -0.9531366229057312,
781
+ "logits/rejected": -0.5800803303718567,
782
+ "logps/chosen": -267.22320556640625,
783
+ "logps/rejected": -1683.7496337890625,
784
+ "loss": 0.0001,
785
+ "rewards/accuracies": 1.0,
786
+ "rewards/chosen": 0.01091139018535614,
787
+ "rewards/margins": 14.13904094696045,
788
+ "rewards/rejected": -14.128130912780762,
789
+ "step": 510
790
+ },
791
+ {
792
+ "epoch": 2.2707423580786026,
793
+ "grad_norm": 0.0008743192695472521,
794
+ "learning_rate": 3.3231266931254546e-06,
795
+ "logits/chosen": -1.0591729879379272,
796
+ "logits/rejected": -0.7667123079299927,
797
+ "logps/chosen": -288.68048095703125,
798
+ "logps/rejected": -1706.4267578125,
799
+ "loss": 0.0001,
800
+ "rewards/accuracies": 1.0,
801
+ "rewards/chosen": 0.048968568444252014,
802
+ "rewards/margins": 14.315594673156738,
803
+ "rewards/rejected": -14.266626358032227,
804
+ "step": 520
805
+ },
806
+ {
807
+ "epoch": 2.314410480349345,
808
+ "grad_norm": 0.0006214259809055745,
809
+ "learning_rate": 3.250754388804495e-06,
810
+ "logits/chosen": -1.0545308589935303,
811
+ "logits/rejected": -0.6721210479736328,
812
+ "logps/chosen": -258.18707275390625,
813
+ "logps/rejected": -1773.6435546875,
814
+ "loss": 0.0001,
815
+ "rewards/accuracies": 1.0,
816
+ "rewards/chosen": 0.016913337633013725,
817
+ "rewards/margins": 14.813578605651855,
818
+ "rewards/rejected": -14.796663284301758,
819
+ "step": 530
820
+ },
821
+ {
822
+ "epoch": 2.3580786026200875,
823
+ "grad_norm": 0.0009986936165298707,
824
+ "learning_rate": 3.1776837081613893e-06,
825
+ "logits/chosen": -0.9897233247756958,
826
+ "logits/rejected": -0.6502217650413513,
827
+ "logps/chosen": -265.50994873046875,
828
+ "logps/rejected": -1707.3509521484375,
829
+ "loss": 0.0001,
830
+ "rewards/accuracies": 1.0,
831
+ "rewards/chosen": 0.05818755552172661,
832
+ "rewards/margins": 14.338269233703613,
833
+ "rewards/rejected": -14.280080795288086,
834
+ "step": 540
835
+ },
836
+ {
837
+ "epoch": 2.4017467248908297,
838
+ "grad_norm": 0.00984202902471183,
839
+ "learning_rate": 3.1039826239365754e-06,
840
+ "logits/chosen": -1.026111364364624,
841
+ "logits/rejected": -0.7162208557128906,
842
+ "logps/chosen": -260.0814514160156,
843
+ "logps/rejected": -1701.0667724609375,
844
+ "loss": 0.0002,
845
+ "rewards/accuracies": 1.0,
846
+ "rewards/chosen": 0.060255538672208786,
847
+ "rewards/margins": 14.235310554504395,
848
+ "rewards/rejected": -14.175054550170898,
849
+ "step": 550
850
+ },
851
+ {
852
+ "epoch": 2.445414847161572,
853
+ "grad_norm": 0.0006428554350249238,
854
+ "learning_rate": 3.0297196952925533e-06,
855
+ "logits/chosen": -1.0254216194152832,
856
+ "logits/rejected": -0.7658672332763672,
857
+ "logps/chosen": -304.80194091796875,
858
+ "logps/rejected": -1645.176025390625,
859
+ "loss": 0.0001,
860
+ "rewards/accuracies": 1.0,
861
+ "rewards/chosen": -0.019381849095225334,
862
+ "rewards/margins": 13.853315353393555,
863
+ "rewards/rejected": -13.872695922851562,
864
+ "step": 560
865
+ },
866
+ {
867
+ "epoch": 2.489082969432314,
868
+ "grad_norm": 0.0006532892925360383,
869
+ "learning_rate": 2.9549640040379043e-06,
870
+ "logits/chosen": -1.0421395301818848,
871
+ "logits/rejected": -0.7338224649429321,
872
+ "logps/chosen": -289.6170349121094,
873
+ "logps/rejected": -1806.13671875,
874
+ "loss": 0.0,
875
+ "rewards/accuracies": 1.0,
876
+ "rewards/chosen": 0.016583764925599098,
877
+ "rewards/margins": 15.094789505004883,
878
+ "rewards/rejected": -15.078204154968262,
879
+ "step": 570
880
+ },
881
+ {
882
+ "epoch": 2.532751091703057,
883
+ "grad_norm": 0.0006677213433183142,
884
+ "learning_rate": 2.8797850903651274e-06,
885
+ "logits/chosen": -1.0362173318862915,
886
+ "logits/rejected": -0.716796875,
887
+ "logps/chosen": -265.3944091796875,
888
+ "logps/rejected": -1734.1741943359375,
889
+ "loss": 0.0001,
890
+ "rewards/accuracies": 1.0,
891
+ "rewards/chosen": 0.07199420034885406,
892
+ "rewards/margins": 14.672683715820312,
893
+ "rewards/rejected": -14.600687980651855,
894
+ "step": 580
895
+ },
896
+ {
897
+ "epoch": 2.576419213973799,
898
+ "grad_norm": 0.0017150618685269877,
899
+ "learning_rate": 2.804252888162079e-06,
900
+ "logits/chosen": -1.1411523818969727,
901
+ "logits/rejected": -0.7371311187744141,
902
+ "logps/chosen": -265.2336120605469,
903
+ "logps/rejected": -1891.736572265625,
904
+ "loss": 0.0,
905
+ "rewards/accuracies": 1.0,
906
+ "rewards/chosen": 0.006541900336742401,
907
+ "rewards/margins": 15.778755187988281,
908
+ "rewards/rejected": -15.772212028503418,
909
+ "step": 590
910
+ },
911
+ {
912
+ "epoch": 2.6200873362445414,
913
+ "grad_norm": 0.0014112958209940734,
914
+ "learning_rate": 2.7284376599571776e-06,
915
+ "logits/chosen": -1.054970622062683,
916
+ "logits/rejected": -0.7131912112236023,
917
+ "logps/chosen": -264.7176818847656,
918
+ "logps/rejected": -1720.33984375,
919
+ "loss": 0.0001,
920
+ "rewards/accuracies": 1.0,
921
+ "rewards/chosen": 0.05957148224115372,
922
+ "rewards/margins": 14.534815788269043,
923
+ "rewards/rejected": -14.475244522094727,
924
+ "step": 600
925
+ },
926
+ {
927
+ "epoch": 2.6637554585152836,
928
+ "grad_norm": 0.0006698680618471773,
929
+ "learning_rate": 2.652409931558898e-06,
930
+ "logits/chosen": -1.053966760635376,
931
+ "logits/rejected": -0.7491308450698853,
932
+ "logps/chosen": -280.0963134765625,
933
+ "logps/rejected": -1777.6761474609375,
934
+ "loss": 0.0,
935
+ "rewards/accuracies": 1.0,
936
+ "rewards/chosen": 0.008257086388766766,
937
+ "rewards/margins": 14.989705085754395,
938
+ "rewards/rejected": -14.98144817352295,
939
+ "step": 610
940
+ },
941
+ {
942
+ "epoch": 2.7074235807860263,
943
+ "grad_norm": 0.00042568362060489706,
944
+ "learning_rate": 2.5762404264503538e-06,
945
+ "logits/chosen": -0.9900799989700317,
946
+ "logits/rejected": -0.6423364281654358,
947
+ "logps/chosen": -269.9592590332031,
948
+ "logps/rejected": -1832.5003662109375,
949
+ "loss": 0.0001,
950
+ "rewards/accuracies": 1.0,
951
+ "rewards/chosen": 0.042371202260255814,
952
+ "rewards/margins": 15.435081481933594,
953
+ "rewards/rejected": -15.392709732055664,
954
+ "step": 620
955
+ },
956
+ {
957
+ "epoch": 2.7510917030567685,
958
+ "grad_norm": 0.006748202311280723,
959
+ "learning_rate": 2.5e-06,
960
+ "logits/chosen": -1.0268778800964355,
961
+ "logits/rejected": -0.7622432708740234,
962
+ "logps/chosen": -285.1914978027344,
963
+ "logps/rejected": -1769.4827880859375,
964
+ "loss": 0.0001,
965
+ "rewards/accuracies": 1.0,
966
+ "rewards/chosen": -0.037803106009960175,
967
+ "rewards/margins": 14.83598804473877,
968
+ "rewards/rejected": -14.873788833618164,
969
+ "step": 630
970
+ },
971
+ {
972
+ "epoch": 2.7947598253275108,
973
+ "grad_norm": 0.0002702166687339878,
974
+ "learning_rate": 2.423759573549647e-06,
975
+ "logits/chosen": -1.0029267072677612,
976
+ "logits/rejected": -0.6316828727722168,
977
+ "logps/chosen": -260.8511962890625,
978
+ "logps/rejected": -1813.537841796875,
979
+ "loss": 0.0,
980
+ "rewards/accuracies": 1.0,
981
+ "rewards/chosen": 0.039741456508636475,
982
+ "rewards/margins": 15.310914993286133,
983
+ "rewards/rejected": -15.271173477172852,
984
+ "step": 640
985
+ },
986
+ {
987
+ "epoch": 2.8384279475982535,
988
+ "grad_norm": 0.0006646355698341602,
989
+ "learning_rate": 2.3475900684411027e-06,
990
+ "logits/chosen": -1.006449818611145,
991
+ "logits/rejected": -0.6019884943962097,
992
+ "logps/chosen": -250.7771453857422,
993
+ "logps/rejected": -1806.491455078125,
994
+ "loss": 0.0,
995
+ "rewards/accuracies": 1.0,
996
+ "rewards/chosen": -0.015752162784337997,
997
+ "rewards/margins": 15.226987838745117,
998
+ "rewards/rejected": -15.2427396774292,
999
+ "step": 650
1000
+ },
1001
+ {
1002
+ "epoch": 2.8820960698689957,
1003
+ "grad_norm": 0.00047399361927177104,
1004
+ "learning_rate": 2.2715623400428228e-06,
1005
+ "logits/chosen": -1.0256198644638062,
1006
+ "logits/rejected": -0.7082680463790894,
1007
+ "logps/chosen": -287.2723083496094,
1008
+ "logps/rejected": -1791.8154296875,
1009
+ "loss": 0.0001,
1010
+ "rewards/accuracies": 1.0,
1011
+ "rewards/chosen": -0.028542857617139816,
1012
+ "rewards/margins": 15.111543655395508,
1013
+ "rewards/rejected": -15.14008617401123,
1014
+ "step": 660
1015
+ },
1016
+ {
1017
+ "epoch": 2.925764192139738,
1018
+ "grad_norm": 0.054595514895715194,
1019
+ "learning_rate": 2.1957471118379213e-06,
1020
+ "logits/chosen": -1.1037609577178955,
1021
+ "logits/rejected": -0.7577636241912842,
1022
+ "logps/chosen": -303.9775390625,
1023
+ "logps/rejected": -1865.5306396484375,
1024
+ "loss": 0.0002,
1025
+ "rewards/accuracies": 1.0,
1026
+ "rewards/chosen": -0.036632128059864044,
1027
+ "rewards/margins": 15.698193550109863,
1028
+ "rewards/rejected": -15.734827041625977,
1029
+ "step": 670
1030
+ },
1031
+ {
1032
+ "epoch": 2.96943231441048,
1033
+ "grad_norm": 0.0032068293343592976,
1034
+ "learning_rate": 2.120214909634873e-06,
1035
+ "logits/chosen": -1.0517921447753906,
1036
+ "logits/rejected": -0.770795464515686,
1037
+ "logps/chosen": -298.4246520996094,
1038
+ "logps/rejected": -1795.1331787109375,
1039
+ "loss": 0.0001,
1040
+ "rewards/accuracies": 1.0,
1041
+ "rewards/chosen": -0.014761227183043957,
1042
+ "rewards/margins": 15.1719970703125,
1043
+ "rewards/rejected": -15.186758041381836,
1044
+ "step": 680
1045
+ },
1046
+ {
1047
+ "epoch": 3.013100436681223,
1048
+ "grad_norm": 0.0011657499181104522,
1049
+ "learning_rate": 2.045035995962097e-06,
1050
+ "logits/chosen": -1.1193125247955322,
1051
+ "logits/rejected": -0.7743921279907227,
1052
+ "logps/chosen": -279.6126403808594,
1053
+ "logps/rejected": -1916.1976318359375,
1054
+ "loss": 0.0001,
1055
+ "rewards/accuracies": 1.0,
1056
+ "rewards/chosen": -0.046258725225925446,
1057
+ "rewards/margins": 16.126333236694336,
1058
+ "rewards/rejected": -16.172592163085938,
1059
+ "step": 690
1060
+ },
1061
+ {
1062
+ "epoch": 3.056768558951965,
1063
+ "grad_norm": 0.005031019960938767,
1064
+ "learning_rate": 1.970280304707447e-06,
1065
+ "logits/chosen": -1.0536220073699951,
1066
+ "logits/rejected": -0.6994236707687378,
1067
+ "logps/chosen": -274.2372131347656,
1068
+ "logps/rejected": -1818.864013671875,
1069
+ "loss": 0.0001,
1070
+ "rewards/accuracies": 1.0,
1071
+ "rewards/chosen": -0.030473018065094948,
1072
+ "rewards/margins": 15.32104206085205,
1073
+ "rewards/rejected": -15.351513862609863,
1074
+ "step": 700
1075
+ },
1076
+ {
1077
+ "epoch": 3.1004366812227073,
1078
+ "grad_norm": 0.0003812225537618957,
1079
+ "learning_rate": 1.8960173760634257e-06,
1080
+ "logits/chosen": -1.1193227767944336,
1081
+ "logits/rejected": -0.8343600034713745,
1082
+ "logps/chosen": -317.20367431640625,
1083
+ "logps/rejected": -1797.347900390625,
1084
+ "loss": 0.0001,
1085
+ "rewards/accuracies": 1.0,
1086
+ "rewards/chosen": -0.015081268735229969,
1087
+ "rewards/margins": 15.233189582824707,
1088
+ "rewards/rejected": -15.248270034790039,
1089
+ "step": 710
1090
+ },
1091
+ {
1092
+ "epoch": 3.14410480349345,
1093
+ "grad_norm": 0.000800716771561942,
1094
+ "learning_rate": 1.8223162918386122e-06,
1095
+ "logits/chosen": -1.0911037921905518,
1096
+ "logits/rejected": -0.7263038158416748,
1097
+ "logps/chosen": -281.10723876953125,
1098
+ "logps/rejected": -1893.2275390625,
1099
+ "loss": 0.0,
1100
+ "rewards/accuracies": 1.0,
1101
+ "rewards/chosen": -0.07366519421339035,
1102
+ "rewards/margins": 15.97996711730957,
1103
+ "rewards/rejected": -16.053630828857422,
1104
+ "step": 720
1105
+ },
1106
+ {
1107
+ "epoch": 3.1877729257641922,
1108
+ "grad_norm": 0.0033435888991239673,
1109
+ "learning_rate": 1.7492456111955052e-06,
1110
+ "logits/chosen": -1.0024917125701904,
1111
+ "logits/rejected": -0.6330265998840332,
1112
+ "logps/chosen": -266.6875915527344,
1113
+ "logps/rejected": -1799.3509521484375,
1114
+ "loss": 0.0,
1115
+ "rewards/accuracies": 1.0,
1116
+ "rewards/chosen": 0.05265193060040474,
1117
+ "rewards/margins": 15.307687759399414,
1118
+ "rewards/rejected": -15.255037307739258,
1119
+ "step": 730
1120
+ },
1121
+ {
1122
+ "epoch": 3.2314410480349345,
1123
+ "grad_norm": 0.003033628662190874,
1124
+ "learning_rate": 1.6768733068745468e-06,
1125
+ "logits/chosen": -1.0493130683898926,
1126
+ "logits/rejected": -0.7165160775184631,
1127
+ "logps/chosen": -277.95294189453125,
1128
+ "logps/rejected": -1927.5621337890625,
1129
+ "loss": 0.0,
1130
+ "rewards/accuracies": 1.0,
1131
+ "rewards/chosen": -0.008759565651416779,
1132
+ "rewards/margins": 16.3278865814209,
1133
+ "rewards/rejected": -16.336645126342773,
1134
+ "step": 740
1135
+ },
1136
+ {
1137
+ "epoch": 3.2751091703056767,
1138
+ "grad_norm": 0.00030739766784929366,
1139
+ "learning_rate": 1.6052667019636462e-06,
1140
+ "logits/chosen": -1.0858592987060547,
1141
+ "logits/rejected": -0.7486026883125305,
1142
+ "logps/chosen": -285.6336669921875,
1143
+ "logps/rejected": -1904.0074462890625,
1144
+ "loss": 0.0001,
1145
+ "rewards/accuracies": 1.0,
1146
+ "rewards/chosen": -0.039479680359363556,
1147
+ "rewards/margins": 15.980633735656738,
1148
+ "rewards/rejected": -16.02011489868164,
1149
+ "step": 750
1150
+ },
1151
+ {
1152
+ "epoch": 3.3187772925764194,
1153
+ "grad_norm": 0.0099847141974217,
1154
+ "learning_rate": 1.5344924072720434e-06,
1155
+ "logits/chosen": -1.0522048473358154,
1156
+ "logits/rejected": -0.7434892058372498,
1157
+ "logps/chosen": -294.0987548828125,
1158
+ "logps/rejected": -1748.886962890625,
1159
+ "loss": 0.0001,
1160
+ "rewards/accuracies": 1.0,
1161
+ "rewards/chosen": -0.06646230816841125,
1162
+ "rewards/margins": 14.771476745605469,
1163
+ "rewards/rejected": -14.837939262390137,
1164
+ "step": 760
1165
+ },
1166
+ {
1167
+ "epoch": 3.3624454148471616,
1168
+ "grad_norm": 0.00046452138658813734,
1169
+ "learning_rate": 1.4646162593667535e-06,
1170
+ "logits/chosen": -1.0780597925186157,
1171
+ "logits/rejected": -0.8004827499389648,
1172
+ "logps/chosen": -309.17120361328125,
1173
+ "logps/rejected": -1823.7366943359375,
1174
+ "loss": 0.0,
1175
+ "rewards/accuracies": 1.0,
1176
+ "rewards/chosen": 0.043299149721860886,
1177
+ "rewards/margins": 15.575587272644043,
1178
+ "rewards/rejected": -15.532289505004883,
1179
+ "step": 770
1180
+ },
1181
+ {
1182
+ "epoch": 3.406113537117904,
1183
+ "grad_norm": 0.0018748318848776301,
1184
+ "learning_rate": 1.3957032593292319e-06,
1185
+ "logits/chosen": -1.042887568473816,
1186
+ "logits/rejected": -0.6965152621269226,
1187
+ "logps/chosen": -282.84088134765625,
1188
+ "logps/rejected": -1889.387451171875,
1189
+ "loss": 0.0001,
1190
+ "rewards/accuracies": 1.0,
1191
+ "rewards/chosen": 0.013044556602835655,
1192
+ "rewards/margins": 15.971423149108887,
1193
+ "rewards/rejected": -15.958378791809082,
1194
+ "step": 780
1195
+ },
1196
+ {
1197
+ "epoch": 3.449781659388646,
1198
+ "grad_norm": 0.0009191121751634733,
1199
+ "learning_rate": 1.3278175122892416e-06,
1200
+ "logits/chosen": -1.029362678527832,
1201
+ "logits/rejected": -0.6999324560165405,
1202
+ "logps/chosen": -269.5941467285156,
1203
+ "logps/rejected": -1812.1949462890625,
1204
+ "loss": 0.0,
1205
+ "rewards/accuracies": 1.0,
1206
+ "rewards/chosen": -0.10188305377960205,
1207
+ "rewards/margins": 15.274584770202637,
1208
+ "rewards/rejected": -15.376466751098633,
1209
+ "step": 790
1210
+ },
1211
+ {
1212
+ "epoch": 3.493449781659389,
1213
+ "grad_norm": 0.003126132491902847,
1214
+ "learning_rate": 1.261022167792161e-06,
1215
+ "logits/chosen": -1.0470654964447021,
1216
+ "logits/rejected": -0.6794711351394653,
1217
+ "logps/chosen": -278.49285888671875,
1218
+ "logps/rejected": -1961.002197265625,
1219
+ "loss": 0.0,
1220
+ "rewards/accuracies": 1.0,
1221
+ "rewards/chosen": 0.009503474459052086,
1222
+ "rewards/margins": 16.650577545166016,
1223
+ "rewards/rejected": -16.641077041625977,
1224
+ "step": 800
1225
+ },
1226
+ {
1227
+ "epoch": 3.537117903930131,
1228
+ "grad_norm": 0.0059086305628955095,
1229
+ "learning_rate": 1.195379361055209e-06,
1230
+ "logits/chosen": -0.9826908111572266,
1231
+ "logits/rejected": -0.6099307537078857,
1232
+ "logps/chosen": -278.61492919921875,
1233
+ "logps/rejected": -1975.528076171875,
1234
+ "loss": 0.0,
1235
+ "rewards/accuracies": 1.0,
1236
+ "rewards/chosen": -0.03355104476213455,
1237
+ "rewards/margins": 16.752765655517578,
1238
+ "rewards/rejected": -16.78631591796875,
1239
+ "step": 810
1240
+ },
1241
+ {
1242
+ "epoch": 3.5807860262008733,
1243
+ "grad_norm": 0.003101839804676522,
1244
+ "learning_rate": 1.1309501551672303e-06,
1245
+ "logits/chosen": -0.957851231098175,
1246
+ "logits/rejected": -0.5697217583656311,
1247
+ "logps/chosen": -272.66534423828125,
1248
+ "logps/rejected": -1837.3662109375,
1249
+ "loss": 0.0001,
1250
+ "rewards/accuracies": 1.0,
1251
+ "rewards/chosen": -0.07275199890136719,
1252
+ "rewards/margins": 15.55827522277832,
1253
+ "rewards/rejected": -15.631025314331055,
1254
+ "step": 820
1255
+ },
1256
+ {
1257
+ "epoch": 3.6244541484716155,
1258
+ "grad_norm": 0.006553990432039902,
1259
+ "learning_rate": 1.0677944842858112e-06,
1260
+ "logits/chosen": -1.0141226053237915,
1261
+ "logits/rejected": -0.6257001757621765,
1262
+ "logps/chosen": -280.70672607421875,
1263
+ "logps/rejected": -1861.9078369140625,
1264
+ "loss": 0.0,
1265
+ "rewards/accuracies": 1.0,
1266
+ "rewards/chosen": -0.05027519538998604,
1267
+ "rewards/margins": 15.878219604492188,
1268
+ "rewards/rejected": -15.928494453430176,
1269
+ "step": 830
1270
+ },
1271
+ {
1272
+ "epoch": 3.668122270742358,
1273
+ "grad_norm": 0.0003416983892558372,
1274
+ "learning_rate": 1.005971097884561e-06,
1275
+ "logits/chosen": -1.0156476497650146,
1276
+ "logits/rejected": -0.668128252029419,
1277
+ "logps/chosen": -302.67767333984375,
1278
+ "logps/rejected": -1935.9398193359375,
1279
+ "loss": 0.0001,
1280
+ "rewards/accuracies": 1.0,
1281
+ "rewards/chosen": -0.08607998490333557,
1282
+ "rewards/margins": 16.394611358642578,
1283
+ "rewards/rejected": -16.480690002441406,
1284
+ "step": 840
1285
+ },
1286
+ {
1287
+ "epoch": 3.7117903930131004,
1288
+ "grad_norm": 0.0015327333742183899,
1289
+ "learning_rate": 9.455375061024319e-07,
1290
+ "logits/chosen": -1.0618703365325928,
1291
+ "logits/rejected": -0.7237562537193298,
1292
+ "logps/chosen": -283.15557861328125,
1293
+ "logps/rejected": -1865.9681396484375,
1294
+ "loss": 0.0001,
1295
+ "rewards/accuracies": 1.0,
1296
+ "rewards/chosen": -0.054165177047252655,
1297
+ "rewards/margins": 15.781875610351562,
1298
+ "rewards/rejected": -15.836041450500488,
1299
+ "step": 850
1300
+ },
1301
+ {
1302
+ "epoch": 3.7554585152838427,
1303
+ "grad_norm": 0.004269773041594158,
1304
+ "learning_rate": 8.86549926245898e-07,
1305
+ "logits/chosen": -1.060253381729126,
1306
+ "logits/rejected": -0.6907247304916382,
1307
+ "logps/chosen": -279.11090087890625,
1308
+ "logps/rejected": -1889.4056396484375,
1309
+ "loss": 0.0001,
1310
+ "rewards/accuracies": 1.0,
1311
+ "rewards/chosen": -0.0650867372751236,
1312
+ "rewards/margins": 15.91883659362793,
1313
+ "rewards/rejected": -15.983922958374023,
1314
+ "step": 860
1315
+ },
1316
+ {
1317
+ "epoch": 3.7991266375545854,
1318
+ "grad_norm": 0.0010427555714299119,
1319
+ "learning_rate": 8.29063230493779e-07,
1320
+ "logits/chosen": -1.0654280185699463,
1321
+ "logits/rejected": -0.6975089907646179,
1322
+ "logps/chosen": -275.31011962890625,
1323
+ "logps/rejected": -1959.466796875,
1324
+ "loss": 0.0001,
1325
+ "rewards/accuracies": 1.0,
1326
+ "rewards/chosen": -0.10659889876842499,
1327
+ "rewards/margins": 16.469921112060547,
1328
+ "rewards/rejected": -16.576520919799805,
1329
+ "step": 870
1330
+ },
1331
+ {
1332
+ "epoch": 3.8427947598253276,
1333
+ "grad_norm": 0.0051211999521691845,
1334
+ "learning_rate": 7.731308948533431e-07,
1335
+ "logits/chosen": -1.0260651111602783,
1336
+ "logits/rejected": -0.7374225854873657,
1337
+ "logps/chosen": -309.5072937011719,
1338
+ "logps/rejected": -1826.8656005859375,
1339
+ "loss": 0.0,
1340
+ "rewards/accuracies": 1.0,
1341
+ "rewards/chosen": -0.08689963072538376,
1342
+ "rewards/margins": 15.4946928024292,
1343
+ "rewards/rejected": -15.58159351348877,
1344
+ "step": 880
1345
+ },
1346
+ {
1347
+ "epoch": 3.88646288209607,
1348
+ "grad_norm": 0.002864032933131896,
1349
+ "learning_rate": 7.188049494151703e-07,
1350
+ "logits/chosen": -0.9869282841682434,
1351
+ "logits/rejected": -0.6170369386672974,
1352
+ "logps/chosen": -270.72308349609375,
1353
+ "logps/rejected": -1809.623046875,
1354
+ "loss": 0.0001,
1355
+ "rewards/accuracies": 1.0,
1356
+ "rewards/chosen": -0.03863278776407242,
1357
+ "rewards/margins": 15.385714530944824,
1358
+ "rewards/rejected": -15.424346923828125,
1359
+ "step": 890
1360
+ },
1361
+ {
1362
+ "epoch": 3.930131004366812,
1363
+ "grad_norm": 0.0005105250492242013,
1364
+ "learning_rate": 6.661359299530626e-07,
1365
+ "logits/chosen": -1.0458041429519653,
1366
+ "logits/rejected": -0.8353503346443176,
1367
+ "logps/chosen": -325.7237548828125,
1368
+ "logps/rejected": -1614.521484375,
1369
+ "loss": 0.0001,
1370
+ "rewards/accuracies": 1.0,
1371
+ "rewards/chosen": -0.15461190044879913,
1372
+ "rewards/margins": 13.781686782836914,
1373
+ "rewards/rejected": -13.936299324035645,
1374
+ "step": 900
1375
+ },
1376
+ {
1377
+ "epoch": 3.9737991266375547,
1378
+ "grad_norm": 0.0019150602563612286,
1379
+ "learning_rate": 6.151728309140071e-07,
1380
+ "logits/chosen": -1.0450153350830078,
1381
+ "logits/rejected": -0.7124982476234436,
1382
+ "logps/chosen": -285.0978698730469,
1383
+ "logps/rejected": -1913.976318359375,
1384
+ "loss": 0.0001,
1385
+ "rewards/accuracies": 1.0,
1386
+ "rewards/chosen": -0.026266271248459816,
1387
+ "rewards/margins": 16.297687530517578,
1388
+ "rewards/rejected": -16.32395362854004,
1389
+ "step": 910
1390
+ },
1391
+ {
1392
+ "epoch": 4.0174672489082965,
1393
+ "grad_norm": 0.003658964581230378,
1394
+ "learning_rate": 5.659630598419391e-07,
1395
+ "logits/chosen": -1.0644596815109253,
1396
+ "logits/rejected": -0.6967732906341553,
1397
+ "logps/chosen": -284.8916320800781,
1398
+ "logps/rejected": -1840.668212890625,
1399
+ "loss": 0.0001,
1400
+ "rewards/accuracies": 1.0,
1401
+ "rewards/chosen": -0.04237327724695206,
1402
+ "rewards/margins": 15.650690078735352,
1403
+ "rewards/rejected": -15.69306468963623,
1404
+ "step": 920
1405
+ },
1406
+ {
1407
+ "epoch": 4.06113537117904,
1408
+ "grad_norm": 0.015359415097265612,
1409
+ "learning_rate": 5.185523932776923e-07,
1410
+ "logits/chosen": -1.0555784702301025,
1411
+ "logits/rejected": -0.7345298528671265,
1412
+ "logps/chosen": -284.2557067871094,
1413
+ "logps/rejected": -1905.572509765625,
1414
+ "loss": 0.0001,
1415
+ "rewards/accuracies": 1.0,
1416
+ "rewards/chosen": -0.04300800338387489,
1417
+ "rewards/margins": 16.220272064208984,
1418
+ "rewards/rejected": -16.26327896118164,
1419
+ "step": 930
1420
+ },
1421
+ {
1422
+ "epoch": 4.104803493449782,
1423
+ "grad_norm": 0.0043225257316477665,
1424
+ "learning_rate": 4.7298493417616024e-07,
1425
+ "logits/chosen": -1.024458646774292,
1426
+ "logits/rejected": -0.7369360327720642,
1427
+ "logps/chosen": -297.3896484375,
1428
+ "logps/rejected": -1880.3125,
1429
+ "loss": 0.0,
1430
+ "rewards/accuracies": 1.0,
1431
+ "rewards/chosen": -0.06699924170970917,
1432
+ "rewards/margins": 15.987092971801758,
1433
+ "rewards/rejected": -16.05409049987793,
1434
+ "step": 940
1435
+ },
1436
+ {
1437
+ "epoch": 4.148471615720524,
1438
+ "grad_norm": 0.0007120346767669291,
1439
+ "learning_rate": 4.293030708802834e-07,
1440
+ "logits/chosen": -1.0639699697494507,
1441
+ "logits/rejected": -0.7229939699172974,
1442
+ "logps/chosen": -270.7474060058594,
1443
+ "logps/rejected": -1974.702392578125,
1444
+ "loss": 0.0,
1445
+ "rewards/accuracies": 1.0,
1446
+ "rewards/chosen": 0.030464956536889076,
1447
+ "rewards/margins": 16.875165939331055,
1448
+ "rewards/rejected": -16.84469985961914,
1449
+ "step": 950
1450
+ },
1451
+ {
1452
+ "epoch": 4.192139737991266,
1453
+ "grad_norm": 0.0004859166750622356,
1454
+ "learning_rate": 3.875474376900254e-07,
1455
+ "logits/chosen": -0.9850033521652222,
1456
+ "logits/rejected": -0.6558709144592285,
1457
+ "logps/chosen": -287.7508544921875,
1458
+ "logps/rejected": -1824.685791015625,
1459
+ "loss": 0.0,
1460
+ "rewards/accuracies": 1.0,
1461
+ "rewards/chosen": -0.10610561072826385,
1462
+ "rewards/margins": 15.522783279418945,
1463
+ "rewards/rejected": -15.628889083862305,
1464
+ "step": 960
1465
+ },
1466
+ {
1467
+ "epoch": 4.235807860262009,
1468
+ "grad_norm": 0.002799456551978829,
1469
+ "learning_rate": 3.4775687706301437e-07,
1470
+ "logits/chosen": -1.1057827472686768,
1471
+ "logits/rejected": -0.7324444055557251,
1472
+ "logps/chosen": -268.9285583496094,
1473
+ "logps/rejected": -2027.9287109375,
1474
+ "loss": 0.0,
1475
+ "rewards/accuracies": 1.0,
1476
+ "rewards/chosen": 0.012243424542248249,
1477
+ "rewards/margins": 17.31172752380371,
1478
+ "rewards/rejected": -17.299488067626953,
1479
+ "step": 970
1480
+ },
1481
+ {
1482
+ "epoch": 4.279475982532751,
1483
+ "grad_norm": 0.0022373731136834133,
1484
+ "learning_rate": 3.0996840348201717e-07,
1485
+ "logits/chosen": -1.1121981143951416,
1486
+ "logits/rejected": -0.8360017538070679,
1487
+ "logps/chosen": -309.40155029296875,
1488
+ "logps/rejected": -1771.3427734375,
1489
+ "loss": 0.0,
1490
+ "rewards/accuracies": 1.0,
1491
+ "rewards/chosen": -0.12068891525268555,
1492
+ "rewards/margins": 14.9802827835083,
1493
+ "rewards/rejected": -15.100972175598145,
1494
+ "step": 980
1495
+ },
1496
+ {
1497
+ "epoch": 4.323144104803493,
1498
+ "grad_norm": 0.00044941977383980663,
1499
+ "learning_rate": 2.742171690228562e-07,
1500
+ "logits/chosen": -1.0458042621612549,
1501
+ "logits/rejected": -0.7098134756088257,
1502
+ "logps/chosen": -300.3705749511719,
1503
+ "logps/rejected": -1846.2822265625,
1504
+ "loss": 0.0001,
1505
+ "rewards/accuracies": 1.0,
1506
+ "rewards/chosen": -0.13983380794525146,
1507
+ "rewards/margins": 15.580610275268555,
1508
+ "rewards/rejected": -15.720443725585938,
1509
+ "step": 990
1510
+ },
1511
+ {
1512
+ "epoch": 4.366812227074236,
1513
+ "grad_norm": 0.002386856484501028,
1514
+ "learning_rate": 2.405364306547955e-07,
1515
+ "logits/chosen": -1.082747459411621,
1516
+ "logits/rejected": -0.8248895406723022,
1517
+ "logps/chosen": -306.8914794921875,
1518
+ "logps/rejected": -1777.709228515625,
1519
+ "loss": 0.0,
1520
+ "rewards/accuracies": 1.0,
1521
+ "rewards/chosen": -0.0018977939616888762,
1522
+ "rewards/margins": 15.146817207336426,
1523
+ "rewards/rejected": -15.148715019226074,
1524
+ "step": 1000
1525
+ },
1526
+ {
1527
+ "epoch": 4.4104803493449785,
1528
+ "grad_norm": 0.0029792452832107918,
1529
+ "learning_rate": 2.0895751930382125e-07,
1530
+ "logits/chosen": -1.0091092586517334,
1531
+ "logits/rejected": -0.6796575784683228,
1532
+ "logps/chosen": -295.8236389160156,
1533
+ "logps/rejected": -1882.607421875,
1534
+ "loss": 0.0,
1535
+ "rewards/accuracies": 1.0,
1536
+ "rewards/chosen": -0.007821875624358654,
1537
+ "rewards/margins": 16.001935958862305,
1538
+ "rewards/rejected": -16.00975799560547,
1539
+ "step": 1010
1540
+ },
1541
+ {
1542
+ "epoch": 4.454148471615721,
1543
+ "grad_norm": 0.0002768349770005794,
1544
+ "learning_rate": 1.7950981070758488e-07,
1545
+ "logits/chosen": -1.0478919744491577,
1546
+ "logits/rejected": -0.7594279050827026,
1547
+ "logps/chosen": -287.52288818359375,
1548
+ "logps/rejected": -1790.5562744140625,
1549
+ "loss": 0.0001,
1550
+ "rewards/accuracies": 1.0,
1551
+ "rewards/chosen": -0.0713457316160202,
1552
+ "rewards/margins": 15.24449634552002,
1553
+ "rewards/rejected": -15.315841674804688,
1554
+ "step": 1020
1555
+ },
1556
+ {
1557
+ "epoch": 4.497816593886463,
1558
+ "grad_norm": 0.005069342731202753,
1559
+ "learning_rate": 1.5222069808913303e-07,
1560
+ "logits/chosen": -1.0174686908721924,
1561
+ "logits/rejected": -0.683228611946106,
1562
+ "logps/chosen": -301.92144775390625,
1563
+ "logps/rejected": -1872.2666015625,
1564
+ "loss": 0.0001,
1565
+ "rewards/accuracies": 1.0,
1566
+ "rewards/chosen": -0.04044654220342636,
1567
+ "rewards/margins": 15.973681449890137,
1568
+ "rewards/rejected": -16.01412582397461,
1569
+ "step": 1030
1570
+ },
1571
+ {
1572
+ "epoch": 4.541484716157205,
1573
+ "grad_norm": 0.0004959573429988225,
1574
+ "learning_rate": 1.271155666748311e-07,
1575
+ "logits/chosen": -1.0908715724945068,
1576
+ "logits/rejected": -0.7660378217697144,
1577
+ "logps/chosen": -306.63043212890625,
1578
+ "logps/rejected": -1871.072509765625,
1579
+ "loss": 0.0001,
1580
+ "rewards/accuracies": 1.0,
1581
+ "rewards/chosen": -0.09966543316841125,
1582
+ "rewards/margins": 15.866052627563477,
1583
+ "rewards/rejected": -15.965716361999512,
1584
+ "step": 1040
1585
+ },
1586
+ {
1587
+ "epoch": 4.585152838427947,
1588
+ "grad_norm": 0.00019462022413360382,
1589
+ "learning_rate": 1.0421777008019663e-07,
1590
+ "logits/chosen": -1.0952246189117432,
1591
+ "logits/rejected": -0.709916353225708,
1592
+ "logps/chosen": -295.17877197265625,
1593
+ "logps/rejected": -2033.247314453125,
1594
+ "loss": 0.0001,
1595
+ "rewards/accuracies": 1.0,
1596
+ "rewards/chosen": -0.07402383536100388,
1597
+ "rewards/margins": 17.205215454101562,
1598
+ "rewards/rejected": -17.279239654541016,
1599
+ "step": 1050
1600
+ },
1601
+ {
1602
+ "epoch": 4.62882096069869,
1603
+ "grad_norm": 0.0036632249464871505,
1604
+ "learning_rate": 8.354860858560021e-08,
1605
+ "logits/chosen": -1.041784405708313,
1606
+ "logits/rejected": -0.6548532247543335,
1607
+ "logps/chosen": -293.025146484375,
1608
+ "logps/rejected": -2029.672607421875,
1609
+ "loss": 0.0001,
1610
+ "rewards/accuracies": 1.0,
1611
+ "rewards/chosen": -0.0596146360039711,
1612
+ "rewards/margins": 17.3281307220459,
1613
+ "rewards/rejected": -17.387744903564453,
1614
+ "step": 1060
1615
+ },
1616
+ {
1617
+ "epoch": 4.672489082969433,
1618
+ "grad_norm": 0.00032895380717983924,
1619
+ "learning_rate": 6.512730932204698e-08,
1620
+ "logits/chosen": -1.0059609413146973,
1621
+ "logits/rejected": -0.7132547497749329,
1622
+ "logps/chosen": -317.78680419921875,
1623
+ "logps/rejected": -1786.2691650390625,
1624
+ "loss": 0.0,
1625
+ "rewards/accuracies": 1.0,
1626
+ "rewards/chosen": -0.03703427314758301,
1627
+ "rewards/margins": 15.275805473327637,
1628
+ "rewards/rejected": -15.312841415405273,
1629
+ "step": 1070
1630
+ },
1631
+ {
1632
+ "epoch": 4.716157205240175,
1633
+ "grad_norm": 0.0031606785955010904,
1634
+ "learning_rate": 4.897100838547081e-08,
1635
+ "logits/chosen": -1.0450844764709473,
1636
+ "logits/rejected": -0.7194000482559204,
1637
+ "logps/chosen": -301.86541748046875,
1638
+ "logps/rejected": -1955.65625,
1639
+ "loss": 0.0001,
1640
+ "rewards/accuracies": 1.0,
1641
+ "rewards/chosen": -0.05521082133054733,
1642
+ "rewards/margins": 16.574329376220703,
1643
+ "rewards/rejected": -16.62953758239746,
1644
+ "step": 1080
1645
+ },
1646
+ {
1647
+ "epoch": 4.759825327510917,
1648
+ "grad_norm": 0.000347629698955204,
1649
+ "learning_rate": 3.5094734896174985e-08,
1650
+ "logits/chosen": -1.080841302871704,
1651
+ "logits/rejected": -0.6831713914871216,
1652
+ "logps/chosen": -268.0194091796875,
1653
+ "logps/rejected": -1998.539306640625,
1654
+ "loss": 0.0002,
1655
+ "rewards/accuracies": 1.0,
1656
+ "rewards/chosen": -0.04163534566760063,
1657
+ "rewards/margins": 16.94226837158203,
1658
+ "rewards/rejected": -16.98390007019043,
1659
+ "step": 1090
1660
+ },
1661
+ {
1662
+ "epoch": 4.8034934497816595,
1663
+ "grad_norm": 0.004295417718120041,
1664
+ "learning_rate": 2.351139701825267e-08,
1665
+ "logits/chosen": -0.9642395973205566,
1666
+ "logits/rejected": -0.6139574646949768,
1667
+ "logps/chosen": -276.0238037109375,
1668
+ "logps/rejected": -1780.542236328125,
1669
+ "loss": 0.0001,
1670
+ "rewards/accuracies": 1.0,
1671
+ "rewards/chosen": -0.0755363255739212,
1672
+ "rewards/margins": 15.09814167022705,
1673
+ "rewards/rejected": -15.173677444458008,
1674
+ "step": 1100
1675
+ },
1676
+ {
1677
+ "epoch": 4.847161572052402,
1678
+ "grad_norm": 0.00037246559243259353,
1679
+ "learning_rate": 1.4231769951990326e-08,
1680
+ "logits/chosen": -1.033734679222107,
1681
+ "logits/rejected": -0.6827409267425537,
1682
+ "logps/chosen": -287.67584228515625,
1683
+ "logps/rejected": -1885.5576171875,
1684
+ "loss": 0.0,
1685
+ "rewards/accuracies": 1.0,
1686
+ "rewards/chosen": -0.08407995104789734,
1687
+ "rewards/margins": 16.050437927246094,
1688
+ "rewards/rejected": -16.134517669677734,
1689
+ "step": 1110
1690
+ },
1691
+ {
1692
+ "epoch": 4.890829694323144,
1693
+ "grad_norm": 0.0003852961387970997,
1694
+ "learning_rate": 7.264485910423447e-09,
1695
+ "logits/chosen": -0.9720686078071594,
1696
+ "logits/rejected": -0.5935501456260681,
1697
+ "logps/chosen": -265.44622802734375,
1698
+ "logps/rejected": -1915.1038818359375,
1699
+ "loss": 0.0,
1700
+ "rewards/accuracies": 1.0,
1701
+ "rewards/chosen": -0.024737151339650154,
1702
+ "rewards/margins": 16.284820556640625,
1703
+ "rewards/rejected": -16.309558868408203,
1704
+ "step": 1120
1705
+ },
1706
+ {
1707
+ "epoch": 4.934497816593886,
1708
+ "grad_norm": 0.015272373299215733,
1709
+ "learning_rate": 2.6160260893692833e-09,
1710
+ "logits/chosen": -1.0101922750473022,
1711
+ "logits/rejected": -0.6879107356071472,
1712
+ "logps/chosen": -289.36138916015625,
1713
+ "logps/rejected": -1796.4605712890625,
1714
+ "loss": 0.0001,
1715
+ "rewards/accuracies": 1.0,
1716
+ "rewards/chosen": 0.028253447264432907,
1717
+ "rewards/margins": 15.371627807617188,
1718
+ "rewards/rejected": -15.34337329864502,
1719
+ "step": 1130
1720
+ },
1721
+ {
1722
+ "epoch": 4.978165938864628,
1723
+ "grad_norm": 0.0002424398537573296,
1724
+ "learning_rate": 2.9071463840540936e-10,
1725
+ "logits/chosen": -1.0956311225891113,
1726
+ "logits/rejected": -0.8314415216445923,
1727
+ "logps/chosen": -341.66766357421875,
1728
+ "logps/rejected": -1909.0863037109375,
1729
+ "loss": 0.0,
1730
+ "rewards/accuracies": 1.0,
1731
+ "rewards/chosen": -0.11481740325689316,
1732
+ "rewards/margins": 16.255878448486328,
1733
+ "rewards/rejected": -16.370695114135742,
1734
+ "step": 1140
1735
+ },
1736
+ {
1737
+ "epoch": 5.0,
1738
+ "step": 1145,
1739
+ "total_flos": 0.0,
1740
+ "train_loss": 0.046525747915877304,
1741
+ "train_runtime": 10909.6849,
1742
+ "train_samples_per_second": 6.711,
1743
+ "train_steps_per_second": 0.105
1744
+ }
1745
+ ],
1746
+ "logging_steps": 10,
1747
+ "max_steps": 1145,
1748
+ "num_input_tokens_seen": 0,
1749
+ "num_train_epochs": 5,
1750
+ "save_steps": 100,
1751
+ "stateful_callbacks": {
1752
+ "TrainerControl": {
1753
+ "args": {
1754
+ "should_epoch_stop": false,
1755
+ "should_evaluate": false,
1756
+ "should_log": false,
1757
+ "should_save": true,
1758
+ "should_training_stop": true
1759
+ },
1760
+ "attributes": {}
1761
+ }
1762
+ },
1763
+ "total_flos": 0.0,
1764
+ "train_batch_size": 2,
1765
+ "trial_name": null,
1766
+ "trial_params": null
1767
+ }