xls-r-1b-fr / trainer_state.json
Plim's picture
End of training
f313a54
{
"best_metric": 0.2463931441307068,
"best_model_checkpoint": "./checkpoint-6000",
"epoch": 4.999878498238225,
"global_step": 15430,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03,
"learning_rate": 3.7125e-06,
"loss": 7.2804,
"step": 100
},
{
"epoch": 0.06,
"learning_rate": 7.4625e-06,
"loss": 2.9939,
"step": 200
},
{
"epoch": 0.1,
"learning_rate": 1.1212499999999998e-05,
"loss": 2.8161,
"step": 300
},
{
"epoch": 0.13,
"learning_rate": 1.49625e-05,
"loss": 1.8175,
"step": 400
},
{
"epoch": 0.16,
"learning_rate": 1.8675e-05,
"loss": 1.3167,
"step": 500
},
{
"epoch": 0.19,
"learning_rate": 2.2424999999999996e-05,
"loss": 1.1703,
"step": 600
},
{
"epoch": 0.23,
"learning_rate": 2.6174999999999996e-05,
"loss": 1.1128,
"step": 700
},
{
"epoch": 0.26,
"learning_rate": 2.9925e-05,
"loss": 1.0763,
"step": 800
},
{
"epoch": 0.29,
"learning_rate": 3.3675e-05,
"loss": 1.0522,
"step": 900
},
{
"epoch": 0.32,
"learning_rate": 3.7424999999999995e-05,
"loss": 1.0326,
"step": 1000
},
{
"epoch": 0.32,
"eval_loss": 0.30916452407836914,
"eval_runtime": 1003.5092,
"eval_samples_per_second": 15.885,
"eval_steps_per_second": 0.994,
"eval_wer": 0.2718170286472251,
"step": 1000
},
{
"epoch": 0.36,
"learning_rate": 4.1175e-05,
"loss": 1.0398,
"step": 1100
},
{
"epoch": 0.39,
"learning_rate": 4.4924999999999994e-05,
"loss": 1.0289,
"step": 1200
},
{
"epoch": 0.42,
"learning_rate": 4.8675e-05,
"loss": 1.0403,
"step": 1300
},
{
"epoch": 0.45,
"learning_rate": 5.2424999999999994e-05,
"loss": 1.0439,
"step": 1400
},
{
"epoch": 0.49,
"learning_rate": 5.6175e-05,
"loss": 1.0455,
"step": 1500
},
{
"epoch": 0.52,
"learning_rate": 5.9925e-05,
"loss": 1.0567,
"step": 1600
},
{
"epoch": 0.55,
"learning_rate": 6.36375e-05,
"loss": 1.0687,
"step": 1700
},
{
"epoch": 0.58,
"learning_rate": 6.738749999999999e-05,
"loss": 1.0601,
"step": 1800
},
{
"epoch": 0.62,
"learning_rate": 7.11375e-05,
"loss": 1.0657,
"step": 1900
},
{
"epoch": 0.65,
"learning_rate": 7.48875e-05,
"loss": 1.0828,
"step": 2000
},
{
"epoch": 0.65,
"eval_loss": 0.28432372212409973,
"eval_runtime": 1004.4407,
"eval_samples_per_second": 15.871,
"eval_steps_per_second": 0.993,
"eval_wer": 0.26055517279125323,
"step": 2000
},
{
"epoch": 0.68,
"learning_rate": 7.4463886820551e-05,
"loss": 1.09,
"step": 2100
},
{
"epoch": 0.71,
"learning_rate": 7.390543559195829e-05,
"loss": 1.0696,
"step": 2200
},
{
"epoch": 0.75,
"learning_rate": 7.33469843633656e-05,
"loss": 1.0794,
"step": 2300
},
{
"epoch": 0.78,
"learning_rate": 7.279411764705881e-05,
"loss": 1.0906,
"step": 2400
},
{
"epoch": 0.81,
"learning_rate": 7.223566641846612e-05,
"loss": 1.0739,
"step": 2500
},
{
"epoch": 0.84,
"learning_rate": 7.16772151898734e-05,
"loss": 1.0753,
"step": 2600
},
{
"epoch": 0.87,
"learning_rate": 7.111876396128072e-05,
"loss": 1.0694,
"step": 2700
},
{
"epoch": 0.91,
"learning_rate": 7.0560312732688e-05,
"loss": 1.0652,
"step": 2800
},
{
"epoch": 0.94,
"learning_rate": 7.00018615040953e-05,
"loss": 1.0627,
"step": 2900
},
{
"epoch": 0.97,
"learning_rate": 6.94434102755026e-05,
"loss": 1.0771,
"step": 3000
},
{
"epoch": 0.97,
"eval_loss": 0.27740851044654846,
"eval_runtime": 1003.9914,
"eval_samples_per_second": 15.878,
"eval_steps_per_second": 0.993,
"eval_wer": 0.24881682903743005,
"step": 3000
},
{
"epoch": 1.0,
"learning_rate": 6.88849590469099e-05,
"loss": 1.0548,
"step": 3100
},
{
"epoch": 1.04,
"learning_rate": 6.832650781831719e-05,
"loss": 1.0594,
"step": 3200
},
{
"epoch": 1.07,
"learning_rate": 6.776805658972449e-05,
"loss": 1.0605,
"step": 3300
},
{
"epoch": 1.1,
"learning_rate": 6.720960536113179e-05,
"loss": 1.0501,
"step": 3400
},
{
"epoch": 1.13,
"learning_rate": 6.665673864482501e-05,
"loss": 1.06,
"step": 3500
},
{
"epoch": 1.17,
"learning_rate": 6.609828741623231e-05,
"loss": 1.0539,
"step": 3600
},
{
"epoch": 1.2,
"learning_rate": 6.553983618763961e-05,
"loss": 1.0483,
"step": 3700
},
{
"epoch": 1.23,
"learning_rate": 6.49813849590469e-05,
"loss": 1.0507,
"step": 3800
},
{
"epoch": 1.26,
"learning_rate": 6.44229337304542e-05,
"loss": 1.0416,
"step": 3900
},
{
"epoch": 1.3,
"learning_rate": 6.38644825018615e-05,
"loss": 1.0306,
"step": 4000
},
{
"epoch": 1.3,
"eval_loss": 0.25875675678253174,
"eval_runtime": 1004.8682,
"eval_samples_per_second": 15.864,
"eval_steps_per_second": 0.992,
"eval_wer": 0.23513389953832187,
"step": 4000
},
{
"epoch": 1.33,
"learning_rate": 6.33060312732688e-05,
"loss": 1.0413,
"step": 4100
},
{
"epoch": 1.36,
"learning_rate": 6.275316455696202e-05,
"loss": 1.0475,
"step": 4200
},
{
"epoch": 1.39,
"learning_rate": 6.219471332836931e-05,
"loss": 1.0328,
"step": 4300
},
{
"epoch": 1.43,
"learning_rate": 6.163626209977662e-05,
"loss": 1.036,
"step": 4400
},
{
"epoch": 1.46,
"learning_rate": 6.10778108711839e-05,
"loss": 1.0248,
"step": 4500
},
{
"epoch": 1.49,
"learning_rate": 6.051935964259121e-05,
"loss": 1.0184,
"step": 4600
},
{
"epoch": 1.52,
"learning_rate": 5.996090841399851e-05,
"loss": 1.0268,
"step": 4700
},
{
"epoch": 1.56,
"learning_rate": 5.94024571854058e-05,
"loss": 1.0177,
"step": 4800
},
{
"epoch": 1.59,
"learning_rate": 5.88440059568131e-05,
"loss": 1.0155,
"step": 4900
},
{
"epoch": 1.62,
"learning_rate": 5.82855547282204e-05,
"loss": 1.0052,
"step": 5000
},
{
"epoch": 1.62,
"eval_loss": 0.24826321005821228,
"eval_runtime": 1000.3482,
"eval_samples_per_second": 15.935,
"eval_steps_per_second": 0.997,
"eval_wer": 0.22835360553241082,
"step": 5000
},
{
"epoch": 1.65,
"learning_rate": 5.7727103499627696e-05,
"loss": 1.0076,
"step": 5100
},
{
"epoch": 1.68,
"learning_rate": 5.717423678332092e-05,
"loss": 1.0107,
"step": 5200
},
{
"epoch": 1.72,
"learning_rate": 5.661578555472821e-05,
"loss": 1.0135,
"step": 5300
},
{
"epoch": 1.75,
"learning_rate": 5.605733432613552e-05,
"loss": 1.012,
"step": 5400
},
{
"epoch": 1.78,
"learning_rate": 5.549888309754281e-05,
"loss": 0.9943,
"step": 5500
},
{
"epoch": 1.81,
"learning_rate": 5.4940431868950106e-05,
"loss": 1.0049,
"step": 5600
},
{
"epoch": 1.85,
"learning_rate": 5.4381980640357404e-05,
"loss": 1.0082,
"step": 5700
},
{
"epoch": 1.88,
"learning_rate": 5.38235294117647e-05,
"loss": 0.9948,
"step": 5800
},
{
"epoch": 1.91,
"learning_rate": 5.326507818317199e-05,
"loss": 0.9925,
"step": 5900
},
{
"epoch": 1.94,
"learning_rate": 5.2706626954579296e-05,
"loss": 0.9865,
"step": 6000
},
{
"epoch": 1.94,
"eval_loss": 0.2463931441307068,
"eval_runtime": 999.3892,
"eval_samples_per_second": 15.951,
"eval_steps_per_second": 0.998,
"eval_wer": 0.22199184819353135,
"step": 6000
},
{
"epoch": 1.98,
"learning_rate": 5.2148175725986594e-05,
"loss": 0.987,
"step": 6100
},
{
"epoch": 2.01,
"learning_rate": 5.158972449739389e-05,
"loss": 0.9789,
"step": 6200
},
{
"epoch": 2.04,
"learning_rate": 5.103127326880119e-05,
"loss": 0.9722,
"step": 6300
},
{
"epoch": 2.07,
"learning_rate": 5.047282204020848e-05,
"loss": 0.9747,
"step": 6400
},
{
"epoch": 2.11,
"learning_rate": 4.9914370811615785e-05,
"loss": 0.9771,
"step": 6500
},
{
"epoch": 2.14,
"learning_rate": 4.9355919583023076e-05,
"loss": 0.9777,
"step": 6600
},
{
"epoch": 2.17,
"learning_rate": 4.88030528667163e-05,
"loss": 0.9804,
"step": 6700
},
{
"epoch": 2.2,
"learning_rate": 4.82446016381236e-05,
"loss": 0.9575,
"step": 6800
},
{
"epoch": 2.24,
"learning_rate": 4.76861504095309e-05,
"loss": 0.9613,
"step": 6900
},
{
"epoch": 2.27,
"learning_rate": 4.713328369322412e-05,
"loss": 0.978,
"step": 7000
},
{
"epoch": 2.27,
"eval_loss": 0.2513602077960968,
"eval_runtime": 1010.6008,
"eval_samples_per_second": 15.774,
"eval_steps_per_second": 0.987,
"eval_wer": 0.21723340824066503,
"step": 7000
},
{
"epoch": 2.3,
"learning_rate": 4.658041697691735e-05,
"loss": 0.9837,
"step": 7100
},
{
"epoch": 2.33,
"learning_rate": 4.602196574832464e-05,
"loss": 1.0137,
"step": 7200
},
{
"epoch": 2.37,
"learning_rate": 4.546351451973194e-05,
"loss": 1.0971,
"step": 7300
},
{
"epoch": 2.4,
"learning_rate": 4.4905063291139234e-05,
"loss": 1.1683,
"step": 7400
},
{
"epoch": 2.43,
"learning_rate": 4.434661206254654e-05,
"loss": 1.2993,
"step": 7500
},
{
"epoch": 2.46,
"learning_rate": 4.378816083395383e-05,
"loss": 1.3806,
"step": 7600
},
{
"epoch": 2.5,
"learning_rate": 4.322970960536113e-05,
"loss": 1.4648,
"step": 7700
},
{
"epoch": 2.53,
"learning_rate": 4.2671258376768425e-05,
"loss": 1.5643,
"step": 7800
},
{
"epoch": 2.56,
"learning_rate": 4.211280714817572e-05,
"loss": 1.5596,
"step": 7900
},
{
"epoch": 2.59,
"learning_rate": 4.155435591958301e-05,
"loss": 1.7438,
"step": 8000
},
{
"epoch": 2.59,
"eval_loss": 0.7983415722846985,
"eval_runtime": 1011.5068,
"eval_samples_per_second": 15.76,
"eval_steps_per_second": 0.986,
"eval_wer": 0.5072471233652923,
"step": 8000
},
{
"epoch": 2.62,
"learning_rate": 4.099590469099032e-05,
"loss": 1.7779,
"step": 8100
},
{
"epoch": 2.66,
"learning_rate": 4.0437453462397615e-05,
"loss": 1.9193,
"step": 8200
},
{
"epoch": 2.69,
"learning_rate": 3.987900223380491e-05,
"loss": 1.8549,
"step": 8300
},
{
"epoch": 2.72,
"learning_rate": 3.932055100521221e-05,
"loss": 1.9323,
"step": 8400
},
{
"epoch": 2.75,
"learning_rate": 3.87620997766195e-05,
"loss": 2.0832,
"step": 8500
},
{
"epoch": 2.79,
"learning_rate": 3.8203648548026806e-05,
"loss": 2.268,
"step": 8600
},
{
"epoch": 2.82,
"learning_rate": 3.76451973194341e-05,
"loss": 1.9962,
"step": 8700
},
{
"epoch": 2.85,
"learning_rate": 3.7086746090841395e-05,
"loss": 2.2105,
"step": 8800
},
{
"epoch": 2.88,
"learning_rate": 3.65282948622487e-05,
"loss": 2.3688,
"step": 8900
},
{
"epoch": 2.92,
"learning_rate": 3.596984363365599e-05,
"loss": 2.3309,
"step": 9000
},
{
"epoch": 2.92,
"eval_loss": 1.8916891813278198,
"eval_runtime": 1006.0568,
"eval_samples_per_second": 15.845,
"eval_steps_per_second": 0.991,
"eval_wer": 0.9416173544619229,
"step": 9000
},
{
"epoch": 2.95,
"learning_rate": 3.541139240506329e-05,
"loss": 2.2688,
"step": 9100
},
{
"epoch": 2.98,
"learning_rate": 3.4852941176470585e-05,
"loss": 2.0991,
"step": 9200
},
{
"epoch": 3.01,
"learning_rate": 3.430007446016381e-05,
"loss": 2.0663,
"step": 9300
},
{
"epoch": 3.05,
"learning_rate": 3.374162323157111e-05,
"loss": 2.0995,
"step": 9400
},
{
"epoch": 3.08,
"learning_rate": 3.3183172002978406e-05,
"loss": 2.2691,
"step": 9500
},
{
"epoch": 3.11,
"learning_rate": 3.26247207743857e-05,
"loss": 2.1691,
"step": 9600
},
{
"epoch": 3.14,
"learning_rate": 3.2066269545792995e-05,
"loss": 2.2706,
"step": 9700
},
{
"epoch": 3.18,
"learning_rate": 3.150781831720029e-05,
"loss": 2.0075,
"step": 9800
},
{
"epoch": 3.21,
"learning_rate": 3.094936708860759e-05,
"loss": 2.1615,
"step": 9900
},
{
"epoch": 3.24,
"learning_rate": 3.0390915860014888e-05,
"loss": 2.1834,
"step": 10000
},
{
"epoch": 3.24,
"eval_loss": 1.7495689392089844,
"eval_runtime": 1013.0296,
"eval_samples_per_second": 15.736,
"eval_steps_per_second": 0.984,
"eval_wer": 0.9029832005820879,
"step": 10000
},
{
"epoch": 3.27,
"learning_rate": 2.983246463142219e-05,
"loss": 2.2615,
"step": 10100
},
{
"epoch": 3.31,
"learning_rate": 2.9274013402829486e-05,
"loss": 2.387,
"step": 10200
},
{
"epoch": 3.34,
"learning_rate": 2.871556217423678e-05,
"loss": 2.3751,
"step": 10300
},
{
"epoch": 3.37,
"learning_rate": 2.8157110945644078e-05,
"loss": 2.3223,
"step": 10400
},
{
"epoch": 3.4,
"learning_rate": 2.7604244229337304e-05,
"loss": 2.2708,
"step": 10500
},
{
"epoch": 3.43,
"learning_rate": 2.7045793000744598e-05,
"loss": 2.2369,
"step": 10600
},
{
"epoch": 3.47,
"learning_rate": 2.6487341772151896e-05,
"loss": 2.3768,
"step": 10700
},
{
"epoch": 3.5,
"learning_rate": 2.5928890543559194e-05,
"loss": 2.2929,
"step": 10800
},
{
"epoch": 3.53,
"learning_rate": 2.5376023827252416e-05,
"loss": 2.3779,
"step": 10900
},
{
"epoch": 3.56,
"learning_rate": 2.4817572598659713e-05,
"loss": 2.3047,
"step": 11000
},
{
"epoch": 3.56,
"eval_loss": 1.5376594066619873,
"eval_runtime": 1009.9366,
"eval_samples_per_second": 15.784,
"eval_steps_per_second": 0.987,
"eval_wer": 0.874715877993342,
"step": 11000
},
{
"epoch": 3.6,
"learning_rate": 2.4259121370067014e-05,
"loss": 2.0797,
"step": 11100
},
{
"epoch": 3.63,
"learning_rate": 2.3700670141474312e-05,
"loss": 2.1414,
"step": 11200
},
{
"epoch": 3.66,
"learning_rate": 2.3142218912881606e-05,
"loss": 2.1064,
"step": 11300
},
{
"epoch": 3.69,
"learning_rate": 2.2583767684288904e-05,
"loss": 2.0681,
"step": 11400
},
{
"epoch": 3.73,
"learning_rate": 2.2025316455696202e-05,
"loss": 2.0853,
"step": 11500
},
{
"epoch": 3.76,
"learning_rate": 2.14668652271035e-05,
"loss": 2.1272,
"step": 11600
},
{
"epoch": 3.79,
"learning_rate": 2.0908413998510794e-05,
"loss": 2.1257,
"step": 11700
},
{
"epoch": 3.82,
"learning_rate": 2.034996276991809e-05,
"loss": 2.0966,
"step": 11800
},
{
"epoch": 3.86,
"learning_rate": 1.979151154132539e-05,
"loss": 2.2297,
"step": 11900
},
{
"epoch": 3.89,
"learning_rate": 1.9233060312732687e-05,
"loss": 2.1378,
"step": 12000
},
{
"epoch": 3.89,
"eval_loss": 1.3500523567199707,
"eval_runtime": 1013.8007,
"eval_samples_per_second": 15.724,
"eval_steps_per_second": 0.983,
"eval_wer": 0.7923285448445941,
"step": 12000
},
{
"epoch": 3.92,
"learning_rate": 1.8680193596425912e-05,
"loss": 2.2683,
"step": 12100
},
{
"epoch": 3.95,
"learning_rate": 1.8127326880119135e-05,
"loss": 2.311,
"step": 12200
},
{
"epoch": 3.99,
"learning_rate": 1.7568875651526432e-05,
"loss": 2.0594,
"step": 12300
},
{
"epoch": 4.02,
"learning_rate": 1.701042442293373e-05,
"loss": 1.9798,
"step": 12400
},
{
"epoch": 4.05,
"learning_rate": 1.6457557706626952e-05,
"loss": 1.8521,
"step": 12500
},
{
"epoch": 4.08,
"learning_rate": 1.589910647803425e-05,
"loss": 1.8466,
"step": 12600
},
{
"epoch": 4.12,
"learning_rate": 1.5340655249441547e-05,
"loss": 1.9372,
"step": 12700
},
{
"epoch": 4.15,
"learning_rate": 1.4782204020848843e-05,
"loss": 1.9591,
"step": 12800
},
{
"epoch": 4.18,
"learning_rate": 1.4223752792256143e-05,
"loss": 1.9641,
"step": 12900
},
{
"epoch": 4.21,
"learning_rate": 1.3670886075949367e-05,
"loss": 1.9812,
"step": 13000
},
{
"epoch": 4.21,
"eval_loss": 1.266192078590393,
"eval_runtime": 1008.4278,
"eval_samples_per_second": 15.808,
"eval_steps_per_second": 0.989,
"eval_wer": 0.7696760526197176,
"step": 13000
},
{
"epoch": 4.24,
"learning_rate": 1.3112434847356663e-05,
"loss": 2.1185,
"step": 13100
},
{
"epoch": 4.28,
"learning_rate": 1.255398361876396e-05,
"loss": 2.2783,
"step": 13200
},
{
"epoch": 4.31,
"learning_rate": 1.2001116902457184e-05,
"loss": 2.3792,
"step": 13300
},
{
"epoch": 4.34,
"learning_rate": 1.1442665673864482e-05,
"loss": 2.4678,
"step": 13400
},
{
"epoch": 4.37,
"learning_rate": 1.0884214445271778e-05,
"loss": 2.5792,
"step": 13500
},
{
"epoch": 4.41,
"learning_rate": 1.0325763216679075e-05,
"loss": 2.6183,
"step": 13600
},
{
"epoch": 4.44,
"learning_rate": 9.767311988086373e-06,
"loss": 2.6677,
"step": 13700
},
{
"epoch": 4.47,
"learning_rate": 9.20886075949367e-06,
"loss": 2.7043,
"step": 13800
},
{
"epoch": 4.5,
"learning_rate": 8.650409530900967e-06,
"loss": 2.6737,
"step": 13900
},
{
"epoch": 4.54,
"learning_rate": 8.091958302308264e-06,
"loss": 2.6855,
"step": 14000
},
{
"epoch": 4.54,
"eval_loss": 2.4120402336120605,
"eval_runtime": 1002.3228,
"eval_samples_per_second": 15.904,
"eval_steps_per_second": 0.995,
"eval_wer": 0.9901740468632286,
"step": 14000
},
{
"epoch": 4.57,
"learning_rate": 7.533507073715562e-06,
"loss": 2.7223,
"step": 14100
},
{
"epoch": 4.6,
"learning_rate": 6.975055845122859e-06,
"loss": 2.7861,
"step": 14200
},
{
"epoch": 4.63,
"learning_rate": 6.416604616530156e-06,
"loss": 2.7397,
"step": 14300
},
{
"epoch": 4.67,
"learning_rate": 5.8581533879374525e-06,
"loss": 2.7574,
"step": 14400
},
{
"epoch": 4.7,
"learning_rate": 5.3052866716306765e-06,
"loss": 2.8119,
"step": 14500
},
{
"epoch": 4.73,
"learning_rate": 4.746835443037975e-06,
"loss": 2.725,
"step": 14600
},
{
"epoch": 4.76,
"learning_rate": 4.188384214445272e-06,
"loss": 2.7319,
"step": 14700
},
{
"epoch": 4.8,
"learning_rate": 3.629932985852568e-06,
"loss": 2.7426,
"step": 14800
},
{
"epoch": 4.83,
"learning_rate": 3.071481757259866e-06,
"loss": 2.7441,
"step": 14900
},
{
"epoch": 4.86,
"learning_rate": 2.513030528667163e-06,
"loss": 2.7482,
"step": 15000
},
{
"epoch": 4.86,
"eval_loss": 2.5340933799743652,
"eval_runtime": 1000.6978,
"eval_samples_per_second": 15.93,
"eval_steps_per_second": 0.996,
"eval_wer": 0.9874181438864671,
"step": 15000
},
{
"epoch": 4.89,
"learning_rate": 1.95457930007446e-06,
"loss": 2.752,
"step": 15100
},
{
"epoch": 4.93,
"learning_rate": 1.3961280714817571e-06,
"loss": 2.7476,
"step": 15200
},
{
"epoch": 4.96,
"learning_rate": 8.376768428890544e-07,
"loss": 2.7484,
"step": 15300
},
{
"epoch": 4.99,
"learning_rate": 2.7922561429635143e-07,
"loss": 2.7494,
"step": 15400
},
{
"epoch": 5.0,
"step": 15430,
"total_flos": 8.828199178194905e+20,
"train_loss": 1.7380384819737829,
"train_runtime": 214447.5226,
"train_samples_per_second": 9.211,
"train_steps_per_second": 0.072
}
],
"max_steps": 15430,
"num_train_epochs": 5,
"total_flos": 8.828199178194905e+20,
"trial_name": null,
"trial_params": null
}