| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 5.0, | |
| "eval_steps": 1000, | |
| "global_step": 125, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 5.49035456239289, | |
| "learning_rate": 1.5384615384615387e-05, | |
| "logits/chosen": -1.73323655128479, | |
| "logits/rejected": -1.963712453842163, | |
| "logps/chosen": -64.71795654296875, | |
| "logps/rejected": -92.56527709960938, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 28.445585026780677, | |
| "learning_rate": 0.00015384615384615385, | |
| "logits/chosen": -1.9578092098236084, | |
| "logits/rejected": -2.1086513996124268, | |
| "logps/chosen": -241.50640869140625, | |
| "logps/rejected": -268.8847351074219, | |
| "loss": 1.1389, | |
| "rewards/accuracies": 0.4722222089767456, | |
| "rewards/chosen": -1.7811455726623535, | |
| "rewards/margins": 0.029786672443151474, | |
| "rewards/rejected": -1.8109323978424072, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 5.507863124304794, | |
| "learning_rate": 0.00019807852804032305, | |
| "logits/chosen": -2.51621675491333, | |
| "logits/rejected": -2.6432676315307617, | |
| "logps/chosen": -570.351806640625, | |
| "logps/rejected": -496.91827392578125, | |
| "loss": 1.8146, | |
| "rewards/accuracies": 0.35624998807907104, | |
| "rewards/chosen": -5.092972755432129, | |
| "rewards/margins": -1.0333130359649658, | |
| "rewards/rejected": -4.059659957885742, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 0.9111283957421439, | |
| "learning_rate": 0.00018884456359788724, | |
| "logits/chosen": -3.369929790496826, | |
| "logits/rejected": -3.283642292022705, | |
| "logps/chosen": -507.5963439941406, | |
| "logps/rejected": -572.895263671875, | |
| "loss": 0.7716, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -4.45947265625, | |
| "rewards/margins": 0.3659425377845764, | |
| "rewards/rejected": -4.825415134429932, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 1.6, | |
| "grad_norm": 1.795612553020093, | |
| "learning_rate": 0.0001726660322034027, | |
| "logits/chosen": -4.420574188232422, | |
| "logits/rejected": -4.248013496398926, | |
| "logps/chosen": -548.0731811523438, | |
| "logps/rejected": -683.3683471679688, | |
| "loss": 0.639, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": -4.906264305114746, | |
| "rewards/margins": 0.9860424995422363, | |
| "rewards/rejected": -5.892307758331299, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 0.8538474858298817, | |
| "learning_rate": 0.00015080753452465296, | |
| "logits/chosen": -5.002673149108887, | |
| "logits/rejected": -4.903468132019043, | |
| "logps/chosen": -598.4771728515625, | |
| "logps/rejected": -708.3795166015625, | |
| "loss": 0.6112, | |
| "rewards/accuracies": 0.706250011920929, | |
| "rewards/chosen": -5.362724781036377, | |
| "rewards/margins": 0.840530514717102, | |
| "rewards/rejected": -6.203255653381348, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 2.4, | |
| "grad_norm": 0.6914654231679778, | |
| "learning_rate": 0.0001249776478167227, | |
| "logits/chosen": -5.280072212219238, | |
| "logits/rejected": -5.135374069213867, | |
| "logps/chosen": -556.451904296875, | |
| "logps/rejected": -724.9661254882812, | |
| "loss": 0.494, | |
| "rewards/accuracies": 0.731249988079071, | |
| "rewards/chosen": -4.981103420257568, | |
| "rewards/margins": 1.3216989040374756, | |
| "rewards/rejected": -6.302803039550781, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 2.8, | |
| "grad_norm": 0.8698084153477929, | |
| "learning_rate": 9.719537437241312e-05, | |
| "logits/chosen": -5.114866733551025, | |
| "logits/rejected": -5.01845121383667, | |
| "logps/chosen": -614.8002319335938, | |
| "logps/rejected": -744.7828369140625, | |
| "loss": 0.5243, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -5.515006065368652, | |
| "rewards/margins": 1.0367931127548218, | |
| "rewards/rejected": -6.5517988204956055, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "grad_norm": 0.7549954414499753, | |
| "learning_rate": 6.963232548903853e-05, | |
| "logits/chosen": -4.472952365875244, | |
| "logits/rejected": -4.432765007019043, | |
| "logps/chosen": -761.5465087890625, | |
| "logps/rejected": -860.75537109375, | |
| "loss": 1.6723, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": -7.0114946365356445, | |
| "rewards/margins": 0.6902900338172913, | |
| "rewards/rejected": -7.701784610748291, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 3.6, | |
| "grad_norm": 1.1425715631983737, | |
| "learning_rate": 4.444297669803981e-05, | |
| "logits/chosen": -5.620628356933594, | |
| "logits/rejected": -5.535118103027344, | |
| "logps/chosen": -655.6566772460938, | |
| "logps/rejected": -807.3014526367188, | |
| "loss": 0.4738, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -5.936182498931885, | |
| "rewards/margins": 1.246588945388794, | |
| "rewards/rejected": -7.1827712059021, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "grad_norm": 1.9528369738814884, | |
| "learning_rate": 2.3596262417839255e-05, | |
| "logits/chosen": -5.550174236297607, | |
| "logits/rejected": -5.511833667755127, | |
| "logps/chosen": -628.8092041015625, | |
| "logps/rejected": -796.935302734375, | |
| "loss": 0.449, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -5.668156623840332, | |
| "rewards/margins": 1.3851256370544434, | |
| "rewards/rejected": -7.053282260894775, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 4.4, | |
| "grad_norm": 2.890782481250775, | |
| "learning_rate": 8.72167349386811e-06, | |
| "logits/chosen": -5.907111167907715, | |
| "logits/rejected": -5.945315361022949, | |
| "logps/chosen": -695.7470092773438, | |
| "logps/rejected": -886.4703369140625, | |
| "loss": 0.3633, | |
| "rewards/accuracies": 0.856249988079071, | |
| "rewards/chosen": -6.3334550857543945, | |
| "rewards/margins": 1.6471582651138306, | |
| "rewards/rejected": -7.980613708496094, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 4.8, | |
| "grad_norm": 0.7286712015593821, | |
| "learning_rate": 9.818874663554357e-07, | |
| "logits/chosen": -5.903909683227539, | |
| "logits/rejected": -5.952976226806641, | |
| "logps/chosen": -664.2296752929688, | |
| "logps/rejected": -862.1025390625, | |
| "loss": 0.3324, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -6.032934665679932, | |
| "rewards/margins": 1.685362458229065, | |
| "rewards/rejected": -7.718297004699707, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 125, | |
| "total_flos": 0.0, | |
| "train_loss": 0.7484294853210449, | |
| "train_runtime": 12665.918, | |
| "train_samples_per_second": 0.632, | |
| "train_steps_per_second": 0.01 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 125, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 1000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 0.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |