File size: 4,447 Bytes
ef014ec |
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 |
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 1000,
"global_step": 8976,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0002228163992869875,
"grad_norm": 5.737454891204834,
"learning_rate": 0.0,
"loss": 1.9528,
"step": 1
},
{
"epoch": 0.22281639928698752,
"grad_norm": 3.56156587600708,
"learning_rate": 8.903743315508023e-06,
"loss": 1.6859,
"step": 1000
},
{
"epoch": 0.22281639928698752,
"eval_cosine_accuracy": 0.9496266841888428,
"eval_loss": 0.5503689050674438,
"eval_runtime": 34.9667,
"eval_samples_per_second": 271.944,
"eval_steps_per_second": 1.087,
"step": 1000
},
{
"epoch": 0.44563279857397503,
"grad_norm": 4.381109237670898,
"learning_rate": 1.7816399286987524e-05,
"loss": 1.1429,
"step": 2000
},
{
"epoch": 0.44563279857397503,
"eval_cosine_accuracy": 0.9598275423049927,
"eval_loss": 0.4875295162200928,
"eval_runtime": 34.4488,
"eval_samples_per_second": 276.033,
"eval_steps_per_second": 1.103,
"step": 2000
},
{
"epoch": 0.6684491978609626,
"grad_norm": 4.41341495513916,
"learning_rate": 1.9252327193503666e-05,
"loss": 0.7836,
"step": 3000
},
{
"epoch": 0.6684491978609626,
"eval_cosine_accuracy": 0.9656115174293518,
"eval_loss": 0.45152774453163147,
"eval_runtime": 37.3805,
"eval_samples_per_second": 254.384,
"eval_steps_per_second": 1.017,
"step": 3000
},
{
"epoch": 0.8912655971479501,
"grad_norm": 4.65971040725708,
"learning_rate": 1.8262032085561498e-05,
"loss": 0.601,
"step": 4000
},
{
"epoch": 0.8912655971479501,
"eval_cosine_accuracy": 0.9693974256515503,
"eval_loss": 0.4410804212093353,
"eval_runtime": 36.7023,
"eval_samples_per_second": 259.085,
"eval_steps_per_second": 1.035,
"step": 4000
},
{
"epoch": 1.1140819964349375,
"grad_norm": 3.0785202980041504,
"learning_rate": 1.7272727272727274e-05,
"loss": 0.7514,
"step": 5000
},
{
"epoch": 1.1140819964349375,
"eval_cosine_accuracy": 0.9719213247299194,
"eval_loss": 0.4301469624042511,
"eval_runtime": 37.2185,
"eval_samples_per_second": 255.491,
"eval_steps_per_second": 1.021,
"step": 5000
},
{
"epoch": 1.3368983957219251,
"grad_norm": 3.4481372833251953,
"learning_rate": 1.6282432164785106e-05,
"loss": 0.8103,
"step": 6000
},
{
"epoch": 1.3368983957219251,
"eval_cosine_accuracy": 0.9718161821365356,
"eval_loss": 0.4272350072860718,
"eval_runtime": 36.6881,
"eval_samples_per_second": 259.185,
"eval_steps_per_second": 1.036,
"step": 6000
},
{
"epoch": 1.5597147950089125,
"grad_norm": 3.873903512954712,
"learning_rate": 1.529312735195088e-05,
"loss": 0.5951,
"step": 7000
},
{
"epoch": 1.5597147950089125,
"eval_cosine_accuracy": 0.9736039638519287,
"eval_loss": 0.3999342918395996,
"eval_runtime": 34.8565,
"eval_samples_per_second": 272.804,
"eval_steps_per_second": 1.09,
"step": 7000
},
{
"epoch": 1.7825311942959001,
"grad_norm": 2.972379446029663,
"learning_rate": 1.4302832244008715e-05,
"loss": 0.4595,
"step": 8000
},
{
"epoch": 1.7825311942959001,
"eval_cosine_accuracy": 0.9741297960281372,
"eval_loss": 0.39611557126045227,
"eval_runtime": 35.4253,
"eval_samples_per_second": 268.424,
"eval_steps_per_second": 1.073,
"step": 8000
}
],
"logging_steps": 1000,
"max_steps": 22440,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"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": 256,
"trial_name": null,
"trial_params": null
}
|