{ "best_metric": 20.935172487967517, "best_model_checkpoint": "/content/checkpoint/whisper-small-hi/checkpoint-2000", "epoch": 2.5873221216041395, "global_step": 2000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.03, "learning_rate": 4.2000000000000006e-07, "loss": 4.1984, "step": 25 }, { "epoch": 0.06, "learning_rate": 9.200000000000001e-07, "loss": 3.3699, "step": 50 }, { "epoch": 0.1, "learning_rate": 1.42e-06, "loss": 2.4858, "step": 75 }, { "epoch": 0.13, "learning_rate": 1.9200000000000003e-06, "loss": 1.4969, "step": 100 }, { "epoch": 0.16, "learning_rate": 2.42e-06, "loss": 1.1419, "step": 125 }, { "epoch": 0.19, "learning_rate": 2.92e-06, "loss": 1.0148, "step": 150 }, { "epoch": 0.23, "learning_rate": 3.4200000000000007e-06, "loss": 0.9297, "step": 175 }, { "epoch": 0.26, "learning_rate": 3.920000000000001e-06, "loss": 0.8621, "step": 200 }, { "epoch": 0.29, "learning_rate": 4.42e-06, "loss": 0.712, "step": 225 }, { "epoch": 0.32, "learning_rate": 4.92e-06, "loss": 0.6186, "step": 250 }, { "epoch": 0.36, "learning_rate": 5.420000000000001e-06, "loss": 0.4717, "step": 275 }, { "epoch": 0.39, "learning_rate": 5.92e-06, "loss": 0.3114, "step": 300 }, { "epoch": 0.42, "learning_rate": 6.42e-06, "loss": 0.3384, "step": 325 }, { "epoch": 0.45, "learning_rate": 6.92e-06, "loss": 0.2962, "step": 350 }, { "epoch": 0.49, "learning_rate": 7.420000000000001e-06, "loss": 0.3173, "step": 375 }, { "epoch": 0.52, "learning_rate": 7.92e-06, "loss": 0.3202, "step": 400 }, { "epoch": 0.55, "learning_rate": 8.42e-06, "loss": 0.3133, "step": 425 }, { "epoch": 0.58, "learning_rate": 8.920000000000001e-06, "loss": 0.3045, "step": 450 }, { "epoch": 0.61, "learning_rate": 9.42e-06, "loss": 0.2761, "step": 475 }, { "epoch": 0.65, "learning_rate": 9.920000000000002e-06, "loss": 0.3105, "step": 500 }, { "epoch": 0.65, "eval_loss": 0.32677161693573, "eval_runtime": 1909.0381, "eval_samples_per_second": 2.655, "eval_steps_per_second": 0.332, "eval_wer": 23.903632599284773, "step": 500 }, { "epoch": 0.68, "learning_rate": 4.800000000000001e-07, "loss": 0.2945, "step": 525 }, { "epoch": 0.71, "learning_rate": 9.800000000000001e-07, "loss": 0.2947, "step": 550 }, { "epoch": 0.74, "learning_rate": 1.48e-06, "loss": 0.2997, "step": 575 }, { "epoch": 0.78, "learning_rate": 1.98e-06, "loss": 0.301, "step": 600 }, { "epoch": 0.81, "learning_rate": 2.4800000000000004e-06, "loss": 0.2875, "step": 625 }, { "epoch": 0.84, "learning_rate": 2.9800000000000003e-06, "loss": 0.3012, "step": 650 }, { "epoch": 0.87, "learning_rate": 3.48e-06, "loss": 0.2679, "step": 675 }, { "epoch": 0.91, "learning_rate": 3.980000000000001e-06, "loss": 0.2838, "step": 700 }, { "epoch": 0.94, "learning_rate": 4.48e-06, "loss": 0.2789, "step": 725 }, { "epoch": 0.97, "learning_rate": 4.960000000000001e-06, "loss": 0.273, "step": 750 }, { "epoch": 1.0, "learning_rate": 5.460000000000001e-06, "loss": 0.2716, "step": 775 }, { "epoch": 1.03, "learning_rate": 5.9600000000000005e-06, "loss": 0.1651, "step": 800 }, { "epoch": 1.07, "learning_rate": 6.460000000000001e-06, "loss": 0.1718, "step": 825 }, { "epoch": 1.1, "learning_rate": 6.96e-06, "loss": 0.1631, "step": 850 }, { "epoch": 1.13, "learning_rate": 7.4600000000000006e-06, "loss": 0.1572, "step": 875 }, { "epoch": 1.16, "learning_rate": 7.960000000000002e-06, "loss": 0.1742, "step": 900 }, { "epoch": 1.2, "learning_rate": 8.46e-06, "loss": 0.1899, "step": 925 }, { "epoch": 1.23, "learning_rate": 8.96e-06, "loss": 0.1787, "step": 950 }, { "epoch": 1.26, "learning_rate": 9.460000000000001e-06, "loss": 0.1836, "step": 975 }, { "epoch": 1.29, "learning_rate": 9.960000000000001e-06, "loss": 0.1775, "step": 1000 }, { "epoch": 1.29, "eval_loss": 0.3070180416107178, "eval_runtime": 2136.5113, "eval_samples_per_second": 2.373, "eval_steps_per_second": 0.297, "eval_wer": 22.419402543626145, "step": 1000 }, { "epoch": 1.33, "learning_rate": 9.934285714285715e-06, "loss": 0.1974, "step": 1025 }, { "epoch": 1.36, "learning_rate": 9.862857142857144e-06, "loss": 0.1784, "step": 1050 }, { "epoch": 1.39, "learning_rate": 9.791428571428571e-06, "loss": 0.1785, "step": 1075 }, { "epoch": 1.42, "learning_rate": 9.72e-06, "loss": 0.1861, "step": 1100 }, { "epoch": 1.46, "learning_rate": 9.648571428571429e-06, "loss": 0.1808, "step": 1125 }, { "epoch": 1.49, "learning_rate": 9.577142857142858e-06, "loss": 0.1741, "step": 1150 }, { "epoch": 1.52, "learning_rate": 9.505714285714287e-06, "loss": 0.184, "step": 1175 }, { "epoch": 1.55, "learning_rate": 9.434285714285714e-06, "loss": 0.1962, "step": 1200 }, { "epoch": 1.58, "learning_rate": 9.362857142857143e-06, "loss": 0.1834, "step": 1225 }, { "epoch": 1.62, "learning_rate": 9.291428571428572e-06, "loss": 0.177, "step": 1250 }, { "epoch": 1.65, "learning_rate": 9.220000000000002e-06, "loss": 0.1855, "step": 1275 }, { "epoch": 1.68, "learning_rate": 9.148571428571429e-06, "loss": 0.1675, "step": 1300 }, { "epoch": 1.71, "learning_rate": 9.077142857142858e-06, "loss": 0.1822, "step": 1325 }, { "epoch": 1.75, "learning_rate": 9.005714285714287e-06, "loss": 0.1666, "step": 1350 }, { "epoch": 1.78, "learning_rate": 8.934285714285716e-06, "loss": 0.196, "step": 1375 }, { "epoch": 1.81, "learning_rate": 8.862857142857143e-06, "loss": 0.1843, "step": 1400 }, { "epoch": 1.84, "learning_rate": 8.791428571428572e-06, "loss": 0.1698, "step": 1425 }, { "epoch": 1.88, "learning_rate": 8.720000000000001e-06, "loss": 0.161, "step": 1450 }, { "epoch": 1.91, "learning_rate": 8.64857142857143e-06, "loss": 0.1897, "step": 1475 }, { "epoch": 1.94, "learning_rate": 8.577142857142858e-06, "loss": 0.1658, "step": 1500 }, { "epoch": 1.94, "eval_loss": 0.2858450710773468, "eval_runtime": 1954.9187, "eval_samples_per_second": 2.593, "eval_steps_per_second": 0.324, "eval_wer": 21.284719421365384, "step": 1500 }, { "epoch": 1.97, "learning_rate": 8.505714285714287e-06, "loss": 0.1836, "step": 1525 }, { "epoch": 2.01, "learning_rate": 8.434285714285716e-06, "loss": 0.1407, "step": 1550 }, { "epoch": 2.04, "learning_rate": 8.362857142857143e-06, "loss": 0.067, "step": 1575 }, { "epoch": 2.07, "learning_rate": 8.291428571428572e-06, "loss": 0.0688, "step": 1600 }, { "epoch": 2.1, "learning_rate": 8.220000000000001e-06, "loss": 0.0763, "step": 1625 }, { "epoch": 2.13, "learning_rate": 8.148571428571428e-06, "loss": 0.0751, "step": 1650 }, { "epoch": 2.17, "learning_rate": 8.077142857142857e-06, "loss": 0.0774, "step": 1675 }, { "epoch": 2.2, "learning_rate": 8.005714285714286e-06, "loss": 0.0796, "step": 1700 }, { "epoch": 2.23, "learning_rate": 7.934285714285715e-06, "loss": 0.0671, "step": 1725 }, { "epoch": 2.26, "learning_rate": 7.862857142857143e-06, "loss": 0.0655, "step": 1750 }, { "epoch": 2.3, "learning_rate": 7.791428571428572e-06, "loss": 0.0691, "step": 1775 }, { "epoch": 2.33, "learning_rate": 7.72e-06, "loss": 0.0634, "step": 1800 }, { "epoch": 2.36, "learning_rate": 7.64857142857143e-06, "loss": 0.0695, "step": 1825 }, { "epoch": 2.39, "learning_rate": 7.577142857142857e-06, "loss": 0.0704, "step": 1850 }, { "epoch": 2.43, "learning_rate": 7.505714285714286e-06, "loss": 0.0706, "step": 1875 }, { "epoch": 2.46, "learning_rate": 7.434285714285715e-06, "loss": 0.0641, "step": 1900 }, { "epoch": 2.49, "learning_rate": 7.362857142857144e-06, "loss": 0.0683, "step": 1925 }, { "epoch": 2.52, "learning_rate": 7.291428571428571e-06, "loss": 0.0605, "step": 1950 }, { "epoch": 2.55, "learning_rate": 7.22e-06, "loss": 0.0674, "step": 1975 }, { "epoch": 2.59, "learning_rate": 7.148571428571429e-06, "loss": 0.0684, "step": 2000 }, { "epoch": 2.59, "eval_loss": 0.29735004901885986, "eval_runtime": 1980.7436, "eval_samples_per_second": 2.559, "eval_steps_per_second": 0.32, "eval_wer": 20.935172487967517, "step": 2000 } ], "max_steps": 4000, "num_train_epochs": 6, "total_flos": 9.23011544383488e+18, "trial_name": null, "trial_params": null }