File size: 6,402 Bytes
3b135c1 |
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 |
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.971563981042654,
"eval_steps": 128,
"global_step": 104,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.018957345971563982,
"grad_norm": 191.735526981635,
"learning_rate": 4.545454545454545e-08,
"logits/chosen": 117.53560638427734,
"logits/rejected": 126.8960952758789,
"logps/chosen": -335.40118408203125,
"logps/rejected": -439.16552734375,
"loss": 0.8466,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.1895734597156398,
"grad_norm": 215.27789653097236,
"learning_rate": 4.545454545454545e-07,
"logits/chosen": 135.01699829101562,
"logits/rejected": 138.35781860351562,
"logps/chosen": -395.863525390625,
"logps/rejected": -439.87109375,
"loss": 0.7497,
"rewards/accuracies": 0.4791666567325592,
"rewards/chosen": 0.0066477167420089245,
"rewards/margins": 0.033824387937784195,
"rewards/rejected": -0.02717665769159794,
"step": 10
},
{
"epoch": 0.3791469194312796,
"grad_norm": 151.6525661192009,
"learning_rate": 4.885348141000122e-07,
"logits/chosen": 121.6771240234375,
"logits/rejected": 125.45533752441406,
"logps/chosen": -372.70947265625,
"logps/rejected": -423.9148864746094,
"loss": 0.66,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.01487281359732151,
"rewards/margins": 0.18130120635032654,
"rewards/rejected": -0.1961739957332611,
"step": 20
},
{
"epoch": 0.5687203791469194,
"grad_norm": 127.760794507037,
"learning_rate": 4.5025027361734613e-07,
"logits/chosen": 140.84371948242188,
"logits/rejected": 134.668701171875,
"logps/chosen": -408.40216064453125,
"logps/rejected": -448.96710205078125,
"loss": 0.6211,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.7968470454216003,
"rewards/margins": 0.6513954997062683,
"rewards/rejected": -1.4482426643371582,
"step": 30
},
{
"epoch": 0.7582938388625592,
"grad_norm": 117.3200007673402,
"learning_rate": 3.893311157806091e-07,
"logits/chosen": 123.24369049072266,
"logits/rejected": 112.3370361328125,
"logps/chosen": -381.4453125,
"logps/rejected": -402.53936767578125,
"loss": 0.6035,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.3625555038452148,
"rewards/margins": 0.8706506490707397,
"rewards/rejected": -2.233206033706665,
"step": 40
},
{
"epoch": 0.9478672985781991,
"grad_norm": 205.0988018258791,
"learning_rate": 3.126631330646801e-07,
"logits/chosen": 140.59500122070312,
"logits/rejected": 144.99911499023438,
"logps/chosen": -465.456298828125,
"logps/rejected": -541.2003173828125,
"loss": 0.5469,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.3156235218048096,
"rewards/margins": 0.9280710220336914,
"rewards/rejected": -3.243694305419922,
"step": 50
},
{
"epoch": 1.1374407582938388,
"grad_norm": 103.63573519510389,
"learning_rate": 2.2891223348923882e-07,
"logits/chosen": 132.75270080566406,
"logits/rejected": 136.41439819335938,
"logps/chosen": -441.16741943359375,
"logps/rejected": -510.4600524902344,
"loss": 0.463,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -2.1580872535705566,
"rewards/margins": 1.546640157699585,
"rewards/rejected": -3.7047271728515625,
"step": 60
},
{
"epoch": 1.3270142180094786,
"grad_norm": 88.9555531672405,
"learning_rate": 1.4754491880085317e-07,
"logits/chosen": 127.8622055053711,
"logits/rejected": 129.76100158691406,
"logps/chosen": -398.2987365722656,
"logps/rejected": -478.58526611328125,
"loss": 0.4087,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -1.5577425956726074,
"rewards/margins": 1.6958301067352295,
"rewards/rejected": -3.253572940826416,
"step": 70
},
{
"epoch": 1.5165876777251186,
"grad_norm": 91.77009057587127,
"learning_rate": 7.775827023107834e-08,
"logits/chosen": 115.83001708984375,
"logits/rejected": 132.09730529785156,
"logps/chosen": -392.9648742675781,
"logps/rejected": -491.1802673339844,
"loss": 0.3483,
"rewards/accuracies": 0.875,
"rewards/chosen": -1.9656927585601807,
"rewards/margins": 1.931434988975525,
"rewards/rejected": -3.897127628326416,
"step": 80
},
{
"epoch": 1.7061611374407581,
"grad_norm": 83.21809324441205,
"learning_rate": 2.7440387297912122e-08,
"logits/chosen": 115.92982482910156,
"logits/rejected": 129.11094665527344,
"logps/chosen": -424.06640625,
"logps/rejected": -514.6354370117188,
"loss": 0.3396,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -1.8817684650421143,
"rewards/margins": 1.9659159183502197,
"rewards/rejected": -3.847684383392334,
"step": 90
},
{
"epoch": 1.8957345971563981,
"grad_norm": 71.96649220747696,
"learning_rate": 2.27878296044029e-09,
"logits/chosen": 122.7137680053711,
"logits/rejected": 122.6277084350586,
"logps/chosen": -415.48455810546875,
"logps/rejected": -488.2542419433594,
"loss": 0.3434,
"rewards/accuracies": 0.875,
"rewards/chosen": -1.6754734516143799,
"rewards/margins": 1.8990375995635986,
"rewards/rejected": -3.5745112895965576,
"step": 100
},
{
"epoch": 1.971563981042654,
"step": 104,
"total_flos": 0.0,
"train_loss": 0.5053261174605443,
"train_runtime": 2173.3813,
"train_samples_per_second": 6.212,
"train_steps_per_second": 0.048
}
],
"logging_steps": 10,
"max_steps": 104,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 500,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}
|