|
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 2.291937332823844,
|
|
"eval_steps": 500,
|
|
"global_step": 3000,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.07642338555598013,
|
|
"grad_norm": 0.04852772876620293,
|
|
"learning_rate": 0.00019991780772074993,
|
|
"loss": 0.631,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 0.15284677111196027,
|
|
"grad_norm": 0.06019178777933121,
|
|
"learning_rate": 0.00019926107993313918,
|
|
"loss": 0.4967,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 0.2292701566679404,
|
|
"grad_norm": 0.06423385441303253,
|
|
"learning_rate": 0.00019795194081958614,
|
|
"loss": 0.4377,
|
|
"step": 300
|
|
},
|
|
{
|
|
"epoch": 0.30569354222392053,
|
|
"grad_norm": 0.05076967179775238,
|
|
"learning_rate": 0.00019599899493270587,
|
|
"loss": 0.4121,
|
|
"step": 400
|
|
},
|
|
{
|
|
"epoch": 0.38211692777990064,
|
|
"grad_norm": 0.06871291249990463,
|
|
"learning_rate": 0.00019341507836108832,
|
|
"loss": 0.3974,
|
|
"step": 500
|
|
},
|
|
{
|
|
"epoch": 0.4585403133358808,
|
|
"grad_norm": 0.062317609786987305,
|
|
"learning_rate": 0.00019021717436179406,
|
|
"loss": 0.3826,
|
|
"step": 600
|
|
},
|
|
{
|
|
"epoch": 0.5349636988918609,
|
|
"grad_norm": 0.054332610219717026,
|
|
"learning_rate": 0.00018642630173483835,
|
|
"loss": 0.3695,
|
|
"step": 700
|
|
},
|
|
{
|
|
"epoch": 0.6113870844478411,
|
|
"grad_norm": 0.07528349757194519,
|
|
"learning_rate": 0.00018206737667334231,
|
|
"loss": 0.3589,
|
|
"step": 800
|
|
},
|
|
{
|
|
"epoch": 0.6878104700038211,
|
|
"grad_norm": 0.07792173326015472,
|
|
"learning_rate": 0.00017716904899736617,
|
|
"loss": 0.3527,
|
|
"step": 900
|
|
},
|
|
{
|
|
"epoch": 0.7642338555598013,
|
|
"grad_norm": 0.08550075441598892,
|
|
"learning_rate": 0.00017176351384780628,
|
|
"loss": 0.3407,
|
|
"step": 1000
|
|
},
|
|
{
|
|
"epoch": 0.8406572411157814,
|
|
"grad_norm": 0.09895286709070206,
|
|
"learning_rate": 0.00016588630007803324,
|
|
"loss": 0.3292,
|
|
"step": 1100
|
|
},
|
|
{
|
|
"epoch": 0.9170806266717616,
|
|
"grad_norm": 0.05582467094063759,
|
|
"learning_rate": 0.00015957603673410394,
|
|
"loss": 0.321,
|
|
"step": 1200
|
|
},
|
|
{
|
|
"epoch": 0.9935040122277417,
|
|
"grad_norm": 0.05442088469862938,
|
|
"learning_rate": 0.00015287419915839849,
|
|
"loss": 0.3202,
|
|
"step": 1300
|
|
},
|
|
{
|
|
"epoch": 1.0695452808559418,
|
|
"grad_norm": 0.07129843533039093,
|
|
"learning_rate": 0.00014582483638546267,
|
|
"loss": 0.3075,
|
|
"step": 1400
|
|
},
|
|
{
|
|
"epoch": 1.145968666411922,
|
|
"grad_norm": 0.06552577763795853,
|
|
"learning_rate": 0.00013847428162179422,
|
|
"loss": 0.3008,
|
|
"step": 1500
|
|
},
|
|
{
|
|
"epoch": 1.2223920519679021,
|
|
"grad_norm": 0.07303871214389801,
|
|
"learning_rate": 0.00013087084771249831,
|
|
"loss": 0.2948,
|
|
"step": 1600
|
|
},
|
|
{
|
|
"epoch": 1.2988154375238823,
|
|
"grad_norm": 0.12717650830745697,
|
|
"learning_rate": 0.00012306450959641117,
|
|
"loss": 0.2948,
|
|
"step": 1700
|
|
},
|
|
{
|
|
"epoch": 1.3752388230798624,
|
|
"grad_norm": 0.05392363294959068,
|
|
"learning_rate": 0.00011510657583681364,
|
|
"loss": 0.2887,
|
|
"step": 1800
|
|
},
|
|
{
|
|
"epoch": 1.4516622086358426,
|
|
"grad_norm": 0.07771366834640503,
|
|
"learning_rate": 0.00010704935138665758,
|
|
"loss": 0.286,
|
|
"step": 1900
|
|
},
|
|
{
|
|
"epoch": 1.5280855941918228,
|
|
"grad_norm": 0.056173793971538544,
|
|
"learning_rate": 9.894579380484204e-05,
|
|
"loss": 0.2874,
|
|
"step": 2000
|
|
},
|
|
{
|
|
"epoch": 1.604508979747803,
|
|
"grad_norm": 0.05204881355166435,
|
|
"learning_rate": 9.084916518311908e-05,
|
|
"loss": 0.2831,
|
|
"step": 2100
|
|
},
|
|
{
|
|
"epoch": 1.680932365303783,
|
|
"grad_norm": 0.08619089424610138,
|
|
"learning_rate": 8.281268207140247e-05,
|
|
"loss": 0.2813,
|
|
"step": 2200
|
|
},
|
|
{
|
|
"epoch": 1.757355750859763,
|
|
"grad_norm": 0.05422086641192436,
|
|
"learning_rate": 7.488916570240859e-05,
|
|
"loss": 0.2852,
|
|
"step": 2300
|
|
},
|
|
{
|
|
"epoch": 1.8337791364157432,
|
|
"grad_norm": 0.08088803291320801,
|
|
"learning_rate": 6.713069481459069e-05,
|
|
"loss": 0.2788,
|
|
"step": 2400
|
|
},
|
|
{
|
|
"epoch": 1.9102025219717234,
|
|
"grad_norm": 0.06791087239980698,
|
|
"learning_rate": 5.958826335525025e-05,
|
|
"loss": 0.2801,
|
|
"step": 2500
|
|
},
|
|
{
|
|
"epoch": 1.9866259075277035,
|
|
"grad_norm": 0.0687682032585144,
|
|
"learning_rate": 5.231144531363399e-05,
|
|
"loss": 0.2776,
|
|
"step": 2600
|
|
},
|
|
{
|
|
"epoch": 2.062667176155904,
|
|
"grad_norm": 0.05683884769678116,
|
|
"learning_rate": 4.534806888696046e-05,
|
|
"loss": 0.2742,
|
|
"step": 2700
|
|
},
|
|
{
|
|
"epoch": 2.1390905617118836,
|
|
"grad_norm": 0.05977335199713707,
|
|
"learning_rate": 3.8743902120979594e-05,
|
|
"loss": 0.2649,
|
|
"step": 2800
|
|
},
|
|
{
|
|
"epoch": 2.2155139472678638,
|
|
"grad_norm": 0.07473505288362503,
|
|
"learning_rate": 3.254235209124844e-05,
|
|
"loss": 0.2688,
|
|
"step": 2900
|
|
},
|
|
{
|
|
"epoch": 2.291937332823844,
|
|
"grad_norm": 0.0591265894472599,
|
|
"learning_rate": 2.6784179602309123e-05,
|
|
"loss": 0.2683,
|
|
"step": 3000
|
|
}
|
|
],
|
|
"logging_steps": 100,
|
|
"max_steps": 3924,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 3,
|
|
"save_steps": 1000,
|
|
"stateful_callbacks": {
|
|
"TrainerControl": {
|
|
"args": {
|
|
"should_epoch_stop": false,
|
|
"should_evaluate": false,
|
|
"should_log": false,
|
|
"should_save": true,
|
|
"should_training_stop": false
|
|
},
|
|
"attributes": {}
|
|
}
|
|
},
|
|
"total_flos": 2.1009319455872778e+18,
|
|
"train_batch_size": 2,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|
|
|