|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 100, |
|
"global_step": 91, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"debug/policy_chosen_logits": 22.537443161010742, |
|
"debug/policy_chosen_logps": -454.7864685058594, |
|
"debug/policy_rejected_logits": 24.08443260192871, |
|
"debug/policy_rejected_logps": -485.6905517578125, |
|
"debug/reference_chosen_logps": -454.7864685058594, |
|
"debug/reference_rejected_logps": -485.6905517578125, |
|
"epoch": 0.01098901098901099, |
|
"grad_norm": 6.833481499698497, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.537443161010742, |
|
"logits/rejected": 24.08443260192871, |
|
"logps/chosen": -454.7864685058594, |
|
"logps/rejected": -485.6905517578125, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.2491512298584, |
|
"debug/policy_chosen_logps": -426.63800048828125, |
|
"debug/policy_rejected_logits": 25.061065673828125, |
|
"debug/policy_rejected_logps": -446.8348083496094, |
|
"debug/reference_chosen_logps": -426.80908203125, |
|
"debug/reference_rejected_logps": -446.17877197265625, |
|
"epoch": 0.02197802197802198, |
|
"grad_norm": 8.518360012009484, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.2491512298584, |
|
"logits/rejected": 25.061065673828125, |
|
"logps/chosen": -426.63800048828125, |
|
"logps/rejected": -446.8348083496094, |
|
"loss": 0.5003, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.0017108535394072533, |
|
"rewards/margins": 0.008270950056612492, |
|
"rewards/rejected": -0.006560096517205238, |
|
"step": 2 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.04501724243164, |
|
"debug/policy_chosen_logps": -418.5323791503906, |
|
"debug/policy_rejected_logits": 26.21137809753418, |
|
"debug/policy_rejected_logps": -445.2972412109375, |
|
"debug/reference_chosen_logps": -418.4830627441406, |
|
"debug/reference_rejected_logps": -445.174560546875, |
|
"epoch": 0.03296703296703297, |
|
"grad_norm": 8.234274066877381, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.04501724243164, |
|
"logits/rejected": 26.21137809753418, |
|
"logps/chosen": -418.5323791503906, |
|
"logps/rejected": -445.2972412109375, |
|
"loss": 0.497, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.0004932782612740993, |
|
"rewards/margins": 0.0007336426060646772, |
|
"rewards/rejected": -0.001226921333000064, |
|
"step": 3 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.864124298095703, |
|
"debug/policy_chosen_logps": -424.31500244140625, |
|
"debug/policy_rejected_logits": 24.804826736450195, |
|
"debug/policy_rejected_logps": -440.4050598144531, |
|
"debug/reference_chosen_logps": -424.2529296875, |
|
"debug/reference_rejected_logps": -440.0723876953125, |
|
"epoch": 0.04395604395604396, |
|
"grad_norm": 8.267600116778256, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.864124298095703, |
|
"logits/rejected": 24.804826736450195, |
|
"logps/chosen": -424.31500244140625, |
|
"logps/rejected": -440.4050598144531, |
|
"loss": 0.4948, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.000620613107457757, |
|
"rewards/margins": 0.0027059551794081926, |
|
"rewards/rejected": -0.0033265682868659496, |
|
"step": 4 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.74032211303711, |
|
"debug/policy_chosen_logps": -425.2148132324219, |
|
"debug/policy_rejected_logits": 24.66983985900879, |
|
"debug/policy_rejected_logps": -428.9949951171875, |
|
"debug/reference_chosen_logps": -425.707763671875, |
|
"debug/reference_rejected_logps": -427.61578369140625, |
|
"epoch": 0.054945054945054944, |
|
"grad_norm": 7.1377920370536465, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.74032211303711, |
|
"logits/rejected": 24.66983985900879, |
|
"logps/chosen": -425.2148132324219, |
|
"logps/rejected": -428.9949951171875, |
|
"loss": 0.4915, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.004929542541503906, |
|
"rewards/margins": 0.018722152337431908, |
|
"rewards/rejected": -0.013792609795928001, |
|
"step": 5 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.908384323120117, |
|
"debug/policy_chosen_logps": -436.1788635253906, |
|
"debug/policy_rejected_logits": 26.066926956176758, |
|
"debug/policy_rejected_logps": -458.21295166015625, |
|
"debug/reference_chosen_logps": -437.0372619628906, |
|
"debug/reference_rejected_logps": -455.97064208984375, |
|
"epoch": 0.06593406593406594, |
|
"grad_norm": 6.840813228978009, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.908384323120117, |
|
"logits/rejected": 26.066926956176758, |
|
"logps/chosen": -436.1788635253906, |
|
"logps/rejected": -458.21295166015625, |
|
"loss": 0.4897, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.008583908900618553, |
|
"rewards/margins": 0.031006699427962303, |
|
"rewards/rejected": -0.02242279052734375, |
|
"step": 6 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.81578826904297, |
|
"debug/policy_chosen_logps": -456.6412353515625, |
|
"debug/policy_rejected_logits": 22.0266170501709, |
|
"debug/policy_rejected_logps": -461.65155029296875, |
|
"debug/reference_chosen_logps": -457.8715515136719, |
|
"debug/reference_rejected_logps": -459.9510498046875, |
|
"epoch": 0.07692307692307693, |
|
"grad_norm": 7.815334550310639, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.81578826904297, |
|
"logits/rejected": 22.0266170501709, |
|
"logps/chosen": -456.6412353515625, |
|
"logps/rejected": -461.65155029296875, |
|
"loss": 0.4758, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.012303275987505913, |
|
"rewards/margins": 0.029308240860700607, |
|
"rewards/rejected": -0.017004966735839844, |
|
"step": 7 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.05910301208496, |
|
"debug/policy_chosen_logps": -420.5210876464844, |
|
"debug/policy_rejected_logits": 21.690637588500977, |
|
"debug/policy_rejected_logps": -461.24737548828125, |
|
"debug/reference_chosen_logps": -421.00885009765625, |
|
"debug/reference_rejected_logps": -459.6142578125, |
|
"epoch": 0.08791208791208792, |
|
"grad_norm": 6.940663555739121, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.05910301208496, |
|
"logits/rejected": 21.690637588500977, |
|
"logps/chosen": -420.5210876464844, |
|
"logps/rejected": -461.24737548828125, |
|
"loss": 0.4748, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": 0.004877815023064613, |
|
"rewards/margins": 0.02120864763855934, |
|
"rewards/rejected": -0.016330832615494728, |
|
"step": 8 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.456287384033203, |
|
"debug/policy_chosen_logps": -440.22540283203125, |
|
"debug/policy_rejected_logits": 26.438932418823242, |
|
"debug/policy_rejected_logps": -419.1558837890625, |
|
"debug/reference_chosen_logps": -439.4281921386719, |
|
"debug/reference_rejected_logps": -417.99652099609375, |
|
"epoch": 0.0989010989010989, |
|
"grad_norm": 5.7812586478558865, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.456287384033203, |
|
"logits/rejected": 26.438932418823242, |
|
"logps/chosen": -440.22540283203125, |
|
"logps/rejected": -419.1558837890625, |
|
"loss": 0.4769, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.007972106337547302, |
|
"rewards/margins": 0.003621369134634733, |
|
"rewards/rejected": -0.011593475937843323, |
|
"step": 9 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.315250396728516, |
|
"debug/policy_chosen_logps": -425.8302001953125, |
|
"debug/policy_rejected_logits": 28.707416534423828, |
|
"debug/policy_rejected_logps": -453.5276184082031, |
|
"debug/reference_chosen_logps": -427.7785339355469, |
|
"debug/reference_rejected_logps": -450.7994079589844, |
|
"epoch": 0.10989010989010989, |
|
"grad_norm": 6.765369287722621, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.315250396728516, |
|
"logits/rejected": 28.707416534423828, |
|
"logps/chosen": -425.8302001953125, |
|
"logps/rejected": -453.5276184082031, |
|
"loss": 0.4488, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.019483529031276703, |
|
"rewards/margins": 0.04676567018032074, |
|
"rewards/rejected": -0.027282143011689186, |
|
"step": 10 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.35513687133789, |
|
"debug/policy_chosen_logps": -409.0987548828125, |
|
"debug/policy_rejected_logits": 25.19911003112793, |
|
"debug/policy_rejected_logps": -457.9080810546875, |
|
"debug/reference_chosen_logps": -411.0341796875, |
|
"debug/reference_rejected_logps": -452.00286865234375, |
|
"epoch": 0.12087912087912088, |
|
"grad_norm": 6.194229784289876, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.35513687133789, |
|
"logits/rejected": 25.19911003112793, |
|
"logps/chosen": -409.0987548828125, |
|
"logps/rejected": -457.9080810546875, |
|
"loss": 0.4512, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.01935398206114769, |
|
"rewards/margins": 0.07840606570243835, |
|
"rewards/rejected": -0.059052083641290665, |
|
"step": 11 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.74183464050293, |
|
"debug/policy_chosen_logps": -443.5227355957031, |
|
"debug/policy_rejected_logits": 23.557279586791992, |
|
"debug/policy_rejected_logps": -438.05023193359375, |
|
"debug/reference_chosen_logps": -444.56549072265625, |
|
"debug/reference_rejected_logps": -433.29443359375, |
|
"epoch": 0.13186813186813187, |
|
"grad_norm": 6.295588890883081, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.74183464050293, |
|
"logits/rejected": 23.557279586791992, |
|
"logps/chosen": -443.5227355957031, |
|
"logps/rejected": -438.05023193359375, |
|
"loss": 0.444, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.010427666828036308, |
|
"rewards/margins": 0.057986069470644, |
|
"rewards/rejected": -0.04755840077996254, |
|
"step": 12 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.26060676574707, |
|
"debug/policy_chosen_logps": -401.9018249511719, |
|
"debug/policy_rejected_logits": 25.899250030517578, |
|
"debug/policy_rejected_logps": -447.271240234375, |
|
"debug/reference_chosen_logps": -407.1501159667969, |
|
"debug/reference_rejected_logps": -443.3197326660156, |
|
"epoch": 0.14285714285714285, |
|
"grad_norm": 6.651428295339092, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.26060676574707, |
|
"logits/rejected": 25.899250030517578, |
|
"logps/chosen": -401.9018249511719, |
|
"logps/rejected": -447.271240234375, |
|
"loss": 0.4438, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.05248305946588516, |
|
"rewards/margins": 0.0919981375336647, |
|
"rewards/rejected": -0.03951507434248924, |
|
"step": 13 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.350528717041016, |
|
"debug/policy_chosen_logps": -431.6693115234375, |
|
"debug/policy_rejected_logits": 27.774742126464844, |
|
"debug/policy_rejected_logps": -437.4266052246094, |
|
"debug/reference_chosen_logps": -437.1637268066406, |
|
"debug/reference_rejected_logps": -433.93536376953125, |
|
"epoch": 0.15384615384615385, |
|
"grad_norm": 6.44432847940225, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.350528717041016, |
|
"logits/rejected": 27.774742126464844, |
|
"logps/chosen": -431.6693115234375, |
|
"logps/rejected": -437.4266052246094, |
|
"loss": 0.4203, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.05494399741292, |
|
"rewards/margins": 0.08985621482133865, |
|
"rewards/rejected": -0.034912221133708954, |
|
"step": 14 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.464439392089844, |
|
"debug/policy_chosen_logps": -441.0960998535156, |
|
"debug/policy_rejected_logits": 22.606828689575195, |
|
"debug/policy_rejected_logps": -450.34637451171875, |
|
"debug/reference_chosen_logps": -443.2272033691406, |
|
"debug/reference_rejected_logps": -445.57098388671875, |
|
"epoch": 0.16483516483516483, |
|
"grad_norm": 6.034773405123565, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.464439392089844, |
|
"logits/rejected": 22.606828689575195, |
|
"logps/chosen": -441.0960998535156, |
|
"logps/rejected": -450.34637451171875, |
|
"loss": 0.4508, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": 0.021311109885573387, |
|
"rewards/margins": 0.06906504929065704, |
|
"rewards/rejected": -0.047753941267728806, |
|
"step": 15 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.063968658447266, |
|
"debug/policy_chosen_logps": -430.00677490234375, |
|
"debug/policy_rejected_logits": 30.106121063232422, |
|
"debug/policy_rejected_logps": -459.87164306640625, |
|
"debug/reference_chosen_logps": -435.9080810546875, |
|
"debug/reference_rejected_logps": -454.621337890625, |
|
"epoch": 0.17582417582417584, |
|
"grad_norm": 5.694131268125543, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.063968658447266, |
|
"logits/rejected": 30.106121063232422, |
|
"logps/chosen": -430.00677490234375, |
|
"logps/rejected": -459.87164306640625, |
|
"loss": 0.4059, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.05901290848851204, |
|
"rewards/margins": 0.11151611059904099, |
|
"rewards/rejected": -0.05250319838523865, |
|
"step": 16 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 29.839580535888672, |
|
"debug/policy_chosen_logps": -436.97039794921875, |
|
"debug/policy_rejected_logits": 28.98643684387207, |
|
"debug/policy_rejected_logps": -476.59881591796875, |
|
"debug/reference_chosen_logps": -443.8123779296875, |
|
"debug/reference_rejected_logps": -466.94873046875, |
|
"epoch": 0.18681318681318682, |
|
"grad_norm": 6.738937580645961, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 29.839580535888672, |
|
"logits/rejected": 28.98643684387207, |
|
"logps/chosen": -436.97039794921875, |
|
"logps/rejected": -476.59881591796875, |
|
"loss": 0.4118, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.06841972470283508, |
|
"rewards/margins": 0.16492080688476562, |
|
"rewards/rejected": -0.09650108218193054, |
|
"step": 17 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 29.84876823425293, |
|
"debug/policy_chosen_logps": -441.1398620605469, |
|
"debug/policy_rejected_logits": 21.603878021240234, |
|
"debug/policy_rejected_logps": -443.14556884765625, |
|
"debug/reference_chosen_logps": -448.20770263671875, |
|
"debug/reference_rejected_logps": -441.1372375488281, |
|
"epoch": 0.1978021978021978, |
|
"grad_norm": 5.137388346436403, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 29.84876823425293, |
|
"logits/rejected": 21.603878021240234, |
|
"logps/chosen": -441.1398620605469, |
|
"logps/rejected": -443.14556884765625, |
|
"loss": 0.4207, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.07067855447530746, |
|
"rewards/margins": 0.09076160192489624, |
|
"rewards/rejected": -0.020083043724298477, |
|
"step": 18 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.974803924560547, |
|
"debug/policy_chosen_logps": -453.483642578125, |
|
"debug/policy_rejected_logits": 26.663101196289062, |
|
"debug/policy_rejected_logps": -489.6366271972656, |
|
"debug/reference_chosen_logps": -460.9517822265625, |
|
"debug/reference_rejected_logps": -470.0023193359375, |
|
"epoch": 0.2087912087912088, |
|
"grad_norm": 7.17158785388267, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.974803924560547, |
|
"logits/rejected": 26.663101196289062, |
|
"logps/chosen": -453.483642578125, |
|
"logps/rejected": -489.6366271972656, |
|
"loss": 0.3364, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.07468143105506897, |
|
"rewards/margins": 0.271024614572525, |
|
"rewards/rejected": -0.19634319841861725, |
|
"step": 19 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.026630401611328, |
|
"debug/policy_chosen_logps": -433.65960693359375, |
|
"debug/policy_rejected_logits": 25.77982521057129, |
|
"debug/policy_rejected_logps": -471.0029296875, |
|
"debug/reference_chosen_logps": -436.43359375, |
|
"debug/reference_rejected_logps": -460.9761962890625, |
|
"epoch": 0.21978021978021978, |
|
"grad_norm": 4.898911355829737, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.026630401611328, |
|
"logits/rejected": 25.77982521057129, |
|
"logps/chosen": -433.65960693359375, |
|
"logps/rejected": -471.0029296875, |
|
"loss": 0.3395, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.027739638462662697, |
|
"rewards/margins": 0.12800678610801697, |
|
"rewards/rejected": -0.10026714205741882, |
|
"step": 20 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.650062561035156, |
|
"debug/policy_chosen_logps": -435.83905029296875, |
|
"debug/policy_rejected_logits": 24.800756454467773, |
|
"debug/policy_rejected_logps": -481.5117492675781, |
|
"debug/reference_chosen_logps": -443.297607421875, |
|
"debug/reference_rejected_logps": -455.4820556640625, |
|
"epoch": 0.23076923076923078, |
|
"grad_norm": 5.689326640738984, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.650062561035156, |
|
"logits/rejected": 24.800756454467773, |
|
"logps/chosen": -435.83905029296875, |
|
"logps/rejected": -481.5117492675781, |
|
"loss": 0.3623, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.07458514720201492, |
|
"rewards/margins": 0.3348817229270935, |
|
"rewards/rejected": -0.2602965533733368, |
|
"step": 21 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.519716262817383, |
|
"debug/policy_chosen_logps": -405.88226318359375, |
|
"debug/policy_rejected_logits": 27.31670570373535, |
|
"debug/policy_rejected_logps": -470.7935791015625, |
|
"debug/reference_chosen_logps": -416.4732360839844, |
|
"debug/reference_rejected_logps": -464.93170166015625, |
|
"epoch": 0.24175824175824176, |
|
"grad_norm": 4.9841849550976765, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.519716262817383, |
|
"logits/rejected": 27.31670570373535, |
|
"logps/chosen": -405.88226318359375, |
|
"logps/rejected": -470.7935791015625, |
|
"loss": 0.3153, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.10590964555740356, |
|
"rewards/margins": 0.16452865302562714, |
|
"rewards/rejected": -0.058619000017642975, |
|
"step": 22 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.54558753967285, |
|
"debug/policy_chosen_logps": -427.68450927734375, |
|
"debug/policy_rejected_logits": 28.938701629638672, |
|
"debug/policy_rejected_logps": -456.0726623535156, |
|
"debug/reference_chosen_logps": -445.5054626464844, |
|
"debug/reference_rejected_logps": -441.0499267578125, |
|
"epoch": 0.25274725274725274, |
|
"grad_norm": 4.453741538704258, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.54558753967285, |
|
"logits/rejected": 28.938701629638672, |
|
"logps/chosen": -427.68450927734375, |
|
"logps/rejected": -456.0726623535156, |
|
"loss": 0.3437, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.17820952832698822, |
|
"rewards/margins": 0.328436940908432, |
|
"rewards/rejected": -0.1502273976802826, |
|
"step": 23 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.3718318939209, |
|
"debug/policy_chosen_logps": -407.55877685546875, |
|
"debug/policy_rejected_logits": 25.362825393676758, |
|
"debug/policy_rejected_logps": -428.3117370605469, |
|
"debug/reference_chosen_logps": -422.75518798828125, |
|
"debug/reference_rejected_logps": -419.3153381347656, |
|
"epoch": 0.26373626373626374, |
|
"grad_norm": 5.345678599763925, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.3718318939209, |
|
"logits/rejected": 25.362825393676758, |
|
"logps/chosen": -407.55877685546875, |
|
"logps/rejected": -428.3117370605469, |
|
"loss": 0.3313, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.1519637256860733, |
|
"rewards/margins": 0.24192774295806885, |
|
"rewards/rejected": -0.08996403217315674, |
|
"step": 24 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.36195945739746, |
|
"debug/policy_chosen_logps": -407.60107421875, |
|
"debug/policy_rejected_logits": 21.827068328857422, |
|
"debug/policy_rejected_logps": -436.4155578613281, |
|
"debug/reference_chosen_logps": -428.3585205078125, |
|
"debug/reference_rejected_logps": -423.5442199707031, |
|
"epoch": 0.27472527472527475, |
|
"grad_norm": 6.983328118149161, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.36195945739746, |
|
"logits/rejected": 21.827068328857422, |
|
"logps/chosen": -407.60107421875, |
|
"logps/rejected": -436.4155578613281, |
|
"loss": 0.3456, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.20757438242435455, |
|
"rewards/margins": 0.3362876772880554, |
|
"rewards/rejected": -0.12871329486370087, |
|
"step": 25 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.60111427307129, |
|
"debug/policy_chosen_logps": -424.568115234375, |
|
"debug/policy_rejected_logits": 27.44658088684082, |
|
"debug/policy_rejected_logps": -500.7226867675781, |
|
"debug/reference_chosen_logps": -442.6761779785156, |
|
"debug/reference_rejected_logps": -472.3526916503906, |
|
"epoch": 0.2857142857142857, |
|
"grad_norm": 4.621624916905314, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.60111427307129, |
|
"logits/rejected": 27.44658088684082, |
|
"logps/chosen": -424.568115234375, |
|
"logps/rejected": -500.7226867675781, |
|
"loss": 0.3364, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.18108057975769043, |
|
"rewards/margins": 0.46478039026260376, |
|
"rewards/rejected": -0.28369981050491333, |
|
"step": 26 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.04281234741211, |
|
"debug/policy_chosen_logps": -420.4295349121094, |
|
"debug/policy_rejected_logits": 24.854686737060547, |
|
"debug/policy_rejected_logps": -453.5174560546875, |
|
"debug/reference_chosen_logps": -440.0169677734375, |
|
"debug/reference_rejected_logps": -441.5909118652344, |
|
"epoch": 0.2967032967032967, |
|
"grad_norm": 4.722287417720129, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.04281234741211, |
|
"logits/rejected": 24.854686737060547, |
|
"logps/chosen": -420.4295349121094, |
|
"logps/rejected": -453.5174560546875, |
|
"loss": 0.3288, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.19587397575378418, |
|
"rewards/margins": 0.3151397109031677, |
|
"rewards/rejected": -0.11926574259996414, |
|
"step": 27 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.117721557617188, |
|
"debug/policy_chosen_logps": -437.7601318359375, |
|
"debug/policy_rejected_logits": 28.745031356811523, |
|
"debug/policy_rejected_logps": -477.95904541015625, |
|
"debug/reference_chosen_logps": -448.17889404296875, |
|
"debug/reference_rejected_logps": -449.2839050292969, |
|
"epoch": 0.3076923076923077, |
|
"grad_norm": 4.51629611753504, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.117721557617188, |
|
"logits/rejected": 28.745031356811523, |
|
"logps/chosen": -437.7601318359375, |
|
"logps/rejected": -477.95904541015625, |
|
"loss": 0.3783, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.10418765991926193, |
|
"rewards/margins": 0.3909391462802887, |
|
"rewards/rejected": -0.28675150871276855, |
|
"step": 28 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.643272399902344, |
|
"debug/policy_chosen_logps": -445.78875732421875, |
|
"debug/policy_rejected_logits": 30.509065628051758, |
|
"debug/policy_rejected_logps": -490.1319274902344, |
|
"debug/reference_chosen_logps": -458.77532958984375, |
|
"debug/reference_rejected_logps": -467.02667236328125, |
|
"epoch": 0.31868131868131866, |
|
"grad_norm": 5.726900857769083, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.643272399902344, |
|
"logits/rejected": 30.509065628051758, |
|
"logps/chosen": -445.78875732421875, |
|
"logps/rejected": -490.1319274902344, |
|
"loss": 0.3291, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.1298656463623047, |
|
"rewards/margins": 0.3609181046485901, |
|
"rewards/rejected": -0.2310524582862854, |
|
"step": 29 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.9057674407959, |
|
"debug/policy_chosen_logps": -429.0777587890625, |
|
"debug/policy_rejected_logits": 27.041988372802734, |
|
"debug/policy_rejected_logps": -489.99102783203125, |
|
"debug/reference_chosen_logps": -435.3116149902344, |
|
"debug/reference_rejected_logps": -468.8659973144531, |
|
"epoch": 0.32967032967032966, |
|
"grad_norm": 5.589091335127157, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.9057674407959, |
|
"logits/rejected": 27.041988372802734, |
|
"logps/chosen": -429.0777587890625, |
|
"logps/rejected": -489.99102783203125, |
|
"loss": 0.3149, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.06233878806233406, |
|
"rewards/margins": 0.2735891342163086, |
|
"rewards/rejected": -0.21125034987926483, |
|
"step": 30 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 20.52339744567871, |
|
"debug/policy_chosen_logps": -417.83367919921875, |
|
"debug/policy_rejected_logits": 25.475711822509766, |
|
"debug/policy_rejected_logps": -477.3974914550781, |
|
"debug/reference_chosen_logps": -429.0983581542969, |
|
"debug/reference_rejected_logps": -454.0728759765625, |
|
"epoch": 0.34065934065934067, |
|
"grad_norm": 5.919614811999328, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 20.52339744567871, |
|
"logits/rejected": 25.475711822509766, |
|
"logps/chosen": -417.83367919921875, |
|
"logps/rejected": -477.3974914550781, |
|
"loss": 0.2959, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.11264674365520477, |
|
"rewards/margins": 0.34589269757270813, |
|
"rewards/rejected": -0.23324593901634216, |
|
"step": 31 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.121295928955078, |
|
"debug/policy_chosen_logps": -410.53643798828125, |
|
"debug/policy_rejected_logits": 27.618637084960938, |
|
"debug/policy_rejected_logps": -471.409912109375, |
|
"debug/reference_chosen_logps": -427.6445007324219, |
|
"debug/reference_rejected_logps": -452.9000244140625, |
|
"epoch": 0.3516483516483517, |
|
"grad_norm": 6.235065911009125, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.121295928955078, |
|
"logits/rejected": 27.618637084960938, |
|
"logps/chosen": -410.53643798828125, |
|
"logps/rejected": -471.409912109375, |
|
"loss": 0.3322, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.1710808128118515, |
|
"rewards/margins": 0.35617977380752563, |
|
"rewards/rejected": -0.18509894609451294, |
|
"step": 32 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.020137786865234, |
|
"debug/policy_chosen_logps": -421.2449951171875, |
|
"debug/policy_rejected_logits": 26.06194305419922, |
|
"debug/policy_rejected_logps": -455.33154296875, |
|
"debug/reference_chosen_logps": -445.73223876953125, |
|
"debug/reference_rejected_logps": -430.2513427734375, |
|
"epoch": 0.3626373626373626, |
|
"grad_norm": 4.220132361607984, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.020137786865234, |
|
"logits/rejected": 26.06194305419922, |
|
"logps/chosen": -421.2449951171875, |
|
"logps/rejected": -455.33154296875, |
|
"loss": 0.2825, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.24487201869487762, |
|
"rewards/margins": 0.4956740140914917, |
|
"rewards/rejected": -0.25080201029777527, |
|
"step": 33 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.49104881286621, |
|
"debug/policy_chosen_logps": -432.42578125, |
|
"debug/policy_rejected_logits": 26.045143127441406, |
|
"debug/policy_rejected_logps": -500.81494140625, |
|
"debug/reference_chosen_logps": -455.1983642578125, |
|
"debug/reference_rejected_logps": -481.32403564453125, |
|
"epoch": 0.37362637362637363, |
|
"grad_norm": 4.102855456579325, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.49104881286621, |
|
"logits/rejected": 26.045143127441406, |
|
"logps/chosen": -432.42578125, |
|
"logps/rejected": -500.81494140625, |
|
"loss": 0.2778, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.22772569954395294, |
|
"rewards/margins": 0.422635018825531, |
|
"rewards/rejected": -0.19490931928157806, |
|
"step": 34 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.481491088867188, |
|
"debug/policy_chosen_logps": -410.7614440917969, |
|
"debug/policy_rejected_logits": 26.54258155822754, |
|
"debug/policy_rejected_logps": -461.57659912109375, |
|
"debug/reference_chosen_logps": -431.06304931640625, |
|
"debug/reference_rejected_logps": -428.4720458984375, |
|
"epoch": 0.38461538461538464, |
|
"grad_norm": 3.930918282446488, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.481491088867188, |
|
"logits/rejected": 26.54258155822754, |
|
"logps/chosen": -410.7614440917969, |
|
"logps/rejected": -461.57659912109375, |
|
"loss": 0.2688, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.20301616191864014, |
|
"rewards/margins": 0.5340613126754761, |
|
"rewards/rejected": -0.33104515075683594, |
|
"step": 35 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.361486434936523, |
|
"debug/policy_chosen_logps": -412.67987060546875, |
|
"debug/policy_rejected_logits": 27.827251434326172, |
|
"debug/policy_rejected_logps": -452.28857421875, |
|
"debug/reference_chosen_logps": -430.6239318847656, |
|
"debug/reference_rejected_logps": -440.8873596191406, |
|
"epoch": 0.3956043956043956, |
|
"grad_norm": 5.7094820058187645, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.361486434936523, |
|
"logits/rejected": 27.827251434326172, |
|
"logps/chosen": -412.67987060546875, |
|
"logps/rejected": -452.28857421875, |
|
"loss": 0.3088, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.1794406771659851, |
|
"rewards/margins": 0.29345306754112244, |
|
"rewards/rejected": -0.11401237547397614, |
|
"step": 36 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.789180755615234, |
|
"debug/policy_chosen_logps": -419.154296875, |
|
"debug/policy_rejected_logits": 24.381994247436523, |
|
"debug/policy_rejected_logps": -443.3766174316406, |
|
"debug/reference_chosen_logps": -440.4947509765625, |
|
"debug/reference_rejected_logps": -457.65887451171875, |
|
"epoch": 0.4065934065934066, |
|
"grad_norm": 3.962286836465319, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.789180755615234, |
|
"logits/rejected": 24.381994247436523, |
|
"logps/chosen": -419.154296875, |
|
"logps/rejected": -443.3766174316406, |
|
"loss": 0.2823, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.21340444684028625, |
|
"rewards/margins": 0.07058170437812805, |
|
"rewards/rejected": 0.1428227573633194, |
|
"step": 37 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.922195434570312, |
|
"debug/policy_chosen_logps": -399.93096923828125, |
|
"debug/policy_rejected_logits": 27.783912658691406, |
|
"debug/policy_rejected_logps": -472.2608947753906, |
|
"debug/reference_chosen_logps": -427.5108642578125, |
|
"debug/reference_rejected_logps": -462.4599914550781, |
|
"epoch": 0.4175824175824176, |
|
"grad_norm": 4.150324748851986, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.922195434570312, |
|
"logits/rejected": 27.783912658691406, |
|
"logps/chosen": -399.93096923828125, |
|
"logps/rejected": -472.2608947753906, |
|
"loss": 0.2398, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.27579906582832336, |
|
"rewards/margins": 0.37380802631378174, |
|
"rewards/rejected": -0.09800895303487778, |
|
"step": 38 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.96780014038086, |
|
"debug/policy_chosen_logps": -420.74261474609375, |
|
"debug/policy_rejected_logits": 23.919336318969727, |
|
"debug/policy_rejected_logps": -439.653076171875, |
|
"debug/reference_chosen_logps": -433.999755859375, |
|
"debug/reference_rejected_logps": -421.92962646484375, |
|
"epoch": 0.42857142857142855, |
|
"grad_norm": 4.365666660792002, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.96780014038086, |
|
"logits/rejected": 23.919336318969727, |
|
"logps/chosen": -420.74261474609375, |
|
"logps/rejected": -439.653076171875, |
|
"loss": 0.2833, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.13257166743278503, |
|
"rewards/margins": 0.3098059296607971, |
|
"rewards/rejected": -0.17723426222801208, |
|
"step": 39 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.75818634033203, |
|
"debug/policy_chosen_logps": -455.0806884765625, |
|
"debug/policy_rejected_logits": 27.270263671875, |
|
"debug/policy_rejected_logps": -468.50927734375, |
|
"debug/reference_chosen_logps": -462.73138427734375, |
|
"debug/reference_rejected_logps": -473.1136474609375, |
|
"epoch": 0.43956043956043955, |
|
"grad_norm": 4.570887002489462, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.75818634033203, |
|
"logits/rejected": 27.270263671875, |
|
"logps/chosen": -455.0806884765625, |
|
"logps/rejected": -468.50927734375, |
|
"loss": 0.3436, |
|
"rewards/accuracies": 0.375, |
|
"rewards/chosen": 0.07650664448738098, |
|
"rewards/margins": 0.030462883412837982, |
|
"rewards/rejected": 0.04604377597570419, |
|
"step": 40 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.347915649414062, |
|
"debug/policy_chosen_logps": -406.87591552734375, |
|
"debug/policy_rejected_logits": 28.548446655273438, |
|
"debug/policy_rejected_logps": -464.4383544921875, |
|
"debug/reference_chosen_logps": -433.459716796875, |
|
"debug/reference_rejected_logps": -448.8690185546875, |
|
"epoch": 0.45054945054945056, |
|
"grad_norm": 3.677831065051595, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.347915649414062, |
|
"logits/rejected": 28.548446655273438, |
|
"logps/chosen": -406.87591552734375, |
|
"logps/rejected": -464.4383544921875, |
|
"loss": 0.2758, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.26583802700042725, |
|
"rewards/margins": 0.4215315878391266, |
|
"rewards/rejected": -0.15569356083869934, |
|
"step": 41 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.77235221862793, |
|
"debug/policy_chosen_logps": -402.58416748046875, |
|
"debug/policy_rejected_logits": 26.393054962158203, |
|
"debug/policy_rejected_logps": -471.61260986328125, |
|
"debug/reference_chosen_logps": -426.26690673828125, |
|
"debug/reference_rejected_logps": -453.03125, |
|
"epoch": 0.46153846153846156, |
|
"grad_norm": 3.808742477450644, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.77235221862793, |
|
"logits/rejected": 26.393054962158203, |
|
"logps/chosen": -402.58416748046875, |
|
"logps/rejected": -471.61260986328125, |
|
"loss": 0.2739, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.23682719469070435, |
|
"rewards/margins": 0.4226408302783966, |
|
"rewards/rejected": -0.18581363558769226, |
|
"step": 42 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.165273666381836, |
|
"debug/policy_chosen_logps": -409.06304931640625, |
|
"debug/policy_rejected_logits": 26.427730560302734, |
|
"debug/policy_rejected_logps": -448.95489501953125, |
|
"debug/reference_chosen_logps": -434.5325927734375, |
|
"debug/reference_rejected_logps": -412.31060791015625, |
|
"epoch": 0.4725274725274725, |
|
"grad_norm": 7.512415554484962, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.165273666381836, |
|
"logits/rejected": 26.427730560302734, |
|
"logps/chosen": -409.06304931640625, |
|
"logps/rejected": -448.95489501953125, |
|
"loss": 0.3186, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.25469571352005005, |
|
"rewards/margins": 0.6211386322975159, |
|
"rewards/rejected": -0.36644288897514343, |
|
"step": 43 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.82545280456543, |
|
"debug/policy_chosen_logps": -409.12420654296875, |
|
"debug/policy_rejected_logits": 23.266063690185547, |
|
"debug/policy_rejected_logps": -493.0911865234375, |
|
"debug/reference_chosen_logps": -437.0023193359375, |
|
"debug/reference_rejected_logps": -456.59796142578125, |
|
"epoch": 0.4835164835164835, |
|
"grad_norm": 5.005310078975177, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.82545280456543, |
|
"logits/rejected": 23.266063690185547, |
|
"logps/chosen": -409.12420654296875, |
|
"logps/rejected": -493.0911865234375, |
|
"loss": 0.2517, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.278780996799469, |
|
"rewards/margins": 0.6437131762504578, |
|
"rewards/rejected": -0.36493217945098877, |
|
"step": 44 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.779375076293945, |
|
"debug/policy_chosen_logps": -419.06585693359375, |
|
"debug/policy_rejected_logits": 25.911027908325195, |
|
"debug/policy_rejected_logps": -493.4896240234375, |
|
"debug/reference_chosen_logps": -444.6504211425781, |
|
"debug/reference_rejected_logps": -467.607177734375, |
|
"epoch": 0.4945054945054945, |
|
"grad_norm": 5.276153180435082, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.779375076293945, |
|
"logits/rejected": 25.911027908325195, |
|
"logps/chosen": -419.06585693359375, |
|
"logps/rejected": -493.4896240234375, |
|
"loss": 0.3159, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2558458745479584, |
|
"rewards/margins": 0.5146701335906982, |
|
"rewards/rejected": -0.25882428884506226, |
|
"step": 45 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.05695152282715, |
|
"debug/policy_chosen_logps": -428.0582275390625, |
|
"debug/policy_rejected_logits": 27.517444610595703, |
|
"debug/policy_rejected_logps": -499.21636962890625, |
|
"debug/reference_chosen_logps": -452.95599365234375, |
|
"debug/reference_rejected_logps": -468.61138916015625, |
|
"epoch": 0.5054945054945055, |
|
"grad_norm": 4.904417802927379, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.05695152282715, |
|
"logits/rejected": 27.517444610595703, |
|
"logps/chosen": -428.0582275390625, |
|
"logps/rejected": -499.21636962890625, |
|
"loss": 0.242, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.2489778846502304, |
|
"rewards/margins": 0.5550275444984436, |
|
"rewards/rejected": -0.306049644947052, |
|
"step": 46 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.79572105407715, |
|
"debug/policy_chosen_logps": -412.3216552734375, |
|
"debug/policy_rejected_logits": 26.50978660583496, |
|
"debug/policy_rejected_logps": -466.15557861328125, |
|
"debug/reference_chosen_logps": -430.951416015625, |
|
"debug/reference_rejected_logps": -448.00433349609375, |
|
"epoch": 0.5164835164835165, |
|
"grad_norm": 5.334062642354243, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.79572105407715, |
|
"logits/rejected": 26.50978660583496, |
|
"logps/chosen": -412.3216552734375, |
|
"logps/rejected": -466.15557861328125, |
|
"loss": 0.3266, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.18629732728004456, |
|
"rewards/margins": 0.3678096830844879, |
|
"rewards/rejected": -0.18151238560676575, |
|
"step": 47 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.24400520324707, |
|
"debug/policy_chosen_logps": -392.45751953125, |
|
"debug/policy_rejected_logits": 27.10297393798828, |
|
"debug/policy_rejected_logps": -463.1430969238281, |
|
"debug/reference_chosen_logps": -417.45147705078125, |
|
"debug/reference_rejected_logps": -438.20220947265625, |
|
"epoch": 0.5274725274725275, |
|
"grad_norm": 7.066668036628892, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.24400520324707, |
|
"logits/rejected": 27.10297393798828, |
|
"logps/chosen": -392.45751953125, |
|
"logps/rejected": -463.1430969238281, |
|
"loss": 0.3255, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.24993950128555298, |
|
"rewards/margins": 0.49934816360473633, |
|
"rewards/rejected": -0.24940869212150574, |
|
"step": 48 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.664649963378906, |
|
"debug/policy_chosen_logps": -394.15032958984375, |
|
"debug/policy_rejected_logits": 25.70164680480957, |
|
"debug/policy_rejected_logps": -452.9449768066406, |
|
"debug/reference_chosen_logps": -416.6987609863281, |
|
"debug/reference_rejected_logps": -439.2945251464844, |
|
"epoch": 0.5384615384615384, |
|
"grad_norm": 5.135410370061012, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.664649963378906, |
|
"logits/rejected": 25.70164680480957, |
|
"logps/chosen": -394.15032958984375, |
|
"logps/rejected": -452.9449768066406, |
|
"loss": 0.2879, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.22548410296440125, |
|
"rewards/margins": 0.3619886040687561, |
|
"rewards/rejected": -0.13650450110435486, |
|
"step": 49 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.666519165039062, |
|
"debug/policy_chosen_logps": -396.6944885253906, |
|
"debug/policy_rejected_logits": 21.13740348815918, |
|
"debug/policy_rejected_logps": -453.7715148925781, |
|
"debug/reference_chosen_logps": -424.0322265625, |
|
"debug/reference_rejected_logps": -440.89862060546875, |
|
"epoch": 0.5494505494505495, |
|
"grad_norm": 3.9756803394109825, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.666519165039062, |
|
"logits/rejected": 21.13740348815918, |
|
"logps/chosen": -396.6944885253906, |
|
"logps/rejected": -453.7715148925781, |
|
"loss": 0.2293, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.27337703108787537, |
|
"rewards/margins": 0.40210601687431335, |
|
"rewards/rejected": -0.1287289708852768, |
|
"step": 50 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.19964027404785, |
|
"debug/policy_chosen_logps": -397.15765380859375, |
|
"debug/policy_rejected_logits": 24.41975212097168, |
|
"debug/policy_rejected_logps": -475.76104736328125, |
|
"debug/reference_chosen_logps": -424.8116760253906, |
|
"debug/reference_rejected_logps": -434.97528076171875, |
|
"epoch": 0.5604395604395604, |
|
"grad_norm": 4.600442889147922, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.19964027404785, |
|
"logits/rejected": 24.41975212097168, |
|
"logps/chosen": -397.15765380859375, |
|
"logps/rejected": -475.76104736328125, |
|
"loss": 0.2753, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.2765401601791382, |
|
"rewards/margins": 0.6843976974487305, |
|
"rewards/rejected": -0.4078575372695923, |
|
"step": 51 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.027416229248047, |
|
"debug/policy_chosen_logps": -433.012451171875, |
|
"debug/policy_rejected_logits": 24.729970932006836, |
|
"debug/policy_rejected_logps": -465.9927978515625, |
|
"debug/reference_chosen_logps": -459.7962951660156, |
|
"debug/reference_rejected_logps": -439.83599853515625, |
|
"epoch": 0.5714285714285714, |
|
"grad_norm": 3.8308230544471087, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.027416229248047, |
|
"logits/rejected": 24.729970932006836, |
|
"logps/chosen": -433.012451171875, |
|
"logps/rejected": -465.9927978515625, |
|
"loss": 0.2846, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.26783838868141174, |
|
"rewards/margins": 0.529405951499939, |
|
"rewards/rejected": -0.26156753301620483, |
|
"step": 52 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.013721466064453, |
|
"debug/policy_chosen_logps": -409.18426513671875, |
|
"debug/policy_rejected_logits": 29.04503059387207, |
|
"debug/policy_rejected_logps": -455.4191589355469, |
|
"debug/reference_chosen_logps": -434.70855712890625, |
|
"debug/reference_rejected_logps": -437.6434020996094, |
|
"epoch": 0.5824175824175825, |
|
"grad_norm": 5.361663918449101, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.013721466064453, |
|
"logits/rejected": 29.04503059387207, |
|
"logps/chosen": -409.18426513671875, |
|
"logps/rejected": -455.4191589355469, |
|
"loss": 0.3145, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.2552429437637329, |
|
"rewards/margins": 0.433000773191452, |
|
"rewards/rejected": -0.17775781452655792, |
|
"step": 53 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.417736053466797, |
|
"debug/policy_chosen_logps": -419.6263427734375, |
|
"debug/policy_rejected_logits": 28.78829002380371, |
|
"debug/policy_rejected_logps": -475.8312683105469, |
|
"debug/reference_chosen_logps": -440.78143310546875, |
|
"debug/reference_rejected_logps": -462.00067138671875, |
|
"epoch": 0.5934065934065934, |
|
"grad_norm": 4.416192021518405, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.417736053466797, |
|
"logits/rejected": 28.78829002380371, |
|
"logps/chosen": -419.6263427734375, |
|
"logps/rejected": -475.8312683105469, |
|
"loss": 0.3234, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.2115507870912552, |
|
"rewards/margins": 0.34985676407814026, |
|
"rewards/rejected": -0.13830597698688507, |
|
"step": 54 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.22123908996582, |
|
"debug/policy_chosen_logps": -429.47418212890625, |
|
"debug/policy_rejected_logits": 25.79034996032715, |
|
"debug/policy_rejected_logps": -471.1182861328125, |
|
"debug/reference_chosen_logps": -445.8591003417969, |
|
"debug/reference_rejected_logps": -455.14202880859375, |
|
"epoch": 0.6043956043956044, |
|
"grad_norm": 6.671276859718192, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.22123908996582, |
|
"logits/rejected": 25.79034996032715, |
|
"logps/chosen": -429.47418212890625, |
|
"logps/rejected": -471.1182861328125, |
|
"loss": 0.2846, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.16384944319725037, |
|
"rewards/margins": 0.32361170649528503, |
|
"rewards/rejected": -0.15976226329803467, |
|
"step": 55 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.506807327270508, |
|
"debug/policy_chosen_logps": -420.41082763671875, |
|
"debug/policy_rejected_logits": 27.579261779785156, |
|
"debug/policy_rejected_logps": -452.91162109375, |
|
"debug/reference_chosen_logps": -448.4418029785156, |
|
"debug/reference_rejected_logps": -432.2170104980469, |
|
"epoch": 0.6153846153846154, |
|
"grad_norm": 5.601125759934195, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.506807327270508, |
|
"logits/rejected": 27.579261779785156, |
|
"logps/chosen": -420.41082763671875, |
|
"logps/rejected": -452.91162109375, |
|
"loss": 0.2563, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2803099453449249, |
|
"rewards/margins": 0.48725610971450806, |
|
"rewards/rejected": -0.20694613456726074, |
|
"step": 56 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.477113723754883, |
|
"debug/policy_chosen_logps": -410.53009033203125, |
|
"debug/policy_rejected_logits": 24.827415466308594, |
|
"debug/policy_rejected_logps": -463.63214111328125, |
|
"debug/reference_chosen_logps": -430.95086669921875, |
|
"debug/reference_rejected_logps": -428.8712158203125, |
|
"epoch": 0.6263736263736264, |
|
"grad_norm": 4.278326605774298, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.477113723754883, |
|
"logits/rejected": 24.827415466308594, |
|
"logps/chosen": -410.53009033203125, |
|
"logps/rejected": -463.63214111328125, |
|
"loss": 0.2727, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.20420792698860168, |
|
"rewards/margins": 0.5518174171447754, |
|
"rewards/rejected": -0.3476094603538513, |
|
"step": 57 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.6320858001709, |
|
"debug/policy_chosen_logps": -408.2926025390625, |
|
"debug/policy_rejected_logits": 27.864049911499023, |
|
"debug/policy_rejected_logps": -461.57208251953125, |
|
"debug/reference_chosen_logps": -426.1760559082031, |
|
"debug/reference_rejected_logps": -447.0111083984375, |
|
"epoch": 0.6373626373626373, |
|
"grad_norm": 4.643648875549545, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.6320858001709, |
|
"logits/rejected": 27.864049911499023, |
|
"logps/chosen": -408.2926025390625, |
|
"logps/rejected": -461.57208251953125, |
|
"loss": 0.2516, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.17883440852165222, |
|
"rewards/margins": 0.3244439363479614, |
|
"rewards/rejected": -0.1456095427274704, |
|
"step": 58 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 20.885883331298828, |
|
"debug/policy_chosen_logps": -412.9990539550781, |
|
"debug/policy_rejected_logits": 21.050661087036133, |
|
"debug/policy_rejected_logps": -522.6400756835938, |
|
"debug/reference_chosen_logps": -440.10931396484375, |
|
"debug/reference_rejected_logps": -487.7521667480469, |
|
"epoch": 0.6483516483516484, |
|
"grad_norm": 4.05657059697319, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 20.885883331298828, |
|
"logits/rejected": 21.050661087036133, |
|
"logps/chosen": -412.9990539550781, |
|
"logps/rejected": -522.6400756835938, |
|
"loss": 0.2678, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2711024880409241, |
|
"rewards/margins": 0.6199814081192017, |
|
"rewards/rejected": -0.3488789200782776, |
|
"step": 59 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.035409927368164, |
|
"debug/policy_chosen_logps": -399.28765869140625, |
|
"debug/policy_rejected_logits": 24.84585952758789, |
|
"debug/policy_rejected_logps": -453.340576171875, |
|
"debug/reference_chosen_logps": -420.63153076171875, |
|
"debug/reference_rejected_logps": -434.2669677734375, |
|
"epoch": 0.6593406593406593, |
|
"grad_norm": 3.784225958996139, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.035409927368164, |
|
"logits/rejected": 24.84585952758789, |
|
"logps/chosen": -399.28765869140625, |
|
"logps/rejected": -453.340576171875, |
|
"loss": 0.2552, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": 0.2134389877319336, |
|
"rewards/margins": 0.4041747748851776, |
|
"rewards/rejected": -0.19073577225208282, |
|
"step": 60 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.771154403686523, |
|
"debug/policy_chosen_logps": -403.99932861328125, |
|
"debug/policy_rejected_logits": 25.54006576538086, |
|
"debug/policy_rejected_logps": -431.0410461425781, |
|
"debug/reference_chosen_logps": -427.13372802734375, |
|
"debug/reference_rejected_logps": -426.09619140625, |
|
"epoch": 0.6703296703296703, |
|
"grad_norm": 4.4444525496723815, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.771154403686523, |
|
"logits/rejected": 25.54006576538086, |
|
"logps/chosen": -403.99932861328125, |
|
"logps/rejected": -431.0410461425781, |
|
"loss": 0.2411, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2313441038131714, |
|
"rewards/margins": 0.2807927131652832, |
|
"rewards/rejected": -0.04944861680269241, |
|
"step": 61 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.56890869140625, |
|
"debug/policy_chosen_logps": -418.62939453125, |
|
"debug/policy_rejected_logits": 26.892208099365234, |
|
"debug/policy_rejected_logps": -432.6386413574219, |
|
"debug/reference_chosen_logps": -446.4366149902344, |
|
"debug/reference_rejected_logps": -437.1725769042969, |
|
"epoch": 0.6813186813186813, |
|
"grad_norm": 6.207177159902563, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.56890869140625, |
|
"logits/rejected": 26.892208099365234, |
|
"logps/chosen": -418.62939453125, |
|
"logps/rejected": -432.6386413574219, |
|
"loss": 0.2765, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": 0.27807170152664185, |
|
"rewards/margins": 0.23273253440856934, |
|
"rewards/rejected": 0.04533915966749191, |
|
"step": 62 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.32245445251465, |
|
"debug/policy_chosen_logps": -440.50726318359375, |
|
"debug/policy_rejected_logits": 25.365680694580078, |
|
"debug/policy_rejected_logps": -464.0654296875, |
|
"debug/reference_chosen_logps": -475.3460998535156, |
|
"debug/reference_rejected_logps": -457.081787109375, |
|
"epoch": 0.6923076923076923, |
|
"grad_norm": 5.48574215281406, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.32245445251465, |
|
"logits/rejected": 25.365680694580078, |
|
"logps/chosen": -440.50726318359375, |
|
"logps/rejected": -464.0654296875, |
|
"loss": 0.2435, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.34838855266571045, |
|
"rewards/margins": 0.4182246923446655, |
|
"rewards/rejected": -0.06983615458011627, |
|
"step": 63 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.090259552001953, |
|
"debug/policy_chosen_logps": -396.6990051269531, |
|
"debug/policy_rejected_logits": 27.255340576171875, |
|
"debug/policy_rejected_logps": -454.5365295410156, |
|
"debug/reference_chosen_logps": -436.3953552246094, |
|
"debug/reference_rejected_logps": -430.6047058105469, |
|
"epoch": 0.7032967032967034, |
|
"grad_norm": 8.801285897449953, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.090259552001953, |
|
"logits/rejected": 27.255340576171875, |
|
"logps/chosen": -396.6990051269531, |
|
"logps/rejected": -454.5365295410156, |
|
"loss": 0.3064, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.3969634771347046, |
|
"rewards/margins": 0.6362816095352173, |
|
"rewards/rejected": -0.23931819200515747, |
|
"step": 64 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 28.422582626342773, |
|
"debug/policy_chosen_logps": -421.810302734375, |
|
"debug/policy_rejected_logits": 22.969980239868164, |
|
"debug/policy_rejected_logps": -464.9712219238281, |
|
"debug/reference_chosen_logps": -445.0280456542969, |
|
"debug/reference_rejected_logps": -451.85345458984375, |
|
"epoch": 0.7142857142857143, |
|
"grad_norm": 3.3417383455535807, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 28.422582626342773, |
|
"logits/rejected": 22.969980239868164, |
|
"logps/chosen": -421.810302734375, |
|
"logps/rejected": -464.9712219238281, |
|
"loss": 0.216, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2321772277355194, |
|
"rewards/margins": 0.36335471272468567, |
|
"rewards/rejected": -0.13117747008800507, |
|
"step": 65 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.602392196655273, |
|
"debug/policy_chosen_logps": -409.503662109375, |
|
"debug/policy_rejected_logits": 26.957948684692383, |
|
"debug/policy_rejected_logps": -518.37353515625, |
|
"debug/reference_chosen_logps": -443.19598388671875, |
|
"debug/reference_rejected_logps": -475.41815185546875, |
|
"epoch": 0.7252747252747253, |
|
"grad_norm": 7.382953809999285, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.602392196655273, |
|
"logits/rejected": 26.957948684692383, |
|
"logps/chosen": -409.503662109375, |
|
"logps/rejected": -518.37353515625, |
|
"loss": 0.1895, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.3369232416152954, |
|
"rewards/margins": 0.7664777040481567, |
|
"rewards/rejected": -0.42955446243286133, |
|
"step": 66 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.386083602905273, |
|
"debug/policy_chosen_logps": -382.5001220703125, |
|
"debug/policy_rejected_logits": 16.90252113342285, |
|
"debug/policy_rejected_logps": -475.36181640625, |
|
"debug/reference_chosen_logps": -412.29132080078125, |
|
"debug/reference_rejected_logps": -449.7041015625, |
|
"epoch": 0.7362637362637363, |
|
"grad_norm": 5.308279817037437, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.386083602905273, |
|
"logits/rejected": 16.90252113342285, |
|
"logps/chosen": -382.5001220703125, |
|
"logps/rejected": -475.36181640625, |
|
"loss": 0.2588, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.2979119122028351, |
|
"rewards/margins": 0.5544889569282532, |
|
"rewards/rejected": -0.2565770447254181, |
|
"step": 67 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 17.921504974365234, |
|
"debug/policy_chosen_logps": -390.6407470703125, |
|
"debug/policy_rejected_logits": 24.367341995239258, |
|
"debug/policy_rejected_logps": -484.7353515625, |
|
"debug/reference_chosen_logps": -416.8155822753906, |
|
"debug/reference_rejected_logps": -466.807861328125, |
|
"epoch": 0.7472527472527473, |
|
"grad_norm": 5.3343691933569595, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 17.921504974365234, |
|
"logits/rejected": 24.367341995239258, |
|
"logps/chosen": -390.6407470703125, |
|
"logps/rejected": -484.7353515625, |
|
"loss": 0.2477, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.2617482841014862, |
|
"rewards/margins": 0.441023588180542, |
|
"rewards/rejected": -0.17927534878253937, |
|
"step": 68 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 22.636571884155273, |
|
"debug/policy_chosen_logps": -391.5332336425781, |
|
"debug/policy_rejected_logits": 27.57520294189453, |
|
"debug/policy_rejected_logps": -491.03448486328125, |
|
"debug/reference_chosen_logps": -418.2889709472656, |
|
"debug/reference_rejected_logps": -467.211181640625, |
|
"epoch": 0.7582417582417582, |
|
"grad_norm": 4.491987560088097, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 22.636571884155273, |
|
"logits/rejected": 27.57520294189453, |
|
"logps/chosen": -391.5332336425781, |
|
"logps/rejected": -491.03448486328125, |
|
"loss": 0.2792, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.26755744218826294, |
|
"rewards/margins": 0.5057904124259949, |
|
"rewards/rejected": -0.23823297023773193, |
|
"step": 69 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.65935516357422, |
|
"debug/policy_chosen_logps": -403.2398986816406, |
|
"debug/policy_rejected_logits": 29.533405303955078, |
|
"debug/policy_rejected_logps": -492.2855224609375, |
|
"debug/reference_chosen_logps": -433.0833435058594, |
|
"debug/reference_rejected_logps": -444.8511962890625, |
|
"epoch": 0.7692307692307693, |
|
"grad_norm": 3.4461660659523656, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.65935516357422, |
|
"logits/rejected": 29.533405303955078, |
|
"logps/chosen": -403.2398986816406, |
|
"logps/rejected": -492.2855224609375, |
|
"loss": 0.2317, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.29843446612358093, |
|
"rewards/margins": 0.7727776169776917, |
|
"rewards/rejected": -0.47434312105178833, |
|
"step": 70 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.503950119018555, |
|
"debug/policy_chosen_logps": -446.45513916015625, |
|
"debug/policy_rejected_logits": 22.433277130126953, |
|
"debug/policy_rejected_logps": -470.9190979003906, |
|
"debug/reference_chosen_logps": -472.0450134277344, |
|
"debug/reference_rejected_logps": -448.1822814941406, |
|
"epoch": 0.7802197802197802, |
|
"grad_norm": 5.717761852601763, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.503950119018555, |
|
"logits/rejected": 22.433277130126953, |
|
"logps/chosen": -446.45513916015625, |
|
"logps/rejected": -470.9190979003906, |
|
"loss": 0.3119, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.2558988630771637, |
|
"rewards/margins": 0.4832671284675598, |
|
"rewards/rejected": -0.22736826539039612, |
|
"step": 71 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 19.840513229370117, |
|
"debug/policy_chosen_logps": -389.92559814453125, |
|
"debug/policy_rejected_logits": 24.959850311279297, |
|
"debug/policy_rejected_logps": -467.8755187988281, |
|
"debug/reference_chosen_logps": -414.42547607421875, |
|
"debug/reference_rejected_logps": -441.78948974609375, |
|
"epoch": 0.7912087912087912, |
|
"grad_norm": 5.324274056234465, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 19.840513229370117, |
|
"logits/rejected": 24.959850311279297, |
|
"logps/chosen": -389.92559814453125, |
|
"logps/rejected": -467.8755187988281, |
|
"loss": 0.225, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.24499861896038055, |
|
"rewards/margins": 0.5058590173721313, |
|
"rewards/rejected": -0.260860413312912, |
|
"step": 72 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.089357376098633, |
|
"debug/policy_chosen_logps": -417.7742614746094, |
|
"debug/policy_rejected_logits": 26.09589958190918, |
|
"debug/policy_rejected_logps": -444.3390808105469, |
|
"debug/reference_chosen_logps": -446.7651672363281, |
|
"debug/reference_rejected_logps": -417.492919921875, |
|
"epoch": 0.8021978021978022, |
|
"grad_norm": 3.7830573167149204, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.089357376098633, |
|
"logits/rejected": 26.09589958190918, |
|
"logps/chosen": -417.7742614746094, |
|
"logps/rejected": -444.3390808105469, |
|
"loss": 0.2977, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2899090647697449, |
|
"rewards/margins": 0.5583702921867371, |
|
"rewards/rejected": -0.2684612572193146, |
|
"step": 73 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.799409866333008, |
|
"debug/policy_chosen_logps": -405.78070068359375, |
|
"debug/policy_rejected_logits": 26.489110946655273, |
|
"debug/policy_rejected_logps": -476.9659423828125, |
|
"debug/reference_chosen_logps": -435.2866516113281, |
|
"debug/reference_rejected_logps": -449.07806396484375, |
|
"epoch": 0.8131868131868132, |
|
"grad_norm": 3.635552390173908, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.799409866333008, |
|
"logits/rejected": 26.489110946655273, |
|
"logps/chosen": -405.78070068359375, |
|
"logps/rejected": -476.9659423828125, |
|
"loss": 0.2695, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2950596511363983, |
|
"rewards/margins": 0.573938250541687, |
|
"rewards/rejected": -0.2788785696029663, |
|
"step": 74 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.563961029052734, |
|
"debug/policy_chosen_logps": -382.45452880859375, |
|
"debug/policy_rejected_logits": 27.974708557128906, |
|
"debug/policy_rejected_logps": -446.31573486328125, |
|
"debug/reference_chosen_logps": -403.13330078125, |
|
"debug/reference_rejected_logps": -445.6566467285156, |
|
"epoch": 0.8241758241758241, |
|
"grad_norm": 6.518551807981199, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.563961029052734, |
|
"logits/rejected": 27.974708557128906, |
|
"logps/chosen": -382.45452880859375, |
|
"logps/rejected": -446.31573486328125, |
|
"loss": 0.339, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.2067876011133194, |
|
"rewards/margins": 0.2133782058954239, |
|
"rewards/rejected": -0.006590619683265686, |
|
"step": 75 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.57479476928711, |
|
"debug/policy_chosen_logps": -421.49554443359375, |
|
"debug/policy_rejected_logits": 25.88642692565918, |
|
"debug/policy_rejected_logps": -463.5325622558594, |
|
"debug/reference_chosen_logps": -445.17974853515625, |
|
"debug/reference_rejected_logps": -448.434814453125, |
|
"epoch": 0.8351648351648352, |
|
"grad_norm": 4.642048379046934, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.57479476928711, |
|
"logits/rejected": 25.88642692565918, |
|
"logps/chosen": -421.49554443359375, |
|
"logps/rejected": -463.5325622558594, |
|
"loss": 0.3014, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.23684199154376984, |
|
"rewards/margins": 0.3878192901611328, |
|
"rewards/rejected": -0.15097728371620178, |
|
"step": 76 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.36321258544922, |
|
"debug/policy_chosen_logps": -417.56976318359375, |
|
"debug/policy_rejected_logits": 27.49226951599121, |
|
"debug/policy_rejected_logps": -499.5318603515625, |
|
"debug/reference_chosen_logps": -444.1575927734375, |
|
"debug/reference_rejected_logps": -475.24273681640625, |
|
"epoch": 0.8461538461538461, |
|
"grad_norm": 4.237463467519612, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.36321258544922, |
|
"logits/rejected": 27.49226951599121, |
|
"logps/chosen": -417.56976318359375, |
|
"logps/rejected": -499.5318603515625, |
|
"loss": 0.2199, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2658780515193939, |
|
"rewards/margins": 0.5087695121765137, |
|
"rewards/rejected": -0.24289149045944214, |
|
"step": 77 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.683202743530273, |
|
"debug/policy_chosen_logps": -409.4302673339844, |
|
"debug/policy_rejected_logits": 23.952590942382812, |
|
"debug/policy_rejected_logps": -463.2736511230469, |
|
"debug/reference_chosen_logps": -432.84686279296875, |
|
"debug/reference_rejected_logps": -437.69964599609375, |
|
"epoch": 0.8571428571428571, |
|
"grad_norm": 4.166956738488574, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.683202743530273, |
|
"logits/rejected": 23.952590942382812, |
|
"logps/chosen": -409.4302673339844, |
|
"logps/rejected": -463.2736511230469, |
|
"loss": 0.2483, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.2341659516096115, |
|
"rewards/margins": 0.48990583419799805, |
|
"rewards/rejected": -0.25573989748954773, |
|
"step": 78 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.540374755859375, |
|
"debug/policy_chosen_logps": -422.86810302734375, |
|
"debug/policy_rejected_logits": 30.67388916015625, |
|
"debug/policy_rejected_logps": -492.8260192871094, |
|
"debug/reference_chosen_logps": -438.4207763671875, |
|
"debug/reference_rejected_logps": -458.2353515625, |
|
"epoch": 0.8681318681318682, |
|
"grad_norm": 5.889314624750013, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.540374755859375, |
|
"logits/rejected": 30.67388916015625, |
|
"logps/chosen": -422.86810302734375, |
|
"logps/rejected": -492.8260192871094, |
|
"loss": 0.264, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.15552687644958496, |
|
"rewards/margins": 0.5014333724975586, |
|
"rewards/rejected": -0.34590649604797363, |
|
"step": 79 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.57747459411621, |
|
"debug/policy_chosen_logps": -390.2710266113281, |
|
"debug/policy_rejected_logits": 27.390230178833008, |
|
"debug/policy_rejected_logps": -489.9441223144531, |
|
"debug/reference_chosen_logps": -416.788818359375, |
|
"debug/reference_rejected_logps": -440.2818298339844, |
|
"epoch": 0.8791208791208791, |
|
"grad_norm": 4.745834840744495, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.57747459411621, |
|
"logits/rejected": 27.390230178833008, |
|
"logps/chosen": -390.2710266113281, |
|
"logps/rejected": -489.9441223144531, |
|
"loss": 0.2894, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.2651780843734741, |
|
"rewards/margins": 0.76180100440979, |
|
"rewards/rejected": -0.49662283062934875, |
|
"step": 80 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 25.34842300415039, |
|
"debug/policy_chosen_logps": -390.0006103515625, |
|
"debug/policy_rejected_logits": 27.055194854736328, |
|
"debug/policy_rejected_logps": -460.668701171875, |
|
"debug/reference_chosen_logps": -408.5838623046875, |
|
"debug/reference_rejected_logps": -444.5193786621094, |
|
"epoch": 0.8901098901098901, |
|
"grad_norm": 4.590683530349925, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 25.34842300415039, |
|
"logits/rejected": 27.055194854736328, |
|
"logps/chosen": -390.0006103515625, |
|
"logps/rejected": -460.668701171875, |
|
"loss": 0.2902, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.1858326643705368, |
|
"rewards/margins": 0.3473258316516876, |
|
"rewards/rejected": -0.16149315237998962, |
|
"step": 81 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.74835968017578, |
|
"debug/policy_chosen_logps": -413.76629638671875, |
|
"debug/policy_rejected_logits": 27.351459503173828, |
|
"debug/policy_rejected_logps": -438.90216064453125, |
|
"debug/reference_chosen_logps": -443.37615966796875, |
|
"debug/reference_rejected_logps": -417.8377380371094, |
|
"epoch": 0.9010989010989011, |
|
"grad_norm": 4.49887483206356, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.74835968017578, |
|
"logits/rejected": 27.351459503173828, |
|
"logps/chosen": -413.76629638671875, |
|
"logps/rejected": -438.90216064453125, |
|
"loss": 0.252, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.29609841108322144, |
|
"rewards/margins": 0.5067427754402161, |
|
"rewards/rejected": -0.21064436435699463, |
|
"step": 82 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.60724639892578, |
|
"debug/policy_chosen_logps": -380.8109130859375, |
|
"debug/policy_rejected_logits": 30.683988571166992, |
|
"debug/policy_rejected_logps": -487.4014892578125, |
|
"debug/reference_chosen_logps": -419.3782958984375, |
|
"debug/reference_rejected_logps": -462.61505126953125, |
|
"epoch": 0.9120879120879121, |
|
"grad_norm": 4.452776489719901, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.60724639892578, |
|
"logits/rejected": 30.683988571166992, |
|
"logps/chosen": -380.8109130859375, |
|
"logps/rejected": -487.4014892578125, |
|
"loss": 0.281, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.38567405939102173, |
|
"rewards/margins": 0.6335387825965881, |
|
"rewards/rejected": -0.2478647232055664, |
|
"step": 83 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.03528594970703, |
|
"debug/policy_chosen_logps": -416.22314453125, |
|
"debug/policy_rejected_logits": 23.581472396850586, |
|
"debug/policy_rejected_logps": -434.1588439941406, |
|
"debug/reference_chosen_logps": -440.158935546875, |
|
"debug/reference_rejected_logps": -421.5128173828125, |
|
"epoch": 0.9230769230769231, |
|
"grad_norm": 4.971404649364657, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.03528594970703, |
|
"logits/rejected": 23.581472396850586, |
|
"logps/chosen": -416.22314453125, |
|
"logps/rejected": -434.1588439941406, |
|
"loss": 0.2359, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.23935794830322266, |
|
"rewards/margins": 0.36581796407699585, |
|
"rewards/rejected": -0.1264600157737732, |
|
"step": 84 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 26.0145320892334, |
|
"debug/policy_chosen_logps": -407.3331298828125, |
|
"debug/policy_rejected_logits": 22.86039924621582, |
|
"debug/policy_rejected_logps": -455.66888427734375, |
|
"debug/reference_chosen_logps": -429.9805908203125, |
|
"debug/reference_rejected_logps": -431.9481201171875, |
|
"epoch": 0.9340659340659341, |
|
"grad_norm": 4.560309498305691, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 26.0145320892334, |
|
"logits/rejected": 22.86039924621582, |
|
"logps/chosen": -407.3331298828125, |
|
"logps/rejected": -455.66888427734375, |
|
"loss": 0.2926, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.22647437453269958, |
|
"rewards/margins": 0.46368181705474854, |
|
"rewards/rejected": -0.23720744252204895, |
|
"step": 85 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 21.526865005493164, |
|
"debug/policy_chosen_logps": -391.841552734375, |
|
"debug/policy_rejected_logits": 25.782846450805664, |
|
"debug/policy_rejected_logps": -485.3880310058594, |
|
"debug/reference_chosen_logps": -410.5163269042969, |
|
"debug/reference_rejected_logps": -450.18896484375, |
|
"epoch": 0.945054945054945, |
|
"grad_norm": 3.115000092835439, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 21.526865005493164, |
|
"logits/rejected": 25.782846450805664, |
|
"logps/chosen": -391.841552734375, |
|
"logps/rejected": -485.3880310058594, |
|
"loss": 0.227, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.1867476999759674, |
|
"rewards/margins": 0.5387383699417114, |
|
"rewards/rejected": -0.35199064016342163, |
|
"step": 86 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 24.497520446777344, |
|
"debug/policy_chosen_logps": -430.42303466796875, |
|
"debug/policy_rejected_logits": 26.349315643310547, |
|
"debug/policy_rejected_logps": -465.03375244140625, |
|
"debug/reference_chosen_logps": -443.20037841796875, |
|
"debug/reference_rejected_logps": -455.8772888183594, |
|
"epoch": 0.9560439560439561, |
|
"grad_norm": 4.513341277064094, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 24.497520446777344, |
|
"logits/rejected": 26.349315643310547, |
|
"logps/chosen": -430.42303466796875, |
|
"logps/rejected": -465.03375244140625, |
|
"loss": 0.2854, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.12777358293533325, |
|
"rewards/margins": 0.2193382978439331, |
|
"rewards/rejected": -0.09156470745801926, |
|
"step": 87 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 18.35858917236328, |
|
"debug/policy_chosen_logps": -389.84100341796875, |
|
"debug/policy_rejected_logits": 21.731107711791992, |
|
"debug/policy_rejected_logps": -423.7160339355469, |
|
"debug/reference_chosen_logps": -406.00408935546875, |
|
"debug/reference_rejected_logps": -423.71820068359375, |
|
"epoch": 0.967032967032967, |
|
"grad_norm": 4.8398310648982354, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 18.35858917236328, |
|
"logits/rejected": 21.731107711791992, |
|
"logps/chosen": -389.84100341796875, |
|
"logps/rejected": -423.7160339355469, |
|
"loss": 0.2595, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.16163063049316406, |
|
"rewards/margins": 0.16160908341407776, |
|
"rewards/rejected": 2.155173569917679e-05, |
|
"step": 88 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 23.2659969329834, |
|
"debug/policy_chosen_logps": -423.4161376953125, |
|
"debug/policy_rejected_logits": 28.146160125732422, |
|
"debug/policy_rejected_logps": -473.67059326171875, |
|
"debug/reference_chosen_logps": -441.9681091308594, |
|
"debug/reference_rejected_logps": -431.1749267578125, |
|
"epoch": 0.978021978021978, |
|
"grad_norm": 6.467539115961516, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 23.2659969329834, |
|
"logits/rejected": 28.146160125732422, |
|
"logps/chosen": -423.4161376953125, |
|
"logps/rejected": -473.67059326171875, |
|
"loss": 0.2244, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.1855195164680481, |
|
"rewards/margins": 0.610476553440094, |
|
"rewards/rejected": -0.4249570369720459, |
|
"step": 89 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 27.019420623779297, |
|
"debug/policy_chosen_logps": -413.4447021484375, |
|
"debug/policy_rejected_logits": 31.003211975097656, |
|
"debug/policy_rejected_logps": -500.8101501464844, |
|
"debug/reference_chosen_logps": -434.320068359375, |
|
"debug/reference_rejected_logps": -467.6002502441406, |
|
"epoch": 0.989010989010989, |
|
"grad_norm": 4.1029333690791745, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 27.019420623779297, |
|
"logits/rejected": 31.003211975097656, |
|
"logps/chosen": -413.4447021484375, |
|
"logps/rejected": -500.8101501464844, |
|
"loss": 0.2992, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 0.20875364542007446, |
|
"rewards/margins": 0.5408526062965393, |
|
"rewards/rejected": -0.33209896087646484, |
|
"step": 90 |
|
}, |
|
{ |
|
"debug/policy_chosen_logits": 29.092975616455078, |
|
"debug/policy_chosen_logps": -404.4205017089844, |
|
"debug/policy_rejected_logits": 23.611309051513672, |
|
"debug/policy_rejected_logps": -429.15985107421875, |
|
"debug/reference_chosen_logps": -424.5293884277344, |
|
"debug/reference_rejected_logps": -418.74169921875, |
|
"epoch": 1.0, |
|
"grad_norm": 5.419471225413546, |
|
"learning_rate": 1e-06, |
|
"logits/chosen": 29.092975616455078, |
|
"logits/rejected": 23.611309051513672, |
|
"logps/chosen": -404.4205017089844, |
|
"logps/rejected": -429.15985107421875, |
|
"loss": 0.2937, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 0.20108896493911743, |
|
"rewards/margins": 0.30527064204216003, |
|
"rewards/rejected": -0.10418166220188141, |
|
"step": 91 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 91, |
|
"total_flos": 0.0, |
|
"train_loss": 0.3198396195094664, |
|
"train_runtime": 711.7161, |
|
"train_samples_per_second": 8.179, |
|
"train_steps_per_second": 0.128 |
|
} |
|
], |
|
"logging_steps": 1, |
|
"max_steps": 91, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 500, |
|
"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": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|