{
  "best_metric": 0.8638888888888889,
  "best_model_checkpoint": "vit-base-patch16-224-in21k-finetuned-inaturalist/checkpoint-140",
  "epoch": 80.0,
  "eval_steps": 500,
  "global_step": 200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.8,
      "eval_accuracy": 0.03333333333333333,
      "eval_loss": 3.200693368911743,
      "eval_runtime": 3.9107,
      "eval_samples_per_second": 184.111,
      "eval_steps_per_second": 1.534,
      "step": 2
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.044444444444444446,
      "eval_loss": 3.188920021057129,
      "eval_runtime": 3.5624,
      "eval_samples_per_second": 202.11,
      "eval_steps_per_second": 1.684,
      "step": 5
    },
    {
      "epoch": 2.8,
      "eval_accuracy": 0.06388888888888888,
      "eval_loss": 3.1746575832366943,
      "eval_runtime": 3.358,
      "eval_samples_per_second": 214.412,
      "eval_steps_per_second": 1.787,
      "step": 7
    },
    {
      "epoch": 4.0,
      "grad_norm": 0.6863054633140564,
      "learning_rate": 2.5e-05,
      "loss": 3.1888,
      "step": 10
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.10972222222222222,
      "eval_loss": 3.144199848175049,
      "eval_runtime": 3.3224,
      "eval_samples_per_second": 216.708,
      "eval_steps_per_second": 1.806,
      "step": 10
    },
    {
      "epoch": 4.8,
      "eval_accuracy": 0.14583333333333334,
      "eval_loss": 3.1183247566223145,
      "eval_runtime": 3.3659,
      "eval_samples_per_second": 213.913,
      "eval_steps_per_second": 1.783,
      "step": 12
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.21944444444444444,
      "eval_loss": 3.071033477783203,
      "eval_runtime": 3.1013,
      "eval_samples_per_second": 232.158,
      "eval_steps_per_second": 1.935,
      "step": 15
    },
    {
      "epoch": 6.8,
      "eval_accuracy": 0.30416666666666664,
      "eval_loss": 3.0330796241760254,
      "eval_runtime": 3.0337,
      "eval_samples_per_second": 237.333,
      "eval_steps_per_second": 1.978,
      "step": 17
    },
    {
      "epoch": 8.0,
      "grad_norm": 0.7356524467468262,
      "learning_rate": 5e-05,
      "loss": 3.0673,
      "step": 20
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.4388888888888889,
      "eval_loss": 2.96268367767334,
      "eval_runtime": 3.2316,
      "eval_samples_per_second": 222.8,
      "eval_steps_per_second": 1.857,
      "step": 20
    },
    {
      "epoch": 8.8,
      "eval_accuracy": 0.49444444444444446,
      "eval_loss": 2.9109385013580322,
      "eval_runtime": 3.1222,
      "eval_samples_per_second": 230.605,
      "eval_steps_per_second": 1.922,
      "step": 22
    },
    {
      "epoch": 10.0,
      "eval_accuracy": 0.5763888888888888,
      "eval_loss": 2.8359875679016113,
      "eval_runtime": 3.0808,
      "eval_samples_per_second": 233.708,
      "eval_steps_per_second": 1.948,
      "step": 25
    },
    {
      "epoch": 10.8,
      "eval_accuracy": 0.6055555555555555,
      "eval_loss": 2.780921459197998,
      "eval_runtime": 3.0798,
      "eval_samples_per_second": 233.781,
      "eval_steps_per_second": 1.948,
      "step": 27
    },
    {
      "epoch": 12.0,
      "grad_norm": 0.76748126745224,
      "learning_rate": 4.722222222222222e-05,
      "loss": 2.8151,
      "step": 30
    },
    {
      "epoch": 12.0,
      "eval_accuracy": 0.6541666666666667,
      "eval_loss": 2.695770025253296,
      "eval_runtime": 3.3541,
      "eval_samples_per_second": 214.662,
      "eval_steps_per_second": 1.789,
      "step": 30
    },
    {
      "epoch": 12.8,
      "eval_accuracy": 0.6763888888888889,
      "eval_loss": 2.640125036239624,
      "eval_runtime": 3.1085,
      "eval_samples_per_second": 231.622,
      "eval_steps_per_second": 1.93,
      "step": 32
    },
    {
      "epoch": 14.0,
      "eval_accuracy": 0.6944444444444444,
      "eval_loss": 2.558335304260254,
      "eval_runtime": 3.0528,
      "eval_samples_per_second": 235.852,
      "eval_steps_per_second": 1.965,
      "step": 35
    },
    {
      "epoch": 14.8,
      "eval_accuracy": 0.7083333333333334,
      "eval_loss": 2.503415822982788,
      "eval_runtime": 3.1925,
      "eval_samples_per_second": 225.525,
      "eval_steps_per_second": 1.879,
      "step": 37
    },
    {
      "epoch": 16.0,
      "grad_norm": 0.8363860249519348,
      "learning_rate": 4.4444444444444447e-05,
      "loss": 2.5143,
      "step": 40
    },
    {
      "epoch": 16.0,
      "eval_accuracy": 0.7347222222222223,
      "eval_loss": 2.4201643466949463,
      "eval_runtime": 3.0856,
      "eval_samples_per_second": 233.339,
      "eval_steps_per_second": 1.944,
      "step": 40
    },
    {
      "epoch": 16.8,
      "eval_accuracy": 0.7375,
      "eval_loss": 2.3662188053131104,
      "eval_runtime": 3.0872,
      "eval_samples_per_second": 233.218,
      "eval_steps_per_second": 1.943,
      "step": 42
    },
    {
      "epoch": 18.0,
      "eval_accuracy": 0.7444444444444445,
      "eval_loss": 2.2883973121643066,
      "eval_runtime": 3.1262,
      "eval_samples_per_second": 230.314,
      "eval_steps_per_second": 1.919,
      "step": 45
    },
    {
      "epoch": 18.8,
      "eval_accuracy": 0.7569444444444444,
      "eval_loss": 2.237414598464966,
      "eval_runtime": 3.0834,
      "eval_samples_per_second": 233.51,
      "eval_steps_per_second": 1.946,
      "step": 47
    },
    {
      "epoch": 20.0,
      "grad_norm": 0.8517465591430664,
      "learning_rate": 4.166666666666667e-05,
      "loss": 2.2236,
      "step": 50
    },
    {
      "epoch": 20.0,
      "eval_accuracy": 0.7777777777777778,
      "eval_loss": 2.1632001399993896,
      "eval_runtime": 3.0769,
      "eval_samples_per_second": 234.001,
      "eval_steps_per_second": 1.95,
      "step": 50
    },
    {
      "epoch": 20.8,
      "eval_accuracy": 0.7833333333333333,
      "eval_loss": 2.1174519062042236,
      "eval_runtime": 3.0592,
      "eval_samples_per_second": 235.356,
      "eval_steps_per_second": 1.961,
      "step": 52
    },
    {
      "epoch": 22.0,
      "eval_accuracy": 0.7930555555555555,
      "eval_loss": 2.052760124206543,
      "eval_runtime": 3.21,
      "eval_samples_per_second": 224.301,
      "eval_steps_per_second": 1.869,
      "step": 55
    },
    {
      "epoch": 22.8,
      "eval_accuracy": 0.7958333333333333,
      "eval_loss": 2.009880304336548,
      "eval_runtime": 3.0534,
      "eval_samples_per_second": 235.801,
      "eval_steps_per_second": 1.965,
      "step": 57
    },
    {
      "epoch": 24.0,
      "grad_norm": 0.8587987422943115,
      "learning_rate": 3.888888888888889e-05,
      "loss": 1.9677,
      "step": 60
    },
    {
      "epoch": 24.0,
      "eval_accuracy": 0.8013888888888889,
      "eval_loss": 1.9488461017608643,
      "eval_runtime": 3.1017,
      "eval_samples_per_second": 232.129,
      "eval_steps_per_second": 1.934,
      "step": 60
    },
    {
      "epoch": 24.8,
      "eval_accuracy": 0.8097222222222222,
      "eval_loss": 1.9112929105758667,
      "eval_runtime": 3.0366,
      "eval_samples_per_second": 237.104,
      "eval_steps_per_second": 1.976,
      "step": 62
    },
    {
      "epoch": 26.0,
      "eval_accuracy": 0.8138888888888889,
      "eval_loss": 1.8581663370132446,
      "eval_runtime": 3.0592,
      "eval_samples_per_second": 235.359,
      "eval_steps_per_second": 1.961,
      "step": 65
    },
    {
      "epoch": 26.8,
      "eval_accuracy": 0.8138888888888889,
      "eval_loss": 1.8241873979568481,
      "eval_runtime": 3.2249,
      "eval_samples_per_second": 223.26,
      "eval_steps_per_second": 1.861,
      "step": 67
    },
    {
      "epoch": 28.0,
      "grad_norm": 0.8794375061988831,
      "learning_rate": 3.611111111111111e-05,
      "loss": 1.7467,
      "step": 70
    },
    {
      "epoch": 28.0,
      "eval_accuracy": 0.8111111111111111,
      "eval_loss": 1.7740373611450195,
      "eval_runtime": 3.1535,
      "eval_samples_per_second": 228.319,
      "eval_steps_per_second": 1.903,
      "step": 70
    },
    {
      "epoch": 28.8,
      "eval_accuracy": 0.8055555555555556,
      "eval_loss": 1.7457906007766724,
      "eval_runtime": 3.0886,
      "eval_samples_per_second": 233.114,
      "eval_steps_per_second": 1.943,
      "step": 72
    },
    {
      "epoch": 30.0,
      "eval_accuracy": 0.8180555555555555,
      "eval_loss": 1.7013169527053833,
      "eval_runtime": 3.0478,
      "eval_samples_per_second": 236.234,
      "eval_steps_per_second": 1.969,
      "step": 75
    },
    {
      "epoch": 30.8,
      "eval_accuracy": 0.8194444444444444,
      "eval_loss": 1.6714116334915161,
      "eval_runtime": 3.1668,
      "eval_samples_per_second": 227.361,
      "eval_steps_per_second": 1.895,
      "step": 77
    },
    {
      "epoch": 32.0,
      "grad_norm": 0.9110932946205139,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 1.5765,
      "step": 80
    },
    {
      "epoch": 32.0,
      "eval_accuracy": 0.8263888888888888,
      "eval_loss": 1.631589412689209,
      "eval_runtime": 3.3174,
      "eval_samples_per_second": 217.038,
      "eval_steps_per_second": 1.809,
      "step": 80
    },
    {
      "epoch": 32.8,
      "eval_accuracy": 0.8236111111111111,
      "eval_loss": 1.6082607507705688,
      "eval_runtime": 3.107,
      "eval_samples_per_second": 231.734,
      "eval_steps_per_second": 1.931,
      "step": 82
    },
    {
      "epoch": 34.0,
      "eval_accuracy": 0.8291666666666667,
      "eval_loss": 1.5738186836242676,
      "eval_runtime": 3.1316,
      "eval_samples_per_second": 229.912,
      "eval_steps_per_second": 1.916,
      "step": 85
    },
    {
      "epoch": 34.8,
      "eval_accuracy": 0.8347222222222223,
      "eval_loss": 1.553105354309082,
      "eval_runtime": 3.2321,
      "eval_samples_per_second": 222.767,
      "eval_steps_per_second": 1.856,
      "step": 87
    },
    {
      "epoch": 36.0,
      "grad_norm": 0.9191176891326904,
      "learning_rate": 3.055555555555556e-05,
      "loss": 1.4431,
      "step": 90
    },
    {
      "epoch": 36.0,
      "eval_accuracy": 0.8430555555555556,
      "eval_loss": 1.5228244066238403,
      "eval_runtime": 3.0649,
      "eval_samples_per_second": 234.921,
      "eval_steps_per_second": 1.958,
      "step": 90
    },
    {
      "epoch": 36.8,
      "eval_accuracy": 0.8444444444444444,
      "eval_loss": 1.5046004056930542,
      "eval_runtime": 3.2047,
      "eval_samples_per_second": 224.669,
      "eval_steps_per_second": 1.872,
      "step": 92
    },
    {
      "epoch": 38.0,
      "eval_accuracy": 0.8472222222222222,
      "eval_loss": 1.4780092239379883,
      "eval_runtime": 3.0681,
      "eval_samples_per_second": 234.673,
      "eval_steps_per_second": 1.956,
      "step": 95
    },
    {
      "epoch": 38.8,
      "eval_accuracy": 0.8458333333333333,
      "eval_loss": 1.4607552289962769,
      "eval_runtime": 3.3287,
      "eval_samples_per_second": 216.303,
      "eval_steps_per_second": 1.803,
      "step": 97
    },
    {
      "epoch": 40.0,
      "grad_norm": 0.896637499332428,
      "learning_rate": 2.777777777777778e-05,
      "loss": 1.3049,
      "step": 100
    },
    {
      "epoch": 40.0,
      "eval_accuracy": 0.8458333333333333,
      "eval_loss": 1.435728907585144,
      "eval_runtime": 3.1142,
      "eval_samples_per_second": 231.201,
      "eval_steps_per_second": 1.927,
      "step": 100
    },
    {
      "epoch": 40.8,
      "eval_accuracy": 0.85,
      "eval_loss": 1.4187902212142944,
      "eval_runtime": 3.144,
      "eval_samples_per_second": 229.01,
      "eval_steps_per_second": 1.908,
      "step": 102
    },
    {
      "epoch": 42.0,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.3949499130249023,
      "eval_runtime": 3.0808,
      "eval_samples_per_second": 233.709,
      "eval_steps_per_second": 1.948,
      "step": 105
    },
    {
      "epoch": 42.8,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.3807573318481445,
      "eval_runtime": 3.1783,
      "eval_samples_per_second": 226.537,
      "eval_steps_per_second": 1.888,
      "step": 107
    },
    {
      "epoch": 44.0,
      "grad_norm": 0.8933643102645874,
      "learning_rate": 2.5e-05,
      "loss": 1.2312,
      "step": 110
    },
    {
      "epoch": 44.0,
      "eval_accuracy": 0.8458333333333333,
      "eval_loss": 1.3636168241500854,
      "eval_runtime": 3.356,
      "eval_samples_per_second": 214.543,
      "eval_steps_per_second": 1.788,
      "step": 110
    },
    {
      "epoch": 44.8,
      "eval_accuracy": 0.8486111111111111,
      "eval_loss": 1.3513400554656982,
      "eval_runtime": 3.2597,
      "eval_samples_per_second": 220.878,
      "eval_steps_per_second": 1.841,
      "step": 112
    },
    {
      "epoch": 46.0,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.3329037427902222,
      "eval_runtime": 3.0929,
      "eval_samples_per_second": 232.788,
      "eval_steps_per_second": 1.94,
      "step": 115
    },
    {
      "epoch": 46.8,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.319313883781433,
      "eval_runtime": 3.1007,
      "eval_samples_per_second": 232.203,
      "eval_steps_per_second": 1.935,
      "step": 117
    },
    {
      "epoch": 48.0,
      "grad_norm": 0.9017526507377625,
      "learning_rate": 2.2222222222222223e-05,
      "loss": 1.1368,
      "step": 120
    },
    {
      "epoch": 48.0,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.3025320768356323,
      "eval_runtime": 3.1234,
      "eval_samples_per_second": 230.52,
      "eval_steps_per_second": 1.921,
      "step": 120
    },
    {
      "epoch": 48.8,
      "eval_accuracy": 0.8541666666666666,
      "eval_loss": 1.2945308685302734,
      "eval_runtime": 3.0692,
      "eval_samples_per_second": 234.59,
      "eval_steps_per_second": 1.955,
      "step": 122
    },
    {
      "epoch": 50.0,
      "eval_accuracy": 0.8527777777777777,
      "eval_loss": 1.2819503545761108,
      "eval_runtime": 3.1162,
      "eval_samples_per_second": 231.052,
      "eval_steps_per_second": 1.925,
      "step": 125
    },
    {
      "epoch": 50.8,
      "eval_accuracy": 0.8569444444444444,
      "eval_loss": 1.2704639434814453,
      "eval_runtime": 3.2771,
      "eval_samples_per_second": 219.706,
      "eval_steps_per_second": 1.831,
      "step": 127
    },
    {
      "epoch": 52.0,
      "grad_norm": 0.8930371999740601,
      "learning_rate": 1.9444444444444445e-05,
      "loss": 1.0821,
      "step": 130
    },
    {
      "epoch": 52.0,
      "eval_accuracy": 0.8583333333333333,
      "eval_loss": 1.2615665197372437,
      "eval_runtime": 3.1151,
      "eval_samples_per_second": 231.134,
      "eval_steps_per_second": 1.926,
      "step": 130
    },
    {
      "epoch": 52.8,
      "eval_accuracy": 0.8555555555555555,
      "eval_loss": 1.2545220851898193,
      "eval_runtime": 3.0635,
      "eval_samples_per_second": 235.024,
      "eval_steps_per_second": 1.959,
      "step": 132
    },
    {
      "epoch": 54.0,
      "eval_accuracy": 0.8541666666666666,
      "eval_loss": 1.2422840595245361,
      "eval_runtime": 3.0988,
      "eval_samples_per_second": 232.348,
      "eval_steps_per_second": 1.936,
      "step": 135
    },
    {
      "epoch": 54.8,
      "eval_accuracy": 0.8597222222222223,
      "eval_loss": 1.233168125152588,
      "eval_runtime": 3.1982,
      "eval_samples_per_second": 225.127,
      "eval_steps_per_second": 1.876,
      "step": 137
    },
    {
      "epoch": 56.0,
      "grad_norm": 0.8568278551101685,
      "learning_rate": 1.6666666666666667e-05,
      "loss": 1.0232,
      "step": 140
    },
    {
      "epoch": 56.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.2210274934768677,
      "eval_runtime": 3.355,
      "eval_samples_per_second": 214.604,
      "eval_steps_per_second": 1.788,
      "step": 140
    },
    {
      "epoch": 56.8,
      "eval_accuracy": 0.8625,
      "eval_loss": 1.2160966396331787,
      "eval_runtime": 3.127,
      "eval_samples_per_second": 230.252,
      "eval_steps_per_second": 1.919,
      "step": 142
    },
    {
      "epoch": 58.0,
      "eval_accuracy": 0.8569444444444444,
      "eval_loss": 1.209418535232544,
      "eval_runtime": 3.1685,
      "eval_samples_per_second": 227.239,
      "eval_steps_per_second": 1.894,
      "step": 145
    },
    {
      "epoch": 58.8,
      "eval_accuracy": 0.8541666666666666,
      "eval_loss": 1.205717921257019,
      "eval_runtime": 3.1918,
      "eval_samples_per_second": 225.576,
      "eval_steps_per_second": 1.88,
      "step": 147
    },
    {
      "epoch": 60.0,
      "grad_norm": 0.8556333780288696,
      "learning_rate": 1.388888888888889e-05,
      "loss": 0.9814,
      "step": 150
    },
    {
      "epoch": 60.0,
      "eval_accuracy": 0.85,
      "eval_loss": 1.1972941160202026,
      "eval_runtime": 3.1515,
      "eval_samples_per_second": 228.465,
      "eval_steps_per_second": 1.904,
      "step": 150
    },
    {
      "epoch": 60.8,
      "eval_accuracy": 0.8486111111111111,
      "eval_loss": 1.1918764114379883,
      "eval_runtime": 3.349,
      "eval_samples_per_second": 214.987,
      "eval_steps_per_second": 1.792,
      "step": 152
    },
    {
      "epoch": 62.0,
      "eval_accuracy": 0.8625,
      "eval_loss": 1.1825212240219116,
      "eval_runtime": 3.1401,
      "eval_samples_per_second": 229.295,
      "eval_steps_per_second": 1.911,
      "step": 155
    },
    {
      "epoch": 62.8,
      "eval_accuracy": 0.8597222222222223,
      "eval_loss": 1.179900050163269,
      "eval_runtime": 3.239,
      "eval_samples_per_second": 222.293,
      "eval_steps_per_second": 1.852,
      "step": 157
    },
    {
      "epoch": 64.0,
      "grad_norm": 0.827712893486023,
      "learning_rate": 1.1111111111111112e-05,
      "loss": 0.9415,
      "step": 160
    },
    {
      "epoch": 64.0,
      "eval_accuracy": 0.8597222222222223,
      "eval_loss": 1.1716293096542358,
      "eval_runtime": 3.1517,
      "eval_samples_per_second": 228.447,
      "eval_steps_per_second": 1.904,
      "step": 160
    },
    {
      "epoch": 64.8,
      "eval_accuracy": 0.8625,
      "eval_loss": 1.166538119316101,
      "eval_runtime": 3.22,
      "eval_samples_per_second": 223.601,
      "eval_steps_per_second": 1.863,
      "step": 162
    },
    {
      "epoch": 66.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1611206531524658,
      "eval_runtime": 3.323,
      "eval_samples_per_second": 216.67,
      "eval_steps_per_second": 1.806,
      "step": 165
    },
    {
      "epoch": 66.8,
      "eval_accuracy": 0.8625,
      "eval_loss": 1.1600357294082642,
      "eval_runtime": 3.2253,
      "eval_samples_per_second": 223.233,
      "eval_steps_per_second": 1.86,
      "step": 167
    },
    {
      "epoch": 68.0,
      "grad_norm": 0.9616327285766602,
      "learning_rate": 8.333333333333334e-06,
      "loss": 0.9135,
      "step": 170
    },
    {
      "epoch": 68.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1577341556549072,
      "eval_runtime": 3.1095,
      "eval_samples_per_second": 231.548,
      "eval_steps_per_second": 1.93,
      "step": 170
    },
    {
      "epoch": 68.8,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1546900272369385,
      "eval_runtime": 3.2289,
      "eval_samples_per_second": 222.988,
      "eval_steps_per_second": 1.858,
      "step": 172
    },
    {
      "epoch": 70.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1493217945098877,
      "eval_runtime": 3.103,
      "eval_samples_per_second": 232.033,
      "eval_steps_per_second": 1.934,
      "step": 175
    },
    {
      "epoch": 70.8,
      "eval_accuracy": 0.8611111111111112,
      "eval_loss": 1.1463639736175537,
      "eval_runtime": 3.3191,
      "eval_samples_per_second": 216.926,
      "eval_steps_per_second": 1.808,
      "step": 177
    },
    {
      "epoch": 72.0,
      "grad_norm": 0.8919360637664795,
      "learning_rate": 5.555555555555556e-06,
      "loss": 0.8946,
      "step": 180
    },
    {
      "epoch": 72.0,
      "eval_accuracy": 0.8555555555555555,
      "eval_loss": 1.1423206329345703,
      "eval_runtime": 3.0106,
      "eval_samples_per_second": 239.152,
      "eval_steps_per_second": 1.993,
      "step": 180
    },
    {
      "epoch": 72.8,
      "eval_accuracy": 0.8611111111111112,
      "eval_loss": 1.1402053833007812,
      "eval_runtime": 3.4453,
      "eval_samples_per_second": 208.981,
      "eval_steps_per_second": 1.742,
      "step": 182
    },
    {
      "epoch": 74.0,
      "eval_accuracy": 0.8583333333333333,
      "eval_loss": 1.1375410556793213,
      "eval_runtime": 3.1143,
      "eval_samples_per_second": 231.188,
      "eval_steps_per_second": 1.927,
      "step": 185
    },
    {
      "epoch": 74.8,
      "eval_accuracy": 0.8597222222222223,
      "eval_loss": 1.1360384225845337,
      "eval_runtime": 3.2332,
      "eval_samples_per_second": 222.688,
      "eval_steps_per_second": 1.856,
      "step": 187
    },
    {
      "epoch": 76.0,
      "grad_norm": 0.9335024356842041,
      "learning_rate": 2.777777777777778e-06,
      "loss": 0.8866,
      "step": 190
    },
    {
      "epoch": 76.0,
      "eval_accuracy": 0.8625,
      "eval_loss": 1.134353756904602,
      "eval_runtime": 3.1331,
      "eval_samples_per_second": 229.808,
      "eval_steps_per_second": 1.915,
      "step": 190
    },
    {
      "epoch": 76.8,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1333543062210083,
      "eval_runtime": 3.2199,
      "eval_samples_per_second": 223.611,
      "eval_steps_per_second": 1.863,
      "step": 192
    },
    {
      "epoch": 78.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1323890686035156,
      "eval_runtime": 3.0815,
      "eval_samples_per_second": 233.653,
      "eval_steps_per_second": 1.947,
      "step": 195
    },
    {
      "epoch": 78.8,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1320444345474243,
      "eval_runtime": 3.2638,
      "eval_samples_per_second": 220.604,
      "eval_steps_per_second": 1.838,
      "step": 197
    },
    {
      "epoch": 80.0,
      "grad_norm": 0.8473469614982605,
      "learning_rate": 0.0,
      "loss": 0.8798,
      "step": 200
    },
    {
      "epoch": 80.0,
      "eval_accuracy": 0.8638888888888889,
      "eval_loss": 1.1318904161453247,
      "eval_runtime": 3.2096,
      "eval_samples_per_second": 224.325,
      "eval_steps_per_second": 1.869,
      "step": 200
    },
    {
      "epoch": 80.0,
      "step": 200,
      "total_flos": 7.440697863438336e+18,
      "train_loss": 1.5909362745285034,
      "train_runtime": 1036.0381,
      "train_samples_per_second": 115.826,
      "train_steps_per_second": 0.193
    }
  ],
  "logging_steps": 10,
  "max_steps": 200,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 100,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 7.440697863438336e+18,
  "train_batch_size": 128,
  "trial_name": null,
  "trial_params": null
}