|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 2.960662525879917, |
|
"eval_steps": 500, |
|
"global_step": 180, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.16563146997929606, |
|
"grad_norm": 1.1045403480529785, |
|
"learning_rate": 5.555555555555556e-05, |
|
"loss": 0.7952, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.33126293995859213, |
|
"grad_norm": 0.3828948438167572, |
|
"learning_rate": 9.996239762521151e-05, |
|
"loss": 0.5763, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.4968944099378882, |
|
"grad_norm": 0.315212219953537, |
|
"learning_rate": 9.865224352899119e-05, |
|
"loss": 0.5143, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.6625258799171843, |
|
"grad_norm": 0.28653863072395325, |
|
"learning_rate": 9.551814704830734e-05, |
|
"loss": 0.4931, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.8281573498964804, |
|
"grad_norm": 0.26458311080932617, |
|
"learning_rate": 9.067760351314838e-05, |
|
"loss": 0.4686, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.9937888198757764, |
|
"grad_norm": 0.25187617540359497, |
|
"learning_rate": 8.43120818934367e-05, |
|
"loss": 0.474, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 1.1490683229813665, |
|
"grad_norm": 0.25587961077690125, |
|
"learning_rate": 7.666022164008457e-05, |
|
"loss": 0.4457, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 1.3146997929606625, |
|
"grad_norm": 0.27590256929397583, |
|
"learning_rate": 6.800888624023553e-05, |
|
"loss": 0.4406, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 1.4803312629399585, |
|
"grad_norm": 0.25824370980262756, |
|
"learning_rate": 5.868240888334653e-05, |
|
"loss": 0.4363, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 1.6459627329192545, |
|
"grad_norm": 0.24047276377677917, |
|
"learning_rate": 4.903043341140879e-05, |
|
"loss": 0.4315, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.8115942028985508, |
|
"grad_norm": 0.2541721761226654, |
|
"learning_rate": 3.941480638852948e-05, |
|
"loss": 0.4356, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 1.9772256728778468, |
|
"grad_norm": 0.26237165927886963, |
|
"learning_rate": 3.019601169804216e-05, |
|
"loss": 0.4263, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 2.132505175983437, |
|
"grad_norm": 0.23833103477954865, |
|
"learning_rate": 2.171965622567308e-05, |
|
"loss": 0.4202, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 2.298136645962733, |
|
"grad_norm": 0.2733284831047058, |
|
"learning_rate": 1.4303513272105057e-05, |
|
"loss": 0.4178, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 2.463768115942029, |
|
"grad_norm": 0.2536739110946655, |
|
"learning_rate": 8.225609429353187e-06, |
|
"loss": 0.4221, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 2.629399585921325, |
|
"grad_norm": 0.22531658411026, |
|
"learning_rate": 3.7138015365554833e-06, |
|
"loss": 0.4202, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 2.795031055900621, |
|
"grad_norm": 0.23673856258392334, |
|
"learning_rate": 9.372344686307655e-07, |
|
"loss": 0.4125, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 2.960662525879917, |
|
"grad_norm": 0.2418087124824524, |
|
"learning_rate": 0.0, |
|
"loss": 0.4152, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 2.960662525879917, |
|
"step": 180, |
|
"total_flos": 4.858579275680317e+18, |
|
"train_loss": 0.46919142405192055, |
|
"train_runtime": 5669.8178, |
|
"train_samples_per_second": 6.132, |
|
"train_steps_per_second": 0.032 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 180, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"stateful_callbacks": { |
|
"TrainerControl": { |
|
"args": { |
|
"should_epoch_stop": false, |
|
"should_evaluate": false, |
|
"should_log": false, |
|
"should_save": true, |
|
"should_training_stop": true |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 4.858579275680317e+18, |
|
"train_batch_size": 6, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|