{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.9955555555555557,
  "eval_steps": 500,
  "global_step": 1011,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "grad_norm": 72.2578351791039,
      "learning_rate": 1.96078431372549e-06,
      "loss": 16.2543,
      "step": 1
    },
    {
      "epoch": 0.01,
      "grad_norm": 69.14793459091129,
      "learning_rate": 9.803921568627451e-06,
      "loss": 16.1546,
      "step": 5
    },
    {
      "epoch": 0.03,
      "grad_norm": 61.19418424850778,
      "learning_rate": 1.9607843137254903e-05,
      "loss": 15.7851,
      "step": 10
    },
    {
      "epoch": 0.04,
      "grad_norm": 50.95192336423662,
      "learning_rate": 2.9411764705882354e-05,
      "loss": 13.304,
      "step": 15
    },
    {
      "epoch": 0.06,
      "grad_norm": 44.16161556810121,
      "learning_rate": 3.9215686274509805e-05,
      "loss": 10.1286,
      "step": 20
    },
    {
      "epoch": 0.07,
      "grad_norm": 22.013360927764538,
      "learning_rate": 4.901960784313725e-05,
      "loss": 6.5136,
      "step": 25
    },
    {
      "epoch": 0.09,
      "grad_norm": 13.13769527146034,
      "learning_rate": 5.882352941176471e-05,
      "loss": 4.6002,
      "step": 30
    },
    {
      "epoch": 0.1,
      "grad_norm": 9.380927030189634,
      "learning_rate": 6.862745098039216e-05,
      "loss": 3.4775,
      "step": 35
    },
    {
      "epoch": 0.12,
      "grad_norm": 8.047209493969948,
      "learning_rate": 7.843137254901961e-05,
      "loss": 2.5528,
      "step": 40
    },
    {
      "epoch": 0.13,
      "grad_norm": 4.835270040905147,
      "learning_rate": 8.823529411764706e-05,
      "loss": 1.9181,
      "step": 45
    },
    {
      "epoch": 0.15,
      "grad_norm": 4.461126160581573,
      "learning_rate": 9.80392156862745e-05,
      "loss": 1.7189,
      "step": 50
    },
    {
      "epoch": 0.16,
      "grad_norm": 2.4713562833452647,
      "learning_rate": 0.00010784313725490196,
      "loss": 1.4816,
      "step": 55
    },
    {
      "epoch": 0.18,
      "grad_norm": 3.006847832744875,
      "learning_rate": 0.00011764705882352942,
      "loss": 1.3855,
      "step": 60
    },
    {
      "epoch": 0.19,
      "grad_norm": 1.6003996046085147,
      "learning_rate": 0.00012745098039215687,
      "loss": 1.2728,
      "step": 65
    },
    {
      "epoch": 0.21,
      "grad_norm": 1.6221503977277323,
      "learning_rate": 0.0001372549019607843,
      "loss": 1.2497,
      "step": 70
    },
    {
      "epoch": 0.22,
      "grad_norm": 1.584552829167114,
      "learning_rate": 0.00014705882352941178,
      "loss": 1.2432,
      "step": 75
    },
    {
      "epoch": 0.24,
      "grad_norm": 1.5050158166503569,
      "learning_rate": 0.00015686274509803922,
      "loss": 1.1723,
      "step": 80
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.7107243802135035,
      "learning_rate": 0.0001666666666666667,
      "loss": 1.083,
      "step": 85
    },
    {
      "epoch": 0.27,
      "grad_norm": 1.7381735421551534,
      "learning_rate": 0.00017647058823529413,
      "loss": 1.0726,
      "step": 90
    },
    {
      "epoch": 0.28,
      "grad_norm": 1.4981023359065837,
      "learning_rate": 0.00018627450980392157,
      "loss": 1.1057,
      "step": 95
    },
    {
      "epoch": 0.3,
      "grad_norm": 1.7877995266746756,
      "learning_rate": 0.000196078431372549,
      "loss": 1.0783,
      "step": 100
    },
    {
      "epoch": 0.31,
      "grad_norm": 1.5140220512101432,
      "learning_rate": 0.00019999462497359466,
      "loss": 1.1228,
      "step": 105
    },
    {
      "epoch": 0.33,
      "grad_norm": 1.329498665050575,
      "learning_rate": 0.00019996177968249334,
      "loss": 1.0441,
      "step": 110
    },
    {
      "epoch": 0.34,
      "grad_norm": 1.7238891644878542,
      "learning_rate": 0.00019989908502204292,
      "loss": 1.1001,
      "step": 115
    },
    {
      "epoch": 0.36,
      "grad_norm": 1.8266342354047433,
      "learning_rate": 0.00019980655971335945,
      "loss": 1.0601,
      "step": 120
    },
    {
      "epoch": 0.37,
      "grad_norm": 1.6200107092414249,
      "learning_rate": 0.0001996842313852238,
      "loss": 1.0045,
      "step": 125
    },
    {
      "epoch": 0.39,
      "grad_norm": 1.5172675508165814,
      "learning_rate": 0.00019953213656583168,
      "loss": 1.006,
      "step": 130
    },
    {
      "epoch": 0.4,
      "grad_norm": 1.478313589538863,
      "learning_rate": 0.0001993503206718859,
      "loss": 0.9678,
      "step": 135
    },
    {
      "epoch": 0.41,
      "grad_norm": 1.8797845296046498,
      "learning_rate": 0.0001991388379950346,
      "loss": 1.0181,
      "step": 140
    },
    {
      "epoch": 0.43,
      "grad_norm": 1.8037331639091116,
      "learning_rate": 0.00019889775168565943,
      "loss": 1.0096,
      "step": 145
    },
    {
      "epoch": 0.44,
      "grad_norm": 1.4294890809484986,
      "learning_rate": 0.0001986271337340182,
      "loss": 0.9842,
      "step": 150
    },
    {
      "epoch": 0.46,
      "grad_norm": 1.4637597540944496,
      "learning_rate": 0.0001983270649487481,
      "loss": 0.9992,
      "step": 155
    },
    {
      "epoch": 0.47,
      "grad_norm": 1.519557023555869,
      "learning_rate": 0.0001979976349327357,
      "loss": 1.036,
      "step": 160
    },
    {
      "epoch": 0.49,
      "grad_norm": 1.6521810842892943,
      "learning_rate": 0.00019763894205636072,
      "loss": 0.9961,
      "step": 165
    },
    {
      "epoch": 0.5,
      "grad_norm": 1.7888048328412252,
      "learning_rate": 0.0001972510934281218,
      "loss": 0.9796,
      "step": 170
    },
    {
      "epoch": 0.52,
      "grad_norm": 1.4704141321403639,
      "learning_rate": 0.00019683420486265327,
      "loss": 0.9766,
      "step": 175
    },
    {
      "epoch": 0.53,
      "grad_norm": 1.1929169794016272,
      "learning_rate": 0.00019638840084614182,
      "loss": 1.0146,
      "step": 180
    },
    {
      "epoch": 0.55,
      "grad_norm": 1.1785071553887134,
      "learning_rate": 0.00019591381449915397,
      "loss": 1.0056,
      "step": 185
    },
    {
      "epoch": 0.56,
      "grad_norm": 1.2250942438326087,
      "learning_rate": 0.00019541058753688538,
      "loss": 0.9809,
      "step": 190
    },
    {
      "epoch": 0.58,
      "grad_norm": 1.4932596567461078,
      "learning_rate": 0.00019487887022684336,
      "loss": 0.9827,
      "step": 195
    },
    {
      "epoch": 0.59,
      "grad_norm": 1.2885727190352654,
      "learning_rate": 0.00019431882134397598,
      "loss": 1.0027,
      "step": 200
    },
    {
      "epoch": 0.61,
      "grad_norm": 2.57327759834574,
      "learning_rate": 0.00019373060812326052,
      "loss": 0.9351,
      "step": 205
    },
    {
      "epoch": 0.62,
      "grad_norm": 1.3232509112686246,
      "learning_rate": 0.00019311440620976597,
      "loss": 0.9811,
      "step": 210
    },
    {
      "epoch": 0.64,
      "grad_norm": 1.3160771870305465,
      "learning_rate": 0.0001924703996062038,
      "loss": 0.9925,
      "step": 215
    },
    {
      "epoch": 0.65,
      "grad_norm": 1.2477864155169314,
      "learning_rate": 0.00019179878061798347,
      "loss": 0.9212,
      "step": 220
    },
    {
      "epoch": 0.67,
      "grad_norm": 1.093292440232411,
      "learning_rate": 0.0001910997497957885,
      "loss": 0.9851,
      "step": 225
    },
    {
      "epoch": 0.68,
      "grad_norm": 1.362427572762746,
      "learning_rate": 0.0001903735158756905,
      "loss": 0.9893,
      "step": 230
    },
    {
      "epoch": 0.7,
      "grad_norm": 1.346631495980947,
      "learning_rate": 0.00018962029571681886,
      "loss": 0.945,
      "step": 235
    },
    {
      "epoch": 0.71,
      "grad_norm": 1.090827112192976,
      "learning_rate": 0.0001888403142366049,
      "loss": 1.0224,
      "step": 240
    },
    {
      "epoch": 0.73,
      "grad_norm": 1.6700272825279434,
      "learning_rate": 0.00018803380434362,
      "loss": 0.9305,
      "step": 245
    },
    {
      "epoch": 0.74,
      "grad_norm": 1.0096253684038474,
      "learning_rate": 0.00018720100686802694,
      "loss": 1.0128,
      "step": 250
    },
    {
      "epoch": 0.76,
      "grad_norm": 1.7873981051539485,
      "learning_rate": 0.00018634217048966637,
      "loss": 0.953,
      "step": 255
    },
    {
      "epoch": 0.77,
      "grad_norm": 1.3129933811195174,
      "learning_rate": 0.000185457551663799,
      "loss": 0.9363,
      "step": 260
    },
    {
      "epoch": 0.79,
      "grad_norm": 1.2927375510712122,
      "learning_rate": 0.00018454741454452603,
      "loss": 0.9862,
      "step": 265
    },
    {
      "epoch": 0.8,
      "grad_norm": 1.1686920395458062,
      "learning_rate": 0.00018361203090591071,
      "loss": 0.967,
      "step": 270
    },
    {
      "epoch": 0.81,
      "grad_norm": 1.1863754899949552,
      "learning_rate": 0.00018265168006082437,
      "loss": 0.9056,
      "step": 275
    },
    {
      "epoch": 0.83,
      "grad_norm": 1.1941161746530229,
      "learning_rate": 0.0001816666487775416,
      "loss": 0.9791,
      "step": 280
    },
    {
      "epoch": 0.84,
      "grad_norm": 1.1930577483488491,
      "learning_rate": 0.00018065723119410884,
      "loss": 0.9537,
      "step": 285
    },
    {
      "epoch": 0.86,
      "grad_norm": 1.4589361109870367,
      "learning_rate": 0.00017962372873051252,
      "loss": 0.9954,
      "step": 290
    },
    {
      "epoch": 0.87,
      "grad_norm": 1.3290947156035955,
      "learning_rate": 0.00017856644999867264,
      "loss": 0.922,
      "step": 295
    },
    {
      "epoch": 0.89,
      "grad_norm": 1.2441967467176946,
      "learning_rate": 0.000177485710710289,
      "loss": 0.9639,
      "step": 300
    },
    {
      "epoch": 0.9,
      "grad_norm": 1.1382757452000494,
      "learning_rate": 0.00017638183358256696,
      "loss": 0.9811,
      "step": 305
    },
    {
      "epoch": 0.92,
      "grad_norm": 1.147116712031671,
      "learning_rate": 0.00017525514824185185,
      "loss": 0.9457,
      "step": 310
    },
    {
      "epoch": 0.93,
      "grad_norm": 1.0220320338471351,
      "learning_rate": 0.0001741059911251997,
      "loss": 0.9233,
      "step": 315
    },
    {
      "epoch": 0.95,
      "grad_norm": 1.1472331976712749,
      "learning_rate": 0.00017293470537991463,
      "loss": 0.9756,
      "step": 320
    },
    {
      "epoch": 0.96,
      "grad_norm": 1.2410203245777724,
      "learning_rate": 0.0001717416407610824,
      "loss": 0.943,
      "step": 325
    },
    {
      "epoch": 0.98,
      "grad_norm": 1.3981773401938695,
      "learning_rate": 0.00017052715352713075,
      "loss": 0.9477,
      "step": 330
    },
    {
      "epoch": 0.99,
      "grad_norm": 1.2121344807882255,
      "learning_rate": 0.0001692916063334479,
      "loss": 0.9328,
      "step": 335
    },
    {
      "epoch": 1.0,
      "eval_loss": 1.1519585847854614,
      "eval_runtime": 157.1947,
      "eval_samples_per_second": 14.695,
      "eval_steps_per_second": 0.464,
      "step": 337
    },
    {
      "epoch": 1.01,
      "grad_norm": 1.0908262897233953,
      "learning_rate": 0.00016803536812409075,
      "loss": 0.9661,
      "step": 340
    },
    {
      "epoch": 1.02,
      "grad_norm": 1.7756060681085633,
      "learning_rate": 0.00016675881402161536,
      "loss": 0.889,
      "step": 345
    },
    {
      "epoch": 1.04,
      "grad_norm": 1.1136805577794664,
      "learning_rate": 0.0001654623252150624,
      "loss": 0.9091,
      "step": 350
    },
    {
      "epoch": 1.05,
      "grad_norm": 1.3395157303175598,
      "learning_rate": 0.00016414628884613107,
      "loss": 0.9285,
      "step": 355
    },
    {
      "epoch": 1.07,
      "grad_norm": 1.1278036304088261,
      "learning_rate": 0.0001628110978935756,
      "loss": 0.8679,
      "step": 360
    },
    {
      "epoch": 1.08,
      "grad_norm": 1.1866907922967995,
      "learning_rate": 0.0001614571510558588,
      "loss": 0.9031,
      "step": 365
    },
    {
      "epoch": 1.1,
      "grad_norm": 1.1180348411654952,
      "learning_rate": 0.00016008485263209742,
      "loss": 0.9585,
      "step": 370
    },
    {
      "epoch": 1.11,
      "grad_norm": 1.0617311418169892,
      "learning_rate": 0.0001586946124013354,
      "loss": 0.9212,
      "step": 375
    },
    {
      "epoch": 1.13,
      "grad_norm": 0.9259620753456954,
      "learning_rate": 0.00015728684550018064,
      "loss": 0.8746,
      "step": 380
    },
    {
      "epoch": 1.14,
      "grad_norm": 1.1221480241476138,
      "learning_rate": 0.00015586197229884184,
      "loss": 0.924,
      "step": 385
    },
    {
      "epoch": 1.16,
      "grad_norm": 0.9589792763467693,
      "learning_rate": 0.00015442041827560274,
      "loss": 0.8658,
      "step": 390
    },
    {
      "epoch": 1.17,
      "grad_norm": 1.4212740627448033,
      "learning_rate": 0.00015296261388977108,
      "loss": 0.8592,
      "step": 395
    },
    {
      "epoch": 1.19,
      "grad_norm": 1.0480944734011124,
      "learning_rate": 0.00015148899445313981,
      "loss": 0.9497,
      "step": 400
    },
    {
      "epoch": 1.2,
      "grad_norm": 1.0225393084718664,
      "learning_rate": 0.00015000000000000001,
      "loss": 0.9505,
      "step": 405
    },
    {
      "epoch": 1.21,
      "grad_norm": 0.9776923989427987,
      "learning_rate": 0.00014849607515574276,
      "loss": 0.9189,
      "step": 410
    },
    {
      "epoch": 1.23,
      "grad_norm": 0.9981587820990419,
      "learning_rate": 0.00014697766900409074,
      "loss": 0.9761,
      "step": 415
    },
    {
      "epoch": 1.24,
      "grad_norm": 1.1240954604826308,
      "learning_rate": 0.00014544523495299842,
      "loss": 0.9174,
      "step": 420
    },
    {
      "epoch": 1.26,
      "grad_norm": 1.4387424633384827,
      "learning_rate": 0.00014389923059926062,
      "loss": 0.905,
      "step": 425
    },
    {
      "epoch": 1.27,
      "grad_norm": 1.3190393196712298,
      "learning_rate": 0.00014234011759187083,
      "loss": 0.9378,
      "step": 430
    },
    {
      "epoch": 1.29,
      "grad_norm": 0.9858027932238713,
      "learning_rate": 0.00014076836149416887,
      "loss": 0.9164,
      "step": 435
    },
    {
      "epoch": 1.3,
      "grad_norm": 0.8937095468783902,
      "learning_rate": 0.00013918443164482046,
      "loss": 0.9296,
      "step": 440
    },
    {
      "epoch": 1.32,
      "grad_norm": 1.08163207719712,
      "learning_rate": 0.0001375888010176686,
      "loss": 0.914,
      "step": 445
    },
    {
      "epoch": 1.33,
      "grad_norm": 1.0364463188268707,
      "learning_rate": 0.0001359819460805001,
      "loss": 0.8906,
      "step": 450
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.9266587461121177,
      "learning_rate": 0.00013436434665276865,
      "loss": 0.9395,
      "step": 455
    },
    {
      "epoch": 1.36,
      "grad_norm": 1.0746148135081544,
      "learning_rate": 0.0001327364857623168,
      "loss": 0.9365,
      "step": 460
    },
    {
      "epoch": 1.38,
      "grad_norm": 1.077761213751867,
      "learning_rate": 0.00013109884950114007,
      "loss": 0.9157,
      "step": 465
    },
    {
      "epoch": 1.39,
      "grad_norm": 1.4149324327770756,
      "learning_rate": 0.00012945192688023624,
      "loss": 0.8864,
      "step": 470
    },
    {
      "epoch": 1.41,
      "grad_norm": 1.0160083472695705,
      "learning_rate": 0.00012779620968358273,
      "loss": 0.8757,
      "step": 475
    },
    {
      "epoch": 1.42,
      "grad_norm": 1.1590619362809342,
      "learning_rate": 0.00012613219232128608,
      "loss": 0.8985,
      "step": 480
    },
    {
      "epoch": 1.44,
      "grad_norm": 1.1120230320518538,
      "learning_rate": 0.00012446037168194714,
      "loss": 0.9227,
      "step": 485
    },
    {
      "epoch": 1.45,
      "grad_norm": 1.1834009436730764,
      "learning_rate": 0.0001227812469842864,
      "loss": 0.9204,
      "step": 490
    },
    {
      "epoch": 1.47,
      "grad_norm": 1.0355132778907057,
      "learning_rate": 0.00012109531962807332,
      "loss": 0.889,
      "step": 495
    },
    {
      "epoch": 1.48,
      "grad_norm": 1.0214392183969367,
      "learning_rate": 0.00011940309304440433,
      "loss": 0.8882,
      "step": 500
    },
    {
      "epoch": 1.5,
      "grad_norm": 1.0496957066021488,
      "learning_rate": 0.00011770507254537453,
      "loss": 0.9334,
      "step": 505
    },
    {
      "epoch": 1.51,
      "grad_norm": 1.0238376160697826,
      "learning_rate": 0.00011600176517318741,
      "loss": 0.9181,
      "step": 510
    },
    {
      "epoch": 1.53,
      "grad_norm": 1.2008352860962,
      "learning_rate": 0.00011429367954874819,
      "loss": 0.9335,
      "step": 515
    },
    {
      "epoch": 1.54,
      "grad_norm": 1.037759449688711,
      "learning_rate": 0.00011258132571978555,
      "loss": 0.9394,
      "step": 520
    },
    {
      "epoch": 1.56,
      "grad_norm": 0.98060881335434,
      "learning_rate": 0.00011086521500854745,
      "loss": 0.9237,
      "step": 525
    },
    {
      "epoch": 1.57,
      "grad_norm": 1.137978604803647,
      "learning_rate": 0.00010914585985911632,
      "loss": 0.873,
      "step": 530
    },
    {
      "epoch": 1.59,
      "grad_norm": 1.0209942184078924,
      "learning_rate": 0.00010742377368438914,
      "loss": 0.989,
      "step": 535
    },
    {
      "epoch": 1.6,
      "grad_norm": 1.0005848887814885,
      "learning_rate": 0.00010569947071276847,
      "loss": 0.9054,
      "step": 540
    },
    {
      "epoch": 1.61,
      "grad_norm": 1.2509234265685367,
      "learning_rate": 0.00010397346583460971,
      "loss": 0.9068,
      "step": 545
    },
    {
      "epoch": 1.63,
      "grad_norm": 1.118860273669544,
      "learning_rate": 0.0001022462744484709,
      "loss": 0.9185,
      "step": 550
    },
    {
      "epoch": 1.64,
      "grad_norm": 0.9619304215105235,
      "learning_rate": 0.00010051841230721065,
      "loss": 0.9152,
      "step": 555
    },
    {
      "epoch": 1.66,
      "grad_norm": 0.9312722485058544,
      "learning_rate": 9.879039536398024e-05,
      "loss": 0.9451,
      "step": 560
    },
    {
      "epoch": 1.67,
      "grad_norm": 0.9758156277152181,
      "learning_rate": 9.70627396181561e-05,
      "loss": 0.9016,
      "step": 565
    },
    {
      "epoch": 1.69,
      "grad_norm": 0.8860842880872017,
      "learning_rate": 9.533596096125825e-05,
      "loss": 0.9069,
      "step": 570
    },
    {
      "epoch": 1.7,
      "grad_norm": 1.0727044167911228,
      "learning_rate": 9.361057502290113e-05,
      "loss": 0.8748,
      "step": 575
    },
    {
      "epoch": 1.72,
      "grad_norm": 0.8634186584908737,
      "learning_rate": 9.188709701682247e-05,
      "loss": 0.893,
      "step": 580
    },
    {
      "epoch": 1.73,
      "grad_norm": 1.1477573965693937,
      "learning_rate": 9.016604158703654e-05,
      "loss": 0.8893,
      "step": 585
    },
    {
      "epoch": 1.75,
      "grad_norm": 0.9383547560329538,
      "learning_rate": 8.844792265415738e-05,
      "loss": 0.8651,
      "step": 590
    },
    {
      "epoch": 1.76,
      "grad_norm": 1.1121008461123547,
      "learning_rate": 8.673325326193806e-05,
      "loss": 0.9379,
      "step": 595
    },
    {
      "epoch": 1.78,
      "grad_norm": 1.0975479479362362,
      "learning_rate": 8.502254542407186e-05,
      "loss": 0.8481,
      "step": 600
    },
    {
      "epoch": 1.79,
      "grad_norm": 1.113767179456745,
      "learning_rate": 8.33163099713009e-05,
      "loss": 0.8327,
      "step": 605
    },
    {
      "epoch": 1.81,
      "grad_norm": 0.89921971591069,
      "learning_rate": 8.161505639887817e-05,
      "loss": 0.9102,
      "step": 610
    },
    {
      "epoch": 1.82,
      "grad_norm": 1.0388536628309168,
      "learning_rate": 7.991929271442817e-05,
      "loss": 0.859,
      "step": 615
    },
    {
      "epoch": 1.84,
      "grad_norm": 1.0521026013607777,
      "learning_rate": 7.822952528625191e-05,
      "loss": 0.8731,
      "step": 620
    },
    {
      "epoch": 1.85,
      "grad_norm": 0.9421918430486114,
      "learning_rate": 7.654625869212146e-05,
      "loss": 0.9012,
      "step": 625
    },
    {
      "epoch": 1.87,
      "grad_norm": 0.9484722686972686,
      "learning_rate": 7.48699955686089e-05,
      "loss": 0.9432,
      "step": 630
    },
    {
      "epoch": 1.88,
      "grad_norm": 0.9768965209508316,
      "learning_rate": 7.320123646099519e-05,
      "loss": 0.8999,
      "step": 635
    },
    {
      "epoch": 1.9,
      "grad_norm": 0.9453535215942106,
      "learning_rate": 7.154047967380354e-05,
      "loss": 0.9724,
      "step": 640
    },
    {
      "epoch": 1.91,
      "grad_norm": 1.0424636870577482,
      "learning_rate": 6.988822112200156e-05,
      "loss": 0.8789,
      "step": 645
    },
    {
      "epoch": 1.93,
      "grad_norm": 0.9224956346414365,
      "learning_rate": 6.82449541829174e-05,
      "loss": 0.8569,
      "step": 650
    },
    {
      "epoch": 1.94,
      "grad_norm": 1.1516517831276118,
      "learning_rate": 6.661116954891328e-05,
      "loss": 0.9306,
      "step": 655
    },
    {
      "epoch": 1.96,
      "grad_norm": 0.9883581608392334,
      "learning_rate": 6.498735508086093e-05,
      "loss": 0.871,
      "step": 660
    },
    {
      "epoch": 1.97,
      "grad_norm": 0.9438307796156034,
      "learning_rate": 6.337399566246257e-05,
      "loss": 0.9302,
      "step": 665
    },
    {
      "epoch": 1.99,
      "grad_norm": 0.8930439870325309,
      "learning_rate": 6.177157305546078e-05,
      "loss": 0.9477,
      "step": 670
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.9118609372127487,
      "learning_rate": 6.018056575578075e-05,
      "loss": 0.8771,
      "step": 675
    },
    {
      "epoch": 2.0,
      "eval_loss": 1.1150912046432495,
      "eval_runtime": 156.2319,
      "eval_samples_per_second": 14.786,
      "eval_steps_per_second": 0.467,
      "step": 675
    },
    {
      "epoch": 2.01,
      "grad_norm": 0.828432991647057,
      "learning_rate": 5.860144885064751e-05,
      "loss": 0.8378,
      "step": 680
    },
    {
      "epoch": 2.03,
      "grad_norm": 1.0448906737327117,
      "learning_rate": 5.7034693876721376e-05,
      "loss": 0.8421,
      "step": 685
    },
    {
      "epoch": 2.04,
      "grad_norm": 1.0620773533713173,
      "learning_rate": 5.54807686792933e-05,
      "loss": 0.8794,
      "step": 690
    },
    {
      "epoch": 2.06,
      "grad_norm": 1.0869539484563016,
      "learning_rate": 5.394013727258254e-05,
      "loss": 0.8679,
      "step": 695
    },
    {
      "epoch": 2.07,
      "grad_norm": 0.9798911637537467,
      "learning_rate": 5.2413259701178505e-05,
      "loss": 0.8898,
      "step": 700
    },
    {
      "epoch": 2.09,
      "grad_norm": 0.9047442392594395,
      "learning_rate": 5.090059190266779e-05,
      "loss": 0.8724,
      "step": 705
    },
    {
      "epoch": 2.1,
      "grad_norm": 1.0781903357123614,
      "learning_rate": 4.940258557148765e-05,
      "loss": 0.8573,
      "step": 710
    },
    {
      "epoch": 2.12,
      "grad_norm": 1.0342896185783423,
      "learning_rate": 4.791968802404648e-05,
      "loss": 0.8281,
      "step": 715
    },
    {
      "epoch": 2.13,
      "grad_norm": 0.9191153541590816,
      "learning_rate": 4.645234206515171e-05,
      "loss": 0.9009,
      "step": 720
    },
    {
      "epoch": 2.15,
      "grad_norm": 1.1845629044153207,
      "learning_rate": 4.5000985855784746e-05,
      "loss": 0.857,
      "step": 725
    },
    {
      "epoch": 2.16,
      "grad_norm": 1.0434089629049308,
      "learning_rate": 4.3566052782262735e-05,
      "loss": 0.858,
      "step": 730
    },
    {
      "epoch": 2.18,
      "grad_norm": 0.990812188911313,
      "learning_rate": 4.2147971326825966e-05,
      "loss": 0.8516,
      "step": 735
    },
    {
      "epoch": 2.19,
      "grad_norm": 0.905736044179964,
      "learning_rate": 4.074716493968975e-05,
      "loss": 0.8476,
      "step": 740
    },
    {
      "epoch": 2.21,
      "grad_norm": 1.016168128211076,
      "learning_rate": 3.936405191259891e-05,
      "loss": 0.8388,
      "step": 745
    },
    {
      "epoch": 2.22,
      "grad_norm": 0.9657889846503783,
      "learning_rate": 3.79990452539225e-05,
      "loss": 0.8796,
      "step": 750
    },
    {
      "epoch": 2.24,
      "grad_norm": 0.8488273704750955,
      "learning_rate": 3.665255256532638e-05,
      "loss": 0.8226,
      "step": 755
    },
    {
      "epoch": 2.25,
      "grad_norm": 0.8515631446002137,
      "learning_rate": 3.53249759200601e-05,
      "loss": 0.8848,
      "step": 760
    },
    {
      "epoch": 2.27,
      "grad_norm": 0.9625777666321612,
      "learning_rate": 3.401671174289469e-05,
      "loss": 0.8581,
      "step": 765
    },
    {
      "epoch": 2.28,
      "grad_norm": 1.1027956646622936,
      "learning_rate": 3.2728150691747115e-05,
      "loss": 0.8205,
      "step": 770
    },
    {
      "epoch": 2.3,
      "grad_norm": 0.959658890342703,
      "learning_rate": 3.145967754102691e-05,
      "loss": 0.8332,
      "step": 775
    },
    {
      "epoch": 2.31,
      "grad_norm": 1.014525985812299,
      "learning_rate": 3.021167106673928e-05,
      "loss": 0.8485,
      "step": 780
    },
    {
      "epoch": 2.33,
      "grad_norm": 1.0119872246269643,
      "learning_rate": 2.898450393337977e-05,
      "loss": 0.856,
      "step": 785
    },
    {
      "epoch": 2.34,
      "grad_norm": 1.321701548482505,
      "learning_rate": 2.7778542582653744e-05,
      "loss": 0.8304,
      "step": 790
    },
    {
      "epoch": 2.36,
      "grad_norm": 1.0899857889914895,
      "learning_rate": 2.659414712405398e-05,
      "loss": 0.8442,
      "step": 795
    },
    {
      "epoch": 2.37,
      "grad_norm": 1.0994834827590023,
      "learning_rate": 2.543167122732918e-05,
      "loss": 0.8645,
      "step": 800
    },
    {
      "epoch": 2.39,
      "grad_norm": 0.8030087173097793,
      "learning_rate": 2.429146201687538e-05,
      "loss": 0.8567,
      "step": 805
    },
    {
      "epoch": 2.4,
      "grad_norm": 1.01607316930185,
      "learning_rate": 2.3173859968081944e-05,
      "loss": 0.9123,
      "step": 810
    },
    {
      "epoch": 2.41,
      "grad_norm": 0.9294802520942274,
      "learning_rate": 2.2079198805662914e-05,
      "loss": 0.8374,
      "step": 815
    },
    {
      "epoch": 2.43,
      "grad_norm": 0.9523609743858644,
      "learning_rate": 2.1007805404004242e-05,
      "loss": 0.8568,
      "step": 820
    },
    {
      "epoch": 2.44,
      "grad_norm": 0.8957405654003763,
      "learning_rate": 1.995999968955641e-05,
      "loss": 0.8873,
      "step": 825
    },
    {
      "epoch": 2.46,
      "grad_norm": 1.0700819784398359,
      "learning_rate": 1.8936094545302095e-05,
      "loss": 0.8564,
      "step": 830
    },
    {
      "epoch": 2.47,
      "grad_norm": 1.0564758507529628,
      "learning_rate": 1.7936395717326704e-05,
      "loss": 0.8221,
      "step": 835
    },
    {
      "epoch": 2.49,
      "grad_norm": 1.015095925986807,
      "learning_rate": 1.696120172352025e-05,
      "loss": 0.8416,
      "step": 840
    },
    {
      "epoch": 2.5,
      "grad_norm": 1.2346772662125298,
      "learning_rate": 1.601080376443763e-05,
      "loss": 0.8742,
      "step": 845
    },
    {
      "epoch": 2.52,
      "grad_norm": 0.9234353972167334,
      "learning_rate": 1.5085485636343755e-05,
      "loss": 0.9049,
      "step": 850
    },
    {
      "epoch": 2.53,
      "grad_norm": 1.0570174679975244,
      "learning_rate": 1.4185523646469822e-05,
      "loss": 0.8367,
      "step": 855
    },
    {
      "epoch": 2.55,
      "grad_norm": 0.8795669021553979,
      "learning_rate": 1.3311186530505838e-05,
      "loss": 0.8661,
      "step": 860
    },
    {
      "epoch": 2.56,
      "grad_norm": 0.9638299921267924,
      "learning_rate": 1.2462735372353996e-05,
      "loss": 0.8363,
      "step": 865
    },
    {
      "epoch": 2.58,
      "grad_norm": 1.2030489918077059,
      "learning_rate": 1.1640423526166988e-05,
      "loss": 0.8583,
      "step": 870
    },
    {
      "epoch": 2.59,
      "grad_norm": 1.0716487215374508,
      "learning_rate": 1.0844496540694515e-05,
      "loss": 0.8588,
      "step": 875
    },
    {
      "epoch": 2.61,
      "grad_norm": 1.0669537623700072,
      "learning_rate": 1.007519208596045e-05,
      "loss": 0.9143,
      "step": 880
    },
    {
      "epoch": 2.62,
      "grad_norm": 0.9415770157718919,
      "learning_rate": 9.332739882292752e-06,
      "loss": 0.906,
      "step": 885
    },
    {
      "epoch": 2.64,
      "grad_norm": 0.8466881399276756,
      "learning_rate": 8.617361631727138e-06,
      "loss": 0.8823,
      "step": 890
    },
    {
      "epoch": 2.65,
      "grad_norm": 1.173003775227567,
      "learning_rate": 7.929270951805178e-06,
      "loss": 0.8866,
      "step": 895
    },
    {
      "epoch": 2.67,
      "grad_norm": 0.9695275155903833,
      "learning_rate": 7.2686733117863784e-06,
      "loss": 0.8717,
      "step": 900
    },
    {
      "epoch": 2.68,
      "grad_norm": 0.9278647388922164,
      "learning_rate": 6.635765971293484e-06,
      "loss": 0.8999,
      "step": 905
    },
    {
      "epoch": 2.7,
      "grad_norm": 0.9905531986540063,
      "learning_rate": 6.030737921409169e-06,
      "loss": 0.8772,
      "step": 910
    },
    {
      "epoch": 2.71,
      "grad_norm": 1.068092774140916,
      "learning_rate": 5.453769828241872e-06,
      "loss": 0.8974,
      "step": 915
    },
    {
      "epoch": 2.73,
      "grad_norm": 1.1961165825237712,
      "learning_rate": 4.905033978977491e-06,
      "loss": 0.86,
      "step": 920
    },
    {
      "epoch": 2.74,
      "grad_norm": 1.0848450931313236,
      "learning_rate": 4.384694230432984e-06,
      "loss": 0.8363,
      "step": 925
    },
    {
      "epoch": 2.76,
      "grad_norm": 1.0929622662706844,
      "learning_rate": 3.892905960127546e-06,
      "loss": 0.8727,
      "step": 930
    },
    {
      "epoch": 2.77,
      "grad_norm": 1.0564158056197959,
      "learning_rate": 3.4298160198856568e-06,
      "loss": 0.8169,
      "step": 935
    },
    {
      "epoch": 2.79,
      "grad_norm": 1.0136059442310454,
      "learning_rate": 2.995562691985898e-06,
      "loss": 0.8574,
      "step": 940
    },
    {
      "epoch": 2.8,
      "grad_norm": 1.092221087850955,
      "learning_rate": 2.590275647868867e-06,
      "loss": 0.8386,
      "step": 945
    },
    {
      "epoch": 2.81,
      "grad_norm": 1.0072864796741217,
      "learning_rate": 2.2140759094162467e-06,
      "loss": 0.8685,
      "step": 950
    },
    {
      "epoch": 2.83,
      "grad_norm": 0.8433396823974946,
      "learning_rate": 1.8670758128126909e-06,
      "loss": 0.849,
      "step": 955
    },
    {
      "epoch": 2.84,
      "grad_norm": 0.9684221245764032,
      "learning_rate": 1.5493789750014031e-06,
      "loss": 0.8525,
      "step": 960
    },
    {
      "epoch": 2.86,
      "grad_norm": 0.9141081630302206,
      "learning_rate": 1.261080262743297e-06,
      "loss": 0.8315,
      "step": 965
    },
    {
      "epoch": 2.87,
      "grad_norm": 0.9014966731205527,
      "learning_rate": 1.0022657642890231e-06,
      "loss": 0.8316,
      "step": 970
    },
    {
      "epoch": 2.89,
      "grad_norm": 0.9464472386186885,
      "learning_rate": 7.730127636723539e-07,
      "loss": 0.8605,
      "step": 975
    },
    {
      "epoch": 2.9,
      "grad_norm": 0.896721952301425,
      "learning_rate": 5.733897176325665e-07,
      "loss": 0.8137,
      "step": 980
    },
    {
      "epoch": 2.92,
      "grad_norm": 1.0368172773007387,
      "learning_rate": 4.034562351727389e-07,
      "loss": 0.8664,
      "step": 985
    },
    {
      "epoch": 2.93,
      "grad_norm": 0.9777429571054969,
      "learning_rate": 2.6326305976001055e-07,
      "loss": 0.8557,
      "step": 990
    },
    {
      "epoch": 2.95,
      "grad_norm": 0.96487435678213,
      "learning_rate": 1.5285205417319149e-07,
      "loss": 0.8691,
      "step": 995
    },
    {
      "epoch": 2.96,
      "grad_norm": 0.8599085672586297,
      "learning_rate": 7.225618800222877e-08,
      "loss": 0.8806,
      "step": 1000
    },
    {
      "epoch": 2.98,
      "grad_norm": 1.0670556057589013,
      "learning_rate": 2.1499527803214846e-08,
      "loss": 0.8669,
      "step": 1005
    },
    {
      "epoch": 2.99,
      "grad_norm": 1.0796304331427407,
      "learning_rate": 5.972299119250125e-10,
      "loss": 0.8486,
      "step": 1010
    },
    {
      "epoch": 3.0,
      "eval_loss": 1.1171541213989258,
      "eval_runtime": 156.2002,
      "eval_samples_per_second": 14.789,
      "eval_steps_per_second": 0.467,
      "step": 1011
    },
    {
      "epoch": 3.0,
      "step": 1011,
      "total_flos": 2597843400392704.0,
      "train_loss": 1.2570129054000658,
      "train_runtime": 12020.0319,
      "train_samples_per_second": 5.39,
      "train_steps_per_second": 0.084
    }
  ],
  "logging_steps": 5,
  "max_steps": 1011,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "total_flos": 2597843400392704.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}