gary109's picture
End of training
65ea447
{
"best_metric": 1.076346755027771,
"best_model_checkpoint": "ai-light-dance_stepmania_ft_wav2vec2-large-xlsr-53-v1/checkpoint-1350",
"epoch": 9.997091815538013,
"global_step": 1500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.07,
"learning_rate": 4.000000000000001e-06,
"loss": 1.2075,
"step": 10
},
{
"epoch": 0.13,
"learning_rate": 8.000000000000001e-06,
"loss": 1.1818,
"step": 20
},
{
"epoch": 0.2,
"learning_rate": 1.16e-05,
"loss": 1.1731,
"step": 30
},
{
"epoch": 0.27,
"learning_rate": 1.5600000000000003e-05,
"loss": 1.1906,
"step": 40
},
{
"epoch": 0.33,
"learning_rate": 1.9600000000000002e-05,
"loss": 1.1936,
"step": 50
},
{
"epoch": 0.4,
"learning_rate": 2.36e-05,
"loss": 1.1728,
"step": 60
},
{
"epoch": 0.47,
"learning_rate": 2.76e-05,
"loss": 1.1748,
"step": 70
},
{
"epoch": 0.53,
"learning_rate": 3.16e-05,
"loss": 1.1953,
"step": 80
},
{
"epoch": 0.6,
"learning_rate": 3.5600000000000005e-05,
"loss": 1.1795,
"step": 90
},
{
"epoch": 0.66,
"learning_rate": 3.96e-05,
"loss": 1.1763,
"step": 100
},
{
"epoch": 0.73,
"learning_rate": 3.9742857142857146e-05,
"loss": 1.1647,
"step": 110
},
{
"epoch": 0.8,
"learning_rate": 3.945714285714286e-05,
"loss": 1.203,
"step": 120
},
{
"epoch": 0.86,
"learning_rate": 3.917142857142858e-05,
"loss": 1.164,
"step": 130
},
{
"epoch": 0.93,
"learning_rate": 3.8885714285714294e-05,
"loss": 1.185,
"step": 140
},
{
"epoch": 1.0,
"learning_rate": 3.86e-05,
"loss": 1.1632,
"step": 150
},
{
"epoch": 1.0,
"eval_loss": 1.2007131576538086,
"eval_runtime": 347.7999,
"eval_samples_per_second": 7.688,
"eval_steps_per_second": 0.771,
"eval_wer": 0.9874654499394933,
"step": 150
},
{
"epoch": 1.07,
"learning_rate": 3.831428571428572e-05,
"loss": 1.239,
"step": 160
},
{
"epoch": 1.13,
"learning_rate": 3.8028571428571435e-05,
"loss": 1.1681,
"step": 170
},
{
"epoch": 1.2,
"learning_rate": 3.7742857142857145e-05,
"loss": 1.1626,
"step": 180
},
{
"epoch": 1.27,
"learning_rate": 3.745714285714286e-05,
"loss": 1.1778,
"step": 190
},
{
"epoch": 1.33,
"learning_rate": 3.717142857142858e-05,
"loss": 1.1747,
"step": 200
},
{
"epoch": 1.4,
"learning_rate": 3.6885714285714286e-05,
"loss": 1.1547,
"step": 210
},
{
"epoch": 1.47,
"learning_rate": 3.66e-05,
"loss": 1.1504,
"step": 220
},
{
"epoch": 1.53,
"learning_rate": 3.631428571428572e-05,
"loss": 1.1893,
"step": 230
},
{
"epoch": 1.6,
"learning_rate": 3.602857142857143e-05,
"loss": 1.1549,
"step": 240
},
{
"epoch": 1.66,
"learning_rate": 3.574285714285714e-05,
"loss": 1.1574,
"step": 250
},
{
"epoch": 1.73,
"learning_rate": 3.545714285714286e-05,
"loss": 1.1671,
"step": 260
},
{
"epoch": 1.8,
"learning_rate": 3.5171428571428575e-05,
"loss": 1.1832,
"step": 270
},
{
"epoch": 1.86,
"learning_rate": 3.4885714285714285e-05,
"loss": 1.1662,
"step": 280
},
{
"epoch": 1.93,
"learning_rate": 3.46e-05,
"loss": 1.175,
"step": 290
},
{
"epoch": 2.0,
"learning_rate": 3.431428571428572e-05,
"loss": 1.1615,
"step": 300
},
{
"epoch": 2.0,
"eval_loss": 1.1911734342575073,
"eval_runtime": 346.7276,
"eval_samples_per_second": 7.712,
"eval_steps_per_second": 0.773,
"eval_wer": 0.9874701586359848,
"step": 300
},
{
"epoch": 2.07,
"learning_rate": 3.402857142857143e-05,
"loss": 1.2418,
"step": 310
},
{
"epoch": 2.13,
"learning_rate": 3.374285714285714e-05,
"loss": 1.1563,
"step": 320
},
{
"epoch": 2.2,
"learning_rate": 3.345714285714286e-05,
"loss": 1.1407,
"step": 330
},
{
"epoch": 2.27,
"learning_rate": 3.3171428571428574e-05,
"loss": 1.1596,
"step": 340
},
{
"epoch": 2.33,
"learning_rate": 3.288571428571429e-05,
"loss": 1.1629,
"step": 350
},
{
"epoch": 2.4,
"learning_rate": 3.26e-05,
"loss": 1.1473,
"step": 360
},
{
"epoch": 2.47,
"learning_rate": 3.2314285714285716e-05,
"loss": 1.1506,
"step": 370
},
{
"epoch": 2.53,
"learning_rate": 3.202857142857143e-05,
"loss": 1.1651,
"step": 380
},
{
"epoch": 2.6,
"learning_rate": 3.174285714285715e-05,
"loss": 1.1321,
"step": 390
},
{
"epoch": 2.66,
"learning_rate": 3.1457142857142864e-05,
"loss": 1.1463,
"step": 400
},
{
"epoch": 2.73,
"learning_rate": 3.117142857142857e-05,
"loss": 1.1402,
"step": 410
},
{
"epoch": 2.8,
"learning_rate": 3.088571428571429e-05,
"loss": 1.1679,
"step": 420
},
{
"epoch": 2.86,
"learning_rate": 3.0600000000000005e-05,
"loss": 1.1486,
"step": 430
},
{
"epoch": 2.93,
"learning_rate": 3.0314285714285718e-05,
"loss": 1.1389,
"step": 440
},
{
"epoch": 3.0,
"learning_rate": 3.002857142857143e-05,
"loss": 1.1487,
"step": 450
},
{
"epoch": 3.0,
"eval_loss": 1.1941583156585693,
"eval_runtime": 346.4981,
"eval_samples_per_second": 7.717,
"eval_steps_per_second": 0.773,
"eval_wer": 0.9874701586359848,
"step": 450
},
{
"epoch": 3.07,
"learning_rate": 2.9742857142857143e-05,
"loss": 1.2107,
"step": 460
},
{
"epoch": 3.13,
"learning_rate": 2.945714285714286e-05,
"loss": 1.1357,
"step": 470
},
{
"epoch": 3.2,
"learning_rate": 2.9171428571428575e-05,
"loss": 1.1376,
"step": 480
},
{
"epoch": 3.27,
"learning_rate": 2.888571428571429e-05,
"loss": 1.1519,
"step": 490
},
{
"epoch": 3.33,
"learning_rate": 2.86e-05,
"loss": 1.1355,
"step": 500
},
{
"epoch": 3.4,
"learning_rate": 2.8314285714285717e-05,
"loss": 1.133,
"step": 510
},
{
"epoch": 3.47,
"learning_rate": 2.8028571428571433e-05,
"loss": 1.1274,
"step": 520
},
{
"epoch": 3.53,
"learning_rate": 2.7742857142857145e-05,
"loss": 1.1595,
"step": 530
},
{
"epoch": 3.6,
"learning_rate": 2.7457142857142858e-05,
"loss": 1.1319,
"step": 540
},
{
"epoch": 3.66,
"learning_rate": 2.7171428571428574e-05,
"loss": 1.1248,
"step": 550
},
{
"epoch": 3.73,
"learning_rate": 2.6885714285714287e-05,
"loss": 1.1312,
"step": 560
},
{
"epoch": 3.8,
"learning_rate": 2.6600000000000003e-05,
"loss": 1.1549,
"step": 570
},
{
"epoch": 3.86,
"learning_rate": 2.6314285714285715e-05,
"loss": 1.1301,
"step": 580
},
{
"epoch": 3.93,
"learning_rate": 2.6028571428571428e-05,
"loss": 1.1261,
"step": 590
},
{
"epoch": 4.0,
"learning_rate": 2.5742857142857144e-05,
"loss": 1.1207,
"step": 600
},
{
"epoch": 4.0,
"eval_loss": 1.1753019094467163,
"eval_runtime": 353.8634,
"eval_samples_per_second": 7.557,
"eval_steps_per_second": 0.757,
"eval_wer": 0.9874701586359848,
"step": 600
},
{
"epoch": 4.07,
"learning_rate": 2.545714285714286e-05,
"loss": 1.185,
"step": 610
},
{
"epoch": 4.13,
"learning_rate": 2.5171428571428576e-05,
"loss": 1.1175,
"step": 620
},
{
"epoch": 4.2,
"learning_rate": 2.491428571428572e-05,
"loss": 1.1091,
"step": 630
},
{
"epoch": 4.27,
"learning_rate": 2.462857142857143e-05,
"loss": 1.1214,
"step": 640
},
{
"epoch": 4.33,
"learning_rate": 2.4342857142857144e-05,
"loss": 1.1384,
"step": 650
},
{
"epoch": 4.4,
"learning_rate": 2.405714285714286e-05,
"loss": 1.1156,
"step": 660
},
{
"epoch": 4.47,
"learning_rate": 2.3771428571428573e-05,
"loss": 1.104,
"step": 670
},
{
"epoch": 4.53,
"learning_rate": 2.348571428571429e-05,
"loss": 1.1295,
"step": 680
},
{
"epoch": 4.6,
"learning_rate": 2.32e-05,
"loss": 1.1031,
"step": 690
},
{
"epoch": 4.66,
"learning_rate": 2.2914285714285714e-05,
"loss": 1.1016,
"step": 700
},
{
"epoch": 4.73,
"learning_rate": 2.262857142857143e-05,
"loss": 1.0909,
"step": 710
},
{
"epoch": 4.8,
"learning_rate": 2.2342857142857146e-05,
"loss": 1.1106,
"step": 720
},
{
"epoch": 4.86,
"learning_rate": 2.2057142857142856e-05,
"loss": 1.0826,
"step": 730
},
{
"epoch": 4.93,
"learning_rate": 2.177142857142857e-05,
"loss": 1.0738,
"step": 740
},
{
"epoch": 5.0,
"learning_rate": 2.1485714285714288e-05,
"loss": 1.0638,
"step": 750
},
{
"epoch": 5.0,
"eval_loss": 1.134511947631836,
"eval_runtime": 349.9878,
"eval_samples_per_second": 7.64,
"eval_steps_per_second": 0.766,
"eval_wer": 0.8213991420754992,
"step": 750
},
{
"epoch": 5.07,
"learning_rate": 2.1200000000000004e-05,
"loss": 1.1294,
"step": 760
},
{
"epoch": 5.13,
"learning_rate": 2.091428571428572e-05,
"loss": 1.0545,
"step": 770
},
{
"epoch": 5.2,
"learning_rate": 2.062857142857143e-05,
"loss": 1.0526,
"step": 780
},
{
"epoch": 5.27,
"learning_rate": 2.0342857142857145e-05,
"loss": 1.0592,
"step": 790
},
{
"epoch": 5.33,
"learning_rate": 2.005714285714286e-05,
"loss": 1.0525,
"step": 800
},
{
"epoch": 5.4,
"learning_rate": 1.9771428571428574e-05,
"loss": 1.0453,
"step": 810
},
{
"epoch": 5.47,
"learning_rate": 1.9485714285714286e-05,
"loss": 1.0356,
"step": 820
},
{
"epoch": 5.53,
"learning_rate": 1.9200000000000003e-05,
"loss": 1.0539,
"step": 830
},
{
"epoch": 5.6,
"learning_rate": 1.8914285714285715e-05,
"loss": 1.0441,
"step": 840
},
{
"epoch": 5.66,
"learning_rate": 1.8628571428571428e-05,
"loss": 1.0297,
"step": 850
},
{
"epoch": 5.73,
"learning_rate": 1.8342857142857144e-05,
"loss": 1.0192,
"step": 860
},
{
"epoch": 5.8,
"learning_rate": 1.8057142857142857e-05,
"loss": 1.0391,
"step": 870
},
{
"epoch": 5.86,
"learning_rate": 1.7771428571428573e-05,
"loss": 1.017,
"step": 880
},
{
"epoch": 5.93,
"learning_rate": 1.748571428571429e-05,
"loss": 1.0152,
"step": 890
},
{
"epoch": 6.0,
"learning_rate": 1.72e-05,
"loss": 1.0174,
"step": 900
},
{
"epoch": 6.0,
"eval_loss": 1.154058814048767,
"eval_runtime": 347.8027,
"eval_samples_per_second": 7.688,
"eval_steps_per_second": 0.771,
"eval_wer": 0.7665004496805149,
"step": 900
},
{
"epoch": 6.07,
"learning_rate": 1.6914285714285717e-05,
"loss": 1.0812,
"step": 910
},
{
"epoch": 6.13,
"learning_rate": 1.662857142857143e-05,
"loss": 1.0024,
"step": 920
},
{
"epoch": 6.2,
"learning_rate": 1.6342857142857146e-05,
"loss": 1.006,
"step": 930
},
{
"epoch": 6.27,
"learning_rate": 1.605714285714286e-05,
"loss": 0.9978,
"step": 940
},
{
"epoch": 6.33,
"learning_rate": 1.577142857142857e-05,
"loss": 1.0089,
"step": 950
},
{
"epoch": 6.4,
"learning_rate": 1.5485714285714287e-05,
"loss": 0.9975,
"step": 960
},
{
"epoch": 6.47,
"learning_rate": 1.5200000000000002e-05,
"loss": 0.993,
"step": 970
},
{
"epoch": 6.53,
"learning_rate": 1.4914285714285715e-05,
"loss": 1.0142,
"step": 980
},
{
"epoch": 6.6,
"learning_rate": 1.462857142857143e-05,
"loss": 0.9909,
"step": 990
},
{
"epoch": 6.66,
"learning_rate": 1.4342857142857145e-05,
"loss": 0.9961,
"step": 1000
},
{
"epoch": 6.73,
"learning_rate": 1.4057142857142858e-05,
"loss": 0.9873,
"step": 1010
},
{
"epoch": 6.8,
"learning_rate": 1.3771428571428574e-05,
"loss": 1.0009,
"step": 1020
},
{
"epoch": 6.86,
"learning_rate": 1.3485714285714286e-05,
"loss": 0.9802,
"step": 1030
},
{
"epoch": 6.93,
"learning_rate": 1.3200000000000002e-05,
"loss": 0.9912,
"step": 1040
},
{
"epoch": 7.0,
"learning_rate": 1.2914285714285715e-05,
"loss": 0.9946,
"step": 1050
},
{
"epoch": 7.0,
"eval_loss": 1.079897403717041,
"eval_runtime": 353.1403,
"eval_samples_per_second": 7.572,
"eval_steps_per_second": 0.759,
"eval_wer": 0.7716470549457793,
"step": 1050
},
{
"epoch": 7.07,
"learning_rate": 1.262857142857143e-05,
"loss": 1.0446,
"step": 1060
},
{
"epoch": 7.13,
"learning_rate": 1.2342857142857144e-05,
"loss": 0.9733,
"step": 1070
},
{
"epoch": 7.2,
"learning_rate": 1.2057142857142858e-05,
"loss": 0.9724,
"step": 1080
},
{
"epoch": 7.27,
"learning_rate": 1.177142857142857e-05,
"loss": 0.9857,
"step": 1090
},
{
"epoch": 7.33,
"learning_rate": 1.1485714285714287e-05,
"loss": 0.99,
"step": 1100
},
{
"epoch": 7.4,
"learning_rate": 1.1200000000000001e-05,
"loss": 0.9732,
"step": 1110
},
{
"epoch": 7.47,
"learning_rate": 1.0914285714285716e-05,
"loss": 0.9712,
"step": 1120
},
{
"epoch": 7.53,
"learning_rate": 1.062857142857143e-05,
"loss": 0.9932,
"step": 1130
},
{
"epoch": 7.6,
"learning_rate": 1.0342857142857143e-05,
"loss": 0.9754,
"step": 1140
},
{
"epoch": 7.66,
"learning_rate": 1.0057142857142859e-05,
"loss": 0.9697,
"step": 1150
},
{
"epoch": 7.73,
"learning_rate": 9.771428571428571e-06,
"loss": 0.9716,
"step": 1160
},
{
"epoch": 7.8,
"learning_rate": 9.485714285714287e-06,
"loss": 0.9877,
"step": 1170
},
{
"epoch": 7.86,
"learning_rate": 9.200000000000002e-06,
"loss": 0.9635,
"step": 1180
},
{
"epoch": 7.93,
"learning_rate": 8.914285714285716e-06,
"loss": 0.9608,
"step": 1190
},
{
"epoch": 8.0,
"learning_rate": 8.628571428571429e-06,
"loss": 0.9694,
"step": 1200
},
{
"epoch": 8.0,
"eval_loss": 1.0847891569137573,
"eval_runtime": 351.0567,
"eval_samples_per_second": 7.617,
"eval_steps_per_second": 0.763,
"eval_wer": 0.7417656670104015,
"step": 1200
},
{
"epoch": 8.07,
"learning_rate": 8.342857142857143e-06,
"loss": 1.0234,
"step": 1210
},
{
"epoch": 8.13,
"learning_rate": 8.057142857142857e-06,
"loss": 0.9587,
"step": 1220
},
{
"epoch": 8.2,
"learning_rate": 7.771428571428572e-06,
"loss": 0.9652,
"step": 1230
},
{
"epoch": 8.27,
"learning_rate": 7.485714285714286e-06,
"loss": 0.9725,
"step": 1240
},
{
"epoch": 8.33,
"learning_rate": 7.2000000000000005e-06,
"loss": 0.9743,
"step": 1250
},
{
"epoch": 8.4,
"learning_rate": 6.914285714285715e-06,
"loss": 0.9559,
"step": 1260
},
{
"epoch": 8.47,
"learning_rate": 6.628571428571428e-06,
"loss": 0.9502,
"step": 1270
},
{
"epoch": 8.53,
"learning_rate": 6.342857142857143e-06,
"loss": 0.9843,
"step": 1280
},
{
"epoch": 8.6,
"learning_rate": 6.057142857142858e-06,
"loss": 0.9639,
"step": 1290
},
{
"epoch": 8.66,
"learning_rate": 5.771428571428572e-06,
"loss": 0.9642,
"step": 1300
},
{
"epoch": 8.73,
"learning_rate": 5.485714285714287e-06,
"loss": 0.9511,
"step": 1310
},
{
"epoch": 8.8,
"learning_rate": 5.2e-06,
"loss": 0.9808,
"step": 1320
},
{
"epoch": 8.86,
"learning_rate": 4.9142857142857145e-06,
"loss": 0.9569,
"step": 1330
},
{
"epoch": 8.93,
"learning_rate": 4.628571428571429e-06,
"loss": 0.9536,
"step": 1340
},
{
"epoch": 9.0,
"learning_rate": 4.342857142857143e-06,
"loss": 0.9566,
"step": 1350
},
{
"epoch": 9.0,
"eval_loss": 1.076346755027771,
"eval_runtime": 348.8389,
"eval_samples_per_second": 7.665,
"eval_steps_per_second": 0.768,
"eval_wer": 0.7344389352695493,
"step": 1350
},
{
"epoch": 9.07,
"learning_rate": 4.057142857142858e-06,
"loss": 1.0231,
"step": 1360
},
{
"epoch": 9.13,
"learning_rate": 3.771428571428572e-06,
"loss": 0.9566,
"step": 1370
},
{
"epoch": 9.2,
"learning_rate": 3.4857142857142863e-06,
"loss": 0.9549,
"step": 1380
},
{
"epoch": 9.27,
"learning_rate": 3.2000000000000003e-06,
"loss": 0.9602,
"step": 1390
},
{
"epoch": 9.33,
"learning_rate": 2.9142857142857146e-06,
"loss": 0.9586,
"step": 1400
},
{
"epoch": 9.4,
"learning_rate": 2.6285714285714286e-06,
"loss": 0.9533,
"step": 1410
},
{
"epoch": 9.47,
"learning_rate": 2.342857142857143e-06,
"loss": 0.9522,
"step": 1420
},
{
"epoch": 9.53,
"learning_rate": 2.0571428571428573e-06,
"loss": 0.9786,
"step": 1430
},
{
"epoch": 9.6,
"learning_rate": 1.7714285714285714e-06,
"loss": 0.9473,
"step": 1440
},
{
"epoch": 9.66,
"learning_rate": 1.4857142857142858e-06,
"loss": 0.9531,
"step": 1450
},
{
"epoch": 9.73,
"learning_rate": 1.2000000000000002e-06,
"loss": 0.9488,
"step": 1460
},
{
"epoch": 9.8,
"learning_rate": 9.142857142857144e-07,
"loss": 0.9673,
"step": 1470
},
{
"epoch": 9.86,
"learning_rate": 6.285714285714287e-07,
"loss": 0.9485,
"step": 1480
},
{
"epoch": 9.93,
"learning_rate": 3.428571428571429e-07,
"loss": 0.9524,
"step": 1490
},
{
"epoch": 10.0,
"learning_rate": 5.714285714285715e-08,
"loss": 0.9466,
"step": 1500
},
{
"epoch": 10.0,
"eval_loss": 1.0791033506393433,
"eval_runtime": 348.3672,
"eval_samples_per_second": 7.676,
"eval_steps_per_second": 0.769,
"eval_wer": 0.7239809203618163,
"step": 1500
},
{
"epoch": 10.0,
"step": 1500,
"total_flos": 7.288767136199884e+19,
"train_loss": 1.0715523401896159,
"train_runtime": 76023.7044,
"train_samples_per_second": 3.165,
"train_steps_per_second": 0.02
}
],
"max_steps": 1500,
"num_train_epochs": 10,
"total_flos": 7.288767136199884e+19,
"trial_name": null,
"trial_params": null
}