{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.4690157958687728,
  "eval_steps": 103,
  "global_step": 309,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "grad_norm": 0.3593529462814331,
      "learning_rate": 2e-05,
      "loss": 0.4337,
      "step": 1
    },
    {
      "epoch": 0.0,
      "eval_loss": 0.3783022463321686,
      "eval_runtime": 12.9552,
      "eval_samples_per_second": 1.93,
      "eval_steps_per_second": 1.93,
      "step": 1
    },
    {
      "epoch": 0.01,
      "grad_norm": 0.3430859446525574,
      "learning_rate": 4e-05,
      "loss": 0.4605,
      "step": 2
    },
    {
      "epoch": 0.01,
      "grad_norm": 0.3275960683822632,
      "learning_rate": 6e-05,
      "loss": 0.39,
      "step": 3
    },
    {
      "epoch": 0.02,
      "grad_norm": 0.3221317529678345,
      "learning_rate": 8e-05,
      "loss": 0.3372,
      "step": 4
    },
    {
      "epoch": 0.02,
      "grad_norm": 0.3926704227924347,
      "learning_rate": 0.0001,
      "loss": 0.3333,
      "step": 5
    },
    {
      "epoch": 0.03,
      "grad_norm": 0.2960835099220276,
      "learning_rate": 0.00012,
      "loss": 0.3671,
      "step": 6
    },
    {
      "epoch": 0.03,
      "grad_norm": 0.3393571078777313,
      "learning_rate": 0.00014,
      "loss": 0.327,
      "step": 7
    },
    {
      "epoch": 0.04,
      "grad_norm": 0.2799758017063141,
      "learning_rate": 0.00016,
      "loss": 0.2933,
      "step": 8
    },
    {
      "epoch": 0.04,
      "grad_norm": 0.3084808886051178,
      "learning_rate": 0.00018,
      "loss": 0.3505,
      "step": 9
    },
    {
      "epoch": 0.05,
      "grad_norm": 0.23642300069332123,
      "learning_rate": 0.0002,
      "loss": 0.3289,
      "step": 10
    },
    {
      "epoch": 0.05,
      "grad_norm": 0.369229793548584,
      "learning_rate": 0.00019999691576447898,
      "loss": 0.3049,
      "step": 11
    },
    {
      "epoch": 0.06,
      "grad_norm": 0.2706857919692993,
      "learning_rate": 0.00019998766324816607,
      "loss": 0.3425,
      "step": 12
    },
    {
      "epoch": 0.06,
      "grad_norm": 0.21327799558639526,
      "learning_rate": 0.00019997224302180006,
      "loss": 0.2686,
      "step": 13
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.26732948422431946,
      "learning_rate": 0.00019995065603657316,
      "loss": 0.2987,
      "step": 14
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.2009548544883728,
      "learning_rate": 0.0001999229036240723,
      "loss": 0.2668,
      "step": 15
    },
    {
      "epoch": 0.08,
      "grad_norm": 0.23616977035999298,
      "learning_rate": 0.00019988898749619702,
      "loss": 0.2962,
      "step": 16
    },
    {
      "epoch": 0.08,
      "grad_norm": 0.18399174511432648,
      "learning_rate": 0.00019984890974505381,
      "loss": 0.2238,
      "step": 17
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.24744661152362823,
      "learning_rate": 0.00019980267284282717,
      "loss": 0.2628,
      "step": 18
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.22109034657478333,
      "learning_rate": 0.00019975027964162702,
      "loss": 0.3143,
      "step": 19
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.21471348404884338,
      "learning_rate": 0.0001996917333733128,
      "loss": 0.3436,
      "step": 20
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.2112409919500351,
      "learning_rate": 0.00019962703764929413,
      "loss": 0.2714,
      "step": 21
    },
    {
      "epoch": 0.11,
      "grad_norm": 0.19044426083564758,
      "learning_rate": 0.00019955619646030802,
      "loss": 0.2335,
      "step": 22
    },
    {
      "epoch": 0.11,
      "grad_norm": 0.21945005655288696,
      "learning_rate": 0.00019947921417617267,
      "loss": 0.2381,
      "step": 23
    },
    {
      "epoch": 0.12,
      "grad_norm": 0.1864914447069168,
      "learning_rate": 0.000199396095545518,
      "loss": 0.287,
      "step": 24
    },
    {
      "epoch": 0.12,
      "grad_norm": 0.2413000762462616,
      "learning_rate": 0.00019930684569549264,
      "loss": 0.1813,
      "step": 25
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.20038250088691711,
      "learning_rate": 0.0001992114701314478,
      "loss": 0.2722,
      "step": 26
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.16239097714424133,
      "learning_rate": 0.0001991099747365975,
      "loss": 0.1917,
      "step": 27
    },
    {
      "epoch": 0.14,
      "grad_norm": 0.19039247930049896,
      "learning_rate": 0.00019900236577165576,
      "loss": 0.2665,
      "step": 28
    },
    {
      "epoch": 0.14,
      "grad_norm": 0.17717348039150238,
      "learning_rate": 0.0001988886498744505,
      "loss": 0.2667,
      "step": 29
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.18755286931991577,
      "learning_rate": 0.00019876883405951377,
      "loss": 0.1775,
      "step": 30
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.1621539294719696,
      "learning_rate": 0.00019864292571764955,
      "loss": 0.2292,
      "step": 31
    },
    {
      "epoch": 0.16,
      "grad_norm": 0.1834522783756256,
      "learning_rate": 0.0001985109326154774,
      "loss": 0.2327,
      "step": 32
    },
    {
      "epoch": 0.16,
      "grad_norm": 0.18009088933467865,
      "learning_rate": 0.00019837286289495361,
      "loss": 0.2325,
      "step": 33
    },
    {
      "epoch": 0.17,
      "grad_norm": 0.16372188925743103,
      "learning_rate": 0.0001982287250728689,
      "loss": 0.2748,
      "step": 34
    },
    {
      "epoch": 0.17,
      "grad_norm": 0.16966991126537323,
      "learning_rate": 0.00019807852804032305,
      "loss": 0.2353,
      "step": 35
    },
    {
      "epoch": 0.17,
      "grad_norm": 0.18791744112968445,
      "learning_rate": 0.00019792228106217658,
      "loss": 0.2693,
      "step": 36
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.16828736662864685,
      "learning_rate": 0.0001977599937764791,
      "loss": 0.1752,
      "step": 37
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.1636415272951126,
      "learning_rate": 0.00019759167619387476,
      "loss": 0.2961,
      "step": 38
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.152165487408638,
      "learning_rate": 0.00019741733869698495,
      "loss": 0.1954,
      "step": 39
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.1838696449995041,
      "learning_rate": 0.00019723699203976766,
      "loss": 0.2039,
      "step": 40
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.170082688331604,
      "learning_rate": 0.00019705064734685425,
      "loss": 0.2228,
      "step": 41
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.1489935666322708,
      "learning_rate": 0.0001968583161128631,
      "loss": 0.185,
      "step": 42
    },
    {
      "epoch": 0.21,
      "grad_norm": 0.1501648873090744,
      "learning_rate": 0.00019666001020169073,
      "loss": 0.2198,
      "step": 43
    },
    {
      "epoch": 0.21,
      "grad_norm": 0.24484610557556152,
      "learning_rate": 0.00019645574184577982,
      "loss": 0.2514,
      "step": 44
    },
    {
      "epoch": 0.22,
      "grad_norm": 0.16400669515132904,
      "learning_rate": 0.00019624552364536473,
      "loss": 0.283,
      "step": 45
    },
    {
      "epoch": 0.22,
      "grad_norm": 0.15034696459770203,
      "learning_rate": 0.0001960293685676943,
      "loss": 0.1637,
      "step": 46
    },
    {
      "epoch": 0.23,
      "grad_norm": 0.15385685861110687,
      "learning_rate": 0.00019580728994623195,
      "loss": 0.2905,
      "step": 47
    },
    {
      "epoch": 0.23,
      "grad_norm": 0.16004744172096252,
      "learning_rate": 0.00019557930147983302,
      "loss": 0.2173,
      "step": 48
    },
    {
      "epoch": 0.24,
      "grad_norm": 0.17182469367980957,
      "learning_rate": 0.0001953454172319001,
      "loss": 0.2147,
      "step": 49
    },
    {
      "epoch": 0.24,
      "grad_norm": 0.1576426774263382,
      "learning_rate": 0.00019510565162951537,
      "loss": 0.284,
      "step": 50
    },
    {
      "epoch": 0.25,
      "grad_norm": 0.14034013450145721,
      "learning_rate": 0.00019486001946255046,
      "loss": 0.1671,
      "step": 51
    },
    {
      "epoch": 0.25,
      "grad_norm": 0.1770170032978058,
      "learning_rate": 0.00019460853588275454,
      "loss": 0.205,
      "step": 52
    },
    {
      "epoch": 0.26,
      "grad_norm": 0.14077183604240417,
      "learning_rate": 0.00019435121640281938,
      "loss": 0.209,
      "step": 53
    },
    {
      "epoch": 0.26,
      "grad_norm": 0.2193373441696167,
      "learning_rate": 0.00019408807689542257,
      "loss": 0.2125,
      "step": 54
    },
    {
      "epoch": 0.27,
      "grad_norm": 0.1439114212989807,
      "learning_rate": 0.00019381913359224842,
      "loss": 0.1611,
      "step": 55
    },
    {
      "epoch": 0.27,
      "grad_norm": 0.16939564049243927,
      "learning_rate": 0.00019354440308298675,
      "loss": 0.2189,
      "step": 56
    },
    {
      "epoch": 0.28,
      "grad_norm": 0.17016972601413727,
      "learning_rate": 0.00019326390231430942,
      "loss": 0.2129,
      "step": 57
    },
    {
      "epoch": 0.28,
      "grad_norm": 0.16602838039398193,
      "learning_rate": 0.00019297764858882514,
      "loss": 0.23,
      "step": 58
    },
    {
      "epoch": 0.29,
      "grad_norm": 0.21853233873844147,
      "learning_rate": 0.00019268565956401208,
      "loss": 0.1879,
      "step": 59
    },
    {
      "epoch": 0.29,
      "grad_norm": 0.18649564683437347,
      "learning_rate": 0.0001923879532511287,
      "loss": 0.2063,
      "step": 60
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.16304822266101837,
      "learning_rate": 0.00019208454801410266,
      "loss": 0.2182,
      "step": 61
    },
    {
      "epoch": 0.3,
      "grad_norm": 0.1357814073562622,
      "learning_rate": 0.00019177546256839812,
      "loss": 0.1912,
      "step": 62
    },
    {
      "epoch": 0.31,
      "grad_norm": 0.1645856499671936,
      "learning_rate": 0.00019146071597986138,
      "loss": 0.2498,
      "step": 63
    },
    {
      "epoch": 0.31,
      "grad_norm": 0.20048978924751282,
      "learning_rate": 0.00019114032766354453,
      "loss": 0.3207,
      "step": 64
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.15142077207565308,
      "learning_rate": 0.00019081431738250814,
      "loss": 0.2102,
      "step": 65
    },
    {
      "epoch": 0.32,
      "grad_norm": 0.15482479333877563,
      "learning_rate": 0.00019048270524660196,
      "loss": 0.2706,
      "step": 66
    },
    {
      "epoch": 0.33,
      "grad_norm": 0.15685780346393585,
      "learning_rate": 0.00019014551171122457,
      "loss": 0.1898,
      "step": 67
    },
    {
      "epoch": 0.33,
      "grad_norm": 0.18191099166870117,
      "learning_rate": 0.00018980275757606157,
      "loss": 0.2838,
      "step": 68
    },
    {
      "epoch": 0.34,
      "grad_norm": 0.1533990502357483,
      "learning_rate": 0.0001894544639838025,
      "loss": 0.1954,
      "step": 69
    },
    {
      "epoch": 0.34,
      "grad_norm": 0.14591443538665771,
      "learning_rate": 0.0001891006524188368,
      "loss": 0.248,
      "step": 70
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.19209636747837067,
      "learning_rate": 0.00018874134470592835,
      "loss": 0.2677,
      "step": 71
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.14589589834213257,
      "learning_rate": 0.00018837656300886937,
      "loss": 0.2031,
      "step": 72
    },
    {
      "epoch": 0.35,
      "grad_norm": 0.17039361596107483,
      "learning_rate": 0.00018800632982911322,
      "loss": 0.2679,
      "step": 73
    },
    {
      "epoch": 0.36,
      "grad_norm": 0.1550627052783966,
      "learning_rate": 0.00018763066800438636,
      "loss": 0.2082,
      "step": 74
    },
    {
      "epoch": 0.36,
      "grad_norm": 0.15761853754520416,
      "learning_rate": 0.00018724960070727972,
      "loss": 0.2542,
      "step": 75
    },
    {
      "epoch": 0.37,
      "grad_norm": 0.1586439311504364,
      "learning_rate": 0.00018686315144381913,
      "loss": 0.1673,
      "step": 76
    },
    {
      "epoch": 0.37,
      "grad_norm": 0.16859450936317444,
      "learning_rate": 0.0001864713440520155,
      "loss": 0.2679,
      "step": 77
    },
    {
      "epoch": 0.38,
      "grad_norm": 0.15938322246074677,
      "learning_rate": 0.0001860742027003944,
      "loss": 0.2812,
      "step": 78
    },
    {
      "epoch": 0.38,
      "grad_norm": 0.15844044089317322,
      "learning_rate": 0.00018567175188650498,
      "loss": 0.2443,
      "step": 79
    },
    {
      "epoch": 0.39,
      "grad_norm": 0.1673026829957962,
      "learning_rate": 0.00018526401643540922,
      "loss": 0.1635,
      "step": 80
    },
    {
      "epoch": 0.39,
      "grad_norm": 0.18364693224430084,
      "learning_rate": 0.00018485102149815038,
      "loss": 0.2023,
      "step": 81
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.17101280391216278,
      "learning_rate": 0.00018443279255020152,
      "loss": 0.1827,
      "step": 82
    },
    {
      "epoch": 0.4,
      "grad_norm": 0.15244990587234497,
      "learning_rate": 0.0001840093553898942,
      "loss": 0.2362,
      "step": 83
    },
    {
      "epoch": 0.41,
      "grad_norm": 0.16682085394859314,
      "learning_rate": 0.00018358073613682706,
      "loss": 0.2006,
      "step": 84
    },
    {
      "epoch": 0.41,
      "grad_norm": 0.18180780112743378,
      "learning_rate": 0.00018314696123025454,
      "loss": 0.2275,
      "step": 85
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.1552531123161316,
      "learning_rate": 0.00018270805742745617,
      "loss": 0.2165,
      "step": 86
    },
    {
      "epoch": 0.42,
      "grad_norm": 0.1649930328130722,
      "learning_rate": 0.000182264051802086,
      "loss": 0.2714,
      "step": 87
    },
    {
      "epoch": 0.43,
      "grad_norm": 0.16403907537460327,
      "learning_rate": 0.00018181497174250236,
      "loss": 0.1963,
      "step": 88
    },
    {
      "epoch": 0.43,
      "grad_norm": 0.1535252332687378,
      "learning_rate": 0.00018136084495007872,
      "loss": 0.1944,
      "step": 89
    },
    {
      "epoch": 0.44,
      "grad_norm": 0.17279598116874695,
      "learning_rate": 0.00018090169943749476,
      "loss": 0.2072,
      "step": 90
    },
    {
      "epoch": 0.44,
      "grad_norm": 0.16793234646320343,
      "learning_rate": 0.00018043756352700846,
      "loss": 0.1753,
      "step": 91
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.2446856051683426,
      "learning_rate": 0.00017996846584870908,
      "loss": 0.1976,
      "step": 92
    },
    {
      "epoch": 0.45,
      "grad_norm": 0.20265690982341766,
      "learning_rate": 0.000179494435338751,
      "loss": 0.2656,
      "step": 93
    },
    {
      "epoch": 0.46,
      "grad_norm": 0.1686105579137802,
      "learning_rate": 0.00017901550123756906,
      "loss": 0.2032,
      "step": 94
    },
    {
      "epoch": 0.46,
      "grad_norm": 0.16633199155330658,
      "learning_rate": 0.00017853169308807448,
      "loss": 0.1939,
      "step": 95
    },
    {
      "epoch": 0.47,
      "grad_norm": 0.15133249759674072,
      "learning_rate": 0.000178043040733833,
      "loss": 0.214,
      "step": 96
    },
    {
      "epoch": 0.47,
      "grad_norm": 0.1566823273897171,
      "learning_rate": 0.00017754957431722346,
      "loss": 0.1764,
      "step": 97
    },
    {
      "epoch": 0.48,
      "grad_norm": 0.15705688297748566,
      "learning_rate": 0.00017705132427757895,
      "loss": 0.1941,
      "step": 98
    },
    {
      "epoch": 0.48,
      "grad_norm": 0.1403038203716278,
      "learning_rate": 0.00017654832134930882,
      "loss": 0.171,
      "step": 99
    },
    {
      "epoch": 0.49,
      "grad_norm": 0.14009466767311096,
      "learning_rate": 0.0001760405965600031,
      "loss": 0.2162,
      "step": 100
    },
    {
      "epoch": 0.49,
      "grad_norm": 0.14049287140369415,
      "learning_rate": 0.00017552818122851838,
      "loss": 0.1668,
      "step": 101
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.1339244395494461,
      "learning_rate": 0.00017501110696304596,
      "loss": 0.1511,
      "step": 102
    },
    {
      "epoch": 0.5,
      "grad_norm": 0.13987047970294952,
      "learning_rate": 0.00017448940565916222,
      "loss": 0.2537,
      "step": 103
    },
    {
      "epoch": 0.5,
      "eval_loss": 0.2344847470521927,
      "eval_runtime": 13.3513,
      "eval_samples_per_second": 1.872,
      "eval_steps_per_second": 1.872,
      "step": 103
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.1495981365442276,
      "learning_rate": 0.000173963109497861,
      "loss": 0.2417,
      "step": 104
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.129630908370018,
      "learning_rate": 0.00017343225094356855,
      "loss": 0.1913,
      "step": 105
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.17110762000083923,
      "learning_rate": 0.00017289686274214118,
      "loss": 0.3276,
      "step": 106
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.1344563066959381,
      "learning_rate": 0.00017235697791884494,
      "loss": 0.1805,
      "step": 107
    },
    {
      "epoch": 0.52,
      "grad_norm": 0.14707708358764648,
      "learning_rate": 0.00017181262977631888,
      "loss": 0.198,
      "step": 108
    },
    {
      "epoch": 0.53,
      "grad_norm": 0.14540620148181915,
      "learning_rate": 0.00017126385189252053,
      "loss": 0.1923,
      "step": 109
    },
    {
      "epoch": 0.53,
      "grad_norm": 0.14539222419261932,
      "learning_rate": 0.00017071067811865476,
      "loss": 0.1822,
      "step": 110
    },
    {
      "epoch": 0.54,
      "grad_norm": 0.16808092594146729,
      "learning_rate": 0.0001701531425770856,
      "loss": 0.2384,
      "step": 111
    },
    {
      "epoch": 0.54,
      "grad_norm": 0.1316782385110855,
      "learning_rate": 0.00016959127965923142,
      "loss": 0.1967,
      "step": 112
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.15255320072174072,
      "learning_rate": 0.00016902512402344373,
      "loss": 0.245,
      "step": 113
    },
    {
      "epoch": 0.55,
      "grad_norm": 0.152465358376503,
      "learning_rate": 0.00016845471059286887,
      "loss": 0.1942,
      "step": 114
    },
    {
      "epoch": 0.56,
      "grad_norm": 0.15641257166862488,
      "learning_rate": 0.0001678800745532942,
      "loss": 0.2091,
      "step": 115
    },
    {
      "epoch": 0.56,
      "grad_norm": 0.16389591991901398,
      "learning_rate": 0.00016730125135097735,
      "loss": 0.2006,
      "step": 116
    },
    {
      "epoch": 0.57,
      "grad_norm": 0.1631624549627304,
      "learning_rate": 0.00016671827669045998,
      "loss": 0.2716,
      "step": 117
    },
    {
      "epoch": 0.57,
      "grad_norm": 0.16853250563144684,
      "learning_rate": 0.00016613118653236518,
      "loss": 0.2376,
      "step": 118
    },
    {
      "epoch": 0.58,
      "grad_norm": 0.13790781795978546,
      "learning_rate": 0.0001655400170911794,
      "loss": 0.1943,
      "step": 119
    },
    {
      "epoch": 0.58,
      "grad_norm": 0.2789172828197479,
      "learning_rate": 0.00016494480483301836,
      "loss": 0.3018,
      "step": 120
    },
    {
      "epoch": 0.59,
      "grad_norm": 0.14194965362548828,
      "learning_rate": 0.0001643455864733779,
      "loss": 0.2133,
      "step": 121
    },
    {
      "epoch": 0.59,
      "grad_norm": 0.13147993385791779,
      "learning_rate": 0.000163742398974869,
      "loss": 0.2281,
      "step": 122
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.16474007070064545,
      "learning_rate": 0.00016313527954493778,
      "loss": 0.2507,
      "step": 123
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.14342117309570312,
      "learning_rate": 0.00016252426563357055,
      "loss": 0.2118,
      "step": 124
    },
    {
      "epoch": 0.61,
      "grad_norm": 0.13474318385124207,
      "learning_rate": 0.00016190939493098344,
      "loss": 0.2172,
      "step": 125
    },
    {
      "epoch": 0.61,
      "grad_norm": 0.12630698084831238,
      "learning_rate": 0.00016129070536529766,
      "loss": 0.1549,
      "step": 126
    },
    {
      "epoch": 0.62,
      "grad_norm": 0.12225893884897232,
      "learning_rate": 0.00016066823510019998,
      "loss": 0.2312,
      "step": 127
    },
    {
      "epoch": 0.62,
      "grad_norm": 0.17443059384822845,
      "learning_rate": 0.00016004202253258842,
      "loss": 0.3303,
      "step": 128
    },
    {
      "epoch": 0.63,
      "grad_norm": 0.1610797941684723,
      "learning_rate": 0.00015941210629020388,
      "loss": 0.2463,
      "step": 129
    },
    {
      "epoch": 0.63,
      "grad_norm": 0.14372020959854126,
      "learning_rate": 0.00015877852522924732,
      "loss": 0.3351,
      "step": 130
    },
    {
      "epoch": 0.64,
      "grad_norm": 0.15952154994010925,
      "learning_rate": 0.00015814131843198308,
      "loss": 0.2147,
      "step": 131
    },
    {
      "epoch": 0.64,
      "grad_norm": 0.18614554405212402,
      "learning_rate": 0.00015750052520432787,
      "loss": 0.1983,
      "step": 132
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.15100689232349396,
      "learning_rate": 0.0001568561850734264,
      "loss": 0.182,
      "step": 133
    },
    {
      "epoch": 0.65,
      "grad_norm": 0.15041258931159973,
      "learning_rate": 0.00015620833778521307,
      "loss": 0.1979,
      "step": 134
    },
    {
      "epoch": 0.66,
      "grad_norm": 0.14446201920509338,
      "learning_rate": 0.00015555702330196023,
      "loss": 0.2039,
      "step": 135
    },
    {
      "epoch": 0.66,
      "grad_norm": 0.12656830251216888,
      "learning_rate": 0.0001549022817998132,
      "loss": 0.2032,
      "step": 136
    },
    {
      "epoch": 0.67,
      "grad_norm": 0.147608682513237,
      "learning_rate": 0.00015424415366631188,
      "loss": 0.1889,
      "step": 137
    },
    {
      "epoch": 0.67,
      "grad_norm": 0.1701640486717224,
      "learning_rate": 0.00015358267949789966,
      "loss": 0.2654,
      "step": 138
    },
    {
      "epoch": 0.68,
      "grad_norm": 0.1424136757850647,
      "learning_rate": 0.00015291790009741907,
      "loss": 0.2052,
      "step": 139
    },
    {
      "epoch": 0.68,
      "grad_norm": 0.1711564064025879,
      "learning_rate": 0.0001522498564715949,
      "loss": 0.226,
      "step": 140
    },
    {
      "epoch": 0.69,
      "grad_norm": 0.17195338010787964,
      "learning_rate": 0.00015157858982850475,
      "loss": 0.2366,
      "step": 141
    },
    {
      "epoch": 0.69,
      "grad_norm": 0.1596439927816391,
      "learning_rate": 0.00015090414157503714,
      "loss": 0.2387,
      "step": 142
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.1538238674402237,
      "learning_rate": 0.00015022655331433727,
      "loss": 0.2326,
      "step": 143
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.13117662072181702,
      "learning_rate": 0.00014954586684324078,
      "loss": 0.1571,
      "step": 144
    },
    {
      "epoch": 0.7,
      "grad_norm": 0.1380062848329544,
      "learning_rate": 0.00014886212414969553,
      "loss": 0.1581,
      "step": 145
    },
    {
      "epoch": 0.71,
      "grad_norm": 0.15343894064426422,
      "learning_rate": 0.00014817536741017152,
      "loss": 0.2154,
      "step": 146
    },
    {
      "epoch": 0.71,
      "grad_norm": 0.12983113527297974,
      "learning_rate": 0.00014748563898705946,
      "loss": 0.1988,
      "step": 147
    },
    {
      "epoch": 0.72,
      "grad_norm": 0.1460677534341812,
      "learning_rate": 0.00014679298142605734,
      "loss": 0.208,
      "step": 148
    },
    {
      "epoch": 0.72,
      "grad_norm": 0.18640218675136566,
      "learning_rate": 0.00014609743745354624,
      "loss": 0.2976,
      "step": 149
    },
    {
      "epoch": 0.73,
      "grad_norm": 0.1410883516073227,
      "learning_rate": 0.00014539904997395468,
      "loss": 0.1779,
      "step": 150
    },
    {
      "epoch": 0.73,
      "grad_norm": 0.1352994590997696,
      "learning_rate": 0.00014469786206711214,
      "loss": 0.1932,
      "step": 151
    },
    {
      "epoch": 0.74,
      "grad_norm": 0.16276930272579193,
      "learning_rate": 0.00014399391698559152,
      "loss": 0.19,
      "step": 152
    },
    {
      "epoch": 0.74,
      "grad_norm": 0.1330641359090805,
      "learning_rate": 0.00014328725815204144,
      "loss": 0.1529,
      "step": 153
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.1461140215396881,
      "learning_rate": 0.00014257792915650728,
      "loss": 0.1822,
      "step": 154
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.1400836855173111,
      "learning_rate": 0.0001418659737537428,
      "loss": 0.1984,
      "step": 155
    },
    {
      "epoch": 0.76,
      "grad_norm": 0.15338997542858124,
      "learning_rate": 0.00014115143586051088,
      "loss": 0.1856,
      "step": 156
    },
    {
      "epoch": 0.76,
      "grad_norm": 0.15661920607089996,
      "learning_rate": 0.00014043435955287452,
      "loss": 0.2219,
      "step": 157
    },
    {
      "epoch": 0.77,
      "grad_norm": 0.22158733010292053,
      "learning_rate": 0.00013971478906347806,
      "loss": 0.3549,
      "step": 158
    },
    {
      "epoch": 0.77,
      "grad_norm": 0.1338592767715454,
      "learning_rate": 0.00013899276877881884,
      "loss": 0.1785,
      "step": 159
    },
    {
      "epoch": 0.78,
      "grad_norm": 0.14615756273269653,
      "learning_rate": 0.000138268343236509,
      "loss": 0.2202,
      "step": 160
    },
    {
      "epoch": 0.78,
      "grad_norm": 0.1247783973813057,
      "learning_rate": 0.00013754155712252832,
      "loss": 0.2008,
      "step": 161
    },
    {
      "epoch": 0.79,
      "grad_norm": 0.18145664036273956,
      "learning_rate": 0.00013681245526846783,
      "loss": 0.2144,
      "step": 162
    },
    {
      "epoch": 0.79,
      "grad_norm": 0.1396200805902481,
      "learning_rate": 0.0001360810826487642,
      "loss": 0.2194,
      "step": 163
    },
    {
      "epoch": 0.8,
      "grad_norm": 0.15508796274662018,
      "learning_rate": 0.00013534748437792573,
      "loss": 0.2766,
      "step": 164
    },
    {
      "epoch": 0.8,
      "grad_norm": 0.15434060990810394,
      "learning_rate": 0.0001346117057077493,
      "loss": 0.188,
      "step": 165
    },
    {
      "epoch": 0.81,
      "grad_norm": 0.14731119573116302,
      "learning_rate": 0.00013387379202452917,
      "loss": 0.233,
      "step": 166
    },
    {
      "epoch": 0.81,
      "grad_norm": 0.1317255198955536,
      "learning_rate": 0.0001331337888462571,
      "loss": 0.233,
      "step": 167
    },
    {
      "epoch": 0.82,
      "grad_norm": 0.16242042183876038,
      "learning_rate": 0.00013239174181981495,
      "loss": 0.2541,
      "step": 168
    },
    {
      "epoch": 0.82,
      "grad_norm": 0.14940407872200012,
      "learning_rate": 0.00013164769671815862,
      "loss": 0.2244,
      "step": 169
    },
    {
      "epoch": 0.83,
      "grad_norm": 0.14262273907661438,
      "learning_rate": 0.00013090169943749476,
      "loss": 0.2458,
      "step": 170
    },
    {
      "epoch": 0.83,
      "grad_norm": 0.13066165149211884,
      "learning_rate": 0.00013015379599444957,
      "loss": 0.1865,
      "step": 171
    },
    {
      "epoch": 0.84,
      "grad_norm": 0.14521218836307526,
      "learning_rate": 0.0001294040325232304,
      "loss": 0.1838,
      "step": 172
    },
    {
      "epoch": 0.84,
      "grad_norm": 0.14943768084049225,
      "learning_rate": 0.00012865245527277986,
      "loss": 0.2369,
      "step": 173
    },
    {
      "epoch": 0.85,
      "grad_norm": 0.14182843267917633,
      "learning_rate": 0.00012789911060392294,
      "loss": 0.1953,
      "step": 174
    },
    {
      "epoch": 0.85,
      "grad_norm": 0.16210182011127472,
      "learning_rate": 0.00012714404498650743,
      "loss": 0.2624,
      "step": 175
    },
    {
      "epoch": 0.86,
      "grad_norm": 0.15228134393692017,
      "learning_rate": 0.0001263873049965373,
      "loss": 0.2441,
      "step": 176
    },
    {
      "epoch": 0.86,
      "grad_norm": 0.13629144430160522,
      "learning_rate": 0.00012562893731329967,
      "loss": 0.2003,
      "step": 177
    },
    {
      "epoch": 0.87,
      "grad_norm": 0.15732020139694214,
      "learning_rate": 0.0001248689887164855,
      "loss": 0.2311,
      "step": 178
    },
    {
      "epoch": 0.87,
      "grad_norm": 0.13804040849208832,
      "learning_rate": 0.00012410750608330388,
      "loss": 0.1768,
      "step": 179
    },
    {
      "epoch": 0.87,
      "grad_norm": 0.13766342401504517,
      "learning_rate": 0.00012334453638559057,
      "loss": 0.2182,
      "step": 180
    },
    {
      "epoch": 0.88,
      "grad_norm": 0.15981349349021912,
      "learning_rate": 0.0001225801266869104,
      "loss": 0.2061,
      "step": 181
    },
    {
      "epoch": 0.88,
      "grad_norm": 0.12819038331508636,
      "learning_rate": 0.00012181432413965428,
      "loss": 0.1549,
      "step": 182
    },
    {
      "epoch": 0.89,
      "grad_norm": 0.13831038773059845,
      "learning_rate": 0.00012104717598213056,
      "loss": 0.1653,
      "step": 183
    },
    {
      "epoch": 0.89,
      "grad_norm": 0.16673411428928375,
      "learning_rate": 0.00012027872953565125,
      "loss": 0.1933,
      "step": 184
    },
    {
      "epoch": 0.9,
      "grad_norm": 0.14682242274284363,
      "learning_rate": 0.00011950903220161285,
      "loss": 0.202,
      "step": 185
    },
    {
      "epoch": 0.9,
      "grad_norm": 0.1462392807006836,
      "learning_rate": 0.00011873813145857249,
      "loss": 0.2377,
      "step": 186
    },
    {
      "epoch": 0.91,
      "grad_norm": 0.12843555212020874,
      "learning_rate": 0.00011796607485931928,
      "loss": 0.2125,
      "step": 187
    },
    {
      "epoch": 0.91,
      "grad_norm": 0.15481220185756683,
      "learning_rate": 0.00011719291002794096,
      "loss": 0.2539,
      "step": 188
    },
    {
      "epoch": 0.92,
      "grad_norm": 0.13950768113136292,
      "learning_rate": 0.0001164186846568863,
      "loss": 0.1972,
      "step": 189
    },
    {
      "epoch": 0.92,
      "grad_norm": 0.11621260643005371,
      "learning_rate": 0.0001156434465040231,
      "loss": 0.1856,
      "step": 190
    },
    {
      "epoch": 0.93,
      "grad_norm": 0.14099712669849396,
      "learning_rate": 0.00011486724338969232,
      "loss": 0.1916,
      "step": 191
    },
    {
      "epoch": 0.93,
      "grad_norm": 0.1282750517129898,
      "learning_rate": 0.00011409012319375827,
      "loss": 0.1836,
      "step": 192
    },
    {
      "epoch": 0.94,
      "grad_norm": 0.15066012740135193,
      "learning_rate": 0.00011331213385265524,
      "loss": 0.2432,
      "step": 193
    },
    {
      "epoch": 0.94,
      "grad_norm": 0.14334805309772491,
      "learning_rate": 0.00011253332335643043,
      "loss": 0.1749,
      "step": 194
    },
    {
      "epoch": 0.95,
      "grad_norm": 0.15670594573020935,
      "learning_rate": 0.00011175373974578378,
      "loss": 0.2479,
      "step": 195
    },
    {
      "epoch": 0.95,
      "grad_norm": 0.15438470244407654,
      "learning_rate": 0.00011097343110910452,
      "loss": 0.2356,
      "step": 196
    },
    {
      "epoch": 0.96,
      "grad_norm": 0.1420874148607254,
      "learning_rate": 0.000110192445579505,
      "loss": 0.2662,
      "step": 197
    },
    {
      "epoch": 0.96,
      "grad_norm": 0.1418399214744568,
      "learning_rate": 0.00010941083133185146,
      "loss": 0.1785,
      "step": 198
    },
    {
      "epoch": 0.97,
      "grad_norm": 0.1280946284532547,
      "learning_rate": 0.00010862863657979237,
      "loss": 0.1652,
      "step": 199
    },
    {
      "epoch": 0.97,
      "grad_norm": 0.14323323965072632,
      "learning_rate": 0.0001078459095727845,
      "loss": 0.2104,
      "step": 200
    },
    {
      "epoch": 0.98,
      "grad_norm": 0.10913383960723877,
      "learning_rate": 0.00010706269859311669,
      "loss": 0.1448,
      "step": 201
    },
    {
      "epoch": 0.98,
      "grad_norm": 0.12007103115320206,
      "learning_rate": 0.00010627905195293135,
      "loss": 0.1263,
      "step": 202
    },
    {
      "epoch": 0.99,
      "grad_norm": 0.1433536857366562,
      "learning_rate": 0.0001054950179912446,
      "loss": 0.1831,
      "step": 203
    },
    {
      "epoch": 0.99,
      "grad_norm": 0.1542392522096634,
      "learning_rate": 0.00010471064507096426,
      "loss": 0.2828,
      "step": 204
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.18856601417064667,
      "learning_rate": 0.00010392598157590688,
      "loss": 0.183,
      "step": 205
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.13536609709262848,
      "learning_rate": 0.00010314107590781284,
      "loss": 0.2161,
      "step": 206
    },
    {
      "epoch": 1.0,
      "eval_loss": 0.22577418386936188,
      "eval_runtime": 13.2186,
      "eval_samples_per_second": 1.891,
      "eval_steps_per_second": 1.891,
      "step": 206
    },
    {
      "epoch": 1.01,
      "grad_norm": 0.1510825753211975,
      "learning_rate": 0.00010235597648336104,
      "loss": 0.174,
      "step": 207
    },
    {
      "epoch": 1.01,
      "grad_norm": 0.13914929330348969,
      "learning_rate": 0.00010157073173118208,
      "loss": 0.2656,
      "step": 208
    },
    {
      "epoch": 1.02,
      "grad_norm": 0.15691936016082764,
      "learning_rate": 0.00010078539008887114,
      "loss": 0.2394,
      "step": 209
    },
    {
      "epoch": 1.02,
      "grad_norm": 0.15231585502624512,
      "learning_rate": 0.0001,
      "loss": 0.2392,
      "step": 210
    },
    {
      "epoch": 1.03,
      "grad_norm": 0.14635689556598663,
      "learning_rate": 9.921460991112891e-05,
      "loss": 0.1702,
      "step": 211
    },
    {
      "epoch": 1.03,
      "grad_norm": 0.1327444165945053,
      "learning_rate": 9.842926826881796e-05,
      "loss": 0.2196,
      "step": 212
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.1320512443780899,
      "learning_rate": 9.764402351663901e-05,
      "loss": 0.1945,
      "step": 213
    },
    {
      "epoch": 1.01,
      "grad_norm": 0.13175004720687866,
      "learning_rate": 9.685892409218717e-05,
      "loss": 0.209,
      "step": 214
    },
    {
      "epoch": 1.01,
      "grad_norm": 0.11285891383886337,
      "learning_rate": 9.607401842409317e-05,
      "loss": 0.1671,
      "step": 215
    },
    {
      "epoch": 1.02,
      "grad_norm": 0.13914702832698822,
      "learning_rate": 9.528935492903575e-05,
      "loss": 0.2156,
      "step": 216
    },
    {
      "epoch": 1.02,
      "grad_norm": 0.11163745075464249,
      "learning_rate": 9.450498200875546e-05,
      "loss": 0.1547,
      "step": 217
    },
    {
      "epoch": 1.03,
      "grad_norm": 0.12759771943092346,
      "learning_rate": 9.372094804706867e-05,
      "loss": 0.1698,
      "step": 218
    },
    {
      "epoch": 1.03,
      "grad_norm": 0.15159080922603607,
      "learning_rate": 9.293730140688336e-05,
      "loss": 0.2386,
      "step": 219
    },
    {
      "epoch": 1.04,
      "grad_norm": 0.1276952624320984,
      "learning_rate": 9.215409042721552e-05,
      "loss": 0.1755,
      "step": 220
    },
    {
      "epoch": 1.04,
      "grad_norm": 0.13293467462062836,
      "learning_rate": 9.137136342020768e-05,
      "loss": 0.1732,
      "step": 221
    },
    {
      "epoch": 1.05,
      "grad_norm": 0.13667437434196472,
      "learning_rate": 9.058916866814858e-05,
      "loss": 0.1606,
      "step": 222
    },
    {
      "epoch": 1.05,
      "grad_norm": 0.1378675252199173,
      "learning_rate": 8.980755442049502e-05,
      "loss": 0.1656,
      "step": 223
    },
    {
      "epoch": 1.06,
      "grad_norm": 0.13927237689495087,
      "learning_rate": 8.902656889089548e-05,
      "loss": 0.1615,
      "step": 224
    },
    {
      "epoch": 1.06,
      "grad_norm": 0.12705956399440765,
      "learning_rate": 8.824626025421626e-05,
      "loss": 0.1235,
      "step": 225
    },
    {
      "epoch": 1.07,
      "grad_norm": 0.17281681299209595,
      "learning_rate": 8.746667664356956e-05,
      "loss": 0.1835,
      "step": 226
    },
    {
      "epoch": 1.07,
      "grad_norm": 0.15210822224617004,
      "learning_rate": 8.668786614734478e-05,
      "loss": 0.1514,
      "step": 227
    },
    {
      "epoch": 1.08,
      "grad_norm": 0.17541825771331787,
      "learning_rate": 8.590987680624174e-05,
      "loss": 0.1403,
      "step": 228
    },
    {
      "epoch": 1.08,
      "grad_norm": 0.13942864537239075,
      "learning_rate": 8.51327566103077e-05,
      "loss": 0.1423,
      "step": 229
    },
    {
      "epoch": 1.09,
      "grad_norm": 0.16645929217338562,
      "learning_rate": 8.435655349597689e-05,
      "loss": 0.1773,
      "step": 230
    },
    {
      "epoch": 1.09,
      "grad_norm": 0.16734780371189117,
      "learning_rate": 8.358131534311372e-05,
      "loss": 0.2327,
      "step": 231
    },
    {
      "epoch": 1.09,
      "grad_norm": 0.18149012327194214,
      "learning_rate": 8.280708997205904e-05,
      "loss": 0.182,
      "step": 232
    },
    {
      "epoch": 1.1,
      "grad_norm": 0.1811237931251526,
      "learning_rate": 8.203392514068074e-05,
      "loss": 0.2129,
      "step": 233
    },
    {
      "epoch": 1.1,
      "grad_norm": 0.17224758863449097,
      "learning_rate": 8.126186854142752e-05,
      "loss": 0.1836,
      "step": 234
    },
    {
      "epoch": 1.11,
      "grad_norm": 0.15472543239593506,
      "learning_rate": 8.049096779838719e-05,
      "loss": 0.1418,
      "step": 235
    },
    {
      "epoch": 1.11,
      "grad_norm": 0.15701377391815186,
      "learning_rate": 7.972127046434878e-05,
      "loss": 0.1749,
      "step": 236
    },
    {
      "epoch": 1.12,
      "grad_norm": 0.15067677199840546,
      "learning_rate": 7.895282401786945e-05,
      "loss": 0.1347,
      "step": 237
    },
    {
      "epoch": 1.12,
      "grad_norm": 0.12470359355211258,
      "learning_rate": 7.818567586034577e-05,
      "loss": 0.1155,
      "step": 238
    },
    {
      "epoch": 1.13,
      "grad_norm": 0.15522097051143646,
      "learning_rate": 7.741987331308964e-05,
      "loss": 0.1537,
      "step": 239
    },
    {
      "epoch": 1.13,
      "grad_norm": 0.17973066866397858,
      "learning_rate": 7.66554636144095e-05,
      "loss": 0.1844,
      "step": 240
    },
    {
      "epoch": 1.14,
      "grad_norm": 0.14860330522060394,
      "learning_rate": 7.589249391669616e-05,
      "loss": 0.1643,
      "step": 241
    },
    {
      "epoch": 1.14,
      "grad_norm": 0.16414889693260193,
      "learning_rate": 7.513101128351454e-05,
      "loss": 0.1533,
      "step": 242
    },
    {
      "epoch": 1.15,
      "grad_norm": 0.14017800986766815,
      "learning_rate": 7.437106268670034e-05,
      "loss": 0.1556,
      "step": 243
    },
    {
      "epoch": 1.15,
      "grad_norm": 0.1377851516008377,
      "learning_rate": 7.361269500346274e-05,
      "loss": 0.1175,
      "step": 244
    },
    {
      "epoch": 1.16,
      "grad_norm": 0.13657085597515106,
      "learning_rate": 7.285595501349258e-05,
      "loss": 0.1182,
      "step": 245
    },
    {
      "epoch": 1.16,
      "grad_norm": 0.16157828271389008,
      "learning_rate": 7.210088939607708e-05,
      "loss": 0.2083,
      "step": 246
    },
    {
      "epoch": 1.17,
      "grad_norm": 0.15126171708106995,
      "learning_rate": 7.134754472722017e-05,
      "loss": 0.1384,
      "step": 247
    },
    {
      "epoch": 1.17,
      "grad_norm": 0.1582876294851303,
      "learning_rate": 7.059596747676962e-05,
      "loss": 0.135,
      "step": 248
    },
    {
      "epoch": 1.18,
      "grad_norm": 0.15393657982349396,
      "learning_rate": 6.984620400555044e-05,
      "loss": 0.1529,
      "step": 249
    },
    {
      "epoch": 1.18,
      "grad_norm": 0.16819702088832855,
      "learning_rate": 6.909830056250527e-05,
      "loss": 0.1589,
      "step": 250
    },
    {
      "epoch": 1.19,
      "grad_norm": 0.17828214168548584,
      "learning_rate": 6.835230328184138e-05,
      "loss": 0.1549,
      "step": 251
    },
    {
      "epoch": 1.19,
      "grad_norm": 0.17099595069885254,
      "learning_rate": 6.760825818018508e-05,
      "loss": 0.1767,
      "step": 252
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.17799484729766846,
      "learning_rate": 6.68662111537429e-05,
      "loss": 0.1274,
      "step": 253
    },
    {
      "epoch": 1.2,
      "grad_norm": 0.21973256766796112,
      "learning_rate": 6.612620797547087e-05,
      "loss": 0.1664,
      "step": 254
    },
    {
      "epoch": 1.21,
      "grad_norm": 0.19034262001514435,
      "learning_rate": 6.538829429225069e-05,
      "loss": 0.1932,
      "step": 255
    },
    {
      "epoch": 1.21,
      "grad_norm": 0.20569321513175964,
      "learning_rate": 6.465251562207431e-05,
      "loss": 0.1971,
      "step": 256
    },
    {
      "epoch": 1.22,
      "grad_norm": 0.1574014127254486,
      "learning_rate": 6.391891735123582e-05,
      "loss": 0.1255,
      "step": 257
    },
    {
      "epoch": 1.22,
      "grad_norm": 0.1868090033531189,
      "learning_rate": 6.318754473153221e-05,
      "loss": 0.202,
      "step": 258
    },
    {
      "epoch": 1.23,
      "grad_norm": 0.15771806240081787,
      "learning_rate": 6.245844287747168e-05,
      "loss": 0.1348,
      "step": 259
    },
    {
      "epoch": 1.23,
      "grad_norm": 0.1910022646188736,
      "learning_rate": 6.173165676349103e-05,
      "loss": 0.1399,
      "step": 260
    },
    {
      "epoch": 1.24,
      "grad_norm": 0.192474827170372,
      "learning_rate": 6.1007231221181206e-05,
      "loss": 0.1497,
      "step": 261
    },
    {
      "epoch": 1.24,
      "grad_norm": 0.20803901553153992,
      "learning_rate": 6.0285210936521955e-05,
      "loss": 0.1459,
      "step": 262
    },
    {
      "epoch": 1.25,
      "grad_norm": 0.16716165840625763,
      "learning_rate": 5.956564044712551e-05,
      "loss": 0.1352,
      "step": 263
    },
    {
      "epoch": 1.25,
      "grad_norm": 0.2101832777261734,
      "learning_rate": 5.884856413948913e-05,
      "loss": 0.1676,
      "step": 264
    },
    {
      "epoch": 1.26,
      "grad_norm": 0.19041313230991364,
      "learning_rate": 5.8134026246257225e-05,
      "loss": 0.1492,
      "step": 265
    },
    {
      "epoch": 1.26,
      "grad_norm": 0.17860420048236847,
      "learning_rate": 5.7422070843492734e-05,
      "loss": 0.1058,
      "step": 266
    },
    {
      "epoch": 1.26,
      "grad_norm": 0.15917859971523285,
      "learning_rate": 5.671274184795865e-05,
      "loss": 0.1315,
      "step": 267
    },
    {
      "epoch": 1.27,
      "grad_norm": 0.19338840246200562,
      "learning_rate": 5.6006083014408484e-05,
      "loss": 0.1436,
      "step": 268
    },
    {
      "epoch": 1.27,
      "grad_norm": 0.27872714400291443,
      "learning_rate": 5.53021379328879e-05,
      "loss": 0.1233,
      "step": 269
    },
    {
      "epoch": 1.28,
      "grad_norm": 0.16847865283489227,
      "learning_rate": 5.4600950026045326e-05,
      "loss": 0.1166,
      "step": 270
    },
    {
      "epoch": 1.28,
      "grad_norm": 0.16588595509529114,
      "learning_rate": 5.390256254645378e-05,
      "loss": 0.1285,
      "step": 271
    },
    {
      "epoch": 1.29,
      "grad_norm": 0.18695320188999176,
      "learning_rate": 5.320701857394268e-05,
      "loss": 0.1741,
      "step": 272
    },
    {
      "epoch": 1.29,
      "grad_norm": 0.18936707079410553,
      "learning_rate": 5.251436101294056e-05,
      "loss": 0.1559,
      "step": 273
    },
    {
      "epoch": 1.3,
      "grad_norm": 0.1948186159133911,
      "learning_rate": 5.182463258982846e-05,
      "loss": 0.1555,
      "step": 274
    },
    {
      "epoch": 1.3,
      "grad_norm": 0.1960403323173523,
      "learning_rate": 5.113787585030454e-05,
      "loss": 0.1562,
      "step": 275
    },
    {
      "epoch": 1.31,
      "grad_norm": 0.17642433941364288,
      "learning_rate": 5.045413315675924e-05,
      "loss": 0.1545,
      "step": 276
    },
    {
      "epoch": 1.31,
      "grad_norm": 0.2207774519920349,
      "learning_rate": 4.977344668566275e-05,
      "loss": 0.1866,
      "step": 277
    },
    {
      "epoch": 1.32,
      "grad_norm": 0.1559211015701294,
      "learning_rate": 4.909585842496287e-05,
      "loss": 0.1201,
      "step": 278
    },
    {
      "epoch": 1.32,
      "grad_norm": 0.16633960604667664,
      "learning_rate": 4.842141017149526e-05,
      "loss": 0.1229,
      "step": 279
    },
    {
      "epoch": 1.33,
      "grad_norm": 0.1892048567533493,
      "learning_rate": 4.7750143528405126e-05,
      "loss": 0.166,
      "step": 280
    },
    {
      "epoch": 1.33,
      "grad_norm": 0.1674378365278244,
      "learning_rate": 4.708209990258095e-05,
      "loss": 0.1496,
      "step": 281
    },
    {
      "epoch": 1.34,
      "grad_norm": 0.17542515695095062,
      "learning_rate": 4.6417320502100316e-05,
      "loss": 0.1326,
      "step": 282
    },
    {
      "epoch": 1.34,
      "grad_norm": 0.17793145775794983,
      "learning_rate": 4.575584633368815e-05,
      "loss": 0.1599,
      "step": 283
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.2022085189819336,
      "learning_rate": 4.5097718200186814e-05,
      "loss": 0.1681,
      "step": 284
    },
    {
      "epoch": 1.35,
      "grad_norm": 0.1831180453300476,
      "learning_rate": 4.444297669803981e-05,
      "loss": 0.176,
      "step": 285
    },
    {
      "epoch": 1.36,
      "grad_norm": 0.15007005631923676,
      "learning_rate": 4.379166221478697e-05,
      "loss": 0.1247,
      "step": 286
    },
    {
      "epoch": 1.36,
      "grad_norm": 0.17191900312900543,
      "learning_rate": 4.31438149265736e-05,
      "loss": 0.1494,
      "step": 287
    },
    {
      "epoch": 1.37,
      "grad_norm": 0.21848122775554657,
      "learning_rate": 4.249947479567218e-05,
      "loss": 0.2025,
      "step": 288
    },
    {
      "epoch": 1.37,
      "grad_norm": 0.21705280244350433,
      "learning_rate": 4.185868156801694e-05,
      "loss": 0.1822,
      "step": 289
    },
    {
      "epoch": 1.38,
      "grad_norm": 0.20829765498638153,
      "learning_rate": 4.12214747707527e-05,
      "loss": 0.1926,
      "step": 290
    },
    {
      "epoch": 1.38,
      "grad_norm": 0.18017052114009857,
      "learning_rate": 4.058789370979615e-05,
      "loss": 0.1399,
      "step": 291
    },
    {
      "epoch": 1.39,
      "grad_norm": 0.1806417554616928,
      "learning_rate": 3.9957977467411615e-05,
      "loss": 0.1443,
      "step": 292
    },
    {
      "epoch": 1.39,
      "grad_norm": 0.1974053829908371,
      "learning_rate": 3.933176489980005e-05,
      "loss": 0.1555,
      "step": 293
    },
    {
      "epoch": 1.4,
      "grad_norm": 0.18204301595687866,
      "learning_rate": 3.8709294634702376e-05,
      "loss": 0.1732,
      "step": 294
    },
    {
      "epoch": 1.4,
      "grad_norm": 0.20640143752098083,
      "learning_rate": 3.8090605069016595e-05,
      "loss": 0.1682,
      "step": 295
    },
    {
      "epoch": 1.41,
      "grad_norm": 0.15330561995506287,
      "learning_rate": 3.747573436642951e-05,
      "loss": 0.1061,
      "step": 296
    },
    {
      "epoch": 1.41,
      "grad_norm": 0.17504094541072845,
      "learning_rate": 3.686472045506223e-05,
      "loss": 0.1208,
      "step": 297
    },
    {
      "epoch": 1.42,
      "grad_norm": 0.19445376098155975,
      "learning_rate": 3.6257601025131026e-05,
      "loss": 0.1637,
      "step": 298
    },
    {
      "epoch": 1.42,
      "grad_norm": 0.20897901058197021,
      "learning_rate": 3.565441352662211e-05,
      "loss": 0.1518,
      "step": 299
    },
    {
      "epoch": 1.43,
      "grad_norm": 0.17896625399589539,
      "learning_rate": 3.5055195166981645e-05,
      "loss": 0.1512,
      "step": 300
    },
    {
      "epoch": 1.43,
      "grad_norm": 0.20567168295383453,
      "learning_rate": 3.445998290882062e-05,
      "loss": 0.1551,
      "step": 301
    },
    {
      "epoch": 1.43,
      "grad_norm": 0.19793061912059784,
      "learning_rate": 3.386881346763483e-05,
      "loss": 0.1436,
      "step": 302
    },
    {
      "epoch": 1.44,
      "grad_norm": 0.1717611402273178,
      "learning_rate": 3.328172330954001e-05,
      "loss": 0.153,
      "step": 303
    },
    {
      "epoch": 1.44,
      "grad_norm": 0.18997669219970703,
      "learning_rate": 3.269874864902269e-05,
      "loss": 0.1352,
      "step": 304
    },
    {
      "epoch": 1.45,
      "grad_norm": 0.18684212863445282,
      "learning_rate": 3.211992544670582e-05,
      "loss": 0.133,
      "step": 305
    },
    {
      "epoch": 1.45,
      "grad_norm": 0.1856592297554016,
      "learning_rate": 3.154528940713113e-05,
      "loss": 0.14,
      "step": 306
    },
    {
      "epoch": 1.46,
      "grad_norm": 0.1783703714609146,
      "learning_rate": 3.0974875976556284e-05,
      "loss": 0.1368,
      "step": 307
    },
    {
      "epoch": 1.46,
      "grad_norm": 0.16114623844623566,
      "learning_rate": 3.0408720340768572e-05,
      "loss": 0.1353,
      "step": 308
    },
    {
      "epoch": 1.47,
      "grad_norm": 0.24124671518802643,
      "learning_rate": 2.9846857422914433e-05,
      "loss": 0.1821,
      "step": 309
    },
    {
      "epoch": 1.47,
      "eval_loss": 0.235604926943779,
      "eval_runtime": 13.2989,
      "eval_samples_per_second": 1.88,
      "eval_steps_per_second": 1.88,
      "step": 309
    }
  ],
  "logging_steps": 1,
  "max_steps": 410,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 103,
  "total_flos": 1.927039310801584e+18,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}