diff --git "a/checkpoint-1240/trainer_state.json" "b/checkpoint-1240/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-1240/trainer_state.json"
@@ -0,0 +1,9564 @@
+{
+  "best_metric": 0.22187493741512299,
+  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF-RGB-b5/checkpoint-1240",
+  "epoch": 112.72727272727273,
+  "global_step": 1240,
+  "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
+    },
+    {
+      "epoch": 40.09,
+      "learning_rate": 6.264367816091955e-07,
+      "loss": 0.4279,
+      "step": 441
+    },
+    {
+      "epoch": 40.18,
+      "learning_rate": 6.206896551724138e-07,
+      "loss": 0.4672,
+      "step": 442
+    },
+    {
+      "epoch": 40.27,
+      "learning_rate": 6.149425287356323e-07,
+      "loss": 0.4416,
+      "step": 443
+    },
+    {
+      "epoch": 40.36,
+      "learning_rate": 6.091954022988506e-07,
+      "loss": 0.4589,
+      "step": 444
+    },
+    {
+      "epoch": 40.45,
+      "learning_rate": 6.03448275862069e-07,
+      "loss": 0.4292,
+      "step": 445
+    },
+    {
+      "epoch": 40.55,
+      "learning_rate": 5.977011494252874e-07,
+      "loss": 0.4247,
+      "step": 446
+    },
+    {
+      "epoch": 40.64,
+      "learning_rate": 5.919540229885057e-07,
+      "loss": 0.4454,
+      "step": 447
+    },
+    {
+      "epoch": 40.73,
+      "learning_rate": 5.862068965517241e-07,
+      "loss": 0.4544,
+      "step": 448
+    },
+    {
+      "epoch": 40.82,
+      "learning_rate": 5.804597701149425e-07,
+      "loss": 0.4194,
+      "step": 449
+    },
+    {
+      "epoch": 40.91,
+      "learning_rate": 5.747126436781609e-07,
+      "loss": 0.4805,
+      "step": 450
+    },
+    {
+      "epoch": 40.91,
+      "eval_accuracy_safe": 0.6469790016229698,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9723798619242439,
+      "eval_iou_safe": 0.37541578774301354,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9620762119956281,
+      "eval_loss": 0.40121129155158997,
+      "eval_mean_accuracy": 0.8096794317736069,
+      "eval_mean_iou": 0.44583066657954723,
+      "eval_overall_accuracy": 0.9627680422654793,
+      "eval_runtime": 10.0604,
+      "eval_samples_per_second": 6.66,
+      "eval_steps_per_second": 0.497,
+      "step": 450
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 5.689655172413793e-07,
+      "loss": 0.5147,
+      "step": 451
+    },
+    {
+      "epoch": 41.09,
+      "learning_rate": 5.632183908045977e-07,
+      "loss": 0.455,
+      "step": 452
+    },
+    {
+      "epoch": 41.18,
+      "learning_rate": 5.574712643678161e-07,
+      "loss": 0.4551,
+      "step": 453
+    },
+    {
+      "epoch": 41.27,
+      "learning_rate": 5.517241379310345e-07,
+      "loss": 0.4154,
+      "step": 454
+    },
+    {
+      "epoch": 41.36,
+      "learning_rate": 5.459770114942529e-07,
+      "loss": 0.4427,
+      "step": 455
+    },
+    {
+      "epoch": 41.45,
+      "learning_rate": 5.402298850574712e-07,
+      "loss": 0.4494,
+      "step": 456
+    },
+    {
+      "epoch": 41.55,
+      "learning_rate": 5.344827586206897e-07,
+      "loss": 0.417,
+      "step": 457
+    },
+    {
+      "epoch": 41.64,
+      "learning_rate": 5.28735632183908e-07,
+      "loss": 0.4539,
+      "step": 458
+    },
+    {
+      "epoch": 41.73,
+      "learning_rate": 5.229885057471265e-07,
+      "loss": 0.4351,
+      "step": 459
+    },
+    {
+      "epoch": 41.82,
+      "learning_rate": 5.172413793103449e-07,
+      "loss": 0.4611,
+      "step": 460
+    },
+    {
+      "epoch": 41.82,
+      "eval_accuracy_safe": 0.6525167597657681,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9718386426019924,
+      "eval_iou_safe": 0.37157282960052246,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9616940748738755,
+      "eval_loss": 0.4025232493877411,
+      "eval_mean_accuracy": 0.8121777011838802,
+      "eval_mean_iou": 0.44442230149146594,
+      "eval_overall_accuracy": 0.962406386190386,
+      "eval_runtime": 9.1949,
+      "eval_samples_per_second": 7.287,
+      "eval_steps_per_second": 0.544,
+      "step": 460
+    },
+    {
+      "epoch": 41.91,
+      "learning_rate": 5.114942528735632e-07,
+      "loss": 0.4298,
+      "step": 461
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 5.057471264367817e-07,
+      "loss": 0.4254,
+      "step": 462
+    },
+    {
+      "epoch": 42.09,
+      "learning_rate": 5e-07,
+      "loss": 0.4288,
+      "step": 463
+    },
+    {
+      "epoch": 42.18,
+      "learning_rate": 4.942528735632184e-07,
+      "loss": 0.423,
+      "step": 464
+    },
+    {
+      "epoch": 42.27,
+      "learning_rate": 4.885057471264368e-07,
+      "loss": 0.4418,
+      "step": 465
+    },
+    {
+      "epoch": 42.36,
+      "learning_rate": 4.827586206896552e-07,
+      "loss": 0.4314,
+      "step": 466
+    },
+    {
+      "epoch": 42.45,
+      "learning_rate": 4.770114942528736e-07,
+      "loss": 0.4333,
+      "step": 467
+    },
+    {
+      "epoch": 42.55,
+      "learning_rate": 4.7126436781609196e-07,
+      "loss": 0.4233,
+      "step": 468
+    },
+    {
+      "epoch": 42.64,
+      "learning_rate": 4.6551724137931035e-07,
+      "loss": 0.4519,
+      "step": 469
+    },
+    {
+      "epoch": 42.73,
+      "learning_rate": 4.597701149425287e-07,
+      "loss": 0.4339,
+      "step": 470
+    },
+    {
+      "epoch": 42.73,
+      "eval_accuracy_safe": 0.6487465352870652,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9726293214969499,
+      "eval_iou_safe": 0.3744354059959327,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9623660982402702,
+      "eval_loss": 0.39508846402168274,
+      "eval_mean_accuracy": 0.8106879283920075,
+      "eval_mean_iou": 0.44560050141206764,
+      "eval_overall_accuracy": 0.9630623433127332,
+      "eval_runtime": 9.952,
+      "eval_samples_per_second": 6.732,
+      "eval_steps_per_second": 0.502,
+      "step": 470
+    },
+    {
+      "epoch": 42.82,
+      "learning_rate": 4.5402298850574714e-07,
+      "loss": 0.4261,
+      "step": 471
+    },
+    {
+      "epoch": 42.91,
+      "learning_rate": 4.4827586206896554e-07,
+      "loss": 0.4671,
+      "step": 472
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 4.425287356321839e-07,
+      "loss": 0.4605,
+      "step": 473
+    },
+    {
+      "epoch": 43.09,
+      "learning_rate": 4.3678160919540233e-07,
+      "loss": 0.416,
+      "step": 474
+    },
+    {
+      "epoch": 43.18,
+      "learning_rate": 4.310344827586207e-07,
+      "loss": 0.4621,
+      "step": 475
+    },
+    {
+      "epoch": 43.27,
+      "learning_rate": 4.252873563218391e-07,
+      "loss": 0.3928,
+      "step": 476
+    },
+    {
+      "epoch": 43.36,
+      "learning_rate": 4.1954022988505747e-07,
+      "loss": 0.4083,
+      "step": 477
+    },
+    {
+      "epoch": 43.45,
+      "learning_rate": 4.1379310344827586e-07,
+      "loss": 0.4524,
+      "step": 478
+    },
+    {
+      "epoch": 43.55,
+      "learning_rate": 4.080459770114943e-07,
+      "loss": 0.4233,
+      "step": 479
+    },
+    {
+      "epoch": 43.64,
+      "learning_rate": 4.0229885057471266e-07,
+      "loss": 0.4361,
+      "step": 480
+    },
+    {
+      "epoch": 43.64,
+      "eval_accuracy_safe": 0.6447700664222574,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9739781749861512,
+      "eval_iou_safe": 0.3768989563134417,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9635797860139171,
+      "eval_loss": 0.3945651650428772,
+      "eval_mean_accuracy": 0.8093741207042043,
+      "eval_mean_iou": 0.4468262474424529,
+      "eval_overall_accuracy": 0.9642538953183303,
+      "eval_runtime": 9.4487,
+      "eval_samples_per_second": 7.091,
+      "eval_steps_per_second": 0.529,
+      "step": 480
+    },
+    {
+      "epoch": 43.73,
+      "learning_rate": 3.9655172413793105e-07,
+      "loss": 0.4263,
+      "step": 481
+    },
+    {
+      "epoch": 43.82,
+      "learning_rate": 3.908045977011494e-07,
+      "loss": 0.3853,
+      "step": 482
+    },
+    {
+      "epoch": 43.91,
+      "learning_rate": 3.8505747126436784e-07,
+      "loss": 0.4272,
+      "step": 483
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 3.7931034482758624e-07,
+      "loss": 0.4668,
+      "step": 484
+    },
+    {
+      "epoch": 44.09,
+      "learning_rate": 3.735632183908046e-07,
+      "loss": 0.4326,
+      "step": 485
+    },
+    {
+      "epoch": 44.18,
+      "learning_rate": 3.67816091954023e-07,
+      "loss": 0.4139,
+      "step": 486
+    },
+    {
+      "epoch": 44.27,
+      "learning_rate": 3.6206896551724143e-07,
+      "loss": 0.4106,
+      "step": 487
+    },
+    {
+      "epoch": 44.36,
+      "learning_rate": 3.5632183908045977e-07,
+      "loss": 0.4596,
+      "step": 488
+    },
+    {
+      "epoch": 44.45,
+      "learning_rate": 3.5057471264367817e-07,
+      "loss": 0.4191,
+      "step": 489
+    },
+    {
+      "epoch": 44.55,
+      "learning_rate": 3.4482758620689656e-07,
+      "loss": 0.4416,
+      "step": 490
+    },
+    {
+      "epoch": 44.55,
+      "eval_accuracy_safe": 0.6447026033053073,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9746319796611832,
+      "eval_iou_safe": 0.37830962816343844,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9642232525873471,
+      "eval_loss": 0.3870936930179596,
+      "eval_mean_accuracy": 0.8096672914832452,
+      "eval_mean_iou": 0.4475109602502618,
+      "eval_overall_accuracy": 0.9648863948992829,
+      "eval_runtime": 10.7764,
+      "eval_samples_per_second": 6.217,
+      "eval_steps_per_second": 0.464,
+      "step": 490
+    },
+    {
+      "epoch": 44.64,
+      "learning_rate": 3.3908045977011496e-07,
+      "loss": 0.3936,
+      "step": 491
+    },
+    {
+      "epoch": 44.73,
+      "learning_rate": 3.333333333333333e-07,
+      "loss": 0.4229,
+      "step": 492
+    },
+    {
+      "epoch": 44.82,
+      "learning_rate": 3.2758620689655175e-07,
+      "loss": 0.4332,
+      "step": 493
+    },
+    {
+      "epoch": 44.91,
+      "learning_rate": 3.2183908045977015e-07,
+      "loss": 0.4375,
+      "step": 494
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 3.1609195402298854e-07,
+      "loss": 0.4713,
+      "step": 495
+    },
+    {
+      "epoch": 45.09,
+      "learning_rate": 3.103448275862069e-07,
+      "loss": 0.4549,
+      "step": 496
+    },
+    {
+      "epoch": 45.18,
+      "learning_rate": 3.045977011494253e-07,
+      "loss": 0.3869,
+      "step": 497
+    },
+    {
+      "epoch": 45.27,
+      "learning_rate": 2.988505747126437e-07,
+      "loss": 0.4323,
+      "step": 498
+    },
+    {
+      "epoch": 45.36,
+      "learning_rate": 2.9310344827586207e-07,
+      "loss": 0.4158,
+      "step": 499
+    },
+    {
+      "epoch": 45.45,
+      "learning_rate": 2.8735632183908047e-07,
+      "loss": 0.4524,
+      "step": 500
+    },
+    {
+      "epoch": 45.45,
+      "eval_accuracy_safe": 0.6588968431116302,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9712046679682527,
+      "eval_iou_safe": 0.3701253070960435,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9612466938703839,
+      "eval_loss": 0.40250322222709656,
+      "eval_mean_accuracy": 0.8150507555399414,
+      "eval_mean_iou": 0.44379066698880915,
+      "eval_overall_accuracy": 0.9619795955828766,
+      "eval_runtime": 9.5625,
+      "eval_samples_per_second": 7.007,
+      "eval_steps_per_second": 0.523,
+      "step": 500
+    },
+    {
+      "epoch": 45.55,
+      "learning_rate": 2.8160919540229886e-07,
+      "loss": 0.4077,
+      "step": 501
+    },
+    {
+      "epoch": 45.64,
+      "learning_rate": 2.7586206896551726e-07,
+      "loss": 0.4103,
+      "step": 502
+    },
+    {
+      "epoch": 45.73,
+      "learning_rate": 2.701149425287356e-07,
+      "loss": 0.4242,
+      "step": 503
+    },
+    {
+      "epoch": 45.82,
+      "learning_rate": 2.64367816091954e-07,
+      "loss": 0.3912,
+      "step": 504
+    },
+    {
+      "epoch": 45.91,
+      "learning_rate": 2.5862068965517245e-07,
+      "loss": 0.4475,
+      "step": 505
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 2.5287356321839084e-07,
+      "loss": 0.4558,
+      "step": 506
+    },
+    {
+      "epoch": 46.09,
+      "learning_rate": 2.471264367816092e-07,
+      "loss": 0.4301,
+      "step": 507
+    },
+    {
+      "epoch": 46.18,
+      "learning_rate": 2.413793103448276e-07,
+      "loss": 0.4541,
+      "step": 508
+    },
+    {
+      "epoch": 46.27,
+      "learning_rate": 2.3563218390804598e-07,
+      "loss": 0.4438,
+      "step": 509
+    },
+    {
+      "epoch": 46.36,
+      "learning_rate": 2.2988505747126435e-07,
+      "loss": 0.4319,
+      "step": 510
+    },
+    {
+      "epoch": 46.36,
+      "eval_accuracy_safe": 0.6730409674596474,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.967265588671203,
+      "eval_iou_safe": 0.3593928995264361,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9577608849356718,
+      "eval_loss": 0.4168514609336853,
+      "eval_mean_accuracy": 0.8201532780654253,
+      "eval_mean_iou": 0.43905126148736934,
+      "eval_overall_accuracy": 0.9585746651265159,
+      "eval_runtime": 10.2031,
+      "eval_samples_per_second": 6.567,
+      "eval_steps_per_second": 0.49,
+      "step": 510
+    },
+    {
+      "epoch": 46.45,
+      "learning_rate": 2.2413793103448277e-07,
+      "loss": 0.4356,
+      "step": 511
+    },
+    {
+      "epoch": 46.55,
+      "learning_rate": 2.1839080459770117e-07,
+      "loss": 0.3902,
+      "step": 512
+    },
+    {
+      "epoch": 46.64,
+      "learning_rate": 2.1264367816091956e-07,
+      "loss": 0.4067,
+      "step": 513
+    },
+    {
+      "epoch": 46.73,
+      "learning_rate": 2.0689655172413793e-07,
+      "loss": 0.3947,
+      "step": 514
+    },
+    {
+      "epoch": 46.82,
+      "learning_rate": 2.0114942528735633e-07,
+      "loss": 0.4378,
+      "step": 515
+    },
+    {
+      "epoch": 46.91,
+      "learning_rate": 1.954022988505747e-07,
+      "loss": 0.3984,
+      "step": 516
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 1.8965517241379312e-07,
+      "loss": 0.4473,
+      "step": 517
+    },
+    {
+      "epoch": 47.09,
+      "learning_rate": 1.839080459770115e-07,
+      "loss": 0.4235,
+      "step": 518
+    },
+    {
+      "epoch": 47.18,
+      "learning_rate": 1.7816091954022988e-07,
+      "loss": 0.4192,
+      "step": 519
+    },
+    {
+      "epoch": 47.27,
+      "learning_rate": 1.7241379310344828e-07,
+      "loss": 0.4224,
+      "step": 520
+    },
+    {
+      "epoch": 47.27,
+      "eval_accuracy_safe": 0.6603020034618217,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9711994464485041,
+      "eval_iou_safe": 0.37161152869427305,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9612871860526603,
+      "eval_loss": 0.39859944581985474,
+      "eval_mean_accuracy": 0.8157507249551629,
+      "eval_mean_iou": 0.4442995715823111,
+      "eval_overall_accuracy": 0.9620160344821304,
+      "eval_runtime": 9.2903,
+      "eval_samples_per_second": 7.212,
+      "eval_steps_per_second": 0.538,
+      "step": 520
+    },
+    {
+      "epoch": 47.36,
+      "learning_rate": 1.0218878248974007e-06,
+      "loss": 0.4479,
+      "step": 521
+    },
+    {
+      "epoch": 47.45,
+      "learning_rate": 1.0177838577291383e-06,
+      "loss": 0.4097,
+      "step": 522
+    },
+    {
+      "epoch": 47.55,
+      "learning_rate": 1.0136798905608756e-06,
+      "loss": 0.4073,
+      "step": 523
+    },
+    {
+      "epoch": 47.64,
+      "learning_rate": 1.009575923392613e-06,
+      "loss": 0.4305,
+      "step": 524
+    },
+    {
+      "epoch": 47.73,
+      "learning_rate": 1.00547195622435e-06,
+      "loss": 0.4537,
+      "step": 525
+    },
+    {
+      "epoch": 47.82,
+      "learning_rate": 1.0013679890560876e-06,
+      "loss": 0.4054,
+      "step": 526
+    },
+    {
+      "epoch": 47.91,
+      "learning_rate": 9.97264021887825e-07,
+      "loss": 0.3876,
+      "step": 527
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 9.931600547195623e-07,
+      "loss": 0.4008,
+      "step": 528
+    },
+    {
+      "epoch": 48.09,
+      "learning_rate": 9.890560875512995e-07,
+      "loss": 0.4183,
+      "step": 529
+    },
+    {
+      "epoch": 48.18,
+      "learning_rate": 9.84952120383037e-07,
+      "loss": 0.4333,
+      "step": 530
+    },
+    {
+      "epoch": 48.18,
+      "eval_accuracy_safe": 0.6649550310137586,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9702578714996897,
+      "eval_iou_safe": 0.3723805021059608,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9604921504914465,
+      "eval_loss": 0.4038476347923279,
+      "eval_mean_accuracy": 0.8176064512567242,
+      "eval_mean_iou": 0.44429088419913576,
+      "eval_overall_accuracy": 0.9612397151206856,
+      "eval_runtime": 8.6363,
+      "eval_samples_per_second": 7.758,
+      "eval_steps_per_second": 0.579,
+      "step": 530
+    },
+    {
+      "epoch": 48.27,
+      "learning_rate": 9.808481532147744e-07,
+      "loss": 0.414,
+      "step": 531
+    },
+    {
+      "epoch": 48.36,
+      "learning_rate": 9.767441860465117e-07,
+      "loss": 0.4344,
+      "step": 532
+    },
+    {
+      "epoch": 48.45,
+      "learning_rate": 9.726402188782489e-07,
+      "loss": 0.4448,
+      "step": 533
+    },
+    {
+      "epoch": 48.55,
+      "learning_rate": 9.685362517099864e-07,
+      "loss": 0.4283,
+      "step": 534
+    },
+    {
+      "epoch": 48.64,
+      "learning_rate": 9.644322845417238e-07,
+      "loss": 0.3871,
+      "step": 535
+    },
+    {
+      "epoch": 48.73,
+      "learning_rate": 9.603283173734611e-07,
+      "loss": 0.3983,
+      "step": 536
+    },
+    {
+      "epoch": 48.82,
+      "learning_rate": 9.562243502051983e-07,
+      "loss": 0.3946,
+      "step": 537
+    },
+    {
+      "epoch": 48.91,
+      "learning_rate": 9.521203830369357e-07,
+      "loss": 0.4091,
+      "step": 538
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 9.480164158686731e-07,
+      "loss": 0.438,
+      "step": 539
+    },
+    {
+      "epoch": 49.09,
+      "learning_rate": 9.439124487004105e-07,
+      "loss": 0.3916,
+      "step": 540
+    },
+    {
+      "epoch": 49.09,
+      "eval_accuracy_safe": 0.6624396205103296,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9724313730965947,
+      "eval_iou_safe": 0.3781414596199825,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9625620999872877,
+      "eval_loss": 0.39684730768203735,
+      "eval_mean_accuracy": 0.8174354968034621,
+      "eval_mean_iou": 0.4469011865357568,
+      "eval_overall_accuracy": 0.9632747137724463,
+      "eval_runtime": 8.8446,
+      "eval_samples_per_second": 7.575,
+      "eval_steps_per_second": 0.565,
+      "step": 540
+    },
+    {
+      "epoch": 49.18,
+      "learning_rate": 9.398084815321478e-07,
+      "loss": 0.4174,
+      "step": 541
+    },
+    {
+      "epoch": 49.27,
+      "learning_rate": 9.357045143638851e-07,
+      "loss": 0.3837,
+      "step": 542
+    },
+    {
+      "epoch": 49.36,
+      "learning_rate": 9.316005471956224e-07,
+      "loss": 0.4472,
+      "step": 543
+    },
+    {
+      "epoch": 49.45,
+      "learning_rate": 9.274965800273599e-07,
+      "loss": 0.3876,
+      "step": 544
+    },
+    {
+      "epoch": 49.55,
+      "learning_rate": 9.233926128590971e-07,
+      "loss": 0.3935,
+      "step": 545
+    },
+    {
+      "epoch": 49.64,
+      "learning_rate": 9.192886456908345e-07,
+      "loss": 0.4331,
+      "step": 546
+    },
+    {
+      "epoch": 49.73,
+      "learning_rate": 9.151846785225718e-07,
+      "loss": 0.3941,
+      "step": 547
+    },
+    {
+      "epoch": 49.82,
+      "learning_rate": 9.110807113543093e-07,
+      "loss": 0.3938,
+      "step": 548
+    },
+    {
+      "epoch": 49.91,
+      "learning_rate": 9.069767441860465e-07,
+      "loss": 0.4009,
+      "step": 549
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 9.028727770177839e-07,
+      "loss": 0.4803,
+      "step": 550
+    },
+    {
+      "epoch": 50.0,
+      "eval_accuracy_safe": 0.6680390592171965,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9725606203775616,
+      "eval_iou_safe": 0.38091072150808025,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.962854988299414,
+      "eval_loss": 0.39420995116233826,
+      "eval_mean_accuracy": 0.8202998397973791,
+      "eval_mean_iou": 0.44792190326916476,
+      "eval_overall_accuracy": 0.9635655417371152,
+      "eval_runtime": 8.6063,
+      "eval_samples_per_second": 7.785,
+      "eval_steps_per_second": 0.581,
+      "step": 550
+    },
+    {
+      "epoch": 50.09,
+      "learning_rate": 8.987688098495212e-07,
+      "loss": 0.4009,
+      "step": 551
+    },
+    {
+      "epoch": 50.18,
+      "learning_rate": 8.946648426812587e-07,
+      "loss": 0.4059,
+      "step": 552
+    },
+    {
+      "epoch": 50.27,
+      "learning_rate": 8.905608755129958e-07,
+      "loss": 0.412,
+      "step": 553
+    },
+    {
+      "epoch": 50.36,
+      "learning_rate": 8.864569083447333e-07,
+      "loss": 0.3805,
+      "step": 554
+    },
+    {
+      "epoch": 50.45,
+      "learning_rate": 8.823529411764706e-07,
+      "loss": 0.407,
+      "step": 555
+    },
+    {
+      "epoch": 50.55,
+      "learning_rate": 8.782489740082081e-07,
+      "loss": 0.3856,
+      "step": 556
+    },
+    {
+      "epoch": 50.64,
+      "learning_rate": 8.741450068399452e-07,
+      "loss": 0.413,
+      "step": 557
+    },
+    {
+      "epoch": 50.73,
+      "learning_rate": 8.700410396716827e-07,
+      "loss": 0.3965,
+      "step": 558
+    },
+    {
+      "epoch": 50.82,
+      "learning_rate": 8.6593707250342e-07,
+      "loss": 0.4005,
+      "step": 559
+    },
+    {
+      "epoch": 50.91,
+      "learning_rate": 8.618331053351575e-07,
+      "loss": 0.3543,
+      "step": 560
+    },
+    {
+      "epoch": 50.91,
+      "eval_accuracy_safe": 0.6472642742317879,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9777157857571726,
+      "eval_iou_safe": 0.3951853111506791,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9673436905478155,
+      "eval_loss": 0.3696909546852112,
+      "eval_mean_accuracy": 0.8124900299944802,
+      "eval_mean_iou": 0.4541763338994982,
+      "eval_overall_accuracy": 0.9679547779595674,
+      "eval_runtime": 9.4902,
+      "eval_samples_per_second": 7.06,
+      "eval_steps_per_second": 0.527,
+      "step": 560
+    },
+    {
+      "epoch": 51.0,
+      "learning_rate": 8.577291381668946e-07,
+      "loss": 0.3975,
+      "step": 561
+    },
+    {
+      "epoch": 51.09,
+      "learning_rate": 8.53625170998632e-07,
+      "loss": 0.3652,
+      "step": 562
+    },
+    {
+      "epoch": 51.18,
+      "learning_rate": 8.495212038303694e-07,
+      "loss": 0.3648,
+      "step": 563
+    },
+    {
+      "epoch": 51.27,
+      "learning_rate": 8.454172366621067e-07,
+      "loss": 0.4468,
+      "step": 564
+    },
+    {
+      "epoch": 51.36,
+      "learning_rate": 8.41313269493844e-07,
+      "loss": 0.4287,
+      "step": 565
+    },
+    {
+      "epoch": 51.45,
+      "learning_rate": 8.372093023255814e-07,
+      "loss": 0.3904,
+      "step": 566
+    },
+    {
+      "epoch": 51.55,
+      "learning_rate": 8.331053351573188e-07,
+      "loss": 0.3856,
+      "step": 567
+    },
+    {
+      "epoch": 51.64,
+      "learning_rate": 8.290013679890561e-07,
+      "loss": 0.407,
+      "step": 568
+    },
+    {
+      "epoch": 51.73,
+      "learning_rate": 8.248974008207934e-07,
+      "loss": 0.3889,
+      "step": 569
+    },
+    {
+      "epoch": 51.82,
+      "learning_rate": 8.207934336525308e-07,
+      "loss": 0.3684,
+      "step": 570
+    },
+    {
+      "epoch": 51.82,
+      "eval_accuracy_safe": 0.6514681901765992,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9772261362760333,
+      "eval_iou_safe": 0.3951423109973169,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9669824966129977,
+      "eval_loss": 0.3708346486091614,
+      "eval_mean_accuracy": 0.8143471632263162,
+      "eval_mean_iou": 0.45404160253677156,
+      "eval_overall_accuracy": 0.9676037688753498,
+      "eval_runtime": 10.3386,
+      "eval_samples_per_second": 6.481,
+      "eval_steps_per_second": 0.484,
+      "step": 570
+    },
+    {
+      "epoch": 51.91,
+      "learning_rate": 8.166894664842682e-07,
+      "loss": 0.413,
+      "step": 571
+    },
+    {
+      "epoch": 52.0,
+      "learning_rate": 8.125854993160055e-07,
+      "loss": 0.3516,
+      "step": 572
+    },
+    {
+      "epoch": 52.09,
+      "learning_rate": 8.084815321477428e-07,
+      "loss": 0.3804,
+      "step": 573
+    },
+    {
+      "epoch": 52.18,
+      "learning_rate": 8.043775649794802e-07,
+      "loss": 0.3756,
+      "step": 574
+    },
+    {
+      "epoch": 52.27,
+      "learning_rate": 8.002735978112176e-07,
+      "loss": 0.412,
+      "step": 575
+    },
+    {
+      "epoch": 52.36,
+      "learning_rate": 7.961696306429549e-07,
+      "loss": 0.4109,
+      "step": 576
+    },
+    {
+      "epoch": 52.45,
+      "learning_rate": 7.920656634746922e-07,
+      "loss": 0.3654,
+      "step": 577
+    },
+    {
+      "epoch": 52.55,
+      "learning_rate": 7.879616963064296e-07,
+      "loss": 0.3624,
+      "step": 578
+    },
+    {
+      "epoch": 52.64,
+      "learning_rate": 7.83857729138167e-07,
+      "loss": 0.3876,
+      "step": 579
+    },
+    {
+      "epoch": 52.73,
+      "learning_rate": 7.797537619699043e-07,
+      "loss": 0.4004,
+      "step": 580
+    },
+    {
+      "epoch": 52.73,
+      "eval_accuracy_safe": 0.64372727938597,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9793208457266203,
+      "eval_iou_safe": 0.401423873377919,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.968824985839696,
+      "eval_loss": 0.3584851026535034,
+      "eval_mean_accuracy": 0.8115240625562952,
+      "eval_mean_iou": 0.45674961973920497,
+      "eval_overall_accuracy": 0.9694079498746502,
+      "eval_runtime": 9.5054,
+      "eval_samples_per_second": 7.049,
+      "eval_steps_per_second": 0.526,
+      "step": 580
+    },
+    {
+      "epoch": 52.82,
+      "learning_rate": 7.756497948016415e-07,
+      "loss": 0.3849,
+      "step": 581
+    },
+    {
+      "epoch": 52.91,
+      "learning_rate": 7.715458276333789e-07,
+      "loss": 0.3474,
+      "step": 582
+    },
+    {
+      "epoch": 53.0,
+      "learning_rate": 7.674418604651163e-07,
+      "loss": 0.3701,
+      "step": 583
+    },
+    {
+      "epoch": 53.09,
+      "learning_rate": 7.633378932968537e-07,
+      "loss": 0.3447,
+      "step": 584
+    },
+    {
+      "epoch": 53.18,
+      "learning_rate": 7.592339261285909e-07,
+      "loss": 0.4049,
+      "step": 585
+    },
+    {
+      "epoch": 53.27,
+      "learning_rate": 7.551299589603283e-07,
+      "loss": 0.4026,
+      "step": 586
+    },
+    {
+      "epoch": 53.36,
+      "learning_rate": 7.510259917920657e-07,
+      "loss": 0.3855,
+      "step": 587
+    },
+    {
+      "epoch": 53.45,
+      "learning_rate": 7.46922024623803e-07,
+      "loss": 0.372,
+      "step": 588
+    },
+    {
+      "epoch": 53.55,
+      "learning_rate": 7.428180574555404e-07,
+      "loss": 0.358,
+      "step": 589
+    },
+    {
+      "epoch": 53.64,
+      "learning_rate": 7.387140902872777e-07,
+      "loss": 0.3656,
+      "step": 590
+    },
+    {
+      "epoch": 53.64,
+      "eval_accuracy_safe": 0.6558667854017525,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9780078388505241,
+      "eval_iou_safe": 0.40104638732360337,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9678817565011693,
+      "eval_loss": 0.3653949499130249,
+      "eval_mean_accuracy": 0.8169373121261383,
+      "eval_mean_iou": 0.4563093812749242,
+      "eval_overall_accuracy": 0.96849230865934,
+      "eval_runtime": 10.8561,
+      "eval_samples_per_second": 6.172,
+      "eval_steps_per_second": 0.461,
+      "step": 590
+    },
+    {
+      "epoch": 53.73,
+      "learning_rate": 7.346101231190151e-07,
+      "loss": 0.4163,
+      "step": 591
+    },
+    {
+      "epoch": 53.82,
+      "learning_rate": 7.305061559507524e-07,
+      "loss": 0.3652,
+      "step": 592
+    },
+    {
+      "epoch": 53.91,
+      "learning_rate": 7.264021887824898e-07,
+      "loss": 0.3957,
+      "step": 593
+    },
+    {
+      "epoch": 54.0,
+      "learning_rate": 7.222982216142271e-07,
+      "loss": 0.4365,
+      "step": 594
+    },
+    {
+      "epoch": 54.09,
+      "learning_rate": 7.181942544459645e-07,
+      "loss": 0.4133,
+      "step": 595
+    },
+    {
+      "epoch": 54.18,
+      "learning_rate": 7.140902872777018e-07,
+      "loss": 0.4297,
+      "step": 596
+    },
+    {
+      "epoch": 54.27,
+      "learning_rate": 7.099863201094392e-07,
+      "loss": 0.327,
+      "step": 597
+    },
+    {
+      "epoch": 54.36,
+      "learning_rate": 7.058823529411765e-07,
+      "loss": 0.3749,
+      "step": 598
+    },
+    {
+      "epoch": 54.45,
+      "learning_rate": 7.017783857729139e-07,
+      "loss": 0.3516,
+      "step": 599
+    },
+    {
+      "epoch": 54.55,
+      "learning_rate": 6.976744186046511e-07,
+      "loss": 0.3918,
+      "step": 600
+    },
+    {
+      "epoch": 54.55,
+      "eval_accuracy_safe": 0.6432222697676571,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.980886949638618,
+      "eval_iou_safe": 0.41152118741259774,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9703598872160661,
+      "eval_loss": 0.3526640236377716,
+      "eval_mean_accuracy": 0.8120546097031376,
+      "eval_mean_iou": 0.46062702487622126,
+      "eval_overall_accuracy": 0.9709128764138293,
+      "eval_runtime": 9.0586,
+      "eval_samples_per_second": 7.396,
+      "eval_steps_per_second": 0.552,
+      "step": 600
+    },
+    {
+      "epoch": 54.64,
+      "learning_rate": 6.935704514363886e-07,
+      "loss": 0.3599,
+      "step": 601
+    },
+    {
+      "epoch": 54.73,
+      "learning_rate": 6.894664842681258e-07,
+      "loss": 0.3603,
+      "step": 602
+    },
+    {
+      "epoch": 54.82,
+      "learning_rate": 6.853625170998633e-07,
+      "loss": 0.3782,
+      "step": 603
+    },
+    {
+      "epoch": 54.91,
+      "learning_rate": 6.812585499316005e-07,
+      "loss": 0.3599,
+      "step": 604
+    },
+    {
+      "epoch": 55.0,
+      "learning_rate": 6.77154582763338e-07,
+      "loss": 0.4199,
+      "step": 605
+    },
+    {
+      "epoch": 55.09,
+      "learning_rate": 6.730506155950752e-07,
+      "loss": 0.3504,
+      "step": 606
+    },
+    {
+      "epoch": 55.18,
+      "learning_rate": 6.689466484268127e-07,
+      "loss": 0.3583,
+      "step": 607
+    },
+    {
+      "epoch": 55.27,
+      "learning_rate": 6.648426812585499e-07,
+      "loss": 0.3537,
+      "step": 608
+    },
+    {
+      "epoch": 55.36,
+      "learning_rate": 6.607387140902874e-07,
+      "loss": 0.4038,
+      "step": 609
+    },
+    {
+      "epoch": 55.45,
+      "learning_rate": 6.566347469220246e-07,
+      "loss": 0.3741,
+      "step": 610
+    },
+    {
+      "epoch": 55.45,
+      "eval_accuracy_safe": 0.6392901338082737,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9826956488782591,
+      "eval_iou_safe": 0.418450346516478,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9720286712517688,
+      "eval_loss": 0.3361377418041229,
+      "eval_mean_accuracy": 0.8109928913432665,
+      "eval_mean_iou": 0.46349300592274895,
+      "eval_overall_accuracy": 0.9725520005866662,
+      "eval_runtime": 9.4221,
+      "eval_samples_per_second": 7.111,
+      "eval_steps_per_second": 0.531,
+      "step": 610
+    },
+    {
+      "epoch": 55.55,
+      "learning_rate": 6.52530779753762e-07,
+      "loss": 0.3866,
+      "step": 611
+    },
+    {
+      "epoch": 55.64,
+      "learning_rate": 6.484268125854993e-07,
+      "loss": 0.3333,
+      "step": 612
+    },
+    {
+      "epoch": 55.73,
+      "learning_rate": 6.443228454172367e-07,
+      "loss": 0.3843,
+      "step": 613
+    },
+    {
+      "epoch": 55.82,
+      "learning_rate": 6.40218878248974e-07,
+      "loss": 0.3664,
+      "step": 614
+    },
+    {
+      "epoch": 55.91,
+      "learning_rate": 6.361149110807114e-07,
+      "loss": 0.3787,
+      "step": 615
+    },
+    {
+      "epoch": 56.0,
+      "learning_rate": 6.320109439124487e-07,
+      "loss": 0.4437,
+      "step": 616
+    },
+    {
+      "epoch": 56.09,
+      "learning_rate": 6.27906976744186e-07,
+      "loss": 0.3822,
+      "step": 617
+    },
+    {
+      "epoch": 56.18,
+      "learning_rate": 6.238030095759234e-07,
+      "loss": 0.3639,
+      "step": 618
+    },
+    {
+      "epoch": 56.27,
+      "learning_rate": 6.196990424076607e-07,
+      "loss": 0.3971,
+      "step": 619
+    },
+    {
+      "epoch": 56.36,
+      "learning_rate": 6.155950752393981e-07,
+      "loss": 0.3656,
+      "step": 620
+    },
+    {
+      "epoch": 56.36,
+      "eval_accuracy_safe": 0.6540472087617242,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9806777955048699,
+      "eval_iou_safe": 0.41467641232391395,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9704675533352136,
+      "eval_loss": 0.3472718894481659,
+      "eval_mean_accuracy": 0.8173625021332971,
+      "eval_mean_iou": 0.46171465521970917,
+      "eval_overall_accuracy": 0.9710296516987815,
+      "eval_runtime": 9.6643,
+      "eval_samples_per_second": 6.933,
+      "eval_steps_per_second": 0.517,
+      "step": 620
+    },
+    {
+      "epoch": 56.45,
+      "learning_rate": 6.114911080711354e-07,
+      "loss": 0.3344,
+      "step": 621
+    },
+    {
+      "epoch": 56.55,
+      "learning_rate": 6.073871409028728e-07,
+      "loss": 0.3539,
+      "step": 622
+    },
+    {
+      "epoch": 56.64,
+      "learning_rate": 6.032831737346101e-07,
+      "loss": 0.3563,
+      "step": 623
+    },
+    {
+      "epoch": 56.73,
+      "learning_rate": 5.991792065663475e-07,
+      "loss": 0.3685,
+      "step": 624
+    },
+    {
+      "epoch": 56.82,
+      "learning_rate": 5.950752393980848e-07,
+      "loss": 0.3558,
+      "step": 625
+    },
+    {
+      "epoch": 56.91,
+      "learning_rate": 5.909712722298222e-07,
+      "loss": 0.3618,
+      "step": 626
+    },
+    {
+      "epoch": 57.0,
+      "learning_rate": 5.868673050615595e-07,
+      "loss": 0.422,
+      "step": 627
+    },
+    {
+      "epoch": 57.09,
+      "learning_rate": 5.827633378932969e-07,
+      "loss": 0.3629,
+      "step": 628
+    },
+    {
+      "epoch": 57.18,
+      "learning_rate": 5.786593707250342e-07,
+      "loss": 0.348,
+      "step": 629
+    },
+    {
+      "epoch": 57.27,
+      "learning_rate": 5.745554035567716e-07,
+      "loss": 0.3341,
+      "step": 630
+    },
+    {
+      "epoch": 57.27,
+      "eval_accuracy_safe": 0.6257917278653513,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9845255862094618,
+      "eval_iou_safe": 0.4246501164098674,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9734443258687906,
+      "eval_loss": 0.3334999680519104,
+      "eval_mean_accuracy": 0.8051586570374065,
+      "eval_mean_iou": 0.4660314807595527,
+      "eval_overall_accuracy": 0.973929163235337,
+      "eval_runtime": 9.6696,
+      "eval_samples_per_second": 6.929,
+      "eval_steps_per_second": 0.517,
+      "step": 630
+    },
+    {
+      "epoch": 57.36,
+      "learning_rate": 5.704514363885089e-07,
+      "loss": 0.3766,
+      "step": 631
+    },
+    {
+      "epoch": 57.45,
+      "learning_rate": 5.663474692202463e-07,
+      "loss": 0.3456,
+      "step": 632
+    },
+    {
+      "epoch": 57.55,
+      "learning_rate": 5.622435020519836e-07,
+      "loss": 0.3709,
+      "step": 633
+    },
+    {
+      "epoch": 57.64,
+      "learning_rate": 5.58139534883721e-07,
+      "loss": 0.355,
+      "step": 634
+    },
+    {
+      "epoch": 57.73,
+      "learning_rate": 5.540355677154583e-07,
+      "loss": 0.3679,
+      "step": 635
+    },
+    {
+      "epoch": 57.82,
+      "learning_rate": 5.499316005471957e-07,
+      "loss": 0.3541,
+      "step": 636
+    },
+    {
+      "epoch": 57.91,
+      "learning_rate": 5.45827633378933e-07,
+      "loss": 0.3495,
+      "step": 637
+    },
+    {
+      "epoch": 58.0,
+      "learning_rate": 5.417236662106703e-07,
+      "loss": 0.384,
+      "step": 638
+    },
+    {
+      "epoch": 58.09,
+      "learning_rate": 5.376196990424077e-07,
+      "loss": 0.393,
+      "step": 639
+    },
+    {
+      "epoch": 58.18,
+      "learning_rate": 5.33515731874145e-07,
+      "loss": 0.3669,
+      "step": 640
+    },
+    {
+      "epoch": 58.18,
+      "eval_accuracy_safe": 0.6495040497145347,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9815480879088024,
+      "eval_iou_safe": 0.41900261749180867,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.971195827459642,
+      "eval_loss": 0.33950886130332947,
+      "eval_mean_accuracy": 0.8155260688116686,
+      "eval_mean_iou": 0.46339948165048356,
+      "eval_overall_accuracy": 0.971740039426889,
+      "eval_runtime": 9.4613,
+      "eval_samples_per_second": 7.082,
+      "eval_steps_per_second": 0.528,
+      "step": 640
+    },
+    {
+      "epoch": 58.27,
+      "learning_rate": 5.294117647058824e-07,
+      "loss": 0.3638,
+      "step": 641
+    },
+    {
+      "epoch": 58.36,
+      "learning_rate": 5.253077975376197e-07,
+      "loss": 0.3462,
+      "step": 642
+    },
+    {
+      "epoch": 58.45,
+      "learning_rate": 5.212038303693571e-07,
+      "loss": 0.3637,
+      "step": 643
+    },
+    {
+      "epoch": 58.55,
+      "learning_rate": 5.170998632010944e-07,
+      "loss": 0.3705,
+      "step": 644
+    },
+    {
+      "epoch": 58.64,
+      "learning_rate": 5.129958960328318e-07,
+      "loss": 0.3323,
+      "step": 645
+    },
+    {
+      "epoch": 58.73,
+      "learning_rate": 5.088919288645691e-07,
+      "loss": 0.336,
+      "step": 646
+    },
+    {
+      "epoch": 58.82,
+      "learning_rate": 5.047879616963065e-07,
+      "loss": 0.3403,
+      "step": 647
+    },
+    {
+      "epoch": 58.91,
+      "learning_rate": 5.006839945280438e-07,
+      "loss": 0.3547,
+      "step": 648
+    },
+    {
+      "epoch": 59.0,
+      "learning_rate": 4.965800273597812e-07,
+      "loss": 0.3592,
+      "step": 649
+    },
+    {
+      "epoch": 59.09,
+      "learning_rate": 4.924760601915185e-07,
+      "loss": 0.3347,
+      "step": 650
+    },
+    {
+      "epoch": 59.09,
+      "eval_accuracy_safe": 0.66117516894692,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9799751784205032,
+      "eval_iou_safe": 0.41742094387885487,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9699836589351479,
+      "eval_loss": 0.3415713906288147,
+      "eval_mean_accuracy": 0.8205751736837116,
+      "eval_mean_iou": 0.46246820093800095,
+      "eval_overall_accuracy": 0.9705583373112465,
+      "eval_runtime": 11.7726,
+      "eval_samples_per_second": 5.691,
+      "eval_steps_per_second": 0.425,
+      "step": 650
+    },
+    {
+      "epoch": 59.18,
+      "learning_rate": 4.883720930232559e-07,
+      "loss": 0.3449,
+      "step": 651
+    },
+    {
+      "epoch": 59.27,
+      "learning_rate": 4.842681258549932e-07,
+      "loss": 0.3679,
+      "step": 652
+    },
+    {
+      "epoch": 59.36,
+      "learning_rate": 4.801641586867306e-07,
+      "loss": 0.3559,
+      "step": 653
+    },
+    {
+      "epoch": 59.45,
+      "learning_rate": 4.7606019151846786e-07,
+      "loss": 0.3403,
+      "step": 654
+    },
+    {
+      "epoch": 59.55,
+      "learning_rate": 4.7195622435020526e-07,
+      "loss": 0.353,
+      "step": 655
+    },
+    {
+      "epoch": 59.64,
+      "learning_rate": 4.6785225718194255e-07,
+      "loss": 0.3617,
+      "step": 656
+    },
+    {
+      "epoch": 59.73,
+      "learning_rate": 4.6374829001367995e-07,
+      "loss": 0.3504,
+      "step": 657
+    },
+    {
+      "epoch": 59.82,
+      "learning_rate": 4.5964432284541724e-07,
+      "loss": 0.3331,
+      "step": 658
+    },
+    {
+      "epoch": 59.91,
+      "learning_rate": 4.5554035567715465e-07,
+      "loss": 0.38,
+      "step": 659
+    },
+    {
+      "epoch": 60.0,
+      "learning_rate": 4.5143638850889194e-07,
+      "loss": 0.4287,
+      "step": 660
+    },
+    {
+      "epoch": 60.0,
+      "eval_accuracy_safe": 0.6673470803890501,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9796976751799341,
+      "eval_iou_safe": 0.4184647553664668,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9698893282294503,
+      "eval_loss": 0.34193894267082214,
+      "eval_mean_accuracy": 0.8235223777844921,
+      "eval_mean_iou": 0.4627846945319724,
+      "eval_overall_accuracy": 0.9704713394392782,
+      "eval_runtime": 9.3579,
+      "eval_samples_per_second": 7.16,
+      "eval_steps_per_second": 0.534,
+      "step": 660
+    },
+    {
+      "epoch": 60.09,
+      "learning_rate": 4.4733242134062934e-07,
+      "loss": 0.3861,
+      "step": 661
+    },
+    {
+      "epoch": 60.18,
+      "learning_rate": 4.4322845417236663e-07,
+      "loss": 0.37,
+      "step": 662
+    },
+    {
+      "epoch": 60.27,
+      "learning_rate": 4.3912448700410404e-07,
+      "loss": 0.3201,
+      "step": 663
+    },
+    {
+      "epoch": 60.36,
+      "learning_rate": 4.3502051983584133e-07,
+      "loss": 0.3348,
+      "step": 664
+    },
+    {
+      "epoch": 60.45,
+      "learning_rate": 4.3091655266757873e-07,
+      "loss": 0.3363,
+      "step": 665
+    },
+    {
+      "epoch": 60.55,
+      "learning_rate": 4.26812585499316e-07,
+      "loss": 0.3438,
+      "step": 666
+    },
+    {
+      "epoch": 60.64,
+      "learning_rate": 4.2270861833105337e-07,
+      "loss": 0.3656,
+      "step": 667
+    },
+    {
+      "epoch": 60.73,
+      "learning_rate": 4.186046511627907e-07,
+      "loss": 0.3284,
+      "step": 668
+    },
+    {
+      "epoch": 60.82,
+      "learning_rate": 4.1450068399452807e-07,
+      "loss": 0.3491,
+      "step": 669
+    },
+    {
+      "epoch": 60.91,
+      "learning_rate": 4.103967168262654e-07,
+      "loss": 0.3838,
+      "step": 670
+    },
+    {
+      "epoch": 60.91,
+      "eval_accuracy_safe": 0.661071082995054,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9812073984123999,
+      "eval_iou_safe": 0.42273980606218975,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9711982397733865,
+      "eval_loss": 0.3380735516548157,
+      "eval_mean_accuracy": 0.821139240703727,
+      "eval_mean_iou": 0.4646460152785254,
+      "eval_overall_accuracy": 0.9717510849682253,
+      "eval_runtime": 10.5907,
+      "eval_samples_per_second": 6.326,
+      "eval_steps_per_second": 0.472,
+      "step": 670
+    },
+    {
+      "epoch": 61.0,
+      "learning_rate": 4.0629274965800276e-07,
+      "loss": 0.3348,
+      "step": 671
+    },
+    {
+      "epoch": 61.09,
+      "learning_rate": 4.021887824897401e-07,
+      "loss": 0.3757,
+      "step": 672
+    },
+    {
+      "epoch": 61.18,
+      "learning_rate": 3.9808481532147746e-07,
+      "loss": 0.3252,
+      "step": 673
+    },
+    {
+      "epoch": 61.27,
+      "learning_rate": 3.939808481532148e-07,
+      "loss": 0.3364,
+      "step": 674
+    },
+    {
+      "epoch": 61.36,
+      "learning_rate": 3.8987688098495215e-07,
+      "loss": 0.377,
+      "step": 675
+    },
+    {
+      "epoch": 61.45,
+      "learning_rate": 3.8577291381668945e-07,
+      "loss": 0.3288,
+      "step": 676
+    },
+    {
+      "epoch": 61.55,
+      "learning_rate": 3.8166894664842685e-07,
+      "loss": 0.362,
+      "step": 677
+    },
+    {
+      "epoch": 61.64,
+      "learning_rate": 3.7756497948016414e-07,
+      "loss": 0.3206,
+      "step": 678
+    },
+    {
+      "epoch": 61.73,
+      "learning_rate": 3.734610123119015e-07,
+      "loss": 0.3152,
+      "step": 679
+    },
+    {
+      "epoch": 61.82,
+      "learning_rate": 3.6935704514363884e-07,
+      "loss": 0.352,
+      "step": 680
+    },
+    {
+      "epoch": 61.82,
+      "eval_accuracy_safe": 0.6406548162882949,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9844651573854055,
+      "eval_iou_safe": 0.4317581022923749,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9738170049692355,
+      "eval_loss": 0.3216310143470764,
+      "eval_mean_accuracy": 0.8125599868368503,
+      "eval_mean_iou": 0.46852503575387017,
+      "eval_overall_accuracy": 0.9743095511820778,
+      "eval_runtime": 9.4908,
+      "eval_samples_per_second": 7.059,
+      "eval_steps_per_second": 0.527,
+      "step": 680
+    },
+    {
+      "epoch": 61.91,
+      "learning_rate": 3.652530779753762e-07,
+      "loss": 0.3738,
+      "step": 681
+    },
+    {
+      "epoch": 62.0,
+      "learning_rate": 3.6114911080711353e-07,
+      "loss": 0.3422,
+      "step": 682
+    },
+    {
+      "epoch": 62.09,
+      "learning_rate": 3.570451436388509e-07,
+      "loss": 0.3377,
+      "step": 683
+    },
+    {
+      "epoch": 62.18,
+      "learning_rate": 3.529411764705882e-07,
+      "loss": 0.3689,
+      "step": 684
+    },
+    {
+      "epoch": 62.27,
+      "learning_rate": 3.4883720930232557e-07,
+      "loss": 0.3431,
+      "step": 685
+    },
+    {
+      "epoch": 62.36,
+      "learning_rate": 3.447332421340629e-07,
+      "loss": 0.3775,
+      "step": 686
+    },
+    {
+      "epoch": 62.45,
+      "learning_rate": 3.4062927496580027e-07,
+      "loss": 0.3317,
+      "step": 687
+    },
+    {
+      "epoch": 62.55,
+      "learning_rate": 3.365253077975376e-07,
+      "loss": 0.3552,
+      "step": 688
+    },
+    {
+      "epoch": 62.64,
+      "learning_rate": 3.3242134062927496e-07,
+      "loss": 0.3333,
+      "step": 689
+    },
+    {
+      "epoch": 62.73,
+      "learning_rate": 3.283173734610123e-07,
+      "loss": 0.3343,
+      "step": 690
+    },
+    {
+      "epoch": 62.73,
+      "eval_accuracy_safe": 0.6499146109691173,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9836740091403583,
+      "eval_iou_safe": 0.4311119919192952,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9733058809214871,
+      "eval_loss": 0.3275364339351654,
+      "eval_mean_accuracy": 0.8167943100547378,
+      "eval_mean_iou": 0.46813929094692747,
+      "eval_overall_accuracy": 0.9738152916751691,
+      "eval_runtime": 9.716,
+      "eval_samples_per_second": 6.896,
+      "eval_steps_per_second": 0.515,
+      "step": 690
+    },
+    {
+      "epoch": 62.82,
+      "learning_rate": 3.2421340629274966e-07,
+      "loss": 0.3756,
+      "step": 691
+    },
+    {
+      "epoch": 62.91,
+      "learning_rate": 3.20109439124487e-07,
+      "loss": 0.3231,
+      "step": 692
+    },
+    {
+      "epoch": 63.0,
+      "learning_rate": 3.1600547195622435e-07,
+      "loss": 0.3676,
+      "step": 693
+    },
+    {
+      "epoch": 63.09,
+      "learning_rate": 3.119015047879617e-07,
+      "loss": 0.297,
+      "step": 694
+    },
+    {
+      "epoch": 63.18,
+      "learning_rate": 3.0779753761969905e-07,
+      "loss": 0.3796,
+      "step": 695
+    },
+    {
+      "epoch": 63.27,
+      "learning_rate": 3.036935704514364e-07,
+      "loss": 0.3462,
+      "step": 696
+    },
+    {
+      "epoch": 63.36,
+      "learning_rate": 2.9958960328317374e-07,
+      "loss": 0.3536,
+      "step": 697
+    },
+    {
+      "epoch": 63.45,
+      "learning_rate": 2.954856361149111e-07,
+      "loss": 0.3426,
+      "step": 698
+    },
+    {
+      "epoch": 63.55,
+      "learning_rate": 2.9138166894664844e-07,
+      "loss": 0.332,
+      "step": 699
+    },
+    {
+      "epoch": 63.64,
+      "learning_rate": 2.872777017783858e-07,
+      "loss": 0.3443,
+      "step": 700
+    },
+    {
+      "epoch": 63.64,
+      "eval_accuracy_safe": 0.6527538444339074,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.983601025201401,
+      "eval_iou_safe": 0.4323750817126982,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9733172883777333,
+      "eval_loss": 0.3273077607154846,
+      "eval_mean_accuracy": 0.8181774348176543,
+      "eval_mean_iou": 0.4685641233634772,
+      "eval_overall_accuracy": 0.9738283299688083,
+      "eval_runtime": 9.4474,
+      "eval_samples_per_second": 7.092,
+      "eval_steps_per_second": 0.529,
+      "step": 700
+    },
+    {
+      "epoch": 63.73,
+      "learning_rate": 2.8317373461012313e-07,
+      "loss": 0.3492,
+      "step": 701
+    },
+    {
+      "epoch": 63.82,
+      "learning_rate": 2.790697674418605e-07,
+      "loss": 0.3385,
+      "step": 702
+    },
+    {
+      "epoch": 63.91,
+      "learning_rate": 2.7496580027359783e-07,
+      "loss": 0.3446,
+      "step": 703
+    },
+    {
+      "epoch": 64.0,
+      "learning_rate": 2.708618331053352e-07,
+      "loss": 0.3284,
+      "step": 704
+    },
+    {
+      "epoch": 64.09,
+      "learning_rate": 2.667578659370725e-07,
+      "loss": 0.3588,
+      "step": 705
+    },
+    {
+      "epoch": 64.18,
+      "learning_rate": 2.6265389876880987e-07,
+      "loss": 0.3259,
+      "step": 706
+    },
+    {
+      "epoch": 64.27,
+      "learning_rate": 2.585499316005472e-07,
+      "loss": 0.3423,
+      "step": 707
+    },
+    {
+      "epoch": 64.36,
+      "learning_rate": 2.5444596443228456e-07,
+      "loss": 0.3579,
+      "step": 708
+    },
+    {
+      "epoch": 64.45,
+      "learning_rate": 2.503419972640219e-07,
+      "loss": 0.3768,
+      "step": 709
+    },
+    {
+      "epoch": 64.55,
+      "learning_rate": 2.4623803009575926e-07,
+      "loss": 0.3183,
+      "step": 710
+    },
+    {
+      "epoch": 64.55,
+      "eval_accuracy_safe": 0.6455776963080327,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9848008600371044,
+      "eval_iou_safe": 0.4367295259604276,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9742930333126004,
+      "eval_loss": 0.315491646528244,
+      "eval_mean_accuracy": 0.8151892781725686,
+      "eval_mean_iou": 0.4703408530910093,
+      "eval_overall_accuracy": 0.9747807516980527,
+      "eval_runtime": 9.6459,
+      "eval_samples_per_second": 6.946,
+      "eval_steps_per_second": 0.518,
+      "step": 710
+    },
+    {
+      "epoch": 64.64,
+      "learning_rate": 2.421340629274966e-07,
+      "loss": 0.3502,
+      "step": 711
+    },
+    {
+      "epoch": 64.73,
+      "learning_rate": 2.3803009575923393e-07,
+      "loss": 0.3373,
+      "step": 712
+    },
+    {
+      "epoch": 64.82,
+      "learning_rate": 2.3392612859097128e-07,
+      "loss": 0.3172,
+      "step": 713
+    },
+    {
+      "epoch": 64.91,
+      "learning_rate": 2.2982216142270862e-07,
+      "loss": 0.3584,
+      "step": 714
+    },
+    {
+      "epoch": 65.0,
+      "learning_rate": 2.2571819425444597e-07,
+      "loss": 0.3384,
+      "step": 715
+    },
+    {
+      "epoch": 65.09,
+      "learning_rate": 2.2161422708618332e-07,
+      "loss": 0.3099,
+      "step": 716
+    },
+    {
+      "epoch": 65.18,
+      "learning_rate": 2.1751025991792066e-07,
+      "loss": 0.356,
+      "step": 717
+    },
+    {
+      "epoch": 65.27,
+      "learning_rate": 2.13406292749658e-07,
+      "loss": 0.3414,
+      "step": 718
+    },
+    {
+      "epoch": 65.36,
+      "learning_rate": 2.0930232558139536e-07,
+      "loss": 0.3399,
+      "step": 719
+    },
+    {
+      "epoch": 65.45,
+      "learning_rate": 2.051983584131327e-07,
+      "loss": 0.3346,
+      "step": 720
+    },
+    {
+      "epoch": 65.45,
+      "eval_accuracy_safe": 0.6517303325738913,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9841023497660231,
+      "eval_iou_safe": 0.4356108967463073,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9737832392255908,
+      "eval_loss": 0.3212249279022217,
+      "eval_mean_accuracy": 0.8179163411699573,
+      "eval_mean_iou": 0.4697980453239661,
+      "eval_overall_accuracy": 0.9742846133104012,
+      "eval_runtime": 9.9806,
+      "eval_samples_per_second": 6.713,
+      "eval_steps_per_second": 0.501,
+      "step": 720
+    },
+    {
+      "epoch": 65.55,
+      "learning_rate": 2.0109439124487005e-07,
+      "loss": 0.337,
+      "step": 721
+    },
+    {
+      "epoch": 65.64,
+      "learning_rate": 1.969904240766074e-07,
+      "loss": 0.348,
+      "step": 722
+    },
+    {
+      "epoch": 65.73,
+      "learning_rate": 1.9288645690834472e-07,
+      "loss": 0.3375,
+      "step": 723
+    },
+    {
+      "epoch": 65.82,
+      "learning_rate": 1.8878248974008207e-07,
+      "loss": 0.3443,
+      "step": 724
+    },
+    {
+      "epoch": 65.91,
+      "learning_rate": 1.8467852257181942e-07,
+      "loss": 0.3469,
+      "step": 725
+    },
+    {
+      "epoch": 66.0,
+      "learning_rate": 1.8057455540355677e-07,
+      "loss": 0.3453,
+      "step": 726
+    },
+    {
+      "epoch": 66.09,
+      "learning_rate": 1.764705882352941e-07,
+      "loss": 0.3513,
+      "step": 727
+    },
+    {
+      "epoch": 66.18,
+      "learning_rate": 1.7236662106703146e-07,
+      "loss": 0.3488,
+      "step": 728
+    },
+    {
+      "epoch": 66.27,
+      "learning_rate": 1.682626538987688e-07,
+      "loss": 0.3507,
+      "step": 729
+    },
+    {
+      "epoch": 66.36,
+      "learning_rate": 1.6415868673050616e-07,
+      "loss": 0.3225,
+      "step": 730
+    },
+    {
+      "epoch": 66.36,
+      "eval_accuracy_safe": 0.6367342454346745,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9862625922228925,
+      "eval_iou_safe": 0.4432164889806017,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9754777569208625,
+      "eval_loss": 0.30519598722457886,
+      "eval_mean_accuracy": 0.8114984188287835,
+      "eval_mean_iou": 0.4728980819671547,
+      "eval_overall_accuracy": 0.9759380852998193,
+      "eval_runtime": 10.6181,
+      "eval_samples_per_second": 6.31,
+      "eval_steps_per_second": 0.471,
+      "step": 730
+    },
+    {
+      "epoch": 66.45,
+      "learning_rate": 1.600547195622435e-07,
+      "loss": 0.3436,
+      "step": 731
+    },
+    {
+      "epoch": 66.55,
+      "learning_rate": 1.5595075239398085e-07,
+      "loss": 0.3231,
+      "step": 732
+    },
+    {
+      "epoch": 66.64,
+      "learning_rate": 1.518467852257182e-07,
+      "loss": 0.3333,
+      "step": 733
+    },
+    {
+      "epoch": 66.73,
+      "learning_rate": 1.4774281805745554e-07,
+      "loss": 0.3432,
+      "step": 734
+    },
+    {
+      "epoch": 66.82,
+      "learning_rate": 1.436388508891929e-07,
+      "loss": 0.3159,
+      "step": 735
+    },
+    {
+      "epoch": 66.91,
+      "learning_rate": 1.3953488372093024e-07,
+      "loss": 0.3653,
+      "step": 736
+    },
+    {
+      "epoch": 67.0,
+      "learning_rate": 1.354309165526676e-07,
+      "loss": 0.3593,
+      "step": 737
+    },
+    {
+      "epoch": 67.09,
+      "learning_rate": 1.3132694938440493e-07,
+      "loss": 0.3153,
+      "step": 738
+    },
+    {
+      "epoch": 67.18,
+      "learning_rate": 1.2722298221614228e-07,
+      "loss": 0.3259,
+      "step": 739
+    },
+    {
+      "epoch": 67.27,
+      "learning_rate": 1.2311901504787963e-07,
+      "loss": 0.3792,
+      "step": 740
+    },
+    {
+      "epoch": 67.27,
+      "eval_accuracy_safe": 0.6381124205380858,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.986087524639413,
+      "eval_iou_safe": 0.44292960173610785,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9753461456304894,
+      "eval_loss": 0.3036666810512543,
+      "eval_mean_accuracy": 0.8120999725887494,
+      "eval_mean_iou": 0.47275858245553243,
+      "eval_overall_accuracy": 0.9758088980148087,
+      "eval_runtime": 9.8604,
+      "eval_samples_per_second": 6.795,
+      "eval_steps_per_second": 0.507,
+      "step": 740
+    },
+    {
+      "epoch": 67.36,
+      "learning_rate": 1.1901504787961696e-07,
+      "loss": 0.3021,
+      "step": 741
+    },
+    {
+      "epoch": 67.45,
+      "learning_rate": 1.1491108071135431e-07,
+      "loss": 0.353,
+      "step": 742
+    },
+    {
+      "epoch": 67.55,
+      "learning_rate": 1.1080711354309166e-07,
+      "loss": 0.3839,
+      "step": 743
+    },
+    {
+      "epoch": 67.64,
+      "learning_rate": 1.06703146374829e-07,
+      "loss": 0.3346,
+      "step": 744
+    },
+    {
+      "epoch": 67.73,
+      "learning_rate": 1.0259917920656635e-07,
+      "loss": 0.3334,
+      "step": 745
+    },
+    {
+      "epoch": 67.82,
+      "learning_rate": 9.84952120383037e-08,
+      "loss": 0.3281,
+      "step": 746
+    },
+    {
+      "epoch": 67.91,
+      "learning_rate": 9.439124487004104e-08,
+      "loss": 0.327,
+      "step": 747
+    },
+    {
+      "epoch": 68.0,
+      "learning_rate": 9.028727770177838e-08,
+      "loss": 0.3521,
+      "step": 748
+    },
+    {
+      "epoch": 68.09,
+      "learning_rate": 8.618331053351573e-08,
+      "loss": 0.3113,
+      "step": 749
+    },
+    {
+      "epoch": 68.18,
+      "learning_rate": 8.207934336525308e-08,
+      "loss": 0.3177,
+      "step": 750
+    },
+    {
+      "epoch": 68.18,
+      "eval_accuracy_safe": 0.6345195276810807,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9864928084419184,
+      "eval_iou_safe": 0.444597209973218,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9756419916569254,
+      "eval_loss": 0.2988866865634918,
+      "eval_mean_accuracy": 0.8105061680614996,
+      "eval_mean_iou": 0.47341306721004783,
+      "eval_overall_accuracy": 0.9760960820895522,
+      "eval_runtime": 10.085,
+      "eval_samples_per_second": 6.644,
+      "eval_steps_per_second": 0.496,
+      "step": 750
+    },
+    {
+      "epoch": 68.27,
+      "learning_rate": 7.797537619699043e-08,
+      "loss": 0.3472,
+      "step": 751
+    },
+    {
+      "epoch": 68.36,
+      "learning_rate": 7.387140902872777e-08,
+      "loss": 0.3095,
+      "step": 752
+    },
+    {
+      "epoch": 68.45,
+      "learning_rate": 6.976744186046512e-08,
+      "loss": 0.3655,
+      "step": 753
+    },
+    {
+      "epoch": 68.55,
+      "learning_rate": 6.566347469220247e-08,
+      "loss": 0.3232,
+      "step": 754
+    },
+    {
+      "epoch": 68.64,
+      "learning_rate": 6.155950752393981e-08,
+      "loss": 0.3631,
+      "step": 755
+    },
+    {
+      "epoch": 68.73,
+      "learning_rate": 5.7455540355677156e-08,
+      "loss": 0.3515,
+      "step": 756
+    },
+    {
+      "epoch": 68.82,
+      "learning_rate": 5.33515731874145e-08,
+      "loss": 0.3406,
+      "step": 757
+    },
+    {
+      "epoch": 68.91,
+      "learning_rate": 4.924760601915185e-08,
+      "loss": 0.3381,
+      "step": 758
+    },
+    {
+      "epoch": 69.0,
+      "learning_rate": 4.514363885088919e-08,
+      "loss": 0.3706,
+      "step": 759
+    },
+    {
+      "epoch": 69.09,
+      "learning_rate": 4.103967168262654e-08,
+      "loss": 0.3295,
+      "step": 760
+    },
+    {
+      "epoch": 69.09,
+      "eval_accuracy_safe": 0.6403599060913412,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9858564283889687,
+      "eval_iou_safe": 0.44260650493805614,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.975184116097936,
+      "eval_loss": 0.3063713610172272,
+      "eval_mean_accuracy": 0.813108167240155,
+      "eval_mean_iou": 0.47259687367866404,
+      "eval_overall_accuracy": 0.975651015096636,
+      "eval_runtime": 9.6213,
+      "eval_samples_per_second": 6.964,
+      "eval_steps_per_second": 0.52,
+      "step": 760
+    },
+    {
+      "epoch": 69.18,
+      "learning_rate": 5.878378378378378e-07,
+      "loss": 0.3508,
+      "step": 761
+    },
+    {
+      "epoch": 69.27,
+      "learning_rate": 5.844594594594596e-07,
+      "loss": 0.3553,
+      "step": 762
+    },
+    {
+      "epoch": 69.36,
+      "learning_rate": 5.810810810810811e-07,
+      "loss": 0.3406,
+      "step": 763
+    },
+    {
+      "epoch": 69.45,
+      "learning_rate": 5.777027027027027e-07,
+      "loss": 0.3433,
+      "step": 764
+    },
+    {
+      "epoch": 69.55,
+      "learning_rate": 5.743243243243243e-07,
+      "loss": 0.3606,
+      "step": 765
+    },
+    {
+      "epoch": 69.64,
+      "learning_rate": 5.70945945945946e-07,
+      "loss": 0.323,
+      "step": 766
+    },
+    {
+      "epoch": 69.73,
+      "learning_rate": 5.675675675675676e-07,
+      "loss": 0.3178,
+      "step": 767
+    },
+    {
+      "epoch": 69.82,
+      "learning_rate": 5.641891891891891e-07,
+      "loss": 0.3124,
+      "step": 768
+    },
+    {
+      "epoch": 69.91,
+      "learning_rate": 5.608108108108109e-07,
+      "loss": 0.3003,
+      "step": 769
+    },
+    {
+      "epoch": 70.0,
+      "learning_rate": 5.574324324324324e-07,
+      "loss": 0.3847,
+      "step": 770
+    },
+    {
+      "epoch": 70.0,
+      "eval_accuracy_safe": 0.6429312146059576,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9856813608054892,
+      "eval_iou_safe": 0.44386336810476407,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9750875071024252,
+      "eval_loss": 0.30536746978759766,
+      "eval_mean_accuracy": 0.8143062877057234,
+      "eval_mean_iou": 0.4729836250690631,
+      "eval_overall_accuracy": 0.9755570710594974,
+      "eval_runtime": 8.333,
+      "eval_samples_per_second": 8.04,
+      "eval_steps_per_second": 0.6,
+      "step": 770
+    },
+    {
+      "epoch": 70.09,
+      "learning_rate": 5.54054054054054e-07,
+      "loss": 0.3266,
+      "step": 771
+    },
+    {
+      "epoch": 70.18,
+      "learning_rate": 5.506756756756757e-07,
+      "loss": 0.328,
+      "step": 772
+    },
+    {
+      "epoch": 70.27,
+      "learning_rate": 5.472972972972973e-07,
+      "loss": 0.3455,
+      "step": 773
+    },
+    {
+      "epoch": 70.36,
+      "learning_rate": 5.43918918918919e-07,
+      "loss": 0.3372,
+      "step": 774
+    },
+    {
+      "epoch": 70.45,
+      "learning_rate": 5.405405405405405e-07,
+      "loss": 0.3266,
+      "step": 775
+    },
+    {
+      "epoch": 70.55,
+      "learning_rate": 5.371621621621622e-07,
+      "loss": 0.362,
+      "step": 776
+    },
+    {
+      "epoch": 70.64,
+      "learning_rate": 5.337837837837837e-07,
+      "loss": 0.3509,
+      "step": 777
+    },
+    {
+      "epoch": 70.73,
+      "learning_rate": 5.304054054054054e-07,
+      "loss": 0.2946,
+      "step": 778
+    },
+    {
+      "epoch": 70.82,
+      "learning_rate": 5.270270270270271e-07,
+      "loss": 0.3261,
+      "step": 779
+    },
+    {
+      "epoch": 70.91,
+      "learning_rate": 5.236486486486486e-07,
+      "loss": 0.3406,
+      "step": 780
+    },
+    {
+      "epoch": 70.91,
+      "eval_accuracy_safe": 0.6442862594978431,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9861615646160722,
+      "eval_iou_safe": 0.4475671202721901,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9756000336170195,
+      "eval_loss": 0.30749550461769104,
+      "eval_mean_accuracy": 0.8152239120569577,
+      "eval_mean_iou": 0.47438905129640324,
+      "eval_overall_accuracy": 0.9760631162728837,
+      "eval_runtime": 9.0467,
+      "eval_samples_per_second": 7.406,
+      "eval_steps_per_second": 0.553,
+      "step": 780
+    },
+    {
+      "epoch": 71.0,
+      "learning_rate": 5.202702702702703e-07,
+      "loss": 0.2752,
+      "step": 781
+    },
+    {
+      "epoch": 71.09,
+      "learning_rate": 5.168918918918919e-07,
+      "loss": 0.3096,
+      "step": 782
+    },
+    {
+      "epoch": 71.18,
+      "learning_rate": 5.135135135135135e-07,
+      "loss": 0.3476,
+      "step": 783
+    },
+    {
+      "epoch": 71.27,
+      "learning_rate": 5.101351351351352e-07,
+      "loss": 0.3235,
+      "step": 784
+    },
+    {
+      "epoch": 71.36,
+      "learning_rate": 5.067567567567567e-07,
+      "loss": 0.3134,
+      "step": 785
+    },
+    {
+      "epoch": 71.45,
+      "learning_rate": 5.033783783783784e-07,
+      "loss": 0.3702,
+      "step": 786
+    },
+    {
+      "epoch": 71.55,
+      "learning_rate": 5e-07,
+      "loss": 0.3547,
+      "step": 787
+    },
+    {
+      "epoch": 71.64,
+      "learning_rate": 4.966216216216216e-07,
+      "loss": 0.2856,
+      "step": 788
+    },
+    {
+      "epoch": 71.73,
+      "learning_rate": 4.932432432432432e-07,
+      "loss": 0.3265,
+      "step": 789
+    },
+    {
+      "epoch": 71.82,
+      "learning_rate": 4.898648648648649e-07,
+      "loss": 0.3847,
+      "step": 790
+    },
+    {
+      "epoch": 71.82,
+      "eval_accuracy_safe": 0.6342573852837884,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9877470878880337,
+      "eval_iou_safe": 0.45462382943213125,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9768733490672094,
+      "eval_loss": 0.29106298089027405,
+      "eval_mean_accuracy": 0.8110022365859111,
+      "eval_mean_iou": 0.47716572616644687,
+      "eval_overall_accuracy": 0.9773055688658757,
+      "eval_runtime": 9.0041,
+      "eval_samples_per_second": 7.441,
+      "eval_steps_per_second": 0.555,
+      "step": 790
+    },
+    {
+      "epoch": 71.91,
+      "learning_rate": 4.864864864864865e-07,
+      "loss": 0.3186,
+      "step": 791
+    },
+    {
+      "epoch": 72.0,
+      "learning_rate": 4.831081081081081e-07,
+      "loss": 0.369,
+      "step": 792
+    },
+    {
+      "epoch": 72.09,
+      "learning_rate": 4.797297297297298e-07,
+      "loss": 0.3192,
+      "step": 793
+    },
+    {
+      "epoch": 72.18,
+      "learning_rate": 4.763513513513513e-07,
+      "loss": 0.315,
+      "step": 794
+    },
+    {
+      "epoch": 72.27,
+      "learning_rate": 4.72972972972973e-07,
+      "loss": 0.3422,
+      "step": 795
+    },
+    {
+      "epoch": 72.36,
+      "learning_rate": 4.6959459459459463e-07,
+      "loss": 0.3231,
+      "step": 796
+    },
+    {
+      "epoch": 72.45,
+      "learning_rate": 4.662162162162162e-07,
+      "loss": 0.3281,
+      "step": 797
+    },
+    {
+      "epoch": 72.55,
+      "learning_rate": 4.6283783783783784e-07,
+      "loss": 0.3152,
+      "step": 798
+    },
+    {
+      "epoch": 72.64,
+      "learning_rate": 4.594594594594594e-07,
+      "loss": 0.3132,
+      "step": 799
+    },
+    {
+      "epoch": 72.73,
+      "learning_rate": 4.560810810810811e-07,
+      "loss": 0.3292,
+      "step": 800
+    },
+    {
+      "epoch": 72.73,
+      "eval_accuracy_safe": 0.6328483698983427,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9880534561591228,
+      "eval_iou_safe": 0.45670444681845496,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9771349564164951,
+      "eval_loss": 0.29046520590782166,
+      "eval_mean_accuracy": 0.8104509130287327,
+      "eval_mean_iou": 0.47794646774498334,
+      "eval_overall_accuracy": 0.9775612674542328,
+      "eval_runtime": 9.9385,
+      "eval_samples_per_second": 6.741,
+      "eval_steps_per_second": 0.503,
+      "step": 800
+    },
+    {
+      "epoch": 72.82,
+      "learning_rate": 4.5270270270270275e-07,
+      "loss": 0.3247,
+      "step": 801
+    },
+    {
+      "epoch": 72.91,
+      "learning_rate": 4.4932432432432433e-07,
+      "loss": 0.3317,
+      "step": 802
+    },
+    {
+      "epoch": 73.0,
+      "learning_rate": 4.4594594594594596e-07,
+      "loss": 0.3452,
+      "step": 803
+    },
+    {
+      "epoch": 73.09,
+      "learning_rate": 4.4256756756756754e-07,
+      "loss": 0.34,
+      "step": 804
+    },
+    {
+      "epoch": 73.18,
+      "learning_rate": 4.3918918918918923e-07,
+      "loss": 0.3101,
+      "step": 805
+    },
+    {
+      "epoch": 73.27,
+      "learning_rate": 4.3581081081081076e-07,
+      "loss": 0.3173,
+      "step": 806
+    },
+    {
+      "epoch": 73.36,
+      "learning_rate": 4.3243243243243244e-07,
+      "loss": 0.3542,
+      "step": 807
+    },
+    {
+      "epoch": 73.45,
+      "learning_rate": 4.290540540540541e-07,
+      "loss": 0.3141,
+      "step": 808
+    },
+    {
+      "epoch": 73.55,
+      "learning_rate": 4.2567567567567566e-07,
+      "loss": 0.3388,
+      "step": 809
+    },
+    {
+      "epoch": 73.64,
+      "learning_rate": 4.2229729729729735e-07,
+      "loss": 0.3156,
+      "step": 810
+    },
+    {
+      "epoch": 73.64,
+      "eval_accuracy_safe": 0.6318094378973096,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9882225395289579,
+      "eval_iou_safe": 0.4578583531101107,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.977271212196021,
+      "eval_loss": 0.28654035925865173,
+      "eval_mean_accuracy": 0.8100159887131337,
+      "eval_mean_iou": 0.47837652176871054,
+      "eval_overall_accuracy": 0.9776946679869695,
+      "eval_runtime": 9.2746,
+      "eval_samples_per_second": 7.224,
+      "eval_steps_per_second": 0.539,
+      "step": 810
+    },
+    {
+      "epoch": 73.73,
+      "learning_rate": 4.189189189189189e-07,
+      "loss": 0.3512,
+      "step": 811
+    },
+    {
+      "epoch": 73.82,
+      "learning_rate": 4.1554054054054056e-07,
+      "loss": 0.2992,
+      "step": 812
+    },
+    {
+      "epoch": 73.91,
+      "learning_rate": 4.121621621621622e-07,
+      "loss": 0.34,
+      "step": 813
+    },
+    {
+      "epoch": 74.0,
+      "learning_rate": 4.087837837837838e-07,
+      "loss": 0.3609,
+      "step": 814
+    },
+    {
+      "epoch": 74.09,
+      "learning_rate": 4.0540540540540546e-07,
+      "loss": 0.2993,
+      "step": 815
+    },
+    {
+      "epoch": 74.18,
+      "learning_rate": 4.02027027027027e-07,
+      "loss": 0.3078,
+      "step": 816
+    },
+    {
+      "epoch": 74.27,
+      "learning_rate": 3.986486486486487e-07,
+      "loss": 0.3215,
+      "step": 817
+    },
+    {
+      "epoch": 74.36,
+      "learning_rate": 3.9527027027027026e-07,
+      "loss": 0.2968,
+      "step": 818
+    },
+    {
+      "epoch": 74.45,
+      "learning_rate": 3.918918918918919e-07,
+      "loss": 0.3119,
+      "step": 819
+    },
+    {
+      "epoch": 74.55,
+      "learning_rate": 3.8851351351351353e-07,
+      "loss": 0.3106,
+      "step": 820
+    },
+    {
+      "epoch": 74.55,
+      "eval_accuracy_safe": 0.6333418144108928,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9883734942515761,
+      "eval_iou_safe": 0.460039398466344,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9774654093148158,
+      "eval_loss": 0.2812236547470093,
+      "eval_mean_accuracy": 0.8108576543312345,
+      "eval_mean_iou": 0.47916826926038664,
+      "eval_overall_accuracy": 0.9778864276942922,
+      "eval_runtime": 9.2069,
+      "eval_samples_per_second": 7.277,
+      "eval_steps_per_second": 0.543,
+      "step": 820
+    },
+    {
+      "epoch": 74.64,
+      "learning_rate": 3.851351351351351e-07,
+      "loss": 0.327,
+      "step": 821
+    },
+    {
+      "epoch": 74.73,
+      "learning_rate": 3.817567567567568e-07,
+      "loss": 0.3093,
+      "step": 822
+    },
+    {
+      "epoch": 74.82,
+      "learning_rate": 3.783783783783784e-07,
+      "loss": 0.326,
+      "step": 823
+    },
+    {
+      "epoch": 74.91,
+      "learning_rate": 3.75e-07,
+      "loss": 0.3625,
+      "step": 824
+    },
+    {
+      "epoch": 75.0,
+      "learning_rate": 3.7162162162162164e-07,
+      "loss": 0.3852,
+      "step": 825
+    },
+    {
+      "epoch": 75.09,
+      "learning_rate": 3.682432432432432e-07,
+      "loss": 0.3337,
+      "step": 826
+    },
+    {
+      "epoch": 75.18,
+      "learning_rate": 3.648648648648649e-07,
+      "loss": 0.3197,
+      "step": 827
+    },
+    {
+      "epoch": 75.27,
+      "learning_rate": 3.614864864864865e-07,
+      "loss": 0.3149,
+      "step": 828
+    },
+    {
+      "epoch": 75.36,
+      "learning_rate": 3.581081081081081e-07,
+      "loss": 0.3312,
+      "step": 829
+    },
+    {
+      "epoch": 75.45,
+      "learning_rate": 3.5472972972972976e-07,
+      "loss": 0.3004,
+      "step": 830
+    },
+    {
+      "epoch": 75.45,
+      "eval_accuracy_safe": 0.6232358394917521,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.989330147072024,
+      "eval_iou_safe": 0.4632075377093191,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9781148642197699,
+      "eval_loss": 0.27982932329177856,
+      "eval_mean_accuracy": 0.806282993281888,
+      "eval_mean_iou": 0.4804408006430297,
+      "eval_overall_accuracy": 0.978516308229361,
+      "eval_runtime": 9.385,
+      "eval_samples_per_second": 7.139,
+      "eval_steps_per_second": 0.533,
+      "step": 830
+    },
+    {
+      "epoch": 75.55,
+      "learning_rate": 3.5135135135135134e-07,
+      "loss": 0.2947,
+      "step": 831
+    },
+    {
+      "epoch": 75.64,
+      "learning_rate": 3.47972972972973e-07,
+      "loss": 0.3226,
+      "step": 832
+    },
+    {
+      "epoch": 75.73,
+      "learning_rate": 3.445945945945946e-07,
+      "loss": 0.3221,
+      "step": 833
+    },
+    {
+      "epoch": 75.82,
+      "learning_rate": 3.4121621621621624e-07,
+      "loss": 0.3153,
+      "step": 834
+    },
+    {
+      "epoch": 75.91,
+      "learning_rate": 3.378378378378379e-07,
+      "loss": 0.3358,
+      "step": 835
+    },
+    {
+      "epoch": 76.0,
+      "learning_rate": 3.3445945945945946e-07,
+      "loss": 0.3165,
+      "step": 836
+    },
+    {
+      "epoch": 76.09,
+      "learning_rate": 3.310810810810811e-07,
+      "loss": 0.3004,
+      "step": 837
+    },
+    {
+      "epoch": 76.18,
+      "learning_rate": 3.2770270270270267e-07,
+      "loss": 0.2937,
+      "step": 838
+    },
+    {
+      "epoch": 76.27,
+      "learning_rate": 3.2432432432432436e-07,
+      "loss": 0.3298,
+      "step": 839
+    },
+    {
+      "epoch": 76.36,
+      "learning_rate": 3.2094594594594594e-07,
+      "loss": 0.3336,
+      "step": 840
+    },
+    {
+      "epoch": 76.36,
+      "eval_accuracy_safe": 0.6484612626782472,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.987219421049624,
+      "eval_iou_safe": 0.4592642445455861,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9767689022192217,
+      "eval_loss": 0.29541948437690735,
+      "eval_mean_accuracy": 0.8178403418639356,
+      "eval_mean_iou": 0.4786777155882693,
+      "eval_overall_accuracy": 0.9772130482232393,
+      "eval_runtime": 10.9829,
+      "eval_samples_per_second": 6.1,
+      "eval_steps_per_second": 0.455,
+      "step": 840
+    },
+    {
+      "epoch": 76.45,
+      "learning_rate": 3.175675675675676e-07,
+      "loss": 0.329,
+      "step": 841
+    },
+    {
+      "epoch": 76.55,
+      "learning_rate": 3.141891891891892e-07,
+      "loss": 0.3153,
+      "step": 842
+    },
+    {
+      "epoch": 76.64,
+      "learning_rate": 3.108108108108108e-07,
+      "loss": 0.2911,
+      "step": 843
+    },
+    {
+      "epoch": 76.73,
+      "learning_rate": 3.074324324324324e-07,
+      "loss": 0.3222,
+      "step": 844
+    },
+    {
+      "epoch": 76.82,
+      "learning_rate": 3.0405405405405406e-07,
+      "loss": 0.2909,
+      "step": 845
+    },
+    {
+      "epoch": 76.91,
+      "learning_rate": 3.006756756756757e-07,
+      "loss": 0.3212,
+      "step": 846
+    },
+    {
+      "epoch": 77.0,
+      "learning_rate": 2.972972972972973e-07,
+      "loss": 0.3321,
+      "step": 847
+    },
+    {
+      "epoch": 77.09,
+      "learning_rate": 2.939189189189189e-07,
+      "loss": 0.3263,
+      "step": 848
+    },
+    {
+      "epoch": 77.18,
+      "learning_rate": 2.9054054054054054e-07,
+      "loss": 0.3369,
+      "step": 849
+    },
+    {
+      "epoch": 77.27,
+      "learning_rate": 2.871621621621622e-07,
+      "loss": 0.299,
+      "step": 850
+    },
+    {
+      "epoch": 77.27,
+      "eval_accuracy_safe": 0.6490125327196117,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9873693784033015,
+      "eval_iou_safe": 0.4612546833154106,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9769332073058408,
+      "eval_loss": 0.2908638119697571,
+      "eval_mean_accuracy": 0.8181909555614566,
+      "eval_mean_iou": 0.4793959635404171,
+      "eval_overall_accuracy": 0.9773748597102379,
+      "eval_runtime": 10.9162,
+      "eval_samples_per_second": 6.138,
+      "eval_steps_per_second": 0.458,
+      "step": 850
+    },
+    {
+      "epoch": 77.36,
+      "learning_rate": 2.837837837837838e-07,
+      "loss": 0.2851,
+      "step": 851
+    },
+    {
+      "epoch": 77.45,
+      "learning_rate": 2.8040540540540544e-07,
+      "loss": 0.2774,
+      "step": 852
+    },
+    {
+      "epoch": 77.55,
+      "learning_rate": 2.77027027027027e-07,
+      "loss": 0.3091,
+      "step": 853
+    },
+    {
+      "epoch": 77.64,
+      "learning_rate": 2.7364864864864866e-07,
+      "loss": 0.3395,
+      "step": 854
+    },
+    {
+      "epoch": 77.73,
+      "learning_rate": 2.7027027027027024e-07,
+      "loss": 0.3364,
+      "step": 855
+    },
+    {
+      "epoch": 77.82,
+      "learning_rate": 2.6689189189189187e-07,
+      "loss": 0.3395,
+      "step": 856
+    },
+    {
+      "epoch": 77.91,
+      "learning_rate": 2.6351351351351356e-07,
+      "loss": 0.2895,
+      "step": 857
+    },
+    {
+      "epoch": 78.0,
+      "learning_rate": 2.6013513513513514e-07,
+      "loss": 0.3205,
+      "step": 858
+    },
+    {
+      "epoch": 78.09,
+      "learning_rate": 2.567567567567568e-07,
+      "loss": 0.3041,
+      "step": 859
+    },
+    {
+      "epoch": 78.18,
+      "learning_rate": 2.5337837837837835e-07,
+      "loss": 0.292,
+      "step": 860
+    },
+    {
+      "epoch": 78.18,
+      "eval_accuracy_safe": 0.6497314967945382,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9875053725918087,
+      "eval_iou_safe": 0.462851791638289,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9770889201532796,
+      "eval_loss": 0.2852531969547272,
+      "eval_mean_accuracy": 0.8186184346931735,
+      "eval_mean_iou": 0.4799802372638562,
+      "eval_overall_accuracy": 0.9775280738944438,
+      "eval_runtime": 10.619,
+      "eval_samples_per_second": 6.309,
+      "eval_steps_per_second": 0.471,
+      "step": 860
+    },
+    {
+      "epoch": 78.27,
+      "learning_rate": 2.5e-07,
+      "loss": 0.3158,
+      "step": 861
+    },
+    {
+      "epoch": 78.36,
+      "learning_rate": 2.466216216216216e-07,
+      "loss": 0.3292,
+      "step": 862
+    },
+    {
+      "epoch": 78.45,
+      "learning_rate": 2.4324324324324326e-07,
+      "loss": 0.3412,
+      "step": 863
+    },
+    {
+      "epoch": 78.55,
+      "learning_rate": 2.398648648648649e-07,
+      "loss": 0.2824,
+      "step": 864
+    },
+    {
+      "epoch": 78.64,
+      "learning_rate": 2.364864864864865e-07,
+      "loss": 0.3077,
+      "step": 865
+    },
+    {
+      "epoch": 78.73,
+      "learning_rate": 2.331081081081081e-07,
+      "loss": 0.3257,
+      "step": 866
+    },
+    {
+      "epoch": 78.82,
+      "learning_rate": 2.297297297297297e-07,
+      "loss": 0.3032,
+      "step": 867
+    },
+    {
+      "epoch": 78.91,
+      "learning_rate": 2.2635135135135137e-07,
+      "loss": 0.3132,
+      "step": 868
+    },
+    {
+      "epoch": 79.0,
+      "learning_rate": 2.2297297297297298e-07,
+      "loss": 0.3469,
+      "step": 869
+    },
+    {
+      "epoch": 79.09,
+      "learning_rate": 2.1959459459459461e-07,
+      "loss": 0.2922,
+      "step": 870
+    },
+    {
+      "epoch": 79.09,
+      "eval_accuracy_safe": 0.6586057879499309,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.986641475082849,
+      "eval_iou_safe": 0.4601317293824394,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.976495115071741,
+      "eval_loss": 0.2916642725467682,
+      "eval_mean_accuracy": 0.82262363151639,
+      "eval_mean_iou": 0.4788756148180601,
+      "eval_overall_accuracy": 0.9769518268642141,
+      "eval_runtime": 9.4334,
+      "eval_samples_per_second": 7.102,
+      "eval_steps_per_second": 0.53,
+      "step": 870
+    },
+    {
+      "epoch": 79.18,
+      "learning_rate": 2.1621621621621622e-07,
+      "loss": 0.2991,
+      "step": 871
+    },
+    {
+      "epoch": 79.27,
+      "learning_rate": 2.1283783783783783e-07,
+      "loss": 0.3065,
+      "step": 872
+    },
+    {
+      "epoch": 79.36,
+      "learning_rate": 2.0945945945945944e-07,
+      "loss": 0.3218,
+      "step": 873
+    },
+    {
+      "epoch": 79.45,
+      "learning_rate": 2.060810810810811e-07,
+      "loss": 0.2962,
+      "step": 874
+    },
+    {
+      "epoch": 79.55,
+      "learning_rate": 2.0270270270270273e-07,
+      "loss": 0.3047,
+      "step": 875
+    },
+    {
+      "epoch": 79.64,
+      "learning_rate": 1.9932432432432434e-07,
+      "loss": 0.3408,
+      "step": 876
+    },
+    {
+      "epoch": 79.73,
+      "learning_rate": 1.9594594594594595e-07,
+      "loss": 0.2935,
+      "step": 877
+    },
+    {
+      "epoch": 79.82,
+      "learning_rate": 1.9256756756756755e-07,
+      "loss": 0.3327,
+      "step": 878
+    },
+    {
+      "epoch": 79.91,
+      "learning_rate": 1.891891891891892e-07,
+      "loss": 0.3071,
+      "step": 879
+    },
+    {
+      "epoch": 80.0,
+      "learning_rate": 1.8581081081081082e-07,
+      "loss": 0.3583,
+      "step": 880
+    },
+    {
+      "epoch": 80.0,
+      "eval_accuracy_safe": 0.6515260157054136,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9875674441411791,
+      "eval_iou_safe": 0.4643852215774202,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9772031493777439,
+      "eval_loss": 0.2875562906265259,
+      "eval_mean_accuracy": 0.8195467299232964,
+      "eval_mean_iou": 0.48052945698505467,
+      "eval_overall_accuracy": 0.9776413191610308,
+      "eval_runtime": 10.47,
+      "eval_samples_per_second": 6.399,
+      "eval_steps_per_second": 0.478,
+      "step": 880
+    },
+    {
+      "epoch": 80.09,
+      "learning_rate": 1.8243243243243246e-07,
+      "loss": 0.3193,
+      "step": 881
+    },
+    {
+      "epoch": 80.18,
+      "learning_rate": 1.7905405405405406e-07,
+      "loss": 0.3032,
+      "step": 882
+    },
+    {
+      "epoch": 80.27,
+      "learning_rate": 1.7567567567567567e-07,
+      "loss": 0.2931,
+      "step": 883
+    },
+    {
+      "epoch": 80.36,
+      "learning_rate": 1.722972972972973e-07,
+      "loss": 0.3185,
+      "step": 884
+    },
+    {
+      "epoch": 80.45,
+      "learning_rate": 1.6891891891891894e-07,
+      "loss": 0.3118,
+      "step": 885
+    },
+    {
+      "epoch": 80.55,
+      "learning_rate": 1.6554054054054055e-07,
+      "loss": 0.2959,
+      "step": 886
+    },
+    {
+      "epoch": 80.64,
+      "learning_rate": 1.6216216216216218e-07,
+      "loss": 0.3005,
+      "step": 887
+    },
+    {
+      "epoch": 80.73,
+      "learning_rate": 1.587837837837838e-07,
+      "loss": 0.3257,
+      "step": 888
+    },
+    {
+      "epoch": 80.82,
+      "learning_rate": 1.554054054054054e-07,
+      "loss": 0.287,
+      "step": 889
+    },
+    {
+      "epoch": 80.91,
+      "learning_rate": 1.5202702702702703e-07,
+      "loss": 0.293,
+      "step": 890
+    },
+    {
+      "epoch": 80.91,
+      "eval_accuracy_safe": 0.6464797745575384,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9881851088592998,
+      "eval_iou_safe": 0.46743123613294535,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9776656866480414,
+      "eval_loss": 0.27672818303108215,
+      "eval_mean_accuracy": 0.8173324417084191,
+      "eval_mean_iou": 0.48169897426032887,
+      "eval_overall_accuracy": 0.9780916811814949,
+      "eval_runtime": 10.4393,
+      "eval_samples_per_second": 6.418,
+      "eval_steps_per_second": 0.479,
+      "step": 890
+    },
+    {
+      "epoch": 81.0,
+      "learning_rate": 1.4864864864864866e-07,
+      "loss": 0.3292,
+      "step": 891
+    },
+    {
+      "epoch": 81.09,
+      "learning_rate": 1.4527027027027027e-07,
+      "loss": 0.3271,
+      "step": 892
+    },
+    {
+      "epoch": 81.18,
+      "learning_rate": 1.418918918918919e-07,
+      "loss": 0.3114,
+      "step": 893
+    },
+    {
+      "epoch": 81.27,
+      "learning_rate": 1.385135135135135e-07,
+      "loss": 0.298,
+      "step": 894
+    },
+    {
+      "epoch": 81.36,
+      "learning_rate": 1.3513513513513512e-07,
+      "loss": 0.292,
+      "step": 895
+    },
+    {
+      "epoch": 81.45,
+      "learning_rate": 1.3175675675675678e-07,
+      "loss": 0.2806,
+      "step": 896
+    },
+    {
+      "epoch": 81.55,
+      "learning_rate": 1.283783783783784e-07,
+      "loss": 0.2994,
+      "step": 897
+    },
+    {
+      "epoch": 81.64,
+      "learning_rate": 1.25e-07,
+      "loss": 0.3274,
+      "step": 898
+    },
+    {
+      "epoch": 81.73,
+      "learning_rate": 1.2162162162162163e-07,
+      "loss": 0.3284,
+      "step": 899
+    },
+    {
+      "epoch": 81.82,
+      "learning_rate": 1.1824324324324325e-07,
+      "loss": 0.3287,
+      "step": 900
+    },
+    {
+      "epoch": 81.82,
+      "eval_accuracy_safe": 0.651817070867113,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9876163152192751,
+      "eval_iou_safe": 0.4652088022143105,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9772601877034197,
+      "eval_loss": 0.28578364849090576,
+      "eval_mean_accuracy": 0.8197166930431941,
+      "eval_mean_iou": 0.48082299663924344,
+      "eval_overall_accuracy": 0.9776973439686334,
+      "eval_runtime": 9.8853,
+      "eval_samples_per_second": 6.778,
+      "eval_steps_per_second": 0.506,
+      "step": 900
+    },
+    {
+      "epoch": 81.91,
+      "learning_rate": 1.1486486486486486e-07,
+      "loss": 0.2998,
+      "step": 901
+    },
+    {
+      "epoch": 82.0,
+      "learning_rate": 1.1148648648648649e-07,
+      "loss": 0.3326,
+      "step": 902
+    },
+    {
+      "epoch": 82.09,
+      "learning_rate": 1.0810810810810811e-07,
+      "loss": 0.287,
+      "step": 903
+    },
+    {
+      "epoch": 82.18,
+      "learning_rate": 1.0472972972972972e-07,
+      "loss": 0.2957,
+      "step": 904
+    },
+    {
+      "epoch": 82.27,
+      "learning_rate": 1.0135135135135137e-07,
+      "loss": 0.3262,
+      "step": 905
+    },
+    {
+      "epoch": 82.36,
+      "learning_rate": 9.797297297297297e-08,
+      "loss": 0.3166,
+      "step": 906
+    },
+    {
+      "epoch": 82.45,
+      "learning_rate": 9.45945945945946e-08,
+      "loss": 0.2848,
+      "step": 907
+    },
+    {
+      "epoch": 82.55,
+      "learning_rate": 9.121621621621623e-08,
+      "loss": 0.2995,
+      "step": 908
+    },
+    {
+      "epoch": 82.64,
+      "learning_rate": 8.783783783783784e-08,
+      "loss": 0.324,
+      "step": 909
+    },
+    {
+      "epoch": 82.73,
+      "learning_rate": 8.445945945945947e-08,
+      "loss": 0.3067,
+      "step": 910
+    },
+    {
+      "epoch": 82.73,
+      "eval_accuracy_safe": 0.6528463652800105,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.987525202633101,
+      "eval_iou_safe": 0.4653979333846323,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.977200381052192,
+      "eval_loss": 0.2860513925552368,
+      "eval_mean_accuracy": 0.8201857839565557,
+      "eval_mean_iou": 0.4808661048122748,
+      "eval_overall_accuracy": 0.9776393264087279,
+      "eval_runtime": 9.3628,
+      "eval_samples_per_second": 7.156,
+      "eval_steps_per_second": 0.534,
+      "step": 910
+    },
+    {
+      "epoch": 82.82,
+      "learning_rate": 8.108108108108109e-08,
+      "loss": 0.2896,
+      "step": 911
+    },
+    {
+      "epoch": 82.91,
+      "learning_rate": 7.77027027027027e-08,
+      "loss": 0.3404,
+      "step": 912
+    },
+    {
+      "epoch": 83.0,
+      "learning_rate": 7.432432432432433e-08,
+      "loss": 0.3558,
+      "step": 913
+    },
+    {
+      "epoch": 83.09,
+      "learning_rate": 7.094594594594595e-08,
+      "loss": 0.301,
+      "step": 914
+    },
+    {
+      "epoch": 83.18,
+      "learning_rate": 6.756756756756756e-08,
+      "loss": 0.3109,
+      "step": 915
+    },
+    {
+      "epoch": 83.27,
+      "learning_rate": 6.41891891891892e-08,
+      "loss": 0.2861,
+      "step": 916
+    },
+    {
+      "epoch": 83.36,
+      "learning_rate": 6.081081081081081e-08,
+      "loss": 0.2973,
+      "step": 917
+    },
+    {
+      "epoch": 83.45,
+      "learning_rate": 5.743243243243243e-08,
+      "loss": 0.3144,
+      "step": 918
+    },
+    {
+      "epoch": 83.55,
+      "learning_rate": 5.4054054054054056e-08,
+      "loss": 0.3096,
+      "step": 919
+    },
+    {
+      "epoch": 83.64,
+      "learning_rate": 5.067567567567568e-08,
+      "loss": 0.3374,
+      "step": 920
+    },
+    {
+      "epoch": 83.64,
+      "eval_accuracy_safe": 0.6577017821827981,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9870246994311359,
+      "eval_iou_safe": 0.46313936379874804,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9768479096925932,
+      "eval_loss": 0.28693443536758423,
+      "eval_mean_accuracy": 0.822363240806967,
+      "eval_mean_iou": 0.47999575783044707,
+      "eval_overall_accuracy": 0.9772970284988631,
+      "eval_runtime": 11.1318,
+      "eval_samples_per_second": 6.019,
+      "eval_steps_per_second": 0.449,
+      "step": 920
+    },
+    {
+      "epoch": 83.73,
+      "learning_rate": 7.545691906005222e-07,
+      "loss": 0.3124,
+      "step": 921
+    },
+    {
+      "epoch": 83.82,
+      "learning_rate": 7.519582245430809e-07,
+      "loss": 0.2825,
+      "step": 922
+    },
+    {
+      "epoch": 83.91,
+      "learning_rate": 7.493472584856398e-07,
+      "loss": 0.2987,
+      "step": 923
+    },
+    {
+      "epoch": 84.0,
+      "learning_rate": 7.467362924281985e-07,
+      "loss": 0.3105,
+      "step": 924
+    },
+    {
+      "epoch": 84.09,
+      "learning_rate": 7.441253263707572e-07,
+      "loss": 0.2889,
+      "step": 925
+    },
+    {
+      "epoch": 84.18,
+      "learning_rate": 7.415143603133159e-07,
+      "loss": 0.3116,
+      "step": 926
+    },
+    {
+      "epoch": 84.27,
+      "learning_rate": 7.389033942558747e-07,
+      "loss": 0.3094,
+      "step": 927
+    },
+    {
+      "epoch": 84.36,
+      "learning_rate": 7.362924281984335e-07,
+      "loss": 0.3213,
+      "step": 928
+    },
+    {
+      "epoch": 84.45,
+      "learning_rate": 7.336814621409922e-07,
+      "loss": 0.3122,
+      "step": 929
+    },
+    {
+      "epoch": 84.55,
+      "learning_rate": 7.310704960835509e-07,
+      "loss": 0.3171,
+      "step": 930
+    },
+    {
+      "epoch": 84.55,
+      "eval_accuracy_safe": 0.6441841010636042,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9887108982973504,
+      "eval_iou_safe": 0.47137576040519213,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9781178068263727,
+      "eval_loss": 0.27060091495513916,
+      "eval_mean_accuracy": 0.8164474996804774,
+      "eval_mean_iou": 0.4831645224105216,
+      "eval_overall_accuracy": 0.9785341291285273,
+      "eval_runtime": 10.3282,
+      "eval_samples_per_second": 6.487,
+      "eval_steps_per_second": 0.484,
+      "step": 930
+    },
+    {
+      "epoch": 84.64,
+      "learning_rate": 7.284595300261096e-07,
+      "loss": 0.2831,
+      "step": 931
+    },
+    {
+      "epoch": 84.73,
+      "learning_rate": 7.258485639686684e-07,
+      "loss": 0.3347,
+      "step": 932
+    },
+    {
+      "epoch": 84.82,
+      "learning_rate": 7.232375979112272e-07,
+      "loss": 0.3065,
+      "step": 933
+    },
+    {
+      "epoch": 84.91,
+      "learning_rate": 7.206266318537859e-07,
+      "loss": 0.3036,
+      "step": 934
+    },
+    {
+      "epoch": 85.0,
+      "learning_rate": 7.180156657963447e-07,
+      "loss": 0.3049,
+      "step": 935
+    },
+    {
+      "epoch": 85.09,
+      "learning_rate": 7.154046997389033e-07,
+      "loss": 0.3003,
+      "step": 936
+    },
+    {
+      "epoch": 85.18,
+      "learning_rate": 7.127937336814622e-07,
+      "loss": 0.3006,
+      "step": 937
+    },
+    {
+      "epoch": 85.27,
+      "learning_rate": 7.101827676240209e-07,
+      "loss": 0.284,
+      "step": 938
+    },
+    {
+      "epoch": 85.36,
+      "learning_rate": 7.075718015665797e-07,
+      "loss": 0.3199,
+      "step": 939
+    },
+    {
+      "epoch": 85.45,
+      "learning_rate": 7.049608355091384e-07,
+      "loss": 0.3156,
+      "step": 940
+    },
+    {
+      "epoch": 85.45,
+      "eval_accuracy_safe": 0.6320773628474833,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9898702516878123,
+      "eval_iou_safe": 0.4757064376163252,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9789079344104102,
+      "eval_loss": 0.2708066403865814,
+      "eval_mean_accuracy": 0.8109738072676478,
+      "eval_mean_iou": 0.4848714573422452,
+      "eval_overall_accuracy": 0.979301623444059,
+      "eval_runtime": 9.3749,
+      "eval_samples_per_second": 7.147,
+      "eval_steps_per_second": 0.533,
+      "step": 940
+    },
+    {
+      "epoch": 85.55,
+      "learning_rate": 7.023498694516971e-07,
+      "loss": 0.2918,
+      "step": 941
+    },
+    {
+      "epoch": 85.64,
+      "learning_rate": 6.997389033942559e-07,
+      "loss": 0.2911,
+      "step": 942
+    },
+    {
+      "epoch": 85.73,
+      "learning_rate": 6.971279373368146e-07,
+      "loss": 0.315,
+      "step": 943
+    },
+    {
+      "epoch": 85.82,
+      "learning_rate": 6.945169712793734e-07,
+      "loss": 0.2923,
+      "step": 944
+    },
+    {
+      "epoch": 85.91,
+      "learning_rate": 6.919060052219321e-07,
+      "loss": 0.3048,
+      "step": 945
+    },
+    {
+      "epoch": 86.0,
+      "learning_rate": 6.89295039164491e-07,
+      "loss": 0.2949,
+      "step": 946
+    },
+    {
+      "epoch": 86.09,
+      "learning_rate": 6.866840731070496e-07,
+      "loss": 0.2835,
+      "step": 947
+    },
+    {
+      "epoch": 86.18,
+      "learning_rate": 6.840731070496084e-07,
+      "loss": 0.3182,
+      "step": 948
+    },
+    {
+      "epoch": 86.27,
+      "learning_rate": 6.814621409921671e-07,
+      "loss": 0.2974,
+      "step": 949
+    },
+    {
+      "epoch": 86.36,
+      "learning_rate": 6.788511749347258e-07,
+      "loss": 0.2749,
+      "step": 950
+    },
+    {
+      "epoch": 86.36,
+      "eval_accuracy_safe": 0.6518093607966045,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9886810358978896,
+      "eval_iou_safe": 0.4765481543905536,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9783128350267929,
+      "eval_loss": 0.2760426700115204,
+      "eval_mean_accuracy": 0.820245198347247,
+      "eval_mean_iou": 0.48495366313911553,
+      "eval_overall_accuracy": 0.9787303867624767,
+      "eval_runtime": 9.6435,
+      "eval_samples_per_second": 6.948,
+      "eval_steps_per_second": 0.518,
+      "step": 950
+    },
+    {
+      "epoch": 86.45,
+      "learning_rate": 6.762402088772847e-07,
+      "loss": 0.319,
+      "step": 951
+    },
+    {
+      "epoch": 86.55,
+      "learning_rate": 6.736292428198434e-07,
+      "loss": 0.2734,
+      "step": 952
+    },
+    {
+      "epoch": 86.64,
+      "learning_rate": 6.710182767624021e-07,
+      "loss": 0.2848,
+      "step": 953
+    },
+    {
+      "epoch": 86.73,
+      "learning_rate": 6.684073107049608e-07,
+      "loss": 0.2765,
+      "step": 954
+    },
+    {
+      "epoch": 86.82,
+      "learning_rate": 6.657963446475195e-07,
+      "loss": 0.2782,
+      "step": 955
+    },
+    {
+      "epoch": 86.91,
+      "learning_rate": 6.631853785900784e-07,
+      "loss": 0.2816,
+      "step": 956
+    },
+    {
+      "epoch": 87.0,
+      "learning_rate": 6.605744125326371e-07,
+      "loss": 0.3259,
+      "step": 957
+    },
+    {
+      "epoch": 87.09,
+      "learning_rate": 6.579634464751959e-07,
+      "loss": 0.2913,
+      "step": 958
+    },
+    {
+      "epoch": 87.18,
+      "learning_rate": 6.553524804177545e-07,
+      "loss": 0.2994,
+      "step": 959
+    },
+    {
+      "epoch": 87.27,
+      "learning_rate": 6.527415143603134e-07,
+      "loss": 0.2725,
+      "step": 960
+    },
+    {
+      "epoch": 87.27,
+      "eval_accuracy_safe": 0.6680833921226209,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9876196006699034,
+      "eval_iou_safe": 0.4764950831797042,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9777417669524591,
+      "eval_loss": 0.2780168652534485,
+      "eval_mean_accuracy": 0.8278514963962622,
+      "eval_mean_iou": 0.4847456167107211,
+      "eval_overall_accuracy": 0.9781810134204466,
+      "eval_runtime": 10.2148,
+      "eval_samples_per_second": 6.559,
+      "eval_steps_per_second": 0.489,
+      "step": 960
+    },
+    {
+      "epoch": 87.36,
+      "learning_rate": 6.501305483028721e-07,
+      "loss": 0.2836,
+      "step": 961
+    },
+    {
+      "epoch": 87.45,
+      "learning_rate": 6.475195822454308e-07,
+      "loss": 0.2821,
+      "step": 962
+    },
+    {
+      "epoch": 87.55,
+      "learning_rate": 6.449086161879896e-07,
+      "loss": 0.3042,
+      "step": 963
+    },
+    {
+      "epoch": 87.64,
+      "learning_rate": 6.422976501305482e-07,
+      "loss": 0.3242,
+      "step": 964
+    },
+    {
+      "epoch": 87.73,
+      "learning_rate": 6.396866840731071e-07,
+      "loss": 0.2735,
+      "step": 965
+    },
+    {
+      "epoch": 87.82,
+      "learning_rate": 6.370757180156658e-07,
+      "loss": 0.3072,
+      "step": 966
+    },
+    {
+      "epoch": 87.91,
+      "learning_rate": 6.344647519582246e-07,
+      "loss": 0.2815,
+      "step": 967
+    },
+    {
+      "epoch": 88.0,
+      "learning_rate": 6.318537859007833e-07,
+      "loss": 0.3286,
+      "step": 968
+    },
+    {
+      "epoch": 88.09,
+      "learning_rate": 6.29242819843342e-07,
+      "loss": 0.2896,
+      "step": 969
+    },
+    {
+      "epoch": 88.18,
+      "learning_rate": 6.266318537859008e-07,
+      "loss": 0.2948,
+      "step": 970
+    },
+    {
+      "epoch": 88.18,
+      "eval_accuracy_safe": 0.6564527507604057,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.989077226042406,
+      "eval_iou_safe": 0.4845336330525837,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9788417622656205,
+      "eval_loss": 0.26358866691589355,
+      "eval_mean_accuracy": 0.8227649884014059,
+      "eval_mean_iou": 0.48779179843940135,
+      "eval_overall_accuracy": 0.9792520323796059,
+      "eval_runtime": 10.7789,
+      "eval_samples_per_second": 6.216,
+      "eval_steps_per_second": 0.464,
+      "step": 970
+    },
+    {
+      "epoch": 88.27,
+      "learning_rate": 6.240208877284595e-07,
+      "loss": 0.3121,
+      "step": 971
+    },
+    {
+      "epoch": 88.36,
+      "learning_rate": 6.214099216710183e-07,
+      "loss": 0.2941,
+      "step": 972
+    },
+    {
+      "epoch": 88.45,
+      "learning_rate": 6.18798955613577e-07,
+      "loss": 0.2852,
+      "step": 973
+    },
+    {
+      "epoch": 88.55,
+      "learning_rate": 6.161879895561358e-07,
+      "loss": 0.2894,
+      "step": 974
+    },
+    {
+      "epoch": 88.64,
+      "learning_rate": 6.135770234986946e-07,
+      "loss": 0.3066,
+      "step": 975
+    },
+    {
+      "epoch": 88.73,
+      "learning_rate": 6.109660574412532e-07,
+      "loss": 0.2867,
+      "step": 976
+    },
+    {
+      "epoch": 88.82,
+      "learning_rate": 6.08355091383812e-07,
+      "loss": 0.2914,
+      "step": 977
+    },
+    {
+      "epoch": 88.91,
+      "learning_rate": 6.057441253263707e-07,
+      "loss": 0.2741,
+      "step": 978
+    },
+    {
+      "epoch": 89.0,
+      "learning_rate": 6.031331592689296e-07,
+      "loss": 0.3029,
+      "step": 979
+    },
+    {
+      "epoch": 89.09,
+      "learning_rate": 6.005221932114883e-07,
+      "loss": 0.2972,
+      "step": 980
+    },
+    {
+      "epoch": 89.09,
+      "eval_accuracy_safe": 0.6722256275033636,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9879379373682813,
+      "eval_iou_safe": 0.48294905349452316,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9781790142676435,
+      "eval_loss": 0.2769739031791687,
+      "eval_mean_accuracy": 0.8300817824358224,
+      "eval_mean_iou": 0.4870426892540556,
+      "eval_overall_accuracy": 0.9786123019545826,
+      "eval_runtime": 10.5252,
+      "eval_samples_per_second": 6.366,
+      "eval_steps_per_second": 0.475,
+      "step": 980
+    },
+    {
+      "epoch": 89.18,
+      "learning_rate": 5.97911227154047e-07,
+      "loss": 0.2912,
+      "step": 981
+    },
+    {
+      "epoch": 89.27,
+      "learning_rate": 5.953002610966057e-07,
+      "loss": 0.3064,
+      "step": 982
+    },
+    {
+      "epoch": 89.36,
+      "learning_rate": 5.926892950391645e-07,
+      "loss": 0.2837,
+      "step": 983
+    },
+    {
+      "epoch": 89.45,
+      "learning_rate": 5.900783289817233e-07,
+      "loss": 0.2857,
+      "step": 984
+    },
+    {
+      "epoch": 89.55,
+      "learning_rate": 5.87467362924282e-07,
+      "loss": 0.2545,
+      "step": 985
+    },
+    {
+      "epoch": 89.64,
+      "learning_rate": 5.848563968668408e-07,
+      "loss": 0.2777,
+      "step": 986
+    },
+    {
+      "epoch": 89.73,
+      "learning_rate": 5.822454308093994e-07,
+      "loss": 0.292,
+      "step": 987
+    },
+    {
+      "epoch": 89.82,
+      "learning_rate": 5.796344647519583e-07,
+      "loss": 0.3179,
+      "step": 988
+    },
+    {
+      "epoch": 89.91,
+      "learning_rate": 5.77023498694517e-07,
+      "loss": 0.2939,
+      "step": 989
+    },
+    {
+      "epoch": 90.0,
+      "learning_rate": 5.744125326370757e-07,
+      "loss": 0.3101,
+      "step": 990
+    },
+    {
+      "epoch": 90.0,
+      "eval_accuracy_safe": 0.6711462176321602,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9881590599293182,
+      "eval_iou_safe": 0.4859155224409652,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9783666378224161,
+      "eval_loss": 0.27649641036987305,
+      "eval_mean_accuracy": 0.8296526387807392,
+      "eval_mean_iou": 0.4880940534211271,
+      "eval_overall_accuracy": 0.978795008872872,
+      "eval_runtime": 9.7226,
+      "eval_samples_per_second": 6.891,
+      "eval_steps_per_second": 0.514,
+      "step": 990
+    },
+    {
+      "epoch": 90.09,
+      "learning_rate": 5.718015665796345e-07,
+      "loss": 0.297,
+      "step": 991
+    },
+    {
+      "epoch": 90.18,
+      "learning_rate": 5.691906005221933e-07,
+      "loss": 0.288,
+      "step": 992
+    },
+    {
+      "epoch": 90.27,
+      "learning_rate": 5.66579634464752e-07,
+      "loss": 0.2584,
+      "step": 993
+    },
+    {
+      "epoch": 90.36,
+      "learning_rate": 5.639686684073107e-07,
+      "loss": 0.2859,
+      "step": 994
+    },
+    {
+      "epoch": 90.45,
+      "learning_rate": 5.613577023498694e-07,
+      "loss": 0.279,
+      "step": 995
+    },
+    {
+      "epoch": 90.55,
+      "learning_rate": 5.587467362924282e-07,
+      "loss": 0.2944,
+      "step": 996
+    },
+    {
+      "epoch": 90.64,
+      "learning_rate": 5.56135770234987e-07,
+      "loss": 0.2943,
+      "step": 997
+    },
+    {
+      "epoch": 90.73,
+      "learning_rate": 5.535248041775457e-07,
+      "loss": 0.2939,
+      "step": 998
+    },
+    {
+      "epoch": 90.82,
+      "learning_rate": 5.509138381201044e-07,
+      "loss": 0.2731,
+      "step": 999
+    },
+    {
+      "epoch": 90.91,
+      "learning_rate": 5.483028720626632e-07,
+      "loss": 0.2874,
+      "step": 1000
+    },
+    {
+      "epoch": 90.91,
+      "eval_accuracy_safe": 0.6689314998785664,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9887533744804734,
+      "eval_iou_safe": 0.4898762051296529,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9788892177058136,
+      "eval_loss": 0.2690369188785553,
+      "eval_mean_accuracy": 0.8288424371795199,
+      "eval_mean_iou": 0.48958847427848884,
+      "eval_overall_accuracy": 0.979306349113806,
+      "eval_runtime": 12.0357,
+      "eval_samples_per_second": 5.567,
+      "eval_steps_per_second": 0.415,
+      "step": 1000
+    },
+    {
+      "epoch": 91.0,
+      "learning_rate": 5.45691906005222e-07,
+      "loss": 0.2744,
+      "step": 1001
+    },
+    {
+      "epoch": 91.09,
+      "learning_rate": 5.430809399477807e-07,
+      "loss": 0.2804,
+      "step": 1002
+    },
+    {
+      "epoch": 91.18,
+      "learning_rate": 5.404699738903395e-07,
+      "loss": 0.3046,
+      "step": 1003
+    },
+    {
+      "epoch": 91.27,
+      "learning_rate": 5.378590078328981e-07,
+      "loss": 0.2818,
+      "step": 1004
+    },
+    {
+      "epoch": 91.36,
+      "learning_rate": 5.352480417754569e-07,
+      "loss": 0.2463,
+      "step": 1005
+    },
+    {
+      "epoch": 91.45,
+      "learning_rate": 5.326370757180157e-07,
+      "loss": 0.2533,
+      "step": 1006
+    },
+    {
+      "epoch": 91.55,
+      "learning_rate": 5.300261096605745e-07,
+      "loss": 0.2837,
+      "step": 1007
+    },
+    {
+      "epoch": 91.64,
+      "learning_rate": 5.274151436031332e-07,
+      "loss": 0.276,
+      "step": 1008
+    },
+    {
+      "epoch": 91.73,
+      "learning_rate": 5.248041775456919e-07,
+      "loss": 0.2684,
+      "step": 1009
+    },
+    {
+      "epoch": 91.82,
+      "learning_rate": 5.221932114882506e-07,
+      "loss": 0.275,
+      "step": 1010
+    },
+    {
+      "epoch": 91.82,
+      "eval_accuracy_safe": 0.6541802074779974,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9901128470154555,
+      "eval_iou_safe": 0.4947347245788654,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9797995942698954,
+      "eval_loss": 0.2592916190624237,
+      "eval_mean_accuracy": 0.8221465272467265,
+      "eval_mean_iou": 0.4915114396162536,
+      "eval_overall_accuracy": 0.9801899354849288,
+      "eval_runtime": 12.6805,
+      "eval_samples_per_second": 5.284,
+      "eval_steps_per_second": 0.394,
+      "step": 1010
+    },
+    {
+      "epoch": 91.91,
+      "learning_rate": 5.195822454308094e-07,
+      "loss": 0.2927,
+      "step": 1011
+    },
+    {
+      "epoch": 92.0,
+      "learning_rate": 5.169712793733682e-07,
+      "loss": 0.3249,
+      "step": 1012
+    },
+    {
+      "epoch": 92.09,
+      "learning_rate": 5.143603133159269e-07,
+      "loss": 0.2613,
+      "step": 1013
+    },
+    {
+      "epoch": 92.18,
+      "learning_rate": 5.117493472584857e-07,
+      "loss": 0.2562,
+      "step": 1014
+    },
+    {
+      "epoch": 92.27,
+      "learning_rate": 5.091383812010444e-07,
+      "loss": 0.283,
+      "step": 1015
+    },
+    {
+      "epoch": 92.36,
+      "learning_rate": 5.065274151436031e-07,
+      "loss": 0.2957,
+      "step": 1016
+    },
+    {
+      "epoch": 92.45,
+      "learning_rate": 5.039164490861619e-07,
+      "loss": 0.2785,
+      "step": 1017
+    },
+    {
+      "epoch": 92.55,
+      "learning_rate": 5.013054830287206e-07,
+      "loss": 0.2555,
+      "step": 1018
+    },
+    {
+      "epoch": 92.64,
+      "learning_rate": 4.986945169712794e-07,
+      "loss": 0.2776,
+      "step": 1019
+    },
+    {
+      "epoch": 92.73,
+      "learning_rate": 4.960835509138382e-07,
+      "loss": 0.2711,
+      "step": 1020
+    },
+    {
+      "epoch": 92.73,
+      "eval_accuracy_safe": 0.6672584145782013,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9893351339167277,
+      "eval_iou_safe": 0.49565376756999,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9794157946407118,
+      "eval_loss": 0.26075512170791626,
+      "eval_mean_accuracy": 0.8282967742474645,
+      "eval_mean_iou": 0.49168985407023397,
+      "eval_overall_accuracy": 0.9798215040520056,
+      "eval_runtime": 11.1156,
+      "eval_samples_per_second": 6.028,
+      "eval_steps_per_second": 0.45,
+      "step": 1020
+    },
+    {
+      "epoch": 92.82,
+      "learning_rate": 4.934725848563969e-07,
+      "loss": 0.2631,
+      "step": 1021
+    },
+    {
+      "epoch": 92.91,
+      "learning_rate": 4.908616187989556e-07,
+      "loss": 0.2891,
+      "step": 1022
+    },
+    {
+      "epoch": 93.0,
+      "learning_rate": 4.882506527415143e-07,
+      "loss": 0.3368,
+      "step": 1023
+    },
+    {
+      "epoch": 93.09,
+      "learning_rate": 4.856396866840732e-07,
+      "loss": 0.2506,
+      "step": 1024
+    },
+    {
+      "epoch": 93.18,
+      "learning_rate": 4.830287206266319e-07,
+      "loss": 0.2542,
+      "step": 1025
+    },
+    {
+      "epoch": 93.27,
+      "learning_rate": 4.804177545691907e-07,
+      "loss": 0.2926,
+      "step": 1026
+    },
+    {
+      "epoch": 93.36,
+      "learning_rate": 4.778067885117493e-07,
+      "loss": 0.2756,
+      "step": 1027
+    },
+    {
+      "epoch": 93.45,
+      "learning_rate": 4.751958224543081e-07,
+      "loss": 0.2448,
+      "step": 1028
+    },
+    {
+      "epoch": 93.55,
+      "learning_rate": 4.7258485639686686e-07,
+      "loss": 0.3061,
+      "step": 1029
+    },
+    {
+      "epoch": 93.64,
+      "learning_rate": 4.6997389033942563e-07,
+      "loss": 0.2691,
+      "step": 1030
+    },
+    {
+      "epoch": 93.64,
+      "eval_accuracy_safe": 0.6818844183330056,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9884273521743757,
+      "eval_iou_safe": 0.4954192037733101,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9789485519513931,
+      "eval_loss": 0.2608976364135742,
+      "eval_mean_accuracy": 0.8351558852536907,
+      "eval_mean_iou": 0.49145591857490106,
+      "eval_overall_accuracy": 0.9793725654260436,
+      "eval_runtime": 9.5589,
+      "eval_samples_per_second": 7.009,
+      "eval_steps_per_second": 0.523,
+      "step": 1030
+    },
+    {
+      "epoch": 93.73,
+      "learning_rate": 4.6736292428198435e-07,
+      "loss": 0.2734,
+      "step": 1031
+    },
+    {
+      "epoch": 93.82,
+      "learning_rate": 4.6475195822454307e-07,
+      "loss": 0.2837,
+      "step": 1032
+    },
+    {
+      "epoch": 93.91,
+      "learning_rate": 4.621409921671018e-07,
+      "loss": 0.2912,
+      "step": 1033
+    },
+    {
+      "epoch": 94.0,
+      "learning_rate": 4.5953002610966057e-07,
+      "loss": 0.2975,
+      "step": 1034
+    },
+    {
+      "epoch": 94.09,
+      "learning_rate": 4.5691906005221934e-07,
+      "loss": 0.2788,
+      "step": 1035
+    },
+    {
+      "epoch": 94.18,
+      "learning_rate": 4.543080939947781e-07,
+      "loss": 0.3211,
+      "step": 1036
+    },
+    {
+      "epoch": 94.27,
+      "learning_rate": 4.516971279373369e-07,
+      "loss": 0.3114,
+      "step": 1037
+    },
+    {
+      "epoch": 94.36,
+      "learning_rate": 4.4908616187989555e-07,
+      "loss": 0.2451,
+      "step": 1038
+    },
+    {
+      "epoch": 94.45,
+      "learning_rate": 4.464751958224543e-07,
+      "loss": 0.3075,
+      "step": 1039
+    },
+    {
+      "epoch": 94.55,
+      "learning_rate": 4.4386422976501305e-07,
+      "loss": 0.274,
+      "step": 1040
+    },
+    {
+      "epoch": 94.55,
+      "eval_accuracy_safe": 0.6722159899152278,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9895092041312664,
+      "eval_iou_safe": 0.5006991902601504,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9797344753826932,
+      "eval_loss": 0.2541664242744446,
+      "eval_mean_accuracy": 0.830862597023247,
+      "eval_mean_iou": 0.49347788854761454,
+      "eval_overall_accuracy": 0.9801368713378906,
+      "eval_runtime": 10.6958,
+      "eval_samples_per_second": 6.264,
+      "eval_steps_per_second": 0.467,
+      "step": 1040
+    },
+    {
+      "epoch": 94.64,
+      "learning_rate": 4.412532637075718e-07,
+      "loss": 0.24,
+      "step": 1041
+    },
+    {
+      "epoch": 94.73,
+      "learning_rate": 4.386422976501306e-07,
+      "loss": 0.2447,
+      "step": 1042
+    },
+    {
+      "epoch": 94.82,
+      "learning_rate": 4.3603133159268937e-07,
+      "loss": 0.2805,
+      "step": 1043
+    },
+    {
+      "epoch": 94.91,
+      "learning_rate": 4.3342036553524803e-07,
+      "loss": 0.2669,
+      "step": 1044
+    },
+    {
+      "epoch": 95.0,
+      "learning_rate": 4.308093994778068e-07,
+      "loss": 0.303,
+      "step": 1045
+    },
+    {
+      "epoch": 95.09,
+      "learning_rate": 4.2819843342036553e-07,
+      "loss": 0.2528,
+      "step": 1046
+    },
+    {
+      "epoch": 95.18,
+      "learning_rate": 4.255874673629243e-07,
+      "loss": 0.2796,
+      "step": 1047
+    },
+    {
+      "epoch": 95.27,
+      "learning_rate": 4.229765013054831e-07,
+      "loss": 0.2867,
+      "step": 1048
+    },
+    {
+      "epoch": 95.36,
+      "learning_rate": 4.2036553524804174e-07,
+      "loss": 0.2571,
+      "step": 1049
+    },
+    {
+      "epoch": 95.45,
+      "learning_rate": 4.177545691906005e-07,
+      "loss": 0.27,
+      "step": 1050
+    },
+    {
+      "epoch": 95.45,
+      "eval_accuracy_safe": 0.6435942806696967,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9918861689920813,
+      "eval_iou_safe": 0.5086651838761769,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9812415734412862,
+      "eval_loss": 0.23565810918807983,
+      "eval_mean_accuracy": 0.8177402248308889,
+      "eval_mean_iou": 0.4966355857724876,
+      "eval_overall_accuracy": 0.9815981850695255,
+      "eval_runtime": 9.9504,
+      "eval_samples_per_second": 6.733,
+      "eval_steps_per_second": 0.502,
+      "step": 1050
+    },
+    {
+      "epoch": 95.55,
+      "learning_rate": 4.151436031331593e-07,
+      "loss": 0.2617,
+      "step": 1051
+    },
+    {
+      "epoch": 95.64,
+      "learning_rate": 4.12532637075718e-07,
+      "loss": 0.2838,
+      "step": 1052
+    },
+    {
+      "epoch": 95.73,
+      "learning_rate": 4.099216710182768e-07,
+      "loss": 0.2938,
+      "step": 1053
+    },
+    {
+      "epoch": 95.82,
+      "learning_rate": 4.0731070496083555e-07,
+      "loss": 0.2417,
+      "step": 1054
+    },
+    {
+      "epoch": 95.91,
+      "learning_rate": 4.046997389033942e-07,
+      "loss": 0.2567,
+      "step": 1055
+    },
+    {
+      "epoch": 96.0,
+      "learning_rate": 4.02088772845953e-07,
+      "loss": 0.3307,
+      "step": 1056
+    },
+    {
+      "epoch": 96.09,
+      "learning_rate": 3.9947780678851177e-07,
+      "loss": 0.2799,
+      "step": 1057
+    },
+    {
+      "epoch": 96.18,
+      "learning_rate": 3.9686684073107054e-07,
+      "loss": 0.2739,
+      "step": 1058
+    },
+    {
+      "epoch": 96.27,
+      "learning_rate": 3.9425587467362926e-07,
+      "loss": 0.2708,
+      "step": 1059
+    },
+    {
+      "epoch": 96.36,
+      "learning_rate": 3.9164490861618804e-07,
+      "loss": 0.255,
+      "step": 1060
+    },
+    {
+      "epoch": 96.36,
+      "eval_accuracy_safe": 0.6671022856504023,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9903374310334044,
+      "eval_iou_safe": 0.5074312624715747,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.980403430127183,
+      "eval_loss": 0.2459607720375061,
+      "eval_mean_accuracy": 0.8287198583419033,
+      "eval_mean_iou": 0.49594489753291926,
+      "eval_overall_accuracy": 0.9807895831207731,
+      "eval_runtime": 11.3645,
+      "eval_samples_per_second": 5.896,
+      "eval_steps_per_second": 0.44,
+      "step": 1060
+    },
+    {
+      "epoch": 96.45,
+      "learning_rate": 3.890339425587467e-07,
+      "loss": 0.2801,
+      "step": 1061
+    },
+    {
+      "epoch": 96.55,
+      "learning_rate": 3.864229765013055e-07,
+      "loss": 0.273,
+      "step": 1062
+    },
+    {
+      "epoch": 96.64,
+      "learning_rate": 3.8381201044386425e-07,
+      "loss": 0.2592,
+      "step": 1063
+    },
+    {
+      "epoch": 96.73,
+      "learning_rate": 3.81201044386423e-07,
+      "loss": 0.2829,
+      "step": 1064
+    },
+    {
+      "epoch": 96.82,
+      "learning_rate": 3.7859007832898174e-07,
+      "loss": 0.2865,
+      "step": 1065
+    },
+    {
+      "epoch": 96.91,
+      "learning_rate": 3.7597911227154046e-07,
+      "loss": 0.2452,
+      "step": 1066
+    },
+    {
+      "epoch": 97.0,
+      "learning_rate": 3.7336814621409924e-07,
+      "loss": 0.2683,
+      "step": 1067
+    },
+    {
+      "epoch": 97.09,
+      "learning_rate": 3.7075718015665796e-07,
+      "loss": 0.2888,
+      "step": 1068
+    },
+    {
+      "epoch": 97.18,
+      "learning_rate": 3.6814621409921673e-07,
+      "loss": 0.265,
+      "step": 1069
+    },
+    {
+      "epoch": 97.27,
+      "learning_rate": 3.6553524804177545e-07,
+      "loss": 0.2756,
+      "step": 1070
+    },
+    {
+      "epoch": 97.27,
+      "eval_accuracy_safe": 0.6633551913832252,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9909028805540396,
+      "eval_iou_safe": 0.5112561019451947,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9808524615525958,
+      "eval_loss": 0.2411348670721054,
+      "eval_mean_accuracy": 0.8271290359686324,
+      "eval_mean_iou": 0.49736952116593014,
+      "eval_overall_accuracy": 0.981227647012739,
+      "eval_runtime": 11.7692,
+      "eval_samples_per_second": 5.693,
+      "eval_steps_per_second": 0.425,
+      "step": 1070
+    },
+    {
+      "epoch": 97.36,
+      "learning_rate": 3.629242819843342e-07,
+      "loss": 0.2541,
+      "step": 1071
+    },
+    {
+      "epoch": 97.45,
+      "learning_rate": 3.6031331592689294e-07,
+      "loss": 0.3003,
+      "step": 1072
+    },
+    {
+      "epoch": 97.55,
+      "learning_rate": 3.5770234986945166e-07,
+      "loss": 0.2535,
+      "step": 1073
+    },
+    {
+      "epoch": 97.64,
+      "learning_rate": 3.5509138381201044e-07,
+      "loss": 0.2349,
+      "step": 1074
+    },
+    {
+      "epoch": 97.73,
+      "learning_rate": 3.524804177545692e-07,
+      "loss": 0.3017,
+      "step": 1075
+    },
+    {
+      "epoch": 97.82,
+      "learning_rate": 3.4986945169712793e-07,
+      "loss": 0.2767,
+      "step": 1076
+    },
+    {
+      "epoch": 97.91,
+      "learning_rate": 3.472584856396867e-07,
+      "loss": 0.2357,
+      "step": 1077
+    },
+    {
+      "epoch": 98.0,
+      "learning_rate": 3.446475195822455e-07,
+      "loss": 0.2671,
+      "step": 1078
+    },
+    {
+      "epoch": 98.09,
+      "learning_rate": 3.420365535248042e-07,
+      "loss": 0.2846,
+      "step": 1079
+    },
+    {
+      "epoch": 98.18,
+      "learning_rate": 3.394255874673629e-07,
+      "loss": 0.2473,
+      "step": 1080
+    },
+    {
+      "epoch": 98.18,
+      "eval_accuracy_safe": 0.6724203067837056,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9903802005603336,
+      "eval_iou_safe": 0.5115604181605686,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9806029058144462,
+      "eval_loss": 0.2446996420621872,
+      "eval_mean_accuracy": 0.8314002536720195,
+      "eval_mean_iou": 0.49738777465833833,
+      "eval_overall_accuracy": 0.9809881751217059,
+      "eval_runtime": 10.9453,
+      "eval_samples_per_second": 6.121,
+      "eval_steps_per_second": 0.457,
+      "step": 1080
+    },
+    {
+      "epoch": 98.27,
+      "learning_rate": 3.368146214099217e-07,
+      "loss": 0.26,
+      "step": 1081
+    },
+    {
+      "epoch": 98.36,
+      "learning_rate": 3.342036553524804e-07,
+      "loss": 0.2765,
+      "step": 1082
+    },
+    {
+      "epoch": 98.45,
+      "learning_rate": 3.315926892950392e-07,
+      "loss": 0.2686,
+      "step": 1083
+    },
+    {
+      "epoch": 98.55,
+      "learning_rate": 3.2898172323759796e-07,
+      "loss": 0.2977,
+      "step": 1084
+    },
+    {
+      "epoch": 98.64,
+      "learning_rate": 3.263707571801567e-07,
+      "loss": 0.2444,
+      "step": 1085
+    },
+    {
+      "epoch": 98.73,
+      "learning_rate": 3.237597911227154e-07,
+      "loss": 0.262,
+      "step": 1086
+    },
+    {
+      "epoch": 98.82,
+      "learning_rate": 3.211488250652741e-07,
+      "loss": 0.2618,
+      "step": 1087
+    },
+    {
+      "epoch": 98.91,
+      "learning_rate": 3.185378590078329e-07,
+      "loss": 0.246,
+      "step": 1088
+    },
+    {
+      "epoch": 99.0,
+      "learning_rate": 3.1592689295039167e-07,
+      "loss": 0.285,
+      "step": 1089
+    },
+    {
+      "epoch": 99.09,
+      "learning_rate": 3.133159268929504e-07,
+      "loss": 0.266,
+      "step": 1090
+    },
+    {
+      "epoch": 99.09,
+      "eval_accuracy_safe": 0.6777942259281962,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9900612771743434,
+      "eval_iou_safe": 0.511777068191146,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9804459747457676,
+      "eval_loss": 0.24548684060573578,
+      "eval_mean_accuracy": 0.8339277515512697,
+      "eval_mean_iou": 0.4974076809789712,
+      "eval_overall_accuracy": 0.9808374091760436,
+      "eval_runtime": 11.7993,
+      "eval_samples_per_second": 5.678,
+      "eval_steps_per_second": 0.424,
+      "step": 1090
+    },
+    {
+      "epoch": 99.18,
+      "learning_rate": 3.1070496083550916e-07,
+      "loss": 0.2705,
+      "step": 1091
+    },
+    {
+      "epoch": 99.27,
+      "learning_rate": 3.080939947780679e-07,
+      "loss": 0.298,
+      "step": 1092
+    },
+    {
+      "epoch": 99.36,
+      "learning_rate": 3.054830287206266e-07,
+      "loss": 0.2351,
+      "step": 1093
+    },
+    {
+      "epoch": 99.45,
+      "learning_rate": 3.0287206266318537e-07,
+      "loss": 0.2721,
+      "step": 1094
+    },
+    {
+      "epoch": 99.55,
+      "learning_rate": 3.0026109660574415e-07,
+      "loss": 0.2686,
+      "step": 1095
+    },
+    {
+      "epoch": 99.64,
+      "learning_rate": 2.9765013054830287e-07,
+      "loss": 0.2837,
+      "step": 1096
+    },
+    {
+      "epoch": 99.73,
+      "learning_rate": 2.9503916449086164e-07,
+      "loss": 0.28,
+      "step": 1097
+    },
+    {
+      "epoch": 99.82,
+      "learning_rate": 2.924281984334204e-07,
+      "loss": 0.2622,
+      "step": 1098
+    },
+    {
+      "epoch": 99.91,
+      "learning_rate": 2.8981723237597913e-07,
+      "loss": 0.2638,
+      "step": 1099
+    },
+    {
+      "epoch": 100.0,
+      "learning_rate": 2.8720626631853785e-07,
+      "loss": 0.2682,
+      "step": 1100
+    },
+    {
+      "epoch": 100.0,
+      "eval_accuracy_safe": 0.6651053773886763,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9911382596240529,
+      "eval_iou_safe": 0.5156260740404243,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.981137303476917,
+      "eval_loss": 0.2358776330947876,
+      "eval_mean_accuracy": 0.8281218185063646,
+      "eval_mean_iou": 0.49892112583911374,
+      "eval_overall_accuracy": 0.9815077710507522,
+      "eval_runtime": 9.6315,
+      "eval_samples_per_second": 6.956,
+      "eval_steps_per_second": 0.519,
+      "step": 1100
+    },
+    {
+      "epoch": 100.09,
+      "learning_rate": 2.8459530026109663e-07,
+      "loss": 0.2801,
+      "step": 1101
+    },
+    {
+      "epoch": 100.18,
+      "learning_rate": 2.8198433420365535e-07,
+      "loss": 0.286,
+      "step": 1102
+    },
+    {
+      "epoch": 100.27,
+      "learning_rate": 2.793733681462141e-07,
+      "loss": 0.2534,
+      "step": 1103
+    },
+    {
+      "epoch": 100.36,
+      "learning_rate": 2.7676240208877284e-07,
+      "loss": 0.2624,
+      "step": 1104
+    },
+    {
+      "epoch": 100.45,
+      "learning_rate": 2.741514360313316e-07,
+      "loss": 0.2565,
+      "step": 1105
+    },
+    {
+      "epoch": 100.55,
+      "learning_rate": 2.7154046997389033e-07,
+      "loss": 0.2646,
+      "step": 1106
+    },
+    {
+      "epoch": 100.64,
+      "learning_rate": 2.6892950391644905e-07,
+      "loss": 0.2764,
+      "step": 1107
+    },
+    {
+      "epoch": 100.73,
+      "learning_rate": 2.6631853785900783e-07,
+      "loss": 0.3186,
+      "step": 1108
+    },
+    {
+      "epoch": 100.82,
+      "learning_rate": 2.637075718015666e-07,
+      "loss": 0.2657,
+      "step": 1109
+    },
+    {
+      "epoch": 100.91,
+      "learning_rate": 2.610966057441253e-07,
+      "loss": 0.2607,
+      "step": 1110
+    },
+    {
+      "epoch": 100.91,
+      "eval_accuracy_safe": 0.6738698000393214,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9905278698323235,
+      "eval_iou_safe": 0.5144442580542631,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9807920754923263,
+      "eval_loss": 0.23757004737854004,
+      "eval_mean_accuracy": 0.8321988349358225,
+      "eval_mean_iou": 0.4984121111821964,
+      "eval_overall_accuracy": 0.9811742981868004,
+      "eval_runtime": 10.0274,
+      "eval_samples_per_second": 6.682,
+      "eval_steps_per_second": 0.499,
+      "step": 1110
+    },
+    {
+      "epoch": 101.0,
+      "learning_rate": 2.584856396866841e-07,
+      "loss": 0.2811,
+      "step": 1111
+    },
+    {
+      "epoch": 101.09,
+      "learning_rate": 2.5587467362924287e-07,
+      "loss": 0.2518,
+      "step": 1112
+    },
+    {
+      "epoch": 101.18,
+      "learning_rate": 2.5326370757180154e-07,
+      "loss": 0.2703,
+      "step": 1113
+    },
+    {
+      "epoch": 101.27,
+      "learning_rate": 2.506527415143603e-07,
+      "loss": 0.2491,
+      "step": 1114
+    },
+    {
+      "epoch": 101.36,
+      "learning_rate": 2.480417754569191e-07,
+      "loss": 0.2838,
+      "step": 1115
+    },
+    {
+      "epoch": 101.45,
+      "learning_rate": 2.454308093994778e-07,
+      "loss": 0.2518,
+      "step": 1116
+    },
+    {
+      "epoch": 101.55,
+      "learning_rate": 2.428198433420366e-07,
+      "loss": 0.2552,
+      "step": 1117
+    },
+    {
+      "epoch": 101.64,
+      "learning_rate": 2.4020887728459535e-07,
+      "loss": 0.2887,
+      "step": 1118
+    },
+    {
+      "epoch": 101.73,
+      "learning_rate": 2.3759791122715404e-07,
+      "loss": 0.277,
+      "step": 1119
+    },
+    {
+      "epoch": 101.82,
+      "learning_rate": 2.3498694516971282e-07,
+      "loss": 0.2506,
+      "step": 1120
+    },
+    {
+      "epoch": 101.82,
+      "eval_accuracy_safe": 0.6726824491809977,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9906665041150856,
+      "eval_iou_safe": 0.5153487204477325,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9808942461066753,
+      "eval_loss": 0.23799440264701843,
+      "eval_mean_accuracy": 0.8316744766480417,
+      "eval_mean_iou": 0.4987476555181359,
+      "eval_overall_accuracy": 0.9812737649946071,
+      "eval_runtime": 10.407,
+      "eval_samples_per_second": 6.438,
+      "eval_steps_per_second": 0.48,
+      "step": 1120
+    },
+    {
+      "epoch": 101.91,
+      "learning_rate": 2.3237597911227154e-07,
+      "loss": 0.2496,
+      "step": 1121
+    },
+    {
+      "epoch": 102.0,
+      "learning_rate": 2.2976501305483028e-07,
+      "loss": 0.273,
+      "step": 1122
+    },
+    {
+      "epoch": 102.09,
+      "learning_rate": 2.2715404699738906e-07,
+      "loss": 0.2565,
+      "step": 1123
+    },
+    {
+      "epoch": 102.18,
+      "learning_rate": 2.2454308093994778e-07,
+      "loss": 0.2824,
+      "step": 1124
+    },
+    {
+      "epoch": 102.27,
+      "learning_rate": 2.2193211488250652e-07,
+      "loss": 0.2724,
+      "step": 1125
+    },
+    {
+      "epoch": 102.36,
+      "learning_rate": 2.193211488250653e-07,
+      "loss": 0.2681,
+      "step": 1126
+    },
+    {
+      "epoch": 102.45,
+      "learning_rate": 2.1671018276762402e-07,
+      "loss": 0.2424,
+      "step": 1127
+    },
+    {
+      "epoch": 102.55,
+      "learning_rate": 2.1409921671018276e-07,
+      "loss": 0.2529,
+      "step": 1128
+    },
+    {
+      "epoch": 102.64,
+      "learning_rate": 2.1148825065274154e-07,
+      "loss": 0.2577,
+      "step": 1129
+    },
+    {
+      "epoch": 102.73,
+      "learning_rate": 2.0887728459530026e-07,
+      "loss": 0.2729,
+      "step": 1130
+    },
+    {
+      "epoch": 102.73,
+      "eval_accuracy_safe": 0.6802209706207764,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9901609553996558,
+      "eval_iou_safe": 0.5145597517748594,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9806164706631543,
+      "eval_loss": 0.23748475313186646,
+      "eval_mean_accuracy": 0.8351909630102161,
+      "eval_mean_iou": 0.49839207414600456,
+      "eval_overall_accuracy": 0.981005825213532,
+      "eval_runtime": 10.7695,
+      "eval_samples_per_second": 6.221,
+      "eval_steps_per_second": 0.464,
+      "step": 1130
+    },
+    {
+      "epoch": 102.82,
+      "learning_rate": 2.06266318537859e-07,
+      "loss": 0.2593,
+      "step": 1131
+    },
+    {
+      "epoch": 102.91,
+      "learning_rate": 2.0365535248041778e-07,
+      "loss": 0.267,
+      "step": 1132
+    },
+    {
+      "epoch": 103.0,
+      "learning_rate": 2.010443864229765e-07,
+      "loss": 0.2543,
+      "step": 1133
+    },
+    {
+      "epoch": 103.09,
+      "learning_rate": 1.9843342036553527e-07,
+      "loss": 0.2673,
+      "step": 1134
+    },
+    {
+      "epoch": 103.18,
+      "learning_rate": 1.9582245430809402e-07,
+      "loss": 0.2496,
+      "step": 1135
+    },
+    {
+      "epoch": 103.27,
+      "learning_rate": 1.9321148825065274e-07,
+      "loss": 0.2757,
+      "step": 1136
+    },
+    {
+      "epoch": 103.36,
+      "learning_rate": 1.906005221932115e-07,
+      "loss": 0.2433,
+      "step": 1137
+    },
+    {
+      "epoch": 103.45,
+      "learning_rate": 1.8798955613577023e-07,
+      "loss": 0.2917,
+      "step": 1138
+    },
+    {
+      "epoch": 103.55,
+      "learning_rate": 1.8537859007832898e-07,
+      "loss": 0.2546,
+      "step": 1139
+    },
+    {
+      "epoch": 103.64,
+      "learning_rate": 1.8276762402088773e-07,
+      "loss": 0.2348,
+      "step": 1140
+    },
+    {
+      "epoch": 103.64,
+      "eval_accuracy_safe": 0.6730949379532076,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9907904712075427,
+      "eval_iou_safe": 0.5171906657661821,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.981029185783124,
+      "eval_loss": 0.23353296518325806,
+      "eval_mean_accuracy": 0.8319427045803751,
+      "eval_mean_iou": 0.49940661718310203,
+      "eval_overall_accuracy": 0.9814062545548624,
+      "eval_runtime": 10.7501,
+      "eval_samples_per_second": 6.232,
+      "eval_steps_per_second": 0.465,
+      "step": 1140
+    },
+    {
+      "epoch": 103.73,
+      "learning_rate": 1.8015665796344647e-07,
+      "loss": 0.2609,
+      "step": 1141
+    },
+    {
+      "epoch": 103.82,
+      "learning_rate": 1.7754569190600522e-07,
+      "loss": 0.2461,
+      "step": 1142
+    },
+    {
+      "epoch": 103.91,
+      "learning_rate": 1.7493472584856397e-07,
+      "loss": 0.3057,
+      "step": 1143
+    },
+    {
+      "epoch": 104.0,
+      "learning_rate": 1.7232375979112274e-07,
+      "loss": 0.2862,
+      "step": 1144
+    },
+    {
+      "epoch": 104.09,
+      "learning_rate": 1.6971279373368146e-07,
+      "loss": 0.2562,
+      "step": 1145
+    },
+    {
+      "epoch": 104.18,
+      "learning_rate": 1.671018276762402e-07,
+      "loss": 0.241,
+      "step": 1146
+    },
+    {
+      "epoch": 104.27,
+      "learning_rate": 1.6449086161879898e-07,
+      "loss": 0.2644,
+      "step": 1147
+    },
+    {
+      "epoch": 104.36,
+      "learning_rate": 1.618798955613577e-07,
+      "loss": 0.2752,
+      "step": 1148
+    },
+    {
+      "epoch": 104.45,
+      "learning_rate": 1.5926892950391645e-07,
+      "loss": 0.2514,
+      "step": 1149
+    },
+    {
+      "epoch": 104.55,
+      "learning_rate": 1.566579634464752e-07,
+      "loss": 0.2409,
+      "step": 1150
+    },
+    {
+      "epoch": 104.55,
+      "eval_accuracy_safe": 0.6781007012309127,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9904347625082679,
+      "eval_iou_safe": 0.5164347505684745,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.980824950444014,
+      "eval_loss": 0.2385135442018509,
+      "eval_mean_accuracy": 0.8342677318695904,
+      "eval_mean_iou": 0.4990865670041629,
+      "eval_overall_accuracy": 0.9812089151410914,
+      "eval_runtime": 9.9272,
+      "eval_samples_per_second": 6.749,
+      "eval_steps_per_second": 0.504,
+      "step": 1150
+    },
+    {
+      "epoch": 104.64,
+      "learning_rate": 1.5404699738903394e-07,
+      "loss": 0.2578,
+      "step": 1151
+    },
+    {
+      "epoch": 104.73,
+      "learning_rate": 1.5143603133159269e-07,
+      "loss": 0.2695,
+      "step": 1152
+    },
+    {
+      "epoch": 104.82,
+      "learning_rate": 1.4882506527415143e-07,
+      "loss": 0.2637,
+      "step": 1153
+    },
+    {
+      "epoch": 104.91,
+      "learning_rate": 1.462140992167102e-07,
+      "loss": 0.2437,
+      "step": 1154
+    },
+    {
+      "epoch": 105.0,
+      "learning_rate": 1.4360313315926893e-07,
+      "loss": 0.2994,
+      "step": 1155
+    },
+    {
+      "epoch": 105.09,
+      "learning_rate": 1.4099216710182767e-07,
+      "loss": 0.2443,
+      "step": 1156
+    },
+    {
+      "epoch": 105.18,
+      "learning_rate": 1.3838120104438642e-07,
+      "loss": 0.2707,
+      "step": 1157
+    },
+    {
+      "epoch": 105.27,
+      "learning_rate": 1.3577023498694517e-07,
+      "loss": 0.2544,
+      "step": 1158
+    },
+    {
+      "epoch": 105.36,
+      "learning_rate": 1.3315926892950391e-07,
+      "loss": 0.2371,
+      "step": 1159
+    },
+    {
+      "epoch": 105.45,
+      "learning_rate": 1.3054830287206266e-07,
+      "loss": 0.2737,
+      "step": 1160
+    },
+    {
+      "epoch": 105.45,
+      "eval_accuracy_safe": 0.677416432473275,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9905725754283728,
+      "eval_iou_safe": 0.517675149619011,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9809411939821061,
+      "eval_loss": 0.23569297790527344,
+      "eval_mean_accuracy": 0.8339945039508239,
+      "eval_mean_iou": 0.4995387812003724,
+      "eval_overall_accuracy": 0.9813224450865788,
+      "eval_runtime": 10.724,
+      "eval_samples_per_second": 6.248,
+      "eval_steps_per_second": 0.466,
+      "step": 1160
+    },
+    {
+      "epoch": 105.55,
+      "learning_rate": 1.2793733681462143e-07,
+      "loss": 0.272,
+      "step": 1161
+    },
+    {
+      "epoch": 105.64,
+      "learning_rate": 1.2532637075718015e-07,
+      "loss": 0.2526,
+      "step": 1162
+    },
+    {
+      "epoch": 105.73,
+      "learning_rate": 1.227154046997389e-07,
+      "loss": 0.2923,
+      "step": 1163
+    },
+    {
+      "epoch": 105.82,
+      "learning_rate": 1.2010443864229767e-07,
+      "loss": 0.2535,
+      "step": 1164
+    },
+    {
+      "epoch": 105.91,
+      "learning_rate": 1.1749347258485641e-07,
+      "loss": 0.257,
+      "step": 1165
+    },
+    {
+      "epoch": 106.0,
+      "learning_rate": 1.1488250652741514e-07,
+      "loss": 0.2114,
+      "step": 1166
+    },
+    {
+      "epoch": 106.09,
+      "learning_rate": 1.1227154046997389e-07,
+      "loss": 0.2743,
+      "step": 1167
+    },
+    {
+      "epoch": 106.18,
+      "learning_rate": 1.0966057441253265e-07,
+      "loss": 0.2485,
+      "step": 1168
+    },
+    {
+      "epoch": 106.27,
+      "learning_rate": 1.0704960835509138e-07,
+      "loss": 0.2781,
+      "step": 1169
+    },
+    {
+      "epoch": 106.36,
+      "learning_rate": 1.0443864229765013e-07,
+      "loss": 0.2857,
+      "step": 1170
+    },
+    {
+      "epoch": 106.36,
+      "eval_accuracy_safe": 0.6658706018866543,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9914503187649804,
+      "eval_iou_safe": 0.5202995990686102,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9814688430393181,
+      "eval_loss": 0.23036494851112366,
+      "eval_mean_accuracy": 0.8286604603258173,
+      "eval_mean_iou": 0.5005894807026428,
+      "eval_overall_accuracy": 0.981833215969712,
+      "eval_runtime": 10.0808,
+      "eval_samples_per_second": 6.646,
+      "eval_steps_per_second": 0.496,
+      "step": 1170
+    },
+    {
+      "epoch": 106.45,
+      "learning_rate": 1.0182767624020889e-07,
+      "loss": 0.2707,
+      "step": 1171
+    },
+    {
+      "epoch": 106.55,
+      "learning_rate": 9.921671018276764e-08,
+      "loss": 0.2403,
+      "step": 1172
+    },
+    {
+      "epoch": 106.64,
+      "learning_rate": 9.660574412532637e-08,
+      "loss": 0.2787,
+      "step": 1173
+    },
+    {
+      "epoch": 106.73,
+      "learning_rate": 9.399477806788512e-08,
+      "loss": 0.2574,
+      "step": 1174
+    },
+    {
+      "epoch": 106.82,
+      "learning_rate": 9.138381201044386e-08,
+      "loss": 0.226,
+      "step": 1175
+    },
+    {
+      "epoch": 106.91,
+      "learning_rate": 8.877284595300261e-08,
+      "loss": 0.2629,
+      "step": 1176
+    },
+    {
+      "epoch": 107.0,
+      "learning_rate": 8.616187989556137e-08,
+      "loss": 0.2674,
+      "step": 1177
+    },
+    {
+      "epoch": 107.09,
+      "learning_rate": 8.35509138381201e-08,
+      "loss": 0.2729,
+      "step": 1178
+    },
+    {
+      "epoch": 107.18,
+      "learning_rate": 8.093994778067885e-08,
+      "loss": 0.2824,
+      "step": 1179
+    },
+    {
+      "epoch": 107.27,
+      "learning_rate": 7.83289817232376e-08,
+      "loss": 0.2503,
+      "step": 1180
+    },
+    {
+      "epoch": 107.27,
+      "eval_accuracy_safe": 0.682225588953011,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9902289524939093,
+      "eval_iou_safe": 0.5171544669118453,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9807430760293058,
+      "eval_loss": 0.23972667753696442,
+      "eval_mean_accuracy": 0.8362272707234601,
+      "eval_mean_iou": 0.49929918098038373,
+      "eval_overall_accuracy": 0.9811310269939366,
+      "eval_runtime": 10.3086,
+      "eval_samples_per_second": 6.499,
+      "eval_steps_per_second": 0.485,
+      "step": 1180
+    },
+    {
+      "epoch": 107.36,
+      "learning_rate": 7.571801566579634e-08,
+      "loss": 0.2528,
+      "step": 1181
+    },
+    {
+      "epoch": 107.45,
+      "learning_rate": 7.31070496083551e-08,
+      "loss": 0.2749,
+      "step": 1182
+    },
+    {
+      "epoch": 107.55,
+      "learning_rate": 7.049608355091384e-08,
+      "loss": 0.2361,
+      "step": 1183
+    },
+    {
+      "epoch": 107.64,
+      "learning_rate": 6.788511749347258e-08,
+      "loss": 0.2599,
+      "step": 1184
+    },
+    {
+      "epoch": 107.73,
+      "learning_rate": 6.527415143603133e-08,
+      "loss": 0.2649,
+      "step": 1185
+    },
+    {
+      "epoch": 107.82,
+      "learning_rate": 6.266318537859008e-08,
+      "loss": 0.2451,
+      "step": 1186
+    },
+    {
+      "epoch": 107.91,
+      "learning_rate": 6.005221932114884e-08,
+      "loss": 0.265,
+      "step": 1187
+    },
+    {
+      "epoch": 108.0,
+      "learning_rate": 5.744125326370757e-08,
+      "loss": 0.2867,
+      "step": 1188
+    },
+    {
+      "epoch": 108.09,
+      "learning_rate": 5.4830287206266324e-08,
+      "loss": 0.2771,
+      "step": 1189
+    },
+    {
+      "epoch": 108.18,
+      "learning_rate": 5.2219321148825064e-08,
+      "loss": 0.2524,
+      "step": 1190
+    },
+    {
+      "epoch": 108.18,
+      "eval_accuracy_safe": 0.6740837544959348,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9909394311922795,
+      "eval_iou_safe": 0.5199116919028611,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9812059186949175,
+      "eval_loss": 0.22940021753311157,
+      "eval_mean_accuracy": 0.8325115928441071,
+      "eval_mean_iou": 0.5003725368659262,
+      "eval_overall_accuracy": 0.9815800225556787,
+      "eval_runtime": 9.8553,
+      "eval_samples_per_second": 6.798,
+      "eval_steps_per_second": 0.507,
+      "step": 1190
+    },
+    {
+      "epoch": 108.27,
+      "learning_rate": 4.960835509138382e-08,
+      "loss": 0.2572,
+      "step": 1191
+    },
+    {
+      "epoch": 108.36,
+      "learning_rate": 4.699738903394256e-08,
+      "loss": 0.2375,
+      "step": 1192
+    },
+    {
+      "epoch": 108.45,
+      "learning_rate": 4.4386422976501305e-08,
+      "loss": 0.2592,
+      "step": 1193
+    },
+    {
+      "epoch": 108.55,
+      "learning_rate": 4.177545691906005e-08,
+      "loss": 0.2463,
+      "step": 1194
+    },
+    {
+      "epoch": 108.64,
+      "learning_rate": 3.91644908616188e-08,
+      "loss": 0.2377,
+      "step": 1195
+    },
+    {
+      "epoch": 108.73,
+      "learning_rate": 3.655352480417755e-08,
+      "loss": 0.2416,
+      "step": 1196
+    },
+    {
+      "epoch": 108.82,
+      "learning_rate": 3.394255874673629e-08,
+      "loss": 0.2558,
+      "step": 1197
+    },
+    {
+      "epoch": 108.91,
+      "learning_rate": 3.133159268929504e-08,
+      "loss": 0.2902,
+      "step": 1198
+    },
+    {
+      "epoch": 109.0,
+      "learning_rate": 2.8720626631853785e-08,
+      "loss": 0.2877,
+      "step": 1199
+    },
+    {
+      "epoch": 109.09,
+      "learning_rate": 2.6109660574412532e-08,
+      "loss": 0.2722,
+      "step": 1200
+    },
+    {
+      "epoch": 109.09,
+      "eval_accuracy_safe": 0.6927633278206329,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9893742073117,
+      "eval_iou_safe": 0.5144704521640547,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9802079010876331,
+      "eval_loss": 0.2461329847574234,
+      "eval_mean_accuracy": 0.8410687675661664,
+      "eval_mean_iou": 0.49822611775056264,
+      "eval_overall_accuracy": 0.9806127975236124,
+      "eval_runtime": 10.7536,
+      "eval_samples_per_second": 6.23,
+      "eval_steps_per_second": 0.465,
+      "step": 1200
+    },
+    {
+      "epoch": 109.18,
+      "learning_rate": 5.058910162002946e-07,
+      "loss": 0.2403,
+      "step": 1201
+    },
+    {
+      "epoch": 109.27,
+      "learning_rate": 5.03681885125184e-07,
+      "loss": 0.2633,
+      "step": 1202
+    },
+    {
+      "epoch": 109.36,
+      "learning_rate": 5.014727540500737e-07,
+      "loss": 0.2314,
+      "step": 1203
+    },
+    {
+      "epoch": 109.45,
+      "learning_rate": 4.992636229749632e-07,
+      "loss": 0.2628,
+      "step": 1204
+    },
+    {
+      "epoch": 109.55,
+      "learning_rate": 4.970544918998527e-07,
+      "loss": 0.2858,
+      "step": 1205
+    },
+    {
+      "epoch": 109.64,
+      "learning_rate": 4.948453608247422e-07,
+      "loss": 0.2463,
+      "step": 1206
+    },
+    {
+      "epoch": 109.73,
+      "learning_rate": 4.926362297496319e-07,
+      "loss": 0.2232,
+      "step": 1207
+    },
+    {
+      "epoch": 109.82,
+      "learning_rate": 4.904270986745214e-07,
+      "loss": 0.2384,
+      "step": 1208
+    },
+    {
+      "epoch": 109.91,
+      "learning_rate": 4.882179675994109e-07,
+      "loss": 0.2542,
+      "step": 1209
+    },
+    {
+      "epoch": 110.0,
+      "learning_rate": 4.860088365243004e-07,
+      "loss": 0.2761,
+      "step": 1210
+    },
+    {
+      "epoch": 110.0,
+      "eval_accuracy_safe": 0.6981218268241063,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.988942346560362,
+      "eval_iou_safe": 0.5136464391621959,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9799386156578034,
+      "eval_loss": 0.2529892325401306,
+      "eval_mean_accuracy": 0.8435320866922341,
+      "eval_mean_iou": 0.4978616849399997,
+      "eval_overall_accuracy": 0.9803519747150478,
+      "eval_runtime": 11.8704,
+      "eval_samples_per_second": 5.644,
+      "eval_steps_per_second": 0.421,
+      "step": 1210
+    },
+    {
+      "epoch": 110.09,
+      "learning_rate": 4.837997054491901e-07,
+      "loss": 0.245,
+      "step": 1211
+    },
+    {
+      "epoch": 110.18,
+      "learning_rate": 4.815905743740796e-07,
+      "loss": 0.269,
+      "step": 1212
+    },
+    {
+      "epoch": 110.27,
+      "learning_rate": 4.79381443298969e-07,
+      "loss": 0.2476,
+      "step": 1213
+    },
+    {
+      "epoch": 110.36,
+      "learning_rate": 4.771723122238586e-07,
+      "loss": 0.2507,
+      "step": 1214
+    },
+    {
+      "epoch": 110.45,
+      "learning_rate": 4.7496318114874815e-07,
+      "loss": 0.2477,
+      "step": 1215
+    },
+    {
+      "epoch": 110.55,
+      "learning_rate": 4.727540500736377e-07,
+      "loss": 0.2485,
+      "step": 1216
+    },
+    {
+      "epoch": 110.64,
+      "learning_rate": 4.7054491899852725e-07,
+      "loss": 0.2638,
+      "step": 1217
+    },
+    {
+      "epoch": 110.73,
+      "learning_rate": 4.683357879234168e-07,
+      "loss": 0.2381,
+      "step": 1218
+    },
+    {
+      "epoch": 110.82,
+      "learning_rate": 4.6612665684830635e-07,
+      "loss": 0.2491,
+      "step": 1219
+    },
+    {
+      "epoch": 110.91,
+      "learning_rate": 4.6391752577319593e-07,
+      "loss": 0.2595,
+      "step": 1220
+    },
+    {
+      "epoch": 110.91,
+      "eval_accuracy_safe": 0.6890586389412531,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9900403324265881,
+      "eval_iou_safe": 0.5196969785076293,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9807582876854402,
+      "eval_loss": 0.23863691091537476,
+      "eval_mean_accuracy": 0.8395494856839205,
+      "eval_mean_iou": 0.5001517553976899,
+      "eval_overall_accuracy": 0.9811498158013643,
+      "eval_runtime": 11.7338,
+      "eval_samples_per_second": 5.71,
+      "eval_steps_per_second": 0.426,
+      "step": 1220
+    },
+    {
+      "epoch": 111.0,
+      "learning_rate": 4.6170839469808545e-07,
+      "loss": 0.2617,
+      "step": 1221
+    },
+    {
+      "epoch": 111.09,
+      "learning_rate": 4.5949926362297503e-07,
+      "loss": 0.2358,
+      "step": 1222
+    },
+    {
+      "epoch": 111.18,
+      "learning_rate": 4.5729013254786455e-07,
+      "loss": 0.251,
+      "step": 1223
+    },
+    {
+      "epoch": 111.27,
+      "learning_rate": 4.55081001472754e-07,
+      "loss": 0.2633,
+      "step": 1224
+    },
+    {
+      "epoch": 111.36,
+      "learning_rate": 4.528718703976436e-07,
+      "loss": 0.2447,
+      "step": 1225
+    },
+    {
+      "epoch": 111.45,
+      "learning_rate": 4.506627393225331e-07,
+      "loss": 0.2509,
+      "step": 1226
+    },
+    {
+      "epoch": 111.55,
+      "learning_rate": 4.484536082474227e-07,
+      "loss": 0.2705,
+      "step": 1227
+    },
+    {
+      "epoch": 111.64,
+      "learning_rate": 4.462444771723122e-07,
+      "loss": 0.2494,
+      "step": 1228
+    },
+    {
+      "epoch": 111.73,
+      "learning_rate": 4.440353460972018e-07,
+      "loss": 0.2625,
+      "step": 1229
+    },
+    {
+      "epoch": 111.82,
+      "learning_rate": 4.418262150220913e-07,
+      "loss": 0.2877,
+      "step": 1230
+    },
+    {
+      "epoch": 111.82,
+      "eval_accuracy_safe": 0.6779060219505707,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9911065784929943,
+      "eval_iou_safe": 0.5250508331865317,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.981484545675428,
+      "eval_loss": 0.22597333788871765,
+      "eval_mean_accuracy": 0.8345063002217825,
+      "eval_mean_iou": 0.5021784596206532,
+      "eval_overall_accuracy": 0.9818551362450443,
+      "eval_runtime": 10.6701,
+      "eval_samples_per_second": 6.279,
+      "eval_steps_per_second": 0.469,
+      "step": 1230
+    },
+    {
+      "epoch": 111.91,
+      "learning_rate": 4.396170839469809e-07,
+      "loss": 0.2425,
+      "step": 1231
+    },
+    {
+      "epoch": 112.0,
+      "learning_rate": 4.374079528718704e-07,
+      "loss": 0.274,
+      "step": 1232
+    },
+    {
+      "epoch": 112.09,
+      "learning_rate": 4.3519882179676e-07,
+      "loss": 0.2509,
+      "step": 1233
+    },
+    {
+      "epoch": 112.18,
+      "learning_rate": 4.3298969072164947e-07,
+      "loss": 0.2546,
+      "step": 1234
+    },
+    {
+      "epoch": 112.27,
+      "learning_rate": 4.30780559646539e-07,
+      "loss": 0.2595,
+      "step": 1235
+    },
+    {
+      "epoch": 112.36,
+      "learning_rate": 4.2857142857142857e-07,
+      "loss": 0.2535,
+      "step": 1236
+    },
+    {
+      "epoch": 112.45,
+      "learning_rate": 4.263622974963181e-07,
+      "loss": 0.2524,
+      "step": 1237
+    },
+    {
+      "epoch": 112.55,
+      "learning_rate": 4.2415316642120767e-07,
+      "loss": 0.2372,
+      "step": 1238
+    },
+    {
+      "epoch": 112.64,
+      "learning_rate": 4.219440353460972e-07,
+      "loss": 0.2495,
+      "step": 1239
+    },
+    {
+      "epoch": 112.73,
+      "learning_rate": 4.1973490427098677e-07,
+      "loss": 0.2307,
+      "step": 1240
+    },
+    {
+      "epoch": 112.73,
+      "eval_accuracy_safe": 0.6717495306494577,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9916446297021398,
+      "eval_iou_safe": 0.5274904076829351,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9818352893830999,
+      "eval_loss": 0.22187493741512299,
+      "eval_mean_accuracy": 0.8316970801757988,
+      "eval_mean_iou": 0.5031085656886783,
+      "eval_overall_accuracy": 0.9821954414026061,
+      "eval_runtime": 10.6767,
+      "eval_samples_per_second": 6.275,
+      "eval_steps_per_second": 0.468,
+      "step": 1240
+    }
+  ],
+  "max_steps": 1430,
+  "num_train_epochs": 130,
+  "total_flos": 6.932825583969632e+18,
+  "trial_name": null,
+  "trial_params": null
+}