autoprogrammer's picture
Add checkpoint-228 (epoch ckpt)
b8f9e51 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 500,
"global_step": 228,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.043859649122807015,
"grad_norm": 56.5707893371582,
"learning_rate": 3.6363636363636366e-06,
"loss": 3.5797,
"step": 5
},
{
"epoch": 0.08771929824561403,
"grad_norm": 29.886016845703125,
"learning_rate": 8.181818181818183e-06,
"loss": 1.7984,
"step": 10
},
{
"epoch": 0.13157894736842105,
"grad_norm": 29.6209774017334,
"learning_rate": 9.997973265157192e-06,
"loss": 0.5489,
"step": 15
},
{
"epoch": 0.17543859649122806,
"grad_norm": 5.7636399269104,
"learning_rate": 9.985593613021873e-06,
"loss": 0.3193,
"step": 20
},
{
"epoch": 0.21929824561403508,
"grad_norm": 6.628712177276611,
"learning_rate": 9.961988113473708e-06,
"loss": 0.2388,
"step": 25
},
{
"epoch": 0.2631578947368421,
"grad_norm": 3.4682586193084717,
"learning_rate": 9.927209918004095e-06,
"loss": 0.1914,
"step": 30
},
{
"epoch": 0.30701754385964913,
"grad_norm": 2.8765273094177246,
"learning_rate": 9.881337335184879e-06,
"loss": 0.1452,
"step": 35
},
{
"epoch": 0.3508771929824561,
"grad_norm": 3.5053396224975586,
"learning_rate": 9.824473654344297e-06,
"loss": 0.1294,
"step": 40
},
{
"epoch": 0.39473684210526316,
"grad_norm": 4.058867931365967,
"learning_rate": 9.756746912994832e-06,
"loss": 0.1301,
"step": 45
},
{
"epoch": 0.43859649122807015,
"grad_norm": 2.281430244445801,
"learning_rate": 9.678309608536626e-06,
"loss": 0.1142,
"step": 50
},
{
"epoch": 0.4824561403508772,
"grad_norm": 5.376229286193848,
"learning_rate": 9.58933835488563e-06,
"loss": 0.1276,
"step": 55
},
{
"epoch": 0.5263157894736842,
"grad_norm": 3.5856456756591797,
"learning_rate": 9.490033484799608e-06,
"loss": 0.1227,
"step": 60
},
{
"epoch": 0.5701754385964912,
"grad_norm": 3.554062843322754,
"learning_rate": 9.380618598797473e-06,
"loss": 0.1237,
"step": 65
},
{
"epoch": 0.6140350877192983,
"grad_norm": 3.1894707679748535,
"learning_rate": 9.26134006168757e-06,
"loss": 0.1014,
"step": 70
},
{
"epoch": 0.6578947368421053,
"grad_norm": 2.8556902408599854,
"learning_rate": 9.132466447838598e-06,
"loss": 0.0928,
"step": 75
},
{
"epoch": 0.7017543859649122,
"grad_norm": 2.012010097503662,
"learning_rate": 8.994287936442226e-06,
"loss": 0.0915,
"step": 80
},
{
"epoch": 0.7456140350877193,
"grad_norm": 2.8497273921966553,
"learning_rate": 8.84711565812904e-06,
"loss": 0.1006,
"step": 85
},
{
"epoch": 0.7894736842105263,
"grad_norm": 2.1030471324920654,
"learning_rate": 8.691280994409044e-06,
"loss": 0.1018,
"step": 90
},
{
"epoch": 0.8333333333333334,
"grad_norm": 2.3723015785217285,
"learning_rate": 8.527134831514116e-06,
"loss": 0.0849,
"step": 95
},
{
"epoch": 0.8771929824561403,
"grad_norm": 2.6458847522735596,
"learning_rate": 8.355046770322528e-06,
"loss": 0.0925,
"step": 100
},
{
"epoch": 0.9210526315789473,
"grad_norm": 3.4910573959350586,
"learning_rate": 8.175404294144482e-06,
"loss": 0.0842,
"step": 105
},
{
"epoch": 0.9649122807017544,
"grad_norm": 2.0795462131500244,
"learning_rate": 7.98861189624256e-06,
"loss": 0.1061,
"step": 110
},
{
"epoch": 1.0087719298245614,
"grad_norm": 1.9848618507385254,
"learning_rate": 7.79509016905158e-06,
"loss": 0.0732,
"step": 115
},
{
"epoch": 1.0526315789473684,
"grad_norm": 1.0972084999084473,
"learning_rate": 7.595274857148651e-06,
"loss": 0.0557,
"step": 120
},
{
"epoch": 1.0964912280701755,
"grad_norm": 1.0524636507034302,
"learning_rate": 7.389615876105773e-06,
"loss": 0.0495,
"step": 125
},
{
"epoch": 1.1403508771929824,
"grad_norm": 1.3278300762176514,
"learning_rate": 7.178576299434239e-06,
"loss": 0.0575,
"step": 130
},
{
"epoch": 1.1842105263157894,
"grad_norm": 1.2919989824295044,
"learning_rate": 6.962631315901861e-06,
"loss": 0.0417,
"step": 135
},
{
"epoch": 1.2280701754385965,
"grad_norm": 1.8236581087112427,
"learning_rate": 6.742267159570796e-06,
"loss": 0.0503,
"step": 140
},
{
"epoch": 1.2719298245614035,
"grad_norm": 1.1337546110153198,
"learning_rate": 6.51798001496514e-06,
"loss": 0.0362,
"step": 145
},
{
"epoch": 1.3157894736842106,
"grad_norm": 0.995852530002594,
"learning_rate": 6.290274899833517e-06,
"loss": 0.0503,
"step": 150
},
{
"epoch": 1.3596491228070176,
"grad_norm": 1.819360613822937,
"learning_rate": 6.059664528022267e-06,
"loss": 0.0493,
"step": 155
},
{
"epoch": 1.4035087719298245,
"grad_norm": 0.9036810398101807,
"learning_rate": 5.82666815501964e-06,
"loss": 0.0412,
"step": 160
},
{
"epoch": 1.4473684210526316,
"grad_norm": 1.3854684829711914,
"learning_rate": 5.5918104087704925e-06,
"loss": 0.0328,
"step": 165
},
{
"epoch": 1.4912280701754386,
"grad_norm": 1.4495540857315063,
"learning_rate": 5.355620108394018e-06,
"loss": 0.0416,
"step": 170
},
{
"epoch": 1.5350877192982457,
"grad_norm": 0.9957414865493774,
"learning_rate": 5.118629073464424e-06,
"loss": 0.0446,
"step": 175
},
{
"epoch": 1.5789473684210527,
"grad_norm": 1.6074761152267456,
"learning_rate": 4.8813709265355766e-06,
"loss": 0.0543,
"step": 180
},
{
"epoch": 1.6228070175438596,
"grad_norm": 1.2443127632141113,
"learning_rate": 4.644379891605983e-06,
"loss": 0.0346,
"step": 185
},
{
"epoch": 1.6666666666666665,
"grad_norm": 1.4209715127944946,
"learning_rate": 4.40818959122951e-06,
"loss": 0.0476,
"step": 190
},
{
"epoch": 1.7105263157894737,
"grad_norm": 1.8186906576156616,
"learning_rate": 4.173331844980362e-06,
"loss": 0.0498,
"step": 195
},
{
"epoch": 1.7543859649122808,
"grad_norm": 1.7225463390350342,
"learning_rate": 3.940335471977733e-06,
"loss": 0.0401,
"step": 200
},
{
"epoch": 1.7982456140350878,
"grad_norm": 0.9068968892097473,
"learning_rate": 3.7097251001664824e-06,
"loss": 0.0315,
"step": 205
},
{
"epoch": 1.8421052631578947,
"grad_norm": 1.039904236793518,
"learning_rate": 3.482019985034861e-06,
"loss": 0.0364,
"step": 210
},
{
"epoch": 1.8859649122807016,
"grad_norm": 1.110565185546875,
"learning_rate": 3.257732840429206e-06,
"loss": 0.0356,
"step": 215
},
{
"epoch": 1.9298245614035088,
"grad_norm": 1.3308241367340088,
"learning_rate": 3.0373686840981396e-06,
"loss": 0.037,
"step": 220
},
{
"epoch": 1.973684210526316,
"grad_norm": 1.6800122261047363,
"learning_rate": 2.821423700565763e-06,
"loss": 0.0424,
"step": 225
}
],
"logging_steps": 5,
"max_steps": 342,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"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": 3.6062732656155034e+17,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}