roberta-base-qnli-sft / trainer_state.json
EvaristeL's picture
Upload folder using huggingface_hub
8d5fcd4 verified
{
"best_metric": 0.9143675417661098,
"best_model_checkpoint": "./save_models/qnli/roberta-base_lr1e-05_run0/checkpoint-17676",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 58920,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.08,
"learning_rate": 1.4140271493212672e-06,
"loss": 0.6933,
"step": 500
},
{
"epoch": 0.17,
"learning_rate": 2.8280542986425343e-06,
"loss": 0.542,
"step": 1000
},
{
"epoch": 0.25,
"learning_rate": 4.242081447963801e-06,
"loss": 0.4328,
"step": 1500
},
{
"epoch": 0.34,
"learning_rate": 5.656108597285069e-06,
"loss": 0.3932,
"step": 2000
},
{
"epoch": 0.42,
"learning_rate": 7.070135746606336e-06,
"loss": 0.3613,
"step": 2500
},
{
"epoch": 0.51,
"learning_rate": 8.484162895927603e-06,
"loss": 0.3553,
"step": 3000
},
{
"epoch": 0.59,
"learning_rate": 9.89819004524887e-06,
"loss": 0.3362,
"step": 3500
},
{
"epoch": 0.68,
"learning_rate": 9.916221291347683e-06,
"loss": 0.3362,
"step": 4000
},
{
"epoch": 0.76,
"learning_rate": 9.825942510472339e-06,
"loss": 0.3298,
"step": 4500
},
{
"epoch": 0.85,
"learning_rate": 9.735663729596995e-06,
"loss": 0.3075,
"step": 5000
},
{
"epoch": 0.93,
"learning_rate": 9.645384948721653e-06,
"loss": 0.3071,
"step": 5500
},
{
"epoch": 1.0,
"eval_accuracy": 0.8982338902147972,
"eval_loss": 0.273178368806839,
"eval_runtime": 14.5434,
"eval_samples_per_second": 720.258,
"eval_steps_per_second": 45.038,
"step": 5892
},
{
"epoch": 1.02,
"learning_rate": 9.55510616784631e-06,
"loss": 0.2836,
"step": 6000
},
{
"epoch": 1.1,
"learning_rate": 9.464827386970968e-06,
"loss": 0.2656,
"step": 6500
},
{
"epoch": 1.19,
"learning_rate": 9.374548606095624e-06,
"loss": 0.2548,
"step": 7000
},
{
"epoch": 1.27,
"learning_rate": 9.28426982522028e-06,
"loss": 0.2535,
"step": 7500
},
{
"epoch": 1.36,
"learning_rate": 9.193991044344937e-06,
"loss": 0.258,
"step": 8000
},
{
"epoch": 1.44,
"learning_rate": 9.103712263469595e-06,
"loss": 0.2539,
"step": 8500
},
{
"epoch": 1.53,
"learning_rate": 9.013433482594251e-06,
"loss": 0.2465,
"step": 9000
},
{
"epoch": 1.61,
"learning_rate": 8.92315470171891e-06,
"loss": 0.2586,
"step": 9500
},
{
"epoch": 1.7,
"learning_rate": 8.832875920843566e-06,
"loss": 0.2407,
"step": 10000
},
{
"epoch": 1.78,
"learning_rate": 8.742597139968224e-06,
"loss": 0.2453,
"step": 10500
},
{
"epoch": 1.87,
"learning_rate": 8.65231835909288e-06,
"loss": 0.2506,
"step": 11000
},
{
"epoch": 1.95,
"learning_rate": 8.562039578217536e-06,
"loss": 0.2426,
"step": 11500
},
{
"epoch": 2.0,
"eval_accuracy": 0.911217183770883,
"eval_loss": 0.24326153099536896,
"eval_runtime": 14.4104,
"eval_samples_per_second": 726.905,
"eval_steps_per_second": 45.453,
"step": 11784
},
{
"epoch": 2.04,
"learning_rate": 8.471760797342193e-06,
"loss": 0.2253,
"step": 12000
},
{
"epoch": 2.12,
"learning_rate": 8.381482016466849e-06,
"loss": 0.2004,
"step": 12500
},
{
"epoch": 2.21,
"learning_rate": 8.291203235591507e-06,
"loss": 0.2032,
"step": 13000
},
{
"epoch": 2.29,
"learning_rate": 8.200924454716163e-06,
"loss": 0.2098,
"step": 13500
},
{
"epoch": 2.38,
"learning_rate": 8.110645673840822e-06,
"loss": 0.1996,
"step": 14000
},
{
"epoch": 2.46,
"learning_rate": 8.020366892965478e-06,
"loss": 0.2065,
"step": 14500
},
{
"epoch": 2.55,
"learning_rate": 7.930088112090136e-06,
"loss": 0.2011,
"step": 15000
},
{
"epoch": 2.63,
"learning_rate": 7.839809331214792e-06,
"loss": 0.2135,
"step": 15500
},
{
"epoch": 2.72,
"learning_rate": 7.749530550339449e-06,
"loss": 0.2116,
"step": 16000
},
{
"epoch": 2.8,
"learning_rate": 7.659251769464105e-06,
"loss": 0.221,
"step": 16500
},
{
"epoch": 2.89,
"learning_rate": 7.568972988588763e-06,
"loss": 0.2179,
"step": 17000
},
{
"epoch": 2.97,
"learning_rate": 7.478694207713419e-06,
"loss": 0.2141,
"step": 17500
},
{
"epoch": 3.0,
"eval_accuracy": 0.9143675417661098,
"eval_loss": 0.27570974826812744,
"eval_runtime": 14.419,
"eval_samples_per_second": 726.472,
"eval_steps_per_second": 45.426,
"step": 17676
},
{
"epoch": 3.05,
"learning_rate": 7.388415426838077e-06,
"loss": 0.1871,
"step": 18000
},
{
"epoch": 3.14,
"learning_rate": 7.298136645962734e-06,
"loss": 0.183,
"step": 18500
},
{
"epoch": 3.22,
"learning_rate": 7.207857865087391e-06,
"loss": 0.1757,
"step": 19000
},
{
"epoch": 3.31,
"learning_rate": 7.117579084212047e-06,
"loss": 0.1783,
"step": 19500
},
{
"epoch": 3.39,
"learning_rate": 7.027300303336704e-06,
"loss": 0.1787,
"step": 20000
},
{
"epoch": 3.48,
"learning_rate": 6.937021522461362e-06,
"loss": 0.1834,
"step": 20500
},
{
"epoch": 3.56,
"learning_rate": 6.846742741586018e-06,
"loss": 0.1722,
"step": 21000
},
{
"epoch": 3.65,
"learning_rate": 6.756463960710675e-06,
"loss": 0.1894,
"step": 21500
},
{
"epoch": 3.73,
"learning_rate": 6.6661851798353315e-06,
"loss": 0.1708,
"step": 22000
},
{
"epoch": 3.82,
"learning_rate": 6.57590639895999e-06,
"loss": 0.1962,
"step": 22500
},
{
"epoch": 3.9,
"learning_rate": 6.485627618084646e-06,
"loss": 0.1831,
"step": 23000
},
{
"epoch": 3.99,
"learning_rate": 6.395348837209303e-06,
"loss": 0.1893,
"step": 23500
},
{
"epoch": 4.0,
"eval_accuracy": 0.9129355608591886,
"eval_loss": 0.39158299565315247,
"eval_runtime": 14.3752,
"eval_samples_per_second": 728.688,
"eval_steps_per_second": 45.565,
"step": 23568
},
{
"epoch": 4.07,
"learning_rate": 6.3050700563339595e-06,
"loss": 0.1408,
"step": 24000
},
{
"epoch": 4.16,
"learning_rate": 6.214791275458617e-06,
"loss": 0.1505,
"step": 24500
},
{
"epoch": 4.24,
"learning_rate": 6.124512494583274e-06,
"loss": 0.1364,
"step": 25000
},
{
"epoch": 4.33,
"learning_rate": 6.034233713707931e-06,
"loss": 0.1519,
"step": 25500
},
{
"epoch": 4.41,
"learning_rate": 5.943954932832587e-06,
"loss": 0.1367,
"step": 26000
},
{
"epoch": 4.5,
"learning_rate": 5.853676151957245e-06,
"loss": 0.154,
"step": 26500
},
{
"epoch": 4.58,
"learning_rate": 5.763397371081901e-06,
"loss": 0.1437,
"step": 27000
},
{
"epoch": 4.67,
"learning_rate": 5.673118590206559e-06,
"loss": 0.1496,
"step": 27500
},
{
"epoch": 4.75,
"learning_rate": 5.582839809331215e-06,
"loss": 0.1668,
"step": 28000
},
{
"epoch": 4.84,
"learning_rate": 5.492561028455872e-06,
"loss": 0.1448,
"step": 28500
},
{
"epoch": 4.92,
"learning_rate": 5.402282247580529e-06,
"loss": 0.1638,
"step": 29000
},
{
"epoch": 5.0,
"eval_accuracy": 0.9121718377088306,
"eval_loss": 0.4099659323692322,
"eval_runtime": 14.3864,
"eval_samples_per_second": 728.119,
"eval_steps_per_second": 45.529,
"step": 29460
},
{
"epoch": 5.01,
"learning_rate": 5.312003466705185e-06,
"loss": 0.1375,
"step": 29500
},
{
"epoch": 5.09,
"learning_rate": 5.221724685829843e-06,
"loss": 0.1032,
"step": 30000
},
{
"epoch": 5.18,
"learning_rate": 5.1314459049545e-06,
"loss": 0.1106,
"step": 30500
},
{
"epoch": 5.26,
"learning_rate": 5.041167124079157e-06,
"loss": 0.1091,
"step": 31000
},
{
"epoch": 5.35,
"learning_rate": 4.950888343203813e-06,
"loss": 0.1094,
"step": 31500
},
{
"epoch": 5.43,
"learning_rate": 4.86060956232847e-06,
"loss": 0.1223,
"step": 32000
},
{
"epoch": 5.52,
"learning_rate": 4.7703307814531275e-06,
"loss": 0.1084,
"step": 32500
},
{
"epoch": 5.6,
"learning_rate": 4.680052000577785e-06,
"loss": 0.1097,
"step": 33000
},
{
"epoch": 5.69,
"learning_rate": 4.589773219702441e-06,
"loss": 0.1216,
"step": 33500
},
{
"epoch": 5.77,
"learning_rate": 4.499494438827098e-06,
"loss": 0.1191,
"step": 34000
},
{
"epoch": 5.86,
"learning_rate": 4.4092156579517554e-06,
"loss": 0.1215,
"step": 34500
},
{
"epoch": 5.94,
"learning_rate": 4.318936877076413e-06,
"loss": 0.1356,
"step": 35000
},
{
"epoch": 6.0,
"eval_accuracy": 0.9077804295942721,
"eval_loss": 0.484077125787735,
"eval_runtime": 14.4023,
"eval_samples_per_second": 727.317,
"eval_steps_per_second": 45.479,
"step": 35352
},
{
"epoch": 6.03,
"learning_rate": 4.228658096201069e-06,
"loss": 0.1124,
"step": 35500
},
{
"epoch": 6.11,
"learning_rate": 4.138379315325726e-06,
"loss": 0.0893,
"step": 36000
},
{
"epoch": 6.19,
"learning_rate": 4.048100534450383e-06,
"loss": 0.0763,
"step": 36500
},
{
"epoch": 6.28,
"learning_rate": 3.9578217535750406e-06,
"loss": 0.0889,
"step": 37000
},
{
"epoch": 6.36,
"learning_rate": 3.867542972699697e-06,
"loss": 0.0941,
"step": 37500
},
{
"epoch": 6.45,
"learning_rate": 3.777264191824354e-06,
"loss": 0.0931,
"step": 38000
},
{
"epoch": 6.53,
"learning_rate": 3.686985410949011e-06,
"loss": 0.0916,
"step": 38500
},
{
"epoch": 6.62,
"learning_rate": 3.596706630073668e-06,
"loss": 0.0916,
"step": 39000
},
{
"epoch": 6.7,
"learning_rate": 3.5064278491983244e-06,
"loss": 0.1038,
"step": 39500
},
{
"epoch": 6.79,
"learning_rate": 3.4161490683229816e-06,
"loss": 0.089,
"step": 40000
},
{
"epoch": 6.87,
"learning_rate": 3.3258702874476384e-06,
"loss": 0.1008,
"step": 40500
},
{
"epoch": 6.96,
"learning_rate": 3.2355915065722956e-06,
"loss": 0.0867,
"step": 41000
},
{
"epoch": 7.0,
"eval_accuracy": 0.9097852028639618,
"eval_loss": 0.5044658184051514,
"eval_runtime": 14.3742,
"eval_samples_per_second": 728.736,
"eval_steps_per_second": 45.568,
"step": 41244
},
{
"epoch": 7.04,
"learning_rate": 3.1453127256969523e-06,
"loss": 0.0746,
"step": 41500
},
{
"epoch": 7.13,
"learning_rate": 3.055033944821609e-06,
"loss": 0.0624,
"step": 42000
},
{
"epoch": 7.21,
"learning_rate": 2.9647551639462663e-06,
"loss": 0.0695,
"step": 42500
},
{
"epoch": 7.3,
"learning_rate": 2.874476383070923e-06,
"loss": 0.0804,
"step": 43000
},
{
"epoch": 7.38,
"learning_rate": 2.7841976021955803e-06,
"loss": 0.0695,
"step": 43500
},
{
"epoch": 7.47,
"learning_rate": 2.693918821320237e-06,
"loss": 0.0737,
"step": 44000
},
{
"epoch": 7.55,
"learning_rate": 2.6036400404448942e-06,
"loss": 0.0773,
"step": 44500
},
{
"epoch": 7.64,
"learning_rate": 2.513361259569551e-06,
"loss": 0.0756,
"step": 45000
},
{
"epoch": 7.72,
"learning_rate": 2.4230824786942078e-06,
"loss": 0.0697,
"step": 45500
},
{
"epoch": 7.81,
"learning_rate": 2.332803697818865e-06,
"loss": 0.0731,
"step": 46000
},
{
"epoch": 7.89,
"learning_rate": 2.2425249169435217e-06,
"loss": 0.0584,
"step": 46500
},
{
"epoch": 7.98,
"learning_rate": 2.152246136068179e-06,
"loss": 0.071,
"step": 47000
},
{
"epoch": 8.0,
"eval_accuracy": 0.9092124105011933,
"eval_loss": 0.5765216946601868,
"eval_runtime": 14.3597,
"eval_samples_per_second": 729.473,
"eval_steps_per_second": 45.614,
"step": 47136
},
{
"epoch": 8.06,
"learning_rate": 2.0619673551928357e-06,
"loss": 0.0614,
"step": 47500
},
{
"epoch": 8.15,
"learning_rate": 1.9716885743174925e-06,
"loss": 0.0588,
"step": 48000
},
{
"epoch": 8.23,
"learning_rate": 1.8814097934421497e-06,
"loss": 0.0575,
"step": 48500
},
{
"epoch": 8.32,
"learning_rate": 1.7911310125668062e-06,
"loss": 0.0454,
"step": 49000
},
{
"epoch": 8.4,
"learning_rate": 1.7008522316914632e-06,
"loss": 0.0554,
"step": 49500
},
{
"epoch": 8.49,
"learning_rate": 1.6105734508161202e-06,
"loss": 0.0559,
"step": 50000
},
{
"epoch": 8.57,
"learning_rate": 1.5202946699407772e-06,
"loss": 0.0561,
"step": 50500
},
{
"epoch": 8.66,
"learning_rate": 1.4300158890654341e-06,
"loss": 0.0505,
"step": 51000
},
{
"epoch": 8.74,
"learning_rate": 1.3397371081900911e-06,
"loss": 0.0552,
"step": 51500
},
{
"epoch": 8.83,
"learning_rate": 1.249458327314748e-06,
"loss": 0.0604,
"step": 52000
},
{
"epoch": 8.91,
"learning_rate": 1.1591795464394049e-06,
"loss": 0.0626,
"step": 52500
},
{
"epoch": 9.0,
"learning_rate": 1.0689007655640619e-06,
"loss": 0.0636,
"step": 53000
},
{
"epoch": 9.0,
"eval_accuracy": 0.9113126491646778,
"eval_loss": 0.5765402913093567,
"eval_runtime": 14.3825,
"eval_samples_per_second": 728.315,
"eval_steps_per_second": 45.541,
"step": 53028
},
{
"epoch": 9.08,
"learning_rate": 9.786219846887188e-07,
"loss": 0.0439,
"step": 53500
},
{
"epoch": 9.16,
"learning_rate": 8.883432038133757e-07,
"loss": 0.0447,
"step": 54000
},
{
"epoch": 9.25,
"learning_rate": 7.980644229380327e-07,
"loss": 0.0391,
"step": 54500
},
{
"epoch": 9.33,
"learning_rate": 7.077856420626897e-07,
"loss": 0.0404,
"step": 55000
},
{
"epoch": 9.42,
"learning_rate": 6.175068611873466e-07,
"loss": 0.0535,
"step": 55500
},
{
"epoch": 9.5,
"learning_rate": 5.272280803120034e-07,
"loss": 0.038,
"step": 56000
},
{
"epoch": 9.59,
"learning_rate": 4.3694929943666047e-07,
"loss": 0.047,
"step": 56500
},
{
"epoch": 9.67,
"learning_rate": 3.4667051856131735e-07,
"loss": 0.0414,
"step": 57000
},
{
"epoch": 9.76,
"learning_rate": 2.5639173768597433e-07,
"loss": 0.0447,
"step": 57500
},
{
"epoch": 9.84,
"learning_rate": 1.6611295681063126e-07,
"loss": 0.0364,
"step": 58000
},
{
"epoch": 9.93,
"learning_rate": 7.583417593528817e-08,
"loss": 0.0492,
"step": 58500
},
{
"epoch": 10.0,
"eval_accuracy": 0.9117899761336515,
"eval_loss": 0.6481573581695557,
"eval_runtime": 14.4339,
"eval_samples_per_second": 725.723,
"eval_steps_per_second": 45.379,
"step": 58920
},
{
"epoch": 10.0,
"step": 58920,
"total_flos": 4.325458183470288e+16,
"train_loss": 0.15599560614596883,
"train_runtime": 4172.8624,
"train_samples_per_second": 225.907,
"train_steps_per_second": 14.12
}
],
"logging_steps": 500,
"max_steps": 58920,
"num_train_epochs": 10,
"save_steps": 500,
"total_flos": 4.325458183470288e+16,
"trial_name": null,
"trial_params": null
}