| { | |
| "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 | |
| } | |