{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.21745039412883935,
  "eval_steps": 50,
  "global_step": 300,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.007248346470961312,
      "grad_norm": 0.07201674580574036,
      "learning_rate": 4.999451708687114e-06,
      "logits/chosen": -1.9793262481689453,
      "logits/rejected": -2.5381760597229004,
      "logps/chosen": -0.28126341104507446,
      "logps/rejected": -0.3779803514480591,
      "loss": 7.3904,
      "rewards/accuracies": 0.550000011920929,
      "rewards/chosen": -0.4218950867652893,
      "rewards/margins": 0.14507544040679932,
      "rewards/rejected": -0.5669704675674438,
      "step": 10
    },
    {
      "epoch": 0.014496692941922623,
      "grad_norm": 0.07562297582626343,
      "learning_rate": 4.997807075247147e-06,
      "logits/chosen": -2.0567643642425537,
      "logits/rejected": -2.4989147186279297,
      "logps/chosen": -0.27690139412879944,
      "logps/rejected": -0.33544114232063293,
      "loss": 7.3756,
      "rewards/accuracies": 0.574999988079071,
      "rewards/chosen": -0.41535210609436035,
      "rewards/margins": 0.08780960738658905,
      "rewards/rejected": -0.5031617283821106,
      "step": 20
    },
    {
      "epoch": 0.021745039412883936,
      "grad_norm": 0.09685570746660233,
      "learning_rate": 4.9950668210706795e-06,
      "logits/chosen": -2.10174298286438,
      "logits/rejected": -2.378197431564331,
      "logps/chosen": -0.26717427372932434,
      "logps/rejected": -0.30565372109413147,
      "loss": 7.451,
      "rewards/accuracies": 0.4375,
      "rewards/chosen": -0.4007614254951477,
      "rewards/margins": 0.05771917849779129,
      "rewards/rejected": -0.4584805369377136,
      "step": 30
    },
    {
      "epoch": 0.028993385883845247,
      "grad_norm": 0.08213861286640167,
      "learning_rate": 4.9912321481237616e-06,
      "logits/chosen": -2.1633317470550537,
      "logits/rejected": -2.387866497039795,
      "logps/chosen": -0.27634260058403015,
      "logps/rejected": -0.37035584449768066,
      "loss": 7.3892,
      "rewards/accuracies": 0.6499999761581421,
      "rewards/chosen": -0.4145139157772064,
      "rewards/margins": 0.14101983606815338,
      "rewards/rejected": -0.555533766746521,
      "step": 40
    },
    {
      "epoch": 0.03624173235480656,
      "grad_norm": 0.08846044540405273,
      "learning_rate": 4.986304738420684e-06,
      "logits/chosen": -2.1402599811553955,
      "logits/rejected": -2.4459526538848877,
      "logps/chosen": -0.2535383999347687,
      "logps/rejected": -0.3090876042842865,
      "loss": 7.5171,
      "rewards/accuracies": 0.512499988079071,
      "rewards/chosen": -0.3803076148033142,
      "rewards/margins": 0.08332376182079315,
      "rewards/rejected": -0.4636313319206238,
      "step": 50
    },
    {
      "epoch": 0.03624173235480656,
      "eval_logits/chosen": -2.1165692806243896,
      "eval_logits/rejected": -2.476428747177124,
      "eval_logps/chosen": -0.2828062176704407,
      "eval_logps/rejected": -0.3432886600494385,
      "eval_loss": 0.9120001792907715,
      "eval_rewards/accuracies": 0.5089285969734192,
      "eval_rewards/chosen": -0.4242093861103058,
      "eval_rewards/margins": 0.09072363376617432,
      "eval_rewards/rejected": -0.5149329900741577,
      "eval_runtime": 30.971,
      "eval_samples_per_second": 28.801,
      "eval_steps_per_second": 3.616,
      "step": 50
    },
    {
      "epoch": 0.04349007882576787,
      "grad_norm": 0.11753705143928528,
      "learning_rate": 4.980286753286196e-06,
      "logits/chosen": -2.0575368404388428,
      "logits/rejected": -2.5456700325012207,
      "logps/chosen": -0.24622221291065216,
      "logps/rejected": -0.32402220368385315,
      "loss": 7.3926,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.36933332681655884,
      "rewards/margins": 0.11669999361038208,
      "rewards/rejected": -0.48603329062461853,
      "step": 60
    },
    {
      "epoch": 0.05073842529672919,
      "grad_norm": 0.09996571391820908,
      "learning_rate": 4.973180832407471e-06,
      "logits/chosen": -1.9278606176376343,
      "logits/rejected": -2.4620182514190674,
      "logps/chosen": -0.2596542239189148,
      "logps/rejected": -0.3665880560874939,
      "loss": 7.185,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.3894812762737274,
      "rewards/margins": 0.160400852560997,
      "rewards/rejected": -0.5498821139335632,
      "step": 70
    },
    {
      "epoch": 0.057986771767690494,
      "grad_norm": 0.07287321239709854,
      "learning_rate": 4.964990092676263e-06,
      "logits/chosen": -2.078031063079834,
      "logits/rejected": -2.461479663848877,
      "logps/chosen": -0.24513795971870422,
      "logps/rejected": -0.3448730707168579,
      "loss": 7.365,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.36770695447921753,
      "rewards/margins": 0.14960262179374695,
      "rewards/rejected": -0.5173095464706421,
      "step": 80
    },
    {
      "epoch": 0.0652351182386518,
      "grad_norm": 0.09656044095754623,
      "learning_rate": 4.9557181268217225e-06,
      "logits/chosen": -1.9979238510131836,
      "logits/rejected": -2.4381277561187744,
      "logps/chosen": -0.24047240614891052,
      "logps/rejected": -0.3277527987957001,
      "loss": 7.2664,
      "rewards/accuracies": 0.6000000238418579,
      "rewards/chosen": -0.36070865392684937,
      "rewards/margins": 0.13092057406902313,
      "rewards/rejected": -0.4916292130947113,
      "step": 90
    },
    {
      "epoch": 0.07248346470961312,
      "grad_norm": 0.08125138282775879,
      "learning_rate": 4.9453690018345144e-06,
      "logits/chosen": -1.8948112726211548,
      "logits/rejected": -2.4755520820617676,
      "logps/chosen": -0.20189261436462402,
      "logps/rejected": -0.29732149839401245,
      "loss": 7.2542,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.30283889174461365,
      "rewards/margins": 0.14314329624176025,
      "rewards/rejected": -0.44598227739334106,
      "step": 100
    },
    {
      "epoch": 0.07248346470961312,
      "eval_logits/chosen": -2.12322735786438,
      "eval_logits/rejected": -2.481174945831299,
      "eval_logps/chosen": -0.2438412606716156,
      "eval_logps/rejected": -0.3260033428668976,
      "eval_loss": 0.891861081123352,
      "eval_rewards/accuracies": 0.5625,
      "eval_rewards/chosen": -0.3657619059085846,
      "eval_rewards/margins": 0.12324309349060059,
      "eval_rewards/rejected": -0.48900502920150757,
      "eval_runtime": 30.3299,
      "eval_samples_per_second": 29.41,
      "eval_steps_per_second": 3.693,
      "step": 100
    },
    {
      "epoch": 0.07973181118057443,
      "grad_norm": 0.05962231010198593,
      "learning_rate": 4.933947257182901e-06,
      "logits/chosen": -2.1134355068206787,
      "logits/rejected": -2.516538381576538,
      "logps/chosen": -0.237023264169693,
      "logps/rejected": -0.31476154923439026,
      "loss": 7.1749,
      "rewards/accuracies": 0.550000011920929,
      "rewards/chosen": -0.3555349111557007,
      "rewards/margins": 0.11660744249820709,
      "rewards/rejected": -0.4721423089504242,
      "step": 110
    },
    {
      "epoch": 0.08698015765153574,
      "grad_norm": 0.06015922501683235,
      "learning_rate": 4.921457902821578e-06,
      "logits/chosen": -2.0215041637420654,
      "logits/rejected": -2.4902031421661377,
      "logps/chosen": -0.1890055537223816,
      "logps/rejected": -0.3192065358161926,
      "loss": 7.142,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.2835083603858948,
      "rewards/margins": 0.19530144333839417,
      "rewards/rejected": -0.47880974411964417,
      "step": 120
    },
    {
      "epoch": 0.09422850412249706,
      "grad_norm": 0.06430571526288986,
      "learning_rate": 4.907906416994146e-06,
      "logits/chosen": -2.0684752464294434,
      "logits/rejected": -2.510018825531006,
      "logps/chosen": -0.2073744535446167,
      "logps/rejected": -0.3121300935745239,
      "loss": 7.1438,
      "rewards/accuracies": 0.612500011920929,
      "rewards/chosen": -0.31106168031692505,
      "rewards/margins": 0.15713343024253845,
      "rewards/rejected": -0.4681951403617859,
      "step": 130
    },
    {
      "epoch": 0.10147685059345837,
      "grad_norm": 0.08829955011606216,
      "learning_rate": 4.893298743830168e-06,
      "logits/chosen": -2.026458263397217,
      "logits/rejected": -2.496157646179199,
      "logps/chosen": -0.19946983456611633,
      "logps/rejected": -0.32050156593322754,
      "loss": 7.1196,
      "rewards/accuracies": 0.625,
      "rewards/chosen": -0.2992047667503357,
      "rewards/margins": 0.18154758214950562,
      "rewards/rejected": -0.4807523787021637,
      "step": 140
    },
    {
      "epoch": 0.10872519706441967,
      "grad_norm": 0.09773921221494675,
      "learning_rate": 4.8776412907378845e-06,
      "logits/chosen": -2.101029872894287,
      "logits/rejected": -2.5849032402038574,
      "logps/chosen": -0.18889756500720978,
      "logps/rejected": -0.36427801847457886,
      "loss": 7.1227,
      "rewards/accuracies": 0.6499999761581421,
      "rewards/chosen": -0.28334635496139526,
      "rewards/margins": 0.26307064294815063,
      "rewards/rejected": -0.5464169979095459,
      "step": 150
    },
    {
      "epoch": 0.10872519706441967,
      "eval_logits/chosen": -2.1945455074310303,
      "eval_logits/rejected": -2.559415578842163,
      "eval_logps/chosen": -0.22209034860134125,
      "eval_logps/rejected": -0.32476040720939636,
      "eval_loss": 0.8760393261909485,
      "eval_rewards/accuracies": 0.5803571343421936,
      "eval_rewards/chosen": -0.33313554525375366,
      "eval_rewards/margins": 0.15400508046150208,
      "eval_rewards/rejected": -0.48714062571525574,
      "eval_runtime": 30.3484,
      "eval_samples_per_second": 29.392,
      "eval_steps_per_second": 3.69,
      "step": 150
    },
    {
      "epoch": 0.11597354353538099,
      "grad_norm": 0.09142427891492844,
      "learning_rate": 4.860940925593703e-06,
      "logits/chosen": -2.235069751739502,
      "logits/rejected": -2.5513346195220947,
      "logps/chosen": -0.19598741829395294,
      "logps/rejected": -0.2772120535373688,
      "loss": 7.0272,
      "rewards/accuracies": 0.5375000238418579,
      "rewards/chosen": -0.2939811050891876,
      "rewards/margins": 0.12183700501918793,
      "rewards/rejected": -0.41581812500953674,
      "step": 160
    },
    {
      "epoch": 0.1232218900063423,
      "grad_norm": 0.11735275387763977,
      "learning_rate": 4.84320497372973e-06,
      "logits/chosen": -2.154263496398926,
      "logits/rejected": -2.512010335922241,
      "logps/chosen": -0.1798369437456131,
      "logps/rejected": -0.30910637974739075,
      "loss": 7.1603,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.26975542306900024,
      "rewards/margins": 0.19390416145324707,
      "rewards/rejected": -0.4636595845222473,
      "step": 170
    },
    {
      "epoch": 0.1304702364773036,
      "grad_norm": 0.09398588538169861,
      "learning_rate": 4.824441214720629e-06,
      "logits/chosen": -2.292147159576416,
      "logits/rejected": -2.581425189971924,
      "logps/chosen": -0.18498703837394714,
      "logps/rejected": -0.31237050890922546,
      "loss": 6.995,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.2774805426597595,
      "rewards/margins": 0.19107523560523987,
      "rewards/rejected": -0.4685557782649994,
      "step": 180
    },
    {
      "epoch": 0.13771858294826492,
      "grad_norm": 0.13551996648311615,
      "learning_rate": 4.804657878971252e-06,
      "logits/chosen": -2.274120330810547,
      "logits/rejected": -2.5797386169433594,
      "logps/chosen": -0.21168990433216095,
      "logps/rejected": -0.34927254915237427,
      "loss": 6.9941,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.317534863948822,
      "rewards/margins": 0.20637397468090057,
      "rewards/rejected": -0.5239088535308838,
      "step": 190
    },
    {
      "epoch": 0.14496692941922623,
      "grad_norm": 0.15515944361686707,
      "learning_rate": 4.783863644106502e-06,
      "logits/chosen": -2.2116379737854004,
      "logits/rejected": -2.6693203449249268,
      "logps/chosen": -0.1971816122531891,
      "logps/rejected": -0.3498842120170593,
      "loss": 7.0413,
      "rewards/accuracies": 0.6499999761581421,
      "rewards/chosen": -0.2957724332809448,
      "rewards/margins": 0.22905388474464417,
      "rewards/rejected": -0.5248263478279114,
      "step": 200
    },
    {
      "epoch": 0.14496692941922623,
      "eval_logits/chosen": -2.3369696140289307,
      "eval_logits/rejected": -2.7298672199249268,
      "eval_logps/chosen": -0.22749511897563934,
      "eval_logps/rejected": -0.3613782525062561,
      "eval_loss": 0.8543878793716431,
      "eval_rewards/accuracies": 0.5714285969734192,
      "eval_rewards/chosen": -0.34124264121055603,
      "eval_rewards/margins": 0.2008247673511505,
      "eval_rewards/rejected": -0.5420674681663513,
      "eval_runtime": 30.3397,
      "eval_samples_per_second": 29.4,
      "eval_steps_per_second": 3.692,
      "step": 200
    },
    {
      "epoch": 0.15221527589018755,
      "grad_norm": 0.13039635121822357,
      "learning_rate": 4.762067631165049e-06,
      "logits/chosen": -2.2575619220733643,
      "logits/rejected": -2.755174160003662,
      "logps/chosen": -0.199218288064003,
      "logps/rejected": -0.33260637521743774,
      "loss": 6.882,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.2988274097442627,
      "rewards/margins": 0.20008206367492676,
      "rewards/rejected": -0.49890947341918945,
      "step": 210
    },
    {
      "epoch": 0.15946362236114886,
      "grad_norm": 0.15020275115966797,
      "learning_rate": 4.7392794005985324e-06,
      "logits/chosen": -2.2949752807617188,
      "logits/rejected": -2.7288482189178467,
      "logps/chosen": -0.21987763047218323,
      "logps/rejected": -0.385539174079895,
      "loss": 6.9158,
      "rewards/accuracies": 0.699999988079071,
      "rewards/chosen": -0.3298164904117584,
      "rewards/margins": 0.24849233031272888,
      "rewards/rejected": -0.5783087611198425,
      "step": 220
    },
    {
      "epoch": 0.16671196883211017,
      "grad_norm": 0.16104522347450256,
      "learning_rate": 4.715508948078037e-06,
      "logits/chosen": -2.1837964057922363,
      "logits/rejected": -2.813563585281372,
      "logps/chosen": -0.2069036215543747,
      "logps/rejected": -0.3996518552303314,
      "loss": 6.8503,
      "rewards/accuracies": 0.699999988079071,
      "rewards/chosen": -0.31035539507865906,
      "rewards/margins": 0.28912240266799927,
      "rewards/rejected": -0.5994777679443359,
      "step": 230
    },
    {
      "epoch": 0.1739603153030715,
      "grad_norm": 0.16533692181110382,
      "learning_rate": 4.690766700109659e-06,
      "logits/chosen": -2.314570188522339,
      "logits/rejected": -2.8867621421813965,
      "logps/chosen": -0.18964803218841553,
      "logps/rejected": -0.3997463583946228,
      "loss": 6.667,
      "rewards/accuracies": 0.6499999761581421,
      "rewards/chosen": -0.2844720482826233,
      "rewards/margins": 0.3151474893093109,
      "rewards/rejected": -0.5996195077896118,
      "step": 240
    },
    {
      "epoch": 0.1812086617740328,
      "grad_norm": 0.24991220235824585,
      "learning_rate": 4.665063509461098e-06,
      "logits/chosen": -2.46189546585083,
      "logits/rejected": -2.847446918487549,
      "logps/chosen": -0.24189543724060059,
      "logps/rejected": -0.4183991551399231,
      "loss": 6.7146,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.3628431558609009,
      "rewards/margins": 0.26475557684898376,
      "rewards/rejected": -0.6275987029075623,
      "step": 250
    },
    {
      "epoch": 0.1812086617740328,
      "eval_logits/chosen": -2.5230772495269775,
      "eval_logits/rejected": -2.9182300567626953,
      "eval_logps/chosen": -0.2819940149784088,
      "eval_logps/rejected": -0.48641347885131836,
      "eval_loss": 0.8070082664489746,
      "eval_rewards/accuracies": 0.5803571343421936,
      "eval_rewards/chosen": -0.4229910671710968,
      "eval_rewards/margins": 0.3066291809082031,
      "eval_rewards/rejected": -0.7296201586723328,
      "eval_runtime": 30.3523,
      "eval_samples_per_second": 29.388,
      "eval_steps_per_second": 3.69,
      "step": 250
    },
    {
      "epoch": 0.18845700824499412,
      "grad_norm": 0.3010155260562897,
      "learning_rate": 4.638410650401267e-06,
      "logits/chosen": -2.4178171157836914,
      "logits/rejected": -2.914424419403076,
      "logps/chosen": -0.24933743476867676,
      "logps/rejected": -0.4757954478263855,
      "loss": 6.5743,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.3740061819553375,
      "rewards/margins": 0.33968693017959595,
      "rewards/rejected": -0.7136931419372559,
      "step": 260
    },
    {
      "epoch": 0.19570535471595543,
      "grad_norm": 0.3765665888786316,
      "learning_rate": 4.610819813755038e-06,
      "logits/chosen": -2.407275676727295,
      "logits/rejected": -2.886014938354492,
      "logps/chosen": -0.27837619185447693,
      "logps/rejected": -0.6118712425231934,
      "loss": 6.4931,
      "rewards/accuracies": 0.5874999761581421,
      "rewards/chosen": -0.417564332485199,
      "rewards/margins": 0.5002425909042358,
      "rewards/rejected": -0.9178068041801453,
      "step": 270
    },
    {
      "epoch": 0.20295370118691675,
      "grad_norm": 0.29865291714668274,
      "learning_rate": 4.582303101775249e-06,
      "logits/chosen": -2.5404791831970215,
      "logits/rejected": -2.861341953277588,
      "logps/chosen": -0.3124132752418518,
      "logps/rejected": -0.558597207069397,
      "loss": 6.2272,
      "rewards/accuracies": 0.550000011920929,
      "rewards/chosen": -0.4686199128627777,
      "rewards/margins": 0.3692759573459625,
      "rewards/rejected": -0.8378958702087402,
      "step": 280
    },
    {
      "epoch": 0.21020204765787803,
      "grad_norm": 0.4311545789241791,
      "learning_rate": 4.55287302283426e-06,
      "logits/chosen": -2.393683910369873,
      "logits/rejected": -2.833466053009033,
      "logps/chosen": -0.3574284017086029,
      "logps/rejected": -0.8571383357048035,
      "loss": 5.7937,
      "rewards/accuracies": 0.7124999761581421,
      "rewards/chosen": -0.5361425876617432,
      "rewards/margins": 0.7495648264884949,
      "rewards/rejected": -1.2857074737548828,
      "step": 290
    },
    {
      "epoch": 0.21745039412883935,
      "grad_norm": 0.7148597836494446,
      "learning_rate": 4.522542485937369e-06,
      "logits/chosen": -2.369452476501465,
      "logits/rejected": -2.7431299686431885,
      "logps/chosen": -0.46852874755859375,
      "logps/rejected": -0.9673423767089844,
      "loss": 5.741,
      "rewards/accuracies": 0.6875,
      "rewards/chosen": -0.7027931213378906,
      "rewards/margins": 0.7482204437255859,
      "rewards/rejected": -1.4510136842727661,
      "step": 300
    },
    {
      "epoch": 0.21745039412883935,
      "eval_logits/chosen": -2.5259616374969482,
      "eval_logits/rejected": -2.8154189586639404,
      "eval_logps/chosen": -0.5109023451805115,
      "eval_logps/rejected": -0.9421015381813049,
      "eval_loss": 0.6990054845809937,
      "eval_rewards/accuracies": 0.5714285969734192,
      "eval_rewards/chosen": -0.7663536071777344,
      "eval_rewards/margins": 0.6467987895011902,
      "eval_rewards/rejected": -1.4131524562835693,
      "eval_runtime": 30.3501,
      "eval_samples_per_second": 29.39,
      "eval_steps_per_second": 3.69,
      "step": 300
    }
  ],
  "logging_steps": 10,
  "max_steps": 1500,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 50,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 1.1139461409794949e+18,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}