| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9992429977289932, | |
| "eval_steps": 100, | |
| "global_step": 165, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 2.9411764705882356e-07, | |
| "logits/chosen": -2.4571897983551025, | |
| "logits/rejected": -2.364729166030884, | |
| "logps/chosen": -393.0217590332031, | |
| "logps/rejected": -392.042236328125, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 2.9411764705882355e-06, | |
| "logits/chosen": -2.4598512649536133, | |
| "logits/rejected": -2.5237348079681396, | |
| "logps/chosen": -319.2628479003906, | |
| "logps/rejected": -363.0235290527344, | |
| "loss": 0.6922, | |
| "rewards/accuracies": 0.4513888955116272, | |
| "rewards/chosen": -0.005169111769646406, | |
| "rewards/margins": 0.0021531255915760994, | |
| "rewards/rejected": -0.007322237826883793, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.994932636402032e-06, | |
| "logits/chosen": -2.4682602882385254, | |
| "logits/rejected": -2.506861925125122, | |
| "logps/chosen": -310.16632080078125, | |
| "logps/rejected": -378.22735595703125, | |
| "loss": 0.6703, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": 0.03243725374341011, | |
| "rewards/margins": 0.057667654007673264, | |
| "rewards/rejected": -0.02523040398955345, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.905416503522124e-06, | |
| "logits/chosen": -2.458878993988037, | |
| "logits/rejected": -2.5221245288848877, | |
| "logps/chosen": -329.98956298828125, | |
| "logps/rejected": -392.0107421875, | |
| "loss": 0.6549, | |
| "rewards/accuracies": 0.6468750238418579, | |
| "rewards/chosen": -0.022809982299804688, | |
| "rewards/margins": 0.09913644939661026, | |
| "rewards/rejected": -0.12194643169641495, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.707922373336524e-06, | |
| "logits/chosen": -2.3397133350372314, | |
| "logits/rejected": -2.440927505493164, | |
| "logps/chosen": -318.717041015625, | |
| "logps/rejected": -400.6865539550781, | |
| "loss": 0.6365, | |
| "rewards/accuracies": 0.690625011920929, | |
| "rewards/chosen": -0.11206980794668198, | |
| "rewards/margins": 0.15860508382320404, | |
| "rewards/rejected": -0.2706748843193054, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 4.411315662967732e-06, | |
| "logits/chosen": -2.3495843410491943, | |
| "logits/rejected": -2.3737149238586426, | |
| "logps/chosen": -344.6326599121094, | |
| "logps/rejected": -397.68157958984375, | |
| "loss": 0.627, | |
| "rewards/accuracies": 0.6656249761581421, | |
| "rewards/chosen": -0.17710785567760468, | |
| "rewards/margins": 0.15342721343040466, | |
| "rewards/rejected": -0.33053505420684814, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.028910905897229e-06, | |
| "logits/chosen": -2.30446720123291, | |
| "logits/rejected": -2.3595499992370605, | |
| "logps/chosen": -353.98602294921875, | |
| "logps/rejected": -411.59515380859375, | |
| "loss": 0.6064, | |
| "rewards/accuracies": 0.715624988079071, | |
| "rewards/chosen": -0.18608322739601135, | |
| "rewards/margins": 0.23026804625988007, | |
| "rewards/rejected": -0.4163512587547302, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 3.577874068920446e-06, | |
| "logits/chosen": -2.239515781402588, | |
| "logits/rejected": -2.3043527603149414, | |
| "logps/chosen": -348.39581298828125, | |
| "logps/rejected": -414.6133728027344, | |
| "loss": 0.5915, | |
| "rewards/accuracies": 0.734375, | |
| "rewards/chosen": -0.22679059207439423, | |
| "rewards/margins": 0.23025627434253693, | |
| "rewards/rejected": -0.45704689621925354, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 3.0784519801008546e-06, | |
| "logits/chosen": -2.16733980178833, | |
| "logits/rejected": -2.2091336250305176, | |
| "logps/chosen": -339.76129150390625, | |
| "logps/rejected": -419.670654296875, | |
| "loss": 0.595, | |
| "rewards/accuracies": 0.715624988079071, | |
| "rewards/chosen": -0.2677019238471985, | |
| "rewards/margins": 0.26828354597091675, | |
| "rewards/rejected": -0.53598552942276, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 2.553063458334059e-06, | |
| "logits/chosen": -2.177290439605713, | |
| "logits/rejected": -2.2133171558380127, | |
| "logps/chosen": -337.5870666503906, | |
| "logps/rejected": -410.07464599609375, | |
| "loss": 0.5779, | |
| "rewards/accuracies": 0.734375, | |
| "rewards/chosen": -0.21148499846458435, | |
| "rewards/margins": 0.3073815107345581, | |
| "rewards/rejected": -0.5188664793968201, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 2.025292943281429e-06, | |
| "logits/chosen": -2.137645721435547, | |
| "logits/rejected": -2.2139902114868164, | |
| "logps/chosen": -362.3526611328125, | |
| "logps/rejected": -451.25543212890625, | |
| "loss": 0.5749, | |
| "rewards/accuracies": 0.734375, | |
| "rewards/chosen": -0.38997265696525574, | |
| "rewards/margins": 0.3522176146507263, | |
| "rewards/rejected": -0.7421903610229492, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "eval_logits/chosen": -2.1221249103546143, | |
| "eval_logits/rejected": -2.0325379371643066, | |
| "eval_logps/chosen": -324.94329833984375, | |
| "eval_logps/rejected": -311.2961730957031, | |
| "eval_loss": 0.688376247882843, | |
| "eval_rewards/accuracies": 0.5440000295639038, | |
| "eval_rewards/chosen": -0.18393579125404358, | |
| "eval_rewards/margins": 0.05143474042415619, | |
| "eval_rewards/rejected": -0.23537051677703857, | |
| "eval_runtime": 384.2989, | |
| "eval_samples_per_second": 5.204, | |
| "eval_steps_per_second": 0.651, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 1.5188318011445907e-06, | |
| "logits/chosen": -2.0991530418395996, | |
| "logits/rejected": -2.1866981983184814, | |
| "logps/chosen": -343.84478759765625, | |
| "logps/rejected": -443.0243225097656, | |
| "loss": 0.5695, | |
| "rewards/accuracies": 0.753125011920929, | |
| "rewards/chosen": -0.24366803467273712, | |
| "rewards/margins": 0.37467828392982483, | |
| "rewards/rejected": -0.6183463335037231, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 1.0564148305586296e-06, | |
| "logits/chosen": -2.12058424949646, | |
| "logits/rejected": -2.183859348297119, | |
| "logps/chosen": -358.75238037109375, | |
| "logps/rejected": -454.56982421875, | |
| "loss": 0.5503, | |
| "rewards/accuracies": 0.796875, | |
| "rewards/chosen": -0.3431812822818756, | |
| "rewards/margins": 0.4134922921657562, | |
| "rewards/rejected": -0.7566735148429871, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "learning_rate": 6.587997083462197e-07, | |
| "logits/chosen": -2.1187376976013184, | |
| "logits/rejected": -2.1680569648742676, | |
| "logps/chosen": -377.47918701171875, | |
| "logps/rejected": -470.26470947265625, | |
| "loss": 0.5556, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": -0.4671626091003418, | |
| "rewards/margins": 0.353724867105484, | |
| "rewards/rejected": -0.8208874464035034, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "learning_rate": 3.438351873250492e-07, | |
| "logits/chosen": -2.0515809059143066, | |
| "logits/rejected": -2.1326162815093994, | |
| "logps/chosen": -359.46771240234375, | |
| "logps/rejected": -465.7957458496094, | |
| "loss": 0.5475, | |
| "rewards/accuracies": 0.7906249761581421, | |
| "rewards/chosen": -0.36117881536483765, | |
| "rewards/margins": 0.44570302963256836, | |
| "rewards/rejected": -0.8068817853927612, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "learning_rate": 1.2565987432367032e-07, | |
| "logits/chosen": -2.070026159286499, | |
| "logits/rejected": -2.1689717769622803, | |
| "logps/chosen": -349.19305419921875, | |
| "logps/rejected": -467.4447326660156, | |
| "loss": 0.541, | |
| "rewards/accuracies": 0.7875000238418579, | |
| "rewards/chosen": -0.32100656628608704, | |
| "rewards/margins": 0.48250946402549744, | |
| "rewards/rejected": -0.8035160303115845, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "learning_rate": 1.4067554877743861e-08, | |
| "logits/chosen": -2.0285654067993164, | |
| "logits/rejected": -2.1381633281707764, | |
| "logps/chosen": -329.7977294921875, | |
| "logps/rejected": -446.18780517578125, | |
| "loss": 0.5574, | |
| "rewards/accuracies": 0.778124988079071, | |
| "rewards/chosen": -0.29442131519317627, | |
| "rewards/margins": 0.48129719495773315, | |
| "rewards/rejected": -0.7757185697555542, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 165, | |
| "total_flos": 0.0, | |
| "train_loss": 0.5940336465835572, | |
| "train_runtime": 8165.5058, | |
| "train_samples_per_second": 2.588, | |
| "train_steps_per_second": 0.02 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 165, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |