{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9953917050691244,
  "eval_steps": 500,
  "global_step": 72,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.013824884792626729,
      "grad_norm": 31.00213623046875,
      "learning_rate": 5.0000000000000004e-08,
      "loss": 2.2089,
      "step": 1
    },
    {
      "epoch": 0.027649769585253458,
      "grad_norm": 30.27136993408203,
      "learning_rate": 1.0000000000000001e-07,
      "loss": 2.1536,
      "step": 2
    },
    {
      "epoch": 0.041474654377880185,
      "grad_norm": 30.48703384399414,
      "learning_rate": 1.5000000000000002e-07,
      "loss": 2.1581,
      "step": 3
    },
    {
      "epoch": 0.055299539170506916,
      "grad_norm": 30.779329299926758,
      "learning_rate": 2.0000000000000002e-07,
      "loss": 2.1741,
      "step": 4
    },
    {
      "epoch": 0.06912442396313365,
      "grad_norm": 31.22808837890625,
      "learning_rate": 2.5000000000000004e-07,
      "loss": 2.1864,
      "step": 5
    },
    {
      "epoch": 0.08294930875576037,
      "grad_norm": 30.783327102661133,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 2.1993,
      "step": 6
    },
    {
      "epoch": 0.0967741935483871,
      "grad_norm": 30.57423210144043,
      "learning_rate": 3.5000000000000004e-07,
      "loss": 2.1506,
      "step": 7
    },
    {
      "epoch": 0.11059907834101383,
      "grad_norm": 30.952186584472656,
      "learning_rate": 4.0000000000000003e-07,
      "loss": 2.1599,
      "step": 8
    },
    {
      "epoch": 0.12442396313364056,
      "grad_norm": 30.37245750427246,
      "learning_rate": 4.5000000000000003e-07,
      "loss": 2.1572,
      "step": 9
    },
    {
      "epoch": 0.1382488479262673,
      "grad_norm": 30.930192947387695,
      "learning_rate": 5.000000000000001e-07,
      "loss": 2.1447,
      "step": 10
    },
    {
      "epoch": 0.15207373271889402,
      "grad_norm": 29.735448837280273,
      "learning_rate": 5.5e-07,
      "loss": 2.0742,
      "step": 11
    },
    {
      "epoch": 0.16589861751152074,
      "grad_norm": 29.62826156616211,
      "learning_rate": 6.000000000000001e-07,
      "loss": 2.061,
      "step": 12
    },
    {
      "epoch": 0.17972350230414746,
      "grad_norm": 28.937463760375977,
      "learning_rate": 6.5e-07,
      "loss": 1.9974,
      "step": 13
    },
    {
      "epoch": 0.1935483870967742,
      "grad_norm": 29.24833106994629,
      "learning_rate": 7.000000000000001e-07,
      "loss": 1.9833,
      "step": 14
    },
    {
      "epoch": 0.2073732718894009,
      "grad_norm": 28.122018814086914,
      "learning_rate": 7.5e-07,
      "loss": 1.8934,
      "step": 15
    },
    {
      "epoch": 0.22119815668202766,
      "grad_norm": 28.059659957885742,
      "learning_rate": 8.000000000000001e-07,
      "loss": 1.875,
      "step": 16
    },
    {
      "epoch": 0.2350230414746544,
      "grad_norm": 27.361961364746094,
      "learning_rate": 8.500000000000001e-07,
      "loss": 1.8009,
      "step": 17
    },
    {
      "epoch": 0.2488479262672811,
      "grad_norm": 26.721765518188477,
      "learning_rate": 9.000000000000001e-07,
      "loss": 1.7116,
      "step": 18
    },
    {
      "epoch": 0.2626728110599078,
      "grad_norm": 25.37330436706543,
      "learning_rate": 9.500000000000001e-07,
      "loss": 1.5608,
      "step": 19
    },
    {
      "epoch": 0.2764976958525346,
      "grad_norm": 25.81206703186035,
      "learning_rate": 1.0000000000000002e-06,
      "loss": 1.5043,
      "step": 20
    },
    {
      "epoch": 0.2903225806451613,
      "grad_norm": 25.539344787597656,
      "learning_rate": 1.0500000000000001e-06,
      "loss": 1.3673,
      "step": 21
    },
    {
      "epoch": 0.30414746543778803,
      "grad_norm": 25.097164154052734,
      "learning_rate": 1.1e-06,
      "loss": 1.2029,
      "step": 22
    },
    {
      "epoch": 0.31797235023041476,
      "grad_norm": 24.619497299194336,
      "learning_rate": 1.1500000000000002e-06,
      "loss": 1.0458,
      "step": 23
    },
    {
      "epoch": 0.3317972350230415,
      "grad_norm": 23.820302963256836,
      "learning_rate": 1.2000000000000002e-06,
      "loss": 0.8723,
      "step": 24
    },
    {
      "epoch": 0.3456221198156682,
      "grad_norm": 23.12735939025879,
      "learning_rate": 1.25e-06,
      "loss": 0.7183,
      "step": 25
    },
    {
      "epoch": 0.35944700460829493,
      "grad_norm": 20.127134323120117,
      "learning_rate": 1.3e-06,
      "loss": 0.5248,
      "step": 26
    },
    {
      "epoch": 0.37327188940092165,
      "grad_norm": 15.901495933532715,
      "learning_rate": 1.3500000000000002e-06,
      "loss": 0.3689,
      "step": 27
    },
    {
      "epoch": 0.3870967741935484,
      "grad_norm": 11.053832054138184,
      "learning_rate": 1.4000000000000001e-06,
      "loss": 0.2482,
      "step": 28
    },
    {
      "epoch": 0.4009216589861751,
      "grad_norm": 7.248495578765869,
      "learning_rate": 1.45e-06,
      "loss": 0.1847,
      "step": 29
    },
    {
      "epoch": 0.4147465437788018,
      "grad_norm": 5.378540515899658,
      "learning_rate": 1.5e-06,
      "loss": 0.1423,
      "step": 30
    },
    {
      "epoch": 0.42857142857142855,
      "grad_norm": 3.8371808528900146,
      "learning_rate": 1.5500000000000002e-06,
      "loss": 0.1152,
      "step": 31
    },
    {
      "epoch": 0.4423963133640553,
      "grad_norm": 2.2655274868011475,
      "learning_rate": 1.6000000000000001e-06,
      "loss": 0.0845,
      "step": 32
    },
    {
      "epoch": 0.45622119815668205,
      "grad_norm": 1.5746861696243286,
      "learning_rate": 1.6500000000000003e-06,
      "loss": 0.0711,
      "step": 33
    },
    {
      "epoch": 0.4700460829493088,
      "grad_norm": 1.3510947227478027,
      "learning_rate": 1.7000000000000002e-06,
      "loss": 0.0734,
      "step": 34
    },
    {
      "epoch": 0.4838709677419355,
      "grad_norm": 0.9737389087677002,
      "learning_rate": 1.75e-06,
      "loss": 0.0651,
      "step": 35
    },
    {
      "epoch": 0.4976958525345622,
      "grad_norm": 0.9815284609794617,
      "learning_rate": 1.8000000000000001e-06,
      "loss": 0.0593,
      "step": 36
    },
    {
      "epoch": 0.511520737327189,
      "grad_norm": 0.8567912578582764,
      "learning_rate": 1.85e-06,
      "loss": 0.0543,
      "step": 37
    },
    {
      "epoch": 0.5253456221198156,
      "grad_norm": 0.6773302555084229,
      "learning_rate": 1.9000000000000002e-06,
      "loss": 0.0622,
      "step": 38
    },
    {
      "epoch": 0.5391705069124424,
      "grad_norm": 0.49936285614967346,
      "learning_rate": 1.9500000000000004e-06,
      "loss": 0.0511,
      "step": 39
    },
    {
      "epoch": 0.5529953917050692,
      "grad_norm": 0.6253588795661926,
      "learning_rate": 2.0000000000000003e-06,
      "loss": 0.0478,
      "step": 40
    },
    {
      "epoch": 0.5668202764976958,
      "grad_norm": 0.5103089809417725,
      "learning_rate": 2.05e-06,
      "loss": 0.0465,
      "step": 41
    },
    {
      "epoch": 0.5806451612903226,
      "grad_norm": 0.29294702410697937,
      "learning_rate": 2.1000000000000002e-06,
      "loss": 0.0456,
      "step": 42
    },
    {
      "epoch": 0.5944700460829493,
      "grad_norm": 0.4237954616546631,
      "learning_rate": 2.15e-06,
      "loss": 0.0501,
      "step": 43
    },
    {
      "epoch": 0.6082949308755761,
      "grad_norm": 0.42243412137031555,
      "learning_rate": 2.2e-06,
      "loss": 0.0388,
      "step": 44
    },
    {
      "epoch": 0.6221198156682027,
      "grad_norm": 0.37881818413734436,
      "learning_rate": 2.25e-06,
      "loss": 0.0415,
      "step": 45
    },
    {
      "epoch": 0.6359447004608295,
      "grad_norm": 0.4941152036190033,
      "learning_rate": 2.3000000000000004e-06,
      "loss": 0.045,
      "step": 46
    },
    {
      "epoch": 0.6497695852534562,
      "grad_norm": 0.3046450912952423,
      "learning_rate": 2.35e-06,
      "loss": 0.0386,
      "step": 47
    },
    {
      "epoch": 0.663594470046083,
      "grad_norm": 0.39361852407455444,
      "learning_rate": 2.4000000000000003e-06,
      "loss": 0.0447,
      "step": 48
    },
    {
      "epoch": 0.6774193548387096,
      "grad_norm": 0.5190001130104065,
      "learning_rate": 2.4500000000000003e-06,
      "loss": 0.0364,
      "step": 49
    },
    {
      "epoch": 0.6912442396313364,
      "grad_norm": 0.372072696685791,
      "learning_rate": 2.5e-06,
      "loss": 0.043,
      "step": 50
    },
    {
      "epoch": 0.7050691244239631,
      "grad_norm": 0.3756551146507263,
      "learning_rate": 2.55e-06,
      "loss": 0.0424,
      "step": 51
    },
    {
      "epoch": 0.7188940092165899,
      "grad_norm": 0.4593554437160492,
      "learning_rate": 2.6e-06,
      "loss": 0.0387,
      "step": 52
    },
    {
      "epoch": 0.7327188940092166,
      "grad_norm": 0.2931855618953705,
      "learning_rate": 2.6500000000000005e-06,
      "loss": 0.0396,
      "step": 53
    },
    {
      "epoch": 0.7465437788018433,
      "grad_norm": 0.38429534435272217,
      "learning_rate": 2.7000000000000004e-06,
      "loss": 0.0373,
      "step": 54
    },
    {
      "epoch": 0.7603686635944701,
      "grad_norm": 0.3506857752799988,
      "learning_rate": 2.7500000000000004e-06,
      "loss": 0.04,
      "step": 55
    },
    {
      "epoch": 0.7741935483870968,
      "grad_norm": 0.29847028851509094,
      "learning_rate": 2.8000000000000003e-06,
      "loss": 0.0369,
      "step": 56
    },
    {
      "epoch": 0.7880184331797235,
      "grad_norm": 0.3653375506401062,
      "learning_rate": 2.85e-06,
      "loss": 0.0396,
      "step": 57
    },
    {
      "epoch": 0.8018433179723502,
      "grad_norm": 0.3163083791732788,
      "learning_rate": 2.9e-06,
      "loss": 0.0337,
      "step": 58
    },
    {
      "epoch": 0.815668202764977,
      "grad_norm": 0.3734363615512848,
      "learning_rate": 2.95e-06,
      "loss": 0.0327,
      "step": 59
    },
    {
      "epoch": 0.8294930875576036,
      "grad_norm": 0.29547712206840515,
      "learning_rate": 3e-06,
      "loss": 0.0365,
      "step": 60
    },
    {
      "epoch": 0.8433179723502304,
      "grad_norm": 0.4041007161140442,
      "learning_rate": 3.05e-06,
      "loss": 0.038,
      "step": 61
    },
    {
      "epoch": 0.8571428571428571,
      "grad_norm": 0.3602149784564972,
      "learning_rate": 3.1000000000000004e-06,
      "loss": 0.033,
      "step": 62
    },
    {
      "epoch": 0.8709677419354839,
      "grad_norm": 0.2948857545852661,
      "learning_rate": 3.1500000000000003e-06,
      "loss": 0.0386,
      "step": 63
    },
    {
      "epoch": 0.8847926267281107,
      "grad_norm": 0.39098358154296875,
      "learning_rate": 3.2000000000000003e-06,
      "loss": 0.0323,
      "step": 64
    },
    {
      "epoch": 0.8986175115207373,
      "grad_norm": 0.3692062795162201,
      "learning_rate": 3.2500000000000002e-06,
      "loss": 0.0309,
      "step": 65
    },
    {
      "epoch": 0.9124423963133641,
      "grad_norm": 0.3967229425907135,
      "learning_rate": 3.3000000000000006e-06,
      "loss": 0.0346,
      "step": 66
    },
    {
      "epoch": 0.9262672811059908,
      "grad_norm": 0.47776708006858826,
      "learning_rate": 3.3500000000000005e-06,
      "loss": 0.0355,
      "step": 67
    },
    {
      "epoch": 0.9400921658986175,
      "grad_norm": 0.21545131504535675,
      "learning_rate": 3.4000000000000005e-06,
      "loss": 0.0294,
      "step": 68
    },
    {
      "epoch": 0.9539170506912442,
      "grad_norm": 0.23738539218902588,
      "learning_rate": 3.45e-06,
      "loss": 0.0308,
      "step": 69
    },
    {
      "epoch": 0.967741935483871,
      "grad_norm": 0.29174014925956726,
      "learning_rate": 3.5e-06,
      "loss": 0.0312,
      "step": 70
    },
    {
      "epoch": 0.9815668202764977,
      "grad_norm": 0.38475602865219116,
      "learning_rate": 3.5500000000000003e-06,
      "loss": 0.0324,
      "step": 71
    },
    {
      "epoch": 0.9953917050691244,
      "grad_norm": 0.4077378809452057,
      "learning_rate": 3.6000000000000003e-06,
      "loss": 0.0297,
      "step": 72
    }
  ],
  "logging_steps": 1,
  "max_steps": 432,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 6,
  "save_steps": 72,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 4.631891949769458e+18,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}