| { | |
| "best_metric": 160.91330033530258, | |
| "best_model_checkpoint": "./whisper-training-blog/checkpoint-396", | |
| "epoch": 2.330357142857143, | |
| "global_step": 448, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 1.6666666666666668e-07, | |
| "loss": 1.7185, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 3.3333333333333335e-07, | |
| "loss": 1.6809, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 5.555555555555555e-07, | |
| "loss": 1.712, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 7.777777777777778e-07, | |
| "loss": 1.543, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 1e-06, | |
| "loss": 1.6091, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 1.2222222222222223e-06, | |
| "loss": 1.5657, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 1.4444444444444445e-06, | |
| "loss": 1.6081, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 1.6666666666666667e-06, | |
| "loss": 1.5076, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 1.8888888888888888e-06, | |
| "loss": 1.5183, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "learning_rate": 2.1111111111111114e-06, | |
| "loss": 1.4948, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 2.3333333333333336e-06, | |
| "loss": 1.4112, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "eval_loss": 1.4919402599334717, | |
| "eval_runtime": 119.3982, | |
| "eval_samples_per_second": 2.764, | |
| "eval_steps_per_second": 0.352, | |
| "eval_wer": 245.34568098355422, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "learning_rate": 2.5555555555555557e-06, | |
| "loss": 1.3193, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 2.7777777777777775e-06, | |
| "loss": 1.3334, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 3e-06, | |
| "loss": 1.2896, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 3.2222222222222222e-06, | |
| "loss": 1.2749, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 3.4444444444444444e-06, | |
| "loss": 1.3855, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 3.6666666666666666e-06, | |
| "loss": 1.2078, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 3.888888888888889e-06, | |
| "loss": 1.2185, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 4.111111111111111e-06, | |
| "loss": 1.1663, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.333333333333333e-06, | |
| "loss": 1.1352, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.555555555555555e-06, | |
| "loss": 1.0731, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 4.777777777777778e-06, | |
| "loss": 1.0502, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "eval_loss": 1.2254914045333862, | |
| "eval_runtime": 123.9853, | |
| "eval_samples_per_second": 2.662, | |
| "eval_steps_per_second": 0.339, | |
| "eval_wer": 220.1500878173399, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.9999999999999996e-06, | |
| "loss": 1.128, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 5.2222222222222226e-06, | |
| "loss": 1.0564, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 5.444444444444445e-06, | |
| "loss": 0.993, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 5.666666666666667e-06, | |
| "loss": 1.0879, | |
| "step": 104 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 5.888888888888889e-06, | |
| "loss": 1.0417, | |
| "step": 108 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 6.111111111111111e-06, | |
| "loss": 0.9337, | |
| "step": 112 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 6.333333333333333e-06, | |
| "loss": 0.9687, | |
| "step": 116 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 6.555555555555556e-06, | |
| "loss": 0.9558, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 6.777777777777778e-06, | |
| "loss": 1.0192, | |
| "step": 124 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 7e-06, | |
| "loss": 0.8509, | |
| "step": 128 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 7.222222222222222e-06, | |
| "loss": 0.9033, | |
| "step": 132 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "eval_loss": 1.1202749013900757, | |
| "eval_runtime": 108.7054, | |
| "eval_samples_per_second": 3.036, | |
| "eval_steps_per_second": 0.386, | |
| "eval_wer": 206.24301452977804, | |
| "step": 132 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 7.444444444444444e-06, | |
| "loss": 0.917, | |
| "step": 136 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 7.428115015974441e-06, | |
| "loss": 0.9444, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 7.332268370607029e-06, | |
| "loss": 0.8927, | |
| "step": 144 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 7.236421725239617e-06, | |
| "loss": 0.9975, | |
| "step": 148 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 7.140575079872205e-06, | |
| "loss": 1.2391, | |
| "step": 152 | |
| }, | |
| { | |
| "epoch": 1.01, | |
| "learning_rate": 7.044728434504793e-06, | |
| "loss": 0.9148, | |
| "step": 156 | |
| }, | |
| { | |
| "epoch": 1.02, | |
| "learning_rate": 6.948881789137381e-06, | |
| "loss": 1.0116, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.03, | |
| "learning_rate": 6.853035143769968e-06, | |
| "loss": 0.9082, | |
| "step": 164 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "learning_rate": 6.7571884984025565e-06, | |
| "loss": 0.8434, | |
| "step": 168 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "learning_rate": 6.6613418530351436e-06, | |
| "loss": 0.8671, | |
| "step": 172 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "learning_rate": 6.565495207667732e-06, | |
| "loss": 0.8141, | |
| "step": 176 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "eval_loss": 1.067484736442566, | |
| "eval_runtime": 108.347, | |
| "eval_samples_per_second": 3.046, | |
| "eval_steps_per_second": 0.388, | |
| "eval_wer": 201.96391505668208, | |
| "step": 176 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "learning_rate": 6.469648562300319e-06, | |
| "loss": 0.8329, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.08, | |
| "learning_rate": 6.373801916932908e-06, | |
| "loss": 0.8191, | |
| "step": 184 | |
| }, | |
| { | |
| "epoch": 1.08, | |
| "learning_rate": 6.277955271565495e-06, | |
| "loss": 0.8659, | |
| "step": 188 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "learning_rate": 6.182108626198084e-06, | |
| "loss": 0.8211, | |
| "step": 192 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "learning_rate": 6.086261980830671e-06, | |
| "loss": 0.8517, | |
| "step": 196 | |
| }, | |
| { | |
| "epoch": 1.11, | |
| "learning_rate": 5.990415335463259e-06, | |
| "loss": 0.7814, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.12, | |
| "learning_rate": 5.894568690095847e-06, | |
| "loss": 0.7542, | |
| "step": 204 | |
| }, | |
| { | |
| "epoch": 1.13, | |
| "learning_rate": 5.798722044728435e-06, | |
| "loss": 0.7938, | |
| "step": 208 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "learning_rate": 5.702875399361023e-06, | |
| "loss": 0.8181, | |
| "step": 212 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "learning_rate": 5.607028753993611e-06, | |
| "loss": 0.7462, | |
| "step": 216 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "learning_rate": 5.5111821086261985e-06, | |
| "loss": 0.8029, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "eval_loss": 1.0393874645233154, | |
| "eval_runtime": 100.1655, | |
| "eval_samples_per_second": 3.295, | |
| "eval_steps_per_second": 0.419, | |
| "eval_wer": 178.36500079833945, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "learning_rate": 5.415335463258786e-06, | |
| "loss": 0.754, | |
| "step": 224 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "learning_rate": 5.319488817891374e-06, | |
| "loss": 0.7663, | |
| "step": 228 | |
| }, | |
| { | |
| "epoch": 1.18, | |
| "learning_rate": 5.223642172523962e-06, | |
| "loss": 0.7254, | |
| "step": 232 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "learning_rate": 5.127795527156549e-06, | |
| "loss": 0.6944, | |
| "step": 236 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "learning_rate": 5.031948881789138e-06, | |
| "loss": 0.7281, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.21, | |
| "learning_rate": 4.936102236421725e-06, | |
| "loss": 0.7842, | |
| "step": 244 | |
| }, | |
| { | |
| "epoch": 1.22, | |
| "learning_rate": 4.840255591054313e-06, | |
| "loss": 0.7154, | |
| "step": 248 | |
| }, | |
| { | |
| "epoch": 1.23, | |
| "learning_rate": 4.744408945686901e-06, | |
| "loss": 0.7075, | |
| "step": 252 | |
| }, | |
| { | |
| "epoch": 1.24, | |
| "learning_rate": 4.648562300319489e-06, | |
| "loss": 0.7811, | |
| "step": 256 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 4.552715654952077e-06, | |
| "loss": 0.6817, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "learning_rate": 4.456869009584665e-06, | |
| "loss": 0.6324, | |
| "step": 264 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "eval_loss": 1.030145525932312, | |
| "eval_runtime": 114.3297, | |
| "eval_samples_per_second": 2.886, | |
| "eval_steps_per_second": 0.367, | |
| "eval_wer": 221.2996966310075, | |
| "step": 264 | |
| }, | |
| { | |
| "epoch": 1.26, | |
| "learning_rate": 4.361022364217253e-06, | |
| "loss": 0.6826, | |
| "step": 268 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "learning_rate": 4.26517571884984e-06, | |
| "loss": 0.7186, | |
| "step": 272 | |
| }, | |
| { | |
| "epoch": 1.28, | |
| "learning_rate": 4.169329073482428e-06, | |
| "loss": 0.6698, | |
| "step": 276 | |
| }, | |
| { | |
| "epoch": 1.29, | |
| "learning_rate": 4.0734824281150155e-06, | |
| "loss": 0.5678, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 1.3, | |
| "learning_rate": 3.977635782747604e-06, | |
| "loss": 0.6833, | |
| "step": 284 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "learning_rate": 3.881789137380191e-06, | |
| "loss": 0.6293, | |
| "step": 288 | |
| }, | |
| { | |
| "epoch": 1.32, | |
| "learning_rate": 3.7859424920127796e-06, | |
| "loss": 0.672, | |
| "step": 292 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 3.6900958466453675e-06, | |
| "loss": 0.7121, | |
| "step": 296 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 3.5942492012779555e-06, | |
| "loss": 0.8765, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.01, | |
| "learning_rate": 3.4984025559105434e-06, | |
| "loss": 0.6726, | |
| "step": 304 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "learning_rate": 3.4025559105431313e-06, | |
| "loss": 0.6972, | |
| "step": 308 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "eval_loss": 1.0134402513504028, | |
| "eval_runtime": 102.1008, | |
| "eval_samples_per_second": 3.232, | |
| "eval_steps_per_second": 0.411, | |
| "eval_wer": 176.67252115599553, | |
| "step": 308 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "learning_rate": 3.306709265175719e-06, | |
| "loss": 0.7431, | |
| "step": 312 | |
| }, | |
| { | |
| "epoch": 2.04, | |
| "learning_rate": 3.2108626198083067e-06, | |
| "loss": 0.6265, | |
| "step": 316 | |
| }, | |
| { | |
| "epoch": 2.04, | |
| "learning_rate": 3.1150159744408946e-06, | |
| "loss": 0.6606, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "learning_rate": 3.0191693290734825e-06, | |
| "loss": 0.6409, | |
| "step": 324 | |
| }, | |
| { | |
| "epoch": 2.06, | |
| "learning_rate": 2.9233226837060704e-06, | |
| "loss": 0.654, | |
| "step": 328 | |
| }, | |
| { | |
| "epoch": 2.07, | |
| "learning_rate": 2.8274760383386583e-06, | |
| "loss": 0.6111, | |
| "step": 332 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "learning_rate": 2.7316293929712462e-06, | |
| "loss": 0.6828, | |
| "step": 336 | |
| }, | |
| { | |
| "epoch": 2.09, | |
| "learning_rate": 2.635782747603834e-06, | |
| "loss": 0.6499, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "learning_rate": 2.539936102236422e-06, | |
| "loss": 0.6643, | |
| "step": 344 | |
| }, | |
| { | |
| "epoch": 2.11, | |
| "learning_rate": 2.44408945686901e-06, | |
| "loss": 0.6146, | |
| "step": 348 | |
| }, | |
| { | |
| "epoch": 2.12, | |
| "learning_rate": 2.3482428115015974e-06, | |
| "loss": 0.6052, | |
| "step": 352 | |
| }, | |
| { | |
| "epoch": 2.12, | |
| "eval_loss": 1.006484866142273, | |
| "eval_runtime": 105.4228, | |
| "eval_samples_per_second": 3.13, | |
| "eval_steps_per_second": 0.398, | |
| "eval_wer": 194.7149928149449, | |
| "step": 352 | |
| }, | |
| { | |
| "epoch": 2.12, | |
| "learning_rate": 2.2523961661341854e-06, | |
| "loss": 0.6237, | |
| "step": 356 | |
| }, | |
| { | |
| "epoch": 2.13, | |
| "learning_rate": 2.1565495207667733e-06, | |
| "loss": 0.5992, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.14, | |
| "learning_rate": 2.060702875399361e-06, | |
| "loss": 0.6777, | |
| "step": 364 | |
| }, | |
| { | |
| "epoch": 2.15, | |
| "learning_rate": 1.964856230031949e-06, | |
| "loss": 0.6017, | |
| "step": 368 | |
| }, | |
| { | |
| "epoch": 2.16, | |
| "learning_rate": 1.8690095846645368e-06, | |
| "loss": 0.622, | |
| "step": 372 | |
| }, | |
| { | |
| "epoch": 2.17, | |
| "learning_rate": 1.7731629392971245e-06, | |
| "loss": 0.6289, | |
| "step": 376 | |
| }, | |
| { | |
| "epoch": 2.18, | |
| "learning_rate": 1.6773162939297124e-06, | |
| "loss": 0.608, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 2.19, | |
| "learning_rate": 1.5814696485623003e-06, | |
| "loss": 0.5888, | |
| "step": 384 | |
| }, | |
| { | |
| "epoch": 2.2, | |
| "learning_rate": 1.4856230031948882e-06, | |
| "loss": 0.5932, | |
| "step": 388 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "learning_rate": 1.389776357827476e-06, | |
| "loss": 0.6373, | |
| "step": 392 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "learning_rate": 1.2939297124600638e-06, | |
| "loss": 0.6047, | |
| "step": 396 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "eval_loss": 1.002995491027832, | |
| "eval_runtime": 93.4309, | |
| "eval_samples_per_second": 3.532, | |
| "eval_steps_per_second": 0.45, | |
| "eval_wer": 160.91330033530258, | |
| "step": 396 | |
| }, | |
| { | |
| "epoch": 2.22, | |
| "learning_rate": 1.1980830670926517e-06, | |
| "loss": 0.5772, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.23, | |
| "learning_rate": 1.1022364217252397e-06, | |
| "loss": 0.6325, | |
| "step": 404 | |
| }, | |
| { | |
| "epoch": 2.24, | |
| "learning_rate": 1.0063897763578274e-06, | |
| "loss": 0.601, | |
| "step": 408 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "learning_rate": 9.105431309904153e-07, | |
| "loss": 0.5617, | |
| "step": 412 | |
| }, | |
| { | |
| "epoch": 2.26, | |
| "learning_rate": 8.146964856230032e-07, | |
| "loss": 0.5728, | |
| "step": 416 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "learning_rate": 7.188498402555911e-07, | |
| "loss": 0.5755, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 2.28, | |
| "learning_rate": 6.230031948881789e-07, | |
| "loss": 0.6207, | |
| "step": 424 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "learning_rate": 5.271565495207668e-07, | |
| "loss": 0.5041, | |
| "step": 428 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "learning_rate": 4.313099041533546e-07, | |
| "loss": 0.5543, | |
| "step": 432 | |
| }, | |
| { | |
| "epoch": 2.3, | |
| "learning_rate": 3.3546325878594247e-07, | |
| "loss": 0.5608, | |
| "step": 436 | |
| }, | |
| { | |
| "epoch": 2.31, | |
| "learning_rate": 2.3961661341853033e-07, | |
| "loss": 0.5849, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 2.31, | |
| "eval_loss": 1.0050277709960938, | |
| "eval_runtime": 101.6428, | |
| "eval_samples_per_second": 3.247, | |
| "eval_steps_per_second": 0.413, | |
| "eval_wer": 180.05748044068338, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 2.32, | |
| "learning_rate": 1.437699680511182e-07, | |
| "loss": 0.5486, | |
| "step": 444 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "learning_rate": 4.792332268370607e-08, | |
| "loss": 0.6387, | |
| "step": 448 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "step": 448, | |
| "total_flos": 1.7572960198656e+17, | |
| "train_loss": 0.8751970188958305, | |
| "train_runtime": 2136.6213, | |
| "train_samples_per_second": 3.355, | |
| "train_steps_per_second": 0.21 | |
| } | |
| ], | |
| "max_steps": 448, | |
| "num_train_epochs": 9223372036854775807, | |
| "total_flos": 1.7572960198656e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |