| { | |
| "best_metric": 84.941730294506, | |
| "best_model_checkpoint": "./whisper-small-khmer/checkpoint-1200", | |
| "epoch": 200.0, | |
| "global_step": 2000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 2.5, | |
| "learning_rate": 4.4e-07, | |
| "loss": 1.9756, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "learning_rate": 9.400000000000001e-07, | |
| "loss": 1.8391, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 7.5, | |
| "learning_rate": 1.44e-06, | |
| "loss": 1.6744, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "learning_rate": 1.94e-06, | |
| "loss": 1.55, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 12.5, | |
| "learning_rate": 2.4400000000000004e-06, | |
| "loss": 1.4472, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "learning_rate": 2.9400000000000002e-06, | |
| "loss": 1.3592, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 17.5, | |
| "learning_rate": 3.44e-06, | |
| "loss": 1.2751, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "learning_rate": 3.94e-06, | |
| "loss": 1.1928, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 22.5, | |
| "learning_rate": 4.440000000000001e-06, | |
| "loss": 1.11, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "learning_rate": 4.94e-06, | |
| "loss": 1.0331, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 27.5, | |
| "learning_rate": 5.4400000000000004e-06, | |
| "loss": 0.9638, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 30.0, | |
| "learning_rate": 5.94e-06, | |
| "loss": 0.8933, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 32.5, | |
| "learning_rate": 6.440000000000001e-06, | |
| "loss": 0.8188, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 35.0, | |
| "learning_rate": 6.9400000000000005e-06, | |
| "loss": 0.7242, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 37.5, | |
| "learning_rate": 7.440000000000001e-06, | |
| "loss": 0.6423, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 40.0, | |
| "learning_rate": 7.94e-06, | |
| "loss": 0.5571, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 40.0, | |
| "eval_loss": 1.2021982669830322, | |
| "eval_runtime": 406.1515, | |
| "eval_samples_per_second": 1.768, | |
| "eval_steps_per_second": 0.03, | |
| "eval_wer": 89.95636948160055, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 42.5, | |
| "learning_rate": 8.44e-06, | |
| "loss": 0.4581, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 45.0, | |
| "learning_rate": 8.94e-06, | |
| "loss": 0.3935, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 47.5, | |
| "learning_rate": 9.440000000000001e-06, | |
| "loss": 0.3306, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 50.0, | |
| "learning_rate": 9.940000000000001e-06, | |
| "loss": 0.2705, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 52.5, | |
| "learning_rate": 9.853333333333334e-06, | |
| "loss": 0.2152, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 55.0, | |
| "learning_rate": 9.686666666666668e-06, | |
| "loss": 0.1493, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 57.5, | |
| "learning_rate": 9.52e-06, | |
| "loss": 0.1033, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 60.0, | |
| "learning_rate": 9.353333333333334e-06, | |
| "loss": 0.076, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 62.5, | |
| "learning_rate": 9.186666666666666e-06, | |
| "loss": 0.0503, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 65.0, | |
| "learning_rate": 9.020000000000002e-06, | |
| "loss": 0.0326, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 67.5, | |
| "learning_rate": 8.853333333333334e-06, | |
| "loss": 0.0283, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 70.0, | |
| "learning_rate": 8.686666666666668e-06, | |
| "loss": 0.0189, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 72.5, | |
| "learning_rate": 8.52e-06, | |
| "loss": 0.0151, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 75.0, | |
| "learning_rate": 8.353333333333335e-06, | |
| "loss": 0.015, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 77.5, | |
| "learning_rate": 8.186666666666667e-06, | |
| "loss": 0.0114, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 80.0, | |
| "learning_rate": 8.020000000000001e-06, | |
| "loss": 0.0088, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 80.0, | |
| "eval_loss": 1.7979915142059326, | |
| "eval_runtime": 405.7271, | |
| "eval_samples_per_second": 1.77, | |
| "eval_steps_per_second": 0.03, | |
| "eval_wer": 86.66685802858947, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 82.5, | |
| "learning_rate": 7.853333333333333e-06, | |
| "loss": 0.0097, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 85.0, | |
| "learning_rate": 7.686666666666667e-06, | |
| "loss": 0.0075, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 87.5, | |
| "learning_rate": 7.520000000000001e-06, | |
| "loss": 0.0062, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 90.0, | |
| "learning_rate": 7.353333333333334e-06, | |
| "loss": 0.0069, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 92.5, | |
| "learning_rate": 7.186666666666668e-06, | |
| "loss": 0.006, | |
| "step": 925 | |
| }, | |
| { | |
| "epoch": 95.0, | |
| "learning_rate": 7.0200000000000006e-06, | |
| "loss": 0.0049, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 97.5, | |
| "learning_rate": 6.853333333333334e-06, | |
| "loss": 0.0055, | |
| "step": 975 | |
| }, | |
| { | |
| "epoch": 100.0, | |
| "learning_rate": 6.6866666666666665e-06, | |
| "loss": 0.0044, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 102.5, | |
| "learning_rate": 6.520000000000001e-06, | |
| "loss": 0.0043, | |
| "step": 1025 | |
| }, | |
| { | |
| "epoch": 105.0, | |
| "learning_rate": 6.353333333333333e-06, | |
| "loss": 0.0037, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 107.5, | |
| "learning_rate": 6.186666666666668e-06, | |
| "loss": 0.0035, | |
| "step": 1075 | |
| }, | |
| { | |
| "epoch": 110.0, | |
| "learning_rate": 6.02e-06, | |
| "loss": 0.0032, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 112.5, | |
| "learning_rate": 5.853333333333335e-06, | |
| "loss": 0.0029, | |
| "step": 1125 | |
| }, | |
| { | |
| "epoch": 115.0, | |
| "learning_rate": 5.686666666666667e-06, | |
| "loss": 0.0031, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 117.5, | |
| "learning_rate": 5.5200000000000005e-06, | |
| "loss": 0.0023, | |
| "step": 1175 | |
| }, | |
| { | |
| "epoch": 120.0, | |
| "learning_rate": 5.3533333333333335e-06, | |
| "loss": 0.0023, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 120.0, | |
| "eval_loss": 1.9220576286315918, | |
| "eval_runtime": 406.1406, | |
| "eval_samples_per_second": 1.768, | |
| "eval_steps_per_second": 0.03, | |
| "eval_wer": 84.941730294506, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 122.5, | |
| "learning_rate": 5.186666666666667e-06, | |
| "loss": 0.0016, | |
| "step": 1225 | |
| }, | |
| { | |
| "epoch": 125.0, | |
| "learning_rate": 5.02e-06, | |
| "loss": 0.0016, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 127.5, | |
| "learning_rate": 4.853333333333334e-06, | |
| "loss": 0.0016, | |
| "step": 1275 | |
| }, | |
| { | |
| "epoch": 130.0, | |
| "learning_rate": 4.686666666666667e-06, | |
| "loss": 0.0016, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 132.5, | |
| "learning_rate": 4.520000000000001e-06, | |
| "loss": 0.001, | |
| "step": 1325 | |
| }, | |
| { | |
| "epoch": 135.0, | |
| "learning_rate": 4.353333333333334e-06, | |
| "loss": 0.001, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 137.5, | |
| "learning_rate": 4.1866666666666675e-06, | |
| "loss": 0.0009, | |
| "step": 1375 | |
| }, | |
| { | |
| "epoch": 140.0, | |
| "learning_rate": 4.0200000000000005e-06, | |
| "loss": 0.0006, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 142.5, | |
| "learning_rate": 3.853333333333334e-06, | |
| "loss": 0.0005, | |
| "step": 1425 | |
| }, | |
| { | |
| "epoch": 145.0, | |
| "learning_rate": 3.686666666666667e-06, | |
| "loss": 0.0004, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 147.5, | |
| "learning_rate": 3.52e-06, | |
| "loss": 0.0003, | |
| "step": 1475 | |
| }, | |
| { | |
| "epoch": 150.0, | |
| "learning_rate": 3.3533333333333336e-06, | |
| "loss": 0.0003, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 152.5, | |
| "learning_rate": 3.186666666666667e-06, | |
| "loss": 0.0002, | |
| "step": 1525 | |
| }, | |
| { | |
| "epoch": 155.0, | |
| "learning_rate": 3.0200000000000003e-06, | |
| "loss": 0.0002, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 157.5, | |
| "learning_rate": 2.8533333333333337e-06, | |
| "loss": 0.0002, | |
| "step": 1575 | |
| }, | |
| { | |
| "epoch": 160.0, | |
| "learning_rate": 2.686666666666667e-06, | |
| "loss": 0.0002, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 160.0, | |
| "eval_loss": 2.055905818939209, | |
| "eval_runtime": 406.1308, | |
| "eval_samples_per_second": 1.768, | |
| "eval_steps_per_second": 0.03, | |
| "eval_wer": 85.4325736264998, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 162.5, | |
| "learning_rate": 2.52e-06, | |
| "loss": 0.0002, | |
| "step": 1625 | |
| }, | |
| { | |
| "epoch": 165.0, | |
| "learning_rate": 2.3533333333333334e-06, | |
| "loss": 0.0002, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 167.5, | |
| "learning_rate": 2.1866666666666668e-06, | |
| "loss": 0.0002, | |
| "step": 1675 | |
| }, | |
| { | |
| "epoch": 170.0, | |
| "learning_rate": 2.02e-06, | |
| "loss": 0.0002, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 172.5, | |
| "learning_rate": 1.8533333333333333e-06, | |
| "loss": 0.0002, | |
| "step": 1725 | |
| }, | |
| { | |
| "epoch": 175.0, | |
| "learning_rate": 1.6866666666666667e-06, | |
| "loss": 0.0002, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 177.5, | |
| "learning_rate": 1.52e-06, | |
| "loss": 0.0002, | |
| "step": 1775 | |
| }, | |
| { | |
| "epoch": 180.0, | |
| "learning_rate": 1.3533333333333334e-06, | |
| "loss": 0.0002, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 182.5, | |
| "learning_rate": 1.1866666666666668e-06, | |
| "loss": 0.0002, | |
| "step": 1825 | |
| }, | |
| { | |
| "epoch": 185.0, | |
| "learning_rate": 1.02e-06, | |
| "loss": 0.0002, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 187.5, | |
| "learning_rate": 8.533333333333334e-07, | |
| "loss": 0.0002, | |
| "step": 1875 | |
| }, | |
| { | |
| "epoch": 190.0, | |
| "learning_rate": 6.866666666666667e-07, | |
| "loss": 0.0002, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 192.5, | |
| "learning_rate": 5.2e-07, | |
| "loss": 0.0002, | |
| "step": 1925 | |
| }, | |
| { | |
| "epoch": 195.0, | |
| "learning_rate": 3.533333333333334e-07, | |
| "loss": 0.0002, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 197.5, | |
| "learning_rate": 1.866666666666667e-07, | |
| "loss": 0.0002, | |
| "step": 1975 | |
| }, | |
| { | |
| "epoch": 200.0, | |
| "learning_rate": 2e-08, | |
| "loss": 0.0002, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 200.0, | |
| "eval_loss": 2.078744649887085, | |
| "eval_runtime": 406.1707, | |
| "eval_samples_per_second": 1.768, | |
| "eval_steps_per_second": 0.03, | |
| "eval_wer": 85.65359664733911, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 200.0, | |
| "step": 2000, | |
| "total_flos": 3.5322852829571318e+19, | |
| "train_loss": 0.2665567479010206, | |
| "train_runtime": 5656.7741, | |
| "train_samples_per_second": 22.628, | |
| "train_steps_per_second": 0.354 | |
| } | |
| ], | |
| "max_steps": 2000, | |
| "num_train_epochs": 200, | |
| "total_flos": 3.5322852829571318e+19, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |