| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9985553308292401, | |
| "eval_steps": 100, | |
| "global_step": 432, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.023114706732158336, | |
| "grad_norm": 67.55843353271484, | |
| "learning_rate": 2.2727272727272726e-07, | |
| "logits/chosen": -0.33515310287475586, | |
| "logits/rejected": -0.31493520736694336, | |
| "logps/chosen": -269.29345703125, | |
| "logps/rejected": -267.6085205078125, | |
| "loss": 2.9222, | |
| "nll_loss": 1.0535928010940552, | |
| "rewards/accuracies": 0.4906249940395355, | |
| "rewards/chosen": -26.929346084594727, | |
| "rewards/margins": -0.16849184036254883, | |
| "rewards/rejected": -26.760854721069336, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.04622941346431667, | |
| "grad_norm": 57.353782653808594, | |
| "learning_rate": 4.545454545454545e-07, | |
| "logits/chosen": -0.3512418866157532, | |
| "logits/rejected": -0.3330627977848053, | |
| "logps/chosen": -260.02117919921875, | |
| "logps/rejected": -266.5718078613281, | |
| "loss": 2.8619, | |
| "nll_loss": 0.9778251647949219, | |
| "rewards/accuracies": 0.565625011920929, | |
| "rewards/chosen": -26.002117156982422, | |
| "rewards/margins": 0.6550623178482056, | |
| "rewards/rejected": -26.657180786132812, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.06934412019647501, | |
| "grad_norm": 57.872154235839844, | |
| "learning_rate": 6.818181818181817e-07, | |
| "logits/chosen": -0.38214609026908875, | |
| "logits/rejected": -0.36934739351272583, | |
| "logps/chosen": -243.16397094726562, | |
| "logps/rejected": -246.39990234375, | |
| "loss": 2.6585, | |
| "nll_loss": 1.018157958984375, | |
| "rewards/accuracies": 0.512499988079071, | |
| "rewards/chosen": -24.316394805908203, | |
| "rewards/margins": 0.3235955238342285, | |
| "rewards/rejected": -24.639989852905273, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.09245882692863334, | |
| "grad_norm": 51.343589782714844, | |
| "learning_rate": 9.09090909090909e-07, | |
| "logits/chosen": -0.6919909119606018, | |
| "logits/rejected": -0.6750722527503967, | |
| "logps/chosen": -202.0379180908203, | |
| "logps/rejected": -203.7409210205078, | |
| "loss": 2.385, | |
| "nll_loss": 0.866622805595398, | |
| "rewards/accuracies": 0.5218750238418579, | |
| "rewards/chosen": -20.203792572021484, | |
| "rewards/margins": 0.17029908299446106, | |
| "rewards/rejected": -20.37409019470215, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.11557353366079168, | |
| "grad_norm": 45.946136474609375, | |
| "learning_rate": 9.845360824742267e-07, | |
| "logits/chosen": -0.8110788464546204, | |
| "logits/rejected": -0.7866124510765076, | |
| "logps/chosen": -176.52212524414062, | |
| "logps/rejected": -175.50418090820312, | |
| "loss": 2.1668, | |
| "nll_loss": 0.4587995409965515, | |
| "rewards/accuracies": 0.5062500238418579, | |
| "rewards/chosen": -17.65221405029297, | |
| "rewards/margins": -0.10179616510868073, | |
| "rewards/rejected": -17.550418853759766, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.13868824039295002, | |
| "grad_norm": 55.89133834838867, | |
| "learning_rate": 9.587628865979382e-07, | |
| "logits/chosen": -0.635937511920929, | |
| "logits/rejected": -0.6386198997497559, | |
| "logps/chosen": -158.75927734375, | |
| "logps/rejected": -159.04640197753906, | |
| "loss": 1.9369, | |
| "nll_loss": 0.41199779510498047, | |
| "rewards/accuracies": 0.528124988079071, | |
| "rewards/chosen": -15.87592887878418, | |
| "rewards/margins": 0.028712665662169456, | |
| "rewards/rejected": -15.904638290405273, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.16180294712510834, | |
| "grad_norm": 51.20368957519531, | |
| "learning_rate": 9.329896907216495e-07, | |
| "logits/chosen": -0.46112680435180664, | |
| "logits/rejected": -0.4352455139160156, | |
| "logps/chosen": -154.1770782470703, | |
| "logps/rejected": -156.22377014160156, | |
| "loss": 1.8871, | |
| "nll_loss": 0.3339909613132477, | |
| "rewards/accuracies": 0.5218750238418579, | |
| "rewards/chosen": -15.417707443237305, | |
| "rewards/margins": 0.20467153191566467, | |
| "rewards/rejected": -15.622377395629883, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.1849176538572667, | |
| "grad_norm": 46.16290283203125, | |
| "learning_rate": 9.072164948453608e-07, | |
| "logits/chosen": -0.36660072207450867, | |
| "logits/rejected": -0.3421391248703003, | |
| "logps/chosen": -158.54318237304688, | |
| "logps/rejected": -161.6597442626953, | |
| "loss": 1.7235, | |
| "nll_loss": 0.2865411043167114, | |
| "rewards/accuracies": 0.5093749761581421, | |
| "rewards/chosen": -15.85431957244873, | |
| "rewards/margins": 0.311655193567276, | |
| "rewards/rejected": -16.165973663330078, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.208032360589425, | |
| "grad_norm": 49.6414680480957, | |
| "learning_rate": 8.814432989690721e-07, | |
| "logits/chosen": -0.3728088140487671, | |
| "logits/rejected": -0.3610958456993103, | |
| "logps/chosen": -153.40731811523438, | |
| "logps/rejected": -161.5341796875, | |
| "loss": 1.6191, | |
| "nll_loss": 0.255443274974823, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -15.340731620788574, | |
| "rewards/margins": 0.812686562538147, | |
| "rewards/rejected": -16.153419494628906, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.23114706732158335, | |
| "grad_norm": 45.195945739746094, | |
| "learning_rate": 8.556701030927834e-07, | |
| "logits/chosen": -0.34290799498558044, | |
| "logits/rejected": -0.3276888430118561, | |
| "logps/chosen": -156.65350341796875, | |
| "logps/rejected": -159.73312377929688, | |
| "loss": 1.7833, | |
| "nll_loss": 0.2992493510246277, | |
| "rewards/accuracies": 0.546875, | |
| "rewards/chosen": -15.665351867675781, | |
| "rewards/margins": 0.30796217918395996, | |
| "rewards/rejected": -15.973315238952637, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.23114706732158335, | |
| "eval_logits/chosen": -0.3332868814468384, | |
| "eval_logits/rejected": -0.30650395154953003, | |
| "eval_logps/chosen": -153.93896484375, | |
| "eval_logps/rejected": -157.79205322265625, | |
| "eval_loss": 1.6389189958572388, | |
| "eval_nll_loss": 0.26784786581993103, | |
| "eval_rewards/accuracies": 0.5782608985900879, | |
| "eval_rewards/chosen": -15.393896102905273, | |
| "eval_rewards/margins": 0.38530847430229187, | |
| "eval_rewards/rejected": -15.779205322265625, | |
| "eval_runtime": 77.324, | |
| "eval_samples_per_second": 23.615, | |
| "eval_steps_per_second": 1.487, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.2542617740537417, | |
| "grad_norm": 46.2556037902832, | |
| "learning_rate": 8.298969072164948e-07, | |
| "logits/chosen": -0.28695493936538696, | |
| "logits/rejected": -0.2580588459968567, | |
| "logps/chosen": -150.8438262939453, | |
| "logps/rejected": -155.1436767578125, | |
| "loss": 1.5662, | |
| "nll_loss": 0.2598631978034973, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -15.084381103515625, | |
| "rewards/margins": 0.4299860894680023, | |
| "rewards/rejected": -15.514368057250977, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.27737648078590005, | |
| "grad_norm": 44.54212188720703, | |
| "learning_rate": 8.041237113402062e-07, | |
| "logits/chosen": -0.3080243170261383, | |
| "logits/rejected": -0.2822437286376953, | |
| "logps/chosen": -157.50094604492188, | |
| "logps/rejected": -159.069580078125, | |
| "loss": 1.6159, | |
| "nll_loss": 0.30583077669143677, | |
| "rewards/accuracies": 0.5406249761581421, | |
| "rewards/chosen": -15.750096321105957, | |
| "rewards/margins": 0.1568617820739746, | |
| "rewards/rejected": -15.906957626342773, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.30049118751805837, | |
| "grad_norm": 55.04152297973633, | |
| "learning_rate": 7.783505154639175e-07, | |
| "logits/chosen": -0.30607444047927856, | |
| "logits/rejected": -0.2829833924770355, | |
| "logps/chosen": -157.97000122070312, | |
| "logps/rejected": -166.240478515625, | |
| "loss": 1.6775, | |
| "nll_loss": 0.2982478737831116, | |
| "rewards/accuracies": 0.596875011920929, | |
| "rewards/chosen": -15.79699993133545, | |
| "rewards/margins": 0.8270493745803833, | |
| "rewards/rejected": -16.624048233032227, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.3236058942502167, | |
| "grad_norm": 68.56768035888672, | |
| "learning_rate": 7.525773195876288e-07, | |
| "logits/chosen": -0.38547760248184204, | |
| "logits/rejected": -0.3799718916416168, | |
| "logps/chosen": -148.39646911621094, | |
| "logps/rejected": -153.57923889160156, | |
| "loss": 1.6639, | |
| "nll_loss": 0.2873283624649048, | |
| "rewards/accuracies": 0.581250011920929, | |
| "rewards/chosen": -14.83964729309082, | |
| "rewards/margins": 0.5182766318321228, | |
| "rewards/rejected": -15.35792350769043, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.34672060098237506, | |
| "grad_norm": 61.17930603027344, | |
| "learning_rate": 7.268041237113402e-07, | |
| "logits/chosen": -0.4227396845817566, | |
| "logits/rejected": -0.41468995809555054, | |
| "logps/chosen": -147.5921173095703, | |
| "logps/rejected": -153.60935974121094, | |
| "loss": 1.7192, | |
| "nll_loss": 0.28114941716194153, | |
| "rewards/accuracies": 0.609375, | |
| "rewards/chosen": -14.759210586547852, | |
| "rewards/margins": 0.6017246246337891, | |
| "rewards/rejected": -15.360937118530273, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.3698353077145334, | |
| "grad_norm": 41.435455322265625, | |
| "learning_rate": 7.010309278350515e-07, | |
| "logits/chosen": -0.45233017206192017, | |
| "logits/rejected": -0.4274457097053528, | |
| "logps/chosen": -162.49929809570312, | |
| "logps/rejected": -166.28240966796875, | |
| "loss": 1.5534, | |
| "nll_loss": 0.28693827986717224, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -16.249929428100586, | |
| "rewards/margins": 0.3783140480518341, | |
| "rewards/rejected": -16.62824249267578, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.3929500144466917, | |
| "grad_norm": 47.89544677734375, | |
| "learning_rate": 6.752577319587629e-07, | |
| "logits/chosen": -0.36802831292152405, | |
| "logits/rejected": -0.35963720083236694, | |
| "logps/chosen": -158.8728790283203, | |
| "logps/rejected": -166.11460876464844, | |
| "loss": 1.4425, | |
| "nll_loss": 0.2846711277961731, | |
| "rewards/accuracies": 0.578125, | |
| "rewards/chosen": -15.887288093566895, | |
| "rewards/margins": 0.7241734862327576, | |
| "rewards/rejected": -16.611461639404297, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.41606472117885, | |
| "grad_norm": 47.56120681762695, | |
| "learning_rate": 6.494845360824742e-07, | |
| "logits/chosen": -0.4111746847629547, | |
| "logits/rejected": -0.4110477566719055, | |
| "logps/chosen": -152.79598999023438, | |
| "logps/rejected": -159.20770263671875, | |
| "loss": 1.4742, | |
| "nll_loss": 0.2701548635959625, | |
| "rewards/accuracies": 0.5531250238418579, | |
| "rewards/chosen": -15.279600143432617, | |
| "rewards/margins": 0.6411706805229187, | |
| "rewards/rejected": -15.920770645141602, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.4391794279110084, | |
| "grad_norm": 51.05569076538086, | |
| "learning_rate": 6.237113402061855e-07, | |
| "logits/chosen": -0.4827125668525696, | |
| "logits/rejected": -0.4709135591983795, | |
| "logps/chosen": -161.96566772460938, | |
| "logps/rejected": -168.30032348632812, | |
| "loss": 1.546, | |
| "nll_loss": 0.29454725980758667, | |
| "rewards/accuracies": 0.5843750238418579, | |
| "rewards/chosen": -16.19656753540039, | |
| "rewards/margins": 0.63346266746521, | |
| "rewards/rejected": -16.83003044128418, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.4622941346431667, | |
| "grad_norm": 44.95571517944336, | |
| "learning_rate": 5.979381443298969e-07, | |
| "logits/chosen": -0.426901638507843, | |
| "logits/rejected": -0.42319783568382263, | |
| "logps/chosen": -160.36892700195312, | |
| "logps/rejected": -166.8142547607422, | |
| "loss": 1.5321, | |
| "nll_loss": 0.29501739144325256, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -16.03689193725586, | |
| "rewards/margins": 0.644534707069397, | |
| "rewards/rejected": -16.681427001953125, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.4622941346431667, | |
| "eval_logits/chosen": -0.4423229396343231, | |
| "eval_logits/rejected": -0.42438310384750366, | |
| "eval_logps/chosen": -158.98841857910156, | |
| "eval_logps/rejected": -165.12060546875, | |
| "eval_loss": 1.524242877960205, | |
| "eval_nll_loss": 0.2763634920120239, | |
| "eval_rewards/accuracies": 0.5978260636329651, | |
| "eval_rewards/chosen": -15.898841857910156, | |
| "eval_rewards/margins": 0.6132183074951172, | |
| "eval_rewards/rejected": -16.512060165405273, | |
| "eval_runtime": 77.1561, | |
| "eval_samples_per_second": 23.666, | |
| "eval_steps_per_second": 1.49, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.48540884137532503, | |
| "grad_norm": 52.92426300048828, | |
| "learning_rate": 5.721649484536082e-07, | |
| "logits/chosen": -0.45326417684555054, | |
| "logits/rejected": -0.42941370606422424, | |
| "logps/chosen": -156.1173553466797, | |
| "logps/rejected": -158.58352661132812, | |
| "loss": 1.5104, | |
| "nll_loss": 0.28336626291275024, | |
| "rewards/accuracies": 0.5531250238418579, | |
| "rewards/chosen": -15.611738204956055, | |
| "rewards/margins": 0.246616929769516, | |
| "rewards/rejected": -15.858355522155762, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.5085235481074833, | |
| "grad_norm": 50.266258239746094, | |
| "learning_rate": 5.463917525773195e-07, | |
| "logits/chosen": -0.471625953912735, | |
| "logits/rejected": -0.4460209906101227, | |
| "logps/chosen": -168.0143585205078, | |
| "logps/rejected": -173.90097045898438, | |
| "loss": 1.569, | |
| "nll_loss": 0.28898459672927856, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -16.801433563232422, | |
| "rewards/margins": 0.5886625051498413, | |
| "rewards/rejected": -17.390098571777344, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.5316382548396418, | |
| "grad_norm": 55.33750534057617, | |
| "learning_rate": 5.20618556701031e-07, | |
| "logits/chosen": -0.42029595375061035, | |
| "logits/rejected": -0.4011649191379547, | |
| "logps/chosen": -167.03567504882812, | |
| "logps/rejected": -173.90760803222656, | |
| "loss": 1.5782, | |
| "nll_loss": 0.29872751235961914, | |
| "rewards/accuracies": 0.6031249761581421, | |
| "rewards/chosen": -16.703567504882812, | |
| "rewards/margins": 0.687191903591156, | |
| "rewards/rejected": -17.39076042175293, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.5547529615718001, | |
| "grad_norm": 48.64881896972656, | |
| "learning_rate": 4.948453608247422e-07, | |
| "logits/chosen": -0.37133297324180603, | |
| "logits/rejected": -0.3616218566894531, | |
| "logps/chosen": -166.56610107421875, | |
| "logps/rejected": -171.88787841796875, | |
| "loss": 1.4277, | |
| "nll_loss": 0.2906336784362793, | |
| "rewards/accuracies": 0.559374988079071, | |
| "rewards/chosen": -16.6566104888916, | |
| "rewards/margins": 0.532177746295929, | |
| "rewards/rejected": -17.18878746032715, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.5778676683039584, | |
| "grad_norm": 53.73638916015625, | |
| "learning_rate": 4.6907216494845357e-07, | |
| "logits/chosen": -0.3499279022216797, | |
| "logits/rejected": -0.3348368704319, | |
| "logps/chosen": -161.66017150878906, | |
| "logps/rejected": -165.3099822998047, | |
| "loss": 1.5715, | |
| "nll_loss": 0.24935810267925262, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": -16.166015625, | |
| "rewards/margins": 0.3649832606315613, | |
| "rewards/rejected": -16.5310001373291, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.6009823750361167, | |
| "grad_norm": 51.90835952758789, | |
| "learning_rate": 4.432989690721649e-07, | |
| "logits/chosen": -0.46265679597854614, | |
| "logits/rejected": -0.4500146508216858, | |
| "logps/chosen": -158.95870971679688, | |
| "logps/rejected": -167.43258666992188, | |
| "loss": 1.4266, | |
| "nll_loss": 0.2810400724411011, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -15.895869255065918, | |
| "rewards/margins": 0.8473905324935913, | |
| "rewards/rejected": -16.743261337280273, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.624097081768275, | |
| "grad_norm": 43.483646392822266, | |
| "learning_rate": 4.175257731958763e-07, | |
| "logits/chosen": -0.4157690107822418, | |
| "logits/rejected": -0.41007503867149353, | |
| "logps/chosen": -158.64816284179688, | |
| "logps/rejected": -165.8525848388672, | |
| "loss": 1.4656, | |
| "nll_loss": 0.2797052264213562, | |
| "rewards/accuracies": 0.6156250238418579, | |
| "rewards/chosen": -15.864816665649414, | |
| "rewards/margins": 0.7204429507255554, | |
| "rewards/rejected": -16.58526039123535, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.6472117885004334, | |
| "grad_norm": 55.48883819580078, | |
| "learning_rate": 3.917525773195876e-07, | |
| "logits/chosen": -0.34205523133277893, | |
| "logits/rejected": -0.3176972568035126, | |
| "logps/chosen": -161.7351531982422, | |
| "logps/rejected": -171.16500854492188, | |
| "loss": 1.4658, | |
| "nll_loss": 0.29218602180480957, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": -16.173513412475586, | |
| "rewards/margins": 0.9429864883422852, | |
| "rewards/rejected": -17.116500854492188, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.6703264952325917, | |
| "grad_norm": 58.60624694824219, | |
| "learning_rate": 3.659793814432989e-07, | |
| "logits/chosen": -0.3818144202232361, | |
| "logits/rejected": -0.35421329736709595, | |
| "logps/chosen": -165.26620483398438, | |
| "logps/rejected": -171.03085327148438, | |
| "loss": 1.4217, | |
| "nll_loss": 0.29271894693374634, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -16.526620864868164, | |
| "rewards/margins": 0.5764636397361755, | |
| "rewards/rejected": -17.103084564208984, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.6934412019647501, | |
| "grad_norm": 43.18596267700195, | |
| "learning_rate": 3.402061855670103e-07, | |
| "logits/chosen": -0.44146934151649475, | |
| "logits/rejected": -0.4326046109199524, | |
| "logps/chosen": -168.69503784179688, | |
| "logps/rejected": -178.41549682617188, | |
| "loss": 1.4722, | |
| "nll_loss": 0.3029026389122009, | |
| "rewards/accuracies": 0.6156250238418579, | |
| "rewards/chosen": -16.869503021240234, | |
| "rewards/margins": 0.9720472097396851, | |
| "rewards/rejected": -17.841550827026367, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.6934412019647501, | |
| "eval_logits/chosen": -0.3856337070465088, | |
| "eval_logits/rejected": -0.36413437128067017, | |
| "eval_logps/chosen": -160.80313110351562, | |
| "eval_logps/rejected": -168.14109802246094, | |
| "eval_loss": 1.463273048400879, | |
| "eval_nll_loss": 0.27904874086380005, | |
| "eval_rewards/accuracies": 0.6217391490936279, | |
| "eval_rewards/chosen": -16.080310821533203, | |
| "eval_rewards/margins": 0.7337984442710876, | |
| "eval_rewards/rejected": -16.814109802246094, | |
| "eval_runtime": 76.8594, | |
| "eval_samples_per_second": 23.758, | |
| "eval_steps_per_second": 1.496, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.7165559086969084, | |
| "grad_norm": 48.37858200073242, | |
| "learning_rate": 3.1443298969072163e-07, | |
| "logits/chosen": -0.43834060430526733, | |
| "logits/rejected": -0.4327595829963684, | |
| "logps/chosen": -170.76321411132812, | |
| "logps/rejected": -174.53939819335938, | |
| "loss": 1.5353, | |
| "nll_loss": 0.30162352323532104, | |
| "rewards/accuracies": 0.559374988079071, | |
| "rewards/chosen": -17.07632064819336, | |
| "rewards/margins": 0.37761738896369934, | |
| "rewards/rejected": -17.45393943786621, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.7396706154290668, | |
| "grad_norm": 45.166900634765625, | |
| "learning_rate": 2.8865979381443296e-07, | |
| "logits/chosen": -0.4189019799232483, | |
| "logits/rejected": -0.41374340653419495, | |
| "logps/chosen": -159.76388549804688, | |
| "logps/rejected": -168.59652709960938, | |
| "loss": 1.521, | |
| "nll_loss": 0.28271549940109253, | |
| "rewards/accuracies": 0.628125011920929, | |
| "rewards/chosen": -15.976388931274414, | |
| "rewards/margins": 0.8832640647888184, | |
| "rewards/rejected": -16.85965347290039, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.7627853221612251, | |
| "grad_norm": 46.85552215576172, | |
| "learning_rate": 2.6288659793814435e-07, | |
| "logits/chosen": -0.4116067886352539, | |
| "logits/rejected": -0.4161633551120758, | |
| "logps/chosen": -159.32391357421875, | |
| "logps/rejected": -166.09951782226562, | |
| "loss": 1.4975, | |
| "nll_loss": 0.299803227186203, | |
| "rewards/accuracies": 0.628125011920929, | |
| "rewards/chosen": -15.932391166687012, | |
| "rewards/margins": 0.6775625944137573, | |
| "rewards/rejected": -16.609954833984375, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.7859000288933834, | |
| "grad_norm": 47.712646484375, | |
| "learning_rate": 2.3711340206185566e-07, | |
| "logits/chosen": -0.4130411148071289, | |
| "logits/rejected": -0.40607109665870667, | |
| "logps/chosen": -162.58676147460938, | |
| "logps/rejected": -167.87673950195312, | |
| "loss": 1.3896, | |
| "nll_loss": 0.27490609884262085, | |
| "rewards/accuracies": 0.581250011920929, | |
| "rewards/chosen": -16.258676528930664, | |
| "rewards/margins": 0.5289959907531738, | |
| "rewards/rejected": -16.78767204284668, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.8090147356255417, | |
| "grad_norm": 45.12944412231445, | |
| "learning_rate": 2.11340206185567e-07, | |
| "logits/chosen": -0.40707340836524963, | |
| "logits/rejected": -0.4076949656009674, | |
| "logps/chosen": -164.3544921875, | |
| "logps/rejected": -173.22579956054688, | |
| "loss": 1.4861, | |
| "nll_loss": 0.279820054769516, | |
| "rewards/accuracies": 0.6156250238418579, | |
| "rewards/chosen": -16.435449600219727, | |
| "rewards/margins": 0.8871291279792786, | |
| "rewards/rejected": -17.32257843017578, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.8321294423577, | |
| "grad_norm": 55.76411437988281, | |
| "learning_rate": 1.8556701030927835e-07, | |
| "logits/chosen": -0.41376179456710815, | |
| "logits/rejected": -0.4020335078239441, | |
| "logps/chosen": -167.02381896972656, | |
| "logps/rejected": -172.5863800048828, | |
| "loss": 1.5281, | |
| "nll_loss": 0.2744566798210144, | |
| "rewards/accuracies": 0.596875011920929, | |
| "rewards/chosen": -16.702383041381836, | |
| "rewards/margins": 0.5562567114830017, | |
| "rewards/rejected": -17.258638381958008, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.8552441490898585, | |
| "grad_norm": 48.47395706176758, | |
| "learning_rate": 1.5979381443298966e-07, | |
| "logits/chosen": -0.4272083342075348, | |
| "logits/rejected": -0.4035620093345642, | |
| "logps/chosen": -161.60487365722656, | |
| "logps/rejected": -171.45762634277344, | |
| "loss": 1.3967, | |
| "nll_loss": 0.2558661103248596, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": -16.160486221313477, | |
| "rewards/margins": 0.9852760434150696, | |
| "rewards/rejected": -17.145763397216797, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.8783588558220168, | |
| "grad_norm": 50.784507751464844, | |
| "learning_rate": 1.3402061855670102e-07, | |
| "logits/chosen": -0.4153415262699127, | |
| "logits/rejected": -0.403484582901001, | |
| "logps/chosen": -159.54061889648438, | |
| "logps/rejected": -166.57180786132812, | |
| "loss": 1.414, | |
| "nll_loss": 0.2998126149177551, | |
| "rewards/accuracies": 0.612500011920929, | |
| "rewards/chosen": -15.954061508178711, | |
| "rewards/margins": 0.7031179666519165, | |
| "rewards/rejected": -16.657180786132812, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.9014735625541751, | |
| "grad_norm": 55.96210861206055, | |
| "learning_rate": 1.0824742268041237e-07, | |
| "logits/chosen": -0.4493354856967926, | |
| "logits/rejected": -0.43461060523986816, | |
| "logps/chosen": -163.90878295898438, | |
| "logps/rejected": -174.5413055419922, | |
| "loss": 1.4429, | |
| "nll_loss": 0.27769935131073, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -16.39087677001953, | |
| "rewards/margins": 1.063251256942749, | |
| "rewards/rejected": -17.45412826538086, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.9245882692863334, | |
| "grad_norm": 49.90566635131836, | |
| "learning_rate": 8.24742268041237e-08, | |
| "logits/chosen": -0.3982897698879242, | |
| "logits/rejected": -0.37326449155807495, | |
| "logps/chosen": -166.83291625976562, | |
| "logps/rejected": -172.27056884765625, | |
| "loss": 1.4589, | |
| "nll_loss": 0.3044116795063019, | |
| "rewards/accuracies": 0.578125, | |
| "rewards/chosen": -16.683292388916016, | |
| "rewards/margins": 0.5437662601470947, | |
| "rewards/rejected": -17.227060317993164, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.9245882692863334, | |
| "eval_logits/chosen": -0.360569566488266, | |
| "eval_logits/rejected": -0.33890673518180847, | |
| "eval_logps/chosen": -160.21502685546875, | |
| "eval_logps/rejected": -168.04132080078125, | |
| "eval_loss": 1.4446684122085571, | |
| "eval_nll_loss": 0.2798323929309845, | |
| "eval_rewards/accuracies": 0.626086950302124, | |
| "eval_rewards/chosen": -16.021503448486328, | |
| "eval_rewards/margins": 0.7826284766197205, | |
| "eval_rewards/rejected": -16.80413246154785, | |
| "eval_runtime": 76.9997, | |
| "eval_samples_per_second": 23.714, | |
| "eval_steps_per_second": 1.494, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.9477029760184917, | |
| "grad_norm": 50.10642623901367, | |
| "learning_rate": 5.670103092783505e-08, | |
| "logits/chosen": -0.3168153762817383, | |
| "logits/rejected": -0.3060024678707123, | |
| "logps/chosen": -152.92799377441406, | |
| "logps/rejected": -162.26162719726562, | |
| "loss": 1.4003, | |
| "nll_loss": 0.2777474522590637, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -15.29279899597168, | |
| "rewards/margins": 0.9333623051643372, | |
| "rewards/rejected": -16.22616195678711, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.9708176827506501, | |
| "grad_norm": 46.16789245605469, | |
| "learning_rate": 3.092783505154639e-08, | |
| "logits/chosen": -0.3549926280975342, | |
| "logits/rejected": -0.33413809537887573, | |
| "logps/chosen": -155.63693237304688, | |
| "logps/rejected": -161.7698211669922, | |
| "loss": 1.4104, | |
| "nll_loss": 0.28400248289108276, | |
| "rewards/accuracies": 0.596875011920929, | |
| "rewards/chosen": -15.563693046569824, | |
| "rewards/margins": 0.6132909059524536, | |
| "rewards/rejected": -16.176982879638672, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.9939323894828085, | |
| "grad_norm": 47.918479919433594, | |
| "learning_rate": 5.154639175257731e-09, | |
| "logits/chosen": -0.37286701798439026, | |
| "logits/rejected": -0.38741737604141235, | |
| "logps/chosen": -163.7307586669922, | |
| "logps/rejected": -171.1845245361328, | |
| "loss": 1.4847, | |
| "nll_loss": 0.26778078079223633, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -16.373075485229492, | |
| "rewards/margins": 0.7453770041465759, | |
| "rewards/rejected": -17.118452072143555, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.9985553308292401, | |
| "step": 432, | |
| "total_flos": 0.0, | |
| "train_loss": 1.6635627029118714, | |
| "train_runtime": 10783.4889, | |
| "train_samples_per_second": 5.135, | |
| "train_steps_per_second": 0.04 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 432, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "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": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |