Qwen2.5-1.5B-Open-R1-Distill / trainer_state.json
flyingbugs's picture
Model save
bb1bde1 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.8357123991549114,
"learning_rate": 1.785714285714286e-05,
"loss": 0.8456,
"step": 5
},
{
"epoch": 0.03731343283582089,
"grad_norm": 0.7888674392020409,
"learning_rate": 3.571428571428572e-05,
"loss": 0.7682,
"step": 10
},
{
"epoch": 0.055970149253731345,
"grad_norm": 0.5088001784298014,
"learning_rate": 4.999827900623038e-05,
"loss": 0.7026,
"step": 15
},
{
"epoch": 0.07462686567164178,
"grad_norm": 0.3967331836313917,
"learning_rate": 4.993807186343243e-05,
"loss": 0.6745,
"step": 20
},
{
"epoch": 0.09328358208955224,
"grad_norm": 0.38953492827269537,
"learning_rate": 4.979207812402531e-05,
"loss": 0.6436,
"step": 25
},
{
"epoch": 0.11194029850746269,
"grad_norm": 0.3056894462367877,
"learning_rate": 4.956085596012407e-05,
"loss": 0.6362,
"step": 30
},
{
"epoch": 0.13059701492537312,
"grad_norm": 0.2848383986273845,
"learning_rate": 4.924528939432311e-05,
"loss": 0.6199,
"step": 35
},
{
"epoch": 0.14925373134328357,
"grad_norm": 0.24821786610124724,
"learning_rate": 4.884658491984735e-05,
"loss": 0.6106,
"step": 40
},
{
"epoch": 0.16791044776119404,
"grad_norm": 0.21630773743864942,
"learning_rate": 4.8366266887814235e-05,
"loss": 0.6112,
"step": 45
},
{
"epoch": 0.1865671641791045,
"grad_norm": 0.2542614286063211,
"learning_rate": 4.780617167924209e-05,
"loss": 0.5939,
"step": 50
},
{
"epoch": 0.20522388059701493,
"grad_norm": 0.273652684856057,
"learning_rate": 4.716844068408693e-05,
"loss": 0.5965,
"step": 55
},
{
"epoch": 0.22388059701492538,
"grad_norm": 0.2919315422944058,
"learning_rate": 4.6455512114150546e-05,
"loss": 0.5919,
"step": 60
},
{
"epoch": 0.24253731343283583,
"grad_norm": 0.4896363894073648,
"learning_rate": 4.5670111681161296e-05,
"loss": 0.5829,
"step": 65
},
{
"epoch": 0.26119402985074625,
"grad_norm": 0.35893234362656307,
"learning_rate": 4.481524217566783e-05,
"loss": 0.5799,
"step": 70
},
{
"epoch": 0.2798507462686567,
"grad_norm": 0.47821339713150723,
"learning_rate": 4.3894171986588217e-05,
"loss": 0.5787,
"step": 75
},
{
"epoch": 0.29850746268656714,
"grad_norm": 0.42779548633021597,
"learning_rate": 4.29104226053073e-05,
"loss": 0.5775,
"step": 80
},
{
"epoch": 0.31716417910447764,
"grad_norm": 0.34493135242732464,
"learning_rate": 4.186775516209732e-05,
"loss": 0.5728,
"step": 85
},
{
"epoch": 0.3358208955223881,
"grad_norm": 0.3450752761118264,
"learning_rate": 4.077015604633669e-05,
"loss": 0.5752,
"step": 90
},
{
"epoch": 0.35447761194029853,
"grad_norm": 0.3254883369412446,
"learning_rate": 3.962182166550441e-05,
"loss": 0.5737,
"step": 95
},
{
"epoch": 0.373134328358209,
"grad_norm": 0.3021937652365121,
"learning_rate": 3.8427142401220634e-05,
"loss": 0.5697,
"step": 100
},
{
"epoch": 0.3917910447761194,
"grad_norm": 0.22543629034027707,
"learning_rate": 3.71906858236735e-05,
"loss": 0.566,
"step": 105
},
{
"epoch": 0.41044776119402987,
"grad_norm": 0.26150626383864994,
"learning_rate": 3.591717922860785e-05,
"loss": 0.5733,
"step": 110
},
{
"epoch": 0.4291044776119403,
"grad_norm": 0.25406069172974854,
"learning_rate": 3.46114915636416e-05,
"loss": 0.5641,
"step": 115
},
{
"epoch": 0.44776119402985076,
"grad_norm": 0.2761171780274853,
"learning_rate": 3.3278614813010034e-05,
"loss": 0.565,
"step": 120
},
{
"epoch": 0.4664179104477612,
"grad_norm": 0.27444350076630936,
"learning_rate": 3.1923644911909e-05,
"loss": 0.5619,
"step": 125
},
{
"epoch": 0.48507462686567165,
"grad_norm": 0.22218188459643629,
"learning_rate": 3.0551762263406576e-05,
"loss": 0.5606,
"step": 130
},
{
"epoch": 0.503731343283582,
"grad_norm": 0.23885769537130422,
"learning_rate": 2.9168211932412042e-05,
"loss": 0.5579,
"step": 135
},
{
"epoch": 0.5223880597014925,
"grad_norm": 0.19581242444906968,
"learning_rate": 2.777828359242567e-05,
"loss": 0.5632,
"step": 140
},
{
"epoch": 0.5410447761194029,
"grad_norm": 0.23903378838505404,
"learning_rate": 2.6387291301738377e-05,
"loss": 0.5559,
"step": 145
},
{
"epoch": 0.5597014925373134,
"grad_norm": 0.22131995247320724,
"learning_rate": 2.50005531864019e-05,
"loss": 0.5537,
"step": 150
},
{
"epoch": 0.5783582089552238,
"grad_norm": 0.20388629120046767,
"learning_rate": 2.362337110764688e-05,
"loss": 0.554,
"step": 155
},
{
"epoch": 0.5970149253731343,
"grad_norm": 0.19865451240052354,
"learning_rate": 2.226101039148557e-05,
"loss": 0.5523,
"step": 160
},
{
"epoch": 0.6156716417910447,
"grad_norm": 0.21792292980419078,
"learning_rate": 2.0918679697998252e-05,
"loss": 0.5511,
"step": 165
},
{
"epoch": 0.6343283582089553,
"grad_norm": 0.20734976300061925,
"learning_rate": 1.9601511107268255e-05,
"loss": 0.5516,
"step": 170
},
{
"epoch": 0.6529850746268657,
"grad_norm": 0.1990824685948915,
"learning_rate": 1.8314540498102216e-05,
"loss": 0.5512,
"step": 175
},
{
"epoch": 0.6716417910447762,
"grad_norm": 0.1759699937325297,
"learning_rate": 1.7062688294552992e-05,
"loss": 0.5433,
"step": 180
},
{
"epoch": 0.6902985074626866,
"grad_norm": 0.19952904233106478,
"learning_rate": 1.5850740653856096e-05,
"loss": 0.5467,
"step": 185
},
{
"epoch": 0.7089552238805971,
"grad_norm": 0.17940456799398388,
"learning_rate": 1.4683331167703218e-05,
"loss": 0.5503,
"step": 190
},
{
"epoch": 0.7276119402985075,
"grad_norm": 0.17212507506796912,
"learning_rate": 1.356492314681356e-05,
"loss": 0.5531,
"step": 195
},
{
"epoch": 0.746268656716418,
"grad_norm": 0.16216686994925128,
"learning_rate": 1.2499792556533716e-05,
"loss": 0.5474,
"step": 200
},
{
"epoch": 0.7649253731343284,
"grad_norm": 0.17719961464188644,
"learning_rate": 1.1492011668707753e-05,
"loss": 0.5449,
"step": 205
},
{
"epoch": 0.7835820895522388,
"grad_norm": 0.17440876918971734,
"learning_rate": 1.0545433492320603e-05,
"loss": 0.5501,
"step": 210
},
{
"epoch": 0.8022388059701493,
"grad_norm": 0.1578415381103267,
"learning_rate": 9.663677042440537e-06,
"loss": 0.5444,
"step": 215
},
{
"epoch": 0.8208955223880597,
"grad_norm": 0.16294954656469435,
"learning_rate": 8.850113503781367e-06,
"loss": 0.5443,
"step": 220
},
{
"epoch": 0.8395522388059702,
"grad_norm": 0.14550383873413048,
"learning_rate": 8.107853341784671e-06,
"loss": 0.5507,
"step": 225
},
{
"epoch": 0.8582089552238806,
"grad_norm": 0.1571896983780636,
"learning_rate": 7.439734410499752e-06,
"loss": 0.547,
"step": 230
},
{
"epoch": 0.8768656716417911,
"grad_norm": 0.1811708441839716,
"learning_rate": 6.848311102728011e-06,
"loss": 0.5472,
"step": 235
},
{
"epoch": 0.8955223880597015,
"grad_norm": 0.1768578492896203,
"learning_rate": 6.335844583913515e-06,
"loss": 0.5433,
"step": 240
},
{
"epoch": 0.914179104477612,
"grad_norm": 0.14375585775141816,
"learning_rate": 5.904294147118193e-06,
"loss": 0.547,
"step": 245
},
{
"epoch": 0.9328358208955224,
"grad_norm": 0.14507740115855672,
"learning_rate": 5.555309722133842e-06,
"loss": 0.5436,
"step": 250
},
{
"epoch": 0.9514925373134329,
"grad_norm": 0.14189135037515943,
"learning_rate": 5.290225567370509e-06,
"loss": 0.5396,
"step": 255
},
{
"epoch": 0.9701492537313433,
"grad_norm": 0.15104419515678771,
"learning_rate": 5.110055168638854e-06,
"loss": 0.5433,
"step": 260
},
{
"epoch": 0.9888059701492538,
"grad_norm": 0.14422216331517992,
"learning_rate": 5.0154873643297575e-06,
"loss": 0.547,
"step": 265
},
{
"epoch": 1.0,
"step": 268,
"total_flos": 487709642588160.0,
"train_loss": 0.5792717831348305,
"train_runtime": 20654.3187,
"train_samples_per_second": 1.661,
"train_steps_per_second": 0.013
}
],
"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": 487709642588160.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}