{ "best_metric": 6.008976459503174, "best_model_checkpoint": "bert_base_24/checkpoint-80000", "epoch": 0.6554343017254093, "global_step": 80001, "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": 9.9822, "step": 500 }, { "epoch": 0.01, "learning_rate": 1.0000000000000002e-06, "loss": 9.2026, "step": 1000 }, { "epoch": 0.01, "learning_rate": 1.5e-06, "loss": 8.7769, "step": 1500 }, { "epoch": 0.02, "learning_rate": 2.0000000000000003e-06, "loss": 8.297, "step": 2000 }, { "epoch": 0.02, "learning_rate": 2.5e-06, "loss": 7.7815, "step": 2500 }, { "epoch": 0.02, "learning_rate": 3e-06, "loss": 7.3833, "step": 3000 }, { "epoch": 0.03, "learning_rate": 3.5e-06, "loss": 7.1337, "step": 3500 }, { "epoch": 0.03, "learning_rate": 4.000000000000001e-06, "loss": 7.0008, "step": 4000 }, { "epoch": 0.04, "learning_rate": 4.5e-06, "loss": 6.8954, "step": 4500 }, { "epoch": 0.04, "learning_rate": 5e-06, "loss": 6.8278, "step": 5000 }, { "epoch": 0.05, "learning_rate": 5.500000000000001e-06, "loss": 6.763, "step": 5500 }, { "epoch": 0.05, "learning_rate": 6e-06, "loss": 6.7141, "step": 6000 }, { "epoch": 0.05, "learning_rate": 6.5000000000000004e-06, "loss": 6.6796, "step": 6500 }, { "epoch": 0.06, "learning_rate": 7e-06, "loss": 6.6416, "step": 7000 }, { "epoch": 0.06, "learning_rate": 7.500000000000001e-06, "loss": 6.6118, "step": 7500 }, { "epoch": 0.07, "learning_rate": 8.000000000000001e-06, "loss": 6.5762, "step": 8000 }, { "epoch": 0.07, "learning_rate": 8.5e-06, "loss": 6.5481, "step": 8500 }, { "epoch": 0.07, "learning_rate": 9e-06, "loss": 6.5293, "step": 9000 }, { "epoch": 0.08, "learning_rate": 9.5e-06, "loss": 6.506, "step": 9500 }, { "epoch": 0.08, "learning_rate": 1e-05, "loss": 6.4917, "step": 10000 }, { "epoch": 0.08, "eval_accuracy": 0.1406164011008725, "eval_loss": 6.442233562469482, "eval_runtime": 2026.7135, "eval_samples_per_second": 152.146, "eval_steps_per_second": 3.17, "step": 10000 }, { "epoch": 0.09, "learning_rate": 9.991670692498626e-06, "loss": 6.4711, "step": 10500 }, { "epoch": 0.09, "learning_rate": 9.983341384997253e-06, "loss": 6.4602, "step": 11000 }, { "epoch": 0.09, "learning_rate": 9.975012077495878e-06, "loss": 6.4399, "step": 11500 }, { "epoch": 0.1, "learning_rate": 9.966682769994504e-06, "loss": 6.428, "step": 12000 }, { "epoch": 0.1, "learning_rate": 9.95835346249313e-06, "loss": 6.4067, "step": 12500 }, { "epoch": 0.11, "learning_rate": 9.950024154991756e-06, "loss": 6.3975, "step": 13000 }, { "epoch": 0.11, "learning_rate": 9.941694847490381e-06, "loss": 6.3889, "step": 13500 }, { "epoch": 0.11, "learning_rate": 9.933365539989006e-06, "loss": 6.3798, "step": 14000 }, { "epoch": 0.12, "learning_rate": 9.925036232487631e-06, "loss": 6.363, "step": 14500 }, { "epoch": 0.12, "learning_rate": 9.916706924986258e-06, "loss": 6.3547, "step": 15000 }, { "epoch": 0.13, "learning_rate": 9.908377617484883e-06, "loss": 6.3391, "step": 15500 }, { "epoch": 0.13, "learning_rate": 9.90004830998351e-06, "loss": 6.3392, "step": 16000 }, { "epoch": 0.14, "learning_rate": 9.891719002482134e-06, "loss": 6.3375, "step": 16500 }, { "epoch": 0.14, "learning_rate": 9.883389694980761e-06, "loss": 6.319, "step": 17000 }, { "epoch": 0.14, "learning_rate": 9.875060387479386e-06, "loss": 6.3203, "step": 17500 }, { "epoch": 0.15, "learning_rate": 9.866731079978011e-06, "loss": 6.3119, "step": 18000 }, { "epoch": 0.15, "learning_rate": 9.858401772476636e-06, "loss": 6.311, "step": 18500 }, { "epoch": 0.16, "learning_rate": 9.850072464975263e-06, "loss": 6.296, "step": 19000 }, { "epoch": 0.16, "learning_rate": 9.84174315747389e-06, "loss": 6.2894, "step": 19500 }, { "epoch": 0.16, "learning_rate": 9.833413849972514e-06, "loss": 6.2848, "step": 20000 }, { "epoch": 0.16, "eval_accuracy": 0.14784890360476463, "eval_loss": 6.264363765716553, "eval_runtime": 2026.7257, "eval_samples_per_second": 152.145, "eval_steps_per_second": 3.17, "step": 20000 }, { "epoch": 0.17, "learning_rate": 9.82508454247114e-06, "loss": 6.279, "step": 20500 }, { "epoch": 0.17, "learning_rate": 9.816755234969766e-06, "loss": 6.2699, "step": 21000 }, { "epoch": 0.18, "learning_rate": 9.808425927468391e-06, "loss": 6.2756, "step": 21500 }, { "epoch": 0.18, "learning_rate": 9.800096619967016e-06, "loss": 6.2597, "step": 22000 }, { "epoch": 0.18, "learning_rate": 9.791767312465643e-06, "loss": 6.2627, "step": 22500 }, { "epoch": 0.19, "learning_rate": 9.783438004964268e-06, "loss": 6.2457, "step": 23000 }, { "epoch": 0.19, "learning_rate": 9.775108697462895e-06, "loss": 6.2424, "step": 23500 }, { "epoch": 0.2, "learning_rate": 9.76677938996152e-06, "loss": 6.2365, "step": 24000 }, { "epoch": 0.2, "learning_rate": 9.758450082460145e-06, "loss": 6.2395, "step": 24500 }, { "epoch": 0.2, "learning_rate": 9.750120774958771e-06, "loss": 6.2347, "step": 25000 }, { "epoch": 0.21, "learning_rate": 9.741791467457396e-06, "loss": 6.2256, "step": 25500 }, { "epoch": 0.21, "learning_rate": 9.733462159956021e-06, "loss": 6.2239, "step": 26000 }, { "epoch": 0.22, "learning_rate": 9.725132852454648e-06, "loss": 6.2282, "step": 26500 }, { "epoch": 0.22, "learning_rate": 9.716803544953275e-06, "loss": 6.2222, "step": 27000 }, { "epoch": 0.23, "learning_rate": 9.7084742374519e-06, "loss": 6.213, "step": 27500 }, { "epoch": 0.23, "learning_rate": 9.700144929950525e-06, "loss": 6.2109, "step": 28000 }, { "epoch": 0.23, "learning_rate": 9.69181562244915e-06, "loss": 6.2101, "step": 28500 }, { "epoch": 0.24, "learning_rate": 9.683486314947776e-06, "loss": 6.2061, "step": 29000 }, { "epoch": 0.24, "learning_rate": 9.675157007446401e-06, "loss": 6.199, "step": 29500 }, { "epoch": 0.25, "learning_rate": 9.666827699945028e-06, "loss": 6.1988, "step": 30000 }, { "epoch": 0.25, "eval_accuracy": 0.1493027153779554, "eval_loss": 6.185213088989258, "eval_runtime": 2025.4966, "eval_samples_per_second": 152.237, "eval_steps_per_second": 3.172, "step": 30000 }, { "epoch": 0.25, "learning_rate": 9.658498392443653e-06, "loss": 6.1916, "step": 30500 }, { "epoch": 0.25, "learning_rate": 9.65016908494228e-06, "loss": 6.1918, "step": 31000 }, { "epoch": 0.26, "learning_rate": 9.641839777440905e-06, "loss": 6.1931, "step": 31500 }, { "epoch": 0.26, "learning_rate": 9.63351046993953e-06, "loss": 6.1847, "step": 32000 }, { "epoch": 0.27, "learning_rate": 9.625181162438155e-06, "loss": 6.1801, "step": 32500 }, { "epoch": 0.27, "learning_rate": 9.616851854936781e-06, "loss": 6.178, "step": 33000 }, { "epoch": 0.27, "learning_rate": 9.608522547435406e-06, "loss": 6.1729, "step": 33500 }, { "epoch": 0.28, "learning_rate": 9.600193239934033e-06, "loss": 6.1726, "step": 34000 }, { "epoch": 0.28, "learning_rate": 9.591863932432658e-06, "loss": 6.1659, "step": 34500 }, { "epoch": 0.29, "learning_rate": 9.583534624931285e-06, "loss": 6.1674, "step": 35000 }, { "epoch": 0.29, "learning_rate": 9.57520531742991e-06, "loss": 6.1718, "step": 35500 }, { "epoch": 0.29, "learning_rate": 9.566876009928535e-06, "loss": 6.1651, "step": 36000 }, { "epoch": 0.3, "learning_rate": 9.55854670242716e-06, "loss": 6.1618, "step": 36500 }, { "epoch": 0.3, "learning_rate": 9.550217394925787e-06, "loss": 6.1585, "step": 37000 }, { "epoch": 0.31, "learning_rate": 9.541888087424413e-06, "loss": 6.1559, "step": 37500 }, { "epoch": 0.31, "learning_rate": 9.533558779923038e-06, "loss": 6.1451, "step": 38000 }, { "epoch": 0.32, "learning_rate": 9.525229472421663e-06, "loss": 6.1481, "step": 38500 }, { "epoch": 0.32, "learning_rate": 9.51690016492029e-06, "loss": 6.1411, "step": 39000 }, { "epoch": 0.32, "learning_rate": 9.508570857418915e-06, "loss": 6.1441, "step": 39500 }, { "epoch": 0.33, "learning_rate": 9.50024154991754e-06, "loss": 6.148, "step": 40000 }, { "epoch": 0.33, "eval_accuracy": 0.15008841811581047, "eval_loss": 6.128693580627441, "eval_runtime": 2030.9811, "eval_samples_per_second": 151.826, "eval_steps_per_second": 3.163, "step": 40000 }, { "epoch": 0.33, "learning_rate": 9.491912242416167e-06, "loss": 6.1396, "step": 40500 }, { "epoch": 0.34, "learning_rate": 9.483582934914792e-06, "loss": 6.1376, "step": 41000 }, { "epoch": 0.34, "learning_rate": 9.475253627413418e-06, "loss": 6.1338, "step": 41500 }, { "epoch": 0.34, "learning_rate": 9.466924319912043e-06, "loss": 6.1375, "step": 42000 }, { "epoch": 0.35, "learning_rate": 9.458595012410668e-06, "loss": 6.1254, "step": 42500 }, { "epoch": 0.35, "learning_rate": 9.450265704909295e-06, "loss": 6.1252, "step": 43000 }, { "epoch": 0.36, "learning_rate": 9.44193639740792e-06, "loss": 6.1209, "step": 43500 }, { "epoch": 0.36, "learning_rate": 9.433607089906545e-06, "loss": 6.1252, "step": 44000 }, { "epoch": 0.36, "learning_rate": 9.425277782405172e-06, "loss": 6.1205, "step": 44500 }, { "epoch": 0.37, "learning_rate": 9.416948474903798e-06, "loss": 6.1177, "step": 45000 }, { "epoch": 0.37, "learning_rate": 9.408619167402423e-06, "loss": 6.1204, "step": 45500 }, { "epoch": 0.38, "learning_rate": 9.400289859901048e-06, "loss": 6.1101, "step": 46000 }, { "epoch": 0.38, "learning_rate": 9.391960552399673e-06, "loss": 6.1094, "step": 46500 }, { "epoch": 0.39, "learning_rate": 9.3836312448983e-06, "loss": 6.1038, "step": 47000 }, { "epoch": 0.39, "learning_rate": 9.375301937396925e-06, "loss": 6.1052, "step": 47500 }, { "epoch": 0.39, "learning_rate": 9.366972629895552e-06, "loss": 6.1056, "step": 48000 }, { "epoch": 0.4, "learning_rate": 9.358643322394177e-06, "loss": 6.1009, "step": 48500 }, { "epoch": 0.4, "learning_rate": 9.350314014892804e-06, "loss": 6.1006, "step": 49000 }, { "epoch": 0.41, "learning_rate": 9.341984707391429e-06, "loss": 6.1009, "step": 49500 }, { "epoch": 0.41, "learning_rate": 9.333655399890054e-06, "loss": 6.1007, "step": 50000 }, { "epoch": 0.41, "eval_accuracy": 0.1500883878026566, "eval_loss": 6.088829040527344, "eval_runtime": 2025.8132, "eval_samples_per_second": 152.213, "eval_steps_per_second": 3.172, "step": 50000 }, { "epoch": 0.41, "learning_rate": 9.325326092388679e-06, "loss": 6.0929, "step": 50500 }, { "epoch": 0.42, "learning_rate": 9.316996784887305e-06, "loss": 6.0942, "step": 51000 }, { "epoch": 0.42, "learning_rate": 9.30866747738593e-06, "loss": 6.097, "step": 51500 }, { "epoch": 0.43, "learning_rate": 9.300338169884557e-06, "loss": 6.0922, "step": 52000 }, { "epoch": 0.43, "learning_rate": 9.292008862383182e-06, "loss": 6.088, "step": 52500 }, { "epoch": 0.43, "learning_rate": 9.283679554881809e-06, "loss": 6.0922, "step": 53000 }, { "epoch": 0.44, "learning_rate": 9.275350247380434e-06, "loss": 6.0847, "step": 53500 }, { "epoch": 0.44, "learning_rate": 9.267020939879059e-06, "loss": 6.0872, "step": 54000 }, { "epoch": 0.45, "learning_rate": 9.258691632377685e-06, "loss": 6.0861, "step": 54500 }, { "epoch": 0.45, "learning_rate": 9.25036232487631e-06, "loss": 6.0777, "step": 55000 }, { "epoch": 0.45, "learning_rate": 9.242033017374937e-06, "loss": 6.0849, "step": 55500 }, { "epoch": 0.46, "learning_rate": 9.233703709873562e-06, "loss": 6.0792, "step": 56000 }, { "epoch": 0.46, "learning_rate": 9.225374402372189e-06, "loss": 6.0751, "step": 56500 }, { "epoch": 0.47, "learning_rate": 9.217045094870814e-06, "loss": 6.066, "step": 57000 }, { "epoch": 0.47, "learning_rate": 9.208715787369439e-06, "loss": 6.0767, "step": 57500 }, { "epoch": 0.48, "learning_rate": 9.200386479868064e-06, "loss": 6.0771, "step": 58000 }, { "epoch": 0.48, "learning_rate": 9.19205717236669e-06, "loss": 6.0733, "step": 58500 }, { "epoch": 0.48, "learning_rate": 9.183727864865315e-06, "loss": 6.0693, "step": 59000 }, { "epoch": 0.49, "learning_rate": 9.175398557363942e-06, "loss": 6.0739, "step": 59500 }, { "epoch": 0.49, "learning_rate": 9.167069249862567e-06, "loss": 6.0721, "step": 60000 }, { "epoch": 0.49, "eval_accuracy": 0.14991843929852017, "eval_loss": 6.0554890632629395, "eval_runtime": 2030.6269, "eval_samples_per_second": 151.853, "eval_steps_per_second": 3.164, "step": 60000 }, { "epoch": 0.5, "learning_rate": 9.158739942361194e-06, "loss": 6.062, "step": 60500 }, { "epoch": 0.5, "learning_rate": 9.150410634859819e-06, "loss": 6.0598, "step": 61000 }, { "epoch": 0.5, "learning_rate": 9.142081327358444e-06, "loss": 6.0644, "step": 61500 }, { "epoch": 0.51, "learning_rate": 9.133752019857069e-06, "loss": 6.0561, "step": 62000 }, { "epoch": 0.51, "learning_rate": 9.125422712355696e-06, "loss": 6.0606, "step": 62500 }, { "epoch": 0.52, "learning_rate": 9.11709340485432e-06, "loss": 6.0611, "step": 63000 }, { "epoch": 0.52, "learning_rate": 9.108764097352947e-06, "loss": 6.0494, "step": 63500 }, { "epoch": 0.52, "learning_rate": 9.100434789851572e-06, "loss": 6.0507, "step": 64000 }, { "epoch": 0.53, "learning_rate": 9.092105482350199e-06, "loss": 6.0536, "step": 64500 }, { "epoch": 0.53, "learning_rate": 9.083776174848824e-06, "loss": 6.0523, "step": 65000 }, { "epoch": 0.54, "learning_rate": 9.075446867347449e-06, "loss": 6.0545, "step": 65500 }, { "epoch": 0.54, "learning_rate": 9.067117559846076e-06, "loss": 6.0559, "step": 66000 }, { "epoch": 0.54, "learning_rate": 9.0587882523447e-06, "loss": 6.0449, "step": 66500 }, { "epoch": 0.55, "learning_rate": 9.050458944843327e-06, "loss": 6.0449, "step": 67000 }, { "epoch": 0.55, "learning_rate": 9.042129637341952e-06, "loss": 6.0459, "step": 67500 }, { "epoch": 0.56, "learning_rate": 9.033800329840577e-06, "loss": 6.0498, "step": 68000 }, { "epoch": 0.56, "learning_rate": 9.025471022339204e-06, "loss": 6.0462, "step": 68500 }, { "epoch": 0.57, "learning_rate": 9.017141714837829e-06, "loss": 6.043, "step": 69000 }, { "epoch": 0.57, "learning_rate": 9.008812407336454e-06, "loss": 6.0461, "step": 69500 }, { "epoch": 0.57, "learning_rate": 9.00048309983508e-06, "loss": 6.0414, "step": 70000 }, { "epoch": 0.57, "eval_accuracy": 0.15135291944253934, "eval_loss": 6.027411937713623, "eval_runtime": 2027.398, "eval_samples_per_second": 152.094, "eval_steps_per_second": 3.169, "step": 70000 }, { "epoch": 0.58, "learning_rate": 8.992153792333706e-06, "loss": 6.0374, "step": 70500 }, { "epoch": 0.58, "learning_rate": 8.983824484832332e-06, "loss": 6.0307, "step": 71000 }, { "epoch": 0.59, "learning_rate": 8.975495177330957e-06, "loss": 6.0401, "step": 71500 }, { "epoch": 0.59, "learning_rate": 8.967165869829582e-06, "loss": 6.0376, "step": 72000 }, { "epoch": 0.59, "learning_rate": 8.958836562328209e-06, "loss": 6.0373, "step": 72500 }, { "epoch": 0.6, "learning_rate": 8.950507254826834e-06, "loss": 6.0303, "step": 73000 }, { "epoch": 0.6, "learning_rate": 8.942177947325459e-06, "loss": 6.0365, "step": 73500 }, { "epoch": 0.61, "learning_rate": 8.933848639824086e-06, "loss": 6.0295, "step": 74000 }, { "epoch": 0.61, "learning_rate": 8.925519332322713e-06, "loss": 6.0279, "step": 74500 }, { "epoch": 0.61, "learning_rate": 8.917190024821338e-06, "loss": 6.0316, "step": 75000 }, { "epoch": 0.62, "learning_rate": 8.908860717319963e-06, "loss": 6.0249, "step": 75500 }, { "epoch": 0.62, "learning_rate": 8.900531409818588e-06, "loss": 6.0282, "step": 76000 }, { "epoch": 0.63, "learning_rate": 8.892202102317214e-06, "loss": 6.0228, "step": 76500 }, { "epoch": 0.63, "learning_rate": 8.88387279481584e-06, "loss": 6.0238, "step": 77000 }, { "epoch": 0.63, "learning_rate": 8.875543487314466e-06, "loss": 6.0215, "step": 77500 }, { "epoch": 0.64, "learning_rate": 8.867214179813091e-06, "loss": 6.0204, "step": 78000 }, { "epoch": 0.64, "learning_rate": 8.858884872311718e-06, "loss": 6.0249, "step": 78500 }, { "epoch": 0.65, "learning_rate": 8.850555564810343e-06, "loss": 6.0199, "step": 79000 }, { "epoch": 0.65, "learning_rate": 8.842226257308968e-06, "loss": 6.0208, "step": 79500 }, { "epoch": 0.66, "learning_rate": 8.833896949807593e-06, "loss": 6.0229, "step": 80000 }, { "epoch": 0.66, "eval_accuracy": 0.1511706409765356, "eval_loss": 6.008976459503174, "eval_runtime": 2027.8147, "eval_samples_per_second": 152.063, "eval_steps_per_second": 3.168, "step": 80000 }, { "epoch": 0.66, "step": 80001, "total_flos": 1.0107191126130688e+18, "train_loss": 6.2887065917574985, "train_runtime": 86523.957, "train_samples_per_second": 338.563, "train_steps_per_second": 7.053 } ], "max_steps": 610290, "num_train_epochs": 5, "total_flos": 1.0107191126130688e+18, "trial_name": null, "trial_params": null }