gary109's picture
End of training
4309d28
{
"best_metric": 1.0753477811813354,
"best_model_checkpoint": "ai-light-dance_stepmania_ft_wav2vec2-large-xlsr-53-v2/checkpoint-750",
"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": 1.0000000000000002e-06,
"loss": 0.9783,
"step": 10
},
{
"epoch": 0.13,
"learning_rate": 2.0000000000000003e-06,
"loss": 0.9545,
"step": 20
},
{
"epoch": 0.2,
"learning_rate": 3e-06,
"loss": 0.9517,
"step": 30
},
{
"epoch": 0.27,
"learning_rate": 4.000000000000001e-06,
"loss": 0.9666,
"step": 40
},
{
"epoch": 0.33,
"learning_rate": 5e-06,
"loss": 0.9694,
"step": 50
},
{
"epoch": 0.4,
"learning_rate": 6e-06,
"loss": 0.9526,
"step": 60
},
{
"epoch": 0.47,
"learning_rate": 7e-06,
"loss": 0.9504,
"step": 70
},
{
"epoch": 0.53,
"learning_rate": 8.000000000000001e-06,
"loss": 0.9766,
"step": 80
},
{
"epoch": 0.6,
"learning_rate": 9e-06,
"loss": 0.951,
"step": 90
},
{
"epoch": 0.66,
"learning_rate": 1e-05,
"loss": 0.9501,
"step": 100
},
{
"epoch": 0.73,
"learning_rate": 9.92857142857143e-06,
"loss": 0.9476,
"step": 110
},
{
"epoch": 0.8,
"learning_rate": 9.857142857142859e-06,
"loss": 0.9734,
"step": 120
},
{
"epoch": 0.86,
"learning_rate": 9.785714285714286e-06,
"loss": 0.9455,
"step": 130
},
{
"epoch": 0.93,
"learning_rate": 9.714285714285715e-06,
"loss": 0.9558,
"step": 140
},
{
"epoch": 1.0,
"learning_rate": 9.642857142857144e-06,
"loss": 0.945,
"step": 150
},
{
"epoch": 1.0,
"eval_loss": 1.0766645669937134,
"eval_runtime": 396.7921,
"eval_samples_per_second": 6.739,
"eval_steps_per_second": 0.675,
"eval_wer": 0.7282187472042114,
"step": 150
},
{
"epoch": 1.07,
"learning_rate": 9.571428571428573e-06,
"loss": 1.009,
"step": 160
},
{
"epoch": 1.13,
"learning_rate": 9.5e-06,
"loss": 0.9516,
"step": 170
},
{
"epoch": 1.2,
"learning_rate": 9.42857142857143e-06,
"loss": 0.9492,
"step": 180
},
{
"epoch": 1.27,
"learning_rate": 9.357142857142859e-06,
"loss": 0.9623,
"step": 190
},
{
"epoch": 1.33,
"learning_rate": 9.285714285714288e-06,
"loss": 0.9597,
"step": 200
},
{
"epoch": 1.4,
"learning_rate": 9.214285714285715e-06,
"loss": 0.9379,
"step": 210
},
{
"epoch": 1.47,
"learning_rate": 9.142857142857144e-06,
"loss": 0.9417,
"step": 220
},
{
"epoch": 1.53,
"learning_rate": 9.071428571428573e-06,
"loss": 0.9649,
"step": 230
},
{
"epoch": 1.6,
"learning_rate": 9e-06,
"loss": 0.9412,
"step": 240
},
{
"epoch": 1.66,
"learning_rate": 8.92857142857143e-06,
"loss": 0.944,
"step": 250
},
{
"epoch": 1.73,
"learning_rate": 8.857142857142858e-06,
"loss": 0.9537,
"step": 260
},
{
"epoch": 1.8,
"learning_rate": 8.785714285714286e-06,
"loss": 0.9674,
"step": 270
},
{
"epoch": 1.86,
"learning_rate": 8.714285714285715e-06,
"loss": 0.9479,
"step": 280
},
{
"epoch": 1.93,
"learning_rate": 8.642857142857144e-06,
"loss": 0.9496,
"step": 290
},
{
"epoch": 2.0,
"learning_rate": 8.571428571428571e-06,
"loss": 0.9445,
"step": 300
},
{
"epoch": 2.0,
"eval_loss": 1.077294111251831,
"eval_runtime": 387.2792,
"eval_samples_per_second": 6.905,
"eval_steps_per_second": 0.692,
"eval_wer": 0.7164846755472682,
"step": 300
},
{
"epoch": 2.07,
"learning_rate": 8.5e-06,
"loss": 1.0108,
"step": 310
},
{
"epoch": 2.13,
"learning_rate": 8.428571428571429e-06,
"loss": 0.939,
"step": 320
},
{
"epoch": 2.2,
"learning_rate": 8.357142857142858e-06,
"loss": 0.9354,
"step": 330
},
{
"epoch": 2.27,
"learning_rate": 8.285714285714287e-06,
"loss": 0.9539,
"step": 340
},
{
"epoch": 2.33,
"learning_rate": 8.214285714285714e-06,
"loss": 0.9509,
"step": 350
},
{
"epoch": 2.4,
"learning_rate": 8.142857142857143e-06,
"loss": 0.94,
"step": 360
},
{
"epoch": 2.47,
"learning_rate": 8.071428571428572e-06,
"loss": 0.9391,
"step": 370
},
{
"epoch": 2.53,
"learning_rate": 8.000000000000001e-06,
"loss": 0.9588,
"step": 380
},
{
"epoch": 2.6,
"learning_rate": 7.928571428571429e-06,
"loss": 0.9306,
"step": 390
},
{
"epoch": 2.66,
"learning_rate": 7.857142857142858e-06,
"loss": 0.9358,
"step": 400
},
{
"epoch": 2.73,
"learning_rate": 7.785714285714287e-06,
"loss": 0.938,
"step": 410
},
{
"epoch": 2.8,
"learning_rate": 7.714285714285716e-06,
"loss": 0.9591,
"step": 420
},
{
"epoch": 2.86,
"learning_rate": 7.642857142857143e-06,
"loss": 0.9435,
"step": 430
},
{
"epoch": 2.93,
"learning_rate": 7.571428571428572e-06,
"loss": 0.9339,
"step": 440
},
{
"epoch": 3.0,
"learning_rate": 7.500000000000001e-06,
"loss": 0.9392,
"step": 450
},
{
"epoch": 3.0,
"eval_loss": 1.081303358078003,
"eval_runtime": 381.1488,
"eval_samples_per_second": 7.016,
"eval_steps_per_second": 0.703,
"eval_wer": 0.7141350359979847,
"step": 450
},
{
"epoch": 3.07,
"learning_rate": 7.428571428571429e-06,
"loss": 0.9972,
"step": 460
},
{
"epoch": 3.13,
"learning_rate": 7.357142857142858e-06,
"loss": 0.9387,
"step": 470
},
{
"epoch": 3.2,
"learning_rate": 7.285714285714286e-06,
"loss": 0.936,
"step": 480
},
{
"epoch": 3.27,
"learning_rate": 7.2142857142857145e-06,
"loss": 0.9475,
"step": 490
},
{
"epoch": 3.33,
"learning_rate": 7.1428571428571436e-06,
"loss": 0.9346,
"step": 500
},
{
"epoch": 3.4,
"learning_rate": 7.0714285714285726e-06,
"loss": 0.9356,
"step": 510
},
{
"epoch": 3.47,
"learning_rate": 7e-06,
"loss": 0.93,
"step": 520
},
{
"epoch": 3.53,
"learning_rate": 6.928571428571429e-06,
"loss": 0.9631,
"step": 530
},
{
"epoch": 3.6,
"learning_rate": 6.857142857142858e-06,
"loss": 0.9329,
"step": 540
},
{
"epoch": 3.66,
"learning_rate": 6.785714285714287e-06,
"loss": 0.9321,
"step": 550
},
{
"epoch": 3.73,
"learning_rate": 6.714285714285714e-06,
"loss": 0.9318,
"step": 560
},
{
"epoch": 3.8,
"learning_rate": 6.642857142857143e-06,
"loss": 0.9546,
"step": 570
},
{
"epoch": 3.86,
"learning_rate": 6.571428571428572e-06,
"loss": 0.9314,
"step": 580
},
{
"epoch": 3.93,
"learning_rate": 6.5000000000000004e-06,
"loss": 0.9306,
"step": 590
},
{
"epoch": 4.0,
"learning_rate": 6.4285714285714295e-06,
"loss": 0.933,
"step": 600
},
{
"epoch": 4.0,
"eval_loss": 1.0858107805252075,
"eval_runtime": 378.2791,
"eval_samples_per_second": 7.069,
"eval_steps_per_second": 0.708,
"eval_wer": 0.7032296949235544,
"step": 600
},
{
"epoch": 4.07,
"learning_rate": 6.357142857142858e-06,
"loss": 0.9928,
"step": 610
},
{
"epoch": 4.13,
"learning_rate": 6.285714285714286e-06,
"loss": 0.9338,
"step": 620
},
{
"epoch": 4.2,
"learning_rate": 6.214285714285715e-06,
"loss": 0.9252,
"step": 630
},
{
"epoch": 4.27,
"learning_rate": 6.142857142857144e-06,
"loss": 0.9404,
"step": 640
},
{
"epoch": 4.33,
"learning_rate": 6.071428571428571e-06,
"loss": 0.943,
"step": 650
},
{
"epoch": 4.4,
"learning_rate": 6e-06,
"loss": 0.9282,
"step": 660
},
{
"epoch": 4.47,
"learning_rate": 5.928571428571429e-06,
"loss": 0.9262,
"step": 670
},
{
"epoch": 4.53,
"learning_rate": 5.857142857142858e-06,
"loss": 0.953,
"step": 680
},
{
"epoch": 4.6,
"learning_rate": 5.785714285714286e-06,
"loss": 0.9247,
"step": 690
},
{
"epoch": 4.66,
"learning_rate": 5.7142857142857145e-06,
"loss": 0.9275,
"step": 700
},
{
"epoch": 4.73,
"learning_rate": 5.6428571428571435e-06,
"loss": 0.9281,
"step": 710
},
{
"epoch": 4.8,
"learning_rate": 5.571428571428572e-06,
"loss": 0.9534,
"step": 720
},
{
"epoch": 4.86,
"learning_rate": 5.500000000000001e-06,
"loss": 0.9313,
"step": 730
},
{
"epoch": 4.93,
"learning_rate": 5.428571428571429e-06,
"loss": 0.924,
"step": 740
},
{
"epoch": 5.0,
"learning_rate": 5.357142857142857e-06,
"loss": 0.921,
"step": 750
},
{
"epoch": 5.0,
"eval_loss": 1.0753477811813354,
"eval_runtime": 381.1227,
"eval_samples_per_second": 7.016,
"eval_steps_per_second": 0.703,
"eval_wer": 0.7017134946532751,
"step": 750
},
{
"epoch": 5.07,
"learning_rate": 5.285714285714286e-06,
"loss": 0.9959,
"step": 760
},
{
"epoch": 5.13,
"learning_rate": 5.214285714285715e-06,
"loss": 0.9258,
"step": 770
},
{
"epoch": 5.2,
"learning_rate": 5.142857142857142e-06,
"loss": 0.9243,
"step": 780
},
{
"epoch": 5.27,
"learning_rate": 5.071428571428571e-06,
"loss": 0.9383,
"step": 790
},
{
"epoch": 5.33,
"learning_rate": 5e-06,
"loss": 0.9343,
"step": 800
},
{
"epoch": 5.4,
"learning_rate": 4.928571428571429e-06,
"loss": 0.9226,
"step": 810
},
{
"epoch": 5.47,
"learning_rate": 4.857142857142858e-06,
"loss": 0.9173,
"step": 820
},
{
"epoch": 5.53,
"learning_rate": 4.785714285714287e-06,
"loss": 0.9431,
"step": 830
},
{
"epoch": 5.6,
"learning_rate": 4.714285714285715e-06,
"loss": 0.9244,
"step": 840
},
{
"epoch": 5.66,
"learning_rate": 4.642857142857144e-06,
"loss": 0.9241,
"step": 850
},
{
"epoch": 5.73,
"learning_rate": 4.571428571428572e-06,
"loss": 0.9178,
"step": 860
},
{
"epoch": 5.8,
"learning_rate": 4.5e-06,
"loss": 0.9462,
"step": 870
},
{
"epoch": 5.86,
"learning_rate": 4.428571428571429e-06,
"loss": 0.9264,
"step": 880
},
{
"epoch": 5.93,
"learning_rate": 4.357142857142857e-06,
"loss": 0.9198,
"step": 890
},
{
"epoch": 6.0,
"learning_rate": 4.2857142857142855e-06,
"loss": 0.9241,
"step": 900
},
{
"epoch": 6.0,
"eval_loss": 1.0786956548690796,
"eval_runtime": 379.7908,
"eval_samples_per_second": 7.041,
"eval_steps_per_second": 0.706,
"eval_wer": 0.6975557156512363,
"step": 900
},
{
"epoch": 6.07,
"learning_rate": 4.2142857142857145e-06,
"loss": 0.9893,
"step": 910
},
{
"epoch": 6.13,
"learning_rate": 4.1428571428571435e-06,
"loss": 0.9191,
"step": 920
},
{
"epoch": 6.2,
"learning_rate": 4.071428571428572e-06,
"loss": 0.9263,
"step": 930
},
{
"epoch": 6.27,
"learning_rate": 4.000000000000001e-06,
"loss": 0.9254,
"step": 940
},
{
"epoch": 6.33,
"learning_rate": 3.928571428571429e-06,
"loss": 0.9332,
"step": 950
},
{
"epoch": 6.4,
"learning_rate": 3.857142857142858e-06,
"loss": 0.9199,
"step": 960
},
{
"epoch": 6.47,
"learning_rate": 3.785714285714286e-06,
"loss": 0.9154,
"step": 970
},
{
"epoch": 6.53,
"learning_rate": 3.7142857142857146e-06,
"loss": 0.9424,
"step": 980
},
{
"epoch": 6.6,
"learning_rate": 3.642857142857143e-06,
"loss": 0.9172,
"step": 990
},
{
"epoch": 6.66,
"learning_rate": 3.5714285714285718e-06,
"loss": 0.9224,
"step": 1000
},
{
"epoch": 6.73,
"learning_rate": 3.5e-06,
"loss": 0.918,
"step": 1010
},
{
"epoch": 6.8,
"learning_rate": 3.428571428571429e-06,
"loss": 0.9374,
"step": 1020
},
{
"epoch": 6.86,
"learning_rate": 3.357142857142857e-06,
"loss": 0.9137,
"step": 1030
},
{
"epoch": 6.93,
"learning_rate": 3.285714285714286e-06,
"loss": 0.9242,
"step": 1040
},
{
"epoch": 7.0,
"learning_rate": 3.2142857142857147e-06,
"loss": 0.9282,
"step": 1050
},
{
"epoch": 7.0,
"eval_loss": 1.082495927810669,
"eval_runtime": 383.2073,
"eval_samples_per_second": 6.978,
"eval_steps_per_second": 0.699,
"eval_wer": 0.6959123805756853,
"step": 1050
},
{
"epoch": 7.07,
"learning_rate": 3.142857142857143e-06,
"loss": 0.9848,
"step": 1060
},
{
"epoch": 7.13,
"learning_rate": 3.071428571428572e-06,
"loss": 0.9162,
"step": 1070
},
{
"epoch": 7.2,
"learning_rate": 3e-06,
"loss": 0.9152,
"step": 1080
},
{
"epoch": 7.27,
"learning_rate": 2.928571428571429e-06,
"loss": 0.929,
"step": 1090
},
{
"epoch": 7.33,
"learning_rate": 2.8571428571428573e-06,
"loss": 0.9288,
"step": 1100
},
{
"epoch": 7.4,
"learning_rate": 2.785714285714286e-06,
"loss": 0.9184,
"step": 1110
},
{
"epoch": 7.47,
"learning_rate": 2.7142857142857144e-06,
"loss": 0.9173,
"step": 1120
},
{
"epoch": 7.53,
"learning_rate": 2.642857142857143e-06,
"loss": 0.9402,
"step": 1130
},
{
"epoch": 7.6,
"learning_rate": 2.571428571428571e-06,
"loss": 0.921,
"step": 1140
},
{
"epoch": 7.66,
"learning_rate": 2.5e-06,
"loss": 0.915,
"step": 1150
},
{
"epoch": 7.73,
"learning_rate": 2.428571428571429e-06,
"loss": 0.9187,
"step": 1160
},
{
"epoch": 7.8,
"learning_rate": 2.3642857142857144e-06,
"loss": 0.9379,
"step": 1170
},
{
"epoch": 7.86,
"learning_rate": 2.292857142857143e-06,
"loss": 0.9123,
"step": 1180
},
{
"epoch": 7.93,
"learning_rate": 2.2214285714285716e-06,
"loss": 0.9096,
"step": 1190
},
{
"epoch": 8.0,
"learning_rate": 2.15e-06,
"loss": 0.9184,
"step": 1200
},
{
"epoch": 8.0,
"eval_loss": 1.075991153717041,
"eval_runtime": 379.1679,
"eval_samples_per_second": 7.052,
"eval_steps_per_second": 0.707,
"eval_wer": 0.6929788626614495,
"step": 1200
},
{
"epoch": 8.07,
"learning_rate": 2.0785714285714288e-06,
"loss": 0.9763,
"step": 1210
},
{
"epoch": 8.13,
"learning_rate": 2.0071428571428573e-06,
"loss": 0.9114,
"step": 1220
},
{
"epoch": 8.2,
"learning_rate": 1.935714285714286e-06,
"loss": 0.9196,
"step": 1230
},
{
"epoch": 8.27,
"learning_rate": 1.8642857142857143e-06,
"loss": 0.9268,
"step": 1240
},
{
"epoch": 8.33,
"learning_rate": 1.7928571428571431e-06,
"loss": 0.9293,
"step": 1250
},
{
"epoch": 8.4,
"learning_rate": 1.7214285714285717e-06,
"loss": 0.9128,
"step": 1260
},
{
"epoch": 8.47,
"learning_rate": 1.6500000000000003e-06,
"loss": 0.907,
"step": 1270
},
{
"epoch": 8.53,
"learning_rate": 1.5785714285714287e-06,
"loss": 0.9437,
"step": 1280
},
{
"epoch": 8.6,
"learning_rate": 1.5071428571428573e-06,
"loss": 0.9165,
"step": 1290
},
{
"epoch": 8.66,
"learning_rate": 1.4357142857142859e-06,
"loss": 0.9177,
"step": 1300
},
{
"epoch": 8.73,
"learning_rate": 1.3642857142857144e-06,
"loss": 0.9063,
"step": 1310
},
{
"epoch": 8.8,
"learning_rate": 1.2928571428571428e-06,
"loss": 0.9389,
"step": 1320
},
{
"epoch": 8.86,
"learning_rate": 1.2214285714285714e-06,
"loss": 0.9135,
"step": 1330
},
{
"epoch": 8.93,
"learning_rate": 1.1500000000000002e-06,
"loss": 0.9128,
"step": 1340
},
{
"epoch": 9.0,
"learning_rate": 1.0785714285714286e-06,
"loss": 0.915,
"step": 1350
},
{
"epoch": 9.0,
"eval_loss": 1.0773109197616577,
"eval_runtime": 373.9554,
"eval_samples_per_second": 7.151,
"eval_steps_per_second": 0.717,
"eval_wer": 0.6905538839683011,
"step": 1350
},
{
"epoch": 9.07,
"learning_rate": 1.0071428571428572e-06,
"loss": 0.9846,
"step": 1360
},
{
"epoch": 9.13,
"learning_rate": 9.357142857142858e-07,
"loss": 0.9192,
"step": 1370
},
{
"epoch": 9.2,
"learning_rate": 8.642857142857144e-07,
"loss": 0.9162,
"step": 1380
},
{
"epoch": 9.27,
"learning_rate": 7.928571428571428e-07,
"loss": 0.9243,
"step": 1390
},
{
"epoch": 9.33,
"learning_rate": 7.214285714285715e-07,
"loss": 0.9208,
"step": 1400
},
{
"epoch": 9.4,
"learning_rate": 6.5e-07,
"loss": 0.9158,
"step": 1410
},
{
"epoch": 9.47,
"learning_rate": 5.785714285714286e-07,
"loss": 0.9122,
"step": 1420
},
{
"epoch": 9.53,
"learning_rate": 5.071428571428572e-07,
"loss": 0.9437,
"step": 1430
},
{
"epoch": 9.6,
"learning_rate": 4.357142857142858e-07,
"loss": 0.9085,
"step": 1440
},
{
"epoch": 9.66,
"learning_rate": 3.6428571428571433e-07,
"loss": 0.9151,
"step": 1450
},
{
"epoch": 9.73,
"learning_rate": 2.9285714285714287e-07,
"loss": 0.9117,
"step": 1460
},
{
"epoch": 9.8,
"learning_rate": 2.2142857142857143e-07,
"loss": 0.9325,
"step": 1470
},
{
"epoch": 9.86,
"learning_rate": 1.5000000000000002e-07,
"loss": 0.9114,
"step": 1480
},
{
"epoch": 9.93,
"learning_rate": 7.857142857142858e-08,
"loss": 0.9152,
"step": 1490
},
{
"epoch": 10.0,
"learning_rate": 7.142857142857144e-09,
"loss": 0.9094,
"step": 1500
},
{
"epoch": 10.0,
"eval_loss": 1.0785654783248901,
"eval_runtime": 377.596,
"eval_samples_per_second": 7.082,
"eval_steps_per_second": 0.71,
"eval_wer": 0.6899935490858066,
"step": 1500
},
{
"epoch": 10.0,
"step": 1500,
"total_flos": 7.288767136199884e+19,
"train_loss": 0.9371341470082601,
"train_runtime": 79168.6766,
"train_samples_per_second": 3.039,
"train_steps_per_second": 0.019
}
],
"max_steps": 1500,
"num_train_epochs": 10,
"total_flos": 7.288767136199884e+19,
"trial_name": null,
"trial_params": null
}