{
  "best_metric": 0.7926363320597429,
  "best_model_checkpoint": "logs/indian_build_rr/MHGanainy/xmod-roberta-base-legal-multi/seed_1/checkpoint-992",
  "epoch": 11.0,
  "eval_steps": 500,
  "global_step": 1364,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.7513025356026398,
      "eval_loss": 0.8476096391677856,
      "eval_macro-f1": 0.49195531709931745,
      "eval_micro-f1": 0.7513025356026398,
      "eval_precision-macro": 0.6248331901101877,
      "eval_precision-micro": 0.7513025356026398,
      "eval_recall-macro": 0.4796676051190033,
      "eval_recall-micro": 0.7513025356026398,
      "eval_runtime": 4.4512,
      "eval_samples_per_second": 6.74,
      "eval_steps_per_second": 3.37,
      "step": 124
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.698853768669677,
      "eval_loss": 0.8966864347457886,
      "eval_macro-f1": 0.4971826438165766,
      "eval_micro-f1": 0.698853768669677,
      "eval_precision-macro": 0.4955191538482637,
      "eval_precision-micro": 0.698853768669677,
      "eval_recall-macro": 0.5356613467712165,
      "eval_recall-micro": 0.698853768669677,
      "eval_runtime": 4.1291,
      "eval_samples_per_second": 7.266,
      "eval_steps_per_second": 3.633,
      "step": 248
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7676276484890587,
      "eval_loss": 0.7835128903388977,
      "eval_macro-f1": 0.5488805777446147,
      "eval_micro-f1": 0.7676276484890587,
      "eval_precision-macro": 0.5898986319845603,
      "eval_precision-micro": 0.7676276484890587,
      "eval_recall-macro": 0.5434855468928687,
      "eval_recall-micro": 0.7676276484890587,
      "eval_runtime": 3.9959,
      "eval_samples_per_second": 7.508,
      "eval_steps_per_second": 3.754,
      "step": 372
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.767280305661688,
      "eval_loss": 0.7243838906288147,
      "eval_macro-f1": 0.5413535210773716,
      "eval_micro-f1": 0.767280305661688,
      "eval_precision-macro": 0.6550974631087091,
      "eval_precision-micro": 0.767280305661688,
      "eval_recall-macro": 0.5427393182281322,
      "eval_recall-micro": 0.767280305661688,
      "eval_runtime": 3.8286,
      "eval_samples_per_second": 7.836,
      "eval_steps_per_second": 3.918,
      "step": 496
    },
    {
      "epoch": 4.032258064516129,
      "grad_norm": 5.5077667236328125,
      "learning_rate": 2.398790322580645e-05,
      "loss": 0.9624,
      "step": 500
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.7641542202153525,
      "eval_loss": 0.7272040247917175,
      "eval_macro-f1": 0.5751739461849497,
      "eval_micro-f1": 0.7641542202153525,
      "eval_precision-macro": 0.5824199451464319,
      "eval_precision-micro": 0.7641542202153525,
      "eval_recall-macro": 0.5938184715362318,
      "eval_recall-micro": 0.7641542202153525,
      "eval_runtime": 4.0236,
      "eval_samples_per_second": 7.456,
      "eval_steps_per_second": 3.728,
      "step": 620
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.7794373046196597,
      "eval_loss": 0.7374066710472107,
      "eval_macro-f1": 0.5727356505277835,
      "eval_micro-f1": 0.7794373046196597,
      "eval_precision-macro": 0.6265669234034454,
      "eval_precision-micro": 0.7794373046196597,
      "eval_recall-macro": 0.5738879184407369,
      "eval_recall-micro": 0.7794373046196597,
      "eval_runtime": 4.8957,
      "eval_samples_per_second": 6.128,
      "eval_steps_per_second": 3.064,
      "step": 744
    },
    {
      "epoch": 7.0,
      "eval_accuracy": 0.7849947898575894,
      "eval_loss": 0.7611834406852722,
      "eval_macro-f1": 0.5809327835705382,
      "eval_micro-f1": 0.7849947898575894,
      "eval_precision-macro": 0.6258277538647542,
      "eval_precision-micro": 0.7849947898575894,
      "eval_recall-macro": 0.5706237921365069,
      "eval_recall-micro": 0.7849947898575894,
      "eval_runtime": 4.0551,
      "eval_samples_per_second": 7.398,
      "eval_steps_per_second": 3.699,
      "step": 868
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.7926363320597429,
      "eval_loss": 0.771350085735321,
      "eval_macro-f1": 0.5940511190085977,
      "eval_micro-f1": 0.7926363320597429,
      "eval_precision-macro": 0.6249375951572697,
      "eval_precision-micro": 0.7926363320597429,
      "eval_recall-macro": 0.582532042975821,
      "eval_recall-micro": 0.7926363320597429,
      "eval_runtime": 4.0804,
      "eval_samples_per_second": 7.352,
      "eval_steps_per_second": 3.676,
      "step": 992
    },
    {
      "epoch": 8.064516129032258,
      "grad_norm": 4.190323352813721,
      "learning_rate": 1.793951612903226e-05,
      "loss": 0.4834,
      "step": 1000
    },
    {
      "epoch": 9.0,
      "eval_accuracy": 0.779089961792289,
      "eval_loss": 0.8429863452911377,
      "eval_macro-f1": 0.5897140550046736,
      "eval_micro-f1": 0.779089961792289,
      "eval_precision-macro": 0.5989913985221759,
      "eval_precision-micro": 0.779089961792289,
      "eval_recall-macro": 0.5902144051148294,
      "eval_recall-micro": 0.779089961792289,
      "eval_runtime": 3.8272,
      "eval_samples_per_second": 7.839,
      "eval_steps_per_second": 3.919,
      "step": 1116
    },
    {
      "epoch": 10.0,
      "eval_accuracy": 0.7752691906912123,
      "eval_loss": 0.8700084686279297,
      "eval_macro-f1": 0.5805579351569137,
      "eval_micro-f1": 0.7752691906912123,
      "eval_precision-macro": 0.5904226342711424,
      "eval_precision-micro": 0.7752691906912123,
      "eval_recall-macro": 0.5790010403564043,
      "eval_recall-micro": 0.7752691906912123,
      "eval_runtime": 4.028,
      "eval_samples_per_second": 7.448,
      "eval_steps_per_second": 3.724,
      "step": 1240
    },
    {
      "epoch": 11.0,
      "eval_accuracy": 0.789857589440778,
      "eval_loss": 0.9059827327728271,
      "eval_macro-f1": 0.6018767746622149,
      "eval_micro-f1": 0.789857589440778,
      "eval_precision-macro": 0.6549734000220255,
      "eval_precision-micro": 0.789857589440778,
      "eval_recall-macro": 0.5864660941250599,
      "eval_recall-micro": 0.789857589440778,
      "eval_runtime": 4.1199,
      "eval_samples_per_second": 7.282,
      "eval_steps_per_second": 3.641,
      "step": 1364
    },
    {
      "epoch": 11.0,
      "step": 1364,
      "total_flos": 6.8167278845952e+16,
      "train_loss": 0.6006245151642831,
      "train_runtime": 980.8913,
      "train_samples_per_second": 5.036,
      "train_steps_per_second": 2.528
    }
  ],
  "logging_steps": 500,
  "max_steps": 2480,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 3,
        "early_stopping_threshold": 0.0
      },
      "attributes": {
        "early_stopping_patience_counter": 0
      }
    },
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 6.8167278845952e+16,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}