{
  "best_metric": 0.6213392615318298,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF30-RGBD-b0/checkpoint-40",
  "epoch": 20.0,
  "global_step": 40,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.5,
      "learning_rate": 5e-06,
      "loss": 1.0565,
      "step": 1
    },
    {
      "epoch": 1.0,
      "learning_rate": 1e-05,
      "loss": 1.0485,
      "step": 2
    },
    {
      "epoch": 1.5,
      "learning_rate": 1.5e-05,
      "loss": 1.0559,
      "step": 3
    },
    {
      "epoch": 2.0,
      "learning_rate": 2e-05,
      "loss": 1.0501,
      "step": 4
    },
    {
      "epoch": 2.5,
      "learning_rate": 2.5e-05,
      "loss": 1.0391,
      "step": 5
    },
    {
      "epoch": 3.0,
      "learning_rate": 3e-05,
      "loss": 1.0435,
      "step": 6
    },
    {
      "epoch": 3.5,
      "learning_rate": 3.5e-05,
      "loss": 1.0235,
      "step": 7
    },
    {
      "epoch": 4.0,
      "learning_rate": 4e-05,
      "loss": 1.0155,
      "step": 8
    },
    {
      "epoch": 4.5,
      "learning_rate": 4.5e-05,
      "loss": 1.0107,
      "step": 9
    },
    {
      "epoch": 5.0,
      "learning_rate": 5e-05,
      "loss": 0.9925,
      "step": 10
    },
    {
      "epoch": 5.0,
      "eval_accuracy_safe": 0.16253187340632969,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9084522788185269,
      "eval_iou_safe": 0.04617705465733694,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8841402740524446,
      "eval_loss": 1.0611684322357178,
      "eval_mean_accuracy": 0.5354920761124282,
      "eval_mean_iou": 0.3101057762365938,
      "eval_overall_accuracy": 0.8847389221191406,
      "eval_runtime": 2.0635,
      "eval_samples_per_second": 3.877,
      "eval_steps_per_second": 0.485,
      "step": 10
    },
    {
      "epoch": 5.5,
      "learning_rate": 4.973684210526316e-05,
      "loss": 0.9711,
      "step": 11
    },
    {
      "epoch": 6.0,
      "learning_rate": 4.9473684210526315e-05,
      "loss": 0.9794,
      "step": 12
    },
    {
      "epoch": 6.5,
      "learning_rate": 4.921052631578947e-05,
      "loss": 0.9568,
      "step": 13
    },
    {
      "epoch": 7.0,
      "learning_rate": 4.8947368421052635e-05,
      "loss": 0.9109,
      "step": 14
    },
    {
      "epoch": 7.5,
      "learning_rate": 4.868421052631579e-05,
      "loss": 0.9157,
      "step": 15
    },
    {
      "epoch": 8.0,
      "learning_rate": 4.842105263157895e-05,
      "loss": 0.8739,
      "step": 16
    },
    {
      "epoch": 8.5,
      "learning_rate": 4.8157894736842105e-05,
      "loss": 0.8838,
      "step": 17
    },
    {
      "epoch": 9.0,
      "learning_rate": 4.789473684210526e-05,
      "loss": 0.9098,
      "step": 18
    },
    {
      "epoch": 9.5,
      "learning_rate": 4.7631578947368424e-05,
      "loss": 0.8599,
      "step": 19
    },
    {
      "epoch": 10.0,
      "learning_rate": 4.736842105263158e-05,
      "loss": 0.8589,
      "step": 20
    },
    {
      "epoch": 10.0,
      "eval_accuracy_safe": 0.05294735263236838,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9833241565303213,
      "eval_iou_safe": 0.03730081575721712,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9536688093648232,
      "eval_loss": 0.9441336989402771,
      "eval_mean_accuracy": 0.5181357545813449,
      "eval_mean_iou": 0.33032320837401347,
      "eval_overall_accuracy": 0.9537467956542969,
      "eval_runtime": 1.9246,
      "eval_samples_per_second": 4.157,
      "eval_steps_per_second": 0.52,
      "step": 20
    },
    {
      "epoch": 10.5,
      "learning_rate": 4.7105263157894744e-05,
      "loss": 0.8438,
      "step": 21
    },
    {
      "epoch": 11.0,
      "learning_rate": 4.68421052631579e-05,
      "loss": 0.828,
      "step": 22
    },
    {
      "epoch": 11.5,
      "learning_rate": 4.657894736842106e-05,
      "loss": 0.8353,
      "step": 23
    },
    {
      "epoch": 12.0,
      "learning_rate": 4.6315789473684214e-05,
      "loss": 0.8821,
      "step": 24
    },
    {
      "epoch": 12.5,
      "learning_rate": 4.605263157894737e-05,
      "loss": 0.8043,
      "step": 25
    },
    {
      "epoch": 13.0,
      "learning_rate": 4.5789473684210527e-05,
      "loss": 0.7807,
      "step": 26
    },
    {
      "epoch": 13.5,
      "learning_rate": 4.552631578947369e-05,
      "loss": 0.7855,
      "step": 27
    },
    {
      "epoch": 14.0,
      "learning_rate": 4.5263157894736846e-05,
      "loss": 0.7243,
      "step": 28
    },
    {
      "epoch": 14.5,
      "learning_rate": 4.5e-05,
      "loss": 0.7318,
      "step": 29
    },
    {
      "epoch": 15.0,
      "learning_rate": 4.473684210526316e-05,
      "loss": 0.7016,
      "step": 30
    },
    {
      "epoch": 15.0,
      "eval_accuracy_safe": 0.017234138293085347,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9965333354346406,
      "eval_iou_safe": 0.016899046946699612,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.965381728938365,
      "eval_loss": 0.7763598561286926,
      "eval_mean_accuracy": 0.506883736863863,
      "eval_mean_iou": 0.32742692529502154,
      "eval_overall_accuracy": 0.9654006958007812,
      "eval_runtime": 1.8724,
      "eval_samples_per_second": 4.273,
      "eval_steps_per_second": 0.534,
      "step": 30
    },
    {
      "epoch": 15.5,
      "learning_rate": 4.4473684210526316e-05,
      "loss": 0.6872,
      "step": 31
    },
    {
      "epoch": 16.0,
      "learning_rate": 4.421052631578947e-05,
      "loss": 0.6695,
      "step": 32
    },
    {
      "epoch": 16.5,
      "learning_rate": 4.394736842105263e-05,
      "loss": 0.6728,
      "step": 33
    },
    {
      "epoch": 17.0,
      "learning_rate": 4.368421052631579e-05,
      "loss": 0.757,
      "step": 34
    },
    {
      "epoch": 17.5,
      "learning_rate": 4.342105263157895e-05,
      "loss": 0.6676,
      "step": 35
    },
    {
      "epoch": 18.0,
      "learning_rate": 4.3157894736842105e-05,
      "loss": 0.6491,
      "step": 36
    },
    {
      "epoch": 18.5,
      "learning_rate": 4.289473684210527e-05,
      "loss": 0.6541,
      "step": 37
    },
    {
      "epoch": 19.0,
      "learning_rate": 4.2631578947368425e-05,
      "loss": 0.6635,
      "step": 38
    },
    {
      "epoch": 19.5,
      "learning_rate": 4.236842105263158e-05,
      "loss": 0.6657,
      "step": 39
    },
    {
      "epoch": 20.0,
      "learning_rate": 4.210526315789474e-05,
      "loss": 0.6093,
      "step": 40
    },
    {
      "epoch": 20.0,
      "eval_accuracy_safe": 0.053772311384430775,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9900870827714798,
      "eval_iou_safe": 0.04153054841176062,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9602530036058077,
      "eval_loss": 0.6213392615318298,
      "eval_mean_accuracy": 0.5219296970779553,
      "eval_mean_iou": 0.33392785067252273,
      "eval_overall_accuracy": 0.9603209495544434,
      "eval_runtime": 1.9138,
      "eval_samples_per_second": 4.18,
      "eval_steps_per_second": 0.523,
      "step": 40
    }
  ],
  "max_steps": 200,
  "num_train_epochs": 100,
  "total_flos": 1.40410915651584e+16,
  "trial_name": null,
  "trial_params": null
}