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