{
  "best_metric": 0.43396273255348206,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF30-RGBD-b0/checkpoint-80",
  "epoch": 40.0,
  "global_step": 80,
  "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
    },
    {
      "epoch": 20.5,
      "learning_rate": 4.18421052631579e-05,
      "loss": 0.6155,
      "step": 41
    },
    {
      "epoch": 21.0,
      "learning_rate": 4.157894736842106e-05,
      "loss": 0.6127,
      "step": 42
    },
    {
      "epoch": 21.5,
      "learning_rate": 4.1315789473684214e-05,
      "loss": 0.5792,
      "step": 43
    },
    {
      "epoch": 22.0,
      "learning_rate": 4.105263157894737e-05,
      "loss": 0.5893,
      "step": 44
    },
    {
      "epoch": 22.5,
      "learning_rate": 4.078947368421053e-05,
      "loss": 0.6444,
      "step": 45
    },
    {
      "epoch": 23.0,
      "learning_rate": 4.0526315789473684e-05,
      "loss": 0.5666,
      "step": 46
    },
    {
      "epoch": 23.5,
      "learning_rate": 4.026315789473684e-05,
      "loss": 0.5816,
      "step": 47
    },
    {
      "epoch": 24.0,
      "learning_rate": 4e-05,
      "loss": 0.5644,
      "step": 48
    },
    {
      "epoch": 24.5,
      "learning_rate": 3.973684210526316e-05,
      "loss": 0.5985,
      "step": 49
    },
    {
      "epoch": 25.0,
      "learning_rate": 3.9473684210526316e-05,
      "loss": 0.5281,
      "step": 50
    },
    {
      "epoch": 25.0,
      "eval_accuracy_safe": 0.04757762111894405,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9951110130501034,
      "eval_iou_safe": 0.041666666666666664,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9649351951785595,
      "eval_loss": 0.5431288480758667,
      "eval_mean_accuracy": 0.5213443170845238,
      "eval_mean_iou": 0.33553395394840874,
      "eval_overall_accuracy": 0.9649882316589355,
      "eval_runtime": 1.915,
      "eval_samples_per_second": 4.178,
      "eval_steps_per_second": 0.522,
      "step": 50
    },
    {
      "epoch": 25.5,
      "learning_rate": 3.921052631578947e-05,
      "loss": 0.5531,
      "step": 51
    },
    {
      "epoch": 26.0,
      "learning_rate": 3.894736842105263e-05,
      "loss": 0.5125,
      "step": 52
    },
    {
      "epoch": 26.5,
      "learning_rate": 3.868421052631579e-05,
      "loss": 0.5446,
      "step": 53
    },
    {
      "epoch": 27.0,
      "learning_rate": 3.842105263157895e-05,
      "loss": 0.6699,
      "step": 54
    },
    {
      "epoch": 27.5,
      "learning_rate": 3.815789473684211e-05,
      "loss": 0.5265,
      "step": 55
    },
    {
      "epoch": 28.0,
      "learning_rate": 3.789473684210527e-05,
      "loss": 0.5342,
      "step": 56
    },
    {
      "epoch": 28.5,
      "learning_rate": 3.7631578947368425e-05,
      "loss": 0.5428,
      "step": 57
    },
    {
      "epoch": 29.0,
      "learning_rate": 3.736842105263158e-05,
      "loss": 0.5174,
      "step": 58
    },
    {
      "epoch": 29.5,
      "learning_rate": 3.710526315789474e-05,
      "loss": 0.5173,
      "step": 59
    },
    {
      "epoch": 30.0,
      "learning_rate": 3.6842105263157895e-05,
      "loss": 0.5077,
      "step": 60
    },
    {
      "epoch": 30.0,
      "eval_accuracy_safe": 0.05236238188090595,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9937724146286449,
      "eval_iou_safe": 0.0444283241702301,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.963784012789081,
      "eval_loss": 0.5042787194252014,
      "eval_mean_accuracy": 0.5230673982547754,
      "eval_mean_iou": 0.336070778986437,
      "eval_overall_accuracy": 0.9638442993164062,
      "eval_runtime": 1.8605,
      "eval_samples_per_second": 4.3,
      "eval_steps_per_second": 0.537,
      "step": 60
    },
    {
      "epoch": 30.5,
      "learning_rate": 3.657894736842106e-05,
      "loss": 0.5175,
      "step": 61
    },
    {
      "epoch": 31.0,
      "learning_rate": 3.6315789473684214e-05,
      "loss": 0.5155,
      "step": 62
    },
    {
      "epoch": 31.5,
      "learning_rate": 3.605263157894737e-05,
      "loss": 0.5188,
      "step": 63
    },
    {
      "epoch": 32.0,
      "learning_rate": 3.578947368421053e-05,
      "loss": 0.4911,
      "step": 64
    },
    {
      "epoch": 32.5,
      "learning_rate": 3.5526315789473684e-05,
      "loss": 0.4856,
      "step": 65
    },
    {
      "epoch": 33.0,
      "learning_rate": 3.526315789473684e-05,
      "loss": 0.5214,
      "step": 66
    },
    {
      "epoch": 33.5,
      "learning_rate": 3.5e-05,
      "loss": 0.4863,
      "step": 67
    },
    {
      "epoch": 34.0,
      "learning_rate": 3.473684210526316e-05,
      "loss": 0.5344,
      "step": 68
    },
    {
      "epoch": 34.5,
      "learning_rate": 3.447368421052632e-05,
      "loss": 0.4722,
      "step": 69
    },
    {
      "epoch": 35.0,
      "learning_rate": 3.421052631578947e-05,
      "loss": 0.5197,
      "step": 70
    },
    {
      "epoch": 35.0,
      "eval_accuracy_safe": 0.05428228588570572,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9955749423043395,
      "eval_iou_safe": 0.04808982791841074,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9655911800769321,
      "eval_loss": 0.45792385935783386,
      "eval_mean_accuracy": 0.5249286140950227,
      "eval_mean_iou": 0.33789366933178094,
      "eval_overall_accuracy": 0.9656505584716797,
      "eval_runtime": 1.9378,
      "eval_samples_per_second": 4.128,
      "eval_steps_per_second": 0.516,
      "step": 70
    },
    {
      "epoch": 35.5,
      "learning_rate": 3.3947368421052636e-05,
      "loss": 0.4697,
      "step": 71
    },
    {
      "epoch": 36.0,
      "learning_rate": 3.368421052631579e-05,
      "loss": 0.5519,
      "step": 72
    },
    {
      "epoch": 36.5,
      "learning_rate": 3.342105263157895e-05,
      "loss": 0.4762,
      "step": 73
    },
    {
      "epoch": 37.0,
      "learning_rate": 3.3157894736842106e-05,
      "loss": 0.4986,
      "step": 74
    },
    {
      "epoch": 37.5,
      "learning_rate": 3.289473684210527e-05,
      "loss": 0.4565,
      "step": 75
    },
    {
      "epoch": 38.0,
      "learning_rate": 3.2631578947368426e-05,
      "loss": 0.4981,
      "step": 76
    },
    {
      "epoch": 38.5,
      "learning_rate": 3.236842105263158e-05,
      "loss": 0.4549,
      "step": 77
    },
    {
      "epoch": 39.0,
      "learning_rate": 3.210526315789474e-05,
      "loss": 0.4852,
      "step": 78
    },
    {
      "epoch": 39.5,
      "learning_rate": 3.1842105263157895e-05,
      "loss": 0.4397,
      "step": 79
    },
    {
      "epoch": 40.0,
      "learning_rate": 3.157894736842105e-05,
      "loss": 0.4477,
      "step": 80
    },
    {
      "epoch": 40.0,
      "eval_accuracy_safe": 0.05825708714564272,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.995995532095335,
      "eval_iou_safe": 0.05227315550052489,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9661213948715597,
      "eval_loss": 0.43396273255348206,
      "eval_mean_accuracy": 0.5271263096204889,
      "eval_mean_iou": 0.3394648501240282,
      "eval_overall_accuracy": 0.966184139251709,
      "eval_runtime": 1.8717,
      "eval_samples_per_second": 4.274,
      "eval_steps_per_second": 0.534,
      "step": 80
    }
  ],
  "max_steps": 200,
  "num_train_epochs": 100,
  "total_flos": 2.80821831303168e+16,
  "trial_name": null,
  "trial_params": null
}