trainer_state.json

{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.9047619047619047,
  "eval_steps": 500,
  "global_step": 200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.95,
      "learning_rate": 3.0050000000000002e-05,
      "loss": 2.069,
      "step": 100
    },
    {
      "epoch": 1.0,
      "eval_accuracy_background": NaN,
      "eval_accuracy_baseball-diamond": 0.0,
      "eval_accuracy_basketball-court": 0.0,
      "eval_accuracy_bridge": 0.0,
      "eval_accuracy_ground-track-field": 0.0,
      "eval_accuracy_harbor": 0.47334601346835964,
      "eval_accuracy_helicopter": NaN,
      "eval_accuracy_large-vehicle": 0.35059128278327106,
      "eval_accuracy_plane": 0.0,
      "eval_accuracy_roundabout": 0.0,
      "eval_accuracy_ship": 0.4326614520657842,
      "eval_accuracy_small-vehicle": 0.0001883516491043023,
      "eval_accuracy_soccer-ball-field": 0.00015808777033008727,
      "eval_accuracy_storage-tank": 0.0,
      "eval_accuracy_swimming-pool": 0.0,
      "eval_accuracy_tennis-court": 0.8373677834210894,
      "eval_iou_background": 0.0,
      "eval_iou_baseball-diamond": 0.0,
      "eval_iou_basketball-court": 0.0,
      "eval_iou_bridge": 0.0,
      "eval_iou_ground-track-field": 0.0,
      "eval_iou_harbor": 0.22125573683831126,
      "eval_iou_helicopter": NaN,
      "eval_iou_large-vehicle": 0.34602565881179354,
      "eval_iou_plane": 0.0,
      "eval_iou_roundabout": 0.0,
      "eval_iou_ship": 0.07936099180546487,
      "eval_iou_small-vehicle": 0.00018768512578315883,
      "eval_iou_soccer-ball-field": 0.00015805841048559496,
      "eval_iou_storage-tank": 0.0,
      "eval_iou_swimming-pool": 0.0,
      "eval_iou_tennis-court": 0.7660002983738624,
      "eval_loss": 1.4974907636642456,
      "eval_mean_accuracy": 0.1495937836541385,
      "eval_mean_iou": 0.09419922862438006,
      "eval_overall_accuracy": 0.2837216122470019,
      "eval_runtime": 131.7329,
      "eval_samples_per_second": 0.752,
      "eval_steps_per_second": 0.099,
      "step": 105
    },
    {
      "epoch": 1.9,
      "learning_rate": 1e-07,
      "loss": 1.5141,
      "step": 200
    },
    {
      "epoch": 1.9,
      "eval_accuracy_background": NaN,
      "eval_accuracy_baseball-diamond": 0.0,
      "eval_accuracy_basketball-court": 0.0,
      "eval_accuracy_bridge": 0.0,
      "eval_accuracy_ground-track-field": 0.0,
      "eval_accuracy_harbor": 0.5786254048218784,
      "eval_accuracy_helicopter": NaN,
      "eval_accuracy_large-vehicle": 0.2588257343134778,
      "eval_accuracy_plane": 0.0,
      "eval_accuracy_roundabout": 0.0,
      "eval_accuracy_ship": 0.1352787805856398,
      "eval_accuracy_small-vehicle": 0.00013013386665388158,
      "eval_accuracy_soccer-ball-field": 0.0,
      "eval_accuracy_storage-tank": 0.0,
      "eval_accuracy_swimming-pool": 0.0,
      "eval_accuracy_tennis-court": 0.7306043520805181,
      "eval_iou_background": 0.0,
      "eval_iou_baseball-diamond": 0.0,
      "eval_iou_basketball-court": 0.0,
      "eval_iou_bridge": 0.0,
      "eval_iou_ground-track-field": 0.0,
      "eval_iou_harbor": 0.2867916836526702,
      "eval_iou_helicopter": NaN,
      "eval_iou_large-vehicle": 0.2563089613847982,
      "eval_iou_plane": 0.0,
      "eval_iou_roundabout": 0.0,
      "eval_iou_ship": 0.053192957552098734,
      "eval_iou_small-vehicle": 0.00012985237834882448,
      "eval_iou_soccer-ball-field": 0.0,
      "eval_iou_storage-tank": 0.0,
      "eval_iou_swimming-pool": 0.0,
      "eval_iou_tennis-court": 0.7061565483696631,
      "eval_loss": 1.313840627670288,
      "eval_mean_accuracy": 0.1216760289762977,
      "eval_mean_iou": 0.08683866688917194,
      "eval_overall_accuracy": 0.22847871034583,
      "eval_runtime": 130.8677,
      "eval_samples_per_second": 0.756,
      "eval_steps_per_second": 0.099,
      "step": 200
    },
    {
      "epoch": 1.9,
      "step": 200,
      "total_flos": 2.80718928248832e+16,
      "train_loss": 1.7915446472167968,
      "train_runtime": 623.7007,
      "train_samples_per_second": 2.565,
      "train_steps_per_second": 0.321
    }
  ],
  "logging_steps": 100,
  "max_steps": 200,
  "num_train_epochs": 2,
  "save_steps": 500,
  "total_flos": 2.80718928248832e+16,
  "trial_name": null,
  "trial_params": null
}