{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.3161794977356937, "eval_steps": 500, "global_step": 24, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.013174145738987238, "grad_norm": 0.5080101490020752, "learning_rate": 1.25e-07, "logits/chosen": 10.32492733001709, "logits/rejected": 10.282785415649414, "logps/chosen": -1.0583709478378296, "logps/rejected": -1.1253927946090698, "loss": 1.3887, "rewards/accuracies": 0.5390625, "rewards/chosen": -2.116741895675659, "rewards/margins": 0.1340436041355133, "rewards/rejected": -2.2507855892181396, "step": 1 }, { "epoch": 0.026348291477974475, "grad_norm": 0.5300341844558716, "learning_rate": 2.5e-07, "logits/chosen": 10.530074119567871, "logits/rejected": 10.672085762023926, "logps/chosen": -1.1093882322311401, "logps/rejected": -1.2041704654693604, "loss": 1.3086, "rewards/accuracies": 0.6171875, "rewards/chosen": -2.2187764644622803, "rewards/margins": 0.18956437706947327, "rewards/rejected": -2.4083409309387207, "step": 2 }, { "epoch": 0.03952243721696171, "grad_norm": 1.1035377979278564, "learning_rate": 3.75e-07, "logits/chosen": 10.240287780761719, "logits/rejected": 10.40180492401123, "logps/chosen": -1.1902998685836792, "logps/rejected": -1.311903715133667, "loss": 1.3245, "rewards/accuracies": 0.6484375, "rewards/chosen": -2.3805997371673584, "rewards/margins": 0.24320749938488007, "rewards/rejected": -2.623807430267334, "step": 3 }, { "epoch": 0.05269658295594895, "grad_norm": 1.8411834239959717, "learning_rate": 5e-07, "logits/chosen": 10.358713150024414, "logits/rejected": 10.405998229980469, "logps/chosen": -1.0821034908294678, "logps/rejected": -1.1429979801177979, "loss": 1.3768, "rewards/accuracies": 0.578125, "rewards/chosen": -2.1642069816589355, "rewards/margins": 0.12178920209407806, "rewards/rejected": -2.2859959602355957, "step": 4 }, { "epoch": 0.06587072869493618, "grad_norm": 0.5591928362846375, "learning_rate": 6.249999999999999e-07, "logits/chosen": 10.748247146606445, "logits/rejected": 10.783531188964844, "logps/chosen": -1.1161627769470215, "logps/rejected": -1.1165733337402344, "loss": 1.4573, "rewards/accuracies": 0.5859375, "rewards/chosen": -2.232325553894043, "rewards/margins": 0.0008210651576519012, "rewards/rejected": -2.2331466674804688, "step": 5 }, { "epoch": 0.07904487443392343, "grad_norm": 1.0748578310012817, "learning_rate": 7.5e-07, "logits/chosen": 10.895101547241211, "logits/rejected": 10.935002326965332, "logps/chosen": -1.051721453666687, "logps/rejected": -1.1617565155029297, "loss": 1.3341, "rewards/accuracies": 0.609375, "rewards/chosen": -2.103442907333374, "rewards/margins": 0.22007031738758087, "rewards/rejected": -2.3235130310058594, "step": 6 }, { "epoch": 0.09221902017291066, "grad_norm": 1.2474150657653809, "learning_rate": 8.75e-07, "logits/chosen": 10.060023307800293, "logits/rejected": 10.135562896728516, "logps/chosen": -1.0141205787658691, "logps/rejected": -1.1586568355560303, "loss": 1.3246, "rewards/accuracies": 0.6328125, "rewards/chosen": -2.0282411575317383, "rewards/margins": 0.28907278180122375, "rewards/rejected": -2.3173136711120605, "step": 7 }, { "epoch": 0.1053931659118979, "grad_norm": 1.7001811265945435, "learning_rate": 1e-06, "logits/chosen": 10.837862014770508, "logits/rejected": 10.750317573547363, "logps/chosen": -1.0156888961791992, "logps/rejected": -1.1156163215637207, "loss": 1.314, "rewards/accuracies": 0.671875, "rewards/chosen": -2.0313777923583984, "rewards/margins": 0.19985482096672058, "rewards/rejected": -2.2312326431274414, "step": 8 }, { "epoch": 0.11856731165088513, "grad_norm": 1.0888780355453491, "learning_rate": 9.994504457428556e-07, "logits/chosen": 10.434420585632324, "logits/rejected": 10.604791641235352, "logps/chosen": -1.1994280815124512, "logps/rejected": -1.1771385669708252, "loss": 1.4926, "rewards/accuracies": 0.5234375, "rewards/chosen": -2.3988561630249023, "rewards/margins": -0.04457877576351166, "rewards/rejected": -2.3542771339416504, "step": 9 }, { "epoch": 0.13174145738987236, "grad_norm": 0.9426828622817993, "learning_rate": 9.97802991010949e-07, "logits/chosen": 10.330013275146484, "logits/rejected": 10.423635482788086, "logps/chosen": -1.1276183128356934, "logps/rejected": -1.1535909175872803, "loss": 1.4741, "rewards/accuracies": 0.5625, "rewards/chosen": -2.2552366256713867, "rewards/margins": 0.05194506794214249, "rewards/rejected": -2.3071818351745605, "step": 10 }, { "epoch": 0.14491560312885962, "grad_norm": 1.174481987953186, "learning_rate": 9.950612572673255e-07, "logits/chosen": 10.557825088500977, "logits/rejected": 10.500927925109863, "logps/chosen": -1.2195600271224976, "logps/rejected": -1.3650047779083252, "loss": 1.3192, "rewards/accuracies": 0.640625, "rewards/chosen": -2.439120054244995, "rewards/margins": 0.2908894717693329, "rewards/rejected": -2.7300095558166504, "step": 11 }, { "epoch": 0.15808974886784685, "grad_norm": 0.9938077330589294, "learning_rate": 9.912312714377879e-07, "logits/chosen": 10.221735000610352, "logits/rejected": 10.339948654174805, "logps/chosen": -1.072077989578247, "logps/rejected": -1.114700198173523, "loss": 1.3846, "rewards/accuracies": 0.6015625, "rewards/chosen": -2.144155979156494, "rewards/margins": 0.08524461090564728, "rewards/rejected": -2.229400396347046, "step": 12 }, { "epoch": 0.17126389460683408, "grad_norm": 0.9858765006065369, "learning_rate": 9.863214526624063e-07, "logits/chosen": 10.290239334106445, "logits/rejected": 10.385334014892578, "logps/chosen": -1.1751586198806763, "logps/rejected": -1.3674362897872925, "loss": 1.2879, "rewards/accuracies": 0.6015625, "rewards/chosen": -2.3503172397613525, "rewards/margins": 0.38455531001091003, "rewards/rejected": -2.734872579574585, "step": 13 }, { "epoch": 0.1844380403458213, "grad_norm": 1.6941860914230347, "learning_rate": 9.8034259378842e-07, "logits/chosen": 10.756081581115723, "logits/rejected": 10.947944641113281, "logps/chosen": -1.0922513008117676, "logps/rejected": -1.1827645301818848, "loss": 1.3465, "rewards/accuracies": 0.5859375, "rewards/chosen": -2.184502601623535, "rewards/margins": 0.18102645874023438, "rewards/rejected": -2.3655290603637695, "step": 14 }, { "epoch": 0.19761218608480857, "grad_norm": 0.33364373445510864, "learning_rate": 9.73307837645217e-07, "logits/chosen": 10.299134254455566, "logits/rejected": 10.412195205688477, "logps/chosen": -1.1260679960250854, "logps/rejected": -1.2360444068908691, "loss": 1.3164, "rewards/accuracies": 0.609375, "rewards/chosen": -2.252135992050171, "rewards/margins": 0.219952791929245, "rewards/rejected": -2.4720888137817383, "step": 15 }, { "epoch": 0.2107863318237958, "grad_norm": 0.7868668437004089, "learning_rate": 9.652326481535433e-07, "logits/chosen": 10.79407024383545, "logits/rejected": 10.986239433288574, "logps/chosen": -1.0521718263626099, "logps/rejected": -1.0945489406585693, "loss": 1.415, "rewards/accuracies": 0.53125, "rewards/chosen": -2.1043436527252197, "rewards/margins": 0.08475431054830551, "rewards/rejected": -2.1890978813171387, "step": 16 }, { "epoch": 0.22396047756278303, "grad_norm": 0.6708123683929443, "learning_rate": 9.561347763324483e-07, "logits/chosen": 10.472009658813477, "logits/rejected": 10.457694053649902, "logps/chosen": -1.0896780490875244, "logps/rejected": -1.1595150232315063, "loss": 1.3736, "rewards/accuracies": 0.5625, "rewards/chosen": -2.179356098175049, "rewards/margins": 0.13967394828796387, "rewards/rejected": -2.3190300464630127, "step": 17 }, { "epoch": 0.23713462330177026, "grad_norm": 0.9609116911888123, "learning_rate": 9.460342212786932e-07, "logits/chosen": 10.532232284545898, "logits/rejected": 10.626241683959961, "logps/chosen": -1.1494747400283813, "logps/rejected": -1.1549882888793945, "loss": 1.4623, "rewards/accuracies": 0.6171875, "rewards/chosen": -2.2989494800567627, "rewards/margins": 0.011027364060282707, "rewards/rejected": -2.309976577758789, "step": 18 }, { "epoch": 0.2503087690407575, "grad_norm": 0.49065420031547546, "learning_rate": 9.349531862043951e-07, "logits/chosen": 10.552773475646973, "logits/rejected": 10.44497013092041, "logps/chosen": -1.123961329460144, "logps/rejected": -1.2145094871520996, "loss": 1.3068, "rewards/accuracies": 0.671875, "rewards/chosen": -2.247922658920288, "rewards/margins": 0.18109644949436188, "rewards/rejected": -2.429018974304199, "step": 19 }, { "epoch": 0.2634829147797447, "grad_norm": 1.4662220478057861, "learning_rate": 9.229160296295487e-07, "logits/chosen": 10.48397445678711, "logits/rejected": 10.557746887207031, "logps/chosen": -1.0979560613632202, "logps/rejected": -1.1902183294296265, "loss": 1.355, "rewards/accuracies": 0.625, "rewards/chosen": -2.1959121227264404, "rewards/margins": 0.1845243275165558, "rewards/rejected": -2.380436658859253, "step": 20 }, { "epoch": 0.276657060518732, "grad_norm": 1.0163906812667847, "learning_rate": 9.099492118367122e-07, "logits/chosen": 10.452715873718262, "logits/rejected": 10.512972831726074, "logps/chosen": -1.0967323780059814, "logps/rejected": -1.2304078340530396, "loss": 1.2888, "rewards/accuracies": 0.625, "rewards/chosen": -2.193464756011963, "rewards/margins": 0.26735079288482666, "rewards/rejected": -2.460815668106079, "step": 21 }, { "epoch": 0.28983120625771924, "grad_norm": 3.908339738845825, "learning_rate": 8.960812367055646e-07, "logits/chosen": 10.526410102844238, "logits/rejected": 10.603084564208984, "logps/chosen": -1.224048376083374, "logps/rejected": -1.2536935806274414, "loss": 1.4348, "rewards/accuracies": 0.578125, "rewards/chosen": -2.448096752166748, "rewards/margins": 0.0592900887131691, "rewards/rejected": -2.507387161254883, "step": 22 }, { "epoch": 0.3030053519967065, "grad_norm": 0.31968942284584045, "learning_rate": 8.813425890551909e-07, "logits/chosen": 10.193231582641602, "logits/rejected": 10.474884986877441, "logps/chosen": -1.1644542217254639, "logps/rejected": -1.1897979974746704, "loss": 1.4224, "rewards/accuracies": 0.59375, "rewards/chosen": -2.3289084434509277, "rewards/margins": 0.05068742483854294, "rewards/rejected": -2.379595994949341, "step": 23 }, { "epoch": 0.3161794977356937, "grad_norm": 1.0304734706878662, "learning_rate": 8.657656676318345e-07, "logits/chosen": 10.281815528869629, "logits/rejected": 10.21823501586914, "logps/chosen": -1.0919454097747803, "logps/rejected": -1.210888385772705, "loss": 1.325, "rewards/accuracies": 0.5703125, "rewards/chosen": -2.1838908195495605, "rewards/margins": 0.23788578808307648, "rewards/rejected": -2.42177677154541, "step": 24 } ], "logging_steps": 1, "max_steps": 75, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 12, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 0.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }