{
  "best_metric": 0.8600206611570248,
  "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-MM/NMM_Classification_base_V10/checkpoint-90",
  "epoch": 7.0,
  "eval_steps": 500,
  "global_step": 105,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 4.08638334274292,
      "learning_rate": 4.545454545454546e-05,
      "loss": 0.8429,
      "step": 10
    },
    {
      "epoch": 1.0,
      "step": 15,
      "train_accuracy": 0.828215987294865,
      "train_auc_class_0": 0.9164869255677786,
      "train_auc_class_1": 0.770497463263411,
      "train_auc_class_2": 0.8879816559016815,
      "train_auc_overall": 0.858322014910957,
      "train_loss": 0.4563606083393097,
      "train_runtime": 687.587,
      "train_samples_per_second": 10.989,
      "train_steps_per_second": 0.087
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.8274793388429752,
      "eval_auc_class_0": 0.9340616414835164,
      "eval_auc_class_1": 0.7507519271533845,
      "eval_auc_class_2": 0.8937217290626166,
      "eval_auc_overall": 0.8595117658998391,
      "eval_loss": 0.49912166595458984,
      "eval_runtime": 208.8157,
      "eval_samples_per_second": 9.271,
      "eval_steps_per_second": 0.077,
      "step": 15
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 1.5998913049697876,
      "learning_rate": 4.5212765957446815e-05,
      "loss": 0.4716,
      "step": 20
    },
    {
      "epoch": 2.0,
      "grad_norm": 2.4967057704925537,
      "learning_rate": 3.9893617021276594e-05,
      "loss": 0.3887,
      "step": 30
    },
    {
      "epoch": 2.0,
      "step": 30,
      "train_accuracy": 0.8553467443091582,
      "train_auc_class_0": 0.9472708924769339,
      "train_auc_class_1": 0.9283657851250956,
      "train_auc_class_2": 0.9247901532600991,
      "train_auc_overall": 0.9334756102873761,
      "train_loss": 0.35952258110046387,
      "train_runtime": 685.7612,
      "train_samples_per_second": 11.018,
      "train_steps_per_second": 0.087
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.8502066115702479,
      "eval_auc_class_0": 0.9477621336996336,
      "eval_auc_class_1": 0.9035008683616039,
      "eval_auc_class_2": 0.9096346486873113,
      "eval_auc_overall": 0.9202992169161829,
      "eval_loss": 0.3957720398902893,
      "eval_runtime": 208.9311,
      "eval_samples_per_second": 9.266,
      "eval_steps_per_second": 0.077,
      "step": 30
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 1.716306209564209,
      "learning_rate": 3.4574468085106386e-05,
      "loss": 0.3533,
      "step": 40
    },
    {
      "epoch": 3.0,
      "step": 45,
      "train_accuracy": 0.8750661725780836,
      "train_auc_class_0": 0.9582223466702152,
      "train_auc_class_1": 0.9392220276138971,
      "train_auc_class_2": 0.9390688589921182,
      "train_auc_overall": 0.9455044110920768,
      "train_loss": 0.3131944239139557,
      "train_runtime": 681.8829,
      "train_samples_per_second": 11.081,
      "train_steps_per_second": 0.088
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.8589876033057852,
      "eval_auc_class_0": 0.9578196543040294,
      "eval_auc_class_1": 0.9092855930322143,
      "eval_auc_class_2": 0.920674647298944,
      "eval_auc_overall": 0.9292599648783959,
      "eval_loss": 0.37213757634162903,
      "eval_runtime": 207.6482,
      "eval_samples_per_second": 9.323,
      "eval_steps_per_second": 0.077,
      "step": 45
    },
    {
      "epoch": 3.3333333333333335,
      "grad_norm": 1.727489948272705,
      "learning_rate": 2.925531914893617e-05,
      "loss": 0.3208,
      "step": 50
    },
    {
      "epoch": 4.0,
      "grad_norm": 10.33216381072998,
      "learning_rate": 2.393617021276596e-05,
      "loss": 0.3092,
      "step": 60
    },
    {
      "epoch": 4.0,
      "step": 60,
      "train_accuracy": 0.8848597141344626,
      "train_auc_class_0": 0.9659353652856636,
      "train_auc_class_1": 0.9407955383480827,
      "train_auc_class_2": 0.946874731546507,
      "train_auc_overall": 0.9512018783934177,
      "train_loss": 0.29102450609207153,
      "train_runtime": 680.3084,
      "train_samples_per_second": 11.107,
      "train_steps_per_second": 0.088
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.8579545454545454,
      "eval_auc_class_0": 0.9606026785714286,
      "eval_auc_class_1": 0.9083715281858423,
      "eval_auc_class_2": 0.9231885836547031,
      "eval_auc_overall": 0.9307209301373248,
      "eval_loss": 0.37202903628349304,
      "eval_runtime": 208.7004,
      "eval_samples_per_second": 9.276,
      "eval_steps_per_second": 0.077,
      "step": 60
    },
    {
      "epoch": 4.666666666666667,
      "grad_norm": 2.016211748123169,
      "learning_rate": 1.8617021276595745e-05,
      "loss": 0.2891,
      "step": 70
    },
    {
      "epoch": 5.0,
      "step": 75,
      "train_accuracy": 0.8913446267866596,
      "train_auc_class_0": 0.9681433160338301,
      "train_auc_class_1": 0.9542337553261225,
      "train_auc_class_2": 0.9506045992123575,
      "train_auc_overall": 0.9576605568574367,
      "train_loss": 0.2791108191013336,
      "train_runtime": 690.7325,
      "train_samples_per_second": 10.939,
      "train_steps_per_second": 0.087
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.8584710743801653,
      "eval_auc_class_0": 0.9613524496336998,
      "eval_auc_class_1": 0.9090984273731954,
      "eval_auc_class_2": 0.9242633286351978,
      "eval_auc_overall": 0.9315714018806975,
      "eval_loss": 0.36684560775756836,
      "eval_runtime": 211.7912,
      "eval_samples_per_second": 9.141,
      "eval_steps_per_second": 0.076,
      "step": 75
    },
    {
      "epoch": 5.333333333333333,
      "grad_norm": 2.7578961849212646,
      "learning_rate": 1.3297872340425532e-05,
      "loss": 0.301,
      "step": 80
    },
    {
      "epoch": 6.0,
      "grad_norm": 2.526148557662964,
      "learning_rate": 7.97872340425532e-06,
      "loss": 0.2852,
      "step": 90
    },
    {
      "epoch": 6.0,
      "step": 90,
      "train_accuracy": 0.8955796717840127,
      "train_auc_class_0": 0.9719376552519517,
      "train_auc_class_1": 0.9603400183000109,
      "train_auc_class_2": 0.9560110428344394,
      "train_auc_overall": 0.9627629054621339,
      "train_loss": 0.2651444673538208,
      "train_runtime": 689.8251,
      "train_samples_per_second": 10.954,
      "train_steps_per_second": 0.087
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.8600206611570248,
      "eval_auc_class_0": 0.9623798076923077,
      "eval_auc_class_1": 0.9106044580248364,
      "eval_auc_class_2": 0.9252550195053197,
      "eval_auc_overall": 0.932746428407488,
      "eval_loss": 0.358356237411499,
      "eval_runtime": 209.7236,
      "eval_samples_per_second": 9.231,
      "eval_steps_per_second": 0.076,
      "step": 90
    },
    {
      "epoch": 6.666666666666667,
      "grad_norm": 1.6473604440689087,
      "learning_rate": 2.6595744680851065e-06,
      "loss": 0.2804,
      "step": 100
    },
    {
      "epoch": 7.0,
      "step": 105,
      "train_accuracy": 0.8926680783483325,
      "train_auc_class_0": 0.9692773945765316,
      "train_auc_class_1": 0.9595504889107397,
      "train_auc_class_2": 0.9538150093690271,
      "train_auc_overall": 0.9608809642854328,
      "train_loss": 0.2725497782230377,
      "train_runtime": 674.8314,
      "train_samples_per_second": 11.197,
      "train_steps_per_second": 0.089
    },
    {
      "epoch": 7.0,
      "eval_accuracy": 0.8600206611570248,
      "eval_auc_class_0": 0.9628448374542126,
      "eval_auc_class_1": 0.90828447439095,
      "eval_auc_class_2": 0.924919084223066,
      "eval_auc_overall": 0.9320161320227428,
      "eval_loss": 0.3619421422481537,
      "eval_runtime": 204.9165,
      "eval_samples_per_second": 9.448,
      "eval_steps_per_second": 0.078,
      "step": 105
    },
    {
      "epoch": 7.0,
      "step": 105,
      "total_flos": 4.143902092898771e+18,
      "train_loss": 0.37870263712746755,
      "train_runtime": 11542.069,
      "train_samples_per_second": 4.583,
      "train_steps_per_second": 0.009
    }
  ],
  "logging_steps": 10,
  "max_steps": 105,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 7,
  "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": 4.143902092898771e+18,
  "train_batch_size": 128,
  "trial_name": null,
  "trial_params": null
}