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