{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9345794392523364, "eval_steps": 50, "global_step": 100, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04672897196261682, "grad_norm": 61.970988574529464, "learning_rate": 5e-07, "logits/chosen": -2.7293832302093506, "logits/rejected": -2.7098002433776855, "logps/chosen": -282.619384765625, "logps/rejected": -220.626708984375, "loss": 0.6898, "rewards/accuracies": 0.34375, "rewards/chosen": 0.016255810856819153, "rewards/margins": 0.00715771596878767, "rewards/rejected": 0.009098095819354057, "step": 5 }, { "epoch": 0.09345794392523364, "grad_norm": 59.149513121461624, "learning_rate": 1e-06, "logits/chosen": -2.6811907291412354, "logits/rejected": -2.6537957191467285, "logps/chosen": -256.4613037109375, "logps/rejected": -214.9097442626953, "loss": 0.6465, "rewards/accuracies": 0.625, "rewards/chosen": 0.46442437171936035, "rewards/margins": 0.15177568793296814, "rewards/rejected": 0.3126486837863922, "step": 10 }, { "epoch": 0.14018691588785046, "grad_norm": 43.868507115118234, "learning_rate": 9.993623730611148e-07, "logits/chosen": -2.4990592002868652, "logits/rejected": -2.4942288398742676, "logps/chosen": -252.8370361328125, "logps/rejected": -212.9906768798828, "loss": 0.6365, "rewards/accuracies": 0.65625, "rewards/chosen": 1.3768469095230103, "rewards/margins": 0.5569905042648315, "rewards/rejected": 0.8198563456535339, "step": 15 }, { "epoch": 0.18691588785046728, "grad_norm": 40.58644668089117, "learning_rate": 9.97451118516912e-07, "logits/chosen": -2.357009172439575, "logits/rejected": -2.3200223445892334, "logps/chosen": -245.38623046875, "logps/rejected": -190.89620971679688, "loss": 0.6273, "rewards/accuracies": 0.731249988079071, "rewards/chosen": 1.5086402893066406, "rewards/margins": 0.8777653574943542, "rewards/rejected": 0.6308748722076416, "step": 20 }, { "epoch": 0.2336448598130841, "grad_norm": 46.4139582288858, "learning_rate": 9.94271111036929e-07, "logits/chosen": -2.2126636505126953, "logits/rejected": -2.2065536975860596, "logps/chosen": -242.53775024414062, "logps/rejected": -225.9445343017578, "loss": 0.6348, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 1.3302034139633179, "rewards/margins": 0.9245456457138062, "rewards/rejected": 0.4056577682495117, "step": 25 }, { "epoch": 0.2803738317757009, "grad_norm": 64.74245067517272, "learning_rate": 9.898304612549066e-07, "logits/chosen": -2.183245897293091, "logits/rejected": -2.1688549518585205, "logps/chosen": -233.8648223876953, "logps/rejected": -205.1978759765625, "loss": 0.5938, "rewards/accuracies": 0.71875, "rewards/chosen": 1.1376529932022095, "rewards/margins": 0.8598777651786804, "rewards/rejected": 0.27777519822120667, "step": 30 }, { "epoch": 0.32710280373831774, "grad_norm": 51.84730256605764, "learning_rate": 9.841404950825536e-07, "logits/chosen": -2.227497100830078, "logits/rejected": -2.1977126598358154, "logps/chosen": -237.3373565673828, "logps/rejected": -212.15512084960938, "loss": 0.5785, "rewards/accuracies": 0.6937500238418579, "rewards/chosen": 1.2570827007293701, "rewards/margins": 0.9708169102668762, "rewards/rejected": 0.2862659692764282, "step": 35 }, { "epoch": 0.37383177570093457, "grad_norm": 40.60340275031408, "learning_rate": 9.77215724822721e-07, "logits/chosen": -2.243427276611328, "logits/rejected": -2.2088351249694824, "logps/chosen": -248.1513671875, "logps/rejected": -229.47531127929688, "loss": 0.5829, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.9161311984062195, "rewards/margins": 1.0557386875152588, "rewards/rejected": -0.13960735499858856, "step": 40 }, { "epoch": 0.4205607476635514, "grad_norm": 34.21102418831377, "learning_rate": 9.69073812155662e-07, "logits/chosen": -2.2237961292266846, "logits/rejected": -2.2018837928771973, "logps/chosen": -269.6021423339844, "logps/rejected": -224.58645629882812, "loss": 0.5896, "rewards/accuracies": 0.71875, "rewards/chosen": 1.0438940525054932, "rewards/margins": 0.9452205896377563, "rewards/rejected": 0.09867370873689651, "step": 45 }, { "epoch": 0.4672897196261682, "grad_norm": 42.29092168681447, "learning_rate": 9.597355230927788e-07, "logits/chosen": -2.0892410278320312, "logits/rejected": -2.0624351501464844, "logps/chosen": -240.1529083251953, "logps/rejected": -198.04843139648438, "loss": 0.5529, "rewards/accuracies": 0.78125, "rewards/chosen": 1.232414722442627, "rewards/margins": 1.3885271549224854, "rewards/rejected": -0.156112402677536, "step": 50 }, { "epoch": 0.4672897196261682, "eval_logits/chosen": -2.0593109130859375, "eval_logits/rejected": -2.02968692779541, "eval_logps/chosen": -239.67454528808594, "eval_logps/rejected": -216.6856689453125, "eval_loss": 0.5947180390357971, "eval_rewards/accuracies": 0.7317708134651184, "eval_rewards/chosen": 0.8738771080970764, "eval_rewards/margins": 1.1332703828811646, "eval_rewards/rejected": -0.2593933641910553, "eval_runtime": 202.7096, "eval_samples_per_second": 14.997, "eval_steps_per_second": 0.237, "step": 50 }, { "epoch": 0.514018691588785, "grad_norm": 35.93324426868974, "learning_rate": 9.4922467501275e-07, "logits/chosen": -2.0398037433624268, "logits/rejected": -2.0083518028259277, "logps/chosen": -261.76177978515625, "logps/rejected": -210.04684448242188, "loss": 0.5467, "rewards/accuracies": 0.75, "rewards/chosen": 0.9814583659172058, "rewards/margins": 1.524784803390503, "rewards/rejected": -0.5433263778686523, "step": 55 }, { "epoch": 0.5607476635514018, "grad_norm": 50.252777089157334, "learning_rate": 9.375680759151206e-07, "logits/chosen": -2.092087984085083, "logits/rejected": -2.098419427871704, "logps/chosen": -252.8424530029297, "logps/rejected": -214.93258666992188, "loss": 0.5409, "rewards/accuracies": 0.768750011920929, "rewards/chosen": 0.7585053443908691, "rewards/margins": 1.2850215435028076, "rewards/rejected": -0.5265161395072937, "step": 60 }, { "epoch": 0.6074766355140186, "grad_norm": 43.0211288247323, "learning_rate": 9.247954560462927e-07, "logits/chosen": -2.0893867015838623, "logits/rejected": -2.0975987911224365, "logps/chosen": -241.7879180908203, "logps/rejected": -239.7193145751953, "loss": 0.5337, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": 0.8641234636306763, "rewards/margins": 1.493786096572876, "rewards/rejected": -0.6296626329421997, "step": 65 }, { "epoch": 0.6542056074766355, "grad_norm": 38.40101091003172, "learning_rate": 9.109393920723001e-07, "logits/chosen": -2.0060245990753174, "logits/rejected": -1.9721952676773071, "logps/chosen": -256.7455139160156, "logps/rejected": -214.79403686523438, "loss": 0.533, "rewards/accuracies": 0.762499988079071, "rewards/chosen": 0.6738287210464478, "rewards/margins": 1.2377485036849976, "rewards/rejected": -0.563919723033905, "step": 70 }, { "epoch": 0.7009345794392523, "grad_norm": 40.40382308924294, "learning_rate": 8.960352239917699e-07, "logits/chosen": -1.9107002019882202, "logits/rejected": -1.8710416555404663, "logps/chosen": -246.01028442382812, "logps/rejected": -224.3345947265625, "loss": 0.5452, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": 0.6804067492485046, "rewards/margins": 1.3526126146316528, "rewards/rejected": -0.6722058653831482, "step": 75 }, { "epoch": 0.7476635514018691, "grad_norm": 37.2632538141041, "learning_rate": 8.801209650009814e-07, "logits/chosen": -1.8342845439910889, "logits/rejected": -1.7722011804580688, "logps/chosen": -238.6627960205078, "logps/rejected": -204.77720642089844, "loss": 0.4954, "rewards/accuracies": 0.8062499761581421, "rewards/chosen": 0.8959934115409851, "rewards/margins": 1.759472131729126, "rewards/rejected": -0.8634785413742065, "step": 80 }, { "epoch": 0.794392523364486, "grad_norm": 41.835599660746425, "learning_rate": 8.632372045409141e-07, "logits/chosen": -1.9798192977905273, "logits/rejected": -1.9717410802841187, "logps/chosen": -251.6102294921875, "logps/rejected": -240.1038360595703, "loss": 0.5641, "rewards/accuracies": 0.731249988079071, "rewards/chosen": 1.3537020683288574, "rewards/margins": 1.3792067766189575, "rewards/rejected": -0.025504767894744873, "step": 85 }, { "epoch": 0.8411214953271028, "grad_norm": 33.80081107648868, "learning_rate": 8.454270047735642e-07, "logits/chosen": -2.012608051300049, "logits/rejected": -2.0120816230773926, "logps/chosen": -253.6774139404297, "logps/rejected": -194.4556884765625, "loss": 0.5553, "rewards/accuracies": 0.706250011920929, "rewards/chosen": 0.8388773798942566, "rewards/margins": 1.2849448919296265, "rewards/rejected": -0.4460674822330475, "step": 90 }, { "epoch": 0.8878504672897196, "grad_norm": 29.38305905850477, "learning_rate": 8.267357907515661e-07, "logits/chosen": -1.9744971990585327, "logits/rejected": -1.9980430603027344, "logps/chosen": -264.6416320800781, "logps/rejected": -246.7568817138672, "loss": 0.538, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": 0.25706252455711365, "rewards/margins": 2.062177896499634, "rewards/rejected": -1.8051154613494873, "step": 95 }, { "epoch": 0.9345794392523364, "grad_norm": 33.767105871104185, "learning_rate": 8.072112345612433e-07, "logits/chosen": -1.9495391845703125, "logits/rejected": -1.8999382257461548, "logps/chosen": -248.95272827148438, "logps/rejected": -255.79019165039062, "loss": 0.5159, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -0.18092432618141174, "rewards/margins": 1.6218032836914062, "rewards/rejected": -1.8027276992797852, "step": 100 }, { "epoch": 0.9345794392523364, "eval_logits/chosen": -1.9645830392837524, "eval_logits/rejected": -1.9168211221694946, "eval_logps/chosen": -250.5727081298828, "eval_logps/rejected": -234.28236389160156, "eval_loss": 0.5285552144050598, "eval_rewards/accuracies": 0.78125, "eval_rewards/chosen": -0.21594171226024628, "eval_rewards/margins": 1.8031220436096191, "eval_rewards/rejected": -2.019063711166382, "eval_runtime": 202.4154, "eval_samples_per_second": 15.019, "eval_steps_per_second": 0.237, "step": 100 } ], "logging_steps": 5, "max_steps": 321, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 100, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 1178822762299392.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }