XLS-R-300m-english-ipa / trainer_state.json
speech31's picture
upload model
77ea096
raw
history blame
13.1 kB
{
"best_metric": 0.1282939612865448,
"best_model_checkpoint": "XLSR-english-phoneme-v2/checkpoint-7500",
"epoch": 57.69230769230769,
"global_step": 7500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.77,
"learning_rate": 2.9999999999999997e-05,
"loss": 5.4253,
"step": 100
},
{
"epoch": 1.54,
"learning_rate": 5.9999999999999995e-05,
"loss": 2.4222,
"step": 200
},
{
"epoch": 2.31,
"learning_rate": 8.999999999999999e-05,
"loss": 2.2277,
"step": 300
},
{
"epoch": 3.08,
"learning_rate": 0.00011999999999999999,
"loss": 2.1449,
"step": 400
},
{
"epoch": 3.85,
"learning_rate": 0.00015,
"loss": 1.3868,
"step": 500
},
{
"epoch": 3.85,
"eval_cer": 0.25636523266022826,
"eval_loss": 0.6820988059043884,
"eval_runtime": 22.2857,
"eval_samples_per_second": 20.731,
"eval_steps_per_second": 2.603,
"step": 500
},
{
"epoch": 4.62,
"learning_rate": 0.00017999999999999998,
"loss": 0.5653,
"step": 600
},
{
"epoch": 5.38,
"learning_rate": 0.00020999999999999998,
"loss": 0.3812,
"step": 700
},
{
"epoch": 6.15,
"learning_rate": 0.00023999999999999998,
"loss": 0.2902,
"step": 800
},
{
"epoch": 6.92,
"learning_rate": 0.00027,
"loss": 0.2423,
"step": 900
},
{
"epoch": 7.69,
"learning_rate": 0.0003,
"loss": 0.2164,
"step": 1000
},
{
"epoch": 7.69,
"eval_cer": 0.09801261074307606,
"eval_loss": 0.20735949277877808,
"eval_runtime": 20.162,
"eval_samples_per_second": 22.914,
"eval_steps_per_second": 2.877,
"step": 1000
},
{
"epoch": 8.46,
"learning_rate": 0.00029558823529411763,
"loss": 0.2116,
"step": 1100
},
{
"epoch": 9.23,
"learning_rate": 0.00029117647058823524,
"loss": 0.1857,
"step": 1200
},
{
"epoch": 10.0,
"learning_rate": 0.00028676470588235296,
"loss": 0.1561,
"step": 1300
},
{
"epoch": 10.77,
"learning_rate": 0.00028235294117647056,
"loss": 0.1368,
"step": 1400
},
{
"epoch": 11.54,
"learning_rate": 0.0002779411764705882,
"loss": 0.1317,
"step": 1500
},
{
"epoch": 11.54,
"eval_cer": 0.07494612499002315,
"eval_loss": 0.16971056163311005,
"eval_runtime": 39.1088,
"eval_samples_per_second": 11.813,
"eval_steps_per_second": 1.483,
"step": 1500
},
{
"epoch": 12.31,
"learning_rate": 0.00027352941176470583,
"loss": 0.1151,
"step": 1600
},
{
"epoch": 13.08,
"learning_rate": 0.0002691176470588235,
"loss": 0.1042,
"step": 1700
},
{
"epoch": 13.85,
"learning_rate": 0.00026470588235294115,
"loss": 0.1044,
"step": 1800
},
{
"epoch": 14.62,
"learning_rate": 0.0002602941176470588,
"loss": 0.089,
"step": 1900
},
{
"epoch": 15.38,
"learning_rate": 0.0002558823529411764,
"loss": 0.0856,
"step": 2000
},
{
"epoch": 15.38,
"eval_cer": 0.0616968632771969,
"eval_loss": 0.1550830602645874,
"eval_runtime": 20.1049,
"eval_samples_per_second": 22.979,
"eval_steps_per_second": 2.885,
"step": 2000
},
{
"epoch": 16.15,
"learning_rate": 0.0002514705882352941,
"loss": 0.0821,
"step": 2100
},
{
"epoch": 16.92,
"learning_rate": 0.00024705882352941174,
"loss": 0.0816,
"step": 2200
},
{
"epoch": 17.69,
"learning_rate": 0.0002426470588235294,
"loss": 0.0708,
"step": 2300
},
{
"epoch": 18.46,
"learning_rate": 0.000238235294117647,
"loss": 0.0693,
"step": 2400
},
{
"epoch": 19.23,
"learning_rate": 0.0002338235294117647,
"loss": 0.0651,
"step": 2500
},
{
"epoch": 19.23,
"eval_cer": 0.06121797429962487,
"eval_loss": 0.15284913778305054,
"eval_runtime": 29.2987,
"eval_samples_per_second": 15.769,
"eval_steps_per_second": 1.98,
"step": 2500
},
{
"epoch": 20.0,
"learning_rate": 0.0002294117647058823,
"loss": 0.0597,
"step": 2600
},
{
"epoch": 20.77,
"learning_rate": 0.000225,
"loss": 0.0558,
"step": 2700
},
{
"epoch": 21.54,
"learning_rate": 0.00022058823529411765,
"loss": 0.0543,
"step": 2800
},
{
"epoch": 22.31,
"learning_rate": 0.00021617647058823526,
"loss": 0.0527,
"step": 2900
},
{
"epoch": 23.08,
"learning_rate": 0.00021176470588235295,
"loss": 0.0502,
"step": 3000
},
{
"epoch": 23.08,
"eval_cer": 0.05475297310240242,
"eval_loss": 0.15015824139118195,
"eval_runtime": 28.2669,
"eval_samples_per_second": 16.344,
"eval_steps_per_second": 2.052,
"step": 3000
},
{
"epoch": 23.85,
"learning_rate": 0.00020735294117647056,
"loss": 0.0503,
"step": 3100
},
{
"epoch": 24.62,
"learning_rate": 0.00020294117647058822,
"loss": 0.0513,
"step": 3200
},
{
"epoch": 25.38,
"learning_rate": 0.00019852941176470585,
"loss": 0.0469,
"step": 3300
},
{
"epoch": 26.15,
"learning_rate": 0.0001941176470588235,
"loss": 0.0417,
"step": 3400
},
{
"epoch": 26.92,
"learning_rate": 0.00018970588235294115,
"loss": 0.0407,
"step": 3500
},
{
"epoch": 26.92,
"eval_cer": 0.06385186367627105,
"eval_loss": 0.13443736732006073,
"eval_runtime": 20.0653,
"eval_samples_per_second": 23.025,
"eval_steps_per_second": 2.891,
"step": 3500
},
{
"epoch": 27.69,
"learning_rate": 0.0001852941176470588,
"loss": 0.0432,
"step": 3600
},
{
"epoch": 28.46,
"learning_rate": 0.00018088235294117644,
"loss": 0.0379,
"step": 3700
},
{
"epoch": 29.23,
"learning_rate": 0.0001764705882352941,
"loss": 0.0432,
"step": 3800
},
{
"epoch": 30.0,
"learning_rate": 0.00017205882352941174,
"loss": 0.0394,
"step": 3900
},
{
"epoch": 30.77,
"learning_rate": 0.0001676470588235294,
"loss": 0.0374,
"step": 4000
},
{
"epoch": 30.77,
"eval_cer": 0.058903344241360046,
"eval_loss": 0.12998488545417786,
"eval_runtime": 16.186,
"eval_samples_per_second": 28.543,
"eval_steps_per_second": 3.583,
"step": 4000
},
{
"epoch": 31.54,
"learning_rate": 0.00016323529411764703,
"loss": 0.0328,
"step": 4100
},
{
"epoch": 32.31,
"learning_rate": 0.0001588235294117647,
"loss": 0.034,
"step": 4200
},
{
"epoch": 33.08,
"learning_rate": 0.00015441176470588233,
"loss": 0.0334,
"step": 4300
},
{
"epoch": 33.85,
"learning_rate": 0.00015,
"loss": 0.0329,
"step": 4400
},
{
"epoch": 34.62,
"learning_rate": 0.00014558823529411762,
"loss": 0.0285,
"step": 4500
},
{
"epoch": 34.62,
"eval_cer": 0.04541463803974778,
"eval_loss": 0.14117006957530975,
"eval_runtime": 25.9649,
"eval_samples_per_second": 17.793,
"eval_steps_per_second": 2.234,
"step": 4500
},
{
"epoch": 35.38,
"learning_rate": 0.00014117647058823528,
"loss": 0.0266,
"step": 4600
},
{
"epoch": 36.15,
"learning_rate": 0.00013676470588235292,
"loss": 0.0257,
"step": 4700
},
{
"epoch": 36.92,
"learning_rate": 0.00013235294117647058,
"loss": 0.0255,
"step": 4800
},
{
"epoch": 37.69,
"learning_rate": 0.0001279411764705882,
"loss": 0.0254,
"step": 4900
},
{
"epoch": 38.46,
"learning_rate": 0.00012352941176470587,
"loss": 0.0212,
"step": 5000
},
{
"epoch": 38.46,
"eval_cer": 0.05012371298587277,
"eval_loss": 0.14315766096115112,
"eval_runtime": 26.1379,
"eval_samples_per_second": 17.675,
"eval_steps_per_second": 2.219,
"step": 5000
},
{
"epoch": 39.23,
"learning_rate": 0.0001191176470588235,
"loss": 0.0205,
"step": 5100
},
{
"epoch": 40.0,
"learning_rate": 0.00011470588235294115,
"loss": 0.0195,
"step": 5200
},
{
"epoch": 40.77,
"learning_rate": 0.00011029411764705883,
"loss": 0.0209,
"step": 5300
},
{
"epoch": 41.54,
"learning_rate": 0.00010588235294117647,
"loss": 0.0178,
"step": 5400
},
{
"epoch": 42.31,
"learning_rate": 0.00010147058823529411,
"loss": 0.016,
"step": 5500
},
{
"epoch": 42.31,
"eval_cer": 0.05722723281985793,
"eval_loss": 0.14901842176914215,
"eval_runtime": 26.3386,
"eval_samples_per_second": 17.541,
"eval_steps_per_second": 2.202,
"step": 5500
},
{
"epoch": 43.08,
"learning_rate": 9.705882352941176e-05,
"loss": 0.0177,
"step": 5600
},
{
"epoch": 43.85,
"learning_rate": 9.26470588235294e-05,
"loss": 0.0159,
"step": 5700
},
{
"epoch": 44.62,
"learning_rate": 8.823529411764705e-05,
"loss": 0.0181,
"step": 5800
},
{
"epoch": 45.38,
"learning_rate": 8.38235294117647e-05,
"loss": 0.0158,
"step": 5900
},
{
"epoch": 46.15,
"learning_rate": 7.941176470588235e-05,
"loss": 0.0153,
"step": 6000
},
{
"epoch": 46.15,
"eval_cer": 0.05068241679304015,
"eval_loss": 0.1339481920003891,
"eval_runtime": 26.3988,
"eval_samples_per_second": 17.501,
"eval_steps_per_second": 2.197,
"step": 6000
},
{
"epoch": 46.92,
"learning_rate": 7.5e-05,
"loss": 0.0142,
"step": 6100
},
{
"epoch": 47.69,
"learning_rate": 7.058823529411764e-05,
"loss": 0.0138,
"step": 6200
},
{
"epoch": 48.46,
"learning_rate": 6.617647058823529e-05,
"loss": 0.0122,
"step": 6300
},
{
"epoch": 49.23,
"learning_rate": 6.176470588235294e-05,
"loss": 0.0125,
"step": 6400
},
{
"epoch": 50.0,
"learning_rate": 5.7352941176470576e-05,
"loss": 0.0111,
"step": 6500
},
{
"epoch": 50.0,
"eval_cer": 0.05004389815627744,
"eval_loss": 0.13527342677116394,
"eval_runtime": 16.9816,
"eval_samples_per_second": 27.206,
"eval_steps_per_second": 3.415,
"step": 6500
},
{
"epoch": 50.77,
"learning_rate": 5.294117647058824e-05,
"loss": 0.0128,
"step": 6600
},
{
"epoch": 51.54,
"learning_rate": 4.85735294117647e-05,
"loss": 0.0098,
"step": 6700
},
{
"epoch": 52.31,
"learning_rate": 4.416176470588235e-05,
"loss": 0.0114,
"step": 6800
},
{
"epoch": 53.08,
"learning_rate": 3.975e-05,
"loss": 0.0105,
"step": 6900
},
{
"epoch": 53.85,
"learning_rate": 3.5338235294117645e-05,
"loss": 0.0103,
"step": 7000
},
{
"epoch": 53.85,
"eval_cer": 0.044616489743794396,
"eval_loss": 0.1340005248785019,
"eval_runtime": 16.089,
"eval_samples_per_second": 28.715,
"eval_steps_per_second": 3.605,
"step": 7000
},
{
"epoch": 54.62,
"learning_rate": 3.092647058823529e-05,
"loss": 0.0083,
"step": 7100
},
{
"epoch": 55.38,
"learning_rate": 2.651470588235294e-05,
"loss": 0.0085,
"step": 7200
},
{
"epoch": 56.15,
"learning_rate": 2.2102941176470584e-05,
"loss": 0.0082,
"step": 7300
},
{
"epoch": 56.92,
"learning_rate": 1.7691176470588234e-05,
"loss": 0.0074,
"step": 7400
},
{
"epoch": 57.69,
"learning_rate": 1.3279411764705882e-05,
"loss": 0.0072,
"step": 7500
},
{
"epoch": 57.69,
"eval_cer": 0.04333945247026898,
"eval_loss": 0.1282939612865448,
"eval_runtime": 16.8303,
"eval_samples_per_second": 27.45,
"eval_steps_per_second": 3.446,
"step": 7500
}
],
"max_steps": 7800,
"num_train_epochs": 60,
"total_flos": 2.2858916955207467e+19,
"trial_name": null,
"trial_params": null
}