Qwen2.5-7B-dpo-2k / trainer_state.json
AmberYifan's picture
Model save
d5a99e6 verified
raw
history blame
4.45 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.992,
"eval_steps": 200,
"global_step": 62,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.016,
"grad_norm": 54.80237307470059,
"learning_rate": 7.142857142857142e-08,
"logits/generated": -1.112015962600708,
"logits/real": -0.9725905656814575,
"logps/generated": -192.91046142578125,
"logps/real": -197.25250244140625,
"loss": 0.9186,
"rewards/accuracies": 0.0,
"rewards/generated": 0.0,
"rewards/margins": 0.0,
"rewards/real": 0.0,
"step": 1
},
{
"epoch": 0.16,
"grad_norm": 627.6270610217415,
"learning_rate": 4.727272727272727e-07,
"logits/generated": -0.9520471096038818,
"logits/real": -0.920363187789917,
"logps/generated": -202.67523193359375,
"logps/real": -215.9678497314453,
"loss": 0.8982,
"rewards/accuracies": 0.4861111044883728,
"rewards/generated": 0.18911296129226685,
"rewards/margins": 0.008780588395893574,
"rewards/real": 0.19789355993270874,
"step": 10
},
{
"epoch": 0.32,
"grad_norm": 50.96666235352248,
"learning_rate": 3.818181818181818e-07,
"logits/generated": -1.0125733613967896,
"logits/real": -0.9775249361991882,
"logps/generated": -208.7979278564453,
"logps/real": -219.83773803710938,
"loss": 0.8919,
"rewards/accuracies": 0.625,
"rewards/generated": 0.7978582382202148,
"rewards/margins": 0.03471279889345169,
"rewards/real": 0.8325710296630859,
"step": 20
},
{
"epoch": 0.48,
"grad_norm": 46.130447914310665,
"learning_rate": 2.909090909090909e-07,
"logits/generated": -1.005338191986084,
"logits/real": -0.8279012441635132,
"logps/generated": -176.2639617919922,
"logps/real": -183.01602172851562,
"loss": 0.8324,
"rewards/accuracies": 0.675000011920929,
"rewards/generated": 1.5164999961853027,
"rewards/margins": 0.1330079436302185,
"rewards/real": 1.6495081186294556,
"step": 30
},
{
"epoch": 0.64,
"grad_norm": 51.04951332078808,
"learning_rate": 2e-07,
"logits/generated": -1.0122597217559814,
"logits/real": -0.854594349861145,
"logps/generated": -179.2203826904297,
"logps/real": -211.2075958251953,
"loss": 0.8282,
"rewards/accuracies": 0.675000011920929,
"rewards/generated": 2.0012855529785156,
"rewards/margins": 0.24582402408123016,
"rewards/real": 2.2471091747283936,
"step": 40
},
{
"epoch": 0.8,
"grad_norm": 46.26306651530627,
"learning_rate": 1.0909090909090908e-07,
"logits/generated": -1.0062984228134155,
"logits/real": -0.7997950315475464,
"logps/generated": -180.25086975097656,
"logps/real": -208.59616088867188,
"loss": 0.823,
"rewards/accuracies": 0.625,
"rewards/generated": 2.424903154373169,
"rewards/margins": 0.1960751712322235,
"rewards/real": 2.620978355407715,
"step": 50
},
{
"epoch": 0.96,
"grad_norm": 44.84748297178086,
"learning_rate": 1.818181818181818e-08,
"logits/generated": -1.0795314311981201,
"logits/real": -0.9678686857223511,
"logps/generated": -172.9720001220703,
"logps/real": -177.3105926513672,
"loss": 0.8174,
"rewards/accuracies": 0.637499988079071,
"rewards/generated": 2.505340099334717,
"rewards/margins": 0.2063644826412201,
"rewards/real": 2.711704730987549,
"step": 60
},
{
"epoch": 0.992,
"step": 62,
"total_flos": 0.0,
"train_loss": 0.8468392750909252,
"train_runtime": 825.6155,
"train_samples_per_second": 2.413,
"train_steps_per_second": 0.075
}
],
"logging_steps": 10,
"max_steps": 62,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 200,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}