{
  "best_metric": 0.5702608823776245,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_7/checkpoint-40",
  "epoch": 13.333333333333334,
  "global_step": 40,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.33,
      "learning_rate": 5e-06,
      "loss": 1.2178,
      "step": 1
    },
    {
      "epoch": 0.67,
      "learning_rate": 1e-05,
      "loss": 1.2142,
      "step": 2
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.5e-05,
      "loss": 1.2223,
      "step": 3
    },
    {
      "epoch": 1.33,
      "learning_rate": 2e-05,
      "loss": 1.208,
      "step": 4
    },
    {
      "epoch": 1.67,
      "learning_rate": 2.5e-05,
      "loss": 1.2072,
      "step": 5
    },
    {
      "epoch": 2.0,
      "learning_rate": 3e-05,
      "loss": 1.2015,
      "step": 6
    },
    {
      "epoch": 2.33,
      "learning_rate": 3.5000000000000004e-05,
      "loss": 1.1888,
      "step": 7
    },
    {
      "epoch": 2.67,
      "learning_rate": 4e-05,
      "loss": 1.1768,
      "step": 8
    },
    {
      "epoch": 3.0,
      "learning_rate": 4.5e-05,
      "loss": 1.1543,
      "step": 9
    },
    {
      "epoch": 3.33,
      "learning_rate": 5e-05,
      "loss": 1.1505,
      "step": 10
    },
    {
      "epoch": 3.33,
      "eval_accuracy_dropoff": 0.9456199600248122,
      "eval_accuracy_undropoff": 0.2616499186280552,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.07032648592680048,
      "eval_iou_undropoff": 0.2615922132145178,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.1103304624557495,
      "eval_mean_accuracy": 0.6036349393264338,
      "eval_mean_iou": 0.11063956638043942,
      "eval_overall_accuracy": 0.2919347127278646,
      "eval_runtime": 1.7484,
      "eval_samples_per_second": 8.579,
      "eval_steps_per_second": 0.572,
      "step": 10
    },
    {
      "epoch": 3.67,
      "learning_rate": 5.500000000000001e-05,
      "loss": 1.1405,
      "step": 11
    },
    {
      "epoch": 4.0,
      "learning_rate": 6e-05,
      "loss": 1.1039,
      "step": 12
    },
    {
      "epoch": 4.33,
      "learning_rate": 6.500000000000001e-05,
      "loss": 1.1211,
      "step": 13
    },
    {
      "epoch": 4.67,
      "learning_rate": 7.000000000000001e-05,
      "loss": 1.0914,
      "step": 14
    },
    {
      "epoch": 5.0,
      "learning_rate": 7.500000000000001e-05,
      "loss": 1.0583,
      "step": 15
    },
    {
      "epoch": 5.33,
      "learning_rate": 8e-05,
      "loss": 1.0527,
      "step": 16
    },
    {
      "epoch": 5.67,
      "learning_rate": 8.5e-05,
      "loss": 1.0227,
      "step": 17
    },
    {
      "epoch": 6.0,
      "learning_rate": 9e-05,
      "loss": 1.1078,
      "step": 18
    },
    {
      "epoch": 6.33,
      "learning_rate": 8.973684210526316e-05,
      "loss": 1.0001,
      "step": 19
    },
    {
      "epoch": 6.67,
      "learning_rate": 8.947368421052632e-05,
      "loss": 0.9635,
      "step": 20
    },
    {
      "epoch": 6.67,
      "eval_accuracy_dropoff": 0.8177051025800078,
      "eval_accuracy_undropoff": 0.8763050644323176,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.2434656953279978,
      "eval_iou_undropoff": 0.8694137027963167,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.0114336013793945,
      "eval_mean_accuracy": 0.8470050835061627,
      "eval_mean_iou": 0.3709597993747715,
      "eval_overall_accuracy": 0.8737103780110677,
      "eval_runtime": 1.7139,
      "eval_samples_per_second": 8.752,
      "eval_steps_per_second": 0.583,
      "step": 20
    },
    {
      "epoch": 7.0,
      "learning_rate": 8.921052631578948e-05,
      "loss": 1.0405,
      "step": 21
    },
    {
      "epoch": 7.33,
      "learning_rate": 8.894736842105263e-05,
      "loss": 0.9095,
      "step": 22
    },
    {
      "epoch": 7.67,
      "learning_rate": 8.86842105263158e-05,
      "loss": 0.9104,
      "step": 23
    },
    {
      "epoch": 8.0,
      "learning_rate": 8.842105263157894e-05,
      "loss": 0.8736,
      "step": 24
    },
    {
      "epoch": 8.33,
      "learning_rate": 8.815789473684211e-05,
      "loss": 0.8849,
      "step": 25
    },
    {
      "epoch": 8.67,
      "learning_rate": 8.789473684210527e-05,
      "loss": 0.8143,
      "step": 26
    },
    {
      "epoch": 9.0,
      "learning_rate": 8.763157894736843e-05,
      "loss": 0.7846,
      "step": 27
    },
    {
      "epoch": 9.33,
      "learning_rate": 8.736842105263158e-05,
      "loss": 0.7748,
      "step": 28
    },
    {
      "epoch": 9.67,
      "learning_rate": 8.710526315789474e-05,
      "loss": 0.7513,
      "step": 29
    },
    {
      "epoch": 10.0,
      "learning_rate": 8.68421052631579e-05,
      "loss": 0.9358,
      "step": 30
    },
    {
      "epoch": 10.0,
      "eval_accuracy_dropoff": 0.5848209157534404,
      "eval_accuracy_undropoff": 0.9605968730608304,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3193744373731639,
      "eval_iou_undropoff": 0.9425019881457949,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.8242360353469849,
      "eval_mean_accuracy": 0.7727088944071354,
      "eval_mean_iou": 0.42062547517298626,
      "eval_overall_accuracy": 0.9439582824707031,
      "eval_runtime": 1.7189,
      "eval_samples_per_second": 8.726,
      "eval_steps_per_second": 0.582,
      "step": 30
    },
    {
      "epoch": 10.33,
      "learning_rate": 8.657894736842107e-05,
      "loss": 0.7191,
      "step": 31
    },
    {
      "epoch": 10.67,
      "learning_rate": 8.631578947368421e-05,
      "loss": 0.73,
      "step": 32
    },
    {
      "epoch": 11.0,
      "learning_rate": 8.605263157894737e-05,
      "loss": 0.6168,
      "step": 33
    },
    {
      "epoch": 11.33,
      "learning_rate": 8.578947368421054e-05,
      "loss": 0.6559,
      "step": 34
    },
    {
      "epoch": 11.67,
      "learning_rate": 8.552631578947369e-05,
      "loss": 0.6277,
      "step": 35
    },
    {
      "epoch": 12.0,
      "learning_rate": 8.526315789473684e-05,
      "loss": 0.6168,
      "step": 36
    },
    {
      "epoch": 12.33,
      "learning_rate": 8.5e-05,
      "loss": 0.6095,
      "step": 37
    },
    {
      "epoch": 12.67,
      "learning_rate": 8.473684210526316e-05,
      "loss": 0.5927,
      "step": 38
    },
    {
      "epoch": 13.0,
      "learning_rate": 8.447368421052632e-05,
      "loss": 0.8162,
      "step": 39
    },
    {
      "epoch": 13.33,
      "learning_rate": 8.421052631578948e-05,
      "loss": 0.579,
      "step": 40
    },
    {
      "epoch": 13.33,
      "eval_accuracy_dropoff": 0.5401589817814231,
      "eval_accuracy_undropoff": 0.982923599779886,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3951496002117638,
      "eval_iou_undropoff": 0.9624271055937778,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.5702608823776245,
      "eval_mean_accuracy": 0.7615412907806545,
      "eval_mean_iou": 0.4525255686018472,
      "eval_overall_accuracy": 0.9633188883463542,
      "eval_runtime": 1.8153,
      "eval_samples_per_second": 8.263,
      "eval_steps_per_second": 0.551,
      "step": 40
    }
  ],
  "max_steps": 360,
  "num_train_epochs": 120,
  "total_flos": 8028359035453440.0,
  "trial_name": null,
  "trial_params": null
}