bert-base-uncased-mnli / trainer_state.json
ZhangYunchenY
[Model] bert-base-uncased-mnli
f8102ed
{
"best_metric": 0.41607236862182617,
"best_model_checkpoint": "output_dir/finetuned_best_mnli/checkpoint-12000",
"epoch": 1.9149282920469362,
"global_step": 23500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"learning_rate": 1.983702737940026e-05,
"loss": 0.8277,
"step": 500
},
{
"epoch": 0.04,
"eval_accuracy": 0.7111563932755985,
"eval_loss": 0.6876205801963806,
"eval_runtime": 133.5092,
"eval_samples_per_second": 73.516,
"eval_steps_per_second": 9.19,
"step": 500
},
{
"epoch": 0.08,
"learning_rate": 1.9674054758800523e-05,
"loss": 0.6598,
"step": 1000
},
{
"epoch": 0.08,
"eval_accuracy": 0.7539480387162506,
"eval_loss": 0.6054476499557495,
"eval_runtime": 151.2877,
"eval_samples_per_second": 64.876,
"eval_steps_per_second": 8.11,
"step": 1000
},
{
"epoch": 0.12,
"learning_rate": 1.9511082138200782e-05,
"loss": 0.6012,
"step": 1500
},
{
"epoch": 0.12,
"eval_accuracy": 0.7814569536423841,
"eval_loss": 0.5584900975227356,
"eval_runtime": 160.2112,
"eval_samples_per_second": 61.263,
"eval_steps_per_second": 7.659,
"step": 1500
},
{
"epoch": 0.16,
"learning_rate": 1.9348109517601044e-05,
"loss": 0.5589,
"step": 2000
},
{
"epoch": 0.16,
"eval_accuracy": 0.7909322465613856,
"eval_loss": 0.5319538116455078,
"eval_runtime": 160.4115,
"eval_samples_per_second": 61.186,
"eval_steps_per_second": 7.649,
"step": 2000
},
{
"epoch": 0.2,
"learning_rate": 1.9185136897001307e-05,
"loss": 0.5631,
"step": 2500
},
{
"epoch": 0.2,
"eval_accuracy": 0.7811512990320937,
"eval_loss": 0.5446802973747253,
"eval_runtime": 190.5286,
"eval_samples_per_second": 51.515,
"eval_steps_per_second": 6.44,
"step": 2500
},
{
"epoch": 0.24,
"learning_rate": 1.9022164276401566e-05,
"loss": 0.5556,
"step": 3000
},
{
"epoch": 0.24,
"eval_accuracy": 0.795109526235354,
"eval_loss": 0.5091724991798401,
"eval_runtime": 100.2906,
"eval_samples_per_second": 97.866,
"eval_steps_per_second": 12.234,
"step": 3000
},
{
"epoch": 0.29,
"learning_rate": 1.8859191655801828e-05,
"loss": 0.5311,
"step": 3500
},
{
"epoch": 0.29,
"eval_accuracy": 0.8002037697401936,
"eval_loss": 0.5131673216819763,
"eval_runtime": 106.1752,
"eval_samples_per_second": 92.442,
"eval_steps_per_second": 11.556,
"step": 3500
},
{
"epoch": 0.33,
"learning_rate": 1.8696219035202087e-05,
"loss": 0.5119,
"step": 4000
},
{
"epoch": 0.33,
"eval_accuracy": 0.8132450331125828,
"eval_loss": 0.47580984234809875,
"eval_runtime": 84.6709,
"eval_samples_per_second": 115.919,
"eval_steps_per_second": 14.491,
"step": 4000
},
{
"epoch": 0.37,
"learning_rate": 1.853324641460235e-05,
"loss": 0.4988,
"step": 4500
},
{
"epoch": 0.37,
"eval_accuracy": 0.8033622007131941,
"eval_loss": 0.5015984177589417,
"eval_runtime": 120.1676,
"eval_samples_per_second": 81.678,
"eval_steps_per_second": 10.211,
"step": 4500
},
{
"epoch": 0.41,
"learning_rate": 1.837027379400261e-05,
"loss": 0.509,
"step": 5000
},
{
"epoch": 0.41,
"eval_accuracy": 0.8169128884360672,
"eval_loss": 0.4686170220375061,
"eval_runtime": 115.84,
"eval_samples_per_second": 84.729,
"eval_steps_per_second": 10.592,
"step": 5000
},
{
"epoch": 0.45,
"learning_rate": 1.820730117340287e-05,
"loss": 0.5033,
"step": 5500
},
{
"epoch": 0.45,
"eval_accuracy": 0.8121242995415181,
"eval_loss": 0.4700980484485626,
"eval_runtime": 103.4737,
"eval_samples_per_second": 94.855,
"eval_steps_per_second": 11.858,
"step": 5500
},
{
"epoch": 0.49,
"learning_rate": 1.804432855280313e-05,
"loss": 0.4834,
"step": 6000
},
{
"epoch": 0.49,
"eval_accuracy": 0.8227203260315843,
"eval_loss": 0.4637387990951538,
"eval_runtime": 110.8909,
"eval_samples_per_second": 88.51,
"eval_steps_per_second": 11.065,
"step": 6000
},
{
"epoch": 0.53,
"learning_rate": 1.788135593220339e-05,
"loss": 0.4863,
"step": 6500
},
{
"epoch": 0.53,
"eval_accuracy": 0.8237391747325522,
"eval_loss": 0.45791763067245483,
"eval_runtime": 95.3317,
"eval_samples_per_second": 102.956,
"eval_steps_per_second": 12.871,
"step": 6500
},
{
"epoch": 0.57,
"learning_rate": 1.771838331160365e-05,
"loss": 0.4829,
"step": 7000
},
{
"epoch": 0.57,
"eval_accuracy": 0.823331635252165,
"eval_loss": 0.463431179523468,
"eval_runtime": 94.439,
"eval_samples_per_second": 103.93,
"eval_steps_per_second": 12.993,
"step": 7000
},
{
"epoch": 0.61,
"learning_rate": 1.7555410691003914e-05,
"loss": 0.4819,
"step": 7500
},
{
"epoch": 0.61,
"eval_accuracy": 0.8260825267447784,
"eval_loss": 0.4441950023174286,
"eval_runtime": 106.424,
"eval_samples_per_second": 92.225,
"eval_steps_per_second": 11.529,
"step": 7500
},
{
"epoch": 0.65,
"learning_rate": 1.7392438070404173e-05,
"loss": 0.4763,
"step": 8000
},
{
"epoch": 0.65,
"eval_accuracy": 0.8212939378502292,
"eval_loss": 0.44791945815086365,
"eval_runtime": 88.0696,
"eval_samples_per_second": 111.446,
"eval_steps_per_second": 13.932,
"step": 8000
},
{
"epoch": 0.69,
"learning_rate": 1.7229465449804435e-05,
"loss": 0.4738,
"step": 8500
},
{
"epoch": 0.69,
"eval_accuracy": 0.8275089149261334,
"eval_loss": 0.45088809728622437,
"eval_runtime": 92.5901,
"eval_samples_per_second": 106.005,
"eval_steps_per_second": 13.252,
"step": 8500
},
{
"epoch": 0.73,
"learning_rate": 1.7066492829204694e-05,
"loss": 0.4687,
"step": 9000
},
{
"epoch": 0.73,
"eval_accuracy": 0.8311767702496179,
"eval_loss": 0.43536150455474854,
"eval_runtime": 126.1698,
"eval_samples_per_second": 77.792,
"eval_steps_per_second": 9.725,
"step": 9000
},
{
"epoch": 0.77,
"learning_rate": 1.6903520208604957e-05,
"loss": 0.468,
"step": 9500
},
{
"epoch": 0.77,
"eval_accuracy": 0.8322975038206827,
"eval_loss": 0.43157872557640076,
"eval_runtime": 113.9939,
"eval_samples_per_second": 86.101,
"eval_steps_per_second": 10.764,
"step": 9500
},
{
"epoch": 0.81,
"learning_rate": 1.6740547588005215e-05,
"loss": 0.4529,
"step": 10000
},
{
"epoch": 0.81,
"eval_accuracy": 0.8317880794701987,
"eval_loss": 0.42749524116516113,
"eval_runtime": 115.1753,
"eval_samples_per_second": 85.218,
"eval_steps_per_second": 10.653,
"step": 10000
},
{
"epoch": 0.86,
"learning_rate": 1.6577574967405478e-05,
"loss": 0.4617,
"step": 10500
},
{
"epoch": 0.86,
"eval_accuracy": 0.8309730005094244,
"eval_loss": 0.43376246094703674,
"eval_runtime": 139.6786,
"eval_samples_per_second": 70.268,
"eval_steps_per_second": 8.784,
"step": 10500
},
{
"epoch": 0.9,
"learning_rate": 1.6414602346805737e-05,
"loss": 0.4534,
"step": 11000
},
{
"epoch": 0.9,
"eval_accuracy": 0.8359653591441671,
"eval_loss": 0.42552220821380615,
"eval_runtime": 96.9513,
"eval_samples_per_second": 101.236,
"eval_steps_per_second": 12.656,
"step": 11000
},
{
"epoch": 0.94,
"learning_rate": 1.6251629726206e-05,
"loss": 0.4536,
"step": 11500
},
{
"epoch": 0.94,
"eval_accuracy": 0.8359653591441671,
"eval_loss": 0.41881263256073,
"eval_runtime": 108.0698,
"eval_samples_per_second": 90.821,
"eval_steps_per_second": 11.354,
"step": 11500
},
{
"epoch": 0.98,
"learning_rate": 1.608865710560626e-05,
"loss": 0.4551,
"step": 12000
},
{
"epoch": 0.98,
"eval_accuracy": 0.8391237901171676,
"eval_loss": 0.41607236862182617,
"eval_runtime": 89.8988,
"eval_samples_per_second": 109.178,
"eval_steps_per_second": 13.649,
"step": 12000
},
{
"epoch": 1.02,
"learning_rate": 1.592568448500652e-05,
"loss": 0.4006,
"step": 12500
},
{
"epoch": 1.02,
"eval_accuracy": 0.8356597045338767,
"eval_loss": 0.4454086124897003,
"eval_runtime": 91.709,
"eval_samples_per_second": 107.023,
"eval_steps_per_second": 13.379,
"step": 12500
},
{
"epoch": 1.06,
"learning_rate": 1.576271186440678e-05,
"loss": 0.3355,
"step": 13000
},
{
"epoch": 1.06,
"eval_accuracy": 0.840855832908813,
"eval_loss": 0.44230031967163086,
"eval_runtime": 93.3849,
"eval_samples_per_second": 105.103,
"eval_steps_per_second": 13.139,
"step": 13000
},
{
"epoch": 1.1,
"learning_rate": 1.5599739243807042e-05,
"loss": 0.3309,
"step": 13500
},
{
"epoch": 1.1,
"eval_accuracy": 0.8339276617422313,
"eval_loss": 0.4691074788570404,
"eval_runtime": 81.8679,
"eval_samples_per_second": 119.888,
"eval_steps_per_second": 14.988,
"step": 13500
},
{
"epoch": 1.14,
"learning_rate": 1.54367666232073e-05,
"loss": 0.3416,
"step": 14000
},
{
"epoch": 1.14,
"eval_accuracy": 0.8397350993377484,
"eval_loss": 0.4406871497631073,
"eval_runtime": 80.3421,
"eval_samples_per_second": 122.165,
"eval_steps_per_second": 15.272,
"step": 14000
},
{
"epoch": 1.18,
"learning_rate": 1.5273794002607563e-05,
"loss": 0.3463,
"step": 14500
},
{
"epoch": 1.18,
"eval_accuracy": 0.8371879775853286,
"eval_loss": 0.445261150598526,
"eval_runtime": 83.1773,
"eval_samples_per_second": 118.001,
"eval_steps_per_second": 14.752,
"step": 14500
},
{
"epoch": 1.22,
"learning_rate": 1.5110821382007822e-05,
"loss": 0.3447,
"step": 15000
},
{
"epoch": 1.22,
"eval_accuracy": 0.8397350993377484,
"eval_loss": 0.4480521082878113,
"eval_runtime": 93.0739,
"eval_samples_per_second": 105.454,
"eval_steps_per_second": 13.183,
"step": 15000
},
{
"epoch": 1.26,
"learning_rate": 1.4947848761408083e-05,
"loss": 0.3302,
"step": 15500
},
{
"epoch": 1.26,
"eval_accuracy": 0.8403464085583291,
"eval_loss": 0.462993860244751,
"eval_runtime": 98.4254,
"eval_samples_per_second": 99.72,
"eval_steps_per_second": 12.466,
"step": 15500
},
{
"epoch": 1.3,
"learning_rate": 1.4784876140808346e-05,
"loss": 0.3366,
"step": 16000
},
{
"epoch": 1.3,
"eval_accuracy": 0.8402445236882323,
"eval_loss": 0.4278508424758911,
"eval_runtime": 80.9392,
"eval_samples_per_second": 121.264,
"eval_steps_per_second": 15.16,
"step": 16000
},
{
"epoch": 1.34,
"learning_rate": 1.4621903520208606e-05,
"loss": 0.3446,
"step": 16500
},
{
"epoch": 1.34,
"eval_accuracy": 0.8444218033622007,
"eval_loss": 0.4328218996524811,
"eval_runtime": 88.6359,
"eval_samples_per_second": 110.734,
"eval_steps_per_second": 13.843,
"step": 16500
},
{
"epoch": 1.39,
"learning_rate": 1.4458930899608867e-05,
"loss": 0.3442,
"step": 17000
},
{
"epoch": 1.39,
"eval_accuracy": 0.8421803362200713,
"eval_loss": 0.4372538924217224,
"eval_runtime": 90.5486,
"eval_samples_per_second": 108.395,
"eval_steps_per_second": 13.551,
"step": 17000
},
{
"epoch": 1.43,
"learning_rate": 1.4295958279009128e-05,
"loss": 0.3431,
"step": 17500
},
{
"epoch": 1.43,
"eval_accuracy": 0.8425878757004585,
"eval_loss": 0.444231778383255,
"eval_runtime": 104.7466,
"eval_samples_per_second": 93.702,
"eval_steps_per_second": 11.714,
"step": 17500
},
{
"epoch": 1.47,
"learning_rate": 1.4132985658409388e-05,
"loss": 0.3388,
"step": 18000
},
{
"epoch": 1.47,
"eval_accuracy": 0.8358634742740703,
"eval_loss": 0.4574286937713623,
"eval_runtime": 79.912,
"eval_samples_per_second": 122.823,
"eval_steps_per_second": 15.354,
"step": 18000
},
{
"epoch": 1.51,
"learning_rate": 1.3970013037809649e-05,
"loss": 0.3403,
"step": 18500
},
{
"epoch": 1.51,
"eval_accuracy": 0.8423841059602649,
"eval_loss": 0.4366081953048706,
"eval_runtime": 80.6484,
"eval_samples_per_second": 121.701,
"eval_steps_per_second": 15.214,
"step": 18500
},
{
"epoch": 1.55,
"learning_rate": 1.380704041720991e-05,
"loss": 0.3421,
"step": 19000
},
{
"epoch": 1.55,
"eval_accuracy": 0.8441161487519103,
"eval_loss": 0.4383392930030823,
"eval_runtime": 105.6168,
"eval_samples_per_second": 92.93,
"eval_steps_per_second": 11.617,
"step": 19000
},
{
"epoch": 1.59,
"learning_rate": 1.364406779661017e-05,
"loss": 0.3405,
"step": 19500
},
{
"epoch": 1.59,
"eval_accuracy": 0.8411614875191035,
"eval_loss": 0.4455229938030243,
"eval_runtime": 99.6387,
"eval_samples_per_second": 98.506,
"eval_steps_per_second": 12.314,
"step": 19500
},
{
"epoch": 1.63,
"learning_rate": 1.3481095176010431e-05,
"loss": 0.3448,
"step": 20000
},
{
"epoch": 1.63,
"eval_accuracy": 0.8385124808965868,
"eval_loss": 0.4386064112186432,
"eval_runtime": 86.9871,
"eval_samples_per_second": 112.833,
"eval_steps_per_second": 14.106,
"step": 20000
},
{
"epoch": 1.67,
"learning_rate": 1.3318122555410693e-05,
"loss": 0.3346,
"step": 20500
},
{
"epoch": 1.67,
"eval_accuracy": 0.8416709118695873,
"eval_loss": 0.4354933500289917,
"eval_runtime": 90.9919,
"eval_samples_per_second": 107.867,
"eval_steps_per_second": 13.485,
"step": 20500
},
{
"epoch": 1.71,
"learning_rate": 1.3155149934810954e-05,
"loss": 0.3475,
"step": 21000
},
{
"epoch": 1.71,
"eval_accuracy": 0.8372898624554254,
"eval_loss": 0.4339876174926758,
"eval_runtime": 89.1677,
"eval_samples_per_second": 110.073,
"eval_steps_per_second": 13.761,
"step": 21000
},
{
"epoch": 1.75,
"learning_rate": 1.2992177314211213e-05,
"loss": 0.3378,
"step": 21500
},
{
"epoch": 1.75,
"eval_accuracy": 0.8469689251146205,
"eval_loss": 0.4287373125553131,
"eval_runtime": 93.5816,
"eval_samples_per_second": 104.882,
"eval_steps_per_second": 13.112,
"step": 21500
},
{
"epoch": 1.79,
"learning_rate": 1.2829204693611474e-05,
"loss": 0.3454,
"step": 22000
},
{
"epoch": 1.79,
"eval_accuracy": 0.8434029546612328,
"eval_loss": 0.43279626965522766,
"eval_runtime": 108.513,
"eval_samples_per_second": 90.45,
"eval_steps_per_second": 11.307,
"step": 22000
},
{
"epoch": 1.83,
"learning_rate": 1.2666232073011735e-05,
"loss": 0.3423,
"step": 22500
},
{
"epoch": 1.83,
"eval_accuracy": 0.8433010697911361,
"eval_loss": 0.4425649344921112,
"eval_runtime": 91.9329,
"eval_samples_per_second": 106.763,
"eval_steps_per_second": 13.347,
"step": 22500
},
{
"epoch": 1.87,
"learning_rate": 1.2503259452411995e-05,
"loss": 0.3292,
"step": 23000
},
{
"epoch": 1.87,
"eval_accuracy": 0.845236882322975,
"eval_loss": 0.424352765083313,
"eval_runtime": 101.4887,
"eval_samples_per_second": 96.71,
"eval_steps_per_second": 12.09,
"step": 23000
},
{
"epoch": 1.91,
"learning_rate": 1.2340286831812256e-05,
"loss": 0.3363,
"step": 23500
},
{
"epoch": 1.91,
"eval_accuracy": 0.8494141619969434,
"eval_loss": 0.4183717668056488,
"eval_runtime": 104.207,
"eval_samples_per_second": 94.188,
"eval_steps_per_second": 11.775,
"step": 23500
}
],
"max_steps": 61360,
"num_train_epochs": 5,
"total_flos": 4.946519097615514e+16,
"trial_name": null,
"trial_params": null
}