{
  "best_metric": 0.41405442357063293,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF-RGB-b5/checkpoint-440",
  "epoch": 40.0,
  "global_step": 440,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.09,
      "learning_rate": 1.764705882352941e-07,
      "loss": 1.2293,
      "step": 1
    },
    {
      "epoch": 0.18,
      "learning_rate": 3.529411764705882e-07,
      "loss": 1.2349,
      "step": 2
    },
    {
      "epoch": 0.27,
      "learning_rate": 5.294117647058824e-07,
      "loss": 1.2357,
      "step": 3
    },
    {
      "epoch": 0.36,
      "learning_rate": 7.058823529411765e-07,
      "loss": 1.2242,
      "step": 4
    },
    {
      "epoch": 0.45,
      "learning_rate": 8.823529411764706e-07,
      "loss": 1.2247,
      "step": 5
    },
    {
      "epoch": 0.55,
      "learning_rate": 1.0588235294117648e-06,
      "loss": 1.227,
      "step": 6
    },
    {
      "epoch": 0.64,
      "learning_rate": 1.2352941176470588e-06,
      "loss": 1.2292,
      "step": 7
    },
    {
      "epoch": 0.73,
      "learning_rate": 1.411764705882353e-06,
      "loss": 1.2257,
      "step": 8
    },
    {
      "epoch": 0.82,
      "learning_rate": 1.5882352941176472e-06,
      "loss": 1.2173,
      "step": 9
    },
    {
      "epoch": 0.91,
      "learning_rate": 1.7647058823529412e-06,
      "loss": 1.2239,
      "step": 10
    },
    {
      "epoch": 0.91,
      "eval_accuracy_safe": 0.39920432072351303,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.29510386893492613,
      "eval_iou_safe": 0.031418797454432365,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.29392505413655395,
      "eval_loss": 1.1102662086486816,
      "eval_mean_accuracy": 0.34715409482921955,
      "eval_mean_iou": 0.10844795053032878,
      "eval_overall_accuracy": 0.2981788293639226,
      "eval_runtime": 9.3061,
      "eval_samples_per_second": 7.2,
      "eval_steps_per_second": 0.537,
      "step": 10
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.9411764705882353e-06,
      "loss": 1.221,
      "step": 11
    },
    {
      "epoch": 1.09,
      "learning_rate": 2.1176470588235296e-06,
      "loss": 1.2149,
      "step": 12
    },
    {
      "epoch": 1.18,
      "learning_rate": 2.2941176470588234e-06,
      "loss": 1.2197,
      "step": 13
    },
    {
      "epoch": 1.27,
      "learning_rate": 2.4705882352941177e-06,
      "loss": 1.2064,
      "step": 14
    },
    {
      "epoch": 1.36,
      "learning_rate": 2.647058823529412e-06,
      "loss": 1.2042,
      "step": 15
    },
    {
      "epoch": 1.45,
      "learning_rate": 2.823529411764706e-06,
      "loss": 1.2046,
      "step": 16
    },
    {
      "epoch": 1.55,
      "learning_rate": 3e-06,
      "loss": 1.2054,
      "step": 17
    },
    {
      "epoch": 1.64,
      "learning_rate": 2.990415335463259e-06,
      "loss": 1.2002,
      "step": 18
    },
    {
      "epoch": 1.73,
      "learning_rate": 2.9808306709265177e-06,
      "loss": 1.1982,
      "step": 19
    },
    {
      "epoch": 1.82,
      "learning_rate": 2.9712460063897764e-06,
      "loss": 1.1948,
      "step": 20
    },
    {
      "epoch": 1.82,
      "eval_accuracy_safe": 0.5218599774094934,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.3705393407485172,
      "eval_iou_safe": 0.0439841259320199,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.36893471456827454,
      "eval_loss": 1.0963102579116821,
      "eval_mean_accuracy": 0.4461996590790053,
      "eval_mean_iou": 0.13763961350009815,
      "eval_overall_accuracy": 0.37500910972481344,
      "eval_runtime": 8.3942,
      "eval_samples_per_second": 7.982,
      "eval_steps_per_second": 0.596,
      "step": 20
    },
    {
      "epoch": 1.91,
      "learning_rate": 2.9616613418530352e-06,
      "loss": 1.1776,
      "step": 21
    },
    {
      "epoch": 2.0,
      "learning_rate": 2.952076677316294e-06,
      "loss": 1.1936,
      "step": 22
    },
    {
      "epoch": 2.09,
      "learning_rate": 2.942492012779553e-06,
      "loss": 1.1664,
      "step": 23
    },
    {
      "epoch": 2.18,
      "learning_rate": 2.9329073482428116e-06,
      "loss": 1.1809,
      "step": 24
    },
    {
      "epoch": 2.27,
      "learning_rate": 2.9233226837060704e-06,
      "loss": 1.1731,
      "step": 25
    },
    {
      "epoch": 2.36,
      "learning_rate": 2.913738019169329e-06,
      "loss": 1.1794,
      "step": 26
    },
    {
      "epoch": 2.45,
      "learning_rate": 2.904153354632588e-06,
      "loss": 1.1684,
      "step": 27
    },
    {
      "epoch": 2.55,
      "learning_rate": 2.8945686900958468e-06,
      "loss": 1.1594,
      "step": 28
    },
    {
      "epoch": 2.64,
      "learning_rate": 2.884984025559105e-06,
      "loss": 1.146,
      "step": 29
    },
    {
      "epoch": 2.73,
      "learning_rate": 2.8753993610223644e-06,
      "loss": 1.1661,
      "step": 30
    },
    {
      "epoch": 2.73,
      "eval_accuracy_safe": 0.5863200218966003,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.49882181393718666,
      "eval_iou_safe": 0.06472096392925002,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.49613792944888824,
      "eval_loss": 1.0516184568405151,
      "eval_mean_accuracy": 0.5425709179168935,
      "eval_mean_iou": 0.1869529644593794,
      "eval_overall_accuracy": 0.5014063707038537,
      "eval_runtime": 9.3615,
      "eval_samples_per_second": 7.157,
      "eval_steps_per_second": 0.534,
      "step": 30
    },
    {
      "epoch": 2.82,
      "learning_rate": 2.865814696485623e-06,
      "loss": 1.153,
      "step": 31
    },
    {
      "epoch": 2.91,
      "learning_rate": 2.856230031948882e-06,
      "loss": 1.149,
      "step": 32
    },
    {
      "epoch": 3.0,
      "learning_rate": 2.8466453674121407e-06,
      "loss": 1.164,
      "step": 33
    },
    {
      "epoch": 3.09,
      "learning_rate": 2.8370607028753995e-06,
      "loss": 1.1479,
      "step": 34
    },
    {
      "epoch": 3.18,
      "learning_rate": 2.8274760383386583e-06,
      "loss": 1.1428,
      "step": 35
    },
    {
      "epoch": 3.27,
      "learning_rate": 2.817891373801917e-06,
      "loss": 1.1398,
      "step": 36
    },
    {
      "epoch": 3.36,
      "learning_rate": 2.8083067092651755e-06,
      "loss": 1.1415,
      "step": 37
    },
    {
      "epoch": 3.45,
      "learning_rate": 2.7987220447284347e-06,
      "loss": 1.1146,
      "step": 38
    },
    {
      "epoch": 3.55,
      "learning_rate": 2.7891373801916935e-06,
      "loss": 1.1234,
      "step": 39
    },
    {
      "epoch": 3.64,
      "learning_rate": 2.7795527156549523e-06,
      "loss": 1.1112,
      "step": 40
    },
    {
      "epoch": 3.64,
      "eval_accuracy_safe": 0.5458517893146133,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.5794390867479824,
      "eval_iou_safe": 0.0899865428032596,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.5753581085678546,
      "eval_loss": 1.0047743320465088,
      "eval_mean_accuracy": 0.5626454380312979,
      "eval_mean_iou": 0.22178155045703807,
      "eval_overall_accuracy": 0.5784469718363747,
      "eval_runtime": 9.38,
      "eval_samples_per_second": 7.143,
      "eval_steps_per_second": 0.533,
      "step": 40
    },
    {
      "epoch": 3.73,
      "learning_rate": 2.769968051118211e-06,
      "loss": 1.1212,
      "step": 41
    },
    {
      "epoch": 3.82,
      "learning_rate": 2.7603833865814694e-06,
      "loss": 1.1117,
      "step": 42
    },
    {
      "epoch": 3.91,
      "learning_rate": 2.7507987220447286e-06,
      "loss": 1.1051,
      "step": 43
    },
    {
      "epoch": 4.0,
      "learning_rate": 2.7412140575079874e-06,
      "loss": 1.1382,
      "step": 44
    },
    {
      "epoch": 4.09,
      "learning_rate": 2.731629392971246e-06,
      "loss": 1.1075,
      "step": 45
    },
    {
      "epoch": 4.18,
      "learning_rate": 2.722044728434505e-06,
      "loss": 1.1096,
      "step": 46
    },
    {
      "epoch": 4.27,
      "learning_rate": 2.7124600638977634e-06,
      "loss": 1.0974,
      "step": 47
    },
    {
      "epoch": 4.36,
      "learning_rate": 2.7028753993610226e-06,
      "loss": 1.0851,
      "step": 48
    },
    {
      "epoch": 4.45,
      "learning_rate": 2.6932907348242814e-06,
      "loss": 1.0851,
      "step": 49
    },
    {
      "epoch": 4.55,
      "learning_rate": 2.6837060702875398e-06,
      "loss": 1.0907,
      "step": 50
    },
    {
      "epoch": 4.55,
      "eval_accuracy_safe": 0.5993018531154467,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.6366664738420048,
      "eval_iou_safe": 0.10939424133801141,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.6320960572783577,
      "eval_loss": 0.9689871668815613,
      "eval_mean_accuracy": 0.6179841634787258,
      "eval_mean_iou": 0.24716343287212306,
      "eval_overall_accuracy": 0.6355627828569554,
      "eval_runtime": 9.8119,
      "eval_samples_per_second": 6.828,
      "eval_steps_per_second": 0.51,
      "step": 50
    },
    {
      "epoch": 4.64,
      "learning_rate": 2.674121405750799e-06,
      "loss": 1.0789,
      "step": 51
    },
    {
      "epoch": 4.73,
      "learning_rate": 2.6645367412140573e-06,
      "loss": 1.0566,
      "step": 52
    },
    {
      "epoch": 4.82,
      "learning_rate": 2.654952076677316e-06,
      "loss": 1.0718,
      "step": 53
    },
    {
      "epoch": 4.91,
      "learning_rate": 2.6453674121405753e-06,
      "loss": 1.0808,
      "step": 54
    },
    {
      "epoch": 5.0,
      "learning_rate": 2.6357827476038337e-06,
      "loss": 1.0808,
      "step": 55
    },
    {
      "epoch": 5.09,
      "learning_rate": 2.626198083067093e-06,
      "loss": 1.0625,
      "step": 56
    },
    {
      "epoch": 5.18,
      "learning_rate": 2.6166134185303517e-06,
      "loss": 1.0508,
      "step": 57
    },
    {
      "epoch": 5.27,
      "learning_rate": 2.60702875399361e-06,
      "loss": 1.0521,
      "step": 58
    },
    {
      "epoch": 5.36,
      "learning_rate": 2.5974440894568693e-06,
      "loss": 1.052,
      "step": 59
    },
    {
      "epoch": 5.45,
      "learning_rate": 2.5878594249201277e-06,
      "loss": 1.047,
      "step": 60
    },
    {
      "epoch": 5.45,
      "eval_accuracy_safe": 0.6691570194409427,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.669918812994849,
      "eval_iou_safe": 0.11592410370525706,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.6655783206367077,
      "eval_loss": 0.9437160491943359,
      "eval_mean_accuracy": 0.6695379162178958,
      "eval_mean_iou": 0.2605008081139883,
      "eval_overall_accuracy": 0.6698963108347423,
      "eval_runtime": 9.3506,
      "eval_samples_per_second": 7.165,
      "eval_steps_per_second": 0.535,
      "step": 60
    },
    {
      "epoch": 5.55,
      "learning_rate": 2.5782747603833865e-06,
      "loss": 1.0456,
      "step": 61
    },
    {
      "epoch": 5.64,
      "learning_rate": 2.5686900958466457e-06,
      "loss": 1.0296,
      "step": 62
    },
    {
      "epoch": 5.73,
      "learning_rate": 2.559105431309904e-06,
      "loss": 1.0334,
      "step": 63
    },
    {
      "epoch": 5.82,
      "learning_rate": 2.5495207667731633e-06,
      "loss": 1.0525,
      "step": 64
    },
    {
      "epoch": 5.91,
      "learning_rate": 2.5399361022364216e-06,
      "loss": 1.0397,
      "step": 65
    },
    {
      "epoch": 6.0,
      "learning_rate": 2.5303514376996804e-06,
      "loss": 1.0575,
      "step": 66
    },
    {
      "epoch": 6.09,
      "learning_rate": 2.5207667731629396e-06,
      "loss": 1.0245,
      "step": 67
    },
    {
      "epoch": 6.18,
      "learning_rate": 2.511182108626198e-06,
      "loss": 1.0284,
      "step": 68
    },
    {
      "epoch": 6.27,
      "learning_rate": 2.501597444089457e-06,
      "loss": 1.0127,
      "step": 69
    },
    {
      "epoch": 6.36,
      "learning_rate": 2.4920127795527156e-06,
      "loss": 1.0112,
      "step": 70
    },
    {
      "epoch": 6.36,
      "eval_accuracy_safe": 0.6673046750012529,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.718865515124044,
      "eval_iou_safe": 0.13494764234639625,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.7136723036690169,
      "eval_loss": 0.9084014892578125,
      "eval_mean_accuracy": 0.6930850950626484,
      "eval_mean_iou": 0.28287331533847104,
      "eval_overall_accuracy": 0.7173424905805446,
      "eval_runtime": 9.0845,
      "eval_samples_per_second": 7.375,
      "eval_steps_per_second": 0.55,
      "step": 70
    },
    {
      "epoch": 6.45,
      "learning_rate": 2.4824281150159744e-06,
      "loss": 1.0107,
      "step": 71
    },
    {
      "epoch": 6.55,
      "learning_rate": 2.4728434504792336e-06,
      "loss": 1.0122,
      "step": 72
    },
    {
      "epoch": 6.64,
      "learning_rate": 2.463258785942492e-06,
      "loss": 1.0087,
      "step": 73
    },
    {
      "epoch": 6.73,
      "learning_rate": 2.4536741214057507e-06,
      "loss": 0.991,
      "step": 74
    },
    {
      "epoch": 6.82,
      "learning_rate": 2.44408945686901e-06,
      "loss": 1.0042,
      "step": 75
    },
    {
      "epoch": 6.91,
      "learning_rate": 2.4345047923322683e-06,
      "loss": 0.9788,
      "step": 76
    },
    {
      "epoch": 7.0,
      "learning_rate": 2.424920127795527e-06,
      "loss": 1.0371,
      "step": 77
    },
    {
      "epoch": 7.09,
      "learning_rate": 2.415335463258786e-06,
      "loss": 0.9884,
      "step": 78
    },
    {
      "epoch": 7.18,
      "learning_rate": 2.4057507987220447e-06,
      "loss": 0.9759,
      "step": 79
    },
    {
      "epoch": 7.27,
      "learning_rate": 2.396166134185304e-06,
      "loss": 0.9925,
      "step": 80
    },
    {
      "epoch": 7.27,
      "eval_accuracy_safe": 0.6842282797676185,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.7664939888573942,
      "eval_iou_safe": 0.14521835177647324,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.7605469001045981,
      "eval_loss": 0.8647122979164124,
      "eval_mean_accuracy": 0.7253611343125064,
      "eval_mean_iou": 0.3019217506270238,
      "eval_overall_accuracy": 0.7640639917174382,
      "eval_runtime": 9.9787,
      "eval_samples_per_second": 6.714,
      "eval_steps_per_second": 0.501,
      "step": 80
    },
    {
      "epoch": 7.36,
      "learning_rate": 2.3865814696485623e-06,
      "loss": 0.9784,
      "step": 81
    },
    {
      "epoch": 7.45,
      "learning_rate": 2.376996805111821e-06,
      "loss": 0.9806,
      "step": 82
    },
    {
      "epoch": 7.55,
      "learning_rate": 2.36741214057508e-06,
      "loss": 0.973,
      "step": 83
    },
    {
      "epoch": 7.64,
      "learning_rate": 2.3578274760383387e-06,
      "loss": 0.9551,
      "step": 84
    },
    {
      "epoch": 7.73,
      "learning_rate": 2.3482428115015974e-06,
      "loss": 0.9484,
      "step": 85
    },
    {
      "epoch": 7.82,
      "learning_rate": 2.3386581469648562e-06,
      "loss": 0.939,
      "step": 86
    },
    {
      "epoch": 7.91,
      "learning_rate": 2.329073482428115e-06,
      "loss": 0.9533,
      "step": 87
    },
    {
      "epoch": 8.0,
      "learning_rate": 2.319488817891374e-06,
      "loss": 0.9533,
      "step": 88
    },
    {
      "epoch": 8.09,
      "learning_rate": 2.3099041533546326e-06,
      "loss": 0.9469,
      "step": 89
    },
    {
      "epoch": 8.18,
      "learning_rate": 2.3003194888178914e-06,
      "loss": 0.9395,
      "step": 90
    },
    {
      "epoch": 8.18,
      "eval_accuracy_safe": 0.6817822598987667,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.7920860651953089,
      "eval_iou_safe": 0.16203325137990066,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.785611270080723,
      "eval_loss": 0.8319001197814941,
      "eval_mean_accuracy": 0.7369341625470378,
      "eval_mean_iou": 0.3158815071535412,
      "eval_overall_accuracy": 0.788827867650274,
      "eval_runtime": 9.682,
      "eval_samples_per_second": 6.92,
      "eval_steps_per_second": 0.516,
      "step": 90
    },
    {
      "epoch": 8.27,
      "learning_rate": 2.29073482428115e-06,
      "loss": 0.9422,
      "step": 91
    },
    {
      "epoch": 8.36,
      "learning_rate": 2.281150159744409e-06,
      "loss": 0.9012,
      "step": 92
    },
    {
      "epoch": 8.45,
      "learning_rate": 2.2715654952076678e-06,
      "loss": 0.937,
      "step": 93
    },
    {
      "epoch": 8.55,
      "learning_rate": 2.2619808306709266e-06,
      "loss": 0.9122,
      "step": 94
    },
    {
      "epoch": 8.64,
      "learning_rate": 2.2523961661341854e-06,
      "loss": 0.9272,
      "step": 95
    },
    {
      "epoch": 8.73,
      "learning_rate": 2.242811501597444e-06,
      "loss": 0.9122,
      "step": 96
    },
    {
      "epoch": 8.82,
      "learning_rate": 2.233226837060703e-06,
      "loss": 0.9261,
      "step": 97
    },
    {
      "epoch": 8.91,
      "learning_rate": 2.2236421725239617e-06,
      "loss": 0.9004,
      "step": 98
    },
    {
      "epoch": 9.0,
      "learning_rate": 2.2140575079872205e-06,
      "loss": 0.8981,
      "step": 99
    },
    {
      "epoch": 9.09,
      "learning_rate": 2.2044728434504793e-06,
      "loss": 0.8902,
      "step": 100
    },
    {
      "epoch": 9.09,
      "eval_accuracy_safe": 0.6805698513112902,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8141590718977455,
      "eval_iou_safe": 0.1769983948474704,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8072069422555583,
      "eval_loss": 0.8013909459114075,
      "eval_mean_accuracy": 0.7473644616045179,
      "eval_mean_iou": 0.32806844570100957,
      "eval_overall_accuracy": 0.8102130605213678,
      "eval_runtime": 9.6492,
      "eval_samples_per_second": 6.944,
      "eval_steps_per_second": 0.518,
      "step": 100
    },
    {
      "epoch": 9.18,
      "learning_rate": 2.194888178913738e-06,
      "loss": 0.9085,
      "step": 101
    },
    {
      "epoch": 9.27,
      "learning_rate": 2.185303514376997e-06,
      "loss": 0.8976,
      "step": 102
    },
    {
      "epoch": 9.36,
      "learning_rate": 2.1757188498402557e-06,
      "loss": 0.8909,
      "step": 103
    },
    {
      "epoch": 9.45,
      "learning_rate": 2.1661341853035145e-06,
      "loss": 0.8878,
      "step": 104
    },
    {
      "epoch": 9.55,
      "learning_rate": 2.1565495207667733e-06,
      "loss": 0.8909,
      "step": 105
    },
    {
      "epoch": 9.64,
      "learning_rate": 2.146964856230032e-06,
      "loss": 0.8529,
      "step": 106
    },
    {
      "epoch": 9.73,
      "learning_rate": 2.137380191693291e-06,
      "loss": 0.9097,
      "step": 107
    },
    {
      "epoch": 9.82,
      "learning_rate": 2.1277955271565496e-06,
      "loss": 0.8778,
      "step": 108
    },
    {
      "epoch": 9.91,
      "learning_rate": 2.1182108626198084e-06,
      "loss": 0.8819,
      "step": 109
    },
    {
      "epoch": 10.0,
      "learning_rate": 2.1086261980830672e-06,
      "loss": 0.9057,
      "step": 110
    },
    {
      "epoch": 10.0,
      "eval_accuracy_safe": 0.6984244470915687,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8178638868312451,
      "eval_iou_safe": 0.17330681673606707,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8109252159162895,
      "eval_loss": 0.7867299914360046,
      "eval_mean_accuracy": 0.7581441669614069,
      "eval_mean_iou": 0.3280773442174522,
      "eval_overall_accuracy": 0.814335837293027,
      "eval_runtime": 9.8362,
      "eval_samples_per_second": 6.812,
      "eval_steps_per_second": 0.508,
      "step": 110
    },
    {
      "epoch": 10.09,
      "learning_rate": 2.099041533546326e-06,
      "loss": 0.8509,
      "step": 111
    },
    {
      "epoch": 10.18,
      "learning_rate": 2.089456869009585e-06,
      "loss": 0.8746,
      "step": 112
    },
    {
      "epoch": 10.27,
      "learning_rate": 2.0798722044728436e-06,
      "loss": 0.8532,
      "step": 113
    },
    {
      "epoch": 10.36,
      "learning_rate": 2.0702875399361024e-06,
      "loss": 0.8509,
      "step": 114
    },
    {
      "epoch": 10.45,
      "learning_rate": 2.060702875399361e-06,
      "loss": 0.8559,
      "step": 115
    },
    {
      "epoch": 10.55,
      "learning_rate": 2.05111821086262e-06,
      "loss": 0.8435,
      "step": 116
    },
    {
      "epoch": 10.64,
      "learning_rate": 2.0415335463258783e-06,
      "loss": 0.8465,
      "step": 117
    },
    {
      "epoch": 10.73,
      "learning_rate": 2.0319488817891376e-06,
      "loss": 0.8345,
      "step": 118
    },
    {
      "epoch": 10.82,
      "learning_rate": 2.0223642172523963e-06,
      "loss": 0.8403,
      "step": 119
    },
    {
      "epoch": 10.91,
      "learning_rate": 2.012779552715655e-06,
      "loss": 0.8321,
      "step": 120
    },
    {
      "epoch": 10.91,
      "eval_accuracy_safe": 0.6743748096576343,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8493713583566551,
      "eval_iou_safe": 0.18617264978095113,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8413265356526651,
      "eval_loss": 0.7439588308334351,
      "eval_mean_accuracy": 0.7618730840071447,
      "eval_mean_iou": 0.3424997284778721,
      "eval_overall_accuracy": 0.8442022409012069,
      "eval_runtime": 9.4999,
      "eval_samples_per_second": 7.053,
      "eval_steps_per_second": 0.526,
      "step": 120
    },
    {
      "epoch": 11.0,
      "learning_rate": 2.003194888178914e-06,
      "loss": 0.8343,
      "step": 121
    },
    {
      "epoch": 11.09,
      "learning_rate": 1.9936102236421723e-06,
      "loss": 0.8413,
      "step": 122
    },
    {
      "epoch": 11.18,
      "learning_rate": 1.9840255591054315e-06,
      "loss": 0.8037,
      "step": 123
    },
    {
      "epoch": 11.27,
      "learning_rate": 1.9744408945686903e-06,
      "loss": 0.8284,
      "step": 124
    },
    {
      "epoch": 11.36,
      "learning_rate": 1.9648562300319487e-06,
      "loss": 0.8023,
      "step": 125
    },
    {
      "epoch": 11.45,
      "learning_rate": 1.955271565495208e-06,
      "loss": 0.8176,
      "step": 126
    },
    {
      "epoch": 11.55,
      "learning_rate": 1.9456869009584667e-06,
      "loss": 0.8078,
      "step": 127
    },
    {
      "epoch": 11.64,
      "learning_rate": 1.936102236421725e-06,
      "loss": 0.8074,
      "step": 128
    },
    {
      "epoch": 11.73,
      "learning_rate": 1.9265175718849843e-06,
      "loss": 0.8139,
      "step": 129
    },
    {
      "epoch": 11.82,
      "learning_rate": 1.9169329073482426e-06,
      "loss": 0.8152,
      "step": 130
    },
    {
      "epoch": 11.82,
      "eval_accuracy_safe": 0.6688004286799203,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8589712690862681,
      "eval_iou_safe": 0.20063653201077156,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8506617582762998,
      "eval_loss": 0.7270171046257019,
      "eval_mean_accuracy": 0.7638858488830942,
      "eval_mean_iou": 0.3504327634290238,
      "eval_overall_accuracy": 0.8533539273845616,
      "eval_runtime": 11.0671,
      "eval_samples_per_second": 6.054,
      "eval_steps_per_second": 0.452,
      "step": 130
    },
    {
      "epoch": 11.91,
      "learning_rate": 1.9073482428115018e-06,
      "loss": 0.8102,
      "step": 131
    },
    {
      "epoch": 12.0,
      "learning_rate": 1.8977635782747604e-06,
      "loss": 0.8192,
      "step": 132
    },
    {
      "epoch": 12.09,
      "learning_rate": 1.888178913738019e-06,
      "loss": 0.8246,
      "step": 133
    },
    {
      "epoch": 12.18,
      "learning_rate": 1.878594249201278e-06,
      "loss": 0.7864,
      "step": 134
    },
    {
      "epoch": 12.27,
      "learning_rate": 1.8690095846645368e-06,
      "loss": 0.7718,
      "step": 135
    },
    {
      "epoch": 12.36,
      "learning_rate": 1.8594249201277954e-06,
      "loss": 0.8074,
      "step": 136
    },
    {
      "epoch": 12.45,
      "learning_rate": 1.8498402555910544e-06,
      "loss": 0.7569,
      "step": 137
    },
    {
      "epoch": 12.55,
      "learning_rate": 1.8402555910543132e-06,
      "loss": 0.7989,
      "step": 138
    },
    {
      "epoch": 12.64,
      "learning_rate": 1.8306709265175722e-06,
      "loss": 0.7987,
      "step": 139
    },
    {
      "epoch": 12.73,
      "learning_rate": 1.8210862619808307e-06,
      "loss": 0.7929,
      "step": 140
    },
    {
      "epoch": 12.73,
      "eval_accuracy_safe": 0.6659920354971646,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8656634973410731,
      "eval_iou_safe": 0.20852726465106847,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8572303764577973,
      "eval_loss": 0.7045122981071472,
      "eval_mean_accuracy": 0.7658277664191189,
      "eval_mean_iou": 0.3552525470362886,
      "eval_overall_accuracy": 0.8597655225155959,
      "eval_runtime": 9.4502,
      "eval_samples_per_second": 7.09,
      "eval_steps_per_second": 0.529,
      "step": 140
    },
    {
      "epoch": 12.82,
      "learning_rate": 1.8115015974440893e-06,
      "loss": 0.7703,
      "step": 141
    },
    {
      "epoch": 12.91,
      "learning_rate": 1.8019169329073483e-06,
      "loss": 0.768,
      "step": 142
    },
    {
      "epoch": 13.0,
      "learning_rate": 1.7923322683706071e-06,
      "loss": 0.8397,
      "step": 143
    },
    {
      "epoch": 13.09,
      "learning_rate": 1.7827476038338657e-06,
      "loss": 0.775,
      "step": 144
    },
    {
      "epoch": 13.18,
      "learning_rate": 1.7731629392971247e-06,
      "loss": 0.7718,
      "step": 145
    },
    {
      "epoch": 13.27,
      "learning_rate": 1.7635782747603833e-06,
      "loss": 0.7544,
      "step": 146
    },
    {
      "epoch": 13.36,
      "learning_rate": 1.7539936102236423e-06,
      "loss": 0.7585,
      "step": 147
    },
    {
      "epoch": 13.45,
      "learning_rate": 1.744408945686901e-06,
      "loss": 0.742,
      "step": 148
    },
    {
      "epoch": 13.55,
      "learning_rate": 1.7348242811501597e-06,
      "loss": 0.7629,
      "step": 149
    },
    {
      "epoch": 13.64,
      "learning_rate": 1.7252396166134187e-06,
      "loss": 0.7568,
      "step": 150
    },
    {
      "epoch": 13.64,
      "eval_accuracy_safe": 0.657079193989229,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8838084544735693,
      "eval_iou_safe": 0.21846087789622765,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.874847055157359,
      "eval_loss": 0.6744459867477417,
      "eval_mean_accuracy": 0.7704438242313991,
      "eval_mean_iou": 0.3644359776845289,
      "eval_overall_accuracy": 0.8771112356612931,
      "eval_runtime": 10.8117,
      "eval_samples_per_second": 6.197,
      "eval_steps_per_second": 0.462,
      "step": 150
    },
    {
      "epoch": 13.73,
      "learning_rate": 1.7156549520766772e-06,
      "loss": 0.7311,
      "step": 151
    },
    {
      "epoch": 13.82,
      "learning_rate": 1.706070287539936e-06,
      "loss": 0.7352,
      "step": 152
    },
    {
      "epoch": 13.91,
      "learning_rate": 1.696485623003195e-06,
      "loss": 0.7661,
      "step": 153
    },
    {
      "epoch": 14.0,
      "learning_rate": 1.6869009584664536e-06,
      "loss": 0.7335,
      "step": 154
    },
    {
      "epoch": 14.09,
      "learning_rate": 1.6773162939297126e-06,
      "loss": 0.7491,
      "step": 155
    },
    {
      "epoch": 14.18,
      "learning_rate": 1.6677316293929714e-06,
      "loss": 0.715,
      "step": 156
    },
    {
      "epoch": 14.27,
      "learning_rate": 1.65814696485623e-06,
      "loss": 0.7349,
      "step": 157
    },
    {
      "epoch": 14.36,
      "learning_rate": 1.648562300319489e-06,
      "loss": 0.7127,
      "step": 158
    },
    {
      "epoch": 14.45,
      "learning_rate": 1.6389776357827476e-06,
      "loss": 0.718,
      "step": 159
    },
    {
      "epoch": 14.55,
      "learning_rate": 1.6293929712460064e-06,
      "loss": 0.7085,
      "step": 160
    },
    {
      "epoch": 14.55,
      "eval_accuracy_safe": 0.6519423595128777,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8934174588611713,
      "eval_iou_safe": 0.2259934305828019,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8842020655315154,
      "eval_loss": 0.65559983253479,
      "eval_mean_accuracy": 0.7726799091870244,
      "eval_mean_iou": 0.3700651653714391,
      "eval_overall_accuracy": 0.8862846716126399,
      "eval_runtime": 9.325,
      "eval_samples_per_second": 7.185,
      "eval_steps_per_second": 0.536,
      "step": 160
    },
    {
      "epoch": 14.64,
      "learning_rate": 1.6198083067092654e-06,
      "loss": 0.753,
      "step": 161
    },
    {
      "epoch": 14.73,
      "learning_rate": 1.610223642172524e-06,
      "loss": 0.7376,
      "step": 162
    },
    {
      "epoch": 14.82,
      "learning_rate": 1.600638977635783e-06,
      "loss": 0.7256,
      "step": 163
    },
    {
      "epoch": 14.91,
      "learning_rate": 1.5910543130990415e-06,
      "loss": 0.7132,
      "step": 164
    },
    {
      "epoch": 15.0,
      "learning_rate": 1.5814696485623003e-06,
      "loss": 0.722,
      "step": 165
    },
    {
      "epoch": 15.09,
      "learning_rate": 1.5718849840255593e-06,
      "loss": 0.7015,
      "step": 166
    },
    {
      "epoch": 15.18,
      "learning_rate": 1.562300319488818e-06,
      "loss": 0.7176,
      "step": 167
    },
    {
      "epoch": 15.27,
      "learning_rate": 1.5527156549520765e-06,
      "loss": 0.7147,
      "step": 168
    },
    {
      "epoch": 15.36,
      "learning_rate": 1.5431309904153355e-06,
      "loss": 0.7217,
      "step": 169
    },
    {
      "epoch": 15.45,
      "learning_rate": 1.5335463258785943e-06,
      "loss": 0.7147,
      "step": 170
    },
    {
      "epoch": 15.45,
      "eval_accuracy_safe": 0.6561000150346374,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8963761831582403,
      "eval_iou_safe": 0.22827906452703248,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8872234626234047,
      "eval_loss": 0.6508952975273132,
      "eval_mean_accuracy": 0.7762380990964388,
      "eval_mean_iou": 0.3718341757168124,
      "eval_overall_accuracy": 0.889278810415695,
      "eval_runtime": 10.5537,
      "eval_samples_per_second": 6.348,
      "eval_steps_per_second": 0.474,
      "step": 170
    },
    {
      "epoch": 15.55,
      "learning_rate": 1.5239616613418533e-06,
      "loss": 0.7212,
      "step": 171
    },
    {
      "epoch": 15.64,
      "learning_rate": 1.5143769968051119e-06,
      "loss": 0.7196,
      "step": 172
    },
    {
      "epoch": 15.73,
      "learning_rate": 1.5047923322683706e-06,
      "loss": 0.6797,
      "step": 173
    },
    {
      "epoch": 15.82,
      "learning_rate": 1.4952076677316294e-06,
      "loss": 0.6626,
      "step": 174
    },
    {
      "epoch": 15.91,
      "learning_rate": 1.4856230031948882e-06,
      "loss": 0.6783,
      "step": 175
    },
    {
      "epoch": 16.0,
      "learning_rate": 1.476038338658147e-06,
      "loss": 0.7436,
      "step": 176
    },
    {
      "epoch": 16.09,
      "learning_rate": 1.4664536741214058e-06,
      "loss": 0.7123,
      "step": 177
    },
    {
      "epoch": 16.18,
      "learning_rate": 1.4568690095846646e-06,
      "loss": 0.7241,
      "step": 178
    },
    {
      "epoch": 16.27,
      "learning_rate": 1.4472843450479234e-06,
      "loss": 0.6914,
      "step": 179
    },
    {
      "epoch": 16.36,
      "learning_rate": 1.4376996805111822e-06,
      "loss": 0.6991,
      "step": 180
    },
    {
      "epoch": 16.36,
      "eval_accuracy_safe": 0.6619847263503225,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8964291023808605,
      "eval_iou_safe": 0.22672413565462626,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.8874142561414083,
      "eval_loss": 0.6502141952514648,
      "eval_mean_accuracy": 0.7792069143655915,
      "eval_mean_iou": 0.37137946393201154,
      "eval_overall_accuracy": 0.889503991425927,
      "eval_runtime": 9.3648,
      "eval_samples_per_second": 7.154,
      "eval_steps_per_second": 0.534,
      "step": 180
    },
    {
      "epoch": 16.45,
      "learning_rate": 1.428115015974441e-06,
      "loss": 0.6552,
      "step": 181
    },
    {
      "epoch": 16.55,
      "learning_rate": 1.4185303514376998e-06,
      "loss": 0.6621,
      "step": 182
    },
    {
      "epoch": 16.64,
      "learning_rate": 1.4089456869009586e-06,
      "loss": 0.6742,
      "step": 183
    },
    {
      "epoch": 16.73,
      "learning_rate": 1.3993610223642173e-06,
      "loss": 0.6462,
      "step": 184
    },
    {
      "epoch": 16.82,
      "learning_rate": 1.3897763578274761e-06,
      "loss": 0.6759,
      "step": 185
    },
    {
      "epoch": 16.91,
      "learning_rate": 1.3801916932907347e-06,
      "loss": 0.6773,
      "step": 186
    },
    {
      "epoch": 17.0,
      "learning_rate": 1.3706070287539937e-06,
      "loss": 0.7158,
      "step": 187
    },
    {
      "epoch": 17.09,
      "learning_rate": 1.3610223642172525e-06,
      "loss": 0.6958,
      "step": 188
    },
    {
      "epoch": 17.18,
      "learning_rate": 1.3514376996805113e-06,
      "loss": 0.6747,
      "step": 189
    },
    {
      "epoch": 17.27,
      "learning_rate": 1.3418530351437699e-06,
      "loss": 0.6357,
      "step": 190
    },
    {
      "epoch": 17.27,
      "eval_accuracy_safe": 0.6611674588764114,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9051011079830231,
      "eval_iou_safe": 0.24110413837507133,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.895975219611748,
      "eval_loss": 0.6229549646377563,
      "eval_mean_accuracy": 0.7831342834297172,
      "eval_mean_iou": 0.3790264526622731,
      "eval_overall_accuracy": 0.8978956991167211,
      "eval_runtime": 10.2719,
      "eval_samples_per_second": 6.523,
      "eval_steps_per_second": 0.487,
      "step": 190
    },
    {
      "epoch": 17.36,
      "learning_rate": 1.3322683706070287e-06,
      "loss": 0.6769,
      "step": 191
    },
    {
      "epoch": 17.45,
      "learning_rate": 1.3226837060702877e-06,
      "loss": 0.6396,
      "step": 192
    },
    {
      "epoch": 17.55,
      "learning_rate": 1.3130990415335465e-06,
      "loss": 0.6567,
      "step": 193
    },
    {
      "epoch": 17.64,
      "learning_rate": 1.303514376996805e-06,
      "loss": 0.6538,
      "step": 194
    },
    {
      "epoch": 17.73,
      "learning_rate": 1.2939297124600638e-06,
      "loss": 0.6654,
      "step": 195
    },
    {
      "epoch": 17.82,
      "learning_rate": 1.2843450479233228e-06,
      "loss": 0.6637,
      "step": 196
    },
    {
      "epoch": 17.91,
      "learning_rate": 1.2747603833865816e-06,
      "loss": 0.6417,
      "step": 197
    },
    {
      "epoch": 18.0,
      "learning_rate": 1.2651757188498402e-06,
      "loss": 0.6707,
      "step": 198
    },
    {
      "epoch": 18.09,
      "learning_rate": 1.255591054313099e-06,
      "loss": 0.6229,
      "step": 199
    },
    {
      "epoch": 18.18,
      "learning_rate": 1.2460063897763578e-06,
      "loss": 0.6815,
      "step": 200
    },
    {
      "epoch": 18.18,
      "eval_accuracy_safe": 0.6484207848080771,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9178010760554833,
      "eval_iou_safe": 0.2593923131195133,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.908185465668198,
      "eval_loss": 0.5992904305458069,
      "eval_mean_accuracy": 0.7831109304317803,
      "eval_mean_iou": 0.3891925929292371,
      "eval_overall_accuracy": 0.9098440141820195,
      "eval_runtime": 9.7351,
      "eval_samples_per_second": 6.882,
      "eval_steps_per_second": 0.514,
      "step": 200
    },
    {
      "epoch": 18.27,
      "learning_rate": 1.2364217252396168e-06,
      "loss": 0.6677,
      "step": 201
    },
    {
      "epoch": 18.36,
      "learning_rate": 1.2268370607028754e-06,
      "loss": 0.6658,
      "step": 202
    },
    {
      "epoch": 18.45,
      "learning_rate": 1.2172523961661342e-06,
      "loss": 0.649,
      "step": 203
    },
    {
      "epoch": 18.55,
      "learning_rate": 1.207667731629393e-06,
      "loss": 0.6342,
      "step": 204
    },
    {
      "epoch": 18.64,
      "learning_rate": 1.198083067092652e-06,
      "loss": 0.6441,
      "step": 205
    },
    {
      "epoch": 18.73,
      "learning_rate": 1.1884984025559105e-06,
      "loss": 0.6429,
      "step": 206
    },
    {
      "epoch": 18.82,
      "learning_rate": 1.1789137380191693e-06,
      "loss": 0.6155,
      "step": 207
    },
    {
      "epoch": 18.91,
      "learning_rate": 1.1693290734824281e-06,
      "loss": 0.6347,
      "step": 208
    },
    {
      "epoch": 19.0,
      "learning_rate": 1.159744408945687e-06,
      "loss": 0.6611,
      "step": 209
    },
    {
      "epoch": 19.09,
      "learning_rate": 1.1501597444089457e-06,
      "loss": 0.6398,
      "step": 210
    },
    {
      "epoch": 19.09,
      "eval_accuracy_safe": 0.6413602877398313,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9258418644556835,
      "eval_iou_safe": 0.26823363821482293,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9159351131912487,
      "eval_loss": 0.5784907341003418,
      "eval_mean_accuracy": 0.7836010760977574,
      "eval_mean_iou": 0.39472291713535723,
      "eval_overall_accuracy": 0.9174387348231985,
      "eval_runtime": 11.2723,
      "eval_samples_per_second": 5.944,
      "eval_steps_per_second": 0.444,
      "step": 210
    },
    {
      "epoch": 19.18,
      "learning_rate": 1.1405750798722045e-06,
      "loss": 0.6403,
      "step": 211
    },
    {
      "epoch": 19.27,
      "learning_rate": 1.1309904153354633e-06,
      "loss": 0.6271,
      "step": 212
    },
    {
      "epoch": 19.36,
      "learning_rate": 1.121405750798722e-06,
      "loss": 0.679,
      "step": 213
    },
    {
      "epoch": 19.45,
      "learning_rate": 1.1118210862619809e-06,
      "loss": 0.6239,
      "step": 214
    },
    {
      "epoch": 19.55,
      "learning_rate": 1.1022364217252397e-06,
      "loss": 0.6527,
      "step": 215
    },
    {
      "epoch": 19.64,
      "learning_rate": 1.0926517571884984e-06,
      "loss": 0.6228,
      "step": 216
    },
    {
      "epoch": 19.73,
      "learning_rate": 1.0830670926517572e-06,
      "loss": 0.6227,
      "step": 217
    },
    {
      "epoch": 19.82,
      "learning_rate": 1.073482428115016e-06,
      "loss": 0.6347,
      "step": 218
    },
    {
      "epoch": 19.91,
      "learning_rate": 1.0638977635782748e-06,
      "loss": 0.6374,
      "step": 219
    },
    {
      "epoch": 20.0,
      "learning_rate": 1.0543130990415336e-06,
      "loss": 0.5845,
      "step": 220
    },
    {
      "epoch": 20.0,
      "eval_accuracy_safe": 0.6426151017151052,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9286315640516787,
      "eval_iou_safe": 0.26982054857603477,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9187221572756564,
      "eval_loss": 0.5641139149665833,
      "eval_mean_accuracy": 0.7856233328833919,
      "eval_mean_iou": 0.39618090195056377,
      "eval_overall_accuracy": 0.9201830963590252,
      "eval_runtime": 9.4874,
      "eval_samples_per_second": 7.062,
      "eval_steps_per_second": 0.527,
      "step": 220
    },
    {
      "epoch": 20.09,
      "learning_rate": 1.0447284345047924e-06,
      "loss": 0.606,
      "step": 221
    },
    {
      "epoch": 20.18,
      "learning_rate": 1.0351437699680512e-06,
      "loss": 0.6491,
      "step": 222
    },
    {
      "epoch": 20.27,
      "learning_rate": 1.02555910543131e-06,
      "loss": 0.6396,
      "step": 223
    },
    {
      "epoch": 20.36,
      "learning_rate": 1.0159744408945688e-06,
      "loss": 0.617,
      "step": 224
    },
    {
      "epoch": 20.45,
      "learning_rate": 1.0063897763578276e-06,
      "loss": 0.614,
      "step": 225
    },
    {
      "epoch": 20.55,
      "learning_rate": 9.968051118210861e-07,
      "loss": 0.6281,
      "step": 226
    },
    {
      "epoch": 20.64,
      "learning_rate": 9.872204472843452e-07,
      "loss": 0.6374,
      "step": 227
    },
    {
      "epoch": 20.73,
      "learning_rate": 9.77635782747604e-07,
      "loss": 0.6394,
      "step": 228
    },
    {
      "epoch": 20.82,
      "learning_rate": 9.680511182108625e-07,
      "loss": 0.6351,
      "step": 229
    },
    {
      "epoch": 20.91,
      "learning_rate": 9.584664536741213e-07,
      "loss": 0.6062,
      "step": 230
    },
    {
      "epoch": 20.91,
      "eval_accuracy_safe": 0.6519963300064379,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9252195649054266,
      "eval_iou_safe": 0.2640576051310034,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9156001170929594,
      "eval_loss": 0.5693235397338867,
      "eval_mean_accuracy": 0.7886079474559322,
      "eval_mean_iou": 0.3932192407413209,
      "eval_overall_accuracy": 0.9171489886383513,
      "eval_runtime": 11.0762,
      "eval_samples_per_second": 6.049,
      "eval_steps_per_second": 0.451,
      "step": 230
    },
    {
      "epoch": 21.0,
      "learning_rate": 9.488817891373802e-07,
      "loss": 0.6456,
      "step": 231
    },
    {
      "epoch": 21.09,
      "learning_rate": 9.39297124600639e-07,
      "loss": 0.6055,
      "step": 232
    },
    {
      "epoch": 21.18,
      "learning_rate": 9.297124600638977e-07,
      "loss": 0.6134,
      "step": 233
    },
    {
      "epoch": 21.27,
      "learning_rate": 9.201277955271566e-07,
      "loss": 0.5718,
      "step": 234
    },
    {
      "epoch": 21.36,
      "learning_rate": 9.105431309904154e-07,
      "loss": 0.5939,
      "step": 235
    },
    {
      "epoch": 21.45,
      "learning_rate": 9.009584664536742e-07,
      "loss": 0.5917,
      "step": 236
    },
    {
      "epoch": 21.55,
      "learning_rate": 8.913738019169329e-07,
      "loss": 0.599,
      "step": 237
    },
    {
      "epoch": 21.64,
      "learning_rate": 8.817891373801916e-07,
      "loss": 0.6239,
      "step": 238
    },
    {
      "epoch": 21.73,
      "learning_rate": 8.722044728434505e-07,
      "loss": 0.5946,
      "step": 239
    },
    {
      "epoch": 21.82,
      "learning_rate": 8.626198083067093e-07,
      "loss": 0.6071,
      "step": 240
    },
    {
      "epoch": 21.82,
      "eval_accuracy_safe": 0.6591840432380754,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9282916959179331,
      "eval_iou_safe": 0.267531459702011,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9188227497665283,
      "eval_loss": 0.5627052187919617,
      "eval_mean_accuracy": 0.7937378695780043,
      "eval_mean_iou": 0.39545140315617977,
      "eval_overall_accuracy": 0.9203426873506005,
      "eval_runtime": 9.7383,
      "eval_samples_per_second": 6.88,
      "eval_steps_per_second": 0.513,
      "step": 240
    },
    {
      "epoch": 21.91,
      "learning_rate": 8.53035143769968e-07,
      "loss": 0.5784,
      "step": 241
    },
    {
      "epoch": 22.0,
      "learning_rate": 8.434504792332268e-07,
      "loss": 0.6049,
      "step": 242
    },
    {
      "epoch": 22.09,
      "learning_rate": 8.338658146964857e-07,
      "loss": 0.5871,
      "step": 243
    },
    {
      "epoch": 22.18,
      "learning_rate": 8.242811501597445e-07,
      "loss": 0.5894,
      "step": 244
    },
    {
      "epoch": 22.27,
      "learning_rate": 8.146964856230032e-07,
      "loss": 0.5801,
      "step": 245
    },
    {
      "epoch": 22.36,
      "learning_rate": 8.05111821086262e-07,
      "loss": 0.5719,
      "step": 246
    },
    {
      "epoch": 22.45,
      "learning_rate": 7.955271565495208e-07,
      "loss": 0.5979,
      "step": 247
    },
    {
      "epoch": 22.55,
      "learning_rate": 7.859424920127797e-07,
      "loss": 0.5855,
      "step": 248
    },
    {
      "epoch": 22.64,
      "learning_rate": 7.763578274760382e-07,
      "loss": 0.5975,
      "step": 249
    },
    {
      "epoch": 22.73,
      "learning_rate": 7.667731629392971e-07,
      "loss": 0.6209,
      "step": 250
    },
    {
      "epoch": 22.73,
      "eval_accuracy_safe": 0.6618690752926936,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9299528432231069,
      "eval_iou_safe": 0.2724302598559697,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9205411497943506,
      "eval_loss": 0.5631816983222961,
      "eval_mean_accuracy": 0.7959109592579002,
      "eval_mean_iou": 0.39765713655010676,
      "eval_overall_accuracy": 0.9220340785695545,
      "eval_runtime": 10.1606,
      "eval_samples_per_second": 6.594,
      "eval_steps_per_second": 0.492,
      "step": 250
    },
    {
      "epoch": 22.82,
      "learning_rate": 7.571884984025559e-07,
      "loss": 0.5852,
      "step": 251
    },
    {
      "epoch": 22.91,
      "learning_rate": 7.476038338658147e-07,
      "loss": 0.5775,
      "step": 252
    },
    {
      "epoch": 23.0,
      "learning_rate": 7.380191693290735e-07,
      "loss": 0.6365,
      "step": 253
    },
    {
      "epoch": 23.09,
      "learning_rate": 7.284345047923323e-07,
      "loss": 0.6243,
      "step": 254
    },
    {
      "epoch": 23.18,
      "learning_rate": 7.188498402555911e-07,
      "loss": 0.5995,
      "step": 255
    },
    {
      "epoch": 23.27,
      "learning_rate": 7.092651757188499e-07,
      "loss": 0.6019,
      "step": 256
    },
    {
      "epoch": 23.36,
      "learning_rate": 6.996805111821087e-07,
      "loss": 0.5883,
      "step": 257
    },
    {
      "epoch": 23.45,
      "learning_rate": 6.900958466453674e-07,
      "loss": 0.589,
      "step": 258
    },
    {
      "epoch": 23.55,
      "learning_rate": 6.805111821086263e-07,
      "loss": 0.5717,
      "step": 259
    },
    {
      "epoch": 23.64,
      "learning_rate": 6.709265175718849e-07,
      "loss": 0.5609,
      "step": 260
    },
    {
      "epoch": 23.64,
      "eval_accuracy_safe": 0.6504639534928547,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9378781128324656,
      "eval_iou_safe": 0.2868443063281154,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9280609954701114,
      "eval_loss": 0.5416346192359924,
      "eval_mean_accuracy": 0.7941710331626601,
      "eval_mean_iou": 0.40496843393274223,
      "eval_overall_accuracy": 0.9293883594114389,
      "eval_runtime": 9.5225,
      "eval_samples_per_second": 7.036,
      "eval_steps_per_second": 0.525,
      "step": 260
    },
    {
      "epoch": 23.73,
      "learning_rate": 6.613418530351438e-07,
      "loss": 0.5557,
      "step": 261
    },
    {
      "epoch": 23.82,
      "learning_rate": 6.517571884984025e-07,
      "loss": 0.5968,
      "step": 262
    },
    {
      "epoch": 23.91,
      "learning_rate": 6.421725239616614e-07,
      "loss": 0.5673,
      "step": 263
    },
    {
      "epoch": 24.0,
      "learning_rate": 6.325878594249201e-07,
      "loss": 0.6091,
      "step": 264
    },
    {
      "epoch": 24.09,
      "learning_rate": 6.230031948881789e-07,
      "loss": 0.5776,
      "step": 265
    },
    {
      "epoch": 24.18,
      "learning_rate": 6.134185303514377e-07,
      "loss": 0.5562,
      "step": 266
    },
    {
      "epoch": 24.27,
      "learning_rate": 6.038338658146965e-07,
      "loss": 0.5863,
      "step": 267
    },
    {
      "epoch": 24.36,
      "learning_rate": 5.942492012779553e-07,
      "loss": 0.5565,
      "step": 268
    },
    {
      "epoch": 24.45,
      "learning_rate": 5.846645367412141e-07,
      "loss": 0.5858,
      "step": 269
    },
    {
      "epoch": 24.55,
      "learning_rate": 5.750798722044729e-07,
      "loss": 0.5752,
      "step": 270
    },
    {
      "epoch": 24.55,
      "eval_accuracy_safe": 0.6412195789530495,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.945148932410419,
      "eval_iou_safe": 0.2982960328042603,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9349957034087992,
      "eval_loss": 0.5140887498855591,
      "eval_mean_accuracy": 0.7931842556817342,
      "eval_mean_iou": 0.41109724540435316,
      "eval_overall_accuracy": 0.9361713466359608,
      "eval_runtime": 10.0336,
      "eval_samples_per_second": 6.678,
      "eval_steps_per_second": 0.498,
      "step": 270
    },
    {
      "epoch": 24.64,
      "learning_rate": 5.654952076677316e-07,
      "loss": 0.5881,
      "step": 271
    },
    {
      "epoch": 24.73,
      "learning_rate": 5.559105431309904e-07,
      "loss": 0.6124,
      "step": 272
    },
    {
      "epoch": 24.82,
      "learning_rate": 5.463258785942492e-07,
      "loss": 0.5648,
      "step": 273
    },
    {
      "epoch": 24.91,
      "learning_rate": 5.36741214057508e-07,
      "loss": 0.578,
      "step": 274
    },
    {
      "epoch": 25.0,
      "learning_rate": 5.271565495207668e-07,
      "loss": 0.5605,
      "step": 275
    },
    {
      "epoch": 25.09,
      "learning_rate": 5.175718849840256e-07,
      "loss": 0.588,
      "step": 276
    },
    {
      "epoch": 25.18,
      "learning_rate": 5.079872204472844e-07,
      "loss": 0.5814,
      "step": 277
    },
    {
      "epoch": 25.27,
      "learning_rate": 4.984025559105431e-07,
      "loss": 0.5671,
      "step": 278
    },
    {
      "epoch": 25.36,
      "learning_rate": 4.88817891373802e-07,
      "loss": 0.5237,
      "step": 279
    },
    {
      "epoch": 25.45,
      "learning_rate": 4.792332268370607e-07,
      "loss": 0.6004,
      "step": 280
    },
    {
      "epoch": 25.45,
      "eval_accuracy_safe": 0.6491917918589365,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9412261630289883,
      "eval_iou_safe": 0.2907094372975922,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9313366391036164,
      "eval_loss": 0.5255351662635803,
      "eval_mean_accuracy": 0.7952089774439624,
      "eval_mean_iou": 0.4073486921337362,
      "eval_overall_accuracy": 0.9325999359586346,
      "eval_runtime": 9.8685,
      "eval_samples_per_second": 6.789,
      "eval_steps_per_second": 0.507,
      "step": 280
    },
    {
      "epoch": 25.55,
      "learning_rate": 4.696485623003195e-07,
      "loss": 0.5482,
      "step": 281
    },
    {
      "epoch": 25.64,
      "learning_rate": 4.600638977635783e-07,
      "loss": 0.5764,
      "step": 282
    },
    {
      "epoch": 25.73,
      "learning_rate": 4.504792332268371e-07,
      "loss": 0.5608,
      "step": 283
    },
    {
      "epoch": 25.82,
      "learning_rate": 4.408945686900958e-07,
      "loss": 0.5698,
      "step": 284
    },
    {
      "epoch": 25.91,
      "learning_rate": 4.3130990415335466e-07,
      "loss": 0.5502,
      "step": 285
    },
    {
      "epoch": 26.0,
      "learning_rate": 4.217252396166134e-07,
      "loss": 0.5975,
      "step": 286
    },
    {
      "epoch": 26.09,
      "learning_rate": 4.1214057507987225e-07,
      "loss": 0.5736,
      "step": 287
    },
    {
      "epoch": 26.18,
      "learning_rate": 4.02555910543131e-07,
      "loss": 0.5498,
      "step": 288
    },
    {
      "epoch": 26.27,
      "learning_rate": 3.9297124600638983e-07,
      "loss": 0.5682,
      "step": 289
    },
    {
      "epoch": 26.36,
      "learning_rate": 3.8338658146964857e-07,
      "loss": 0.5524,
      "step": 290
    },
    {
      "epoch": 26.36,
      "eval_accuracy_safe": 0.6587715544658655,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9386905578378355,
      "eval_iou_safe": 0.28676309565062713,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9290998927343179,
      "eval_loss": 0.5313962697982788,
      "eval_mean_accuracy": 0.7987310561518506,
      "eval_mean_iou": 0.4052876627949817,
      "eval_overall_accuracy": 0.9304221993062034,
      "eval_runtime": 9.5415,
      "eval_samples_per_second": 7.022,
      "eval_steps_per_second": 0.524,
      "step": 290
    },
    {
      "epoch": 26.45,
      "learning_rate": 3.7380191693290736e-07,
      "loss": 0.5519,
      "step": 291
    },
    {
      "epoch": 26.55,
      "learning_rate": 3.6421725239616615e-07,
      "loss": 0.5424,
      "step": 292
    },
    {
      "epoch": 26.64,
      "learning_rate": 3.5463258785942494e-07,
      "loss": 0.583,
      "step": 293
    },
    {
      "epoch": 26.73,
      "learning_rate": 3.450479233226837e-07,
      "loss": 0.5798,
      "step": 294
    },
    {
      "epoch": 26.82,
      "learning_rate": 3.3546325878594247e-07,
      "loss": 0.5284,
      "step": 295
    },
    {
      "epoch": 26.91,
      "learning_rate": 3.2587859424920126e-07,
      "loss": 0.5681,
      "step": 296
    },
    {
      "epoch": 27.0,
      "learning_rate": 3.1629392971246005e-07,
      "loss": 0.6352,
      "step": 297
    },
    {
      "epoch": 27.09,
      "learning_rate": 3.0670926517571884e-07,
      "loss": 0.5651,
      "step": 298
    },
    {
      "epoch": 27.18,
      "learning_rate": 2.9712460063897763e-07,
      "loss": 0.5707,
      "step": 299
    },
    {
      "epoch": 27.27,
      "learning_rate": 2.875399361022364e-07,
      "loss": 0.5758,
      "step": 300
    },
    {
      "epoch": 27.27,
      "eval_accuracy_safe": 0.6544115095932552,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9423355893036522,
      "eval_iou_safe": 0.291325116977104,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9325746804551616,
      "eval_loss": 0.526845395565033,
      "eval_mean_accuracy": 0.7983735494484537,
      "eval_mean_iou": 0.40796659914408856,
      "eval_overall_accuracy": 0.9338307736524895,
      "eval_runtime": 10.1793,
      "eval_samples_per_second": 6.582,
      "eval_steps_per_second": 0.491,
      "step": 300
    },
    {
      "epoch": 27.36,
      "learning_rate": 2.779552715654952e-07,
      "loss": 0.5541,
      "step": 301
    },
    {
      "epoch": 27.45,
      "learning_rate": 2.68370607028754e-07,
      "loss": 0.5143,
      "step": 302
    },
    {
      "epoch": 27.55,
      "learning_rate": 2.587859424920128e-07,
      "loss": 0.5392,
      "step": 303
    },
    {
      "epoch": 27.64,
      "learning_rate": 2.4920127795527154e-07,
      "loss": 0.5803,
      "step": 304
    },
    {
      "epoch": 27.73,
      "learning_rate": 2.3961661341853033e-07,
      "loss": 0.5733,
      "step": 305
    },
    {
      "epoch": 27.82,
      "learning_rate": 2.3003194888178915e-07,
      "loss": 0.5799,
      "step": 306
    },
    {
      "epoch": 27.91,
      "learning_rate": 2.204472843450479e-07,
      "loss": 0.5737,
      "step": 307
    },
    {
      "epoch": 28.0,
      "learning_rate": 2.108626198083067e-07,
      "loss": 0.507,
      "step": 308
    },
    {
      "epoch": 28.09,
      "learning_rate": 2.012779552715655e-07,
      "loss": 0.5913,
      "step": 309
    },
    {
      "epoch": 28.18,
      "learning_rate": 1.9169329073482428e-07,
      "loss": 0.5598,
      "step": 310
    },
    {
      "epoch": 28.18,
      "eval_accuracy_safe": 0.6604831901187737,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9407892567641855,
      "eval_iou_safe": 0.2896690945241332,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9312167321135931,
      "eval_loss": 0.5239936709403992,
      "eval_mean_accuracy": 0.8006362234414796,
      "eval_mean_iou": 0.4069619422125754,
      "eval_overall_accuracy": 0.9325094650040812,
      "eval_runtime": 9.6243,
      "eval_samples_per_second": 6.962,
      "eval_steps_per_second": 0.52,
      "step": 310
    },
    {
      "epoch": 28.27,
      "learning_rate": 1.8210862619808307e-07,
      "loss": 0.5394,
      "step": 311
    },
    {
      "epoch": 28.36,
      "learning_rate": 1.7252396166134184e-07,
      "loss": 0.5312,
      "step": 312
    },
    {
      "epoch": 28.45,
      "learning_rate": 1.6293929712460063e-07,
      "loss": 0.5313,
      "step": 313
    },
    {
      "epoch": 28.55,
      "learning_rate": 1.5335463258785942e-07,
      "loss": 0.5188,
      "step": 314
    },
    {
      "epoch": 28.64,
      "learning_rate": 1.437699680511182e-07,
      "loss": 0.5497,
      "step": 315
    },
    {
      "epoch": 28.73,
      "learning_rate": 1.34185303514377e-07,
      "loss": 0.5908,
      "step": 316
    },
    {
      "epoch": 28.82,
      "learning_rate": 1.2460063897763577e-07,
      "loss": 0.56,
      "step": 317
    },
    {
      "epoch": 28.91,
      "learning_rate": 1.1501597444089457e-07,
      "loss": 0.5792,
      "step": 318
    },
    {
      "epoch": 29.0,
      "learning_rate": 1.0543130990415335e-07,
      "loss": 0.6318,
      "step": 319
    },
    {
      "epoch": 29.09,
      "learning_rate": 9.584664536741214e-08,
      "loss": 0.5505,
      "step": 320
    },
    {
      "epoch": 29.09,
      "eval_accuracy_safe": 0.6582414871183997,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9420817295738547,
      "eval_iou_safe": 0.29588640634788055,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9324414181256692,
      "eval_loss": 0.5165032744407654,
      "eval_mean_accuracy": 0.8001616083461272,
      "eval_mean_iou": 0.4094426081578499,
      "eval_overall_accuracy": 0.933697543927093,
      "eval_runtime": 11.4418,
      "eval_samples_per_second": 5.856,
      "eval_steps_per_second": 0.437,
      "step": 320
    },
    {
      "epoch": 29.18,
      "learning_rate": 1.3160919540229885e-06,
      "loss": 0.5725,
      "step": 321
    },
    {
      "epoch": 29.27,
      "learning_rate": 1.310344827586207e-06,
      "loss": 0.5195,
      "step": 322
    },
    {
      "epoch": 29.36,
      "learning_rate": 1.3045977011494253e-06,
      "loss": 0.5441,
      "step": 323
    },
    {
      "epoch": 29.45,
      "learning_rate": 1.2988505747126436e-06,
      "loss": 0.5896,
      "step": 324
    },
    {
      "epoch": 29.55,
      "learning_rate": 1.293103448275862e-06,
      "loss": 0.5837,
      "step": 325
    },
    {
      "epoch": 29.64,
      "learning_rate": 1.2873563218390806e-06,
      "loss": 0.5317,
      "step": 326
    },
    {
      "epoch": 29.73,
      "learning_rate": 1.2816091954022989e-06,
      "loss": 0.5395,
      "step": 327
    },
    {
      "epoch": 29.82,
      "learning_rate": 1.2758620689655172e-06,
      "loss": 0.5642,
      "step": 328
    },
    {
      "epoch": 29.91,
      "learning_rate": 1.2701149425287357e-06,
      "loss": 0.5314,
      "step": 329
    },
    {
      "epoch": 30.0,
      "learning_rate": 1.2643678160919542e-06,
      "loss": 0.5754,
      "step": 330
    },
    {
      "epoch": 30.0,
      "eval_accuracy_safe": 0.657778882887884,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9432738201330771,
      "eval_iou_safe": 0.2958553359937371,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9336061303908306,
      "eval_loss": 0.5145326852798462,
      "eval_mean_accuracy": 0.8005263515104806,
      "eval_mean_iou": 0.40982048879485594,
      "eval_overall_accuracy": 0.9348407574553987,
      "eval_runtime": 8.5617,
      "eval_samples_per_second": 7.826,
      "eval_steps_per_second": 0.584,
      "step": 330
    },
    {
      "epoch": 30.09,
      "learning_rate": 1.2586206896551725e-06,
      "loss": 0.5332,
      "step": 331
    },
    {
      "epoch": 30.18,
      "learning_rate": 1.252873563218391e-06,
      "loss": 0.5386,
      "step": 332
    },
    {
      "epoch": 30.27,
      "learning_rate": 1.2471264367816092e-06,
      "loss": 0.5446,
      "step": 333
    },
    {
      "epoch": 30.36,
      "learning_rate": 1.2413793103448275e-06,
      "loss": 0.524,
      "step": 334
    },
    {
      "epoch": 30.45,
      "learning_rate": 1.235632183908046e-06,
      "loss": 0.5337,
      "step": 335
    },
    {
      "epoch": 30.55,
      "learning_rate": 1.2298850574712645e-06,
      "loss": 0.5228,
      "step": 336
    },
    {
      "epoch": 30.64,
      "learning_rate": 1.2241379310344828e-06,
      "loss": 0.5856,
      "step": 337
    },
    {
      "epoch": 30.73,
      "learning_rate": 1.2183908045977011e-06,
      "loss": 0.5221,
      "step": 338
    },
    {
      "epoch": 30.82,
      "learning_rate": 1.2126436781609196e-06,
      "loss": 0.5505,
      "step": 339
    },
    {
      "epoch": 30.91,
      "learning_rate": 1.206896551724138e-06,
      "loss": 0.5284,
      "step": 340
    },
    {
      "epoch": 30.91,
      "eval_accuracy_safe": 0.6718902394362396,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9410974437668724,
      "eval_iou_safe": 0.29405467602594526,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9318466838959976,
      "eval_loss": 0.5174660086631775,
      "eval_mean_accuracy": 0.806493841601556,
      "eval_mean_iou": 0.4086337866406476,
      "eval_overall_accuracy": 0.933145494603399,
      "eval_runtime": 8.5465,
      "eval_samples_per_second": 7.839,
      "eval_steps_per_second": 0.585,
      "step": 340
    },
    {
      "epoch": 31.0,
      "learning_rate": 1.2011494252873564e-06,
      "loss": 0.5315,
      "step": 341
    },
    {
      "epoch": 31.09,
      "learning_rate": 1.1954022988505747e-06,
      "loss": 0.511,
      "step": 342
    },
    {
      "epoch": 31.18,
      "learning_rate": 1.1896551724137932e-06,
      "loss": 0.5607,
      "step": 343
    },
    {
      "epoch": 31.27,
      "learning_rate": 1.1839080459770115e-06,
      "loss": 0.5254,
      "step": 344
    },
    {
      "epoch": 31.36,
      "learning_rate": 1.1781609195402298e-06,
      "loss": 0.4814,
      "step": 345
    },
    {
      "epoch": 31.45,
      "learning_rate": 1.1724137931034483e-06,
      "loss": 0.5376,
      "step": 346
    },
    {
      "epoch": 31.55,
      "learning_rate": 1.1666666666666668e-06,
      "loss": 0.5253,
      "step": 347
    },
    {
      "epoch": 31.64,
      "learning_rate": 1.160919540229885e-06,
      "loss": 0.5591,
      "step": 348
    },
    {
      "epoch": 31.73,
      "learning_rate": 1.1551724137931034e-06,
      "loss": 0.5161,
      "step": 349
    },
    {
      "epoch": 31.82,
      "learning_rate": 1.1494252873563219e-06,
      "loss": 0.5463,
      "step": 350
    },
    {
      "epoch": 31.82,
      "eval_accuracy_safe": 0.6683590271433032,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9448280729553086,
      "eval_iou_safe": 0.3019912053572595,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9354427723232134,
      "eval_loss": 0.5016354918479919,
      "eval_mean_accuracy": 0.8065935500493059,
      "eval_mean_iou": 0.4124779925601576,
      "eval_overall_accuracy": 0.9366616206382637,
      "eval_runtime": 8.7504,
      "eval_samples_per_second": 7.657,
      "eval_steps_per_second": 0.571,
      "step": 350
    },
    {
      "epoch": 31.91,
      "learning_rate": 1.1436781609195402e-06,
      "loss": 0.4994,
      "step": 351
    },
    {
      "epoch": 32.0,
      "learning_rate": 1.1379310344827587e-06,
      "loss": 0.5251,
      "step": 352
    },
    {
      "epoch": 32.09,
      "learning_rate": 1.1321839080459772e-06,
      "loss": 0.5379,
      "step": 353
    },
    {
      "epoch": 32.18,
      "learning_rate": 1.1264367816091955e-06,
      "loss": 0.4885,
      "step": 354
    },
    {
      "epoch": 32.27,
      "learning_rate": 1.1206896551724137e-06,
      "loss": 0.5029,
      "step": 355
    },
    {
      "epoch": 32.36,
      "learning_rate": 1.1149425287356322e-06,
      "loss": 0.5273,
      "step": 356
    },
    {
      "epoch": 32.45,
      "learning_rate": 1.1091954022988505e-06,
      "loss": 0.5111,
      "step": 357
    },
    {
      "epoch": 32.55,
      "learning_rate": 1.103448275862069e-06,
      "loss": 0.5102,
      "step": 358
    },
    {
      "epoch": 32.64,
      "learning_rate": 1.0977011494252873e-06,
      "loss": 0.5197,
      "step": 359
    },
    {
      "epoch": 32.73,
      "learning_rate": 1.0919540229885058e-06,
      "loss": 0.4923,
      "step": 360
    },
    {
      "epoch": 32.73,
      "eval_accuracy_safe": 0.6687907910917845,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9462536065154241,
      "eval_iou_safe": 0.3065972713134455,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.936872865769379,
      "eval_loss": 0.4946950674057007,
      "eval_mean_accuracy": 0.8075221988036043,
      "eval_mean_iou": 0.41449004569427483,
      "eval_overall_accuracy": 0.9380577998374825,
      "eval_runtime": 9.7999,
      "eval_samples_per_second": 6.837,
      "eval_steps_per_second": 0.51,
      "step": 360
    },
    {
      "epoch": 32.82,
      "learning_rate": 1.0862068965517241e-06,
      "loss": 0.4919,
      "step": 361
    },
    {
      "epoch": 32.91,
      "learning_rate": 1.0804597701149424e-06,
      "loss": 0.5359,
      "step": 362
    },
    {
      "epoch": 33.0,
      "learning_rate": 1.0747126436781611e-06,
      "loss": 0.5646,
      "step": 363
    },
    {
      "epoch": 33.09,
      "learning_rate": 1.0689655172413794e-06,
      "loss": 0.5186,
      "step": 364
    },
    {
      "epoch": 33.18,
      "learning_rate": 1.0632183908045977e-06,
      "loss": 0.4801,
      "step": 365
    },
    {
      "epoch": 33.27,
      "learning_rate": 1.057471264367816e-06,
      "loss": 0.4874,
      "step": 366
    },
    {
      "epoch": 33.36,
      "learning_rate": 1.0517241379310345e-06,
      "loss": 0.5092,
      "step": 367
    },
    {
      "epoch": 33.45,
      "learning_rate": 1.045977011494253e-06,
      "loss": 0.503,
      "step": 368
    },
    {
      "epoch": 33.55,
      "learning_rate": 1.0402298850574713e-06,
      "loss": 0.5327,
      "step": 369
    },
    {
      "epoch": 33.64,
      "learning_rate": 1.0344827586206898e-06,
      "loss": 0.4922,
      "step": 370
    },
    {
      "epoch": 33.64,
      "eval_accuracy_safe": 0.6685016634477122,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9503706868340143,
      "eval_iou_safe": 0.3165045602136191,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9409432282460215,
      "eval_loss": 0.4738222360610962,
      "eval_mean_accuracy": 0.8094361751408632,
      "eval_mean_iou": 0.41914926281988024,
      "eval_overall_accuracy": 0.9420447278378615,
      "eval_runtime": 10.1885,
      "eval_samples_per_second": 6.576,
      "eval_steps_per_second": 0.491,
      "step": 370
    },
    {
      "epoch": 33.73,
      "learning_rate": 1.028735632183908e-06,
      "loss": 0.5358,
      "step": 371
    },
    {
      "epoch": 33.82,
      "learning_rate": 1.0229885057471264e-06,
      "loss": 0.4633,
      "step": 372
    },
    {
      "epoch": 33.91,
      "learning_rate": 1.0172413793103449e-06,
      "loss": 0.4886,
      "step": 373
    },
    {
      "epoch": 34.0,
      "learning_rate": 1.0114942528735634e-06,
      "loss": 0.531,
      "step": 374
    },
    {
      "epoch": 34.09,
      "learning_rate": 1.0057471264367817e-06,
      "loss": 0.5054,
      "step": 375
    },
    {
      "epoch": 34.18,
      "learning_rate": 1e-06,
      "loss": 0.498,
      "step": 376
    },
    {
      "epoch": 34.27,
      "learning_rate": 9.942528735632185e-07,
      "loss": 0.5071,
      "step": 377
    },
    {
      "epoch": 34.36,
      "learning_rate": 9.885057471264367e-07,
      "loss": 0.4907,
      "step": 378
    },
    {
      "epoch": 34.45,
      "learning_rate": 9.827586206896552e-07,
      "loss": 0.4967,
      "step": 379
    },
    {
      "epoch": 34.55,
      "learning_rate": 9.770114942528735e-07,
      "loss": 0.4976,
      "step": 380
    },
    {
      "epoch": 34.55,
      "eval_accuracy_safe": 0.6747757333240813,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9535474829165368,
      "eval_iou_safe": 0.3232939737984814,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.944264577460996,
      "eval_loss": 0.46629098057746887,
      "eval_mean_accuracy": 0.814161608120309,
      "eval_mean_iou": 0.42251951708649244,
      "eval_overall_accuracy": 0.9453130124220207,
      "eval_runtime": 8.7895,
      "eval_samples_per_second": 7.623,
      "eval_steps_per_second": 0.569,
      "step": 380
    },
    {
      "epoch": 34.64,
      "learning_rate": 9.71264367816092e-07,
      "loss": 0.4647,
      "step": 381
    },
    {
      "epoch": 34.73,
      "learning_rate": 9.655172413793103e-07,
      "loss": 0.4782,
      "step": 382
    },
    {
      "epoch": 34.82,
      "learning_rate": 9.597701149425286e-07,
      "loss": 0.4785,
      "step": 383
    },
    {
      "epoch": 34.91,
      "learning_rate": 9.540229885057471e-07,
      "loss": 0.4996,
      "step": 384
    },
    {
      "epoch": 35.0,
      "learning_rate": 9.482758620689655e-07,
      "loss": 0.437,
      "step": 385
    },
    {
      "epoch": 35.09,
      "learning_rate": 9.425287356321839e-07,
      "loss": 0.4617,
      "step": 386
    },
    {
      "epoch": 35.18,
      "learning_rate": 9.367816091954024e-07,
      "loss": 0.4486,
      "step": 387
    },
    {
      "epoch": 35.27,
      "learning_rate": 9.310344827586207e-07,
      "loss": 0.5012,
      "step": 388
    },
    {
      "epoch": 35.36,
      "learning_rate": 9.252873563218391e-07,
      "loss": 0.4752,
      "step": 389
    },
    {
      "epoch": 35.45,
      "learning_rate": 9.195402298850574e-07,
      "loss": 0.4922,
      "step": 390
    },
    {
      "epoch": 35.45,
      "eval_accuracy_safe": 0.650909210064726,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9652786537349766,
      "eval_iou_safe": 0.3483788236580638,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9551696231808672,
      "eval_loss": 0.4295278489589691,
      "eval_mean_accuracy": 0.8080939318998512,
      "eval_mean_iou": 0.4345161489463103,
      "eval_overall_accuracy": 0.9559926844354886,
      "eval_runtime": 9.5593,
      "eval_samples_per_second": 7.009,
      "eval_steps_per_second": 0.523,
      "step": 390
    },
    {
      "epoch": 35.55,
      "learning_rate": 9.137931034482759e-07,
      "loss": 0.4963,
      "step": 391
    },
    {
      "epoch": 35.64,
      "learning_rate": 9.080459770114943e-07,
      "loss": 0.4951,
      "step": 392
    },
    {
      "epoch": 35.73,
      "learning_rate": 9.022988505747126e-07,
      "loss": 0.4667,
      "step": 393
    },
    {
      "epoch": 35.82,
      "learning_rate": 8.965517241379311e-07,
      "loss": 0.4694,
      "step": 394
    },
    {
      "epoch": 35.91,
      "learning_rate": 8.908045977011495e-07,
      "loss": 0.471,
      "step": 395
    },
    {
      "epoch": 36.0,
      "learning_rate": 8.850574712643678e-07,
      "loss": 0.5268,
      "step": 396
    },
    {
      "epoch": 36.09,
      "learning_rate": 8.793103448275862e-07,
      "loss": 0.4383,
      "step": 397
    },
    {
      "epoch": 36.18,
      "learning_rate": 8.735632183908047e-07,
      "loss": 0.48,
      "step": 398
    },
    {
      "epoch": 36.27,
      "learning_rate": 8.67816091954023e-07,
      "loss": 0.4727,
      "step": 399
    },
    {
      "epoch": 36.36,
      "learning_rate": 8.620689655172414e-07,
      "loss": 0.4608,
      "step": 400
    },
    {
      "epoch": 36.36,
      "eval_accuracy_safe": 0.6580178950736505,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9636894929998194,
      "eval_iou_safe": 0.35071600798452407,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9538031459361976,
      "eval_loss": 0.44339337944984436,
      "eval_mean_accuracy": 0.810853694036735,
      "eval_mean_iou": 0.4348397179735739,
      "eval_overall_accuracy": 0.9546604441173041,
      "eval_runtime": 9.0957,
      "eval_samples_per_second": 7.366,
      "eval_steps_per_second": 0.55,
      "step": 400
    },
    {
      "epoch": 36.45,
      "learning_rate": 8.563218390804599e-07,
      "loss": 0.5072,
      "step": 401
    },
    {
      "epoch": 36.55,
      "learning_rate": 8.505747126436782e-07,
      "loss": 0.4273,
      "step": 402
    },
    {
      "epoch": 36.64,
      "learning_rate": 8.448275862068965e-07,
      "loss": 0.4543,
      "step": 403
    },
    {
      "epoch": 36.73,
      "learning_rate": 8.390804597701149e-07,
      "loss": 0.4856,
      "step": 404
    },
    {
      "epoch": 36.82,
      "learning_rate": 8.333333333333334e-07,
      "loss": 0.4673,
      "step": 405
    },
    {
      "epoch": 36.91,
      "learning_rate": 8.275862068965517e-07,
      "loss": 0.4909,
      "step": 406
    },
    {
      "epoch": 37.0,
      "learning_rate": 8.218390804597701e-07,
      "loss": 0.5135,
      "step": 407
    },
    {
      "epoch": 37.09,
      "learning_rate": 8.160919540229886e-07,
      "loss": 0.4513,
      "step": 408
    },
    {
      "epoch": 37.18,
      "learning_rate": 8.103448275862069e-07,
      "loss": 0.4915,
      "step": 409
    },
    {
      "epoch": 37.27,
      "learning_rate": 8.045977011494253e-07,
      "loss": 0.4836,
      "step": 410
    },
    {
      "epoch": 37.27,
      "eval_accuracy_safe": 0.6521813716986442,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9662073802250839,
      "eval_iou_safe": 0.3588467371591596,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.956125924319263,
      "eval_loss": 0.4327642321586609,
      "eval_mean_accuracy": 0.809194375961864,
      "eval_mean_iou": 0.43832422049280756,
      "eval_overall_accuracy": 0.956931555449073,
      "eval_runtime": 10.5295,
      "eval_samples_per_second": 6.363,
      "eval_steps_per_second": 0.475,
      "step": 410
    },
    {
      "epoch": 37.36,
      "learning_rate": 7.988505747126438e-07,
      "loss": 0.4822,
      "step": 411
    },
    {
      "epoch": 37.45,
      "learning_rate": 7.931034482758621e-07,
      "loss": 0.4407,
      "step": 412
    },
    {
      "epoch": 37.55,
      "learning_rate": 7.873563218390805e-07,
      "loss": 0.471,
      "step": 413
    },
    {
      "epoch": 37.64,
      "learning_rate": 7.816091954022988e-07,
      "loss": 0.4699,
      "step": 414
    },
    {
      "epoch": 37.73,
      "learning_rate": 7.758620689655173e-07,
      "loss": 0.4575,
      "step": 415
    },
    {
      "epoch": 37.82,
      "learning_rate": 7.701149425287357e-07,
      "loss": 0.4705,
      "step": 416
    },
    {
      "epoch": 37.91,
      "learning_rate": 7.64367816091954e-07,
      "loss": 0.435,
      "step": 417
    },
    {
      "epoch": 38.0,
      "learning_rate": 7.586206896551725e-07,
      "loss": 0.4484,
      "step": 418
    },
    {
      "epoch": 38.09,
      "learning_rate": 7.528735632183909e-07,
      "loss": 0.447,
      "step": 419
    },
    {
      "epoch": 38.18,
      "learning_rate": 7.471264367816092e-07,
      "loss": 0.459,
      "step": 420
    },
    {
      "epoch": 38.18,
      "eval_accuracy_safe": 0.6477345885328122,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9690508790751174,
      "eval_iou_safe": 0.36317283147412854,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9587996460225279,
      "eval_loss": 0.42110592126846313,
      "eval_mean_accuracy": 0.8083927338039648,
      "eval_mean_iou": 0.4406574924988855,
      "eval_overall_accuracy": 0.9595597110577484,
      "eval_runtime": 9.3518,
      "eval_samples_per_second": 7.164,
      "eval_steps_per_second": 0.535,
      "step": 420
    },
    {
      "epoch": 38.27,
      "learning_rate": 7.413793103448277e-07,
      "loss": 0.4621,
      "step": 421
    },
    {
      "epoch": 38.36,
      "learning_rate": 7.35632183908046e-07,
      "loss": 0.4535,
      "step": 422
    },
    {
      "epoch": 38.45,
      "learning_rate": 7.298850574712644e-07,
      "loss": 0.494,
      "step": 423
    },
    {
      "epoch": 38.55,
      "learning_rate": 7.241379310344829e-07,
      "loss": 0.4429,
      "step": 424
    },
    {
      "epoch": 38.64,
      "learning_rate": 7.183908045977011e-07,
      "loss": 0.4367,
      "step": 425
    },
    {
      "epoch": 38.73,
      "learning_rate": 7.126436781609195e-07,
      "loss": 0.425,
      "step": 426
    },
    {
      "epoch": 38.82,
      "learning_rate": 7.068965517241379e-07,
      "loss": 0.4453,
      "step": 427
    },
    {
      "epoch": 38.91,
      "learning_rate": 7.011494252873563e-07,
      "loss": 0.4533,
      "step": 428
    },
    {
      "epoch": 39.0,
      "learning_rate": 6.954022988505747e-07,
      "loss": 0.476,
      "step": 429
    },
    {
      "epoch": 39.09,
      "learning_rate": 6.896551724137931e-07,
      "loss": 0.4528,
      "step": 430
    },
    {
      "epoch": 39.09,
      "eval_accuracy_safe": 0.6592746365665514,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9668307944818041,
      "eval_iou_safe": 0.357427398033294,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9569448389901313,
      "eval_loss": 0.4239484369754791,
      "eval_mean_accuracy": 0.8130527155241778,
      "eval_mean_iou": 0.4381240790078084,
      "eval_overall_accuracy": 0.957746078718954,
      "eval_runtime": 9.9636,
      "eval_samples_per_second": 6.724,
      "eval_steps_per_second": 0.502,
      "step": 430
    },
    {
      "epoch": 39.18,
      "learning_rate": 6.839080459770115e-07,
      "loss": 0.4644,
      "step": 431
    },
    {
      "epoch": 39.27,
      "learning_rate": 6.781609195402299e-07,
      "loss": 0.4788,
      "step": 432
    },
    {
      "epoch": 39.36,
      "learning_rate": 6.724137931034483e-07,
      "loss": 0.4112,
      "step": 433
    },
    {
      "epoch": 39.45,
      "learning_rate": 6.666666666666666e-07,
      "loss": 0.4425,
      "step": 434
    },
    {
      "epoch": 39.55,
      "learning_rate": 6.609195402298851e-07,
      "loss": 0.4941,
      "step": 435
    },
    {
      "epoch": 39.64,
      "learning_rate": 6.551724137931035e-07,
      "loss": 0.4367,
      "step": 436
    },
    {
      "epoch": 39.73,
      "learning_rate": 6.494252873563218e-07,
      "loss": 0.4364,
      "step": 437
    },
    {
      "epoch": 39.82,
      "learning_rate": 6.436781609195403e-07,
      "loss": 0.4365,
      "step": 438
    },
    {
      "epoch": 39.91,
      "learning_rate": 6.379310344827586e-07,
      "loss": 0.4137,
      "step": 439
    },
    {
      "epoch": 40.0,
      "learning_rate": 6.321839080459771e-07,
      "loss": 0.4202,
      "step": 440
    },
    {
      "epoch": 40.0,
      "eval_accuracy_safe": 0.657156294694315,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9689412271603979,
      "eval_iou_safe": 0.3649963867997752,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9589753883685492,
      "eval_loss": 0.41405442357063293,
      "eval_mean_accuracy": 0.8130487609273565,
      "eval_mean_iou": 0.4413239250561081,
      "eval_overall_accuracy": 0.9597316001778218,
      "eval_runtime": 9.3639,
      "eval_samples_per_second": 7.155,
      "eval_steps_per_second": 0.534,
      "step": 440
    }
  ],
  "max_steps": 550,
  "num_train_epochs": 50,
  "total_flos": 2.4589017501872947e+18,
  "trial_name": null,
  "trial_params": null
}