{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.0, "eval_steps": 100, "global_step": 200, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "grad_norm": 4.600574569217216, "learning_rate": 2.5e-08, "logits/chosen": -0.665952742099762, "logits/rejected": -0.7915734052658081, "logps/chosen": -817.56982421875, "logps/rejected": -1233.5159912109375, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.05, "grad_norm": 4.6956668605469805, "learning_rate": 2.5e-07, "logits/chosen": -0.6826351881027222, "logits/rejected": -0.8164599537849426, "logps/chosen": -999.0946044921875, "logps/rejected": -1313.702880859375, "loss": 0.6928, "rewards/accuracies": 0.4791666567325592, "rewards/chosen": 0.00046035187551751733, "rewards/margins": 0.0011585307074710727, "rewards/rejected": -0.0006981787737458944, "step": 10 }, { "epoch": 0.1, "grad_norm": 4.568367716631967, "learning_rate": 5e-07, "logits/chosen": -0.7570823431015015, "logits/rejected": -0.8555054664611816, "logps/chosen": -1019.2281494140625, "logps/rejected": -1418.564453125, "loss": 0.6925, "rewards/accuracies": 0.5625, "rewards/chosen": 0.0018463224405422807, "rewards/margins": 0.0015529071679338813, "rewards/rejected": 0.0002934155927505344, "step": 20 }, { "epoch": 0.15, "grad_norm": 4.602323361612275, "learning_rate": 4.96201938253052e-07, "logits/chosen": -0.7583962678909302, "logits/rejected": -0.8155515789985657, "logps/chosen": -1024.038818359375, "logps/rejected": -1309.645263671875, "loss": 0.689, "rewards/accuracies": 0.675000011920929, "rewards/chosen": 0.004936753772199154, "rewards/margins": 0.006319736130535603, "rewards/rejected": -0.0013829817762598395, "step": 30 }, { "epoch": 0.2, "grad_norm": 4.543733206128541, "learning_rate": 4.849231551964771e-07, "logits/chosen": -0.7214664220809937, "logits/rejected": -0.8081992268562317, "logps/chosen": -998.6472778320312, "logps/rejected": -1311.484130859375, "loss": 0.6835, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": 0.009940719231963158, "rewards/margins": 0.02101372741162777, "rewards/rejected": -0.011073010042309761, "step": 40 }, { "epoch": 0.25, "grad_norm": 5.088718809722254, "learning_rate": 4.6650635094610966e-07, "logits/chosen": -0.6799867749214172, "logits/rejected": -0.7925044298171997, "logps/chosen": -994.2998046875, "logps/rejected": -1354.9202880859375, "loss": 0.6734, "rewards/accuracies": 0.8125, "rewards/chosen": 0.02278929203748703, "rewards/margins": 0.04287552461028099, "rewards/rejected": -0.02008623257279396, "step": 50 }, { "epoch": 0.3, "grad_norm": 4.7938552939091, "learning_rate": 4.415111107797445e-07, "logits/chosen": -0.7093895673751831, "logits/rejected": -0.7959045171737671, "logps/chosen": -1028.694091796875, "logps/rejected": -1305.4276123046875, "loss": 0.6663, "rewards/accuracies": 0.831250011920929, "rewards/chosen": 0.02415267750620842, "rewards/margins": 0.06639216095209122, "rewards/rejected": -0.0422394797205925, "step": 60 }, { "epoch": 0.35, "grad_norm": 5.054883757482895, "learning_rate": 4.106969024216348e-07, "logits/chosen": -0.7344843149185181, "logits/rejected": -0.8120015859603882, "logps/chosen": -1003.9986572265625, "logps/rejected": -1363.428955078125, "loss": 0.6526, "rewards/accuracies": 0.831250011920929, "rewards/chosen": 0.020409032702445984, "rewards/margins": 0.08794097602367401, "rewards/rejected": -0.06753192842006683, "step": 70 }, { "epoch": 0.4, "grad_norm": 5.482448828665516, "learning_rate": 3.75e-07, "logits/chosen": -0.7606137990951538, "logits/rejected": -0.8274203538894653, "logps/chosen": -1024.800048828125, "logps/rejected": -1341.73095703125, "loss": 0.6352, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -0.019309956580400467, "rewards/margins": 0.10660652071237564, "rewards/rejected": -0.1259164810180664, "step": 80 }, { "epoch": 0.45, "grad_norm": 5.606863838837641, "learning_rate": 3.355050358314172e-07, "logits/chosen": -0.7421795725822449, "logits/rejected": -0.8470423817634583, "logps/chosen": -994.8064575195312, "logps/rejected": -1336.379150390625, "loss": 0.6015, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -0.0030502979643642902, "rewards/margins": 0.22511807084083557, "rewards/rejected": -0.22816836833953857, "step": 90 }, { "epoch": 0.5, "grad_norm": 5.3547776029335505, "learning_rate": 2.934120444167326e-07, "logits/chosen": -0.8171275854110718, "logits/rejected": -0.8435714840888977, "logps/chosen": -1030.83984375, "logps/rejected": -1344.560791015625, "loss": 0.5643, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": -0.050915710628032684, "rewards/margins": 0.47365838289260864, "rewards/rejected": -0.5245740413665771, "step": 100 }, { "epoch": 0.5, "eval_logits/chosen": -0.6995198726654053, "eval_logits/rejected": -0.8644800186157227, "eval_logps/chosen": -818.2396850585938, "eval_logps/rejected": -1334.0771484375, "eval_loss": 0.5890205502510071, "eval_rewards/accuracies": 0.7727272510528564, "eval_rewards/chosen": -0.04838310554623604, "eval_rewards/margins": 0.2467191368341446, "eval_rewards/rejected": -0.2951022684574127, "eval_runtime": 80.5531, "eval_samples_per_second": 8.392, "eval_steps_per_second": 0.273, "step": 100 }, { "epoch": 0.55, "grad_norm": 6.918657542547641, "learning_rate": 2.5e-07, "logits/chosen": -0.8259917497634888, "logits/rejected": -0.8844992518424988, "logps/chosen": -1003.9481201171875, "logps/rejected": -1435.4678955078125, "loss": 0.5411, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -0.06742595881223679, "rewards/margins": 0.6114541292190552, "rewards/rejected": -0.6788800954818726, "step": 110 }, { "epoch": 0.6, "grad_norm": 7.371942457502656, "learning_rate": 2.065879555832674e-07, "logits/chosen": -0.8082677125930786, "logits/rejected": -0.8919416666030884, "logps/chosen": -1007.8558349609375, "logps/rejected": -1447.9739990234375, "loss": 0.506, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -0.1385834515094757, "rewards/margins": 0.7867234349250793, "rewards/rejected": -0.9253069162368774, "step": 120 }, { "epoch": 0.65, "grad_norm": 12.09684601520378, "learning_rate": 1.6449496416858282e-07, "logits/chosen": -0.8252215385437012, "logits/rejected": -0.907401442527771, "logps/chosen": -990.1673583984375, "logps/rejected": -1487.195068359375, "loss": 0.4781, "rewards/accuracies": 0.887499988079071, "rewards/chosen": -0.21052339673042297, "rewards/margins": 1.0436497926712036, "rewards/rejected": -1.2541732788085938, "step": 130 }, { "epoch": 0.7, "grad_norm": 7.3453809585551495, "learning_rate": 1.2500000000000005e-07, "logits/chosen": -0.8709625005722046, "logits/rejected": -0.9511100053787231, "logps/chosen": -987.8455810546875, "logps/rejected": -1538.638916015625, "loss": 0.4291, "rewards/accuracies": 0.893750011920929, "rewards/chosen": -0.31802910566329956, "rewards/margins": 1.2913882732391357, "rewards/rejected": -1.6094173192977905, "step": 140 }, { "epoch": 0.75, "grad_norm": 8.079843532621563, "learning_rate": 8.930309757836516e-08, "logits/chosen": -0.8827797770500183, "logits/rejected": -0.9422929883003235, "logps/chosen": -1121.2596435546875, "logps/rejected": -1523.8992919921875, "loss": 0.4159, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -0.4621273875236511, "rewards/margins": 1.4384998083114624, "rewards/rejected": -1.9006273746490479, "step": 150 }, { "epoch": 0.8, "grad_norm": 7.053680331045143, "learning_rate": 5.848888922025552e-08, "logits/chosen": -0.8414640426635742, "logits/rejected": -0.9282405972480774, "logps/chosen": -1008.7062377929688, "logps/rejected": -1539.466796875, "loss": 0.3966, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -0.3745620548725128, "rewards/margins": 1.7526756525039673, "rewards/rejected": -2.1272377967834473, "step": 160 }, { "epoch": 0.85, "grad_norm": 8.515809533873794, "learning_rate": 3.349364905389032e-08, "logits/chosen": -0.8843591809272766, "logits/rejected": -0.9637457728385925, "logps/chosen": -1041.8538818359375, "logps/rejected": -1473.283203125, "loss": 0.3949, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -0.4412936568260193, "rewards/margins": 1.4416576623916626, "rewards/rejected": -1.8829511404037476, "step": 170 }, { "epoch": 0.9, "grad_norm": 8.34230704065185, "learning_rate": 1.507684480352292e-08, "logits/chosen": -0.9421852827072144, "logits/rejected": -0.9663949012756348, "logps/chosen": -1073.7769775390625, "logps/rejected": -1609.543701171875, "loss": 0.3902, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -0.5003660917282104, "rewards/margins": 1.6850690841674805, "rewards/rejected": -2.1854348182678223, "step": 180 }, { "epoch": 0.95, "grad_norm": 8.155374618910328, "learning_rate": 3.798061746947995e-09, "logits/chosen": -0.8831602931022644, "logits/rejected": -0.9347723722457886, "logps/chosen": -1112.270263671875, "logps/rejected": -1539.087890625, "loss": 0.4006, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -0.5728045701980591, "rewards/margins": 1.4792672395706177, "rewards/rejected": -2.0520718097686768, "step": 190 }, { "epoch": 1.0, "grad_norm": 7.802479752642293, "learning_rate": 0.0, "logits/chosen": -0.8749731183052063, "logits/rejected": -0.9254159927368164, "logps/chosen": -1024.6395263671875, "logps/rejected": -1483.4818115234375, "loss": 0.3959, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": -0.5080244541168213, "rewards/margins": 1.5510140657424927, "rewards/rejected": -2.0590386390686035, "step": 200 }, { "epoch": 1.0, "eval_logits/chosen": -0.8309964537620544, "eval_logits/rejected": -0.9202789068222046, "eval_logps/chosen": -859.0581665039062, "eval_logps/rejected": -1444.6895751953125, "eval_loss": 0.4459438920021057, "eval_rewards/accuracies": 0.8068181872367859, "eval_rewards/chosen": -0.4565673768520355, "eval_rewards/margins": 0.9446592926979065, "eval_rewards/rejected": -1.4012266397476196, "eval_runtime": 80.5966, "eval_samples_per_second": 8.387, "eval_steps_per_second": 0.273, "step": 200 }, { "epoch": 1.0, "step": 200, "total_flos": 0.0, "train_loss": 0.0, "train_runtime": 2.1286, "train_samples_per_second": 6009.015, "train_steps_per_second": 93.957 } ], "logging_steps": 10, "max_steps": 200, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 100, "total_flos": 0.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }