|
{ |
|
"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 |
|
} |
|
|