{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0,
  "eval_steps": 500,
  "global_step": 5914,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0033818058843422386,
      "grad_norm": 1.5605809688568115,
      "learning_rate": 9.991538331358945e-05,
      "loss": 1.5931,
      "step": 10
    },
    {
      "epoch": 0.006763611768684477,
      "grad_norm": 0.2704204320907593,
      "learning_rate": 9.974614994076833e-05,
      "loss": 0.6298,
      "step": 20
    },
    {
      "epoch": 0.010145417653026716,
      "grad_norm": 0.20275332033634186,
      "learning_rate": 9.957691656794721e-05,
      "loss": 0.6147,
      "step": 30
    },
    {
      "epoch": 0.013527223537368955,
      "grad_norm": 0.1859731376171112,
      "learning_rate": 9.940768319512609e-05,
      "loss": 0.5784,
      "step": 40
    },
    {
      "epoch": 0.016909029421711193,
      "grad_norm": 0.20401914417743683,
      "learning_rate": 9.923844982230497e-05,
      "loss": 0.5696,
      "step": 50
    },
    {
      "epoch": 0.020290835306053433,
      "grad_norm": 0.21193653345108032,
      "learning_rate": 9.906921644948385e-05,
      "loss": 0.565,
      "step": 60
    },
    {
      "epoch": 0.023672641190395673,
      "grad_norm": 0.20135045051574707,
      "learning_rate": 9.889998307666272e-05,
      "loss": 0.5251,
      "step": 70
    },
    {
      "epoch": 0.02705444707473791,
      "grad_norm": 0.20686441659927368,
      "learning_rate": 9.873074970384161e-05,
      "loss": 0.5295,
      "step": 80
    },
    {
      "epoch": 0.03043625295908015,
      "grad_norm": 0.20559485256671906,
      "learning_rate": 9.856151633102048e-05,
      "loss": 0.5272,
      "step": 90
    },
    {
      "epoch": 0.033818058843422386,
      "grad_norm": 0.23076851665973663,
      "learning_rate": 9.839228295819937e-05,
      "loss": 0.5284,
      "step": 100
    },
    {
      "epoch": 0.037199864727764625,
      "grad_norm": 0.21439723670482635,
      "learning_rate": 9.822304958537824e-05,
      "loss": 0.5088,
      "step": 110
    },
    {
      "epoch": 0.040581670612106865,
      "grad_norm": 0.23800402879714966,
      "learning_rate": 9.805381621255713e-05,
      "loss": 0.5123,
      "step": 120
    },
    {
      "epoch": 0.043963476496449105,
      "grad_norm": 0.2768985629081726,
      "learning_rate": 9.7884582839736e-05,
      "loss": 0.4884,
      "step": 130
    },
    {
      "epoch": 0.047345282380791345,
      "grad_norm": 0.26437142491340637,
      "learning_rate": 9.771534946691489e-05,
      "loss": 0.4748,
      "step": 140
    },
    {
      "epoch": 0.05072708826513358,
      "grad_norm": 0.2534578740596771,
      "learning_rate": 9.754611609409376e-05,
      "loss": 0.4766,
      "step": 150
    },
    {
      "epoch": 0.05410889414947582,
      "grad_norm": 0.2550220787525177,
      "learning_rate": 9.737688272127265e-05,
      "loss": 0.4613,
      "step": 160
    },
    {
      "epoch": 0.05749070003381806,
      "grad_norm": 0.2937515377998352,
      "learning_rate": 9.720764934845152e-05,
      "loss": 0.4631,
      "step": 170
    },
    {
      "epoch": 0.0608725059181603,
      "grad_norm": 0.31693148612976074,
      "learning_rate": 9.703841597563041e-05,
      "loss": 0.4454,
      "step": 180
    },
    {
      "epoch": 0.06425431180250253,
      "grad_norm": 0.3209972679615021,
      "learning_rate": 9.686918260280928e-05,
      "loss": 0.4389,
      "step": 190
    },
    {
      "epoch": 0.06763611768684477,
      "grad_norm": 0.26218757033348083,
      "learning_rate": 9.669994922998817e-05,
      "loss": 0.4393,
      "step": 200
    },
    {
      "epoch": 0.07101792357118701,
      "grad_norm": 0.23638683557510376,
      "learning_rate": 9.653071585716704e-05,
      "loss": 0.393,
      "step": 210
    },
    {
      "epoch": 0.07439972945552925,
      "grad_norm": 0.2807103395462036,
      "learning_rate": 9.636148248434593e-05,
      "loss": 0.4012,
      "step": 220
    },
    {
      "epoch": 0.07778153533987149,
      "grad_norm": 0.29169249534606934,
      "learning_rate": 9.61922491115248e-05,
      "loss": 0.4146,
      "step": 230
    },
    {
      "epoch": 0.08116334122421373,
      "grad_norm": 0.26288753747940063,
      "learning_rate": 9.602301573870369e-05,
      "loss": 0.4185,
      "step": 240
    },
    {
      "epoch": 0.08454514710855597,
      "grad_norm": 0.2438131868839264,
      "learning_rate": 9.585378236588256e-05,
      "loss": 0.3786,
      "step": 250
    },
    {
      "epoch": 0.08792695299289821,
      "grad_norm": 0.2577499449253082,
      "learning_rate": 9.568454899306143e-05,
      "loss": 0.4075,
      "step": 260
    },
    {
      "epoch": 0.09130875887724045,
      "grad_norm": 0.2468397617340088,
      "learning_rate": 9.551531562024032e-05,
      "loss": 0.382,
      "step": 270
    },
    {
      "epoch": 0.09469056476158269,
      "grad_norm": 0.27210426330566406,
      "learning_rate": 9.53460822474192e-05,
      "loss": 0.3853,
      "step": 280
    },
    {
      "epoch": 0.09807237064592493,
      "grad_norm": 0.2579999566078186,
      "learning_rate": 9.517684887459808e-05,
      "loss": 0.3816,
      "step": 290
    },
    {
      "epoch": 0.10145417653026716,
      "grad_norm": 0.24862495064735413,
      "learning_rate": 9.500761550177695e-05,
      "loss": 0.3481,
      "step": 300
    },
    {
      "epoch": 0.1048359824146094,
      "grad_norm": 0.289042204618454,
      "learning_rate": 9.483838212895584e-05,
      "loss": 0.3961,
      "step": 310
    },
    {
      "epoch": 0.10821778829895164,
      "grad_norm": 0.2919853627681732,
      "learning_rate": 9.466914875613471e-05,
      "loss": 0.3332,
      "step": 320
    },
    {
      "epoch": 0.11159959418329388,
      "grad_norm": 0.25639480352401733,
      "learning_rate": 9.44999153833136e-05,
      "loss": 0.3644,
      "step": 330
    },
    {
      "epoch": 0.11498140006763612,
      "grad_norm": 0.26778921484947205,
      "learning_rate": 9.433068201049247e-05,
      "loss": 0.3695,
      "step": 340
    },
    {
      "epoch": 0.11836320595197836,
      "grad_norm": 0.24935244023799896,
      "learning_rate": 9.416144863767136e-05,
      "loss": 0.3433,
      "step": 350
    },
    {
      "epoch": 0.1217450118363206,
      "grad_norm": 0.2668505609035492,
      "learning_rate": 9.399221526485023e-05,
      "loss": 0.3689,
      "step": 360
    },
    {
      "epoch": 0.12512681772066284,
      "grad_norm": 0.2972855865955353,
      "learning_rate": 9.382298189202912e-05,
      "loss": 0.3153,
      "step": 370
    },
    {
      "epoch": 0.12850862360500506,
      "grad_norm": 0.2680118978023529,
      "learning_rate": 9.3653748519208e-05,
      "loss": 0.3332,
      "step": 380
    },
    {
      "epoch": 0.13189042948934732,
      "grad_norm": 0.28004977107048035,
      "learning_rate": 9.348451514638688e-05,
      "loss": 0.342,
      "step": 390
    },
    {
      "epoch": 0.13527223537368954,
      "grad_norm": 0.2519853115081787,
      "learning_rate": 9.331528177356575e-05,
      "loss": 0.3194,
      "step": 400
    },
    {
      "epoch": 0.1386540412580318,
      "grad_norm": 0.2775399386882782,
      "learning_rate": 9.314604840074464e-05,
      "loss": 0.3113,
      "step": 410
    },
    {
      "epoch": 0.14203584714237402,
      "grad_norm": 0.27479955554008484,
      "learning_rate": 9.297681502792351e-05,
      "loss": 0.3148,
      "step": 420
    },
    {
      "epoch": 0.14541765302671628,
      "grad_norm": 0.3637036681175232,
      "learning_rate": 9.28075816551024e-05,
      "loss": 0.3065,
      "step": 430
    },
    {
      "epoch": 0.1487994589110585,
      "grad_norm": 0.2898998260498047,
      "learning_rate": 9.263834828228127e-05,
      "loss": 0.2799,
      "step": 440
    },
    {
      "epoch": 0.15218126479540076,
      "grad_norm": 0.295907199382782,
      "learning_rate": 9.246911490946015e-05,
      "loss": 0.2984,
      "step": 450
    },
    {
      "epoch": 0.15556307067974298,
      "grad_norm": 0.3111323118209839,
      "learning_rate": 9.229988153663903e-05,
      "loss": 0.2854,
      "step": 460
    },
    {
      "epoch": 0.1589448765640852,
      "grad_norm": 0.2768048346042633,
      "learning_rate": 9.21306481638179e-05,
      "loss": 0.2895,
      "step": 470
    },
    {
      "epoch": 0.16232668244842746,
      "grad_norm": 0.34783947467803955,
      "learning_rate": 9.19614147909968e-05,
      "loss": 0.2686,
      "step": 480
    },
    {
      "epoch": 0.1657084883327697,
      "grad_norm": 0.3182440400123596,
      "learning_rate": 9.179218141817567e-05,
      "loss": 0.2603,
      "step": 490
    },
    {
      "epoch": 0.16909029421711194,
      "grad_norm": 0.3452625572681427,
      "learning_rate": 9.162294804535455e-05,
      "loss": 0.296,
      "step": 500
    },
    {
      "epoch": 0.17247210010145417,
      "grad_norm": 0.329106867313385,
      "learning_rate": 9.145371467253343e-05,
      "loss": 0.2665,
      "step": 510
    },
    {
      "epoch": 0.17585390598579642,
      "grad_norm": 0.3779897689819336,
      "learning_rate": 9.128448129971231e-05,
      "loss": 0.2545,
      "step": 520
    },
    {
      "epoch": 0.17923571187013865,
      "grad_norm": 0.33585628867149353,
      "learning_rate": 9.111524792689119e-05,
      "loss": 0.2683,
      "step": 530
    },
    {
      "epoch": 0.1826175177544809,
      "grad_norm": 0.27421489357948303,
      "learning_rate": 9.094601455407007e-05,
      "loss": 0.2585,
      "step": 540
    },
    {
      "epoch": 0.18599932363882313,
      "grad_norm": 0.37735405564308167,
      "learning_rate": 9.077678118124895e-05,
      "loss": 0.2676,
      "step": 550
    },
    {
      "epoch": 0.18938112952316538,
      "grad_norm": 0.30768734216690063,
      "learning_rate": 9.060754780842783e-05,
      "loss": 0.2543,
      "step": 560
    },
    {
      "epoch": 0.1927629354075076,
      "grad_norm": 0.3587827682495117,
      "learning_rate": 9.04383144356067e-05,
      "loss": 0.2249,
      "step": 570
    },
    {
      "epoch": 0.19614474129184986,
      "grad_norm": 0.3213902711868286,
      "learning_rate": 9.026908106278559e-05,
      "loss": 0.2443,
      "step": 580
    },
    {
      "epoch": 0.1995265471761921,
      "grad_norm": 0.4303820729255676,
      "learning_rate": 9.009984768996447e-05,
      "loss": 0.2256,
      "step": 590
    },
    {
      "epoch": 0.2029083530605343,
      "grad_norm": 0.2948843538761139,
      "learning_rate": 8.993061431714335e-05,
      "loss": 0.2295,
      "step": 600
    },
    {
      "epoch": 0.20629015894487657,
      "grad_norm": 0.3263229727745056,
      "learning_rate": 8.976138094432223e-05,
      "loss": 0.239,
      "step": 610
    },
    {
      "epoch": 0.2096719648292188,
      "grad_norm": 0.3471197783946991,
      "learning_rate": 8.95921475715011e-05,
      "loss": 0.2227,
      "step": 620
    },
    {
      "epoch": 0.21305377071356105,
      "grad_norm": 0.3152846693992615,
      "learning_rate": 8.942291419867999e-05,
      "loss": 0.2359,
      "step": 630
    },
    {
      "epoch": 0.21643557659790327,
      "grad_norm": 0.4284135401248932,
      "learning_rate": 8.925368082585886e-05,
      "loss": 0.2453,
      "step": 640
    },
    {
      "epoch": 0.21981738248224553,
      "grad_norm": 0.37850794196128845,
      "learning_rate": 8.908444745303775e-05,
      "loss": 0.2245,
      "step": 650
    },
    {
      "epoch": 0.22319918836658775,
      "grad_norm": 0.38802939653396606,
      "learning_rate": 8.891521408021662e-05,
      "loss": 0.2227,
      "step": 660
    },
    {
      "epoch": 0.22658099425093,
      "grad_norm": 0.32752034068107605,
      "learning_rate": 8.87459807073955e-05,
      "loss": 0.2008,
      "step": 670
    },
    {
      "epoch": 0.22996280013527223,
      "grad_norm": 0.3419714570045471,
      "learning_rate": 8.857674733457438e-05,
      "loss": 0.2049,
      "step": 680
    },
    {
      "epoch": 0.2333446060196145,
      "grad_norm": 0.3725152313709259,
      "learning_rate": 8.840751396175327e-05,
      "loss": 0.1922,
      "step": 690
    },
    {
      "epoch": 0.2367264119039567,
      "grad_norm": 0.3231143653392792,
      "learning_rate": 8.823828058893214e-05,
      "loss": 0.2094,
      "step": 700
    },
    {
      "epoch": 0.24010821778829894,
      "grad_norm": 0.3786492347717285,
      "learning_rate": 8.806904721611103e-05,
      "loss": 0.1917,
      "step": 710
    },
    {
      "epoch": 0.2434900236726412,
      "grad_norm": 0.3543280363082886,
      "learning_rate": 8.78998138432899e-05,
      "loss": 0.2005,
      "step": 720
    },
    {
      "epoch": 0.24687182955698342,
      "grad_norm": 0.336761474609375,
      "learning_rate": 8.773058047046879e-05,
      "loss": 0.174,
      "step": 730
    },
    {
      "epoch": 0.2502536354413257,
      "grad_norm": 0.3629204332828522,
      "learning_rate": 8.756134709764766e-05,
      "loss": 0.1851,
      "step": 740
    },
    {
      "epoch": 0.2536354413256679,
      "grad_norm": 0.3555070459842682,
      "learning_rate": 8.739211372482655e-05,
      "loss": 0.1832,
      "step": 750
    },
    {
      "epoch": 0.2570172472100101,
      "grad_norm": 0.41113823652267456,
      "learning_rate": 8.722288035200542e-05,
      "loss": 0.1772,
      "step": 760
    },
    {
      "epoch": 0.2603990530943524,
      "grad_norm": 0.32061767578125,
      "learning_rate": 8.70536469791843e-05,
      "loss": 0.1707,
      "step": 770
    },
    {
      "epoch": 0.26378085897869463,
      "grad_norm": 0.35991913080215454,
      "learning_rate": 8.688441360636318e-05,
      "loss": 0.1487,
      "step": 780
    },
    {
      "epoch": 0.26716266486303686,
      "grad_norm": 0.3859257102012634,
      "learning_rate": 8.671518023354207e-05,
      "loss": 0.1688,
      "step": 790
    },
    {
      "epoch": 0.2705444707473791,
      "grad_norm": 0.31017613410949707,
      "learning_rate": 8.654594686072094e-05,
      "loss": 0.1624,
      "step": 800
    },
    {
      "epoch": 0.27392627663172137,
      "grad_norm": 0.35503777861595154,
      "learning_rate": 8.637671348789981e-05,
      "loss": 0.1642,
      "step": 810
    },
    {
      "epoch": 0.2773080825160636,
      "grad_norm": 0.3433260917663574,
      "learning_rate": 8.62074801150787e-05,
      "loss": 0.1844,
      "step": 820
    },
    {
      "epoch": 0.2806898884004058,
      "grad_norm": 0.3865962624549866,
      "learning_rate": 8.603824674225757e-05,
      "loss": 0.1506,
      "step": 830
    },
    {
      "epoch": 0.28407169428474804,
      "grad_norm": 0.3324780762195587,
      "learning_rate": 8.586901336943646e-05,
      "loss": 0.1632,
      "step": 840
    },
    {
      "epoch": 0.28745350016909027,
      "grad_norm": 0.4064556062221527,
      "learning_rate": 8.569977999661533e-05,
      "loss": 0.1519,
      "step": 850
    },
    {
      "epoch": 0.29083530605343255,
      "grad_norm": 0.3946615159511566,
      "learning_rate": 8.553054662379422e-05,
      "loss": 0.157,
      "step": 860
    },
    {
      "epoch": 0.2942171119377748,
      "grad_norm": 0.3700334429740906,
      "learning_rate": 8.536131325097309e-05,
      "loss": 0.141,
      "step": 870
    },
    {
      "epoch": 0.297598917822117,
      "grad_norm": 0.420950710773468,
      "learning_rate": 8.519207987815198e-05,
      "loss": 0.1488,
      "step": 880
    },
    {
      "epoch": 0.30098072370645923,
      "grad_norm": 0.39633265137672424,
      "learning_rate": 8.502284650533085e-05,
      "loss": 0.1545,
      "step": 890
    },
    {
      "epoch": 0.3043625295908015,
      "grad_norm": 0.38442814350128174,
      "learning_rate": 8.485361313250974e-05,
      "loss": 0.141,
      "step": 900
    },
    {
      "epoch": 0.30774433547514374,
      "grad_norm": 0.41429537534713745,
      "learning_rate": 8.468437975968861e-05,
      "loss": 0.144,
      "step": 910
    },
    {
      "epoch": 0.31112614135948596,
      "grad_norm": 0.3561369478702545,
      "learning_rate": 8.45151463868675e-05,
      "loss": 0.134,
      "step": 920
    },
    {
      "epoch": 0.3145079472438282,
      "grad_norm": 0.3710062801837921,
      "learning_rate": 8.434591301404637e-05,
      "loss": 0.1457,
      "step": 930
    },
    {
      "epoch": 0.3178897531281704,
      "grad_norm": 0.3607446849346161,
      "learning_rate": 8.417667964122526e-05,
      "loss": 0.122,
      "step": 940
    },
    {
      "epoch": 0.3212715590125127,
      "grad_norm": 0.34604352712631226,
      "learning_rate": 8.400744626840413e-05,
      "loss": 0.1414,
      "step": 950
    },
    {
      "epoch": 0.3246533648968549,
      "grad_norm": 0.27822059392929077,
      "learning_rate": 8.383821289558302e-05,
      "loss": 0.1281,
      "step": 960
    },
    {
      "epoch": 0.32803517078119715,
      "grad_norm": 0.38080519437789917,
      "learning_rate": 8.366897952276189e-05,
      "loss": 0.1392,
      "step": 970
    },
    {
      "epoch": 0.3314169766655394,
      "grad_norm": 0.4061764180660248,
      "learning_rate": 8.349974614994078e-05,
      "loss": 0.1179,
      "step": 980
    },
    {
      "epoch": 0.33479878254988166,
      "grad_norm": 0.28504157066345215,
      "learning_rate": 8.333051277711965e-05,
      "loss": 0.1284,
      "step": 990
    },
    {
      "epoch": 0.3381805884342239,
      "grad_norm": 0.3242844343185425,
      "learning_rate": 8.316127940429852e-05,
      "loss": 0.1184,
      "step": 1000
    },
    {
      "epoch": 0.3415623943185661,
      "grad_norm": 0.3317604064941406,
      "learning_rate": 8.299204603147741e-05,
      "loss": 0.13,
      "step": 1010
    },
    {
      "epoch": 0.34494420020290834,
      "grad_norm": 0.39583227038383484,
      "learning_rate": 8.282281265865628e-05,
      "loss": 0.1146,
      "step": 1020
    },
    {
      "epoch": 0.3483260060872506,
      "grad_norm": 0.33169683814048767,
      "learning_rate": 8.265357928583517e-05,
      "loss": 0.1162,
      "step": 1030
    },
    {
      "epoch": 0.35170781197159284,
      "grad_norm": 0.38136041164398193,
      "learning_rate": 8.248434591301404e-05,
      "loss": 0.1268,
      "step": 1040
    },
    {
      "epoch": 0.35508961785593507,
      "grad_norm": 0.28301218152046204,
      "learning_rate": 8.231511254019293e-05,
      "loss": 0.1117,
      "step": 1050
    },
    {
      "epoch": 0.3584714237402773,
      "grad_norm": 0.3950181007385254,
      "learning_rate": 8.21458791673718e-05,
      "loss": 0.1172,
      "step": 1060
    },
    {
      "epoch": 0.3618532296246195,
      "grad_norm": 0.40106379985809326,
      "learning_rate": 8.197664579455069e-05,
      "loss": 0.1134,
      "step": 1070
    },
    {
      "epoch": 0.3652350355089618,
      "grad_norm": 0.3036629259586334,
      "learning_rate": 8.180741242172956e-05,
      "loss": 0.0977,
      "step": 1080
    },
    {
      "epoch": 0.36861684139330403,
      "grad_norm": 0.35834598541259766,
      "learning_rate": 8.163817904890845e-05,
      "loss": 0.1226,
      "step": 1090
    },
    {
      "epoch": 0.37199864727764625,
      "grad_norm": 0.3107944130897522,
      "learning_rate": 8.146894567608732e-05,
      "loss": 0.0932,
      "step": 1100
    },
    {
      "epoch": 0.3753804531619885,
      "grad_norm": 0.2708600163459778,
      "learning_rate": 8.129971230326621e-05,
      "loss": 0.1071,
      "step": 1110
    },
    {
      "epoch": 0.37876225904633076,
      "grad_norm": 0.3161390721797943,
      "learning_rate": 8.113047893044508e-05,
      "loss": 0.0951,
      "step": 1120
    },
    {
      "epoch": 0.382144064930673,
      "grad_norm": 0.2967537045478821,
      "learning_rate": 8.096124555762397e-05,
      "loss": 0.0961,
      "step": 1130
    },
    {
      "epoch": 0.3855258708150152,
      "grad_norm": 0.3615525960922241,
      "learning_rate": 8.079201218480284e-05,
      "loss": 0.0955,
      "step": 1140
    },
    {
      "epoch": 0.38890767669935744,
      "grad_norm": 0.33469459414482117,
      "learning_rate": 8.062277881198173e-05,
      "loss": 0.1028,
      "step": 1150
    },
    {
      "epoch": 0.3922894825836997,
      "grad_norm": 0.44299551844596863,
      "learning_rate": 8.04535454391606e-05,
      "loss": 0.1065,
      "step": 1160
    },
    {
      "epoch": 0.39567128846804195,
      "grad_norm": 0.3324688971042633,
      "learning_rate": 8.028431206633949e-05,
      "loss": 0.1026,
      "step": 1170
    },
    {
      "epoch": 0.3990530943523842,
      "grad_norm": 0.3845025599002838,
      "learning_rate": 8.011507869351836e-05,
      "loss": 0.1166,
      "step": 1180
    },
    {
      "epoch": 0.4024349002367264,
      "grad_norm": 0.36462751030921936,
      "learning_rate": 7.994584532069724e-05,
      "loss": 0.0955,
      "step": 1190
    },
    {
      "epoch": 0.4058167061210686,
      "grad_norm": 0.33478081226348877,
      "learning_rate": 7.977661194787612e-05,
      "loss": 0.0966,
      "step": 1200
    },
    {
      "epoch": 0.4091985120054109,
      "grad_norm": 0.3839697539806366,
      "learning_rate": 7.9607378575055e-05,
      "loss": 0.1011,
      "step": 1210
    },
    {
      "epoch": 0.41258031788975313,
      "grad_norm": 0.33828845620155334,
      "learning_rate": 7.943814520223388e-05,
      "loss": 0.094,
      "step": 1220
    },
    {
      "epoch": 0.41596212377409536,
      "grad_norm": 0.35532712936401367,
      "learning_rate": 7.926891182941276e-05,
      "loss": 0.0771,
      "step": 1230
    },
    {
      "epoch": 0.4193439296584376,
      "grad_norm": 0.32812246680259705,
      "learning_rate": 7.909967845659164e-05,
      "loss": 0.0931,
      "step": 1240
    },
    {
      "epoch": 0.42272573554277987,
      "grad_norm": 0.4312772750854492,
      "learning_rate": 7.893044508377052e-05,
      "loss": 0.091,
      "step": 1250
    },
    {
      "epoch": 0.4261075414271221,
      "grad_norm": 0.38822853565216064,
      "learning_rate": 7.87612117109494e-05,
      "loss": 0.0936,
      "step": 1260
    },
    {
      "epoch": 0.4294893473114643,
      "grad_norm": 0.30626174807548523,
      "learning_rate": 7.859197833812828e-05,
      "loss": 0.0773,
      "step": 1270
    },
    {
      "epoch": 0.43287115319580655,
      "grad_norm": 0.36472904682159424,
      "learning_rate": 7.842274496530716e-05,
      "loss": 0.0776,
      "step": 1280
    },
    {
      "epoch": 0.43625295908014877,
      "grad_norm": 0.312610000371933,
      "learning_rate": 7.825351159248604e-05,
      "loss": 0.0882,
      "step": 1290
    },
    {
      "epoch": 0.43963476496449105,
      "grad_norm": 0.34325119853019714,
      "learning_rate": 7.808427821966492e-05,
      "loss": 0.0836,
      "step": 1300
    },
    {
      "epoch": 0.4430165708488333,
      "grad_norm": 0.39519986510276794,
      "learning_rate": 7.79150448468438e-05,
      "loss": 0.084,
      "step": 1310
    },
    {
      "epoch": 0.4463983767331755,
      "grad_norm": 0.35263752937316895,
      "learning_rate": 7.774581147402268e-05,
      "loss": 0.0713,
      "step": 1320
    },
    {
      "epoch": 0.44978018261751773,
      "grad_norm": 0.36037376523017883,
      "learning_rate": 7.757657810120156e-05,
      "loss": 0.069,
      "step": 1330
    },
    {
      "epoch": 0.45316198850186,
      "grad_norm": 0.38161391019821167,
      "learning_rate": 7.740734472838044e-05,
      "loss": 0.0693,
      "step": 1340
    },
    {
      "epoch": 0.45654379438620224,
      "grad_norm": 0.3896481394767761,
      "learning_rate": 7.723811135555932e-05,
      "loss": 0.0793,
      "step": 1350
    },
    {
      "epoch": 0.45992560027054447,
      "grad_norm": 0.3284771740436554,
      "learning_rate": 7.70688779827382e-05,
      "loss": 0.0874,
      "step": 1360
    },
    {
      "epoch": 0.4633074061548867,
      "grad_norm": 0.3116714358329773,
      "learning_rate": 7.689964460991708e-05,
      "loss": 0.0734,
      "step": 1370
    },
    {
      "epoch": 0.466689212039229,
      "grad_norm": 0.29977601766586304,
      "learning_rate": 7.673041123709595e-05,
      "loss": 0.0696,
      "step": 1380
    },
    {
      "epoch": 0.4700710179235712,
      "grad_norm": 0.30298420786857605,
      "learning_rate": 7.656117786427484e-05,
      "loss": 0.0755,
      "step": 1390
    },
    {
      "epoch": 0.4734528238079134,
      "grad_norm": 0.3856925368309021,
      "learning_rate": 7.639194449145371e-05,
      "loss": 0.0791,
      "step": 1400
    },
    {
      "epoch": 0.47683462969225565,
      "grad_norm": 0.3421929180622101,
      "learning_rate": 7.62227111186326e-05,
      "loss": 0.0674,
      "step": 1410
    },
    {
      "epoch": 0.4802164355765979,
      "grad_norm": 0.3304542899131775,
      "learning_rate": 7.605347774581147e-05,
      "loss": 0.076,
      "step": 1420
    },
    {
      "epoch": 0.48359824146094016,
      "grad_norm": 0.29870203137397766,
      "learning_rate": 7.588424437299036e-05,
      "loss": 0.0703,
      "step": 1430
    },
    {
      "epoch": 0.4869800473452824,
      "grad_norm": 0.2472551167011261,
      "learning_rate": 7.571501100016923e-05,
      "loss": 0.069,
      "step": 1440
    },
    {
      "epoch": 0.4903618532296246,
      "grad_norm": 0.2860611081123352,
      "learning_rate": 7.554577762734812e-05,
      "loss": 0.0667,
      "step": 1450
    },
    {
      "epoch": 0.49374365911396684,
      "grad_norm": 0.3618302345275879,
      "learning_rate": 7.537654425452699e-05,
      "loss": 0.0597,
      "step": 1460
    },
    {
      "epoch": 0.4971254649983091,
      "grad_norm": 0.34842416644096375,
      "learning_rate": 7.520731088170588e-05,
      "loss": 0.0715,
      "step": 1470
    },
    {
      "epoch": 0.5005072708826513,
      "grad_norm": 0.2862137258052826,
      "learning_rate": 7.503807750888475e-05,
      "loss": 0.0778,
      "step": 1480
    },
    {
      "epoch": 0.5038890767669936,
      "grad_norm": 0.3283160626888275,
      "learning_rate": 7.486884413606364e-05,
      "loss": 0.073,
      "step": 1490
    },
    {
      "epoch": 0.5072708826513358,
      "grad_norm": 0.2540145218372345,
      "learning_rate": 7.469961076324251e-05,
      "loss": 0.067,
      "step": 1500
    },
    {
      "epoch": 0.510652688535678,
      "grad_norm": 0.32044336199760437,
      "learning_rate": 7.45303773904214e-05,
      "loss": 0.063,
      "step": 1510
    },
    {
      "epoch": 0.5140344944200202,
      "grad_norm": 0.3898891508579254,
      "learning_rate": 7.436114401760027e-05,
      "loss": 0.0695,
      "step": 1520
    },
    {
      "epoch": 0.5174163003043625,
      "grad_norm": 0.30922603607177734,
      "learning_rate": 7.419191064477916e-05,
      "loss": 0.071,
      "step": 1530
    },
    {
      "epoch": 0.5207981061887048,
      "grad_norm": 0.3513735234737396,
      "learning_rate": 7.402267727195803e-05,
      "loss": 0.0722,
      "step": 1540
    },
    {
      "epoch": 0.524179912073047,
      "grad_norm": 0.2806611955165863,
      "learning_rate": 7.38534438991369e-05,
      "loss": 0.0662,
      "step": 1550
    },
    {
      "epoch": 0.5275617179573893,
      "grad_norm": 0.32641470432281494,
      "learning_rate": 7.368421052631579e-05,
      "loss": 0.0645,
      "step": 1560
    },
    {
      "epoch": 0.5309435238417315,
      "grad_norm": 0.2627265751361847,
      "learning_rate": 7.351497715349466e-05,
      "loss": 0.0594,
      "step": 1570
    },
    {
      "epoch": 0.5343253297260737,
      "grad_norm": 0.2586800158023834,
      "learning_rate": 7.334574378067355e-05,
      "loss": 0.0603,
      "step": 1580
    },
    {
      "epoch": 0.5377071356104159,
      "grad_norm": 0.24359896779060364,
      "learning_rate": 7.317651040785242e-05,
      "loss": 0.0638,
      "step": 1590
    },
    {
      "epoch": 0.5410889414947582,
      "grad_norm": 0.3084506392478943,
      "learning_rate": 7.300727703503131e-05,
      "loss": 0.0658,
      "step": 1600
    },
    {
      "epoch": 0.5444707473791004,
      "grad_norm": 0.29731103777885437,
      "learning_rate": 7.283804366221018e-05,
      "loss": 0.0548,
      "step": 1610
    },
    {
      "epoch": 0.5478525532634427,
      "grad_norm": 0.25320005416870117,
      "learning_rate": 7.266881028938907e-05,
      "loss": 0.0533,
      "step": 1620
    },
    {
      "epoch": 0.551234359147785,
      "grad_norm": 0.274166077375412,
      "learning_rate": 7.249957691656794e-05,
      "loss": 0.048,
      "step": 1630
    },
    {
      "epoch": 0.5546161650321272,
      "grad_norm": 0.29488497972488403,
      "learning_rate": 7.233034354374683e-05,
      "loss": 0.054,
      "step": 1640
    },
    {
      "epoch": 0.5579979709164694,
      "grad_norm": 0.2893562316894531,
      "learning_rate": 7.21611101709257e-05,
      "loss": 0.0589,
      "step": 1650
    },
    {
      "epoch": 0.5613797768008116,
      "grad_norm": 0.29615455865859985,
      "learning_rate": 7.199187679810459e-05,
      "loss": 0.0524,
      "step": 1660
    },
    {
      "epoch": 0.5647615826851539,
      "grad_norm": 0.3037727475166321,
      "learning_rate": 7.182264342528346e-05,
      "loss": 0.0486,
      "step": 1670
    },
    {
      "epoch": 0.5681433885694961,
      "grad_norm": 0.2997746765613556,
      "learning_rate": 7.165341005246235e-05,
      "loss": 0.047,
      "step": 1680
    },
    {
      "epoch": 0.5715251944538383,
      "grad_norm": 0.31440579891204834,
      "learning_rate": 7.148417667964122e-05,
      "loss": 0.0467,
      "step": 1690
    },
    {
      "epoch": 0.5749070003381805,
      "grad_norm": 0.345804363489151,
      "learning_rate": 7.131494330682011e-05,
      "loss": 0.0543,
      "step": 1700
    },
    {
      "epoch": 0.5782888062225229,
      "grad_norm": 0.3310382664203644,
      "learning_rate": 7.114570993399898e-05,
      "loss": 0.0561,
      "step": 1710
    },
    {
      "epoch": 0.5816706121068651,
      "grad_norm": 0.33962053060531616,
      "learning_rate": 7.097647656117787e-05,
      "loss": 0.0468,
      "step": 1720
    },
    {
      "epoch": 0.5850524179912073,
      "grad_norm": 0.303245484828949,
      "learning_rate": 7.080724318835674e-05,
      "loss": 0.0503,
      "step": 1730
    },
    {
      "epoch": 0.5884342238755496,
      "grad_norm": 0.3114894926548004,
      "learning_rate": 7.063800981553562e-05,
      "loss": 0.0482,
      "step": 1740
    },
    {
      "epoch": 0.5918160297598918,
      "grad_norm": 0.23478536307811737,
      "learning_rate": 7.04687764427145e-05,
      "loss": 0.0496,
      "step": 1750
    },
    {
      "epoch": 0.595197835644234,
      "grad_norm": 0.3116193413734436,
      "learning_rate": 7.029954306989338e-05,
      "loss": 0.0498,
      "step": 1760
    },
    {
      "epoch": 0.5985796415285762,
      "grad_norm": 0.30709898471832275,
      "learning_rate": 7.013030969707226e-05,
      "loss": 0.0528,
      "step": 1770
    },
    {
      "epoch": 0.6019614474129185,
      "grad_norm": 0.3018944561481476,
      "learning_rate": 6.996107632425114e-05,
      "loss": 0.0466,
      "step": 1780
    },
    {
      "epoch": 0.6053432532972607,
      "grad_norm": 0.32787784934043884,
      "learning_rate": 6.979184295143002e-05,
      "loss": 0.0513,
      "step": 1790
    },
    {
      "epoch": 0.608725059181603,
      "grad_norm": 0.30039939284324646,
      "learning_rate": 6.962260957860891e-05,
      "loss": 0.0504,
      "step": 1800
    },
    {
      "epoch": 0.6121068650659452,
      "grad_norm": 0.3414636552333832,
      "learning_rate": 6.94533762057878e-05,
      "loss": 0.0499,
      "step": 1810
    },
    {
      "epoch": 0.6154886709502875,
      "grad_norm": 0.20751340687274933,
      "learning_rate": 6.928414283296667e-05,
      "loss": 0.0437,
      "step": 1820
    },
    {
      "epoch": 0.6188704768346297,
      "grad_norm": 0.3419632315635681,
      "learning_rate": 6.911490946014556e-05,
      "loss": 0.049,
      "step": 1830
    },
    {
      "epoch": 0.6222522827189719,
      "grad_norm": 0.27273499965667725,
      "learning_rate": 6.894567608732443e-05,
      "loss": 0.0444,
      "step": 1840
    },
    {
      "epoch": 0.6256340886033142,
      "grad_norm": 0.31872832775115967,
      "learning_rate": 6.87764427145033e-05,
      "loss": 0.0535,
      "step": 1850
    },
    {
      "epoch": 0.6290158944876564,
      "grad_norm": 0.24325081706047058,
      "learning_rate": 6.860720934168219e-05,
      "loss": 0.049,
      "step": 1860
    },
    {
      "epoch": 0.6323977003719986,
      "grad_norm": 0.3723507821559906,
      "learning_rate": 6.843797596886106e-05,
      "loss": 0.0434,
      "step": 1870
    },
    {
      "epoch": 0.6357795062563408,
      "grad_norm": 0.24727103114128113,
      "learning_rate": 6.826874259603995e-05,
      "loss": 0.0452,
      "step": 1880
    },
    {
      "epoch": 0.6391613121406832,
      "grad_norm": 0.3109544515609741,
      "learning_rate": 6.809950922321882e-05,
      "loss": 0.044,
      "step": 1890
    },
    {
      "epoch": 0.6425431180250254,
      "grad_norm": 0.27794697880744934,
      "learning_rate": 6.793027585039771e-05,
      "loss": 0.0403,
      "step": 1900
    },
    {
      "epoch": 0.6459249239093676,
      "grad_norm": 0.3061220347881317,
      "learning_rate": 6.776104247757658e-05,
      "loss": 0.0425,
      "step": 1910
    },
    {
      "epoch": 0.6493067297937098,
      "grad_norm": 0.25127366185188293,
      "learning_rate": 6.759180910475547e-05,
      "loss": 0.0432,
      "step": 1920
    },
    {
      "epoch": 0.6526885356780521,
      "grad_norm": 0.26195022463798523,
      "learning_rate": 6.742257573193434e-05,
      "loss": 0.042,
      "step": 1930
    },
    {
      "epoch": 0.6560703415623943,
      "grad_norm": 0.23891383409500122,
      "learning_rate": 6.725334235911323e-05,
      "loss": 0.0404,
      "step": 1940
    },
    {
      "epoch": 0.6594521474467365,
      "grad_norm": 0.26952946186065674,
      "learning_rate": 6.70841089862921e-05,
      "loss": 0.0414,
      "step": 1950
    },
    {
      "epoch": 0.6628339533310788,
      "grad_norm": 0.2608715295791626,
      "learning_rate": 6.691487561347099e-05,
      "loss": 0.0418,
      "step": 1960
    },
    {
      "epoch": 0.6662157592154211,
      "grad_norm": 0.3117476999759674,
      "learning_rate": 6.674564224064986e-05,
      "loss": 0.0417,
      "step": 1970
    },
    {
      "epoch": 0.6695975650997633,
      "grad_norm": 0.38972756266593933,
      "learning_rate": 6.657640886782875e-05,
      "loss": 0.0432,
      "step": 1980
    },
    {
      "epoch": 0.6729793709841055,
      "grad_norm": 0.23405376076698303,
      "learning_rate": 6.640717549500762e-05,
      "loss": 0.0448,
      "step": 1990
    },
    {
      "epoch": 0.6763611768684478,
      "grad_norm": 0.2317565679550171,
      "learning_rate": 6.623794212218651e-05,
      "loss": 0.0391,
      "step": 2000
    },
    {
      "epoch": 0.67974298275279,
      "grad_norm": 0.2788334786891937,
      "learning_rate": 6.606870874936538e-05,
      "loss": 0.041,
      "step": 2010
    },
    {
      "epoch": 0.6831247886371322,
      "grad_norm": 0.2472919374704361,
      "learning_rate": 6.589947537654427e-05,
      "loss": 0.0413,
      "step": 2020
    },
    {
      "epoch": 0.6865065945214744,
      "grad_norm": 0.31981515884399414,
      "learning_rate": 6.573024200372314e-05,
      "loss": 0.0394,
      "step": 2030
    },
    {
      "epoch": 0.6898884004058167,
      "grad_norm": 0.3317371606826782,
      "learning_rate": 6.556100863090201e-05,
      "loss": 0.0405,
      "step": 2040
    },
    {
      "epoch": 0.6932702062901589,
      "grad_norm": 0.22539494931697845,
      "learning_rate": 6.53917752580809e-05,
      "loss": 0.0364,
      "step": 2050
    },
    {
      "epoch": 0.6966520121745012,
      "grad_norm": 0.2573080360889435,
      "learning_rate": 6.522254188525977e-05,
      "loss": 0.0404,
      "step": 2060
    },
    {
      "epoch": 0.7000338180588435,
      "grad_norm": 0.3429962992668152,
      "learning_rate": 6.505330851243866e-05,
      "loss": 0.0432,
      "step": 2070
    },
    {
      "epoch": 0.7034156239431857,
      "grad_norm": 0.26425597071647644,
      "learning_rate": 6.488407513961753e-05,
      "loss": 0.0383,
      "step": 2080
    },
    {
      "epoch": 0.7067974298275279,
      "grad_norm": 0.28535112738609314,
      "learning_rate": 6.471484176679642e-05,
      "loss": 0.042,
      "step": 2090
    },
    {
      "epoch": 0.7101792357118701,
      "grad_norm": 0.27274271845817566,
      "learning_rate": 6.45456083939753e-05,
      "loss": 0.0358,
      "step": 2100
    },
    {
      "epoch": 0.7135610415962124,
      "grad_norm": 0.24476689100265503,
      "learning_rate": 6.437637502115418e-05,
      "loss": 0.0358,
      "step": 2110
    },
    {
      "epoch": 0.7169428474805546,
      "grad_norm": 0.295635461807251,
      "learning_rate": 6.420714164833305e-05,
      "loss": 0.0364,
      "step": 2120
    },
    {
      "epoch": 0.7203246533648968,
      "grad_norm": 0.3371022343635559,
      "learning_rate": 6.403790827551194e-05,
      "loss": 0.0381,
      "step": 2130
    },
    {
      "epoch": 0.723706459249239,
      "grad_norm": 0.24469931423664093,
      "learning_rate": 6.386867490269081e-05,
      "loss": 0.0384,
      "step": 2140
    },
    {
      "epoch": 0.7270882651335814,
      "grad_norm": 0.20693735778331757,
      "learning_rate": 6.36994415298697e-05,
      "loss": 0.0369,
      "step": 2150
    },
    {
      "epoch": 0.7304700710179236,
      "grad_norm": 0.24583208560943604,
      "learning_rate": 6.353020815704857e-05,
      "loss": 0.0367,
      "step": 2160
    },
    {
      "epoch": 0.7338518769022658,
      "grad_norm": 0.29009056091308594,
      "learning_rate": 6.336097478422746e-05,
      "loss": 0.038,
      "step": 2170
    },
    {
      "epoch": 0.7372336827866081,
      "grad_norm": 0.260128378868103,
      "learning_rate": 6.319174141140633e-05,
      "loss": 0.039,
      "step": 2180
    },
    {
      "epoch": 0.7406154886709503,
      "grad_norm": 0.2804548442363739,
      "learning_rate": 6.302250803858522e-05,
      "loss": 0.0399,
      "step": 2190
    },
    {
      "epoch": 0.7439972945552925,
      "grad_norm": 0.3354940116405487,
      "learning_rate": 6.28532746657641e-05,
      "loss": 0.0363,
      "step": 2200
    },
    {
      "epoch": 0.7473791004396347,
      "grad_norm": 0.23447370529174805,
      "learning_rate": 6.268404129294298e-05,
      "loss": 0.042,
      "step": 2210
    },
    {
      "epoch": 0.750760906323977,
      "grad_norm": 0.26304641366004944,
      "learning_rate": 6.251480792012185e-05,
      "loss": 0.0364,
      "step": 2220
    },
    {
      "epoch": 0.7541427122083192,
      "grad_norm": 0.26174917817115784,
      "learning_rate": 6.234557454730073e-05,
      "loss": 0.0333,
      "step": 2230
    },
    {
      "epoch": 0.7575245180926615,
      "grad_norm": 0.2674359977245331,
      "learning_rate": 6.217634117447961e-05,
      "loss": 0.0387,
      "step": 2240
    },
    {
      "epoch": 0.7609063239770038,
      "grad_norm": 0.21239162981510162,
      "learning_rate": 6.200710780165849e-05,
      "loss": 0.0374,
      "step": 2250
    },
    {
      "epoch": 0.764288129861346,
      "grad_norm": 0.2588539719581604,
      "learning_rate": 6.183787442883737e-05,
      "loss": 0.0354,
      "step": 2260
    },
    {
      "epoch": 0.7676699357456882,
      "grad_norm": 0.253009557723999,
      "learning_rate": 6.166864105601625e-05,
      "loss": 0.0346,
      "step": 2270
    },
    {
      "epoch": 0.7710517416300304,
      "grad_norm": 0.2577788531780243,
      "learning_rate": 6.149940768319513e-05,
      "loss": 0.032,
      "step": 2280
    },
    {
      "epoch": 0.7744335475143727,
      "grad_norm": 0.2186667025089264,
      "learning_rate": 6.133017431037401e-05,
      "loss": 0.0329,
      "step": 2290
    },
    {
      "epoch": 0.7778153533987149,
      "grad_norm": 0.278125524520874,
      "learning_rate": 6.11609409375529e-05,
      "loss": 0.0338,
      "step": 2300
    },
    {
      "epoch": 0.7811971592830571,
      "grad_norm": 0.3377946615219116,
      "learning_rate": 6.0991707564731774e-05,
      "loss": 0.037,
      "step": 2310
    },
    {
      "epoch": 0.7845789651673994,
      "grad_norm": 0.29264363646507263,
      "learning_rate": 6.082247419191065e-05,
      "loss": 0.0361,
      "step": 2320
    },
    {
      "epoch": 0.7879607710517417,
      "grad_norm": 0.23570716381072998,
      "learning_rate": 6.065324081908953e-05,
      "loss": 0.0368,
      "step": 2330
    },
    {
      "epoch": 0.7913425769360839,
      "grad_norm": 0.27007460594177246,
      "learning_rate": 6.048400744626841e-05,
      "loss": 0.0334,
      "step": 2340
    },
    {
      "epoch": 0.7947243828204261,
      "grad_norm": 0.2671580910682678,
      "learning_rate": 6.031477407344729e-05,
      "loss": 0.0342,
      "step": 2350
    },
    {
      "epoch": 0.7981061887047683,
      "grad_norm": 0.2466389387845993,
      "learning_rate": 6.014554070062617e-05,
      "loss": 0.0403,
      "step": 2360
    },
    {
      "epoch": 0.8014879945891106,
      "grad_norm": 0.252189964056015,
      "learning_rate": 5.997630732780505e-05,
      "loss": 0.0336,
      "step": 2370
    },
    {
      "epoch": 0.8048698004734528,
      "grad_norm": 0.20659616589546204,
      "learning_rate": 5.980707395498393e-05,
      "loss": 0.0349,
      "step": 2380
    },
    {
      "epoch": 0.808251606357795,
      "grad_norm": 0.32111021876335144,
      "learning_rate": 5.963784058216281e-05,
      "loss": 0.0334,
      "step": 2390
    },
    {
      "epoch": 0.8116334122421373,
      "grad_norm": 0.21712100505828857,
      "learning_rate": 5.946860720934169e-05,
      "loss": 0.0325,
      "step": 2400
    },
    {
      "epoch": 0.8150152181264796,
      "grad_norm": 0.2319028079509735,
      "learning_rate": 5.929937383652057e-05,
      "loss": 0.0316,
      "step": 2410
    },
    {
      "epoch": 0.8183970240108218,
      "grad_norm": 0.2731321156024933,
      "learning_rate": 5.9130140463699447e-05,
      "loss": 0.0319,
      "step": 2420
    },
    {
      "epoch": 0.821778829895164,
      "grad_norm": 0.25032860040664673,
      "learning_rate": 5.8960907090878327e-05,
      "loss": 0.0332,
      "step": 2430
    },
    {
      "epoch": 0.8251606357795063,
      "grad_norm": 0.2227882593870163,
      "learning_rate": 5.8791673718057207e-05,
      "loss": 0.0329,
      "step": 2440
    },
    {
      "epoch": 0.8285424416638485,
      "grad_norm": 0.2757587134838104,
      "learning_rate": 5.8622440345236087e-05,
      "loss": 0.0305,
      "step": 2450
    },
    {
      "epoch": 0.8319242475481907,
      "grad_norm": 0.23499512672424316,
      "learning_rate": 5.8453206972414966e-05,
      "loss": 0.0342,
      "step": 2460
    },
    {
      "epoch": 0.835306053432533,
      "grad_norm": 0.19104419648647308,
      "learning_rate": 5.8283973599593846e-05,
      "loss": 0.0341,
      "step": 2470
    },
    {
      "epoch": 0.8386878593168752,
      "grad_norm": 0.21760663390159607,
      "learning_rate": 5.8114740226772726e-05,
      "loss": 0.0287,
      "step": 2480
    },
    {
      "epoch": 0.8420696652012174,
      "grad_norm": 0.25679340958595276,
      "learning_rate": 5.7945506853951606e-05,
      "loss": 0.0356,
      "step": 2490
    },
    {
      "epoch": 0.8454514710855597,
      "grad_norm": 0.25008776783943176,
      "learning_rate": 5.777627348113048e-05,
      "loss": 0.0288,
      "step": 2500
    },
    {
      "epoch": 0.848833276969902,
      "grad_norm": 0.25389808416366577,
      "learning_rate": 5.760704010830936e-05,
      "loss": 0.0316,
      "step": 2510
    },
    {
      "epoch": 0.8522150828542442,
      "grad_norm": 0.2889176905155182,
      "learning_rate": 5.743780673548824e-05,
      "loss": 0.0333,
      "step": 2520
    },
    {
      "epoch": 0.8555968887385864,
      "grad_norm": 0.2402850240468979,
      "learning_rate": 5.726857336266712e-05,
      "loss": 0.0313,
      "step": 2530
    },
    {
      "epoch": 0.8589786946229286,
      "grad_norm": 0.217300683259964,
      "learning_rate": 5.7099339989846e-05,
      "loss": 0.0306,
      "step": 2540
    },
    {
      "epoch": 0.8623605005072709,
      "grad_norm": 0.32617250084877014,
      "learning_rate": 5.693010661702488e-05,
      "loss": 0.0293,
      "step": 2550
    },
    {
      "epoch": 0.8657423063916131,
      "grad_norm": 0.2338167428970337,
      "learning_rate": 5.676087324420376e-05,
      "loss": 0.0275,
      "step": 2560
    },
    {
      "epoch": 0.8691241122759553,
      "grad_norm": 0.2444329410791397,
      "learning_rate": 5.659163987138264e-05,
      "loss": 0.0281,
      "step": 2570
    },
    {
      "epoch": 0.8725059181602975,
      "grad_norm": 0.2600383758544922,
      "learning_rate": 5.642240649856152e-05,
      "loss": 0.0293,
      "step": 2580
    },
    {
      "epoch": 0.8758877240446399,
      "grad_norm": 0.19937843084335327,
      "learning_rate": 5.62531731257404e-05,
      "loss": 0.0327,
      "step": 2590
    },
    {
      "epoch": 0.8792695299289821,
      "grad_norm": 0.25247564911842346,
      "learning_rate": 5.608393975291928e-05,
      "loss": 0.0353,
      "step": 2600
    },
    {
      "epoch": 0.8826513358133243,
      "grad_norm": 0.20916889607906342,
      "learning_rate": 5.591470638009816e-05,
      "loss": 0.0311,
      "step": 2610
    },
    {
      "epoch": 0.8860331416976666,
      "grad_norm": 0.26659512519836426,
      "learning_rate": 5.574547300727704e-05,
      "loss": 0.0299,
      "step": 2620
    },
    {
      "epoch": 0.8894149475820088,
      "grad_norm": 0.25376033782958984,
      "learning_rate": 5.557623963445592e-05,
      "loss": 0.03,
      "step": 2630
    },
    {
      "epoch": 0.892796753466351,
      "grad_norm": 0.20384323596954346,
      "learning_rate": 5.54070062616348e-05,
      "loss": 0.0302,
      "step": 2640
    },
    {
      "epoch": 0.8961785593506932,
      "grad_norm": 0.2751913070678711,
      "learning_rate": 5.523777288881368e-05,
      "loss": 0.0296,
      "step": 2650
    },
    {
      "epoch": 0.8995603652350355,
      "grad_norm": 0.20069202780723572,
      "learning_rate": 5.506853951599256e-05,
      "loss": 0.0261,
      "step": 2660
    },
    {
      "epoch": 0.9029421711193778,
      "grad_norm": 0.21684013307094574,
      "learning_rate": 5.489930614317144e-05,
      "loss": 0.0311,
      "step": 2670
    },
    {
      "epoch": 0.90632397700372,
      "grad_norm": 0.23668934404850006,
      "learning_rate": 5.473007277035032e-05,
      "loss": 0.0303,
      "step": 2680
    },
    {
      "epoch": 0.9097057828880623,
      "grad_norm": 0.24122962355613708,
      "learning_rate": 5.456083939752919e-05,
      "loss": 0.0294,
      "step": 2690
    },
    {
      "epoch": 0.9130875887724045,
      "grad_norm": 0.19830548763275146,
      "learning_rate": 5.439160602470807e-05,
      "loss": 0.0264,
      "step": 2700
    },
    {
      "epoch": 0.9164693946567467,
      "grad_norm": 0.20941804349422455,
      "learning_rate": 5.422237265188695e-05,
      "loss": 0.029,
      "step": 2710
    },
    {
      "epoch": 0.9198512005410889,
      "grad_norm": 0.2681092619895935,
      "learning_rate": 5.405313927906583e-05,
      "loss": 0.0304,
      "step": 2720
    },
    {
      "epoch": 0.9232330064254312,
      "grad_norm": 0.20790086686611176,
      "learning_rate": 5.388390590624471e-05,
      "loss": 0.027,
      "step": 2730
    },
    {
      "epoch": 0.9266148123097734,
      "grad_norm": 0.227376326918602,
      "learning_rate": 5.371467253342359e-05,
      "loss": 0.0267,
      "step": 2740
    },
    {
      "epoch": 0.9299966181941156,
      "grad_norm": 0.26523780822753906,
      "learning_rate": 5.354543916060247e-05,
      "loss": 0.0279,
      "step": 2750
    },
    {
      "epoch": 0.933378424078458,
      "grad_norm": 0.21116450428962708,
      "learning_rate": 5.337620578778135e-05,
      "loss": 0.027,
      "step": 2760
    },
    {
      "epoch": 0.9367602299628002,
      "grad_norm": 0.22488407790660858,
      "learning_rate": 5.320697241496023e-05,
      "loss": 0.0258,
      "step": 2770
    },
    {
      "epoch": 0.9401420358471424,
      "grad_norm": 0.23635748028755188,
      "learning_rate": 5.303773904213911e-05,
      "loss": 0.0277,
      "step": 2780
    },
    {
      "epoch": 0.9435238417314846,
      "grad_norm": 0.1881859302520752,
      "learning_rate": 5.286850566931799e-05,
      "loss": 0.026,
      "step": 2790
    },
    {
      "epoch": 0.9469056476158269,
      "grad_norm": 0.28154221177101135,
      "learning_rate": 5.269927229649687e-05,
      "loss": 0.0269,
      "step": 2800
    },
    {
      "epoch": 0.9502874535001691,
      "grad_norm": 0.2318260818719864,
      "learning_rate": 5.253003892367575e-05,
      "loss": 0.0271,
      "step": 2810
    },
    {
      "epoch": 0.9536692593845113,
      "grad_norm": 0.2238345593214035,
      "learning_rate": 5.236080555085463e-05,
      "loss": 0.0253,
      "step": 2820
    },
    {
      "epoch": 0.9570510652688535,
      "grad_norm": 0.2091434746980667,
      "learning_rate": 5.219157217803351e-05,
      "loss": 0.0251,
      "step": 2830
    },
    {
      "epoch": 0.9604328711531958,
      "grad_norm": 0.2255164235830307,
      "learning_rate": 5.202233880521239e-05,
      "loss": 0.0281,
      "step": 2840
    },
    {
      "epoch": 0.9638146770375381,
      "grad_norm": 0.245024174451828,
      "learning_rate": 5.185310543239127e-05,
      "loss": 0.0257,
      "step": 2850
    },
    {
      "epoch": 0.9671964829218803,
      "grad_norm": 0.18670466542243958,
      "learning_rate": 5.168387205957015e-05,
      "loss": 0.027,
      "step": 2860
    },
    {
      "epoch": 0.9705782888062225,
      "grad_norm": 0.22900165617465973,
      "learning_rate": 5.1514638686749025e-05,
      "loss": 0.0268,
      "step": 2870
    },
    {
      "epoch": 0.9739600946905648,
      "grad_norm": 0.2123824506998062,
      "learning_rate": 5.1345405313927905e-05,
      "loss": 0.0255,
      "step": 2880
    },
    {
      "epoch": 0.977341900574907,
      "grad_norm": 0.17296041548252106,
      "learning_rate": 5.1176171941106785e-05,
      "loss": 0.0256,
      "step": 2890
    },
    {
      "epoch": 0.9807237064592492,
      "grad_norm": 0.21980832517147064,
      "learning_rate": 5.1006938568285665e-05,
      "loss": 0.027,
      "step": 2900
    },
    {
      "epoch": 0.9841055123435914,
      "grad_norm": 0.21194496750831604,
      "learning_rate": 5.0837705195464545e-05,
      "loss": 0.0253,
      "step": 2910
    },
    {
      "epoch": 0.9874873182279337,
      "grad_norm": 0.28237712383270264,
      "learning_rate": 5.0668471822643425e-05,
      "loss": 0.0261,
      "step": 2920
    },
    {
      "epoch": 0.9908691241122759,
      "grad_norm": 0.22561867535114288,
      "learning_rate": 5.0499238449822305e-05,
      "loss": 0.0263,
      "step": 2930
    },
    {
      "epoch": 0.9942509299966182,
      "grad_norm": 0.19687789678573608,
      "learning_rate": 5.0330005077001185e-05,
      "loss": 0.0253,
      "step": 2940
    },
    {
      "epoch": 0.9976327358809605,
      "grad_norm": 0.26815053820610046,
      "learning_rate": 5.0160771704180065e-05,
      "loss": 0.0315,
      "step": 2950
    },
    {
      "epoch": 1.0,
      "eval_loss": 0.025125615298748016,
      "eval_runtime": 350.9983,
      "eval_samples_per_second": 11.35,
      "eval_steps_per_second": 0.356,
      "step": 2957
    },
    {
      "epoch": 1.0010145417653027,
      "grad_norm": 0.17259906232357025,
      "learning_rate": 4.9991538331358945e-05,
      "loss": 0.0235,
      "step": 2960
    },
    {
      "epoch": 1.004396347649645,
      "grad_norm": 0.23677606880664825,
      "learning_rate": 4.9822304958537825e-05,
      "loss": 0.0231,
      "step": 2970
    },
    {
      "epoch": 1.0077781535339871,
      "grad_norm": 0.2753681540489197,
      "learning_rate": 4.9653071585716705e-05,
      "loss": 0.0225,
      "step": 2980
    },
    {
      "epoch": 1.0111599594183294,
      "grad_norm": 0.1379474550485611,
      "learning_rate": 4.9483838212895585e-05,
      "loss": 0.0209,
      "step": 2990
    },
    {
      "epoch": 1.0145417653026716,
      "grad_norm": 0.1725788563489914,
      "learning_rate": 4.9314604840074465e-05,
      "loss": 0.0231,
      "step": 3000
    },
    {
      "epoch": 1.0179235711870138,
      "grad_norm": 0.16469170153141022,
      "learning_rate": 4.9145371467253345e-05,
      "loss": 0.0235,
      "step": 3010
    },
    {
      "epoch": 1.021305377071356,
      "grad_norm": 0.1656704694032669,
      "learning_rate": 4.8976138094432225e-05,
      "loss": 0.0214,
      "step": 3020
    },
    {
      "epoch": 1.0246871829556983,
      "grad_norm": 0.22718942165374756,
      "learning_rate": 4.8806904721611104e-05,
      "loss": 0.0235,
      "step": 3030
    },
    {
      "epoch": 1.0280689888400405,
      "grad_norm": 0.18668334186077118,
      "learning_rate": 4.8637671348789984e-05,
      "loss": 0.0226,
      "step": 3040
    },
    {
      "epoch": 1.0314507947243827,
      "grad_norm": 0.1626449078321457,
      "learning_rate": 4.8468437975968864e-05,
      "loss": 0.0211,
      "step": 3050
    },
    {
      "epoch": 1.0348326006087252,
      "grad_norm": 0.14618806540966034,
      "learning_rate": 4.829920460314774e-05,
      "loss": 0.0224,
      "step": 3060
    },
    {
      "epoch": 1.0382144064930674,
      "grad_norm": 0.20951628684997559,
      "learning_rate": 4.812997123032662e-05,
      "loss": 0.0225,
      "step": 3070
    },
    {
      "epoch": 1.0415962123774096,
      "grad_norm": 0.15292339026927948,
      "learning_rate": 4.79607378575055e-05,
      "loss": 0.0236,
      "step": 3080
    },
    {
      "epoch": 1.0449780182617519,
      "grad_norm": 0.23356632888317108,
      "learning_rate": 4.779150448468438e-05,
      "loss": 0.022,
      "step": 3090
    },
    {
      "epoch": 1.048359824146094,
      "grad_norm": 0.17371255159378052,
      "learning_rate": 4.762227111186326e-05,
      "loss": 0.0221,
      "step": 3100
    },
    {
      "epoch": 1.0517416300304363,
      "grad_norm": 0.20166213810443878,
      "learning_rate": 4.745303773904214e-05,
      "loss": 0.0227,
      "step": 3110
    },
    {
      "epoch": 1.0551234359147785,
      "grad_norm": 0.18544991314411163,
      "learning_rate": 4.728380436622102e-05,
      "loss": 0.0227,
      "step": 3120
    },
    {
      "epoch": 1.0585052417991208,
      "grad_norm": 0.23601798713207245,
      "learning_rate": 4.71145709933999e-05,
      "loss": 0.022,
      "step": 3130
    },
    {
      "epoch": 1.061887047683463,
      "grad_norm": 0.20726293325424194,
      "learning_rate": 4.6945337620578784e-05,
      "loss": 0.021,
      "step": 3140
    },
    {
      "epoch": 1.0652688535678052,
      "grad_norm": 0.20292699337005615,
      "learning_rate": 4.6776104247757664e-05,
      "loss": 0.0214,
      "step": 3150
    },
    {
      "epoch": 1.0686506594521474,
      "grad_norm": 0.2641526162624359,
      "learning_rate": 4.6606870874936544e-05,
      "loss": 0.0222,
      "step": 3160
    },
    {
      "epoch": 1.0720324653364897,
      "grad_norm": 0.19379054009914398,
      "learning_rate": 4.6437637502115424e-05,
      "loss": 0.0213,
      "step": 3170
    },
    {
      "epoch": 1.0754142712208319,
      "grad_norm": 0.17844127118587494,
      "learning_rate": 4.6268404129294304e-05,
      "loss": 0.0247,
      "step": 3180
    },
    {
      "epoch": 1.0787960771051741,
      "grad_norm": 0.19252558052539825,
      "learning_rate": 4.6099170756473184e-05,
      "loss": 0.0215,
      "step": 3190
    },
    {
      "epoch": 1.0821778829895163,
      "grad_norm": 0.18375921249389648,
      "learning_rate": 4.5929937383652064e-05,
      "loss": 0.0225,
      "step": 3200
    },
    {
      "epoch": 1.0855596888738586,
      "grad_norm": 0.20377720892429352,
      "learning_rate": 4.576070401083094e-05,
      "loss": 0.0218,
      "step": 3210
    },
    {
      "epoch": 1.0889414947582008,
      "grad_norm": 0.17277295887470245,
      "learning_rate": 4.559147063800982e-05,
      "loss": 0.0203,
      "step": 3220
    },
    {
      "epoch": 1.092323300642543,
      "grad_norm": 0.17021779716014862,
      "learning_rate": 4.54222372651887e-05,
      "loss": 0.0195,
      "step": 3230
    },
    {
      "epoch": 1.0957051065268852,
      "grad_norm": 0.1649063676595688,
      "learning_rate": 4.525300389236758e-05,
      "loss": 0.0221,
      "step": 3240
    },
    {
      "epoch": 1.0990869124112277,
      "grad_norm": 0.12819623947143555,
      "learning_rate": 4.508377051954646e-05,
      "loss": 0.0216,
      "step": 3250
    },
    {
      "epoch": 1.10246871829557,
      "grad_norm": 0.20793533325195312,
      "learning_rate": 4.491453714672534e-05,
      "loss": 0.0205,
      "step": 3260
    },
    {
      "epoch": 1.1058505241799121,
      "grad_norm": 0.23723694682121277,
      "learning_rate": 4.474530377390422e-05,
      "loss": 0.0208,
      "step": 3270
    },
    {
      "epoch": 1.1092323300642544,
      "grad_norm": 0.20238593220710754,
      "learning_rate": 4.45760704010831e-05,
      "loss": 0.0212,
      "step": 3280
    },
    {
      "epoch": 1.1126141359485966,
      "grad_norm": 0.1898476779460907,
      "learning_rate": 4.440683702826198e-05,
      "loss": 0.0215,
      "step": 3290
    },
    {
      "epoch": 1.1159959418329388,
      "grad_norm": 0.1884363740682602,
      "learning_rate": 4.423760365544086e-05,
      "loss": 0.021,
      "step": 3300
    },
    {
      "epoch": 1.119377747717281,
      "grad_norm": 0.1632470339536667,
      "learning_rate": 4.406837028261974e-05,
      "loss": 0.0213,
      "step": 3310
    },
    {
      "epoch": 1.1227595536016233,
      "grad_norm": 0.1657821387052536,
      "learning_rate": 4.389913690979862e-05,
      "loss": 0.0203,
      "step": 3320
    },
    {
      "epoch": 1.1261413594859655,
      "grad_norm": 0.12199438363313675,
      "learning_rate": 4.37299035369775e-05,
      "loss": 0.0202,
      "step": 3330
    },
    {
      "epoch": 1.1295231653703077,
      "grad_norm": 0.16210108995437622,
      "learning_rate": 4.356067016415638e-05,
      "loss": 0.0205,
      "step": 3340
    },
    {
      "epoch": 1.13290497125465,
      "grad_norm": 0.18802976608276367,
      "learning_rate": 4.339143679133526e-05,
      "loss": 0.0207,
      "step": 3350
    },
    {
      "epoch": 1.1362867771389922,
      "grad_norm": 0.1435530036687851,
      "learning_rate": 4.322220341851414e-05,
      "loss": 0.0215,
      "step": 3360
    },
    {
      "epoch": 1.1396685830233344,
      "grad_norm": 0.25706106424331665,
      "learning_rate": 4.305297004569302e-05,
      "loss": 0.022,
      "step": 3370
    },
    {
      "epoch": 1.1430503889076766,
      "grad_norm": 0.19478261470794678,
      "learning_rate": 4.28837366728719e-05,
      "loss": 0.0207,
      "step": 3380
    },
    {
      "epoch": 1.1464321947920189,
      "grad_norm": 0.14196892082691193,
      "learning_rate": 4.271450330005077e-05,
      "loss": 0.0203,
      "step": 3390
    },
    {
      "epoch": 1.149814000676361,
      "grad_norm": 0.15680982172489166,
      "learning_rate": 4.254526992722965e-05,
      "loss": 0.0213,
      "step": 3400
    },
    {
      "epoch": 1.1531958065607033,
      "grad_norm": 0.1810910403728485,
      "learning_rate": 4.237603655440853e-05,
      "loss": 0.0213,
      "step": 3410
    },
    {
      "epoch": 1.1565776124450458,
      "grad_norm": 0.13093525171279907,
      "learning_rate": 4.220680318158741e-05,
      "loss": 0.0193,
      "step": 3420
    },
    {
      "epoch": 1.159959418329388,
      "grad_norm": 0.1480521410703659,
      "learning_rate": 4.203756980876629e-05,
      "loss": 0.0192,
      "step": 3430
    },
    {
      "epoch": 1.1633412242137302,
      "grad_norm": 0.20289231836795807,
      "learning_rate": 4.186833643594517e-05,
      "loss": 0.0189,
      "step": 3440
    },
    {
      "epoch": 1.1667230300980724,
      "grad_norm": 0.15804171562194824,
      "learning_rate": 4.169910306312405e-05,
      "loss": 0.0204,
      "step": 3450
    },
    {
      "epoch": 1.1701048359824147,
      "grad_norm": 0.12550042569637299,
      "learning_rate": 4.152986969030293e-05,
      "loss": 0.0193,
      "step": 3460
    },
    {
      "epoch": 1.1734866418667569,
      "grad_norm": 0.153706893324852,
      "learning_rate": 4.136063631748181e-05,
      "loss": 0.0204,
      "step": 3470
    },
    {
      "epoch": 1.1768684477510991,
      "grad_norm": 0.17369280755519867,
      "learning_rate": 4.119140294466069e-05,
      "loss": 0.0194,
      "step": 3480
    },
    {
      "epoch": 1.1802502536354413,
      "grad_norm": 0.20219433307647705,
      "learning_rate": 4.102216957183957e-05,
      "loss": 0.0234,
      "step": 3490
    },
    {
      "epoch": 1.1836320595197836,
      "grad_norm": 0.14274387061595917,
      "learning_rate": 4.085293619901845e-05,
      "loss": 0.0201,
      "step": 3500
    },
    {
      "epoch": 1.1870138654041258,
      "grad_norm": 0.1277850866317749,
      "learning_rate": 4.068370282619733e-05,
      "loss": 0.0197,
      "step": 3510
    },
    {
      "epoch": 1.190395671288468,
      "grad_norm": 0.12567327916622162,
      "learning_rate": 4.051446945337621e-05,
      "loss": 0.0193,
      "step": 3520
    },
    {
      "epoch": 1.1937774771728102,
      "grad_norm": 0.18639731407165527,
      "learning_rate": 4.034523608055509e-05,
      "loss": 0.0208,
      "step": 3530
    },
    {
      "epoch": 1.1971592830571525,
      "grad_norm": 0.17870990931987762,
      "learning_rate": 4.017600270773397e-05,
      "loss": 0.0207,
      "step": 3540
    },
    {
      "epoch": 1.2005410889414947,
      "grad_norm": 0.1992855817079544,
      "learning_rate": 4.000676933491285e-05,
      "loss": 0.0197,
      "step": 3550
    },
    {
      "epoch": 1.203922894825837,
      "grad_norm": 0.19009998440742493,
      "learning_rate": 3.983753596209173e-05,
      "loss": 0.0193,
      "step": 3560
    },
    {
      "epoch": 1.2073047007101791,
      "grad_norm": 0.2152644246816635,
      "learning_rate": 3.966830258927061e-05,
      "loss": 0.0193,
      "step": 3570
    },
    {
      "epoch": 1.2106865065945214,
      "grad_norm": 0.17644502222537994,
      "learning_rate": 3.949906921644948e-05,
      "loss": 0.0189,
      "step": 3580
    },
    {
      "epoch": 1.2140683124788638,
      "grad_norm": 0.16940535604953766,
      "learning_rate": 3.932983584362836e-05,
      "loss": 0.0195,
      "step": 3590
    },
    {
      "epoch": 1.217450118363206,
      "grad_norm": 0.1200951337814331,
      "learning_rate": 3.916060247080724e-05,
      "loss": 0.0205,
      "step": 3600
    },
    {
      "epoch": 1.2208319242475483,
      "grad_norm": 0.12686198949813843,
      "learning_rate": 3.899136909798612e-05,
      "loss": 0.0208,
      "step": 3610
    },
    {
      "epoch": 1.2242137301318905,
      "grad_norm": 0.16317637264728546,
      "learning_rate": 3.8822135725165e-05,
      "loss": 0.0198,
      "step": 3620
    },
    {
      "epoch": 1.2275955360162327,
      "grad_norm": 0.16301840543746948,
      "learning_rate": 3.865290235234388e-05,
      "loss": 0.0196,
      "step": 3630
    },
    {
      "epoch": 1.230977341900575,
      "grad_norm": 0.14271463453769684,
      "learning_rate": 3.848366897952276e-05,
      "loss": 0.0198,
      "step": 3640
    },
    {
      "epoch": 1.2343591477849172,
      "grad_norm": 0.13855944573879242,
      "learning_rate": 3.831443560670164e-05,
      "loss": 0.0192,
      "step": 3650
    },
    {
      "epoch": 1.2377409536692594,
      "grad_norm": 0.16393686830997467,
      "learning_rate": 3.814520223388052e-05,
      "loss": 0.0181,
      "step": 3660
    },
    {
      "epoch": 1.2411227595536016,
      "grad_norm": 0.13381467759609222,
      "learning_rate": 3.79759688610594e-05,
      "loss": 0.0199,
      "step": 3670
    },
    {
      "epoch": 1.2445045654379439,
      "grad_norm": 0.20619070529937744,
      "learning_rate": 3.780673548823828e-05,
      "loss": 0.02,
      "step": 3680
    },
    {
      "epoch": 1.247886371322286,
      "grad_norm": 0.1646890640258789,
      "learning_rate": 3.763750211541716e-05,
      "loss": 0.0207,
      "step": 3690
    },
    {
      "epoch": 1.2512681772066283,
      "grad_norm": 0.18202926218509674,
      "learning_rate": 3.746826874259604e-05,
      "loss": 0.0194,
      "step": 3700
    },
    {
      "epoch": 1.2546499830909705,
      "grad_norm": 0.16690859198570251,
      "learning_rate": 3.729903536977492e-05,
      "loss": 0.0198,
      "step": 3710
    },
    {
      "epoch": 1.2580317889753128,
      "grad_norm": 0.14021560549736023,
      "learning_rate": 3.71298019969538e-05,
      "loss": 0.019,
      "step": 3720
    },
    {
      "epoch": 1.261413594859655,
      "grad_norm": 0.1812976896762848,
      "learning_rate": 3.696056862413268e-05,
      "loss": 0.0202,
      "step": 3730
    },
    {
      "epoch": 1.2647954007439974,
      "grad_norm": 0.19519537687301636,
      "learning_rate": 3.679133525131156e-05,
      "loss": 0.0201,
      "step": 3740
    },
    {
      "epoch": 1.2681772066283394,
      "grad_norm": 0.1934451311826706,
      "learning_rate": 3.662210187849044e-05,
      "loss": 0.0191,
      "step": 3750
    },
    {
      "epoch": 1.2715590125126819,
      "grad_norm": 0.16278310120105743,
      "learning_rate": 3.645286850566932e-05,
      "loss": 0.0198,
      "step": 3760
    },
    {
      "epoch": 1.274940818397024,
      "grad_norm": 0.13240766525268555,
      "learning_rate": 3.6283635132848195e-05,
      "loss": 0.0185,
      "step": 3770
    },
    {
      "epoch": 1.2783226242813663,
      "grad_norm": 0.1842852085828781,
      "learning_rate": 3.6114401760027075e-05,
      "loss": 0.0192,
      "step": 3780
    },
    {
      "epoch": 1.2817044301657086,
      "grad_norm": 0.15183594822883606,
      "learning_rate": 3.5945168387205955e-05,
      "loss": 0.0208,
      "step": 3790
    },
    {
      "epoch": 1.2850862360500508,
      "grad_norm": 0.131156325340271,
      "learning_rate": 3.5775935014384835e-05,
      "loss": 0.0196,
      "step": 3800
    },
    {
      "epoch": 1.288468041934393,
      "grad_norm": 0.12825001776218414,
      "learning_rate": 3.5606701641563715e-05,
      "loss": 0.0192,
      "step": 3810
    },
    {
      "epoch": 1.2918498478187352,
      "grad_norm": 0.1289099007844925,
      "learning_rate": 3.5437468268742595e-05,
      "loss": 0.0193,
      "step": 3820
    },
    {
      "epoch": 1.2952316537030775,
      "grad_norm": 0.18116918206214905,
      "learning_rate": 3.5268234895921475e-05,
      "loss": 0.0181,
      "step": 3830
    },
    {
      "epoch": 1.2986134595874197,
      "grad_norm": 0.1824859082698822,
      "learning_rate": 3.5099001523100355e-05,
      "loss": 0.0189,
      "step": 3840
    },
    {
      "epoch": 1.301995265471762,
      "grad_norm": 0.18777935206890106,
      "learning_rate": 3.4929768150279235e-05,
      "loss": 0.0184,
      "step": 3850
    },
    {
      "epoch": 1.3053770713561041,
      "grad_norm": 0.11837860941886902,
      "learning_rate": 3.4760534777458115e-05,
      "loss": 0.0191,
      "step": 3860
    },
    {
      "epoch": 1.3087588772404464,
      "grad_norm": 0.13773775100708008,
      "learning_rate": 3.4591301404636995e-05,
      "loss": 0.0189,
      "step": 3870
    },
    {
      "epoch": 1.3121406831247886,
      "grad_norm": 0.148502916097641,
      "learning_rate": 3.4422068031815875e-05,
      "loss": 0.0184,
      "step": 3880
    },
    {
      "epoch": 1.3155224890091308,
      "grad_norm": 0.12410616874694824,
      "learning_rate": 3.4252834658994755e-05,
      "loss": 0.018,
      "step": 3890
    },
    {
      "epoch": 1.318904294893473,
      "grad_norm": 0.12317556142807007,
      "learning_rate": 3.4083601286173635e-05,
      "loss": 0.0177,
      "step": 3900
    },
    {
      "epoch": 1.3222861007778153,
      "grad_norm": 0.133961483836174,
      "learning_rate": 3.3914367913352515e-05,
      "loss": 0.0184,
      "step": 3910
    },
    {
      "epoch": 1.3256679066621575,
      "grad_norm": 0.1317528784275055,
      "learning_rate": 3.3745134540531395e-05,
      "loss": 0.02,
      "step": 3920
    },
    {
      "epoch": 1.3290497125465,
      "grad_norm": 0.14328566193580627,
      "learning_rate": 3.3575901167710275e-05,
      "loss": 0.0185,
      "step": 3930
    },
    {
      "epoch": 1.332431518430842,
      "grad_norm": 0.1426452249288559,
      "learning_rate": 3.3406667794889155e-05,
      "loss": 0.0187,
      "step": 3940
    },
    {
      "epoch": 1.3358133243151844,
      "grad_norm": 0.13055744767189026,
      "learning_rate": 3.323743442206803e-05,
      "loss": 0.0178,
      "step": 3950
    },
    {
      "epoch": 1.3391951301995266,
      "grad_norm": 0.14115405082702637,
      "learning_rate": 3.306820104924691e-05,
      "loss": 0.0201,
      "step": 3960
    },
    {
      "epoch": 1.3425769360838689,
      "grad_norm": 0.11101550608873367,
      "learning_rate": 3.289896767642579e-05,
      "loss": 0.0177,
      "step": 3970
    },
    {
      "epoch": 1.345958741968211,
      "grad_norm": 0.10724030435085297,
      "learning_rate": 3.272973430360467e-05,
      "loss": 0.0197,
      "step": 3980
    },
    {
      "epoch": 1.3493405478525533,
      "grad_norm": 0.15012751519680023,
      "learning_rate": 3.256050093078355e-05,
      "loss": 0.0184,
      "step": 3990
    },
    {
      "epoch": 1.3527223537368955,
      "grad_norm": 0.16174957156181335,
      "learning_rate": 3.239126755796243e-05,
      "loss": 0.0187,
      "step": 4000
    },
    {
      "epoch": 1.3561041596212378,
      "grad_norm": 0.12000412493944168,
      "learning_rate": 3.222203418514131e-05,
      "loss": 0.0176,
      "step": 4010
    },
    {
      "epoch": 1.35948596550558,
      "grad_norm": 0.14733338356018066,
      "learning_rate": 3.205280081232019e-05,
      "loss": 0.0177,
      "step": 4020
    },
    {
      "epoch": 1.3628677713899222,
      "grad_norm": 0.12138606607913971,
      "learning_rate": 3.188356743949907e-05,
      "loss": 0.0176,
      "step": 4030
    },
    {
      "epoch": 1.3662495772742644,
      "grad_norm": 0.1111988052725792,
      "learning_rate": 3.1714334066677954e-05,
      "loss": 0.0177,
      "step": 4040
    },
    {
      "epoch": 1.3696313831586067,
      "grad_norm": 0.22300873696804047,
      "learning_rate": 3.1545100693856834e-05,
      "loss": 0.0182,
      "step": 4050
    },
    {
      "epoch": 1.373013189042949,
      "grad_norm": 0.11778800189495087,
      "learning_rate": 3.1375867321035714e-05,
      "loss": 0.0176,
      "step": 4060
    },
    {
      "epoch": 1.3763949949272911,
      "grad_norm": 0.14025694131851196,
      "learning_rate": 3.1206633948214594e-05,
      "loss": 0.018,
      "step": 4070
    },
    {
      "epoch": 1.3797768008116333,
      "grad_norm": 0.22355903685092926,
      "learning_rate": 3.1037400575393474e-05,
      "loss": 0.02,
      "step": 4080
    },
    {
      "epoch": 1.3831586066959756,
      "grad_norm": 0.15728473663330078,
      "learning_rate": 3.0868167202572354e-05,
      "loss": 0.0175,
      "step": 4090
    },
    {
      "epoch": 1.386540412580318,
      "grad_norm": 0.13040538132190704,
      "learning_rate": 3.069893382975123e-05,
      "loss": 0.0177,
      "step": 4100
    },
    {
      "epoch": 1.38992221846466,
      "grad_norm": 0.13674882054328918,
      "learning_rate": 3.052970045693011e-05,
      "loss": 0.0189,
      "step": 4110
    },
    {
      "epoch": 1.3933040243490025,
      "grad_norm": 0.13062675297260284,
      "learning_rate": 3.036046708410899e-05,
      "loss": 0.0174,
      "step": 4120
    },
    {
      "epoch": 1.3966858302333445,
      "grad_norm": 0.07586494088172913,
      "learning_rate": 3.019123371128787e-05,
      "loss": 0.0174,
      "step": 4130
    },
    {
      "epoch": 1.400067636117687,
      "grad_norm": 0.12583595514297485,
      "learning_rate": 3.002200033846675e-05,
      "loss": 0.0174,
      "step": 4140
    },
    {
      "epoch": 1.4034494420020291,
      "grad_norm": 0.11460530757904053,
      "learning_rate": 2.9852766965645627e-05,
      "loss": 0.017,
      "step": 4150
    },
    {
      "epoch": 1.4068312478863714,
      "grad_norm": 0.14381511509418488,
      "learning_rate": 2.9683533592824507e-05,
      "loss": 0.0189,
      "step": 4160
    },
    {
      "epoch": 1.4102130537707136,
      "grad_norm": 0.12002012878656387,
      "learning_rate": 2.9514300220003387e-05,
      "loss": 0.0172,
      "step": 4170
    },
    {
      "epoch": 1.4135948596550558,
      "grad_norm": 0.1379200518131256,
      "learning_rate": 2.9345066847182267e-05,
      "loss": 0.018,
      "step": 4180
    },
    {
      "epoch": 1.416976665539398,
      "grad_norm": 0.098530113697052,
      "learning_rate": 2.9175833474361147e-05,
      "loss": 0.0172,
      "step": 4190
    },
    {
      "epoch": 1.4203584714237403,
      "grad_norm": 0.11576182395219803,
      "learning_rate": 2.9006600101540027e-05,
      "loss": 0.0187,
      "step": 4200
    },
    {
      "epoch": 1.4237402773080825,
      "grad_norm": 0.15769566595554352,
      "learning_rate": 2.8837366728718907e-05,
      "loss": 0.0179,
      "step": 4210
    },
    {
      "epoch": 1.4271220831924247,
      "grad_norm": 0.1257631480693817,
      "learning_rate": 2.8668133355897787e-05,
      "loss": 0.0174,
      "step": 4220
    },
    {
      "epoch": 1.430503889076767,
      "grad_norm": 0.09543973207473755,
      "learning_rate": 2.8498899983076667e-05,
      "loss": 0.0174,
      "step": 4230
    },
    {
      "epoch": 1.4338856949611092,
      "grad_norm": 0.12915478646755219,
      "learning_rate": 2.8329666610255544e-05,
      "loss": 0.0169,
      "step": 4240
    },
    {
      "epoch": 1.4372675008454514,
      "grad_norm": 0.14382801949977875,
      "learning_rate": 2.8160433237434424e-05,
      "loss": 0.0169,
      "step": 4250
    },
    {
      "epoch": 1.4406493067297936,
      "grad_norm": 0.1201377660036087,
      "learning_rate": 2.7991199864613304e-05,
      "loss": 0.0167,
      "step": 4260
    },
    {
      "epoch": 1.444031112614136,
      "grad_norm": 0.11754725128412247,
      "learning_rate": 2.7821966491792184e-05,
      "loss": 0.0213,
      "step": 4270
    },
    {
      "epoch": 1.447412918498478,
      "grad_norm": 0.0879068523645401,
      "learning_rate": 2.7652733118971064e-05,
      "loss": 0.0175,
      "step": 4280
    },
    {
      "epoch": 1.4507947243828205,
      "grad_norm": 0.10662252455949783,
      "learning_rate": 2.7483499746149944e-05,
      "loss": 0.019,
      "step": 4290
    },
    {
      "epoch": 1.4541765302671625,
      "grad_norm": 0.09592264890670776,
      "learning_rate": 2.7314266373328824e-05,
      "loss": 0.0182,
      "step": 4300
    },
    {
      "epoch": 1.457558336151505,
      "grad_norm": 0.19000816345214844,
      "learning_rate": 2.7145033000507703e-05,
      "loss": 0.0181,
      "step": 4310
    },
    {
      "epoch": 1.4609401420358472,
      "grad_norm": 0.18730321526527405,
      "learning_rate": 2.6975799627686583e-05,
      "loss": 0.017,
      "step": 4320
    },
    {
      "epoch": 1.4643219479201894,
      "grad_norm": 0.11948239058256149,
      "learning_rate": 2.680656625486546e-05,
      "loss": 0.0163,
      "step": 4330
    },
    {
      "epoch": 1.4677037538045317,
      "grad_norm": 0.11915738880634308,
      "learning_rate": 2.663733288204434e-05,
      "loss": 0.0176,
      "step": 4340
    },
    {
      "epoch": 1.471085559688874,
      "grad_norm": 0.11469963937997818,
      "learning_rate": 2.646809950922322e-05,
      "loss": 0.0174,
      "step": 4350
    },
    {
      "epoch": 1.4744673655732161,
      "grad_norm": 0.13674333691596985,
      "learning_rate": 2.62988661364021e-05,
      "loss": 0.0169,
      "step": 4360
    },
    {
      "epoch": 1.4778491714575583,
      "grad_norm": 0.1072223111987114,
      "learning_rate": 2.612963276358098e-05,
      "loss": 0.017,
      "step": 4370
    },
    {
      "epoch": 1.4812309773419006,
      "grad_norm": 0.12718436121940613,
      "learning_rate": 2.596039939075986e-05,
      "loss": 0.0185,
      "step": 4380
    },
    {
      "epoch": 1.4846127832262428,
      "grad_norm": 0.09312810003757477,
      "learning_rate": 2.579116601793874e-05,
      "loss": 0.0167,
      "step": 4390
    },
    {
      "epoch": 1.487994589110585,
      "grad_norm": 0.1793229579925537,
      "learning_rate": 2.562193264511762e-05,
      "loss": 0.0177,
      "step": 4400
    },
    {
      "epoch": 1.4913763949949272,
      "grad_norm": 0.10192491114139557,
      "learning_rate": 2.54526992722965e-05,
      "loss": 0.0163,
      "step": 4410
    },
    {
      "epoch": 1.4947582008792695,
      "grad_norm": 0.10123202204704285,
      "learning_rate": 2.528346589947538e-05,
      "loss": 0.017,
      "step": 4420
    },
    {
      "epoch": 1.4981400067636117,
      "grad_norm": 0.13236495852470398,
      "learning_rate": 2.5114232526654256e-05,
      "loss": 0.0172,
      "step": 4430
    },
    {
      "epoch": 1.5015218126479541,
      "grad_norm": 0.11559689790010452,
      "learning_rate": 2.4944999153833136e-05,
      "loss": 0.018,
      "step": 4440
    },
    {
      "epoch": 1.5049036185322962,
      "grad_norm": 0.1355433166027069,
      "learning_rate": 2.4775765781012016e-05,
      "loss": 0.0177,
      "step": 4450
    },
    {
      "epoch": 1.5082854244166386,
      "grad_norm": 0.09492843598127365,
      "learning_rate": 2.4606532408190896e-05,
      "loss": 0.0166,
      "step": 4460
    },
    {
      "epoch": 1.5116672303009806,
      "grad_norm": 0.14215636253356934,
      "learning_rate": 2.4437299035369776e-05,
      "loss": 0.0169,
      "step": 4470
    },
    {
      "epoch": 1.515049036185323,
      "grad_norm": 0.10210006684064865,
      "learning_rate": 2.4268065662548656e-05,
      "loss": 0.0168,
      "step": 4480
    },
    {
      "epoch": 1.518430842069665,
      "grad_norm": 0.14845940470695496,
      "learning_rate": 2.4098832289727536e-05,
      "loss": 0.0179,
      "step": 4490
    },
    {
      "epoch": 1.5218126479540075,
      "grad_norm": 0.18286077678203583,
      "learning_rate": 2.3929598916906416e-05,
      "loss": 0.0172,
      "step": 4500
    },
    {
      "epoch": 1.5251944538383497,
      "grad_norm": 0.11884652823209763,
      "learning_rate": 2.3760365544085296e-05,
      "loss": 0.0161,
      "step": 4510
    },
    {
      "epoch": 1.528576259722692,
      "grad_norm": 0.1169377937912941,
      "learning_rate": 2.3591132171264173e-05,
      "loss": 0.0168,
      "step": 4520
    },
    {
      "epoch": 1.5319580656070342,
      "grad_norm": 0.10672768950462341,
      "learning_rate": 2.3421898798443053e-05,
      "loss": 0.017,
      "step": 4530
    },
    {
      "epoch": 1.5353398714913764,
      "grad_norm": 0.06240490451455116,
      "learning_rate": 2.3252665425621933e-05,
      "loss": 0.0173,
      "step": 4540
    },
    {
      "epoch": 1.5387216773757186,
      "grad_norm": 0.12996572256088257,
      "learning_rate": 2.3083432052800813e-05,
      "loss": 0.0164,
      "step": 4550
    },
    {
      "epoch": 1.5421034832600609,
      "grad_norm": 0.10913591086864471,
      "learning_rate": 2.2914198679979693e-05,
      "loss": 0.0166,
      "step": 4560
    },
    {
      "epoch": 1.545485289144403,
      "grad_norm": 0.09768196195363998,
      "learning_rate": 2.2744965307158573e-05,
      "loss": 0.0173,
      "step": 4570
    },
    {
      "epoch": 1.5488670950287453,
      "grad_norm": 0.06998474895954132,
      "learning_rate": 2.2575731934337453e-05,
      "loss": 0.017,
      "step": 4580
    },
    {
      "epoch": 1.5522489009130875,
      "grad_norm": 0.07349732518196106,
      "learning_rate": 2.2406498561516333e-05,
      "loss": 0.0212,
      "step": 4590
    },
    {
      "epoch": 1.5556307067974298,
      "grad_norm": 0.13526155054569244,
      "learning_rate": 2.2237265188695212e-05,
      "loss": 0.016,
      "step": 4600
    },
    {
      "epoch": 1.5590125126817722,
      "grad_norm": 0.12724880874156952,
      "learning_rate": 2.206803181587409e-05,
      "loss": 0.0171,
      "step": 4610
    },
    {
      "epoch": 1.5623943185661142,
      "grad_norm": 0.08398682624101639,
      "learning_rate": 2.189879844305297e-05,
      "loss": 0.0168,
      "step": 4620
    },
    {
      "epoch": 1.5657761244504567,
      "grad_norm": 0.0917474552989006,
      "learning_rate": 2.172956507023185e-05,
      "loss": 0.0179,
      "step": 4630
    },
    {
      "epoch": 1.5691579303347987,
      "grad_norm": 0.12279586493968964,
      "learning_rate": 2.156033169741073e-05,
      "loss": 0.0166,
      "step": 4640
    },
    {
      "epoch": 1.5725397362191411,
      "grad_norm": 0.14498373866081238,
      "learning_rate": 2.139109832458961e-05,
      "loss": 0.0174,
      "step": 4650
    },
    {
      "epoch": 1.5759215421034831,
      "grad_norm": 0.1070031225681305,
      "learning_rate": 2.122186495176849e-05,
      "loss": 0.0164,
      "step": 4660
    },
    {
      "epoch": 1.5793033479878256,
      "grad_norm": 0.0644792914390564,
      "learning_rate": 2.105263157894737e-05,
      "loss": 0.0164,
      "step": 4670
    },
    {
      "epoch": 1.5826851538721676,
      "grad_norm": 0.13356371223926544,
      "learning_rate": 2.088339820612625e-05,
      "loss": 0.0171,
      "step": 4680
    },
    {
      "epoch": 1.58606695975651,
      "grad_norm": 0.061863575130701065,
      "learning_rate": 2.071416483330513e-05,
      "loss": 0.0158,
      "step": 4690
    },
    {
      "epoch": 1.5894487656408522,
      "grad_norm": 0.15974536538124084,
      "learning_rate": 2.0544931460484005e-05,
      "loss": 0.0168,
      "step": 4700
    },
    {
      "epoch": 1.5928305715251945,
      "grad_norm": 0.10037930309772491,
      "learning_rate": 2.037569808766289e-05,
      "loss": 0.016,
      "step": 4710
    },
    {
      "epoch": 1.5962123774095367,
      "grad_norm": 0.09223894774913788,
      "learning_rate": 2.020646471484177e-05,
      "loss": 0.0159,
      "step": 4720
    },
    {
      "epoch": 1.599594183293879,
      "grad_norm": 0.08334498107433319,
      "learning_rate": 2.003723134202065e-05,
      "loss": 0.0169,
      "step": 4730
    },
    {
      "epoch": 1.6029759891782212,
      "grad_norm": 0.127294659614563,
      "learning_rate": 1.986799796919953e-05,
      "loss": 0.0168,
      "step": 4740
    },
    {
      "epoch": 1.6063577950625634,
      "grad_norm": 0.0788227841258049,
      "learning_rate": 1.969876459637841e-05,
      "loss": 0.0159,
      "step": 4750
    },
    {
      "epoch": 1.6097396009469056,
      "grad_norm": 0.11868683248758316,
      "learning_rate": 1.952953122355729e-05,
      "loss": 0.0156,
      "step": 4760
    },
    {
      "epoch": 1.6131214068312478,
      "grad_norm": 0.09021388739347458,
      "learning_rate": 1.936029785073617e-05,
      "loss": 0.0169,
      "step": 4770
    },
    {
      "epoch": 1.6165032127155903,
      "grad_norm": 0.10817821323871613,
      "learning_rate": 1.9191064477915045e-05,
      "loss": 0.0168,
      "step": 4780
    },
    {
      "epoch": 1.6198850185999323,
      "grad_norm": 0.09980209171772003,
      "learning_rate": 1.9021831105093925e-05,
      "loss": 0.0163,
      "step": 4790
    },
    {
      "epoch": 1.6232668244842747,
      "grad_norm": 0.09977928549051285,
      "learning_rate": 1.8852597732272805e-05,
      "loss": 0.0165,
      "step": 4800
    },
    {
      "epoch": 1.6266486303686167,
      "grad_norm": 0.07304428517818451,
      "learning_rate": 1.8683364359451685e-05,
      "loss": 0.0159,
      "step": 4810
    },
    {
      "epoch": 1.6300304362529592,
      "grad_norm": 0.10551423579454422,
      "learning_rate": 1.8514130986630565e-05,
      "loss": 0.0167,
      "step": 4820
    },
    {
      "epoch": 1.6334122421373012,
      "grad_norm": 0.06610710173845291,
      "learning_rate": 1.8344897613809445e-05,
      "loss": 0.0164,
      "step": 4830
    },
    {
      "epoch": 1.6367940480216436,
      "grad_norm": 0.19381451606750488,
      "learning_rate": 1.8175664240988325e-05,
      "loss": 0.0166,
      "step": 4840
    },
    {
      "epoch": 1.6401758539059856,
      "grad_norm": 0.09147250652313232,
      "learning_rate": 1.8006430868167205e-05,
      "loss": 0.0163,
      "step": 4850
    },
    {
      "epoch": 1.643557659790328,
      "grad_norm": 0.1002357229590416,
      "learning_rate": 1.7837197495346085e-05,
      "loss": 0.0164,
      "step": 4860
    },
    {
      "epoch": 1.6469394656746703,
      "grad_norm": 0.08374287933111191,
      "learning_rate": 1.766796412252496e-05,
      "loss": 0.0168,
      "step": 4870
    },
    {
      "epoch": 1.6503212715590125,
      "grad_norm": 0.11019756644964218,
      "learning_rate": 1.749873074970384e-05,
      "loss": 0.0175,
      "step": 4880
    },
    {
      "epoch": 1.6537030774433548,
      "grad_norm": 0.06535135954618454,
      "learning_rate": 1.732949737688272e-05,
      "loss": 0.0157,
      "step": 4890
    },
    {
      "epoch": 1.657084883327697,
      "grad_norm": 0.0966026782989502,
      "learning_rate": 1.71602640040616e-05,
      "loss": 0.0159,
      "step": 4900
    },
    {
      "epoch": 1.6604666892120392,
      "grad_norm": 0.07409293204545975,
      "learning_rate": 1.699103063124048e-05,
      "loss": 0.0168,
      "step": 4910
    },
    {
      "epoch": 1.6638484950963814,
      "grad_norm": 0.13448843359947205,
      "learning_rate": 1.682179725841936e-05,
      "loss": 0.017,
      "step": 4920
    },
    {
      "epoch": 1.6672303009807237,
      "grad_norm": 0.11097908765077591,
      "learning_rate": 1.665256388559824e-05,
      "loss": 0.0164,
      "step": 4930
    },
    {
      "epoch": 1.670612106865066,
      "grad_norm": 0.08099315315485,
      "learning_rate": 1.648333051277712e-05,
      "loss": 0.0169,
      "step": 4940
    },
    {
      "epoch": 1.6739939127494083,
      "grad_norm": 0.10239287465810776,
      "learning_rate": 1.6314097139956e-05,
      "loss": 0.0168,
      "step": 4950
    },
    {
      "epoch": 1.6773757186337503,
      "grad_norm": 0.07663831859827042,
      "learning_rate": 1.614486376713488e-05,
      "loss": 0.0152,
      "step": 4960
    },
    {
      "epoch": 1.6807575245180928,
      "grad_norm": 0.07033158838748932,
      "learning_rate": 1.5975630394313758e-05,
      "loss": 0.0162,
      "step": 4970
    },
    {
      "epoch": 1.6841393304024348,
      "grad_norm": 0.10326120257377625,
      "learning_rate": 1.5806397021492638e-05,
      "loss": 0.0157,
      "step": 4980
    },
    {
      "epoch": 1.6875211362867772,
      "grad_norm": 0.12648841738700867,
      "learning_rate": 1.5637163648671518e-05,
      "loss": 0.0165,
      "step": 4990
    },
    {
      "epoch": 1.6909029421711192,
      "grad_norm": 0.07291960716247559,
      "learning_rate": 1.5467930275850398e-05,
      "loss": 0.0163,
      "step": 5000
    },
    {
      "epoch": 1.6942847480554617,
      "grad_norm": 0.062065865844488144,
      "learning_rate": 1.5298696903029278e-05,
      "loss": 0.0156,
      "step": 5010
    },
    {
      "epoch": 1.6976665539398037,
      "grad_norm": 0.08348127454519272,
      "learning_rate": 1.5129463530208158e-05,
      "loss": 0.0161,
      "step": 5020
    },
    {
      "epoch": 1.7010483598241462,
      "grad_norm": 0.05632692202925682,
      "learning_rate": 1.4960230157387036e-05,
      "loss": 0.0162,
      "step": 5030
    },
    {
      "epoch": 1.7044301657084884,
      "grad_norm": 0.08220499753952026,
      "learning_rate": 1.4790996784565916e-05,
      "loss": 0.0156,
      "step": 5040
    },
    {
      "epoch": 1.7078119715928306,
      "grad_norm": 0.09087113291025162,
      "learning_rate": 1.4621763411744796e-05,
      "loss": 0.0163,
      "step": 5050
    },
    {
      "epoch": 1.7111937774771728,
      "grad_norm": 0.07073336839675903,
      "learning_rate": 1.4452530038923676e-05,
      "loss": 0.0157,
      "step": 5060
    },
    {
      "epoch": 1.714575583361515,
      "grad_norm": 0.0676584392786026,
      "learning_rate": 1.4283296666102556e-05,
      "loss": 0.0156,
      "step": 5070
    },
    {
      "epoch": 1.7179573892458573,
      "grad_norm": 0.08832427114248276,
      "learning_rate": 1.4114063293281434e-05,
      "loss": 0.016,
      "step": 5080
    },
    {
      "epoch": 1.7213391951301995,
      "grad_norm": 0.14378510415554047,
      "learning_rate": 1.3944829920460314e-05,
      "loss": 0.0193,
      "step": 5090
    },
    {
      "epoch": 1.7247210010145417,
      "grad_norm": 0.0655590146780014,
      "learning_rate": 1.3775596547639194e-05,
      "loss": 0.0158,
      "step": 5100
    },
    {
      "epoch": 1.728102806898884,
      "grad_norm": 0.0975392535328865,
      "learning_rate": 1.3606363174818074e-05,
      "loss": 0.0163,
      "step": 5110
    },
    {
      "epoch": 1.7314846127832264,
      "grad_norm": 0.060749635100364685,
      "learning_rate": 1.3437129801996954e-05,
      "loss": 0.0154,
      "step": 5120
    },
    {
      "epoch": 1.7348664186675684,
      "grad_norm": 0.07566985487937927,
      "learning_rate": 1.3267896429175832e-05,
      "loss": 0.0165,
      "step": 5130
    },
    {
      "epoch": 1.7382482245519109,
      "grad_norm": 0.09878894686698914,
      "learning_rate": 1.3098663056354712e-05,
      "loss": 0.0159,
      "step": 5140
    },
    {
      "epoch": 1.7416300304362529,
      "grad_norm": 0.07339813560247421,
      "learning_rate": 1.2929429683533592e-05,
      "loss": 0.0166,
      "step": 5150
    },
    {
      "epoch": 1.7450118363205953,
      "grad_norm": 0.11021166294813156,
      "learning_rate": 1.2760196310712474e-05,
      "loss": 0.0153,
      "step": 5160
    },
    {
      "epoch": 1.7483936422049373,
      "grad_norm": 0.15555544197559357,
      "learning_rate": 1.2590962937891354e-05,
      "loss": 0.0156,
      "step": 5170
    },
    {
      "epoch": 1.7517754480892798,
      "grad_norm": 0.07689628005027771,
      "learning_rate": 1.2421729565070232e-05,
      "loss": 0.0156,
      "step": 5180
    },
    {
      "epoch": 1.7551572539736218,
      "grad_norm": 0.06506321579217911,
      "learning_rate": 1.2252496192249112e-05,
      "loss": 0.0151,
      "step": 5190
    },
    {
      "epoch": 1.7585390598579642,
      "grad_norm": 0.07474467903375626,
      "learning_rate": 1.2083262819427992e-05,
      "loss": 0.0152,
      "step": 5200
    },
    {
      "epoch": 1.7619208657423064,
      "grad_norm": 0.0740489810705185,
      "learning_rate": 1.191402944660687e-05,
      "loss": 0.0156,
      "step": 5210
    },
    {
      "epoch": 1.7653026716266487,
      "grad_norm": 0.05762564763426781,
      "learning_rate": 1.174479607378575e-05,
      "loss": 0.0148,
      "step": 5220
    },
    {
      "epoch": 1.768684477510991,
      "grad_norm": 0.06422838568687439,
      "learning_rate": 1.157556270096463e-05,
      "loss": 0.016,
      "step": 5230
    },
    {
      "epoch": 1.7720662833953331,
      "grad_norm": 0.06867840886116028,
      "learning_rate": 1.140632932814351e-05,
      "loss": 0.0148,
      "step": 5240
    },
    {
      "epoch": 1.7754480892796753,
      "grad_norm": 0.09431049972772598,
      "learning_rate": 1.123709595532239e-05,
      "loss": 0.0162,
      "step": 5250
    },
    {
      "epoch": 1.7788298951640176,
      "grad_norm": 0.13230328261852264,
      "learning_rate": 1.1067862582501269e-05,
      "loss": 0.016,
      "step": 5260
    },
    {
      "epoch": 1.7822117010483598,
      "grad_norm": 0.08878026902675629,
      "learning_rate": 1.089862920968015e-05,
      "loss": 0.0156,
      "step": 5270
    },
    {
      "epoch": 1.785593506932702,
      "grad_norm": 0.11089925467967987,
      "learning_rate": 1.072939583685903e-05,
      "loss": 0.0157,
      "step": 5280
    },
    {
      "epoch": 1.7889753128170442,
      "grad_norm": 0.08735585957765579,
      "learning_rate": 1.056016246403791e-05,
      "loss": 0.0161,
      "step": 5290
    },
    {
      "epoch": 1.7923571187013865,
      "grad_norm": 0.08069667965173721,
      "learning_rate": 1.0390929091216788e-05,
      "loss": 0.0154,
      "step": 5300
    },
    {
      "epoch": 1.795738924585729,
      "grad_norm": 0.09315579384565353,
      "learning_rate": 1.0221695718395668e-05,
      "loss": 0.0152,
      "step": 5310
    },
    {
      "epoch": 1.799120730470071,
      "grad_norm": 0.10615842044353485,
      "learning_rate": 1.0052462345574548e-05,
      "loss": 0.0158,
      "step": 5320
    },
    {
      "epoch": 1.8025025363544134,
      "grad_norm": 0.08738986402750015,
      "learning_rate": 9.883228972753428e-06,
      "loss": 0.0159,
      "step": 5330
    },
    {
      "epoch": 1.8058843422387554,
      "grad_norm": 0.08308973908424377,
      "learning_rate": 9.713995599932307e-06,
      "loss": 0.0154,
      "step": 5340
    },
    {
      "epoch": 1.8092661481230978,
      "grad_norm": 0.056414760649204254,
      "learning_rate": 9.544762227111187e-06,
      "loss": 0.0159,
      "step": 5350
    },
    {
      "epoch": 1.8126479540074398,
      "grad_norm": 0.11174483597278595,
      "learning_rate": 9.375528854290067e-06,
      "loss": 0.0157,
      "step": 5360
    },
    {
      "epoch": 1.8160297598917823,
      "grad_norm": 0.12031450867652893,
      "learning_rate": 9.206295481468947e-06,
      "loss": 0.016,
      "step": 5370
    },
    {
      "epoch": 1.8194115657761243,
      "grad_norm": 0.07165198028087616,
      "learning_rate": 9.037062108647826e-06,
      "loss": 0.0154,
      "step": 5380
    },
    {
      "epoch": 1.8227933716604667,
      "grad_norm": 0.09684468805789948,
      "learning_rate": 8.867828735826705e-06,
      "loss": 0.0161,
      "step": 5390
    },
    {
      "epoch": 1.826175177544809,
      "grad_norm": 0.06861387938261032,
      "learning_rate": 8.698595363005585e-06,
      "loss": 0.0157,
      "step": 5400
    },
    {
      "epoch": 1.8295569834291512,
      "grad_norm": 0.06754298508167267,
      "learning_rate": 8.529361990184465e-06,
      "loss": 0.0156,
      "step": 5410
    },
    {
      "epoch": 1.8329387893134934,
      "grad_norm": 0.07989432662725449,
      "learning_rate": 8.360128617363345e-06,
      "loss": 0.0166,
      "step": 5420
    },
    {
      "epoch": 1.8363205951978356,
      "grad_norm": 0.07799744606018066,
      "learning_rate": 8.190895244542223e-06,
      "loss": 0.0153,
      "step": 5430
    },
    {
      "epoch": 1.8397024010821779,
      "grad_norm": 0.07402709126472473,
      "learning_rate": 8.021661871721103e-06,
      "loss": 0.0158,
      "step": 5440
    },
    {
      "epoch": 1.84308420696652,
      "grad_norm": 0.05629491060972214,
      "learning_rate": 7.852428498899983e-06,
      "loss": 0.0158,
      "step": 5450
    },
    {
      "epoch": 1.8464660128508623,
      "grad_norm": 0.05271398648619652,
      "learning_rate": 7.683195126078863e-06,
      "loss": 0.0161,
      "step": 5460
    },
    {
      "epoch": 1.8498478187352045,
      "grad_norm": 0.06697215884923935,
      "learning_rate": 7.513961753257742e-06,
      "loss": 0.0151,
      "step": 5470
    },
    {
      "epoch": 1.853229624619547,
      "grad_norm": 0.06437050551176071,
      "learning_rate": 7.344728380436622e-06,
      "loss": 0.0162,
      "step": 5480
    },
    {
      "epoch": 1.856611430503889,
      "grad_norm": 0.0728522315621376,
      "learning_rate": 7.175495007615503e-06,
      "loss": 0.0157,
      "step": 5490
    },
    {
      "epoch": 1.8599932363882314,
      "grad_norm": 0.08476530015468597,
      "learning_rate": 7.006261634794383e-06,
      "loss": 0.0153,
      "step": 5500
    },
    {
      "epoch": 1.8633750422725734,
      "grad_norm": 0.06566642969846725,
      "learning_rate": 6.837028261973262e-06,
      "loss": 0.0148,
      "step": 5510
    },
    {
      "epoch": 1.866756848156916,
      "grad_norm": 0.09338746964931488,
      "learning_rate": 6.667794889152142e-06,
      "loss": 0.0157,
      "step": 5520
    },
    {
      "epoch": 1.870138654041258,
      "grad_norm": 0.05955313891172409,
      "learning_rate": 6.498561516331021e-06,
      "loss": 0.0157,
      "step": 5530
    },
    {
      "epoch": 1.8735204599256003,
      "grad_norm": 0.11766281723976135,
      "learning_rate": 6.329328143509901e-06,
      "loss": 0.0157,
      "step": 5540
    },
    {
      "epoch": 1.8769022658099423,
      "grad_norm": 0.06120260804891586,
      "learning_rate": 6.16009477068878e-06,
      "loss": 0.0154,
      "step": 5550
    },
    {
      "epoch": 1.8802840716942848,
      "grad_norm": 0.09879166632890701,
      "learning_rate": 5.99086139786766e-06,
      "loss": 0.015,
      "step": 5560
    },
    {
      "epoch": 1.883665877578627,
      "grad_norm": 0.0957876592874527,
      "learning_rate": 5.82162802504654e-06,
      "loss": 0.0161,
      "step": 5570
    },
    {
      "epoch": 1.8870476834629692,
      "grad_norm": 0.07821828126907349,
      "learning_rate": 5.652394652225419e-06,
      "loss": 0.0162,
      "step": 5580
    },
    {
      "epoch": 1.8904294893473115,
      "grad_norm": 0.06772307306528091,
      "learning_rate": 5.483161279404299e-06,
      "loss": 0.0153,
      "step": 5590
    },
    {
      "epoch": 1.8938112952316537,
      "grad_norm": 0.09948345273733139,
      "learning_rate": 5.313927906583178e-06,
      "loss": 0.0156,
      "step": 5600
    },
    {
      "epoch": 1.897193101115996,
      "grad_norm": 0.07948364317417145,
      "learning_rate": 5.144694533762058e-06,
      "loss": 0.017,
      "step": 5610
    },
    {
      "epoch": 1.9005749070003382,
      "grad_norm": 0.10279282182455063,
      "learning_rate": 4.975461160940937e-06,
      "loss": 0.0151,
      "step": 5620
    },
    {
      "epoch": 1.9039567128846804,
      "grad_norm": 0.08826450258493423,
      "learning_rate": 4.806227788119817e-06,
      "loss": 0.0154,
      "step": 5630
    },
    {
      "epoch": 1.9073385187690226,
      "grad_norm": 0.09827665984630585,
      "learning_rate": 4.636994415298697e-06,
      "loss": 0.0156,
      "step": 5640
    },
    {
      "epoch": 1.910720324653365,
      "grad_norm": 0.06779050081968307,
      "learning_rate": 4.467761042477576e-06,
      "loss": 0.0154,
      "step": 5650
    },
    {
      "epoch": 1.914102130537707,
      "grad_norm": 0.05314110219478607,
      "learning_rate": 4.298527669656456e-06,
      "loss": 0.0156,
      "step": 5660
    },
    {
      "epoch": 1.9174839364220495,
      "grad_norm": 0.0623813197016716,
      "learning_rate": 4.129294296835336e-06,
      "loss": 0.0163,
      "step": 5670
    },
    {
      "epoch": 1.9208657423063915,
      "grad_norm": 0.06000444293022156,
      "learning_rate": 3.960060924014216e-06,
      "loss": 0.0152,
      "step": 5680
    },
    {
      "epoch": 1.924247548190734,
      "grad_norm": 0.059466857463121414,
      "learning_rate": 3.7908275511930954e-06,
      "loss": 0.0152,
      "step": 5690
    },
    {
      "epoch": 1.927629354075076,
      "grad_norm": 0.04620284587144852,
      "learning_rate": 3.6215941783719754e-06,
      "loss": 0.0151,
      "step": 5700
    },
    {
      "epoch": 1.9310111599594184,
      "grad_norm": 0.129261776804924,
      "learning_rate": 3.452360805550855e-06,
      "loss": 0.0152,
      "step": 5710
    },
    {
      "epoch": 1.9343929658437604,
      "grad_norm": 0.07549330592155457,
      "learning_rate": 3.2831274327297345e-06,
      "loss": 0.0157,
      "step": 5720
    },
    {
      "epoch": 1.9377747717281029,
      "grad_norm": 0.0915280282497406,
      "learning_rate": 3.113894059908614e-06,
      "loss": 0.0148,
      "step": 5730
    },
    {
      "epoch": 1.941156577612445,
      "grad_norm": 0.09341620653867722,
      "learning_rate": 2.9446606870874936e-06,
      "loss": 0.0151,
      "step": 5740
    },
    {
      "epoch": 1.9445383834967873,
      "grad_norm": 0.07050041854381561,
      "learning_rate": 2.7754273142663736e-06,
      "loss": 0.0153,
      "step": 5750
    },
    {
      "epoch": 1.9479201893811295,
      "grad_norm": 0.05427863821387291,
      "learning_rate": 2.606193941445253e-06,
      "loss": 0.0149,
      "step": 5760
    },
    {
      "epoch": 1.9513019952654718,
      "grad_norm": 0.08598488569259644,
      "learning_rate": 2.4369605686241327e-06,
      "loss": 0.0149,
      "step": 5770
    },
    {
      "epoch": 1.954683801149814,
      "grad_norm": 0.08857986330986023,
      "learning_rate": 2.267727195803012e-06,
      "loss": 0.0147,
      "step": 5780
    },
    {
      "epoch": 1.9580656070341562,
      "grad_norm": 0.09314049780368805,
      "learning_rate": 2.098493822981892e-06,
      "loss": 0.0158,
      "step": 5790
    },
    {
      "epoch": 1.9614474129184984,
      "grad_norm": 0.057003553956747055,
      "learning_rate": 1.929260450160772e-06,
      "loss": 0.0155,
      "step": 5800
    },
    {
      "epoch": 1.9648292188028407,
      "grad_norm": 0.0754491537809372,
      "learning_rate": 1.7600270773396515e-06,
      "loss": 0.0152,
      "step": 5810
    },
    {
      "epoch": 1.9682110246871831,
      "grad_norm": 0.07122204452753067,
      "learning_rate": 1.5907937045185312e-06,
      "loss": 0.0152,
      "step": 5820
    },
    {
      "epoch": 1.9715928305715251,
      "grad_norm": 0.06150359660387039,
      "learning_rate": 1.4215603316974108e-06,
      "loss": 0.0144,
      "step": 5830
    },
    {
      "epoch": 1.9749746364558676,
      "grad_norm": 0.04940520599484444,
      "learning_rate": 1.2523269588762905e-06,
      "loss": 0.0149,
      "step": 5840
    },
    {
      "epoch": 1.9783564423402096,
      "grad_norm": 0.06114586442708969,
      "learning_rate": 1.08309358605517e-06,
      "loss": 0.0157,
      "step": 5850
    },
    {
      "epoch": 1.981738248224552,
      "grad_norm": 0.06917215138673782,
      "learning_rate": 9.138602132340497e-07,
      "loss": 0.0155,
      "step": 5860
    },
    {
      "epoch": 1.985120054108894,
      "grad_norm": 0.04913664981722832,
      "learning_rate": 7.446268404129294e-07,
      "loss": 0.0155,
      "step": 5870
    },
    {
      "epoch": 1.9885018599932365,
      "grad_norm": 0.07977385073900223,
      "learning_rate": 5.753934675918092e-07,
      "loss": 0.0144,
      "step": 5880
    },
    {
      "epoch": 1.9918836658775785,
      "grad_norm": 0.07336314022541046,
      "learning_rate": 4.061600947706888e-07,
      "loss": 0.015,
      "step": 5890
    },
    {
      "epoch": 1.995265471761921,
      "grad_norm": 0.05290009453892708,
      "learning_rate": 2.3692672194956846e-07,
      "loss": 0.0154,
      "step": 5900
    },
    {
      "epoch": 1.9986472776462632,
      "grad_norm": 0.06475949287414551,
      "learning_rate": 6.769334912844813e-08,
      "loss": 0.0151,
      "step": 5910
    },
    {
      "epoch": 2.0,
      "eval_loss": 0.016348782926797867,
      "eval_runtime": 342.3753,
      "eval_samples_per_second": 11.636,
      "eval_steps_per_second": 0.365,
      "step": 5914
    }
  ],
  "logging_steps": 10,
  "max_steps": 5914,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 500,
  "total_flos": 8.364323669652013e+18,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}