{ "best_metric": 0.08970743743390906, "best_model_checkpoint": "./whisper-medium-sr-yodas/checkpoint-1000", "epoch": 0.4884004884004884, "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.01, "grad_norm": 9.130462646484375, "learning_rate": 4.600000000000001e-06, "loss": 1.914, "step": 25 }, { "epoch": 0.02, "grad_norm": 11.731131553649902, "learning_rate": 9.600000000000001e-06, "loss": 1.4161, "step": 50 }, { "epoch": 0.04, "grad_norm": 7.232382297515869, "learning_rate": 9.841379310344828e-06, "loss": 1.1372, "step": 75 }, { "epoch": 0.05, "grad_norm": 6.721960067749023, "learning_rate": 9.66896551724138e-06, "loss": 0.8027, "step": 100 }, { "epoch": 0.06, "grad_norm": 7.100645065307617, "learning_rate": 9.496551724137932e-06, "loss": 0.7147, "step": 125 }, { "epoch": 0.07, "grad_norm": 9.62185001373291, "learning_rate": 9.324137931034484e-06, "loss": 0.8326, "step": 150 }, { "epoch": 0.09, "grad_norm": 7.819121360778809, "learning_rate": 9.151724137931035e-06, "loss": 0.7793, "step": 175 }, { "epoch": 0.1, "grad_norm": 5.436564922332764, "learning_rate": 8.979310344827587e-06, "loss": 0.6697, "step": 200 }, { "epoch": 0.11, "grad_norm": 6.4688029289245605, "learning_rate": 8.806896551724139e-06, "loss": 0.6794, "step": 225 }, { "epoch": 0.12, "grad_norm": 6.041417121887207, "learning_rate": 8.634482758620691e-06, "loss": 0.6636, "step": 250 }, { "epoch": 0.13, "grad_norm": 9.54858112335205, "learning_rate": 8.462068965517241e-06, "loss": 0.7643, "step": 275 }, { "epoch": 0.15, "grad_norm": 8.129537582397461, "learning_rate": 8.289655172413793e-06, "loss": 0.6648, "step": 300 }, { "epoch": 0.16, "grad_norm": 7.3216872215271, "learning_rate": 8.117241379310346e-06, "loss": 0.6526, "step": 325 }, { "epoch": 0.17, "grad_norm": 6.333176136016846, "learning_rate": 7.944827586206898e-06, "loss": 0.6882, "step": 350 }, { "epoch": 0.18, "grad_norm": 7.942104816436768, "learning_rate": 7.77241379310345e-06, "loss": 0.6674, "step": 375 }, { "epoch": 0.2, "grad_norm": 5.19351863861084, "learning_rate": 7.600000000000001e-06, "loss": 0.6029, "step": 400 }, { "epoch": 0.21, "grad_norm": 5.838120937347412, "learning_rate": 7.427586206896552e-06, "loss": 0.5979, "step": 425 }, { "epoch": 0.22, "grad_norm": 6.667115688323975, "learning_rate": 7.255172413793104e-06, "loss": 0.6611, "step": 450 }, { "epoch": 0.23, "grad_norm": 4.763759613037109, "learning_rate": 7.0827586206896555e-06, "loss": 0.5857, "step": 475 }, { "epoch": 0.24, "grad_norm": 5.6900129318237305, "learning_rate": 6.910344827586208e-06, "loss": 0.5918, "step": 500 }, { "epoch": 0.24, "eval_loss": 0.17614366114139557, "eval_runtime": 446.8109, "eval_samples_per_second": 3.453, "eval_steps_per_second": 0.432, "eval_wer": 0.10345435318998943, "eval_wer_ortho": 0.20109366731345915, "step": 500 }, { "epoch": 0.26, "grad_norm": 4.805812358856201, "learning_rate": 6.737931034482759e-06, "loss": 0.6014, "step": 525 }, { "epoch": 0.27, "grad_norm": 6.178523063659668, "learning_rate": 6.565517241379311e-06, "loss": 0.6268, "step": 550 }, { "epoch": 0.28, "grad_norm": 7.9498066902160645, "learning_rate": 6.393103448275862e-06, "loss": 0.6575, "step": 575 }, { "epoch": 0.29, "grad_norm": 5.547674655914307, "learning_rate": 6.220689655172414e-06, "loss": 0.587, "step": 600 }, { "epoch": 0.31, "grad_norm": 8.781354904174805, "learning_rate": 6.048275862068966e-06, "loss": 0.6366, "step": 625 }, { "epoch": 0.32, "grad_norm": 5.279805660247803, "learning_rate": 5.875862068965518e-06, "loss": 0.5603, "step": 650 }, { "epoch": 0.33, "grad_norm": 7.07759952545166, "learning_rate": 5.703448275862069e-06, "loss": 0.5903, "step": 675 }, { "epoch": 0.34, "grad_norm": 7.5665202140808105, "learning_rate": 5.531034482758622e-06, "loss": 0.5792, "step": 700 }, { "epoch": 0.35, "grad_norm": 4.364047050476074, "learning_rate": 5.358620689655173e-06, "loss": 0.5298, "step": 725 }, { "epoch": 0.37, "grad_norm": 7.198115348815918, "learning_rate": 5.193103448275863e-06, "loss": 0.5717, "step": 750 }, { "epoch": 0.38, "grad_norm": 5.364291667938232, "learning_rate": 5.020689655172414e-06, "loss": 0.5509, "step": 775 }, { "epoch": 0.39, "grad_norm": 5.191427707672119, "learning_rate": 4.848275862068966e-06, "loss": 0.5267, "step": 800 }, { "epoch": 0.4, "grad_norm": 6.82424783706665, "learning_rate": 4.675862068965517e-06, "loss": 0.5319, "step": 825 }, { "epoch": 0.42, "grad_norm": 8.202191352844238, "learning_rate": 4.503448275862069e-06, "loss": 0.521, "step": 850 }, { "epoch": 0.43, "grad_norm": 5.929942607879639, "learning_rate": 4.3310344827586206e-06, "loss": 0.5218, "step": 875 }, { "epoch": 0.44, "grad_norm": 6.061079025268555, "learning_rate": 4.158620689655173e-06, "loss": 0.5195, "step": 900 }, { "epoch": 0.45, "grad_norm": 5.912203311920166, "learning_rate": 3.986206896551724e-06, "loss": 0.5596, "step": 925 }, { "epoch": 0.46, "grad_norm": 4.799466133117676, "learning_rate": 3.813793103448276e-06, "loss": 0.5251, "step": 950 }, { "epoch": 0.48, "grad_norm": 5.564664840698242, "learning_rate": 3.641379310344828e-06, "loss": 0.599, "step": 975 }, { "epoch": 0.49, "grad_norm": 5.804991722106934, "learning_rate": 3.4689655172413793e-06, "loss": 0.5499, "step": 1000 }, { "epoch": 0.49, "eval_loss": 0.1544894278049469, "eval_runtime": 442.7866, "eval_samples_per_second": 3.485, "eval_steps_per_second": 0.436, "eval_wer": 0.08970743743390906, "eval_wer_ortho": 0.18680543305697653, "step": 1000 } ], "logging_steps": 25, "max_steps": 1500, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 500, "total_flos": 1.632967852032e+19, "train_batch_size": 8, "trial_name": null, "trial_params": null }