{
  "best_metric": 0.11641146242618561,
  "best_model_checkpoint": "/home/projects/Project - Fundus Photo Glaucoma Detection (AI)//model_results/vit-base-gpu/checkpoint-200",
  "epoch": 5.0,
  "eval_steps": 100,
  "global_step": 285,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.17543859649122806,
      "grad_norm": 1.276576280593872,
      "learning_rate": 9.649122807017545e-06,
      "loss": 0.6232,
      "step": 10
    },
    {
      "epoch": 0.3508771929824561,
      "grad_norm": 1.7671723365783691,
      "learning_rate": 9.298245614035088e-06,
      "loss": 0.4728,
      "step": 20
    },
    {
      "epoch": 0.5263157894736842,
      "grad_norm": 1.358464241027832,
      "learning_rate": 8.947368421052632e-06,
      "loss": 0.3828,
      "step": 30
    },
    {
      "epoch": 0.7017543859649122,
      "grad_norm": 1.4662245512008667,
      "learning_rate": 8.596491228070176e-06,
      "loss": 0.3042,
      "step": 40
    },
    {
      "epoch": 0.8771929824561403,
      "grad_norm": 1.1634291410446167,
      "learning_rate": 8.24561403508772e-06,
      "loss": 0.286,
      "step": 50
    },
    {
      "epoch": 1.0526315789473684,
      "grad_norm": 0.8176971077919006,
      "learning_rate": 7.894736842105265e-06,
      "loss": 0.2169,
      "step": 60
    },
    {
      "epoch": 1.2280701754385965,
      "grad_norm": 0.6673957705497742,
      "learning_rate": 7.5438596491228074e-06,
      "loss": 0.2044,
      "step": 70
    },
    {
      "epoch": 1.4035087719298245,
      "grad_norm": 1.1810154914855957,
      "learning_rate": 7.192982456140352e-06,
      "loss": 0.1821,
      "step": 80
    },
    {
      "epoch": 1.5789473684210527,
      "grad_norm": 0.759559690952301,
      "learning_rate": 6.842105263157896e-06,
      "loss": 0.1558,
      "step": 90
    },
    {
      "epoch": 1.7543859649122808,
      "grad_norm": 0.4498167037963867,
      "learning_rate": 6.491228070175439e-06,
      "loss": 0.149,
      "step": 100
    },
    {
      "epoch": 1.7543859649122808,
      "eval_accuracy": 0.9647577092511013,
      "eval_confusion_matrix": [
        [
          58,
          8
        ],
        [
          0,
          161
        ]
      ],
      "eval_loss": 0.17813891172409058,
      "eval_runtime": 25.2489,
      "eval_samples_per_second": 8.99,
      "eval_steps_per_second": 0.317,
      "step": 100
    },
    {
      "epoch": 1.9298245614035088,
      "grad_norm": 4.42154598236084,
      "learning_rate": 6.140350877192983e-06,
      "loss": 0.1435,
      "step": 110
    },
    {
      "epoch": 2.1052631578947367,
      "grad_norm": 0.48428404331207275,
      "learning_rate": 5.789473684210527e-06,
      "loss": 0.1275,
      "step": 120
    },
    {
      "epoch": 2.280701754385965,
      "grad_norm": 0.41149669885635376,
      "learning_rate": 5.438596491228071e-06,
      "loss": 0.1291,
      "step": 130
    },
    {
      "epoch": 2.456140350877193,
      "grad_norm": 0.49232661724090576,
      "learning_rate": 5.087719298245615e-06,
      "loss": 0.0896,
      "step": 140
    },
    {
      "epoch": 2.6315789473684212,
      "grad_norm": 0.36095741391181946,
      "learning_rate": 4.736842105263158e-06,
      "loss": 0.0997,
      "step": 150
    },
    {
      "epoch": 2.807017543859649,
      "grad_norm": 0.4267736077308655,
      "learning_rate": 4.385964912280702e-06,
      "loss": 0.0925,
      "step": 160
    },
    {
      "epoch": 2.982456140350877,
      "grad_norm": 1.9955291748046875,
      "learning_rate": 4.035087719298246e-06,
      "loss": 0.0907,
      "step": 170
    },
    {
      "epoch": 3.1578947368421053,
      "grad_norm": 0.3319329619407654,
      "learning_rate": 3.6842105263157896e-06,
      "loss": 0.0979,
      "step": 180
    },
    {
      "epoch": 3.3333333333333335,
      "grad_norm": 0.35440099239349365,
      "learning_rate": 3.3333333333333333e-06,
      "loss": 0.0886,
      "step": 190
    },
    {
      "epoch": 3.5087719298245617,
      "grad_norm": 0.5289174318313599,
      "learning_rate": 2.9824561403508774e-06,
      "loss": 0.0996,
      "step": 200
    },
    {
      "epoch": 3.5087719298245617,
      "eval_accuracy": 0.973568281938326,
      "eval_confusion_matrix": [
        [
          60,
          6
        ],
        [
          0,
          161
        ]
      ],
      "eval_loss": 0.11641146242618561,
      "eval_runtime": 25.5874,
      "eval_samples_per_second": 8.872,
      "eval_steps_per_second": 0.313,
      "step": 200
    },
    {
      "epoch": 3.6842105263157894,
      "grad_norm": 0.3807942569255829,
      "learning_rate": 2.631578947368421e-06,
      "loss": 0.1028,
      "step": 210
    },
    {
      "epoch": 3.8596491228070176,
      "grad_norm": 0.6208366751670837,
      "learning_rate": 2.280701754385965e-06,
      "loss": 0.0741,
      "step": 220
    },
    {
      "epoch": 4.035087719298246,
      "grad_norm": 0.7710980176925659,
      "learning_rate": 1.929824561403509e-06,
      "loss": 0.1073,
      "step": 230
    },
    {
      "epoch": 4.2105263157894735,
      "grad_norm": 0.36691102385520935,
      "learning_rate": 1.5789473684210526e-06,
      "loss": 0.0777,
      "step": 240
    },
    {
      "epoch": 4.385964912280702,
      "grad_norm": 0.29168033599853516,
      "learning_rate": 1.2280701754385965e-06,
      "loss": 0.0614,
      "step": 250
    },
    {
      "epoch": 4.56140350877193,
      "grad_norm": 0.2981802821159363,
      "learning_rate": 8.771929824561404e-07,
      "loss": 0.0601,
      "step": 260
    },
    {
      "epoch": 4.7368421052631575,
      "grad_norm": 0.28405195474624634,
      "learning_rate": 5.263157894736843e-07,
      "loss": 0.0525,
      "step": 270
    },
    {
      "epoch": 4.912280701754386,
      "grad_norm": 0.5508224964141846,
      "learning_rate": 1.7543859649122808e-07,
      "loss": 0.0904,
      "step": 280
    },
    {
      "epoch": 5.0,
      "step": 285,
      "total_flos": 7.020774259082035e+17,
      "train_loss": 0.16532422492378637,
      "train_runtime": 2902.1592,
      "train_samples_per_second": 3.142,
      "train_steps_per_second": 0.098
    }
  ],
  "logging_steps": 10,
  "max_steps": 285,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 100,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 7.020774259082035e+17,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}