{ "best_metric": 2.3904106616973877, "best_model_checkpoint": "bert_base_120/checkpoint-80000", "epoch": 0.6903848989824509, "global_step": 84267, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 5.000000000000001e-07, "loss": 2.7523, "step": 500 }, { "epoch": 0.01, "learning_rate": 1.0000000000000002e-06, "loss": 2.7474, "step": 1000 }, { "epoch": 0.01, "learning_rate": 1.5e-06, "loss": 2.7487, "step": 1500 }, { "epoch": 0.02, "learning_rate": 2.0000000000000003e-06, "loss": 2.7421, "step": 2000 }, { "epoch": 0.02, "learning_rate": 2.5e-06, "loss": 2.7429, "step": 2500 }, { "epoch": 0.02, "learning_rate": 3e-06, "loss": 2.7478, "step": 3000 }, { "epoch": 0.03, "learning_rate": 3.5e-06, "loss": 2.7434, "step": 3500 }, { "epoch": 0.03, "learning_rate": 4.000000000000001e-06, "loss": 2.7475, "step": 4000 }, { "epoch": 0.04, "learning_rate": 4.5e-06, "loss": 2.7418, "step": 4500 }, { "epoch": 0.04, "learning_rate": 5e-06, "loss": 2.7485, "step": 5000 }, { "epoch": 0.05, "learning_rate": 5.500000000000001e-06, "loss": 2.7412, "step": 5500 }, { "epoch": 0.05, "learning_rate": 6e-06, "loss": 2.7444, "step": 6000 }, { "epoch": 0.05, "learning_rate": 6.5000000000000004e-06, "loss": 2.7471, "step": 6500 }, { "epoch": 0.06, "learning_rate": 7e-06, "loss": 2.7385, "step": 7000 }, { "epoch": 0.06, "learning_rate": 7.500000000000001e-06, "loss": 2.7401, "step": 7500 }, { "epoch": 0.07, "learning_rate": 8.000000000000001e-06, "loss": 2.7419, "step": 8000 }, { "epoch": 0.07, "learning_rate": 8.5e-06, "loss": 2.7371, "step": 8500 }, { "epoch": 0.07, "learning_rate": 9e-06, "loss": 2.7368, "step": 9000 }, { "epoch": 0.08, "learning_rate": 9.5e-06, "loss": 2.7372, "step": 9500 }, { "epoch": 0.08, "learning_rate": 1e-05, "loss": 2.7403, "step": 10000 }, { "epoch": 0.08, "eval_accuracy": 0.5307275776278991, "eval_loss": 2.6149816513061523, "eval_runtime": 1998.4296, "eval_samples_per_second": 154.299, "eval_steps_per_second": 3.215, "step": 10000 }, { "epoch": 0.09, "learning_rate": 9.991670692498626e-06, "loss": 2.7349, "step": 10500 }, { "epoch": 0.09, "learning_rate": 9.983341384997253e-06, "loss": 2.7466, "step": 11000 }, { "epoch": 0.09, "learning_rate": 9.975012077495878e-06, "loss": 2.7357, "step": 11500 }, { "epoch": 0.1, "learning_rate": 9.966682769994504e-06, "loss": 2.7319, "step": 12000 }, { "epoch": 0.1, "learning_rate": 9.95835346249313e-06, "loss": 2.7272, "step": 12500 }, { "epoch": 0.11, "learning_rate": 9.950024154991756e-06, "loss": 2.7236, "step": 13000 }, { "epoch": 0.11, "learning_rate": 9.941694847490381e-06, "loss": 2.7258, "step": 13500 }, { "epoch": 0.11, "learning_rate": 9.933365539989006e-06, "loss": 2.7251, "step": 14000 }, { "epoch": 0.12, "learning_rate": 9.925036232487631e-06, "loss": 2.7124, "step": 14500 }, { "epoch": 0.12, "learning_rate": 9.916706924986258e-06, "loss": 2.7167, "step": 15000 }, { "epoch": 0.13, "learning_rate": 9.908377617484883e-06, "loss": 2.7122, "step": 15500 }, { "epoch": 0.13, "learning_rate": 9.90004830998351e-06, "loss": 2.7135, "step": 16000 }, { "epoch": 0.14, "learning_rate": 9.891719002482134e-06, "loss": 2.7134, "step": 16500 }, { "epoch": 0.14, "learning_rate": 9.883389694980761e-06, "loss": 2.7056, "step": 17000 }, { "epoch": 0.14, "learning_rate": 9.875060387479386e-06, "loss": 2.7087, "step": 17500 }, { "epoch": 0.15, "learning_rate": 9.866731079978011e-06, "loss": 2.7075, "step": 18000 }, { "epoch": 0.15, "learning_rate": 9.858401772476636e-06, "loss": 2.708, "step": 18500 }, { "epoch": 0.16, "learning_rate": 9.850072464975263e-06, "loss": 2.7016, "step": 19000 }, { "epoch": 0.16, "learning_rate": 9.84174315747389e-06, "loss": 2.6974, "step": 19500 }, { "epoch": 0.16, "learning_rate": 9.833413849972514e-06, "loss": 2.6939, "step": 20000 }, { "epoch": 0.16, "eval_accuracy": 0.5359725723899631, "eval_loss": 2.5743441581726074, "eval_runtime": 1998.0838, "eval_samples_per_second": 154.326, "eval_steps_per_second": 3.216, "step": 20000 }, { "epoch": 0.17, "learning_rate": 9.82508454247114e-06, "loss": 2.6884, "step": 20500 }, { "epoch": 0.17, "learning_rate": 9.816755234969766e-06, "loss": 2.6943, "step": 21000 }, { "epoch": 0.18, "learning_rate": 9.808425927468391e-06, "loss": 2.6888, "step": 21500 }, { "epoch": 0.18, "learning_rate": 9.800096619967016e-06, "loss": 2.6861, "step": 22000 }, { "epoch": 0.18, "learning_rate": 9.791767312465643e-06, "loss": 2.6776, "step": 22500 }, { "epoch": 0.19, "learning_rate": 9.783438004964268e-06, "loss": 2.675, "step": 23000 }, { "epoch": 0.19, "learning_rate": 9.775108697462895e-06, "loss": 2.6808, "step": 23500 }, { "epoch": 0.2, "learning_rate": 9.76677938996152e-06, "loss": 2.6745, "step": 24000 }, { "epoch": 0.2, "learning_rate": 9.758450082460145e-06, "loss": 2.6728, "step": 24500 }, { "epoch": 0.2, "learning_rate": 9.750120774958771e-06, "loss": 2.6678, "step": 25000 }, { "epoch": 0.21, "learning_rate": 9.741791467457396e-06, "loss": 2.6678, "step": 25500 }, { "epoch": 0.21, "learning_rate": 9.733462159956021e-06, "loss": 2.6693, "step": 26000 }, { "epoch": 0.22, "learning_rate": 9.725132852454648e-06, "loss": 2.6714, "step": 26500 }, { "epoch": 0.22, "learning_rate": 9.716803544953275e-06, "loss": 2.6665, "step": 27000 }, { "epoch": 0.23, "learning_rate": 9.7084742374519e-06, "loss": 2.6631, "step": 27500 }, { "epoch": 0.23, "learning_rate": 9.700144929950525e-06, "loss": 2.6601, "step": 28000 }, { "epoch": 0.23, "learning_rate": 9.69181562244915e-06, "loss": 2.6616, "step": 28500 }, { "epoch": 0.24, "learning_rate": 9.683486314947776e-06, "loss": 2.6594, "step": 29000 }, { "epoch": 0.24, "learning_rate": 9.675157007446401e-06, "loss": 2.6583, "step": 29500 }, { "epoch": 0.25, "learning_rate": 9.666827699945028e-06, "loss": 2.6549, "step": 30000 }, { "epoch": 0.25, "eval_accuracy": 0.5407637846718732, "eval_loss": 2.5380136966705322, "eval_runtime": 1998.1587, "eval_samples_per_second": 154.32, "eval_steps_per_second": 3.215, "step": 30000 }, { "epoch": 0.25, "learning_rate": 9.658498392443653e-06, "loss": 2.6479, "step": 30500 }, { "epoch": 0.25, "learning_rate": 9.65016908494228e-06, "loss": 2.6497, "step": 31000 }, { "epoch": 0.26, "learning_rate": 9.641839777440905e-06, "loss": 2.6506, "step": 31500 }, { "epoch": 0.26, "learning_rate": 9.63351046993953e-06, "loss": 2.6488, "step": 32000 }, { "epoch": 0.27, "learning_rate": 9.625181162438155e-06, "loss": 2.6441, "step": 32500 }, { "epoch": 0.27, "learning_rate": 9.616851854936781e-06, "loss": 2.6479, "step": 33000 }, { "epoch": 0.27, "learning_rate": 9.608522547435406e-06, "loss": 2.644, "step": 33500 }, { "epoch": 0.28, "learning_rate": 9.600193239934033e-06, "loss": 2.6387, "step": 34000 }, { "epoch": 0.28, "learning_rate": 9.591863932432658e-06, "loss": 2.6371, "step": 34500 }, { "epoch": 0.29, "learning_rate": 9.583534624931285e-06, "loss": 2.6357, "step": 35000 }, { "epoch": 0.29, "learning_rate": 9.57520531742991e-06, "loss": 2.6427, "step": 35500 }, { "epoch": 0.29, "learning_rate": 9.566876009928535e-06, "loss": 2.6358, "step": 36000 }, { "epoch": 0.3, "learning_rate": 9.55854670242716e-06, "loss": 2.6348, "step": 36500 }, { "epoch": 0.3, "learning_rate": 9.550217394925787e-06, "loss": 2.6298, "step": 37000 }, { "epoch": 0.31, "learning_rate": 9.541888087424413e-06, "loss": 2.6265, "step": 37500 }, { "epoch": 0.31, "learning_rate": 9.533558779923038e-06, "loss": 2.6227, "step": 38000 }, { "epoch": 0.32, "learning_rate": 9.525229472421663e-06, "loss": 2.6138, "step": 38500 }, { "epoch": 0.32, "learning_rate": 9.51690016492029e-06, "loss": 2.6198, "step": 39000 }, { "epoch": 0.32, "learning_rate": 9.508570857418915e-06, "loss": 2.6176, "step": 39500 }, { "epoch": 0.33, "learning_rate": 9.50024154991754e-06, "loss": 2.6298, "step": 40000 }, { "epoch": 0.33, "eval_accuracy": 0.5454560769274944, "eval_loss": 2.5020463466644287, "eval_runtime": 1995.7897, "eval_samples_per_second": 154.503, "eval_steps_per_second": 3.219, "step": 40000 }, { "epoch": 0.33, "learning_rate": 9.491912242416167e-06, "loss": 2.6179, "step": 40500 }, { "epoch": 0.34, "learning_rate": 9.483582934914792e-06, "loss": 2.6155, "step": 41000 }, { "epoch": 0.34, "learning_rate": 9.475253627413418e-06, "loss": 2.6142, "step": 41500 }, { "epoch": 0.34, "learning_rate": 9.466924319912043e-06, "loss": 2.618, "step": 42000 }, { "epoch": 0.35, "learning_rate": 9.458595012410668e-06, "loss": 2.607, "step": 42500 }, { "epoch": 0.35, "learning_rate": 9.450265704909295e-06, "loss": 2.608, "step": 43000 }, { "epoch": 0.36, "learning_rate": 9.44193639740792e-06, "loss": 2.5987, "step": 43500 }, { "epoch": 0.36, "learning_rate": 9.433607089906545e-06, "loss": 2.6078, "step": 44000 }, { "epoch": 0.36, "learning_rate": 9.425277782405172e-06, "loss": 2.604, "step": 44500 }, { "epoch": 0.37, "learning_rate": 9.416948474903798e-06, "loss": 2.6037, "step": 45000 }, { "epoch": 0.37, "learning_rate": 9.408619167402423e-06, "loss": 2.606, "step": 45500 }, { "epoch": 0.38, "learning_rate": 9.400289859901048e-06, "loss": 2.5939, "step": 46000 }, { "epoch": 0.38, "learning_rate": 9.391960552399673e-06, "loss": 2.5937, "step": 46500 }, { "epoch": 0.39, "learning_rate": 9.3836312448983e-06, "loss": 2.5915, "step": 47000 }, { "epoch": 0.39, "learning_rate": 9.375301937396925e-06, "loss": 2.5918, "step": 47500 }, { "epoch": 0.39, "learning_rate": 9.366972629895552e-06, "loss": 2.5911, "step": 48000 }, { "epoch": 0.4, "learning_rate": 9.358643322394177e-06, "loss": 2.5931, "step": 48500 }, { "epoch": 0.4, "learning_rate": 9.350314014892804e-06, "loss": 2.5857, "step": 49000 }, { "epoch": 0.41, "learning_rate": 9.341984707391429e-06, "loss": 2.5877, "step": 49500 }, { "epoch": 0.41, "learning_rate": 9.333655399890054e-06, "loss": 2.5883, "step": 50000 }, { "epoch": 0.41, "eval_accuracy": 0.549350897420584, "eval_loss": 2.4715354442596436, "eval_runtime": 1998.6634, "eval_samples_per_second": 154.281, "eval_steps_per_second": 3.215, "step": 50000 }, { "epoch": 0.41, "learning_rate": 9.325326092388679e-06, "loss": 2.5816, "step": 50500 }, { "epoch": 0.42, "learning_rate": 9.316996784887305e-06, "loss": 2.5851, "step": 51000 }, { "epoch": 0.42, "learning_rate": 9.30866747738593e-06, "loss": 2.5841, "step": 51500 }, { "epoch": 0.43, "learning_rate": 9.300338169884557e-06, "loss": 2.585, "step": 52000 }, { "epoch": 0.43, "learning_rate": 9.292008862383182e-06, "loss": 2.5799, "step": 52500 }, { "epoch": 0.43, "learning_rate": 9.283679554881809e-06, "loss": 2.5759, "step": 53000 }, { "epoch": 0.44, "learning_rate": 9.275350247380434e-06, "loss": 2.5769, "step": 53500 }, { "epoch": 0.44, "learning_rate": 9.267020939879059e-06, "loss": 2.5702, "step": 54000 }, { "epoch": 0.45, "learning_rate": 9.258691632377685e-06, "loss": 2.573, "step": 54500 }, { "epoch": 0.45, "learning_rate": 9.25036232487631e-06, "loss": 2.5696, "step": 55000 }, { "epoch": 0.45, "learning_rate": 9.242033017374937e-06, "loss": 2.5704, "step": 55500 }, { "epoch": 0.46, "learning_rate": 9.233703709873562e-06, "loss": 2.5646, "step": 56000 }, { "epoch": 0.46, "learning_rate": 9.225374402372189e-06, "loss": 2.5668, "step": 56500 }, { "epoch": 0.47, "learning_rate": 9.217045094870814e-06, "loss": 2.5582, "step": 57000 }, { "epoch": 0.47, "learning_rate": 9.208715787369439e-06, "loss": 2.5651, "step": 57500 }, { "epoch": 0.48, "learning_rate": 9.200386479868064e-06, "loss": 2.5591, "step": 58000 }, { "epoch": 0.48, "learning_rate": 9.19205717236669e-06, "loss": 2.5582, "step": 58500 }, { "epoch": 0.48, "learning_rate": 9.183727864865315e-06, "loss": 2.5635, "step": 59000 }, { "epoch": 0.49, "learning_rate": 9.175398557363942e-06, "loss": 2.5627, "step": 59500 }, { "epoch": 0.49, "learning_rate": 9.167069249862567e-06, "loss": 2.5629, "step": 60000 }, { "epoch": 0.49, "eval_accuracy": 0.5533242043441243, "eval_loss": 2.443215847015381, "eval_runtime": 1998.0183, "eval_samples_per_second": 154.331, "eval_steps_per_second": 3.216, "step": 60000 }, { "epoch": 0.5, "learning_rate": 9.158739942361194e-06, "loss": 2.5553, "step": 60500 }, { "epoch": 0.5, "learning_rate": 9.150410634859819e-06, "loss": 2.5485, "step": 61000 }, { "epoch": 0.5, "learning_rate": 9.142081327358444e-06, "loss": 2.5559, "step": 61500 }, { "epoch": 0.51, "learning_rate": 9.133752019857069e-06, "loss": 2.5456, "step": 62000 }, { "epoch": 0.51, "learning_rate": 9.125422712355696e-06, "loss": 2.5513, "step": 62500 }, { "epoch": 0.52, "learning_rate": 9.11709340485432e-06, "loss": 2.5507, "step": 63000 }, { "epoch": 0.52, "learning_rate": 9.108764097352947e-06, "loss": 2.539, "step": 63500 }, { "epoch": 0.52, "learning_rate": 9.100434789851572e-06, "loss": 2.5409, "step": 64000 }, { "epoch": 0.53, "learning_rate": 9.092105482350199e-06, "loss": 2.5423, "step": 64500 }, { "epoch": 0.53, "learning_rate": 9.083776174848824e-06, "loss": 2.5396, "step": 65000 }, { "epoch": 0.54, "learning_rate": 9.075446867347449e-06, "loss": 2.5476, "step": 65500 }, { "epoch": 0.54, "learning_rate": 9.067117559846076e-06, "loss": 2.5416, "step": 66000 }, { "epoch": 0.54, "learning_rate": 9.0587882523447e-06, "loss": 2.5371, "step": 66500 }, { "epoch": 0.55, "learning_rate": 9.050458944843327e-06, "loss": 2.5287, "step": 67000 }, { "epoch": 0.55, "learning_rate": 9.042129637341952e-06, "loss": 2.5351, "step": 67500 }, { "epoch": 0.56, "learning_rate": 9.033800329840577e-06, "loss": 2.5331, "step": 68000 }, { "epoch": 0.56, "learning_rate": 9.025471022339204e-06, "loss": 2.5372, "step": 68500 }, { "epoch": 0.57, "learning_rate": 9.017141714837829e-06, "loss": 2.5306, "step": 69000 }, { "epoch": 0.57, "learning_rate": 9.008812407336454e-06, "loss": 2.5352, "step": 69500 }, { "epoch": 0.57, "learning_rate": 9.00048309983508e-06, "loss": 2.5274, "step": 70000 }, { "epoch": 0.57, "eval_accuracy": 0.5567950069381908, "eval_loss": 2.416321039199829, "eval_runtime": 1992.5879, "eval_samples_per_second": 154.752, "eval_steps_per_second": 3.224, "step": 70000 }, { "epoch": 0.58, "learning_rate": 8.992153792333706e-06, "loss": 2.5254, "step": 70500 }, { "epoch": 0.58, "learning_rate": 8.983824484832332e-06, "loss": 2.5241, "step": 71000 }, { "epoch": 0.59, "learning_rate": 8.975495177330957e-06, "loss": 2.5314, "step": 71500 }, { "epoch": 0.59, "learning_rate": 8.967165869829582e-06, "loss": 2.5209, "step": 72000 }, { "epoch": 0.59, "learning_rate": 8.958836562328209e-06, "loss": 2.5258, "step": 72500 }, { "epoch": 0.6, "learning_rate": 8.950507254826834e-06, "loss": 2.5182, "step": 73000 }, { "epoch": 0.6, "learning_rate": 8.942177947325459e-06, "loss": 2.5229, "step": 73500 }, { "epoch": 0.61, "learning_rate": 8.933848639824086e-06, "loss": 2.5199, "step": 74000 }, { "epoch": 0.61, "learning_rate": 8.925519332322713e-06, "loss": 2.5187, "step": 74500 }, { "epoch": 0.61, "learning_rate": 8.917190024821338e-06, "loss": 2.5218, "step": 75000 }, { "epoch": 0.62, "learning_rate": 8.908860717319963e-06, "loss": 2.5148, "step": 75500 }, { "epoch": 0.62, "learning_rate": 8.900531409818588e-06, "loss": 2.5116, "step": 76000 }, { "epoch": 0.63, "learning_rate": 8.892202102317214e-06, "loss": 2.5101, "step": 76500 }, { "epoch": 0.63, "learning_rate": 8.88387279481584e-06, "loss": 2.5088, "step": 77000 }, { "epoch": 0.63, "learning_rate": 8.875543487314466e-06, "loss": 2.5087, "step": 77500 }, { "epoch": 0.64, "learning_rate": 8.867214179813091e-06, "loss": 2.5029, "step": 78000 }, { "epoch": 0.64, "learning_rate": 8.858884872311718e-06, "loss": 2.5093, "step": 78500 }, { "epoch": 0.65, "learning_rate": 8.850555564810343e-06, "loss": 2.5005, "step": 79000 }, { "epoch": 0.65, "learning_rate": 8.842226257308968e-06, "loss": 2.5108, "step": 79500 }, { "epoch": 0.66, "learning_rate": 8.833896949807593e-06, "loss": 2.5059, "step": 80000 }, { "epoch": 0.66, "eval_accuracy": 0.5602219706284388, "eval_loss": 2.3904106616973877, "eval_runtime": 1994.2916, "eval_samples_per_second": 154.619, "eval_steps_per_second": 3.222, "step": 80000 }, { "epoch": 0.66, "learning_rate": 8.82556764230622e-06, "loss": 2.503, "step": 80500 }, { "epoch": 0.66, "learning_rate": 8.817238334804844e-06, "loss": 2.5056, "step": 81000 }, { "epoch": 0.67, "learning_rate": 8.808909027303471e-06, "loss": 2.501, "step": 81500 }, { "epoch": 0.67, "learning_rate": 8.800579719802096e-06, "loss": 2.4976, "step": 82000 }, { "epoch": 0.68, "learning_rate": 8.792250412300723e-06, "loss": 2.4901, "step": 82500 }, { "epoch": 0.68, "learning_rate": 8.783921104799348e-06, "loss": 2.4993, "step": 83000 }, { "epoch": 0.68, "learning_rate": 8.775591797297973e-06, "loss": 2.4963, "step": 83500 }, { "epoch": 0.69, "learning_rate": 8.767262489796598e-06, "loss": 2.4949, "step": 84000 }, { "epoch": 0.69, "step": 84267, "total_flos": 1.0646150356758692e+18, "train_loss": 2.6181558959414812, "train_runtime": 89967.2417, "train_samples_per_second": 325.605, "train_steps_per_second": 6.783 } ], "max_steps": 610290, "num_train_epochs": 5, "total_flos": 1.0646150356758692e+18, "trial_name": null, "trial_params": null }