|
{ |
|
"best_metric": 34.22500634893761, |
|
"best_model_checkpoint": "./content/drive/MyDrive/whisper-small-hi/checkpoint-2000", |
|
"epoch": 4.88997555012225, |
|
"eval_steps": 1000, |
|
"global_step": 2000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 4.4e-07, |
|
"loss": 2.073, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 9.400000000000001e-07, |
|
"loss": 1.6171, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 1.44e-06, |
|
"loss": 1.1505, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 1.94e-06, |
|
"loss": 0.8303, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 2.4400000000000004e-06, |
|
"loss": 0.6825, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 2.9400000000000002e-06, |
|
"loss": 0.6282, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.44e-06, |
|
"loss": 0.5957, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.94e-06, |
|
"loss": 0.5421, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 4.440000000000001e-06, |
|
"loss": 0.4863, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 4.94e-06, |
|
"loss": 0.4577, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 5.4400000000000004e-06, |
|
"loss": 0.4241, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 5.94e-06, |
|
"loss": 0.3798, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.440000000000001e-06, |
|
"loss": 0.3282, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 6.9400000000000005e-06, |
|
"loss": 0.2936, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 7.440000000000001e-06, |
|
"loss": 0.2704, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 7.94e-06, |
|
"loss": 0.2582, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"learning_rate": 8.44e-06, |
|
"loss": 0.218, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"learning_rate": 8.94e-06, |
|
"loss": 0.1932, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 1.16, |
|
"learning_rate": 9.440000000000001e-06, |
|
"loss": 0.1963, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"learning_rate": 9.940000000000001e-06, |
|
"loss": 0.1856, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 1.28, |
|
"learning_rate": 9.937142857142858e-06, |
|
"loss": 0.1859, |
|
"step": 525 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"learning_rate": 9.865714285714285e-06, |
|
"loss": 0.1849, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 9.794285714285714e-06, |
|
"loss": 0.1733, |
|
"step": 575 |
|
}, |
|
{ |
|
"epoch": 1.47, |
|
"learning_rate": 9.722857142857143e-06, |
|
"loss": 0.1758, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"learning_rate": 9.651428571428572e-06, |
|
"loss": 0.1719, |
|
"step": 625 |
|
}, |
|
{ |
|
"epoch": 1.59, |
|
"learning_rate": 9.58e-06, |
|
"loss": 0.1663, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"learning_rate": 9.508571428571429e-06, |
|
"loss": 0.1743, |
|
"step": 675 |
|
}, |
|
{ |
|
"epoch": 1.71, |
|
"learning_rate": 9.437142857142858e-06, |
|
"loss": 0.164, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 1.77, |
|
"learning_rate": 9.365714285714287e-06, |
|
"loss": 0.1464, |
|
"step": 725 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"learning_rate": 9.294285714285714e-06, |
|
"loss": 0.1725, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 9.222857142857143e-06, |
|
"loss": 0.1588, |
|
"step": 775 |
|
}, |
|
{ |
|
"epoch": 1.96, |
|
"learning_rate": 9.151428571428572e-06, |
|
"loss": 0.1598, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"learning_rate": 9.080000000000001e-06, |
|
"loss": 0.133, |
|
"step": 825 |
|
}, |
|
{ |
|
"epoch": 2.08, |
|
"learning_rate": 9.00857142857143e-06, |
|
"loss": 0.0881, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 8.937142857142857e-06, |
|
"loss": 0.0899, |
|
"step": 875 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"learning_rate": 8.865714285714287e-06, |
|
"loss": 0.0875, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"learning_rate": 8.794285714285716e-06, |
|
"loss": 0.0796, |
|
"step": 925 |
|
}, |
|
{ |
|
"epoch": 2.32, |
|
"learning_rate": 8.722857142857145e-06, |
|
"loss": 0.0873, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 2.38, |
|
"learning_rate": 8.651428571428572e-06, |
|
"loss": 0.0855, |
|
"step": 975 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"learning_rate": 8.580000000000001e-06, |
|
"loss": 0.0825, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"eval_loss": 0.2965221107006073, |
|
"eval_runtime": 1430.9073, |
|
"eval_samples_per_second": 2.022, |
|
"eval_steps_per_second": 0.253, |
|
"eval_wer": 35.4355371201219, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 2.51, |
|
"learning_rate": 8.50857142857143e-06, |
|
"loss": 0.0807, |
|
"step": 1025 |
|
}, |
|
{ |
|
"epoch": 2.57, |
|
"learning_rate": 8.437142857142859e-06, |
|
"loss": 0.0894, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 2.63, |
|
"learning_rate": 8.365714285714286e-06, |
|
"loss": 0.0875, |
|
"step": 1075 |
|
}, |
|
{ |
|
"epoch": 2.69, |
|
"learning_rate": 8.294285714285715e-06, |
|
"loss": 0.0869, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"learning_rate": 8.222857142857144e-06, |
|
"loss": 0.0915, |
|
"step": 1125 |
|
}, |
|
{ |
|
"epoch": 2.81, |
|
"learning_rate": 8.151428571428572e-06, |
|
"loss": 0.0779, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 2.87, |
|
"learning_rate": 8.08e-06, |
|
"loss": 0.0855, |
|
"step": 1175 |
|
}, |
|
{ |
|
"epoch": 2.93, |
|
"learning_rate": 8.00857142857143e-06, |
|
"loss": 0.0836, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 7.937142857142857e-06, |
|
"loss": 0.0844, |
|
"step": 1225 |
|
}, |
|
{ |
|
"epoch": 3.06, |
|
"learning_rate": 7.865714285714286e-06, |
|
"loss": 0.0469, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 3.12, |
|
"learning_rate": 7.794285714285715e-06, |
|
"loss": 0.0405, |
|
"step": 1275 |
|
}, |
|
{ |
|
"epoch": 3.18, |
|
"learning_rate": 7.722857142857142e-06, |
|
"loss": 0.039, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 3.24, |
|
"learning_rate": 7.651428571428571e-06, |
|
"loss": 0.045, |
|
"step": 1325 |
|
}, |
|
{ |
|
"epoch": 3.3, |
|
"learning_rate": 7.58e-06, |
|
"loss": 0.0443, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"learning_rate": 7.508571428571429e-06, |
|
"loss": 0.0406, |
|
"step": 1375 |
|
}, |
|
{ |
|
"epoch": 3.42, |
|
"learning_rate": 7.4371428571428575e-06, |
|
"loss": 0.0442, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 3.48, |
|
"learning_rate": 7.365714285714286e-06, |
|
"loss": 0.0468, |
|
"step": 1425 |
|
}, |
|
{ |
|
"epoch": 3.55, |
|
"learning_rate": 7.294285714285715e-06, |
|
"loss": 0.0462, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 3.61, |
|
"learning_rate": 7.222857142857144e-06, |
|
"loss": 0.0407, |
|
"step": 1475 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 7.151428571428573e-06, |
|
"loss": 0.041, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 3.73, |
|
"learning_rate": 7.08e-06, |
|
"loss": 0.0408, |
|
"step": 1525 |
|
}, |
|
{ |
|
"epoch": 3.79, |
|
"learning_rate": 7.008571428571429e-06, |
|
"loss": 0.0446, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 3.85, |
|
"learning_rate": 6.937142857142858e-06, |
|
"loss": 0.0392, |
|
"step": 1575 |
|
}, |
|
{ |
|
"epoch": 3.91, |
|
"learning_rate": 6.865714285714287e-06, |
|
"loss": 0.0398, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"learning_rate": 6.794285714285714e-06, |
|
"loss": 0.04, |
|
"step": 1625 |
|
}, |
|
{ |
|
"epoch": 4.03, |
|
"learning_rate": 6.722857142857143e-06, |
|
"loss": 0.0281, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 4.1, |
|
"learning_rate": 6.651428571428572e-06, |
|
"loss": 0.0233, |
|
"step": 1675 |
|
}, |
|
{ |
|
"epoch": 4.16, |
|
"learning_rate": 6.5800000000000005e-06, |
|
"loss": 0.0199, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 4.22, |
|
"learning_rate": 6.5085714285714295e-06, |
|
"loss": 0.0194, |
|
"step": 1725 |
|
}, |
|
{ |
|
"epoch": 4.28, |
|
"learning_rate": 6.437142857142858e-06, |
|
"loss": 0.0203, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 4.34, |
|
"learning_rate": 6.365714285714286e-06, |
|
"loss": 0.0207, |
|
"step": 1775 |
|
}, |
|
{ |
|
"epoch": 4.4, |
|
"learning_rate": 6.294285714285715e-06, |
|
"loss": 0.02, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 4.46, |
|
"learning_rate": 6.222857142857144e-06, |
|
"loss": 0.0203, |
|
"step": 1825 |
|
}, |
|
{ |
|
"epoch": 4.52, |
|
"learning_rate": 6.151428571428571e-06, |
|
"loss": 0.0178, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"learning_rate": 6.08e-06, |
|
"loss": 0.0222, |
|
"step": 1875 |
|
}, |
|
{ |
|
"epoch": 4.65, |
|
"learning_rate": 6.008571428571429e-06, |
|
"loss": 0.0214, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 4.71, |
|
"learning_rate": 5.937142857142858e-06, |
|
"loss": 0.0225, |
|
"step": 1925 |
|
}, |
|
{ |
|
"epoch": 4.77, |
|
"learning_rate": 5.865714285714286e-06, |
|
"loss": 0.0181, |
|
"step": 1950 |
|
}, |
|
{ |
|
"epoch": 4.83, |
|
"learning_rate": 5.794285714285715e-06, |
|
"loss": 0.0215, |
|
"step": 1975 |
|
}, |
|
{ |
|
"epoch": 4.89, |
|
"learning_rate": 5.722857142857144e-06, |
|
"loss": 0.022, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 4.89, |
|
"eval_loss": 0.3495931327342987, |
|
"eval_runtime": 1434.3119, |
|
"eval_samples_per_second": 2.018, |
|
"eval_steps_per_second": 0.252, |
|
"eval_wer": 34.22500634893761, |
|
"step": 2000 |
|
} |
|
], |
|
"logging_steps": 25, |
|
"max_steps": 4000, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 10, |
|
"save_steps": 1000, |
|
"total_flos": 9.23011544383488e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|