SLM_vs_LLM_experiments
/
max_seq_length_128_experiments
/Qwen
/Qwen1.5_1.8B_patent
/checkpoint-750
/trainer_state.json
{ | |
"best_metric": 0.8989499807357788, | |
"best_model_checkpoint": "../../experiments_checkpoints/MAdAiLab/Qwen/Qwen1.5_1.8B_patent/checkpoint-750", | |
"epoch": 1.918158567774936, | |
"eval_steps": 50, | |
"global_step": 750, | |
"is_hyper_param_search": false, | |
"is_local_process_zero": true, | |
"is_world_process_zero": true, | |
"log_history": [ | |
{ | |
"epoch": 0.03, | |
"grad_norm": 117.48015594482422, | |
"learning_rate": 4.9573742540494466e-06, | |
"loss": 3.9523, | |
"step": 10 | |
}, | |
{ | |
"epoch": 0.05, | |
"grad_norm": 108.45614624023438, | |
"learning_rate": 4.914748508098892e-06, | |
"loss": 2.4902, | |
"step": 20 | |
}, | |
{ | |
"epoch": 0.08, | |
"grad_norm": 70.28152465820312, | |
"learning_rate": 4.872122762148338e-06, | |
"loss": 1.9918, | |
"step": 30 | |
}, | |
{ | |
"epoch": 0.1, | |
"grad_norm": 79.78052520751953, | |
"learning_rate": 4.829497016197784e-06, | |
"loss": 1.6672, | |
"step": 40 | |
}, | |
{ | |
"epoch": 0.13, | |
"grad_norm": 45.827728271484375, | |
"learning_rate": 4.7868712702472295e-06, | |
"loss": 1.4469, | |
"step": 50 | |
}, | |
{ | |
"epoch": 0.13, | |
"eval_accuracy": 0.528, | |
"eval_f1_macro": 0.3841980536108302, | |
"eval_f1_micro": 0.528, | |
"eval_loss": 1.3521000146865845, | |
"eval_runtime": 12.8673, | |
"eval_samples_per_second": 388.581, | |
"eval_steps_per_second": 6.14, | |
"step": 50 | |
}, | |
{ | |
"epoch": 0.15, | |
"grad_norm": 34.13399887084961, | |
"learning_rate": 4.744245524296676e-06, | |
"loss": 1.3984, | |
"step": 60 | |
}, | |
{ | |
"epoch": 0.18, | |
"grad_norm": 39.08888626098633, | |
"learning_rate": 4.701619778346122e-06, | |
"loss": 1.2457, | |
"step": 70 | |
}, | |
{ | |
"epoch": 0.2, | |
"grad_norm": 70.29698181152344, | |
"learning_rate": 4.658994032395567e-06, | |
"loss": 1.3391, | |
"step": 80 | |
}, | |
{ | |
"epoch": 0.23, | |
"grad_norm": 38.475982666015625, | |
"learning_rate": 4.616368286445013e-06, | |
"loss": 1.2006, | |
"step": 90 | |
}, | |
{ | |
"epoch": 0.26, | |
"grad_norm": 34.23910903930664, | |
"learning_rate": 4.573742540494459e-06, | |
"loss": 1.1465, | |
"step": 100 | |
}, | |
{ | |
"epoch": 0.26, | |
"eval_accuracy": 0.596, | |
"eval_f1_macro": 0.4990873064564526, | |
"eval_f1_micro": 0.596, | |
"eval_loss": 1.1614124774932861, | |
"eval_runtime": 12.7706, | |
"eval_samples_per_second": 391.525, | |
"eval_steps_per_second": 6.186, | |
"step": 100 | |
}, | |
{ | |
"epoch": 0.28, | |
"grad_norm": 54.537071228027344, | |
"learning_rate": 4.531116794543905e-06, | |
"loss": 1.1721, | |
"step": 110 | |
}, | |
{ | |
"epoch": 0.31, | |
"grad_norm": 80.3447494506836, | |
"learning_rate": 4.488491048593351e-06, | |
"loss": 1.1453, | |
"step": 120 | |
}, | |
{ | |
"epoch": 0.33, | |
"grad_norm": 37.36294937133789, | |
"learning_rate": 4.445865302642797e-06, | |
"loss": 1.0252, | |
"step": 130 | |
}, | |
{ | |
"epoch": 0.36, | |
"grad_norm": 65.64375305175781, | |
"learning_rate": 4.403239556692242e-06, | |
"loss": 1.0926, | |
"step": 140 | |
}, | |
{ | |
"epoch": 0.38, | |
"grad_norm": 76.81836700439453, | |
"learning_rate": 4.3606138107416884e-06, | |
"loss": 1.1717, | |
"step": 150 | |
}, | |
{ | |
"epoch": 0.38, | |
"eval_accuracy": 0.6286, | |
"eval_f1_macro": 0.5523456257544254, | |
"eval_f1_micro": 0.6286, | |
"eval_loss": 1.0560749769210815, | |
"eval_runtime": 13.0362, | |
"eval_samples_per_second": 383.546, | |
"eval_steps_per_second": 6.06, | |
"step": 150 | |
}, | |
{ | |
"epoch": 0.41, | |
"grad_norm": 26.359981536865234, | |
"learning_rate": 4.317988064791135e-06, | |
"loss": 0.9779, | |
"step": 160 | |
}, | |
{ | |
"epoch": 0.43, | |
"grad_norm": 34.32497024536133, | |
"learning_rate": 4.27536231884058e-06, | |
"loss": 1.0311, | |
"step": 170 | |
}, | |
{ | |
"epoch": 0.46, | |
"grad_norm": 38.703914642333984, | |
"learning_rate": 4.232736572890026e-06, | |
"loss": 0.9883, | |
"step": 180 | |
}, | |
{ | |
"epoch": 0.49, | |
"grad_norm": 40.25131607055664, | |
"learning_rate": 4.190110826939472e-06, | |
"loss": 1.0908, | |
"step": 190 | |
}, | |
{ | |
"epoch": 0.51, | |
"grad_norm": 34.45612335205078, | |
"learning_rate": 4.1474850809889175e-06, | |
"loss": 0.9861, | |
"step": 200 | |
}, | |
{ | |
"epoch": 0.51, | |
"eval_accuracy": 0.6682, | |
"eval_f1_macro": 0.581301697324148, | |
"eval_f1_micro": 0.6682, | |
"eval_loss": 0.9591749906539917, | |
"eval_runtime": 12.8147, | |
"eval_samples_per_second": 390.178, | |
"eval_steps_per_second": 6.165, | |
"step": 200 | |
}, | |
{ | |
"epoch": 0.54, | |
"grad_norm": 48.11906051635742, | |
"learning_rate": 4.104859335038364e-06, | |
"loss": 1.0355, | |
"step": 210 | |
}, | |
{ | |
"epoch": 0.56, | |
"grad_norm": 40.41707992553711, | |
"learning_rate": 4.06223358908781e-06, | |
"loss": 0.968, | |
"step": 220 | |
}, | |
{ | |
"epoch": 0.59, | |
"grad_norm": 34.4398307800293, | |
"learning_rate": 4.019607843137255e-06, | |
"loss": 0.9777, | |
"step": 230 | |
}, | |
{ | |
"epoch": 0.61, | |
"grad_norm": 52.77057647705078, | |
"learning_rate": 3.976982097186701e-06, | |
"loss": 0.9646, | |
"step": 240 | |
}, | |
{ | |
"epoch": 0.64, | |
"grad_norm": 37.42899703979492, | |
"learning_rate": 3.934356351236147e-06, | |
"loss": 0.9701, | |
"step": 250 | |
}, | |
{ | |
"epoch": 0.64, | |
"eval_accuracy": 0.6658, | |
"eval_f1_macro": 0.5948802716943312, | |
"eval_f1_micro": 0.6658, | |
"eval_loss": 0.957937479019165, | |
"eval_runtime": 12.8026, | |
"eval_samples_per_second": 390.547, | |
"eval_steps_per_second": 6.171, | |
"step": 250 | |
}, | |
{ | |
"epoch": 0.66, | |
"grad_norm": 51.38178634643555, | |
"learning_rate": 3.891730605285593e-06, | |
"loss": 0.9826, | |
"step": 260 | |
}, | |
{ | |
"epoch": 0.69, | |
"grad_norm": 31.820234298706055, | |
"learning_rate": 3.849104859335039e-06, | |
"loss": 0.9791, | |
"step": 270 | |
}, | |
{ | |
"epoch": 0.72, | |
"grad_norm": 39.74692916870117, | |
"learning_rate": 3.8064791133844846e-06, | |
"loss": 0.9047, | |
"step": 280 | |
}, | |
{ | |
"epoch": 0.74, | |
"grad_norm": 28.091707229614258, | |
"learning_rate": 3.7638533674339308e-06, | |
"loss": 0.8908, | |
"step": 290 | |
}, | |
{ | |
"epoch": 0.77, | |
"grad_norm": 37.695556640625, | |
"learning_rate": 3.7212276214833765e-06, | |
"loss": 0.9389, | |
"step": 300 | |
}, | |
{ | |
"epoch": 0.77, | |
"eval_accuracy": 0.679, | |
"eval_f1_macro": 0.6287401468652754, | |
"eval_f1_micro": 0.679, | |
"eval_loss": 0.9363999962806702, | |
"eval_runtime": 12.797, | |
"eval_samples_per_second": 390.715, | |
"eval_steps_per_second": 6.173, | |
"step": 300 | |
}, | |
{ | |
"epoch": 0.79, | |
"grad_norm": 45.27925109863281, | |
"learning_rate": 3.678601875532822e-06, | |
"loss": 0.9262, | |
"step": 310 | |
}, | |
{ | |
"epoch": 0.82, | |
"grad_norm": 33.71603775024414, | |
"learning_rate": 3.6359761295822684e-06, | |
"loss": 0.9182, | |
"step": 320 | |
}, | |
{ | |
"epoch": 0.84, | |
"grad_norm": 40.315677642822266, | |
"learning_rate": 3.593350383631714e-06, | |
"loss": 0.9084, | |
"step": 330 | |
}, | |
{ | |
"epoch": 0.87, | |
"grad_norm": 38.41544723510742, | |
"learning_rate": 3.55072463768116e-06, | |
"loss": 1.0266, | |
"step": 340 | |
}, | |
{ | |
"epoch": 0.9, | |
"grad_norm": 43.862613677978516, | |
"learning_rate": 3.508098891730606e-06, | |
"loss": 0.9914, | |
"step": 350 | |
}, | |
{ | |
"epoch": 0.9, | |
"eval_accuracy": 0.6756, | |
"eval_f1_macro": 0.6115341208469629, | |
"eval_f1_micro": 0.6756, | |
"eval_loss": 0.924637496471405, | |
"eval_runtime": 12.8058, | |
"eval_samples_per_second": 390.447, | |
"eval_steps_per_second": 6.169, | |
"step": 350 | |
}, | |
{ | |
"epoch": 0.92, | |
"grad_norm": 26.705493927001953, | |
"learning_rate": 3.4654731457800517e-06, | |
"loss": 0.9863, | |
"step": 360 | |
}, | |
{ | |
"epoch": 0.95, | |
"grad_norm": 31.23122215270996, | |
"learning_rate": 3.4228473998294974e-06, | |
"loss": 0.8785, | |
"step": 370 | |
}, | |
{ | |
"epoch": 0.97, | |
"grad_norm": 17.27379608154297, | |
"learning_rate": 3.3802216538789436e-06, | |
"loss": 0.8624, | |
"step": 380 | |
}, | |
{ | |
"epoch": 1.0, | |
"grad_norm": 29.550363540649414, | |
"learning_rate": 3.3375959079283893e-06, | |
"loss": 0.9793, | |
"step": 390 | |
}, | |
{ | |
"epoch": 1.02, | |
"grad_norm": 40.75907897949219, | |
"learning_rate": 3.294970161977835e-06, | |
"loss": 0.7508, | |
"step": 400 | |
}, | |
{ | |
"epoch": 1.02, | |
"eval_accuracy": 0.6812, | |
"eval_f1_macro": 0.6406138110649251, | |
"eval_f1_micro": 0.6812, | |
"eval_loss": 0.9046875238418579, | |
"eval_runtime": 12.966, | |
"eval_samples_per_second": 385.623, | |
"eval_steps_per_second": 6.093, | |
"step": 400 | |
}, | |
{ | |
"epoch": 1.05, | |
"grad_norm": 38.431182861328125, | |
"learning_rate": 3.252344416027281e-06, | |
"loss": 0.7557, | |
"step": 410 | |
}, | |
{ | |
"epoch": 1.07, | |
"grad_norm": 20.32223892211914, | |
"learning_rate": 3.2097186700767265e-06, | |
"loss": 0.7158, | |
"step": 420 | |
}, | |
{ | |
"epoch": 1.1, | |
"grad_norm": 22.297561645507812, | |
"learning_rate": 3.1670929241261722e-06, | |
"loss": 0.6851, | |
"step": 430 | |
}, | |
{ | |
"epoch": 1.13, | |
"grad_norm": 23.710554122924805, | |
"learning_rate": 3.124467178175618e-06, | |
"loss": 0.6733, | |
"step": 440 | |
}, | |
{ | |
"epoch": 1.15, | |
"grad_norm": 19.360496520996094, | |
"learning_rate": 3.081841432225064e-06, | |
"loss": 0.6312, | |
"step": 450 | |
}, | |
{ | |
"epoch": 1.15, | |
"eval_accuracy": 0.6844, | |
"eval_f1_macro": 0.641013130489217, | |
"eval_f1_micro": 0.6844, | |
"eval_loss": 0.9342374801635742, | |
"eval_runtime": 12.8113, | |
"eval_samples_per_second": 390.281, | |
"eval_steps_per_second": 6.166, | |
"step": 450 | |
}, | |
{ | |
"epoch": 1.18, | |
"grad_norm": 31.652124404907227, | |
"learning_rate": 3.03921568627451e-06, | |
"loss": 0.7273, | |
"step": 460 | |
}, | |
{ | |
"epoch": 1.2, | |
"grad_norm": 27.453479766845703, | |
"learning_rate": 2.9965899403239555e-06, | |
"loss": 0.6845, | |
"step": 470 | |
}, | |
{ | |
"epoch": 1.23, | |
"grad_norm": 31.358789443969727, | |
"learning_rate": 2.9539641943734017e-06, | |
"loss": 0.6445, | |
"step": 480 | |
}, | |
{ | |
"epoch": 1.25, | |
"grad_norm": 32.35221862792969, | |
"learning_rate": 2.9113384484228474e-06, | |
"loss": 0.6479, | |
"step": 490 | |
}, | |
{ | |
"epoch": 1.28, | |
"grad_norm": 26.45619773864746, | |
"learning_rate": 2.868712702472293e-06, | |
"loss": 0.6436, | |
"step": 500 | |
}, | |
{ | |
"epoch": 1.28, | |
"eval_accuracy": 0.6848, | |
"eval_f1_macro": 0.6409585523940933, | |
"eval_f1_micro": 0.6848, | |
"eval_loss": 0.9463624954223633, | |
"eval_runtime": 12.8005, | |
"eval_samples_per_second": 390.608, | |
"eval_steps_per_second": 6.172, | |
"step": 500 | |
}, | |
{ | |
"epoch": 1.3, | |
"grad_norm": 29.645923614501953, | |
"learning_rate": 2.8260869565217393e-06, | |
"loss": 0.7069, | |
"step": 510 | |
}, | |
{ | |
"epoch": 1.33, | |
"grad_norm": 29.035226821899414, | |
"learning_rate": 2.783461210571185e-06, | |
"loss": 0.7149, | |
"step": 520 | |
}, | |
{ | |
"epoch": 1.36, | |
"grad_norm": 35.058223724365234, | |
"learning_rate": 2.7408354646206308e-06, | |
"loss": 0.628, | |
"step": 530 | |
}, | |
{ | |
"epoch": 1.38, | |
"grad_norm": 38.28398513793945, | |
"learning_rate": 2.698209718670077e-06, | |
"loss": 0.6717, | |
"step": 540 | |
}, | |
{ | |
"epoch": 1.41, | |
"grad_norm": 29.30170440673828, | |
"learning_rate": 2.6555839727195226e-06, | |
"loss": 0.6429, | |
"step": 550 | |
}, | |
{ | |
"epoch": 1.41, | |
"eval_accuracy": 0.6846, | |
"eval_f1_macro": 0.6298638266651704, | |
"eval_f1_micro": 0.6846, | |
"eval_loss": 0.9365624785423279, | |
"eval_runtime": 12.804, | |
"eval_samples_per_second": 390.503, | |
"eval_steps_per_second": 6.17, | |
"step": 550 | |
}, | |
{ | |
"epoch": 1.43, | |
"grad_norm": 42.15592956542969, | |
"learning_rate": 2.6129582267689684e-06, | |
"loss": 0.6613, | |
"step": 560 | |
}, | |
{ | |
"epoch": 1.46, | |
"grad_norm": 40.29536819458008, | |
"learning_rate": 2.5703324808184145e-06, | |
"loss": 0.6224, | |
"step": 570 | |
}, | |
{ | |
"epoch": 1.48, | |
"grad_norm": 31.332948684692383, | |
"learning_rate": 2.5277067348678602e-06, | |
"loss": 0.6877, | |
"step": 580 | |
}, | |
{ | |
"epoch": 1.51, | |
"grad_norm": 29.125625610351562, | |
"learning_rate": 2.4850809889173064e-06, | |
"loss": 0.5654, | |
"step": 590 | |
}, | |
{ | |
"epoch": 1.53, | |
"grad_norm": 27.54241180419922, | |
"learning_rate": 2.442455242966752e-06, | |
"loss": 0.6471, | |
"step": 600 | |
}, | |
{ | |
"epoch": 1.53, | |
"eval_accuracy": 0.6812, | |
"eval_f1_macro": 0.6490292737514075, | |
"eval_f1_micro": 0.6812, | |
"eval_loss": 0.9346874952316284, | |
"eval_runtime": 12.8127, | |
"eval_samples_per_second": 390.237, | |
"eval_steps_per_second": 6.166, | |
"step": 600 | |
}, | |
{ | |
"epoch": 1.56, | |
"grad_norm": 45.75101089477539, | |
"learning_rate": 2.3998294970161983e-06, | |
"loss": 0.6867, | |
"step": 610 | |
}, | |
{ | |
"epoch": 1.59, | |
"grad_norm": 24.49323844909668, | |
"learning_rate": 2.357203751065644e-06, | |
"loss": 0.6749, | |
"step": 620 | |
}, | |
{ | |
"epoch": 1.61, | |
"grad_norm": 36.94871139526367, | |
"learning_rate": 2.3145780051150897e-06, | |
"loss": 0.7851, | |
"step": 630 | |
}, | |
{ | |
"epoch": 1.64, | |
"grad_norm": 37.22614288330078, | |
"learning_rate": 2.2719522591645355e-06, | |
"loss": 0.6278, | |
"step": 640 | |
}, | |
{ | |
"epoch": 1.66, | |
"grad_norm": 25.08321189880371, | |
"learning_rate": 2.229326513213981e-06, | |
"loss": 0.7045, | |
"step": 650 | |
}, | |
{ | |
"epoch": 1.66, | |
"eval_accuracy": 0.6696, | |
"eval_f1_macro": 0.6265175426669454, | |
"eval_f1_micro": 0.6696, | |
"eval_loss": 0.9456750154495239, | |
"eval_runtime": 12.9615, | |
"eval_samples_per_second": 385.758, | |
"eval_steps_per_second": 6.095, | |
"step": 650 | |
}, | |
{ | |
"epoch": 1.69, | |
"grad_norm": 22.992191314697266, | |
"learning_rate": 2.1867007672634273e-06, | |
"loss": 0.6695, | |
"step": 660 | |
}, | |
{ | |
"epoch": 1.71, | |
"grad_norm": 30.21503257751465, | |
"learning_rate": 2.144075021312873e-06, | |
"loss": 0.7161, | |
"step": 670 | |
}, | |
{ | |
"epoch": 1.74, | |
"grad_norm": 31.841976165771484, | |
"learning_rate": 2.101449275362319e-06, | |
"loss": 0.7071, | |
"step": 680 | |
}, | |
{ | |
"epoch": 1.76, | |
"grad_norm": 23.938678741455078, | |
"learning_rate": 2.058823529411765e-06, | |
"loss": 0.6467, | |
"step": 690 | |
}, | |
{ | |
"epoch": 1.79, | |
"grad_norm": 58.5498161315918, | |
"learning_rate": 2.0161977834612107e-06, | |
"loss": 0.6311, | |
"step": 700 | |
}, | |
{ | |
"epoch": 1.79, | |
"eval_accuracy": 0.6924, | |
"eval_f1_macro": 0.6303167192886554, | |
"eval_f1_micro": 0.6924, | |
"eval_loss": 0.9205750226974487, | |
"eval_runtime": 12.809, | |
"eval_samples_per_second": 390.35, | |
"eval_steps_per_second": 6.168, | |
"step": 700 | |
}, | |
{ | |
"epoch": 1.82, | |
"grad_norm": 30.269773483276367, | |
"learning_rate": 1.9735720375106564e-06, | |
"loss": 0.6146, | |
"step": 710 | |
}, | |
{ | |
"epoch": 1.84, | |
"grad_norm": 41.80614471435547, | |
"learning_rate": 1.9309462915601026e-06, | |
"loss": 0.7465, | |
"step": 720 | |
}, | |
{ | |
"epoch": 1.87, | |
"grad_norm": 25.003660202026367, | |
"learning_rate": 1.8883205456095483e-06, | |
"loss": 0.6379, | |
"step": 730 | |
}, | |
{ | |
"epoch": 1.89, | |
"grad_norm": 29.400390625, | |
"learning_rate": 1.8456947996589942e-06, | |
"loss": 0.7268, | |
"step": 740 | |
}, | |
{ | |
"epoch": 1.92, | |
"grad_norm": 20.223796844482422, | |
"learning_rate": 1.80306905370844e-06, | |
"loss": 0.6659, | |
"step": 750 | |
}, | |
{ | |
"epoch": 1.92, | |
"eval_accuracy": 0.6976, | |
"eval_f1_macro": 0.650699169639119, | |
"eval_f1_micro": 0.6976, | |
"eval_loss": 0.8989499807357788, | |
"eval_runtime": 12.7898, | |
"eval_samples_per_second": 390.935, | |
"eval_steps_per_second": 6.177, | |
"step": 750 | |
} | |
], | |
"logging_steps": 10, | |
"max_steps": 1173, | |
"num_input_tokens_seen": 0, | |
"num_train_epochs": 3, | |
"save_steps": 50, | |
"total_flos": 4.4771963830272e+16, | |
"train_batch_size": 32, | |
"trial_name": null, | |
"trial_params": null | |
} | |