Qwen2.5-1.5B-Open-R1-Distill / trainer_state.json
AggarwalAnkit's picture
Model save
63b1717 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 100,
"global_step": 169,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.029585798816568046,
"grad_norm": 2.3443965310318395,
"learning_rate": 5.882352941176471e-06,
"loss": 1.0993,
"step": 5
},
{
"epoch": 0.05917159763313609,
"grad_norm": 1.636263409121339,
"learning_rate": 1.1764705882352942e-05,
"loss": 1.0401,
"step": 10
},
{
"epoch": 0.08875739644970414,
"grad_norm": 0.8972296280656706,
"learning_rate": 1.7647058823529414e-05,
"loss": 0.9534,
"step": 15
},
{
"epoch": 0.11834319526627218,
"grad_norm": 0.6127327175054608,
"learning_rate": 1.9980782984658682e-05,
"loss": 0.8922,
"step": 20
},
{
"epoch": 0.14792899408284024,
"grad_norm": 0.548045389840513,
"learning_rate": 1.9863613034027224e-05,
"loss": 0.8551,
"step": 25
},
{
"epoch": 0.17751479289940827,
"grad_norm": 0.4353812948230317,
"learning_rate": 1.9641197940012136e-05,
"loss": 0.8282,
"step": 30
},
{
"epoch": 0.20710059171597633,
"grad_norm": 0.3859967015713015,
"learning_rate": 1.9315910880512792e-05,
"loss": 0.8229,
"step": 35
},
{
"epoch": 0.23668639053254437,
"grad_norm": 0.4091494861330922,
"learning_rate": 1.8891222681391853e-05,
"loss": 0.8224,
"step": 40
},
{
"epoch": 0.26627218934911245,
"grad_norm": 0.3533354972227181,
"learning_rate": 1.8371664782625287e-05,
"loss": 0.8073,
"step": 45
},
{
"epoch": 0.2958579881656805,
"grad_norm": 0.34460234770269144,
"learning_rate": 1.7762780887657576e-05,
"loss": 0.7977,
"step": 50
},
{
"epoch": 0.3254437869822485,
"grad_norm": 0.38463501256015875,
"learning_rate": 1.7071067811865477e-05,
"loss": 0.7877,
"step": 55
},
{
"epoch": 0.35502958579881655,
"grad_norm": 0.3851597019414931,
"learning_rate": 1.6303906161279554e-05,
"loss": 0.7982,
"step": 60
},
{
"epoch": 0.38461538461538464,
"grad_norm": 0.35155649922397253,
"learning_rate": 1.5469481581224274e-05,
"loss": 0.7722,
"step": 65
},
{
"epoch": 0.41420118343195267,
"grad_norm": 0.3563209304455789,
"learning_rate": 1.4576697415156818e-05,
"loss": 0.7741,
"step": 70
},
{
"epoch": 0.4437869822485207,
"grad_norm": 0.34462046903935517,
"learning_rate": 1.3635079705638298e-05,
"loss": 0.7852,
"step": 75
},
{
"epoch": 0.47337278106508873,
"grad_norm": 0.3579907623029016,
"learning_rate": 1.2654675551080724e-05,
"loss": 0.7666,
"step": 80
},
{
"epoch": 0.5029585798816568,
"grad_norm": 0.38314828122342875,
"learning_rate": 1.164594590280734e-05,
"loss": 0.7706,
"step": 85
},
{
"epoch": 0.5325443786982249,
"grad_norm": 0.36676887475259434,
"learning_rate": 1.0619653946285948e-05,
"loss": 0.7701,
"step": 90
},
{
"epoch": 0.5621301775147929,
"grad_norm": 0.38552080123996796,
"learning_rate": 9.586750257511868e-06,
"loss": 0.7772,
"step": 95
},
{
"epoch": 0.591715976331361,
"grad_norm": 0.3469107480877405,
"learning_rate": 8.558255959926533e-06,
"loss": 0.753,
"step": 100
},
{
"epoch": 0.591715976331361,
"eval_loss": 0.7880496978759766,
"eval_runtime": 1.669,
"eval_samples_per_second": 76.692,
"eval_steps_per_second": 2.397,
"step": 100
},
{
"epoch": 0.621301775147929,
"grad_norm": 0.34173840549923107,
"learning_rate": 7.545145128592009e-06,
"loss": 0.757,
"step": 105
},
{
"epoch": 0.650887573964497,
"grad_norm": 0.3542603327303965,
"learning_rate": 6.558227696373617e-06,
"loss": 0.7681,
"step": 110
},
{
"epoch": 0.6804733727810651,
"grad_norm": 0.31554387020683655,
"learning_rate": 5.608034111526298e-06,
"loss": 0.7624,
"step": 115
},
{
"epoch": 0.7100591715976331,
"grad_norm": 0.30172395532928425,
"learning_rate": 4.704702977392914e-06,
"loss": 0.7514,
"step": 120
},
{
"epoch": 0.7396449704142012,
"grad_norm": 0.35444141011660624,
"learning_rate": 3.857872873103322e-06,
"loss": 0.7539,
"step": 125
},
{
"epoch": 0.7692307692307693,
"grad_norm": 0.3199997185213819,
"learning_rate": 3.0765795095517026e-06,
"loss": 0.7555,
"step": 130
},
{
"epoch": 0.7988165680473372,
"grad_norm": 0.3293673552765145,
"learning_rate": 2.369159318001937e-06,
"loss": 0.7585,
"step": 135
},
{
"epoch": 0.8284023668639053,
"grad_norm": 0.30541127430395704,
"learning_rate": 1.743160500034443e-06,
"loss": 0.7498,
"step": 140
},
{
"epoch": 0.8579881656804734,
"grad_norm": 0.3196994335551187,
"learning_rate": 1.2052624879351105e-06,
"loss": 0.7566,
"step": 145
},
{
"epoch": 0.8875739644970414,
"grad_norm": 0.30216600713724134,
"learning_rate": 7.612046748871327e-07,
"loss": 0.7667,
"step": 150
},
{
"epoch": 0.9171597633136095,
"grad_norm": 0.30343707555784966,
"learning_rate": 4.1572517541747294e-07,
"loss": 0.7614,
"step": 155
},
{
"epoch": 0.9467455621301775,
"grad_norm": 0.2913107753392487,
"learning_rate": 1.7251026952640583e-07,
"loss": 0.7607,
"step": 160
},
{
"epoch": 0.9763313609467456,
"grad_norm": 0.28187999370417227,
"learning_rate": 3.4155069933301535e-08,
"loss": 0.7446,
"step": 165
},
{
"epoch": 1.0,
"step": 169,
"total_flos": 76916824473600.0,
"train_loss": 0.8026797080181054,
"train_runtime": 1042.2023,
"train_samples_per_second": 20.735,
"train_steps_per_second": 0.162
}
],
"logging_steps": 5,
"max_steps": 169,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 76916824473600.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}