Model save
Browse files- README.md +1 -1
- all_results.json +5 -5
- model-00001-of-00004.safetensors +1 -1
- model-00002-of-00004.safetensors +1 -1
- model-00003-of-00004.safetensors +1 -1
- model-00004-of-00004.safetensors +1 -1
- train_results.json +5 -5
- trainer_state.json +279 -234
- training_args.bin +1 -1
README.md
CHANGED
|
@@ -26,7 +26,7 @@ print(output["generated_text"])
|
|
| 26 |
|
| 27 |
## Training procedure
|
| 28 |
|
| 29 |
-
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/ggbetz/argunauts-training/runs/
|
| 30 |
|
| 31 |
|
| 32 |
This model was trained with DPO, a method introduced in [Direct Preference Optimization: Your Language Model is Secretly a Reward Model](https://huggingface.co/papers/2305.18290).
|
|
|
|
| 26 |
|
| 27 |
## Training procedure
|
| 28 |
|
| 29 |
+
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/ggbetz/argunauts-training/runs/8afyfsbz)
|
| 30 |
|
| 31 |
|
| 32 |
This model was trained with DPO, a method introduced in [Direct Preference Optimization: Your Language Model is Secretly a Reward Model](https://huggingface.co/papers/2305.18290).
|
all_results.json
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
{
|
| 2 |
-
"epoch": 1.
|
| 3 |
"total_flos": 0.0,
|
| 4 |
-
"train_loss": 0.
|
| 5 |
-
"train_runtime":
|
| 6 |
-
"train_samples":
|
| 7 |
-
"train_samples_per_second": 1.
|
| 8 |
"train_steps_per_second": 0.056
|
| 9 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"epoch": 1.9820224719101125,
|
| 3 |
"total_flos": 0.0,
|
| 4 |
+
"train_loss": 0.340618884563446,
|
| 5 |
+
"train_runtime": 1955.8801,
|
| 6 |
+
"train_samples": 1779,
|
| 7 |
+
"train_samples_per_second": 1.819,
|
| 8 |
"train_steps_per_second": 0.056
|
| 9 |
}
|
model-00001-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4976698672
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:872f675707c09411dbb21f0c56ce21a75a907a172ce62eab35fbc9ac386d87d3
|
| 3 |
size 4976698672
|
model-00002-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4999802720
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fb9bba22ea8a58699ae221ee9488c2cffe45b95c03ff0f7d585b957bceadb231
|
| 3 |
size 4999802720
|
model-00003-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4915916176
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e8d81c8cb4860860e1e8ebc9f7920a5497b153e08513ac9b6af2d4656427d313
|
| 3 |
size 4915916176
|
model-00004-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1168138808
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bacab750cca6f653341a857fdbe0f2616bae9c79632f48a574760daf38647180
|
| 3 |
size 1168138808
|
train_results.json
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
{
|
| 2 |
-
"epoch": 1.
|
| 3 |
"total_flos": 0.0,
|
| 4 |
-
"train_loss": 0.
|
| 5 |
-
"train_runtime":
|
| 6 |
-
"train_samples":
|
| 7 |
-
"train_samples_per_second": 1.
|
| 8 |
"train_steps_per_second": 0.056
|
| 9 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"epoch": 1.9820224719101125,
|
| 3 |
"total_flos": 0.0,
|
| 4 |
+
"train_loss": 0.340618884563446,
|
| 5 |
+
"train_runtime": 1955.8801,
|
| 6 |
+
"train_samples": 1779,
|
| 7 |
+
"train_samples_per_second": 1.819,
|
| 8 |
"train_steps_per_second": 0.056
|
| 9 |
}
|
trainer_state.json
CHANGED
|
@@ -1,310 +1,355 @@
|
|
| 1 |
{
|
| 2 |
"best_metric": null,
|
| 3 |
"best_model_checkpoint": null,
|
| 4 |
-
"epoch": 1.
|
| 5 |
"eval_steps": 500,
|
| 6 |
-
"global_step":
|
| 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.
|
| 13 |
-
"grad_norm":
|
| 14 |
-
"learning_rate":
|
| 15 |
-
"logits/chosen": -1.
|
| 16 |
-
"logits/rejected": -1.
|
| 17 |
-
"logps/chosen": -
|
| 18 |
-
"logps/rejected": -
|
| 19 |
-
"loss": 0.
|
| 20 |
-
"rewards/accuracies": 0.
|
| 21 |
-
"rewards/chosen": -
|
| 22 |
-
"rewards/margins": 2.
|
| 23 |
-
"rewards/rejected": -4.
|
| 24 |
"step": 5
|
| 25 |
},
|
| 26 |
{
|
| 27 |
-
"epoch": 0.
|
| 28 |
-
"grad_norm":
|
| 29 |
-
"learning_rate":
|
| 30 |
-
"logits/chosen": -1.
|
| 31 |
-
"logits/rejected": -1.
|
| 32 |
-
"logps/chosen": -
|
| 33 |
-
"logps/rejected": -
|
| 34 |
-
"loss": 0.
|
| 35 |
-
"rewards/accuracies": 0.
|
| 36 |
-
"rewards/chosen": -
|
| 37 |
-
"rewards/margins": 2.
|
| 38 |
-
"rewards/rejected": -
|
| 39 |
"step": 10
|
| 40 |
},
|
| 41 |
{
|
| 42 |
-
"epoch": 0.
|
| 43 |
-
"grad_norm":
|
| 44 |
-
"learning_rate": 1.
|
| 45 |
-
"logits/chosen": -1.
|
| 46 |
-
"logits/rejected": -1.
|
| 47 |
-
"logps/chosen": -
|
| 48 |
-
"logps/rejected": -
|
| 49 |
-
"loss": 0.
|
| 50 |
-
"rewards/accuracies": 0.
|
| 51 |
-
"rewards/chosen": -
|
| 52 |
-
"rewards/margins": 2.
|
| 53 |
-
"rewards/rejected": -
|
| 54 |
"step": 15
|
| 55 |
},
|
| 56 |
{
|
| 57 |
-
"epoch": 0.
|
| 58 |
-
"grad_norm":
|
| 59 |
-
"learning_rate":
|
| 60 |
-
"logits/chosen": -1.
|
| 61 |
-
"logits/rejected": -1.
|
| 62 |
-
"logps/chosen": -
|
| 63 |
-
"logps/rejected": -
|
| 64 |
-
"loss": 0.
|
| 65 |
-
"rewards/accuracies": 0.
|
| 66 |
-
"rewards/chosen": -
|
| 67 |
-
"rewards/margins": 2.
|
| 68 |
-
"rewards/rejected": -
|
| 69 |
"step": 20
|
| 70 |
},
|
| 71 |
{
|
| 72 |
-
"epoch": 0.
|
| 73 |
-
"grad_norm":
|
| 74 |
-
"learning_rate": 2.
|
| 75 |
-
"logits/chosen": -1.
|
| 76 |
-
"logits/rejected": -1.
|
| 77 |
-
"logps/chosen": -
|
| 78 |
-
"logps/rejected": -
|
| 79 |
-
"loss": 0.
|
| 80 |
-
"rewards/accuracies": 0.
|
| 81 |
-
"rewards/chosen": -
|
| 82 |
-
"rewards/margins":
|
| 83 |
-
"rewards/rejected": -
|
| 84 |
"step": 25
|
| 85 |
},
|
| 86 |
{
|
| 87 |
-
"epoch": 0.
|
| 88 |
-
"grad_norm":
|
| 89 |
-
"learning_rate":
|
| 90 |
-
"logits/chosen": -1.
|
| 91 |
-
"logits/rejected": -1.
|
| 92 |
-
"logps/chosen": -
|
| 93 |
-
"logps/rejected": -
|
| 94 |
-
"loss": 0.
|
| 95 |
-
"rewards/accuracies": 0.
|
| 96 |
-
"rewards/chosen": -1.
|
| 97 |
-
"rewards/margins": 2.
|
| 98 |
-
"rewards/rejected": -
|
| 99 |
"step": 30
|
| 100 |
},
|
| 101 |
{
|
| 102 |
-
"epoch": 0.
|
| 103 |
-
"grad_norm":
|
| 104 |
-
"learning_rate": 3.
|
| 105 |
-
"logits/chosen": -1.
|
| 106 |
-
"logits/rejected": -1.
|
| 107 |
-
"logps/chosen": -
|
| 108 |
-
"logps/rejected": -
|
| 109 |
-
"loss": 0.
|
| 110 |
-
"rewards/accuracies": 0.
|
| 111 |
-
"rewards/chosen": -
|
| 112 |
-
"rewards/margins": 2.
|
| 113 |
-
"rewards/rejected": -4.
|
| 114 |
"step": 35
|
| 115 |
},
|
| 116 |
{
|
| 117 |
-
"epoch": 0.
|
| 118 |
-
"grad_norm":
|
| 119 |
-
"learning_rate":
|
| 120 |
-
"logits/chosen": -1.
|
| 121 |
-
"logits/rejected": -1.
|
| 122 |
-
"logps/chosen": -
|
| 123 |
-
"logps/rejected": -
|
| 124 |
-
"loss": 0.
|
| 125 |
"rewards/accuracies": 0.8500000238418579,
|
| 126 |
-
"rewards/chosen": -1.
|
| 127 |
-
"rewards/margins": 2.
|
| 128 |
-
"rewards/rejected": -4.
|
| 129 |
"step": 40
|
| 130 |
},
|
| 131 |
{
|
| 132 |
-
"epoch": 0.
|
| 133 |
-
"grad_norm":
|
| 134 |
-
"learning_rate": 4.
|
| 135 |
-
"logits/chosen": -1.
|
| 136 |
-
"logits/rejected": -1.
|
| 137 |
-
"logps/chosen": -
|
| 138 |
-
"logps/rejected": -
|
| 139 |
-
"loss": 0.
|
| 140 |
-
"rewards/accuracies": 0.
|
| 141 |
-
"rewards/chosen": -
|
| 142 |
-
"rewards/margins": 2.
|
| 143 |
-
"rewards/rejected": -4.
|
| 144 |
"step": 45
|
| 145 |
},
|
| 146 |
{
|
| 147 |
-
"epoch":
|
| 148 |
-
"grad_norm":
|
| 149 |
-
"learning_rate": 4.
|
| 150 |
-
"logits/chosen": -1.
|
| 151 |
-
"logits/rejected": -1.
|
| 152 |
-
"logps/chosen": -
|
| 153 |
-
"logps/rejected": -
|
| 154 |
-
"loss": 0.
|
| 155 |
-
"rewards/accuracies": 0.
|
| 156 |
-
"rewards/chosen": -
|
| 157 |
-
"rewards/margins": 2.
|
| 158 |
-
"rewards/rejected": -4.
|
| 159 |
"step": 50
|
| 160 |
},
|
| 161 |
{
|
| 162 |
-
"epoch":
|
| 163 |
-
"grad_norm":
|
| 164 |
-
"learning_rate":
|
| 165 |
-
"logits/chosen": -1.
|
| 166 |
-
"logits/rejected": -1.
|
| 167 |
-
"logps/chosen": -
|
| 168 |
-
"logps/rejected": -
|
| 169 |
-
"loss": 0.
|
| 170 |
-
"rewards/accuracies": 0.
|
| 171 |
-
"rewards/chosen": -1.
|
| 172 |
-
"rewards/margins": 2.
|
| 173 |
-
"rewards/rejected": -3.
|
| 174 |
"step": 55
|
| 175 |
},
|
| 176 |
{
|
| 177 |
-
"epoch": 1.
|
| 178 |
-
"grad_norm":
|
| 179 |
-
"learning_rate":
|
| 180 |
-
"logits/chosen": -1.
|
| 181 |
-
"logits/rejected": -1.
|
| 182 |
-
"logps/chosen": -
|
| 183 |
-
"logps/rejected": -
|
| 184 |
-
"loss": 0.
|
| 185 |
-
"rewards/accuracies": 0.
|
| 186 |
-
"rewards/chosen": -1.
|
| 187 |
-
"rewards/margins": 2.
|
| 188 |
-
"rewards/rejected": -
|
| 189 |
"step": 60
|
| 190 |
},
|
| 191 |
{
|
| 192 |
-
"epoch": 1.
|
| 193 |
-
"grad_norm":
|
| 194 |
-
"learning_rate":
|
| 195 |
-
"logits/chosen": -1.
|
| 196 |
-
"logits/rejected": -1.
|
| 197 |
-
"logps/chosen": -
|
| 198 |
-
"logps/rejected": -
|
| 199 |
-
"loss": 0.
|
| 200 |
-
"rewards/accuracies": 0.
|
| 201 |
-
"rewards/chosen": -
|
| 202 |
-
"rewards/margins": 2.
|
| 203 |
-
"rewards/rejected": -4.
|
| 204 |
"step": 65
|
| 205 |
},
|
| 206 |
{
|
| 207 |
-
"epoch": 1.
|
| 208 |
-
"grad_norm":
|
| 209 |
-
"learning_rate":
|
| 210 |
-
"logits/chosen": -1.
|
| 211 |
-
"logits/rejected": -1.
|
| 212 |
-
"logps/chosen": -
|
| 213 |
-
"logps/rejected": -
|
| 214 |
-
"loss": 0.
|
| 215 |
-
"rewards/accuracies": 0.
|
| 216 |
-
"rewards/chosen": -1.
|
| 217 |
-
"rewards/margins": 2.
|
| 218 |
-
"rewards/rejected": -
|
| 219 |
"step": 70
|
| 220 |
},
|
| 221 |
{
|
| 222 |
-
"epoch": 1.
|
| 223 |
-
"grad_norm":
|
| 224 |
-
"learning_rate":
|
| 225 |
-
"logits/chosen": -1.
|
| 226 |
-
"logits/rejected": -1.
|
| 227 |
-
"logps/chosen": -
|
| 228 |
-
"logps/rejected": -
|
| 229 |
-
"loss": 0.
|
| 230 |
"rewards/accuracies": 0.8687499761581421,
|
| 231 |
-
"rewards/chosen": -1.
|
| 232 |
-
"rewards/margins": 2.
|
| 233 |
-
"rewards/rejected": -4.
|
| 234 |
"step": 75
|
| 235 |
},
|
| 236 |
{
|
| 237 |
-
"epoch": 1.
|
| 238 |
-
"grad_norm":
|
| 239 |
-
"learning_rate":
|
| 240 |
-
"logits/chosen": -1.
|
| 241 |
-
"logits/rejected": -1.
|
| 242 |
-
"logps/chosen": -
|
| 243 |
-
"logps/rejected": -
|
| 244 |
-
"loss": 0.
|
| 245 |
-
"rewards/accuracies": 0.
|
| 246 |
-
"rewards/chosen": -1.
|
| 247 |
-
"rewards/margins": 2.
|
| 248 |
-
"rewards/rejected": -
|
| 249 |
"step": 80
|
| 250 |
},
|
| 251 |
{
|
| 252 |
-
"epoch": 1.
|
| 253 |
-
"grad_norm":
|
| 254 |
-
"learning_rate":
|
| 255 |
-
"logits/chosen": -1.
|
| 256 |
-
"logits/rejected": -1.
|
| 257 |
-
"logps/chosen": -
|
| 258 |
-
"logps/rejected": -
|
| 259 |
-
"loss": 0.
|
| 260 |
-
"rewards/accuracies": 0.
|
| 261 |
-
"rewards/chosen": -1.
|
| 262 |
-
"rewards/margins": 2.
|
| 263 |
-
"rewards/rejected": -4.
|
| 264 |
"step": 85
|
| 265 |
},
|
| 266 |
{
|
| 267 |
-
"epoch": 1.
|
| 268 |
-
"grad_norm":
|
| 269 |
-
"learning_rate":
|
| 270 |
-
"logits/chosen": -1.
|
| 271 |
-
"logits/rejected": -1.
|
| 272 |
-
"logps/chosen": -
|
| 273 |
-
"logps/rejected": -
|
| 274 |
-
"loss": 0.
|
| 275 |
-
"rewards/accuracies": 0.
|
| 276 |
-
"rewards/chosen": -1.
|
| 277 |
-
"rewards/margins": 2.
|
| 278 |
-
"rewards/rejected": -
|
| 279 |
"step": 90
|
| 280 |
},
|
| 281 |
{
|
| 282 |
-
"epoch": 1.
|
| 283 |
-
"grad_norm":
|
| 284 |
-
"learning_rate":
|
| 285 |
-
"logits/chosen": -1.
|
| 286 |
-
"logits/rejected": -1.
|
| 287 |
-
"logps/chosen": -
|
| 288 |
-
"logps/rejected": -
|
| 289 |
-
"loss": 0.
|
| 290 |
-
"rewards/accuracies": 0.
|
| 291 |
-
"rewards/chosen": -1.
|
| 292 |
-
"rewards/margins": 3.
|
| 293 |
-
"rewards/rejected": -
|
| 294 |
"step": 95
|
| 295 |
},
|
| 296 |
{
|
| 297 |
-
"epoch": 1.
|
| 298 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 299 |
"total_flos": 0.0,
|
| 300 |
-
"train_loss": 0.
|
| 301 |
-
"train_runtime":
|
| 302 |
-
"train_samples_per_second": 1.
|
| 303 |
"train_steps_per_second": 0.056
|
| 304 |
}
|
| 305 |
],
|
| 306 |
"logging_steps": 5,
|
| 307 |
-
"max_steps":
|
| 308 |
"num_input_tokens_seen": 0,
|
| 309 |
"num_train_epochs": 2,
|
| 310 |
"save_steps": 50,
|
|
|
|
| 1 |
{
|
| 2 |
"best_metric": null,
|
| 3 |
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 1.9820224719101125,
|
| 5 |
"eval_steps": 500,
|
| 6 |
+
"global_step": 110,
|
| 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.0898876404494382,
|
| 13 |
+
"grad_norm": 250.8028214771956,
|
| 14 |
+
"learning_rate": 4.545454545454545e-08,
|
| 15 |
+
"logits/chosen": -1.31005859375,
|
| 16 |
+
"logits/rejected": -1.302734375,
|
| 17 |
+
"logps/chosen": -219.6999969482422,
|
| 18 |
+
"logps/rejected": -277.82501220703125,
|
| 19 |
+
"loss": 0.3525,
|
| 20 |
+
"rewards/accuracies": 0.84375,
|
| 21 |
+
"rewards/chosen": -2.008740186691284,
|
| 22 |
+
"rewards/margins": 2.612988233566284,
|
| 23 |
+
"rewards/rejected": -4.619116306304932,
|
| 24 |
"step": 5
|
| 25 |
},
|
| 26 |
{
|
| 27 |
+
"epoch": 0.1797752808988764,
|
| 28 |
+
"grad_norm": 170.2256638749618,
|
| 29 |
+
"learning_rate": 9.09090909090909e-08,
|
| 30 |
+
"logits/chosen": -1.200585961341858,
|
| 31 |
+
"logits/rejected": -1.272363305091858,
|
| 32 |
+
"logps/chosen": -205.9250030517578,
|
| 33 |
+
"logps/rejected": -230.5124969482422,
|
| 34 |
+
"loss": 0.3659,
|
| 35 |
+
"rewards/accuracies": 0.824999988079071,
|
| 36 |
+
"rewards/chosen": -1.0198485851287842,
|
| 37 |
+
"rewards/margins": 2.195556640625,
|
| 38 |
+
"rewards/rejected": -3.2144532203674316,
|
| 39 |
"step": 10
|
| 40 |
},
|
| 41 |
{
|
| 42 |
+
"epoch": 0.2696629213483146,
|
| 43 |
+
"grad_norm": 295.1115992445591,
|
| 44 |
+
"learning_rate": 1.3636363636363635e-07,
|
| 45 |
+
"logits/chosen": -1.2916991710662842,
|
| 46 |
+
"logits/rejected": -1.23388671875,
|
| 47 |
+
"logps/chosen": -180.6999969482422,
|
| 48 |
+
"logps/rejected": -212.14999389648438,
|
| 49 |
+
"loss": 0.437,
|
| 50 |
+
"rewards/accuracies": 0.8374999761581421,
|
| 51 |
+
"rewards/chosen": -2.502392530441284,
|
| 52 |
+
"rewards/margins": 2.3900389671325684,
|
| 53 |
+
"rewards/rejected": -4.890429496765137,
|
| 54 |
"step": 15
|
| 55 |
},
|
| 56 |
{
|
| 57 |
+
"epoch": 0.3595505617977528,
|
| 58 |
+
"grad_norm": 277.7225526041515,
|
| 59 |
+
"learning_rate": 1.818181818181818e-07,
|
| 60 |
+
"logits/chosen": -1.259765625,
|
| 61 |
+
"logits/rejected": -1.2590820789337158,
|
| 62 |
+
"logps/chosen": -197.78750610351562,
|
| 63 |
+
"logps/rejected": -218.91250610351562,
|
| 64 |
+
"loss": 0.3354,
|
| 65 |
+
"rewards/accuracies": 0.8374999761581421,
|
| 66 |
+
"rewards/chosen": -2.3748536109924316,
|
| 67 |
+
"rewards/margins": 2.6907958984375,
|
| 68 |
+
"rewards/rejected": -5.071179389953613,
|
| 69 |
"step": 20
|
| 70 |
},
|
| 71 |
{
|
| 72 |
+
"epoch": 0.449438202247191,
|
| 73 |
+
"grad_norm": 115.01279948814654,
|
| 74 |
+
"learning_rate": 2.2727272727272726e-07,
|
| 75 |
+
"logits/chosen": -1.180078148841858,
|
| 76 |
+
"logits/rejected": -1.196386694908142,
|
| 77 |
+
"logps/chosen": -206.75,
|
| 78 |
+
"logps/rejected": -224.52499389648438,
|
| 79 |
+
"loss": 0.4124,
|
| 80 |
+
"rewards/accuracies": 0.8125,
|
| 81 |
+
"rewards/chosen": -2.122851610183716,
|
| 82 |
+
"rewards/margins": 1.923669457435608,
|
| 83 |
+
"rewards/rejected": -4.044787406921387,
|
| 84 |
"step": 25
|
| 85 |
},
|
| 86 |
{
|
| 87 |
+
"epoch": 0.5393258426966292,
|
| 88 |
+
"grad_norm": 72.00062004716999,
|
| 89 |
+
"learning_rate": 2.727272727272727e-07,
|
| 90 |
+
"logits/chosen": -1.245996117591858,
|
| 91 |
+
"logits/rejected": -1.2501952648162842,
|
| 92 |
+
"logps/chosen": -208.91250610351562,
|
| 93 |
+
"logps/rejected": -217.8125,
|
| 94 |
+
"loss": 0.3735,
|
| 95 |
+
"rewards/accuracies": 0.8187500238418579,
|
| 96 |
+
"rewards/chosen": -1.261322021484375,
|
| 97 |
+
"rewards/margins": 2.326733350753784,
|
| 98 |
+
"rewards/rejected": -3.585009813308716,
|
| 99 |
"step": 30
|
| 100 |
},
|
| 101 |
{
|
| 102 |
+
"epoch": 0.6292134831460674,
|
| 103 |
+
"grad_norm": 159.67880662723658,
|
| 104 |
+
"learning_rate": 3.1818181818181815e-07,
|
| 105 |
+
"logits/chosen": -1.2498047351837158,
|
| 106 |
+
"logits/rejected": -1.212011694908142,
|
| 107 |
+
"logps/chosen": -200.75,
|
| 108 |
+
"logps/rejected": -203.1750030517578,
|
| 109 |
+
"loss": 0.3959,
|
| 110 |
+
"rewards/accuracies": 0.831250011920929,
|
| 111 |
+
"rewards/chosen": -2.0471434593200684,
|
| 112 |
+
"rewards/margins": 2.1449217796325684,
|
| 113 |
+
"rewards/rejected": -4.191888332366943,
|
| 114 |
"step": 35
|
| 115 |
},
|
| 116 |
{
|
| 117 |
+
"epoch": 0.7191011235955056,
|
| 118 |
+
"grad_norm": 159.3228331743839,
|
| 119 |
+
"learning_rate": 3.636363636363636e-07,
|
| 120 |
+
"logits/chosen": -1.304296851158142,
|
| 121 |
+
"logits/rejected": -1.2903320789337158,
|
| 122 |
+
"logps/chosen": -211.1750030517578,
|
| 123 |
+
"logps/rejected": -218.71249389648438,
|
| 124 |
+
"loss": 0.3872,
|
| 125 |
"rewards/accuracies": 0.8500000238418579,
|
| 126 |
+
"rewards/chosen": -1.924902319908142,
|
| 127 |
+
"rewards/margins": 2.205249071121216,
|
| 128 |
+
"rewards/rejected": -4.130883693695068,
|
| 129 |
"step": 40
|
| 130 |
},
|
| 131 |
{
|
| 132 |
+
"epoch": 0.8089887640449438,
|
| 133 |
+
"grad_norm": 209.54928027137586,
|
| 134 |
+
"learning_rate": 4.090909090909091e-07,
|
| 135 |
+
"logits/chosen": -1.2551758289337158,
|
| 136 |
+
"logits/rejected": -1.2458007335662842,
|
| 137 |
+
"logps/chosen": -222.3249969482422,
|
| 138 |
+
"logps/rejected": -237.64999389648438,
|
| 139 |
+
"loss": 0.4076,
|
| 140 |
+
"rewards/accuracies": 0.8125,
|
| 141 |
+
"rewards/chosen": -1.8654601573944092,
|
| 142 |
+
"rewards/margins": 2.3793578147888184,
|
| 143 |
+
"rewards/rejected": -4.247656345367432,
|
| 144 |
"step": 45
|
| 145 |
},
|
| 146 |
{
|
| 147 |
+
"epoch": 0.898876404494382,
|
| 148 |
+
"grad_norm": 134.6125923552392,
|
| 149 |
+
"learning_rate": 4.545454545454545e-07,
|
| 150 |
+
"logits/chosen": -1.2243163585662842,
|
| 151 |
+
"logits/rejected": -1.208593726158142,
|
| 152 |
+
"logps/chosen": -203.58749389648438,
|
| 153 |
+
"logps/rejected": -223.9875030517578,
|
| 154 |
+
"loss": 0.4013,
|
| 155 |
+
"rewards/accuracies": 0.793749988079071,
|
| 156 |
+
"rewards/chosen": -2.0265870094299316,
|
| 157 |
+
"rewards/margins": 2.5132813453674316,
|
| 158 |
+
"rewards/rejected": -4.545800685882568,
|
| 159 |
"step": 50
|
| 160 |
},
|
| 161 |
{
|
| 162 |
+
"epoch": 0.9887640449438202,
|
| 163 |
+
"grad_norm": 189.31983158924976,
|
| 164 |
+
"learning_rate": 5e-07,
|
| 165 |
+
"logits/chosen": -1.2102539539337158,
|
| 166 |
+
"logits/rejected": -1.205078125,
|
| 167 |
+
"logps/chosen": -204.6750030517578,
|
| 168 |
+
"logps/rejected": -235.71249389648438,
|
| 169 |
+
"loss": 0.4268,
|
| 170 |
+
"rewards/accuracies": 0.78125,
|
| 171 |
+
"rewards/chosen": -1.649023413658142,
|
| 172 |
+
"rewards/margins": 2.216552734375,
|
| 173 |
+
"rewards/rejected": -3.8661131858825684,
|
| 174 |
"step": 55
|
| 175 |
},
|
| 176 |
{
|
| 177 |
+
"epoch": 1.0831460674157303,
|
| 178 |
+
"grad_norm": 125.21413207415151,
|
| 179 |
+
"learning_rate": 4.545454545454545e-07,
|
| 180 |
+
"logits/chosen": -1.2853515148162842,
|
| 181 |
+
"logits/rejected": -1.318261742591858,
|
| 182 |
+
"logps/chosen": -202.3000030517578,
|
| 183 |
+
"logps/rejected": -216.0124969482422,
|
| 184 |
+
"loss": 0.3312,
|
| 185 |
+
"rewards/accuracies": 0.8812500238418579,
|
| 186 |
+
"rewards/chosen": -1.6000487804412842,
|
| 187 |
+
"rewards/margins": 2.2996582984924316,
|
| 188 |
+
"rewards/rejected": -3.900146484375,
|
| 189 |
"step": 60
|
| 190 |
},
|
| 191 |
{
|
| 192 |
+
"epoch": 1.1730337078651685,
|
| 193 |
+
"grad_norm": 164.73999131891642,
|
| 194 |
+
"learning_rate": 4.090909090909091e-07,
|
| 195 |
+
"logits/chosen": -1.240136742591858,
|
| 196 |
+
"logits/rejected": -1.2451171875,
|
| 197 |
+
"logps/chosen": -229.66250610351562,
|
| 198 |
+
"logps/rejected": -257.3999938964844,
|
| 199 |
+
"loss": 0.2984,
|
| 200 |
+
"rewards/accuracies": 0.84375,
|
| 201 |
+
"rewards/chosen": -1.916986107826233,
|
| 202 |
+
"rewards/margins": 2.175122022628784,
|
| 203 |
+
"rewards/rejected": -4.092577934265137,
|
| 204 |
"step": 65
|
| 205 |
},
|
| 206 |
{
|
| 207 |
+
"epoch": 1.2629213483146067,
|
| 208 |
+
"grad_norm": 99.20074290357307,
|
| 209 |
+
"learning_rate": 3.636363636363636e-07,
|
| 210 |
+
"logits/chosen": -1.254492163658142,
|
| 211 |
+
"logits/rejected": -1.2688477039337158,
|
| 212 |
+
"logps/chosen": -222.0625,
|
| 213 |
+
"logps/rejected": -236.8625030517578,
|
| 214 |
+
"loss": 0.3024,
|
| 215 |
+
"rewards/accuracies": 0.862500011920929,
|
| 216 |
+
"rewards/chosen": -1.540551781654358,
|
| 217 |
+
"rewards/margins": 2.271484375,
|
| 218 |
+
"rewards/rejected": -3.810473680496216,
|
| 219 |
"step": 70
|
| 220 |
},
|
| 221 |
{
|
| 222 |
+
"epoch": 1.3528089887640449,
|
| 223 |
+
"grad_norm": 126.64899063848813,
|
| 224 |
+
"learning_rate": 3.1818181818181815e-07,
|
| 225 |
+
"logits/chosen": -1.2423827648162842,
|
| 226 |
+
"logits/rejected": -1.212158203125,
|
| 227 |
+
"logps/chosen": -208.27499389648438,
|
| 228 |
+
"logps/rejected": -230.75,
|
| 229 |
+
"loss": 0.2873,
|
| 230 |
"rewards/accuracies": 0.8687499761581421,
|
| 231 |
+
"rewards/chosen": -1.522576928138733,
|
| 232 |
+
"rewards/margins": 2.6622557640075684,
|
| 233 |
+
"rewards/rejected": -4.184668064117432,
|
| 234 |
"step": 75
|
| 235 |
},
|
| 236 |
{
|
| 237 |
+
"epoch": 1.442696629213483,
|
| 238 |
+
"grad_norm": 117.37315757338318,
|
| 239 |
+
"learning_rate": 2.727272727272727e-07,
|
| 240 |
+
"logits/chosen": -1.244140625,
|
| 241 |
+
"logits/rejected": -1.20654296875,
|
| 242 |
+
"logps/chosen": -203.5812530517578,
|
| 243 |
+
"logps/rejected": -249.3249969482422,
|
| 244 |
+
"loss": 0.3876,
|
| 245 |
+
"rewards/accuracies": 0.793749988079071,
|
| 246 |
+
"rewards/chosen": -1.797338843345642,
|
| 247 |
+
"rewards/margins": 2.134106397628784,
|
| 248 |
+
"rewards/rejected": -3.930224657058716,
|
| 249 |
"step": 80
|
| 250 |
},
|
| 251 |
{
|
| 252 |
+
"epoch": 1.5325842696629213,
|
| 253 |
+
"grad_norm": 133.10246641657412,
|
| 254 |
+
"learning_rate": 2.2727272727272726e-07,
|
| 255 |
+
"logits/chosen": -1.252050757408142,
|
| 256 |
+
"logits/rejected": -1.2130858898162842,
|
| 257 |
+
"logps/chosen": -202.4499969482422,
|
| 258 |
+
"logps/rejected": -219.47500610351562,
|
| 259 |
+
"loss": 0.3168,
|
| 260 |
+
"rewards/accuracies": 0.84375,
|
| 261 |
+
"rewards/chosen": -1.588891625404358,
|
| 262 |
+
"rewards/margins": 2.5704102516174316,
|
| 263 |
+
"rewards/rejected": -4.15966796875,
|
| 264 |
"step": 85
|
| 265 |
},
|
| 266 |
{
|
| 267 |
+
"epoch": 1.6224719101123597,
|
| 268 |
+
"grad_norm": 129.44212708304207,
|
| 269 |
+
"learning_rate": 1.818181818181818e-07,
|
| 270 |
+
"logits/chosen": -1.2580077648162842,
|
| 271 |
+
"logits/rejected": -1.262597680091858,
|
| 272 |
+
"logps/chosen": -165.0124969482422,
|
| 273 |
+
"logps/rejected": -193.21249389648438,
|
| 274 |
+
"loss": 0.2857,
|
| 275 |
+
"rewards/accuracies": 0.893750011920929,
|
| 276 |
+
"rewards/chosen": -1.4154541492462158,
|
| 277 |
+
"rewards/margins": 2.55810546875,
|
| 278 |
+
"rewards/rejected": -3.975329637527466,
|
| 279 |
"step": 90
|
| 280 |
},
|
| 281 |
{
|
| 282 |
+
"epoch": 1.7123595505617977,
|
| 283 |
+
"grad_norm": 133.23298622428365,
|
| 284 |
+
"learning_rate": 1.3636363636363635e-07,
|
| 285 |
+
"logits/chosen": -1.26171875,
|
| 286 |
+
"logits/rejected": -1.2370116710662842,
|
| 287 |
+
"logps/chosen": -191.8874969482422,
|
| 288 |
+
"logps/rejected": -232.5124969482422,
|
| 289 |
+
"loss": 0.2636,
|
| 290 |
+
"rewards/accuracies": 0.862500011920929,
|
| 291 |
+
"rewards/chosen": -1.913476586341858,
|
| 292 |
+
"rewards/margins": 3.128124952316284,
|
| 293 |
+
"rewards/rejected": -5.037988185882568,
|
| 294 |
"step": 95
|
| 295 |
},
|
| 296 |
{
|
| 297 |
+
"epoch": 1.802247191011236,
|
| 298 |
+
"grad_norm": 121.7166467503501,
|
| 299 |
+
"learning_rate": 9.09090909090909e-08,
|
| 300 |
+
"logits/chosen": -1.2141602039337158,
|
| 301 |
+
"logits/rejected": -1.228906273841858,
|
| 302 |
+
"logps/chosen": -199.4375,
|
| 303 |
+
"logps/rejected": -229.64999389648438,
|
| 304 |
+
"loss": 0.2476,
|
| 305 |
+
"rewards/accuracies": 0.8999999761581421,
|
| 306 |
+
"rewards/chosen": -1.579687476158142,
|
| 307 |
+
"rewards/margins": 2.887890577316284,
|
| 308 |
+
"rewards/rejected": -4.467577934265137,
|
| 309 |
+
"step": 100
|
| 310 |
+
},
|
| 311 |
+
{
|
| 312 |
+
"epoch": 1.892134831460674,
|
| 313 |
+
"grad_norm": 124.25297719784706,
|
| 314 |
+
"learning_rate": 4.545454545454545e-08,
|
| 315 |
+
"logits/chosen": -1.291894555091858,
|
| 316 |
+
"logits/rejected": -1.346093773841858,
|
| 317 |
+
"logps/chosen": -214.0625,
|
| 318 |
+
"logps/rejected": -233.0749969482422,
|
| 319 |
+
"loss": 0.2455,
|
| 320 |
+
"rewards/accuracies": 0.8999999761581421,
|
| 321 |
+
"rewards/chosen": -2.6200804710388184,
|
| 322 |
+
"rewards/margins": 2.843066453933716,
|
| 323 |
+
"rewards/rejected": -5.462133884429932,
|
| 324 |
+
"step": 105
|
| 325 |
+
},
|
| 326 |
+
{
|
| 327 |
+
"epoch": 1.9820224719101125,
|
| 328 |
+
"grad_norm": 113.18883753786767,
|
| 329 |
+
"learning_rate": 0.0,
|
| 330 |
+
"logits/chosen": -1.2438476085662842,
|
| 331 |
+
"logits/rejected": -1.2707030773162842,
|
| 332 |
+
"logps/chosen": -215.46249389648438,
|
| 333 |
+
"logps/rejected": -218.7375030517578,
|
| 334 |
+
"loss": 0.2322,
|
| 335 |
+
"rewards/accuracies": 0.8999999761581421,
|
| 336 |
+
"rewards/chosen": -2.182504177093506,
|
| 337 |
+
"rewards/margins": 2.7933106422424316,
|
| 338 |
+
"rewards/rejected": -4.975976467132568,
|
| 339 |
+
"step": 110
|
| 340 |
+
},
|
| 341 |
+
{
|
| 342 |
+
"epoch": 1.9820224719101125,
|
| 343 |
+
"step": 110,
|
| 344 |
"total_flos": 0.0,
|
| 345 |
+
"train_loss": 0.340618884563446,
|
| 346 |
+
"train_runtime": 1955.8801,
|
| 347 |
+
"train_samples_per_second": 1.819,
|
| 348 |
"train_steps_per_second": 0.056
|
| 349 |
}
|
| 350 |
],
|
| 351 |
"logging_steps": 5,
|
| 352 |
+
"max_steps": 110,
|
| 353 |
"num_input_tokens_seen": 0,
|
| 354 |
"num_train_epochs": 2,
|
| 355 |
"save_steps": 50,
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7672
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:03d967e1eb02dcc35b71618607ff5d6e5a726deda0443e58f9972946a8d826b4
|
| 3 |
size 7672
|