| { |
| "best_global_step": null, |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 1.0, |
| "eval_steps": 50, |
| "global_step": 537, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.009324009324009324, |
| "grad_norm": 269.7618103027344, |
| "learning_rate": 1.8604651162790698e-07, |
| "loss": 13.883856201171875, |
| "step": 5 |
| }, |
| { |
| "epoch": 0.018648018648018648, |
| "grad_norm": 256.5934143066406, |
| "learning_rate": 4.186046511627907e-07, |
| "loss": 13.883564758300782, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.027972027972027972, |
| "grad_norm": 247.4810028076172, |
| "learning_rate": 6.511627906976745e-07, |
| "loss": 13.845947265625, |
| "step": 15 |
| }, |
| { |
| "epoch": 0.037296037296037296, |
| "grad_norm": 243.70272827148438, |
| "learning_rate": 8.837209302325582e-07, |
| "loss": 13.705046081542969, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.046620046620046623, |
| "grad_norm": 238.2322235107422, |
| "learning_rate": 1.116279069767442e-06, |
| "loss": 13.556683349609376, |
| "step": 25 |
| }, |
| { |
| "epoch": 0.055944055944055944, |
| "grad_norm": 241.71214294433594, |
| "learning_rate": 1.3488372093023258e-06, |
| "loss": 13.285650634765625, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.06526806526806526, |
| "grad_norm": 221.19960021972656, |
| "learning_rate": 1.5813953488372093e-06, |
| "loss": 13.05435791015625, |
| "step": 35 |
| }, |
| { |
| "epoch": 0.07459207459207459, |
| "grad_norm": 205.08326721191406, |
| "learning_rate": 1.8139534883720933e-06, |
| "loss": 12.770874786376954, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.08391608391608392, |
| "grad_norm": 170.5838623046875, |
| "learning_rate": 2.0465116279069768e-06, |
| "loss": 12.280790710449219, |
| "step": 45 |
| }, |
| { |
| "epoch": 0.09324009324009325, |
| "grad_norm": 96.90670776367188, |
| "learning_rate": 2.2790697674418607e-06, |
| "loss": 11.730733489990234, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.09324009324009325, |
| "eval_com_accuracy": 0.2, |
| "eval_rew_accuracy": 0.276, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.10256410256410256, |
| "grad_norm": 61.14415740966797, |
| "learning_rate": 2.5116279069767446e-06, |
| "loss": 11.25308609008789, |
| "step": 55 |
| }, |
| { |
| "epoch": 0.11188811188811189, |
| "grad_norm": 56.027095794677734, |
| "learning_rate": 2.744186046511628e-06, |
| "loss": 11.197045135498048, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.12121212121212122, |
| "grad_norm": 46.54409408569336, |
| "learning_rate": 2.9767441860465116e-06, |
| "loss": 10.874276733398437, |
| "step": 65 |
| }, |
| { |
| "epoch": 0.13053613053613053, |
| "grad_norm": 40.62468719482422, |
| "learning_rate": 3.2093023255813956e-06, |
| "loss": 10.69852294921875, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.13986013986013987, |
| "grad_norm": 37.44975662231445, |
| "learning_rate": 3.4418604651162795e-06, |
| "loss": 10.560739135742187, |
| "step": 75 |
| }, |
| { |
| "epoch": 0.14918414918414918, |
| "grad_norm": 32.19042205810547, |
| "learning_rate": 3.674418604651163e-06, |
| "loss": 10.408053588867187, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.1585081585081585, |
| "grad_norm": 27.208942413330078, |
| "learning_rate": 3.906976744186047e-06, |
| "loss": 10.108213806152344, |
| "step": 85 |
| }, |
| { |
| "epoch": 0.16783216783216784, |
| "grad_norm": 24.76529312133789, |
| "learning_rate": 4.1395348837209304e-06, |
| "loss": 9.929393005371093, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.17715617715617715, |
| "grad_norm": 23.917495727539062, |
| "learning_rate": 4.372093023255815e-06, |
| "loss": 9.677659606933593, |
| "step": 95 |
| }, |
| { |
| "epoch": 0.1864801864801865, |
| "grad_norm": 22.98892593383789, |
| "learning_rate": 4.604651162790698e-06, |
| "loss": 9.481817626953125, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.1864801864801865, |
| "eval_com_accuracy": 0.176, |
| "eval_rew_accuracy": 0.264, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.1958041958041958, |
| "grad_norm": 25.177200317382812, |
| "learning_rate": 4.837209302325582e-06, |
| "loss": 9.42150650024414, |
| "step": 105 |
| }, |
| { |
| "epoch": 0.20512820512820512, |
| "grad_norm": 18.95603370666504, |
| "learning_rate": 5.069767441860466e-06, |
| "loss": 9.267645263671875, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.21445221445221446, |
| "grad_norm": 22.155452728271484, |
| "learning_rate": 5.302325581395349e-06, |
| "loss": 9.06854248046875, |
| "step": 115 |
| }, |
| { |
| "epoch": 0.22377622377622378, |
| "grad_norm": 20.26909065246582, |
| "learning_rate": 5.534883720930233e-06, |
| "loss": 9.045303344726562, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.2331002331002331, |
| "grad_norm": 26.49457359313965, |
| "learning_rate": 5.7674418604651175e-06, |
| "loss": 8.91327362060547, |
| "step": 125 |
| }, |
| { |
| "epoch": 0.24242424242424243, |
| "grad_norm": 19.886178970336914, |
| "learning_rate": 6e-06, |
| "loss": 8.950289154052735, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.2517482517482518, |
| "grad_norm": 18.22088050842285, |
| "learning_rate": 6.2325581395348845e-06, |
| "loss": 8.710862731933593, |
| "step": 135 |
| }, |
| { |
| "epoch": 0.26107226107226106, |
| "grad_norm": 21.09984016418457, |
| "learning_rate": 6.465116279069767e-06, |
| "loss": 8.762718200683594, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.2703962703962704, |
| "grad_norm": 17.96813201904297, |
| "learning_rate": 6.6976744186046515e-06, |
| "loss": 8.533232116699219, |
| "step": 145 |
| }, |
| { |
| "epoch": 0.27972027972027974, |
| "grad_norm": 20.52909278869629, |
| "learning_rate": 6.930232558139536e-06, |
| "loss": 8.279258728027344, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.27972027972027974, |
| "eval_com_accuracy": 0.152, |
| "eval_rew_accuracy": 0.252, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.289044289044289, |
| "grad_norm": 19.475040435791016, |
| "learning_rate": 7.1627906976744185e-06, |
| "loss": 8.369435882568359, |
| "step": 155 |
| }, |
| { |
| "epoch": 0.29836829836829837, |
| "grad_norm": 16.741973876953125, |
| "learning_rate": 7.395348837209303e-06, |
| "loss": 8.280664825439453, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.3076923076923077, |
| "grad_norm": 19.79235076904297, |
| "learning_rate": 7.627906976744187e-06, |
| "loss": 8.200827026367188, |
| "step": 165 |
| }, |
| { |
| "epoch": 0.317016317016317, |
| "grad_norm": 32.991554260253906, |
| "learning_rate": 7.86046511627907e-06, |
| "loss": 7.973581695556641, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.32634032634032634, |
| "grad_norm": 17.949445724487305, |
| "learning_rate": 8.093023255813955e-06, |
| "loss": 8.141377258300782, |
| "step": 175 |
| }, |
| { |
| "epoch": 0.3356643356643357, |
| "grad_norm": 16.19559097290039, |
| "learning_rate": 8.325581395348837e-06, |
| "loss": 7.837094116210937, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.34498834498834496, |
| "grad_norm": 23.260622024536133, |
| "learning_rate": 8.558139534883722e-06, |
| "loss": 7.763467407226562, |
| "step": 185 |
| }, |
| { |
| "epoch": 0.3543123543123543, |
| "grad_norm": 24.858335494995117, |
| "learning_rate": 8.790697674418606e-06, |
| "loss": 7.886874389648438, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.36363636363636365, |
| "grad_norm": 18.59360122680664, |
| "learning_rate": 9.023255813953489e-06, |
| "loss": 7.7624969482421875, |
| "step": 195 |
| }, |
| { |
| "epoch": 0.372960372960373, |
| "grad_norm": 18.056495666503906, |
| "learning_rate": 9.255813953488373e-06, |
| "loss": 7.4621437072753904, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.372960372960373, |
| "eval_com_accuracy": 0.144, |
| "eval_rew_accuracy": 0.332, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.3822843822843823, |
| "grad_norm": 16.496261596679688, |
| "learning_rate": 9.488372093023258e-06, |
| "loss": 7.526261901855468, |
| "step": 205 |
| }, |
| { |
| "epoch": 0.3916083916083916, |
| "grad_norm": 16.77888298034668, |
| "learning_rate": 9.72093023255814e-06, |
| "loss": 7.624967193603515, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.40093240093240096, |
| "grad_norm": 17.096778869628906, |
| "learning_rate": 9.953488372093025e-06, |
| "loss": 7.337849426269531, |
| "step": 215 |
| }, |
| { |
| "epoch": 0.41025641025641024, |
| "grad_norm": 18.3544979095459, |
| "learning_rate": 9.999894343918674e-06, |
| "loss": 7.354197692871094, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.4195804195804196, |
| "grad_norm": 15.287137985229492, |
| "learning_rate": 9.999465123741172e-06, |
| "loss": 7.085916137695312, |
| "step": 225 |
| }, |
| { |
| "epoch": 0.4289044289044289, |
| "grad_norm": 17.03432273864746, |
| "learning_rate": 9.998705764284132e-06, |
| "loss": 7.059596252441406, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.4382284382284382, |
| "grad_norm": 15.470928192138672, |
| "learning_rate": 9.99761631569179e-06, |
| "loss": 7.118296813964844, |
| "step": 235 |
| }, |
| { |
| "epoch": 0.44755244755244755, |
| "grad_norm": 15.732117652893066, |
| "learning_rate": 9.9961968499058e-06, |
| "loss": 6.981436157226563, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.4568764568764569, |
| "grad_norm": 20.92630958557129, |
| "learning_rate": 9.994447460660473e-06, |
| "loss": 6.96953125, |
| "step": 245 |
| }, |
| { |
| "epoch": 0.4662004662004662, |
| "grad_norm": 14.848095893859863, |
| "learning_rate": 9.992368263476585e-06, |
| "loss": 6.831211090087891, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.4662004662004662, |
| "eval_com_accuracy": 0.132, |
| "eval_rew_accuracy": 0.312, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.4755244755244755, |
| "grad_norm": 18.59755516052246, |
| "learning_rate": 9.989959395653756e-06, |
| "loss": 6.873024749755859, |
| "step": 255 |
| }, |
| { |
| "epoch": 0.48484848484848486, |
| "grad_norm": 15.705364227294922, |
| "learning_rate": 9.98722101626138e-06, |
| "loss": 6.633563232421875, |
| "step": 260 |
| }, |
| { |
| "epoch": 0.49417249417249415, |
| "grad_norm": 39.2294807434082, |
| "learning_rate": 9.984153306128124e-06, |
| "loss": 6.63948974609375, |
| "step": 265 |
| }, |
| { |
| "epoch": 0.5034965034965035, |
| "grad_norm": 13.209358215332031, |
| "learning_rate": 9.980756467829977e-06, |
| "loss": 6.688297271728516, |
| "step": 270 |
| }, |
| { |
| "epoch": 0.5128205128205128, |
| "grad_norm": 14.645393371582031, |
| "learning_rate": 9.977030725676887e-06, |
| "loss": 6.4885505676269535, |
| "step": 275 |
| }, |
| { |
| "epoch": 0.5221445221445221, |
| "grad_norm": 17.008819580078125, |
| "learning_rate": 9.972976325697938e-06, |
| "loss": 6.78935546875, |
| "step": 280 |
| }, |
| { |
| "epoch": 0.5314685314685315, |
| "grad_norm": 16.029518127441406, |
| "learning_rate": 9.96859353562511e-06, |
| "loss": 6.681609344482422, |
| "step": 285 |
| }, |
| { |
| "epoch": 0.5407925407925408, |
| "grad_norm": 13.878348350524902, |
| "learning_rate": 9.963882644875594e-06, |
| "loss": 6.528868103027344, |
| "step": 290 |
| }, |
| { |
| "epoch": 0.5501165501165501, |
| "grad_norm": 13.996464729309082, |
| "learning_rate": 9.958843964532683e-06, |
| "loss": 6.431344604492187, |
| "step": 295 |
| }, |
| { |
| "epoch": 0.5594405594405595, |
| "grad_norm": 15.72003173828125, |
| "learning_rate": 9.953477827325229e-06, |
| "loss": 6.5927978515625, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.5594405594405595, |
| "eval_com_accuracy": 0.148, |
| "eval_rew_accuracy": 0.368, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.5687645687645687, |
| "grad_norm": 16.416725158691406, |
| "learning_rate": 9.947784587605678e-06, |
| "loss": 6.411885833740234, |
| "step": 305 |
| }, |
| { |
| "epoch": 0.578088578088578, |
| "grad_norm": 14.880790710449219, |
| "learning_rate": 9.941764621326655e-06, |
| "loss": 6.5298927307128904, |
| "step": 310 |
| }, |
| { |
| "epoch": 0.5874125874125874, |
| "grad_norm": 17.828296661376953, |
| "learning_rate": 9.935418326016153e-06, |
| "loss": 6.045618057250977, |
| "step": 315 |
| }, |
| { |
| "epoch": 0.5967365967365967, |
| "grad_norm": 13.30245590209961, |
| "learning_rate": 9.928746120751275e-06, |
| "loss": 6.1650840759277346, |
| "step": 320 |
| }, |
| { |
| "epoch": 0.6060606060606061, |
| "grad_norm": 13.527432441711426, |
| "learning_rate": 9.921748446130564e-06, |
| "loss": 6.349945831298828, |
| "step": 325 |
| }, |
| { |
| "epoch": 0.6153846153846154, |
| "grad_norm": 14.491817474365234, |
| "learning_rate": 9.9144257642449e-06, |
| "loss": 6.232943725585938, |
| "step": 330 |
| }, |
| { |
| "epoch": 0.6247086247086248, |
| "grad_norm": 15.7533597946167, |
| "learning_rate": 9.906778558647e-06, |
| "loss": 6.564260864257813, |
| "step": 335 |
| }, |
| { |
| "epoch": 0.634032634032634, |
| "grad_norm": 16.538602828979492, |
| "learning_rate": 9.898807334319471e-06, |
| "loss": 6.122506332397461, |
| "step": 340 |
| }, |
| { |
| "epoch": 0.6433566433566433, |
| "grad_norm": 16.459688186645508, |
| "learning_rate": 9.890512617641474e-06, |
| "loss": 6.332431030273438, |
| "step": 345 |
| }, |
| { |
| "epoch": 0.6526806526806527, |
| "grad_norm": 14.879058837890625, |
| "learning_rate": 9.881894956353963e-06, |
| "loss": 6.118003082275391, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.6526806526806527, |
| "eval_com_accuracy": 0.164, |
| "eval_rew_accuracy": 0.364, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.662004662004662, |
| "grad_norm": 16.02516746520996, |
| "learning_rate": 9.87295491952351e-06, |
| "loss": 6.241584014892578, |
| "step": 355 |
| }, |
| { |
| "epoch": 0.6713286713286714, |
| "grad_norm": 14.9837646484375, |
| "learning_rate": 9.863693097504733e-06, |
| "loss": 6.083467864990235, |
| "step": 360 |
| }, |
| { |
| "epoch": 0.6806526806526807, |
| "grad_norm": 14.138517379760742, |
| "learning_rate": 9.854110101901308e-06, |
| "loss": 6.143560791015625, |
| "step": 365 |
| }, |
| { |
| "epoch": 0.6899766899766899, |
| "grad_norm": 12.696243286132812, |
| "learning_rate": 9.844206565525585e-06, |
| "loss": 6.061418914794922, |
| "step": 370 |
| }, |
| { |
| "epoch": 0.6993006993006993, |
| "grad_norm": 15.22846508026123, |
| "learning_rate": 9.833983142356792e-06, |
| "loss": 6.193332672119141, |
| "step": 375 |
| }, |
| { |
| "epoch": 0.7086247086247086, |
| "grad_norm": 15.284857749938965, |
| "learning_rate": 9.823440507497863e-06, |
| "loss": 6.0868377685546875, |
| "step": 380 |
| }, |
| { |
| "epoch": 0.717948717948718, |
| "grad_norm": 14.117541313171387, |
| "learning_rate": 9.812579357130848e-06, |
| "loss": 6.289948272705078, |
| "step": 385 |
| }, |
| { |
| "epoch": 0.7272727272727273, |
| "grad_norm": 12.648367881774902, |
| "learning_rate": 9.801400408470943e-06, |
| "loss": 6.194795227050781, |
| "step": 390 |
| }, |
| { |
| "epoch": 0.7365967365967366, |
| "grad_norm": 12.394021034240723, |
| "learning_rate": 9.789904399719124e-06, |
| "loss": 6.031880950927734, |
| "step": 395 |
| }, |
| { |
| "epoch": 0.745920745920746, |
| "grad_norm": 15.566425323486328, |
| "learning_rate": 9.778092090013416e-06, |
| "loss": 6.0966644287109375, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.745920745920746, |
| "eval_com_accuracy": 0.184, |
| "eval_rew_accuracy": 0.364, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.7552447552447552, |
| "grad_norm": 14.885455131530762, |
| "learning_rate": 9.76596425937874e-06, |
| "loss": 6.476510620117187, |
| "step": 405 |
| }, |
| { |
| "epoch": 0.7645687645687645, |
| "grad_norm": 16.627117156982422, |
| "learning_rate": 9.753521708675426e-06, |
| "loss": 5.9605262756347654, |
| "step": 410 |
| }, |
| { |
| "epoch": 0.7738927738927739, |
| "grad_norm": 14.517923355102539, |
| "learning_rate": 9.740765259546312e-06, |
| "loss": 6.1104694366455075, |
| "step": 415 |
| }, |
| { |
| "epoch": 0.7832167832167832, |
| "grad_norm": 12.007637977600098, |
| "learning_rate": 9.727695754362498e-06, |
| "loss": 5.964785766601563, |
| "step": 420 |
| }, |
| { |
| "epoch": 0.7925407925407926, |
| "grad_norm": 16.17314338684082, |
| "learning_rate": 9.714314056167711e-06, |
| "loss": 5.976337432861328, |
| "step": 425 |
| }, |
| { |
| "epoch": 0.8018648018648019, |
| "grad_norm": 13.246196746826172, |
| "learning_rate": 9.700621048621322e-06, |
| "loss": 5.936331939697266, |
| "step": 430 |
| }, |
| { |
| "epoch": 0.8111888111888111, |
| "grad_norm": 15.030324935913086, |
| "learning_rate": 9.686617635939988e-06, |
| "loss": 6.209032821655273, |
| "step": 435 |
| }, |
| { |
| "epoch": 0.8205128205128205, |
| "grad_norm": 13.853086471557617, |
| "learning_rate": 9.672304742837945e-06, |
| "loss": 5.985638427734375, |
| "step": 440 |
| }, |
| { |
| "epoch": 0.8298368298368298, |
| "grad_norm": 13.534188270568848, |
| "learning_rate": 9.657683314465948e-06, |
| "loss": 5.7971649169921875, |
| "step": 445 |
| }, |
| { |
| "epoch": 0.8391608391608392, |
| "grad_norm": 26.1879825592041, |
| "learning_rate": 9.642754316348846e-06, |
| "loss": 5.949985504150391, |
| "step": 450 |
| }, |
| { |
| "epoch": 0.8391608391608392, |
| "eval_com_accuracy": 0.176, |
| "eval_rew_accuracy": 0.34, |
| "step": 450 |
| }, |
| { |
| "epoch": 0.8484848484848485, |
| "grad_norm": 16.848834991455078, |
| "learning_rate": 9.627518734321837e-06, |
| "loss": 6.1193115234375, |
| "step": 455 |
| }, |
| { |
| "epoch": 0.8578088578088578, |
| "grad_norm": 30.765796661376953, |
| "learning_rate": 9.61197757446536e-06, |
| "loss": 5.74780158996582, |
| "step": 460 |
| }, |
| { |
| "epoch": 0.8671328671328671, |
| "grad_norm": 13.27775764465332, |
| "learning_rate": 9.596131863038664e-06, |
| "loss": 5.994970703125, |
| "step": 465 |
| }, |
| { |
| "epoch": 0.8764568764568764, |
| "grad_norm": 14.923792839050293, |
| "learning_rate": 9.579982646412039e-06, |
| "loss": 5.810473251342773, |
| "step": 470 |
| }, |
| { |
| "epoch": 0.8857808857808858, |
| "grad_norm": 14.351722717285156, |
| "learning_rate": 9.563530990997707e-06, |
| "loss": 6.09692268371582, |
| "step": 475 |
| }, |
| { |
| "epoch": 0.8951048951048951, |
| "grad_norm": 11.83583927154541, |
| "learning_rate": 9.546777983179421e-06, |
| "loss": 5.867227935791016, |
| "step": 480 |
| }, |
| { |
| "epoch": 0.9044289044289044, |
| "grad_norm": 19.415884017944336, |
| "learning_rate": 9.529724729240712e-06, |
| "loss": 5.834615707397461, |
| "step": 485 |
| }, |
| { |
| "epoch": 0.9137529137529138, |
| "grad_norm": 12.147261619567871, |
| "learning_rate": 9.512372355291838e-06, |
| "loss": 6.039286041259766, |
| "step": 490 |
| }, |
| { |
| "epoch": 0.9230769230769231, |
| "grad_norm": 12.706372261047363, |
| "learning_rate": 9.494722007195427e-06, |
| "loss": 6.06304702758789, |
| "step": 495 |
| }, |
| { |
| "epoch": 0.9324009324009324, |
| "grad_norm": 12.642041206359863, |
| "learning_rate": 9.476774850490803e-06, |
| "loss": 5.9729866027832035, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.9324009324009324, |
| "eval_com_accuracy": 0.204, |
| "eval_rew_accuracy": 0.328, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.9417249417249417, |
| "grad_norm": 12.66169548034668, |
| "learning_rate": 9.458532070317021e-06, |
| "loss": 5.8281097412109375, |
| "step": 505 |
| }, |
| { |
| "epoch": 0.951048951048951, |
| "grad_norm": 13.191854476928711, |
| "learning_rate": 9.439994871334614e-06, |
| "loss": 6.183546447753907, |
| "step": 510 |
| }, |
| { |
| "epoch": 0.9603729603729604, |
| "grad_norm": 12.52676010131836, |
| "learning_rate": 9.421164477646031e-06, |
| "loss": 5.785601043701172, |
| "step": 515 |
| }, |
| { |
| "epoch": 0.9696969696969697, |
| "grad_norm": 13.225418090820312, |
| "learning_rate": 9.402042132714817e-06, |
| "loss": 5.784825134277344, |
| "step": 520 |
| }, |
| { |
| "epoch": 0.9790209790209791, |
| "grad_norm": 12.035611152648926, |
| "learning_rate": 9.382629099283486e-06, |
| "loss": 5.9122272491455075, |
| "step": 525 |
| }, |
| { |
| "epoch": 0.9883449883449883, |
| "grad_norm": 12.166351318359375, |
| "learning_rate": 9.362926659290149e-06, |
| "loss": 5.790053939819336, |
| "step": 530 |
| }, |
| { |
| "epoch": 0.9976689976689976, |
| "grad_norm": 13.22154426574707, |
| "learning_rate": 9.342936113783855e-06, |
| "loss": 5.745006561279297, |
| "step": 535 |
| } |
| ], |
| "logging_steps": 5, |
| "max_steps": 2148, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 4, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 0.0, |
| "train_batch_size": 8, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
