{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.9047619047619047, "eval_steps": 500, "global_step": 20, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.19047619047619047, "grad_norm": 115.44032311397486, "learning_rate": 9.931806517013612e-06, "logits/chosen": -2.1214675903320312, "logits/rejected": -2.09596586227417, "logps/chosen": -107.49108123779297, "logps/rejected": -15.508064270019531, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 2 }, { "epoch": 0.38095238095238093, "grad_norm": 313.3498585768591, "learning_rate": 9.397368756032445e-06, "logits/chosen": -2.0437583923339844, "logits/rejected": -2.048534393310547, "logps/chosen": -121.86276245117188, "logps/rejected": -16.88058090209961, "loss": 0.9007, "rewards/accuracies": 0.5625, "rewards/chosen": -0.31137895584106445, "rewards/margins": -0.054485566914081573, "rewards/rejected": -0.25689342617988586, "step": 4 }, { "epoch": 0.5714285714285714, "grad_norm": 455.64541668703777, "learning_rate": 8.386407858128707e-06, "logits/chosen": -1.9633965492248535, "logits/rejected": -2.0469534397125244, "logps/chosen": -152.4108428955078, "logps/rejected": -19.865943908691406, "loss": 2.869, "rewards/accuracies": 0.25, "rewards/chosen": -3.2351627349853516, "rewards/margins": -2.399106502532959, "rewards/rejected": -0.8360563516616821, "step": 6 }, { "epoch": 0.7619047619047619, "grad_norm": 249.13606400963317, "learning_rate": 7.008477123264849e-06, "logits/chosen": -2.021455764770508, "logits/rejected": -2.0236542224884033, "logps/chosen": -156.01495361328125, "logps/rejected": -31.431161880493164, "loss": 3.0671, "rewards/accuracies": 0.21875, "rewards/chosen": -4.389909267425537, "rewards/margins": -2.582636594772339, "rewards/rejected": -1.8072729110717773, "step": 8 }, { "epoch": 0.9523809523809523, "grad_norm": 189.45328579107624, "learning_rate": 5.412896727361663e-06, "logits/chosen": -2.025608539581299, "logits/rejected": -2.0653796195983887, "logps/chosen": -149.14700317382812, "logps/rejected": -42.3158073425293, "loss": 1.3723, "rewards/accuracies": 0.5, "rewards/chosen": -3.427455425262451, "rewards/margins": -0.4787706732749939, "rewards/rejected": -2.9486846923828125, "step": 10 }, { "epoch": 1.1428571428571428, "grad_norm": 244.97288427388463, "learning_rate": 3.7725725642960047e-06, "logits/chosen": -2.0566327571868896, "logits/rejected": -2.057169198989868, "logps/chosen": -126.93424987792969, "logps/rejected": -56.91143035888672, "loss": 1.9476, "rewards/accuracies": 0.875, "rewards/chosen": -1.397817850112915, "rewards/margins": 3.1280388832092285, "rewards/rejected": -4.525856971740723, "step": 12 }, { "epoch": 1.3333333333333333, "grad_norm": 183.45909354907522, "learning_rate": 2.265259209387867e-06, "logits/chosen": -2.049287796020508, "logits/rejected": -2.0284643173217773, "logps/chosen": -130.31076049804688, "logps/rejected": -76.90144348144531, "loss": 0.5622, "rewards/accuracies": 0.90625, "rewards/chosen": 1.7893621921539307, "rewards/margins": 8.1104154586792, "rewards/rejected": -6.3210530281066895, "step": 14 }, { "epoch": 1.5238095238095237, "grad_norm": 123.873213446449, "learning_rate": 1.0542974530180327e-06, "logits/chosen": -2.060037136077881, "logits/rejected": -2.0083541870117188, "logps/chosen": -103.64411926269531, "logps/rejected": -78.95755004882812, "loss": 0.2122, "rewards/accuracies": 0.90625, "rewards/chosen": 2.1228506565093994, "rewards/margins": 8.57053279876709, "rewards/rejected": -6.447681903839111, "step": 16 }, { "epoch": 1.7142857142857144, "grad_norm": 23.274773765505252, "learning_rate": 2.7091379149682683e-07, "logits/chosen": -2.054797649383545, "logits/rejected": -1.9887886047363281, "logps/chosen": -112.17431640625, "logps/rejected": -83.83834838867188, "loss": 0.3109, "rewards/accuracies": 0.90625, "rewards/chosen": 0.16159486770629883, "rewards/margins": 7.233631134033203, "rewards/rejected": -7.0720367431640625, "step": 18 }, { "epoch": 1.9047619047619047, "grad_norm": 155.65960541381844, "learning_rate": 0.0, "logits/chosen": -2.08943510055542, "logits/rejected": -2.03725528717041, "logps/chosen": -96.60086822509766, "logps/rejected": -91.33654022216797, "loss": 0.8478, "rewards/accuracies": 0.875, "rewards/chosen": 0.384034126996994, "rewards/margins": 8.193408966064453, "rewards/rejected": -7.809375286102295, "step": 20 }, { "epoch": 1.9047619047619047, "step": 20, "total_flos": 474810286080.0, "train_loss": 1.2783152908086777, "train_runtime": 432.4363, "train_samples_per_second": 0.777, "train_steps_per_second": 0.046 } ], "logging_steps": 2, "max_steps": 20, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 474810286080.0, "train_batch_size": 1, "trial_name": null, "trial_params": null }