| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.31840796019900497, | |
| "eval_steps": 500, | |
| "global_step": 800, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.03980099502487562, | |
| "grad_norm": 1.5459058284759521, | |
| "learning_rate": 4.980474401576887e-07, | |
| "logits/chosen": -1.4838274717330933, | |
| "logits/rejected": -1.5693057775497437, | |
| "logps/chosen": -190.30685424804688, | |
| "logps/rejected": -190.12179565429688, | |
| "loss": 0.7107, | |
| "rewards/accuracies": 0.49312499165534973, | |
| "rewards/chosen": -0.8337415456771851, | |
| "rewards/margins": 0.0017802910879254341, | |
| "rewards/rejected": -0.8355217576026917, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.07960199004975124, | |
| "grad_norm": 1.247958779335022, | |
| "learning_rate": 4.922202605502572e-07, | |
| "logits/chosen": -1.4577226638793945, | |
| "logits/rejected": -1.544972538948059, | |
| "logps/chosen": -191.1748504638672, | |
| "logps/rejected": -191.21063232421875, | |
| "loss": 0.7063, | |
| "rewards/accuracies": 0.5017187595367432, | |
| "rewards/chosen": -0.7915948629379272, | |
| "rewards/margins": 0.004952423740178347, | |
| "rewards/rejected": -0.7965472340583801, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.11940298507462686, | |
| "grad_norm": 1.1431488990783691, | |
| "learning_rate": 4.82609484512869e-07, | |
| "logits/chosen": -1.4753738641738892, | |
| "logits/rejected": -1.5457732677459717, | |
| "logps/chosen": -189.42088317871094, | |
| "logps/rejected": -190.7891082763672, | |
| "loss": 0.7009, | |
| "rewards/accuracies": 0.5146874785423279, | |
| "rewards/chosen": -0.7856764197349548, | |
| "rewards/margins": 0.013686330057680607, | |
| "rewards/rejected": -0.7993627190589905, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.15920398009950248, | |
| "grad_norm": 1.1773804426193237, | |
| "learning_rate": 4.6936523696827614e-07, | |
| "logits/chosen": -1.4836928844451904, | |
| "logits/rejected": -1.5391887426376343, | |
| "logps/chosen": -191.47879028320312, | |
| "logps/rejected": -192.72312927246094, | |
| "loss": 0.7031, | |
| "rewards/accuracies": 0.4935937523841858, | |
| "rewards/chosen": -0.7633911967277527, | |
| "rewards/margins": 0.00844556000083685, | |
| "rewards/rejected": -0.7718366384506226, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.19900497512437812, | |
| "grad_norm": 1.1332217454910278, | |
| "learning_rate": 4.5269439940365644e-07, | |
| "logits/chosen": -1.5031734704971313, | |
| "logits/rejected": -1.5771981477737427, | |
| "logps/chosen": -188.72503662109375, | |
| "logps/rejected": -188.9233856201172, | |
| "loss": 0.6982, | |
| "rewards/accuracies": 0.5159375071525574, | |
| "rewards/chosen": -0.7517531514167786, | |
| "rewards/margins": 0.017128920182585716, | |
| "rewards/rejected": -0.7688820362091064, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.23880597014925373, | |
| "grad_norm": 1.1783969402313232, | |
| "learning_rate": 4.328573782827409e-07, | |
| "logits/chosen": -1.4775351285934448, | |
| "logits/rejected": -1.5578263998031616, | |
| "logps/chosen": -187.82449340820312, | |
| "logps/rejected": -186.08935546875, | |
| "loss": 0.6956, | |
| "rewards/accuracies": 0.5301562547683716, | |
| "rewards/chosen": -0.7211830615997314, | |
| "rewards/margins": 0.02292461320757866, | |
| "rewards/rejected": -0.744107723236084, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.27860696517412936, | |
| "grad_norm": 1.2275896072387695, | |
| "learning_rate": 4.1016403737218373e-07, | |
| "logits/chosen": -1.4353595972061157, | |
| "logits/rejected": -1.5325896739959717, | |
| "logps/chosen": -190.53240966796875, | |
| "logps/rejected": -186.2373504638672, | |
| "loss": 0.6918, | |
| "rewards/accuracies": 0.5306249856948853, | |
| "rewards/chosen": -0.7291114926338196, | |
| "rewards/margins": 0.02977893128991127, | |
| "rewards/rejected": -0.7588903903961182, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.31840796019900497, | |
| "grad_norm": 1.2193249464035034, | |
| "learning_rate": 3.849688575211836e-07, | |
| "logits/chosen": -1.4531009197235107, | |
| "logits/rejected": -1.528568148612976, | |
| "logps/chosen": -193.5086669921875, | |
| "logps/rejected": -190.48219299316406, | |
| "loss": 0.6947, | |
| "rewards/accuracies": 0.5360937714576721, | |
| "rewards/chosen": -0.7271183729171753, | |
| "rewards/margins": 0.027006981894373894, | |
| "rewards/rejected": -0.7541253566741943, | |
| "step": 800 | |
| } | |
| ], | |
| "logging_steps": 100, | |
| "max_steps": 2512, | |
| "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": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |