| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.0, | |
| "eval_steps": 100, | |
| "global_step": 385, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 656.0, | |
| "learning_rate": 1.282051282051282e-07, | |
| "logits/chosen": 88.18099975585938, | |
| "logits/rejected": 88.25153350830078, | |
| "logps/chosen": -29.073104858398438, | |
| "logps/rejected": -26.25731658935547, | |
| "loss": 6.25, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 536.0, | |
| "learning_rate": 1.282051282051282e-06, | |
| "logits/chosen": 81.07795715332031, | |
| "logits/rejected": 80.78132629394531, | |
| "logps/chosen": -34.14677047729492, | |
| "logps/rejected": -33.015296936035156, | |
| "loss": 6.1804, | |
| "rewards/accuracies": 0.5277777910232544, | |
| "rewards/chosen": 0.0189919825643301, | |
| "rewards/margins": 0.030369587242603302, | |
| "rewards/rejected": -0.011377605609595776, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 592.0, | |
| "learning_rate": 2.564102564102564e-06, | |
| "logits/chosen": 80.67132568359375, | |
| "logits/rejected": 80.56291198730469, | |
| "logps/chosen": -33.51841354370117, | |
| "logps/rejected": -30.76877212524414, | |
| "loss": 6.6549, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.034299030900001526, | |
| "rewards/margins": 0.03255577012896538, | |
| "rewards/rejected": 0.0017432533204555511, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 448.0, | |
| "learning_rate": 3.846153846153847e-06, | |
| "logits/chosen": 82.46919250488281, | |
| "logits/rejected": 82.5018539428711, | |
| "logps/chosen": -33.83555603027344, | |
| "logps/rejected": -31.259449005126953, | |
| "loss": 7.4075, | |
| "rewards/accuracies": 0.44999998807907104, | |
| "rewards/chosen": 0.03545312210917473, | |
| "rewards/margins": 0.0025764047168195248, | |
| "rewards/rejected": 0.03287670761346817, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 612.0, | |
| "learning_rate": 4.999896948438434e-06, | |
| "logits/chosen": 80.91734313964844, | |
| "logits/rejected": 80.9122314453125, | |
| "logps/chosen": -32.870399475097656, | |
| "logps/rejected": -33.192291259765625, | |
| "loss": 6.9708, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.06073303893208504, | |
| "rewards/margins": 0.025485849007964134, | |
| "rewards/rejected": 0.03524719178676605, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 528.0, | |
| "learning_rate": 4.987541037542187e-06, | |
| "logits/chosen": 78.42672729492188, | |
| "logits/rejected": 78.44120788574219, | |
| "logps/chosen": -30.90542984008789, | |
| "logps/rejected": -31.027896881103516, | |
| "loss": 6.7099, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.059963710606098175, | |
| "rewards/margins": 0.05088571831583977, | |
| "rewards/rejected": 0.009077982977032661, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 498.0, | |
| "learning_rate": 4.954691471941119e-06, | |
| "logits/chosen": 82.86418151855469, | |
| "logits/rejected": 82.91536712646484, | |
| "logps/chosen": -31.078943252563477, | |
| "logps/rejected": -29.606725692749023, | |
| "loss": 7.0829, | |
| "rewards/accuracies": 0.4749999940395355, | |
| "rewards/chosen": 0.0192436333745718, | |
| "rewards/margins": 0.022484425455331802, | |
| "rewards/rejected": -0.0032407897524535656, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 960.0, | |
| "learning_rate": 4.901618883413549e-06, | |
| "logits/chosen": 83.54774475097656, | |
| "logits/rejected": 83.57620239257812, | |
| "logps/chosen": -30.6588077545166, | |
| "logps/rejected": -33.22206115722656, | |
| "loss": 6.976, | |
| "rewards/accuracies": 0.5874999761581421, | |
| "rewards/chosen": -0.0059564015828073025, | |
| "rewards/margins": 0.0309152714908123, | |
| "rewards/rejected": -0.03687167167663574, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 556.0, | |
| "learning_rate": 4.828760511501322e-06, | |
| "logits/chosen": 81.12471771240234, | |
| "logits/rejected": 81.11027526855469, | |
| "logps/chosen": -31.388225555419922, | |
| "logps/rejected": -30.88912010192871, | |
| "loss": 6.8226, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.023343011736869812, | |
| "rewards/margins": 0.049637455493211746, | |
| "rewards/rejected": -0.026294440031051636, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 680.0, | |
| "learning_rate": 4.7367166013034295e-06, | |
| "logits/chosen": 78.06856536865234, | |
| "logits/rejected": 78.03953552246094, | |
| "logps/chosen": -32.45366668701172, | |
| "logps/rejected": -31.23443603515625, | |
| "loss": 6.1663, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.03690256550908089, | |
| "rewards/margins": 0.09349562227725983, | |
| "rewards/rejected": -0.056593067944049835, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 396.0, | |
| "learning_rate": 4.626245458345211e-06, | |
| "logits/chosen": 83.3121566772461, | |
| "logits/rejected": 83.32379913330078, | |
| "logps/chosen": -34.064453125, | |
| "logps/rejected": -31.79021644592285, | |
| "loss": 6.7218, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": 0.04810718446969986, | |
| "rewards/margins": 0.05604150891304016, | |
| "rewards/rejected": -0.00793431792408228, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "eval_logits/chosen": 98.67085266113281, | |
| "eval_logits/rejected": 98.65489959716797, | |
| "eval_logps/chosen": -32.457313537597656, | |
| "eval_logps/rejected": -36.04214859008789, | |
| "eval_loss": 7.570494651794434, | |
| "eval_rewards/accuracies": 0.5083056688308716, | |
| "eval_rewards/chosen": -0.002827130490913987, | |
| "eval_rewards/margins": 0.0122976154088974, | |
| "eval_rewards/rejected": -0.015124746598303318, | |
| "eval_runtime": 104.1006, | |
| "eval_samples_per_second": 3.295, | |
| "eval_steps_per_second": 0.413, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 752.0, | |
| "learning_rate": 4.498257201263691e-06, | |
| "logits/chosen": 83.45378112792969, | |
| "logits/rejected": 83.33932495117188, | |
| "logps/chosen": -32.50124740600586, | |
| "logps/rejected": -32.954078674316406, | |
| "loss": 5.6892, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.104311004281044, | |
| "rewards/margins": 0.16394877433776855, | |
| "rewards/rejected": -0.05963778495788574, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "grad_norm": 620.0, | |
| "learning_rate": 4.353806263777678e-06, | |
| "logits/chosen": 83.64091491699219, | |
| "logits/rejected": 83.74860382080078, | |
| "logps/chosen": -28.32021141052246, | |
| "logps/rejected": -35.66899871826172, | |
| "loss": 5.3716, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": 0.12187393009662628, | |
| "rewards/margins": 0.16384465992450714, | |
| "rewards/rejected": -0.04197072237730026, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "grad_norm": 484.0, | |
| "learning_rate": 4.1940827077152755e-06, | |
| "logits/chosen": 80.95816040039062, | |
| "logits/rejected": 80.97350311279297, | |
| "logps/chosen": -30.554763793945312, | |
| "logps/rejected": -32.3652458190918, | |
| "loss": 6.2467, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": 0.07051900774240494, | |
| "rewards/margins": 0.15724298357963562, | |
| "rewards/rejected": -0.08672399073839188, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "grad_norm": 428.0, | |
| "learning_rate": 4.0204024186666215e-06, | |
| "logits/chosen": 82.39835357666016, | |
| "logits/rejected": 82.39920806884766, | |
| "logps/chosen": -27.08688735961914, | |
| "logps/rejected": -33.00067901611328, | |
| "loss": 5.3964, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.07208801805973053, | |
| "rewards/margins": 0.18754826486110687, | |
| "rewards/rejected": -0.11546025425195694, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "grad_norm": 482.0, | |
| "learning_rate": 3.834196265035119e-06, | |
| "logits/chosen": 81.19883728027344, | |
| "logits/rejected": 81.17581939697266, | |
| "logps/chosen": -28.9885311126709, | |
| "logps/rejected": -32.907562255859375, | |
| "loss": 5.3313, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": 0.08011577278375626, | |
| "rewards/margins": 0.14832790195941925, | |
| "rewards/rejected": -0.06821214407682419, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "grad_norm": 1040.0, | |
| "learning_rate": 3.636998309800573e-06, | |
| "logits/chosen": 83.12278747558594, | |
| "logits/rejected": 83.13607025146484, | |
| "logps/chosen": -33.499732971191406, | |
| "logps/rejected": -30.342330932617188, | |
| "loss": 6.478, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.1274038851261139, | |
| "rewards/margins": 0.2053629606962204, | |
| "rewards/rejected": -0.07795912027359009, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "grad_norm": 684.0, | |
| "learning_rate": 3.4304331721118078e-06, | |
| "logits/chosen": 84.02305603027344, | |
| "logits/rejected": 83.95697784423828, | |
| "logps/chosen": -30.76401710510254, | |
| "logps/rejected": -32.3089599609375, | |
| "loss": 5.8957, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": 0.09905388206243515, | |
| "rewards/margins": 0.1568334400653839, | |
| "rewards/rejected": -0.057779550552368164, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "grad_norm": 498.0, | |
| "learning_rate": 3.2162026428305436e-06, | |
| "logits/chosen": 81.92304992675781, | |
| "logits/rejected": 81.90460205078125, | |
| "logps/chosen": -30.485610961914062, | |
| "logps/rejected": -31.443450927734375, | |
| "loss": 5.0645, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.1423078328371048, | |
| "rewards/margins": 0.19955609738826752, | |
| "rewards/rejected": -0.05724828690290451, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "grad_norm": 502.0, | |
| "learning_rate": 2.996071664294641e-06, | |
| "logits/chosen": 83.73831176757812, | |
| "logits/rejected": 83.71122741699219, | |
| "logps/chosen": -30.49334716796875, | |
| "logps/rejected": -30.74527931213379, | |
| "loss": 7.3479, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.05461405590176582, | |
| "rewards/margins": 0.11006426811218262, | |
| "rewards/rejected": -0.055450212210416794, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "grad_norm": 294.0, | |
| "learning_rate": 2.7718537898066833e-06, | |
| "logits/chosen": 79.29591369628906, | |
| "logits/rejected": 79.25688171386719, | |
| "logps/chosen": -34.00603485107422, | |
| "logps/rejected": -32.55823516845703, | |
| "loss": 6.2997, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.1493411809206009, | |
| "rewards/margins": 0.16230711340904236, | |
| "rewards/rejected": -0.012965935282409191, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "eval_logits/chosen": 98.92082977294922, | |
| "eval_logits/rejected": 98.90645599365234, | |
| "eval_logps/chosen": -32.54380416870117, | |
| "eval_logps/rejected": -36.24779510498047, | |
| "eval_loss": 7.145691394805908, | |
| "eval_rewards/accuracies": 0.5402824282646179, | |
| "eval_rewards/chosen": -0.02012600749731064, | |
| "eval_rewards/margins": 0.036127228289842606, | |
| "eval_rewards/rejected": -0.056253232061862946, | |
| "eval_runtime": 104.0138, | |
| "eval_samples_per_second": 3.298, | |
| "eval_steps_per_second": 0.413, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "grad_norm": 1192.0, | |
| "learning_rate": 2.5453962426402006e-06, | |
| "logits/chosen": 81.78749084472656, | |
| "logits/rejected": 81.68329620361328, | |
| "logps/chosen": -33.50835037231445, | |
| "logps/rejected": -35.296775817871094, | |
| "loss": 5.7897, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": 0.08747398853302002, | |
| "rewards/margins": 0.14960959553718567, | |
| "rewards/rejected": -0.062135614454746246, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "grad_norm": 382.0, | |
| "learning_rate": 2.3185646976551794e-06, | |
| "logits/chosen": 83.9058609008789, | |
| "logits/rejected": 83.9858627319336, | |
| "logps/chosen": -31.16668128967285, | |
| "logps/rejected": -30.917465209960938, | |
| "loss": 6.2677, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.15645496547222137, | |
| "rewards/margins": 0.1941571831703186, | |
| "rewards/rejected": -0.03770221397280693, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "grad_norm": 688.0, | |
| "learning_rate": 2.0932279108998323e-06, | |
| "logits/chosen": 81.14820098876953, | |
| "logits/rejected": 81.19589233398438, | |
| "logps/chosen": -32.21388626098633, | |
| "logps/rejected": -34.05015182495117, | |
| "loss": 6.1846, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": 0.11837179958820343, | |
| "rewards/margins": 0.12597264349460602, | |
| "rewards/rejected": -0.007600849959999323, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "grad_norm": 764.0, | |
| "learning_rate": 1.8712423238279358e-06, | |
| "logits/chosen": 83.51747131347656, | |
| "logits/rejected": 83.77994537353516, | |
| "logps/chosen": -30.83194351196289, | |
| "logps/rejected": -31.766277313232422, | |
| "loss": 3.8985, | |
| "rewards/accuracies": 0.7875000238418579, | |
| "rewards/chosen": 0.17122383415699005, | |
| "rewards/margins": 0.21807047724723816, | |
| "rewards/rejected": -0.0468466654419899, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "grad_norm": 528.0, | |
| "learning_rate": 1.6544367689701824e-06, | |
| "logits/chosen": 82.23722076416016, | |
| "logits/rejected": 82.29248809814453, | |
| "logps/chosen": -26.981786727905273, | |
| "logps/rejected": -29.9254093170166, | |
| "loss": 5.8549, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.122822105884552, | |
| "rewards/margins": 0.12775607407093048, | |
| "rewards/rejected": -0.0049339523538947105, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "grad_norm": 744.0, | |
| "learning_rate": 1.4445974030621963e-06, | |
| "logits/chosen": 79.6116943359375, | |
| "logits/rejected": 79.72075653076172, | |
| "logps/chosen": -30.516536712646484, | |
| "logps/rejected": -36.090003967285156, | |
| "loss": 4.7707, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.20889195799827576, | |
| "rewards/margins": 0.20656757056713104, | |
| "rewards/rejected": 0.0023243515752255917, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "grad_norm": 348.0, | |
| "learning_rate": 1.243452991757889e-06, | |
| "logits/chosen": 79.08982849121094, | |
| "logits/rejected": 79.111083984375, | |
| "logps/chosen": -30.892492294311523, | |
| "logps/rejected": -31.574138641357422, | |
| "loss": 4.8727, | |
| "rewards/accuracies": 0.762499988079071, | |
| "rewards/chosen": 0.1772860586643219, | |
| "rewards/margins": 0.1910295933485031, | |
| "rewards/rejected": -0.013743534684181213, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "grad_norm": 700.0, | |
| "learning_rate": 1.0526606671603523e-06, | |
| "logits/chosen": 81.58210754394531, | |
| "logits/rejected": 81.3693618774414, | |
| "logps/chosen": -31.06771469116211, | |
| "logps/rejected": -29.591297149658203, | |
| "loss": 7.3884, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.14843526482582092, | |
| "rewards/margins": 0.13329505920410156, | |
| "rewards/rejected": 0.015140200033783913, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 428.0, | |
| "learning_rate": 8.737922755071455e-07, | |
| "logits/chosen": 81.82354736328125, | |
| "logits/rejected": 81.73272705078125, | |
| "logps/chosen": -33.02164077758789, | |
| "logps/rejected": -32.17236328125, | |
| "loss": 4.7078, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.21652066707611084, | |
| "rewards/margins": 0.24762043356895447, | |
| "rewards/rejected": -0.03109976276755333, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 704.0, | |
| "learning_rate": 7.08321427484816e-07, | |
| "logits/chosen": 77.42338562011719, | |
| "logits/rejected": 77.51200866699219, | |
| "logps/chosen": -32.24098205566406, | |
| "logps/rejected": -29.014667510986328, | |
| "loss": 6.0651, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": 0.2329459935426712, | |
| "rewards/margins": 0.21547456085681915, | |
| "rewards/rejected": 0.017471451312303543, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "eval_logits/chosen": 99.00227355957031, | |
| "eval_logits/rejected": 98.98667907714844, | |
| "eval_logps/chosen": -32.43557357788086, | |
| "eval_logps/rejected": -36.024574279785156, | |
| "eval_loss": 7.648494243621826, | |
| "eval_rewards/accuracies": 0.5390365719795227, | |
| "eval_rewards/chosen": 0.0015206302050501108, | |
| "eval_rewards/margins": 0.01312990766018629, | |
| "eval_rewards/rejected": -0.011609279550611973, | |
| "eval_runtime": 104.0603, | |
| "eval_samples_per_second": 3.296, | |
| "eval_steps_per_second": 0.413, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "grad_norm": 1200.0, | |
| "learning_rate": 5.576113578589035e-07, | |
| "logits/chosen": 84.3228759765625, | |
| "logits/rejected": 84.34573364257812, | |
| "logps/chosen": -29.936237335205078, | |
| "logps/rejected": -32.30895233154297, | |
| "loss": 5.3743, | |
| "rewards/accuracies": 0.75, | |
| "rewards/chosen": 0.18545907735824585, | |
| "rewards/margins": 0.2072969228029251, | |
| "rewards/rejected": -0.02183784916996956, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "grad_norm": 406.0, | |
| "learning_rate": 4.229036944380913e-07, | |
| "logits/chosen": 81.96125030517578, | |
| "logits/rejected": 81.95556640625, | |
| "logps/chosen": -30.652542114257812, | |
| "logps/rejected": -28.881006240844727, | |
| "loss": 5.1557, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.18920376896858215, | |
| "rewards/margins": 0.18150660395622253, | |
| "rewards/rejected": 0.007697162684053183, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "grad_norm": 576.0, | |
| "learning_rate": 3.053082288996112e-07, | |
| "logits/chosen": 79.27213287353516, | |
| "logits/rejected": 79.31598663330078, | |
| "logps/chosen": -29.210277557373047, | |
| "logps/rejected": -32.73192596435547, | |
| "loss": 5.7764, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": 0.22846612334251404, | |
| "rewards/margins": 0.2478782832622528, | |
| "rewards/rejected": -0.019412150606513023, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 708.0, | |
| "learning_rate": 2.0579377374915805e-07, | |
| "logits/chosen": 83.35621643066406, | |
| "logits/rejected": 83.36161041259766, | |
| "logps/chosen": -32.25800323486328, | |
| "logps/rejected": -33.48296356201172, | |
| "loss": 5.3324, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.19082824885845184, | |
| "rewards/margins": 0.213682621717453, | |
| "rewards/rejected": -0.022854382172226906, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "grad_norm": 322.0, | |
| "learning_rate": 1.2518018074041684e-07, | |
| "logits/chosen": 82.49974822998047, | |
| "logits/rejected": 82.50485229492188, | |
| "logps/chosen": -32.731300354003906, | |
| "logps/rejected": -33.191688537597656, | |
| "loss": 4.9862, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.18248799443244934, | |
| "rewards/margins": 0.20246508717536926, | |
| "rewards/rejected": -0.019977085292339325, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 520.0, | |
| "learning_rate": 6.41315865106129e-08, | |
| "logits/chosen": 83.8789291381836, | |
| "logits/rejected": 83.89030456542969, | |
| "logps/chosen": -28.570327758789062, | |
| "logps/rejected": -31.54669761657715, | |
| "loss": 5.4694, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": 0.19637688994407654, | |
| "rewards/margins": 0.17574702203273773, | |
| "rewards/rejected": 0.02062985673546791, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 370.0, | |
| "learning_rate": 2.3150941078050325e-08, | |
| "logits/chosen": 83.33052062988281, | |
| "logits/rejected": 83.347900390625, | |
| "logps/chosen": -31.880367279052734, | |
| "logps/rejected": -35.12144470214844, | |
| "loss": 7.7502, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.1991589069366455, | |
| "rewards/margins": 0.20157304406166077, | |
| "rewards/rejected": -0.0024141333997249603, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "grad_norm": 552.0, | |
| "learning_rate": 2.575864278703266e-09, | |
| "logits/chosen": 77.32600402832031, | |
| "logits/rejected": 77.20854187011719, | |
| "logps/chosen": -29.861562728881836, | |
| "logps/rejected": -28.112417221069336, | |
| "loss": 5.5564, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.14291271567344666, | |
| "rewards/margins": 0.13328143954277039, | |
| "rewards/rejected": 0.009631268680095673, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 385, | |
| "total_flos": 0.0, | |
| "train_loss": 5.985443219271573, | |
| "train_runtime": 2557.1257, | |
| "train_samples_per_second": 1.204, | |
| "train_steps_per_second": 0.151 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 385, | |
| "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 | |
| } | |