infinitejoy's picture
End of training
29b40dc
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 50.0,
"global_step": 46900,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.07,
"learning_rate": 1.74475e-05,
"loss": 7.5225,
"step": 1000
},
{
"epoch": 2.13,
"learning_rate": 3.49475e-05,
"loss": 2.9185,
"step": 2000
},
{
"epoch": 3.2,
"learning_rate": 5.2447499999999995e-05,
"loss": 1.5385,
"step": 3000
},
{
"epoch": 4.26,
"learning_rate": 6.99475e-05,
"loss": 1.2881,
"step": 4000
},
{
"epoch": 4.26,
"eval_loss": 0.37639039754867554,
"eval_runtime": 433.5071,
"eval_samples_per_second": 21.185,
"eval_steps_per_second": 1.324,
"eval_wer": 0.6460503785612609,
"step": 4000
},
{
"epoch": 5.33,
"learning_rate": 6.837482517482516e-05,
"loss": 1.2081,
"step": 5000
},
{
"epoch": 6.4,
"learning_rate": 6.674475524475524e-05,
"loss": 1.1501,
"step": 6000
},
{
"epoch": 7.46,
"learning_rate": 6.511305361305361e-05,
"loss": 1.1115,
"step": 7000
},
{
"epoch": 8.53,
"learning_rate": 6.348135198135198e-05,
"loss": 1.0767,
"step": 8000
},
{
"epoch": 8.53,
"eval_loss": 0.26572874188423157,
"eval_runtime": 423.456,
"eval_samples_per_second": 21.688,
"eval_steps_per_second": 1.356,
"eval_wer": 0.5164111979865079,
"step": 8000
},
{
"epoch": 9.59,
"learning_rate": 6.185128205128204e-05,
"loss": 1.0506,
"step": 9000
},
{
"epoch": 10.66,
"learning_rate": 6.021958041958042e-05,
"loss": 1.0261,
"step": 10000
},
{
"epoch": 11.73,
"learning_rate": 5.858787878787878e-05,
"loss": 1.0029,
"step": 11000
},
{
"epoch": 12.79,
"learning_rate": 5.695617715617715e-05,
"loss": 0.9841,
"step": 12000
},
{
"epoch": 12.79,
"eval_loss": 0.23304273188114166,
"eval_runtime": 424.6539,
"eval_samples_per_second": 21.627,
"eval_steps_per_second": 1.352,
"eval_wer": 0.4444742433932292,
"step": 12000
},
{
"epoch": 13.86,
"learning_rate": 5.532447552447552e-05,
"loss": 0.9694,
"step": 13000
},
{
"epoch": 14.93,
"learning_rate": 5.369603729603729e-05,
"loss": 0.9593,
"step": 14000
},
{
"epoch": 15.99,
"learning_rate": 5.206433566433566e-05,
"loss": 0.9428,
"step": 15000
},
{
"epoch": 17.06,
"learning_rate": 5.0432634032634027e-05,
"loss": 0.9274,
"step": 16000
},
{
"epoch": 17.06,
"eval_loss": 0.213353231549263,
"eval_runtime": 424.1743,
"eval_samples_per_second": 21.651,
"eval_steps_per_second": 1.353,
"eval_wer": 0.39289913972727086,
"step": 16000
},
{
"epoch": 18.12,
"learning_rate": 4.8800932400932396e-05,
"loss": 0.9171,
"step": 17000
},
{
"epoch": 19.19,
"learning_rate": 4.7170862470862466e-05,
"loss": 0.9057,
"step": 18000
},
{
"epoch": 20.26,
"learning_rate": 4.5539160839160836e-05,
"loss": 0.8881,
"step": 19000
},
{
"epoch": 21.32,
"learning_rate": 4.3909090909090906e-05,
"loss": 0.8781,
"step": 20000
},
{
"epoch": 21.32,
"eval_loss": 0.19452312588691711,
"eval_runtime": 418.381,
"eval_samples_per_second": 21.951,
"eval_steps_per_second": 1.372,
"eval_wer": 0.3886287211437295,
"step": 20000
},
{
"epoch": 22.39,
"learning_rate": 4.2277389277389276e-05,
"loss": 0.8633,
"step": 21000
},
{
"epoch": 23.45,
"learning_rate": 4.064731934731934e-05,
"loss": 0.8537,
"step": 22000
},
{
"epoch": 24.52,
"learning_rate": 3.901561771561771e-05,
"loss": 0.8462,
"step": 23000
},
{
"epoch": 25.59,
"learning_rate": 3.738391608391608e-05,
"loss": 0.8381,
"step": 24000
},
{
"epoch": 25.59,
"eval_loss": 0.184038907289505,
"eval_runtime": 425.8794,
"eval_samples_per_second": 21.565,
"eval_steps_per_second": 1.348,
"eval_wer": 0.3736719410806016,
"step": 24000
},
{
"epoch": 26.65,
"learning_rate": 3.575221445221445e-05,
"loss": 0.8237,
"step": 25000
},
{
"epoch": 27.72,
"learning_rate": 3.412214452214452e-05,
"loss": 0.819,
"step": 26000
},
{
"epoch": 28.78,
"learning_rate": 3.249207459207459e-05,
"loss": 0.8114,
"step": 27000
},
{
"epoch": 29.85,
"learning_rate": 3.086037296037296e-05,
"loss": 0.8054,
"step": 28000
},
{
"epoch": 29.85,
"eval_loss": 0.17562612891197205,
"eval_runtime": 420.9127,
"eval_samples_per_second": 21.819,
"eval_steps_per_second": 1.364,
"eval_wer": 0.3523198481628948,
"step": 28000
},
{
"epoch": 30.92,
"learning_rate": 2.9228671328671326e-05,
"loss": 0.7997,
"step": 29000
},
{
"epoch": 31.98,
"learning_rate": 2.7598601398601396e-05,
"loss": 0.7867,
"step": 30000
},
{
"epoch": 33.05,
"learning_rate": 2.5966899766899766e-05,
"loss": 0.7764,
"step": 31000
},
{
"epoch": 34.12,
"learning_rate": 2.4336829836829836e-05,
"loss": 0.7763,
"step": 32000
},
{
"epoch": 34.12,
"eval_loss": 0.17449165880680084,
"eval_runtime": 413.6966,
"eval_samples_per_second": 22.2,
"eval_steps_per_second": 1.387,
"eval_wer": 0.3299156231304025,
"step": 32000
},
{
"epoch": 35.18,
"learning_rate": 2.2705128205128205e-05,
"loss": 0.7676,
"step": 33000
},
{
"epoch": 36.25,
"learning_rate": 2.1075058275058275e-05,
"loss": 0.7571,
"step": 34000
},
{
"epoch": 37.31,
"learning_rate": 1.944335664335664e-05,
"loss": 0.7556,
"step": 35000
},
{
"epoch": 38.38,
"learning_rate": 1.7814918414918412e-05,
"loss": 0.7474,
"step": 36000
},
{
"epoch": 38.38,
"eval_loss": 0.1677420288324356,
"eval_runtime": 414.2329,
"eval_samples_per_second": 22.171,
"eval_steps_per_second": 1.386,
"eval_wer": 0.30740824789057825,
"step": 36000
},
{
"epoch": 39.45,
"learning_rate": 1.6183216783216782e-05,
"loss": 0.7411,
"step": 37000
},
{
"epoch": 40.51,
"learning_rate": 1.455151515151515e-05,
"loss": 0.7369,
"step": 38000
},
{
"epoch": 41.58,
"learning_rate": 1.2919813519813518e-05,
"loss": 0.7309,
"step": 39000
},
{
"epoch": 42.64,
"learning_rate": 1.1288111888111887e-05,
"loss": 0.7298,
"step": 40000
},
{
"epoch": 42.64,
"eval_loss": 0.1649060845375061,
"eval_runtime": 413.307,
"eval_samples_per_second": 22.221,
"eval_steps_per_second": 1.389,
"eval_wer": 0.2962886555401976,
"step": 40000
},
{
"epoch": 43.71,
"learning_rate": 9.65967365967366e-06,
"loss": 0.7201,
"step": 41000
},
{
"epoch": 44.78,
"learning_rate": 8.027972027972028e-06,
"loss": 0.72,
"step": 42000
},
{
"epoch": 45.84,
"learning_rate": 6.396270396270396e-06,
"loss": 0.7092,
"step": 43000
},
{
"epoch": 46.91,
"learning_rate": 4.764568764568764e-06,
"loss": 0.7125,
"step": 44000
},
{
"epoch": 46.91,
"eval_loss": 0.161702960729599,
"eval_runtime": 419.2465,
"eval_samples_per_second": 21.906,
"eval_steps_per_second": 1.369,
"eval_wer": 0.29307036907144185,
"step": 44000
},
{
"epoch": 47.97,
"learning_rate": 3.1344988344988342e-06,
"loss": 0.7093,
"step": 45000
},
{
"epoch": 49.04,
"learning_rate": 1.5044289044289044e-06,
"loss": 0.7049,
"step": 46000
},
{
"epoch": 50.0,
"step": 46900,
"total_flos": 2.4630771835403823e+20,
"train_loss": 1.0648517145022656,
"train_runtime": 95151.9589,
"train_samples_per_second": 15.765,
"train_steps_per_second": 0.493
}
],
"max_steps": 46900,
"num_train_epochs": 50,
"total_flos": 2.4630771835403823e+20,
"trial_name": null,
"trial_params": null
}