DeepSeek-R1-Distill-Qwen-14B_lora / trainer_state.json
wf8888884's picture
Upload folder using huggingface_hub
f0934ce verified
raw
history blame
4.11 kB
{
"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
}