{
  "best_metric": 0.45347079634666443,
  "best_model_checkpoint": "model_output/e2e_opentable_5_way__approximate__0-shot__seed-77__roberta-base/checkpoint-500",
  "epoch": 0.8576329331046312,
  "global_step": 500,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03,
      "learning_rate": 4.9783362218370885e-05,
      "loss": 16.4408,
      "step": 20
    },
    {
      "epoch": 0.07,
      "learning_rate": 4.956672443674177e-05,
      "loss": 13.3709,
      "step": 40
    },
    {
      "epoch": 0.09,
      "eval_distillation_accuracy_counterfactual": 0.3304258594150847,
      "eval_distillation_accuracy_factual": 0.85351462288353,
      "eval_distillation_f1_counterfactual": 0.243675431005225,
      "eval_distillation_f1_factual": 0.8405624277794843,
      "eval_groundtruth_accuracy_counterfactual": 0.3258081067213956,
      "eval_groundtruth_f1_counterfactual": 0.24573702955510618,
      "eval_groundtruth_f1_factual": 0.7133962491941664,
      "eval_icace_cosine": 0.5112652778625488,
      "eval_icace_l2": 0.8116109371185303,
      "eval_icace_normdiff": 0.5299729108810425,
      "eval_loss": 10.612378120422363,
      "eval_runtime": 12.8754,
      "eval_samples_per_second": 302.748,
      "eval_steps_per_second": 2.408,
      "step": 50
    },
    {
      "epoch": 0.1,
      "learning_rate": 4.935008665511265e-05,
      "loss": 11.6048,
      "step": 60
    },
    {
      "epoch": 0.14,
      "learning_rate": 4.913344887348354e-05,
      "loss": 11.1238,
      "step": 80
    },
    {
      "epoch": 0.17,
      "learning_rate": 4.891681109185442e-05,
      "loss": 10.5122,
      "step": 100
    },
    {
      "epoch": 0.17,
      "eval_distillation_accuracy_counterfactual": 0.5010261672652643,
      "eval_distillation_accuracy_factual": 0.8737814263724987,
      "eval_distillation_f1_counterfactual": 0.3809463454212182,
      "eval_distillation_f1_factual": 0.8520946306923461,
      "eval_groundtruth_accuracy_counterfactual": 0.47998973832734737,
      "eval_groundtruth_f1_counterfactual": 0.36220047443175507,
      "eval_groundtruth_f1_factual": 0.7127756432715844,
      "eval_icace_cosine": 0.46907880902290344,
      "eval_icace_l2": 0.6739468574523926,
      "eval_icace_normdiff": 0.44848179817199707,
      "eval_loss": 8.415631294250488,
      "eval_runtime": 12.9894,
      "eval_samples_per_second": 300.091,
      "eval_steps_per_second": 2.387,
      "step": 100
    },
    {
      "epoch": 0.21,
      "learning_rate": 4.8700173310225307e-05,
      "loss": 10.4539,
      "step": 120
    },
    {
      "epoch": 0.24,
      "learning_rate": 4.848353552859619e-05,
      "loss": 10.4598,
      "step": 140
    },
    {
      "epoch": 0.26,
      "eval_distillation_accuracy_counterfactual": 0.5100051308363264,
      "eval_distillation_accuracy_factual": 0.8648024628014367,
      "eval_distillation_f1_counterfactual": 0.4378381593277977,
      "eval_distillation_f1_factual": 0.8505368829186162,
      "eval_groundtruth_accuracy_counterfactual": 0.4979476654694715,
      "eval_groundtruth_f1_counterfactual": 0.42720537829821525,
      "eval_groundtruth_f1_factual": 0.6962013496902839,
      "eval_icace_cosine": 0.4693450629711151,
      "eval_icace_l2": 0.6681450009346008,
      "eval_icace_normdiff": 0.41851383447647095,
      "eval_loss": 8.433810234069824,
      "eval_runtime": 12.8826,
      "eval_samples_per_second": 302.578,
      "eval_steps_per_second": 2.406,
      "step": 150
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.826689774696707e-05,
      "loss": 10.219,
      "step": 160
    },
    {
      "epoch": 0.31,
      "learning_rate": 4.8050259965337955e-05,
      "loss": 10.1164,
      "step": 180
    },
    {
      "epoch": 0.34,
      "learning_rate": 4.7833622183708845e-05,
      "loss": 10.1494,
      "step": 200
    },
    {
      "epoch": 0.34,
      "eval_distillation_accuracy_counterfactual": 0.48691636736788096,
      "eval_distillation_accuracy_factual": 0.8858388917393535,
      "eval_distillation_f1_counterfactual": 0.3500647816598059,
      "eval_distillation_f1_factual": 0.8801680477409537,
      "eval_groundtruth_accuracy_counterfactual": 0.46793227296049256,
      "eval_groundtruth_f1_counterfactual": 0.3413720392476469,
      "eval_groundtruth_f1_factual": 0.696679629826118,
      "eval_icace_cosine": 0.4742945730686188,
      "eval_icace_l2": 0.6967979073524475,
      "eval_icace_normdiff": 0.4476469159126282,
      "eval_loss": 8.772797584533691,
      "eval_runtime": 13.0849,
      "eval_samples_per_second": 297.901,
      "eval_steps_per_second": 2.369,
      "step": 200
    },
    {
      "epoch": 0.38,
      "learning_rate": 4.761698440207972e-05,
      "loss": 9.9977,
      "step": 220
    },
    {
      "epoch": 0.41,
      "learning_rate": 4.740034662045061e-05,
      "loss": 10.0073,
      "step": 240
    },
    {
      "epoch": 0.43,
      "eval_distillation_accuracy_counterfactual": 0.43560800410466904,
      "eval_distillation_accuracy_factual": 0.8519753719856337,
      "eval_distillation_f1_counterfactual": 0.35530824798064053,
      "eval_distillation_f1_factual": 0.8365066372023451,
      "eval_groundtruth_accuracy_counterfactual": 0.43766033863519754,
      "eval_groundtruth_f1_counterfactual": 0.364318999243899,
      "eval_groundtruth_f1_factual": 0.6915401170991118,
      "eval_icace_cosine": 0.48830506205558777,
      "eval_icace_l2": 0.7442983388900757,
      "eval_icace_normdiff": 0.4745895266532898,
      "eval_loss": 9.400979042053223,
      "eval_runtime": 12.983,
      "eval_samples_per_second": 300.24,
      "eval_steps_per_second": 2.388,
      "step": 250
    },
    {
      "epoch": 0.45,
      "learning_rate": 4.7183708838821494e-05,
      "loss": 9.7738,
      "step": 260
    },
    {
      "epoch": 0.48,
      "learning_rate": 4.6967071057192376e-05,
      "loss": 9.9329,
      "step": 280
    },
    {
      "epoch": 0.51,
      "learning_rate": 4.675043327556326e-05,
      "loss": 9.7779,
      "step": 300
    },
    {
      "epoch": 0.51,
      "eval_distillation_accuracy_counterfactual": 0.5359158542842484,
      "eval_distillation_accuracy_factual": 0.8619805028219599,
      "eval_distillation_f1_counterfactual": 0.47616723466717426,
      "eval_distillation_f1_factual": 0.852650039087773,
      "eval_groundtruth_accuracy_counterfactual": 0.5010261672652643,
      "eval_groundtruth_f1_counterfactual": 0.44417532302390744,
      "eval_groundtruth_f1_factual": 0.7073800556776944,
      "eval_icace_cosine": 0.46865805983543396,
      "eval_icace_l2": 0.6562190651893616,
      "eval_icace_normdiff": 0.4228982925415039,
      "eval_loss": 8.363639831542969,
      "eval_runtime": 13.0389,
      "eval_samples_per_second": 298.951,
      "eval_steps_per_second": 2.377,
      "step": 300
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.653379549393415e-05,
      "loss": 9.6834,
      "step": 320
    },
    {
      "epoch": 0.58,
      "learning_rate": 4.6317157712305025e-05,
      "loss": 9.547,
      "step": 340
    },
    {
      "epoch": 0.6,
      "eval_distillation_accuracy_counterfactual": 0.5025654181631606,
      "eval_distillation_accuracy_factual": 0.8855823499230374,
      "eval_distillation_f1_counterfactual": 0.4398602753096072,
      "eval_distillation_f1_factual": 0.8717503260210726,
      "eval_groundtruth_accuracy_counterfactual": 0.4989738327347358,
      "eval_groundtruth_f1_counterfactual": 0.44441757184034925,
      "eval_groundtruth_f1_factual": 0.7229789758781793,
      "eval_icace_cosine": 0.4650568664073944,
      "eval_icace_l2": 0.6811135411262512,
      "eval_icace_normdiff": 0.4423196017742157,
      "eval_loss": 8.341341972351074,
      "eval_runtime": 13.1584,
      "eval_samples_per_second": 296.238,
      "eval_steps_per_second": 2.356,
      "step": 350
    },
    {
      "epoch": 0.62,
      "learning_rate": 4.6100519930675915e-05,
      "loss": 9.4159,
      "step": 360
    },
    {
      "epoch": 0.65,
      "learning_rate": 4.58838821490468e-05,
      "loss": 9.6264,
      "step": 380
    },
    {
      "epoch": 0.69,
      "learning_rate": 4.566724436741768e-05,
      "loss": 9.2833,
      "step": 400
    },
    {
      "epoch": 0.69,
      "eval_distillation_accuracy_counterfactual": 0.5495125705489995,
      "eval_distillation_accuracy_factual": 0.8494099538224731,
      "eval_distillation_f1_counterfactual": 0.48234074968553386,
      "eval_distillation_f1_factual": 0.834773556914336,
      "eval_groundtruth_accuracy_counterfactual": 0.5266803488968702,
      "eval_groundtruth_f1_counterfactual": 0.4613217521128874,
      "eval_groundtruth_f1_factual": 0.7080154702931097,
      "eval_icace_cosine": 0.4618784785270691,
      "eval_icace_l2": 0.6433539986610413,
      "eval_icace_normdiff": 0.40616923570632935,
      "eval_loss": 8.343696594238281,
      "eval_runtime": 15.6191,
      "eval_samples_per_second": 249.565,
      "eval_steps_per_second": 1.985,
      "step": 400
    },
    {
      "epoch": 0.72,
      "learning_rate": 4.5450606585788563e-05,
      "loss": 9.3389,
      "step": 420
    },
    {
      "epoch": 0.75,
      "learning_rate": 4.5233968804159446e-05,
      "loss": 9.2901,
      "step": 440
    },
    {
      "epoch": 0.77,
      "eval_distillation_accuracy_counterfactual": 0.4989738327347358,
      "eval_distillation_accuracy_factual": 0.8601847101077476,
      "eval_distillation_f1_counterfactual": 0.33860612154049907,
      "eval_distillation_f1_factual": 0.8490636185423319,
      "eval_groundtruth_accuracy_counterfactual": 0.4892252437147255,
      "eval_groundtruth_f1_counterfactual": 0.33971729579938115,
      "eval_groundtruth_f1_factual": 0.7307035270822468,
      "eval_icace_cosine": 0.475065141916275,
      "eval_icace_l2": 0.6995478272438049,
      "eval_icace_normdiff": 0.440874308347702,
      "eval_loss": 8.762495994567871,
      "eval_runtime": 13.0872,
      "eval_samples_per_second": 297.848,
      "eval_steps_per_second": 2.369,
      "step": 450
    },
    {
      "epoch": 0.79,
      "learning_rate": 4.501733102253033e-05,
      "loss": 9.3355,
      "step": 460
    },
    {
      "epoch": 0.82,
      "learning_rate": 4.480069324090121e-05,
      "loss": 9.1845,
      "step": 480
    },
    {
      "epoch": 0.86,
      "learning_rate": 4.45840554592721e-05,
      "loss": 9.2082,
      "step": 500
    },
    {
      "epoch": 0.86,
      "eval_distillation_accuracy_counterfactual": 0.5823499230374551,
      "eval_distillation_accuracy_factual": 0.8514622883530015,
      "eval_distillation_f1_counterfactual": 0.49132364435744985,
      "eval_distillation_f1_factual": 0.8334607656933123,
      "eval_groundtruth_accuracy_counterfactual": 0.5482298614674191,
      "eval_groundtruth_f1_counterfactual": 0.46237812507137493,
      "eval_groundtruth_f1_factual": 0.7129235668323691,
      "eval_icace_cosine": 0.45347079634666443,
      "eval_icace_l2": 0.6185017824172974,
      "eval_icace_normdiff": 0.3950257897377014,
      "eval_loss": 8.00942611694336,
      "eval_runtime": 13.0722,
      "eval_samples_per_second": 298.191,
      "eval_steps_per_second": 2.371,
      "step": 500
    }
  ],
  "max_steps": 4616,
  "num_train_epochs": 8,
  "total_flos": 4209890279424000.0,
  "trial_name": null,
  "trial_params": null
}