| { |
| "best_global_step": 1500, |
| "best_metric": 0.33901429176330566, |
| "best_model_checkpoint": "./stt/checkpoint-1500", |
| "epoch": 1.854140914709518, |
| "eval_steps": 500, |
| "global_step": 1500, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.12360939431396786, |
| "grad_norm": 10.66793155670166, |
| "learning_rate": 9.999173944549022e-06, |
| "loss": 1.6423, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.24721878862793573, |
| "grad_norm": 8.940858840942383, |
| "learning_rate": 9.991680624320292e-06, |
| "loss": 0.5818, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.37082818294190356, |
| "grad_norm": 7.630367755889893, |
| "learning_rate": 9.976393387344834e-06, |
| "loss": 0.4997, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.49443757725587145, |
| "grad_norm": 8.54523754119873, |
| "learning_rate": 9.953336102828482e-06, |
| "loss": 0.4622, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.6180469715698393, |
| "grad_norm": 8.749971389770508, |
| "learning_rate": 9.922544771983945e-06, |
| "loss": 0.4135, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.6180469715698393, |
| "eval_loss": 0.4068942070007324, |
| "eval_runtime": 1305.4198, |
| "eval_samples_per_second": 2.477, |
| "eval_steps_per_second": 0.156, |
| "eval_wer": 21.631878557874764, |
| "eval_wer_ortho": 29.911516425399192, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.7416563658838071, |
| "grad_norm": 7.349922180175781, |
| "learning_rate": 9.884067471819184e-06, |
| "loss": 0.3882, |
| "step": 600 |
| }, |
| { |
| "epoch": 0.865265760197775, |
| "grad_norm": 8.007939338684082, |
| "learning_rate": 9.837964280070878e-06, |
| "loss": 0.3613, |
| "step": 700 |
| }, |
| { |
| "epoch": 0.9888751545117429, |
| "grad_norm": 7.204431056976318, |
| "learning_rate": 9.784307181400159e-06, |
| "loss": 0.3561, |
| "step": 800 |
| }, |
| { |
| "epoch": 1.1124845488257107, |
| "grad_norm": 6.045726776123047, |
| "learning_rate": 9.723179954997127e-06, |
| "loss": 0.2156, |
| "step": 900 |
| }, |
| { |
| "epoch": 1.2360939431396787, |
| "grad_norm": 5.328393459320068, |
| "learning_rate": 9.654678043769566e-06, |
| "loss": 0.2036, |
| "step": 1000 |
| }, |
| { |
| "epoch": 1.2360939431396787, |
| "eval_loss": 0.35839423537254333, |
| "eval_runtime": 1301.6256, |
| "eval_samples_per_second": 2.484, |
| "eval_steps_per_second": 0.156, |
| "eval_wer": 18.355245323936025, |
| "eval_wer_ortho": 25.873817676373868, |
| "step": 1000 |
| }, |
| { |
| "epoch": 1.3597033374536465, |
| "grad_norm": 6.864717960357666, |
| "learning_rate": 9.578908405320196e-06, |
| "loss": 0.2019, |
| "step": 1100 |
| }, |
| { |
| "epoch": 1.4833127317676142, |
| "grad_norm": 5.546298027038574, |
| "learning_rate": 9.495989344945056e-06, |
| "loss": 0.211, |
| "step": 1200 |
| }, |
| { |
| "epoch": 1.6069221260815822, |
| "grad_norm": 4.569760322570801, |
| "learning_rate": 9.406050330913839e-06, |
| "loss": 0.196, |
| "step": 1300 |
| }, |
| { |
| "epoch": 1.73053152039555, |
| "grad_norm": 6.44041109085083, |
| "learning_rate": 9.309231792320548e-06, |
| "loss": 0.2003, |
| "step": 1400 |
| }, |
| { |
| "epoch": 1.854140914709518, |
| "grad_norm": 6.337347507476807, |
| "learning_rate": 9.205684899820146e-06, |
| "loss": 0.1899, |
| "step": 1500 |
| }, |
| { |
| "epoch": 1.854140914709518, |
| "eval_loss": 0.33901429176330566, |
| "eval_runtime": 1299.601, |
| "eval_samples_per_second": 2.488, |
| "eval_steps_per_second": 0.156, |
| "eval_wer": 16.481431282190297, |
| "eval_wer_ortho": 24.09397565854155, |
| "step": 1500 |
| } |
| ], |
| "logging_steps": 100, |
| "max_steps": 8000, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 10, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 2.448022931177472e+19, |
| "train_batch_size": 16, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
