{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.999259807549963, "eval_steps": 500, "global_step": 675, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.050333086602516654, "grad_norm": 3.643551826477051, "learning_rate": 5e-07, "logits/chosen": -0.5447720289230347, "logits/rejected": -0.4665735065937042, "logps/chosen": -70.74549102783203, "logps/rejected": -10.341827392578125, "loss": 0.6933, "rewards/accuracies": 0.4852941334247589, "rewards/chosen": 0.0009019060526043177, "rewards/margins": 2.6691925086197443e-05, "rewards/rejected": 0.0008752142894081771, "step": 34 }, { "epoch": 0.10066617320503331, "grad_norm": 8.25416374206543, "learning_rate": 1e-06, "logits/chosen": -0.5530118346214294, "logits/rejected": -0.46221238374710083, "logps/chosen": -64.53274536132812, "logps/rejected": -8.40461254119873, "loss": 0.6902, "rewards/accuracies": 0.5625, "rewards/chosen": 0.004430013243108988, "rewards/margins": 0.006295821163803339, "rewards/rejected": -0.0018658083863556385, "step": 68 }, { "epoch": 0.15099925980754997, "grad_norm": 5.755356788635254, "learning_rate": 9.43986820428336e-07, "logits/chosen": -0.531723141670227, "logits/rejected": -0.4240272343158722, "logps/chosen": -74.01490783691406, "logps/rejected": -9.048775672912598, "loss": 0.6686, "rewards/accuracies": 0.8713235259056091, "rewards/chosen": 0.0434609092772007, "rewards/margins": 0.05061572045087814, "rewards/rejected": -0.007154808845371008, "step": 102 }, { "epoch": 0.20133234641006661, "grad_norm": 6.1793928146362305, "learning_rate": 8.879736408566722e-07, "logits/chosen": -0.5414551496505737, "logits/rejected": -0.44891485571861267, "logps/chosen": -74.40818786621094, "logps/rejected": -11.396381378173828, "loss": 0.6245, "rewards/accuracies": 0.9779411554336548, "rewards/chosen": 0.12970797717571259, "rewards/margins": 0.14640042185783386, "rewards/rejected": -0.016692442819476128, "step": 136 }, { "epoch": 0.25166543301258326, "grad_norm": 4.775036811828613, "learning_rate": 8.319604612850081e-07, "logits/chosen": -0.5510014891624451, "logits/rejected": -0.4638059437274933, "logps/chosen": -65.10930633544922, "logps/rejected": -9.851871490478516, "loss": 0.573, "rewards/accuracies": 0.9852941036224365, "rewards/chosen": 0.23002856969833374, "rewards/margins": 0.26800811290740967, "rewards/rejected": -0.037979576736688614, "step": 170 }, { "epoch": 0.30199851961509994, "grad_norm": 4.691457271575928, "learning_rate": 7.759472817133442e-07, "logits/chosen": -0.5684951543807983, "logits/rejected": -0.48908984661102295, "logps/chosen": -63.056541442871094, "logps/rejected": -9.765806198120117, "loss": 0.4989, "rewards/accuracies": 0.9742646813392639, "rewards/chosen": 0.39861220121383667, "rewards/margins": 0.48014047741889954, "rewards/rejected": -0.08152831345796585, "step": 204 }, { "epoch": 0.35233160621761656, "grad_norm": 3.0710372924804688, "learning_rate": 7.199341021416804e-07, "logits/chosen": -0.5603327751159668, "logits/rejected": -0.474366694688797, "logps/chosen": -68.22887420654297, "logps/rejected": -11.145505905151367, "loss": 0.4096, "rewards/accuracies": 0.9742646813392639, "rewards/chosen": 0.6239223480224609, "rewards/margins": 0.7788197994232178, "rewards/rejected": -0.15489746630191803, "step": 238 }, { "epoch": 0.40266469282013323, "grad_norm": 3.50852108001709, "learning_rate": 6.639209225700164e-07, "logits/chosen": -0.554121732711792, "logits/rejected": -0.47616252303123474, "logps/chosen": -59.60889434814453, "logps/rejected": -12.72043514251709, "loss": 0.3624, "rewards/accuracies": 0.9779411554336548, "rewards/chosen": 0.769890308380127, "rewards/margins": 1.017148494720459, "rewards/rejected": -0.24725815653800964, "step": 272 }, { "epoch": 0.4529977794226499, "grad_norm": 2.061458110809326, "learning_rate": 6.079077429983526e-07, "logits/chosen": -0.5355279445648193, "logits/rejected": -0.46640804409980774, "logps/chosen": -64.37019348144531, "logps/rejected": -15.296799659729004, "loss": 0.2901, "rewards/accuracies": 0.9779411554336548, "rewards/chosen": 0.9912306666374207, "rewards/margins": 1.3743596076965332, "rewards/rejected": -0.3831288516521454, "step": 306 }, { "epoch": 0.5033308660251665, "grad_norm": 2.341501474380493, "learning_rate": 5.518945634266886e-07, "logits/chosen": -0.5389317870140076, "logits/rejected": -0.4714396893978119, "logps/chosen": -62.30534744262695, "logps/rejected": -16.425796508789062, "loss": 0.2661, "rewards/accuracies": 0.9816176295280457, "rewards/chosen": 1.0894008874893188, "rewards/margins": 1.5544438362121582, "rewards/rejected": -0.46504291892051697, "step": 340 }, { "epoch": 0.5536639526276832, "grad_norm": 1.6648075580596924, "learning_rate": 4.958813838550247e-07, "logits/chosen": -0.5351024270057678, "logits/rejected": -0.48344936966896057, "logps/chosen": -53.63265609741211, "logps/rejected": -17.541196823120117, "loss": 0.2411, "rewards/accuracies": 0.9705882668495178, "rewards/chosen": 1.1532927751541138, "rewards/margins": 1.7396329641342163, "rewards/rejected": -0.5863401889801025, "step": 374 }, { "epoch": 0.6039970392301999, "grad_norm": 2.055748701095581, "learning_rate": 4.3986820428336077e-07, "logits/chosen": -0.538494884967804, "logits/rejected": -0.4838006794452667, "logps/chosen": -57.38086700439453, "logps/rejected": -18.22422981262207, "loss": 0.2084, "rewards/accuracies": 0.9705882668495178, "rewards/chosen": 1.300463318824768, "rewards/margins": 1.98318350315094, "rewards/rejected": -0.6827201843261719, "step": 408 }, { "epoch": 0.6543301258327165, "grad_norm": 3.2196602821350098, "learning_rate": 3.8385502471169683e-07, "logits/chosen": -0.5572548508644104, "logits/rejected": -0.5117458701133728, "logps/chosen": -54.00746154785156, "logps/rejected": -17.898473739624023, "loss": 0.1869, "rewards/accuracies": 0.9889705777168274, "rewards/chosen": 1.3422279357910156, "rewards/margins": 2.1278655529022217, "rewards/rejected": -0.7856374382972717, "step": 442 }, { "epoch": 0.7046632124352331, "grad_norm": 1.63054358959198, "learning_rate": 3.2784184514003294e-07, "logits/chosen": -0.5344655513763428, "logits/rejected": -0.47442930936813354, "logps/chosen": -56.39248275756836, "logps/rejected": -17.505578994750977, "loss": 0.1707, "rewards/accuracies": 0.9889705777168274, "rewards/chosen": 1.3500103950500488, "rewards/margins": 2.1952497959136963, "rewards/rejected": -0.8452394008636475, "step": 476 }, { "epoch": 0.7549962990377498, "grad_norm": 1.9224135875701904, "learning_rate": 2.7182866556836905e-07, "logits/chosen": -0.5313306450843811, "logits/rejected": -0.48571935296058655, "logps/chosen": -50.30500411987305, "logps/rejected": -18.38837242126465, "loss": 0.1665, "rewards/accuracies": 0.9816176295280457, "rewards/chosen": 1.3861879110336304, "rewards/margins": 2.343000888824463, "rewards/rejected": -0.9568131566047668, "step": 510 }, { "epoch": 0.8053293856402665, "grad_norm": 2.294525146484375, "learning_rate": 2.158154859967051e-07, "logits/chosen": -0.5495839715003967, "logits/rejected": -0.4935465455055237, "logps/chosen": -54.87118911743164, "logps/rejected": -20.08263397216797, "loss": 0.1558, "rewards/accuracies": 0.9889705777168274, "rewards/chosen": 1.3922532796859741, "rewards/margins": 2.4232118129730225, "rewards/rejected": -1.0309585332870483, "step": 544 }, { "epoch": 0.8556624722427831, "grad_norm": 1.459617257118225, "learning_rate": 1.5980230642504117e-07, "logits/chosen": -0.5833328366279602, "logits/rejected": -0.517787516117096, "logps/chosen": -62.456932067871094, "logps/rejected": -20.363330841064453, "loss": 0.1342, "rewards/accuracies": 0.9926470518112183, "rewards/chosen": 1.5634468793869019, "rewards/margins": 2.619420289993286, "rewards/rejected": -1.0559734106063843, "step": 578 }, { "epoch": 0.9059955588452998, "grad_norm": 1.3194255828857422, "learning_rate": 1.0378912685337726e-07, "logits/chosen": -0.5344783663749695, "logits/rejected": -0.4842905104160309, "logps/chosen": -54.44826126098633, "logps/rejected": -20.549943923950195, "loss": 0.1334, "rewards/accuracies": 0.9852941036224365, "rewards/chosen": 1.394426703453064, "rewards/margins": 2.494863271713257, "rewards/rejected": -1.1004364490509033, "step": 612 }, { "epoch": 0.9563286454478165, "grad_norm": 1.6856123208999634, "learning_rate": 4.777594728171334e-08, "logits/chosen": -0.5274606347084045, "logits/rejected": -0.48015105724334717, "logps/chosen": -56.23059844970703, "logps/rejected": -23.505773544311523, "loss": 0.1363, "rewards/accuracies": 0.9852941036224365, "rewards/chosen": 1.4247695207595825, "rewards/margins": 2.5551772117614746, "rewards/rejected": -1.130407691001892, "step": 646 }, { "epoch": 0.999259807549963, "step": 675, "total_flos": 3.8465011975952794e+17, "train_loss": 0.33879759046766494, "train_runtime": 9841.7629, "train_samples_per_second": 0.549, "train_steps_per_second": 0.069 } ], "logging_steps": 34, "max_steps": 675, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 3.8465011975952794e+17, "train_batch_size": 2, "trial_name": null, "trial_params": null }