QA_TIME1 / trainer_state.json
vitorandrade's picture
Upload 13 files
2c01633 verified
raw
history blame
11.4 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.1880341880341883,
"eval_steps": 100,
"global_step": 2800,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.07814407814407814,
"grad_norm": 0.389704167842865,
"learning_rate": 4.000000000000001e-06,
"loss": 2.4124,
"step": 100
},
{
"epoch": 0.07814407814407814,
"eval_loss": 2.2181031703948975,
"eval_runtime": 3367.0243,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 100
},
{
"epoch": 0.1562881562881563,
"grad_norm": 0.27927058935165405,
"learning_rate": 8.000000000000001e-06,
"loss": 2.2928,
"step": 200
},
{
"epoch": 0.1562881562881563,
"eval_loss": 2.0591659545898438,
"eval_runtime": 3366.419,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 200
},
{
"epoch": 0.23443223443223443,
"grad_norm": 0.22626566886901855,
"learning_rate": 1.2e-05,
"loss": 2.1104,
"step": 300
},
{
"epoch": 0.23443223443223443,
"eval_loss": 1.8923490047454834,
"eval_runtime": 3360.9485,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 300
},
{
"epoch": 0.3125763125763126,
"grad_norm": 0.210145503282547,
"learning_rate": 1.6000000000000003e-05,
"loss": 1.9296,
"step": 400
},
{
"epoch": 0.3125763125763126,
"eval_loss": 1.7505871057510376,
"eval_runtime": 3357.6413,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 400
},
{
"epoch": 0.3907203907203907,
"grad_norm": 0.2200106978416443,
"learning_rate": 2e-05,
"loss": 1.8345,
"step": 500
},
{
"epoch": 0.3907203907203907,
"eval_loss": 1.7071024179458618,
"eval_runtime": 3361.7985,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 500
},
{
"epoch": 0.46886446886446886,
"grad_norm": 0.22381120920181274,
"learning_rate": 1.9980028422948323e-05,
"loss": 1.8103,
"step": 600
},
{
"epoch": 0.46886446886446886,
"eval_loss": 1.6838117837905884,
"eval_runtime": 3360.7216,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 600
},
{
"epoch": 0.5470085470085471,
"grad_norm": 0.2300158590078354,
"learning_rate": 1.9920193464571277e-05,
"loss": 1.7827,
"step": 700
},
{
"epoch": 0.5470085470085471,
"eval_loss": 1.6691471338272095,
"eval_runtime": 3362.2,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 700
},
{
"epoch": 0.6251526251526252,
"grad_norm": 0.25236302614212036,
"learning_rate": 1.982073412456518e-05,
"loss": 1.765,
"step": 800
},
{
"epoch": 0.6251526251526252,
"eval_loss": 1.6575521230697632,
"eval_runtime": 3363.3293,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 800
},
{
"epoch": 0.7032967032967034,
"grad_norm": 0.26079344749450684,
"learning_rate": 1.9682047674904527e-05,
"loss": 1.7425,
"step": 900
},
{
"epoch": 0.7032967032967034,
"eval_loss": 1.6489633321762085,
"eval_runtime": 3359.1797,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 900
},
{
"epoch": 0.7814407814407814,
"grad_norm": 0.2565068304538727,
"learning_rate": 1.9504688073012397e-05,
"loss": 1.7521,
"step": 1000
},
{
"epoch": 0.7814407814407814,
"eval_loss": 1.6409879922866821,
"eval_runtime": 3358.9412,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 1000
},
{
"epoch": 0.8595848595848596,
"grad_norm": 0.26189491152763367,
"learning_rate": 1.9289363749079798e-05,
"loss": 1.7326,
"step": 1100
},
{
"epoch": 0.8595848595848596,
"eval_loss": 1.6336146593093872,
"eval_runtime": 3363.4368,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 1100
},
{
"epoch": 0.9377289377289377,
"grad_norm": 0.2778639495372772,
"learning_rate": 1.903693477637204e-05,
"loss": 1.7298,
"step": 1200
},
{
"epoch": 0.9377289377289377,
"eval_loss": 1.627331256866455,
"eval_runtime": 3361.7893,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 1200
},
{
"epoch": 1.0158730158730158,
"grad_norm": 0.2954370975494385,
"learning_rate": 1.874840943582482e-05,
"loss": 1.7128,
"step": 1300
},
{
"epoch": 1.0158730158730158,
"eval_loss": 1.62191641330719,
"eval_runtime": 3358.2318,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 1300
},
{
"epoch": 1.0940170940170941,
"grad_norm": 0.2985123097896576,
"learning_rate": 1.842494018865216e-05,
"loss": 1.7265,
"step": 1400
},
{
"epoch": 1.0940170940170941,
"eval_loss": 1.616808295249939,
"eval_runtime": 3366.219,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 1400
},
{
"epoch": 1.1721611721611722,
"grad_norm": 0.3037821054458618,
"learning_rate": 1.8067819073052813e-05,
"loss": 1.7148,
"step": 1500
},
{
"epoch": 1.1721611721611722,
"eval_loss": 1.612409234046936,
"eval_runtime": 3364.8773,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 1500
},
{
"epoch": 1.2503052503052503,
"grad_norm": 0.3078468441963196,
"learning_rate": 1.7678472543402166e-05,
"loss": 1.7003,
"step": 1600
},
{
"epoch": 1.2503052503052503,
"eval_loss": 1.607498049736023,
"eval_runtime": 3366.2319,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 1600
},
{
"epoch": 1.3284493284493284,
"grad_norm": 0.3602350056171417,
"learning_rate": 1.7258455772543573e-05,
"loss": 1.6981,
"step": 1700
},
{
"epoch": 1.3284493284493284,
"eval_loss": 1.6040035486221313,
"eval_runtime": 3369.5225,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 1700
},
{
"epoch": 1.4065934065934065,
"grad_norm": 0.34948959946632385,
"learning_rate": 1.6809446439937472e-05,
"loss": 1.703,
"step": 1800
},
{
"epoch": 1.4065934065934065,
"eval_loss": 1.6002304553985596,
"eval_runtime": 3359.2322,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 1800
},
{
"epoch": 1.4847374847374848,
"grad_norm": 0.33941686153411865,
"learning_rate": 1.6333238030480473e-05,
"loss": 1.6842,
"step": 1900
},
{
"epoch": 1.4847374847374848,
"eval_loss": 1.5976512432098389,
"eval_runtime": 3363.9968,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 1900
},
{
"epoch": 1.5628815628815629,
"grad_norm": 0.3594741225242615,
"learning_rate": 1.5831732670761e-05,
"loss": 1.6813,
"step": 2000
},
{
"epoch": 1.5628815628815629,
"eval_loss": 1.5942325592041016,
"eval_runtime": 3356.6247,
"eval_samples_per_second": 0.753,
"eval_steps_per_second": 0.094,
"step": 2000
},
{
"epoch": 1.641025641025641,
"grad_norm": 0.3513031303882599,
"learning_rate": 1.5306933531365748e-05,
"loss": 1.6747,
"step": 2100
},
{
"epoch": 1.641025641025641,
"eval_loss": 1.591470718383789,
"eval_runtime": 3360.2103,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 2100
},
{
"epoch": 1.7191697191697193,
"grad_norm": 0.3548774719238281,
"learning_rate": 1.4760936825584535e-05,
"loss": 1.6821,
"step": 2200
},
{
"epoch": 1.7191697191697193,
"eval_loss": 1.5890487432479858,
"eval_runtime": 3360.7245,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 2200
},
{
"epoch": 1.7973137973137974,
"grad_norm": 0.37473219633102417,
"learning_rate": 1.4195923436473257e-05,
"loss": 1.685,
"step": 2300
},
{
"epoch": 1.7973137973137974,
"eval_loss": 1.586594820022583,
"eval_runtime": 3365.8879,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 2300
},
{
"epoch": 1.8754578754578755,
"grad_norm": 0.37071430683135986,
"learning_rate": 1.3614150205719086e-05,
"loss": 1.6602,
"step": 2400
},
{
"epoch": 1.8754578754578755,
"eval_loss": 1.5838119983673096,
"eval_runtime": 3363.0708,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 2400
},
{
"epoch": 1.9536019536019538,
"grad_norm": 0.4014996588230133,
"learning_rate": 1.3017940919102943e-05,
"loss": 1.682,
"step": 2500
},
{
"epoch": 1.9536019536019538,
"eval_loss": 1.5820069313049316,
"eval_runtime": 3367.2545,
"eval_samples_per_second": 0.75,
"eval_steps_per_second": 0.094,
"step": 2500
},
{
"epoch": 2.0317460317460316,
"grad_norm": 0.3767947256565094,
"learning_rate": 1.2409677024566145e-05,
"loss": 1.6595,
"step": 2600
},
{
"epoch": 2.0317460317460316,
"eval_loss": 1.5797280073165894,
"eval_runtime": 3364.0566,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 2600
},
{
"epoch": 2.10989010989011,
"grad_norm": 0.3852591812610626,
"learning_rate": 1.1791788119956191e-05,
"loss": 1.6606,
"step": 2700
},
{
"epoch": 2.10989010989011,
"eval_loss": 1.578330636024475,
"eval_runtime": 3359.3638,
"eval_samples_per_second": 0.752,
"eval_steps_per_second": 0.094,
"step": 2700
},
{
"epoch": 2.1880341880341883,
"grad_norm": 0.3954846262931824,
"learning_rate": 1.116674224844664e-05,
"loss": 1.6611,
"step": 2800
},
{
"epoch": 2.1880341880341883,
"eval_loss": 1.5769294500350952,
"eval_runtime": 3365.2078,
"eval_samples_per_second": 0.751,
"eval_steps_per_second": 0.094,
"step": 2800
}
],
"logging_steps": 100,
"max_steps": 5470,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 100,
"total_flos": 1.141344360972288e+18,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}