{ "best_metric": 20.827384221123925, "best_model_checkpoint": "whisper-small-vietnamese/checkpoint-3000", "epoch": 52.63157894736842, "eval_steps": 1000, "global_step": 5000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 5.2631578947368425, "grad_norm": 2.647732734680176, "learning_rate": 1.7429999999999997e-05, "loss": 0.4843, "step": 500 }, { "epoch": 10.526315789473685, "grad_norm": 2.083451986312866, "learning_rate": 1.556333333333333e-05, "loss": 0.0586, "step": 1000 }, { "epoch": 10.526315789473685, "eval_loss": 0.6817760467529297, "eval_runtime": 119.8409, "eval_samples_per_second": 9.488, "eval_steps_per_second": 0.401, "eval_wer": 21.848801025077798, "step": 1000 }, { "epoch": 15.789473684210526, "grad_norm": 0.8228249549865723, "learning_rate": 1.3618888888888889e-05, "loss": 0.0083, "step": 1500 }, { "epoch": 21.05263157894737, "grad_norm": 0.2391805350780487, "learning_rate": 1.1674444444444444e-05, "loss": 0.0031, "step": 2000 }, { "epoch": 21.05263157894737, "eval_loss": 0.7675614953041077, "eval_runtime": 118.5639, "eval_samples_per_second": 9.59, "eval_steps_per_second": 0.405, "eval_wer": 21.14222954420648, "step": 2000 }, { "epoch": 26.31578947368421, "grad_norm": 0.11328469961881638, "learning_rate": 9.73e-06, "loss": 0.0012, "step": 2500 }, { "epoch": 31.57894736842105, "grad_norm": 0.008894503116607666, "learning_rate": 7.785555555555555e-06, "loss": 0.0005, "step": 3000 }, { "epoch": 31.57894736842105, "eval_loss": 0.8094444870948792, "eval_runtime": 119.7395, "eval_samples_per_second": 9.496, "eval_steps_per_second": 0.401, "eval_wer": 20.827384221123925, "step": 3000 }, { "epoch": 36.8421052631579, "grad_norm": 0.006032014265656471, "learning_rate": 5.8411111111111104e-06, "loss": 0.0003, "step": 3500 }, { "epoch": 42.10526315789474, "grad_norm": 0.004732669331133366, "learning_rate": 3.896666666666667e-06, "loss": 0.0002, "step": 4000 }, { "epoch": 42.10526315789474, "eval_loss": 0.8392577767372131, "eval_runtime": 119.9053, "eval_samples_per_second": 9.482, "eval_steps_per_second": 0.4, "eval_wer": 21.149551528464215, "step": 4000 }, { "epoch": 47.36842105263158, "grad_norm": 0.004539730492979288, "learning_rate": 1.952222222222222e-06, "loss": 0.0001, "step": 4500 }, { "epoch": 52.63157894736842, "grad_norm": 0.003727754345163703, "learning_rate": 7.777777777777777e-09, "loss": 0.0001, "step": 5000 }, { "epoch": 52.63157894736842, "eval_loss": 0.8485473394393921, "eval_runtime": 119.3939, "eval_samples_per_second": 9.523, "eval_steps_per_second": 0.402, "eval_wer": 21.21911037891269, "step": 5000 } ], "logging_steps": 500, "max_steps": 5000, "num_input_tokens_seen": 0, "num_train_epochs": 53, "save_steps": 1000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 6.908041878700032e+19, "train_batch_size": 48, "trial_name": null, "trial_params": null }