{ "best_metric": null, "best_model_checkpoint": null, "epoch": 7.149240393208221, "eval_steps": 200, "global_step": 4000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.18, "learning_rate": 5e-05, "loss": 0.4292, "step": 100 }, { "epoch": 0.36, "learning_rate": 4.994863481875841e-05, "loss": 0.2746, "step": 200 }, { "epoch": 0.36, "eval_f1_score": 0.6564825695260478, "eval_label_f1": 0.8343125734430082, "eval_loss": 0.26023727655410767, "eval_runtime": 344.2386, "eval_samples_per_second": 2.905, "eval_steps_per_second": 0.363, "eval_wer": 0.10898676368139949, "step": 200 }, { "epoch": 0.54, "learning_rate": 4.979475034558115e-05, "loss": 0.2568, "step": 300 }, { "epoch": 0.71, "learning_rate": 4.9538978924776634e-05, "loss": 0.2481, "step": 400 }, { "epoch": 0.71, "eval_f1_score": 0.6577916992952232, "eval_label_f1": 0.8347689898198903, "eval_loss": 0.246540829539299, "eval_runtime": 341.9234, "eval_samples_per_second": 2.925, "eval_steps_per_second": 0.366, "eval_wer": 0.10217509095131203, "step": 400 }, { "epoch": 0.89, "learning_rate": 4.9182371575975736e-05, "loss": 0.2412, "step": 500 }, { "epoch": 1.07, "learning_rate": 4.8726393675266716e-05, "loss": 0.2385, "step": 600 }, { "epoch": 1.07, "eval_f1_score": 0.6684952978056427, "eval_label_f1": 0.8322884012539185, "eval_loss": 0.24104812741279602, "eval_runtime": 342.1463, "eval_samples_per_second": 2.923, "eval_steps_per_second": 0.365, "eval_wer": 0.10484557628299404, "step": 600 }, { "epoch": 1.25, "learning_rate": 4.817291893365055e-05, "loss": 0.2325, "step": 700 }, { "epoch": 1.43, "learning_rate": 4.752422169756048e-05, "loss": 0.2316, "step": 800 }, { "epoch": 1.43, "eval_f1_score": 0.6724477729601892, "eval_label_f1": 0.8316909735908553, "eval_loss": 0.23740312457084656, "eval_runtime": 344.4982, "eval_samples_per_second": 2.903, "eval_steps_per_second": 0.363, "eval_wer": 0.10221379363727842, "step": 800 }, { "epoch": 1.61, "learning_rate": 4.678296760308474e-05, "loss": 0.2304, "step": 900 }, { "epoch": 1.79, "learning_rate": 4.595220262229601e-05, "loss": 0.2291, "step": 1000 }, { "epoch": 1.79, "eval_f1_score": 0.6698076168040833, "eval_label_f1": 0.8292108362779742, "eval_loss": 0.2348490208387375, "eval_runtime": 338.1818, "eval_samples_per_second": 2.957, "eval_steps_per_second": 0.37, "eval_wer": 0.09683412028794798, "step": 1000 }, { "epoch": 1.97, "learning_rate": 4.503534054669892e-05, "loss": 0.2275, "step": 1100 }, { "epoch": 2.14, "learning_rate": 4.4036148959228365e-05, "loss": 0.2205, "step": 1200 }, { "epoch": 2.14, "eval_f1_score": 0.6744822196170379, "eval_label_f1": 0.8339194998046112, "eval_loss": 0.2333754003047943, "eval_runtime": 340.8077, "eval_samples_per_second": 2.934, "eval_steps_per_second": 0.367, "eval_wer": 0.09636968805635111, "step": 1200 }, { "epoch": 2.32, "learning_rate": 4.2958733752443195e-05, "loss": 0.2224, "step": 1300 }, { "epoch": 2.5, "learning_rate": 4.180752225653292e-05, "loss": 0.2211, "step": 1400 }, { "epoch": 2.5, "eval_f1_score": 0.672933803368586, "eval_label_f1": 0.8327457892675283, "eval_loss": 0.2319139689207077, "eval_runtime": 341.1059, "eval_samples_per_second": 2.932, "eval_steps_per_second": 0.366, "eval_wer": 0.09606006656861986, "step": 1400 }, { "epoch": 2.68, "learning_rate": 4.058724504646834e-05, "loss": 0.2196, "step": 1500 }, { "epoch": 2.86, "learning_rate": 3.9302916503054246e-05, "loss": 0.2163, "step": 1600 }, { "epoch": 2.86, "eval_f1_score": 0.6731669266770671, "eval_label_f1": 0.829953198127925, "eval_loss": 0.2304619699716568, "eval_runtime": 340.688, "eval_samples_per_second": 2.935, "eval_steps_per_second": 0.367, "eval_wer": 0.09807260623887298, "step": 1600 }, { "epoch": 3.04, "learning_rate": 3.7959814207763135e-05, "loss": 0.219, "step": 1700 }, { "epoch": 3.22, "learning_rate": 3.656345725602089e-05, "loss": 0.2108, "step": 1800 }, { "epoch": 3.22, "eval_f1_score": 0.6734375, "eval_label_f1": 0.8328125000000001, "eval_loss": 0.22986993193626404, "eval_runtime": 338.4413, "eval_samples_per_second": 2.955, "eval_steps_per_second": 0.369, "eval_wer": 0.09544082359315736, "step": 1800 }, { "epoch": 3.4, "learning_rate": 3.5119583578059846e-05, "loss": 0.2132, "step": 1900 }, { "epoch": 3.57, "learning_rate": 3.363412636053269e-05, "loss": 0.2104, "step": 2000 }, { "epoch": 3.57, "eval_f1_score": 0.6792156862745098, "eval_label_f1": 0.8368627450980393, "eval_loss": 0.22974026203155518, "eval_runtime": 343.1743, "eval_samples_per_second": 2.914, "eval_steps_per_second": 0.364, "eval_wer": 0.09919498413189876, "step": 2000 }, { "epoch": 3.75, "learning_rate": 3.211318966577581e-05, "loss": 0.2128, "step": 2100 }, { "epoch": 3.93, "learning_rate": 3.056302334890786e-05, "loss": 0.2124, "step": 2200 }, { "epoch": 3.93, "eval_f1_score": 0.6781925343811395, "eval_label_f1": 0.8345776031434184, "eval_loss": 0.2278670072555542, "eval_runtime": 338.3142, "eval_samples_per_second": 2.956, "eval_steps_per_second": 0.369, "eval_wer": 0.09447325644399722, "step": 2200 }, { "epoch": 4.11, "learning_rate": 2.8989997375834482e-05, "loss": 0.2077, "step": 2300 }, { "epoch": 4.29, "learning_rate": 2.7400575647692046e-05, "loss": 0.2027, "step": 2400 }, { "epoch": 4.29, "eval_f1_score": 0.6789638932496076, "eval_label_f1": 0.8335949764521193, "eval_loss": 0.22793905436992645, "eval_runtime": 338.197, "eval_samples_per_second": 2.957, "eval_steps_per_second": 0.37, "eval_wer": 0.09443455375803081, "step": 2400 }, { "epoch": 4.47, "learning_rate": 2.5801289439291388e-05, "loss": 0.2077, "step": 2500 }, { "epoch": 4.65, "learning_rate": 2.419871056070862e-05, "loss": 0.2055, "step": 2600 }, { "epoch": 4.65, "eval_f1_score": 0.6832347140039449, "eval_label_f1": 0.8347140039447732, "eval_loss": 0.2275087535381317, "eval_runtime": 337.608, "eval_samples_per_second": 2.962, "eval_steps_per_second": 0.37, "eval_wer": 0.09493768867559409, "step": 2600 }, { "epoch": 4.83, "learning_rate": 2.2599424352307957e-05, "loss": 0.2073, "step": 2700 }, { "epoch": 5.0, "learning_rate": 2.1010002624165527e-05, "loss": 0.209, "step": 2800 }, { "epoch": 5.0, "eval_f1_score": 0.6821766561514195, "eval_label_f1": 0.833596214511041, "eval_loss": 0.22685159742832184, "eval_runtime": 338.9186, "eval_samples_per_second": 2.951, "eval_steps_per_second": 0.369, "eval_wer": 0.09826611966870501, "step": 2800 }, { "epoch": 5.18, "learning_rate": 1.9436976651092144e-05, "loss": 0.2011, "step": 2900 }, { "epoch": 5.36, "learning_rate": 1.7886810334224192e-05, "loss": 0.2017, "step": 3000 }, { "epoch": 5.36, "eval_f1_score": 0.6834645669291338, "eval_label_f1": 0.8346456692913385, "eval_loss": 0.22723452746868134, "eval_runtime": 337.6021, "eval_samples_per_second": 2.962, "eval_steps_per_second": 0.37, "eval_wer": 0.09791779549500736, "step": 3000 }, { "epoch": 5.54, "learning_rate": 1.6365873639467315e-05, "loss": 0.2034, "step": 3100 }, { "epoch": 5.72, "learning_rate": 1.4880416421940155e-05, "loss": 0.2029, "step": 3200 }, { "epoch": 5.72, "eval_f1_score": 0.6818718049547778, "eval_label_f1": 0.8320880849390484, "eval_loss": 0.22658005356788635, "eval_runtime": 338.8852, "eval_samples_per_second": 2.951, "eval_steps_per_second": 0.369, "eval_wer": 0.09656320148618314, "step": 3200 }, { "epoch": 5.9, "learning_rate": 1.3436542743979125e-05, "loss": 0.202, "step": 3300 }, { "epoch": 6.08, "learning_rate": 1.2040185792236874e-05, "loss": 0.201, "step": 3400 }, { "epoch": 6.08, "eval_f1_score": 0.6800472255017709, "eval_label_f1": 0.8327430145611965, "eval_loss": 0.22656936943531036, "eval_runtime": 337.9074, "eval_samples_per_second": 2.959, "eval_steps_per_second": 0.37, "eval_wer": 0.09776298475114173, "step": 3400 }, { "epoch": 6.26, "learning_rate": 1.0697083496945765e-05, "loss": 0.1979, "step": 3500 }, { "epoch": 6.43, "learning_rate": 9.412754953531663e-06, "loss": 0.1985, "step": 3600 }, { "epoch": 6.43, "eval_f1_score": 0.6856465005931198, "eval_label_f1": 0.8335310399367339, "eval_loss": 0.22672241926193237, "eval_runtime": 339.7265, "eval_samples_per_second": 2.944, "eval_steps_per_second": 0.368, "eval_wer": 0.0994659029336636, "step": 3600 }, { "epoch": 6.61, "learning_rate": 8.192477743467078e-06, "loss": 0.2006, "step": 3700 }, { "epoch": 6.79, "learning_rate": 7.041266247556813e-06, "loss": 0.1996, "step": 3800 }, { "epoch": 6.79, "eval_f1_score": 0.686437327006722, "eval_label_f1": 0.8343218663503361, "eval_loss": 0.226467102766037, "eval_runtime": 338.3929, "eval_samples_per_second": 2.955, "eval_steps_per_second": 0.369, "eval_wer": 0.0988079572722347, "step": 3800 }, { "epoch": 6.97, "learning_rate": 5.9638510407716394e-06, "loss": 0.2002, "step": 3900 }, { "epoch": 7.15, "learning_rate": 4.9646594533010875e-06, "loss": 0.197, "step": 4000 }, { "epoch": 7.15, "eval_f1_score": 0.6843354430379748, "eval_label_f1": 0.8346518987341772, "eval_loss": 0.2263847291469574, "eval_runtime": 338.0877, "eval_samples_per_second": 2.958, "eval_steps_per_second": 0.37, "eval_wer": 0.09857574115643626, "step": 4000 } ], "logging_steps": 100, "max_steps": 5000, "num_input_tokens_seen": 0, "num_train_epochs": 9, "save_steps": 200, "total_flos": 1.5746441158097424e+20, "train_batch_size": 4, "trial_name": null, "trial_params": null }