| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.971563981042654, | |
| "eval_steps": 128, | |
| "global_step": 104, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.018957345971563982, | |
| "grad_norm": 157.75602672892865, | |
| "learning_rate": 4.545454545454545e-08, | |
| "logits/chosen": 117.53560638427734, | |
| "logits/rejected": 126.8960952758789, | |
| "logps/chosen": -335.40118408203125, | |
| "logps/rejected": -439.16552734375, | |
| "loss": 0.8464, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.1895734597156398, | |
| "grad_norm": 159.84694184422614, | |
| "learning_rate": 4.545454545454545e-07, | |
| "logits/chosen": 135.0513153076172, | |
| "logits/rejected": 138.41073608398438, | |
| "logps/chosen": -395.8233642578125, | |
| "logps/rejected": -439.4460144042969, | |
| "loss": 0.7992, | |
| "rewards/accuracies": 0.4375, | |
| "rewards/chosen": 0.008658057078719139, | |
| "rewards/margins": 0.014583365991711617, | |
| "rewards/rejected": -0.005925314500927925, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.3791469194312796, | |
| "grad_norm": 152.55623162543827, | |
| "learning_rate": 4.885348141000122e-07, | |
| "logits/chosen": 121.6577377319336, | |
| "logits/rejected": 125.38299560546875, | |
| "logps/chosen": -372.36334228515625, | |
| "logps/rejected": -426.1581115722656, | |
| "loss": 0.6773, | |
| "rewards/accuracies": 0.59375, | |
| "rewards/chosen": 0.0024319454096257687, | |
| "rewards/margins": 0.3107669949531555, | |
| "rewards/rejected": -0.30833500623703003, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.5687203791469194, | |
| "grad_norm": 119.76688287434446, | |
| "learning_rate": 4.5025027361734613e-07, | |
| "logits/chosen": 142.59500122070312, | |
| "logits/rejected": 136.1927947998047, | |
| "logps/chosen": -427.85626220703125, | |
| "logps/rejected": -474.54498291015625, | |
| "loss": 0.5982, | |
| "rewards/accuracies": 0.71875, | |
| "rewards/chosen": -1.7695486545562744, | |
| "rewards/margins": 0.9575872421264648, | |
| "rewards/rejected": -2.7271358966827393, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.7582938388625592, | |
| "grad_norm": 101.19518327182783, | |
| "learning_rate": 3.893311157806091e-07, | |
| "logits/chosen": 125.58316802978516, | |
| "logits/rejected": 114.24357604980469, | |
| "logps/chosen": -413.34722900390625, | |
| "logps/rejected": -439.4224548339844, | |
| "loss": 0.557, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": -2.9576516151428223, | |
| "rewards/margins": 1.1197086572647095, | |
| "rewards/rejected": -4.077361106872559, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.9478672985781991, | |
| "grad_norm": 124.4654073535278, | |
| "learning_rate": 3.126631330646801e-07, | |
| "logits/chosen": 140.7157440185547, | |
| "logits/rejected": 144.85781860351562, | |
| "logps/chosen": -476.4051208496094, | |
| "logps/rejected": -560.07568359375, | |
| "loss": 0.5143, | |
| "rewards/accuracies": 0.762499988079071, | |
| "rewards/chosen": -2.863062620162964, | |
| "rewards/margins": 1.3244019746780396, | |
| "rewards/rejected": -4.187464237213135, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.1374407582938388, | |
| "grad_norm": 60.3803202767405, | |
| "learning_rate": 2.2891223348923882e-07, | |
| "logits/chosen": 133.24928283691406, | |
| "logits/rejected": 136.88035583496094, | |
| "logps/chosen": -462.84130859375, | |
| "logps/rejected": -546.5650024414062, | |
| "loss": 0.3397, | |
| "rewards/accuracies": 0.875, | |
| "rewards/chosen": -3.2417819499969482, | |
| "rewards/margins": 2.2681899070739746, | |
| "rewards/rejected": -5.509971618652344, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.3270142180094786, | |
| "grad_norm": 48.59955026130948, | |
| "learning_rate": 1.4754491880085317e-07, | |
| "logits/chosen": 126.54573822021484, | |
| "logits/rejected": 128.51889038085938, | |
| "logps/chosen": -433.62042236328125, | |
| "logps/rejected": -533.0380859375, | |
| "loss": 0.2374, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": -3.3238282203674316, | |
| "rewards/margins": 2.6523845195770264, | |
| "rewards/rejected": -5.976212978363037, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.5165876777251186, | |
| "grad_norm": 54.174748869518474, | |
| "learning_rate": 7.775827023107834e-08, | |
| "logits/chosen": 112.75434875488281, | |
| "logits/rejected": 129.4608154296875, | |
| "logps/chosen": -436.5595703125, | |
| "logps/rejected": -553.3507080078125, | |
| "loss": 0.2095, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": -4.145426273345947, | |
| "rewards/margins": 2.8602230548858643, | |
| "rewards/rejected": -7.005648612976074, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.7061611374407581, | |
| "grad_norm": 54.737453014104844, | |
| "learning_rate": 2.7440387297912122e-08, | |
| "logits/chosen": 112.3515396118164, | |
| "logits/rejected": 125.18294525146484, | |
| "logps/chosen": -466.541259765625, | |
| "logps/rejected": -578.0580444335938, | |
| "loss": 0.2032, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": -4.005508899688721, | |
| "rewards/margins": 3.0133039951324463, | |
| "rewards/rejected": -7.018812656402588, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.8957345971563981, | |
| "grad_norm": 52.477271491484125, | |
| "learning_rate": 2.27878296044029e-09, | |
| "logits/chosen": 118.53382873535156, | |
| "logits/rejected": 118.45182800292969, | |
| "logps/chosen": -454.86248779296875, | |
| "logps/rejected": -543.6682739257812, | |
| "loss": 0.1998, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": -3.6443698406219482, | |
| "rewards/margins": 2.700843095779419, | |
| "rewards/rejected": -6.345212936401367, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.971563981042654, | |
| "step": 104, | |
| "total_flos": 0.0, | |
| "train_loss": 0.4319001900462004, | |
| "train_runtime": 2177.8844, | |
| "train_samples_per_second": 6.199, | |
| "train_steps_per_second": 0.048 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 104, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 500, | |
| "total_flos": 0.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |