{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.985781990521327, "eval_steps": 100, "global_step": 52, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.02, "grad_norm": 532.1433587155178, "learning_rate": 8.333333333333333e-08, "logits/chosen": 123.11854553222656, "logits/rejected": 97.00198364257812, "logps/chosen": -425.18585205078125, "logps/rejected": -424.1869201660156, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.19, "grad_norm": 566.1002259089971, "learning_rate": 4.907293218369498e-07, "logits/chosen": 117.24137115478516, "logits/rejected": 136.07931518554688, "logps/chosen": -441.024169921875, "logps/rejected": -523.3809204101562, "loss": 0.9796, "rewards/accuracies": 0.5277777910232544, "rewards/chosen": 0.47284841537475586, "rewards/margins": 0.3260529637336731, "rewards/rejected": 0.14679540693759918, "step": 10 }, { "epoch": 0.38, "grad_norm": 404.0281115107519, "learning_rate": 3.941700805287168e-07, "logits/chosen": 126.44390869140625, "logits/rejected": 133.97540283203125, "logps/chosen": -428.3494567871094, "logps/rejected": -497.9369201660156, "loss": 0.8584, "rewards/accuracies": 0.668749988079071, "rewards/chosen": -0.416983425617218, "rewards/margins": 1.6682897806167603, "rewards/rejected": -2.085273265838623, "step": 20 }, { "epoch": 0.57, "grad_norm": 390.20247413651066, "learning_rate": 2.3293939665883228e-07, "logits/chosen": 124.90443420410156, "logits/rejected": 128.2018585205078, "logps/chosen": -471.1220703125, "logps/rejected": -534.6819458007812, "loss": 0.9208, "rewards/accuracies": 0.706250011920929, "rewards/chosen": -3.200387954711914, "rewards/margins": 2.2930397987365723, "rewards/rejected": -5.493427753448486, "step": 30 }, { "epoch": 0.76, "grad_norm": 433.63122435275017, "learning_rate": 7.936171419533652e-08, "logits/chosen": 125.5661392211914, "logits/rejected": 120.38105773925781, "logps/chosen": -487.0335388183594, "logps/rejected": -507.2274475097656, "loss": 0.8419, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": -2.770435333251953, "rewards/margins": 1.7221254110336304, "rewards/rejected": -4.492560386657715, "step": 40 }, { "epoch": 0.95, "grad_norm": 521.2062578099036, "learning_rate": 2.328513490917311e-09, "logits/chosen": 135.242919921875, "logits/rejected": 135.51016235351562, "logps/chosen": -495.3255920410156, "logps/rejected": -513.388916015625, "loss": 0.8142, "rewards/accuracies": 0.6812499761581421, "rewards/chosen": -2.5005745887756348, "rewards/margins": 1.9275367259979248, "rewards/rejected": -4.428111553192139, "step": 50 }, { "epoch": 0.99, "step": 52, "total_flos": 0.0, "train_loss": 0.8682076243253855, "train_runtime": 573.101, "train_samples_per_second": 11.778, "train_steps_per_second": 0.091 } ], "logging_steps": 10, "max_steps": 52, "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 }