trainer_state.json

{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.045454545454545456,
  "eval_steps": 500,
  "global_step": 2,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.022727272727272728,
      "grad_norm": 5.735886096954346,
      "learning_rate": 3.0050000000000002e-05,
      "loss": 2.2694,
      "step": 1
    },
    {
      "epoch": 0.045454545454545456,
      "grad_norm": 5.947845935821533,
      "learning_rate": 1e-07,
      "loss": 2.2475,
      "step": 2
    },
    {
      "epoch": 0.045454545454545456,
      "eval_accuracy__backgound_": NaN,
      "eval_accuracy_aim": 0.2144114770133681,
      "eval_accuracy_arrow": 0.08341416100872939,
      "eval_accuracy_chevron": 0.0022865095933989462,
      "eval_accuracy_desig": 0.09223437061956827,
      "eval_accuracy_disp": 0.27102803738317754,
      "eval_accuracy_rwythr": 0.17653715579026782,
      "eval_accuracy_tdz": 0.04933458099396964,
      "eval_accuracy_thrbar": 0.01395139513951395,
      "eval_iou__backgound_": 0.0,
      "eval_iou_aim": 0.11579095647274264,
      "eval_iou_arrow": 0.008531746031746031,
      "eval_iou_chevron": 0.0019953153465776005,
      "eval_iou_desig": 0.04427994616419919,
      "eval_iou_disp": 0.02664951295717699,
      "eval_iou_rwythr": 0.09797764099987438,
      "eval_iou_tdz": 0.039930993856770174,
      "eval_iou_thrbar": 0.0131286394917946,
      "eval_loss": 2.2161576747894287,
      "eval_mean_accuracy": 0.1128997109427492,
      "eval_mean_iou": 0.038698305702320177,
      "eval_overall_accuracy": 0.10500469317517107,
      "eval_runtime": 192.9187,
      "eval_samples_per_second": 0.451,
      "eval_steps_per_second": 0.057,
      "step": 2
    },
    {
      "epoch": 0.045454545454545456,
      "step": 2,
      "total_flos": 280583108296704.0,
      "train_loss": 2.258466958999634,
      "train_runtime": 267.1515,
      "train_samples_per_second": 0.06,
      "train_steps_per_second": 0.007
    }
  ],
  "logging_steps": 1.0,
  "max_steps": 2,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "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": 280583108296704.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}