whisper-base-hac / trainer_state.json
razhan's picture
End of training
4cee845
Raw History Blame Contribute Delete
4.61 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 5.0,
"eval_steps": 500,
"global_step": 135,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.37037037037037035,
"grad_norm": 34.00384521484375,
"learning_rate": 9.776119402985076e-06,
"loss": 3.5666,
"step": 10
},
{
"epoch": 0.7407407407407407,
"grad_norm": 14.095426559448242,
"learning_rate": 9.029850746268657e-06,
"loss": 1.9468,
"step": 20
},
{
"epoch": 1.0,
"eval_cer": 0.3796859980775393,
"eval_loss": 0.9999122619628906,
"eval_runtime": 91.0467,
"eval_samples_per_second": 13.872,
"eval_steps_per_second": 0.055,
"eval_wer": 0.9633358058705097,
"step": 27
},
{
"epoch": 1.1111111111111112,
"grad_norm": 6.282424449920654,
"learning_rate": 8.283582089552239e-06,
"loss": 1.1917,
"step": 30
},
{
"epoch": 1.4814814814814814,
"grad_norm": 4.256287574768066,
"learning_rate": 7.537313432835821e-06,
"loss": 0.8244,
"step": 40
},
{
"epoch": 1.8518518518518519,
"grad_norm": 2.4645814895629883,
"learning_rate": 6.791044776119403e-06,
"loss": 0.6132,
"step": 50
},
{
"epoch": 2.0,
"eval_cer": 0.1469174661213412,
"eval_loss": 0.468128502368927,
"eval_runtime": 87.6022,
"eval_samples_per_second": 14.417,
"eval_steps_per_second": 0.057,
"eval_wer": 0.6078202818692381,
"step": 54
},
{
"epoch": 2.2222222222222223,
"grad_norm": 2.1577141284942627,
"learning_rate": 6.044776119402986e-06,
"loss": 0.4936,
"step": 60
},
{
"epoch": 2.5925925925925926,
"grad_norm": 2.0963618755340576,
"learning_rate": 5.298507462686567e-06,
"loss": 0.4327,
"step": 70
},
{
"epoch": 2.962962962962963,
"grad_norm": 1.5172204971313477,
"learning_rate": 4.5522388059701495e-06,
"loss": 0.3976,
"step": 80
},
{
"epoch": 3.0,
"eval_cer": 0.11282205929472078,
"eval_loss": 0.366842657327652,
"eval_runtime": 86.2276,
"eval_samples_per_second": 14.647,
"eval_steps_per_second": 0.058,
"eval_wer": 0.5160538306665253,
"step": 81
},
{
"epoch": 3.3333333333333335,
"grad_norm": 1.841143012046814,
"learning_rate": 3.8059701492537314e-06,
"loss": 0.3747,
"step": 90
},
{
"epoch": 3.7037037037037037,
"grad_norm": 1.7737436294555664,
"learning_rate": 3.0597014925373137e-06,
"loss": 0.3485,
"step": 100
},
{
"epoch": 4.0,
"eval_cer": 0.10645155210434061,
"eval_loss": 0.33595842123031616,
"eval_runtime": 86.4694,
"eval_samples_per_second": 14.606,
"eval_steps_per_second": 0.058,
"eval_wer": 0.488926565645862,
"step": 108
},
{
"epoch": 4.074074074074074,
"grad_norm": 1.6087193489074707,
"learning_rate": 2.3134328358208956e-06,
"loss": 0.3447,
"step": 110
},
{
"epoch": 4.444444444444445,
"grad_norm": 1.753946304321289,
"learning_rate": 1.5671641791044779e-06,
"loss": 0.338,
"step": 120
},
{
"epoch": 4.814814814814815,
"grad_norm": 1.5161445140838623,
"learning_rate": 8.208955223880598e-07,
"loss": 0.3292,
"step": 130
},
{
"epoch": 5.0,
"eval_cer": 0.10385057579584221,
"eval_loss": 0.32716336846351624,
"eval_runtime": 85.8417,
"eval_samples_per_second": 14.713,
"eval_steps_per_second": 0.058,
"eval_wer": 0.47917770477906113,
"step": 135
},
{
"epoch": 5.0,
"step": 135,
"total_flos": 3.3623422081150484e+18,
"train_loss": 0.8413535683243363,
"train_runtime": 3448.7065,
"train_samples_per_second": 14.737,
"train_steps_per_second": 0.039
}
],
"logging_steps": 10,
"max_steps": 135,
"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": true
},
"attributes": {}
}
},
"total_flos": 3.3623422081150484e+18,
"train_batch_size": 192,
"trial_name": null,
"trial_params": null
}