| { | |
| "best_metric": 0.47801004237740824, | |
| "best_model_checkpoint": "./whisper-tiny-ckb/checkpoint-600", | |
| "epoch": 4.3478260869565215, | |
| "eval_steps": 100, | |
| "global_step": 600, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 2.6164804129562553e-06, | |
| "loss": 3.7615, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 4.980934789368156e-06, | |
| "loss": 2.8759, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 5.99823228863792e-06, | |
| "loss": 2.0115, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 6.655623437887147e-06, | |
| "loss": 1.4273, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 7.1422479480192775e-06, | |
| "loss": 1.0629, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "learning_rate": 7.528775214088733e-06, | |
| "loss": 0.864, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 7.849441238868767e-06, | |
| "loss": 0.69, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 8.123456553166724e-06, | |
| "loss": 0.5759, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 8.36268666504982e-06, | |
| "loss": 0.5048, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "learning_rate": 8.574976672697987e-06, | |
| "loss": 0.4502, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "eval_loss": 0.498779296875, | |
| "eval_runtime": 154.3428, | |
| "eval_samples_per_second": 32.007, | |
| "eval_steps_per_second": 0.065, | |
| "eval_wer": 0.7165699227235497, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 8.765784640355151e-06, | |
| "loss": 0.4391, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 8.93906144297322e-06, | |
| "loss": 0.4092, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 9.09775921698996e-06, | |
| "loss": 0.386, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.01, | |
| "learning_rate": 9.244141981517345e-06, | |
| "loss": 0.3711, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "learning_rate": 9.379984164319582e-06, | |
| "loss": 0.3478, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "learning_rate": 9.50670232443118e-06, | |
| "loss": 0.3501, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.23, | |
| "learning_rate": 9.625445353695127e-06, | |
| "loss": 0.3344, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.3, | |
| "learning_rate": 9.73715793581418e-06, | |
| "loss": 0.3167, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.38, | |
| "learning_rate": 9.842626246843015e-06, | |
| "loss": 0.3082, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "learning_rate": 9.942511545878664e-06, | |
| "loss": 0.2977, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "eval_loss": 0.355712890625, | |
| "eval_runtime": 154.4865, | |
| "eval_samples_per_second": 31.977, | |
| "eval_steps_per_second": 0.065, | |
| "eval_wer": 0.5858765713471742, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.52, | |
| "learning_rate": 9.925e-06, | |
| "loss": 0.3017, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.59, | |
| "learning_rate": 9.675000000000001e-06, | |
| "loss": 0.2896, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.67, | |
| "learning_rate": 9.425e-06, | |
| "loss": 0.2844, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.74, | |
| "learning_rate": 9.175000000000001e-06, | |
| "loss": 0.2729, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.81, | |
| "learning_rate": 8.925e-06, | |
| "loss": 0.2675, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "learning_rate": 8.675e-06, | |
| "loss": 0.2796, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 1.96, | |
| "learning_rate": 8.425000000000001e-06, | |
| "loss": 0.2726, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "learning_rate": 8.175e-06, | |
| "loss": 0.2623, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "learning_rate": 7.925000000000001e-06, | |
| "loss": 0.2456, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 2.17, | |
| "learning_rate": 7.675e-06, | |
| "loss": 0.2494, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.17, | |
| "eval_loss": 0.3095703125, | |
| "eval_runtime": 165.9821, | |
| "eval_samples_per_second": 29.762, | |
| "eval_steps_per_second": 0.06, | |
| "eval_wer": 0.5314625547523236, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "learning_rate": 7.425000000000001e-06, | |
| "loss": 0.2531, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.32, | |
| "learning_rate": 7.175000000000001e-06, | |
| "loss": 0.2496, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.39, | |
| "learning_rate": 6.925000000000001e-06, | |
| "loss": 0.2421, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.46, | |
| "learning_rate": 6.6750000000000005e-06, | |
| "loss": 0.2337, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.54, | |
| "learning_rate": 6.425e-06, | |
| "loss": 0.2358, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "learning_rate": 6.175000000000001e-06, | |
| "loss": 0.2433, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.68, | |
| "learning_rate": 5.925000000000001e-06, | |
| "loss": 0.2335, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 2.75, | |
| "learning_rate": 5.675000000000001e-06, | |
| "loss": 0.2301, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 2.83, | |
| "learning_rate": 5.4250000000000006e-06, | |
| "loss": 0.2223, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 2.9, | |
| "learning_rate": 5.1750000000000004e-06, | |
| "loss": 0.2224, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.9, | |
| "eval_loss": 0.28173828125, | |
| "eval_runtime": 152.093, | |
| "eval_samples_per_second": 32.48, | |
| "eval_steps_per_second": 0.066, | |
| "eval_wer": 0.5008012535166126, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "learning_rate": 4.925e-06, | |
| "loss": 0.2293, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 3.04, | |
| "learning_rate": 4.675000000000001e-06, | |
| "loss": 0.2255, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "learning_rate": 4.425e-06, | |
| "loss": 0.2131, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 3.19, | |
| "learning_rate": 4.175e-06, | |
| "loss": 0.2106, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 3.26, | |
| "learning_rate": 3.9250000000000005e-06, | |
| "loss": 0.2176, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "learning_rate": 3.6750000000000004e-06, | |
| "loss": 0.22, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 3.41, | |
| "learning_rate": 3.4250000000000007e-06, | |
| "loss": 0.2151, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 3.48, | |
| "learning_rate": 3.175e-06, | |
| "loss": 0.2082, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 3.55, | |
| "learning_rate": 2.925e-06, | |
| "loss": 0.2014, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 3.62, | |
| "learning_rate": 2.6750000000000002e-06, | |
| "loss": 0.2148, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.62, | |
| "eval_loss": 0.2666015625, | |
| "eval_runtime": 152.4154, | |
| "eval_samples_per_second": 32.411, | |
| "eval_steps_per_second": 0.066, | |
| "eval_wer": 0.481927281791959, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.7, | |
| "learning_rate": 2.425e-06, | |
| "loss": 0.2158, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 3.77, | |
| "learning_rate": 2.1750000000000004e-06, | |
| "loss": 0.2103, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 3.84, | |
| "learning_rate": 1.925e-06, | |
| "loss": 0.2091, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 3.91, | |
| "learning_rate": 1.6750000000000003e-06, | |
| "loss": 0.1998, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 3.99, | |
| "learning_rate": 1.425e-06, | |
| "loss": 0.2068, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 4.06, | |
| "learning_rate": 1.175e-06, | |
| "loss": 0.212, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 4.13, | |
| "learning_rate": 9.25e-07, | |
| "loss": 0.2034, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 4.2, | |
| "learning_rate": 6.750000000000001e-07, | |
| "loss": 0.2013, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 4.28, | |
| "learning_rate": 4.2500000000000006e-07, | |
| "loss": 0.1919, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 4.35, | |
| "learning_rate": 1.7500000000000002e-07, | |
| "loss": 0.2096, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 4.35, | |
| "eval_loss": 0.26123046875, | |
| "eval_runtime": 154.28, | |
| "eval_samples_per_second": 32.02, | |
| "eval_steps_per_second": 0.065, | |
| "eval_wer": 0.47801004237740824, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 4.35, | |
| "step": 600, | |
| "total_flos": 1.1344381987478569e+19, | |
| "train_loss": 0.4515265909830729, | |
| "train_runtime": 14971.1378, | |
| "train_samples_per_second": 30.779, | |
| "train_steps_per_second": 0.04 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 600, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 100, | |
| "total_flos": 1.1344381987478569e+19, | |
| "train_batch_size": 192, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |