whisper-small-yoruba / trainer_state.json
steja's picture
yoruba whisper small ft
5fde137
raw
history blame
11.4 kB
{
"best_metric": 67.88663748364095,
"best_model_checkpoint": "./whisper-small-yoruba/checkpoint-1600",
"epoch": 181.7826086956522,
"global_step": 2000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 2.26,
"learning_rate": 4.6000000000000004e-07,
"loss": 4.1287,
"step": 25
},
{
"epoch": 4.52,
"learning_rate": 9.600000000000001e-07,
"loss": 3.4626,
"step": 50
},
{
"epoch": 6.78,
"learning_rate": 1.46e-06,
"loss": 2.6201,
"step": 75
},
{
"epoch": 9.09,
"learning_rate": 1.9600000000000003e-06,
"loss": 1.9421,
"step": 100
},
{
"epoch": 11.35,
"learning_rate": 2.46e-06,
"loss": 1.4255,
"step": 125
},
{
"epoch": 13.61,
"learning_rate": 2.96e-06,
"loss": 1.0901,
"step": 150
},
{
"epoch": 15.87,
"learning_rate": 3.46e-06,
"loss": 0.8583,
"step": 175
},
{
"epoch": 18.17,
"learning_rate": 3.96e-06,
"loss": 0.6715,
"step": 200
},
{
"epoch": 20.43,
"learning_rate": 4.4600000000000005e-06,
"loss": 0.483,
"step": 225
},
{
"epoch": 22.7,
"learning_rate": 4.960000000000001e-06,
"loss": 0.3442,
"step": 250
},
{
"epoch": 24.96,
"learning_rate": 5.460000000000001e-06,
"loss": 0.222,
"step": 275
},
{
"epoch": 27.26,
"learning_rate": 5.9600000000000005e-06,
"loss": 0.1322,
"step": 300
},
{
"epoch": 29.52,
"learning_rate": 6.460000000000001e-06,
"loss": 0.0683,
"step": 325
},
{
"epoch": 31.78,
"learning_rate": 6.96e-06,
"loss": 0.0342,
"step": 350
},
{
"epoch": 34.09,
"learning_rate": 7.4600000000000006e-06,
"loss": 0.0191,
"step": 375
},
{
"epoch": 36.35,
"learning_rate": 7.960000000000002e-06,
"loss": 0.013,
"step": 400
},
{
"epoch": 36.35,
"eval_loss": 1.4067741632461548,
"eval_runtime": 308.7178,
"eval_samples_per_second": 2.692,
"eval_steps_per_second": 0.042,
"eval_wer": 72.96809422807888,
"step": 400
},
{
"epoch": 38.61,
"learning_rate": 8.46e-06,
"loss": 0.0113,
"step": 425
},
{
"epoch": 40.87,
"learning_rate": 8.96e-06,
"loss": 0.0087,
"step": 450
},
{
"epoch": 43.17,
"learning_rate": 9.460000000000001e-06,
"loss": 0.0087,
"step": 475
},
{
"epoch": 45.43,
"learning_rate": 9.960000000000001e-06,
"loss": 0.0082,
"step": 500
},
{
"epoch": 47.7,
"learning_rate": 9.846666666666668e-06,
"loss": 0.0078,
"step": 525
},
{
"epoch": 49.96,
"learning_rate": 9.68e-06,
"loss": 0.0056,
"step": 550
},
{
"epoch": 52.26,
"learning_rate": 9.513333333333334e-06,
"loss": 0.0038,
"step": 575
},
{
"epoch": 54.52,
"learning_rate": 9.346666666666666e-06,
"loss": 0.0028,
"step": 600
},
{
"epoch": 56.78,
"learning_rate": 9.180000000000002e-06,
"loss": 0.0025,
"step": 625
},
{
"epoch": 59.09,
"learning_rate": 9.013333333333334e-06,
"loss": 0.0036,
"step": 650
},
{
"epoch": 61.35,
"learning_rate": 8.846666666666668e-06,
"loss": 0.0025,
"step": 675
},
{
"epoch": 63.61,
"learning_rate": 8.68e-06,
"loss": 0.002,
"step": 700
},
{
"epoch": 65.87,
"learning_rate": 8.513333333333335e-06,
"loss": 0.0013,
"step": 725
},
{
"epoch": 68.17,
"learning_rate": 8.346666666666668e-06,
"loss": 0.0012,
"step": 750
},
{
"epoch": 70.43,
"learning_rate": 8.18e-06,
"loss": 0.001,
"step": 775
},
{
"epoch": 72.7,
"learning_rate": 8.013333333333333e-06,
"loss": 0.0008,
"step": 800
},
{
"epoch": 72.7,
"eval_loss": 1.5545728206634521,
"eval_runtime": 309.8001,
"eval_samples_per_second": 2.682,
"eval_steps_per_second": 0.042,
"eval_wer": 68.45074236201995,
"step": 800
},
{
"epoch": 74.96,
"learning_rate": 7.846666666666667e-06,
"loss": 0.0007,
"step": 825
},
{
"epoch": 77.26,
"learning_rate": 7.680000000000001e-06,
"loss": 0.0006,
"step": 850
},
{
"epoch": 79.52,
"learning_rate": 7.513333333333334e-06,
"loss": 0.0006,
"step": 875
},
{
"epoch": 81.78,
"learning_rate": 7.346666666666668e-06,
"loss": 0.0005,
"step": 900
},
{
"epoch": 84.09,
"learning_rate": 7.180000000000001e-06,
"loss": 0.0005,
"step": 925
},
{
"epoch": 86.35,
"learning_rate": 7.0133333333333345e-06,
"loss": 0.0005,
"step": 950
},
{
"epoch": 88.61,
"learning_rate": 6.846666666666667e-06,
"loss": 0.0005,
"step": 975
},
{
"epoch": 90.87,
"learning_rate": 6.680000000000001e-06,
"loss": 0.0004,
"step": 1000
},
{
"epoch": 93.17,
"learning_rate": 6.513333333333333e-06,
"loss": 0.0004,
"step": 1025
},
{
"epoch": 95.43,
"learning_rate": 6.346666666666668e-06,
"loss": 0.0004,
"step": 1050
},
{
"epoch": 97.7,
"learning_rate": 6.18e-06,
"loss": 0.0004,
"step": 1075
},
{
"epoch": 99.96,
"learning_rate": 6.013333333333335e-06,
"loss": 0.0004,
"step": 1100
},
{
"epoch": 102.26,
"learning_rate": 5.846666666666667e-06,
"loss": 0.0004,
"step": 1125
},
{
"epoch": 104.52,
"learning_rate": 5.68e-06,
"loss": 0.0004,
"step": 1150
},
{
"epoch": 106.78,
"learning_rate": 5.513333333333334e-06,
"loss": 0.0004,
"step": 1175
},
{
"epoch": 109.09,
"learning_rate": 5.346666666666667e-06,
"loss": 0.0003,
"step": 1200
},
{
"epoch": 109.09,
"eval_loss": 1.6399505138397217,
"eval_runtime": 309.455,
"eval_samples_per_second": 2.685,
"eval_steps_per_second": 0.042,
"eval_wer": 67.91371451780314,
"step": 1200
},
{
"epoch": 111.35,
"learning_rate": 5.18e-06,
"loss": 0.0003,
"step": 1225
},
{
"epoch": 113.61,
"learning_rate": 5.013333333333333e-06,
"loss": 0.0003,
"step": 1250
},
{
"epoch": 115.87,
"learning_rate": 4.846666666666667e-06,
"loss": 0.0003,
"step": 1275
},
{
"epoch": 118.17,
"learning_rate": 4.680000000000001e-06,
"loss": 0.0003,
"step": 1300
},
{
"epoch": 120.43,
"learning_rate": 4.513333333333333e-06,
"loss": 0.0003,
"step": 1325
},
{
"epoch": 122.7,
"learning_rate": 4.346666666666667e-06,
"loss": 0.0003,
"step": 1350
},
{
"epoch": 124.96,
"learning_rate": 4.18e-06,
"loss": 0.0003,
"step": 1375
},
{
"epoch": 127.26,
"learning_rate": 4.013333333333334e-06,
"loss": 0.0003,
"step": 1400
},
{
"epoch": 129.52,
"learning_rate": 3.8466666666666665e-06,
"loss": 0.0003,
"step": 1425
},
{
"epoch": 131.78,
"learning_rate": 3.6800000000000003e-06,
"loss": 0.0003,
"step": 1450
},
{
"epoch": 134.09,
"learning_rate": 3.5133333333333337e-06,
"loss": 0.0003,
"step": 1475
},
{
"epoch": 136.35,
"learning_rate": 3.346666666666667e-06,
"loss": 0.0003,
"step": 1500
},
{
"epoch": 138.61,
"learning_rate": 3.1800000000000005e-06,
"loss": 0.0003,
"step": 1525
},
{
"epoch": 140.87,
"learning_rate": 3.013333333333334e-06,
"loss": 0.0003,
"step": 1550
},
{
"epoch": 143.17,
"learning_rate": 2.8466666666666672e-06,
"loss": 0.0003,
"step": 1575
},
{
"epoch": 145.43,
"learning_rate": 2.68e-06,
"loss": 0.0002,
"step": 1600
},
{
"epoch": 145.43,
"eval_loss": 1.677327036857605,
"eval_runtime": 306.2007,
"eval_samples_per_second": 2.714,
"eval_steps_per_second": 0.042,
"eval_wer": 67.88663748364095,
"step": 1600
},
{
"epoch": 147.7,
"learning_rate": 2.5133333333333336e-06,
"loss": 0.0003,
"step": 1625
},
{
"epoch": 149.96,
"learning_rate": 2.346666666666667e-06,
"loss": 0.0002,
"step": 1650
},
{
"epoch": 152.26,
"learning_rate": 2.1800000000000003e-06,
"loss": 0.0002,
"step": 1675
},
{
"epoch": 154.52,
"learning_rate": 2.0133333333333337e-06,
"loss": 0.0002,
"step": 1700
},
{
"epoch": 156.78,
"learning_rate": 1.8466666666666668e-06,
"loss": 0.0002,
"step": 1725
},
{
"epoch": 159.09,
"learning_rate": 1.6800000000000002e-06,
"loss": 0.0002,
"step": 1750
},
{
"epoch": 161.35,
"learning_rate": 1.5133333333333334e-06,
"loss": 0.0002,
"step": 1775
},
{
"epoch": 163.61,
"learning_rate": 1.3466666666666668e-06,
"loss": 0.0002,
"step": 1800
},
{
"epoch": 165.87,
"learning_rate": 1.1800000000000001e-06,
"loss": 0.0002,
"step": 1825
},
{
"epoch": 168.17,
"learning_rate": 1.0133333333333333e-06,
"loss": 0.0002,
"step": 1850
},
{
"epoch": 170.43,
"learning_rate": 8.466666666666668e-07,
"loss": 0.0002,
"step": 1875
},
{
"epoch": 172.7,
"learning_rate": 6.800000000000001e-07,
"loss": 0.0002,
"step": 1900
},
{
"epoch": 174.96,
"learning_rate": 5.133333333333334e-07,
"loss": 0.0002,
"step": 1925
},
{
"epoch": 177.26,
"learning_rate": 3.466666666666667e-07,
"loss": 0.0002,
"step": 1950
},
{
"epoch": 179.52,
"learning_rate": 1.8e-07,
"loss": 0.0002,
"step": 1975
},
{
"epoch": 181.78,
"learning_rate": 1.3333333333333334e-08,
"loss": 0.0002,
"step": 2000
},
{
"epoch": 181.78,
"eval_loss": 1.6901103258132935,
"eval_runtime": 305.9854,
"eval_samples_per_second": 2.716,
"eval_steps_per_second": 0.042,
"eval_wer": 68.11227943499254,
"step": 2000
},
{
"epoch": 181.78,
"step": 2000,
"total_flos": 3.861041754179161e+19,
"train_loss": 0.22003283377131447,
"train_runtime": 4964.2974,
"train_samples_per_second": 25.784,
"train_steps_per_second": 0.403
}
],
"max_steps": 2000,
"num_train_epochs": 182,
"total_flos": 3.861041754179161e+19,
"trial_name": null,
"trial_params": null
}