|
{ |
|
"best_metric": 47.79359103847213, |
|
"best_model_checkpoint": "./whisper-small-ar/checkpoint-2000", |
|
"epoch": 0.8312551953449709, |
|
"global_step": 2000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 4.2000000000000006e-07, |
|
"loss": 3.262, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 9.200000000000001e-07, |
|
"loss": 2.5217, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.42e-06, |
|
"loss": 1.7958, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 1.9200000000000003e-06, |
|
"loss": 1.2313, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.42e-06, |
|
"loss": 1.0969, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 2.92e-06, |
|
"loss": 0.9611, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.4200000000000007e-06, |
|
"loss": 0.9175, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 3.920000000000001e-06, |
|
"loss": 0.819, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.42e-06, |
|
"loss": 0.7742, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.92e-06, |
|
"loss": 0.6782, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 5.420000000000001e-06, |
|
"loss": 0.5778, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 5.92e-06, |
|
"loss": 0.456, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 6.42e-06, |
|
"loss": 0.4114, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 6.92e-06, |
|
"loss": 0.4197, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 7.420000000000001e-06, |
|
"loss": 0.3929, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 7.92e-06, |
|
"loss": 0.4104, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 8.42e-06, |
|
"loss": 0.4269, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 8.920000000000001e-06, |
|
"loss": 0.4303, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 9.42e-06, |
|
"loss": 0.3523, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 9.920000000000002e-06, |
|
"loss": 0.3633, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 9.940000000000001e-06, |
|
"loss": 0.361, |
|
"step": 525 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 9.86857142857143e-06, |
|
"loss": 0.3368, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 9.797142857142858e-06, |
|
"loss": 0.329, |
|
"step": 575 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 9.725714285714287e-06, |
|
"loss": 0.3671, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 9.654285714285716e-06, |
|
"loss": 0.3832, |
|
"step": 625 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 9.582857142857143e-06, |
|
"loss": 0.3494, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 9.511428571428572e-06, |
|
"loss": 0.3334, |
|
"step": 675 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 9.440000000000001e-06, |
|
"loss": 0.3466, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 9.368571428571428e-06, |
|
"loss": 0.3464, |
|
"step": 725 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 9.297142857142857e-06, |
|
"loss": 0.3481, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 9.225714285714286e-06, |
|
"loss": 0.3188, |
|
"step": 775 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 9.154285714285715e-06, |
|
"loss": 0.3396, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 9.082857142857143e-06, |
|
"loss": 0.3187, |
|
"step": 825 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 9.011428571428572e-06, |
|
"loss": 0.3079, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 8.94e-06, |
|
"loss": 0.311, |
|
"step": 875 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 8.86857142857143e-06, |
|
"loss": 0.3162, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 8.797142857142857e-06, |
|
"loss": 0.3474, |
|
"step": 925 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 8.725714285714286e-06, |
|
"loss": 0.3154, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 8.654285714285715e-06, |
|
"loss": 0.3315, |
|
"step": 975 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 8.582857142857144e-06, |
|
"loss": 0.2811, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"eval_loss": 0.3930181860923767, |
|
"eval_runtime": 2355.0366, |
|
"eval_samples_per_second": 4.433, |
|
"eval_steps_per_second": 0.554, |
|
"eval_wer": 53.77834909558351, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 8.511428571428571e-06, |
|
"loss": 0.2884, |
|
"step": 1025 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 8.44e-06, |
|
"loss": 0.3166, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 8.36857142857143e-06, |
|
"loss": 0.2723, |
|
"step": 1075 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 8.297142857142859e-06, |
|
"loss": 0.2841, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 8.225714285714288e-06, |
|
"loss": 0.3115, |
|
"step": 1125 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 8.154285714285715e-06, |
|
"loss": 0.3161, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 8.082857142857144e-06, |
|
"loss": 0.2771, |
|
"step": 1175 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 8.011428571428573e-06, |
|
"loss": 0.2826, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 7.94e-06, |
|
"loss": 0.3057, |
|
"step": 1225 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 7.86857142857143e-06, |
|
"loss": 0.2849, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 7.797142857142858e-06, |
|
"loss": 0.2959, |
|
"step": 1275 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 7.725714285714286e-06, |
|
"loss": 0.3015, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 7.654285714285715e-06, |
|
"loss": 0.2686, |
|
"step": 1325 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 7.5828571428571444e-06, |
|
"loss": 0.2554, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 7.511428571428572e-06, |
|
"loss": 0.3009, |
|
"step": 1375 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 7.440000000000001e-06, |
|
"loss": 0.2896, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 7.36857142857143e-06, |
|
"loss": 0.271, |
|
"step": 1425 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 7.297142857142858e-06, |
|
"loss": 0.2695, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 7.225714285714286e-06, |
|
"loss": 0.287, |
|
"step": 1475 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 7.154285714285715e-06, |
|
"loss": 0.2473, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 7.082857142857143e-06, |
|
"loss": 0.2454, |
|
"step": 1525 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 7.011428571428572e-06, |
|
"loss": 0.2952, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 6.9400000000000005e-06, |
|
"loss": 0.2765, |
|
"step": 1575 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 6.868571428571429e-06, |
|
"loss": 0.2615, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 6.797142857142858e-06, |
|
"loss": 0.26, |
|
"step": 1625 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 6.725714285714287e-06, |
|
"loss": 0.2687, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 6.654285714285716e-06, |
|
"loss": 0.2568, |
|
"step": 1675 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 6.582857142857143e-06, |
|
"loss": 0.2634, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 6.511428571428572e-06, |
|
"loss": 0.2655, |
|
"step": 1725 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 6.440000000000001e-06, |
|
"loss": 0.2457, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 6.368571428571429e-06, |
|
"loss": 0.2427, |
|
"step": 1775 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 6.297142857142857e-06, |
|
"loss": 0.236, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 6.225714285714286e-06, |
|
"loss": 0.2553, |
|
"step": 1825 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 6.1542857142857145e-06, |
|
"loss": 0.2521, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 6.0828571428571435e-06, |
|
"loss": 0.2536, |
|
"step": 1875 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.011428571428572e-06, |
|
"loss": 0.2696, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.94e-06, |
|
"loss": 0.2436, |
|
"step": 1925 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.868571428571429e-06, |
|
"loss": 0.2197, |
|
"step": 1950 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 5.797142857142858e-06, |
|
"loss": 0.2526, |
|
"step": 1975 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 5.725714285714287e-06, |
|
"loss": 0.2356, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"eval_loss": 0.3487941324710846, |
|
"eval_runtime": 2297.3957, |
|
"eval_samples_per_second": 4.544, |
|
"eval_steps_per_second": 0.568, |
|
"eval_wer": 47.79359103847213, |
|
"step": 2000 |
|
} |
|
], |
|
"max_steps": 4000, |
|
"num_train_epochs": 2, |
|
"total_flos": 9.23473281024e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|