whisper-medium-mix-jp-ver2 / trainer_state.json
vumichien's picture
End of training
623cdec
{
"best_metric": 8.39864864864865,
"best_model_checkpoint": "./checkpoint-10000",
"epoch": 10.090817356205852,
"global_step": 10000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.5,
"learning_rate": 9.940000000000001e-06,
"loss": 0.3462,
"step": 500
},
{
"epoch": 1.01,
"learning_rate": 9.476842105263158e-06,
"loss": 0.1691,
"step": 1000
},
{
"epoch": 1.01,
"eval_cer": 6.350948206752441,
"eval_loss": 0.1870781034231186,
"eval_runtime": 4246.4995,
"eval_samples_per_second": 1.084,
"eval_steps_per_second": 0.068,
"eval_wer": 10.173986486486486,
"step": 1000
},
{
"epoch": 1.51,
"learning_rate": 8.950526315789474e-06,
"loss": 0.097,
"step": 1500
},
{
"epoch": 2.02,
"learning_rate": 8.42421052631579e-06,
"loss": 0.0916,
"step": 2000
},
{
"epoch": 2.02,
"eval_cer": 5.5499336320984085,
"eval_loss": 0.16914600133895874,
"eval_runtime": 4299.9346,
"eval_samples_per_second": 1.071,
"eval_steps_per_second": 0.067,
"eval_wer": 8.97972972972973,
"step": 2000
},
{
"epoch": 2.52,
"learning_rate": 7.897894736842106e-06,
"loss": 0.0453,
"step": 2500
},
{
"epoch": 3.03,
"learning_rate": 7.371578947368422e-06,
"loss": 0.0452,
"step": 3000
},
{
"epoch": 3.03,
"eval_cer": 5.58673167654519,
"eval_loss": 0.19017226994037628,
"eval_runtime": 4203.2916,
"eval_samples_per_second": 1.095,
"eval_steps_per_second": 0.069,
"eval_wer": 8.98141891891892,
"step": 3000
},
{
"epoch": 3.53,
"learning_rate": 6.845263157894737e-06,
"loss": 0.0214,
"step": 3500
},
{
"epoch": 4.04,
"learning_rate": 6.318947368421054e-06,
"loss": 0.0213,
"step": 4000
},
{
"epoch": 4.04,
"eval_cer": 5.653099578136705,
"eval_loss": 0.20616890490055084,
"eval_runtime": 4260.0725,
"eval_samples_per_second": 1.081,
"eval_steps_per_second": 0.068,
"eval_wer": 8.9375,
"step": 4000
},
{
"epoch": 4.54,
"learning_rate": 5.7926315789473685e-06,
"loss": 0.0096,
"step": 4500
},
{
"epoch": 5.05,
"learning_rate": 5.266315789473685e-06,
"loss": 0.0096,
"step": 5000
},
{
"epoch": 5.05,
"eval_cer": 5.620244181309221,
"eval_loss": 0.22839371860027313,
"eval_runtime": 4320.8285,
"eval_samples_per_second": 1.066,
"eval_steps_per_second": 0.067,
"eval_wer": 8.733108108108107,
"step": 5000
},
{
"epoch": 5.55,
"learning_rate": 4.74e-06,
"loss": 0.0045,
"step": 5500
},
{
"epoch": 6.05,
"learning_rate": 4.213684210526316e-06,
"loss": 0.0041,
"step": 6000
},
{
"epoch": 6.05,
"eval_cer": 5.300889724146089,
"eval_loss": 0.2394615262746811,
"eval_runtime": 4207.8926,
"eval_samples_per_second": 1.094,
"eval_steps_per_second": 0.068,
"eval_wer": 8.505067567567567,
"step": 6000
},
{
"epoch": 6.56,
"learning_rate": 3.687368421052632e-06,
"loss": 0.0024,
"step": 6500
},
{
"epoch": 7.06,
"learning_rate": 3.1610526315789474e-06,
"loss": 0.0022,
"step": 7000
},
{
"epoch": 7.06,
"eval_cer": 5.363972086054855,
"eval_loss": 0.2534642517566681,
"eval_runtime": 4214.0551,
"eval_samples_per_second": 1.093,
"eval_steps_per_second": 0.068,
"eval_wer": 8.550675675675675,
"step": 7000
},
{
"epoch": 7.57,
"learning_rate": 2.6347368421052633e-06,
"loss": 0.0013,
"step": 7500
},
{
"epoch": 8.07,
"learning_rate": 2.1084210526315792e-06,
"loss": 0.001,
"step": 8000
},
{
"epoch": 8.07,
"eval_cer": 5.379085568595497,
"eval_loss": 0.2655620574951172,
"eval_runtime": 4212.0826,
"eval_samples_per_second": 1.093,
"eval_steps_per_second": 0.068,
"eval_wer": 8.555743243243242,
"step": 8000
},
{
"epoch": 8.58,
"learning_rate": 1.5821052631578948e-06,
"loss": 0.0007,
"step": 8500
},
{
"epoch": 9.08,
"learning_rate": 1.0557894736842105e-06,
"loss": 0.0006,
"step": 9000
},
{
"epoch": 9.08,
"eval_cer": 5.273948298747552,
"eval_loss": 0.2720721364021301,
"eval_runtime": 4204.0827,
"eval_samples_per_second": 1.095,
"eval_steps_per_second": 0.069,
"eval_wer": 8.403716216216216,
"step": 9000
},
{
"epoch": 9.59,
"learning_rate": 5.294736842105263e-07,
"loss": 0.0005,
"step": 9500
},
{
"epoch": 10.09,
"learning_rate": 3.1578947368421054e-09,
"loss": 0.0004,
"step": 10000
},
{
"epoch": 10.09,
"eval_cer": 5.2581777082703605,
"eval_loss": 0.2789727747440338,
"eval_runtime": 4206.3116,
"eval_samples_per_second": 1.095,
"eval_steps_per_second": 0.068,
"eval_wer": 8.39864864864865,
"step": 10000
},
{
"epoch": 10.09,
"step": 10000,
"total_flos": 3.265119220137984e+20,
"train_loss": 0.04369454556107521,
"train_runtime": 84974.4249,
"train_samples_per_second": 3.766,
"train_steps_per_second": 0.118
}
],
"max_steps": 10000,
"num_train_epochs": 11,
"total_flos": 3.265119220137984e+20,
"trial_name": null,
"trial_params": null
}