{ "best_metric": 5.11147362697118, "best_model_checkpoint": "./final-whisper-for-initial-publish-v2/checkpoint-1000", "epoch": 1.6474464579901154, "eval_steps": 500, "global_step": 1000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04118616144975288, "grad_norm": 67.08155822753906, "learning_rate": 4.2000000000000006e-07, "loss": 2.8368, "step": 25 }, { "epoch": 0.08237232289950576, "grad_norm": 17.44784164428711, "learning_rate": 9.200000000000001e-07, "loss": 2.0282, "step": 50 }, { "epoch": 0.12355848434925865, "grad_norm": 12.481098175048828, "learning_rate": 1.42e-06, "loss": 1.3152, "step": 75 }, { "epoch": 0.16474464579901152, "grad_norm": 8.900260925292969, "learning_rate": 1.9200000000000003e-06, "loss": 0.7248, "step": 100 }, { "epoch": 0.20593080724876442, "grad_norm": 7.8956146240234375, "learning_rate": 2.42e-06, "loss": 0.5731, "step": 125 }, { "epoch": 0.2471169686985173, "grad_norm": 9.542287826538086, "learning_rate": 2.92e-06, "loss": 0.474, "step": 150 }, { "epoch": 0.2883031301482702, "grad_norm": 6.937229633331299, "learning_rate": 3.4200000000000007e-06, "loss": 0.4301, "step": 175 }, { "epoch": 0.32948929159802304, "grad_norm": 6.919783115386963, "learning_rate": 3.920000000000001e-06, "loss": 0.351, "step": 200 }, { "epoch": 0.37067545304777594, "grad_norm": 6.427859783172607, "learning_rate": 4.42e-06, "loss": 0.2726, "step": 225 }, { "epoch": 0.41186161449752884, "grad_norm": 5.83529806137085, "learning_rate": 4.92e-06, "loss": 0.1663, "step": 250 }, { "epoch": 0.45304777594728174, "grad_norm": 2.4859721660614014, "learning_rate": 5.420000000000001e-06, "loss": 0.0893, "step": 275 }, { "epoch": 0.4942339373970346, "grad_norm": 3.2526772022247314, "learning_rate": 5.92e-06, "loss": 0.076, "step": 300 }, { "epoch": 0.5354200988467874, "grad_norm": 4.417870044708252, "learning_rate": 6.42e-06, "loss": 0.0764, "step": 325 }, { "epoch": 0.5766062602965404, "grad_norm": 3.149733781814575, "learning_rate": 6.92e-06, "loss": 0.0679, "step": 350 }, { "epoch": 0.6177924217462932, "grad_norm": 3.8823468685150146, "learning_rate": 7.420000000000001e-06, "loss": 0.0525, "step": 375 }, { "epoch": 0.6589785831960461, "grad_norm": 6.1009745597839355, "learning_rate": 7.92e-06, "loss": 0.0552, "step": 400 }, { "epoch": 0.700164744645799, "grad_norm": 5.714720726013184, "learning_rate": 8.42e-06, "loss": 0.0529, "step": 425 }, { "epoch": 0.7413509060955519, "grad_norm": 4.403444290161133, "learning_rate": 8.920000000000001e-06, "loss": 0.0472, "step": 450 }, { "epoch": 0.7825370675453048, "grad_norm": 3.7249884605407715, "learning_rate": 9.42e-06, "loss": 0.0424, "step": 475 }, { "epoch": 0.8237232289950577, "grad_norm": 3.8490548133850098, "learning_rate": 9.920000000000002e-06, "loss": 0.0461, "step": 500 }, { "epoch": 0.8237232289950577, "eval_loss": 0.04240331053733826, "eval_runtime": 1653.3644, "eval_samples_per_second": 1.468, "eval_steps_per_second": 0.184, "eval_wer": 11.04637613609881, "step": 500 }, { "epoch": 0.8649093904448105, "grad_norm": 2.538515329360962, "learning_rate": 9.961818181818183e-06, "loss": 0.0384, "step": 525 }, { "epoch": 0.9060955518945635, "grad_norm": 1.6598188877105713, "learning_rate": 9.916363636363637e-06, "loss": 0.0375, "step": 550 }, { "epoch": 0.9472817133443163, "grad_norm": 3.256746292114258, "learning_rate": 9.870909090909092e-06, "loss": 0.0346, "step": 575 }, { "epoch": 0.9884678747940692, "grad_norm": 3.04457426071167, "learning_rate": 9.825454545454546e-06, "loss": 0.034, "step": 600 }, { "epoch": 1.029654036243822, "grad_norm": 0.6082770228385925, "learning_rate": 9.780000000000001e-06, "loss": 0.0272, "step": 625 }, { "epoch": 1.0708401976935749, "grad_norm": 1.6969329118728638, "learning_rate": 9.734545454545455e-06, "loss": 0.0206, "step": 650 }, { "epoch": 1.112026359143328, "grad_norm": 1.9652377367019653, "learning_rate": 9.68909090909091e-06, "loss": 0.0197, "step": 675 }, { "epoch": 1.1532125205930808, "grad_norm": 2.402329206466675, "learning_rate": 9.643636363636364e-06, "loss": 0.0221, "step": 700 }, { "epoch": 1.1943986820428336, "grad_norm": 3.167133092880249, "learning_rate": 9.598181818181818e-06, "loss": 0.0176, "step": 725 }, { "epoch": 1.2355848434925865, "grad_norm": 0.4075870215892792, "learning_rate": 9.552727272727273e-06, "loss": 0.0165, "step": 750 }, { "epoch": 1.2767710049423393, "grad_norm": 2.739176034927368, "learning_rate": 9.507272727272729e-06, "loss": 0.0156, "step": 775 }, { "epoch": 1.3179571663920924, "grad_norm": 2.4662702083587646, "learning_rate": 9.461818181818183e-06, "loss": 0.0133, "step": 800 }, { "epoch": 1.3591433278418452, "grad_norm": 2.955909252166748, "learning_rate": 9.416363636363636e-06, "loss": 0.0184, "step": 825 }, { "epoch": 1.400329489291598, "grad_norm": 2.0838463306427, "learning_rate": 9.370909090909092e-06, "loss": 0.0138, "step": 850 }, { "epoch": 1.441515650741351, "grad_norm": 2.2245779037475586, "learning_rate": 9.325454545454547e-06, "loss": 0.0132, "step": 875 }, { "epoch": 1.4827018121911038, "grad_norm": 2.146961212158203, "learning_rate": 9.280000000000001e-06, "loss": 0.0126, "step": 900 }, { "epoch": 1.5238879736408566, "grad_norm": 1.955930471420288, "learning_rate": 9.234545454545455e-06, "loss": 0.0111, "step": 925 }, { "epoch": 1.5650741350906094, "grad_norm": 2.04664945602417, "learning_rate": 9.18909090909091e-06, "loss": 0.0118, "step": 950 }, { "epoch": 1.6062602965403623, "grad_norm": 2.1349849700927734, "learning_rate": 9.143636363636364e-06, "loss": 0.0116, "step": 975 }, { "epoch": 1.6474464579901154, "grad_norm": 2.336024761199951, "learning_rate": 9.09818181818182e-06, "loss": 0.0124, "step": 1000 }, { "epoch": 1.6474464579901154, "eval_loss": 0.02211674302816391, "eval_runtime": 1633.7086, "eval_samples_per_second": 1.486, "eval_steps_per_second": 0.186, "eval_wer": 5.11147362697118, "step": 1000 } ], "logging_steps": 25, "max_steps": 6000, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 1000, "total_flos": 4.61621206351872e+18, "train_batch_size": 16, "trial_name": null, "trial_params": null }