| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.971563981042654, | |
| "eval_steps": 100, | |
| "global_step": 104, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 133.0057671141645, | |
| "learning_rate": 4.545454545454545e-08, | |
| "logits/chosen": 123.11854553222656, | |
| "logits/rejected": 97.00198364257812, | |
| "logps/chosen": -425.18585205078125, | |
| "logps/rejected": -424.1869201660156, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 182.25849293095231, | |
| "learning_rate": 4.545454545454545e-07, | |
| "logits/chosen": 117.33037567138672, | |
| "logits/rejected": 136.23989868164062, | |
| "logps/chosen": -442.517333984375, | |
| "logps/rejected": -524.100830078125, | |
| "loss": 0.7193, | |
| "rewards/accuracies": 0.4791666567325592, | |
| "rewards/chosen": 0.04355551674962044, | |
| "rewards/margins": 0.04285159707069397, | |
| "rewards/rejected": 0.0007039175252430141, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "grad_norm": 98.47580983421109, | |
| "learning_rate": 4.885348141000122e-07, | |
| "logits/chosen": 124.34559631347656, | |
| "logits/rejected": 131.87884521484375, | |
| "logps/chosen": -425.8033142089844, | |
| "logps/rejected": -496.58966064453125, | |
| "loss": 0.5981, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.02306445501744747, | |
| "rewards/margins": 0.4770180284976959, | |
| "rewards/rejected": -0.4539535939693451, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "grad_norm": 114.087847259367, | |
| "learning_rate": 4.5025027361734613e-07, | |
| "logits/chosen": 121.997314453125, | |
| "logits/rejected": 125.30668640136719, | |
| "logps/chosen": -473.92950439453125, | |
| "logps/rejected": -546.4301147460938, | |
| "loss": 0.5828, | |
| "rewards/accuracies": 0.731249988079071, | |
| "rewards/chosen": -0.9404676556587219, | |
| "rewards/margins": 1.0203006267547607, | |
| "rewards/rejected": -1.9607683420181274, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "grad_norm": 102.23106994021863, | |
| "learning_rate": 3.893311157806091e-07, | |
| "logits/chosen": 123.2125473022461, | |
| "logits/rejected": 118.11393737792969, | |
| "logps/chosen": -498.98480224609375, | |
| "logps/rejected": -529.0015869140625, | |
| "loss": 0.5432, | |
| "rewards/accuracies": 0.731249988079071, | |
| "rewards/chosen": -1.29017174243927, | |
| "rewards/margins": 0.921674907207489, | |
| "rewards/rejected": -2.2118465900421143, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "grad_norm": 116.89709640403532, | |
| "learning_rate": 3.126631330646801e-07, | |
| "logits/chosen": 131.3624267578125, | |
| "logits/rejected": 131.5041046142578, | |
| "logps/chosen": -517.221923828125, | |
| "logps/rejected": -547.076904296875, | |
| "loss": 0.5007, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": -1.7199589014053345, | |
| "rewards/margins": 1.0714657306671143, | |
| "rewards/rejected": -2.791424512863159, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "grad_norm": 53.67728401735822, | |
| "learning_rate": 2.2891223348923882e-07, | |
| "logits/chosen": 130.25340270996094, | |
| "logits/rejected": 133.3688507080078, | |
| "logps/chosen": -482.68798828125, | |
| "logps/rejected": -531.1278076171875, | |
| "loss": 0.2897, | |
| "rewards/accuracies": 0.8687499761581421, | |
| "rewards/chosen": -1.2961251735687256, | |
| "rewards/margins": 2.0452933311462402, | |
| "rewards/rejected": -3.341418504714966, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "grad_norm": 46.49732419269788, | |
| "learning_rate": 1.4754491880085317e-07, | |
| "logits/chosen": 125.44151306152344, | |
| "logits/rejected": 126.0068588256836, | |
| "logps/chosen": -451.8477478027344, | |
| "logps/rejected": -573.8129272460938, | |
| "loss": 0.1884, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": -1.1466975212097168, | |
| "rewards/margins": 2.4370884895324707, | |
| "rewards/rejected": -3.5837860107421875, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.52, | |
| "grad_norm": 48.86628168194261, | |
| "learning_rate": 7.775827023107834e-08, | |
| "logits/chosen": 129.2606201171875, | |
| "logits/rejected": 130.1272735595703, | |
| "logps/chosen": -510.95819091796875, | |
| "logps/rejected": -606.6770629882812, | |
| "loss": 0.1877, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": -1.0929005146026611, | |
| "rewards/margins": 2.9041507244110107, | |
| "rewards/rejected": -3.9970507621765137, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.71, | |
| "grad_norm": 51.734548215896965, | |
| "learning_rate": 2.7440387297912122e-08, | |
| "logits/chosen": 116.0444564819336, | |
| "logits/rejected": 118.35429382324219, | |
| "logps/chosen": -488.314697265625, | |
| "logps/rejected": -571.8872680664062, | |
| "loss": 0.1774, | |
| "rewards/accuracies": 0.96875, | |
| "rewards/chosen": -1.3073575496673584, | |
| "rewards/margins": 3.0684092044830322, | |
| "rewards/rejected": -4.375766277313232, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "grad_norm": 58.09774424636419, | |
| "learning_rate": 2.27878296044029e-09, | |
| "logits/chosen": 129.8248748779297, | |
| "logits/rejected": 120.5040512084961, | |
| "logps/chosen": -505.01513671875, | |
| "logps/rejected": -610.0936279296875, | |
| "loss": 0.193, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": -1.331251859664917, | |
| "rewards/margins": 3.034428596496582, | |
| "rewards/rejected": -4.36568021774292, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "eval_logits/chosen": 99.29187774658203, | |
| "eval_logits/rejected": 93.2354736328125, | |
| "eval_logps/chosen": -493.443359375, | |
| "eval_logps/rejected": -520.8292236328125, | |
| "eval_loss": 0.4772518575191498, | |
| "eval_rewards/accuracies": 0.7291666865348816, | |
| "eval_rewards/chosen": -2.0019426345825195, | |
| "eval_rewards/margins": 1.4620394706726074, | |
| "eval_rewards/rejected": -3.463982343673706, | |
| "eval_runtime": 49.3044, | |
| "eval_samples_per_second": 15.212, | |
| "eval_steps_per_second": 0.487, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.97, | |
| "step": 104, | |
| "total_flos": 0.0, | |
| "train_loss": 0.3880494168171516, | |
| "train_runtime": 1239.5887, | |
| "train_samples_per_second": 10.891, | |
| "train_steps_per_second": 0.084 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 104, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |