convnext-base-wd1e-8-4e-5 / trainer_state.json
vuongnhathien's picture
End of training
5787190 verified
raw
history blame contribute delete
No virus
20.6 kB
{
"best_metric": 0.23118199408054352,
"best_model_checkpoint": "./convnext-base-wd1e-8-4e-5/checkpoint-7693",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 10990,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.09,
"grad_norm": 23.269968032836914,
"learning_rate": 3.999182900797692e-05,
"loss": 2.4371,
"step": 100
},
{
"epoch": 0.18,
"grad_norm": 30.385623931884766,
"learning_rate": 3.996732270841873e-05,
"loss": 1.2813,
"step": 200
},
{
"epoch": 0.27,
"grad_norm": 27.450511932373047,
"learning_rate": 3.992650112540325e-05,
"loss": 0.9817,
"step": 300
},
{
"epoch": 0.36,
"grad_norm": 23.956308364868164,
"learning_rate": 3.986939761421341e-05,
"loss": 0.8368,
"step": 400
},
{
"epoch": 0.45,
"grad_norm": 20.52164649963379,
"learning_rate": 3.9796058834082656e-05,
"loss": 0.77,
"step": 500
},
{
"epoch": 0.55,
"grad_norm": 12.296116828918457,
"learning_rate": 3.970654471006971e-05,
"loss": 0.661,
"step": 600
},
{
"epoch": 0.64,
"grad_norm": 17.72109031677246,
"learning_rate": 3.960092838409391e-05,
"loss": 0.7014,
"step": 700
},
{
"epoch": 0.73,
"grad_norm": 18.131221771240234,
"learning_rate": 3.947929615517096e-05,
"loss": 0.635,
"step": 800
},
{
"epoch": 0.82,
"grad_norm": 17.65522575378418,
"learning_rate": 3.93417474088981e-05,
"loss": 0.5691,
"step": 900
},
{
"epoch": 0.91,
"grad_norm": 12.841033935546875,
"learning_rate": 3.9188394536246174e-05,
"loss": 0.6271,
"step": 1000
},
{
"epoch": 1.0,
"eval_accuracy": 0.9053677932405566,
"eval_loss": 0.34362277388572693,
"eval_runtime": 106.9883,
"eval_samples_per_second": 23.507,
"eval_steps_per_second": 1.477,
"step": 1099
},
{
"epoch": 1.0,
"grad_norm": 26.4544677734375,
"learning_rate": 3.90193628417251e-05,
"loss": 0.6271,
"step": 1100
},
{
"epoch": 1.09,
"grad_norm": 13.971123695373535,
"learning_rate": 3.883479044099763e-05,
"loss": 0.5298,
"step": 1200
},
{
"epoch": 1.18,
"grad_norm": 16.64786720275879,
"learning_rate": 3.8634828148025175e-05,
"loss": 0.478,
"step": 1300
},
{
"epoch": 1.27,
"grad_norm": 22.18141746520996,
"learning_rate": 3.841963935183782e-05,
"loss": 0.4858,
"step": 1400
},
{
"epoch": 1.36,
"grad_norm": 20.354249954223633,
"learning_rate": 3.818939988302927e-05,
"loss": 0.4578,
"step": 1500
},
{
"epoch": 1.46,
"grad_norm": 18.752660751342773,
"learning_rate": 3.7944297870085823e-05,
"loss": 0.4527,
"step": 1600
},
{
"epoch": 1.55,
"grad_norm": 36.6127815246582,
"learning_rate": 3.768453358566675e-05,
"loss": 0.4937,
"step": 1700
},
{
"epoch": 1.64,
"grad_norm": 23.436214447021484,
"learning_rate": 3.741031928296162e-05,
"loss": 0.4916,
"step": 1800
},
{
"epoch": 1.73,
"grad_norm": 27.608442306518555,
"learning_rate": 3.712187902225845e-05,
"loss": 0.4758,
"step": 1900
},
{
"epoch": 1.82,
"grad_norm": 22.647695541381836,
"learning_rate": 3.681944848786418e-05,
"loss": 0.4312,
"step": 2000
},
{
"epoch": 1.91,
"grad_norm": 14.084342002868652,
"learning_rate": 3.6503274795527193e-05,
"loss": 0.4528,
"step": 2100
},
{
"epoch": 2.0,
"eval_accuracy": 0.9256461232604374,
"eval_loss": 0.27671584486961365,
"eval_runtime": 108.0363,
"eval_samples_per_second": 23.279,
"eval_steps_per_second": 1.462,
"step": 2198
},
{
"epoch": 2.0,
"grad_norm": 21.57204818725586,
"learning_rate": 3.617361629051932e-05,
"loss": 0.4516,
"step": 2200
},
{
"epoch": 2.09,
"grad_norm": 21.228384017944336,
"learning_rate": 3.5830742336542e-05,
"loss": 0.3546,
"step": 2300
},
{
"epoch": 2.18,
"grad_norm": 37.46253204345703,
"learning_rate": 3.547493309562956e-05,
"loss": 0.3906,
"step": 2400
},
{
"epoch": 2.27,
"grad_norm": 17.27825355529785,
"learning_rate": 3.51064792992289e-05,
"loss": 0.3866,
"step": 2500
},
{
"epoch": 2.37,
"grad_norm": 15.454079627990723,
"learning_rate": 3.472568201064314e-05,
"loss": 0.3813,
"step": 2600
},
{
"epoch": 2.46,
"grad_norm": 14.477165222167969,
"learning_rate": 3.433285237903305e-05,
"loss": 0.3714,
"step": 2700
},
{
"epoch": 2.55,
"grad_norm": 16.274826049804688,
"learning_rate": 3.392831138517724e-05,
"loss": 0.3561,
"step": 2800
},
{
"epoch": 2.64,
"grad_norm": 11.944202423095703,
"learning_rate": 3.3512389579199094e-05,
"loss": 0.3941,
"step": 2900
},
{
"epoch": 2.73,
"grad_norm": 10.119762420654297,
"learning_rate": 3.308542681047451e-05,
"loss": 0.3632,
"step": 3000
},
{
"epoch": 2.82,
"grad_norm": 14.887975692749023,
"learning_rate": 3.264777194994121e-05,
"loss": 0.38,
"step": 3100
},
{
"epoch": 2.91,
"grad_norm": 26.09409523010254,
"learning_rate": 3.219978260503665e-05,
"loss": 0.3762,
"step": 3200
},
{
"epoch": 3.0,
"eval_accuracy": 0.9268389662027833,
"eval_loss": 0.25190117955207825,
"eval_runtime": 107.1043,
"eval_samples_per_second": 23.482,
"eval_steps_per_second": 1.475,
"step": 3297
},
{
"epoch": 3.0,
"grad_norm": 18.360994338989258,
"learning_rate": 3.174182482749717e-05,
"loss": 0.3766,
"step": 3300
},
{
"epoch": 3.09,
"grad_norm": 20.89484977722168,
"learning_rate": 3.12742728142575e-05,
"loss": 0.3003,
"step": 3400
},
{
"epoch": 3.18,
"grad_norm": 11.302007675170898,
"learning_rate": 3.079750860169469e-05,
"loss": 0.3166,
"step": 3500
},
{
"epoch": 3.28,
"grad_norm": 11.336188316345215,
"learning_rate": 3.031192175346651e-05,
"loss": 0.3069,
"step": 3600
},
{
"epoch": 3.37,
"grad_norm": 30.606346130371094,
"learning_rate": 2.9817909042199313e-05,
"loss": 0.2799,
"step": 3700
},
{
"epoch": 3.46,
"grad_norm": 15.9932222366333,
"learning_rate": 2.93158741252854e-05,
"loss": 0.3054,
"step": 3800
},
{
"epoch": 3.55,
"grad_norm": 12.633625984191895,
"learning_rate": 2.8806227215054902e-05,
"loss": 0.3178,
"step": 3900
},
{
"epoch": 3.64,
"grad_norm": 33.243408203125,
"learning_rate": 2.828938474359164e-05,
"loss": 0.3152,
"step": 4000
},
{
"epoch": 3.73,
"grad_norm": 18.574337005615234,
"learning_rate": 2.7765769022466762e-05,
"loss": 0.3448,
"step": 4100
},
{
"epoch": 3.82,
"grad_norm": 4.619294166564941,
"learning_rate": 2.723580789766831e-05,
"loss": 0.3059,
"step": 4200
},
{
"epoch": 3.91,
"grad_norm": 26.067737579345703,
"learning_rate": 2.669993440000862e-05,
"loss": 0.2979,
"step": 4300
},
{
"epoch": 4.0,
"eval_accuracy": 0.9371769383697813,
"eval_loss": 0.2414184808731079,
"eval_runtime": 107.4842,
"eval_samples_per_second": 23.399,
"eval_steps_per_second": 1.47,
"step": 4396
},
{
"epoch": 4.0,
"grad_norm": 17.691198348999023,
"learning_rate": 2.6158586391295163e-05,
"loss": 0.3153,
"step": 4400
},
{
"epoch": 4.09,
"grad_norm": 10.9254150390625,
"learning_rate": 2.5612206206554034e-05,
"loss": 0.2663,
"step": 4500
},
{
"epoch": 4.19,
"grad_norm": 29.32948875427246,
"learning_rate": 2.5061240292598336e-05,
"loss": 0.2546,
"step": 4600
},
{
"epoch": 4.28,
"grad_norm": 15.644346237182617,
"learning_rate": 2.4506138843236867e-05,
"loss": 0.2597,
"step": 4700
},
{
"epoch": 4.37,
"grad_norm": 12.25552749633789,
"learning_rate": 2.3947355431421094e-05,
"loss": 0.2506,
"step": 4800
},
{
"epoch": 4.46,
"grad_norm": 25.368473052978516,
"learning_rate": 2.3385346638631082e-05,
"loss": 0.2458,
"step": 4900
},
{
"epoch": 4.55,
"grad_norm": 17.984514236450195,
"learning_rate": 2.2820571681803108e-05,
"loss": 0.2692,
"step": 5000
},
{
"epoch": 4.64,
"grad_norm": 10.834085464477539,
"learning_rate": 2.225349203810388e-05,
"loss": 0.2347,
"step": 5100
},
{
"epoch": 4.73,
"grad_norm": 23.141050338745117,
"learning_rate": 2.16845710678579e-05,
"loss": 0.2799,
"step": 5200
},
{
"epoch": 4.82,
"grad_norm": 8.481077194213867,
"learning_rate": 2.111427363593615e-05,
"loss": 0.2713,
"step": 5300
},
{
"epoch": 4.91,
"grad_norm": 1.8054014444351196,
"learning_rate": 2.0543065731915316e-05,
"loss": 0.2901,
"step": 5400
},
{
"epoch": 5.0,
"eval_accuracy": 0.9427435387673956,
"eval_loss": 0.23892664909362793,
"eval_runtime": 111.9885,
"eval_samples_per_second": 22.458,
"eval_steps_per_second": 1.411,
"step": 5495
},
{
"epoch": 5.0,
"grad_norm": 5.089294910430908,
"learning_rate": 1.997141408931814e-05,
"loss": 0.2719,
"step": 5500
},
{
"epoch": 5.1,
"grad_norm": 18.415925979614258,
"learning_rate": 1.9399785804245775e-05,
"loss": 0.2031,
"step": 5600
},
{
"epoch": 5.19,
"grad_norm": 5.980634689331055,
"learning_rate": 1.8828647953713982e-05,
"loss": 0.2062,
"step": 5700
},
{
"epoch": 5.28,
"grad_norm": 12.657254219055176,
"learning_rate": 1.825846721400482e-05,
"loss": 0.2194,
"step": 5800
},
{
"epoch": 5.37,
"grad_norm": 0.39924386143684387,
"learning_rate": 1.7689709479345895e-05,
"loss": 0.1904,
"step": 5900
},
{
"epoch": 5.46,
"grad_norm": 14.926392555236816,
"learning_rate": 1.7122839481228503e-05,
"loss": 0.2239,
"step": 6000
},
{
"epoch": 5.55,
"grad_norm": 18.79764175415039,
"learning_rate": 1.65583204086759e-05,
"loss": 0.2309,
"step": 6100
},
{
"epoch": 5.64,
"grad_norm": 8.083695411682129,
"learning_rate": 1.599661352977198e-05,
"loss": 0.2275,
"step": 6200
},
{
"epoch": 5.73,
"grad_norm": 32.457725524902344,
"learning_rate": 1.543817781475941e-05,
"loss": 0.2326,
"step": 6300
},
{
"epoch": 5.82,
"grad_norm": 15.148175239562988,
"learning_rate": 1.4883469561015482e-05,
"loss": 0.1995,
"step": 6400
},
{
"epoch": 5.91,
"grad_norm": 18.512500762939453,
"learning_rate": 1.433294202021182e-05,
"loss": 0.2381,
"step": 6500
},
{
"epoch": 6.0,
"eval_accuracy": 0.941948310139165,
"eval_loss": 0.2407774031162262,
"eval_runtime": 112.0327,
"eval_samples_per_second": 22.449,
"eval_steps_per_second": 1.41,
"step": 6594
},
{
"epoch": 6.01,
"grad_norm": 0.42633116245269775,
"learning_rate": 1.378704502796289e-05,
"loss": 0.2012,
"step": 6600
},
{
"epoch": 6.1,
"grad_norm": 22.5328426361084,
"learning_rate": 1.3246224636265588e-05,
"loss": 0.2241,
"step": 6700
},
{
"epoch": 6.19,
"grad_norm": 12.483528137207031,
"learning_rate": 1.2710922749030558e-05,
"loss": 0.2037,
"step": 6800
},
{
"epoch": 6.28,
"grad_norm": 1.0468084812164307,
"learning_rate": 1.2181576761002872e-05,
"loss": 0.1827,
"step": 6900
},
{
"epoch": 6.37,
"grad_norm": 1.2360355854034424,
"learning_rate": 1.1658619200367073e-05,
"loss": 0.1708,
"step": 7000
},
{
"epoch": 6.46,
"grad_norm": 17.188753128051758,
"learning_rate": 1.1142477375328796e-05,
"loss": 0.2357,
"step": 7100
},
{
"epoch": 6.55,
"grad_norm": 1.7141854763031006,
"learning_rate": 1.0633573024961584e-05,
"loss": 0.1534,
"step": 7200
},
{
"epoch": 6.64,
"grad_norm": 1.2022801637649536,
"learning_rate": 1.013232197460415e-05,
"loss": 0.1753,
"step": 7300
},
{
"epoch": 6.73,
"grad_norm": 15.08975601196289,
"learning_rate": 9.639133796089901e-06,
"loss": 0.1601,
"step": 7400
},
{
"epoch": 6.82,
"grad_norm": 29.775014877319336,
"learning_rate": 9.154411473086087e-06,
"loss": 0.1623,
"step": 7500
},
{
"epoch": 6.92,
"grad_norm": 6.941558361053467,
"learning_rate": 8.678551071816193e-06,
"loss": 0.2084,
"step": 7600
},
{
"epoch": 7.0,
"eval_accuracy": 0.9463220675944334,
"eval_loss": 0.23118199408054352,
"eval_runtime": 111.9458,
"eval_samples_per_second": 22.466,
"eval_steps_per_second": 1.411,
"step": 7693
},
{
"epoch": 7.01,
"grad_norm": 13.732728958129883,
"learning_rate": 8.211941417434488e-06,
"loss": 0.169,
"step": 7700
},
{
"epoch": 7.1,
"grad_norm": 0.291477233171463,
"learning_rate": 7.754963776317355e-06,
"loss": 0.1618,
"step": 7800
},
{
"epoch": 7.19,
"grad_norm": 4.71694278717041,
"learning_rate": 7.307991544530839e-06,
"loss": 0.1636,
"step": 7900
},
{
"epoch": 7.28,
"grad_norm": 0.7449569702148438,
"learning_rate": 6.871389942728983e-06,
"loss": 0.1638,
"step": 8000
},
{
"epoch": 7.37,
"grad_norm": 1.087550401687622,
"learning_rate": 6.445515717732332e-06,
"loss": 0.1695,
"step": 8100
},
{
"epoch": 7.46,
"grad_norm": 14.644157409667969,
"learning_rate": 6.03071685103043e-06,
"loss": 0.1437,
"step": 8200
},
{
"epoch": 7.55,
"grad_norm": 35.039981842041016,
"learning_rate": 5.627332274446377e-06,
"loss": 0.174,
"step": 8300
},
{
"epoch": 7.64,
"grad_norm": 7.3837080001831055,
"learning_rate": 5.2356915931959086e-06,
"loss": 0.1604,
"step": 8400
},
{
"epoch": 7.73,
"grad_norm": 10.73990249633789,
"learning_rate": 4.856114816567281e-06,
"loss": 0.1443,
"step": 8500
},
{
"epoch": 7.83,
"grad_norm": 14.905359268188477,
"learning_rate": 4.488912096441886e-06,
"loss": 0.162,
"step": 8600
},
{
"epoch": 7.92,
"grad_norm": 0.12942950427532196,
"learning_rate": 4.13438347386943e-06,
"loss": 0.1742,
"step": 8700
},
{
"epoch": 8.0,
"eval_accuracy": 0.9451292246520875,
"eval_loss": 0.23588231205940247,
"eval_runtime": 112.1261,
"eval_samples_per_second": 22.43,
"eval_steps_per_second": 1.409,
"step": 8792
},
{
"epoch": 8.01,
"grad_norm": 17.91644859313965,
"learning_rate": 3.7928186339045982e-06,
"loss": 0.1865,
"step": 8800
},
{
"epoch": 8.1,
"grad_norm": 2.5071592330932617,
"learning_rate": 3.464496668905677e-06,
"loss": 0.1669,
"step": 8900
},
{
"epoch": 8.19,
"grad_norm": 0.10696269571781158,
"learning_rate": 3.14968585048836e-06,
"loss": 0.1535,
"step": 9000
},
{
"epoch": 8.28,
"grad_norm": 0.128840371966362,
"learning_rate": 2.848643410321257e-06,
"loss": 0.1154,
"step": 9100
},
{
"epoch": 8.37,
"grad_norm": 3.628100633621216,
"learning_rate": 2.561615329942089e-06,
"loss": 0.1564,
"step": 9200
},
{
"epoch": 8.46,
"grad_norm": 11.36262321472168,
"learning_rate": 2.288836139766373e-06,
"loss": 0.1267,
"step": 9300
},
{
"epoch": 8.55,
"grad_norm": 6.693118095397949,
"learning_rate": 2.0305287274528073e-06,
"loss": 0.119,
"step": 9400
},
{
"epoch": 8.64,
"grad_norm": 10.11449146270752,
"learning_rate": 1.7869041557819456e-06,
"loss": 0.1749,
"step": 9500
},
{
"epoch": 8.74,
"grad_norm": 20.39699363708496,
"learning_rate": 1.558161490196961e-06,
"loss": 0.1405,
"step": 9600
},
{
"epoch": 8.83,
"grad_norm": 23.157207489013672,
"learning_rate": 1.34448763614744e-06,
"loss": 0.1765,
"step": 9700
},
{
"epoch": 8.92,
"grad_norm": 13.510420799255371,
"learning_rate": 1.1460571863690762e-06,
"loss": 0.1582,
"step": 9800
},
{
"epoch": 9.0,
"eval_accuracy": 0.9479125248508946,
"eval_loss": 0.23640932142734528,
"eval_runtime": 111.9275,
"eval_samples_per_second": 22.47,
"eval_steps_per_second": 1.412,
"step": 9891
},
{
"epoch": 9.01,
"grad_norm": 16.904611587524414,
"learning_rate": 9.63032278224101e-07,
"loss": 0.1588,
"step": 9900
},
{
"epoch": 9.1,
"grad_norm": 0.36576932668685913,
"learning_rate": 7.955624612189594e-07,
"loss": 0.1708,
"step": 10000
},
{
"epoch": 9.19,
"grad_norm": 26.957815170288086,
"learning_rate": 6.4378457480754e-07,
"loss": 0.1847,
"step": 10100
},
{
"epoch": 9.28,
"grad_norm": 17.24859619140625,
"learning_rate": 5.078226365797556e-07,
"loss": 0.1401,
"step": 10200
},
{
"epoch": 9.37,
"grad_norm": 10.3253755569458,
"learning_rate": 3.877877409268771e-07,
"loss": 0.1398,
"step": 10300
},
{
"epoch": 9.46,
"grad_norm": 0.26469576358795166,
"learning_rate": 2.837779682663899e-07,
"loss": 0.1309,
"step": 10400
},
{
"epoch": 9.55,
"grad_norm": 7.175375461578369,
"learning_rate": 1.958783049005719e-07,
"loss": 0.1637,
"step": 10500
},
{
"epoch": 9.65,
"grad_norm": 0.38347867131233215,
"learning_rate": 1.2416057357423684e-07,
"loss": 0.1888,
"step": 10600
},
{
"epoch": 9.74,
"grad_norm": 11.241958618164062,
"learning_rate": 6.868337478844523e-08,
"loss": 0.156,
"step": 10700
},
{
"epoch": 9.83,
"grad_norm": 19.296527862548828,
"learning_rate": 2.9492038918070844e-08,
"loss": 0.135,
"step": 10800
},
{
"epoch": 9.92,
"grad_norm": 14.173076629638672,
"learning_rate": 6.618589172393553e-09,
"loss": 0.1451,
"step": 10900
},
{
"epoch": 10.0,
"eval_accuracy": 0.9495029821073558,
"eval_loss": 0.2357349991798401,
"eval_runtime": 111.9428,
"eval_samples_per_second": 22.467,
"eval_steps_per_second": 1.411,
"step": 10990
},
{
"epoch": 10.0,
"step": 10990,
"total_flos": 4.09349935387607e+19,
"train_loss": 0.3212610531113601,
"train_runtime": 19604.4461,
"train_samples_per_second": 8.968,
"train_steps_per_second": 0.561
}
],
"logging_steps": 100,
"max_steps": 10990,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"save_steps": 500,
"total_flos": 4.09349935387607e+19,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}