whisper-small-kannada / trainer_state.json
steja's picture
kannada whisper small ft
e3a128f
{
"best_metric": 23.110834371108343,
"best_model_checkpoint": "/ml_disk/home/saiteja.malyala/projects/whisper/outdir/kn_in/2022-12-08_16-18-24/checkpoint-3500",
"epoch": 48.775757575757574,
"global_step": 4000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.3,
"learning_rate": 4.6000000000000004e-07,
"loss": 2.0129,
"step": 25
},
{
"epoch": 0.61,
"learning_rate": 9.600000000000001e-07,
"loss": 1.8503,
"step": 50
},
{
"epoch": 0.91,
"learning_rate": 1.46e-06,
"loss": 1.6612,
"step": 75
},
{
"epoch": 1.22,
"learning_rate": 1.9600000000000003e-06,
"loss": 1.2994,
"step": 100
},
{
"epoch": 1.52,
"learning_rate": 2.46e-06,
"loss": 0.8061,
"step": 125
},
{
"epoch": 1.82,
"learning_rate": 2.96e-06,
"loss": 0.5326,
"step": 150
},
{
"epoch": 2.13,
"learning_rate": 3.46e-06,
"loss": 0.4048,
"step": 175
},
{
"epoch": 2.44,
"learning_rate": 3.96e-06,
"loss": 0.3291,
"step": 200
},
{
"epoch": 2.74,
"learning_rate": 4.4600000000000005e-06,
"loss": 0.2936,
"step": 225
},
{
"epoch": 3.05,
"learning_rate": 4.960000000000001e-06,
"loss": 0.2742,
"step": 250
},
{
"epoch": 3.35,
"learning_rate": 5.460000000000001e-06,
"loss": 0.2294,
"step": 275
},
{
"epoch": 3.65,
"learning_rate": 5.9600000000000005e-06,
"loss": 0.2148,
"step": 300
},
{
"epoch": 3.96,
"learning_rate": 6.460000000000001e-06,
"loss": 0.2013,
"step": 325
},
{
"epoch": 4.27,
"learning_rate": 6.96e-06,
"loss": 0.1752,
"step": 350
},
{
"epoch": 4.57,
"learning_rate": 7.4600000000000006e-06,
"loss": 0.1587,
"step": 375
},
{
"epoch": 4.87,
"learning_rate": 7.960000000000002e-06,
"loss": 0.1406,
"step": 400
},
{
"epoch": 5.18,
"learning_rate": 8.46e-06,
"loss": 0.1077,
"step": 425
},
{
"epoch": 5.48,
"learning_rate": 8.96e-06,
"loss": 0.0902,
"step": 450
},
{
"epoch": 5.79,
"learning_rate": 9.460000000000001e-06,
"loss": 0.0872,
"step": 475
},
{
"epoch": 6.1,
"learning_rate": 9.960000000000001e-06,
"loss": 0.0792,
"step": 500
},
{
"epoch": 6.1,
"eval_loss": 0.1414351910352707,
"eval_runtime": 112.7148,
"eval_samples_per_second": 7.275,
"eval_steps_per_second": 0.231,
"eval_wer": 24.70485678704857,
"step": 500
},
{
"epoch": 6.4,
"learning_rate": 9.934285714285715e-06,
"loss": 0.0633,
"step": 525
},
{
"epoch": 6.7,
"learning_rate": 9.862857142857144e-06,
"loss": 0.0578,
"step": 550
},
{
"epoch": 7.01,
"learning_rate": 9.791428571428571e-06,
"loss": 0.0613,
"step": 575
},
{
"epoch": 7.32,
"learning_rate": 9.72e-06,
"loss": 0.0386,
"step": 600
},
{
"epoch": 7.62,
"learning_rate": 9.648571428571429e-06,
"loss": 0.0381,
"step": 625
},
{
"epoch": 7.92,
"learning_rate": 9.577142857142858e-06,
"loss": 0.0368,
"step": 650
},
{
"epoch": 8.23,
"learning_rate": 9.505714285714287e-06,
"loss": 0.0263,
"step": 675
},
{
"epoch": 8.53,
"learning_rate": 9.434285714285714e-06,
"loss": 0.0216,
"step": 700
},
{
"epoch": 8.84,
"learning_rate": 9.362857142857143e-06,
"loss": 0.0234,
"step": 725
},
{
"epoch": 9.15,
"learning_rate": 9.291428571428572e-06,
"loss": 0.0194,
"step": 750
},
{
"epoch": 9.45,
"learning_rate": 9.220000000000002e-06,
"loss": 0.0125,
"step": 775
},
{
"epoch": 9.75,
"learning_rate": 9.148571428571429e-06,
"loss": 0.0151,
"step": 800
},
{
"epoch": 10.06,
"learning_rate": 9.077142857142858e-06,
"loss": 0.0136,
"step": 825
},
{
"epoch": 10.36,
"learning_rate": 9.005714285714287e-06,
"loss": 0.0079,
"step": 850
},
{
"epoch": 10.67,
"learning_rate": 8.934285714285716e-06,
"loss": 0.008,
"step": 875
},
{
"epoch": 10.97,
"learning_rate": 8.862857142857143e-06,
"loss": 0.0105,
"step": 900
},
{
"epoch": 11.28,
"learning_rate": 8.791428571428572e-06,
"loss": 0.0058,
"step": 925
},
{
"epoch": 11.58,
"learning_rate": 8.720000000000001e-06,
"loss": 0.0062,
"step": 950
},
{
"epoch": 11.88,
"learning_rate": 8.64857142857143e-06,
"loss": 0.0061,
"step": 975
},
{
"epoch": 12.19,
"learning_rate": 8.577142857142858e-06,
"loss": 0.0067,
"step": 1000
},
{
"epoch": 12.19,
"eval_loss": 0.1930890679359436,
"eval_runtime": 107.7523,
"eval_samples_per_second": 7.61,
"eval_steps_per_second": 0.241,
"eval_wer": 23.775840597758403,
"step": 1000
},
{
"epoch": 12.5,
"learning_rate": 8.505714285714287e-06,
"loss": 0.0043,
"step": 1025
},
{
"epoch": 12.8,
"learning_rate": 8.434285714285716e-06,
"loss": 0.0052,
"step": 1050
},
{
"epoch": 13.11,
"learning_rate": 8.362857142857143e-06,
"loss": 0.0033,
"step": 1075
},
{
"epoch": 13.41,
"learning_rate": 8.291428571428572e-06,
"loss": 0.0036,
"step": 1100
},
{
"epoch": 13.72,
"learning_rate": 8.220000000000001e-06,
"loss": 0.0033,
"step": 1125
},
{
"epoch": 14.02,
"learning_rate": 8.148571428571428e-06,
"loss": 0.0038,
"step": 1150
},
{
"epoch": 14.33,
"learning_rate": 8.077142857142857e-06,
"loss": 0.0028,
"step": 1175
},
{
"epoch": 14.63,
"learning_rate": 8.005714285714286e-06,
"loss": 0.004,
"step": 1200
},
{
"epoch": 14.93,
"learning_rate": 7.934285714285715e-06,
"loss": 0.0034,
"step": 1225
},
{
"epoch": 15.24,
"learning_rate": 7.862857142857143e-06,
"loss": 0.0029,
"step": 1250
},
{
"epoch": 15.55,
"learning_rate": 7.791428571428572e-06,
"loss": 0.0019,
"step": 1275
},
{
"epoch": 15.85,
"learning_rate": 7.72e-06,
"loss": 0.002,
"step": 1300
},
{
"epoch": 16.16,
"learning_rate": 7.64857142857143e-06,
"loss": 0.0017,
"step": 1325
},
{
"epoch": 16.46,
"learning_rate": 7.577142857142857e-06,
"loss": 0.0019,
"step": 1350
},
{
"epoch": 16.76,
"learning_rate": 7.505714285714286e-06,
"loss": 0.0017,
"step": 1375
},
{
"epoch": 17.07,
"learning_rate": 7.434285714285715e-06,
"loss": 0.0023,
"step": 1400
},
{
"epoch": 17.38,
"learning_rate": 7.362857142857144e-06,
"loss": 0.0021,
"step": 1425
},
{
"epoch": 17.68,
"learning_rate": 7.291428571428571e-06,
"loss": 0.0013,
"step": 1450
},
{
"epoch": 17.98,
"learning_rate": 7.22e-06,
"loss": 0.0016,
"step": 1475
},
{
"epoch": 18.29,
"learning_rate": 7.148571428571429e-06,
"loss": 0.0011,
"step": 1500
},
{
"epoch": 18.29,
"eval_loss": 0.2161669284105301,
"eval_runtime": 105.1782,
"eval_samples_per_second": 7.796,
"eval_steps_per_second": 0.247,
"eval_wer": 23.322540473225406,
"step": 1500
},
{
"epoch": 18.59,
"learning_rate": 7.077142857142858e-06,
"loss": 0.0013,
"step": 1525
},
{
"epoch": 18.9,
"learning_rate": 7.0057142857142865e-06,
"loss": 0.0014,
"step": 1550
},
{
"epoch": 19.21,
"learning_rate": 6.934285714285715e-06,
"loss": 0.0011,
"step": 1575
},
{
"epoch": 19.51,
"learning_rate": 6.862857142857144e-06,
"loss": 0.0012,
"step": 1600
},
{
"epoch": 19.81,
"learning_rate": 6.791428571428572e-06,
"loss": 0.0009,
"step": 1625
},
{
"epoch": 20.12,
"learning_rate": 6.720000000000001e-06,
"loss": 0.0009,
"step": 1650
},
{
"epoch": 20.42,
"learning_rate": 6.648571428571429e-06,
"loss": 0.0005,
"step": 1675
},
{
"epoch": 20.73,
"learning_rate": 6.577142857142857e-06,
"loss": 0.0013,
"step": 1700
},
{
"epoch": 21.04,
"learning_rate": 6.505714285714286e-06,
"loss": 0.001,
"step": 1725
},
{
"epoch": 21.34,
"learning_rate": 6.434285714285715e-06,
"loss": 0.0005,
"step": 1750
},
{
"epoch": 21.64,
"learning_rate": 6.3628571428571426e-06,
"loss": 0.0007,
"step": 1775
},
{
"epoch": 21.95,
"learning_rate": 6.2914285714285716e-06,
"loss": 0.0008,
"step": 1800
},
{
"epoch": 22.25,
"learning_rate": 6.220000000000001e-06,
"loss": 0.0009,
"step": 1825
},
{
"epoch": 22.56,
"learning_rate": 6.14857142857143e-06,
"loss": 0.0006,
"step": 1850
},
{
"epoch": 22.86,
"learning_rate": 6.077142857142858e-06,
"loss": 0.0005,
"step": 1875
},
{
"epoch": 23.17,
"learning_rate": 6.005714285714286e-06,
"loss": 0.0004,
"step": 1900
},
{
"epoch": 23.47,
"learning_rate": 5.934285714285715e-06,
"loss": 0.0003,
"step": 1925
},
{
"epoch": 23.78,
"learning_rate": 5.862857142857143e-06,
"loss": 0.0004,
"step": 1950
},
{
"epoch": 24.08,
"learning_rate": 5.791428571428572e-06,
"loss": 0.0003,
"step": 1975
},
{
"epoch": 24.39,
"learning_rate": 5.72e-06,
"loss": 0.0002,
"step": 2000
},
{
"epoch": 24.39,
"eval_loss": 0.22942006587982178,
"eval_runtime": 105.2283,
"eval_samples_per_second": 7.793,
"eval_steps_per_second": 0.247,
"eval_wer": 23.133250311332503,
"step": 2000
},
{
"epoch": 24.69,
"learning_rate": 5.6485714285714285e-06,
"loss": 0.0002,
"step": 2025
},
{
"epoch": 24.99,
"learning_rate": 5.5771428571428575e-06,
"loss": 0.0002,
"step": 2050
},
{
"epoch": 25.3,
"learning_rate": 5.5057142857142865e-06,
"loss": 0.0002,
"step": 2075
},
{
"epoch": 25.61,
"learning_rate": 5.4342857142857155e-06,
"loss": 0.0001,
"step": 2100
},
{
"epoch": 25.91,
"learning_rate": 5.362857142857143e-06,
"loss": 0.0001,
"step": 2125
},
{
"epoch": 26.22,
"learning_rate": 5.291428571428572e-06,
"loss": 0.0001,
"step": 2150
},
{
"epoch": 26.52,
"learning_rate": 5.220000000000001e-06,
"loss": 0.0001,
"step": 2175
},
{
"epoch": 26.82,
"learning_rate": 5.14857142857143e-06,
"loss": 0.0001,
"step": 2200
},
{
"epoch": 27.13,
"learning_rate": 5.077142857142857e-06,
"loss": 0.0001,
"step": 2225
},
{
"epoch": 27.44,
"learning_rate": 5.005714285714286e-06,
"loss": 0.0001,
"step": 2250
},
{
"epoch": 27.74,
"learning_rate": 4.934285714285715e-06,
"loss": 0.0001,
"step": 2275
},
{
"epoch": 28.05,
"learning_rate": 4.862857142857143e-06,
"loss": 0.0001,
"step": 2300
},
{
"epoch": 28.35,
"learning_rate": 4.7914285714285715e-06,
"loss": 0.0001,
"step": 2325
},
{
"epoch": 28.65,
"learning_rate": 4.7200000000000005e-06,
"loss": 0.0001,
"step": 2350
},
{
"epoch": 28.96,
"learning_rate": 4.648571428571429e-06,
"loss": 0.0001,
"step": 2375
},
{
"epoch": 29.27,
"learning_rate": 4.577142857142858e-06,
"loss": 0.0001,
"step": 2400
},
{
"epoch": 29.57,
"learning_rate": 4.505714285714286e-06,
"loss": 0.0001,
"step": 2425
},
{
"epoch": 29.87,
"learning_rate": 4.434285714285715e-06,
"loss": 0.0001,
"step": 2450
},
{
"epoch": 30.18,
"learning_rate": 4.362857142857143e-06,
"loss": 0.0001,
"step": 2475
},
{
"epoch": 30.48,
"learning_rate": 4.291428571428572e-06,
"loss": 0.0001,
"step": 2500
},
{
"epoch": 30.48,
"eval_loss": 0.24061080813407898,
"eval_runtime": 105.1705,
"eval_samples_per_second": 7.797,
"eval_steps_per_second": 0.247,
"eval_wer": 23.16562889165629,
"step": 2500
},
{
"epoch": 30.79,
"learning_rate": 4.22e-06,
"loss": 0.0001,
"step": 2525
},
{
"epoch": 31.1,
"learning_rate": 4.148571428571429e-06,
"loss": 0.0001,
"step": 2550
},
{
"epoch": 31.4,
"learning_rate": 4.0771428571428574e-06,
"loss": 0.0001,
"step": 2575
},
{
"epoch": 31.7,
"learning_rate": 4.0057142857142864e-06,
"loss": 0.0001,
"step": 2600
},
{
"epoch": 32.01,
"learning_rate": 3.934285714285715e-06,
"loss": 0.0001,
"step": 2625
},
{
"epoch": 32.32,
"learning_rate": 3.862857142857143e-06,
"loss": 0.0001,
"step": 2650
},
{
"epoch": 32.62,
"learning_rate": 3.7914285714285722e-06,
"loss": 0.0001,
"step": 2675
},
{
"epoch": 32.92,
"learning_rate": 3.7200000000000004e-06,
"loss": 0.0001,
"step": 2700
},
{
"epoch": 33.23,
"learning_rate": 3.648571428571429e-06,
"loss": 0.0001,
"step": 2725
},
{
"epoch": 33.53,
"learning_rate": 3.5771428571428576e-06,
"loss": 0.0001,
"step": 2750
},
{
"epoch": 33.84,
"learning_rate": 3.505714285714286e-06,
"loss": 0.0001,
"step": 2775
},
{
"epoch": 34.15,
"learning_rate": 3.4342857142857143e-06,
"loss": 0.0001,
"step": 2800
},
{
"epoch": 34.45,
"learning_rate": 3.3628571428571433e-06,
"loss": 0.0001,
"step": 2825
},
{
"epoch": 34.75,
"learning_rate": 3.2914285714285715e-06,
"loss": 0.0001,
"step": 2850
},
{
"epoch": 35.06,
"learning_rate": 3.2200000000000005e-06,
"loss": 0.0001,
"step": 2875
},
{
"epoch": 35.36,
"learning_rate": 3.1485714285714287e-06,
"loss": 0.0001,
"step": 2900
},
{
"epoch": 35.67,
"learning_rate": 3.0771428571428573e-06,
"loss": 0.0001,
"step": 2925
},
{
"epoch": 35.97,
"learning_rate": 3.005714285714286e-06,
"loss": 0.0001,
"step": 2950
},
{
"epoch": 36.28,
"learning_rate": 2.9342857142857144e-06,
"loss": 0.0001,
"step": 2975
},
{
"epoch": 36.58,
"learning_rate": 2.8628571428571435e-06,
"loss": 0.0001,
"step": 3000
},
{
"epoch": 36.58,
"eval_loss": 0.24610796570777893,
"eval_runtime": 105.0431,
"eval_samples_per_second": 7.806,
"eval_steps_per_second": 0.248,
"eval_wer": 23.153175591531756,
"step": 3000
},
{
"epoch": 36.88,
"learning_rate": 2.7914285714285716e-06,
"loss": 0.0001,
"step": 3025
},
{
"epoch": 37.19,
"learning_rate": 2.7200000000000002e-06,
"loss": 0.0001,
"step": 3050
},
{
"epoch": 37.5,
"learning_rate": 2.648571428571429e-06,
"loss": 0.0001,
"step": 3075
},
{
"epoch": 37.8,
"learning_rate": 2.5771428571428574e-06,
"loss": 0.0001,
"step": 3100
},
{
"epoch": 38.11,
"learning_rate": 2.5057142857142856e-06,
"loss": 0.0001,
"step": 3125
},
{
"epoch": 38.41,
"learning_rate": 2.4342857142857146e-06,
"loss": 0.0001,
"step": 3150
},
{
"epoch": 38.72,
"learning_rate": 2.362857142857143e-06,
"loss": 0.0001,
"step": 3175
},
{
"epoch": 39.02,
"learning_rate": 2.2914285714285718e-06,
"loss": 0.0001,
"step": 3200
},
{
"epoch": 39.33,
"learning_rate": 2.2200000000000003e-06,
"loss": 0.0001,
"step": 3225
},
{
"epoch": 39.63,
"learning_rate": 2.148571428571429e-06,
"loss": 0.0001,
"step": 3250
},
{
"epoch": 39.93,
"learning_rate": 2.077142857142857e-06,
"loss": 0.0001,
"step": 3275
},
{
"epoch": 40.24,
"learning_rate": 2.0057142857142857e-06,
"loss": 0.0001,
"step": 3300
},
{
"epoch": 40.55,
"learning_rate": 1.9342857142857143e-06,
"loss": 0.0001,
"step": 3325
},
{
"epoch": 40.85,
"learning_rate": 1.8628571428571429e-06,
"loss": 0.0001,
"step": 3350
},
{
"epoch": 41.16,
"learning_rate": 1.7914285714285715e-06,
"loss": 0.0001,
"step": 3375
},
{
"epoch": 41.46,
"learning_rate": 1.72e-06,
"loss": 0.0001,
"step": 3400
},
{
"epoch": 41.76,
"learning_rate": 1.6485714285714289e-06,
"loss": 0.0001,
"step": 3425
},
{
"epoch": 42.07,
"learning_rate": 1.5771428571428574e-06,
"loss": 0.0001,
"step": 3450
},
{
"epoch": 42.38,
"learning_rate": 1.5057142857142858e-06,
"loss": 0.0001,
"step": 3475
},
{
"epoch": 42.68,
"learning_rate": 1.4342857142857144e-06,
"loss": 0.0001,
"step": 3500
},
{
"epoch": 42.68,
"eval_loss": 0.24934539198875427,
"eval_runtime": 105.7821,
"eval_samples_per_second": 7.752,
"eval_steps_per_second": 0.246,
"eval_wer": 23.110834371108343,
"step": 3500
},
{
"epoch": 42.98,
"learning_rate": 1.362857142857143e-06,
"loss": 0.0001,
"step": 3525
},
{
"epoch": 43.29,
"learning_rate": 1.2914285714285716e-06,
"loss": 0.0001,
"step": 3550
},
{
"epoch": 43.59,
"learning_rate": 1.2200000000000002e-06,
"loss": 0.0001,
"step": 3575
},
{
"epoch": 43.9,
"learning_rate": 1.1485714285714286e-06,
"loss": 0.0001,
"step": 3600
},
{
"epoch": 44.21,
"learning_rate": 1.0771428571428574e-06,
"loss": 0.0001,
"step": 3625
},
{
"epoch": 44.51,
"learning_rate": 1.0057142857142857e-06,
"loss": 0.0001,
"step": 3650
},
{
"epoch": 44.81,
"learning_rate": 9.342857142857144e-07,
"loss": 0.0001,
"step": 3675
},
{
"epoch": 45.12,
"learning_rate": 8.628571428571429e-07,
"loss": 0.0001,
"step": 3700
},
{
"epoch": 45.42,
"learning_rate": 7.914285714285715e-07,
"loss": 0.0001,
"step": 3725
},
{
"epoch": 45.73,
"learning_rate": 7.2e-07,
"loss": 0.0001,
"step": 3750
},
{
"epoch": 46.04,
"learning_rate": 6.485714285714287e-07,
"loss": 0.0001,
"step": 3775
},
{
"epoch": 46.34,
"learning_rate": 5.771428571428572e-07,
"loss": 0.0001,
"step": 3800
},
{
"epoch": 46.64,
"learning_rate": 5.057142857142858e-07,
"loss": 0.0001,
"step": 3825
},
{
"epoch": 46.95,
"learning_rate": 4.342857142857143e-07,
"loss": 0.0001,
"step": 3850
},
{
"epoch": 47.25,
"learning_rate": 3.6285714285714283e-07,
"loss": 0.0001,
"step": 3875
},
{
"epoch": 47.56,
"learning_rate": 2.914285714285715e-07,
"loss": 0.0001,
"step": 3900
},
{
"epoch": 47.86,
"learning_rate": 2.2e-07,
"loss": 0.0001,
"step": 3925
},
{
"epoch": 48.17,
"learning_rate": 1.4857142857142857e-07,
"loss": 0.0001,
"step": 3950
},
{
"epoch": 48.47,
"learning_rate": 7.714285714285715e-08,
"loss": 0.0,
"step": 3975
},
{
"epoch": 48.78,
"learning_rate": 5.714285714285715e-09,
"loss": 0.0001,
"step": 4000
},
{
"epoch": 48.78,
"eval_loss": 0.2507011592388153,
"eval_runtime": 104.9754,
"eval_samples_per_second": 7.811,
"eval_steps_per_second": 0.248,
"eval_wer": 23.12577833125778,
"step": 4000
}
],
"max_steps": 4000,
"num_train_epochs": 49,
"total_flos": 3.7049748266057466e+19,
"trial_name": null,
"trial_params": null
}