{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9221902017291066, "eval_steps": 500, "global_step": 10, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.09221902017291066, "grad_norm": 0.3283297121524811, "learning_rate": 5e-07, "logits/chosen": 0.08735769987106323, "logits/rejected": 0.2894211411476135, "logps/chosen": -131.21682739257812, "logps/ref_chosen": -121.79497528076172, "logps/ref_rejected": -161.43960571289062, "logps/rejected": -170.8655242919922, "loss": 0.6172, "rewards/accuracies": 0.5, "rewards/chosen": -0.09421855211257935, "rewards/margins": 4.066087421961129e-05, "rewards/rejected": -0.09425921738147736, "sft_loss": 0.4654408097267151, "step": 1 }, { "epoch": 0.1844380403458213, "grad_norm": 0.3361700177192688, "learning_rate": 4.849231551964771e-07, "logits/chosen": 0.13249221444129944, "logits/rejected": 0.2818200886249542, "logps/chosen": -133.42129516601562, "logps/ref_chosen": -124.1944808959961, "logps/ref_rejected": -155.79832458496094, "logps/rejected": -164.98802185058594, "loss": 0.6221, "rewards/accuracies": 0.4921875, "rewards/chosen": -0.092268206179142, "rewards/margins": -0.00037108425749465823, "rewards/rejected": -0.09189712256193161, "sft_loss": 0.4797438383102417, "step": 2 }, { "epoch": 0.276657060518732, "grad_norm": 0.3492422103881836, "learning_rate": 4.415111107797445e-07, "logits/chosen": 0.1395522505044937, "logits/rejected": 0.3097372353076935, "logps/chosen": -134.55520629882812, "logps/ref_chosen": -126.107666015625, "logps/ref_rejected": -166.2165985107422, "logps/rejected": -174.74267578125, "loss": 0.6219, "rewards/accuracies": 0.546875, "rewards/chosen": -0.08447550237178802, "rewards/margins": 0.0007849468383938074, "rewards/rejected": -0.08526045083999634, "sft_loss": 0.48026716709136963, "step": 3 }, { "epoch": 0.3688760806916426, "grad_norm": 0.40037524700164795, "learning_rate": 3.75e-07, "logits/chosen": 0.1005653589963913, "logits/rejected": 0.24767020344734192, "logps/chosen": -132.58848571777344, "logps/ref_chosen": -125.09526062011719, "logps/ref_rejected": -163.86485290527344, "logps/rejected": -171.28343200683594, "loss": 0.6207, "rewards/accuracies": 0.4453125, "rewards/chosen": -0.07493225485086441, "rewards/margins": -0.0007463583024218678, "rewards/rejected": -0.07418590039014816, "sft_loss": 0.47509297728538513, "step": 4 }, { "epoch": 0.4610951008645533, "grad_norm": 0.3622381389141083, "learning_rate": 2.934120444167326e-07, "logits/chosen": 0.09210827201604843, "logits/rejected": 0.27451103925704956, "logps/chosen": -127.89833068847656, "logps/ref_chosen": -121.33724212646484, "logps/ref_rejected": -157.88401794433594, "logps/rejected": -164.47955322265625, "loss": 0.6192, "rewards/accuracies": 0.4921875, "rewards/chosen": -0.06561093032360077, "rewards/margins": 0.00034457340370863676, "rewards/rejected": -0.0659554973244667, "sft_loss": 0.4717320203781128, "step": 5 }, { "epoch": 0.553314121037464, "grad_norm": 0.3330410420894623, "learning_rate": 2.065879555832674e-07, "logits/chosen": 0.08373218774795532, "logits/rejected": 0.31112056970596313, "logps/chosen": -130.05555725097656, "logps/ref_chosen": -124.18991088867188, "logps/ref_rejected": -162.19686889648438, "logps/rejected": -168.12203979492188, "loss": 0.6132, "rewards/accuracies": 0.515625, "rewards/chosen": -0.05865645781159401, "rewards/margins": 0.0005952669307589531, "rewards/rejected": -0.059251733124256134, "sft_loss": 0.4540290832519531, "step": 6 }, { "epoch": 0.6455331412103746, "grad_norm": 0.3554254174232483, "learning_rate": 1.2500000000000005e-07, "logits/chosen": 0.0741545632481575, "logits/rejected": 0.2337346076965332, "logps/chosen": -128.43069458007812, "logps/ref_chosen": -124.93177032470703, "logps/ref_rejected": -159.1948699951172, "logps/rejected": -162.75405883789062, "loss": 0.6147, "rewards/accuracies": 0.5078125, "rewards/chosen": -0.03498939424753189, "rewards/margins": 0.0006023755413480103, "rewards/rejected": -0.035591769963502884, "sft_loss": 0.45834434032440186, "step": 7 }, { "epoch": 0.7377521613832853, "grad_norm": 0.3500949740409851, "learning_rate": 5.848888922025552e-08, "logits/chosen": 0.14493891596794128, "logits/rejected": 0.311847060918808, "logps/chosen": -127.07202911376953, "logps/ref_chosen": -124.96209716796875, "logps/ref_rejected": -158.53147888183594, "logps/rejected": -160.7977752685547, "loss": 0.6134, "rewards/accuracies": 0.6328125, "rewards/chosen": -0.021099410951137543, "rewards/margins": 0.001563590602017939, "rewards/rejected": -0.022663000971078873, "sft_loss": 0.4554786682128906, "step": 8 }, { "epoch": 0.829971181556196, "grad_norm": 0.4512203335762024, "learning_rate": 1.507684480352292e-08, "logits/chosen": 0.10647760331630707, "logits/rejected": 0.3222857713699341, "logps/chosen": -131.0767059326172, "logps/ref_chosen": -129.31468200683594, "logps/ref_rejected": -173.19229125976562, "logps/rejected": -174.9317169189453, "loss": 0.6109, "rewards/accuracies": 0.4453125, "rewards/chosen": -0.01762016862630844, "rewards/margins": -0.00022587231069337577, "rewards/rejected": -0.0173942968249321, "sft_loss": 0.44625312089920044, "step": 9 }, { "epoch": 0.9221902017291066, "grad_norm": 0.2977113425731659, "learning_rate": 0.0, "logits/chosen": 0.1302759051322937, "logits/rejected": 0.2790878713130951, "logps/chosen": -127.37201690673828, "logps/ref_chosen": -125.88874816894531, "logps/ref_rejected": -154.29812622070312, "logps/rejected": -155.82936096191406, "loss": 0.6057, "rewards/accuracies": 0.546875, "rewards/chosen": -0.014832701534032822, "rewards/margins": 0.00047985377022996545, "rewards/rejected": -0.015312554314732552, "sft_loss": 0.4313240647315979, "step": 10 }, { "epoch": 0.9221902017291066, "step": 10, "total_flos": 0.0, "train_loss": 0.6159235239028931, "train_runtime": 636.1291, "train_samples_per_second": 2.18, "train_steps_per_second": 0.016 } ], "logging_steps": 1, "max_steps": 10, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 100000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": false, "should_training_stop": false }, "attributes": {} } }, "total_flos": 0.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }