trainer_state.json

{
  "best_metric": 0.8583717942237854,
  "best_model_checkpoint": "vit-base-patch16-224-in21k-Brain_Tumors_Image_Classification/checkpoint-540",
  "epoch": 3.0,
  "global_step": 540,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 0.00019962962962962963,
      "loss": 1.3668,
      "step": 1
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.6428228240288554,
      "eval_Macro Precision": 0.7865541878512177,
      "eval_Macro Recall": 0.6529559062167758,
      "eval_Micro F1": 0.6852791878172588,
      "eval_Micro Precision": 0.6852791878172588,
      "eval_Micro Recall": 0.6852791878172588,
      "eval_Weighted F1": 0.6523871249366453,
      "eval_Weighted Precision": 0.7636872189071013,
      "eval_Weighted Recall": 0.6852791878172588,
      "eval_accuracy": 0.6852791878172588,
      "eval_loss": 1.073616862297058,
      "eval_runtime": 129.3739,
      "eval_samples_per_second": 3.045,
      "eval_steps_per_second": 0.386,
      "step": 180
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.7410889275122396,
      "eval_Macro Precision": 0.8527626317702652,
      "eval_Macro Recall": 0.7757990599294948,
      "eval_Micro F1": 0.7791878172588832,
      "eval_Micro Precision": 0.7791878172588832,
      "eval_Micro Recall": 0.7791878172588832,
      "eval_Weighted F1": 0.733526477593851,
      "eval_Weighted Precision": 0.8390822349786973,
      "eval_Weighted Recall": 0.7791878172588832,
      "eval_accuracy": 0.7791878172588832,
      "eval_loss": 1.0248740911483765,
      "eval_runtime": 129.3384,
      "eval_samples_per_second": 3.046,
      "eval_steps_per_second": 0.387,
      "step": 360
    },
    {
      "epoch": 2.78,
      "learning_rate": 1.4814814814814815e-05,
      "loss": 0.1864,
      "step": 500
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.8054073624217509,
      "eval_Macro Precision": 0.8769299591002045,
      "eval_Macro Recall": 0.8149206815511164,
      "eval_Micro F1": 0.8197969543147208,
      "eval_Micro Precision": 0.8197969543147208,
      "eval_Micro Recall": 0.8197969543147208,
      "eval_Weighted F1": 0.7987157640227063,
      "eval_Weighted Precision": 0.8614960605400019,
      "eval_Weighted Recall": 0.8197969543147208,
      "eval_accuracy": 0.8197969543147208,
      "eval_loss": 0.8583717942237854,
      "eval_runtime": 95.2031,
      "eval_samples_per_second": 4.139,
      "eval_steps_per_second": 0.525,
      "step": 540
    },
    {
      "epoch": 3.0,
      "step": 540,
      "total_flos": 6.672179904948634e+17,
      "train_loss": 0.17559059135339877,
      "train_runtime": 8350.3418,
      "train_samples_per_second": 1.031,
      "train_steps_per_second": 0.065
    }
  ],
  "max_steps": 540,
  "num_train_epochs": 3,
  "total_flos": 6.672179904948634e+17,
  "trial_name": null,
  "trial_params": null
}