|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.9993998799759952, |
|
"eval_steps": 100, |
|
"global_step": 1249, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 4e-08, |
|
"logits/chosen": -2.6092891693115234, |
|
"logits/rejected": -2.629662275314331, |
|
"logps/chosen": -260.39178466796875, |
|
"logps/rejected": -256.2431945800781, |
|
"loss": 0.6931, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 4.0000000000000003e-07, |
|
"logits/chosen": -2.542802095413208, |
|
"logits/rejected": -2.5701138973236084, |
|
"logps/chosen": -287.18267822265625, |
|
"logps/rejected": -308.441162109375, |
|
"loss": 0.6934, |
|
"rewards/accuracies": 0.4166666567325592, |
|
"rewards/chosen": 0.0005982329603284597, |
|
"rewards/margins": -0.0005221270839683712, |
|
"rewards/rejected": 0.0011203601025044918, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 8.000000000000001e-07, |
|
"logits/chosen": -2.520120859146118, |
|
"logits/rejected": -2.5234670639038086, |
|
"logps/chosen": -255.07748413085938, |
|
"logps/rejected": -273.0997009277344, |
|
"loss": 0.693, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.00023935179342515767, |
|
"rewards/margins": 0.00027643097564578056, |
|
"rewards/rejected": -0.0005157828563824296, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.2000000000000002e-06, |
|
"logits/chosen": -2.5063071250915527, |
|
"logits/rejected": -2.5213770866394043, |
|
"logps/chosen": -276.6528625488281, |
|
"logps/rejected": -294.9043884277344, |
|
"loss": 0.693, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.0008763422374613583, |
|
"rewards/margins": 0.0003111693658865988, |
|
"rewards/rejected": -0.001187511719763279, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.6000000000000001e-06, |
|
"logits/chosen": -2.542746067047119, |
|
"logits/rejected": -2.549612522125244, |
|
"logps/chosen": -251.32003784179688, |
|
"logps/rejected": -259.71685791015625, |
|
"loss": 0.6928, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.004003011621534824, |
|
"rewards/margins": 0.0007247755420394242, |
|
"rewards/rejected": -0.004727787338197231, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.0000000000000003e-06, |
|
"logits/chosen": -2.4885387420654297, |
|
"logits/rejected": -2.4705405235290527, |
|
"logps/chosen": -251.05563354492188, |
|
"logps/rejected": -255.7452392578125, |
|
"loss": 0.6933, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.006660168059170246, |
|
"rewards/margins": -0.00038892580778338015, |
|
"rewards/rejected": -0.006271242164075375, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.4000000000000003e-06, |
|
"logits/chosen": -2.5540900230407715, |
|
"logits/rejected": -2.5607895851135254, |
|
"logps/chosen": -262.1423034667969, |
|
"logps/rejected": -280.08209228515625, |
|
"loss": 0.6928, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.009707225486636162, |
|
"rewards/margins": 0.000684750615619123, |
|
"rewards/rejected": -0.010391974821686745, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 2.8000000000000003e-06, |
|
"logits/chosen": -2.5549094676971436, |
|
"logits/rejected": -2.5427558422088623, |
|
"logps/chosen": -266.07891845703125, |
|
"logps/rejected": -271.27740478515625, |
|
"loss": 0.6926, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.014160098508000374, |
|
"rewards/margins": 0.0011990683851763606, |
|
"rewards/rejected": -0.015359166078269482, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.2000000000000003e-06, |
|
"logits/chosen": -2.5750041007995605, |
|
"logits/rejected": -2.5791172981262207, |
|
"logps/chosen": -262.36407470703125, |
|
"logps/rejected": -282.44720458984375, |
|
"loss": 0.6928, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.015319536440074444, |
|
"rewards/margins": 0.0008405897533521056, |
|
"rewards/rejected": -0.01616012677550316, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.6000000000000003e-06, |
|
"logits/chosen": -2.5781426429748535, |
|
"logits/rejected": -2.5637879371643066, |
|
"logps/chosen": -295.14898681640625, |
|
"logps/rejected": -300.16131591796875, |
|
"loss": 0.6906, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.020102154463529587, |
|
"rewards/margins": 0.005331115797162056, |
|
"rewards/rejected": -0.025433272123336792, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 4.000000000000001e-06, |
|
"logits/chosen": -2.5696635246276855, |
|
"logits/rejected": -2.569899797439575, |
|
"logps/chosen": -263.42047119140625, |
|
"logps/rejected": -276.1015319824219, |
|
"loss": 0.6917, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.024685552343726158, |
|
"rewards/margins": 0.0029262492898851633, |
|
"rewards/rejected": -0.027611803263425827, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"eval_logits/chosen": -2.530292272567749, |
|
"eval_logits/rejected": -2.5272903442382812, |
|
"eval_logps/chosen": -210.00074768066406, |
|
"eval_logps/rejected": -215.47608947753906, |
|
"eval_loss": 0.6923908591270447, |
|
"eval_rewards/accuracies": 0.5286666750907898, |
|
"eval_rewards/chosen": -0.016008317470550537, |
|
"eval_rewards/margins": 0.001649469486437738, |
|
"eval_rewards/rejected": -0.01765778660774231, |
|
"eval_runtime": 1564.6229, |
|
"eval_samples_per_second": 1.914, |
|
"eval_steps_per_second": 0.24, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.4e-06, |
|
"logits/chosen": -2.573535680770874, |
|
"logits/rejected": -2.5764663219451904, |
|
"logps/chosen": -283.31744384765625, |
|
"logps/rejected": -293.5372314453125, |
|
"loss": 0.6918, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.027460157871246338, |
|
"rewards/margins": 0.0029141721315681934, |
|
"rewards/rejected": -0.030374327674508095, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.800000000000001e-06, |
|
"logits/chosen": -2.593564033508301, |
|
"logits/rejected": -2.6072158813476562, |
|
"logps/chosen": -265.5439758300781, |
|
"logps/rejected": -282.11444091796875, |
|
"loss": 0.6911, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.03183265030384064, |
|
"rewards/margins": 0.004463006276637316, |
|
"rewards/rejected": -0.036295659840106964, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.999755876225375e-06, |
|
"logits/chosen": -2.559654712677002, |
|
"logits/rejected": -2.543823480606079, |
|
"logps/chosen": -279.4471435546875, |
|
"logps/rejected": -297.8502502441406, |
|
"loss": 0.6892, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.0379730649292469, |
|
"rewards/margins": 0.008286955766379833, |
|
"rewards/rejected": -0.04626002162694931, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.997803172081864e-06, |
|
"logits/chosen": -2.5811429023742676, |
|
"logits/rejected": -2.581789255142212, |
|
"logps/chosen": -274.745849609375, |
|
"logps/rejected": -287.67303466796875, |
|
"loss": 0.6865, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.04451751708984375, |
|
"rewards/margins": 0.014026105403900146, |
|
"rewards/rejected": -0.058543622493743896, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.9938992891651825e-06, |
|
"logits/chosen": -2.545161724090576, |
|
"logits/rejected": -2.5361251831054688, |
|
"logps/chosen": -262.94903564453125, |
|
"logps/rejected": -284.0696716308594, |
|
"loss": 0.6833, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.0502193346619606, |
|
"rewards/margins": 0.020780760794878006, |
|
"rewards/rejected": -0.07100008428096771, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.988047277024456e-06, |
|
"logits/chosen": -2.5855746269226074, |
|
"logits/rejected": -2.5990371704101562, |
|
"logps/chosen": -274.5702819824219, |
|
"logps/rejected": -288.3578796386719, |
|
"loss": 0.6819, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.07741515338420868, |
|
"rewards/margins": 0.023703744634985924, |
|
"rewards/rejected": -0.10111888498067856, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.980251707005417e-06, |
|
"logits/chosen": -2.5723748207092285, |
|
"logits/rejected": -2.547863006591797, |
|
"logps/chosen": -291.48736572265625, |
|
"logps/rejected": -298.7716979980469, |
|
"loss": 0.6899, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.10616284608840942, |
|
"rewards/margins": 0.007837113924324512, |
|
"rewards/rejected": -0.11399996280670166, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.970518668679459e-06, |
|
"logits/chosen": -2.5809338092803955, |
|
"logits/rejected": -2.566509246826172, |
|
"logps/chosen": -289.52642822265625, |
|
"logps/rejected": -295.8729248046875, |
|
"loss": 0.6849, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.09603477269411087, |
|
"rewards/margins": 0.01864427700638771, |
|
"rewards/rejected": -0.11467906087636948, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.958855765086722e-06, |
|
"logits/chosen": -2.608503818511963, |
|
"logits/rejected": -2.6062302589416504, |
|
"logps/chosen": -269.84283447265625, |
|
"logps/rejected": -280.5111083984375, |
|
"loss": 0.6851, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.05638309568166733, |
|
"rewards/margins": 0.017940795049071312, |
|
"rewards/rejected": -0.07432388514280319, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.945272106796919e-06, |
|
"logits/chosen": -2.605029344558716, |
|
"logits/rejected": -2.6143765449523926, |
|
"logps/chosen": -270.00830078125, |
|
"logps/rejected": -285.1636657714844, |
|
"loss": 0.6854, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.06747416406869888, |
|
"rewards/margins": 0.018329909071326256, |
|
"rewards/rejected": -0.08580406755208969, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"eval_logits/chosen": -2.57635235786438, |
|
"eval_logits/rejected": -2.573589563369751, |
|
"eval_logps/chosen": -215.42254638671875, |
|
"eval_logps/rejected": -222.06101989746094, |
|
"eval_loss": 0.6875421404838562, |
|
"eval_rewards/accuracies": 0.5563333630561829, |
|
"eval_rewards/chosen": -0.07022587954998016, |
|
"eval_rewards/margins": 0.013281372375786304, |
|
"eval_rewards/rejected": -0.08350725471973419, |
|
"eval_runtime": 1564.649, |
|
"eval_samples_per_second": 1.914, |
|
"eval_steps_per_second": 0.24, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.929778304792537e-06, |
|
"logits/chosen": -2.594109058380127, |
|
"logits/rejected": -2.5997750759124756, |
|
"logps/chosen": -294.9931640625, |
|
"logps/rejected": -299.47515869140625, |
|
"loss": 0.6783, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.08973054587841034, |
|
"rewards/margins": 0.03516639396548271, |
|
"rewards/rejected": -0.12489694356918335, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.912386462179987e-06, |
|
"logits/chosen": -2.6223044395446777, |
|
"logits/rejected": -2.6194934844970703, |
|
"logps/chosen": -282.9168701171875, |
|
"logps/rejected": -308.5869140625, |
|
"loss": 0.6757, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.08990589529275894, |
|
"rewards/margins": 0.039798181504011154, |
|
"rewards/rejected": -0.1297040730714798, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.893110164735167e-06, |
|
"logits/chosen": -2.686192512512207, |
|
"logits/rejected": -2.6861042976379395, |
|
"logps/chosen": -288.02984619140625, |
|
"logps/rejected": -301.3773193359375, |
|
"loss": 0.6819, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.11503495275974274, |
|
"rewards/margins": 0.02641122415661812, |
|
"rewards/rejected": -0.14144618809223175, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.871964470290823e-06, |
|
"logits/chosen": -2.660379409790039, |
|
"logits/rejected": -2.6843981742858887, |
|
"logps/chosen": -289.6697998046875, |
|
"logps/rejected": -310.08245849609375, |
|
"loss": 0.6712, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.12107042968273163, |
|
"rewards/margins": 0.05008827522397041, |
|
"rewards/rejected": -0.17115873098373413, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.848965896974006e-06, |
|
"logits/chosen": -2.6900529861450195, |
|
"logits/rejected": -2.691603183746338, |
|
"logps/chosen": -281.68792724609375, |
|
"logps/rejected": -301.64129638671875, |
|
"loss": 0.6784, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.1229969710111618, |
|
"rewards/margins": 0.03473496437072754, |
|
"rewards/rejected": -0.15773193538188934, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.8241324103028055e-06, |
|
"logits/chosen": -2.794003963470459, |
|
"logits/rejected": -2.756747007369995, |
|
"logps/chosen": -288.9064025878906, |
|
"logps/rejected": -300.8970947265625, |
|
"loss": 0.6737, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.1542636603116989, |
|
"rewards/margins": 0.044557444751262665, |
|
"rewards/rejected": -0.19882109761238098, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.797483409152438e-06, |
|
"logits/chosen": -2.8427796363830566, |
|
"logits/rejected": -2.8422060012817383, |
|
"logps/chosen": -280.02099609375, |
|
"logps/rejected": -300.798583984375, |
|
"loss": 0.667, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.17782548069953918, |
|
"rewards/margins": 0.0627431869506836, |
|
"rewards/rejected": -0.24056866765022278, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.769039710601669e-06, |
|
"logits/chosen": -2.926100015640259, |
|
"logits/rejected": -2.9281368255615234, |
|
"logps/chosen": -280.8794250488281, |
|
"logps/rejected": -298.9492492675781, |
|
"loss": 0.6732, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.2086031138896942, |
|
"rewards/margins": 0.047397904098033905, |
|
"rewards/rejected": -0.25600099563598633, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.738823533671383e-06, |
|
"logits/chosen": -2.972533702850342, |
|
"logits/rejected": -2.9721083641052246, |
|
"logps/chosen": -306.55865478515625, |
|
"logps/rejected": -322.1519470214844, |
|
"loss": 0.6769, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.24026934802532196, |
|
"rewards/margins": 0.041927777230739594, |
|
"rewards/rejected": -0.28219714760780334, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.706858481968017e-06, |
|
"logits/chosen": -2.922761917114258, |
|
"logits/rejected": -2.9264841079711914, |
|
"logps/chosen": -295.0879821777344, |
|
"logps/rejected": -304.8949279785156, |
|
"loss": 0.682, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.2886089086532593, |
|
"rewards/margins": 0.03215286135673523, |
|
"rewards/rejected": -0.3207617700099945, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"eval_logits/chosen": -2.920912981033325, |
|
"eval_logits/rejected": -2.9179892539978027, |
|
"eval_logps/chosen": -232.2801055908203, |
|
"eval_logps/rejected": -240.21971130371094, |
|
"eval_loss": 0.684112012386322, |
|
"eval_rewards/accuracies": 0.5450000166893005, |
|
"eval_rewards/chosen": -0.23880183696746826, |
|
"eval_rewards/margins": 0.026292217895388603, |
|
"eval_rewards/rejected": -0.2650940418243408, |
|
"eval_runtime": 1563.5865, |
|
"eval_samples_per_second": 1.915, |
|
"eval_steps_per_second": 0.24, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.673169525245416e-06, |
|
"logits/chosen": -2.912137269973755, |
|
"logits/rejected": -2.890232563018799, |
|
"logps/chosen": -291.8182678222656, |
|
"logps/rejected": -320.7854309082031, |
|
"loss": 0.6652, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.257565438747406, |
|
"rewards/margins": 0.06874646991491318, |
|
"rewards/rejected": -0.3263119161128998, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.63778297989952e-06, |
|
"logits/chosen": -3.007861375808716, |
|
"logits/rejected": -2.9914684295654297, |
|
"logps/chosen": -297.81695556640625, |
|
"logps/rejected": -314.2324523925781, |
|
"loss": 0.6678, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.28555119037628174, |
|
"rewards/margins": 0.06271664053201675, |
|
"rewards/rejected": -0.3482677936553955, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.60072648841109e-06, |
|
"logits/chosen": -3.0622572898864746, |
|
"logits/rejected": -3.0570547580718994, |
|
"logps/chosen": -311.02105712890625, |
|
"logps/rejected": -333.46630859375, |
|
"loss": 0.6444, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.3358006477355957, |
|
"rewards/margins": 0.11871840804815292, |
|
"rewards/rejected": -0.454519122838974, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.562028997752574e-06, |
|
"logits/chosen": -3.153225898742676, |
|
"logits/rejected": -3.1641342639923096, |
|
"logps/chosen": -311.9708557128906, |
|
"logps/rejected": -338.3099670410156, |
|
"loss": 0.6682, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.3542831838130951, |
|
"rewards/margins": 0.068769671022892, |
|
"rewards/rejected": -0.4230528473854065, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.521720736775947e-06, |
|
"logits/chosen": -3.2916831970214844, |
|
"logits/rejected": -3.3004555702209473, |
|
"logps/chosen": -328.3070373535156, |
|
"logps/rejected": -338.16192626953125, |
|
"loss": 0.6564, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.4777124524116516, |
|
"rewards/margins": 0.09214212745428085, |
|
"rewards/rejected": -0.5698545575141907, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.479833192599198e-06, |
|
"logits/chosen": -3.4253318309783936, |
|
"logits/rejected": -3.4066455364227295, |
|
"logps/chosen": -329.5577697753906, |
|
"logps/rejected": -350.79742431640625, |
|
"loss": 0.6555, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.46095147728919983, |
|
"rewards/margins": 0.09699266403913498, |
|
"rewards/rejected": -0.5579441785812378, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.436399086009928e-06, |
|
"logits/chosen": -3.4043822288513184, |
|
"logits/rejected": -3.3838889598846436, |
|
"logps/chosen": -317.8493957519531, |
|
"logps/rejected": -334.6202087402344, |
|
"loss": 0.6576, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.4469628930091858, |
|
"rewards/margins": 0.08755813539028168, |
|
"rewards/rejected": -0.5345209836959839, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.391452345905239e-06, |
|
"logits/chosen": -3.347275972366333, |
|
"logits/rejected": -3.35723614692688, |
|
"logps/chosen": -325.1240234375, |
|
"logps/rejected": -339.3655090332031, |
|
"loss": 0.6618, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.4934774935245514, |
|
"rewards/margins": 0.09260674566030502, |
|
"rewards/rejected": -0.586084246635437, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.3450280827879125e-06, |
|
"logits/chosen": -3.4295692443847656, |
|
"logits/rejected": -3.4535210132598877, |
|
"logps/chosen": -323.19415283203125, |
|
"logps/rejected": -340.07989501953125, |
|
"loss": 0.6639, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.4877980351448059, |
|
"rewards/margins": 0.08489425480365753, |
|
"rewards/rejected": -0.5726922154426575, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.297162561339554e-06, |
|
"logits/chosen": -3.366345167160034, |
|
"logits/rejected": -3.3470451831817627, |
|
"logps/chosen": -330.50311279296875, |
|
"logps/rejected": -352.4704895019531, |
|
"loss": 0.6634, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.5246509313583374, |
|
"rewards/margins": 0.09757552295923233, |
|
"rewards/rejected": -0.6222264766693115, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"eval_logits/chosen": -3.460310935974121, |
|
"eval_logits/rejected": -3.4548885822296143, |
|
"eval_logps/chosen": -256.72369384765625, |
|
"eval_logps/rejected": -266.585693359375, |
|
"eval_loss": 0.681212306022644, |
|
"eval_rewards/accuracies": 0.5486666560173035, |
|
"eval_rewards/chosen": -0.4832378029823303, |
|
"eval_rewards/margins": 0.04551612213253975, |
|
"eval_rewards/rejected": -0.528753936290741, |
|
"eval_runtime": 1563.8204, |
|
"eval_samples_per_second": 1.915, |
|
"eval_steps_per_second": 0.24, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.247893172092157e-06, |
|
"logits/chosen": -3.3930275440216064, |
|
"logits/rejected": -3.3995296955108643, |
|
"logps/chosen": -320.64154052734375, |
|
"logps/rejected": -351.81414794921875, |
|
"loss": 0.6576, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.5477547645568848, |
|
"rewards/margins": 0.10663916915655136, |
|
"rewards/rejected": -0.6543939709663391, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.197258402220187e-06, |
|
"logits/chosen": -3.4664406776428223, |
|
"logits/rejected": -3.4772841930389404, |
|
"logps/chosen": -328.4302673339844, |
|
"logps/rejected": -347.1799011230469, |
|
"loss": 0.6516, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.5463472604751587, |
|
"rewards/margins": 0.14043554663658142, |
|
"rewards/rejected": -0.686782717704773, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.145297805476023e-06, |
|
"logits/chosen": -3.5236573219299316, |
|
"logits/rejected": -3.5294857025146484, |
|
"logps/chosen": -326.1194763183594, |
|
"logps/rejected": -350.4049987792969, |
|
"loss": 0.6477, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.5520918965339661, |
|
"rewards/margins": 0.13327525556087494, |
|
"rewards/rejected": -0.685367226600647, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.092051971292228e-06, |
|
"logits/chosen": -3.597280502319336, |
|
"logits/rejected": -3.597381591796875, |
|
"logps/chosen": -324.05584716796875, |
|
"logps/rejected": -346.5114440917969, |
|
"loss": 0.654, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.5706614255905151, |
|
"rewards/margins": 0.11499401181936264, |
|
"rewards/rejected": -0.6856554746627808, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.037562493074792e-06, |
|
"logits/chosen": -3.742100238800049, |
|
"logits/rejected": -3.7566616535186768, |
|
"logps/chosen": -349.8996887207031, |
|
"logps/rejected": -364.50506591796875, |
|
"loss": 0.6477, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.6562848091125488, |
|
"rewards/margins": 0.1411934792995453, |
|
"rewards/rejected": -0.7974783182144165, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.981871935712112e-06, |
|
"logits/chosen": -3.968726396560669, |
|
"logits/rejected": -3.941209077835083, |
|
"logps/chosen": -332.73443603515625, |
|
"logps/rejected": -360.47625732421875, |
|
"loss": 0.6493, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.6563684344291687, |
|
"rewards/margins": 0.13294601440429688, |
|
"rewards/rejected": -0.7893144488334656, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.925023802325094e-06, |
|
"logits/chosen": -4.149798393249512, |
|
"logits/rejected": -4.129707336425781, |
|
"logps/chosen": -354.22802734375, |
|
"logps/rejected": -380.85040283203125, |
|
"loss": 0.6505, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.7849351763725281, |
|
"rewards/margins": 0.16557727754116058, |
|
"rewards/rejected": -0.9505125284194946, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.867062500284342e-06, |
|
"logits/chosen": -4.189931392669678, |
|
"logits/rejected": -4.1646270751953125, |
|
"logps/chosen": -344.6732482910156, |
|
"logps/rejected": -376.7868957519531, |
|
"loss": 0.647, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.8410800099372864, |
|
"rewards/margins": 0.15282076597213745, |
|
"rewards/rejected": -0.9939007759094238, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.8080333065209885e-06, |
|
"logits/chosen": -4.1885223388671875, |
|
"logits/rejected": -4.1956305503845215, |
|
"logps/chosen": -351.62237548828125, |
|
"logps/rejected": -351.5168151855469, |
|
"loss": 0.6998, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.8627797961235046, |
|
"rewards/margins": 0.05857674032449722, |
|
"rewards/rejected": -0.9213564991950989, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 3.7479823321582624e-06, |
|
"logits/chosen": -4.04768705368042, |
|
"logits/rejected": -4.022109031677246, |
|
"logps/chosen": -345.1391296386719, |
|
"logps/rejected": -380.88482666015625, |
|
"loss": 0.6296, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.7372187972068787, |
|
"rewards/margins": 0.17785954475402832, |
|
"rewards/rejected": -0.915078341960907, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"eval_logits/chosen": -4.174871921539307, |
|
"eval_logits/rejected": -4.1668291091918945, |
|
"eval_logps/chosen": -277.3629150390625, |
|
"eval_logps/rejected": -289.35430908203125, |
|
"eval_loss": 0.678155779838562, |
|
"eval_rewards/accuracies": 0.5600000023841858, |
|
"eval_rewards/chosen": -0.6896299123764038, |
|
"eval_rewards/margins": 0.06681026518344879, |
|
"eval_rewards/rejected": -0.7564401626586914, |
|
"eval_runtime": 1565.1126, |
|
"eval_samples_per_second": 1.913, |
|
"eval_steps_per_second": 0.24, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.686956486491419e-06, |
|
"logits/chosen": -4.035499572753906, |
|
"logits/rejected": -4.026083469390869, |
|
"logps/chosen": -354.2256774902344, |
|
"logps/rejected": -383.2754821777344, |
|
"loss": 0.642, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.7423269748687744, |
|
"rewards/margins": 0.21402092278003693, |
|
"rewards/rejected": -0.9563478231430054, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.625003440344166e-06, |
|
"logits/chosen": -4.110401630401611, |
|
"logits/rejected": -4.130579948425293, |
|
"logps/chosen": -333.2506103515625, |
|
"logps/rejected": -345.42529296875, |
|
"loss": 0.6662, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.6982457637786865, |
|
"rewards/margins": 0.08196356892585754, |
|
"rewards/rejected": -0.7802093625068665, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.562171588830231e-06, |
|
"logits/chosen": -4.092467308044434, |
|
"logits/rejected": -4.074677467346191, |
|
"logps/chosen": -342.26513671875, |
|
"logps/rejected": -370.6748352050781, |
|
"loss": 0.6627, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.6801224946975708, |
|
"rewards/margins": 0.11174317449331284, |
|
"rewards/rejected": -0.7918656468391418, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.4985100135491245e-06, |
|
"logits/chosen": -4.104280471801758, |
|
"logits/rejected": -4.07126522064209, |
|
"logps/chosen": -353.3872375488281, |
|
"logps/rejected": -391.28289794921875, |
|
"loss": 0.6383, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.7574983239173889, |
|
"rewards/margins": 0.18485240638256073, |
|
"rewards/rejected": -0.9423506855964661, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.4340684442456673e-06, |
|
"logits/chosen": -4.108157634735107, |
|
"logits/rejected": -4.100740432739258, |
|
"logps/chosen": -353.9938049316406, |
|
"logps/rejected": -375.70562744140625, |
|
"loss": 0.6498, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.8029891848564148, |
|
"rewards/margins": 0.12438172101974487, |
|
"rewards/rejected": -0.9273708462715149, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3688972199631974e-06, |
|
"logits/chosen": -4.0825605392456055, |
|
"logits/rejected": -4.094653129577637, |
|
"logps/chosen": -352.5472412109375, |
|
"logps/rejected": -372.22125244140625, |
|
"loss": 0.646, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.7980318069458008, |
|
"rewards/margins": 0.1836821585893631, |
|
"rewards/rejected": -0.9817139506340027, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.3030472497208354e-06, |
|
"logits/chosen": -4.089273452758789, |
|
"logits/rejected": -4.051713943481445, |
|
"logps/chosen": -349.65447998046875, |
|
"logps/rejected": -411.3336486816406, |
|
"loss": 0.6145, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.8661645650863647, |
|
"rewards/margins": 0.2521691918373108, |
|
"rewards/rejected": -1.1183335781097412, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.236569972745492e-06, |
|
"logits/chosen": -4.18710994720459, |
|
"logits/rejected": -4.17570161819458, |
|
"logps/chosen": -347.78521728515625, |
|
"logps/rejected": -368.5164489746094, |
|
"loss": 0.6671, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.878218948841095, |
|
"rewards/margins": 0.11807211488485336, |
|
"rewards/rejected": -0.9962911605834961, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.1695173182897126e-06, |
|
"logits/chosen": -4.180907249450684, |
|
"logits/rejected": -4.169510841369629, |
|
"logps/chosen": -363.77081298828125, |
|
"logps/rejected": -398.118408203125, |
|
"loss": 0.6542, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.982130229473114, |
|
"rewards/margins": 0.1635296642780304, |
|
"rewards/rejected": -1.1456598043441772, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.10194166506673e-06, |
|
"logits/chosen": -4.227340221405029, |
|
"logits/rejected": -4.191034317016602, |
|
"logps/chosen": -362.04327392578125, |
|
"logps/rejected": -401.2042541503906, |
|
"loss": 0.6503, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.9192866086959839, |
|
"rewards/margins": 0.18541966378688812, |
|
"rewards/rejected": -1.1047064065933228, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"eval_logits/chosen": -4.443284511566162, |
|
"eval_logits/rejected": -4.434456825256348, |
|
"eval_logps/chosen": -304.2834167480469, |
|
"eval_logps/rejected": -318.1134033203125, |
|
"eval_loss": 0.6769664287567139, |
|
"eval_rewards/accuracies": 0.5533333420753479, |
|
"eval_rewards/chosen": -0.9588348269462585, |
|
"eval_rewards/margins": 0.0851958841085434, |
|
"eval_rewards/rejected": -1.0440306663513184, |
|
"eval_runtime": 1565.9018, |
|
"eval_samples_per_second": 1.912, |
|
"eval_steps_per_second": 0.239, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.0338958003344115e-06, |
|
"logits/chosen": -4.3410210609436035, |
|
"logits/rejected": -4.294773101806641, |
|
"logps/chosen": -360.5263977050781, |
|
"logps/rejected": -391.7284240722656, |
|
"loss": 0.6595, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.9942121505737305, |
|
"rewards/margins": 0.21516022086143494, |
|
"rewards/rejected": -1.2093722820281982, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.9654328786600823e-06, |
|
"logits/chosen": -4.326909065246582, |
|
"logits/rejected": -4.2857208251953125, |
|
"logps/chosen": -359.14569091796875, |
|
"logps/rejected": -399.8367004394531, |
|
"loss": 0.627, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.9758028984069824, |
|
"rewards/margins": 0.18271294236183167, |
|
"rewards/rejected": -1.1585158109664917, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.896606380398402e-06, |
|
"logits/chosen": -4.396684169769287, |
|
"logits/rejected": -4.430912971496582, |
|
"logps/chosen": -377.33551025390625, |
|
"logps/rejected": -398.4137878417969, |
|
"loss": 0.6688, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.0871975421905518, |
|
"rewards/margins": 0.13647660613059998, |
|
"rewards/rejected": -1.2236744165420532, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.827470069914772e-06, |
|
"logits/chosen": -4.29737663269043, |
|
"logits/rejected": -4.263535499572754, |
|
"logps/chosen": -376.9189453125, |
|
"logps/rejected": -404.60247802734375, |
|
"loss": 0.6551, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.0662126541137695, |
|
"rewards/margins": 0.14261041581630707, |
|
"rewards/rejected": -1.2088232040405273, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.7580779535868675e-06, |
|
"logits/chosen": -4.269147872924805, |
|
"logits/rejected": -4.276608467102051, |
|
"logps/chosen": -366.36492919921875, |
|
"logps/rejected": -390.6036376953125, |
|
"loss": 0.6583, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.9485187530517578, |
|
"rewards/margins": 0.15207722783088684, |
|
"rewards/rejected": -1.1005959510803223, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.688484237617129e-06, |
|
"logits/chosen": -4.208004474639893, |
|
"logits/rejected": -4.182279586791992, |
|
"logps/chosen": -362.94586181640625, |
|
"logps/rejected": -393.9739990234375, |
|
"loss": 0.6322, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.8923367261886597, |
|
"rewards/margins": 0.20138171315193176, |
|
"rewards/rejected": -1.0937182903289795, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.6187432856891585e-06, |
|
"logits/chosen": -4.1537322998046875, |
|
"logits/rejected": -4.151703357696533, |
|
"logps/chosen": -372.31201171875, |
|
"logps/rejected": -405.5547180175781, |
|
"loss": 0.6489, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.953785240650177, |
|
"rewards/margins": 0.15488819777965546, |
|
"rewards/rejected": -1.108673334121704, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.548909576501096e-06, |
|
"logits/chosen": -4.182751655578613, |
|
"logits/rejected": -4.181895732879639, |
|
"logps/chosen": -372.65435791015625, |
|
"logps/rejected": -398.80902099609375, |
|
"loss": 0.6442, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.9745942950248718, |
|
"rewards/margins": 0.160622701048851, |
|
"rewards/rejected": -1.1352169513702393, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4790376612091503e-06, |
|
"logits/chosen": -4.2613630294799805, |
|
"logits/rejected": -4.227023124694824, |
|
"logps/chosen": -391.2955017089844, |
|
"logps/rejected": -419.54656982421875, |
|
"loss": 0.6233, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.0887335538864136, |
|
"rewards/margins": 0.2128380537033081, |
|
"rewards/rejected": -1.3015714883804321, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 2.40918212081453e-06, |
|
"logits/chosen": -4.308588981628418, |
|
"logits/rejected": -4.264666557312012, |
|
"logps/chosen": -360.5702819824219, |
|
"logps/rejected": -409.14422607421875, |
|
"loss": 0.5974, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.0383028984069824, |
|
"rewards/margins": 0.29866236448287964, |
|
"rewards/rejected": -1.3369653224945068, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"eval_logits/chosen": -4.447990417480469, |
|
"eval_logits/rejected": -4.436967849731445, |
|
"eval_logps/chosen": -322.94854736328125, |
|
"eval_logps/rejected": -338.0312194824219, |
|
"eval_loss": 0.6777762174606323, |
|
"eval_rewards/accuracies": 0.5653333067893982, |
|
"eval_rewards/chosen": -1.1454858779907227, |
|
"eval_rewards/margins": 0.09772318601608276, |
|
"eval_rewards/rejected": -1.2432090044021606, |
|
"eval_runtime": 1566.0394, |
|
"eval_samples_per_second": 1.912, |
|
"eval_steps_per_second": 0.239, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.3393975235270654e-06, |
|
"logits/chosen": -4.278247833251953, |
|
"logits/rejected": -4.263747692108154, |
|
"logps/chosen": -394.7868347167969, |
|
"logps/rejected": -433.32373046875, |
|
"loss": 0.6411, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -1.1327521800994873, |
|
"rewards/margins": 0.22177498042583466, |
|
"rewards/rejected": -1.354527235031128, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2697383821388153e-06, |
|
"logits/chosen": -4.283724308013916, |
|
"logits/rejected": -4.2910637855529785, |
|
"logps/chosen": -383.4498596191406, |
|
"logps/rejected": -406.142578125, |
|
"loss": 0.6479, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.1030470132827759, |
|
"rewards/margins": 0.19455528259277344, |
|
"rewards/rejected": -1.2976024150848389, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2002591114409657e-06, |
|
"logits/chosen": -4.243215560913086, |
|
"logits/rejected": -4.230997562408447, |
|
"logps/chosen": -392.0690612792969, |
|
"logps/rejected": -419.01092529296875, |
|
"loss": 0.6517, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -1.1523319482803345, |
|
"rewards/margins": 0.17104671895503998, |
|
"rewards/rejected": -1.323378562927246, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.131013985717285e-06, |
|
"logits/chosen": -4.2963361740112305, |
|
"logits/rejected": -4.248631477355957, |
|
"logps/chosen": -402.02130126953125, |
|
"logps/rejected": -451.04400634765625, |
|
"loss": 0.6211, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.197863221168518, |
|
"rewards/margins": 0.26271212100982666, |
|
"rewards/rejected": -1.4605753421783447, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.062057096347338e-06, |
|
"logits/chosen": -4.2642927169799805, |
|
"logits/rejected": -4.245224952697754, |
|
"logps/chosen": -386.35394287109375, |
|
"logps/rejected": -399.6516418457031, |
|
"loss": 0.6611, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.1031478643417358, |
|
"rewards/margins": 0.12747173011302948, |
|
"rewards/rejected": -1.2306196689605713, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 1.9934423095525733e-06, |
|
"logits/chosen": -4.144771575927734, |
|
"logits/rejected": -4.146181583404541, |
|
"logps/chosen": -381.7077941894531, |
|
"logps/rejected": -401.769287109375, |
|
"loss": 0.6555, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.9921267628669739, |
|
"rewards/margins": 0.19879314303398132, |
|
"rewards/rejected": -1.190920114517212, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.9252232243182986e-06, |
|
"logits/chosen": -4.229475498199463, |
|
"logits/rejected": -4.177742958068848, |
|
"logps/chosen": -345.7065734863281, |
|
"logps/rejected": -388.74462890625, |
|
"loss": 0.6144, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.8882685899734497, |
|
"rewards/margins": 0.26380977034568787, |
|
"rewards/rejected": -1.15207839012146, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.8574531305244043e-06, |
|
"logits/chosen": -4.0229105949401855, |
|
"logits/rejected": -4.011281490325928, |
|
"logps/chosen": -381.92205810546875, |
|
"logps/rejected": -418.5514221191406, |
|
"loss": 0.6197, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.013808250427246, |
|
"rewards/margins": 0.23891642689704895, |
|
"rewards/rejected": -1.2527246475219727, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.7901849673175559e-06, |
|
"logits/chosen": -4.090008735656738, |
|
"logits/rejected": -4.056280612945557, |
|
"logps/chosen": -375.6288146972656, |
|
"logps/rejected": -409.71490478515625, |
|
"loss": 0.634, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.9446262121200562, |
|
"rewards/margins": 0.2019636631011963, |
|
"rewards/rejected": -1.1465898752212524, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.7234712817573555e-06, |
|
"logits/chosen": -4.081742763519287, |
|
"logits/rejected": -4.069535255432129, |
|
"logps/chosen": -408.01580810546875, |
|
"logps/rejected": -428.19830322265625, |
|
"loss": 0.6508, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.1142655611038208, |
|
"rewards/margins": 0.18682199716567993, |
|
"rewards/rejected": -1.301087498664856, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"eval_logits/chosen": -4.271145343780518, |
|
"eval_logits/rejected": -4.261835098266602, |
|
"eval_logps/chosen": -318.4167785644531, |
|
"eval_logps/rejected": -333.9435119628906, |
|
"eval_loss": 0.6748104691505432, |
|
"eval_rewards/accuracies": 0.5649999976158142, |
|
"eval_rewards/chosen": -1.1001687049865723, |
|
"eval_rewards/margins": 0.10216327011585236, |
|
"eval_rewards/rejected": -1.2023320198059082, |
|
"eval_runtime": 1566.0255, |
|
"eval_samples_per_second": 1.912, |
|
"eval_steps_per_second": 0.239, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6573641877687936e-06, |
|
"logits/chosen": -4.110816478729248, |
|
"logits/rejected": -4.093571662902832, |
|
"logps/chosen": -381.05877685546875, |
|
"logps/rejected": -423.17364501953125, |
|
"loss": 0.6202, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.0917795896530151, |
|
"rewards/margins": 0.21704904735088348, |
|
"rewards/rejected": -1.308828592300415, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.591915325433034e-06, |
|
"logits/chosen": -4.177929878234863, |
|
"logits/rejected": -4.188526630401611, |
|
"logps/chosen": -370.15655517578125, |
|
"logps/rejected": -401.153564453125, |
|
"loss": 0.6176, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.020612359046936, |
|
"rewards/margins": 0.26860466599464417, |
|
"rewards/rejected": -1.2892169952392578, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.5271758206483664e-06, |
|
"logits/chosen": -4.208674430847168, |
|
"logits/rejected": -4.197338104248047, |
|
"logps/chosen": -399.302490234375, |
|
"logps/rejected": -428.2151794433594, |
|
"loss": 0.6565, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.2150713205337524, |
|
"rewards/margins": 0.18397213518619537, |
|
"rewards/rejected": -1.3990434408187866, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4631962451927966e-06, |
|
"logits/chosen": -4.116250514984131, |
|
"logits/rejected": -4.103257656097412, |
|
"logps/chosen": -396.60467529296875, |
|
"logps/rejected": -433.5016174316406, |
|
"loss": 0.6155, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.1897716522216797, |
|
"rewards/margins": 0.23897738754749298, |
|
"rewards/rejected": -1.4287490844726562, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.4000265772195032e-06, |
|
"logits/chosen": -4.289558410644531, |
|
"logits/rejected": -4.233942985534668, |
|
"logps/chosen": -391.7837219238281, |
|
"logps/rejected": -433.24261474609375, |
|
"loss": 0.6144, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.1339555978775024, |
|
"rewards/margins": 0.25826430320739746, |
|
"rewards/rejected": -1.3922197818756104, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.3377161622160077e-06, |
|
"logits/chosen": -4.226014137268066, |
|
"logits/rejected": -4.22386360168457, |
|
"logps/chosen": -392.7188415527344, |
|
"logps/rejected": -428.2996520996094, |
|
"loss": 0.6116, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.1736845970153809, |
|
"rewards/margins": 0.23641009628772736, |
|
"rewards/rejected": -1.4100948572158813, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.276313674457553e-06, |
|
"logits/chosen": -4.321101188659668, |
|
"logits/rejected": -4.316857814788818, |
|
"logps/chosen": -376.7432556152344, |
|
"logps/rejected": -420.355224609375, |
|
"loss": 0.6062, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.1783860921859741, |
|
"rewards/margins": 0.2604502737522125, |
|
"rewards/rejected": -1.4388363361358643, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2158670789848095e-06, |
|
"logits/chosen": -4.357269287109375, |
|
"logits/rejected": -4.3427653312683105, |
|
"logps/chosen": -412.49603271484375, |
|
"logps/rejected": -446.0122985839844, |
|
"loss": 0.6207, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.3438339233398438, |
|
"rewards/margins": 0.2622073292732239, |
|
"rewards/rejected": -1.6060411930084229, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.1564235941356016e-06, |
|
"logits/chosen": -4.453230381011963, |
|
"logits/rejected": -4.401095390319824, |
|
"logps/chosen": -398.6358337402344, |
|
"logps/rejected": -450.500244140625, |
|
"loss": 0.6262, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.3652451038360596, |
|
"rewards/margins": 0.3177439570426941, |
|
"rewards/rejected": -1.6829888820648193, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 1.0980296546599254e-06, |
|
"logits/chosen": -4.331755638122559, |
|
"logits/rejected": -4.3203840255737305, |
|
"logps/chosen": -422.264404296875, |
|
"logps/rejected": -445.89739990234375, |
|
"loss": 0.6746, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.3867082595825195, |
|
"rewards/margins": 0.28304582834243774, |
|
"rewards/rejected": -1.6697540283203125, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"eval_logits/chosen": -4.577242374420166, |
|
"eval_logits/rejected": -4.566162586212158, |
|
"eval_logps/chosen": -341.29400634765625, |
|
"eval_logps/rejected": -358.15576171875, |
|
"eval_loss": 0.6757350564002991, |
|
"eval_rewards/accuracies": 0.5686666369438171, |
|
"eval_rewards/chosen": -1.3289406299591064, |
|
"eval_rewards/margins": 0.11551385372877121, |
|
"eval_rewards/rejected": -1.4444544315338135, |
|
"eval_runtime": 1566.2371, |
|
"eval_samples_per_second": 1.912, |
|
"eval_steps_per_second": 0.239, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.040730875447083e-06, |
|
"logits/chosen": -4.31623649597168, |
|
"logits/rejected": -4.329934597015381, |
|
"logps/chosen": -407.89544677734375, |
|
"logps/rejected": -431.8106384277344, |
|
"loss": 0.6508, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.282503604888916, |
|
"rewards/margins": 0.16822777688503265, |
|
"rewards/rejected": -1.4507312774658203, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.845720158932414e-07, |
|
"logits/chosen": -4.317078590393066, |
|
"logits/rejected": -4.328789710998535, |
|
"logps/chosen": -384.6829833984375, |
|
"logps/rejected": -404.79107666015625, |
|
"loss": 0.6351, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.2274483442306519, |
|
"rewards/margins": 0.19847750663757324, |
|
"rewards/rejected": -1.4259259700775146, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.295969449372796e-07, |
|
"logits/chosen": -4.379042625427246, |
|
"logits/rejected": -4.360438346862793, |
|
"logps/chosen": -385.20245361328125, |
|
"logps/rejected": -416.4458923339844, |
|
"loss": 0.6352, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.1640335321426392, |
|
"rewards/margins": 0.19592450559139252, |
|
"rewards/rejected": -1.3599579334259033, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 8.758486067922176e-07, |
|
"logits/chosen": -4.383969306945801, |
|
"logits/rejected": -4.344205379486084, |
|
"logps/chosen": -384.3717346191406, |
|
"logps/rejected": -427.7298889160156, |
|
"loss": 0.6135, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.207880973815918, |
|
"rewards/margins": 0.2520582675933838, |
|
"rewards/rejected": -1.4599391222000122, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 8.233689873990006e-07, |
|
"logits/chosen": -4.391127109527588, |
|
"logits/rejected": -4.357823848724365, |
|
"logps/chosen": -389.40570068359375, |
|
"logps/rejected": -443.3841247558594, |
|
"loss": 0.6165, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -1.2565414905548096, |
|
"rewards/margins": 0.3726215362548828, |
|
"rewards/rejected": -1.6291630268096924, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.721990816288555e-07, |
|
"logits/chosen": -4.3561787605285645, |
|
"logits/rejected": -4.322731971740723, |
|
"logps/chosen": -363.4510803222656, |
|
"logps/rejected": -400.2040710449219, |
|
"loss": 0.6128, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.1432923078536987, |
|
"rewards/margins": 0.24396805465221405, |
|
"rewards/rejected": -1.3872603178024292, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.223788612598148e-07, |
|
"logits/chosen": -4.374851703643799, |
|
"logits/rejected": -4.3536810874938965, |
|
"logps/chosen": -397.66680908203125, |
|
"logps/rejected": -428.77227783203125, |
|
"loss": 0.639, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.2013370990753174, |
|
"rewards/margins": 0.15685971081256866, |
|
"rewards/rejected": -1.3581968545913696, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 6.73947243752448e-07, |
|
"logits/chosen": -4.431571006774902, |
|
"logits/rejected": -4.440985679626465, |
|
"logps/chosen": -415.3916931152344, |
|
"logps/rejected": -431.7449645996094, |
|
"loss": 0.656, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.3690643310546875, |
|
"rewards/margins": 0.22862248122692108, |
|
"rewards/rejected": -1.5976868867874146, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.269420618491759e-07, |
|
"logits/chosen": -4.3968610763549805, |
|
"logits/rejected": -4.365904808044434, |
|
"logps/chosen": -384.7822265625, |
|
"logps/rejected": -417.7525939941406, |
|
"loss": 0.6459, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.2827244997024536, |
|
"rewards/margins": 0.17925159633159637, |
|
"rewards/rejected": -1.461976170539856, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.814000340209267e-07, |
|
"logits/chosen": -4.392641067504883, |
|
"logits/rejected": -4.334468841552734, |
|
"logps/chosen": -398.47662353515625, |
|
"logps/rejected": -445.6675720214844, |
|
"loss": 0.6151, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.2670553922653198, |
|
"rewards/margins": 0.2884874939918518, |
|
"rewards/rejected": -1.5555428266525269, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"eval_logits/chosen": -4.618439197540283, |
|
"eval_logits/rejected": -4.607028484344482, |
|
"eval_logps/chosen": -343.98931884765625, |
|
"eval_logps/rejected": -361.1741943359375, |
|
"eval_loss": 0.6755098700523376, |
|
"eval_rewards/accuracies": 0.5690000057220459, |
|
"eval_rewards/chosen": -1.3558939695358276, |
|
"eval_rewards/margins": 0.11874517053365707, |
|
"eval_rewards/rejected": -1.4746389389038086, |
|
"eval_runtime": 1565.397, |
|
"eval_samples_per_second": 1.913, |
|
"eval_steps_per_second": 0.24, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.373567357842111e-07, |
|
"logits/chosen": -4.363351821899414, |
|
"logits/rejected": -4.336891174316406, |
|
"logps/chosen": -393.9075012207031, |
|
"logps/rejected": -432.1922302246094, |
|
"loss": 0.6011, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.2202444076538086, |
|
"rewards/margins": 0.2966518700122833, |
|
"rewards/rejected": -1.516896367073059, |
|
"step": 1010 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.948465719110226e-07, |
|
"logits/chosen": -4.430235862731934, |
|
"logits/rejected": -4.40949010848999, |
|
"logps/chosen": -379.41375732421875, |
|
"logps/rejected": -404.0538635253906, |
|
"loss": 0.6481, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.1881725788116455, |
|
"rewards/margins": 0.17870518565177917, |
|
"rewards/rejected": -1.366877794265747, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.539027495532766e-07, |
|
"logits/chosen": -4.403011322021484, |
|
"logits/rejected": -4.427884578704834, |
|
"logps/chosen": -381.40582275390625, |
|
"logps/rejected": -415.64044189453125, |
|
"loss": 0.6201, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.2115557193756104, |
|
"rewards/margins": 0.2422032654285431, |
|
"rewards/rejected": -1.4537590742111206, |
|
"step": 1030 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.14557252302783e-07, |
|
"logits/chosen": -4.416018962860107, |
|
"logits/rejected": -4.378045082092285, |
|
"logps/chosen": -401.25286865234375, |
|
"logps/rejected": -434.80523681640625, |
|
"loss": 0.6795, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -1.350279450416565, |
|
"rewards/margins": 0.18791106343269348, |
|
"rewards/rejected": -1.538190484046936, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.7684081520700884e-07, |
|
"logits/chosen": -4.323966026306152, |
|
"logits/rejected": -4.319721221923828, |
|
"logps/chosen": -411.38812255859375, |
|
"logps/rejected": -426.6982421875, |
|
"loss": 0.6426, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.2739022970199585, |
|
"rewards/margins": 0.20139846205711365, |
|
"rewards/rejected": -1.475300908088684, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.407829007601507e-07, |
|
"logits/chosen": -4.3614420890808105, |
|
"logits/rejected": -4.317663192749023, |
|
"logps/chosen": -395.0092468261719, |
|
"logps/rejected": -438.6324157714844, |
|
"loss": 0.6132, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.2276242971420288, |
|
"rewards/margins": 0.28230905532836914, |
|
"rewards/rejected": -1.5099334716796875, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 3.064116758882724e-07, |
|
"logits/chosen": -4.330022811889648, |
|
"logits/rejected": -4.274947643280029, |
|
"logps/chosen": -406.9677429199219, |
|
"logps/rejected": -459.18048095703125, |
|
"loss": 0.6063, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.304404616355896, |
|
"rewards/margins": 0.33513855934143066, |
|
"rewards/rejected": -1.6395431756973267, |
|
"step": 1070 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.737539899464908e-07, |
|
"logits/chosen": -4.385619163513184, |
|
"logits/rejected": -4.378711700439453, |
|
"logps/chosen": -370.54156494140625, |
|
"logps/rejected": -411.8634338378906, |
|
"loss": 0.6197, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.1883795261383057, |
|
"rewards/margins": 0.2932683825492859, |
|
"rewards/rejected": -1.4816478490829468, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.4283535374538645e-07, |
|
"logits/chosen": -4.3373942375183105, |
|
"logits/rejected": -4.339685916900635, |
|
"logps/chosen": -400.1960754394531, |
|
"logps/rejected": -430.27557373046875, |
|
"loss": 0.6378, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.2183153629302979, |
|
"rewards/margins": 0.2266024649143219, |
|
"rewards/rejected": -1.444917917251587, |
|
"step": 1090 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.1367991962303298e-07, |
|
"logits/chosen": -4.3288164138793945, |
|
"logits/rejected": -4.31368350982666, |
|
"logps/chosen": -382.65814208984375, |
|
"logps/rejected": -405.8753967285156, |
|
"loss": 0.6837, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -1.2913353443145752, |
|
"rewards/margins": 0.10410255193710327, |
|
"rewards/rejected": -1.3954378366470337, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"eval_logits/chosen": -4.571716785430908, |
|
"eval_logits/rejected": -4.5607380867004395, |
|
"eval_logps/chosen": -340.8576354980469, |
|
"eval_logps/rejected": -358.08392333984375, |
|
"eval_loss": 0.6748068928718567, |
|
"eval_rewards/accuracies": 0.5709999799728394, |
|
"eval_rewards/chosen": -1.3245766162872314, |
|
"eval_rewards/margins": 0.11915931850671768, |
|
"eval_rewards/rejected": -1.4437358379364014, |
|
"eval_runtime": 1563.7394, |
|
"eval_samples_per_second": 1.915, |
|
"eval_steps_per_second": 0.24, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.8631046257820278e-07, |
|
"logits/chosen": -4.340999603271484, |
|
"logits/rejected": -4.353137969970703, |
|
"logps/chosen": -399.4720153808594, |
|
"logps/rejected": -427.0414123535156, |
|
"loss": 0.6277, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.2465407848358154, |
|
"rewards/margins": 0.23380056023597717, |
|
"rewards/rejected": -1.4803413152694702, |
|
"step": 1110 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.6074836247950143e-07, |
|
"logits/chosen": -4.328183174133301, |
|
"logits/rejected": -4.327004432678223, |
|
"logps/chosen": -406.1539306640625, |
|
"logps/rejected": -430.02044677734375, |
|
"loss": 0.6483, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.2271990776062012, |
|
"rewards/margins": 0.17618249356746674, |
|
"rewards/rejected": -1.4033817052841187, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.370135873643097e-07, |
|
"logits/chosen": -4.33071231842041, |
|
"logits/rejected": -4.364874839782715, |
|
"logps/chosen": -403.37628173828125, |
|
"logps/rejected": -413.484375, |
|
"loss": 0.6427, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.2655575275421143, |
|
"rewards/margins": 0.19985264539718628, |
|
"rewards/rejected": -1.4654099941253662, |
|
"step": 1130 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.1512467784059372e-07, |
|
"logits/chosen": -4.4061384201049805, |
|
"logits/rejected": -4.369538307189941, |
|
"logps/chosen": -367.4423522949219, |
|
"logps/rejected": -402.65911865234375, |
|
"loss": 0.6186, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.1879181861877441, |
|
"rewards/margins": 0.27168983221054077, |
|
"rewards/rejected": -1.4596080780029297, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 9.509873260376251e-08, |
|
"logits/chosen": -4.3491435050964355, |
|
"logits/rejected": -4.2928547859191895, |
|
"logps/chosen": -394.6898498535156, |
|
"logps/rejected": -461.10845947265625, |
|
"loss": 0.583, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.2356455326080322, |
|
"rewards/margins": 0.3976030647754669, |
|
"rewards/rejected": -1.6332485675811768, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.695139507988559e-08, |
|
"logits/chosen": -4.359416961669922, |
|
"logits/rejected": -4.381959915161133, |
|
"logps/chosen": -409.65643310546875, |
|
"logps/rejected": -434.98675537109375, |
|
"loss": 0.6521, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -1.2608633041381836, |
|
"rewards/margins": 0.1726626455783844, |
|
"rewards/rejected": -1.4335259199142456, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 6.069684120570684e-08, |
|
"logits/chosen": -4.381880283355713, |
|
"logits/rejected": -4.29121208190918, |
|
"logps/chosen": -395.4755554199219, |
|
"logps/rejected": -438.13427734375, |
|
"loss": 0.6418, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.3113107681274414, |
|
"rewards/margins": 0.23519746959209442, |
|
"rewards/rejected": -1.5465081930160522, |
|
"step": 1170 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 4.634776835499871e-08, |
|
"logits/chosen": -4.314273357391357, |
|
"logits/rejected": -4.296536445617676, |
|
"logps/chosen": -382.25958251953125, |
|
"logps/rejected": -419.88897705078125, |
|
"loss": 0.6354, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.2596309185028076, |
|
"rewards/margins": 0.24562938511371613, |
|
"rewards/rejected": -1.5052604675292969, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 3.3915385419908964e-08, |
|
"logits/chosen": -4.292089939117432, |
|
"logits/rejected": -4.3260178565979, |
|
"logps/chosen": -391.3450927734375, |
|
"logps/rejected": -419.9061584472656, |
|
"loss": 0.6332, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -1.2074798345565796, |
|
"rewards/margins": 0.2409057915210724, |
|
"rewards/rejected": -1.4483857154846191, |
|
"step": 1190 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.3409404055043938e-08, |
|
"logits/chosen": -4.403561592102051, |
|
"logits/rejected": -4.3759870529174805, |
|
"logps/chosen": -400.16522216796875, |
|
"logps/rejected": -427.40606689453125, |
|
"loss": 0.6539, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.288109302520752, |
|
"rewards/margins": 0.17616990208625793, |
|
"rewards/rejected": -1.4642791748046875, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"eval_logits/chosen": -4.559507846832275, |
|
"eval_logits/rejected": -4.548335075378418, |
|
"eval_logps/chosen": -340.22308349609375, |
|
"eval_logps/rejected": -357.4035949707031, |
|
"eval_loss": 0.6746300458908081, |
|
"eval_rewards/accuracies": 0.5730000138282776, |
|
"eval_rewards/chosen": -1.318231225013733, |
|
"eval_rewards/margins": 0.11870139837265015, |
|
"eval_rewards/rejected": -1.4369325637817383, |
|
"eval_runtime": 1563.0275, |
|
"eval_samples_per_second": 1.916, |
|
"eval_steps_per_second": 0.24, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.4838031091134186e-08, |
|
"logits/chosen": -4.341926574707031, |
|
"logits/rejected": -4.2704925537109375, |
|
"logps/chosen": -373.17681884765625, |
|
"logps/rejected": -431.8099060058594, |
|
"loss": 0.6065, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.2158408164978027, |
|
"rewards/margins": 0.33925992250442505, |
|
"rewards/rejected": -1.555100679397583, |
|
"step": 1210 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 8.207962124201774e-09, |
|
"logits/chosen": -4.371233940124512, |
|
"logits/rejected": -4.343363285064697, |
|
"logps/chosen": -401.8610534667969, |
|
"logps/rejected": -437.0784606933594, |
|
"loss": 0.6005, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.1877615451812744, |
|
"rewards/margins": 0.29727452993392944, |
|
"rewards/rejected": -1.4850361347198486, |
|
"step": 1220 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 3.5243762852441023e-09, |
|
"logits/chosen": -4.321781158447266, |
|
"logits/rejected": -4.277449131011963, |
|
"logps/chosen": -392.8436279296875, |
|
"logps/rejected": -430.36279296875, |
|
"loss": 0.6401, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.2518620491027832, |
|
"rewards/margins": 0.2195155918598175, |
|
"rewards/rejected": -1.4713777303695679, |
|
"step": 1230 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 7.909321945129278e-10, |
|
"logits/chosen": -4.296388626098633, |
|
"logits/rejected": -4.257707118988037, |
|
"logps/chosen": -408.13409423828125, |
|
"logps/rejected": -451.42462158203125, |
|
"loss": 0.6106, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.2207772731781006, |
|
"rewards/margins": 0.29706281423568726, |
|
"rewards/rejected": -1.5178401470184326, |
|
"step": 1240 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 1249, |
|
"total_flos": 0.0, |
|
"train_loss": 0.6508825262419409, |
|
"train_runtime": 40390.1729, |
|
"train_samples_per_second": 0.495, |
|
"train_steps_per_second": 0.031 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 1249, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 100, |
|
"total_flos": 0.0, |
|
"train_batch_size": 4, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|