whisper-medium-v2-amet / trainer_state.json
drmeeseeks's picture
End of training
c6a3800
{
"best_metric": 141.1764705882353,
"best_model_checkpoint": "./whisper-medium-v2-amet/checkpoint-1000",
"epoch": 3000.0,
"global_step": 3000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 25.0,
"learning_rate": 4.2000000000000006e-07,
"loss": 2.8559,
"step": 25
},
{
"epoch": 50.0,
"learning_rate": 9.200000000000001e-07,
"loss": 1.9972,
"step": 50
},
{
"epoch": 75.0,
"learning_rate": 1.42e-06,
"loss": 0.8333,
"step": 75
},
{
"epoch": 100.0,
"learning_rate": 1.9200000000000003e-06,
"loss": 0.0194,
"step": 100
},
{
"epoch": 100.0,
"eval_loss": 3.8540303707122803,
"eval_runtime": 172.9892,
"eval_samples_per_second": 0.289,
"eval_steps_per_second": 0.023,
"eval_wer": 147.99465240641712,
"step": 100
},
{
"epoch": 125.0,
"learning_rate": 2.42e-06,
"loss": 0.0002,
"step": 125
},
{
"epoch": 150.0,
"learning_rate": 2.92e-06,
"loss": 0.0001,
"step": 150
},
{
"epoch": 175.0,
"learning_rate": 3.4200000000000007e-06,
"loss": 0.0001,
"step": 175
},
{
"epoch": 200.0,
"learning_rate": 3.920000000000001e-06,
"loss": 0.0001,
"step": 200
},
{
"epoch": 200.0,
"eval_loss": 4.147902011871338,
"eval_runtime": 173.0433,
"eval_samples_per_second": 0.289,
"eval_steps_per_second": 0.023,
"eval_wer": 148.1283422459893,
"step": 200
},
{
"epoch": 225.0,
"learning_rate": 4.42e-06,
"loss": 0.0001,
"step": 225
},
{
"epoch": 250.0,
"learning_rate": 4.92e-06,
"loss": 0.0001,
"step": 250
},
{
"epoch": 275.0,
"learning_rate": 5.420000000000001e-06,
"loss": 0.0001,
"step": 275
},
{
"epoch": 300.0,
"learning_rate": 5.92e-06,
"loss": 0.0001,
"step": 300
},
{
"epoch": 300.0,
"eval_loss": 4.183959007263184,
"eval_runtime": 212.6643,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 150.53475935828877,
"step": 300
},
{
"epoch": 325.0,
"learning_rate": 6.42e-06,
"loss": 0.0001,
"step": 325
},
{
"epoch": 350.0,
"learning_rate": 6.92e-06,
"loss": 0.0001,
"step": 350
},
{
"epoch": 375.0,
"learning_rate": 7.420000000000001e-06,
"loss": 0.0001,
"step": 375
},
{
"epoch": 400.0,
"learning_rate": 7.92e-06,
"loss": 0.0001,
"step": 400
},
{
"epoch": 400.0,
"eval_loss": 4.333920001983643,
"eval_runtime": 205.1394,
"eval_samples_per_second": 0.244,
"eval_steps_per_second": 0.019,
"eval_wer": 177.94117647058823,
"step": 400
},
{
"epoch": 425.0,
"learning_rate": 8.42e-06,
"loss": 0.0001,
"step": 425
},
{
"epoch": 450.0,
"learning_rate": 8.920000000000001e-06,
"loss": 0.0001,
"step": 450
},
{
"epoch": 475.0,
"learning_rate": 9.42e-06,
"loss": 0.0001,
"step": 475
},
{
"epoch": 500.0,
"learning_rate": 9.920000000000002e-06,
"loss": 0.0,
"step": 500
},
{
"epoch": 500.0,
"eval_loss": 4.583085536956787,
"eval_runtime": 212.5309,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 151.06951871657753,
"step": 500
},
{
"epoch": 525.0,
"learning_rate": 9.916000000000001e-06,
"loss": 0.0,
"step": 525
},
{
"epoch": 550.0,
"learning_rate": 9.816e-06,
"loss": 0.0,
"step": 550
},
{
"epoch": 575.0,
"learning_rate": 9.716000000000002e-06,
"loss": 0.0,
"step": 575
},
{
"epoch": 600.0,
"learning_rate": 9.616e-06,
"loss": 0.0,
"step": 600
},
{
"epoch": 600.0,
"eval_loss": 4.931729793548584,
"eval_runtime": 212.8362,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 164.0374331550802,
"step": 600
},
{
"epoch": 625.0,
"learning_rate": 9.516e-06,
"loss": 0.0,
"step": 625
},
{
"epoch": 650.0,
"learning_rate": 9.416000000000001e-06,
"loss": 0.0,
"step": 650
},
{
"epoch": 675.0,
"learning_rate": 9.316e-06,
"loss": 0.0,
"step": 675
},
{
"epoch": 700.0,
"learning_rate": 9.216000000000001e-06,
"loss": 0.0,
"step": 700
},
{
"epoch": 700.0,
"eval_loss": 5.303104877471924,
"eval_runtime": 204.8283,
"eval_samples_per_second": 0.244,
"eval_steps_per_second": 0.02,
"eval_wer": 141.0427807486631,
"step": 700
},
{
"epoch": 725.0,
"learning_rate": 9.116e-06,
"loss": 0.0,
"step": 725
},
{
"epoch": 750.0,
"learning_rate": 9.016e-06,
"loss": 0.0,
"step": 750
},
{
"epoch": 775.0,
"learning_rate": 8.916e-06,
"loss": 0.0,
"step": 775
},
{
"epoch": 800.0,
"learning_rate": 8.816000000000002e-06,
"loss": 0.0,
"step": 800
},
{
"epoch": 800.0,
"eval_loss": 5.658377170562744,
"eval_runtime": 204.6203,
"eval_samples_per_second": 0.244,
"eval_steps_per_second": 0.02,
"eval_wer": 122.32620320855614,
"step": 800
},
{
"epoch": 825.0,
"learning_rate": 8.716000000000001e-06,
"loss": 0.0,
"step": 825
},
{
"epoch": 850.0,
"learning_rate": 8.616000000000002e-06,
"loss": 0.0,
"step": 850
},
{
"epoch": 875.0,
"learning_rate": 8.516000000000001e-06,
"loss": 0.0,
"step": 875
},
{
"epoch": 900.0,
"learning_rate": 8.416e-06,
"loss": 0.0,
"step": 900
},
{
"epoch": 900.0,
"eval_loss": 5.971085071563721,
"eval_runtime": 213.3134,
"eval_samples_per_second": 0.234,
"eval_steps_per_second": 0.019,
"eval_wer": 157.4866310160428,
"step": 900
},
{
"epoch": 925.0,
"learning_rate": 8.316000000000001e-06,
"loss": 0.0,
"step": 925
},
{
"epoch": 950.0,
"learning_rate": 8.216e-06,
"loss": 0.0,
"step": 950
},
{
"epoch": 975.0,
"learning_rate": 8.116e-06,
"loss": 0.0,
"step": 975
},
{
"epoch": 1000.0,
"learning_rate": 8.016e-06,
"loss": 0.0,
"step": 1000
},
{
"epoch": 1000.0,
"eval_loss": 6.246512413024902,
"eval_runtime": 213.4328,
"eval_samples_per_second": 0.234,
"eval_steps_per_second": 0.019,
"eval_wer": 141.1764705882353,
"step": 1000
},
{
"epoch": 1025.0,
"learning_rate": 7.916e-06,
"loss": 0.0,
"step": 1025
},
{
"epoch": 1050.0,
"learning_rate": 7.816000000000001e-06,
"loss": 0.0,
"step": 1050
},
{
"epoch": 1075.0,
"learning_rate": 7.716e-06,
"loss": 0.0,
"step": 1075
},
{
"epoch": 1100.0,
"learning_rate": 7.616000000000001e-06,
"loss": 0.0,
"step": 1100
},
{
"epoch": 1100.0,
"eval_loss": 6.483173847198486,
"eval_runtime": 213.8254,
"eval_samples_per_second": 0.234,
"eval_steps_per_second": 0.019,
"eval_wer": 169.6524064171123,
"step": 1100
},
{
"epoch": 1125.0,
"learning_rate": 7.516000000000001e-06,
"loss": 0.0,
"step": 1125
},
{
"epoch": 1150.0,
"learning_rate": 7.416000000000001e-06,
"loss": 0.0,
"step": 1150
},
{
"epoch": 1175.0,
"learning_rate": 7.316000000000001e-06,
"loss": 0.0,
"step": 1175
},
{
"epoch": 1200.0,
"learning_rate": 7.216000000000001e-06,
"loss": 0.0,
"step": 1200
},
{
"epoch": 1200.0,
"eval_loss": 6.6890459060668945,
"eval_runtime": 209.6765,
"eval_samples_per_second": 0.238,
"eval_steps_per_second": 0.019,
"eval_wer": 155.0802139037433,
"step": 1200
},
{
"epoch": 1225.0,
"learning_rate": 7.116000000000001e-06,
"loss": 0.0,
"step": 1225
},
{
"epoch": 1250.0,
"learning_rate": 7.016e-06,
"loss": 0.0,
"step": 1250
},
{
"epoch": 1275.0,
"learning_rate": 6.916e-06,
"loss": 0.0,
"step": 1275
},
{
"epoch": 1300.0,
"learning_rate": 6.8160000000000005e-06,
"loss": 0.0,
"step": 1300
},
{
"epoch": 1300.0,
"eval_loss": 6.867943286895752,
"eval_runtime": 209.9316,
"eval_samples_per_second": 0.238,
"eval_steps_per_second": 0.019,
"eval_wer": 159.75935828877004,
"step": 1300
},
{
"epoch": 1325.0,
"learning_rate": 6.716000000000001e-06,
"loss": 0.0,
"step": 1325
},
{
"epoch": 1350.0,
"learning_rate": 6.616e-06,
"loss": 0.0,
"step": 1350
},
{
"epoch": 1375.0,
"learning_rate": 6.516e-06,
"loss": 0.0,
"step": 1375
},
{
"epoch": 1400.0,
"learning_rate": 6.416e-06,
"loss": 0.0,
"step": 1400
},
{
"epoch": 1400.0,
"eval_loss": 7.025048732757568,
"eval_runtime": 212.7993,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 155.0802139037433,
"step": 1400
},
{
"epoch": 1425.0,
"learning_rate": 6.316000000000001e-06,
"loss": 0.0,
"step": 1425
},
{
"epoch": 1450.0,
"learning_rate": 6.216000000000001e-06,
"loss": 0.0,
"step": 1450
},
{
"epoch": 1475.0,
"learning_rate": 6.116000000000001e-06,
"loss": 0.0,
"step": 1475
},
{
"epoch": 1500.0,
"learning_rate": 6.0160000000000005e-06,
"loss": 0.0,
"step": 1500
},
{
"epoch": 1500.0,
"eval_loss": 7.161492347717285,
"eval_runtime": 212.8178,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 146.2566844919786,
"step": 1500
},
{
"epoch": 1525.0,
"learning_rate": 5.916000000000001e-06,
"loss": 0.0,
"step": 1525
},
{
"epoch": 1550.0,
"learning_rate": 5.816000000000001e-06,
"loss": 0.0,
"step": 1550
},
{
"epoch": 1575.0,
"learning_rate": 5.716000000000001e-06,
"loss": 0.0,
"step": 1575
},
{
"epoch": 1600.0,
"learning_rate": 5.616e-06,
"loss": 0.0,
"step": 1600
},
{
"epoch": 1600.0,
"eval_loss": 7.2876811027526855,
"eval_runtime": 212.9353,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 143.04812834224597,
"step": 1600
},
{
"epoch": 1625.0,
"learning_rate": 5.516e-06,
"loss": 0.0,
"step": 1625
},
{
"epoch": 1650.0,
"learning_rate": 5.416e-06,
"loss": 0.0,
"step": 1650
},
{
"epoch": 1675.0,
"learning_rate": 5.3160000000000004e-06,
"loss": 0.0,
"step": 1675
},
{
"epoch": 1700.0,
"learning_rate": 5.216e-06,
"loss": 0.0,
"step": 1700
},
{
"epoch": 1700.0,
"eval_loss": 7.3986992835998535,
"eval_runtime": 212.9015,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 148.52941176470588,
"step": 1700
},
{
"epoch": 1725.0,
"learning_rate": 5.116000000000001e-06,
"loss": 0.0,
"step": 1725
},
{
"epoch": 1750.0,
"learning_rate": 5.016000000000001e-06,
"loss": 0.0,
"step": 1750
},
{
"epoch": 1775.0,
"learning_rate": 4.916e-06,
"loss": 0.0,
"step": 1775
},
{
"epoch": 1800.0,
"learning_rate": 4.816e-06,
"loss": 0.0,
"step": 1800
},
{
"epoch": 1800.0,
"eval_loss": 7.501037120819092,
"eval_runtime": 213.1608,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 142.5133689839572,
"step": 1800
},
{
"epoch": 1825.0,
"learning_rate": 4.716e-06,
"loss": 0.0,
"step": 1825
},
{
"epoch": 1850.0,
"learning_rate": 4.616e-06,
"loss": 0.0,
"step": 1850
},
{
"epoch": 1875.0,
"learning_rate": 4.5160000000000005e-06,
"loss": 0.0,
"step": 1875
},
{
"epoch": 1900.0,
"learning_rate": 4.416000000000001e-06,
"loss": 0.0,
"step": 1900
},
{
"epoch": 1900.0,
"eval_loss": 7.584920883178711,
"eval_runtime": 213.8853,
"eval_samples_per_second": 0.234,
"eval_steps_per_second": 0.019,
"eval_wer": 136.76470588235296,
"step": 1900
},
{
"epoch": 1925.0,
"learning_rate": 4.316e-06,
"loss": 0.0,
"step": 1925
},
{
"epoch": 1950.0,
"learning_rate": 4.216e-06,
"loss": 0.0,
"step": 1950
},
{
"epoch": 1975.0,
"learning_rate": 4.116000000000001e-06,
"loss": 0.0,
"step": 1975
},
{
"epoch": 2000.0,
"learning_rate": 4.016e-06,
"loss": 0.0,
"step": 2000
},
{
"epoch": 2000.0,
"eval_loss": 7.668895244598389,
"eval_runtime": 212.6893,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 148.26203208556151,
"step": 2000
},
{
"epoch": 2025.0,
"learning_rate": 3.916e-06,
"loss": 0.0,
"step": 2025
},
{
"epoch": 2050.0,
"learning_rate": 3.816e-06,
"loss": 0.0,
"step": 2050
},
{
"epoch": 2075.0,
"learning_rate": 3.716e-06,
"loss": 0.0,
"step": 2075
},
{
"epoch": 2100.0,
"learning_rate": 3.616e-06,
"loss": 0.0,
"step": 2100
},
{
"epoch": 2100.0,
"eval_loss": 7.695452690124512,
"eval_runtime": 213.8269,
"eval_samples_per_second": 0.234,
"eval_steps_per_second": 0.019,
"eval_wer": 165.37433155080214,
"step": 2100
},
{
"epoch": 2125.0,
"learning_rate": 3.5160000000000007e-06,
"loss": 0.0,
"step": 2125
},
{
"epoch": 2150.0,
"learning_rate": 3.4160000000000004e-06,
"loss": 0.0,
"step": 2150
},
{
"epoch": 2175.0,
"learning_rate": 3.3160000000000005e-06,
"loss": 0.0,
"step": 2175
},
{
"epoch": 2200.0,
"learning_rate": 3.216e-06,
"loss": 0.0,
"step": 2200
},
{
"epoch": 2200.0,
"eval_loss": 7.724720001220703,
"eval_runtime": 213.0213,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 162.96791443850267,
"step": 2200
},
{
"epoch": 2225.0,
"learning_rate": 3.1160000000000003e-06,
"loss": 0.0,
"step": 2225
},
{
"epoch": 2250.0,
"learning_rate": 3.016e-06,
"loss": 0.0,
"step": 2250
},
{
"epoch": 2275.0,
"learning_rate": 2.9160000000000005e-06,
"loss": 0.0,
"step": 2275
},
{
"epoch": 2300.0,
"learning_rate": 2.8160000000000002e-06,
"loss": 0.0,
"step": 2300
},
{
"epoch": 2300.0,
"eval_loss": 7.755704402923584,
"eval_runtime": 212.9136,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 161.63101604278074,
"step": 2300
},
{
"epoch": 2325.0,
"learning_rate": 2.7160000000000003e-06,
"loss": 0.0,
"step": 2325
},
{
"epoch": 2350.0,
"learning_rate": 2.616e-06,
"loss": 0.0,
"step": 2350
},
{
"epoch": 2375.0,
"learning_rate": 2.516e-06,
"loss": 0.0,
"step": 2375
},
{
"epoch": 2400.0,
"learning_rate": 2.4160000000000002e-06,
"loss": 0.0,
"step": 2400
},
{
"epoch": 2400.0,
"eval_loss": 7.784246444702148,
"eval_runtime": 213.1347,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 162.29946524064172,
"step": 2400
},
{
"epoch": 2425.0,
"learning_rate": 2.3160000000000004e-06,
"loss": 0.0,
"step": 2425
},
{
"epoch": 2450.0,
"learning_rate": 2.216e-06,
"loss": 0.0,
"step": 2450
},
{
"epoch": 2475.0,
"learning_rate": 2.116e-06,
"loss": 0.0,
"step": 2475
},
{
"epoch": 2500.0,
"learning_rate": 2.0160000000000003e-06,
"loss": 0.0,
"step": 2500
},
{
"epoch": 2500.0,
"eval_loss": 7.807443141937256,
"eval_runtime": 212.982,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 150.93582887700535,
"step": 2500
},
{
"epoch": 2525.0,
"learning_rate": 1.916e-06,
"loss": 0.0,
"step": 2525
},
{
"epoch": 2550.0,
"learning_rate": 1.8160000000000003e-06,
"loss": 0.0,
"step": 2550
},
{
"epoch": 2575.0,
"learning_rate": 1.7160000000000002e-06,
"loss": 0.0,
"step": 2575
},
{
"epoch": 2600.0,
"learning_rate": 1.616e-06,
"loss": 0.0,
"step": 2600
},
{
"epoch": 2600.0,
"eval_loss": 7.8286895751953125,
"eval_runtime": 213.1069,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 154.81283422459893,
"step": 2600
},
{
"epoch": 2625.0,
"learning_rate": 1.5160000000000002e-06,
"loss": 0.0,
"step": 2625
},
{
"epoch": 2650.0,
"learning_rate": 1.416e-06,
"loss": 0.0,
"step": 2650
},
{
"epoch": 2675.0,
"learning_rate": 1.316e-06,
"loss": 0.0,
"step": 2675
},
{
"epoch": 2700.0,
"learning_rate": 1.216e-06,
"loss": 0.0,
"step": 2700
},
{
"epoch": 2700.0,
"eval_loss": 7.843414783477783,
"eval_runtime": 212.8223,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 155.4812834224599,
"step": 2700
},
{
"epoch": 2725.0,
"learning_rate": 1.1160000000000002e-06,
"loss": 0.0,
"step": 2725
},
{
"epoch": 2750.0,
"learning_rate": 1.016e-06,
"loss": 0.0,
"step": 2750
},
{
"epoch": 2775.0,
"learning_rate": 9.160000000000001e-07,
"loss": 0.0,
"step": 2775
},
{
"epoch": 2800.0,
"learning_rate": 8.160000000000001e-07,
"loss": 0.0,
"step": 2800
},
{
"epoch": 2800.0,
"eval_loss": 7.856691837310791,
"eval_runtime": 212.6712,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 154.41176470588235,
"step": 2800
},
{
"epoch": 2825.0,
"learning_rate": 7.16e-07,
"loss": 0.0,
"step": 2825
},
{
"epoch": 2850.0,
"learning_rate": 6.160000000000001e-07,
"loss": 0.0,
"step": 2850
},
{
"epoch": 2875.0,
"learning_rate": 5.16e-07,
"loss": 0.0,
"step": 2875
},
{
"epoch": 2900.0,
"learning_rate": 4.16e-07,
"loss": 0.0,
"step": 2900
},
{
"epoch": 2900.0,
"eval_loss": 7.863497257232666,
"eval_runtime": 212.9291,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 154.41176470588235,
"step": 2900
},
{
"epoch": 2925.0,
"learning_rate": 3.160000000000001e-07,
"loss": 0.0,
"step": 2925
},
{
"epoch": 2950.0,
"learning_rate": 2.1600000000000003e-07,
"loss": 0.0,
"step": 2950
},
{
"epoch": 2975.0,
"learning_rate": 1.16e-07,
"loss": 0.0,
"step": 2975
},
{
"epoch": 3000.0,
"learning_rate": 1.6e-08,
"loss": 0.0,
"step": 3000
},
{
"epoch": 3000.0,
"eval_loss": 7.86700439453125,
"eval_runtime": 212.9038,
"eval_samples_per_second": 0.235,
"eval_steps_per_second": 0.019,
"eval_wer": 154.41176470588235,
"step": 3000
},
{
"epoch": 3000.0,
"step": 3000,
"total_flos": 3.06181472256e+18,
"train_loss": 0.047578999360693465,
"train_runtime": 7446.2425,
"train_samples_per_second": 12.892,
"train_steps_per_second": 0.403
}
],
"max_steps": 3000,
"num_train_epochs": 3000,
"total_flos": 3.06181472256e+18,
"trial_name": null,
"trial_params": null
}