Qwen2.5-1.5B-Open-R1-Distill / trainer_state.json
Rano23's picture
Model save
0e6ade2 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.8355981820926557,
"learning_rate": 1.785714285714286e-05,
"loss": 0.8456,
"step": 5
},
{
"epoch": 0.03731343283582089,
"grad_norm": 0.7819205180343736,
"learning_rate": 3.571428571428572e-05,
"loss": 0.7682,
"step": 10
},
{
"epoch": 0.055970149253731345,
"grad_norm": 0.5028382045807719,
"learning_rate": 4.999827900623038e-05,
"loss": 0.7026,
"step": 15
},
{
"epoch": 0.07462686567164178,
"grad_norm": 0.38189058158585737,
"learning_rate": 4.993807186343243e-05,
"loss": 0.6746,
"step": 20
},
{
"epoch": 0.09328358208955224,
"grad_norm": 0.3554256422824799,
"learning_rate": 4.979207812402531e-05,
"loss": 0.6436,
"step": 25
},
{
"epoch": 0.11194029850746269,
"grad_norm": 0.26680075742409215,
"learning_rate": 4.956085596012407e-05,
"loss": 0.6363,
"step": 30
},
{
"epoch": 0.13059701492537312,
"grad_norm": 0.24138373910929908,
"learning_rate": 4.924528939432311e-05,
"loss": 0.6199,
"step": 35
},
{
"epoch": 0.14925373134328357,
"grad_norm": 0.23298141204427875,
"learning_rate": 4.884658491984735e-05,
"loss": 0.6106,
"step": 40
},
{
"epoch": 0.16791044776119404,
"grad_norm": 0.20744698870861822,
"learning_rate": 4.8366266887814235e-05,
"loss": 0.6112,
"step": 45
},
{
"epoch": 0.1865671641791045,
"grad_norm": 0.22238620533604506,
"learning_rate": 4.780617167924209e-05,
"loss": 0.5938,
"step": 50
},
{
"epoch": 0.20522388059701493,
"grad_norm": 0.2276132160605588,
"learning_rate": 4.716844068408693e-05,
"loss": 0.5964,
"step": 55
},
{
"epoch": 0.22388059701492538,
"grad_norm": 0.24338894179158121,
"learning_rate": 4.6455512114150546e-05,
"loss": 0.5918,
"step": 60
},
{
"epoch": 0.24253731343283583,
"grad_norm": 0.49788551357919736,
"learning_rate": 4.5670111681161296e-05,
"loss": 0.5829,
"step": 65
},
{
"epoch": 0.26119402985074625,
"grad_norm": 0.393010996842019,
"learning_rate": 4.481524217566783e-05,
"loss": 0.5797,
"step": 70
},
{
"epoch": 0.2798507462686567,
"grad_norm": 0.42897779811059983,
"learning_rate": 4.3894171986588217e-05,
"loss": 0.5785,
"step": 75
},
{
"epoch": 0.29850746268656714,
"grad_norm": 0.38805947922007195,
"learning_rate": 4.29104226053073e-05,
"loss": 0.5773,
"step": 80
},
{
"epoch": 0.31716417910447764,
"grad_norm": 0.2926662624946126,
"learning_rate": 4.186775516209732e-05,
"loss": 0.5728,
"step": 85
},
{
"epoch": 0.3358208955223881,
"grad_norm": 0.3077872574971531,
"learning_rate": 4.077015604633669e-05,
"loss": 0.5752,
"step": 90
},
{
"epoch": 0.35447761194029853,
"grad_norm": 0.28837233371551474,
"learning_rate": 3.962182166550441e-05,
"loss": 0.5736,
"step": 95
},
{
"epoch": 0.373134328358209,
"grad_norm": 0.2755000734486141,
"learning_rate": 3.8427142401220634e-05,
"loss": 0.5696,
"step": 100
},
{
"epoch": 0.3917910447761194,
"grad_norm": 0.2454481524983729,
"learning_rate": 3.71906858236735e-05,
"loss": 0.5659,
"step": 105
},
{
"epoch": 0.41044776119402987,
"grad_norm": 0.22976578272395923,
"learning_rate": 3.591717922860785e-05,
"loss": 0.5731,
"step": 110
},
{
"epoch": 0.4291044776119403,
"grad_norm": 0.20945706948866985,
"learning_rate": 3.46114915636416e-05,
"loss": 0.5639,
"step": 115
},
{
"epoch": 0.44776119402985076,
"grad_norm": 0.21707312317165936,
"learning_rate": 3.3278614813010034e-05,
"loss": 0.5649,
"step": 120
},
{
"epoch": 0.4664179104477612,
"grad_norm": 0.2161425231604288,
"learning_rate": 3.1923644911909e-05,
"loss": 0.5618,
"step": 125
},
{
"epoch": 0.48507462686567165,
"grad_norm": 0.2119187448421958,
"learning_rate": 3.0551762263406576e-05,
"loss": 0.5605,
"step": 130
},
{
"epoch": 0.503731343283582,
"grad_norm": 0.2072839164242438,
"learning_rate": 2.9168211932412042e-05,
"loss": 0.5578,
"step": 135
},
{
"epoch": 0.5223880597014925,
"grad_norm": 0.1945962599413367,
"learning_rate": 2.777828359242567e-05,
"loss": 0.5632,
"step": 140
},
{
"epoch": 0.5410447761194029,
"grad_norm": 0.23118838276778292,
"learning_rate": 2.6387291301738377e-05,
"loss": 0.5559,
"step": 145
},
{
"epoch": 0.5597014925373134,
"grad_norm": 0.19803620006104744,
"learning_rate": 2.50005531864019e-05,
"loss": 0.5537,
"step": 150
},
{
"epoch": 0.5783582089552238,
"grad_norm": 0.19045030832075327,
"learning_rate": 2.362337110764688e-05,
"loss": 0.554,
"step": 155
},
{
"epoch": 0.5970149253731343,
"grad_norm": 0.1997848852674072,
"learning_rate": 2.226101039148557e-05,
"loss": 0.5521,
"step": 160
},
{
"epoch": 0.6156716417910447,
"grad_norm": 0.2106306143571984,
"learning_rate": 2.0918679697998252e-05,
"loss": 0.551,
"step": 165
},
{
"epoch": 0.6343283582089553,
"grad_norm": 0.19304692647798125,
"learning_rate": 1.9601511107268255e-05,
"loss": 0.5514,
"step": 170
},
{
"epoch": 0.6529850746268657,
"grad_norm": 0.18776569092787673,
"learning_rate": 1.8314540498102216e-05,
"loss": 0.551,
"step": 175
},
{
"epoch": 0.6716417910447762,
"grad_norm": 0.16444794445801264,
"learning_rate": 1.7062688294552992e-05,
"loss": 0.5432,
"step": 180
},
{
"epoch": 0.6902985074626866,
"grad_norm": 0.1821192450738552,
"learning_rate": 1.5850740653856096e-05,
"loss": 0.5465,
"step": 185
},
{
"epoch": 0.7089552238805971,
"grad_norm": 0.16488410850720042,
"learning_rate": 1.4683331167703218e-05,
"loss": 0.5501,
"step": 190
},
{
"epoch": 0.7276119402985075,
"grad_norm": 0.16409414248406526,
"learning_rate": 1.356492314681356e-05,
"loss": 0.5529,
"step": 195
},
{
"epoch": 0.746268656716418,
"grad_norm": 0.15257643452142564,
"learning_rate": 1.2499792556533716e-05,
"loss": 0.5473,
"step": 200
},
{
"epoch": 0.7649253731343284,
"grad_norm": 0.16748658501135685,
"learning_rate": 1.1492011668707753e-05,
"loss": 0.5448,
"step": 205
},
{
"epoch": 0.7835820895522388,
"grad_norm": 0.16228397016239318,
"learning_rate": 1.0545433492320603e-05,
"loss": 0.5499,
"step": 210
},
{
"epoch": 0.8022388059701493,
"grad_norm": 0.15112912823821292,
"learning_rate": 9.663677042440537e-06,
"loss": 0.5443,
"step": 215
},
{
"epoch": 0.8208955223880597,
"grad_norm": 0.15495931429665755,
"learning_rate": 8.850113503781367e-06,
"loss": 0.5442,
"step": 220
},
{
"epoch": 0.8395522388059702,
"grad_norm": 0.13761158555242786,
"learning_rate": 8.107853341784671e-06,
"loss": 0.5505,
"step": 225
},
{
"epoch": 0.8582089552238806,
"grad_norm": 0.14992222321390744,
"learning_rate": 7.439734410499752e-06,
"loss": 0.5469,
"step": 230
},
{
"epoch": 0.8768656716417911,
"grad_norm": 0.16741403717291473,
"learning_rate": 6.848311102728011e-06,
"loss": 0.547,
"step": 235
},
{
"epoch": 0.8955223880597015,
"grad_norm": 0.16841892108770962,
"learning_rate": 6.335844583913515e-06,
"loss": 0.5432,
"step": 240
},
{
"epoch": 0.914179104477612,
"grad_norm": 0.1376017090301476,
"learning_rate": 5.904294147118193e-06,
"loss": 0.547,
"step": 245
},
{
"epoch": 0.9328358208955224,
"grad_norm": 0.14289320412837506,
"learning_rate": 5.555309722133842e-06,
"loss": 0.5434,
"step": 250
},
{
"epoch": 0.9514925373134329,
"grad_norm": 0.136012415879584,
"learning_rate": 5.290225567370509e-06,
"loss": 0.5395,
"step": 255
},
{
"epoch": 0.9701492537313433,
"grad_norm": 0.14123958244150933,
"learning_rate": 5.110055168638854e-06,
"loss": 0.5432,
"step": 260
},
{
"epoch": 0.9888059701492538,
"grad_norm": 0.13723916512843948,
"learning_rate": 5.0154873643297575e-06,
"loss": 0.5469,
"step": 265
},
{
"epoch": 1.0,
"step": 268,
"total_flos": 488621249396736.0,
"train_loss": 0.5792123831030148,
"train_runtime": 3418.6445,
"train_samples_per_second": 10.034,
"train_steps_per_second": 0.078
}
],
"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
}