trainer_state.json

{
  "best_metric": 0.8388264444322653,
  "best_model_checkpoint": "google/vit-base-patch16-224-in21k_Human_Activity_Recognition/checkpoint-3150",
  "epoch": 5.0,
  "global_step": 3150,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "learning_rate": 0.00019993650793650796,
      "loss": 2.7024,
      "step": 1
    },
    {
      "epoch": 0.79,
      "learning_rate": 0.00016825396825396826,
      "loss": 1.0814,
      "step": 500
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.7797509984431872,
      "eval_Macro Precision": 0.7896156938204444,
      "eval_Macro Recall": 0.7797368978315249,
      "eval_Micro F1": 0.7793650793650795,
      "eval_Micro Precision": 0.7793650793650794,
      "eval_Micro Recall": 0.7793650793650794,
      "eval_Weighted F1": 0.7795333736972855,
      "eval_Weighted Precision": 0.7896010748173106,
      "eval_Weighted Recall": 0.7793650793650794,
      "eval_accuracy": 0.7793650793650794,
      "eval_loss": 0.7368100881576538,
      "eval_runtime": 2176.4899,
      "eval_samples_per_second": 1.158,
      "eval_steps_per_second": 0.145,
      "step": 630
    },
    {
      "epoch": 1.59,
      "learning_rate": 0.0001365079365079365,
      "loss": 0.5149,
      "step": 1000
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.8036019693964128,
      "eval_Macro Precision": 0.8129758647375759,
      "eval_Macro Recall": 0.8050702488240186,
      "eval_Micro F1": 0.8059523809523811,
      "eval_Micro Precision": 0.805952380952381,
      "eval_Micro Recall": 0.805952380952381,
      "eval_Weighted F1": 0.8048628753859928,
      "eval_Weighted Precision": 0.8136261756908056,
      "eval_Weighted Recall": 0.805952380952381,
      "eval_accuracy": 0.805952380952381,
      "eval_loss": 0.6439331769943237,
      "eval_runtime": 2173.3694,
      "eval_samples_per_second": 1.159,
      "eval_steps_per_second": 0.145,
      "step": 1260
    },
    {
      "epoch": 2.38,
      "learning_rate": 0.00010476190476190477,
      "loss": 0.3023,
      "step": 1500
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.8277822707615001,
      "eval_Macro Precision": 0.8345405192710199,
      "eval_Macro Recall": 0.8255515227375826,
      "eval_Micro F1": 0.8253968253968254,
      "eval_Micro Precision": 0.8253968253968254,
      "eval_Micro Recall": 0.8253968253968254,
      "eval_Weighted F1": 0.827247305487527,
      "eval_Weighted Precision": 0.8335207337542807,
      "eval_Weighted Recall": 0.8253968253968254,
      "eval_accuracy": 0.8253968253968254,
      "eval_loss": 0.7026447057723999,
      "eval_runtime": 2195.4202,
      "eval_samples_per_second": 1.148,
      "eval_steps_per_second": 0.143,
      "step": 1890
    },
    {
      "epoch": 3.17,
      "learning_rate": 7.301587301587302e-05,
      "loss": 0.134,
      "step": 2000
    },
    {
      "epoch": 3.97,
      "learning_rate": 4.126984126984127e-05,
      "loss": 0.0507,
      "step": 2500
    },
    {
      "epoch": 4.0,
      "eval_Macro F1": 0.8348492115859122,
      "eval_Macro Precision": 0.8438045216325151,
      "eval_Macro Recall": 0.8321202050428085,
      "eval_Micro F1": 0.8317460317460317,
      "eval_Micro Precision": 0.8317460317460318,
      "eval_Micro Recall": 0.8317460317460318,
      "eval_Weighted F1": 0.834177405595237,
      "eval_Weighted Precision": 0.842675236484586,
      "eval_Weighted Recall": 0.8317460317460318,
      "eval_accuracy": 0.8317460317460318,
      "eval_loss": 0.7413991689682007,
      "eval_runtime": 2218.4996,
      "eval_samples_per_second": 1.136,
      "eval_steps_per_second": 0.142,
      "step": 2520
    },
    {
      "epoch": 4.76,
      "learning_rate": 9.523809523809523e-06,
      "loss": 0.0128,
      "step": 3000
    },
    {
      "epoch": 5.0,
      "eval_Macro F1": 0.8394073191246441,
      "eval_Macro Precision": 0.842423886079004,
      "eval_Macro Recall": 0.8389614636720059,
      "eval_Micro F1": 0.8380952380952381,
      "eval_Micro Precision": 0.8380952380952381,
      "eval_Micro Recall": 0.8380952380952381,
      "eval_Weighted F1": 0.8388264444322653,
      "eval_Weighted Precision": 0.8421267923064243,
      "eval_Weighted Recall": 0.8380952380952381,
      "eval_accuracy": 0.8380952380952381,
      "eval_loss": 0.74034184217453,
      "eval_runtime": 2176.4146,
      "eval_samples_per_second": 1.158,
      "eval_steps_per_second": 0.145,
      "step": 3150
    },
    {
      "epoch": 5.0,
      "step": 3150,
      "total_flos": 3.906051336688435e+18,
      "train_loss": 0.334039562164791,
      "train_runtime": 140934.4445,
      "train_samples_per_second": 0.358,
      "train_steps_per_second": 0.022
    }
  ],
  "max_steps": 3150,
  "num_train_epochs": 5,
  "total_flos": 3.906051336688435e+18,
  "trial_name": null,
  "trial_params": null
}