{
  "best_metric": 0.7629496455192566,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/INTERNAL_BEST-safety-utcustom-train-SF-RGB-b5/checkpoint-80",
  "epoch": 8.0,
  "global_step": 80,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.1,
      "learning_rate": 1.0000000000000001e-07,
      "loss": 1.2416,
      "step": 1
    },
    {
      "epoch": 0.2,
      "learning_rate": 2.0000000000000002e-07,
      "loss": 1.2501,
      "step": 2
    },
    {
      "epoch": 0.3,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 1.2448,
      "step": 3
    },
    {
      "epoch": 0.4,
      "learning_rate": 4.0000000000000003e-07,
      "loss": 1.2497,
      "step": 4
    },
    {
      "epoch": 0.5,
      "learning_rate": 5.000000000000001e-07,
      "loss": 1.2596,
      "step": 5
    },
    {
      "epoch": 0.6,
      "learning_rate": 6.000000000000001e-07,
      "loss": 1.2563,
      "step": 6
    },
    {
      "epoch": 0.7,
      "learning_rate": 7.000000000000001e-07,
      "loss": 1.2528,
      "step": 7
    },
    {
      "epoch": 0.8,
      "learning_rate": 8.000000000000001e-07,
      "loss": 1.2488,
      "step": 8
    },
    {
      "epoch": 0.9,
      "learning_rate": 9e-07,
      "loss": 1.246,
      "step": 9
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.0000000000000002e-06,
      "loss": 1.2569,
      "step": 10
    },
    {
      "epoch": 1.1,
      "learning_rate": 1.1e-06,
      "loss": 1.2506,
      "step": 11
    },
    {
      "epoch": 1.2,
      "learning_rate": 1.2000000000000002e-06,
      "loss": 1.2516,
      "step": 12
    },
    {
      "epoch": 1.3,
      "learning_rate": 1.3e-06,
      "loss": 1.2334,
      "step": 13
    },
    {
      "epoch": 1.4,
      "learning_rate": 1.4000000000000001e-06,
      "loss": 1.2405,
      "step": 14
    },
    {
      "epoch": 1.5,
      "learning_rate": 1.5e-06,
      "loss": 1.2353,
      "step": 15
    },
    {
      "epoch": 1.6,
      "learning_rate": 1.6000000000000001e-06,
      "loss": 1.2381,
      "step": 16
    },
    {
      "epoch": 1.7,
      "learning_rate": 1.7000000000000002e-06,
      "loss": 1.2353,
      "step": 17
    },
    {
      "epoch": 1.8,
      "learning_rate": 1.8e-06,
      "loss": 1.2275,
      "step": 18
    },
    {
      "epoch": 1.9,
      "learning_rate": 1.9e-06,
      "loss": 1.2315,
      "step": 19
    },
    {
      "epoch": 2.0,
      "learning_rate": 2.0000000000000003e-06,
      "loss": 1.2199,
      "step": 20
    },
    {
      "epoch": 2.0,
      "eval_accuracy_safe": 0.6769557557603865,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.20277138320874238,
      "eval_iou_safe": 0.027914850351298256,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.20144342824057224,
      "eval_loss": 1.147387981414795,
      "eval_mean_accuracy": 0.4398635694845644,
      "eval_mean_iou": 0.0764527595306235,
      "eval_overall_accuracy": 0.2167780292567922,
      "eval_runtime": 9.4386,
      "eval_samples_per_second": 7.098,
      "eval_steps_per_second": 0.53,
      "step": 20
    },
    {
      "epoch": 2.1,
      "learning_rate": 2.1000000000000002e-06,
      "loss": 1.2169,
      "step": 21
    },
    {
      "epoch": 2.2,
      "learning_rate": 2.2e-06,
      "loss": 1.2233,
      "step": 22
    },
    {
      "epoch": 2.3,
      "learning_rate": 2.3e-06,
      "loss": 1.225,
      "step": 23
    },
    {
      "epoch": 2.4,
      "learning_rate": 2.4000000000000003e-06,
      "loss": 1.2193,
      "step": 24
    },
    {
      "epoch": 2.5,
      "learning_rate": 2.5e-06,
      "loss": 1.2153,
      "step": 25
    },
    {
      "epoch": 2.6,
      "learning_rate": 2.6e-06,
      "loss": 1.2085,
      "step": 26
    },
    {
      "epoch": 2.7,
      "learning_rate": 2.7e-06,
      "loss": 1.2061,
      "step": 27
    },
    {
      "epoch": 2.8,
      "learning_rate": 2.8000000000000003e-06,
      "loss": 1.2082,
      "step": 28
    },
    {
      "epoch": 2.9,
      "learning_rate": 2.9e-06,
      "loss": 1.208,
      "step": 29
    },
    {
      "epoch": 3.0,
      "learning_rate": 3e-06,
      "loss": 1.205,
      "step": 30
    },
    {
      "epoch": 3.1,
      "learning_rate": 3.1e-06,
      "loss": 1.1978,
      "step": 31
    },
    {
      "epoch": 3.2,
      "learning_rate": 3.2000000000000003e-06,
      "loss": 1.1951,
      "step": 32
    },
    {
      "epoch": 3.3,
      "learning_rate": 3.3e-06,
      "loss": 1.1905,
      "step": 33
    },
    {
      "epoch": 3.4,
      "learning_rate": 3.4000000000000005e-06,
      "loss": 1.1892,
      "step": 34
    },
    {
      "epoch": 3.5,
      "learning_rate": 3.5000000000000004e-06,
      "loss": 1.1726,
      "step": 35
    },
    {
      "epoch": 3.6,
      "learning_rate": 3.6e-06,
      "loss": 1.1638,
      "step": 36
    },
    {
      "epoch": 3.7,
      "learning_rate": 3.7e-06,
      "loss": 1.1697,
      "step": 37
    },
    {
      "epoch": 3.8,
      "learning_rate": 3.8e-06,
      "loss": 1.1557,
      "step": 38
    },
    {
      "epoch": 3.9,
      "learning_rate": 3.9e-06,
      "loss": 1.1568,
      "step": 39
    },
    {
      "epoch": 4.0,
      "learning_rate": 4.000000000000001e-06,
      "loss": 1.1542,
      "step": 40
    },
    {
      "epoch": 4.0,
      "eval_accuracy_safe": 0.7907563964672457,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.42571478791888173,
      "eval_iou_safe": 0.04364337256292872,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.4237482501114813,
      "eval_loss": 1.0615942478179932,
      "eval_mean_accuracy": 0.6082355921930638,
      "eval_mean_iou": 0.15579720755813667,
      "eval_overall_accuracy": 0.4364975317200618,
      "eval_runtime": 9.575,
      "eval_samples_per_second": 6.997,
      "eval_steps_per_second": 0.522,
      "step": 40
    },
    {
      "epoch": 4.1,
      "learning_rate": 4.1000000000000006e-06,
      "loss": 1.1516,
      "step": 41
    },
    {
      "epoch": 4.2,
      "learning_rate": 4.2000000000000004e-06,
      "loss": 1.1479,
      "step": 42
    },
    {
      "epoch": 4.3,
      "learning_rate": 4.2999999999999995e-06,
      "loss": 1.1341,
      "step": 43
    },
    {
      "epoch": 4.4,
      "learning_rate": 4.4e-06,
      "loss": 1.1384,
      "step": 44
    },
    {
      "epoch": 4.5,
      "learning_rate": 4.5e-06,
      "loss": 1.1202,
      "step": 45
    },
    {
      "epoch": 4.6,
      "learning_rate": 4.6e-06,
      "loss": 1.1091,
      "step": 46
    },
    {
      "epoch": 4.7,
      "learning_rate": 4.7e-06,
      "loss": 1.112,
      "step": 47
    },
    {
      "epoch": 4.8,
      "learning_rate": 4.800000000000001e-06,
      "loss": 1.1014,
      "step": 48
    },
    {
      "epoch": 4.9,
      "learning_rate": 4.9000000000000005e-06,
      "loss": 1.1065,
      "step": 49
    },
    {
      "epoch": 5.0,
      "learning_rate": 5e-06,
      "loss": 1.1028,
      "step": 50
    },
    {
      "epoch": 5.1,
      "learning_rate": 5.1e-06,
      "loss": 1.0823,
      "step": 51
    },
    {
      "epoch": 5.2,
      "learning_rate": 5.2e-06,
      "loss": 1.0721,
      "step": 52
    },
    {
      "epoch": 5.3,
      "learning_rate": 5.3e-06,
      "loss": 1.0809,
      "step": 53
    },
    {
      "epoch": 5.4,
      "learning_rate": 5.4e-06,
      "loss": 1.0711,
      "step": 54
    },
    {
      "epoch": 5.5,
      "learning_rate": 5.500000000000001e-06,
      "loss": 1.0566,
      "step": 55
    },
    {
      "epoch": 5.6,
      "learning_rate": 5.600000000000001e-06,
      "loss": 1.0401,
      "step": 56
    },
    {
      "epoch": 5.7,
      "learning_rate": 5.7000000000000005e-06,
      "loss": 1.073,
      "step": 57
    },
    {
      "epoch": 5.8,
      "learning_rate": 5.8e-06,
      "loss": 1.0281,
      "step": 58
    },
    {
      "epoch": 5.9,
      "learning_rate": 5.9e-06,
      "loss": 1.0327,
      "step": 59
    },
    {
      "epoch": 6.0,
      "learning_rate": 6e-06,
      "loss": 1.0324,
      "step": 60
    },
    {
      "epoch": 6.0,
      "eval_accuracy_safe": 0.6667148546073454,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.7104100559195431,
      "eval_iou_safe": 0.06715586533607736,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.7035684769007364,
      "eval_loss": 0.9139688611030579,
      "eval_mean_accuracy": 0.6885624552634442,
      "eval_mean_iou": 0.25690811407893793,
      "eval_overall_accuracy": 0.709119369734579,
      "eval_runtime": 9.3129,
      "eval_samples_per_second": 7.194,
      "eval_steps_per_second": 0.537,
      "step": 60
    },
    {
      "epoch": 6.1,
      "learning_rate": 6.1e-06,
      "loss": 0.996,
      "step": 61
    },
    {
      "epoch": 6.2,
      "learning_rate": 6.2e-06,
      "loss": 1.0074,
      "step": 62
    },
    {
      "epoch": 6.3,
      "learning_rate": 6.300000000000001e-06,
      "loss": 1.0368,
      "step": 63
    },
    {
      "epoch": 6.4,
      "learning_rate": 6.4000000000000006e-06,
      "loss": 0.981,
      "step": 64
    },
    {
      "epoch": 6.5,
      "learning_rate": 6.5000000000000004e-06,
      "loss": 0.9959,
      "step": 65
    },
    {
      "epoch": 6.6,
      "learning_rate": 6.6e-06,
      "loss": 0.9577,
      "step": 66
    },
    {
      "epoch": 6.7,
      "learning_rate": 6.700000000000001e-06,
      "loss": 0.9664,
      "step": 67
    },
    {
      "epoch": 6.8,
      "learning_rate": 6.800000000000001e-06,
      "loss": 0.9371,
      "step": 68
    },
    {
      "epoch": 6.9,
      "learning_rate": 6.900000000000001e-06,
      "loss": 0.9334,
      "step": 69
    },
    {
      "epoch": 7.0,
      "learning_rate": 7.000000000000001e-06,
      "loss": 0.9258,
      "step": 70
    },
    {
      "epoch": 7.1,
      "learning_rate": 7.1e-06,
      "loss": 0.9073,
      "step": 71
    },
    {
      "epoch": 7.2,
      "learning_rate": 7.2e-06,
      "loss": 1.004,
      "step": 72
    },
    {
      "epoch": 7.3,
      "learning_rate": 7.2999999999999996e-06,
      "loss": 0.9029,
      "step": 73
    },
    {
      "epoch": 7.4,
      "learning_rate": 7.4e-06,
      "loss": 0.8715,
      "step": 74
    },
    {
      "epoch": 7.5,
      "learning_rate": 7.5e-06,
      "loss": 0.8641,
      "step": 75
    },
    {
      "epoch": 7.6,
      "learning_rate": 7.6e-06,
      "loss": 0.8934,
      "step": 76
    },
    {
      "epoch": 7.7,
      "learning_rate": 7.7e-06,
      "loss": 0.8487,
      "step": 77
    },
    {
      "epoch": 7.8,
      "learning_rate": 7.8e-06,
      "loss": 0.8626,
      "step": 78
    },
    {
      "epoch": 7.9,
      "learning_rate": 7.9e-06,
      "loss": 0.8309,
      "step": 79
    },
    {
      "epoch": 8.0,
      "learning_rate": 8.000000000000001e-06,
      "loss": 0.8058,
      "step": 80
    },
    {
      "epoch": 8.0,
      "eval_accuracy_safe": 0.5359944641693748,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8199187015242027,
      "eval_iou_safe": 0.08228570650745816,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8086088911065791,
      "eval_loss": 0.7629496455192566,
      "eval_mean_accuracy": 0.6779565828467887,
      "eval_mean_iou": 0.29696486587134574,
      "eval_overall_accuracy": 0.8115320348027927,
      "eval_runtime": 10.0198,
      "eval_samples_per_second": 6.687,
      "eval_steps_per_second": 0.499,
      "step": 80
    }
  ],
  "max_steps": 20000,
  "num_train_epochs": 2000,
  "total_flos": 4.9178035003745894e+17,
  "trial_name": null,
  "trial_params": null
}