{ "best_metric": 0.06226002424955368, "best_model_checkpoint": "training_output/reranker_focal_answer/checkpoint-26182", "epoch": 26.0, "global_step": 26182, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.5, "learning_rate": 4.950347567030785e-05, "loss": 0.0273, "step": 500 }, { "epoch": 0.99, "learning_rate": 4.900695134061569e-05, "loss": 0.013, "step": 1000 }, { "epoch": 1.0, "eval_loss": 0.06812480837106705, "eval_runtime": 56.2831, "eval_samples_per_second": 556.508, "step": 1007 }, { "epoch": 1.49, "learning_rate": 4.8510427010923536e-05, "loss": 0.0109, "step": 1500 }, { "epoch": 1.99, "learning_rate": 4.801390268123138e-05, "loss": 0.0106, "step": 2000 }, { "epoch": 2.0, "eval_loss": 0.07386605441570282, "eval_runtime": 56.2854, "eval_samples_per_second": 556.486, "step": 2014 }, { "epoch": 2.48, "learning_rate": 4.7517378351539225e-05, "loss": 0.0095, "step": 2500 }, { "epoch": 2.98, "learning_rate": 4.702085402184707e-05, "loss": 0.0095, "step": 3000 }, { "epoch": 3.0, "eval_loss": 0.06689883023500443, "eval_runtime": 56.2997, "eval_samples_per_second": 556.344, "step": 3021 }, { "epoch": 3.48, "learning_rate": 4.6524329692154915e-05, "loss": 0.0083, "step": 3500 }, { "epoch": 3.97, "learning_rate": 4.602780536246276e-05, "loss": 0.0085, "step": 4000 }, { "epoch": 4.0, "eval_loss": 0.06948165595531464, "eval_runtime": 56.3068, "eval_samples_per_second": 556.274, "step": 4028 }, { "epoch": 4.47, "learning_rate": 4.5531281032770604e-05, "loss": 0.0072, "step": 4500 }, { "epoch": 4.97, "learning_rate": 4.503475670307845e-05, "loss": 0.0073, "step": 5000 }, { "epoch": 5.0, "eval_loss": 0.06620071828365326, "eval_runtime": 56.3226, "eval_samples_per_second": 556.118, "step": 5035 }, { "epoch": 5.46, "learning_rate": 4.453823237338629e-05, "loss": 0.0062, "step": 5500 }, { "epoch": 5.96, "learning_rate": 4.404170804369414e-05, "loss": 0.0064, "step": 6000 }, { "epoch": 6.0, "eval_loss": 0.06671101599931717, "eval_runtime": 56.334, "eval_samples_per_second": 556.005, "step": 6042 }, { "epoch": 6.45, "learning_rate": 4.354518371400199e-05, "loss": 0.0054, "step": 6500 }, { "epoch": 6.95, "learning_rate": 4.3048659384309834e-05, "loss": 0.0056, "step": 7000 }, { "epoch": 7.0, "eval_loss": 0.06806950271129608, "eval_runtime": 56.3386, "eval_samples_per_second": 555.96, "step": 7049 }, { "epoch": 7.45, "learning_rate": 4.255213505461768e-05, "loss": 0.0049, "step": 7500 }, { "epoch": 7.94, "learning_rate": 4.205561072492552e-05, "loss": 0.0049, "step": 8000 }, { "epoch": 8.0, "eval_loss": 0.06494650989770889, "eval_runtime": 56.3526, "eval_samples_per_second": 555.821, "step": 8056 }, { "epoch": 8.44, "learning_rate": 4.155908639523337e-05, "loss": 0.0043, "step": 8500 }, { "epoch": 8.94, "learning_rate": 4.106256206554122e-05, "loss": 0.0044, "step": 9000 }, { "epoch": 9.0, "eval_loss": 0.06944319605827332, "eval_runtime": 56.3534, "eval_samples_per_second": 555.814, "step": 9063 }, { "epoch": 9.43, "learning_rate": 4.0566037735849064e-05, "loss": 0.0038, "step": 9500 }, { "epoch": 9.93, "learning_rate": 4.006951340615691e-05, "loss": 0.0039, "step": 10000 }, { "epoch": 10.0, "eval_loss": 0.06448082625865936, "eval_runtime": 56.3612, "eval_samples_per_second": 555.737, "step": 10070 }, { "epoch": 10.43, "learning_rate": 3.957298907646475e-05, "loss": 0.0035, "step": 10500 }, { "epoch": 10.92, "learning_rate": 3.90764647467726e-05, "loss": 0.0035, "step": 11000 }, { "epoch": 11.0, "eval_loss": 0.06930028647184372, "eval_runtime": 56.3501, "eval_samples_per_second": 555.846, "step": 11077 }, { "epoch": 11.42, "learning_rate": 3.857994041708044e-05, "loss": 0.0031, "step": 11500 }, { "epoch": 11.92, "learning_rate": 3.8083416087388287e-05, "loss": 0.0032, "step": 12000 }, { "epoch": 12.0, "eval_loss": 0.06855376809835434, "eval_runtime": 56.3685, "eval_samples_per_second": 555.665, "step": 12084 }, { "epoch": 12.41, "learning_rate": 3.758689175769613e-05, "loss": 0.0028, "step": 12500 }, { "epoch": 12.91, "learning_rate": 3.7090367428003976e-05, "loss": 0.003, "step": 13000 }, { "epoch": 13.0, "eval_loss": 0.06912727653980255, "eval_runtime": 56.3545, "eval_samples_per_second": 555.803, "step": 13091 }, { "epoch": 13.41, "learning_rate": 3.659384309831182e-05, "loss": 0.0027, "step": 13500 }, { "epoch": 13.9, "learning_rate": 3.6097318768619665e-05, "loss": 0.0027, "step": 14000 }, { "epoch": 14.0, "eval_loss": 0.06820587813854218, "eval_runtime": 56.3582, "eval_samples_per_second": 555.767, "step": 14098 }, { "epoch": 14.4, "learning_rate": 3.560079443892751e-05, "loss": 0.0025, "step": 14500 }, { "epoch": 14.9, "learning_rate": 3.5104270109235354e-05, "loss": 0.0025, "step": 15000 }, { "epoch": 15.0, "eval_loss": 0.06939652562141418, "eval_runtime": 56.3589, "eval_samples_per_second": 555.76, "step": 15105 }, { "epoch": 15.39, "learning_rate": 3.46077457795432e-05, "loss": 0.0023, "step": 15500 }, { "epoch": 15.89, "learning_rate": 3.4111221449851043e-05, "loss": 0.0023, "step": 16000 }, { "epoch": 16.0, "eval_loss": 0.06309447437524796, "eval_runtime": 56.3564, "eval_samples_per_second": 555.784, "step": 16112 }, { "epoch": 16.39, "learning_rate": 3.361469712015889e-05, "loss": 0.0022, "step": 16500 }, { "epoch": 16.88, "learning_rate": 3.311817279046673e-05, "loss": 0.0023, "step": 17000 }, { "epoch": 17.0, "eval_loss": 0.06566380709409714, "eval_runtime": 56.4037, "eval_samples_per_second": 555.318, "step": 17119 }, { "epoch": 17.38, "learning_rate": 3.262164846077458e-05, "loss": 0.0021, "step": 17500 }, { "epoch": 17.87, "learning_rate": 3.212512413108242e-05, "loss": 0.002, "step": 18000 }, { "epoch": 18.0, "eval_loss": 0.06591986119747162, "eval_runtime": 56.3845, "eval_samples_per_second": 555.507, "step": 18126 }, { "epoch": 18.37, "learning_rate": 3.1628599801390267e-05, "loss": 0.002, "step": 18500 }, { "epoch": 18.87, "learning_rate": 3.113207547169811e-05, "loss": 0.002, "step": 19000 }, { "epoch": 19.0, "eval_loss": 0.0679154321551323, "eval_runtime": 56.3981, "eval_samples_per_second": 555.373, "step": 19133 }, { "epoch": 19.36, "learning_rate": 3.0635551142005956e-05, "loss": 0.0018, "step": 19500 }, { "epoch": 19.86, "learning_rate": 3.0139026812313804e-05, "loss": 0.0019, "step": 20000 }, { "epoch": 20.0, "eval_loss": 0.06612461805343628, "eval_runtime": 56.3877, "eval_samples_per_second": 555.476, "step": 20140 }, { "epoch": 20.36, "learning_rate": 2.964250248262165e-05, "loss": 0.0019, "step": 20500 }, { "epoch": 20.85, "learning_rate": 2.9145978152929493e-05, "loss": 0.0017, "step": 21000 }, { "epoch": 21.0, "eval_loss": 0.06515755504369736, "eval_runtime": 56.3934, "eval_samples_per_second": 555.42, "step": 21147 }, { "epoch": 21.35, "learning_rate": 2.8649453823237338e-05, "loss": 0.0017, "step": 21500 }, { "epoch": 21.85, "learning_rate": 2.8152929493545182e-05, "loss": 0.0018, "step": 22000 }, { "epoch": 22.0, "eval_loss": 0.062369268387556076, "eval_runtime": 56.3907, "eval_samples_per_second": 555.446, "step": 22154 }, { "epoch": 22.34, "learning_rate": 2.7656405163853027e-05, "loss": 0.0017, "step": 22500 }, { "epoch": 22.84, "learning_rate": 2.7159880834160878e-05, "loss": 0.0017, "step": 23000 }, { "epoch": 23.0, "eval_loss": 0.064911849796772, "eval_runtime": 56.3943, "eval_samples_per_second": 555.411, "step": 23161 }, { "epoch": 23.34, "learning_rate": 2.6663356504468723e-05, "loss": 0.0017, "step": 23500 }, { "epoch": 23.83, "learning_rate": 2.6166832174776567e-05, "loss": 0.0016, "step": 24000 }, { "epoch": 24.0, "eval_loss": 0.06586817651987076, "eval_runtime": 56.3912, "eval_samples_per_second": 555.441, "step": 24168 }, { "epoch": 24.33, "learning_rate": 2.5670307845084412e-05, "loss": 0.0016, "step": 24500 }, { "epoch": 24.83, "learning_rate": 2.5173783515392257e-05, "loss": 0.0015, "step": 25000 }, { "epoch": 25.0, "eval_loss": 0.06666699051856995, "eval_runtime": 56.3936, "eval_samples_per_second": 555.418, "step": 25175 }, { "epoch": 25.32, "learning_rate": 2.46772591857001e-05, "loss": 0.0015, "step": 25500 }, { "epoch": 25.82, "learning_rate": 2.4180734856007946e-05, "loss": 0.0016, "step": 26000 }, { "epoch": 26.0, "eval_loss": 0.06226002424955368, "eval_runtime": 56.4017, "eval_samples_per_second": 555.338, "step": 26182 } ], "max_steps": 50350, "num_train_epochs": 50, "total_flos": 1.4079942379067136e+17, "trial_name": null, "trial_params": null }