dzungpham's picture
upload model weights after 800 training steps with frozen pretrained weights
4a296ec verified
raw
history blame
12.6 kB
{
"best_global_step": 600,
"best_metric": 0.621848143053056,
"best_model_checkpoint": "./taskA-codebert-base/checkpoint-600",
"epoch": 0.07679508511455267,
"eval_steps": 200,
"global_step": 600,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0012799180852425445,
"grad_norm": 3.6954128742218018,
"learning_rate": 3.840409643695328e-08,
"loss": 0.7763,
"step": 10
},
{
"epoch": 0.002559836170485089,
"grad_norm": 3.3660645484924316,
"learning_rate": 8.10753147002347e-08,
"loss": 0.7763,
"step": 20
},
{
"epoch": 0.0038397542557276334,
"grad_norm": 4.491067886352539,
"learning_rate": 1.2374653296351612e-07,
"loss": 0.7889,
"step": 30
},
{
"epoch": 0.005119672340970178,
"grad_norm": 1.5920521020889282,
"learning_rate": 1.6641775122679754e-07,
"loss": 0.7888,
"step": 40
},
{
"epoch": 0.006399590426212722,
"grad_norm": 4.410898208618164,
"learning_rate": 2.0908896949007894e-07,
"loss": 0.7692,
"step": 50
},
{
"epoch": 0.007679508511455267,
"grad_norm": 2.5669264793395996,
"learning_rate": 2.517601877533604e-07,
"loss": 0.7735,
"step": 60
},
{
"epoch": 0.008959426596697812,
"grad_norm": 6.190986156463623,
"learning_rate": 2.944314060166418e-07,
"loss": 0.7679,
"step": 70
},
{
"epoch": 0.010239344681940356,
"grad_norm": 5.736413478851318,
"learning_rate": 3.371026242799232e-07,
"loss": 0.744,
"step": 80
},
{
"epoch": 0.011519262767182901,
"grad_norm": 4.278866767883301,
"learning_rate": 3.7977384254320464e-07,
"loss": 0.7542,
"step": 90
},
{
"epoch": 0.012799180852425445,
"grad_norm": 5.565455436706543,
"learning_rate": 4.22445060806486e-07,
"loss": 0.752,
"step": 100
},
{
"epoch": 0.01407909893766799,
"grad_norm": 4.031781196594238,
"learning_rate": 4.651162790697675e-07,
"loss": 0.7614,
"step": 110
},
{
"epoch": 0.015359017022910534,
"grad_norm": 2.484126329421997,
"learning_rate": 5.077874973330489e-07,
"loss": 0.7664,
"step": 120
},
{
"epoch": 0.016638935108153077,
"grad_norm": 2.7525155544281006,
"learning_rate": 5.504587155963304e-07,
"loss": 0.7521,
"step": 130
},
{
"epoch": 0.017918853193395624,
"grad_norm": 1.6374385356903076,
"learning_rate": 5.931299338596117e-07,
"loss": 0.7436,
"step": 140
},
{
"epoch": 0.019198771278638168,
"grad_norm": 2.370774269104004,
"learning_rate": 6.358011521228932e-07,
"loss": 0.7429,
"step": 150
},
{
"epoch": 0.02047868936388071,
"grad_norm": 2.8948566913604736,
"learning_rate": 6.784723703861745e-07,
"loss": 0.7333,
"step": 160
},
{
"epoch": 0.021758607449123255,
"grad_norm": 2.7481186389923096,
"learning_rate": 7.21143588649456e-07,
"loss": 0.7294,
"step": 170
},
{
"epoch": 0.023038525534365802,
"grad_norm": 2.0159404277801514,
"learning_rate": 7.638148069127374e-07,
"loss": 0.731,
"step": 180
},
{
"epoch": 0.024318443619608346,
"grad_norm": 3.6898131370544434,
"learning_rate": 8.064860251760189e-07,
"loss": 0.7153,
"step": 190
},
{
"epoch": 0.02559836170485089,
"grad_norm": 3.2842342853546143,
"learning_rate": 8.491572434393003e-07,
"loss": 0.7245,
"step": 200
},
{
"epoch": 0.02559836170485089,
"eval_accuracy": 0.47695,
"eval_loss": 0.7148504853248596,
"eval_macro_f1": 0.3229290091066048,
"eval_precision": 0.238475,
"eval_recall": 0.5,
"eval_runtime": 1641.324,
"eval_samples_per_second": 60.926,
"eval_steps_per_second": 0.476,
"step": 200
},
{
"epoch": 0.026878279790093433,
"grad_norm": 2.294558048248291,
"learning_rate": 8.918284617025817e-07,
"loss": 0.718,
"step": 210
},
{
"epoch": 0.02815819787533598,
"grad_norm": 2.9467175006866455,
"learning_rate": 9.344996799658632e-07,
"loss": 0.7076,
"step": 220
},
{
"epoch": 0.029438115960578524,
"grad_norm": 2.6889193058013916,
"learning_rate": 9.771708982291445e-07,
"loss": 0.7019,
"step": 230
},
{
"epoch": 0.030718034045821067,
"grad_norm": 0.7491384744644165,
"learning_rate": 1.0198421164924258e-06,
"loss": 0.7046,
"step": 240
},
{
"epoch": 0.031997952131063614,
"grad_norm": 2.0363433361053467,
"learning_rate": 1.0625133347557074e-06,
"loss": 0.7085,
"step": 250
},
{
"epoch": 0.033277870216306155,
"grad_norm": 1.1431337594985962,
"learning_rate": 1.1051845530189888e-06,
"loss": 0.7008,
"step": 260
},
{
"epoch": 0.0345577883015487,
"grad_norm": 3.7735233306884766,
"learning_rate": 1.1478557712822702e-06,
"loss": 0.7015,
"step": 270
},
{
"epoch": 0.03583770638679125,
"grad_norm": 0.7889328002929688,
"learning_rate": 1.1905269895455517e-06,
"loss": 0.699,
"step": 280
},
{
"epoch": 0.03711762447203379,
"grad_norm": 0.8796226382255554,
"learning_rate": 1.233198207808833e-06,
"loss": 0.6922,
"step": 290
},
{
"epoch": 0.038397542557276336,
"grad_norm": 0.7966652512550354,
"learning_rate": 1.2758694260721145e-06,
"loss": 0.6949,
"step": 300
},
{
"epoch": 0.039677460642518876,
"grad_norm": 0.7029069066047668,
"learning_rate": 1.318540644335396e-06,
"loss": 0.695,
"step": 310
},
{
"epoch": 0.04095737872776142,
"grad_norm": 2.04327392578125,
"learning_rate": 1.3612118625986772e-06,
"loss": 0.6944,
"step": 320
},
{
"epoch": 0.04223729681300397,
"grad_norm": 0.918025553226471,
"learning_rate": 1.4038830808619588e-06,
"loss": 0.6863,
"step": 330
},
{
"epoch": 0.04351721489824651,
"grad_norm": 0.9528706073760986,
"learning_rate": 1.4465542991252401e-06,
"loss": 0.6907,
"step": 340
},
{
"epoch": 0.04479713298348906,
"grad_norm": 1.071359634399414,
"learning_rate": 1.4892255173885215e-06,
"loss": 0.6873,
"step": 350
},
{
"epoch": 0.046077051068731605,
"grad_norm": 1.9875625371932983,
"learning_rate": 1.531896735651803e-06,
"loss": 0.6942,
"step": 360
},
{
"epoch": 0.047356969153974145,
"grad_norm": 2.0222439765930176,
"learning_rate": 1.5745679539150842e-06,
"loss": 0.6898,
"step": 370
},
{
"epoch": 0.04863688723921669,
"grad_norm": 1.7827857732772827,
"learning_rate": 1.6172391721783658e-06,
"loss": 0.6909,
"step": 380
},
{
"epoch": 0.04991680532445923,
"grad_norm": 1.065674901008606,
"learning_rate": 1.6599103904416472e-06,
"loss": 0.6877,
"step": 390
},
{
"epoch": 0.05119672340970178,
"grad_norm": 0.8346193432807922,
"learning_rate": 1.7025816087049288e-06,
"loss": 0.6889,
"step": 400
},
{
"epoch": 0.05119672340970178,
"eval_accuracy": 0.54043,
"eval_loss": 0.6827595233917236,
"eval_macro_f1": 0.41833840347598905,
"eval_precision": 0.5839205994464474,
"eval_recall": 0.5204152767503025,
"eval_runtime": 1634.6244,
"eval_samples_per_second": 61.176,
"eval_steps_per_second": 0.478,
"step": 400
},
{
"epoch": 0.052476641494944326,
"grad_norm": 0.8074935674667358,
"learning_rate": 1.7452528269682101e-06,
"loss": 0.684,
"step": 410
},
{
"epoch": 0.053756559580186866,
"grad_norm": 0.8965069651603699,
"learning_rate": 1.7879240452314913e-06,
"loss": 0.6922,
"step": 420
},
{
"epoch": 0.05503647766542941,
"grad_norm": 2.597215414047241,
"learning_rate": 1.8305952634947729e-06,
"loss": 0.6909,
"step": 430
},
{
"epoch": 0.05631639575067196,
"grad_norm": 0.8233373165130615,
"learning_rate": 1.8732664817580542e-06,
"loss": 0.6831,
"step": 440
},
{
"epoch": 0.0575963138359145,
"grad_norm": 1.0624489784240723,
"learning_rate": 1.915937700021336e-06,
"loss": 0.6817,
"step": 450
},
{
"epoch": 0.05887623192115705,
"grad_norm": 2.0623559951782227,
"learning_rate": 1.958608918284617e-06,
"loss": 0.6819,
"step": 460
},
{
"epoch": 0.06015615000639959,
"grad_norm": 0.912433922290802,
"learning_rate": 2.0012801365478988e-06,
"loss": 0.686,
"step": 470
},
{
"epoch": 0.061436068091642135,
"grad_norm": 0.6926271319389343,
"learning_rate": 2.04395135481118e-06,
"loss": 0.6834,
"step": 480
},
{
"epoch": 0.06271598617688467,
"grad_norm": 1.498634934425354,
"learning_rate": 2.0866225730744615e-06,
"loss": 0.6805,
"step": 490
},
{
"epoch": 0.06399590426212723,
"grad_norm": 1.1818156242370605,
"learning_rate": 2.129293791337743e-06,
"loss": 0.6805,
"step": 500
},
{
"epoch": 0.06527582234736977,
"grad_norm": 1.4476051330566406,
"learning_rate": 2.1719650096010242e-06,
"loss": 0.6773,
"step": 510
},
{
"epoch": 0.06655574043261231,
"grad_norm": 1.1143743991851807,
"learning_rate": 2.214636227864306e-06,
"loss": 0.6801,
"step": 520
},
{
"epoch": 0.06783565851785486,
"grad_norm": 1.7030247449874878,
"learning_rate": 2.257307446127587e-06,
"loss": 0.6794,
"step": 530
},
{
"epoch": 0.0691155766030974,
"grad_norm": 0.8592290282249451,
"learning_rate": 2.2999786643908685e-06,
"loss": 0.6736,
"step": 540
},
{
"epoch": 0.07039549468833994,
"grad_norm": 0.8231940865516663,
"learning_rate": 2.34264988265415e-06,
"loss": 0.67,
"step": 550
},
{
"epoch": 0.0716754127735825,
"grad_norm": 0.88897705078125,
"learning_rate": 2.3853211009174317e-06,
"loss": 0.685,
"step": 560
},
{
"epoch": 0.07295533085882504,
"grad_norm": 1.6297541856765747,
"learning_rate": 2.427992319180713e-06,
"loss": 0.6797,
"step": 570
},
{
"epoch": 0.07423524894406758,
"grad_norm": 2.9297447204589844,
"learning_rate": 2.470663537443994e-06,
"loss": 0.6766,
"step": 580
},
{
"epoch": 0.07551516702931012,
"grad_norm": 0.9277703762054443,
"learning_rate": 2.5133347557072756e-06,
"loss": 0.6824,
"step": 590
},
{
"epoch": 0.07679508511455267,
"grad_norm": 0.5883333683013916,
"learning_rate": 2.556005973970557e-06,
"loss": 0.6818,
"step": 600
},
{
"epoch": 0.07679508511455267,
"eval_accuracy": 0.63909,
"eval_loss": 0.6700075268745422,
"eval_macro_f1": 0.621848143053056,
"eval_precision": 0.6524313071088819,
"eval_recall": 0.6305863955136095,
"eval_runtime": 1633.9732,
"eval_samples_per_second": 61.201,
"eval_steps_per_second": 0.479,
"step": 600
}
],
"logging_steps": 10,
"max_steps": 23439,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 200,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 3,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 0
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 1.0103464525824e+16,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}