{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0,
  "eval_steps": 500,
  "global_step": 17092,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "grad_norm": 11.189342496234838,
      "learning_rate": 4.2000000000000006e-07,
      "loss": 0.6334,
      "step": 20
    },
    {
      "epoch": 0.0,
      "grad_norm": 12.102946252189938,
      "learning_rate": 6.515789473684211e-07,
      "loss": 0.6334,
      "step": 40
    },
    {
      "epoch": 0.01,
      "grad_norm": 9.23868714422811,
      "learning_rate": 8.831578947368421e-07,
      "loss": 0.5919,
      "step": 60
    },
    {
      "epoch": 0.01,
      "grad_norm": 10.163427827153294,
      "learning_rate": 1.1147368421052632e-06,
      "loss": 0.591,
      "step": 80
    },
    {
      "epoch": 0.01,
      "grad_norm": 7.703301789458378,
      "learning_rate": 1.3463157894736842e-06,
      "loss": 0.5663,
      "step": 100
    },
    {
      "epoch": 0.01,
      "grad_norm": 10.292384648986614,
      "learning_rate": 1.5778947368421053e-06,
      "loss": 0.5649,
      "step": 120
    },
    {
      "epoch": 0.02,
      "grad_norm": 9.622475032798782,
      "learning_rate": 1.8094736842105263e-06,
      "loss": 0.5215,
      "step": 140
    },
    {
      "epoch": 0.02,
      "grad_norm": 12.80974448742081,
      "learning_rate": 2.0410526315789475e-06,
      "loss": 0.5166,
      "step": 160
    },
    {
      "epoch": 0.02,
      "grad_norm": 7.2494201428270495,
      "learning_rate": 2.2726315789473687e-06,
      "loss": 0.4708,
      "step": 180
    },
    {
      "epoch": 0.02,
      "grad_norm": 7.525527996410342,
      "learning_rate": 2.5042105263157895e-06,
      "loss": 0.4809,
      "step": 200
    },
    {
      "epoch": 0.03,
      "grad_norm": 9.553244905277726,
      "learning_rate": 2.735789473684211e-06,
      "loss": 0.4149,
      "step": 220
    },
    {
      "epoch": 0.03,
      "grad_norm": 10.08934392514418,
      "learning_rate": 2.967368421052632e-06,
      "loss": 0.4709,
      "step": 240
    },
    {
      "epoch": 0.03,
      "grad_norm": 8.77781615537256,
      "learning_rate": 3.198947368421053e-06,
      "loss": 0.4706,
      "step": 260
    },
    {
      "epoch": 0.03,
      "grad_norm": 8.842231539252408,
      "learning_rate": 3.430526315789474e-06,
      "loss": 0.4033,
      "step": 280
    },
    {
      "epoch": 0.04,
      "grad_norm": 10.007575408247385,
      "learning_rate": 3.662105263157895e-06,
      "loss": 0.4411,
      "step": 300
    },
    {
      "epoch": 0.04,
      "grad_norm": 8.652379461692668,
      "learning_rate": 3.893684210526316e-06,
      "loss": 0.4011,
      "step": 320
    },
    {
      "epoch": 0.04,
      "grad_norm": 9.03023367578369,
      "learning_rate": 4.125263157894737e-06,
      "loss": 0.4252,
      "step": 340
    },
    {
      "epoch": 0.04,
      "grad_norm": 8.141636176102654,
      "learning_rate": 4.356842105263159e-06,
      "loss": 0.4364,
      "step": 360
    },
    {
      "epoch": 0.04,
      "grad_norm": 8.361423248626743,
      "learning_rate": 4.58842105263158e-06,
      "loss": 0.4193,
      "step": 380
    },
    {
      "epoch": 0.05,
      "grad_norm": 9.373028764113052,
      "learning_rate": 4.8200000000000004e-06,
      "loss": 0.4078,
      "step": 400
    },
    {
      "epoch": 0.05,
      "grad_norm": 8.799844515033747,
      "learning_rate": 5.051578947368422e-06,
      "loss": 0.4291,
      "step": 420
    },
    {
      "epoch": 0.05,
      "grad_norm": 8.642174028335415,
      "learning_rate": 5.283157894736842e-06,
      "loss": 0.382,
      "step": 440
    },
    {
      "epoch": 0.05,
      "grad_norm": 9.451813409785364,
      "learning_rate": 5.514736842105264e-06,
      "loss": 0.4089,
      "step": 460
    },
    {
      "epoch": 0.06,
      "grad_norm": 11.329024075313082,
      "learning_rate": 5.7463157894736845e-06,
      "loss": 0.4015,
      "step": 480
    },
    {
      "epoch": 0.06,
      "grad_norm": 7.271736970582471,
      "learning_rate": 5.977894736842106e-06,
      "loss": 0.4028,
      "step": 500
    },
    {
      "epoch": 0.06,
      "grad_norm": 10.856360235187129,
      "learning_rate": 6.209473684210528e-06,
      "loss": 0.3814,
      "step": 520
    },
    {
      "epoch": 0.06,
      "grad_norm": 10.40735773829138,
      "learning_rate": 6.441052631578948e-06,
      "loss": 0.3868,
      "step": 540
    },
    {
      "epoch": 0.07,
      "grad_norm": 7.176143531188922,
      "learning_rate": 6.672631578947369e-06,
      "loss": 0.4089,
      "step": 560
    },
    {
      "epoch": 0.07,
      "grad_norm": 9.628282656024224,
      "learning_rate": 6.90421052631579e-06,
      "loss": 0.3771,
      "step": 580
    },
    {
      "epoch": 0.07,
      "grad_norm": 11.199839686200068,
      "learning_rate": 7.135789473684212e-06,
      "loss": 0.383,
      "step": 600
    },
    {
      "epoch": 0.07,
      "grad_norm": 8.042449860326105,
      "learning_rate": 7.367368421052632e-06,
      "loss": 0.3875,
      "step": 620
    },
    {
      "epoch": 0.07,
      "grad_norm": 6.256777636127039,
      "learning_rate": 7.598947368421053e-06,
      "loss": 0.399,
      "step": 640
    },
    {
      "epoch": 0.08,
      "grad_norm": 12.140400652487944,
      "learning_rate": 7.830526315789473e-06,
      "loss": 0.399,
      "step": 660
    },
    {
      "epoch": 0.08,
      "grad_norm": 9.31740323216306,
      "learning_rate": 8.062105263157896e-06,
      "loss": 0.383,
      "step": 680
    },
    {
      "epoch": 0.08,
      "grad_norm": 11.518507949844242,
      "learning_rate": 8.293684210526317e-06,
      "loss": 0.3673,
      "step": 700
    },
    {
      "epoch": 0.08,
      "grad_norm": 12.23485079993539,
      "learning_rate": 8.525263157894737e-06,
      "loss": 0.4105,
      "step": 720
    },
    {
      "epoch": 0.09,
      "grad_norm": 7.506176757455166,
      "learning_rate": 8.756842105263158e-06,
      "loss": 0.425,
      "step": 740
    },
    {
      "epoch": 0.09,
      "grad_norm": 28.3327677352401,
      "learning_rate": 8.988421052631579e-06,
      "loss": 0.3874,
      "step": 760
    },
    {
      "epoch": 0.09,
      "grad_norm": 10.006030362522525,
      "learning_rate": 9.220000000000002e-06,
      "loss": 0.4055,
      "step": 780
    },
    {
      "epoch": 0.09,
      "grad_norm": 10.010446813711976,
      "learning_rate": 9.451578947368422e-06,
      "loss": 0.3599,
      "step": 800
    },
    {
      "epoch": 0.1,
      "grad_norm": 7.043688000572858,
      "learning_rate": 9.683157894736843e-06,
      "loss": 0.3624,
      "step": 820
    },
    {
      "epoch": 0.1,
      "grad_norm": 9.607021831412599,
      "learning_rate": 9.914736842105264e-06,
      "loss": 0.3952,
      "step": 840
    },
    {
      "epoch": 0.1,
      "grad_norm": 9.147395412600936,
      "learning_rate": 1.0146315789473686e-05,
      "loss": 0.3819,
      "step": 860
    },
    {
      "epoch": 0.1,
      "grad_norm": 7.993375419567414,
      "learning_rate": 1.0377894736842107e-05,
      "loss": 0.3678,
      "step": 880
    },
    {
      "epoch": 0.11,
      "grad_norm": 7.52609874523302,
      "learning_rate": 1.0609473684210528e-05,
      "loss": 0.375,
      "step": 900
    },
    {
      "epoch": 0.11,
      "grad_norm": 9.52738539095041,
      "learning_rate": 1.084105263157895e-05,
      "loss": 0.3681,
      "step": 920
    },
    {
      "epoch": 0.11,
      "grad_norm": 11.370588516129013,
      "learning_rate": 1.107263157894737e-05,
      "loss": 0.3911,
      "step": 940
    },
    {
      "epoch": 0.11,
      "grad_norm": 9.391054368953709,
      "learning_rate": 1.130421052631579e-05,
      "loss": 0.329,
      "step": 960
    },
    {
      "epoch": 0.11,
      "grad_norm": 7.997549873904863,
      "learning_rate": 1.1535789473684211e-05,
      "loss": 0.359,
      "step": 980
    },
    {
      "epoch": 0.12,
      "grad_norm": 10.755576173466588,
      "learning_rate": 1.1767368421052634e-05,
      "loss": 0.393,
      "step": 1000
    },
    {
      "epoch": 0.12,
      "grad_norm": 6.458117606549104,
      "learning_rate": 1.1998947368421054e-05,
      "loss": 0.363,
      "step": 1020
    },
    {
      "epoch": 0.12,
      "grad_norm": 9.937055469862932,
      "learning_rate": 1.2230526315789475e-05,
      "loss": 0.3766,
      "step": 1040
    },
    {
      "epoch": 0.12,
      "grad_norm": 7.485816768913991,
      "learning_rate": 1.2462105263157898e-05,
      "loss": 0.3845,
      "step": 1060
    },
    {
      "epoch": 0.13,
      "grad_norm": 8.01333127284645,
      "learning_rate": 1.2693684210526319e-05,
      "loss": 0.3617,
      "step": 1080
    },
    {
      "epoch": 0.13,
      "grad_norm": 8.911089094498925,
      "learning_rate": 1.292526315789474e-05,
      "loss": 0.3599,
      "step": 1100
    },
    {
      "epoch": 0.13,
      "grad_norm": 8.32079434857686,
      "learning_rate": 1.3156842105263158e-05,
      "loss": 0.3722,
      "step": 1120
    },
    {
      "epoch": 0.13,
      "grad_norm": 9.806231592184165,
      "learning_rate": 1.3388421052631581e-05,
      "loss": 0.3755,
      "step": 1140
    },
    {
      "epoch": 0.14,
      "grad_norm": 8.798076540805445,
      "learning_rate": 1.3620000000000002e-05,
      "loss": 0.3616,
      "step": 1160
    },
    {
      "epoch": 0.14,
      "grad_norm": 8.343613755617508,
      "learning_rate": 1.3851578947368423e-05,
      "loss": 0.3631,
      "step": 1180
    },
    {
      "epoch": 0.14,
      "grad_norm": 8.35814690526204,
      "learning_rate": 1.4083157894736843e-05,
      "loss": 0.3638,
      "step": 1200
    },
    {
      "epoch": 0.14,
      "grad_norm": 7.706542458286787,
      "learning_rate": 1.4314736842105266e-05,
      "loss": 0.375,
      "step": 1220
    },
    {
      "epoch": 0.15,
      "grad_norm": 10.89212756963322,
      "learning_rate": 1.4546315789473687e-05,
      "loss": 0.4049,
      "step": 1240
    },
    {
      "epoch": 0.15,
      "grad_norm": 11.46198790559678,
      "learning_rate": 1.4777894736842107e-05,
      "loss": 0.332,
      "step": 1260
    },
    {
      "epoch": 0.15,
      "grad_norm": 8.365156167551179,
      "learning_rate": 1.500947368421053e-05,
      "loss": 0.3586,
      "step": 1280
    },
    {
      "epoch": 0.15,
      "grad_norm": 7.841036992292662,
      "learning_rate": 1.5241052631578949e-05,
      "loss": 0.368,
      "step": 1300
    },
    {
      "epoch": 0.15,
      "grad_norm": 9.84275168548854,
      "learning_rate": 1.547263157894737e-05,
      "loss": 0.3767,
      "step": 1320
    },
    {
      "epoch": 0.16,
      "grad_norm": 10.506254240871392,
      "learning_rate": 1.5704210526315792e-05,
      "loss": 0.378,
      "step": 1340
    },
    {
      "epoch": 0.16,
      "grad_norm": 7.844581104033147,
      "learning_rate": 1.5935789473684213e-05,
      "loss": 0.3453,
      "step": 1360
    },
    {
      "epoch": 0.16,
      "grad_norm": 8.193758138016792,
      "learning_rate": 1.6167368421052634e-05,
      "loss": 0.3752,
      "step": 1380
    },
    {
      "epoch": 0.16,
      "grad_norm": 7.475620318034102,
      "learning_rate": 1.6398947368421055e-05,
      "loss": 0.3642,
      "step": 1400
    },
    {
      "epoch": 0.17,
      "grad_norm": 7.4964850772334355,
      "learning_rate": 1.6630526315789475e-05,
      "loss": 0.3377,
      "step": 1420
    },
    {
      "epoch": 0.17,
      "grad_norm": 7.903753436810423,
      "learning_rate": 1.6862105263157896e-05,
      "loss": 0.3586,
      "step": 1440
    },
    {
      "epoch": 0.17,
      "grad_norm": 10.49594728277505,
      "learning_rate": 1.7093684210526317e-05,
      "loss": 0.3468,
      "step": 1460
    },
    {
      "epoch": 0.17,
      "grad_norm": 8.027713457118994,
      "learning_rate": 1.7325263157894738e-05,
      "loss": 0.3491,
      "step": 1480
    },
    {
      "epoch": 0.18,
      "grad_norm": 7.2748681125287415,
      "learning_rate": 1.755684210526316e-05,
      "loss": 0.3581,
      "step": 1500
    },
    {
      "epoch": 0.18,
      "grad_norm": 6.077326940397276,
      "learning_rate": 1.7788421052631583e-05,
      "loss": 0.3356,
      "step": 1520
    },
    {
      "epoch": 0.18,
      "grad_norm": 6.724980645701413,
      "learning_rate": 1.802e-05,
      "loss": 0.3656,
      "step": 1540
    },
    {
      "epoch": 0.18,
      "grad_norm": 8.592315221136559,
      "learning_rate": 1.825157894736842e-05,
      "loss": 0.3467,
      "step": 1560
    },
    {
      "epoch": 0.18,
      "grad_norm": 9.036253074638084,
      "learning_rate": 1.8483157894736845e-05,
      "loss": 0.3414,
      "step": 1580
    },
    {
      "epoch": 0.19,
      "grad_norm": 11.349080832480995,
      "learning_rate": 1.8714736842105266e-05,
      "loss": 0.3655,
      "step": 1600
    },
    {
      "epoch": 0.19,
      "grad_norm": 9.361475025905081,
      "learning_rate": 1.8946315789473687e-05,
      "loss": 0.3564,
      "step": 1620
    },
    {
      "epoch": 0.19,
      "grad_norm": 6.53643855564322,
      "learning_rate": 1.9177894736842108e-05,
      "loss": 0.3714,
      "step": 1640
    },
    {
      "epoch": 0.19,
      "grad_norm": 6.65699298275954,
      "learning_rate": 1.940947368421053e-05,
      "loss": 0.3546,
      "step": 1660
    },
    {
      "epoch": 0.2,
      "grad_norm": 9.856127510930383,
      "learning_rate": 1.964105263157895e-05,
      "loss": 0.3308,
      "step": 1680
    },
    {
      "epoch": 0.2,
      "grad_norm": 8.532146476622113,
      "learning_rate": 1.987263157894737e-05,
      "loss": 0.366,
      "step": 1700
    },
    {
      "epoch": 0.2,
      "grad_norm": 8.327666937449752,
      "learning_rate": 1.999999532098205e-05,
      "loss": 0.3247,
      "step": 1720
    },
    {
      "epoch": 0.2,
      "grad_norm": 10.876035443701728,
      "learning_rate": 1.99999578888647e-05,
      "loss": 0.3543,
      "step": 1740
    },
    {
      "epoch": 0.21,
      "grad_norm": 8.263711805253918,
      "learning_rate": 1.999988302477013e-05,
      "loss": 0.3664,
      "step": 1760
    },
    {
      "epoch": 0.21,
      "grad_norm": 7.575419338189409,
      "learning_rate": 1.9999770728978607e-05,
      "loss": 0.345,
      "step": 1780
    },
    {
      "epoch": 0.21,
      "grad_norm": 7.814279338388738,
      "learning_rate": 1.9999621001910508e-05,
      "loss": 0.3573,
      "step": 1800
    },
    {
      "epoch": 0.21,
      "grad_norm": 13.040107479471628,
      "learning_rate": 1.999943384412636e-05,
      "loss": 0.3344,
      "step": 1820
    },
    {
      "epoch": 0.22,
      "grad_norm": 7.535441012066771,
      "learning_rate": 1.99992092563268e-05,
      "loss": 0.3552,
      "step": 1840
    },
    {
      "epoch": 0.22,
      "grad_norm": 8.059491683038381,
      "learning_rate": 1.9998947239352593e-05,
      "loss": 0.3442,
      "step": 1860
    },
    {
      "epoch": 0.22,
      "grad_norm": 6.700609139607897,
      "learning_rate": 1.999864779418462e-05,
      "loss": 0.3584,
      "step": 1880
    },
    {
      "epoch": 0.22,
      "grad_norm": 6.831136707088909,
      "learning_rate": 1.999831092194389e-05,
      "loss": 0.3341,
      "step": 1900
    },
    {
      "epoch": 0.22,
      "grad_norm": 10.30262237054047,
      "learning_rate": 1.9997936623891498e-05,
      "loss": 0.3704,
      "step": 1920
    },
    {
      "epoch": 0.23,
      "grad_norm": 6.703691792033561,
      "learning_rate": 1.9997524901428677e-05,
      "loss": 0.3769,
      "step": 1940
    },
    {
      "epoch": 0.23,
      "grad_norm": 7.251026212477321,
      "learning_rate": 1.9997075756096734e-05,
      "loss": 0.3234,
      "step": 1960
    },
    {
      "epoch": 0.23,
      "grad_norm": 6.14471190830255,
      "learning_rate": 1.9996589189577094e-05,
      "loss": 0.3443,
      "step": 1980
    },
    {
      "epoch": 0.23,
      "grad_norm": 7.254885638733633,
      "learning_rate": 1.999606520369125e-05,
      "loss": 0.3178,
      "step": 2000
    },
    {
      "epoch": 0.24,
      "grad_norm": 7.330977870771951,
      "learning_rate": 1.9995503800400805e-05,
      "loss": 0.3309,
      "step": 2020
    },
    {
      "epoch": 0.24,
      "grad_norm": 7.291895603037368,
      "learning_rate": 1.9994904981807407e-05,
      "loss": 0.3372,
      "step": 2040
    },
    {
      "epoch": 0.24,
      "grad_norm": 12.772894555548028,
      "learning_rate": 1.9994268750152793e-05,
      "loss": 0.3747,
      "step": 2060
    },
    {
      "epoch": 0.24,
      "grad_norm": 9.497788021161657,
      "learning_rate": 1.999359510781875e-05,
      "loss": 0.3529,
      "step": 2080
    },
    {
      "epoch": 0.25,
      "grad_norm": 6.581748417513793,
      "learning_rate": 1.9992884057327118e-05,
      "loss": 0.3395,
      "step": 2100
    },
    {
      "epoch": 0.25,
      "grad_norm": 9.523030924078585,
      "learning_rate": 1.9992135601339776e-05,
      "loss": 0.3469,
      "step": 2120
    },
    {
      "epoch": 0.25,
      "grad_norm": 17.640537636796516,
      "learning_rate": 1.999134974265864e-05,
      "loss": 0.3365,
      "step": 2140
    },
    {
      "epoch": 0.25,
      "grad_norm": 5.4385511763432985,
      "learning_rate": 1.999052648422563e-05,
      "loss": 0.3845,
      "step": 2160
    },
    {
      "epoch": 0.26,
      "grad_norm": 7.585921156892153,
      "learning_rate": 1.9989665829122704e-05,
      "loss": 0.3464,
      "step": 2180
    },
    {
      "epoch": 0.26,
      "grad_norm": 12.163408304702841,
      "learning_rate": 1.9988767780571778e-05,
      "loss": 0.3127,
      "step": 2200
    },
    {
      "epoch": 0.26,
      "grad_norm": 8.246994395835456,
      "learning_rate": 1.9987832341934793e-05,
      "loss": 0.3424,
      "step": 2220
    },
    {
      "epoch": 0.26,
      "grad_norm": 9.628684787640838,
      "learning_rate": 1.9986859516713636e-05,
      "loss": 0.3472,
      "step": 2240
    },
    {
      "epoch": 0.26,
      "grad_norm": 10.912326639203211,
      "learning_rate": 1.9985849308550167e-05,
      "loss": 0.3326,
      "step": 2260
    },
    {
      "epoch": 0.27,
      "grad_norm": 6.419410251137725,
      "learning_rate": 1.9984801721226188e-05,
      "loss": 0.3297,
      "step": 2280
    },
    {
      "epoch": 0.27,
      "grad_norm": 7.798212276898864,
      "learning_rate": 1.998371675866344e-05,
      "loss": 0.3389,
      "step": 2300
    },
    {
      "epoch": 0.27,
      "grad_norm": 7.811483576458649,
      "learning_rate": 1.9982594424923566e-05,
      "loss": 0.332,
      "step": 2320
    },
    {
      "epoch": 0.27,
      "grad_norm": 6.967736687994363,
      "learning_rate": 1.9981434724208126e-05,
      "loss": 0.3099,
      "step": 2340
    },
    {
      "epoch": 0.28,
      "grad_norm": 9.635115819134949,
      "learning_rate": 1.9980237660858565e-05,
      "loss": 0.3285,
      "step": 2360
    },
    {
      "epoch": 0.28,
      "grad_norm": 6.720069081541748,
      "learning_rate": 1.997900323935619e-05,
      "loss": 0.343,
      "step": 2380
    },
    {
      "epoch": 0.28,
      "grad_norm": 10.300889386045979,
      "learning_rate": 1.9977731464322162e-05,
      "loss": 0.3667,
      "step": 2400
    },
    {
      "epoch": 0.28,
      "grad_norm": 7.888854177455323,
      "learning_rate": 1.9976422340517495e-05,
      "loss": 0.3397,
      "step": 2420
    },
    {
      "epoch": 0.29,
      "grad_norm": 9.643051064776678,
      "learning_rate": 1.9975075872843e-05,
      "loss": 0.3371,
      "step": 2440
    },
    {
      "epoch": 0.29,
      "grad_norm": 6.166262467342929,
      "learning_rate": 1.9973692066339298e-05,
      "loss": 0.3424,
      "step": 2460
    },
    {
      "epoch": 0.29,
      "grad_norm": 7.520601841401043,
      "learning_rate": 1.9972270926186794e-05,
      "loss": 0.3083,
      "step": 2480
    },
    {
      "epoch": 0.29,
      "grad_norm": 6.095472200047568,
      "learning_rate": 1.9970812457705642e-05,
      "loss": 0.3512,
      "step": 2500
    },
    {
      "epoch": 0.29,
      "grad_norm": 7.259492119293066,
      "learning_rate": 1.9969316666355757e-05,
      "loss": 0.3309,
      "step": 2520
    },
    {
      "epoch": 0.3,
      "grad_norm": 8.355978245335697,
      "learning_rate": 1.996778355773676e-05,
      "loss": 0.3296,
      "step": 2540
    },
    {
      "epoch": 0.3,
      "grad_norm": 9.612502269967832,
      "learning_rate": 1.996621313758798e-05,
      "loss": 0.3391,
      "step": 2560
    },
    {
      "epoch": 0.3,
      "grad_norm": 8.148285817739712,
      "learning_rate": 1.9964605411788424e-05,
      "loss": 0.3416,
      "step": 2580
    },
    {
      "epoch": 0.3,
      "grad_norm": 7.645206131186397,
      "learning_rate": 1.9962960386356748e-05,
      "loss": 0.352,
      "step": 2600
    },
    {
      "epoch": 0.31,
      "grad_norm": 5.406075006199623,
      "learning_rate": 1.9961278067451258e-05,
      "loss": 0.3194,
      "step": 2620
    },
    {
      "epoch": 0.31,
      "grad_norm": 7.23674984010716,
      "learning_rate": 1.995955846136986e-05,
      "loss": 0.3216,
      "step": 2640
    },
    {
      "epoch": 0.31,
      "grad_norm": 8.044071872202302,
      "learning_rate": 1.9957801574550044e-05,
      "loss": 0.321,
      "step": 2660
    },
    {
      "epoch": 0.31,
      "grad_norm": 7.2624583573058485,
      "learning_rate": 1.9956007413568874e-05,
      "loss": 0.3352,
      "step": 2680
    },
    {
      "epoch": 0.32,
      "grad_norm": 9.277277549105909,
      "learning_rate": 1.9954175985142955e-05,
      "loss": 0.3391,
      "step": 2700
    },
    {
      "epoch": 0.32,
      "grad_norm": 10.975275864316648,
      "learning_rate": 1.9952307296128395e-05,
      "loss": 0.3298,
      "step": 2720
    },
    {
      "epoch": 0.32,
      "grad_norm": 11.416959735607907,
      "learning_rate": 1.9950401353520795e-05,
      "loss": 0.3382,
      "step": 2740
    },
    {
      "epoch": 0.32,
      "grad_norm": 7.384993942351236,
      "learning_rate": 1.994845816445522e-05,
      "loss": 0.3422,
      "step": 2760
    },
    {
      "epoch": 0.33,
      "grad_norm": 6.775839902047113,
      "learning_rate": 1.9946477736206167e-05,
      "loss": 0.3694,
      "step": 2780
    },
    {
      "epoch": 0.33,
      "grad_norm": 8.246414330131564,
      "learning_rate": 1.9944460076187546e-05,
      "loss": 0.3458,
      "step": 2800
    },
    {
      "epoch": 0.33,
      "grad_norm": 7.152660285783071,
      "learning_rate": 1.9942405191952646e-05,
      "loss": 0.3426,
      "step": 2820
    },
    {
      "epoch": 0.33,
      "grad_norm": 6.9408668390055315,
      "learning_rate": 1.9940313091194096e-05,
      "loss": 0.3097,
      "step": 2840
    },
    {
      "epoch": 0.33,
      "grad_norm": 5.797513942002627,
      "learning_rate": 1.9938183781743873e-05,
      "loss": 0.3102,
      "step": 2860
    },
    {
      "epoch": 0.34,
      "grad_norm": 7.361935180345544,
      "learning_rate": 1.9936017271573223e-05,
      "loss": 0.3102,
      "step": 2880
    },
    {
      "epoch": 0.34,
      "grad_norm": 7.4796257480282,
      "learning_rate": 1.9933813568792678e-05,
      "loss": 0.3256,
      "step": 2900
    },
    {
      "epoch": 0.34,
      "grad_norm": 5.500192812227557,
      "learning_rate": 1.9931572681651984e-05,
      "loss": 0.3107,
      "step": 2920
    },
    {
      "epoch": 0.34,
      "grad_norm": 13.84776864376109,
      "learning_rate": 1.9929294618540103e-05,
      "loss": 0.3062,
      "step": 2940
    },
    {
      "epoch": 0.35,
      "grad_norm": 7.937410128828084,
      "learning_rate": 1.992697938798516e-05,
      "loss": 0.3293,
      "step": 2960
    },
    {
      "epoch": 0.35,
      "grad_norm": 5.447364190742796,
      "learning_rate": 1.9924626998654433e-05,
      "loss": 0.3318,
      "step": 2980
    },
    {
      "epoch": 0.35,
      "grad_norm": 7.825239369201605,
      "learning_rate": 1.9922237459354287e-05,
      "loss": 0.3238,
      "step": 3000
    },
    {
      "epoch": 0.35,
      "grad_norm": 6.466926354506395,
      "learning_rate": 1.9919810779030177e-05,
      "loss": 0.3136,
      "step": 3020
    },
    {
      "epoch": 0.36,
      "grad_norm": 10.115420483963225,
      "learning_rate": 1.9917346966766596e-05,
      "loss": 0.3329,
      "step": 3040
    },
    {
      "epoch": 0.36,
      "grad_norm": 6.08100630613984,
      "learning_rate": 1.991484603178704e-05,
      "loss": 0.2999,
      "step": 3060
    },
    {
      "epoch": 0.36,
      "grad_norm": 6.153474908414458,
      "learning_rate": 1.991230798345398e-05,
      "loss": 0.3344,
      "step": 3080
    },
    {
      "epoch": 0.36,
      "grad_norm": 6.956842076140817,
      "learning_rate": 1.990973283126882e-05,
      "loss": 0.3356,
      "step": 3100
    },
    {
      "epoch": 0.37,
      "grad_norm": 7.977473013654998,
      "learning_rate": 1.990712058487187e-05,
      "loss": 0.2984,
      "step": 3120
    },
    {
      "epoch": 0.37,
      "grad_norm": 7.472067890046691,
      "learning_rate": 1.9904471254042304e-05,
      "loss": 0.308,
      "step": 3140
    },
    {
      "epoch": 0.37,
      "grad_norm": 6.98090482864238,
      "learning_rate": 1.9901784848698125e-05,
      "loss": 0.3036,
      "step": 3160
    },
    {
      "epoch": 0.37,
      "grad_norm": 7.229553593404789,
      "learning_rate": 1.989906137889613e-05,
      "loss": 0.3027,
      "step": 3180
    },
    {
      "epoch": 0.37,
      "grad_norm": 5.9700766482137295,
      "learning_rate": 1.9896300854831862e-05,
      "loss": 0.3329,
      "step": 3200
    },
    {
      "epoch": 0.38,
      "grad_norm": 7.864131148182889,
      "learning_rate": 1.989350328683959e-05,
      "loss": 0.308,
      "step": 3220
    },
    {
      "epoch": 0.38,
      "grad_norm": 20.04577205534678,
      "learning_rate": 1.989066868539225e-05,
      "loss": 0.3217,
      "step": 3240
    },
    {
      "epoch": 0.38,
      "grad_norm": 8.46751929768339,
      "learning_rate": 1.9887797061101424e-05,
      "loss": 0.3036,
      "step": 3260
    },
    {
      "epoch": 0.38,
      "grad_norm": 7.667111535917501,
      "learning_rate": 1.9884888424717296e-05,
      "loss": 0.3084,
      "step": 3280
    },
    {
      "epoch": 0.39,
      "grad_norm": 6.615069450077278,
      "learning_rate": 1.9881942787128594e-05,
      "loss": 0.3177,
      "step": 3300
    },
    {
      "epoch": 0.39,
      "grad_norm": 9.324876177990959,
      "learning_rate": 1.9878960159362572e-05,
      "loss": 0.3496,
      "step": 3320
    },
    {
      "epoch": 0.39,
      "grad_norm": 7.1979554134343635,
      "learning_rate": 1.987594055258496e-05,
      "loss": 0.3167,
      "step": 3340
    },
    {
      "epoch": 0.39,
      "grad_norm": 29.076306154493622,
      "learning_rate": 1.987288397809993e-05,
      "loss": 0.34,
      "step": 3360
    },
    {
      "epoch": 0.4,
      "grad_norm": 7.861933206527291,
      "learning_rate": 1.986979044735002e-05,
      "loss": 0.3424,
      "step": 3380
    },
    {
      "epoch": 0.4,
      "grad_norm": 5.7878723156599445,
      "learning_rate": 1.9866659971916146e-05,
      "loss": 0.3157,
      "step": 3400
    },
    {
      "epoch": 0.4,
      "grad_norm": 8.470531768859948,
      "learning_rate": 1.9863492563517513e-05,
      "loss": 0.3201,
      "step": 3420
    },
    {
      "epoch": 0.4,
      "grad_norm": 8.104846079245535,
      "learning_rate": 1.98602882340116e-05,
      "loss": 0.322,
      "step": 3440
    },
    {
      "epoch": 0.4,
      "grad_norm": 6.180289951670451,
      "learning_rate": 1.9857046995394084e-05,
      "loss": 0.2991,
      "step": 3460
    },
    {
      "epoch": 0.41,
      "grad_norm": 4.962160167060266,
      "learning_rate": 1.985376885979884e-05,
      "loss": 0.3097,
      "step": 3480
    },
    {
      "epoch": 0.41,
      "grad_norm": 6.813966925636774,
      "learning_rate": 1.9850453839497844e-05,
      "loss": 0.3294,
      "step": 3500
    },
    {
      "epoch": 0.41,
      "grad_norm": 5.902763508299491,
      "learning_rate": 1.9847101946901172e-05,
      "loss": 0.2967,
      "step": 3520
    },
    {
      "epoch": 0.41,
      "grad_norm": 7.359798475906595,
      "learning_rate": 1.984371319455693e-05,
      "loss": 0.3268,
      "step": 3540
    },
    {
      "epoch": 0.42,
      "grad_norm": 7.377881796357637,
      "learning_rate": 1.9840287595151205e-05,
      "loss": 0.3251,
      "step": 3560
    },
    {
      "epoch": 0.42,
      "grad_norm": 7.261601866242624,
      "learning_rate": 1.9836825161508032e-05,
      "loss": 0.284,
      "step": 3580
    },
    {
      "epoch": 0.42,
      "grad_norm": 6.123609423918951,
      "learning_rate": 1.983332590658934e-05,
      "loss": 0.3137,
      "step": 3600
    },
    {
      "epoch": 0.42,
      "grad_norm": 7.205290164862431,
      "learning_rate": 1.9829789843494885e-05,
      "loss": 0.3,
      "step": 3620
    },
    {
      "epoch": 0.43,
      "grad_norm": 6.0471563544848035,
      "learning_rate": 1.9826216985462245e-05,
      "loss": 0.2991,
      "step": 3640
    },
    {
      "epoch": 0.43,
      "grad_norm": 7.436589545934819,
      "learning_rate": 1.9822607345866715e-05,
      "loss": 0.3299,
      "step": 3660
    },
    {
      "epoch": 0.43,
      "grad_norm": 8.240853065109379,
      "learning_rate": 1.9818960938221308e-05,
      "loss": 0.2886,
      "step": 3680
    },
    {
      "epoch": 0.43,
      "grad_norm": 5.451034188795461,
      "learning_rate": 1.9815277776176663e-05,
      "loss": 0.3138,
      "step": 3700
    },
    {
      "epoch": 0.44,
      "grad_norm": 7.874196329434049,
      "learning_rate": 1.9811557873521027e-05,
      "loss": 0.2946,
      "step": 3720
    },
    {
      "epoch": 0.44,
      "grad_norm": 6.547528436867746,
      "learning_rate": 1.9807801244180174e-05,
      "loss": 0.2978,
      "step": 3740
    },
    {
      "epoch": 0.44,
      "grad_norm": 5.816214082411462,
      "learning_rate": 1.980400790221739e-05,
      "loss": 0.3067,
      "step": 3760
    },
    {
      "epoch": 0.44,
      "grad_norm": 6.245697982302123,
      "learning_rate": 1.9800177861833372e-05,
      "loss": 0.3191,
      "step": 3780
    },
    {
      "epoch": 0.44,
      "grad_norm": 7.932045331416635,
      "learning_rate": 1.9796311137366212e-05,
      "loss": 0.3439,
      "step": 3800
    },
    {
      "epoch": 0.45,
      "grad_norm": 6.088006858952151,
      "learning_rate": 1.9792407743291343e-05,
      "loss": 0.33,
      "step": 3820
    },
    {
      "epoch": 0.45,
      "grad_norm": 5.770262468473867,
      "learning_rate": 1.9788467694221453e-05,
      "loss": 0.2897,
      "step": 3840
    },
    {
      "epoch": 0.45,
      "grad_norm": 7.978864884873927,
      "learning_rate": 1.9784491004906468e-05,
      "loss": 0.3118,
      "step": 3860
    },
    {
      "epoch": 0.45,
      "grad_norm": 6.631368778757732,
      "learning_rate": 1.9780477690233476e-05,
      "loss": 0.3165,
      "step": 3880
    },
    {
      "epoch": 0.46,
      "grad_norm": 6.541290304773085,
      "learning_rate": 1.9776427765226664e-05,
      "loss": 0.3379,
      "step": 3900
    },
    {
      "epoch": 0.46,
      "grad_norm": 6.42886929503141,
      "learning_rate": 1.9772341245047285e-05,
      "loss": 0.3136,
      "step": 3920
    },
    {
      "epoch": 0.46,
      "grad_norm": 9.004651033459043,
      "learning_rate": 1.9768218144993593e-05,
      "loss": 0.311,
      "step": 3940
    },
    {
      "epoch": 0.46,
      "grad_norm": 6.241942743461084,
      "learning_rate": 1.976405848050077e-05,
      "loss": 0.3224,
      "step": 3960
    },
    {
      "epoch": 0.47,
      "grad_norm": 7.030346892956473,
      "learning_rate": 1.975986226714089e-05,
      "loss": 0.3513,
      "step": 3980
    },
    {
      "epoch": 0.47,
      "grad_norm": 6.984184655106913,
      "learning_rate": 1.9755629520622835e-05,
      "loss": 0.3031,
      "step": 4000
    },
    {
      "epoch": 0.47,
      "grad_norm": 3.6594620639728253,
      "learning_rate": 1.9751360256792274e-05,
      "loss": 0.2819,
      "step": 4020
    },
    {
      "epoch": 0.47,
      "grad_norm": 3.9453364305431418,
      "learning_rate": 1.9747054491631567e-05,
      "loss": 0.2806,
      "step": 4040
    },
    {
      "epoch": 0.48,
      "grad_norm": 3.8004852838340475,
      "learning_rate": 1.9742712241259726e-05,
      "loss": 0.2712,
      "step": 4060
    },
    {
      "epoch": 0.48,
      "grad_norm": 4.671864793440983,
      "learning_rate": 1.9738333521932348e-05,
      "loss": 0.264,
      "step": 4080
    },
    {
      "epoch": 0.48,
      "grad_norm": 3.9158215253486452,
      "learning_rate": 1.9733918350041546e-05,
      "loss": 0.2466,
      "step": 4100
    },
    {
      "epoch": 0.48,
      "grad_norm": 3.232730170815344,
      "learning_rate": 1.972946674211591e-05,
      "loss": 0.28,
      "step": 4120
    },
    {
      "epoch": 0.48,
      "grad_norm": 5.26102706338969,
      "learning_rate": 1.972497871482043e-05,
      "loss": 0.2793,
      "step": 4140
    },
    {
      "epoch": 0.49,
      "grad_norm": 6.036896745386087,
      "learning_rate": 1.972045428495642e-05,
      "loss": 0.2551,
      "step": 4160
    },
    {
      "epoch": 0.49,
      "grad_norm": 3.4899809349872823,
      "learning_rate": 1.971589346946149e-05,
      "loss": 0.2746,
      "step": 4180
    },
    {
      "epoch": 0.49,
      "grad_norm": 4.003065603442425,
      "learning_rate": 1.9711296285409443e-05,
      "loss": 0.2639,
      "step": 4200
    },
    {
      "epoch": 0.49,
      "grad_norm": 2.8477299266017626,
      "learning_rate": 1.9706662750010255e-05,
      "loss": 0.2689,
      "step": 4220
    },
    {
      "epoch": 0.5,
      "grad_norm": 3.5491058459283837,
      "learning_rate": 1.970199288060996e-05,
      "loss": 0.2642,
      "step": 4240
    },
    {
      "epoch": 0.5,
      "grad_norm": 3.658049963633355,
      "learning_rate": 1.969728669469063e-05,
      "loss": 0.2699,
      "step": 4260
    },
    {
      "epoch": 0.5,
      "grad_norm": 3.7093939532444127,
      "learning_rate": 1.9692544209870287e-05,
      "loss": 0.2697,
      "step": 4280
    },
    {
      "epoch": 0.5,
      "grad_norm": 3.0342042036143444,
      "learning_rate": 1.968776544390283e-05,
      "loss": 0.2654,
      "step": 4300
    },
    {
      "epoch": 0.51,
      "grad_norm": 3.8174199112021174,
      "learning_rate": 1.9682950414678004e-05,
      "loss": 0.2525,
      "step": 4320
    },
    {
      "epoch": 0.51,
      "grad_norm": 5.35961771156863,
      "learning_rate": 1.9678099140221284e-05,
      "loss": 0.2464,
      "step": 4340
    },
    {
      "epoch": 0.51,
      "grad_norm": 5.063501894675061,
      "learning_rate": 1.9673211638693844e-05,
      "loss": 0.2788,
      "step": 4360
    },
    {
      "epoch": 0.51,
      "grad_norm": 3.8581072620927723,
      "learning_rate": 1.9668287928392478e-05,
      "loss": 0.2743,
      "step": 4380
    },
    {
      "epoch": 0.51,
      "grad_norm": 3.386524586463125,
      "learning_rate": 1.9663328027749523e-05,
      "loss": 0.2547,
      "step": 4400
    },
    {
      "epoch": 0.52,
      "grad_norm": 4.049063661443696,
      "learning_rate": 1.9658331955332807e-05,
      "loss": 0.2787,
      "step": 4420
    },
    {
      "epoch": 0.52,
      "grad_norm": 5.12761026294579,
      "learning_rate": 1.9653299729845564e-05,
      "loss": 0.2577,
      "step": 4440
    },
    {
      "epoch": 0.52,
      "grad_norm": 3.808933278135121,
      "learning_rate": 1.9648231370126375e-05,
      "loss": 0.2654,
      "step": 4460
    },
    {
      "epoch": 0.52,
      "grad_norm": 4.402659150192465,
      "learning_rate": 1.9643126895149087e-05,
      "loss": 0.2603,
      "step": 4480
    },
    {
      "epoch": 0.53,
      "grad_norm": 8.492295307503369,
      "learning_rate": 1.963798632402275e-05,
      "loss": 0.2691,
      "step": 4500
    },
    {
      "epoch": 0.53,
      "grad_norm": 3.628014593555578,
      "learning_rate": 1.9632809675991544e-05,
      "loss": 0.2711,
      "step": 4520
    },
    {
      "epoch": 0.53,
      "grad_norm": 5.207256154885955,
      "learning_rate": 1.9627596970434708e-05,
      "loss": 0.2604,
      "step": 4540
    },
    {
      "epoch": 0.53,
      "grad_norm": 3.892539047603965,
      "learning_rate": 1.962234822686646e-05,
      "loss": 0.2693,
      "step": 4560
    },
    {
      "epoch": 0.54,
      "grad_norm": 4.005334872796194,
      "learning_rate": 1.961706346493594e-05,
      "loss": 0.2774,
      "step": 4580
    },
    {
      "epoch": 0.54,
      "grad_norm": 4.456881895914548,
      "learning_rate": 1.9611742704427106e-05,
      "loss": 0.2531,
      "step": 4600
    },
    {
      "epoch": 0.54,
      "grad_norm": 3.925512807597538,
      "learning_rate": 1.9606385965258697e-05,
      "loss": 0.2507,
      "step": 4620
    },
    {
      "epoch": 0.54,
      "grad_norm": 4.401211069070843,
      "learning_rate": 1.9600993267484147e-05,
      "loss": 0.262,
      "step": 4640
    },
    {
      "epoch": 0.55,
      "grad_norm": 5.3437799820979075,
      "learning_rate": 1.9595564631291478e-05,
      "loss": 0.2516,
      "step": 4660
    },
    {
      "epoch": 0.55,
      "grad_norm": 4.553539308066039,
      "learning_rate": 1.9590100077003278e-05,
      "loss": 0.2603,
      "step": 4680
    },
    {
      "epoch": 0.55,
      "grad_norm": 3.169670353545694,
      "learning_rate": 1.958459962507658e-05,
      "loss": 0.2875,
      "step": 4700
    },
    {
      "epoch": 0.55,
      "grad_norm": 3.572973282405317,
      "learning_rate": 1.9579063296102812e-05,
      "loss": 0.2671,
      "step": 4720
    },
    {
      "epoch": 0.55,
      "grad_norm": 3.59782258517998,
      "learning_rate": 1.957349111080771e-05,
      "loss": 0.2837,
      "step": 4740
    },
    {
      "epoch": 0.56,
      "grad_norm": 5.199297439358037,
      "learning_rate": 1.956788309005123e-05,
      "loss": 0.2753,
      "step": 4760
    },
    {
      "epoch": 0.56,
      "grad_norm": 4.993257936146954,
      "learning_rate": 1.95622392548275e-05,
      "loss": 0.2714,
      "step": 4780
    },
    {
      "epoch": 0.56,
      "grad_norm": 3.7288744650364642,
      "learning_rate": 1.955655962626471e-05,
      "loss": 0.2609,
      "step": 4800
    },
    {
      "epoch": 0.56,
      "grad_norm": 5.377784540016259,
      "learning_rate": 1.9550844225625043e-05,
      "loss": 0.2605,
      "step": 4820
    },
    {
      "epoch": 0.57,
      "grad_norm": 3.1237811954279353,
      "learning_rate": 1.954509307430461e-05,
      "loss": 0.2569,
      "step": 4840
    },
    {
      "epoch": 0.57,
      "grad_norm": 4.101705029145583,
      "learning_rate": 1.953930619383335e-05,
      "loss": 0.2719,
      "step": 4860
    },
    {
      "epoch": 0.57,
      "grad_norm": 2.734092828632754,
      "learning_rate": 1.953348360587496e-05,
      "loss": 0.2503,
      "step": 4880
    },
    {
      "epoch": 0.57,
      "grad_norm": 3.4516150854343604,
      "learning_rate": 1.9527625332226804e-05,
      "loss": 0.2588,
      "step": 4900
    },
    {
      "epoch": 0.58,
      "grad_norm": 3.0561153265178085,
      "learning_rate": 1.952173139481985e-05,
      "loss": 0.2779,
      "step": 4920
    },
    {
      "epoch": 0.58,
      "grad_norm": 3.5096519082422986,
      "learning_rate": 1.9515801815718566e-05,
      "loss": 0.2567,
      "step": 4940
    },
    {
      "epoch": 0.58,
      "grad_norm": 3.1690240083159993,
      "learning_rate": 1.9509836617120856e-05,
      "loss": 0.2466,
      "step": 4960
    },
    {
      "epoch": 0.58,
      "grad_norm": 3.2970371296868484,
      "learning_rate": 1.950383582135796e-05,
      "loss": 0.2762,
      "step": 4980
    },
    {
      "epoch": 0.59,
      "grad_norm": 4.326207366177782,
      "learning_rate": 1.9497799450894387e-05,
      "loss": 0.2681,
      "step": 5000
    },
    {
      "epoch": 0.59,
      "grad_norm": 3.977255410491104,
      "learning_rate": 1.949172752832782e-05,
      "loss": 0.235,
      "step": 5020
    },
    {
      "epoch": 0.59,
      "grad_norm": 3.5634772315616665,
      "learning_rate": 1.9485620076389036e-05,
      "loss": 0.254,
      "step": 5040
    },
    {
      "epoch": 0.59,
      "grad_norm": 2.9667663371741377,
      "learning_rate": 1.9479477117941815e-05,
      "loss": 0.2582,
      "step": 5060
    },
    {
      "epoch": 0.59,
      "grad_norm": 3.628580495546253,
      "learning_rate": 1.9473298675982863e-05,
      "loss": 0.2526,
      "step": 5080
    },
    {
      "epoch": 0.6,
      "grad_norm": 7.393689026504584,
      "learning_rate": 1.946708477364172e-05,
      "loss": 0.2587,
      "step": 5100
    },
    {
      "epoch": 0.6,
      "grad_norm": 3.1975448011133407,
      "learning_rate": 1.9460835434180677e-05,
      "loss": 0.2679,
      "step": 5120
    },
    {
      "epoch": 0.6,
      "grad_norm": 3.231683615655966,
      "learning_rate": 1.9454550680994684e-05,
      "loss": 0.2589,
      "step": 5140
    },
    {
      "epoch": 0.6,
      "grad_norm": 2.956736460495959,
      "learning_rate": 1.9448230537611266e-05,
      "loss": 0.2594,
      "step": 5160
    },
    {
      "epoch": 0.61,
      "grad_norm": 3.326543297808526,
      "learning_rate": 1.9441875027690435e-05,
      "loss": 0.2556,
      "step": 5180
    },
    {
      "epoch": 0.61,
      "grad_norm": 3.635188025395725,
      "learning_rate": 1.94354841750246e-05,
      "loss": 0.2391,
      "step": 5200
    },
    {
      "epoch": 0.61,
      "grad_norm": 4.934642109517512,
      "learning_rate": 1.942905800353848e-05,
      "loss": 0.2576,
      "step": 5220
    },
    {
      "epoch": 0.61,
      "grad_norm": 3.8685357103793767,
      "learning_rate": 1.9422596537289016e-05,
      "loss": 0.2735,
      "step": 5240
    },
    {
      "epoch": 0.62,
      "grad_norm": 4.387694986472,
      "learning_rate": 1.941609980046527e-05,
      "loss": 0.2584,
      "step": 5260
    },
    {
      "epoch": 0.62,
      "grad_norm": 2.8046325848231977,
      "learning_rate": 1.9409567817388347e-05,
      "loss": 0.2527,
      "step": 5280
    },
    {
      "epoch": 0.62,
      "grad_norm": 3.4012361074638466,
      "learning_rate": 1.94030006125113e-05,
      "loss": 0.2566,
      "step": 5300
    },
    {
      "epoch": 0.62,
      "grad_norm": 5.450089536158873,
      "learning_rate": 1.9396398210419038e-05,
      "loss": 0.2667,
      "step": 5320
    },
    {
      "epoch": 0.62,
      "grad_norm": 3.7967573233998677,
      "learning_rate": 1.938976063582824e-05,
      "loss": 0.2536,
      "step": 5340
    },
    {
      "epoch": 0.63,
      "grad_norm": 3.0487888683110866,
      "learning_rate": 1.938308791358723e-05,
      "loss": 0.2673,
      "step": 5360
    },
    {
      "epoch": 0.63,
      "grad_norm": 3.2736066169116285,
      "learning_rate": 1.9376380068675956e-05,
      "loss": 0.2592,
      "step": 5380
    },
    {
      "epoch": 0.63,
      "grad_norm": 9.727952335441667,
      "learning_rate": 1.93696371262058e-05,
      "loss": 0.273,
      "step": 5400
    },
    {
      "epoch": 0.63,
      "grad_norm": 3.3824664284674686,
      "learning_rate": 1.936285911141958e-05,
      "loss": 0.2787,
      "step": 5420
    },
    {
      "epoch": 0.64,
      "grad_norm": 6.879080376487243,
      "learning_rate": 1.935604604969138e-05,
      "loss": 0.2698,
      "step": 5440
    },
    {
      "epoch": 0.64,
      "grad_norm": 4.579168728014292,
      "learning_rate": 1.934919796652649e-05,
      "loss": 0.2589,
      "step": 5460
    },
    {
      "epoch": 0.64,
      "grad_norm": 4.6776324362587784,
      "learning_rate": 1.9342314887561323e-05,
      "loss": 0.2544,
      "step": 5480
    },
    {
      "epoch": 0.64,
      "grad_norm": 3.693997622998789,
      "learning_rate": 1.9335396838563282e-05,
      "loss": 0.2446,
      "step": 5500
    },
    {
      "epoch": 0.65,
      "grad_norm": 5.06873813162095,
      "learning_rate": 1.932844384543069e-05,
      "loss": 0.2618,
      "step": 5520
    },
    {
      "epoch": 0.65,
      "grad_norm": 6.816411566527297,
      "learning_rate": 1.9321455934192693e-05,
      "loss": 0.2614,
      "step": 5540
    },
    {
      "epoch": 0.65,
      "grad_norm": 3.5475232040366853,
      "learning_rate": 1.9314433131009146e-05,
      "loss": 0.2942,
      "step": 5560
    },
    {
      "epoch": 0.65,
      "grad_norm": 5.287550198652632,
      "learning_rate": 1.9307375462170527e-05,
      "loss": 0.2482,
      "step": 5580
    },
    {
      "epoch": 0.66,
      "grad_norm": 11.389854038766874,
      "learning_rate": 1.9300282954097847e-05,
      "loss": 0.245,
      "step": 5600
    },
    {
      "epoch": 0.66,
      "grad_norm": 4.178435806663215,
      "learning_rate": 1.9293155633342534e-05,
      "loss": 0.2534,
      "step": 5620
    },
    {
      "epoch": 0.66,
      "grad_norm": 3.1710922703595035,
      "learning_rate": 1.928599352658633e-05,
      "loss": 0.2622,
      "step": 5640
    },
    {
      "epoch": 0.66,
      "grad_norm": 2.6946627953356996,
      "learning_rate": 1.927879666064122e-05,
      "loss": 0.2626,
      "step": 5660
    },
    {
      "epoch": 0.66,
      "grad_norm": 4.8581812333670955,
      "learning_rate": 1.92715650624493e-05,
      "loss": 0.2346,
      "step": 5680
    },
    {
      "epoch": 0.67,
      "grad_norm": 3.092927765677092,
      "learning_rate": 1.926429875908269e-05,
      "loss": 0.2585,
      "step": 5700
    },
    {
      "epoch": 0.67,
      "grad_norm": 4.029661826813332,
      "learning_rate": 1.9256997777743437e-05,
      "loss": 0.2467,
      "step": 5720
    },
    {
      "epoch": 0.67,
      "grad_norm": 4.642691104141709,
      "learning_rate": 1.924966214576341e-05,
      "loss": 0.2681,
      "step": 5740
    },
    {
      "epoch": 0.67,
      "grad_norm": 7.315047122076067,
      "learning_rate": 1.9242291890604186e-05,
      "loss": 0.2508,
      "step": 5760
    },
    {
      "epoch": 0.68,
      "grad_norm": 3.540549397482083,
      "learning_rate": 1.923488703985696e-05,
      "loss": 0.2592,
      "step": 5780
    },
    {
      "epoch": 0.68,
      "grad_norm": 3.475018859373003,
      "learning_rate": 1.9227447621242458e-05,
      "loss": 0.2422,
      "step": 5800
    },
    {
      "epoch": 0.68,
      "grad_norm": 3.8676644079356355,
      "learning_rate": 1.921997366261078e-05,
      "loss": 0.2603,
      "step": 5820
    },
    {
      "epoch": 0.68,
      "grad_norm": 3.5885966628483,
      "learning_rate": 1.9212465191941346e-05,
      "loss": 0.2542,
      "step": 5840
    },
    {
      "epoch": 0.69,
      "grad_norm": 3.6495890137770886,
      "learning_rate": 1.920492223734279e-05,
      "loss": 0.2374,
      "step": 5860
    },
    {
      "epoch": 0.69,
      "grad_norm": 4.323488251305847,
      "learning_rate": 1.9197344827052813e-05,
      "loss": 0.2687,
      "step": 5880
    },
    {
      "epoch": 0.69,
      "grad_norm": 5.051497760408599,
      "learning_rate": 1.9189732989438123e-05,
      "loss": 0.2678,
      "step": 5900
    },
    {
      "epoch": 0.69,
      "grad_norm": 3.5559342765441495,
      "learning_rate": 1.9182086752994295e-05,
      "loss": 0.2726,
      "step": 5920
    },
    {
      "epoch": 0.7,
      "grad_norm": 3.029184640641301,
      "learning_rate": 1.9174406146345694e-05,
      "loss": 0.246,
      "step": 5940
    },
    {
      "epoch": 0.7,
      "grad_norm": 3.4792605785250994,
      "learning_rate": 1.9166691198245343e-05,
      "loss": 0.2843,
      "step": 5960
    },
    {
      "epoch": 0.7,
      "grad_norm": 3.0247490770922,
      "learning_rate": 1.915894193757483e-05,
      "loss": 0.2528,
      "step": 5980
    },
    {
      "epoch": 0.7,
      "grad_norm": 3.7727567007247855,
      "learning_rate": 1.915115839334419e-05,
      "loss": 0.2442,
      "step": 6000
    },
    {
      "epoch": 0.7,
      "grad_norm": 2.31881357193246,
      "learning_rate": 1.9143340594691806e-05,
      "loss": 0.2561,
      "step": 6020
    },
    {
      "epoch": 0.71,
      "grad_norm": 3.138315746753083,
      "learning_rate": 1.913548857088429e-05,
      "loss": 0.2636,
      "step": 6040
    },
    {
      "epoch": 0.71,
      "grad_norm": 8.010145906221986,
      "learning_rate": 1.9127602351316393e-05,
      "loss": 0.2876,
      "step": 6060
    },
    {
      "epoch": 0.71,
      "grad_norm": 3.867596475572881,
      "learning_rate": 1.9119681965510858e-05,
      "loss": 0.2659,
      "step": 6080
    },
    {
      "epoch": 0.71,
      "grad_norm": 3.3587166074477777,
      "learning_rate": 1.9111727443118357e-05,
      "loss": 0.2634,
      "step": 6100
    },
    {
      "epoch": 0.72,
      "grad_norm": 2.89895288044486,
      "learning_rate": 1.9103738813917336e-05,
      "loss": 0.2533,
      "step": 6120
    },
    {
      "epoch": 0.72,
      "grad_norm": 4.037752097559083,
      "learning_rate": 1.909571610781393e-05,
      "loss": 0.2485,
      "step": 6140
    },
    {
      "epoch": 0.72,
      "grad_norm": 3.763608653829743,
      "learning_rate": 1.908765935484185e-05,
      "loss": 0.259,
      "step": 6160
    },
    {
      "epoch": 0.72,
      "grad_norm": 3.161229351032073,
      "learning_rate": 1.9079568585162248e-05,
      "loss": 0.2508,
      "step": 6180
    },
    {
      "epoch": 0.73,
      "grad_norm": 2.725004696623228,
      "learning_rate": 1.9071443829063644e-05,
      "loss": 0.2524,
      "step": 6200
    },
    {
      "epoch": 0.73,
      "grad_norm": 3.7116897623627985,
      "learning_rate": 1.906328511696177e-05,
      "loss": 0.2598,
      "step": 6220
    },
    {
      "epoch": 0.73,
      "grad_norm": 2.9004110110815544,
      "learning_rate": 1.9055092479399474e-05,
      "loss": 0.2553,
      "step": 6240
    },
    {
      "epoch": 0.73,
      "grad_norm": 4.311030925472568,
      "learning_rate": 1.9046865947046627e-05,
      "loss": 0.2678,
      "step": 6260
    },
    {
      "epoch": 0.73,
      "grad_norm": 3.5377476420175977,
      "learning_rate": 1.903860555069997e-05,
      "loss": 0.2379,
      "step": 6280
    },
    {
      "epoch": 0.74,
      "grad_norm": 3.58934533910836,
      "learning_rate": 1.9030311321283022e-05,
      "loss": 0.2608,
      "step": 6300
    },
    {
      "epoch": 0.74,
      "grad_norm": 3.4612535457241624,
      "learning_rate": 1.9021983289845956e-05,
      "loss": 0.246,
      "step": 6320
    },
    {
      "epoch": 0.74,
      "grad_norm": 3.9763337015158258,
      "learning_rate": 1.9013621487565498e-05,
      "loss": 0.238,
      "step": 6340
    },
    {
      "epoch": 0.74,
      "grad_norm": 3.05488683337997,
      "learning_rate": 1.9005225945744782e-05,
      "loss": 0.2472,
      "step": 6360
    },
    {
      "epoch": 0.75,
      "grad_norm": 3.2350295137123313,
      "learning_rate": 1.8996796695813256e-05,
      "loss": 0.261,
      "step": 6380
    },
    {
      "epoch": 0.75,
      "grad_norm": 3.2500906711548123,
      "learning_rate": 1.8988333769326564e-05,
      "loss": 0.2641,
      "step": 6400
    },
    {
      "epoch": 0.75,
      "grad_norm": 2.815087548041305,
      "learning_rate": 1.8979837197966415e-05,
      "loss": 0.2583,
      "step": 6420
    },
    {
      "epoch": 0.75,
      "grad_norm": 3.7397965536014257,
      "learning_rate": 1.8971307013540464e-05,
      "loss": 0.2485,
      "step": 6440
    },
    {
      "epoch": 0.76,
      "grad_norm": 2.7592373806083423,
      "learning_rate": 1.8962743247982205e-05,
      "loss": 0.2641,
      "step": 6460
    },
    {
      "epoch": 0.76,
      "grad_norm": 2.818080452105616,
      "learning_rate": 1.895414593335085e-05,
      "loss": 0.2554,
      "step": 6480
    },
    {
      "epoch": 0.76,
      "grad_norm": 3.627385505536174,
      "learning_rate": 1.8945515101831203e-05,
      "loss": 0.2693,
      "step": 6500
    },
    {
      "epoch": 0.76,
      "grad_norm": 3.255509767944512,
      "learning_rate": 1.893685078573354e-05,
      "loss": 0.2459,
      "step": 6520
    },
    {
      "epoch": 0.77,
      "grad_norm": 4.135182845698409,
      "learning_rate": 1.8928153017493488e-05,
      "loss": 0.2499,
      "step": 6540
    },
    {
      "epoch": 0.77,
      "grad_norm": 5.892816677202276,
      "learning_rate": 1.8919421829671904e-05,
      "loss": 0.2786,
      "step": 6560
    },
    {
      "epoch": 0.77,
      "grad_norm": 3.479892875732629,
      "learning_rate": 1.891065725495476e-05,
      "loss": 0.265,
      "step": 6580
    },
    {
      "epoch": 0.77,
      "grad_norm": 3.8343809741781216,
      "learning_rate": 1.890185932615301e-05,
      "loss": 0.2527,
      "step": 6600
    },
    {
      "epoch": 0.77,
      "grad_norm": 3.1643033159708107,
      "learning_rate": 1.8893028076202472e-05,
      "loss": 0.2741,
      "step": 6620
    },
    {
      "epoch": 0.78,
      "grad_norm": 4.477352585232187,
      "learning_rate": 1.8884163538163708e-05,
      "loss": 0.2525,
      "step": 6640
    },
    {
      "epoch": 0.78,
      "grad_norm": 3.529760400999529,
      "learning_rate": 1.8875265745221894e-05,
      "loss": 0.2643,
      "step": 6660
    },
    {
      "epoch": 0.78,
      "grad_norm": 2.3110189076619525,
      "learning_rate": 1.8866334730686703e-05,
      "loss": 0.2721,
      "step": 6680
    },
    {
      "epoch": 0.78,
      "grad_norm": 3.685226774238388,
      "learning_rate": 1.8857370527992165e-05,
      "loss": 0.2549,
      "step": 6700
    },
    {
      "epoch": 0.79,
      "grad_norm": 3.8074891978039864,
      "learning_rate": 1.8848373170696567e-05,
      "loss": 0.2418,
      "step": 6720
    },
    {
      "epoch": 0.79,
      "grad_norm": 7.507877790185021,
      "learning_rate": 1.8839342692482302e-05,
      "loss": 0.2451,
      "step": 6740
    },
    {
      "epoch": 0.79,
      "grad_norm": 2.7359733051983186,
      "learning_rate": 1.8830279127155764e-05,
      "loss": 0.2484,
      "step": 6760
    },
    {
      "epoch": 0.79,
      "grad_norm": 2.4635290638854657,
      "learning_rate": 1.88211825086472e-05,
      "loss": 0.2407,
      "step": 6780
    },
    {
      "epoch": 0.8,
      "grad_norm": 3.040130192327906,
      "learning_rate": 1.8812052871010605e-05,
      "loss": 0.2544,
      "step": 6800
    },
    {
      "epoch": 0.8,
      "grad_norm": 5.610778632978326,
      "learning_rate": 1.8802890248423574e-05,
      "loss": 0.2397,
      "step": 6820
    },
    {
      "epoch": 0.8,
      "grad_norm": 4.156547822535594,
      "learning_rate": 1.8793694675187197e-05,
      "loss": 0.2635,
      "step": 6840
    },
    {
      "epoch": 0.8,
      "grad_norm": 4.057314806159278,
      "learning_rate": 1.8784466185725902e-05,
      "loss": 0.2598,
      "step": 6860
    },
    {
      "epoch": 0.81,
      "grad_norm": 8.314874066232434,
      "learning_rate": 1.877520481458736e-05,
      "loss": 0.2706,
      "step": 6880
    },
    {
      "epoch": 0.81,
      "grad_norm": 3.7767818313058195,
      "learning_rate": 1.8765910596442316e-05,
      "loss": 0.2483,
      "step": 6900
    },
    {
      "epoch": 0.81,
      "grad_norm": 3.53905010431742,
      "learning_rate": 1.8756583566084502e-05,
      "loss": 0.2683,
      "step": 6920
    },
    {
      "epoch": 0.81,
      "grad_norm": 3.2053331160944625,
      "learning_rate": 1.8747223758430467e-05,
      "loss": 0.2567,
      "step": 6940
    },
    {
      "epoch": 0.81,
      "grad_norm": 3.655977418308093,
      "learning_rate": 1.8737831208519478e-05,
      "loss": 0.2433,
      "step": 6960
    },
    {
      "epoch": 0.82,
      "grad_norm": 3.334646935711793,
      "learning_rate": 1.8728405951513376e-05,
      "loss": 0.2442,
      "step": 6980
    },
    {
      "epoch": 0.82,
      "grad_norm": 3.3678493336120505,
      "learning_rate": 1.8718948022696427e-05,
      "loss": 0.2634,
      "step": 7000
    },
    {
      "epoch": 0.82,
      "grad_norm": 2.9941614875638956,
      "learning_rate": 1.8709457457475224e-05,
      "loss": 0.2719,
      "step": 7020
    },
    {
      "epoch": 0.82,
      "grad_norm": 3.3212764283242495,
      "learning_rate": 1.8699934291378534e-05,
      "loss": 0.2616,
      "step": 7040
    },
    {
      "epoch": 0.83,
      "grad_norm": 3.7449113015848963,
      "learning_rate": 1.8690378560057165e-05,
      "loss": 0.251,
      "step": 7060
    },
    {
      "epoch": 0.83,
      "grad_norm": 2.588981565454773,
      "learning_rate": 1.868079029928383e-05,
      "loss": 0.2345,
      "step": 7080
    },
    {
      "epoch": 0.83,
      "grad_norm": 4.188807496871825,
      "learning_rate": 1.8671169544953033e-05,
      "loss": 0.2537,
      "step": 7100
    },
    {
      "epoch": 0.83,
      "grad_norm": 4.66518430781191,
      "learning_rate": 1.866151633308091e-05,
      "loss": 0.2397,
      "step": 7120
    },
    {
      "epoch": 0.84,
      "grad_norm": 4.197676860816234,
      "learning_rate": 1.8651830699805108e-05,
      "loss": 0.2446,
      "step": 7140
    },
    {
      "epoch": 0.84,
      "grad_norm": 3.6434672643458543,
      "learning_rate": 1.8642112681384645e-05,
      "loss": 0.2402,
      "step": 7160
    },
    {
      "epoch": 0.84,
      "grad_norm": 3.950387723218738,
      "learning_rate": 1.8632362314199778e-05,
      "loss": 0.2536,
      "step": 7180
    },
    {
      "epoch": 0.84,
      "grad_norm": 9.658211616844094,
      "learning_rate": 1.8622579634751864e-05,
      "loss": 0.2541,
      "step": 7200
    },
    {
      "epoch": 0.84,
      "grad_norm": 3.799337836345952,
      "learning_rate": 1.8612764679663222e-05,
      "loss": 0.2586,
      "step": 7220
    },
    {
      "epoch": 0.85,
      "grad_norm": 3.737906856782671,
      "learning_rate": 1.8602917485677e-05,
      "loss": 0.2539,
      "step": 7240
    },
    {
      "epoch": 0.85,
      "grad_norm": 3.2570157254536016,
      "learning_rate": 1.859303808965703e-05,
      "loss": 0.2533,
      "step": 7260
    },
    {
      "epoch": 0.85,
      "grad_norm": 2.8736525985585226,
      "learning_rate": 1.85831265285877e-05,
      "loss": 0.2532,
      "step": 7280
    },
    {
      "epoch": 0.85,
      "grad_norm": 3.5447389329516503,
      "learning_rate": 1.857318283957381e-05,
      "loss": 0.2387,
      "step": 7300
    },
    {
      "epoch": 0.86,
      "grad_norm": 2.7468496831393403,
      "learning_rate": 1.8563207059840436e-05,
      "loss": 0.2455,
      "step": 7320
    },
    {
      "epoch": 0.86,
      "grad_norm": 3.483830293921577,
      "learning_rate": 1.8553199226732784e-05,
      "loss": 0.2456,
      "step": 7340
    },
    {
      "epoch": 0.86,
      "grad_norm": 3.3132796899404635,
      "learning_rate": 1.8543159377716054e-05,
      "loss": 0.2564,
      "step": 7360
    },
    {
      "epoch": 0.86,
      "grad_norm": 3.3795508326649575,
      "learning_rate": 1.8533087550375302e-05,
      "loss": 0.2527,
      "step": 7380
    },
    {
      "epoch": 0.87,
      "grad_norm": 4.001941448174156,
      "learning_rate": 1.8522983782415306e-05,
      "loss": 0.2433,
      "step": 7400
    },
    {
      "epoch": 0.87,
      "grad_norm": 3.4205382361927117,
      "learning_rate": 1.85128481116604e-05,
      "loss": 0.2538,
      "step": 7420
    },
    {
      "epoch": 0.87,
      "grad_norm": 3.896330422632185,
      "learning_rate": 1.8502680576054365e-05,
      "loss": 0.2478,
      "step": 7440
    },
    {
      "epoch": 0.87,
      "grad_norm": 5.020228094868969,
      "learning_rate": 1.849248121366026e-05,
      "loss": 0.2427,
      "step": 7460
    },
    {
      "epoch": 0.88,
      "grad_norm": 3.574276782956127,
      "learning_rate": 1.8482250062660293e-05,
      "loss": 0.2389,
      "step": 7480
    },
    {
      "epoch": 0.88,
      "grad_norm": 3.5309579103911184,
      "learning_rate": 1.847198716135568e-05,
      "loss": 0.2567,
      "step": 7500
    },
    {
      "epoch": 0.88,
      "grad_norm": 3.1760414458004,
      "learning_rate": 1.846169254816649e-05,
      "loss": 0.2291,
      "step": 7520
    },
    {
      "epoch": 0.88,
      "grad_norm": 4.214334621561301,
      "learning_rate": 1.8451366261631514e-05,
      "loss": 0.2624,
      "step": 7540
    },
    {
      "epoch": 0.88,
      "grad_norm": 2.977221959069712,
      "learning_rate": 1.8441008340408115e-05,
      "loss": 0.2425,
      "step": 7560
    },
    {
      "epoch": 0.89,
      "grad_norm": 3.78383188925913,
      "learning_rate": 1.8430618823272076e-05,
      "loss": 0.2437,
      "step": 7580
    },
    {
      "epoch": 0.89,
      "grad_norm": 3.2432934448936224,
      "learning_rate": 1.842019774911747e-05,
      "loss": 0.2628,
      "step": 7600
    },
    {
      "epoch": 0.89,
      "grad_norm": 3.05382414419798,
      "learning_rate": 1.8409745156956503e-05,
      "loss": 0.2332,
      "step": 7620
    },
    {
      "epoch": 0.89,
      "grad_norm": 3.098744070302099,
      "learning_rate": 1.839926108591937e-05,
      "loss": 0.2359,
      "step": 7640
    },
    {
      "epoch": 0.9,
      "grad_norm": 3.427098715766002,
      "learning_rate": 1.8388745575254117e-05,
      "loss": 0.2429,
      "step": 7660
    },
    {
      "epoch": 0.9,
      "grad_norm": 5.630711008711337,
      "learning_rate": 1.8378198664326474e-05,
      "loss": 0.2416,
      "step": 7680
    },
    {
      "epoch": 0.9,
      "grad_norm": 3.1144723181358622,
      "learning_rate": 1.836762039261974e-05,
      "loss": 0.2406,
      "step": 7700
    },
    {
      "epoch": 0.9,
      "grad_norm": 4.070301722184627,
      "learning_rate": 1.8357010799734588e-05,
      "loss": 0.234,
      "step": 7720
    },
    {
      "epoch": 0.91,
      "grad_norm": 3.255428182680105,
      "learning_rate": 1.834636992538898e-05,
      "loss": 0.2365,
      "step": 7740
    },
    {
      "epoch": 0.91,
      "grad_norm": 3.4823775104086576,
      "learning_rate": 1.8335697809417952e-05,
      "loss": 0.2376,
      "step": 7760
    },
    {
      "epoch": 0.91,
      "grad_norm": 5.528269163918348,
      "learning_rate": 1.8324994491773507e-05,
      "loss": 0.2679,
      "step": 7780
    },
    {
      "epoch": 0.91,
      "grad_norm": 2.7221100312640467,
      "learning_rate": 1.8314260012524462e-05,
      "loss": 0.2544,
      "step": 7800
    },
    {
      "epoch": 0.92,
      "grad_norm": 3.224515621271091,
      "learning_rate": 1.8303494411856278e-05,
      "loss": 0.2586,
      "step": 7820
    },
    {
      "epoch": 0.92,
      "grad_norm": 2.798714485709818,
      "learning_rate": 1.829269773007092e-05,
      "loss": 0.2277,
      "step": 7840
    },
    {
      "epoch": 0.92,
      "grad_norm": 3.890849401904209,
      "learning_rate": 1.8281870007586735e-05,
      "loss": 0.2565,
      "step": 7860
    },
    {
      "epoch": 0.92,
      "grad_norm": 3.701194549732654,
      "learning_rate": 1.8271011284938233e-05,
      "loss": 0.2307,
      "step": 7880
    },
    {
      "epoch": 0.92,
      "grad_norm": 3.9057568048025875,
      "learning_rate": 1.8260121602776003e-05,
      "loss": 0.2464,
      "step": 7900
    },
    {
      "epoch": 0.93,
      "grad_norm": 3.397196786583742,
      "learning_rate": 1.8249201001866533e-05,
      "loss": 0.25,
      "step": 7920
    },
    {
      "epoch": 0.93,
      "grad_norm": 3.6136942859367496,
      "learning_rate": 1.8238249523092046e-05,
      "loss": 0.2398,
      "step": 7940
    },
    {
      "epoch": 0.93,
      "grad_norm": 3.6009293204428765,
      "learning_rate": 1.8227267207450363e-05,
      "loss": 0.2356,
      "step": 7960
    },
    {
      "epoch": 0.93,
      "grad_norm": 3.1462270393488585,
      "learning_rate": 1.8216254096054747e-05,
      "loss": 0.2292,
      "step": 7980
    },
    {
      "epoch": 0.94,
      "grad_norm": 3.453766085807792,
      "learning_rate": 1.8205210230133756e-05,
      "loss": 0.2532,
      "step": 8000
    },
    {
      "epoch": 0.94,
      "grad_norm": 4.718419953986993,
      "learning_rate": 1.8194135651031063e-05,
      "loss": 0.2379,
      "step": 8020
    },
    {
      "epoch": 0.94,
      "grad_norm": 4.074122309800583,
      "learning_rate": 1.8183030400205335e-05,
      "loss": 0.247,
      "step": 8040
    },
    {
      "epoch": 0.94,
      "grad_norm": 3.597629476607082,
      "learning_rate": 1.817189451923005e-05,
      "loss": 0.2563,
      "step": 8060
    },
    {
      "epoch": 0.95,
      "grad_norm": 3.2550443969136436,
      "learning_rate": 1.8160728049793357e-05,
      "loss": 0.2425,
      "step": 8080
    },
    {
      "epoch": 0.95,
      "grad_norm": 2.6027515890006505,
      "learning_rate": 1.8149531033697913e-05,
      "loss": 0.2329,
      "step": 8100
    },
    {
      "epoch": 0.95,
      "grad_norm": 3.705349797095471,
      "learning_rate": 1.8138303512860737e-05,
      "loss": 0.2521,
      "step": 8120
    },
    {
      "epoch": 0.95,
      "grad_norm": 3.510055266811937,
      "learning_rate": 1.812704552931303e-05,
      "loss": 0.2413,
      "step": 8140
    },
    {
      "epoch": 0.95,
      "grad_norm": 3.2863076604207997,
      "learning_rate": 1.8115757125200054e-05,
      "loss": 0.2471,
      "step": 8160
    },
    {
      "epoch": 0.96,
      "grad_norm": 3.174018473859266,
      "learning_rate": 1.810443834278094e-05,
      "loss": 0.2429,
      "step": 8180
    },
    {
      "epoch": 0.96,
      "grad_norm": 4.78684161965386,
      "learning_rate": 1.8093089224428533e-05,
      "loss": 0.21,
      "step": 8200
    },
    {
      "epoch": 0.96,
      "grad_norm": 3.778736256263061,
      "learning_rate": 1.8081709812629265e-05,
      "loss": 0.2438,
      "step": 8220
    },
    {
      "epoch": 0.96,
      "grad_norm": 2.011964890319697,
      "learning_rate": 1.807030014998296e-05,
      "loss": 0.2235,
      "step": 8240
    },
    {
      "epoch": 0.97,
      "grad_norm": 2.469231474076661,
      "learning_rate": 1.8058860279202698e-05,
      "loss": 0.2361,
      "step": 8260
    },
    {
      "epoch": 0.97,
      "grad_norm": 4.194473608952758,
      "learning_rate": 1.804739024311463e-05,
      "loss": 0.2485,
      "step": 8280
    },
    {
      "epoch": 0.97,
      "grad_norm": 3.648802255045255,
      "learning_rate": 1.8035890084657854e-05,
      "loss": 0.2448,
      "step": 8300
    },
    {
      "epoch": 0.97,
      "grad_norm": 2.8256525011924443,
      "learning_rate": 1.8024359846884213e-05,
      "loss": 0.2688,
      "step": 8320
    },
    {
      "epoch": 0.98,
      "grad_norm": 4.409589867929299,
      "learning_rate": 1.8012799572958173e-05,
      "loss": 0.2405,
      "step": 8340
    },
    {
      "epoch": 0.98,
      "grad_norm": 5.024619525530542,
      "learning_rate": 1.8001209306156634e-05,
      "loss": 0.2284,
      "step": 8360
    },
    {
      "epoch": 0.98,
      "grad_norm": 3.5087106485202026,
      "learning_rate": 1.7989589089868774e-05,
      "loss": 0.2298,
      "step": 8380
    },
    {
      "epoch": 0.98,
      "grad_norm": 4.40165070000877,
      "learning_rate": 1.7977938967595897e-05,
      "loss": 0.2507,
      "step": 8400
    },
    {
      "epoch": 0.99,
      "grad_norm": 3.9660729217673705,
      "learning_rate": 1.7966258982951255e-05,
      "loss": 0.2459,
      "step": 8420
    },
    {
      "epoch": 0.99,
      "grad_norm": 4.665706172372416,
      "learning_rate": 1.7954549179659903e-05,
      "loss": 0.2458,
      "step": 8440
    },
    {
      "epoch": 0.99,
      "grad_norm": 3.2908069759526333,
      "learning_rate": 1.794280960155851e-05,
      "loss": 0.2251,
      "step": 8460
    },
    {
      "epoch": 0.99,
      "grad_norm": 6.7504752662905725,
      "learning_rate": 1.7931040292595227e-05,
      "loss": 0.254,
      "step": 8480
    },
    {
      "epoch": 0.99,
      "grad_norm": 2.2047073417435787,
      "learning_rate": 1.791924129682949e-05,
      "loss": 0.2456,
      "step": 8500
    },
    {
      "epoch": 1.0,
      "grad_norm": 3.9048911210632236,
      "learning_rate": 1.7907412658431877e-05,
      "loss": 0.2307,
      "step": 8520
    },
    {
      "epoch": 1.0,
      "grad_norm": 3.33290909610624,
      "learning_rate": 1.7895554421683938e-05,
      "loss": 0.2514,
      "step": 8540
    },
    {
      "epoch": 1.0,
      "grad_norm": 3.7515500997507605,
      "learning_rate": 1.7883666630978024e-05,
      "loss": 0.2189,
      "step": 8560
    },
    {
      "epoch": 1.0,
      "grad_norm": 2.495653761409699,
      "learning_rate": 1.7871749330817114e-05,
      "loss": 0.216,
      "step": 8580
    },
    {
      "epoch": 1.01,
      "grad_norm": 3.1012959017387556,
      "learning_rate": 1.785980256581468e-05,
      "loss": 0.2183,
      "step": 8600
    },
    {
      "epoch": 1.01,
      "grad_norm": 2.7851575340492745,
      "learning_rate": 1.7847826380694482e-05,
      "loss": 0.2086,
      "step": 8620
    },
    {
      "epoch": 1.01,
      "grad_norm": 5.673400169424141,
      "learning_rate": 1.783582082029042e-05,
      "loss": 0.2216,
      "step": 8640
    },
    {
      "epoch": 1.01,
      "grad_norm": 3.9347827557672113,
      "learning_rate": 1.7823785929546363e-05,
      "loss": 0.2351,
      "step": 8660
    },
    {
      "epoch": 1.02,
      "grad_norm": 4.263640336790709,
      "learning_rate": 1.781172175351599e-05,
      "loss": 0.2237,
      "step": 8680
    },
    {
      "epoch": 1.02,
      "grad_norm": 2.8181611623874034,
      "learning_rate": 1.779962833736259e-05,
      "loss": 0.2078,
      "step": 8700
    },
    {
      "epoch": 1.02,
      "grad_norm": 3.826255497546828,
      "learning_rate": 1.7787505726358937e-05,
      "loss": 0.2224,
      "step": 8720
    },
    {
      "epoch": 1.02,
      "grad_norm": 3.2239814391820603,
      "learning_rate": 1.777535396588709e-05,
      "loss": 0.2196,
      "step": 8740
    },
    {
      "epoch": 1.03,
      "grad_norm": 3.4517898394902753,
      "learning_rate": 1.7763173101438232e-05,
      "loss": 0.2056,
      "step": 8760
    },
    {
      "epoch": 1.03,
      "grad_norm": 4.604513417070624,
      "learning_rate": 1.7750963178612498e-05,
      "loss": 0.2434,
      "step": 8780
    },
    {
      "epoch": 1.03,
      "grad_norm": 3.835720949606093,
      "learning_rate": 1.7738724243118807e-05,
      "loss": 0.2282,
      "step": 8800
    },
    {
      "epoch": 1.03,
      "grad_norm": 3.437975625165332,
      "learning_rate": 1.7726456340774684e-05,
      "loss": 0.25,
      "step": 8820
    },
    {
      "epoch": 1.03,
      "grad_norm": 3.362299298390976,
      "learning_rate": 1.771415951750611e-05,
      "loss": 0.2301,
      "step": 8840
    },
    {
      "epoch": 1.04,
      "grad_norm": 3.469988077896261,
      "learning_rate": 1.7701833819347317e-05,
      "loss": 0.2268,
      "step": 8860
    },
    {
      "epoch": 1.04,
      "grad_norm": 3.3257530965726834,
      "learning_rate": 1.7689479292440635e-05,
      "loss": 0.2362,
      "step": 8880
    },
    {
      "epoch": 1.04,
      "grad_norm": 4.357144130197379,
      "learning_rate": 1.767709598303633e-05,
      "loss": 0.2179,
      "step": 8900
    },
    {
      "epoch": 1.04,
      "grad_norm": 2.833662313173985,
      "learning_rate": 1.7664683937492394e-05,
      "loss": 0.2082,
      "step": 8920
    },
    {
      "epoch": 1.05,
      "grad_norm": 3.6248674039596347,
      "learning_rate": 1.7652243202274427e-05,
      "loss": 0.2399,
      "step": 8940
    },
    {
      "epoch": 1.05,
      "grad_norm": 3.374713885577454,
      "learning_rate": 1.7639773823955405e-05,
      "loss": 0.2303,
      "step": 8960
    },
    {
      "epoch": 1.05,
      "grad_norm": 4.623143210009699,
      "learning_rate": 1.7627275849215543e-05,
      "loss": 0.2197,
      "step": 8980
    },
    {
      "epoch": 1.05,
      "grad_norm": 2.7733450860758713,
      "learning_rate": 1.7614749324842106e-05,
      "loss": 0.2419,
      "step": 9000
    },
    {
      "epoch": 1.06,
      "grad_norm": 4.136130142481253,
      "learning_rate": 1.7602194297729238e-05,
      "loss": 0.2315,
      "step": 9020
    },
    {
      "epoch": 1.06,
      "grad_norm": 2.5713300459015476,
      "learning_rate": 1.758961081487779e-05,
      "loss": 0.2285,
      "step": 9040
    },
    {
      "epoch": 1.06,
      "grad_norm": 2.90938161486602,
      "learning_rate": 1.7576998923395128e-05,
      "loss": 0.2357,
      "step": 9060
    },
    {
      "epoch": 1.06,
      "grad_norm": 3.2699691862001896,
      "learning_rate": 1.756435867049498e-05,
      "loss": 0.2161,
      "step": 9080
    },
    {
      "epoch": 1.06,
      "grad_norm": 3.353066842604925,
      "learning_rate": 1.7551690103497248e-05,
      "loss": 0.2379,
      "step": 9100
    },
    {
      "epoch": 1.07,
      "grad_norm": 3.1865972100619815,
      "learning_rate": 1.7538993269827817e-05,
      "loss": 0.2324,
      "step": 9120
    },
    {
      "epoch": 1.07,
      "grad_norm": 4.115179219937697,
      "learning_rate": 1.7526268217018404e-05,
      "loss": 0.2165,
      "step": 9140
    },
    {
      "epoch": 1.07,
      "grad_norm": 3.0706453021663553,
      "learning_rate": 1.751351499270636e-05,
      "loss": 0.2128,
      "step": 9160
    },
    {
      "epoch": 1.07,
      "grad_norm": 3.0307089725880774,
      "learning_rate": 1.7500733644634503e-05,
      "loss": 0.2153,
      "step": 9180
    },
    {
      "epoch": 1.08,
      "grad_norm": 3.0414096427563435,
      "learning_rate": 1.748792422065093e-05,
      "loss": 0.2159,
      "step": 9200
    },
    {
      "epoch": 1.08,
      "grad_norm": 3.581867909009255,
      "learning_rate": 1.7475086768708844e-05,
      "loss": 0.2108,
      "step": 9220
    },
    {
      "epoch": 1.08,
      "grad_norm": 3.3902710057413707,
      "learning_rate": 1.7462221336866373e-05,
      "loss": 0.2171,
      "step": 9240
    },
    {
      "epoch": 1.08,
      "grad_norm": 4.186442227351193,
      "learning_rate": 1.74493279732864e-05,
      "loss": 0.213,
      "step": 9260
    },
    {
      "epoch": 1.09,
      "grad_norm": 2.111492196482593,
      "learning_rate": 1.743640672623636e-05,
      "loss": 0.231,
      "step": 9280
    },
    {
      "epoch": 1.09,
      "grad_norm": 3.3151227536495744,
      "learning_rate": 1.7423457644088067e-05,
      "loss": 0.2255,
      "step": 9300
    },
    {
      "epoch": 1.09,
      "grad_norm": 2.979480504575997,
      "learning_rate": 1.741048077531756e-05,
      "loss": 0.2193,
      "step": 9320
    },
    {
      "epoch": 1.09,
      "grad_norm": 8.144688622580304,
      "learning_rate": 1.7397476168504877e-05,
      "loss": 0.2213,
      "step": 9340
    },
    {
      "epoch": 1.1,
      "grad_norm": 2.3573272104543395,
      "learning_rate": 1.738444387233392e-05,
      "loss": 0.23,
      "step": 9360
    },
    {
      "epoch": 1.1,
      "grad_norm": 5.626318713423072,
      "learning_rate": 1.737138393559222e-05,
      "loss": 0.2231,
      "step": 9380
    },
    {
      "epoch": 1.1,
      "grad_norm": 3.659177993436024,
      "learning_rate": 1.7358296407170803e-05,
      "loss": 0.2201,
      "step": 9400
    },
    {
      "epoch": 1.1,
      "grad_norm": 4.1924943889109505,
      "learning_rate": 1.734518133606398e-05,
      "loss": 0.2211,
      "step": 9420
    },
    {
      "epoch": 1.1,
      "grad_norm": 3.1828991465696626,
      "learning_rate": 1.7332038771369178e-05,
      "loss": 0.2178,
      "step": 9440
    },
    {
      "epoch": 1.11,
      "grad_norm": 3.210581516107296,
      "learning_rate": 1.7318868762286735e-05,
      "loss": 0.2252,
      "step": 9460
    },
    {
      "epoch": 1.11,
      "grad_norm": 3.1129128729970272,
      "learning_rate": 1.7305671358119743e-05,
      "loss": 0.2218,
      "step": 9480
    },
    {
      "epoch": 1.11,
      "grad_norm": 3.041995792784782,
      "learning_rate": 1.7292446608273838e-05,
      "loss": 0.2089,
      "step": 9500
    },
    {
      "epoch": 1.11,
      "grad_norm": 4.604556704379658,
      "learning_rate": 1.7279194562257034e-05,
      "loss": 0.2104,
      "step": 9520
    },
    {
      "epoch": 1.12,
      "grad_norm": 2.4749100524278944,
      "learning_rate": 1.7265915269679533e-05,
      "loss": 0.2077,
      "step": 9540
    },
    {
      "epoch": 1.12,
      "grad_norm": 3.256783082638953,
      "learning_rate": 1.7252608780253527e-05,
      "loss": 0.2222,
      "step": 9560
    },
    {
      "epoch": 1.12,
      "grad_norm": 3.250990203251271,
      "learning_rate": 1.723927514379304e-05,
      "loss": 0.238,
      "step": 9580
    },
    {
      "epoch": 1.12,
      "grad_norm": 4.17869256557804,
      "learning_rate": 1.7225914410213695e-05,
      "loss": 0.2264,
      "step": 9600
    },
    {
      "epoch": 1.13,
      "grad_norm": 3.001794596192333,
      "learning_rate": 1.7212526629532585e-05,
      "loss": 0.2156,
      "step": 9620
    },
    {
      "epoch": 1.13,
      "grad_norm": 4.51471973043563,
      "learning_rate": 1.7199111851868038e-05,
      "loss": 0.2212,
      "step": 9640
    },
    {
      "epoch": 1.13,
      "grad_norm": 3.343622223025876,
      "learning_rate": 1.7185670127439453e-05,
      "loss": 0.2292,
      "step": 9660
    },
    {
      "epoch": 1.13,
      "grad_norm": 5.351871498264467,
      "learning_rate": 1.7172201506567103e-05,
      "loss": 0.233,
      "step": 9680
    },
    {
      "epoch": 1.14,
      "grad_norm": 2.480488358067184,
      "learning_rate": 1.7158706039671962e-05,
      "loss": 0.2253,
      "step": 9700
    },
    {
      "epoch": 1.14,
      "grad_norm": 9.126953333979776,
      "learning_rate": 1.7145183777275484e-05,
      "loss": 0.2216,
      "step": 9720
    },
    {
      "epoch": 1.14,
      "grad_norm": 3.0992364496831994,
      "learning_rate": 1.7131634769999458e-05,
      "loss": 0.2339,
      "step": 9740
    },
    {
      "epoch": 1.14,
      "grad_norm": 3.536084536918467,
      "learning_rate": 1.7118059068565775e-05,
      "loss": 0.2244,
      "step": 9760
    },
    {
      "epoch": 1.14,
      "grad_norm": 2.48499093727593,
      "learning_rate": 1.710445672379627e-05,
      "loss": 0.221,
      "step": 9780
    },
    {
      "epoch": 1.15,
      "grad_norm": 3.7310914483561097,
      "learning_rate": 1.7090827786612515e-05,
      "loss": 0.2356,
      "step": 9800
    },
    {
      "epoch": 1.15,
      "grad_norm": 3.0512655853035975,
      "learning_rate": 1.7077172308035627e-05,
      "loss": 0.2157,
      "step": 9820
    },
    {
      "epoch": 1.15,
      "grad_norm": 2.6139447976363868,
      "learning_rate": 1.70634903391861e-05,
      "loss": 0.2251,
      "step": 9840
    },
    {
      "epoch": 1.15,
      "grad_norm": 2.7800692505753135,
      "learning_rate": 1.7049781931283576e-05,
      "loss": 0.2385,
      "step": 9860
    },
    {
      "epoch": 1.16,
      "grad_norm": 3.8229638009279614,
      "learning_rate": 1.7036047135646695e-05,
      "loss": 0.2163,
      "step": 9880
    },
    {
      "epoch": 1.16,
      "grad_norm": 3.4670030473465205,
      "learning_rate": 1.7022286003692863e-05,
      "loss": 0.2172,
      "step": 9900
    },
    {
      "epoch": 1.16,
      "grad_norm": 3.1067639606886908,
      "learning_rate": 1.700849858693809e-05,
      "loss": 0.228,
      "step": 9920
    },
    {
      "epoch": 1.16,
      "grad_norm": 2.775258672531786,
      "learning_rate": 1.6994684936996782e-05,
      "loss": 0.2174,
      "step": 9940
    },
    {
      "epoch": 1.17,
      "grad_norm": 3.0648920487627045,
      "learning_rate": 1.698084510558155e-05,
      "loss": 0.2148,
      "step": 9960
    },
    {
      "epoch": 1.17,
      "grad_norm": 3.2600668007495064,
      "learning_rate": 1.696697914450302e-05,
      "loss": 0.2133,
      "step": 9980
    },
    {
      "epoch": 1.17,
      "grad_norm": 5.51464299005287,
      "learning_rate": 1.6953087105669627e-05,
      "loss": 0.2297,
      "step": 10000
    },
    {
      "epoch": 1.17,
      "grad_norm": 3.872665563483291,
      "learning_rate": 1.693916904108745e-05,
      "loss": 0.2277,
      "step": 10020
    },
    {
      "epoch": 1.17,
      "grad_norm": 3.0527595230989917,
      "learning_rate": 1.692522500285998e-05,
      "loss": 0.2084,
      "step": 10040
    },
    {
      "epoch": 1.18,
      "grad_norm": 3.009517988727613,
      "learning_rate": 1.6911255043187952e-05,
      "loss": 0.2105,
      "step": 10060
    },
    {
      "epoch": 1.18,
      "grad_norm": 4.461421265567992,
      "learning_rate": 1.6897259214369137e-05,
      "loss": 0.2156,
      "step": 10080
    },
    {
      "epoch": 1.18,
      "grad_norm": 3.2651682265027757,
      "learning_rate": 1.6883237568798148e-05,
      "loss": 0.2223,
      "step": 10100
    },
    {
      "epoch": 1.18,
      "grad_norm": 4.560070279232337,
      "learning_rate": 1.6869190158966248e-05,
      "loss": 0.2032,
      "step": 10120
    },
    {
      "epoch": 1.19,
      "grad_norm": 2.17671049536392,
      "learning_rate": 1.6855117037461147e-05,
      "loss": 0.2086,
      "step": 10140
    },
    {
      "epoch": 1.19,
      "grad_norm": 2.835463807449662,
      "learning_rate": 1.6841018256966817e-05,
      "loss": 0.2285,
      "step": 10160
    },
    {
      "epoch": 1.19,
      "grad_norm": 2.9456662236305045,
      "learning_rate": 1.682689387026328e-05,
      "loss": 0.2288,
      "step": 10180
    },
    {
      "epoch": 1.19,
      "grad_norm": 2.6149578560812046,
      "learning_rate": 1.681274393022641e-05,
      "loss": 0.2196,
      "step": 10200
    },
    {
      "epoch": 1.2,
      "grad_norm": 2.9578551501440025,
      "learning_rate": 1.6798568489827764e-05,
      "loss": 0.2233,
      "step": 10220
    },
    {
      "epoch": 1.2,
      "grad_norm": 3.191017752898972,
      "learning_rate": 1.6784367602134337e-05,
      "loss": 0.2093,
      "step": 10240
    },
    {
      "epoch": 1.2,
      "grad_norm": 3.353192695439166,
      "learning_rate": 1.6770141320308405e-05,
      "loss": 0.233,
      "step": 10260
    },
    {
      "epoch": 1.2,
      "grad_norm": 3.8431182590288397,
      "learning_rate": 1.6755889697607306e-05,
      "loss": 0.2206,
      "step": 10280
    },
    {
      "epoch": 1.21,
      "grad_norm": 3.479325951377326,
      "learning_rate": 1.6741612787383236e-05,
      "loss": 0.2205,
      "step": 10300
    },
    {
      "epoch": 1.21,
      "grad_norm": 2.5880779155399316,
      "learning_rate": 1.6727310643083066e-05,
      "loss": 0.2103,
      "step": 10320
    },
    {
      "epoch": 1.21,
      "grad_norm": 7.419706714519934,
      "learning_rate": 1.671298331824813e-05,
      "loss": 0.207,
      "step": 10340
    },
    {
      "epoch": 1.21,
      "grad_norm": 4.622811908173908,
      "learning_rate": 1.6698630866514022e-05,
      "loss": 0.2205,
      "step": 10360
    },
    {
      "epoch": 1.21,
      "grad_norm": 4.038977026265583,
      "learning_rate": 1.6684253341610414e-05,
      "loss": 0.2046,
      "step": 10380
    },
    {
      "epoch": 1.22,
      "grad_norm": 3.2506861329229912,
      "learning_rate": 1.6669850797360825e-05,
      "loss": 0.2161,
      "step": 10400
    },
    {
      "epoch": 1.22,
      "grad_norm": 3.502316934183813,
      "learning_rate": 1.665542328768245e-05,
      "loss": 0.2202,
      "step": 10420
    },
    {
      "epoch": 1.22,
      "grad_norm": 2.774170254901442,
      "learning_rate": 1.6640970866585937e-05,
      "loss": 0.2313,
      "step": 10440
    },
    {
      "epoch": 1.22,
      "grad_norm": 3.3567765943400345,
      "learning_rate": 1.6626493588175194e-05,
      "loss": 0.2118,
      "step": 10460
    },
    {
      "epoch": 1.23,
      "grad_norm": 3.147468966936314,
      "learning_rate": 1.6611991506647186e-05,
      "loss": 0.2015,
      "step": 10480
    },
    {
      "epoch": 1.23,
      "grad_norm": 2.794186200047937,
      "learning_rate": 1.659746467629173e-05,
      "loss": 0.2267,
      "step": 10500
    },
    {
      "epoch": 1.23,
      "grad_norm": 3.208243100437357,
      "learning_rate": 1.6582913151491286e-05,
      "loss": 0.2272,
      "step": 10520
    },
    {
      "epoch": 1.23,
      "grad_norm": 2.8524752057641996,
      "learning_rate": 1.6568336986720773e-05,
      "loss": 0.2169,
      "step": 10540
    },
    {
      "epoch": 1.24,
      "grad_norm": 3.3114099688426513,
      "learning_rate": 1.6553736236547336e-05,
      "loss": 0.2161,
      "step": 10560
    },
    {
      "epoch": 1.24,
      "grad_norm": 3.763983657663605,
      "learning_rate": 1.653911095563017e-05,
      "loss": 0.2252,
      "step": 10580
    },
    {
      "epoch": 1.24,
      "grad_norm": 3.266499087518214,
      "learning_rate": 1.6524461198720303e-05,
      "loss": 0.1967,
      "step": 10600
    },
    {
      "epoch": 1.24,
      "grad_norm": 3.9946972745895017,
      "learning_rate": 1.650978702066038e-05,
      "loss": 0.2174,
      "step": 10620
    },
    {
      "epoch": 1.25,
      "grad_norm": 2.8854543268016055,
      "learning_rate": 1.649508847638448e-05,
      "loss": 0.2136,
      "step": 10640
    },
    {
      "epoch": 1.25,
      "grad_norm": 3.5610681969299653,
      "learning_rate": 1.648036562091789e-05,
      "loss": 0.225,
      "step": 10660
    },
    {
      "epoch": 1.25,
      "grad_norm": 8.713288377423561,
      "learning_rate": 1.6465618509376923e-05,
      "loss": 0.2172,
      "step": 10680
    },
    {
      "epoch": 1.25,
      "grad_norm": 3.2387951119923932,
      "learning_rate": 1.6450847196968676e-05,
      "loss": 0.2131,
      "step": 10700
    },
    {
      "epoch": 1.25,
      "grad_norm": 3.8639660913863234,
      "learning_rate": 1.643605173899086e-05,
      "loss": 0.2126,
      "step": 10720
    },
    {
      "epoch": 1.26,
      "grad_norm": 4.013150770232071,
      "learning_rate": 1.642123219083156e-05,
      "loss": 0.2253,
      "step": 10740
    },
    {
      "epoch": 1.26,
      "grad_norm": 3.448212871140601,
      "learning_rate": 1.640638860796908e-05,
      "loss": 0.2122,
      "step": 10760
    },
    {
      "epoch": 1.26,
      "grad_norm": 4.008865307397649,
      "learning_rate": 1.6391521045971654e-05,
      "loss": 0.2198,
      "step": 10780
    },
    {
      "epoch": 1.26,
      "grad_norm": 2.668310542854165,
      "learning_rate": 1.6376629560497318e-05,
      "loss": 0.2303,
      "step": 10800
    },
    {
      "epoch": 1.27,
      "grad_norm": 4.253939653866208,
      "learning_rate": 1.636171420729365e-05,
      "loss": 0.2165,
      "step": 10820
    },
    {
      "epoch": 1.27,
      "grad_norm": 3.662446989645766,
      "learning_rate": 1.634677504219759e-05,
      "loss": 0.2093,
      "step": 10840
    },
    {
      "epoch": 1.27,
      "grad_norm": 3.7845535349608115,
      "learning_rate": 1.6331812121135216e-05,
      "loss": 0.1997,
      "step": 10860
    },
    {
      "epoch": 1.27,
      "grad_norm": 4.21364980338036,
      "learning_rate": 1.6316825500121528e-05,
      "loss": 0.2186,
      "step": 10880
    },
    {
      "epoch": 1.28,
      "grad_norm": 3.4198924553007153,
      "learning_rate": 1.630181523526027e-05,
      "loss": 0.2303,
      "step": 10900
    },
    {
      "epoch": 1.28,
      "grad_norm": 3.569247097261353,
      "learning_rate": 1.6286781382743678e-05,
      "loss": 0.2084,
      "step": 10920
    },
    {
      "epoch": 1.28,
      "grad_norm": 3.2057387689879966,
      "learning_rate": 1.6271723998852305e-05,
      "loss": 0.2223,
      "step": 10940
    },
    {
      "epoch": 1.28,
      "grad_norm": 3.438084084566052,
      "learning_rate": 1.6256643139954787e-05,
      "loss": 0.2087,
      "step": 10960
    },
    {
      "epoch": 1.28,
      "grad_norm": 3.1629151324387075,
      "learning_rate": 1.6241538862507647e-05,
      "loss": 0.2225,
      "step": 10980
    },
    {
      "epoch": 1.29,
      "grad_norm": 3.1696025055249097,
      "learning_rate": 1.622641122305507e-05,
      "loss": 0.2315,
      "step": 11000
    },
    {
      "epoch": 1.29,
      "grad_norm": 3.053371916892243,
      "learning_rate": 1.621126027822871e-05,
      "loss": 0.2116,
      "step": 11020
    },
    {
      "epoch": 1.29,
      "grad_norm": 2.9543774083504766,
      "learning_rate": 1.619608608474745e-05,
      "loss": 0.2254,
      "step": 11040
    },
    {
      "epoch": 1.29,
      "grad_norm": 3.3693833415742453,
      "learning_rate": 1.618088869941723e-05,
      "loss": 0.2169,
      "step": 11060
    },
    {
      "epoch": 1.3,
      "grad_norm": 3.9995183654737376,
      "learning_rate": 1.6165668179130784e-05,
      "loss": 0.2365,
      "step": 11080
    },
    {
      "epoch": 1.3,
      "grad_norm": 13.880528903525311,
      "learning_rate": 1.6150424580867476e-05,
      "loss": 0.2333,
      "step": 11100
    },
    {
      "epoch": 1.3,
      "grad_norm": 3.1836707357447778,
      "learning_rate": 1.613515796169305e-05,
      "loss": 0.2237,
      "step": 11120
    },
    {
      "epoch": 1.3,
      "grad_norm": 3.8509352600789506,
      "learning_rate": 1.6119868378759436e-05,
      "loss": 0.2053,
      "step": 11140
    },
    {
      "epoch": 1.31,
      "grad_norm": 3.2583341112322906,
      "learning_rate": 1.610455588930453e-05,
      "loss": 0.2139,
      "step": 11160
    },
    {
      "epoch": 1.31,
      "grad_norm": 5.386703878469319,
      "learning_rate": 1.6089220550651985e-05,
      "loss": 0.2104,
      "step": 11180
    },
    {
      "epoch": 1.31,
      "grad_norm": 3.1002668081338114,
      "learning_rate": 1.607386242021099e-05,
      "loss": 0.2048,
      "step": 11200
    },
    {
      "epoch": 1.31,
      "grad_norm": 4.67566499111419,
      "learning_rate": 1.605848155547605e-05,
      "loss": 0.2182,
      "step": 11220
    },
    {
      "epoch": 1.32,
      "grad_norm": 1.8252132735330482,
      "learning_rate": 1.6043078014026786e-05,
      "loss": 0.2211,
      "step": 11240
    },
    {
      "epoch": 1.32,
      "grad_norm": 3.5104873755754635,
      "learning_rate": 1.6027651853527706e-05,
      "loss": 0.2082,
      "step": 11260
    },
    {
      "epoch": 1.32,
      "grad_norm": 2.596015767829866,
      "learning_rate": 1.6012203131728e-05,
      "loss": 0.2053,
      "step": 11280
    },
    {
      "epoch": 1.32,
      "grad_norm": 3.9320961337285247,
      "learning_rate": 1.5996731906461314e-05,
      "loss": 0.2124,
      "step": 11300
    },
    {
      "epoch": 1.32,
      "grad_norm": 3.239895886290778,
      "learning_rate": 1.598123823564554e-05,
      "loss": 0.2126,
      "step": 11320
    },
    {
      "epoch": 1.33,
      "grad_norm": 7.536836064136331,
      "learning_rate": 1.5965722177282595e-05,
      "loss": 0.2199,
      "step": 11340
    },
    {
      "epoch": 1.33,
      "grad_norm": 5.562084911160559,
      "learning_rate": 1.5950183789458206e-05,
      "loss": 0.2246,
      "step": 11360
    },
    {
      "epoch": 1.33,
      "grad_norm": 2.742122377671127,
      "learning_rate": 1.5934623130341695e-05,
      "loss": 0.2178,
      "step": 11380
    },
    {
      "epoch": 1.33,
      "grad_norm": 4.143294024804004,
      "learning_rate": 1.5919040258185753e-05,
      "loss": 0.2189,
      "step": 11400
    },
    {
      "epoch": 1.34,
      "grad_norm": 3.6948127005341975,
      "learning_rate": 1.5903435231326233e-05,
      "loss": 0.2311,
      "step": 11420
    },
    {
      "epoch": 1.34,
      "grad_norm": 2.8456773200400005,
      "learning_rate": 1.5887808108181922e-05,
      "loss": 0.2047,
      "step": 11440
    },
    {
      "epoch": 1.34,
      "grad_norm": 4.524975240228682,
      "learning_rate": 1.5872158947254328e-05,
      "loss": 0.216,
      "step": 11460
    },
    {
      "epoch": 1.34,
      "grad_norm": 3.22887435738582,
      "learning_rate": 1.585648780712746e-05,
      "loss": 0.2139,
      "step": 11480
    },
    {
      "epoch": 1.35,
      "grad_norm": 2.40402306888702,
      "learning_rate": 1.5840794746467615e-05,
      "loss": 0.2066,
      "step": 11500
    },
    {
      "epoch": 1.35,
      "grad_norm": 3.4971814386926967,
      "learning_rate": 1.5825079824023132e-05,
      "loss": 0.2131,
      "step": 11520
    },
    {
      "epoch": 1.35,
      "grad_norm": 3.210588645083174,
      "learning_rate": 1.5809343098624212e-05,
      "loss": 0.2114,
      "step": 11540
    },
    {
      "epoch": 1.35,
      "grad_norm": 2.632446897574305,
      "learning_rate": 1.5793584629182667e-05,
      "loss": 0.2151,
      "step": 11560
    },
    {
      "epoch": 1.36,
      "grad_norm": 4.5614133612079755,
      "learning_rate": 1.577780447469171e-05,
      "loss": 0.2151,
      "step": 11580
    },
    {
      "epoch": 1.36,
      "grad_norm": 2.394202553621421,
      "learning_rate": 1.5762002694225735e-05,
      "loss": 0.2079,
      "step": 11600
    },
    {
      "epoch": 1.36,
      "grad_norm": 2.6251453177782604,
      "learning_rate": 1.5746179346940102e-05,
      "loss": 0.2159,
      "step": 11620
    },
    {
      "epoch": 1.36,
      "grad_norm": 3.3437649735445256,
      "learning_rate": 1.5730334492070893e-05,
      "loss": 0.22,
      "step": 11640
    },
    {
      "epoch": 1.36,
      "grad_norm": 2.63771302867986,
      "learning_rate": 1.5714468188934716e-05,
      "loss": 0.2136,
      "step": 11660
    },
    {
      "epoch": 1.37,
      "grad_norm": 2.655161017296502,
      "learning_rate": 1.5698580496928475e-05,
      "loss": 0.2067,
      "step": 11680
    },
    {
      "epoch": 1.37,
      "grad_norm": 2.948812575507089,
      "learning_rate": 1.5682671475529137e-05,
      "loss": 0.2008,
      "step": 11700
    },
    {
      "epoch": 1.37,
      "grad_norm": 3.469779471829569,
      "learning_rate": 1.566674118429352e-05,
      "loss": 0.2228,
      "step": 11720
    },
    {
      "epoch": 1.37,
      "grad_norm": 3.2002593829411747,
      "learning_rate": 1.5650789682858073e-05,
      "loss": 0.2149,
      "step": 11740
    },
    {
      "epoch": 1.38,
      "grad_norm": 3.0178239938264864,
      "learning_rate": 1.563481703093864e-05,
      "loss": 0.2173,
      "step": 11760
    },
    {
      "epoch": 1.38,
      "grad_norm": 3.359858420431831,
      "learning_rate": 1.5618823288330245e-05,
      "loss": 0.2109,
      "step": 11780
    },
    {
      "epoch": 1.38,
      "grad_norm": 4.244464186067146,
      "learning_rate": 1.5602808514906874e-05,
      "loss": 0.2172,
      "step": 11800
    },
    {
      "epoch": 1.38,
      "grad_norm": 3.02678280463322,
      "learning_rate": 1.5586772770621236e-05,
      "loss": 0.2055,
      "step": 11820
    },
    {
      "epoch": 1.39,
      "grad_norm": 3.2941320498135602,
      "learning_rate": 1.5570716115504544e-05,
      "loss": 0.2212,
      "step": 11840
    },
    {
      "epoch": 1.39,
      "grad_norm": 3.2434629574232474,
      "learning_rate": 1.5554638609666298e-05,
      "loss": 0.2183,
      "step": 11860
    },
    {
      "epoch": 1.39,
      "grad_norm": 2.7694471424497342,
      "learning_rate": 1.5538540313294058e-05,
      "loss": 0.2254,
      "step": 11880
    },
    {
      "epoch": 1.39,
      "grad_norm": 3.5269104616476983,
      "learning_rate": 1.5522421286653208e-05,
      "loss": 0.2201,
      "step": 11900
    },
    {
      "epoch": 1.39,
      "grad_norm": 3.8299784007845363,
      "learning_rate": 1.550628159008674e-05,
      "loss": 0.2133,
      "step": 11920
    },
    {
      "epoch": 1.4,
      "grad_norm": 5.026447159875988,
      "learning_rate": 1.5490121284015026e-05,
      "loss": 0.2177,
      "step": 11940
    },
    {
      "epoch": 1.4,
      "grad_norm": 3.55665309344499,
      "learning_rate": 1.5473940428935594e-05,
      "loss": 0.2269,
      "step": 11960
    },
    {
      "epoch": 1.4,
      "grad_norm": 3.8149873717812732,
      "learning_rate": 1.545773908542289e-05,
      "loss": 0.2135,
      "step": 11980
    },
    {
      "epoch": 1.4,
      "grad_norm": 4.453465498906573,
      "learning_rate": 1.5441517314128075e-05,
      "loss": 0.2066,
      "step": 12000
    },
    {
      "epoch": 1.41,
      "grad_norm": 3.4558864961105167,
      "learning_rate": 1.5425275175778765e-05,
      "loss": 0.2221,
      "step": 12020
    },
    {
      "epoch": 1.41,
      "grad_norm": 3.0249571615921726,
      "learning_rate": 1.5409012731178844e-05,
      "loss": 0.2176,
      "step": 12040
    },
    {
      "epoch": 1.41,
      "grad_norm": 3.378968625237132,
      "learning_rate": 1.539273004120819e-05,
      "loss": 0.2129,
      "step": 12060
    },
    {
      "epoch": 1.41,
      "grad_norm": 3.599643022004301,
      "learning_rate": 1.5376427166822493e-05,
      "loss": 0.2005,
      "step": 12080
    },
    {
      "epoch": 1.42,
      "grad_norm": 4.369469771352546,
      "learning_rate": 1.5360104169052983e-05,
      "loss": 0.2072,
      "step": 12100
    },
    {
      "epoch": 1.42,
      "grad_norm": 2.9804449117964618,
      "learning_rate": 1.534376110900625e-05,
      "loss": 0.2235,
      "step": 12120
    },
    {
      "epoch": 1.42,
      "grad_norm": 3.477739662012998,
      "learning_rate": 1.532739804786397e-05,
      "loss": 0.2044,
      "step": 12140
    },
    {
      "epoch": 1.42,
      "grad_norm": 4.365666269531915,
      "learning_rate": 1.5311015046882695e-05,
      "loss": 0.2068,
      "step": 12160
    },
    {
      "epoch": 1.43,
      "grad_norm": 3.097981651118829,
      "learning_rate": 1.5294612167393633e-05,
      "loss": 0.2034,
      "step": 12180
    },
    {
      "epoch": 1.43,
      "grad_norm": 3.1761710102538485,
      "learning_rate": 1.5278189470802402e-05,
      "loss": 0.2148,
      "step": 12200
    },
    {
      "epoch": 1.43,
      "grad_norm": 6.183507999444912,
      "learning_rate": 1.526174701858881e-05,
      "loss": 0.2252,
      "step": 12220
    },
    {
      "epoch": 1.43,
      "grad_norm": 3.794441982049424,
      "learning_rate": 1.5245284872306627e-05,
      "loss": 0.2194,
      "step": 12240
    },
    {
      "epoch": 1.43,
      "grad_norm": 3.061187462874714,
      "learning_rate": 1.5228803093583333e-05,
      "loss": 0.2255,
      "step": 12260
    },
    {
      "epoch": 1.44,
      "grad_norm": 3.26228124827874,
      "learning_rate": 1.5212301744119922e-05,
      "loss": 0.2297,
      "step": 12280
    },
    {
      "epoch": 1.44,
      "grad_norm": 2.8942060403720355,
      "learning_rate": 1.5195780885690641e-05,
      "loss": 0.2013,
      "step": 12300
    },
    {
      "epoch": 1.44,
      "grad_norm": 2.6600186708698534,
      "learning_rate": 1.5179240580142782e-05,
      "loss": 0.2047,
      "step": 12320
    },
    {
      "epoch": 1.44,
      "grad_norm": 3.969215305715658,
      "learning_rate": 1.5162680889396424e-05,
      "loss": 0.2173,
      "step": 12340
    },
    {
      "epoch": 1.45,
      "grad_norm": 2.6829571142305904,
      "learning_rate": 1.5146101875444226e-05,
      "loss": 0.2023,
      "step": 12360
    },
    {
      "epoch": 1.45,
      "grad_norm": 3.6657811020233684,
      "learning_rate": 1.5129503600351186e-05,
      "loss": 0.2147,
      "step": 12380
    },
    {
      "epoch": 1.45,
      "grad_norm": 3.90125498508867,
      "learning_rate": 1.5112886126254405e-05,
      "loss": 0.2137,
      "step": 12400
    },
    {
      "epoch": 1.45,
      "grad_norm": 3.147105349280156,
      "learning_rate": 1.5096249515362853e-05,
      "loss": 0.2118,
      "step": 12420
    },
    {
      "epoch": 1.46,
      "grad_norm": 2.7729387721425645,
      "learning_rate": 1.5079593829957144e-05,
      "loss": 0.2149,
      "step": 12440
    },
    {
      "epoch": 1.46,
      "grad_norm": 5.342410092487611,
      "learning_rate": 1.5062919132389305e-05,
      "loss": 0.2169,
      "step": 12460
    },
    {
      "epoch": 1.46,
      "grad_norm": 3.9559478933684638,
      "learning_rate": 1.5046225485082526e-05,
      "loss": 0.2205,
      "step": 12480
    },
    {
      "epoch": 1.46,
      "grad_norm": 2.726490183748406,
      "learning_rate": 1.5029512950530943e-05,
      "loss": 0.224,
      "step": 12500
    },
    {
      "epoch": 1.47,
      "grad_norm": 2.990487116043663,
      "learning_rate": 1.5012781591299399e-05,
      "loss": 0.2287,
      "step": 12520
    },
    {
      "epoch": 1.47,
      "grad_norm": 3.8406463746608215,
      "learning_rate": 1.4996031470023205e-05,
      "loss": 0.2198,
      "step": 12540
    },
    {
      "epoch": 1.47,
      "grad_norm": 3.29662607708646,
      "learning_rate": 1.4979262649407914e-05,
      "loss": 0.2198,
      "step": 12560
    },
    {
      "epoch": 1.47,
      "grad_norm": 2.7040482195837834,
      "learning_rate": 1.496247519222908e-05,
      "loss": 0.2079,
      "step": 12580
    },
    {
      "epoch": 1.47,
      "grad_norm": 2.6175719021573918,
      "learning_rate": 1.4945669161332021e-05,
      "loss": 0.2198,
      "step": 12600
    },
    {
      "epoch": 1.48,
      "grad_norm": 2.9074738909532285,
      "learning_rate": 1.4928844619631588e-05,
      "loss": 0.2084,
      "step": 12620
    },
    {
      "epoch": 1.48,
      "grad_norm": 3.1468612473612563,
      "learning_rate": 1.4912001630111936e-05,
      "loss": 0.2013,
      "step": 12640
    },
    {
      "epoch": 1.48,
      "grad_norm": 3.779361095424968,
      "learning_rate": 1.4895140255826273e-05,
      "loss": 0.1918,
      "step": 12660
    },
    {
      "epoch": 1.48,
      "grad_norm": 2.696019176777178,
      "learning_rate": 1.4878260559896634e-05,
      "loss": 0.2126,
      "step": 12680
    },
    {
      "epoch": 1.49,
      "grad_norm": 3.7345031971654103,
      "learning_rate": 1.4861362605513643e-05,
      "loss": 0.2212,
      "step": 12700
    },
    {
      "epoch": 1.49,
      "grad_norm": 4.465836435468654,
      "learning_rate": 1.4844446455936277e-05,
      "loss": 0.2211,
      "step": 12720
    },
    {
      "epoch": 1.49,
      "grad_norm": 2.9078487797789645,
      "learning_rate": 1.4827512174491631e-05,
      "loss": 0.2267,
      "step": 12740
    },
    {
      "epoch": 1.49,
      "grad_norm": 3.296252006498326,
      "learning_rate": 1.4810559824574675e-05,
      "loss": 0.2216,
      "step": 12760
    },
    {
      "epoch": 1.5,
      "grad_norm": 3.227367236112649,
      "learning_rate": 1.4793589469648017e-05,
      "loss": 0.2233,
      "step": 12780
    },
    {
      "epoch": 1.5,
      "grad_norm": 3.4962503238332774,
      "learning_rate": 1.4776601173241671e-05,
      "loss": 0.2269,
      "step": 12800
    },
    {
      "epoch": 1.5,
      "grad_norm": 2.878724712305482,
      "learning_rate": 1.4759594998952821e-05,
      "loss": 0.2119,
      "step": 12820
    },
    {
      "epoch": 1.5,
      "grad_norm": 3.682489644312493,
      "learning_rate": 1.4742571010445572e-05,
      "loss": 0.2011,
      "step": 12840
    },
    {
      "epoch": 1.5,
      "grad_norm": 5.266174287756561,
      "learning_rate": 1.472552927145072e-05,
      "loss": 0.2258,
      "step": 12860
    },
    {
      "epoch": 1.51,
      "grad_norm": 4.045996375421248,
      "learning_rate": 1.4708469845765512e-05,
      "loss": 0.1922,
      "step": 12880
    },
    {
      "epoch": 1.51,
      "grad_norm": 3.266689729284762,
      "learning_rate": 1.4691392797253405e-05,
      "loss": 0.2075,
      "step": 12900
    },
    {
      "epoch": 1.51,
      "grad_norm": 2.242467882655721,
      "learning_rate": 1.4674298189843837e-05,
      "loss": 0.212,
      "step": 12920
    },
    {
      "epoch": 1.51,
      "grad_norm": 2.807279276540352,
      "learning_rate": 1.4657186087531962e-05,
      "loss": 0.1924,
      "step": 12940
    },
    {
      "epoch": 1.52,
      "grad_norm": 6.9144252531074875,
      "learning_rate": 1.4640056554378448e-05,
      "loss": 0.224,
      "step": 12960
    },
    {
      "epoch": 1.52,
      "grad_norm": 3.5667764527961645,
      "learning_rate": 1.4622909654509204e-05,
      "loss": 0.2175,
      "step": 12980
    },
    {
      "epoch": 1.52,
      "grad_norm": 2.431697492432426,
      "learning_rate": 1.4605745452115155e-05,
      "loss": 0.2172,
      "step": 13000
    },
    {
      "epoch": 1.52,
      "grad_norm": 3.336010970081616,
      "learning_rate": 1.4588564011451998e-05,
      "loss": 0.2219,
      "step": 13020
    },
    {
      "epoch": 1.53,
      "grad_norm": 2.9985996792871656,
      "learning_rate": 1.4571365396839969e-05,
      "loss": 0.2226,
      "step": 13040
    },
    {
      "epoch": 1.53,
      "grad_norm": 2.7817912914440677,
      "learning_rate": 1.4554149672663591e-05,
      "loss": 0.2166,
      "step": 13060
    },
    {
      "epoch": 1.53,
      "grad_norm": 3.5654978937601007,
      "learning_rate": 1.4536916903371443e-05,
      "loss": 0.209,
      "step": 13080
    },
    {
      "epoch": 1.53,
      "grad_norm": 3.140872774741023,
      "learning_rate": 1.4519667153475908e-05,
      "loss": 0.2059,
      "step": 13100
    },
    {
      "epoch": 1.54,
      "grad_norm": 3.7735270929856304,
      "learning_rate": 1.4502400487552937e-05,
      "loss": 0.2078,
      "step": 13120
    },
    {
      "epoch": 1.54,
      "grad_norm": 3.3704008625810826,
      "learning_rate": 1.4485116970241814e-05,
      "loss": 0.2288,
      "step": 13140
    },
    {
      "epoch": 1.54,
      "grad_norm": 3.058405104235901,
      "learning_rate": 1.4467816666244904e-05,
      "loss": 0.2145,
      "step": 13160
    },
    {
      "epoch": 1.54,
      "grad_norm": 3.2628541717630295,
      "learning_rate": 1.4450499640327418e-05,
      "loss": 0.2096,
      "step": 13180
    },
    {
      "epoch": 1.54,
      "grad_norm": 2.8358576319353475,
      "learning_rate": 1.443316595731716e-05,
      "loss": 0.2162,
      "step": 13200
    },
    {
      "epoch": 1.55,
      "grad_norm": 2.130034709343158,
      "learning_rate": 1.4415815682104295e-05,
      "loss": 0.2038,
      "step": 13220
    },
    {
      "epoch": 1.55,
      "grad_norm": 3.679976984449645,
      "learning_rate": 1.4398448879641102e-05,
      "loss": 0.216,
      "step": 13240
    },
    {
      "epoch": 1.55,
      "grad_norm": 3.5668380826834647,
      "learning_rate": 1.4381065614941733e-05,
      "loss": 0.2262,
      "step": 13260
    },
    {
      "epoch": 1.55,
      "grad_norm": 3.0243831604612765,
      "learning_rate": 1.4363665953081969e-05,
      "loss": 0.201,
      "step": 13280
    },
    {
      "epoch": 1.56,
      "grad_norm": 3.2212853677485933,
      "learning_rate": 1.4346249959198965e-05,
      "loss": 0.2163,
      "step": 13300
    },
    {
      "epoch": 1.56,
      "grad_norm": 2.596122759445438,
      "learning_rate": 1.4328817698491031e-05,
      "loss": 0.22,
      "step": 13320
    },
    {
      "epoch": 1.56,
      "grad_norm": 3.0666724605781996,
      "learning_rate": 1.4311369236217367e-05,
      "loss": 0.2139,
      "step": 13340
    },
    {
      "epoch": 1.56,
      "grad_norm": 3.657252769223528,
      "learning_rate": 1.4293904637697817e-05,
      "loss": 0.215,
      "step": 13360
    },
    {
      "epoch": 1.57,
      "grad_norm": 3.4144546667640383,
      "learning_rate": 1.4276423968312645e-05,
      "loss": 0.227,
      "step": 13380
    },
    {
      "epoch": 1.57,
      "grad_norm": 3.2413250982644675,
      "learning_rate": 1.4258927293502273e-05,
      "loss": 0.2077,
      "step": 13400
    },
    {
      "epoch": 1.57,
      "grad_norm": 3.5063106320828488,
      "learning_rate": 1.4241414678767039e-05,
      "loss": 0.2047,
      "step": 13420
    },
    {
      "epoch": 1.57,
      "grad_norm": 2.1504143515238514,
      "learning_rate": 1.4223886189666951e-05,
      "loss": 0.1982,
      "step": 13440
    },
    {
      "epoch": 1.58,
      "grad_norm": 2.9525594825793005,
      "learning_rate": 1.4206341891821456e-05,
      "loss": 0.2066,
      "step": 13460
    },
    {
      "epoch": 1.58,
      "grad_norm": 5.213240221920547,
      "learning_rate": 1.4188781850909165e-05,
      "loss": 0.2246,
      "step": 13480
    },
    {
      "epoch": 1.58,
      "grad_norm": 4.305790200898134,
      "learning_rate": 1.417120613266764e-05,
      "loss": 0.1994,
      "step": 13500
    },
    {
      "epoch": 1.58,
      "grad_norm": 3.370920187857142,
      "learning_rate": 1.4153614802893125e-05,
      "loss": 0.2085,
      "step": 13520
    },
    {
      "epoch": 1.58,
      "grad_norm": 3.425048053710641,
      "learning_rate": 1.4136007927440307e-05,
      "loss": 0.207,
      "step": 13540
    },
    {
      "epoch": 1.59,
      "grad_norm": 3.24691919878633,
      "learning_rate": 1.4118385572222072e-05,
      "loss": 0.2194,
      "step": 13560
    },
    {
      "epoch": 1.59,
      "grad_norm": 2.8908834805780064,
      "learning_rate": 1.4100747803209252e-05,
      "loss": 0.2024,
      "step": 13580
    },
    {
      "epoch": 1.59,
      "grad_norm": 3.5087368773081473,
      "learning_rate": 1.4083094686430387e-05,
      "loss": 0.2102,
      "step": 13600
    },
    {
      "epoch": 1.59,
      "grad_norm": 2.5718331983652485,
      "learning_rate": 1.4065426287971473e-05,
      "loss": 0.2205,
      "step": 13620
    },
    {
      "epoch": 1.6,
      "grad_norm": 3.0529680412811375,
      "learning_rate": 1.4047742673975706e-05,
      "loss": 0.214,
      "step": 13640
    },
    {
      "epoch": 1.6,
      "grad_norm": 3.6927881631261505,
      "learning_rate": 1.4030043910643254e-05,
      "loss": 0.2124,
      "step": 13660
    },
    {
      "epoch": 1.6,
      "grad_norm": 3.3041534353009743,
      "learning_rate": 1.401233006423099e-05,
      "loss": 0.2033,
      "step": 13680
    },
    {
      "epoch": 1.6,
      "grad_norm": 3.539906112805195,
      "learning_rate": 1.3994601201052253e-05,
      "loss": 0.2022,
      "step": 13700
    },
    {
      "epoch": 1.61,
      "grad_norm": 3.981704115906401,
      "learning_rate": 1.39768573874766e-05,
      "loss": 0.213,
      "step": 13720
    },
    {
      "epoch": 1.61,
      "grad_norm": 2.5831456167458375,
      "learning_rate": 1.3959098689929555e-05,
      "loss": 0.1958,
      "step": 13740
    },
    {
      "epoch": 1.61,
      "grad_norm": 2.4960100759889725,
      "learning_rate": 1.3941325174892371e-05,
      "loss": 0.2243,
      "step": 13760
    },
    {
      "epoch": 1.61,
      "grad_norm": 2.9399908340726424,
      "learning_rate": 1.392353690890175e-05,
      "loss": 0.2052,
      "step": 13780
    },
    {
      "epoch": 1.61,
      "grad_norm": 3.161745178851194,
      "learning_rate": 1.3905733958549632e-05,
      "loss": 0.1905,
      "step": 13800
    },
    {
      "epoch": 1.62,
      "grad_norm": 3.6763150256287305,
      "learning_rate": 1.3887916390482927e-05,
      "loss": 0.2097,
      "step": 13820
    },
    {
      "epoch": 1.62,
      "grad_norm": 2.589478710103983,
      "learning_rate": 1.3870084271403269e-05,
      "loss": 0.2108,
      "step": 13840
    },
    {
      "epoch": 1.62,
      "grad_norm": 2.9786347312251205,
      "learning_rate": 1.3852237668066763e-05,
      "loss": 0.2021,
      "step": 13860
    },
    {
      "epoch": 1.62,
      "grad_norm": 2.918896140906093,
      "learning_rate": 1.3834376647283729e-05,
      "loss": 0.2184,
      "step": 13880
    },
    {
      "epoch": 1.63,
      "grad_norm": 3.4250636463795074,
      "learning_rate": 1.3816501275918475e-05,
      "loss": 0.1989,
      "step": 13900
    },
    {
      "epoch": 1.63,
      "grad_norm": 2.7399597682922923,
      "learning_rate": 1.3798611620889017e-05,
      "loss": 0.2089,
      "step": 13920
    },
    {
      "epoch": 1.63,
      "grad_norm": 3.678704067595457,
      "learning_rate": 1.378070774916686e-05,
      "loss": 0.2192,
      "step": 13940
    },
    {
      "epoch": 1.63,
      "grad_norm": 3.1242276572434133,
      "learning_rate": 1.3762789727776708e-05,
      "loss": 0.2157,
      "step": 13960
    },
    {
      "epoch": 1.64,
      "grad_norm": 2.885737395564972,
      "learning_rate": 1.3744857623796255e-05,
      "loss": 0.2133,
      "step": 13980
    },
    {
      "epoch": 1.64,
      "grad_norm": 2.998690955346801,
      "learning_rate": 1.3726911504355908e-05,
      "loss": 0.2129,
      "step": 14000
    },
    {
      "epoch": 1.64,
      "grad_norm": 4.013414301345931,
      "learning_rate": 1.3708951436638542e-05,
      "loss": 0.2031,
      "step": 14020
    },
    {
      "epoch": 1.64,
      "grad_norm": 3.6060060573956503,
      "learning_rate": 1.3690977487879247e-05,
      "loss": 0.2149,
      "step": 14040
    },
    {
      "epoch": 1.65,
      "grad_norm": 2.8295749719869105,
      "learning_rate": 1.3672989725365076e-05,
      "loss": 0.2113,
      "step": 14060
    },
    {
      "epoch": 1.65,
      "grad_norm": 4.634259826093287,
      "learning_rate": 1.3654988216434802e-05,
      "loss": 0.2013,
      "step": 14080
    },
    {
      "epoch": 1.65,
      "grad_norm": 4.019301812951925,
      "learning_rate": 1.3636973028478655e-05,
      "loss": 0.2079,
      "step": 14100
    },
    {
      "epoch": 1.65,
      "grad_norm": 3.3314202699929387,
      "learning_rate": 1.3618944228938072e-05,
      "loss": 0.1968,
      "step": 14120
    },
    {
      "epoch": 1.65,
      "grad_norm": 4.011000289395802,
      "learning_rate": 1.3600901885305446e-05,
      "loss": 0.2075,
      "step": 14140
    },
    {
      "epoch": 1.66,
      "grad_norm": 3.1307277830268183,
      "learning_rate": 1.358284606512388e-05,
      "loss": 0.2004,
      "step": 14160
    },
    {
      "epoch": 1.66,
      "grad_norm": 2.6402421481891047,
      "learning_rate": 1.3564776835986921e-05,
      "loss": 0.2102,
      "step": 14180
    },
    {
      "epoch": 1.66,
      "grad_norm": 3.1118142226597745,
      "learning_rate": 1.3546694265538316e-05,
      "loss": 0.2124,
      "step": 14200
    },
    {
      "epoch": 1.66,
      "grad_norm": 3.4864022554268614,
      "learning_rate": 1.3528598421471755e-05,
      "loss": 0.2067,
      "step": 14220
    },
    {
      "epoch": 1.67,
      "grad_norm": 2.8051441214080457,
      "learning_rate": 1.3510489371530615e-05,
      "loss": 0.2163,
      "step": 14240
    },
    {
      "epoch": 1.67,
      "grad_norm": 3.1532788786376638,
      "learning_rate": 1.3492367183507726e-05,
      "loss": 0.203,
      "step": 14260
    },
    {
      "epoch": 1.67,
      "grad_norm": 3.6383798904923412,
      "learning_rate": 1.3474231925245087e-05,
      "loss": 0.2117,
      "step": 14280
    },
    {
      "epoch": 1.67,
      "grad_norm": 3.2573585834385925,
      "learning_rate": 1.3456083664633626e-05,
      "loss": 0.2097,
      "step": 14300
    },
    {
      "epoch": 1.68,
      "grad_norm": 4.797438398942751,
      "learning_rate": 1.3437922469612951e-05,
      "loss": 0.2116,
      "step": 14320
    },
    {
      "epoch": 1.68,
      "grad_norm": 2.5155576140103646,
      "learning_rate": 1.3419748408171094e-05,
      "loss": 0.2225,
      "step": 14340
    },
    {
      "epoch": 1.68,
      "grad_norm": 5.78637039662545,
      "learning_rate": 1.3401561548344253e-05,
      "loss": 0.2074,
      "step": 14360
    },
    {
      "epoch": 1.68,
      "grad_norm": 5.171051974083877,
      "learning_rate": 1.3383361958216531e-05,
      "loss": 0.1966,
      "step": 14380
    },
    {
      "epoch": 1.68,
      "grad_norm": 3.6388434115246278,
      "learning_rate": 1.3365149705919693e-05,
      "loss": 0.2202,
      "step": 14400
    },
    {
      "epoch": 1.69,
      "grad_norm": 3.6368767430520426,
      "learning_rate": 1.3346924859632908e-05,
      "loss": 0.2156,
      "step": 14420
    },
    {
      "epoch": 1.69,
      "grad_norm": 2.6252892652713293,
      "learning_rate": 1.3328687487582486e-05,
      "loss": 0.2047,
      "step": 14440
    },
    {
      "epoch": 1.69,
      "grad_norm": 3.2310005313042853,
      "learning_rate": 1.3310437658041631e-05,
      "loss": 0.1995,
      "step": 14460
    },
    {
      "epoch": 1.69,
      "grad_norm": 2.591667902252132,
      "learning_rate": 1.3292175439330182e-05,
      "loss": 0.1936,
      "step": 14480
    },
    {
      "epoch": 1.7,
      "grad_norm": 6.920226165515457,
      "learning_rate": 1.3273900899814365e-05,
      "loss": 0.2008,
      "step": 14500
    },
    {
      "epoch": 1.7,
      "grad_norm": 3.051444827700318,
      "learning_rate": 1.3255614107906521e-05,
      "loss": 0.1976,
      "step": 14520
    },
    {
      "epoch": 1.7,
      "grad_norm": 2.9835362079482435,
      "learning_rate": 1.3237315132064863e-05,
      "loss": 0.2148,
      "step": 14540
    },
    {
      "epoch": 1.7,
      "grad_norm": 2.8495541541728024,
      "learning_rate": 1.321900404079321e-05,
      "loss": 0.1977,
      "step": 14560
    },
    {
      "epoch": 1.71,
      "grad_norm": 2.7688725257095443,
      "learning_rate": 1.3200680902640744e-05,
      "loss": 0.2178,
      "step": 14580
    },
    {
      "epoch": 1.71,
      "grad_norm": 2.7073866045339576,
      "learning_rate": 1.318234578620174e-05,
      "loss": 0.2146,
      "step": 14600
    },
    {
      "epoch": 1.71,
      "grad_norm": 3.161990242988256,
      "learning_rate": 1.3163998760115325e-05,
      "loss": 0.2134,
      "step": 14620
    },
    {
      "epoch": 1.71,
      "grad_norm": 2.7928417256953812,
      "learning_rate": 1.314563989306519e-05,
      "loss": 0.2065,
      "step": 14640
    },
    {
      "epoch": 1.72,
      "grad_norm": 2.356147630324507,
      "learning_rate": 1.3127269253779376e-05,
      "loss": 0.2105,
      "step": 14660
    },
    {
      "epoch": 1.72,
      "grad_norm": 4.346477722398746,
      "learning_rate": 1.310888691102998e-05,
      "loss": 0.2134,
      "step": 14680
    },
    {
      "epoch": 1.72,
      "grad_norm": 2.7518253337355763,
      "learning_rate": 1.3090492933632926e-05,
      "loss": 0.2107,
      "step": 14700
    },
    {
      "epoch": 1.72,
      "grad_norm": 3.241061463203708,
      "learning_rate": 1.307208739044767e-05,
      "loss": 0.2094,
      "step": 14720
    },
    {
      "epoch": 1.72,
      "grad_norm": 2.666615356507985,
      "learning_rate": 1.3053670350376993e-05,
      "loss": 0.1989,
      "step": 14740
    },
    {
      "epoch": 1.73,
      "grad_norm": 3.0536397322406663,
      "learning_rate": 1.3035241882366692e-05,
      "loss": 0.2156,
      "step": 14760
    },
    {
      "epoch": 1.73,
      "grad_norm": 2.3893756469119007,
      "learning_rate": 1.3016802055405363e-05,
      "loss": 0.2091,
      "step": 14780
    },
    {
      "epoch": 1.73,
      "grad_norm": 3.252027759325605,
      "learning_rate": 1.2998350938524114e-05,
      "loss": 0.1993,
      "step": 14800
    },
    {
      "epoch": 1.73,
      "grad_norm": 3.9445454371124105,
      "learning_rate": 1.2979888600796322e-05,
      "loss": 0.2021,
      "step": 14820
    },
    {
      "epoch": 1.74,
      "grad_norm": 3.811107068357804,
      "learning_rate": 1.2961415111337374e-05,
      "loss": 0.2257,
      "step": 14840
    },
    {
      "epoch": 1.74,
      "grad_norm": 2.437893371140398,
      "learning_rate": 1.2942930539304397e-05,
      "loss": 0.1892,
      "step": 14860
    },
    {
      "epoch": 1.74,
      "grad_norm": 2.5823821090570105,
      "learning_rate": 1.2924434953896014e-05,
      "loss": 0.1954,
      "step": 14880
    },
    {
      "epoch": 1.74,
      "grad_norm": 2.817062999748064,
      "learning_rate": 1.2905928424352068e-05,
      "loss": 0.2016,
      "step": 14900
    },
    {
      "epoch": 1.75,
      "grad_norm": 5.869940669265181,
      "learning_rate": 1.288741101995338e-05,
      "loss": 0.211,
      "step": 14920
    },
    {
      "epoch": 1.75,
      "grad_norm": 3.047154496896096,
      "learning_rate": 1.286888281002148e-05,
      "loss": 0.1963,
      "step": 14940
    },
    {
      "epoch": 1.75,
      "grad_norm": 3.7639733962315756,
      "learning_rate": 1.2850343863918358e-05,
      "loss": 0.2057,
      "step": 14960
    },
    {
      "epoch": 1.75,
      "grad_norm": 2.7118107601240107,
      "learning_rate": 1.2831794251046173e-05,
      "loss": 0.2064,
      "step": 14980
    },
    {
      "epoch": 1.76,
      "grad_norm": 4.293885901020362,
      "learning_rate": 1.281323404084704e-05,
      "loss": 0.189,
      "step": 15000
    },
    {
      "epoch": 1.76,
      "grad_norm": 3.136084771606493,
      "learning_rate": 1.2794663302802732e-05,
      "loss": 0.222,
      "step": 15020
    },
    {
      "epoch": 1.76,
      "grad_norm": 2.33967520945844,
      "learning_rate": 1.2776082106434442e-05,
      "loss": 0.21,
      "step": 15040
    },
    {
      "epoch": 1.76,
      "grad_norm": 3.058105585694254,
      "learning_rate": 1.2757490521302506e-05,
      "loss": 0.2066,
      "step": 15060
    },
    {
      "epoch": 1.76,
      "grad_norm": 3.783134432254202,
      "learning_rate": 1.2738888617006158e-05,
      "loss": 0.204,
      "step": 15080
    },
    {
      "epoch": 1.77,
      "grad_norm": 3.3088612273817577,
      "learning_rate": 1.2720276463183264e-05,
      "loss": 0.2088,
      "step": 15100
    },
    {
      "epoch": 1.77,
      "grad_norm": 2.5164740415038063,
      "learning_rate": 1.2701654129510053e-05,
      "loss": 0.2083,
      "step": 15120
    },
    {
      "epoch": 1.77,
      "grad_norm": 2.680468920777465,
      "learning_rate": 1.2683021685700867e-05,
      "loss": 0.2085,
      "step": 15140
    },
    {
      "epoch": 1.77,
      "grad_norm": 2.7932332800591206,
      "learning_rate": 1.26643792015079e-05,
      "loss": 0.218,
      "step": 15160
    },
    {
      "epoch": 1.78,
      "grad_norm": 4.372169232777801,
      "learning_rate": 1.2645726746720925e-05,
      "loss": 0.1995,
      "step": 15180
    },
    {
      "epoch": 1.78,
      "grad_norm": 3.1330033622307245,
      "learning_rate": 1.2627064391167045e-05,
      "loss": 0.2057,
      "step": 15200
    },
    {
      "epoch": 1.78,
      "grad_norm": 3.6755304848957446,
      "learning_rate": 1.2608392204710436e-05,
      "loss": 0.2154,
      "step": 15220
    },
    {
      "epoch": 1.78,
      "grad_norm": 2.54686994493603,
      "learning_rate": 1.2589710257252058e-05,
      "loss": 0.2056,
      "step": 15240
    },
    {
      "epoch": 1.79,
      "grad_norm": 3.1076386937733926,
      "learning_rate": 1.2571018618729422e-05,
      "loss": 0.2006,
      "step": 15260
    },
    {
      "epoch": 1.79,
      "grad_norm": 4.9484787578205855,
      "learning_rate": 1.2552317359116326e-05,
      "loss": 0.2089,
      "step": 15280
    },
    {
      "epoch": 1.79,
      "grad_norm": 2.4893284006088634,
      "learning_rate": 1.2533606548422577e-05,
      "loss": 0.2024,
      "step": 15300
    },
    {
      "epoch": 1.79,
      "grad_norm": 3.1084832681147594,
      "learning_rate": 1.2514886256693732e-05,
      "loss": 0.2106,
      "step": 15320
    },
    {
      "epoch": 1.79,
      "grad_norm": 3.0389123403917804,
      "learning_rate": 1.249615655401085e-05,
      "loss": 0.2096,
      "step": 15340
    },
    {
      "epoch": 1.8,
      "grad_norm": 4.392113049940431,
      "learning_rate": 1.2477417510490223e-05,
      "loss": 0.21,
      "step": 15360
    },
    {
      "epoch": 1.8,
      "grad_norm": 2.4290921871050077,
      "learning_rate": 1.245866919628311e-05,
      "loss": 0.2101,
      "step": 15380
    },
    {
      "epoch": 1.8,
      "grad_norm": 2.53729167515282,
      "learning_rate": 1.2439911681575459e-05,
      "loss": 0.2085,
      "step": 15400
    },
    {
      "epoch": 1.8,
      "grad_norm": 0.955267140537957,
      "learning_rate": 1.2421145036587681e-05,
      "loss": 0.1946,
      "step": 15420
    },
    {
      "epoch": 1.81,
      "grad_norm": 2.3147872747972924,
      "learning_rate": 1.2402369331574363e-05,
      "loss": 0.2133,
      "step": 15440
    },
    {
      "epoch": 1.81,
      "grad_norm": 2.6069168116002013,
      "learning_rate": 1.2383584636824002e-05,
      "loss": 0.2151,
      "step": 15460
    },
    {
      "epoch": 1.81,
      "grad_norm": 3.6384307404668057,
      "learning_rate": 1.2364791022658756e-05,
      "loss": 0.2055,
      "step": 15480
    },
    {
      "epoch": 1.81,
      "grad_norm": 2.7467900528862796,
      "learning_rate": 1.2345988559434163e-05,
      "loss": 0.1968,
      "step": 15500
    },
    {
      "epoch": 1.82,
      "grad_norm": 3.712915796699261,
      "learning_rate": 1.2327177317538903e-05,
      "loss": 0.2022,
      "step": 15520
    },
    {
      "epoch": 1.82,
      "grad_norm": 3.0273517830034553,
      "learning_rate": 1.2308357367394515e-05,
      "loss": 0.2059,
      "step": 15540
    },
    {
      "epoch": 1.82,
      "grad_norm": 2.9422286911490256,
      "learning_rate": 1.2289528779455129e-05,
      "loss": 0.2109,
      "step": 15560
    },
    {
      "epoch": 1.82,
      "grad_norm": 2.7461702382273097,
      "learning_rate": 1.227069162420722e-05,
      "loss": 0.2008,
      "step": 15580
    },
    {
      "epoch": 1.83,
      "grad_norm": 2.9259269484541868,
      "learning_rate": 1.2251845972169336e-05,
      "loss": 0.2087,
      "step": 15600
    },
    {
      "epoch": 1.83,
      "grad_norm": 2.351163202901331,
      "learning_rate": 1.2232991893891826e-05,
      "loss": 0.2162,
      "step": 15620
    },
    {
      "epoch": 1.83,
      "grad_norm": 3.3864085617483486,
      "learning_rate": 1.2214129459956595e-05,
      "loss": 0.2109,
      "step": 15640
    },
    {
      "epoch": 1.83,
      "grad_norm": 3.1532912786097556,
      "learning_rate": 1.2195258740976816e-05,
      "loss": 0.1927,
      "step": 15660
    },
    {
      "epoch": 1.83,
      "grad_norm": 2.32467980179587,
      "learning_rate": 1.2176379807596684e-05,
      "loss": 0.2083,
      "step": 15680
    },
    {
      "epoch": 1.84,
      "grad_norm": 3.692322632410928,
      "learning_rate": 1.2157492730491142e-05,
      "loss": 0.1952,
      "step": 15700
    },
    {
      "epoch": 1.84,
      "grad_norm": 3.5592767958117704,
      "learning_rate": 1.2138597580365635e-05,
      "loss": 0.2259,
      "step": 15720
    },
    {
      "epoch": 1.84,
      "grad_norm": 3.312387140618671,
      "learning_rate": 1.21196944279558e-05,
      "loss": 0.1996,
      "step": 15740
    },
    {
      "epoch": 1.84,
      "grad_norm": 2.4404373077236974,
      "learning_rate": 1.2100783344027257e-05,
      "loss": 0.2061,
      "step": 15760
    },
    {
      "epoch": 1.85,
      "grad_norm": 3.0929231405689337,
      "learning_rate": 1.2081864399375306e-05,
      "loss": 0.1989,
      "step": 15780
    },
    {
      "epoch": 1.85,
      "grad_norm": 3.6918547181282237,
      "learning_rate": 1.2062937664824692e-05,
      "loss": 0.2039,
      "step": 15800
    },
    {
      "epoch": 1.85,
      "grad_norm": 25.439570612132734,
      "learning_rate": 1.2044003211229292e-05,
      "loss": 0.2023,
      "step": 15820
    },
    {
      "epoch": 1.85,
      "grad_norm": 3.238254008621471,
      "learning_rate": 1.2025061109471903e-05,
      "loss": 0.195,
      "step": 15840
    },
    {
      "epoch": 1.86,
      "grad_norm": 2.717237819317576,
      "learning_rate": 1.2006111430463945e-05,
      "loss": 0.2159,
      "step": 15860
    },
    {
      "epoch": 1.86,
      "grad_norm": 3.3138924317200207,
      "learning_rate": 1.198715424514521e-05,
      "loss": 0.2037,
      "step": 15880
    },
    {
      "epoch": 1.86,
      "grad_norm": 2.477738254765999,
      "learning_rate": 1.1968189624483583e-05,
      "loss": 0.2081,
      "step": 15900
    },
    {
      "epoch": 1.86,
      "grad_norm": 5.595275329960663,
      "learning_rate": 1.194921763947479e-05,
      "loss": 0.2048,
      "step": 15920
    },
    {
      "epoch": 1.87,
      "grad_norm": 6.622925343547304,
      "learning_rate": 1.1930238361142116e-05,
      "loss": 0.2149,
      "step": 15940
    },
    {
      "epoch": 1.87,
      "grad_norm": 3.576100405484642,
      "learning_rate": 1.1911251860536169e-05,
      "loss": 0.2103,
      "step": 15960
    },
    {
      "epoch": 1.87,
      "grad_norm": 2.5035188705559004,
      "learning_rate": 1.1892258208734573e-05,
      "loss": 0.2108,
      "step": 15980
    },
    {
      "epoch": 1.87,
      "grad_norm": 2.6938599141436494,
      "learning_rate": 1.1873257476841734e-05,
      "loss": 0.2029,
      "step": 16000
    },
    {
      "epoch": 1.87,
      "grad_norm": 2.7076420610940914,
      "learning_rate": 1.1854249735988559e-05,
      "loss": 0.203,
      "step": 16020
    },
    {
      "epoch": 1.88,
      "grad_norm": 2.53155657006711,
      "learning_rate": 1.18352350573322e-05,
      "loss": 0.2073,
      "step": 16040
    },
    {
      "epoch": 1.88,
      "grad_norm": 2.8351358589034037,
      "learning_rate": 1.1816213512055778e-05,
      "loss": 0.1972,
      "step": 16060
    },
    {
      "epoch": 1.88,
      "grad_norm": 1.994239617859996,
      "learning_rate": 1.1797185171368113e-05,
      "loss": 0.2157,
      "step": 16080
    },
    {
      "epoch": 1.88,
      "grad_norm": 8.626140574352826,
      "learning_rate": 1.1778150106503472e-05,
      "loss": 0.2219,
      "step": 16100
    },
    {
      "epoch": 1.89,
      "grad_norm": 3.0368386887010326,
      "learning_rate": 1.1759108388721294e-05,
      "loss": 0.212,
      "step": 16120
    },
    {
      "epoch": 1.89,
      "grad_norm": 3.720694025784477,
      "learning_rate": 1.1740060089305923e-05,
      "loss": 0.2188,
      "step": 16140
    },
    {
      "epoch": 1.89,
      "grad_norm": 4.140749601972492,
      "learning_rate": 1.1721005279566335e-05,
      "loss": 0.2205,
      "step": 16160
    },
    {
      "epoch": 1.89,
      "grad_norm": 3.696009886666488,
      "learning_rate": 1.170194403083589e-05,
      "loss": 0.202,
      "step": 16180
    },
    {
      "epoch": 1.9,
      "grad_norm": 2.913336533021661,
      "learning_rate": 1.1682876414472041e-05,
      "loss": 0.2122,
      "step": 16200
    },
    {
      "epoch": 1.9,
      "grad_norm": 3.7716338149283706,
      "learning_rate": 1.1663802501856097e-05,
      "loss": 0.2256,
      "step": 16220
    },
    {
      "epoch": 1.9,
      "grad_norm": 4.6668357591421215,
      "learning_rate": 1.1644722364392916e-05,
      "loss": 0.1985,
      "step": 16240
    },
    {
      "epoch": 1.9,
      "grad_norm": 2.246832949864308,
      "learning_rate": 1.1625636073510669e-05,
      "loss": 0.2221,
      "step": 16260
    },
    {
      "epoch": 1.9,
      "grad_norm": 3.962744426709658,
      "learning_rate": 1.1606543700660566e-05,
      "loss": 0.2083,
      "step": 16280
    },
    {
      "epoch": 1.91,
      "grad_norm": 2.801003078896173,
      "learning_rate": 1.1587445317316579e-05,
      "loss": 0.2005,
      "step": 16300
    },
    {
      "epoch": 1.91,
      "grad_norm": 2.1091323147793117,
      "learning_rate": 1.1568340994975185e-05,
      "loss": 0.2045,
      "step": 16320
    },
    {
      "epoch": 1.91,
      "grad_norm": 3.198361161909217,
      "learning_rate": 1.1549230805155091e-05,
      "loss": 0.206,
      "step": 16340
    },
    {
      "epoch": 1.91,
      "grad_norm": 10.406894789872744,
      "learning_rate": 1.1530114819396976e-05,
      "loss": 0.2123,
      "step": 16360
    },
    {
      "epoch": 1.92,
      "grad_norm": 3.2619393462547737,
      "learning_rate": 1.1510993109263206e-05,
      "loss": 0.2041,
      "step": 16380
    },
    {
      "epoch": 1.92,
      "grad_norm": 3.409171994718834,
      "learning_rate": 1.149186574633759e-05,
      "loss": 0.2056,
      "step": 16400
    },
    {
      "epoch": 1.92,
      "grad_norm": 2.5059832499949737,
      "learning_rate": 1.1472732802225081e-05,
      "loss": 0.2034,
      "step": 16420
    },
    {
      "epoch": 1.92,
      "grad_norm": 3.267940886449616,
      "learning_rate": 1.1453594348551547e-05,
      "loss": 0.2012,
      "step": 16440
    },
    {
      "epoch": 1.93,
      "grad_norm": 3.2066882155626875,
      "learning_rate": 1.1434450456963462e-05,
      "loss": 0.1971,
      "step": 16460
    },
    {
      "epoch": 1.93,
      "grad_norm": 3.1243325092793883,
      "learning_rate": 1.1415301199127671e-05,
      "loss": 0.2041,
      "step": 16480
    },
    {
      "epoch": 1.93,
      "grad_norm": 2.6001065965955448,
      "learning_rate": 1.1396146646731104e-05,
      "loss": 0.1996,
      "step": 16500
    },
    {
      "epoch": 1.93,
      "grad_norm": 2.424714226942306,
      "learning_rate": 1.1376986871480505e-05,
      "loss": 0.1946,
      "step": 16520
    },
    {
      "epoch": 1.94,
      "grad_norm": 2.4162929070497,
      "learning_rate": 1.1357821945102177e-05,
      "loss": 0.1907,
      "step": 16540
    },
    {
      "epoch": 1.94,
      "grad_norm": 3.868349706003454,
      "learning_rate": 1.133865193934171e-05,
      "loss": 0.2202,
      "step": 16560
    },
    {
      "epoch": 1.94,
      "grad_norm": 5.899437583367488,
      "learning_rate": 1.1319476925963702e-05,
      "loss": 0.203,
      "step": 16580
    },
    {
      "epoch": 1.94,
      "grad_norm": 4.061783477977705,
      "learning_rate": 1.1300296976751496e-05,
      "loss": 0.2078,
      "step": 16600
    },
    {
      "epoch": 1.94,
      "grad_norm": 3.279344277780531,
      "learning_rate": 1.1281112163506923e-05,
      "loss": 0.2004,
      "step": 16620
    },
    {
      "epoch": 1.95,
      "grad_norm": 3.012593064512875,
      "learning_rate": 1.1261922558050012e-05,
      "loss": 0.1992,
      "step": 16640
    },
    {
      "epoch": 1.95,
      "grad_norm": 3.108910147265754,
      "learning_rate": 1.1242728232218743e-05,
      "loss": 0.2006,
      "step": 16660
    },
    {
      "epoch": 1.95,
      "grad_norm": 7.0034021556776525,
      "learning_rate": 1.122352925786875e-05,
      "loss": 0.2147,
      "step": 16680
    },
    {
      "epoch": 1.95,
      "grad_norm": 1.8028644419788564,
      "learning_rate": 1.1204325706873091e-05,
      "loss": 0.2024,
      "step": 16700
    },
    {
      "epoch": 1.96,
      "grad_norm": 2.515029836822745,
      "learning_rate": 1.1185117651121942e-05,
      "loss": 0.2128,
      "step": 16720
    },
    {
      "epoch": 1.96,
      "grad_norm": 5.397378511249806,
      "learning_rate": 1.1165905162522347e-05,
      "loss": 0.2158,
      "step": 16740
    },
    {
      "epoch": 1.96,
      "grad_norm": 3.52178009365664,
      "learning_rate": 1.1146688312997946e-05,
      "loss": 0.2038,
      "step": 16760
    },
    {
      "epoch": 1.96,
      "grad_norm": 4.3841535451132305,
      "learning_rate": 1.1127467174488697e-05,
      "loss": 0.2101,
      "step": 16780
    },
    {
      "epoch": 1.97,
      "grad_norm": 3.869824799113014,
      "learning_rate": 1.1108241818950627e-05,
      "loss": 0.1925,
      "step": 16800
    },
    {
      "epoch": 1.97,
      "grad_norm": 4.329393772389082,
      "learning_rate": 1.1089012318355544e-05,
      "loss": 0.2014,
      "step": 16820
    },
    {
      "epoch": 1.97,
      "grad_norm": 3.781154410675153,
      "learning_rate": 1.1069778744690771e-05,
      "loss": 0.1978,
      "step": 16840
    },
    {
      "epoch": 1.97,
      "grad_norm": 2.96129756942421,
      "learning_rate": 1.1050541169958878e-05,
      "loss": 0.1964,
      "step": 16860
    },
    {
      "epoch": 1.98,
      "grad_norm": 3.730945241165853,
      "learning_rate": 1.1031299666177414e-05,
      "loss": 0.1997,
      "step": 16880
    },
    {
      "epoch": 1.98,
      "grad_norm": 3.6034480060136094,
      "learning_rate": 1.1012054305378647e-05,
      "loss": 0.2064,
      "step": 16900
    },
    {
      "epoch": 1.98,
      "grad_norm": 3.2852161959917856,
      "learning_rate": 1.0992805159609274e-05,
      "loss": 0.2013,
      "step": 16920
    },
    {
      "epoch": 1.98,
      "grad_norm": 3.2808070292511253,
      "learning_rate": 1.0973552300930157e-05,
      "loss": 0.2114,
      "step": 16940
    },
    {
      "epoch": 1.98,
      "grad_norm": 2.8603265565226517,
      "learning_rate": 1.0954295801416068e-05,
      "loss": 0.2045,
      "step": 16960
    },
    {
      "epoch": 1.99,
      "grad_norm": 4.057061413690102,
      "learning_rate": 1.0935035733155402e-05,
      "loss": 0.1918,
      "step": 16980
    },
    {
      "epoch": 1.99,
      "grad_norm": 3.9395052935238755,
      "learning_rate": 1.0915772168249923e-05,
      "loss": 0.213,
      "step": 17000
    },
    {
      "epoch": 1.99,
      "grad_norm": 9.969773195187368,
      "learning_rate": 1.0896505178814471e-05,
      "loss": 0.2098,
      "step": 17020
    },
    {
      "epoch": 1.99,
      "grad_norm": 3.0741477095309926,
      "learning_rate": 1.0877234836976714e-05,
      "loss": 0.2178,
      "step": 17040
    },
    {
      "epoch": 2.0,
      "grad_norm": 3.0763547812686323,
      "learning_rate": 1.0857961214876875e-05,
      "loss": 0.2009,
      "step": 17060
    },
    {
      "epoch": 2.0,
      "grad_norm": 2.581952300141561,
      "learning_rate": 1.0838684384667446e-05,
      "loss": 0.2033,
      "step": 17080
    }
  ],
  "logging_steps": 20,
  "max_steps": 34184,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 4,
  "save_steps": 500,
  "total_flos": 0.0,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}