| { | |
| "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.018957345971563982, | |
| "grad_norm": 133.64062565621384, | |
| "learning_rate": 4.545454545454545e-08, | |
| "logits/chosen": 119.0696792602539, | |
| "logits/rejected": 120.28123474121094, | |
| "logps/chosen": -394.1268310546875, | |
| "logps/rejected": -419.3145446777344, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.1895734597156398, | |
| "grad_norm": 130.60842697521545, | |
| "learning_rate": 4.545454545454545e-07, | |
| "logits/chosen": 133.6595001220703, | |
| "logits/rejected": 136.7303466796875, | |
| "logps/chosen": -410.0771484375, | |
| "logps/rejected": -445.1907653808594, | |
| "loss": 0.7019, | |
| "rewards/accuracies": 0.46875, | |
| "rewards/chosen": 0.020121444016695023, | |
| "rewards/margins": 0.041466910392045975, | |
| "rewards/rejected": -0.021345460787415504, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.3791469194312796, | |
| "grad_norm": 127.29787487076526, | |
| "learning_rate": 4.885348141000122e-07, | |
| "logits/chosen": 122.2022476196289, | |
| "logits/rejected": 128.57586669921875, | |
| "logps/chosen": -357.1582336425781, | |
| "logps/rejected": -416.08087158203125, | |
| "loss": 0.6346, | |
| "rewards/accuracies": 0.653124988079071, | |
| "rewards/chosen": 0.25420495867729187, | |
| "rewards/margins": 0.4108888506889343, | |
| "rewards/rejected": -0.15668384730815887, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.5687203791469194, | |
| "grad_norm": 110.05011163607695, | |
| "learning_rate": 4.5025027361734613e-07, | |
| "logits/chosen": 121.9586181640625, | |
| "logits/rejected": 125.2878646850586, | |
| "logps/chosen": -387.713134765625, | |
| "logps/rejected": -442.55206298828125, | |
| "loss": 0.5698, | |
| "rewards/accuracies": 0.703125, | |
| "rewards/chosen": -1.2848999500274658, | |
| "rewards/margins": 0.9555079340934753, | |
| "rewards/rejected": -2.240407943725586, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.7582938388625592, | |
| "grad_norm": 111.08969508053838, | |
| "learning_rate": 3.893311157806091e-07, | |
| "logits/chosen": 121.52265930175781, | |
| "logits/rejected": 119.2688980102539, | |
| "logps/chosen": -402.15716552734375, | |
| "logps/rejected": -444.649169921875, | |
| "loss": 0.5496, | |
| "rewards/accuracies": 0.7593749761581421, | |
| "rewards/chosen": -2.0494799613952637, | |
| "rewards/margins": 1.070623517036438, | |
| "rewards/rejected": -3.120103359222412, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.9478672985781991, | |
| "grad_norm": 122.82358054602282, | |
| "learning_rate": 3.126631330646801e-07, | |
| "logits/chosen": 128.3933868408203, | |
| "logits/rejected": 133.44308471679688, | |
| "logps/chosen": -431.0421447753906, | |
| "logps/rejected": -497.99420166015625, | |
| "loss": 0.4959, | |
| "rewards/accuracies": 0.746874988079071, | |
| "rewards/chosen": -2.2058186531066895, | |
| "rewards/margins": 1.2984471321105957, | |
| "rewards/rejected": -3.504265546798706, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.1374407582938388, | |
| "grad_norm": 65.56687198861316, | |
| "learning_rate": 2.2891223348923882e-07, | |
| "logits/chosen": 124.70857238769531, | |
| "logits/rejected": 126.91219329833984, | |
| "logps/chosen": -420.981201171875, | |
| "logps/rejected": -505.5345153808594, | |
| "loss": 0.307, | |
| "rewards/accuracies": 0.893750011920929, | |
| "rewards/chosen": -2.4674336910247803, | |
| "rewards/margins": 2.305318832397461, | |
| "rewards/rejected": -4.772752285003662, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.3270142180094786, | |
| "grad_norm": 56.106028687537446, | |
| "learning_rate": 1.4754491880085317e-07, | |
| "logits/chosen": 121.775146484375, | |
| "logits/rejected": 125.95316314697266, | |
| "logps/chosen": -425.7054138183594, | |
| "logps/rejected": -518.8656005859375, | |
| "loss": 0.1907, | |
| "rewards/accuracies": 0.940625011920929, | |
| "rewards/chosen": -2.623661518096924, | |
| "rewards/margins": 2.869920253753662, | |
| "rewards/rejected": -5.493582248687744, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.5165876777251186, | |
| "grad_norm": 50.43661058282089, | |
| "learning_rate": 7.775827023107834e-08, | |
| "logits/chosen": 114.5962142944336, | |
| "logits/rejected": 126.1790771484375, | |
| "logps/chosen": -426.8082580566406, | |
| "logps/rejected": -527.3065185546875, | |
| "loss": 0.1761, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": -2.979158401489258, | |
| "rewards/margins": 3.0644469261169434, | |
| "rewards/rejected": -6.043605804443359, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.7061611374407581, | |
| "grad_norm": 45.81843583580765, | |
| "learning_rate": 2.7440387297912122e-08, | |
| "logits/chosen": 117.46388244628906, | |
| "logits/rejected": 123.80489349365234, | |
| "logps/chosen": -449.65399169921875, | |
| "logps/rejected": -544.6094970703125, | |
| "loss": 0.1515, | |
| "rewards/accuracies": 0.984375, | |
| "rewards/chosen": -2.8478360176086426, | |
| "rewards/margins": 3.247156858444214, | |
| "rewards/rejected": -6.0949931144714355, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.8957345971563981, | |
| "grad_norm": 42.75820426735574, | |
| "learning_rate": 2.27878296044029e-09, | |
| "logits/chosen": 114.7729721069336, | |
| "logits/rejected": 119.34477233886719, | |
| "logps/chosen": -437.2296447753906, | |
| "logps/rejected": -523.9191284179688, | |
| "loss": 0.1578, | |
| "rewards/accuracies": 0.9593750238418579, | |
| "rewards/chosen": -2.8138155937194824, | |
| "rewards/margins": 3.170293092727661, | |
| "rewards/rejected": -5.984108924865723, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.8957345971563981, | |
| "eval_logits/chosen": 91.35408782958984, | |
| "eval_logits/rejected": 94.07221221923828, | |
| "eval_logps/chosen": -428.1683349609375, | |
| "eval_logps/rejected": -515.7637939453125, | |
| "eval_loss": 0.4643263816833496, | |
| "eval_rewards/accuracies": 0.75, | |
| "eval_rewards/chosen": -3.5909416675567627, | |
| "eval_rewards/margins": 1.7481167316436768, | |
| "eval_rewards/rejected": -5.339057922363281, | |
| "eval_runtime": 88.3612, | |
| "eval_samples_per_second": 8.488, | |
| "eval_steps_per_second": 0.532, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.971563981042654, | |
| "step": 104, | |
| "total_flos": 0.0, | |
| "train_loss": 0.3883641087091886, | |
| "train_runtime": 2802.2739, | |
| "train_samples_per_second": 4.818, | |
| "train_steps_per_second": 0.037 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 104, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 500, | |
| "total_flos": 0.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |