whisper-small-tr-2 / trainer_state.json
Sercan's picture
End of training
3e825ad
raw
history blame
No virus
15.9 kB
{
"best_metric": 16.632698616044568,
"best_model_checkpoint": "./checkpoint-4000",
"epoch": 7.1138,
"global_step": 5000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 9.200000000000001e-07,
"loss": 1.1412,
"step": 50
},
{
"epoch": 0.02,
"learning_rate": 1.9200000000000003e-06,
"loss": 0.5256,
"step": 100
},
{
"epoch": 0.03,
"learning_rate": 2.92e-06,
"loss": 0.1965,
"step": 150
},
{
"epoch": 0.04,
"learning_rate": 3.920000000000001e-06,
"loss": 0.1692,
"step": 200
},
{
"epoch": 0.05,
"learning_rate": 4.92e-06,
"loss": 0.1573,
"step": 250
},
{
"epoch": 0.06,
"learning_rate": 5.92e-06,
"loss": 0.1267,
"step": 300
},
{
"epoch": 0.07,
"learning_rate": 6.92e-06,
"loss": 0.1248,
"step": 350
},
{
"epoch": 0.08,
"learning_rate": 7.92e-06,
"loss": 0.1412,
"step": 400
},
{
"epoch": 0.08,
"eval_cer": 5.2393487430695105,
"eval_loss": 0.2656303644180298,
"eval_runtime": 3483.4243,
"eval_samples_per_second": 2.912,
"eval_steps_per_second": 0.182,
"eval_wer": 19.8334595003785,
"step": 400
},
{
"epoch": 0.09,
"learning_rate": 8.920000000000001e-06,
"loss": 0.1807,
"step": 450
},
{
"epoch": 0.1,
"learning_rate": 9.920000000000002e-06,
"loss": 0.1606,
"step": 500
},
{
"epoch": 0.11,
"learning_rate": 9.89777777777778e-06,
"loss": 0.1767,
"step": 550
},
{
"epoch": 0.12,
"learning_rate": 9.786666666666667e-06,
"loss": 0.1367,
"step": 600
},
{
"epoch": 1.0,
"learning_rate": 9.675555555555555e-06,
"loss": 0.1498,
"step": 650
},
{
"epoch": 1.01,
"learning_rate": 9.564444444444445e-06,
"loss": 0.1635,
"step": 700
},
{
"epoch": 1.02,
"learning_rate": 9.453333333333335e-06,
"loss": 0.1518,
"step": 750
},
{
"epoch": 1.03,
"learning_rate": 9.342222222222223e-06,
"loss": 0.0851,
"step": 800
},
{
"epoch": 1.03,
"eval_cer": 4.891634975507227,
"eval_loss": 0.23817911744117737,
"eval_runtime": 3529.8171,
"eval_samples_per_second": 2.874,
"eval_steps_per_second": 0.18,
"eval_wer": 18.63001999262408,
"step": 800
},
{
"epoch": 1.04,
"learning_rate": 9.231111111111111e-06,
"loss": 0.0817,
"step": 850
},
{
"epoch": 1.05,
"learning_rate": 9.12e-06,
"loss": 0.0717,
"step": 900
},
{
"epoch": 1.06,
"learning_rate": 9.008888888888889e-06,
"loss": 0.0514,
"step": 950
},
{
"epoch": 1.07,
"learning_rate": 8.897777777777779e-06,
"loss": 0.0455,
"step": 1000
},
{
"epoch": 1.08,
"learning_rate": 8.786666666666668e-06,
"loss": 0.0604,
"step": 1050
},
{
"epoch": 1.09,
"learning_rate": 8.675555555555556e-06,
"loss": 0.0633,
"step": 1100
},
{
"epoch": 1.1,
"learning_rate": 8.564444444444445e-06,
"loss": 0.0533,
"step": 1150
},
{
"epoch": 1.11,
"learning_rate": 8.453333333333334e-06,
"loss": 0.0525,
"step": 1200
},
{
"epoch": 1.11,
"eval_cer": 5.223779469895081,
"eval_loss": 0.253223180770874,
"eval_runtime": 3540.1304,
"eval_samples_per_second": 2.865,
"eval_steps_per_second": 0.179,
"eval_wer": 19.169626739649452,
"step": 1200
},
{
"epoch": 1.12,
"learning_rate": 8.342222222222222e-06,
"loss": 0.0427,
"step": 1250
},
{
"epoch": 2.01,
"learning_rate": 8.231111111111112e-06,
"loss": 0.0513,
"step": 1300
},
{
"epoch": 2.02,
"learning_rate": 8.120000000000002e-06,
"loss": 0.058,
"step": 1350
},
{
"epoch": 2.03,
"learning_rate": 8.00888888888889e-06,
"loss": 0.0368,
"step": 1400
},
{
"epoch": 2.04,
"learning_rate": 7.897777777777778e-06,
"loss": 0.0268,
"step": 1450
},
{
"epoch": 2.05,
"learning_rate": 7.786666666666666e-06,
"loss": 0.0265,
"step": 1500
},
{
"epoch": 2.06,
"learning_rate": 7.675555555555556e-06,
"loss": 0.0202,
"step": 1550
},
{
"epoch": 2.07,
"learning_rate": 7.564444444444446e-06,
"loss": 0.0163,
"step": 1600
},
{
"epoch": 2.07,
"eval_cer": 4.583997670375421,
"eval_loss": 0.24465897679328918,
"eval_runtime": 3488.6047,
"eval_samples_per_second": 2.907,
"eval_steps_per_second": 0.182,
"eval_wer": 17.201420835031737,
"step": 1600
},
{
"epoch": 2.08,
"learning_rate": 7.453333333333334e-06,
"loss": 0.0171,
"step": 1650
},
{
"epoch": 2.09,
"learning_rate": 7.342222222222223e-06,
"loss": 0.0244,
"step": 1700
},
{
"epoch": 2.1,
"learning_rate": 7.231111111111112e-06,
"loss": 0.0202,
"step": 1750
},
{
"epoch": 2.11,
"learning_rate": 7.1200000000000004e-06,
"loss": 0.0171,
"step": 1800
},
{
"epoch": 2.12,
"learning_rate": 7.008888888888889e-06,
"loss": 0.0175,
"step": 1850
},
{
"epoch": 3.0,
"learning_rate": 6.897777777777779e-06,
"loss": 0.0135,
"step": 1900
},
{
"epoch": 3.01,
"learning_rate": 6.786666666666667e-06,
"loss": 0.0219,
"step": 1950
},
{
"epoch": 3.02,
"learning_rate": 6.675555555555556e-06,
"loss": 0.0202,
"step": 2000
},
{
"epoch": 3.02,
"eval_cer": 4.493465230064843,
"eval_loss": 0.2471683919429779,
"eval_runtime": 3536.1026,
"eval_samples_per_second": 2.868,
"eval_steps_per_second": 0.179,
"eval_wer": 17.10631029328985,
"step": 2000
},
{
"epoch": 3.03,
"learning_rate": 6.564444444444446e-06,
"loss": 0.0114,
"step": 2050
},
{
"epoch": 3.04,
"learning_rate": 6.453333333333334e-06,
"loss": 0.01,
"step": 2100
},
{
"epoch": 3.05,
"learning_rate": 6.342222222222223e-06,
"loss": 0.0101,
"step": 2150
},
{
"epoch": 3.06,
"learning_rate": 6.231111111111111e-06,
"loss": 0.008,
"step": 2200
},
{
"epoch": 3.07,
"learning_rate": 6.120000000000001e-06,
"loss": 0.005,
"step": 2250
},
{
"epoch": 3.08,
"learning_rate": 6.00888888888889e-06,
"loss": 0.0083,
"step": 2300
},
{
"epoch": 3.09,
"learning_rate": 5.897777777777778e-06,
"loss": 0.0096,
"step": 2350
},
{
"epoch": 3.1,
"learning_rate": 5.7866666666666674e-06,
"loss": 0.0075,
"step": 2400
},
{
"epoch": 3.1,
"eval_cer": 4.4317647771143225,
"eval_loss": 0.25027787685394287,
"eval_runtime": 3586.7167,
"eval_samples_per_second": 2.828,
"eval_steps_per_second": 0.177,
"eval_wer": 17.01508181447621,
"step": 2400
},
{
"epoch": 3.11,
"learning_rate": 5.675555555555556e-06,
"loss": 0.0068,
"step": 2450
},
{
"epoch": 3.12,
"learning_rate": 5.5644444444444444e-06,
"loss": 0.0051,
"step": 2500
},
{
"epoch": 4.0,
"learning_rate": 5.453333333333334e-06,
"loss": 0.0072,
"step": 2550
},
{
"epoch": 4.01,
"learning_rate": 5.342222222222223e-06,
"loss": 0.0077,
"step": 2600
},
{
"epoch": 4.02,
"learning_rate": 5.231111111111111e-06,
"loss": 0.0062,
"step": 2650
},
{
"epoch": 4.03,
"learning_rate": 5.12e-06,
"loss": 0.0053,
"step": 2700
},
{
"epoch": 4.04,
"learning_rate": 5.00888888888889e-06,
"loss": 0.0041,
"step": 2750
},
{
"epoch": 4.05,
"learning_rate": 4.897777777777778e-06,
"loss": 0.0039,
"step": 2800
},
{
"epoch": 4.05,
"eval_cer": 4.365451206186191,
"eval_loss": 0.2513594627380371,
"eval_runtime": 3531.9344,
"eval_samples_per_second": 2.872,
"eval_steps_per_second": 0.18,
"eval_wer": 16.74333740949941,
"step": 2800
},
{
"epoch": 4.06,
"learning_rate": 4.786666666666667e-06,
"loss": 0.0033,
"step": 2850
},
{
"epoch": 4.07,
"learning_rate": 4.675555555555556e-06,
"loss": 0.0026,
"step": 2900
},
{
"epoch": 4.08,
"learning_rate": 4.564444444444445e-06,
"loss": 0.0035,
"step": 2950
},
{
"epoch": 4.09,
"learning_rate": 4.453333333333334e-06,
"loss": 0.0033,
"step": 3000
},
{
"epoch": 4.1,
"learning_rate": 4.3422222222222225e-06,
"loss": 0.0028,
"step": 3050
},
{
"epoch": 4.11,
"learning_rate": 4.2311111111111114e-06,
"loss": 0.0032,
"step": 3100
},
{
"epoch": 4.12,
"learning_rate": 4.12e-06,
"loss": 0.0023,
"step": 3150
},
{
"epoch": 5.01,
"learning_rate": 4.008888888888889e-06,
"loss": 0.0038,
"step": 3200
},
{
"epoch": 5.01,
"eval_cer": 4.358243209346177,
"eval_loss": 0.2565072774887085,
"eval_runtime": 3521.7324,
"eval_samples_per_second": 2.88,
"eval_steps_per_second": 0.18,
"eval_wer": 16.88697373784429,
"step": 3200
},
{
"epoch": 5.02,
"learning_rate": 3.897777777777778e-06,
"loss": 0.0037,
"step": 3250
},
{
"epoch": 5.03,
"learning_rate": 3.7866666666666667e-06,
"loss": 0.0025,
"step": 3300
},
{
"epoch": 5.04,
"learning_rate": 3.675555555555556e-06,
"loss": 0.002,
"step": 3350
},
{
"epoch": 5.05,
"learning_rate": 3.564444444444445e-06,
"loss": 0.0023,
"step": 3400
},
{
"epoch": 5.06,
"learning_rate": 3.4533333333333334e-06,
"loss": 0.0029,
"step": 3450
},
{
"epoch": 5.07,
"learning_rate": 3.3422222222222224e-06,
"loss": 0.0016,
"step": 3500
},
{
"epoch": 5.08,
"learning_rate": 3.2311111111111117e-06,
"loss": 0.0017,
"step": 3550
},
{
"epoch": 5.09,
"learning_rate": 3.12e-06,
"loss": 0.0023,
"step": 3600
},
{
"epoch": 5.09,
"eval_cer": 4.333736020090129,
"eval_loss": 0.2590169310569763,
"eval_runtime": 3522.239,
"eval_samples_per_second": 2.88,
"eval_steps_per_second": 0.18,
"eval_wer": 16.698693685824647,
"step": 3600
},
{
"epoch": 5.1,
"learning_rate": 3.008888888888889e-06,
"loss": 0.0018,
"step": 3650
},
{
"epoch": 5.11,
"learning_rate": 2.8977777777777785e-06,
"loss": 0.0017,
"step": 3700
},
{
"epoch": 5.12,
"learning_rate": 2.786666666666667e-06,
"loss": 0.0016,
"step": 3750
},
{
"epoch": 6.0,
"learning_rate": 2.675555555555556e-06,
"loss": 0.0014,
"step": 3800
},
{
"epoch": 6.01,
"learning_rate": 2.5644444444444444e-06,
"loss": 0.0024,
"step": 3850
},
{
"epoch": 6.02,
"learning_rate": 2.4533333333333333e-06,
"loss": 0.002,
"step": 3900
},
{
"epoch": 6.03,
"learning_rate": 2.342222222222222e-06,
"loss": 0.0014,
"step": 3950
},
{
"epoch": 6.04,
"learning_rate": 2.2311111111111115e-06,
"loss": 0.0013,
"step": 4000
},
{
"epoch": 6.04,
"eval_cer": 4.285298281325233,
"eval_loss": 0.2575719356536865,
"eval_runtime": 3539.1562,
"eval_samples_per_second": 2.866,
"eval_steps_per_second": 0.179,
"eval_wer": 16.632698616044568,
"step": 4000
},
{
"epoch": 6.05,
"learning_rate": 2.12e-06,
"loss": 0.0013,
"step": 4050
},
{
"epoch": 6.06,
"learning_rate": 2.008888888888889e-06,
"loss": 0.0011,
"step": 4100
},
{
"epoch": 6.07,
"learning_rate": 1.8977777777777779e-06,
"loss": 0.0011,
"step": 4150
},
{
"epoch": 6.08,
"learning_rate": 1.7866666666666668e-06,
"loss": 0.0013,
"step": 4200
},
{
"epoch": 6.09,
"learning_rate": 1.675555555555556e-06,
"loss": 0.0016,
"step": 4250
},
{
"epoch": 6.1,
"learning_rate": 1.5644444444444446e-06,
"loss": 0.0013,
"step": 4300
},
{
"epoch": 6.11,
"learning_rate": 1.4533333333333335e-06,
"loss": 0.0014,
"step": 4350
},
{
"epoch": 6.12,
"learning_rate": 1.3422222222222222e-06,
"loss": 0.0011,
"step": 4400
},
{
"epoch": 6.12,
"eval_cer": 4.355648330483771,
"eval_loss": 0.26468709111213684,
"eval_runtime": 3546.9734,
"eval_samples_per_second": 2.86,
"eval_steps_per_second": 0.179,
"eval_wer": 16.91220714687785,
"step": 4400
},
{
"epoch": 7.0,
"learning_rate": 1.2311111111111112e-06,
"loss": 0.0013,
"step": 4450
},
{
"epoch": 7.01,
"learning_rate": 1.12e-06,
"loss": 0.0016,
"step": 4500
},
{
"epoch": 7.02,
"learning_rate": 1.008888888888889e-06,
"loss": 0.0013,
"step": 4550
},
{
"epoch": 7.03,
"learning_rate": 8.977777777777778e-07,
"loss": 0.0011,
"step": 4600
},
{
"epoch": 7.04,
"learning_rate": 7.866666666666667e-07,
"loss": 0.0011,
"step": 4650
},
{
"epoch": 7.05,
"learning_rate": 6.755555555555555e-07,
"loss": 0.001,
"step": 4700
},
{
"epoch": 7.06,
"learning_rate": 5.644444444444445e-07,
"loss": 0.001,
"step": 4750
},
{
"epoch": 7.07,
"learning_rate": 4.533333333333334e-07,
"loss": 0.001,
"step": 4800
},
{
"epoch": 7.07,
"eval_cer": 4.28385668195723,
"eval_loss": 0.26154619455337524,
"eval_runtime": 3542.9557,
"eval_samples_per_second": 2.863,
"eval_steps_per_second": 0.179,
"eval_wer": 16.634639647508685,
"step": 4800
},
{
"epoch": 7.08,
"learning_rate": 3.422222222222223e-07,
"loss": 0.0012,
"step": 4850
},
{
"epoch": 7.09,
"learning_rate": 2.3111111111111112e-07,
"loss": 0.0011,
"step": 4900
},
{
"epoch": 7.1,
"learning_rate": 1.2000000000000002e-07,
"loss": 0.0012,
"step": 4950
},
{
"epoch": 7.11,
"learning_rate": 8.88888888888889e-09,
"loss": 0.0012,
"step": 5000
},
{
"epoch": 7.11,
"step": 5000,
"total_flos": 4.616962385559552e+19,
"train_loss": 0.04895092924162745,
"train_runtime": 68546.6851,
"train_samples_per_second": 2.334,
"train_steps_per_second": 0.073
}
],
"max_steps": 5000,
"num_train_epochs": 9223372036854775807,
"total_flos": 4.616962385559552e+19,
"trial_name": null,
"trial_params": null
}