File size: 7,613 Bytes
9a2f4f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 | {
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.23529411764705882,
"eval_steps": 3,
"global_step": 9,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0,
"eval_logits/chosen": -2.3223893642425537,
"eval_logits/rejected": -1.7441076040267944,
"eval_logps/chosen": -205.81431579589844,
"eval_logps/rejected": -160.84881591796875,
"eval_loss": 0.6931471824645996,
"eval_rewards/accuracies": 0.0,
"eval_rewards/chosen": 0.0,
"eval_rewards/margins": 0.0,
"eval_rewards/rejected": 0.0,
"eval_runtime": 13.1039,
"eval_samples_per_second": 1.297,
"eval_steps_per_second": 0.687,
"step": 0
},
{
"epoch": 0.026143790849673203,
"grad_norm": 11.304482460021973,
"learning_rate": 0.0,
"logits/chosen": -2.2234325408935547,
"logits/rejected": -2.2330026626586914,
"logps/chosen": -251.4762420654297,
"logps/rejected": -287.5872497558594,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.05228758169934641,
"grad_norm": 11.91282844543457,
"learning_rate": 2e-05,
"logits/chosen": -2.2173714637756348,
"logits/rejected": -2.2363157272338867,
"logps/chosen": -227.2211151123047,
"logps/rejected": -262.26251220703125,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 2
},
{
"epoch": 0.0784313725490196,
"grad_norm": 9.456698417663574,
"learning_rate": 4e-05,
"logits/chosen": -1.7777986526489258,
"logits/rejected": -1.5732976198196411,
"logps/chosen": -166.3558807373047,
"logps/rejected": -206.31166076660156,
"loss": 0.6976,
"rewards/accuracies": 0.375,
"rewards/chosen": 0.005008268635720015,
"rewards/margins": -0.007055856287479401,
"rewards/rejected": 0.012064123526215553,
"step": 3
},
{
"epoch": 0.0784313725490196,
"eval_logits/chosen": -2.3259360790252686,
"eval_logits/rejected": -1.7466574907302856,
"eval_logps/chosen": -205.6438751220703,
"eval_logps/rejected": -160.8700408935547,
"eval_loss": 0.6828210353851318,
"eval_rewards/accuracies": 0.6666666865348816,
"eval_rewards/chosen": 0.017044195905327797,
"eval_rewards/margins": 0.019166946411132812,
"eval_rewards/rejected": -0.0021227519027888775,
"eval_runtime": 12.9552,
"eval_samples_per_second": 1.312,
"eval_steps_per_second": 0.695,
"step": 3
},
{
"epoch": 0.10457516339869281,
"grad_norm": 11.230616569519043,
"learning_rate": 6e-05,
"logits/chosen": -2.7505903244018555,
"logits/rejected": -1.348630428314209,
"logps/chosen": -209.1761474609375,
"logps/rejected": -190.92935180664062,
"loss": 0.6807,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.005297848954796791,
"rewards/margins": 0.027892589569091797,
"rewards/rejected": -0.033190444111824036,
"step": 4
},
{
"epoch": 0.13071895424836602,
"grad_norm": 9.93552017211914,
"learning_rate": 8e-05,
"logits/chosen": -1.8584319353103638,
"logits/rejected": -1.2940995693206787,
"logps/chosen": -245.94874572753906,
"logps/rejected": -135.0790252685547,
"loss": 0.6947,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.0011208057403564453,
"rewards/margins": 0.0006671445444226265,
"rewards/rejected": -0.0017879479564726353,
"step": 5
},
{
"epoch": 0.1568627450980392,
"grad_norm": 11.156682014465332,
"learning_rate": 0.0001,
"logits/chosen": -2.436879873275757,
"logits/rejected": -2.5771214962005615,
"logps/chosen": -204.76026916503906,
"logps/rejected": -262.71990966796875,
"loss": 0.6572,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.015309764072299004,
"rewards/margins": 0.07809243351221085,
"rewards/rejected": -0.093402199447155,
"step": 6
},
{
"epoch": 0.1568627450980392,
"eval_logits/chosen": -2.3328909873962402,
"eval_logits/rejected": -1.7542129755020142,
"eval_logps/chosen": -205.6370391845703,
"eval_logps/rejected": -161.59091186523438,
"eval_loss": 0.6492722034454346,
"eval_rewards/accuracies": 0.6666666865348816,
"eval_rewards/chosen": 0.017727533355355263,
"eval_rewards/margins": 0.09193826466798782,
"eval_rewards/rejected": -0.07421071827411652,
"eval_runtime": 11.3826,
"eval_samples_per_second": 1.494,
"eval_steps_per_second": 0.791,
"step": 6
},
{
"epoch": 0.1830065359477124,
"grad_norm": 9.182978630065918,
"learning_rate": 0.00012,
"logits/chosen": -3.110889434814453,
"logits/rejected": -2.425046443939209,
"logps/chosen": -183.1647186279297,
"logps/rejected": -238.2554168701172,
"loss": 0.6602,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.02616577222943306,
"rewards/margins": 0.07209749519824982,
"rewards/rejected": -0.04593172296881676,
"step": 7
},
{
"epoch": 0.20915032679738563,
"grad_norm": 10.583196640014648,
"learning_rate": 0.00014,
"logits/chosen": -2.9872114658355713,
"logits/rejected": -2.2816989421844482,
"logps/chosen": -195.46536254882812,
"logps/rejected": -200.18411254882812,
"loss": 0.6859,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.03734993934631348,
"rewards/margins": 0.020749665796756744,
"rewards/rejected": -0.05809960514307022,
"step": 8
},
{
"epoch": 0.23529411764705882,
"grad_norm": 8.478883743286133,
"learning_rate": 0.00016,
"logits/chosen": -2.448885440826416,
"logits/rejected": -1.7898311614990234,
"logps/chosen": -121.42498779296875,
"logps/rejected": -155.28976440429688,
"loss": 0.6212,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.11882638931274414,
"rewards/margins": 0.1818961203098297,
"rewards/rejected": -0.06306972354650497,
"step": 9
},
{
"epoch": 0.23529411764705882,
"eval_logits/chosen": -2.3386030197143555,
"eval_logits/rejected": -1.7669346332550049,
"eval_logps/chosen": -205.10494995117188,
"eval_logps/rejected": -162.58966064453125,
"eval_loss": 0.5985996127128601,
"eval_rewards/accuracies": 0.5555555820465088,
"eval_rewards/chosen": 0.07093840092420578,
"eval_rewards/margins": 0.245022714138031,
"eval_rewards/rejected": -0.17408432066440582,
"eval_runtime": 10.3489,
"eval_samples_per_second": 1.643,
"eval_steps_per_second": 0.87,
"step": 9
}
],
"logging_steps": 1,
"max_steps": 10,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 3,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}
|