| { | |
| "best_metric": 94.92795389048992, | |
| "best_model_checkpoint": "./whisper-tiny-en/checkpoint-2000", | |
| "epoch": 48.78048780487805, | |
| "eval_steps": 1000, | |
| "global_step": 2000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.6097560975609756, | |
| "grad_norm": 76.21949768066406, | |
| "learning_rate": 4.2000000000000006e-07, | |
| "loss": 7.5702, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 1.2195121951219512, | |
| "grad_norm": 72.27240753173828, | |
| "learning_rate": 9.200000000000001e-07, | |
| "loss": 7.1361, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.8292682926829267, | |
| "grad_norm": 27.645973205566406, | |
| "learning_rate": 1.42e-06, | |
| "loss": 6.4445, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 2.4390243902439024, | |
| "grad_norm": 21.712514877319336, | |
| "learning_rate": 1.9200000000000003e-06, | |
| "loss": 5.8148, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 3.048780487804878, | |
| "grad_norm": 19.066011428833008, | |
| "learning_rate": 2.42e-06, | |
| "loss": 5.1377, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 3.658536585365854, | |
| "grad_norm": 14.726506233215332, | |
| "learning_rate": 2.92e-06, | |
| "loss": 4.6, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 4.2682926829268295, | |
| "grad_norm": 13.597981452941895, | |
| "learning_rate": 3.4200000000000007e-06, | |
| "loss": 4.3595, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 4.878048780487805, | |
| "grad_norm": 16.399568557739258, | |
| "learning_rate": 3.920000000000001e-06, | |
| "loss": 4.171, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 5.487804878048781, | |
| "grad_norm": 15.264267921447754, | |
| "learning_rate": 4.42e-06, | |
| "loss": 3.9727, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 6.097560975609756, | |
| "grad_norm": 14.83893871307373, | |
| "learning_rate": 4.92e-06, | |
| "loss": 3.816, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 6.7073170731707314, | |
| "grad_norm": 14.460186004638672, | |
| "learning_rate": 5.420000000000001e-06, | |
| "loss": 3.6683, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 7.317073170731708, | |
| "grad_norm": 13.663064002990723, | |
| "learning_rate": 5.92e-06, | |
| "loss": 3.5392, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 7.926829268292683, | |
| "grad_norm": 14.359073638916016, | |
| "learning_rate": 6.42e-06, | |
| "loss": 3.4807, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 8.536585365853659, | |
| "grad_norm": 12.90373706817627, | |
| "learning_rate": 6.92e-06, | |
| "loss": 3.335, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 9.146341463414634, | |
| "grad_norm": 15.809270858764648, | |
| "learning_rate": 7.420000000000001e-06, | |
| "loss": 3.2196, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 9.75609756097561, | |
| "grad_norm": 14.483990669250488, | |
| "learning_rate": 7.92e-06, | |
| "loss": 3.1224, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 10.365853658536585, | |
| "grad_norm": 15.841504096984863, | |
| "learning_rate": 8.42e-06, | |
| "loss": 2.9858, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 10.975609756097562, | |
| "grad_norm": 16.420425415039062, | |
| "learning_rate": 8.920000000000001e-06, | |
| "loss": 2.9395, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 11.585365853658537, | |
| "grad_norm": 14.952699661254883, | |
| "learning_rate": 9.42e-06, | |
| "loss": 2.7501, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 12.195121951219512, | |
| "grad_norm": 15.765605926513672, | |
| "learning_rate": 9.920000000000002e-06, | |
| "loss": 2.7014, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 12.804878048780488, | |
| "grad_norm": 17.627931594848633, | |
| "learning_rate": 9.967692307692308e-06, | |
| "loss": 2.601, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 13.414634146341463, | |
| "grad_norm": 16.36449432373047, | |
| "learning_rate": 9.92923076923077e-06, | |
| "loss": 2.4699, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 14.024390243902438, | |
| "grad_norm": 15.8140869140625, | |
| "learning_rate": 9.890769230769231e-06, | |
| "loss": 2.3859, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 14.634146341463415, | |
| "grad_norm": 15.983495712280273, | |
| "learning_rate": 9.852307692307693e-06, | |
| "loss": 2.2235, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 15.24390243902439, | |
| "grad_norm": 15.25550651550293, | |
| "learning_rate": 9.813846153846155e-06, | |
| "loss": 2.1583, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 15.853658536585366, | |
| "grad_norm": 18.167879104614258, | |
| "learning_rate": 9.775384615384616e-06, | |
| "loss": 2.0671, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 16.463414634146343, | |
| "grad_norm": 16.86029624938965, | |
| "learning_rate": 9.736923076923078e-06, | |
| "loss": 1.9547, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 17.073170731707318, | |
| "grad_norm": 16.195270538330078, | |
| "learning_rate": 9.698461538461539e-06, | |
| "loss": 1.8877, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 17.682926829268293, | |
| "grad_norm": 17.331113815307617, | |
| "learning_rate": 9.66e-06, | |
| "loss": 1.7336, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 18.29268292682927, | |
| "grad_norm": 18.00438690185547, | |
| "learning_rate": 9.621538461538463e-06, | |
| "loss": 1.6845, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 18.902439024390244, | |
| "grad_norm": 17.1490478515625, | |
| "learning_rate": 9.583076923076923e-06, | |
| "loss": 1.6027, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 19.51219512195122, | |
| "grad_norm": 17.924156188964844, | |
| "learning_rate": 9.544615384615385e-06, | |
| "loss": 1.4647, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 20.121951219512194, | |
| "grad_norm": 17.920028686523438, | |
| "learning_rate": 9.506153846153848e-06, | |
| "loss": 1.4291, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 20.73170731707317, | |
| "grad_norm": 17.502954483032227, | |
| "learning_rate": 9.467692307692308e-06, | |
| "loss": 1.3113, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 21.341463414634145, | |
| "grad_norm": 18.125986099243164, | |
| "learning_rate": 9.42923076923077e-06, | |
| "loss": 1.232, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 21.951219512195124, | |
| "grad_norm": 16.026338577270508, | |
| "learning_rate": 9.39076923076923e-06, | |
| "loss": 1.1903, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 22.5609756097561, | |
| "grad_norm": 16.795467376708984, | |
| "learning_rate": 9.352307692307693e-06, | |
| "loss": 1.0742, | |
| "step": 925 | |
| }, | |
| { | |
| "epoch": 23.170731707317074, | |
| "grad_norm": 19.86144256591797, | |
| "learning_rate": 9.313846153846155e-06, | |
| "loss": 1.0346, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 23.78048780487805, | |
| "grad_norm": 17.94236183166504, | |
| "learning_rate": 9.275384615384616e-06, | |
| "loss": 0.9589, | |
| "step": 975 | |
| }, | |
| { | |
| "epoch": 24.390243902439025, | |
| "grad_norm": 17.72712516784668, | |
| "learning_rate": 9.236923076923078e-06, | |
| "loss": 0.8757, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 24.390243902439025, | |
| "eval_loss": 4.123460292816162, | |
| "eval_runtime": 64.6053, | |
| "eval_samples_per_second": 2.492, | |
| "eval_steps_per_second": 0.325, | |
| "eval_wer": 97.89625360230548, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "grad_norm": 31.632688522338867, | |
| "learning_rate": 9.19846153846154e-06, | |
| "loss": 0.8421, | |
| "step": 1025 | |
| }, | |
| { | |
| "epoch": 25.609756097560975, | |
| "grad_norm": 18.91576385498047, | |
| "learning_rate": 9.16e-06, | |
| "loss": 0.7294, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 26.21951219512195, | |
| "grad_norm": 20.973003387451172, | |
| "learning_rate": 9.121538461538463e-06, | |
| "loss": 0.7139, | |
| "step": 1075 | |
| }, | |
| { | |
| "epoch": 26.829268292682926, | |
| "grad_norm": 18.86784553527832, | |
| "learning_rate": 9.083076923076923e-06, | |
| "loss": 0.6638, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 27.4390243902439, | |
| "grad_norm": 14.628901481628418, | |
| "learning_rate": 9.044615384615385e-06, | |
| "loss": 0.605, | |
| "step": 1125 | |
| }, | |
| { | |
| "epoch": 28.048780487804876, | |
| "grad_norm": 13.873974800109863, | |
| "learning_rate": 9.006153846153847e-06, | |
| "loss": 0.5613, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 28.658536585365855, | |
| "grad_norm": 15.853562355041504, | |
| "learning_rate": 8.967692307692308e-06, | |
| "loss": 0.4886, | |
| "step": 1175 | |
| }, | |
| { | |
| "epoch": 29.26829268292683, | |
| "grad_norm": 12.60745906829834, | |
| "learning_rate": 8.92923076923077e-06, | |
| "loss": 0.4652, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 29.878048780487806, | |
| "grad_norm": 17.16914176940918, | |
| "learning_rate": 8.890769230769232e-06, | |
| "loss": 0.4324, | |
| "step": 1225 | |
| }, | |
| { | |
| "epoch": 30.48780487804878, | |
| "grad_norm": 11.67048168182373, | |
| "learning_rate": 8.852307692307693e-06, | |
| "loss": 0.3793, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 31.097560975609756, | |
| "grad_norm": 10.136366844177246, | |
| "learning_rate": 8.813846153846155e-06, | |
| "loss": 0.3689, | |
| "step": 1275 | |
| }, | |
| { | |
| "epoch": 31.70731707317073, | |
| "grad_norm": 15.549346923828125, | |
| "learning_rate": 8.775384615384615e-06, | |
| "loss": 0.3165, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 32.31707317073171, | |
| "grad_norm": 10.75722599029541, | |
| "learning_rate": 8.736923076923077e-06, | |
| "loss": 0.2891, | |
| "step": 1325 | |
| }, | |
| { | |
| "epoch": 32.926829268292686, | |
| "grad_norm": 10.687921524047852, | |
| "learning_rate": 8.69846153846154e-06, | |
| "loss": 0.2786, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 33.53658536585366, | |
| "grad_norm": 15.220375061035156, | |
| "learning_rate": 8.66e-06, | |
| "loss": 0.2349, | |
| "step": 1375 | |
| }, | |
| { | |
| "epoch": 34.146341463414636, | |
| "grad_norm": 9.226957321166992, | |
| "learning_rate": 8.621538461538462e-06, | |
| "loss": 0.2279, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 34.75609756097561, | |
| "grad_norm": 10.428820610046387, | |
| "learning_rate": 8.583076923076924e-06, | |
| "loss": 0.2022, | |
| "step": 1425 | |
| }, | |
| { | |
| "epoch": 35.36585365853659, | |
| "grad_norm": 10.70584487915039, | |
| "learning_rate": 8.544615384615385e-06, | |
| "loss": 0.186, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 35.97560975609756, | |
| "grad_norm": 11.436528205871582, | |
| "learning_rate": 8.506153846153847e-06, | |
| "loss": 0.1763, | |
| "step": 1475 | |
| }, | |
| { | |
| "epoch": 36.58536585365854, | |
| "grad_norm": 8.337956428527832, | |
| "learning_rate": 8.467692307692308e-06, | |
| "loss": 0.1482, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 37.19512195121951, | |
| "grad_norm": 8.083637237548828, | |
| "learning_rate": 8.42923076923077e-06, | |
| "loss": 0.1515, | |
| "step": 1525 | |
| }, | |
| { | |
| "epoch": 37.80487804878049, | |
| "grad_norm": 8.621879577636719, | |
| "learning_rate": 8.390769230769232e-06, | |
| "loss": 0.1345, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 38.41463414634146, | |
| "grad_norm": 8.559672355651855, | |
| "learning_rate": 8.352307692307692e-06, | |
| "loss": 0.1204, | |
| "step": 1575 | |
| }, | |
| { | |
| "epoch": 39.02439024390244, | |
| "grad_norm": 6.850173473358154, | |
| "learning_rate": 8.313846153846155e-06, | |
| "loss": 0.1208, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 39.63414634146341, | |
| "grad_norm": 8.43338680267334, | |
| "learning_rate": 8.275384615384617e-06, | |
| "loss": 0.1007, | |
| "step": 1625 | |
| }, | |
| { | |
| "epoch": 40.24390243902439, | |
| "grad_norm": 9.348348617553711, | |
| "learning_rate": 8.236923076923077e-06, | |
| "loss": 0.0995, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 40.853658536585364, | |
| "grad_norm": 6.383693695068359, | |
| "learning_rate": 8.19846153846154e-06, | |
| "loss": 0.0956, | |
| "step": 1675 | |
| }, | |
| { | |
| "epoch": 41.46341463414634, | |
| "grad_norm": 6.797173976898193, | |
| "learning_rate": 8.16e-06, | |
| "loss": 0.0863, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 42.073170731707314, | |
| "grad_norm": 5.753118515014648, | |
| "learning_rate": 8.121538461538462e-06, | |
| "loss": 0.0874, | |
| "step": 1725 | |
| }, | |
| { | |
| "epoch": 42.68292682926829, | |
| "grad_norm": 8.442388534545898, | |
| "learning_rate": 8.083076923076924e-06, | |
| "loss": 0.0737, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 43.292682926829265, | |
| "grad_norm": 4.439198017120361, | |
| "learning_rate": 8.044615384615385e-06, | |
| "loss": 0.0733, | |
| "step": 1775 | |
| }, | |
| { | |
| "epoch": 43.90243902439025, | |
| "grad_norm": 6.787204742431641, | |
| "learning_rate": 8.006153846153847e-06, | |
| "loss": 0.0714, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 44.51219512195122, | |
| "grad_norm": 5.47157621383667, | |
| "learning_rate": 7.967692307692309e-06, | |
| "loss": 0.0627, | |
| "step": 1825 | |
| }, | |
| { | |
| "epoch": 45.1219512195122, | |
| "grad_norm": 5.718803882598877, | |
| "learning_rate": 7.92923076923077e-06, | |
| "loss": 0.0666, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 45.73170731707317, | |
| "grad_norm": 4.490105628967285, | |
| "learning_rate": 7.890769230769232e-06, | |
| "loss": 0.055, | |
| "step": 1875 | |
| }, | |
| { | |
| "epoch": 46.34146341463415, | |
| "grad_norm": 5.873692035675049, | |
| "learning_rate": 7.852307692307692e-06, | |
| "loss": 0.0569, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 46.951219512195124, | |
| "grad_norm": 10.3510103225708, | |
| "learning_rate": 7.813846153846154e-06, | |
| "loss": 0.0601, | |
| "step": 1925 | |
| }, | |
| { | |
| "epoch": 47.5609756097561, | |
| "grad_norm": 5.004169464111328, | |
| "learning_rate": 7.775384615384616e-06, | |
| "loss": 0.0524, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 48.170731707317074, | |
| "grad_norm": 3.3900997638702393, | |
| "learning_rate": 7.736923076923077e-06, | |
| "loss": 0.0508, | |
| "step": 1975 | |
| }, | |
| { | |
| "epoch": 48.78048780487805, | |
| "grad_norm": 6.310654640197754, | |
| "learning_rate": 7.698461538461539e-06, | |
| "loss": 0.0518, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 48.78048780487805, | |
| "eval_loss": 4.87410306930542, | |
| "eval_runtime": 62.3697, | |
| "eval_samples_per_second": 2.581, | |
| "eval_steps_per_second": 0.337, | |
| "eval_wer": 94.92795389048992, | |
| "step": 2000 | |
| } | |
| ], | |
| "logging_steps": 25, | |
| "max_steps": 7000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 171, | |
| "save_steps": 1000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 7.7244213952512e+17, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |