{
  "best_metric": 0.9584578601315002,
  "best_model_checkpoint": "/hdd1/mujeen/retrieval_prf/output/roberta-base_mnli_bc/checkpoint-49089",
  "epoch": 3.0,
  "global_step": 49089,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03,
      "learning_rate": 1.9797918067184097e-05,
      "loss": 0.3858,
      "step": 500
    },
    {
      "epoch": 0.06,
      "learning_rate": 1.959420644136161e-05,
      "loss": 0.284,
      "step": 1000
    },
    {
      "epoch": 0.09,
      "learning_rate": 1.9390494815539124e-05,
      "loss": 0.2536,
      "step": 1500
    },
    {
      "epoch": 0.12,
      "learning_rate": 1.918719061296828e-05,
      "loss": 0.2583,
      "step": 2000
    },
    {
      "epoch": 0.15,
      "learning_rate": 1.89834789871458e-05,
      "loss": 0.2522,
      "step": 2500
    },
    {
      "epoch": 0.18,
      "learning_rate": 1.8779767361323312e-05,
      "loss": 0.2526,
      "step": 3000
    },
    {
      "epoch": 0.21,
      "learning_rate": 1.8576055735500826e-05,
      "loss": 0.2457,
      "step": 3500
    },
    {
      "epoch": 0.24,
      "learning_rate": 1.8372344109678343e-05,
      "loss": 0.2335,
      "step": 4000
    },
    {
      "epoch": 0.28,
      "learning_rate": 1.8168632483855853e-05,
      "loss": 0.224,
      "step": 4500
    },
    {
      "epoch": 0.31,
      "learning_rate": 1.796492085803337e-05,
      "loss": 0.2234,
      "step": 5000
    },
    {
      "epoch": 0.34,
      "learning_rate": 1.7761209232210884e-05,
      "loss": 0.2335,
      "step": 5500
    },
    {
      "epoch": 0.37,
      "learning_rate": 1.7557905029640045e-05,
      "loss": 0.2341,
      "step": 6000
    },
    {
      "epoch": 0.4,
      "learning_rate": 1.7354193403817555e-05,
      "loss": 0.2235,
      "step": 6500
    },
    {
      "epoch": 0.43,
      "learning_rate": 1.7150481777995072e-05,
      "loss": 0.231,
      "step": 7000
    },
    {
      "epoch": 0.46,
      "learning_rate": 1.6946770152172586e-05,
      "loss": 0.2146,
      "step": 7500
    },
    {
      "epoch": 0.49,
      "learning_rate": 1.67430585263501e-05,
      "loss": 0.219,
      "step": 8000
    },
    {
      "epoch": 0.52,
      "learning_rate": 1.653975432377926e-05,
      "loss": 0.2263,
      "step": 8500
    },
    {
      "epoch": 0.55,
      "learning_rate": 1.6336042697956774e-05,
      "loss": 0.2197,
      "step": 9000
    },
    {
      "epoch": 0.58,
      "learning_rate": 1.6132331072134288e-05,
      "loss": 0.2072,
      "step": 9500
    },
    {
      "epoch": 0.61,
      "learning_rate": 1.59286194463118e-05,
      "loss": 0.2196,
      "step": 10000
    },
    {
      "epoch": 0.64,
      "learning_rate": 1.5724907820489318e-05,
      "loss": 0.2039,
      "step": 10500
    },
    {
      "epoch": 0.67,
      "learning_rate": 1.5521603617918476e-05,
      "loss": 0.201,
      "step": 11000
    },
    {
      "epoch": 0.7,
      "learning_rate": 1.531789199209599e-05,
      "loss": 0.2081,
      "step": 11500
    },
    {
      "epoch": 0.73,
      "learning_rate": 1.5114180366273505e-05,
      "loss": 0.2072,
      "step": 12000
    },
    {
      "epoch": 0.76,
      "learning_rate": 1.4910468740451018e-05,
      "loss": 0.1949,
      "step": 12500
    },
    {
      "epoch": 0.79,
      "learning_rate": 1.4706757114628534e-05,
      "loss": 0.2119,
      "step": 13000
    },
    {
      "epoch": 0.83,
      "learning_rate": 1.4503452912057691e-05,
      "loss": 0.209,
      "step": 13500
    },
    {
      "epoch": 0.86,
      "learning_rate": 1.4299741286235206e-05,
      "loss": 0.1992,
      "step": 14000
    },
    {
      "epoch": 0.89,
      "learning_rate": 1.409602966041272e-05,
      "loss": 0.1953,
      "step": 14500
    },
    {
      "epoch": 0.92,
      "learning_rate": 1.3892318034590235e-05,
      "loss": 0.2025,
      "step": 15000
    },
    {
      "epoch": 0.95,
      "learning_rate": 1.368860640876775e-05,
      "loss": 0.198,
      "step": 15500
    },
    {
      "epoch": 0.98,
      "learning_rate": 1.3485302206196908e-05,
      "loss": 0.2015,
      "step": 16000
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.9469515839808726,
      "eval_loss": 0.18203075230121613,
      "eval_runtime": 18.9219,
      "eval_samples_per_second": 353.664,
      "eval_steps_per_second": 44.234,
      "step": 16363
    },
    {
      "epoch": 1.01,
      "learning_rate": 1.3281590580374423e-05,
      "loss": 0.1932,
      "step": 16500
    },
    {
      "epoch": 1.04,
      "learning_rate": 1.3077878954551937e-05,
      "loss": 0.1541,
      "step": 17000
    },
    {
      "epoch": 1.07,
      "learning_rate": 1.2874574751981098e-05,
      "loss": 0.1556,
      "step": 17500
    },
    {
      "epoch": 1.1,
      "learning_rate": 1.267086312615861e-05,
      "loss": 0.1607,
      "step": 18000
    },
    {
      "epoch": 1.13,
      "learning_rate": 1.2467151500336125e-05,
      "loss": 0.1618,
      "step": 18500
    },
    {
      "epoch": 1.16,
      "learning_rate": 1.2263439874513639e-05,
      "loss": 0.1593,
      "step": 19000
    },
    {
      "epoch": 1.19,
      "learning_rate": 1.2059728248691154e-05,
      "loss": 0.1511,
      "step": 19500
    },
    {
      "epoch": 1.22,
      "learning_rate": 1.185601662286867e-05,
      "loss": 0.165,
      "step": 20000
    },
    {
      "epoch": 1.25,
      "learning_rate": 1.1652304997046182e-05,
      "loss": 0.157,
      "step": 20500
    },
    {
      "epoch": 1.28,
      "learning_rate": 1.1448593371223697e-05,
      "loss": 0.1385,
      "step": 21000
    },
    {
      "epoch": 1.31,
      "learning_rate": 1.1245289168652856e-05,
      "loss": 0.1646,
      "step": 21500
    },
    {
      "epoch": 1.34,
      "learning_rate": 1.1041577542830371e-05,
      "loss": 0.163,
      "step": 22000
    },
    {
      "epoch": 1.38,
      "learning_rate": 1.0837865917007883e-05,
      "loss": 0.156,
      "step": 22500
    },
    {
      "epoch": 1.41,
      "learning_rate": 1.0634154291185399e-05,
      "loss": 0.1556,
      "step": 23000
    },
    {
      "epoch": 1.44,
      "learning_rate": 1.043085008861456e-05,
      "loss": 0.1626,
      "step": 23500
    },
    {
      "epoch": 1.47,
      "learning_rate": 1.0227545886043717e-05,
      "loss": 0.1605,
      "step": 24000
    },
    {
      "epoch": 1.5,
      "learning_rate": 1.0023834260221232e-05,
      "loss": 0.1651,
      "step": 24500
    },
    {
      "epoch": 1.53,
      "learning_rate": 9.820122634398746e-06,
      "loss": 0.1438,
      "step": 25000
    },
    {
      "epoch": 1.56,
      "learning_rate": 9.616411008576261e-06,
      "loss": 0.1501,
      "step": 25500
    },
    {
      "epoch": 1.59,
      "learning_rate": 9.412699382753775e-06,
      "loss": 0.1562,
      "step": 26000
    },
    {
      "epoch": 1.62,
      "learning_rate": 9.209395180182934e-06,
      "loss": 0.158,
      "step": 26500
    },
    {
      "epoch": 1.65,
      "learning_rate": 9.005683554360448e-06,
      "loss": 0.149,
      "step": 27000
    },
    {
      "epoch": 1.68,
      "learning_rate": 8.801971928537963e-06,
      "loss": 0.1483,
      "step": 27500
    },
    {
      "epoch": 1.71,
      "learning_rate": 8.598260302715477e-06,
      "loss": 0.163,
      "step": 28000
    },
    {
      "epoch": 1.74,
      "learning_rate": 8.39454867689299e-06,
      "loss": 0.1507,
      "step": 28500
    },
    {
      "epoch": 1.77,
      "learning_rate": 8.19124447432215e-06,
      "loss": 0.148,
      "step": 29000
    },
    {
      "epoch": 1.8,
      "learning_rate": 7.987532848499665e-06,
      "loss": 0.1571,
      "step": 29500
    },
    {
      "epoch": 1.83,
      "learning_rate": 7.78382122267718e-06,
      "loss": 0.1467,
      "step": 30000
    },
    {
      "epoch": 1.86,
      "learning_rate": 7.580109596854694e-06,
      "loss": 0.1417,
      "step": 30500
    },
    {
      "epoch": 1.89,
      "learning_rate": 7.376397971032207e-06,
      "loss": 0.1526,
      "step": 31000
    },
    {
      "epoch": 1.93,
      "learning_rate": 7.172686345209722e-06,
      "loss": 0.1327,
      "step": 31500
    },
    {
      "epoch": 1.96,
      "learning_rate": 6.969382142638881e-06,
      "loss": 0.151,
      "step": 32000
    },
    {
      "epoch": 1.99,
      "learning_rate": 6.7656705168163946e-06,
      "loss": 0.1463,
      "step": 32500
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.9559175134488942,
      "eval_loss": 0.19088995456695557,
      "eval_runtime": 18.8218,
      "eval_samples_per_second": 355.545,
      "eval_steps_per_second": 44.47,
      "step": 32726
    },
    {
      "epoch": 2.02,
      "learning_rate": 6.561958890993909e-06,
      "loss": 0.1198,
      "step": 33000
    },
    {
      "epoch": 2.05,
      "learning_rate": 6.358247265171424e-06,
      "loss": 0.0851,
      "step": 33500
    },
    {
      "epoch": 2.08,
      "learning_rate": 6.155350485852228e-06,
      "loss": 0.1081,
      "step": 34000
    },
    {
      "epoch": 2.11,
      "learning_rate": 5.9516388600297426e-06,
      "loss": 0.1071,
      "step": 34500
    },
    {
      "epoch": 2.14,
      "learning_rate": 5.747927234207257e-06,
      "loss": 0.1046,
      "step": 35000
    },
    {
      "epoch": 2.17,
      "learning_rate": 5.544215608384771e-06,
      "loss": 0.0965,
      "step": 35500
    },
    {
      "epoch": 2.2,
      "learning_rate": 5.340503982562285e-06,
      "loss": 0.0906,
      "step": 36000
    },
    {
      "epoch": 2.23,
      "learning_rate": 5.137199779991444e-06,
      "loss": 0.0927,
      "step": 36500
    },
    {
      "epoch": 2.26,
      "learning_rate": 4.933488154168959e-06,
      "loss": 0.0947,
      "step": 37000
    },
    {
      "epoch": 2.29,
      "learning_rate": 4.729776528346473e-06,
      "loss": 0.1007,
      "step": 37500
    },
    {
      "epoch": 2.32,
      "learning_rate": 4.526064902523987e-06,
      "loss": 0.0981,
      "step": 38000
    },
    {
      "epoch": 2.35,
      "learning_rate": 4.322353276701502e-06,
      "loss": 0.105,
      "step": 38500
    },
    {
      "epoch": 2.38,
      "learning_rate": 4.118641650879016e-06,
      "loss": 0.1025,
      "step": 39000
    },
    {
      "epoch": 2.41,
      "learning_rate": 3.915337448308175e-06,
      "loss": 0.089,
      "step": 39500
    },
    {
      "epoch": 2.44,
      "learning_rate": 3.7116258224856896e-06,
      "loss": 0.1049,
      "step": 40000
    },
    {
      "epoch": 2.48,
      "learning_rate": 3.507914196663204e-06,
      "loss": 0.0982,
      "step": 40500
    },
    {
      "epoch": 2.51,
      "learning_rate": 3.304202570840718e-06,
      "loss": 0.1007,
      "step": 41000
    },
    {
      "epoch": 2.54,
      "learning_rate": 3.1004909450182323e-06,
      "loss": 0.098,
      "step": 41500
    },
    {
      "epoch": 2.57,
      "learning_rate": 2.8967793191957468e-06,
      "loss": 0.0889,
      "step": 42000
    },
    {
      "epoch": 2.6,
      "learning_rate": 2.693067693373261e-06,
      "loss": 0.0975,
      "step": 42500
    },
    {
      "epoch": 2.63,
      "learning_rate": 2.4893560675507754e-06,
      "loss": 0.1031,
      "step": 43000
    },
    {
      "epoch": 2.66,
      "learning_rate": 2.28564444172829e-06,
      "loss": 0.0882,
      "step": 43500
    },
    {
      "epoch": 2.69,
      "learning_rate": 2.082747662409094e-06,
      "loss": 0.0917,
      "step": 44000
    },
    {
      "epoch": 2.72,
      "learning_rate": 1.879036036586608e-06,
      "loss": 0.0935,
      "step": 44500
    },
    {
      "epoch": 2.75,
      "learning_rate": 1.6753244107641225e-06,
      "loss": 0.099,
      "step": 45000
    },
    {
      "epoch": 2.78,
      "learning_rate": 1.4716127849416368e-06,
      "loss": 0.0938,
      "step": 45500
    },
    {
      "epoch": 2.81,
      "learning_rate": 1.2679011591191509e-06,
      "loss": 0.0977,
      "step": 46000
    },
    {
      "epoch": 2.84,
      "learning_rate": 1.0641895332966654e-06,
      "loss": 0.0785,
      "step": 46500
    },
    {
      "epoch": 2.87,
      "learning_rate": 8.604779074741797e-07,
      "loss": 0.106,
      "step": 47000
    },
    {
      "epoch": 2.9,
      "learning_rate": 6.571737049033389e-07,
      "loss": 0.088,
      "step": 47500
    },
    {
      "epoch": 2.93,
      "learning_rate": 4.5346207908085317e-07,
      "loss": 0.087,
      "step": 48000
    },
    {
      "epoch": 2.96,
      "learning_rate": 2.4975045325836747e-07,
      "loss": 0.1096,
      "step": 48500
    },
    {
      "epoch": 2.99,
      "learning_rate": 4.6038827435881766e-08,
      "loss": 0.0768,
      "step": 49000
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.9584578601315002,
      "eval_loss": 0.21173284947872162,
      "eval_runtime": 18.7035,
      "eval_samples_per_second": 357.794,
      "eval_steps_per_second": 44.751,
      "step": 49089
    },
    {
      "epoch": 3.0,
      "step": 49089,
      "total_flos": 3.330874138659072e+16,
      "train_loss": 0.1589348805035952,
      "train_runtime": 5162.793,
      "train_samples_per_second": 152.128,
      "train_steps_per_second": 9.508
    }
  ],
  "max_steps": 49089,
  "num_train_epochs": 3,
  "total_flos": 3.330874138659072e+16,
  "trial_name": null,
  "trial_params": null
}