Muqeeth's picture
Training in progress, step 14, checkpoint
5eccea2 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"episode": 7168,
"epoch": 0.030705962988348184,
"eval_steps": 500,
"global_step": 14,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"episode": 512,
"epoch": 0.0021932830705962986,
"eps": 6,
"loss/policy_avg": 0.18060211837291718,
"lr": 3e-06,
"objective/entropy": -44.39540100097656,
"objective/kl": 0.23749515414237976,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 4.621584892272949,
"objective/scores": 4.621584892272949,
"policy/approxkl_avg": 0.09178640693426132,
"policy/clipfrac_avg": 0.271484375,
"policy/entropy_avg": 0.7151485681533813,
"step": 0,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 5024,
"val/ratio": 1.0001094341278076,
"val/ratio_var": 1.1996758075838443e-06
},
{
"episode": 1024,
"epoch": 0.004386566141192597,
"eps": 6,
"loss/policy_avg": 0.10823587328195572,
"lr": 2.9882812500000002e-06,
"objective/entropy": -40.33261489868164,
"objective/kl": 0.8562099933624268,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 4.949611186981201,
"objective/scores": 4.949611186981201,
"policy/approxkl_avg": 0.09872549772262573,
"policy/clipfrac_avg": 0.29296875,
"policy/entropy_avg": 0.6909540891647339,
"step": 1,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3988,
"val/ratio": 1.0000126361846924,
"val/ratio_var": 1.5850138197492925e-06
},
{
"episode": 1536,
"epoch": 0.006579849211788897,
"eps": 6,
"loss/policy_avg": 0.24130496382713318,
"lr": 2.9765625e-06,
"objective/entropy": -37.140586853027344,
"objective/kl": 2.1461260318756104,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 5.433126449584961,
"objective/scores": 5.433126449584961,
"policy/approxkl_avg": 0.1151532381772995,
"policy/clipfrac_avg": 0.30078125,
"policy/entropy_avg": 0.6479897499084473,
"step": 2,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 2924,
"val/ratio": 0.999896764755249,
"val/ratio_var": 1.4944761232982273e-06
},
{
"episode": 2048,
"epoch": 0.008773132282385195,
"eps": 6,
"loss/policy_avg": 0.11201904714107513,
"lr": 2.96484375e-06,
"objective/entropy": -23.162431716918945,
"objective/kl": 4.706851005554199,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 5.616954803466797,
"objective/scores": 5.616954803466797,
"policy/approxkl_avg": 0.13956432044506073,
"policy/clipfrac_avg": 0.365234375,
"policy/entropy_avg": 0.6664378643035889,
"step": 3,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 2146,
"val/ratio": 0.9996535778045654,
"val/ratio_var": 2.1856692455912707e-06
},
{
"episode": 2560,
"epoch": 0.010966415352981495,
"eps": 6,
"loss/policy_avg": 0.21055178344249725,
"lr": 2.953125e-06,
"objective/entropy": -17.498517990112305,
"objective/kl": 6.893625259399414,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 5.975703716278076,
"objective/scores": 5.975703716278076,
"policy/approxkl_avg": 0.14695778489112854,
"policy/clipfrac_avg": 0.33203125,
"policy/entropy_avg": 0.6391204595565796,
"step": 4,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 1715,
"val/ratio": 0.9999969005584717,
"val/ratio_var": 2.4044477413553977e-06
},
{
"episode": 3072,
"epoch": 0.013159698423577794,
"eps": 6,
"loss/policy_avg": 0.28335219621658325,
"lr": 2.94140625e-06,
"objective/entropy": -16.186603546142578,
"objective/kl": 8.215164184570312,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 6.314523696899414,
"objective/scores": 6.314523696899414,
"policy/approxkl_avg": 0.15406616032123566,
"policy/clipfrac_avg": 0.34375,
"policy/entropy_avg": 0.6234536170959473,
"step": 5,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 1743,
"val/ratio": 1.0000813007354736,
"val/ratio_var": 2.2491783511213725e-06
},
{
"episode": 3584,
"epoch": 0.015352981494174092,
"eps": 6,
"loss/policy_avg": 0.2581750154495239,
"lr": 2.9296875e-06,
"objective/entropy": -16.82525634765625,
"objective/kl": 10.012128829956055,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 6.68141508102417,
"objective/scores": 6.68141508102417,
"policy/approxkl_avg": 0.15727216005325317,
"policy/clipfrac_avg": 0.306640625,
"policy/entropy_avg": 0.6279575824737549,
"step": 6,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 2024,
"val/ratio": 1.0000149011611938,
"val/ratio_var": 3.910876330337487e-06
},
{
"episode": 4096,
"epoch": 0.01754626456477039,
"eps": 6,
"loss/policy_avg": 0.4508611857891083,
"lr": 2.91796875e-06,
"objective/entropy": -13.531968116760254,
"objective/kl": 11.674339294433594,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 6.797825813293457,
"objective/scores": 6.797825813293457,
"policy/approxkl_avg": 0.17787906527519226,
"policy/clipfrac_avg": 0.333984375,
"policy/entropy_avg": 0.6140714287757874,
"step": 7,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 2248,
"val/ratio": 0.9998712539672852,
"val/ratio_var": 2.3673501345911063e-06
},
{
"episode": 4608,
"epoch": 0.01973954763536669,
"eps": 6,
"loss/policy_avg": 0.25109419226646423,
"lr": 2.90625e-06,
"objective/entropy": -15.348489761352539,
"objective/kl": 12.88166332244873,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 7.097723484039307,
"objective/scores": 7.097723484039307,
"policy/approxkl_avg": 0.18050235509872437,
"policy/clipfrac_avg": 0.302734375,
"policy/entropy_avg": 0.6190727949142456,
"step": 8,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 2698,
"val/ratio": 1.000298023223877,
"val/ratio_var": 2.175341933252639e-06
},
{
"episode": 5120,
"epoch": 0.02193283070596299,
"eps": 6,
"loss/policy_avg": 0.2478555142879486,
"lr": 2.89453125e-06,
"objective/entropy": -20.807220458984375,
"objective/kl": 12.126293182373047,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 7.294116020202637,
"objective/scores": 7.294116020202637,
"policy/approxkl_avg": 0.16978076100349426,
"policy/clipfrac_avg": 0.31640625,
"policy/entropy_avg": 0.5916793346405029,
"step": 9,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3166,
"val/ratio": 0.999906063079834,
"val/ratio_var": 2.3764121124258963e-06
},
{
"episode": 5632,
"epoch": 0.024126113776559287,
"eps": 6,
"loss/policy_avg": 0.2194754183292389,
"lr": 2.8828125e-06,
"objective/entropy": -23.919815063476562,
"objective/kl": 13.048431396484375,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 7.783618927001953,
"objective/scores": 7.783618927001953,
"policy/approxkl_avg": 0.1719370186328888,
"policy/clipfrac_avg": 0.3359375,
"policy/entropy_avg": 0.6023513078689575,
"step": 10,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3395,
"val/ratio": 0.9998449683189392,
"val/ratio_var": 1.784112214409106e-06
},
{
"episode": 6144,
"epoch": 0.026319396847155587,
"eps": 6,
"loss/policy_avg": 0.13436169922351837,
"lr": 2.87109375e-06,
"objective/entropy": -21.395191192626953,
"objective/kl": 15.173453330993652,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 7.682435512542725,
"objective/scores": 7.682435512542725,
"policy/approxkl_avg": 0.19314789772033691,
"policy/clipfrac_avg": 0.369140625,
"policy/entropy_avg": 0.6016405820846558,
"step": 11,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3629,
"val/ratio": 0.9996920824050903,
"val/ratio_var": 1.8341333998250775e-06
},
{
"episode": 6656,
"epoch": 0.028512679917751884,
"eps": 6,
"loss/policy_avg": 0.27384719252586365,
"lr": 2.859375e-06,
"objective/entropy": -23.660961151123047,
"objective/kl": 15.459722518920898,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 8.303886413574219,
"objective/scores": 8.303886413574219,
"policy/approxkl_avg": 0.1933055967092514,
"policy/clipfrac_avg": 0.359375,
"policy/entropy_avg": 0.5889922976493835,
"step": 12,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3434,
"val/ratio": 1.0000567436218262,
"val/ratio_var": 2.589923496998381e-06
},
{
"episode": 7168,
"epoch": 0.030705962988348184,
"eps": 6,
"loss/policy_avg": 0.20046143233776093,
"lr": 2.84765625e-06,
"objective/entropy": -24.642715454101562,
"objective/kl": 17.541698455810547,
"objective/non_score_reward": 0.0,
"objective/rlhf_reward": 8.798042297363281,
"objective/scores": 8.798042297363281,
"policy/approxkl_avg": 0.1510610282421112,
"policy/clipfrac_avg": 0.337890625,
"policy/entropy_avg": 0.552618145942688,
"step": 13,
"val/clipfrac_avg": 0.0,
"val/num_eos_tokens": 3045,
"val/ratio": 1.0000343322753906,
"val/ratio_var": 2.4570517780375667e-06
}
],
"logging_steps": 1,
"max_steps": 128,
"num_input_tokens_seen": 0,
"num_train_epochs": 1.122960932145305,
"save_steps": 2,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": true,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0,
"train_batch_size": null,
"trial_name": null,
"trial_params": null
}