| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9968652037617555, | |
| "eval_steps": 500, | |
| "global_step": 159, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.006269592476489028, | |
| "grad_norm": 48.727538993129954, | |
| "learning_rate": 2.5e-08, | |
| "logits/chosen": -1.7884750366210938, | |
| "logits/rejected": -1.7773182392120361, | |
| "logps/chosen": -178.7047882080078, | |
| "logps/pi_response": -149.36985778808594, | |
| "logps/ref_response": -149.36985778808594, | |
| "logps/rejected": -327.8221435546875, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.06269592476489028, | |
| "grad_norm": 44.27733622419657, | |
| "learning_rate": 2.5e-07, | |
| "logits/chosen": -1.925902009010315, | |
| "logits/rejected": -1.8669484853744507, | |
| "logps/chosen": -259.4127197265625, | |
| "logps/pi_response": -168.5211639404297, | |
| "logps/ref_response": -168.7109375, | |
| "logps/rejected": -363.8843994140625, | |
| "loss": 0.6877, | |
| "rewards/accuracies": 0.5208333134651184, | |
| "rewards/chosen": -0.007133293431252241, | |
| "rewards/margins": 0.008144436404109001, | |
| "rewards/rejected": -0.01527773030102253, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.12539184952978055, | |
| "grad_norm": 77.06299399731483, | |
| "learning_rate": 3.9922826507434426e-07, | |
| "logits/chosen": -1.7760452032089233, | |
| "logits/rejected": -1.7109419107437134, | |
| "logps/chosen": -287.03887939453125, | |
| "logps/pi_response": -184.39227294921875, | |
| "logps/ref_response": -179.93296813964844, | |
| "logps/rejected": -452.4369201660156, | |
| "loss": 0.6483, | |
| "rewards/accuracies": 0.5874999761581421, | |
| "rewards/chosen": -0.2718208432197571, | |
| "rewards/margins": 0.36123865842819214, | |
| "rewards/rejected": -0.6330594420433044, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.18808777429467086, | |
| "grad_norm": 37.12655893399871, | |
| "learning_rate": 3.9061450010948387e-07, | |
| "logits/chosen": -1.6445729732513428, | |
| "logits/rejected": -1.5510730743408203, | |
| "logps/chosen": -272.7137756347656, | |
| "logps/pi_response": -159.47608947753906, | |
| "logps/ref_response": -154.75852966308594, | |
| "logps/rejected": -406.7279968261719, | |
| "loss": 0.6095, | |
| "rewards/accuracies": 0.59375, | |
| "rewards/chosen": -0.29393836855888367, | |
| "rewards/margins": 0.42385944724082947, | |
| "rewards/rejected": -0.7177978157997131, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.2507836990595611, | |
| "grad_norm": 43.17045113037534, | |
| "learning_rate": 3.7283776752336964e-07, | |
| "logits/chosen": -1.585517168045044, | |
| "logits/rejected": -1.5017468929290771, | |
| "logps/chosen": -241.1661376953125, | |
| "logps/pi_response": -170.72608947753906, | |
| "logps/ref_response": -165.39320373535156, | |
| "logps/rejected": -458.1199645996094, | |
| "loss": 0.5888, | |
| "rewards/accuracies": 0.6937500238418579, | |
| "rewards/chosen": -0.12789960205554962, | |
| "rewards/margins": 0.4231465756893158, | |
| "rewards/rejected": -0.5510461330413818, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.31347962382445144, | |
| "grad_norm": 56.15552525880845, | |
| "learning_rate": 3.4675260684476076e-07, | |
| "logits/chosen": -1.31333327293396, | |
| "logits/rejected": -1.1443239450454712, | |
| "logps/chosen": -280.60247802734375, | |
| "logps/pi_response": -187.87725830078125, | |
| "logps/ref_response": -162.92739868164062, | |
| "logps/rejected": -448.25238037109375, | |
| "loss": 0.56, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -0.38959115743637085, | |
| "rewards/margins": 0.5290592908859253, | |
| "rewards/rejected": -0.9186504483222961, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.3761755485893417, | |
| "grad_norm": 59.31824188989313, | |
| "learning_rate": 3.1361294934623114e-07, | |
| "logits/chosen": -0.8963761329650879, | |
| "logits/rejected": -0.7264960408210754, | |
| "logps/chosen": -326.63250732421875, | |
| "logps/pi_response": -210.930419921875, | |
| "logps/ref_response": -159.56680297851562, | |
| "logps/rejected": -436.368408203125, | |
| "loss": 0.5461, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": -0.676125168800354, | |
| "rewards/margins": 0.5084071755409241, | |
| "rewards/rejected": -1.1845324039459229, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.438871473354232, | |
| "grad_norm": 48.22413613389395, | |
| "learning_rate": 2.750118407219124e-07, | |
| "logits/chosen": -0.902199923992157, | |
| "logits/rejected": -0.6483638882637024, | |
| "logps/chosen": -299.6310729980469, | |
| "logps/pi_response": -208.26123046875, | |
| "logps/ref_response": -157.38909912109375, | |
| "logps/rejected": -434.2081604003906, | |
| "loss": 0.5688, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": -0.5099861025810242, | |
| "rewards/margins": 0.5872552394866943, | |
| "rewards/rejected": -1.0972412824630737, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.5015673981191222, | |
| "grad_norm": 45.59894096049591, | |
| "learning_rate": 2.3280486230620434e-07, | |
| "logits/chosen": -0.756037175655365, | |
| "logits/rejected": -0.477339506149292, | |
| "logps/chosen": -285.1630554199219, | |
| "logps/pi_response": -204.59786987304688, | |
| "logps/ref_response": -150.49502563476562, | |
| "logps/rejected": -451.710693359375, | |
| "loss": 0.531, | |
| "rewards/accuracies": 0.71875, | |
| "rewards/chosen": -0.5814803242683411, | |
| "rewards/margins": 0.663509726524353, | |
| "rewards/rejected": -1.2449901103973389, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.5642633228840125, | |
| "grad_norm": 51.60912561342815, | |
| "learning_rate": 1.890209320271345e-07, | |
| "logits/chosen": -0.6508065462112427, | |
| "logits/rejected": -0.365335613489151, | |
| "logps/chosen": -306.8770751953125, | |
| "logps/pi_response": -212.5884246826172, | |
| "logps/ref_response": -147.56716918945312, | |
| "logps/rejected": -485.7850036621094, | |
| "loss": 0.4977, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": -0.6989418864250183, | |
| "rewards/margins": 0.8249226808547974, | |
| "rewards/rejected": -1.523864507675171, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.6269592476489029, | |
| "grad_norm": 47.512990245630185, | |
| "learning_rate": 1.457647729527166e-07, | |
| "logits/chosen": -0.5415297746658325, | |
| "logits/rejected": -0.25003066658973694, | |
| "logps/chosen": -314.3260192871094, | |
| "logps/pi_response": -253.16586303710938, | |
| "logps/ref_response": -160.88829040527344, | |
| "logps/rejected": -611.5823974609375, | |
| "loss": 0.5027, | |
| "rewards/accuracies": 0.8062499761581421, | |
| "rewards/chosen": -0.8261867761611938, | |
| "rewards/margins": 1.2384675741195679, | |
| "rewards/rejected": -2.06465482711792, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.6896551724137931, | |
| "grad_norm": 40.671499365551824, | |
| "learning_rate": 1.0511573783308798e-07, | |
| "logits/chosen": -0.45151621103286743, | |
| "logits/rejected": -0.2101474106311798, | |
| "logps/chosen": -330.63751220703125, | |
| "logps/pi_response": -239.79678344726562, | |
| "logps/ref_response": -149.11666870117188, | |
| "logps/rejected": -537.8972778320312, | |
| "loss": 0.4933, | |
| "rewards/accuracies": 0.75, | |
| "rewards/chosen": -0.9507951736450195, | |
| "rewards/margins": 0.9308554530143738, | |
| "rewards/rejected": -1.881650686264038, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.7523510971786834, | |
| "grad_norm": 49.168767905757875, | |
| "learning_rate": 6.9027853210943e-08, | |
| "logits/chosen": -0.5997277498245239, | |
| "logits/rejected": -0.22453825175762177, | |
| "logps/chosen": -349.99212646484375, | |
| "logps/pi_response": -280.00726318359375, | |
| "logps/ref_response": -178.89942932128906, | |
| "logps/rejected": -565.8890380859375, | |
| "loss": 0.5114, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": -0.9609189033508301, | |
| "rewards/margins": 0.9585698843002319, | |
| "rewards/rejected": -1.9194889068603516, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.8150470219435737, | |
| "grad_norm": 49.00943764848692, | |
| "learning_rate": 3.9235888047312215e-08, | |
| "logits/chosen": -0.6853595972061157, | |
| "logits/rejected": -0.2808631658554077, | |
| "logps/chosen": -327.33953857421875, | |
| "logps/pi_response": -256.1371154785156, | |
| "logps/ref_response": -166.0455780029297, | |
| "logps/rejected": -572.0056762695312, | |
| "loss": 0.5109, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": -0.8519734144210815, | |
| "rewards/margins": 1.0642752647399902, | |
| "rewards/rejected": -1.9162486791610718, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.877742946708464, | |
| "grad_norm": 65.70628626767385, | |
| "learning_rate": 1.7171962207216682e-08, | |
| "logits/chosen": -0.5600570440292358, | |
| "logits/rejected": -0.11410878598690033, | |
| "logps/chosen": -324.69439697265625, | |
| "logps/pi_response": -264.94036865234375, | |
| "logps/ref_response": -165.58712768554688, | |
| "logps/rejected": -562.1802978515625, | |
| "loss": 0.5045, | |
| "rewards/accuracies": 0.78125, | |
| "rewards/chosen": -0.8904104232788086, | |
| "rewards/margins": 0.9812215566635132, | |
| "rewards/rejected": -1.8716322183609009, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.9404388714733543, | |
| "grad_norm": 44.83563629682317, | |
| "learning_rate": 3.8967034915552865e-09, | |
| "logits/chosen": -0.47401466965675354, | |
| "logits/rejected": -0.1014849916100502, | |
| "logps/chosen": -298.55377197265625, | |
| "logps/pi_response": -241.1999969482422, | |
| "logps/ref_response": -146.22021484375, | |
| "logps/rejected": -519.8775634765625, | |
| "loss": 0.5097, | |
| "rewards/accuracies": 0.762499988079071, | |
| "rewards/chosen": -0.8489562273025513, | |
| "rewards/margins": 1.053095817565918, | |
| "rewards/rejected": -1.9020519256591797, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.9968652037617555, | |
| "step": 159, | |
| "total_flos": 0.0, | |
| "train_loss": 0.5458380771133134, | |
| "train_runtime": 3385.2715, | |
| "train_samples_per_second": 6.02, | |
| "train_steps_per_second": 0.047 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 159, | |
| "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 | |
| } | |