| { |
| "best_metric": 0.26187804341316223, |
| "best_model_checkpoint": "./checkpoint-6000", |
| "epoch": 1.9998784982382245, |
| "global_step": 6172, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.03, |
| "learning_rate": 3.7499999999999997e-06, |
| "loss": 12.1043, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.06, |
| "learning_rate": 7.499999999999999e-06, |
| "loss": 6.4771, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.1, |
| "learning_rate": 1.1249999999999999e-05, |
| "loss": 4.4866, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.13, |
| "learning_rate": 1.4999999999999999e-05, |
| "loss": 3.8842, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.16, |
| "learning_rate": 1.8712499999999997e-05, |
| "loss": 3.495, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.16, |
| "eval_loss": 3.3882696628570557, |
| "eval_runtime": 721.337, |
| "eval_samples_per_second": 22.099, |
| "eval_steps_per_second": 1.382, |
| "eval_wer": 1.0, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.19, |
| "learning_rate": 2.2462499999999997e-05, |
| "loss": 3.171, |
| "step": 600 |
| }, |
| { |
| "epoch": 0.23, |
| "learning_rate": 2.6212499999999997e-05, |
| "loss": 3.0275, |
| "step": 700 |
| }, |
| { |
| "epoch": 0.26, |
| "learning_rate": 2.99625e-05, |
| "loss": 2.9681, |
| "step": 800 |
| }, |
| { |
| "epoch": 0.29, |
| "learning_rate": 3.37125e-05, |
| "loss": 2.9347, |
| "step": 900 |
| }, |
| { |
| "epoch": 0.32, |
| "learning_rate": 3.7462499999999996e-05, |
| "loss": 2.9095, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.32, |
| "eval_loss": 2.9152133464813232, |
| "eval_runtime": 718.1623, |
| "eval_samples_per_second": 22.197, |
| "eval_steps_per_second": 1.388, |
| "eval_wer": 0.9999871219487068, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.36, |
| "learning_rate": 4.12125e-05, |
| "loss": 2.8888, |
| "step": 1100 |
| }, |
| { |
| "epoch": 0.39, |
| "learning_rate": 4.4962499999999995e-05, |
| "loss": 2.8347, |
| "step": 1200 |
| }, |
| { |
| "epoch": 0.42, |
| "learning_rate": 4.871249999999999e-05, |
| "loss": 2.5318, |
| "step": 1300 |
| }, |
| { |
| "epoch": 0.45, |
| "learning_rate": 5.2462499999999994e-05, |
| "loss": 2.0502, |
| "step": 1400 |
| }, |
| { |
| "epoch": 0.49, |
| "learning_rate": 5.62125e-05, |
| "loss": 1.8434, |
| "step": 1500 |
| }, |
| { |
| "epoch": 0.49, |
| "eval_loss": 1.0473320484161377, |
| "eval_runtime": 720.1235, |
| "eval_samples_per_second": 22.136, |
| "eval_steps_per_second": 1.384, |
| "eval_wer": 0.7446153648029981, |
| "step": 1500 |
| }, |
| { |
| "epoch": 0.52, |
| "learning_rate": 5.9962499999999994e-05, |
| "loss": 1.7339, |
| "step": 1600 |
| }, |
| { |
| "epoch": 0.55, |
| "learning_rate": 6.367499999999999e-05, |
| "loss": 1.6535, |
| "step": 1700 |
| }, |
| { |
| "epoch": 0.58, |
| "learning_rate": 6.7425e-05, |
| "loss": 1.5793, |
| "step": 1800 |
| }, |
| { |
| "epoch": 0.62, |
| "learning_rate": 7.1175e-05, |
| "loss": 1.5056, |
| "step": 1900 |
| }, |
| { |
| "epoch": 0.65, |
| "learning_rate": 7.492499999999999e-05, |
| "loss": 1.4298, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.65, |
| "eval_loss": 0.5728740692138672, |
| "eval_runtime": 712.5783, |
| "eval_samples_per_second": 22.371, |
| "eval_steps_per_second": 1.399, |
| "eval_wer": 0.5129521000882147, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.68, |
| "learning_rate": 7.325623202301054e-05, |
| "loss": 1.3592, |
| "step": 2100 |
| }, |
| { |
| "epoch": 0.71, |
| "learning_rate": 7.145853307766058e-05, |
| "loss": 1.2917, |
| "step": 2200 |
| }, |
| { |
| "epoch": 0.75, |
| "learning_rate": 6.966083413231063e-05, |
| "loss": 1.2536, |
| "step": 2300 |
| }, |
| { |
| "epoch": 0.78, |
| "learning_rate": 6.788111217641419e-05, |
| "loss": 1.2345, |
| "step": 2400 |
| }, |
| { |
| "epoch": 0.81, |
| "learning_rate": 6.608341323106423e-05, |
| "loss": 1.1937, |
| "step": 2500 |
| }, |
| { |
| "epoch": 0.81, |
| "eval_loss": 0.3795304000377655, |
| "eval_runtime": 716.4435, |
| "eval_samples_per_second": 22.25, |
| "eval_steps_per_second": 1.392, |
| "eval_wer": 0.34504806732645216, |
| "step": 2500 |
| }, |
| { |
| "epoch": 0.84, |
| "learning_rate": 6.428571428571427e-05, |
| "loss": 1.1806, |
| "step": 2600 |
| }, |
| { |
| "epoch": 0.87, |
| "learning_rate": 6.248801534036433e-05, |
| "loss": 1.1651, |
| "step": 2700 |
| }, |
| { |
| "epoch": 0.91, |
| "learning_rate": 6.069031639501438e-05, |
| "loss": 1.1455, |
| "step": 2800 |
| }, |
| { |
| "epoch": 0.94, |
| "learning_rate": 5.889261744966442e-05, |
| "loss": 1.1312, |
| "step": 2900 |
| }, |
| { |
| "epoch": 0.97, |
| "learning_rate": 5.709491850431447e-05, |
| "loss": 1.1248, |
| "step": 3000 |
| }, |
| { |
| "epoch": 0.97, |
| "eval_loss": 0.3320523500442505, |
| "eval_runtime": 716.2808, |
| "eval_samples_per_second": 22.255, |
| "eval_steps_per_second": 1.392, |
| "eval_wer": 0.30515830344552264, |
| "step": 3000 |
| }, |
| { |
| "epoch": 1.0, |
| "learning_rate": 5.5297219558964525e-05, |
| "loss": 1.1017, |
| "step": 3100 |
| }, |
| { |
| "epoch": 1.04, |
| "learning_rate": 5.3499520613614567e-05, |
| "loss": 1.0978, |
| "step": 3200 |
| }, |
| { |
| "epoch": 1.07, |
| "learning_rate": 5.1701821668264615e-05, |
| "loss": 1.0954, |
| "step": 3300 |
| }, |
| { |
| "epoch": 1.1, |
| "learning_rate": 4.990412272291467e-05, |
| "loss": 1.0867, |
| "step": 3400 |
| }, |
| { |
| "epoch": 1.13, |
| "learning_rate": 4.812440076701821e-05, |
| "loss": 1.0835, |
| "step": 3500 |
| }, |
| { |
| "epoch": 1.13, |
| "eval_loss": 0.3037940561771393, |
| "eval_runtime": 714.1597, |
| "eval_samples_per_second": 22.321, |
| "eval_steps_per_second": 1.396, |
| "eval_wer": 0.2805032742445413, |
| "step": 3500 |
| }, |
| { |
| "epoch": 1.17, |
| "learning_rate": 4.632670182166826e-05, |
| "loss": 1.0808, |
| "step": 3600 |
| }, |
| { |
| "epoch": 1.2, |
| "learning_rate": 4.4529002876318304e-05, |
| "loss": 1.0648, |
| "step": 3700 |
| }, |
| { |
| "epoch": 1.23, |
| "learning_rate": 4.273130393096836e-05, |
| "loss": 1.0541, |
| "step": 3800 |
| }, |
| { |
| "epoch": 1.26, |
| "learning_rate": 4.093360498561841e-05, |
| "loss": 1.0621, |
| "step": 3900 |
| }, |
| { |
| "epoch": 1.3, |
| "learning_rate": 3.913590604026845e-05, |
| "loss": 1.0479, |
| "step": 4000 |
| }, |
| { |
| "epoch": 1.3, |
| "eval_loss": 0.2910499572753906, |
| "eval_runtime": 718.1665, |
| "eval_samples_per_second": 22.197, |
| "eval_steps_per_second": 1.388, |
| "eval_wer": 0.26888727197800427, |
| "step": 4000 |
| }, |
| { |
| "epoch": 1.33, |
| "learning_rate": 3.7338207094918506e-05, |
| "loss": 1.0428, |
| "step": 4100 |
| }, |
| { |
| "epoch": 1.36, |
| "learning_rate": 3.555848513902205e-05, |
| "loss": 1.047, |
| "step": 4200 |
| }, |
| { |
| "epoch": 1.39, |
| "learning_rate": 3.37607861936721e-05, |
| "loss": 1.0397, |
| "step": 4300 |
| }, |
| { |
| "epoch": 1.43, |
| "learning_rate": 3.1963087248322145e-05, |
| "loss": 1.0347, |
| "step": 4400 |
| }, |
| { |
| "epoch": 1.46, |
| "learning_rate": 3.0165388302972194e-05, |
| "loss": 1.0413, |
| "step": 4500 |
| }, |
| { |
| "epoch": 1.46, |
| "eval_loss": 0.27976545691490173, |
| "eval_runtime": 713.7382, |
| "eval_samples_per_second": 22.335, |
| "eval_steps_per_second": 1.397, |
| "eval_wer": 0.2592995627901586, |
| "step": 4500 |
| }, |
| { |
| "epoch": 1.49, |
| "learning_rate": 2.836768935762224e-05, |
| "loss": 1.0238, |
| "step": 4600 |
| }, |
| { |
| "epoch": 1.52, |
| "learning_rate": 2.656999041227229e-05, |
| "loss": 1.0269, |
| "step": 4700 |
| }, |
| { |
| "epoch": 1.56, |
| "learning_rate": 2.4772291466922337e-05, |
| "loss": 1.021, |
| "step": 4800 |
| }, |
| { |
| "epoch": 1.59, |
| "learning_rate": 2.2974592521572386e-05, |
| "loss": 1.0186, |
| "step": 4900 |
| }, |
| { |
| "epoch": 1.62, |
| "learning_rate": 2.1176893576222434e-05, |
| "loss": 1.014, |
| "step": 5000 |
| }, |
| { |
| "epoch": 1.62, |
| "eval_loss": 0.27265554666519165, |
| "eval_runtime": 707.3075, |
| "eval_samples_per_second": 22.538, |
| "eval_steps_per_second": 1.41, |
| "eval_wer": 0.25117351242409997, |
| "step": 5000 |
| }, |
| { |
| "epoch": 1.65, |
| "learning_rate": 1.9379194630872483e-05, |
| "loss": 1.0074, |
| "step": 5100 |
| }, |
| { |
| "epoch": 1.68, |
| "learning_rate": 1.759947267497603e-05, |
| "loss": 1.0193, |
| "step": 5200 |
| }, |
| { |
| "epoch": 1.72, |
| "learning_rate": 1.5801773729626078e-05, |
| "loss": 1.0044, |
| "step": 5300 |
| }, |
| { |
| "epoch": 1.75, |
| "learning_rate": 1.4004074784276125e-05, |
| "loss": 1.0005, |
| "step": 5400 |
| }, |
| { |
| "epoch": 1.78, |
| "learning_rate": 1.2206375838926173e-05, |
| "loss": 1.004, |
| "step": 5500 |
| }, |
| { |
| "epoch": 1.78, |
| "eval_loss": 0.26460376381874084, |
| "eval_runtime": 719.6956, |
| "eval_samples_per_second": 22.15, |
| "eval_steps_per_second": 1.385, |
| "eval_wer": 0.2470782921128375, |
| "step": 5500 |
| }, |
| { |
| "epoch": 1.81, |
| "learning_rate": 1.0408676893576222e-05, |
| "loss": 1.0048, |
| "step": 5600 |
| }, |
| { |
| "epoch": 1.85, |
| "learning_rate": 8.610977948226269e-06, |
| "loss": 0.9988, |
| "step": 5700 |
| }, |
| { |
| "epoch": 1.88, |
| "learning_rate": 6.813279002876318e-06, |
| "loss": 0.9919, |
| "step": 5800 |
| }, |
| { |
| "epoch": 1.91, |
| "learning_rate": 5.015580057526366e-06, |
| "loss": 0.9886, |
| "step": 5900 |
| }, |
| { |
| "epoch": 1.94, |
| "learning_rate": 3.217881112176414e-06, |
| "loss": 0.9949, |
| "step": 6000 |
| }, |
| { |
| "epoch": 1.94, |
| "eval_loss": 0.26187804341316223, |
| "eval_runtime": 717.4473, |
| "eval_samples_per_second": 22.219, |
| "eval_steps_per_second": 1.39, |
| "eval_wer": 0.24574541380398318, |
| "step": 6000 |
| }, |
| { |
| "epoch": 1.98, |
| "learning_rate": 1.4201821668264622e-06, |
| "loss": 0.9931, |
| "step": 6100 |
| }, |
| { |
| "epoch": 2.0, |
| "step": 6172, |
| "total_flos": 1.1573983785360925e+20, |
| "train_loss": 1.788894365302016, |
| "train_runtime": 52105.5599, |
| "train_samples_per_second": 15.163, |
| "train_steps_per_second": 0.118 |
| } |
| ], |
| "max_steps": 6172, |
| "num_train_epochs": 2, |
| "total_flos": 1.1573983785360925e+20, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
