two_agent_2_dpo_iter_2 / trainer_state.json
YYYYYYibo's picture
Model save
5d7e1dd verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9945,
"eval_steps": 500,
"global_step": 153,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 3.125e-08,
"logits/chosen": -2.004561424255371,
"logits/rejected": -2.004561424255371,
"logps/chosen": -230.59202575683594,
"logps/rejected": -230.59202575683594,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.07,
"learning_rate": 3.1249999999999997e-07,
"logits/chosen": -1.9280829429626465,
"logits/rejected": -1.9280829429626465,
"logps/chosen": -184.71426391601562,
"logps/rejected": -184.71426391601562,
"loss": 0.6931,
"rewards/accuracies": 0.15811966359615326,
"rewards/chosen": -0.0006816095556132495,
"rewards/margins": 6.696617482759848e-09,
"rewards/rejected": -0.0006816161912865937,
"step": 10
},
{
"epoch": 0.13,
"learning_rate": 4.989490450759331e-07,
"logits/chosen": -1.9418092966079712,
"logits/rejected": -1.9418092966079712,
"logps/chosen": -197.00942993164062,
"logps/rejected": -197.00942993164062,
"loss": 0.6931,
"rewards/accuracies": 0.15000000596046448,
"rewards/chosen": -0.02568601444363594,
"rewards/margins": -1.1841985170235603e-08,
"rewards/rejected": -0.025686005130410194,
"step": 20
},
{
"epoch": 0.2,
"learning_rate": 4.872270441827174e-07,
"logits/chosen": -1.768611192703247,
"logits/rejected": -1.768611192703247,
"logps/chosen": -218.05352783203125,
"logps/rejected": -218.0535125732422,
"loss": 0.6931,
"rewards/accuracies": 0.13461539149284363,
"rewards/chosen": -0.2483261078596115,
"rewards/margins": -2.0081643015146255e-09,
"rewards/rejected": -0.24832607805728912,
"step": 30
},
{
"epoch": 0.26,
"learning_rate": 4.6308512113530063e-07,
"logits/chosen": -1.6662472486495972,
"logits/rejected": -1.6662472486495972,
"logps/chosen": -275.301513671875,
"logps/rejected": -275.301513671875,
"loss": 0.6931,
"rewards/accuracies": 0.14615385234355927,
"rewards/chosen": -0.9216616153717041,
"rewards/margins": 1.6548885595213392e-09,
"rewards/rejected": -0.9216616153717041,
"step": 40
},
{
"epoch": 0.33,
"learning_rate": 4.277872161641681e-07,
"logits/chosen": -1.4318017959594727,
"logits/rejected": -1.4318017959594727,
"logps/chosen": -325.6317443847656,
"logps/rejected": -325.6317443847656,
"loss": 0.6931,
"rewards/accuracies": 0.15000000596046448,
"rewards/chosen": -1.4385263919830322,
"rewards/margins": 6.762834736662171e-09,
"rewards/rejected": -1.4385263919830322,
"step": 50
},
{
"epoch": 0.39,
"learning_rate": 3.8318133624280046e-07,
"logits/chosen": -1.1741256713867188,
"logits/rejected": -1.1741256713867188,
"logps/chosen": -389.4248352050781,
"logps/rejected": -389.4248352050781,
"loss": 0.6931,
"rewards/accuracies": 0.14230769872665405,
"rewards/chosen": -1.963220238685608,
"rewards/margins": 4.241099915702762e-09,
"rewards/rejected": -1.963220238685608,
"step": 60
},
{
"epoch": 0.46,
"learning_rate": 3.316028034595861e-07,
"logits/chosen": -1.0726745128631592,
"logits/rejected": -1.0726745128631592,
"logps/chosen": -422.3594665527344,
"logps/rejected": -422.3594665527344,
"loss": 0.6931,
"rewards/accuracies": 0.16538462042808533,
"rewards/chosen": -2.3579981327056885,
"rewards/margins": 4.58497284583359e-09,
"rewards/rejected": -2.3579981327056885,
"step": 70
},
{
"epoch": 0.52,
"learning_rate": 2.7575199021178855e-07,
"logits/chosen": -0.9876527190208435,
"logits/rejected": -0.9876527190208435,
"logps/chosen": -491.9862365722656,
"logps/rejected": -491.9862365722656,
"loss": 0.6931,
"rewards/accuracies": 0.13076923787593842,
"rewards/chosen": -3.068056344985962,
"rewards/margins": -3.2094809032656713e-09,
"rewards/rejected": -3.068056106567383,
"step": 80
},
{
"epoch": 0.58,
"learning_rate": 2.1855294234408068e-07,
"logits/chosen": -0.7263882756233215,
"logits/rejected": -0.7263882756233215,
"logps/chosen": -615.2776489257812,
"logps/rejected": -615.2776489257812,
"loss": 0.6931,
"rewards/accuracies": 0.13846154510974884,
"rewards/chosen": -4.217284202575684,
"rewards/margins": 2.063237758420655e-08,
"rewards/rejected": -4.217283725738525,
"step": 90
},
{
"epoch": 0.65,
"learning_rate": 1.6300029195778453e-07,
"logits/chosen": -0.43152233958244324,
"logits/rejected": -0.43152233958244324,
"logps/chosen": -721.6929321289062,
"logps/rejected": -721.69287109375,
"loss": 0.6931,
"rewards/accuracies": 0.1230769231915474,
"rewards/chosen": -5.2063374519348145,
"rewards/margins": -2.2007869304729866e-08,
"rewards/rejected": -5.2063374519348145,
"step": 100
},
{
"epoch": 0.71,
"learning_rate": 1.1200247470632392e-07,
"logits/chosen": -0.6822031736373901,
"logits/rejected": -0.6822031736373901,
"logps/chosen": -729.0514526367188,
"logps/rejected": -729.0514526367188,
"loss": 0.6931,
"rewards/accuracies": 0.1269230842590332,
"rewards/chosen": -5.433934688568115,
"rewards/margins": 9.169945358600273e-10,
"rewards/rejected": -5.433934688568115,
"step": 110
},
{
"epoch": 0.78,
"learning_rate": 6.822945986946385e-08,
"logits/chosen": -0.5164610147476196,
"logits/rejected": -0.5164610147476196,
"logps/chosen": -775.7005004882812,
"logps/rejected": -775.7005004882812,
"loss": 0.6931,
"rewards/accuracies": 0.1269230842590332,
"rewards/chosen": -5.883157253265381,
"rewards/margins": 4.7683716530855236e-08,
"rewards/rejected": -5.883157253265381,
"step": 120
},
{
"epoch": 0.84,
"learning_rate": 3.397296523427806e-08,
"logits/chosen": -0.38993218541145325,
"logits/rejected": -0.38993218541145325,
"logps/chosen": -804.00830078125,
"logps/rejected": -804.0083618164062,
"loss": 0.6931,
"rewards/accuracies": 0.13076923787593842,
"rewards/chosen": -6.021703720092773,
"rewards/margins": 1.9256885863683237e-08,
"rewards/rejected": -6.021703243255615,
"step": 130
},
{
"epoch": 0.91,
"learning_rate": 1.1026475173977978e-08,
"logits/chosen": -0.47965455055236816,
"logits/rejected": -0.47965455055236816,
"logps/chosen": -805.8763427734375,
"logps/rejected": -805.8763427734375,
"loss": 0.6931,
"rewards/accuracies": 0.1269230842590332,
"rewards/chosen": -6.223758220672607,
"rewards/margins": 2.6592841706474246e-08,
"rewards/rejected": -6.223758220672607,
"step": 140
},
{
"epoch": 0.97,
"learning_rate": 5.913435276374834e-10,
"logits/chosen": -0.5421052575111389,
"logits/rejected": -0.5421052575111389,
"logps/chosen": -826.6334838867188,
"logps/rejected": -826.6334838867188,
"loss": 0.6931,
"rewards/accuracies": 0.11923076957464218,
"rewards/chosen": -6.313894271850586,
"rewards/margins": 4.401573860945973e-08,
"rewards/rejected": -6.313894271850586,
"step": 150
},
{
"epoch": 0.99,
"step": 153,
"total_flos": 0.0,
"train_loss": 0.0,
"train_runtime": 0.0097,
"train_samples_per_second": 2062857.002,
"train_steps_per_second": 15780.856
}
],
"logging_steps": 10,
"max_steps": 153,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 200,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}