| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9984, | |
| "eval_steps": 500, | |
| "global_step": 156, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0064, | |
| "grad_norm": 1341.948853106591, | |
| "learning_rate": 3.125e-09, | |
| "logits/chosen": -3.9499800205230713, | |
| "logits/rejected": -4.237819194793701, | |
| "logps/chosen": -300.693115234375, | |
| "logps/rejected": -249.96307373046875, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.064, | |
| "grad_norm": 1338.549586928997, | |
| "learning_rate": 3.125e-08, | |
| "logits/chosen": -4.129465579986572, | |
| "logits/rejected": -4.351754188537598, | |
| "logps/chosen": -351.4585266113281, | |
| "logps/rejected": -308.89202880859375, | |
| "loss": 0.7278, | |
| "rewards/accuracies": 0.375, | |
| "rewards/chosen": -0.016052477061748505, | |
| "rewards/margins": -0.04994054138660431, | |
| "rewards/rejected": 0.033888060599565506, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.128, | |
| "grad_norm": 1285.3502810697903, | |
| "learning_rate": 4.9899357349880975e-08, | |
| "logits/chosen": -4.195886611938477, | |
| "logits/rejected": -4.383217811584473, | |
| "logps/chosen": -334.9483337402344, | |
| "logps/rejected": -293.82867431640625, | |
| "loss": 0.6803, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.20189008116722107, | |
| "rewards/margins": 0.08841639757156372, | |
| "rewards/rejected": 0.11347369104623795, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.192, | |
| "grad_norm": 938.2936106934807, | |
| "learning_rate": 4.877641290737884e-08, | |
| "logits/chosen": -4.2334418296813965, | |
| "logits/rejected": -4.3670477867126465, | |
| "logps/chosen": -327.77716064453125, | |
| "logps/rejected": -295.27203369140625, | |
| "loss": 0.5536, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.9395688772201538, | |
| "rewards/margins": 0.4487342834472656, | |
| "rewards/rejected": 0.49083465337753296, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.256, | |
| "grad_norm": 899.5533271284629, | |
| "learning_rate": 4.646121984004665e-08, | |
| "logits/chosen": -4.14696741104126, | |
| "logits/rejected": -4.348996162414551, | |
| "logps/chosen": -330.3467712402344, | |
| "logps/rejected": -288.37066650390625, | |
| "loss": 0.4332, | |
| "rewards/accuracies": 0.8218749761581421, | |
| "rewards/chosen": 1.8148998022079468, | |
| "rewards/margins": 1.0534685850143433, | |
| "rewards/rejected": 0.7614310383796692, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "grad_norm": 737.601892593307, | |
| "learning_rate": 4.3069871595684784e-08, | |
| "logits/chosen": -4.238839149475098, | |
| "logits/rejected": -4.417731285095215, | |
| "logps/chosen": -329.99560546875, | |
| "logps/rejected": -291.286865234375, | |
| "loss": 0.3783, | |
| "rewards/accuracies": 0.8531249761581421, | |
| "rewards/chosen": 2.4285898208618164, | |
| "rewards/margins": 1.3909423351287842, | |
| "rewards/rejected": 1.0376476049423218, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.384, | |
| "grad_norm": 725.6613657693058, | |
| "learning_rate": 3.8772424536302564e-08, | |
| "logits/chosen": -4.256905555725098, | |
| "logits/rejected": -4.42824125289917, | |
| "logps/chosen": -321.1641845703125, | |
| "logps/rejected": -291.2255859375, | |
| "loss": 0.3554, | |
| "rewards/accuracies": 0.846875011920929, | |
| "rewards/chosen": 2.799910306930542, | |
| "rewards/margins": 1.64876389503479, | |
| "rewards/rejected": 1.151146411895752, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.448, | |
| "grad_norm": 650.256008502672, | |
| "learning_rate": 3.378437060203357e-08, | |
| "logits/chosen": -4.1877875328063965, | |
| "logits/rejected": -4.377414703369141, | |
| "logps/chosen": -320.77117919921875, | |
| "logps/rejected": -288.512451171875, | |
| "loss": 0.3377, | |
| "rewards/accuracies": 0.7906249761581421, | |
| "rewards/chosen": 3.0348763465881348, | |
| "rewards/margins": 1.8613855838775635, | |
| "rewards/rejected": 1.1734905242919922, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.512, | |
| "grad_norm": 643.4652639779088, | |
| "learning_rate": 2.8355831645441387e-08, | |
| "logits/chosen": -4.054490089416504, | |
| "logits/rejected": -4.3433709144592285, | |
| "logps/chosen": -346.53253173828125, | |
| "logps/rejected": -307.34930419921875, | |
| "loss": 0.3369, | |
| "rewards/accuracies": 0.8656250238418579, | |
| "rewards/chosen": 3.375593662261963, | |
| "rewards/margins": 2.1429190635681152, | |
| "rewards/rejected": 1.2326747179031372, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.576, | |
| "grad_norm": 658.4601656800384, | |
| "learning_rate": 2.2759017277414164e-08, | |
| "logits/chosen": -4.182233810424805, | |
| "logits/rejected": -4.391860485076904, | |
| "logps/chosen": -333.4163513183594, | |
| "logps/rejected": -295.03515625, | |
| "loss": 0.3301, | |
| "rewards/accuracies": 0.859375, | |
| "rewards/chosen": 2.9584574699401855, | |
| "rewards/margins": 1.9474881887435913, | |
| "rewards/rejected": 1.0109691619873047, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "grad_norm": 716.6222925151064, | |
| "learning_rate": 1.7274575140626317e-08, | |
| "logits/chosen": -4.168593406677246, | |
| "logits/rejected": -4.387479305267334, | |
| "logps/chosen": -330.75775146484375, | |
| "logps/rejected": -285.5531311035156, | |
| "loss": 0.3315, | |
| "rewards/accuracies": 0.862500011920929, | |
| "rewards/chosen": 3.267747163772583, | |
| "rewards/margins": 2.245748281478882, | |
| "rewards/rejected": 1.0219987630844116, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.704, | |
| "grad_norm": 660.4496184068971, | |
| "learning_rate": 1.217751806485235e-08, | |
| "logits/chosen": -4.1479597091674805, | |
| "logits/rejected": -4.388964653015137, | |
| "logps/chosen": -312.3727722167969, | |
| "logps/rejected": -276.13372802734375, | |
| "loss": 0.3225, | |
| "rewards/accuracies": 0.8656250238418579, | |
| "rewards/chosen": 3.277716875076294, | |
| "rewards/margins": 2.2097737789154053, | |
| "rewards/rejected": 1.0679429769515991, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.768, | |
| "grad_norm": 637.7261383555538, | |
| "learning_rate": 7.723433775328384e-09, | |
| "logits/chosen": -4.145724296569824, | |
| "logits/rejected": -4.354761123657227, | |
| "logps/chosen": -326.25091552734375, | |
| "logps/rejected": -280.4151916503906, | |
| "loss": 0.3134, | |
| "rewards/accuracies": 0.8531249761581421, | |
| "rewards/chosen": 3.3363654613494873, | |
| "rewards/margins": 2.202853202819824, | |
| "rewards/rejected": 1.1335121393203735, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.832, | |
| "grad_norm": 768.5710075733069, | |
| "learning_rate": 4.135668656967433e-09, | |
| "logits/chosen": -4.2324018478393555, | |
| "logits/rejected": -4.393607139587402, | |
| "logps/chosen": -331.71051025390625, | |
| "logps/rejected": -286.67938232421875, | |
| "loss": 0.3203, | |
| "rewards/accuracies": 0.875, | |
| "rewards/chosen": 3.384364604949951, | |
| "rewards/margins": 2.2786831855773926, | |
| "rewards/rejected": 1.1056816577911377, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.896, | |
| "grad_norm": 742.2155895768275, | |
| "learning_rate": 1.5941282340065698e-09, | |
| "logits/chosen": -4.18549919128418, | |
| "logits/rejected": -4.4001359939575195, | |
| "logps/chosen": -333.18035888671875, | |
| "logps/rejected": -303.45220947265625, | |
| "loss": 0.3282, | |
| "rewards/accuracies": 0.84375, | |
| "rewards/chosen": 3.2540221214294434, | |
| "rewards/margins": 2.2057249546051025, | |
| "rewards/rejected": 1.048297643661499, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 668.7483012920786, | |
| "learning_rate": 2.262559558016325e-10, | |
| "logits/chosen": -4.1221604347229, | |
| "logits/rejected": -4.351316452026367, | |
| "logps/chosen": -339.6897888183594, | |
| "logps/rejected": -295.03692626953125, | |
| "loss": 0.3208, | |
| "rewards/accuracies": 0.859375, | |
| "rewards/chosen": 3.4081547260284424, | |
| "rewards/margins": 2.2413132190704346, | |
| "rewards/rejected": 1.1668416261672974, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.9984, | |
| "step": 156, | |
| "total_flos": 0.0, | |
| "train_loss": 0.4014626894241724, | |
| "train_runtime": 5052.0322, | |
| "train_samples_per_second": 7.906, | |
| "train_steps_per_second": 0.031 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 156, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "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": 0.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |