{ "best_metric": 0.5802612119808278, "best_model_checkpoint": "whisper-tiny-luganda-v2/checkpoint-3000", "epoch": 0.8701326952360235, "eval_steps": 500, "global_step": 4000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.021753317380900587, "grad_norm": 47.60859680175781, "learning_rate": 4.5238095238095235e-06, "loss": 9.6365, "step": 100 }, { "epoch": 0.04350663476180117, "grad_norm": 41.293460845947266, "learning_rate": 9.285714285714286e-06, "loss": 4.282, "step": 200 }, { "epoch": 0.06525995214270176, "grad_norm": 29.167564392089844, "learning_rate": 1.4047619047619048e-05, "loss": 2.8974, "step": 300 }, { "epoch": 0.08701326952360235, "grad_norm": 30.911415100097656, "learning_rate": 1.880952380952381e-05, "loss": 2.2716, "step": 400 }, { "epoch": 0.10876658690450294, "grad_norm": 25.824464797973633, "learning_rate": 2.357142857142857e-05, "loss": 2.0498, "step": 500 }, { "epoch": 0.10876658690450294, "eval_loss": 0.9947264194488525, "eval_runtime": 2111.5938, "eval_samples_per_second": 6.458, "eval_steps_per_second": 1.615, "eval_wer": 0.8350676193361011, "step": 500 }, { "epoch": 0.13051990428540353, "grad_norm": 22.513357162475586, "learning_rate": 2.8333333333333335e-05, "loss": 1.8357, "step": 600 }, { "epoch": 0.1522732216663041, "grad_norm": 22.64017105102539, "learning_rate": 3.309523809523809e-05, "loss": 1.7019, "step": 700 }, { "epoch": 0.1740265390472047, "grad_norm": 22.659271240234375, "learning_rate": 3.785714285714286e-05, "loss": 1.6458, "step": 800 }, { "epoch": 0.19577985642810528, "grad_norm": 19.53955841064453, "learning_rate": 4.261904761904762e-05, "loss": 1.6001, "step": 900 }, { "epoch": 0.2175331738090059, "grad_norm": 21.89055061340332, "learning_rate": 4.738095238095238e-05, "loss": 1.5134, "step": 1000 }, { "epoch": 0.2175331738090059, "eval_loss": 0.7854410409927368, "eval_runtime": 2131.7562, "eval_samples_per_second": 6.397, "eval_steps_per_second": 1.6, "eval_wer": 0.7198385688576876, "step": 1000 }, { "epoch": 0.23928649118990647, "grad_norm": 24.862058639526367, "learning_rate": 4.988721804511278e-05, "loss": 1.421, "step": 1100 }, { "epoch": 0.26103980857080705, "grad_norm": 18.665178298950195, "learning_rate": 4.963659147869674e-05, "loss": 1.4254, "step": 1200 }, { "epoch": 0.28279312595170764, "grad_norm": 16.429445266723633, "learning_rate": 4.93859649122807e-05, "loss": 1.3466, "step": 1300 }, { "epoch": 0.3045464433326082, "grad_norm": 24.4464054107666, "learning_rate": 4.913533834586466e-05, "loss": 1.3342, "step": 1400 }, { "epoch": 0.3262997607135088, "grad_norm": 14.964367866516113, "learning_rate": 4.888471177944862e-05, "loss": 1.2494, "step": 1500 }, { "epoch": 0.3262997607135088, "eval_loss": 0.6914966106414795, "eval_runtime": 2154.6429, "eval_samples_per_second": 6.329, "eval_steps_per_second": 1.583, "eval_wer": 0.7817383247509934, "step": 1500 }, { "epoch": 0.3480530780944094, "grad_norm": 17.042715072631836, "learning_rate": 4.863408521303258e-05, "loss": 1.2668, "step": 1600 }, { "epoch": 0.36980639547530997, "grad_norm": 14.82885456085205, "learning_rate": 4.838345864661654e-05, "loss": 1.2279, "step": 1700 }, { "epoch": 0.39155971285621055, "grad_norm": 19.486909866333008, "learning_rate": 4.81328320802005e-05, "loss": 1.2124, "step": 1800 }, { "epoch": 0.41331303023711113, "grad_norm": 18.727062225341797, "learning_rate": 4.7882205513784464e-05, "loss": 1.2099, "step": 1900 }, { "epoch": 0.4350663476180118, "grad_norm": 17.26986312866211, "learning_rate": 4.7631578947368424e-05, "loss": 1.1587, "step": 2000 }, { "epoch": 0.4350663476180118, "eval_loss": 0.6377026438713074, "eval_runtime": 2184.0106, "eval_samples_per_second": 6.244, "eval_steps_per_second": 1.561, "eval_wer": 0.7916005915578284, "step": 2000 }, { "epoch": 0.45681966499891236, "grad_norm": 20.09056282043457, "learning_rate": 4.738095238095238e-05, "loss": 1.1293, "step": 2100 }, { "epoch": 0.47857298237981294, "grad_norm": 12.891928672790527, "learning_rate": 4.713032581453634e-05, "loss": 1.1347, "step": 2200 }, { "epoch": 0.5003262997607135, "grad_norm": 14.460548400878906, "learning_rate": 4.68796992481203e-05, "loss": 1.1219, "step": 2300 }, { "epoch": 0.5220796171416141, "grad_norm": 14.870162963867188, "learning_rate": 4.662907268170426e-05, "loss": 1.0758, "step": 2400 }, { "epoch": 0.5438329345225147, "grad_norm": 15.290679931640625, "learning_rate": 4.637844611528822e-05, "loss": 1.0453, "step": 2500 }, { "epoch": 0.5438329345225147, "eval_loss": 0.5957969427108765, "eval_runtime": 2164.8981, "eval_samples_per_second": 6.299, "eval_steps_per_second": 1.575, "eval_wer": 0.7275804928460703, "step": 2500 }, { "epoch": 0.5655862519034153, "grad_norm": 14.45456314086914, "learning_rate": 4.612781954887218e-05, "loss": 1.0498, "step": 2600 }, { "epoch": 0.5873395692843159, "grad_norm": 14.366482734680176, "learning_rate": 4.587719298245614e-05, "loss": 1.0699, "step": 2700 }, { "epoch": 0.6090928866652164, "grad_norm": 20.212724685668945, "learning_rate": 4.56265664160401e-05, "loss": 1.0341, "step": 2800 }, { "epoch": 0.630846204046117, "grad_norm": 14.940664291381836, "learning_rate": 4.537593984962406e-05, "loss": 1.0165, "step": 2900 }, { "epoch": 0.6525995214270176, "grad_norm": 12.089710235595703, "learning_rate": 4.512531328320802e-05, "loss": 1.0065, "step": 3000 }, { "epoch": 0.6525995214270176, "eval_loss": 0.5708178281784058, "eval_runtime": 2103.4657, "eval_samples_per_second": 6.483, "eval_steps_per_second": 1.621, "eval_wer": 0.5802612119808278, "step": 3000 }, { "epoch": 0.6743528388079182, "grad_norm": 15.426619529724121, "learning_rate": 4.487468671679198e-05, "loss": 0.9918, "step": 3100 }, { "epoch": 0.6961061561888188, "grad_norm": 15.387158393859863, "learning_rate": 4.4624060150375944e-05, "loss": 0.9898, "step": 3200 }, { "epoch": 0.7178594735697194, "grad_norm": 13.513629913330078, "learning_rate": 4.43734335839599e-05, "loss": 0.9918, "step": 3300 }, { "epoch": 0.7396127909506199, "grad_norm": 14.379338264465332, "learning_rate": 4.412280701754386e-05, "loss": 0.9797, "step": 3400 }, { "epoch": 0.7613661083315205, "grad_norm": 15.046808242797852, "learning_rate": 4.387218045112782e-05, "loss": 0.935, "step": 3500 }, { "epoch": 0.7613661083315205, "eval_loss": 0.547825276851654, "eval_runtime": 2134.8624, "eval_samples_per_second": 6.388, "eval_steps_per_second": 1.597, "eval_wer": 0.5881724070345491, "step": 3500 }, { "epoch": 0.7831194257124211, "grad_norm": 19.1199893951416, "learning_rate": 4.362155388471178e-05, "loss": 0.9624, "step": 3600 }, { "epoch": 0.8048727430933217, "grad_norm": 14.161909103393555, "learning_rate": 4.337092731829574e-05, "loss": 0.9398, "step": 3700 }, { "epoch": 0.8266260604742223, "grad_norm": 14.970125198364258, "learning_rate": 4.31203007518797e-05, "loss": 0.9781, "step": 3800 }, { "epoch": 0.8483793778551229, "grad_norm": 11.029887199401855, "learning_rate": 4.286967418546366e-05, "loss": 0.9815, "step": 3900 }, { "epoch": 0.8701326952360235, "grad_norm": 12.273273468017578, "learning_rate": 4.261904761904762e-05, "loss": 0.8838, "step": 4000 }, { "epoch": 0.8701326952360235, "eval_loss": 0.5256088376045227, "eval_runtime": 2143.6891, "eval_samples_per_second": 6.361, "eval_steps_per_second": 1.591, "eval_wer": 0.5876378668282166, "step": 4000 } ], "logging_steps": 100, "max_steps": 21000, "num_input_tokens_seen": 0, "num_train_epochs": 5, "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": 1.57560864768e+18, "train_batch_size": 8, "trial_name": null, "trial_params": null }