diff --git "a/checkpoint-2160/trainer_state.json" "b/checkpoint-2160/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-2160/trainer_state.json"
+++ /dev/null
@@ -1,15028 +0,0 @@
-{
-  "best_metric": 0.01912616565823555,
-  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/parking-terrain/checkpoint-560",
-  "epoch": 540.0,
-  "global_step": 2160,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.25,
-      "learning_rate": 4.1666666666666667e-07,
-      "loss": 1.3152,
-      "step": 1
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 8.333333333333333e-07,
-      "loss": 1.3189,
-      "step": 2
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.25e-06,
-      "loss": 1.3046,
-      "step": 3
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.6666666666666667e-06,
-      "loss": 1.3124,
-      "step": 4
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 2.0833333333333334e-06,
-      "loss": 1.3014,
-      "step": 5
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 2.5e-06,
-      "loss": 1.2884,
-      "step": 6
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 2.916666666666667e-06,
-      "loss": 1.2836,
-      "step": 7
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 3.3333333333333333e-06,
-      "loss": 1.266,
-      "step": 8
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 3.75e-06,
-      "loss": 1.2544,
-      "step": 9
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 4.166666666666667e-06,
-      "loss": 1.2427,
-      "step": 10
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 4.583333333333333e-06,
-      "loss": 1.2271,
-      "step": 11
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 5e-06,
-      "loss": 1.2266,
-      "step": 12
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 5.416666666666667e-06,
-      "loss": 1.2002,
-      "step": 13
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 5.833333333333334e-06,
-      "loss": 1.1886,
-      "step": 14
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 6.25e-06,
-      "loss": 1.1605,
-      "step": 15
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 6.666666666666667e-06,
-      "loss": 1.1389,
-      "step": 16
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 7.083333333333334e-06,
-      "loss": 1.1115,
-      "step": 17
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 7.5e-06,
-      "loss": 1.0863,
-      "step": 18
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 7.916666666666667e-06,
-      "loss": 1.0763,
-      "step": 19
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 8.333333333333334e-06,
-      "loss": 1.0322,
-      "step": 20
-    },
-    {
-      "epoch": 5.0,
-      "eval_accuracy_ELSE": 0.7236194227710022,
-      "eval_accuracy_road": 0.9371333371304893,
-      "eval_accuracy_sidewalk": 0.5361869895803999,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.7051860054312004,
-      "eval_iou_road": 0.6015263687048716,
-      "eval_iou_sidewalk": 0.40198458777578383,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 1.1061346530914307,
-      "eval_mean_accuracy": 0.7323132498272971,
-      "eval_mean_iou": 0.42717424047796393,
-      "eval_overall_accuracy": 0.7757415771484375,
-      "eval_runtime": 0.7193,
-      "eval_samples_per_second": 1.39,
-      "eval_steps_per_second": 1.39,
-      "step": 20
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 8.75e-06,
-      "loss": 1.0201,
-      "step": 21
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 9.166666666666666e-06,
-      "loss": 0.9859,
-      "step": 22
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 9.583333333333334e-06,
-      "loss": 0.9649,
-      "step": 23
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1e-05,
-      "loss": 0.9171,
-      "step": 24
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.0416666666666668e-05,
-      "loss": 0.8769,
-      "step": 25
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.0833333333333334e-05,
-      "loss": 0.882,
-      "step": 26
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.125e-05,
-      "loss": 0.8217,
-      "step": 27
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 1.1666666666666668e-05,
-      "loss": 0.7833,
-      "step": 28
-    },
-    {
-      "epoch": 7.25,
-      "learning_rate": 1.2083333333333333e-05,
-      "loss": 0.7459,
-      "step": 29
-    },
-    {
-      "epoch": 7.5,
-      "learning_rate": 1.25e-05,
-      "loss": 0.7156,
-      "step": 30
-    },
-    {
-      "epoch": 7.75,
-      "learning_rate": 1.2916666666666668e-05,
-      "loss": 0.6706,
-      "step": 31
-    },
-    {
-      "epoch": 8.0,
-      "learning_rate": 1.3333333333333333e-05,
-      "loss": 0.6556,
-      "step": 32
-    },
-    {
-      "epoch": 8.25,
-      "learning_rate": 1.3750000000000002e-05,
-      "loss": 0.6085,
-      "step": 33
-    },
-    {
-      "epoch": 8.5,
-      "learning_rate": 1.4166666666666668e-05,
-      "loss": 0.5801,
-      "step": 34
-    },
-    {
-      "epoch": 8.75,
-      "learning_rate": 1.4583333333333335e-05,
-      "loss": 0.5008,
-      "step": 35
-    },
-    {
-      "epoch": 9.0,
-      "learning_rate": 1.5e-05,
-      "loss": 0.5359,
-      "step": 36
-    },
-    {
-      "epoch": 9.25,
-      "learning_rate": 1.5416666666666668e-05,
-      "loss": 0.4526,
-      "step": 37
-    },
-    {
-      "epoch": 9.5,
-      "learning_rate": 1.5833333333333333e-05,
-      "loss": 0.461,
-      "step": 38
-    },
-    {
-      "epoch": 9.75,
-      "learning_rate": 1.6250000000000002e-05,
-      "loss": 0.3945,
-      "step": 39
-    },
-    {
-      "epoch": 10.0,
-      "learning_rate": 1.6666666666666667e-05,
-      "loss": 0.3821,
-      "step": 40
-    },
-    {
-      "epoch": 10.0,
-      "eval_accuracy_ELSE": 0.9475472637354313,
-      "eval_accuracy_road": 0.9957851569174688,
-      "eval_accuracy_sidewalk": 0.7203604618417347,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9428740645022344,
-      "eval_iou_road": 0.876176157363904,
-      "eval_iou_sidewalk": 0.5966180758017493,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.332696795463562,
-      "eval_mean_accuracy": 0.8878976274982117,
-      "eval_mean_iou": 0.805222765889296,
-      "eval_overall_accuracy": 0.954315185546875,
-      "eval_runtime": 0.6036,
-      "eval_samples_per_second": 1.657,
-      "eval_steps_per_second": 1.657,
-      "step": 40
-    },
-    {
-      "epoch": 10.25,
-      "learning_rate": 1.7083333333333333e-05,
-      "loss": 0.3487,
-      "step": 41
-    },
-    {
-      "epoch": 10.5,
-      "learning_rate": 1.75e-05,
-      "loss": 0.3294,
-      "step": 42
-    },
-    {
-      "epoch": 10.75,
-      "learning_rate": 1.7916666666666667e-05,
-      "loss": 0.3092,
-      "step": 43
-    },
-    {
-      "epoch": 11.0,
-      "learning_rate": 1.8333333333333333e-05,
-      "loss": 0.2778,
-      "step": 44
-    },
-    {
-      "epoch": 11.25,
-      "learning_rate": 1.8750000000000002e-05,
-      "loss": 0.2508,
-      "step": 45
-    },
-    {
-      "epoch": 11.5,
-      "learning_rate": 1.9166666666666667e-05,
-      "loss": 0.2453,
-      "step": 46
-    },
-    {
-      "epoch": 11.75,
-      "learning_rate": 1.9583333333333333e-05,
-      "loss": 0.2169,
-      "step": 47
-    },
-    {
-      "epoch": 12.0,
-      "learning_rate": 2e-05,
-      "loss": 0.2321,
-      "step": 48
-    },
-    {
-      "epoch": 12.25,
-      "learning_rate": 2.0416666666666667e-05,
-      "loss": 0.2083,
-      "step": 49
-    },
-    {
-      "epoch": 12.5,
-      "learning_rate": 2.0833333333333336e-05,
-      "loss": 0.1772,
-      "step": 50
-    },
-    {
-      "epoch": 12.75,
-      "learning_rate": 2.125e-05,
-      "loss": 0.1643,
-      "step": 51
-    },
-    {
-      "epoch": 13.0,
-      "learning_rate": 2.1666666666666667e-05,
-      "loss": 0.1781,
-      "step": 52
-    },
-    {
-      "epoch": 13.25,
-      "learning_rate": 2.2083333333333333e-05,
-      "loss": 0.1629,
-      "step": 53
-    },
-    {
-      "epoch": 13.5,
-      "learning_rate": 2.25e-05,
-      "loss": 0.1367,
-      "step": 54
-    },
-    {
-      "epoch": 13.75,
-      "learning_rate": 2.2916666666666667e-05,
-      "loss": 0.1678,
-      "step": 55
-    },
-    {
-      "epoch": 14.0,
-      "learning_rate": 2.3333333333333336e-05,
-      "loss": 0.1151,
-      "step": 56
-    },
-    {
-      "epoch": 14.25,
-      "learning_rate": 2.375e-05,
-      "loss": 0.1134,
-      "step": 57
-    },
-    {
-      "epoch": 14.5,
-      "learning_rate": 2.4166666666666667e-05,
-      "loss": 0.113,
-      "step": 58
-    },
-    {
-      "epoch": 14.75,
-      "learning_rate": 2.4583333333333332e-05,
-      "loss": 0.1298,
-      "step": 59
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 2.5e-05,
-      "loss": 0.1146,
-      "step": 60
-    },
-    {
-      "epoch": 15.0,
-      "eval_accuracy_ELSE": 0.9833345958639497,
-      "eval_accuracy_road": 0.9951443868542462,
-      "eval_accuracy_sidewalk": 0.7050126724866235,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9793074460861975,
-      "eval_iou_road": 0.9435900897859988,
-      "eval_iou_sidewalk": 0.6242363795038025,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09062469005584717,
-      "eval_mean_accuracy": 0.8944972184016065,
-      "eval_mean_iou": 0.8490446384586662,
-      "eval_overall_accuracy": 0.9789581298828125,
-      "eval_runtime": 0.5877,
-      "eval_samples_per_second": 1.701,
-      "eval_steps_per_second": 1.701,
-      "step": 60
-    },
-    {
-      "epoch": 15.25,
-      "learning_rate": 2.5416666666666667e-05,
-      "loss": 0.1079,
-      "step": 61
-    },
-    {
-      "epoch": 15.5,
-      "learning_rate": 2.5833333333333336e-05,
-      "loss": 0.0953,
-      "step": 62
-    },
-    {
-      "epoch": 15.75,
-      "learning_rate": 2.625e-05,
-      "loss": 0.094,
-      "step": 63
-    },
-    {
-      "epoch": 16.0,
-      "learning_rate": 2.6666666666666667e-05,
-      "loss": 0.1031,
-      "step": 64
-    },
-    {
-      "epoch": 16.25,
-      "learning_rate": 2.7083333333333332e-05,
-      "loss": 0.0991,
-      "step": 65
-    },
-    {
-      "epoch": 16.5,
-      "learning_rate": 2.7500000000000004e-05,
-      "loss": 0.09,
-      "step": 66
-    },
-    {
-      "epoch": 16.75,
-      "learning_rate": 2.791666666666667e-05,
-      "loss": 0.0938,
-      "step": 67
-    },
-    {
-      "epoch": 17.0,
-      "learning_rate": 2.8333333333333335e-05,
-      "loss": 0.0784,
-      "step": 68
-    },
-    {
-      "epoch": 17.25,
-      "learning_rate": 2.8749999999999997e-05,
-      "loss": 0.074,
-      "step": 69
-    },
-    {
-      "epoch": 17.5,
-      "learning_rate": 2.916666666666667e-05,
-      "loss": 0.0754,
-      "step": 70
-    },
-    {
-      "epoch": 17.75,
-      "learning_rate": 2.9583333333333335e-05,
-      "loss": 0.0908,
-      "step": 71
-    },
-    {
-      "epoch": 18.0,
-      "learning_rate": 3e-05,
-      "loss": 0.0746,
-      "step": 72
-    },
-    {
-      "epoch": 18.25,
-      "learning_rate": 3.0416666666666666e-05,
-      "loss": 0.0705,
-      "step": 73
-    },
-    {
-      "epoch": 18.5,
-      "learning_rate": 3.0833333333333335e-05,
-      "loss": 0.0786,
-      "step": 74
-    },
-    {
-      "epoch": 18.75,
-      "learning_rate": 3.125e-05,
-      "loss": 0.0759,
-      "step": 75
-    },
-    {
-      "epoch": 19.0,
-      "learning_rate": 3.1666666666666666e-05,
-      "loss": 0.0743,
-      "step": 76
-    },
-    {
-      "epoch": 19.25,
-      "learning_rate": 3.208333333333334e-05,
-      "loss": 0.0665,
-      "step": 77
-    },
-    {
-      "epoch": 19.5,
-      "learning_rate": 3.2500000000000004e-05,
-      "loss": 0.0642,
-      "step": 78
-    },
-    {
-      "epoch": 19.75,
-      "learning_rate": 3.291666666666667e-05,
-      "loss": 0.0707,
-      "step": 79
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 3.3333333333333335e-05,
-      "loss": 0.0668,
-      "step": 80
-    },
-    {
-      "epoch": 20.0,
-      "eval_accuracy_ELSE": 0.9925168006752735,
-      "eval_accuracy_road": 0.9965967989975508,
-      "eval_accuracy_sidewalk": 0.6823430019712757,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9843781863455368,
-      "eval_iou_road": 0.9669927325983033,
-      "eval_iou_sidewalk": 0.663653793481238,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.05583948642015457,
-      "eval_mean_accuracy": 0.8904855338813666,
-      "eval_mean_iou": 0.8716749041416927,
-      "eval_overall_accuracy": 0.9852066040039062,
-      "eval_runtime": 0.5774,
-      "eval_samples_per_second": 1.732,
-      "eval_steps_per_second": 1.732,
-      "step": 80
-    },
-    {
-      "epoch": 20.25,
-      "learning_rate": 3.375000000000001e-05,
-      "loss": 0.0614,
-      "step": 81
-    },
-    {
-      "epoch": 20.5,
-      "learning_rate": 3.4166666666666666e-05,
-      "loss": 0.0692,
-      "step": 82
-    },
-    {
-      "epoch": 20.75,
-      "learning_rate": 3.458333333333333e-05,
-      "loss": 0.0609,
-      "step": 83
-    },
-    {
-      "epoch": 21.0,
-      "learning_rate": 3.5e-05,
-      "loss": 0.085,
-      "step": 84
-    },
-    {
-      "epoch": 21.25,
-      "learning_rate": 3.541666666666667e-05,
-      "loss": 0.0666,
-      "step": 85
-    },
-    {
-      "epoch": 21.5,
-      "learning_rate": 3.5833333333333335e-05,
-      "loss": 0.059,
-      "step": 86
-    },
-    {
-      "epoch": 21.75,
-      "learning_rate": 3.625e-05,
-      "loss": 0.0564,
-      "step": 87
-    },
-    {
-      "epoch": 22.0,
-      "learning_rate": 3.6666666666666666e-05,
-      "loss": 0.0598,
-      "step": 88
-    },
-    {
-      "epoch": 22.25,
-      "learning_rate": 3.708333333333334e-05,
-      "loss": 0.0542,
-      "step": 89
-    },
-    {
-      "epoch": 22.5,
-      "learning_rate": 3.7500000000000003e-05,
-      "loss": 0.0556,
-      "step": 90
-    },
-    {
-      "epoch": 22.75,
-      "learning_rate": 3.791666666666667e-05,
-      "loss": 0.0584,
-      "step": 91
-    },
-    {
-      "epoch": 23.0,
-      "learning_rate": 3.8333333333333334e-05,
-      "loss": 0.0563,
-      "step": 92
-    },
-    {
-      "epoch": 23.25,
-      "learning_rate": 3.875e-05,
-      "loss": 0.0526,
-      "step": 93
-    },
-    {
-      "epoch": 23.5,
-      "learning_rate": 3.9166666666666665e-05,
-      "loss": 0.0595,
-      "step": 94
-    },
-    {
-      "epoch": 23.75,
-      "learning_rate": 3.958333333333333e-05,
-      "loss": 0.0521,
-      "step": 95
-    },
-    {
-      "epoch": 24.0,
-      "learning_rate": 4e-05,
-      "loss": 0.0637,
-      "step": 96
-    },
-    {
-      "epoch": 24.25,
-      "learning_rate": 4.041666666666667e-05,
-      "loss": 0.0526,
-      "step": 97
-    },
-    {
-      "epoch": 24.5,
-      "learning_rate": 4.0833333333333334e-05,
-      "loss": 0.0589,
-      "step": 98
-    },
-    {
-      "epoch": 24.75,
-      "learning_rate": 4.125e-05,
-      "loss": 0.0513,
-      "step": 99
-    },
-    {
-      "epoch": 25.0,
-      "learning_rate": 4.166666666666667e-05,
-      "loss": 0.0491,
-      "step": 100
-    },
-    {
-      "epoch": 25.0,
-      "eval_accuracy_ELSE": 0.9903145865572954,
-      "eval_accuracy_road": 0.9915560745001993,
-      "eval_accuracy_sidewalk": 0.8296254576175726,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9854569928657962,
-      "eval_iou_road": 0.9708338561490094,
-      "eval_iou_sidewalk": 0.7111647555823778,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.047440506517887115,
-      "eval_mean_accuracy": 0.937165372891689,
-      "eval_mean_iou": 0.8891518681990611,
-      "eval_overall_accuracy": 0.9862937927246094,
-      "eval_runtime": 0.7221,
-      "eval_samples_per_second": 1.385,
-      "eval_steps_per_second": 1.385,
-      "step": 100
-    },
-    {
-      "epoch": 25.25,
-      "learning_rate": 4.208333333333334e-05,
-      "loss": 0.0548,
-      "step": 101
-    },
-    {
-      "epoch": 25.5,
-      "learning_rate": 4.25e-05,
-      "loss": 0.0491,
-      "step": 102
-    },
-    {
-      "epoch": 25.75,
-      "learning_rate": 4.291666666666667e-05,
-      "loss": 0.0552,
-      "step": 103
-    },
-    {
-      "epoch": 26.0,
-      "learning_rate": 4.3333333333333334e-05,
-      "loss": 0.0476,
-      "step": 104
-    },
-    {
-      "epoch": 26.25,
-      "learning_rate": 4.375e-05,
-      "loss": 0.051,
-      "step": 105
-    },
-    {
-      "epoch": 26.5,
-      "learning_rate": 4.4166666666666665e-05,
-      "loss": 0.0465,
-      "step": 106
-    },
-    {
-      "epoch": 26.75,
-      "learning_rate": 4.458333333333334e-05,
-      "loss": 0.0502,
-      "step": 107
-    },
-    {
-      "epoch": 27.0,
-      "learning_rate": 4.5e-05,
-      "loss": 0.0479,
-      "step": 108
-    },
-    {
-      "epoch": 27.25,
-      "learning_rate": 4.541666666666667e-05,
-      "loss": 0.0526,
-      "step": 109
-    },
-    {
-      "epoch": 27.5,
-      "learning_rate": 4.5833333333333334e-05,
-      "loss": 0.0471,
-      "step": 110
-    },
-    {
-      "epoch": 27.75,
-      "learning_rate": 4.6250000000000006e-05,
-      "loss": 0.0405,
-      "step": 111
-    },
-    {
-      "epoch": 28.0,
-      "learning_rate": 4.666666666666667e-05,
-      "loss": 0.0449,
-      "step": 112
-    },
-    {
-      "epoch": 28.25,
-      "learning_rate": 4.708333333333334e-05,
-      "loss": 0.0443,
-      "step": 113
-    },
-    {
-      "epoch": 28.5,
-      "learning_rate": 4.75e-05,
-      "loss": 0.0406,
-      "step": 114
-    },
-    {
-      "epoch": 28.75,
-      "learning_rate": 4.791666666666667e-05,
-      "loss": 0.0436,
-      "step": 115
-    },
-    {
-      "epoch": 29.0,
-      "learning_rate": 4.8333333333333334e-05,
-      "loss": 0.052,
-      "step": 116
-    },
-    {
-      "epoch": 29.25,
-      "learning_rate": 4.875e-05,
-      "loss": 0.0468,
-      "step": 117
-    },
-    {
-      "epoch": 29.5,
-      "learning_rate": 4.9166666666666665e-05,
-      "loss": 0.0402,
-      "step": 118
-    },
-    {
-      "epoch": 29.75,
-      "learning_rate": 4.958333333333334e-05,
-      "loss": 0.0456,
-      "step": 119
-    },
-    {
-      "epoch": 30.0,
-      "learning_rate": 5e-05,
-      "loss": 0.04,
-      "step": 120
-    },
-    {
-      "epoch": 30.0,
-      "eval_accuracy_ELSE": 0.9936638999210017,
-      "eval_accuracy_road": 0.9914136811528166,
-      "eval_accuracy_sidewalk": 0.877640101379893,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9889762507404815,
-      "eval_iou_road": 0.9755362822434882,
-      "eval_iou_sidewalk": 0.8067564069376133,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.03659357875585556,
-      "eval_mean_accuracy": 0.9542392274845705,
-      "eval_mean_iou": 0.9237563133071943,
-      "eval_overall_accuracy": 0.9899177551269531,
-      "eval_runtime": 0.6094,
-      "eval_samples_per_second": 1.641,
-      "eval_steps_per_second": 1.641,
-      "step": 120
-    },
-    {
-      "epoch": 30.25,
-      "learning_rate": 4.99780701754386e-05,
-      "loss": 0.0374,
-      "step": 121
-    },
-    {
-      "epoch": 30.5,
-      "learning_rate": 4.9956140350877195e-05,
-      "loss": 0.0435,
-      "step": 122
-    },
-    {
-      "epoch": 30.75,
-      "learning_rate": 4.9934210526315795e-05,
-      "loss": 0.0442,
-      "step": 123
-    },
-    {
-      "epoch": 31.0,
-      "learning_rate": 4.991228070175439e-05,
-      "loss": 0.0387,
-      "step": 124
-    },
-    {
-      "epoch": 31.25,
-      "learning_rate": 4.989035087719299e-05,
-      "loss": 0.0422,
-      "step": 125
-    },
-    {
-      "epoch": 31.5,
-      "learning_rate": 4.986842105263158e-05,
-      "loss": 0.0398,
-      "step": 126
-    },
-    {
-      "epoch": 31.75,
-      "learning_rate": 4.984649122807018e-05,
-      "loss": 0.0428,
-      "step": 127
-    },
-    {
-      "epoch": 32.0,
-      "learning_rate": 4.9824561403508773e-05,
-      "loss": 0.0375,
-      "step": 128
-    },
-    {
-      "epoch": 32.25,
-      "learning_rate": 4.980263157894737e-05,
-      "loss": 0.0447,
-      "step": 129
-    },
-    {
-      "epoch": 32.5,
-      "learning_rate": 4.9780701754385966e-05,
-      "loss": 0.0358,
-      "step": 130
-    },
-    {
-      "epoch": 32.75,
-      "learning_rate": 4.9758771929824566e-05,
-      "loss": 0.0357,
-      "step": 131
-    },
-    {
-      "epoch": 33.0,
-      "learning_rate": 4.973684210526316e-05,
-      "loss": 0.0369,
-      "step": 132
-    },
-    {
-      "epoch": 33.25,
-      "learning_rate": 4.971491228070176e-05,
-      "loss": 0.0333,
-      "step": 133
-    },
-    {
-      "epoch": 33.5,
-      "learning_rate": 4.969298245614035e-05,
-      "loss": 0.0381,
-      "step": 134
-    },
-    {
-      "epoch": 33.75,
-      "learning_rate": 4.967105263157895e-05,
-      "loss": 0.036,
-      "step": 135
-    },
-    {
-      "epoch": 34.0,
-      "learning_rate": 4.9649122807017544e-05,
-      "loss": 0.0396,
-      "step": 136
-    },
-    {
-      "epoch": 34.25,
-      "learning_rate": 4.9627192982456144e-05,
-      "loss": 0.0376,
-      "step": 137
-    },
-    {
-      "epoch": 34.5,
-      "learning_rate": 4.960526315789474e-05,
-      "loss": 0.0388,
-      "step": 138
-    },
-    {
-      "epoch": 34.75,
-      "learning_rate": 4.958333333333334e-05,
-      "loss": 0.0311,
-      "step": 139
-    },
-    {
-      "epoch": 35.0,
-      "learning_rate": 4.956140350877193e-05,
-      "loss": 0.0339,
-      "step": 140
-    },
-    {
-      "epoch": 35.0,
-      "eval_accuracy_ELSE": 0.9937234192214875,
-      "eval_accuracy_road": 0.9900751836874181,
-      "eval_accuracy_sidewalk": 0.9129822585187272,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9891420261754726,
-      "eval_iou_road": 0.9801381449111926,
-      "eval_iou_sidewalk": 0.8095892121363466,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.03316768258810043,
-      "eval_mean_accuracy": 0.9655936204758776,
-      "eval_mean_iou": 0.9262897944076706,
-      "eval_overall_accuracy": 0.9905586242675781,
-      "eval_runtime": 0.6199,
-      "eval_samples_per_second": 1.613,
-      "eval_steps_per_second": 1.613,
-      "step": 140
-    },
-    {
-      "epoch": 35.25,
-      "learning_rate": 4.953947368421053e-05,
-      "loss": 0.0351,
-      "step": 141
-    },
-    {
-      "epoch": 35.5,
-      "learning_rate": 4.951754385964912e-05,
-      "loss": 0.0318,
-      "step": 142
-    },
-    {
-      "epoch": 35.75,
-      "learning_rate": 4.949561403508772e-05,
-      "loss": 0.0354,
-      "step": 143
-    },
-    {
-      "epoch": 36.0,
-      "learning_rate": 4.9473684210526315e-05,
-      "loss": 0.0385,
-      "step": 144
-    },
-    {
-      "epoch": 36.25,
-      "learning_rate": 4.9451754385964915e-05,
-      "loss": 0.0342,
-      "step": 145
-    },
-    {
-      "epoch": 36.5,
-      "learning_rate": 4.942982456140351e-05,
-      "loss": 0.033,
-      "step": 146
-    },
-    {
-      "epoch": 36.75,
-      "learning_rate": 4.940789473684211e-05,
-      "loss": 0.0337,
-      "step": 147
-    },
-    {
-      "epoch": 37.0,
-      "learning_rate": 4.93859649122807e-05,
-      "loss": 0.0324,
-      "step": 148
-    },
-    {
-      "epoch": 37.25,
-      "learning_rate": 4.93640350877193e-05,
-      "loss": 0.0328,
-      "step": 149
-    },
-    {
-      "epoch": 37.5,
-      "learning_rate": 4.9342105263157894e-05,
-      "loss": 0.0363,
-      "step": 150
-    },
-    {
-      "epoch": 37.75,
-      "learning_rate": 4.932017543859649e-05,
-      "loss": 0.0309,
-      "step": 151
-    },
-    {
-      "epoch": 38.0,
-      "learning_rate": 4.9298245614035086e-05,
-      "loss": 0.034,
-      "step": 152
-    },
-    {
-      "epoch": 38.25,
-      "learning_rate": 4.9276315789473686e-05,
-      "loss": 0.0297,
-      "step": 153
-    },
-    {
-      "epoch": 38.5,
-      "learning_rate": 4.925438596491228e-05,
-      "loss": 0.031,
-      "step": 154
-    },
-    {
-      "epoch": 38.75,
-      "learning_rate": 4.923245614035088e-05,
-      "loss": 0.0308,
-      "step": 155
-    },
-    {
-      "epoch": 39.0,
-      "learning_rate": 4.921052631578947e-05,
-      "loss": 0.0368,
-      "step": 156
-    },
-    {
-      "epoch": 39.25,
-      "learning_rate": 4.918859649122807e-05,
-      "loss": 0.032,
-      "step": 157
-    },
-    {
-      "epoch": 39.5,
-      "learning_rate": 4.9166666666666665e-05,
-      "loss": 0.0336,
-      "step": 158
-    },
-    {
-      "epoch": 39.75,
-      "learning_rate": 4.9144736842105264e-05,
-      "loss": 0.0285,
-      "step": 159
-    },
-    {
-      "epoch": 40.0,
-      "learning_rate": 4.912280701754386e-05,
-      "loss": 0.0341,
-      "step": 160
-    },
-    {
-      "epoch": 40.0,
-      "eval_accuracy_ELSE": 0.9941238217883926,
-      "eval_accuracy_road": 0.9888790795694025,
-      "eval_accuracy_sidewalk": 0.9129822585187272,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9896685088824968,
-      "eval_iou_road": 0.978181869401093,
-      "eval_iou_sidewalk": 0.8118192062100914,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.03096950612962246,
-      "eval_mean_accuracy": 0.9653283866255075,
-      "eval_mean_iou": 0.9265565281645604,
-      "eval_overall_accuracy": 0.9905204772949219,
-      "eval_runtime": 0.6028,
-      "eval_samples_per_second": 1.659,
-      "eval_steps_per_second": 1.659,
-      "step": 160
-    },
-    {
-      "epoch": 40.25,
-      "learning_rate": 4.910087719298246e-05,
-      "loss": 0.0302,
-      "step": 161
-    },
-    {
-      "epoch": 40.5,
-      "learning_rate": 4.907894736842106e-05,
-      "loss": 0.0324,
-      "step": 162
-    },
-    {
-      "epoch": 40.75,
-      "learning_rate": 4.905701754385965e-05,
-      "loss": 0.0301,
-      "step": 163
-    },
-    {
-      "epoch": 41.0,
-      "learning_rate": 4.903508771929825e-05,
-      "loss": 0.0294,
-      "step": 164
-    },
-    {
-      "epoch": 41.25,
-      "learning_rate": 4.901315789473684e-05,
-      "loss": 0.031,
-      "step": 165
-    },
-    {
-      "epoch": 41.5,
-      "learning_rate": 4.899122807017544e-05,
-      "loss": 0.0311,
-      "step": 166
-    },
-    {
-      "epoch": 41.75,
-      "learning_rate": 4.8969298245614035e-05,
-      "loss": 0.0287,
-      "step": 167
-    },
-    {
-      "epoch": 42.0,
-      "learning_rate": 4.8947368421052635e-05,
-      "loss": 0.0281,
-      "step": 168
-    },
-    {
-      "epoch": 42.25,
-      "learning_rate": 4.892543859649123e-05,
-      "loss": 0.0295,
-      "step": 169
-    },
-    {
-      "epoch": 42.5,
-      "learning_rate": 4.890350877192983e-05,
-      "loss": 0.0336,
-      "step": 170
-    },
-    {
-      "epoch": 42.75,
-      "learning_rate": 4.888157894736842e-05,
-      "loss": 0.0288,
-      "step": 171
-    },
-    {
-      "epoch": 43.0,
-      "learning_rate": 4.885964912280702e-05,
-      "loss": 0.0327,
-      "step": 172
-    },
-    {
-      "epoch": 43.25,
-      "learning_rate": 4.8837719298245614e-05,
-      "loss": 0.029,
-      "step": 173
-    },
-    {
-      "epoch": 43.5,
-      "learning_rate": 4.881578947368421e-05,
-      "loss": 0.0321,
-      "step": 174
-    },
-    {
-      "epoch": 43.75,
-      "learning_rate": 4.8793859649122806e-05,
-      "loss": 0.0291,
-      "step": 175
-    },
-    {
-      "epoch": 44.0,
-      "learning_rate": 4.8771929824561406e-05,
-      "loss": 0.0296,
-      "step": 176
-    },
-    {
-      "epoch": 44.25,
-      "learning_rate": 4.875e-05,
-      "loss": 0.0268,
-      "step": 177
-    },
-    {
-      "epoch": 44.5,
-      "learning_rate": 4.87280701754386e-05,
-      "loss": 0.0298,
-      "step": 178
-    },
-    {
-      "epoch": 44.75,
-      "learning_rate": 4.870614035087719e-05,
-      "loss": 0.0286,
-      "step": 179
-    },
-    {
-      "epoch": 45.0,
-      "learning_rate": 4.868421052631579e-05,
-      "loss": 0.0295,
-      "step": 180
-    },
-    {
-      "epoch": 45.0,
-      "eval_accuracy_ELSE": 0.9963422684428669,
-      "eval_accuracy_road": 0.9892920202768126,
-      "eval_accuracy_sidewalk": 0.916784004505773,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9912789758718332,
-      "eval_iou_road": 0.9822428320986258,
-      "eval_iou_sidewalk": 0.8485598853121334,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02526058815419674,
-      "eval_mean_accuracy": 0.9674727644084843,
-      "eval_mean_iou": 0.9406938977608642,
-      "eval_overall_accuracy": 0.9922981262207031,
-      "eval_runtime": 0.6252,
-      "eval_samples_per_second": 1.6,
-      "eval_steps_per_second": 1.6,
-      "step": 180
-    },
-    {
-      "epoch": 45.25,
-      "learning_rate": 4.8662280701754385e-05,
-      "loss": 0.0255,
-      "step": 181
-    },
-    {
-      "epoch": 45.5,
-      "learning_rate": 4.8640350877192984e-05,
-      "loss": 0.0291,
-      "step": 182
-    },
-    {
-      "epoch": 45.75,
-      "learning_rate": 4.861842105263158e-05,
-      "loss": 0.0353,
-      "step": 183
-    },
-    {
-      "epoch": 46.0,
-      "learning_rate": 4.859649122807018e-05,
-      "loss": 0.0296,
-      "step": 184
-    },
-    {
-      "epoch": 46.25,
-      "learning_rate": 4.857456140350877e-05,
-      "loss": 0.0275,
-      "step": 185
-    },
-    {
-      "epoch": 46.5,
-      "learning_rate": 4.855263157894737e-05,
-      "loss": 0.0301,
-      "step": 186
-    },
-    {
-      "epoch": 46.75,
-      "learning_rate": 4.853070175438596e-05,
-      "loss": 0.0315,
-      "step": 187
-    },
-    {
-      "epoch": 47.0,
-      "learning_rate": 4.850877192982456e-05,
-      "loss": 0.0401,
-      "step": 188
-    },
-    {
-      "epoch": 47.25,
-      "learning_rate": 4.8486842105263156e-05,
-      "loss": 0.0274,
-      "step": 189
-    },
-    {
-      "epoch": 47.5,
-      "learning_rate": 4.8464912280701755e-05,
-      "loss": 0.028,
-      "step": 190
-    },
-    {
-      "epoch": 47.75,
-      "learning_rate": 4.844298245614035e-05,
-      "loss": 0.0309,
-      "step": 191
-    },
-    {
-      "epoch": 48.0,
-      "learning_rate": 4.842105263157895e-05,
-      "loss": 0.0291,
-      "step": 192
-    },
-    {
-      "epoch": 48.25,
-      "learning_rate": 4.839912280701754e-05,
-      "loss": 0.0273,
-      "step": 193
-    },
-    {
-      "epoch": 48.5,
-      "learning_rate": 4.837719298245614e-05,
-      "loss": 0.0302,
-      "step": 194
-    },
-    {
-      "epoch": 48.75,
-      "learning_rate": 4.8355263157894734e-05,
-      "loss": 0.0313,
-      "step": 195
-    },
-    {
-      "epoch": 49.0,
-      "learning_rate": 4.8333333333333334e-05,
-      "loss": 0.0268,
-      "step": 196
-    },
-    {
-      "epoch": 49.25,
-      "learning_rate": 4.8311403508771927e-05,
-      "loss": 0.0263,
-      "step": 197
-    },
-    {
-      "epoch": 49.5,
-      "learning_rate": 4.8289473684210526e-05,
-      "loss": 0.0262,
-      "step": 198
-    },
-    {
-      "epoch": 49.75,
-      "learning_rate": 4.826754385964912e-05,
-      "loss": 0.0267,
-      "step": 199
-    },
-    {
-      "epoch": 50.0,
-      "learning_rate": 4.824561403508772e-05,
-      "loss": 0.058,
-      "step": 200
-    },
-    {
-      "epoch": 50.0,
-      "eval_accuracy_ELSE": 0.9955793392275477,
-      "eval_accuracy_road": 0.990217577034801,
-      "eval_accuracy_sidewalk": 0.9076316530554773,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9915928798159056,
-      "eval_iou_road": 0.980873661791049,
-      "eval_iou_sidewalk": 0.8211464968152866,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02616145648062229,
-      "eval_mean_accuracy": 0.9644761897726086,
-      "eval_mean_iou": 0.931204346140747,
-      "eval_overall_accuracy": 0.99176025390625,
-      "eval_runtime": 0.6133,
-      "eval_samples_per_second": 1.63,
-      "eval_steps_per_second": 1.63,
-      "step": 200
-    },
-    {
-      "epoch": 50.25,
-      "learning_rate": 4.822368421052631e-05,
-      "loss": 0.0327,
-      "step": 201
-    },
-    {
-      "epoch": 50.5,
-      "learning_rate": 4.820175438596491e-05,
-      "loss": 0.0261,
-      "step": 202
-    },
-    {
-      "epoch": 50.75,
-      "learning_rate": 4.817982456140351e-05,
-      "loss": 0.026,
-      "step": 203
-    },
-    {
-      "epoch": 51.0,
-      "learning_rate": 4.8157894736842105e-05,
-      "loss": 0.0241,
-      "step": 204
-    },
-    {
-      "epoch": 51.25,
-      "learning_rate": 4.8135964912280704e-05,
-      "loss": 0.0252,
-      "step": 205
-    },
-    {
-      "epoch": 51.5,
-      "learning_rate": 4.81140350877193e-05,
-      "loss": 0.0259,
-      "step": 206
-    },
-    {
-      "epoch": 51.75,
-      "learning_rate": 4.80921052631579e-05,
-      "loss": 0.0293,
-      "step": 207
-    },
-    {
-      "epoch": 52.0,
-      "learning_rate": 4.807017543859649e-05,
-      "loss": 0.0289,
-      "step": 208
-    },
-    {
-      "epoch": 52.25,
-      "learning_rate": 4.804824561403509e-05,
-      "loss": 0.0263,
-      "step": 209
-    },
-    {
-      "epoch": 52.5,
-      "learning_rate": 4.802631578947368e-05,
-      "loss": 0.0233,
-      "step": 210
-    },
-    {
-      "epoch": 52.75,
-      "learning_rate": 4.800438596491228e-05,
-      "loss": 0.0256,
-      "step": 211
-    },
-    {
-      "epoch": 53.0,
-      "learning_rate": 4.7982456140350876e-05,
-      "loss": 0.0285,
-      "step": 212
-    },
-    {
-      "epoch": 53.25,
-      "learning_rate": 4.7960526315789475e-05,
-      "loss": 0.0247,
-      "step": 213
-    },
-    {
-      "epoch": 53.5,
-      "learning_rate": 4.793859649122807e-05,
-      "loss": 0.024,
-      "step": 214
-    },
-    {
-      "epoch": 53.75,
-      "learning_rate": 4.791666666666667e-05,
-      "loss": 0.0237,
-      "step": 215
-    },
-    {
-      "epoch": 54.0,
-      "learning_rate": 4.789473684210526e-05,
-      "loss": 0.0269,
-      "step": 216
-    },
-    {
-      "epoch": 54.25,
-      "learning_rate": 4.787280701754386e-05,
-      "loss": 0.0226,
-      "step": 217
-    },
-    {
-      "epoch": 54.5,
-      "learning_rate": 4.785087719298246e-05,
-      "loss": 0.0279,
-      "step": 218
-    },
-    {
-      "epoch": 54.75,
-      "learning_rate": 4.7828947368421054e-05,
-      "loss": 0.0241,
-      "step": 219
-    },
-    {
-      "epoch": 55.0,
-      "learning_rate": 4.780701754385965e-05,
-      "loss": 0.0242,
-      "step": 220
-    },
-    {
-      "epoch": 55.0,
-      "eval_accuracy_ELSE": 0.9960013851764477,
-      "eval_accuracy_road": 0.9922680412371134,
-      "eval_accuracy_sidewalk": 0.9050971557307801,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.991820812207357,
-      "eval_iou_road": 0.9837928649075995,
-      "eval_iou_sidewalk": 0.8376335678915819,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.024843668565154076,
-      "eval_mean_accuracy": 0.964455527381447,
-      "eval_mean_iou": 0.9377490816688462,
-      "eval_overall_accuracy": 0.9925384521484375,
-      "eval_runtime": 0.6054,
-      "eval_samples_per_second": 1.652,
-      "eval_steps_per_second": 1.652,
-      "step": 220
-    },
-    {
-      "epoch": 55.25,
-      "learning_rate": 4.7785087719298246e-05,
-      "loss": 0.0225,
-      "step": 221
-    },
-    {
-      "epoch": 55.5,
-      "learning_rate": 4.7763157894736846e-05,
-      "loss": 0.0233,
-      "step": 222
-    },
-    {
-      "epoch": 55.75,
-      "learning_rate": 4.774122807017544e-05,
-      "loss": 0.0305,
-      "step": 223
-    },
-    {
-      "epoch": 56.0,
-      "learning_rate": 4.771929824561404e-05,
-      "loss": 0.0237,
-      "step": 224
-    },
-    {
-      "epoch": 56.25,
-      "learning_rate": 4.769736842105263e-05,
-      "loss": 0.0222,
-      "step": 225
-    },
-    {
-      "epoch": 56.5,
-      "learning_rate": 4.767543859649123e-05,
-      "loss": 0.0234,
-      "step": 226
-    },
-    {
-      "epoch": 56.75,
-      "learning_rate": 4.7653508771929825e-05,
-      "loss": 0.0232,
-      "step": 227
-    },
-    {
-      "epoch": 57.0,
-      "learning_rate": 4.7631578947368424e-05,
-      "loss": 0.0227,
-      "step": 228
-    },
-    {
-      "epoch": 57.25,
-      "learning_rate": 4.760964912280702e-05,
-      "loss": 0.0236,
-      "step": 229
-    },
-    {
-      "epoch": 57.5,
-      "learning_rate": 4.758771929824562e-05,
-      "loss": 0.0255,
-      "step": 230
-    },
-    {
-      "epoch": 57.75,
-      "learning_rate": 4.756578947368422e-05,
-      "loss": 0.0216,
-      "step": 231
-    },
-    {
-      "epoch": 58.0,
-      "learning_rate": 4.754385964912281e-05,
-      "loss": 0.0236,
-      "step": 232
-    },
-    {
-      "epoch": 58.25,
-      "learning_rate": 4.752192982456141e-05,
-      "loss": 0.0237,
-      "step": 233
-    },
-    {
-      "epoch": 58.5,
-      "learning_rate": 4.75e-05,
-      "loss": 0.0232,
-      "step": 234
-    },
-    {
-      "epoch": 58.75,
-      "learning_rate": 4.74780701754386e-05,
-      "loss": 0.0212,
-      "step": 235
-    },
-    {
-      "epoch": 59.0,
-      "learning_rate": 4.7456140350877195e-05,
-      "loss": 0.0423,
-      "step": 236
-    },
-    {
-      "epoch": 59.25,
-      "learning_rate": 4.7434210526315795e-05,
-      "loss": 0.0226,
-      "step": 237
-    },
-    {
-      "epoch": 59.5,
-      "learning_rate": 4.741228070175439e-05,
-      "loss": 0.023,
-      "step": 238
-    },
-    {
-      "epoch": 59.75,
-      "learning_rate": 4.739035087719299e-05,
-      "loss": 0.0233,
-      "step": 239
-    },
-    {
-      "epoch": 60.0,
-      "learning_rate": 4.736842105263158e-05,
-      "loss": 0.0217,
-      "step": 240
-    },
-    {
-      "epoch": 60.0,
-      "eval_accuracy_ELSE": 0.9959797417944528,
-      "eval_accuracy_road": 0.9896622429800079,
-      "eval_accuracy_sidewalk": 0.9128414531117995,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9915428165114387,
-      "eval_iou_road": 0.9828744361008585,
-      "eval_iou_sidewalk": 0.8229246001523229,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02502310648560524,
-      "eval_mean_accuracy": 0.9661611459620868,
-      "eval_mean_iou": 0.9324472842548733,
-      "eval_overall_accuracy": 0.992034912109375,
-      "eval_runtime": 0.6032,
-      "eval_samples_per_second": 1.658,
-      "eval_steps_per_second": 1.658,
-      "step": 240
-    },
-    {
-      "epoch": 60.25,
-      "learning_rate": 4.734649122807018e-05,
-      "loss": 0.0313,
-      "step": 241
-    },
-    {
-      "epoch": 60.5,
-      "learning_rate": 4.7324561403508774e-05,
-      "loss": 0.0209,
-      "step": 242
-    },
-    {
-      "epoch": 60.75,
-      "learning_rate": 4.730263157894737e-05,
-      "loss": 0.0228,
-      "step": 243
-    },
-    {
-      "epoch": 61.0,
-      "learning_rate": 4.728070175438597e-05,
-      "loss": 0.022,
-      "step": 244
-    },
-    {
-      "epoch": 61.25,
-      "learning_rate": 4.7258771929824566e-05,
-      "loss": 0.0235,
-      "step": 245
-    },
-    {
-      "epoch": 61.5,
-      "learning_rate": 4.7236842105263166e-05,
-      "loss": 0.0232,
-      "step": 246
-    },
-    {
-      "epoch": 61.75,
-      "learning_rate": 4.721491228070176e-05,
-      "loss": 0.0241,
-      "step": 247
-    },
-    {
-      "epoch": 62.0,
-      "learning_rate": 4.719298245614036e-05,
-      "loss": 0.0213,
-      "step": 248
-    },
-    {
-      "epoch": 62.25,
-      "learning_rate": 4.717105263157895e-05,
-      "loss": 0.022,
-      "step": 249
-    },
-    {
-      "epoch": 62.5,
-      "learning_rate": 4.714912280701755e-05,
-      "loss": 0.0212,
-      "step": 250
-    },
-    {
-      "epoch": 62.75,
-      "learning_rate": 4.7127192982456144e-05,
-      "loss": 0.0221,
-      "step": 251
-    },
-    {
-      "epoch": 63.0,
-      "learning_rate": 4.7105263157894744e-05,
-      "loss": 0.0248,
-      "step": 252
-    },
-    {
-      "epoch": 63.25,
-      "learning_rate": 4.708333333333334e-05,
-      "loss": 0.0244,
-      "step": 253
-    },
-    {
-      "epoch": 63.5,
-      "learning_rate": 4.706140350877194e-05,
-      "loss": 0.0227,
-      "step": 254
-    },
-    {
-      "epoch": 63.75,
-      "learning_rate": 4.703947368421053e-05,
-      "loss": 0.0238,
-      "step": 255
-    },
-    {
-      "epoch": 64.0,
-      "learning_rate": 4.701754385964913e-05,
-      "loss": 0.0223,
-      "step": 256
-    },
-    {
-      "epoch": 64.25,
-      "learning_rate": 4.699561403508772e-05,
-      "loss": 0.0224,
-      "step": 257
-    },
-    {
-      "epoch": 64.5,
-      "learning_rate": 4.697368421052632e-05,
-      "loss": 0.0208,
-      "step": 258
-    },
-    {
-      "epoch": 64.75,
-      "learning_rate": 4.6951754385964915e-05,
-      "loss": 0.0201,
-      "step": 259
-    },
-    {
-      "epoch": 65.0,
-      "learning_rate": 4.6929824561403515e-05,
-      "loss": 0.0223,
-      "step": 260
-    },
-    {
-      "epoch": 65.0,
-      "eval_accuracy_ELSE": 0.9958877574209746,
-      "eval_accuracy_road": 0.9915703138349377,
-      "eval_accuracy_sidewalk": 0.8920022528865108,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9908907863426399,
-      "eval_iou_road": 0.9826294325990941,
-      "eval_iou_sidewalk": 0.8281045751633986,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.024835307151079178,
-      "eval_mean_accuracy": 0.9598201080474743,
-      "eval_mean_iou": 0.9338749313683775,
-      "eval_overall_accuracy": 0.9919166564941406,
-      "eval_runtime": 0.6119,
-      "eval_samples_per_second": 1.634,
-      "eval_steps_per_second": 1.634,
-      "step": 260
-    },
-    {
-      "epoch": 65.25,
-      "learning_rate": 4.690789473684211e-05,
-      "loss": 0.0239,
-      "step": 261
-    },
-    {
-      "epoch": 65.5,
-      "learning_rate": 4.688596491228071e-05,
-      "loss": 0.0228,
-      "step": 262
-    },
-    {
-      "epoch": 65.75,
-      "learning_rate": 4.68640350877193e-05,
-      "loss": 0.0219,
-      "step": 263
-    },
-    {
-      "epoch": 66.0,
-      "learning_rate": 4.68421052631579e-05,
-      "loss": 0.0197,
-      "step": 264
-    },
-    {
-      "epoch": 66.25,
-      "learning_rate": 4.6820175438596494e-05,
-      "loss": 0.0251,
-      "step": 265
-    },
-    {
-      "epoch": 66.5,
-      "learning_rate": 4.679824561403509e-05,
-      "loss": 0.0219,
-      "step": 266
-    },
-    {
-      "epoch": 66.75,
-      "learning_rate": 4.6776315789473686e-05,
-      "loss": 0.0187,
-      "step": 267
-    },
-    {
-      "epoch": 67.0,
-      "learning_rate": 4.6754385964912286e-05,
-      "loss": 0.0204,
-      "step": 268
-    },
-    {
-      "epoch": 67.25,
-      "learning_rate": 4.673245614035088e-05,
-      "loss": 0.0196,
-      "step": 269
-    },
-    {
-      "epoch": 67.5,
-      "learning_rate": 4.671052631578948e-05,
-      "loss": 0.0199,
-      "step": 270
-    },
-    {
-      "epoch": 67.75,
-      "learning_rate": 4.668859649122807e-05,
-      "loss": 0.0219,
-      "step": 271
-    },
-    {
-      "epoch": 68.0,
-      "learning_rate": 4.666666666666667e-05,
-      "loss": 0.0209,
-      "step": 272
-    },
-    {
-      "epoch": 68.25,
-      "learning_rate": 4.6644736842105265e-05,
-      "loss": 0.0223,
-      "step": 273
-    },
-    {
-      "epoch": 68.5,
-      "learning_rate": 4.6622807017543864e-05,
-      "loss": 0.0192,
-      "step": 274
-    },
-    {
-      "epoch": 68.75,
-      "learning_rate": 4.660087719298246e-05,
-      "loss": 0.0193,
-      "step": 275
-    },
-    {
-      "epoch": 69.0,
-      "learning_rate": 4.657894736842106e-05,
-      "loss": 0.0207,
-      "step": 276
-    },
-    {
-      "epoch": 69.25,
-      "learning_rate": 4.655701754385965e-05,
-      "loss": 0.0211,
-      "step": 277
-    },
-    {
-      "epoch": 69.5,
-      "learning_rate": 4.653508771929825e-05,
-      "loss": 0.0209,
-      "step": 278
-    },
-    {
-      "epoch": 69.75,
-      "learning_rate": 4.651315789473684e-05,
-      "loss": 0.0211,
-      "step": 279
-    },
-    {
-      "epoch": 70.0,
-      "learning_rate": 4.649122807017544e-05,
-      "loss": 0.0206,
-      "step": 280
-    },
-    {
-      "epoch": 70.0,
-      "eval_accuracy_ELSE": 0.9964775395803348,
-      "eval_accuracy_road": 0.9925813066013556,
-      "eval_accuracy_sidewalk": 0.9038299070684315,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9921025270835915,
-      "eval_iou_road": 0.9847290501214895,
-      "eval_iou_sidewalk": 0.8466103930361383,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02149595506489277,
-      "eval_mean_accuracy": 0.964296251083374,
-      "eval_mean_iou": 0.9411473234137397,
-      "eval_overall_accuracy": 0.9929237365722656,
-      "eval_runtime": 0.6222,
-      "eval_samples_per_second": 1.607,
-      "eval_steps_per_second": 1.607,
-      "step": 280
-    },
-    {
-      "epoch": 70.25,
-      "learning_rate": 4.6469298245614036e-05,
-      "loss": 0.0207,
-      "step": 281
-    },
-    {
-      "epoch": 70.5,
-      "learning_rate": 4.6447368421052635e-05,
-      "loss": 0.0191,
-      "step": 282
-    },
-    {
-      "epoch": 70.75,
-      "learning_rate": 4.6425438596491235e-05,
-      "loss": 0.0212,
-      "step": 283
-    },
-    {
-      "epoch": 71.0,
-      "learning_rate": 4.640350877192983e-05,
-      "loss": 0.0197,
-      "step": 284
-    },
-    {
-      "epoch": 71.25,
-      "learning_rate": 4.638157894736843e-05,
-      "loss": 0.0189,
-      "step": 285
-    },
-    {
-      "epoch": 71.5,
-      "learning_rate": 4.635964912280702e-05,
-      "loss": 0.0192,
-      "step": 286
-    },
-    {
-      "epoch": 71.75,
-      "learning_rate": 4.633771929824562e-05,
-      "loss": 0.0176,
-      "step": 287
-    },
-    {
-      "epoch": 72.0,
-      "learning_rate": 4.6315789473684214e-05,
-      "loss": 0.0207,
-      "step": 288
-    },
-    {
-      "epoch": 72.25,
-      "learning_rate": 4.629385964912281e-05,
-      "loss": 0.0209,
-      "step": 289
-    },
-    {
-      "epoch": 72.5,
-      "learning_rate": 4.6271929824561406e-05,
-      "loss": 0.0198,
-      "step": 290
-    },
-    {
-      "epoch": 72.75,
-      "learning_rate": 4.6250000000000006e-05,
-      "loss": 0.0196,
-      "step": 291
-    },
-    {
-      "epoch": 73.0,
-      "learning_rate": 4.62280701754386e-05,
-      "loss": 0.0186,
-      "step": 292
-    },
-    {
-      "epoch": 73.25,
-      "learning_rate": 4.62061403508772e-05,
-      "loss": 0.0191,
-      "step": 293
-    },
-    {
-      "epoch": 73.5,
-      "learning_rate": 4.618421052631579e-05,
-      "loss": 0.0196,
-      "step": 294
-    },
-    {
-      "epoch": 73.75,
-      "learning_rate": 4.616228070175439e-05,
-      "loss": 0.0193,
-      "step": 295
-    },
-    {
-      "epoch": 74.0,
-      "learning_rate": 4.6140350877192985e-05,
-      "loss": 0.0176,
-      "step": 296
-    },
-    {
-      "epoch": 74.25,
-      "learning_rate": 4.6118421052631584e-05,
-      "loss": 0.0196,
-      "step": 297
-    },
-    {
-      "epoch": 74.5,
-      "learning_rate": 4.609649122807018e-05,
-      "loss": 0.0184,
-      "step": 298
-    },
-    {
-      "epoch": 74.75,
-      "learning_rate": 4.607456140350878e-05,
-      "loss": 0.0193,
-      "step": 299
-    },
-    {
-      "epoch": 75.0,
-      "learning_rate": 4.605263157894737e-05,
-      "loss": 0.0194,
-      "step": 300
-    },
-    {
-      "epoch": 75.0,
-      "eval_accuracy_ELSE": 0.9953520837166016,
-      "eval_accuracy_road": 0.9927236999487384,
-      "eval_accuracy_sidewalk": 0.9125598422979443,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9915268397591726,
-      "eval_iou_road": 0.9848564042435972,
-      "eval_iou_sidewalk": 0.828878373193503,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022591305896639824,
-      "eval_mean_accuracy": 0.9668785419877614,
-      "eval_mean_iou": 0.9350872057320908,
-      "eval_overall_accuracy": 0.9924049377441406,
-      "eval_runtime": 0.6107,
-      "eval_samples_per_second": 1.637,
-      "eval_steps_per_second": 1.637,
-      "step": 300
-    },
-    {
-      "epoch": 75.25,
-      "learning_rate": 4.603070175438597e-05,
-      "loss": 0.0205,
-      "step": 301
-    },
-    {
-      "epoch": 75.5,
-      "learning_rate": 4.600877192982456e-05,
-      "loss": 0.0206,
-      "step": 302
-    },
-    {
-      "epoch": 75.75,
-      "learning_rate": 4.598684210526316e-05,
-      "loss": 0.0197,
-      "step": 303
-    },
-    {
-      "epoch": 76.0,
-      "learning_rate": 4.5964912280701756e-05,
-      "loss": 0.0208,
-      "step": 304
-    },
-    {
-      "epoch": 76.25,
-      "learning_rate": 4.5942982456140355e-05,
-      "loss": 0.019,
-      "step": 305
-    },
-    {
-      "epoch": 76.5,
-      "learning_rate": 4.592105263157895e-05,
-      "loss": 0.0198,
-      "step": 306
-    },
-    {
-      "epoch": 76.75,
-      "learning_rate": 4.589912280701755e-05,
-      "loss": 0.0196,
-      "step": 307
-    },
-    {
-      "epoch": 77.0,
-      "learning_rate": 4.587719298245614e-05,
-      "loss": 0.0171,
-      "step": 308
-    },
-    {
-      "epoch": 77.25,
-      "learning_rate": 4.585526315789474e-05,
-      "loss": 0.0189,
-      "step": 309
-    },
-    {
-      "epoch": 77.5,
-      "learning_rate": 4.5833333333333334e-05,
-      "loss": 0.0186,
-      "step": 310
-    },
-    {
-      "epoch": 77.75,
-      "learning_rate": 4.5811403508771934e-05,
-      "loss": 0.0184,
-      "step": 311
-    },
-    {
-      "epoch": 78.0,
-      "learning_rate": 4.5789473684210527e-05,
-      "loss": 0.0154,
-      "step": 312
-    },
-    {
-      "epoch": 78.25,
-      "learning_rate": 4.5767543859649126e-05,
-      "loss": 0.0177,
-      "step": 313
-    },
-    {
-      "epoch": 78.5,
-      "learning_rate": 4.574561403508772e-05,
-      "loss": 0.0172,
-      "step": 314
-    },
-    {
-      "epoch": 78.75,
-      "learning_rate": 4.572368421052632e-05,
-      "loss": 0.0201,
-      "step": 315
-    },
-    {
-      "epoch": 79.0,
-      "learning_rate": 4.570175438596491e-05,
-      "loss": 0.0175,
-      "step": 316
-    },
-    {
-      "epoch": 79.25,
-      "learning_rate": 4.567982456140351e-05,
-      "loss": 0.0167,
-      "step": 317
-    },
-    {
-      "epoch": 79.5,
-      "learning_rate": 4.5657894736842105e-05,
-      "loss": 0.0182,
-      "step": 318
-    },
-    {
-      "epoch": 79.75,
-      "learning_rate": 4.5635964912280705e-05,
-      "loss": 0.0164,
-      "step": 319
-    },
-    {
-      "epoch": 80.0,
-      "learning_rate": 4.56140350877193e-05,
-      "loss": 0.02,
-      "step": 320
-    },
-    {
-      "epoch": 80.0,
-      "eval_accuracy_ELSE": 0.9966560974817925,
-      "eval_accuracy_road": 0.9917981431907501,
-      "eval_accuracy_sidewalk": 0.8993241340467474,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9916873048347152,
-      "eval_iou_road": 0.9854137487090248,
-      "eval_iou_sidewalk": 0.8370904325032765,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021616503596305847,
-      "eval_mean_accuracy": 0.9625927915730967,
-      "eval_mean_iou": 0.9380638286823388,
-      "eval_overall_accuracy": 0.9927177429199219,
-      "eval_runtime": 0.6134,
-      "eval_samples_per_second": 1.63,
-      "eval_steps_per_second": 1.63,
-      "step": 320
-    },
-    {
-      "epoch": 80.25,
-      "learning_rate": 4.55921052631579e-05,
-      "loss": 0.0188,
-      "step": 321
-    },
-    {
-      "epoch": 80.5,
-      "learning_rate": 4.557017543859649e-05,
-      "loss": 0.0165,
-      "step": 322
-    },
-    {
-      "epoch": 80.75,
-      "learning_rate": 4.554824561403509e-05,
-      "loss": 0.0179,
-      "step": 323
-    },
-    {
-      "epoch": 81.0,
-      "learning_rate": 4.552631578947369e-05,
-      "loss": 0.018,
-      "step": 324
-    },
-    {
-      "epoch": 81.25,
-      "learning_rate": 4.550438596491228e-05,
-      "loss": 0.0163,
-      "step": 325
-    },
-    {
-      "epoch": 81.5,
-      "learning_rate": 4.548245614035088e-05,
-      "loss": 0.0206,
-      "step": 326
-    },
-    {
-      "epoch": 81.75,
-      "learning_rate": 4.5460526315789476e-05,
-      "loss": 0.0161,
-      "step": 327
-    },
-    {
-      "epoch": 82.0,
-      "learning_rate": 4.5438596491228075e-05,
-      "loss": 0.0194,
-      "step": 328
-    },
-    {
-      "epoch": 82.25,
-      "learning_rate": 4.541666666666667e-05,
-      "loss": 0.0181,
-      "step": 329
-    },
-    {
-      "epoch": 82.5,
-      "learning_rate": 4.539473684210527e-05,
-      "loss": 0.0157,
-      "step": 330
-    },
-    {
-      "epoch": 82.75,
-      "learning_rate": 4.537280701754386e-05,
-      "loss": 0.02,
-      "step": 331
-    },
-    {
-      "epoch": 83.0,
-      "learning_rate": 4.535087719298246e-05,
-      "loss": 0.02,
-      "step": 332
-    },
-    {
-      "epoch": 83.25,
-      "learning_rate": 4.5328947368421054e-05,
-      "loss": 0.0192,
-      "step": 333
-    },
-    {
-      "epoch": 83.5,
-      "learning_rate": 4.5307017543859654e-05,
-      "loss": 0.0178,
-      "step": 334
-    },
-    {
-      "epoch": 83.75,
-      "learning_rate": 4.5285087719298247e-05,
-      "loss": 0.0177,
-      "step": 335
-    },
-    {
-      "epoch": 84.0,
-      "learning_rate": 4.5263157894736846e-05,
-      "loss": 0.0169,
-      "step": 336
-    },
-    {
-      "epoch": 84.25,
-      "learning_rate": 4.524122807017544e-05,
-      "loss": 0.0183,
-      "step": 337
-    },
-    {
-      "epoch": 84.5,
-      "learning_rate": 4.521929824561404e-05,
-      "loss": 0.0187,
-      "step": 338
-    },
-    {
-      "epoch": 84.75,
-      "learning_rate": 4.519736842105263e-05,
-      "loss": 0.0157,
-      "step": 339
-    },
-    {
-      "epoch": 85.0,
-      "learning_rate": 4.517543859649123e-05,
-      "loss": 0.0181,
-      "step": 340
-    },
-    {
-      "epoch": 85.0,
-      "eval_accuracy_ELSE": 0.9960879587044271,
-      "eval_accuracy_road": 0.9906874750811642,
-      "eval_accuracy_sidewalk": 0.9091805125316812,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9915063446582071,
-      "eval_iou_road": 0.9846166909610676,
-      "eval_iou_sidewalk": 0.8239122113053464,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022249778732657433,
-      "eval_mean_accuracy": 0.9653186487724241,
-      "eval_mean_iou": 0.933345082308207,
-      "eval_overall_accuracy": 0.9922866821289062,
-      "eval_runtime": 0.6261,
-      "eval_samples_per_second": 1.597,
-      "eval_steps_per_second": 1.597,
-      "step": 340
-    },
-    {
-      "epoch": 85.25,
-      "learning_rate": 4.5153508771929825e-05,
-      "loss": 0.0168,
-      "step": 341
-    },
-    {
-      "epoch": 85.5,
-      "learning_rate": 4.5131578947368425e-05,
-      "loss": 0.019,
-      "step": 342
-    },
-    {
-      "epoch": 85.75,
-      "learning_rate": 4.510964912280702e-05,
-      "loss": 0.0154,
-      "step": 343
-    },
-    {
-      "epoch": 86.0,
-      "learning_rate": 4.508771929824562e-05,
-      "loss": 0.0172,
-      "step": 344
-    },
-    {
-      "epoch": 86.25,
-      "learning_rate": 4.506578947368421e-05,
-      "loss": 0.0174,
-      "step": 345
-    },
-    {
-      "epoch": 86.5,
-      "learning_rate": 4.504385964912281e-05,
-      "loss": 0.0172,
-      "step": 346
-    },
-    {
-      "epoch": 86.75,
-      "learning_rate": 4.50219298245614e-05,
-      "loss": 0.0158,
-      "step": 347
-    },
-    {
-      "epoch": 87.0,
-      "learning_rate": 4.5e-05,
-      "loss": 0.0185,
-      "step": 348
-    },
-    {
-      "epoch": 87.25,
-      "learning_rate": 4.4978070175438596e-05,
-      "loss": 0.0167,
-      "step": 349
-    },
-    {
-      "epoch": 87.5,
-      "learning_rate": 4.4956140350877196e-05,
-      "loss": 0.0164,
-      "step": 350
-    },
-    {
-      "epoch": 87.75,
-      "learning_rate": 4.493421052631579e-05,
-      "loss": 0.0172,
-      "step": 351
-    },
-    {
-      "epoch": 88.0,
-      "learning_rate": 4.491228070175439e-05,
-      "loss": 0.0177,
-      "step": 352
-    },
-    {
-      "epoch": 88.25,
-      "learning_rate": 4.489035087719298e-05,
-      "loss": 0.0163,
-      "step": 353
-    },
-    {
-      "epoch": 88.5,
-      "learning_rate": 4.486842105263158e-05,
-      "loss": 0.0171,
-      "step": 354
-    },
-    {
-      "epoch": 88.75,
-      "learning_rate": 4.4846491228070174e-05,
-      "loss": 0.0163,
-      "step": 355
-    },
-    {
-      "epoch": 89.0,
-      "learning_rate": 4.4824561403508774e-05,
-      "loss": 0.0174,
-      "step": 356
-    },
-    {
-      "epoch": 89.25,
-      "learning_rate": 4.480263157894737e-05,
-      "loss": 0.017,
-      "step": 357
-    },
-    {
-      "epoch": 89.5,
-      "learning_rate": 4.4780701754385967e-05,
-      "loss": 0.0153,
-      "step": 358
-    },
-    {
-      "epoch": 89.75,
-      "learning_rate": 4.475877192982456e-05,
-      "loss": 0.0155,
-      "step": 359
-    },
-    {
-      "epoch": 90.0,
-      "learning_rate": 4.473684210526316e-05,
-      "loss": 0.018,
-      "step": 360
-    },
-    {
-      "epoch": 90.0,
-      "eval_accuracy_ELSE": 0.9963801443613579,
-      "eval_accuracy_road": 0.9914136811528166,
-      "eval_accuracy_sidewalk": 0.8993241340467474,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9916369589333218,
-      "eval_iou_road": 0.9848366974553375,
-      "eval_iou_sidewalk": 0.8256204756980352,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022521808743476868,
-      "eval_mean_accuracy": 0.962372653186974,
-      "eval_mean_iou": 0.9340313773622316,
-      "eval_overall_accuracy": 0.9924201965332031,
-      "eval_runtime": 0.6094,
-      "eval_samples_per_second": 1.641,
-      "eval_steps_per_second": 1.641,
-      "step": 360
-    },
-    {
-      "epoch": 90.25,
-      "learning_rate": 4.471491228070175e-05,
-      "loss": 0.0152,
-      "step": 361
-    },
-    {
-      "epoch": 90.5,
-      "learning_rate": 4.469298245614035e-05,
-      "loss": 0.0175,
-      "step": 362
-    },
-    {
-      "epoch": 90.75,
-      "learning_rate": 4.4671052631578945e-05,
-      "loss": 0.0161,
-      "step": 363
-    },
-    {
-      "epoch": 91.0,
-      "learning_rate": 4.4649122807017545e-05,
-      "loss": 0.0171,
-      "step": 364
-    },
-    {
-      "epoch": 91.25,
-      "learning_rate": 4.4627192982456145e-05,
-      "loss": 0.017,
-      "step": 365
-    },
-    {
-      "epoch": 91.5,
-      "learning_rate": 4.460526315789474e-05,
-      "loss": 0.0149,
-      "step": 366
-    },
-    {
-      "epoch": 91.75,
-      "learning_rate": 4.458333333333334e-05,
-      "loss": 0.0158,
-      "step": 367
-    },
-    {
-      "epoch": 92.0,
-      "learning_rate": 4.456140350877193e-05,
-      "loss": 0.0176,
-      "step": 368
-    },
-    {
-      "epoch": 92.25,
-      "learning_rate": 4.453947368421053e-05,
-      "loss": 0.0168,
-      "step": 369
-    },
-    {
-      "epoch": 92.5,
-      "learning_rate": 4.451754385964912e-05,
-      "loss": 0.0159,
-      "step": 370
-    },
-    {
-      "epoch": 92.75,
-      "learning_rate": 4.449561403508772e-05,
-      "loss": 0.018,
-      "step": 371
-    },
-    {
-      "epoch": 93.0,
-      "learning_rate": 4.4473684210526316e-05,
-      "loss": 0.0157,
-      "step": 372
-    },
-    {
-      "epoch": 93.25,
-      "learning_rate": 4.4451754385964916e-05,
-      "loss": 0.0154,
-      "step": 373
-    },
-    {
-      "epoch": 93.5,
-      "learning_rate": 4.442982456140351e-05,
-      "loss": 0.0161,
-      "step": 374
-    },
-    {
-      "epoch": 93.75,
-      "learning_rate": 4.440789473684211e-05,
-      "loss": 0.0163,
-      "step": 375
-    },
-    {
-      "epoch": 94.0,
-      "learning_rate": 4.43859649122807e-05,
-      "loss": 0.018,
-      "step": 376
-    },
-    {
-      "epoch": 94.25,
-      "learning_rate": 4.43640350877193e-05,
-      "loss": 0.0165,
-      "step": 377
-    },
-    {
-      "epoch": 94.5,
-      "learning_rate": 4.4342105263157894e-05,
-      "loss": 0.0157,
-      "step": 378
-    },
-    {
-      "epoch": 94.75,
-      "learning_rate": 4.4320175438596494e-05,
-      "loss": 0.0159,
-      "step": 379
-    },
-    {
-      "epoch": 95.0,
-      "learning_rate": 4.429824561403509e-05,
-      "loss": 0.0169,
-      "step": 380
-    },
-    {
-      "epoch": 95.0,
-      "eval_accuracy_ELSE": 0.9958120055839925,
-      "eval_accuracy_road": 0.9931366406561486,
-      "eval_accuracy_sidewalk": 0.9098845395663193,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9922096126371405,
-      "eval_iou_road": 0.9854609678558813,
-      "eval_iou_sidewalk": 0.8305912596401028,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020703531801700592,
-      "eval_mean_accuracy": 0.9662777286021536,
-      "eval_mean_iou": 0.9360872800443749,
-      "eval_overall_accuracy": 0.992767333984375,
-      "eval_runtime": 0.6103,
-      "eval_samples_per_second": 1.638,
-      "eval_steps_per_second": 1.638,
-      "step": 380
-    },
-    {
-      "epoch": 95.25,
-      "learning_rate": 4.4276315789473687e-05,
-      "loss": 0.0169,
-      "step": 381
-    },
-    {
-      "epoch": 95.5,
-      "learning_rate": 4.425438596491228e-05,
-      "loss": 0.0148,
-      "step": 382
-    },
-    {
-      "epoch": 95.75,
-      "learning_rate": 4.423245614035088e-05,
-      "loss": 0.0165,
-      "step": 383
-    },
-    {
-      "epoch": 96.0,
-      "learning_rate": 4.421052631578947e-05,
-      "loss": 0.018,
-      "step": 384
-    },
-    {
-      "epoch": 96.25,
-      "learning_rate": 4.418859649122807e-05,
-      "loss": 0.0165,
-      "step": 385
-    },
-    {
-      "epoch": 96.5,
-      "learning_rate": 4.4166666666666665e-05,
-      "loss": 0.0156,
-      "step": 386
-    },
-    {
-      "epoch": 96.75,
-      "learning_rate": 4.4144736842105265e-05,
-      "loss": 0.0147,
-      "step": 387
-    },
-    {
-      "epoch": 97.0,
-      "learning_rate": 4.412280701754386e-05,
-      "loss": 0.0176,
-      "step": 388
-    },
-    {
-      "epoch": 97.25,
-      "learning_rate": 4.410087719298246e-05,
-      "loss": 0.0153,
-      "step": 389
-    },
-    {
-      "epoch": 97.5,
-      "learning_rate": 4.407894736842105e-05,
-      "loss": 0.0159,
-      "step": 390
-    },
-    {
-      "epoch": 97.75,
-      "learning_rate": 4.405701754385965e-05,
-      "loss": 0.0159,
-      "step": 391
-    },
-    {
-      "epoch": 98.0,
-      "learning_rate": 4.403508771929824e-05,
-      "loss": 0.0154,
-      "step": 392
-    },
-    {
-      "epoch": 98.25,
-      "learning_rate": 4.401315789473684e-05,
-      "loss": 0.0141,
-      "step": 393
-    },
-    {
-      "epoch": 98.5,
-      "learning_rate": 4.3991228070175436e-05,
-      "loss": 0.0151,
-      "step": 394
-    },
-    {
-      "epoch": 98.75,
-      "learning_rate": 4.3969298245614036e-05,
-      "loss": 0.0167,
-      "step": 395
-    },
-    {
-      "epoch": 99.0,
-      "learning_rate": 4.394736842105263e-05,
-      "loss": 0.0178,
-      "step": 396
-    },
-    {
-      "epoch": 99.25,
-      "learning_rate": 4.392543859649123e-05,
-      "loss": 0.0161,
-      "step": 397
-    },
-    {
-      "epoch": 99.5,
-      "learning_rate": 4.390350877192982e-05,
-      "loss": 0.015,
-      "step": 398
-    },
-    {
-      "epoch": 99.75,
-      "learning_rate": 4.388157894736842e-05,
-      "loss": 0.0146,
-      "step": 399
-    },
-    {
-      "epoch": 100.0,
-      "learning_rate": 4.3859649122807014e-05,
-      "loss": 0.0157,
-      "step": 400
-    },
-    {
-      "epoch": 100.0,
-      "eval_accuracy_ELSE": 0.995974330948954,
-      "eval_accuracy_road": 0.99229651990659,
-      "eval_accuracy_sidewalk": 0.9091805125316812,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9920878741821082,
-      "eval_iou_road": 0.9857555096613574,
-      "eval_iou_sidewalk": 0.8233868910992094,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020867783576250076,
-      "eval_mean_accuracy": 0.965817121129075,
-      "eval_mean_iou": 0.9337434249808917,
-      "eval_overall_accuracy": 0.9926376342773438,
-      "eval_runtime": 0.6759,
-      "eval_samples_per_second": 1.479,
-      "eval_steps_per_second": 1.479,
-      "step": 400
-    },
-    {
-      "epoch": 100.25,
-      "learning_rate": 4.3837719298245614e-05,
-      "loss": 0.0166,
-      "step": 401
-    },
-    {
-      "epoch": 100.5,
-      "learning_rate": 4.381578947368421e-05,
-      "loss": 0.0153,
-      "step": 402
-    },
-    {
-      "epoch": 100.75,
-      "learning_rate": 4.379385964912281e-05,
-      "loss": 0.0143,
-      "step": 403
-    },
-    {
-      "epoch": 101.0,
-      "learning_rate": 4.37719298245614e-05,
-      "loss": 0.0163,
-      "step": 404
-    },
-    {
-      "epoch": 101.25,
-      "learning_rate": 4.375e-05,
-      "loss": 0.0148,
-      "step": 405
-    },
-    {
-      "epoch": 101.5,
-      "learning_rate": 4.37280701754386e-05,
-      "loss": 0.0151,
-      "step": 406
-    },
-    {
-      "epoch": 101.75,
-      "learning_rate": 4.370614035087719e-05,
-      "loss": 0.0147,
-      "step": 407
-    },
-    {
-      "epoch": 102.0,
-      "learning_rate": 4.368421052631579e-05,
-      "loss": 0.0173,
-      "step": 408
-    },
-    {
-      "epoch": 102.25,
-      "learning_rate": 4.3662280701754385e-05,
-      "loss": 0.0156,
-      "step": 409
-    },
-    {
-      "epoch": 102.5,
-      "learning_rate": 4.3640350877192985e-05,
-      "loss": 0.0143,
-      "step": 410
-    },
-    {
-      "epoch": 102.75,
-      "learning_rate": 4.361842105263158e-05,
-      "loss": 0.0167,
-      "step": 411
-    },
-    {
-      "epoch": 103.0,
-      "learning_rate": 4.359649122807018e-05,
-      "loss": 0.0158,
-      "step": 412
-    },
-    {
-      "epoch": 103.25,
-      "learning_rate": 4.357456140350877e-05,
-      "loss": 0.014,
-      "step": 413
-    },
-    {
-      "epoch": 103.5,
-      "learning_rate": 4.355263157894737e-05,
-      "loss": 0.0139,
-      "step": 414
-    },
-    {
-      "epoch": 103.75,
-      "learning_rate": 4.353070175438596e-05,
-      "loss": 0.0141,
-      "step": 415
-    },
-    {
-      "epoch": 104.0,
-      "learning_rate": 4.350877192982456e-05,
-      "loss": 0.0174,
-      "step": 416
-    },
-    {
-      "epoch": 104.25,
-      "learning_rate": 4.3486842105263156e-05,
-      "loss": 0.0158,
-      "step": 417
-    },
-    {
-      "epoch": 104.5,
-      "learning_rate": 4.3464912280701756e-05,
-      "loss": 0.0156,
-      "step": 418
-    },
-    {
-      "epoch": 104.75,
-      "learning_rate": 4.344298245614035e-05,
-      "loss": 0.0162,
-      "step": 419
-    },
-    {
-      "epoch": 105.0,
-      "learning_rate": 4.342105263157895e-05,
-      "loss": 0.0147,
-      "step": 420
-    },
-    {
-      "epoch": 105.0,
-      "eval_accuracy_ELSE": 0.9954440680900798,
-      "eval_accuracy_road": 0.9913994418180783,
-      "eval_accuracy_sidewalk": 0.9119966206702337,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9918483534967976,
-      "eval_iou_road": 0.9835704295986551,
-      "eval_iou_sidewalk": 0.8153323262839879,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022910170257091522,
-      "eval_mean_accuracy": 0.9662800435261305,
-      "eval_mean_iou": 0.9302503697931469,
-      "eval_overall_accuracy": 0.9920997619628906,
-      "eval_runtime": 0.6023,
-      "eval_samples_per_second": 1.66,
-      "eval_steps_per_second": 1.66,
-      "step": 420
-    },
-    {
-      "epoch": 105.25,
-      "learning_rate": 4.339912280701754e-05,
-      "loss": 0.015,
-      "step": 421
-    },
-    {
-      "epoch": 105.5,
-      "learning_rate": 4.337719298245614e-05,
-      "loss": 0.0146,
-      "step": 422
-    },
-    {
-      "epoch": 105.75,
-      "learning_rate": 4.335526315789474e-05,
-      "loss": 0.0162,
-      "step": 423
-    },
-    {
-      "epoch": 106.0,
-      "learning_rate": 4.3333333333333334e-05,
-      "loss": 0.0146,
-      "step": 424
-    },
-    {
-      "epoch": 106.25,
-      "learning_rate": 4.3311403508771934e-05,
-      "loss": 0.0141,
-      "step": 425
-    },
-    {
-      "epoch": 106.5,
-      "learning_rate": 4.328947368421053e-05,
-      "loss": 0.0158,
-      "step": 426
-    },
-    {
-      "epoch": 106.75,
-      "learning_rate": 4.3267543859649127e-05,
-      "loss": 0.0166,
-      "step": 427
-    },
-    {
-      "epoch": 107.0,
-      "learning_rate": 4.324561403508772e-05,
-      "loss": 0.0145,
-      "step": 428
-    },
-    {
-      "epoch": 107.25,
-      "learning_rate": 4.322368421052632e-05,
-      "loss": 0.0151,
-      "step": 429
-    },
-    {
-      "epoch": 107.5,
-      "learning_rate": 4.320175438596491e-05,
-      "loss": 0.0148,
-      "step": 430
-    },
-    {
-      "epoch": 107.75,
-      "learning_rate": 4.317982456140351e-05,
-      "loss": 0.0137,
-      "step": 431
-    },
-    {
-      "epoch": 108.0,
-      "learning_rate": 4.3157894736842105e-05,
-      "loss": 0.0154,
-      "step": 432
-    },
-    {
-      "epoch": 108.25,
-      "learning_rate": 4.3135964912280705e-05,
-      "loss": 0.0143,
-      "step": 433
-    },
-    {
-      "epoch": 108.5,
-      "learning_rate": 4.31140350877193e-05,
-      "loss": 0.0142,
-      "step": 434
-    },
-    {
-      "epoch": 108.75,
-      "learning_rate": 4.30921052631579e-05,
-      "loss": 0.0152,
-      "step": 435
-    },
-    {
-      "epoch": 109.0,
-      "learning_rate": 4.30701754385965e-05,
-      "loss": 0.017,
-      "step": 436
-    },
-    {
-      "epoch": 109.25,
-      "learning_rate": 4.304824561403509e-05,
-      "loss": 0.017,
-      "step": 437
-    },
-    {
-      "epoch": 109.5,
-      "learning_rate": 4.302631578947369e-05,
-      "loss": 0.0139,
-      "step": 438
-    },
-    {
-      "epoch": 109.75,
-      "learning_rate": 4.300438596491228e-05,
-      "loss": 0.0152,
-      "step": 439
-    },
-    {
-      "epoch": 110.0,
-      "learning_rate": 4.298245614035088e-05,
-      "loss": 0.014,
-      "step": 440
-    },
-    {
-      "epoch": 110.0,
-      "eval_accuracy_ELSE": 0.9961961756144015,
-      "eval_accuracy_road": 0.9930227259782423,
-      "eval_accuracy_sidewalk": 0.9081948746831878,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9925602859437924,
-      "eval_iou_road": 0.9856543185447967,
-      "eval_iou_sidewalk": 0.8327953518398967,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02052408643066883,
-      "eval_mean_accuracy": 0.9658045920919438,
-      "eval_mean_iou": 0.9370033187761618,
-      "eval_overall_accuracy": 0.9929618835449219,
-      "eval_runtime": 0.6167,
-      "eval_samples_per_second": 1.622,
-      "eval_steps_per_second": 1.622,
-      "step": 440
-    },
-    {
-      "epoch": 110.25,
-      "learning_rate": 4.2960526315789476e-05,
-      "loss": 0.0149,
-      "step": 441
-    },
-    {
-      "epoch": 110.5,
-      "learning_rate": 4.2938596491228076e-05,
-      "loss": 0.0133,
-      "step": 442
-    },
-    {
-      "epoch": 110.75,
-      "learning_rate": 4.291666666666667e-05,
-      "loss": 0.0157,
-      "step": 443
-    },
-    {
-      "epoch": 111.0,
-      "learning_rate": 4.289473684210527e-05,
-      "loss": 0.0138,
-      "step": 444
-    },
-    {
-      "epoch": 111.25,
-      "learning_rate": 4.287280701754386e-05,
-      "loss": 0.0134,
-      "step": 445
-    },
-    {
-      "epoch": 111.5,
-      "learning_rate": 4.285087719298246e-05,
-      "loss": 0.0144,
-      "step": 446
-    },
-    {
-      "epoch": 111.75,
-      "learning_rate": 4.2828947368421054e-05,
-      "loss": 0.0153,
-      "step": 447
-    },
-    {
-      "epoch": 112.0,
-      "learning_rate": 4.2807017543859654e-05,
-      "loss": 0.0143,
-      "step": 448
-    },
-    {
-      "epoch": 112.25,
-      "learning_rate": 4.2785087719298254e-05,
-      "loss": 0.0158,
-      "step": 449
-    },
-    {
-      "epoch": 112.5,
-      "learning_rate": 4.2763157894736847e-05,
-      "loss": 0.0139,
-      "step": 450
-    },
-    {
-      "epoch": 112.75,
-      "learning_rate": 4.2741228070175446e-05,
-      "loss": 0.0144,
-      "step": 451
-    },
-    {
-      "epoch": 113.0,
-      "learning_rate": 4.271929824561404e-05,
-      "loss": 0.0133,
-      "step": 452
-    },
-    {
-      "epoch": 113.25,
-      "learning_rate": 4.269736842105264e-05,
-      "loss": 0.0139,
-      "step": 453
-    },
-    {
-      "epoch": 113.5,
-      "learning_rate": 4.267543859649123e-05,
-      "loss": 0.0135,
-      "step": 454
-    },
-    {
-      "epoch": 113.75,
-      "learning_rate": 4.265350877192983e-05,
-      "loss": 0.0148,
-      "step": 455
-    },
-    {
-      "epoch": 114.0,
-      "learning_rate": 4.2631578947368425e-05,
-      "loss": 0.0143,
-      "step": 456
-    },
-    {
-      "epoch": 114.25,
-      "learning_rate": 4.2609649122807025e-05,
-      "loss": 0.0157,
-      "step": 457
-    },
-    {
-      "epoch": 114.5,
-      "learning_rate": 4.258771929824562e-05,
-      "loss": 0.0139,
-      "step": 458
-    },
-    {
-      "epoch": 114.75,
-      "learning_rate": 4.256578947368422e-05,
-      "loss": 0.0142,
-      "step": 459
-    },
-    {
-      "epoch": 115.0,
-      "learning_rate": 4.254385964912281e-05,
-      "loss": 0.013,
-      "step": 460
-    },
-    {
-      "epoch": 115.0,
-      "eval_accuracy_ELSE": 0.9967589035462682,
-      "eval_accuracy_road": 0.992752178618215,
-      "eval_accuracy_sidewalk": 0.8943959448042805,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9922223000231607,
-      "eval_iou_road": 0.9856922707158106,
-      "eval_iou_sidewalk": 0.8344718864950079,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021300747990608215,
-      "eval_mean_accuracy": 0.9613023423229213,
-      "eval_mean_iou": 0.9374621524113264,
-      "eval_overall_accuracy": 0.9929122924804688,
-      "eval_runtime": 0.6042,
-      "eval_samples_per_second": 1.655,
-      "eval_steps_per_second": 1.655,
-      "step": 460
-    },
-    {
-      "epoch": 115.25,
-      "learning_rate": 4.252192982456141e-05,
-      "loss": 0.014,
-      "step": 461
-    },
-    {
-      "epoch": 115.5,
-      "learning_rate": 4.25e-05,
-      "loss": 0.0127,
-      "step": 462
-    },
-    {
-      "epoch": 115.75,
-      "learning_rate": 4.24780701754386e-05,
-      "loss": 0.0142,
-      "step": 463
-    },
-    {
-      "epoch": 116.0,
-      "learning_rate": 4.2456140350877196e-05,
-      "loss": 0.0149,
-      "step": 464
-    },
-    {
-      "epoch": 116.25,
-      "learning_rate": 4.2434210526315796e-05,
-      "loss": 0.0149,
-      "step": 465
-    },
-    {
-      "epoch": 116.5,
-      "learning_rate": 4.241228070175439e-05,
-      "loss": 0.0135,
-      "step": 466
-    },
-    {
-      "epoch": 116.75,
-      "learning_rate": 4.239035087719299e-05,
-      "loss": 0.0149,
-      "step": 467
-    },
-    {
-      "epoch": 117.0,
-      "learning_rate": 4.236842105263158e-05,
-      "loss": 0.0137,
-      "step": 468
-    },
-    {
-      "epoch": 117.25,
-      "learning_rate": 4.234649122807018e-05,
-      "loss": 0.0128,
-      "step": 469
-    },
-    {
-      "epoch": 117.5,
-      "learning_rate": 4.2324561403508774e-05,
-      "loss": 0.0145,
-      "step": 470
-    },
-    {
-      "epoch": 117.75,
-      "learning_rate": 4.2302631578947374e-05,
-      "loss": 0.0137,
-      "step": 471
-    },
-    {
-      "epoch": 118.0,
-      "learning_rate": 4.228070175438597e-05,
-      "loss": 0.0168,
-      "step": 472
-    },
-    {
-      "epoch": 118.25,
-      "learning_rate": 4.2258771929824567e-05,
-      "loss": 0.0135,
-      "step": 473
-    },
-    {
-      "epoch": 118.5,
-      "learning_rate": 4.223684210526316e-05,
-      "loss": 0.0141,
-      "step": 474
-    },
-    {
-      "epoch": 118.75,
-      "learning_rate": 4.221491228070176e-05,
-      "loss": 0.0139,
-      "step": 475
-    },
-    {
-      "epoch": 119.0,
-      "learning_rate": 4.219298245614035e-05,
-      "loss": 0.0136,
-      "step": 476
-    },
-    {
-      "epoch": 119.25,
-      "learning_rate": 4.217105263157895e-05,
-      "loss": 0.0132,
-      "step": 477
-    },
-    {
-      "epoch": 119.5,
-      "learning_rate": 4.2149122807017545e-05,
-      "loss": 0.0151,
-      "step": 478
-    },
-    {
-      "epoch": 119.75,
-      "learning_rate": 4.2127192982456145e-05,
-      "loss": 0.0154,
-      "step": 479
-    },
-    {
-      "epoch": 120.0,
-      "learning_rate": 4.210526315789474e-05,
-      "loss": 0.0146,
-      "step": 480
-    },
-    {
-      "epoch": 120.0,
-      "eval_accuracy_ELSE": 0.9978789485645027,
-      "eval_accuracy_road": 0.9921683658939454,
-      "eval_accuracy_sidewalk": 0.8914390312588003,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9929949440833931,
-      "eval_iou_road": 0.9865771812080537,
-      "eval_iou_sidewalk": 0.843233883857219,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020083561539649963,
-      "eval_mean_accuracy": 0.9604954485724161,
-      "eval_mean_iou": 0.9409353363828886,
-      "eval_overall_accuracy": 0.9934654235839844,
-      "eval_runtime": 0.5941,
-      "eval_samples_per_second": 1.683,
-      "eval_steps_per_second": 1.683,
-      "step": 480
-    },
-    {
-      "epoch": 120.25,
-      "learning_rate": 4.208333333333334e-05,
-      "loss": 0.0128,
-      "step": 481
-    },
-    {
-      "epoch": 120.5,
-      "learning_rate": 4.206140350877193e-05,
-      "loss": 0.015,
-      "step": 482
-    },
-    {
-      "epoch": 120.75,
-      "learning_rate": 4.203947368421053e-05,
-      "loss": 0.0138,
-      "step": 483
-    },
-    {
-      "epoch": 121.0,
-      "learning_rate": 4.201754385964912e-05,
-      "loss": 0.0142,
-      "step": 484
-    },
-    {
-      "epoch": 121.25,
-      "learning_rate": 4.199561403508772e-05,
-      "loss": 0.014,
-      "step": 485
-    },
-    {
-      "epoch": 121.5,
-      "learning_rate": 4.1973684210526316e-05,
-      "loss": 0.0135,
-      "step": 486
-    },
-    {
-      "epoch": 121.75,
-      "learning_rate": 4.1951754385964916e-05,
-      "loss": 0.016,
-      "step": 487
-    },
-    {
-      "epoch": 122.0,
-      "learning_rate": 4.1929824561403516e-05,
-      "loss": 0.013,
-      "step": 488
-    },
-    {
-      "epoch": 122.25,
-      "learning_rate": 4.190789473684211e-05,
-      "loss": 0.0137,
-      "step": 489
-    },
-    {
-      "epoch": 122.5,
-      "learning_rate": 4.188596491228071e-05,
-      "loss": 0.0131,
-      "step": 490
-    },
-    {
-      "epoch": 122.75,
-      "learning_rate": 4.18640350877193e-05,
-      "loss": 0.0132,
-      "step": 491
-    },
-    {
-      "epoch": 123.0,
-      "learning_rate": 4.18421052631579e-05,
-      "loss": 0.0147,
-      "step": 492
-    },
-    {
-      "epoch": 123.25,
-      "learning_rate": 4.1820175438596494e-05,
-      "loss": 0.0142,
-      "step": 493
-    },
-    {
-      "epoch": 123.5,
-      "learning_rate": 4.1798245614035094e-05,
-      "loss": 0.015,
-      "step": 494
-    },
-    {
-      "epoch": 123.75,
-      "learning_rate": 4.177631578947369e-05,
-      "loss": 0.0138,
-      "step": 495
-    },
-    {
-      "epoch": 124.0,
-      "learning_rate": 4.1754385964912287e-05,
-      "loss": 0.0139,
-      "step": 496
-    },
-    {
-      "epoch": 124.25,
-      "learning_rate": 4.173245614035088e-05,
-      "loss": 0.0134,
-      "step": 497
-    },
-    {
-      "epoch": 124.5,
-      "learning_rate": 4.171052631578948e-05,
-      "loss": 0.0136,
-      "step": 498
-    },
-    {
-      "epoch": 124.75,
-      "learning_rate": 4.168859649122807e-05,
-      "loss": 0.0148,
-      "step": 499
-    },
-    {
-      "epoch": 125.0,
-      "learning_rate": 4.166666666666667e-05,
-      "loss": 0.0141,
-      "step": 500
-    },
-    {
-      "epoch": 125.0,
-      "eval_accuracy_ELSE": 0.9958769357299772,
-      "eval_accuracy_road": 0.993720453380418,
-      "eval_accuracy_sidewalk": 0.9110109828217403,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9926007420830097,
-      "eval_iou_road": 0.9864444632912108,
-      "eval_iou_sidewalk": 0.8285311819695224,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020598702132701874,
-      "eval_mean_accuracy": 0.9668694573107118,
-      "eval_mean_iou": 0.9358587957812476,
-      "eval_overall_accuracy": 0.9930000305175781,
-      "eval_runtime": 0.6022,
-      "eval_samples_per_second": 1.661,
-      "eval_steps_per_second": 1.661,
-      "step": 500
-    },
-    {
-      "epoch": 125.25,
-      "learning_rate": 4.1644736842105265e-05,
-      "loss": 0.0126,
-      "step": 501
-    },
-    {
-      "epoch": 125.5,
-      "learning_rate": 4.1622807017543865e-05,
-      "loss": 0.0142,
-      "step": 502
-    },
-    {
-      "epoch": 125.75,
-      "learning_rate": 4.160087719298246e-05,
-      "loss": 0.0137,
-      "step": 503
-    },
-    {
-      "epoch": 126.0,
-      "learning_rate": 4.157894736842106e-05,
-      "loss": 0.0144,
-      "step": 504
-    },
-    {
-      "epoch": 126.25,
-      "learning_rate": 4.155701754385965e-05,
-      "loss": 0.0152,
-      "step": 505
-    },
-    {
-      "epoch": 126.5,
-      "learning_rate": 4.153508771929825e-05,
-      "loss": 0.0118,
-      "step": 506
-    },
-    {
-      "epoch": 126.75,
-      "learning_rate": 4.151315789473684e-05,
-      "loss": 0.0143,
-      "step": 507
-    },
-    {
-      "epoch": 127.0,
-      "learning_rate": 4.149122807017544e-05,
-      "loss": 0.0132,
-      "step": 508
-    },
-    {
-      "epoch": 127.25,
-      "learning_rate": 4.1469298245614036e-05,
-      "loss": 0.0139,
-      "step": 509
-    },
-    {
-      "epoch": 127.5,
-      "learning_rate": 4.1447368421052636e-05,
-      "loss": 0.0132,
-      "step": 510
-    },
-    {
-      "epoch": 127.75,
-      "learning_rate": 4.142543859649123e-05,
-      "loss": 0.0117,
-      "step": 511
-    },
-    {
-      "epoch": 128.0,
-      "learning_rate": 4.140350877192983e-05,
-      "loss": 0.0141,
-      "step": 512
-    },
-    {
-      "epoch": 128.25,
-      "learning_rate": 4.138157894736842e-05,
-      "loss": 0.0126,
-      "step": 513
-    },
-    {
-      "epoch": 128.5,
-      "learning_rate": 4.135964912280702e-05,
-      "loss": 0.0155,
-      "step": 514
-    },
-    {
-      "epoch": 128.75,
-      "learning_rate": 4.1337719298245614e-05,
-      "loss": 0.0143,
-      "step": 515
-    },
-    {
-      "epoch": 129.0,
-      "learning_rate": 4.1315789473684214e-05,
-      "loss": 0.0122,
-      "step": 516
-    },
-    {
-      "epoch": 129.25,
-      "learning_rate": 4.129385964912281e-05,
-      "loss": 0.0137,
-      "step": 517
-    },
-    {
-      "epoch": 129.5,
-      "learning_rate": 4.127192982456141e-05,
-      "loss": 0.0126,
-      "step": 518
-    },
-    {
-      "epoch": 129.75,
-      "learning_rate": 4.125e-05,
-      "loss": 0.013,
-      "step": 519
-    },
-    {
-      "epoch": 130.0,
-      "learning_rate": 4.12280701754386e-05,
-      "loss": 0.0146,
-      "step": 520
-    },
-    {
-      "epoch": 130.0,
-      "eval_accuracy_ELSE": 0.9963260359063707,
-      "eval_accuracy_road": 0.994176112092043,
-      "eval_accuracy_sidewalk": 0.898338496198254,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9925451977705667,
-      "eval_iou_road": 0.9862555090970732,
-      "eval_iou_sidewalk": 0.8346415489272632,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020625747740268707,
-      "eval_mean_accuracy": 0.9629468813988892,
-      "eval_mean_iou": 0.9378140852649678,
-      "eval_overall_accuracy": 0.9930953979492188,
-      "eval_runtime": 0.6069,
-      "eval_samples_per_second": 1.648,
-      "eval_steps_per_second": 1.648,
-      "step": 520
-    },
-    {
-      "epoch": 130.25,
-      "learning_rate": 4.120614035087719e-05,
-      "loss": 0.0121,
-      "step": 521
-    },
-    {
-      "epoch": 130.5,
-      "learning_rate": 4.118421052631579e-05,
-      "loss": 0.0127,
-      "step": 522
-    },
-    {
-      "epoch": 130.75,
-      "learning_rate": 4.1162280701754385e-05,
-      "loss": 0.0131,
-      "step": 523
-    },
-    {
-      "epoch": 131.0,
-      "learning_rate": 4.1140350877192985e-05,
-      "loss": 0.0149,
-      "step": 524
-    },
-    {
-      "epoch": 131.25,
-      "learning_rate": 4.111842105263158e-05,
-      "loss": 0.0128,
-      "step": 525
-    },
-    {
-      "epoch": 131.5,
-      "learning_rate": 4.109649122807018e-05,
-      "loss": 0.013,
-      "step": 526
-    },
-    {
-      "epoch": 131.75,
-      "learning_rate": 4.107456140350877e-05,
-      "loss": 0.0135,
-      "step": 527
-    },
-    {
-      "epoch": 132.0,
-      "learning_rate": 4.105263157894737e-05,
-      "loss": 0.0125,
-      "step": 528
-    },
-    {
-      "epoch": 132.25,
-      "learning_rate": 4.103070175438597e-05,
-      "loss": 0.013,
-      "step": 529
-    },
-    {
-      "epoch": 132.5,
-      "learning_rate": 4.100877192982456e-05,
-      "loss": 0.0131,
-      "step": 530
-    },
-    {
-      "epoch": 132.75,
-      "learning_rate": 4.098684210526316e-05,
-      "loss": 0.0134,
-      "step": 531
-    },
-    {
-      "epoch": 133.0,
-      "learning_rate": 4.0964912280701756e-05,
-      "loss": 0.015,
-      "step": 532
-    },
-    {
-      "epoch": 133.25,
-      "learning_rate": 4.0942982456140356e-05,
-      "loss": 0.0124,
-      "step": 533
-    },
-    {
-      "epoch": 133.5,
-      "learning_rate": 4.092105263157895e-05,
-      "loss": 0.0141,
-      "step": 534
-    },
-    {
-      "epoch": 133.75,
-      "learning_rate": 4.089912280701755e-05,
-      "loss": 0.0128,
-      "step": 535
-    },
-    {
-      "epoch": 134.0,
-      "learning_rate": 4.087719298245614e-05,
-      "loss": 0.0141,
-      "step": 536
-    },
-    {
-      "epoch": 134.25,
-      "learning_rate": 4.085526315789474e-05,
-      "loss": 0.0135,
-      "step": 537
-    },
-    {
-      "epoch": 134.5,
-      "learning_rate": 4.0833333333333334e-05,
-      "loss": 0.0139,
-      "step": 538
-    },
-    {
-      "epoch": 134.75,
-      "learning_rate": 4.0811403508771934e-05,
-      "loss": 0.0124,
-      "step": 539
-    },
-    {
-      "epoch": 135.0,
-      "learning_rate": 4.078947368421053e-05,
-      "loss": 0.0132,
-      "step": 540
-    },
-    {
-      "epoch": 135.0,
-      "eval_accuracy_ELSE": 0.9962881599878797,
-      "eval_accuracy_road": 0.9947456854815743,
-      "eval_accuracy_sidewalk": 0.8931286961419318,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9926465831410519,
-      "eval_iou_road": 0.986792666045145,
-      "eval_iou_sidewalk": 0.8267726798748697,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020599104464054108,
-      "eval_mean_accuracy": 0.961387513870462,
-      "eval_mean_iou": 0.9354039763536889,
-      "eval_overall_accuracy": 0.9930801391601562,
-      "eval_runtime": 0.5792,
-      "eval_samples_per_second": 1.726,
-      "eval_steps_per_second": 1.726,
-      "step": 540
-    },
-    {
-      "epoch": 135.25,
-      "learning_rate": 4.076754385964913e-05,
-      "loss": 0.0135,
-      "step": 541
-    },
-    {
-      "epoch": 135.5,
-      "learning_rate": 4.074561403508772e-05,
-      "loss": 0.0129,
-      "step": 542
-    },
-    {
-      "epoch": 135.75,
-      "learning_rate": 4.072368421052632e-05,
-      "loss": 0.0121,
-      "step": 543
-    },
-    {
-      "epoch": 136.0,
-      "learning_rate": 4.070175438596491e-05,
-      "loss": 0.0135,
-      "step": 544
-    },
-    {
-      "epoch": 136.25,
-      "learning_rate": 4.067982456140351e-05,
-      "loss": 0.0136,
-      "step": 545
-    },
-    {
-      "epoch": 136.5,
-      "learning_rate": 4.0657894736842105e-05,
-      "loss": 0.0126,
-      "step": 546
-    },
-    {
-      "epoch": 136.75,
-      "learning_rate": 4.0635964912280705e-05,
-      "loss": 0.0139,
-      "step": 547
-    },
-    {
-      "epoch": 137.0,
-      "learning_rate": 4.06140350877193e-05,
-      "loss": 0.0146,
-      "step": 548
-    },
-    {
-      "epoch": 137.25,
-      "learning_rate": 4.05921052631579e-05,
-      "loss": 0.0133,
-      "step": 549
-    },
-    {
-      "epoch": 137.5,
-      "learning_rate": 4.057017543859649e-05,
-      "loss": 0.012,
-      "step": 550
-    },
-    {
-      "epoch": 137.75,
-      "learning_rate": 4.054824561403509e-05,
-      "loss": 0.0128,
-      "step": 551
-    },
-    {
-      "epoch": 138.0,
-      "learning_rate": 4.0526315789473684e-05,
-      "loss": 0.0128,
-      "step": 552
-    },
-    {
-      "epoch": 138.25,
-      "learning_rate": 4.050438596491228e-05,
-      "loss": 0.0117,
-      "step": 553
-    },
-    {
-      "epoch": 138.5,
-      "learning_rate": 4.0482456140350876e-05,
-      "loss": 0.0134,
-      "step": 554
-    },
-    {
-      "epoch": 138.75,
-      "learning_rate": 4.0460526315789476e-05,
-      "loss": 0.0121,
-      "step": 555
-    },
-    {
-      "epoch": 139.0,
-      "learning_rate": 4.043859649122807e-05,
-      "loss": 0.0125,
-      "step": 556
-    },
-    {
-      "epoch": 139.25,
-      "learning_rate": 4.041666666666667e-05,
-      "loss": 0.0137,
-      "step": 557
-    },
-    {
-      "epoch": 139.5,
-      "learning_rate": 4.039473684210526e-05,
-      "loss": 0.0126,
-      "step": 558
-    },
-    {
-      "epoch": 139.75,
-      "learning_rate": 4.037280701754386e-05,
-      "loss": 0.0118,
-      "step": 559
-    },
-    {
-      "epoch": 140.0,
-      "learning_rate": 4.0350877192982455e-05,
-      "loss": 0.0116,
-      "step": 560
-    },
-    {
-      "epoch": 140.0,
-      "eval_accuracy_ELSE": 0.9966236324088003,
-      "eval_accuracy_road": 0.9937489320498947,
-      "eval_accuracy_sidewalk": 0.9062235989862011,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931253875394279,
-      "eval_iou_road": 0.9870028851049386,
-      "eval_iou_sidewalk": 0.8358441558441558,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.01912616565823555,
-      "eval_mean_accuracy": 0.965532054481632,
-      "eval_mean_iou": 0.9386574761628408,
-      "eval_overall_accuracy": 0.9934043884277344,
-      "eval_runtime": 0.6102,
-      "eval_samples_per_second": 1.639,
-      "eval_steps_per_second": 1.639,
-      "step": 560
-    },
-    {
-      "epoch": 140.25,
-      "learning_rate": 4.0328947368421054e-05,
-      "loss": 0.0125,
-      "step": 561
-    },
-    {
-      "epoch": 140.5,
-      "learning_rate": 4.030701754385965e-05,
-      "loss": 0.0127,
-      "step": 562
-    },
-    {
-      "epoch": 140.75,
-      "learning_rate": 4.028508771929825e-05,
-      "loss": 0.0137,
-      "step": 563
-    },
-    {
-      "epoch": 141.0,
-      "learning_rate": 4.026315789473684e-05,
-      "loss": 0.0128,
-      "step": 564
-    },
-    {
-      "epoch": 141.25,
-      "learning_rate": 4.024122807017544e-05,
-      "loss": 0.0133,
-      "step": 565
-    },
-    {
-      "epoch": 141.5,
-      "learning_rate": 4.021929824561403e-05,
-      "loss": 0.0131,
-      "step": 566
-    },
-    {
-      "epoch": 141.75,
-      "learning_rate": 4.019736842105263e-05,
-      "loss": 0.0115,
-      "step": 567
-    },
-    {
-      "epoch": 142.0,
-      "learning_rate": 4.017543859649123e-05,
-      "loss": 0.0129,
-      "step": 568
-    },
-    {
-      "epoch": 142.25,
-      "learning_rate": 4.0153508771929825e-05,
-      "loss": 0.0119,
-      "step": 569
-    },
-    {
-      "epoch": 142.5,
-      "learning_rate": 4.0131578947368425e-05,
-      "loss": 0.0133,
-      "step": 570
-    },
-    {
-      "epoch": 142.75,
-      "learning_rate": 4.010964912280702e-05,
-      "loss": 0.0121,
-      "step": 571
-    },
-    {
-      "epoch": 143.0,
-      "learning_rate": 4.008771929824562e-05,
-      "loss": 0.0129,
-      "step": 572
-    },
-    {
-      "epoch": 143.25,
-      "learning_rate": 4.006578947368421e-05,
-      "loss": 0.0132,
-      "step": 573
-    },
-    {
-      "epoch": 143.5,
-      "learning_rate": 4.004385964912281e-05,
-      "loss": 0.0113,
-      "step": 574
-    },
-    {
-      "epoch": 143.75,
-      "learning_rate": 4.0021929824561404e-05,
-      "loss": 0.0121,
-      "step": 575
-    },
-    {
-      "epoch": 144.0,
-      "learning_rate": 4e-05,
-      "loss": 0.0132,
-      "step": 576
-    },
-    {
-      "epoch": 144.25,
-      "learning_rate": 3.9978070175438596e-05,
-      "loss": 0.0117,
-      "step": 577
-    },
-    {
-      "epoch": 144.5,
-      "learning_rate": 3.9956140350877196e-05,
-      "loss": 0.0125,
-      "step": 578
-    },
-    {
-      "epoch": 144.75,
-      "learning_rate": 3.993421052631579e-05,
-      "loss": 0.013,
-      "step": 579
-    },
-    {
-      "epoch": 145.0,
-      "learning_rate": 3.991228070175439e-05,
-      "loss": 0.012,
-      "step": 580
-    },
-    {
-      "epoch": 145.0,
-      "eval_accuracy_ELSE": 0.9972945772506412,
-      "eval_accuracy_road": 0.9927236999487384,
-      "eval_accuracy_sidewalk": 0.8953815826527739,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9930497187560613,
-      "eval_iou_road": 0.986305439626512,
-      "eval_iou_sidewalk": 0.8357208568800105,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020623385906219482,
-      "eval_mean_accuracy": 0.9617999532840512,
-      "eval_mean_iou": 0.9383586717541946,
-      "eval_overall_accuracy": 0.9933090209960938,
-      "eval_runtime": 0.6203,
-      "eval_samples_per_second": 1.612,
-      "eval_steps_per_second": 1.612,
-      "step": 580
-    },
-    {
-      "epoch": 145.25,
-      "learning_rate": 3.989035087719298e-05,
-      "loss": 0.0111,
-      "step": 581
-    },
-    {
-      "epoch": 145.5,
-      "learning_rate": 3.986842105263158e-05,
-      "loss": 0.0143,
-      "step": 582
-    },
-    {
-      "epoch": 145.75,
-      "learning_rate": 3.9846491228070175e-05,
-      "loss": 0.0126,
-      "step": 583
-    },
-    {
-      "epoch": 146.0,
-      "learning_rate": 3.9824561403508774e-05,
-      "loss": 0.0127,
-      "step": 584
-    },
-    {
-      "epoch": 146.25,
-      "learning_rate": 3.980263157894737e-05,
-      "loss": 0.0124,
-      "step": 585
-    },
-    {
-      "epoch": 146.5,
-      "learning_rate": 3.978070175438597e-05,
-      "loss": 0.0129,
-      "step": 586
-    },
-    {
-      "epoch": 146.75,
-      "learning_rate": 3.975877192982456e-05,
-      "loss": 0.0129,
-      "step": 587
-    },
-    {
-      "epoch": 147.0,
-      "learning_rate": 3.973684210526316e-05,
-      "loss": 0.0122,
-      "step": 588
-    },
-    {
-      "epoch": 147.25,
-      "learning_rate": 3.971491228070175e-05,
-      "loss": 0.0119,
-      "step": 589
-    },
-    {
-      "epoch": 147.5,
-      "learning_rate": 3.969298245614035e-05,
-      "loss": 0.0141,
-      "step": 590
-    },
-    {
-      "epoch": 147.75,
-      "learning_rate": 3.9671052631578945e-05,
-      "loss": 0.0141,
-      "step": 591
-    },
-    {
-      "epoch": 148.0,
-      "learning_rate": 3.9649122807017545e-05,
-      "loss": 0.0107,
-      "step": 592
-    },
-    {
-      "epoch": 148.25,
-      "learning_rate": 3.962719298245614e-05,
-      "loss": 0.0111,
-      "step": 593
-    },
-    {
-      "epoch": 148.5,
-      "learning_rate": 3.960526315789474e-05,
-      "loss": 0.0136,
-      "step": 594
-    },
-    {
-      "epoch": 148.75,
-      "learning_rate": 3.958333333333333e-05,
-      "loss": 0.013,
-      "step": 595
-    },
-    {
-      "epoch": 149.0,
-      "learning_rate": 3.956140350877193e-05,
-      "loss": 0.0122,
-      "step": 596
-    },
-    {
-      "epoch": 149.25,
-      "learning_rate": 3.9539473684210524e-05,
-      "loss": 0.012,
-      "step": 597
-    },
-    {
-      "epoch": 149.5,
-      "learning_rate": 3.9517543859649124e-05,
-      "loss": 0.0122,
-      "step": 598
-    },
-    {
-      "epoch": 149.75,
-      "learning_rate": 3.9495614035087716e-05,
-      "loss": 0.0123,
-      "step": 599
-    },
-    {
-      "epoch": 150.0,
-      "learning_rate": 3.9473684210526316e-05,
-      "loss": 0.0131,
-      "step": 600
-    },
-    {
-      "epoch": 150.0,
-      "eval_accuracy_ELSE": 0.9959039899574708,
-      "eval_accuracy_road": 0.9936777353762032,
-      "eval_accuracy_sidewalk": 0.9103069557871022,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9926277073086548,
-      "eval_iou_road": 0.9865554534530289,
-      "eval_iou_sidewalk": 0.82640930589288,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020645052194595337,
-      "eval_mean_accuracy": 0.966629560373592,
-      "eval_mean_iou": 0.9351974888848545,
-      "eval_overall_accuracy": 0.9929885864257812,
-      "eval_runtime": 0.6055,
-      "eval_samples_per_second": 1.652,
-      "eval_steps_per_second": 1.652,
-      "step": 600
-    },
-    {
-      "epoch": 150.25,
-      "learning_rate": 3.945175438596491e-05,
-      "loss": 0.0122,
-      "step": 601
-    },
-    {
-      "epoch": 150.5,
-      "learning_rate": 3.942982456140351e-05,
-      "loss": 0.0111,
-      "step": 602
-    },
-    {
-      "epoch": 150.75,
-      "learning_rate": 3.94078947368421e-05,
-      "loss": 0.0117,
-      "step": 603
-    },
-    {
-      "epoch": 151.0,
-      "learning_rate": 3.93859649122807e-05,
-      "loss": 0.0127,
-      "step": 604
-    },
-    {
-      "epoch": 151.25,
-      "learning_rate": 3.9364035087719295e-05,
-      "loss": 0.0106,
-      "step": 605
-    },
-    {
-      "epoch": 151.5,
-      "learning_rate": 3.9342105263157895e-05,
-      "loss": 0.0119,
-      "step": 606
-    },
-    {
-      "epoch": 151.75,
-      "learning_rate": 3.932017543859649e-05,
-      "loss": 0.0128,
-      "step": 607
-    },
-    {
-      "epoch": 152.0,
-      "learning_rate": 3.929824561403509e-05,
-      "loss": 0.0126,
-      "step": 608
-    },
-    {
-      "epoch": 152.25,
-      "learning_rate": 3.927631578947369e-05,
-      "loss": 0.0122,
-      "step": 609
-    },
-    {
-      "epoch": 152.5,
-      "learning_rate": 3.925438596491228e-05,
-      "loss": 0.0117,
-      "step": 610
-    },
-    {
-      "epoch": 152.75,
-      "learning_rate": 3.923245614035088e-05,
-      "loss": 0.0117,
-      "step": 611
-    },
-    {
-      "epoch": 153.0,
-      "learning_rate": 3.921052631578947e-05,
-      "loss": 0.0113,
-      "step": 612
-    },
-    {
-      "epoch": 153.25,
-      "learning_rate": 3.918859649122807e-05,
-      "loss": 0.0125,
-      "step": 613
-    },
-    {
-      "epoch": 153.5,
-      "learning_rate": 3.9166666666666665e-05,
-      "loss": 0.0118,
-      "step": 614
-    },
-    {
-      "epoch": 153.75,
-      "learning_rate": 3.9144736842105265e-05,
-      "loss": 0.0117,
-      "step": 615
-    },
-    {
-      "epoch": 154.0,
-      "learning_rate": 3.912280701754386e-05,
-      "loss": 0.0118,
-      "step": 616
-    },
-    {
-      "epoch": 154.25,
-      "learning_rate": 3.910087719298246e-05,
-      "loss": 0.0117,
-      "step": 617
-    },
-    {
-      "epoch": 154.5,
-      "learning_rate": 3.907894736842105e-05,
-      "loss": 0.0113,
-      "step": 618
-    },
-    {
-      "epoch": 154.75,
-      "learning_rate": 3.905701754385965e-05,
-      "loss": 0.0113,
-      "step": 619
-    },
-    {
-      "epoch": 155.0,
-      "learning_rate": 3.9035087719298244e-05,
-      "loss": 0.0126,
-      "step": 620
-    },
-    {
-      "epoch": 155.0,
-      "eval_accuracy_ELSE": 0.9965803456448105,
-      "eval_accuracy_road": 0.9936350173719883,
-      "eval_accuracy_sidewalk": 0.8967896367220501,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9927182765330157,
-      "eval_iou_road": 0.9859694237997004,
-      "eval_iou_sidewalk": 0.8326578637730422,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020528439432382584,
-      "eval_mean_accuracy": 0.9623349999129496,
-      "eval_mean_iou": 0.9371151880352527,
-      "eval_overall_accuracy": 0.9930877685546875,
-      "eval_runtime": 0.616,
-      "eval_samples_per_second": 1.623,
-      "eval_steps_per_second": 1.623,
-      "step": 620
-    },
-    {
-      "epoch": 155.25,
-      "learning_rate": 3.9013157894736844e-05,
-      "loss": 0.0113,
-      "step": 621
-    },
-    {
-      "epoch": 155.5,
-      "learning_rate": 3.8991228070175436e-05,
-      "loss": 0.0119,
-      "step": 622
-    },
-    {
-      "epoch": 155.75,
-      "learning_rate": 3.8969298245614036e-05,
-      "loss": 0.0117,
-      "step": 623
-    },
-    {
-      "epoch": 156.0,
-      "learning_rate": 3.894736842105263e-05,
-      "loss": 0.0121,
-      "step": 624
-    },
-    {
-      "epoch": 156.25,
-      "learning_rate": 3.892543859649123e-05,
-      "loss": 0.0106,
-      "step": 625
-    },
-    {
-      "epoch": 156.5,
-      "learning_rate": 3.890350877192982e-05,
-      "loss": 0.0117,
-      "step": 626
-    },
-    {
-      "epoch": 156.75,
-      "learning_rate": 3.888157894736842e-05,
-      "loss": 0.0117,
-      "step": 627
-    },
-    {
-      "epoch": 157.0,
-      "learning_rate": 3.8859649122807015e-05,
-      "loss": 0.0141,
-      "step": 628
-    },
-    {
-      "epoch": 157.25,
-      "learning_rate": 3.8837719298245614e-05,
-      "loss": 0.0118,
-      "step": 629
-    },
-    {
-      "epoch": 157.5,
-      "learning_rate": 3.8815789473684214e-05,
-      "loss": 0.0113,
-      "step": 630
-    },
-    {
-      "epoch": 157.75,
-      "learning_rate": 3.879385964912281e-05,
-      "loss": 0.0117,
-      "step": 631
-    },
-    {
-      "epoch": 158.0,
-      "learning_rate": 3.877192982456141e-05,
-      "loss": 0.0117,
-      "step": 632
-    },
-    {
-      "epoch": 158.25,
-      "learning_rate": 3.875e-05,
-      "loss": 0.0128,
-      "step": 633
-    },
-    {
-      "epoch": 158.5,
-      "learning_rate": 3.87280701754386e-05,
-      "loss": 0.0111,
-      "step": 634
-    },
-    {
-      "epoch": 158.75,
-      "learning_rate": 3.870614035087719e-05,
-      "loss": 0.0118,
-      "step": 635
-    },
-    {
-      "epoch": 159.0,
-      "learning_rate": 3.868421052631579e-05,
-      "loss": 0.012,
-      "step": 636
-    },
-    {
-      "epoch": 159.25,
-      "learning_rate": 3.8662280701754385e-05,
-      "loss": 0.0109,
-      "step": 637
-    },
-    {
-      "epoch": 159.5,
-      "learning_rate": 3.8640350877192985e-05,
-      "loss": 0.012,
-      "step": 638
-    },
-    {
-      "epoch": 159.75,
-      "learning_rate": 3.861842105263158e-05,
-      "loss": 0.0132,
-      "step": 639
-    },
-    {
-      "epoch": 160.0,
-      "learning_rate": 3.859649122807018e-05,
-      "loss": 0.011,
-      "step": 640
-    },
-    {
-      "epoch": 160.0,
-      "eval_accuracy_ELSE": 0.9968454770742476,
-      "eval_accuracy_road": 0.9944608987868087,
-      "eval_accuracy_sidewalk": 0.8958039988735568,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932714754770082,
-      "eval_iou_road": 0.9868028768033007,
-      "eval_iou_sidewalk": 0.836664913203577,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.019651338458061218,
-      "eval_mean_accuracy": 0.9623701249115376,
-      "eval_mean_iou": 0.9389130884946285,
-      "eval_overall_accuracy": 0.99346923828125,
-      "eval_runtime": 0.6037,
-      "eval_samples_per_second": 1.656,
-      "eval_steps_per_second": 1.656,
-      "step": 640
-    },
-    {
-      "epoch": 160.25,
-      "learning_rate": 3.857456140350877e-05,
-      "loss": 0.012,
-      "step": 641
-    },
-    {
-      "epoch": 160.5,
-      "learning_rate": 3.855263157894737e-05,
-      "loss": 0.011,
-      "step": 642
-    },
-    {
-      "epoch": 160.75,
-      "learning_rate": 3.853070175438597e-05,
-      "loss": 0.0109,
-      "step": 643
-    },
-    {
-      "epoch": 161.0,
-      "learning_rate": 3.8508771929824563e-05,
-      "loss": 0.0117,
-      "step": 644
-    },
-    {
-      "epoch": 161.25,
-      "learning_rate": 3.848684210526316e-05,
-      "loss": 0.0119,
-      "step": 645
-    },
-    {
-      "epoch": 161.5,
-      "learning_rate": 3.8464912280701756e-05,
-      "loss": 0.0122,
-      "step": 646
-    },
-    {
-      "epoch": 161.75,
-      "learning_rate": 3.8442982456140356e-05,
-      "loss": 0.0111,
-      "step": 647
-    },
-    {
-      "epoch": 162.0,
-      "learning_rate": 3.842105263157895e-05,
-      "loss": 0.0102,
-      "step": 648
-    },
-    {
-      "epoch": 162.25,
-      "learning_rate": 3.839912280701755e-05,
-      "loss": 0.0117,
-      "step": 649
-    },
-    {
-      "epoch": 162.5,
-      "learning_rate": 3.837719298245614e-05,
-      "loss": 0.0116,
-      "step": 650
-    },
-    {
-      "epoch": 162.75,
-      "learning_rate": 3.835526315789474e-05,
-      "loss": 0.0117,
-      "step": 651
-    },
-    {
-      "epoch": 163.0,
-      "learning_rate": 3.8333333333333334e-05,
-      "loss": 0.0118,
-      "step": 652
-    },
-    {
-      "epoch": 163.25,
-      "learning_rate": 3.8311403508771934e-05,
-      "loss": 0.0124,
-      "step": 653
-    },
-    {
-      "epoch": 163.5,
-      "learning_rate": 3.828947368421053e-05,
-      "loss": 0.0119,
-      "step": 654
-    },
-    {
-      "epoch": 163.75,
-      "learning_rate": 3.826754385964913e-05,
-      "loss": 0.0117,
-      "step": 655
-    },
-    {
-      "epoch": 164.0,
-      "learning_rate": 3.824561403508773e-05,
-      "loss": 0.0118,
-      "step": 656
-    },
-    {
-      "epoch": 164.25,
-      "learning_rate": 3.822368421052632e-05,
-      "loss": 0.0106,
-      "step": 657
-    },
-    {
-      "epoch": 164.5,
-      "learning_rate": 3.820175438596492e-05,
-      "loss": 0.0122,
-      "step": 658
-    },
-    {
-      "epoch": 164.75,
-      "learning_rate": 3.817982456140351e-05,
-      "loss": 0.0111,
-      "step": 659
-    },
-    {
-      "epoch": 165.0,
-      "learning_rate": 3.815789473684211e-05,
-      "loss": 0.0124,
-      "step": 660
-    },
-    {
-      "epoch": 165.0,
-      "eval_accuracy_ELSE": 0.9969645156752194,
-      "eval_accuracy_road": 0.993663496041465,
-      "eval_accuracy_sidewalk": 0.8981976907913264,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931437811615685,
-      "eval_iou_road": 0.9871553663125433,
-      "eval_iou_sidewalk": 0.8330939010056158,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.019842373207211494,
-      "eval_mean_accuracy": 0.9629419008360035,
-      "eval_mean_iou": 0.9377976828265759,
-      "eval_overall_accuracy": 0.9934043884277344,
-      "eval_runtime": 0.6163,
-      "eval_samples_per_second": 1.623,
-      "eval_steps_per_second": 1.623,
-      "step": 660
-    },
-    {
-      "epoch": 165.25,
-      "learning_rate": 3.8135964912280705e-05,
-      "loss": 0.0137,
-      "step": 661
-    },
-    {
-      "epoch": 165.5,
-      "learning_rate": 3.8114035087719305e-05,
-      "loss": 0.0108,
-      "step": 662
-    },
-    {
-      "epoch": 165.75,
-      "learning_rate": 3.80921052631579e-05,
-      "loss": 0.0109,
-      "step": 663
-    },
-    {
-      "epoch": 166.0,
-      "learning_rate": 3.80701754385965e-05,
-      "loss": 0.012,
-      "step": 664
-    },
-    {
-      "epoch": 166.25,
-      "learning_rate": 3.804824561403509e-05,
-      "loss": 0.0114,
-      "step": 665
-    },
-    {
-      "epoch": 166.5,
-      "learning_rate": 3.802631578947369e-05,
-      "loss": 0.0106,
-      "step": 666
-    },
-    {
-      "epoch": 166.75,
-      "learning_rate": 3.8004385964912283e-05,
-      "loss": 0.0124,
-      "step": 667
-    },
-    {
-      "epoch": 167.0,
-      "learning_rate": 3.798245614035088e-05,
-      "loss": 0.0111,
-      "step": 668
-    },
-    {
-      "epoch": 167.25,
-      "learning_rate": 3.7960526315789476e-05,
-      "loss": 0.0108,
-      "step": 669
-    },
-    {
-      "epoch": 167.5,
-      "learning_rate": 3.7938596491228076e-05,
-      "loss": 0.0111,
-      "step": 670
-    },
-    {
-      "epoch": 167.75,
-      "learning_rate": 3.791666666666667e-05,
-      "loss": 0.0109,
-      "step": 671
-    },
-    {
-      "epoch": 168.0,
-      "learning_rate": 3.789473684210527e-05,
-      "loss": 0.0114,
-      "step": 672
-    },
-    {
-      "epoch": 168.25,
-      "learning_rate": 3.787280701754386e-05,
-      "loss": 0.0107,
-      "step": 673
-    },
-    {
-      "epoch": 168.5,
-      "learning_rate": 3.785087719298246e-05,
-      "loss": 0.0114,
-      "step": 674
-    },
-    {
-      "epoch": 168.75,
-      "learning_rate": 3.7828947368421054e-05,
-      "loss": 0.0112,
-      "step": 675
-    },
-    {
-      "epoch": 169.0,
-      "learning_rate": 3.7807017543859654e-05,
-      "loss": 0.0111,
-      "step": 676
-    },
-    {
-      "epoch": 169.25,
-      "learning_rate": 3.778508771929825e-05,
-      "loss": 0.0111,
-      "step": 677
-    },
-    {
-      "epoch": 169.5,
-      "learning_rate": 3.776315789473685e-05,
-      "loss": 0.0102,
-      "step": 678
-    },
-    {
-      "epoch": 169.75,
-      "learning_rate": 3.774122807017544e-05,
-      "loss": 0.0108,
-      "step": 679
-    },
-    {
-      "epoch": 170.0,
-      "learning_rate": 3.771929824561404e-05,
-      "loss": 0.0119,
-      "step": 680
-    },
-    {
-      "epoch": 170.0,
-      "eval_accuracy_ELSE": 0.9962394623783912,
-      "eval_accuracy_road": 0.9943327447741641,
-      "eval_accuracy_sidewalk": 0.9036891016615038,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9928710479343834,
-      "eval_iou_road": 0.9872336815913364,
-      "eval_iou_sidewalk": 0.8284497224732155,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02057482674717903,
-      "eval_mean_accuracy": 0.9647537696046863,
-      "eval_mean_iou": 0.9361848173329784,
-      "eval_overall_accuracy": 0.9932212829589844,
-      "eval_runtime": 0.6071,
-      "eval_samples_per_second": 1.647,
-      "eval_steps_per_second": 1.647,
-      "step": 680
-    },
-    {
-      "epoch": 170.25,
-      "learning_rate": 3.769736842105263e-05,
-      "loss": 0.0126,
-      "step": 681
-    },
-    {
-      "epoch": 170.5,
-      "learning_rate": 3.767543859649123e-05,
-      "loss": 0.0111,
-      "step": 682
-    },
-    {
-      "epoch": 170.75,
-      "learning_rate": 3.7653508771929825e-05,
-      "loss": 0.0106,
-      "step": 683
-    },
-    {
-      "epoch": 171.0,
-      "learning_rate": 3.7631578947368425e-05,
-      "loss": 0.0108,
-      "step": 684
-    },
-    {
-      "epoch": 171.25,
-      "learning_rate": 3.760964912280702e-05,
-      "loss": 0.0117,
-      "step": 685
-    },
-    {
-      "epoch": 171.5,
-      "learning_rate": 3.758771929824562e-05,
-      "loss": 0.0114,
-      "step": 686
-    },
-    {
-      "epoch": 171.75,
-      "learning_rate": 3.756578947368421e-05,
-      "loss": 0.0106,
-      "step": 687
-    },
-    {
-      "epoch": 172.0,
-      "learning_rate": 3.754385964912281e-05,
-      "loss": 0.012,
-      "step": 688
-    },
-    {
-      "epoch": 172.25,
-      "learning_rate": 3.7521929824561404e-05,
-      "loss": 0.0103,
-      "step": 689
-    },
-    {
-      "epoch": 172.5,
-      "learning_rate": 3.7500000000000003e-05,
-      "loss": 0.0108,
-      "step": 690
-    },
-    {
-      "epoch": 172.75,
-      "learning_rate": 3.74780701754386e-05,
-      "loss": 0.0115,
-      "step": 691
-    },
-    {
-      "epoch": 173.0,
-      "learning_rate": 3.7456140350877196e-05,
-      "loss": 0.011,
-      "step": 692
-    },
-    {
-      "epoch": 173.25,
-      "learning_rate": 3.7434210526315796e-05,
-      "loss": 0.0112,
-      "step": 693
-    },
-    {
-      "epoch": 173.5,
-      "learning_rate": 3.741228070175439e-05,
-      "loss": 0.0107,
-      "step": 694
-    },
-    {
-      "epoch": 173.75,
-      "learning_rate": 3.739035087719299e-05,
-      "loss": 0.0108,
-      "step": 695
-    },
-    {
-      "epoch": 174.0,
-      "learning_rate": 3.736842105263158e-05,
-      "loss": 0.012,
-      "step": 696
-    },
-    {
-      "epoch": 174.25,
-      "learning_rate": 3.734649122807018e-05,
-      "loss": 0.0112,
-      "step": 697
-    },
-    {
-      "epoch": 174.5,
-      "learning_rate": 3.7324561403508774e-05,
-      "loss": 0.0109,
-      "step": 698
-    },
-    {
-      "epoch": 174.75,
-      "learning_rate": 3.7302631578947374e-05,
-      "loss": 0.0106,
-      "step": 699
-    },
-    {
-      "epoch": 175.0,
-      "learning_rate": 3.728070175438597e-05,
-      "loss": 0.0121,
-      "step": 700
-    },
-    {
-      "epoch": 175.0,
-      "eval_accuracy_ELSE": 0.9965370588808207,
-      "eval_accuracy_road": 0.9929942473087657,
-      "eval_accuracy_sidewalk": 0.9063644043931287,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9928945722726586,
-      "eval_iou_road": 0.9866440294284098,
-      "eval_iou_sidewalk": 0.8285493628523619,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020968396216630936,
-      "eval_mean_accuracy": 0.9652985701942384,
-      "eval_mean_iou": 0.9360293215178102,
-      "eval_overall_accuracy": 0.9931449890136719,
-      "eval_runtime": 0.6029,
-      "eval_samples_per_second": 1.659,
-      "eval_steps_per_second": 1.659,
-      "step": 700
-    },
-    {
-      "epoch": 175.25,
-      "learning_rate": 3.725877192982457e-05,
-      "loss": 0.0114,
-      "step": 701
-    },
-    {
-      "epoch": 175.5,
-      "learning_rate": 3.723684210526316e-05,
-      "loss": 0.012,
-      "step": 702
-    },
-    {
-      "epoch": 175.75,
-      "learning_rate": 3.721491228070176e-05,
-      "loss": 0.0105,
-      "step": 703
-    },
-    {
-      "epoch": 176.0,
-      "learning_rate": 3.719298245614035e-05,
-      "loss": 0.013,
-      "step": 704
-    },
-    {
-      "epoch": 176.25,
-      "learning_rate": 3.717105263157895e-05,
-      "loss": 0.0105,
-      "step": 705
-    },
-    {
-      "epoch": 176.5,
-      "learning_rate": 3.7149122807017545e-05,
-      "loss": 0.0105,
-      "step": 706
-    },
-    {
-      "epoch": 176.75,
-      "learning_rate": 3.7127192982456145e-05,
-      "loss": 0.013,
-      "step": 707
-    },
-    {
-      "epoch": 177.0,
-      "learning_rate": 3.710526315789474e-05,
-      "loss": 0.0123,
-      "step": 708
-    },
-    {
-      "epoch": 177.25,
-      "learning_rate": 3.708333333333334e-05,
-      "loss": 0.0129,
-      "step": 709
-    },
-    {
-      "epoch": 177.5,
-      "learning_rate": 3.706140350877193e-05,
-      "loss": 0.0119,
-      "step": 710
-    },
-    {
-      "epoch": 177.75,
-      "learning_rate": 3.703947368421053e-05,
-      "loss": 0.01,
-      "step": 711
-    },
-    {
-      "epoch": 178.0,
-      "learning_rate": 3.7017543859649124e-05,
-      "loss": 0.0109,
-      "step": 712
-    },
-    {
-      "epoch": 178.25,
-      "learning_rate": 3.6995614035087723e-05,
-      "loss": 0.0114,
-      "step": 713
-    },
-    {
-      "epoch": 178.5,
-      "learning_rate": 3.6973684210526316e-05,
-      "loss": 0.0115,
-      "step": 714
-    },
-    {
-      "epoch": 178.75,
-      "learning_rate": 3.6951754385964916e-05,
-      "loss": 0.0111,
-      "step": 715
-    },
-    {
-      "epoch": 179.0,
-      "learning_rate": 3.692982456140351e-05,
-      "loss": 0.0101,
-      "step": 716
-    },
-    {
-      "epoch": 179.25,
-      "learning_rate": 3.690789473684211e-05,
-      "loss": 0.0133,
-      "step": 717
-    },
-    {
-      "epoch": 179.5,
-      "learning_rate": 3.68859649122807e-05,
-      "loss": 0.0103,
-      "step": 718
-    },
-    {
-      "epoch": 179.75,
-      "learning_rate": 3.68640350877193e-05,
-      "loss": 0.0115,
-      "step": 719
-    },
-    {
-      "epoch": 180.0,
-      "learning_rate": 3.6842105263157895e-05,
-      "loss": 0.0123,
-      "step": 720
-    },
-    {
-      "epoch": 180.0,
-      "eval_accuracy_ELSE": 0.9957795405110003,
-      "eval_accuracy_road": 0.9942615481004727,
-      "eval_accuracy_sidewalk": 0.9069276260208392,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9928838486563476,
-      "eval_iou_road": 0.9861173878657779,
-      "eval_iou_sidewalk": 0.8229206592564201,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02133314125239849,
-      "eval_mean_accuracy": 0.9656562382107707,
-      "eval_mean_iou": 0.9339739652595153,
-      "eval_overall_accuracy": 0.9929656982421875,
-      "eval_runtime": 0.6336,
-      "eval_samples_per_second": 1.578,
-      "eval_steps_per_second": 1.578,
-      "step": 720
-    },
-    {
-      "epoch": 180.25,
-      "learning_rate": 3.6820175438596494e-05,
-      "loss": 0.0113,
-      "step": 721
-    },
-    {
-      "epoch": 180.5,
-      "learning_rate": 3.679824561403509e-05,
-      "loss": 0.0116,
-      "step": 722
-    },
-    {
-      "epoch": 180.75,
-      "learning_rate": 3.677631578947369e-05,
-      "loss": 0.0101,
-      "step": 723
-    },
-    {
-      "epoch": 181.0,
-      "learning_rate": 3.675438596491228e-05,
-      "loss": 0.0107,
-      "step": 724
-    },
-    {
-      "epoch": 181.25,
-      "learning_rate": 3.673245614035088e-05,
-      "loss": 0.0123,
-      "step": 725
-    },
-    {
-      "epoch": 181.5,
-      "learning_rate": 3.671052631578947e-05,
-      "loss": 0.011,
-      "step": 726
-    },
-    {
-      "epoch": 181.75,
-      "learning_rate": 3.668859649122807e-05,
-      "loss": 0.0101,
-      "step": 727
-    },
-    {
-      "epoch": 182.0,
-      "learning_rate": 3.6666666666666666e-05,
-      "loss": 0.0096,
-      "step": 728
-    },
-    {
-      "epoch": 182.25,
-      "learning_rate": 3.6644736842105265e-05,
-      "loss": 0.0106,
-      "step": 729
-    },
-    {
-      "epoch": 182.5,
-      "learning_rate": 3.662280701754386e-05,
-      "loss": 0.0108,
-      "step": 730
-    },
-    {
-      "epoch": 182.75,
-      "learning_rate": 3.660087719298246e-05,
-      "loss": 0.0129,
-      "step": 731
-    },
-    {
-      "epoch": 183.0,
-      "learning_rate": 3.657894736842106e-05,
-      "loss": 0.0095,
-      "step": 732
-    },
-    {
-      "epoch": 183.25,
-      "learning_rate": 3.655701754385965e-05,
-      "loss": 0.012,
-      "step": 733
-    },
-    {
-      "epoch": 183.5,
-      "learning_rate": 3.653508771929825e-05,
-      "loss": 0.0118,
-      "step": 734
-    },
-    {
-      "epoch": 183.75,
-      "learning_rate": 3.6513157894736844e-05,
-      "loss": 0.0112,
-      "step": 735
-    },
-    {
-      "epoch": 184.0,
-      "learning_rate": 3.6491228070175443e-05,
-      "loss": 0.0105,
-      "step": 736
-    },
-    {
-      "epoch": 184.25,
-      "learning_rate": 3.6469298245614036e-05,
-      "loss": 0.0107,
-      "step": 737
-    },
-    {
-      "epoch": 184.5,
-      "learning_rate": 3.6447368421052636e-05,
-      "loss": 0.0113,
-      "step": 738
-    },
-    {
-      "epoch": 184.75,
-      "learning_rate": 3.642543859649123e-05,
-      "loss": 0.0105,
-      "step": 739
-    },
-    {
-      "epoch": 185.0,
-      "learning_rate": 3.640350877192983e-05,
-      "loss": 0.0098,
-      "step": 740
-    },
-    {
-      "epoch": 185.0,
-      "eval_accuracy_ELSE": 0.9969374614477258,
-      "eval_accuracy_road": 0.993563820698297,
-      "eval_accuracy_sidewalk": 0.9025626584060827,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932506738544474,
-      "eval_iou_road": 0.987209960384833,
-      "eval_iou_sidewalk": 0.8352879854052645,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020464804023504257,
-      "eval_mean_accuracy": 0.9643546468507017,
-      "eval_mean_iou": 0.9385828732148483,
-      "eval_overall_accuracy": 0.9934768676757812,
-      "eval_runtime": 0.6167,
-      "eval_samples_per_second": 1.621,
-      "eval_steps_per_second": 1.621,
-      "step": 740
-    },
-    {
-      "epoch": 185.25,
-      "learning_rate": 3.638157894736842e-05,
-      "loss": 0.0115,
-      "step": 741
-    },
-    {
-      "epoch": 185.5,
-      "learning_rate": 3.635964912280702e-05,
-      "loss": 0.0102,
-      "step": 742
-    },
-    {
-      "epoch": 185.75,
-      "learning_rate": 3.6337719298245615e-05,
-      "loss": 0.0122,
-      "step": 743
-    },
-    {
-      "epoch": 186.0,
-      "learning_rate": 3.6315789473684214e-05,
-      "loss": 0.0104,
-      "step": 744
-    },
-    {
-      "epoch": 186.25,
-      "learning_rate": 3.629385964912281e-05,
-      "loss": 0.0104,
-      "step": 745
-    },
-    {
-      "epoch": 186.5,
-      "learning_rate": 3.627192982456141e-05,
-      "loss": 0.0099,
-      "step": 746
-    },
-    {
-      "epoch": 186.75,
-      "learning_rate": 3.625e-05,
-      "loss": 0.0111,
-      "step": 747
-    },
-    {
-      "epoch": 187.0,
-      "learning_rate": 3.62280701754386e-05,
-      "loss": 0.0121,
-      "step": 748
-    },
-    {
-      "epoch": 187.25,
-      "learning_rate": 3.620614035087719e-05,
-      "loss": 0.0111,
-      "step": 749
-    },
-    {
-      "epoch": 187.5,
-      "learning_rate": 3.618421052631579e-05,
-      "loss": 0.0116,
-      "step": 750
-    },
-    {
-      "epoch": 187.75,
-      "learning_rate": 3.6162280701754386e-05,
-      "loss": 0.0099,
-      "step": 751
-    },
-    {
-      "epoch": 188.0,
-      "learning_rate": 3.6140350877192985e-05,
-      "loss": 0.0096,
-      "step": 752
-    },
-    {
-      "epoch": 188.25,
-      "learning_rate": 3.611842105263158e-05,
-      "loss": 0.0097,
-      "step": 753
-    },
-    {
-      "epoch": 188.5,
-      "learning_rate": 3.609649122807018e-05,
-      "loss": 0.0102,
-      "step": 754
-    },
-    {
-      "epoch": 188.75,
-      "learning_rate": 3.607456140350877e-05,
-      "loss": 0.0108,
-      "step": 755
-    },
-    {
-      "epoch": 189.0,
-      "learning_rate": 3.605263157894737e-05,
-      "loss": 0.0104,
-      "step": 756
-    },
-    {
-      "epoch": 189.25,
-      "learning_rate": 3.6030701754385964e-05,
-      "loss": 0.0106,
-      "step": 757
-    },
-    {
-      "epoch": 189.5,
-      "learning_rate": 3.6008771929824564e-05,
-      "loss": 0.0095,
-      "step": 758
-    },
-    {
-      "epoch": 189.75,
-      "learning_rate": 3.598684210526316e-05,
-      "loss": 0.0111,
-      "step": 759
-    },
-    {
-      "epoch": 190.0,
-      "learning_rate": 3.5964912280701756e-05,
-      "loss": 0.0112,
-      "step": 760
-    },
-    {
-      "epoch": 190.0,
-      "eval_accuracy_ELSE": 0.9968617096107438,
-      "eval_accuracy_road": 0.9943042661046876,
-      "eval_accuracy_sidewalk": 0.8973528583497606,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931537864412627,
-      "eval_iou_road": 0.9874287653605215,
-      "eval_iou_sidewalk": 0.8350366876310272,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02040124125778675,
-      "eval_mean_accuracy": 0.962839611355064,
-      "eval_mean_iou": 0.9385397464776037,
-      "eval_overall_accuracy": 0.9934806823730469,
-      "eval_runtime": 0.6142,
-      "eval_samples_per_second": 1.628,
-      "eval_steps_per_second": 1.628,
-      "step": 760
-    },
-    {
-      "epoch": 190.25,
-      "learning_rate": 3.594298245614035e-05,
-      "loss": 0.0105,
-      "step": 761
-    },
-    {
-      "epoch": 190.5,
-      "learning_rate": 3.592105263157895e-05,
-      "loss": 0.0108,
-      "step": 762
-    },
-    {
-      "epoch": 190.75,
-      "learning_rate": 3.589912280701754e-05,
-      "loss": 0.0106,
-      "step": 763
-    },
-    {
-      "epoch": 191.0,
-      "learning_rate": 3.587719298245614e-05,
-      "loss": 0.0103,
-      "step": 764
-    },
-    {
-      "epoch": 191.25,
-      "learning_rate": 3.5855263157894735e-05,
-      "loss": 0.0104,
-      "step": 765
-    },
-    {
-      "epoch": 191.5,
-      "learning_rate": 3.5833333333333335e-05,
-      "loss": 0.0103,
-      "step": 766
-    },
-    {
-      "epoch": 191.75,
-      "learning_rate": 3.581140350877193e-05,
-      "loss": 0.0116,
-      "step": 767
-    },
-    {
-      "epoch": 192.0,
-      "learning_rate": 3.578947368421053e-05,
-      "loss": 0.0104,
-      "step": 768
-    },
-    {
-      "epoch": 192.25,
-      "learning_rate": 3.576754385964912e-05,
-      "loss": 0.0107,
-      "step": 769
-    },
-    {
-      "epoch": 192.5,
-      "learning_rate": 3.574561403508772e-05,
-      "loss": 0.0107,
-      "step": 770
-    },
-    {
-      "epoch": 192.75,
-      "learning_rate": 3.572368421052631e-05,
-      "loss": 0.0113,
-      "step": 771
-    },
-    {
-      "epoch": 193.0,
-      "learning_rate": 3.570175438596491e-05,
-      "loss": 0.0101,
-      "step": 772
-    },
-    {
-      "epoch": 193.25,
-      "learning_rate": 3.567982456140351e-05,
-      "loss": 0.0126,
-      "step": 773
-    },
-    {
-      "epoch": 193.5,
-      "learning_rate": 3.5657894736842106e-05,
-      "loss": 0.0108,
-      "step": 774
-    },
-    {
-      "epoch": 193.75,
-      "learning_rate": 3.5635964912280705e-05,
-      "loss": 0.0107,
-      "step": 775
-    },
-    {
-      "epoch": 194.0,
-      "learning_rate": 3.56140350877193e-05,
-      "loss": 0.0108,
-      "step": 776
-    },
-    {
-      "epoch": 194.25,
-      "learning_rate": 3.55921052631579e-05,
-      "loss": 0.0101,
-      "step": 777
-    },
-    {
-      "epoch": 194.5,
-      "learning_rate": 3.557017543859649e-05,
-      "loss": 0.0099,
-      "step": 778
-    },
-    {
-      "epoch": 194.75,
-      "learning_rate": 3.554824561403509e-05,
-      "loss": 0.0113,
-      "step": 779
-    },
-    {
-      "epoch": 195.0,
-      "learning_rate": 3.5526315789473684e-05,
-      "loss": 0.0104,
-      "step": 780
-    },
-    {
-      "epoch": 195.0,
-      "eval_accuracy_ELSE": 0.9963368575973681,
-      "eval_accuracy_road": 0.9941903514267814,
-      "eval_accuracy_sidewalk": 0.9021402421852999,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.992978823224888,
-      "eval_iou_road": 0.9865065347933593,
-      "eval_iou_sidewalk": 0.8309987029831388,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02063138037919998,
-      "eval_mean_accuracy": 0.9642224837364831,
-      "eval_mean_iou": 0.9368280203337953,
-      "eval_overall_accuracy": 0.9932098388671875,
-      "eval_runtime": 0.609,
-      "eval_samples_per_second": 1.642,
-      "eval_steps_per_second": 1.642,
-      "step": 780
-    },
-    {
-      "epoch": 195.25,
-      "learning_rate": 3.5504385964912284e-05,
-      "loss": 0.0107,
-      "step": 781
-    },
-    {
-      "epoch": 195.5,
-      "learning_rate": 3.548245614035088e-05,
-      "loss": 0.0106,
-      "step": 782
-    },
-    {
-      "epoch": 195.75,
-      "learning_rate": 3.5460526315789476e-05,
-      "loss": 0.0101,
-      "step": 783
-    },
-    {
-      "epoch": 196.0,
-      "learning_rate": 3.543859649122807e-05,
-      "loss": 0.0104,
-      "step": 784
-    },
-    {
-      "epoch": 196.25,
-      "learning_rate": 3.541666666666667e-05,
-      "loss": 0.0106,
-      "step": 785
-    },
-    {
-      "epoch": 196.5,
-      "learning_rate": 3.539473684210526e-05,
-      "loss": 0.0106,
-      "step": 786
-    },
-    {
-      "epoch": 196.75,
-      "learning_rate": 3.537280701754386e-05,
-      "loss": 0.0107,
-      "step": 787
-    },
-    {
-      "epoch": 197.0,
-      "learning_rate": 3.5350877192982455e-05,
-      "loss": 0.0102,
-      "step": 788
-    },
-    {
-      "epoch": 197.25,
-      "learning_rate": 3.5328947368421055e-05,
-      "loss": 0.0106,
-      "step": 789
-    },
-    {
-      "epoch": 197.5,
-      "learning_rate": 3.530701754385965e-05,
-      "loss": 0.0094,
-      "step": 790
-    },
-    {
-      "epoch": 197.75,
-      "learning_rate": 3.528508771929825e-05,
-      "loss": 0.0115,
-      "step": 791
-    },
-    {
-      "epoch": 198.0,
-      "learning_rate": 3.526315789473684e-05,
-      "loss": 0.011,
-      "step": 792
-    },
-    {
-      "epoch": 198.25,
-      "learning_rate": 3.524122807017544e-05,
-      "loss": 0.0109,
-      "step": 793
-    },
-    {
-      "epoch": 198.5,
-      "learning_rate": 3.521929824561403e-05,
-      "loss": 0.0106,
-      "step": 794
-    },
-    {
-      "epoch": 198.75,
-      "learning_rate": 3.519736842105263e-05,
-      "loss": 0.0099,
-      "step": 795
-    },
-    {
-      "epoch": 199.0,
-      "learning_rate": 3.5175438596491226e-05,
-      "loss": 0.0094,
-      "step": 796
-    },
-    {
-      "epoch": 199.25,
-      "learning_rate": 3.5153508771929826e-05,
-      "loss": 0.0098,
-      "step": 797
-    },
-    {
-      "epoch": 199.5,
-      "learning_rate": 3.513157894736842e-05,
-      "loss": 0.0102,
-      "step": 798
-    },
-    {
-      "epoch": 199.75,
-      "learning_rate": 3.510964912280702e-05,
-      "loss": 0.01,
-      "step": 799
-    },
-    {
-      "epoch": 200.0,
-      "learning_rate": 3.508771929824561e-05,
-      "loss": 0.0112,
-      "step": 800
-    },
-    {
-      "epoch": 200.0,
-      "eval_accuracy_ELSE": 0.9971484844221759,
-      "eval_accuracy_road": 0.9926240246055704,
-      "eval_accuracy_sidewalk": 0.904393128696142,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9930915185187181,
-      "eval_iou_road": 0.9867368748849916,
-      "eval_iou_sidewalk": 0.8393883951907998,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02015109732747078,
-      "eval_mean_accuracy": 0.964721879241296,
-      "eval_mean_iou": 0.9397389295315032,
-      "eval_overall_accuracy": 0.9934234619140625,
-      "eval_runtime": 0.7225,
-      "eval_samples_per_second": 1.384,
-      "eval_steps_per_second": 1.384,
-      "step": 800
-    },
-    {
-      "epoch": 200.25,
-      "learning_rate": 3.506578947368421e-05,
-      "loss": 0.0097,
-      "step": 801
-    },
-    {
-      "epoch": 200.5,
-      "learning_rate": 3.5043859649122804e-05,
-      "loss": 0.0102,
-      "step": 802
-    },
-    {
-      "epoch": 200.75,
-      "learning_rate": 3.5021929824561404e-05,
-      "loss": 0.0107,
-      "step": 803
-    },
-    {
-      "epoch": 201.0,
-      "learning_rate": 3.5e-05,
-      "loss": 0.0111,
-      "step": 804
-    },
-    {
-      "epoch": 201.25,
-      "learning_rate": 3.49780701754386e-05,
-      "loss": 0.0109,
-      "step": 805
-    },
-    {
-      "epoch": 201.5,
-      "learning_rate": 3.495614035087719e-05,
-      "loss": 0.0093,
-      "step": 806
-    },
-    {
-      "epoch": 201.75,
-      "learning_rate": 3.493421052631579e-05,
-      "loss": 0.0098,
-      "step": 807
-    },
-    {
-      "epoch": 202.0,
-      "learning_rate": 3.491228070175438e-05,
-      "loss": 0.0117,
-      "step": 808
-    },
-    {
-      "epoch": 202.25,
-      "learning_rate": 3.489035087719298e-05,
-      "loss": 0.0103,
-      "step": 809
-    },
-    {
-      "epoch": 202.5,
-      "learning_rate": 3.4868421052631575e-05,
-      "loss": 0.0105,
-      "step": 810
-    },
-    {
-      "epoch": 202.75,
-      "learning_rate": 3.4846491228070175e-05,
-      "loss": 0.0099,
-      "step": 811
-    },
-    {
-      "epoch": 203.0,
-      "learning_rate": 3.482456140350877e-05,
-      "loss": 0.0109,
-      "step": 812
-    },
-    {
-      "epoch": 203.25,
-      "learning_rate": 3.480263157894737e-05,
-      "loss": 0.0094,
-      "step": 813
-    },
-    {
-      "epoch": 203.5,
-      "learning_rate": 3.478070175438597e-05,
-      "loss": 0.0097,
-      "step": 814
-    },
-    {
-      "epoch": 203.75,
-      "learning_rate": 3.475877192982456e-05,
-      "loss": 0.0108,
-      "step": 815
-    },
-    {
-      "epoch": 204.0,
-      "learning_rate": 3.473684210526316e-05,
-      "loss": 0.0097,
-      "step": 816
-    },
-    {
-      "epoch": 204.25,
-      "learning_rate": 3.471491228070175e-05,
-      "loss": 0.0098,
-      "step": 817
-    },
-    {
-      "epoch": 204.5,
-      "learning_rate": 3.469298245614035e-05,
-      "loss": 0.0093,
-      "step": 818
-    },
-    {
-      "epoch": 204.75,
-      "learning_rate": 3.4671052631578946e-05,
-      "loss": 0.0101,
-      "step": 819
-    },
-    {
-      "epoch": 205.0,
-      "learning_rate": 3.4649122807017546e-05,
-      "loss": 0.0109,
-      "step": 820
-    },
-    {
-      "epoch": 205.0,
-      "eval_accuracy_ELSE": 0.9966669191727899,
-      "eval_accuracy_road": 0.9937062140456798,
-      "eval_accuracy_sidewalk": 0.9098845395663193,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933077723671935,
-      "eval_iou_road": 0.9874352661516258,
-      "eval_iou_sidewalk": 0.8360719368611722,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02013850398361683,
-      "eval_mean_accuracy": 0.9667525575949297,
-      "eval_mean_iou": 0.9389383251266638,
-      "eval_overall_accuracy": 0.9935226440429688,
-      "eval_runtime": 0.6142,
-      "eval_samples_per_second": 1.628,
-      "eval_steps_per_second": 1.628,
-      "step": 820
-    },
-    {
-      "epoch": 205.25,
-      "learning_rate": 3.462719298245614e-05,
-      "loss": 0.0094,
-      "step": 821
-    },
-    {
-      "epoch": 205.5,
-      "learning_rate": 3.460526315789474e-05,
-      "loss": 0.0102,
-      "step": 822
-    },
-    {
-      "epoch": 205.75,
-      "learning_rate": 3.458333333333333e-05,
-      "loss": 0.0109,
-      "step": 823
-    },
-    {
-      "epoch": 206.0,
-      "learning_rate": 3.456140350877193e-05,
-      "loss": 0.0101,
-      "step": 824
-    },
-    {
-      "epoch": 206.25,
-      "learning_rate": 3.4539473684210524e-05,
-      "loss": 0.0122,
-      "step": 825
-    },
-    {
-      "epoch": 206.5,
-      "learning_rate": 3.4517543859649124e-05,
-      "loss": 0.0109,
-      "step": 826
-    },
-    {
-      "epoch": 206.75,
-      "learning_rate": 3.449561403508772e-05,
-      "loss": 0.0099,
-      "step": 827
-    },
-    {
-      "epoch": 207.0,
-      "learning_rate": 3.447368421052632e-05,
-      "loss": 0.0099,
-      "step": 828
-    },
-    {
-      "epoch": 207.25,
-      "learning_rate": 3.445175438596491e-05,
-      "loss": 0.0104,
-      "step": 829
-    },
-    {
-      "epoch": 207.5,
-      "learning_rate": 3.442982456140351e-05,
-      "loss": 0.0103,
-      "step": 830
-    },
-    {
-      "epoch": 207.75,
-      "learning_rate": 3.44078947368421e-05,
-      "loss": 0.0093,
-      "step": 831
-    },
-    {
-      "epoch": 208.0,
-      "learning_rate": 3.43859649122807e-05,
-      "loss": 0.0107,
-      "step": 832
-    },
-    {
-      "epoch": 208.25,
-      "learning_rate": 3.4364035087719295e-05,
-      "loss": 0.0111,
-      "step": 833
-    },
-    {
-      "epoch": 208.5,
-      "learning_rate": 3.4342105263157895e-05,
-      "loss": 0.0107,
-      "step": 834
-    },
-    {
-      "epoch": 208.75,
-      "learning_rate": 3.4320175438596495e-05,
-      "loss": 0.0111,
-      "step": 835
-    },
-    {
-      "epoch": 209.0,
-      "learning_rate": 3.429824561403509e-05,
-      "loss": 0.0088,
-      "step": 836
-    },
-    {
-      "epoch": 209.25,
-      "learning_rate": 3.427631578947369e-05,
-      "loss": 0.0104,
-      "step": 837
-    },
-    {
-      "epoch": 209.5,
-      "learning_rate": 3.425438596491228e-05,
-      "loss": 0.0113,
-      "step": 838
-    },
-    {
-      "epoch": 209.75,
-      "learning_rate": 3.423245614035088e-05,
-      "loss": 0.0108,
-      "step": 839
-    },
-    {
-      "epoch": 210.0,
-      "learning_rate": 3.421052631578947e-05,
-      "loss": 0.0104,
-      "step": 840
-    },
-    {
-      "epoch": 210.0,
-      "eval_accuracy_ELSE": 0.9968130120012553,
-      "eval_accuracy_road": 0.9936207780372501,
-      "eval_accuracy_sidewalk": 0.9041115178822867,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932123525479286,
-      "eval_iou_road": 0.9867360502276649,
-      "eval_iou_sidewalk": 0.8382506527415143,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020421767607331276,
-      "eval_mean_accuracy": 0.9648484359735973,
-      "eval_mean_iou": 0.9393996851723694,
-      "eval_overall_accuracy": 0.9934463500976562,
-      "eval_runtime": 0.6941,
-      "eval_samples_per_second": 1.441,
-      "eval_steps_per_second": 1.441,
-      "step": 840
-    },
-    {
-      "epoch": 210.25,
-      "learning_rate": 3.418859649122807e-05,
-      "loss": 0.01,
-      "step": 841
-    },
-    {
-      "epoch": 210.5,
-      "learning_rate": 3.4166666666666666e-05,
-      "loss": 0.0097,
-      "step": 842
-    },
-    {
-      "epoch": 210.75,
-      "learning_rate": 3.4144736842105266e-05,
-      "loss": 0.0117,
-      "step": 843
-    },
-    {
-      "epoch": 211.0,
-      "learning_rate": 3.412280701754386e-05,
-      "loss": 0.0105,
-      "step": 844
-    },
-    {
-      "epoch": 211.25,
-      "learning_rate": 3.410087719298246e-05,
-      "loss": 0.011,
-      "step": 845
-    },
-    {
-      "epoch": 211.5,
-      "learning_rate": 3.407894736842105e-05,
-      "loss": 0.011,
-      "step": 846
-    },
-    {
-      "epoch": 211.75,
-      "learning_rate": 3.405701754385965e-05,
-      "loss": 0.0098,
-      "step": 847
-    },
-    {
-      "epoch": 212.0,
-      "learning_rate": 3.403508771929825e-05,
-      "loss": 0.0084,
-      "step": 848
-    },
-    {
-      "epoch": 212.25,
-      "learning_rate": 3.4013157894736844e-05,
-      "loss": 0.0112,
-      "step": 849
-    },
-    {
-      "epoch": 212.5,
-      "learning_rate": 3.3991228070175444e-05,
-      "loss": 0.0103,
-      "step": 850
-    },
-    {
-      "epoch": 212.75,
-      "learning_rate": 3.396929824561404e-05,
-      "loss": 0.0108,
-      "step": 851
-    },
-    {
-      "epoch": 213.0,
-      "learning_rate": 3.3947368421052636e-05,
-      "loss": 0.0089,
-      "step": 852
-    },
-    {
-      "epoch": 213.25,
-      "learning_rate": 3.392543859649123e-05,
-      "loss": 0.0101,
-      "step": 853
-    },
-    {
-      "epoch": 213.5,
-      "learning_rate": 3.390350877192983e-05,
-      "loss": 0.0101,
-      "step": 854
-    },
-    {
-      "epoch": 213.75,
-      "learning_rate": 3.388157894736842e-05,
-      "loss": 0.0104,
-      "step": 855
-    },
-    {
-      "epoch": 214.0,
-      "learning_rate": 3.385964912280702e-05,
-      "loss": 0.01,
-      "step": 856
-    },
-    {
-      "epoch": 214.25,
-      "learning_rate": 3.3837719298245615e-05,
-      "loss": 0.0096,
-      "step": 857
-    },
-    {
-      "epoch": 214.5,
-      "learning_rate": 3.3815789473684215e-05,
-      "loss": 0.01,
-      "step": 858
-    },
-    {
-      "epoch": 214.75,
-      "learning_rate": 3.379385964912281e-05,
-      "loss": 0.0094,
-      "step": 859
-    },
-    {
-      "epoch": 215.0,
-      "learning_rate": 3.377192982456141e-05,
-      "loss": 0.011,
-      "step": 860
-    },
-    {
-      "epoch": 215.0,
-      "eval_accuracy_ELSE": 0.996704795091281,
-      "eval_accuracy_road": 0.9945890527994532,
-      "eval_accuracy_sidewalk": 0.900591382709096,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933669479871653,
-      "eval_iou_road": 0.9870695137289262,
-      "eval_iou_sidewalk": 0.8370632116215155,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02035897597670555,
-      "eval_mean_accuracy": 0.9639617435332767,
-      "eval_mean_iou": 0.9391665577792022,
-      "eval_overall_accuracy": 0.9935340881347656,
-      "eval_runtime": 0.5991,
-      "eval_samples_per_second": 1.669,
-      "eval_steps_per_second": 1.669,
-      "step": 860
-    },
-    {
-      "epoch": 215.25,
-      "learning_rate": 3.375000000000001e-05,
-      "loss": 0.0095,
-      "step": 861
-    },
-    {
-      "epoch": 215.5,
-      "learning_rate": 3.37280701754386e-05,
-      "loss": 0.0105,
-      "step": 862
-    },
-    {
-      "epoch": 215.75,
-      "learning_rate": 3.37061403508772e-05,
-      "loss": 0.0096,
-      "step": 863
-    },
-    {
-      "epoch": 216.0,
-      "learning_rate": 3.368421052631579e-05,
-      "loss": 0.01,
-      "step": 864
-    },
-    {
-      "epoch": 216.25,
-      "learning_rate": 3.366228070175439e-05,
-      "loss": 0.0109,
-      "step": 865
-    },
-    {
-      "epoch": 216.5,
-      "learning_rate": 3.3640350877192986e-05,
-      "loss": 0.0113,
-      "step": 866
-    },
-    {
-      "epoch": 216.75,
-      "learning_rate": 3.3618421052631585e-05,
-      "loss": 0.0095,
-      "step": 867
-    },
-    {
-      "epoch": 217.0,
-      "learning_rate": 3.359649122807018e-05,
-      "loss": 0.0085,
-      "step": 868
-    },
-    {
-      "epoch": 217.25,
-      "learning_rate": 3.357456140350878e-05,
-      "loss": 0.0104,
-      "step": 869
-    },
-    {
-      "epoch": 217.5,
-      "learning_rate": 3.355263157894737e-05,
-      "loss": 0.0109,
-      "step": 870
-    },
-    {
-      "epoch": 217.75,
-      "learning_rate": 3.353070175438597e-05,
-      "loss": 0.0101,
-      "step": 871
-    },
-    {
-      "epoch": 218.0,
-      "learning_rate": 3.3508771929824564e-05,
-      "loss": 0.0092,
-      "step": 872
-    },
-    {
-      "epoch": 218.25,
-      "learning_rate": 3.3486842105263164e-05,
-      "loss": 0.0099,
-      "step": 873
-    },
-    {
-      "epoch": 218.5,
-      "learning_rate": 3.346491228070176e-05,
-      "loss": 0.0102,
-      "step": 874
-    },
-    {
-      "epoch": 218.75,
-      "learning_rate": 3.3442982456140356e-05,
-      "loss": 0.0106,
-      "step": 875
-    },
-    {
-      "epoch": 219.0,
-      "learning_rate": 3.342105263157895e-05,
-      "loss": 0.0108,
-      "step": 876
-    },
-    {
-      "epoch": 219.25,
-      "learning_rate": 3.339912280701755e-05,
-      "loss": 0.0105,
-      "step": 877
-    },
-    {
-      "epoch": 219.5,
-      "learning_rate": 3.337719298245614e-05,
-      "loss": 0.0095,
-      "step": 878
-    },
-    {
-      "epoch": 219.75,
-      "learning_rate": 3.335526315789474e-05,
-      "loss": 0.0099,
-      "step": 879
-    },
-    {
-      "epoch": 220.0,
-      "learning_rate": 3.3333333333333335e-05,
-      "loss": 0.0103,
-      "step": 880
-    },
-    {
-      "epoch": 220.0,
-      "eval_accuracy_ELSE": 0.9970078024392092,
-      "eval_accuracy_road": 0.992851853961383,
-      "eval_accuracy_sidewalk": 0.9027034638130104,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9929942175349346,
-      "eval_iou_road": 0.9866979877168652,
-      "eval_iou_sidewalk": 0.8366175127234764,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020980387926101685,
-      "eval_mean_accuracy": 0.9641877067378676,
-      "eval_mean_iou": 0.9387699059917587,
-      "eval_overall_accuracy": 0.9933395385742188,
-      "eval_runtime": 0.6067,
-      "eval_samples_per_second": 1.648,
-      "eval_steps_per_second": 1.648,
-      "step": 880
-    },
-    {
-      "epoch": 220.25,
-      "learning_rate": 3.3311403508771935e-05,
-      "loss": 0.01,
-      "step": 881
-    },
-    {
-      "epoch": 220.5,
-      "learning_rate": 3.328947368421053e-05,
-      "loss": 0.0094,
-      "step": 882
-    },
-    {
-      "epoch": 220.75,
-      "learning_rate": 3.326754385964913e-05,
-      "loss": 0.0097,
-      "step": 883
-    },
-    {
-      "epoch": 221.0,
-      "learning_rate": 3.324561403508772e-05,
-      "loss": 0.0103,
-      "step": 884
-    },
-    {
-      "epoch": 221.25,
-      "learning_rate": 3.322368421052632e-05,
-      "loss": 0.0096,
-      "step": 885
-    },
-    {
-      "epoch": 221.5,
-      "learning_rate": 3.320175438596491e-05,
-      "loss": 0.0095,
-      "step": 886
-    },
-    {
-      "epoch": 221.75,
-      "learning_rate": 3.317982456140351e-05,
-      "loss": 0.0097,
-      "step": 887
-    },
-    {
-      "epoch": 222.0,
-      "learning_rate": 3.3157894736842106e-05,
-      "loss": 0.0107,
-      "step": 888
-    },
-    {
-      "epoch": 222.25,
-      "learning_rate": 3.3135964912280706e-05,
-      "loss": 0.01,
-      "step": 889
-    },
-    {
-      "epoch": 222.5,
-      "learning_rate": 3.31140350877193e-05,
-      "loss": 0.0107,
-      "step": 890
-    },
-    {
-      "epoch": 222.75,
-      "learning_rate": 3.30921052631579e-05,
-      "loss": 0.0096,
-      "step": 891
-    },
-    {
-      "epoch": 223.0,
-      "learning_rate": 3.307017543859649e-05,
-      "loss": 0.0103,
-      "step": 892
-    },
-    {
-      "epoch": 223.25,
-      "learning_rate": 3.304824561403509e-05,
-      "loss": 0.0095,
-      "step": 893
-    },
-    {
-      "epoch": 223.5,
-      "learning_rate": 3.302631578947369e-05,
-      "loss": 0.009,
-      "step": 894
-    },
-    {
-      "epoch": 223.75,
-      "learning_rate": 3.3004385964912284e-05,
-      "loss": 0.0102,
-      "step": 895
-    },
-    {
-      "epoch": 224.0,
-      "learning_rate": 3.2982456140350884e-05,
-      "loss": 0.0101,
-      "step": 896
-    },
-    {
-      "epoch": 224.25,
-      "learning_rate": 3.296052631578948e-05,
-      "loss": 0.0107,
-      "step": 897
-    },
-    {
-      "epoch": 224.5,
-      "learning_rate": 3.2938596491228076e-05,
-      "loss": 0.0091,
-      "step": 898
-    },
-    {
-      "epoch": 224.75,
-      "learning_rate": 3.291666666666667e-05,
-      "loss": 0.0097,
-      "step": 899
-    },
-    {
-      "epoch": 225.0,
-      "learning_rate": 3.289473684210527e-05,
-      "loss": 0.0091,
-      "step": 900
-    },
-    {
-      "epoch": 225.0,
-      "eval_accuracy_ELSE": 0.9966560974817925,
-      "eval_accuracy_road": 0.9926667426097853,
-      "eval_accuracy_sidewalk": 0.904393128696142,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9926867256255625,
-      "eval_iou_road": 0.9865419449790558,
-      "eval_iou_sidewalk": 0.8299521902054529,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022020306438207626,
-      "eval_mean_accuracy": 0.9645719895959065,
-      "eval_mean_iou": 0.9363936202700236,
-      "eval_overall_accuracy": 0.9930877685546875,
-      "eval_runtime": 0.6031,
-      "eval_samples_per_second": 1.658,
-      "eval_steps_per_second": 1.658,
-      "step": 900
-    },
-    {
-      "epoch": 225.25,
-      "learning_rate": 3.287280701754386e-05,
-      "loss": 0.0097,
-      "step": 901
-    },
-    {
-      "epoch": 225.5,
-      "learning_rate": 3.285087719298246e-05,
-      "loss": 0.0097,
-      "step": 902
-    },
-    {
-      "epoch": 225.75,
-      "learning_rate": 3.2828947368421055e-05,
-      "loss": 0.01,
-      "step": 903
-    },
-    {
-      "epoch": 226.0,
-      "learning_rate": 3.2807017543859655e-05,
-      "loss": 0.0098,
-      "step": 904
-    },
-    {
-      "epoch": 226.25,
-      "learning_rate": 3.278508771929825e-05,
-      "loss": 0.0103,
-      "step": 905
-    },
-    {
-      "epoch": 226.5,
-      "learning_rate": 3.276315789473685e-05,
-      "loss": 0.0089,
-      "step": 906
-    },
-    {
-      "epoch": 226.75,
-      "learning_rate": 3.274122807017544e-05,
-      "loss": 0.0091,
-      "step": 907
-    },
-    {
-      "epoch": 227.0,
-      "learning_rate": 3.271929824561404e-05,
-      "loss": 0.0113,
-      "step": 908
-    },
-    {
-      "epoch": 227.25,
-      "learning_rate": 3.269736842105263e-05,
-      "loss": 0.0091,
-      "step": 909
-    },
-    {
-      "epoch": 227.5,
-      "learning_rate": 3.267543859649123e-05,
-      "loss": 0.0095,
-      "step": 910
-    },
-    {
-      "epoch": 227.75,
-      "learning_rate": 3.2653508771929826e-05,
-      "loss": 0.0119,
-      "step": 911
-    },
-    {
-      "epoch": 228.0,
-      "learning_rate": 3.2631578947368426e-05,
-      "loss": 0.0091,
-      "step": 912
-    },
-    {
-      "epoch": 228.25,
-      "learning_rate": 3.260964912280702e-05,
-      "loss": 0.0104,
-      "step": 913
-    },
-    {
-      "epoch": 228.5,
-      "learning_rate": 3.258771929824562e-05,
-      "loss": 0.0094,
-      "step": 914
-    },
-    {
-      "epoch": 228.75,
-      "learning_rate": 3.256578947368421e-05,
-      "loss": 0.0092,
-      "step": 915
-    },
-    {
-      "epoch": 229.0,
-      "learning_rate": 3.254385964912281e-05,
-      "loss": 0.0094,
-      "step": 916
-    },
-    {
-      "epoch": 229.25,
-      "learning_rate": 3.2521929824561404e-05,
-      "loss": 0.009,
-      "step": 917
-    },
-    {
-      "epoch": 229.5,
-      "learning_rate": 3.2500000000000004e-05,
-      "loss": 0.0099,
-      "step": 918
-    },
-    {
-      "epoch": 229.75,
-      "learning_rate": 3.24780701754386e-05,
-      "loss": 0.01,
-      "step": 919
-    },
-    {
-      "epoch": 230.0,
-      "learning_rate": 3.24561403508772e-05,
-      "loss": 0.0091,
-      "step": 920
-    },
-    {
-      "epoch": 230.0,
-      "eval_accuracy_ELSE": 0.9971376627311783,
-      "eval_accuracy_road": 0.9940052400751836,
-      "eval_accuracy_sidewalk": 0.9058011827654182,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936590442195394,
-      "eval_iou_road": 0.9874389985147465,
-      "eval_iou_sidewalk": 0.845668463257526,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.019371982663869858,
-      "eval_mean_accuracy": 0.9656480285239267,
-      "eval_mean_iou": 0.9422555019972707,
-      "eval_overall_accuracy": 0.9938240051269531,
-      "eval_runtime": 0.6116,
-      "eval_samples_per_second": 1.635,
-      "eval_steps_per_second": 1.635,
-      "step": 920
-    },
-    {
-      "epoch": 230.25,
-      "learning_rate": 3.243421052631579e-05,
-      "loss": 0.009,
-      "step": 921
-    },
-    {
-      "epoch": 230.5,
-      "learning_rate": 3.241228070175439e-05,
-      "loss": 0.0103,
-      "step": 922
-    },
-    {
-      "epoch": 230.75,
-      "learning_rate": 3.239035087719298e-05,
-      "loss": 0.0094,
-      "step": 923
-    },
-    {
-      "epoch": 231.0,
-      "learning_rate": 3.236842105263158e-05,
-      "loss": 0.0098,
-      "step": 924
-    },
-    {
-      "epoch": 231.25,
-      "learning_rate": 3.2346491228070175e-05,
-      "loss": 0.0102,
-      "step": 925
-    },
-    {
-      "epoch": 231.5,
-      "learning_rate": 3.2324561403508775e-05,
-      "loss": 0.0097,
-      "step": 926
-    },
-    {
-      "epoch": 231.75,
-      "learning_rate": 3.230263157894737e-05,
-      "loss": 0.0097,
-      "step": 927
-    },
-    {
-      "epoch": 232.0,
-      "learning_rate": 3.228070175438597e-05,
-      "loss": 0.0086,
-      "step": 928
-    },
-    {
-      "epoch": 232.25,
-      "learning_rate": 3.225877192982456e-05,
-      "loss": 0.0099,
-      "step": 929
-    },
-    {
-      "epoch": 232.5,
-      "learning_rate": 3.223684210526316e-05,
-      "loss": 0.0095,
-      "step": 930
-    },
-    {
-      "epoch": 232.75,
-      "learning_rate": 3.2214912280701753e-05,
-      "loss": 0.0094,
-      "step": 931
-    },
-    {
-      "epoch": 233.0,
-      "learning_rate": 3.219298245614035e-05,
-      "loss": 0.0099,
-      "step": 932
-    },
-    {
-      "epoch": 233.25,
-      "learning_rate": 3.2171052631578946e-05,
-      "loss": 0.0095,
-      "step": 933
-    },
-    {
-      "epoch": 233.5,
-      "learning_rate": 3.2149122807017546e-05,
-      "loss": 0.0087,
-      "step": 934
-    },
-    {
-      "epoch": 233.75,
-      "learning_rate": 3.2127192982456146e-05,
-      "loss": 0.0098,
-      "step": 935
-    },
-    {
-      "epoch": 234.0,
-      "learning_rate": 3.210526315789474e-05,
-      "loss": 0.0102,
-      "step": 936
-    },
-    {
-      "epoch": 234.25,
-      "learning_rate": 3.208333333333334e-05,
-      "loss": 0.0089,
-      "step": 937
-    },
-    {
-      "epoch": 234.5,
-      "learning_rate": 3.206140350877193e-05,
-      "loss": 0.0097,
-      "step": 938
-    },
-    {
-      "epoch": 234.75,
-      "learning_rate": 3.203947368421053e-05,
-      "loss": 0.0101,
-      "step": 939
-    },
-    {
-      "epoch": 235.0,
-      "learning_rate": 3.2017543859649124e-05,
-      "loss": 0.0089,
-      "step": 940
-    },
-    {
-      "epoch": 235.0,
-      "eval_accuracy_ELSE": 0.9966344540997977,
-      "eval_accuracy_road": 0.993663496041465,
-      "eval_accuracy_sidewalk": 0.9034074908476486,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9930290859099119,
-      "eval_iou_road": 0.986806380451383,
-      "eval_iou_sidewalk": 0.8336798336798337,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021239805966615677,
-      "eval_mean_accuracy": 0.9645684803296372,
-      "eval_mean_iou": 0.9378384333470429,
-      "eval_overall_accuracy": 0.9933128356933594,
-      "eval_runtime": 0.6263,
-      "eval_samples_per_second": 1.597,
-      "eval_steps_per_second": 1.597,
-      "step": 940
-    },
-    {
-      "epoch": 235.25,
-      "learning_rate": 3.1995614035087724e-05,
-      "loss": 0.01,
-      "step": 941
-    },
-    {
-      "epoch": 235.5,
-      "learning_rate": 3.197368421052632e-05,
-      "loss": 0.0099,
-      "step": 942
-    },
-    {
-      "epoch": 235.75,
-      "learning_rate": 3.195175438596492e-05,
-      "loss": 0.0088,
-      "step": 943
-    },
-    {
-      "epoch": 236.0,
-      "learning_rate": 3.192982456140351e-05,
-      "loss": 0.0089,
-      "step": 944
-    },
-    {
-      "epoch": 236.25,
-      "learning_rate": 3.190789473684211e-05,
-      "loss": 0.0098,
-      "step": 945
-    },
-    {
-      "epoch": 236.5,
-      "learning_rate": 3.18859649122807e-05,
-      "loss": 0.009,
-      "step": 946
-    },
-    {
-      "epoch": 236.75,
-      "learning_rate": 3.18640350877193e-05,
-      "loss": 0.0094,
-      "step": 947
-    },
-    {
-      "epoch": 237.0,
-      "learning_rate": 3.1842105263157895e-05,
-      "loss": 0.009,
-      "step": 948
-    },
-    {
-      "epoch": 237.25,
-      "learning_rate": 3.1820175438596495e-05,
-      "loss": 0.0083,
-      "step": 949
-    },
-    {
-      "epoch": 237.5,
-      "learning_rate": 3.179824561403509e-05,
-      "loss": 0.0101,
-      "step": 950
-    },
-    {
-      "epoch": 237.75,
-      "learning_rate": 3.177631578947369e-05,
-      "loss": 0.0108,
-      "step": 951
-    },
-    {
-      "epoch": 238.0,
-      "learning_rate": 3.175438596491228e-05,
-      "loss": 0.0087,
-      "step": 952
-    },
-    {
-      "epoch": 238.25,
-      "learning_rate": 3.173245614035088e-05,
-      "loss": 0.0093,
-      "step": 953
-    },
-    {
-      "epoch": 238.5,
-      "learning_rate": 3.1710526315789473e-05,
-      "loss": 0.0099,
-      "step": 954
-    },
-    {
-      "epoch": 238.75,
-      "learning_rate": 3.168859649122807e-05,
-      "loss": 0.0097,
-      "step": 955
-    },
-    {
-      "epoch": 239.0,
-      "learning_rate": 3.1666666666666666e-05,
-      "loss": 0.0109,
-      "step": 956
-    },
-    {
-      "epoch": 239.25,
-      "learning_rate": 3.1644736842105266e-05,
-      "loss": 0.0108,
-      "step": 957
-    },
-    {
-      "epoch": 239.5,
-      "learning_rate": 3.162280701754386e-05,
-      "loss": 0.0106,
-      "step": 958
-    },
-    {
-      "epoch": 239.75,
-      "learning_rate": 3.160087719298246e-05,
-      "loss": 0.0089,
-      "step": 959
-    },
-    {
-      "epoch": 240.0,
-      "learning_rate": 3.157894736842105e-05,
-      "loss": 0.008,
-      "step": 960
-    },
-    {
-      "epoch": 240.0,
-      "eval_accuracy_ELSE": 0.9970348566667028,
-      "eval_accuracy_road": 0.9932932733382697,
-      "eval_accuracy_sidewalk": 0.9090397071247536,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993604814183724,
-      "eval_iou_road": 0.9868992544176111,
-      "eval_iou_sidewalk": 0.8412822517591869,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020116329193115234,
-      "eval_mean_accuracy": 0.9664559457099086,
-      "eval_mean_iou": 0.940595440120174,
-      "eval_overall_accuracy": 0.9936485290527344,
-      "eval_runtime": 0.6098,
-      "eval_samples_per_second": 1.64,
-      "eval_steps_per_second": 1.64,
-      "step": 960
-    },
-    {
-      "epoch": 240.25,
-      "learning_rate": 3.155701754385965e-05,
-      "loss": 0.0093,
-      "step": 961
-    },
-    {
-      "epoch": 240.5,
-      "learning_rate": 3.1535087719298244e-05,
-      "loss": 0.0103,
-      "step": 962
-    },
-    {
-      "epoch": 240.75,
-      "learning_rate": 3.1513157894736844e-05,
-      "loss": 0.008,
-      "step": 963
-    },
-    {
-      "epoch": 241.0,
-      "learning_rate": 3.149122807017544e-05,
-      "loss": 0.0097,
-      "step": 964
-    },
-    {
-      "epoch": 241.25,
-      "learning_rate": 3.146929824561404e-05,
-      "loss": 0.0089,
-      "step": 965
-    },
-    {
-      "epoch": 241.5,
-      "learning_rate": 3.144736842105263e-05,
-      "loss": 0.0095,
-      "step": 966
-    },
-    {
-      "epoch": 241.75,
-      "learning_rate": 3.142543859649123e-05,
-      "loss": 0.0096,
-      "step": 967
-    },
-    {
-      "epoch": 242.0,
-      "learning_rate": 3.140350877192982e-05,
-      "loss": 0.0091,
-      "step": 968
-    },
-    {
-      "epoch": 242.25,
-      "learning_rate": 3.138157894736842e-05,
-      "loss": 0.0102,
-      "step": 969
-    },
-    {
-      "epoch": 242.5,
-      "learning_rate": 3.1359649122807015e-05,
-      "loss": 0.0097,
-      "step": 970
-    },
-    {
-      "epoch": 242.75,
-      "learning_rate": 3.1337719298245615e-05,
-      "loss": 0.0095,
-      "step": 971
-    },
-    {
-      "epoch": 243.0,
-      "learning_rate": 3.131578947368421e-05,
-      "loss": 0.0086,
-      "step": 972
-    },
-    {
-      "epoch": 243.25,
-      "learning_rate": 3.129385964912281e-05,
-      "loss": 0.0113,
-      "step": 973
-    },
-    {
-      "epoch": 243.5,
-      "learning_rate": 3.12719298245614e-05,
-      "loss": 0.0088,
-      "step": 974
-    },
-    {
-      "epoch": 243.75,
-      "learning_rate": 3.125e-05,
-      "loss": 0.0091,
-      "step": 975
-    },
-    {
-      "epoch": 244.0,
-      "learning_rate": 3.12280701754386e-05,
-      "loss": 0.0099,
-      "step": 976
-    },
-    {
-      "epoch": 244.25,
-      "learning_rate": 3.1206140350877193e-05,
-      "loss": 0.011,
-      "step": 977
-    },
-    {
-      "epoch": 244.5,
-      "learning_rate": 3.118421052631579e-05,
-      "loss": 0.0087,
-      "step": 978
-    },
-    {
-      "epoch": 244.75,
-      "learning_rate": 3.1162280701754386e-05,
-      "loss": 0.009,
-      "step": 979
-    },
-    {
-      "epoch": 245.0,
-      "learning_rate": 3.1140350877192986e-05,
-      "loss": 0.0094,
-      "step": 980
-    },
-    {
-      "epoch": 245.0,
-      "eval_accuracy_ELSE": 0.9966344540997977,
-      "eval_accuracy_road": 0.9923392379108048,
-      "eval_accuracy_sidewalk": 0.9124190368910167,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9928631492701439,
-      "eval_iou_road": 0.986747090306686,
-      "eval_iou_sidewalk": 0.8324768756423433,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021078480407595634,
-      "eval_mean_accuracy": 0.967130909633873,
-      "eval_mean_iou": 0.9373623717397243,
-      "eval_overall_accuracy": 0.9932022094726562,
-      "eval_runtime": 0.6524,
-      "eval_samples_per_second": 1.533,
-      "eval_steps_per_second": 1.533,
-      "step": 980
-    },
-    {
-      "epoch": 245.25,
-      "learning_rate": 3.111842105263158e-05,
-      "loss": 0.0093,
-      "step": 981
-    },
-    {
-      "epoch": 245.5,
-      "learning_rate": 3.109649122807018e-05,
-      "loss": 0.0099,
-      "step": 982
-    },
-    {
-      "epoch": 245.75,
-      "learning_rate": 3.107456140350877e-05,
-      "loss": 0.0091,
-      "step": 983
-    },
-    {
-      "epoch": 246.0,
-      "learning_rate": 3.105263157894737e-05,
-      "loss": 0.0086,
-      "step": 984
-    },
-    {
-      "epoch": 246.25,
-      "learning_rate": 3.1030701754385964e-05,
-      "loss": 0.0095,
-      "step": 985
-    },
-    {
-      "epoch": 246.5,
-      "learning_rate": 3.1008771929824564e-05,
-      "loss": 0.0097,
-      "step": 986
-    },
-    {
-      "epoch": 246.75,
-      "learning_rate": 3.098684210526316e-05,
-      "loss": 0.0099,
-      "step": 987
-    },
-    {
-      "epoch": 247.0,
-      "learning_rate": 3.096491228070176e-05,
-      "loss": 0.0093,
-      "step": 988
-    },
-    {
-      "epoch": 247.25,
-      "learning_rate": 3.094298245614035e-05,
-      "loss": 0.009,
-      "step": 989
-    },
-    {
-      "epoch": 247.5,
-      "learning_rate": 3.092105263157895e-05,
-      "loss": 0.0089,
-      "step": 990
-    },
-    {
-      "epoch": 247.75,
-      "learning_rate": 3.089912280701754e-05,
-      "loss": 0.0097,
-      "step": 991
-    },
-    {
-      "epoch": 248.0,
-      "learning_rate": 3.087719298245614e-05,
-      "loss": 0.0108,
-      "step": 992
-    },
-    {
-      "epoch": 248.25,
-      "learning_rate": 3.0855263157894735e-05,
-      "loss": 0.008,
-      "step": 993
-    },
-    {
-      "epoch": 248.5,
-      "learning_rate": 3.0833333333333335e-05,
-      "loss": 0.01,
-      "step": 994
-    },
-    {
-      "epoch": 248.75,
-      "learning_rate": 3.081140350877193e-05,
-      "loss": 0.0094,
-      "step": 995
-    },
-    {
-      "epoch": 249.0,
-      "learning_rate": 3.078947368421053e-05,
-      "loss": 0.0095,
-      "step": 996
-    },
-    {
-      "epoch": 249.25,
-      "learning_rate": 3.076754385964912e-05,
-      "loss": 0.009,
-      "step": 997
-    },
-    {
-      "epoch": 249.5,
-      "learning_rate": 3.074561403508772e-05,
-      "loss": 0.0087,
-      "step": 998
-    },
-    {
-      "epoch": 249.75,
-      "learning_rate": 3.0723684210526314e-05,
-      "loss": 0.0113,
-      "step": 999
-    },
-    {
-      "epoch": 250.0,
-      "learning_rate": 3.0701754385964913e-05,
-      "loss": 0.0106,
-      "step": 1000
-    },
-    {
-      "epoch": 250.0,
-      "eval_accuracy_ELSE": 0.9969212289112297,
-      "eval_accuracy_road": 0.9921398872244689,
-      "eval_accuracy_sidewalk": 0.9117150098563784,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931488389141637,
-      "eval_iou_road": 0.9864091964437397,
-      "eval_iou_sidewalk": 0.83656330749354,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020775219425559044,
-      "eval_mean_accuracy": 0.9669253753306922,
-      "eval_mean_iou": 0.9387071142838145,
-      "eval_overall_accuracy": 0.9933319091796875,
-      "eval_runtime": 0.6387,
-      "eval_samples_per_second": 1.566,
-      "eval_steps_per_second": 1.566,
-      "step": 1000
-    },
-    {
-      "epoch": 250.25,
-      "learning_rate": 3.0679824561403506e-05,
-      "loss": 0.0095,
-      "step": 1001
-    },
-    {
-      "epoch": 250.5,
-      "learning_rate": 3.0657894736842106e-05,
-      "loss": 0.0095,
-      "step": 1002
-    },
-    {
-      "epoch": 250.75,
-      "learning_rate": 3.06359649122807e-05,
-      "loss": 0.0101,
-      "step": 1003
-    },
-    {
-      "epoch": 251.0,
-      "learning_rate": 3.06140350877193e-05,
-      "loss": 0.0097,
-      "step": 1004
-    },
-    {
-      "epoch": 251.25,
-      "learning_rate": 3.059210526315789e-05,
-      "loss": 0.0145,
-      "step": 1005
-    },
-    {
-      "epoch": 251.5,
-      "learning_rate": 3.057017543859649e-05,
-      "loss": 0.0094,
-      "step": 1006
-    },
-    {
-      "epoch": 251.75,
-      "learning_rate": 3.0548245614035085e-05,
-      "loss": 0.0103,
-      "step": 1007
-    },
-    {
-      "epoch": 252.0,
-      "learning_rate": 3.0526315789473684e-05,
-      "loss": 0.0086,
-      "step": 1008
-    },
-    {
-      "epoch": 252.25,
-      "learning_rate": 3.050438596491228e-05,
-      "loss": 0.0098,
-      "step": 1009
-    },
-    {
-      "epoch": 252.5,
-      "learning_rate": 3.048245614035088e-05,
-      "loss": 0.0107,
-      "step": 1010
-    },
-    {
-      "epoch": 252.75,
-      "learning_rate": 3.0460526315789474e-05,
-      "loss": 0.0104,
-      "step": 1011
-    },
-    {
-      "epoch": 253.0,
-      "learning_rate": 3.0438596491228073e-05,
-      "loss": 0.0084,
-      "step": 1012
-    },
-    {
-      "epoch": 253.25,
-      "learning_rate": 3.0416666666666666e-05,
-      "loss": 0.0098,
-      "step": 1013
-    },
-    {
-      "epoch": 253.5,
-      "learning_rate": 3.0394736842105266e-05,
-      "loss": 0.01,
-      "step": 1014
-    },
-    {
-      "epoch": 253.75,
-      "learning_rate": 3.037280701754386e-05,
-      "loss": 0.0096,
-      "step": 1015
-    },
-    {
-      "epoch": 254.0,
-      "learning_rate": 3.035087719298246e-05,
-      "loss": 0.0092,
-      "step": 1016
-    },
-    {
-      "epoch": 254.25,
-      "learning_rate": 3.0328947368421055e-05,
-      "loss": 0.0094,
-      "step": 1017
-    },
-    {
-      "epoch": 254.5,
-      "learning_rate": 3.030701754385965e-05,
-      "loss": 0.0097,
-      "step": 1018
-    },
-    {
-      "epoch": 254.75,
-      "learning_rate": 3.028508771929825e-05,
-      "loss": 0.0116,
-      "step": 1019
-    },
-    {
-      "epoch": 255.0,
-      "learning_rate": 3.0263157894736844e-05,
-      "loss": 0.0088,
-      "step": 1020
-    },
-    {
-      "epoch": 255.0,
-      "eval_accuracy_ELSE": 0.9960067960219464,
-      "eval_accuracy_road": 0.9936350173719883,
-      "eval_accuracy_sidewalk": 0.9062235989862011,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9924518536090923,
-      "eval_iou_road": 0.9864712035963698,
-      "eval_iou_sidewalk": 0.8271430407402648,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021984245628118515,
-      "eval_mean_accuracy": 0.9652884707933787,
-      "eval_mean_iou": 0.9353553659819088,
-      "eval_overall_accuracy": 0.9929389953613281,
-      "eval_runtime": 0.6464,
-      "eval_samples_per_second": 1.547,
-      "eval_steps_per_second": 1.547,
-      "step": 1020
-    },
-    {
-      "epoch": 255.25,
-      "learning_rate": 3.0241228070175444e-05,
-      "loss": 0.0107,
-      "step": 1021
-    },
-    {
-      "epoch": 255.5,
-      "learning_rate": 3.0219298245614037e-05,
-      "loss": 0.0104,
-      "step": 1022
-    },
-    {
-      "epoch": 255.75,
-      "learning_rate": 3.0197368421052637e-05,
-      "loss": 0.009,
-      "step": 1023
-    },
-    {
-      "epoch": 256.0,
-      "learning_rate": 3.017543859649123e-05,
-      "loss": 0.0089,
-      "step": 1024
-    },
-    {
-      "epoch": 256.25,
-      "learning_rate": 3.015350877192983e-05,
-      "loss": 0.0092,
-      "step": 1025
-    },
-    {
-      "epoch": 256.5,
-      "learning_rate": 3.0131578947368423e-05,
-      "loss": 0.0096,
-      "step": 1026
-    },
-    {
-      "epoch": 256.75,
-      "learning_rate": 3.0109649122807022e-05,
-      "loss": 0.0097,
-      "step": 1027
-    },
-    {
-      "epoch": 257.0,
-      "learning_rate": 3.0087719298245615e-05,
-      "loss": 0.0099,
-      "step": 1028
-    },
-    {
-      "epoch": 257.25,
-      "learning_rate": 3.0065789473684215e-05,
-      "loss": 0.0094,
-      "step": 1029
-    },
-    {
-      "epoch": 257.5,
-      "learning_rate": 3.0043859649122808e-05,
-      "loss": 0.0107,
-      "step": 1030
-    },
-    {
-      "epoch": 257.75,
-      "learning_rate": 3.0021929824561408e-05,
-      "loss": 0.0096,
-      "step": 1031
-    },
-    {
-      "epoch": 258.0,
-      "learning_rate": 3e-05,
-      "loss": 0.0094,
-      "step": 1032
-    },
-    {
-      "epoch": 258.25,
-      "learning_rate": 2.99780701754386e-05,
-      "loss": 0.0103,
-      "step": 1033
-    },
-    {
-      "epoch": 258.5,
-      "learning_rate": 2.9956140350877194e-05,
-      "loss": 0.0098,
-      "step": 1034
-    },
-    {
-      "epoch": 258.75,
-      "learning_rate": 2.9934210526315793e-05,
-      "loss": 0.01,
-      "step": 1035
-    },
-    {
-      "epoch": 259.0,
-      "learning_rate": 2.9912280701754386e-05,
-      "loss": 0.0089,
-      "step": 1036
-    },
-    {
-      "epoch": 259.25,
-      "learning_rate": 2.9890350877192986e-05,
-      "loss": 0.0088,
-      "step": 1037
-    },
-    {
-      "epoch": 259.5,
-      "learning_rate": 2.986842105263158e-05,
-      "loss": 0.0097,
-      "step": 1038
-    },
-    {
-      "epoch": 259.75,
-      "learning_rate": 2.984649122807018e-05,
-      "loss": 0.0098,
-      "step": 1039
-    },
-    {
-      "epoch": 260.0,
-      "learning_rate": 2.9824561403508772e-05,
-      "loss": 0.0091,
-      "step": 1040
-    },
-    {
-      "epoch": 260.0,
-      "eval_accuracy_ELSE": 0.9972134145681604,
-      "eval_accuracy_road": 0.9928660932961212,
-      "eval_accuracy_sidewalk": 0.908898901717826,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933649544548051,
-      "eval_iou_road": 0.9870613383162753,
-      "eval_iou_sidewalk": 0.8453378732320587,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020471524447202682,
-      "eval_mean_accuracy": 0.9663261365273691,
-      "eval_mean_iou": 0.941921388667713,
-      "eval_overall_accuracy": 0.9936561584472656,
-      "eval_runtime": 0.6474,
-      "eval_samples_per_second": 1.545,
-      "eval_steps_per_second": 1.545,
-      "step": 1040
-    },
-    {
-      "epoch": 260.25,
-      "learning_rate": 2.980263157894737e-05,
-      "loss": 0.0098,
-      "step": 1041
-    },
-    {
-      "epoch": 260.5,
-      "learning_rate": 2.9780701754385965e-05,
-      "loss": 0.0087,
-      "step": 1042
-    },
-    {
-      "epoch": 260.75,
-      "learning_rate": 2.9758771929824564e-05,
-      "loss": 0.0109,
-      "step": 1043
-    },
-    {
-      "epoch": 261.0,
-      "learning_rate": 2.9736842105263157e-05,
-      "loss": 0.0101,
-      "step": 1044
-    },
-    {
-      "epoch": 261.25,
-      "learning_rate": 2.9714912280701757e-05,
-      "loss": 0.0092,
-      "step": 1045
-    },
-    {
-      "epoch": 261.5,
-      "learning_rate": 2.969298245614035e-05,
-      "loss": 0.0086,
-      "step": 1046
-    },
-    {
-      "epoch": 261.75,
-      "learning_rate": 2.967105263157895e-05,
-      "loss": 0.0085,
-      "step": 1047
-    },
-    {
-      "epoch": 262.0,
-      "learning_rate": 2.9649122807017543e-05,
-      "loss": 0.0105,
-      "step": 1048
-    },
-    {
-      "epoch": 262.25,
-      "learning_rate": 2.9627192982456143e-05,
-      "loss": 0.0096,
-      "step": 1049
-    },
-    {
-      "epoch": 262.5,
-      "learning_rate": 2.9605263157894735e-05,
-      "loss": 0.0081,
-      "step": 1050
-    },
-    {
-      "epoch": 262.75,
-      "learning_rate": 2.9583333333333335e-05,
-      "loss": 0.0098,
-      "step": 1051
-    },
-    {
-      "epoch": 263.0,
-      "learning_rate": 2.9561403508771928e-05,
-      "loss": 0.0097,
-      "step": 1052
-    },
-    {
-      "epoch": 263.25,
-      "learning_rate": 2.9539473684210528e-05,
-      "loss": 0.0092,
-      "step": 1053
-    },
-    {
-      "epoch": 263.5,
-      "learning_rate": 2.951754385964912e-05,
-      "loss": 0.0089,
-      "step": 1054
-    },
-    {
-      "epoch": 263.75,
-      "learning_rate": 2.949561403508772e-05,
-      "loss": 0.0098,
-      "step": 1055
-    },
-    {
-      "epoch": 264.0,
-      "learning_rate": 2.9473684210526314e-05,
-      "loss": 0.0094,
-      "step": 1056
-    },
-    {
-      "epoch": 264.25,
-      "learning_rate": 2.9451754385964914e-05,
-      "loss": 0.009,
-      "step": 1057
-    },
-    {
-      "epoch": 264.5,
-      "learning_rate": 2.9429824561403513e-05,
-      "loss": 0.0094,
-      "step": 1058
-    },
-    {
-      "epoch": 264.75,
-      "learning_rate": 2.9407894736842106e-05,
-      "loss": 0.0104,
-      "step": 1059
-    },
-    {
-      "epoch": 265.0,
-      "learning_rate": 2.9385964912280706e-05,
-      "loss": 0.0098,
-      "step": 1060
-    },
-    {
-      "epoch": 265.0,
-      "eval_accuracy_ELSE": 0.9967210276277771,
-      "eval_accuracy_road": 0.9942615481004727,
-      "eval_accuracy_sidewalk": 0.903266685440721,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932385070796174,
-      "eval_iou_road": 0.9869536948040935,
-      "eval_iou_sidewalk": 0.8407601572739187,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020581580698490143,
-      "eval_mean_accuracy": 0.9647497537229904,
-      "eval_mean_iou": 0.9403174530525432,
-      "eval_overall_accuracy": 0.9935302734375,
-      "eval_runtime": 0.6658,
-      "eval_samples_per_second": 1.502,
-      "eval_steps_per_second": 1.502,
-      "step": 1060
-    },
-    {
-      "epoch": 265.25,
-      "learning_rate": 2.93640350877193e-05,
-      "loss": 0.0084,
-      "step": 1061
-    },
-    {
-      "epoch": 265.5,
-      "learning_rate": 2.93421052631579e-05,
-      "loss": 0.0089,
-      "step": 1062
-    },
-    {
-      "epoch": 265.75,
-      "learning_rate": 2.9320175438596492e-05,
-      "loss": 0.0099,
-      "step": 1063
-    },
-    {
-      "epoch": 266.0,
-      "learning_rate": 2.929824561403509e-05,
-      "loss": 0.0091,
-      "step": 1064
-    },
-    {
-      "epoch": 266.25,
-      "learning_rate": 2.9276315789473684e-05,
-      "loss": 0.0095,
-      "step": 1065
-    },
-    {
-      "epoch": 266.5,
-      "learning_rate": 2.9254385964912284e-05,
-      "loss": 0.0087,
-      "step": 1066
-    },
-    {
-      "epoch": 266.75,
-      "learning_rate": 2.9232456140350877e-05,
-      "loss": 0.0096,
-      "step": 1067
-    },
-    {
-      "epoch": 267.0,
-      "learning_rate": 2.9210526315789477e-05,
-      "loss": 0.0081,
-      "step": 1068
-    },
-    {
-      "epoch": 267.25,
-      "learning_rate": 2.918859649122807e-05,
-      "loss": 0.0098,
-      "step": 1069
-    },
-    {
-      "epoch": 267.5,
-      "learning_rate": 2.916666666666667e-05,
-      "loss": 0.0089,
-      "step": 1070
-    },
-    {
-      "epoch": 267.75,
-      "learning_rate": 2.9144736842105263e-05,
-      "loss": 0.0085,
-      "step": 1071
-    },
-    {
-      "epoch": 268.0,
-      "learning_rate": 2.9122807017543863e-05,
-      "loss": 0.0097,
-      "step": 1072
-    },
-    {
-      "epoch": 268.25,
-      "learning_rate": 2.9100877192982455e-05,
-      "loss": 0.0093,
-      "step": 1073
-    },
-    {
-      "epoch": 268.5,
-      "learning_rate": 2.9078947368421055e-05,
-      "loss": 0.0083,
-      "step": 1074
-    },
-    {
-      "epoch": 268.75,
-      "learning_rate": 2.9057017543859648e-05,
-      "loss": 0.0088,
-      "step": 1075
-    },
-    {
-      "epoch": 269.0,
-      "learning_rate": 2.9035087719298248e-05,
-      "loss": 0.0094,
-      "step": 1076
-    },
-    {
-      "epoch": 269.25,
-      "learning_rate": 2.901315789473684e-05,
-      "loss": 0.0078,
-      "step": 1077
-    },
-    {
-      "epoch": 269.5,
-      "learning_rate": 2.899122807017544e-05,
-      "loss": 0.0092,
-      "step": 1078
-    },
-    {
-      "epoch": 269.75,
-      "learning_rate": 2.8969298245614034e-05,
-      "loss": 0.0091,
-      "step": 1079
-    },
-    {
-      "epoch": 270.0,
-      "learning_rate": 2.8947368421052634e-05,
-      "loss": 0.0105,
-      "step": 1080
-    },
-    {
-      "epoch": 270.0,
-      "eval_accuracy_ELSE": 0.9967751360827642,
-      "eval_accuracy_road": 0.9934071880161759,
-      "eval_accuracy_sidewalk": 0.9083356800901154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932602928807126,
-      "eval_iou_road": 0.9867890634945332,
-      "eval_iou_sidewalk": 0.8392090542474308,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02029246836900711,
-      "eval_mean_accuracy": 0.9661726680630185,
-      "eval_mean_iou": 0.9397528035408923,
-      "eval_overall_accuracy": 0.9934768676757812,
-      "eval_runtime": 0.6652,
-      "eval_samples_per_second": 1.503,
-      "eval_steps_per_second": 1.503,
-      "step": 1080
-    },
-    {
-      "epoch": 270.25,
-      "learning_rate": 2.8925438596491226e-05,
-      "loss": 0.0081,
-      "step": 1081
-    },
-    {
-      "epoch": 270.5,
-      "learning_rate": 2.8903508771929826e-05,
-      "loss": 0.0088,
-      "step": 1082
-    },
-    {
-      "epoch": 270.75,
-      "learning_rate": 2.888157894736842e-05,
-      "loss": 0.0082,
-      "step": 1083
-    },
-    {
-      "epoch": 271.0,
-      "learning_rate": 2.885964912280702e-05,
-      "loss": 0.0107,
-      "step": 1084
-    },
-    {
-      "epoch": 271.25,
-      "learning_rate": 2.8837719298245612e-05,
-      "loss": 0.009,
-      "step": 1085
-    },
-    {
-      "epoch": 271.5,
-      "learning_rate": 2.8815789473684212e-05,
-      "loss": 0.0087,
-      "step": 1086
-    },
-    {
-      "epoch": 271.75,
-      "learning_rate": 2.8793859649122805e-05,
-      "loss": 0.0081,
-      "step": 1087
-    },
-    {
-      "epoch": 272.0,
-      "learning_rate": 2.8771929824561404e-05,
-      "loss": 0.0092,
-      "step": 1088
-    },
-    {
-      "epoch": 272.25,
-      "learning_rate": 2.8749999999999997e-05,
-      "loss": 0.01,
-      "step": 1089
-    },
-    {
-      "epoch": 272.5,
-      "learning_rate": 2.8728070175438597e-05,
-      "loss": 0.0086,
-      "step": 1090
-    },
-    {
-      "epoch": 272.75,
-      "learning_rate": 2.8706140350877194e-05,
-      "loss": 0.0089,
-      "step": 1091
-    },
-    {
-      "epoch": 273.0,
-      "learning_rate": 2.868421052631579e-05,
-      "loss": 0.0081,
-      "step": 1092
-    },
-    {
-      "epoch": 273.25,
-      "learning_rate": 2.8662280701754386e-05,
-      "loss": 0.0085,
-      "step": 1093
-    },
-    {
-      "epoch": 273.5,
-      "learning_rate": 2.8640350877192983e-05,
-      "loss": 0.01,
-      "step": 1094
-    },
-    {
-      "epoch": 273.75,
-      "learning_rate": 2.861842105263158e-05,
-      "loss": 0.0099,
-      "step": 1095
-    },
-    {
-      "epoch": 274.0,
-      "learning_rate": 2.8596491228070175e-05,
-      "loss": 0.0086,
-      "step": 1096
-    },
-    {
-      "epoch": 274.25,
-      "learning_rate": 2.8574561403508772e-05,
-      "loss": 0.0091,
-      "step": 1097
-    },
-    {
-      "epoch": 274.5,
-      "learning_rate": 2.8552631578947368e-05,
-      "loss": 0.0095,
-      "step": 1098
-    },
-    {
-      "epoch": 274.75,
-      "learning_rate": 2.8530701754385968e-05,
-      "loss": 0.0086,
-      "step": 1099
-    },
-    {
-      "epoch": 275.0,
-      "learning_rate": 2.850877192982456e-05,
-      "loss": 0.0088,
-      "step": 1100
-    },
-    {
-      "epoch": 275.0,
-      "eval_accuracy_ELSE": 0.9965965781813066,
-      "eval_accuracy_road": 0.99411915475309,
-      "eval_accuracy_sidewalk": 0.9056603773584906,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931733989032143,
-      "eval_iou_road": 0.9871611781174441,
-      "eval_iou_sidewalk": 0.8372819578234835,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02087252028286457,
-      "eval_mean_accuracy": 0.9654587034309624,
-      "eval_mean_iou": 0.9392055116147139,
-      "eval_overall_accuracy": 0.99346923828125,
-      "eval_runtime": 0.6549,
-      "eval_samples_per_second": 1.527,
-      "eval_steps_per_second": 1.527,
-      "step": 1100
-    },
-    {
-      "epoch": 275.25,
-      "learning_rate": 2.848684210526316e-05,
-      "loss": 0.009,
-      "step": 1101
-    },
-    {
-      "epoch": 275.5,
-      "learning_rate": 2.8464912280701754e-05,
-      "loss": 0.0086,
-      "step": 1102
-    },
-    {
-      "epoch": 275.75,
-      "learning_rate": 2.8442982456140353e-05,
-      "loss": 0.0105,
-      "step": 1103
-    },
-    {
-      "epoch": 276.0,
-      "learning_rate": 2.842105263157895e-05,
-      "loss": 0.0101,
-      "step": 1104
-    },
-    {
-      "epoch": 276.25,
-      "learning_rate": 2.8399122807017546e-05,
-      "loss": 0.009,
-      "step": 1105
-    },
-    {
-      "epoch": 276.5,
-      "learning_rate": 2.8377192982456143e-05,
-      "loss": 0.009,
-      "step": 1106
-    },
-    {
-      "epoch": 276.75,
-      "learning_rate": 2.835526315789474e-05,
-      "loss": 0.0092,
-      "step": 1107
-    },
-    {
-      "epoch": 277.0,
-      "learning_rate": 2.8333333333333335e-05,
-      "loss": 0.0084,
-      "step": 1108
-    },
-    {
-      "epoch": 277.25,
-      "learning_rate": 2.8311403508771932e-05,
-      "loss": 0.0095,
-      "step": 1109
-    },
-    {
-      "epoch": 277.5,
-      "learning_rate": 2.8289473684210528e-05,
-      "loss": 0.0092,
-      "step": 1110
-    },
-    {
-      "epoch": 277.75,
-      "learning_rate": 2.8267543859649124e-05,
-      "loss": 0.0079,
-      "step": 1111
-    },
-    {
-      "epoch": 278.0,
-      "learning_rate": 2.824561403508772e-05,
-      "loss": 0.0087,
-      "step": 1112
-    },
-    {
-      "epoch": 278.25,
-      "learning_rate": 2.8223684210526317e-05,
-      "loss": 0.0092,
-      "step": 1113
-    },
-    {
-      "epoch": 278.5,
-      "learning_rate": 2.8201754385964914e-05,
-      "loss": 0.0084,
-      "step": 1114
-    },
-    {
-      "epoch": 278.75,
-      "learning_rate": 2.817982456140351e-05,
-      "loss": 0.009,
-      "step": 1115
-    },
-    {
-      "epoch": 279.0,
-      "learning_rate": 2.8157894736842106e-05,
-      "loss": 0.0084,
-      "step": 1116
-    },
-    {
-      "epoch": 279.25,
-      "learning_rate": 2.8135964912280706e-05,
-      "loss": 0.0082,
-      "step": 1117
-    },
-    {
-      "epoch": 279.5,
-      "learning_rate": 2.81140350877193e-05,
-      "loss": 0.0094,
-      "step": 1118
-    },
-    {
-      "epoch": 279.75,
-      "learning_rate": 2.80921052631579e-05,
-      "loss": 0.0091,
-      "step": 1119
-    },
-    {
-      "epoch": 280.0,
-      "learning_rate": 2.8070175438596492e-05,
-      "loss": 0.0089,
-      "step": 1120
-    },
-    {
-      "epoch": 280.0,
-      "eval_accuracy_ELSE": 0.9970510892031988,
-      "eval_accuracy_road": 0.9927664179529533,
-      "eval_accuracy_sidewalk": 0.907772458462405,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931550779081487,
-      "eval_iou_road": 0.9867387519990942,
-      "eval_iou_sidewalk": 0.8420846394984326,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021570276468992233,
-      "eval_mean_accuracy": 0.9658633218728524,
-      "eval_mean_iou": 0.9406594898018917,
-      "eval_overall_accuracy": 0.9934844970703125,
-      "eval_runtime": 0.6589,
-      "eval_samples_per_second": 1.518,
-      "eval_steps_per_second": 1.518,
-      "step": 1120
-    },
-    {
-      "epoch": 280.25,
-      "learning_rate": 2.804824561403509e-05,
-      "loss": 0.0084,
-      "step": 1121
-    },
-    {
-      "epoch": 280.5,
-      "learning_rate": 2.8026315789473685e-05,
-      "loss": 0.0093,
-      "step": 1122
-    },
-    {
-      "epoch": 280.75,
-      "learning_rate": 2.8004385964912284e-05,
-      "loss": 0.009,
-      "step": 1123
-    },
-    {
-      "epoch": 281.0,
-      "learning_rate": 2.7982456140350877e-05,
-      "loss": 0.0075,
-      "step": 1124
-    },
-    {
-      "epoch": 281.25,
-      "learning_rate": 2.7960526315789477e-05,
-      "loss": 0.0095,
-      "step": 1125
-    },
-    {
-      "epoch": 281.5,
-      "learning_rate": 2.793859649122807e-05,
-      "loss": 0.0084,
-      "step": 1126
-    },
-    {
-      "epoch": 281.75,
-      "learning_rate": 2.791666666666667e-05,
-      "loss": 0.0086,
-      "step": 1127
-    },
-    {
-      "epoch": 282.0,
-      "learning_rate": 2.7894736842105263e-05,
-      "loss": 0.0094,
-      "step": 1128
-    },
-    {
-      "epoch": 282.25,
-      "learning_rate": 2.7872807017543863e-05,
-      "loss": 0.009,
-      "step": 1129
-    },
-    {
-      "epoch": 282.5,
-      "learning_rate": 2.7850877192982456e-05,
-      "loss": 0.0079,
-      "step": 1130
-    },
-    {
-      "epoch": 282.75,
-      "learning_rate": 2.7828947368421055e-05,
-      "loss": 0.0095,
-      "step": 1131
-    },
-    {
-      "epoch": 283.0,
-      "learning_rate": 2.780701754385965e-05,
-      "loss": 0.0086,
-      "step": 1132
-    },
-    {
-      "epoch": 283.25,
-      "learning_rate": 2.7785087719298248e-05,
-      "loss": 0.0087,
-      "step": 1133
-    },
-    {
-      "epoch": 283.5,
-      "learning_rate": 2.776315789473684e-05,
-      "loss": 0.0089,
-      "step": 1134
-    },
-    {
-      "epoch": 283.75,
-      "learning_rate": 2.774122807017544e-05,
-      "loss": 0.0075,
-      "step": 1135
-    },
-    {
-      "epoch": 284.0,
-      "learning_rate": 2.7719298245614034e-05,
-      "loss": 0.0099,
-      "step": 1136
-    },
-    {
-      "epoch": 284.25,
-      "learning_rate": 2.7697368421052634e-05,
-      "loss": 0.0086,
-      "step": 1137
-    },
-    {
-      "epoch": 284.5,
-      "learning_rate": 2.7675438596491233e-05,
-      "loss": 0.0085,
-      "step": 1138
-    },
-    {
-      "epoch": 284.75,
-      "learning_rate": 2.7653508771929826e-05,
-      "loss": 0.0093,
-      "step": 1139
-    },
-    {
-      "epoch": 285.0,
-      "learning_rate": 2.7631578947368426e-05,
-      "loss": 0.0088,
-      "step": 1140
-    },
-    {
-      "epoch": 285.0,
-      "eval_accuracy_ELSE": 0.9964071985888515,
-      "eval_accuracy_road": 0.9933217520077462,
-      "eval_accuracy_sidewalk": 0.9149535342157139,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993123904543616,
-      "eval_iou_road": 0.9867879422290751,
-      "eval_iou_sidewalk": 0.8370475331701662,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020417595282197,
-      "eval_mean_accuracy": 0.9682274949374373,
-      "eval_mean_iou": 0.9389864599809524,
-      "eval_overall_accuracy": 0.9933738708496094,
-      "eval_runtime": 0.6361,
-      "eval_samples_per_second": 1.572,
-      "eval_steps_per_second": 1.572,
-      "step": 1140
-    },
-    {
-      "epoch": 285.25,
-      "learning_rate": 2.760964912280702e-05,
-      "loss": 0.0094,
-      "step": 1141
-    },
-    {
-      "epoch": 285.5,
-      "learning_rate": 2.758771929824562e-05,
-      "loss": 0.0087,
-      "step": 1142
-    },
-    {
-      "epoch": 285.75,
-      "learning_rate": 2.7565789473684212e-05,
-      "loss": 0.0085,
-      "step": 1143
-    },
-    {
-      "epoch": 286.0,
-      "learning_rate": 2.754385964912281e-05,
-      "loss": 0.0086,
-      "step": 1144
-    },
-    {
-      "epoch": 286.25,
-      "learning_rate": 2.7521929824561405e-05,
-      "loss": 0.0086,
-      "step": 1145
-    },
-    {
-      "epoch": 286.5,
-      "learning_rate": 2.7500000000000004e-05,
-      "loss": 0.008,
-      "step": 1146
-    },
-    {
-      "epoch": 286.75,
-      "learning_rate": 2.7478070175438597e-05,
-      "loss": 0.0098,
-      "step": 1147
-    },
-    {
-      "epoch": 287.0,
-      "learning_rate": 2.7456140350877197e-05,
-      "loss": 0.0093,
-      "step": 1148
-    },
-    {
-      "epoch": 287.25,
-      "learning_rate": 2.743421052631579e-05,
-      "loss": 0.0075,
-      "step": 1149
-    },
-    {
-      "epoch": 287.5,
-      "learning_rate": 2.741228070175439e-05,
-      "loss": 0.0095,
-      "step": 1150
-    },
-    {
-      "epoch": 287.75,
-      "learning_rate": 2.7390350877192983e-05,
-      "loss": 0.0092,
-      "step": 1151
-    },
-    {
-      "epoch": 288.0,
-      "learning_rate": 2.7368421052631583e-05,
-      "loss": 0.0089,
-      "step": 1152
-    },
-    {
-      "epoch": 288.25,
-      "learning_rate": 2.7346491228070176e-05,
-      "loss": 0.0092,
-      "step": 1153
-    },
-    {
-      "epoch": 288.5,
-      "learning_rate": 2.7324561403508775e-05,
-      "loss": 0.0088,
-      "step": 1154
-    },
-    {
-      "epoch": 288.75,
-      "learning_rate": 2.730263157894737e-05,
-      "loss": 0.0083,
-      "step": 1155
-    },
-    {
-      "epoch": 289.0,
-      "learning_rate": 2.7280701754385968e-05,
-      "loss": 0.0078,
-      "step": 1156
-    },
-    {
-      "epoch": 289.25,
-      "learning_rate": 2.725877192982456e-05,
-      "loss": 0.0082,
-      "step": 1157
-    },
-    {
-      "epoch": 289.5,
-      "learning_rate": 2.723684210526316e-05,
-      "loss": 0.0099,
-      "step": 1158
-    },
-    {
-      "epoch": 289.75,
-      "learning_rate": 2.7214912280701754e-05,
-      "loss": 0.0086,
-      "step": 1159
-    },
-    {
-      "epoch": 290.0,
-      "learning_rate": 2.7192982456140354e-05,
-      "loss": 0.0088,
-      "step": 1160
-    },
-    {
-      "epoch": 290.0,
-      "eval_accuracy_ELSE": 0.996320625060872,
-      "eval_accuracy_road": 0.9950731901805547,
-      "eval_accuracy_sidewalk": 0.9012954097437341,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9931233111661246,
-      "eval_iou_road": 0.9875918598078011,
-      "eval_iou_sidewalk": 0.8310828356271098,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02152002975344658,
-      "eval_mean_accuracy": 0.9642297416617204,
-      "eval_mean_iou": 0.9372660022003453,
-      "eval_overall_accuracy": 0.9934120178222656,
-      "eval_runtime": 0.6673,
-      "eval_samples_per_second": 1.499,
-      "eval_steps_per_second": 1.499,
-      "step": 1160
-    },
-    {
-      "epoch": 290.25,
-      "learning_rate": 2.7171052631578947e-05,
-      "loss": 0.0084,
-      "step": 1161
-    },
-    {
-      "epoch": 290.5,
-      "learning_rate": 2.7149122807017546e-05,
-      "loss": 0.0092,
-      "step": 1162
-    },
-    {
-      "epoch": 290.75,
-      "learning_rate": 2.712719298245614e-05,
-      "loss": 0.0085,
-      "step": 1163
-    },
-    {
-      "epoch": 291.0,
-      "learning_rate": 2.710526315789474e-05,
-      "loss": 0.0102,
-      "step": 1164
-    },
-    {
-      "epoch": 291.25,
-      "learning_rate": 2.7083333333333332e-05,
-      "loss": 0.0082,
-      "step": 1165
-    },
-    {
-      "epoch": 291.5,
-      "learning_rate": 2.7061403508771932e-05,
-      "loss": 0.0089,
-      "step": 1166
-    },
-    {
-      "epoch": 291.75,
-      "learning_rate": 2.7039473684210525e-05,
-      "loss": 0.0092,
-      "step": 1167
-    },
-    {
-      "epoch": 292.0,
-      "learning_rate": 2.7017543859649125e-05,
-      "loss": 0.0077,
-      "step": 1168
-    },
-    {
-      "epoch": 292.25,
-      "learning_rate": 2.6995614035087718e-05,
-      "loss": 0.0078,
-      "step": 1169
-    },
-    {
-      "epoch": 292.5,
-      "learning_rate": 2.6973684210526317e-05,
-      "loss": 0.0087,
-      "step": 1170
-    },
-    {
-      "epoch": 292.75,
-      "learning_rate": 2.695175438596491e-05,
-      "loss": 0.0081,
-      "step": 1171
-    },
-    {
-      "epoch": 293.0,
-      "learning_rate": 2.692982456140351e-05,
-      "loss": 0.0092,
-      "step": 1172
-    },
-    {
-      "epoch": 293.25,
-      "learning_rate": 2.6907894736842103e-05,
-      "loss": 0.009,
-      "step": 1173
-    },
-    {
-      "epoch": 293.5,
-      "learning_rate": 2.6885964912280703e-05,
-      "loss": 0.0085,
-      "step": 1174
-    },
-    {
-      "epoch": 293.75,
-      "learning_rate": 2.6864035087719296e-05,
-      "loss": 0.0079,
-      "step": 1175
-    },
-    {
-      "epoch": 294.0,
-      "learning_rate": 2.6842105263157896e-05,
-      "loss": 0.0093,
-      "step": 1176
-    },
-    {
-      "epoch": 294.25,
-      "learning_rate": 2.682017543859649e-05,
-      "loss": 0.0089,
-      "step": 1177
-    },
-    {
-      "epoch": 294.5,
-      "learning_rate": 2.679824561403509e-05,
-      "loss": 0.0084,
-      "step": 1178
-    },
-    {
-      "epoch": 294.75,
-      "learning_rate": 2.6776315789473688e-05,
-      "loss": 0.0092,
-      "step": 1179
-    },
-    {
-      "epoch": 295.0,
-      "learning_rate": 2.675438596491228e-05,
-      "loss": 0.0086,
-      "step": 1180
-    },
-    {
-      "epoch": 295.0,
-      "eval_accuracy_ELSE": 0.9971755386496693,
-      "eval_accuracy_road": 0.9932505553340548,
-      "eval_accuracy_sidewalk": 0.9039707124753591,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932897481364903,
-      "eval_iou_road": 0.9869405888761549,
-      "eval_iou_sidewalk": 0.8438485804416404,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021573306992650032,
-      "eval_mean_accuracy": 0.964798935486361,
-      "eval_mean_iou": 0.9413596391514285,
-      "eval_overall_accuracy": 0.9935989379882812,
-      "eval_runtime": 0.6642,
-      "eval_samples_per_second": 1.506,
-      "eval_steps_per_second": 1.506,
-      "step": 1180
-    },
-    {
-      "epoch": 295.25,
-      "learning_rate": 2.673245614035088e-05,
-      "loss": 0.0088,
-      "step": 1181
-    },
-    {
-      "epoch": 295.5,
-      "learning_rate": 2.6710526315789474e-05,
-      "loss": 0.0081,
-      "step": 1182
-    },
-    {
-      "epoch": 295.75,
-      "learning_rate": 2.6688596491228074e-05,
-      "loss": 0.008,
-      "step": 1183
-    },
-    {
-      "epoch": 296.0,
-      "learning_rate": 2.6666666666666667e-05,
-      "loss": 0.0091,
-      "step": 1184
-    },
-    {
-      "epoch": 296.25,
-      "learning_rate": 2.6644736842105266e-05,
-      "loss": 0.008,
-      "step": 1185
-    },
-    {
-      "epoch": 296.5,
-      "learning_rate": 2.662280701754386e-05,
-      "loss": 0.0096,
-      "step": 1186
-    },
-    {
-      "epoch": 296.75,
-      "learning_rate": 2.660087719298246e-05,
-      "loss": 0.0081,
-      "step": 1187
-    },
-    {
-      "epoch": 297.0,
-      "learning_rate": 2.6578947368421052e-05,
-      "loss": 0.0078,
-      "step": 1188
-    },
-    {
-      "epoch": 297.25,
-      "learning_rate": 2.6557017543859652e-05,
-      "loss": 0.0082,
-      "step": 1189
-    },
-    {
-      "epoch": 297.5,
-      "learning_rate": 2.6535087719298245e-05,
-      "loss": 0.0085,
-      "step": 1190
-    },
-    {
-      "epoch": 297.75,
-      "learning_rate": 2.6513157894736845e-05,
-      "loss": 0.0078,
-      "step": 1191
-    },
-    {
-      "epoch": 298.0,
-      "learning_rate": 2.6491228070175438e-05,
-      "loss": 0.0083,
-      "step": 1192
-    },
-    {
-      "epoch": 298.25,
-      "learning_rate": 2.6469298245614037e-05,
-      "loss": 0.0077,
-      "step": 1193
-    },
-    {
-      "epoch": 298.5,
-      "learning_rate": 2.644736842105263e-05,
-      "loss": 0.008,
-      "step": 1194
-    },
-    {
-      "epoch": 298.75,
-      "learning_rate": 2.642543859649123e-05,
-      "loss": 0.0092,
-      "step": 1195
-    },
-    {
-      "epoch": 299.0,
-      "learning_rate": 2.6403508771929823e-05,
-      "loss": 0.0089,
-      "step": 1196
-    },
-    {
-      "epoch": 299.25,
-      "learning_rate": 2.6381578947368423e-05,
-      "loss": 0.0082,
-      "step": 1197
-    },
-    {
-      "epoch": 299.5,
-      "learning_rate": 2.6359649122807016e-05,
-      "loss": 0.0084,
-      "step": 1198
-    },
-    {
-      "epoch": 299.75,
-      "learning_rate": 2.6337719298245616e-05,
-      "loss": 0.0099,
-      "step": 1199
-    },
-    {
-      "epoch": 300.0,
-      "learning_rate": 2.6315789473684212e-05,
-      "loss": 0.0081,
-      "step": 1200
-    },
-    {
-      "epoch": 300.0,
-      "eval_accuracy_ELSE": 0.9968130120012553,
-      "eval_accuracy_road": 0.9933502306772227,
-      "eval_accuracy_sidewalk": 0.9105885666009574,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993292679639185,
-      "eval_iou_road": 0.9870535966947762,
-      "eval_iou_sidewalk": 0.8409622886866059,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020722003653645515,
-      "eval_mean_accuracy": 0.9669172697598117,
-      "eval_mean_iou": 0.9404361883401892,
-      "eval_overall_accuracy": 0.9935493469238281,
-      "eval_runtime": 0.6472,
-      "eval_samples_per_second": 1.545,
-      "eval_steps_per_second": 1.545,
-      "step": 1200
-    },
-    {
-      "epoch": 300.25,
-      "learning_rate": 2.629385964912281e-05,
-      "loss": 0.0083,
-      "step": 1201
-    },
-    {
-      "epoch": 300.5,
-      "learning_rate": 2.6271929824561405e-05,
-      "loss": 0.0079,
-      "step": 1202
-    },
-    {
-      "epoch": 300.75,
-      "learning_rate": 2.625e-05,
-      "loss": 0.0088,
-      "step": 1203
-    },
-    {
-      "epoch": 301.0,
-      "learning_rate": 2.6228070175438597e-05,
-      "loss": 0.0083,
-      "step": 1204
-    },
-    {
-      "epoch": 301.25,
-      "learning_rate": 2.6206140350877194e-05,
-      "loss": 0.0084,
-      "step": 1205
-    },
-    {
-      "epoch": 301.5,
-      "learning_rate": 2.618421052631579e-05,
-      "loss": 0.0088,
-      "step": 1206
-    },
-    {
-      "epoch": 301.75,
-      "learning_rate": 2.6162280701754387e-05,
-      "loss": 0.0077,
-      "step": 1207
-    },
-    {
-      "epoch": 302.0,
-      "learning_rate": 2.6140350877192983e-05,
-      "loss": 0.0094,
-      "step": 1208
-    },
-    {
-      "epoch": 302.25,
-      "learning_rate": 2.611842105263158e-05,
-      "loss": 0.0088,
-      "step": 1209
-    },
-    {
-      "epoch": 302.5,
-      "learning_rate": 2.6096491228070176e-05,
-      "loss": 0.0081,
-      "step": 1210
-    },
-    {
-      "epoch": 302.75,
-      "learning_rate": 2.6074561403508772e-05,
-      "loss": 0.0087,
-      "step": 1211
-    },
-    {
-      "epoch": 303.0,
-      "learning_rate": 2.605263157894737e-05,
-      "loss": 0.0077,
-      "step": 1212
-    },
-    {
-      "epoch": 303.25,
-      "learning_rate": 2.6030701754385968e-05,
-      "loss": 0.0081,
-      "step": 1213
-    },
-    {
-      "epoch": 303.5,
-      "learning_rate": 2.600877192982456e-05,
-      "loss": 0.0084,
-      "step": 1214
-    },
-    {
-      "epoch": 303.75,
-      "learning_rate": 2.598684210526316e-05,
-      "loss": 0.0088,
-      "step": 1215
-    },
-    {
-      "epoch": 304.0,
-      "learning_rate": 2.5964912280701754e-05,
-      "loss": 0.0087,
-      "step": 1216
-    },
-    {
-      "epoch": 304.25,
-      "learning_rate": 2.5942982456140354e-05,
-      "loss": 0.0081,
-      "step": 1217
-    },
-    {
-      "epoch": 304.5,
-      "learning_rate": 2.5921052631578947e-05,
-      "loss": 0.0084,
-      "step": 1218
-    },
-    {
-      "epoch": 304.75,
-      "learning_rate": 2.5899122807017546e-05,
-      "loss": 0.008,
-      "step": 1219
-    },
-    {
-      "epoch": 305.0,
-      "learning_rate": 2.5877192982456143e-05,
-      "loss": 0.009,
-      "step": 1220
-    },
-    {
-      "epoch": 305.0,
-      "eval_accuracy_ELSE": 0.9967859577737618,
-      "eval_accuracy_road": 0.9938770860625391,
-      "eval_accuracy_sidewalk": 0.9072092368346945,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9932710764121034,
-      "eval_iou_road": 0.9871720528958348,
-      "eval_iou_sidewalk": 0.8417820747321662,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020752139389514923,
-      "eval_mean_accuracy": 0.9659574268903318,
-      "eval_mean_iou": 0.9407417346800347,
-      "eval_overall_accuracy": 0.9935798645019531,
-      "eval_runtime": 0.6679,
-      "eval_samples_per_second": 1.497,
-      "eval_steps_per_second": 1.497,
-      "step": 1220
-    },
-    {
-      "epoch": 305.25,
-      "learning_rate": 2.585526315789474e-05,
-      "loss": 0.0082,
-      "step": 1221
-    },
-    {
-      "epoch": 305.5,
-      "learning_rate": 2.5833333333333336e-05,
-      "loss": 0.0087,
-      "step": 1222
-    },
-    {
-      "epoch": 305.75,
-      "learning_rate": 2.5811403508771932e-05,
-      "loss": 0.0088,
-      "step": 1223
-    },
-    {
-      "epoch": 306.0,
-      "learning_rate": 2.578947368421053e-05,
-      "loss": 0.0086,
-      "step": 1224
-    },
-    {
-      "epoch": 306.25,
-      "learning_rate": 2.5767543859649125e-05,
-      "loss": 0.0092,
-      "step": 1225
-    },
-    {
-      "epoch": 306.5,
-      "learning_rate": 2.5745614035087724e-05,
-      "loss": 0.0082,
-      "step": 1226
-    },
-    {
-      "epoch": 306.75,
-      "learning_rate": 2.5723684210526317e-05,
-      "loss": 0.0078,
-      "step": 1227
-    },
-    {
-      "epoch": 307.0,
-      "learning_rate": 2.5701754385964917e-05,
-      "loss": 0.008,
-      "step": 1228
-    },
-    {
-      "epoch": 307.25,
-      "learning_rate": 2.567982456140351e-05,
-      "loss": 0.008,
-      "step": 1229
-    },
-    {
-      "epoch": 307.5,
-      "learning_rate": 2.565789473684211e-05,
-      "loss": 0.0086,
-      "step": 1230
-    },
-    {
-      "epoch": 307.75,
-      "learning_rate": 2.5635964912280703e-05,
-      "loss": 0.0078,
-      "step": 1231
-    },
-    {
-      "epoch": 308.0,
-      "learning_rate": 2.5614035087719303e-05,
-      "loss": 0.0083,
-      "step": 1232
-    },
-    {
-      "epoch": 308.25,
-      "learning_rate": 2.5592105263157896e-05,
-      "loss": 0.0084,
-      "step": 1233
-    },
-    {
-      "epoch": 308.5,
-      "learning_rate": 2.5570175438596495e-05,
-      "loss": 0.0081,
-      "step": 1234
-    },
-    {
-      "epoch": 308.75,
-      "learning_rate": 2.554824561403509e-05,
-      "loss": 0.0088,
-      "step": 1235
-    },
-    {
-      "epoch": 309.0,
-      "learning_rate": 2.5526315789473688e-05,
-      "loss": 0.0086,
-      "step": 1236
-    },
-    {
-      "epoch": 309.25,
-      "learning_rate": 2.550438596491228e-05,
-      "loss": 0.0077,
-      "step": 1237
-    },
-    {
-      "epoch": 309.5,
-      "learning_rate": 2.548245614035088e-05,
-      "loss": 0.0085,
-      "step": 1238
-    },
-    {
-      "epoch": 309.75,
-      "learning_rate": 2.5460526315789474e-05,
-      "loss": 0.0086,
-      "step": 1239
-    },
-    {
-      "epoch": 310.0,
-      "learning_rate": 2.5438596491228074e-05,
-      "loss": 0.0083,
-      "step": 1240
-    },
-    {
-      "epoch": 310.0,
-      "eval_accuracy_ELSE": 0.9970889651216899,
-      "eval_accuracy_road": 0.9933075126730079,
-      "eval_accuracy_sidewalk": 0.9069276260208392,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933694867012388,
-      "eval_iou_road": 0.9869692553658088,
-      "eval_iou_sidewalk": 0.8443890928159413,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021018290892243385,
-      "eval_mean_accuracy": 0.9657747012718456,
-      "eval_mean_iou": 0.9415759449609963,
-      "eval_overall_accuracy": 0.9936332702636719,
-      "eval_runtime": 0.7945,
-      "eval_samples_per_second": 1.259,
-      "eval_steps_per_second": 1.259,
-      "step": 1240
-    },
-    {
-      "epoch": 310.25,
-      "learning_rate": 2.5416666666666667e-05,
-      "loss": 0.0084,
-      "step": 1241
-    },
-    {
-      "epoch": 310.5,
-      "learning_rate": 2.5394736842105266e-05,
-      "loss": 0.0086,
-      "step": 1242
-    },
-    {
-      "epoch": 310.75,
-      "learning_rate": 2.537280701754386e-05,
-      "loss": 0.0081,
-      "step": 1243
-    },
-    {
-      "epoch": 311.0,
-      "learning_rate": 2.535087719298246e-05,
-      "loss": 0.0081,
-      "step": 1244
-    },
-    {
-      "epoch": 311.25,
-      "learning_rate": 2.5328947368421052e-05,
-      "loss": 0.0089,
-      "step": 1245
-    },
-    {
-      "epoch": 311.5,
-      "learning_rate": 2.5307017543859652e-05,
-      "loss": 0.0085,
-      "step": 1246
-    },
-    {
-      "epoch": 311.75,
-      "learning_rate": 2.5285087719298245e-05,
-      "loss": 0.0084,
-      "step": 1247
-    },
-    {
-      "epoch": 312.0,
-      "learning_rate": 2.5263157894736845e-05,
-      "loss": 0.0081,
-      "step": 1248
-    },
-    {
-      "epoch": 312.25,
-      "learning_rate": 2.5241228070175438e-05,
-      "loss": 0.0079,
-      "step": 1249
-    },
-    {
-      "epoch": 312.5,
-      "learning_rate": 2.5219298245614037e-05,
-      "loss": 0.008,
-      "step": 1250
-    },
-    {
-      "epoch": 312.75,
-      "learning_rate": 2.519736842105263e-05,
-      "loss": 0.0094,
-      "step": 1251
-    },
-    {
-      "epoch": 313.0,
-      "learning_rate": 2.517543859649123e-05,
-      "loss": 0.0075,
-      "step": 1252
-    },
-    {
-      "epoch": 313.25,
-      "learning_rate": 2.5153508771929823e-05,
-      "loss": 0.0082,
-      "step": 1253
-    },
-    {
-      "epoch": 313.5,
-      "learning_rate": 2.5131578947368423e-05,
-      "loss": 0.0085,
-      "step": 1254
-    },
-    {
-      "epoch": 313.75,
-      "learning_rate": 2.5109649122807016e-05,
-      "loss": 0.0086,
-      "step": 1255
-    },
-    {
-      "epoch": 314.0,
-      "learning_rate": 2.5087719298245616e-05,
-      "loss": 0.0082,
-      "step": 1256
-    },
-    {
-      "epoch": 314.25,
-      "learning_rate": 2.506578947368421e-05,
-      "loss": 0.0081,
-      "step": 1257
-    },
-    {
-      "epoch": 314.5,
-      "learning_rate": 2.504385964912281e-05,
-      "loss": 0.0078,
-      "step": 1258
-    },
-    {
-      "epoch": 314.75,
-      "learning_rate": 2.50219298245614e-05,
-      "loss": 0.0088,
-      "step": 1259
-    },
-    {
-      "epoch": 315.0,
-      "learning_rate": 2.5e-05,
-      "loss": 0.0094,
-      "step": 1260
-    },
-    {
-      "epoch": 315.0,
-      "eval_accuracy_ELSE": 0.9969158180657309,
-      "eval_accuracy_road": 0.994631770803668,
-      "eval_accuracy_sidewalk": 0.9031258800337933,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.99360405543871,
-      "eval_iou_road": 0.9872932862190813,
-      "eval_iou_sidewalk": 0.8440584287406238,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020621269941329956,
-      "eval_mean_accuracy": 0.9648911563010641,
-      "eval_mean_iou": 0.9416519234661384,
-      "eval_overall_accuracy": 0.9937629699707031,
-      "eval_runtime": 0.6661,
-      "eval_samples_per_second": 1.501,
-      "eval_steps_per_second": 1.501,
-      "step": 1260
-    },
-    {
-      "epoch": 315.25,
-      "learning_rate": 2.4978070175438598e-05,
-      "loss": 0.0083,
-      "step": 1261
-    },
-    {
-      "epoch": 315.5,
-      "learning_rate": 2.4956140350877194e-05,
-      "loss": 0.0095,
-      "step": 1262
-    },
-    {
-      "epoch": 315.75,
-      "learning_rate": 2.493421052631579e-05,
-      "loss": 0.008,
-      "step": 1263
-    },
-    {
-      "epoch": 316.0,
-      "learning_rate": 2.4912280701754387e-05,
-      "loss": 0.0083,
-      "step": 1264
-    },
-    {
-      "epoch": 316.25,
-      "learning_rate": 2.4890350877192983e-05,
-      "loss": 0.0081,
-      "step": 1265
-    },
-    {
-      "epoch": 316.5,
-      "learning_rate": 2.486842105263158e-05,
-      "loss": 0.0089,
-      "step": 1266
-    },
-    {
-      "epoch": 316.75,
-      "learning_rate": 2.4846491228070176e-05,
-      "loss": 0.0079,
-      "step": 1267
-    },
-    {
-      "epoch": 317.0,
-      "learning_rate": 2.4824561403508772e-05,
-      "loss": 0.0086,
-      "step": 1268
-    },
-    {
-      "epoch": 317.25,
-      "learning_rate": 2.480263157894737e-05,
-      "loss": 0.0079,
-      "step": 1269
-    },
-    {
-      "epoch": 317.5,
-      "learning_rate": 2.4780701754385965e-05,
-      "loss": 0.008,
-      "step": 1270
-    },
-    {
-      "epoch": 317.75,
-      "learning_rate": 2.475877192982456e-05,
-      "loss": 0.0088,
-      "step": 1271
-    },
-    {
-      "epoch": 318.0,
-      "learning_rate": 2.4736842105263158e-05,
-      "loss": 0.0095,
-      "step": 1272
-    },
-    {
-      "epoch": 318.25,
-      "learning_rate": 2.4714912280701754e-05,
-      "loss": 0.0084,
-      "step": 1273
-    },
-    {
-      "epoch": 318.5,
-      "learning_rate": 2.469298245614035e-05,
-      "loss": 0.0084,
-      "step": 1274
-    },
-    {
-      "epoch": 318.75,
-      "learning_rate": 2.4671052631578947e-05,
-      "loss": 0.0084,
-      "step": 1275
-    },
-    {
-      "epoch": 319.0,
-      "learning_rate": 2.4649122807017543e-05,
-      "loss": 0.0089,
-      "step": 1276
-    },
-    {
-      "epoch": 319.25,
-      "learning_rate": 2.462719298245614e-05,
-      "loss": 0.0091,
-      "step": 1277
-    },
-    {
-      "epoch": 319.5,
-      "learning_rate": 2.4605263157894736e-05,
-      "loss": 0.0084,
-      "step": 1278
-    },
-    {
-      "epoch": 319.75,
-      "learning_rate": 2.4583333333333332e-05,
-      "loss": 0.0077,
-      "step": 1279
-    },
-    {
-      "epoch": 320.0,
-      "learning_rate": 2.456140350877193e-05,
-      "loss": 0.0084,
-      "step": 1280
-    },
-    {
-      "epoch": 320.0,
-      "eval_accuracy_ELSE": 0.9971755386496693,
-      "eval_accuracy_road": 0.994631770803668,
-      "eval_accuracy_sidewalk": 0.901717825964517,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937932410498105,
-      "eval_iou_road": 0.9874607707314315,
-      "eval_iou_sidewalk": 0.8470899470899471,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02046687714755535,
-      "eval_mean_accuracy": 0.964508378472618,
-      "eval_mean_iou": 0.9427813196237297,
-      "eval_overall_accuracy": 0.9939079284667969,
-      "eval_runtime": 0.6697,
-      "eval_samples_per_second": 1.493,
-      "eval_steps_per_second": 1.493,
-      "step": 1280
-    },
-    {
-      "epoch": 320.25,
-      "learning_rate": 2.453947368421053e-05,
-      "loss": 0.0088,
-      "step": 1281
-    },
-    {
-      "epoch": 320.5,
-      "learning_rate": 2.4517543859649125e-05,
-      "loss": 0.0078,
-      "step": 1282
-    },
-    {
-      "epoch": 320.75,
-      "learning_rate": 2.449561403508772e-05,
-      "loss": 0.0072,
-      "step": 1283
-    },
-    {
-      "epoch": 321.0,
-      "learning_rate": 2.4473684210526318e-05,
-      "loss": 0.0079,
-      "step": 1284
-    },
-    {
-      "epoch": 321.25,
-      "learning_rate": 2.4451754385964914e-05,
-      "loss": 0.0088,
-      "step": 1285
-    },
-    {
-      "epoch": 321.5,
-      "learning_rate": 2.442982456140351e-05,
-      "loss": 0.0082,
-      "step": 1286
-    },
-    {
-      "epoch": 321.75,
-      "learning_rate": 2.4407894736842107e-05,
-      "loss": 0.009,
-      "step": 1287
-    },
-    {
-      "epoch": 322.0,
-      "learning_rate": 2.4385964912280703e-05,
-      "loss": 0.0074,
-      "step": 1288
-    },
-    {
-      "epoch": 322.25,
-      "learning_rate": 2.43640350877193e-05,
-      "loss": 0.0083,
-      "step": 1289
-    },
-    {
-      "epoch": 322.5,
-      "learning_rate": 2.4342105263157896e-05,
-      "loss": 0.0078,
-      "step": 1290
-    },
-    {
-      "epoch": 322.75,
-      "learning_rate": 2.4320175438596492e-05,
-      "loss": 0.0091,
-      "step": 1291
-    },
-    {
-      "epoch": 323.0,
-      "learning_rate": 2.429824561403509e-05,
-      "loss": 0.0081,
-      "step": 1292
-    },
-    {
-      "epoch": 323.25,
-      "learning_rate": 2.4276315789473685e-05,
-      "loss": 0.0079,
-      "step": 1293
-    },
-    {
-      "epoch": 323.5,
-      "learning_rate": 2.425438596491228e-05,
-      "loss": 0.0074,
-      "step": 1294
-    },
-    {
-      "epoch": 323.75,
-      "learning_rate": 2.4232456140350878e-05,
-      "loss": 0.0089,
-      "step": 1295
-    },
-    {
-      "epoch": 324.0,
-      "learning_rate": 2.4210526315789474e-05,
-      "loss": 0.008,
-      "step": 1296
-    },
-    {
-      "epoch": 324.25,
-      "learning_rate": 2.418859649122807e-05,
-      "loss": 0.0084,
-      "step": 1297
-    },
-    {
-      "epoch": 324.5,
-      "learning_rate": 2.4166666666666667e-05,
-      "loss": 0.0077,
-      "step": 1298
-    },
-    {
-      "epoch": 324.75,
-      "learning_rate": 2.4144736842105263e-05,
-      "loss": 0.0086,
-      "step": 1299
-    },
-    {
-      "epoch": 325.0,
-      "learning_rate": 2.412280701754386e-05,
-      "loss": 0.0085,
-      "step": 1300
-    },
-    {
-      "epoch": 325.0,
-      "eval_accuracy_ELSE": 0.9963909660523553,
-      "eval_accuracy_road": 0.994019479409922,
-      "eval_accuracy_sidewalk": 0.9093213179386088,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9930863025740311,
-      "eval_iou_road": 0.9872436713336161,
-      "eval_iou_sidewalk": 0.8347983453981386,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021792393177747726,
-      "eval_mean_accuracy": 0.9665772544669621,
-      "eval_mean_iou": 0.9383761064352619,
-      "eval_overall_accuracy": 0.9933967590332031,
-      "eval_runtime": 0.6382,
-      "eval_samples_per_second": 1.567,
-      "eval_steps_per_second": 1.567,
-      "step": 1300
-    },
-    {
-      "epoch": 325.25,
-      "learning_rate": 2.4100877192982456e-05,
-      "loss": 0.0084,
-      "step": 1301
-    },
-    {
-      "epoch": 325.5,
-      "learning_rate": 2.4078947368421052e-05,
-      "loss": 0.0085,
-      "step": 1302
-    },
-    {
-      "epoch": 325.75,
-      "learning_rate": 2.405701754385965e-05,
-      "loss": 0.0085,
-      "step": 1303
-    },
-    {
-      "epoch": 326.0,
-      "learning_rate": 2.4035087719298245e-05,
-      "loss": 0.0083,
-      "step": 1304
-    },
-    {
-      "epoch": 326.25,
-      "learning_rate": 2.401315789473684e-05,
-      "loss": 0.0077,
-      "step": 1305
-    },
-    {
-      "epoch": 326.5,
-      "learning_rate": 2.3991228070175438e-05,
-      "loss": 0.0089,
-      "step": 1306
-    },
-    {
-      "epoch": 326.75,
-      "learning_rate": 2.3969298245614034e-05,
-      "loss": 0.0092,
-      "step": 1307
-    },
-    {
-      "epoch": 327.0,
-      "learning_rate": 2.394736842105263e-05,
-      "loss": 0.0086,
-      "step": 1308
-    },
-    {
-      "epoch": 327.25,
-      "learning_rate": 2.392543859649123e-05,
-      "loss": 0.0076,
-      "step": 1309
-    },
-    {
-      "epoch": 327.5,
-      "learning_rate": 2.3903508771929827e-05,
-      "loss": 0.0097,
-      "step": 1310
-    },
-    {
-      "epoch": 327.75,
-      "learning_rate": 2.3881578947368423e-05,
-      "loss": 0.0092,
-      "step": 1311
-    },
-    {
-      "epoch": 328.0,
-      "learning_rate": 2.385964912280702e-05,
-      "loss": 0.008,
-      "step": 1312
-    },
-    {
-      "epoch": 328.25,
-      "learning_rate": 2.3837719298245616e-05,
-      "loss": 0.0081,
-      "step": 1313
-    },
-    {
-      "epoch": 328.5,
-      "learning_rate": 2.3815789473684212e-05,
-      "loss": 0.0083,
-      "step": 1314
-    },
-    {
-      "epoch": 328.75,
-      "learning_rate": 2.379385964912281e-05,
-      "loss": 0.009,
-      "step": 1315
-    },
-    {
-      "epoch": 329.0,
-      "learning_rate": 2.3771929824561405e-05,
-      "loss": 0.0088,
-      "step": 1316
-    },
-    {
-      "epoch": 329.25,
-      "learning_rate": 2.375e-05,
-      "loss": 0.009,
-      "step": 1317
-    },
-    {
-      "epoch": 329.5,
-      "learning_rate": 2.3728070175438598e-05,
-      "loss": 0.0077,
-      "step": 1318
-    },
-    {
-      "epoch": 329.75,
-      "learning_rate": 2.3706140350877194e-05,
-      "loss": 0.0083,
-      "step": 1319
-    },
-    {
-      "epoch": 330.0,
-      "learning_rate": 2.368421052631579e-05,
-      "loss": 0.0089,
-      "step": 1320
-    },
-    {
-      "epoch": 330.0,
-      "eval_accuracy_ELSE": 0.9968400662287489,
-      "eval_accuracy_road": 0.9942757874352111,
-      "eval_accuracy_sidewalk": 0.9079132638693326,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933785547131965,
-      "eval_iou_road": 0.9877776205969727,
-      "eval_iou_sidewalk": 0.8444211629125197,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020651984959840775,
-      "eval_mean_accuracy": 0.9663430391777642,
-      "eval_mean_iou": 0.9418591127408963,
-      "eval_overall_accuracy": 0.993743896484375,
-      "eval_runtime": 0.6434,
-      "eval_samples_per_second": 1.554,
-      "eval_steps_per_second": 1.554,
-      "step": 1320
-    },
-    {
-      "epoch": 330.25,
-      "learning_rate": 2.3662280701754387e-05,
-      "loss": 0.0094,
-      "step": 1321
-    },
-    {
-      "epoch": 330.5,
-      "learning_rate": 2.3640350877192987e-05,
-      "loss": 0.0078,
-      "step": 1322
-    },
-    {
-      "epoch": 330.75,
-      "learning_rate": 2.3618421052631583e-05,
-      "loss": 0.0085,
-      "step": 1323
-    },
-    {
-      "epoch": 331.0,
-      "learning_rate": 2.359649122807018e-05,
-      "loss": 0.0082,
-      "step": 1324
-    },
-    {
-      "epoch": 331.25,
-      "learning_rate": 2.3574561403508776e-05,
-      "loss": 0.0078,
-      "step": 1325
-    },
-    {
-      "epoch": 331.5,
-      "learning_rate": 2.3552631578947372e-05,
-      "loss": 0.009,
-      "step": 1326
-    },
-    {
-      "epoch": 331.75,
-      "learning_rate": 2.353070175438597e-05,
-      "loss": 0.0092,
-      "step": 1327
-    },
-    {
-      "epoch": 332.0,
-      "learning_rate": 2.3508771929824565e-05,
-      "loss": 0.0084,
-      "step": 1328
-    },
-    {
-      "epoch": 332.25,
-      "learning_rate": 2.348684210526316e-05,
-      "loss": 0.0093,
-      "step": 1329
-    },
-    {
-      "epoch": 332.5,
-      "learning_rate": 2.3464912280701758e-05,
-      "loss": 0.0085,
-      "step": 1330
-    },
-    {
-      "epoch": 332.75,
-      "learning_rate": 2.3442982456140354e-05,
-      "loss": 0.0078,
-      "step": 1331
-    },
-    {
-      "epoch": 333.0,
-      "learning_rate": 2.342105263157895e-05,
-      "loss": 0.009,
-      "step": 1332
-    },
-    {
-      "epoch": 333.25,
-      "learning_rate": 2.3399122807017547e-05,
-      "loss": 0.0085,
-      "step": 1333
-    },
-    {
-      "epoch": 333.5,
-      "learning_rate": 2.3377192982456143e-05,
-      "loss": 0.0084,
-      "step": 1334
-    },
-    {
-      "epoch": 333.75,
-      "learning_rate": 2.335526315789474e-05,
-      "loss": 0.0086,
-      "step": 1335
-    },
-    {
-      "epoch": 334.0,
-      "learning_rate": 2.3333333333333336e-05,
-      "loss": 0.0096,
-      "step": 1336
-    },
-    {
-      "epoch": 334.25,
-      "learning_rate": 2.3311403508771932e-05,
-      "loss": 0.0089,
-      "step": 1337
-    },
-    {
-      "epoch": 334.5,
-      "learning_rate": 2.328947368421053e-05,
-      "loss": 0.0081,
-      "step": 1338
-    },
-    {
-      "epoch": 334.75,
-      "learning_rate": 2.3267543859649125e-05,
-      "loss": 0.0089,
-      "step": 1339
-    },
-    {
-      "epoch": 335.0,
-      "learning_rate": 2.324561403508772e-05,
-      "loss": 0.008,
-      "step": 1340
-    },
-    {
-      "epoch": 335.0,
-      "eval_accuracy_ELSE": 0.9971214301946822,
-      "eval_accuracy_road": 0.9942045907615196,
-      "eval_accuracy_sidewalk": 0.9045339341030696,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936428683119362,
-      "eval_iou_road": 0.9876510029139672,
-      "eval_iou_sidewalk": 0.844596371285827,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02121923491358757,
-      "eval_mean_accuracy": 0.9652866516864238,
-      "eval_mean_iou": 0.9419634141705768,
-      "eval_overall_accuracy": 0.9938316345214844,
-      "eval_runtime": 0.6519,
-      "eval_samples_per_second": 1.534,
-      "eval_steps_per_second": 1.534,
-      "step": 1340
-    },
-    {
-      "epoch": 335.25,
-      "learning_rate": 2.3223684210526318e-05,
-      "loss": 0.0075,
-      "step": 1341
-    },
-    {
-      "epoch": 335.5,
-      "learning_rate": 2.3201754385964914e-05,
-      "loss": 0.0089,
-      "step": 1342
-    },
-    {
-      "epoch": 335.75,
-      "learning_rate": 2.317982456140351e-05,
-      "loss": 0.0081,
-      "step": 1343
-    },
-    {
-      "epoch": 336.0,
-      "learning_rate": 2.3157894736842107e-05,
-      "loss": 0.0092,
-      "step": 1344
-    },
-    {
-      "epoch": 336.25,
-      "learning_rate": 2.3135964912280703e-05,
-      "loss": 0.0087,
-      "step": 1345
-    },
-    {
-      "epoch": 336.5,
-      "learning_rate": 2.31140350877193e-05,
-      "loss": 0.0085,
-      "step": 1346
-    },
-    {
-      "epoch": 336.75,
-      "learning_rate": 2.3092105263157896e-05,
-      "loss": 0.0085,
-      "step": 1347
-    },
-    {
-      "epoch": 337.0,
-      "learning_rate": 2.3070175438596492e-05,
-      "loss": 0.0069,
-      "step": 1348
-    },
-    {
-      "epoch": 337.25,
-      "learning_rate": 2.304824561403509e-05,
-      "loss": 0.0078,
-      "step": 1349
-    },
-    {
-      "epoch": 337.5,
-      "learning_rate": 2.3026315789473685e-05,
-      "loss": 0.009,
-      "step": 1350
-    },
-    {
-      "epoch": 337.75,
-      "learning_rate": 2.300438596491228e-05,
-      "loss": 0.0078,
-      "step": 1351
-    },
-    {
-      "epoch": 338.0,
-      "learning_rate": 2.2982456140350878e-05,
-      "loss": 0.0076,
-      "step": 1352
-    },
-    {
-      "epoch": 338.25,
-      "learning_rate": 2.2960526315789474e-05,
-      "loss": 0.0086,
-      "step": 1353
-    },
-    {
-      "epoch": 338.5,
-      "learning_rate": 2.293859649122807e-05,
-      "loss": 0.0081,
-      "step": 1354
-    },
-    {
-      "epoch": 338.75,
-      "learning_rate": 2.2916666666666667e-05,
-      "loss": 0.0079,
-      "step": 1355
-    },
-    {
-      "epoch": 339.0,
-      "learning_rate": 2.2894736842105263e-05,
-      "loss": 0.0083,
-      "step": 1356
-    },
-    {
-      "epoch": 339.25,
-      "learning_rate": 2.287280701754386e-05,
-      "loss": 0.0079,
-      "step": 1357
-    },
-    {
-      "epoch": 339.5,
-      "learning_rate": 2.2850877192982456e-05,
-      "loss": 0.0082,
-      "step": 1358
-    },
-    {
-      "epoch": 339.75,
-      "learning_rate": 2.2828947368421052e-05,
-      "loss": 0.0083,
-      "step": 1359
-    },
-    {
-      "epoch": 340.0,
-      "learning_rate": 2.280701754385965e-05,
-      "loss": 0.0076,
-      "step": 1360
-    },
-    {
-      "epoch": 340.0,
-      "eval_accuracy_ELSE": 0.9971755386496693,
-      "eval_accuracy_road": 0.9944608987868087,
-      "eval_accuracy_sidewalk": 0.9041115178822867,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937718054214951,
-      "eval_iou_road": 0.9877100186683261,
-      "eval_iou_sidewalk": 0.8469858857670493,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021072784438729286,
-      "eval_mean_accuracy": 0.9652493184395882,
-      "eval_mean_iou": 0.9428225699522902,
-      "eval_overall_accuracy": 0.993927001953125,
-      "eval_runtime": 0.6757,
-      "eval_samples_per_second": 1.48,
-      "eval_steps_per_second": 1.48,
-      "step": 1360
-    },
-    {
-      "epoch": 340.25,
-      "learning_rate": 2.2785087719298245e-05,
-      "loss": 0.0086,
-      "step": 1361
-    },
-    {
-      "epoch": 340.5,
-      "learning_rate": 2.2763157894736845e-05,
-      "loss": 0.0076,
-      "step": 1362
-    },
-    {
-      "epoch": 340.75,
-      "learning_rate": 2.274122807017544e-05,
-      "loss": 0.008,
-      "step": 1363
-    },
-    {
-      "epoch": 341.0,
-      "learning_rate": 2.2719298245614038e-05,
-      "loss": 0.0075,
-      "step": 1364
-    },
-    {
-      "epoch": 341.25,
-      "learning_rate": 2.2697368421052634e-05,
-      "loss": 0.0096,
-      "step": 1365
-    },
-    {
-      "epoch": 341.5,
-      "learning_rate": 2.267543859649123e-05,
-      "loss": 0.0083,
-      "step": 1366
-    },
-    {
-      "epoch": 341.75,
-      "learning_rate": 2.2653508771929827e-05,
-      "loss": 0.0084,
-      "step": 1367
-    },
-    {
-      "epoch": 342.0,
-      "learning_rate": 2.2631578947368423e-05,
-      "loss": 0.0075,
-      "step": 1368
-    },
-    {
-      "epoch": 342.25,
-      "learning_rate": 2.260964912280702e-05,
-      "loss": 0.0076,
-      "step": 1369
-    },
-    {
-      "epoch": 342.5,
-      "learning_rate": 2.2587719298245616e-05,
-      "loss": 0.0088,
-      "step": 1370
-    },
-    {
-      "epoch": 342.75,
-      "learning_rate": 2.2565789473684212e-05,
-      "loss": 0.0081,
-      "step": 1371
-    },
-    {
-      "epoch": 343.0,
-      "learning_rate": 2.254385964912281e-05,
-      "loss": 0.0082,
-      "step": 1372
-    },
-    {
-      "epoch": 343.25,
-      "learning_rate": 2.2521929824561405e-05,
-      "loss": 0.0088,
-      "step": 1373
-    },
-    {
-      "epoch": 343.5,
-      "learning_rate": 2.25e-05,
-      "loss": 0.0083,
-      "step": 1374
-    },
-    {
-      "epoch": 343.75,
-      "learning_rate": 2.2478070175438598e-05,
-      "loss": 0.0075,
-      "step": 1375
-    },
-    {
-      "epoch": 344.0,
-      "learning_rate": 2.2456140350877194e-05,
-      "loss": 0.0077,
-      "step": 1376
-    },
-    {
-      "epoch": 344.25,
-      "learning_rate": 2.243421052631579e-05,
-      "loss": 0.0079,
-      "step": 1377
-    },
-    {
-      "epoch": 344.5,
-      "learning_rate": 2.2412280701754387e-05,
-      "loss": 0.0077,
-      "step": 1378
-    },
-    {
-      "epoch": 344.75,
-      "learning_rate": 2.2390350877192983e-05,
-      "loss": 0.0078,
-      "step": 1379
-    },
-    {
-      "epoch": 345.0,
-      "learning_rate": 2.236842105263158e-05,
-      "loss": 0.0088,
-      "step": 1380
-    },
-    {
-      "epoch": 345.0,
-      "eval_accuracy_ELSE": 0.9970565000486976,
-      "eval_accuracy_road": 0.9947884034857891,
-      "eval_accuracy_sidewalk": 0.904393128696142,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937228340002373,
-      "eval_iou_road": 0.9879514664705716,
-      "eval_iou_sidewalk": 0.846802900461437,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021011579781770706,
-      "eval_mean_accuracy": 0.9654126774102095,
-      "eval_mean_iou": 0.942825733644082,
-      "eval_overall_accuracy": 0.9939384460449219,
-      "eval_runtime": 0.6305,
-      "eval_samples_per_second": 1.586,
-      "eval_steps_per_second": 1.586,
-      "step": 1380
-    },
-    {
-      "epoch": 345.25,
-      "learning_rate": 2.2346491228070176e-05,
-      "loss": 0.0072,
-      "step": 1381
-    },
-    {
-      "epoch": 345.5,
-      "learning_rate": 2.2324561403508772e-05,
-      "loss": 0.0077,
-      "step": 1382
-    },
-    {
-      "epoch": 345.75,
-      "learning_rate": 2.230263157894737e-05,
-      "loss": 0.0079,
-      "step": 1383
-    },
-    {
-      "epoch": 346.0,
-      "learning_rate": 2.2280701754385965e-05,
-      "loss": 0.0082,
-      "step": 1384
-    },
-    {
-      "epoch": 346.25,
-      "learning_rate": 2.225877192982456e-05,
-      "loss": 0.008,
-      "step": 1385
-    },
-    {
-      "epoch": 346.5,
-      "learning_rate": 2.2236842105263158e-05,
-      "loss": 0.0089,
-      "step": 1386
-    },
-    {
-      "epoch": 346.75,
-      "learning_rate": 2.2214912280701754e-05,
-      "loss": 0.0081,
-      "step": 1387
-    },
-    {
-      "epoch": 347.0,
-      "learning_rate": 2.219298245614035e-05,
-      "loss": 0.0084,
-      "step": 1388
-    },
-    {
-      "epoch": 347.25,
-      "learning_rate": 2.2171052631578947e-05,
-      "loss": 0.0074,
-      "step": 1389
-    },
-    {
-      "epoch": 347.5,
-      "learning_rate": 2.2149122807017543e-05,
-      "loss": 0.0085,
-      "step": 1390
-    },
-    {
-      "epoch": 347.75,
-      "learning_rate": 2.212719298245614e-05,
-      "loss": 0.0088,
-      "step": 1391
-    },
-    {
-      "epoch": 348.0,
-      "learning_rate": 2.2105263157894736e-05,
-      "loss": 0.0079,
-      "step": 1392
-    },
-    {
-      "epoch": 348.25,
-      "learning_rate": 2.2083333333333333e-05,
-      "loss": 0.0075,
-      "step": 1393
-    },
-    {
-      "epoch": 348.5,
-      "learning_rate": 2.206140350877193e-05,
-      "loss": 0.0079,
-      "step": 1394
-    },
-    {
-      "epoch": 348.75,
-      "learning_rate": 2.2039473684210525e-05,
-      "loss": 0.0077,
-      "step": 1395
-    },
-    {
-      "epoch": 349.0,
-      "learning_rate": 2.201754385964912e-05,
-      "loss": 0.0086,
-      "step": 1396
-    },
-    {
-      "epoch": 349.25,
-      "learning_rate": 2.1995614035087718e-05,
-      "loss": 0.0078,
-      "step": 1397
-    },
-    {
-      "epoch": 349.5,
-      "learning_rate": 2.1973684210526314e-05,
-      "loss": 0.0085,
-      "step": 1398
-    },
-    {
-      "epoch": 349.75,
-      "learning_rate": 2.195175438596491e-05,
-      "loss": 0.0079,
-      "step": 1399
-    },
-    {
-      "epoch": 350.0,
-      "learning_rate": 2.1929824561403507e-05,
-      "loss": 0.0084,
-      "step": 1400
-    },
-    {
-      "epoch": 350.0,
-      "eval_accuracy_ELSE": 0.996818422846754,
-      "eval_accuracy_road": 0.9943469841089024,
-      "eval_accuracy_sidewalk": 0.9067868206139116,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993431979465499,
-      "eval_iou_road": 0.9876667185267952,
-      "eval_iou_sidewalk": 0.8424908424908425,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0213001798838377,
-      "eval_mean_accuracy": 0.9659840758565226,
-      "eval_mean_iou": 0.9411965134943788,
-      "eval_overall_accuracy": 0.9937171936035156,
-      "eval_runtime": 0.6126,
-      "eval_samples_per_second": 1.632,
-      "eval_steps_per_second": 1.632,
-      "step": 1400
-    },
-    {
-      "epoch": 350.25,
-      "learning_rate": 2.1907894736842104e-05,
-      "loss": 0.0089,
-      "step": 1401
-    },
-    {
-      "epoch": 350.5,
-      "learning_rate": 2.18859649122807e-05,
-      "loss": 0.0071,
-      "step": 1402
-    },
-    {
-      "epoch": 350.75,
-      "learning_rate": 2.18640350877193e-05,
-      "loss": 0.0077,
-      "step": 1403
-    },
-    {
-      "epoch": 351.0,
-      "learning_rate": 2.1842105263157896e-05,
-      "loss": 0.0079,
-      "step": 1404
-    },
-    {
-      "epoch": 351.25,
-      "learning_rate": 2.1820175438596492e-05,
-      "loss": 0.0073,
-      "step": 1405
-    },
-    {
-      "epoch": 351.5,
-      "learning_rate": 2.179824561403509e-05,
-      "loss": 0.0073,
-      "step": 1406
-    },
-    {
-      "epoch": 351.75,
-      "learning_rate": 2.1776315789473685e-05,
-      "loss": 0.0091,
-      "step": 1407
-    },
-    {
-      "epoch": 352.0,
-      "learning_rate": 2.175438596491228e-05,
-      "loss": 0.008,
-      "step": 1408
-    },
-    {
-      "epoch": 352.25,
-      "learning_rate": 2.1732456140350878e-05,
-      "loss": 0.0075,
-      "step": 1409
-    },
-    {
-      "epoch": 352.5,
-      "learning_rate": 2.1710526315789474e-05,
-      "loss": 0.0088,
-      "step": 1410
-    },
-    {
-      "epoch": 352.75,
-      "learning_rate": 2.168859649122807e-05,
-      "loss": 0.0085,
-      "step": 1411
-    },
-    {
-      "epoch": 353.0,
-      "learning_rate": 2.1666666666666667e-05,
-      "loss": 0.0077,
-      "step": 1412
-    },
-    {
-      "epoch": 353.25,
-      "learning_rate": 2.1644736842105263e-05,
-      "loss": 0.008,
-      "step": 1413
-    },
-    {
-      "epoch": 353.5,
-      "learning_rate": 2.162280701754386e-05,
-      "loss": 0.0083,
-      "step": 1414
-    },
-    {
-      "epoch": 353.75,
-      "learning_rate": 2.1600877192982456e-05,
-      "loss": 0.0072,
-      "step": 1415
-    },
-    {
-      "epoch": 354.0,
-      "learning_rate": 2.1578947368421053e-05,
-      "loss": 0.0078,
-      "step": 1416
-    },
-    {
-      "epoch": 354.25,
-      "learning_rate": 2.155701754385965e-05,
-      "loss": 0.0087,
-      "step": 1417
-    },
-    {
-      "epoch": 354.5,
-      "learning_rate": 2.153508771929825e-05,
-      "loss": 0.0076,
-      "step": 1418
-    },
-    {
-      "epoch": 354.75,
-      "learning_rate": 2.1513157894736845e-05,
-      "loss": 0.008,
-      "step": 1419
-    },
-    {
-      "epoch": 355.0,
-      "learning_rate": 2.149122807017544e-05,
-      "loss": 0.0077,
-      "step": 1420
-    },
-    {
-      "epoch": 355.0,
-      "eval_accuracy_ELSE": 0.9966452757907951,
-      "eval_accuracy_road": 0.9952440621974141,
-      "eval_accuracy_sidewalk": 0.901717825964517,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935005043177147,
-      "eval_iou_road": 0.9877195709622261,
-      "eval_iou_sidewalk": 0.8388787005501703,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021512934938073158,
-      "eval_mean_accuracy": 0.9645357213175755,
-      "eval_mean_iou": 0.9400329252767037,
-      "eval_overall_accuracy": 0.9936981201171875,
-      "eval_runtime": 0.6125,
-      "eval_samples_per_second": 1.633,
-      "eval_steps_per_second": 1.633,
-      "step": 1420
-    },
-    {
-      "epoch": 355.25,
-      "learning_rate": 2.1469298245614038e-05,
-      "loss": 0.0089,
-      "step": 1421
-    },
-    {
-      "epoch": 355.5,
-      "learning_rate": 2.1447368421052634e-05,
-      "loss": 0.0079,
-      "step": 1422
-    },
-    {
-      "epoch": 355.75,
-      "learning_rate": 2.142543859649123e-05,
-      "loss": 0.008,
-      "step": 1423
-    },
-    {
-      "epoch": 356.0,
-      "learning_rate": 2.1403508771929827e-05,
-      "loss": 0.0085,
-      "step": 1424
-    },
-    {
-      "epoch": 356.25,
-      "learning_rate": 2.1381578947368423e-05,
-      "loss": 0.0072,
-      "step": 1425
-    },
-    {
-      "epoch": 356.5,
-      "learning_rate": 2.135964912280702e-05,
-      "loss": 0.0082,
-      "step": 1426
-    },
-    {
-      "epoch": 356.75,
-      "learning_rate": 2.1337719298245616e-05,
-      "loss": 0.0078,
-      "step": 1427
-    },
-    {
-      "epoch": 357.0,
-      "learning_rate": 2.1315789473684212e-05,
-      "loss": 0.0082,
-      "step": 1428
-    },
-    {
-      "epoch": 357.25,
-      "learning_rate": 2.129385964912281e-05,
-      "loss": 0.0085,
-      "step": 1429
-    },
-    {
-      "epoch": 357.5,
-      "learning_rate": 2.1271929824561405e-05,
-      "loss": 0.0086,
-      "step": 1430
-    },
-    {
-      "epoch": 357.75,
-      "learning_rate": 2.125e-05,
-      "loss": 0.008,
-      "step": 1431
-    },
-    {
-      "epoch": 358.0,
-      "learning_rate": 2.1228070175438598e-05,
-      "loss": 0.0074,
-      "step": 1432
-    },
-    {
-      "epoch": 358.25,
-      "learning_rate": 2.1206140350877194e-05,
-      "loss": 0.008,
-      "step": 1433
-    },
-    {
-      "epoch": 358.5,
-      "learning_rate": 2.118421052631579e-05,
-      "loss": 0.0085,
-      "step": 1434
-    },
-    {
-      "epoch": 358.75,
-      "learning_rate": 2.1162280701754387e-05,
-      "loss": 0.0078,
-      "step": 1435
-    },
-    {
-      "epoch": 359.0,
-      "learning_rate": 2.1140350877192983e-05,
-      "loss": 0.0078,
-      "step": 1436
-    },
-    {
-      "epoch": 359.25,
-      "learning_rate": 2.111842105263158e-05,
-      "loss": 0.0082,
-      "step": 1437
-    },
-    {
-      "epoch": 359.5,
-      "learning_rate": 2.1096491228070176e-05,
-      "loss": 0.0081,
-      "step": 1438
-    },
-    {
-      "epoch": 359.75,
-      "learning_rate": 2.1074561403508773e-05,
-      "loss": 0.008,
-      "step": 1439
-    },
-    {
-      "epoch": 360.0,
-      "learning_rate": 2.105263157894737e-05,
-      "loss": 0.0069,
-      "step": 1440
-    },
-    {
-      "epoch": 360.0,
-      "eval_accuracy_ELSE": 0.9970510892031988,
-      "eval_accuracy_road": 0.9936492567067267,
-      "eval_accuracy_sidewalk": 0.9086172909039707,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935888449136732,
-      "eval_iou_road": 0.9873926393389272,
-      "eval_iou_sidewalk": 0.8431987455899648,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020957402884960175,
-      "eval_mean_accuracy": 0.9664392122712987,
-      "eval_mean_iou": 0.9413934099475217,
-      "eval_overall_accuracy": 0.993743896484375,
-      "eval_runtime": 0.605,
-      "eval_samples_per_second": 1.653,
-      "eval_steps_per_second": 1.653,
-      "step": 1440
-    },
-    {
-      "epoch": 360.25,
-      "learning_rate": 2.1030701754385965e-05,
-      "loss": 0.008,
-      "step": 1441
-    },
-    {
-      "epoch": 360.5,
-      "learning_rate": 2.100877192982456e-05,
-      "loss": 0.0081,
-      "step": 1442
-    },
-    {
-      "epoch": 360.75,
-      "learning_rate": 2.0986842105263158e-05,
-      "loss": 0.0099,
-      "step": 1443
-    },
-    {
-      "epoch": 361.0,
-      "learning_rate": 2.0964912280701758e-05,
-      "loss": 0.0078,
-      "step": 1444
-    },
-    {
-      "epoch": 361.25,
-      "learning_rate": 2.0942982456140354e-05,
-      "loss": 0.0079,
-      "step": 1445
-    },
-    {
-      "epoch": 361.5,
-      "learning_rate": 2.092105263157895e-05,
-      "loss": 0.0076,
-      "step": 1446
-    },
-    {
-      "epoch": 361.75,
-      "learning_rate": 2.0899122807017547e-05,
-      "loss": 0.0076,
-      "step": 1447
-    },
-    {
-      "epoch": 362.0,
-      "learning_rate": 2.0877192982456143e-05,
-      "loss": 0.009,
-      "step": 1448
-    },
-    {
-      "epoch": 362.25,
-      "learning_rate": 2.085526315789474e-05,
-      "loss": 0.0082,
-      "step": 1449
-    },
-    {
-      "epoch": 362.5,
-      "learning_rate": 2.0833333333333336e-05,
-      "loss": 0.008,
-      "step": 1450
-    },
-    {
-      "epoch": 362.75,
-      "learning_rate": 2.0811403508771932e-05,
-      "loss": 0.0079,
-      "step": 1451
-    },
-    {
-      "epoch": 363.0,
-      "learning_rate": 2.078947368421053e-05,
-      "loss": 0.0081,
-      "step": 1452
-    },
-    {
-      "epoch": 363.25,
-      "learning_rate": 2.0767543859649125e-05,
-      "loss": 0.0084,
-      "step": 1453
-    },
-    {
-      "epoch": 363.5,
-      "learning_rate": 2.074561403508772e-05,
-      "loss": 0.0077,
-      "step": 1454
-    },
-    {
-      "epoch": 363.75,
-      "learning_rate": 2.0723684210526318e-05,
-      "loss": 0.0081,
-      "step": 1455
-    },
-    {
-      "epoch": 364.0,
-      "learning_rate": 2.0701754385964914e-05,
-      "loss": 0.0082,
-      "step": 1456
-    },
-    {
-      "epoch": 364.25,
-      "learning_rate": 2.067982456140351e-05,
-      "loss": 0.0067,
-      "step": 1457
-    },
-    {
-      "epoch": 364.5,
-      "learning_rate": 2.0657894736842107e-05,
-      "loss": 0.0078,
-      "step": 1458
-    },
-    {
-      "epoch": 364.75,
-      "learning_rate": 2.0635964912280703e-05,
-      "loss": 0.0076,
-      "step": 1459
-    },
-    {
-      "epoch": 365.0,
-      "learning_rate": 2.06140350877193e-05,
-      "loss": 0.0071,
-      "step": 1460
-    },
-    {
-      "epoch": 365.0,
-      "eval_accuracy_ELSE": 0.9975056002250912,
-      "eval_accuracy_road": 0.9929942473087657,
-      "eval_accuracy_sidewalk": 0.9083356800901154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936613683036075,
-      "eval_iou_road": 0.9873984085181094,
-      "eval_iou_sidewalk": 0.8495983142367971,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02040543593466282,
-      "eval_mean_accuracy": 0.9662785092079909,
-      "eval_mean_iou": 0.9435526970195047,
-      "eval_overall_accuracy": 0.9938812255859375,
-      "eval_runtime": 0.6059,
-      "eval_samples_per_second": 1.65,
-      "eval_steps_per_second": 1.65,
-      "step": 1460
-    },
-    {
-      "epoch": 365.25,
-      "learning_rate": 2.0592105263157896e-05,
-      "loss": 0.0078,
-      "step": 1461
-    },
-    {
-      "epoch": 365.5,
-      "learning_rate": 2.0570175438596493e-05,
-      "loss": 0.0084,
-      "step": 1462
-    },
-    {
-      "epoch": 365.75,
-      "learning_rate": 2.054824561403509e-05,
-      "loss": 0.0084,
-      "step": 1463
-    },
-    {
-      "epoch": 366.0,
-      "learning_rate": 2.0526315789473685e-05,
-      "loss": 0.008,
-      "step": 1464
-    },
-    {
-      "epoch": 366.25,
-      "learning_rate": 2.050438596491228e-05,
-      "loss": 0.0078,
-      "step": 1465
-    },
-    {
-      "epoch": 366.5,
-      "learning_rate": 2.0482456140350878e-05,
-      "loss": 0.0078,
-      "step": 1466
-    },
-    {
-      "epoch": 366.75,
-      "learning_rate": 2.0460526315789474e-05,
-      "loss": 0.0088,
-      "step": 1467
-    },
-    {
-      "epoch": 367.0,
-      "learning_rate": 2.043859649122807e-05,
-      "loss": 0.0083,
-      "step": 1468
-    },
-    {
-      "epoch": 367.25,
-      "learning_rate": 2.0416666666666667e-05,
-      "loss": 0.0079,
-      "step": 1469
-    },
-    {
-      "epoch": 367.5,
-      "learning_rate": 2.0394736842105264e-05,
-      "loss": 0.009,
-      "step": 1470
-    },
-    {
-      "epoch": 367.75,
-      "learning_rate": 2.037280701754386e-05,
-      "loss": 0.0074,
-      "step": 1471
-    },
-    {
-      "epoch": 368.0,
-      "learning_rate": 2.0350877192982456e-05,
-      "loss": 0.008,
-      "step": 1472
-    },
-    {
-      "epoch": 368.25,
-      "learning_rate": 2.0328947368421053e-05,
-      "loss": 0.0078,
-      "step": 1473
-    },
-    {
-      "epoch": 368.5,
-      "learning_rate": 2.030701754385965e-05,
-      "loss": 0.0068,
-      "step": 1474
-    },
-    {
-      "epoch": 368.75,
-      "learning_rate": 2.0285087719298245e-05,
-      "loss": 0.0083,
-      "step": 1475
-    },
-    {
-      "epoch": 369.0,
-      "learning_rate": 2.0263157894736842e-05,
-      "loss": 0.0083,
-      "step": 1476
-    },
-    {
-      "epoch": 369.25,
-      "learning_rate": 2.0241228070175438e-05,
-      "loss": 0.0078,
-      "step": 1477
-    },
-    {
-      "epoch": 369.5,
-      "learning_rate": 2.0219298245614035e-05,
-      "loss": 0.0081,
-      "step": 1478
-    },
-    {
-      "epoch": 369.75,
-      "learning_rate": 2.019736842105263e-05,
-      "loss": 0.0076,
-      "step": 1479
-    },
-    {
-      "epoch": 370.0,
-      "learning_rate": 2.0175438596491227e-05,
-      "loss": 0.0079,
-      "step": 1480
-    },
-    {
-      "epoch": 370.0,
-      "eval_accuracy_ELSE": 0.9973649182421245,
-      "eval_accuracy_road": 0.9934783846898673,
-      "eval_accuracy_sidewalk": 0.9072092368346945,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937354775754896,
-      "eval_iou_road": 0.9876141269728926,
-      "eval_iou_sidewalk": 0.8462043603887576,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020821254700422287,
-      "eval_mean_accuracy": 0.9660175132555621,
-      "eval_mean_iou": 0.94251798831238,
-      "eval_overall_accuracy": 0.9938812255859375,
-      "eval_runtime": 0.5982,
-      "eval_samples_per_second": 1.672,
-      "eval_steps_per_second": 1.672,
-      "step": 1480
-    },
-    {
-      "epoch": 370.25,
-      "learning_rate": 2.0153508771929824e-05,
-      "loss": 0.0076,
-      "step": 1481
-    },
-    {
-      "epoch": 370.5,
-      "learning_rate": 2.013157894736842e-05,
-      "loss": 0.0086,
-      "step": 1482
-    },
-    {
-      "epoch": 370.75,
-      "learning_rate": 2.0109649122807016e-05,
-      "loss": 0.0072,
-      "step": 1483
-    },
-    {
-      "epoch": 371.0,
-      "learning_rate": 2.0087719298245616e-05,
-      "loss": 0.0076,
-      "step": 1484
-    },
-    {
-      "epoch": 371.25,
-      "learning_rate": 2.0065789473684213e-05,
-      "loss": 0.0082,
-      "step": 1485
-    },
-    {
-      "epoch": 371.5,
-      "learning_rate": 2.004385964912281e-05,
-      "loss": 0.0074,
-      "step": 1486
-    },
-    {
-      "epoch": 371.75,
-      "learning_rate": 2.0021929824561405e-05,
-      "loss": 0.0082,
-      "step": 1487
-    },
-    {
-      "epoch": 372.0,
-      "learning_rate": 2e-05,
-      "loss": 0.0079,
-      "step": 1488
-    },
-    {
-      "epoch": 372.25,
-      "learning_rate": 1.9978070175438598e-05,
-      "loss": 0.0079,
-      "step": 1489
-    },
-    {
-      "epoch": 372.5,
-      "learning_rate": 1.9956140350877194e-05,
-      "loss": 0.0084,
-      "step": 1490
-    },
-    {
-      "epoch": 372.75,
-      "learning_rate": 1.993421052631579e-05,
-      "loss": 0.0091,
-      "step": 1491
-    },
-    {
-      "epoch": 373.0,
-      "learning_rate": 1.9912280701754387e-05,
-      "loss": 0.0071,
-      "step": 1492
-    },
-    {
-      "epoch": 373.25,
-      "learning_rate": 1.9890350877192984e-05,
-      "loss": 0.0075,
-      "step": 1493
-    },
-    {
-      "epoch": 373.5,
-      "learning_rate": 1.986842105263158e-05,
-      "loss": 0.0078,
-      "step": 1494
-    },
-    {
-      "epoch": 373.75,
-      "learning_rate": 1.9846491228070176e-05,
-      "loss": 0.0078,
-      "step": 1495
-    },
-    {
-      "epoch": 374.0,
-      "learning_rate": 1.9824561403508773e-05,
-      "loss": 0.0079,
-      "step": 1496
-    },
-    {
-      "epoch": 374.25,
-      "learning_rate": 1.980263157894737e-05,
-      "loss": 0.0082,
-      "step": 1497
-    },
-    {
-      "epoch": 374.5,
-      "learning_rate": 1.9780701754385965e-05,
-      "loss": 0.0082,
-      "step": 1498
-    },
-    {
-      "epoch": 374.75,
-      "learning_rate": 1.9758771929824562e-05,
-      "loss": 0.0072,
-      "step": 1499
-    },
-    {
-      "epoch": 375.0,
-      "learning_rate": 1.9736842105263158e-05,
-      "loss": 0.0085,
-      "step": 1500
-    },
-    {
-      "epoch": 375.0,
-      "eval_accuracy_ELSE": 0.9972621121776489,
-      "eval_accuracy_road": 0.993364470011961,
-      "eval_accuracy_sidewalk": 0.907772458462405,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993568767823354,
-      "eval_iou_road": 0.9875428215509188,
-      "eval_iou_sidewalk": 0.8451756685894074,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02099161222577095,
-      "eval_mean_accuracy": 0.9661330135506717,
-      "eval_mean_iou": 0.9420957526545601,
-      "eval_overall_accuracy": 0.9937934875488281,
-      "eval_runtime": 0.6019,
-      "eval_samples_per_second": 1.661,
-      "eval_steps_per_second": 1.661,
-      "step": 1500
-    },
-    {
-      "epoch": 375.25,
-      "learning_rate": 1.9714912280701755e-05,
-      "loss": 0.0085,
-      "step": 1501
-    },
-    {
-      "epoch": 375.5,
-      "learning_rate": 1.969298245614035e-05,
-      "loss": 0.0076,
-      "step": 1502
-    },
-    {
-      "epoch": 375.75,
-      "learning_rate": 1.9671052631578947e-05,
-      "loss": 0.0068,
-      "step": 1503
-    },
-    {
-      "epoch": 376.0,
-      "learning_rate": 1.9649122807017544e-05,
-      "loss": 0.0072,
-      "step": 1504
-    },
-    {
-      "epoch": 376.25,
-      "learning_rate": 1.962719298245614e-05,
-      "loss": 0.007,
-      "step": 1505
-    },
-    {
-      "epoch": 376.5,
-      "learning_rate": 1.9605263157894736e-05,
-      "loss": 0.0076,
-      "step": 1506
-    },
-    {
-      "epoch": 376.75,
-      "learning_rate": 1.9583333333333333e-05,
-      "loss": 0.0071,
-      "step": 1507
-    },
-    {
-      "epoch": 377.0,
-      "learning_rate": 1.956140350877193e-05,
-      "loss": 0.0078,
-      "step": 1508
-    },
-    {
-      "epoch": 377.25,
-      "learning_rate": 1.9539473684210525e-05,
-      "loss": 0.0073,
-      "step": 1509
-    },
-    {
-      "epoch": 377.5,
-      "learning_rate": 1.9517543859649122e-05,
-      "loss": 0.0071,
-      "step": 1510
-    },
-    {
-      "epoch": 377.75,
-      "learning_rate": 1.9495614035087718e-05,
-      "loss": 0.0076,
-      "step": 1511
-    },
-    {
-      "epoch": 378.0,
-      "learning_rate": 1.9473684210526315e-05,
-      "loss": 0.0084,
-      "step": 1512
-    },
-    {
-      "epoch": 378.25,
-      "learning_rate": 1.945175438596491e-05,
-      "loss": 0.0079,
-      "step": 1513
-    },
-    {
-      "epoch": 378.5,
-      "learning_rate": 1.9429824561403507e-05,
-      "loss": 0.0082,
-      "step": 1514
-    },
-    {
-      "epoch": 378.75,
-      "learning_rate": 1.9407894736842107e-05,
-      "loss": 0.0082,
-      "step": 1515
-    },
-    {
-      "epoch": 379.0,
-      "learning_rate": 1.9385964912280704e-05,
-      "loss": 0.0069,
-      "step": 1516
-    },
-    {
-      "epoch": 379.25,
-      "learning_rate": 1.93640350877193e-05,
-      "loss": 0.0078,
-      "step": 1517
-    },
-    {
-      "epoch": 379.5,
-      "learning_rate": 1.9342105263157896e-05,
-      "loss": 0.0067,
-      "step": 1518
-    },
-    {
-      "epoch": 379.75,
-      "learning_rate": 1.9320175438596493e-05,
-      "loss": 0.0079,
-      "step": 1519
-    },
-    {
-      "epoch": 380.0,
-      "learning_rate": 1.929824561403509e-05,
-      "loss": 0.0086,
-      "step": 1520
-    },
-    {
-      "epoch": 380.0,
-      "eval_accuracy_ELSE": 0.9972729338686463,
-      "eval_accuracy_road": 0.9935922993677735,
-      "eval_accuracy_sidewalk": 0.907772458462405,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936277575313222,
-      "eval_iou_road": 0.9877693157045384,
-      "eval_iou_sidewalk": 0.8461740385877412,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.020808037370443344,
-      "eval_mean_accuracy": 0.9662125638996083,
-      "eval_mean_iou": 0.9425237039412006,
-      "eval_overall_accuracy": 0.9938621520996094,
-      "eval_runtime": 0.6048,
-      "eval_samples_per_second": 1.653,
-      "eval_steps_per_second": 1.653,
-      "step": 1520
-    },
-    {
-      "epoch": 380.25,
-      "learning_rate": 1.9276315789473685e-05,
-      "loss": 0.0073,
-      "step": 1521
-    },
-    {
-      "epoch": 380.5,
-      "learning_rate": 1.9254385964912282e-05,
-      "loss": 0.0079,
-      "step": 1522
-    },
-    {
-      "epoch": 380.75,
-      "learning_rate": 1.9232456140350878e-05,
-      "loss": 0.0075,
-      "step": 1523
-    },
-    {
-      "epoch": 381.0,
-      "learning_rate": 1.9210526315789474e-05,
-      "loss": 0.0074,
-      "step": 1524
-    },
-    {
-      "epoch": 381.25,
-      "learning_rate": 1.918859649122807e-05,
-      "loss": 0.0077,
-      "step": 1525
-    },
-    {
-      "epoch": 381.5,
-      "learning_rate": 1.9166666666666667e-05,
-      "loss": 0.0067,
-      "step": 1526
-    },
-    {
-      "epoch": 381.75,
-      "learning_rate": 1.9144736842105264e-05,
-      "loss": 0.0074,
-      "step": 1527
-    },
-    {
-      "epoch": 382.0,
-      "learning_rate": 1.9122807017543863e-05,
-      "loss": 0.0084,
-      "step": 1528
-    },
-    {
-      "epoch": 382.25,
-      "learning_rate": 1.910087719298246e-05,
-      "loss": 0.0079,
-      "step": 1529
-    },
-    {
-      "epoch": 382.5,
-      "learning_rate": 1.9078947368421056e-05,
-      "loss": 0.0075,
-      "step": 1530
-    },
-    {
-      "epoch": 382.75,
-      "learning_rate": 1.9057017543859653e-05,
-      "loss": 0.0071,
-      "step": 1531
-    },
-    {
-      "epoch": 383.0,
-      "learning_rate": 1.903508771929825e-05,
-      "loss": 0.008,
-      "step": 1532
-    },
-    {
-      "epoch": 383.25,
-      "learning_rate": 1.9013157894736845e-05,
-      "loss": 0.0079,
-      "step": 1533
-    },
-    {
-      "epoch": 383.5,
-      "learning_rate": 1.899122807017544e-05,
-      "loss": 0.0083,
-      "step": 1534
-    },
-    {
-      "epoch": 383.75,
-      "learning_rate": 1.8969298245614038e-05,
-      "loss": 0.0071,
-      "step": 1535
-    },
-    {
-      "epoch": 384.0,
-      "learning_rate": 1.8947368421052634e-05,
-      "loss": 0.0067,
-      "step": 1536
-    },
-    {
-      "epoch": 384.25,
-      "learning_rate": 1.892543859649123e-05,
-      "loss": 0.0075,
-      "step": 1537
-    },
-    {
-      "epoch": 384.5,
-      "learning_rate": 1.8903508771929827e-05,
-      "loss": 0.0079,
-      "step": 1538
-    },
-    {
-      "epoch": 384.75,
-      "learning_rate": 1.8881578947368423e-05,
-      "loss": 0.0084,
-      "step": 1539
-    },
-    {
-      "epoch": 385.0,
-      "learning_rate": 1.885964912280702e-05,
-      "loss": 0.0076,
-      "step": 1540
-    },
-    {
-      "epoch": 385.0,
-      "eval_accuracy_ELSE": 0.9971268410401809,
-      "eval_accuracy_road": 0.9932932733382697,
-      "eval_accuracy_sidewalk": 0.9062235989862011,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933054844360598,
-      "eval_iou_road": 0.987458063785513,
-      "eval_iou_sidewalk": 0.8419675562532706,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02179449051618576,
-      "eval_mean_accuracy": 0.9655479044548838,
-      "eval_mean_iou": 0.940910368158281,
-      "eval_overall_accuracy": 0.9936370849609375,
-      "eval_runtime": 0.6438,
-      "eval_samples_per_second": 1.553,
-      "eval_steps_per_second": 1.553,
-      "step": 1540
-    },
-    {
-      "epoch": 385.25,
-      "learning_rate": 1.8837719298245616e-05,
-      "loss": 0.0076,
-      "step": 1541
-    },
-    {
-      "epoch": 385.5,
-      "learning_rate": 1.8815789473684213e-05,
-      "loss": 0.0076,
-      "step": 1542
-    },
-    {
-      "epoch": 385.75,
-      "learning_rate": 1.879385964912281e-05,
-      "loss": 0.0073,
-      "step": 1543
-    },
-    {
-      "epoch": 386.0,
-      "learning_rate": 1.8771929824561405e-05,
-      "loss": 0.0079,
-      "step": 1544
-    },
-    {
-      "epoch": 386.25,
-      "learning_rate": 1.8750000000000002e-05,
-      "loss": 0.008,
-      "step": 1545
-    },
-    {
-      "epoch": 386.5,
-      "learning_rate": 1.8728070175438598e-05,
-      "loss": 0.0073,
-      "step": 1546
-    },
-    {
-      "epoch": 386.75,
-      "learning_rate": 1.8706140350877194e-05,
-      "loss": 0.0076,
-      "step": 1547
-    },
-    {
-      "epoch": 387.0,
-      "learning_rate": 1.868421052631579e-05,
-      "loss": 0.0081,
-      "step": 1548
-    },
-    {
-      "epoch": 387.25,
-      "learning_rate": 1.8662280701754387e-05,
-      "loss": 0.0082,
-      "step": 1549
-    },
-    {
-      "epoch": 387.5,
-      "learning_rate": 1.8640350877192984e-05,
-      "loss": 0.0072,
-      "step": 1550
-    },
-    {
-      "epoch": 387.75,
-      "learning_rate": 1.861842105263158e-05,
-      "loss": 0.0077,
-      "step": 1551
-    },
-    {
-      "epoch": 388.0,
-      "learning_rate": 1.8596491228070176e-05,
-      "loss": 0.009,
-      "step": 1552
-    },
-    {
-      "epoch": 388.25,
-      "learning_rate": 1.8574561403508773e-05,
-      "loss": 0.0081,
-      "step": 1553
-    },
-    {
-      "epoch": 388.5,
-      "learning_rate": 1.855263157894737e-05,
-      "loss": 0.0076,
-      "step": 1554
-    },
-    {
-      "epoch": 388.75,
-      "learning_rate": 1.8530701754385965e-05,
-      "loss": 0.0069,
-      "step": 1555
-    },
-    {
-      "epoch": 389.0,
-      "learning_rate": 1.8508771929824562e-05,
-      "loss": 0.0074,
-      "step": 1556
-    },
-    {
-      "epoch": 389.25,
-      "learning_rate": 1.8486842105263158e-05,
-      "loss": 0.0078,
-      "step": 1557
-    },
-    {
-      "epoch": 389.5,
-      "learning_rate": 1.8464912280701755e-05,
-      "loss": 0.0079,
-      "step": 1558
-    },
-    {
-      "epoch": 389.75,
-      "learning_rate": 1.844298245614035e-05,
-      "loss": 0.0065,
-      "step": 1559
-    },
-    {
-      "epoch": 390.0,
-      "learning_rate": 1.8421052631578947e-05,
-      "loss": 0.0077,
-      "step": 1560
-    },
-    {
-      "epoch": 390.0,
-      "eval_accuracy_ELSE": 0.9969482831387233,
-      "eval_accuracy_road": 0.9945178561257618,
-      "eval_accuracy_sidewalk": 0.9052379611377077,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935078240426197,
-      "eval_iou_road": 0.9879202795026663,
-      "eval_iou_sidewalk": 0.8442547603414314,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02136794663965702,
-      "eval_mean_accuracy": 0.9655680334673976,
-      "eval_mean_iou": 0.9418942879622391,
-      "eval_overall_accuracy": 0.9938125610351562,
-      "eval_runtime": 0.596,
-      "eval_samples_per_second": 1.678,
-      "eval_steps_per_second": 1.678,
-      "step": 1560
-    },
-    {
-      "epoch": 390.25,
-      "learning_rate": 1.8399122807017544e-05,
-      "loss": 0.0073,
-      "step": 1561
-    },
-    {
-      "epoch": 390.5,
-      "learning_rate": 1.837719298245614e-05,
-      "loss": 0.0076,
-      "step": 1562
-    },
-    {
-      "epoch": 390.75,
-      "learning_rate": 1.8355263157894736e-05,
-      "loss": 0.008,
-      "step": 1563
-    },
-    {
-      "epoch": 391.0,
-      "learning_rate": 1.8333333333333333e-05,
-      "loss": 0.0081,
-      "step": 1564
-    },
-    {
-      "epoch": 391.25,
-      "learning_rate": 1.831140350877193e-05,
-      "loss": 0.0075,
-      "step": 1565
-    },
-    {
-      "epoch": 391.5,
-      "learning_rate": 1.828947368421053e-05,
-      "loss": 0.0074,
-      "step": 1566
-    },
-    {
-      "epoch": 391.75,
-      "learning_rate": 1.8267543859649125e-05,
-      "loss": 0.0083,
-      "step": 1567
-    },
-    {
-      "epoch": 392.0,
-      "learning_rate": 1.8245614035087722e-05,
-      "loss": 0.0077,
-      "step": 1568
-    },
-    {
-      "epoch": 392.25,
-      "learning_rate": 1.8223684210526318e-05,
-      "loss": 0.0076,
-      "step": 1569
-    },
-    {
-      "epoch": 392.5,
-      "learning_rate": 1.8201754385964914e-05,
-      "loss": 0.0064,
-      "step": 1570
-    },
-    {
-      "epoch": 392.75,
-      "learning_rate": 1.817982456140351e-05,
-      "loss": 0.0074,
-      "step": 1571
-    },
-    {
-      "epoch": 393.0,
-      "learning_rate": 1.8157894736842107e-05,
-      "loss": 0.0084,
-      "step": 1572
-    },
-    {
-      "epoch": 393.25,
-      "learning_rate": 1.8135964912280704e-05,
-      "loss": 0.0073,
-      "step": 1573
-    },
-    {
-      "epoch": 393.5,
-      "learning_rate": 1.81140350877193e-05,
-      "loss": 0.0075,
-      "step": 1574
-    },
-    {
-      "epoch": 393.75,
-      "learning_rate": 1.8092105263157896e-05,
-      "loss": 0.0077,
-      "step": 1575
-    },
-    {
-      "epoch": 394.0,
-      "learning_rate": 1.8070175438596493e-05,
-      "loss": 0.0079,
-      "step": 1576
-    },
-    {
-      "epoch": 394.25,
-      "learning_rate": 1.804824561403509e-05,
-      "loss": 0.0067,
-      "step": 1577
-    },
-    {
-      "epoch": 394.5,
-      "learning_rate": 1.8026315789473685e-05,
-      "loss": 0.0085,
-      "step": 1578
-    },
-    {
-      "epoch": 394.75,
-      "learning_rate": 1.8004385964912282e-05,
-      "loss": 0.0073,
-      "step": 1579
-    },
-    {
-      "epoch": 395.0,
-      "learning_rate": 1.7982456140350878e-05,
-      "loss": 0.0077,
-      "step": 1580
-    },
-    {
-      "epoch": 395.0,
-      "eval_accuracy_ELSE": 0.9966398649452963,
-      "eval_accuracy_road": 0.9948026428205274,
-      "eval_accuracy_sidewalk": 0.9074908476485497,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935004692607255,
-      "eval_iou_road": 0.987937666157588,
-      "eval_iou_sidewalk": 0.8399582953212563,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02159702032804489,
-      "eval_mean_accuracy": 0.9663111184714578,
-      "eval_mean_iou": 0.9404654769131899,
-      "eval_overall_accuracy": 0.9937324523925781,
-      "eval_runtime": 0.6052,
-      "eval_samples_per_second": 1.652,
-      "eval_steps_per_second": 1.652,
-      "step": 1580
-    },
-    {
-      "epoch": 395.25,
-      "learning_rate": 1.7960526315789475e-05,
-      "loss": 0.0076,
-      "step": 1581
-    },
-    {
-      "epoch": 395.5,
-      "learning_rate": 1.793859649122807e-05,
-      "loss": 0.0074,
-      "step": 1582
-    },
-    {
-      "epoch": 395.75,
-      "learning_rate": 1.7916666666666667e-05,
-      "loss": 0.0071,
-      "step": 1583
-    },
-    {
-      "epoch": 396.0,
-      "learning_rate": 1.7894736842105264e-05,
-      "loss": 0.0089,
-      "step": 1584
-    },
-    {
-      "epoch": 396.25,
-      "learning_rate": 1.787280701754386e-05,
-      "loss": 0.0076,
-      "step": 1585
-    },
-    {
-      "epoch": 396.5,
-      "learning_rate": 1.7850877192982456e-05,
-      "loss": 0.0077,
-      "step": 1586
-    },
-    {
-      "epoch": 396.75,
-      "learning_rate": 1.7828947368421053e-05,
-      "loss": 0.0076,
-      "step": 1587
-    },
-    {
-      "epoch": 397.0,
-      "learning_rate": 1.780701754385965e-05,
-      "loss": 0.0072,
-      "step": 1588
-    },
-    {
-      "epoch": 397.25,
-      "learning_rate": 1.7785087719298246e-05,
-      "loss": 0.007,
-      "step": 1589
-    },
-    {
-      "epoch": 397.5,
-      "learning_rate": 1.7763157894736842e-05,
-      "loss": 0.0076,
-      "step": 1590
-    },
-    {
-      "epoch": 397.75,
-      "learning_rate": 1.774122807017544e-05,
-      "loss": 0.008,
-      "step": 1591
-    },
-    {
-      "epoch": 398.0,
-      "learning_rate": 1.7719298245614035e-05,
-      "loss": 0.0073,
-      "step": 1592
-    },
-    {
-      "epoch": 398.25,
-      "learning_rate": 1.769736842105263e-05,
-      "loss": 0.008,
-      "step": 1593
-    },
-    {
-      "epoch": 398.5,
-      "learning_rate": 1.7675438596491227e-05,
-      "loss": 0.0065,
-      "step": 1594
-    },
-    {
-      "epoch": 398.75,
-      "learning_rate": 1.7653508771929824e-05,
-      "loss": 0.0076,
-      "step": 1595
-    },
-    {
-      "epoch": 399.0,
-      "learning_rate": 1.763157894736842e-05,
-      "loss": 0.0084,
-      "step": 1596
-    },
-    {
-      "epoch": 399.25,
-      "learning_rate": 1.7609649122807017e-05,
-      "loss": 0.0077,
-      "step": 1597
-    },
-    {
-      "epoch": 399.5,
-      "learning_rate": 1.7587719298245613e-05,
-      "loss": 0.0071,
-      "step": 1598
-    },
-    {
-      "epoch": 399.75,
-      "learning_rate": 1.756578947368421e-05,
-      "loss": 0.0079,
-      "step": 1599
-    },
-    {
-      "epoch": 400.0,
-      "learning_rate": 1.7543859649122806e-05,
-      "loss": 0.008,
-      "step": 1600
-    },
-    {
-      "epoch": 400.0,
-      "eval_accuracy_ELSE": 0.9970240349757054,
-      "eval_accuracy_road": 0.9945463347952384,
-      "eval_accuracy_sidewalk": 0.9059419881723458,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936851187747728,
-      "eval_iou_road": 0.9878647299265944,
-      "eval_iou_sidewalk": 0.8457999211252794,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021236859261989594,
-      "eval_mean_accuracy": 0.9658374526477632,
-      "eval_mean_iou": 0.9424499232755489,
-      "eval_overall_accuracy": 0.9938926696777344,
-      "eval_runtime": 0.6139,
-      "eval_samples_per_second": 1.629,
-      "eval_steps_per_second": 1.629,
-      "step": 1600
-    },
-    {
-      "epoch": 400.25,
-      "learning_rate": 1.7521929824561402e-05,
-      "loss": 0.0076,
-      "step": 1601
-    },
-    {
-      "epoch": 400.5,
-      "learning_rate": 1.75e-05,
-      "loss": 0.0077,
-      "step": 1602
-    },
-    {
-      "epoch": 400.75,
-      "learning_rate": 1.7478070175438595e-05,
-      "loss": 0.0072,
-      "step": 1603
-    },
-    {
-      "epoch": 401.0,
-      "learning_rate": 1.745614035087719e-05,
-      "loss": 0.0078,
-      "step": 1604
-    },
-    {
-      "epoch": 401.25,
-      "learning_rate": 1.7434210526315788e-05,
-      "loss": 0.0077,
-      "step": 1605
-    },
-    {
-      "epoch": 401.5,
-      "learning_rate": 1.7412280701754384e-05,
-      "loss": 0.0066,
-      "step": 1606
-    },
-    {
-      "epoch": 401.75,
-      "learning_rate": 1.7390350877192984e-05,
-      "loss": 0.0087,
-      "step": 1607
-    },
-    {
-      "epoch": 402.0,
-      "learning_rate": 1.736842105263158e-05,
-      "loss": 0.0074,
-      "step": 1608
-    },
-    {
-      "epoch": 402.25,
-      "learning_rate": 1.7346491228070176e-05,
-      "loss": 0.008,
-      "step": 1609
-    },
-    {
-      "epoch": 402.5,
-      "learning_rate": 1.7324561403508773e-05,
-      "loss": 0.0073,
-      "step": 1610
-    },
-    {
-      "epoch": 402.75,
-      "learning_rate": 1.730263157894737e-05,
-      "loss": 0.0074,
-      "step": 1611
-    },
-    {
-      "epoch": 403.0,
-      "learning_rate": 1.7280701754385966e-05,
-      "loss": 0.007,
-      "step": 1612
-    },
-    {
-      "epoch": 403.25,
-      "learning_rate": 1.7258771929824562e-05,
-      "loss": 0.0078,
-      "step": 1613
-    },
-    {
-      "epoch": 403.5,
-      "learning_rate": 1.723684210526316e-05,
-      "loss": 0.0078,
-      "step": 1614
-    },
-    {
-      "epoch": 403.75,
-      "learning_rate": 1.7214912280701755e-05,
-      "loss": 0.0078,
-      "step": 1615
-    },
-    {
-      "epoch": 404.0,
-      "learning_rate": 1.719298245614035e-05,
-      "loss": 0.0071,
-      "step": 1616
-    },
-    {
-      "epoch": 404.25,
-      "learning_rate": 1.7171052631578947e-05,
-      "loss": 0.0072,
-      "step": 1617
-    },
-    {
-      "epoch": 404.5,
-      "learning_rate": 1.7149122807017544e-05,
-      "loss": 0.0079,
-      "step": 1618
-    },
-    {
-      "epoch": 404.75,
-      "learning_rate": 1.712719298245614e-05,
-      "loss": 0.007,
-      "step": 1619
-    },
-    {
-      "epoch": 405.0,
-      "learning_rate": 1.7105263157894737e-05,
-      "loss": 0.0071,
-      "step": 1620
-    },
-    {
-      "epoch": 405.0,
-      "eval_accuracy_ELSE": 0.996818422846754,
-      "eval_accuracy_road": 0.9945178561257618,
-      "eval_accuracy_sidewalk": 0.907772458462405,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936355889237674,
-      "eval_iou_road": 0.9877107138816609,
-      "eval_iou_sidewalk": 0.8423046772929187,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021291302517056465,
-      "eval_mean_accuracy": 0.9663695791449737,
-      "eval_mean_iou": 0.9412169933661158,
-      "eval_overall_accuracy": 0.9937896728515625,
-      "eval_runtime": 0.6001,
-      "eval_samples_per_second": 1.667,
-      "eval_steps_per_second": 1.667,
-      "step": 1620
-    },
-    {
-      "epoch": 405.25,
-      "learning_rate": 1.7083333333333333e-05,
-      "loss": 0.0075,
-      "step": 1621
-    },
-    {
-      "epoch": 405.5,
-      "learning_rate": 1.706140350877193e-05,
-      "loss": 0.0071,
-      "step": 1622
-    },
-    {
-      "epoch": 405.75,
-      "learning_rate": 1.7039473684210526e-05,
-      "loss": 0.0079,
-      "step": 1623
-    },
-    {
-      "epoch": 406.0,
-      "learning_rate": 1.7017543859649125e-05,
-      "loss": 0.0069,
-      "step": 1624
-    },
-    {
-      "epoch": 406.25,
-      "learning_rate": 1.6995614035087722e-05,
-      "loss": 0.0073,
-      "step": 1625
-    },
-    {
-      "epoch": 406.5,
-      "learning_rate": 1.6973684210526318e-05,
-      "loss": 0.0074,
-      "step": 1626
-    },
-    {
-      "epoch": 406.75,
-      "learning_rate": 1.6951754385964915e-05,
-      "loss": 0.0072,
-      "step": 1627
-    },
-    {
-      "epoch": 407.0,
-      "learning_rate": 1.692982456140351e-05,
-      "loss": 0.0078,
-      "step": 1628
-    },
-    {
-      "epoch": 407.25,
-      "learning_rate": 1.6907894736842107e-05,
-      "loss": 0.0064,
-      "step": 1629
-    },
-    {
-      "epoch": 407.5,
-      "learning_rate": 1.6885964912280704e-05,
-      "loss": 0.0074,
-      "step": 1630
-    },
-    {
-      "epoch": 407.75,
-      "learning_rate": 1.68640350877193e-05,
-      "loss": 0.0076,
-      "step": 1631
-    },
-    {
-      "epoch": 408.0,
-      "learning_rate": 1.6842105263157896e-05,
-      "loss": 0.0077,
-      "step": 1632
-    },
-    {
-      "epoch": 408.25,
-      "learning_rate": 1.6820175438596493e-05,
-      "loss": 0.0079,
-      "step": 1633
-    },
-    {
-      "epoch": 408.5,
-      "learning_rate": 1.679824561403509e-05,
-      "loss": 0.007,
-      "step": 1634
-    },
-    {
-      "epoch": 408.75,
-      "learning_rate": 1.6776315789473686e-05,
-      "loss": 0.0071,
-      "step": 1635
-    },
-    {
-      "epoch": 409.0,
-      "learning_rate": 1.6754385964912282e-05,
-      "loss": 0.0086,
-      "step": 1636
-    },
-    {
-      "epoch": 409.25,
-      "learning_rate": 1.673245614035088e-05,
-      "loss": 0.0075,
-      "step": 1637
-    },
-    {
-      "epoch": 409.5,
-      "learning_rate": 1.6710526315789475e-05,
-      "loss": 0.0068,
-      "step": 1638
-    },
-    {
-      "epoch": 409.75,
-      "learning_rate": 1.668859649122807e-05,
-      "loss": 0.0079,
-      "step": 1639
-    },
-    {
-      "epoch": 410.0,
-      "learning_rate": 1.6666666666666667e-05,
-      "loss": 0.0082,
-      "step": 1640
-    },
-    {
-      "epoch": 410.0,
-      "eval_accuracy_ELSE": 0.996953693984222,
-      "eval_accuracy_road": 0.9943897021131173,
-      "eval_accuracy_sidewalk": 0.9063644043931287,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936257300480497,
-      "eval_iou_road": 0.987611370386084,
-      "eval_iou_sidewalk": 0.8444182080545717,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02179083414375782,
-      "eval_mean_accuracy": 0.9659026001634894,
-      "eval_mean_iou": 0.9418851028295685,
-      "eval_overall_accuracy": 0.9938125610351562,
-      "eval_runtime": 0.6102,
-      "eval_samples_per_second": 1.639,
-      "eval_steps_per_second": 1.639,
-      "step": 1640
-    },
-    {
-      "epoch": 410.25,
-      "learning_rate": 1.6644736842105264e-05,
-      "loss": 0.0077,
-      "step": 1641
-    },
-    {
-      "epoch": 410.5,
-      "learning_rate": 1.662280701754386e-05,
-      "loss": 0.0077,
-      "step": 1642
-    },
-    {
-      "epoch": 410.75,
-      "learning_rate": 1.6600877192982457e-05,
-      "loss": 0.007,
-      "step": 1643
-    },
-    {
-      "epoch": 411.0,
-      "learning_rate": 1.6578947368421053e-05,
-      "loss": 0.0074,
-      "step": 1644
-    },
-    {
-      "epoch": 411.25,
-      "learning_rate": 1.655701754385965e-05,
-      "loss": 0.008,
-      "step": 1645
-    },
-    {
-      "epoch": 411.5,
-      "learning_rate": 1.6535087719298246e-05,
-      "loss": 0.0076,
-      "step": 1646
-    },
-    {
-      "epoch": 411.75,
-      "learning_rate": 1.6513157894736845e-05,
-      "loss": 0.0076,
-      "step": 1647
-    },
-    {
-      "epoch": 412.0,
-      "learning_rate": 1.6491228070175442e-05,
-      "loss": 0.0076,
-      "step": 1648
-    },
-    {
-      "epoch": 412.25,
-      "learning_rate": 1.6469298245614038e-05,
-      "loss": 0.0083,
-      "step": 1649
-    },
-    {
-      "epoch": 412.5,
-      "learning_rate": 1.6447368421052635e-05,
-      "loss": 0.008,
-      "step": 1650
-    },
-    {
-      "epoch": 412.75,
-      "learning_rate": 1.642543859649123e-05,
-      "loss": 0.0071,
-      "step": 1651
-    },
-    {
-      "epoch": 413.0,
-      "learning_rate": 1.6403508771929827e-05,
-      "loss": 0.0081,
-      "step": 1652
-    },
-    {
-      "epoch": 413.25,
-      "learning_rate": 1.6381578947368424e-05,
-      "loss": 0.0066,
-      "step": 1653
-    },
-    {
-      "epoch": 413.5,
-      "learning_rate": 1.635964912280702e-05,
-      "loss": 0.0075,
-      "step": 1654
-    },
-    {
-      "epoch": 413.75,
-      "learning_rate": 1.6337719298245616e-05,
-      "loss": 0.0081,
-      "step": 1655
-    },
-    {
-      "epoch": 414.0,
-      "learning_rate": 1.6315789473684213e-05,
-      "loss": 0.0079,
-      "step": 1656
-    },
-    {
-      "epoch": 414.25,
-      "learning_rate": 1.629385964912281e-05,
-      "loss": 0.0073,
-      "step": 1657
-    },
-    {
-      "epoch": 414.5,
-      "learning_rate": 1.6271929824561406e-05,
-      "loss": 0.0068,
-      "step": 1658
-    },
-    {
-      "epoch": 414.75,
-      "learning_rate": 1.6250000000000002e-05,
-      "loss": 0.0071,
-      "step": 1659
-    },
-    {
-      "epoch": 415.0,
-      "learning_rate": 1.62280701754386e-05,
-      "loss": 0.0077,
-      "step": 1660
-    },
-    {
-      "epoch": 415.0,
-      "eval_accuracy_ELSE": 0.9969645156752194,
-      "eval_accuracy_road": 0.9937489320498947,
-      "eval_accuracy_sidewalk": 0.9084764854970431,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935615027474157,
-      "eval_iou_road": 0.9874916870657818,
-      "eval_iou_sidewalk": 0.8407610112066719,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02190292254090309,
-      "eval_mean_accuracy": 0.9663966444073857,
-      "eval_mean_iou": 0.9406047336732898,
-      "eval_overall_accuracy": 0.9937057495117188,
-      "eval_runtime": 0.6039,
-      "eval_samples_per_second": 1.656,
-      "eval_steps_per_second": 1.656,
-      "step": 1660
-    },
-    {
-      "epoch": 415.25,
-      "learning_rate": 1.6206140350877195e-05,
-      "loss": 0.007,
-      "step": 1661
-    },
-    {
-      "epoch": 415.5,
-      "learning_rate": 1.618421052631579e-05,
-      "loss": 0.0082,
-      "step": 1662
-    },
-    {
-      "epoch": 415.75,
-      "learning_rate": 1.6162280701754387e-05,
-      "loss": 0.0073,
-      "step": 1663
-    },
-    {
-      "epoch": 416.0,
-      "learning_rate": 1.6140350877192984e-05,
-      "loss": 0.0072,
-      "step": 1664
-    },
-    {
-      "epoch": 416.25,
-      "learning_rate": 1.611842105263158e-05,
-      "loss": 0.0078,
-      "step": 1665
-    },
-    {
-      "epoch": 416.5,
-      "learning_rate": 1.6096491228070177e-05,
-      "loss": 0.0062,
-      "step": 1666
-    },
-    {
-      "epoch": 416.75,
-      "learning_rate": 1.6074561403508773e-05,
-      "loss": 0.0078,
-      "step": 1667
-    },
-    {
-      "epoch": 417.0,
-      "learning_rate": 1.605263157894737e-05,
-      "loss": 0.0075,
-      "step": 1668
-    },
-    {
-      "epoch": 417.25,
-      "learning_rate": 1.6030701754385966e-05,
-      "loss": 0.0077,
-      "step": 1669
-    },
-    {
-      "epoch": 417.5,
-      "learning_rate": 1.6008771929824562e-05,
-      "loss": 0.0082,
-      "step": 1670
-    },
-    {
-      "epoch": 417.75,
-      "learning_rate": 1.598684210526316e-05,
-      "loss": 0.0076,
-      "step": 1671
-    },
-    {
-      "epoch": 418.0,
-      "learning_rate": 1.5964912280701755e-05,
-      "loss": 0.0069,
-      "step": 1672
-    },
-    {
-      "epoch": 418.25,
-      "learning_rate": 1.594298245614035e-05,
-      "loss": 0.0076,
-      "step": 1673
-    },
-    {
-      "epoch": 418.5,
-      "learning_rate": 1.5921052631578948e-05,
-      "loss": 0.0072,
-      "step": 1674
-    },
-    {
-      "epoch": 418.75,
-      "learning_rate": 1.5899122807017544e-05,
-      "loss": 0.0077,
-      "step": 1675
-    },
-    {
-      "epoch": 419.0,
-      "learning_rate": 1.587719298245614e-05,
-      "loss": 0.008,
-      "step": 1676
-    },
-    {
-      "epoch": 419.25,
-      "learning_rate": 1.5855263157894737e-05,
-      "loss": 0.0076,
-      "step": 1677
-    },
-    {
-      "epoch": 419.5,
-      "learning_rate": 1.5833333333333333e-05,
-      "loss": 0.0079,
-      "step": 1678
-    },
-    {
-      "epoch": 419.75,
-      "learning_rate": 1.581140350877193e-05,
-      "loss": 0.0076,
-      "step": 1679
-    },
-    {
-      "epoch": 420.0,
-      "learning_rate": 1.5789473684210526e-05,
-      "loss": 0.0065,
-      "step": 1680
-    },
-    {
-      "epoch": 420.0,
-      "eval_accuracy_ELSE": 0.9968995855292347,
-      "eval_accuracy_road": 0.9946744888078829,
-      "eval_accuracy_sidewalk": 0.9052379611377077,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936950542041961,
-      "eval_iou_road": 0.9877265914425497,
-      "eval_iou_sidewalk": 0.842374213836478,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021578041836619377,
-      "eval_mean_accuracy": 0.9656040118249418,
-      "eval_mean_iou": 0.9412652864944079,
-      "eval_overall_accuracy": 0.9938201904296875,
-      "eval_runtime": 0.6036,
-      "eval_samples_per_second": 1.657,
-      "eval_steps_per_second": 1.657,
-      "step": 1680
-    },
-    {
-      "epoch": 420.25,
-      "learning_rate": 1.5767543859649122e-05,
-      "loss": 0.0074,
-      "step": 1681
-    },
-    {
-      "epoch": 420.5,
-      "learning_rate": 1.574561403508772e-05,
-      "loss": 0.0073,
-      "step": 1682
-    },
-    {
-      "epoch": 420.75,
-      "learning_rate": 1.5723684210526315e-05,
-      "loss": 0.0078,
-      "step": 1683
-    },
-    {
-      "epoch": 421.0,
-      "learning_rate": 1.570175438596491e-05,
-      "loss": 0.0074,
-      "step": 1684
-    },
-    {
-      "epoch": 421.25,
-      "learning_rate": 1.5679824561403508e-05,
-      "loss": 0.0075,
-      "step": 1685
-    },
-    {
-      "epoch": 421.5,
-      "learning_rate": 1.5657894736842104e-05,
-      "loss": 0.0076,
-      "step": 1686
-    },
-    {
-      "epoch": 421.75,
-      "learning_rate": 1.56359649122807e-05,
-      "loss": 0.007,
-      "step": 1687
-    },
-    {
-      "epoch": 422.0,
-      "learning_rate": 1.56140350877193e-05,
-      "loss": 0.0077,
-      "step": 1688
-    },
-    {
-      "epoch": 422.25,
-      "learning_rate": 1.5592105263157897e-05,
-      "loss": 0.0068,
-      "step": 1689
-    },
-    {
-      "epoch": 422.5,
-      "learning_rate": 1.5570175438596493e-05,
-      "loss": 0.0072,
-      "step": 1690
-    },
-    {
-      "epoch": 422.75,
-      "learning_rate": 1.554824561403509e-05,
-      "loss": 0.0066,
-      "step": 1691
-    },
-    {
-      "epoch": 423.0,
-      "learning_rate": 1.5526315789473686e-05,
-      "loss": 0.0082,
-      "step": 1692
-    },
-    {
-      "epoch": 423.25,
-      "learning_rate": 1.5504385964912282e-05,
-      "loss": 0.0074,
-      "step": 1693
-    },
-    {
-      "epoch": 423.5,
-      "learning_rate": 1.548245614035088e-05,
-      "loss": 0.0069,
-      "step": 1694
-    },
-    {
-      "epoch": 423.75,
-      "learning_rate": 1.5460526315789475e-05,
-      "loss": 0.0074,
-      "step": 1695
-    },
-    {
-      "epoch": 424.0,
-      "learning_rate": 1.543859649122807e-05,
-      "loss": 0.0069,
-      "step": 1696
-    },
-    {
-      "epoch": 424.25,
-      "learning_rate": 1.5416666666666668e-05,
-      "loss": 0.0073,
-      "step": 1697
-    },
-    {
-      "epoch": 424.5,
-      "learning_rate": 1.5394736842105264e-05,
-      "loss": 0.0076,
-      "step": 1698
-    },
-    {
-      "epoch": 424.75,
-      "learning_rate": 1.537280701754386e-05,
-      "loss": 0.0072,
-      "step": 1699
-    },
-    {
-      "epoch": 425.0,
-      "learning_rate": 1.5350877192982457e-05,
-      "loss": 0.0075,
-      "step": 1700
-    },
-    {
-      "epoch": 425.0,
-      "eval_accuracy_ELSE": 0.9969266397567283,
-      "eval_accuracy_road": 0.9952298228626758,
-      "eval_accuracy_sidewalk": 0.9028442692199381,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937809804799379,
-      "eval_iou_road": 0.9880964162013147,
-      "eval_iou_sidewalk": 0.8433513086939366,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02155287005007267,
-      "eval_mean_accuracy": 0.9650002439464473,
-      "eval_mean_iou": 0.9417429017917297,
-      "eval_overall_accuracy": 0.9939231872558594,
-      "eval_runtime": 0.6218,
-      "eval_samples_per_second": 1.608,
-      "eval_steps_per_second": 1.608,
-      "step": 1700
-    },
-    {
-      "epoch": 425.25,
-      "learning_rate": 1.5328947368421053e-05,
-      "loss": 0.007,
-      "step": 1701
-    },
-    {
-      "epoch": 425.5,
-      "learning_rate": 1.530701754385965e-05,
-      "loss": 0.0069,
-      "step": 1702
-    },
-    {
-      "epoch": 425.75,
-      "learning_rate": 1.5285087719298246e-05,
-      "loss": 0.0075,
-      "step": 1703
-    },
-    {
-      "epoch": 426.0,
-      "learning_rate": 1.5263157894736842e-05,
-      "loss": 0.0074,
-      "step": 1704
-    },
-    {
-      "epoch": 426.25,
-      "learning_rate": 1.524122807017544e-05,
-      "loss": 0.007,
-      "step": 1705
-    },
-    {
-      "epoch": 426.5,
-      "learning_rate": 1.5219298245614037e-05,
-      "loss": 0.0074,
-      "step": 1706
-    },
-    {
-      "epoch": 426.75,
-      "learning_rate": 1.5197368421052633e-05,
-      "loss": 0.0078,
-      "step": 1707
-    },
-    {
-      "epoch": 427.0,
-      "learning_rate": 1.517543859649123e-05,
-      "loss": 0.0077,
-      "step": 1708
-    },
-    {
-      "epoch": 427.25,
-      "learning_rate": 1.5153508771929826e-05,
-      "loss": 0.0077,
-      "step": 1709
-    },
-    {
-      "epoch": 427.5,
-      "learning_rate": 1.5131578947368422e-05,
-      "loss": 0.0068,
-      "step": 1710
-    },
-    {
-      "epoch": 427.75,
-      "learning_rate": 1.5109649122807019e-05,
-      "loss": 0.007,
-      "step": 1711
-    },
-    {
-      "epoch": 428.0,
-      "learning_rate": 1.5087719298245615e-05,
-      "loss": 0.0071,
-      "step": 1712
-    },
-    {
-      "epoch": 428.25,
-      "learning_rate": 1.5065789473684211e-05,
-      "loss": 0.0074,
-      "step": 1713
-    },
-    {
-      "epoch": 428.5,
-      "learning_rate": 1.5043859649122808e-05,
-      "loss": 0.0075,
-      "step": 1714
-    },
-    {
-      "epoch": 428.75,
-      "learning_rate": 1.5021929824561404e-05,
-      "loss": 0.0081,
-      "step": 1715
-    },
-    {
-      "epoch": 429.0,
-      "learning_rate": 1.5e-05,
-      "loss": 0.0069,
-      "step": 1716
-    },
-    {
-      "epoch": 429.25,
-      "learning_rate": 1.4978070175438597e-05,
-      "loss": 0.0079,
-      "step": 1717
-    },
-    {
-      "epoch": 429.5,
-      "learning_rate": 1.4956140350877193e-05,
-      "loss": 0.0072,
-      "step": 1718
-    },
-    {
-      "epoch": 429.75,
-      "learning_rate": 1.493421052631579e-05,
-      "loss": 0.0067,
-      "step": 1719
-    },
-    {
-      "epoch": 430.0,
-      "learning_rate": 1.4912280701754386e-05,
-      "loss": 0.0065,
-      "step": 1720
-    },
-    {
-      "epoch": 430.0,
-      "eval_accuracy_ELSE": 0.9968671204562425,
-      "eval_accuracy_road": 0.9940764367488751,
-      "eval_accuracy_sidewalk": 0.9076316530554773,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9934965838191123,
-      "eval_iou_road": 0.9877053239201483,
-      "eval_iou_sidewalk": 0.8400886224423303,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02224127948284149,
-      "eval_mean_accuracy": 0.9661917367535316,
-      "eval_mean_iou": 0.940430176727197,
-      "eval_overall_accuracy": 0.9937019348144531,
-      "eval_runtime": 0.6175,
-      "eval_samples_per_second": 1.619,
-      "eval_steps_per_second": 1.619,
-      "step": 1720
-    },
-    {
-      "epoch": 430.25,
-      "learning_rate": 1.4890350877192982e-05,
-      "loss": 0.0075,
-      "step": 1721
-    },
-    {
-      "epoch": 430.5,
-      "learning_rate": 1.4868421052631579e-05,
-      "loss": 0.0073,
-      "step": 1722
-    },
-    {
-      "epoch": 430.75,
-      "learning_rate": 1.4846491228070175e-05,
-      "loss": 0.0072,
-      "step": 1723
-    },
-    {
-      "epoch": 431.0,
-      "learning_rate": 1.4824561403508771e-05,
-      "loss": 0.007,
-      "step": 1724
-    },
-    {
-      "epoch": 431.25,
-      "learning_rate": 1.4802631578947368e-05,
-      "loss": 0.0069,
-      "step": 1725
-    },
-    {
-      "epoch": 431.5,
-      "learning_rate": 1.4780701754385964e-05,
-      "loss": 0.0077,
-      "step": 1726
-    },
-    {
-      "epoch": 431.75,
-      "learning_rate": 1.475877192982456e-05,
-      "loss": 0.0077,
-      "step": 1727
-    },
-    {
-      "epoch": 432.0,
-      "learning_rate": 1.4736842105263157e-05,
-      "loss": 0.0068,
-      "step": 1728
-    },
-    {
-      "epoch": 432.25,
-      "learning_rate": 1.4714912280701757e-05,
-      "loss": 0.0076,
-      "step": 1729
-    },
-    {
-      "epoch": 432.5,
-      "learning_rate": 1.4692982456140353e-05,
-      "loss": 0.0072,
-      "step": 1730
-    },
-    {
-      "epoch": 432.75,
-      "learning_rate": 1.467105263157895e-05,
-      "loss": 0.0072,
-      "step": 1731
-    },
-    {
-      "epoch": 433.0,
-      "learning_rate": 1.4649122807017546e-05,
-      "loss": 0.0075,
-      "step": 1732
-    },
-    {
-      "epoch": 433.25,
-      "learning_rate": 1.4627192982456142e-05,
-      "loss": 0.0071,
-      "step": 1733
-    },
-    {
-      "epoch": 433.5,
-      "learning_rate": 1.4605263157894739e-05,
-      "loss": 0.0076,
-      "step": 1734
-    },
-    {
-      "epoch": 433.75,
-      "learning_rate": 1.4583333333333335e-05,
-      "loss": 0.0081,
-      "step": 1735
-    },
-    {
-      "epoch": 434.0,
-      "learning_rate": 1.4561403508771931e-05,
-      "loss": 0.0067,
-      "step": 1736
-    },
-    {
-      "epoch": 434.25,
-      "learning_rate": 1.4539473684210528e-05,
-      "loss": 0.0072,
-      "step": 1737
-    },
-    {
-      "epoch": 434.5,
-      "learning_rate": 1.4517543859649124e-05,
-      "loss": 0.0066,
-      "step": 1738
-    },
-    {
-      "epoch": 434.75,
-      "learning_rate": 1.449561403508772e-05,
-      "loss": 0.0073,
-      "step": 1739
-    },
-    {
-      "epoch": 435.0,
-      "learning_rate": 1.4473684210526317e-05,
-      "loss": 0.008,
-      "step": 1740
-    },
-    {
-      "epoch": 435.0,
-      "eval_accuracy_ELSE": 0.9966182215633015,
-      "eval_accuracy_road": 0.9944893774562853,
-      "eval_accuracy_sidewalk": 0.9076316530554773,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9933610182288858,
-      "eval_iou_road": 0.9879339123546553,
-      "eval_iou_sidewalk": 0.8373603533385295,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02256837673485279,
-      "eval_mean_accuracy": 0.9662464173583546,
-      "eval_mean_iou": 0.9395517613073568,
-      "eval_overall_accuracy": 0.9936370849609375,
-      "eval_runtime": 0.6025,
-      "eval_samples_per_second": 1.66,
-      "eval_steps_per_second": 1.66,
-      "step": 1740
-    },
-    {
-      "epoch": 435.25,
-      "learning_rate": 1.4451754385964913e-05,
-      "loss": 0.0072,
-      "step": 1741
-    },
-    {
-      "epoch": 435.5,
-      "learning_rate": 1.442982456140351e-05,
-      "loss": 0.0066,
-      "step": 1742
-    },
-    {
-      "epoch": 435.75,
-      "learning_rate": 1.4407894736842106e-05,
-      "loss": 0.0076,
-      "step": 1743
-    },
-    {
-      "epoch": 436.0,
-      "learning_rate": 1.4385964912280702e-05,
-      "loss": 0.0077,
-      "step": 1744
-    },
-    {
-      "epoch": 436.25,
-      "learning_rate": 1.4364035087719299e-05,
-      "loss": 0.0068,
-      "step": 1745
-    },
-    {
-      "epoch": 436.5,
-      "learning_rate": 1.4342105263157895e-05,
-      "loss": 0.0078,
-      "step": 1746
-    },
-    {
-      "epoch": 436.75,
-      "learning_rate": 1.4320175438596491e-05,
-      "loss": 0.0073,
-      "step": 1747
-    },
-    {
-      "epoch": 437.0,
-      "learning_rate": 1.4298245614035088e-05,
-      "loss": 0.008,
-      "step": 1748
-    },
-    {
-      "epoch": 437.25,
-      "learning_rate": 1.4276315789473684e-05,
-      "loss": 0.0071,
-      "step": 1749
-    },
-    {
-      "epoch": 437.5,
-      "learning_rate": 1.425438596491228e-05,
-      "loss": 0.0067,
-      "step": 1750
-    },
-    {
-      "epoch": 437.75,
-      "learning_rate": 1.4232456140350877e-05,
-      "loss": 0.0072,
-      "step": 1751
-    },
-    {
-      "epoch": 438.0,
-      "learning_rate": 1.4210526315789475e-05,
-      "loss": 0.007,
-      "step": 1752
-    },
-    {
-      "epoch": 438.25,
-      "learning_rate": 1.4188596491228071e-05,
-      "loss": 0.0069,
-      "step": 1753
-    },
-    {
-      "epoch": 438.5,
-      "learning_rate": 1.4166666666666668e-05,
-      "loss": 0.0078,
-      "step": 1754
-    },
-    {
-      "epoch": 438.75,
-      "learning_rate": 1.4144736842105264e-05,
-      "loss": 0.0067,
-      "step": 1755
-    },
-    {
-      "epoch": 439.0,
-      "learning_rate": 1.412280701754386e-05,
-      "loss": 0.0074,
-      "step": 1756
-    },
-    {
-      "epoch": 439.25,
-      "learning_rate": 1.4100877192982457e-05,
-      "loss": 0.0069,
-      "step": 1757
-    },
-    {
-      "epoch": 439.5,
-      "learning_rate": 1.4078947368421053e-05,
-      "loss": 0.0067,
-      "step": 1758
-    },
-    {
-      "epoch": 439.75,
-      "learning_rate": 1.405701754385965e-05,
-      "loss": 0.0073,
-      "step": 1759
-    },
-    {
-      "epoch": 440.0,
-      "learning_rate": 1.4035087719298246e-05,
-      "loss": 0.0083,
-      "step": 1760
-    },
-    {
-      "epoch": 440.0,
-      "eval_accuracy_ELSE": 0.9968833529927387,
-      "eval_accuracy_road": 0.9949450361679102,
-      "eval_accuracy_sidewalk": 0.9058011827654182,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937539105482319,
-      "eval_iou_road": 0.9878695338677524,
-      "eval_iou_sidewalk": 0.844558225022975,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021862179040908813,
-      "eval_mean_accuracy": 0.9658765239753557,
-      "eval_mean_iou": 0.9420605564796531,
-      "eval_overall_accuracy": 0.993896484375,
-      "eval_runtime": 0.6066,
-      "eval_samples_per_second": 1.649,
-      "eval_steps_per_second": 1.649,
-      "step": 1760
-    },
-    {
-      "epoch": 440.25,
-      "learning_rate": 1.4013157894736842e-05,
-      "loss": 0.0065,
-      "step": 1761
-    },
-    {
-      "epoch": 440.5,
-      "learning_rate": 1.3991228070175439e-05,
-      "loss": 0.0075,
-      "step": 1762
-    },
-    {
-      "epoch": 440.75,
-      "learning_rate": 1.3969298245614035e-05,
-      "loss": 0.0073,
-      "step": 1763
-    },
-    {
-      "epoch": 441.0,
-      "learning_rate": 1.3947368421052631e-05,
-      "loss": 0.0066,
-      "step": 1764
-    },
-    {
-      "epoch": 441.25,
-      "learning_rate": 1.3925438596491228e-05,
-      "loss": 0.0076,
-      "step": 1765
-    },
-    {
-      "epoch": 441.5,
-      "learning_rate": 1.3903508771929824e-05,
-      "loss": 0.0068,
-      "step": 1766
-    },
-    {
-      "epoch": 441.75,
-      "learning_rate": 1.388157894736842e-05,
-      "loss": 0.0077,
-      "step": 1767
-    },
-    {
-      "epoch": 442.0,
-      "learning_rate": 1.3859649122807017e-05,
-      "loss": 0.0074,
-      "step": 1768
-    },
-    {
-      "epoch": 442.25,
-      "learning_rate": 1.3837719298245617e-05,
-      "loss": 0.007,
-      "step": 1769
-    },
-    {
-      "epoch": 442.5,
-      "learning_rate": 1.3815789473684213e-05,
-      "loss": 0.0064,
-      "step": 1770
-    },
-    {
-      "epoch": 442.75,
-      "learning_rate": 1.379385964912281e-05,
-      "loss": 0.0074,
-      "step": 1771
-    },
-    {
-      "epoch": 443.0,
-      "learning_rate": 1.3771929824561406e-05,
-      "loss": 0.0079,
-      "step": 1772
-    },
-    {
-      "epoch": 443.25,
-      "learning_rate": 1.3750000000000002e-05,
-      "loss": 0.0076,
-      "step": 1773
-    },
-    {
-      "epoch": 443.5,
-      "learning_rate": 1.3728070175438599e-05,
-      "loss": 0.0072,
-      "step": 1774
-    },
-    {
-      "epoch": 443.75,
-      "learning_rate": 1.3706140350877195e-05,
-      "loss": 0.0071,
-      "step": 1775
-    },
-    {
-      "epoch": 444.0,
-      "learning_rate": 1.3684210526315791e-05,
-      "loss": 0.0076,
-      "step": 1776
-    },
-    {
-      "epoch": 444.25,
-      "learning_rate": 1.3662280701754388e-05,
-      "loss": 0.0068,
-      "step": 1777
-    },
-    {
-      "epoch": 444.5,
-      "learning_rate": 1.3640350877192984e-05,
-      "loss": 0.0073,
-      "step": 1778
-    },
-    {
-      "epoch": 444.75,
-      "learning_rate": 1.361842105263158e-05,
-      "loss": 0.0064,
-      "step": 1779
-    },
-    {
-      "epoch": 445.0,
-      "learning_rate": 1.3596491228070177e-05,
-      "loss": 0.0072,
-      "step": 1780
-    },
-    {
-      "epoch": 445.0,
-      "eval_accuracy_ELSE": 0.9968995855292347,
-      "eval_accuracy_road": 0.9942188300962579,
-      "eval_accuracy_sidewalk": 0.9087580963108983,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993737931629648,
-      "eval_iou_road": 0.9874695932567743,
-      "eval_iou_sidewalk": 0.8424487664795719,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02201010286808014,
-      "eval_mean_accuracy": 0.966625503978797,
-      "eval_mean_iou": 0.9412187637886648,
-      "eval_overall_accuracy": 0.9937934875488281,
-      "eval_runtime": 0.6197,
-      "eval_samples_per_second": 1.614,
-      "eval_steps_per_second": 1.614,
-      "step": 1780
-    },
-    {
-      "epoch": 445.25,
-      "learning_rate": 1.3574561403508773e-05,
-      "loss": 0.0072,
-      "step": 1781
-    },
-    {
-      "epoch": 445.5,
-      "learning_rate": 1.355263157894737e-05,
-      "loss": 0.0071,
-      "step": 1782
-    },
-    {
-      "epoch": 445.75,
-      "learning_rate": 1.3530701754385966e-05,
-      "loss": 0.0071,
-      "step": 1783
-    },
-    {
-      "epoch": 446.0,
-      "learning_rate": 1.3508771929824562e-05,
-      "loss": 0.0073,
-      "step": 1784
-    },
-    {
-      "epoch": 446.25,
-      "learning_rate": 1.3486842105263159e-05,
-      "loss": 0.0076,
-      "step": 1785
-    },
-    {
-      "epoch": 446.5,
-      "learning_rate": 1.3464912280701755e-05,
-      "loss": 0.0074,
-      "step": 1786
-    },
-    {
-      "epoch": 446.75,
-      "learning_rate": 1.3442982456140351e-05,
-      "loss": 0.0073,
-      "step": 1787
-    },
-    {
-      "epoch": 447.0,
-      "learning_rate": 1.3421052631578948e-05,
-      "loss": 0.0072,
-      "step": 1788
-    },
-    {
-      "epoch": 447.25,
-      "learning_rate": 1.3399122807017544e-05,
-      "loss": 0.0069,
-      "step": 1789
-    },
-    {
-      "epoch": 447.5,
-      "learning_rate": 1.337719298245614e-05,
-      "loss": 0.008,
-      "step": 1790
-    },
-    {
-      "epoch": 447.75,
-      "learning_rate": 1.3355263157894737e-05,
-      "loss": 0.0072,
-      "step": 1791
-    },
-    {
-      "epoch": 448.0,
-      "learning_rate": 1.3333333333333333e-05,
-      "loss": 0.0074,
-      "step": 1792
-    },
-    {
-      "epoch": 448.25,
-      "learning_rate": 1.331140350877193e-05,
-      "loss": 0.0071,
-      "step": 1793
-    },
-    {
-      "epoch": 448.5,
-      "learning_rate": 1.3289473684210526e-05,
-      "loss": 0.0072,
-      "step": 1794
-    },
-    {
-      "epoch": 448.75,
-      "learning_rate": 1.3267543859649122e-05,
-      "loss": 0.0081,
-      "step": 1795
-    },
-    {
-      "epoch": 449.0,
-      "learning_rate": 1.3245614035087719e-05,
-      "loss": 0.0066,
-      "step": 1796
-    },
-    {
-      "epoch": 449.25,
-      "learning_rate": 1.3223684210526315e-05,
-      "loss": 0.0074,
-      "step": 1797
-    },
-    {
-      "epoch": 449.5,
-      "learning_rate": 1.3201754385964912e-05,
-      "loss": 0.0075,
-      "step": 1798
-    },
-    {
-      "epoch": 449.75,
-      "learning_rate": 1.3179824561403508e-05,
-      "loss": 0.0076,
-      "step": 1799
-    },
-    {
-      "epoch": 450.0,
-      "learning_rate": 1.3157894736842106e-05,
-      "loss": 0.0074,
-      "step": 1800
-    },
-    {
-      "epoch": 450.0,
-      "eval_accuracy_ELSE": 0.9968833529927387,
-      "eval_accuracy_road": 0.9951586261889844,
-      "eval_accuracy_sidewalk": 0.9021402421852999,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937539105482319,
-      "eval_iou_road": 0.9878441793407587,
-      "eval_iou_sidewalk": 0.8418079096045198,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022430066019296646,
-      "eval_mean_accuracy": 0.964727407122341,
-      "eval_mean_iou": 0.9411353331645035,
-      "eval_overall_accuracy": 0.9938545227050781,
-      "eval_runtime": 0.6157,
-      "eval_samples_per_second": 1.624,
-      "eval_steps_per_second": 1.624,
-      "step": 1800
-    },
-    {
-      "epoch": 450.25,
-      "learning_rate": 1.3135964912280702e-05,
-      "loss": 0.0074,
-      "step": 1801
-    },
-    {
-      "epoch": 450.5,
-      "learning_rate": 1.3114035087719299e-05,
-      "loss": 0.0066,
-      "step": 1802
-    },
-    {
-      "epoch": 450.75,
-      "learning_rate": 1.3092105263157895e-05,
-      "loss": 0.0067,
-      "step": 1803
-    },
-    {
-      "epoch": 451.0,
-      "learning_rate": 1.3070175438596491e-05,
-      "loss": 0.0069,
-      "step": 1804
-    },
-    {
-      "epoch": 451.25,
-      "learning_rate": 1.3048245614035088e-05,
-      "loss": 0.0075,
-      "step": 1805
-    },
-    {
-      "epoch": 451.5,
-      "learning_rate": 1.3026315789473684e-05,
-      "loss": 0.0074,
-      "step": 1806
-    },
-    {
-      "epoch": 451.75,
-      "learning_rate": 1.300438596491228e-05,
-      "loss": 0.0072,
-      "step": 1807
-    },
-    {
-      "epoch": 452.0,
-      "learning_rate": 1.2982456140350877e-05,
-      "loss": 0.0071,
-      "step": 1808
-    },
-    {
-      "epoch": 452.25,
-      "learning_rate": 1.2960526315789473e-05,
-      "loss": 0.0074,
-      "step": 1809
-    },
-    {
-      "epoch": 452.5,
-      "learning_rate": 1.2938596491228071e-05,
-      "loss": 0.0077,
-      "step": 1810
-    },
-    {
-      "epoch": 452.75,
-      "learning_rate": 1.2916666666666668e-05,
-      "loss": 0.0074,
-      "step": 1811
-    },
-    {
-      "epoch": 453.0,
-      "learning_rate": 1.2894736842105264e-05,
-      "loss": 0.0066,
-      "step": 1812
-    },
-    {
-      "epoch": 453.25,
-      "learning_rate": 1.2872807017543862e-05,
-      "loss": 0.0077,
-      "step": 1813
-    },
-    {
-      "epoch": 453.5,
-      "learning_rate": 1.2850877192982459e-05,
-      "loss": 0.0074,
-      "step": 1814
-    },
-    {
-      "epoch": 453.75,
-      "learning_rate": 1.2828947368421055e-05,
-      "loss": 0.0069,
-      "step": 1815
-    },
-    {
-      "epoch": 454.0,
-      "learning_rate": 1.2807017543859651e-05,
-      "loss": 0.0068,
-      "step": 1816
-    },
-    {
-      "epoch": 454.25,
-      "learning_rate": 1.2785087719298248e-05,
-      "loss": 0.0069,
-      "step": 1817
-    },
-    {
-      "epoch": 454.5,
-      "learning_rate": 1.2763157894736844e-05,
-      "loss": 0.0074,
-      "step": 1818
-    },
-    {
-      "epoch": 454.75,
-      "learning_rate": 1.274122807017544e-05,
-      "loss": 0.0067,
-      "step": 1819
-    },
-    {
-      "epoch": 455.0,
-      "learning_rate": 1.2719298245614037e-05,
-      "loss": 0.0069,
-      "step": 1820
-    },
-    {
-      "epoch": 455.0,
-      "eval_accuracy_ELSE": 0.9971376627311783,
-      "eval_accuracy_road": 0.994375462778379,
-      "eval_accuracy_sidewalk": 0.9072092368346945,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9938734016103893,
-      "eval_iou_road": 0.987667067392688,
-      "eval_iou_sidewalk": 0.8476516247862124,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.021911904215812683,
-      "eval_mean_accuracy": 0.966240787448084,
-      "eval_mean_iou": 0.9430640312630966,
-      "eval_overall_accuracy": 0.9939613342285156,
-      "eval_runtime": 0.6105,
-      "eval_samples_per_second": 1.638,
-      "eval_steps_per_second": 1.638,
-      "step": 1820
-    },
-    {
-      "epoch": 455.25,
-      "learning_rate": 1.2697368421052633e-05,
-      "loss": 0.0064,
-      "step": 1821
-    },
-    {
-      "epoch": 455.5,
-      "learning_rate": 1.267543859649123e-05,
-      "loss": 0.0079,
-      "step": 1822
-    },
-    {
-      "epoch": 455.75,
-      "learning_rate": 1.2653508771929826e-05,
-      "loss": 0.0067,
-      "step": 1823
-    },
-    {
-      "epoch": 456.0,
-      "learning_rate": 1.2631578947368422e-05,
-      "loss": 0.0075,
-      "step": 1824
-    },
-    {
-      "epoch": 456.25,
-      "learning_rate": 1.2609649122807019e-05,
-      "loss": 0.0074,
-      "step": 1825
-    },
-    {
-      "epoch": 456.5,
-      "learning_rate": 1.2587719298245615e-05,
-      "loss": 0.0073,
-      "step": 1826
-    },
-    {
-      "epoch": 456.75,
-      "learning_rate": 1.2565789473684211e-05,
-      "loss": 0.007,
-      "step": 1827
-    },
-    {
-      "epoch": 457.0,
-      "learning_rate": 1.2543859649122808e-05,
-      "loss": 0.0069,
-      "step": 1828
-    },
-    {
-      "epoch": 457.25,
-      "learning_rate": 1.2521929824561404e-05,
-      "loss": 0.0076,
-      "step": 1829
-    },
-    {
-      "epoch": 457.5,
-      "learning_rate": 1.25e-05,
-      "loss": 0.0069,
-      "step": 1830
-    },
-    {
-      "epoch": 457.75,
-      "learning_rate": 1.2478070175438597e-05,
-      "loss": 0.0075,
-      "step": 1831
-    },
-    {
-      "epoch": 458.0,
-      "learning_rate": 1.2456140350877193e-05,
-      "loss": 0.0066,
-      "step": 1832
-    },
-    {
-      "epoch": 458.25,
-      "learning_rate": 1.243421052631579e-05,
-      "loss": 0.0075,
-      "step": 1833
-    },
-    {
-      "epoch": 458.5,
-      "learning_rate": 1.2412280701754386e-05,
-      "loss": 0.0075,
-      "step": 1834
-    },
-    {
-      "epoch": 458.75,
-      "learning_rate": 1.2390350877192982e-05,
-      "loss": 0.0076,
-      "step": 1835
-    },
-    {
-      "epoch": 459.0,
-      "learning_rate": 1.2368421052631579e-05,
-      "loss": 0.0067,
-      "step": 1836
-    },
-    {
-      "epoch": 459.25,
-      "learning_rate": 1.2346491228070175e-05,
-      "loss": 0.0065,
-      "step": 1837
-    },
-    {
-      "epoch": 459.5,
-      "learning_rate": 1.2324561403508772e-05,
-      "loss": 0.0076,
-      "step": 1838
-    },
-    {
-      "epoch": 459.75,
-      "learning_rate": 1.2302631578947368e-05,
-      "loss": 0.007,
-      "step": 1839
-    },
-    {
-      "epoch": 460.0,
-      "learning_rate": 1.2280701754385964e-05,
-      "loss": 0.0072,
-      "step": 1840
-    },
-    {
-      "epoch": 460.0,
-      "eval_accuracy_ELSE": 0.9970186241302066,
-      "eval_accuracy_road": 0.9939482827362306,
-      "eval_accuracy_sidewalk": 0.9086172909039707,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935886374911028,
-      "eval_iou_road": 0.9876758072275519,
-      "eval_iou_sidewalk": 0.8443019756640063,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022085409611463547,
-      "eval_mean_accuracy": 0.9665280659234693,
-      "eval_mean_iou": 0.9418554734608869,
-      "eval_overall_accuracy": 0.9938011169433594,
-      "eval_runtime": 0.7254,
-      "eval_samples_per_second": 1.379,
-      "eval_steps_per_second": 1.379,
-      "step": 1840
-    },
-    {
-      "epoch": 460.25,
-      "learning_rate": 1.2258771929824562e-05,
-      "loss": 0.0074,
-      "step": 1841
-    },
-    {
-      "epoch": 460.5,
-      "learning_rate": 1.2236842105263159e-05,
-      "loss": 0.0068,
-      "step": 1842
-    },
-    {
-      "epoch": 460.75,
-      "learning_rate": 1.2214912280701755e-05,
-      "loss": 0.006,
-      "step": 1843
-    },
-    {
-      "epoch": 461.0,
-      "learning_rate": 1.2192982456140352e-05,
-      "loss": 0.0081,
-      "step": 1844
-    },
-    {
-      "epoch": 461.25,
-      "learning_rate": 1.2171052631578948e-05,
-      "loss": 0.0074,
-      "step": 1845
-    },
-    {
-      "epoch": 461.5,
-      "learning_rate": 1.2149122807017544e-05,
-      "loss": 0.0068,
-      "step": 1846
-    },
-    {
-      "epoch": 461.75,
-      "learning_rate": 1.212719298245614e-05,
-      "loss": 0.0075,
-      "step": 1847
-    },
-    {
-      "epoch": 462.0,
-      "learning_rate": 1.2105263157894737e-05,
-      "loss": 0.0076,
-      "step": 1848
-    },
-    {
-      "epoch": 462.25,
-      "learning_rate": 1.2083333333333333e-05,
-      "loss": 0.0062,
-      "step": 1849
-    },
-    {
-      "epoch": 462.5,
-      "learning_rate": 1.206140350877193e-05,
-      "loss": 0.0085,
-      "step": 1850
-    },
-    {
-      "epoch": 462.75,
-      "learning_rate": 1.2039473684210526e-05,
-      "loss": 0.0074,
-      "step": 1851
-    },
-    {
-      "epoch": 463.0,
-      "learning_rate": 1.2017543859649123e-05,
-      "loss": 0.0072,
-      "step": 1852
-    },
-    {
-      "epoch": 463.25,
-      "learning_rate": 1.1995614035087719e-05,
-      "loss": 0.0071,
-      "step": 1853
-    },
-    {
-      "epoch": 463.5,
-      "learning_rate": 1.1973684210526315e-05,
-      "loss": 0.0065,
-      "step": 1854
-    },
-    {
-      "epoch": 463.75,
-      "learning_rate": 1.1951754385964913e-05,
-      "loss": 0.0076,
-      "step": 1855
-    },
-    {
-      "epoch": 464.0,
-      "learning_rate": 1.192982456140351e-05,
-      "loss": 0.0075,
-      "step": 1856
-    },
-    {
-      "epoch": 464.25,
-      "learning_rate": 1.1907894736842106e-05,
-      "loss": 0.0071,
-      "step": 1857
-    },
-    {
-      "epoch": 464.5,
-      "learning_rate": 1.1885964912280702e-05,
-      "loss": 0.007,
-      "step": 1858
-    },
-    {
-      "epoch": 464.75,
-      "learning_rate": 1.1864035087719299e-05,
-      "loss": 0.0071,
-      "step": 1859
-    },
-    {
-      "epoch": 465.0,
-      "learning_rate": 1.1842105263157895e-05,
-      "loss": 0.0078,
-      "step": 1860
-    },
-    {
-      "epoch": 465.0,
-      "eval_accuracy_ELSE": 0.9972891664051424,
-      "eval_accuracy_road": 0.9945320954605,
-      "eval_accuracy_sidewalk": 0.9018586313714446,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9938582490347907,
-      "eval_iou_road": 0.9876969199875555,
-      "eval_iou_sidewalk": 0.8472222222222222,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02216063253581524,
-      "eval_mean_accuracy": 0.9645599644123624,
-      "eval_mean_iou": 0.9429257970815228,
-      "eval_overall_accuracy": 0.9939651489257812,
-      "eval_runtime": 0.6064,
-      "eval_samples_per_second": 1.649,
-      "eval_steps_per_second": 1.649,
-      "step": 1860
-    },
-    {
-      "epoch": 465.25,
-      "learning_rate": 1.1820175438596493e-05,
-      "loss": 0.0077,
-      "step": 1861
-    },
-    {
-      "epoch": 465.5,
-      "learning_rate": 1.179824561403509e-05,
-      "loss": 0.0072,
-      "step": 1862
-    },
-    {
-      "epoch": 465.75,
-      "learning_rate": 1.1776315789473686e-05,
-      "loss": 0.0079,
-      "step": 1863
-    },
-    {
-      "epoch": 466.0,
-      "learning_rate": 1.1754385964912282e-05,
-      "loss": 0.0065,
-      "step": 1864
-    },
-    {
-      "epoch": 466.25,
-      "learning_rate": 1.1732456140350879e-05,
-      "loss": 0.007,
-      "step": 1865
-    },
-    {
-      "epoch": 466.5,
-      "learning_rate": 1.1710526315789475e-05,
-      "loss": 0.0066,
-      "step": 1866
-    },
-    {
-      "epoch": 466.75,
-      "learning_rate": 1.1688596491228072e-05,
-      "loss": 0.0078,
-      "step": 1867
-    },
-    {
-      "epoch": 467.0,
-      "learning_rate": 1.1666666666666668e-05,
-      "loss": 0.0073,
-      "step": 1868
-    },
-    {
-      "epoch": 467.25,
-      "learning_rate": 1.1644736842105264e-05,
-      "loss": 0.0069,
-      "step": 1869
-    },
-    {
-      "epoch": 467.5,
-      "learning_rate": 1.162280701754386e-05,
-      "loss": 0.0085,
-      "step": 1870
-    },
-    {
-      "epoch": 467.75,
-      "learning_rate": 1.1600877192982457e-05,
-      "loss": 0.0068,
-      "step": 1871
-    },
-    {
-      "epoch": 468.0,
-      "learning_rate": 1.1578947368421053e-05,
-      "loss": 0.0072,
-      "step": 1872
-    },
-    {
-      "epoch": 468.25,
-      "learning_rate": 1.155701754385965e-05,
-      "loss": 0.0063,
-      "step": 1873
-    },
-    {
-      "epoch": 468.5,
-      "learning_rate": 1.1535087719298246e-05,
-      "loss": 0.0076,
-      "step": 1874
-    },
-    {
-      "epoch": 468.75,
-      "learning_rate": 1.1513157894736843e-05,
-      "loss": 0.0081,
-      "step": 1875
-    },
-    {
-      "epoch": 469.0,
-      "learning_rate": 1.1491228070175439e-05,
-      "loss": 0.0065,
-      "step": 1876
-    },
-    {
-      "epoch": 469.25,
-      "learning_rate": 1.1469298245614035e-05,
-      "loss": 0.0069,
-      "step": 1877
-    },
-    {
-      "epoch": 469.5,
-      "learning_rate": 1.1447368421052632e-05,
-      "loss": 0.0075,
-      "step": 1878
-    },
-    {
-      "epoch": 469.75,
-      "learning_rate": 1.1425438596491228e-05,
-      "loss": 0.007,
-      "step": 1879
-    },
-    {
-      "epoch": 470.0,
-      "learning_rate": 1.1403508771929824e-05,
-      "loss": 0.0072,
-      "step": 1880
-    },
-    {
-      "epoch": 470.0,
-      "eval_accuracy_ELSE": 0.9968941746837361,
-      "eval_accuracy_road": 0.9946175314689297,
-      "eval_accuracy_sidewalk": 0.9049563503238525,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936896607518473,
-      "eval_iou_road": 0.9875722829391056,
-      "eval_iou_sidewalk": 0.8421121593291404,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.023017089813947678,
-      "eval_mean_accuracy": 0.9654893521588394,
-      "eval_mean_iou": 0.9411247010066978,
-      "eval_overall_accuracy": 0.9937934875488281,
-      "eval_runtime": 0.6071,
-      "eval_samples_per_second": 1.647,
-      "eval_steps_per_second": 1.647,
-      "step": 1880
-    },
-    {
-      "epoch": 470.25,
-      "learning_rate": 1.1381578947368422e-05,
-      "loss": 0.0065,
-      "step": 1881
-    },
-    {
-      "epoch": 470.5,
-      "learning_rate": 1.1359649122807019e-05,
-      "loss": 0.0072,
-      "step": 1882
-    },
-    {
-      "epoch": 470.75,
-      "learning_rate": 1.1337719298245615e-05,
-      "loss": 0.0064,
-      "step": 1883
-    },
-    {
-      "epoch": 471.0,
-      "learning_rate": 1.1315789473684212e-05,
-      "loss": 0.0073,
-      "step": 1884
-    },
-    {
-      "epoch": 471.25,
-      "learning_rate": 1.1293859649122808e-05,
-      "loss": 0.0081,
-      "step": 1885
-    },
-    {
-      "epoch": 471.5,
-      "learning_rate": 1.1271929824561404e-05,
-      "loss": 0.0066,
-      "step": 1886
-    },
-    {
-      "epoch": 471.75,
-      "learning_rate": 1.125e-05,
-      "loss": 0.0068,
-      "step": 1887
-    },
-    {
-      "epoch": 472.0,
-      "learning_rate": 1.1228070175438597e-05,
-      "loss": 0.0066,
-      "step": 1888
-    },
-    {
-      "epoch": 472.25,
-      "learning_rate": 1.1206140350877193e-05,
-      "loss": 0.0077,
-      "step": 1889
-    },
-    {
-      "epoch": 472.5,
-      "learning_rate": 1.118421052631579e-05,
-      "loss": 0.0069,
-      "step": 1890
-    },
-    {
-      "epoch": 472.75,
-      "learning_rate": 1.1162280701754386e-05,
-      "loss": 0.007,
-      "step": 1891
-    },
-    {
-      "epoch": 473.0,
-      "learning_rate": 1.1140350877192983e-05,
-      "loss": 0.0077,
-      "step": 1892
-    },
-    {
-      "epoch": 473.25,
-      "learning_rate": 1.1118421052631579e-05,
-      "loss": 0.007,
-      "step": 1893
-    },
-    {
-      "epoch": 473.5,
-      "learning_rate": 1.1096491228070175e-05,
-      "loss": 0.0069,
-      "step": 1894
-    },
-    {
-      "epoch": 473.75,
-      "learning_rate": 1.1074561403508772e-05,
-      "loss": 0.0069,
-      "step": 1895
-    },
-    {
-      "epoch": 474.0,
-      "learning_rate": 1.1052631578947368e-05,
-      "loss": 0.0069,
-      "step": 1896
-    },
-    {
-      "epoch": 474.25,
-      "learning_rate": 1.1030701754385964e-05,
-      "loss": 0.0069,
-      "step": 1897
-    },
-    {
-      "epoch": 474.5,
-      "learning_rate": 1.100877192982456e-05,
-      "loss": 0.0074,
-      "step": 1898
-    },
-    {
-      "epoch": 474.75,
-      "learning_rate": 1.0986842105263157e-05,
-      "loss": 0.0079,
-      "step": 1899
-    },
-    {
-      "epoch": 475.0,
-      "learning_rate": 1.0964912280701754e-05,
-      "loss": 0.0076,
-      "step": 1900
-    },
-    {
-      "epoch": 475.0,
-      "eval_accuracy_ELSE": 0.9968454770742476,
-      "eval_accuracy_road": 0.994631770803668,
-      "eval_accuracy_sidewalk": 0.905519571951563,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935928896175689,
-      "eval_iou_road": 0.9876562411628301,
-      "eval_iou_sidewalk": 0.8426362683438156,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022327013313770294,
-      "eval_mean_accuracy": 0.9656656066098263,
-      "eval_mean_iou": 0.9412951330414049,
-      "eval_overall_accuracy": 0.9937782287597656,
-      "eval_runtime": 0.5969,
-      "eval_samples_per_second": 1.675,
-      "eval_steps_per_second": 1.675,
-      "step": 1900
-    },
-    {
-      "epoch": 475.25,
-      "learning_rate": 1.094298245614035e-05,
-      "loss": 0.0071,
-      "step": 1901
-    },
-    {
-      "epoch": 475.5,
-      "learning_rate": 1.0921052631578948e-05,
-      "loss": 0.0084,
-      "step": 1902
-    },
-    {
-      "epoch": 475.75,
-      "learning_rate": 1.0899122807017544e-05,
-      "loss": 0.0068,
-      "step": 1903
-    },
-    {
-      "epoch": 476.0,
-      "learning_rate": 1.087719298245614e-05,
-      "loss": 0.0063,
-      "step": 1904
-    },
-    {
-      "epoch": 476.25,
-      "learning_rate": 1.0855263157894737e-05,
-      "loss": 0.0073,
-      "step": 1905
-    },
-    {
-      "epoch": 476.5,
-      "learning_rate": 1.0833333333333334e-05,
-      "loss": 0.007,
-      "step": 1906
-    },
-    {
-      "epoch": 476.75,
-      "learning_rate": 1.081140350877193e-05,
-      "loss": 0.0071,
-      "step": 1907
-    },
-    {
-      "epoch": 477.0,
-      "learning_rate": 1.0789473684210526e-05,
-      "loss": 0.0062,
-      "step": 1908
-    },
-    {
-      "epoch": 477.25,
-      "learning_rate": 1.0767543859649124e-05,
-      "loss": 0.0069,
-      "step": 1909
-    },
-    {
-      "epoch": 477.5,
-      "learning_rate": 1.074561403508772e-05,
-      "loss": 0.007,
-      "step": 1910
-    },
-    {
-      "epoch": 477.75,
-      "learning_rate": 1.0723684210526317e-05,
-      "loss": 0.0067,
-      "step": 1911
-    },
-    {
-      "epoch": 478.0,
-      "learning_rate": 1.0701754385964913e-05,
-      "loss": 0.0073,
-      "step": 1912
-    },
-    {
-      "epoch": 478.25,
-      "learning_rate": 1.067982456140351e-05,
-      "loss": 0.0068,
-      "step": 1913
-    },
-    {
-      "epoch": 478.5,
-      "learning_rate": 1.0657894736842106e-05,
-      "loss": 0.0081,
-      "step": 1914
-    },
-    {
-      "epoch": 478.75,
-      "learning_rate": 1.0635964912280703e-05,
-      "loss": 0.0069,
-      "step": 1915
-    },
-    {
-      "epoch": 479.0,
-      "learning_rate": 1.0614035087719299e-05,
-      "loss": 0.0071,
-      "step": 1916
-    },
-    {
-      "epoch": 479.25,
-      "learning_rate": 1.0592105263157895e-05,
-      "loss": 0.0069,
-      "step": 1917
-    },
-    {
-      "epoch": 479.5,
-      "learning_rate": 1.0570175438596492e-05,
-      "loss": 0.0073,
-      "step": 1918
-    },
-    {
-      "epoch": 479.75,
-      "learning_rate": 1.0548245614035088e-05,
-      "loss": 0.0073,
-      "step": 1919
-    },
-    {
-      "epoch": 480.0,
-      "learning_rate": 1.0526315789473684e-05,
-      "loss": 0.0072,
-      "step": 1920
-    },
-    {
-      "epoch": 480.0,
-      "eval_accuracy_ELSE": 0.9967697252372656,
-      "eval_accuracy_road": 0.9946602494731446,
-      "eval_accuracy_sidewalk": 0.9080540692762602,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935924058142984,
-      "eval_iou_road": 0.9878940446053543,
-      "eval_iou_sidewalk": 0.8424559111691705,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022111717611551285,
-      "eval_mean_accuracy": 0.9664946813288902,
-      "eval_mean_iou": 0.9413141205296078,
-      "eval_overall_accuracy": 0.9938011169433594,
-      "eval_runtime": 0.7331,
-      "eval_samples_per_second": 1.364,
-      "eval_steps_per_second": 1.364,
-      "step": 1920
-    },
-    {
-      "epoch": 480.25,
-      "learning_rate": 1.050438596491228e-05,
-      "loss": 0.0069,
-      "step": 1921
-    },
-    {
-      "epoch": 480.5,
-      "learning_rate": 1.0482456140350879e-05,
-      "loss": 0.0064,
-      "step": 1922
-    },
-    {
-      "epoch": 480.75,
-      "learning_rate": 1.0460526315789475e-05,
-      "loss": 0.0079,
-      "step": 1923
-    },
-    {
-      "epoch": 481.0,
-      "learning_rate": 1.0438596491228072e-05,
-      "loss": 0.0072,
-      "step": 1924
-    },
-    {
-      "epoch": 481.25,
-      "learning_rate": 1.0416666666666668e-05,
-      "loss": 0.0071,
-      "step": 1925
-    },
-    {
-      "epoch": 481.5,
-      "learning_rate": 1.0394736842105264e-05,
-      "loss": 0.0069,
-      "step": 1926
-    },
-    {
-      "epoch": 481.75,
-      "learning_rate": 1.037280701754386e-05,
-      "loss": 0.007,
-      "step": 1927
-    },
-    {
-      "epoch": 482.0,
-      "learning_rate": 1.0350877192982457e-05,
-      "loss": 0.0069,
-      "step": 1928
-    },
-    {
-      "epoch": 482.25,
-      "learning_rate": 1.0328947368421054e-05,
-      "loss": 0.007,
-      "step": 1929
-    },
-    {
-      "epoch": 482.5,
-      "learning_rate": 1.030701754385965e-05,
-      "loss": 0.0068,
-      "step": 1930
-    },
-    {
-      "epoch": 482.75,
-      "learning_rate": 1.0285087719298246e-05,
-      "loss": 0.0071,
-      "step": 1931
-    },
-    {
-      "epoch": 483.0,
-      "learning_rate": 1.0263157894736843e-05,
-      "loss": 0.0065,
-      "step": 1932
-    },
-    {
-      "epoch": 483.25,
-      "learning_rate": 1.0241228070175439e-05,
-      "loss": 0.0078,
-      "step": 1933
-    },
-    {
-      "epoch": 483.5,
-      "learning_rate": 1.0219298245614035e-05,
-      "loss": 0.0069,
-      "step": 1934
-    },
-    {
-      "epoch": 483.75,
-      "learning_rate": 1.0197368421052632e-05,
-      "loss": 0.0065,
-      "step": 1935
-    },
-    {
-      "epoch": 484.0,
-      "learning_rate": 1.0175438596491228e-05,
-      "loss": 0.0074,
-      "step": 1936
-    },
-    {
-      "epoch": 484.25,
-      "learning_rate": 1.0153508771929825e-05,
-      "loss": 0.0073,
-      "step": 1937
-    },
-    {
-      "epoch": 484.5,
-      "learning_rate": 1.0131578947368421e-05,
-      "loss": 0.0076,
-      "step": 1938
-    },
-    {
-      "epoch": 484.75,
-      "learning_rate": 1.0109649122807017e-05,
-      "loss": 0.0067,
-      "step": 1939
-    },
-    {
-      "epoch": 485.0,
-      "learning_rate": 1.0087719298245614e-05,
-      "loss": 0.0073,
-      "step": 1940
-    },
-    {
-      "epoch": 485.0,
-      "eval_accuracy_ELSE": 0.9971809494951681,
-      "eval_accuracy_road": 0.9935780600330353,
-      "eval_accuracy_sidewalk": 0.9083356800901154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936432453416149,
-      "eval_iou_road": 0.9875035380696292,
-      "eval_iou_sidewalk": 0.8449246889325475,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022527044638991356,
-      "eval_mean_accuracy": 0.9663648965394396,
-      "eval_mean_iou": 0.9420238241145972,
-      "eval_overall_accuracy": 0.9938087463378906,
-      "eval_runtime": 0.603,
-      "eval_samples_per_second": 1.658,
-      "eval_steps_per_second": 1.658,
-      "step": 1940
-    },
-    {
-      "epoch": 485.25,
-      "learning_rate": 1.006578947368421e-05,
-      "loss": 0.0079,
-      "step": 1941
-    },
-    {
-      "epoch": 485.5,
-      "learning_rate": 1.0043859649122808e-05,
-      "loss": 0.0075,
-      "step": 1942
-    },
-    {
-      "epoch": 485.75,
-      "learning_rate": 1.0021929824561404e-05,
-      "loss": 0.0066,
-      "step": 1943
-    },
-    {
-      "epoch": 486.0,
-      "learning_rate": 1e-05,
-      "loss": 0.007,
-      "step": 1944
-    },
-    {
-      "epoch": 486.25,
-      "learning_rate": 9.978070175438597e-06,
-      "loss": 0.0069,
-      "step": 1945
-    },
-    {
-      "epoch": 486.5,
-      "learning_rate": 9.956140350877194e-06,
-      "loss": 0.0068,
-      "step": 1946
-    },
-    {
-      "epoch": 486.75,
-      "learning_rate": 9.93421052631579e-06,
-      "loss": 0.0072,
-      "step": 1947
-    },
-    {
-      "epoch": 487.0,
-      "learning_rate": 9.912280701754386e-06,
-      "loss": 0.0064,
-      "step": 1948
-    },
-    {
-      "epoch": 487.25,
-      "learning_rate": 9.890350877192983e-06,
-      "loss": 0.0069,
-      "step": 1949
-    },
-    {
-      "epoch": 487.5,
-      "learning_rate": 9.868421052631579e-06,
-      "loss": 0.0073,
-      "step": 1950
-    },
-    {
-      "epoch": 487.75,
-      "learning_rate": 9.846491228070175e-06,
-      "loss": 0.0072,
-      "step": 1951
-    },
-    {
-      "epoch": 488.0,
-      "learning_rate": 9.824561403508772e-06,
-      "loss": 0.0082,
-      "step": 1952
-    },
-    {
-      "epoch": 488.25,
-      "learning_rate": 9.802631578947368e-06,
-      "loss": 0.0068,
-      "step": 1953
-    },
-    {
-      "epoch": 488.5,
-      "learning_rate": 9.780701754385965e-06,
-      "loss": 0.0074,
-      "step": 1954
-    },
-    {
-      "epoch": 488.75,
-      "learning_rate": 9.758771929824561e-06,
-      "loss": 0.0068,
-      "step": 1955
-    },
-    {
-      "epoch": 489.0,
-      "learning_rate": 9.736842105263157e-06,
-      "loss": 0.007,
-      "step": 1956
-    },
-    {
-      "epoch": 489.25,
-      "learning_rate": 9.714912280701754e-06,
-      "loss": 0.0072,
-      "step": 1957
-    },
-    {
-      "epoch": 489.5,
-      "learning_rate": 9.692982456140352e-06,
-      "loss": 0.0069,
-      "step": 1958
-    },
-    {
-      "epoch": 489.75,
-      "learning_rate": 9.671052631578948e-06,
-      "loss": 0.007,
-      "step": 1959
-    },
-    {
-      "epoch": 490.0,
-      "learning_rate": 9.649122807017545e-06,
-      "loss": 0.0079,
-      "step": 1960
-    },
-    {
-      "epoch": 490.0,
-      "eval_accuracy_ELSE": 0.9969212289112297,
-      "eval_accuracy_road": 0.9938913253972774,
-      "eval_accuracy_sidewalk": 0.9083356800901154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9934165830933972,
-      "eval_iou_road": 0.9875213989615313,
-      "eval_iou_sidewalk": 0.843709128956317,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022614937275648117,
-      "eval_mean_accuracy": 0.9663827447995409,
-      "eval_mean_iou": 0.9415490370037484,
-      "eval_overall_accuracy": 0.9937095642089844,
-      "eval_runtime": 0.6147,
-      "eval_samples_per_second": 1.627,
-      "eval_steps_per_second": 1.627,
-      "step": 1960
-    },
-    {
-      "epoch": 490.25,
-      "learning_rate": 9.627192982456141e-06,
-      "loss": 0.0068,
-      "step": 1961
-    },
-    {
-      "epoch": 490.5,
-      "learning_rate": 9.605263157894737e-06,
-      "loss": 0.0073,
-      "step": 1962
-    },
-    {
-      "epoch": 490.75,
-      "learning_rate": 9.583333333333334e-06,
-      "loss": 0.0071,
-      "step": 1963
-    },
-    {
-      "epoch": 491.0,
-      "learning_rate": 9.561403508771932e-06,
-      "loss": 0.0069,
-      "step": 1964
-    },
-    {
-      "epoch": 491.25,
-      "learning_rate": 9.539473684210528e-06,
-      "loss": 0.0072,
-      "step": 1965
-    },
-    {
-      "epoch": 491.5,
-      "learning_rate": 9.517543859649124e-06,
-      "loss": 0.0063,
-      "step": 1966
-    },
-    {
-      "epoch": 491.75,
-      "learning_rate": 9.49561403508772e-06,
-      "loss": 0.0076,
-      "step": 1967
-    },
-    {
-      "epoch": 492.0,
-      "learning_rate": 9.473684210526317e-06,
-      "loss": 0.0071,
-      "step": 1968
-    },
-    {
-      "epoch": 492.25,
-      "learning_rate": 9.451754385964914e-06,
-      "loss": 0.0067,
-      "step": 1969
-    },
-    {
-      "epoch": 492.5,
-      "learning_rate": 9.42982456140351e-06,
-      "loss": 0.0068,
-      "step": 1970
-    },
-    {
-      "epoch": 492.75,
-      "learning_rate": 9.407894736842106e-06,
-      "loss": 0.0077,
-      "step": 1971
-    },
-    {
-      "epoch": 493.0,
-      "learning_rate": 9.385964912280703e-06,
-      "loss": 0.0066,
-      "step": 1972
-    },
-    {
-      "epoch": 493.25,
-      "learning_rate": 9.364035087719299e-06,
-      "loss": 0.0067,
-      "step": 1973
-    },
-    {
-      "epoch": 493.5,
-      "learning_rate": 9.342105263157895e-06,
-      "loss": 0.0072,
-      "step": 1974
-    },
-    {
-      "epoch": 493.75,
-      "learning_rate": 9.320175438596492e-06,
-      "loss": 0.0066,
-      "step": 1975
-    },
-    {
-      "epoch": 494.0,
-      "learning_rate": 9.298245614035088e-06,
-      "loss": 0.0065,
-      "step": 1976
-    },
-    {
-      "epoch": 494.25,
-      "learning_rate": 9.276315789473685e-06,
-      "loss": 0.0077,
-      "step": 1977
-    },
-    {
-      "epoch": 494.5,
-      "learning_rate": 9.254385964912281e-06,
-      "loss": 0.0063,
-      "step": 1978
-    },
-    {
-      "epoch": 494.75,
-      "learning_rate": 9.232456140350877e-06,
-      "loss": 0.0068,
-      "step": 1979
-    },
-    {
-      "epoch": 495.0,
-      "learning_rate": 9.210526315789474e-06,
-      "loss": 0.007,
-      "step": 1980
-    },
-    {
-      "epoch": 495.0,
-      "eval_accuracy_ELSE": 0.996991569902713,
-      "eval_accuracy_road": 0.9940764367488751,
-      "eval_accuracy_sidewalk": 0.9059419881723458,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935241751546163,
-      "eval_iou_road": 0.9874957564784429,
-      "eval_iou_sidewalk": 0.843360859876786,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02254437282681465,
-      "eval_mean_accuracy": 0.9656699982746447,
-      "eval_mean_iou": 0.9414602638366151,
-      "eval_overall_accuracy": 0.993743896484375,
-      "eval_runtime": 0.5975,
-      "eval_samples_per_second": 1.674,
-      "eval_steps_per_second": 1.674,
-      "step": 1980
-    },
-    {
-      "epoch": 495.25,
-      "learning_rate": 9.18859649122807e-06,
-      "loss": 0.007,
-      "step": 1981
-    },
-    {
-      "epoch": 495.5,
-      "learning_rate": 9.166666666666666e-06,
-      "loss": 0.0067,
-      "step": 1982
-    },
-    {
-      "epoch": 495.75,
-      "learning_rate": 9.144736842105264e-06,
-      "loss": 0.0075,
-      "step": 1983
-    },
-    {
-      "epoch": 496.0,
-      "learning_rate": 9.122807017543861e-06,
-      "loss": 0.0077,
-      "step": 1984
-    },
-    {
-      "epoch": 496.25,
-      "learning_rate": 9.100877192982457e-06,
-      "loss": 0.0079,
-      "step": 1985
-    },
-    {
-      "epoch": 496.5,
-      "learning_rate": 9.078947368421054e-06,
-      "loss": 0.0066,
-      "step": 1986
-    },
-    {
-      "epoch": 496.75,
-      "learning_rate": 9.05701754385965e-06,
-      "loss": 0.0069,
-      "step": 1987
-    },
-    {
-      "epoch": 497.0,
-      "learning_rate": 9.035087719298246e-06,
-      "loss": 0.0066,
-      "step": 1988
-    },
-    {
-      "epoch": 497.25,
-      "learning_rate": 9.013157894736843e-06,
-      "loss": 0.0068,
-      "step": 1989
-    },
-    {
-      "epoch": 497.5,
-      "learning_rate": 8.991228070175439e-06,
-      "loss": 0.0066,
-      "step": 1990
-    },
-    {
-      "epoch": 497.75,
-      "learning_rate": 8.969298245614035e-06,
-      "loss": 0.0071,
-      "step": 1991
-    },
-    {
-      "epoch": 498.0,
-      "learning_rate": 8.947368421052632e-06,
-      "loss": 0.0064,
-      "step": 1992
-    },
-    {
-      "epoch": 498.25,
-      "learning_rate": 8.925438596491228e-06,
-      "loss": 0.0067,
-      "step": 1993
-    },
-    {
-      "epoch": 498.5,
-      "learning_rate": 8.903508771929825e-06,
-      "loss": 0.0078,
-      "step": 1994
-    },
-    {
-      "epoch": 498.75,
-      "learning_rate": 8.881578947368421e-06,
-      "loss": 0.007,
-      "step": 1995
-    },
-    {
-      "epoch": 499.0,
-      "learning_rate": 8.859649122807017e-06,
-      "loss": 0.0065,
-      "step": 1996
-    },
-    {
-      "epoch": 499.25,
-      "learning_rate": 8.837719298245614e-06,
-      "loss": 0.007,
-      "step": 1997
-    },
-    {
-      "epoch": 499.5,
-      "learning_rate": 8.81578947368421e-06,
-      "loss": 0.0069,
-      "step": 1998
-    },
-    {
-      "epoch": 499.75,
-      "learning_rate": 8.793859649122806e-06,
-      "loss": 0.0069,
-      "step": 1999
-    },
-    {
-      "epoch": 500.0,
-      "learning_rate": 8.771929824561403e-06,
-      "loss": 0.0072,
-      "step": 2000
-    },
-    {
-      "epoch": 500.0,
-      "eval_accuracy_ELSE": 0.9968995855292347,
-      "eval_accuracy_road": 0.9945036167910235,
-      "eval_accuracy_sidewalk": 0.905519571951563,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935021515697292,
-      "eval_iou_road": 0.9878083276759447,
-      "eval_iou_sidewalk": 0.843520461699895,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022378191351890564,
-      "eval_mean_accuracy": 0.9656409247572738,
-      "eval_mean_iou": 0.941610313648523,
-      "eval_overall_accuracy": 0.9937820434570312,
-      "eval_runtime": 0.7261,
-      "eval_samples_per_second": 1.377,
-      "eval_steps_per_second": 1.377,
-      "step": 2000
-    },
-    {
-      "epoch": 500.25,
-      "learning_rate": 8.75e-06,
-      "loss": 0.0065,
-      "step": 2001
-    },
-    {
-      "epoch": 500.5,
-      "learning_rate": 8.728070175438596e-06,
-      "loss": 0.0073,
-      "step": 2002
-    },
-    {
-      "epoch": 500.75,
-      "learning_rate": 8.706140350877192e-06,
-      "loss": 0.0068,
-      "step": 2003
-    },
-    {
-      "epoch": 501.0,
-      "learning_rate": 8.68421052631579e-06,
-      "loss": 0.0073,
-      "step": 2004
-    },
-    {
-      "epoch": 501.25,
-      "learning_rate": 8.662280701754386e-06,
-      "loss": 0.0071,
-      "step": 2005
-    },
-    {
-      "epoch": 501.5,
-      "learning_rate": 8.640350877192983e-06,
-      "loss": 0.007,
-      "step": 2006
-    },
-    {
-      "epoch": 501.75,
-      "learning_rate": 8.61842105263158e-06,
-      "loss": 0.0067,
-      "step": 2007
-    },
-    {
-      "epoch": 502.0,
-      "learning_rate": 8.596491228070176e-06,
-      "loss": 0.0074,
-      "step": 2008
-    },
-    {
-      "epoch": 502.25,
-      "learning_rate": 8.574561403508772e-06,
-      "loss": 0.0067,
-      "step": 2009
-    },
-    {
-      "epoch": 502.5,
-      "learning_rate": 8.552631578947368e-06,
-      "loss": 0.0068,
-      "step": 2010
-    },
-    {
-      "epoch": 502.75,
-      "learning_rate": 8.530701754385965e-06,
-      "loss": 0.0075,
-      "step": 2011
-    },
-    {
-      "epoch": 503.0,
-      "learning_rate": 8.508771929824563e-06,
-      "loss": 0.0074,
-      "step": 2012
-    },
-    {
-      "epoch": 503.25,
-      "learning_rate": 8.486842105263159e-06,
-      "loss": 0.0062,
-      "step": 2013
-    },
-    {
-      "epoch": 503.5,
-      "learning_rate": 8.464912280701755e-06,
-      "loss": 0.0073,
-      "step": 2014
-    },
-    {
-      "epoch": 503.75,
-      "learning_rate": 8.442982456140352e-06,
-      "loss": 0.0066,
-      "step": 2015
-    },
-    {
-      "epoch": 504.0,
-      "learning_rate": 8.421052631578948e-06,
-      "loss": 0.0076,
-      "step": 2016
-    },
-    {
-      "epoch": 504.25,
-      "learning_rate": 8.399122807017545e-06,
-      "loss": 0.0075,
-      "step": 2017
-    },
-    {
-      "epoch": 504.5,
-      "learning_rate": 8.377192982456141e-06,
-      "loss": 0.0077,
-      "step": 2018
-    },
-    {
-      "epoch": 504.75,
-      "learning_rate": 8.355263157894737e-06,
-      "loss": 0.0068,
-      "step": 2019
-    },
-    {
-      "epoch": 505.0,
-      "learning_rate": 8.333333333333334e-06,
-      "loss": 0.0068,
-      "step": 2020
-    },
-    {
-      "epoch": 505.0,
-      "eval_accuracy_ELSE": 0.9969266397567283,
-      "eval_accuracy_road": 0.994176112092043,
-      "eval_accuracy_sidewalk": 0.9080540692762602,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936201996451473,
-      "eval_iou_road": 0.9876366825569717,
-      "eval_iou_sidewalk": 0.8427861996863565,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02252509817481041,
-      "eval_mean_accuracy": 0.9663856070416772,
-      "eval_mean_iou": 0.941347693962825,
-      "eval_overall_accuracy": 0.9937820434570312,
-      "eval_runtime": 0.6101,
-      "eval_samples_per_second": 1.639,
-      "eval_steps_per_second": 1.639,
-      "step": 2020
-    },
-    {
-      "epoch": 505.25,
-      "learning_rate": 8.31140350877193e-06,
-      "loss": 0.0063,
-      "step": 2021
-    },
-    {
-      "epoch": 505.5,
-      "learning_rate": 8.289473684210526e-06,
-      "loss": 0.0071,
-      "step": 2022
-    },
-    {
-      "epoch": 505.75,
-      "learning_rate": 8.267543859649123e-06,
-      "loss": 0.0074,
-      "step": 2023
-    },
-    {
-      "epoch": 506.0,
-      "learning_rate": 8.245614035087721e-06,
-      "loss": 0.0068,
-      "step": 2024
-    },
-    {
-      "epoch": 506.25,
-      "learning_rate": 8.223684210526317e-06,
-      "loss": 0.007,
-      "step": 2025
-    },
-    {
-      "epoch": 506.5,
-      "learning_rate": 8.201754385964914e-06,
-      "loss": 0.0064,
-      "step": 2026
-    },
-    {
-      "epoch": 506.75,
-      "learning_rate": 8.17982456140351e-06,
-      "loss": 0.0065,
-      "step": 2027
-    },
-    {
-      "epoch": 507.0,
-      "learning_rate": 8.157894736842106e-06,
-      "loss": 0.0073,
-      "step": 2028
-    },
-    {
-      "epoch": 507.25,
-      "learning_rate": 8.135964912280703e-06,
-      "loss": 0.0072,
-      "step": 2029
-    },
-    {
-      "epoch": 507.5,
-      "learning_rate": 8.1140350877193e-06,
-      "loss": 0.0065,
-      "step": 2030
-    },
-    {
-      "epoch": 507.75,
-      "learning_rate": 8.092105263157896e-06,
-      "loss": 0.0064,
-      "step": 2031
-    },
-    {
-      "epoch": 508.0,
-      "learning_rate": 8.070175438596492e-06,
-      "loss": 0.0076,
-      "step": 2032
-    },
-    {
-      "epoch": 508.25,
-      "learning_rate": 8.048245614035088e-06,
-      "loss": 0.0075,
-      "step": 2033
-    },
-    {
-      "epoch": 508.5,
-      "learning_rate": 8.026315789473685e-06,
-      "loss": 0.0062,
-      "step": 2034
-    },
-    {
-      "epoch": 508.75,
-      "learning_rate": 8.004385964912281e-06,
-      "loss": 0.0071,
-      "step": 2035
-    },
-    {
-      "epoch": 509.0,
-      "learning_rate": 7.982456140350877e-06,
-      "loss": 0.0067,
-      "step": 2036
-    },
-    {
-      "epoch": 509.25,
-      "learning_rate": 7.960526315789474e-06,
-      "loss": 0.0075,
-      "step": 2037
-    },
-    {
-      "epoch": 509.5,
-      "learning_rate": 7.93859649122807e-06,
-      "loss": 0.0073,
-      "step": 2038
-    },
-    {
-      "epoch": 509.75,
-      "learning_rate": 7.916666666666667e-06,
-      "loss": 0.0066,
-      "step": 2039
-    },
-    {
-      "epoch": 510.0,
-      "learning_rate": 7.894736842105263e-06,
-      "loss": 0.0061,
-      "step": 2040
-    },
-    {
-      "epoch": 510.0,
-      "eval_accuracy_ELSE": 0.9972675230231476,
-      "eval_accuracy_road": 0.993720453380418,
-      "eval_accuracy_sidewalk": 0.9065052098000563,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9936598646790845,
-      "eval_iou_road": 0.9873935312261242,
-      "eval_iou_sidewalk": 0.8481096034778026,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02259432151913643,
-      "eval_mean_accuracy": 0.9658310620678741,
-      "eval_mean_iou": 0.9430543331276704,
-      "eval_overall_accuracy": 0.9938583374023438,
-      "eval_runtime": 0.6162,
-      "eval_samples_per_second": 1.623,
-      "eval_steps_per_second": 1.623,
-      "step": 2040
-    },
-    {
-      "epoch": 510.25,
-      "learning_rate": 7.87280701754386e-06,
-      "loss": 0.007,
-      "step": 2041
-    },
-    {
-      "epoch": 510.5,
-      "learning_rate": 7.850877192982456e-06,
-      "loss": 0.0066,
-      "step": 2042
-    },
-    {
-      "epoch": 510.75,
-      "learning_rate": 7.828947368421052e-06,
-      "loss": 0.0065,
-      "step": 2043
-    },
-    {
-      "epoch": 511.0,
-      "learning_rate": 7.80701754385965e-06,
-      "loss": 0.008,
-      "step": 2044
-    },
-    {
-      "epoch": 511.25,
-      "learning_rate": 7.785087719298246e-06,
-      "loss": 0.0064,
-      "step": 2045
-    },
-    {
-      "epoch": 511.5,
-      "learning_rate": 7.763157894736843e-06,
-      "loss": 0.0073,
-      "step": 2046
-    },
-    {
-      "epoch": 511.75,
-      "learning_rate": 7.74122807017544e-06,
-      "loss": 0.007,
-      "step": 2047
-    },
-    {
-      "epoch": 512.0,
-      "learning_rate": 7.719298245614036e-06,
-      "loss": 0.0076,
-      "step": 2048
-    },
-    {
-      "epoch": 512.25,
-      "learning_rate": 7.697368421052632e-06,
-      "loss": 0.0069,
-      "step": 2049
-    },
-    {
-      "epoch": 512.5,
-      "learning_rate": 7.675438596491228e-06,
-      "loss": 0.0071,
-      "step": 2050
-    },
-    {
-      "epoch": 512.75,
-      "learning_rate": 7.653508771929825e-06,
-      "loss": 0.0062,
-      "step": 2051
-    },
-    {
-      "epoch": 513.0,
-      "learning_rate": 7.631578947368421e-06,
-      "loss": 0.0072,
-      "step": 2052
-    },
-    {
-      "epoch": 513.25,
-      "learning_rate": 7.609649122807018e-06,
-      "loss": 0.0068,
-      "step": 2053
-    },
-    {
-      "epoch": 513.5,
-      "learning_rate": 7.587719298245615e-06,
-      "loss": 0.0079,
-      "step": 2054
-    },
-    {
-      "epoch": 513.75,
-      "learning_rate": 7.565789473684211e-06,
-      "loss": 0.0072,
-      "step": 2055
-    },
-    {
-      "epoch": 514.0,
-      "learning_rate": 7.5438596491228074e-06,
-      "loss": 0.0065,
-      "step": 2056
-    },
-    {
-      "epoch": 514.25,
-      "learning_rate": 7.521929824561404e-06,
-      "loss": 0.0065,
-      "step": 2057
-    },
-    {
-      "epoch": 514.5,
-      "learning_rate": 7.5e-06,
-      "loss": 0.0071,
-      "step": 2058
-    },
-    {
-      "epoch": 514.75,
-      "learning_rate": 7.4780701754385966e-06,
-      "loss": 0.0073,
-      "step": 2059
-    },
-    {
-      "epoch": 515.0,
-      "learning_rate": 7.456140350877193e-06,
-      "loss": 0.0066,
-      "step": 2060
-    },
-    {
-      "epoch": 515.0,
-      "eval_accuracy_ELSE": 0.9968725313017412,
-      "eval_accuracy_road": 0.9952440621974141,
-      "eval_accuracy_sidewalk": 0.9021402421852999,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937216828478964,
-      "eval_iou_road": 0.9877474880230636,
-      "eval_iou_sidewalk": 0.844137022397892,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02251315861940384,
-      "eval_mean_accuracy": 0.9647522785614852,
-      "eval_mean_iou": 0.9418687310896173,
-      "eval_overall_accuracy": 0.9938697814941406,
-      "eval_runtime": 0.7112,
-      "eval_samples_per_second": 1.406,
-      "eval_steps_per_second": 1.406,
-      "step": 2060
-    },
-    {
-      "epoch": 515.25,
-      "learning_rate": 7.434210526315789e-06,
-      "loss": 0.0071,
-      "step": 2061
-    },
-    {
-      "epoch": 515.5,
-      "learning_rate": 7.412280701754386e-06,
-      "loss": 0.0069,
-      "step": 2062
-    },
-    {
-      "epoch": 515.75,
-      "learning_rate": 7.390350877192982e-06,
-      "loss": 0.0065,
-      "step": 2063
-    },
-    {
-      "epoch": 516.0,
-      "learning_rate": 7.3684210526315784e-06,
-      "loss": 0.0069,
-      "step": 2064
-    },
-    {
-      "epoch": 516.25,
-      "learning_rate": 7.3464912280701765e-06,
-      "loss": 0.0068,
-      "step": 2065
-    },
-    {
-      "epoch": 516.5,
-      "learning_rate": 7.324561403508773e-06,
-      "loss": 0.0073,
-      "step": 2066
-    },
-    {
-      "epoch": 516.75,
-      "learning_rate": 7.302631578947369e-06,
-      "loss": 0.0072,
-      "step": 2067
-    },
-    {
-      "epoch": 517.0,
-      "learning_rate": 7.280701754385966e-06,
-      "loss": 0.0069,
-      "step": 2068
-    },
-    {
-      "epoch": 517.25,
-      "learning_rate": 7.258771929824562e-06,
-      "loss": 0.0079,
-      "step": 2069
-    },
-    {
-      "epoch": 517.5,
-      "learning_rate": 7.236842105263158e-06,
-      "loss": 0.0065,
-      "step": 2070
-    },
-    {
-      "epoch": 517.75,
-      "learning_rate": 7.214912280701755e-06,
-      "loss": 0.0068,
-      "step": 2071
-    },
-    {
-      "epoch": 518.0,
-      "learning_rate": 7.192982456140351e-06,
-      "loss": 0.0062,
-      "step": 2072
-    },
-    {
-      "epoch": 518.25,
-      "learning_rate": 7.1710526315789475e-06,
-      "loss": 0.0067,
-      "step": 2073
-    },
-    {
-      "epoch": 518.5,
-      "learning_rate": 7.149122807017544e-06,
-      "loss": 0.0072,
-      "step": 2074
-    },
-    {
-      "epoch": 518.75,
-      "learning_rate": 7.12719298245614e-06,
-      "loss": 0.0062,
-      "step": 2075
-    },
-    {
-      "epoch": 519.0,
-      "learning_rate": 7.1052631578947375e-06,
-      "loss": 0.007,
-      "step": 2076
-    },
-    {
-      "epoch": 519.25,
-      "learning_rate": 7.083333333333334e-06,
-      "loss": 0.0064,
-      "step": 2077
-    },
-    {
-      "epoch": 519.5,
-      "learning_rate": 7.06140350877193e-06,
-      "loss": 0.0068,
-      "step": 2078
-    },
-    {
-      "epoch": 519.75,
-      "learning_rate": 7.039473684210527e-06,
-      "loss": 0.0069,
-      "step": 2079
-    },
-    {
-      "epoch": 520.0,
-      "learning_rate": 7.017543859649123e-06,
-      "loss": 0.0089,
-      "step": 2080
-    },
-    {
-      "epoch": 520.0,
-      "eval_accuracy_ELSE": 0.997202592877163,
-      "eval_accuracy_road": 0.9941618727573048,
-      "eval_accuracy_sidewalk": 0.9059419881723458,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9937344642212025,
-      "eval_iou_road": 0.9875526889020906,
-      "eval_iou_sidewalk": 0.8485887628594039,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022204779088497162,
-      "eval_mean_accuracy": 0.9657688179356047,
-      "eval_mean_iou": 0.9432919719942324,
-      "eval_overall_accuracy": 0.9939155578613281,
-      "eval_runtime": 0.6058,
-      "eval_samples_per_second": 1.651,
-      "eval_steps_per_second": 1.651,
-      "step": 2080
-    },
-    {
-      "epoch": 520.25,
-      "learning_rate": 6.995614035087719e-06,
-      "loss": 0.0066,
-      "step": 2081
-    },
-    {
-      "epoch": 520.5,
-      "learning_rate": 6.973684210526316e-06,
-      "loss": 0.0064,
-      "step": 2082
-    },
-    {
-      "epoch": 520.75,
-      "learning_rate": 6.951754385964912e-06,
-      "loss": 0.0073,
-      "step": 2083
-    },
-    {
-      "epoch": 521.0,
-      "learning_rate": 6.9298245614035085e-06,
-      "loss": 0.0069,
-      "step": 2084
-    },
-    {
-      "epoch": 521.25,
-      "learning_rate": 6.9078947368421065e-06,
-      "loss": 0.0074,
-      "step": 2085
-    },
-    {
-      "epoch": 521.5,
-      "learning_rate": 6.885964912280703e-06,
-      "loss": 0.007,
-      "step": 2086
-    },
-    {
-      "epoch": 521.75,
-      "learning_rate": 6.864035087719299e-06,
-      "loss": 0.0066,
-      "step": 2087
-    },
-    {
-      "epoch": 522.0,
-      "learning_rate": 6.842105263157896e-06,
-      "loss": 0.0064,
-      "step": 2088
-    },
-    {
-      "epoch": 522.25,
-      "learning_rate": 6.820175438596492e-06,
-      "loss": 0.0072,
-      "step": 2089
-    },
-    {
-      "epoch": 522.5,
-      "learning_rate": 6.798245614035088e-06,
-      "loss": 0.0067,
-      "step": 2090
-    },
-    {
-      "epoch": 522.75,
-      "learning_rate": 6.776315789473685e-06,
-      "loss": 0.0065,
-      "step": 2091
-    },
-    {
-      "epoch": 523.0,
-      "learning_rate": 6.754385964912281e-06,
-      "loss": 0.0075,
-      "step": 2092
-    },
-    {
-      "epoch": 523.25,
-      "learning_rate": 6.7324561403508775e-06,
-      "loss": 0.0065,
-      "step": 2093
-    },
-    {
-      "epoch": 523.5,
-      "learning_rate": 6.710526315789474e-06,
-      "loss": 0.0066,
-      "step": 2094
-    },
-    {
-      "epoch": 523.75,
-      "learning_rate": 6.68859649122807e-06,
-      "loss": 0.0072,
-      "step": 2095
-    },
-    {
-      "epoch": 524.0,
-      "learning_rate": 6.666666666666667e-06,
-      "loss": 0.0065,
-      "step": 2096
-    },
-    {
-      "epoch": 524.25,
-      "learning_rate": 6.644736842105263e-06,
-      "loss": 0.0067,
-      "step": 2097
-    },
-    {
-      "epoch": 524.5,
-      "learning_rate": 6.622807017543859e-06,
-      "loss": 0.0092,
-      "step": 2098
-    },
-    {
-      "epoch": 524.75,
-      "learning_rate": 6.600877192982456e-06,
-      "loss": 0.0064,
-      "step": 2099
-    },
-    {
-      "epoch": 525.0,
-      "learning_rate": 6.578947368421053e-06,
-      "loss": 0.0066,
-      "step": 2100
-    },
-    {
-      "epoch": 525.0,
-      "eval_accuracy_ELSE": 0.9970186241302066,
-      "eval_accuracy_road": 0.9940052400751836,
-      "eval_accuracy_sidewalk": 0.9074908476485497,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993636858765227,
-      "eval_iou_road": 0.9875367813490267,
-      "eval_iou_sidewalk": 0.8433656110965716,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.022893136367201805,
-      "eval_mean_accuracy": 0.96617157061798,
-      "eval_mean_iou": 0.9415130837369418,
-      "eval_overall_accuracy": 0.9937858581542969,
-      "eval_runtime": 0.5997,
-      "eval_samples_per_second": 1.667,
-      "eval_steps_per_second": 1.667,
-      "step": 2100
-    },
-    {
-      "epoch": 525.25,
-      "learning_rate": 6.557017543859649e-06,
-      "loss": 0.0069,
-      "step": 2101
-    },
-    {
-      "epoch": 525.5,
-      "learning_rate": 6.535087719298246e-06,
-      "loss": 0.0063,
-      "step": 2102
-    },
-    {
-      "epoch": 525.75,
-      "learning_rate": 6.513157894736842e-06,
-      "loss": 0.0076,
-      "step": 2103
-    },
-    {
-      "epoch": 526.0,
-      "learning_rate": 6.4912280701754385e-06,
-      "loss": 0.0063,
-      "step": 2104
-    },
-    {
-      "epoch": 526.25,
-      "learning_rate": 6.469298245614036e-06,
-      "loss": 0.0063,
-      "step": 2105
-    },
-    {
-      "epoch": 526.5,
-      "learning_rate": 6.447368421052632e-06,
-      "loss": 0.0067,
-      "step": 2106
-    },
-    {
-      "epoch": 526.75,
-      "learning_rate": 6.425438596491229e-06,
-      "loss": 0.0071,
-      "step": 2107
-    },
-    {
-      "epoch": 527.0,
-      "learning_rate": 6.403508771929826e-06,
-      "loss": 0.0067,
-      "step": 2108
-    },
-    {
-      "epoch": 527.25,
-      "learning_rate": 6.381578947368422e-06,
-      "loss": 0.0064,
-      "step": 2109
-    },
-    {
-      "epoch": 527.5,
-      "learning_rate": 6.3596491228070184e-06,
-      "loss": 0.0071,
-      "step": 2110
-    },
-    {
-      "epoch": 527.75,
-      "learning_rate": 6.337719298245615e-06,
-      "loss": 0.0066,
-      "step": 2111
-    },
-    {
-      "epoch": 528.0,
-      "learning_rate": 6.315789473684211e-06,
-      "loss": 0.0071,
-      "step": 2112
-    },
-    {
-      "epoch": 528.25,
-      "learning_rate": 6.2938596491228076e-06,
-      "loss": 0.0075,
-      "step": 2113
-    },
-    {
-      "epoch": 528.5,
-      "learning_rate": 6.271929824561404e-06,
-      "loss": 0.0067,
-      "step": 2114
-    },
-    {
-      "epoch": 528.75,
-      "learning_rate": 6.25e-06,
-      "loss": 0.0067,
-      "step": 2115
-    },
-    {
-      "epoch": 529.0,
-      "learning_rate": 6.228070175438597e-06,
-      "loss": 0.0068,
-      "step": 2116
-    },
-    {
-      "epoch": 529.25,
-      "learning_rate": 6.206140350877193e-06,
-      "loss": 0.0072,
-      "step": 2117
-    },
-    {
-      "epoch": 529.5,
-      "learning_rate": 6.184210526315789e-06,
-      "loss": 0.006,
-      "step": 2118
-    },
-    {
-      "epoch": 529.75,
-      "learning_rate": 6.162280701754386e-06,
-      "loss": 0.0069,
-      "step": 2119
-    },
-    {
-      "epoch": 530.0,
-      "learning_rate": 6.140350877192982e-06,
-      "loss": 0.0075,
-      "step": 2120
-    },
-    {
-      "epoch": 530.0,
-      "eval_accuracy_ELSE": 0.9968562987652451,
-      "eval_accuracy_road": 0.9940337187446603,
-      "eval_accuracy_sidewalk": 0.9067868206139116,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993480441323972,
-      "eval_iou_road": 0.9874812572495544,
-      "eval_iou_sidewalk": 0.8396349413298566,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.023344410583376884,
-      "eval_mean_accuracy": 0.9658922793746058,
-      "eval_mean_iou": 0.9401988799677943,
-      "eval_overall_accuracy": 0.9936599731445312,
-      "eval_runtime": 0.7035,
-      "eval_samples_per_second": 1.421,
-      "eval_steps_per_second": 1.421,
-      "step": 2120
-    },
-    {
-      "epoch": 530.25,
-      "learning_rate": 6.118421052631579e-06,
-      "loss": 0.0073,
-      "step": 2121
-    },
-    {
-      "epoch": 530.5,
-      "learning_rate": 6.096491228070176e-06,
-      "loss": 0.0061,
-      "step": 2122
-    },
-    {
-      "epoch": 530.75,
-      "learning_rate": 6.074561403508772e-06,
-      "loss": 0.0064,
-      "step": 2123
-    },
-    {
-      "epoch": 531.0,
-      "learning_rate": 6.0526315789473685e-06,
-      "loss": 0.007,
-      "step": 2124
-    },
-    {
-      "epoch": 531.25,
-      "learning_rate": 6.030701754385965e-06,
-      "loss": 0.006,
-      "step": 2125
-    },
-    {
-      "epoch": 531.5,
-      "learning_rate": 6.008771929824561e-06,
-      "loss": 0.0072,
-      "step": 2126
-    },
-    {
-      "epoch": 531.75,
-      "learning_rate": 5.986842105263158e-06,
-      "loss": 0.008,
-      "step": 2127
-    },
-    {
-      "epoch": 532.0,
-      "learning_rate": 5.964912280701755e-06,
-      "loss": 0.0071,
-      "step": 2128
-    },
-    {
-      "epoch": 532.25,
-      "learning_rate": 5.942982456140351e-06,
-      "loss": 0.0068,
-      "step": 2129
-    },
-    {
-      "epoch": 532.5,
-      "learning_rate": 5.921052631578948e-06,
-      "loss": 0.0067,
-      "step": 2130
-    },
-    {
-      "epoch": 532.75,
-      "learning_rate": 5.899122807017545e-06,
-      "loss": 0.0067,
-      "step": 2131
-    },
-    {
-      "epoch": 533.0,
-      "learning_rate": 5.877192982456141e-06,
-      "loss": 0.0065,
-      "step": 2132
-    },
-    {
-      "epoch": 533.25,
-      "learning_rate": 5.855263157894738e-06,
-      "loss": 0.0061,
-      "step": 2133
-    },
-    {
-      "epoch": 533.5,
-      "learning_rate": 5.833333333333334e-06,
-      "loss": 0.0069,
-      "step": 2134
-    },
-    {
-      "epoch": 533.75,
-      "learning_rate": 5.81140350877193e-06,
-      "loss": 0.0066,
-      "step": 2135
-    },
-    {
-      "epoch": 534.0,
-      "learning_rate": 5.789473684210527e-06,
-      "loss": 0.007,
-      "step": 2136
-    },
-    {
-      "epoch": 534.25,
-      "learning_rate": 5.767543859649123e-06,
-      "loss": 0.0073,
-      "step": 2137
-    },
-    {
-      "epoch": 534.5,
-      "learning_rate": 5.7456140350877194e-06,
-      "loss": 0.0068,
-      "step": 2138
-    },
-    {
-      "epoch": 534.75,
-      "learning_rate": 5.723684210526316e-06,
-      "loss": 0.0055,
-      "step": 2139
-    },
-    {
-      "epoch": 535.0,
-      "learning_rate": 5.701754385964912e-06,
-      "loss": 0.0075,
-      "step": 2140
-    },
-    {
-      "epoch": 535.0,
-      "eval_accuracy_ELSE": 0.9969212289112297,
-      "eval_accuracy_road": 0.9943612234436408,
-      "eval_accuracy_sidewalk": 0.9053787665446353,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.993593373347786,
-      "eval_iou_road": 0.9877088018557022,
-      "eval_iou_sidewalk": 0.8408526219432457,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.023249920457601547,
-      "eval_mean_accuracy": 0.9655537396331685,
-      "eval_mean_iou": 0.9407182657155779,
-      "eval_overall_accuracy": 0.9937553405761719,
-      "eval_runtime": 0.6056,
-      "eval_samples_per_second": 1.651,
-      "eval_steps_per_second": 1.651,
-      "step": 2140
-    },
-    {
-      "epoch": 535.25,
-      "learning_rate": 5.679824561403509e-06,
-      "loss": 0.0074,
-      "step": 2141
-    },
-    {
-      "epoch": 535.5,
-      "learning_rate": 5.657894736842106e-06,
-      "loss": 0.0069,
-      "step": 2142
-    },
-    {
-      "epoch": 535.75,
-      "learning_rate": 5.635964912280702e-06,
-      "loss": 0.0078,
-      "step": 2143
-    },
-    {
-      "epoch": 536.0,
-      "learning_rate": 5.6140350877192985e-06,
-      "loss": 0.0066,
-      "step": 2144
-    },
-    {
-      "epoch": 536.25,
-      "learning_rate": 5.592105263157895e-06,
-      "loss": 0.0065,
-      "step": 2145
-    },
-    {
-      "epoch": 536.5,
-      "learning_rate": 5.570175438596491e-06,
-      "loss": 0.0075,
-      "step": 2146
-    },
-    {
-      "epoch": 536.75,
-      "learning_rate": 5.548245614035088e-06,
-      "loss": 0.0072,
-      "step": 2147
-    },
-    {
-      "epoch": 537.0,
-      "learning_rate": 5.526315789473684e-06,
-      "loss": 0.0065,
-      "step": 2148
-    },
-    {
-      "epoch": 537.25,
-      "learning_rate": 5.50438596491228e-06,
-      "loss": 0.0068,
-      "step": 2149
-    },
-    {
-      "epoch": 537.5,
-      "learning_rate": 5.482456140350877e-06,
-      "loss": 0.007,
-      "step": 2150
-    },
-    {
-      "epoch": 537.75,
-      "learning_rate": 5.460526315789474e-06,
-      "loss": 0.0078,
-      "step": 2151
-    },
-    {
-      "epoch": 538.0,
-      "learning_rate": 5.43859649122807e-06,
-      "loss": 0.006,
-      "step": 2152
-    },
-    {
-      "epoch": 538.25,
-      "learning_rate": 5.416666666666667e-06,
-      "loss": 0.0066,
-      "step": 2153
-    },
-    {
-      "epoch": 538.5,
-      "learning_rate": 5.394736842105263e-06,
-      "loss": 0.0062,
-      "step": 2154
-    },
-    {
-      "epoch": 538.75,
-      "learning_rate": 5.37280701754386e-06,
-      "loss": 0.0071,
-      "step": 2155
-    },
-    {
-      "epoch": 539.0,
-      "learning_rate": 5.350877192982457e-06,
-      "loss": 0.0071,
-      "step": 2156
-    },
-    {
-      "epoch": 539.25,
-      "learning_rate": 5.328947368421053e-06,
-      "loss": 0.0064,
-      "step": 2157
-    },
-    {
-      "epoch": 539.5,
-      "learning_rate": 5.3070175438596495e-06,
-      "loss": 0.0069,
-      "step": 2158
-    },
-    {
-      "epoch": 539.75,
-      "learning_rate": 5.285087719298246e-06,
-      "loss": 0.0065,
-      "step": 2159
-    },
-    {
-      "epoch": 540.0,
-      "learning_rate": 5.263157894736842e-06,
-      "loss": 0.007,
-      "step": 2160
-    },
-    {
-      "epoch": 540.0,
-      "eval_accuracy_ELSE": 0.9968779421472399,
-      "eval_accuracy_road": 0.9945748134647149,
-      "eval_accuracy_sidewalk": 0.9070684314277668,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_ELSE": 0.9935984554318751,
-      "eval_iou_road": 0.9880048093924606,
-      "eval_iou_sidewalk": 0.8428627502289677,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.02291146293282509,
-      "eval_mean_accuracy": 0.9661737290132405,
-      "eval_mean_iou": 0.9414886716844345,
-      "eval_overall_accuracy": 0.9938278198242188,
-      "eval_runtime": 0.6091,
-      "eval_samples_per_second": 1.642,
-      "eval_steps_per_second": 1.642,
-      "step": 2160
-    }
-  ],
-  "max_steps": 2400,
-  "num_train_epochs": 600,
-  "total_flos": 1.3579967201594573e+19,
-  "trial_name": null,
-  "trial_params": null
-}