Qwen2.5-1.5B-Open-R1-Distill / trainer_state.json
PrismXu's picture
Model save
d2ef4e5 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 268,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.018656716417910446,
"grad_norm": 1.8349594315547968,
"learning_rate": 1.785714285714286e-05,
"loss": 0.8456,
"step": 5
},
{
"epoch": 0.03731343283582089,
"grad_norm": 0.782634121444219,
"learning_rate": 3.571428571428572e-05,
"loss": 0.7682,
"step": 10
},
{
"epoch": 0.055970149253731345,
"grad_norm": 0.5018670527820713,
"learning_rate": 4.999827900623038e-05,
"loss": 0.7026,
"step": 15
},
{
"epoch": 0.07462686567164178,
"grad_norm": 0.3828243717102519,
"learning_rate": 4.993807186343243e-05,
"loss": 0.6746,
"step": 20
},
{
"epoch": 0.09328358208955224,
"grad_norm": 0.35587545892731653,
"learning_rate": 4.979207812402531e-05,
"loss": 0.6436,
"step": 25
},
{
"epoch": 0.11194029850746269,
"grad_norm": 0.27236143821993597,
"learning_rate": 4.956085596012407e-05,
"loss": 0.6363,
"step": 30
},
{
"epoch": 0.13059701492537312,
"grad_norm": 0.244361998294939,
"learning_rate": 4.924528939432311e-05,
"loss": 0.6199,
"step": 35
},
{
"epoch": 0.14925373134328357,
"grad_norm": 0.2352972939407857,
"learning_rate": 4.884658491984735e-05,
"loss": 0.6106,
"step": 40
},
{
"epoch": 0.16791044776119404,
"grad_norm": 0.20430970659123662,
"learning_rate": 4.8366266887814235e-05,
"loss": 0.6112,
"step": 45
},
{
"epoch": 0.1865671641791045,
"grad_norm": 0.22885994080124342,
"learning_rate": 4.780617167924209e-05,
"loss": 0.5938,
"step": 50
},
{
"epoch": 0.20522388059701493,
"grad_norm": 0.2316567095700394,
"learning_rate": 4.716844068408693e-05,
"loss": 0.5965,
"step": 55
},
{
"epoch": 0.22388059701492538,
"grad_norm": 0.24259188670212606,
"learning_rate": 4.6455512114150546e-05,
"loss": 0.5918,
"step": 60
},
{
"epoch": 0.24253731343283583,
"grad_norm": 0.4672987834548283,
"learning_rate": 4.5670111681161296e-05,
"loss": 0.5827,
"step": 65
},
{
"epoch": 0.26119402985074625,
"grad_norm": 0.34703822530692985,
"learning_rate": 4.481524217566783e-05,
"loss": 0.5796,
"step": 70
},
{
"epoch": 0.2798507462686567,
"grad_norm": 0.454248199319451,
"learning_rate": 4.3894171986588217e-05,
"loss": 0.5785,
"step": 75
},
{
"epoch": 0.29850746268656714,
"grad_norm": 0.3532889581573952,
"learning_rate": 4.29104226053073e-05,
"loss": 0.5772,
"step": 80
},
{
"epoch": 0.31716417910447764,
"grad_norm": 0.2741268834576412,
"learning_rate": 4.186775516209732e-05,
"loss": 0.5727,
"step": 85
},
{
"epoch": 0.3358208955223881,
"grad_norm": 0.30913456448335347,
"learning_rate": 4.077015604633669e-05,
"loss": 0.5752,
"step": 90
},
{
"epoch": 0.35447761194029853,
"grad_norm": 0.3178209548506241,
"learning_rate": 3.962182166550441e-05,
"loss": 0.5736,
"step": 95
},
{
"epoch": 0.373134328358209,
"grad_norm": 0.2835035495635748,
"learning_rate": 3.8427142401220634e-05,
"loss": 0.5695,
"step": 100
},
{
"epoch": 0.3917910447761194,
"grad_norm": 0.21964235563572918,
"learning_rate": 3.71906858236735e-05,
"loss": 0.5659,
"step": 105
},
{
"epoch": 0.41044776119402987,
"grad_norm": 0.2345510668840131,
"learning_rate": 3.591717922860785e-05,
"loss": 0.5731,
"step": 110
},
{
"epoch": 0.4291044776119403,
"grad_norm": 0.22112443876109653,
"learning_rate": 3.46114915636416e-05,
"loss": 0.564,
"step": 115
},
{
"epoch": 0.44776119402985076,
"grad_norm": 0.20253061555942384,
"learning_rate": 3.3278614813010034e-05,
"loss": 0.5649,
"step": 120
},
{
"epoch": 0.4664179104477612,
"grad_norm": 0.22309759687545724,
"learning_rate": 3.1923644911909e-05,
"loss": 0.5618,
"step": 125
},
{
"epoch": 0.48507462686567165,
"grad_norm": 0.20874782850901671,
"learning_rate": 3.0551762263406576e-05,
"loss": 0.5604,
"step": 130
},
{
"epoch": 0.503731343283582,
"grad_norm": 0.21464706866941952,
"learning_rate": 2.9168211932412042e-05,
"loss": 0.5577,
"step": 135
},
{
"epoch": 0.5223880597014925,
"grad_norm": 0.19611581523563187,
"learning_rate": 2.777828359242567e-05,
"loss": 0.5631,
"step": 140
},
{
"epoch": 0.5410447761194029,
"grad_norm": 0.2261324456297022,
"learning_rate": 2.6387291301738377e-05,
"loss": 0.5557,
"step": 145
},
{
"epoch": 0.5597014925373134,
"grad_norm": 0.19704521098901392,
"learning_rate": 2.50005531864019e-05,
"loss": 0.5536,
"step": 150
},
{
"epoch": 0.5783582089552238,
"grad_norm": 0.1835561626961263,
"learning_rate": 2.362337110764688e-05,
"loss": 0.5539,
"step": 155
},
{
"epoch": 0.5970149253731343,
"grad_norm": 0.19724667606725876,
"learning_rate": 2.226101039148557e-05,
"loss": 0.5521,
"step": 160
},
{
"epoch": 0.6156716417910447,
"grad_norm": 0.2060188374035982,
"learning_rate": 2.0918679697998252e-05,
"loss": 0.5509,
"step": 165
},
{
"epoch": 0.6343283582089553,
"grad_norm": 0.19026425790845444,
"learning_rate": 1.9601511107268255e-05,
"loss": 0.5514,
"step": 170
},
{
"epoch": 0.6529850746268657,
"grad_norm": 0.18483764400787342,
"learning_rate": 1.8314540498102216e-05,
"loss": 0.551,
"step": 175
},
{
"epoch": 0.6716417910447762,
"grad_norm": 0.1648214313425333,
"learning_rate": 1.7062688294552992e-05,
"loss": 0.5431,
"step": 180
},
{
"epoch": 0.6902985074626866,
"grad_norm": 0.18674644446841088,
"learning_rate": 1.5850740653856096e-05,
"loss": 0.5465,
"step": 185
},
{
"epoch": 0.7089552238805971,
"grad_norm": 0.16703240301177977,
"learning_rate": 1.4683331167703218e-05,
"loss": 0.5501,
"step": 190
},
{
"epoch": 0.7276119402985075,
"grad_norm": 0.16383969711245713,
"learning_rate": 1.356492314681356e-05,
"loss": 0.5528,
"step": 195
},
{
"epoch": 0.746268656716418,
"grad_norm": 0.15304132098607015,
"learning_rate": 1.2499792556533716e-05,
"loss": 0.5472,
"step": 200
},
{
"epoch": 0.7649253731343284,
"grad_norm": 0.1677131828200418,
"learning_rate": 1.1492011668707753e-05,
"loss": 0.5447,
"step": 205
},
{
"epoch": 0.7835820895522388,
"grad_norm": 0.16485103710830218,
"learning_rate": 1.0545433492320603e-05,
"loss": 0.5499,
"step": 210
},
{
"epoch": 0.8022388059701493,
"grad_norm": 0.1525388962626507,
"learning_rate": 9.663677042440537e-06,
"loss": 0.5443,
"step": 215
},
{
"epoch": 0.8208955223880597,
"grad_norm": 0.15621002592594435,
"learning_rate": 8.850113503781367e-06,
"loss": 0.5441,
"step": 220
},
{
"epoch": 0.8395522388059702,
"grad_norm": 0.13793420371358125,
"learning_rate": 8.107853341784671e-06,
"loss": 0.5504,
"step": 225
},
{
"epoch": 0.8582089552238806,
"grad_norm": 0.14914444954968403,
"learning_rate": 7.439734410499752e-06,
"loss": 0.5469,
"step": 230
},
{
"epoch": 0.8768656716417911,
"grad_norm": 0.17083770106883414,
"learning_rate": 6.848311102728011e-06,
"loss": 0.547,
"step": 235
},
{
"epoch": 0.8955223880597015,
"grad_norm": 0.17270380122448753,
"learning_rate": 6.335844583913515e-06,
"loss": 0.5432,
"step": 240
},
{
"epoch": 0.914179104477612,
"grad_norm": 0.1382488878975219,
"learning_rate": 5.904294147118193e-06,
"loss": 0.5469,
"step": 245
},
{
"epoch": 0.9328358208955224,
"grad_norm": 0.1424405593133097,
"learning_rate": 5.555309722133842e-06,
"loss": 0.5433,
"step": 250
},
{
"epoch": 0.9514925373134329,
"grad_norm": 0.13668151767554876,
"learning_rate": 5.290225567370509e-06,
"loss": 0.5394,
"step": 255
},
{
"epoch": 0.9701492537313433,
"grad_norm": 0.1417515581555094,
"learning_rate": 5.110055168638854e-06,
"loss": 0.5431,
"step": 260
},
{
"epoch": 0.9888059701492538,
"grad_norm": 0.13736660142133392,
"learning_rate": 5.0154873643297575e-06,
"loss": 0.5468,
"step": 265
},
{
"epoch": 1.0,
"step": 268,
"total_flos": 488621249396736.0,
"train_loss": 0.0,
"train_runtime": 2.2105,
"train_samples_per_second": 15517.054,
"train_steps_per_second": 121.238
}
],
"logging_steps": 5,
"max_steps": 268,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 488621249396736.0,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}