{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 25638,
  "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
    },
    {
      "epoch": 2.0,
      "grad_norm": 3.187089482219754,
      "learning_rate": 1.0819404418512938e-05,
      "loss": 0.1985,
      "step": 17100
    },
    {
      "epoch": 2.0,
      "grad_norm": 2.6884529730734985,
      "learning_rate": 1.0800121388589594e-05,
      "loss": 0.1817,
      "step": 17120
    },
    {
      "epoch": 2.01,
      "grad_norm": 2.647278478646676,
      "learning_rate": 1.0780835367085132e-05,
      "loss": 0.1963,
      "step": 17140
    },
    {
      "epoch": 2.01,
      "grad_norm": 4.435409913327422,
      "learning_rate": 1.0761546426198472e-05,
      "loss": 0.1834,
      "step": 17160
    },
    {
      "epoch": 2.01,
      "grad_norm": 2.906528582603181,
      "learning_rate": 1.0742254638139455e-05,
      "loss": 0.1775,
      "step": 17180
    },
    {
      "epoch": 2.01,
      "grad_norm": 2.0907428839367546,
      "learning_rate": 1.0722960075128581e-05,
      "loss": 0.1772,
      "step": 17200
    },
    {
      "epoch": 2.01,
      "grad_norm": 2.4500955718726187,
      "learning_rate": 1.070366280939675e-05,
      "loss": 0.1839,
      "step": 17220
    },
    {
      "epoch": 2.02,
      "grad_norm": 2.55696802104814,
      "learning_rate": 1.0684362913184958e-05,
      "loss": 0.1875,
      "step": 17240
    },
    {
      "epoch": 2.02,
      "grad_norm": 3.7661813665264834,
      "learning_rate": 1.0665060458744079e-05,
      "loss": 0.1876,
      "step": 17260
    },
    {
      "epoch": 2.02,
      "grad_norm": 3.3501159818698785,
      "learning_rate": 1.0645755518334535e-05,
      "loss": 0.1872,
      "step": 17280
    },
    {
      "epoch": 2.02,
      "grad_norm": 4.545667769892908,
      "learning_rate": 1.0626448164226067e-05,
      "loss": 0.1732,
      "step": 17300
    },
    {
      "epoch": 2.03,
      "grad_norm": 2.5282252571210164,
      "learning_rate": 1.0607138468697453e-05,
      "loss": 0.1885,
      "step": 17320
    },
    {
      "epoch": 2.03,
      "grad_norm": 3.466764276915403,
      "learning_rate": 1.058782650403624e-05,
      "loss": 0.1836,
      "step": 17340
    },
    {
      "epoch": 2.03,
      "grad_norm": 4.565230297283624,
      "learning_rate": 1.0568512342538455e-05,
      "loss": 0.1808,
      "step": 17360
    },
    {
      "epoch": 2.03,
      "grad_norm": 2.702805693184825,
      "learning_rate": 1.0549196056508364e-05,
      "loss": 0.1799,
      "step": 17380
    },
    {
      "epoch": 2.04,
      "grad_norm": 3.6330853759662816,
      "learning_rate": 1.0529877718258179e-05,
      "loss": 0.1832,
      "step": 17400
    },
    {
      "epoch": 2.04,
      "grad_norm": 3.075596889467387,
      "learning_rate": 1.0510557400107796e-05,
      "loss": 0.1901,
      "step": 17420
    },
    {
      "epoch": 2.04,
      "grad_norm": 4.726698086898242,
      "learning_rate": 1.0491235174384524e-05,
      "loss": 0.1959,
      "step": 17440
    },
    {
      "epoch": 2.04,
      "grad_norm": 4.278041333313194,
      "learning_rate": 1.0471911113422813e-05,
      "loss": 0.1851,
      "step": 17460
    },
    {
      "epoch": 2.05,
      "grad_norm": 3.492295368073298,
      "learning_rate": 1.0452585289563979e-05,
      "loss": 0.1888,
      "step": 17480
    },
    {
      "epoch": 2.05,
      "grad_norm": 2.54721890801404,
      "learning_rate": 1.0433257775155945e-05,
      "loss": 0.188,
      "step": 17500
    },
    {
      "epoch": 2.05,
      "grad_norm": 3.690291043514027,
      "learning_rate": 1.0413928642552955e-05,
      "loss": 0.1848,
      "step": 17520
    },
    {
      "epoch": 2.05,
      "grad_norm": 2.420677528904685,
      "learning_rate": 1.0394597964115322e-05,
      "loss": 0.1812,
      "step": 17540
    },
    {
      "epoch": 2.05,
      "grad_norm": 3.833408368799868,
      "learning_rate": 1.0375265812209132e-05,
      "loss": 0.1873,
      "step": 17560
    },
    {
      "epoch": 2.06,
      "grad_norm": 2.594813668795708,
      "learning_rate": 1.0355932259205996e-05,
      "loss": 0.1869,
      "step": 17580
    },
    {
      "epoch": 2.06,
      "grad_norm": 2.7847592791858102,
      "learning_rate": 1.0336597377482773e-05,
      "loss": 0.1969,
      "step": 17600
    },
    {
      "epoch": 2.06,
      "grad_norm": 3.2040901032730926,
      "learning_rate": 1.0317261239421283e-05,
      "loss": 0.1831,
      "step": 17620
    },
    {
      "epoch": 2.06,
      "grad_norm": 3.7875189399875167,
      "learning_rate": 1.0297923917408064e-05,
      "loss": 0.1832,
      "step": 17640
    },
    {
      "epoch": 2.07,
      "grad_norm": 2.5234037698197986,
      "learning_rate": 1.0278585483834073e-05,
      "loss": 0.1634,
      "step": 17660
    },
    {
      "epoch": 2.07,
      "grad_norm": 2.9933831039856162,
      "learning_rate": 1.0259246011094443e-05,
      "loss": 0.177,
      "step": 17680
    },
    {
      "epoch": 2.07,
      "grad_norm": 3.610660208566244,
      "learning_rate": 1.0239905571588182e-05,
      "loss": 0.1933,
      "step": 17700
    },
    {
      "epoch": 2.07,
      "grad_norm": 2.552193547159119,
      "learning_rate": 1.0220564237717925e-05,
      "loss": 0.1851,
      "step": 17720
    },
    {
      "epoch": 2.08,
      "grad_norm": 2.3741283825079127,
      "learning_rate": 1.0201222081889656e-05,
      "loss": 0.1894,
      "step": 17740
    },
    {
      "epoch": 2.08,
      "grad_norm": 3.0761058400650367,
      "learning_rate": 1.0181879176512441e-05,
      "loss": 0.1848,
      "step": 17760
    },
    {
      "epoch": 2.08,
      "grad_norm": 2.8149565247181823,
      "learning_rate": 1.0162535593998133e-05,
      "loss": 0.1832,
      "step": 17780
    },
    {
      "epoch": 2.08,
      "grad_norm": 3.3079895404618025,
      "learning_rate": 1.0143191406761144e-05,
      "loss": 0.2008,
      "step": 17800
    },
    {
      "epoch": 2.09,
      "grad_norm": 3.4780019459030735,
      "learning_rate": 1.0123846687218129e-05,
      "loss": 0.177,
      "step": 17820
    },
    {
      "epoch": 2.09,
      "grad_norm": 2.956742911346678,
      "learning_rate": 1.010450150778775e-05,
      "loss": 0.1829,
      "step": 17840
    },
    {
      "epoch": 2.09,
      "grad_norm": 4.258286867927119,
      "learning_rate": 1.0085155940890386e-05,
      "loss": 0.1738,
      "step": 17860
    },
    {
      "epoch": 2.09,
      "grad_norm": 3.661775245173711,
      "learning_rate": 1.0065810058947862e-05,
      "loss": 0.1764,
      "step": 17880
    },
    {
      "epoch": 2.09,
      "grad_norm": 3.0954483456868456,
      "learning_rate": 1.0046463934383193e-05,
      "loss": 0.1961,
      "step": 17900
    },
    {
      "epoch": 2.1,
      "grad_norm": 3.6220892697165,
      "learning_rate": 1.0027117639620287e-05,
      "loss": 0.197,
      "step": 17920
    },
    {
      "epoch": 2.1,
      "grad_norm": 2.184218752270426,
      "learning_rate": 1.0007771247083709e-05,
      "loss": 0.1813,
      "step": 17940
    },
    {
      "epoch": 2.1,
      "grad_norm": 6.033162974838077,
      "learning_rate": 9.988424829198365e-06,
      "loss": 0.201,
      "step": 17960
    },
    {
      "epoch": 2.1,
      "grad_norm": 3.198884119737886,
      "learning_rate": 9.969078458389281e-06,
      "loss": 0.1868,
      "step": 17980
    },
    {
      "epoch": 2.11,
      "grad_norm": 2.1035316062893963,
      "learning_rate": 9.949732207081292e-06,
      "loss": 0.1803,
      "step": 18000
    },
    {
      "epoch": 2.11,
      "grad_norm": 2.7102647105925195,
      "learning_rate": 9.930386147698793e-06,
      "loss": 0.1879,
      "step": 18020
    },
    {
      "epoch": 2.11,
      "grad_norm": 2.797729106040711,
      "learning_rate": 9.911040352665449e-06,
      "loss": 0.1763,
      "step": 18040
    },
    {
      "epoch": 2.11,
      "grad_norm": 2.5070535337242945,
      "learning_rate": 9.891694894403947e-06,
      "loss": 0.1761,
      "step": 18060
    },
    {
      "epoch": 2.12,
      "grad_norm": 2.7653928325208987,
      "learning_rate": 9.872349845335711e-06,
      "loss": 0.1874,
      "step": 18080
    },
    {
      "epoch": 2.12,
      "grad_norm": 2.4731031258102307,
      "learning_rate": 9.853005277880631e-06,
      "loss": 0.1695,
      "step": 18100
    },
    {
      "epoch": 2.12,
      "grad_norm": 4.932234068564112,
      "learning_rate": 9.833661264456793e-06,
      "loss": 0.1788,
      "step": 18120
    },
    {
      "epoch": 2.12,
      "grad_norm": 2.7969357787619646,
      "learning_rate": 9.814317877480212e-06,
      "loss": 0.1864,
      "step": 18140
    },
    {
      "epoch": 2.12,
      "grad_norm": 6.958001987387912,
      "learning_rate": 9.794975189364554e-06,
      "loss": 0.1906,
      "step": 18160
    },
    {
      "epoch": 2.13,
      "grad_norm": 13.874898274366679,
      "learning_rate": 9.775633272520874e-06,
      "loss": 0.1887,
      "step": 18180
    },
    {
      "epoch": 2.13,
      "grad_norm": 2.4563842309668944,
      "learning_rate": 9.756292199357337e-06,
      "loss": 0.1718,
      "step": 18200
    },
    {
      "epoch": 2.13,
      "grad_norm": 3.701500516205849,
      "learning_rate": 9.736952042278947e-06,
      "loss": 0.2008,
      "step": 18220
    },
    {
      "epoch": 2.13,
      "grad_norm": 3.8540761301215185,
      "learning_rate": 9.717612873687285e-06,
      "loss": 0.1944,
      "step": 18240
    },
    {
      "epoch": 2.14,
      "grad_norm": 2.5734610245537937,
      "learning_rate": 9.698274765980221e-06,
      "loss": 0.1953,
      "step": 18260
    },
    {
      "epoch": 2.14,
      "grad_norm": 3.6988039197017955,
      "learning_rate": 9.678937791551665e-06,
      "loss": 0.1782,
      "step": 18280
    },
    {
      "epoch": 2.14,
      "grad_norm": 3.5093985117721687,
      "learning_rate": 9.659602022791278e-06,
      "loss": 0.1737,
      "step": 18300
    },
    {
      "epoch": 2.14,
      "grad_norm": 2.806102002835496,
      "learning_rate": 9.640267532084208e-06,
      "loss": 0.1947,
      "step": 18320
    },
    {
      "epoch": 2.15,
      "grad_norm": 4.637926543739354,
      "learning_rate": 9.620934391810822e-06,
      "loss": 0.1896,
      "step": 18340
    },
    {
      "epoch": 2.15,
      "grad_norm": 4.20279507955276,
      "learning_rate": 9.601602674346428e-06,
      "loss": 0.1808,
      "step": 18360
    },
    {
      "epoch": 2.15,
      "grad_norm": 3.93610696522427,
      "learning_rate": 9.582272452061003e-06,
      "loss": 0.1786,
      "step": 18380
    },
    {
      "epoch": 2.15,
      "grad_norm": 2.5104348800652727,
      "learning_rate": 9.562943797318939e-06,
      "loss": 0.1715,
      "step": 18400
    },
    {
      "epoch": 2.16,
      "grad_norm": 5.142293270927721,
      "learning_rate": 9.543616782478752e-06,
      "loss": 0.1771,
      "step": 18420
    },
    {
      "epoch": 2.16,
      "grad_norm": 2.7896695371322764,
      "learning_rate": 9.52429147989282e-06,
      "loss": 0.181,
      "step": 18440
    },
    {
      "epoch": 2.16,
      "grad_norm": 2.8005597440629466,
      "learning_rate": 9.50496796190711e-06,
      "loss": 0.1936,
      "step": 18460
    },
    {
      "epoch": 2.16,
      "grad_norm": 3.2682102319338617,
      "learning_rate": 9.485646300860904e-06,
      "loss": 0.1943,
      "step": 18480
    },
    {
      "epoch": 2.16,
      "grad_norm": 3.1744273773216425,
      "learning_rate": 9.466326569086545e-06,
      "loss": 0.1826,
      "step": 18500
    },
    {
      "epoch": 2.17,
      "grad_norm": 3.1673830292863894,
      "learning_rate": 9.447008838909144e-06,
      "loss": 0.1749,
      "step": 18520
    },
    {
      "epoch": 2.17,
      "grad_norm": 2.2288241494638825,
      "learning_rate": 9.42769318264632e-06,
      "loss": 0.1826,
      "step": 18540
    },
    {
      "epoch": 2.17,
      "grad_norm": 51.583019118001666,
      "learning_rate": 9.408379672607929e-06,
      "loss": 0.1862,
      "step": 18560
    },
    {
      "epoch": 2.17,
      "grad_norm": 3.645865274925182,
      "learning_rate": 9.389068381095792e-06,
      "loss": 0.1833,
      "step": 18580
    },
    {
      "epoch": 2.18,
      "grad_norm": 3.6838881050372,
      "learning_rate": 9.369759380403422e-06,
      "loss": 0.1863,
      "step": 18600
    },
    {
      "epoch": 2.18,
      "grad_norm": 2.20917195165408,
      "learning_rate": 9.35045274281577e-06,
      "loss": 0.1701,
      "step": 18620
    },
    {
      "epoch": 2.18,
      "grad_norm": 3.5141625647505466,
      "learning_rate": 9.331148540608918e-06,
      "loss": 0.1827,
      "step": 18640
    },
    {
      "epoch": 2.18,
      "grad_norm": 3.633798116884379,
      "learning_rate": 9.311846846049849e-06,
      "loss": 0.1821,
      "step": 18660
    },
    {
      "epoch": 2.19,
      "grad_norm": 2.9164096900721406,
      "learning_rate": 9.29254773139615e-06,
      "loss": 0.1884,
      "step": 18680
    },
    {
      "epoch": 2.19,
      "grad_norm": 2.5084531922287368,
      "learning_rate": 9.273251268895755e-06,
      "loss": 0.17,
      "step": 18700
    },
    {
      "epoch": 2.19,
      "grad_norm": 3.6314194500201795,
      "learning_rate": 9.253957530786663e-06,
      "loss": 0.1865,
      "step": 18720
    },
    {
      "epoch": 2.19,
      "grad_norm": 5.532515903593115,
      "learning_rate": 9.234666589296678e-06,
      "loss": 0.1784,
      "step": 18740
    },
    {
      "epoch": 2.2,
      "grad_norm": 2.285776324792612,
      "learning_rate": 9.215378516643135e-06,
      "loss": 0.1966,
      "step": 18760
    },
    {
      "epoch": 2.2,
      "grad_norm": 1.6912381482889705,
      "learning_rate": 9.19609338503263e-06,
      "loss": 0.1684,
      "step": 18780
    },
    {
      "epoch": 2.2,
      "grad_norm": 4.691665019728991,
      "learning_rate": 9.176811266660744e-06,
      "loss": 0.1989,
      "step": 18800
    },
    {
      "epoch": 2.2,
      "grad_norm": 3.5520723553410516,
      "learning_rate": 9.157532233711778e-06,
      "loss": 0.1919,
      "step": 18820
    },
    {
      "epoch": 2.2,
      "grad_norm": 2.354445404385382,
      "learning_rate": 9.138256358358495e-06,
      "loss": 0.1646,
      "step": 18840
    },
    {
      "epoch": 2.21,
      "grad_norm": 2.3996097048660343,
      "learning_rate": 9.11898371276182e-06,
      "loss": 0.1848,
      "step": 18860
    },
    {
      "epoch": 2.21,
      "grad_norm": 3.179901640708416,
      "learning_rate": 9.099714369070602e-06,
      "loss": 0.1848,
      "step": 18880
    },
    {
      "epoch": 2.21,
      "grad_norm": 4.23191486133106,
      "learning_rate": 9.080448399421314e-06,
      "loss": 0.1988,
      "step": 18900
    },
    {
      "epoch": 2.21,
      "grad_norm": 2.6002312997490864,
      "learning_rate": 9.061185875937807e-06,
      "loss": 0.174,
      "step": 18920
    },
    {
      "epoch": 2.22,
      "grad_norm": 2.203729533047983,
      "learning_rate": 9.041926870731034e-06,
      "loss": 0.1899,
      "step": 18940
    },
    {
      "epoch": 2.22,
      "grad_norm": 2.4547031453081436,
      "learning_rate": 9.022671455898773e-06,
      "loss": 0.1906,
      "step": 18960
    },
    {
      "epoch": 2.22,
      "grad_norm": 3.0866563949523904,
      "learning_rate": 9.003419703525355e-06,
      "loss": 0.1809,
      "step": 18980
    },
    {
      "epoch": 2.22,
      "grad_norm": 2.2096591664201033,
      "learning_rate": 8.98417168568141e-06,
      "loss": 0.1886,
      "step": 19000
    },
    {
      "epoch": 2.23,
      "grad_norm": 2.06152413140976,
      "learning_rate": 8.96492747442358e-06,
      "loss": 0.1755,
      "step": 19020
    },
    {
      "epoch": 2.23,
      "grad_norm": 2.3654089563201643,
      "learning_rate": 8.945687141794268e-06,
      "loss": 0.1846,
      "step": 19040
    },
    {
      "epoch": 2.23,
      "grad_norm": 3.656689609280586,
      "learning_rate": 8.926450759821344e-06,
      "loss": 0.1778,
      "step": 19060
    },
    {
      "epoch": 2.23,
      "grad_norm": 3.573948851020321,
      "learning_rate": 8.907218400517894e-06,
      "loss": 0.1856,
      "step": 19080
    },
    {
      "epoch": 2.23,
      "grad_norm": 3.059058610806695,
      "learning_rate": 8.887990135881947e-06,
      "loss": 0.1871,
      "step": 19100
    },
    {
      "epoch": 2.24,
      "grad_norm": 2.9163171467938107,
      "learning_rate": 8.868766037896199e-06,
      "loss": 0.1759,
      "step": 19120
    },
    {
      "epoch": 2.24,
      "grad_norm": 2.690429067178921,
      "learning_rate": 8.84954617852775e-06,
      "loss": 0.1983,
      "step": 19140
    },
    {
      "epoch": 2.24,
      "grad_norm": 3.48897409766989,
      "learning_rate": 8.830330629727828e-06,
      "loss": 0.1824,
      "step": 19160
    },
    {
      "epoch": 2.24,
      "grad_norm": 2.505307095828609,
      "learning_rate": 8.811119463431534e-06,
      "loss": 0.1861,
      "step": 19180
    },
    {
      "epoch": 2.25,
      "grad_norm": 2.860339392951664,
      "learning_rate": 8.791912751557554e-06,
      "loss": 0.1875,
      "step": 19200
    },
    {
      "epoch": 2.25,
      "grad_norm": 2.3013803320616724,
      "learning_rate": 8.772710566007902e-06,
      "loss": 0.1734,
      "step": 19220
    },
    {
      "epoch": 2.25,
      "grad_norm": 3.269140660006422,
      "learning_rate": 8.753512978667644e-06,
      "loss": 0.1717,
      "step": 19240
    },
    {
      "epoch": 2.25,
      "grad_norm": 2.5218023429151177,
      "learning_rate": 8.734320061404637e-06,
      "loss": 0.194,
      "step": 19260
    },
    {
      "epoch": 2.26,
      "grad_norm": 3.1714206614450915,
      "learning_rate": 8.71513188606925e-06,
      "loss": 0.1797,
      "step": 19280
    },
    {
      "epoch": 2.26,
      "grad_norm": 2.008888164295546,
      "learning_rate": 8.69594852449411e-06,
      "loss": 0.1928,
      "step": 19300
    },
    {
      "epoch": 2.26,
      "grad_norm": 3.074172062016426,
      "learning_rate": 8.676770048493806e-06,
      "loss": 0.1933,
      "step": 19320
    },
    {
      "epoch": 2.26,
      "grad_norm": 3.0341910026543117,
      "learning_rate": 8.657596529864656e-06,
      "loss": 0.1905,
      "step": 19340
    },
    {
      "epoch": 2.27,
      "grad_norm": 2.6160519086313183,
      "learning_rate": 8.638428040384407e-06,
      "loss": 0.1963,
      "step": 19360
    },
    {
      "epoch": 2.27,
      "grad_norm": 7.1675419235999245,
      "learning_rate": 8.619264651811987e-06,
      "loss": 0.2065,
      "step": 19380
    },
    {
      "epoch": 2.27,
      "grad_norm": 5.44898895544822,
      "learning_rate": 8.60010643588722e-06,
      "loss": 0.1876,
      "step": 19400
    },
    {
      "epoch": 2.27,
      "grad_norm": 2.248782358238912,
      "learning_rate": 8.580953464330571e-06,
      "loss": 0.1908,
      "step": 19420
    },
    {
      "epoch": 2.27,
      "grad_norm": 2.6269853441042317,
      "learning_rate": 8.561805808842873e-06,
      "loss": 0.1737,
      "step": 19440
    },
    {
      "epoch": 2.28,
      "grad_norm": 2.654278461092079,
      "learning_rate": 8.542663541105058e-06,
      "loss": 0.1735,
      "step": 19460
    },
    {
      "epoch": 2.28,
      "grad_norm": 3.5278579489207997,
      "learning_rate": 8.523526732777881e-06,
      "loss": 0.1867,
      "step": 19480
    },
    {
      "epoch": 2.28,
      "grad_norm": 2.5779515063546503,
      "learning_rate": 8.504395455501673e-06,
      "loss": 0.1661,
      "step": 19500
    },
    {
      "epoch": 2.28,
      "grad_norm": 3.514009367259985,
      "learning_rate": 8.485269780896044e-06,
      "loss": 0.1862,
      "step": 19520
    },
    {
      "epoch": 2.29,
      "grad_norm": 2.5847053473530592,
      "learning_rate": 8.466149780559642e-06,
      "loss": 0.1879,
      "step": 19540
    },
    {
      "epoch": 2.29,
      "grad_norm": 2.9447271843079865,
      "learning_rate": 8.447035526069868e-06,
      "loss": 0.1923,
      "step": 19560
    },
    {
      "epoch": 2.29,
      "grad_norm": 1.9904986834010254,
      "learning_rate": 8.427927088982608e-06,
      "loss": 0.1758,
      "step": 19580
    },
    {
      "epoch": 2.29,
      "grad_norm": 3.0672202032272557,
      "learning_rate": 8.408824540831977e-06,
      "loss": 0.1809,
      "step": 19600
    },
    {
      "epoch": 2.3,
      "grad_norm": 2.2937829620737027,
      "learning_rate": 8.389727953130044e-06,
      "loss": 0.1839,
      "step": 19620
    },
    {
      "epoch": 2.3,
      "grad_norm": 3.797253750720813,
      "learning_rate": 8.370637397366563e-06,
      "loss": 0.1839,
      "step": 19640
    },
    {
      "epoch": 2.3,
      "grad_norm": 2.796604164692134,
      "learning_rate": 8.351552945008701e-06,
      "loss": 0.1743,
      "step": 19660
    },
    {
      "epoch": 2.3,
      "grad_norm": 4.4643765903487695,
      "learning_rate": 8.332474667500786e-06,
      "loss": 0.1803,
      "step": 19680
    },
    {
      "epoch": 2.31,
      "grad_norm": 3.6583758311891637,
      "learning_rate": 8.313402636264024e-06,
      "loss": 0.186,
      "step": 19700
    },
    {
      "epoch": 2.31,
      "grad_norm": 3.7600154645074215,
      "learning_rate": 8.294336922696247e-06,
      "loss": 0.173,
      "step": 19720
    },
    {
      "epoch": 2.31,
      "grad_norm": 2.943679478280475,
      "learning_rate": 8.275277598171615e-06,
      "loss": 0.1901,
      "step": 19740
    },
    {
      "epoch": 2.31,
      "grad_norm": 2.521023283388264,
      "learning_rate": 8.256224734040397e-06,
      "loss": 0.1853,
      "step": 19760
    },
    {
      "epoch": 2.31,
      "grad_norm": 9.148411230347637,
      "learning_rate": 8.237178401628654e-06,
      "loss": 0.1889,
      "step": 19780
    },
    {
      "epoch": 2.32,
      "grad_norm": 2.3073382118186014,
      "learning_rate": 8.218138672238008e-06,
      "loss": 0.1904,
      "step": 19800
    },
    {
      "epoch": 2.32,
      "grad_norm": 2.4455256262578975,
      "learning_rate": 8.19910561714536e-06,
      "loss": 0.1942,
      "step": 19820
    },
    {
      "epoch": 2.32,
      "grad_norm": 3.1936382953495905,
      "learning_rate": 8.180079307602623e-06,
      "loss": 0.1732,
      "step": 19840
    },
    {
      "epoch": 2.32,
      "grad_norm": 2.5890410547283422,
      "learning_rate": 8.161059814836456e-06,
      "loss": 0.1905,
      "step": 19860
    },
    {
      "epoch": 2.33,
      "grad_norm": 2.3357113119483905,
      "learning_rate": 8.142047210048e-06,
      "loss": 0.1868,
      "step": 19880
    },
    {
      "epoch": 2.33,
      "grad_norm": 3.8518130702595963,
      "learning_rate": 8.123041564412619e-06,
      "loss": 0.1772,
      "step": 19900
    },
    {
      "epoch": 2.33,
      "grad_norm": 3.2440906500139315,
      "learning_rate": 8.104042949079605e-06,
      "loss": 0.1894,
      "step": 19920
    },
    {
      "epoch": 2.33,
      "grad_norm": 2.8690166407648023,
      "learning_rate": 8.085051435171953e-06,
      "loss": 0.1972,
      "step": 19940
    },
    {
      "epoch": 2.34,
      "grad_norm": 5.11949923182371,
      "learning_rate": 8.06606709378606e-06,
      "loss": 0.1646,
      "step": 19960
    },
    {
      "epoch": 2.34,
      "grad_norm": 3.6058833420132292,
      "learning_rate": 8.04708999599148e-06,
      "loss": 0.1686,
      "step": 19980
    },
    {
      "epoch": 2.34,
      "grad_norm": 3.190984429599093,
      "learning_rate": 8.028120212830638e-06,
      "loss": 0.196,
      "step": 20000
    },
    {
      "epoch": 2.34,
      "grad_norm": 3.5551243775576666,
      "learning_rate": 8.009157815318589e-06,
      "loss": 0.1698,
      "step": 20020
    },
    {
      "epoch": 2.34,
      "grad_norm": 3.102970447513887,
      "learning_rate": 7.99020287444273e-06,
      "loss": 0.2009,
      "step": 20040
    },
    {
      "epoch": 2.35,
      "grad_norm": 3.059829014258245,
      "learning_rate": 7.971255461162554e-06,
      "loss": 0.1845,
      "step": 20060
    },
    {
      "epoch": 2.35,
      "grad_norm": 2.030503708119072,
      "learning_rate": 7.952315646409361e-06,
      "loss": 0.1703,
      "step": 20080
    },
    {
      "epoch": 2.35,
      "grad_norm": 3.7340316295377285,
      "learning_rate": 7.933383501086012e-06,
      "loss": 0.1806,
      "step": 20100
    },
    {
      "epoch": 2.35,
      "grad_norm": 2.381515800479722,
      "learning_rate": 7.914459096066656e-06,
      "loss": 0.1764,
      "step": 20120
    },
    {
      "epoch": 2.36,
      "grad_norm": 3.213566578542309,
      "learning_rate": 7.895542502196467e-06,
      "loss": 0.1943,
      "step": 20140
    },
    {
      "epoch": 2.36,
      "grad_norm": 3.5176444124185204,
      "learning_rate": 7.876633790291378e-06,
      "loss": 0.1798,
      "step": 20160
    },
    {
      "epoch": 2.36,
      "grad_norm": 2.418359580060765,
      "learning_rate": 7.85773303113781e-06,
      "loss": 0.1846,
      "step": 20180
    },
    {
      "epoch": 2.36,
      "grad_norm": 3.620888713476074,
      "learning_rate": 7.838840295492415e-06,
      "loss": 0.1838,
      "step": 20200
    },
    {
      "epoch": 2.37,
      "grad_norm": 2.332237145106214,
      "learning_rate": 7.81995565408181e-06,
      "loss": 0.1828,
      "step": 20220
    },
    {
      "epoch": 2.37,
      "grad_norm": 3.043301095307801,
      "learning_rate": 7.80107917760231e-06,
      "loss": 0.1766,
      "step": 20240
    },
    {
      "epoch": 2.37,
      "grad_norm": 3.2505637926978066,
      "learning_rate": 7.782210936719661e-06,
      "loss": 0.1846,
      "step": 20260
    },
    {
      "epoch": 2.37,
      "grad_norm": 2.9027355643269837,
      "learning_rate": 7.763351002068778e-06,
      "loss": 0.1754,
      "step": 20280
    },
    {
      "epoch": 2.38,
      "grad_norm": 3.373330869133493,
      "learning_rate": 7.744499444253488e-06,
      "loss": 0.1889,
      "step": 20300
    },
    {
      "epoch": 2.38,
      "grad_norm": 2.562535634607174,
      "learning_rate": 7.725656333846251e-06,
      "loss": 0.1905,
      "step": 20320
    },
    {
      "epoch": 2.38,
      "grad_norm": 3.269212276527265,
      "learning_rate": 7.706821741387906e-06,
      "loss": 0.1735,
      "step": 20340
    },
    {
      "epoch": 2.38,
      "grad_norm": 2.694612185401067,
      "learning_rate": 7.687995737387405e-06,
      "loss": 0.1814,
      "step": 20360
    },
    {
      "epoch": 2.38,
      "grad_norm": 2.550536910734683,
      "learning_rate": 7.669178392321549e-06,
      "loss": 0.1951,
      "step": 20380
    },
    {
      "epoch": 2.39,
      "grad_norm": 3.7192454288549706,
      "learning_rate": 7.65036977663472e-06,
      "loss": 0.1811,
      "step": 20400
    },
    {
      "epoch": 2.39,
      "grad_norm": 4.583412424040906,
      "learning_rate": 7.631569960738625e-06,
      "loss": 0.1912,
      "step": 20420
    },
    {
      "epoch": 2.39,
      "grad_norm": 3.6102709958551844,
      "learning_rate": 7.612779015012024e-06,
      "loss": 0.1857,
      "step": 20440
    },
    {
      "epoch": 2.39,
      "grad_norm": 2.132609263367914,
      "learning_rate": 7.593997009800476e-06,
      "loss": 0.1852,
      "step": 20460
    },
    {
      "epoch": 2.4,
      "grad_norm": 3.4880324528085604,
      "learning_rate": 7.575224015416064e-06,
      "loss": 0.1715,
      "step": 20480
    },
    {
      "epoch": 2.4,
      "grad_norm": 1.8270056794678053,
      "learning_rate": 7.5564601021371455e-06,
      "loss": 0.1859,
      "step": 20500
    },
    {
      "epoch": 2.4,
      "grad_norm": 3.2661614228423366,
      "learning_rate": 7.537705340208075e-06,
      "loss": 0.1763,
      "step": 20520
    },
    {
      "epoch": 2.4,
      "grad_norm": 2.652309411884222,
      "learning_rate": 7.518959799838952e-06,
      "loss": 0.1848,
      "step": 20540
    },
    {
      "epoch": 2.41,
      "grad_norm": 2.840447321036168,
      "learning_rate": 7.5002235512053515e-06,
      "loss": 0.1885,
      "step": 20560
    },
    {
      "epoch": 2.41,
      "grad_norm": 2.8852660946629722,
      "learning_rate": 7.481496664448073e-06,
      "loss": 0.1823,
      "step": 20580
    },
    {
      "epoch": 2.41,
      "grad_norm": 3.287773695311469,
      "learning_rate": 7.462779209672856e-06,
      "loss": 0.1617,
      "step": 20600
    },
    {
      "epoch": 2.41,
      "grad_norm": 3.3685486912072653,
      "learning_rate": 7.444071256950138e-06,
      "loss": 0.1796,
      "step": 20620
    },
    {
      "epoch": 2.42,
      "grad_norm": 4.499428818798512,
      "learning_rate": 7.425372876314783e-06,
      "loss": 0.2029,
      "step": 20640
    },
    {
      "epoch": 2.42,
      "grad_norm": 3.1348393987132894,
      "learning_rate": 7.406684137765827e-06,
      "loss": 0.1828,
      "step": 20660
    },
    {
      "epoch": 2.42,
      "grad_norm": 2.3227046875533435,
      "learning_rate": 7.388005111266197e-06,
      "loss": 0.1909,
      "step": 20680
    },
    {
      "epoch": 2.42,
      "grad_norm": 3.142369938528309,
      "learning_rate": 7.369335866742471e-06,
      "loss": 0.1769,
      "step": 20700
    },
    {
      "epoch": 2.42,
      "grad_norm": 2.4074339492943193,
      "learning_rate": 7.350676474084608e-06,
      "loss": 0.1833,
      "step": 20720
    },
    {
      "epoch": 2.43,
      "grad_norm": 2.4803162054779415,
      "learning_rate": 7.3320270031456855e-06,
      "loss": 0.1754,
      "step": 20740
    },
    {
      "epoch": 2.43,
      "grad_norm": 1.998861465638246,
      "learning_rate": 7.31338752374163e-06,
      "loss": 0.1832,
      "step": 20760
    },
    {
      "epoch": 2.43,
      "grad_norm": 3.1286831322602557,
      "learning_rate": 7.294758105650972e-06,
      "loss": 0.1896,
      "step": 20780
    },
    {
      "epoch": 2.43,
      "grad_norm": 2.5799109544099244,
      "learning_rate": 7.276138818614579e-06,
      "loss": 0.1748,
      "step": 20800
    },
    {
      "epoch": 2.44,
      "grad_norm": 3.3287032077321066,
      "learning_rate": 7.257529732335385e-06,
      "loss": 0.1987,
      "step": 20820
    },
    {
      "epoch": 2.44,
      "grad_norm": 6.680428732630699,
      "learning_rate": 7.238930916478139e-06,
      "loss": 0.1866,
      "step": 20840
    },
    {
      "epoch": 2.44,
      "grad_norm": 2.5614379915099614,
      "learning_rate": 7.2203424406691416e-06,
      "loss": 0.1848,
      "step": 20860
    },
    {
      "epoch": 2.44,
      "grad_norm": 3.3339886656936364,
      "learning_rate": 7.201764374495986e-06,
      "loss": 0.1811,
      "step": 20880
    },
    {
      "epoch": 2.45,
      "grad_norm": 3.0393813095952913,
      "learning_rate": 7.1831967875072935e-06,
      "loss": 0.1604,
      "step": 20900
    },
    {
      "epoch": 2.45,
      "grad_norm": 1.7498486998411649,
      "learning_rate": 7.1646397492124605e-06,
      "loss": 0.1858,
      "step": 20920
    },
    {
      "epoch": 2.45,
      "grad_norm": 3.448053147917569,
      "learning_rate": 7.146093329081384e-06,
      "loss": 0.1735,
      "step": 20940
    },
    {
      "epoch": 2.45,
      "grad_norm": 3.7318359426110197,
      "learning_rate": 7.127557596544221e-06,
      "loss": 0.1803,
      "step": 20960
    },
    {
      "epoch": 2.45,
      "grad_norm": 2.294152755094677,
      "learning_rate": 7.109032620991115e-06,
      "loss": 0.1812,
      "step": 20980
    },
    {
      "epoch": 2.46,
      "grad_norm": 2.695852120602769,
      "learning_rate": 7.0905184717719374e-06,
      "loss": 0.1698,
      "step": 21000
    },
    {
      "epoch": 2.46,
      "grad_norm": 1.7682884854314216,
      "learning_rate": 7.072015218196033e-06,
      "loss": 0.1794,
      "step": 21020
    },
    {
      "epoch": 2.46,
      "grad_norm": 2.8223377680550117,
      "learning_rate": 7.053522929531956e-06,
      "loss": 0.1879,
      "step": 21040
    },
    {
      "epoch": 2.46,
      "grad_norm": 2.3633329023281004,
      "learning_rate": 7.035041675007215e-06,
      "loss": 0.1774,
      "step": 21060
    },
    {
      "epoch": 2.47,
      "grad_norm": 2.5439977014459267,
      "learning_rate": 7.016571523808011e-06,
      "loss": 0.1801,
      "step": 21080
    },
    {
      "epoch": 2.47,
      "grad_norm": 3.017965722730571,
      "learning_rate": 6.998112545078976e-06,
      "loss": 0.1835,
      "step": 21100
    },
    {
      "epoch": 2.47,
      "grad_norm": 3.2340560977012465,
      "learning_rate": 6.9796648079229154e-06,
      "loss": 0.183,
      "step": 21120
    },
    {
      "epoch": 2.47,
      "grad_norm": 2.7076792196060855,
      "learning_rate": 6.961228381400557e-06,
      "loss": 0.17,
      "step": 21140
    },
    {
      "epoch": 2.48,
      "grad_norm": 2.5294049921088604,
      "learning_rate": 6.9428033345302836e-06,
      "loss": 0.1748,
      "step": 21160
    },
    {
      "epoch": 2.48,
      "grad_norm": 1.9755482842123588,
      "learning_rate": 6.924389736287876e-06,
      "loss": 0.1797,
      "step": 21180
    },
    {
      "epoch": 2.48,
      "grad_norm": 2.6170283482684837,
      "learning_rate": 6.905987655606256e-06,
      "loss": 0.1742,
      "step": 21200
    },
    {
      "epoch": 2.48,
      "grad_norm": 2.706014158619269,
      "learning_rate": 6.887597161375228e-06,
      "loss": 0.1881,
      "step": 21220
    },
    {
      "epoch": 2.49,
      "grad_norm": 2.4950356785549106,
      "learning_rate": 6.869218322441225e-06,
      "loss": 0.1644,
      "step": 21240
    },
    {
      "epoch": 2.49,
      "grad_norm": 4.077341289413537,
      "learning_rate": 6.850851207607047e-06,
      "loss": 0.1901,
      "step": 21260
    },
    {
      "epoch": 2.49,
      "grad_norm": 2.5754376826477765,
      "learning_rate": 6.832495885631597e-06,
      "loss": 0.1783,
      "step": 21280
    },
    {
      "epoch": 2.49,
      "grad_norm": 6.354918922129214,
      "learning_rate": 6.814152425229636e-06,
      "loss": 0.1678,
      "step": 21300
    },
    {
      "epoch": 2.49,
      "grad_norm": 2.394268674906887,
      "learning_rate": 6.795820895071523e-06,
      "loss": 0.1782,
      "step": 21320
    },
    {
      "epoch": 2.5,
      "grad_norm": 3.186718863720809,
      "learning_rate": 6.777501363782952e-06,
      "loss": 0.175,
      "step": 21340
    },
    {
      "epoch": 2.5,
      "grad_norm": 2.5749398641138184,
      "learning_rate": 6.759193899944691e-06,
      "loss": 0.1827,
      "step": 21360
    },
    {
      "epoch": 2.5,
      "grad_norm": 2.8466404907249983,
      "learning_rate": 6.740898572092345e-06,
      "loss": 0.1846,
      "step": 21380
    },
    {
      "epoch": 2.5,
      "grad_norm": 1.8342194005424137,
      "learning_rate": 6.722615448716077e-06,
      "loss": 0.1707,
      "step": 21400
    },
    {
      "epoch": 2.51,
      "grad_norm": 3.129190153464715,
      "learning_rate": 6.704344598260372e-06,
      "loss": 0.1676,
      "step": 21420
    },
    {
      "epoch": 2.51,
      "grad_norm": 3.723062274962248,
      "learning_rate": 6.6860860891237524e-06,
      "loss": 0.1897,
      "step": 21440
    },
    {
      "epoch": 2.51,
      "grad_norm": 5.087016329631688,
      "learning_rate": 6.667839989658562e-06,
      "loss": 0.1752,
      "step": 21460
    },
    {
      "epoch": 2.51,
      "grad_norm": 3.2186600755889,
      "learning_rate": 6.649606368170671e-06,
      "loss": 0.1899,
      "step": 21480
    },
    {
      "epoch": 2.52,
      "grad_norm": 3.005904268619833,
      "learning_rate": 6.631385292919246e-06,
      "loss": 0.1844,
      "step": 21500
    },
    {
      "epoch": 2.52,
      "grad_norm": 2.6779566031964586,
      "learning_rate": 6.613176832116484e-06,
      "loss": 0.1836,
      "step": 21520
    },
    {
      "epoch": 2.52,
      "grad_norm": 3.702777315066449,
      "learning_rate": 6.594981053927354e-06,
      "loss": 0.1896,
      "step": 21540
    },
    {
      "epoch": 2.52,
      "grad_norm": 3.289749907290013,
      "learning_rate": 6.576798026469352e-06,
      "loss": 0.1784,
      "step": 21560
    },
    {
      "epoch": 2.53,
      "grad_norm": 2.1363438027971995,
      "learning_rate": 6.558627817812236e-06,
      "loss": 0.1841,
      "step": 21580
    },
    {
      "epoch": 2.53,
      "grad_norm": 2.8492415506482476,
      "learning_rate": 6.540470495977783e-06,
      "loss": 0.1745,
      "step": 21600
    },
    {
      "epoch": 2.53,
      "grad_norm": 2.4044134884900337,
      "learning_rate": 6.522326128939518e-06,
      "loss": 0.1825,
      "step": 21620
    },
    {
      "epoch": 2.53,
      "grad_norm": 2.280363668416859,
      "learning_rate": 6.504194784622475e-06,
      "loss": 0.1967,
      "step": 21640
    },
    {
      "epoch": 2.53,
      "grad_norm": 5.67258820896823,
      "learning_rate": 6.486076530902932e-06,
      "loss": 0.1881,
      "step": 21660
    },
    {
      "epoch": 2.54,
      "grad_norm": 2.4280569609507388,
      "learning_rate": 6.467971435608168e-06,
      "loss": 0.18,
      "step": 21680
    },
    {
      "epoch": 2.54,
      "grad_norm": 2.7465414493529705,
      "learning_rate": 6.449879566516194e-06,
      "loss": 0.1893,
      "step": 21700
    },
    {
      "epoch": 2.54,
      "grad_norm": 3.7485576398899934,
      "learning_rate": 6.4318009913555115e-06,
      "loss": 0.1684,
      "step": 21720
    },
    {
      "epoch": 2.54,
      "grad_norm": 2.1872776463442554,
      "learning_rate": 6.413735777804855e-06,
      "loss": 0.1695,
      "step": 21740
    },
    {
      "epoch": 2.55,
      "grad_norm": 2.6147402124846324,
      "learning_rate": 6.395683993492941e-06,
      "loss": 0.1918,
      "step": 21760
    },
    {
      "epoch": 2.55,
      "grad_norm": 3.2226855467418334,
      "learning_rate": 6.377645705998208e-06,
      "loss": 0.1685,
      "step": 21780
    },
    {
      "epoch": 2.55,
      "grad_norm": 4.304393945228268,
      "learning_rate": 6.359620982848567e-06,
      "loss": 0.1824,
      "step": 21800
    },
    {
      "epoch": 2.55,
      "grad_norm": 2.460673581458399,
      "learning_rate": 6.341609891521153e-06,
      "loss": 0.1804,
      "step": 21820
    },
    {
      "epoch": 2.56,
      "grad_norm": 2.4334232766585817,
      "learning_rate": 6.3236124994420746e-06,
      "loss": 0.1856,
      "step": 21840
    },
    {
      "epoch": 2.56,
      "grad_norm": 12.823001484263724,
      "learning_rate": 6.305628873986145e-06,
      "loss": 0.1741,
      "step": 21860
    },
    {
      "epoch": 2.56,
      "grad_norm": 3.3642127400826882,
      "learning_rate": 6.2876590824766445e-06,
      "loss": 0.181,
      "step": 21880
    },
    {
      "epoch": 2.56,
      "grad_norm": 4.637360423164878,
      "learning_rate": 6.269703192185069e-06,
      "loss": 0.1872,
      "step": 21900
    },
    {
      "epoch": 2.56,
      "grad_norm": 3.0708541586457465,
      "learning_rate": 6.251761270330871e-06,
      "loss": 0.1802,
      "step": 21920
    },
    {
      "epoch": 2.57,
      "grad_norm": 3.0814400264184236,
      "learning_rate": 6.233833384081214e-06,
      "loss": 0.1793,
      "step": 21940
    },
    {
      "epoch": 2.57,
      "grad_norm": 2.43342072926522,
      "learning_rate": 6.215919600550707e-06,
      "loss": 0.1758,
      "step": 21960
    },
    {
      "epoch": 2.57,
      "grad_norm": 2.881630756102204,
      "learning_rate": 6.198019986801178e-06,
      "loss": 0.1757,
      "step": 21980
    },
    {
      "epoch": 2.57,
      "grad_norm": 2.211538603792218,
      "learning_rate": 6.1801346098414036e-06,
      "loss": 0.1783,
      "step": 22000
    },
    {
      "epoch": 2.58,
      "grad_norm": 8.512703770734728,
      "learning_rate": 6.162263536626864e-06,
      "loss": 0.1805,
      "step": 22020
    },
    {
      "epoch": 2.58,
      "grad_norm": 2.2659396249079604,
      "learning_rate": 6.144406834059488e-06,
      "loss": 0.1768,
      "step": 22040
    },
    {
      "epoch": 2.58,
      "grad_norm": 2.9857345599922196,
      "learning_rate": 6.126564568987412e-06,
      "loss": 0.1814,
      "step": 22060
    },
    {
      "epoch": 2.58,
      "grad_norm": 2.3473919864135215,
      "learning_rate": 6.108736808204723e-06,
      "loss": 0.1655,
      "step": 22080
    },
    {
      "epoch": 2.59,
      "grad_norm": 1.7312928039547535,
      "learning_rate": 6.090923618451211e-06,
      "loss": 0.1737,
      "step": 22100
    },
    {
      "epoch": 2.59,
      "grad_norm": 2.807589503939605,
      "learning_rate": 6.073125066412114e-06,
      "loss": 0.1844,
      "step": 22120
    },
    {
      "epoch": 2.59,
      "grad_norm": 3.4617900960703105,
      "learning_rate": 6.055341218717877e-06,
      "loss": 0.1907,
      "step": 22140
    },
    {
      "epoch": 2.59,
      "grad_norm": 3.4841808042594486,
      "learning_rate": 6.037572141943895e-06,
      "loss": 0.1787,
      "step": 22160
    },
    {
      "epoch": 2.6,
      "grad_norm": 2.6092051376516756,
      "learning_rate": 6.019817902610269e-06,
      "loss": 0.1656,
      "step": 22180
    },
    {
      "epoch": 2.6,
      "grad_norm": 3.136799318411884,
      "learning_rate": 6.002078567181554e-06,
      "loss": 0.1732,
      "step": 22200
    },
    {
      "epoch": 2.6,
      "grad_norm": 3.4455409936543546,
      "learning_rate": 5.984354202066508e-06,
      "loss": 0.1815,
      "step": 22220
    },
    {
      "epoch": 2.6,
      "grad_norm": 2.612732332110929,
      "learning_rate": 5.966644873617851e-06,
      "loss": 0.1779,
      "step": 22240
    },
    {
      "epoch": 2.6,
      "grad_norm": 2.3915440407218425,
      "learning_rate": 5.948950648132007e-06,
      "loss": 0.1795,
      "step": 22260
    },
    {
      "epoch": 2.61,
      "grad_norm": 4.31280781849793,
      "learning_rate": 5.931271591848869e-06,
      "loss": 0.1631,
      "step": 22280
    },
    {
      "epoch": 2.61,
      "grad_norm": 4.414478599718549,
      "learning_rate": 5.913607770951527e-06,
      "loss": 0.1811,
      "step": 22300
    },
    {
      "epoch": 2.61,
      "grad_norm": 8.180075970509124,
      "learning_rate": 5.895959251566058e-06,
      "loss": 0.1801,
      "step": 22320
    },
    {
      "epoch": 2.61,
      "grad_norm": 2.0782483286580717,
      "learning_rate": 5.878326099761238e-06,
      "loss": 0.1739,
      "step": 22340
    },
    {
      "epoch": 2.62,
      "grad_norm": 2.5269629826915856,
      "learning_rate": 5.860708381548322e-06,
      "loss": 0.181,
      "step": 22360
    },
    {
      "epoch": 2.62,
      "grad_norm": 2.5519186991832656,
      "learning_rate": 5.843106162880785e-06,
      "loss": 0.1834,
      "step": 22380
    },
    {
      "epoch": 2.62,
      "grad_norm": 3.435157671423404,
      "learning_rate": 5.825519509654079e-06,
      "loss": 0.1763,
      "step": 22400
    },
    {
      "epoch": 2.62,
      "grad_norm": 2.6878877404420196,
      "learning_rate": 5.807948487705384e-06,
      "loss": 0.1941,
      "step": 22420
    },
    {
      "epoch": 2.63,
      "grad_norm": 2.5448246291018366,
      "learning_rate": 5.790393162813365e-06,
      "loss": 0.1815,
      "step": 22440
    },
    {
      "epoch": 2.63,
      "grad_norm": 4.743917083813302,
      "learning_rate": 5.772853600697924e-06,
      "loss": 0.1831,
      "step": 22460
    },
    {
      "epoch": 2.63,
      "grad_norm": 2.379670720282324,
      "learning_rate": 5.755329867019952e-06,
      "loss": 0.1811,
      "step": 22480
    },
    {
      "epoch": 2.63,
      "grad_norm": 2.610666195409938,
      "learning_rate": 5.737822027381088e-06,
      "loss": 0.1743,
      "step": 22500
    },
    {
      "epoch": 2.64,
      "grad_norm": 3.0100376370126853,
      "learning_rate": 5.720330147323458e-06,
      "loss": 0.1902,
      "step": 22520
    },
    {
      "epoch": 2.64,
      "grad_norm": 2.99934268744031,
      "learning_rate": 5.702854292329461e-06,
      "loss": 0.1736,
      "step": 22540
    },
    {
      "epoch": 2.64,
      "grad_norm": 3.5108678209005477,
      "learning_rate": 5.685394527821494e-06,
      "loss": 0.1882,
      "step": 22560
    },
    {
      "epoch": 2.64,
      "grad_norm": 6.166234628475323,
      "learning_rate": 5.66795091916172e-06,
      "loss": 0.1776,
      "step": 22580
    },
    {
      "epoch": 2.64,
      "grad_norm": 2.265912267954542,
      "learning_rate": 5.650523531651817e-06,
      "loss": 0.1775,
      "step": 22600
    },
    {
      "epoch": 2.65,
      "grad_norm": 2.875442305003155,
      "learning_rate": 5.63311243053274e-06,
      "loss": 0.182,
      "step": 22620
    },
    {
      "epoch": 2.65,
      "grad_norm": 4.23319985387031,
      "learning_rate": 5.615717680984476e-06,
      "loss": 0.1862,
      "step": 22640
    },
    {
      "epoch": 2.65,
      "grad_norm": 2.813515458637544,
      "learning_rate": 5.598339348125805e-06,
      "loss": 0.1754,
      "step": 22660
    },
    {
      "epoch": 2.65,
      "grad_norm": 2.465967179258659,
      "learning_rate": 5.580977497014037e-06,
      "loss": 0.1752,
      "step": 22680
    },
    {
      "epoch": 2.66,
      "grad_norm": 3.0256927516690193,
      "learning_rate": 5.563632192644785e-06,
      "loss": 0.1836,
      "step": 22700
    },
    {
      "epoch": 2.66,
      "grad_norm": 7.287674455861031,
      "learning_rate": 5.546303499951722e-06,
      "loss": 0.1891,
      "step": 22720
    },
    {
      "epoch": 2.66,
      "grad_norm": 2.3533914240438594,
      "learning_rate": 5.528991483806333e-06,
      "loss": 0.1849,
      "step": 22740
    },
    {
      "epoch": 2.66,
      "grad_norm": 3.6142213990161087,
      "learning_rate": 5.511696209017668e-06,
      "loss": 0.1799,
      "step": 22760
    },
    {
      "epoch": 2.67,
      "grad_norm": 4.110370703915694,
      "learning_rate": 5.494417740332109e-06,
      "loss": 0.1884,
      "step": 22780
    },
    {
      "epoch": 2.67,
      "grad_norm": 3.28314953271809,
      "learning_rate": 5.477156142433121e-06,
      "loss": 0.1792,
      "step": 22800
    },
    {
      "epoch": 2.67,
      "grad_norm": 3.4918275515348984,
      "learning_rate": 5.459911479941013e-06,
      "loss": 0.1678,
      "step": 22820
    },
    {
      "epoch": 2.67,
      "grad_norm": 3.5246556918764216,
      "learning_rate": 5.442683817412696e-06,
      "loss": 0.1834,
      "step": 22840
    },
    {
      "epoch": 2.67,
      "grad_norm": 3.9386810469092604,
      "learning_rate": 5.4254732193414325e-06,
      "loss": 0.1855,
      "step": 22860
    },
    {
      "epoch": 2.68,
      "grad_norm": 2.7961797436144646,
      "learning_rate": 5.408279750156611e-06,
      "loss": 0.177,
      "step": 22880
    },
    {
      "epoch": 2.68,
      "grad_norm": 1.8885662755469237,
      "learning_rate": 5.3911034742234935e-06,
      "loss": 0.1877,
      "step": 22900
    },
    {
      "epoch": 2.68,
      "grad_norm": 3.5211002015468824,
      "learning_rate": 5.3739444558429785e-06,
      "loss": 0.173,
      "step": 22920
    },
    {
      "epoch": 2.68,
      "grad_norm": 2.889315251733768,
      "learning_rate": 5.356802759251359e-06,
      "loss": 0.1721,
      "step": 22940
    },
    {
      "epoch": 2.69,
      "grad_norm": 2.6526195174893497,
      "learning_rate": 5.33967844862008e-06,
      "loss": 0.1891,
      "step": 22960
    },
    {
      "epoch": 2.69,
      "grad_norm": 2.3210056695240207,
      "learning_rate": 5.322571588055504e-06,
      "loss": 0.1779,
      "step": 22980
    },
    {
      "epoch": 2.69,
      "grad_norm": 2.93332602832347,
      "learning_rate": 5.305482241598673e-06,
      "loss": 0.1816,
      "step": 23000
    },
    {
      "epoch": 2.69,
      "grad_norm": 2.3996137785091047,
      "learning_rate": 5.288410473225043e-06,
      "loss": 0.1767,
      "step": 23020
    },
    {
      "epoch": 2.7,
      "grad_norm": 3.156745343004959,
      "learning_rate": 5.271356346844285e-06,
      "loss": 0.1745,
      "step": 23040
    },
    {
      "epoch": 2.7,
      "grad_norm": 2.7810676600659145,
      "learning_rate": 5.254319926300022e-06,
      "loss": 0.207,
      "step": 23060
    },
    {
      "epoch": 2.7,
      "grad_norm": 3.7710824255400843,
      "learning_rate": 5.237301275369584e-06,
      "loss": 0.1686,
      "step": 23080
    },
    {
      "epoch": 2.7,
      "grad_norm": 2.4391670395260356,
      "learning_rate": 5.220300457763794e-06,
      "loss": 0.1783,
      "step": 23100
    },
    {
      "epoch": 2.71,
      "grad_norm": 2.4162439656268617,
      "learning_rate": 5.203317537126702e-06,
      "loss": 0.1804,
      "step": 23120
    },
    {
      "epoch": 2.71,
      "grad_norm": 2.584051729623389,
      "learning_rate": 5.186352577035364e-06,
      "loss": 0.1782,
      "step": 23140
    },
    {
      "epoch": 2.71,
      "grad_norm": 4.071893718029455,
      "learning_rate": 5.169405640999598e-06,
      "loss": 0.1774,
      "step": 23160
    },
    {
      "epoch": 2.71,
      "grad_norm": 3.8461652161356787,
      "learning_rate": 5.152476792461754e-06,
      "loss": 0.1788,
      "step": 23180
    },
    {
      "epoch": 2.71,
      "grad_norm": 2.2006858146864485,
      "learning_rate": 5.135566094796451e-06,
      "loss": 0.1821,
      "step": 23200
    },
    {
      "epoch": 2.72,
      "grad_norm": 3.4525315823446716,
      "learning_rate": 5.11867361131038e-06,
      "loss": 0.176,
      "step": 23220
    },
    {
      "epoch": 2.72,
      "grad_norm": 3.2240358670987,
      "learning_rate": 5.1017994052420315e-06,
      "loss": 0.1821,
      "step": 23240
    },
    {
      "epoch": 2.72,
      "grad_norm": 3.174379459324645,
      "learning_rate": 5.084943539761484e-06,
      "loss": 0.1742,
      "step": 23260
    },
    {
      "epoch": 2.72,
      "grad_norm": 2.002869335885304,
      "learning_rate": 5.068106077970147e-06,
      "loss": 0.1714,
      "step": 23280
    },
    {
      "epoch": 2.73,
      "grad_norm": 3.039475753654841,
      "learning_rate": 5.05128708290054e-06,
      "loss": 0.1862,
      "step": 23300
    },
    {
      "epoch": 2.73,
      "grad_norm": 11.415182049603324,
      "learning_rate": 5.034486617516047e-06,
      "loss": 0.1864,
      "step": 23320
    },
    {
      "epoch": 2.73,
      "grad_norm": 2.215650946245953,
      "learning_rate": 5.01770474471069e-06,
      "loss": 0.1707,
      "step": 23340
    },
    {
      "epoch": 2.73,
      "grad_norm": 1.5861810205971256,
      "learning_rate": 5.000941527308877e-06,
      "loss": 0.1723,
      "step": 23360
    },
    {
      "epoch": 2.74,
      "grad_norm": 2.736141048682862,
      "learning_rate": 4.984197028065187e-06,
      "loss": 0.1817,
      "step": 23380
    },
    {
      "epoch": 2.74,
      "grad_norm": 2.785186296597876,
      "learning_rate": 4.967471309664126e-06,
      "loss": 0.1764,
      "step": 23400
    },
    {
      "epoch": 2.74,
      "grad_norm": 2.885900151662176,
      "learning_rate": 4.950764434719889e-06,
      "loss": 0.1828,
      "step": 23420
    },
    {
      "epoch": 2.74,
      "grad_norm": 3.137029763048628,
      "learning_rate": 4.934076465776131e-06,
      "loss": 0.1809,
      "step": 23440
    },
    {
      "epoch": 2.75,
      "grad_norm": 2.923199358588338,
      "learning_rate": 4.917407465305728e-06,
      "loss": 0.1674,
      "step": 23460
    },
    {
      "epoch": 2.75,
      "grad_norm": 2.964303953866439,
      "learning_rate": 4.90075749571055e-06,
      "loss": 0.1773,
      "step": 23480
    },
    {
      "epoch": 2.75,
      "grad_norm": 4.009182404005084,
      "learning_rate": 4.884126619321215e-06,
      "loss": 0.1728,
      "step": 23500
    },
    {
      "epoch": 2.75,
      "grad_norm": 2.9239608744705476,
      "learning_rate": 4.867514898396881e-06,
      "loss": 0.1854,
      "step": 23520
    },
    {
      "epoch": 2.75,
      "grad_norm": 3.0183117218620064,
      "learning_rate": 4.8509223951249725e-06,
      "loss": 0.1688,
      "step": 23540
    },
    {
      "epoch": 2.76,
      "grad_norm": 2.9178365450200583,
      "learning_rate": 4.8343491716209906e-06,
      "loss": 0.1804,
      "step": 23560
    },
    {
      "epoch": 2.76,
      "grad_norm": 1.9094178814607006,
      "learning_rate": 4.81779528992825e-06,
      "loss": 0.1607,
      "step": 23580
    },
    {
      "epoch": 2.76,
      "grad_norm": 4.365735954125327,
      "learning_rate": 4.801260812017662e-06,
      "loss": 0.1806,
      "step": 23600
    },
    {
      "epoch": 2.76,
      "grad_norm": 2.428095255980094,
      "learning_rate": 4.7847457997874975e-06,
      "loss": 0.1755,
      "step": 23620
    },
    {
      "epoch": 2.77,
      "grad_norm": 1.5540702783116471,
      "learning_rate": 4.768250315063154e-06,
      "loss": 0.174,
      "step": 23640
    },
    {
      "epoch": 2.77,
      "grad_norm": 2.549882669180469,
      "learning_rate": 4.751774419596929e-06,
      "loss": 0.1722,
      "step": 23660
    },
    {
      "epoch": 2.77,
      "grad_norm": 3.5723809525533547,
      "learning_rate": 4.735318175067789e-06,
      "loss": 0.1929,
      "step": 23680
    },
    {
      "epoch": 2.77,
      "grad_norm": 3.2517252524328635,
      "learning_rate": 4.7188816430811215e-06,
      "loss": 0.1901,
      "step": 23700
    },
    {
      "epoch": 2.78,
      "grad_norm": 3.250214496283374,
      "learning_rate": 4.702464885168532e-06,
      "loss": 0.1726,
      "step": 23720
    },
    {
      "epoch": 2.78,
      "grad_norm": 3.0741278551325983,
      "learning_rate": 4.686067962787599e-06,
      "loss": 0.1789,
      "step": 23740
    },
    {
      "epoch": 2.78,
      "grad_norm": 3.4015668119894724,
      "learning_rate": 4.669690937321637e-06,
      "loss": 0.1731,
      "step": 23760
    },
    {
      "epoch": 2.78,
      "grad_norm": 2.162316454556547,
      "learning_rate": 4.653333870079482e-06,
      "loss": 0.1727,
      "step": 23780
    },
    {
      "epoch": 2.78,
      "grad_norm": 2.954847529781422,
      "learning_rate": 4.636996822295253e-06,
      "loss": 0.1926,
      "step": 23800
    },
    {
      "epoch": 2.79,
      "grad_norm": 3.218453402881352,
      "learning_rate": 4.620679855128122e-06,
      "loss": 0.1752,
      "step": 23820
    },
    {
      "epoch": 2.79,
      "grad_norm": 2.3305631837756633,
      "learning_rate": 4.604383029662088e-06,
      "loss": 0.1681,
      "step": 23840
    },
    {
      "epoch": 2.79,
      "grad_norm": 3.2650324087593714,
      "learning_rate": 4.588106406905756e-06,
      "loss": 0.167,
      "step": 23860
    },
    {
      "epoch": 2.79,
      "grad_norm": 2.7496586501069826,
      "learning_rate": 4.5718500477920825e-06,
      "loss": 0.1646,
      "step": 23880
    },
    {
      "epoch": 2.8,
      "grad_norm": 3.211021403680614,
      "learning_rate": 4.555614013178179e-06,
      "loss": 0.1704,
      "step": 23900
    },
    {
      "epoch": 2.8,
      "grad_norm": 2.902403059465085,
      "learning_rate": 4.5393983638450714e-06,
      "loss": 0.1814,
      "step": 23920
    },
    {
      "epoch": 2.8,
      "grad_norm": 2.460405952329076,
      "learning_rate": 4.5232031604974605e-06,
      "loss": 0.1753,
      "step": 23940
    },
    {
      "epoch": 2.8,
      "grad_norm": 3.1027053533889952,
      "learning_rate": 4.507028463763521e-06,
      "loss": 0.1821,
      "step": 23960
    },
    {
      "epoch": 2.81,
      "grad_norm": 2.4814554970152076,
      "learning_rate": 4.490874334194647e-06,
      "loss": 0.1754,
      "step": 23980
    },
    {
      "epoch": 2.81,
      "grad_norm": 3.0816354630415317,
      "learning_rate": 4.474740832265242e-06,
      "loss": 0.1647,
      "step": 24000
    },
    {
      "epoch": 2.81,
      "grad_norm": 4.258299409515962,
      "learning_rate": 4.45862801837249e-06,
      "loss": 0.1651,
      "step": 24020
    },
    {
      "epoch": 2.81,
      "grad_norm": 2.5109760616168644,
      "learning_rate": 4.442535952836121e-06,
      "loss": 0.1737,
      "step": 24040
    },
    {
      "epoch": 2.82,
      "grad_norm": 3.670190952540861,
      "learning_rate": 4.426464695898196e-06,
      "loss": 0.1815,
      "step": 24060
    },
    {
      "epoch": 2.82,
      "grad_norm": 2.246500684943945,
      "learning_rate": 4.41041430772288e-06,
      "loss": 0.1778,
      "step": 24080
    },
    {
      "epoch": 2.82,
      "grad_norm": 2.8719645522541355,
      "learning_rate": 4.394384848396209e-06,
      "loss": 0.1729,
      "step": 24100
    },
    {
      "epoch": 2.82,
      "grad_norm": 2.5285651956593784,
      "learning_rate": 4.378376377925873e-06,
      "loss": 0.1812,
      "step": 24120
    },
    {
      "epoch": 2.82,
      "grad_norm": 1.589113859923903,
      "learning_rate": 4.362388956240989e-06,
      "loss": 0.1765,
      "step": 24140
    },
    {
      "epoch": 2.83,
      "grad_norm": 3.212563652862593,
      "learning_rate": 4.346422643191872e-06,
      "loss": 0.1861,
      "step": 24160
    },
    {
      "epoch": 2.83,
      "grad_norm": 1.9036122817924004,
      "learning_rate": 4.330477498549818e-06,
      "loss": 0.1757,
      "step": 24180
    },
    {
      "epoch": 2.83,
      "grad_norm": 2.773805921557294,
      "learning_rate": 4.314553582006882e-06,
      "loss": 0.1795,
      "step": 24200
    },
    {
      "epoch": 2.83,
      "grad_norm": 4.356117261837307,
      "learning_rate": 4.298650953175635e-06,
      "loss": 0.1817,
      "step": 24220
    },
    {
      "epoch": 2.84,
      "grad_norm": 2.629481169141941,
      "learning_rate": 4.282769671588969e-06,
      "loss": 0.1777,
      "step": 24240
    },
    {
      "epoch": 2.84,
      "grad_norm": 3.3316093437094025,
      "learning_rate": 4.2669097966998586e-06,
      "loss": 0.1684,
      "step": 24260
    },
    {
      "epoch": 2.84,
      "grad_norm": 3.671906945921538,
      "learning_rate": 4.251071387881136e-06,
      "loss": 0.1828,
      "step": 24280
    },
    {
      "epoch": 2.84,
      "grad_norm": 2.67347838717775,
      "learning_rate": 4.235254504425278e-06,
      "loss": 0.1708,
      "step": 24300
    },
    {
      "epoch": 2.85,
      "grad_norm": 2.862267370641254,
      "learning_rate": 4.2194592055441765e-06,
      "loss": 0.1706,
      "step": 24320
    },
    {
      "epoch": 2.85,
      "grad_norm": 2.953771449788878,
      "learning_rate": 4.203685550368921e-06,
      "loss": 0.1683,
      "step": 24340
    },
    {
      "epoch": 2.85,
      "grad_norm": 2.5105236764052354,
      "learning_rate": 4.187933597949578e-06,
      "loss": 0.186,
      "step": 24360
    },
    {
      "epoch": 2.85,
      "grad_norm": 2.218245972886692,
      "learning_rate": 4.172203407254954e-06,
      "loss": 0.1627,
      "step": 24380
    },
    {
      "epoch": 2.86,
      "grad_norm": 3.517470625569687,
      "learning_rate": 4.1564950371724115e-06,
      "loss": 0.1745,
      "step": 24400
    },
    {
      "epoch": 2.86,
      "grad_norm": 2.3898984515591137,
      "learning_rate": 4.140808546507611e-06,
      "loss": 0.1757,
      "step": 24420
    },
    {
      "epoch": 2.86,
      "grad_norm": 3.7119551698756417,
      "learning_rate": 4.125143993984306e-06,
      "loss": 0.171,
      "step": 24440
    },
    {
      "epoch": 2.86,
      "grad_norm": 1.8304933384323714,
      "learning_rate": 4.109501438244126e-06,
      "loss": 0.1804,
      "step": 24460
    },
    {
      "epoch": 2.86,
      "grad_norm": 2.967840597504324,
      "learning_rate": 4.093880937846353e-06,
      "loss": 0.182,
      "step": 24480
    },
    {
      "epoch": 2.87,
      "grad_norm": 3.1239513926711426,
      "learning_rate": 4.078282551267702e-06,
      "loss": 0.1856,
      "step": 24500
    },
    {
      "epoch": 2.87,
      "grad_norm": 3.283381033484747,
      "learning_rate": 4.062706336902103e-06,
      "loss": 0.1816,
      "step": 24520
    },
    {
      "epoch": 2.87,
      "grad_norm": 2.6231464926634684,
      "learning_rate": 4.047152353060487e-06,
      "loss": 0.1738,
      "step": 24540
    },
    {
      "epoch": 2.87,
      "grad_norm": 3.169329895288349,
      "learning_rate": 4.031620657970553e-06,
      "loss": 0.1751,
      "step": 24560
    },
    {
      "epoch": 2.88,
      "grad_norm": 3.054745100410017,
      "learning_rate": 4.0161113097765655e-06,
      "loss": 0.1731,
      "step": 24580
    },
    {
      "epoch": 2.88,
      "grad_norm": 2.1135692576460454,
      "learning_rate": 4.000624366539136e-06,
      "loss": 0.1724,
      "step": 24600
    },
    {
      "epoch": 2.88,
      "grad_norm": 1.9962103225500736,
      "learning_rate": 3.985159886234995e-06,
      "loss": 0.1721,
      "step": 24620
    },
    {
      "epoch": 2.88,
      "grad_norm": 3.3958365473507763,
      "learning_rate": 3.969717926756783e-06,
      "loss": 0.1764,
      "step": 24640
    },
    {
      "epoch": 2.89,
      "grad_norm": 2.0383790723400077,
      "learning_rate": 3.954298545912833e-06,
      "loss": 0.1777,
      "step": 24660
    },
    {
      "epoch": 2.89,
      "grad_norm": 2.7379240709118675,
      "learning_rate": 3.93890180142695e-06,
      "loss": 0.1809,
      "step": 24680
    },
    {
      "epoch": 2.89,
      "grad_norm": 2.0472564341865227,
      "learning_rate": 3.923527750938205e-06,
      "loss": 0.1754,
      "step": 24700
    },
    {
      "epoch": 2.89,
      "grad_norm": 2.8384107833691874,
      "learning_rate": 3.908176452000697e-06,
      "loss": 0.1731,
      "step": 24720
    },
    {
      "epoch": 2.89,
      "grad_norm": 2.6243764954263122,
      "learning_rate": 3.892847962083369e-06,
      "loss": 0.1791,
      "step": 24740
    },
    {
      "epoch": 2.9,
      "grad_norm": 2.7421433317534447,
      "learning_rate": 3.877542338569767e-06,
      "loss": 0.1749,
      "step": 24760
    },
    {
      "epoch": 2.9,
      "grad_norm": 3.215031225548206,
      "learning_rate": 3.862259638757837e-06,
      "loss": 0.1749,
      "step": 24780
    },
    {
      "epoch": 2.9,
      "grad_norm": 2.5967047533401417,
      "learning_rate": 3.846999919859708e-06,
      "loss": 0.1749,
      "step": 24800
    },
    {
      "epoch": 2.9,
      "grad_norm": 3.203524383209205,
      "learning_rate": 3.831763239001478e-06,
      "loss": 0.1704,
      "step": 24820
    },
    {
      "epoch": 2.91,
      "grad_norm": 4.166137407385502,
      "learning_rate": 3.816549653223004e-06,
      "loss": 0.1847,
      "step": 24840
    },
    {
      "epoch": 2.91,
      "grad_norm": 2.4957440389496837,
      "learning_rate": 3.801359219477679e-06,
      "loss": 0.174,
      "step": 24860
    },
    {
      "epoch": 2.91,
      "grad_norm": 2.552841966458733,
      "learning_rate": 3.7861919946322273e-06,
      "loss": 0.1823,
      "step": 24880
    },
    {
      "epoch": 2.91,
      "grad_norm": 3.4426880959489736,
      "learning_rate": 3.7710480354664838e-06,
      "loss": 0.1678,
      "step": 24900
    },
    {
      "epoch": 2.92,
      "grad_norm": 2.4191332648960073,
      "learning_rate": 3.7559273986731896e-06,
      "loss": 0.1774,
      "step": 24920
    },
    {
      "epoch": 2.92,
      "grad_norm": 3.6865799612802497,
      "learning_rate": 3.7408301408577813e-06,
      "loss": 0.1681,
      "step": 24940
    },
    {
      "epoch": 2.92,
      "grad_norm": 2.135317941259829,
      "learning_rate": 3.7257563185381663e-06,
      "loss": 0.1681,
      "step": 24960
    },
    {
      "epoch": 2.92,
      "grad_norm": 3.460522766752715,
      "learning_rate": 3.7107059881445233e-06,
      "loss": 0.1847,
      "step": 24980
    },
    {
      "epoch": 2.93,
      "grad_norm": 2.320078520103796,
      "learning_rate": 3.695679206019088e-06,
      "loss": 0.1775,
      "step": 25000
    },
    {
      "epoch": 2.93,
      "grad_norm": 4.046108335242948,
      "learning_rate": 3.6806760284159377e-06,
      "loss": 0.1775,
      "step": 25020
    },
    {
      "epoch": 2.93,
      "grad_norm": 2.7317959202771216,
      "learning_rate": 3.665696511500789e-06,
      "loss": 0.1801,
      "step": 25040
    },
    {
      "epoch": 2.93,
      "grad_norm": 3.2533030231293756,
      "learning_rate": 3.6507407113507725e-06,
      "loss": 0.169,
      "step": 25060
    },
    {
      "epoch": 2.93,
      "grad_norm": 3.2224973327294224,
      "learning_rate": 3.635808683954245e-06,
      "loss": 0.18,
      "step": 25080
    },
    {
      "epoch": 2.94,
      "grad_norm": 1.6534753986947448,
      "learning_rate": 3.6209004852105604e-06,
      "loss": 0.1709,
      "step": 25100
    },
    {
      "epoch": 2.94,
      "grad_norm": 2.6494173021082297,
      "learning_rate": 3.6060161709298714e-06,
      "loss": 0.1736,
      "step": 25120
    },
    {
      "epoch": 2.94,
      "grad_norm": 3.1748671676603415,
      "learning_rate": 3.5911557968329165e-06,
      "loss": 0.1733,
      "step": 25140
    },
    {
      "epoch": 2.94,
      "grad_norm": 3.125506855868197,
      "learning_rate": 3.5763194185508114e-06,
      "loss": 0.1747,
      "step": 25160
    },
    {
      "epoch": 2.95,
      "grad_norm": 3.1171335010105934,
      "learning_rate": 3.5615070916248402e-06,
      "loss": 0.1897,
      "step": 25180
    },
    {
      "epoch": 2.95,
      "grad_norm": 2.136778222185654,
      "learning_rate": 3.5467188715062528e-06,
      "loss": 0.1711,
      "step": 25200
    },
    {
      "epoch": 2.95,
      "grad_norm": 3.0959637360728443,
      "learning_rate": 3.5319548135560493e-06,
      "loss": 0.1781,
      "step": 25220
    },
    {
      "epoch": 2.95,
      "grad_norm": 2.5024311642320893,
      "learning_rate": 3.5172149730447746e-06,
      "loss": 0.1812,
      "step": 25240
    },
    {
      "epoch": 2.96,
      "grad_norm": 4.167284093251525,
      "learning_rate": 3.5024994051523154e-06,
      "loss": 0.1808,
      "step": 25260
    },
    {
      "epoch": 2.96,
      "grad_norm": 2.348151384844568,
      "learning_rate": 3.4878081649676994e-06,
      "loss": 0.1718,
      "step": 25280
    },
    {
      "epoch": 2.96,
      "grad_norm": 3.29314656529847,
      "learning_rate": 3.4731413074888713e-06,
      "loss": 0.1703,
      "step": 25300
    },
    {
      "epoch": 2.96,
      "grad_norm": 2.343550304806459,
      "learning_rate": 3.4584988876225e-06,
      "loss": 0.1674,
      "step": 25320
    },
    {
      "epoch": 2.97,
      "grad_norm": 6.679659231025708,
      "learning_rate": 3.443880960183772e-06,
      "loss": 0.1747,
      "step": 25340
    },
    {
      "epoch": 2.97,
      "grad_norm": 2.41908162132978,
      "learning_rate": 3.4292875798961815e-06,
      "loss": 0.1787,
      "step": 25360
    },
    {
      "epoch": 2.97,
      "grad_norm": 2.3049198356805327,
      "learning_rate": 3.414718801391333e-06,
      "loss": 0.1854,
      "step": 25380
    },
    {
      "epoch": 2.97,
      "grad_norm": 2.3512446293219433,
      "learning_rate": 3.400174679208722e-06,
      "loss": 0.1708,
      "step": 25400
    },
    {
      "epoch": 2.97,
      "grad_norm": 3.478845425392513,
      "learning_rate": 3.3856552677955525e-06,
      "loss": 0.17,
      "step": 25420
    },
    {
      "epoch": 2.98,
      "grad_norm": 2.268869221394817,
      "learning_rate": 3.3711606215065136e-06,
      "loss": 0.1798,
      "step": 25440
    },
    {
      "epoch": 2.98,
      "grad_norm": 4.871263294588487,
      "learning_rate": 3.356690794603589e-06,
      "loss": 0.1749,
      "step": 25460
    },
    {
      "epoch": 2.98,
      "grad_norm": 2.3442633511697126,
      "learning_rate": 3.342245841255847e-06,
      "loss": 0.1828,
      "step": 25480
    },
    {
      "epoch": 2.98,
      "grad_norm": 3.5581851728863607,
      "learning_rate": 3.3278258155392384e-06,
      "loss": 0.1833,
      "step": 25500
    },
    {
      "epoch": 2.99,
      "grad_norm": 2.573504752674091,
      "learning_rate": 3.3134307714363954e-06,
      "loss": 0.1587,
      "step": 25520
    },
    {
      "epoch": 2.99,
      "grad_norm": 2.330468758184399,
      "learning_rate": 3.2990607628364324e-06,
      "loss": 0.1722,
      "step": 25540
    },
    {
      "epoch": 2.99,
      "grad_norm": 3.2204463895739854,
      "learning_rate": 3.2847158435347417e-06,
      "loss": 0.1916,
      "step": 25560
    },
    {
      "epoch": 2.99,
      "grad_norm": 2.557634339742982,
      "learning_rate": 3.270396067232781e-06,
      "loss": 0.1808,
      "step": 25580
    },
    {
      "epoch": 3.0,
      "grad_norm": 3.6599295595821943,
      "learning_rate": 3.2561014875378948e-06,
      "loss": 0.1777,
      "step": 25600
    },
    {
      "epoch": 3.0,
      "grad_norm": 2.50785271443005,
      "learning_rate": 3.2418321579630986e-06,
      "loss": 0.1799,
      "step": 25620
    }
  ],
  "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
}