whisper-small-ar / checkpoint-2000 /trainer_state.json
MMars's picture
Upload 53 files
cba1d2e
raw
history blame
10.3 kB
{
"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
}