{
  "best_metric": 78.98580203731828,
  "best_model_checkpoint": "/home/fbravo/data/all_results/tar/albeto_xxlarge/epochs_2_bs_16_lr_5e-6/checkpoint-7800",
  "epoch": 2.0,
  "global_step": 10970,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.05,
      "eval_exact_match": 50.73793755912961,
      "eval_f1": 69.4730775689932,
      "step": 300
    },
    {
      "epoch": 0.09,
      "learning_rate": 4.774384685505926e-06,
      "loss": 2.2004,
      "step": 500
    },
    {
      "epoch": 0.11,
      "eval_exact_match": 54.370860927152314,
      "eval_f1": 73.05853038044215,
      "step": 600
    },
    {
      "epoch": 0.16,
      "eval_exact_match": 54.89120151371807,
      "eval_f1": 73.65027892880356,
      "step": 900
    },
    {
      "epoch": 0.18,
      "learning_rate": 4.546490428441203e-06,
      "loss": 1.6573,
      "step": 1000
    },
    {
      "epoch": 0.22,
      "eval_exact_match": 56.9914853358562,
      "eval_f1": 75.0513540652836,
      "step": 1200
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.319051959890611e-06,
      "loss": 1.5668,
      "step": 1500
    },
    {
      "epoch": 0.27,
      "eval_exact_match": 57.8240302743614,
      "eval_f1": 75.63669001183044,
      "step": 1500
    },
    {
      "epoch": 0.33,
      "eval_exact_match": 58.81740775780511,
      "eval_f1": 76.36054723510281,
      "step": 1800
    },
    {
      "epoch": 0.36,
      "learning_rate": 4.091157702825889e-06,
      "loss": 1.5322,
      "step": 2000
    },
    {
      "epoch": 0.38,
      "eval_exact_match": 59.14853358561968,
      "eval_f1": 76.483161808874,
      "step": 2100
    },
    {
      "epoch": 0.44,
      "eval_exact_match": 59.25260170293283,
      "eval_f1": 76.67791535670501,
      "step": 2400
    },
    {
      "epoch": 0.46,
      "learning_rate": 3.863263445761167e-06,
      "loss": 1.5015,
      "step": 2500
    },
    {
      "epoch": 0.49,
      "eval_exact_match": 59.290444654683064,
      "eval_f1": 76.94674808406094,
      "step": 2700
    },
    {
      "epoch": 0.55,
      "learning_rate": 3.6353691886964453e-06,
      "loss": 1.4561,
      "step": 3000
    },
    {
      "epoch": 0.55,
      "eval_exact_match": 60.13245033112583,
      "eval_f1": 77.30126888604488,
      "step": 3000
    },
    {
      "epoch": 0.6,
      "eval_exact_match": 59.93377483443709,
      "eval_f1": 77.31924407449924,
      "step": 3300
    },
    {
      "epoch": 0.64,
      "learning_rate": 3.407474931631723e-06,
      "loss": 1.4414,
      "step": 3500
    },
    {
      "epoch": 0.66,
      "eval_exact_match": 60.44465468306528,
      "eval_f1": 77.63204103944324,
      "step": 3600
    },
    {
      "epoch": 0.71,
      "eval_exact_match": 61.31504257332072,
      "eval_f1": 78.13561765566448,
      "step": 3900
    },
    {
      "epoch": 0.73,
      "learning_rate": 3.179580674567001e-06,
      "loss": 1.4206,
      "step": 4000
    },
    {
      "epoch": 0.77,
      "eval_exact_match": 60.52034058656575,
      "eval_f1": 77.94067744854881,
      "step": 4200
    },
    {
      "epoch": 0.82,
      "learning_rate": 2.9516864175022793e-06,
      "loss": 1.3835,
      "step": 4500
    },
    {
      "epoch": 0.82,
      "eval_exact_match": 60.98391674550615,
      "eval_f1": 78.30309425310817,
      "step": 4500
    },
    {
      "epoch": 0.88,
      "eval_exact_match": 61.21097445600757,
      "eval_f1": 78.3458557935229,
      "step": 4800
    },
    {
      "epoch": 0.91,
      "learning_rate": 2.7237921604375573e-06,
      "loss": 1.3656,
      "step": 5000
    },
    {
      "epoch": 0.93,
      "eval_exact_match": 61.30558183538316,
      "eval_f1": 78.35466990254258,
      "step": 5100
    },
    {
      "epoch": 0.98,
      "eval_exact_match": 61.485335856196784,
      "eval_f1": 78.61331349695031,
      "step": 5400
    },
    {
      "epoch": 1.0,
      "learning_rate": 2.4958979033728353e-06,
      "loss": 1.3461,
      "step": 5500
    },
    {
      "epoch": 1.04,
      "eval_exact_match": 61.258278145695364,
      "eval_f1": 78.44203108653505,
      "step": 5700
    },
    {
      "epoch": 1.09,
      "learning_rate": 2.268459434822243e-06,
      "loss": 1.0985,
      "step": 6000
    },
    {
      "epoch": 1.09,
      "eval_exact_match": 61.22989593188269,
      "eval_f1": 78.6338704184745,
      "step": 6000
    },
    {
      "epoch": 1.15,
      "eval_exact_match": 61.21097445600757,
      "eval_f1": 78.69664962146094,
      "step": 6300
    },
    {
      "epoch": 1.19,
      "learning_rate": 2.0405651777575204e-06,
      "loss": 1.0548,
      "step": 6500
    },
    {
      "epoch": 1.2,
      "eval_exact_match": 61.343424787133394,
      "eval_f1": 78.77498098426705,
      "step": 6600
    },
    {
      "epoch": 1.26,
      "eval_exact_match": 61.65562913907285,
      "eval_f1": 78.67261809065106,
      "step": 6900
    },
    {
      "epoch": 1.28,
      "learning_rate": 1.8126709206927986e-06,
      "loss": 1.0615,
      "step": 7000
    },
    {
      "epoch": 1.31,
      "eval_exact_match": 61.53263954588458,
      "eval_f1": 78.5005776920404,
      "step": 7200
    },
    {
      "epoch": 1.37,
      "learning_rate": 1.5847766636280768e-06,
      "loss": 1.0828,
      "step": 7500
    },
    {
      "epoch": 1.37,
      "eval_exact_match": 61.551561021759696,
      "eval_f1": 78.55843313017921,
      "step": 7500
    },
    {
      "epoch": 1.42,
      "eval_exact_match": 61.84484389782403,
      "eval_f1": 78.98580203731828,
      "step": 7800
    },
    {
      "epoch": 1.46,
      "learning_rate": 1.3573381950774841e-06,
      "loss": 1.0492,
      "step": 8000
    },
    {
      "epoch": 1.48,
      "eval_exact_match": 61.73131504257332,
      "eval_f1": 78.56521589229312,
      "step": 8100
    },
    {
      "epoch": 1.53,
      "eval_exact_match": 61.81646168401135,
      "eval_f1": 78.62740589986592,
      "step": 8400
    },
    {
      "epoch": 1.55,
      "learning_rate": 1.1294439380127621e-06,
      "loss": 1.0551,
      "step": 8500
    },
    {
      "epoch": 1.59,
      "eval_exact_match": 61.57994323557237,
      "eval_f1": 78.64237467540597,
      "step": 8700
    },
    {
      "epoch": 1.64,
      "learning_rate": 9.015496809480402e-07,
      "loss": 1.0353,
      "step": 9000
    },
    {
      "epoch": 1.64,
      "eval_exact_match": 61.30558183538316,
      "eval_f1": 78.50591438549111,
      "step": 9000
    },
    {
      "epoch": 1.7,
      "eval_exact_match": 61.63670766319773,
      "eval_f1": 78.6172319874888,
      "step": 9300
    },
    {
      "epoch": 1.73,
      "learning_rate": 6.736554238833181e-07,
      "loss": 1.0373,
      "step": 9500
    },
    {
      "epoch": 1.75,
      "eval_exact_match": 61.66508987701041,
      "eval_f1": 78.81109427991775,
      "step": 9600
    },
    {
      "epoch": 1.8,
      "eval_exact_match": 61.59886471144749,
      "eval_f1": 78.57587678017954,
      "step": 9900
    },
    {
      "epoch": 1.82,
      "learning_rate": 4.457611668185962e-07,
      "loss": 1.0469,
      "step": 10000
    },
    {
      "epoch": 1.86,
      "eval_exact_match": 61.81646168401135,
      "eval_f1": 78.77340428037925,
      "step": 10200
    },
    {
      "epoch": 1.91,
      "learning_rate": 2.1832269826800366e-07,
      "loss": 1.0179,
      "step": 10500
    },
    {
      "epoch": 1.91,
      "eval_exact_match": 61.84484389782403,
      "eval_f1": 78.74115858178497,
      "step": 10500
    },
    {
      "epoch": 1.97,
      "eval_exact_match": 61.74077578051088,
      "eval_f1": 78.74584305622463,
      "step": 10800
    },
    {
      "epoch": 2.0,
      "step": 10970,
      "total_flos": 6.330675040878592e+16,
      "train_loss": 1.2922884611618335,
      "train_runtime": 6069.4707,
      "train_samples_per_second": 28.914,
      "train_steps_per_second": 1.807
    }
  ],
  "max_steps": 10970,
  "num_train_epochs": 2,
  "total_flos": 6.330675040878592e+16,
  "trial_name": null,
  "trial_params": null
}