bert_base_120 / trainer_state.json
gokuls's picture
End of training
156cf90
{
"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
}