diff --git "a/checkpoint-5120/trainer_state.json" "b/checkpoint-5120/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-5120/trainer_state.json"
@@ -0,0 +1,40720 @@
+{
+  "best_metric": 0.07582816481590271,
+  "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-1360",
+  "epoch": 301.1764705882353,
+  "global_step": 5120,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.06,
+      "learning_rate": 2.9411764705882356e-07,
+      "loss": 2.7081,
+      "step": 1
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 5.882352941176471e-07,
+      "loss": 2.7218,
+      "step": 2
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 8.823529411764706e-07,
+      "loss": 2.709,
+      "step": 3
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.1764705882352942e-06,
+      "loss": 2.6939,
+      "step": 4
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.4705882352941177e-06,
+      "loss": 2.6979,
+      "step": 5
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 1.7647058823529412e-06,
+      "loss": 2.6788,
+      "step": 6
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.058823529411765e-06,
+      "loss": 2.6769,
+      "step": 7
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.3529411764705885e-06,
+      "loss": 2.6464,
+      "step": 8
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.647058823529412e-06,
+      "loss": 2.6424,
+      "step": 9
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.9411764705882355e-06,
+      "loss": 2.6198,
+      "step": 10
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 3.2352941176470594e-06,
+      "loss": 2.6228,
+      "step": 11
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 3.5294117647058825e-06,
+      "loss": 2.6158,
+      "step": 12
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 3.823529411764706e-06,
+      "loss": 2.632,
+      "step": 13
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 4.11764705882353e-06,
+      "loss": 2.587,
+      "step": 14
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 4.411764705882353e-06,
+      "loss": 2.5767,
+      "step": 15
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 4.705882352941177e-06,
+      "loss": 2.5638,
+      "step": 16
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 5e-06,
+      "loss": 2.517,
+      "step": 17
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 5.294117647058824e-06,
+      "loss": 2.5202,
+      "step": 18
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 5.588235294117647e-06,
+      "loss": 2.5257,
+      "step": 19
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 5.882352941176471e-06,
+      "loss": 2.51,
+      "step": 20
+    },
+    {
+      "epoch": 1.18,
+      "eval_accuracy_NAT": 0.0901846148886272,
+      "eval_accuracy_REST": 0.1772357723577236,
+      "eval_accuracy_concrete": 0.21836230658804573,
+      "eval_accuracy_concrete tiles": 0.10897949563987744,
+      "eval_accuracy_dark bricks": 0.01268591007407691,
+      "eval_accuracy_grass": 0.027127403891645564,
+      "eval_accuracy_red bricks": 0.0003844463971880492,
+      "eval_accuracy_road": 0.0070585682671182835,
+      "eval_accuracy_rough concrete": 0.002699336785465293,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.6741853129093485,
+      "eval_accuracy_steel": 0.0007721641086127534,
+      "eval_accuracy_tiles": 0.2798933624163107,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.09010334674005777,
+      "eval_iou_REST": 2.227093222444523e-05,
+      "eval_iou_concrete": 0.07965033823417267,
+      "eval_iou_concrete tiles": 0.0011714161291131616,
+      "eval_iou_dark bricks": 0.0030587532148061655,
+      "eval_iou_grass": 0.019163923774789638,
+      "eval_iou_red bricks": 5.560869275085081e-06,
+      "eval_iou_road": 0.006794427417222495,
+      "eval_iou_rough concrete": 0.00013145523557474666,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.4632836614619225,
+      "eval_iou_steel": 0.0006867765509148131,
+      "eval_iou_tiles": 0.0032256814378599824,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 2.507078170776367,
+      "eval_mean_accuracy": 0.12304374571723385,
+      "eval_mean_iou": 0.04766411514270953,
+      "eval_overall_accuracy": 0.20164948802883342,
+      "eval_runtime": 18.8176,
+      "eval_samples_per_second": 6.271,
+      "eval_steps_per_second": 0.425,
+      "step": 20
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 6.1764705882352944e-06,
+      "loss": 2.4771,
+      "step": 21
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 6.470588235294119e-06,
+      "loss": 2.4758,
+      "step": 22
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 6.7647058823529414e-06,
+      "loss": 2.4283,
+      "step": 23
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 2.3783,
+      "step": 24
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 7.3529411764705884e-06,
+      "loss": 2.3059,
+      "step": 25
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 7.647058823529413e-06,
+      "loss": 2.3405,
+      "step": 26
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 7.941176470588235e-06,
+      "loss": 2.2917,
+      "step": 27
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 8.23529411764706e-06,
+      "loss": 2.2857,
+      "step": 28
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 8.529411764705883e-06,
+      "loss": 2.1291,
+      "step": 29
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 2.1234,
+      "step": 30
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 9.117647058823529e-06,
+      "loss": 2.0916,
+      "step": 31
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 9.411764705882354e-06,
+      "loss": 2.0543,
+      "step": 32
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 1.9605,
+      "step": 33
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 1e-05,
+      "loss": 1.9835,
+      "step": 34
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 1.0294117647058824e-05,
+      "loss": 2.0005,
+      "step": 35
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 1.8628,
+      "step": 36
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 1.0882352941176471e-05,
+      "loss": 1.8321,
+      "step": 37
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 1.1176470588235295e-05,
+      "loss": 1.7578,
+      "step": 38
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 1.1470588235294118e-05,
+      "loss": 1.7233,
+      "step": 39
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 1.1764705882352942e-05,
+      "loss": 1.5504,
+      "step": 40
+    },
+    {
+      "epoch": 2.35,
+      "eval_accuracy_NAT": 0.760154401121494,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.3386920222776326,
+      "eval_accuracy_concrete tiles": 0.007895357058684892,
+      "eval_accuracy_dark bricks": 0.0014308136975288556,
+      "eval_accuracy_grass": 0.07454476982063306,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.3332036483520477,
+      "eval_accuracy_rough concrete": 0.0009832021404012656,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9879532008612127,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.00875518797903602,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.7577068331202973,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.2037716446695068,
+      "eval_iou_concrete tiles": 0.00033778197232406094,
+      "eval_iou_dark bricks": 0.0009957339958239116,
+      "eval_iou_grass": 0.06824458242953867,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.31270597640653264,
+      "eval_iou_rough concrete": 0.00024414351495637967,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.571054819328365,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0007685393497465681,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.840782642364502,
+      "eval_mean_accuracy": 0.1933548156391286,
+      "eval_mean_iou": 0.13684500391336366,
+      "eval_overall_accuracy": 0.6935860908637612,
+      "eval_runtime": 7.985,
+      "eval_samples_per_second": 14.778,
+      "eval_steps_per_second": 1.002,
+      "step": 40
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 1.2058823529411765e-05,
+      "loss": 1.7505,
+      "step": 41
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 1.2352941176470589e-05,
+      "loss": 1.4808,
+      "step": 42
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 1.2647058823529412e-05,
+      "loss": 1.4041,
+      "step": 43
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 1.2941176470588238e-05,
+      "loss": 1.3261,
+      "step": 44
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 1.3337,
+      "step": 45
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 1.3529411764705883e-05,
+      "loss": 1.3177,
+      "step": 46
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 1.3823529411764708e-05,
+      "loss": 1.2551,
+      "step": 47
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 1.119,
+      "step": 48
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 1.4411764705882352e-05,
+      "loss": 1.1259,
+      "step": 49
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 1.4705882352941177e-05,
+      "loss": 1.0199,
+      "step": 50
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 1.5e-05,
+      "loss": 1.1612,
+      "step": 51
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 1.5294117647058826e-05,
+      "loss": 0.8821,
+      "step": 52
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 1.558823529411765e-05,
+      "loss": 0.8667,
+      "step": 53
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 1.588235294117647e-05,
+      "loss": 0.8607,
+      "step": 54
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 1.6176470588235296e-05,
+      "loss": 0.8021,
+      "step": 55
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 1.647058823529412e-05,
+      "loss": 0.7418,
+      "step": 56
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 1.676470588235294e-05,
+      "loss": 0.7114,
+      "step": 57
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 1.7058823529411767e-05,
+      "loss": 0.7799,
+      "step": 58
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.735294117647059e-05,
+      "loss": 0.6974,
+      "step": 59
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.6271,
+      "step": 60
+    },
+    {
+      "epoch": 3.53,
+      "eval_accuracy_NAT": 0.9551305368117847,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8126940616930818,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.25898348643298624,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.8978859032247204,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9793115987478005,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 3.0294768093550245e-05,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9404775296004152,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.5259192497701041,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.2391512912549761,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.6993177996437074,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8262999758534085,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 2.891343318105592e-05,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.8082596063613892,
+      "eval_mean_accuracy": 0.30031045243680515,
+      "eval_mean_iou": 0.24855344304275326,
+      "eval_overall_accuracy": 0.8893904281874835,
+      "eval_runtime": 6.7648,
+      "eval_samples_per_second": 17.443,
+      "eval_steps_per_second": 1.183,
+      "step": 60
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.7941176470588237e-05,
+      "loss": 0.6901,
+      "step": 61
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.8235294117647057e-05,
+      "loss": 0.5778,
+      "step": 62
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.6319,
+      "step": 63
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8823529411764708e-05,
+      "loss": 0.5626,
+      "step": 64
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.9117647058823528e-05,
+      "loss": 0.7374,
+      "step": 65
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.6702,
+      "step": 66
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.9705882352941178e-05,
+      "loss": 0.5477,
+      "step": 67
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 2e-05,
+      "loss": 0.586,
+      "step": 68
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.505,
+      "step": 69
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 2.058823529411765e-05,
+      "loss": 0.5758,
+      "step": 70
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 2.0882352941176472e-05,
+      "loss": 0.5694,
+      "step": 71
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.4535,
+      "step": 72
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 2.1470588235294116e-05,
+      "loss": 0.5105,
+      "step": 73
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 2.1764705882352943e-05,
+      "loss": 0.5525,
+      "step": 74
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.465,
+      "step": 75
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 2.235294117647059e-05,
+      "loss": 0.5477,
+      "step": 76
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 2.2647058823529413e-05,
+      "loss": 0.3823,
+      "step": 77
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 2.2941176470588237e-05,
+      "loss": 0.4698,
+      "step": 78
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 2.323529411764706e-05,
+      "loss": 0.4636,
+      "step": 79
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 2.3529411764705884e-05,
+      "loss": 0.3963,
+      "step": 80
+    },
+    {
+      "epoch": 4.71,
+      "eval_accuracy_NAT": 0.969774210544504,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.7972872258293342,
+      "eval_accuracy_concrete tiles": 0.04426113598868725,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7826429655535243,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9650914038630805,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9776932262095467,
+      "eval_accuracy_steel": 0.0003548456421456666,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9531585041831687,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.6900984540650801,
+      "eval_iou_concrete tiles": 0.04426113598868725,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.5632131076991799,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.727967248165449,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8641130344685216,
+      "eval_iou_steel": 0.00035449307490288885,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.4204064607620239,
+      "eval_mean_accuracy": 0.34900807797160177,
+      "eval_mean_iou": 0.29562815212653765,
+      "eval_overall_accuracy": 0.9185845003289691,
+      "eval_runtime": 5.4734,
+      "eval_samples_per_second": 21.559,
+      "eval_steps_per_second": 1.462,
+      "step": 80
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 2.3823529411764707e-05,
+      "loss": 0.3853,
+      "step": 81
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 2.411764705882353e-05,
+      "loss": 0.378,
+      "step": 82
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 2.4411764705882354e-05,
+      "loss": 0.4374,
+      "step": 83
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 2.4705882352941178e-05,
+      "loss": 0.3066,
+      "step": 84
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 2.5e-05,
+      "loss": 0.3689,
+      "step": 85
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 2.5294117647058825e-05,
+      "loss": 0.3284,
+      "step": 86
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 2.5588235294117645e-05,
+      "loss": 0.3295,
+      "step": 87
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 2.5882352941176475e-05,
+      "loss": 0.3386,
+      "step": 88
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 2.6176470588235295e-05,
+      "loss": 0.3517,
+      "step": 89
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.3205,
+      "step": 90
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 2.6764705882352942e-05,
+      "loss": 0.4159,
+      "step": 91
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 2.7058823529411766e-05,
+      "loss": 0.3534,
+      "step": 92
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 2.7352941176470593e-05,
+      "loss": 0.325,
+      "step": 93
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 2.7647058823529416e-05,
+      "loss": 0.2642,
+      "step": 94
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 2.7941176470588236e-05,
+      "loss": 0.3485,
+      "step": 95
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.3246,
+      "step": 96
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 2.8529411764705883e-05,
+      "loss": 0.4612,
+      "step": 97
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 2.8823529411764703e-05,
+      "loss": 0.2731,
+      "step": 98
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 2.9117647058823534e-05,
+      "loss": 0.4205,
+      "step": 99
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 2.9411764705882354e-05,
+      "loss": 0.3558,
+      "step": 100
+    },
+    {
+      "epoch": 5.88,
+      "eval_accuracy_NAT": 0.9806404490547624,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9025246556766533,
+      "eval_accuracy_concrete tiles": 0.30502003299552205,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7816418565745797,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.913452910724182,
+      "eval_accuracy_rough concrete": 0.0030807000399239656,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9632305210940226,
+      "eval_accuracy_steel": 0.28737998970447853,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9617550029709626,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.6868428500250288,
+      "eval_iou_concrete tiles": 0.3027439238344757,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6345173453939056,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8305275811769873,
+      "eval_iou_rough concrete": 0.0030807000399239656,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.891543623735271,
+      "eval_iou_steel": 0.2804400149240265,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.30909788608551025,
+      "eval_mean_accuracy": 0.3951516242972403,
+      "eval_mean_iou": 0.35318854170004477,
+      "eval_overall_accuracy": 0.9322030665510792,
+      "eval_runtime": 5.2886,
+      "eval_samples_per_second": 22.312,
+      "eval_steps_per_second": 1.513,
+      "step": 100
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 2.9705882352941177e-05,
+      "loss": 0.4249,
+      "step": 101
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 3e-05,
+      "loss": 0.4043,
+      "step": 102
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 3.0294117647058824e-05,
+      "loss": 0.3018,
+      "step": 103
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 3.058823529411765e-05,
+      "loss": 0.3319,
+      "step": 104
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 3.0882352941176475e-05,
+      "loss": 0.3451,
+      "step": 105
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 3.11764705882353e-05,
+      "loss": 0.3506,
+      "step": 106
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 3.147058823529412e-05,
+      "loss": 0.3709,
+      "step": 107
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 3.176470588235294e-05,
+      "loss": 0.2675,
+      "step": 108
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 3.205882352941177e-05,
+      "loss": 0.3261,
+      "step": 109
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 3.235294117647059e-05,
+      "loss": 0.2528,
+      "step": 110
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 3.2647058823529416e-05,
+      "loss": 0.2736,
+      "step": 111
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 3.294117647058824e-05,
+      "loss": 0.3192,
+      "step": 112
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 3.3235294117647056e-05,
+      "loss": 0.2764,
+      "step": 113
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 3.352941176470588e-05,
+      "loss": 0.2485,
+      "step": 114
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 3.382352941176471e-05,
+      "loss": 0.295,
+      "step": 115
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 3.411764705882353e-05,
+      "loss": 0.2182,
+      "step": 116
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 3.441176470588236e-05,
+      "loss": 0.2945,
+      "step": 117
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 3.470588235294118e-05,
+      "loss": 0.2284,
+      "step": 118
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 3.5e-05,
+      "loss": 0.3391,
+      "step": 119
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 3.529411764705883e-05,
+      "loss": 0.2654,
+      "step": 120
+    },
+    {
+      "epoch": 7.06,
+      "eval_accuracy_NAT": 0.9793312019858504,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8161395883560691,
+      "eval_accuracy_concrete tiles": 0.38750883808625974,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.8487237279860109,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9769372718377128,
+      "eval_accuracy_rough concrete": 0.00018472282637841962,
+      "eval_accuracy_rough red sidewalk": 0.0009475180291624993,
+      "eval_accuracy_speedway bricks": 0.9861728882500396,
+      "eval_accuracy_steel": 0.4149295056650357,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9635845098252436,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7301235451822234,
+      "eval_iou_concrete tiles": 0.30457172495554236,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.649146550195586,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.7846078239913129,
+      "eval_iou_rough concrete": 0.0001839500605255038,
+      "eval_iou_rough red sidewalk": 0.0009475180291624993,
+      "eval_iou_speedway bricks": 0.9013550388480807,
+      "eval_iou_steel": 0.40151082824726514,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.25543102622032166,
+      "eval_mean_accuracy": 0.41622117407865533,
+      "eval_mean_iou": 0.3643101145642263,
+      "eval_overall_accuracy": 0.9363230042538401,
+      "eval_runtime": 5.3581,
+      "eval_samples_per_second": 22.023,
+      "eval_steps_per_second": 1.493,
+      "step": 120
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 3.558823529411765e-05,
+      "loss": 0.3335,
+      "step": 121
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 3.5882352941176474e-05,
+      "loss": 0.2502,
+      "step": 122
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 3.61764705882353e-05,
+      "loss": 0.2615,
+      "step": 123
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 3.6470588235294114e-05,
+      "loss": 0.2804,
+      "step": 124
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 3.6764705882352945e-05,
+      "loss": 0.2703,
+      "step": 125
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 3.705882352941177e-05,
+      "loss": 0.2235,
+      "step": 126
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 3.735294117647059e-05,
+      "loss": 0.2855,
+      "step": 127
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 3.7647058823529415e-05,
+      "loss": 0.2263,
+      "step": 128
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 3.794117647058824e-05,
+      "loss": 0.1922,
+      "step": 129
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 3.8235294117647055e-05,
+      "loss": 0.2025,
+      "step": 130
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 3.8529411764705886e-05,
+      "loss": 0.2237,
+      "step": 131
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 3.882352941176471e-05,
+      "loss": 0.2142,
+      "step": 132
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 3.911764705882353e-05,
+      "loss": 0.2622,
+      "step": 133
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 3.9411764705882356e-05,
+      "loss": 0.2454,
+      "step": 134
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 3.970588235294117e-05,
+      "loss": 0.2398,
+      "step": 135
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 4e-05,
+      "loss": 0.3103,
+      "step": 136
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 4.029411764705883e-05,
+      "loss": 0.2222,
+      "step": 137
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 4.058823529411765e-05,
+      "loss": 0.2128,
+      "step": 138
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 4.0882352941176474e-05,
+      "loss": 0.2087,
+      "step": 139
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 4.11764705882353e-05,
+      "loss": 0.1938,
+      "step": 140
+    },
+    {
+      "epoch": 8.24,
+      "eval_accuracy_NAT": 0.9830271765403628,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9010176438483495,
+      "eval_accuracy_concrete tiles": 0.6649304737214236,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.8719394262831794,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9517197778325257,
+      "eval_accuracy_rough concrete": 0.1663399257533414,
+      "eval_accuracy_rough red sidewalk": 0.03742696215191872,
+      "eval_accuracy_speedway bricks": 0.9778834415332932,
+      "eval_accuracy_steel": 0.644497143742472,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9680687096130417,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7740222344798603,
+      "eval_iou_concrete tiles": 0.18868164278023367,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6641972532280409,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8746486517016726,
+      "eval_iou_rough concrete": 0.16600853979090596,
+      "eval_iou_rough red sidewalk": 0.03742696215191872,
+      "eval_iou_speedway bricks": 0.9354015193799212,
+      "eval_iou_steel": 0.5996131393140197,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.2058570384979248,
+      "eval_mean_accuracy": 0.4768293824159129,
+      "eval_mean_iou": 0.4006206655722781,
+      "eval_overall_accuracy": 0.9478311700336004,
+      "eval_runtime": 5.5104,
+      "eval_samples_per_second": 21.414,
+      "eval_steps_per_second": 1.452,
+      "step": 140
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 4.147058823529412e-05,
+      "loss": 0.2178,
+      "step": 141
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 4.1764705882352944e-05,
+      "loss": 0.2016,
+      "step": 142
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.205882352941177e-05,
+      "loss": 0.291,
+      "step": 143
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.235294117647059e-05,
+      "loss": 0.2516,
+      "step": 144
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.2647058823529415e-05,
+      "loss": 0.2561,
+      "step": 145
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.294117647058823e-05,
+      "loss": 0.2003,
+      "step": 146
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.323529411764706e-05,
+      "loss": 0.2749,
+      "step": 147
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 4.3529411764705885e-05,
+      "loss": 0.2009,
+      "step": 148
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 4.382352941176471e-05,
+      "loss": 0.2451,
+      "step": 149
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 4.411764705882353e-05,
+      "loss": 0.1868,
+      "step": 150
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 4.4411764705882356e-05,
+      "loss": 0.1741,
+      "step": 151
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 4.470588235294118e-05,
+      "loss": 0.1841,
+      "step": 152
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 4.5e-05,
+      "loss": 0.1486,
+      "step": 153
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 4.5294117647058826e-05,
+      "loss": 0.2007,
+      "step": 154
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 4.558823529411765e-05,
+      "loss": 0.1838,
+      "step": 155
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 4.588235294117647e-05,
+      "loss": 0.1689,
+      "step": 156
+    },
+    {
+      "epoch": 9.24,
+      "learning_rate": 4.61764705882353e-05,
+      "loss": 0.1427,
+      "step": 157
+    },
+    {
+      "epoch": 9.29,
+      "learning_rate": 4.647058823529412e-05,
+      "loss": 0.1998,
+      "step": 158
+    },
+    {
+      "epoch": 9.35,
+      "learning_rate": 4.6764705882352944e-05,
+      "loss": 0.1375,
+      "step": 159
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 4.705882352941177e-05,
+      "loss": 0.202,
+      "step": 160
+    },
+    {
+      "epoch": 9.41,
+      "eval_accuracy_NAT": 0.9873472336178203,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8667412464894995,
+      "eval_accuracy_concrete tiles": 0.7498939429648833,
+      "eval_accuracy_dark bricks": 0.028214306222843252,
+      "eval_accuracy_grass": 0.855339752542514,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9538643897689238,
+      "eval_accuracy_rough concrete": 0.5565400818739237,
+      "eval_accuracy_rough red sidewalk": 0.16307838079696793,
+      "eval_accuracy_speedway bricks": 0.9877744670029537,
+      "eval_accuracy_steel": 0.679419452538146,
+      "eval_accuracy_tiles": 0.01720742827713654,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9704972906134676,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7774759671425241,
+      "eval_iou_concrete tiles": 0.20069383120978931,
+      "eval_iou_dark bricks": 0.028214306222843252,
+      "eval_iou_grass": 0.7106618898582445,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9073164433288142,
+      "eval_iou_rough concrete": 0.49950262592121164,
+      "eval_iou_rough red sidewalk": 0.1493012048192771,
+      "eval_iou_speedway bricks": 0.9308492599585683,
+      "eval_iou_steel": 0.6268308483977987,
+      "eval_iou_tiles": 0.01656459609215515,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.17850728332996368,
+      "eval_mean_accuracy": 0.5265708216996624,
+      "eval_mean_iou": 0.4475314048895918,
+      "eval_overall_accuracy": 0.9523853689937268,
+      "eval_runtime": 5.1814,
+      "eval_samples_per_second": 22.774,
+      "eval_steps_per_second": 1.544,
+      "step": 160
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 4.735294117647059e-05,
+      "loss": 0.1802,
+      "step": 161
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 4.7647058823529414e-05,
+      "loss": 0.1469,
+      "step": 162
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 4.794117647058824e-05,
+      "loss": 0.1746,
+      "step": 163
+    },
+    {
+      "epoch": 9.65,
+      "learning_rate": 4.823529411764706e-05,
+      "loss": 0.1482,
+      "step": 164
+    },
+    {
+      "epoch": 9.71,
+      "learning_rate": 4.8529411764705885e-05,
+      "loss": 0.1973,
+      "step": 165
+    },
+    {
+      "epoch": 9.76,
+      "learning_rate": 4.882352941176471e-05,
+      "loss": 0.2324,
+      "step": 166
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 4.911764705882353e-05,
+      "loss": 0.1866,
+      "step": 167
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 4.9411764705882355e-05,
+      "loss": 0.1695,
+      "step": 168
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 4.970588235294118e-05,
+      "loss": 0.1591,
+      "step": 169
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 5e-05,
+      "loss": 0.1731,
+      "step": 170
+    },
+    {
+      "epoch": 10.06,
+      "learning_rate": 4.998452012383901e-05,
+      "loss": 0.2182,
+      "step": 171
+    },
+    {
+      "epoch": 10.12,
+      "learning_rate": 4.996904024767802e-05,
+      "loss": 0.1816,
+      "step": 172
+    },
+    {
+      "epoch": 10.18,
+      "learning_rate": 4.9953560371517026e-05,
+      "loss": 0.1562,
+      "step": 173
+    },
+    {
+      "epoch": 10.24,
+      "learning_rate": 4.9938080495356036e-05,
+      "loss": 0.1389,
+      "step": 174
+    },
+    {
+      "epoch": 10.29,
+      "learning_rate": 4.9922600619195046e-05,
+      "loss": 0.14,
+      "step": 175
+    },
+    {
+      "epoch": 10.35,
+      "learning_rate": 4.9907120743034056e-05,
+      "loss": 0.1198,
+      "step": 176
+    },
+    {
+      "epoch": 10.41,
+      "learning_rate": 4.9891640866873066e-05,
+      "loss": 0.1172,
+      "step": 177
+    },
+    {
+      "epoch": 10.47,
+      "learning_rate": 4.9876160990712076e-05,
+      "loss": 0.1214,
+      "step": 178
+    },
+    {
+      "epoch": 10.53,
+      "learning_rate": 4.9860681114551086e-05,
+      "loss": 0.1756,
+      "step": 179
+    },
+    {
+      "epoch": 10.59,
+      "learning_rate": 4.9845201238390096e-05,
+      "loss": 0.174,
+      "step": 180
+    },
+    {
+      "epoch": 10.59,
+      "eval_accuracy_NAT": 0.9880436657214318,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9067070625843705,
+      "eval_accuracy_concrete tiles": 0.7554560452510016,
+      "eval_accuracy_dark bricks": 0.0684062936661371,
+      "eval_accuracy_grass": 0.8559283062144455,
+      "eval_accuracy_red bricks": 0.08809314586994728,
+      "eval_accuracy_road": 0.9470382925190827,
+      "eval_accuracy_rough concrete": 0.5194048349710104,
+      "eval_accuracy_rough red sidewalk": 0.47680460824641485,
+      "eval_accuracy_speedway bricks": 0.9830751203154764,
+      "eval_accuracy_steel": 0.7461504245653141,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9711355215780687,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7933555654989429,
+      "eval_iou_concrete tiles": 0.21096068920582062,
+      "eval_iou_dark bricks": 0.06803708604228342,
+      "eval_iou_grass": 0.740562538990161,
+      "eval_iou_red bricks": 0.08809314586994728,
+      "eval_iou_road": 0.9141089895430123,
+      "eval_iou_rough concrete": 0.5090669111763916,
+      "eval_iou_rough red sidewalk": 0.38462705037367756,
+      "eval_iou_speedway bricks": 0.9481883072256478,
+      "eval_iou_steel": 0.6882840269421739,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15712417662143707,
+      "eval_mean_accuracy": 0.564239061532664,
+      "eval_mean_iou": 0.4858784486497021,
+      "eval_overall_accuracy": 0.9572272543179787,
+      "eval_runtime": 5.3736,
+      "eval_samples_per_second": 21.959,
+      "eval_steps_per_second": 1.489,
+      "step": 180
+    },
+    {
+      "epoch": 10.65,
+      "learning_rate": 4.9829721362229106e-05,
+      "loss": 0.1558,
+      "step": 181
+    },
+    {
+      "epoch": 10.71,
+      "learning_rate": 4.9814241486068116e-05,
+      "loss": 0.3122,
+      "step": 182
+    },
+    {
+      "epoch": 10.76,
+      "learning_rate": 4.9798761609907126e-05,
+      "loss": 0.1763,
+      "step": 183
+    },
+    {
+      "epoch": 10.82,
+      "learning_rate": 4.9783281733746136e-05,
+      "loss": 0.1451,
+      "step": 184
+    },
+    {
+      "epoch": 10.88,
+      "learning_rate": 4.976780185758514e-05,
+      "loss": 0.1583,
+      "step": 185
+    },
+    {
+      "epoch": 10.94,
+      "learning_rate": 4.975232198142415e-05,
+      "loss": 0.1434,
+      "step": 186
+    },
+    {
+      "epoch": 11.0,
+      "learning_rate": 4.973684210526316e-05,
+      "loss": 0.1714,
+      "step": 187
+    },
+    {
+      "epoch": 11.06,
+      "learning_rate": 4.972136222910217e-05,
+      "loss": 0.1208,
+      "step": 188
+    },
+    {
+      "epoch": 11.12,
+      "learning_rate": 4.970588235294118e-05,
+      "loss": 0.1512,
+      "step": 189
+    },
+    {
+      "epoch": 11.18,
+      "learning_rate": 4.969040247678019e-05,
+      "loss": 0.1312,
+      "step": 190
+    },
+    {
+      "epoch": 11.24,
+      "learning_rate": 4.967492260061919e-05,
+      "loss": 0.1599,
+      "step": 191
+    },
+    {
+      "epoch": 11.29,
+      "learning_rate": 4.96594427244582e-05,
+      "loss": 0.126,
+      "step": 192
+    },
+    {
+      "epoch": 11.35,
+      "learning_rate": 4.964396284829721e-05,
+      "loss": 0.1264,
+      "step": 193
+    },
+    {
+      "epoch": 11.41,
+      "learning_rate": 4.962848297213622e-05,
+      "loss": 0.1599,
+      "step": 194
+    },
+    {
+      "epoch": 11.47,
+      "learning_rate": 4.961300309597524e-05,
+      "loss": 0.1406,
+      "step": 195
+    },
+    {
+      "epoch": 11.53,
+      "learning_rate": 4.959752321981425e-05,
+      "loss": 0.1432,
+      "step": 196
+    },
+    {
+      "epoch": 11.59,
+      "learning_rate": 4.958204334365325e-05,
+      "loss": 0.1213,
+      "step": 197
+    },
+    {
+      "epoch": 11.65,
+      "learning_rate": 4.956656346749226e-05,
+      "loss": 0.1459,
+      "step": 198
+    },
+    {
+      "epoch": 11.71,
+      "learning_rate": 4.955108359133127e-05,
+      "loss": 0.1344,
+      "step": 199
+    },
+    {
+      "epoch": 11.76,
+      "learning_rate": 4.953560371517028e-05,
+      "loss": 0.1317,
+      "step": 200
+    },
+    {
+      "epoch": 11.76,
+      "eval_accuracy_NAT": 0.9864662076511677,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9249773355816844,
+      "eval_accuracy_concrete tiles": 0.7600047136460052,
+      "eval_accuracy_dark bricks": 0.7836265145569742,
+      "eval_accuracy_grass": 0.894897750627349,
+      "eval_accuracy_red bricks": 0.10456942003514938,
+      "eval_accuracy_road": 0.9491855765361391,
+      "eval_accuracy_rough concrete": 0.5790703078912399,
+      "eval_accuracy_rough red sidewalk": 0.5756548026380105,
+      "eval_accuracy_speedway bricks": 0.9715143127606035,
+      "eval_accuracy_steel": 0.7851359658548531,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9719829052737282,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8087278598858759,
+      "eval_iou_concrete tiles": 0.6147203477067368,
+      "eval_iou_dark bricks": 0.7371681176904863,
+      "eval_iou_grass": 0.7440049088827965,
+      "eval_iou_red bricks": 0.10435735818032338,
+      "eval_iou_road": 0.9256206487360326,
+      "eval_iou_rough concrete": 0.5756129055186671,
+      "eval_iou_rough red sidewalk": 0.4396544792492275,
+      "eval_iou_speedway bricks": 0.9522459713674365,
+      "eval_iou_steel": 0.7135061360560284,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13907073438167572,
+      "eval_mean_accuracy": 0.6396233005983981,
+      "eval_mean_iou": 0.5836616645036414,
+      "eval_overall_accuracy": 0.9632077621201337,
+      "eval_runtime": 5.8073,
+      "eval_samples_per_second": 20.319,
+      "eval_steps_per_second": 1.378,
+      "step": 200
+    },
+    {
+      "epoch": 11.82,
+      "learning_rate": 4.952012383900929e-05,
+      "loss": 0.1202,
+      "step": 201
+    },
+    {
+      "epoch": 11.88,
+      "learning_rate": 4.95046439628483e-05,
+      "loss": 0.1488,
+      "step": 202
+    },
+    {
+      "epoch": 11.94,
+      "learning_rate": 4.9489164086687305e-05,
+      "loss": 0.1498,
+      "step": 203
+    },
+    {
+      "epoch": 12.0,
+      "learning_rate": 4.9473684210526315e-05,
+      "loss": 0.1198,
+      "step": 204
+    },
+    {
+      "epoch": 12.06,
+      "learning_rate": 4.9458204334365325e-05,
+      "loss": 0.1387,
+      "step": 205
+    },
+    {
+      "epoch": 12.12,
+      "learning_rate": 4.9442724458204335e-05,
+      "loss": 0.1239,
+      "step": 206
+    },
+    {
+      "epoch": 12.18,
+      "learning_rate": 4.9427244582043345e-05,
+      "loss": 0.1445,
+      "step": 207
+    },
+    {
+      "epoch": 12.24,
+      "learning_rate": 4.9411764705882355e-05,
+      "loss": 0.1116,
+      "step": 208
+    },
+    {
+      "epoch": 12.29,
+      "learning_rate": 4.9396284829721365e-05,
+      "loss": 0.1086,
+      "step": 209
+    },
+    {
+      "epoch": 12.35,
+      "learning_rate": 4.9380804953560375e-05,
+      "loss": 0.1052,
+      "step": 210
+    },
+    {
+      "epoch": 12.41,
+      "learning_rate": 4.9365325077399385e-05,
+      "loss": 0.185,
+      "step": 211
+    },
+    {
+      "epoch": 12.47,
+      "learning_rate": 4.9349845201238395e-05,
+      "loss": 0.0983,
+      "step": 212
+    },
+    {
+      "epoch": 12.53,
+      "learning_rate": 4.9334365325077405e-05,
+      "loss": 0.1275,
+      "step": 213
+    },
+    {
+      "epoch": 12.59,
+      "learning_rate": 4.9318885448916415e-05,
+      "loss": 0.1293,
+      "step": 214
+    },
+    {
+      "epoch": 12.65,
+      "learning_rate": 4.930340557275542e-05,
+      "loss": 0.1142,
+      "step": 215
+    },
+    {
+      "epoch": 12.71,
+      "learning_rate": 4.928792569659443e-05,
+      "loss": 0.1127,
+      "step": 216
+    },
+    {
+      "epoch": 12.76,
+      "learning_rate": 4.927244582043344e-05,
+      "loss": 0.1655,
+      "step": 217
+    },
+    {
+      "epoch": 12.82,
+      "learning_rate": 4.925696594427245e-05,
+      "loss": 0.1393,
+      "step": 218
+    },
+    {
+      "epoch": 12.88,
+      "learning_rate": 4.924148606811146e-05,
+      "loss": 0.1265,
+      "step": 219
+    },
+    {
+      "epoch": 12.94,
+      "learning_rate": 4.922600619195047e-05,
+      "loss": 0.0915,
+      "step": 220
+    },
+    {
+      "epoch": 12.94,
+      "eval_accuracy_NAT": 0.9860258024913594,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9262420330556528,
+      "eval_accuracy_concrete tiles": 0.7638934716002829,
+      "eval_accuracy_dark bricks": 0.7600061252225179,
+      "eval_accuracy_grass": 0.9003149046013164,
+      "eval_accuracy_red bricks": 0.5878185413005272,
+      "eval_accuracy_road": 0.9337452256598837,
+      "eval_accuracy_rough concrete": 0.6387834512182768,
+      "eval_accuracy_rough red sidewalk": 0.5331292910910745,
+      "eval_accuracy_speedway bricks": 0.9804688276498366,
+      "eval_accuracy_steel": 0.7936647558312134,
+      "eval_accuracy_tiles": 0.011542306643642643,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9732919784830467,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.811574716554503,
+      "eval_iou_concrete tiles": 0.6126800499035954,
+      "eval_iou_dark bricks": 0.7370453216509963,
+      "eval_iou_grass": 0.7637841543433552,
+      "eval_iou_red bricks": 0.5786969451203028,
+      "eval_iou_road": 0.90648387138614,
+      "eval_iou_rough concrete": 0.56731583403895,
+      "eval_iou_rough red sidewalk": 0.4643523825118716,
+      "eval_iou_speedway bricks": 0.956141941385667,
+      "eval_iou_steel": 0.7260183788232067,
+      "eval_iou_tiles": 0.011447629349197765,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12515389919281006,
+      "eval_mean_accuracy": 0.6781257489511988,
+      "eval_mean_iou": 0.623756400273141,
+      "eval_overall_accuracy": 0.9643849841618942,
+      "eval_runtime": 5.8731,
+      "eval_samples_per_second": 20.092,
+      "eval_steps_per_second": 1.362,
+      "step": 220
+    },
+    {
+      "epoch": 13.0,
+      "learning_rate": 4.921052631578947e-05,
+      "loss": 0.1054,
+      "step": 221
+    },
+    {
+      "epoch": 13.06,
+      "learning_rate": 4.919504643962848e-05,
+      "loss": 0.1411,
+      "step": 222
+    },
+    {
+      "epoch": 13.12,
+      "learning_rate": 4.917956656346749e-05,
+      "loss": 0.1017,
+      "step": 223
+    },
+    {
+      "epoch": 13.18,
+      "learning_rate": 4.91640866873065e-05,
+      "loss": 0.0983,
+      "step": 224
+    },
+    {
+      "epoch": 13.24,
+      "learning_rate": 4.914860681114551e-05,
+      "loss": 0.1682,
+      "step": 225
+    },
+    {
+      "epoch": 13.29,
+      "learning_rate": 4.913312693498452e-05,
+      "loss": 0.1545,
+      "step": 226
+    },
+    {
+      "epoch": 13.35,
+      "learning_rate": 4.911764705882353e-05,
+      "loss": 0.1017,
+      "step": 227
+    },
+    {
+      "epoch": 13.41,
+      "learning_rate": 4.910216718266254e-05,
+      "loss": 0.1027,
+      "step": 228
+    },
+    {
+      "epoch": 13.47,
+      "learning_rate": 4.908668730650155e-05,
+      "loss": 0.1169,
+      "step": 229
+    },
+    {
+      "epoch": 13.53,
+      "learning_rate": 4.907120743034056e-05,
+      "loss": 0.0948,
+      "step": 230
+    },
+    {
+      "epoch": 13.59,
+      "learning_rate": 4.905572755417957e-05,
+      "loss": 0.1222,
+      "step": 231
+    },
+    {
+      "epoch": 13.65,
+      "learning_rate": 4.904024767801858e-05,
+      "loss": 0.1146,
+      "step": 232
+    },
+    {
+      "epoch": 13.71,
+      "learning_rate": 4.9024767801857585e-05,
+      "loss": 0.1355,
+      "step": 233
+    },
+    {
+      "epoch": 13.76,
+      "learning_rate": 4.9009287925696595e-05,
+      "loss": 0.0935,
+      "step": 234
+    },
+    {
+      "epoch": 13.82,
+      "learning_rate": 4.8993808049535605e-05,
+      "loss": 0.1122,
+      "step": 235
+    },
+    {
+      "epoch": 13.88,
+      "learning_rate": 4.8978328173374615e-05,
+      "loss": 0.0929,
+      "step": 236
+    },
+    {
+      "epoch": 13.94,
+      "learning_rate": 4.8962848297213625e-05,
+      "loss": 0.083,
+      "step": 237
+    },
+    {
+      "epoch": 14.0,
+      "learning_rate": 4.8947368421052635e-05,
+      "loss": 0.1423,
+      "step": 238
+    },
+    {
+      "epoch": 14.06,
+      "learning_rate": 4.893188854489164e-05,
+      "loss": 0.1276,
+      "step": 239
+    },
+    {
+      "epoch": 14.12,
+      "learning_rate": 4.891640866873065e-05,
+      "loss": 0.1027,
+      "step": 240
+    },
+    {
+      "epoch": 14.12,
+      "eval_accuracy_NAT": 0.9917472418339723,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9190345187117132,
+      "eval_accuracy_concrete tiles": 0.7522743341975018,
+      "eval_accuracy_dark bricks": 0.7271979021112877,
+      "eval_accuracy_grass": 0.8350431669107934,
+      "eval_accuracy_red bricks": 0.6341168717047452,
+      "eval_accuracy_road": 0.9736051872567405,
+      "eval_accuracy_rough concrete": 0.6122906226351009,
+      "eval_accuracy_rough red sidewalk": 0.5560201234781431,
+      "eval_accuracy_speedway bricks": 0.9814349844023434,
+      "eval_accuracy_steel": 0.7841863789251675,
+      "eval_accuracy_tiles": 0.058953618710048776,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9740554919478627,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8296911062490576,
+      "eval_iou_concrete tiles": 0.6982172153560101,
+      "eval_iou_dark bricks": 0.7163416439221454,
+      "eval_iou_grass": 0.7689727795781048,
+      "eval_iou_red bricks": 0.6251218191662155,
+      "eval_iou_road": 0.9231815060770597,
+      "eval_iou_rough concrete": 0.594851191682249,
+      "eval_iou_rough red sidewalk": 0.47678587549491225,
+      "eval_iou_speedway bricks": 0.9582911258333994,
+      "eval_iou_steel": 0.7348050774234488,
+      "eval_iou_tiles": 0.058793317018641045,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11748708039522171,
+      "eval_mean_accuracy": 0.6789157654521198,
+      "eval_mean_iou": 0.6430083192114698,
+      "eval_overall_accuracy": 0.9672419661182469,
+      "eval_runtime": 5.5775,
+      "eval_samples_per_second": 21.156,
+      "eval_steps_per_second": 1.434,
+      "step": 240
+    },
+    {
+      "epoch": 14.18,
+      "learning_rate": 4.890092879256966e-05,
+      "loss": 0.1013,
+      "step": 241
+    },
+    {
+      "epoch": 14.24,
+      "learning_rate": 4.888544891640867e-05,
+      "loss": 0.0807,
+      "step": 242
+    },
+    {
+      "epoch": 14.29,
+      "learning_rate": 4.886996904024768e-05,
+      "loss": 0.0938,
+      "step": 243
+    },
+    {
+      "epoch": 14.35,
+      "learning_rate": 4.8854489164086695e-05,
+      "loss": 0.1197,
+      "step": 244
+    },
+    {
+      "epoch": 14.41,
+      "learning_rate": 4.88390092879257e-05,
+      "loss": 0.1158,
+      "step": 245
+    },
+    {
+      "epoch": 14.47,
+      "learning_rate": 4.882352941176471e-05,
+      "loss": 0.121,
+      "step": 246
+    },
+    {
+      "epoch": 14.53,
+      "learning_rate": 4.880804953560372e-05,
+      "loss": 0.1065,
+      "step": 247
+    },
+    {
+      "epoch": 14.59,
+      "learning_rate": 4.879256965944273e-05,
+      "loss": 0.1111,
+      "step": 248
+    },
+    {
+      "epoch": 14.65,
+      "learning_rate": 4.877708978328174e-05,
+      "loss": 0.0954,
+      "step": 249
+    },
+    {
+      "epoch": 14.71,
+      "learning_rate": 4.876160990712075e-05,
+      "loss": 0.0999,
+      "step": 250
+    },
+    {
+      "epoch": 14.76,
+      "learning_rate": 4.874613003095975e-05,
+      "loss": 0.1146,
+      "step": 251
+    },
+    {
+      "epoch": 14.82,
+      "learning_rate": 4.873065015479876e-05,
+      "loss": 0.1087,
+      "step": 252
+    },
+    {
+      "epoch": 14.88,
+      "learning_rate": 4.871517027863777e-05,
+      "loss": 0.1084,
+      "step": 253
+    },
+    {
+      "epoch": 14.94,
+      "learning_rate": 4.869969040247678e-05,
+      "loss": 0.0894,
+      "step": 254
+    },
+    {
+      "epoch": 15.0,
+      "learning_rate": 4.868421052631579e-05,
+      "loss": 0.1068,
+      "step": 255
+    },
+    {
+      "epoch": 15.06,
+      "learning_rate": 4.86687306501548e-05,
+      "loss": 0.0922,
+      "step": 256
+    },
+    {
+      "epoch": 15.12,
+      "learning_rate": 4.8653250773993805e-05,
+      "loss": 0.0908,
+      "step": 257
+    },
+    {
+      "epoch": 15.18,
+      "learning_rate": 4.863777089783282e-05,
+      "loss": 0.1101,
+      "step": 258
+    },
+    {
+      "epoch": 15.24,
+      "learning_rate": 4.862229102167183e-05,
+      "loss": 0.0947,
+      "step": 259
+    },
+    {
+      "epoch": 15.29,
+      "learning_rate": 4.860681114551084e-05,
+      "loss": 0.1055,
+      "step": 260
+    },
+    {
+      "epoch": 15.29,
+      "eval_accuracy_NAT": 0.9880657156308897,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9169668205146237,
+      "eval_accuracy_concrete tiles": 0.6840678765024747,
+      "eval_accuracy_dark bricks": 0.7566324675076087,
+      "eval_accuracy_grass": 0.8498658249024439,
+      "eval_accuracy_red bricks": 0.6606985940246046,
+      "eval_accuracy_road": 0.9807342984532393,
+      "eval_accuracy_rough concrete": 0.6125766450759449,
+      "eval_accuracy_rough red sidewalk": 0.6318290857955016,
+      "eval_accuracy_speedway bricks": 0.9880714428192173,
+      "eval_accuracy_steel": 0.7434690909454388,
+      "eval_accuracy_tiles": 0.030385652397830896,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9746038269106396,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8274392746410625,
+      "eval_iou_concrete tiles": 0.666950067786484,
+      "eval_iou_dark bricks": 0.738972547040184,
+      "eval_iou_grass": 0.7776974436725635,
+      "eval_iou_red bricks": 0.6507979442791453,
+      "eval_iou_road": 0.9132698891078013,
+      "eval_iou_rough concrete": 0.5830686510277235,
+      "eval_iou_rough red sidewalk": 0.5123983070486785,
+      "eval_iou_speedway bricks": 0.9579088057628538,
+      "eval_iou_steel": 0.7130391226405146,
+      "eval_iou_tiles": 0.030056937368894216,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11388663947582245,
+      "eval_mean_accuracy": 0.6802587318899861,
+      "eval_mean_iou": 0.6420156013297343,
+      "eval_overall_accuracy": 0.9669402494268903,
+      "eval_runtime": 6.1549,
+      "eval_samples_per_second": 19.172,
+      "eval_steps_per_second": 1.3,
+      "step": 260
+    },
+    {
+      "epoch": 15.35,
+      "learning_rate": 4.859133126934985e-05,
+      "loss": 0.0871,
+      "step": 261
+    },
+    {
+      "epoch": 15.41,
+      "learning_rate": 4.857585139318886e-05,
+      "loss": 0.1223,
+      "step": 262
+    },
+    {
+      "epoch": 15.47,
+      "learning_rate": 4.8560371517027865e-05,
+      "loss": 0.0979,
+      "step": 263
+    },
+    {
+      "epoch": 15.53,
+      "learning_rate": 4.8544891640866875e-05,
+      "loss": 0.0874,
+      "step": 264
+    },
+    {
+      "epoch": 15.59,
+      "learning_rate": 4.8529411764705885e-05,
+      "loss": 0.0922,
+      "step": 265
+    },
+    {
+      "epoch": 15.65,
+      "learning_rate": 4.8513931888544895e-05,
+      "loss": 0.0995,
+      "step": 266
+    },
+    {
+      "epoch": 15.71,
+      "learning_rate": 4.8498452012383905e-05,
+      "loss": 0.0835,
+      "step": 267
+    },
+    {
+      "epoch": 15.76,
+      "learning_rate": 4.8482972136222915e-05,
+      "loss": 0.0942,
+      "step": 268
+    },
+    {
+      "epoch": 15.82,
+      "learning_rate": 4.846749226006192e-05,
+      "loss": 0.0956,
+      "step": 269
+    },
+    {
+      "epoch": 15.88,
+      "learning_rate": 4.845201238390093e-05,
+      "loss": 0.0978,
+      "step": 270
+    },
+    {
+      "epoch": 15.94,
+      "learning_rate": 4.843653250773994e-05,
+      "loss": 0.0927,
+      "step": 271
+    },
+    {
+      "epoch": 16.0,
+      "learning_rate": 4.842105263157895e-05,
+      "loss": 0.0825,
+      "step": 272
+    },
+    {
+      "epoch": 16.06,
+      "learning_rate": 4.840557275541796e-05,
+      "loss": 0.0847,
+      "step": 273
+    },
+    {
+      "epoch": 16.12,
+      "learning_rate": 4.839009287925697e-05,
+      "loss": 0.0886,
+      "step": 274
+    },
+    {
+      "epoch": 16.18,
+      "learning_rate": 4.837461300309598e-05,
+      "loss": 0.0814,
+      "step": 275
+    },
+    {
+      "epoch": 16.24,
+      "learning_rate": 4.835913312693499e-05,
+      "loss": 0.0678,
+      "step": 276
+    },
+    {
+      "epoch": 16.29,
+      "learning_rate": 4.8343653250774e-05,
+      "loss": 0.0912,
+      "step": 277
+    },
+    {
+      "epoch": 16.35,
+      "learning_rate": 4.832817337461301e-05,
+      "loss": 0.1018,
+      "step": 278
+    },
+    {
+      "epoch": 16.41,
+      "learning_rate": 4.831269349845202e-05,
+      "loss": 0.0735,
+      "step": 279
+    },
+    {
+      "epoch": 16.47,
+      "learning_rate": 4.829721362229103e-05,
+      "loss": 0.089,
+      "step": 280
+    },
+    {
+      "epoch": 16.47,
+      "eval_accuracy_NAT": 0.9882539215817883,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9266035935906707,
+      "eval_accuracy_concrete tiles": 0.7337732736271506,
+      "eval_accuracy_dark bricks": 0.7679306318549853,
+      "eval_accuracy_grass": 0.8684279750347266,
+      "eval_accuracy_red bricks": 0.7258897188049209,
+      "eval_accuracy_road": 0.9741289150657786,
+      "eval_accuracy_rough concrete": 0.6240711719173634,
+      "eval_accuracy_rough red sidewalk": 0.7053068529617459,
+      "eval_accuracy_speedway bricks": 0.9871671488882171,
+      "eval_accuracy_steel": 0.7572206090350697,
+      "eval_accuracy_tiles": 0.06749674331242994,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9757167219517451,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8338523416250841,
+      "eval_iou_concrete tiles": 0.6868753722946588,
+      "eval_iou_dark bricks": 0.7545810625902224,
+      "eval_iou_grass": 0.7875470668392022,
+      "eval_iou_red bricks": 0.7145483051305617,
+      "eval_iou_road": 0.936732714837276,
+      "eval_iou_rough concrete": 0.5637154375709819,
+      "eval_iou_rough red sidewalk": 0.5270045513288757,
+      "eval_iou_speedway bricks": 0.9597518102996859,
+      "eval_iou_steel": 0.724264609221694,
+      "eval_iou_tiles": 0.06645588498478792,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10488714277744293,
+      "eval_mean_accuracy": 0.7020208119749882,
+      "eval_mean_iou": 0.656234298359598,
+      "eval_overall_accuracy": 0.968710850861113,
+      "eval_runtime": 5.5741,
+      "eval_samples_per_second": 21.169,
+      "eval_steps_per_second": 1.435,
+      "step": 280
+    },
+    {
+      "epoch": 16.53,
+      "learning_rate": 4.828173374613003e-05,
+      "loss": 0.1157,
+      "step": 281
+    },
+    {
+      "epoch": 16.59,
+      "learning_rate": 4.826625386996904e-05,
+      "loss": 0.0849,
+      "step": 282
+    },
+    {
+      "epoch": 16.65,
+      "learning_rate": 4.825077399380805e-05,
+      "loss": 0.0923,
+      "step": 283
+    },
+    {
+      "epoch": 16.71,
+      "learning_rate": 4.823529411764706e-05,
+      "loss": 0.0892,
+      "step": 284
+    },
+    {
+      "epoch": 16.76,
+      "learning_rate": 4.821981424148607e-05,
+      "loss": 0.0897,
+      "step": 285
+    },
+    {
+      "epoch": 16.82,
+      "learning_rate": 4.820433436532508e-05,
+      "loss": 0.0713,
+      "step": 286
+    },
+    {
+      "epoch": 16.88,
+      "learning_rate": 4.8188854489164085e-05,
+      "loss": 0.0838,
+      "step": 287
+    },
+    {
+      "epoch": 16.94,
+      "learning_rate": 4.8173374613003095e-05,
+      "loss": 0.108,
+      "step": 288
+    },
+    {
+      "epoch": 17.0,
+      "learning_rate": 4.8157894736842105e-05,
+      "loss": 0.0924,
+      "step": 289
+    },
+    {
+      "epoch": 17.06,
+      "learning_rate": 4.8142414860681115e-05,
+      "loss": 0.0913,
+      "step": 290
+    },
+    {
+      "epoch": 17.12,
+      "learning_rate": 4.8126934984520125e-05,
+      "loss": 0.0698,
+      "step": 291
+    },
+    {
+      "epoch": 17.18,
+      "learning_rate": 4.811145510835914e-05,
+      "loss": 0.1019,
+      "step": 292
+    },
+    {
+      "epoch": 17.24,
+      "learning_rate": 4.8095975232198145e-05,
+      "loss": 0.074,
+      "step": 293
+    },
+    {
+      "epoch": 17.29,
+      "learning_rate": 4.8080495356037155e-05,
+      "loss": 0.0724,
+      "step": 294
+    },
+    {
+      "epoch": 17.35,
+      "learning_rate": 4.8065015479876165e-05,
+      "loss": 0.0794,
+      "step": 295
+    },
+    {
+      "epoch": 17.41,
+      "learning_rate": 4.8049535603715175e-05,
+      "loss": 0.067,
+      "step": 296
+    },
+    {
+      "epoch": 17.47,
+      "learning_rate": 4.8034055727554185e-05,
+      "loss": 0.0773,
+      "step": 297
+    },
+    {
+      "epoch": 17.53,
+      "learning_rate": 4.8018575851393195e-05,
+      "loss": 0.103,
+      "step": 298
+    },
+    {
+      "epoch": 17.59,
+      "learning_rate": 4.80030959752322e-05,
+      "loss": 0.0926,
+      "step": 299
+    },
+    {
+      "epoch": 17.65,
+      "learning_rate": 4.798761609907121e-05,
+      "loss": 0.0982,
+      "step": 300
+    },
+    {
+      "epoch": 17.65,
+      "eval_accuracy_NAT": 0.9931665767148967,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9314965089916206,
+      "eval_accuracy_concrete tiles": 0.7459816167805797,
+      "eval_accuracy_dark bricks": 0.8016145703730644,
+      "eval_accuracy_grass": 0.8160921171657072,
+      "eval_accuracy_red bricks": 0.8456722319859402,
+      "eval_accuracy_road": 0.9701795798527256,
+      "eval_accuracy_rough concrete": 0.6210738950893522,
+      "eval_accuracy_rough red sidewalk": 0.6152550402695163,
+      "eval_accuracy_speedway bricks": 0.9799415370631988,
+      "eval_accuracy_steel": 0.8352466676995507,
+      "eval_accuracy_tiles": 0.09173255778727014,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9752539993478923,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8405579023170859,
+      "eval_iou_concrete tiles": 0.7205427062465853,
+      "eval_iou_dark bricks": 0.7828242705198423,
+      "eval_iou_grass": 0.7712687805109247,
+      "eval_iou_red bricks": 0.8008946218662228,
+      "eval_iou_road": 0.9368960548615971,
+      "eval_iou_rough concrete": 0.5511559532119212,
+      "eval_iou_rough red sidewalk": 0.5013173858163503,
+      "eval_iou_speedway bricks": 0.9640669157177488,
+      "eval_iou_steel": 0.7744820666819596,
+      "eval_iou_tiles": 0.08814881662833687,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10236875712871552,
+      "eval_mean_accuracy": 0.7113425307518016,
+      "eval_mean_iou": 0.6698007287481899,
+      "eval_overall_accuracy": 0.9696271864034361,
+      "eval_runtime": 5.9798,
+      "eval_samples_per_second": 19.733,
+      "eval_steps_per_second": 1.338,
+      "step": 300
+    },
+    {
+      "epoch": 17.71,
+      "learning_rate": 4.797213622291022e-05,
+      "loss": 0.0788,
+      "step": 301
+    },
+    {
+      "epoch": 17.76,
+      "learning_rate": 4.795665634674923e-05,
+      "loss": 0.0778,
+      "step": 302
+    },
+    {
+      "epoch": 17.82,
+      "learning_rate": 4.794117647058824e-05,
+      "loss": 0.078,
+      "step": 303
+    },
+    {
+      "epoch": 17.88,
+      "learning_rate": 4.792569659442725e-05,
+      "loss": 0.0882,
+      "step": 304
+    },
+    {
+      "epoch": 17.94,
+      "learning_rate": 4.791021671826625e-05,
+      "loss": 0.0745,
+      "step": 305
+    },
+    {
+      "epoch": 18.0,
+      "learning_rate": 4.789473684210526e-05,
+      "loss": 0.0797,
+      "step": 306
+    },
+    {
+      "epoch": 18.06,
+      "learning_rate": 4.787925696594428e-05,
+      "loss": 0.0779,
+      "step": 307
+    },
+    {
+      "epoch": 18.12,
+      "learning_rate": 4.786377708978329e-05,
+      "loss": 0.0661,
+      "step": 308
+    },
+    {
+      "epoch": 18.18,
+      "learning_rate": 4.78482972136223e-05,
+      "loss": 0.0659,
+      "step": 309
+    },
+    {
+      "epoch": 18.24,
+      "learning_rate": 4.783281733746131e-05,
+      "loss": 0.0739,
+      "step": 310
+    },
+    {
+      "epoch": 18.29,
+      "learning_rate": 4.781733746130031e-05,
+      "loss": 0.0877,
+      "step": 311
+    },
+    {
+      "epoch": 18.35,
+      "learning_rate": 4.780185758513932e-05,
+      "loss": 0.0735,
+      "step": 312
+    },
+    {
+      "epoch": 18.41,
+      "learning_rate": 4.778637770897833e-05,
+      "loss": 0.0671,
+      "step": 313
+    },
+    {
+      "epoch": 18.47,
+      "learning_rate": 4.777089783281734e-05,
+      "loss": 0.0732,
+      "step": 314
+    },
+    {
+      "epoch": 18.53,
+      "learning_rate": 4.775541795665635e-05,
+      "loss": 0.0977,
+      "step": 315
+    },
+    {
+      "epoch": 18.59,
+      "learning_rate": 4.773993808049536e-05,
+      "loss": 0.0752,
+      "step": 316
+    },
+    {
+      "epoch": 18.65,
+      "learning_rate": 4.7724458204334364e-05,
+      "loss": 0.0708,
+      "step": 317
+    },
+    {
+      "epoch": 18.71,
+      "learning_rate": 4.7708978328173374e-05,
+      "loss": 0.0807,
+      "step": 318
+    },
+    {
+      "epoch": 18.76,
+      "learning_rate": 4.7693498452012384e-05,
+      "loss": 0.0666,
+      "step": 319
+    },
+    {
+      "epoch": 18.82,
+      "learning_rate": 4.7678018575851394e-05,
+      "loss": 0.0787,
+      "step": 320
+    },
+    {
+      "epoch": 18.82,
+      "eval_accuracy_NAT": 0.9902132366415152,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9379877818528716,
+      "eval_accuracy_concrete tiles": 0.7392411029931653,
+      "eval_accuracy_dark bricks": 0.8253450222996382,
+      "eval_accuracy_grass": 0.8625017505213715,
+      "eval_accuracy_red bricks": 0.8736269771528998,
+      "eval_accuracy_road": 0.9765295123292473,
+      "eval_accuracy_rough concrete": 0.6109022220368373,
+      "eval_accuracy_rough red sidewalk": 0.7007873423623279,
+      "eval_accuracy_speedway bricks": 0.9772675551607303,
+      "eval_accuracy_steel": 0.8889108237916507,
+      "eval_accuracy_tiles": 0.11981580780999121,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9772417072398732,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8491310902138964,
+      "eval_iou_concrete tiles": 0.6966970969103308,
+      "eval_iou_dark bricks": 0.802353926311872,
+      "eval_iou_grass": 0.7964497227547327,
+      "eval_iou_red bricks": 0.8242395979066273,
+      "eval_iou_road": 0.9342291068271499,
+      "eval_iou_rough concrete": 0.5883997084430976,
+      "eval_iou_rough red sidewalk": 0.5513809670319267,
+      "eval_iou_speedway bricks": 0.9625142068997178,
+      "eval_iou_steel": 0.773288725413471,
+      "eval_iou_tiles": 0.11640912435614423,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10046244412660599,
+      "eval_mean_accuracy": 0.7310099334578652,
+      "eval_mean_iou": 0.682487306177603,
+      "eval_overall_accuracy": 0.971074799359855,
+      "eval_runtime": 6.0333,
+      "eval_samples_per_second": 19.558,
+      "eval_steps_per_second": 1.326,
+      "step": 320
+    },
+    {
+      "epoch": 18.88,
+      "learning_rate": 4.7662538699690404e-05,
+      "loss": 0.083,
+      "step": 321
+    },
+    {
+      "epoch": 18.94,
+      "learning_rate": 4.7647058823529414e-05,
+      "loss": 0.0779,
+      "step": 322
+    },
+    {
+      "epoch": 19.0,
+      "learning_rate": 4.7631578947368424e-05,
+      "loss": 0.087,
+      "step": 323
+    },
+    {
+      "epoch": 19.06,
+      "learning_rate": 4.7616099071207434e-05,
+      "loss": 0.0676,
+      "step": 324
+    },
+    {
+      "epoch": 19.12,
+      "learning_rate": 4.7600619195046444e-05,
+      "loss": 0.0646,
+      "step": 325
+    },
+    {
+      "epoch": 19.18,
+      "learning_rate": 4.7585139318885454e-05,
+      "loss": 0.0675,
+      "step": 326
+    },
+    {
+      "epoch": 19.24,
+      "learning_rate": 4.7569659442724464e-05,
+      "loss": 0.0754,
+      "step": 327
+    },
+    {
+      "epoch": 19.29,
+      "learning_rate": 4.755417956656347e-05,
+      "loss": 0.0707,
+      "step": 328
+    },
+    {
+      "epoch": 19.35,
+      "learning_rate": 4.753869969040248e-05,
+      "loss": 0.0819,
+      "step": 329
+    },
+    {
+      "epoch": 19.41,
+      "learning_rate": 4.752321981424149e-05,
+      "loss": 0.077,
+      "step": 330
+    },
+    {
+      "epoch": 19.47,
+      "learning_rate": 4.75077399380805e-05,
+      "loss": 0.0674,
+      "step": 331
+    },
+    {
+      "epoch": 19.53,
+      "learning_rate": 4.749226006191951e-05,
+      "loss": 0.0637,
+      "step": 332
+    },
+    {
+      "epoch": 19.59,
+      "learning_rate": 4.747678018575852e-05,
+      "loss": 0.0711,
+      "step": 333
+    },
+    {
+      "epoch": 19.65,
+      "learning_rate": 4.746130030959752e-05,
+      "loss": 0.0842,
+      "step": 334
+    },
+    {
+      "epoch": 19.71,
+      "learning_rate": 4.744582043343653e-05,
+      "loss": 0.0651,
+      "step": 335
+    },
+    {
+      "epoch": 19.76,
+      "learning_rate": 4.743034055727554e-05,
+      "loss": 0.071,
+      "step": 336
+    },
+    {
+      "epoch": 19.82,
+      "learning_rate": 4.741486068111455e-05,
+      "loss": 0.0744,
+      "step": 337
+    },
+    {
+      "epoch": 19.88,
+      "learning_rate": 4.739938080495356e-05,
+      "loss": 0.0751,
+      "step": 338
+    },
+    {
+      "epoch": 19.94,
+      "learning_rate": 4.738390092879257e-05,
+      "loss": 0.0786,
+      "step": 339
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 4.736842105263158e-05,
+      "loss": 0.0659,
+      "step": 340
+    },
+    {
+      "epoch": 20.0,
+      "eval_accuracy_NAT": 0.9924402235850037,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9213479710568117,
+      "eval_accuracy_concrete tiles": 0.8307801084138581,
+      "eval_accuracy_dark bricks": 0.8274697088605172,
+      "eval_accuracy_grass": 0.867474177444201,
+      "eval_accuracy_red bricks": 0.867805360281195,
+      "eval_accuracy_road": 0.9700555953101778,
+      "eval_accuracy_rough concrete": 0.6168431464852013,
+      "eval_accuracy_rough red sidewalk": 0.7044646147136014,
+      "eval_accuracy_speedway bricks": 0.9838130186794876,
+      "eval_accuracy_steel": 0.8907150389580533,
+      "eval_accuracy_tiles": 0.1436274955315217,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.977106375940702,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8576227324911442,
+      "eval_iou_concrete tiles": 0.7540268241031893,
+      "eval_iou_dark bricks": 0.8016820124806439,
+      "eval_iou_grass": 0.80373527797601,
+      "eval_iou_red bricks": 0.8045315682281059,
+      "eval_iou_road": 0.94770066933286,
+      "eval_iou_rough concrete": 0.5887793058731188,
+      "eval_iou_rough red sidewalk": 0.5287908465372523,
+      "eval_iou_speedway bricks": 0.9634395180447106,
+      "eval_iou_steel": 0.7817665804716818,
+      "eval_iou_tiles": 0.1388002459232368,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09484652429819107,
+      "eval_mean_accuracy": 0.7397566507168947,
+      "eval_mean_iou": 0.688306304415589,
+      "eval_overall_accuracy": 0.9722490795588089,
+      "eval_runtime": 5.4896,
+      "eval_samples_per_second": 21.495,
+      "eval_steps_per_second": 1.457,
+      "step": 340
+    },
+    {
+      "epoch": 20.06,
+      "learning_rate": 4.735294117647059e-05,
+      "loss": 0.0644,
+      "step": 341
+    },
+    {
+      "epoch": 20.12,
+      "learning_rate": 4.73374613003096e-05,
+      "loss": 0.0652,
+      "step": 342
+    },
+    {
+      "epoch": 20.18,
+      "learning_rate": 4.732198142414861e-05,
+      "loss": 0.065,
+      "step": 343
+    },
+    {
+      "epoch": 20.24,
+      "learning_rate": 4.730650154798762e-05,
+      "loss": 0.103,
+      "step": 344
+    },
+    {
+      "epoch": 20.29,
+      "learning_rate": 4.729102167182663e-05,
+      "loss": 0.0546,
+      "step": 345
+    },
+    {
+      "epoch": 20.35,
+      "learning_rate": 4.7275541795665634e-05,
+      "loss": 0.0758,
+      "step": 346
+    },
+    {
+      "epoch": 20.41,
+      "learning_rate": 4.7260061919504644e-05,
+      "loss": 0.0645,
+      "step": 347
+    },
+    {
+      "epoch": 20.47,
+      "learning_rate": 4.7244582043343654e-05,
+      "loss": 0.0779,
+      "step": 348
+    },
+    {
+      "epoch": 20.53,
+      "learning_rate": 4.7229102167182664e-05,
+      "loss": 0.066,
+      "step": 349
+    },
+    {
+      "epoch": 20.59,
+      "learning_rate": 4.7213622291021674e-05,
+      "loss": 0.0758,
+      "step": 350
+    },
+    {
+      "epoch": 20.65,
+      "learning_rate": 4.7198142414860684e-05,
+      "loss": 0.0775,
+      "step": 351
+    },
+    {
+      "epoch": 20.71,
+      "learning_rate": 4.718266253869969e-05,
+      "loss": 0.066,
+      "step": 352
+    },
+    {
+      "epoch": 20.76,
+      "learning_rate": 4.71671826625387e-05,
+      "loss": 0.073,
+      "step": 353
+    },
+    {
+      "epoch": 20.82,
+      "learning_rate": 4.715170278637771e-05,
+      "loss": 0.0638,
+      "step": 354
+    },
+    {
+      "epoch": 20.88,
+      "learning_rate": 4.7136222910216724e-05,
+      "loss": 0.0726,
+      "step": 355
+    },
+    {
+      "epoch": 20.94,
+      "learning_rate": 4.7120743034055734e-05,
+      "loss": 0.0797,
+      "step": 356
+    },
+    {
+      "epoch": 21.0,
+      "learning_rate": 4.7105263157894744e-05,
+      "loss": 0.0823,
+      "step": 357
+    },
+    {
+      "epoch": 21.06,
+      "learning_rate": 4.708978328173375e-05,
+      "loss": 0.0607,
+      "step": 358
+    },
+    {
+      "epoch": 21.12,
+      "learning_rate": 4.707430340557276e-05,
+      "loss": 0.0682,
+      "step": 359
+    },
+    {
+      "epoch": 21.18,
+      "learning_rate": 4.705882352941177e-05,
+      "loss": 0.0615,
+      "step": 360
+    },
+    {
+      "epoch": 21.18,
+      "eval_accuracy_NAT": 0.9913629588153516,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9522984701326538,
+      "eval_accuracy_concrete tiles": 0.7668159321234975,
+      "eval_accuracy_dark bricks": 0.8173152384051452,
+      "eval_accuracy_grass": 0.8567212450843468,
+      "eval_accuracy_red bricks": 0.7849297012302284,
+      "eval_accuracy_road": 0.9600828558770543,
+      "eval_accuracy_rough concrete": 0.608846435743271,
+      "eval_accuracy_rough red sidewalk": 0.6382661924063198,
+      "eval_accuracy_speedway bricks": 0.9756119963835097,
+      "eval_accuracy_steel": 0.8969873105199239,
+      "eval_accuracy_tiles": 0.039383198521615315,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9765665415339276,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8440068262299035,
+      "eval_iou_concrete tiles": 0.7239714292071827,
+      "eval_iou_dark bricks": 0.800682565056982,
+      "eval_iou_grass": 0.7904247081326513,
+      "eval_iou_red bricks": 0.7613466865544428,
+      "eval_iou_road": 0.9455310239636968,
+      "eval_iou_rough concrete": 0.6013111898400443,
+      "eval_iou_rough red sidewalk": 0.5301668405239455,
+      "eval_iou_speedway bricks": 0.9648085398083491,
+      "eval_iou_steel": 0.7869885794342117,
+      "eval_iou_tiles": 0.03864791747183161,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09980905055999756,
+      "eval_mean_accuracy": 0.7145093488648399,
+      "eval_mean_iou": 0.6741886805967052,
+      "eval_overall_accuracy": 0.9711400694766287,
+      "eval_runtime": 6.4095,
+      "eval_samples_per_second": 18.41,
+      "eval_steps_per_second": 1.248,
+      "step": 360
+    },
+    {
+      "epoch": 21.24,
+      "learning_rate": 4.704334365325078e-05,
+      "loss": 0.0638,
+      "step": 361
+    },
+    {
+      "epoch": 21.29,
+      "learning_rate": 4.702786377708979e-05,
+      "loss": 0.0664,
+      "step": 362
+    },
+    {
+      "epoch": 21.35,
+      "learning_rate": 4.70123839009288e-05,
+      "loss": 0.0702,
+      "step": 363
+    },
+    {
+      "epoch": 21.41,
+      "learning_rate": 4.69969040247678e-05,
+      "loss": 0.0618,
+      "step": 364
+    },
+    {
+      "epoch": 21.47,
+      "learning_rate": 4.698142414860681e-05,
+      "loss": 0.0729,
+      "step": 365
+    },
+    {
+      "epoch": 21.53,
+      "learning_rate": 4.696594427244582e-05,
+      "loss": 0.092,
+      "step": 366
+    },
+    {
+      "epoch": 21.59,
+      "learning_rate": 4.695046439628483e-05,
+      "loss": 0.0798,
+      "step": 367
+    },
+    {
+      "epoch": 21.65,
+      "learning_rate": 4.693498452012384e-05,
+      "loss": 0.0659,
+      "step": 368
+    },
+    {
+      "epoch": 21.71,
+      "learning_rate": 4.691950464396285e-05,
+      "loss": 0.0672,
+      "step": 369
+    },
+    {
+      "epoch": 21.76,
+      "learning_rate": 4.690402476780186e-05,
+      "loss": 0.0756,
+      "step": 370
+    },
+    {
+      "epoch": 21.82,
+      "learning_rate": 4.688854489164087e-05,
+      "loss": 0.0689,
+      "step": 371
+    },
+    {
+      "epoch": 21.88,
+      "learning_rate": 4.687306501547988e-05,
+      "loss": 0.0713,
+      "step": 372
+    },
+    {
+      "epoch": 21.94,
+      "learning_rate": 4.685758513931889e-05,
+      "loss": 0.0562,
+      "step": 373
+    },
+    {
+      "epoch": 22.0,
+      "learning_rate": 4.68421052631579e-05,
+      "loss": 0.0657,
+      "step": 374
+    },
+    {
+      "epoch": 22.06,
+      "learning_rate": 4.682662538699691e-05,
+      "loss": 0.0626,
+      "step": 375
+    },
+    {
+      "epoch": 22.12,
+      "learning_rate": 4.6811145510835914e-05,
+      "loss": 0.065,
+      "step": 376
+    },
+    {
+      "epoch": 22.18,
+      "learning_rate": 4.6795665634674924e-05,
+      "loss": 0.0681,
+      "step": 377
+    },
+    {
+      "epoch": 22.24,
+      "learning_rate": 4.6780185758513934e-05,
+      "loss": 0.0616,
+      "step": 378
+    },
+    {
+      "epoch": 22.29,
+      "learning_rate": 4.6764705882352944e-05,
+      "loss": 0.0726,
+      "step": 379
+    },
+    {
+      "epoch": 22.35,
+      "learning_rate": 4.6749226006191954e-05,
+      "loss": 0.0671,
+      "step": 380
+    },
+    {
+      "epoch": 22.35,
+      "eval_accuracy_NAT": 0.9918347945306951,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9463292815111854,
+      "eval_accuracy_concrete tiles": 0.7548432712703276,
+      "eval_accuracy_dark bricks": 0.8054093371360757,
+      "eval_accuracy_grass": 0.8488552536458156,
+      "eval_accuracy_red bricks": 0.8876318101933216,
+      "eval_accuracy_road": 0.967290526244909,
+      "eval_accuracy_rough concrete": 0.6129401319278508,
+      "eval_accuracy_rough red sidewalk": 0.7476669248527963,
+      "eval_accuracy_speedway bricks": 0.9791248107271846,
+      "eval_accuracy_steel": 0.8704263645314289,
+      "eval_accuracy_tiles": 0.1060013935593323,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9774210083281781,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8496639620054871,
+      "eval_iou_concrete tiles": 0.7044694703501672,
+      "eval_iou_dark bricks": 0.7850040111191955,
+      "eval_iou_grass": 0.7914839253842089,
+      "eval_iou_red bricks": 0.8270815209047644,
+      "eval_iou_road": 0.9404892116236474,
+      "eval_iou_rough concrete": 0.5939658159140778,
+      "eval_iou_rough red sidewalk": 0.5748613785248016,
+      "eval_iou_speedway bricks": 0.9670993189974229,
+      "eval_iou_steel": 0.7935183012536479,
+      "eval_iou_tiles": 0.10403472779710403,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09570205211639404,
+      "eval_mean_accuracy": 0.7321810692408403,
+      "eval_mean_iou": 0.685314819400208,
+      "eval_overall_accuracy": 0.9719316514871889,
+      "eval_runtime": 5.6048,
+      "eval_samples_per_second": 21.053,
+      "eval_steps_per_second": 1.427,
+      "step": 380
+    },
+    {
+      "epoch": 22.41,
+      "learning_rate": 4.6733746130030964e-05,
+      "loss": 0.0598,
+      "step": 381
+    },
+    {
+      "epoch": 22.47,
+      "learning_rate": 4.671826625386997e-05,
+      "loss": 0.0697,
+      "step": 382
+    },
+    {
+      "epoch": 22.53,
+      "learning_rate": 4.670278637770898e-05,
+      "loss": 0.0661,
+      "step": 383
+    },
+    {
+      "epoch": 22.59,
+      "learning_rate": 4.668730650154799e-05,
+      "loss": 0.0627,
+      "step": 384
+    },
+    {
+      "epoch": 22.65,
+      "learning_rate": 4.6671826625387e-05,
+      "loss": 0.0702,
+      "step": 385
+    },
+    {
+      "epoch": 22.71,
+      "learning_rate": 4.665634674922601e-05,
+      "loss": 0.0587,
+      "step": 386
+    },
+    {
+      "epoch": 22.76,
+      "learning_rate": 4.664086687306502e-05,
+      "loss": 0.0846,
+      "step": 387
+    },
+    {
+      "epoch": 22.82,
+      "learning_rate": 4.662538699690403e-05,
+      "loss": 0.0624,
+      "step": 388
+    },
+    {
+      "epoch": 22.88,
+      "learning_rate": 4.660990712074304e-05,
+      "loss": 0.0665,
+      "step": 389
+    },
+    {
+      "epoch": 22.94,
+      "learning_rate": 4.659442724458205e-05,
+      "loss": 0.0736,
+      "step": 390
+    },
+    {
+      "epoch": 23.0,
+      "learning_rate": 4.657894736842106e-05,
+      "loss": 0.0619,
+      "step": 391
+    },
+    {
+      "epoch": 23.06,
+      "learning_rate": 4.656346749226007e-05,
+      "loss": 0.08,
+      "step": 392
+    },
+    {
+      "epoch": 23.12,
+      "learning_rate": 4.654798761609908e-05,
+      "loss": 0.063,
+      "step": 393
+    },
+    {
+      "epoch": 23.18,
+      "learning_rate": 4.653250773993808e-05,
+      "loss": 0.0666,
+      "step": 394
+    },
+    {
+      "epoch": 23.24,
+      "learning_rate": 4.651702786377709e-05,
+      "loss": 0.0528,
+      "step": 395
+    },
+    {
+      "epoch": 23.29,
+      "learning_rate": 4.65015479876161e-05,
+      "loss": 0.0542,
+      "step": 396
+    },
+    {
+      "epoch": 23.35,
+      "learning_rate": 4.648606811145511e-05,
+      "loss": 0.0632,
+      "step": 397
+    },
+    {
+      "epoch": 23.41,
+      "learning_rate": 4.647058823529412e-05,
+      "loss": 0.0504,
+      "step": 398
+    },
+    {
+      "epoch": 23.47,
+      "learning_rate": 4.645510835913313e-05,
+      "loss": 0.0689,
+      "step": 399
+    },
+    {
+      "epoch": 23.53,
+      "learning_rate": 4.6439628482972134e-05,
+      "loss": 0.0917,
+      "step": 400
+    },
+    {
+      "epoch": 23.53,
+      "eval_accuracy_NAT": 0.9909434175063019,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9335657359858984,
+      "eval_accuracy_concrete tiles": 0.7849870374734857,
+      "eval_accuracy_dark bricks": 0.8096204276170971,
+      "eval_accuracy_grass": 0.8686938650376409,
+      "eval_accuracy_red bricks": 0.8738466608084359,
+      "eval_accuracy_road": 0.9780349625721663,
+      "eval_accuracy_rough concrete": 0.6157526859294835,
+      "eval_accuracy_rough red sidewalk": 0.6848976154129599,
+      "eval_accuracy_speedway bricks": 0.9868086529807598,
+      "eval_accuracy_steel": 0.8361887578903177,
+      "eval_accuracy_tiles": 0.10460783422702899,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9777327551888214,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8544803622196483,
+      "eval_iou_concrete tiles": 0.732021978021978,
+      "eval_iou_dark bricks": 0.7838293614024684,
+      "eval_iou_grass": 0.8003822054755468,
+      "eval_iou_red bricks": 0.8379502843901412,
+      "eval_iou_road": 0.9416817001622897,
+      "eval_iou_rough concrete": 0.599613545631789,
+      "eval_iou_rough red sidewalk": 0.5411039883077764,
+      "eval_iou_speedway bricks": 0.9665345059616663,
+      "eval_iou_steel": 0.7894853567442969,
+      "eval_iou_tiles": 0.1043202416918429,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09354794770479202,
+      "eval_mean_accuracy": 0.7283036656493521,
+      "eval_mean_iou": 0.6868566373229436,
+      "eval_overall_accuracy": 0.9724575301348153,
+      "eval_runtime": 6.1543,
+      "eval_samples_per_second": 19.174,
+      "eval_steps_per_second": 1.3,
+      "step": 400
+    },
+    {
+      "epoch": 23.59,
+      "learning_rate": 4.6424148606811144e-05,
+      "loss": 0.0554,
+      "step": 401
+    },
+    {
+      "epoch": 23.65,
+      "learning_rate": 4.6408668730650154e-05,
+      "loss": 0.0656,
+      "step": 402
+    },
+    {
+      "epoch": 23.71,
+      "learning_rate": 4.6393188854489164e-05,
+      "loss": 0.0757,
+      "step": 403
+    },
+    {
+      "epoch": 23.76,
+      "learning_rate": 4.637770897832818e-05,
+      "loss": 0.0679,
+      "step": 404
+    },
+    {
+      "epoch": 23.82,
+      "learning_rate": 4.636222910216719e-05,
+      "loss": 0.0602,
+      "step": 405
+    },
+    {
+      "epoch": 23.88,
+      "learning_rate": 4.6346749226006194e-05,
+      "loss": 0.0704,
+      "step": 406
+    },
+    {
+      "epoch": 23.94,
+      "learning_rate": 4.6331269349845204e-05,
+      "loss": 0.0603,
+      "step": 407
+    },
+    {
+      "epoch": 24.0,
+      "learning_rate": 4.6315789473684214e-05,
+      "loss": 0.0736,
+      "step": 408
+    },
+    {
+      "epoch": 24.06,
+      "learning_rate": 4.6300309597523224e-05,
+      "loss": 0.0631,
+      "step": 409
+    },
+    {
+      "epoch": 24.12,
+      "learning_rate": 4.6284829721362234e-05,
+      "loss": 0.058,
+      "step": 410
+    },
+    {
+      "epoch": 24.18,
+      "learning_rate": 4.6269349845201244e-05,
+      "loss": 0.0653,
+      "step": 411
+    },
+    {
+      "epoch": 24.24,
+      "learning_rate": 4.625386996904025e-05,
+      "loss": 0.0701,
+      "step": 412
+    },
+    {
+      "epoch": 24.29,
+      "learning_rate": 4.623839009287926e-05,
+      "loss": 0.0489,
+      "step": 413
+    },
+    {
+      "epoch": 24.35,
+      "learning_rate": 4.622291021671827e-05,
+      "loss": 0.0624,
+      "step": 414
+    },
+    {
+      "epoch": 24.41,
+      "learning_rate": 4.620743034055728e-05,
+      "loss": 0.065,
+      "step": 415
+    },
+    {
+      "epoch": 24.47,
+      "learning_rate": 4.619195046439629e-05,
+      "loss": 0.0596,
+      "step": 416
+    },
+    {
+      "epoch": 24.53,
+      "learning_rate": 4.61764705882353e-05,
+      "loss": 0.0619,
+      "step": 417
+    },
+    {
+      "epoch": 24.59,
+      "learning_rate": 4.616099071207431e-05,
+      "loss": 0.0546,
+      "step": 418
+    },
+    {
+      "epoch": 24.65,
+      "learning_rate": 4.614551083591332e-05,
+      "loss": 0.0572,
+      "step": 419
+    },
+    {
+      "epoch": 24.71,
+      "learning_rate": 4.613003095975233e-05,
+      "loss": 0.0649,
+      "step": 420
+    },
+    {
+      "epoch": 24.71,
+      "eval_accuracy_NAT": 0.9925458906327725,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9282929144836717,
+      "eval_accuracy_concrete tiles": 0.8514023096865425,
+      "eval_accuracy_dark bricks": 0.8465153226269548,
+      "eval_accuracy_grass": 0.8541730537041032,
+      "eval_accuracy_red bricks": 0.9041080843585237,
+      "eval_accuracy_road": 0.9788007808888516,
+      "eval_accuracy_rough concrete": 0.623808984679923,
+      "eval_accuracy_rough red sidewalk": 0.6470946540431196,
+      "eval_accuracy_speedway bricks": 0.9883149527066442,
+      "eval_accuracy_steel": 0.8562525301493851,
+      "eval_accuracy_tiles": 0.13562967675482443,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9782598114699761,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8609499442060025,
+      "eval_iou_concrete tiles": 0.7683064293158085,
+      "eval_iou_dark bricks": 0.816763934547335,
+      "eval_iou_grass": 0.8033950859006216,
+      "eval_iou_red bricks": 0.8423476436575756,
+      "eval_iou_road": 0.9410264736316956,
+      "eval_iou_rough concrete": 0.6038078648963536,
+      "eval_iou_rough red sidewalk": 0.544589232258922,
+      "eval_iou_speedway bricks": 0.9664483963533285,
+      "eval_iou_steel": 0.7954582387326526,
+      "eval_iou_tiles": 0.13351027346196284,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08796583861112595,
+      "eval_mean_accuracy": 0.7389953195934861,
+      "eval_mean_iou": 0.696527948340941,
+      "eval_overall_accuracy": 0.9733388545149464,
+      "eval_runtime": 8.4846,
+      "eval_samples_per_second": 13.908,
+      "eval_steps_per_second": 0.943,
+      "step": 420
+    },
+    {
+      "epoch": 24.76,
+      "learning_rate": 4.611455108359134e-05,
+      "loss": 0.0556,
+      "step": 421
+    },
+    {
+      "epoch": 24.82,
+      "learning_rate": 4.609907120743035e-05,
+      "loss": 0.0623,
+      "step": 422
+    },
+    {
+      "epoch": 24.88,
+      "learning_rate": 4.608359133126936e-05,
+      "loss": 0.0669,
+      "step": 423
+    },
+    {
+      "epoch": 24.94,
+      "learning_rate": 4.606811145510836e-05,
+      "loss": 0.0577,
+      "step": 424
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 4.605263157894737e-05,
+      "loss": 0.0598,
+      "step": 425
+    },
+    {
+      "epoch": 25.06,
+      "learning_rate": 4.603715170278638e-05,
+      "loss": 0.0633,
+      "step": 426
+    },
+    {
+      "epoch": 25.12,
+      "learning_rate": 4.602167182662539e-05,
+      "loss": 0.0547,
+      "step": 427
+    },
+    {
+      "epoch": 25.18,
+      "learning_rate": 4.60061919504644e-05,
+      "loss": 0.0553,
+      "step": 428
+    },
+    {
+      "epoch": 25.24,
+      "learning_rate": 4.599071207430341e-05,
+      "loss": 0.0568,
+      "step": 429
+    },
+    {
+      "epoch": 25.29,
+      "learning_rate": 4.597523219814241e-05,
+      "loss": 0.052,
+      "step": 430
+    },
+    {
+      "epoch": 25.35,
+      "learning_rate": 4.595975232198142e-05,
+      "loss": 0.0671,
+      "step": 431
+    },
+    {
+      "epoch": 25.41,
+      "learning_rate": 4.594427244582043e-05,
+      "loss": 0.0603,
+      "step": 432
+    },
+    {
+      "epoch": 25.47,
+      "learning_rate": 4.592879256965944e-05,
+      "loss": 0.0586,
+      "step": 433
+    },
+    {
+      "epoch": 25.53,
+      "learning_rate": 4.591331269349845e-05,
+      "loss": 0.0734,
+      "step": 434
+    },
+    {
+      "epoch": 25.59,
+      "learning_rate": 4.589783281733746e-05,
+      "loss": 0.0559,
+      "step": 435
+    },
+    {
+      "epoch": 25.65,
+      "learning_rate": 4.588235294117647e-05,
+      "loss": 0.0498,
+      "step": 436
+    },
+    {
+      "epoch": 25.71,
+      "learning_rate": 4.586687306501548e-05,
+      "loss": 0.0595,
+      "step": 437
+    },
+    {
+      "epoch": 25.76,
+      "learning_rate": 4.585139318885449e-05,
+      "loss": 0.0491,
+      "step": 438
+    },
+    {
+      "epoch": 25.82,
+      "learning_rate": 4.58359133126935e-05,
+      "loss": 0.0623,
+      "step": 439
+    },
+    {
+      "epoch": 25.88,
+      "learning_rate": 4.582043343653251e-05,
+      "loss": 0.0648,
+      "step": 440
+    },
+    {
+      "epoch": 25.88,
+      "eval_accuracy_NAT": 0.9919881734852373,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9411478056409559,
+      "eval_accuracy_concrete tiles": 0.8343860476078246,
+      "eval_accuracy_dark bricks": 0.8282257910150642,
+      "eval_accuracy_grass": 0.8704670958755824,
+      "eval_accuracy_red bricks": 0.8488027240773286,
+      "eval_accuracy_road": 0.9759058487035867,
+      "eval_accuracy_rough concrete": 0.6163366484128734,
+      "eval_accuracy_rough red sidewalk": 0.7025846186239932,
+      "eval_accuracy_speedway bricks": 0.9869561983805307,
+      "eval_accuracy_steel": 0.834177132947168,
+      "eval_accuracy_tiles": 0.10306280105425793,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788527237848805,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8594148468537138,
+      "eval_iou_concrete tiles": 0.79045726533893,
+      "eval_iou_dark bricks": 0.8074231652018138,
+      "eval_iou_grass": 0.8148623846709846,
+      "eval_iou_red bricks": 0.8211572180011689,
+      "eval_iou_road": 0.9523440987621357,
+      "eval_iou_rough concrete": 0.5988443790853457,
+      "eval_iou_rough red sidewalk": 0.5235994978591764,
+      "eval_iou_speedway bricks": 0.9685424964705743,
+      "eval_iou_steel": 0.792603356412229,
+      "eval_iou_tiles": 0.1028105167724388,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08859998732805252,
+      "eval_mean_accuracy": 0.733387760448031,
+      "eval_mean_iou": 0.6931470730164146,
+      "eval_overall_accuracy": 0.973915197081485,
+      "eval_runtime": 10.9462,
+      "eval_samples_per_second": 10.78,
+      "eval_steps_per_second": 0.731,
+      "step": 440
+    },
+    {
+      "epoch": 25.94,
+      "learning_rate": 4.580495356037152e-05,
+      "loss": 0.05,
+      "step": 441
+    },
+    {
+      "epoch": 26.0,
+      "learning_rate": 4.5789473684210527e-05,
+      "loss": 0.0547,
+      "step": 442
+    },
+    {
+      "epoch": 26.06,
+      "learning_rate": 4.5773993808049537e-05,
+      "loss": 0.0636,
+      "step": 443
+    },
+    {
+      "epoch": 26.12,
+      "learning_rate": 4.5758513931888547e-05,
+      "loss": 0.0503,
+      "step": 444
+    },
+    {
+      "epoch": 26.18,
+      "learning_rate": 4.5743034055727557e-05,
+      "loss": 0.0486,
+      "step": 445
+    },
+    {
+      "epoch": 26.24,
+      "learning_rate": 4.5727554179566567e-05,
+      "loss": 0.0596,
+      "step": 446
+    },
+    {
+      "epoch": 26.29,
+      "learning_rate": 4.5712074303405577e-05,
+      "loss": 0.0578,
+      "step": 447
+    },
+    {
+      "epoch": 26.35,
+      "learning_rate": 4.569659442724458e-05,
+      "loss": 0.0709,
+      "step": 448
+    },
+    {
+      "epoch": 26.41,
+      "learning_rate": 4.568111455108359e-05,
+      "loss": 0.0521,
+      "step": 449
+    },
+    {
+      "epoch": 26.47,
+      "learning_rate": 4.56656346749226e-05,
+      "loss": 0.0509,
+      "step": 450
+    },
+    {
+      "epoch": 26.53,
+      "learning_rate": 4.565015479876161e-05,
+      "loss": 0.0582,
+      "step": 451
+    },
+    {
+      "epoch": 26.59,
+      "learning_rate": 4.563467492260062e-05,
+      "loss": 0.0571,
+      "step": 452
+    },
+    {
+      "epoch": 26.65,
+      "learning_rate": 4.561919504643964e-05,
+      "loss": 0.0596,
+      "step": 453
+    },
+    {
+      "epoch": 26.71,
+      "learning_rate": 4.560371517027864e-05,
+      "loss": 0.0588,
+      "step": 454
+    },
+    {
+      "epoch": 26.76,
+      "learning_rate": 4.558823529411765e-05,
+      "loss": 0.0409,
+      "step": 455
+    },
+    {
+      "epoch": 26.82,
+      "learning_rate": 4.557275541795666e-05,
+      "loss": 0.0521,
+      "step": 456
+    },
+    {
+      "epoch": 26.88,
+      "learning_rate": 4.555727554179567e-05,
+      "loss": 0.0638,
+      "step": 457
+    },
+    {
+      "epoch": 26.94,
+      "learning_rate": 4.554179566563468e-05,
+      "loss": 0.0627,
+      "step": 458
+    },
+    {
+      "epoch": 27.0,
+      "learning_rate": 4.552631578947369e-05,
+      "loss": 0.0633,
+      "step": 459
+    },
+    {
+      "epoch": 27.06,
+      "learning_rate": 4.551083591331269e-05,
+      "loss": 0.0633,
+      "step": 460
+    },
+    {
+      "epoch": 27.06,
+      "eval_accuracy_NAT": 0.9912690984428182,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9297857849379844,
+      "eval_accuracy_concrete tiles": 0.8931180768324298,
+      "eval_accuracy_dark bricks": 0.8600290948069598,
+      "eval_accuracy_grass": 0.8679898715779673,
+      "eval_accuracy_red bricks": 0.9230008787346221,
+      "eval_accuracy_road": 0.9769875069540899,
+      "eval_accuracy_rough concrete": 0.6186665395455818,
+      "eval_accuracy_rough red sidewalk": 0.7836726099609713,
+      "eval_accuracy_speedway bricks": 0.9904856351443888,
+      "eval_accuracy_steel": 0.8313108797673012,
+      "eval_accuracy_tiles": 0.13214577842406616,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9784511593154107,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8657692444570981,
+      "eval_iou_concrete tiles": 0.7636887608069164,
+      "eval_iou_dark bricks": 0.8113273984723451,
+      "eval_iou_grass": 0.8026909321920439,
+      "eval_iou_red bricks": 0.8375778719162721,
+      "eval_iou_road": 0.945710151690017,
+      "eval_iou_rough concrete": 0.6000277405335429,
+      "eval_iou_rough red sidewalk": 0.6096657754792814,
+      "eval_iou_speedway bricks": 0.9673607821937777,
+      "eval_iou_steel": 0.7921859893603341,
+      "eval_iou_tiles": 0.13162341581170792,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08854084461927414,
+      "eval_mean_accuracy": 0.7537278273176292,
+      "eval_mean_iou": 0.7004676324791345,
+      "eval_overall_accuracy": 0.9738638603081138,
+      "eval_runtime": 5.7428,
+      "eval_samples_per_second": 20.548,
+      "eval_steps_per_second": 1.393,
+      "step": 460
+    },
+    {
+      "epoch": 27.12,
+      "learning_rate": 4.54953560371517e-05,
+      "loss": 0.0449,
+      "step": 461
+    },
+    {
+      "epoch": 27.18,
+      "learning_rate": 4.547987616099071e-05,
+      "loss": 0.0513,
+      "step": 462
+    },
+    {
+      "epoch": 27.24,
+      "learning_rate": 4.546439628482972e-05,
+      "loss": 0.0553,
+      "step": 463
+    },
+    {
+      "epoch": 27.29,
+      "learning_rate": 4.544891640866873e-05,
+      "loss": 0.0509,
+      "step": 464
+    },
+    {
+      "epoch": 27.35,
+      "learning_rate": 4.5433436532507736e-05,
+      "loss": 0.0608,
+      "step": 465
+    },
+    {
+      "epoch": 27.41,
+      "learning_rate": 4.5417956656346746e-05,
+      "loss": 0.0623,
+      "step": 466
+    },
+    {
+      "epoch": 27.47,
+      "learning_rate": 4.540247678018576e-05,
+      "loss": 0.0538,
+      "step": 467
+    },
+    {
+      "epoch": 27.53,
+      "learning_rate": 4.538699690402477e-05,
+      "loss": 0.0662,
+      "step": 468
+    },
+    {
+      "epoch": 27.59,
+      "learning_rate": 4.537151702786378e-05,
+      "loss": 0.0469,
+      "step": 469
+    },
+    {
+      "epoch": 27.65,
+      "learning_rate": 4.535603715170279e-05,
+      "loss": 0.0562,
+      "step": 470
+    },
+    {
+      "epoch": 27.71,
+      "learning_rate": 4.5340557275541796e-05,
+      "loss": 0.0491,
+      "step": 471
+    },
+    {
+      "epoch": 27.76,
+      "learning_rate": 4.5325077399380806e-05,
+      "loss": 0.0563,
+      "step": 472
+    },
+    {
+      "epoch": 27.82,
+      "learning_rate": 4.5309597523219816e-05,
+      "loss": 0.0535,
+      "step": 473
+    },
+    {
+      "epoch": 27.88,
+      "learning_rate": 4.5294117647058826e-05,
+      "loss": 0.0462,
+      "step": 474
+    },
+    {
+      "epoch": 27.94,
+      "learning_rate": 4.5278637770897836e-05,
+      "loss": 0.0533,
+      "step": 475
+    },
+    {
+      "epoch": 28.0,
+      "learning_rate": 4.5263157894736846e-05,
+      "loss": 0.0563,
+      "step": 476
+    },
+    {
+      "epoch": 28.06,
+      "learning_rate": 4.524767801857585e-05,
+      "loss": 0.0529,
+      "step": 477
+    },
+    {
+      "epoch": 28.12,
+      "learning_rate": 4.523219814241486e-05,
+      "loss": 0.0604,
+      "step": 478
+    },
+    {
+      "epoch": 28.18,
+      "learning_rate": 4.521671826625387e-05,
+      "loss": 0.0485,
+      "step": 479
+    },
+    {
+      "epoch": 28.24,
+      "learning_rate": 4.520123839009288e-05,
+      "loss": 0.0504,
+      "step": 480
+    },
+    {
+      "epoch": 28.24,
+      "eval_accuracy_NAT": 0.9911260166342576,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9331674843183628,
+      "eval_accuracy_concrete tiles": 0.8847749234032524,
+      "eval_accuracy_dark bricks": 0.8469077196945045,
+      "eval_accuracy_grass": 0.8925321812064025,
+      "eval_accuracy_red bricks": 0.9044925307557118,
+      "eval_accuracy_road": 0.9779024273715117,
+      "eval_accuracy_rough concrete": 0.6204899326059624,
+      "eval_accuracy_rough red sidewalk": 0.7195572233209755,
+      "eval_accuracy_speedway bricks": 0.9882148754552316,
+      "eval_accuracy_steel": 0.8626022680134142,
+      "eval_accuracy_tiles": 0.17510375958072041,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790429664294764,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8675571401566566,
+      "eval_iou_concrete tiles": 0.7879481151876416,
+      "eval_iou_dark bricks": 0.8161326619076605,
+      "eval_iou_grass": 0.8155797060723207,
+      "eval_iou_red bricks": 0.8463870901428718,
+      "eval_iou_road": 0.9503152653229957,
+      "eval_iou_rough concrete": 0.5960606077952111,
+      "eval_iou_rough red sidewalk": 0.536717522997532,
+      "eval_iou_speedway bricks": 0.9697299086152923,
+      "eval_iou_steel": 0.8085007213925687,
+      "eval_iou_tiles": 0.17393397731034274,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08616523444652557,
+      "eval_mean_accuracy": 0.7536054878738698,
+      "eval_mean_iou": 0.7036850525638899,
+      "eval_overall_accuracy": 0.974603491314387,
+      "eval_runtime": 5.9007,
+      "eval_samples_per_second": 19.997,
+      "eval_steps_per_second": 1.356,
+      "step": 480
+    },
+    {
+      "epoch": 28.29,
+      "learning_rate": 4.518575851393189e-05,
+      "loss": 0.0567,
+      "step": 481
+    },
+    {
+      "epoch": 28.35,
+      "learning_rate": 4.51702786377709e-05,
+      "loss": 0.0511,
+      "step": 482
+    },
+    {
+      "epoch": 28.41,
+      "learning_rate": 4.515479876160991e-05,
+      "loss": 0.0489,
+      "step": 483
+    },
+    {
+      "epoch": 28.47,
+      "learning_rate": 4.513931888544892e-05,
+      "loss": 0.0492,
+      "step": 484
+    },
+    {
+      "epoch": 28.53,
+      "learning_rate": 4.512383900928793e-05,
+      "loss": 0.0502,
+      "step": 485
+    },
+    {
+      "epoch": 28.59,
+      "learning_rate": 4.510835913312694e-05,
+      "loss": 0.0536,
+      "step": 486
+    },
+    {
+      "epoch": 28.65,
+      "learning_rate": 4.509287925696595e-05,
+      "loss": 0.0479,
+      "step": 487
+    },
+    {
+      "epoch": 28.71,
+      "learning_rate": 4.507739938080496e-05,
+      "loss": 0.048,
+      "step": 488
+    },
+    {
+      "epoch": 28.76,
+      "learning_rate": 4.506191950464396e-05,
+      "loss": 0.0573,
+      "step": 489
+    },
+    {
+      "epoch": 28.82,
+      "learning_rate": 4.504643962848297e-05,
+      "loss": 0.0667,
+      "step": 490
+    },
+    {
+      "epoch": 28.88,
+      "learning_rate": 4.503095975232198e-05,
+      "loss": 0.0514,
+      "step": 491
+    },
+    {
+      "epoch": 28.94,
+      "learning_rate": 4.501547987616099e-05,
+      "loss": 0.052,
+      "step": 492
+    },
+    {
+      "epoch": 29.0,
+      "learning_rate": 4.5e-05,
+      "loss": 0.0556,
+      "step": 493
+    },
+    {
+      "epoch": 29.06,
+      "learning_rate": 4.498452012383901e-05,
+      "loss": 0.0601,
+      "step": 494
+    },
+    {
+      "epoch": 29.12,
+      "learning_rate": 4.4969040247678016e-05,
+      "loss": 0.0497,
+      "step": 495
+    },
+    {
+      "epoch": 29.18,
+      "learning_rate": 4.4953560371517026e-05,
+      "loss": 0.0451,
+      "step": 496
+    },
+    {
+      "epoch": 29.24,
+      "learning_rate": 4.4938080495356036e-05,
+      "loss": 0.0537,
+      "step": 497
+    },
+    {
+      "epoch": 29.29,
+      "learning_rate": 4.4922600619195046e-05,
+      "loss": 0.0474,
+      "step": 498
+    },
+    {
+      "epoch": 29.35,
+      "learning_rate": 4.4907120743034056e-05,
+      "loss": 0.0533,
+      "step": 499
+    },
+    {
+      "epoch": 29.41,
+      "learning_rate": 4.4891640866873066e-05,
+      "loss": 0.0603,
+      "step": 500
+    },
+    {
+      "epoch": 29.41,
+      "eval_accuracy_NAT": 0.9932370932957547,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.942857383096701,
+      "eval_accuracy_concrete tiles": 0.8190666980909733,
+      "eval_accuracy_dark bricks": 0.8274314262197806,
+      "eval_accuracy_grass": 0.8522957378116401,
+      "eval_accuracy_red bricks": 0.8900483304042179,
+      "eval_accuracy_road": 0.9719020030451031,
+      "eval_accuracy_rough concrete": 0.6163962364213825,
+      "eval_accuracy_rough red sidewalk": 0.7169628287173163,
+      "eval_accuracy_speedway bricks": 0.9857528722513885,
+      "eval_accuracy_steel": 0.8697291678120018,
+      "eval_accuracy_tiles": 0.138083552970402,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9782592362006232,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8635047513437755,
+      "eval_iou_concrete tiles": 0.7952267630772047,
+      "eval_iou_dark bricks": 0.8104902971782132,
+      "eval_iou_grass": 0.8026895039834602,
+      "eval_iou_red bricks": 0.8525883838383839,
+      "eval_iou_road": 0.9459653181836624,
+      "eval_iou_rough concrete": 0.6043007863159986,
+      "eval_iou_rough red sidewalk": 0.5906685996084554,
+      "eval_iou_speedway bricks": 0.9711059681367812,
+      "eval_iou_steel": 0.8125372075052353,
+      "eval_iou_tiles": 0.137033251157477,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08850965648889542,
+      "eval_mean_accuracy": 0.7402894869335893,
+      "eval_mean_iou": 0.7049515435791746,
+      "eval_overall_accuracy": 0.9742760092525159,
+      "eval_runtime": 8.0307,
+      "eval_samples_per_second": 14.694,
+      "eval_steps_per_second": 0.996,
+      "step": 500
+    },
+    {
+      "epoch": 29.47,
+      "learning_rate": 4.4876160990712076e-05,
+      "loss": 0.0463,
+      "step": 501
+    },
+    {
+      "epoch": 29.53,
+      "learning_rate": 4.4860681114551086e-05,
+      "loss": 0.0528,
+      "step": 502
+    },
+    {
+      "epoch": 29.59,
+      "learning_rate": 4.4845201238390096e-05,
+      "loss": 0.0575,
+      "step": 503
+    },
+    {
+      "epoch": 29.65,
+      "learning_rate": 4.4829721362229106e-05,
+      "loss": 0.055,
+      "step": 504
+    },
+    {
+      "epoch": 29.71,
+      "learning_rate": 4.4814241486068116e-05,
+      "loss": 0.0654,
+      "step": 505
+    },
+    {
+      "epoch": 29.76,
+      "learning_rate": 4.4798761609907126e-05,
+      "loss": 0.057,
+      "step": 506
+    },
+    {
+      "epoch": 29.82,
+      "learning_rate": 4.478328173374613e-05,
+      "loss": 0.0434,
+      "step": 507
+    },
+    {
+      "epoch": 29.88,
+      "learning_rate": 4.476780185758514e-05,
+      "loss": 0.0492,
+      "step": 508
+    },
+    {
+      "epoch": 29.94,
+      "learning_rate": 4.475232198142415e-05,
+      "loss": 0.06,
+      "step": 509
+    },
+    {
+      "epoch": 30.0,
+      "learning_rate": 4.473684210526316e-05,
+      "loss": 0.0457,
+      "step": 510
+    },
+    {
+      "epoch": 30.06,
+      "learning_rate": 4.472136222910217e-05,
+      "loss": 0.0561,
+      "step": 511
+    },
+    {
+      "epoch": 30.12,
+      "learning_rate": 4.470588235294118e-05,
+      "loss": 0.046,
+      "step": 512
+    },
+    {
+      "epoch": 30.18,
+      "learning_rate": 4.469040247678018e-05,
+      "loss": 0.0678,
+      "step": 513
+    },
+    {
+      "epoch": 30.24,
+      "learning_rate": 4.467492260061919e-05,
+      "loss": 0.0558,
+      "step": 514
+    },
+    {
+      "epoch": 30.29,
+      "learning_rate": 4.46594427244582e-05,
+      "loss": 0.0437,
+      "step": 515
+    },
+    {
+      "epoch": 30.35,
+      "learning_rate": 4.464396284829722e-05,
+      "loss": 0.0544,
+      "step": 516
+    },
+    {
+      "epoch": 30.41,
+      "learning_rate": 4.462848297213623e-05,
+      "loss": 0.0559,
+      "step": 517
+    },
+    {
+      "epoch": 30.47,
+      "learning_rate": 4.461300309597524e-05,
+      "loss": 0.0521,
+      "step": 518
+    },
+    {
+      "epoch": 30.53,
+      "learning_rate": 4.459752321981424e-05,
+      "loss": 0.0522,
+      "step": 519
+    },
+    {
+      "epoch": 30.59,
+      "learning_rate": 4.458204334365325e-05,
+      "loss": 0.0447,
+      "step": 520
+    },
+    {
+      "epoch": 30.59,
+      "eval_accuracy_NAT": 0.9901760375277598,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9483530267891132,
+      "eval_accuracy_concrete tiles": 0.8590855526749941,
+      "eval_accuracy_dark bricks": 0.875892464062171,
+      "eval_accuracy_grass": 0.8866419133482459,
+      "eval_accuracy_red bricks": 0.9196507029876977,
+      "eval_accuracy_road": 0.9787553555176595,
+      "eval_accuracy_rough concrete": 0.6182732586894213,
+      "eval_accuracy_rough red sidewalk": 0.736898307251521,
+      "eval_accuracy_speedway bricks": 0.985416996544593,
+      "eval_accuracy_steel": 0.8896280118148605,
+      "eval_accuracy_tiles": 0.2665636644551486,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794976770895875,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8698962107547464,
+      "eval_iou_concrete tiles": 0.8193077095976624,
+      "eval_iou_dark bricks": 0.8351637822077632,
+      "eval_iou_grass": 0.8186954037349569,
+      "eval_iou_red bricks": 0.8540317233641047,
+      "eval_iou_road": 0.9470304594120126,
+      "eval_iou_rough concrete": 0.6008338641496322,
+      "eval_iou_rough red sidewalk": 0.5913070763512168,
+      "eval_iou_speedway bricks": 0.9713626391149616,
+      "eval_iou_steel": 0.8156706945456378,
+      "eval_iou_tiles": 0.24534351996430961,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08243369311094284,
+      "eval_mean_accuracy": 0.7657950224356297,
+      "eval_mean_iou": 0.7190877507912763,
+      "eval_overall_accuracy": 0.9753200724973516,
+      "eval_runtime": 5.6665,
+      "eval_samples_per_second": 20.824,
+      "eval_steps_per_second": 1.412,
+      "step": 520
+    },
+    {
+      "epoch": 30.65,
+      "learning_rate": 4.456656346749226e-05,
+      "loss": 0.0519,
+      "step": 521
+    },
+    {
+      "epoch": 30.71,
+      "learning_rate": 4.455108359133127e-05,
+      "loss": 0.0496,
+      "step": 522
+    },
+    {
+      "epoch": 30.76,
+      "learning_rate": 4.453560371517028e-05,
+      "loss": 0.0504,
+      "step": 523
+    },
+    {
+      "epoch": 30.82,
+      "learning_rate": 4.452012383900929e-05,
+      "loss": 0.0531,
+      "step": 524
+    },
+    {
+      "epoch": 30.88,
+      "learning_rate": 4.4504643962848296e-05,
+      "loss": 0.0479,
+      "step": 525
+    },
+    {
+      "epoch": 30.94,
+      "learning_rate": 4.4489164086687306e-05,
+      "loss": 0.0583,
+      "step": 526
+    },
+    {
+      "epoch": 31.0,
+      "learning_rate": 4.4473684210526316e-05,
+      "loss": 0.0534,
+      "step": 527
+    },
+    {
+      "epoch": 31.06,
+      "learning_rate": 4.4458204334365326e-05,
+      "loss": 0.0472,
+      "step": 528
+    },
+    {
+      "epoch": 31.12,
+      "learning_rate": 4.4442724458204336e-05,
+      "loss": 0.0586,
+      "step": 529
+    },
+    {
+      "epoch": 31.18,
+      "learning_rate": 4.4427244582043346e-05,
+      "loss": 0.0465,
+      "step": 530
+    },
+    {
+      "epoch": 31.24,
+      "learning_rate": 4.4411764705882356e-05,
+      "loss": 0.0502,
+      "step": 531
+    },
+    {
+      "epoch": 31.29,
+      "learning_rate": 4.4396284829721366e-05,
+      "loss": 0.0479,
+      "step": 532
+    },
+    {
+      "epoch": 31.35,
+      "learning_rate": 4.4380804953560376e-05,
+      "loss": 0.0519,
+      "step": 533
+    },
+    {
+      "epoch": 31.41,
+      "learning_rate": 4.4365325077399386e-05,
+      "loss": 0.0551,
+      "step": 534
+    },
+    {
+      "epoch": 31.47,
+      "learning_rate": 4.4349845201238396e-05,
+      "loss": 0.0585,
+      "step": 535
+    },
+    {
+      "epoch": 31.53,
+      "learning_rate": 4.4334365325077406e-05,
+      "loss": 0.0463,
+      "step": 536
+    },
+    {
+      "epoch": 31.59,
+      "learning_rate": 4.431888544891641e-05,
+      "loss": 0.0478,
+      "step": 537
+    },
+    {
+      "epoch": 31.65,
+      "learning_rate": 4.430340557275542e-05,
+      "loss": 0.0489,
+      "step": 538
+    },
+    {
+      "epoch": 31.71,
+      "learning_rate": 4.428792569659443e-05,
+      "loss": 0.0542,
+      "step": 539
+    },
+    {
+      "epoch": 31.76,
+      "learning_rate": 4.427244582043344e-05,
+      "loss": 0.0517,
+      "step": 540
+    },
+    {
+      "epoch": 31.76,
+      "eval_accuracy_NAT": 0.9909922615600154,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9372061842903858,
+      "eval_accuracy_concrete tiles": 0.893683714353052,
+      "eval_accuracy_dark bricks": 0.8919950998219858,
+      "eval_accuracy_grass": 0.8814584776330681,
+      "eval_accuracy_red bricks": 0.9331063268892794,
+      "eval_accuracy_road": 0.9784058473675463,
+      "eval_accuracy_rough concrete": 0.6244704115743748,
+      "eval_accuracy_rough red sidewalk": 0.8062024830988351,
+      "eval_accuracy_speedway bricks": 0.9904345683277433,
+      "eval_accuracy_steel": 0.8430882566083754,
+      "eval_accuracy_tiles": 0.3040079978187767,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9798501078345754,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8732045489507899,
+      "eval_iou_concrete tiles": 0.8217002188657985,
+      "eval_iou_dark bricks": 0.8340544724796298,
+      "eval_iou_grass": 0.8191193203301274,
+      "eval_iou_red bricks": 0.8353820434654342,
+      "eval_iou_road": 0.9494913348518607,
+      "eval_iou_rough concrete": 0.6023739042965943,
+      "eval_iou_rough red sidewalk": 0.5985060795194445,
+      "eval_iou_speedway bricks": 0.9701875387971075,
+      "eval_iou_steel": 0.8044703660606985,
+      "eval_iou_tiles": 0.2819136981683335,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08074591308832169,
+      "eval_mean_accuracy": 0.7750039714879567,
+      "eval_mean_iou": 0.7207887410477227,
+      "eval_overall_accuracy": 0.9755485987259169,
+      "eval_runtime": 5.8591,
+      "eval_samples_per_second": 20.14,
+      "eval_steps_per_second": 1.365,
+      "step": 540
+    },
+    {
+      "epoch": 31.82,
+      "learning_rate": 4.425696594427245e-05,
+      "loss": 0.0505,
+      "step": 541
+    },
+    {
+      "epoch": 31.88,
+      "learning_rate": 4.424148606811146e-05,
+      "loss": 0.054,
+      "step": 542
+    },
+    {
+      "epoch": 31.94,
+      "learning_rate": 4.422600619195046e-05,
+      "loss": 0.0489,
+      "step": 543
+    },
+    {
+      "epoch": 32.0,
+      "learning_rate": 4.421052631578947e-05,
+      "loss": 0.0477,
+      "step": 544
+    },
+    {
+      "epoch": 32.06,
+      "learning_rate": 4.419504643962848e-05,
+      "loss": 0.042,
+      "step": 545
+    },
+    {
+      "epoch": 32.12,
+      "learning_rate": 4.417956656346749e-05,
+      "loss": 0.0537,
+      "step": 546
+    },
+    {
+      "epoch": 32.18,
+      "learning_rate": 4.41640866873065e-05,
+      "loss": 0.0536,
+      "step": 547
+    },
+    {
+      "epoch": 32.24,
+      "learning_rate": 4.414860681114551e-05,
+      "loss": 0.0545,
+      "step": 548
+    },
+    {
+      "epoch": 32.29,
+      "learning_rate": 4.413312693498452e-05,
+      "loss": 0.0469,
+      "step": 549
+    },
+    {
+      "epoch": 32.35,
+      "learning_rate": 4.411764705882353e-05,
+      "loss": 0.0414,
+      "step": 550
+    },
+    {
+      "epoch": 32.41,
+      "learning_rate": 4.410216718266254e-05,
+      "loss": 0.0474,
+      "step": 551
+    },
+    {
+      "epoch": 32.47,
+      "learning_rate": 4.408668730650155e-05,
+      "loss": 0.0579,
+      "step": 552
+    },
+    {
+      "epoch": 32.53,
+      "learning_rate": 4.407120743034056e-05,
+      "loss": 0.0456,
+      "step": 553
+    },
+    {
+      "epoch": 32.59,
+      "learning_rate": 4.405572755417957e-05,
+      "loss": 0.0478,
+      "step": 554
+    },
+    {
+      "epoch": 32.65,
+      "learning_rate": 4.4040247678018576e-05,
+      "loss": 0.0454,
+      "step": 555
+    },
+    {
+      "epoch": 32.71,
+      "learning_rate": 4.4024767801857586e-05,
+      "loss": 0.0446,
+      "step": 556
+    },
+    {
+      "epoch": 32.76,
+      "learning_rate": 4.4009287925696596e-05,
+      "loss": 0.0398,
+      "step": 557
+    },
+    {
+      "epoch": 32.82,
+      "learning_rate": 4.3993808049535606e-05,
+      "loss": 0.0474,
+      "step": 558
+    },
+    {
+      "epoch": 32.88,
+      "learning_rate": 4.3978328173374616e-05,
+      "loss": 0.0519,
+      "step": 559
+    },
+    {
+      "epoch": 32.94,
+      "learning_rate": 4.3962848297213626e-05,
+      "loss": 0.0512,
+      "step": 560
+    },
+    {
+      "epoch": 32.94,
+      "eval_accuracy_NAT": 0.9933581251948573,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9411397794557176,
+      "eval_accuracy_concrete tiles": 0.8627386283290125,
+      "eval_accuracy_dark bricks": 0.7835116666347645,
+      "eval_accuracy_grass": 0.8619198204438187,
+      "eval_accuracy_red bricks": 0.835951230228471,
+      "eval_accuracy_road": 0.9788659796569156,
+      "eval_accuracy_rough concrete": 0.6165869180486119,
+      "eval_accuracy_rough red sidewalk": 0.7804014167650531,
+      "eval_accuracy_speedway bricks": 0.9852581410444912,
+      "eval_accuracy_steel": 0.8938186888703414,
+      "eval_accuracy_tiles": 0.12823775333999818,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9793455250994089,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.873954996019624,
+      "eval_iou_concrete tiles": 0.8109616960942866,
+      "eval_iou_dark bricks": 0.7668982056122042,
+      "eval_iou_grass": 0.8081869093206233,
+      "eval_iou_red bricks": 0.822312263641275,
+      "eval_iou_road": 0.9476417280956264,
+      "eval_iou_rough concrete": 0.6077005749590364,
+      "eval_iou_rough red sidewalk": 0.6209246528214057,
+      "eval_iou_speedway bricks": 0.9704413251282072,
+      "eval_iou_steel": 0.8201518854988787,
+      "eval_iou_tiles": 0.1261774174317396,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08563321828842163,
+      "eval_mean_accuracy": 0.743214472924004,
+      "eval_mean_iou": 0.704207475363255,
+      "eval_overall_accuracy": 0.9751662884728384,
+      "eval_runtime": 5.9098,
+      "eval_samples_per_second": 19.967,
+      "eval_steps_per_second": 1.354,
+      "step": 560
+    },
+    {
+      "epoch": 33.0,
+      "learning_rate": 4.394736842105263e-05,
+      "loss": 0.0474,
+      "step": 561
+    },
+    {
+      "epoch": 33.06,
+      "learning_rate": 4.393188854489164e-05,
+      "loss": 0.0477,
+      "step": 562
+    },
+    {
+      "epoch": 33.12,
+      "learning_rate": 4.391640866873065e-05,
+      "loss": 0.0486,
+      "step": 563
+    },
+    {
+      "epoch": 33.18,
+      "learning_rate": 4.390092879256966e-05,
+      "loss": 0.0484,
+      "step": 564
+    },
+    {
+      "epoch": 33.24,
+      "learning_rate": 4.3885448916408676e-05,
+      "loss": 0.0493,
+      "step": 565
+    },
+    {
+      "epoch": 33.29,
+      "learning_rate": 4.3869969040247686e-05,
+      "loss": 0.0467,
+      "step": 566
+    },
+    {
+      "epoch": 33.35,
+      "learning_rate": 4.385448916408669e-05,
+      "loss": 0.0537,
+      "step": 567
+    },
+    {
+      "epoch": 33.41,
+      "learning_rate": 4.38390092879257e-05,
+      "loss": 0.0462,
+      "step": 568
+    },
+    {
+      "epoch": 33.47,
+      "learning_rate": 4.382352941176471e-05,
+      "loss": 0.0494,
+      "step": 569
+    },
+    {
+      "epoch": 33.53,
+      "learning_rate": 4.380804953560372e-05,
+      "loss": 0.0471,
+      "step": 570
+    },
+    {
+      "epoch": 33.59,
+      "learning_rate": 4.379256965944273e-05,
+      "loss": 0.0422,
+      "step": 571
+    },
+    {
+      "epoch": 33.65,
+      "learning_rate": 4.377708978328174e-05,
+      "loss": 0.0595,
+      "step": 572
+    },
+    {
+      "epoch": 33.71,
+      "learning_rate": 4.376160990712074e-05,
+      "loss": 0.0455,
+      "step": 573
+    },
+    {
+      "epoch": 33.76,
+      "learning_rate": 4.374613003095975e-05,
+      "loss": 0.0476,
+      "step": 574
+    },
+    {
+      "epoch": 33.82,
+      "learning_rate": 4.373065015479876e-05,
+      "loss": 0.0349,
+      "step": 575
+    },
+    {
+      "epoch": 33.88,
+      "learning_rate": 4.371517027863777e-05,
+      "loss": 0.049,
+      "step": 576
+    },
+    {
+      "epoch": 33.94,
+      "learning_rate": 4.369969040247678e-05,
+      "loss": 0.0478,
+      "step": 577
+    },
+    {
+      "epoch": 34.0,
+      "learning_rate": 4.368421052631579e-05,
+      "loss": 0.05,
+      "step": 578
+    },
+    {
+      "epoch": 34.06,
+      "learning_rate": 4.36687306501548e-05,
+      "loss": 0.0458,
+      "step": 579
+    },
+    {
+      "epoch": 34.12,
+      "learning_rate": 4.365325077399381e-05,
+      "loss": 0.0474,
+      "step": 580
+    },
+    {
+      "epoch": 34.12,
+      "eval_accuracy_NAT": 0.9925247572232188,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9510559402179147,
+      "eval_accuracy_concrete tiles": 0.8676879566344567,
+      "eval_accuracy_dark bricks": 0.8660681813831518,
+      "eval_accuracy_grass": 0.8697148447997214,
+      "eval_accuracy_red bricks": 0.8806019332161688,
+      "eval_accuracy_road": 0.9743426815184473,
+      "eval_accuracy_rough concrete": 0.6254416961130742,
+      "eval_accuracy_rough red sidewalk": 0.7290173636438836,
+      "eval_accuracy_speedway bricks": 0.9838051358822872,
+      "eval_accuracy_steel": 0.8986790746025479,
+      "eval_accuracy_tiles": 0.21272986155290982,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979491261992737,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8723648091991095,
+      "eval_iou_concrete tiles": 0.8248605292048484,
+      "eval_iou_dark bricks": 0.8305294749304771,
+      "eval_iou_grass": 0.8163165799109914,
+      "eval_iou_red bricks": 0.8460767241834204,
+      "eval_iou_road": 0.9531798785620255,
+      "eval_iou_rough concrete": 0.5899867344185629,
+      "eval_iou_rough red sidewalk": 0.6061121392487371,
+      "eval_iou_speedway bricks": 0.9723500352882056,
+      "eval_iou_steel": 0.8223677591079971,
+      "eval_iou_tiles": 0.21011999162153266,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07961232215166092,
+      "eval_mean_accuracy": 0.7578207251375216,
+      "eval_mean_iou": 0.7172119936668189,
+      "eval_overall_accuracy": 0.9757948729951503,
+      "eval_runtime": 5.7515,
+      "eval_samples_per_second": 20.516,
+      "eval_steps_per_second": 1.391,
+      "step": 580
+    },
+    {
+      "epoch": 34.18,
+      "learning_rate": 4.363777089783282e-05,
+      "loss": 0.0518,
+      "step": 581
+    },
+    {
+      "epoch": 34.24,
+      "learning_rate": 4.362229102167183e-05,
+      "loss": 0.0598,
+      "step": 582
+    },
+    {
+      "epoch": 34.29,
+      "learning_rate": 4.360681114551084e-05,
+      "loss": 0.0466,
+      "step": 583
+    },
+    {
+      "epoch": 34.35,
+      "learning_rate": 4.359133126934985e-05,
+      "loss": 0.0496,
+      "step": 584
+    },
+    {
+      "epoch": 34.41,
+      "learning_rate": 4.3575851393188855e-05,
+      "loss": 0.0381,
+      "step": 585
+    },
+    {
+      "epoch": 34.47,
+      "learning_rate": 4.3560371517027865e-05,
+      "loss": 0.0472,
+      "step": 586
+    },
+    {
+      "epoch": 34.53,
+      "learning_rate": 4.3544891640866875e-05,
+      "loss": 0.0462,
+      "step": 587
+    },
+    {
+      "epoch": 34.59,
+      "learning_rate": 4.3529411764705885e-05,
+      "loss": 0.0465,
+      "step": 588
+    },
+    {
+      "epoch": 34.65,
+      "learning_rate": 4.3513931888544895e-05,
+      "loss": 0.0469,
+      "step": 589
+    },
+    {
+      "epoch": 34.71,
+      "learning_rate": 4.3498452012383905e-05,
+      "loss": 0.0488,
+      "step": 590
+    },
+    {
+      "epoch": 34.76,
+      "learning_rate": 4.348297213622291e-05,
+      "loss": 0.0476,
+      "step": 591
+    },
+    {
+      "epoch": 34.82,
+      "learning_rate": 4.346749226006192e-05,
+      "loss": 0.0514,
+      "step": 592
+    },
+    {
+      "epoch": 34.88,
+      "learning_rate": 4.345201238390093e-05,
+      "loss": 0.0475,
+      "step": 593
+    },
+    {
+      "epoch": 34.94,
+      "learning_rate": 4.343653250773994e-05,
+      "loss": 0.0436,
+      "step": 594
+    },
+    {
+      "epoch": 35.0,
+      "learning_rate": 4.342105263157895e-05,
+      "loss": 0.0492,
+      "step": 595
+    },
+    {
+      "epoch": 35.06,
+      "learning_rate": 4.340557275541796e-05,
+      "loss": 0.0441,
+      "step": 596
+    },
+    {
+      "epoch": 35.12,
+      "learning_rate": 4.339009287925697e-05,
+      "loss": 0.051,
+      "step": 597
+    },
+    {
+      "epoch": 35.18,
+      "learning_rate": 4.337461300309598e-05,
+      "loss": 0.0404,
+      "step": 598
+    },
+    {
+      "epoch": 35.24,
+      "learning_rate": 4.335913312693499e-05,
+      "loss": 0.0456,
+      "step": 599
+    },
+    {
+      "epoch": 35.29,
+      "learning_rate": 4.3343653250774e-05,
+      "loss": 0.0471,
+      "step": 600
+    },
+    {
+      "epoch": 35.29,
+      "eval_accuracy_NAT": 0.9921806923768611,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9352401511063141,
+      "eval_accuracy_concrete tiles": 0.9168041480084845,
+      "eval_accuracy_dark bricks": 0.8880184905154758,
+      "eval_accuracy_grass": 0.8596167020555852,
+      "eval_accuracy_red bricks": 0.9219573813708261,
+      "eval_accuracy_road": 0.9796841707545048,
+      "eval_accuracy_rough concrete": 0.6205256854110679,
+      "eval_accuracy_rough red sidewalk": 0.8089923972958136,
+      "eval_accuracy_speedway bricks": 0.9875815098367027,
+      "eval_accuracy_steel": 0.8911223617726289,
+      "eval_accuracy_tiles": 0.456420976097428,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9789283325219831,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8784533738659873,
+      "eval_iou_concrete tiles": 0.8173814376667857,
+      "eval_iou_dark bricks": 0.8203265905152598,
+      "eval_iou_grass": 0.8033537812057023,
+      "eval_iou_red bricks": 0.8567418597529856,
+      "eval_iou_road": 0.9529836330101246,
+      "eval_iou_rough concrete": 0.6047773086550244,
+      "eval_iou_rough red sidewalk": 0.5827776182711528,
+      "eval_iou_speedway bricks": 0.9728446971237983,
+      "eval_iou_steel": 0.8207736729324953,
+      "eval_iou_tiles": 0.4101377470463331,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08150813728570938,
+      "eval_mean_accuracy": 0.7890880512770533,
+      "eval_mean_iou": 0.7307292348128948,
+      "eval_overall_accuracy": 0.9756607443599378,
+      "eval_runtime": 8.5574,
+      "eval_samples_per_second": 13.789,
+      "eval_steps_per_second": 0.935,
+      "step": 600
+    },
+    {
+      "epoch": 35.35,
+      "learning_rate": 4.332817337461301e-05,
+      "loss": 0.0514,
+      "step": 601
+    },
+    {
+      "epoch": 35.41,
+      "learning_rate": 4.331269349845202e-05,
+      "loss": 0.052,
+      "step": 602
+    },
+    {
+      "epoch": 35.47,
+      "learning_rate": 4.329721362229102e-05,
+      "loss": 0.0427,
+      "step": 603
+    },
+    {
+      "epoch": 35.53,
+      "learning_rate": 4.328173374613003e-05,
+      "loss": 0.0551,
+      "step": 604
+    },
+    {
+      "epoch": 35.59,
+      "learning_rate": 4.326625386996904e-05,
+      "loss": 0.0451,
+      "step": 605
+    },
+    {
+      "epoch": 35.65,
+      "learning_rate": 4.325077399380805e-05,
+      "loss": 0.0483,
+      "step": 606
+    },
+    {
+      "epoch": 35.71,
+      "learning_rate": 4.323529411764706e-05,
+      "loss": 0.0421,
+      "step": 607
+    },
+    {
+      "epoch": 35.76,
+      "learning_rate": 4.321981424148607e-05,
+      "loss": 0.0522,
+      "step": 608
+    },
+    {
+      "epoch": 35.82,
+      "learning_rate": 4.3204334365325075e-05,
+      "loss": 0.0513,
+      "step": 609
+    },
+    {
+      "epoch": 35.88,
+      "learning_rate": 4.3188854489164085e-05,
+      "loss": 0.0469,
+      "step": 610
+    },
+    {
+      "epoch": 35.94,
+      "learning_rate": 4.3173374613003095e-05,
+      "loss": 0.0493,
+      "step": 611
+    },
+    {
+      "epoch": 36.0,
+      "learning_rate": 4.3157894736842105e-05,
+      "loss": 0.0591,
+      "step": 612
+    },
+    {
+      "epoch": 36.06,
+      "learning_rate": 4.314241486068112e-05,
+      "loss": 0.0483,
+      "step": 613
+    },
+    {
+      "epoch": 36.12,
+      "learning_rate": 4.3126934984520125e-05,
+      "loss": 0.045,
+      "step": 614
+    },
+    {
+      "epoch": 36.18,
+      "learning_rate": 4.3111455108359135e-05,
+      "loss": 0.0452,
+      "step": 615
+    },
+    {
+      "epoch": 36.24,
+      "learning_rate": 4.3095975232198145e-05,
+      "loss": 0.0451,
+      "step": 616
+    },
+    {
+      "epoch": 36.29,
+      "learning_rate": 4.3080495356037155e-05,
+      "loss": 0.0451,
+      "step": 617
+    },
+    {
+      "epoch": 36.35,
+      "learning_rate": 4.3065015479876165e-05,
+      "loss": 0.0414,
+      "step": 618
+    },
+    {
+      "epoch": 36.41,
+      "learning_rate": 4.3049535603715175e-05,
+      "loss": 0.05,
+      "step": 619
+    },
+    {
+      "epoch": 36.47,
+      "learning_rate": 4.303405572755418e-05,
+      "loss": 0.0522,
+      "step": 620
+    },
+    {
+      "epoch": 36.47,
+      "eval_accuracy_NAT": 0.9943427156506753,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9352936590079024,
+      "eval_accuracy_concrete tiles": 0.8649776101814753,
+      "eval_accuracy_dark bricks": 0.8185402829087151,
+      "eval_accuracy_grass": 0.8543376973358011,
+      "eval_accuracy_red bricks": 0.8782952548330404,
+      "eval_accuracy_road": 0.9782813284088668,
+      "eval_accuracy_rough concrete": 0.6238447374850286,
+      "eval_accuracy_rough red sidewalk": 0.7478699644304739,
+      "eval_accuracy_speedway bricks": 0.9850266267180213,
+      "eval_accuracy_steel": 0.9007806604127204,
+      "eval_accuracy_tiles": 0.10754642673210336,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9785511215548038,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8728486742140519,
+      "eval_iou_concrete tiles": 0.8187070580886946,
+      "eval_iou_dark bricks": 0.7977836751255778,
+      "eval_iou_grass": 0.8036221777391295,
+      "eval_iou_red bricks": 0.8518617162946786,
+      "eval_iou_road": 0.9475414435277907,
+      "eval_iou_rough concrete": 0.6092469739292365,
+      "eval_iou_rough red sidewalk": 0.603051305840049,
+      "eval_iou_speedway bricks": 0.9727465494249605,
+      "eval_iou_steel": 0.8254196148473816,
+      "eval_iou_tiles": 0.10750734381151389,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08459129929542542,
+      "eval_mean_accuracy": 0.7453182280080634,
+      "eval_mean_iou": 0.7068375118767591,
+      "eval_overall_accuracy": 0.9751562668105304,
+      "eval_runtime": 5.4145,
+      "eval_samples_per_second": 21.793,
+      "eval_steps_per_second": 1.478,
+      "step": 620
+    },
+    {
+      "epoch": 36.53,
+      "learning_rate": 4.301857585139319e-05,
+      "loss": 0.0406,
+      "step": 621
+    },
+    {
+      "epoch": 36.59,
+      "learning_rate": 4.30030959752322e-05,
+      "loss": 0.0507,
+      "step": 622
+    },
+    {
+      "epoch": 36.65,
+      "learning_rate": 4.298761609907121e-05,
+      "loss": 0.0402,
+      "step": 623
+    },
+    {
+      "epoch": 36.71,
+      "learning_rate": 4.297213622291022e-05,
+      "loss": 0.0433,
+      "step": 624
+    },
+    {
+      "epoch": 36.76,
+      "learning_rate": 4.295665634674923e-05,
+      "loss": 0.0478,
+      "step": 625
+    },
+    {
+      "epoch": 36.82,
+      "learning_rate": 4.294117647058823e-05,
+      "loss": 0.0557,
+      "step": 626
+    },
+    {
+      "epoch": 36.88,
+      "learning_rate": 4.292569659442724e-05,
+      "loss": 0.0417,
+      "step": 627
+    },
+    {
+      "epoch": 36.94,
+      "learning_rate": 4.291021671826626e-05,
+      "loss": 0.041,
+      "step": 628
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 4.289473684210527e-05,
+      "loss": 0.0399,
+      "step": 629
+    },
+    {
+      "epoch": 37.06,
+      "learning_rate": 4.287925696594428e-05,
+      "loss": 0.0414,
+      "step": 630
+    },
+    {
+      "epoch": 37.12,
+      "learning_rate": 4.286377708978329e-05,
+      "loss": 0.0394,
+      "step": 631
+    },
+    {
+      "epoch": 37.18,
+      "learning_rate": 4.284829721362229e-05,
+      "loss": 0.0432,
+      "step": 632
+    },
+    {
+      "epoch": 37.24,
+      "learning_rate": 4.28328173374613e-05,
+      "loss": 0.0461,
+      "step": 633
+    },
+    {
+      "epoch": 37.29,
+      "learning_rate": 4.281733746130031e-05,
+      "loss": 0.0482,
+      "step": 634
+    },
+    {
+      "epoch": 37.35,
+      "learning_rate": 4.280185758513932e-05,
+      "loss": 0.0425,
+      "step": 635
+    },
+    {
+      "epoch": 37.41,
+      "learning_rate": 4.278637770897833e-05,
+      "loss": 0.0429,
+      "step": 636
+    },
+    {
+      "epoch": 37.47,
+      "learning_rate": 4.277089783281734e-05,
+      "loss": 0.0533,
+      "step": 637
+    },
+    {
+      "epoch": 37.53,
+      "learning_rate": 4.2755417956656345e-05,
+      "loss": 0.0495,
+      "step": 638
+    },
+    {
+      "epoch": 37.59,
+      "learning_rate": 4.2739938080495355e-05,
+      "loss": 0.0452,
+      "step": 639
+    },
+    {
+      "epoch": 37.65,
+      "learning_rate": 4.2724458204334365e-05,
+      "loss": 0.043,
+      "step": 640
+    },
+    {
+      "epoch": 37.65,
+      "eval_accuracy_NAT": 0.993806239736212,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9391362907405341,
+      "eval_accuracy_concrete tiles": 0.91006363422107,
+      "eval_accuracy_dark bricks": 0.8166548628524396,
+      "eval_accuracy_grass": 0.8672234270855806,
+      "eval_accuracy_red bricks": 0.9003734622144113,
+      "eval_accuracy_road": 0.980038488649803,
+      "eval_accuracy_rough concrete": 0.6178918954349627,
+      "eval_accuracy_rough red sidewalk": 0.7609171372923544,
+      "eval_accuracy_speedway bricks": 0.9853395394938421,
+      "eval_accuracy_steel": 0.9130003448499903,
+      "eval_accuracy_tiles": 0.17068072343906207,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9796457416816873,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8762803509415095,
+      "eval_iou_concrete tiles": 0.8324853397723353,
+      "eval_iou_dark bricks": 0.7964735586907923,
+      "eval_iou_grass": 0.8144555358175916,
+      "eval_iou_red bricks": 0.8581898131183584,
+      "eval_iou_road": 0.9519905976698708,
+      "eval_iou_rough concrete": 0.6100795443848254,
+      "eval_iou_rough red sidewalk": 0.6210778296096243,
+      "eval_iou_speedway bricks": 0.9727566141361427,
+      "eval_iou_steel": 0.8250808462205521,
+      "eval_iou_tiles": 0.16930612735523035,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08132755756378174,
+      "eval_mean_accuracy": 0.7580866189238662,
+      "eval_mean_iou": 0.7159862999537322,
+      "eval_overall_accuracy": 0.9760765140339479,
+      "eval_runtime": 8.4358,
+      "eval_samples_per_second": 13.988,
+      "eval_steps_per_second": 0.948,
+      "step": 640
+    },
+    {
+      "epoch": 37.71,
+      "learning_rate": 4.2708978328173375e-05,
+      "loss": 0.0445,
+      "step": 641
+    },
+    {
+      "epoch": 37.76,
+      "learning_rate": 4.2693498452012385e-05,
+      "loss": 0.0405,
+      "step": 642
+    },
+    {
+      "epoch": 37.82,
+      "learning_rate": 4.2678018575851395e-05,
+      "loss": 0.0485,
+      "step": 643
+    },
+    {
+      "epoch": 37.88,
+      "learning_rate": 4.2662538699690405e-05,
+      "loss": 0.0433,
+      "step": 644
+    },
+    {
+      "epoch": 37.94,
+      "learning_rate": 4.2647058823529415e-05,
+      "loss": 0.0396,
+      "step": 645
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 4.2631578947368425e-05,
+      "loss": 0.0477,
+      "step": 646
+    },
+    {
+      "epoch": 38.06,
+      "learning_rate": 4.2616099071207435e-05,
+      "loss": 0.0409,
+      "step": 647
+    },
+    {
+      "epoch": 38.12,
+      "learning_rate": 4.2600619195046445e-05,
+      "loss": 0.0468,
+      "step": 648
+    },
+    {
+      "epoch": 38.18,
+      "learning_rate": 4.2585139318885455e-05,
+      "loss": 0.0399,
+      "step": 649
+    },
+    {
+      "epoch": 38.24,
+      "learning_rate": 4.256965944272446e-05,
+      "loss": 0.0434,
+      "step": 650
+    },
+    {
+      "epoch": 38.29,
+      "learning_rate": 4.255417956656347e-05,
+      "loss": 0.0477,
+      "step": 651
+    },
+    {
+      "epoch": 38.35,
+      "learning_rate": 4.253869969040248e-05,
+      "loss": 0.0409,
+      "step": 652
+    },
+    {
+      "epoch": 38.41,
+      "learning_rate": 4.252321981424149e-05,
+      "loss": 0.0421,
+      "step": 653
+    },
+    {
+      "epoch": 38.47,
+      "learning_rate": 4.25077399380805e-05,
+      "loss": 0.0427,
+      "step": 654
+    },
+    {
+      "epoch": 38.53,
+      "learning_rate": 4.249226006191951e-05,
+      "loss": 0.0521,
+      "step": 655
+    },
+    {
+      "epoch": 38.59,
+      "learning_rate": 4.247678018575851e-05,
+      "loss": 0.0439,
+      "step": 656
+    },
+    {
+      "epoch": 38.65,
+      "learning_rate": 4.246130030959752e-05,
+      "loss": 0.0389,
+      "step": 657
+    },
+    {
+      "epoch": 38.71,
+      "learning_rate": 4.244582043343653e-05,
+      "loss": 0.0415,
+      "step": 658
+    },
+    {
+      "epoch": 38.76,
+      "learning_rate": 4.243034055727554e-05,
+      "loss": 0.0498,
+      "step": 659
+    },
+    {
+      "epoch": 38.82,
+      "learning_rate": 4.241486068111455e-05,
+      "loss": 0.0429,
+      "step": 660
+    },
+    {
+      "epoch": 38.82,
+      "eval_accuracy_NAT": 0.993064360019708,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9416037694023474,
+      "eval_accuracy_concrete tiles": 0.8665095451331605,
+      "eval_accuracy_dark bricks": 0.8162672511149819,
+      "eval_accuracy_grass": 0.8894030059763746,
+      "eval_accuracy_red bricks": 0.9094903339191565,
+      "eval_accuracy_road": 0.9792309858748472,
+      "eval_accuracy_rough concrete": 0.6195305656689648,
+      "eval_accuracy_rough red sidewalk": 0.7703095977560367,
+      "eval_accuracy_speedway bricks": 0.9875611173830758,
+      "eval_accuracy_steel": 0.8956428953405269,
+      "eval_accuracy_tiles": 0.19309885182828926,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9804051300764475,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8790174711985862,
+      "eval_iou_concrete tiles": 0.8424067454862065,
+      "eval_iou_dark bricks": 0.7969845067000579,
+      "eval_iou_grass": 0.8264285946653607,
+      "eval_iou_red bricks": 0.8588766142834915,
+      "eval_iou_road": 0.954441135075388,
+      "eval_iou_rough concrete": 0.6086180251481021,
+      "eval_iou_rough red sidewalk": 0.6179531264138992,
+      "eval_iou_speedway bricks": 0.9729326781278663,
+      "eval_iou_steel": 0.8281644253431305,
+      "eval_iou_tiles": 0.19039369137941334,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07921610027551651,
+      "eval_mean_accuracy": 0.7585932522628822,
+      "eval_mean_iou": 0.719740164915227,
+      "eval_overall_accuracy": 0.9767600237312963,
+      "eval_runtime": 5.7289,
+      "eval_samples_per_second": 20.597,
+      "eval_steps_per_second": 1.396,
+      "step": 660
+    },
+    {
+      "epoch": 38.88,
+      "learning_rate": 4.239938080495356e-05,
+      "loss": 0.0436,
+      "step": 661
+    },
+    {
+      "epoch": 38.94,
+      "learning_rate": 4.238390092879257e-05,
+      "loss": 0.0458,
+      "step": 662
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 4.236842105263158e-05,
+      "loss": 0.0416,
+      "step": 663
+    },
+    {
+      "epoch": 39.06,
+      "learning_rate": 4.235294117647059e-05,
+      "loss": 0.0426,
+      "step": 664
+    },
+    {
+      "epoch": 39.12,
+      "learning_rate": 4.23374613003096e-05,
+      "loss": 0.0467,
+      "step": 665
+    },
+    {
+      "epoch": 39.18,
+      "learning_rate": 4.232198142414861e-05,
+      "loss": 0.0417,
+      "step": 666
+    },
+    {
+      "epoch": 39.24,
+      "learning_rate": 4.230650154798762e-05,
+      "loss": 0.0422,
+      "step": 667
+    },
+    {
+      "epoch": 39.29,
+      "learning_rate": 4.2291021671826625e-05,
+      "loss": 0.0403,
+      "step": 668
+    },
+    {
+      "epoch": 39.35,
+      "learning_rate": 4.2275541795665635e-05,
+      "loss": 0.0423,
+      "step": 669
+    },
+    {
+      "epoch": 39.41,
+      "learning_rate": 4.2260061919504645e-05,
+      "loss": 0.0476,
+      "step": 670
+    },
+    {
+      "epoch": 39.47,
+      "learning_rate": 4.2244582043343655e-05,
+      "loss": 0.0569,
+      "step": 671
+    },
+    {
+      "epoch": 39.53,
+      "learning_rate": 4.2229102167182665e-05,
+      "loss": 0.0468,
+      "step": 672
+    },
+    {
+      "epoch": 39.59,
+      "learning_rate": 4.2213622291021675e-05,
+      "loss": 0.0447,
+      "step": 673
+    },
+    {
+      "epoch": 39.65,
+      "learning_rate": 4.219814241486068e-05,
+      "loss": 0.0396,
+      "step": 674
+    },
+    {
+      "epoch": 39.71,
+      "learning_rate": 4.218266253869969e-05,
+      "loss": 0.0423,
+      "step": 675
+    },
+    {
+      "epoch": 39.76,
+      "learning_rate": 4.2167182662538705e-05,
+      "loss": 0.0372,
+      "step": 676
+    },
+    {
+      "epoch": 39.82,
+      "learning_rate": 4.2151702786377715e-05,
+      "loss": 0.042,
+      "step": 677
+    },
+    {
+      "epoch": 39.88,
+      "learning_rate": 4.2136222910216725e-05,
+      "loss": 0.039,
+      "step": 678
+    },
+    {
+      "epoch": 39.94,
+      "learning_rate": 4.2120743034055735e-05,
+      "loss": 0.0381,
+      "step": 679
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 4.210526315789474e-05,
+      "loss": 0.0506,
+      "step": 680
+    },
+    {
+      "epoch": 40.0,
+      "eval_accuracy_NAT": 0.9915816788218814,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9473501358336301,
+      "eval_accuracy_concrete tiles": 0.8905962762196559,
+      "eval_accuracy_dark bricks": 0.8488218517313324,
+      "eval_accuracy_grass": 0.8902943525341872,
+      "eval_accuracy_red bricks": 0.8935632688927944,
+      "eval_accuracy_road": 0.9795794251926971,
+      "eval_accuracy_rough concrete": 0.6164141128239353,
+      "eval_accuracy_rough red sidewalk": 0.7506824385805277,
+      "eval_accuracy_speedway bricks": 0.9894197438707825,
+      "eval_accuracy_steel": 0.8787777316867162,
+      "eval_accuracy_tiles": 0.21100305977157746,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807303099744141,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8769597034864961,
+      "eval_iou_concrete tiles": 0.8471126255380201,
+      "eval_iou_dark bricks": 0.819791839979295,
+      "eval_iou_grass": 0.8269456409679492,
+      "eval_iou_red bricks": 0.8523233275708523,
+      "eval_iou_road": 0.9515856732881748,
+      "eval_iou_rough concrete": 0.6061064140197923,
+      "eval_iou_rough red sidewalk": 0.614039404321804,
+      "eval_iou_speedway bricks": 0.9726276228723906,
+      "eval_iou_steel": 0.8268979803941394,
+      "eval_iou_tiles": 0.2069160155669766,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0794433206319809,
+      "eval_mean_accuracy": 0.7606218519969014,
+      "eval_mean_iou": 0.7216951198446389,
+      "eval_overall_accuracy": 0.9767023830090539,
+      "eval_runtime": 5.9542,
+      "eval_samples_per_second": 19.818,
+      "eval_steps_per_second": 1.344,
+      "step": 680
+    },
+    {
+      "epoch": 40.06,
+      "learning_rate": 4.208978328173375e-05,
+      "loss": 0.0406,
+      "step": 681
+    },
+    {
+      "epoch": 40.12,
+      "learning_rate": 4.207430340557276e-05,
+      "loss": 0.0418,
+      "step": 682
+    },
+    {
+      "epoch": 40.18,
+      "learning_rate": 4.205882352941177e-05,
+      "loss": 0.0452,
+      "step": 683
+    },
+    {
+      "epoch": 40.24,
+      "learning_rate": 4.204334365325078e-05,
+      "loss": 0.0494,
+      "step": 684
+    },
+    {
+      "epoch": 40.29,
+      "learning_rate": 4.202786377708979e-05,
+      "loss": 0.0416,
+      "step": 685
+    },
+    {
+      "epoch": 40.35,
+      "learning_rate": 4.201238390092879e-05,
+      "loss": 0.0399,
+      "step": 686
+    },
+    {
+      "epoch": 40.41,
+      "learning_rate": 4.19969040247678e-05,
+      "loss": 0.0435,
+      "step": 687
+    },
+    {
+      "epoch": 40.47,
+      "learning_rate": 4.198142414860681e-05,
+      "loss": 0.0427,
+      "step": 688
+    },
+    {
+      "epoch": 40.53,
+      "learning_rate": 4.196594427244582e-05,
+      "loss": 0.0414,
+      "step": 689
+    },
+    {
+      "epoch": 40.59,
+      "learning_rate": 4.195046439628483e-05,
+      "loss": 0.0405,
+      "step": 690
+    },
+    {
+      "epoch": 40.65,
+      "learning_rate": 4.193498452012384e-05,
+      "loss": 0.0402,
+      "step": 691
+    },
+    {
+      "epoch": 40.71,
+      "learning_rate": 4.191950464396285e-05,
+      "loss": 0.0497,
+      "step": 692
+    },
+    {
+      "epoch": 40.76,
+      "learning_rate": 4.190402476780186e-05,
+      "loss": 0.039,
+      "step": 693
+    },
+    {
+      "epoch": 40.82,
+      "learning_rate": 4.188854489164087e-05,
+      "loss": 0.0387,
+      "step": 694
+    },
+    {
+      "epoch": 40.88,
+      "learning_rate": 4.187306501547988e-05,
+      "loss": 0.037,
+      "step": 695
+    },
+    {
+      "epoch": 40.94,
+      "learning_rate": 4.185758513931889e-05,
+      "loss": 0.034,
+      "step": 696
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 4.18421052631579e-05,
+      "loss": 0.0512,
+      "step": 697
+    },
+    {
+      "epoch": 41.06,
+      "learning_rate": 4.1826625386996904e-05,
+      "loss": 0.0444,
+      "step": 698
+    },
+    {
+      "epoch": 41.12,
+      "learning_rate": 4.1811145510835914e-05,
+      "loss": 0.0391,
+      "step": 699
+    },
+    {
+      "epoch": 41.18,
+      "learning_rate": 4.1795665634674924e-05,
+      "loss": 0.0426,
+      "step": 700
+    },
+    {
+      "epoch": 41.18,
+      "eval_accuracy_NAT": 0.992102682061493,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.951264621034109,
+      "eval_accuracy_concrete tiles": 0.9061984444968183,
+      "eval_accuracy_dark bricks": 0.8437015485328178,
+      "eval_accuracy_grass": 0.8701397010677234,
+      "eval_accuracy_red bricks": 0.9255821616871704,
+      "eval_accuracy_road": 0.9787628373435029,
+      "eval_accuracy_rough concrete": 0.6143285325261144,
+      "eval_accuracy_rough red sidewalk": 0.7620000150399687,
+      "eval_accuracy_speedway bricks": 0.9887498774739131,
+      "eval_accuracy_steel": 0.8858521543128739,
+      "eval_accuracy_tiles": 0.214062831349026,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9804335180753472,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8763075924244403,
+      "eval_iou_concrete tiles": 0.8416881922858018,
+      "eval_iou_dark bricks": 0.8206457738907016,
+      "eval_iou_grass": 0.8174202747051524,
+      "eval_iou_red bricks": 0.85792099368764,
+      "eval_iou_road": 0.9523834781568671,
+      "eval_iou_rough concrete": 0.6066004930658931,
+      "eval_iou_rough red sidewalk": 0.6145234456492734,
+      "eval_iou_speedway bricks": 0.9741434647027937,
+      "eval_iou_steel": 0.831835460859771,
+      "eval_iou_tiles": 0.21037275217339527,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08128701895475388,
+      "eval_mean_accuracy": 0.7640573389942715,
+      "eval_mean_iou": 0.7218673415136212,
+      "eval_overall_accuracy": 0.9766193325236693,
+      "eval_runtime": 5.6917,
+      "eval_samples_per_second": 20.732,
+      "eval_steps_per_second": 1.406,
+      "step": 700
+    },
+    {
+      "epoch": 41.24,
+      "learning_rate": 4.1780185758513934e-05,
+      "loss": 0.0396,
+      "step": 701
+    },
+    {
+      "epoch": 41.29,
+      "learning_rate": 4.1764705882352944e-05,
+      "loss": 0.0365,
+      "step": 702
+    },
+    {
+      "epoch": 41.35,
+      "learning_rate": 4.1749226006191954e-05,
+      "loss": 0.0408,
+      "step": 703
+    },
+    {
+      "epoch": 41.41,
+      "learning_rate": 4.173374613003096e-05,
+      "loss": 0.0419,
+      "step": 704
+    },
+    {
+      "epoch": 41.47,
+      "learning_rate": 4.171826625386997e-05,
+      "loss": 0.0479,
+      "step": 705
+    },
+    {
+      "epoch": 41.53,
+      "learning_rate": 4.170278637770898e-05,
+      "loss": 0.036,
+      "step": 706
+    },
+    {
+      "epoch": 41.59,
+      "learning_rate": 4.168730650154799e-05,
+      "loss": 0.0452,
+      "step": 707
+    },
+    {
+      "epoch": 41.65,
+      "learning_rate": 4.1671826625387e-05,
+      "loss": 0.0507,
+      "step": 708
+    },
+    {
+      "epoch": 41.71,
+      "learning_rate": 4.165634674922601e-05,
+      "loss": 0.0429,
+      "step": 709
+    },
+    {
+      "epoch": 41.76,
+      "learning_rate": 4.164086687306502e-05,
+      "loss": 0.0424,
+      "step": 710
+    },
+    {
+      "epoch": 41.82,
+      "learning_rate": 4.162538699690403e-05,
+      "loss": 0.0419,
+      "step": 711
+    },
+    {
+      "epoch": 41.88,
+      "learning_rate": 4.160990712074304e-05,
+      "loss": 0.0385,
+      "step": 712
+    },
+    {
+      "epoch": 41.94,
+      "learning_rate": 4.159442724458205e-05,
+      "loss": 0.0481,
+      "step": 713
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 4.157894736842106e-05,
+      "loss": 0.0431,
+      "step": 714
+    },
+    {
+      "epoch": 42.06,
+      "learning_rate": 4.156346749226007e-05,
+      "loss": 0.0409,
+      "step": 715
+    },
+    {
+      "epoch": 42.12,
+      "learning_rate": 4.154798761609907e-05,
+      "loss": 0.0368,
+      "step": 716
+    },
+    {
+      "epoch": 42.18,
+      "learning_rate": 4.153250773993808e-05,
+      "loss": 0.0384,
+      "step": 717
+    },
+    {
+      "epoch": 42.24,
+      "learning_rate": 4.151702786377709e-05,
+      "loss": 0.0476,
+      "step": 718
+    },
+    {
+      "epoch": 42.29,
+      "learning_rate": 4.15015479876161e-05,
+      "loss": 0.0493,
+      "step": 719
+    },
+    {
+      "epoch": 42.35,
+      "learning_rate": 4.148606811145511e-05,
+      "loss": 0.043,
+      "step": 720
+    },
+    {
+      "epoch": 42.35,
+      "eval_accuracy_NAT": 0.9923592481228869,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9485104928995015,
+      "eval_accuracy_concrete tiles": 0.8466650954513316,
+      "eval_accuracy_dark bricks": 0.8488074957410562,
+      "eval_accuracy_grass": 0.8917695216250894,
+      "eval_accuracy_red bricks": 0.914213532513181,
+      "eval_accuracy_road": 0.9799428281622338,
+      "eval_accuracy_rough concrete": 0.6664680399716361,
+      "eval_accuracy_rough red sidewalk": 0.7487197226629768,
+      "eval_accuracy_speedway bricks": 0.9883259200766619,
+      "eval_accuracy_steel": 0.8803970272931275,
+      "eval_accuracy_tiles": 0.1910691023660214,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809562210409813,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8793828877096302,
+      "eval_iou_concrete tiles": 0.8212138530117727,
+      "eval_iou_dark bricks": 0.8281021676307324,
+      "eval_iou_grass": 0.8290248008909109,
+      "eval_iou_red bricks": 0.8632921896068873,
+      "eval_iou_road": 0.9535515005278239,
+      "eval_iou_rough concrete": 0.6520834183569358,
+      "eval_iou_rough red sidewalk": 0.6152192047455742,
+      "eval_iou_speedway bricks": 0.9744320413766746,
+      "eval_iou_steel": 0.8306992207302265,
+      "eval_iou_tiles": 0.1892118921189212,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07583378255367279,
+      "eval_mean_accuracy": 0.7613267712989003,
+      "eval_mean_iou": 0.724397645980544,
+      "eval_overall_accuracy": 0.9773474224543167,
+      "eval_runtime": 8.5023,
+      "eval_samples_per_second": 13.879,
+      "eval_steps_per_second": 0.941,
+      "step": 720
+    },
+    {
+      "epoch": 42.41,
+      "learning_rate": 4.147058823529412e-05,
+      "loss": 0.0457,
+      "step": 721
+    },
+    {
+      "epoch": 42.47,
+      "learning_rate": 4.1455108359133124e-05,
+      "loss": 0.0415,
+      "step": 722
+    },
+    {
+      "epoch": 42.53,
+      "learning_rate": 4.1439628482972134e-05,
+      "loss": 0.0349,
+      "step": 723
+    },
+    {
+      "epoch": 42.59,
+      "learning_rate": 4.1424148606811144e-05,
+      "loss": 0.0385,
+      "step": 724
+    },
+    {
+      "epoch": 42.65,
+      "learning_rate": 4.140866873065016e-05,
+      "loss": 0.0425,
+      "step": 725
+    },
+    {
+      "epoch": 42.71,
+      "learning_rate": 4.139318885448917e-05,
+      "loss": 0.0403,
+      "step": 726
+    },
+    {
+      "epoch": 42.76,
+      "learning_rate": 4.137770897832818e-05,
+      "loss": 0.0443,
+      "step": 727
+    },
+    {
+      "epoch": 42.82,
+      "learning_rate": 4.1362229102167184e-05,
+      "loss": 0.0391,
+      "step": 728
+    },
+    {
+      "epoch": 42.88,
+      "learning_rate": 4.1346749226006194e-05,
+      "loss": 0.0367,
+      "step": 729
+    },
+    {
+      "epoch": 42.94,
+      "learning_rate": 4.1331269349845204e-05,
+      "loss": 0.039,
+      "step": 730
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 4.1315789473684214e-05,
+      "loss": 0.043,
+      "step": 731
+    },
+    {
+      "epoch": 43.06,
+      "learning_rate": 4.1300309597523224e-05,
+      "loss": 0.0381,
+      "step": 732
+    },
+    {
+      "epoch": 43.12,
+      "learning_rate": 4.1284829721362234e-05,
+      "loss": 0.0398,
+      "step": 733
+    },
+    {
+      "epoch": 43.18,
+      "learning_rate": 4.126934984520124e-05,
+      "loss": 0.0358,
+      "step": 734
+    },
+    {
+      "epoch": 43.24,
+      "learning_rate": 4.125386996904025e-05,
+      "loss": 0.0442,
+      "step": 735
+    },
+    {
+      "epoch": 43.29,
+      "learning_rate": 4.123839009287926e-05,
+      "loss": 0.0393,
+      "step": 736
+    },
+    {
+      "epoch": 43.35,
+      "learning_rate": 4.122291021671827e-05,
+      "loss": 0.0439,
+      "step": 737
+    },
+    {
+      "epoch": 43.41,
+      "learning_rate": 4.120743034055728e-05,
+      "loss": 0.0387,
+      "step": 738
+    },
+    {
+      "epoch": 43.47,
+      "learning_rate": 4.119195046439629e-05,
+      "loss": 0.0426,
+      "step": 739
+    },
+    {
+      "epoch": 43.53,
+      "learning_rate": 4.11764705882353e-05,
+      "loss": 0.0369,
+      "step": 740
+    },
+    {
+      "epoch": 43.53,
+      "eval_accuracy_NAT": 0.9927773877262009,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9525602766511392,
+      "eval_accuracy_concrete tiles": 0.8837379212821117,
+      "eval_accuracy_dark bricks": 0.8390884903240625,
+      "eval_accuracy_grass": 0.876139731347012,
+      "eval_accuracy_red bricks": 0.9155316344463972,
+      "eval_accuracy_road": 0.9744639939803367,
+      "eval_accuracy_rough concrete": 0.617081498519238,
+      "eval_accuracy_rough red sidewalk": 0.7683694417915611,
+      "eval_accuracy_speedway bricks": 0.9881799169632998,
+      "eval_accuracy_steel": 0.8889408107473249,
+      "eval_accuracy_tiles": 0.224302462964646,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808259955460622,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8761423222027079,
+      "eval_iou_concrete tiles": 0.8354388075663391,
+      "eval_iou_dark bricks": 0.8188763835053472,
+      "eval_iou_grass": 0.8242899301614433,
+      "eval_iou_red bricks": 0.8647611142812678,
+      "eval_iou_road": 0.9536246163887512,
+      "eval_iou_rough concrete": 0.608209459261043,
+      "eval_iou_rough red sidewalk": 0.6258544652701212,
+      "eval_iou_speedway bricks": 0.9747296153357259,
+      "eval_iou_steel": 0.8328401189333459,
+      "eval_iou_tiles": 0.21711973255916248,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08001357316970825,
+      "eval_mean_accuracy": 0.7631671974417947,
+      "eval_mean_iou": 0.724054812385486,
+      "eval_overall_accuracy": 0.9769962440102787,
+      "eval_runtime": 5.4467,
+      "eval_samples_per_second": 21.664,
+      "eval_steps_per_second": 1.469,
+      "step": 740
+    },
+    {
+      "epoch": 43.59,
+      "learning_rate": 4.116099071207431e-05,
+      "loss": 0.0394,
+      "step": 741
+    },
+    {
+      "epoch": 43.65,
+      "learning_rate": 4.114551083591332e-05,
+      "loss": 0.0407,
+      "step": 742
+    },
+    {
+      "epoch": 43.71,
+      "learning_rate": 4.113003095975233e-05,
+      "loss": 0.0399,
+      "step": 743
+    },
+    {
+      "epoch": 43.76,
+      "learning_rate": 4.111455108359134e-05,
+      "loss": 0.0389,
+      "step": 744
+    },
+    {
+      "epoch": 43.82,
+      "learning_rate": 4.109907120743035e-05,
+      "loss": 0.0447,
+      "step": 745
+    },
+    {
+      "epoch": 43.88,
+      "learning_rate": 4.108359133126935e-05,
+      "loss": 0.0367,
+      "step": 746
+    },
+    {
+      "epoch": 43.94,
+      "learning_rate": 4.106811145510836e-05,
+      "loss": 0.0384,
+      "step": 747
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 4.105263157894737e-05,
+      "loss": 0.0375,
+      "step": 748
+    },
+    {
+      "epoch": 44.06,
+      "learning_rate": 4.103715170278638e-05,
+      "loss": 0.0339,
+      "step": 749
+    },
+    {
+      "epoch": 44.12,
+      "learning_rate": 4.102167182662539e-05,
+      "loss": 0.0487,
+      "step": 750
+    },
+    {
+      "epoch": 44.18,
+      "learning_rate": 4.10061919504644e-05,
+      "loss": 0.0603,
+      "step": 751
+    },
+    {
+      "epoch": 44.24,
+      "learning_rate": 4.0990712074303404e-05,
+      "loss": 0.0491,
+      "step": 752
+    },
+    {
+      "epoch": 44.29,
+      "learning_rate": 4.0975232198142414e-05,
+      "loss": 0.0369,
+      "step": 753
+    },
+    {
+      "epoch": 44.35,
+      "learning_rate": 4.0959752321981424e-05,
+      "loss": 0.0374,
+      "step": 754
+    },
+    {
+      "epoch": 44.41,
+      "learning_rate": 4.0944272445820434e-05,
+      "loss": 0.0528,
+      "step": 755
+    },
+    {
+      "epoch": 44.47,
+      "learning_rate": 4.0928792569659444e-05,
+      "loss": 0.0449,
+      "step": 756
+    },
+    {
+      "epoch": 44.53,
+      "learning_rate": 4.0913312693498454e-05,
+      "loss": 0.0336,
+      "step": 757
+    },
+    {
+      "epoch": 44.59,
+      "learning_rate": 4.0897832817337464e-05,
+      "loss": 0.0357,
+      "step": 758
+    },
+    {
+      "epoch": 44.65,
+      "learning_rate": 4.0882352941176474e-05,
+      "loss": 0.0454,
+      "step": 759
+    },
+    {
+      "epoch": 44.71,
+      "learning_rate": 4.0866873065015484e-05,
+      "loss": 0.0373,
+      "step": 760
+    },
+    {
+      "epoch": 44.71,
+      "eval_accuracy_NAT": 0.9927799215788771,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9443819760926696,
+      "eval_accuracy_concrete tiles": 0.8854819703040302,
+      "eval_accuracy_dark bricks": 0.8560955534712784,
+      "eval_accuracy_grass": 0.8600633215622598,
+      "eval_accuracy_red bricks": 0.8865883128295254,
+      "eval_accuracy_road": 0.9836190767320026,
+      "eval_accuracy_rough concrete": 0.614864824602697,
+      "eval_accuracy_rough red sidewalk": 0.7556982681476022,
+      "eval_accuracy_speedway bricks": 0.9885584625940709,
+      "eval_accuracy_steel": 0.8776082404154193,
+      "eval_accuracy_tiles": 0.20333848344390923,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9802811033779495,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8743327277784108,
+      "eval_iou_concrete tiles": 0.8404769361549819,
+      "eval_iou_dark bricks": 0.8347097412353144,
+      "eval_iou_grass": 0.8130648339724684,
+      "eval_iou_red bricks": 0.8579400510204082,
+      "eval_iou_road": 0.9458303635067712,
+      "eval_iou_rough concrete": 0.6082179507583125,
+      "eval_iou_rough red sidewalk": 0.6063963311609945,
+      "eval_iou_speedway bricks": 0.9741491155827552,
+      "eval_iou_steel": 0.8308712891900332,
+      "eval_iou_tiles": 0.19649287157118184,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08317837119102478,
+      "eval_mean_accuracy": 0.757621416290334,
+      "eval_mean_iou": 0.7202125627161217,
+      "eval_overall_accuracy": 0.9762467529814122,
+      "eval_runtime": 5.9253,
+      "eval_samples_per_second": 19.915,
+      "eval_steps_per_second": 1.35,
+      "step": 760
+    },
+    {
+      "epoch": 44.76,
+      "learning_rate": 4.0851393188854494e-05,
+      "loss": 0.0368,
+      "step": 761
+    },
+    {
+      "epoch": 44.82,
+      "learning_rate": 4.0835913312693504e-05,
+      "loss": 0.042,
+      "step": 762
+    },
+    {
+      "epoch": 44.88,
+      "learning_rate": 4.082043343653251e-05,
+      "loss": 0.0483,
+      "step": 763
+    },
+    {
+      "epoch": 44.94,
+      "learning_rate": 4.080495356037152e-05,
+      "loss": 0.0441,
+      "step": 764
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 4.078947368421053e-05,
+      "loss": 0.0399,
+      "step": 765
+    },
+    {
+      "epoch": 45.06,
+      "learning_rate": 4.077399380804954e-05,
+      "loss": 0.0419,
+      "step": 766
+    },
+    {
+      "epoch": 45.12,
+      "learning_rate": 4.075851393188855e-05,
+      "loss": 0.0378,
+      "step": 767
+    },
+    {
+      "epoch": 45.18,
+      "learning_rate": 4.074303405572756e-05,
+      "loss": 0.0421,
+      "step": 768
+    },
+    {
+      "epoch": 45.24,
+      "learning_rate": 4.072755417956656e-05,
+      "loss": 0.039,
+      "step": 769
+    },
+    {
+      "epoch": 45.29,
+      "learning_rate": 4.071207430340557e-05,
+      "loss": 0.0456,
+      "step": 770
+    },
+    {
+      "epoch": 45.35,
+      "learning_rate": 4.069659442724458e-05,
+      "loss": 0.0489,
+      "step": 771
+    },
+    {
+      "epoch": 45.41,
+      "learning_rate": 4.068111455108359e-05,
+      "loss": 0.0467,
+      "step": 772
+    },
+    {
+      "epoch": 45.47,
+      "learning_rate": 4.06656346749226e-05,
+      "loss": 0.0413,
+      "step": 773
+    },
+    {
+      "epoch": 45.53,
+      "learning_rate": 4.065015479876162e-05,
+      "loss": 0.0379,
+      "step": 774
+    },
+    {
+      "epoch": 45.59,
+      "learning_rate": 4.063467492260062e-05,
+      "loss": 0.0513,
+      "step": 775
+    },
+    {
+      "epoch": 45.65,
+      "learning_rate": 4.061919504643963e-05,
+      "loss": 0.0432,
+      "step": 776
+    },
+    {
+      "epoch": 45.71,
+      "learning_rate": 4.060371517027864e-05,
+      "loss": 0.0403,
+      "step": 777
+    },
+    {
+      "epoch": 45.76,
+      "learning_rate": 4.058823529411765e-05,
+      "loss": 0.0389,
+      "step": 778
+    },
+    {
+      "epoch": 45.82,
+      "learning_rate": 4.057275541795666e-05,
+      "loss": 0.0458,
+      "step": 779
+    },
+    {
+      "epoch": 45.88,
+      "learning_rate": 4.055727554179567e-05,
+      "loss": 0.0332,
+      "step": 780
+    },
+    {
+      "epoch": 45.88,
+      "eval_accuracy_NAT": 0.9922975191587566,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9401074591543611,
+      "eval_accuracy_concrete tiles": 0.8655903841621494,
+      "eval_accuracy_dark bricks": 0.8542053480849109,
+      "eval_accuracy_grass": 0.8879079661023365,
+      "eval_accuracy_red bricks": 0.891146748681898,
+      "eval_accuracy_road": 0.9791310500582247,
+      "eval_accuracy_rough concrete": 0.6247504752143679,
+      "eval_accuracy_rough red sidewalk": 0.8070447213469796,
+      "eval_accuracy_speedway bricks": 0.9859261224246387,
+      "eval_accuracy_steel": 0.9102140568852549,
+      "eval_accuracy_tiles": 0.22954345784483018,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806645176463016,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8770076222436308,
+      "eval_iou_concrete tiles": 0.8293327311730835,
+      "eval_iou_dark bricks": 0.8286216948900772,
+      "eval_iou_grass": 0.8239650451950314,
+      "eval_iou_red bricks": 0.8598378464310318,
+      "eval_iou_road": 0.945054655970685,
+      "eval_iou_rough concrete": 0.5886596896265187,
+      "eval_iou_rough red sidewalk": 0.5982596286241478,
+      "eval_iou_speedway bricks": 0.9745539449151847,
+      "eval_iou_steel": 0.8352553504231953,
+      "eval_iou_tiles": 0.22587568937248473,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08175460249185562,
+      "eval_mean_accuracy": 0.7667588699322084,
+      "eval_mean_iou": 0.7205452628085671,
+      "eval_overall_accuracy": 0.976465936434471,
+      "eval_runtime": 7.852,
+      "eval_samples_per_second": 15.028,
+      "eval_steps_per_second": 1.019,
+      "step": 780
+    },
+    {
+      "epoch": 45.94,
+      "learning_rate": 4.0541795665634674e-05,
+      "loss": 0.0346,
+      "step": 781
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 4.0526315789473684e-05,
+      "loss": 0.0372,
+      "step": 782
+    },
+    {
+      "epoch": 46.06,
+      "learning_rate": 4.0510835913312694e-05,
+      "loss": 0.0364,
+      "step": 783
+    },
+    {
+      "epoch": 46.12,
+      "learning_rate": 4.0495356037151704e-05,
+      "loss": 0.0379,
+      "step": 784
+    },
+    {
+      "epoch": 46.18,
+      "learning_rate": 4.0479876160990714e-05,
+      "loss": 0.0333,
+      "step": 785
+    },
+    {
+      "epoch": 46.24,
+      "learning_rate": 4.0464396284829724e-05,
+      "loss": 0.0403,
+      "step": 786
+    },
+    {
+      "epoch": 46.29,
+      "learning_rate": 4.044891640866873e-05,
+      "loss": 0.0401,
+      "step": 787
+    },
+    {
+      "epoch": 46.35,
+      "learning_rate": 4.0433436532507744e-05,
+      "loss": 0.0395,
+      "step": 788
+    },
+    {
+      "epoch": 46.41,
+      "learning_rate": 4.0417956656346754e-05,
+      "loss": 0.0389,
+      "step": 789
+    },
+    {
+      "epoch": 46.47,
+      "learning_rate": 4.0402476780185764e-05,
+      "loss": 0.0436,
+      "step": 790
+    },
+    {
+      "epoch": 46.53,
+      "learning_rate": 4.0386996904024774e-05,
+      "loss": 0.0323,
+      "step": 791
+    },
+    {
+      "epoch": 46.59,
+      "learning_rate": 4.0371517027863784e-05,
+      "loss": 0.0415,
+      "step": 792
+    },
+    {
+      "epoch": 46.65,
+      "learning_rate": 4.035603715170279e-05,
+      "loss": 0.0397,
+      "step": 793
+    },
+    {
+      "epoch": 46.71,
+      "learning_rate": 4.03405572755418e-05,
+      "loss": 0.0414,
+      "step": 794
+    },
+    {
+      "epoch": 46.76,
+      "learning_rate": 4.032507739938081e-05,
+      "loss": 0.0384,
+      "step": 795
+    },
+    {
+      "epoch": 46.82,
+      "learning_rate": 4.030959752321982e-05,
+      "loss": 0.0424,
+      "step": 796
+    },
+    {
+      "epoch": 46.88,
+      "learning_rate": 4.029411764705883e-05,
+      "loss": 0.0348,
+      "step": 797
+    },
+    {
+      "epoch": 46.94,
+      "learning_rate": 4.027863777089784e-05,
+      "loss": 0.0426,
+      "step": 798
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 4.026315789473684e-05,
+      "loss": 0.0394,
+      "step": 799
+    },
+    {
+      "epoch": 47.06,
+      "learning_rate": 4.024767801857585e-05,
+      "loss": 0.0386,
+      "step": 800
+    },
+    {
+      "epoch": 47.06,
+      "eval_accuracy_NAT": 0.9925614711311426,
+      "eval_accuracy_REST": 0.032520325203252036,
+      "eval_accuracy_concrete": 0.9593274974048668,
+      "eval_accuracy_concrete tiles": 0.844025453688428,
+      "eval_accuracy_dark bricks": 0.8593495779338859,
+      "eval_accuracy_grass": 0.8724560666447142,
+      "eval_accuracy_red bricks": 0.898011862917399,
+      "eval_accuracy_road": 0.9703276131211986,
+      "eval_accuracy_rough concrete": 0.6201145281523546,
+      "eval_accuracy_rough red sidewalk": 0.7499454801134013,
+      "eval_accuracy_speedway bricks": 0.9851378427046082,
+      "eval_accuracy_steel": 0.9042666440098557,
+      "eval_accuracy_tiles": 0.19943045835984125,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810036198475386,
+      "eval_iou_REST": 0.032467532467532464,
+      "eval_iou_concrete": 0.8732396179205504,
+      "eval_iou_concrete tiles": 0.8141311266709103,
+      "eval_iou_dark bricks": 0.8340144900613041,
+      "eval_iou_grass": 0.8221526718918581,
+      "eval_iou_red bricks": 0.8615765623353356,
+      "eval_iou_road": 0.9528672167616816,
+      "eval_iou_rough concrete": 0.57871907375586,
+      "eval_iou_rough red sidewalk": 0.5950582366700081,
+      "eval_iou_speedway bricks": 0.975265346173186,
+      "eval_iou_steel": 0.8392392058091613,
+      "eval_iou_tiles": 0.1944871188844245,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08345451205968857,
+      "eval_mean_accuracy": 0.7605749862603807,
+      "eval_mean_iou": 0.7195555245576423,
+      "eval_overall_accuracy": 0.976670927920584,
+      "eval_runtime": 7.5362,
+      "eval_samples_per_second": 15.658,
+      "eval_steps_per_second": 1.062,
+      "step": 800
+    },
+    {
+      "epoch": 47.12,
+      "learning_rate": 4.023219814241486e-05,
+      "loss": 0.0389,
+      "step": 801
+    },
+    {
+      "epoch": 47.18,
+      "learning_rate": 4.021671826625387e-05,
+      "loss": 0.0433,
+      "step": 802
+    },
+    {
+      "epoch": 47.24,
+      "learning_rate": 4.020123839009288e-05,
+      "loss": 0.0384,
+      "step": 803
+    },
+    {
+      "epoch": 47.29,
+      "learning_rate": 4.018575851393189e-05,
+      "loss": 0.0359,
+      "step": 804
+    },
+    {
+      "epoch": 47.35,
+      "learning_rate": 4.01702786377709e-05,
+      "loss": 0.0408,
+      "step": 805
+    },
+    {
+      "epoch": 47.41,
+      "learning_rate": 4.015479876160991e-05,
+      "loss": 0.043,
+      "step": 806
+    },
+    {
+      "epoch": 47.47,
+      "learning_rate": 4.013931888544892e-05,
+      "loss": 0.0336,
+      "step": 807
+    },
+    {
+      "epoch": 47.53,
+      "learning_rate": 4.012383900928793e-05,
+      "loss": 0.0477,
+      "step": 808
+    },
+    {
+      "epoch": 47.59,
+      "learning_rate": 4.010835913312694e-05,
+      "loss": 0.0382,
+      "step": 809
+    },
+    {
+      "epoch": 47.65,
+      "learning_rate": 4.009287925696595e-05,
+      "loss": 0.0373,
+      "step": 810
+    },
+    {
+      "epoch": 47.71,
+      "learning_rate": 4.007739938080495e-05,
+      "loss": 0.0354,
+      "step": 811
+    },
+    {
+      "epoch": 47.76,
+      "learning_rate": 4.006191950464396e-05,
+      "loss": 0.0483,
+      "step": 812
+    },
+    {
+      "epoch": 47.82,
+      "learning_rate": 4.004643962848297e-05,
+      "loss": 0.0442,
+      "step": 813
+    },
+    {
+      "epoch": 47.88,
+      "learning_rate": 4.003095975232198e-05,
+      "loss": 0.0437,
+      "step": 814
+    },
+    {
+      "epoch": 47.94,
+      "learning_rate": 4.001547987616099e-05,
+      "loss": 0.0392,
+      "step": 815
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 4e-05,
+      "loss": 0.0403,
+      "step": 816
+    },
+    {
+      "epoch": 48.06,
+      "learning_rate": 3.9984520123839006e-05,
+      "loss": 0.0385,
+      "step": 817
+    },
+    {
+      "epoch": 48.12,
+      "learning_rate": 3.9969040247678017e-05,
+      "loss": 0.0353,
+      "step": 818
+    },
+    {
+      "epoch": 48.18,
+      "learning_rate": 3.9953560371517027e-05,
+      "loss": 0.0362,
+      "step": 819
+    },
+    {
+      "epoch": 48.24,
+      "learning_rate": 3.9938080495356037e-05,
+      "loss": 0.0457,
+      "step": 820
+    },
+    {
+      "epoch": 48.24,
+      "eval_accuracy_NAT": 0.993284913026046,
+      "eval_accuracy_REST": 0.011382113821138212,
+      "eval_accuracy_concrete": 0.9413209419225236,
+      "eval_accuracy_concrete tiles": 0.9300730615130803,
+      "eval_accuracy_dark bricks": 0.884673544781119,
+      "eval_accuracy_grass": 0.8748254209767341,
+      "eval_accuracy_red bricks": 0.9043277680140598,
+      "eval_accuracy_road": 0.9777367583706935,
+      "eval_accuracy_rough concrete": 0.6237732318748175,
+      "eval_accuracy_rough red sidewalk": 0.7846050880214169,
+      "eval_accuracy_speedway bricks": 0.9886994961178938,
+      "eval_accuracy_steel": 0.8896030226851319,
+      "eval_accuracy_tiles": 0.24520585294919567,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809656395022599,
+      "eval_iou_REST": 0.011363636363636364,
+      "eval_iou_concrete": 0.8789259208108553,
+      "eval_iou_concrete tiles": 0.8180046846174575,
+      "eval_iou_dark bricks": 0.8469760345988739,
+      "eval_iou_grass": 0.8197962701548546,
+      "eval_iou_red bricks": 0.8569346864428832,
+      "eval_iou_road": 0.9515196400171004,
+      "eval_iou_rough concrete": 0.6110368498164221,
+      "eval_iou_rough red sidewalk": 0.6043103797790945,
+      "eval_iou_speedway bricks": 0.97510861397648,
+      "eval_iou_steel": 0.8364921448745483,
+      "eval_iou_tiles": 0.23762550642945218,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0798560082912445,
+      "eval_mean_accuracy": 0.773039324159527,
+      "eval_mean_iou": 0.7253123082603015,
+      "eval_overall_accuracy": 0.9771031201896021,
+      "eval_runtime": 5.5393,
+      "eval_samples_per_second": 21.302,
+      "eval_steps_per_second": 1.444,
+      "step": 820
+    },
+    {
+      "epoch": 48.29,
+      "learning_rate": 3.9922600619195047e-05,
+      "loss": 0.0346,
+      "step": 821
+    },
+    {
+      "epoch": 48.35,
+      "learning_rate": 3.9907120743034057e-05,
+      "loss": 0.0423,
+      "step": 822
+    },
+    {
+      "epoch": 48.41,
+      "learning_rate": 3.9891640866873067e-05,
+      "loss": 0.0369,
+      "step": 823
+    },
+    {
+      "epoch": 48.47,
+      "learning_rate": 3.9876160990712077e-05,
+      "loss": 0.0436,
+      "step": 824
+    },
+    {
+      "epoch": 48.53,
+      "learning_rate": 3.9860681114551087e-05,
+      "loss": 0.0397,
+      "step": 825
+    },
+    {
+      "epoch": 48.59,
+      "learning_rate": 3.9845201238390097e-05,
+      "loss": 0.0513,
+      "step": 826
+    },
+    {
+      "epoch": 48.65,
+      "learning_rate": 3.9829721362229107e-05,
+      "loss": 0.0333,
+      "step": 827
+    },
+    {
+      "epoch": 48.71,
+      "learning_rate": 3.9814241486068117e-05,
+      "loss": 0.0362,
+      "step": 828
+    },
+    {
+      "epoch": 48.76,
+      "learning_rate": 3.979876160990712e-05,
+      "loss": 0.0405,
+      "step": 829
+    },
+    {
+      "epoch": 48.82,
+      "learning_rate": 3.978328173374613e-05,
+      "loss": 0.0367,
+      "step": 830
+    },
+    {
+      "epoch": 48.88,
+      "learning_rate": 3.976780185758514e-05,
+      "loss": 0.0366,
+      "step": 831
+    },
+    {
+      "epoch": 48.94,
+      "learning_rate": 3.975232198142415e-05,
+      "loss": 0.0449,
+      "step": 832
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 3.973684210526316e-05,
+      "loss": 0.0387,
+      "step": 833
+    },
+    {
+      "epoch": 49.06,
+      "learning_rate": 3.972136222910217e-05,
+      "loss": 0.0348,
+      "step": 834
+    },
+    {
+      "epoch": 49.12,
+      "learning_rate": 3.970588235294117e-05,
+      "loss": 0.0416,
+      "step": 835
+    },
+    {
+      "epoch": 49.18,
+      "learning_rate": 3.969040247678018e-05,
+      "loss": 0.0352,
+      "step": 836
+    },
+    {
+      "epoch": 49.24,
+      "learning_rate": 3.96749226006192e-05,
+      "loss": 0.0413,
+      "step": 837
+    },
+    {
+      "epoch": 49.29,
+      "learning_rate": 3.965944272445821e-05,
+      "loss": 0.0322,
+      "step": 838
+    },
+    {
+      "epoch": 49.35,
+      "learning_rate": 3.964396284829722e-05,
+      "loss": 0.0416,
+      "step": 839
+    },
+    {
+      "epoch": 49.41,
+      "learning_rate": 3.962848297213623e-05,
+      "loss": 0.045,
+      "step": 840
+    },
+    {
+      "epoch": 49.41,
+      "eval_accuracy_NAT": 0.9926908054411413,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9410973553337441,
+      "eval_accuracy_concrete tiles": 0.8221305679943436,
+      "eval_accuracy_dark bricks": 0.8123193537890243,
+      "eval_accuracy_grass": 0.88069865673506,
+      "eval_accuracy_red bricks": 0.8406195079086116,
+      "eval_accuracy_road": 0.9731642939481114,
+      "eval_accuracy_rough concrete": 0.6200370637412927,
+      "eval_accuracy_rough red sidewalk": 0.7597740996698726,
+      "eval_accuracy_speedway bricks": 0.9920198673907873,
+      "eval_accuracy_steel": 0.8703638917071074,
+      "eval_accuracy_tiles": 0.17676997182586568,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807335603411328,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8722015791095635,
+      "eval_iou_concrete tiles": 0.7926333250017042,
+      "eval_iou_dark bricks": 0.7954042808411741,
+      "eval_iou_grass": 0.8199669985895554,
+      "eval_iou_red bricks": 0.8261456253036109,
+      "eval_iou_road": 0.9523591859959761,
+      "eval_iou_rough concrete": 0.6057257951846505,
+      "eval_iou_rough red sidewalk": 0.6041221949163184,
+      "eval_iou_speedway bricks": 0.9719564469144301,
+      "eval_iou_steel": 0.8282038783007764,
+      "eval_iou_tiles": 0.17556792537987062,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0856236144900322,
+      "eval_mean_accuracy": 0.7447450334988432,
+      "eval_mean_iou": 0.7096169842983664,
+      "eval_overall_accuracy": 0.9761548769676079,
+      "eval_runtime": 5.6297,
+      "eval_samples_per_second": 20.96,
+      "eval_steps_per_second": 1.421,
+      "step": 840
+    },
+    {
+      "epoch": 49.47,
+      "learning_rate": 3.961300309597523e-05,
+      "loss": 0.0331,
+      "step": 841
+    },
+    {
+      "epoch": 49.53,
+      "learning_rate": 3.959752321981424e-05,
+      "loss": 0.0397,
+      "step": 842
+    },
+    {
+      "epoch": 49.59,
+      "learning_rate": 3.958204334365325e-05,
+      "loss": 0.0355,
+      "step": 843
+    },
+    {
+      "epoch": 49.65,
+      "learning_rate": 3.956656346749226e-05,
+      "loss": 0.0337,
+      "step": 844
+    },
+    {
+      "epoch": 49.71,
+      "learning_rate": 3.955108359133127e-05,
+      "loss": 0.0476,
+      "step": 845
+    },
+    {
+      "epoch": 49.76,
+      "learning_rate": 3.953560371517028e-05,
+      "loss": 0.0418,
+      "step": 846
+    },
+    {
+      "epoch": 49.82,
+      "learning_rate": 3.9520123839009286e-05,
+      "loss": 0.04,
+      "step": 847
+    },
+    {
+      "epoch": 49.88,
+      "learning_rate": 3.9504643962848296e-05,
+      "loss": 0.0385,
+      "step": 848
+    },
+    {
+      "epoch": 49.94,
+      "learning_rate": 3.9489164086687306e-05,
+      "loss": 0.0404,
+      "step": 849
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 3.9473684210526316e-05,
+      "loss": 0.0332,
+      "step": 850
+    },
+    {
+      "epoch": 50.06,
+      "learning_rate": 3.9458204334365326e-05,
+      "loss": 0.0366,
+      "step": 851
+    },
+    {
+      "epoch": 50.12,
+      "learning_rate": 3.9442724458204336e-05,
+      "loss": 0.0356,
+      "step": 852
+    },
+    {
+      "epoch": 50.18,
+      "learning_rate": 3.9427244582043346e-05,
+      "loss": 0.0462,
+      "step": 853
+    },
+    {
+      "epoch": 50.24,
+      "learning_rate": 3.9411764705882356e-05,
+      "loss": 0.0339,
+      "step": 854
+    },
+    {
+      "epoch": 50.29,
+      "learning_rate": 3.9396284829721366e-05,
+      "loss": 0.0345,
+      "step": 855
+    },
+    {
+      "epoch": 50.35,
+      "learning_rate": 3.9380804953560376e-05,
+      "loss": 0.0393,
+      "step": 856
+    },
+    {
+      "epoch": 50.41,
+      "learning_rate": 3.9365325077399386e-05,
+      "loss": 0.0454,
+      "step": 857
+    },
+    {
+      "epoch": 50.47,
+      "learning_rate": 3.9349845201238396e-05,
+      "loss": 0.0374,
+      "step": 858
+    },
+    {
+      "epoch": 50.53,
+      "learning_rate": 3.93343653250774e-05,
+      "loss": 0.0394,
+      "step": 859
+    },
+    {
+      "epoch": 50.59,
+      "learning_rate": 3.931888544891641e-05,
+      "loss": 0.0414,
+      "step": 860
+    },
+    {
+      "epoch": 50.59,
+      "eval_accuracy_NAT": 0.9925857314227221,
+      "eval_accuracy_REST": 0.12195121951219512,
+      "eval_accuracy_concrete": 0.9468819416947328,
+      "eval_accuracy_concrete tiles": 0.8721659203393826,
+      "eval_accuracy_dark bricks": 0.8332982409126581,
+      "eval_accuracy_grass": 0.8964788972283096,
+      "eval_accuracy_red bricks": 0.8888949912126538,
+      "eval_accuracy_road": 0.9774647405596726,
+      "eval_accuracy_rough concrete": 0.6200728165463982,
+      "eval_accuracy_rough red sidewalk": 0.7201889020070839,
+      "eval_accuracy_speedway bricks": 0.990332606059609,
+      "eval_accuracy_steel": 0.8716133481935359,
+      "eval_accuracy_tiles": 0.21218455572722592,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9814399439643317,
+      "eval_iou_REST": 0.12175324675324675,
+      "eval_iou_concrete": 0.877990003317107,
+      "eval_iou_concrete tiles": 0.8280043854742354,
+      "eval_iou_dark bricks": 0.8171370920438281,
+      "eval_iou_grass": 0.8289540342020185,
+      "eval_iou_red bricks": 0.8561227188574452,
+      "eval_iou_road": 0.9513924722233781,
+      "eval_iou_rough concrete": 0.6119880495894987,
+      "eval_iou_rough red sidewalk": 0.5965045592705167,
+      "eval_iou_speedway bricks": 0.9745007669094427,
+      "eval_iou_steel": 0.8298664065381093,
+      "eval_iou_tiles": 0.2088377362991234,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08020060509443283,
+      "eval_mean_accuracy": 0.7649318393397061,
+      "eval_mean_iou": 0.7295762627263294,
+      "eval_overall_accuracy": 0.9771873021529893,
+      "eval_runtime": 5.851,
+      "eval_samples_per_second": 20.167,
+      "eval_steps_per_second": 1.367,
+      "step": 860
+    },
+    {
+      "epoch": 50.65,
+      "learning_rate": 3.930340557275542e-05,
+      "loss": 0.0366,
+      "step": 861
+    },
+    {
+      "epoch": 50.71,
+      "learning_rate": 3.928792569659443e-05,
+      "loss": 0.0387,
+      "step": 862
+    },
+    {
+      "epoch": 50.76,
+      "learning_rate": 3.927244582043344e-05,
+      "loss": 0.0357,
+      "step": 863
+    },
+    {
+      "epoch": 50.82,
+      "learning_rate": 3.925696594427245e-05,
+      "loss": 0.0407,
+      "step": 864
+    },
+    {
+      "epoch": 50.88,
+      "learning_rate": 3.924148606811145e-05,
+      "loss": 0.0329,
+      "step": 865
+    },
+    {
+      "epoch": 50.94,
+      "learning_rate": 3.922600619195046e-05,
+      "loss": 0.0373,
+      "step": 866
+    },
+    {
+      "epoch": 51.0,
+      "learning_rate": 3.921052631578947e-05,
+      "loss": 0.0337,
+      "step": 867
+    },
+    {
+      "epoch": 51.06,
+      "learning_rate": 3.919504643962848e-05,
+      "loss": 0.0376,
+      "step": 868
+    },
+    {
+      "epoch": 51.12,
+      "learning_rate": 3.917956656346749e-05,
+      "loss": 0.0329,
+      "step": 869
+    },
+    {
+      "epoch": 51.18,
+      "learning_rate": 3.91640866873065e-05,
+      "loss": 0.0431,
+      "step": 870
+    },
+    {
+      "epoch": 51.24,
+      "learning_rate": 3.914860681114551e-05,
+      "loss": 0.0336,
+      "step": 871
+    },
+    {
+      "epoch": 51.29,
+      "learning_rate": 3.913312693498452e-05,
+      "loss": 0.0293,
+      "step": 872
+    },
+    {
+      "epoch": 51.35,
+      "learning_rate": 3.911764705882353e-05,
+      "loss": 0.0379,
+      "step": 873
+    },
+    {
+      "epoch": 51.41,
+      "learning_rate": 3.910216718266254e-05,
+      "loss": 0.0294,
+      "step": 874
+    },
+    {
+      "epoch": 51.47,
+      "learning_rate": 3.908668730650155e-05,
+      "loss": 0.0377,
+      "step": 875
+    },
+    {
+      "epoch": 51.53,
+      "learning_rate": 3.907120743034056e-05,
+      "loss": 0.0342,
+      "step": 876
+    },
+    {
+      "epoch": 51.59,
+      "learning_rate": 3.9055727554179566e-05,
+      "loss": 0.0444,
+      "step": 877
+    },
+    {
+      "epoch": 51.65,
+      "learning_rate": 3.9040247678018576e-05,
+      "loss": 0.0369,
+      "step": 878
+    },
+    {
+      "epoch": 51.71,
+      "learning_rate": 3.9024767801857586e-05,
+      "loss": 0.0404,
+      "step": 879
+    },
+    {
+      "epoch": 51.76,
+      "learning_rate": 3.9009287925696596e-05,
+      "loss": 0.038,
+      "step": 880
+    },
+    {
+      "epoch": 51.76,
+      "eval_accuracy_NAT": 0.9934121986892003,
+      "eval_accuracy_REST": 0.03414634146341464,
+      "eval_accuracy_concrete": 0.9429991790359099,
+      "eval_accuracy_concrete tiles": 0.8840207400424228,
+      "eval_accuracy_dark bricks": 0.8399642057309114,
+      "eval_accuracy_grass": 0.8809948260265625,
+      "eval_accuracy_red bricks": 0.8958699472759226,
+      "eval_accuracy_road": 0.9795115543439749,
+      "eval_accuracy_rough concrete": 0.6205018502076642,
+      "eval_accuracy_rough red sidewalk": 0.7668428849667993,
+      "eval_accuracy_speedway bricks": 0.9909476356063893,
+      "eval_accuracy_steel": 0.8687745830563705,
+      "eval_accuracy_tiles": 0.1804659335332788,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9812413400145908,
+      "eval_iou_REST": 0.03414634146341464,
+      "eval_iou_concrete": 0.8798285343221981,
+      "eval_iou_concrete tiles": 0.8269909163065526,
+      "eval_iou_dark bricks": 0.8211959877987163,
+      "eval_iou_grass": 0.82433791603805,
+      "eval_iou_red bricks": 0.8576235541535226,
+      "eval_iou_road": 0.9539101529545061,
+      "eval_iou_rough concrete": 0.6107806909496158,
+      "eval_iou_rough red sidewalk": 0.6117803748410165,
+      "eval_iou_speedway bricks": 0.97457153797357,
+      "eval_iou_steel": 0.8285751056390896,
+      "eval_iou_tiles": 0.17920640173279986,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08046015352010727,
+      "eval_mean_accuracy": 0.7598809138445246,
+      "eval_mean_iou": 0.7218606810913571,
+      "eval_overall_accuracy": 0.9772612684864109,
+      "eval_runtime": 5.5554,
+      "eval_samples_per_second": 21.241,
+      "eval_steps_per_second": 1.44,
+      "step": 880
+    },
+    {
+      "epoch": 51.82,
+      "learning_rate": 3.8993808049535606e-05,
+      "loss": 0.0367,
+      "step": 881
+    },
+    {
+      "epoch": 51.88,
+      "learning_rate": 3.8978328173374616e-05,
+      "loss": 0.0367,
+      "step": 882
+    },
+    {
+      "epoch": 51.94,
+      "learning_rate": 3.896284829721362e-05,
+      "loss": 0.0403,
+      "step": 883
+    },
+    {
+      "epoch": 52.0,
+      "learning_rate": 3.894736842105263e-05,
+      "loss": 0.0419,
+      "step": 884
+    },
+    {
+      "epoch": 52.06,
+      "learning_rate": 3.893188854489164e-05,
+      "loss": 0.0356,
+      "step": 885
+    },
+    {
+      "epoch": 52.12,
+      "learning_rate": 3.8916408668730656e-05,
+      "loss": 0.0359,
+      "step": 886
+    },
+    {
+      "epoch": 52.18,
+      "learning_rate": 3.8900928792569666e-05,
+      "loss": 0.044,
+      "step": 887
+    },
+    {
+      "epoch": 52.24,
+      "learning_rate": 3.8885448916408676e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 52.29,
+      "learning_rate": 3.886996904024768e-05,
+      "loss": 0.0376,
+      "step": 889
+    },
+    {
+      "epoch": 52.35,
+      "learning_rate": 3.885448916408669e-05,
+      "loss": 0.0388,
+      "step": 890
+    },
+    {
+      "epoch": 52.41,
+      "learning_rate": 3.88390092879257e-05,
+      "loss": 0.0376,
+      "step": 891
+    },
+    {
+      "epoch": 52.47,
+      "learning_rate": 3.882352941176471e-05,
+      "loss": 0.0388,
+      "step": 892
+    },
+    {
+      "epoch": 52.53,
+      "learning_rate": 3.880804953560372e-05,
+      "loss": 0.0333,
+      "step": 893
+    },
+    {
+      "epoch": 52.59,
+      "learning_rate": 3.879256965944273e-05,
+      "loss": 0.0375,
+      "step": 894
+    },
+    {
+      "epoch": 52.65,
+      "learning_rate": 3.877708978328173e-05,
+      "loss": 0.0365,
+      "step": 895
+    },
+    {
+      "epoch": 52.71,
+      "learning_rate": 3.876160990712074e-05,
+      "loss": 0.0364,
+      "step": 896
+    },
+    {
+      "epoch": 52.76,
+      "learning_rate": 3.874613003095975e-05,
+      "loss": 0.0352,
+      "step": 897
+    },
+    {
+      "epoch": 52.82,
+      "learning_rate": 3.873065015479876e-05,
+      "loss": 0.0356,
+      "step": 898
+    },
+    {
+      "epoch": 52.88,
+      "learning_rate": 3.871517027863777e-05,
+      "loss": 0.0347,
+      "step": 899
+    },
+    {
+      "epoch": 52.94,
+      "learning_rate": 3.869969040247678e-05,
+      "loss": 0.0348,
+      "step": 900
+    },
+    {
+      "epoch": 52.94,
+      "eval_accuracy_NAT": 0.9888540133719492,
+      "eval_accuracy_REST": 0.3008130081300813,
+      "eval_accuracy_concrete": 0.9529829890736865,
+      "eval_accuracy_concrete tiles": 0.901249116191374,
+      "eval_accuracy_dark bricks": 0.8564975211990123,
+      "eval_accuracy_grass": 0.9121957404610779,
+      "eval_accuracy_red bricks": 0.9022407732864675,
+      "eval_accuracy_road": 0.9801090315791836,
+      "eval_accuracy_rough concrete": 0.6241903479343817,
+      "eval_accuracy_rough red sidewalk": 0.8273035592085969,
+      "eval_accuracy_speedway bricks": 0.9887646148773745,
+      "eval_accuracy_steel": 0.8903377030991518,
+      "eval_accuracy_tiles": 0.20206610318398013,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810375330982319,
+      "eval_iou_REST": 0.29983792544570503,
+      "eval_iou_concrete": 0.877285158428386,
+      "eval_iou_concrete tiles": 0.8302575014112641,
+      "eval_iou_dark bricks": 0.8291632115111113,
+      "eval_iou_grass": 0.8277350139310445,
+      "eval_iou_red bricks": 0.8709574806489238,
+      "eval_iou_road": 0.9462856099116447,
+      "eval_iou_rough concrete": 0.585037698966769,
+      "eval_iou_rough red sidewalk": 0.6260363851568591,
+      "eval_iou_speedway bricks": 0.9756787477607302,
+      "eval_iou_steel": 0.8367543447627994,
+      "eval_iou_tiles": 0.19808154901553174,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08102715015411377,
+      "eval_mean_accuracy": 0.7944311170458707,
+      "eval_mean_iou": 0.7449344738499231,
+      "eval_overall_accuracy": 0.9767890542240466,
+      "eval_runtime": 5.6704,
+      "eval_samples_per_second": 20.81,
+      "eval_steps_per_second": 1.411,
+      "step": 900
+    },
+    {
+      "epoch": 53.0,
+      "learning_rate": 3.868421052631579e-05,
+      "loss": 0.0401,
+      "step": 901
+    },
+    {
+      "epoch": 53.06,
+      "learning_rate": 3.86687306501548e-05,
+      "loss": 0.0358,
+      "step": 902
+    },
+    {
+      "epoch": 53.12,
+      "learning_rate": 3.865325077399381e-05,
+      "loss": 0.0372,
+      "step": 903
+    },
+    {
+      "epoch": 53.18,
+      "learning_rate": 3.863777089783282e-05,
+      "loss": 0.0398,
+      "step": 904
+    },
+    {
+      "epoch": 53.24,
+      "learning_rate": 3.862229102167183e-05,
+      "loss": 0.0396,
+      "step": 905
+    },
+    {
+      "epoch": 53.29,
+      "learning_rate": 3.8606811145510836e-05,
+      "loss": 0.0405,
+      "step": 906
+    },
+    {
+      "epoch": 53.35,
+      "learning_rate": 3.8591331269349846e-05,
+      "loss": 0.0309,
+      "step": 907
+    },
+    {
+      "epoch": 53.41,
+      "learning_rate": 3.8575851393188856e-05,
+      "loss": 0.0356,
+      "step": 908
+    },
+    {
+      "epoch": 53.47,
+      "learning_rate": 3.8560371517027866e-05,
+      "loss": 0.0368,
+      "step": 909
+    },
+    {
+      "epoch": 53.53,
+      "learning_rate": 3.8544891640866876e-05,
+      "loss": 0.0343,
+      "step": 910
+    },
+    {
+      "epoch": 53.59,
+      "learning_rate": 3.8529411764705886e-05,
+      "loss": 0.0337,
+      "step": 911
+    },
+    {
+      "epoch": 53.65,
+      "learning_rate": 3.851393188854489e-05,
+      "loss": 0.0344,
+      "step": 912
+    },
+    {
+      "epoch": 53.71,
+      "learning_rate": 3.84984520123839e-05,
+      "loss": 0.0422,
+      "step": 913
+    },
+    {
+      "epoch": 53.76,
+      "learning_rate": 3.848297213622291e-05,
+      "loss": 0.0366,
+      "step": 914
+    },
+    {
+      "epoch": 53.82,
+      "learning_rate": 3.846749226006192e-05,
+      "loss": 0.0342,
+      "step": 915
+    },
+    {
+      "epoch": 53.88,
+      "learning_rate": 3.845201238390093e-05,
+      "loss": 0.0374,
+      "step": 916
+    },
+    {
+      "epoch": 53.94,
+      "learning_rate": 3.843653250773994e-05,
+      "loss": 0.0311,
+      "step": 917
+    },
+    {
+      "epoch": 54.0,
+      "learning_rate": 3.842105263157895e-05,
+      "loss": 0.0397,
+      "step": 918
+    },
+    {
+      "epoch": 54.06,
+      "learning_rate": 3.840557275541796e-05,
+      "loss": 0.0368,
+      "step": 919
+    },
+    {
+      "epoch": 54.12,
+      "learning_rate": 3.839009287925697e-05,
+      "loss": 0.0386,
+      "step": 920
+    },
+    {
+      "epoch": 54.12,
+      "eval_accuracy_NAT": 0.9937696336518063,
+      "eval_accuracy_REST": 0.05365853658536585,
+      "eval_accuracy_concrete": 0.9511832125838354,
+      "eval_accuracy_concrete tiles": 0.8834786707518265,
+      "eval_accuracy_dark bricks": 0.8349826771050667,
+      "eval_accuracy_grass": 0.8486016646038901,
+      "eval_accuracy_red bricks": 0.9054261862917399,
+      "eval_accuracy_road": 0.9782642270926534,
+      "eval_accuracy_rough concrete": 0.616437948027339,
+      "eval_accuracy_rough red sidewalk": 0.7137668353649824,
+      "eval_accuracy_speedway bricks": 0.9895650615235185,
+      "eval_accuracy_steel": 0.8807443761963546,
+      "eval_accuracy_tiles": 0.19912751067890574,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9802639671263945,
+      "eval_iou_REST": 0.05365853658536585,
+      "eval_iou_concrete": 0.8779370132905309,
+      "eval_iou_concrete tiles": 0.8351379049146728,
+      "eval_iou_dark bricks": 0.819423311731004,
+      "eval_iou_grass": 0.8058242656770866,
+      "eval_iou_red bricks": 0.8772415261001437,
+      "eval_iou_road": 0.9516963894859102,
+      "eval_iou_rough concrete": 0.6105622248191036,
+      "eval_iou_rough red sidewalk": 0.6058455194776181,
+      "eval_iou_speedway bricks": 0.9750876922217045,
+      "eval_iou_steel": 0.8357127783429009,
+      "eval_iou_tiles": 0.1962734032070232,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08540533483028412,
+      "eval_mean_accuracy": 0.7576158877274832,
+      "eval_mean_iou": 0.7249741948445737,
+      "eval_overall_accuracy": 0.9766210135767015,
+      "eval_runtime": 5.7762,
+      "eval_samples_per_second": 20.429,
+      "eval_steps_per_second": 1.385,
+      "step": 920
+    },
+    {
+      "epoch": 54.18,
+      "learning_rate": 3.837461300309598e-05,
+      "loss": 0.0342,
+      "step": 921
+    },
+    {
+      "epoch": 54.24,
+      "learning_rate": 3.835913312693499e-05,
+      "loss": 0.037,
+      "step": 922
+    },
+    {
+      "epoch": 54.29,
+      "learning_rate": 3.8343653250774e-05,
+      "loss": 0.0303,
+      "step": 923
+    },
+    {
+      "epoch": 54.35,
+      "learning_rate": 3.8328173374613e-05,
+      "loss": 0.0331,
+      "step": 924
+    },
+    {
+      "epoch": 54.41,
+      "learning_rate": 3.831269349845201e-05,
+      "loss": 0.0398,
+      "step": 925
+    },
+    {
+      "epoch": 54.47,
+      "learning_rate": 3.829721362229102e-05,
+      "loss": 0.0359,
+      "step": 926
+    },
+    {
+      "epoch": 54.53,
+      "learning_rate": 3.828173374613003e-05,
+      "loss": 0.0387,
+      "step": 927
+    },
+    {
+      "epoch": 54.59,
+      "learning_rate": 3.826625386996904e-05,
+      "loss": 0.0303,
+      "step": 928
+    },
+    {
+      "epoch": 54.65,
+      "learning_rate": 3.825077399380805e-05,
+      "loss": 0.0363,
+      "step": 929
+    },
+    {
+      "epoch": 54.71,
+      "learning_rate": 3.8235294117647055e-05,
+      "loss": 0.0398,
+      "step": 930
+    },
+    {
+      "epoch": 54.76,
+      "learning_rate": 3.8219814241486065e-05,
+      "loss": 0.0346,
+      "step": 931
+    },
+    {
+      "epoch": 54.82,
+      "learning_rate": 3.8204334365325075e-05,
+      "loss": 0.0339,
+      "step": 932
+    },
+    {
+      "epoch": 54.88,
+      "learning_rate": 3.8188854489164085e-05,
+      "loss": 0.0331,
+      "step": 933
+    },
+    {
+      "epoch": 54.94,
+      "learning_rate": 3.81733746130031e-05,
+      "loss": 0.0343,
+      "step": 934
+    },
+    {
+      "epoch": 55.0,
+      "learning_rate": 3.815789473684211e-05,
+      "loss": 0.0422,
+      "step": 935
+    },
+    {
+      "epoch": 55.06,
+      "learning_rate": 3.8142414860681115e-05,
+      "loss": 0.0307,
+      "step": 936
+    },
+    {
+      "epoch": 55.12,
+      "learning_rate": 3.8126934984520125e-05,
+      "loss": 0.0324,
+      "step": 937
+    },
+    {
+      "epoch": 55.18,
+      "learning_rate": 3.8111455108359135e-05,
+      "loss": 0.034,
+      "step": 938
+    },
+    {
+      "epoch": 55.24,
+      "learning_rate": 3.8095975232198145e-05,
+      "loss": 0.03,
+      "step": 939
+    },
+    {
+      "epoch": 55.29,
+      "learning_rate": 3.8080495356037155e-05,
+      "loss": 0.0387,
+      "step": 940
+    },
+    {
+      "epoch": 55.29,
+      "eval_accuracy_NAT": 0.9936041245514744,
+      "eval_accuracy_REST": 0.08292682926829269,
+      "eval_accuracy_concrete": 0.9422481574171889,
+      "eval_accuracy_concrete tiles": 0.8882394532170634,
+      "eval_accuracy_dark bricks": 0.8624552571636391,
+      "eval_accuracy_grass": 0.8795215493911971,
+      "eval_accuracy_red bricks": 0.8723088752196837,
+      "eval_accuracy_road": 0.9796440895446293,
+      "eval_accuracy_rough concrete": 0.623093928577813,
+      "eval_accuracy_rough red sidewalk": 0.8036532084013265,
+      "eval_accuracy_speedway bricks": 0.9909808804467558,
+      "eval_accuracy_steel": 0.8757065676430753,
+      "eval_accuracy_tiles": 0.2240601048198976,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810823956322647,
+      "eval_iou_REST": 0.08279220779220779,
+      "eval_iou_concrete": 0.8819949505881023,
+      "eval_iou_concrete tiles": 0.827980139724944,
+      "eval_iou_dark bricks": 0.8400208806215713,
+      "eval_iou_grass": 0.826028110636755,
+      "eval_iou_red bricks": 0.8475001333973641,
+      "eval_iou_road": 0.9556556740770245,
+      "eval_iou_rough concrete": 0.612638634193212,
+      "eval_iou_rough red sidewalk": 0.6468049822667135,
+      "eval_iou_speedway bricks": 0.9754247112745985,
+      "eval_iou_steel": 0.8335117212771626,
+      "eval_iou_tiles": 0.22174916799088537,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07935668528079987,
+      "eval_mean_accuracy": 0.7706494635124644,
+      "eval_mean_iou": 0.7333218238056005,
+      "eval_overall_accuracy": 0.9777301206427106,
+      "eval_runtime": 5.3545,
+      "eval_samples_per_second": 22.038,
+      "eval_steps_per_second": 1.494,
+      "step": 940
+    },
+    {
+      "epoch": 55.35,
+      "learning_rate": 3.8065015479876166e-05,
+      "loss": 0.0383,
+      "step": 941
+    },
+    {
+      "epoch": 55.41,
+      "learning_rate": 3.804953560371517e-05,
+      "loss": 0.0361,
+      "step": 942
+    },
+    {
+      "epoch": 55.47,
+      "learning_rate": 3.803405572755418e-05,
+      "loss": 0.045,
+      "step": 943
+    },
+    {
+      "epoch": 55.53,
+      "learning_rate": 3.801857585139319e-05,
+      "loss": 0.0375,
+      "step": 944
+    },
+    {
+      "epoch": 55.59,
+      "learning_rate": 3.80030959752322e-05,
+      "loss": 0.0346,
+      "step": 945
+    },
+    {
+      "epoch": 55.65,
+      "learning_rate": 3.798761609907121e-05,
+      "loss": 0.0341,
+      "step": 946
+    },
+    {
+      "epoch": 55.71,
+      "learning_rate": 3.797213622291022e-05,
+      "loss": 0.0339,
+      "step": 947
+    },
+    {
+      "epoch": 55.76,
+      "learning_rate": 3.795665634674922e-05,
+      "loss": 0.0317,
+      "step": 948
+    },
+    {
+      "epoch": 55.82,
+      "learning_rate": 3.794117647058824e-05,
+      "loss": 0.034,
+      "step": 949
+    },
+    {
+      "epoch": 55.88,
+      "learning_rate": 3.792569659442725e-05,
+      "loss": 0.0385,
+      "step": 950
+    },
+    {
+      "epoch": 55.94,
+      "learning_rate": 3.791021671826626e-05,
+      "loss": 0.0381,
+      "step": 951
+    },
+    {
+      "epoch": 56.0,
+      "learning_rate": 3.789473684210527e-05,
+      "loss": 0.0362,
+      "step": 952
+    },
+    {
+      "epoch": 56.06,
+      "learning_rate": 3.787925696594428e-05,
+      "loss": 0.0383,
+      "step": 953
+    },
+    {
+      "epoch": 56.12,
+      "learning_rate": 3.786377708978328e-05,
+      "loss": 0.0335,
+      "step": 954
+    },
+    {
+      "epoch": 56.18,
+      "learning_rate": 3.784829721362229e-05,
+      "loss": 0.0355,
+      "step": 955
+    },
+    {
+      "epoch": 56.24,
+      "learning_rate": 3.78328173374613e-05,
+      "loss": 0.0321,
+      "step": 956
+    },
+    {
+      "epoch": 56.29,
+      "learning_rate": 3.781733746130031e-05,
+      "loss": 0.0383,
+      "step": 957
+    },
+    {
+      "epoch": 56.35,
+      "learning_rate": 3.780185758513932e-05,
+      "loss": 0.0369,
+      "step": 958
+    },
+    {
+      "epoch": 56.41,
+      "learning_rate": 3.778637770897833e-05,
+      "loss": 0.0379,
+      "step": 959
+    },
+    {
+      "epoch": 56.47,
+      "learning_rate": 3.7770897832817335e-05,
+      "loss": 0.0283,
+      "step": 960
+    },
+    {
+      "epoch": 56.47,
+      "eval_accuracy_NAT": 0.9940290570364307,
+      "eval_accuracy_REST": 0.1073170731707317,
+      "eval_accuracy_concrete": 0.947219423674036,
+      "eval_accuracy_concrete tiles": 0.8996464765496112,
+      "eval_accuracy_dark bricks": 0.8595505617977528,
+      "eval_accuracy_grass": 0.8739274508245429,
+      "eval_accuracy_red bricks": 0.9006480667838312,
+      "eval_accuracy_road": 0.9807049055659974,
+      "eval_accuracy_rough concrete": 0.6202932921778821,
+      "eval_accuracy_rough red sidewalk": 0.786327164439498,
+      "eval_accuracy_speedway bricks": 0.9863600190009686,
+      "eval_accuracy_steel": 0.8993787702349478,
+      "eval_accuracy_tiles": 0.22711987639734618,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810132306957249,
+      "eval_iou_REST": 0.10696920583468396,
+      "eval_iou_concrete": 0.8835801707236987,
+      "eval_iou_concrete tiles": 0.8253762324857291,
+      "eval_iou_dark bricks": 0.8396454848872975,
+      "eval_iou_grass": 0.8187282816821502,
+      "eval_iou_red bricks": 0.858586387434555,
+      "eval_iou_road": 0.954342682108701,
+      "eval_iou_rough concrete": 0.6133852653660089,
+      "eval_iou_rough red sidewalk": 0.6249439693041435,
+      "eval_iou_speedway bricks": 0.9763386266792573,
+      "eval_iou_steel": 0.8405589277331571,
+      "eval_iou_tiles": 0.22450813044650075,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0812796875834465,
+      "eval_mean_accuracy": 0.775578625973352,
+      "eval_mean_iou": 0.7344597381062775,
+      "eval_overall_accuracy": 0.977639764042224,
+      "eval_runtime": 5.7,
+      "eval_samples_per_second": 20.702,
+      "eval_steps_per_second": 1.404,
+      "step": 960
+    },
+    {
+      "epoch": 56.53,
+      "learning_rate": 3.7755417956656345e-05,
+      "loss": 0.0333,
+      "step": 961
+    },
+    {
+      "epoch": 56.59,
+      "learning_rate": 3.7739938080495355e-05,
+      "loss": 0.0327,
+      "step": 962
+    },
+    {
+      "epoch": 56.65,
+      "learning_rate": 3.7724458204334365e-05,
+      "loss": 0.0384,
+      "step": 963
+    },
+    {
+      "epoch": 56.71,
+      "learning_rate": 3.7708978328173375e-05,
+      "loss": 0.032,
+      "step": 964
+    },
+    {
+      "epoch": 56.76,
+      "learning_rate": 3.7693498452012385e-05,
+      "loss": 0.0304,
+      "step": 965
+    },
+    {
+      "epoch": 56.82,
+      "learning_rate": 3.7678018575851395e-05,
+      "loss": 0.0387,
+      "step": 966
+    },
+    {
+      "epoch": 56.88,
+      "learning_rate": 3.7662538699690405e-05,
+      "loss": 0.0335,
+      "step": 967
+    },
+    {
+      "epoch": 56.94,
+      "learning_rate": 3.7647058823529415e-05,
+      "loss": 0.0353,
+      "step": 968
+    },
+    {
+      "epoch": 57.0,
+      "learning_rate": 3.7631578947368425e-05,
+      "loss": 0.0365,
+      "step": 969
+    },
+    {
+      "epoch": 57.06,
+      "learning_rate": 3.7616099071207435e-05,
+      "loss": 0.0321,
+      "step": 970
+    },
+    {
+      "epoch": 57.12,
+      "learning_rate": 3.7600619195046445e-05,
+      "loss": 0.0316,
+      "step": 971
+    },
+    {
+      "epoch": 57.18,
+      "learning_rate": 3.758513931888545e-05,
+      "loss": 0.0374,
+      "step": 972
+    },
+    {
+      "epoch": 57.24,
+      "learning_rate": 3.756965944272446e-05,
+      "loss": 0.0364,
+      "step": 973
+    },
+    {
+      "epoch": 57.29,
+      "learning_rate": 3.755417956656347e-05,
+      "loss": 0.0328,
+      "step": 974
+    },
+    {
+      "epoch": 57.35,
+      "learning_rate": 3.753869969040248e-05,
+      "loss": 0.0309,
+      "step": 975
+    },
+    {
+      "epoch": 57.41,
+      "learning_rate": 3.752321981424149e-05,
+      "loss": 0.0406,
+      "step": 976
+    },
+    {
+      "epoch": 57.47,
+      "learning_rate": 3.75077399380805e-05,
+      "loss": 0.0333,
+      "step": 977
+    },
+    {
+      "epoch": 57.53,
+      "learning_rate": 3.74922600619195e-05,
+      "loss": 0.0378,
+      "step": 978
+    },
+    {
+      "epoch": 57.59,
+      "learning_rate": 3.747678018575851e-05,
+      "loss": 0.0367,
+      "step": 979
+    },
+    {
+      "epoch": 57.65,
+      "learning_rate": 3.746130030959752e-05,
+      "loss": 0.0384,
+      "step": 980
+    },
+    {
+      "epoch": 57.65,
+      "eval_accuracy_NAT": 0.9925220616352655,
+      "eval_accuracy_REST": 0.15772357723577235,
+      "eval_accuracy_concrete": 0.9498726511942199,
+      "eval_accuracy_concrete tiles": 0.9201037002121141,
+      "eval_accuracy_dark bricks": 0.849343452711368,
+      "eval_accuracy_grass": 0.8902697506122094,
+      "eval_accuracy_red bricks": 0.8871375219683656,
+      "eval_accuracy_road": 0.9787938334791398,
+      "eval_accuracy_rough concrete": 0.6250186212526592,
+      "eval_accuracy_rough red sidewalk": 0.7860263650651607,
+      "eval_accuracy_speedway bricks": 0.9885589766895405,
+      "eval_accuracy_steel": 0.8913947432866703,
+      "eval_accuracy_tiles": 0.24911387803326365,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9816768379657402,
+      "eval_iou_REST": 0.15746753246753248,
+      "eval_iou_concrete": 0.8824866621097391,
+      "eval_iou_concrete tiles": 0.8382899229134011,
+      "eval_iou_dark bricks": 0.8279833739964453,
+      "eval_iou_grass": 0.8279874402019143,
+      "eval_iou_red bricks": 0.8572414159104177,
+      "eval_iou_road": 0.9501891537987437,
+      "eval_iou_rough concrete": 0.5993440299872006,
+      "eval_iou_rough red sidewalk": 0.6357272318116022,
+      "eval_iou_speedway bricks": 0.9760409479396596,
+      "eval_iou_steel": 0.8398661725441814,
+      "eval_iou_tiles": 0.24450655645089353,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07997491955757141,
+      "eval_mean_accuracy": 0.7819907025673654,
+      "eval_mean_iou": 0.7399082521613438,
+      "eval_overall_accuracy": 0.9777147648698192,
+      "eval_runtime": 7.695,
+      "eval_samples_per_second": 15.335,
+      "eval_steps_per_second": 1.04,
+      "step": 980
+    },
+    {
+      "epoch": 57.71,
+      "learning_rate": 3.744582043343653e-05,
+      "loss": 0.0299,
+      "step": 981
+    },
+    {
+      "epoch": 57.76,
+      "learning_rate": 3.743034055727554e-05,
+      "loss": 0.0298,
+      "step": 982
+    },
+    {
+      "epoch": 57.82,
+      "learning_rate": 3.741486068111456e-05,
+      "loss": 0.0384,
+      "step": 983
+    },
+    {
+      "epoch": 57.88,
+      "learning_rate": 3.739938080495356e-05,
+      "loss": 0.0324,
+      "step": 984
+    },
+    {
+      "epoch": 57.94,
+      "learning_rate": 3.738390092879257e-05,
+      "loss": 0.0359,
+      "step": 985
+    },
+    {
+      "epoch": 58.0,
+      "learning_rate": 3.736842105263158e-05,
+      "loss": 0.0425,
+      "step": 986
+    },
+    {
+      "epoch": 58.06,
+      "learning_rate": 3.735294117647059e-05,
+      "loss": 0.0325,
+      "step": 987
+    },
+    {
+      "epoch": 58.12,
+      "learning_rate": 3.73374613003096e-05,
+      "loss": 0.0305,
+      "step": 988
+    },
+    {
+      "epoch": 58.18,
+      "learning_rate": 3.732198142414861e-05,
+      "loss": 0.0395,
+      "step": 989
+    },
+    {
+      "epoch": 58.24,
+      "learning_rate": 3.7306501547987615e-05,
+      "loss": 0.0323,
+      "step": 990
+    },
+    {
+      "epoch": 58.29,
+      "learning_rate": 3.7291021671826625e-05,
+      "loss": 0.0319,
+      "step": 991
+    },
+    {
+      "epoch": 58.35,
+      "learning_rate": 3.7275541795665635e-05,
+      "loss": 0.0294,
+      "step": 992
+    },
+    {
+      "epoch": 58.41,
+      "learning_rate": 3.7260061919504645e-05,
+      "loss": 0.0308,
+      "step": 993
+    },
+    {
+      "epoch": 58.47,
+      "learning_rate": 3.7244582043343655e-05,
+      "loss": 0.0353,
+      "step": 994
+    },
+    {
+      "epoch": 58.53,
+      "learning_rate": 3.7229102167182665e-05,
+      "loss": 0.0312,
+      "step": 995
+    },
+    {
+      "epoch": 58.59,
+      "learning_rate": 3.721362229102167e-05,
+      "loss": 0.0331,
+      "step": 996
+    },
+    {
+      "epoch": 58.65,
+      "learning_rate": 3.7198142414860685e-05,
+      "loss": 0.034,
+      "step": 997
+    },
+    {
+      "epoch": 58.71,
+      "learning_rate": 3.7182662538699695e-05,
+      "loss": 0.0392,
+      "step": 998
+    },
+    {
+      "epoch": 58.76,
+      "learning_rate": 3.7167182662538705e-05,
+      "loss": 0.0345,
+      "step": 999
+    },
+    {
+      "epoch": 58.82,
+      "learning_rate": 3.7151702786377715e-05,
+      "loss": 0.0284,
+      "step": 1000
+    },
+    {
+      "epoch": 58.82,
+      "eval_accuracy_NAT": 0.9921813393179699,
+      "eval_accuracy_REST": 0.43414634146341463,
+      "eval_accuracy_concrete": 0.9423184820878477,
+      "eval_accuracy_concrete tiles": 0.8988215884987037,
+      "eval_accuracy_dark bricks": 0.8605794077675478,
+      "eval_accuracy_grass": 0.8817253138637507,
+      "eval_accuracy_red bricks": 0.890322934973638,
+      "eval_accuracy_road": 0.9849903885258718,
+      "eval_accuracy_rough concrete": 0.6222239436535791,
+      "eval_accuracy_rough red sidewalk": 0.7714300754254431,
+      "eval_accuracy_speedway bricks": 0.991181720410207,
+      "eval_accuracy_steel": 0.8787052632105035,
+      "eval_accuracy_tiles": 0.21878881517161985,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.981312315122162,
+      "eval_iou_REST": 0.4070121951219512,
+      "eval_iou_concrete": 0.8805584107693506,
+      "eval_iou_concrete tiles": 0.8373109096098535,
+      "eval_iou_dark bricks": 0.8309975001270731,
+      "eval_iou_grass": 0.8225141163396466,
+      "eval_iou_red bricks": 0.8639415902792582,
+      "eval_iou_road": 0.9464345316648095,
+      "eval_iou_rough concrete": 0.6075898545917922,
+      "eval_iou_rough red sidewalk": 0.6204277143410103,
+      "eval_iou_speedway bricks": 0.9753300572952552,
+      "eval_iou_steel": 0.8363440466365238,
+      "eval_iou_tiles": 0.21588497294711984,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08268687129020691,
+      "eval_mean_accuracy": 0.7974935087976996,
+      "eval_mean_iou": 0.7558198626804467,
+      "eval_overall_accuracy": 0.9772288435596531,
+      "eval_runtime": 5.8338,
+      "eval_samples_per_second": 20.227,
+      "eval_steps_per_second": 1.371,
+      "step": 1000
+    },
+    {
+      "epoch": 58.88,
+      "learning_rate": 3.7136222910216725e-05,
+      "loss": 0.0356,
+      "step": 1001
+    },
+    {
+      "epoch": 58.94,
+      "learning_rate": 3.712074303405573e-05,
+      "loss": 0.0352,
+      "step": 1002
+    },
+    {
+      "epoch": 59.0,
+      "learning_rate": 3.710526315789474e-05,
+      "loss": 0.0406,
+      "step": 1003
+    },
+    {
+      "epoch": 59.06,
+      "learning_rate": 3.708978328173375e-05,
+      "loss": 0.0397,
+      "step": 1004
+    },
+    {
+      "epoch": 59.12,
+      "learning_rate": 3.707430340557276e-05,
+      "loss": 0.0319,
+      "step": 1005
+    },
+    {
+      "epoch": 59.18,
+      "learning_rate": 3.705882352941177e-05,
+      "loss": 0.0337,
+      "step": 1006
+    },
+    {
+      "epoch": 59.24,
+      "learning_rate": 3.704334365325078e-05,
+      "loss": 0.0354,
+      "step": 1007
+    },
+    {
+      "epoch": 59.29,
+      "learning_rate": 3.702786377708978e-05,
+      "loss": 0.0315,
+      "step": 1008
+    },
+    {
+      "epoch": 59.35,
+      "learning_rate": 3.701238390092879e-05,
+      "loss": 0.0377,
+      "step": 1009
+    },
+    {
+      "epoch": 59.41,
+      "learning_rate": 3.69969040247678e-05,
+      "loss": 0.0327,
+      "step": 1010
+    },
+    {
+      "epoch": 59.47,
+      "learning_rate": 3.698142414860681e-05,
+      "loss": 0.0303,
+      "step": 1011
+    },
+    {
+      "epoch": 59.53,
+      "learning_rate": 3.696594427244582e-05,
+      "loss": 0.0369,
+      "step": 1012
+    },
+    {
+      "epoch": 59.59,
+      "learning_rate": 3.695046439628483e-05,
+      "loss": 0.0313,
+      "step": 1013
+    },
+    {
+      "epoch": 59.65,
+      "learning_rate": 3.693498452012384e-05,
+      "loss": 0.0363,
+      "step": 1014
+    },
+    {
+      "epoch": 59.71,
+      "learning_rate": 3.691950464396285e-05,
+      "loss": 0.0306,
+      "step": 1015
+    },
+    {
+      "epoch": 59.76,
+      "learning_rate": 3.690402476780186e-05,
+      "loss": 0.0338,
+      "step": 1016
+    },
+    {
+      "epoch": 59.82,
+      "learning_rate": 3.688854489164087e-05,
+      "loss": 0.0349,
+      "step": 1017
+    },
+    {
+      "epoch": 59.88,
+      "learning_rate": 3.687306501547988e-05,
+      "loss": 0.0354,
+      "step": 1018
+    },
+    {
+      "epoch": 59.94,
+      "learning_rate": 3.685758513931889e-05,
+      "loss": 0.0277,
+      "step": 1019
+    },
+    {
+      "epoch": 60.0,
+      "learning_rate": 3.6842105263157895e-05,
+      "loss": 0.0282,
+      "step": 1020
+    },
+    {
+      "epoch": 60.0,
+      "eval_accuracy_NAT": 0.9918096716509706,
+      "eval_accuracy_REST": 0.4016260162601626,
+      "eval_accuracy_concrete": 0.9463885224022296,
+      "eval_accuracy_concrete tiles": 0.9250765967475842,
+      "eval_accuracy_dark bricks": 0.8827402714239229,
+      "eval_accuracy_grass": 0.8949971045430287,
+      "eval_accuracy_red bricks": 0.8991102811950791,
+      "eval_accuracy_road": 0.983558153292992,
+      "eval_accuracy_rough concrete": 0.623308445408446,
+      "eval_accuracy_rough red sidewalk": 0.7761375856338218,
+      "eval_accuracy_speedway bricks": 0.9871097415607801,
+      "eval_accuracy_steel": 0.9167012349627912,
+      "eval_accuracy_tiles": 0.24320639825502136,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9814752227349574,
+      "eval_iou_REST": 0.39268680445151033,
+      "eval_iou_concrete": 0.8841362393150186,
+      "eval_iou_concrete tiles": 0.8398271177011789,
+      "eval_iou_dark bricks": 0.8436370953726825,
+      "eval_iou_grass": 0.8236017348736243,
+      "eval_iou_red bricks": 0.8689029244732233,
+      "eval_iou_road": 0.9497141984320003,
+      "eval_iou_rough concrete": 0.6083691985576364,
+      "eval_iou_rough red sidewalk": 0.6327748042695899,
+      "eval_iou_speedway bricks": 0.9759902446231568,
+      "eval_iou_steel": 0.8398063253322344,
+      "eval_iou_tiles": 0.23962033250753664,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08094064146280289,
+      "eval_mean_accuracy": 0.80552077102591,
+      "eval_mean_iou": 0.7600417109726423,
+      "eval_overall_accuracy": 0.9776828248622054,
+      "eval_runtime": 5.4926,
+      "eval_samples_per_second": 21.484,
+      "eval_steps_per_second": 1.457,
+      "step": 1020
+    },
+    {
+      "epoch": 60.06,
+      "learning_rate": 3.6826625386996905e-05,
+      "loss": 0.0361,
+      "step": 1021
+    },
+    {
+      "epoch": 60.12,
+      "learning_rate": 3.6811145510835915e-05,
+      "loss": 0.0357,
+      "step": 1022
+    },
+    {
+      "epoch": 60.18,
+      "learning_rate": 3.6795665634674925e-05,
+      "loss": 0.0345,
+      "step": 1023
+    },
+    {
+      "epoch": 60.24,
+      "learning_rate": 3.6780185758513935e-05,
+      "loss": 0.0378,
+      "step": 1024
+    },
+    {
+      "epoch": 60.29,
+      "learning_rate": 3.6764705882352945e-05,
+      "loss": 0.0332,
+      "step": 1025
+    },
+    {
+      "epoch": 60.35,
+      "learning_rate": 3.674922600619195e-05,
+      "loss": 0.0356,
+      "step": 1026
+    },
+    {
+      "epoch": 60.41,
+      "learning_rate": 3.673374613003096e-05,
+      "loss": 0.0365,
+      "step": 1027
+    },
+    {
+      "epoch": 60.47,
+      "learning_rate": 3.671826625386997e-05,
+      "loss": 0.0371,
+      "step": 1028
+    },
+    {
+      "epoch": 60.53,
+      "learning_rate": 3.670278637770898e-05,
+      "loss": 0.0294,
+      "step": 1029
+    },
+    {
+      "epoch": 60.59,
+      "learning_rate": 3.668730650154799e-05,
+      "loss": 0.033,
+      "step": 1030
+    },
+    {
+      "epoch": 60.65,
+      "learning_rate": 3.6671826625387e-05,
+      "loss": 0.0325,
+      "step": 1031
+    },
+    {
+      "epoch": 60.71,
+      "learning_rate": 3.665634674922601e-05,
+      "loss": 0.0361,
+      "step": 1032
+    },
+    {
+      "epoch": 60.76,
+      "learning_rate": 3.664086687306502e-05,
+      "loss": 0.0338,
+      "step": 1033
+    },
+    {
+      "epoch": 60.82,
+      "learning_rate": 3.662538699690403e-05,
+      "loss": 0.0291,
+      "step": 1034
+    },
+    {
+      "epoch": 60.88,
+      "learning_rate": 3.660990712074304e-05,
+      "loss": 0.0327,
+      "step": 1035
+    },
+    {
+      "epoch": 60.94,
+      "learning_rate": 3.659442724458205e-05,
+      "loss": 0.0334,
+      "step": 1036
+    },
+    {
+      "epoch": 61.0,
+      "learning_rate": 3.657894736842106e-05,
+      "loss": 0.033,
+      "step": 1037
+    },
+    {
+      "epoch": 61.06,
+      "learning_rate": 3.656346749226006e-05,
+      "loss": 0.033,
+      "step": 1038
+    },
+    {
+      "epoch": 61.12,
+      "learning_rate": 3.654798761609907e-05,
+      "loss": 0.0321,
+      "step": 1039
+    },
+    {
+      "epoch": 61.18,
+      "learning_rate": 3.653250773993808e-05,
+      "loss": 0.0327,
+      "step": 1040
+    },
+    {
+      "epoch": 61.18,
+      "eval_accuracy_NAT": 0.9918416413240966,
+      "eval_accuracy_REST": 0.4113821138211382,
+      "eval_accuracy_concrete": 0.9499039915365788,
+      "eval_accuracy_concrete tiles": 0.9102050436012256,
+      "eval_accuracy_dark bricks": 0.8241056218057922,
+      "eval_accuracy_grass": 0.9009176516897736,
+      "eval_accuracy_red bricks": 0.8756041300527241,
+      "eval_accuracy_road": 0.9789685875541965,
+      "eval_accuracy_rough concrete": 0.6215505991574256,
+      "eval_accuracy_rough red sidewalk": 0.7633310522714113,
+      "eval_accuracy_speedway bricks": 0.9883005580334957,
+      "eval_accuracy_steel": 0.906825530894061,
+      "eval_accuracy_tiles": 0.22954345784483018,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9817055662529158,
+      "eval_iou_REST": 0.4080645161290323,
+      "eval_iou_concrete": 0.8795901885339427,
+      "eval_iou_concrete tiles": 0.8338011140377392,
+      "eval_iou_dark bricks": 0.8083731147818005,
+      "eval_iou_grass": 0.8279318464824104,
+      "eval_iou_red bricks": 0.8530230069555912,
+      "eval_iou_road": 0.9510194461302683,
+      "eval_iou_rough concrete": 0.6061564030892429,
+      "eval_iou_rough red sidewalk": 0.6305957631856868,
+      "eval_iou_speedway bricks": 0.9756481291007124,
+      "eval_iou_steel": 0.8437646774336057,
+      "eval_iou_tiles": 0.22592957032531233,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08245375007390976,
+      "eval_mean_accuracy": 0.7963446138143653,
+      "eval_mean_iou": 0.75581564172602,
+      "eval_overall_accuracy": 0.9775115514205674,
+      "eval_runtime": 5.7704,
+      "eval_samples_per_second": 20.449,
+      "eval_steps_per_second": 1.386,
+      "step": 1040
+    },
+    {
+      "epoch": 61.24,
+      "learning_rate": 3.651702786377709e-05,
+      "loss": 0.0337,
+      "step": 1041
+    },
+    {
+      "epoch": 61.29,
+      "learning_rate": 3.65015479876161e-05,
+      "loss": 0.0332,
+      "step": 1042
+    },
+    {
+      "epoch": 61.35,
+      "learning_rate": 3.648606811145511e-05,
+      "loss": 0.0339,
+      "step": 1043
+    },
+    {
+      "epoch": 61.41,
+      "learning_rate": 3.6470588235294114e-05,
+      "loss": 0.0358,
+      "step": 1044
+    },
+    {
+      "epoch": 61.47,
+      "learning_rate": 3.6455108359133124e-05,
+      "loss": 0.0273,
+      "step": 1045
+    },
+    {
+      "epoch": 61.53,
+      "learning_rate": 3.643962848297214e-05,
+      "loss": 0.0359,
+      "step": 1046
+    },
+    {
+      "epoch": 61.59,
+      "learning_rate": 3.642414860681115e-05,
+      "loss": 0.0344,
+      "step": 1047
+    },
+    {
+      "epoch": 61.65,
+      "learning_rate": 3.640866873065016e-05,
+      "loss": 0.0306,
+      "step": 1048
+    },
+    {
+      "epoch": 61.71,
+      "learning_rate": 3.639318885448917e-05,
+      "loss": 0.0332,
+      "step": 1049
+    },
+    {
+      "epoch": 61.76,
+      "learning_rate": 3.6377708978328174e-05,
+      "loss": 0.0335,
+      "step": 1050
+    },
+    {
+      "epoch": 61.82,
+      "learning_rate": 3.6362229102167184e-05,
+      "loss": 0.0317,
+      "step": 1051
+    },
+    {
+      "epoch": 61.88,
+      "learning_rate": 3.6346749226006194e-05,
+      "loss": 0.0338,
+      "step": 1052
+    },
+    {
+      "epoch": 61.94,
+      "learning_rate": 3.6331269349845204e-05,
+      "loss": 0.0323,
+      "step": 1053
+    },
+    {
+      "epoch": 62.0,
+      "learning_rate": 3.6315789473684214e-05,
+      "loss": 0.034,
+      "step": 1054
+    },
+    {
+      "epoch": 62.06,
+      "learning_rate": 3.630030959752322e-05,
+      "loss": 0.032,
+      "step": 1055
+    },
+    {
+      "epoch": 62.12,
+      "learning_rate": 3.628482972136223e-05,
+      "loss": 0.0381,
+      "step": 1056
+    },
+    {
+      "epoch": 62.18,
+      "learning_rate": 3.626934984520124e-05,
+      "loss": 0.0295,
+      "step": 1057
+    },
+    {
+      "epoch": 62.24,
+      "learning_rate": 3.625386996904025e-05,
+      "loss": 0.0362,
+      "step": 1058
+    },
+    {
+      "epoch": 62.29,
+      "learning_rate": 3.623839009287926e-05,
+      "loss": 0.0334,
+      "step": 1059
+    },
+    {
+      "epoch": 62.35,
+      "learning_rate": 3.622291021671827e-05,
+      "loss": 0.033,
+      "step": 1060
+    },
+    {
+      "epoch": 62.35,
+      "eval_accuracy_NAT": 0.9923818910616945,
+      "eval_accuracy_REST": 0.3024390243902439,
+      "eval_accuracy_concrete": 0.9527479365059952,
+      "eval_accuracy_concrete tiles": 0.8912326184303558,
+      "eval_accuracy_dark bricks": 0.8402082575656069,
+      "eval_accuracy_grass": 0.884502492363942,
+      "eval_accuracy_red bricks": 0.9115224077328646,
+      "eval_accuracy_road": 0.9812949009753629,
+      "eval_accuracy_rough concrete": 0.6220332620263498,
+      "eval_accuracy_rough red sidewalk": 0.7280021657554953,
+      "eval_accuracy_speedway bricks": 0.9887152617122944,
+      "eval_accuracy_steel": 0.9029147320915402,
+      "eval_accuracy_tiles": 0.22975552122148504,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.981627653224521,
+      "eval_iou_REST": 0.2995169082125604,
+      "eval_iou_concrete": 0.8805582323302258,
+      "eval_iou_concrete tiles": 0.8412680756395996,
+      "eval_iou_dark bricks": 0.8209007560066764,
+      "eval_iou_grass": 0.8232177077315448,
+      "eval_iou_red bricks": 0.870502465121158,
+      "eval_iou_road": 0.9529414451163437,
+      "eval_iou_rough concrete": 0.6107477182307512,
+      "eval_iou_rough red sidewalk": 0.6131847807498148,
+      "eval_iou_speedway bricks": 0.9762220118128542,
+      "eval_iou_steel": 0.8434578402971173,
+      "eval_iou_tiles": 0.2261652700325053,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08240743726491928,
+      "eval_mean_accuracy": 0.7867500362948638,
+      "eval_mean_iou": 0.7492546818850516,
+      "eval_overall_accuracy": 0.9776363049523306,
+      "eval_runtime": 8.754,
+      "eval_samples_per_second": 13.48,
+      "eval_steps_per_second": 0.914,
+      "step": 1060
+    },
+    {
+      "epoch": 62.41,
+      "learning_rate": 3.620743034055728e-05,
+      "loss": 0.0358,
+      "step": 1061
+    },
+    {
+      "epoch": 62.47,
+      "learning_rate": 3.619195046439629e-05,
+      "loss": 0.0284,
+      "step": 1062
+    },
+    {
+      "epoch": 62.53,
+      "learning_rate": 3.61764705882353e-05,
+      "loss": 0.0284,
+      "step": 1063
+    },
+    {
+      "epoch": 62.59,
+      "learning_rate": 3.616099071207431e-05,
+      "loss": 0.032,
+      "step": 1064
+    },
+    {
+      "epoch": 62.65,
+      "learning_rate": 3.614551083591332e-05,
+      "loss": 0.0395,
+      "step": 1065
+    },
+    {
+      "epoch": 62.71,
+      "learning_rate": 3.613003095975233e-05,
+      "loss": 0.0343,
+      "step": 1066
+    },
+    {
+      "epoch": 62.76,
+      "learning_rate": 3.611455108359133e-05,
+      "loss": 0.0366,
+      "step": 1067
+    },
+    {
+      "epoch": 62.82,
+      "learning_rate": 3.609907120743034e-05,
+      "loss": 0.0325,
+      "step": 1068
+    },
+    {
+      "epoch": 62.88,
+      "learning_rate": 3.608359133126935e-05,
+      "loss": 0.0324,
+      "step": 1069
+    },
+    {
+      "epoch": 62.94,
+      "learning_rate": 3.606811145510836e-05,
+      "loss": 0.0373,
+      "step": 1070
+    },
+    {
+      "epoch": 63.0,
+      "learning_rate": 3.605263157894737e-05,
+      "loss": 0.0282,
+      "step": 1071
+    },
+    {
+      "epoch": 63.06,
+      "learning_rate": 3.603715170278638e-05,
+      "loss": 0.0337,
+      "step": 1072
+    },
+    {
+      "epoch": 63.12,
+      "learning_rate": 3.6021671826625384e-05,
+      "loss": 0.0357,
+      "step": 1073
+    },
+    {
+      "epoch": 63.18,
+      "learning_rate": 3.6006191950464394e-05,
+      "loss": 0.0292,
+      "step": 1074
+    },
+    {
+      "epoch": 63.24,
+      "learning_rate": 3.5990712074303404e-05,
+      "loss": 0.0352,
+      "step": 1075
+    },
+    {
+      "epoch": 63.29,
+      "learning_rate": 3.5975232198142414e-05,
+      "loss": 0.0284,
+      "step": 1076
+    },
+    {
+      "epoch": 63.35,
+      "learning_rate": 3.5959752321981424e-05,
+      "loss": 0.0357,
+      "step": 1077
+    },
+    {
+      "epoch": 63.41,
+      "learning_rate": 3.5944272445820434e-05,
+      "loss": 0.0294,
+      "step": 1078
+    },
+    {
+      "epoch": 63.47,
+      "learning_rate": 3.5928792569659444e-05,
+      "loss": 0.0306,
+      "step": 1079
+    },
+    {
+      "epoch": 63.53,
+      "learning_rate": 3.5913312693498454e-05,
+      "loss": 0.0368,
+      "step": 1080
+    },
+    {
+      "epoch": 63.53,
+      "eval_accuracy_NAT": 0.9934459474503754,
+      "eval_accuracy_REST": 0.11056910569105691,
+      "eval_accuracy_concrete": 0.9446781805478903,
+      "eval_accuracy_concrete tiles": 0.8895592740985152,
+      "eval_accuracy_dark bricks": 0.8378969431311372,
+      "eval_accuracy_grass": 0.8800883398244558,
+      "eval_accuracy_red bricks": 0.9107535149384886,
+      "eval_accuracy_road": 0.9836361780482161,
+      "eval_accuracy_rough concrete": 0.6245478759854367,
+      "eval_accuracy_rough red sidewalk": 0.7779198219267703,
+      "eval_accuracy_speedway bricks": 0.99053790151713,
+      "eval_accuracy_steel": 0.8842428543583541,
+      "eval_accuracy_tiles": 0.21472931624708413,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9817367145617313,
+      "eval_iou_REST": 0.11056910569105691,
+      "eval_iou_concrete": 0.8837308416053454,
+      "eval_iou_concrete tiles": 0.837917637917638,
+      "eval_iou_dark bricks": 0.8200726884419757,
+      "eval_iou_grass": 0.8245979860738014,
+      "eval_iou_red bricks": 0.8740775880244571,
+      "eval_iou_road": 0.9521563154077513,
+      "eval_iou_rough concrete": 0.6120636292498336,
+      "eval_iou_rough red sidewalk": 0.6341772927905837,
+      "eval_iou_speedway bricks": 0.9760166340414632,
+      "eval_iou_steel": 0.8398827466710973,
+      "eval_iou_tiles": 0.21284646106723523,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08029883354902267,
+      "eval_mean_accuracy": 0.7725080964434545,
+      "eval_mean_iou": 0.7353727416572284,
+      "eval_overall_accuracy": 0.9778749174990896,
+      "eval_runtime": 6.9943,
+      "eval_samples_per_second": 16.871,
+      "eval_steps_per_second": 1.144,
+      "step": 1080
+    },
+    {
+      "epoch": 63.59,
+      "learning_rate": 3.5897832817337464e-05,
+      "loss": 0.0323,
+      "step": 1081
+    },
+    {
+      "epoch": 63.65,
+      "learning_rate": 3.5882352941176474e-05,
+      "loss": 0.0312,
+      "step": 1082
+    },
+    {
+      "epoch": 63.71,
+      "learning_rate": 3.5866873065015484e-05,
+      "loss": 0.041,
+      "step": 1083
+    },
+    {
+      "epoch": 63.76,
+      "learning_rate": 3.5851393188854494e-05,
+      "loss": 0.0355,
+      "step": 1084
+    },
+    {
+      "epoch": 63.82,
+      "learning_rate": 3.58359133126935e-05,
+      "loss": 0.0352,
+      "step": 1085
+    },
+    {
+      "epoch": 63.88,
+      "learning_rate": 3.582043343653251e-05,
+      "loss": 0.0306,
+      "step": 1086
+    },
+    {
+      "epoch": 63.94,
+      "learning_rate": 3.580495356037152e-05,
+      "loss": 0.032,
+      "step": 1087
+    },
+    {
+      "epoch": 64.0,
+      "learning_rate": 3.578947368421053e-05,
+      "loss": 0.0325,
+      "step": 1088
+    },
+    {
+      "epoch": 64.06,
+      "learning_rate": 3.577399380804954e-05,
+      "loss": 0.0285,
+      "step": 1089
+    },
+    {
+      "epoch": 64.12,
+      "learning_rate": 3.575851393188855e-05,
+      "loss": 0.0371,
+      "step": 1090
+    },
+    {
+      "epoch": 64.18,
+      "learning_rate": 3.574303405572755e-05,
+      "loss": 0.0324,
+      "step": 1091
+    },
+    {
+      "epoch": 64.24,
+      "learning_rate": 3.572755417956656e-05,
+      "loss": 0.0266,
+      "step": 1092
+    },
+    {
+      "epoch": 64.29,
+      "learning_rate": 3.571207430340557e-05,
+      "loss": 0.0286,
+      "step": 1093
+    },
+    {
+      "epoch": 64.35,
+      "learning_rate": 3.569659442724458e-05,
+      "loss": 0.0333,
+      "step": 1094
+    },
+    {
+      "epoch": 64.41,
+      "learning_rate": 3.56811145510836e-05,
+      "loss": 0.0316,
+      "step": 1095
+    },
+    {
+      "epoch": 64.47,
+      "learning_rate": 3.566563467492261e-05,
+      "loss": 0.0336,
+      "step": 1096
+    },
+    {
+      "epoch": 64.53,
+      "learning_rate": 3.565015479876161e-05,
+      "loss": 0.0296,
+      "step": 1097
+    },
+    {
+      "epoch": 64.59,
+      "learning_rate": 3.563467492260062e-05,
+      "loss": 0.0284,
+      "step": 1098
+    },
+    {
+      "epoch": 64.65,
+      "learning_rate": 3.561919504643963e-05,
+      "loss": 0.032,
+      "step": 1099
+    },
+    {
+      "epoch": 64.71,
+      "learning_rate": 3.560371517027864e-05,
+      "loss": 0.0395,
+      "step": 1100
+    },
+    {
+      "epoch": 64.71,
+      "eval_accuracy_NAT": 0.9933450785491634,
+      "eval_accuracy_REST": 0.055284552845528454,
+      "eval_accuracy_concrete": 0.9391015106045016,
+      "eval_accuracy_concrete tiles": 0.9200565637520622,
+      "eval_accuracy_dark bricks": 0.8680636640315449,
+      "eval_accuracy_grass": 0.8875058193007755,
+      "eval_accuracy_red bricks": 0.9199802284710018,
+      "eval_accuracy_road": 0.9820190348337778,
+      "eval_accuracy_rough concrete": 0.6581912655897127,
+      "eval_accuracy_rough red sidewalk": 0.7844246083968145,
+      "eval_accuracy_speedway bricks": 0.9909495206231111,
+      "eval_accuracy_steel": 0.8916371378450374,
+      "eval_accuracy_tiles": 0.23357266200127239,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9816389172822669,
+      "eval_iou_REST": 0.055284552845528454,
+      "eval_iou_concrete": 0.8853906696483588,
+      "eval_iou_concrete tiles": 0.8478596094954717,
+      "eval_iou_dark bricks": 0.8406523129396717,
+      "eval_iou_grass": 0.8273295245555029,
+      "eval_iou_red bricks": 0.8751828631138976,
+      "eval_iou_road": 0.9525531288424973,
+      "eval_iou_rough concrete": 0.644120477009651,
+      "eval_iou_rough red sidewalk": 0.6287756861184952,
+      "eval_iou_speedway bricks": 0.9760494212665044,
+      "eval_iou_steel": 0.8417931011704087,
+      "eval_iou_tiles": 0.23059668012561688,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0765690729022026,
+      "eval_mean_accuracy": 0.7787793574495617,
+      "eval_mean_iou": 0.7374789957241441,
+      "eval_overall_accuracy": 0.9781521942655919,
+      "eval_runtime": 5.6286,
+      "eval_samples_per_second": 20.964,
+      "eval_steps_per_second": 1.421,
+      "step": 1100
+    },
+    {
+      "epoch": 64.76,
+      "learning_rate": 3.558823529411765e-05,
+      "loss": 0.0287,
+      "step": 1101
+    },
+    {
+      "epoch": 64.82,
+      "learning_rate": 3.557275541795666e-05,
+      "loss": 0.036,
+      "step": 1102
+    },
+    {
+      "epoch": 64.88,
+      "learning_rate": 3.5557275541795664e-05,
+      "loss": 0.0326,
+      "step": 1103
+    },
+    {
+      "epoch": 64.94,
+      "learning_rate": 3.5541795665634674e-05,
+      "loss": 0.0356,
+      "step": 1104
+    },
+    {
+      "epoch": 65.0,
+      "learning_rate": 3.5526315789473684e-05,
+      "loss": 0.0336,
+      "step": 1105
+    },
+    {
+      "epoch": 65.06,
+      "learning_rate": 3.5510835913312694e-05,
+      "loss": 0.0334,
+      "step": 1106
+    },
+    {
+      "epoch": 65.12,
+      "learning_rate": 3.5495356037151704e-05,
+      "loss": 0.031,
+      "step": 1107
+    },
+    {
+      "epoch": 65.18,
+      "learning_rate": 3.5479876160990714e-05,
+      "loss": 0.0319,
+      "step": 1108
+    },
+    {
+      "epoch": 65.24,
+      "learning_rate": 3.5464396284829724e-05,
+      "loss": 0.029,
+      "step": 1109
+    },
+    {
+      "epoch": 65.29,
+      "learning_rate": 3.5448916408668734e-05,
+      "loss": 0.0344,
+      "step": 1110
+    },
+    {
+      "epoch": 65.35,
+      "learning_rate": 3.5433436532507744e-05,
+      "loss": 0.0335,
+      "step": 1111
+    },
+    {
+      "epoch": 65.41,
+      "learning_rate": 3.5417956656346754e-05,
+      "loss": 0.0339,
+      "step": 1112
+    },
+    {
+      "epoch": 65.47,
+      "learning_rate": 3.5402476780185764e-05,
+      "loss": 0.0353,
+      "step": 1113
+    },
+    {
+      "epoch": 65.53,
+      "learning_rate": 3.5386996904024774e-05,
+      "loss": 0.0325,
+      "step": 1114
+    },
+    {
+      "epoch": 65.59,
+      "learning_rate": 3.537151702786378e-05,
+      "loss": 0.0299,
+      "step": 1115
+    },
+    {
+      "epoch": 65.65,
+      "learning_rate": 3.535603715170279e-05,
+      "loss": 0.0298,
+      "step": 1116
+    },
+    {
+      "epoch": 65.71,
+      "learning_rate": 3.53405572755418e-05,
+      "loss": 0.0343,
+      "step": 1117
+    },
+    {
+      "epoch": 65.76,
+      "learning_rate": 3.532507739938081e-05,
+      "loss": 0.0285,
+      "step": 1118
+    },
+    {
+      "epoch": 65.82,
+      "learning_rate": 3.530959752321982e-05,
+      "loss": 0.0274,
+      "step": 1119
+    },
+    {
+      "epoch": 65.88,
+      "learning_rate": 3.529411764705883e-05,
+      "loss": 0.0355,
+      "step": 1120
+    },
+    {
+      "epoch": 65.88,
+      "eval_accuracy_NAT": 0.9935288098240596,
+      "eval_accuracy_REST": 0.15772357723577235,
+      "eval_accuracy_concrete": 0.947173941957686,
+      "eval_accuracy_concrete tiles": 0.9117841150129625,
+      "eval_accuracy_dark bricks": 0.8493243113909997,
+      "eval_accuracy_grass": 0.8630335305272002,
+      "eval_accuracy_red bricks": 0.9045474516695958,
+      "eval_accuracy_road": 0.9832118516396688,
+      "eval_accuracy_rough concrete": 0.6616652464857972,
+      "eval_accuracy_rough red sidewalk": 0.7929823505967107,
+      "eval_accuracy_speedway bricks": 0.9890675884741167,
+      "eval_accuracy_steel": 0.8980943289668994,
+      "eval_accuracy_tiles": 0.23039171135144962,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9811323848611445,
+      "eval_iou_REST": 0.15746753246753248,
+      "eval_iou_concrete": 0.8863057884077346,
+      "eval_iou_concrete tiles": 0.8400356103704347,
+      "eval_iou_dark bricks": 0.8323758248251864,
+      "eval_iou_grass": 0.815390612009176,
+      "eval_iou_red bricks": 0.8783531544984268,
+      "eval_iou_road": 0.9513252874989658,
+      "eval_iou_rough concrete": 0.6450675915114125,
+      "eval_iou_rough red sidewalk": 0.62166910342876,
+      "eval_iou_speedway bricks": 0.9768907139886711,
+      "eval_iou_steel": 0.8456050859501099,
+      "eval_iou_tiles": 0.22557394554191137,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07918473333120346,
+      "eval_mean_accuracy": 0.7832714473179168,
+      "eval_mean_iou": 0.7428609719507281,
+      "eval_overall_accuracy": 0.9778169858253608,
+      "eval_runtime": 5.5719,
+      "eval_samples_per_second": 21.178,
+      "eval_steps_per_second": 1.436,
+      "step": 1120
+    },
+    {
+      "epoch": 65.94,
+      "learning_rate": 3.527863777089783e-05,
+      "loss": 0.0304,
+      "step": 1121
+    },
+    {
+      "epoch": 66.0,
+      "learning_rate": 3.526315789473684e-05,
+      "loss": 0.0325,
+      "step": 1122
+    },
+    {
+      "epoch": 66.06,
+      "learning_rate": 3.524767801857585e-05,
+      "loss": 0.0322,
+      "step": 1123
+    },
+    {
+      "epoch": 66.12,
+      "learning_rate": 3.523219814241486e-05,
+      "loss": 0.0287,
+      "step": 1124
+    },
+    {
+      "epoch": 66.18,
+      "learning_rate": 3.521671826625387e-05,
+      "loss": 0.0385,
+      "step": 1125
+    },
+    {
+      "epoch": 66.24,
+      "learning_rate": 3.520123839009288e-05,
+      "loss": 0.0318,
+      "step": 1126
+    },
+    {
+      "epoch": 66.29,
+      "learning_rate": 3.518575851393189e-05,
+      "loss": 0.026,
+      "step": 1127
+    },
+    {
+      "epoch": 66.35,
+      "learning_rate": 3.51702786377709e-05,
+      "loss": 0.0298,
+      "step": 1128
+    },
+    {
+      "epoch": 66.41,
+      "learning_rate": 3.515479876160991e-05,
+      "loss": 0.0311,
+      "step": 1129
+    },
+    {
+      "epoch": 66.47,
+      "learning_rate": 3.513931888544892e-05,
+      "loss": 0.0336,
+      "step": 1130
+    },
+    {
+      "epoch": 66.53,
+      "learning_rate": 3.512383900928793e-05,
+      "loss": 0.0332,
+      "step": 1131
+    },
+    {
+      "epoch": 66.59,
+      "learning_rate": 3.510835913312694e-05,
+      "loss": 0.0305,
+      "step": 1132
+    },
+    {
+      "epoch": 66.65,
+      "learning_rate": 3.5092879256965944e-05,
+      "loss": 0.0365,
+      "step": 1133
+    },
+    {
+      "epoch": 66.71,
+      "learning_rate": 3.5077399380804954e-05,
+      "loss": 0.0289,
+      "step": 1134
+    },
+    {
+      "epoch": 66.76,
+      "learning_rate": 3.5061919504643964e-05,
+      "loss": 0.0302,
+      "step": 1135
+    },
+    {
+      "epoch": 66.82,
+      "learning_rate": 3.5046439628482974e-05,
+      "loss": 0.0345,
+      "step": 1136
+    },
+    {
+      "epoch": 66.88,
+      "learning_rate": 3.5030959752321984e-05,
+      "loss": 0.0273,
+      "step": 1137
+    },
+    {
+      "epoch": 66.94,
+      "learning_rate": 3.5015479876160994e-05,
+      "loss": 0.0344,
+      "step": 1138
+    },
+    {
+      "epoch": 67.0,
+      "learning_rate": 3.5e-05,
+      "loss": 0.0459,
+      "step": 1139
+    },
+    {
+      "epoch": 67.06,
+      "learning_rate": 3.498452012383901e-05,
+      "loss": 0.0326,
+      "step": 1140
+    },
+    {
+      "epoch": 67.06,
+      "eval_accuracy_NAT": 0.9919153386987394,
+      "eval_accuracy_REST": 0.191869918699187,
+      "eval_accuracy_concrete": 0.9546646659807463,
+      "eval_accuracy_concrete tiles": 0.8761253829837379,
+      "eval_accuracy_dark bricks": 0.8499320483126926,
+      "eval_accuracy_grass": 0.8893254152993676,
+      "eval_accuracy_red bricks": 0.9099297012302284,
+      "eval_accuracy_road": 0.9822333357025782,
+      "eval_accuracy_rough concrete": 0.6300418903699819,
+      "eval_accuracy_rough red sidewalk": 0.7587739417502012,
+      "eval_accuracy_speedway bricks": 0.9882316692405713,
+      "eval_accuracy_steel": 0.902150064721846,
+      "eval_accuracy_tiles": 0.1801023963161562,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9819518422256411,
+      "eval_iou_REST": 0.191869918699187,
+      "eval_iou_concrete": 0.8805101971033416,
+      "eval_iou_concrete tiles": 0.8212707670554966,
+      "eval_iou_dark bricks": 0.8274994525640966,
+      "eval_iou_grass": 0.8292444489932045,
+      "eval_iou_red bricks": 0.8830615073019934,
+      "eval_iou_road": 0.9491662320090478,
+      "eval_iou_rough concrete": 0.6198658646687069,
+      "eval_iou_rough red sidewalk": 0.6252889376390464,
+      "eval_iou_speedway bricks": 0.9767151019960202,
+      "eval_iou_steel": 0.8457860285538912,
+      "eval_iou_tiles": 0.17682926829268292,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07983653992414474,
+      "eval_mean_accuracy": 0.7773304437927717,
+      "eval_mean_iou": 0.7391584282386428,
+      "eval_overall_accuracy": 0.9778036667128741,
+      "eval_runtime": 5.6682,
+      "eval_samples_per_second": 20.818,
+      "eval_steps_per_second": 1.411,
+      "step": 1140
+    },
+    {
+      "epoch": 67.12,
+      "learning_rate": 3.496904024767802e-05,
+      "loss": 0.0302,
+      "step": 1141
+    },
+    {
+      "epoch": 67.18,
+      "learning_rate": 3.495356037151703e-05,
+      "loss": 0.0326,
+      "step": 1142
+    },
+    {
+      "epoch": 67.24,
+      "learning_rate": 3.493808049535604e-05,
+      "loss": 0.0379,
+      "step": 1143
+    },
+    {
+      "epoch": 67.29,
+      "learning_rate": 3.4922600619195054e-05,
+      "loss": 0.0328,
+      "step": 1144
+    },
+    {
+      "epoch": 67.35,
+      "learning_rate": 3.490712074303406e-05,
+      "loss": 0.0247,
+      "step": 1145
+    },
+    {
+      "epoch": 67.41,
+      "learning_rate": 3.489164086687307e-05,
+      "loss": 0.0304,
+      "step": 1146
+    },
+    {
+      "epoch": 67.47,
+      "learning_rate": 3.487616099071208e-05,
+      "loss": 0.033,
+      "step": 1147
+    },
+    {
+      "epoch": 67.53,
+      "learning_rate": 3.486068111455109e-05,
+      "loss": 0.0538,
+      "step": 1148
+    },
+    {
+      "epoch": 67.59,
+      "learning_rate": 3.48452012383901e-05,
+      "loss": 0.0312,
+      "step": 1149
+    },
+    {
+      "epoch": 67.65,
+      "learning_rate": 3.482972136222911e-05,
+      "loss": 0.0284,
+      "step": 1150
+    },
+    {
+      "epoch": 67.71,
+      "learning_rate": 3.481424148606811e-05,
+      "loss": 0.034,
+      "step": 1151
+    },
+    {
+      "epoch": 67.76,
+      "learning_rate": 3.479876160990712e-05,
+      "loss": 0.031,
+      "step": 1152
+    },
+    {
+      "epoch": 67.82,
+      "learning_rate": 3.478328173374613e-05,
+      "loss": 0.03,
+      "step": 1153
+    },
+    {
+      "epoch": 67.88,
+      "learning_rate": 3.476780185758514e-05,
+      "loss": 0.031,
+      "step": 1154
+    },
+    {
+      "epoch": 67.94,
+      "learning_rate": 3.475232198142415e-05,
+      "loss": 0.0372,
+      "step": 1155
+    },
+    {
+      "epoch": 68.0,
+      "learning_rate": 3.473684210526316e-05,
+      "loss": 0.0288,
+      "step": 1156
+    },
+    {
+      "epoch": 68.06,
+      "learning_rate": 3.4721362229102163e-05,
+      "loss": 0.0307,
+      "step": 1157
+    },
+    {
+      "epoch": 68.12,
+      "learning_rate": 3.470588235294118e-05,
+      "loss": 0.0331,
+      "step": 1158
+    },
+    {
+      "epoch": 68.18,
+      "learning_rate": 3.469040247678019e-05,
+      "loss": 0.032,
+      "step": 1159
+    },
+    {
+      "epoch": 68.24,
+      "learning_rate": 3.46749226006192e-05,
+      "loss": 0.03,
+      "step": 1160
+    },
+    {
+      "epoch": 68.24,
+      "eval_accuracy_NAT": 0.9934777014764652,
+      "eval_accuracy_REST": 0.3056910569105691,
+      "eval_accuracy_concrete": 0.9438457504788957,
+      "eval_accuracy_concrete tiles": 0.9096629743106293,
+      "eval_accuracy_dark bricks": 0.8682311505847673,
+      "eval_accuracy_grass": 0.8853427426222621,
+      "eval_accuracy_red bricks": 0.9111928822495606,
+      "eval_accuracy_road": 0.9814338491695975,
+      "eval_accuracy_rough concrete": 0.7519768321822916,
+      "eval_accuracy_rough red sidewalk": 0.7964265034328729,
+      "eval_accuracy_speedway bricks": 0.9891467591764328,
+      "eval_accuracy_steel": 0.9062058004767926,
+      "eval_accuracy_tiles": 0.22775606652731073,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9820215251429533,
+      "eval_iou_REST": 0.29984051036682613,
+      "eval_iou_concrete": 0.8899700377609263,
+      "eval_iou_concrete tiles": 0.8444809101848813,
+      "eval_iou_dark bricks": 0.8442967761149579,
+      "eval_iou_grass": 0.828061638998384,
+      "eval_iou_red bricks": 0.8802992518703242,
+      "eval_iou_road": 0.9556096261359331,
+      "eval_iou_rough concrete": 0.738022831477496,
+      "eval_iou_rough red sidewalk": 0.6150754702735979,
+      "eval_iou_speedway bricks": 0.976805894926519,
+      "eval_iou_steel": 0.8482410179640718,
+      "eval_iou_tiles": 0.22415026833631485,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07349683344364166,
+      "eval_mean_accuracy": 0.8054146207383421,
+      "eval_mean_iou": 0.7636058276579374,
+      "eval_overall_accuracy": 0.9789129677465407,
+      "eval_runtime": 5.4514,
+      "eval_samples_per_second": 21.646,
+      "eval_steps_per_second": 1.468,
+      "step": 1160
+    },
+    {
+      "epoch": 68.29,
+      "learning_rate": 3.465944272445821e-05,
+      "loss": 0.032,
+      "step": 1161
+    },
+    {
+      "epoch": 68.35,
+      "learning_rate": 3.464396284829722e-05,
+      "loss": 0.0315,
+      "step": 1162
+    },
+    {
+      "epoch": 68.41,
+      "learning_rate": 3.4628482972136223e-05,
+      "loss": 0.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 68.47,
+      "learning_rate": 3.4613003095975233e-05,
+      "loss": 0.0333,
+      "step": 1164
+    },
+    {
+      "epoch": 68.53,
+      "learning_rate": 3.4597523219814243e-05,
+      "loss": 0.033,
+      "step": 1165
+    },
+    {
+      "epoch": 68.59,
+      "learning_rate": 3.4582043343653253e-05,
+      "loss": 0.0376,
+      "step": 1166
+    },
+    {
+      "epoch": 68.65,
+      "learning_rate": 3.4566563467492263e-05,
+      "loss": 0.0414,
+      "step": 1167
+    },
+    {
+      "epoch": 68.71,
+      "learning_rate": 3.4551083591331273e-05,
+      "loss": 0.033,
+      "step": 1168
+    },
+    {
+      "epoch": 68.76,
+      "learning_rate": 3.453560371517028e-05,
+      "loss": 0.0398,
+      "step": 1169
+    },
+    {
+      "epoch": 68.82,
+      "learning_rate": 3.452012383900929e-05,
+      "loss": 0.0334,
+      "step": 1170
+    },
+    {
+      "epoch": 68.88,
+      "learning_rate": 3.45046439628483e-05,
+      "loss": 0.0396,
+      "step": 1171
+    },
+    {
+      "epoch": 68.94,
+      "learning_rate": 3.448916408668731e-05,
+      "loss": 0.0335,
+      "step": 1172
+    },
+    {
+      "epoch": 69.0,
+      "learning_rate": 3.447368421052632e-05,
+      "loss": 0.0326,
+      "step": 1173
+    },
+    {
+      "epoch": 69.06,
+      "learning_rate": 3.445820433436533e-05,
+      "loss": 0.0343,
+      "step": 1174
+    },
+    {
+      "epoch": 69.12,
+      "learning_rate": 3.444272445820434e-05,
+      "loss": 0.0459,
+      "step": 1175
+    },
+    {
+      "epoch": 69.18,
+      "learning_rate": 3.442724458204335e-05,
+      "loss": 0.0314,
+      "step": 1176
+    },
+    {
+      "epoch": 69.24,
+      "learning_rate": 3.441176470588236e-05,
+      "loss": 0.0305,
+      "step": 1177
+    },
+    {
+      "epoch": 69.29,
+      "learning_rate": 3.439628482972137e-05,
+      "loss": 0.0354,
+      "step": 1178
+    },
+    {
+      "epoch": 69.35,
+      "learning_rate": 3.438080495356038e-05,
+      "loss": 0.0398,
+      "step": 1179
+    },
+    {
+      "epoch": 69.41,
+      "learning_rate": 3.436532507739939e-05,
+      "loss": 0.0304,
+      "step": 1180
+    },
+    {
+      "epoch": 69.41,
+      "eval_accuracy_NAT": 0.9931456589523792,
+      "eval_accuracy_REST": 0.2634146341463415,
+      "eval_accuracy_concrete": 0.945193767399623,
+      "eval_accuracy_concrete tiles": 0.9119019561630921,
+      "eval_accuracy_dark bricks": 0.888994697854258,
+      "eval_accuracy_grass": 0.8893727266877864,
+      "eval_accuracy_red bricks": 0.9082271528998243,
+      "eval_accuracy_road": 0.9828548616637123,
+      "eval_accuracy_rough concrete": 0.6426328365679691,
+      "eval_accuracy_rough red sidewalk": 0.7215800991133938,
+      "eval_accuracy_speedway bricks": 0.9898546686380514,
+      "eval_accuracy_steel": 0.9067880471994683,
+      "eval_accuracy_tiles": 0.2100033324244903,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.98200447224178,
+      "eval_iou_REST": 0.26129032258064516,
+      "eval_iou_concrete": 0.8846762397166505,
+      "eval_iou_concrete tiles": 0.8445084686572376,
+      "eval_iou_dark bricks": 0.8599699108899433,
+      "eval_iou_grass": 0.8291459103928835,
+      "eval_iou_red bricks": 0.8779464854533872,
+      "eval_iou_road": 0.9485877067067877,
+      "eval_iou_rough concrete": 0.6299121536376805,
+      "eval_iou_rough red sidewalk": 0.622748778255875,
+      "eval_iou_speedway bricks": 0.977983933489584,
+      "eval_iou_steel": 0.8494635738180013,
+      "eval_iou_tiles": 0.20392433736357485,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07490196824073792,
+      "eval_mean_accuracy": 0.7887664953623378,
+      "eval_mean_iou": 0.7517047917849256,
+      "eval_overall_accuracy": 0.978399438373113,
+      "eval_runtime": 7.2052,
+      "eval_samples_per_second": 16.377,
+      "eval_steps_per_second": 1.11,
+      "step": 1180
+    },
+    {
+      "epoch": 69.47,
+      "learning_rate": 3.434984520123839e-05,
+      "loss": 0.0381,
+      "step": 1181
+    },
+    {
+      "epoch": 69.53,
+      "learning_rate": 3.43343653250774e-05,
+      "loss": 0.0305,
+      "step": 1182
+    },
+    {
+      "epoch": 69.59,
+      "learning_rate": 3.431888544891641e-05,
+      "loss": 0.0355,
+      "step": 1183
+    },
+    {
+      "epoch": 69.65,
+      "learning_rate": 3.430340557275542e-05,
+      "loss": 0.0428,
+      "step": 1184
+    },
+    {
+      "epoch": 69.71,
+      "learning_rate": 3.428792569659443e-05,
+      "loss": 0.038,
+      "step": 1185
+    },
+    {
+      "epoch": 69.76,
+      "learning_rate": 3.427244582043344e-05,
+      "loss": 0.0352,
+      "step": 1186
+    },
+    {
+      "epoch": 69.82,
+      "learning_rate": 3.425696594427244e-05,
+      "loss": 0.034,
+      "step": 1187
+    },
+    {
+      "epoch": 69.88,
+      "learning_rate": 3.424148606811145e-05,
+      "loss": 0.035,
+      "step": 1188
+    },
+    {
+      "epoch": 69.94,
+      "learning_rate": 3.422600619195046e-05,
+      "loss": 0.0308,
+      "step": 1189
+    },
+    {
+      "epoch": 70.0,
+      "learning_rate": 3.421052631578947e-05,
+      "loss": 0.0307,
+      "step": 1190
+    },
+    {
+      "epoch": 70.06,
+      "learning_rate": 3.419504643962848e-05,
+      "loss": 0.031,
+      "step": 1191
+    },
+    {
+      "epoch": 70.12,
+      "learning_rate": 3.41795665634675e-05,
+      "loss": 0.0365,
+      "step": 1192
+    },
+    {
+      "epoch": 70.18,
+      "learning_rate": 3.41640866873065e-05,
+      "loss": 0.0376,
+      "step": 1193
+    },
+    {
+      "epoch": 70.24,
+      "learning_rate": 3.414860681114551e-05,
+      "loss": 0.0354,
+      "step": 1194
+    },
+    {
+      "epoch": 70.29,
+      "learning_rate": 3.413312693498452e-05,
+      "loss": 0.0276,
+      "step": 1195
+    },
+    {
+      "epoch": 70.35,
+      "learning_rate": 3.411764705882353e-05,
+      "loss": 0.0324,
+      "step": 1196
+    },
+    {
+      "epoch": 70.41,
+      "learning_rate": 3.410216718266254e-05,
+      "loss": 0.0336,
+      "step": 1197
+    },
+    {
+      "epoch": 70.47,
+      "learning_rate": 3.4086687306501546e-05,
+      "loss": 0.0322,
+      "step": 1198
+    },
+    {
+      "epoch": 70.53,
+      "learning_rate": 3.4071207430340556e-05,
+      "loss": 0.0346,
+      "step": 1199
+    },
+    {
+      "epoch": 70.59,
+      "learning_rate": 3.4055727554179566e-05,
+      "loss": 0.0366,
+      "step": 1200
+    },
+    {
+      "epoch": 70.59,
+      "eval_accuracy_NAT": 0.992674955383976,
+      "eval_accuracy_REST": 0.04715447154471545,
+      "eval_accuracy_concrete": 0.958235171813872,
+      "eval_accuracy_concrete tiles": 0.8674994107942493,
+      "eval_accuracy_dark bricks": 0.8251440384357713,
+      "eval_accuracy_grass": 0.8618933260663041,
+      "eval_accuracy_red bricks": 0.9024055360281195,
+      "eval_accuracy_road": 0.9774027482883987,
+      "eval_accuracy_rough concrete": 0.6224027076791067,
+      "eval_accuracy_rough red sidewalk": 0.8008482542356312,
+      "eval_accuracy_speedway bricks": 0.9870333127009685,
+      "eval_accuracy_steel": 0.9060183820038283,
+      "eval_accuracy_tiles": 0.14244599957587326,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9819493271821238,
+      "eval_iou_REST": 0.04707792207792208,
+      "eval_iou_concrete": 0.8767137131319575,
+      "eval_iou_concrete tiles": 0.8272390156197326,
+      "eval_iou_dark bricks": 0.8106283554443995,
+      "eval_iou_grass": 0.8130420850341017,
+      "eval_iou_red bricks": 0.8766940561306157,
+      "eval_iou_road": 0.9598719622793611,
+      "eval_iou_rough concrete": 0.6054954928842642,
+      "eval_iou_rough red sidewalk": 0.5239165834436141,
+      "eval_iou_speedway bricks": 0.9776720750888546,
+      "eval_iou_steel": 0.8490466622641244,
+      "eval_iou_tiles": 0.13535219781801433,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08301250636577606,
+      "eval_mean_accuracy": 0.7608583318885243,
+      "eval_mean_iou": 0.7142076498768526,
+      "eval_overall_accuracy": 0.9770683676509534,
+      "eval_runtime": 5.8395,
+      "eval_samples_per_second": 20.207,
+      "eval_steps_per_second": 1.37,
+      "step": 1200
+    },
+    {
+      "epoch": 70.65,
+      "learning_rate": 3.4040247678018576e-05,
+      "loss": 0.0324,
+      "step": 1201
+    },
+    {
+      "epoch": 70.71,
+      "learning_rate": 3.4024767801857586e-05,
+      "loss": 0.0378,
+      "step": 1202
+    },
+    {
+      "epoch": 70.76,
+      "learning_rate": 3.4009287925696596e-05,
+      "loss": 0.0355,
+      "step": 1203
+    },
+    {
+      "epoch": 70.82,
+      "learning_rate": 3.39938080495356e-05,
+      "loss": 0.038,
+      "step": 1204
+    },
+    {
+      "epoch": 70.88,
+      "learning_rate": 3.397832817337461e-05,
+      "loss": 0.039,
+      "step": 1205
+    },
+    {
+      "epoch": 70.94,
+      "learning_rate": 3.396284829721362e-05,
+      "loss": 0.0297,
+      "step": 1206
+    },
+    {
+      "epoch": 71.0,
+      "learning_rate": 3.3947368421052636e-05,
+      "loss": 0.0321,
+      "step": 1207
+    },
+    {
+      "epoch": 71.06,
+      "learning_rate": 3.3931888544891646e-05,
+      "loss": 0.0328,
+      "step": 1208
+    },
+    {
+      "epoch": 71.12,
+      "learning_rate": 3.3916408668730656e-05,
+      "loss": 0.0335,
+      "step": 1209
+    },
+    {
+      "epoch": 71.18,
+      "learning_rate": 3.390092879256966e-05,
+      "loss": 0.0323,
+      "step": 1210
+    },
+    {
+      "epoch": 71.24,
+      "learning_rate": 3.388544891640867e-05,
+      "loss": 0.0333,
+      "step": 1211
+    },
+    {
+      "epoch": 71.29,
+      "learning_rate": 3.386996904024768e-05,
+      "loss": 0.0339,
+      "step": 1212
+    },
+    {
+      "epoch": 71.35,
+      "learning_rate": 3.385448916408669e-05,
+      "loss": 0.0317,
+      "step": 1213
+    },
+    {
+      "epoch": 71.41,
+      "learning_rate": 3.38390092879257e-05,
+      "loss": 0.0318,
+      "step": 1214
+    },
+    {
+      "epoch": 71.47,
+      "learning_rate": 3.382352941176471e-05,
+      "loss": 0.0317,
+      "step": 1215
+    },
+    {
+      "epoch": 71.53,
+      "learning_rate": 3.380804953560371e-05,
+      "loss": 0.0288,
+      "step": 1216
+    },
+    {
+      "epoch": 71.59,
+      "learning_rate": 3.379256965944272e-05,
+      "loss": 0.0303,
+      "step": 1217
+    },
+    {
+      "epoch": 71.65,
+      "learning_rate": 3.377708978328173e-05,
+      "loss": 0.0386,
+      "step": 1218
+    },
+    {
+      "epoch": 71.71,
+      "learning_rate": 3.376160990712074e-05,
+      "loss": 0.0318,
+      "step": 1219
+    },
+    {
+      "epoch": 71.76,
+      "learning_rate": 3.374613003095975e-05,
+      "loss": 0.0296,
+      "step": 1220
+    },
+    {
+      "epoch": 71.76,
+      "eval_accuracy_NAT": 0.9916076642897511,
+      "eval_accuracy_REST": 0.14308943089430895,
+      "eval_accuracy_concrete": 0.9524127477224744,
+      "eval_accuracy_concrete tiles": 0.9023332547725665,
+      "eval_accuracy_dark bricks": 0.8711693432612981,
+      "eval_accuracy_grass": 0.9014191524070142,
+      "eval_accuracy_red bricks": 0.890322934973638,
+      "eval_accuracy_road": 0.9790663857062923,
+      "eval_accuracy_rough concrete": 0.6242618535445927,
+      "eval_accuracy_rough red sidewalk": 0.7556456282570932,
+      "eval_accuracy_speedway bricks": 0.991555981912065,
+      "eval_accuracy_steel": 0.885989594526381,
+      "eval_accuracy_tiles": 0.17537641249356237,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9818573901591024,
+      "eval_iou_REST": 0.14308943089430895,
+      "eval_iou_concrete": 0.8808498965890104,
+      "eval_iou_concrete tiles": 0.839660503980525,
+      "eval_iou_dark bricks": 0.8471303198667299,
+      "eval_iou_grass": 0.8321373634388728,
+      "eval_iou_red bricks": 0.8664813726014218,
+      "eval_iou_road": 0.9586484952029115,
+      "eval_iou_rough concrete": 0.6103291581706962,
+      "eval_iou_rough red sidewalk": 0.6133155109589292,
+      "eval_iou_speedway bricks": 0.977225058515616,
+      "eval_iou_steel": 0.8448727388841627,
+      "eval_iou_tiles": 0.1710192023633678,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07912295311689377,
+      "eval_mean_accuracy": 0.7741731065200798,
+      "eval_mean_iou": 0.7358935724327427,
+      "eval_overall_accuracy": 0.978185427390923,
+      "eval_runtime": 13.2108,
+      "eval_samples_per_second": 8.932,
+      "eval_steps_per_second": 0.606,
+      "step": 1220
+    },
+    {
+      "epoch": 71.82,
+      "learning_rate": 3.373065015479876e-05,
+      "loss": 0.0351,
+      "step": 1221
+    },
+    {
+      "epoch": 71.88,
+      "learning_rate": 3.371517027863777e-05,
+      "loss": 0.027,
+      "step": 1222
+    },
+    {
+      "epoch": 71.94,
+      "learning_rate": 3.369969040247678e-05,
+      "loss": 0.0298,
+      "step": 1223
+    },
+    {
+      "epoch": 72.0,
+      "learning_rate": 3.368421052631579e-05,
+      "loss": 0.0299,
+      "step": 1224
+    },
+    {
+      "epoch": 72.06,
+      "learning_rate": 3.36687306501548e-05,
+      "loss": 0.0306,
+      "step": 1225
+    },
+    {
+      "epoch": 72.12,
+      "learning_rate": 3.365325077399381e-05,
+      "loss": 0.0309,
+      "step": 1226
+    },
+    {
+      "epoch": 72.18,
+      "learning_rate": 3.363777089783282e-05,
+      "loss": 0.0277,
+      "step": 1227
+    },
+    {
+      "epoch": 72.24,
+      "learning_rate": 3.3622291021671826e-05,
+      "loss": 0.0363,
+      "step": 1228
+    },
+    {
+      "epoch": 72.29,
+      "learning_rate": 3.3606811145510836e-05,
+      "loss": 0.0299,
+      "step": 1229
+    },
+    {
+      "epoch": 72.35,
+      "learning_rate": 3.3591331269349846e-05,
+      "loss": 0.032,
+      "step": 1230
+    },
+    {
+      "epoch": 72.41,
+      "learning_rate": 3.3575851393188856e-05,
+      "loss": 0.0406,
+      "step": 1231
+    },
+    {
+      "epoch": 72.47,
+      "learning_rate": 3.3560371517027866e-05,
+      "loss": 0.0306,
+      "step": 1232
+    },
+    {
+      "epoch": 72.53,
+      "learning_rate": 3.3544891640866876e-05,
+      "loss": 0.0264,
+      "step": 1233
+    },
+    {
+      "epoch": 72.59,
+      "learning_rate": 3.352941176470588e-05,
+      "loss": 0.0306,
+      "step": 1234
+    },
+    {
+      "epoch": 72.65,
+      "learning_rate": 3.351393188854489e-05,
+      "loss": 0.0321,
+      "step": 1235
+    },
+    {
+      "epoch": 72.71,
+      "learning_rate": 3.34984520123839e-05,
+      "loss": 0.0303,
+      "step": 1236
+    },
+    {
+      "epoch": 72.76,
+      "learning_rate": 3.348297213622291e-05,
+      "loss": 0.0339,
+      "step": 1237
+    },
+    {
+      "epoch": 72.82,
+      "learning_rate": 3.346749226006192e-05,
+      "loss": 0.0351,
+      "step": 1238
+    },
+    {
+      "epoch": 72.88,
+      "learning_rate": 3.345201238390093e-05,
+      "loss": 0.0348,
+      "step": 1239
+    },
+    {
+      "epoch": 72.94,
+      "learning_rate": 3.343653250773994e-05,
+      "loss": 0.0313,
+      "step": 1240
+    },
+    {
+      "epoch": 72.94,
+      "eval_accuracy_NAT": 0.9927213194967726,
+      "eval_accuracy_REST": 0.08943089430894309,
+      "eval_accuracy_concrete": 0.9468108526254798,
+      "eval_accuracy_concrete tiles": 0.9154843271270328,
+      "eval_accuracy_dark bricks": 0.856550159830025,
+      "eval_accuracy_grass": 0.8992655380061846,
+      "eval_accuracy_red bricks": 0.8841168717047452,
+      "eval_accuracy_road": 0.9796259193961525,
+      "eval_accuracy_rough concrete": 0.6253403964986086,
+      "eval_accuracy_rough red sidewalk": 0.784311808631438,
+      "eval_accuracy_speedway bricks": 0.9926374674149154,
+      "eval_accuracy_steel": 0.8652661092424795,
+      "eval_accuracy_tiles": 0.2090338998454967,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9821372358073259,
+      "eval_iou_REST": 0.08943089430894309,
+      "eval_iou_concrete": 0.8860208515889054,
+      "eval_iou_concrete tiles": 0.8653152149699265,
+      "eval_iou_dark bricks": 0.839343324455115,
+      "eval_iou_grass": 0.8308539785950206,
+      "eval_iou_red bricks": 0.8594308899685014,
+      "eval_iou_road": 0.9600674794861629,
+      "eval_iou_rough concrete": 0.6149735127279546,
+      "eval_iou_rough red sidewalk": 0.6126972377897618,
+      "eval_iou_speedway bricks": 0.9755335515396683,
+      "eval_iou_steel": 0.8341532160925078,
+      "eval_iou_tiles": 0.2064508407635689,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07735487818717957,
+      "eval_mean_accuracy": 0.7723535049329442,
+      "eval_mean_iou": 0.7351083252379511,
+      "eval_overall_accuracy": 0.9783551490913003,
+      "eval_runtime": 5.5504,
+      "eval_samples_per_second": 21.26,
+      "eval_steps_per_second": 1.441,
+      "step": 1240
+    },
+    {
+      "epoch": 73.0,
+      "learning_rate": 3.342105263157895e-05,
+      "loss": 0.0332,
+      "step": 1241
+    },
+    {
+      "epoch": 73.06,
+      "learning_rate": 3.340557275541796e-05,
+      "loss": 0.031,
+      "step": 1242
+    },
+    {
+      "epoch": 73.12,
+      "learning_rate": 3.339009287925697e-05,
+      "loss": 0.0293,
+      "step": 1243
+    },
+    {
+      "epoch": 73.18,
+      "learning_rate": 3.337461300309598e-05,
+      "loss": 0.0289,
+      "step": 1244
+    },
+    {
+      "epoch": 73.24,
+      "learning_rate": 3.335913312693499e-05,
+      "loss": 0.0255,
+      "step": 1245
+    },
+    {
+      "epoch": 73.29,
+      "learning_rate": 3.334365325077399e-05,
+      "loss": 0.0296,
+      "step": 1246
+    },
+    {
+      "epoch": 73.35,
+      "learning_rate": 3.3328173374613e-05,
+      "loss": 0.0336,
+      "step": 1247
+    },
+    {
+      "epoch": 73.41,
+      "learning_rate": 3.331269349845201e-05,
+      "loss": 0.0289,
+      "step": 1248
+    },
+    {
+      "epoch": 73.47,
+      "learning_rate": 3.329721362229102e-05,
+      "loss": 0.0294,
+      "step": 1249
+    },
+    {
+      "epoch": 73.53,
+      "learning_rate": 3.328173374613003e-05,
+      "loss": 0.034,
+      "step": 1250
+    },
+    {
+      "epoch": 73.59,
+      "learning_rate": 3.326625386996904e-05,
+      "loss": 0.0299,
+      "step": 1251
+    },
+    {
+      "epoch": 73.65,
+      "learning_rate": 3.3250773993808046e-05,
+      "loss": 0.0307,
+      "step": 1252
+    },
+    {
+      "epoch": 73.71,
+      "learning_rate": 3.3235294117647056e-05,
+      "loss": 0.0339,
+      "step": 1253
+    },
+    {
+      "epoch": 73.76,
+      "learning_rate": 3.3219814241486066e-05,
+      "loss": 0.032,
+      "step": 1254
+    },
+    {
+      "epoch": 73.82,
+      "learning_rate": 3.320433436532508e-05,
+      "loss": 0.0294,
+      "step": 1255
+    },
+    {
+      "epoch": 73.88,
+      "learning_rate": 3.318885448916409e-05,
+      "loss": 0.0282,
+      "step": 1256
+    },
+    {
+      "epoch": 73.94,
+      "learning_rate": 3.31733746130031e-05,
+      "loss": 0.0349,
+      "step": 1257
+    },
+    {
+      "epoch": 74.0,
+      "learning_rate": 3.3157894736842106e-05,
+      "loss": 0.0268,
+      "step": 1258
+    },
+    {
+      "epoch": 74.06,
+      "learning_rate": 3.3142414860681116e-05,
+      "loss": 0.0285,
+      "step": 1259
+    },
+    {
+      "epoch": 74.12,
+      "learning_rate": 3.3126934984520126e-05,
+      "loss": 0.0272,
+      "step": 1260
+    },
+    {
+      "epoch": 74.12,
+      "eval_accuracy_NAT": 0.9942002268714645,
+      "eval_accuracy_REST": 0.22276422764227644,
+      "eval_accuracy_concrete": 0.9498260228799787,
+      "eval_accuracy_concrete tiles": 0.9140702333254772,
+      "eval_accuracy_dark bricks": 0.8848888846352622,
+      "eval_accuracy_grass": 0.8712098846737596,
+      "eval_accuracy_red bricks": 0.9234951669595782,
+      "eval_accuracy_road": 0.97808092235949,
+      "eval_accuracy_rough concrete": 0.6260733290032714,
+      "eval_accuracy_rough red sidewalk": 0.772994232171997,
+      "eval_accuracy_speedway bricks": 0.9907025834325538,
+      "eval_accuracy_steel": 0.8942984801611299,
+      "eval_accuracy_tiles": 0.264382441152413,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9816555179486792,
+      "eval_iou_REST": 0.21642969984202212,
+      "eval_iou_concrete": 0.8902363972700686,
+      "eval_iou_concrete tiles": 0.855441351625568,
+      "eval_iou_dark bricks": 0.8587450193652651,
+      "eval_iou_grass": 0.8241404527621606,
+      "eval_iou_red bricks": 0.8853261728005054,
+      "eval_iou_road": 0.9616836897156543,
+      "eval_iou_rough concrete": 0.617017653069614,
+      "eval_iou_rough red sidewalk": 0.6000770586930378,
+      "eval_iou_speedway bricks": 0.977204896135035,
+      "eval_iou_steel": 0.8476513278193066,
+      "eval_iou_tiles": 0.25355918414782963,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07605685293674469,
+      "eval_mean_accuracy": 0.7913066642514347,
+      "eval_mean_iou": 0.7514744939380574,
+      "eval_overall_accuracy": 0.9786853143724344,
+      "eval_runtime": 5.5689,
+      "eval_samples_per_second": 21.189,
+      "eval_steps_per_second": 1.437,
+      "step": 1260
+    },
+    {
+      "epoch": 74.18,
+      "learning_rate": 3.3111455108359136e-05,
+      "loss": 0.0309,
+      "step": 1261
+    },
+    {
+      "epoch": 74.24,
+      "learning_rate": 3.3095975232198146e-05,
+      "loss": 0.0314,
+      "step": 1262
+    },
+    {
+      "epoch": 74.29,
+      "learning_rate": 3.3080495356037156e-05,
+      "loss": 0.0328,
+      "step": 1263
+    },
+    {
+      "epoch": 74.35,
+      "learning_rate": 3.306501547987616e-05,
+      "loss": 0.0331,
+      "step": 1264
+    },
+    {
+      "epoch": 74.41,
+      "learning_rate": 3.304953560371517e-05,
+      "loss": 0.0306,
+      "step": 1265
+    },
+    {
+      "epoch": 74.47,
+      "learning_rate": 3.303405572755418e-05,
+      "loss": 0.0313,
+      "step": 1266
+    },
+    {
+      "epoch": 74.53,
+      "learning_rate": 3.301857585139319e-05,
+      "loss": 0.0261,
+      "step": 1267
+    },
+    {
+      "epoch": 74.59,
+      "learning_rate": 3.30030959752322e-05,
+      "loss": 0.0322,
+      "step": 1268
+    },
+    {
+      "epoch": 74.65,
+      "learning_rate": 3.298761609907121e-05,
+      "loss": 0.0277,
+      "step": 1269
+    },
+    {
+      "epoch": 74.71,
+      "learning_rate": 3.297213622291022e-05,
+      "loss": 0.0288,
+      "step": 1270
+    },
+    {
+      "epoch": 74.76,
+      "learning_rate": 3.295665634674923e-05,
+      "loss": 0.0329,
+      "step": 1271
+    },
+    {
+      "epoch": 74.82,
+      "learning_rate": 3.294117647058824e-05,
+      "loss": 0.0311,
+      "step": 1272
+    },
+    {
+      "epoch": 74.88,
+      "learning_rate": 3.292569659442725e-05,
+      "loss": 0.0312,
+      "step": 1273
+    },
+    {
+      "epoch": 74.94,
+      "learning_rate": 3.291021671826626e-05,
+      "loss": 0.0284,
+      "step": 1274
+    },
+    {
+      "epoch": 75.0,
+      "learning_rate": 3.289473684210527e-05,
+      "loss": 0.0292,
+      "step": 1275
+    },
+    {
+      "epoch": 75.06,
+      "learning_rate": 3.287925696594427e-05,
+      "loss": 0.0302,
+      "step": 1276
+    },
+    {
+      "epoch": 75.12,
+      "learning_rate": 3.286377708978328e-05,
+      "loss": 0.0342,
+      "step": 1277
+    },
+    {
+      "epoch": 75.18,
+      "learning_rate": 3.284829721362229e-05,
+      "loss": 0.0294,
+      "step": 1278
+    },
+    {
+      "epoch": 75.24,
+      "learning_rate": 3.28328173374613e-05,
+      "loss": 0.0272,
+      "step": 1279
+    },
+    {
+      "epoch": 75.29,
+      "learning_rate": 3.281733746130031e-05,
+      "loss": 0.0329,
+      "step": 1280
+    },
+    {
+      "epoch": 75.29,
+      "eval_accuracy_NAT": 0.9928449391603104,
+      "eval_accuracy_REST": 0.1902439024390244,
+      "eval_accuracy_concrete": 0.9563448140906179,
+      "eval_accuracy_concrete tiles": 0.8934716002828188,
+      "eval_accuracy_dark bricks": 0.8582058840418813,
+      "eval_accuracy_grass": 0.8855253645815592,
+      "eval_accuracy_red bricks": 0.8926296133567663,
+      "eval_accuracy_road": 0.9793902418820853,
+      "eval_accuracy_rough concrete": 0.6249173216381936,
+      "eval_accuracy_rough red sidewalk": 0.7686627211815399,
+      "eval_accuracy_speedway bricks": 0.9893386881517445,
+      "eval_accuracy_steel": 0.8970098007366796,
+      "eval_accuracy_tiles": 0.21088188069920324,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9820734284240179,
+      "eval_iou_REST": 0.18962722852512157,
+      "eval_iou_concrete": 0.8856006348028841,
+      "eval_iou_concrete tiles": 0.8492003046458492,
+      "eval_iou_dark bricks": 0.8390065215153868,
+      "eval_iou_grass": 0.8260845002789371,
+      "eval_iou_red bricks": 0.8720356261401437,
+      "eval_iou_road": 0.9609563440959449,
+      "eval_iou_rough concrete": 0.6129052995768755,
+      "eval_iou_rough red sidewalk": 0.5995917313874094,
+      "eval_iou_speedway bricks": 0.9775231301854868,
+      "eval_iou_steel": 0.8492074757511238,
+      "eval_iou_tiles": 0.20707401237505949,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07827547192573547,
+      "eval_mean_accuracy": 0.7799589824801865,
+      "eval_mean_iou": 0.7423758644387879,
+      "eval_overall_accuracy": 0.9784607644808494,
+      "eval_runtime": 6.2956,
+      "eval_samples_per_second": 18.743,
+      "eval_steps_per_second": 1.271,
+      "step": 1280
+    },
+    {
+      "epoch": 75.35,
+      "learning_rate": 3.280185758513932e-05,
+      "loss": 0.0345,
+      "step": 1281
+    },
+    {
+      "epoch": 75.41,
+      "learning_rate": 3.2786377708978326e-05,
+      "loss": 0.0353,
+      "step": 1282
+    },
+    {
+      "epoch": 75.47,
+      "learning_rate": 3.2770897832817336e-05,
+      "loss": 0.0303,
+      "step": 1283
+    },
+    {
+      "epoch": 75.53,
+      "learning_rate": 3.2755417956656346e-05,
+      "loss": 0.0284,
+      "step": 1284
+    },
+    {
+      "epoch": 75.59,
+      "learning_rate": 3.2739938080495356e-05,
+      "loss": 0.0282,
+      "step": 1285
+    },
+    {
+      "epoch": 75.65,
+      "learning_rate": 3.2724458204334366e-05,
+      "loss": 0.0288,
+      "step": 1286
+    },
+    {
+      "epoch": 75.71,
+      "learning_rate": 3.2708978328173376e-05,
+      "loss": 0.0285,
+      "step": 1287
+    },
+    {
+      "epoch": 75.76,
+      "learning_rate": 3.2693498452012386e-05,
+      "loss": 0.0289,
+      "step": 1288
+    },
+    {
+      "epoch": 75.82,
+      "learning_rate": 3.2678018575851396e-05,
+      "loss": 0.0266,
+      "step": 1289
+    },
+    {
+      "epoch": 75.88,
+      "learning_rate": 3.2662538699690406e-05,
+      "loss": 0.0244,
+      "step": 1290
+    },
+    {
+      "epoch": 75.94,
+      "learning_rate": 3.2647058823529416e-05,
+      "loss": 0.033,
+      "step": 1291
+    },
+    {
+      "epoch": 76.0,
+      "learning_rate": 3.2631578947368426e-05,
+      "loss": 0.0308,
+      "step": 1292
+    },
+    {
+      "epoch": 76.06,
+      "learning_rate": 3.2616099071207436e-05,
+      "loss": 0.0316,
+      "step": 1293
+    },
+    {
+      "epoch": 76.12,
+      "learning_rate": 3.260061919504644e-05,
+      "loss": 0.0321,
+      "step": 1294
+    },
+    {
+      "epoch": 76.18,
+      "learning_rate": 3.258513931888545e-05,
+      "loss": 0.0283,
+      "step": 1295
+    },
+    {
+      "epoch": 76.24,
+      "learning_rate": 3.256965944272446e-05,
+      "loss": 0.0281,
+      "step": 1296
+    },
+    {
+      "epoch": 76.29,
+      "learning_rate": 3.255417956656347e-05,
+      "loss": 0.0296,
+      "step": 1297
+    },
+    {
+      "epoch": 76.35,
+      "learning_rate": 3.253869969040248e-05,
+      "loss": 0.0286,
+      "step": 1298
+    },
+    {
+      "epoch": 76.41,
+      "learning_rate": 3.252321981424149e-05,
+      "loss": 0.0279,
+      "step": 1299
+    },
+    {
+      "epoch": 76.47,
+      "learning_rate": 3.250773993808049e-05,
+      "loss": 0.0297,
+      "step": 1300
+    },
+    {
+      "epoch": 76.47,
+      "eval_accuracy_NAT": 0.992488851991681,
+      "eval_accuracy_REST": 0.2601626016260163,
+      "eval_accuracy_concrete": 0.9521742553611096,
+      "eval_accuracy_concrete tiles": 0.9211407023332547,
+      "eval_accuracy_dark bricks": 0.8558323603162146,
+      "eval_accuracy_grass": 0.9004804944607826,
+      "eval_accuracy_red bricks": 0.8834578207381371,
+      "eval_accuracy_road": 0.9812932977269678,
+      "eval_accuracy_rough concrete": 0.6232965278067442,
+      "eval_accuracy_rough red sidewalk": 0.7873047624060941,
+      "eval_accuracy_speedway bricks": 0.9882248146343102,
+      "eval_accuracy_steel": 0.9161164893271427,
+      "eval_accuracy_tiles": 0.2258777909055106,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9823089468549129,
+      "eval_iou_REST": 0.25806451612903225,
+      "eval_iou_concrete": 0.8877026974541958,
+      "eval_iou_concrete tiles": 0.8522459659834278,
+      "eval_iou_dark bricks": 0.8368740231907387,
+      "eval_iou_grass": 0.8291762110430432,
+      "eval_iou_red bricks": 0.8674036128336479,
+      "eval_iou_road": 0.9597156076083275,
+      "eval_iou_rough concrete": 0.614172812250458,
+      "eval_iou_rough red sidewalk": 0.6164766705136963,
+      "eval_iou_speedway bricks": 0.9772265422709538,
+      "eval_iou_steel": 0.8510678800260005,
+      "eval_iou_tiles": 0.22112818079364138,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07827833294868469,
+      "eval_mean_accuracy": 0.7913731361256896,
+      "eval_mean_iou": 0.7502741282270828,
+      "eval_overall_accuracy": 0.9786628464520988,
+      "eval_runtime": 5.5012,
+      "eval_samples_per_second": 21.45,
+      "eval_steps_per_second": 1.454,
+      "step": 1300
+    },
+    {
+      "epoch": 76.53,
+      "learning_rate": 3.24922600619195e-05,
+      "loss": 0.0318,
+      "step": 1301
+    },
+    {
+      "epoch": 76.59,
+      "learning_rate": 3.247678018575851e-05,
+      "loss": 0.0295,
+      "step": 1302
+    },
+    {
+      "epoch": 76.65,
+      "learning_rate": 3.246130030959752e-05,
+      "loss": 0.0265,
+      "step": 1303
+    },
+    {
+      "epoch": 76.71,
+      "learning_rate": 3.244582043343654e-05,
+      "loss": 0.0291,
+      "step": 1304
+    },
+    {
+      "epoch": 76.76,
+      "learning_rate": 3.243034055727555e-05,
+      "loss": 0.0284,
+      "step": 1305
+    },
+    {
+      "epoch": 76.82,
+      "learning_rate": 3.241486068111455e-05,
+      "loss": 0.0298,
+      "step": 1306
+    },
+    {
+      "epoch": 76.88,
+      "learning_rate": 3.239938080495356e-05,
+      "loss": 0.0343,
+      "step": 1307
+    },
+    {
+      "epoch": 76.94,
+      "learning_rate": 3.238390092879257e-05,
+      "loss": 0.0295,
+      "step": 1308
+    },
+    {
+      "epoch": 77.0,
+      "learning_rate": 3.236842105263158e-05,
+      "loss": 0.0333,
+      "step": 1309
+    },
+    {
+      "epoch": 77.06,
+      "learning_rate": 3.235294117647059e-05,
+      "loss": 0.0275,
+      "step": 1310
+    },
+    {
+      "epoch": 77.12,
+      "learning_rate": 3.23374613003096e-05,
+      "loss": 0.0253,
+      "step": 1311
+    },
+    {
+      "epoch": 77.18,
+      "learning_rate": 3.2321981424148605e-05,
+      "loss": 0.0312,
+      "step": 1312
+    },
+    {
+      "epoch": 77.24,
+      "learning_rate": 3.2306501547987615e-05,
+      "loss": 0.0301,
+      "step": 1313
+    },
+    {
+      "epoch": 77.29,
+      "learning_rate": 3.2291021671826625e-05,
+      "loss": 0.0316,
+      "step": 1314
+    },
+    {
+      "epoch": 77.35,
+      "learning_rate": 3.2275541795665635e-05,
+      "loss": 0.0372,
+      "step": 1315
+    },
+    {
+      "epoch": 77.41,
+      "learning_rate": 3.2260061919504645e-05,
+      "loss": 0.037,
+      "step": 1316
+    },
+    {
+      "epoch": 77.47,
+      "learning_rate": 3.2244582043343655e-05,
+      "loss": 0.0305,
+      "step": 1317
+    },
+    {
+      "epoch": 77.53,
+      "learning_rate": 3.2229102167182665e-05,
+      "loss": 0.0297,
+      "step": 1318
+    },
+    {
+      "epoch": 77.59,
+      "learning_rate": 3.2213622291021675e-05,
+      "loss": 0.0282,
+      "step": 1319
+    },
+    {
+      "epoch": 77.65,
+      "learning_rate": 3.2198142414860685e-05,
+      "loss": 0.0316,
+      "step": 1320
+    },
+    {
+      "epoch": 77.65,
+      "eval_accuracy_NAT": 0.9949590887920742,
+      "eval_accuracy_REST": 0.06829268292682927,
+      "eval_accuracy_concrete": 0.9489343519199399,
+      "eval_accuracy_concrete tiles": 0.9051143059156257,
+      "eval_accuracy_dark bricks": 0.8368776678215263,
+      "eval_accuracy_grass": 0.8689285295241989,
+      "eval_accuracy_red bricks": 0.8564367311072056,
+      "eval_accuracy_road": 0.9787991776404565,
+      "eval_accuracy_rough concrete": 0.622158396844219,
+      "eval_accuracy_rough red sidewalk": 0.7372968664225178,
+      "eval_accuracy_speedway bricks": 0.9889505460722078,
+      "eval_accuracy_steel": 0.9112061253354791,
+      "eval_accuracy_tiles": 0.22015207973582962,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.981318691427329,
+      "eval_iou_REST": 0.06807131280388978,
+      "eval_iou_concrete": 0.8872889012142022,
+      "eval_iou_concrete tiles": 0.8525885800550573,
+      "eval_iou_dark bricks": 0.8249517672753346,
+      "eval_iou_grass": 0.8239352772760756,
+      "eval_iou_red bricks": 0.8418268192614986,
+      "eval_iou_road": 0.9598942168303413,
+      "eval_iou_rough concrete": 0.6145488148701861,
+      "eval_iou_rough red sidewalk": 0.6000416164311462,
+      "eval_iou_speedway bricks": 0.9772518101809055,
+      "eval_iou_steel": 0.8517559944406162,
+      "eval_iou_tiles": 0.21822822822822824,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08024384081363678,
+      "eval_mean_accuracy": 0.7644697346198547,
+      "eval_mean_iou": 0.7309001561765238,
+      "eval_overall_accuracy": 0.9783175840216168,
+      "eval_runtime": 7.305,
+      "eval_samples_per_second": 16.153,
+      "eval_steps_per_second": 1.095,
+      "step": 1320
+    },
+    {
+      "epoch": 77.71,
+      "learning_rate": 3.2182662538699695e-05,
+      "loss": 0.0311,
+      "step": 1321
+    },
+    {
+      "epoch": 77.76,
+      "learning_rate": 3.2167182662538705e-05,
+      "loss": 0.0303,
+      "step": 1322
+    },
+    {
+      "epoch": 77.82,
+      "learning_rate": 3.2151702786377715e-05,
+      "loss": 0.0348,
+      "step": 1323
+    },
+    {
+      "epoch": 77.88,
+      "learning_rate": 3.213622291021672e-05,
+      "loss": 0.0324,
+      "step": 1324
+    },
+    {
+      "epoch": 77.94,
+      "learning_rate": 3.212074303405573e-05,
+      "loss": 0.0321,
+      "step": 1325
+    },
+    {
+      "epoch": 78.0,
+      "learning_rate": 3.210526315789474e-05,
+      "loss": 0.0286,
+      "step": 1326
+    },
+    {
+      "epoch": 78.06,
+      "learning_rate": 3.208978328173375e-05,
+      "loss": 0.0312,
+      "step": 1327
+    },
+    {
+      "epoch": 78.12,
+      "learning_rate": 3.207430340557276e-05,
+      "loss": 0.026,
+      "step": 1328
+    },
+    {
+      "epoch": 78.18,
+      "learning_rate": 3.205882352941177e-05,
+      "loss": 0.0293,
+      "step": 1329
+    },
+    {
+      "epoch": 78.24,
+      "learning_rate": 3.204334365325077e-05,
+      "loss": 0.029,
+      "step": 1330
+    },
+    {
+      "epoch": 78.29,
+      "learning_rate": 3.202786377708978e-05,
+      "loss": 0.03,
+      "step": 1331
+    },
+    {
+      "epoch": 78.35,
+      "learning_rate": 3.201238390092879e-05,
+      "loss": 0.0307,
+      "step": 1332
+    },
+    {
+      "epoch": 78.41,
+      "learning_rate": 3.19969040247678e-05,
+      "loss": 0.0253,
+      "step": 1333
+    },
+    {
+      "epoch": 78.47,
+      "learning_rate": 3.198142414860681e-05,
+      "loss": 0.0307,
+      "step": 1334
+    },
+    {
+      "epoch": 78.53,
+      "learning_rate": 3.196594427244582e-05,
+      "loss": 0.0331,
+      "step": 1335
+    },
+    {
+      "epoch": 78.59,
+      "learning_rate": 3.195046439628483e-05,
+      "loss": 0.0307,
+      "step": 1336
+    },
+    {
+      "epoch": 78.65,
+      "learning_rate": 3.193498452012384e-05,
+      "loss": 0.0324,
+      "step": 1337
+    },
+    {
+      "epoch": 78.71,
+      "learning_rate": 3.191950464396285e-05,
+      "loss": 0.0329,
+      "step": 1338
+    },
+    {
+      "epoch": 78.76,
+      "learning_rate": 3.190402476780186e-05,
+      "loss": 0.0285,
+      "step": 1339
+    },
+    {
+      "epoch": 78.82,
+      "learning_rate": 3.188854489164087e-05,
+      "loss": 0.0295,
+      "step": 1340
+    },
+    {
+      "epoch": 78.82,
+      "eval_accuracy_NAT": 0.9931076511622378,
+      "eval_accuracy_REST": 0.23089430894308943,
+      "eval_accuracy_concrete": 0.9574501344577127,
+      "eval_accuracy_concrete tiles": 0.891821824181004,
+      "eval_accuracy_dark bricks": 0.8444624160174569,
+      "eval_accuracy_grass": 0.8917903386359938,
+      "eval_accuracy_red bricks": 0.8780206502636204,
+      "eval_accuracy_road": 0.9784314993418666,
+      "eval_accuracy_rough concrete": 0.6227423593276089,
+      "eval_accuracy_rough red sidewalk": 0.7669857646696094,
+      "eval_accuracy_speedway bricks": 0.9876396026247658,
+      "eval_accuracy_steel": 0.9197898913972422,
+      "eval_accuracy_tiles": 0.21251779817625496,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.982500950046869,
+      "eval_iou_REST": 0.22756410256410256,
+      "eval_iou_concrete": 0.8858469625941018,
+      "eval_iou_concrete tiles": 0.8542532057070615,
+      "eval_iou_dark bricks": 0.8293690518152979,
+      "eval_iou_grass": 0.8328935275195306,
+      "eval_iou_red bricks": 0.8618793465955038,
+      "eval_iou_road": 0.960303673072041,
+      "eval_iou_rough concrete": 0.6135943307049629,
+      "eval_iou_rough red sidewalk": 0.6129829074212083,
+      "eval_iou_speedway bricks": 0.9771381370643655,
+      "eval_iou_steel": 0.8528193994888774,
+      "eval_iou_tiles": 0.2103574427252009,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07932768762111664,
+      "eval_mean_accuracy": 0.7827426337844972,
+      "eval_mean_iou": 0.7462694644091633,
+      "eval_overall_accuracy": 0.9787218449479442,
+      "eval_runtime": 6.1176,
+      "eval_samples_per_second": 19.288,
+      "eval_steps_per_second": 1.308,
+      "step": 1340
+    },
+    {
+      "epoch": 78.88,
+      "learning_rate": 3.187306501547988e-05,
+      "loss": 0.0265,
+      "step": 1341
+    },
+    {
+      "epoch": 78.94,
+      "learning_rate": 3.1857585139318885e-05,
+      "loss": 0.0276,
+      "step": 1342
+    },
+    {
+      "epoch": 79.0,
+      "learning_rate": 3.1842105263157895e-05,
+      "loss": 0.0265,
+      "step": 1343
+    },
+    {
+      "epoch": 79.06,
+      "learning_rate": 3.1826625386996905e-05,
+      "loss": 0.0255,
+      "step": 1344
+    },
+    {
+      "epoch": 79.12,
+      "learning_rate": 3.1811145510835915e-05,
+      "loss": 0.0286,
+      "step": 1345
+    },
+    {
+      "epoch": 79.18,
+      "learning_rate": 3.1795665634674925e-05,
+      "loss": 0.0276,
+      "step": 1346
+    },
+    {
+      "epoch": 79.24,
+      "learning_rate": 3.178018575851393e-05,
+      "loss": 0.0315,
+      "step": 1347
+    },
+    {
+      "epoch": 79.29,
+      "learning_rate": 3.176470588235294e-05,
+      "loss": 0.0312,
+      "step": 1348
+    },
+    {
+      "epoch": 79.35,
+      "learning_rate": 3.174922600619195e-05,
+      "loss": 0.025,
+      "step": 1349
+    },
+    {
+      "epoch": 79.41,
+      "learning_rate": 3.173374613003096e-05,
+      "loss": 0.032,
+      "step": 1350
+    },
+    {
+      "epoch": 79.47,
+      "learning_rate": 3.171826625386997e-05,
+      "loss": 0.0298,
+      "step": 1351
+    },
+    {
+      "epoch": 79.53,
+      "learning_rate": 3.170278637770898e-05,
+      "loss": 0.0309,
+      "step": 1352
+    },
+    {
+      "epoch": 79.59,
+      "learning_rate": 3.168730650154799e-05,
+      "loss": 0.0284,
+      "step": 1353
+    },
+    {
+      "epoch": 79.65,
+      "learning_rate": 3.1671826625387e-05,
+      "loss": 0.0279,
+      "step": 1354
+    },
+    {
+      "epoch": 79.71,
+      "learning_rate": 3.165634674922601e-05,
+      "loss": 0.0271,
+      "step": 1355
+    },
+    {
+      "epoch": 79.76,
+      "learning_rate": 3.164086687306502e-05,
+      "loss": 0.0286,
+      "step": 1356
+    },
+    {
+      "epoch": 79.82,
+      "learning_rate": 3.162538699690403e-05,
+      "loss": 0.0291,
+      "step": 1357
+    },
+    {
+      "epoch": 79.88,
+      "learning_rate": 3.160990712074304e-05,
+      "loss": 0.0302,
+      "step": 1358
+    },
+    {
+      "epoch": 79.94,
+      "learning_rate": 3.159442724458204e-05,
+      "loss": 0.0272,
+      "step": 1359
+    },
+    {
+      "epoch": 80.0,
+      "learning_rate": 3.157894736842105e-05,
+      "loss": 0.0363,
+      "step": 1360
+    },
+    {
+      "epoch": 80.0,
+      "eval_accuracy_NAT": 0.9934704233889913,
+      "eval_accuracy_REST": 0.22926829268292684,
+      "eval_accuracy_concrete": 0.9454731550857731,
+      "eval_accuracy_concrete tiles": 0.9084138581192552,
+      "eval_accuracy_dark bricks": 0.8572009647225466,
+      "eval_accuracy_grass": 0.897609639411522,
+      "eval_accuracy_red bricks": 0.8853251318101933,
+      "eval_accuracy_road": 0.9826731601789439,
+      "eval_accuracy_rough concrete": 0.6364833540898229,
+      "eval_accuracy_rough red sidewalk": 0.7868009234540793,
+      "eval_accuracy_speedway bricks": 0.9907786695620524,
+      "eval_accuracy_steel": 0.8975195789831424,
+      "eval_accuracy_tiles": 0.22757429791874942,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9823985430335701,
+      "eval_iou_REST": 0.2266881028938907,
+      "eval_iou_concrete": 0.8881201981762045,
+      "eval_iou_concrete tiles": 0.8518952370427672,
+      "eval_iou_dark bricks": 0.8386950272259497,
+      "eval_iou_grass": 0.835193096016228,
+      "eval_iou_red bricks": 0.8649460750120728,
+      "eval_iou_road": 0.9614746893655648,
+      "eval_iou_rough concrete": 0.6243365831988964,
+      "eval_iou_rough red sidewalk": 0.6142808496647605,
+      "eval_iou_speedway bricks": 0.9772434400055711,
+      "eval_iou_steel": 0.8497443650292307,
+      "eval_iou_tiles": 0.2253284540164377,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07582816481590271,
+      "eval_mean_accuracy": 0.7875839576467689,
+      "eval_mean_iou": 0.7492572815908572,
+      "eval_overall_accuracy": 0.9789746817895921,
+      "eval_runtime": 5.7191,
+      "eval_samples_per_second": 20.633,
+      "eval_steps_per_second": 1.399,
+      "step": 1360
+    },
+    {
+      "epoch": 80.06,
+      "learning_rate": 3.156346749226006e-05,
+      "loss": 0.028,
+      "step": 1361
+    },
+    {
+      "epoch": 80.12,
+      "learning_rate": 3.154798761609907e-05,
+      "loss": 0.0313,
+      "step": 1362
+    },
+    {
+      "epoch": 80.18,
+      "learning_rate": 3.153250773993808e-05,
+      "loss": 0.0325,
+      "step": 1363
+    },
+    {
+      "epoch": 80.24,
+      "learning_rate": 3.151702786377709e-05,
+      "loss": 0.0319,
+      "step": 1364
+    },
+    {
+      "epoch": 80.29,
+      "learning_rate": 3.1501547987616095e-05,
+      "loss": 0.0263,
+      "step": 1365
+    },
+    {
+      "epoch": 80.35,
+      "learning_rate": 3.1486068111455105e-05,
+      "loss": 0.0297,
+      "step": 1366
+    },
+    {
+      "epoch": 80.41,
+      "learning_rate": 3.147058823529412e-05,
+      "loss": 0.0326,
+      "step": 1367
+    },
+    {
+      "epoch": 80.47,
+      "learning_rate": 3.145510835913313e-05,
+      "loss": 0.0299,
+      "step": 1368
+    },
+    {
+      "epoch": 80.53,
+      "learning_rate": 3.143962848297214e-05,
+      "loss": 0.0261,
+      "step": 1369
+    },
+    {
+      "epoch": 80.59,
+      "learning_rate": 3.142414860681115e-05,
+      "loss": 0.0305,
+      "step": 1370
+    },
+    {
+      "epoch": 80.65,
+      "learning_rate": 3.1408668730650155e-05,
+      "loss": 0.0294,
+      "step": 1371
+    },
+    {
+      "epoch": 80.71,
+      "learning_rate": 3.1393188854489165e-05,
+      "loss": 0.0308,
+      "step": 1372
+    },
+    {
+      "epoch": 80.76,
+      "learning_rate": 3.1377708978328175e-05,
+      "loss": 0.0306,
+      "step": 1373
+    },
+    {
+      "epoch": 80.82,
+      "learning_rate": 3.1362229102167185e-05,
+      "loss": 0.0268,
+      "step": 1374
+    },
+    {
+      "epoch": 80.88,
+      "learning_rate": 3.1346749226006195e-05,
+      "loss": 0.028,
+      "step": 1375
+    },
+    {
+      "epoch": 80.94,
+      "learning_rate": 3.1331269349845205e-05,
+      "loss": 0.0323,
+      "step": 1376
+    },
+    {
+      "epoch": 81.0,
+      "learning_rate": 3.131578947368421e-05,
+      "loss": 0.0297,
+      "step": 1377
+    },
+    {
+      "epoch": 81.06,
+      "learning_rate": 3.130030959752322e-05,
+      "loss": 0.0286,
+      "step": 1378
+    },
+    {
+      "epoch": 81.12,
+      "learning_rate": 3.128482972136223e-05,
+      "loss": 0.0266,
+      "step": 1379
+    },
+    {
+      "epoch": 81.18,
+      "learning_rate": 3.126934984520124e-05,
+      "loss": 0.0278,
+      "step": 1380
+    },
+    {
+      "epoch": 81.18,
+      "eval_accuracy_NAT": 0.99331930872833,
+      "eval_accuracy_REST": 0.16585365853658537,
+      "eval_accuracy_concrete": 0.95582081885435,
+      "eval_accuracy_concrete tiles": 0.9068347867075183,
+      "eval_accuracy_dark bricks": 0.84659667323852,
+      "eval_accuracy_grass": 0.8910891838596252,
+      "eval_accuracy_red bricks": 0.882579086115993,
+      "eval_accuracy_road": 0.9816791461740347,
+      "eval_accuracy_rough concrete": 0.6236957674637555,
+      "eval_accuracy_rough red sidewalk": 0.7761300656494634,
+      "eval_accuracy_speedway bricks": 0.9878529522446436,
+      "eval_accuracy_steel": 0.8980443507074423,
+      "eval_accuracy_tiles": 0.21597140173891968,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9824832905037847,
+      "eval_iou_REST": 0.16558441558441558,
+      "eval_iou_concrete": 0.8856266332083604,
+      "eval_iou_concrete tiles": 0.8531485587583149,
+      "eval_iou_dark bricks": 0.8324314913799593,
+      "eval_iou_grass": 0.8328090662681235,
+      "eval_iou_red bricks": 0.8660738345459444,
+      "eval_iou_road": 0.9614814690744717,
+      "eval_iou_rough concrete": 0.6168588923791394,
+      "eval_iou_rough red sidewalk": 0.6026767726903784,
+      "eval_iou_speedway bricks": 0.9772715097421862,
+      "eval_iou_steel": 0.8500501457063921,
+      "eval_iou_tiles": 0.21380799568125244,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07792044430971146,
+      "eval_mean_accuracy": 0.7788820923091677,
+      "eval_mean_iou": 0.7415618519632864,
+      "eval_overall_accuracy": 0.9787269527629271,
+      "eval_runtime": 5.3595,
+      "eval_samples_per_second": 22.017,
+      "eval_steps_per_second": 1.493,
+      "step": 1380
+    },
+    {
+      "epoch": 81.24,
+      "learning_rate": 3.125386996904025e-05,
+      "loss": 0.0334,
+      "step": 1381
+    },
+    {
+      "epoch": 81.29,
+      "learning_rate": 3.123839009287926e-05,
+      "loss": 0.026,
+      "step": 1382
+    },
+    {
+      "epoch": 81.35,
+      "learning_rate": 3.122291021671827e-05,
+      "loss": 0.0309,
+      "step": 1383
+    },
+    {
+      "epoch": 81.41,
+      "learning_rate": 3.120743034055728e-05,
+      "loss": 0.0328,
+      "step": 1384
+    },
+    {
+      "epoch": 81.47,
+      "learning_rate": 3.119195046439629e-05,
+      "loss": 0.0247,
+      "step": 1385
+    },
+    {
+      "epoch": 81.53,
+      "learning_rate": 3.11764705882353e-05,
+      "loss": 0.0284,
+      "step": 1386
+    },
+    {
+      "epoch": 81.59,
+      "learning_rate": 3.116099071207431e-05,
+      "loss": 0.0274,
+      "step": 1387
+    },
+    {
+      "epoch": 81.65,
+      "learning_rate": 3.114551083591332e-05,
+      "loss": 0.0289,
+      "step": 1388
+    },
+    {
+      "epoch": 81.71,
+      "learning_rate": 3.113003095975232e-05,
+      "loss": 0.0286,
+      "step": 1389
+    },
+    {
+      "epoch": 81.76,
+      "learning_rate": 3.111455108359133e-05,
+      "loss": 0.0284,
+      "step": 1390
+    },
+    {
+      "epoch": 81.82,
+      "learning_rate": 3.109907120743034e-05,
+      "loss": 0.0323,
+      "step": 1391
+    },
+    {
+      "epoch": 81.88,
+      "learning_rate": 3.108359133126935e-05,
+      "loss": 0.0309,
+      "step": 1392
+    },
+    {
+      "epoch": 81.94,
+      "learning_rate": 3.106811145510836e-05,
+      "loss": 0.0295,
+      "step": 1393
+    },
+    {
+      "epoch": 82.0,
+      "learning_rate": 3.105263157894737e-05,
+      "loss": 0.0246,
+      "step": 1394
+    },
+    {
+      "epoch": 82.06,
+      "learning_rate": 3.1037151702786375e-05,
+      "loss": 0.0331,
+      "step": 1395
+    },
+    {
+      "epoch": 82.12,
+      "learning_rate": 3.1021671826625385e-05,
+      "loss": 0.0268,
+      "step": 1396
+    },
+    {
+      "epoch": 82.18,
+      "learning_rate": 3.1006191950464395e-05,
+      "loss": 0.0324,
+      "step": 1397
+    },
+    {
+      "epoch": 82.24,
+      "learning_rate": 3.0990712074303405e-05,
+      "loss": 0.0231,
+      "step": 1398
+    },
+    {
+      "epoch": 82.29,
+      "learning_rate": 3.0975232198142415e-05,
+      "loss": 0.0297,
+      "step": 1399
+    },
+    {
+      "epoch": 82.35,
+      "learning_rate": 3.0959752321981425e-05,
+      "loss": 0.0349,
+      "step": 1400
+    },
+    {
+      "epoch": 82.35,
+      "eval_accuracy_NAT": 0.9918829377315409,
+      "eval_accuracy_REST": 0.22439024390243903,
+      "eval_accuracy_concrete": 0.9583846117390221,
+      "eval_accuracy_concrete tiles": 0.8933066226726373,
+      "eval_accuracy_dark bricks": 0.8434383553777539,
+      "eval_accuracy_grass": 0.9010898651436184,
+      "eval_accuracy_red bricks": 0.8806568541300527,
+      "eval_accuracy_road": 0.9808887447152925,
+      "eval_accuracy_rough concrete": 0.6225755129037832,
+      "eval_accuracy_rough red sidewalk": 0.773167191812241,
+      "eval_accuracy_speedway bricks": 0.9892602029100546,
+      "eval_accuracy_steel": 0.8945408747194971,
+      "eval_accuracy_tiles": 0.20600442303614167,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.982569707737552,
+      "eval_iou_REST": 0.22402597402597402,
+      "eval_iou_concrete": 0.8838863544876526,
+      "eval_iou_concrete tiles": 0.8455772448410486,
+      "eval_iou_dark bricks": 0.8265375507045886,
+      "eval_iou_grass": 0.8310334325550061,
+      "eval_iou_red bricks": 0.8618650900295619,
+      "eval_iou_road": 0.96037455570433,
+      "eval_iou_rough concrete": 0.6151744603681155,
+      "eval_iou_rough red sidewalk": 0.6070222463631211,
+      "eval_iou_speedway bricks": 0.9771983882918063,
+      "eval_iou_steel": 0.8500273074822501,
+      "eval_iou_tiles": 0.20384304085853894,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07893545180559158,
+      "eval_mean_accuracy": 0.7815066492918519,
+      "eval_mean_iou": 0.743779642573042,
+      "eval_overall_accuracy": 0.9785271984035686,
+      "eval_runtime": 5.5112,
+      "eval_samples_per_second": 21.411,
+      "eval_steps_per_second": 1.452,
+      "step": 1400
+    },
+    {
+      "epoch": 82.41,
+      "learning_rate": 3.0944272445820435e-05,
+      "loss": 0.0302,
+      "step": 1401
+    },
+    {
+      "epoch": 82.47,
+      "learning_rate": 3.0928792569659445e-05,
+      "loss": 0.027,
+      "step": 1402
+    },
+    {
+      "epoch": 82.53,
+      "learning_rate": 3.0913312693498455e-05,
+      "loss": 0.0222,
+      "step": 1403
+    },
+    {
+      "epoch": 82.59,
+      "learning_rate": 3.0897832817337465e-05,
+      "loss": 0.03,
+      "step": 1404
+    },
+    {
+      "epoch": 82.65,
+      "learning_rate": 3.0882352941176475e-05,
+      "loss": 0.0258,
+      "step": 1405
+    },
+    {
+      "epoch": 82.71,
+      "learning_rate": 3.0866873065015485e-05,
+      "loss": 0.0286,
+      "step": 1406
+    },
+    {
+      "epoch": 82.76,
+      "learning_rate": 3.085139318885449e-05,
+      "loss": 0.0272,
+      "step": 1407
+    },
+    {
+      "epoch": 82.82,
+      "learning_rate": 3.08359133126935e-05,
+      "loss": 0.0316,
+      "step": 1408
+    },
+    {
+      "epoch": 82.88,
+      "learning_rate": 3.082043343653251e-05,
+      "loss": 0.0311,
+      "step": 1409
+    },
+    {
+      "epoch": 82.94,
+      "learning_rate": 3.080495356037152e-05,
+      "loss": 0.028,
+      "step": 1410
+    },
+    {
+      "epoch": 83.0,
+      "learning_rate": 3.078947368421053e-05,
+      "loss": 0.0273,
+      "step": 1411
+    },
+    {
+      "epoch": 83.06,
+      "learning_rate": 3.077399380804954e-05,
+      "loss": 0.0294,
+      "step": 1412
+    },
+    {
+      "epoch": 83.12,
+      "learning_rate": 3.075851393188854e-05,
+      "loss": 0.0249,
+      "step": 1413
+    },
+    {
+      "epoch": 83.18,
+      "learning_rate": 3.074303405572755e-05,
+      "loss": 0.0322,
+      "step": 1414
+    },
+    {
+      "epoch": 83.24,
+      "learning_rate": 3.072755417956656e-05,
+      "loss": 0.0319,
+      "step": 1415
+    },
+    {
+      "epoch": 83.29,
+      "learning_rate": 3.071207430340558e-05,
+      "loss": 0.027,
+      "step": 1416
+    },
+    {
+      "epoch": 83.35,
+      "learning_rate": 3.069659442724459e-05,
+      "loss": 0.0309,
+      "step": 1417
+    },
+    {
+      "epoch": 83.41,
+      "learning_rate": 3.06811145510836e-05,
+      "loss": 0.0279,
+      "step": 1418
+    },
+    {
+      "epoch": 83.47,
+      "learning_rate": 3.06656346749226e-05,
+      "loss": 0.0286,
+      "step": 1419
+    },
+    {
+      "epoch": 83.53,
+      "learning_rate": 3.065015479876161e-05,
+      "loss": 0.0334,
+      "step": 1420
+    },
+    {
+      "epoch": 83.53,
+      "eval_accuracy_NAT": 0.9937829498562956,
+      "eval_accuracy_REST": 0.208130081300813,
+      "eval_accuracy_concrete": 0.9552586036883761,
+      "eval_accuracy_concrete tiles": 0.9087438133396182,
+      "eval_accuracy_dark bricks": 0.8441705108818406,
+      "eval_accuracy_grass": 0.8773414406128528,
+      "eval_accuracy_red bricks": 0.8930689806678384,
+      "eval_accuracy_road": 0.9788510160052287,
+      "eval_accuracy_rough concrete": 0.6274736472032368,
+      "eval_accuracy_rough red sidewalk": 0.7602854586062461,
+      "eval_accuracy_speedway bricks": 0.9896318939345645,
+      "eval_accuracy_steel": 0.8963600833637367,
+      "eval_accuracy_tiles": 0.22372686237086856,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9822022920545374,
+      "eval_iou_REST": 0.20545746388443017,
+      "eval_iou_concrete": 0.8864829224104481,
+      "eval_iou_concrete tiles": 0.8504753292013146,
+      "eval_iou_dark bricks": 0.8298936335367202,
+      "eval_iou_grass": 0.8271775213017176,
+      "eval_iou_red bricks": 0.8747646457582441,
+      "eval_iou_road": 0.9598746249071112,
+      "eval_iou_rough concrete": 0.6197698713987229,
+      "eval_iou_rough red sidewalk": 0.603590426325813,
+      "eval_iou_speedway bricks": 0.9772318919008559,
+      "eval_iou_steel": 0.8501853484645941,
+      "eval_iou_tiles": 0.2210680716039035,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07888671010732651,
+      "eval_mean_accuracy": 0.7812942570639627,
+      "eval_mean_iou": 0.7452441571344932,
+      "eval_overall_accuracy": 0.9785843542066671,
+      "eval_runtime": 8.3441,
+      "eval_samples_per_second": 14.142,
+      "eval_steps_per_second": 0.959,
+      "step": 1420
+    },
+    {
+      "epoch": 83.59,
+      "learning_rate": 3.063467492260062e-05,
+      "loss": 0.0314,
+      "step": 1421
+    },
+    {
+      "epoch": 83.65,
+      "learning_rate": 3.061919504643963e-05,
+      "loss": 0.0301,
+      "step": 1422
+    },
+    {
+      "epoch": 83.71,
+      "learning_rate": 3.060371517027864e-05,
+      "loss": 0.0298,
+      "step": 1423
+    },
+    {
+      "epoch": 83.76,
+      "learning_rate": 3.058823529411765e-05,
+      "loss": 0.0313,
+      "step": 1424
+    },
+    {
+      "epoch": 83.82,
+      "learning_rate": 3.0572755417956654e-05,
+      "loss": 0.0246,
+      "step": 1425
+    },
+    {
+      "epoch": 83.88,
+      "learning_rate": 3.0557275541795664e-05,
+      "loss": 0.0295,
+      "step": 1426
+    },
+    {
+      "epoch": 83.94,
+      "learning_rate": 3.0541795665634674e-05,
+      "loss": 0.0261,
+      "step": 1427
+    },
+    {
+      "epoch": 84.0,
+      "learning_rate": 3.0526315789473684e-05,
+      "loss": 0.028,
+      "step": 1428
+    },
+    {
+      "epoch": 84.06,
+      "learning_rate": 3.0510835913312698e-05,
+      "loss": 0.0261,
+      "step": 1429
+    },
+    {
+      "epoch": 84.12,
+      "learning_rate": 3.0495356037151708e-05,
+      "loss": 0.0279,
+      "step": 1430
+    },
+    {
+      "epoch": 84.18,
+      "learning_rate": 3.047987616099071e-05,
+      "loss": 0.0315,
+      "step": 1431
+    },
+    {
+      "epoch": 84.24,
+      "learning_rate": 3.046439628482972e-05,
+      "loss": 0.0256,
+      "step": 1432
+    },
+    {
+      "epoch": 84.29,
+      "learning_rate": 3.044891640866873e-05,
+      "loss": 0.0307,
+      "step": 1433
+    },
+    {
+      "epoch": 84.35,
+      "learning_rate": 3.043343653250774e-05,
+      "loss": 0.0248,
+      "step": 1434
+    },
+    {
+      "epoch": 84.41,
+      "learning_rate": 3.041795665634675e-05,
+      "loss": 0.0294,
+      "step": 1435
+    },
+    {
+      "epoch": 84.47,
+      "learning_rate": 3.040247678018576e-05,
+      "loss": 0.0319,
+      "step": 1436
+    },
+    {
+      "epoch": 84.53,
+      "learning_rate": 3.0386996904024768e-05,
+      "loss": 0.0277,
+      "step": 1437
+    },
+    {
+      "epoch": 84.59,
+      "learning_rate": 3.0371517027863778e-05,
+      "loss": 0.0266,
+      "step": 1438
+    },
+    {
+      "epoch": 84.65,
+      "learning_rate": 3.0356037151702788e-05,
+      "loss": 0.0256,
+      "step": 1439
+    },
+    {
+      "epoch": 84.71,
+      "learning_rate": 3.0340557275541798e-05,
+      "loss": 0.0271,
+      "step": 1440
+    },
+    {
+      "epoch": 84.71,
+      "eval_accuracy_NAT": 0.9929208469170749,
+      "eval_accuracy_REST": 0.34959349593495936,
+      "eval_accuracy_concrete": 0.9541162099894666,
+      "eval_accuracy_concrete tiles": 0.9276219655903841,
+      "eval_accuracy_dark bricks": 0.8699682254081886,
+      "eval_accuracy_grass": 0.8941048117574477,
+      "eval_accuracy_red bricks": 0.8968585237258347,
+      "eval_accuracy_road": 0.9809603564769365,
+      "eval_accuracy_rough concrete": 0.6276047408219569,
+      "eval_accuracy_rough red sidewalk": 0.7771678234909272,
+      "eval_accuracy_speedway bricks": 0.9882453784530936,
+      "eval_accuracy_steel": 0.9109712275160305,
+      "eval_accuracy_tiles": 0.2436002302402375,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9825177897285533,
+      "eval_iou_REST": 0.34181240063593005,
+      "eval_iou_concrete": 0.8881662579334739,
+      "eval_iou_concrete tiles": 0.8425705906278766,
+      "eval_iou_dark bricks": 0.8466478521664618,
+      "eval_iou_grass": 0.8327436038216235,
+      "eval_iou_red bricks": 0.8716771645137183,
+      "eval_iou_road": 0.9593965061708843,
+      "eval_iou_rough concrete": 0.6171858519091484,
+      "eval_iou_rough red sidewalk": 0.6108448048608935,
+      "eval_iou_speedway bricks": 0.977506610617669,
+      "eval_iou_steel": 0.8513912438285012,
+      "eval_iou_tiles": 0.24069805729338162,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07791364938020706,
+      "eval_mean_accuracy": 0.8010564489478876,
+      "eval_mean_iou": 0.7587045180083168,
+      "eval_overall_accuracy": 0.9788982585325079,
+      "eval_runtime": 5.4694,
+      "eval_samples_per_second": 21.575,
+      "eval_steps_per_second": 1.463,
+      "step": 1440
+    },
+    {
+      "epoch": 84.76,
+      "learning_rate": 3.0325077399380808e-05,
+      "loss": 0.0263,
+      "step": 1441
+    },
+    {
+      "epoch": 84.82,
+      "learning_rate": 3.0309597523219818e-05,
+      "loss": 0.03,
+      "step": 1442
+    },
+    {
+      "epoch": 84.88,
+      "learning_rate": 3.0294117647058824e-05,
+      "loss": 0.031,
+      "step": 1443
+    },
+    {
+      "epoch": 84.94,
+      "learning_rate": 3.0278637770897834e-05,
+      "loss": 0.031,
+      "step": 1444
+    },
+    {
+      "epoch": 85.0,
+      "learning_rate": 3.0263157894736844e-05,
+      "loss": 0.027,
+      "step": 1445
+    },
+    {
+      "epoch": 85.06,
+      "learning_rate": 3.0247678018575854e-05,
+      "loss": 0.0321,
+      "step": 1446
+    },
+    {
+      "epoch": 85.12,
+      "learning_rate": 3.0232198142414864e-05,
+      "loss": 0.0265,
+      "step": 1447
+    },
+    {
+      "epoch": 85.18,
+      "learning_rate": 3.0216718266253874e-05,
+      "loss": 0.0279,
+      "step": 1448
+    },
+    {
+      "epoch": 85.24,
+      "learning_rate": 3.0201238390092878e-05,
+      "loss": 0.0278,
+      "step": 1449
+    },
+    {
+      "epoch": 85.29,
+      "learning_rate": 3.0185758513931888e-05,
+      "loss": 0.0305,
+      "step": 1450
+    },
+    {
+      "epoch": 85.35,
+      "learning_rate": 3.0170278637770898e-05,
+      "loss": 0.0281,
+      "step": 1451
+    },
+    {
+      "epoch": 85.41,
+      "learning_rate": 3.015479876160991e-05,
+      "loss": 0.0265,
+      "step": 1452
+    },
+    {
+      "epoch": 85.47,
+      "learning_rate": 3.013931888544892e-05,
+      "loss": 0.0268,
+      "step": 1453
+    },
+    {
+      "epoch": 85.53,
+      "learning_rate": 3.012383900928793e-05,
+      "loss": 0.0294,
+      "step": 1454
+    },
+    {
+      "epoch": 85.59,
+      "learning_rate": 3.0108359133126934e-05,
+      "loss": 0.0278,
+      "step": 1455
+    },
+    {
+      "epoch": 85.65,
+      "learning_rate": 3.0092879256965944e-05,
+      "loss": 0.0235,
+      "step": 1456
+    },
+    {
+      "epoch": 85.71,
+      "learning_rate": 3.0077399380804954e-05,
+      "loss": 0.0327,
+      "step": 1457
+    },
+    {
+      "epoch": 85.76,
+      "learning_rate": 3.0061919504643964e-05,
+      "loss": 0.0301,
+      "step": 1458
+    },
+    {
+      "epoch": 85.82,
+      "learning_rate": 3.0046439628482974e-05,
+      "loss": 0.0274,
+      "step": 1459
+    },
+    {
+      "epoch": 85.88,
+      "learning_rate": 3.0030959752321984e-05,
+      "loss": 0.0269,
+      "step": 1460
+    },
+    {
+      "epoch": 85.88,
+      "eval_accuracy_NAT": 0.9929293649750073,
+      "eval_accuracy_REST": 0.2959349593495935,
+      "eval_accuracy_concrete": 0.950695526280788,
+      "eval_accuracy_concrete tiles": 0.9228611831251473,
+      "eval_accuracy_dark bricks": 0.8474436766648163,
+      "eval_accuracy_grass": 0.8994708694319227,
+      "eval_accuracy_red bricks": 0.9106436731107206,
+      "eval_accuracy_road": 0.9824465677391151,
+      "eval_accuracy_rough concrete": 0.6298988791495599,
+      "eval_accuracy_rough red sidewalk": 0.7745132690124005,
+      "eval_accuracy_speedway bricks": 0.9894750948163411,
+      "eval_accuracy_steel": 0.9105938916571291,
+      "eval_accuracy_tiles": 0.23639007543397256,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9826370358008298,
+      "eval_iou_REST": 0.2940226171243942,
+      "eval_iou_concrete": 0.8883696029891439,
+      "eval_iou_concrete tiles": 0.8451394284727618,
+      "eval_iou_dark bricks": 0.8318161748818683,
+      "eval_iou_grass": 0.8361210308734277,
+      "eval_iou_red bricks": 0.8757724607827603,
+      "eval_iou_road": 0.9575628429019907,
+      "eval_iou_rough concrete": 0.6199323234633497,
+      "eval_iou_rough red sidewalk": 0.621539092862143,
+      "eval_iou_speedway bricks": 0.9773567257242782,
+      "eval_iou_steel": 0.8515895031794738,
+      "eval_iou_tiles": 0.23397301349325336,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07780010253190994,
+      "eval_mean_accuracy": 0.7956382331343475,
+      "eval_mean_iou": 0.7550639886576673,
+      "eval_overall_accuracy": 0.9789560608944651,
+      "eval_runtime": 5.6492,
+      "eval_samples_per_second": 20.888,
+      "eval_steps_per_second": 1.416,
+      "step": 1460
+    },
+    {
+      "epoch": 85.94,
+      "learning_rate": 3.001547987616099e-05,
+      "loss": 0.0317,
+      "step": 1461
+    },
+    {
+      "epoch": 86.0,
+      "learning_rate": 3e-05,
+      "loss": 0.0312,
+      "step": 1462
+    },
+    {
+      "epoch": 86.06,
+      "learning_rate": 2.998452012383901e-05,
+      "loss": 0.0308,
+      "step": 1463
+    },
+    {
+      "epoch": 86.12,
+      "learning_rate": 2.996904024767802e-05,
+      "loss": 0.0282,
+      "step": 1464
+    },
+    {
+      "epoch": 86.18,
+      "learning_rate": 2.995356037151703e-05,
+      "loss": 0.0286,
+      "step": 1465
+    },
+    {
+      "epoch": 86.24,
+      "learning_rate": 2.993808049535604e-05,
+      "loss": 0.0257,
+      "step": 1466
+    },
+    {
+      "epoch": 86.29,
+      "learning_rate": 2.9922600619195047e-05,
+      "loss": 0.0242,
+      "step": 1467
+    },
+    {
+      "epoch": 86.35,
+      "learning_rate": 2.9907120743034057e-05,
+      "loss": 0.0295,
+      "step": 1468
+    },
+    {
+      "epoch": 86.41,
+      "learning_rate": 2.9891640866873067e-05,
+      "loss": 0.033,
+      "step": 1469
+    },
+    {
+      "epoch": 86.47,
+      "learning_rate": 2.9876160990712077e-05,
+      "loss": 0.0277,
+      "step": 1470
+    },
+    {
+      "epoch": 86.53,
+      "learning_rate": 2.9860681114551087e-05,
+      "loss": 0.0325,
+      "step": 1471
+    },
+    {
+      "epoch": 86.59,
+      "learning_rate": 2.9845201238390097e-05,
+      "loss": 0.0278,
+      "step": 1472
+    },
+    {
+      "epoch": 86.65,
+      "learning_rate": 2.98297213622291e-05,
+      "loss": 0.0296,
+      "step": 1473
+    },
+    {
+      "epoch": 86.71,
+      "learning_rate": 2.981424148606811e-05,
+      "loss": 0.0286,
+      "step": 1474
+    },
+    {
+      "epoch": 86.76,
+      "learning_rate": 2.979876160990712e-05,
+      "loss": 0.0282,
+      "step": 1475
+    },
+    {
+      "epoch": 86.82,
+      "learning_rate": 2.978328173374613e-05,
+      "loss": 0.0324,
+      "step": 1476
+    },
+    {
+      "epoch": 86.88,
+      "learning_rate": 2.9767801857585144e-05,
+      "loss": 0.0262,
+      "step": 1477
+    },
+    {
+      "epoch": 86.94,
+      "learning_rate": 2.9752321981424154e-05,
+      "loss": 0.0274,
+      "step": 1478
+    },
+    {
+      "epoch": 87.0,
+      "learning_rate": 2.9736842105263157e-05,
+      "loss": 0.0282,
+      "step": 1479
+    },
+    {
+      "epoch": 87.06,
+      "learning_rate": 2.9721362229102167e-05,
+      "loss": 0.0273,
+      "step": 1480
+    },
+    {
+      "epoch": 87.06,
+      "eval_accuracy_NAT": 0.993669411691703,
+      "eval_accuracy_REST": 0.08943089430894309,
+      "eval_accuracy_concrete": 0.9506099136382469,
+      "eval_accuracy_concrete tiles": 0.9190195616309215,
+      "eval_accuracy_dark bricks": 0.8679966694102559,
+      "eval_accuracy_grass": 0.8859767152270758,
+      "eval_accuracy_red bricks": 0.9251427943760984,
+      "eval_accuracy_road": 0.9807241445467376,
+      "eval_accuracy_rough concrete": 0.6401063050071804,
+      "eval_accuracy_rough red sidewalk": 0.7764985448830266,
+      "eval_accuracy_speedway bricks": 0.9898335907237984,
+      "eval_accuracy_steel": 0.9119682937922003,
+      "eval_accuracy_tiles": 0.2602623526916901,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9826311973955166,
+      "eval_iou_REST": 0.08943089430894309,
+      "eval_iou_concrete": 0.8905050531089321,
+      "eval_iou_concrete tiles": 0.8519368158877892,
+      "eval_iou_dark bricks": 0.8465821579590961,
+      "eval_iou_grass": 0.8314788437220273,
+      "eval_iou_red bricks": 0.887419660731219,
+      "eval_iou_road": 0.9591810280844729,
+      "eval_iou_rough concrete": 0.6285406683166869,
+      "eval_iou_rough red sidewalk": 0.6164325498928416,
+      "eval_iou_speedway bricks": 0.9773870673206759,
+      "eval_iou_steel": 0.8540444402840995,
+      "eval_iou_tiles": 0.2561648328711572,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07576427608728409,
+      "eval_mean_accuracy": 0.7839414763021446,
+      "eval_mean_iou": 0.7439796315294966,
+      "eval_overall_accuracy": 0.9791401685294459,
+      "eval_runtime": 5.6589,
+      "eval_samples_per_second": 20.852,
+      "eval_steps_per_second": 1.414,
+      "step": 1480
+    },
+    {
+      "epoch": 87.12,
+      "learning_rate": 2.9705882352941177e-05,
+      "loss": 0.0321,
+      "step": 1481
+    },
+    {
+      "epoch": 87.18,
+      "learning_rate": 2.9690402476780187e-05,
+      "loss": 0.0298,
+      "step": 1482
+    },
+    {
+      "epoch": 87.24,
+      "learning_rate": 2.9674922600619197e-05,
+      "loss": 0.0304,
+      "step": 1483
+    },
+    {
+      "epoch": 87.29,
+      "learning_rate": 2.9659442724458207e-05,
+      "loss": 0.0304,
+      "step": 1484
+    },
+    {
+      "epoch": 87.35,
+      "learning_rate": 2.9643962848297214e-05,
+      "loss": 0.0297,
+      "step": 1485
+    },
+    {
+      "epoch": 87.41,
+      "learning_rate": 2.9628482972136224e-05,
+      "loss": 0.0284,
+      "step": 1486
+    },
+    {
+      "epoch": 87.47,
+      "learning_rate": 2.9613003095975234e-05,
+      "loss": 0.0257,
+      "step": 1487
+    },
+    {
+      "epoch": 87.53,
+      "learning_rate": 2.9597523219814244e-05,
+      "loss": 0.0292,
+      "step": 1488
+    },
+    {
+      "epoch": 87.59,
+      "learning_rate": 2.9582043343653254e-05,
+      "loss": 0.0267,
+      "step": 1489
+    },
+    {
+      "epoch": 87.65,
+      "learning_rate": 2.9566563467492264e-05,
+      "loss": 0.0273,
+      "step": 1490
+    },
+    {
+      "epoch": 87.71,
+      "learning_rate": 2.955108359133127e-05,
+      "loss": 0.0294,
+      "step": 1491
+    },
+    {
+      "epoch": 87.76,
+      "learning_rate": 2.953560371517028e-05,
+      "loss": 0.0278,
+      "step": 1492
+    },
+    {
+      "epoch": 87.82,
+      "learning_rate": 2.952012383900929e-05,
+      "loss": 0.0262,
+      "step": 1493
+    },
+    {
+      "epoch": 87.88,
+      "learning_rate": 2.95046439628483e-05,
+      "loss": 0.0288,
+      "step": 1494
+    },
+    {
+      "epoch": 87.94,
+      "learning_rate": 2.948916408668731e-05,
+      "loss": 0.0244,
+      "step": 1495
+    },
+    {
+      "epoch": 88.0,
+      "learning_rate": 2.9473684210526314e-05,
+      "loss": 0.0244,
+      "step": 1496
+    },
+    {
+      "epoch": 88.06,
+      "learning_rate": 2.9458204334365324e-05,
+      "loss": 0.0265,
+      "step": 1497
+    },
+    {
+      "epoch": 88.12,
+      "learning_rate": 2.9442724458204334e-05,
+      "loss": 0.0243,
+      "step": 1498
+    },
+    {
+      "epoch": 88.18,
+      "learning_rate": 2.9427244582043344e-05,
+      "loss": 0.0324,
+      "step": 1499
+    },
+    {
+      "epoch": 88.24,
+      "learning_rate": 2.9411764705882354e-05,
+      "loss": 0.0288,
+      "step": 1500
+    },
+    {
+      "epoch": 88.24,
+      "eval_accuracy_NAT": 0.9931507266577313,
+      "eval_accuracy_REST": 0.18536585365853658,
+      "eval_accuracy_concrete": 0.9524498210542891,
+      "eval_accuracy_concrete tiles": 0.9074946971482442,
+      "eval_accuracy_dark bricks": 0.8556026644717952,
+      "eval_accuracy_grass": 0.890016161570284,
+      "eval_accuracy_red bricks": 0.890981985940246,
+      "eval_accuracy_road": 0.9808288901085452,
+      "eval_accuracy_rough concrete": 0.6310548865146378,
+      "eval_accuracy_rough red sidewalk": 0.7877785214206754,
+      "eval_accuracy_speedway bricks": 0.9902592617726149,
+      "eval_accuracy_steel": 0.9092619710425964,
+      "eval_accuracy_tiles": 0.23396649398648853,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9827546453868776,
+      "eval_iou_REST": 0.1847649918962723,
+      "eval_iou_concrete": 0.8889152057255291,
+      "eval_iou_concrete tiles": 0.8508264097578222,
+      "eval_iou_dark bricks": 0.837033257181379,
+      "eval_iou_grass": 0.832136639017071,
+      "eval_iou_red bricks": 0.875074167970225,
+      "eval_iou_road": 0.9590789470933866,
+      "eval_iou_rough concrete": 0.6224169546512448,
+      "eval_iou_rough red sidewalk": 0.6146399277156502,
+      "eval_iou_speedway bricks": 0.9774471573154316,
+      "eval_iou_steel": 0.8554739205567377,
+      "eval_iou_tiles": 0.23169926797071883,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07804440706968307,
+      "eval_mean_accuracy": 0.7852470719497451,
+      "eval_mean_iou": 0.7470970378644883,
+      "eval_overall_accuracy": 0.9790279905674821,
+      "eval_runtime": 5.5948,
+      "eval_samples_per_second": 21.091,
+      "eval_steps_per_second": 1.43,
+      "step": 1500
+    },
+    {
+      "epoch": 88.29,
+      "learning_rate": 2.9396284829721367e-05,
+      "loss": 0.0287,
+      "step": 1501
+    },
+    {
+      "epoch": 88.35,
+      "learning_rate": 2.938080495356037e-05,
+      "loss": 0.0298,
+      "step": 1502
+    },
+    {
+      "epoch": 88.41,
+      "learning_rate": 2.936532507739938e-05,
+      "loss": 0.0319,
+      "step": 1503
+    },
+    {
+      "epoch": 88.47,
+      "learning_rate": 2.934984520123839e-05,
+      "loss": 0.03,
+      "step": 1504
+    },
+    {
+      "epoch": 88.53,
+      "learning_rate": 2.93343653250774e-05,
+      "loss": 0.0234,
+      "step": 1505
+    },
+    {
+      "epoch": 88.59,
+      "learning_rate": 2.931888544891641e-05,
+      "loss": 0.0278,
+      "step": 1506
+    },
+    {
+      "epoch": 88.65,
+      "learning_rate": 2.930340557275542e-05,
+      "loss": 0.0297,
+      "step": 1507
+    },
+    {
+      "epoch": 88.71,
+      "learning_rate": 2.9287925696594427e-05,
+      "loss": 0.0258,
+      "step": 1508
+    },
+    {
+      "epoch": 88.76,
+      "learning_rate": 2.9272445820433437e-05,
+      "loss": 0.0271,
+      "step": 1509
+    },
+    {
+      "epoch": 88.82,
+      "learning_rate": 2.9256965944272447e-05,
+      "loss": 0.031,
+      "step": 1510
+    },
+    {
+      "epoch": 88.88,
+      "learning_rate": 2.9241486068111457e-05,
+      "loss": 0.0251,
+      "step": 1511
+    },
+    {
+      "epoch": 88.94,
+      "learning_rate": 2.9226006191950467e-05,
+      "loss": 0.028,
+      "step": 1512
+    },
+    {
+      "epoch": 89.0,
+      "learning_rate": 2.9210526315789477e-05,
+      "loss": 0.0265,
+      "step": 1513
+    },
+    {
+      "epoch": 89.06,
+      "learning_rate": 2.919504643962848e-05,
+      "loss": 0.0253,
+      "step": 1514
+    },
+    {
+      "epoch": 89.12,
+      "learning_rate": 2.9179566563467494e-05,
+      "loss": 0.0319,
+      "step": 1515
+    },
+    {
+      "epoch": 89.18,
+      "learning_rate": 2.9164086687306504e-05,
+      "loss": 0.0288,
+      "step": 1516
+    },
+    {
+      "epoch": 89.24,
+      "learning_rate": 2.9148606811145514e-05,
+      "loss": 0.0293,
+      "step": 1517
+    },
+    {
+      "epoch": 89.29,
+      "learning_rate": 2.9133126934984524e-05,
+      "loss": 0.0258,
+      "step": 1518
+    },
+    {
+      "epoch": 89.35,
+      "learning_rate": 2.9117647058823534e-05,
+      "loss": 0.0268,
+      "step": 1519
+    },
+    {
+      "epoch": 89.41,
+      "learning_rate": 2.9102167182662537e-05,
+      "loss": 0.0258,
+      "step": 1520
+    },
+    {
+      "epoch": 89.41,
+      "eval_accuracy_NAT": 0.9924023236183804,
+      "eval_accuracy_REST": 0.3056910569105691,
+      "eval_accuracy_concrete": 0.957218903882992,
+      "eval_accuracy_concrete tiles": 0.9353523450388875,
+      "eval_accuracy_dark bricks": 0.8572775300040196,
+      "eval_accuracy_grass": 0.893308087976473,
+      "eval_accuracy_red bricks": 0.8876318101933216,
+      "eval_accuracy_road": 0.9823006721351688,
+      "eval_accuracy_rough concrete": 0.6271876247623929,
+      "eval_accuracy_rough red sidewalk": 0.7935463494235931,
+      "eval_accuracy_speedway bricks": 0.9876289779850611,
+      "eval_accuracy_steel": 0.9146271371953201,
+      "eval_accuracy_tiles": 0.230634069496198,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9826423473833403,
+      "eval_iou_REST": 0.3003194888178914,
+      "eval_iou_concrete": 0.8865741732246671,
+      "eval_iou_concrete tiles": 0.8465295849153193,
+      "eval_iou_dark bricks": 0.8399851833791273,
+      "eval_iou_grass": 0.8322465304438511,
+      "eval_iou_red bricks": 0.8710320668283481,
+      "eval_iou_road": 0.9575308683136019,
+      "eval_iou_rough concrete": 0.6191994540662651,
+      "eval_iou_rough red sidewalk": 0.6180667002471681,
+      "eval_iou_speedway bricks": 0.9777841003269798,
+      "eval_iou_steel": 0.854014223848279,
+      "eval_iou_tiles": 0.22587823403750296,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07936197519302368,
+      "eval_mean_accuracy": 0.7972928375863366,
+      "eval_mean_iou": 0.7547540735255647,
+      "eval_overall_accuracy": 0.9788075139967062,
+      "eval_runtime": 5.3616,
+      "eval_samples_per_second": 22.008,
+      "eval_steps_per_second": 1.492,
+      "step": 1520
+    },
+    {
+      "epoch": 89.47,
+      "learning_rate": 2.9086687306501547e-05,
+      "loss": 0.027,
+      "step": 1521
+    },
+    {
+      "epoch": 89.53,
+      "learning_rate": 2.9071207430340557e-05,
+      "loss": 0.0253,
+      "step": 1522
+    },
+    {
+      "epoch": 89.59,
+      "learning_rate": 2.9055727554179567e-05,
+      "loss": 0.0269,
+      "step": 1523
+    },
+    {
+      "epoch": 89.65,
+      "learning_rate": 2.9040247678018577e-05,
+      "loss": 0.0247,
+      "step": 1524
+    },
+    {
+      "epoch": 89.71,
+      "learning_rate": 2.902476780185759e-05,
+      "loss": 0.0286,
+      "step": 1525
+    },
+    {
+      "epoch": 89.76,
+      "learning_rate": 2.9009287925696593e-05,
+      "loss": 0.0251,
+      "step": 1526
+    },
+    {
+      "epoch": 89.82,
+      "learning_rate": 2.8993808049535603e-05,
+      "loss": 0.0299,
+      "step": 1527
+    },
+    {
+      "epoch": 89.88,
+      "learning_rate": 2.8978328173374613e-05,
+      "loss": 0.0331,
+      "step": 1528
+    },
+    {
+      "epoch": 89.94,
+      "learning_rate": 2.8962848297213623e-05,
+      "loss": 0.0286,
+      "step": 1529
+    },
+    {
+      "epoch": 90.0,
+      "learning_rate": 2.8947368421052634e-05,
+      "loss": 0.025,
+      "step": 1530
+    },
+    {
+      "epoch": 90.06,
+      "learning_rate": 2.8931888544891644e-05,
+      "loss": 0.0284,
+      "step": 1531
+    },
+    {
+      "epoch": 90.12,
+      "learning_rate": 2.891640866873065e-05,
+      "loss": 0.0272,
+      "step": 1532
+    },
+    {
+      "epoch": 90.18,
+      "learning_rate": 2.890092879256966e-05,
+      "loss": 0.0279,
+      "step": 1533
+    },
+    {
+      "epoch": 90.24,
+      "learning_rate": 2.888544891640867e-05,
+      "loss": 0.0291,
+      "step": 1534
+    },
+    {
+      "epoch": 90.29,
+      "learning_rate": 2.886996904024768e-05,
+      "loss": 0.0289,
+      "step": 1535
+    },
+    {
+      "epoch": 90.35,
+      "learning_rate": 2.885448916408669e-05,
+      "loss": 0.0268,
+      "step": 1536
+    },
+    {
+      "epoch": 90.41,
+      "learning_rate": 2.88390092879257e-05,
+      "loss": 0.0277,
+      "step": 1537
+    },
+    {
+      "epoch": 90.47,
+      "learning_rate": 2.8823529411764703e-05,
+      "loss": 0.0264,
+      "step": 1538
+    },
+    {
+      "epoch": 90.53,
+      "learning_rate": 2.8808049535603713e-05,
+      "loss": 0.0308,
+      "step": 1539
+    },
+    {
+      "epoch": 90.59,
+      "learning_rate": 2.8792569659442727e-05,
+      "loss": 0.023,
+      "step": 1540
+    },
+    {
+      "epoch": 90.59,
+      "eval_accuracy_NAT": 0.9927307540546091,
+      "eval_accuracy_REST": 0.3252032520325203,
+      "eval_accuracy_concrete": 0.9555395201717145,
+      "eval_accuracy_concrete tiles": 0.9297195380626915,
+      "eval_accuracy_dark bricks": 0.8654700151216431,
+      "eval_accuracy_grass": 0.8867119342031059,
+      "eval_accuracy_red bricks": 0.9231107205623902,
+      "eval_accuracy_road": 0.9818116813746893,
+      "eval_accuracy_rough concrete": 0.6278490516568446,
+      "eval_accuracy_rough red sidewalk": 0.7999458561126193,
+      "eval_accuracy_speedway bricks": 0.9899388089299068,
+      "eval_accuracy_steel": 0.9057410026638413,
+      "eval_accuracy_tiles": 0.23390590445030143,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828049629374469,
+      "eval_iou_REST": 0.31347962382445144,
+      "eval_iou_concrete": 0.8881160639401734,
+      "eval_iou_concrete tiles": 0.8506490705999051,
+      "eval_iou_dark bricks": 0.8438303371887408,
+      "eval_iou_grass": 0.8323462325554956,
+      "eval_iou_red bricks": 0.8904428904428905,
+      "eval_iou_road": 0.9597776360271012,
+      "eval_iou_rough concrete": 0.619196774876002,
+      "eval_iou_rough red sidewalk": 0.6106509147479061,
+      "eval_iou_speedway bricks": 0.9778079746941716,
+      "eval_iou_steel": 0.8547838596325732,
+      "eval_iou_tiles": 0.22806084773297888,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07828875631093979,
+      "eval_mean_accuracy": 0.8013598491843751,
+      "eval_mean_iou": 0.7578420914769105,
+      "eval_overall_accuracy": 0.9790317406088619,
+      "eval_runtime": 5.5415,
+      "eval_samples_per_second": 21.294,
+      "eval_steps_per_second": 1.444,
+      "step": 1540
+    },
+    {
+      "epoch": 90.65,
+      "learning_rate": 2.8777089783281737e-05,
+      "loss": 0.0256,
+      "step": 1541
+    },
+    {
+      "epoch": 90.71,
+      "learning_rate": 2.8761609907120747e-05,
+      "loss": 0.0245,
+      "step": 1542
+    },
+    {
+      "epoch": 90.76,
+      "learning_rate": 2.8746130030959757e-05,
+      "loss": 0.0305,
+      "step": 1543
+    },
+    {
+      "epoch": 90.82,
+      "learning_rate": 2.873065015479876e-05,
+      "loss": 0.0251,
+      "step": 1544
+    },
+    {
+      "epoch": 90.88,
+      "learning_rate": 2.871517027863777e-05,
+      "loss": 0.0298,
+      "step": 1545
+    },
+    {
+      "epoch": 90.94,
+      "learning_rate": 2.869969040247678e-05,
+      "loss": 0.0291,
+      "step": 1546
+    },
+    {
+      "epoch": 91.0,
+      "learning_rate": 2.868421052631579e-05,
+      "loss": 0.0276,
+      "step": 1547
+    },
+    {
+      "epoch": 91.06,
+      "learning_rate": 2.86687306501548e-05,
+      "loss": 0.028,
+      "step": 1548
+    },
+    {
+      "epoch": 91.12,
+      "learning_rate": 2.865325077399381e-05,
+      "loss": 0.0287,
+      "step": 1549
+    },
+    {
+      "epoch": 91.18,
+      "learning_rate": 2.8637770897832817e-05,
+      "loss": 0.0289,
+      "step": 1550
+    },
+    {
+      "epoch": 91.24,
+      "learning_rate": 2.8622291021671827e-05,
+      "loss": 0.0276,
+      "step": 1551
+    },
+    {
+      "epoch": 91.29,
+      "learning_rate": 2.8606811145510837e-05,
+      "loss": 0.0299,
+      "step": 1552
+    },
+    {
+      "epoch": 91.35,
+      "learning_rate": 2.8591331269349847e-05,
+      "loss": 0.0263,
+      "step": 1553
+    },
+    {
+      "epoch": 91.41,
+      "learning_rate": 2.8575851393188857e-05,
+      "loss": 0.0246,
+      "step": 1554
+    },
+    {
+      "epoch": 91.47,
+      "learning_rate": 2.8560371517027867e-05,
+      "loss": 0.0292,
+      "step": 1555
+    },
+    {
+      "epoch": 91.53,
+      "learning_rate": 2.8544891640866873e-05,
+      "loss": 0.031,
+      "step": 1556
+    },
+    {
+      "epoch": 91.59,
+      "learning_rate": 2.8529411764705883e-05,
+      "loss": 0.0278,
+      "step": 1557
+    },
+    {
+      "epoch": 91.65,
+      "learning_rate": 2.8513931888544893e-05,
+      "loss": 0.0312,
+      "step": 1558
+    },
+    {
+      "epoch": 91.71,
+      "learning_rate": 2.8498452012383903e-05,
+      "loss": 0.0292,
+      "step": 1559
+    },
+    {
+      "epoch": 91.76,
+      "learning_rate": 2.8482972136222913e-05,
+      "loss": 0.0263,
+      "step": 1560
+    },
+    {
+      "epoch": 91.76,
+      "eval_accuracy_NAT": 0.9935412634404037,
+      "eval_accuracy_REST": 0.2634146341463415,
+      "eval_accuracy_concrete": 0.95487602219202,
+      "eval_accuracy_concrete tiles": 0.9192081074711289,
+      "eval_accuracy_dark bricks": 0.8575598644794518,
+      "eval_accuracy_grass": 0.8835326089013539,
+      "eval_accuracy_red bricks": 0.9121814586994728,
+      "eval_accuracy_road": 0.9808502667538122,
+      "eval_accuracy_rough concrete": 0.629255328657661,
+      "eval_accuracy_rough red sidewalk": 0.7827551718692425,
+      "eval_accuracy_speedway bricks": 0.9881835156315868,
+      "eval_accuracy_steel": 0.9137275285250915,
+      "eval_accuracy_tiles": 0.24396376745736012,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.982647858053155,
+      "eval_iou_REST": 0.2575516693163752,
+      "eval_iou_concrete": 0.8877119238770678,
+      "eval_iou_concrete tiles": 0.8496057160284058,
+      "eval_iou_dark bricks": 0.8400766919337525,
+      "eval_iou_grass": 0.8322863000267403,
+      "eval_iou_red bricks": 0.8876596654374432,
+      "eval_iou_road": 0.9591895715451144,
+      "eval_iou_rough concrete": 0.6229412458706938,
+      "eval_iou_rough red sidewalk": 0.6038648744292899,
+      "eval_iou_speedway bricks": 0.9776860465352853,
+      "eval_iou_steel": 0.854292489749191,
+      "eval_iou_tiles": 0.23872765541161475,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07876893877983093,
+      "eval_mean_accuracy": 0.7940807337096097,
+      "eval_mean_iou": 0.7534032083241637,
+      "eval_overall_accuracy": 0.9789359852419062,
+      "eval_runtime": 5.3864,
+      "eval_samples_per_second": 21.907,
+      "eval_steps_per_second": 1.485,
+      "step": 1560
+    },
+    {
+      "epoch": 91.82,
+      "learning_rate": 2.8467492260061923e-05,
+      "loss": 0.0251,
+      "step": 1561
+    },
+    {
+      "epoch": 91.88,
+      "learning_rate": 2.8452012383900926e-05,
+      "loss": 0.0263,
+      "step": 1562
+    },
+    {
+      "epoch": 91.94,
+      "learning_rate": 2.8436532507739936e-05,
+      "loss": 0.0294,
+      "step": 1563
+    },
+    {
+      "epoch": 92.0,
+      "learning_rate": 2.842105263157895e-05,
+      "loss": 0.0234,
+      "step": 1564
+    },
+    {
+      "epoch": 92.06,
+      "learning_rate": 2.840557275541796e-05,
+      "loss": 0.0289,
+      "step": 1565
+    },
+    {
+      "epoch": 92.12,
+      "learning_rate": 2.839009287925697e-05,
+      "loss": 0.0283,
+      "step": 1566
+    },
+    {
+      "epoch": 92.18,
+      "learning_rate": 2.837461300309598e-05,
+      "loss": 0.0305,
+      "step": 1567
+    },
+    {
+      "epoch": 92.24,
+      "learning_rate": 2.8359133126934983e-05,
+      "loss": 0.0295,
+      "step": 1568
+    },
+    {
+      "epoch": 92.29,
+      "learning_rate": 2.8343653250773993e-05,
+      "loss": 0.0227,
+      "step": 1569
+    },
+    {
+      "epoch": 92.35,
+      "learning_rate": 2.8328173374613003e-05,
+      "loss": 0.0248,
+      "step": 1570
+    },
+    {
+      "epoch": 92.41,
+      "learning_rate": 2.8312693498452013e-05,
+      "loss": 0.0301,
+      "step": 1571
+    },
+    {
+      "epoch": 92.47,
+      "learning_rate": 2.8297213622291023e-05,
+      "loss": 0.0258,
+      "step": 1572
+    },
+    {
+      "epoch": 92.53,
+      "learning_rate": 2.8281733746130033e-05,
+      "loss": 0.0258,
+      "step": 1573
+    },
+    {
+      "epoch": 92.59,
+      "learning_rate": 2.826625386996904e-05,
+      "loss": 0.0267,
+      "step": 1574
+    },
+    {
+      "epoch": 92.65,
+      "learning_rate": 2.825077399380805e-05,
+      "loss": 0.0259,
+      "step": 1575
+    },
+    {
+      "epoch": 92.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.0252,
+      "step": 1576
+    },
+    {
+      "epoch": 92.76,
+      "learning_rate": 2.821981424148607e-05,
+      "loss": 0.0272,
+      "step": 1577
+    },
+    {
+      "epoch": 92.82,
+      "learning_rate": 2.820433436532508e-05,
+      "loss": 0.029,
+      "step": 1578
+    },
+    {
+      "epoch": 92.88,
+      "learning_rate": 2.818885448916409e-05,
+      "loss": 0.0253,
+      "step": 1579
+    },
+    {
+      "epoch": 92.94,
+      "learning_rate": 2.8173374613003096e-05,
+      "loss": 0.0308,
+      "step": 1580
+    },
+    {
+      "epoch": 92.94,
+      "eval_accuracy_NAT": 0.9932622161754793,
+      "eval_accuracy_REST": 0.21951219512195122,
+      "eval_accuracy_concrete": 0.9518092550324181,
+      "eval_accuracy_concrete tiles": 0.9266556681593212,
+      "eval_accuracy_dark bricks": 0.8570143368489559,
+      "eval_accuracy_grass": 0.8981272260008251,
+      "eval_accuracy_red bricks": 0.898341388400703,
+      "eval_accuracy_road": 0.981349945836925,
+      "eval_accuracy_rough concrete": 0.6284687669453399,
+      "eval_accuracy_rough red sidewalk": 0.7816722941216282,
+      "eval_accuracy_speedway bricks": 0.9883673904445418,
+      "eval_accuracy_steel": 0.9183855023064966,
+      "eval_accuracy_tiles": 0.23402708352267562,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9827872154053462,
+      "eval_iou_REST": 0.21809369951534732,
+      "eval_iou_concrete": 0.8883347976324371,
+      "eval_iou_concrete tiles": 0.8554271914365903,
+      "eval_iou_dark bricks": 0.8390978007252827,
+      "eval_iou_grass": 0.8334842240863791,
+      "eval_iou_red bricks": 0.8765339478055838,
+      "eval_iou_road": 0.9602794391345524,
+      "eval_iou_rough concrete": 0.6198735203883723,
+      "eval_iou_rough red sidewalk": 0.6176489812410498,
+      "eval_iou_speedway bricks": 0.9778192275180463,
+      "eval_iou_steel": 0.8562887645214052,
+      "eval_iou_tiles": 0.231107521091366,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07858654856681824,
+      "eval_mean_accuracy": 0.7905379437628662,
+      "eval_mean_iou": 0.7505212561924429,
+      "eval_overall_accuracy": 0.9791110410528668,
+      "eval_runtime": 5.5585,
+      "eval_samples_per_second": 21.229,
+      "eval_steps_per_second": 1.439,
+      "step": 1580
+    },
+    {
+      "epoch": 93.0,
+      "learning_rate": 2.8157894736842106e-05,
+      "loss": 0.0307,
+      "step": 1581
+    },
+    {
+      "epoch": 93.06,
+      "learning_rate": 2.8142414860681116e-05,
+      "loss": 0.0263,
+      "step": 1582
+    },
+    {
+      "epoch": 93.12,
+      "learning_rate": 2.8126934984520126e-05,
+      "loss": 0.0261,
+      "step": 1583
+    },
+    {
+      "epoch": 93.18,
+      "learning_rate": 2.8111455108359136e-05,
+      "loss": 0.0276,
+      "step": 1584
+    },
+    {
+      "epoch": 93.24,
+      "learning_rate": 2.8095975232198146e-05,
+      "loss": 0.0236,
+      "step": 1585
+    },
+    {
+      "epoch": 93.29,
+      "learning_rate": 2.808049535603715e-05,
+      "loss": 0.0264,
+      "step": 1586
+    },
+    {
+      "epoch": 93.35,
+      "learning_rate": 2.806501547987616e-05,
+      "loss": 0.0301,
+      "step": 1587
+    },
+    {
+      "epoch": 93.41,
+      "learning_rate": 2.8049535603715173e-05,
+      "loss": 0.0239,
+      "step": 1588
+    },
+    {
+      "epoch": 93.47,
+      "learning_rate": 2.8034055727554183e-05,
+      "loss": 0.0244,
+      "step": 1589
+    },
+    {
+      "epoch": 93.53,
+      "learning_rate": 2.8018575851393193e-05,
+      "loss": 0.0312,
+      "step": 1590
+    },
+    {
+      "epoch": 93.59,
+      "learning_rate": 2.8003095975232203e-05,
+      "loss": 0.027,
+      "step": 1591
+    },
+    {
+      "epoch": 93.65,
+      "learning_rate": 2.7987616099071206e-05,
+      "loss": 0.0318,
+      "step": 1592
+    },
+    {
+      "epoch": 93.71,
+      "learning_rate": 2.7972136222910216e-05,
+      "loss": 0.0263,
+      "step": 1593
+    },
+    {
+      "epoch": 93.76,
+      "learning_rate": 2.7956656346749226e-05,
+      "loss": 0.0299,
+      "step": 1594
+    },
+    {
+      "epoch": 93.82,
+      "learning_rate": 2.7941176470588236e-05,
+      "loss": 0.0261,
+      "step": 1595
+    },
+    {
+      "epoch": 93.88,
+      "learning_rate": 2.7925696594427246e-05,
+      "loss": 0.0233,
+      "step": 1596
+    },
+    {
+      "epoch": 93.94,
+      "learning_rate": 2.7910216718266256e-05,
+      "loss": 0.0268,
+      "step": 1597
+    },
+    {
+      "epoch": 94.0,
+      "learning_rate": 2.7894736842105263e-05,
+      "loss": 0.0261,
+      "step": 1598
+    },
+    {
+      "epoch": 94.06,
+      "learning_rate": 2.7879256965944273e-05,
+      "loss": 0.0281,
+      "step": 1599
+    },
+    {
+      "epoch": 94.12,
+      "learning_rate": 2.7863777089783283e-05,
+      "loss": 0.0255,
+      "step": 1600
+    },
+    {
+      "epoch": 94.12,
+      "eval_accuracy_NAT": 0.9936142060504197,
+      "eval_accuracy_REST": 0.20650406504065041,
+      "eval_accuracy_concrete": 0.9600510006741996,
+      "eval_accuracy_concrete tiles": 0.9187131746405844,
+      "eval_accuracy_dark bricks": 0.8598233256130008,
+      "eval_accuracy_grass": 0.8876089581275288,
+      "eval_accuracy_red bricks": 0.8846111599297012,
+      "eval_accuracy_road": 0.9774545866531709,
+      "eval_accuracy_rough concrete": 0.627461729601535,
+      "eval_accuracy_rough red sidewalk": 0.7879815609983531,
+      "eval_accuracy_speedway bricks": 0.9859895275325542,
+      "eval_accuracy_steel": 0.9239056010635374,
+      "eval_accuracy_tiles": 0.22287860886424915,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828774859674867,
+      "eval_iou_REST": 0.20583468395461912,
+      "eval_iou_concrete": 0.887396026714665,
+      "eval_iou_concrete tiles": 0.8558975935359213,
+      "eval_iou_dark bricks": 0.8418717319189609,
+      "eval_iou_grass": 0.833258569092374,
+      "eval_iou_red bricks": 0.8669932177844762,
+      "eval_iou_road": 0.9607099429303793,
+      "eval_iou_rough concrete": 0.6196632750940088,
+      "eval_iou_rough red sidewalk": 0.6095092399239168,
+      "eval_iou_speedway bricks": 0.9777154183027815,
+      "eval_iou_steel": 0.858082813649561,
+      "eval_iou_tiles": 0.22033542976939202,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07987956702709198,
+      "eval_mean_accuracy": 0.7874305772914989,
+      "eval_mean_iou": 0.7477034945106571,
+      "eval_overall_accuracy": 0.9790565684690313,
+      "eval_runtime": 5.496,
+      "eval_samples_per_second": 21.47,
+      "eval_steps_per_second": 1.456,
+      "step": 1600
+    },
+    {
+      "epoch": 94.18,
+      "learning_rate": 2.7848297213622293e-05,
+      "loss": 0.0342,
+      "step": 1601
+    },
+    {
+      "epoch": 94.24,
+      "learning_rate": 2.7832817337461303e-05,
+      "loss": 0.0277,
+      "step": 1602
+    },
+    {
+      "epoch": 94.29,
+      "learning_rate": 2.7817337461300313e-05,
+      "loss": 0.0268,
+      "step": 1603
+    },
+    {
+      "epoch": 94.35,
+      "learning_rate": 2.780185758513932e-05,
+      "loss": 0.0288,
+      "step": 1604
+    },
+    {
+      "epoch": 94.41,
+      "learning_rate": 2.778637770897833e-05,
+      "loss": 0.0274,
+      "step": 1605
+    },
+    {
+      "epoch": 94.47,
+      "learning_rate": 2.777089783281734e-05,
+      "loss": 0.0237,
+      "step": 1606
+    },
+    {
+      "epoch": 94.53,
+      "learning_rate": 2.775541795665635e-05,
+      "loss": 0.0304,
+      "step": 1607
+    },
+    {
+      "epoch": 94.59,
+      "learning_rate": 2.773993808049536e-05,
+      "loss": 0.0312,
+      "step": 1608
+    },
+    {
+      "epoch": 94.65,
+      "learning_rate": 2.772445820433437e-05,
+      "loss": 0.0255,
+      "step": 1609
+    },
+    {
+      "epoch": 94.71,
+      "learning_rate": 2.7708978328173373e-05,
+      "loss": 0.0226,
+      "step": 1610
+    },
+    {
+      "epoch": 94.76,
+      "learning_rate": 2.7693498452012383e-05,
+      "loss": 0.026,
+      "step": 1611
+    },
+    {
+      "epoch": 94.82,
+      "learning_rate": 2.7678018575851393e-05,
+      "loss": 0.0269,
+      "step": 1612
+    },
+    {
+      "epoch": 94.88,
+      "learning_rate": 2.7662538699690406e-05,
+      "loss": 0.0273,
+      "step": 1613
+    },
+    {
+      "epoch": 94.94,
+      "learning_rate": 2.7647058823529416e-05,
+      "loss": 0.027,
+      "step": 1614
+    },
+    {
+      "epoch": 95.0,
+      "learning_rate": 2.7631578947368426e-05,
+      "loss": 0.0291,
+      "step": 1615
+    },
+    {
+      "epoch": 95.06,
+      "learning_rate": 2.761609907120743e-05,
+      "loss": 0.0235,
+      "step": 1616
+    },
+    {
+      "epoch": 95.12,
+      "learning_rate": 2.760061919504644e-05,
+      "loss": 0.0233,
+      "step": 1617
+    },
+    {
+      "epoch": 95.18,
+      "learning_rate": 2.758513931888545e-05,
+      "loss": 0.0275,
+      "step": 1618
+    },
+    {
+      "epoch": 95.24,
+      "learning_rate": 2.756965944272446e-05,
+      "loss": 0.03,
+      "step": 1619
+    },
+    {
+      "epoch": 95.29,
+      "learning_rate": 2.755417956656347e-05,
+      "loss": 0.0288,
+      "step": 1620
+    },
+    {
+      "epoch": 95.29,
+      "eval_accuracy_NAT": 0.9931991394173724,
+      "eval_accuracy_REST": 0.28780487804878047,
+      "eval_accuracy_concrete": 0.9564862278305297,
+      "eval_accuracy_concrete tiles": 0.9127739806740514,
+      "eval_accuracy_dark bricks": 0.8604406331948777,
+      "eval_accuracy_grass": 0.8750127740748731,
+      "eval_accuracy_red bricks": 0.9047122144112478,
+      "eval_accuracy_road": 0.9825390217298944,
+      "eval_accuracy_rough concrete": 0.6288441713989477,
+      "eval_accuracy_rough red sidewalk": 0.7858383654561999,
+      "eval_accuracy_speedway bricks": 0.9909735117450251,
+      "eval_accuracy_steel": 0.8877988075187293,
+      "eval_accuracy_tiles": 0.22493865309461056,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9827858264924875,
+      "eval_iou_REST": 0.2845659163987138,
+      "eval_iou_concrete": 0.8869904452842066,
+      "eval_iou_concrete tiles": 0.8520294797052029,
+      "eval_iou_dark bricks": 0.8408293818916417,
+      "eval_iou_grass": 0.8290379986283378,
+      "eval_iou_red bricks": 0.8824191129205057,
+      "eval_iou_road": 0.9593850010514721,
+      "eval_iou_rough concrete": 0.6187926940103785,
+      "eval_iou_rough red sidewalk": 0.609535586379067,
+      "eval_iou_speedway bricks": 0.9775109620579256,
+      "eval_iou_steel": 0.8493363551169507,
+      "eval_iou_tiles": 0.22240527182866557,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08075219392776489,
+      "eval_mean_accuracy": 0.7916432598919337,
+      "eval_mean_iou": 0.7535095409050427,
+      "eval_overall_accuracy": 0.9788664478366658,
+      "eval_runtime": 5.43,
+      "eval_samples_per_second": 21.731,
+      "eval_steps_per_second": 1.473,
+      "step": 1620
+    },
+    {
+      "epoch": 95.35,
+      "learning_rate": 2.753869969040248e-05,
+      "loss": 0.0272,
+      "step": 1621
+    },
+    {
+      "epoch": 95.41,
+      "learning_rate": 2.7523219814241486e-05,
+      "loss": 0.0315,
+      "step": 1622
+    },
+    {
+      "epoch": 95.47,
+      "learning_rate": 2.7507739938080496e-05,
+      "loss": 0.0278,
+      "step": 1623
+    },
+    {
+      "epoch": 95.53,
+      "learning_rate": 2.7492260061919506e-05,
+      "loss": 0.026,
+      "step": 1624
+    },
+    {
+      "epoch": 95.59,
+      "learning_rate": 2.7476780185758516e-05,
+      "loss": 0.0281,
+      "step": 1625
+    },
+    {
+      "epoch": 95.65,
+      "learning_rate": 2.7461300309597526e-05,
+      "loss": 0.0269,
+      "step": 1626
+    },
+    {
+      "epoch": 95.71,
+      "learning_rate": 2.7445820433436536e-05,
+      "loss": 0.028,
+      "step": 1627
+    },
+    {
+      "epoch": 95.76,
+      "learning_rate": 2.7430340557275543e-05,
+      "loss": 0.0248,
+      "step": 1628
+    },
+    {
+      "epoch": 95.82,
+      "learning_rate": 2.7414860681114553e-05,
+      "loss": 0.031,
+      "step": 1629
+    },
+    {
+      "epoch": 95.88,
+      "learning_rate": 2.7399380804953563e-05,
+      "loss": 0.025,
+      "step": 1630
+    },
+    {
+      "epoch": 95.94,
+      "learning_rate": 2.7383900928792573e-05,
+      "loss": 0.0292,
+      "step": 1631
+    },
+    {
+      "epoch": 96.0,
+      "learning_rate": 2.7368421052631583e-05,
+      "loss": 0.0252,
+      "step": 1632
+    },
+    {
+      "epoch": 96.06,
+      "learning_rate": 2.7352941176470593e-05,
+      "loss": 0.0314,
+      "step": 1633
+    },
+    {
+      "epoch": 96.12,
+      "learning_rate": 2.7337461300309596e-05,
+      "loss": 0.0229,
+      "step": 1634
+    },
+    {
+      "epoch": 96.18,
+      "learning_rate": 2.7321981424148606e-05,
+      "loss": 0.0283,
+      "step": 1635
+    },
+    {
+      "epoch": 96.24,
+      "learning_rate": 2.7306501547987616e-05,
+      "loss": 0.0284,
+      "step": 1636
+    },
+    {
+      "epoch": 96.29,
+      "learning_rate": 2.729102167182663e-05,
+      "loss": 0.0284,
+      "step": 1637
+    },
+    {
+      "epoch": 96.35,
+      "learning_rate": 2.727554179566564e-05,
+      "loss": 0.0251,
+      "step": 1638
+    },
+    {
+      "epoch": 96.41,
+      "learning_rate": 2.7260061919504642e-05,
+      "loss": 0.0289,
+      "step": 1639
+    },
+    {
+      "epoch": 96.47,
+      "learning_rate": 2.7244582043343652e-05,
+      "loss": 0.0278,
+      "step": 1640
+    },
+    {
+      "epoch": 96.47,
+      "eval_accuracy_NAT": 0.9943527432378615,
+      "eval_accuracy_REST": 0.15934959349593497,
+      "eval_accuracy_concrete": 0.9515000558010974,
+      "eval_accuracy_concrete tiles": 0.9112420457223662,
+      "eval_accuracy_dark bricks": 0.8437111191930019,
+      "eval_accuracy_grass": 0.8814499615831526,
+      "eval_accuracy_red bricks": 0.883018453427065,
+      "eval_accuracy_road": 0.9804510579034534,
+      "eval_accuracy_rough concrete": 0.6292136170517045,
+      "eval_accuracy_rough red sidewalk": 0.7453658096391159,
+      "eval_accuracy_speedway bricks": 0.9892458082369061,
+      "eval_accuracy_steel": 0.9099091895025664,
+      "eval_accuracy_tiles": 0.22478717925414282,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9825262471994339,
+      "eval_iou_REST": 0.15934959349593497,
+      "eval_iou_concrete": 0.8880344977894904,
+      "eval_iou_concrete tiles": 0.8545096912502487,
+      "eval_iou_dark bricks": 0.8284987946938335,
+      "eval_iou_grass": 0.831556329994162,
+      "eval_iou_red bricks": 0.8651993757735564,
+      "eval_iou_road": 0.9586238319746976,
+      "eval_iou_rough concrete": 0.6227750451183693,
+      "eval_iou_rough red sidewalk": 0.5959224900346907,
+      "eval_iou_speedway bricks": 0.9777565042920443,
+      "eval_iou_steel": 0.854774570056293,
+      "eval_iou_tiles": 0.22276261671019845,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08022556453943253,
+      "eval_mean_accuracy": 0.7771997410806437,
+      "eval_mean_iou": 0.7417145837217658,
+      "eval_overall_accuracy": 0.9788875903113413,
+      "eval_runtime": 5.4789,
+      "eval_samples_per_second": 21.537,
+      "eval_steps_per_second": 1.46,
+      "step": 1640
+    },
+    {
+      "epoch": 96.53,
+      "learning_rate": 2.7229102167182662e-05,
+      "loss": 0.0314,
+      "step": 1641
+    },
+    {
+      "epoch": 96.59,
+      "learning_rate": 2.7213622291021672e-05,
+      "loss": 0.0246,
+      "step": 1642
+    },
+    {
+      "epoch": 96.65,
+      "learning_rate": 2.7198142414860682e-05,
+      "loss": 0.0279,
+      "step": 1643
+    },
+    {
+      "epoch": 96.71,
+      "learning_rate": 2.7182662538699692e-05,
+      "loss": 0.0295,
+      "step": 1644
+    },
+    {
+      "epoch": 96.76,
+      "learning_rate": 2.71671826625387e-05,
+      "loss": 0.0274,
+      "step": 1645
+    },
+    {
+      "epoch": 96.82,
+      "learning_rate": 2.715170278637771e-05,
+      "loss": 0.0229,
+      "step": 1646
+    },
+    {
+      "epoch": 96.88,
+      "learning_rate": 2.713622291021672e-05,
+      "loss": 0.0316,
+      "step": 1647
+    },
+    {
+      "epoch": 96.94,
+      "learning_rate": 2.712074303405573e-05,
+      "loss": 0.0229,
+      "step": 1648
+    },
+    {
+      "epoch": 97.0,
+      "learning_rate": 2.710526315789474e-05,
+      "loss": 0.025,
+      "step": 1649
+    },
+    {
+      "epoch": 97.06,
+      "learning_rate": 2.708978328173375e-05,
+      "loss": 0.0247,
+      "step": 1650
+    },
+    {
+      "epoch": 97.12,
+      "learning_rate": 2.7074303405572752e-05,
+      "loss": 0.025,
+      "step": 1651
+    },
+    {
+      "epoch": 97.18,
+      "learning_rate": 2.7058823529411766e-05,
+      "loss": 0.0245,
+      "step": 1652
+    },
+    {
+      "epoch": 97.24,
+      "learning_rate": 2.7043343653250776e-05,
+      "loss": 0.0252,
+      "step": 1653
+    },
+    {
+      "epoch": 97.29,
+      "learning_rate": 2.7027863777089786e-05,
+      "loss": 0.0286,
+      "step": 1654
+    },
+    {
+      "epoch": 97.35,
+      "learning_rate": 2.7012383900928796e-05,
+      "loss": 0.0255,
+      "step": 1655
+    },
+    {
+      "epoch": 97.41,
+      "learning_rate": 2.6996904024767806e-05,
+      "loss": 0.025,
+      "step": 1656
+    },
+    {
+      "epoch": 97.47,
+      "learning_rate": 2.698142414860681e-05,
+      "loss": 0.027,
+      "step": 1657
+    },
+    {
+      "epoch": 97.53,
+      "learning_rate": 2.696594427244582e-05,
+      "loss": 0.0261,
+      "step": 1658
+    },
+    {
+      "epoch": 97.59,
+      "learning_rate": 2.695046439628483e-05,
+      "loss": 0.0274,
+      "step": 1659
+    },
+    {
+      "epoch": 97.65,
+      "learning_rate": 2.693498452012384e-05,
+      "loss": 0.0266,
+      "step": 1660
+    },
+    {
+      "epoch": 97.65,
+      "eval_accuracy_NAT": 0.9932863686435408,
+      "eval_accuracy_REST": 0.2032520325203252,
+      "eval_accuracy_concrete": 0.9568707203233712,
+      "eval_accuracy_concrete tiles": 0.9172519443789772,
+      "eval_accuracy_dark bricks": 0.8505589265547537,
+      "eval_accuracy_grass": 0.8897767659448841,
+      "eval_accuracy_red bricks": 0.8823044815465729,
+      "eval_accuracy_road": 0.9814851531182379,
+      "eval_accuracy_rough concrete": 0.6297081975223306,
+      "eval_accuracy_rough red sidewalk": 0.76765504327751,
+      "eval_accuracy_speedway bricks": 0.9874649815302634,
+      "eval_accuracy_steel": 0.9221113815490262,
+      "eval_accuracy_tiles": 0.21118482838013874,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828826278261484,
+      "eval_iou_REST": 0.20292207792207792,
+      "eval_iou_concrete": 0.8874165119755226,
+      "eval_iou_concrete tiles": 0.8550243859572038,
+      "eval_iou_dark bricks": 0.8347563495641719,
+      "eval_iou_grass": 0.8320688703186553,
+      "eval_iou_red bricks": 0.8653846153846154,
+      "eval_iou_road": 0.9596845495760064,
+      "eval_iou_rough concrete": 0.6214941453918853,
+      "eval_iou_rough red sidewalk": 0.6081123746775722,
+      "eval_iou_speedway bricks": 0.977989755311795,
+      "eval_iou_steel": 0.8557947419199228,
+      "eval_iou_tiles": 0.20927021104139773,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0800696462392807,
+      "eval_mean_accuracy": 0.7840700634838409,
+      "eval_mean_iou": 0.7456000936051519,
+      "eval_overall_accuracy": 0.9790103071827,
+      "eval_runtime": 5.5149,
+      "eval_samples_per_second": 21.397,
+      "eval_steps_per_second": 1.451,
+      "step": 1660
+    },
+    {
+      "epoch": 97.71,
+      "learning_rate": 2.691950464396285e-05,
+      "loss": 0.0285,
+      "step": 1661
+    },
+    {
+      "epoch": 97.76,
+      "learning_rate": 2.6904024767801862e-05,
+      "loss": 0.0256,
+      "step": 1662
+    },
+    {
+      "epoch": 97.82,
+      "learning_rate": 2.6888544891640866e-05,
+      "loss": 0.0255,
+      "step": 1663
+    },
+    {
+      "epoch": 97.88,
+      "learning_rate": 2.6873065015479876e-05,
+      "loss": 0.0292,
+      "step": 1664
+    },
+    {
+      "epoch": 97.94,
+      "learning_rate": 2.6857585139318886e-05,
+      "loss": 0.0286,
+      "step": 1665
+    },
+    {
+      "epoch": 98.0,
+      "learning_rate": 2.6842105263157896e-05,
+      "loss": 0.0371,
+      "step": 1666
+    },
+    {
+      "epoch": 98.06,
+      "learning_rate": 2.6826625386996906e-05,
+      "loss": 0.0292,
+      "step": 1667
+    },
+    {
+      "epoch": 98.12,
+      "learning_rate": 2.6811145510835916e-05,
+      "loss": 0.0248,
+      "step": 1668
+    },
+    {
+      "epoch": 98.18,
+      "learning_rate": 2.6795665634674922e-05,
+      "loss": 0.0259,
+      "step": 1669
+    },
+    {
+      "epoch": 98.24,
+      "learning_rate": 2.6780185758513932e-05,
+      "loss": 0.0287,
+      "step": 1670
+    },
+    {
+      "epoch": 98.29,
+      "learning_rate": 2.6764705882352942e-05,
+      "loss": 0.0251,
+      "step": 1671
+    },
+    {
+      "epoch": 98.35,
+      "learning_rate": 2.6749226006191952e-05,
+      "loss": 0.0234,
+      "step": 1672
+    },
+    {
+      "epoch": 98.41,
+      "learning_rate": 2.6733746130030962e-05,
+      "loss": 0.0234,
+      "step": 1673
+    },
+    {
+      "epoch": 98.47,
+      "learning_rate": 2.6718266253869972e-05,
+      "loss": 0.0246,
+      "step": 1674
+    },
+    {
+      "epoch": 98.53,
+      "learning_rate": 2.6702786377708975e-05,
+      "loss": 0.0318,
+      "step": 1675
+    },
+    {
+      "epoch": 98.59,
+      "learning_rate": 2.668730650154799e-05,
+      "loss": 0.0267,
+      "step": 1676
+    },
+    {
+      "epoch": 98.65,
+      "learning_rate": 2.6671826625387e-05,
+      "loss": 0.0247,
+      "step": 1677
+    },
+    {
+      "epoch": 98.71,
+      "learning_rate": 2.665634674922601e-05,
+      "loss": 0.028,
+      "step": 1678
+    },
+    {
+      "epoch": 98.76,
+      "learning_rate": 2.664086687306502e-05,
+      "loss": 0.028,
+      "step": 1679
+    },
+    {
+      "epoch": 98.82,
+      "learning_rate": 2.662538699690403e-05,
+      "loss": 0.025,
+      "step": 1680
+    },
+    {
+      "epoch": 98.82,
+      "eval_accuracy_NAT": 0.9930489412566151,
+      "eval_accuracy_REST": 0.208130081300813,
+      "eval_accuracy_concrete": 0.9596944087300434,
+      "eval_accuracy_concrete tiles": 0.8977610181475372,
+      "eval_accuracy_dark bricks": 0.8492238194590663,
+      "eval_accuracy_grass": 0.901311282441419,
+      "eval_accuracy_red bricks": 0.8684094903339191,
+      "eval_accuracy_road": 0.9798728196489848,
+      "eval_accuracy_rough concrete": 0.633265601630328,
+      "eval_accuracy_rough red sidewalk": 0.7603832184029057,
+      "eval_accuracy_speedway bricks": 0.9864068016887008,
+      "eval_accuracy_steel": 0.92542993797698,
+      "eval_accuracy_tiles": 0.20046048047502196,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9832012542773101,
+      "eval_iou_REST": 0.20745542949756887,
+      "eval_iou_concrete": 0.8871388173644655,
+      "eval_iou_concrete tiles": 0.8506476105404198,
+      "eval_iou_dark bricks": 0.8329296911668075,
+      "eval_iou_grass": 0.8397989132782655,
+      "eval_iou_red bricks": 0.8552112066634215,
+      "eval_iou_road": 0.9606239023919996,
+      "eval_iou_rough concrete": 0.6286914339801231,
+      "eval_iou_rough red sidewalk": 0.6086901558521302,
+      "eval_iou_speedway bricks": 0.9772341350967851,
+      "eval_iou_steel": 0.854526533418863,
+      "eval_iou_tiles": 0.19860731758561695,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07872192561626434,
+      "eval_mean_accuracy": 0.7817998385763334,
+      "eval_mean_iou": 0.7449812616241368,
+      "eval_overall_accuracy": 0.9791793823242188,
+      "eval_runtime": 5.4267,
+      "eval_samples_per_second": 21.744,
+      "eval_steps_per_second": 1.474,
+      "step": 1680
+    },
+    {
+      "epoch": 98.88,
+      "learning_rate": 2.6609907120743032e-05,
+      "loss": 0.0236,
+      "step": 1681
+    },
+    {
+      "epoch": 98.94,
+      "learning_rate": 2.6594427244582042e-05,
+      "loss": 0.0296,
+      "step": 1682
+    },
+    {
+      "epoch": 99.0,
+      "learning_rate": 2.6578947368421052e-05,
+      "loss": 0.0287,
+      "step": 1683
+    },
+    {
+      "epoch": 99.06,
+      "learning_rate": 2.6563467492260062e-05,
+      "loss": 0.0296,
+      "step": 1684
+    },
+    {
+      "epoch": 99.12,
+      "learning_rate": 2.6547987616099072e-05,
+      "loss": 0.0223,
+      "step": 1685
+    },
+    {
+      "epoch": 99.18,
+      "learning_rate": 2.6532507739938085e-05,
+      "loss": 0.0246,
+      "step": 1686
+    },
+    {
+      "epoch": 99.24,
+      "learning_rate": 2.651702786377709e-05,
+      "loss": 0.027,
+      "step": 1687
+    },
+    {
+      "epoch": 99.29,
+      "learning_rate": 2.65015479876161e-05,
+      "loss": 0.0264,
+      "step": 1688
+    },
+    {
+      "epoch": 99.35,
+      "learning_rate": 2.648606811145511e-05,
+      "loss": 0.0297,
+      "step": 1689
+    },
+    {
+      "epoch": 99.41,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.0279,
+      "step": 1690
+    },
+    {
+      "epoch": 99.47,
+      "learning_rate": 2.645510835913313e-05,
+      "loss": 0.0297,
+      "step": 1691
+    },
+    {
+      "epoch": 99.53,
+      "learning_rate": 2.643962848297214e-05,
+      "loss": 0.0272,
+      "step": 1692
+    },
+    {
+      "epoch": 99.59,
+      "learning_rate": 2.6424148606811145e-05,
+      "loss": 0.0302,
+      "step": 1693
+    },
+    {
+      "epoch": 99.65,
+      "learning_rate": 2.6408668730650155e-05,
+      "loss": 0.0251,
+      "step": 1694
+    },
+    {
+      "epoch": 99.71,
+      "learning_rate": 2.6393188854489165e-05,
+      "loss": 0.0242,
+      "step": 1695
+    },
+    {
+      "epoch": 99.76,
+      "learning_rate": 2.6377708978328175e-05,
+      "loss": 0.0251,
+      "step": 1696
+    },
+    {
+      "epoch": 99.82,
+      "learning_rate": 2.6362229102167185e-05,
+      "loss": 0.0257,
+      "step": 1697
+    },
+    {
+      "epoch": 99.88,
+      "learning_rate": 2.6346749226006195e-05,
+      "loss": 0.0279,
+      "step": 1698
+    },
+    {
+      "epoch": 99.94,
+      "learning_rate": 2.63312693498452e-05,
+      "loss": 0.026,
+      "step": 1699
+    },
+    {
+      "epoch": 100.0,
+      "learning_rate": 2.6315789473684212e-05,
+      "loss": 0.0268,
+      "step": 1700
+    },
+    {
+      "epoch": 100.0,
+      "eval_accuracy_NAT": 0.9931490553932003,
+      "eval_accuracy_REST": 0.18699186991869918,
+      "eval_accuracy_concrete": 0.9548618808180288,
+      "eval_accuracy_concrete tiles": 0.9205986330426585,
+      "eval_accuracy_dark bricks": 0.8659485481308501,
+      "eval_accuracy_grass": 0.8962130072253952,
+      "eval_accuracy_red bricks": 0.8848308435852372,
+      "eval_accuracy_road": 0.9825561230461078,
+      "eval_accuracy_rough concrete": 0.6367097885221578,
+      "eval_accuracy_rough red sidewalk": 0.7756863865723159,
+      "eval_accuracy_speedway bricks": 0.9882243005388406,
+      "eval_accuracy_steel": 0.9146821132807229,
+      "eval_accuracy_tiles": 0.2170620133902875,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9829670466292932,
+      "eval_iou_REST": 0.18608414239482202,
+      "eval_iou_concrete": 0.8890780130738865,
+      "eval_iou_concrete tiles": 0.8482670257123002,
+      "eval_iou_dark bricks": 0.8459823098212282,
+      "eval_iou_grass": 0.8332721890107376,
+      "eval_iou_red bricks": 0.867068510844411,
+      "eval_iou_road": 0.9595579236230388,
+      "eval_iou_rough concrete": 0.6280756615684845,
+      "eval_iou_rough red sidewalk": 0.6210338723855167,
+      "eval_iou_speedway bricks": 0.9784463626911498,
+      "eval_iou_steel": 0.8557207882229448,
+      "eval_iou_tiles": 0.21484902096014874,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07878053188323975,
+      "eval_mean_accuracy": 0.7859626587280388,
+      "eval_mean_iou": 0.7469540666875355,
+      "eval_overall_accuracy": 0.9792783704854674,
+      "eval_runtime": 7.6625,
+      "eval_samples_per_second": 15.4,
+      "eval_steps_per_second": 1.044,
+      "step": 1700
+    },
+    {
+      "epoch": 100.06,
+      "learning_rate": 2.6300309597523222e-05,
+      "loss": 0.0232,
+      "step": 1701
+    },
+    {
+      "epoch": 100.12,
+      "learning_rate": 2.6284829721362232e-05,
+      "loss": 0.0266,
+      "step": 1702
+    },
+    {
+      "epoch": 100.18,
+      "learning_rate": 2.6269349845201242e-05,
+      "loss": 0.0256,
+      "step": 1703
+    },
+    {
+      "epoch": 100.24,
+      "learning_rate": 2.6253869969040252e-05,
+      "loss": 0.0244,
+      "step": 1704
+    },
+    {
+      "epoch": 100.29,
+      "learning_rate": 2.6238390092879255e-05,
+      "loss": 0.024,
+      "step": 1705
+    },
+    {
+      "epoch": 100.35,
+      "learning_rate": 2.6222910216718265e-05,
+      "loss": 0.0299,
+      "step": 1706
+    },
+    {
+      "epoch": 100.41,
+      "learning_rate": 2.6207430340557275e-05,
+      "loss": 0.0254,
+      "step": 1707
+    },
+    {
+      "epoch": 100.47,
+      "learning_rate": 2.6191950464396285e-05,
+      "loss": 0.0261,
+      "step": 1708
+    },
+    {
+      "epoch": 100.53,
+      "learning_rate": 2.6176470588235295e-05,
+      "loss": 0.0284,
+      "step": 1709
+    },
+    {
+      "epoch": 100.59,
+      "learning_rate": 2.616099071207431e-05,
+      "loss": 0.0251,
+      "step": 1710
+    },
+    {
+      "epoch": 100.65,
+      "learning_rate": 2.6145510835913312e-05,
+      "loss": 0.0285,
+      "step": 1711
+    },
+    {
+      "epoch": 100.71,
+      "learning_rate": 2.6130030959752322e-05,
+      "loss": 0.027,
+      "step": 1712
+    },
+    {
+      "epoch": 100.76,
+      "learning_rate": 2.6114551083591332e-05,
+      "loss": 0.0234,
+      "step": 1713
+    },
+    {
+      "epoch": 100.82,
+      "learning_rate": 2.6099071207430342e-05,
+      "loss": 0.029,
+      "step": 1714
+    },
+    {
+      "epoch": 100.88,
+      "learning_rate": 2.6083591331269352e-05,
+      "loss": 0.0258,
+      "step": 1715
+    },
+    {
+      "epoch": 100.94,
+      "learning_rate": 2.6068111455108362e-05,
+      "loss": 0.029,
+      "step": 1716
+    },
+    {
+      "epoch": 101.0,
+      "learning_rate": 2.605263157894737e-05,
+      "loss": 0.0301,
+      "step": 1717
+    },
+    {
+      "epoch": 101.06,
+      "learning_rate": 2.603715170278638e-05,
+      "loss": 0.0259,
+      "step": 1718
+    },
+    {
+      "epoch": 101.12,
+      "learning_rate": 2.602167182662539e-05,
+      "loss": 0.0246,
+      "step": 1719
+    },
+    {
+      "epoch": 101.18,
+      "learning_rate": 2.60061919504644e-05,
+      "loss": 0.0256,
+      "step": 1720
+    },
+    {
+      "epoch": 101.18,
+      "eval_accuracy_NAT": 0.9942035154887675,
+      "eval_accuracy_REST": 0.1902439024390244,
+      "eval_accuracy_concrete": 0.9515806998527768,
+      "eval_accuracy_concrete tiles": 0.9300966297431063,
+      "eval_accuracy_dark bricks": 0.846797657102387,
+      "eval_accuracy_grass": 0.8849065316210396,
+      "eval_accuracy_red bricks": 0.9161357644991213,
+      "eval_accuracy_road": 0.9808262180278869,
+      "eval_accuracy_rough concrete": 0.6354405639409125,
+      "eval_accuracy_rough red sidewalk": 0.7714451153941599,
+      "eval_accuracy_speedway bricks": 0.9888542388542388,
+      "eval_accuracy_steel": 0.9183180316562295,
+      "eval_accuracy_tiles": 0.23308794571177557,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9826267529036181,
+      "eval_iou_REST": 0.18962722852512157,
+      "eval_iou_concrete": 0.8895014004801646,
+      "eval_iou_concrete tiles": 0.852372621438908,
+      "eval_iou_dark bricks": 0.8317993795243019,
+      "eval_iou_grass": 0.8334824675232079,
+      "eval_iou_red bricks": 0.8921275002674083,
+      "eval_iou_road": 0.9605173843983954,
+      "eval_iou_rough concrete": 0.6270927299136152,
+      "eval_iou_rough red sidewalk": 0.6190956108217714,
+      "eval_iou_speedway bricks": 0.9779966262600891,
+      "eval_iou_steel": 0.8559406713141425,
+      "eval_iou_tiles": 0.23030411877394635,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07852399349212646,
+      "eval_mean_accuracy": 0.7878412934101096,
+      "eval_mean_iou": 0.7494218840111302,
+      "eval_overall_accuracy": 0.9792022058519266,
+      "eval_runtime": 5.4796,
+      "eval_samples_per_second": 21.534,
+      "eval_steps_per_second": 1.46,
+      "step": 1720
+    },
+    {
+      "epoch": 101.24,
+      "learning_rate": 2.599071207430341e-05,
+      "loss": 0.0285,
+      "step": 1721
+    },
+    {
+      "epoch": 101.29,
+      "learning_rate": 2.597523219814242e-05,
+      "loss": 0.0269,
+      "step": 1722
+    },
+    {
+      "epoch": 101.35,
+      "learning_rate": 2.5959752321981422e-05,
+      "loss": 0.0276,
+      "step": 1723
+    },
+    {
+      "epoch": 101.41,
+      "learning_rate": 2.5944272445820432e-05,
+      "loss": 0.0237,
+      "step": 1724
+    },
+    {
+      "epoch": 101.47,
+      "learning_rate": 2.5928792569659445e-05,
+      "loss": 0.0225,
+      "step": 1725
+    },
+    {
+      "epoch": 101.53,
+      "learning_rate": 2.5913312693498455e-05,
+      "loss": 0.0266,
+      "step": 1726
+    },
+    {
+      "epoch": 101.59,
+      "learning_rate": 2.5897832817337465e-05,
+      "loss": 0.0251,
+      "step": 1727
+    },
+    {
+      "epoch": 101.65,
+      "learning_rate": 2.5882352941176475e-05,
+      "loss": 0.0315,
+      "step": 1728
+    },
+    {
+      "epoch": 101.71,
+      "learning_rate": 2.586687306501548e-05,
+      "loss": 0.0262,
+      "step": 1729
+    },
+    {
+      "epoch": 101.76,
+      "learning_rate": 2.585139318885449e-05,
+      "loss": 0.0292,
+      "step": 1730
+    },
+    {
+      "epoch": 101.82,
+      "learning_rate": 2.58359133126935e-05,
+      "loss": 0.0246,
+      "step": 1731
+    },
+    {
+      "epoch": 101.88,
+      "learning_rate": 2.582043343653251e-05,
+      "loss": 0.026,
+      "step": 1732
+    },
+    {
+      "epoch": 101.94,
+      "learning_rate": 2.580495356037152e-05,
+      "loss": 0.025,
+      "step": 1733
+    },
+    {
+      "epoch": 102.0,
+      "learning_rate": 2.578947368421053e-05,
+      "loss": 0.0223,
+      "step": 1734
+    },
+    {
+      "epoch": 102.06,
+      "learning_rate": 2.5773993808049535e-05,
+      "loss": 0.0262,
+      "step": 1735
+    },
+    {
+      "epoch": 102.12,
+      "learning_rate": 2.5758513931888545e-05,
+      "loss": 0.0277,
+      "step": 1736
+    },
+    {
+      "epoch": 102.18,
+      "learning_rate": 2.5743034055727555e-05,
+      "loss": 0.024,
+      "step": 1737
+    },
+    {
+      "epoch": 102.24,
+      "learning_rate": 2.5727554179566565e-05,
+      "loss": 0.0226,
+      "step": 1738
+    },
+    {
+      "epoch": 102.29,
+      "learning_rate": 2.5712074303405575e-05,
+      "loss": 0.0244,
+      "step": 1739
+    },
+    {
+      "epoch": 102.35,
+      "learning_rate": 2.5696594427244585e-05,
+      "loss": 0.0256,
+      "step": 1740
+    },
+    {
+      "epoch": 102.35,
+      "eval_accuracy_NAT": 0.9930296947586287,
+      "eval_accuracy_REST": 0.32195121951219513,
+      "eval_accuracy_concrete": 0.9563195889370121,
+      "eval_accuracy_concrete tiles": 0.9174169219891586,
+      "eval_accuracy_dark bricks": 0.8542053480849109,
+      "eval_accuracy_grass": 0.8969236242794476,
+      "eval_accuracy_red bricks": 0.9228361159929701,
+      "eval_accuracy_road": 0.9798273942777928,
+      "eval_accuracy_rough concrete": 0.6241784303326798,
+      "eval_accuracy_rough red sidewalk": 0.7796268583761345,
+      "eval_accuracy_speedway bricks": 0.9893088706145086,
+      "eval_accuracy_steel": 0.9138699665645444,
+      "eval_accuracy_tiles": 0.2346632736526402,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830898959521086,
+      "eval_iou_REST": 0.31629392971246006,
+      "eval_iou_concrete": 0.8871943846989002,
+      "eval_iou_concrete tiles": 0.8580434686769828,
+      "eval_iou_dark bricks": 0.8359362926678499,
+      "eval_iou_grass": 0.8383820247774899,
+      "eval_iou_red bricks": 0.895921087709944,
+      "eval_iou_road": 0.9589891733760979,
+      "eval_iou_rough concrete": 0.6186926707106026,
+      "eval_iou_rough red sidewalk": 0.6221174098543628,
+      "eval_iou_speedway bricks": 0.9785152565458746,
+      "eval_iou_steel": 0.8582817795217934,
+      "eval_iou_tiles": 0.22653097034567468,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07949712872505188,
+      "eval_mean_accuracy": 0.7987813313363556,
+      "eval_mean_iou": 0.7598452572730878,
+      "eval_overall_accuracy": 0.9792924977965274,
+      "eval_runtime": 5.6854,
+      "eval_samples_per_second": 20.755,
+      "eval_steps_per_second": 1.407,
+      "step": 1740
+    },
+    {
+      "epoch": 102.41,
+      "learning_rate": 2.568111455108359e-05,
+      "loss": 0.0259,
+      "step": 1741
+    },
+    {
+      "epoch": 102.47,
+      "learning_rate": 2.56656346749226e-05,
+      "loss": 0.0268,
+      "step": 1742
+    },
+    {
+      "epoch": 102.53,
+      "learning_rate": 2.565015479876161e-05,
+      "loss": 0.0275,
+      "step": 1743
+    },
+    {
+      "epoch": 102.59,
+      "learning_rate": 2.563467492260062e-05,
+      "loss": 0.0222,
+      "step": 1744
+    },
+    {
+      "epoch": 102.65,
+      "learning_rate": 2.561919504643963e-05,
+      "loss": 0.0242,
+      "step": 1745
+    },
+    {
+      "epoch": 102.71,
+      "learning_rate": 2.560371517027864e-05,
+      "loss": 0.0306,
+      "step": 1746
+    },
+    {
+      "epoch": 102.76,
+      "learning_rate": 2.5588235294117645e-05,
+      "loss": 0.0261,
+      "step": 1747
+    },
+    {
+      "epoch": 102.82,
+      "learning_rate": 2.5572755417956655e-05,
+      "loss": 0.0279,
+      "step": 1748
+    },
+    {
+      "epoch": 102.88,
+      "learning_rate": 2.5557275541795668e-05,
+      "loss": 0.0253,
+      "step": 1749
+    },
+    {
+      "epoch": 102.94,
+      "learning_rate": 2.5541795665634678e-05,
+      "loss": 0.0261,
+      "step": 1750
+    },
+    {
+      "epoch": 103.0,
+      "learning_rate": 2.5526315789473688e-05,
+      "loss": 0.0299,
+      "step": 1751
+    },
+    {
+      "epoch": 103.06,
+      "learning_rate": 2.5510835913312698e-05,
+      "loss": 0.0243,
+      "step": 1752
+    },
+    {
+      "epoch": 103.12,
+      "learning_rate": 2.54953560371517e-05,
+      "loss": 0.0265,
+      "step": 1753
+    },
+    {
+      "epoch": 103.18,
+      "learning_rate": 2.547987616099071e-05,
+      "loss": 0.0262,
+      "step": 1754
+    },
+    {
+      "epoch": 103.24,
+      "learning_rate": 2.546439628482972e-05,
+      "loss": 0.0238,
+      "step": 1755
+    },
+    {
+      "epoch": 103.29,
+      "learning_rate": 2.544891640866873e-05,
+      "loss": 0.0277,
+      "step": 1756
+    },
+    {
+      "epoch": 103.35,
+      "learning_rate": 2.543343653250774e-05,
+      "loss": 0.0262,
+      "step": 1757
+    },
+    {
+      "epoch": 103.41,
+      "learning_rate": 2.541795665634675e-05,
+      "loss": 0.028,
+      "step": 1758
+    },
+    {
+      "epoch": 103.47,
+      "learning_rate": 2.5402476780185758e-05,
+      "loss": 0.0269,
+      "step": 1759
+    },
+    {
+      "epoch": 103.53,
+      "learning_rate": 2.5386996904024768e-05,
+      "loss": 0.0243,
+      "step": 1760
+    },
+    {
+      "epoch": 103.53,
+      "eval_accuracy_NAT": 0.9933396334614978,
+      "eval_accuracy_REST": 0.1967479674796748,
+      "eval_accuracy_concrete": 0.9556090804437792,
+      "eval_accuracy_concrete tiles": 0.9193730850813104,
+      "eval_accuracy_dark bricks": 0.8592203740214,
+      "eval_accuracy_grass": 0.9019064597077292,
+      "eval_accuracy_red bricks": 0.9078427065026362,
+      "eval_accuracy_road": 0.9790503532223422,
+      "eval_accuracy_rough concrete": 0.6303279128108259,
+      "eval_accuracy_rough red sidewalk": 0.762909933147339,
+      "eval_accuracy_speedway bricks": 0.9882760528161122,
+      "eval_accuracy_steel": 0.9217615337328262,
+      "eval_accuracy_tiles": 0.23042200611954317,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830367578528209,
+      "eval_iou_REST": 0.19611021069692058,
+      "eval_iou_concrete": 0.8895426466659005,
+      "eval_iou_concrete tiles": 0.8594373085989998,
+      "eval_iou_dark bricks": 0.8390444725860643,
+      "eval_iou_grass": 0.8390714543517874,
+      "eval_iou_red bricks": 0.8899057873485868,
+      "eval_iou_road": 0.960197406523577,
+      "eval_iou_rough concrete": 0.6229887629861717,
+      "eval_iou_rough red sidewalk": 0.6181212224605186,
+      "eval_iou_speedway bricks": 0.9782112999791537,
+      "eval_iou_steel": 0.8581289199903221,
+      "eval_iou_tiles": 0.22726186207720808,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07805890589952469,
+      "eval_mean_accuracy": 0.7882143921959244,
+      "eval_mean_iou": 0.7508506240090793,
+      "eval_overall_accuracy": 0.9794309583760924,
+      "eval_runtime": 5.4263,
+      "eval_samples_per_second": 21.746,
+      "eval_steps_per_second": 1.474,
+      "step": 1760
+    },
+    {
+      "epoch": 103.59,
+      "learning_rate": 2.5371517027863778e-05,
+      "loss": 0.0262,
+      "step": 1761
+    },
+    {
+      "epoch": 103.65,
+      "learning_rate": 2.5356037151702788e-05,
+      "loss": 0.0215,
+      "step": 1762
+    },
+    {
+      "epoch": 103.71,
+      "learning_rate": 2.5340557275541798e-05,
+      "loss": 0.0276,
+      "step": 1763
+    },
+    {
+      "epoch": 103.76,
+      "learning_rate": 2.5325077399380808e-05,
+      "loss": 0.0265,
+      "step": 1764
+    },
+    {
+      "epoch": 103.82,
+      "learning_rate": 2.5309597523219815e-05,
+      "loss": 0.0255,
+      "step": 1765
+    },
+    {
+      "epoch": 103.88,
+      "learning_rate": 2.5294117647058825e-05,
+      "loss": 0.0259,
+      "step": 1766
+    },
+    {
+      "epoch": 103.94,
+      "learning_rate": 2.5278637770897835e-05,
+      "loss": 0.0267,
+      "step": 1767
+    },
+    {
+      "epoch": 104.0,
+      "learning_rate": 2.5263157894736845e-05,
+      "loss": 0.0259,
+      "step": 1768
+    },
+    {
+      "epoch": 104.06,
+      "learning_rate": 2.5247678018575855e-05,
+      "loss": 0.0242,
+      "step": 1769
+    },
+    {
+      "epoch": 104.12,
+      "learning_rate": 2.5232198142414865e-05,
+      "loss": 0.0238,
+      "step": 1770
+    },
+    {
+      "epoch": 104.18,
+      "learning_rate": 2.5216718266253868e-05,
+      "loss": 0.0228,
+      "step": 1771
+    },
+    {
+      "epoch": 104.24,
+      "learning_rate": 2.5201238390092878e-05,
+      "loss": 0.0271,
+      "step": 1772
+    },
+    {
+      "epoch": 104.29,
+      "learning_rate": 2.518575851393189e-05,
+      "loss": 0.0261,
+      "step": 1773
+    },
+    {
+      "epoch": 104.35,
+      "learning_rate": 2.51702786377709e-05,
+      "loss": 0.0252,
+      "step": 1774
+    },
+    {
+      "epoch": 104.41,
+      "learning_rate": 2.515479876160991e-05,
+      "loss": 0.0262,
+      "step": 1775
+    },
+    {
+      "epoch": 104.47,
+      "learning_rate": 2.513931888544892e-05,
+      "loss": 0.0236,
+      "step": 1776
+    },
+    {
+      "epoch": 104.53,
+      "learning_rate": 2.5123839009287925e-05,
+      "loss": 0.0276,
+      "step": 1777
+    },
+    {
+      "epoch": 104.59,
+      "learning_rate": 2.5108359133126935e-05,
+      "loss": 0.0311,
+      "step": 1778
+    },
+    {
+      "epoch": 104.65,
+      "learning_rate": 2.5092879256965945e-05,
+      "loss": 0.0275,
+      "step": 1779
+    },
+    {
+      "epoch": 104.71,
+      "learning_rate": 2.5077399380804955e-05,
+      "loss": 0.0299,
+      "step": 1780
+    },
+    {
+      "epoch": 104.71,
+      "eval_accuracy_NAT": 0.9930651147843349,
+      "eval_accuracy_REST": 0.2032520325203252,
+      "eval_accuracy_concrete": 0.9583880515326956,
+      "eval_accuracy_concrete tiles": 0.9249823238274806,
+      "eval_accuracy_dark bricks": 0.8484533813142431,
+      "eval_accuracy_grass": 0.9020767807060374,
+      "eval_accuracy_red bricks": 0.8708809314586995,
+      "eval_accuracy_road": 0.9782508666893616,
+      "eval_accuracy_rough concrete": 0.6224622956876158,
+      "eval_accuracy_rough red sidewalk": 0.7793937388610231,
+      "eval_accuracy_speedway bricks": 0.9883850410556642,
+      "eval_accuracy_steel": 0.9215966054766177,
+      "eval_accuracy_tiles": 0.2051258747614287,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830730174290276,
+      "eval_iou_REST": 0.2025931928687196,
+      "eval_iou_concrete": 0.88835956661153,
+      "eval_iou_concrete tiles": 0.8547565118912798,
+      "eval_iou_dark bricks": 0.8335002185961894,
+      "eval_iou_grass": 0.8378676008532128,
+      "eval_iou_red bricks": 0.8555627495413834,
+      "eval_iou_road": 0.9612219596072362,
+      "eval_iou_rough concrete": 0.6178930557198627,
+      "eval_iou_rough red sidewalk": 0.625187749956267,
+      "eval_iou_speedway bricks": 0.9787689230155185,
+      "eval_iou_steel": 0.8586865413092677,
+      "eval_iou_tiles": 0.20330280738627834,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07847295701503754,
+      "eval_mean_accuracy": 0.7843317722058097,
+      "eval_mean_iou": 0.7462133765219826,
+      "eval_overall_accuracy": 0.9793918092372054,
+      "eval_runtime": 6.5633,
+      "eval_samples_per_second": 17.979,
+      "eval_steps_per_second": 1.219,
+      "step": 1780
+    },
+    {
+      "epoch": 104.76,
+      "learning_rate": 2.5061919504643965e-05,
+      "loss": 0.028,
+      "step": 1781
+    },
+    {
+      "epoch": 104.82,
+      "learning_rate": 2.5046439628482975e-05,
+      "loss": 0.021,
+      "step": 1782
+    },
+    {
+      "epoch": 104.88,
+      "learning_rate": 2.503095975232198e-05,
+      "loss": 0.0282,
+      "step": 1783
+    },
+    {
+      "epoch": 104.94,
+      "learning_rate": 2.501547987616099e-05,
+      "loss": 0.023,
+      "step": 1784
+    },
+    {
+      "epoch": 105.0,
+      "learning_rate": 2.5e-05,
+      "loss": 0.0269,
+      "step": 1785
+    },
+    {
+      "epoch": 105.06,
+      "learning_rate": 2.498452012383901e-05,
+      "loss": 0.0272,
+      "step": 1786
+    },
+    {
+      "epoch": 105.12,
+      "learning_rate": 2.4969040247678018e-05,
+      "loss": 0.0297,
+      "step": 1787
+    },
+    {
+      "epoch": 105.18,
+      "learning_rate": 2.4953560371517028e-05,
+      "loss": 0.0244,
+      "step": 1788
+    },
+    {
+      "epoch": 105.24,
+      "learning_rate": 2.4938080495356038e-05,
+      "loss": 0.028,
+      "step": 1789
+    },
+    {
+      "epoch": 105.29,
+      "learning_rate": 2.4922600619195048e-05,
+      "loss": 0.0297,
+      "step": 1790
+    },
+    {
+      "epoch": 105.35,
+      "learning_rate": 2.4907120743034058e-05,
+      "loss": 0.0235,
+      "step": 1791
+    },
+    {
+      "epoch": 105.41,
+      "learning_rate": 2.4891640866873068e-05,
+      "loss": 0.0194,
+      "step": 1792
+    },
+    {
+      "epoch": 105.47,
+      "learning_rate": 2.4876160990712074e-05,
+      "loss": 0.0238,
+      "step": 1793
+    },
+    {
+      "epoch": 105.53,
+      "learning_rate": 2.4860681114551084e-05,
+      "loss": 0.0255,
+      "step": 1794
+    },
+    {
+      "epoch": 105.59,
+      "learning_rate": 2.4845201238390094e-05,
+      "loss": 0.0243,
+      "step": 1795
+    },
+    {
+      "epoch": 105.65,
+      "learning_rate": 2.48297213622291e-05,
+      "loss": 0.0299,
+      "step": 1796
+    },
+    {
+      "epoch": 105.71,
+      "learning_rate": 2.481424148606811e-05,
+      "loss": 0.0231,
+      "step": 1797
+    },
+    {
+      "epoch": 105.76,
+      "learning_rate": 2.4798761609907124e-05,
+      "loss": 0.0267,
+      "step": 1798
+    },
+    {
+      "epoch": 105.82,
+      "learning_rate": 2.478328173374613e-05,
+      "loss": 0.0251,
+      "step": 1799
+    },
+    {
+      "epoch": 105.88,
+      "learning_rate": 2.476780185758514e-05,
+      "loss": 0.0238,
+      "step": 1800
+    },
+    {
+      "epoch": 105.88,
+      "eval_accuracy_NAT": 0.9938548681428893,
+      "eval_accuracy_REST": 0.17886178861788618,
+      "eval_accuracy_concrete": 0.9530441409612159,
+      "eval_accuracy_concrete tiles": 0.9208814518029695,
+      "eval_accuracy_dark bricks": 0.8513437206898532,
+      "eval_accuracy_grass": 0.8997888019620979,
+      "eval_accuracy_red bricks": 0.8795035149384886,
+      "eval_accuracy_road": 0.9815033232667147,
+      "eval_accuracy_rough concrete": 0.6352320059111305,
+      "eval_accuracy_rough red sidewalk": 0.7671211243880612,
+      "eval_accuracy_speedway bricks": 0.9888307618277945,
+      "eval_accuracy_steel": 0.9115934568462719,
+      "eval_accuracy_tiles": 0.23687479172346937,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830029434348605,
+      "eval_iou_REST": 0.17886178861788618,
+      "eval_iou_concrete": 0.8912250265643479,
+      "eval_iou_concrete tiles": 0.8586151581075439,
+      "eval_iou_dark bricks": 0.8351304282515526,
+      "eval_iou_grass": 0.8372148305711489,
+      "eval_iou_red bricks": 0.865387733045123,
+      "eval_iou_road": 0.9615402738148217,
+      "eval_iou_rough concrete": 0.6287763504028501,
+      "eval_iou_rough red sidewalk": 0.6171451042076288,
+      "eval_iou_speedway bricks": 0.97849911829648,
+      "eval_iou_steel": 0.8580326187687285,
+      "eval_iou_tiles": 0.23510132900354802,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07701998949050903,
+      "eval_mean_accuracy": 0.7844949039291418,
+      "eval_mean_iou": 0.7483486694681939,
+      "eval_overall_accuracy": 0.9795555179401979,
+      "eval_runtime": 8.2948,
+      "eval_samples_per_second": 14.226,
+      "eval_steps_per_second": 0.964,
+      "step": 1800
+    },
+    {
+      "epoch": 105.94,
+      "learning_rate": 2.475232198142415e-05,
+      "loss": 0.0257,
+      "step": 1801
+    },
+    {
+      "epoch": 106.0,
+      "learning_rate": 2.4736842105263158e-05,
+      "loss": 0.0301,
+      "step": 1802
+    },
+    {
+      "epoch": 106.06,
+      "learning_rate": 2.4721362229102168e-05,
+      "loss": 0.0227,
+      "step": 1803
+    },
+    {
+      "epoch": 106.12,
+      "learning_rate": 2.4705882352941178e-05,
+      "loss": 0.0262,
+      "step": 1804
+    },
+    {
+      "epoch": 106.18,
+      "learning_rate": 2.4690402476780188e-05,
+      "loss": 0.0222,
+      "step": 1805
+    },
+    {
+      "epoch": 106.24,
+      "learning_rate": 2.4674922600619198e-05,
+      "loss": 0.0281,
+      "step": 1806
+    },
+    {
+      "epoch": 106.29,
+      "learning_rate": 2.4659442724458208e-05,
+      "loss": 0.0272,
+      "step": 1807
+    },
+    {
+      "epoch": 106.35,
+      "learning_rate": 2.4643962848297214e-05,
+      "loss": 0.0269,
+      "step": 1808
+    },
+    {
+      "epoch": 106.41,
+      "learning_rate": 2.4628482972136224e-05,
+      "loss": 0.0283,
+      "step": 1809
+    },
+    {
+      "epoch": 106.47,
+      "learning_rate": 2.4613003095975234e-05,
+      "loss": 0.0224,
+      "step": 1810
+    },
+    {
+      "epoch": 106.53,
+      "learning_rate": 2.459752321981424e-05,
+      "loss": 0.0258,
+      "step": 1811
+    },
+    {
+      "epoch": 106.59,
+      "learning_rate": 2.458204334365325e-05,
+      "loss": 0.0268,
+      "step": 1812
+    },
+    {
+      "epoch": 106.65,
+      "learning_rate": 2.456656346749226e-05,
+      "loss": 0.027,
+      "step": 1813
+    },
+    {
+      "epoch": 106.71,
+      "learning_rate": 2.455108359133127e-05,
+      "loss": 0.0317,
+      "step": 1814
+    },
+    {
+      "epoch": 106.76,
+      "learning_rate": 2.453560371517028e-05,
+      "loss": 0.0225,
+      "step": 1815
+    },
+    {
+      "epoch": 106.82,
+      "learning_rate": 2.452012383900929e-05,
+      "loss": 0.0243,
+      "step": 1816
+    },
+    {
+      "epoch": 106.88,
+      "learning_rate": 2.4504643962848298e-05,
+      "loss": 0.0243,
+      "step": 1817
+    },
+    {
+      "epoch": 106.94,
+      "learning_rate": 2.4489164086687308e-05,
+      "loss": 0.0253,
+      "step": 1818
+    },
+    {
+      "epoch": 107.0,
+      "learning_rate": 2.4473684210526318e-05,
+      "loss": 0.0252,
+      "step": 1819
+    },
+    {
+      "epoch": 107.06,
+      "learning_rate": 2.4458204334365324e-05,
+      "loss": 0.0283,
+      "step": 1820
+    },
+    {
+      "epoch": 107.06,
+      "eval_accuracy_NAT": 0.9929890452922931,
+      "eval_accuracy_REST": 0.23577235772357724,
+      "eval_accuracy_concrete": 0.9605107864285616,
+      "eval_accuracy_concrete tiles": 0.9186660381805326,
+      "eval_accuracy_dark bricks": 0.8513820033305898,
+      "eval_accuracy_grass": 0.8909822601217985,
+      "eval_accuracy_red bricks": 0.9089411247803163,
+      "eval_accuracy_road": 0.9779820553751307,
+      "eval_accuracy_rough concrete": 0.6272889243768583,
+      "eval_accuracy_rough red sidewalk": 0.8059468036306484,
+      "eval_accuracy_speedway bricks": 0.9887879205386624,
+      "eval_accuracy_steel": 0.909671792770145,
+      "eval_accuracy_tiles": 0.21818291980974885,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828682544078575,
+      "eval_iou_REST": 0.23424878836833601,
+      "eval_iou_concrete": 0.8885194259712985,
+      "eval_iou_concrete tiles": 0.8594959317324866,
+      "eval_iou_dark bricks": 0.835179743317717,
+      "eval_iou_grass": 0.8343515934922321,
+      "eval_iou_red bricks": 0.8891634878848117,
+      "eval_iou_road": 0.9607504769624762,
+      "eval_iou_rough concrete": 0.6210348713047685,
+      "eval_iou_rough red sidewalk": 0.6203455561022198,
+      "eval_iou_speedway bricks": 0.9785727174917339,
+      "eval_iou_steel": 0.858531972047895,
+      "eval_iou_tiles": 0.2165428906461409,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07921610027551651,
+      "eval_mean_accuracy": 0.7913156947968356,
+      "eval_mean_iou": 0.7522773622869211,
+      "eval_overall_accuracy": 0.9792406114481264,
+      "eval_runtime": 5.5587,
+      "eval_samples_per_second": 21.228,
+      "eval_steps_per_second": 1.439,
+      "step": 1820
+    },
+    {
+      "epoch": 107.12,
+      "learning_rate": 2.4442724458204334e-05,
+      "loss": 0.0265,
+      "step": 1821
+    },
+    {
+      "epoch": 107.18,
+      "learning_rate": 2.4427244582043348e-05,
+      "loss": 0.0259,
+      "step": 1822
+    },
+    {
+      "epoch": 107.24,
+      "learning_rate": 2.4411764705882354e-05,
+      "loss": 0.0238,
+      "step": 1823
+    },
+    {
+      "epoch": 107.29,
+      "learning_rate": 2.4396284829721364e-05,
+      "loss": 0.0275,
+      "step": 1824
+    },
+    {
+      "epoch": 107.35,
+      "learning_rate": 2.4380804953560374e-05,
+      "loss": 0.0259,
+      "step": 1825
+    },
+    {
+      "epoch": 107.41,
+      "learning_rate": 2.436532507739938e-05,
+      "loss": 0.0218,
+      "step": 1826
+    },
+    {
+      "epoch": 107.47,
+      "learning_rate": 2.434984520123839e-05,
+      "loss": 0.0239,
+      "step": 1827
+    },
+    {
+      "epoch": 107.53,
+      "learning_rate": 2.43343653250774e-05,
+      "loss": 0.025,
+      "step": 1828
+    },
+    {
+      "epoch": 107.59,
+      "learning_rate": 2.431888544891641e-05,
+      "loss": 0.0244,
+      "step": 1829
+    },
+    {
+      "epoch": 107.65,
+      "learning_rate": 2.430340557275542e-05,
+      "loss": 0.0288,
+      "step": 1830
+    },
+    {
+      "epoch": 107.71,
+      "learning_rate": 2.428792569659443e-05,
+      "loss": 0.0272,
+      "step": 1831
+    },
+    {
+      "epoch": 107.76,
+      "learning_rate": 2.4272445820433437e-05,
+      "loss": 0.0256,
+      "step": 1832
+    },
+    {
+      "epoch": 107.82,
+      "learning_rate": 2.4256965944272447e-05,
+      "loss": 0.0267,
+      "step": 1833
+    },
+    {
+      "epoch": 107.88,
+      "learning_rate": 2.4241486068111457e-05,
+      "loss": 0.0277,
+      "step": 1834
+    },
+    {
+      "epoch": 107.94,
+      "learning_rate": 2.4226006191950464e-05,
+      "loss": 0.0242,
+      "step": 1835
+    },
+    {
+      "epoch": 108.0,
+      "learning_rate": 2.4210526315789474e-05,
+      "loss": 0.0336,
+      "step": 1836
+    },
+    {
+      "epoch": 108.06,
+      "learning_rate": 2.4195046439628484e-05,
+      "loss": 0.0234,
+      "step": 1837
+    },
+    {
+      "epoch": 108.12,
+      "learning_rate": 2.4179566563467494e-05,
+      "loss": 0.0262,
+      "step": 1838
+    },
+    {
+      "epoch": 108.18,
+      "learning_rate": 2.4164086687306504e-05,
+      "loss": 0.026,
+      "step": 1839
+    },
+    {
+      "epoch": 108.24,
+      "learning_rate": 2.4148606811145514e-05,
+      "loss": 0.0236,
+      "step": 1840
+    },
+    {
+      "epoch": 108.24,
+      "eval_accuracy_NAT": 0.9933820620158825,
+      "eval_accuracy_REST": 0.3024390243902439,
+      "eval_accuracy_concrete": 0.9578594699048629,
+      "eval_accuracy_concrete tiles": 0.9290596276219656,
+      "eval_accuracy_dark bricks": 0.8560046321995292,
+      "eval_accuracy_grass": 0.8894881664755286,
+      "eval_accuracy_red bricks": 0.9201449912126538,
+      "eval_accuracy_road": 0.9805317547393358,
+      "eval_accuracy_rough concrete": 0.6351783767034722,
+      "eval_accuracy_rough red sidewalk": 0.7654592078448477,
+      "eval_accuracy_speedway bricks": 0.9883523103107673,
+      "eval_accuracy_steel": 0.9078750743426609,
+      "eval_accuracy_tiles": 0.24090399587991154,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828156761297722,
+      "eval_iou_REST": 0.2980769230769231,
+      "eval_iou_concrete": 0.888829385605746,
+      "eval_iou_concrete tiles": 0.8549677923091938,
+      "eval_iou_dark bricks": 0.8388323563892145,
+      "eval_iou_grass": 0.8342149938190586,
+      "eval_iou_red bricks": 0.892879982946067,
+      "eval_iou_road": 0.9591808675313785,
+      "eval_iou_rough concrete": 0.6230783619167864,
+      "eval_iou_rough red sidewalk": 0.6138769110152881,
+      "eval_iou_speedway bricks": 0.978251769411969,
+      "eval_iou_steel": 0.8569595471164052,
+      "eval_iou_tiles": 0.2367300765085886,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07888595759868622,
+      "eval_mean_accuracy": 0.7974368225878202,
+      "eval_mean_iou": 0.7583611264443378,
+      "eval_overall_accuracy": 0.9791961928545418,
+      "eval_runtime": 5.4523,
+      "eval_samples_per_second": 21.642,
+      "eval_steps_per_second": 1.467,
+      "step": 1840
+    },
+    {
+      "epoch": 108.29,
+      "learning_rate": 2.413312693498452e-05,
+      "loss": 0.0316,
+      "step": 1841
+    },
+    {
+      "epoch": 108.35,
+      "learning_rate": 2.411764705882353e-05,
+      "loss": 0.0211,
+      "step": 1842
+    },
+    {
+      "epoch": 108.41,
+      "learning_rate": 2.410216718266254e-05,
+      "loss": 0.0281,
+      "step": 1843
+    },
+    {
+      "epoch": 108.47,
+      "learning_rate": 2.4086687306501547e-05,
+      "loss": 0.0224,
+      "step": 1844
+    },
+    {
+      "epoch": 108.53,
+      "learning_rate": 2.4071207430340557e-05,
+      "loss": 0.025,
+      "step": 1845
+    },
+    {
+      "epoch": 108.59,
+      "learning_rate": 2.405572755417957e-05,
+      "loss": 0.0232,
+      "step": 1846
+    },
+    {
+      "epoch": 108.65,
+      "learning_rate": 2.4040247678018577e-05,
+      "loss": 0.0237,
+      "step": 1847
+    },
+    {
+      "epoch": 108.71,
+      "learning_rate": 2.4024767801857587e-05,
+      "loss": 0.0253,
+      "step": 1848
+    },
+    {
+      "epoch": 108.76,
+      "learning_rate": 2.4009287925696597e-05,
+      "loss": 0.0325,
+      "step": 1849
+    },
+    {
+      "epoch": 108.82,
+      "learning_rate": 2.3993808049535604e-05,
+      "loss": 0.0227,
+      "step": 1850
+    },
+    {
+      "epoch": 108.88,
+      "learning_rate": 2.3978328173374614e-05,
+      "loss": 0.024,
+      "step": 1851
+    },
+    {
+      "epoch": 108.94,
+      "learning_rate": 2.3962848297213624e-05,
+      "loss": 0.026,
+      "step": 1852
+    },
+    {
+      "epoch": 109.0,
+      "learning_rate": 2.394736842105263e-05,
+      "loss": 0.0241,
+      "step": 1853
+    },
+    {
+      "epoch": 109.06,
+      "learning_rate": 2.3931888544891644e-05,
+      "loss": 0.0234,
+      "step": 1854
+    },
+    {
+      "epoch": 109.12,
+      "learning_rate": 2.3916408668730654e-05,
+      "loss": 0.0252,
+      "step": 1855
+    },
+    {
+      "epoch": 109.18,
+      "learning_rate": 2.390092879256966e-05,
+      "loss": 0.0237,
+      "step": 1856
+    },
+    {
+      "epoch": 109.24,
+      "learning_rate": 2.388544891640867e-05,
+      "loss": 0.0276,
+      "step": 1857
+    },
+    {
+      "epoch": 109.29,
+      "learning_rate": 2.386996904024768e-05,
+      "loss": 0.0253,
+      "step": 1858
+    },
+    {
+      "epoch": 109.35,
+      "learning_rate": 2.3854489164086687e-05,
+      "loss": 0.0303,
+      "step": 1859
+    },
+    {
+      "epoch": 109.41,
+      "learning_rate": 2.3839009287925697e-05,
+      "loss": 0.0248,
+      "step": 1860
+    },
+    {
+      "epoch": 109.41,
+      "eval_accuracy_NAT": 0.9935593238796907,
+      "eval_accuracy_REST": 0.2926829268292683,
+      "eval_accuracy_concrete": 0.9552834466426849,
+      "eval_accuracy_concrete tiles": 0.9286118312514731,
+      "eval_accuracy_dark bricks": 0.8521093735045844,
+      "eval_accuracy_grass": 0.8933251200763038,
+      "eval_accuracy_red bricks": 0.9070188927943761,
+      "eval_accuracy_road": 0.9818966535396251,
+      "eval_accuracy_rough concrete": 0.6298512087427526,
+      "eval_accuracy_rough red sidewalk": 0.7726257529384339,
+      "eval_accuracy_speedway bricks": 0.9893052719462215,
+      "eval_accuracy_steel": 0.9060558656984212,
+      "eval_accuracy_tiles": 0.23896513072192432,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983031373654946,
+      "eval_iou_REST": 0.28846153846153844,
+      "eval_iou_concrete": 0.8900859873949729,
+      "eval_iou_concrete tiles": 0.8462959383121764,
+      "eval_iou_dark bricks": 0.8344439445912764,
+      "eval_iou_grass": 0.8374798190366363,
+      "eval_iou_red bricks": 0.885285446261056,
+      "eval_iou_road": 0.9591957341374461,
+      "eval_iou_rough concrete": 0.6220486805866152,
+      "eval_iou_rough red sidewalk": 0.6224283325659729,
+      "eval_iou_speedway bricks": 0.978403241073025,
+      "eval_iou_steel": 0.8568228221415608,
+      "eval_iou_tiles": 0.23652883144921888,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07837572693824768,
+      "eval_mean_accuracy": 0.7954839075819815,
+      "eval_mean_iou": 0.7569624376666493,
+      "eval_overall_accuracy": 0.9794194819563526,
+      "eval_runtime": 5.2932,
+      "eval_samples_per_second": 22.293,
+      "eval_steps_per_second": 1.511,
+      "step": 1860
+    },
+    {
+      "epoch": 109.47,
+      "learning_rate": 2.3823529411764707e-05,
+      "loss": 0.0271,
+      "step": 1861
+    },
+    {
+      "epoch": 109.53,
+      "learning_rate": 2.3808049535603717e-05,
+      "loss": 0.0254,
+      "step": 1862
+    },
+    {
+      "epoch": 109.59,
+      "learning_rate": 2.3792569659442727e-05,
+      "loss": 0.0242,
+      "step": 1863
+    },
+    {
+      "epoch": 109.65,
+      "learning_rate": 2.3777089783281734e-05,
+      "loss": 0.0245,
+      "step": 1864
+    },
+    {
+      "epoch": 109.71,
+      "learning_rate": 2.3761609907120744e-05,
+      "loss": 0.0221,
+      "step": 1865
+    },
+    {
+      "epoch": 109.76,
+      "learning_rate": 2.3746130030959754e-05,
+      "loss": 0.026,
+      "step": 1866
+    },
+    {
+      "epoch": 109.82,
+      "learning_rate": 2.373065015479876e-05,
+      "loss": 0.022,
+      "step": 1867
+    },
+    {
+      "epoch": 109.88,
+      "learning_rate": 2.371517027863777e-05,
+      "loss": 0.0314,
+      "step": 1868
+    },
+    {
+      "epoch": 109.94,
+      "learning_rate": 2.369969040247678e-05,
+      "loss": 0.0237,
+      "step": 1869
+    },
+    {
+      "epoch": 110.0,
+      "learning_rate": 2.368421052631579e-05,
+      "loss": 0.0258,
+      "step": 1870
+    },
+    {
+      "epoch": 110.06,
+      "learning_rate": 2.36687306501548e-05,
+      "loss": 0.0243,
+      "step": 1871
+    },
+    {
+      "epoch": 110.12,
+      "learning_rate": 2.365325077399381e-05,
+      "loss": 0.0264,
+      "step": 1872
+    },
+    {
+      "epoch": 110.18,
+      "learning_rate": 2.3637770897832817e-05,
+      "loss": 0.0299,
+      "step": 1873
+    },
+    {
+      "epoch": 110.24,
+      "learning_rate": 2.3622291021671827e-05,
+      "loss": 0.0222,
+      "step": 1874
+    },
+    {
+      "epoch": 110.29,
+      "learning_rate": 2.3606811145510837e-05,
+      "loss": 0.0281,
+      "step": 1875
+    },
+    {
+      "epoch": 110.35,
+      "learning_rate": 2.3591331269349844e-05,
+      "loss": 0.0243,
+      "step": 1876
+    },
+    {
+      "epoch": 110.41,
+      "learning_rate": 2.3575851393188854e-05,
+      "loss": 0.0228,
+      "step": 1877
+    },
+    {
+      "epoch": 110.47,
+      "learning_rate": 2.3560371517027867e-05,
+      "loss": 0.0224,
+      "step": 1878
+    },
+    {
+      "epoch": 110.53,
+      "learning_rate": 2.3544891640866874e-05,
+      "loss": 0.0211,
+      "step": 1879
+    },
+    {
+      "epoch": 110.59,
+      "learning_rate": 2.3529411764705884e-05,
+      "loss": 0.0296,
+      "step": 1880
+    },
+    {
+      "epoch": 110.59,
+      "eval_accuracy_NAT": 0.9933384474027983,
+      "eval_accuracy_REST": 0.15609756097560976,
+      "eval_accuracy_concrete": 0.9523030565242184,
+      "eval_accuracy_concrete tiles": 0.934904548668395,
+      "eval_accuracy_dark bricks": 0.8548705089677086,
+      "eval_accuracy_grass": 0.9021893818104744,
+      "eval_accuracy_red bricks": 0.883293057996485,
+      "eval_accuracy_road": 0.9829825871191817,
+      "eval_accuracy_rough concrete": 0.6291480702423444,
+      "eval_accuracy_rough red sidewalk": 0.7901548364779402,
+      "eval_accuracy_speedway bricks": 0.9887961460661757,
+      "eval_accuracy_steel": 0.9161889578033555,
+      "eval_accuracy_tiles": 0.2347844527250144,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830663717385475,
+      "eval_iou_REST": 0.15609756097560976,
+      "eval_iou_concrete": 0.8906924561533328,
+      "eval_iou_concrete tiles": 0.8561685228351895,
+      "eval_iou_dark bricks": 0.838586114631742,
+      "eval_iou_grass": 0.8375069941648682,
+      "eval_iou_red bricks": 0.8651890903222336,
+      "eval_iou_road": 0.9607486008132651,
+      "eval_iou_rough concrete": 0.6204924776680771,
+      "eval_iou_rough red sidewalk": 0.6305448871819491,
+      "eval_iou_speedway bricks": 0.9783077604918503,
+      "eval_iou_steel": 0.8587647242073112,
+      "eval_iou_tiles": 0.23311776206948415,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07774849236011505,
+      "eval_mean_accuracy": 0.7860808932907463,
+      "eval_mean_iou": 0.7468679479425737,
+      "eval_overall_accuracy": 0.9795162071615963,
+      "eval_runtime": 5.5042,
+      "eval_samples_per_second": 21.438,
+      "eval_steps_per_second": 1.453,
+      "step": 1880
+    },
+    {
+      "epoch": 110.65,
+      "learning_rate": 2.3513931888544894e-05,
+      "loss": 0.0256,
+      "step": 1881
+    },
+    {
+      "epoch": 110.71,
+      "learning_rate": 2.34984520123839e-05,
+      "loss": 0.0268,
+      "step": 1882
+    },
+    {
+      "epoch": 110.76,
+      "learning_rate": 2.348297213622291e-05,
+      "loss": 0.0269,
+      "step": 1883
+    },
+    {
+      "epoch": 110.82,
+      "learning_rate": 2.346749226006192e-05,
+      "loss": 0.0269,
+      "step": 1884
+    },
+    {
+      "epoch": 110.88,
+      "learning_rate": 2.345201238390093e-05,
+      "loss": 0.024,
+      "step": 1885
+    },
+    {
+      "epoch": 110.94,
+      "learning_rate": 2.343653250773994e-05,
+      "loss": 0.0224,
+      "step": 1886
+    },
+    {
+      "epoch": 111.0,
+      "learning_rate": 2.342105263157895e-05,
+      "loss": 0.0291,
+      "step": 1887
+    },
+    {
+      "epoch": 111.06,
+      "learning_rate": 2.3405572755417957e-05,
+      "loss": 0.0246,
+      "step": 1888
+    },
+    {
+      "epoch": 111.12,
+      "learning_rate": 2.3390092879256967e-05,
+      "loss": 0.0255,
+      "step": 1889
+    },
+    {
+      "epoch": 111.18,
+      "learning_rate": 2.3374613003095977e-05,
+      "loss": 0.0246,
+      "step": 1890
+    },
+    {
+      "epoch": 111.24,
+      "learning_rate": 2.3359133126934984e-05,
+      "loss": 0.023,
+      "step": 1891
+    },
+    {
+      "epoch": 111.29,
+      "learning_rate": 2.3343653250773994e-05,
+      "loss": 0.024,
+      "step": 1892
+    },
+    {
+      "epoch": 111.35,
+      "learning_rate": 2.3328173374613004e-05,
+      "loss": 0.0251,
+      "step": 1893
+    },
+    {
+      "epoch": 111.41,
+      "learning_rate": 2.3312693498452014e-05,
+      "loss": 0.0249,
+      "step": 1894
+    },
+    {
+      "epoch": 111.47,
+      "learning_rate": 2.3297213622291024e-05,
+      "loss": 0.0274,
+      "step": 1895
+    },
+    {
+      "epoch": 111.53,
+      "learning_rate": 2.3281733746130034e-05,
+      "loss": 0.0252,
+      "step": 1896
+    },
+    {
+      "epoch": 111.59,
+      "learning_rate": 2.326625386996904e-05,
+      "loss": 0.0257,
+      "step": 1897
+    },
+    {
+      "epoch": 111.65,
+      "learning_rate": 2.325077399380805e-05,
+      "loss": 0.0244,
+      "step": 1898
+    },
+    {
+      "epoch": 111.71,
+      "learning_rate": 2.323529411764706e-05,
+      "loss": 0.0271,
+      "step": 1899
+    },
+    {
+      "epoch": 111.76,
+      "learning_rate": 2.3219814241486067e-05,
+      "loss": 0.0234,
+      "step": 1900
+    },
+    {
+      "epoch": 111.76,
+      "eval_accuracy_NAT": 0.9936789001612986,
+      "eval_accuracy_REST": 0.2764227642276423,
+      "eval_accuracy_concrete": 0.9536040629314074,
+      "eval_accuracy_concrete tiles": 0.9373792128211171,
+      "eval_accuracy_dark bricks": 0.8558754282870432,
+      "eval_accuracy_grass": 0.8959206228449662,
+      "eval_accuracy_red bricks": 0.9210786467486819,
+      "eval_accuracy_road": 0.979130515642093,
+      "eval_accuracy_rough concrete": 0.6245478759854367,
+      "eval_accuracy_rough red sidewalk": 0.7705803171929403,
+      "eval_accuracy_speedway bricks": 0.9895931654091892,
+      "eval_accuracy_steel": 0.9138149904791416,
+      "eval_accuracy_tiles": 0.24208549183556,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828312551940153,
+      "eval_iou_REST": 0.27419354838709675,
+      "eval_iou_concrete": 0.8900715076903319,
+      "eval_iou_concrete tiles": 0.8577497897302077,
+      "eval_iou_dark bricks": 0.8385799082904324,
+      "eval_iou_grass": 0.8377786743964636,
+      "eval_iou_red bricks": 0.8940718626719266,
+      "eval_iou_road": 0.9596493166986786,
+      "eval_iou_rough concrete": 0.6196078199542437,
+      "eval_iou_rough red sidewalk": 0.6235456625450297,
+      "eval_iou_speedway bricks": 0.9784859803044292,
+      "eval_iou_steel": 0.8595777389145888,
+      "eval_iou_tiles": 0.23997717649178654,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07985126227140427,
+      "eval_mean_accuracy": 0.7964393841974245,
+      "eval_mean_iou": 0.7581630954822486,
+      "eval_overall_accuracy": 0.9794363571425616,
+      "eval_runtime": 5.5768,
+      "eval_samples_per_second": 21.159,
+      "eval_steps_per_second": 1.435,
+      "step": 1900
+    },
+    {
+      "epoch": 111.82,
+      "learning_rate": 2.3204334365325077e-05,
+      "loss": 0.0257,
+      "step": 1901
+    },
+    {
+      "epoch": 111.88,
+      "learning_rate": 2.318885448916409e-05,
+      "loss": 0.0289,
+      "step": 1902
+    },
+    {
+      "epoch": 111.94,
+      "learning_rate": 2.3173374613003097e-05,
+      "loss": 0.028,
+      "step": 1903
+    },
+    {
+      "epoch": 112.0,
+      "learning_rate": 2.3157894736842107e-05,
+      "loss": 0.019,
+      "step": 1904
+    },
+    {
+      "epoch": 112.06,
+      "learning_rate": 2.3142414860681117e-05,
+      "loss": 0.0231,
+      "step": 1905
+    },
+    {
+      "epoch": 112.12,
+      "learning_rate": 2.3126934984520123e-05,
+      "loss": 0.0193,
+      "step": 1906
+    },
+    {
+      "epoch": 112.18,
+      "learning_rate": 2.3111455108359133e-05,
+      "loss": 0.0259,
+      "step": 1907
+    },
+    {
+      "epoch": 112.24,
+      "learning_rate": 2.3095975232198143e-05,
+      "loss": 0.0286,
+      "step": 1908
+    },
+    {
+      "epoch": 112.29,
+      "learning_rate": 2.3080495356037153e-05,
+      "loss": 0.0255,
+      "step": 1909
+    },
+    {
+      "epoch": 112.35,
+      "learning_rate": 2.3065015479876163e-05,
+      "loss": 0.0247,
+      "step": 1910
+    },
+    {
+      "epoch": 112.41,
+      "learning_rate": 2.3049535603715173e-05,
+      "loss": 0.0268,
+      "step": 1911
+    },
+    {
+      "epoch": 112.47,
+      "learning_rate": 2.303405572755418e-05,
+      "loss": 0.0239,
+      "step": 1912
+    },
+    {
+      "epoch": 112.53,
+      "learning_rate": 2.301857585139319e-05,
+      "loss": 0.0258,
+      "step": 1913
+    },
+    {
+      "epoch": 112.59,
+      "learning_rate": 2.30030959752322e-05,
+      "loss": 0.0244,
+      "step": 1914
+    },
+    {
+      "epoch": 112.65,
+      "learning_rate": 2.2987616099071207e-05,
+      "loss": 0.0268,
+      "step": 1915
+    },
+    {
+      "epoch": 112.71,
+      "learning_rate": 2.2972136222910217e-05,
+      "loss": 0.0241,
+      "step": 1916
+    },
+    {
+      "epoch": 112.76,
+      "learning_rate": 2.2956656346749227e-05,
+      "loss": 0.0237,
+      "step": 1917
+    },
+    {
+      "epoch": 112.82,
+      "learning_rate": 2.2941176470588237e-05,
+      "loss": 0.0217,
+      "step": 1918
+    },
+    {
+      "epoch": 112.88,
+      "learning_rate": 2.2925696594427247e-05,
+      "loss": 0.0231,
+      "step": 1919
+    },
+    {
+      "epoch": 112.94,
+      "learning_rate": 2.2910216718266257e-05,
+      "loss": 0.0256,
+      "step": 1920
+    },
+    {
+      "epoch": 112.94,
+      "eval_accuracy_NAT": 0.9930772988418838,
+      "eval_accuracy_REST": 0.34146341463414637,
+      "eval_accuracy_concrete": 0.9587457900747429,
+      "eval_accuracy_concrete tiles": 0.9101107706811219,
+      "eval_accuracy_dark bricks": 0.8472570487912257,
+      "eval_accuracy_grass": 0.8977004772772864,
+      "eval_accuracy_red bricks": 0.9015817223198594,
+      "eval_accuracy_road": 0.9779959501945542,
+      "eval_accuracy_rough concrete": 0.625459572515627,
+      "eval_accuracy_rough red sidewalk": 0.7750697478549244,
+      "eval_accuracy_speedway bricks": 0.9892122206662266,
+      "eval_accuracy_steel": 0.9107688155652292,
+      "eval_accuracy_tiles": 0.22902844678723985,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9829664138405338,
+      "eval_iou_REST": 0.33707865168539325,
+      "eval_iou_concrete": 0.8883792698742562,
+      "eval_iou_concrete tiles": 0.8634290313925409,
+      "eval_iou_dark bricks": 0.8311996206733049,
+      "eval_iou_grass": 0.8369739551444809,
+      "eval_iou_red bricks": 0.8852459016393442,
+      "eval_iou_road": 0.9599356065811795,
+      "eval_iou_rough concrete": 0.6191397494278367,
+      "eval_iou_rough red sidewalk": 0.6203347597638265,
+      "eval_iou_speedway bricks": 0.9782631352840842,
+      "eval_iou_steel": 0.8604561669417078,
+      "eval_iou_tiles": 0.22731372902760238,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08010214567184448,
+      "eval_mean_accuracy": 0.7967285597080053,
+      "eval_mean_iou": 0.7608243070212377,
+      "eval_overall_accuracy": 0.9792959245584779,
+      "eval_runtime": 5.7267,
+      "eval_samples_per_second": 20.605,
+      "eval_steps_per_second": 1.397,
+      "step": 1920
+    },
+    {
+      "epoch": 113.0,
+      "learning_rate": 2.2894736842105263e-05,
+      "loss": 0.0285,
+      "step": 1921
+    },
+    {
+      "epoch": 113.06,
+      "learning_rate": 2.2879256965944273e-05,
+      "loss": 0.0246,
+      "step": 1922
+    },
+    {
+      "epoch": 113.12,
+      "learning_rate": 2.2863777089783283e-05,
+      "loss": 0.0267,
+      "step": 1923
+    },
+    {
+      "epoch": 113.18,
+      "learning_rate": 2.284829721362229e-05,
+      "loss": 0.0272,
+      "step": 1924
+    },
+    {
+      "epoch": 113.24,
+      "learning_rate": 2.28328173374613e-05,
+      "loss": 0.0217,
+      "step": 1925
+    },
+    {
+      "epoch": 113.29,
+      "learning_rate": 2.281733746130031e-05,
+      "loss": 0.0262,
+      "step": 1926
+    },
+    {
+      "epoch": 113.35,
+      "learning_rate": 2.280185758513932e-05,
+      "loss": 0.0277,
+      "step": 1927
+    },
+    {
+      "epoch": 113.41,
+      "learning_rate": 2.278637770897833e-05,
+      "loss": 0.0223,
+      "step": 1928
+    },
+    {
+      "epoch": 113.47,
+      "learning_rate": 2.277089783281734e-05,
+      "loss": 0.026,
+      "step": 1929
+    },
+    {
+      "epoch": 113.53,
+      "learning_rate": 2.2755417956656347e-05,
+      "loss": 0.0237,
+      "step": 1930
+    },
+    {
+      "epoch": 113.59,
+      "learning_rate": 2.2739938080495357e-05,
+      "loss": 0.0242,
+      "step": 1931
+    },
+    {
+      "epoch": 113.65,
+      "learning_rate": 2.2724458204334367e-05,
+      "loss": 0.0277,
+      "step": 1932
+    },
+    {
+      "epoch": 113.71,
+      "learning_rate": 2.2708978328173373e-05,
+      "loss": 0.0233,
+      "step": 1933
+    },
+    {
+      "epoch": 113.76,
+      "learning_rate": 2.2693498452012387e-05,
+      "loss": 0.0224,
+      "step": 1934
+    },
+    {
+      "epoch": 113.82,
+      "learning_rate": 2.2678018575851397e-05,
+      "loss": 0.0221,
+      "step": 1935
+    },
+    {
+      "epoch": 113.88,
+      "learning_rate": 2.2662538699690403e-05,
+      "loss": 0.0224,
+      "step": 1936
+    },
+    {
+      "epoch": 113.94,
+      "learning_rate": 2.2647058823529413e-05,
+      "loss": 0.0283,
+      "step": 1937
+    },
+    {
+      "epoch": 114.0,
+      "learning_rate": 2.2631578947368423e-05,
+      "loss": 0.0252,
+      "step": 1938
+    },
+    {
+      "epoch": 114.06,
+      "learning_rate": 2.261609907120743e-05,
+      "loss": 0.0266,
+      "step": 1939
+    },
+    {
+      "epoch": 114.12,
+      "learning_rate": 2.260061919504644e-05,
+      "loss": 0.022,
+      "step": 1940
+    },
+    {
+      "epoch": 114.12,
+      "eval_accuracy_NAT": 0.9932994692009938,
+      "eval_accuracy_REST": 0.37886178861788616,
+      "eval_accuracy_concrete": 0.9556320124016028,
+      "eval_accuracy_concrete tiles": 0.9383219420221541,
+      "eval_accuracy_dark bricks": 0.8549949275501024,
+      "eval_accuracy_grass": 0.8948334071390993,
+      "eval_accuracy_red bricks": 0.9146528998242531,
+      "eval_accuracy_road": 0.9788739958988906,
+      "eval_accuracy_rough concrete": 0.6287786245895876,
+      "eval_accuracy_rough red sidewalk": 0.7992389775829266,
+      "eval_accuracy_speedway bricks": 0.9892010819310523,
+      "eval_accuracy_steel": 0.9180506479681338,
+      "eval_accuracy_tiles": 0.2431155139507407,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830294749707805,
+      "eval_iou_REST": 0.3612403100775194,
+      "eval_iou_concrete": 0.8901498766958986,
+      "eval_iou_concrete tiles": 0.8593537525092275,
+      "eval_iou_dark bricks": 0.8374933790821182,
+      "eval_iou_grass": 0.8356831864803986,
+      "eval_iou_red bricks": 0.8921626399528579,
+      "eval_iou_road": 0.9606835508897364,
+      "eval_iou_rough concrete": 0.6224289363007356,
+      "eval_iou_rough red sidewalk": 0.6208786073139385,
+      "eval_iou_speedway bricks": 0.9784748168561811,
+      "eval_iou_steel": 0.8615894070797705,
+      "eval_iou_tiles": 0.24065133295348906,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0785597562789917,
+      "eval_mean_accuracy": 0.8067580991290325,
+      "eval_mean_iou": 0.7649091747048195,
+      "eval_overall_accuracy": 0.979449385303562,
+      "eval_runtime": 10.2248,
+      "eval_samples_per_second": 11.541,
+      "eval_steps_per_second": 0.782,
+      "step": 1940
+    },
+    {
+      "epoch": 114.18,
+      "learning_rate": 2.258513931888545e-05,
+      "loss": 0.0249,
+      "step": 1941
+    },
+    {
+      "epoch": 114.24,
+      "learning_rate": 2.256965944272446e-05,
+      "loss": 0.0252,
+      "step": 1942
+    },
+    {
+      "epoch": 114.29,
+      "learning_rate": 2.255417956656347e-05,
+      "loss": 0.0225,
+      "step": 1943
+    },
+    {
+      "epoch": 114.35,
+      "learning_rate": 2.253869969040248e-05,
+      "loss": 0.0237,
+      "step": 1944
+    },
+    {
+      "epoch": 114.41,
+      "learning_rate": 2.2523219814241486e-05,
+      "loss": 0.0241,
+      "step": 1945
+    },
+    {
+      "epoch": 114.47,
+      "learning_rate": 2.2507739938080496e-05,
+      "loss": 0.0244,
+      "step": 1946
+    },
+    {
+      "epoch": 114.53,
+      "learning_rate": 2.2492260061919506e-05,
+      "loss": 0.024,
+      "step": 1947
+    },
+    {
+      "epoch": 114.59,
+      "learning_rate": 2.2476780185758513e-05,
+      "loss": 0.0267,
+      "step": 1948
+    },
+    {
+      "epoch": 114.65,
+      "learning_rate": 2.2461300309597523e-05,
+      "loss": 0.0246,
+      "step": 1949
+    },
+    {
+      "epoch": 114.71,
+      "learning_rate": 2.2445820433436533e-05,
+      "loss": 0.0294,
+      "step": 1950
+    },
+    {
+      "epoch": 114.76,
+      "learning_rate": 2.2430340557275543e-05,
+      "loss": 0.029,
+      "step": 1951
+    },
+    {
+      "epoch": 114.82,
+      "learning_rate": 2.2414860681114553e-05,
+      "loss": 0.0296,
+      "step": 1952
+    },
+    {
+      "epoch": 114.88,
+      "learning_rate": 2.2399380804953563e-05,
+      "loss": 0.0249,
+      "step": 1953
+    },
+    {
+      "epoch": 114.94,
+      "learning_rate": 2.238390092879257e-05,
+      "loss": 0.0248,
+      "step": 1954
+    },
+    {
+      "epoch": 115.0,
+      "learning_rate": 2.236842105263158e-05,
+      "loss": 0.0248,
+      "step": 1955
+    },
+    {
+      "epoch": 115.06,
+      "learning_rate": 2.235294117647059e-05,
+      "loss": 0.0264,
+      "step": 1956
+    },
+    {
+      "epoch": 115.12,
+      "learning_rate": 2.2337461300309596e-05,
+      "loss": 0.0261,
+      "step": 1957
+    },
+    {
+      "epoch": 115.18,
+      "learning_rate": 2.232198142414861e-05,
+      "loss": 0.0239,
+      "step": 1958
+    },
+    {
+      "epoch": 115.24,
+      "learning_rate": 2.230650154798762e-05,
+      "loss": 0.0207,
+      "step": 1959
+    },
+    {
+      "epoch": 115.29,
+      "learning_rate": 2.2291021671826626e-05,
+      "loss": 0.0235,
+      "step": 1960
+    },
+    {
+      "epoch": 115.29,
+      "eval_accuracy_NAT": 0.9931815641839169,
+      "eval_accuracy_REST": 0.34796747967479674,
+      "eval_accuracy_concrete": 0.955950002216756,
+      "eval_accuracy_concrete tiles": 0.9250530285175583,
+      "eval_accuracy_dark bricks": 0.8506259211760427,
+      "eval_accuracy_grass": 0.893872985954195,
+      "eval_accuracy_red bricks": 0.9053163444639719,
+      "eval_accuracy_road": 0.9798615969102197,
+      "eval_accuracy_rough concrete": 0.6273544711862185,
+      "eval_accuracy_rough red sidewalk": 0.7835598101955948,
+      "eval_accuracy_speedway bricks": 0.9904719259318666,
+      "eval_accuracy_steel": 0.9062832667789511,
+      "eval_accuracy_tiles": 0.236359780665879,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9829906725469955,
+      "eval_iou_REST": 0.3424,
+      "eval_iou_concrete": 0.8900952877136643,
+      "eval_iou_concrete tiles": 0.8614824081999957,
+      "eval_iou_dark bricks": 0.8323905408569422,
+      "eval_iou_grass": 0.8352106350171875,
+      "eval_iou_red bricks": 0.8877638948728996,
+      "eval_iou_road": 0.9598394751637758,
+      "eval_iou_rough concrete": 0.6206713593434968,
+      "eval_iou_rough red sidewalk": 0.6234786563109584,
+      "eval_iou_speedway bricks": 0.9783074014920654,
+      "eval_iou_steel": 0.8596707058065005,
+      "eval_iou_tiles": 0.2342942942942943,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0790090560913086,
+      "eval_mean_accuracy": 0.7996813982966129,
+      "eval_mean_iou": 0.7621996408937518,
+      "eval_overall_accuracy": 0.9793838565632448,
+      "eval_runtime": 5.5086,
+      "eval_samples_per_second": 21.421,
+      "eval_steps_per_second": 1.452,
+      "step": 1960
+    },
+    {
+      "epoch": 115.35,
+      "learning_rate": 2.2275541795665636e-05,
+      "loss": 0.0271,
+      "step": 1961
+    },
+    {
+      "epoch": 115.41,
+      "learning_rate": 2.2260061919504646e-05,
+      "loss": 0.0292,
+      "step": 1962
+    },
+    {
+      "epoch": 115.47,
+      "learning_rate": 2.2244582043343653e-05,
+      "loss": 0.0256,
+      "step": 1963
+    },
+    {
+      "epoch": 115.53,
+      "learning_rate": 2.2229102167182663e-05,
+      "loss": 0.0239,
+      "step": 1964
+    },
+    {
+      "epoch": 115.59,
+      "learning_rate": 2.2213622291021673e-05,
+      "loss": 0.0255,
+      "step": 1965
+    },
+    {
+      "epoch": 115.65,
+      "learning_rate": 2.2198142414860683e-05,
+      "loss": 0.0246,
+      "step": 1966
+    },
+    {
+      "epoch": 115.71,
+      "learning_rate": 2.2182662538699693e-05,
+      "loss": 0.0251,
+      "step": 1967
+    },
+    {
+      "epoch": 115.76,
+      "learning_rate": 2.2167182662538703e-05,
+      "loss": 0.0235,
+      "step": 1968
+    },
+    {
+      "epoch": 115.82,
+      "learning_rate": 2.215170278637771e-05,
+      "loss": 0.0243,
+      "step": 1969
+    },
+    {
+      "epoch": 115.88,
+      "learning_rate": 2.213622291021672e-05,
+      "loss": 0.0226,
+      "step": 1970
+    },
+    {
+      "epoch": 115.94,
+      "learning_rate": 2.212074303405573e-05,
+      "loss": 0.022,
+      "step": 1971
+    },
+    {
+      "epoch": 116.0,
+      "learning_rate": 2.2105263157894736e-05,
+      "loss": 0.0266,
+      "step": 1972
+    },
+    {
+      "epoch": 116.06,
+      "learning_rate": 2.2089783281733746e-05,
+      "loss": 0.0254,
+      "step": 1973
+    },
+    {
+      "epoch": 116.12,
+      "learning_rate": 2.2074303405572756e-05,
+      "loss": 0.0253,
+      "step": 1974
+    },
+    {
+      "epoch": 116.18,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.0246,
+      "step": 1975
+    },
+    {
+      "epoch": 116.24,
+      "learning_rate": 2.2043343653250776e-05,
+      "loss": 0.0204,
+      "step": 1976
+    },
+    {
+      "epoch": 116.29,
+      "learning_rate": 2.2027863777089786e-05,
+      "loss": 0.0211,
+      "step": 1977
+    },
+    {
+      "epoch": 116.35,
+      "learning_rate": 2.2012383900928793e-05,
+      "loss": 0.0257,
+      "step": 1978
+    },
+    {
+      "epoch": 116.41,
+      "learning_rate": 2.1996904024767803e-05,
+      "loss": 0.0282,
+      "step": 1979
+    },
+    {
+      "epoch": 116.47,
+      "learning_rate": 2.1981424148606813e-05,
+      "loss": 0.0292,
+      "step": 1980
+    },
+    {
+      "epoch": 116.47,
+      "eval_accuracy_NAT": 0.9939344418992703,
+      "eval_accuracy_REST": 0.3544715447154472,
+      "eval_accuracy_concrete": 0.9549803626001171,
+      "eval_accuracy_concrete tiles": 0.926749941079425,
+      "eval_accuracy_dark bricks": 0.8587992649732978,
+      "eval_accuracy_grass": 0.8872049188704312,
+      "eval_accuracy_red bricks": 0.9185522847100176,
+      "eval_accuracy_road": 0.9808550764989972,
+      "eval_accuracy_rough concrete": 0.6221405204416663,
+      "eval_accuracy_rough red sidewalk": 0.7914181938501568,
+      "eval_accuracy_speedway bricks": 0.988326605537288,
+      "eval_accuracy_steel": 0.9214966489577034,
+      "eval_accuracy_tiles": 0.243297282559302,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9828808122561766,
+      "eval_iou_REST": 0.34548335974643424,
+      "eval_iou_concrete": 0.8907275950538522,
+      "eval_iou_concrete tiles": 0.8649992300754527,
+      "eval_iou_dark bricks": 0.8401998146050057,
+      "eval_iou_grass": 0.834079532727656,
+      "eval_iou_red bricks": 0.8967347595303201,
+      "eval_iou_road": 0.9594234187140617,
+      "eval_iou_rough concrete": 0.6181952750310853,
+      "eval_iou_rough red sidewalk": 0.6286181892030726,
+      "eval_iou_speedway bricks": 0.9787361355364534,
+      "eval_iou_steel": 0.8614254344982246,
+      "eval_iou_tiles": 0.2409396375855034,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08029722422361374,
+      "eval_mean_accuracy": 0.8032482374379323,
+      "eval_mean_iou": 0.7648033226587153,
+      "eval_overall_accuracy": 0.9794560448598053,
+      "eval_runtime": 5.3602,
+      "eval_samples_per_second": 22.014,
+      "eval_steps_per_second": 1.492,
+      "step": 1980
+    },
+    {
+      "epoch": 116.53,
+      "learning_rate": 2.196594427244582e-05,
+      "loss": 0.0268,
+      "step": 1981
+    },
+    {
+      "epoch": 116.59,
+      "learning_rate": 2.195046439628483e-05,
+      "loss": 0.0254,
+      "step": 1982
+    },
+    {
+      "epoch": 116.65,
+      "learning_rate": 2.1934984520123843e-05,
+      "loss": 0.0236,
+      "step": 1983
+    },
+    {
+      "epoch": 116.71,
+      "learning_rate": 2.191950464396285e-05,
+      "loss": 0.0226,
+      "step": 1984
+    },
+    {
+      "epoch": 116.76,
+      "learning_rate": 2.190402476780186e-05,
+      "loss": 0.0233,
+      "step": 1985
+    },
+    {
+      "epoch": 116.82,
+      "learning_rate": 2.188854489164087e-05,
+      "loss": 0.0264,
+      "step": 1986
+    },
+    {
+      "epoch": 116.88,
+      "learning_rate": 2.1873065015479876e-05,
+      "loss": 0.0274,
+      "step": 1987
+    },
+    {
+      "epoch": 116.94,
+      "learning_rate": 2.1857585139318886e-05,
+      "loss": 0.0263,
+      "step": 1988
+    },
+    {
+      "epoch": 117.0,
+      "learning_rate": 2.1842105263157896e-05,
+      "loss": 0.0245,
+      "step": 1989
+    },
+    {
+      "epoch": 117.06,
+      "learning_rate": 2.1826625386996906e-05,
+      "loss": 0.0261,
+      "step": 1990
+    },
+    {
+      "epoch": 117.12,
+      "learning_rate": 2.1811145510835916e-05,
+      "loss": 0.0266,
+      "step": 1991
+    },
+    {
+      "epoch": 117.18,
+      "learning_rate": 2.1795665634674926e-05,
+      "loss": 0.0261,
+      "step": 1992
+    },
+    {
+      "epoch": 117.24,
+      "learning_rate": 2.1780185758513933e-05,
+      "loss": 0.0254,
+      "step": 1993
+    },
+    {
+      "epoch": 117.29,
+      "learning_rate": 2.1764705882352943e-05,
+      "loss": 0.0285,
+      "step": 1994
+    },
+    {
+      "epoch": 117.35,
+      "learning_rate": 2.1749226006191953e-05,
+      "loss": 0.0233,
+      "step": 1995
+    },
+    {
+      "epoch": 117.41,
+      "learning_rate": 2.173374613003096e-05,
+      "loss": 0.0222,
+      "step": 1996
+    },
+    {
+      "epoch": 117.47,
+      "learning_rate": 2.171826625386997e-05,
+      "loss": 0.0233,
+      "step": 1997
+    },
+    {
+      "epoch": 117.53,
+      "learning_rate": 2.170278637770898e-05,
+      "loss": 0.0227,
+      "step": 1998
+    },
+    {
+      "epoch": 117.59,
+      "learning_rate": 2.168730650154799e-05,
+      "loss": 0.0235,
+      "step": 1999
+    },
+    {
+      "epoch": 117.65,
+      "learning_rate": 2.1671826625387e-05,
+      "loss": 0.0252,
+      "step": 2000
+    },
+    {
+      "epoch": 117.65,
+      "eval_accuracy_NAT": 0.993984256364647,
+      "eval_accuracy_REST": 0.34796747967479674,
+      "eval_accuracy_concrete": 0.9533017432874338,
+      "eval_accuracy_concrete tiles": 0.9367193023803912,
+      "eval_accuracy_dark bricks": 0.8492238194590663,
+      "eval_accuracy_grass": 0.8846434803014304,
+      "eval_accuracy_red bricks": 0.9283282073813708,
+      "eval_accuracy_road": 0.9815647811218571,
+      "eval_accuracy_rough concrete": 0.6247623928160697,
+      "eval_accuracy_rough red sidewalk": 0.7846727678806428,
+      "eval_accuracy_speedway bricks": 0.989387184491042,
+      "eval_accuracy_steel": 0.9175333729827525,
+      "eval_accuracy_tiles": 0.24356993547214395,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.982917496371618,
+      "eval_iou_REST": 0.33860759493670883,
+      "eval_iou_concrete": 0.8907909628530705,
+      "eval_iou_concrete tiles": 0.8639466133379706,
+      "eval_iou_dark bricks": 0.8326131527955672,
+      "eval_iou_grass": 0.8315287094404332,
+      "eval_iou_red bricks": 0.8973773624973455,
+      "eval_iou_road": 0.9588900565872545,
+      "eval_iou_rough concrete": 0.6189242157707702,
+      "eval_iou_rough red sidewalk": 0.6293122165396121,
+      "eval_iou_speedway bricks": 0.97878549151262,
+      "eval_iou_steel": 0.8600409908063478,
+      "eval_iou_tiles": 0.23907936602337268,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08008939772844315,
+      "eval_mean_accuracy": 0.8027429787395111,
+      "eval_mean_iou": 0.7632934022671302,
+      "eval_overall_accuracy": 0.9793882854914261,
+      "eval_runtime": 9.6118,
+      "eval_samples_per_second": 12.277,
+      "eval_steps_per_second": 0.832,
+      "step": 2000
+    },
+    {
+      "epoch": 117.71,
+      "learning_rate": 2.165634674922601e-05,
+      "loss": 0.0255,
+      "step": 2001
+    },
+    {
+      "epoch": 117.76,
+      "learning_rate": 2.1640866873065016e-05,
+      "loss": 0.0238,
+      "step": 2002
+    },
+    {
+      "epoch": 117.82,
+      "learning_rate": 2.1625386996904026e-05,
+      "loss": 0.0235,
+      "step": 2003
+    },
+    {
+      "epoch": 117.88,
+      "learning_rate": 2.1609907120743036e-05,
+      "loss": 0.0239,
+      "step": 2004
+    },
+    {
+      "epoch": 117.94,
+      "learning_rate": 2.1594427244582043e-05,
+      "loss": 0.0251,
+      "step": 2005
+    },
+    {
+      "epoch": 118.0,
+      "learning_rate": 2.1578947368421053e-05,
+      "loss": 0.0245,
+      "step": 2006
+    },
+    {
+      "epoch": 118.06,
+      "learning_rate": 2.1563467492260063e-05,
+      "loss": 0.0251,
+      "step": 2007
+    },
+    {
+      "epoch": 118.12,
+      "learning_rate": 2.1547987616099073e-05,
+      "loss": 0.0263,
+      "step": 2008
+    },
+    {
+      "epoch": 118.18,
+      "learning_rate": 2.1532507739938083e-05,
+      "loss": 0.0208,
+      "step": 2009
+    },
+    {
+      "epoch": 118.24,
+      "learning_rate": 2.151702786377709e-05,
+      "loss": 0.0261,
+      "step": 2010
+    },
+    {
+      "epoch": 118.29,
+      "learning_rate": 2.15015479876161e-05,
+      "loss": 0.03,
+      "step": 2011
+    },
+    {
+      "epoch": 118.35,
+      "learning_rate": 2.148606811145511e-05,
+      "loss": 0.0246,
+      "step": 2012
+    },
+    {
+      "epoch": 118.41,
+      "learning_rate": 2.1470588235294116e-05,
+      "loss": 0.0229,
+      "step": 2013
+    },
+    {
+      "epoch": 118.47,
+      "learning_rate": 2.145510835913313e-05,
+      "loss": 0.0312,
+      "step": 2014
+    },
+    {
+      "epoch": 118.53,
+      "learning_rate": 2.143962848297214e-05,
+      "loss": 0.0237,
+      "step": 2015
+    },
+    {
+      "epoch": 118.59,
+      "learning_rate": 2.1424148606811146e-05,
+      "loss": 0.023,
+      "step": 2016
+    },
+    {
+      "epoch": 118.65,
+      "learning_rate": 2.1408668730650156e-05,
+      "loss": 0.0245,
+      "step": 2017
+    },
+    {
+      "epoch": 118.71,
+      "learning_rate": 2.1393188854489166e-05,
+      "loss": 0.0218,
+      "step": 2018
+    },
+    {
+      "epoch": 118.76,
+      "learning_rate": 2.1377708978328172e-05,
+      "loss": 0.0252,
+      "step": 2019
+    },
+    {
+      "epoch": 118.82,
+      "learning_rate": 2.1362229102167182e-05,
+      "loss": 0.0266,
+      "step": 2020
+    },
+    {
+      "epoch": 118.82,
+      "eval_accuracy_NAT": 0.9923951533544247,
+      "eval_accuracy_REST": 0.44390243902439025,
+      "eval_accuracy_concrete": 0.9605402157744352,
+      "eval_accuracy_concrete tiles": 0.9216120669337733,
+      "eval_accuracy_dark bricks": 0.8812520337652892,
+      "eval_accuracy_grass": 0.9002742168072762,
+      "eval_accuracy_red bricks": 0.9195957820738138,
+      "eval_accuracy_road": 0.9825571918783712,
+      "eval_accuracy_rough concrete": 0.6329497851852294,
+      "eval_accuracy_rough red sidewalk": 0.7890869986990428,
+      "eval_accuracy_speedway bricks": 0.9874523005086804,
+      "eval_accuracy_steel": 0.9215366315652691,
+      "eval_accuracy_tiles": 0.16859038444060712,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833507243515335,
+      "eval_iou_REST": 0.4326465927099842,
+      "eval_iou_concrete": 0.8888080978979331,
+      "eval_iou_concrete tiles": 0.8657073278724817,
+      "eval_iou_dark bricks": 0.8562136477545877,
+      "eval_iou_grass": 0.8397477142562099,
+      "eval_iou_red bricks": 0.8956884561891516,
+      "eval_iou_road": 0.9589388761625797,
+      "eval_iou_rough concrete": 0.62268663716036,
+      "eval_iou_rough red sidewalk": 0.6283466170050959,
+      "eval_iou_speedway bricks": 0.9782129487799988,
+      "eval_iou_steel": 0.8601306610253695,
+      "eval_iou_tiles": 0.16785304940580323,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07900871336460114,
+      "eval_mean_accuracy": 0.8078265538469693,
+      "eval_mean_iou": 0.76756395004393,
+      "eval_overall_accuracy": 0.9795042781506554,
+      "eval_runtime": 5.6096,
+      "eval_samples_per_second": 21.035,
+      "eval_steps_per_second": 1.426,
+      "step": 2020
+    },
+    {
+      "epoch": 118.88,
+      "learning_rate": 2.1346749226006192e-05,
+      "loss": 0.0275,
+      "step": 2021
+    },
+    {
+      "epoch": 118.94,
+      "learning_rate": 2.1331269349845202e-05,
+      "loss": 0.0276,
+      "step": 2022
+    },
+    {
+      "epoch": 119.0,
+      "learning_rate": 2.1315789473684212e-05,
+      "loss": 0.0238,
+      "step": 2023
+    },
+    {
+      "epoch": 119.06,
+      "learning_rate": 2.1300309597523222e-05,
+      "loss": 0.0262,
+      "step": 2024
+    },
+    {
+      "epoch": 119.12,
+      "learning_rate": 2.128482972136223e-05,
+      "loss": 0.0242,
+      "step": 2025
+    },
+    {
+      "epoch": 119.18,
+      "learning_rate": 2.126934984520124e-05,
+      "loss": 0.0258,
+      "step": 2026
+    },
+    {
+      "epoch": 119.24,
+      "learning_rate": 2.125386996904025e-05,
+      "loss": 0.0272,
+      "step": 2027
+    },
+    {
+      "epoch": 119.29,
+      "learning_rate": 2.1238390092879256e-05,
+      "loss": 0.0236,
+      "step": 2028
+    },
+    {
+      "epoch": 119.35,
+      "learning_rate": 2.1222910216718266e-05,
+      "loss": 0.0238,
+      "step": 2029
+    },
+    {
+      "epoch": 119.41,
+      "learning_rate": 2.1207430340557276e-05,
+      "loss": 0.025,
+      "step": 2030
+    },
+    {
+      "epoch": 119.47,
+      "learning_rate": 2.1191950464396286e-05,
+      "loss": 0.0251,
+      "step": 2031
+    },
+    {
+      "epoch": 119.53,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0243,
+      "step": 2032
+    },
+    {
+      "epoch": 119.59,
+      "learning_rate": 2.1160990712074306e-05,
+      "loss": 0.0242,
+      "step": 2033
+    },
+    {
+      "epoch": 119.65,
+      "learning_rate": 2.1145510835913312e-05,
+      "loss": 0.0209,
+      "step": 2034
+    },
+    {
+      "epoch": 119.71,
+      "learning_rate": 2.1130030959752322e-05,
+      "loss": 0.0234,
+      "step": 2035
+    },
+    {
+      "epoch": 119.76,
+      "learning_rate": 2.1114551083591332e-05,
+      "loss": 0.0252,
+      "step": 2036
+    },
+    {
+      "epoch": 119.82,
+      "learning_rate": 2.109907120743034e-05,
+      "loss": 0.0235,
+      "step": 2037
+    },
+    {
+      "epoch": 119.88,
+      "learning_rate": 2.1083591331269352e-05,
+      "loss": 0.0261,
+      "step": 2038
+    },
+    {
+      "epoch": 119.94,
+      "learning_rate": 2.1068111455108362e-05,
+      "loss": 0.0228,
+      "step": 2039
+    },
+    {
+      "epoch": 120.0,
+      "learning_rate": 2.105263157894737e-05,
+      "loss": 0.0257,
+      "step": 2040
+    },
+    {
+      "epoch": 120.0,
+      "eval_accuracy_NAT": 0.9935486493513958,
+      "eval_accuracy_REST": 0.33170731707317075,
+      "eval_accuracy_concrete": 0.9584044861024692,
+      "eval_accuracy_concrete tiles": 0.9307094037237803,
+      "eval_accuracy_dark bricks": 0.8575646498095438,
+      "eval_accuracy_grass": 0.8875219051728379,
+      "eval_accuracy_red bricks": 0.9197605448154658,
+      "eval_accuracy_road": 0.9791396007163314,
+      "eval_accuracy_rough concrete": 0.6242797299471454,
+      "eval_accuracy_rough red sidewalk": 0.802983929793426,
+      "eval_accuracy_speedway bricks": 0.9882309837799452,
+      "eval_accuracy_steel": 0.9261871086077557,
+      "eval_accuracy_tiles": 0.17798176254960768,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9830856783897617,
+      "eval_iou_REST": 0.3258785942492013,
+      "eval_iou_concrete": 0.8903455746778113,
+      "eval_iou_concrete tiles": 0.8654963070111995,
+      "eval_iou_dark bricks": 0.8413237185806972,
+      "eval_iou_grass": 0.8343886717325282,
+      "eval_iou_red bricks": 0.8958489354873221,
+      "eval_iou_road": 0.9606922855890415,
+      "eval_iou_rough concrete": 0.6168547860032149,
+      "eval_iou_rough red sidewalk": 0.6192586062911757,
+      "eval_iou_speedway bricks": 0.9787606999714017,
+      "eval_iou_steel": 0.8633717612144732,
+      "eval_iou_tiles": 0.17715526339595333,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08132188767194748,
+      "eval_mean_accuracy": 0.7983092362648366,
+      "eval_mean_iou": 0.7578816063533679,
+      "eval_overall_accuracy": 0.97945300603317,
+      "eval_runtime": 7.4094,
+      "eval_samples_per_second": 15.926,
+      "eval_steps_per_second": 1.08,
+      "step": 2040
+    },
+    {
+      "epoch": 120.06,
+      "learning_rate": 2.103715170278638e-05,
+      "loss": 0.0263,
+      "step": 2041
+    },
+    {
+      "epoch": 120.12,
+      "learning_rate": 2.102167182662539e-05,
+      "loss": 0.0253,
+      "step": 2042
+    },
+    {
+      "epoch": 120.18,
+      "learning_rate": 2.1006191950464396e-05,
+      "loss": 0.0279,
+      "step": 2043
+    },
+    {
+      "epoch": 120.24,
+      "learning_rate": 2.0990712074303406e-05,
+      "loss": 0.025,
+      "step": 2044
+    },
+    {
+      "epoch": 120.29,
+      "learning_rate": 2.0975232198142416e-05,
+      "loss": 0.0245,
+      "step": 2045
+    },
+    {
+      "epoch": 120.35,
+      "learning_rate": 2.0959752321981426e-05,
+      "loss": 0.0262,
+      "step": 2046
+    },
+    {
+      "epoch": 120.41,
+      "learning_rate": 2.0944272445820436e-05,
+      "loss": 0.0283,
+      "step": 2047
+    },
+    {
+      "epoch": 120.47,
+      "learning_rate": 2.0928792569659446e-05,
+      "loss": 0.0284,
+      "step": 2048
+    },
+    {
+      "epoch": 120.53,
+      "learning_rate": 2.0913312693498452e-05,
+      "loss": 0.027,
+      "step": 2049
+    },
+    {
+      "epoch": 120.59,
+      "learning_rate": 2.0897832817337462e-05,
+      "loss": 0.0234,
+      "step": 2050
+    },
+    {
+      "epoch": 120.65,
+      "learning_rate": 2.0882352941176472e-05,
+      "loss": 0.0258,
+      "step": 2051
+    },
+    {
+      "epoch": 120.71,
+      "learning_rate": 2.086687306501548e-05,
+      "loss": 0.024,
+      "step": 2052
+    },
+    {
+      "epoch": 120.76,
+      "learning_rate": 2.085139318885449e-05,
+      "loss": 0.0236,
+      "step": 2053
+    },
+    {
+      "epoch": 120.82,
+      "learning_rate": 2.08359133126935e-05,
+      "loss": 0.0229,
+      "step": 2054
+    },
+    {
+      "epoch": 120.88,
+      "learning_rate": 2.082043343653251e-05,
+      "loss": 0.0207,
+      "step": 2055
+    },
+    {
+      "epoch": 120.94,
+      "learning_rate": 2.080495356037152e-05,
+      "loss": 0.023,
+      "step": 2056
+    },
+    {
+      "epoch": 121.0,
+      "learning_rate": 2.078947368421053e-05,
+      "loss": 0.0192,
+      "step": 2057
+    },
+    {
+      "epoch": 121.06,
+      "learning_rate": 2.0773993808049535e-05,
+      "loss": 0.0223,
+      "step": 2058
+    },
+    {
+      "epoch": 121.12,
+      "learning_rate": 2.0758513931888545e-05,
+      "loss": 0.0261,
+      "step": 2059
+    },
+    {
+      "epoch": 121.18,
+      "learning_rate": 2.0743034055727555e-05,
+      "loss": 0.0246,
+      "step": 2060
+    },
+    {
+      "epoch": 121.18,
+      "eval_accuracy_NAT": 0.9936819192198063,
+      "eval_accuracy_REST": 0.25040650406504067,
+      "eval_accuracy_concrete": 0.9544124144446874,
+      "eval_accuracy_concrete tiles": 0.9302851755833137,
+      "eval_accuracy_dark bricks": 0.8724565970560649,
+      "eval_accuracy_grass": 0.8938010726437983,
+      "eval_accuracy_red bricks": 0.9228361159929701,
+      "eval_accuracy_road": 0.9799839782043724,
+      "eval_accuracy_rough concrete": 0.6264725686602828,
+      "eval_accuracy_rough red sidewalk": 0.7641356905977635,
+      "eval_accuracy_speedway bricks": 0.9909829368286341,
+      "eval_accuracy_steel": 0.9044990429163314,
+      "eval_accuracy_tiles": 0.2586264352146384,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983268789929746,
+      "eval_iou_REST": 0.24758842443729903,
+      "eval_iou_concrete": 0.892161447717559,
+      "eval_iou_concrete tiles": 0.8670972277140723,
+      "eval_iou_dark bricks": 0.8509161680559315,
+      "eval_iou_grass": 0.8386156273277147,
+      "eval_iou_red bricks": 0.9003375663076676,
+      "eval_iou_road": 0.9602971780963383,
+      "eval_iou_rough concrete": 0.6180390220388104,
+      "eval_iou_rough red sidewalk": 0.621108672929872,
+      "eval_iou_speedway bricks": 0.9781626275483235,
+      "eval_iou_steel": 0.8594491273892912,
+      "eval_iou_tiles": 0.2519775678866588,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07729286700487137,
+      "eval_mean_accuracy": 0.7955831116482849,
+      "eval_mean_iou": 0.7591553421060989,
+      "eval_overall_accuracy": 0.9797306060791016,
+      "eval_runtime": 7.1008,
+      "eval_samples_per_second": 16.618,
+      "eval_steps_per_second": 1.127,
+      "step": 2060
+    },
+    {
+      "epoch": 121.24,
+      "learning_rate": 2.0727554179566562e-05,
+      "loss": 0.02,
+      "step": 2061
+    },
+    {
+      "epoch": 121.29,
+      "learning_rate": 2.0712074303405572e-05,
+      "loss": 0.0227,
+      "step": 2062
+    },
+    {
+      "epoch": 121.35,
+      "learning_rate": 2.0696594427244585e-05,
+      "loss": 0.0259,
+      "step": 2063
+    },
+    {
+      "epoch": 121.41,
+      "learning_rate": 2.0681114551083592e-05,
+      "loss": 0.0272,
+      "step": 2064
+    },
+    {
+      "epoch": 121.47,
+      "learning_rate": 2.0665634674922602e-05,
+      "loss": 0.0233,
+      "step": 2065
+    },
+    {
+      "epoch": 121.53,
+      "learning_rate": 2.0650154798761612e-05,
+      "loss": 0.0294,
+      "step": 2066
+    },
+    {
+      "epoch": 121.59,
+      "learning_rate": 2.063467492260062e-05,
+      "loss": 0.0218,
+      "step": 2067
+    },
+    {
+      "epoch": 121.65,
+      "learning_rate": 2.061919504643963e-05,
+      "loss": 0.0242,
+      "step": 2068
+    },
+    {
+      "epoch": 121.71,
+      "learning_rate": 2.060371517027864e-05,
+      "loss": 0.0201,
+      "step": 2069
+    },
+    {
+      "epoch": 121.76,
+      "learning_rate": 2.058823529411765e-05,
+      "loss": 0.0205,
+      "step": 2070
+    },
+    {
+      "epoch": 121.82,
+      "learning_rate": 2.057275541795666e-05,
+      "loss": 0.0249,
+      "step": 2071
+    },
+    {
+      "epoch": 121.88,
+      "learning_rate": 2.055727554179567e-05,
+      "loss": 0.022,
+      "step": 2072
+    },
+    {
+      "epoch": 121.94,
+      "learning_rate": 2.0541795665634675e-05,
+      "loss": 0.0224,
+      "step": 2073
+    },
+    {
+      "epoch": 122.0,
+      "learning_rate": 2.0526315789473685e-05,
+      "loss": 0.0286,
+      "step": 2074
+    },
+    {
+      "epoch": 122.06,
+      "learning_rate": 2.0510835913312695e-05,
+      "loss": 0.0211,
+      "step": 2075
+    },
+    {
+      "epoch": 122.12,
+      "learning_rate": 2.0495356037151702e-05,
+      "loss": 0.022,
+      "step": 2076
+    },
+    {
+      "epoch": 122.18,
+      "learning_rate": 2.0479876160990712e-05,
+      "loss": 0.0239,
+      "step": 2077
+    },
+    {
+      "epoch": 122.24,
+      "learning_rate": 2.0464396284829722e-05,
+      "loss": 0.0206,
+      "step": 2078
+    },
+    {
+      "epoch": 122.29,
+      "learning_rate": 2.0448916408668732e-05,
+      "loss": 0.0216,
+      "step": 2079
+    },
+    {
+      "epoch": 122.35,
+      "learning_rate": 2.0433436532507742e-05,
+      "loss": 0.0273,
+      "step": 2080
+    },
+    {
+      "epoch": 122.35,
+      "eval_accuracy_NAT": 0.9939416121632261,
+      "eval_accuracy_REST": 0.19349593495934958,
+      "eval_accuracy_concrete": 0.9528755910712129,
+      "eval_accuracy_concrete tiles": 0.9174404902191845,
+      "eval_accuracy_dark bricks": 0.8531334341442873,
+      "eval_accuracy_grass": 0.8845242556026146,
+      "eval_accuracy_red bricks": 0.9047671353251318,
+      "eval_accuracy_road": 0.9813686504015335,
+      "eval_accuracy_rough concrete": 0.6249292392398954,
+      "eval_accuracy_rough red sidewalk": 0.7741372697944788,
+      "eval_accuracy_speedway bricks": 0.9913405759103089,
+      "eval_accuracy_steel": 0.9007656669348834,
+      "eval_accuracy_tiles": 0.20924596322215153,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9831844185417113,
+      "eval_iou_REST": 0.1925566343042071,
+      "eval_iou_concrete": 0.8896219726248533,
+      "eval_iou_concrete tiles": 0.8640459912989434,
+      "eval_iou_dark bricks": 0.8382514740316529,
+      "eval_iou_grass": 0.8355657335725286,
+      "eval_iou_red bricks": 0.8907272235739389,
+      "eval_iou_road": 0.9592453236256351,
+      "eval_iou_rough concrete": 0.6192466889861182,
+      "eval_iou_rough red sidewalk": 0.6204548057161111,
+      "eval_iou_speedway bricks": 0.9774320612435962,
+      "eval_iou_steel": 0.8581389923152372,
+      "eval_iou_tiles": 0.20767311103761388,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08115528523921967,
+      "eval_mean_accuracy": 0.7832281399221738,
+      "eval_mean_iou": 0.7489341869901651,
+      "eval_overall_accuracy": 0.9793643304857157,
+      "eval_runtime": 10.08,
+      "eval_samples_per_second": 11.706,
+      "eval_steps_per_second": 0.794,
+      "step": 2080
+    },
+    {
+      "epoch": 122.41,
+      "learning_rate": 2.0417956656346752e-05,
+      "loss": 0.0216,
+      "step": 2081
+    },
+    {
+      "epoch": 122.47,
+      "learning_rate": 2.040247678018576e-05,
+      "loss": 0.0254,
+      "step": 2082
+    },
+    {
+      "epoch": 122.53,
+      "learning_rate": 2.038699690402477e-05,
+      "loss": 0.031,
+      "step": 2083
+    },
+    {
+      "epoch": 122.59,
+      "learning_rate": 2.037151702786378e-05,
+      "loss": 0.0256,
+      "step": 2084
+    },
+    {
+      "epoch": 122.65,
+      "learning_rate": 2.0356037151702785e-05,
+      "loss": 0.0219,
+      "step": 2085
+    },
+    {
+      "epoch": 122.71,
+      "learning_rate": 2.0340557275541795e-05,
+      "loss": 0.0228,
+      "step": 2086
+    },
+    {
+      "epoch": 122.76,
+      "learning_rate": 2.032507739938081e-05,
+      "loss": 0.0245,
+      "step": 2087
+    },
+    {
+      "epoch": 122.82,
+      "learning_rate": 2.0309597523219815e-05,
+      "loss": 0.0256,
+      "step": 2088
+    },
+    {
+      "epoch": 122.88,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.026,
+      "step": 2089
+    },
+    {
+      "epoch": 122.94,
+      "learning_rate": 2.0278637770897835e-05,
+      "loss": 0.026,
+      "step": 2090
+    },
+    {
+      "epoch": 123.0,
+      "learning_rate": 2.0263157894736842e-05,
+      "loss": 0.0244,
+      "step": 2091
+    },
+    {
+      "epoch": 123.06,
+      "learning_rate": 2.0247678018575852e-05,
+      "loss": 0.0241,
+      "step": 2092
+    },
+    {
+      "epoch": 123.12,
+      "learning_rate": 2.0232198142414862e-05,
+      "loss": 0.0229,
+      "step": 2093
+    },
+    {
+      "epoch": 123.18,
+      "learning_rate": 2.0216718266253872e-05,
+      "loss": 0.0241,
+      "step": 2094
+    },
+    {
+      "epoch": 123.24,
+      "learning_rate": 2.0201238390092882e-05,
+      "loss": 0.0258,
+      "step": 2095
+    },
+    {
+      "epoch": 123.29,
+      "learning_rate": 2.0185758513931892e-05,
+      "loss": 0.0245,
+      "step": 2096
+    },
+    {
+      "epoch": 123.35,
+      "learning_rate": 2.01702786377709e-05,
+      "loss": 0.0229,
+      "step": 2097
+    },
+    {
+      "epoch": 123.41,
+      "learning_rate": 2.015479876160991e-05,
+      "loss": 0.0261,
+      "step": 2098
+    },
+    {
+      "epoch": 123.47,
+      "learning_rate": 2.013931888544892e-05,
+      "loss": 0.0274,
+      "step": 2099
+    },
+    {
+      "epoch": 123.53,
+      "learning_rate": 2.0123839009287925e-05,
+      "loss": 0.0209,
+      "step": 2100
+    },
+    {
+      "epoch": 123.53,
+      "eval_accuracy_NAT": 0.9933261016099723,
+      "eval_accuracy_REST": 0.24552845528455283,
+      "eval_accuracy_concrete": 0.9573205688960097,
+      "eval_accuracy_concrete tiles": 0.9207164741927881,
+      "eval_accuracy_dark bricks": 0.8545259652010796,
+      "eval_accuracy_grass": 0.8999648003270163,
+      "eval_accuracy_red bricks": 0.9154217926186292,
+      "eval_accuracy_road": 0.9800887237661802,
+      "eval_accuracy_rough concrete": 0.6232488573999369,
+      "eval_accuracy_rough red sidewalk": 0.7778220621301107,
+      "eval_accuracy_speedway bricks": 0.9897266588661248,
+      "eval_accuracy_steel": 0.9137100361342816,
+      "eval_accuracy_tiles": 0.20155109212638977,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833629720399898,
+      "eval_iou_REST": 0.24394184168012925,
+      "eval_iou_concrete": 0.8904290452507829,
+      "eval_iou_concrete tiles": 0.8678826116899563,
+      "eval_iou_dark bricks": 0.8394302637145677,
+      "eval_iou_grass": 0.8418687895991885,
+      "eval_iou_red bricks": 0.8980603448275862,
+      "eval_iou_road": 0.96001522252473,
+      "eval_iou_rough concrete": 0.6177457268743282,
+      "eval_iou_rough red sidewalk": 0.6289081025865528,
+      "eval_iou_speedway bricks": 0.9783775973064023,
+      "eval_iou_steel": 0.8626707561637371,
+      "eval_iou_tiles": 0.20016848692721967,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08050906658172607,
+      "eval_mean_accuracy": 0.7902270452733132,
+      "eval_mean_iou": 0.7548355200911671,
+      "eval_overall_accuracy": 0.9797040325100139,
+      "eval_runtime": 5.8225,
+      "eval_samples_per_second": 20.266,
+      "eval_steps_per_second": 1.374,
+      "step": 2100
+    },
+    {
+      "epoch": 123.59,
+      "learning_rate": 2.0108359133126935e-05,
+      "loss": 0.0256,
+      "step": 2101
+    },
+    {
+      "epoch": 123.65,
+      "learning_rate": 2.0092879256965945e-05,
+      "loss": 0.0215,
+      "step": 2102
+    },
+    {
+      "epoch": 123.71,
+      "learning_rate": 2.0077399380804955e-05,
+      "loss": 0.0231,
+      "step": 2103
+    },
+    {
+      "epoch": 123.76,
+      "learning_rate": 2.0061919504643965e-05,
+      "loss": 0.0281,
+      "step": 2104
+    },
+    {
+      "epoch": 123.82,
+      "learning_rate": 2.0046439628482975e-05,
+      "loss": 0.031,
+      "step": 2105
+    },
+    {
+      "epoch": 123.88,
+      "learning_rate": 2.003095975232198e-05,
+      "loss": 0.0242,
+      "step": 2106
+    },
+    {
+      "epoch": 123.94,
+      "learning_rate": 2.001547987616099e-05,
+      "loss": 0.0242,
+      "step": 2107
+    },
+    {
+      "epoch": 124.0,
+      "learning_rate": 2e-05,
+      "loss": 0.021,
+      "step": 2108
+    },
+    {
+      "epoch": 124.06,
+      "learning_rate": 1.9984520123839008e-05,
+      "loss": 0.0249,
+      "step": 2109
+    },
+    {
+      "epoch": 124.12,
+      "learning_rate": 1.9969040247678018e-05,
+      "loss": 0.0254,
+      "step": 2110
+    },
+    {
+      "epoch": 124.18,
+      "learning_rate": 1.9953560371517028e-05,
+      "loss": 0.0227,
+      "step": 2111
+    },
+    {
+      "epoch": 124.24,
+      "learning_rate": 1.9938080495356038e-05,
+      "loss": 0.0242,
+      "step": 2112
+    },
+    {
+      "epoch": 124.29,
+      "learning_rate": 1.9922600619195048e-05,
+      "loss": 0.0238,
+      "step": 2113
+    },
+    {
+      "epoch": 124.35,
+      "learning_rate": 1.9907120743034058e-05,
+      "loss": 0.024,
+      "step": 2114
+    },
+    {
+      "epoch": 124.41,
+      "learning_rate": 1.9891640866873065e-05,
+      "loss": 0.0233,
+      "step": 2115
+    },
+    {
+      "epoch": 124.47,
+      "learning_rate": 1.9876160990712075e-05,
+      "loss": 0.0198,
+      "step": 2116
+    },
+    {
+      "epoch": 124.53,
+      "learning_rate": 1.9860681114551085e-05,
+      "loss": 0.0235,
+      "step": 2117
+    },
+    {
+      "epoch": 124.59,
+      "learning_rate": 1.984520123839009e-05,
+      "loss": 0.021,
+      "step": 2118
+    },
+    {
+      "epoch": 124.65,
+      "learning_rate": 1.9829721362229105e-05,
+      "loss": 0.0252,
+      "step": 2119
+    },
+    {
+      "epoch": 124.71,
+      "learning_rate": 1.9814241486068115e-05,
+      "loss": 0.0233,
+      "step": 2120
+    },
+    {
+      "epoch": 124.71,
+      "eval_accuracy_NAT": 0.9929944364681997,
+      "eval_accuracy_REST": 0.24227642276422764,
+      "eval_accuracy_concrete": 0.9551213941407319,
+      "eval_accuracy_concrete tiles": 0.9354466179589913,
+      "eval_accuracy_dark bricks": 0.85482744099688,
+      "eval_accuracy_grass": 0.902180865760559,
+      "eval_accuracy_red bricks": 0.9183326010544816,
+      "eval_accuracy_road": 0.9806658931883854,
+      "eval_accuracy_rough concrete": 0.6223967488782557,
+      "eval_accuracy_rough red sidewalk": 0.7888388392152145,
+      "eval_accuracy_speedway bricks": 0.9906480893127777,
+      "eval_accuracy_steel": 0.910561405788482,
+      "eval_accuracy_tiles": 0.2289981520191463,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834201441956409,
+      "eval_iou_REST": 0.2407108239095315,
+      "eval_iou_concrete": 0.8902942087024613,
+      "eval_iou_concrete tiles": 0.8647841906878445,
+      "eval_iou_dark bricks": 0.8387801098746303,
+      "eval_iou_grass": 0.8394228060290182,
+      "eval_iou_red bricks": 0.896183942544753,
+      "eval_iou_road": 0.9599921108994098,
+      "eval_iou_rough concrete": 0.6160642665030907,
+      "eval_iou_rough red sidewalk": 0.6273713547522787,
+      "eval_iou_speedway bricks": 0.9785320831328903,
+      "eval_iou_steel": 0.862285779193427,
+      "eval_iou_tiles": 0.22709929397626558,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08027881383895874,
+      "eval_mean_accuracy": 0.7940991467343333,
+      "eval_mean_iou": 0.7557647011077877,
+      "eval_overall_accuracy": 0.9796587733899133,
+      "eval_runtime": 5.757,
+      "eval_samples_per_second": 20.497,
+      "eval_steps_per_second": 1.39,
+      "step": 2120
+    },
+    {
+      "epoch": 124.76,
+      "learning_rate": 1.979876160990712e-05,
+      "loss": 0.0216,
+      "step": 2121
+    },
+    {
+      "epoch": 124.82,
+      "learning_rate": 1.978328173374613e-05,
+      "loss": 0.0196,
+      "step": 2122
+    },
+    {
+      "epoch": 124.88,
+      "learning_rate": 1.976780185758514e-05,
+      "loss": 0.0244,
+      "step": 2123
+    },
+    {
+      "epoch": 124.94,
+      "learning_rate": 1.9752321981424148e-05,
+      "loss": 0.0275,
+      "step": 2124
+    },
+    {
+      "epoch": 125.0,
+      "learning_rate": 1.9736842105263158e-05,
+      "loss": 0.03,
+      "step": 2125
+    },
+    {
+      "epoch": 125.06,
+      "learning_rate": 1.9721362229102168e-05,
+      "loss": 0.0272,
+      "step": 2126
+    },
+    {
+      "epoch": 125.12,
+      "learning_rate": 1.9705882352941178e-05,
+      "loss": 0.0246,
+      "step": 2127
+    },
+    {
+      "epoch": 125.18,
+      "learning_rate": 1.9690402476780188e-05,
+      "loss": 0.0231,
+      "step": 2128
+    },
+    {
+      "epoch": 125.24,
+      "learning_rate": 1.9674922600619198e-05,
+      "loss": 0.0202,
+      "step": 2129
+    },
+    {
+      "epoch": 125.29,
+      "learning_rate": 1.9659442724458205e-05,
+      "loss": 0.0244,
+      "step": 2130
+    },
+    {
+      "epoch": 125.35,
+      "learning_rate": 1.9643962848297215e-05,
+      "loss": 0.0263,
+      "step": 2131
+    },
+    {
+      "epoch": 125.41,
+      "learning_rate": 1.9628482972136225e-05,
+      "loss": 0.0245,
+      "step": 2132
+    },
+    {
+      "epoch": 125.47,
+      "learning_rate": 1.961300309597523e-05,
+      "loss": 0.0245,
+      "step": 2133
+    },
+    {
+      "epoch": 125.53,
+      "learning_rate": 1.959752321981424e-05,
+      "loss": 0.0213,
+      "step": 2134
+    },
+    {
+      "epoch": 125.59,
+      "learning_rate": 1.958204334365325e-05,
+      "loss": 0.023,
+      "step": 2135
+    },
+    {
+      "epoch": 125.65,
+      "learning_rate": 1.956656346749226e-05,
+      "loss": 0.0218,
+      "step": 2136
+    },
+    {
+      "epoch": 125.71,
+      "learning_rate": 1.955108359133127e-05,
+      "loss": 0.0252,
+      "step": 2137
+    },
+    {
+      "epoch": 125.76,
+      "learning_rate": 1.953560371517028e-05,
+      "loss": 0.0288,
+      "step": 2138
+    },
+    {
+      "epoch": 125.82,
+      "learning_rate": 1.9520123839009288e-05,
+      "loss": 0.0242,
+      "step": 2139
+    },
+    {
+      "epoch": 125.88,
+      "learning_rate": 1.9504643962848298e-05,
+      "loss": 0.0206,
+      "step": 2140
+    },
+    {
+      "epoch": 125.88,
+      "eval_accuracy_NAT": 0.9938475900554153,
+      "eval_accuracy_REST": 0.3089430894308943,
+      "eval_accuracy_concrete": 0.9518356267839152,
+      "eval_accuracy_concrete tiles": 0.9176761725194438,
+      "eval_accuracy_dark bricks": 0.8557414390444653,
+      "eval_accuracy_grass": 0.8885599170347492,
+      "eval_accuracy_red bricks": 0.9249231107205624,
+      "eval_accuracy_road": 0.9814188855179107,
+      "eval_accuracy_rough concrete": 0.6287130777802276,
+      "eval_accuracy_rough red sidewalk": 0.7948397867332436,
+      "eval_accuracy_speedway bricks": 0.9914303712523297,
+      "eval_accuracy_steel": 0.90480391029902,
+      "eval_accuracy_tiles": 0.21157866036535491,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9831938137866055,
+      "eval_iou_REST": 0.3035143769968051,
+      "eval_iou_concrete": 0.8902215351069249,
+      "eval_iou_concrete tiles": 0.8658053900204572,
+      "eval_iou_dark bricks": 0.8394798610459112,
+      "eval_iou_grass": 0.8395738897978524,
+      "eval_iou_red bricks": 0.9012629776303115,
+      "eval_iou_road": 0.9591095540562984,
+      "eval_iou_rough concrete": 0.6217295997737237,
+      "eval_iou_rough red sidewalk": 0.6243841637031698,
+      "eval_iou_speedway bricks": 0.9783259326673833,
+      "eval_iou_steel": 0.8598245580706138,
+      "eval_iou_tiles": 0.2103297696130101,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0803886204957962,
+      "eval_mean_accuracy": 0.7964855105798102,
+      "eval_mean_iou": 0.7597504170976206,
+      "eval_overall_accuracy": 0.9795742034912109,
+      "eval_runtime": 5.4573,
+      "eval_samples_per_second": 21.622,
+      "eval_steps_per_second": 1.466,
+      "step": 2140
+    },
+    {
+      "epoch": 125.94,
+      "learning_rate": 1.9489164086687308e-05,
+      "loss": 0.0242,
+      "step": 2141
+    },
+    {
+      "epoch": 126.0,
+      "learning_rate": 1.9473684210526315e-05,
+      "loss": 0.0238,
+      "step": 2142
+    },
+    {
+      "epoch": 126.06,
+      "learning_rate": 1.9458204334365328e-05,
+      "loss": 0.0227,
+      "step": 2143
+    },
+    {
+      "epoch": 126.12,
+      "learning_rate": 1.9442724458204338e-05,
+      "loss": 0.0232,
+      "step": 2144
+    },
+    {
+      "epoch": 126.18,
+      "learning_rate": 1.9427244582043345e-05,
+      "loss": 0.0214,
+      "step": 2145
+    },
+    {
+      "epoch": 126.24,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.0244,
+      "step": 2146
+    },
+    {
+      "epoch": 126.29,
+      "learning_rate": 1.9396284829721365e-05,
+      "loss": 0.0229,
+      "step": 2147
+    },
+    {
+      "epoch": 126.35,
+      "learning_rate": 1.938080495356037e-05,
+      "loss": 0.0266,
+      "step": 2148
+    },
+    {
+      "epoch": 126.41,
+      "learning_rate": 1.936532507739938e-05,
+      "loss": 0.0283,
+      "step": 2149
+    },
+    {
+      "epoch": 126.47,
+      "learning_rate": 1.934984520123839e-05,
+      "loss": 0.0244,
+      "step": 2150
+    },
+    {
+      "epoch": 126.53,
+      "learning_rate": 1.93343653250774e-05,
+      "loss": 0.024,
+      "step": 2151
+    },
+    {
+      "epoch": 126.59,
+      "learning_rate": 1.931888544891641e-05,
+      "loss": 0.0259,
+      "step": 2152
+    },
+    {
+      "epoch": 126.65,
+      "learning_rate": 1.9303405572755418e-05,
+      "loss": 0.0252,
+      "step": 2153
+    },
+    {
+      "epoch": 126.71,
+      "learning_rate": 1.9287925696594428e-05,
+      "loss": 0.0267,
+      "step": 2154
+    },
+    {
+      "epoch": 126.76,
+      "learning_rate": 1.9272445820433438e-05,
+      "loss": 0.0224,
+      "step": 2155
+    },
+    {
+      "epoch": 126.82,
+      "learning_rate": 1.9256965944272444e-05,
+      "loss": 0.0235,
+      "step": 2156
+    },
+    {
+      "epoch": 126.88,
+      "learning_rate": 1.9241486068111454e-05,
+      "loss": 0.0211,
+      "step": 2157
+    },
+    {
+      "epoch": 126.94,
+      "learning_rate": 1.9226006191950464e-05,
+      "loss": 0.0247,
+      "step": 2158
+    },
+    {
+      "epoch": 127.0,
+      "learning_rate": 1.9210526315789474e-05,
+      "loss": 0.0218,
+      "step": 2159
+    },
+    {
+      "epoch": 127.06,
+      "learning_rate": 1.9195046439628485e-05,
+      "loss": 0.0252,
+      "step": 2160
+    },
+    {
+      "epoch": 127.06,
+      "eval_accuracy_NAT": 0.99329391628981,
+      "eval_accuracy_REST": 0.3284552845528455,
+      "eval_accuracy_concrete": 0.9543508803578609,
+      "eval_accuracy_concrete tiles": 0.9381098279519208,
+      "eval_accuracy_dark bricks": 0.8706381716210784,
+      "eval_accuracy_grass": 0.8985454586744485,
+      "eval_accuracy_red bricks": 0.930305360281195,
+      "eval_accuracy_road": 0.981270317833306,
+      "eval_accuracy_rough concrete": 0.6271995423640947,
+      "eval_accuracy_rough red sidewalk": 0.7816497341685529,
+      "eval_accuracy_speedway bricks": 0.9902441816388404,
+      "eval_accuracy_steel": 0.917525876243834,
+      "eval_accuracy_tiles": 0.23814717198339846,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834420677070707,
+      "eval_iou_REST": 0.32114467408585057,
+      "eval_iou_concrete": 0.8908410266692401,
+      "eval_iou_concrete tiles": 0.8685709298012089,
+      "eval_iou_dark bricks": 0.8501782701949991,
+      "eval_iou_grass": 0.8414735942919223,
+      "eval_iou_red bricks": 0.9023065040217333,
+      "eval_iou_road": 0.9596715647972237,
+      "eval_iou_rough concrete": 0.620697382310101,
+      "eval_iou_rough red sidewalk": 0.6263626338529772,
+      "eval_iou_speedway bricks": 0.9787588654344221,
+      "eval_iou_steel": 0.8632895618541552,
+      "eval_iou_tiles": 0.23370098403543715,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0793551504611969,
+      "eval_mean_accuracy": 0.8038258249200911,
+      "eval_mean_iou": 0.7646490814658724,
+      "eval_overall_accuracy": 0.9798235812429654,
+      "eval_runtime": 5.2463,
+      "eval_samples_per_second": 22.492,
+      "eval_steps_per_second": 1.525,
+      "step": 2160
+    },
+    {
+      "epoch": 127.12,
+      "learning_rate": 1.9179566563467495e-05,
+      "loss": 0.0218,
+      "step": 2161
+    },
+    {
+      "epoch": 127.18,
+      "learning_rate": 1.91640866873065e-05,
+      "loss": 0.024,
+      "step": 2162
+    },
+    {
+      "epoch": 127.24,
+      "learning_rate": 1.914860681114551e-05,
+      "loss": 0.0191,
+      "step": 2163
+    },
+    {
+      "epoch": 127.29,
+      "learning_rate": 1.913312693498452e-05,
+      "loss": 0.0178,
+      "step": 2164
+    },
+    {
+      "epoch": 127.35,
+      "learning_rate": 1.9117647058823528e-05,
+      "loss": 0.022,
+      "step": 2165
+    },
+    {
+      "epoch": 127.41,
+      "learning_rate": 1.9102167182662538e-05,
+      "loss": 0.0237,
+      "step": 2166
+    },
+    {
+      "epoch": 127.47,
+      "learning_rate": 1.908668730650155e-05,
+      "loss": 0.0238,
+      "step": 2167
+    },
+    {
+      "epoch": 127.53,
+      "learning_rate": 1.9071207430340558e-05,
+      "loss": 0.0221,
+      "step": 2168
+    },
+    {
+      "epoch": 127.59,
+      "learning_rate": 1.9055727554179568e-05,
+      "loss": 0.0252,
+      "step": 2169
+    },
+    {
+      "epoch": 127.65,
+      "learning_rate": 1.9040247678018578e-05,
+      "loss": 0.0246,
+      "step": 2170
+    },
+    {
+      "epoch": 127.71,
+      "learning_rate": 1.9024767801857584e-05,
+      "loss": 0.0232,
+      "step": 2171
+    },
+    {
+      "epoch": 127.76,
+      "learning_rate": 1.9009287925696594e-05,
+      "loss": 0.0259,
+      "step": 2172
+    },
+    {
+      "epoch": 127.82,
+      "learning_rate": 1.8993808049535604e-05,
+      "loss": 0.0269,
+      "step": 2173
+    },
+    {
+      "epoch": 127.88,
+      "learning_rate": 1.897832817337461e-05,
+      "loss": 0.0257,
+      "step": 2174
+    },
+    {
+      "epoch": 127.94,
+      "learning_rate": 1.8962848297213624e-05,
+      "loss": 0.0242,
+      "step": 2175
+    },
+    {
+      "epoch": 128.0,
+      "learning_rate": 1.8947368421052634e-05,
+      "loss": 0.0248,
+      "step": 2176
+    },
+    {
+      "epoch": 128.06,
+      "learning_rate": 1.893188854489164e-05,
+      "loss": 0.0244,
+      "step": 2177
+    },
+    {
+      "epoch": 128.12,
+      "learning_rate": 1.891640866873065e-05,
+      "loss": 0.0211,
+      "step": 2178
+    },
+    {
+      "epoch": 128.18,
+      "learning_rate": 1.890092879256966e-05,
+      "loss": 0.0257,
+      "step": 2179
+    },
+    {
+      "epoch": 128.24,
+      "learning_rate": 1.8885448916408668e-05,
+      "loss": 0.0274,
+      "step": 2180
+    },
+    {
+      "epoch": 128.24,
+      "eval_accuracy_NAT": 0.9934557593905254,
+      "eval_accuracy_REST": 0.33658536585365856,
+      "eval_accuracy_concrete": 0.956248882067056,
+      "eval_accuracy_concrete tiles": 0.9204807918925288,
+      "eval_accuracy_dark bricks": 0.862656241027506,
+      "eval_accuracy_grass": 0.897614370550364,
+      "eval_accuracy_red bricks": 0.9170144991212654,
+      "eval_accuracy_road": 0.9812382528654057,
+      "eval_accuracy_rough concrete": 0.6233978274212097,
+      "eval_accuracy_rough red sidewalk": 0.7903503560712594,
+      "eval_accuracy_speedway bricks": 0.9889071906876061,
+      "eval_accuracy_steel": 0.9197873924842693,
+      "eval_accuracy_tiles": 0.2291193310915205,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833469507288637,
+      "eval_iou_REST": 0.3312,
+      "eval_iou_concrete": 0.8903719936612573,
+      "eval_iou_concrete tiles": 0.868026848024181,
+      "eval_iou_dark bricks": 0.8469391590321823,
+      "eval_iou_grass": 0.8407603344166807,
+      "eval_iou_red bricks": 0.8969166308551784,
+      "eval_iou_road": 0.9592772143096353,
+      "eval_iou_rough concrete": 0.6176197982159408,
+      "eval_iou_rough red sidewalk": 0.6296995296725683,
+      "eval_iou_speedway bricks": 0.978939624004374,
+      "eval_iou_steel": 0.8638676486464919,
+      "eval_iou_tiles": 0.2272262949164764,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08005253225564957,
+      "eval_mean_accuracy": 0.8012966354249365,
+      "eval_mean_iou": 0.7641686174218332,
+      "eval_overall_accuracy": 0.9797458325402212,
+      "eval_runtime": 5.5667,
+      "eval_samples_per_second": 21.197,
+      "eval_steps_per_second": 1.437,
+      "step": 2180
+    },
+    {
+      "epoch": 128.29,
+      "learning_rate": 1.8869969040247678e-05,
+      "loss": 0.0238,
+      "step": 2181
+    },
+    {
+      "epoch": 128.35,
+      "learning_rate": 1.8854489164086688e-05,
+      "loss": 0.0254,
+      "step": 2182
+    },
+    {
+      "epoch": 128.41,
+      "learning_rate": 1.8839009287925698e-05,
+      "loss": 0.0247,
+      "step": 2183
+    },
+    {
+      "epoch": 128.47,
+      "learning_rate": 1.8823529411764708e-05,
+      "loss": 0.0245,
+      "step": 2184
+    },
+    {
+      "epoch": 128.53,
+      "learning_rate": 1.8808049535603718e-05,
+      "loss": 0.0304,
+      "step": 2185
+    },
+    {
+      "epoch": 128.59,
+      "learning_rate": 1.8792569659442724e-05,
+      "loss": 0.0229,
+      "step": 2186
+    },
+    {
+      "epoch": 128.65,
+      "learning_rate": 1.8777089783281734e-05,
+      "loss": 0.021,
+      "step": 2187
+    },
+    {
+      "epoch": 128.71,
+      "learning_rate": 1.8761609907120744e-05,
+      "loss": 0.0217,
+      "step": 2188
+    },
+    {
+      "epoch": 128.76,
+      "learning_rate": 1.874613003095975e-05,
+      "loss": 0.023,
+      "step": 2189
+    },
+    {
+      "epoch": 128.82,
+      "learning_rate": 1.873065015479876e-05,
+      "loss": 0.0249,
+      "step": 2190
+    },
+    {
+      "epoch": 128.88,
+      "learning_rate": 1.871517027863777e-05,
+      "loss": 0.024,
+      "step": 2191
+    },
+    {
+      "epoch": 128.94,
+      "learning_rate": 1.869969040247678e-05,
+      "loss": 0.0231,
+      "step": 2192
+    },
+    {
+      "epoch": 129.0,
+      "learning_rate": 1.868421052631579e-05,
+      "loss": 0.0215,
+      "step": 2193
+    },
+    {
+      "epoch": 129.06,
+      "learning_rate": 1.86687306501548e-05,
+      "loss": 0.0225,
+      "step": 2194
+    },
+    {
+      "epoch": 129.12,
+      "learning_rate": 1.8653250773993807e-05,
+      "loss": 0.0238,
+      "step": 2195
+    },
+    {
+      "epoch": 129.18,
+      "learning_rate": 1.8637770897832817e-05,
+      "loss": 0.0235,
+      "step": 2196
+    },
+    {
+      "epoch": 129.24,
+      "learning_rate": 1.8622291021671827e-05,
+      "loss": 0.0292,
+      "step": 2197
+    },
+    {
+      "epoch": 129.29,
+      "learning_rate": 1.8606811145510834e-05,
+      "loss": 0.0248,
+      "step": 2198
+    },
+    {
+      "epoch": 129.35,
+      "learning_rate": 1.8591331269349847e-05,
+      "loss": 0.0214,
+      "step": 2199
+    },
+    {
+      "epoch": 129.41,
+      "learning_rate": 1.8575851393188857e-05,
+      "loss": 0.0254,
+      "step": 2200
+    },
+    {
+      "epoch": 129.41,
+      "eval_accuracy_NAT": 0.9935555500565562,
+      "eval_accuracy_REST": 0.25365853658536586,
+      "eval_accuracy_concrete": 0.9551045773716613,
+      "eval_accuracy_concrete tiles": 0.9270091916097101,
+      "eval_accuracy_dark bricks": 0.8635080297838945,
+      "eval_accuracy_grass": 0.8959707729166904,
+      "eval_accuracy_red bricks": 0.9122913005272407,
+      "eval_accuracy_road": 0.981887034049255,
+      "eval_accuracy_rough concrete": 0.6250126624518082,
+      "eval_accuracy_rough red sidewalk": 0.7929447506749185,
+      "eval_accuracy_speedway bricks": 0.9885699440595583,
+      "eval_accuracy_steel": 0.9259597075272257,
+      "eval_accuracy_tiles": 0.22233330303856524,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833154883794909,
+      "eval_iou_REST": 0.25120772946859904,
+      "eval_iou_concrete": 0.8910654903557672,
+      "eval_iou_concrete tiles": 0.8657935285053929,
+      "eval_iou_dark bricks": 0.8460979406579392,
+      "eval_iou_grass": 0.8386083993067149,
+      "eval_iou_red bricks": 0.8924887169568021,
+      "eval_iou_road": 0.9590406399074631,
+      "eval_iou_rough concrete": 0.6188470184257571,
+      "eval_iou_rough red sidewalk": 0.6303955329172346,
+      "eval_iou_speedway bricks": 0.9791585654173303,
+      "eval_iou_steel": 0.863598332210753,
+      "eval_iou_tiles": 0.22064880791317157,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07959654182195663,
+      "eval_mean_accuracy": 0.7952157969732654,
+      "eval_mean_iou": 0.7569435531094165,
+      "eval_overall_accuracy": 0.9797311556541314,
+      "eval_runtime": 5.3202,
+      "eval_samples_per_second": 22.179,
+      "eval_steps_per_second": 1.504,
+      "step": 2200
+    },
+    {
+      "epoch": 129.47,
+      "learning_rate": 1.8560371517027864e-05,
+      "loss": 0.0204,
+      "step": 2201
+    },
+    {
+      "epoch": 129.53,
+      "learning_rate": 1.8544891640866874e-05,
+      "loss": 0.0226,
+      "step": 2202
+    },
+    {
+      "epoch": 129.59,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.02,
+      "step": 2203
+    },
+    {
+      "epoch": 129.65,
+      "learning_rate": 1.851393188854489e-05,
+      "loss": 0.0221,
+      "step": 2204
+    },
+    {
+      "epoch": 129.71,
+      "learning_rate": 1.84984520123839e-05,
+      "loss": 0.0248,
+      "step": 2205
+    },
+    {
+      "epoch": 129.76,
+      "learning_rate": 1.848297213622291e-05,
+      "loss": 0.0301,
+      "step": 2206
+    },
+    {
+      "epoch": 129.82,
+      "learning_rate": 1.846749226006192e-05,
+      "loss": 0.0219,
+      "step": 2207
+    },
+    {
+      "epoch": 129.88,
+      "learning_rate": 1.845201238390093e-05,
+      "loss": 0.0224,
+      "step": 2208
+    },
+    {
+      "epoch": 129.94,
+      "learning_rate": 1.843653250773994e-05,
+      "loss": 0.0234,
+      "step": 2209
+    },
+    {
+      "epoch": 130.0,
+      "learning_rate": 1.8421052631578947e-05,
+      "loss": 0.0247,
+      "step": 2210
+    },
+    {
+      "epoch": 130.06,
+      "learning_rate": 1.8405572755417957e-05,
+      "loss": 0.0215,
+      "step": 2211
+    },
+    {
+      "epoch": 130.12,
+      "learning_rate": 1.8390092879256967e-05,
+      "loss": 0.0233,
+      "step": 2212
+    },
+    {
+      "epoch": 130.18,
+      "learning_rate": 1.8374613003095974e-05,
+      "loss": 0.0265,
+      "step": 2213
+    },
+    {
+      "epoch": 130.24,
+      "learning_rate": 1.8359133126934984e-05,
+      "loss": 0.0244,
+      "step": 2214
+    },
+    {
+      "epoch": 130.29,
+      "learning_rate": 1.8343653250773994e-05,
+      "loss": 0.0237,
+      "step": 2215
+    },
+    {
+      "epoch": 130.35,
+      "learning_rate": 1.8328173374613004e-05,
+      "loss": 0.0201,
+      "step": 2216
+    },
+    {
+      "epoch": 130.41,
+      "learning_rate": 1.8312693498452014e-05,
+      "loss": 0.0247,
+      "step": 2217
+    },
+    {
+      "epoch": 130.47,
+      "learning_rate": 1.8297213622291024e-05,
+      "loss": 0.0199,
+      "step": 2218
+    },
+    {
+      "epoch": 130.53,
+      "learning_rate": 1.828173374613003e-05,
+      "loss": 0.0277,
+      "step": 2219
+    },
+    {
+      "epoch": 130.59,
+      "learning_rate": 1.826625386996904e-05,
+      "loss": 0.024,
+      "step": 2220
+    },
+    {
+      "epoch": 130.59,
+      "eval_accuracy_NAT": 0.9935954447582648,
+      "eval_accuracy_REST": 0.1951219512195122,
+      "eval_accuracy_concrete": 0.9571539300024919,
+      "eval_accuracy_concrete tiles": 0.9302380391232619,
+      "eval_accuracy_dark bricks": 0.8622255613192198,
+      "eval_accuracy_grass": 0.897022031967359,
+      "eval_accuracy_red bricks": 0.9228910369068541,
+      "eval_accuracy_road": 0.9811046488324878,
+      "eval_accuracy_rough concrete": 0.6275153588091933,
+      "eval_accuracy_rough red sidewalk": 0.7900194767594884,
+      "eval_accuracy_speedway bricks": 0.98927819625149,
+      "eval_accuracy_steel": 0.9205870446355835,
+      "eval_accuracy_tiles": 0.22615044381835256,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834839026964168,
+      "eval_iou_REST": 0.1941747572815534,
+      "eval_iou_concrete": 0.8926597734712732,
+      "eval_iou_concrete tiles": 0.8687900332372169,
+      "eval_iou_dark bricks": 0.8467271625069902,
+      "eval_iou_grass": 0.8409859072465106,
+      "eval_iou_red bricks": 0.8988980421525623,
+      "eval_iou_road": 0.9612367707449223,
+      "eval_iou_rough concrete": 0.6217285291738743,
+      "eval_iou_rough red sidewalk": 0.6272299572516896,
+      "eval_iou_speedway bricks": 0.9790627434227006,
+      "eval_iou_steel": 0.8649617641327138,
+      "eval_iou_tiles": 0.22470726347792058,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07831832021474838,
+      "eval_mean_accuracy": 0.7917617818771969,
+      "eval_mean_iou": 0.7542035851381802,
+      "eval_overall_accuracy": 0.9799870313224146,
+      "eval_runtime": 12.6947,
+      "eval_samples_per_second": 9.295,
+      "eval_steps_per_second": 0.63,
+      "step": 2220
+    },
+    {
+      "epoch": 130.65,
+      "learning_rate": 1.825077399380805e-05,
+      "loss": 0.0225,
+      "step": 2221
+    },
+    {
+      "epoch": 130.71,
+      "learning_rate": 1.8235294117647057e-05,
+      "loss": 0.0229,
+      "step": 2222
+    },
+    {
+      "epoch": 130.76,
+      "learning_rate": 1.821981424148607e-05,
+      "loss": 0.0258,
+      "step": 2223
+    },
+    {
+      "epoch": 130.82,
+      "learning_rate": 1.820433436532508e-05,
+      "loss": 0.0201,
+      "step": 2224
+    },
+    {
+      "epoch": 130.88,
+      "learning_rate": 1.8188854489164087e-05,
+      "loss": 0.0273,
+      "step": 2225
+    },
+    {
+      "epoch": 130.94,
+      "learning_rate": 1.8173374613003097e-05,
+      "loss": 0.0278,
+      "step": 2226
+    },
+    {
+      "epoch": 131.0,
+      "learning_rate": 1.8157894736842107e-05,
+      "loss": 0.0206,
+      "step": 2227
+    },
+    {
+      "epoch": 131.06,
+      "learning_rate": 1.8142414860681114e-05,
+      "loss": 0.0211,
+      "step": 2228
+    },
+    {
+      "epoch": 131.12,
+      "learning_rate": 1.8126934984520124e-05,
+      "loss": 0.0232,
+      "step": 2229
+    },
+    {
+      "epoch": 131.18,
+      "learning_rate": 1.8111455108359134e-05,
+      "loss": 0.0236,
+      "step": 2230
+    },
+    {
+      "epoch": 131.24,
+      "learning_rate": 1.8095975232198144e-05,
+      "loss": 0.0227,
+      "step": 2231
+    },
+    {
+      "epoch": 131.29,
+      "learning_rate": 1.8080495356037154e-05,
+      "loss": 0.0198,
+      "step": 2232
+    },
+    {
+      "epoch": 131.35,
+      "learning_rate": 1.8065015479876164e-05,
+      "loss": 0.0268,
+      "step": 2233
+    },
+    {
+      "epoch": 131.41,
+      "learning_rate": 1.804953560371517e-05,
+      "loss": 0.0228,
+      "step": 2234
+    },
+    {
+      "epoch": 131.47,
+      "learning_rate": 1.803405572755418e-05,
+      "loss": 0.0243,
+      "step": 2235
+    },
+    {
+      "epoch": 131.53,
+      "learning_rate": 1.801857585139319e-05,
+      "loss": 0.0226,
+      "step": 2236
+    },
+    {
+      "epoch": 131.59,
+      "learning_rate": 1.8003095975232197e-05,
+      "loss": 0.023,
+      "step": 2237
+    },
+    {
+      "epoch": 131.65,
+      "learning_rate": 1.7987616099071207e-05,
+      "loss": 0.0231,
+      "step": 2238
+    },
+    {
+      "epoch": 131.71,
+      "learning_rate": 1.7972136222910217e-05,
+      "loss": 0.0263,
+      "step": 2239
+    },
+    {
+      "epoch": 131.76,
+      "learning_rate": 1.7956656346749227e-05,
+      "loss": 0.027,
+      "step": 2240
+    },
+    {
+      "epoch": 131.76,
+      "eval_accuracy_NAT": 0.9933460489608266,
+      "eval_accuracy_REST": 0.2813008130081301,
+      "eval_accuracy_concrete": 0.9516517889220298,
+      "eval_accuracy_concrete tiles": 0.9241102993165213,
+      "eval_accuracy_dark bricks": 0.8711549872710219,
+      "eval_accuracy_grass": 0.9026577645558218,
+      "eval_accuracy_red bricks": 0.9344793497363796,
+      "eval_accuracy_road": 0.9833336985176899,
+      "eval_accuracy_rough concrete": 0.6333192308379861,
+      "eval_accuracy_rough red sidewalk": 0.7868610833289467,
+      "eval_accuracy_speedway bricks": 0.9902222468988048,
+      "eval_accuracy_steel": 0.9146421306731571,
+      "eval_accuracy_tiles": 0.22748341361446878,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833660296445369,
+      "eval_iou_REST": 0.27768860353130015,
+      "eval_iou_concrete": 0.8926005456116264,
+      "eval_iou_concrete tiles": 0.8660791200053012,
+      "eval_iou_dark bricks": 0.8496665219806121,
+      "eval_iou_grass": 0.8405310579747688,
+      "eval_iou_red bricks": 0.9017913928344287,
+      "eval_iou_road": 0.9596837033474955,
+      "eval_iou_rough concrete": 0.6236826045114194,
+      "eval_iou_rough red sidewalk": 0.626210396543263,
+      "eval_iou_speedway bricks": 0.9789607023727256,
+      "eval_iou_steel": 0.8631027896337868,
+      "eval_iou_tiles": 0.2257123962967416,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07864497601985931,
+      "eval_mean_accuracy": 0.7995817581262911,
+      "eval_mean_iou": 0.7606981434067698,
+      "eval_overall_accuracy": 0.9798804460945776,
+      "eval_runtime": 5.5547,
+      "eval_samples_per_second": 21.243,
+      "eval_steps_per_second": 1.44,
+      "step": 2240
+    },
+    {
+      "epoch": 131.82,
+      "learning_rate": 1.7941176470588237e-05,
+      "loss": 0.0234,
+      "step": 2241
+    },
+    {
+      "epoch": 131.88,
+      "learning_rate": 1.7925696594427247e-05,
+      "loss": 0.0241,
+      "step": 2242
+    },
+    {
+      "epoch": 131.94,
+      "learning_rate": 1.7910216718266254e-05,
+      "loss": 0.0251,
+      "step": 2243
+    },
+    {
+      "epoch": 132.0,
+      "learning_rate": 1.7894736842105264e-05,
+      "loss": 0.0214,
+      "step": 2244
+    },
+    {
+      "epoch": 132.06,
+      "learning_rate": 1.7879256965944274e-05,
+      "loss": 0.0244,
+      "step": 2245
+    },
+    {
+      "epoch": 132.12,
+      "learning_rate": 1.786377708978328e-05,
+      "loss": 0.0241,
+      "step": 2246
+    },
+    {
+      "epoch": 132.18,
+      "learning_rate": 1.784829721362229e-05,
+      "loss": 0.0231,
+      "step": 2247
+    },
+    {
+      "epoch": 132.24,
+      "learning_rate": 1.7832817337461304e-05,
+      "loss": 0.0234,
+      "step": 2248
+    },
+    {
+      "epoch": 132.29,
+      "learning_rate": 1.781733746130031e-05,
+      "loss": 0.0305,
+      "step": 2249
+    },
+    {
+      "epoch": 132.35,
+      "learning_rate": 1.780185758513932e-05,
+      "loss": 0.0207,
+      "step": 2250
+    },
+    {
+      "epoch": 132.41,
+      "learning_rate": 1.778637770897833e-05,
+      "loss": 0.0217,
+      "step": 2251
+    },
+    {
+      "epoch": 132.47,
+      "learning_rate": 1.7770897832817337e-05,
+      "loss": 0.0219,
+      "step": 2252
+    },
+    {
+      "epoch": 132.53,
+      "learning_rate": 1.7755417956656347e-05,
+      "loss": 0.0259,
+      "step": 2253
+    },
+    {
+      "epoch": 132.59,
+      "learning_rate": 1.7739938080495357e-05,
+      "loss": 0.0252,
+      "step": 2254
+    },
+    {
+      "epoch": 132.65,
+      "learning_rate": 1.7724458204334367e-05,
+      "loss": 0.0255,
+      "step": 2255
+    },
+    {
+      "epoch": 132.71,
+      "learning_rate": 1.7708978328173377e-05,
+      "loss": 0.0209,
+      "step": 2256
+    },
+    {
+      "epoch": 132.76,
+      "learning_rate": 1.7693498452012387e-05,
+      "loss": 0.0211,
+      "step": 2257
+    },
+    {
+      "epoch": 132.82,
+      "learning_rate": 1.7678018575851394e-05,
+      "loss": 0.0235,
+      "step": 2258
+    },
+    {
+      "epoch": 132.88,
+      "learning_rate": 1.7662538699690404e-05,
+      "loss": 0.0237,
+      "step": 2259
+    },
+    {
+      "epoch": 132.94,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.0208,
+      "step": 2260
+    },
+    {
+      "epoch": 132.94,
+      "eval_accuracy_NAT": 0.9935048190912753,
+      "eval_accuracy_REST": 0.23577235772357724,
+      "eval_accuracy_concrete": 0.9550541270644495,
+      "eval_accuracy_concrete tiles": 0.9257365071883101,
+      "eval_accuracy_dark bricks": 0.8583015906437226,
+      "eval_accuracy_grass": 0.8920401427668457,
+      "eval_accuracy_red bricks": 0.9265158172231986,
+      "eval_accuracy_road": 0.9820510998016782,
+      "eval_accuracy_rough concrete": 0.6282363737121541,
+      "eval_accuracy_rough red sidewalk": 0.8028184901375405,
+      "eval_accuracy_speedway bricks": 0.9902823960687462,
+      "eval_accuracy_steel": 0.9136075807023945,
+      "eval_accuracy_tiles": 0.23078554333666576,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833120272223053,
+      "eval_iou_REST": 0.23387096774193547,
+      "eval_iou_concrete": 0.8921378601227081,
+      "eval_iou_concrete tiles": 0.8693506263556283,
+      "eval_iou_dark bricks": 0.8425886343243967,
+      "eval_iou_grass": 0.8364732887147049,
+      "eval_iou_red bricks": 0.8995414311613522,
+      "eval_iou_road": 0.9604321491270936,
+      "eval_iou_rough concrete": 0.6216391509433963,
+      "eval_iou_rough red sidewalk": 0.6271065971957072,
+      "eval_iou_speedway bricks": 0.9790520140098254,
+      "eval_iou_steel": 0.8638070148495552,
+      "eval_iou_tiles": 0.2291127819548872,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0800853818655014,
+      "eval_mean_accuracy": 0.7949774496508122,
+      "eval_mean_iou": 0.7568018879787305,
+      "eval_overall_accuracy": 0.9797754126080012,
+      "eval_runtime": 5.6412,
+      "eval_samples_per_second": 20.918,
+      "eval_steps_per_second": 1.418,
+      "step": 2260
+    },
+    {
+      "epoch": 133.0,
+      "learning_rate": 1.763157894736842e-05,
+      "loss": 0.025,
+      "step": 2261
+    },
+    {
+      "epoch": 133.06,
+      "learning_rate": 1.761609907120743e-05,
+      "loss": 0.0219,
+      "step": 2262
+    },
+    {
+      "epoch": 133.12,
+      "learning_rate": 1.760061919504644e-05,
+      "loss": 0.0245,
+      "step": 2263
+    },
+    {
+      "epoch": 133.18,
+      "learning_rate": 1.758513931888545e-05,
+      "loss": 0.0219,
+      "step": 2264
+    },
+    {
+      "epoch": 133.24,
+      "learning_rate": 1.756965944272446e-05,
+      "loss": 0.0239,
+      "step": 2265
+    },
+    {
+      "epoch": 133.29,
+      "learning_rate": 1.755417956656347e-05,
+      "loss": 0.0236,
+      "step": 2266
+    },
+    {
+      "epoch": 133.35,
+      "learning_rate": 1.7538699690402477e-05,
+      "loss": 0.0256,
+      "step": 2267
+    },
+    {
+      "epoch": 133.41,
+      "learning_rate": 1.7523219814241487e-05,
+      "loss": 0.02,
+      "step": 2268
+    },
+    {
+      "epoch": 133.47,
+      "learning_rate": 1.7507739938080497e-05,
+      "loss": 0.0244,
+      "step": 2269
+    },
+    {
+      "epoch": 133.53,
+      "learning_rate": 1.7492260061919503e-05,
+      "loss": 0.0241,
+      "step": 2270
+    },
+    {
+      "epoch": 133.59,
+      "learning_rate": 1.7476780185758513e-05,
+      "loss": 0.0219,
+      "step": 2271
+    },
+    {
+      "epoch": 133.65,
+      "learning_rate": 1.7461300309597527e-05,
+      "loss": 0.0259,
+      "step": 2272
+    },
+    {
+      "epoch": 133.71,
+      "learning_rate": 1.7445820433436533e-05,
+      "loss": 0.0235,
+      "step": 2273
+    },
+    {
+      "epoch": 133.76,
+      "learning_rate": 1.7430340557275543e-05,
+      "loss": 0.0278,
+      "step": 2274
+    },
+    {
+      "epoch": 133.82,
+      "learning_rate": 1.7414860681114553e-05,
+      "loss": 0.0242,
+      "step": 2275
+    },
+    {
+      "epoch": 133.88,
+      "learning_rate": 1.739938080495356e-05,
+      "loss": 0.0235,
+      "step": 2276
+    },
+    {
+      "epoch": 133.94,
+      "learning_rate": 1.738390092879257e-05,
+      "loss": 0.0209,
+      "step": 2277
+    },
+    {
+      "epoch": 134.0,
+      "learning_rate": 1.736842105263158e-05,
+      "loss": 0.021,
+      "step": 2278
+    },
+    {
+      "epoch": 134.06,
+      "learning_rate": 1.735294117647059e-05,
+      "loss": 0.0271,
+      "step": 2279
+    },
+    {
+      "epoch": 134.12,
+      "learning_rate": 1.73374613003096e-05,
+      "loss": 0.0251,
+      "step": 2280
+    },
+    {
+      "epoch": 134.12,
+      "eval_accuracy_NAT": 0.9937414378018149,
+      "eval_accuracy_REST": 0.2894308943089431,
+      "eval_accuracy_concrete": 0.9541479325311225,
+      "eval_accuracy_concrete tiles": 0.9326891350459581,
+      "eval_accuracy_dark bricks": 0.8686570449629616,
+      "eval_accuracy_grass": 0.8883053817650555,
+      "eval_accuracy_red bricks": 0.9182227592267135,
+      "eval_accuracy_road": 0.9821665336861193,
+      "eval_accuracy_rough concrete": 0.6233501570144024,
+      "eval_accuracy_rough red sidewalk": 0.7845750080839832,
+      "eval_accuracy_speedway bricks": 0.990513567664903,
+      "eval_accuracy_steel": 0.9137450209159016,
+      "eval_accuracy_tiles": 0.2340573782907692,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9832757547279972,
+      "eval_iou_REST": 0.2848,
+      "eval_iou_concrete": 0.8915332682188205,
+      "eval_iou_concrete tiles": 0.8688035126234906,
+      "eval_iou_dark bricks": 0.8490848453381605,
+      "eval_iou_grass": 0.8361874698271485,
+      "eval_iou_red bricks": 0.8970383088314197,
+      "eval_iou_road": 0.960265492366807,
+      "eval_iou_rough concrete": 0.6178315359264815,
+      "eval_iou_rough red sidewalk": 0.6223054624405024,
+      "eval_iou_speedway bricks": 0.9789670917170973,
+      "eval_iou_steel": 0.8633780302654662,
+      "eval_iou_tiles": 0.23225612505636556,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08101151883602142,
+      "eval_mean_accuracy": 0.7979694039460499,
+      "eval_mean_iou": 0.7604405305645968,
+      "eval_overall_accuracy": 0.9797396579031217,
+      "eval_runtime": 5.5371,
+      "eval_samples_per_second": 21.311,
+      "eval_steps_per_second": 1.445,
+      "step": 2280
+    },
+    {
+      "epoch": 134.18,
+      "learning_rate": 1.732198142414861e-05,
+      "loss": 0.0266,
+      "step": 2281
+    },
+    {
+      "epoch": 134.24,
+      "learning_rate": 1.7306501547987617e-05,
+      "loss": 0.0236,
+      "step": 2282
+    },
+    {
+      "epoch": 134.29,
+      "learning_rate": 1.7291021671826627e-05,
+      "loss": 0.0216,
+      "step": 2283
+    },
+    {
+      "epoch": 134.35,
+      "learning_rate": 1.7275541795665637e-05,
+      "loss": 0.0201,
+      "step": 2284
+    },
+    {
+      "epoch": 134.41,
+      "learning_rate": 1.7260061919504643e-05,
+      "loss": 0.024,
+      "step": 2285
+    },
+    {
+      "epoch": 134.47,
+      "learning_rate": 1.7244582043343653e-05,
+      "loss": 0.0221,
+      "step": 2286
+    },
+    {
+      "epoch": 134.53,
+      "learning_rate": 1.7229102167182663e-05,
+      "loss": 0.0252,
+      "step": 2287
+    },
+    {
+      "epoch": 134.59,
+      "learning_rate": 1.7213622291021673e-05,
+      "loss": 0.0259,
+      "step": 2288
+    },
+    {
+      "epoch": 134.65,
+      "learning_rate": 1.7198142414860683e-05,
+      "loss": 0.0251,
+      "step": 2289
+    },
+    {
+      "epoch": 134.71,
+      "learning_rate": 1.7182662538699693e-05,
+      "loss": 0.0234,
+      "step": 2290
+    },
+    {
+      "epoch": 134.76,
+      "learning_rate": 1.71671826625387e-05,
+      "loss": 0.0269,
+      "step": 2291
+    },
+    {
+      "epoch": 134.82,
+      "learning_rate": 1.715170278637771e-05,
+      "loss": 0.0223,
+      "step": 2292
+    },
+    {
+      "epoch": 134.88,
+      "learning_rate": 1.713622291021672e-05,
+      "loss": 0.0224,
+      "step": 2293
+    },
+    {
+      "epoch": 134.94,
+      "learning_rate": 1.7120743034055727e-05,
+      "loss": 0.0211,
+      "step": 2294
+    },
+    {
+      "epoch": 135.0,
+      "learning_rate": 1.7105263157894737e-05,
+      "loss": 0.0218,
+      "step": 2295
+    },
+    {
+      "epoch": 135.06,
+      "learning_rate": 1.708978328173375e-05,
+      "loss": 0.0226,
+      "step": 2296
+    },
+    {
+      "epoch": 135.12,
+      "learning_rate": 1.7074303405572757e-05,
+      "loss": 0.0231,
+      "step": 2297
+    },
+    {
+      "epoch": 135.18,
+      "learning_rate": 1.7058823529411767e-05,
+      "loss": 0.0258,
+      "step": 2298
+    },
+    {
+      "epoch": 135.24,
+      "learning_rate": 1.7043343653250773e-05,
+      "loss": 0.0238,
+      "step": 2299
+    },
+    {
+      "epoch": 135.29,
+      "learning_rate": 1.7027863777089783e-05,
+      "loss": 0.0217,
+      "step": 2300
+    },
+    {
+      "epoch": 135.29,
+      "eval_accuracy_NAT": 0.9942153221640029,
+      "eval_accuracy_REST": 0.3284552845528455,
+      "eval_accuracy_concrete": 0.9531117902367954,
+      "eval_accuracy_concrete tiles": 0.9385104878623616,
+      "eval_accuracy_dark bricks": 0.8568420649656413,
+      "eval_accuracy_grass": 0.888304435537287,
+      "eval_accuracy_red bricks": 0.9183326010544816,
+      "eval_accuracy_road": 0.9802656155057634,
+      "eval_accuracy_rough concrete": 0.6243691119599092,
+      "eval_accuracy_rough red sidewalk": 0.7928018709721084,
+      "eval_accuracy_speedway bricks": 0.9903781891912455,
+      "eval_accuracy_steel": 0.9173159675541139,
+      "eval_accuracy_tiles": 0.23932866793904692,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833148932653317,
+      "eval_iou_REST": 0.3201267828843106,
+      "eval_iou_concrete": 0.8921571924421812,
+      "eval_iou_concrete tiles": 0.8764774503114476,
+      "eval_iou_dark bricks": 0.840271618495211,
+      "eval_iou_grass": 0.8373657700311385,
+      "eval_iou_red bricks": 0.8983023530675835,
+      "eval_iou_road": 0.9603277401115154,
+      "eval_iou_rough concrete": 0.6188341601700922,
+      "eval_iou_rough red sidewalk": 0.6297360420039065,
+      "eval_iou_speedway bricks": 0.9788994430818846,
+      "eval_iou_steel": 0.8644009127065423,
+      "eval_iou_tiles": 0.23658361284139914,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08090127259492874,
+      "eval_mean_accuracy": 0.8017101084227387,
+      "eval_mean_iou": 0.7643690747240419,
+      "eval_overall_accuracy": 0.9798173419499802,
+      "eval_runtime": 5.5735,
+      "eval_samples_per_second": 21.172,
+      "eval_steps_per_second": 1.435,
+      "step": 2300
+    },
+    {
+      "epoch": 135.35,
+      "learning_rate": 1.7012383900928793e-05,
+      "loss": 0.021,
+      "step": 2301
+    },
+    {
+      "epoch": 135.41,
+      "learning_rate": 1.69969040247678e-05,
+      "loss": 0.0261,
+      "step": 2302
+    },
+    {
+      "epoch": 135.47,
+      "learning_rate": 1.698142414860681e-05,
+      "loss": 0.0234,
+      "step": 2303
+    },
+    {
+      "epoch": 135.53,
+      "learning_rate": 1.6965944272445823e-05,
+      "loss": 0.0234,
+      "step": 2304
+    },
+    {
+      "epoch": 135.59,
+      "learning_rate": 1.695046439628483e-05,
+      "loss": 0.0241,
+      "step": 2305
+    },
+    {
+      "epoch": 135.65,
+      "learning_rate": 1.693498452012384e-05,
+      "loss": 0.0216,
+      "step": 2306
+    },
+    {
+      "epoch": 135.71,
+      "learning_rate": 1.691950464396285e-05,
+      "loss": 0.022,
+      "step": 2307
+    },
+    {
+      "epoch": 135.76,
+      "learning_rate": 1.6904024767801856e-05,
+      "loss": 0.0235,
+      "step": 2308
+    },
+    {
+      "epoch": 135.82,
+      "learning_rate": 1.6888544891640866e-05,
+      "loss": 0.0225,
+      "step": 2309
+    },
+    {
+      "epoch": 135.88,
+      "learning_rate": 1.6873065015479876e-05,
+      "loss": 0.022,
+      "step": 2310
+    },
+    {
+      "epoch": 135.94,
+      "learning_rate": 1.6857585139318886e-05,
+      "loss": 0.0256,
+      "step": 2311
+    },
+    {
+      "epoch": 136.0,
+      "learning_rate": 1.6842105263157896e-05,
+      "loss": 0.0266,
+      "step": 2312
+    },
+    {
+      "epoch": 136.06,
+      "learning_rate": 1.6826625386996906e-05,
+      "loss": 0.0214,
+      "step": 2313
+    },
+    {
+      "epoch": 136.12,
+      "learning_rate": 1.6811145510835913e-05,
+      "loss": 0.0235,
+      "step": 2314
+    },
+    {
+      "epoch": 136.18,
+      "learning_rate": 1.6795665634674923e-05,
+      "loss": 0.0217,
+      "step": 2315
+    },
+    {
+      "epoch": 136.24,
+      "learning_rate": 1.6780185758513933e-05,
+      "loss": 0.0207,
+      "step": 2316
+    },
+    {
+      "epoch": 136.29,
+      "learning_rate": 1.676470588235294e-05,
+      "loss": 0.0226,
+      "step": 2317
+    },
+    {
+      "epoch": 136.35,
+      "learning_rate": 1.674922600619195e-05,
+      "loss": 0.0249,
+      "step": 2318
+    },
+    {
+      "epoch": 136.41,
+      "learning_rate": 1.673374613003096e-05,
+      "loss": 0.0228,
+      "step": 2319
+    },
+    {
+      "epoch": 136.47,
+      "learning_rate": 1.671826625386997e-05,
+      "loss": 0.0249,
+      "step": 2320
+    },
+    {
+      "epoch": 136.47,
+      "eval_accuracy_NAT": 0.9936994405415026,
+      "eval_accuracy_REST": 0.2975609756097561,
+      "eval_accuracy_concrete": 0.9559874577478678,
+      "eval_accuracy_concrete tiles": 0.9356351637991986,
+      "eval_accuracy_dark bricks": 0.8588184062936661,
+      "eval_accuracy_grass": 0.8968725279799551,
+      "eval_accuracy_red bricks": 0.890981985940246,
+      "eval_accuracy_road": 0.9817833573197107,
+      "eval_accuracy_rough concrete": 0.6262401754270971,
+      "eval_accuracy_rough red sidewalk": 0.7743929492626656,
+      "eval_accuracy_speedway bricks": 0.9886919560510065,
+      "eval_accuracy_steel": 0.9167287230054926,
+      "eval_accuracy_tiles": 0.23445121027598534,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833042774626791,
+      "eval_iou_REST": 0.2928,
+      "eval_iou_concrete": 0.8912564382595186,
+      "eval_iou_concrete tiles": 0.8724863189820004,
+      "eval_iou_dark bricks": 0.8418857745983347,
+      "eval_iou_grass": 0.8377133635184951,
+      "eval_iou_red bricks": 0.8746495579038172,
+      "eval_iou_road": 0.9608221240016465,
+      "eval_iou_rough concrete": 0.6197698898986266,
+      "eval_iou_rough red sidewalk": 0.6202858743381702,
+      "eval_iou_speedway bricks": 0.9788788311894553,
+      "eval_iou_steel": 0.8643115801179428,
+      "eval_iou_tiles": 0.23236053563922415,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08120080828666687,
+      "eval_mean_accuracy": 0.7962957176349346,
+      "eval_mean_iou": 0.7592711204546085,
+      "eval_overall_accuracy": 0.979728569418697,
+      "eval_runtime": 7.8558,
+      "eval_samples_per_second": 15.021,
+      "eval_steps_per_second": 1.018,
+      "step": 2320
+    },
+    {
+      "epoch": 136.53,
+      "learning_rate": 1.670278637770898e-05,
+      "loss": 0.022,
+      "step": 2321
+    },
+    {
+      "epoch": 136.59,
+      "learning_rate": 1.668730650154799e-05,
+      "loss": 0.0223,
+      "step": 2322
+    },
+    {
+      "epoch": 136.65,
+      "learning_rate": 1.6671826625386996e-05,
+      "loss": 0.0292,
+      "step": 2323
+    },
+    {
+      "epoch": 136.71,
+      "learning_rate": 1.6656346749226006e-05,
+      "loss": 0.024,
+      "step": 2324
+    },
+    {
+      "epoch": 136.76,
+      "learning_rate": 1.6640866873065016e-05,
+      "loss": 0.0235,
+      "step": 2325
+    },
+    {
+      "epoch": 136.82,
+      "learning_rate": 1.6625386996904023e-05,
+      "loss": 0.024,
+      "step": 2326
+    },
+    {
+      "epoch": 136.88,
+      "learning_rate": 1.6609907120743033e-05,
+      "loss": 0.0223,
+      "step": 2327
+    },
+    {
+      "epoch": 136.94,
+      "learning_rate": 1.6594427244582046e-05,
+      "loss": 0.024,
+      "step": 2328
+    },
+    {
+      "epoch": 137.0,
+      "learning_rate": 1.6578947368421053e-05,
+      "loss": 0.0248,
+      "step": 2329
+    },
+    {
+      "epoch": 137.06,
+      "learning_rate": 1.6563467492260063e-05,
+      "loss": 0.0205,
+      "step": 2330
+    },
+    {
+      "epoch": 137.12,
+      "learning_rate": 1.6547987616099073e-05,
+      "loss": 0.0236,
+      "step": 2331
+    },
+    {
+      "epoch": 137.18,
+      "learning_rate": 1.653250773993808e-05,
+      "loss": 0.0215,
+      "step": 2332
+    },
+    {
+      "epoch": 137.24,
+      "learning_rate": 1.651702786377709e-05,
+      "loss": 0.0203,
+      "step": 2333
+    },
+    {
+      "epoch": 137.29,
+      "learning_rate": 1.65015479876161e-05,
+      "loss": 0.0253,
+      "step": 2334
+    },
+    {
+      "epoch": 137.35,
+      "learning_rate": 1.648606811145511e-05,
+      "loss": 0.0254,
+      "step": 2335
+    },
+    {
+      "epoch": 137.41,
+      "learning_rate": 1.647058823529412e-05,
+      "loss": 0.0234,
+      "step": 2336
+    },
+    {
+      "epoch": 137.47,
+      "learning_rate": 1.645510835913313e-05,
+      "loss": 0.0238,
+      "step": 2337
+    },
+    {
+      "epoch": 137.53,
+      "learning_rate": 1.6439628482972136e-05,
+      "loss": 0.0261,
+      "step": 2338
+    },
+    {
+      "epoch": 137.59,
+      "learning_rate": 1.6424148606811146e-05,
+      "loss": 0.0248,
+      "step": 2339
+    },
+    {
+      "epoch": 137.65,
+      "learning_rate": 1.6408668730650156e-05,
+      "loss": 0.0229,
+      "step": 2340
+    },
+    {
+      "epoch": 137.65,
+      "eval_accuracy_NAT": 0.9940121287440841,
+      "eval_accuracy_REST": 0.33983739837398375,
+      "eval_accuracy_concrete": 0.9522331140528566,
+      "eval_accuracy_concrete tiles": 0.9262078717888287,
+      "eval_accuracy_dark bricks": 0.8621968493386674,
+      "eval_accuracy_grass": 0.8958042368294556,
+      "eval_accuracy_red bricks": 0.9253075571177505,
+      "eval_accuracy_road": 0.9808871414668975,
+      "eval_accuracy_rough concrete": 0.625096085663721,
+      "eval_accuracy_rough red sidewalk": 0.797524421149204,
+      "eval_accuracy_speedway bricks": 0.9905301900850862,
+      "eval_accuracy_steel": 0.9102490416668749,
+      "eval_accuracy_tiles": 0.2338150201460208,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834414014568328,
+      "eval_iou_REST": 0.3333333333333333,
+      "eval_iou_concrete": 0.892247962907056,
+      "eval_iou_concrete tiles": 0.8731752838447353,
+      "eval_iou_dark bricks": 0.8429949656579268,
+      "eval_iou_grass": 0.8404846646845179,
+      "eval_iou_red bricks": 0.9012517385257302,
+      "eval_iou_road": 0.9608526151740876,
+      "eval_iou_rough concrete": 0.6188748480879733,
+      "eval_iou_rough red sidewalk": 0.6252667822231656,
+      "eval_iou_speedway bricks": 0.9787665257615278,
+      "eval_iou_steel": 0.8637641981456451,
+      "eval_iou_tiles": 0.23205748819868308,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08003875613212585,
+      "eval_mean_accuracy": 0.8025923889556488,
+      "eval_mean_iou": 0.7651162929231704,
+      "eval_overall_accuracy": 0.9798942501262083,
+      "eval_runtime": 5.5404,
+      "eval_samples_per_second": 21.298,
+      "eval_steps_per_second": 1.444,
+      "step": 2340
+    },
+    {
+      "epoch": 137.71,
+      "learning_rate": 1.6393188854489163e-05,
+      "loss": 0.021,
+      "step": 2341
+    },
+    {
+      "epoch": 137.76,
+      "learning_rate": 1.6377708978328173e-05,
+      "loss": 0.0249,
+      "step": 2342
+    },
+    {
+      "epoch": 137.82,
+      "learning_rate": 1.6362229102167183e-05,
+      "loss": 0.0216,
+      "step": 2343
+    },
+    {
+      "epoch": 137.88,
+      "learning_rate": 1.6346749226006193e-05,
+      "loss": 0.0169,
+      "step": 2344
+    },
+    {
+      "epoch": 137.94,
+      "learning_rate": 1.6331269349845203e-05,
+      "loss": 0.0226,
+      "step": 2345
+    },
+    {
+      "epoch": 138.0,
+      "learning_rate": 1.6315789473684213e-05,
+      "loss": 0.0215,
+      "step": 2346
+    },
+    {
+      "epoch": 138.06,
+      "learning_rate": 1.630030959752322e-05,
+      "loss": 0.0282,
+      "step": 2347
+    },
+    {
+      "epoch": 138.12,
+      "learning_rate": 1.628482972136223e-05,
+      "loss": 0.0239,
+      "step": 2348
+    },
+    {
+      "epoch": 138.18,
+      "learning_rate": 1.626934984520124e-05,
+      "loss": 0.0221,
+      "step": 2349
+    },
+    {
+      "epoch": 138.24,
+      "learning_rate": 1.6253869969040246e-05,
+      "loss": 0.0241,
+      "step": 2350
+    },
+    {
+      "epoch": 138.29,
+      "learning_rate": 1.6238390092879256e-05,
+      "loss": 0.0241,
+      "step": 2351
+    },
+    {
+      "epoch": 138.35,
+      "learning_rate": 1.622291021671827e-05,
+      "loss": 0.0217,
+      "step": 2352
+    },
+    {
+      "epoch": 138.41,
+      "learning_rate": 1.6207430340557276e-05,
+      "loss": 0.0176,
+      "step": 2353
+    },
+    {
+      "epoch": 138.47,
+      "learning_rate": 1.6191950464396286e-05,
+      "loss": 0.0256,
+      "step": 2354
+    },
+    {
+      "epoch": 138.53,
+      "learning_rate": 1.6176470588235296e-05,
+      "loss": 0.0216,
+      "step": 2355
+    },
+    {
+      "epoch": 138.59,
+      "learning_rate": 1.6160990712074303e-05,
+      "loss": 0.0226,
+      "step": 2356
+    },
+    {
+      "epoch": 138.65,
+      "learning_rate": 1.6145510835913313e-05,
+      "loss": 0.0242,
+      "step": 2357
+    },
+    {
+      "epoch": 138.71,
+      "learning_rate": 1.6130030959752323e-05,
+      "loss": 0.0226,
+      "step": 2358
+    },
+    {
+      "epoch": 138.76,
+      "learning_rate": 1.6114551083591333e-05,
+      "loss": 0.0226,
+      "step": 2359
+    },
+    {
+      "epoch": 138.82,
+      "learning_rate": 1.6099071207430343e-05,
+      "loss": 0.0255,
+      "step": 2360
+    },
+    {
+      "epoch": 138.82,
+      "eval_accuracy_NAT": 0.9941619495225278,
+      "eval_accuracy_REST": 0.32195121951219513,
+      "eval_accuracy_concrete": 0.9547120586935817,
+      "eval_accuracy_concrete tiles": 0.9288239453217063,
+      "eval_accuracy_dark bricks": 0.8616273950577111,
+      "eval_accuracy_grass": 0.8915764911603402,
+      "eval_accuracy_red bricks": 0.913829086115993,
+      "eval_accuracy_road": 0.981408731611409,
+      "eval_accuracy_rough concrete": 0.624595546392244,
+      "eval_accuracy_rough red sidewalk": 0.7646320095654201,
+      "eval_accuracy_speedway bricks": 0.9897994890576493,
+      "eval_accuracy_steel": 0.9133751817959188,
+      "eval_accuracy_tiles": 0.24084340634372445,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9832379894118394,
+      "eval_iou_REST": 0.31528662420382164,
+      "eval_iou_concrete": 0.8922952228609967,
+      "eval_iou_concrete tiles": 0.876050326768328,
+      "eval_iou_dark bricks": 0.8445403377110694,
+      "eval_iou_grass": 0.8382691270904616,
+      "eval_iou_red bricks": 0.8958221169376548,
+      "eval_iou_road": 0.9602084166660131,
+      "eval_iou_rough concrete": 0.6179891871495699,
+      "eval_iou_rough red sidewalk": 0.6198299247157792,
+      "eval_iou_speedway bricks": 0.9792335520967602,
+      "eval_iou_steel": 0.864361287027096,
+      "eval_iou_tiles": 0.2385166961687318,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08072586357593536,
+      "eval_mean_accuracy": 0.7985643469346477,
+      "eval_mean_iou": 0.7635108314467787,
+      "eval_overall_accuracy": 0.9798394865908865,
+      "eval_runtime": 5.5814,
+      "eval_samples_per_second": 21.142,
+      "eval_steps_per_second": 1.433,
+      "step": 2360
+    },
+    {
+      "epoch": 138.88,
+      "learning_rate": 1.6083591331269353e-05,
+      "loss": 0.0243,
+      "step": 2361
+    },
+    {
+      "epoch": 138.94,
+      "learning_rate": 1.606811145510836e-05,
+      "loss": 0.0215,
+      "step": 2362
+    },
+    {
+      "epoch": 139.0,
+      "learning_rate": 1.605263157894737e-05,
+      "loss": 0.0236,
+      "step": 2363
+    },
+    {
+      "epoch": 139.06,
+      "learning_rate": 1.603715170278638e-05,
+      "loss": 0.0216,
+      "step": 2364
+    },
+    {
+      "epoch": 139.12,
+      "learning_rate": 1.6021671826625386e-05,
+      "loss": 0.02,
+      "step": 2365
+    },
+    {
+      "epoch": 139.18,
+      "learning_rate": 1.6006191950464396e-05,
+      "loss": 0.0215,
+      "step": 2366
+    },
+    {
+      "epoch": 139.24,
+      "learning_rate": 1.5990712074303406e-05,
+      "loss": 0.0253,
+      "step": 2367
+    },
+    {
+      "epoch": 139.29,
+      "learning_rate": 1.5975232198142416e-05,
+      "loss": 0.0245,
+      "step": 2368
+    },
+    {
+      "epoch": 139.35,
+      "learning_rate": 1.5959752321981426e-05,
+      "loss": 0.0231,
+      "step": 2369
+    },
+    {
+      "epoch": 139.41,
+      "learning_rate": 1.5944272445820436e-05,
+      "loss": 0.025,
+      "step": 2370
+    },
+    {
+      "epoch": 139.47,
+      "learning_rate": 1.5928792569659443e-05,
+      "loss": 0.0221,
+      "step": 2371
+    },
+    {
+      "epoch": 139.53,
+      "learning_rate": 1.5913312693498453e-05,
+      "loss": 0.0223,
+      "step": 2372
+    },
+    {
+      "epoch": 139.59,
+      "learning_rate": 1.5897832817337463e-05,
+      "loss": 0.024,
+      "step": 2373
+    },
+    {
+      "epoch": 139.65,
+      "learning_rate": 1.588235294117647e-05,
+      "loss": 0.0242,
+      "step": 2374
+    },
+    {
+      "epoch": 139.71,
+      "learning_rate": 1.586687306501548e-05,
+      "loss": 0.0218,
+      "step": 2375
+    },
+    {
+      "epoch": 139.76,
+      "learning_rate": 1.585139318885449e-05,
+      "loss": 0.0232,
+      "step": 2376
+    },
+    {
+      "epoch": 139.82,
+      "learning_rate": 1.58359133126935e-05,
+      "loss": 0.0226,
+      "step": 2377
+    },
+    {
+      "epoch": 139.88,
+      "learning_rate": 1.582043343653251e-05,
+      "loss": 0.025,
+      "step": 2378
+    },
+    {
+      "epoch": 139.94,
+      "learning_rate": 1.580495356037152e-05,
+      "loss": 0.0221,
+      "step": 2379
+    },
+    {
+      "epoch": 140.0,
+      "learning_rate": 1.5789473684210526e-05,
+      "loss": 0.025,
+      "step": 2380
+    },
+    {
+      "epoch": 140.0,
+      "eval_accuracy_NAT": 0.9931871710068598,
+      "eval_accuracy_REST": 0.36260162601626017,
+      "eval_accuracy_concrete": 0.9544315244095403,
+      "eval_accuracy_concrete tiles": 0.9284704218713175,
+      "eval_accuracy_dark bricks": 0.8567224317133396,
+      "eval_accuracy_grass": 0.897832002937091,
+      "eval_accuracy_red bricks": 0.9276142355008787,
+      "eval_accuracy_road": 0.9807353672855027,
+      "eval_accuracy_rough concrete": 0.6266155798807048,
+      "eval_accuracy_rough red sidewalk": 0.7963137036674963,
+      "eval_accuracy_speedway bricks": 0.9909522624656156,
+      "eval_accuracy_steel": 0.915901582811477,
+      "eval_accuracy_tiles": 0.23999515283710504,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835667506555436,
+      "eval_iou_REST": 0.35509554140127386,
+      "eval_iou_concrete": 0.8903689890861378,
+      "eval_iou_concrete tiles": 0.8749583564686285,
+      "eval_iou_dark bricks": 0.8393901175411534,
+      "eval_iou_grass": 0.8404643549307153,
+      "eval_iou_red bricks": 0.9007519598954722,
+      "eval_iou_road": 0.9600701655996041,
+      "eval_iou_rough concrete": 0.6176862756615465,
+      "eval_iou_rough red sidewalk": 0.6291171577946768,
+      "eval_iou_speedway bricks": 0.9785274371072907,
+      "eval_iou_steel": 0.8647052023394131,
+      "eval_iou_tiles": 0.2366047428469028,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08112779259681702,
+      "eval_mean_accuracy": 0.8054902355694762,
+      "eval_mean_iou": 0.7670236193329507,
+      "eval_overall_accuracy": 0.9797759298550881,
+      "eval_runtime": 5.558,
+      "eval_samples_per_second": 21.231,
+      "eval_steps_per_second": 1.439,
+      "step": 2380
+    },
+    {
+      "epoch": 140.06,
+      "learning_rate": 1.5773993808049536e-05,
+      "loss": 0.0202,
+      "step": 2381
+    },
+    {
+      "epoch": 140.12,
+      "learning_rate": 1.5758513931888546e-05,
+      "loss": 0.0185,
+      "step": 2382
+    },
+    {
+      "epoch": 140.18,
+      "learning_rate": 1.5743034055727552e-05,
+      "loss": 0.0251,
+      "step": 2383
+    },
+    {
+      "epoch": 140.24,
+      "learning_rate": 1.5727554179566566e-05,
+      "loss": 0.0242,
+      "step": 2384
+    },
+    {
+      "epoch": 140.29,
+      "learning_rate": 1.5712074303405576e-05,
+      "loss": 0.0241,
+      "step": 2385
+    },
+    {
+      "epoch": 140.35,
+      "learning_rate": 1.5696594427244582e-05,
+      "loss": 0.0233,
+      "step": 2386
+    },
+    {
+      "epoch": 140.41,
+      "learning_rate": 1.5681114551083592e-05,
+      "loss": 0.0213,
+      "step": 2387
+    },
+    {
+      "epoch": 140.47,
+      "learning_rate": 1.5665634674922602e-05,
+      "loss": 0.0223,
+      "step": 2388
+    },
+    {
+      "epoch": 140.53,
+      "learning_rate": 1.565015479876161e-05,
+      "loss": 0.0204,
+      "step": 2389
+    },
+    {
+      "epoch": 140.59,
+      "learning_rate": 1.563467492260062e-05,
+      "loss": 0.025,
+      "step": 2390
+    },
+    {
+      "epoch": 140.65,
+      "learning_rate": 1.561919504643963e-05,
+      "loss": 0.0259,
+      "step": 2391
+    },
+    {
+      "epoch": 140.71,
+      "learning_rate": 1.560371517027864e-05,
+      "loss": 0.0241,
+      "step": 2392
+    },
+    {
+      "epoch": 140.76,
+      "learning_rate": 1.558823529411765e-05,
+      "loss": 0.0207,
+      "step": 2393
+    },
+    {
+      "epoch": 140.82,
+      "learning_rate": 1.557275541795666e-05,
+      "loss": 0.0267,
+      "step": 2394
+    },
+    {
+      "epoch": 140.88,
+      "learning_rate": 1.5557275541795666e-05,
+      "loss": 0.0225,
+      "step": 2395
+    },
+    {
+      "epoch": 140.94,
+      "learning_rate": 1.5541795665634676e-05,
+      "loss": 0.0192,
+      "step": 2396
+    },
+    {
+      "epoch": 141.0,
+      "learning_rate": 1.5526315789473686e-05,
+      "loss": 0.0214,
+      "step": 2397
+    },
+    {
+      "epoch": 141.06,
+      "learning_rate": 1.5510835913312692e-05,
+      "loss": 0.0212,
+      "step": 2398
+    },
+    {
+      "epoch": 141.12,
+      "learning_rate": 1.5495356037151702e-05,
+      "loss": 0.0241,
+      "step": 2399
+    },
+    {
+      "epoch": 141.18,
+      "learning_rate": 1.5479876160990712e-05,
+      "loss": 0.0203,
+      "step": 2400
+    },
+    {
+      "epoch": 141.18,
+      "eval_accuracy_NAT": 0.99367189163262,
+      "eval_accuracy_REST": 0.32195121951219513,
+      "eval_accuracy_concrete": 0.9576752498436805,
+      "eval_accuracy_concrete tiles": 0.9230497289653548,
+      "eval_accuracy_dark bricks": 0.874657848898417,
+      "eval_accuracy_grass": 0.8939562539978123,
+      "eval_accuracy_red bricks": 0.9280536028119508,
+      "eval_accuracy_road": 0.9796638629415012,
+      "eval_accuracy_rough concrete": 0.6233024866075951,
+      "eval_accuracy_rough red sidewalk": 0.7674670436685491,
+      "eval_accuracy_speedway bricks": 0.9892506064612889,
+      "eval_accuracy_steel": 0.9159215741152599,
+      "eval_accuracy_tiles": 0.24084340634372445,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834091702867114,
+      "eval_iou_REST": 0.31629392971246006,
+      "eval_iou_concrete": 0.8907013150991123,
+      "eval_iou_concrete tiles": 0.8698694029850746,
+      "eval_iou_dark bricks": 0.8539638191706068,
+      "eval_iou_grass": 0.8388498509217281,
+      "eval_iou_red bricks": 0.8998828416231761,
+      "eval_iou_road": 0.9594483905021755,
+      "eval_iou_rough concrete": 0.6164917223129196,
+      "eval_iou_rough red sidewalk": 0.6187147620491058,
+      "eval_iou_speedway bricks": 0.9793506841399607,
+      "eval_iou_steel": 0.8643264428314728,
+      "eval_iou_tiles": 0.2344717749070961,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08150885254144669,
+      "eval_mean_accuracy": 0.8007280596769193,
+      "eval_mean_iou": 0.7635210851185845,
+      "eval_overall_accuracy": 0.9797950679973021,
+      "eval_runtime": 5.4604,
+      "eval_samples_per_second": 21.61,
+      "eval_steps_per_second": 1.465,
+      "step": 2400
+    },
+    {
+      "epoch": 141.24,
+      "learning_rate": 1.5464396284829722e-05,
+      "loss": 0.0218,
+      "step": 2401
+    },
+    {
+      "epoch": 141.29,
+      "learning_rate": 1.5448916408668732e-05,
+      "loss": 0.0222,
+      "step": 2402
+    },
+    {
+      "epoch": 141.35,
+      "learning_rate": 1.5433436532507742e-05,
+      "loss": 0.0228,
+      "step": 2403
+    },
+    {
+      "epoch": 141.41,
+      "learning_rate": 1.541795665634675e-05,
+      "loss": 0.0218,
+      "step": 2404
+    },
+    {
+      "epoch": 141.47,
+      "learning_rate": 1.540247678018576e-05,
+      "loss": 0.0232,
+      "step": 2405
+    },
+    {
+      "epoch": 141.53,
+      "learning_rate": 1.538699690402477e-05,
+      "loss": 0.0201,
+      "step": 2406
+    },
+    {
+      "epoch": 141.59,
+      "learning_rate": 1.5371517027863776e-05,
+      "loss": 0.022,
+      "step": 2407
+    },
+    {
+      "epoch": 141.65,
+      "learning_rate": 1.535603715170279e-05,
+      "loss": 0.0248,
+      "step": 2408
+    },
+    {
+      "epoch": 141.71,
+      "learning_rate": 1.53405572755418e-05,
+      "loss": 0.0253,
+      "step": 2409
+    },
+    {
+      "epoch": 141.76,
+      "learning_rate": 1.5325077399380806e-05,
+      "loss": 0.0256,
+      "step": 2410
+    },
+    {
+      "epoch": 141.82,
+      "learning_rate": 1.5309597523219816e-05,
+      "loss": 0.022,
+      "step": 2411
+    },
+    {
+      "epoch": 141.88,
+      "learning_rate": 1.5294117647058826e-05,
+      "loss": 0.0244,
+      "step": 2412
+    },
+    {
+      "epoch": 141.94,
+      "learning_rate": 1.5278637770897832e-05,
+      "loss": 0.0241,
+      "step": 2413
+    },
+    {
+      "epoch": 142.0,
+      "learning_rate": 1.5263157894736842e-05,
+      "loss": 0.0218,
+      "step": 2414
+    },
+    {
+      "epoch": 142.06,
+      "learning_rate": 1.5247678018575854e-05,
+      "loss": 0.0238,
+      "step": 2415
+    },
+    {
+      "epoch": 142.12,
+      "learning_rate": 1.523219814241486e-05,
+      "loss": 0.0204,
+      "step": 2416
+    },
+    {
+      "epoch": 142.18,
+      "learning_rate": 1.521671826625387e-05,
+      "loss": 0.0226,
+      "step": 2417
+    },
+    {
+      "epoch": 142.24,
+      "learning_rate": 1.520123839009288e-05,
+      "loss": 0.0263,
+      "step": 2418
+    },
+    {
+      "epoch": 142.29,
+      "learning_rate": 1.5185758513931889e-05,
+      "loss": 0.0237,
+      "step": 2419
+    },
+    {
+      "epoch": 142.35,
+      "learning_rate": 1.5170278637770899e-05,
+      "loss": 0.0245,
+      "step": 2420
+    },
+    {
+      "epoch": 142.35,
+      "eval_accuracy_NAT": 0.9939281881352187,
+      "eval_accuracy_REST": 0.3121951219512195,
+      "eval_accuracy_concrete": 0.9541475503318254,
+      "eval_accuracy_concrete tiles": 0.9387461701626207,
+      "eval_accuracy_dark bricks": 0.869307849855483,
+      "eval_accuracy_grass": 0.8923902470411458,
+      "eval_accuracy_red bricks": 0.9260764499121266,
+      "eval_accuracy_road": 0.982679573172524,
+      "eval_accuracy_rough concrete": 0.6251437560705284,
+      "eval_accuracy_rough red sidewalk": 0.7979605802419931,
+      "eval_accuracy_speedway bricks": 0.9883130676899223,
+      "eval_accuracy_steel": 0.9334189627512032,
+      "eval_accuracy_tiles": 0.24417583083401498,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834206690056343,
+      "eval_iou_REST": 0.3076923076923077,
+      "eval_iou_concrete": 0.8927875090299185,
+      "eval_iou_concrete tiles": 0.8695775570352582,
+      "eval_iou_dark bricks": 0.8510784828154867,
+      "eval_iou_grass": 0.8392537419687828,
+      "eval_iou_red bricks": 0.9012292891501871,
+      "eval_iou_road": 0.9598934026581473,
+      "eval_iou_rough concrete": 0.6196458525291185,
+      "eval_iou_rough red sidewalk": 0.6326735034581445,
+      "eval_iou_speedway bricks": 0.9793491284626138,
+      "eval_iou_steel": 0.8638788862754894,
+      "eval_iou_tiles": 0.24113687359760658,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08004844188690186,
+      "eval_mean_accuracy": 0.8044987190884482,
+      "eval_mean_iou": 0.7647397848983611,
+      "eval_overall_accuracy": 0.9799574512546345,
+      "eval_runtime": 5.8345,
+      "eval_samples_per_second": 20.224,
+      "eval_steps_per_second": 1.371,
+      "step": 2420
+    },
+    {
+      "epoch": 142.41,
+      "learning_rate": 1.5154798761609909e-05,
+      "loss": 0.0228,
+      "step": 2421
+    },
+    {
+      "epoch": 142.47,
+      "learning_rate": 1.5139318885448917e-05,
+      "loss": 0.0232,
+      "step": 2422
+    },
+    {
+      "epoch": 142.53,
+      "learning_rate": 1.5123839009287927e-05,
+      "loss": 0.025,
+      "step": 2423
+    },
+    {
+      "epoch": 142.59,
+      "learning_rate": 1.5108359133126937e-05,
+      "loss": 0.021,
+      "step": 2424
+    },
+    {
+      "epoch": 142.65,
+      "learning_rate": 1.5092879256965944e-05,
+      "loss": 0.0245,
+      "step": 2425
+    },
+    {
+      "epoch": 142.71,
+      "learning_rate": 1.5077399380804955e-05,
+      "loss": 0.0208,
+      "step": 2426
+    },
+    {
+      "epoch": 142.76,
+      "learning_rate": 1.5061919504643965e-05,
+      "loss": 0.0238,
+      "step": 2427
+    },
+    {
+      "epoch": 142.82,
+      "learning_rate": 1.5046439628482972e-05,
+      "loss": 0.0213,
+      "step": 2428
+    },
+    {
+      "epoch": 142.88,
+      "learning_rate": 1.5030959752321982e-05,
+      "loss": 0.0227,
+      "step": 2429
+    },
+    {
+      "epoch": 142.94,
+      "learning_rate": 1.5015479876160992e-05,
+      "loss": 0.0223,
+      "step": 2430
+    },
+    {
+      "epoch": 143.0,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0226,
+      "step": 2431
+    },
+    {
+      "epoch": 143.06,
+      "learning_rate": 1.498452012383901e-05,
+      "loss": 0.028,
+      "step": 2432
+    },
+    {
+      "epoch": 143.12,
+      "learning_rate": 1.496904024767802e-05,
+      "loss": 0.0236,
+      "step": 2433
+    },
+    {
+      "epoch": 143.18,
+      "learning_rate": 1.4953560371517029e-05,
+      "loss": 0.0207,
+      "step": 2434
+    },
+    {
+      "epoch": 143.24,
+      "learning_rate": 1.4938080495356039e-05,
+      "loss": 0.0243,
+      "step": 2435
+    },
+    {
+      "epoch": 143.29,
+      "learning_rate": 1.4922600619195049e-05,
+      "loss": 0.0213,
+      "step": 2436
+    },
+    {
+      "epoch": 143.35,
+      "learning_rate": 1.4907120743034055e-05,
+      "loss": 0.023,
+      "step": 2437
+    },
+    {
+      "epoch": 143.41,
+      "learning_rate": 1.4891640866873065e-05,
+      "loss": 0.0239,
+      "step": 2438
+    },
+    {
+      "epoch": 143.47,
+      "learning_rate": 1.4876160990712077e-05,
+      "loss": 0.0249,
+      "step": 2439
+    },
+    {
+      "epoch": 143.53,
+      "learning_rate": 1.4860681114551084e-05,
+      "loss": 0.021,
+      "step": 2440
+    },
+    {
+      "epoch": 143.53,
+      "eval_accuracy_NAT": 0.9933527879307098,
+      "eval_accuracy_REST": 0.36260162601626017,
+      "eval_accuracy_concrete": 0.9556878134989734,
+      "eval_accuracy_concrete tiles": 0.9448974781993872,
+      "eval_accuracy_dark bricks": 0.8650154087628965,
+      "eval_accuracy_grass": 0.8963341243797477,
+      "eval_accuracy_red bricks": 0.9219573813708261,
+      "eval_accuracy_road": 0.9824300008390333,
+      "eval_accuracy_rough concrete": 0.6283853437334271,
+      "eval_accuracy_rough red sidewalk": 0.796133224042894,
+      "eval_accuracy_speedway bricks": 0.9896418331136432,
+      "eval_accuracy_steel": 0.9209543848425935,
+      "eval_accuracy_tiles": 0.23693538125965646,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835878221190649,
+      "eval_iou_REST": 0.3562300319488818,
+      "eval_iou_concrete": 0.8919868383127242,
+      "eval_iou_concrete tiles": 0.8759258045487317,
+      "eval_iou_dark bricks": 0.8456834885777244,
+      "eval_iou_grass": 0.8395410876347136,
+      "eval_iou_red bricks": 0.8998659876708657,
+      "eval_iou_road": 0.9591424127623807,
+      "eval_iou_rough concrete": 0.6214714236884599,
+      "eval_iou_rough red sidewalk": 0.6309387589692246,
+      "eval_iou_speedway bricks": 0.9793744570027587,
+      "eval_iou_steel": 0.8647201535436582,
+      "eval_iou_tiles": 0.2342247911113773,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.07970796525478363,
+      "eval_mean_accuracy": 0.8072559067684651,
+      "eval_mean_iou": 0.7678994659915822,
+      "eval_overall_accuracy": 0.9799317828679489,
+      "eval_runtime": 5.4521,
+      "eval_samples_per_second": 21.643,
+      "eval_steps_per_second": 1.467,
+      "step": 2440
+    },
+    {
+      "epoch": 143.59,
+      "learning_rate": 1.4845201238390094e-05,
+      "loss": 0.0229,
+      "step": 2441
+    },
+    {
+      "epoch": 143.65,
+      "learning_rate": 1.4829721362229104e-05,
+      "loss": 0.025,
+      "step": 2442
+    },
+    {
+      "epoch": 143.71,
+      "learning_rate": 1.4814241486068112e-05,
+      "loss": 0.0223,
+      "step": 2443
+    },
+    {
+      "epoch": 143.76,
+      "learning_rate": 1.4798761609907122e-05,
+      "loss": 0.0224,
+      "step": 2444
+    },
+    {
+      "epoch": 143.82,
+      "learning_rate": 1.4783281733746132e-05,
+      "loss": 0.02,
+      "step": 2445
+    },
+    {
+      "epoch": 143.88,
+      "learning_rate": 1.476780185758514e-05,
+      "loss": 0.0242,
+      "step": 2446
+    },
+    {
+      "epoch": 143.94,
+      "learning_rate": 1.475232198142415e-05,
+      "loss": 0.0204,
+      "step": 2447
+    },
+    {
+      "epoch": 144.0,
+      "learning_rate": 1.4736842105263157e-05,
+      "loss": 0.0233,
+      "step": 2448
+    },
+    {
+      "epoch": 144.06,
+      "learning_rate": 1.4721362229102167e-05,
+      "loss": 0.0222,
+      "step": 2449
+    },
+    {
+      "epoch": 144.12,
+      "learning_rate": 1.4705882352941177e-05,
+      "loss": 0.0249,
+      "step": 2450
+    },
+    {
+      "epoch": 144.18,
+      "learning_rate": 1.4690402476780185e-05,
+      "loss": 0.0209,
+      "step": 2451
+    },
+    {
+      "epoch": 144.24,
+      "learning_rate": 1.4674922600619195e-05,
+      "loss": 0.0239,
+      "step": 2452
+    },
+    {
+      "epoch": 144.29,
+      "learning_rate": 1.4659442724458205e-05,
+      "loss": 0.0283,
+      "step": 2453
+    },
+    {
+      "epoch": 144.35,
+      "learning_rate": 1.4643962848297213e-05,
+      "loss": 0.0237,
+      "step": 2454
+    },
+    {
+      "epoch": 144.41,
+      "learning_rate": 1.4628482972136224e-05,
+      "loss": 0.025,
+      "step": 2455
+    },
+    {
+      "epoch": 144.47,
+      "learning_rate": 1.4613003095975234e-05,
+      "loss": 0.0219,
+      "step": 2456
+    },
+    {
+      "epoch": 144.53,
+      "learning_rate": 1.459752321981424e-05,
+      "loss": 0.0228,
+      "step": 2457
+    },
+    {
+      "epoch": 144.59,
+      "learning_rate": 1.4582043343653252e-05,
+      "loss": 0.0232,
+      "step": 2458
+    },
+    {
+      "epoch": 144.65,
+      "learning_rate": 1.4566563467492262e-05,
+      "loss": 0.0212,
+      "step": 2459
+    },
+    {
+      "epoch": 144.71,
+      "learning_rate": 1.4551083591331268e-05,
+      "loss": 0.0235,
+      "step": 2460
+    },
+    {
+      "epoch": 144.71,
+      "eval_accuracy_NAT": 0.9942651905411387,
+      "eval_accuracy_REST": 0.2910569105691057,
+      "eval_accuracy_concrete": 0.9581354177973396,
+      "eval_accuracy_concrete tiles": 0.9410558567051615,
+      "eval_accuracy_dark bricks": 0.8602827173018395,
+      "eval_accuracy_grass": 0.8844286865980083,
+      "eval_accuracy_red bricks": 0.9149275043936731,
+      "eval_accuracy_road": 0.9793560392496584,
+      "eval_accuracy_rough concrete": 0.6232071457939804,
+      "eval_accuracy_rough red sidewalk": 0.7764233450394423,
+      "eval_accuracy_speedway bricks": 0.9884074898911694,
+      "eval_accuracy_steel": 0.9206570141988235,
+      "eval_accuracy_tiles": 0.2400254476051986,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9832345635795381,
+      "eval_iou_REST": 0.28685897435897434,
+      "eval_iou_concrete": 0.8909627117053784,
+      "eval_iou_concrete tiles": 0.8680406095784691,
+      "eval_iou_dark bricks": 0.8449621876189716,
+      "eval_iou_grass": 0.8353880886753398,
+      "eval_iou_red bricks": 0.8935793595451376,
+      "eval_iou_road": 0.9600096390086489,
+      "eval_iou_rough concrete": 0.6179051040121943,
+      "eval_iou_rough red sidewalk": 0.6197879773810524,
+      "eval_iou_speedway bricks": 0.9793332679068495,
+      "eval_iou_steel": 0.8658182929122016,
+      "eval_iou_tiles": 0.23686806780471764,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08249124139547348,
+      "eval_mean_accuracy": 0.7978637512065032,
+      "eval_mean_iou": 0.7602114495451903,
+      "eval_overall_accuracy": 0.9797041941497285,
+      "eval_runtime": 5.5302,
+      "eval_samples_per_second": 21.338,
+      "eval_steps_per_second": 1.447,
+      "step": 2460
+    },
+    {
+      "epoch": 144.76,
+      "learning_rate": 1.4535603715170278e-05,
+      "loss": 0.0204,
+      "step": 2461
+    },
+    {
+      "epoch": 144.82,
+      "learning_rate": 1.4520123839009288e-05,
+      "loss": 0.021,
+      "step": 2462
+    },
+    {
+      "epoch": 144.88,
+      "learning_rate": 1.4504643962848297e-05,
+      "loss": 0.024,
+      "step": 2463
+    },
+    {
+      "epoch": 144.94,
+      "learning_rate": 1.4489164086687307e-05,
+      "loss": 0.0207,
+      "step": 2464
+    },
+    {
+      "epoch": 145.0,
+      "learning_rate": 1.4473684210526317e-05,
+      "loss": 0.0201,
+      "step": 2465
+    },
+    {
+      "epoch": 145.06,
+      "learning_rate": 1.4458204334365325e-05,
+      "loss": 0.0256,
+      "step": 2466
+    },
+    {
+      "epoch": 145.12,
+      "learning_rate": 1.4442724458204335e-05,
+      "loss": 0.0243,
+      "step": 2467
+    },
+    {
+      "epoch": 145.18,
+      "learning_rate": 1.4427244582043345e-05,
+      "loss": 0.0219,
+      "step": 2468
+    },
+    {
+      "epoch": 145.24,
+      "learning_rate": 1.4411764705882352e-05,
+      "loss": 0.0224,
+      "step": 2469
+    },
+    {
+      "epoch": 145.29,
+      "learning_rate": 1.4396284829721363e-05,
+      "loss": 0.025,
+      "step": 2470
+    },
+    {
+      "epoch": 145.35,
+      "learning_rate": 1.4380804953560373e-05,
+      "loss": 0.0247,
+      "step": 2471
+    },
+    {
+      "epoch": 145.41,
+      "learning_rate": 1.436532507739938e-05,
+      "loss": 0.0176,
+      "step": 2472
+    },
+    {
+      "epoch": 145.47,
+      "learning_rate": 1.434984520123839e-05,
+      "loss": 0.0207,
+      "step": 2473
+    },
+    {
+      "epoch": 145.53,
+      "learning_rate": 1.43343653250774e-05,
+      "loss": 0.0224,
+      "step": 2474
+    },
+    {
+      "epoch": 145.59,
+      "learning_rate": 1.4318885448916408e-05,
+      "loss": 0.0209,
+      "step": 2475
+    },
+    {
+      "epoch": 145.65,
+      "learning_rate": 1.4303405572755418e-05,
+      "loss": 0.0214,
+      "step": 2476
+    },
+    {
+      "epoch": 145.71,
+      "learning_rate": 1.4287925696594428e-05,
+      "loss": 0.0223,
+      "step": 2477
+    },
+    {
+      "epoch": 145.76,
+      "learning_rate": 1.4272445820433437e-05,
+      "loss": 0.021,
+      "step": 2478
+    },
+    {
+      "epoch": 145.82,
+      "learning_rate": 1.4256965944272447e-05,
+      "loss": 0.0207,
+      "step": 2479
+    },
+    {
+      "epoch": 145.88,
+      "learning_rate": 1.4241486068111457e-05,
+      "loss": 0.0254,
+      "step": 2480
+    },
+    {
+      "epoch": 145.88,
+      "eval_accuracy_NAT": 0.9940816210015199,
+      "eval_accuracy_REST": 0.2845528455284553,
+      "eval_accuracy_concrete": 0.9532226280329426,
+      "eval_accuracy_concrete tiles": 0.9409144473250058,
+      "eval_accuracy_dark bricks": 0.8679966694102559,
+      "eval_accuracy_grass": 0.8930261121014962,
+      "eval_accuracy_red bricks": 0.9247034270650264,
+      "eval_accuracy_road": 0.9797675396710455,
+      "eval_accuracy_rough concrete": 0.6242320595403381,
+      "eval_accuracy_rough red sidewalk": 0.7813790147316494,
+      "eval_accuracy_speedway bricks": 0.9903286646610089,
+      "eval_accuracy_steel": 0.9162139469330841,
+      "eval_accuracy_tiles": 0.24611469599200217,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833423148825381,
+      "eval_iou_REST": 0.2808988764044944,
+      "eval_iou_concrete": 0.8924652843132975,
+      "eval_iou_concrete tiles": 0.8708446034377454,
+      "eval_iou_dark bricks": 0.8467086161343628,
+      "eval_iou_grass": 0.838213717298319,
+      "eval_iou_red bricks": 0.9010489136251739,
+      "eval_iou_road": 0.9606772218394087,
+      "eval_iou_rough concrete": 0.6182601510859301,
+      "eval_iou_rough red sidewalk": 0.6232238716449243,
+      "eval_iou_speedway bricks": 0.9791732393263678,
+      "eval_iou_steel": 0.865265483070867,
+      "eval_iou_tiles": 0.24308069776487837,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08122285455465317,
+      "eval_mean_accuracy": 0.7997333593841408,
+      "eval_mean_iou": 0.7617848454483315,
+      "eval_overall_accuracy": 0.97989305399232,
+      "eval_runtime": 5.4103,
+      "eval_samples_per_second": 21.81,
+      "eval_steps_per_second": 1.479,
+      "step": 2480
+    },
+    {
+      "epoch": 145.94,
+      "learning_rate": 1.4226006191950463e-05,
+      "loss": 0.0216,
+      "step": 2481
+    },
+    {
+      "epoch": 146.0,
+      "learning_rate": 1.4210526315789475e-05,
+      "loss": 0.0244,
+      "step": 2482
+    },
+    {
+      "epoch": 146.06,
+      "learning_rate": 1.4195046439628485e-05,
+      "loss": 0.0249,
+      "step": 2483
+    },
+    {
+      "epoch": 146.12,
+      "learning_rate": 1.4179566563467492e-05,
+      "loss": 0.0197,
+      "step": 2484
+    },
+    {
+      "epoch": 146.18,
+      "learning_rate": 1.4164086687306502e-05,
+      "loss": 0.026,
+      "step": 2485
+    },
+    {
+      "epoch": 146.24,
+      "learning_rate": 1.4148606811145512e-05,
+      "loss": 0.024,
+      "step": 2486
+    },
+    {
+      "epoch": 146.29,
+      "learning_rate": 1.413312693498452e-05,
+      "loss": 0.0224,
+      "step": 2487
+    },
+    {
+      "epoch": 146.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0216,
+      "step": 2488
+    },
+    {
+      "epoch": 146.41,
+      "learning_rate": 1.410216718266254e-05,
+      "loss": 0.0205,
+      "step": 2489
+    },
+    {
+      "epoch": 146.47,
+      "learning_rate": 1.4086687306501548e-05,
+      "loss": 0.0255,
+      "step": 2490
+    },
+    {
+      "epoch": 146.53,
+      "learning_rate": 1.4071207430340558e-05,
+      "loss": 0.0232,
+      "step": 2491
+    },
+    {
+      "epoch": 146.59,
+      "learning_rate": 1.4055727554179568e-05,
+      "loss": 0.0259,
+      "step": 2492
+    },
+    {
+      "epoch": 146.65,
+      "learning_rate": 1.4040247678018575e-05,
+      "loss": 0.0225,
+      "step": 2493
+    },
+    {
+      "epoch": 146.71,
+      "learning_rate": 1.4024767801857586e-05,
+      "loss": 0.0209,
+      "step": 2494
+    },
+    {
+      "epoch": 146.76,
+      "learning_rate": 1.4009287925696596e-05,
+      "loss": 0.0219,
+      "step": 2495
+    },
+    {
+      "epoch": 146.82,
+      "learning_rate": 1.3993808049535603e-05,
+      "loss": 0.0232,
+      "step": 2496
+    },
+    {
+      "epoch": 146.88,
+      "learning_rate": 1.3978328173374613e-05,
+      "loss": 0.0199,
+      "step": 2497
+    },
+    {
+      "epoch": 146.94,
+      "learning_rate": 1.3962848297213623e-05,
+      "loss": 0.0211,
+      "step": 2498
+    },
+    {
+      "epoch": 147.0,
+      "learning_rate": 1.3947368421052631e-05,
+      "loss": 0.0237,
+      "step": 2499
+    },
+    {
+      "epoch": 147.06,
+      "learning_rate": 1.3931888544891641e-05,
+      "loss": 0.0216,
+      "step": 2500
+    },
+    {
+      "epoch": 147.06,
+      "eval_accuracy_NAT": 0.9935082155320963,
+      "eval_accuracy_REST": 0.2926829268292683,
+      "eval_accuracy_concrete": 0.9526069049653804,
+      "eval_accuracy_concrete tiles": 0.9280933301909027,
+      "eval_accuracy_dark bricks": 0.8560955534712784,
+      "eval_accuracy_grass": 0.899503987403816,
+      "eval_accuracy_red bricks": 0.9277240773286467,
+      "eval_accuracy_road": 0.9810020409352068,
+      "eval_accuracy_rough concrete": 0.6234931682348244,
+      "eval_accuracy_rough red sidewalk": 0.7848457275208868,
+      "eval_accuracy_speedway bricks": 0.9912883095375677,
+      "eval_accuracy_steel": 0.907507734135651,
+      "eval_accuracy_tiles": 0.23226998697324971,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834419588562487,
+      "eval_iou_REST": 0.2903225806451613,
+      "eval_iou_concrete": 0.8914125672242745,
+      "eval_iou_concrete tiles": 0.8713710391219684,
+      "eval_iou_dark bricks": 0.8392953484553494,
+      "eval_iou_grass": 0.837533093984714,
+      "eval_iou_red bricks": 0.9033155080213904,
+      "eval_iou_road": 0.9600976598151724,
+      "eval_iou_rough concrete": 0.6180389840519788,
+      "eval_iou_rough red sidewalk": 0.6270231300690898,
+      "eval_iou_speedway bricks": 0.9788321066603133,
+      "eval_iou_steel": 0.862846797358911,
+      "eval_iou_tiles": 0.23051020714951445,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08095356822013855,
+      "eval_mean_accuracy": 0.7977401510045211,
+      "eval_mean_iou": 0.7610800754933913,
+      "eval_overall_accuracy": 0.9797616085763705,
+      "eval_runtime": 5.4495,
+      "eval_samples_per_second": 21.653,
+      "eval_steps_per_second": 1.468,
+      "step": 2500
+    },
+    {
+      "epoch": 147.12,
+      "learning_rate": 1.3916408668730651e-05,
+      "loss": 0.0234,
+      "step": 2501
+    },
+    {
+      "epoch": 147.18,
+      "learning_rate": 1.390092879256966e-05,
+      "loss": 0.0208,
+      "step": 2502
+    },
+    {
+      "epoch": 147.24,
+      "learning_rate": 1.388544891640867e-05,
+      "loss": 0.0221,
+      "step": 2503
+    },
+    {
+      "epoch": 147.29,
+      "learning_rate": 1.386996904024768e-05,
+      "loss": 0.0223,
+      "step": 2504
+    },
+    {
+      "epoch": 147.35,
+      "learning_rate": 1.3854489164086686e-05,
+      "loss": 0.0221,
+      "step": 2505
+    },
+    {
+      "epoch": 147.41,
+      "learning_rate": 1.3839009287925696e-05,
+      "loss": 0.0225,
+      "step": 2506
+    },
+    {
+      "epoch": 147.47,
+      "learning_rate": 1.3823529411764708e-05,
+      "loss": 0.0209,
+      "step": 2507
+    },
+    {
+      "epoch": 147.53,
+      "learning_rate": 1.3808049535603715e-05,
+      "loss": 0.0211,
+      "step": 2508
+    },
+    {
+      "epoch": 147.59,
+      "learning_rate": 1.3792569659442725e-05,
+      "loss": 0.0223,
+      "step": 2509
+    },
+    {
+      "epoch": 147.65,
+      "learning_rate": 1.3777089783281735e-05,
+      "loss": 0.0238,
+      "step": 2510
+    },
+    {
+      "epoch": 147.71,
+      "learning_rate": 1.3761609907120743e-05,
+      "loss": 0.0222,
+      "step": 2511
+    },
+    {
+      "epoch": 147.76,
+      "learning_rate": 1.3746130030959753e-05,
+      "loss": 0.0258,
+      "step": 2512
+    },
+    {
+      "epoch": 147.82,
+      "learning_rate": 1.3730650154798763e-05,
+      "loss": 0.0216,
+      "step": 2513
+    },
+    {
+      "epoch": 147.88,
+      "learning_rate": 1.3715170278637771e-05,
+      "loss": 0.0217,
+      "step": 2514
+    },
+    {
+      "epoch": 147.94,
+      "learning_rate": 1.3699690402476781e-05,
+      "loss": 0.0239,
+      "step": 2515
+    },
+    {
+      "epoch": 148.0,
+      "learning_rate": 1.3684210526315791e-05,
+      "loss": 0.0259,
+      "step": 2516
+    },
+    {
+      "epoch": 148.06,
+      "learning_rate": 1.3668730650154798e-05,
+      "loss": 0.022,
+      "step": 2517
+    },
+    {
+      "epoch": 148.12,
+      "learning_rate": 1.3653250773993808e-05,
+      "loss": 0.0226,
+      "step": 2518
+    },
+    {
+      "epoch": 148.18,
+      "learning_rate": 1.363777089783282e-05,
+      "loss": 0.022,
+      "step": 2519
+    },
+    {
+      "epoch": 148.24,
+      "learning_rate": 1.3622291021671826e-05,
+      "loss": 0.0229,
+      "step": 2520
+    },
+    {
+      "epoch": 148.24,
+      "eval_accuracy_NAT": 0.9935867110532961,
+      "eval_accuracy_REST": 0.3642276422764228,
+      "eval_accuracy_concrete": 0.9541918854502843,
+      "eval_accuracy_concrete tiles": 0.9384162149422578,
+      "eval_accuracy_dark bricks": 0.8601487280592616,
+      "eval_accuracy_grass": 0.8935153118577479,
+      "eval_accuracy_red bricks": 0.9304152021089631,
+      "eval_accuracy_road": 0.9833165972014765,
+      "eval_accuracy_rough concrete": 0.6247802692186224,
+      "eval_accuracy_rough red sidewalk": 0.7976898608050895,
+      "eval_accuracy_speedway bricks": 0.9898164542081456,
+      "eval_accuracy_steel": 0.9154767676060913,
+      "eval_accuracy_tiles": 0.24593292738344089,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834584641418803,
+      "eval_iou_REST": 0.35443037974683544,
+      "eval_iou_concrete": 0.892060157146073,
+      "eval_iou_concrete tiles": 0.8637468002950236,
+      "eval_iou_dark bricks": 0.8429170340079909,
+      "eval_iou_grass": 0.8380164321123731,
+      "eval_iou_red bricks": 0.8966338520165131,
+      "eval_iou_road": 0.9592652585843303,
+      "eval_iou_rough concrete": 0.6181500893178241,
+      "eval_iou_rough red sidewalk": 0.6256879110508154,
+      "eval_iou_speedway bricks": 0.9792214498584502,
+      "eval_iou_steel": 0.8649571358078324,
+      "eval_iou_tiles": 0.2430029634507738,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08096043765544891,
+      "eval_mean_accuracy": 0.8070395824746999,
+      "eval_mean_iou": 0.7662729175028242,
+      "eval_overall_accuracy": 0.9798249066886255,
+      "eval_runtime": 5.3386,
+      "eval_samples_per_second": 22.103,
+      "eval_steps_per_second": 1.499,
+      "step": 2520
+    },
+    {
+      "epoch": 148.29,
+      "learning_rate": 1.3606811145510836e-05,
+      "loss": 0.0229,
+      "step": 2521
+    },
+    {
+      "epoch": 148.35,
+      "learning_rate": 1.3591331269349846e-05,
+      "loss": 0.021,
+      "step": 2522
+    },
+    {
+      "epoch": 148.41,
+      "learning_rate": 1.3575851393188855e-05,
+      "loss": 0.0229,
+      "step": 2523
+    },
+    {
+      "epoch": 148.47,
+      "learning_rate": 1.3560371517027865e-05,
+      "loss": 0.0188,
+      "step": 2524
+    },
+    {
+      "epoch": 148.53,
+      "learning_rate": 1.3544891640866875e-05,
+      "loss": 0.0203,
+      "step": 2525
+    },
+    {
+      "epoch": 148.59,
+      "learning_rate": 1.3529411764705883e-05,
+      "loss": 0.0202,
+      "step": 2526
+    },
+    {
+      "epoch": 148.65,
+      "learning_rate": 1.3513931888544893e-05,
+      "loss": 0.0243,
+      "step": 2527
+    },
+    {
+      "epoch": 148.71,
+      "learning_rate": 1.3498452012383903e-05,
+      "loss": 0.0206,
+      "step": 2528
+    },
+    {
+      "epoch": 148.76,
+      "learning_rate": 1.348297213622291e-05,
+      "loss": 0.0215,
+      "step": 2529
+    },
+    {
+      "epoch": 148.82,
+      "learning_rate": 1.346749226006192e-05,
+      "loss": 0.0278,
+      "step": 2530
+    },
+    {
+      "epoch": 148.88,
+      "learning_rate": 1.3452012383900931e-05,
+      "loss": 0.0271,
+      "step": 2531
+    },
+    {
+      "epoch": 148.94,
+      "learning_rate": 1.3436532507739938e-05,
+      "loss": 0.0238,
+      "step": 2532
+    },
+    {
+      "epoch": 149.0,
+      "learning_rate": 1.3421052631578948e-05,
+      "loss": 0.0257,
+      "step": 2533
+    },
+    {
+      "epoch": 149.06,
+      "learning_rate": 1.3405572755417958e-05,
+      "loss": 0.026,
+      "step": 2534
+    },
+    {
+      "epoch": 149.12,
+      "learning_rate": 1.3390092879256966e-05,
+      "loss": 0.0223,
+      "step": 2535
+    },
+    {
+      "epoch": 149.18,
+      "learning_rate": 1.3374613003095976e-05,
+      "loss": 0.0232,
+      "step": 2536
+    },
+    {
+      "epoch": 149.24,
+      "learning_rate": 1.3359133126934986e-05,
+      "loss": 0.0226,
+      "step": 2537
+    },
+    {
+      "epoch": 149.29,
+      "learning_rate": 1.3343653250773994e-05,
+      "loss": 0.0227,
+      "step": 2538
+    },
+    {
+      "epoch": 149.35,
+      "learning_rate": 1.3328173374613004e-05,
+      "loss": 0.0228,
+      "step": 2539
+    },
+    {
+      "epoch": 149.41,
+      "learning_rate": 1.3312693498452014e-05,
+      "loss": 0.022,
+      "step": 2540
+    },
+    {
+      "epoch": 149.41,
+      "eval_accuracy_NAT": 0.9934164577181664,
+      "eval_accuracy_REST": 0.31382113821138213,
+      "eval_accuracy_concrete": 0.9571764797610184,
+      "eval_accuracy_concrete tiles": 0.9459816167805798,
+      "eval_accuracy_dark bricks": 0.8610818674272152,
+      "eval_accuracy_grass": 0.8923665913469363,
+      "eval_accuracy_red bricks": 0.9240992970123023,
+      "eval_accuracy_road": 0.9814819466214479,
+      "eval_accuracy_rough concrete": 0.6235110446373772,
+      "eval_accuracy_rough red sidewalk": 0.7865452439858925,
+      "eval_accuracy_speedway bricks": 0.9896913576438799,
+      "eval_accuracy_steel": 0.9175633599384267,
+      "eval_accuracy_tiles": 0.23966191038807597,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835405568836345,
+      "eval_iou_REST": 0.3029827315541601,
+      "eval_iou_concrete": 0.8908520857944437,
+      "eval_iou_concrete tiles": 0.8671174577113353,
+      "eval_iou_dark bricks": 0.8420269441883753,
+      "eval_iou_grass": 0.837845449400719,
+      "eval_iou_red bricks": 0.8953333688075347,
+      "eval_iou_road": 0.9596578454399858,
+      "eval_iou_rough concrete": 0.6181516358093978,
+      "eval_iou_rough red sidewalk": 0.6212225601064335,
+      "eval_iou_speedway bricks": 0.9791889938439299,
+      "eval_iou_steel": 0.8651942996635218,
+      "eval_iou_tiles": 0.23766748783272246,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.081676185131073,
+      "eval_mean_accuracy": 0.802030639344054,
+      "eval_mean_iou": 0.7615985705412457,
+      "eval_overall_accuracy": 0.9797789040258376,
+      "eval_runtime": 5.5724,
+      "eval_samples_per_second": 21.176,
+      "eval_steps_per_second": 1.436,
+      "step": 2540
+    },
+    {
+      "epoch": 149.47,
+      "learning_rate": 1.3297213622291021e-05,
+      "loss": 0.0248,
+      "step": 2541
+    },
+    {
+      "epoch": 149.53,
+      "learning_rate": 1.3281733746130031e-05,
+      "loss": 0.0217,
+      "step": 2542
+    },
+    {
+      "epoch": 149.59,
+      "learning_rate": 1.3266253869969043e-05,
+      "loss": 0.0236,
+      "step": 2543
+    },
+    {
+      "epoch": 149.65,
+      "learning_rate": 1.325077399380805e-05,
+      "loss": 0.0236,
+      "step": 2544
+    },
+    {
+      "epoch": 149.71,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.02,
+      "step": 2545
+    },
+    {
+      "epoch": 149.76,
+      "learning_rate": 1.321981424148607e-05,
+      "loss": 0.0251,
+      "step": 2546
+    },
+    {
+      "epoch": 149.82,
+      "learning_rate": 1.3204334365325078e-05,
+      "loss": 0.0231,
+      "step": 2547
+    },
+    {
+      "epoch": 149.88,
+      "learning_rate": 1.3188854489164088e-05,
+      "loss": 0.0191,
+      "step": 2548
+    },
+    {
+      "epoch": 149.94,
+      "learning_rate": 1.3173374613003098e-05,
+      "loss": 0.0224,
+      "step": 2549
+    },
+    {
+      "epoch": 150.0,
+      "learning_rate": 1.3157894736842106e-05,
+      "loss": 0.0185,
+      "step": 2550
+    },
+    {
+      "epoch": 150.06,
+      "learning_rate": 1.3142414860681116e-05,
+      "loss": 0.0262,
+      "step": 2551
+    },
+    {
+      "epoch": 150.12,
+      "learning_rate": 1.3126934984520126e-05,
+      "loss": 0.026,
+      "step": 2552
+    },
+    {
+      "epoch": 150.18,
+      "learning_rate": 1.3111455108359133e-05,
+      "loss": 0.0269,
+      "step": 2553
+    },
+    {
+      "epoch": 150.24,
+      "learning_rate": 1.3095975232198143e-05,
+      "loss": 0.02,
+      "step": 2554
+    },
+    {
+      "epoch": 150.29,
+      "learning_rate": 1.3080495356037154e-05,
+      "loss": 0.022,
+      "step": 2555
+    },
+    {
+      "epoch": 150.35,
+      "learning_rate": 1.3065015479876161e-05,
+      "loss": 0.0197,
+      "step": 2556
+    },
+    {
+      "epoch": 150.41,
+      "learning_rate": 1.3049535603715171e-05,
+      "loss": 0.0205,
+      "step": 2557
+    },
+    {
+      "epoch": 150.47,
+      "learning_rate": 1.3034055727554181e-05,
+      "loss": 0.0189,
+      "step": 2558
+    },
+    {
+      "epoch": 150.53,
+      "learning_rate": 1.301857585139319e-05,
+      "loss": 0.0218,
+      "step": 2559
+    },
+    {
+      "epoch": 150.59,
+      "learning_rate": 1.30030959752322e-05,
+      "loss": 0.0235,
+      "step": 2560
+    },
+    {
+      "epoch": 150.59,
+      "eval_accuracy_NAT": 0.9937834350621271,
+      "eval_accuracy_REST": 0.26666666666666666,
+      "eval_accuracy_concrete": 0.9538612830583282,
+      "eval_accuracy_concrete tiles": 0.9397360358237097,
+      "eval_accuracy_dark bricks": 0.8590098194973489,
+      "eval_accuracy_grass": 0.8940414144969664,
+      "eval_accuracy_red bricks": 0.9200351493848857,
+      "eval_accuracy_road": 0.9817635839228389,
+      "eval_accuracy_rough concrete": 0.6240056251080033,
+      "eval_accuracy_rough red sidewalk": 0.7736334308424638,
+      "eval_accuracy_speedway bricks": 0.9896928999302886,
+      "eval_accuracy_steel": 0.922898539135476,
+      "eval_accuracy_tiles": 0.24278227150171167,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9833997743893802,
+      "eval_iou_REST": 0.2640901771336554,
+      "eval_iou_concrete": 0.8915905192031649,
+      "eval_iou_concrete tiles": 0.8687685201324734,
+      "eval_iou_dark bricks": 0.8424686143376745,
+      "eval_iou_grass": 0.8377931195922617,
+      "eval_iou_red bricks": 0.899194847020934,
+      "eval_iou_road": 0.9601298864878227,
+      "eval_iou_rough concrete": 0.617613059991979,
+      "eval_iou_rough red sidewalk": 0.6176721342971733,
+      "eval_iou_speedway bricks": 0.9790263550017206,
+      "eval_iou_steel": 0.8659644910477298,
+      "eval_iou_tiles": 0.24024942290973408,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08168569207191467,
+      "eval_mean_accuracy": 0.7970700118792935,
+      "eval_mean_iou": 0.7590739170419772,
+      "eval_overall_accuracy": 0.9797867273880263,
+      "eval_runtime": 5.5353,
+      "eval_samples_per_second": 21.318,
+      "eval_steps_per_second": 1.445,
+      "step": 2560
+    },
+    {
+      "epoch": 150.65,
+      "learning_rate": 1.298761609907121e-05,
+      "loss": 0.0233,
+      "step": 2561
+    },
+    {
+      "epoch": 150.71,
+      "learning_rate": 1.2972136222910216e-05,
+      "loss": 0.0216,
+      "step": 2562
+    },
+    {
+      "epoch": 150.76,
+      "learning_rate": 1.2956656346749228e-05,
+      "loss": 0.0191,
+      "step": 2563
+    },
+    {
+      "epoch": 150.82,
+      "learning_rate": 1.2941176470588238e-05,
+      "loss": 0.0228,
+      "step": 2564
+    },
+    {
+      "epoch": 150.88,
+      "learning_rate": 1.2925696594427244e-05,
+      "loss": 0.0192,
+      "step": 2565
+    },
+    {
+      "epoch": 150.94,
+      "learning_rate": 1.2910216718266254e-05,
+      "loss": 0.0234,
+      "step": 2566
+    },
+    {
+      "epoch": 151.0,
+      "learning_rate": 1.2894736842105264e-05,
+      "loss": 0.0218,
+      "step": 2567
+    },
+    {
+      "epoch": 151.06,
+      "learning_rate": 1.2879256965944272e-05,
+      "loss": 0.025,
+      "step": 2568
+    },
+    {
+      "epoch": 151.12,
+      "learning_rate": 1.2863777089783282e-05,
+      "loss": 0.0207,
+      "step": 2569
+    },
+    {
+      "epoch": 151.18,
+      "learning_rate": 1.2848297213622292e-05,
+      "loss": 0.0222,
+      "step": 2570
+    },
+    {
+      "epoch": 151.24,
+      "learning_rate": 1.28328173374613e-05,
+      "loss": 0.0197,
+      "step": 2571
+    },
+    {
+      "epoch": 151.29,
+      "learning_rate": 1.281733746130031e-05,
+      "loss": 0.02,
+      "step": 2572
+    },
+    {
+      "epoch": 151.35,
+      "learning_rate": 1.280185758513932e-05,
+      "loss": 0.0218,
+      "step": 2573
+    },
+    {
+      "epoch": 151.41,
+      "learning_rate": 1.2786377708978327e-05,
+      "loss": 0.0225,
+      "step": 2574
+    },
+    {
+      "epoch": 151.47,
+      "learning_rate": 1.2770897832817339e-05,
+      "loss": 0.0277,
+      "step": 2575
+    },
+    {
+      "epoch": 151.53,
+      "learning_rate": 1.2755417956656349e-05,
+      "loss": 0.021,
+      "step": 2576
+    },
+    {
+      "epoch": 151.59,
+      "learning_rate": 1.2739938080495356e-05,
+      "loss": 0.0206,
+      "step": 2577
+    },
+    {
+      "epoch": 151.65,
+      "learning_rate": 1.2724458204334366e-05,
+      "loss": 0.0231,
+      "step": 2578
+    },
+    {
+      "epoch": 151.71,
+      "learning_rate": 1.2708978328173376e-05,
+      "loss": 0.0261,
+      "step": 2579
+    },
+    {
+      "epoch": 151.76,
+      "learning_rate": 1.2693498452012384e-05,
+      "loss": 0.0224,
+      "step": 2580
+    },
+    {
+      "epoch": 151.76,
+      "eval_accuracy_NAT": 0.9938005250897509,
+      "eval_accuracy_REST": 0.35609756097560974,
+      "eval_accuracy_concrete": 0.9555158238152969,
+      "eval_accuracy_concrete tiles": 0.9403252415743578,
+      "eval_accuracy_dark bricks": 0.8660586107229676,
+      "eval_accuracy_grass": 0.8891806424508056,
+      "eval_accuracy_red bricks": 0.9244288224956063,
+      "eval_accuracy_road": 0.9814386589147825,
+      "eval_accuracy_rough concrete": 0.6237732318748175,
+      "eval_accuracy_rough red sidewalk": 0.7723851134389641,
+      "eval_accuracy_speedway bricks": 0.9899161887292451,
+      "eval_accuracy_steel": 0.9213692043960877,
+      "eval_accuracy_tiles": 0.24532703202156989,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834206731119332,
+      "eval_iou_REST": 0.347068145800317,
+      "eval_iou_concrete": 0.8912362431304774,
+      "eval_iou_concrete tiles": 0.868839964286492,
+      "eval_iou_dark bricks": 0.847012463085772,
+      "eval_iou_grass": 0.8374223697964706,
+      "eval_iou_red bricks": 0.8983774551665243,
+      "eval_iou_road": 0.960038412639996,
+      "eval_iou_rough concrete": 0.6176015953178836,
+      "eval_iou_rough red sidewalk": 0.6209254243845819,
+      "eval_iou_speedway bricks": 0.9792905652649543,
+      "eval_iou_steel": 0.8657576447770376,
+      "eval_iou_tiles": 0.23977023746076864,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08196701109409332,
+      "eval_mean_accuracy": 0.8045858966538355,
+      "eval_mean_iou": 0.7659047072479391,
+      "eval_overall_accuracy": 0.97982241743702,
+      "eval_runtime": 5.4549,
+      "eval_samples_per_second": 21.632,
+      "eval_steps_per_second": 1.467,
+      "step": 2580
+    },
+    {
+      "epoch": 151.82,
+      "learning_rate": 1.2678018575851394e-05,
+      "loss": 0.0194,
+      "step": 2581
+    },
+    {
+      "epoch": 151.88,
+      "learning_rate": 1.2662538699690404e-05,
+      "loss": 0.0214,
+      "step": 2582
+    },
+    {
+      "epoch": 151.94,
+      "learning_rate": 1.2647058823529412e-05,
+      "loss": 0.0275,
+      "step": 2583
+    },
+    {
+      "epoch": 152.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0244,
+      "step": 2584
+    },
+    {
+      "epoch": 152.06,
+      "learning_rate": 1.2616099071207432e-05,
+      "loss": 0.0196,
+      "step": 2585
+    },
+    {
+      "epoch": 152.12,
+      "learning_rate": 1.2600619195046439e-05,
+      "loss": 0.0212,
+      "step": 2586
+    },
+    {
+      "epoch": 152.18,
+      "learning_rate": 1.258513931888545e-05,
+      "loss": 0.0233,
+      "step": 2587
+    },
+    {
+      "epoch": 152.24,
+      "learning_rate": 1.256965944272446e-05,
+      "loss": 0.0213,
+      "step": 2588
+    },
+    {
+      "epoch": 152.29,
+      "learning_rate": 1.2554179566563467e-05,
+      "loss": 0.0273,
+      "step": 2589
+    },
+    {
+      "epoch": 152.35,
+      "learning_rate": 1.2538699690402477e-05,
+      "loss": 0.024,
+      "step": 2590
+    },
+    {
+      "epoch": 152.41,
+      "learning_rate": 1.2523219814241487e-05,
+      "loss": 0.0212,
+      "step": 2591
+    },
+    {
+      "epoch": 152.47,
+      "learning_rate": 1.2507739938080496e-05,
+      "loss": 0.0238,
+      "step": 2592
+    },
+    {
+      "epoch": 152.53,
+      "learning_rate": 1.2492260061919506e-05,
+      "loss": 0.0253,
+      "step": 2593
+    },
+    {
+      "epoch": 152.59,
+      "learning_rate": 1.2476780185758514e-05,
+      "loss": 0.0206,
+      "step": 2594
+    },
+    {
+      "epoch": 152.65,
+      "learning_rate": 1.2461300309597524e-05,
+      "loss": 0.0221,
+      "step": 2595
+    },
+    {
+      "epoch": 152.71,
+      "learning_rate": 1.2445820433436534e-05,
+      "loss": 0.0198,
+      "step": 2596
+    },
+    {
+      "epoch": 152.76,
+      "learning_rate": 1.2430340557275542e-05,
+      "loss": 0.022,
+      "step": 2597
+    },
+    {
+      "epoch": 152.82,
+      "learning_rate": 1.241486068111455e-05,
+      "loss": 0.0251,
+      "step": 2598
+    },
+    {
+      "epoch": 152.88,
+      "learning_rate": 1.2399380804953562e-05,
+      "loss": 0.0216,
+      "step": 2599
+    },
+    {
+      "epoch": 152.94,
+      "learning_rate": 1.238390092879257e-05,
+      "loss": 0.0205,
+      "step": 2600
+    },
+    {
+      "epoch": 152.94,
+      "eval_accuracy_NAT": 0.9938552455252027,
+      "eval_accuracy_REST": 0.2943089430894309,
+      "eval_accuracy_concrete": 0.9546100114812669,
+      "eval_accuracy_concrete tiles": 0.9375441904312987,
+      "eval_accuracy_dark bricks": 0.8546838810941179,
+      "eval_accuracy_grass": 0.8965782511439894,
+      "eval_accuracy_red bricks": 0.9192662565905096,
+      "eval_accuracy_road": 0.9801304082244505,
+      "eval_accuracy_rough concrete": 0.6246372579982005,
+      "eval_accuracy_rough red sidewalk": 0.7737988704983494,
+      "eval_accuracy_speedway bricks": 0.9904212018455342,
+      "eval_accuracy_steel": 0.9152018871790771,
+      "eval_accuracy_tiles": 0.23784422430246296,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834703623329354,
+      "eval_iou_REST": 0.2896,
+      "eval_iou_concrete": 0.8917698035709695,
+      "eval_iou_concrete tiles": 0.8715083798882681,
+      "eval_iou_dark bricks": 0.8391593606404871,
+      "eval_iou_grass": 0.8382882986497557,
+      "eval_iou_red bricks": 0.898443370907139,
+      "eval_iou_road": 0.9610566891190135,
+      "eval_iou_rough concrete": 0.6184643704201919,
+      "eval_iou_rough red sidewalk": 0.6205576026583522,
+      "eval_iou_speedway bricks": 0.9793189934945279,
+      "eval_iou_steel": 0.865692809530563,
+      "eval_iou_tiles": 0.23589327564449253,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.081034816801548,
+      "eval_mean_accuracy": 0.7979138945695301,
+      "eval_mean_iou": 0.7610171782197459,
+      "eval_overall_accuracy": 0.9798855539095603,
+      "eval_runtime": 5.6698,
+      "eval_samples_per_second": 20.812,
+      "eval_steps_per_second": 1.411,
+      "step": 2600
+    },
+    {
+      "epoch": 153.0,
+      "learning_rate": 1.2368421052631579e-05,
+      "loss": 0.0213,
+      "step": 2601
+    },
+    {
+      "epoch": 153.06,
+      "learning_rate": 1.2352941176470589e-05,
+      "loss": 0.023,
+      "step": 2602
+    },
+    {
+      "epoch": 153.12,
+      "learning_rate": 1.2337461300309599e-05,
+      "loss": 0.0251,
+      "step": 2603
+    },
+    {
+      "epoch": 153.18,
+      "learning_rate": 1.2321981424148607e-05,
+      "loss": 0.0208,
+      "step": 2604
+    },
+    {
+      "epoch": 153.24,
+      "learning_rate": 1.2306501547987617e-05,
+      "loss": 0.022,
+      "step": 2605
+    },
+    {
+      "epoch": 153.29,
+      "learning_rate": 1.2291021671826625e-05,
+      "loss": 0.0218,
+      "step": 2606
+    },
+    {
+      "epoch": 153.35,
+      "learning_rate": 1.2275541795665635e-05,
+      "loss": 0.0231,
+      "step": 2607
+    },
+    {
+      "epoch": 153.41,
+      "learning_rate": 1.2260061919504645e-05,
+      "loss": 0.0233,
+      "step": 2608
+    },
+    {
+      "epoch": 153.47,
+      "learning_rate": 1.2244582043343654e-05,
+      "loss": 0.0244,
+      "step": 2609
+    },
+    {
+      "epoch": 153.53,
+      "learning_rate": 1.2229102167182662e-05,
+      "loss": 0.0224,
+      "step": 2610
+    },
+    {
+      "epoch": 153.59,
+      "learning_rate": 1.2213622291021674e-05,
+      "loss": 0.0216,
+      "step": 2611
+    },
+    {
+      "epoch": 153.65,
+      "learning_rate": 1.2198142414860682e-05,
+      "loss": 0.0253,
+      "step": 2612
+    },
+    {
+      "epoch": 153.71,
+      "learning_rate": 1.218266253869969e-05,
+      "loss": 0.0206,
+      "step": 2613
+    },
+    {
+      "epoch": 153.76,
+      "learning_rate": 1.21671826625387e-05,
+      "loss": 0.0224,
+      "step": 2614
+    },
+    {
+      "epoch": 153.82,
+      "learning_rate": 1.215170278637771e-05,
+      "loss": 0.0172,
+      "step": 2615
+    },
+    {
+      "epoch": 153.88,
+      "learning_rate": 1.2136222910216719e-05,
+      "loss": 0.0211,
+      "step": 2616
+    },
+    {
+      "epoch": 153.94,
+      "learning_rate": 1.2120743034055729e-05,
+      "loss": 0.0219,
+      "step": 2617
+    },
+    {
+      "epoch": 154.0,
+      "learning_rate": 1.2105263157894737e-05,
+      "loss": 0.0204,
+      "step": 2618
+    },
+    {
+      "epoch": 154.06,
+      "learning_rate": 1.2089783281733747e-05,
+      "loss": 0.0221,
+      "step": 2619
+    },
+    {
+      "epoch": 154.12,
+      "learning_rate": 1.2074303405572757e-05,
+      "loss": 0.0197,
+      "step": 2620
+    },
+    {
+      "epoch": 154.12,
+      "eval_accuracy_NAT": 0.9929238120638235,
+      "eval_accuracy_REST": 0.3382113821138211,
+      "eval_accuracy_concrete": 0.9570392702133742,
+      "eval_accuracy_concrete tiles": 0.9462172990808391,
+      "eval_accuracy_dark bricks": 0.8656422870049576,
+      "eval_accuracy_grass": 0.9035689818967704,
+      "eval_accuracy_red bricks": 0.9324472759226714,
+      "eval_accuracy_road": 0.9827522537664313,
+      "eval_accuracy_rough concrete": 0.6258171005666819,
+      "eval_accuracy_rough red sidewalk": 0.7920874724580573,
+      "eval_accuracy_speedway bricks": 0.9891049460782398,
+      "eval_accuracy_steel": 0.9219064706852519,
+      "eval_accuracy_tiles": 0.23890454118573723,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983616620546666,
+      "eval_iou_REST": 0.33226837060702874,
+      "eval_iou_concrete": 0.8919103271139055,
+      "eval_iou_concrete tiles": 0.8673897074709416,
+      "eval_iou_dark bricks": 0.8474580241361217,
+      "eval_iou_grass": 0.8405042055268616,
+      "eval_iou_red bricks": 0.9058315104305608,
+      "eval_iou_road": 0.9603900414200968,
+      "eval_iou_rough concrete": 0.6170514033242658,
+      "eval_iou_rough red sidewalk": 0.6254698550500882,
+      "eval_iou_speedway bricks": 0.9794341223167187,
+      "eval_iou_steel": 0.8655552088441136,
+      "eval_iou_tiles": 0.23581125530769692,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08082769811153412,
+      "eval_mean_accuracy": 0.8066633148489737,
+      "eval_mean_iou": 0.7655915886226973,
+      "eval_overall_accuracy": 0.979949078317416,
+      "eval_runtime": 5.6194,
+      "eval_samples_per_second": 20.999,
+      "eval_steps_per_second": 1.424,
+      "step": 2620
+    },
+    {
+      "epoch": 154.18,
+      "learning_rate": 1.2058823529411765e-05,
+      "loss": 0.0246,
+      "step": 2621
+    },
+    {
+      "epoch": 154.24,
+      "learning_rate": 1.2043343653250774e-05,
+      "loss": 0.0219,
+      "step": 2622
+    },
+    {
+      "epoch": 154.29,
+      "learning_rate": 1.2027863777089785e-05,
+      "loss": 0.0179,
+      "step": 2623
+    },
+    {
+      "epoch": 154.35,
+      "learning_rate": 1.2012383900928794e-05,
+      "loss": 0.0212,
+      "step": 2624
+    },
+    {
+      "epoch": 154.41,
+      "learning_rate": 1.1996904024767802e-05,
+      "loss": 0.0216,
+      "step": 2625
+    },
+    {
+      "epoch": 154.47,
+      "learning_rate": 1.1981424148606812e-05,
+      "loss": 0.023,
+      "step": 2626
+    },
+    {
+      "epoch": 154.53,
+      "learning_rate": 1.1965944272445822e-05,
+      "loss": 0.0258,
+      "step": 2627
+    },
+    {
+      "epoch": 154.59,
+      "learning_rate": 1.195046439628483e-05,
+      "loss": 0.022,
+      "step": 2628
+    },
+    {
+      "epoch": 154.65,
+      "learning_rate": 1.193498452012384e-05,
+      "loss": 0.0234,
+      "step": 2629
+    },
+    {
+      "epoch": 154.71,
+      "learning_rate": 1.1919504643962849e-05,
+      "loss": 0.0188,
+      "step": 2630
+    },
+    {
+      "epoch": 154.76,
+      "learning_rate": 1.1904024767801859e-05,
+      "loss": 0.0227,
+      "step": 2631
+    },
+    {
+      "epoch": 154.82,
+      "learning_rate": 1.1888544891640867e-05,
+      "loss": 0.0241,
+      "step": 2632
+    },
+    {
+      "epoch": 154.88,
+      "learning_rate": 1.1873065015479877e-05,
+      "loss": 0.0251,
+      "step": 2633
+    },
+    {
+      "epoch": 154.94,
+      "learning_rate": 1.1857585139318885e-05,
+      "loss": 0.0268,
+      "step": 2634
+    },
+    {
+      "epoch": 155.0,
+      "learning_rate": 1.1842105263157895e-05,
+      "loss": 0.0189,
+      "step": 2635
+    },
+    {
+      "epoch": 155.06,
+      "learning_rate": 1.1826625386996905e-05,
+      "loss": 0.0216,
+      "step": 2636
+    },
+    {
+      "epoch": 155.12,
+      "learning_rate": 1.1811145510835914e-05,
+      "loss": 0.0197,
+      "step": 2637
+    },
+    {
+      "epoch": 155.18,
+      "learning_rate": 1.1795665634674922e-05,
+      "loss": 0.02,
+      "step": 2638
+    },
+    {
+      "epoch": 155.24,
+      "learning_rate": 1.1780185758513934e-05,
+      "loss": 0.0233,
+      "step": 2639
+    },
+    {
+      "epoch": 155.29,
+      "learning_rate": 1.1764705882352942e-05,
+      "loss": 0.0209,
+      "step": 2640
+    },
+    {
+      "epoch": 155.29,
+      "eval_accuracy_NAT": 0.994252305630722,
+      "eval_accuracy_REST": 0.24227642276422764,
+      "eval_accuracy_concrete": 0.9555272897942086,
+      "eval_accuracy_concrete tiles": 0.9340325241574358,
+      "eval_accuracy_dark bricks": 0.8575455084891755,
+      "eval_accuracy_grass": 0.8938559538543642,
+      "eval_accuracy_red bricks": 0.9253624780316344,
+      "eval_accuracy_road": 0.9817170897193834,
+      "eval_accuracy_rough concrete": 0.6239698723028978,
+      "eval_accuracy_rough red sidewalk": 0.7637596913798419,
+      "eval_accuracy_speedway bricks": 0.9885651458351755,
+      "eval_accuracy_steel": 0.9252000379834772,
+      "eval_accuracy_tiles": 0.24426671513829562,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834581839192446,
+      "eval_iou_REST": 0.23993558776167473,
+      "eval_iou_concrete": 0.8926015223034703,
+      "eval_iou_concrete tiles": 0.8722570705403323,
+      "eval_iou_dark bricks": 0.842123318248676,
+      "eval_iou_grass": 0.8397929347840581,
+      "eval_iou_red bricks": 0.9026572377584914,
+      "eval_iou_road": 0.9609784142252782,
+      "eval_iou_rough concrete": 0.6175339686733345,
+      "eval_iou_rough red sidewalk": 0.6173427831605053,
+      "eval_iou_speedway bricks": 0.9793358306937662,
+      "eval_iou_steel": 0.8656578575119418,
+      "eval_iou_tiles": 0.2418996759870395,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08118943125009537,
+      "eval_mean_accuracy": 0.7946408488523722,
+      "eval_mean_iou": 0.7581211065821395,
+      "eval_overall_accuracy": 0.9799605547371557,
+      "eval_runtime": 5.4414,
+      "eval_samples_per_second": 21.686,
+      "eval_steps_per_second": 1.47,
+      "step": 2640
+    },
+    {
+      "epoch": 155.35,
+      "learning_rate": 1.174922600619195e-05,
+      "loss": 0.0206,
+      "step": 2641
+    },
+    {
+      "epoch": 155.41,
+      "learning_rate": 1.173374613003096e-05,
+      "loss": 0.0219,
+      "step": 2642
+    },
+    {
+      "epoch": 155.47,
+      "learning_rate": 1.171826625386997e-05,
+      "loss": 0.0212,
+      "step": 2643
+    },
+    {
+      "epoch": 155.53,
+      "learning_rate": 1.1702786377708978e-05,
+      "loss": 0.0237,
+      "step": 2644
+    },
+    {
+      "epoch": 155.59,
+      "learning_rate": 1.1687306501547988e-05,
+      "loss": 0.0248,
+      "step": 2645
+    },
+    {
+      "epoch": 155.65,
+      "learning_rate": 1.1671826625386997e-05,
+      "loss": 0.0241,
+      "step": 2646
+    },
+    {
+      "epoch": 155.71,
+      "learning_rate": 1.1656346749226007e-05,
+      "loss": 0.0239,
+      "step": 2647
+    },
+    {
+      "epoch": 155.76,
+      "learning_rate": 1.1640866873065017e-05,
+      "loss": 0.0246,
+      "step": 2648
+    },
+    {
+      "epoch": 155.82,
+      "learning_rate": 1.1625386996904025e-05,
+      "loss": 0.0204,
+      "step": 2649
+    },
+    {
+      "epoch": 155.88,
+      "learning_rate": 1.1609907120743033e-05,
+      "loss": 0.0191,
+      "step": 2650
+    },
+    {
+      "epoch": 155.94,
+      "learning_rate": 1.1594427244582045e-05,
+      "loss": 0.0248,
+      "step": 2651
+    },
+    {
+      "epoch": 156.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 0.0219,
+      "step": 2652
+    },
+    {
+      "epoch": 156.06,
+      "learning_rate": 1.1563467492260062e-05,
+      "loss": 0.0237,
+      "step": 2653
+    },
+    {
+      "epoch": 156.12,
+      "learning_rate": 1.1547987616099072e-05,
+      "loss": 0.0216,
+      "step": 2654
+    },
+    {
+      "epoch": 156.18,
+      "learning_rate": 1.1532507739938082e-05,
+      "loss": 0.0244,
+      "step": 2655
+    },
+    {
+      "epoch": 156.24,
+      "learning_rate": 1.151702786377709e-05,
+      "loss": 0.0252,
+      "step": 2656
+    },
+    {
+      "epoch": 156.29,
+      "learning_rate": 1.15015479876161e-05,
+      "loss": 0.0222,
+      "step": 2657
+    },
+    {
+      "epoch": 156.35,
+      "learning_rate": 1.1486068111455108e-05,
+      "loss": 0.0187,
+      "step": 2658
+    },
+    {
+      "epoch": 156.41,
+      "learning_rate": 1.1470588235294118e-05,
+      "loss": 0.0231,
+      "step": 2659
+    },
+    {
+      "epoch": 156.47,
+      "learning_rate": 1.1455108359133128e-05,
+      "loss": 0.0203,
+      "step": 2660
+    },
+    {
+      "epoch": 156.47,
+      "eval_accuracy_NAT": 0.9940302430951301,
+      "eval_accuracy_REST": 0.34959349593495936,
+      "eval_accuracy_concrete": 0.9558009444909029,
+      "eval_accuracy_concrete tiles": 0.9386518972425171,
+      "eval_accuracy_dark bricks": 0.8629002928622016,
+      "eval_accuracy_grass": 0.8938739321819634,
+      "eval_accuracy_red bricks": 0.9250878734622144,
+      "eval_accuracy_road": 0.9807946874761183,
+      "eval_accuracy_rough concrete": 0.6244823291760766,
+      "eval_accuracy_rough red sidewalk": 0.77346047120222,
+      "eval_accuracy_speedway bricks": 0.9892427236640886,
+      "eval_accuracy_steel": 0.9230934543473589,
+      "eval_accuracy_tiles": 0.24987124723560242,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834143702762304,
+      "eval_iou_REST": 0.34290271132376393,
+      "eval_iou_concrete": 0.8925630217609332,
+      "eval_iou_concrete tiles": 0.8694712483080819,
+      "eval_iou_dark bricks": 0.8462802004918433,
+      "eval_iou_grass": 0.8399726135802853,
+      "eval_iou_red bricks": 0.9049588996937624,
+      "eval_iou_road": 0.9607700969688531,
+      "eval_iou_rough concrete": 0.617746052142954,
+      "eval_iou_rough red sidewalk": 0.6219779156537619,
+      "eval_iou_speedway bricks": 0.9795577537998109,
+      "eval_iou_steel": 0.8663931907787495,
+      "eval_iou_tiles": 0.24600334049152947,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08094289898872375,
+      "eval_mean_accuracy": 0.8046833532593349,
+      "eval_mean_iou": 0.7670778011746585,
+      "eval_overall_accuracy": 0.9799908783476231,
+      "eval_runtime": 5.3629,
+      "eval_samples_per_second": 22.003,
+      "eval_steps_per_second": 1.492,
+      "step": 2660
+    },
+    {
+      "epoch": 156.53,
+      "learning_rate": 1.1439628482972137e-05,
+      "loss": 0.021,
+      "step": 2661
+    },
+    {
+      "epoch": 156.59,
+      "learning_rate": 1.1424148606811145e-05,
+      "loss": 0.0237,
+      "step": 2662
+    },
+    {
+      "epoch": 156.65,
+      "learning_rate": 1.1408668730650155e-05,
+      "loss": 0.0207,
+      "step": 2663
+    },
+    {
+      "epoch": 156.71,
+      "learning_rate": 1.1393188854489165e-05,
+      "loss": 0.0208,
+      "step": 2664
+    },
+    {
+      "epoch": 156.76,
+      "learning_rate": 1.1377708978328173e-05,
+      "loss": 0.0235,
+      "step": 2665
+    },
+    {
+      "epoch": 156.82,
+      "learning_rate": 1.1362229102167183e-05,
+      "loss": 0.0214,
+      "step": 2666
+    },
+    {
+      "epoch": 156.88,
+      "learning_rate": 1.1346749226006193e-05,
+      "loss": 0.0224,
+      "step": 2667
+    },
+    {
+      "epoch": 156.94,
+      "learning_rate": 1.1331269349845202e-05,
+      "loss": 0.0215,
+      "step": 2668
+    },
+    {
+      "epoch": 157.0,
+      "learning_rate": 1.1315789473684212e-05,
+      "loss": 0.0237,
+      "step": 2669
+    },
+    {
+      "epoch": 157.06,
+      "learning_rate": 1.130030959752322e-05,
+      "loss": 0.0198,
+      "step": 2670
+    },
+    {
+      "epoch": 157.12,
+      "learning_rate": 1.128482972136223e-05,
+      "loss": 0.0216,
+      "step": 2671
+    },
+    {
+      "epoch": 157.18,
+      "learning_rate": 1.126934984520124e-05,
+      "loss": 0.0188,
+      "step": 2672
+    },
+    {
+      "epoch": 157.24,
+      "learning_rate": 1.1253869969040248e-05,
+      "loss": 0.0204,
+      "step": 2673
+    },
+    {
+      "epoch": 157.29,
+      "learning_rate": 1.1238390092879257e-05,
+      "loss": 0.022,
+      "step": 2674
+    },
+    {
+      "epoch": 157.35,
+      "learning_rate": 1.1222910216718267e-05,
+      "loss": 0.0253,
+      "step": 2675
+    },
+    {
+      "epoch": 157.41,
+      "learning_rate": 1.1207430340557277e-05,
+      "loss": 0.0212,
+      "step": 2676
+    },
+    {
+      "epoch": 157.47,
+      "learning_rate": 1.1191950464396285e-05,
+      "loss": 0.0283,
+      "step": 2677
+    },
+    {
+      "epoch": 157.53,
+      "learning_rate": 1.1176470588235295e-05,
+      "loss": 0.0193,
+      "step": 2678
+    },
+    {
+      "epoch": 157.59,
+      "learning_rate": 1.1160990712074305e-05,
+      "loss": 0.0237,
+      "step": 2679
+    },
+    {
+      "epoch": 157.65,
+      "learning_rate": 1.1145510835913313e-05,
+      "loss": 0.0225,
+      "step": 2680
+    },
+    {
+      "epoch": 157.65,
+      "eval_accuracy_NAT": 0.9938389102622057,
+      "eval_accuracy_REST": 0.3203252032520325,
+      "eval_accuracy_concrete": 0.9571459038172537,
+      "eval_accuracy_concrete tiles": 0.9291774687720952,
+      "eval_accuracy_dark bricks": 0.8643215358995463,
+      "eval_accuracy_grass": 0.8980392268183659,
+      "eval_accuracy_red bricks": 0.9264608963093146,
+      "eval_accuracy_road": 0.98097425129636,
+      "eval_accuracy_rough concrete": 0.6219736740178406,
+      "eval_accuracy_rough red sidewalk": 0.7661660863745403,
+      "eval_accuracy_speedway bricks": 0.9885450961118617,
+      "eval_accuracy_steel": 0.9237181825905731,
+      "eval_accuracy_tiles": 0.242691387197431,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835807089243077,
+      "eval_iou_REST": 0.3152,
+      "eval_iou_concrete": 0.8912357634404546,
+      "eval_iou_concrete tiles": 0.865285428966486,
+      "eval_iou_dark bricks": 0.8466687604591968,
+      "eval_iou_grass": 0.8418850387512319,
+      "eval_iou_red bricks": 0.9053292545483819,
+      "eval_iou_road": 0.9601985261193249,
+      "eval_iou_rough concrete": 0.6168570601202049,
+      "eval_iou_rough red sidewalk": 0.622196166083457,
+      "eval_iou_speedway bricks": 0.9794348540885546,
+      "eval_iou_steel": 0.8654449930815534,
+      "eval_iou_tiles": 0.23876371006199332,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08075359463691711,
+      "eval_mean_accuracy": 0.8010290632861093,
+      "eval_mean_iou": 0.7640061742034729,
+      "eval_overall_accuracy": 0.9799637552035056,
+      "eval_runtime": 5.3885,
+      "eval_samples_per_second": 21.899,
+      "eval_steps_per_second": 1.485,
+      "step": 2680
+    },
+    {
+      "epoch": 157.71,
+      "learning_rate": 1.1130030959752323e-05,
+      "loss": 0.0209,
+      "step": 2681
+    },
+    {
+      "epoch": 157.76,
+      "learning_rate": 1.1114551083591331e-05,
+      "loss": 0.0219,
+      "step": 2682
+    },
+    {
+      "epoch": 157.82,
+      "learning_rate": 1.1099071207430341e-05,
+      "loss": 0.0226,
+      "step": 2683
+    },
+    {
+      "epoch": 157.88,
+      "learning_rate": 1.1083591331269351e-05,
+      "loss": 0.019,
+      "step": 2684
+    },
+    {
+      "epoch": 157.94,
+      "learning_rate": 1.106811145510836e-05,
+      "loss": 0.0239,
+      "step": 2685
+    },
+    {
+      "epoch": 158.0,
+      "learning_rate": 1.1052631578947368e-05,
+      "loss": 0.0245,
+      "step": 2686
+    },
+    {
+      "epoch": 158.06,
+      "learning_rate": 1.1037151702786378e-05,
+      "loss": 0.0236,
+      "step": 2687
+    },
+    {
+      "epoch": 158.12,
+      "learning_rate": 1.1021671826625388e-05,
+      "loss": 0.0214,
+      "step": 2688
+    },
+    {
+      "epoch": 158.18,
+      "learning_rate": 1.1006191950464396e-05,
+      "loss": 0.0217,
+      "step": 2689
+    },
+    {
+      "epoch": 158.24,
+      "learning_rate": 1.0990712074303406e-05,
+      "loss": 0.0202,
+      "step": 2690
+    },
+    {
+      "epoch": 158.29,
+      "learning_rate": 1.0975232198142415e-05,
+      "loss": 0.0218,
+      "step": 2691
+    },
+    {
+      "epoch": 158.35,
+      "learning_rate": 1.0959752321981425e-05,
+      "loss": 0.0237,
+      "step": 2692
+    },
+    {
+      "epoch": 158.41,
+      "learning_rate": 1.0944272445820435e-05,
+      "loss": 0.0223,
+      "step": 2693
+    },
+    {
+      "epoch": 158.47,
+      "learning_rate": 1.0928792569659443e-05,
+      "loss": 0.0202,
+      "step": 2694
+    },
+    {
+      "epoch": 158.53,
+      "learning_rate": 1.0913312693498453e-05,
+      "loss": 0.0201,
+      "step": 2695
+    },
+    {
+      "epoch": 158.59,
+      "learning_rate": 1.0897832817337463e-05,
+      "loss": 0.0232,
+      "step": 2696
+    },
+    {
+      "epoch": 158.65,
+      "learning_rate": 1.0882352941176471e-05,
+      "loss": 0.0235,
+      "step": 2697
+    },
+    {
+      "epoch": 158.71,
+      "learning_rate": 1.086687306501548e-05,
+      "loss": 0.0201,
+      "step": 2698
+    },
+    {
+      "epoch": 158.76,
+      "learning_rate": 1.085139318885449e-05,
+      "loss": 0.0239,
+      "step": 2699
+    },
+    {
+      "epoch": 158.82,
+      "learning_rate": 1.08359133126935e-05,
+      "loss": 0.0269,
+      "step": 2700
+    },
+    {
+      "epoch": 158.82,
+      "eval_accuracy_NAT": 0.9942922003324307,
+      "eval_accuracy_REST": 0.2552845528455285,
+      "eval_accuracy_concrete": 0.9557722795436234,
+      "eval_accuracy_concrete tiles": 0.9247466415272213,
+      "eval_accuracy_dark bricks": 0.8602875026319315,
+      "eval_accuracy_grass": 0.8893642106378711,
+      "eval_accuracy_red bricks": 0.9212434094903339,
+      "eval_accuracy_road": 0.980666427604517,
+      "eval_accuracy_rough concrete": 0.6250424564560628,
+      "eval_accuracy_rough red sidewalk": 0.7760774257589544,
+      "eval_accuracy_speedway bricks": 0.9899199587626887,
+      "eval_accuracy_steel": 0.9176583186313954,
+      "eval_accuracy_tiles": 0.24444848374685693,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834257991120192,
+      "eval_iou_REST": 0.2516025641025641,
+      "eval_iou_concrete": 0.8928482296861182,
+      "eval_iou_concrete tiles": 0.8697107392219883,
+      "eval_iou_dark bricks": 0.843891997446393,
+      "eval_iou_grass": 0.83900462570363,
+      "eval_iou_red bricks": 0.9014402407566638,
+      "eval_iou_road": 0.9608989917238093,
+      "eval_iou_rough concrete": 0.6184423088261305,
+      "eval_iou_rough red sidewalk": 0.6271657156053065,
+      "eval_iou_speedway bricks": 0.9795101626805487,
+      "eval_iou_steel": 0.8653123836543491,
+      "eval_iou_tiles": 0.24177503445796128,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0806136280298233,
+      "eval_mean_accuracy": 0.7949849129207244,
+      "eval_mean_iou": 0.7596175994598062,
+      "eval_overall_accuracy": 0.9800087880280058,
+      "eval_runtime": 5.1938,
+      "eval_samples_per_second": 22.719,
+      "eval_steps_per_second": 1.54,
+      "step": 2700
+    },
+    {
+      "epoch": 158.88,
+      "learning_rate": 1.0820433436532508e-05,
+      "loss": 0.0219,
+      "step": 2701
+    },
+    {
+      "epoch": 158.94,
+      "learning_rate": 1.0804953560371518e-05,
+      "loss": 0.0205,
+      "step": 2702
+    },
+    {
+      "epoch": 159.0,
+      "learning_rate": 1.0789473684210526e-05,
+      "loss": 0.019,
+      "step": 2703
+    },
+    {
+      "epoch": 159.06,
+      "learning_rate": 1.0773993808049536e-05,
+      "loss": 0.0212,
+      "step": 2704
+    },
+    {
+      "epoch": 159.12,
+      "learning_rate": 1.0758513931888545e-05,
+      "loss": 0.0181,
+      "step": 2705
+    },
+    {
+      "epoch": 159.18,
+      "learning_rate": 1.0743034055727555e-05,
+      "loss": 0.024,
+      "step": 2706
+    },
+    {
+      "epoch": 159.24,
+      "learning_rate": 1.0727554179566565e-05,
+      "loss": 0.0213,
+      "step": 2707
+    },
+    {
+      "epoch": 159.29,
+      "learning_rate": 1.0712074303405573e-05,
+      "loss": 0.0236,
+      "step": 2708
+    },
+    {
+      "epoch": 159.35,
+      "learning_rate": 1.0696594427244583e-05,
+      "loss": 0.0215,
+      "step": 2709
+    },
+    {
+      "epoch": 159.41,
+      "learning_rate": 1.0681114551083591e-05,
+      "loss": 0.0214,
+      "step": 2710
+    },
+    {
+      "epoch": 159.47,
+      "learning_rate": 1.0665634674922601e-05,
+      "loss": 0.0248,
+      "step": 2711
+    },
+    {
+      "epoch": 159.53,
+      "learning_rate": 1.0650154798761611e-05,
+      "loss": 0.0214,
+      "step": 2712
+    },
+    {
+      "epoch": 159.59,
+      "learning_rate": 1.063467492260062e-05,
+      "loss": 0.0223,
+      "step": 2713
+    },
+    {
+      "epoch": 159.65,
+      "learning_rate": 1.0619195046439628e-05,
+      "loss": 0.0233,
+      "step": 2714
+    },
+    {
+      "epoch": 159.71,
+      "learning_rate": 1.0603715170278638e-05,
+      "loss": 0.0216,
+      "step": 2715
+    },
+    {
+      "epoch": 159.76,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0201,
+      "step": 2716
+    },
+    {
+      "epoch": 159.82,
+      "learning_rate": 1.0572755417956656e-05,
+      "loss": 0.0237,
+      "step": 2717
+    },
+    {
+      "epoch": 159.88,
+      "learning_rate": 1.0557275541795666e-05,
+      "loss": 0.0226,
+      "step": 2718
+    },
+    {
+      "epoch": 159.94,
+      "learning_rate": 1.0541795665634676e-05,
+      "loss": 0.0192,
+      "step": 2719
+    },
+    {
+      "epoch": 160.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0204,
+      "step": 2720
+    },
+    {
+      "epoch": 160.0,
+      "eval_accuracy_NAT": 0.9939970334515457,
+      "eval_accuracy_REST": 0.34796747967479674,
+      "eval_accuracy_concrete": 0.954261827921646,
+      "eval_accuracy_concrete tiles": 0.9310157907141173,
+      "eval_accuracy_dark bricks": 0.866838619527975,
+      "eval_accuracy_grass": 0.89485043923893,
+      "eval_accuracy_red bricks": 0.9281085237258347,
+      "eval_accuracy_road": 0.9815920363445723,
+      "eval_accuracy_rough concrete": 0.6257038833505145,
+      "eval_accuracy_rough red sidewalk": 0.7981485798509539,
+      "eval_accuracy_speedway bricks": 0.9892302140076621,
+      "eval_accuracy_steel": 0.9247552314743087,
+      "eval_accuracy_tiles": 0.2526886606683026,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983518218056789,
+      "eval_iou_REST": 0.33914421553090335,
+      "eval_iou_concrete": 0.8931911820102879,
+      "eval_iou_concrete tiles": 0.8665028844677444,
+      "eval_iou_dark bricks": 0.8496761150694441,
+      "eval_iou_grass": 0.8404022756638686,
+      "eval_iou_red bricks": 0.9049480561208096,
+      "eval_iou_road": 0.9605439586989672,
+      "eval_iou_rough concrete": 0.6186517568873282,
+      "eval_iou_rough red sidewalk": 0.6307841342667982,
+      "eval_iou_speedway bricks": 0.9796196710495166,
+      "eval_iou_steel": 0.8656120472684063,
+      "eval_iou_tiles": 0.2475074183976261,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08046878129243851,
+      "eval_mean_accuracy": 0.8068583323039353,
+      "eval_mean_iou": 0.7677001487298837,
+      "eval_overall_accuracy": 0.9800751572948391,
+      "eval_runtime": 5.2952,
+      "eval_samples_per_second": 22.284,
+      "eval_steps_per_second": 1.511,
+      "step": 2720
+    },
+    {
+      "epoch": 160.06,
+      "learning_rate": 1.0510835913312694e-05,
+      "loss": 0.0212,
+      "step": 2721
+    },
+    {
+      "epoch": 160.12,
+      "learning_rate": 1.0495356037151703e-05,
+      "loss": 0.0231,
+      "step": 2722
+    },
+    {
+      "epoch": 160.18,
+      "learning_rate": 1.0479876160990713e-05,
+      "loss": 0.0237,
+      "step": 2723
+    },
+    {
+      "epoch": 160.24,
+      "learning_rate": 1.0464396284829723e-05,
+      "loss": 0.0232,
+      "step": 2724
+    },
+    {
+      "epoch": 160.29,
+      "learning_rate": 1.0448916408668731e-05,
+      "loss": 0.0216,
+      "step": 2725
+    },
+    {
+      "epoch": 160.35,
+      "learning_rate": 1.043343653250774e-05,
+      "loss": 0.0204,
+      "step": 2726
+    },
+    {
+      "epoch": 160.41,
+      "learning_rate": 1.041795665634675e-05,
+      "loss": 0.0224,
+      "step": 2727
+    },
+    {
+      "epoch": 160.47,
+      "learning_rate": 1.040247678018576e-05,
+      "loss": 0.0204,
+      "step": 2728
+    },
+    {
+      "epoch": 160.53,
+      "learning_rate": 1.0386996904024768e-05,
+      "loss": 0.0199,
+      "step": 2729
+    },
+    {
+      "epoch": 160.59,
+      "learning_rate": 1.0371517027863778e-05,
+      "loss": 0.0202,
+      "step": 2730
+    },
+    {
+      "epoch": 160.65,
+      "learning_rate": 1.0356037151702786e-05,
+      "loss": 0.0224,
+      "step": 2731
+    },
+    {
+      "epoch": 160.71,
+      "learning_rate": 1.0340557275541796e-05,
+      "loss": 0.0267,
+      "step": 2732
+    },
+    {
+      "epoch": 160.76,
+      "learning_rate": 1.0325077399380806e-05,
+      "loss": 0.0227,
+      "step": 2733
+    },
+    {
+      "epoch": 160.82,
+      "learning_rate": 1.0309597523219814e-05,
+      "loss": 0.0203,
+      "step": 2734
+    },
+    {
+      "epoch": 160.88,
+      "learning_rate": 1.0294117647058824e-05,
+      "loss": 0.0226,
+      "step": 2735
+    },
+    {
+      "epoch": 160.94,
+      "learning_rate": 1.0278637770897834e-05,
+      "loss": 0.0194,
+      "step": 2736
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 1.0263157894736843e-05,
+      "loss": 0.0234,
+      "step": 2737
+    },
+    {
+      "epoch": 161.06,
+      "learning_rate": 1.0247678018575851e-05,
+      "loss": 0.023,
+      "step": 2738
+    },
+    {
+      "epoch": 161.12,
+      "learning_rate": 1.0232198142414861e-05,
+      "loss": 0.0218,
+      "step": 2739
+    },
+    {
+      "epoch": 161.18,
+      "learning_rate": 1.0216718266253871e-05,
+      "loss": 0.0188,
+      "step": 2740
+    },
+    {
+      "epoch": 161.18,
+      "eval_accuracy_NAT": 0.9934050284252445,
+      "eval_accuracy_REST": 0.36585365853658536,
+      "eval_accuracy_concrete": 0.9578820196633895,
+      "eval_accuracy_concrete tiles": 0.9386283290124912,
+      "eval_accuracy_dark bricks": 0.867656910973719,
+      "eval_accuracy_grass": 0.8992210653010707,
+      "eval_accuracy_red bricks": 0.9273945518453427,
+      "eval_accuracy_road": 0.9822787610737703,
+      "eval_accuracy_rough concrete": 0.6254774489181797,
+      "eval_accuracy_rough red sidewalk": 0.7694147196173832,
+      "eval_accuracy_speedway bricks": 0.9895726015904057,
+      "eval_accuracy_steel": 0.9167612088741397,
+      "eval_accuracy_tiles": 0.2531430821897058,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837008827549625,
+      "eval_iou_REST": 0.35714285714285715,
+      "eval_iou_concrete": 0.892026902069264,
+      "eval_iou_concrete tiles": 0.8708753362052,
+      "eval_iou_dark bricks": 0.849418389479947,
+      "eval_iou_grass": 0.8420361509835194,
+      "eval_iou_red bricks": 0.9019335541074671,
+      "eval_iou_road": 0.9608230274555067,
+      "eval_iou_rough concrete": 0.6191464889255905,
+      "eval_iou_rough red sidewalk": 0.6212641933329285,
+      "eval_iou_speedway bricks": 0.9793934745804969,
+      "eval_iou_steel": 0.8653554933883089,
+      "eval_iou_tiles": 0.248439079502884,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08005556464195251,
+      "eval_mean_accuracy": 0.8066684143093406,
+      "eval_mean_iou": 0.768581217686841,
+      "eval_overall_accuracy": 0.9800700494798563,
+      "eval_runtime": 5.4771,
+      "eval_samples_per_second": 21.544,
+      "eval_steps_per_second": 1.461,
+      "step": 2740
+    },
+    {
+      "epoch": 161.24,
+      "learning_rate": 1.020123839009288e-05,
+      "loss": 0.0216,
+      "step": 2741
+    },
+    {
+      "epoch": 161.29,
+      "learning_rate": 1.018575851393189e-05,
+      "loss": 0.024,
+      "step": 2742
+    },
+    {
+      "epoch": 161.35,
+      "learning_rate": 1.0170278637770898e-05,
+      "loss": 0.0208,
+      "step": 2743
+    },
+    {
+      "epoch": 161.41,
+      "learning_rate": 1.0154798761609908e-05,
+      "loss": 0.0194,
+      "step": 2744
+    },
+    {
+      "epoch": 161.47,
+      "learning_rate": 1.0139318885448918e-05,
+      "loss": 0.0217,
+      "step": 2745
+    },
+    {
+      "epoch": 161.53,
+      "learning_rate": 1.0123839009287926e-05,
+      "loss": 0.0198,
+      "step": 2746
+    },
+    {
+      "epoch": 161.59,
+      "learning_rate": 1.0108359133126936e-05,
+      "loss": 0.0206,
+      "step": 2747
+    },
+    {
+      "epoch": 161.65,
+      "learning_rate": 1.0092879256965946e-05,
+      "loss": 0.026,
+      "step": 2748
+    },
+    {
+      "epoch": 161.71,
+      "learning_rate": 1.0077399380804954e-05,
+      "loss": 0.0222,
+      "step": 2749
+    },
+    {
+      "epoch": 161.76,
+      "learning_rate": 1.0061919504643963e-05,
+      "loss": 0.0183,
+      "step": 2750
+    },
+    {
+      "epoch": 161.82,
+      "learning_rate": 1.0046439628482973e-05,
+      "loss": 0.0225,
+      "step": 2751
+    },
+    {
+      "epoch": 161.88,
+      "learning_rate": 1.0030959752321983e-05,
+      "loss": 0.0251,
+      "step": 2752
+    },
+    {
+      "epoch": 161.94,
+      "learning_rate": 1.001547987616099e-05,
+      "loss": 0.0222,
+      "step": 2753
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 1e-05,
+      "loss": 0.0237,
+      "step": 2754
+    },
+    {
+      "epoch": 162.06,
+      "learning_rate": 9.984520123839009e-06,
+      "loss": 0.0187,
+      "step": 2755
+    },
+    {
+      "epoch": 162.12,
+      "learning_rate": 9.969040247678019e-06,
+      "loss": 0.0206,
+      "step": 2756
+    },
+    {
+      "epoch": 162.18,
+      "learning_rate": 9.953560371517029e-06,
+      "loss": 0.0226,
+      "step": 2757
+    },
+    {
+      "epoch": 162.24,
+      "learning_rate": 9.938080495356037e-06,
+      "loss": 0.0226,
+      "step": 2758
+    },
+    {
+      "epoch": 162.29,
+      "learning_rate": 9.922600619195046e-06,
+      "loss": 0.0271,
+      "step": 2759
+    },
+    {
+      "epoch": 162.35,
+      "learning_rate": 9.907120743034057e-06,
+      "loss": 0.0218,
+      "step": 2760
+    },
+    {
+      "epoch": 162.35,
+      "eval_accuracy_NAT": 0.9935705375255764,
+      "eval_accuracy_REST": 0.35772357723577236,
+      "eval_accuracy_concrete": 0.9578369201463365,
+      "eval_accuracy_concrete tiles": 0.932076361065284,
+      "eval_accuracy_dark bricks": 0.8665467143923588,
+      "eval_accuracy_grass": 0.8999950796156044,
+      "eval_accuracy_red bricks": 0.9289872583479789,
+      "eval_accuracy_road": 0.9800422295627247,
+      "eval_accuracy_rough concrete": 0.6247147224092624,
+      "eval_accuracy_rough red sidewalk": 0.7756487866505237,
+      "eval_accuracy_speedway bricks": 0.9893912972547987,
+      "eval_accuracy_steel": 0.9216165967804005,
+      "eval_accuracy_tiles": 0.2514768699445606,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837143175727022,
+      "eval_iou_REST": 0.3514376996805112,
+      "eval_iou_concrete": 0.8918607866436488,
+      "eval_iou_concrete tiles": 0.8731977655604867,
+      "eval_iou_dark bricks": 0.8478191658707418,
+      "eval_iou_grass": 0.8421416112552349,
+      "eval_iou_red bricks": 0.9040619989310529,
+      "eval_iou_road": 0.9609052710893089,
+      "eval_iou_rough concrete": 0.618263843840302,
+      "eval_iou_rough red sidewalk": 0.6240282172437367,
+      "eval_iou_speedway bricks": 0.9795047563853914,
+      "eval_iou_steel": 0.8673546420701349,
+      "eval_iou_tiles": 0.24692863729660589,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08054724335670471,
+      "eval_mean_accuracy": 0.8061251500716294,
+      "eval_mean_iou": 0.7685552856492198,
+      "eval_overall_accuracy": 0.9800904160839016,
+      "eval_runtime": 5.5636,
+      "eval_samples_per_second": 21.209,
+      "eval_steps_per_second": 1.438,
+      "step": 2760
+    },
+    {
+      "epoch": 162.41,
+      "learning_rate": 9.891640866873066e-06,
+      "loss": 0.02,
+      "step": 2761
+    },
+    {
+      "epoch": 162.47,
+      "learning_rate": 9.876160990712074e-06,
+      "loss": 0.0239,
+      "step": 2762
+    },
+    {
+      "epoch": 162.53,
+      "learning_rate": 9.860681114551084e-06,
+      "loss": 0.0216,
+      "step": 2763
+    },
+    {
+      "epoch": 162.59,
+      "learning_rate": 9.845201238390094e-06,
+      "loss": 0.025,
+      "step": 2764
+    },
+    {
+      "epoch": 162.65,
+      "learning_rate": 9.829721362229102e-06,
+      "loss": 0.0201,
+      "step": 2765
+    },
+    {
+      "epoch": 162.71,
+      "learning_rate": 9.814241486068112e-06,
+      "loss": 0.0207,
+      "step": 2766
+    },
+    {
+      "epoch": 162.76,
+      "learning_rate": 9.79876160990712e-06,
+      "loss": 0.0209,
+      "step": 2767
+    },
+    {
+      "epoch": 162.82,
+      "learning_rate": 9.78328173374613e-06,
+      "loss": 0.0259,
+      "step": 2768
+    },
+    {
+      "epoch": 162.88,
+      "learning_rate": 9.76780185758514e-06,
+      "loss": 0.0178,
+      "step": 2769
+    },
+    {
+      "epoch": 162.94,
+      "learning_rate": 9.752321981424149e-06,
+      "loss": 0.0223,
+      "step": 2770
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 9.736842105263157e-06,
+      "loss": 0.0252,
+      "step": 2771
+    },
+    {
+      "epoch": 163.06,
+      "learning_rate": 9.721362229102169e-06,
+      "loss": 0.0217,
+      "step": 2772
+    },
+    {
+      "epoch": 163.12,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0182,
+      "step": 2773
+    },
+    {
+      "epoch": 163.18,
+      "learning_rate": 9.690402476780186e-06,
+      "loss": 0.0217,
+      "step": 2774
+    },
+    {
+      "epoch": 163.24,
+      "learning_rate": 9.674922600619196e-06,
+      "loss": 0.0222,
+      "step": 2775
+    },
+    {
+      "epoch": 163.29,
+      "learning_rate": 9.659442724458206e-06,
+      "loss": 0.0223,
+      "step": 2776
+    },
+    {
+      "epoch": 163.35,
+      "learning_rate": 9.643962848297214e-06,
+      "loss": 0.023,
+      "step": 2777
+    },
+    {
+      "epoch": 163.41,
+      "learning_rate": 9.628482972136222e-06,
+      "loss": 0.0213,
+      "step": 2778
+    },
+    {
+      "epoch": 163.47,
+      "learning_rate": 9.613003095975232e-06,
+      "loss": 0.0225,
+      "step": 2779
+    },
+    {
+      "epoch": 163.53,
+      "learning_rate": 9.597523219814242e-06,
+      "loss": 0.0203,
+      "step": 2780
+    },
+    {
+      "epoch": 163.53,
+      "eval_accuracy_NAT": 0.9931384886884234,
+      "eval_accuracy_REST": 0.32682926829268294,
+      "eval_accuracy_concrete": 0.9579603705192865,
+      "eval_accuracy_concrete tiles": 0.9177468772095215,
+      "eval_accuracy_dark bricks": 0.8569138449170224,
+      "eval_accuracy_grass": 0.9023313159757311,
+      "eval_accuracy_red bricks": 0.9230008787346221,
+      "eval_accuracy_road": 0.9821793596732794,
+      "eval_accuracy_rough concrete": 0.6227483181284599,
+      "eval_accuracy_rough red sidewalk": 0.7698132787883801,
+      "eval_accuracy_speedway bricks": 0.9898034304562494,
+      "eval_accuracy_steel": 0.9155167502136571,
+      "eval_accuracy_tiles": 0.24114635402465995,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835999365252531,
+      "eval_iou_REST": 0.3216,
+      "eval_iou_concrete": 0.8909636457803461,
+      "eval_iou_concrete tiles": 0.871316372423978,
+      "eval_iou_dark bricks": 0.8405195073410687,
+      "eval_iou_grass": 0.8406492755462934,
+      "eval_iou_red bricks": 0.9017545742340506,
+      "eval_iou_road": 0.959926438136008,
+      "eval_iou_rough concrete": 0.6166923353809296,
+      "eval_iou_rough red sidewalk": 0.6241555495939322,
+      "eval_iou_speedway bricks": 0.9795240786802043,
+      "eval_iou_steel": 0.8663756408558618,
+      "eval_iou_tiles": 0.2384161499985024,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08148735761642456,
+      "eval_mean_accuracy": 0.7999329642786136,
+      "eval_mean_iou": 0.7642687311151098,
+      "eval_overall_accuracy": 0.9799140671552238,
+      "eval_runtime": 5.3192,
+      "eval_samples_per_second": 22.184,
+      "eval_steps_per_second": 1.504,
+      "step": 2780
+    },
+    {
+      "epoch": 163.59,
+      "learning_rate": 9.58204334365325e-06,
+      "loss": 0.0227,
+      "step": 2781
+    },
+    {
+      "epoch": 163.65,
+      "learning_rate": 9.56656346749226e-06,
+      "loss": 0.0205,
+      "step": 2782
+    },
+    {
+      "epoch": 163.71,
+      "learning_rate": 9.551083591331269e-06,
+      "loss": 0.0232,
+      "step": 2783
+    },
+    {
+      "epoch": 163.76,
+      "learning_rate": 9.535603715170279e-06,
+      "loss": 0.0217,
+      "step": 2784
+    },
+    {
+      "epoch": 163.82,
+      "learning_rate": 9.520123839009289e-06,
+      "loss": 0.0247,
+      "step": 2785
+    },
+    {
+      "epoch": 163.88,
+      "learning_rate": 9.504643962848297e-06,
+      "loss": 0.021,
+      "step": 2786
+    },
+    {
+      "epoch": 163.94,
+      "learning_rate": 9.489164086687305e-06,
+      "loss": 0.0231,
+      "step": 2787
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0213,
+      "step": 2788
+    },
+    {
+      "epoch": 164.06,
+      "learning_rate": 9.458204334365325e-06,
+      "loss": 0.0208,
+      "step": 2789
+    },
+    {
+      "epoch": 164.12,
+      "learning_rate": 9.442724458204334e-06,
+      "loss": 0.019,
+      "step": 2790
+    },
+    {
+      "epoch": 164.18,
+      "learning_rate": 9.427244582043344e-06,
+      "loss": 0.0186,
+      "step": 2791
+    },
+    {
+      "epoch": 164.24,
+      "learning_rate": 9.411764705882354e-06,
+      "loss": 0.0214,
+      "step": 2792
+    },
+    {
+      "epoch": 164.29,
+      "learning_rate": 9.396284829721362e-06,
+      "loss": 0.0215,
+      "step": 2793
+    },
+    {
+      "epoch": 164.35,
+      "learning_rate": 9.380804953560372e-06,
+      "loss": 0.0224,
+      "step": 2794
+    },
+    {
+      "epoch": 164.41,
+      "learning_rate": 9.36532507739938e-06,
+      "loss": 0.0219,
+      "step": 2795
+    },
+    {
+      "epoch": 164.47,
+      "learning_rate": 9.34984520123839e-06,
+      "loss": 0.022,
+      "step": 2796
+    },
+    {
+      "epoch": 164.53,
+      "learning_rate": 9.3343653250774e-06,
+      "loss": 0.0231,
+      "step": 2797
+    },
+    {
+      "epoch": 164.59,
+      "learning_rate": 9.318885448916409e-06,
+      "loss": 0.0229,
+      "step": 2798
+    },
+    {
+      "epoch": 164.65,
+      "learning_rate": 9.303405572755417e-06,
+      "loss": 0.0228,
+      "step": 2799
+    },
+    {
+      "epoch": 164.71,
+      "learning_rate": 9.287925696594429e-06,
+      "loss": 0.0229,
+      "step": 2800
+    },
+    {
+      "epoch": 164.71,
+      "eval_accuracy_NAT": 0.9933407656084381,
+      "eval_accuracy_REST": 0.3300813008130081,
+      "eval_accuracy_concrete": 0.9548225142904317,
+      "eval_accuracy_concrete tiles": 0.9398538769738393,
+      "eval_accuracy_dark bricks": 0.8573923779262294,
+      "eval_accuracy_grass": 0.9024126915638117,
+      "eval_accuracy_red bricks": 0.930305360281195,
+      "eval_accuracy_road": 0.9801042218339986,
+      "eval_accuracy_rough concrete": 0.6250603328586155,
+      "eval_accuracy_rough red sidewalk": 0.789056918761609,
+      "eval_accuracy_speedway bricks": 0.9911529310639103,
+      "eval_accuracy_steel": 0.9162364371498398,
+      "eval_accuracy_tiles": 0.25123451179981215,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836287408897749,
+      "eval_iou_REST": 0.3248,
+      "eval_iou_concrete": 0.8924182216449549,
+      "eval_iou_concrete tiles": 0.8792997001234786,
+      "eval_iou_dark bricks": 0.8400977146768945,
+      "eval_iou_grass": 0.8409675101076236,
+      "eval_iou_red bricks": 0.9061196105702365,
+      "eval_iou_road": 0.9611498756355792,
+      "eval_iou_rough concrete": 0.6189855192191944,
+      "eval_iou_rough red sidewalk": 0.6315599400509206,
+      "eval_iou_speedway bricks": 0.9792372809616524,
+      "eval_iou_steel": 0.866876300359372,
+      "eval_iou_tiles": 0.24814482345900657,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08053811639547348,
+      "eval_mean_accuracy": 0.8046964800711337,
+      "eval_mean_iou": 0.7671757875152837,
+      "eval_overall_accuracy": 0.9800550493143373,
+      "eval_runtime": 5.3606,
+      "eval_samples_per_second": 22.012,
+      "eval_steps_per_second": 1.492,
+      "step": 2800
+    },
+    {
+      "epoch": 164.76,
+      "learning_rate": 9.272445820433437e-06,
+      "loss": 0.0194,
+      "step": 2801
+    },
+    {
+      "epoch": 164.82,
+      "learning_rate": 9.256965944272445e-06,
+      "loss": 0.0188,
+      "step": 2802
+    },
+    {
+      "epoch": 164.88,
+      "learning_rate": 9.241486068111455e-06,
+      "loss": 0.0224,
+      "step": 2803
+    },
+    {
+      "epoch": 164.94,
+      "learning_rate": 9.226006191950465e-06,
+      "loss": 0.0234,
+      "step": 2804
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 9.210526315789474e-06,
+      "loss": 0.0228,
+      "step": 2805
+    },
+    {
+      "epoch": 165.06,
+      "learning_rate": 9.195046439628484e-06,
+      "loss": 0.0195,
+      "step": 2806
+    },
+    {
+      "epoch": 165.12,
+      "learning_rate": 9.179566563467492e-06,
+      "loss": 0.0222,
+      "step": 2807
+    },
+    {
+      "epoch": 165.18,
+      "learning_rate": 9.164086687306502e-06,
+      "loss": 0.0204,
+      "step": 2808
+    },
+    {
+      "epoch": 165.24,
+      "learning_rate": 9.148606811145512e-06,
+      "loss": 0.0203,
+      "step": 2809
+    },
+    {
+      "epoch": 165.29,
+      "learning_rate": 9.13312693498452e-06,
+      "loss": 0.0199,
+      "step": 2810
+    },
+    {
+      "epoch": 165.35,
+      "learning_rate": 9.117647058823529e-06,
+      "loss": 0.0228,
+      "step": 2811
+    },
+    {
+      "epoch": 165.41,
+      "learning_rate": 9.10216718266254e-06,
+      "loss": 0.0199,
+      "step": 2812
+    },
+    {
+      "epoch": 165.47,
+      "learning_rate": 9.086687306501549e-06,
+      "loss": 0.0229,
+      "step": 2813
+    },
+    {
+      "epoch": 165.53,
+      "learning_rate": 9.071207430340557e-06,
+      "loss": 0.0255,
+      "step": 2814
+    },
+    {
+      "epoch": 165.59,
+      "learning_rate": 9.055727554179567e-06,
+      "loss": 0.0239,
+      "step": 2815
+    },
+    {
+      "epoch": 165.65,
+      "learning_rate": 9.040247678018577e-06,
+      "loss": 0.0251,
+      "step": 2816
+    },
+    {
+      "epoch": 165.71,
+      "learning_rate": 9.024767801857585e-06,
+      "loss": 0.0207,
+      "step": 2817
+    },
+    {
+      "epoch": 165.76,
+      "learning_rate": 9.009287925696595e-06,
+      "loss": 0.0227,
+      "step": 2818
+    },
+    {
+      "epoch": 165.82,
+      "learning_rate": 8.993808049535604e-06,
+      "loss": 0.0207,
+      "step": 2819
+    },
+    {
+      "epoch": 165.88,
+      "learning_rate": 8.978328173374614e-06,
+      "loss": 0.0189,
+      "step": 2820
+    },
+    {
+      "epoch": 165.88,
+      "eval_accuracy_NAT": 0.9940474948580312,
+      "eval_accuracy_REST": 0.33495934959349594,
+      "eval_accuracy_concrete": 0.955562452129538,
+      "eval_accuracy_concrete tiles": 0.9350930945086025,
+      "eval_accuracy_dark bricks": 0.8622303466493119,
+      "eval_accuracy_grass": 0.892068529599897,
+      "eval_accuracy_red bricks": 0.9278339191564148,
+      "eval_accuracy_road": 0.9820655290372333,
+      "eval_accuracy_rough concrete": 0.624816022023728,
+      "eval_accuracy_rough red sidewalk": 0.781582054309327,
+      "eval_accuracy_speedway bricks": 0.9889923591704007,
+      "eval_accuracy_steel": 0.9258422586175014,
+      "eval_accuracy_tiles": 0.25238571298736706,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835063978476642,
+      "eval_iou_REST": 0.32854864433811803,
+      "eval_iou_concrete": 0.8929697642896813,
+      "eval_iou_concrete tiles": 0.8755213275371273,
+      "eval_iou_dark bricks": 0.8449886510720517,
+      "eval_iou_grass": 0.8390086707491481,
+      "eval_iou_red bricks": 0.9041477120685042,
+      "eval_iou_road": 0.9605225675306259,
+      "eval_iou_rough concrete": 0.6186669184067215,
+      "eval_iou_rough red sidewalk": 0.6254641303235823,
+      "eval_iou_speedway bricks": 0.979499484644301,
+      "eval_iou_steel": 0.8664957832649643,
+      "eval_iou_tiles": 0.24715201139195445,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08159501105546951,
+      "eval_mean_accuracy": 0.8044214709723729,
+      "eval_mean_iou": 0.7666532356511111,
+      "eval_overall_accuracy": 0.9800159001754503,
+      "eval_runtime": 5.2209,
+      "eval_samples_per_second": 22.601,
+      "eval_steps_per_second": 1.532,
+      "step": 2820
+    },
+    {
+      "epoch": 165.94,
+      "learning_rate": 8.962848297213624e-06,
+      "loss": 0.0214,
+      "step": 2821
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 8.947368421052632e-06,
+      "loss": 0.0193,
+      "step": 2822
+    },
+    {
+      "epoch": 166.06,
+      "learning_rate": 8.93188854489164e-06,
+      "loss": 0.0234,
+      "step": 2823
+    },
+    {
+      "epoch": 166.12,
+      "learning_rate": 8.916408668730652e-06,
+      "loss": 0.0227,
+      "step": 2824
+    },
+    {
+      "epoch": 166.18,
+      "learning_rate": 8.90092879256966e-06,
+      "loss": 0.0187,
+      "step": 2825
+    },
+    {
+      "epoch": 166.24,
+      "learning_rate": 8.885448916408668e-06,
+      "loss": 0.0223,
+      "step": 2826
+    },
+    {
+      "epoch": 166.29,
+      "learning_rate": 8.869969040247678e-06,
+      "loss": 0.027,
+      "step": 2827
+    },
+    {
+      "epoch": 166.35,
+      "learning_rate": 8.854489164086688e-06,
+      "loss": 0.0221,
+      "step": 2828
+    },
+    {
+      "epoch": 166.41,
+      "learning_rate": 8.839009287925697e-06,
+      "loss": 0.0182,
+      "step": 2829
+    },
+    {
+      "epoch": 166.47,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0195,
+      "step": 2830
+    },
+    {
+      "epoch": 166.53,
+      "learning_rate": 8.808049535603715e-06,
+      "loss": 0.0218,
+      "step": 2831
+    },
+    {
+      "epoch": 166.59,
+      "learning_rate": 8.792569659442725e-06,
+      "loss": 0.0192,
+      "step": 2832
+    },
+    {
+      "epoch": 166.65,
+      "learning_rate": 8.777089783281735e-06,
+      "loss": 0.0189,
+      "step": 2833
+    },
+    {
+      "epoch": 166.71,
+      "learning_rate": 8.761609907120743e-06,
+      "loss": 0.0213,
+      "step": 2834
+    },
+    {
+      "epoch": 166.76,
+      "learning_rate": 8.746130030959752e-06,
+      "loss": 0.0233,
+      "step": 2835
+    },
+    {
+      "epoch": 166.82,
+      "learning_rate": 8.730650154798763e-06,
+      "loss": 0.0246,
+      "step": 2836
+    },
+    {
+      "epoch": 166.88,
+      "learning_rate": 8.715170278637772e-06,
+      "loss": 0.0241,
+      "step": 2837
+    },
+    {
+      "epoch": 166.94,
+      "learning_rate": 8.69969040247678e-06,
+      "loss": 0.0218,
+      "step": 2838
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 8.68421052631579e-06,
+      "loss": 0.0179,
+      "step": 2839
+    },
+    {
+      "epoch": 167.06,
+      "learning_rate": 8.6687306501548e-06,
+      "loss": 0.0235,
+      "step": 2840
+    },
+    {
+      "epoch": 167.06,
+      "eval_accuracy_NAT": 0.9932580110582722,
+      "eval_accuracy_REST": 0.3772357723577236,
+      "eval_accuracy_concrete": 0.9555295829899909,
+      "eval_accuracy_concrete tiles": 0.9379212821117134,
+      "eval_accuracy_dark bricks": 0.8621824933483911,
+      "eval_accuracy_grass": 0.9031819747395035,
+      "eval_accuracy_red bricks": 0.9265158172231986,
+      "eval_accuracy_road": 0.9821911168281762,
+      "eval_accuracy_rough concrete": 0.6231356401837694,
+      "eval_accuracy_rough red sidewalk": 0.7864249242361576,
+      "eval_accuracy_speedway bricks": 0.9898682064854172,
+      "eval_accuracy_steel": 0.9224262445836061,
+      "eval_accuracy_tiles": 0.25387015662395107,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837118504619963,
+      "eval_iou_REST": 0.3700159489633174,
+      "eval_iou_concrete": 0.8925125295724327,
+      "eval_iou_concrete tiles": 0.877240163121349,
+      "eval_iou_dark bricks": 0.8443359310929804,
+      "eval_iou_grass": 0.8411711259965966,
+      "eval_iou_red bricks": 0.9049458212638128,
+      "eval_iou_road": 0.9603020712667065,
+      "eval_iou_rough concrete": 0.6175897381382658,
+      "eval_iou_rough red sidewalk": 0.6271732285825662,
+      "eval_iou_speedway bricks": 0.979474234228307,
+      "eval_iou_steel": 0.8668567282018285,
+      "eval_iou_tiles": 0.24858354839666577,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08091432601213455,
+      "eval_mean_accuracy": 0.8087493248284514,
+      "eval_mean_iou": 0.7703009937912942,
+      "eval_overall_accuracy": 0.9800649416648736,
+      "eval_runtime": 5.3919,
+      "eval_samples_per_second": 21.885,
+      "eval_steps_per_second": 1.484,
+      "step": 2840
+    },
+    {
+      "epoch": 167.12,
+      "learning_rate": 8.653250773993808e-06,
+      "loss": 0.0205,
+      "step": 2841
+    },
+    {
+      "epoch": 167.18,
+      "learning_rate": 8.637770897832818e-06,
+      "loss": 0.0224,
+      "step": 2842
+    },
+    {
+      "epoch": 167.24,
+      "learning_rate": 8.622291021671827e-06,
+      "loss": 0.0242,
+      "step": 2843
+    },
+    {
+      "epoch": 167.29,
+      "learning_rate": 8.606811145510837e-06,
+      "loss": 0.022,
+      "step": 2844
+    },
+    {
+      "epoch": 167.35,
+      "learning_rate": 8.591331269349847e-06,
+      "loss": 0.0235,
+      "step": 2845
+    },
+    {
+      "epoch": 167.41,
+      "learning_rate": 8.575851393188855e-06,
+      "loss": 0.0224,
+      "step": 2846
+    },
+    {
+      "epoch": 167.47,
+      "learning_rate": 8.560371517027863e-06,
+      "loss": 0.0191,
+      "step": 2847
+    },
+    {
+      "epoch": 167.53,
+      "learning_rate": 8.544891640866875e-06,
+      "loss": 0.0206,
+      "step": 2848
+    },
+    {
+      "epoch": 167.59,
+      "learning_rate": 8.529411764705883e-06,
+      "loss": 0.0219,
+      "step": 2849
+    },
+    {
+      "epoch": 167.65,
+      "learning_rate": 8.513931888544892e-06,
+      "loss": 0.0209,
+      "step": 2850
+    },
+    {
+      "epoch": 167.71,
+      "learning_rate": 8.4984520123839e-06,
+      "loss": 0.0214,
+      "step": 2851
+    },
+    {
+      "epoch": 167.76,
+      "learning_rate": 8.482972136222912e-06,
+      "loss": 0.0219,
+      "step": 2852
+    },
+    {
+      "epoch": 167.82,
+      "learning_rate": 8.46749226006192e-06,
+      "loss": 0.0172,
+      "step": 2853
+    },
+    {
+      "epoch": 167.88,
+      "learning_rate": 8.452012383900928e-06,
+      "loss": 0.0218,
+      "step": 2854
+    },
+    {
+      "epoch": 167.94,
+      "learning_rate": 8.436532507739938e-06,
+      "loss": 0.0213,
+      "step": 2855
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0244,
+      "step": 2856
+    },
+    {
+      "epoch": 168.06,
+      "learning_rate": 8.405572755417957e-06,
+      "loss": 0.0188,
+      "step": 2857
+    },
+    {
+      "epoch": 168.12,
+      "learning_rate": 8.390092879256967e-06,
+      "loss": 0.0214,
+      "step": 2858
+    },
+    {
+      "epoch": 168.18,
+      "learning_rate": 8.374613003095975e-06,
+      "loss": 0.0215,
+      "step": 2859
+    },
+    {
+      "epoch": 168.24,
+      "learning_rate": 8.359133126934985e-06,
+      "loss": 0.0218,
+      "step": 2860
+    },
+    {
+      "epoch": 168.24,
+      "eval_accuracy_NAT": 0.9934739276533305,
+      "eval_accuracy_REST": 0.33495934959349594,
+      "eval_accuracy_concrete": 0.9582152974504249,
+      "eval_accuracy_concrete tiles": 0.9313693141645062,
+      "eval_accuracy_dark bricks": 0.8572009647225466,
+      "eval_accuracy_grass": 0.8960862127044326,
+      "eval_accuracy_red bricks": 0.9207491212653779,
+      "eval_accuracy_road": 0.9808614894925772,
+      "eval_accuracy_rough concrete": 0.6219617564161388,
+      "eval_accuracy_rough red sidewalk": 0.7770099038194,
+      "eval_accuracy_speedway bricks": 0.990580057345636,
+      "eval_accuracy_steel": 0.9125030611683917,
+      "eval_accuracy_tiles": 0.2425702081250568,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836868716362848,
+      "eval_iou_REST": 0.329073482428115,
+      "eval_iou_concrete": 0.8914491801833099,
+      "eval_iou_concrete tiles": 0.8748920719962806,
+      "eval_iou_dark bricks": 0.8394771866681663,
+      "eval_iou_grass": 0.8411808752639213,
+      "eval_iou_red bricks": 0.900327587132807,
+      "eval_iou_road": 0.9612753603756716,
+      "eval_iou_rough concrete": 0.6170494516863232,
+      "eval_iou_rough red sidewalk": 0.6243225116464554,
+      "eval_iou_speedway bricks": 0.9792170629935452,
+      "eval_iou_steel": 0.8658209839952579,
+      "eval_iou_tiles": 0.24000359690666026,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08251631259918213,
+      "eval_mean_accuracy": 0.8013492818401011,
+      "eval_mean_iou": 0.7652135556086767,
+      "eval_overall_accuracy": 0.9799988633495266,
+      "eval_runtime": 5.2123,
+      "eval_samples_per_second": 22.639,
+      "eval_steps_per_second": 1.535,
+      "step": 2860
+    },
+    {
+      "epoch": 168.29,
+      "learning_rate": 8.343653250773995e-06,
+      "loss": 0.0212,
+      "step": 2861
+    },
+    {
+      "epoch": 168.35,
+      "learning_rate": 8.328173374613003e-06,
+      "loss": 0.0229,
+      "step": 2862
+    },
+    {
+      "epoch": 168.41,
+      "learning_rate": 8.312693498452011e-06,
+      "loss": 0.0225,
+      "step": 2863
+    },
+    {
+      "epoch": 168.47,
+      "learning_rate": 8.297213622291023e-06,
+      "loss": 0.0222,
+      "step": 2864
+    },
+    {
+      "epoch": 168.53,
+      "learning_rate": 8.281733746130031e-06,
+      "loss": 0.0194,
+      "step": 2865
+    },
+    {
+      "epoch": 168.59,
+      "learning_rate": 8.26625386996904e-06,
+      "loss": 0.0234,
+      "step": 2866
+    },
+    {
+      "epoch": 168.65,
+      "learning_rate": 8.25077399380805e-06,
+      "loss": 0.0221,
+      "step": 2867
+    },
+    {
+      "epoch": 168.71,
+      "learning_rate": 8.23529411764706e-06,
+      "loss": 0.0248,
+      "step": 2868
+    },
+    {
+      "epoch": 168.76,
+      "learning_rate": 8.219814241486068e-06,
+      "loss": 0.0207,
+      "step": 2869
+    },
+    {
+      "epoch": 168.82,
+      "learning_rate": 8.204334365325078e-06,
+      "loss": 0.0188,
+      "step": 2870
+    },
+    {
+      "epoch": 168.88,
+      "learning_rate": 8.188854489164086e-06,
+      "loss": 0.0217,
+      "step": 2871
+    },
+    {
+      "epoch": 168.94,
+      "learning_rate": 8.173374613003096e-06,
+      "loss": 0.0209,
+      "step": 2872
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 8.157894736842106e-06,
+      "loss": 0.0196,
+      "step": 2873
+    },
+    {
+      "epoch": 169.06,
+      "learning_rate": 8.142414860681115e-06,
+      "loss": 0.0237,
+      "step": 2874
+    },
+    {
+      "epoch": 169.12,
+      "learning_rate": 8.126934984520123e-06,
+      "loss": 0.0206,
+      "step": 2875
+    },
+    {
+      "epoch": 169.18,
+      "learning_rate": 8.111455108359135e-06,
+      "loss": 0.0211,
+      "step": 2876
+    },
+    {
+      "epoch": 169.24,
+      "learning_rate": 8.095975232198143e-06,
+      "loss": 0.0223,
+      "step": 2877
+    },
+    {
+      "epoch": 169.29,
+      "learning_rate": 8.080495356037151e-06,
+      "loss": 0.0218,
+      "step": 2878
+    },
+    {
+      "epoch": 169.35,
+      "learning_rate": 8.065015479876161e-06,
+      "loss": 0.0237,
+      "step": 2879
+    },
+    {
+      "epoch": 169.41,
+      "learning_rate": 8.049535603715171e-06,
+      "loss": 0.0225,
+      "step": 2880
+    },
+    {
+      "epoch": 169.41,
+      "eval_accuracy_NAT": 0.9941456142595309,
+      "eval_accuracy_REST": 0.3089430894308943,
+      "eval_accuracy_concrete": 0.9553170801808262,
+      "eval_accuracy_concrete tiles": 0.9302616073532878,
+      "eval_accuracy_dark bricks": 0.8597467603315276,
+      "eval_accuracy_grass": 0.8912774831855326,
+      "eval_accuracy_red bricks": 0.9232205623901582,
+      "eval_accuracy_road": 0.9819297873397888,
+      "eval_accuracy_rough concrete": 0.6236302206543956,
+      "eval_accuracy_rough red sidewalk": 0.7836801299453298,
+      "eval_accuracy_speedway bricks": 0.9895869962635542,
+      "eval_accuracy_steel": 0.9214266793944634,
+      "eval_accuracy_tiles": 0.24496349480444726,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835355386718644,
+      "eval_iou_REST": 0.304,
+      "eval_iou_concrete": 0.8927851069491498,
+      "eval_iou_concrete tiles": 0.8759265012649239,
+      "eval_iou_dark bricks": 0.8426100496196453,
+      "eval_iou_grass": 0.8391563597747466,
+      "eval_iou_red bricks": 0.9028896766569986,
+      "eval_iou_road": 0.9611046655779586,
+      "eval_iou_rough concrete": 0.6184743939770001,
+      "eval_iou_rough red sidewalk": 0.6285881451724783,
+      "eval_iou_speedway bricks": 0.9794404459336035,
+      "eval_iou_steel": 0.8664952777980134,
+      "eval_iou_tiles": 0.24138754552510597,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08182083070278168,
+      "eval_mean_accuracy": 0.800625346579518,
+      "eval_mean_iou": 0.7643379774554991,
+      "eval_overall_accuracy": 0.9800418271856793,
+      "eval_runtime": 5.3416,
+      "eval_samples_per_second": 22.091,
+      "eval_steps_per_second": 1.498,
+      "step": 2880
+    },
+    {
+      "epoch": 169.47,
+      "learning_rate": 8.03405572755418e-06,
+      "loss": 0.0162,
+      "step": 2881
+    },
+    {
+      "epoch": 169.53,
+      "learning_rate": 8.01857585139319e-06,
+      "loss": 0.0217,
+      "step": 2882
+    },
+    {
+      "epoch": 169.59,
+      "learning_rate": 8.003095975232198e-06,
+      "loss": 0.0188,
+      "step": 2883
+    },
+    {
+      "epoch": 169.65,
+      "learning_rate": 7.987616099071208e-06,
+      "loss": 0.0227,
+      "step": 2884
+    },
+    {
+      "epoch": 169.71,
+      "learning_rate": 7.972136222910218e-06,
+      "loss": 0.0198,
+      "step": 2885
+    },
+    {
+      "epoch": 169.76,
+      "learning_rate": 7.956656346749226e-06,
+      "loss": 0.0199,
+      "step": 2886
+    },
+    {
+      "epoch": 169.82,
+      "learning_rate": 7.941176470588235e-06,
+      "loss": 0.0219,
+      "step": 2887
+    },
+    {
+      "epoch": 169.88,
+      "learning_rate": 7.925696594427245e-06,
+      "loss": 0.026,
+      "step": 2888
+    },
+    {
+      "epoch": 169.94,
+      "learning_rate": 7.910216718266255e-06,
+      "loss": 0.0215,
+      "step": 2889
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 7.894736842105263e-06,
+      "loss": 0.0232,
+      "step": 2890
+    },
+    {
+      "epoch": 170.06,
+      "learning_rate": 7.879256965944273e-06,
+      "loss": 0.0207,
+      "step": 2891
+    },
+    {
+      "epoch": 170.12,
+      "learning_rate": 7.863777089783283e-06,
+      "loss": 0.0217,
+      "step": 2892
+    },
+    {
+      "epoch": 170.18,
+      "learning_rate": 7.848297213622291e-06,
+      "loss": 0.0202,
+      "step": 2893
+    },
+    {
+      "epoch": 170.24,
+      "learning_rate": 7.832817337461301e-06,
+      "loss": 0.0191,
+      "step": 2894
+    },
+    {
+      "epoch": 170.29,
+      "learning_rate": 7.81733746130031e-06,
+      "loss": 0.0226,
+      "step": 2895
+    },
+    {
+      "epoch": 170.35,
+      "learning_rate": 7.80185758513932e-06,
+      "loss": 0.0245,
+      "step": 2896
+    },
+    {
+      "epoch": 170.41,
+      "learning_rate": 7.78637770897833e-06,
+      "loss": 0.0233,
+      "step": 2897
+    },
+    {
+      "epoch": 170.47,
+      "learning_rate": 7.770897832817338e-06,
+      "loss": 0.0201,
+      "step": 2898
+    },
+    {
+      "epoch": 170.53,
+      "learning_rate": 7.755417956656346e-06,
+      "loss": 0.018,
+      "step": 2899
+    },
+    {
+      "epoch": 170.59,
+      "learning_rate": 7.739938080495356e-06,
+      "loss": 0.0204,
+      "step": 2900
+    },
+    {
+      "epoch": 170.59,
+      "eval_accuracy_NAT": 0.9937209513333699,
+      "eval_accuracy_REST": 0.38699186991869916,
+      "eval_accuracy_concrete": 0.9556748187228734,
+      "eval_accuracy_concrete tiles": 0.9337968418571765,
+      "eval_accuracy_dark bricks": 0.8589045422353234,
+      "eval_accuracy_grass": 0.8985615445465108,
+      "eval_accuracy_red bricks": 0.9321726713532513,
+      "eval_accuracy_road": 0.9815300440732984,
+      "eval_accuracy_rough concrete": 0.6253344376977577,
+      "eval_accuracy_rough red sidewalk": 0.7735507110145211,
+      "eval_accuracy_speedway bricks": 0.9901903729796905,
+      "eval_accuracy_steel": 0.9196149674891422,
+      "eval_accuracy_tiles": 0.2531127874216123,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837273476144998,
+      "eval_iou_REST": 0.37658227848101267,
+      "eval_iou_concrete": 0.8927871485255489,
+      "eval_iou_concrete tiles": 0.8769976537252645,
+      "eval_iou_dark bricks": 0.8423693177018313,
+      "eval_iou_grass": 0.8411554776859621,
+      "eval_iou_red bricks": 0.906144893492072,
+      "eval_iou_road": 0.9611352096151199,
+      "eval_iou_rough concrete": 0.6191984989556413,
+      "eval_iou_rough red sidewalk": 0.6237516296273838,
+      "eval_iou_speedway bricks": 0.9793305381747947,
+      "eval_iou_steel": 0.8666632126304684,
+      "eval_iou_tiles": 0.24970113568439928,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08075252920389175,
+      "eval_mean_accuracy": 0.807935120049479,
+      "eval_mean_iou": 0.7707341801472308,
+      "eval_overall_accuracy": 0.9801128840042372,
+      "eval_runtime": 5.3903,
+      "eval_samples_per_second": 21.891,
+      "eval_steps_per_second": 1.484,
+      "step": 2900
+    },
+    {
+      "epoch": 170.65,
+      "learning_rate": 7.724458204334366e-06,
+      "loss": 0.0234,
+      "step": 2901
+    },
+    {
+      "epoch": 170.71,
+      "learning_rate": 7.708978328173374e-06,
+      "loss": 0.019,
+      "step": 2902
+    },
+    {
+      "epoch": 170.76,
+      "learning_rate": 7.693498452012384e-06,
+      "loss": 0.0224,
+      "step": 2903
+    },
+    {
+      "epoch": 170.82,
+      "learning_rate": 7.678018575851394e-06,
+      "loss": 0.0221,
+      "step": 2904
+    },
+    {
+      "epoch": 170.88,
+      "learning_rate": 7.662538699690403e-06,
+      "loss": 0.0196,
+      "step": 2905
+    },
+    {
+      "epoch": 170.94,
+      "learning_rate": 7.647058823529413e-06,
+      "loss": 0.0215,
+      "step": 2906
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 7.631578947368421e-06,
+      "loss": 0.0215,
+      "step": 2907
+    },
+    {
+      "epoch": 171.06,
+      "learning_rate": 7.61609907120743e-06,
+      "loss": 0.0223,
+      "step": 2908
+    },
+    {
+      "epoch": 171.12,
+      "learning_rate": 7.60061919504644e-06,
+      "loss": 0.0214,
+      "step": 2909
+    },
+    {
+      "epoch": 171.18,
+      "learning_rate": 7.585139318885449e-06,
+      "loss": 0.0197,
+      "step": 2910
+    },
+    {
+      "epoch": 171.24,
+      "learning_rate": 7.5696594427244586e-06,
+      "loss": 0.0213,
+      "step": 2911
+    },
+    {
+      "epoch": 171.29,
+      "learning_rate": 7.5541795665634686e-06,
+      "loss": 0.0212,
+      "step": 2912
+    },
+    {
+      "epoch": 171.35,
+      "learning_rate": 7.538699690402478e-06,
+      "loss": 0.0223,
+      "step": 2913
+    },
+    {
+      "epoch": 171.41,
+      "learning_rate": 7.523219814241486e-06,
+      "loss": 0.0213,
+      "step": 2914
+    },
+    {
+      "epoch": 171.47,
+      "learning_rate": 7.507739938080496e-06,
+      "loss": 0.0201,
+      "step": 2915
+    },
+    {
+      "epoch": 171.53,
+      "learning_rate": 7.492260061919505e-06,
+      "loss": 0.0231,
+      "step": 2916
+    },
+    {
+      "epoch": 171.59,
+      "learning_rate": 7.476780185758514e-06,
+      "loss": 0.0198,
+      "step": 2917
+    },
+    {
+      "epoch": 171.65,
+      "learning_rate": 7.461300309597524e-06,
+      "loss": 0.0257,
+      "step": 2918
+    },
+    {
+      "epoch": 171.71,
+      "learning_rate": 7.445820433436533e-06,
+      "loss": 0.0203,
+      "step": 2919
+    },
+    {
+      "epoch": 171.76,
+      "learning_rate": 7.430340557275542e-06,
+      "loss": 0.0184,
+      "step": 2920
+    },
+    {
+      "epoch": 171.76,
+      "eval_accuracy_NAT": 0.9939313689290036,
+      "eval_accuracy_REST": 0.35772357723577236,
+      "eval_accuracy_concrete": 0.957523516722748,
+      "eval_accuracy_concrete tiles": 0.9345510252180061,
+      "eval_accuracy_dark bricks": 0.8620676454261815,
+      "eval_accuracy_grass": 0.891021055460302,
+      "eval_accuracy_red bricks": 0.9336555360281195,
+      "eval_accuracy_road": 0.9812890223979145,
+      "eval_accuracy_rough concrete": 0.6253225200960558,
+      "eval_accuracy_rough red sidewalk": 0.7792809390956467,
+      "eval_accuracy_speedway bricks": 0.9891556701645723,
+      "eval_accuracy_steel": 0.9239006032375916,
+      "eval_accuracy_tiles": 0.25447605198582207,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835952775509318,
+      "eval_iou_REST": 0.3486529318541997,
+      "eval_iou_concrete": 0.8926954867781031,
+      "eval_iou_concrete tiles": 0.8714753521900618,
+      "eval_iou_dark bricks": 0.844979784050507,
+      "eval_iou_grass": 0.8385279615881573,
+      "eval_iou_red bricks": 0.9021917953616728,
+      "eval_iou_road": 0.9612035165120488,
+      "eval_iou_rough concrete": 0.6189274150265699,
+      "eval_iou_rough red sidewalk": 0.6237540328405644,
+      "eval_iou_speedway bricks": 0.9795029661467786,
+      "eval_iou_steel": 0.8667543452473052,
+      "eval_iou_tiles": 0.24884464984002844,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08169034123420715,
+      "eval_mean_accuracy": 0.8064537332305951,
+      "eval_mean_iou": 0.7677773473066869,
+      "eval_overall_accuracy": 0.9800321934586864,
+      "eval_runtime": 5.4316,
+      "eval_samples_per_second": 21.725,
+      "eval_steps_per_second": 1.473,
+      "step": 2920
+    },
+    {
+      "epoch": 171.82,
+      "learning_rate": 7.414860681114552e-06,
+      "loss": 0.0216,
+      "step": 2921
+    },
+    {
+      "epoch": 171.88,
+      "learning_rate": 7.399380804953561e-06,
+      "loss": 0.0216,
+      "step": 2922
+    },
+    {
+      "epoch": 171.94,
+      "learning_rate": 7.38390092879257e-06,
+      "loss": 0.0218,
+      "step": 2923
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0226,
+      "step": 2924
+    },
+    {
+      "epoch": 172.06,
+      "learning_rate": 7.3529411764705884e-06,
+      "loss": 0.0215,
+      "step": 2925
+    },
+    {
+      "epoch": 172.12,
+      "learning_rate": 7.337461300309598e-06,
+      "loss": 0.0241,
+      "step": 2926
+    },
+    {
+      "epoch": 172.18,
+      "learning_rate": 7.321981424148607e-06,
+      "loss": 0.0216,
+      "step": 2927
+    },
+    {
+      "epoch": 172.24,
+      "learning_rate": 7.306501547987617e-06,
+      "loss": 0.0217,
+      "step": 2928
+    },
+    {
+      "epoch": 172.29,
+      "learning_rate": 7.291021671826626e-06,
+      "loss": 0.0201,
+      "step": 2929
+    },
+    {
+      "epoch": 172.35,
+      "learning_rate": 7.275541795665634e-06,
+      "loss": 0.0209,
+      "step": 2930
+    },
+    {
+      "epoch": 172.41,
+      "learning_rate": 7.260061919504644e-06,
+      "loss": 0.0198,
+      "step": 2931
+    },
+    {
+      "epoch": 172.47,
+      "learning_rate": 7.244582043343653e-06,
+      "loss": 0.021,
+      "step": 2932
+    },
+    {
+      "epoch": 172.53,
+      "learning_rate": 7.2291021671826625e-06,
+      "loss": 0.0216,
+      "step": 2933
+    },
+    {
+      "epoch": 172.59,
+      "learning_rate": 7.2136222910216725e-06,
+      "loss": 0.0201,
+      "step": 2934
+    },
+    {
+      "epoch": 172.65,
+      "learning_rate": 7.198142414860682e-06,
+      "loss": 0.0212,
+      "step": 2935
+    },
+    {
+      "epoch": 172.71,
+      "learning_rate": 7.18266253869969e-06,
+      "loss": 0.0242,
+      "step": 2936
+    },
+    {
+      "epoch": 172.76,
+      "learning_rate": 7.1671826625387e-06,
+      "loss": 0.022,
+      "step": 2937
+    },
+    {
+      "epoch": 172.82,
+      "learning_rate": 7.151702786377709e-06,
+      "loss": 0.0204,
+      "step": 2938
+    },
+    {
+      "epoch": 172.88,
+      "learning_rate": 7.136222910216718e-06,
+      "loss": 0.0236,
+      "step": 2939
+    },
+    {
+      "epoch": 172.94,
+      "learning_rate": 7.120743034055728e-06,
+      "loss": 0.0234,
+      "step": 2940
+    },
+    {
+      "epoch": 172.94,
+      "eval_accuracy_NAT": 0.9938183698820017,
+      "eval_accuracy_REST": 0.3300813008130081,
+      "eval_accuracy_concrete": 0.9575693806383951,
+      "eval_accuracy_concrete tiles": 0.934291774687721,
+      "eval_accuracy_dark bricks": 0.8639817774630094,
+      "eval_accuracy_grass": 0.8945135821533873,
+      "eval_accuracy_red bricks": 0.9314037785588752,
+      "eval_accuracy_road": 0.9801320114728456,
+      "eval_accuracy_rough concrete": 0.6258945649777439,
+      "eval_accuracy_rough red sidewalk": 0.7919069928334549,
+      "eval_accuracy_speedway bricks": 0.9890096670512101,
+      "eval_accuracy_steel": 0.9267643575044856,
+      "eval_accuracy_tiles": 0.25296131358114454,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837181551881644,
+      "eval_iou_REST": 0.3232484076433121,
+      "eval_iou_concrete": 0.8927789645852301,
+      "eval_iou_concrete tiles": 0.8738454755869062,
+      "eval_iou_dark bricks": 0.8452700867985655,
+      "eval_iou_grass": 0.8399657384231678,
+      "eval_iou_red bricks": 0.9007807935411909,
+      "eval_iou_road": 0.9615253346027678,
+      "eval_iou_rough concrete": 0.619369409211791,
+      "eval_iou_rough red sidewalk": 0.6259034282725501,
+      "eval_iou_speedway bricks": 0.9794250345942486,
+      "eval_iou_steel": 0.8671255585145559,
+      "eval_iou_tiles": 0.24743673324245838,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08140265941619873,
+      "eval_mean_accuracy": 0.8055637593551755,
+      "eval_mean_iou": 0.7661840861696084,
+      "eval_overall_accuracy": 0.9800936165502516,
+      "eval_runtime": 5.3597,
+      "eval_samples_per_second": 22.016,
+      "eval_steps_per_second": 1.493,
+      "step": 2940
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 7.1052631578947375e-06,
+      "loss": 0.0192,
+      "step": 2941
+    },
+    {
+      "epoch": 173.06,
+      "learning_rate": 7.089783281733746e-06,
+      "loss": 0.0183,
+      "step": 2942
+    },
+    {
+      "epoch": 173.12,
+      "learning_rate": 7.074303405572756e-06,
+      "loss": 0.0209,
+      "step": 2943
+    },
+    {
+      "epoch": 173.18,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 0.022,
+      "step": 2944
+    },
+    {
+      "epoch": 173.24,
+      "learning_rate": 7.043343653250774e-06,
+      "loss": 0.0209,
+      "step": 2945
+    },
+    {
+      "epoch": 173.29,
+      "learning_rate": 7.027863777089784e-06,
+      "loss": 0.0192,
+      "step": 2946
+    },
+    {
+      "epoch": 173.35,
+      "learning_rate": 7.012383900928793e-06,
+      "loss": 0.0217,
+      "step": 2947
+    },
+    {
+      "epoch": 173.41,
+      "learning_rate": 6.9969040247678016e-06,
+      "loss": 0.0221,
+      "step": 2948
+    },
+    {
+      "epoch": 173.47,
+      "learning_rate": 6.9814241486068116e-06,
+      "loss": 0.0249,
+      "step": 2949
+    },
+    {
+      "epoch": 173.53,
+      "learning_rate": 6.965944272445821e-06,
+      "loss": 0.0212,
+      "step": 2950
+    },
+    {
+      "epoch": 173.59,
+      "learning_rate": 6.95046439628483e-06,
+      "loss": 0.0233,
+      "step": 2951
+    },
+    {
+      "epoch": 173.65,
+      "learning_rate": 6.93498452012384e-06,
+      "loss": 0.0178,
+      "step": 2952
+    },
+    {
+      "epoch": 173.71,
+      "learning_rate": 6.919504643962848e-06,
+      "loss": 0.0181,
+      "step": 2953
+    },
+    {
+      "epoch": 173.76,
+      "learning_rate": 6.904024767801857e-06,
+      "loss": 0.0214,
+      "step": 2954
+    },
+    {
+      "epoch": 173.82,
+      "learning_rate": 6.888544891640867e-06,
+      "loss": 0.0229,
+      "step": 2955
+    },
+    {
+      "epoch": 173.88,
+      "learning_rate": 6.8730650154798765e-06,
+      "loss": 0.0242,
+      "step": 2956
+    },
+    {
+      "epoch": 173.94,
+      "learning_rate": 6.857585139318886e-06,
+      "loss": 0.0186,
+      "step": 2957
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 6.842105263157896e-06,
+      "loss": 0.0215,
+      "step": 2958
+    },
+    {
+      "epoch": 174.06,
+      "learning_rate": 6.826625386996904e-06,
+      "loss": 0.0231,
+      "step": 2959
+    },
+    {
+      "epoch": 174.12,
+      "learning_rate": 6.811145510835913e-06,
+      "loss": 0.0213,
+      "step": 2960
+    },
+    {
+      "epoch": 174.12,
+      "eval_accuracy_NAT": 0.9934057831898715,
+      "eval_accuracy_REST": 0.35284552845528455,
+      "eval_accuracy_concrete": 0.9570136628604713,
+      "eval_accuracy_concrete tiles": 0.9311807683242989,
+      "eval_accuracy_dark bricks": 0.8599381735352104,
+      "eval_accuracy_grass": 0.9020313617731551,
+      "eval_accuracy_red bricks": 0.9336555360281195,
+      "eval_accuracy_road": 0.9818009930520559,
+      "eval_accuracy_rough concrete": 0.6259481941854022,
+      "eval_accuracy_rough red sidewalk": 0.7750847878236413,
+      "eval_accuracy_speedway bricks": 0.9894387654031571,
+      "eval_accuracy_steel": 0.9217715293847176,
+      "eval_accuracy_tiles": 0.24747796055621193,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983709087337114,
+      "eval_iou_REST": 0.34609250398724084,
+      "eval_iou_concrete": 0.8921336865339017,
+      "eval_iou_concrete tiles": 0.8754514635173163,
+      "eval_iou_dark bricks": 0.8432010135135135,
+      "eval_iou_grass": 0.8416207139489673,
+      "eval_iou_red bricks": 0.9034863945578231,
+      "eval_iou_road": 0.9609152704783583,
+      "eval_iou_rough concrete": 0.6192435567921903,
+      "eval_iou_rough red sidewalk": 0.6199289069595395,
+      "eval_iou_speedway bricks": 0.9795760136241345,
+      "eval_iou_steel": 0.8671924957682904,
+      "eval_iou_tiles": 0.24399641577060932,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08147281408309937,
+      "eval_mean_accuracy": 0.8055071572747383,
+      "eval_mean_iou": 0.7674267325222307,
+      "eval_overall_accuracy": 0.9800656852075609,
+      "eval_runtime": 5.4672,
+      "eval_samples_per_second": 21.583,
+      "eval_steps_per_second": 1.463,
+      "step": 2960
+    },
+    {
+      "epoch": 174.18,
+      "learning_rate": 6.795665634674923e-06,
+      "loss": 0.0239,
+      "step": 2961
+    },
+    {
+      "epoch": 174.24,
+      "learning_rate": 6.780185758513932e-06,
+      "loss": 0.022,
+      "step": 2962
+    },
+    {
+      "epoch": 174.29,
+      "learning_rate": 6.7647058823529414e-06,
+      "loss": 0.0223,
+      "step": 2963
+    },
+    {
+      "epoch": 174.35,
+      "learning_rate": 6.7492260061919514e-06,
+      "loss": 0.0229,
+      "step": 2964
+    },
+    {
+      "epoch": 174.41,
+      "learning_rate": 6.73374613003096e-06,
+      "loss": 0.0201,
+      "step": 2965
+    },
+    {
+      "epoch": 174.47,
+      "learning_rate": 6.718266253869969e-06,
+      "loss": 0.0203,
+      "step": 2966
+    },
+    {
+      "epoch": 174.53,
+      "learning_rate": 6.702786377708979e-06,
+      "loss": 0.0194,
+      "step": 2967
+    },
+    {
+      "epoch": 174.59,
+      "learning_rate": 6.687306501547988e-06,
+      "loss": 0.024,
+      "step": 2968
+    },
+    {
+      "epoch": 174.65,
+      "learning_rate": 6.671826625386997e-06,
+      "loss": 0.0226,
+      "step": 2969
+    },
+    {
+      "epoch": 174.71,
+      "learning_rate": 6.656346749226007e-06,
+      "loss": 0.0189,
+      "step": 2970
+    },
+    {
+      "epoch": 174.76,
+      "learning_rate": 6.6408668730650155e-06,
+      "loss": 0.0194,
+      "step": 2971
+    },
+    {
+      "epoch": 174.82,
+      "learning_rate": 6.625386996904025e-06,
+      "loss": 0.0241,
+      "step": 2972
+    },
+    {
+      "epoch": 174.88,
+      "learning_rate": 6.609907120743035e-06,
+      "loss": 0.0192,
+      "step": 2973
+    },
+    {
+      "epoch": 174.94,
+      "learning_rate": 6.594427244582044e-06,
+      "loss": 0.0193,
+      "step": 2974
+    },
+    {
+      "epoch": 175.0,
+      "learning_rate": 6.578947368421053e-06,
+      "loss": 0.0189,
+      "step": 2975
+    },
+    {
+      "epoch": 175.06,
+      "learning_rate": 6.563467492260063e-06,
+      "loss": 0.0205,
+      "step": 2976
+    },
+    {
+      "epoch": 175.12,
+      "learning_rate": 6.547987616099071e-06,
+      "loss": 0.0229,
+      "step": 2977
+    },
+    {
+      "epoch": 175.18,
+      "learning_rate": 6.5325077399380805e-06,
+      "loss": 0.0212,
+      "step": 2978
+    },
+    {
+      "epoch": 175.24,
+      "learning_rate": 6.5170278637770905e-06,
+      "loss": 0.0215,
+      "step": 2979
+    },
+    {
+      "epoch": 175.29,
+      "learning_rate": 6.5015479876161e-06,
+      "loss": 0.0182,
+      "step": 2980
+    },
+    {
+      "epoch": 175.29,
+      "eval_accuracy_NAT": 0.9934371059218886,
+      "eval_accuracy_REST": 0.36097560975609755,
+      "eval_accuracy_concrete": 0.9573767521926774,
+      "eval_accuracy_concrete tiles": 0.9412679707753948,
+      "eval_accuracy_dark bricks": 0.8612684953008058,
+      "eval_accuracy_grass": 0.8956982593193973,
+      "eval_accuracy_red bricks": 0.9365663444639719,
+      "eval_accuracy_road": 0.9801785056763009,
+      "eval_accuracy_rough concrete": 0.6249054040364918,
+      "eval_accuracy_rough red sidewalk": 0.7774686228652644,
+      "eval_accuracy_speedway bricks": 0.9901620977288633,
+      "eval_accuracy_steel": 0.9251225716813186,
+      "eval_accuracy_tiles": 0.24602381168772153,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836373383541799,
+      "eval_iou_REST": 0.3535031847133758,
+      "eval_iou_concrete": 0.8918223700138139,
+      "eval_iou_concrete tiles": 0.8757565125866152,
+      "eval_iou_dark bricks": 0.8435317716974589,
+      "eval_iou_grass": 0.8402099373170415,
+      "eval_iou_red bricks": 0.9022751322751322,
+      "eval_iou_road": 0.9612066028905727,
+      "eval_iou_rough concrete": 0.6178950407428575,
+      "eval_iou_rough red sidewalk": 0.6215514287260201,
+      "eval_iou_speedway bricks": 0.9793870630217538,
+      "eval_iou_steel": 0.8676627409620906,
+      "eval_iou_tiles": 0.24366167601788233,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08201245218515396,
+      "eval_mean_accuracy": 0.8069578116466302,
+      "eval_mean_iou": 0.7678539076399074,
+      "eval_overall_accuracy": 0.9800082061250331,
+      "eval_runtime": 5.308,
+      "eval_samples_per_second": 22.23,
+      "eval_steps_per_second": 1.507,
+      "step": 2980
+    },
+    {
+      "epoch": 175.35,
+      "learning_rate": 6.486068111455108e-06,
+      "loss": 0.0236,
+      "step": 2981
+    },
+    {
+      "epoch": 175.41,
+      "learning_rate": 6.470588235294119e-06,
+      "loss": 0.0203,
+      "step": 2982
+    },
+    {
+      "epoch": 175.47,
+      "learning_rate": 6.455108359133127e-06,
+      "loss": 0.0188,
+      "step": 2983
+    },
+    {
+      "epoch": 175.53,
+      "learning_rate": 6.439628482972136e-06,
+      "loss": 0.0212,
+      "step": 2984
+    },
+    {
+      "epoch": 175.59,
+      "learning_rate": 6.424148606811146e-06,
+      "loss": 0.0193,
+      "step": 2985
+    },
+    {
+      "epoch": 175.65,
+      "learning_rate": 6.408668730650155e-06,
+      "loss": 0.0189,
+      "step": 2986
+    },
+    {
+      "epoch": 175.71,
+      "learning_rate": 6.393188854489164e-06,
+      "loss": 0.0236,
+      "step": 2987
+    },
+    {
+      "epoch": 175.76,
+      "learning_rate": 6.3777089783281746e-06,
+      "loss": 0.0205,
+      "step": 2988
+    },
+    {
+      "epoch": 175.82,
+      "learning_rate": 6.362229102167183e-06,
+      "loss": 0.0221,
+      "step": 2989
+    },
+    {
+      "epoch": 175.88,
+      "learning_rate": 6.346749226006192e-06,
+      "loss": 0.0235,
+      "step": 2990
+    },
+    {
+      "epoch": 175.94,
+      "learning_rate": 6.331269349845202e-06,
+      "loss": 0.021,
+      "step": 2991
+    },
+    {
+      "epoch": 176.0,
+      "learning_rate": 6.315789473684211e-06,
+      "loss": 0.0247,
+      "step": 2992
+    },
+    {
+      "epoch": 176.06,
+      "learning_rate": 6.3003095975232195e-06,
+      "loss": 0.0242,
+      "step": 2993
+    },
+    {
+      "epoch": 176.12,
+      "learning_rate": 6.28482972136223e-06,
+      "loss": 0.0196,
+      "step": 2994
+    },
+    {
+      "epoch": 176.18,
+      "learning_rate": 6.269349845201239e-06,
+      "loss": 0.0214,
+      "step": 2995
+    },
+    {
+      "epoch": 176.24,
+      "learning_rate": 6.253869969040248e-06,
+      "loss": 0.0231,
+      "step": 2996
+    },
+    {
+      "epoch": 176.29,
+      "learning_rate": 6.238390092879257e-06,
+      "loss": 0.0208,
+      "step": 2997
+    },
+    {
+      "epoch": 176.35,
+      "learning_rate": 6.222910216718267e-06,
+      "loss": 0.02,
+      "step": 2998
+    },
+    {
+      "epoch": 176.41,
+      "learning_rate": 6.207430340557275e-06,
+      "loss": 0.0196,
+      "step": 2999
+    },
+    {
+      "epoch": 176.47,
+      "learning_rate": 6.191950464396285e-06,
+      "loss": 0.0203,
+      "step": 3000
+    },
+    {
+      "epoch": 176.47,
+      "eval_accuracy_NAT": 0.9934044893076539,
+      "eval_accuracy_REST": 0.3235772357723577,
+      "eval_accuracy_concrete": 0.9572571238126979,
+      "eval_accuracy_concrete tiles": 0.9337025689370728,
+      "eval_accuracy_dark bricks": 0.8583063759738147,
+      "eval_accuracy_grass": 0.8991879473291775,
+      "eval_accuracy_red bricks": 0.9322275922671354,
+      "eval_accuracy_road": 0.981058689045164,
+      "eval_accuracy_rough concrete": 0.6247028048075606,
+      "eval_accuracy_rough red sidewalk": 0.7840110092571008,
+      "eval_accuracy_speedway bricks": 0.9893074996932564,
+      "eval_accuracy_steel": 0.9246202901737745,
+      "eval_accuracy_tiles": 0.23184586021994003,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836463664879431,
+      "eval_iou_REST": 0.3184,
+      "eval_iou_concrete": 0.8916525213745806,
+      "eval_iou_concrete tiles": 0.8729287854750573,
+      "eval_iou_dark bricks": 0.8421186070642146,
+      "eval_iou_grass": 0.8399280178117026,
+      "eval_iou_red bricks": 0.9036895064686152,
+      "eval_iou_road": 0.9606550479762882,
+      "eval_iou_rough concrete": 0.6183213310449364,
+      "eval_iou_rough red sidewalk": 0.622693798565362,
+      "eval_iou_speedway bricks": 0.9794570237081325,
+      "eval_iou_steel": 0.8674315507657828,
+      "eval_iou_tiles": 0.23033166797086618,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08248841017484665,
+      "eval_mean_accuracy": 0.8025545758920543,
+      "eval_mean_iou": 0.7639426326702677,
+      "eval_overall_accuracy": 0.9799614922475006,
+      "eval_runtime": 5.2578,
+      "eval_samples_per_second": 22.443,
+      "eval_steps_per_second": 1.522,
+      "step": 3000
+    },
+    {
+      "epoch": 176.53,
+      "learning_rate": 6.1764705882352944e-06,
+      "loss": 0.0217,
+      "step": 3001
+    },
+    {
+      "epoch": 176.59,
+      "learning_rate": 6.160990712074304e-06,
+      "loss": 0.0194,
+      "step": 3002
+    },
+    {
+      "epoch": 176.65,
+      "learning_rate": 6.145510835913313e-06,
+      "loss": 0.0208,
+      "step": 3003
+    },
+    {
+      "epoch": 176.71,
+      "learning_rate": 6.130030959752323e-06,
+      "loss": 0.0215,
+      "step": 3004
+    },
+    {
+      "epoch": 176.76,
+      "learning_rate": 6.114551083591331e-06,
+      "loss": 0.0227,
+      "step": 3005
+    },
+    {
+      "epoch": 176.82,
+      "learning_rate": 6.099071207430341e-06,
+      "loss": 0.0216,
+      "step": 3006
+    },
+    {
+      "epoch": 176.88,
+      "learning_rate": 6.08359133126935e-06,
+      "loss": 0.0219,
+      "step": 3007
+    },
+    {
+      "epoch": 176.94,
+      "learning_rate": 6.068111455108359e-06,
+      "loss": 0.0215,
+      "step": 3008
+    },
+    {
+      "epoch": 177.0,
+      "learning_rate": 6.0526315789473685e-06,
+      "loss": 0.0176,
+      "step": 3009
+    },
+    {
+      "epoch": 177.06,
+      "learning_rate": 6.0371517027863785e-06,
+      "loss": 0.0193,
+      "step": 3010
+    },
+    {
+      "epoch": 177.12,
+      "learning_rate": 6.021671826625387e-06,
+      "loss": 0.0234,
+      "step": 3011
+    },
+    {
+      "epoch": 177.18,
+      "learning_rate": 6.006191950464397e-06,
+      "loss": 0.0215,
+      "step": 3012
+    },
+    {
+      "epoch": 177.24,
+      "learning_rate": 5.990712074303406e-06,
+      "loss": 0.0218,
+      "step": 3013
+    },
+    {
+      "epoch": 177.29,
+      "learning_rate": 5.975232198142415e-06,
+      "loss": 0.0223,
+      "step": 3014
+    },
+    {
+      "epoch": 177.35,
+      "learning_rate": 5.959752321981424e-06,
+      "loss": 0.0218,
+      "step": 3015
+    },
+    {
+      "epoch": 177.41,
+      "learning_rate": 5.9442724458204335e-06,
+      "loss": 0.0187,
+      "step": 3016
+    },
+    {
+      "epoch": 177.47,
+      "learning_rate": 5.928792569659443e-06,
+      "loss": 0.0231,
+      "step": 3017
+    },
+    {
+      "epoch": 177.53,
+      "learning_rate": 5.913312693498453e-06,
+      "loss": 0.0208,
+      "step": 3018
+    },
+    {
+      "epoch": 177.59,
+      "learning_rate": 5.897832817337461e-06,
+      "loss": 0.0245,
+      "step": 3019
+    },
+    {
+      "epoch": 177.65,
+      "learning_rate": 5.882352941176471e-06,
+      "loss": 0.024,
+      "step": 3020
+    },
+    {
+      "epoch": 177.65,
+      "eval_accuracy_NAT": 0.9938076953537067,
+      "eval_accuracy_REST": 0.3235772357723577,
+      "eval_accuracy_concrete": 0.9538719846386459,
+      "eval_accuracy_concrete tiles": 0.9345745934480321,
+      "eval_accuracy_dark bricks": 0.8583781559251957,
+      "eval_accuracy_grass": 0.8971232783385754,
+      "eval_accuracy_red bricks": 0.9300307557117751,
+      "eval_accuracy_road": 0.982477563874752,
+      "eval_accuracy_rough concrete": 0.625882647376042,
+      "eval_accuracy_rough red sidewalk": 0.7842366087878537,
+      "eval_accuracy_speedway bricks": 0.9898503845091382,
+      "eval_accuracy_steel": 0.9240580347548816,
+      "eval_accuracy_tiles": 0.24163107031415673,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836564097430494,
+      "eval_iou_REST": 0.31789137380191695,
+      "eval_iou_concrete": 0.8925666918441969,
+      "eval_iou_concrete tiles": 0.8711335676625659,
+      "eval_iou_dark bricks": 0.8414660461974368,
+      "eval_iou_grass": 0.8404061158479672,
+      "eval_iou_red bricks": 0.9034357661118224,
+      "eval_iou_road": 0.9607297188119102,
+      "eval_iou_rough concrete": 0.6197011086003552,
+      "eval_iou_rough red sidewalk": 0.6243945372138833,
+      "eval_iou_speedway bricks": 0.9793649305379529,
+      "eval_iou_steel": 0.8673188648842626,
+      "eval_iou_tiles": 0.23996630362837715,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08209843188524246,
+      "eval_mean_accuracy": 0.8030384622157778,
+      "eval_mean_iou": 0.7647716488373614,
+      "eval_overall_accuracy": 0.9800455772270591,
+      "eval_runtime": 5.4949,
+      "eval_samples_per_second": 21.475,
+      "eval_steps_per_second": 1.456,
+      "step": 3020
+    },
+    {
+      "epoch": 177.71,
+      "learning_rate": 5.86687306501548e-06,
+      "loss": 0.0235,
+      "step": 3021
+    },
+    {
+      "epoch": 177.76,
+      "learning_rate": 5.851393188854489e-06,
+      "loss": 0.0203,
+      "step": 3022
+    },
+    {
+      "epoch": 177.82,
+      "learning_rate": 5.835913312693498e-06,
+      "loss": 0.021,
+      "step": 3023
+    },
+    {
+      "epoch": 177.88,
+      "learning_rate": 5.820433436532508e-06,
+      "loss": 0.0197,
+      "step": 3024
+    },
+    {
+      "epoch": 177.94,
+      "learning_rate": 5.804953560371517e-06,
+      "loss": 0.0205,
+      "step": 3025
+    },
+    {
+      "epoch": 178.0,
+      "learning_rate": 5.789473684210527e-06,
+      "loss": 0.0251,
+      "step": 3026
+    },
+    {
+      "epoch": 178.06,
+      "learning_rate": 5.773993808049536e-06,
+      "loss": 0.0205,
+      "step": 3027
+    },
+    {
+      "epoch": 178.12,
+      "learning_rate": 5.758513931888545e-06,
+      "loss": 0.0207,
+      "step": 3028
+    },
+    {
+      "epoch": 178.18,
+      "learning_rate": 5.743034055727554e-06,
+      "loss": 0.0219,
+      "step": 3029
+    },
+    {
+      "epoch": 178.24,
+      "learning_rate": 5.727554179566564e-06,
+      "loss": 0.021,
+      "step": 3030
+    },
+    {
+      "epoch": 178.29,
+      "learning_rate": 5.7120743034055725e-06,
+      "loss": 0.0211,
+      "step": 3031
+    },
+    {
+      "epoch": 178.35,
+      "learning_rate": 5.6965944272445825e-06,
+      "loss": 0.0219,
+      "step": 3032
+    },
+    {
+      "epoch": 178.41,
+      "learning_rate": 5.681114551083592e-06,
+      "loss": 0.0234,
+      "step": 3033
+    },
+    {
+      "epoch": 178.47,
+      "learning_rate": 5.665634674922601e-06,
+      "loss": 0.0213,
+      "step": 3034
+    },
+    {
+      "epoch": 178.53,
+      "learning_rate": 5.65015479876161e-06,
+      "loss": 0.019,
+      "step": 3035
+    },
+    {
+      "epoch": 178.59,
+      "learning_rate": 5.63467492260062e-06,
+      "loss": 0.0237,
+      "step": 3036
+    },
+    {
+      "epoch": 178.65,
+      "learning_rate": 5.619195046439628e-06,
+      "loss": 0.0245,
+      "step": 3037
+    },
+    {
+      "epoch": 178.71,
+      "learning_rate": 5.603715170278638e-06,
+      "loss": 0.0223,
+      "step": 3038
+    },
+    {
+      "epoch": 178.76,
+      "learning_rate": 5.588235294117647e-06,
+      "loss": 0.0225,
+      "step": 3039
+    },
+    {
+      "epoch": 178.82,
+      "learning_rate": 5.5727554179566566e-06,
+      "loss": 0.0182,
+      "step": 3040
+    },
+    {
+      "epoch": 178.82,
+      "eval_accuracy_NAT": 0.9932880938198309,
+      "eval_accuracy_REST": 0.37398373983739835,
+      "eval_accuracy_concrete": 0.9574184119160568,
+      "eval_accuracy_concrete tiles": 0.9399717181239688,
+      "eval_accuracy_dark bricks": 0.8569234155772065,
+      "eval_accuracy_grass": 0.9010945962824604,
+      "eval_accuracy_red bricks": 0.9270650263620387,
+      "eval_accuracy_road": 0.9817512923518105,
+      "eval_accuracy_rough concrete": 0.6260792878041223,
+      "eval_accuracy_rough red sidewalk": 0.7774761428496229,
+      "eval_accuracy_speedway bricks": 0.9899631427821339,
+      "eval_accuracy_steel": 0.920457101160995,
+      "eval_accuracy_tiles": 0.23838953012814687,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838396664916071,
+      "eval_iou_REST": 0.3668261562998405,
+      "eval_iou_concrete": 0.8915596271501329,
+      "eval_iou_concrete tiles": 0.871321521420925,
+      "eval_iou_dark bricks": 0.839941462590937,
+      "eval_iou_grass": 0.8421094518916252,
+      "eval_iou_red bricks": 0.9022877913192218,
+      "eval_iou_road": 0.9606802064381985,
+      "eval_iou_rough concrete": 0.6195960489458942,
+      "eval_iou_rough red sidewalk": 0.624670710781352,
+      "eval_iou_speedway bricks": 0.9793727115309173,
+      "eval_iou_steel": 0.8675058878944889,
+      "eval_iou_tiles": 0.23656204906204906,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0814213901758194,
+      "eval_mean_accuracy": 0.8064508845381378,
+      "eval_mean_iou": 0.7681748686013221,
+      "eval_overall_accuracy": 0.9800657821913897,
+      "eval_runtime": 5.4142,
+      "eval_samples_per_second": 21.795,
+      "eval_steps_per_second": 1.478,
+      "step": 3040
+    },
+    {
+      "epoch": 178.88,
+      "learning_rate": 5.557275541795666e-06,
+      "loss": 0.0217,
+      "step": 3041
+    },
+    {
+      "epoch": 178.94,
+      "learning_rate": 5.541795665634676e-06,
+      "loss": 0.0198,
+      "step": 3042
+    },
+    {
+      "epoch": 179.0,
+      "learning_rate": 5.526315789473684e-06,
+      "loss": 0.0213,
+      "step": 3043
+    },
+    {
+      "epoch": 179.06,
+      "learning_rate": 5.510835913312694e-06,
+      "loss": 0.0201,
+      "step": 3044
+    },
+    {
+      "epoch": 179.12,
+      "learning_rate": 5.495356037151703e-06,
+      "loss": 0.0235,
+      "step": 3045
+    },
+    {
+      "epoch": 179.18,
+      "learning_rate": 5.479876160990712e-06,
+      "loss": 0.0188,
+      "step": 3046
+    },
+    {
+      "epoch": 179.24,
+      "learning_rate": 5.4643962848297215e-06,
+      "loss": 0.022,
+      "step": 3047
+    },
+    {
+      "epoch": 179.29,
+      "learning_rate": 5.4489164086687315e-06,
+      "loss": 0.0206,
+      "step": 3048
+    },
+    {
+      "epoch": 179.35,
+      "learning_rate": 5.43343653250774e-06,
+      "loss": 0.0229,
+      "step": 3049
+    },
+    {
+      "epoch": 179.41,
+      "learning_rate": 5.41795665634675e-06,
+      "loss": 0.0208,
+      "step": 3050
+    },
+    {
+      "epoch": 179.47,
+      "learning_rate": 5.402476780185759e-06,
+      "loss": 0.018,
+      "step": 3051
+    },
+    {
+      "epoch": 179.53,
+      "learning_rate": 5.386996904024768e-06,
+      "loss": 0.022,
+      "step": 3052
+    },
+    {
+      "epoch": 179.59,
+      "learning_rate": 5.371517027863777e-06,
+      "loss": 0.0199,
+      "step": 3053
+    },
+    {
+      "epoch": 179.65,
+      "learning_rate": 5.3560371517027864e-06,
+      "loss": 0.0227,
+      "step": 3054
+    },
+    {
+      "epoch": 179.71,
+      "learning_rate": 5.340557275541796e-06,
+      "loss": 0.0199,
+      "step": 3055
+    },
+    {
+      "epoch": 179.76,
+      "learning_rate": 5.325077399380806e-06,
+      "loss": 0.0203,
+      "step": 3056
+    },
+    {
+      "epoch": 179.82,
+      "learning_rate": 5.309597523219814e-06,
+      "loss": 0.0223,
+      "step": 3057
+    },
+    {
+      "epoch": 179.88,
+      "learning_rate": 5.294117647058824e-06,
+      "loss": 0.0221,
+      "step": 3058
+    },
+    {
+      "epoch": 179.94,
+      "learning_rate": 5.278637770897833e-06,
+      "loss": 0.0212,
+      "step": 3059
+    },
+    {
+      "epoch": 180.0,
+      "learning_rate": 5.263157894736842e-06,
+      "loss": 0.0227,
+      "step": 3060
+    },
+    {
+      "epoch": 180.0,
+      "eval_accuracy_NAT": 0.9931777364490233,
+      "eval_accuracy_REST": 0.34471544715447155,
+      "eval_accuracy_concrete": 0.9574275846991862,
+      "eval_accuracy_concrete tiles": 0.9367664388404431,
+      "eval_accuracy_dark bricks": 0.858397297245564,
+      "eval_accuracy_grass": 0.9025413785403112,
+      "eval_accuracy_red bricks": 0.9287126537785588,
+      "eval_accuracy_road": 0.9817245715452269,
+      "eval_accuracy_rough concrete": 0.6249351980407463,
+      "eval_accuracy_rough red sidewalk": 0.7809578956075771,
+      "eval_accuracy_speedway bricks": 0.989646631338026,
+      "eval_accuracy_steel": 0.9228110771814261,
+      "eval_accuracy_tiles": 0.24075252203944378,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837651350616297,
+      "eval_iou_REST": 0.33865814696485624,
+      "eval_iou_concrete": 0.8918068550686767,
+      "eval_iou_concrete tiles": 0.8728314814001493,
+      "eval_iou_dark bricks": 0.840944357715802,
+      "eval_iou_grass": 0.8411171026582775,
+      "eval_iou_red bricks": 0.904955581718934,
+      "eval_iou_road": 0.9605460616651451,
+      "eval_iou_rough concrete": 0.6185841856295196,
+      "eval_iou_rough red sidewalk": 0.626106301396291,
+      "eval_iou_speedway bricks": 0.979459931959371,
+      "eval_iou_steel": 0.8672468900667664,
+      "eval_iou_tiles": 0.23905065575742993,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08132944256067276,
+      "eval_mean_accuracy": 0.8048128024969234,
+      "eval_mean_iou": 0.7665440528509885,
+      "eval_overall_accuracy": 0.9800361697956667,
+      "eval_runtime": 5.4124,
+      "eval_samples_per_second": 21.802,
+      "eval_steps_per_second": 1.478,
+      "step": 3060
+    },
+    {
+      "epoch": 180.06,
+      "learning_rate": 5.247678018575851e-06,
+      "loss": 0.0219,
+      "step": 3061
+    },
+    {
+      "epoch": 180.12,
+      "learning_rate": 5.232198142414861e-06,
+      "loss": 0.021,
+      "step": 3062
+    },
+    {
+      "epoch": 180.18,
+      "learning_rate": 5.21671826625387e-06,
+      "loss": 0.0196,
+      "step": 3063
+    },
+    {
+      "epoch": 180.24,
+      "learning_rate": 5.20123839009288e-06,
+      "loss": 0.0209,
+      "step": 3064
+    },
+    {
+      "epoch": 180.29,
+      "learning_rate": 5.185758513931889e-06,
+      "loss": 0.0195,
+      "step": 3065
+    },
+    {
+      "epoch": 180.35,
+      "learning_rate": 5.170278637770898e-06,
+      "loss": 0.0212,
+      "step": 3066
+    },
+    {
+      "epoch": 180.41,
+      "learning_rate": 5.154798761609907e-06,
+      "loss": 0.0221,
+      "step": 3067
+    },
+    {
+      "epoch": 180.47,
+      "learning_rate": 5.139318885448917e-06,
+      "loss": 0.0229,
+      "step": 3068
+    },
+    {
+      "epoch": 180.53,
+      "learning_rate": 5.1238390092879255e-06,
+      "loss": 0.0185,
+      "step": 3069
+    },
+    {
+      "epoch": 180.59,
+      "learning_rate": 5.1083591331269355e-06,
+      "loss": 0.0212,
+      "step": 3070
+    },
+    {
+      "epoch": 180.65,
+      "learning_rate": 5.092879256965945e-06,
+      "loss": 0.0221,
+      "step": 3071
+    },
+    {
+      "epoch": 180.71,
+      "learning_rate": 5.077399380804954e-06,
+      "loss": 0.0195,
+      "step": 3072
+    },
+    {
+      "epoch": 180.76,
+      "learning_rate": 5.061919504643963e-06,
+      "loss": 0.023,
+      "step": 3073
+    },
+    {
+      "epoch": 180.82,
+      "learning_rate": 5.046439628482973e-06,
+      "loss": 0.0204,
+      "step": 3074
+    },
+    {
+      "epoch": 180.88,
+      "learning_rate": 5.030959752321981e-06,
+      "loss": 0.0211,
+      "step": 3075
+    },
+    {
+      "epoch": 180.94,
+      "learning_rate": 5.015479876160991e-06,
+      "loss": 0.0206,
+      "step": 3076
+    },
+    {
+      "epoch": 181.0,
+      "learning_rate": 5e-06,
+      "loss": 0.0216,
+      "step": 3077
+    },
+    {
+      "epoch": 181.06,
+      "learning_rate": 4.9845201238390096e-06,
+      "loss": 0.0221,
+      "step": 3078
+    },
+    {
+      "epoch": 181.12,
+      "learning_rate": 4.969040247678019e-06,
+      "loss": 0.0211,
+      "step": 3079
+    },
+    {
+      "epoch": 181.18,
+      "learning_rate": 4.953560371517029e-06,
+      "loss": 0.0198,
+      "step": 3080
+    },
+    {
+      "epoch": 181.18,
+      "eval_accuracy_NAT": 0.9939725575129298,
+      "eval_accuracy_REST": 0.359349593495935,
+      "eval_accuracy_concrete": 0.9563501648807767,
+      "eval_accuracy_concrete tiles": 0.9416686306858355,
+      "eval_accuracy_dark bricks": 0.8583494439446433,
+      "eval_accuracy_grass": 0.8938152660603239,
+      "eval_accuracy_red bricks": 0.9339850615114236,
+      "eval_accuracy_road": 0.9814236952630957,
+      "eval_accuracy_rough concrete": 0.6247623928160697,
+      "eval_accuracy_rough red sidewalk": 0.7770851036629843,
+      "eval_accuracy_speedway bricks": 0.9888592084437782,
+      "eval_accuracy_steel": 0.9281687465952311,
+      "eval_accuracy_tiles": 0.24781120300524098,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836857370632204,
+      "eval_iou_REST": 0.35079365079365077,
+      "eval_iou_concrete": 0.8918446806796788,
+      "eval_iou_concrete tiles": 0.8712195547415015,
+      "eval_iou_dark bricks": 0.8428652654233099,
+      "eval_iou_grass": 0.8405542657767044,
+      "eval_iou_red bricks": 0.9036611934746799,
+      "eval_iou_road": 0.9605801640753968,
+      "eval_iou_rough concrete": 0.6177390751029594,
+      "eval_iou_rough red sidewalk": 0.6239569118552779,
+      "eval_iou_speedway bricks": 0.9794151527870869,
+      "eval_iou_steel": 0.8673242311096167,
+      "eval_iou_tiles": 0.24560139314237675,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08163498342037201,
+      "eval_mean_accuracy": 0.8065846975290974,
+      "eval_mean_iou": 0.7676339443096508,
+      "eval_overall_accuracy": 0.9800258248539294,
+      "eval_runtime": 5.4565,
+      "eval_samples_per_second": 21.626,
+      "eval_steps_per_second": 1.466,
+      "step": 3080
+    },
+    {
+      "epoch": 181.24,
+      "learning_rate": 4.938080495356037e-06,
+      "loss": 0.0192,
+      "step": 3081
+    },
+    {
+      "epoch": 181.29,
+      "learning_rate": 4.922600619195047e-06,
+      "loss": 0.0204,
+      "step": 3082
+    },
+    {
+      "epoch": 181.35,
+      "learning_rate": 4.907120743034056e-06,
+      "loss": 0.0198,
+      "step": 3083
+    },
+    {
+      "epoch": 181.41,
+      "learning_rate": 4.891640866873065e-06,
+      "loss": 0.02,
+      "step": 3084
+    },
+    {
+      "epoch": 181.47,
+      "learning_rate": 4.8761609907120745e-06,
+      "loss": 0.0235,
+      "step": 3085
+    },
+    {
+      "epoch": 181.53,
+      "learning_rate": 4.8606811145510845e-06,
+      "loss": 0.0208,
+      "step": 3086
+    },
+    {
+      "epoch": 181.59,
+      "learning_rate": 4.845201238390093e-06,
+      "loss": 0.0195,
+      "step": 3087
+    },
+    {
+      "epoch": 181.65,
+      "learning_rate": 4.829721362229103e-06,
+      "loss": 0.0233,
+      "step": 3088
+    },
+    {
+      "epoch": 181.71,
+      "learning_rate": 4.814241486068111e-06,
+      "loss": 0.0202,
+      "step": 3089
+    },
+    {
+      "epoch": 181.76,
+      "learning_rate": 4.798761609907121e-06,
+      "loss": 0.0254,
+      "step": 3090
+    },
+    {
+      "epoch": 181.82,
+      "learning_rate": 4.78328173374613e-06,
+      "loss": 0.0255,
+      "step": 3091
+    },
+    {
+      "epoch": 181.88,
+      "learning_rate": 4.7678018575851394e-06,
+      "loss": 0.0188,
+      "step": 3092
+    },
+    {
+      "epoch": 181.94,
+      "learning_rate": 4.752321981424149e-06,
+      "loss": 0.0196,
+      "step": 3093
+    },
+    {
+      "epoch": 182.0,
+      "learning_rate": 4.736842105263159e-06,
+      "loss": 0.0219,
+      "step": 3094
+    },
+    {
+      "epoch": 182.06,
+      "learning_rate": 4.721362229102167e-06,
+      "loss": 0.0222,
+      "step": 3095
+    },
+    {
+      "epoch": 182.12,
+      "learning_rate": 4.705882352941177e-06,
+      "loss": 0.0187,
+      "step": 3096
+    },
+    {
+      "epoch": 182.18,
+      "learning_rate": 4.690402476780186e-06,
+      "loss": 0.0212,
+      "step": 3097
+    },
+    {
+      "epoch": 182.24,
+      "learning_rate": 4.674922600619195e-06,
+      "loss": 0.0221,
+      "step": 3098
+    },
+    {
+      "epoch": 182.29,
+      "learning_rate": 4.659442724458204e-06,
+      "loss": 0.0215,
+      "step": 3099
+    },
+    {
+      "epoch": 182.35,
+      "learning_rate": 4.643962848297214e-06,
+      "loss": 0.0249,
+      "step": 3100
+    },
+    {
+      "epoch": 182.35,
+      "eval_accuracy_NAT": 0.9938610679951818,
+      "eval_accuracy_REST": 0.3121951219512195,
+      "eval_accuracy_concrete": 0.9549792160022259,
+      "eval_accuracy_concrete tiles": 0.9415507895357059,
+      "eval_accuracy_dark bricks": 0.8633931818616848,
+      "eval_accuracy_grass": 0.899016680103101,
+      "eval_accuracy_red bricks": 0.9334358523725835,
+      "eval_accuracy_road": 0.9814258329276224,
+      "eval_accuracy_rough concrete": 0.6255787485326453,
+      "eval_accuracy_rough red sidewalk": 0.7774310229434723,
+      "eval_accuracy_speedway bricks": 0.989921672414254,
+      "eval_accuracy_steel": 0.9186104044740537,
+      "eval_accuracy_tiles": 0.25338544033445426,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837510749378784,
+      "eval_iou_REST": 0.3062200956937799,
+      "eval_iou_concrete": 0.8928906572173698,
+      "eval_iou_concrete tiles": 0.8700291824556818,
+      "eval_iou_dark bricks": 0.8448247606115235,
+      "eval_iou_grass": 0.8416232836122339,
+      "eval_iou_red bricks": 0.9050053248136315,
+      "eval_iou_road": 0.9610930150283548,
+      "eval_iou_rough concrete": 0.6180037203607337,
+      "eval_iou_rough red sidewalk": 0.6212337904262862,
+      "eval_iou_speedway bricks": 0.979519664560789,
+      "eval_iou_steel": 0.8676945446305493,
+      "eval_iou_tiles": 0.25009718027688904,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08121335506439209,
+      "eval_mean_accuracy": 0.8034450024190929,
+      "eval_mean_iou": 0.7647681765096692,
+      "eval_overall_accuracy": 0.9801421407925881,
+      "eval_runtime": 5.0994,
+      "eval_samples_per_second": 23.14,
+      "eval_steps_per_second": 1.569,
+      "step": 3100
+    },
+    {
+      "epoch": 182.41,
+      "learning_rate": 4.628482972136223e-06,
+      "loss": 0.0213,
+      "step": 3101
+    },
+    {
+      "epoch": 182.47,
+      "learning_rate": 4.613003095975233e-06,
+      "loss": 0.0205,
+      "step": 3102
+    },
+    {
+      "epoch": 182.53,
+      "learning_rate": 4.597523219814242e-06,
+      "loss": 0.0234,
+      "step": 3103
+    },
+    {
+      "epoch": 182.59,
+      "learning_rate": 4.582043343653251e-06,
+      "loss": 0.0214,
+      "step": 3104
+    },
+    {
+      "epoch": 182.65,
+      "learning_rate": 4.56656346749226e-06,
+      "loss": 0.0177,
+      "step": 3105
+    },
+    {
+      "epoch": 182.71,
+      "learning_rate": 4.55108359133127e-06,
+      "loss": 0.0209,
+      "step": 3106
+    },
+    {
+      "epoch": 182.76,
+      "learning_rate": 4.5356037151702785e-06,
+      "loss": 0.0198,
+      "step": 3107
+    },
+    {
+      "epoch": 182.82,
+      "learning_rate": 4.5201238390092885e-06,
+      "loss": 0.0196,
+      "step": 3108
+    },
+    {
+      "epoch": 182.88,
+      "learning_rate": 4.504643962848298e-06,
+      "loss": 0.0211,
+      "step": 3109
+    },
+    {
+      "epoch": 182.94,
+      "learning_rate": 4.489164086687307e-06,
+      "loss": 0.0217,
+      "step": 3110
+    },
+    {
+      "epoch": 183.0,
+      "learning_rate": 4.473684210526316e-06,
+      "loss": 0.0182,
+      "step": 3111
+    },
+    {
+      "epoch": 183.06,
+      "learning_rate": 4.458204334365326e-06,
+      "loss": 0.0196,
+      "step": 3112
+    },
+    {
+      "epoch": 183.12,
+      "learning_rate": 4.442724458204334e-06,
+      "loss": 0.0189,
+      "step": 3113
+    },
+    {
+      "epoch": 183.18,
+      "learning_rate": 4.427244582043344e-06,
+      "loss": 0.0245,
+      "step": 3114
+    },
+    {
+      "epoch": 183.24,
+      "learning_rate": 4.411764705882353e-06,
+      "loss": 0.0226,
+      "step": 3115
+    },
+    {
+      "epoch": 183.29,
+      "learning_rate": 4.3962848297213626e-06,
+      "loss": 0.0222,
+      "step": 3116
+    },
+    {
+      "epoch": 183.35,
+      "learning_rate": 4.380804953560372e-06,
+      "loss": 0.0242,
+      "step": 3117
+    },
+    {
+      "epoch": 183.41,
+      "learning_rate": 4.365325077399382e-06,
+      "loss": 0.0186,
+      "step": 3118
+    },
+    {
+      "epoch": 183.47,
+      "learning_rate": 4.34984520123839e-06,
+      "loss": 0.0222,
+      "step": 3119
+    },
+    {
+      "epoch": 183.53,
+      "learning_rate": 4.3343653250774e-06,
+      "loss": 0.0201,
+      "step": 3120
+    },
+    {
+      "epoch": 183.53,
+      "eval_accuracy_NAT": 0.9934409875685414,
+      "eval_accuracy_REST": 0.3008130081300813,
+      "eval_accuracy_concrete": 0.9577899096327982,
+      "eval_accuracy_concrete tiles": 0.9378034409615837,
+      "eval_accuracy_dark bricks": 0.854420687939054,
+      "eval_accuracy_grass": 0.8991926784680194,
+      "eval_accuracy_red bricks": 0.9277240773286467,
+      "eval_accuracy_road": 0.9808144608729902,
+      "eval_accuracy_rough concrete": 0.6232548162007877,
+      "eval_accuracy_rough red sidewalk": 0.7769497439445326,
+      "eval_accuracy_speedway bricks": 0.989906763645636,
+      "eval_accuracy_steel": 0.9198748544383193,
+      "eval_accuracy_tiles": 0.2465388227453119,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837633800109976,
+      "eval_iou_REST": 0.2955271565495208,
+      "eval_iou_concrete": 0.8919677922943257,
+      "eval_iou_concrete tiles": 0.8724374575193492,
+      "eval_iou_dark bricks": 0.8381763385941359,
+      "eval_iou_grass": 0.8405245384284721,
+      "eval_iou_red bricks": 0.9022058430807028,
+      "eval_iou_road": 0.9609162338889677,
+      "eval_iou_rough concrete": 0.6165717586861434,
+      "eval_iou_rough red sidewalk": 0.6207820611420881,
+      "eval_iou_speedway bricks": 0.9794039406110813,
+      "eval_iou_steel": 0.8677746346063178,
+      "eval_iou_tiles": 0.24379868184541642,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08220455795526505,
+      "eval_mean_accuracy": 0.8006557116827924,
+      "eval_mean_iou": 0.7626038320967324,
+      "eval_overall_accuracy": 0.9800192299535719,
+      "eval_runtime": 5.4212,
+      "eval_samples_per_second": 21.766,
+      "eval_steps_per_second": 1.476,
+      "step": 3120
+    },
+    {
+      "epoch": 183.59,
+      "learning_rate": 4.318885448916409e-06,
+      "loss": 0.0252,
+      "step": 3121
+    },
+    {
+      "epoch": 183.65,
+      "learning_rate": 4.303405572755418e-06,
+      "loss": 0.0242,
+      "step": 3122
+    },
+    {
+      "epoch": 183.71,
+      "learning_rate": 4.2879256965944275e-06,
+      "loss": 0.0188,
+      "step": 3123
+    },
+    {
+      "epoch": 183.76,
+      "learning_rate": 4.2724458204334375e-06,
+      "loss": 0.0214,
+      "step": 3124
+    },
+    {
+      "epoch": 183.82,
+      "learning_rate": 4.256965944272446e-06,
+      "loss": 0.0199,
+      "step": 3125
+    },
+    {
+      "epoch": 183.88,
+      "learning_rate": 4.241486068111456e-06,
+      "loss": 0.0211,
+      "step": 3126
+    },
+    {
+      "epoch": 183.94,
+      "learning_rate": 4.226006191950464e-06,
+      "loss": 0.0168,
+      "step": 3127
+    },
+    {
+      "epoch": 184.0,
+      "learning_rate": 4.210526315789474e-06,
+      "loss": 0.0212,
+      "step": 3128
+    },
+    {
+      "epoch": 184.06,
+      "learning_rate": 4.195046439628483e-06,
+      "loss": 0.0188,
+      "step": 3129
+    },
+    {
+      "epoch": 184.12,
+      "learning_rate": 4.1795665634674924e-06,
+      "loss": 0.019,
+      "step": 3130
+    },
+    {
+      "epoch": 184.18,
+      "learning_rate": 4.164086687306502e-06,
+      "loss": 0.024,
+      "step": 3131
+    },
+    {
+      "epoch": 184.24,
+      "learning_rate": 4.148606811145512e-06,
+      "loss": 0.0183,
+      "step": 3132
+    },
+    {
+      "epoch": 184.29,
+      "learning_rate": 4.13312693498452e-06,
+      "loss": 0.0209,
+      "step": 3133
+    },
+    {
+      "epoch": 184.35,
+      "learning_rate": 4.11764705882353e-06,
+      "loss": 0.0204,
+      "step": 3134
+    },
+    {
+      "epoch": 184.41,
+      "learning_rate": 4.102167182662539e-06,
+      "loss": 0.0227,
+      "step": 3135
+    },
+    {
+      "epoch": 184.47,
+      "learning_rate": 4.086687306501548e-06,
+      "loss": 0.0214,
+      "step": 3136
+    },
+    {
+      "epoch": 184.53,
+      "learning_rate": 4.071207430340557e-06,
+      "loss": 0.0215,
+      "step": 3137
+    },
+    {
+      "epoch": 184.59,
+      "learning_rate": 4.055727554179567e-06,
+      "loss": 0.0241,
+      "step": 3138
+    },
+    {
+      "epoch": 184.65,
+      "learning_rate": 4.040247678018576e-06,
+      "loss": 0.0218,
+      "step": 3139
+    },
+    {
+      "epoch": 184.71,
+      "learning_rate": 4.024767801857586e-06,
+      "loss": 0.0262,
+      "step": 3140
+    },
+    {
+      "epoch": 184.71,
+      "eval_accuracy_NAT": 0.9936063349335961,
+      "eval_accuracy_REST": 0.3382113821138211,
+      "eval_accuracy_concrete": 0.9553862582535938,
+      "eval_accuracy_concrete tiles": 0.930520857883573,
+      "eval_accuracy_dark bricks": 0.8553729686273759,
+      "eval_accuracy_grass": 0.8991340123463799,
+      "eval_accuracy_red bricks": 0.9307996485061512,
+      "eval_accuracy_road": 0.9814274361760175,
+      "eval_accuracy_rough concrete": 0.6253582729011614,
+      "eval_accuracy_rough red sidewalk": 0.7830785311966552,
+      "eval_accuracy_speedway bricks": 0.9904625008482575,
+      "eval_accuracy_steel": 0.9193475838010465,
+      "eval_accuracy_tiles": 0.24750825532430548,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837519127673471,
+      "eval_iou_REST": 0.33226837060702874,
+      "eval_iou_concrete": 0.8927496304476165,
+      "eval_iou_concrete tiles": 0.8766180421412554,
+      "eval_iou_dark bricks": 0.8390396169733383,
+      "eval_iou_grass": 0.8407976619094937,
+      "eval_iou_red bricks": 0.9037487335359676,
+      "eval_iou_road": 0.9610110970750735,
+      "eval_iou_rough concrete": 0.6179568860441974,
+      "eval_iou_rough red sidewalk": 0.6239544136327685,
+      "eval_iou_speedway bricks": 0.9793561787352019,
+      "eval_iou_steel": 0.867446954496047,
+      "eval_iou_tiles": 0.24552229835316744,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08066441118717194,
+      "eval_mean_accuracy": 0.8038626186855333,
+      "eval_mean_iou": 0.7664785997475771,
+      "eval_overall_accuracy": 0.9800859548277774,
+      "eval_runtime": 5.4023,
+      "eval_samples_per_second": 21.842,
+      "eval_steps_per_second": 1.481,
+      "step": 3140
+    },
+    {
+      "epoch": 184.76,
+      "learning_rate": 4.009287925696595e-06,
+      "loss": 0.0216,
+      "step": 3141
+    },
+    {
+      "epoch": 184.82,
+      "learning_rate": 3.993808049535604e-06,
+      "loss": 0.0196,
+      "step": 3142
+    },
+    {
+      "epoch": 184.88,
+      "learning_rate": 3.978328173374613e-06,
+      "loss": 0.0217,
+      "step": 3143
+    },
+    {
+      "epoch": 184.94,
+      "learning_rate": 3.962848297213622e-06,
+      "loss": 0.0219,
+      "step": 3144
+    },
+    {
+      "epoch": 185.0,
+      "learning_rate": 3.9473684210526315e-06,
+      "loss": 0.0159,
+      "step": 3145
+    },
+    {
+      "epoch": 185.06,
+      "learning_rate": 3.9318885448916415e-06,
+      "loss": 0.0209,
+      "step": 3146
+    },
+    {
+      "epoch": 185.12,
+      "learning_rate": 3.916408668730651e-06,
+      "loss": 0.0198,
+      "step": 3147
+    },
+    {
+      "epoch": 185.18,
+      "learning_rate": 3.90092879256966e-06,
+      "loss": 0.0208,
+      "step": 3148
+    },
+    {
+      "epoch": 185.24,
+      "learning_rate": 3.885448916408669e-06,
+      "loss": 0.0211,
+      "step": 3149
+    },
+    {
+      "epoch": 185.29,
+      "learning_rate": 3.869969040247678e-06,
+      "loss": 0.0235,
+      "step": 3150
+    },
+    {
+      "epoch": 185.35,
+      "learning_rate": 3.854489164086687e-06,
+      "loss": 0.0248,
+      "step": 3151
+    },
+    {
+      "epoch": 185.41,
+      "learning_rate": 3.839009287925697e-06,
+      "loss": 0.0204,
+      "step": 3152
+    },
+    {
+      "epoch": 185.47,
+      "learning_rate": 3.823529411764706e-06,
+      "loss": 0.0237,
+      "step": 3153
+    },
+    {
+      "epoch": 185.53,
+      "learning_rate": 3.808049535603715e-06,
+      "loss": 0.0186,
+      "step": 3154
+    },
+    {
+      "epoch": 185.59,
+      "learning_rate": 3.7925696594427247e-06,
+      "loss": 0.0224,
+      "step": 3155
+    },
+    {
+      "epoch": 185.65,
+      "learning_rate": 3.7770897832817343e-06,
+      "loss": 0.021,
+      "step": 3156
+    },
+    {
+      "epoch": 185.71,
+      "learning_rate": 3.761609907120743e-06,
+      "loss": 0.0213,
+      "step": 3157
+    },
+    {
+      "epoch": 185.76,
+      "learning_rate": 3.7461300309597526e-06,
+      "loss": 0.0207,
+      "step": 3158
+    },
+    {
+      "epoch": 185.82,
+      "learning_rate": 3.730650154798762e-06,
+      "loss": 0.0167,
+      "step": 3159
+    },
+    {
+      "epoch": 185.88,
+      "learning_rate": 3.715170278637771e-06,
+      "loss": 0.0217,
+      "step": 3160
+    },
+    {
+      "epoch": 185.88,
+      "eval_accuracy_NAT": 0.9937045082468549,
+      "eval_accuracy_REST": 0.37398373983739835,
+      "eval_accuracy_concrete": 0.957069846157139,
+      "eval_accuracy_concrete tiles": 0.9371670987508838,
+      "eval_accuracy_dark bricks": 0.8573971632563214,
+      "eval_accuracy_grass": 0.8970570423947889,
+      "eval_accuracy_red bricks": 0.9321726713532513,
+      "eval_accuracy_road": 0.9815963116736257,
+      "eval_accuracy_rough concrete": 0.6256979245496637,
+      "eval_accuracy_rough red sidewalk": 0.7807548560298995,
+      "eval_accuracy_speedway bricks": 0.9893938677321467,
+      "eval_accuracy_steel": 0.923580742377066,
+      "eval_accuracy_tiles": 0.24593292738344089,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837567472291228,
+      "eval_iou_REST": 0.36507936507936506,
+      "eval_iou_concrete": 0.8923564747651522,
+      "eval_iou_concrete tiles": 0.8743760582272357,
+      "eval_iou_dark bricks": 0.8418747797486198,
+      "eval_iou_grass": 0.8408106250415733,
+      "eval_iou_red bricks": 0.9036362668370335,
+      "eval_iou_road": 0.9608476214788806,
+      "eval_iou_rough concrete": 0.6182633937245712,
+      "eval_iou_rough red sidewalk": 0.6234028244788165,
+      "eval_iou_speedway bricks": 0.9795671282409054,
+      "eval_iou_steel": 0.8684290910458001,
+      "eval_iou_tiles": 0.24365939310262028,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08127845823764801,
+      "eval_mean_accuracy": 0.8073468230571138,
+      "eval_mean_iou": 0.7689276745384382,
+      "eval_overall_accuracy": 0.9800941337973385,
+      "eval_runtime": 5.3122,
+      "eval_samples_per_second": 22.213,
+      "eval_steps_per_second": 1.506,
+      "step": 3160
+    },
+    {
+      "epoch": 185.94,
+      "learning_rate": 3.6996904024767805e-06,
+      "loss": 0.0227,
+      "step": 3161
+    },
+    {
+      "epoch": 186.0,
+      "learning_rate": 3.6842105263157892e-06,
+      "loss": 0.0183,
+      "step": 3162
+    },
+    {
+      "epoch": 186.06,
+      "learning_rate": 3.668730650154799e-06,
+      "loss": 0.0214,
+      "step": 3163
+    },
+    {
+      "epoch": 186.12,
+      "learning_rate": 3.6532507739938084e-06,
+      "loss": 0.0212,
+      "step": 3164
+    },
+    {
+      "epoch": 186.18,
+      "learning_rate": 3.637770897832817e-06,
+      "loss": 0.0257,
+      "step": 3165
+    },
+    {
+      "epoch": 186.24,
+      "learning_rate": 3.6222910216718267e-06,
+      "loss": 0.019,
+      "step": 3166
+    },
+    {
+      "epoch": 186.29,
+      "learning_rate": 3.6068111455108363e-06,
+      "loss": 0.0196,
+      "step": 3167
+    },
+    {
+      "epoch": 186.35,
+      "learning_rate": 3.591331269349845e-06,
+      "loss": 0.0219,
+      "step": 3168
+    },
+    {
+      "epoch": 186.41,
+      "learning_rate": 3.5758513931888546e-06,
+      "loss": 0.0199,
+      "step": 3169
+    },
+    {
+      "epoch": 186.47,
+      "learning_rate": 3.560371517027864e-06,
+      "loss": 0.0211,
+      "step": 3170
+    },
+    {
+      "epoch": 186.53,
+      "learning_rate": 3.544891640866873e-06,
+      "loss": 0.0214,
+      "step": 3171
+    },
+    {
+      "epoch": 186.59,
+      "learning_rate": 3.5294117647058825e-06,
+      "loss": 0.022,
+      "step": 3172
+    },
+    {
+      "epoch": 186.65,
+      "learning_rate": 3.513931888544892e-06,
+      "loss": 0.0231,
+      "step": 3173
+    },
+    {
+      "epoch": 186.71,
+      "learning_rate": 3.4984520123839008e-06,
+      "loss": 0.0226,
+      "step": 3174
+    },
+    {
+      "epoch": 186.76,
+      "learning_rate": 3.4829721362229104e-06,
+      "loss": 0.0199,
+      "step": 3175
+    },
+    {
+      "epoch": 186.82,
+      "learning_rate": 3.46749226006192e-06,
+      "loss": 0.0201,
+      "step": 3176
+    },
+    {
+      "epoch": 186.88,
+      "learning_rate": 3.4520123839009287e-06,
+      "loss": 0.0209,
+      "step": 3177
+    },
+    {
+      "epoch": 186.94,
+      "learning_rate": 3.4365325077399382e-06,
+      "loss": 0.0183,
+      "step": 3178
+    },
+    {
+      "epoch": 187.0,
+      "learning_rate": 3.421052631578948e-06,
+      "loss": 0.0227,
+      "step": 3179
+    },
+    {
+      "epoch": 187.06,
+      "learning_rate": 3.4055727554179566e-06,
+      "loss": 0.0193,
+      "step": 3180
+    },
+    {
+      "epoch": 187.06,
+      "eval_accuracy_NAT": 0.9937103307168339,
+      "eval_accuracy_REST": 0.3853658536585366,
+      "eval_accuracy_concrete": 0.9560757457854884,
+      "eval_accuracy_concrete tiles": 0.936177233089795,
+      "eval_accuracy_dark bricks": 0.8591725207204793,
+      "eval_accuracy_grass": 0.8967145079426359,
+      "eval_accuracy_red bricks": 0.9305250439367311,
+      "eval_accuracy_road": 0.9825277989911292,
+      "eval_accuracy_rough concrete": 0.6256204601386017,
+      "eval_accuracy_rough red sidewalk": 0.7831687710089563,
+      "eval_accuracy_speedway bricks": 0.9894730384344628,
+      "eval_accuracy_steel": 0.9242154662721717,
+      "eval_accuracy_tiles": 0.2481747402223636,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837198263476608,
+      "eval_iou_REST": 0.375,
+      "eval_iou_concrete": 0.8928228281818831,
+      "eval_iou_concrete tiles": 0.8724932457662486,
+      "eval_iou_dark bricks": 0.8426123644282167,
+      "eval_iou_grass": 0.8407774018998544,
+      "eval_iou_red bricks": 0.9028562293509539,
+      "eval_iou_road": 0.960602640667534,
+      "eval_iou_rough concrete": 0.6182924243851879,
+      "eval_iou_rough red sidewalk": 0.6243517880159468,
+      "eval_iou_speedway bricks": 0.979585182194271,
+      "eval_iou_steel": 0.8681120844247904,
+      "eval_iou_tiles": 0.2456887502624239,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0812520980834961,
+      "eval_mean_accuracy": 0.8085324239167835,
+      "eval_mean_iou": 0.7697626743019209,
+      "eval_overall_accuracy": 0.9801035735566738,
+      "eval_runtime": 5.3941,
+      "eval_samples_per_second": 21.876,
+      "eval_steps_per_second": 1.483,
+      "step": 3180
+    },
+    {
+      "epoch": 187.12,
+      "learning_rate": 3.390092879256966e-06,
+      "loss": 0.0215,
+      "step": 3181
+    },
+    {
+      "epoch": 187.18,
+      "learning_rate": 3.3746130030959757e-06,
+      "loss": 0.0191,
+      "step": 3182
+    },
+    {
+      "epoch": 187.24,
+      "learning_rate": 3.3591331269349844e-06,
+      "loss": 0.0226,
+      "step": 3183
+    },
+    {
+      "epoch": 187.29,
+      "learning_rate": 3.343653250773994e-06,
+      "loss": 0.0217,
+      "step": 3184
+    },
+    {
+      "epoch": 187.35,
+      "learning_rate": 3.3281733746130036e-06,
+      "loss": 0.0192,
+      "step": 3185
+    },
+    {
+      "epoch": 187.41,
+      "learning_rate": 3.3126934984520123e-06,
+      "loss": 0.0201,
+      "step": 3186
+    },
+    {
+      "epoch": 187.47,
+      "learning_rate": 3.297213622291022e-06,
+      "loss": 0.0223,
+      "step": 3187
+    },
+    {
+      "epoch": 187.53,
+      "learning_rate": 3.2817337461300315e-06,
+      "loss": 0.0205,
+      "step": 3188
+    },
+    {
+      "epoch": 187.59,
+      "learning_rate": 3.2662538699690402e-06,
+      "loss": 0.0217,
+      "step": 3189
+    },
+    {
+      "epoch": 187.65,
+      "learning_rate": 3.25077399380805e-06,
+      "loss": 0.0224,
+      "step": 3190
+    },
+    {
+      "epoch": 187.71,
+      "learning_rate": 3.2352941176470594e-06,
+      "loss": 0.0181,
+      "step": 3191
+    },
+    {
+      "epoch": 187.76,
+      "learning_rate": 3.219814241486068e-06,
+      "loss": 0.0231,
+      "step": 3192
+    },
+    {
+      "epoch": 187.82,
+      "learning_rate": 3.2043343653250777e-06,
+      "loss": 0.0198,
+      "step": 3193
+    },
+    {
+      "epoch": 187.88,
+      "learning_rate": 3.1888544891640873e-06,
+      "loss": 0.0231,
+      "step": 3194
+    },
+    {
+      "epoch": 187.94,
+      "learning_rate": 3.173374613003096e-06,
+      "loss": 0.02,
+      "step": 3195
+    },
+    {
+      "epoch": 188.0,
+      "learning_rate": 3.1578947368421056e-06,
+      "loss": 0.0219,
+      "step": 3196
+    },
+    {
+      "epoch": 188.06,
+      "learning_rate": 3.142414860681115e-06,
+      "loss": 0.0197,
+      "step": 3197
+    },
+    {
+      "epoch": 188.12,
+      "learning_rate": 3.126934984520124e-06,
+      "loss": 0.0202,
+      "step": 3198
+    },
+    {
+      "epoch": 188.18,
+      "learning_rate": 3.1114551083591335e-06,
+      "loss": 0.0175,
+      "step": 3199
+    },
+    {
+      "epoch": 188.24,
+      "learning_rate": 3.0959752321981426e-06,
+      "loss": 0.0252,
+      "step": 3200
+    },
+    {
+      "epoch": 188.24,
+      "eval_accuracy_NAT": 0.9937598217116563,
+      "eval_accuracy_REST": 0.36910569105691055,
+      "eval_accuracy_concrete": 0.9568282962013976,
+      "eval_accuracy_concrete tiles": 0.9325712938958284,
+      "eval_accuracy_dark bricks": 0.8564879505388282,
+      "eval_accuracy_grass": 0.8948021816227427,
+      "eval_accuracy_red bricks": 0.9250329525483304,
+      "eval_accuracy_road": 0.9813237594464731,
+      "eval_accuracy_rough concrete": 0.6245895875913932,
+      "eval_accuracy_rough red sidewalk": 0.773498071124012,
+      "eval_accuracy_speedway bricks": 0.9898867139223222,
+      "eval_accuracy_steel": 0.9237756575889489,
+      "eval_accuracy_tiles": 0.24517555818110212,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837247188180968,
+      "eval_iou_REST": 0.3603174603174603,
+      "eval_iou_concrete": 0.8922803853570042,
+      "eval_iou_concrete tiles": 0.8751105803255484,
+      "eval_iou_dark bricks": 0.8404094473400009,
+      "eval_iou_grass": 0.8403375383445123,
+      "eval_iou_red bricks": 0.9006470242233036,
+      "eval_iou_road": 0.9608004725898415,
+      "eval_iou_rough concrete": 0.6178048119201707,
+      "eval_iou_rough red sidewalk": 0.6209567390700651,
+      "eval_iou_speedway bricks": 0.9794282753377359,
+      "eval_iou_steel": 0.8682573832328859,
+      "eval_iou_tiles": 0.24277786110694466,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08172224462032318,
+      "eval_mean_accuracy": 0.8051413488792266,
+      "eval_mean_iou": 0.7679117459987362,
+      "eval_overall_accuracy": 0.9800540794760494,
+      "eval_runtime": 5.288,
+      "eval_samples_per_second": 22.315,
+      "eval_steps_per_second": 1.513,
+      "step": 3200
+    },
+    {
+      "epoch": 188.29,
+      "learning_rate": 3.080495356037152e-06,
+      "loss": 0.0183,
+      "step": 3201
+    },
+    {
+      "epoch": 188.35,
+      "learning_rate": 3.0650154798761614e-06,
+      "loss": 0.0202,
+      "step": 3202
+    },
+    {
+      "epoch": 188.41,
+      "learning_rate": 3.0495356037151705e-06,
+      "loss": 0.0226,
+      "step": 3203
+    },
+    {
+      "epoch": 188.47,
+      "learning_rate": 3.0340557275541797e-06,
+      "loss": 0.0196,
+      "step": 3204
+    },
+    {
+      "epoch": 188.53,
+      "learning_rate": 3.0185758513931893e-06,
+      "loss": 0.0187,
+      "step": 3205
+    },
+    {
+      "epoch": 188.59,
+      "learning_rate": 3.0030959752321984e-06,
+      "loss": 0.0223,
+      "step": 3206
+    },
+    {
+      "epoch": 188.65,
+      "learning_rate": 2.9876160990712076e-06,
+      "loss": 0.0205,
+      "step": 3207
+    },
+    {
+      "epoch": 188.71,
+      "learning_rate": 2.9721362229102167e-06,
+      "loss": 0.021,
+      "step": 3208
+    },
+    {
+      "epoch": 188.76,
+      "learning_rate": 2.9566563467492263e-06,
+      "loss": 0.0228,
+      "step": 3209
+    },
+    {
+      "epoch": 188.82,
+      "learning_rate": 2.9411764705882355e-06,
+      "loss": 0.0206,
+      "step": 3210
+    },
+    {
+      "epoch": 188.88,
+      "learning_rate": 2.9256965944272446e-06,
+      "loss": 0.0241,
+      "step": 3211
+    },
+    {
+      "epoch": 188.94,
+      "learning_rate": 2.910216718266254e-06,
+      "loss": 0.0211,
+      "step": 3212
+    },
+    {
+      "epoch": 189.0,
+      "learning_rate": 2.8947368421052634e-06,
+      "loss": 0.0217,
+      "step": 3213
+    },
+    {
+      "epoch": 189.06,
+      "learning_rate": 2.8792569659442725e-06,
+      "loss": 0.0205,
+      "step": 3214
+    },
+    {
+      "epoch": 189.12,
+      "learning_rate": 2.863777089783282e-06,
+      "loss": 0.0226,
+      "step": 3215
+    },
+    {
+      "epoch": 189.18,
+      "learning_rate": 2.8482972136222912e-06,
+      "loss": 0.0214,
+      "step": 3216
+    },
+    {
+      "epoch": 189.24,
+      "learning_rate": 2.8328173374613004e-06,
+      "loss": 0.0219,
+      "step": 3217
+    },
+    {
+      "epoch": 189.29,
+      "learning_rate": 2.81733746130031e-06,
+      "loss": 0.0183,
+      "step": 3218
+    },
+    {
+      "epoch": 189.35,
+      "learning_rate": 2.801857585139319e-06,
+      "loss": 0.0197,
+      "step": 3219
+    },
+    {
+      "epoch": 189.41,
+      "learning_rate": 2.7863777089783283e-06,
+      "loss": 0.0204,
+      "step": 3220
+    },
+    {
+      "epoch": 189.41,
+      "eval_accuracy_NAT": 0.9938298530866827,
+      "eval_accuracy_REST": 0.36260162601626017,
+      "eval_accuracy_concrete": 0.9558013266901999,
+      "eval_accuracy_concrete tiles": 0.9326419985859062,
+      "eval_accuracy_dark bricks": 0.8602013666902742,
+      "eval_accuracy_grass": 0.89683467886922,
+      "eval_accuracy_red bricks": 0.929371704745167,
+      "eval_accuracy_road": 0.980541374229706,
+      "eval_accuracy_rough concrete": 0.6238983666926867,
+      "eval_accuracy_rough red sidewalk": 0.7823265327608119,
+      "eval_accuracy_speedway bricks": 0.9902160777531698,
+      "eval_accuracy_steel": 0.9196524511837351,
+      "eval_accuracy_tiles": 0.24899269896088946,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837232222573651,
+      "eval_iou_REST": 0.35566188197767146,
+      "eval_iou_concrete": 0.8927093060779224,
+      "eval_iou_concrete tiles": 0.8750027639579878,
+      "eval_iou_dark bricks": 0.8431069982974452,
+      "eval_iou_grass": 0.8402884535252574,
+      "eval_iou_red bricks": 0.9028919005442322,
+      "eval_iou_road": 0.9609646570691884,
+      "eval_iou_rough concrete": 0.6174959748524113,
+      "eval_iou_rough red sidewalk": 0.6246239935636104,
+      "eval_iou_speedway bricks": 0.9795402411082926,
+      "eval_iou_steel": 0.8681992969874259,
+      "eval_iou_tiles": 0.24563658099222951,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08185406774282455,
+      "eval_mean_accuracy": 0.8059161581742084,
+      "eval_mean_iou": 0.7684496362470029,
+      "eval_overall_accuracy": 0.9801059981523934,
+      "eval_runtime": 5.2883,
+      "eval_samples_per_second": 22.313,
+      "eval_steps_per_second": 1.513,
+      "step": 3220
+    },
+    {
+      "epoch": 189.47,
+      "learning_rate": 2.770897832817338e-06,
+      "loss": 0.0228,
+      "step": 3221
+    },
+    {
+      "epoch": 189.53,
+      "learning_rate": 2.755417956656347e-06,
+      "loss": 0.0223,
+      "step": 3222
+    },
+    {
+      "epoch": 189.59,
+      "learning_rate": 2.739938080495356e-06,
+      "loss": 0.0208,
+      "step": 3223
+    },
+    {
+      "epoch": 189.65,
+      "learning_rate": 2.7244582043343658e-06,
+      "loss": 0.0237,
+      "step": 3224
+    },
+    {
+      "epoch": 189.71,
+      "learning_rate": 2.708978328173375e-06,
+      "loss": 0.0206,
+      "step": 3225
+    },
+    {
+      "epoch": 189.76,
+      "learning_rate": 2.693498452012384e-06,
+      "loss": 0.0247,
+      "step": 3226
+    },
+    {
+      "epoch": 189.82,
+      "learning_rate": 2.6780185758513932e-06,
+      "loss": 0.0219,
+      "step": 3227
+    },
+    {
+      "epoch": 189.88,
+      "learning_rate": 2.662538699690403e-06,
+      "loss": 0.0204,
+      "step": 3228
+    },
+    {
+      "epoch": 189.94,
+      "learning_rate": 2.647058823529412e-06,
+      "loss": 0.0195,
+      "step": 3229
+    },
+    {
+      "epoch": 190.0,
+      "learning_rate": 2.631578947368421e-06,
+      "loss": 0.0214,
+      "step": 3230
+    },
+    {
+      "epoch": 190.06,
+      "learning_rate": 2.6160990712074307e-06,
+      "loss": 0.0199,
+      "step": 3231
+    },
+    {
+      "epoch": 190.12,
+      "learning_rate": 2.60061919504644e-06,
+      "loss": 0.022,
+      "step": 3232
+    },
+    {
+      "epoch": 190.18,
+      "learning_rate": 2.585139318885449e-06,
+      "loss": 0.0203,
+      "step": 3233
+    },
+    {
+      "epoch": 190.24,
+      "learning_rate": 2.5696594427244586e-06,
+      "loss": 0.0238,
+      "step": 3234
+    },
+    {
+      "epoch": 190.29,
+      "learning_rate": 2.5541795665634677e-06,
+      "loss": 0.0173,
+      "step": 3235
+    },
+    {
+      "epoch": 190.35,
+      "learning_rate": 2.538699690402477e-06,
+      "loss": 0.024,
+      "step": 3236
+    },
+    {
+      "epoch": 190.41,
+      "learning_rate": 2.5232198142414865e-06,
+      "loss": 0.0205,
+      "step": 3237
+    },
+    {
+      "epoch": 190.47,
+      "learning_rate": 2.5077399380804956e-06,
+      "loss": 0.0194,
+      "step": 3238
+    },
+    {
+      "epoch": 190.53,
+      "learning_rate": 2.4922600619195048e-06,
+      "loss": 0.0202,
+      "step": 3239
+    },
+    {
+      "epoch": 190.59,
+      "learning_rate": 2.4767801857585144e-06,
+      "loss": 0.0249,
+      "step": 3240
+    },
+    {
+      "epoch": 190.59,
+      "eval_accuracy_NAT": 0.9935654159084651,
+      "eval_accuracy_REST": 0.33658536585365856,
+      "eval_accuracy_concrete": 0.9575946057920011,
+      "eval_accuracy_concrete tiles": 0.9338911147772803,
+      "eval_accuracy_dark bricks": 0.860866527573072,
+      "eval_accuracy_grass": 0.8991595604961261,
+      "eval_accuracy_red bricks": 0.9281085237258347,
+      "eval_accuracy_road": 0.9814140757727257,
+      "eval_accuracy_rough concrete": 0.6232965278067442,
+      "eval_accuracy_rough red sidewalk": 0.7887185194654796,
+      "eval_accuracy_speedway bricks": 0.9893966095746511,
+      "eval_accuracy_steel": 0.9214741587409477,
+      "eval_accuracy_tiles": 0.2499924263079766,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837959441256314,
+      "eval_iou_REST": 0.3306709265175719,
+      "eval_iou_concrete": 0.892698463145638,
+      "eval_iou_concrete tiles": 0.8724898714109565,
+      "eval_iou_dark bricks": 0.8438183063313226,
+      "eval_iou_grass": 0.8409398633975048,
+      "eval_iou_red bricks": 0.902531510361034,
+      "eval_iou_road": 0.9608914139326463,
+      "eval_iou_rough concrete": 0.6170313172843803,
+      "eval_iou_rough red sidewalk": 0.6281622825930717,
+      "eval_iou_speedway bricks": 0.9795796485664254,
+      "eval_iou_steel": 0.8679125379527856,
+      "eval_iou_tiles": 0.24638719694255345,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08217969536781311,
+      "eval_mean_accuracy": 0.8049279562919202,
+      "eval_mean_iou": 0.7666853294278092,
+      "eval_overall_accuracy": 0.9801305350610765,
+      "eval_runtime": 5.2766,
+      "eval_samples_per_second": 22.363,
+      "eval_steps_per_second": 1.516,
+      "step": 3240
+    },
+    {
+      "epoch": 190.65,
+      "learning_rate": 2.4613003095975235e-06,
+      "loss": 0.021,
+      "step": 3241
+    },
+    {
+      "epoch": 190.71,
+      "learning_rate": 2.4458204334365327e-06,
+      "loss": 0.0196,
+      "step": 3242
+    },
+    {
+      "epoch": 190.76,
+      "learning_rate": 2.4303405572755423e-06,
+      "loss": 0.0188,
+      "step": 3243
+    },
+    {
+      "epoch": 190.82,
+      "learning_rate": 2.4148606811145514e-06,
+      "loss": 0.0217,
+      "step": 3244
+    },
+    {
+      "epoch": 190.88,
+      "learning_rate": 2.3993808049535606e-06,
+      "loss": 0.0215,
+      "step": 3245
+    },
+    {
+      "epoch": 190.94,
+      "learning_rate": 2.3839009287925697e-06,
+      "loss": 0.0225,
+      "step": 3246
+    },
+    {
+      "epoch": 191.0,
+      "learning_rate": 2.3684210526315793e-06,
+      "loss": 0.0229,
+      "step": 3247
+    },
+    {
+      "epoch": 191.06,
+      "learning_rate": 2.3529411764705885e-06,
+      "loss": 0.023,
+      "step": 3248
+    },
+    {
+      "epoch": 191.12,
+      "learning_rate": 2.3374613003095976e-06,
+      "loss": 0.0217,
+      "step": 3249
+    },
+    {
+      "epoch": 191.18,
+      "learning_rate": 2.321981424148607e-06,
+      "loss": 0.0221,
+      "step": 3250
+    },
+    {
+      "epoch": 191.24,
+      "learning_rate": 2.3065015479876163e-06,
+      "loss": 0.0216,
+      "step": 3251
+    },
+    {
+      "epoch": 191.29,
+      "learning_rate": 2.2910216718266255e-06,
+      "loss": 0.0201,
+      "step": 3252
+    },
+    {
+      "epoch": 191.35,
+      "learning_rate": 2.275541795665635e-06,
+      "loss": 0.0218,
+      "step": 3253
+    },
+    {
+      "epoch": 191.41,
+      "learning_rate": 2.2600619195046442e-06,
+      "loss": 0.0212,
+      "step": 3254
+    },
+    {
+      "epoch": 191.47,
+      "learning_rate": 2.2445820433436534e-06,
+      "loss": 0.0218,
+      "step": 3255
+    },
+    {
+      "epoch": 191.53,
+      "learning_rate": 2.229102167182663e-06,
+      "loss": 0.0189,
+      "step": 3256
+    },
+    {
+      "epoch": 191.59,
+      "learning_rate": 2.213622291021672e-06,
+      "loss": 0.022,
+      "step": 3257
+    },
+    {
+      "epoch": 191.65,
+      "learning_rate": 2.1981424148606813e-06,
+      "loss": 0.0216,
+      "step": 3258
+    },
+    {
+      "epoch": 191.71,
+      "learning_rate": 2.182662538699691e-06,
+      "loss": 0.018,
+      "step": 3259
+    },
+    {
+      "epoch": 191.76,
+      "learning_rate": 2.1671826625387e-06,
+      "loss": 0.0215,
+      "step": 3260
+    },
+    {
+      "epoch": 191.76,
+      "eval_accuracy_NAT": 0.9938773493464197,
+      "eval_accuracy_REST": 0.3463414634146341,
+      "eval_accuracy_concrete": 0.9556847559045969,
+      "eval_accuracy_concrete tiles": 0.9361065283997172,
+      "eval_accuracy_dark bricks": 0.8646947916467278,
+      "eval_accuracy_grass": 0.8961732656591234,
+      "eval_accuracy_red bricks": 0.9288774165202109,
+      "eval_accuracy_road": 0.9820991972535286,
+      "eval_accuracy_rough concrete": 0.6244108235658656,
+      "eval_accuracy_rough red sidewalk": 0.7838606095699321,
+      "eval_accuracy_speedway bricks": 0.9893791303286852,
+      "eval_accuracy_steel": 0.9244503640916202,
+      "eval_accuracy_tiles": 0.2525977763640219,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837204704609227,
+      "eval_iou_REST": 0.3397129186602871,
+      "eval_iou_concrete": 0.8932266623990672,
+      "eval_iou_concrete tiles": 0.8726957133126799,
+      "eval_iou_dark bricks": 0.8465581942290664,
+      "eval_iou_grass": 0.8405928432084324,
+      "eval_iou_red bricks": 0.9026525057373112,
+      "eval_iou_road": 0.9608628657864837,
+      "eval_iou_rough concrete": 0.6172023630719936,
+      "eval_iou_rough red sidewalk": 0.625454523635229,
+      "eval_iou_speedway bricks": 0.9796387994732514,
+      "eval_iou_steel": 0.8678479100296992,
+      "eval_iou_tiles": 0.2487247561375772,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08203808963298798,
+      "eval_mean_accuracy": 0.8060425747742372,
+      "eval_mean_iou": 0.7676069635493848,
+      "eval_overall_accuracy": 0.9801481537899729,
+      "eval_runtime": 5.2644,
+      "eval_samples_per_second": 22.414,
+      "eval_steps_per_second": 1.52,
+      "step": 3260
+    },
+    {
+      "epoch": 191.82,
+      "learning_rate": 2.151702786377709e-06,
+      "loss": 0.0197,
+      "step": 3261
+    },
+    {
+      "epoch": 191.88,
+      "learning_rate": 2.1362229102167187e-06,
+      "loss": 0.0201,
+      "step": 3262
+    },
+    {
+      "epoch": 191.94,
+      "learning_rate": 2.120743034055728e-06,
+      "loss": 0.0219,
+      "step": 3263
+    },
+    {
+      "epoch": 192.0,
+      "learning_rate": 2.105263157894737e-06,
+      "loss": 0.0194,
+      "step": 3264
+    },
+    {
+      "epoch": 192.06,
+      "learning_rate": 2.0897832817337462e-06,
+      "loss": 0.0198,
+      "step": 3265
+    },
+    {
+      "epoch": 192.12,
+      "learning_rate": 2.074303405572756e-06,
+      "loss": 0.0234,
+      "step": 3266
+    },
+    {
+      "epoch": 192.18,
+      "learning_rate": 2.058823529411765e-06,
+      "loss": 0.0191,
+      "step": 3267
+    },
+    {
+      "epoch": 192.24,
+      "learning_rate": 2.043343653250774e-06,
+      "loss": 0.0202,
+      "step": 3268
+    },
+    {
+      "epoch": 192.29,
+      "learning_rate": 2.0278637770897837e-06,
+      "loss": 0.0215,
+      "step": 3269
+    },
+    {
+      "epoch": 192.35,
+      "learning_rate": 2.012383900928793e-06,
+      "loss": 0.0227,
+      "step": 3270
+    },
+    {
+      "epoch": 192.41,
+      "learning_rate": 1.996904024767802e-06,
+      "loss": 0.0218,
+      "step": 3271
+    },
+    {
+      "epoch": 192.47,
+      "learning_rate": 1.981424148606811e-06,
+      "loss": 0.0198,
+      "step": 3272
+    },
+    {
+      "epoch": 192.53,
+      "learning_rate": 1.9659442724458207e-06,
+      "loss": 0.0198,
+      "step": 3273
+    },
+    {
+      "epoch": 192.59,
+      "learning_rate": 1.95046439628483e-06,
+      "loss": 0.0224,
+      "step": 3274
+    },
+    {
+      "epoch": 192.65,
+      "learning_rate": 1.934984520123839e-06,
+      "loss": 0.0216,
+      "step": 3275
+    },
+    {
+      "epoch": 192.71,
+      "learning_rate": 1.9195046439628486e-06,
+      "loss": 0.0198,
+      "step": 3276
+    },
+    {
+      "epoch": 192.76,
+      "learning_rate": 1.9040247678018576e-06,
+      "loss": 0.0203,
+      "step": 3277
+    },
+    {
+      "epoch": 192.82,
+      "learning_rate": 1.8885448916408671e-06,
+      "loss": 0.021,
+      "step": 3278
+    },
+    {
+      "epoch": 192.88,
+      "learning_rate": 1.8730650154798763e-06,
+      "loss": 0.0235,
+      "step": 3279
+    },
+    {
+      "epoch": 192.94,
+      "learning_rate": 1.8575851393188855e-06,
+      "loss": 0.0225,
+      "step": 3280
+    },
+    {
+      "epoch": 192.94,
+      "eval_accuracy_NAT": 0.9936639666040373,
+      "eval_accuracy_REST": 0.3821138211382114,
+      "eval_accuracy_concrete": 0.9572005583167331,
+      "eval_accuracy_concrete tiles": 0.9379919868017912,
+      "eval_accuracy_dark bricks": 0.8614981911452252,
+      "eval_accuracy_grass": 0.893787825455041,
+      "eval_accuracy_red bricks": 0.9264059753954306,
+      "eval_accuracy_road": 0.9819500951527922,
+      "eval_accuracy_rough concrete": 0.6239877487054505,
+      "eval_accuracy_rough red sidewalk": 0.7830183713217876,
+      "eval_accuracy_speedway bricks": 0.9896745638585401,
+      "eval_accuracy_steel": 0.9225586869711675,
+      "eval_accuracy_tiles": 0.24608440122390862,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837712700811525,
+      "eval_iou_REST": 0.37183544303797467,
+      "eval_iou_concrete": 0.8920704892889022,
+      "eval_iou_concrete tiles": 0.8717527489376615,
+      "eval_iou_dark bricks": 0.8438121039409052,
+      "eval_iou_grass": 0.8402271490024871,
+      "eval_iou_red bricks": 0.8987159678192764,
+      "eval_iou_road": 0.9606607311563057,
+      "eval_iou_rough concrete": 0.6165590169629241,
+      "eval_iou_rough red sidewalk": 0.6232865232434245,
+      "eval_iou_speedway bricks": 0.9794889059432599,
+      "eval_iou_steel": 0.8678799960506458,
+      "eval_iou_tiles": 0.2435317043921451,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08237060904502869,
+      "eval_mean_accuracy": 0.8076873993915475,
+      "eval_mean_iou": 0.7687378499890051,
+      "eval_overall_accuracy": 0.980056536399712,
+      "eval_runtime": 5.2094,
+      "eval_samples_per_second": 22.651,
+      "eval_steps_per_second": 1.536,
+      "step": 3280
+    },
+    {
+      "epoch": 193.0,
+      "learning_rate": 1.8421052631578946e-06,
+      "loss": 0.0256,
+      "step": 3281
+    },
+    {
+      "epoch": 193.06,
+      "learning_rate": 1.8266253869969042e-06,
+      "loss": 0.0209,
+      "step": 3282
+    },
+    {
+      "epoch": 193.12,
+      "learning_rate": 1.8111455108359133e-06,
+      "loss": 0.02,
+      "step": 3283
+    },
+    {
+      "epoch": 193.18,
+      "learning_rate": 1.7956656346749225e-06,
+      "loss": 0.0252,
+      "step": 3284
+    },
+    {
+      "epoch": 193.24,
+      "learning_rate": 1.780185758513932e-06,
+      "loss": 0.019,
+      "step": 3285
+    },
+    {
+      "epoch": 193.29,
+      "learning_rate": 1.7647058823529412e-06,
+      "loss": 0.0197,
+      "step": 3286
+    },
+    {
+      "epoch": 193.35,
+      "learning_rate": 1.7492260061919504e-06,
+      "loss": 0.0213,
+      "step": 3287
+    },
+    {
+      "epoch": 193.41,
+      "learning_rate": 1.73374613003096e-06,
+      "loss": 0.0209,
+      "step": 3288
+    },
+    {
+      "epoch": 193.47,
+      "learning_rate": 1.7182662538699691e-06,
+      "loss": 0.0221,
+      "step": 3289
+    },
+    {
+      "epoch": 193.53,
+      "learning_rate": 1.7027863777089783e-06,
+      "loss": 0.0176,
+      "step": 3290
+    },
+    {
+      "epoch": 193.59,
+      "learning_rate": 1.6873065015479879e-06,
+      "loss": 0.0182,
+      "step": 3291
+    },
+    {
+      "epoch": 193.65,
+      "learning_rate": 1.671826625386997e-06,
+      "loss": 0.019,
+      "step": 3292
+    },
+    {
+      "epoch": 193.71,
+      "learning_rate": 1.6563467492260062e-06,
+      "loss": 0.0213,
+      "step": 3293
+    },
+    {
+      "epoch": 193.76,
+      "learning_rate": 1.6408668730650157e-06,
+      "loss": 0.0259,
+      "step": 3294
+    },
+    {
+      "epoch": 193.82,
+      "learning_rate": 1.625386996904025e-06,
+      "loss": 0.0228,
+      "step": 3295
+    },
+    {
+      "epoch": 193.88,
+      "learning_rate": 1.609907120743034e-06,
+      "loss": 0.0204,
+      "step": 3296
+    },
+    {
+      "epoch": 193.94,
+      "learning_rate": 1.5944272445820436e-06,
+      "loss": 0.0191,
+      "step": 3297
+    },
+    {
+      "epoch": 194.0,
+      "learning_rate": 1.5789473684210528e-06,
+      "loss": 0.0198,
+      "step": 3298
+    },
+    {
+      "epoch": 194.06,
+      "learning_rate": 1.563467492260062e-06,
+      "loss": 0.0214,
+      "step": 3299
+    },
+    {
+      "epoch": 194.12,
+      "learning_rate": 1.5479876160990713e-06,
+      "loss": 0.0221,
+      "step": 3300
+    },
+    {
+      "epoch": 194.12,
+      "eval_accuracy_NAT": 0.9936930789539329,
+      "eval_accuracy_REST": 0.367479674796748,
+      "eval_accuracy_concrete": 0.9553564467084232,
+      "eval_accuracy_concrete tiles": 0.9394532170633986,
+      "eval_accuracy_dark bricks": 0.8610579407767548,
+      "eval_accuracy_grass": 0.897142202893943,
+      "eval_accuracy_red bricks": 0.9268453427065027,
+      "eval_accuracy_road": 0.9818795522234116,
+      "eval_accuracy_rough concrete": 0.6245419171845857,
+      "eval_accuracy_rough red sidewalk": 0.7910647545853104,
+      "eval_accuracy_speedway bricks": 0.9898666641990084,
+      "eval_accuracy_steel": 0.9237931499797588,
+      "eval_accuracy_tiles": 0.24953800478657337,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837538331473746,
+      "eval_iou_REST": 0.36044657097288674,
+      "eval_iou_concrete": 0.8929545786611481,
+      "eval_iou_concrete tiles": 0.8719838995471748,
+      "eval_iou_dark bricks": 0.8433532215655304,
+      "eval_iou_grass": 0.8407964875657451,
+      "eval_iou_red bricks": 0.9010144153764015,
+      "eval_iou_road": 0.9609569746362598,
+      "eval_iou_rough concrete": 0.6169540212970103,
+      "eval_iou_rough red sidewalk": 0.626246450406901,
+      "eval_iou_speedway bricks": 0.9794984519934844,
+      "eval_iou_steel": 0.8679272747762554,
+      "eval_iou_tiles": 0.24717921017885008,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08160500973463058,
+      "eval_mean_accuracy": 0.8078239959121809,
+      "eval_mean_iou": 0.7686973377019247,
+      "eval_overall_accuracy": 0.9801208366781978,
+      "eval_runtime": 5.3699,
+      "eval_samples_per_second": 21.974,
+      "eval_steps_per_second": 1.49,
+      "step": 3300
+    },
+    {
+      "epoch": 194.18,
+      "learning_rate": 1.5325077399380807e-06,
+      "loss": 0.0219,
+      "step": 3301
+    },
+    {
+      "epoch": 194.24,
+      "learning_rate": 1.5170278637770898e-06,
+      "loss": 0.0212,
+      "step": 3302
+    },
+    {
+      "epoch": 194.29,
+      "learning_rate": 1.5015479876160992e-06,
+      "loss": 0.0208,
+      "step": 3303
+    },
+    {
+      "epoch": 194.35,
+      "learning_rate": 1.4860681114551084e-06,
+      "loss": 0.0198,
+      "step": 3304
+    },
+    {
+      "epoch": 194.41,
+      "learning_rate": 1.4705882352941177e-06,
+      "loss": 0.0198,
+      "step": 3305
+    },
+    {
+      "epoch": 194.47,
+      "learning_rate": 1.455108359133127e-06,
+      "loss": 0.0223,
+      "step": 3306
+    },
+    {
+      "epoch": 194.53,
+      "learning_rate": 1.4396284829721363e-06,
+      "loss": 0.0236,
+      "step": 3307
+    },
+    {
+      "epoch": 194.59,
+      "learning_rate": 1.4241486068111456e-06,
+      "loss": 0.0194,
+      "step": 3308
+    },
+    {
+      "epoch": 194.65,
+      "learning_rate": 1.408668730650155e-06,
+      "loss": 0.0188,
+      "step": 3309
+    },
+    {
+      "epoch": 194.71,
+      "learning_rate": 1.3931888544891641e-06,
+      "loss": 0.0221,
+      "step": 3310
+    },
+    {
+      "epoch": 194.76,
+      "learning_rate": 1.3777089783281735e-06,
+      "loss": 0.0217,
+      "step": 3311
+    },
+    {
+      "epoch": 194.82,
+      "learning_rate": 1.3622291021671829e-06,
+      "loss": 0.0183,
+      "step": 3312
+    },
+    {
+      "epoch": 194.88,
+      "learning_rate": 1.346749226006192e-06,
+      "loss": 0.0204,
+      "step": 3313
+    },
+    {
+      "epoch": 194.94,
+      "learning_rate": 1.3312693498452014e-06,
+      "loss": 0.0245,
+      "step": 3314
+    },
+    {
+      "epoch": 195.0,
+      "learning_rate": 1.3157894736842106e-06,
+      "loss": 0.0237,
+      "step": 3315
+    },
+    {
+      "epoch": 195.06,
+      "learning_rate": 1.30030959752322e-06,
+      "loss": 0.0229,
+      "step": 3316
+    },
+    {
+      "epoch": 195.12,
+      "learning_rate": 1.2848297213622293e-06,
+      "loss": 0.0201,
+      "step": 3317
+    },
+    {
+      "epoch": 195.18,
+      "learning_rate": 1.2693498452012384e-06,
+      "loss": 0.0233,
+      "step": 3318
+    },
+    {
+      "epoch": 195.24,
+      "learning_rate": 1.2538699690402478e-06,
+      "loss": 0.0197,
+      "step": 3319
+    },
+    {
+      "epoch": 195.29,
+      "learning_rate": 1.2383900928792572e-06,
+      "loss": 0.0198,
+      "step": 3320
+    },
+    {
+      "epoch": 195.29,
+      "eval_accuracy_NAT": 0.993921880459408,
+      "eval_accuracy_REST": 0.35609756097560974,
+      "eval_accuracy_concrete": 0.9560119185028795,
+      "eval_accuracy_concrete tiles": 0.9345745934480321,
+      "eval_accuracy_dark bricks": 0.8574306605669659,
+      "eval_accuracy_grass": 0.8954796807049018,
+      "eval_accuracy_red bricks": 0.9279986818980668,
+      "eval_accuracy_road": 0.9815220278313233,
+      "eval_accuracy_rough concrete": 0.623737479069712,
+      "eval_accuracy_rough red sidewalk": 0.7884853999503681,
+      "eval_accuracy_speedway bricks": 0.9897150060354808,
+      "eval_accuracy_steel": 0.9186728772983752,
+      "eval_accuracy_tiles": 0.24935623617801206,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983727539278749,
+      "eval_iou_REST": 0.3492822966507177,
+      "eval_iou_concrete": 0.8926383345609438,
+      "eval_iou_concrete tiles": 0.8723792762072379,
+      "eval_iou_dark bricks": 0.8410540694044808,
+      "eval_iou_grass": 0.8403733856186864,
+      "eval_iou_red bricks": 0.9010291686663467,
+      "eval_iou_road": 0.9609076126537138,
+      "eval_iou_rough concrete": 0.6172602901285529,
+      "eval_iou_rough red sidewalk": 0.6256645881194618,
+      "eval_iou_speedway bricks": 0.9795550149270237,
+      "eval_iou_steel": 0.8679440838789035,
+      "eval_iou_tiles": 0.24633387202968815,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08207561820745468,
+      "eval_mean_accuracy": 0.8056156925322411,
+      "eval_mean_iou": 0.7675499640095773,
+      "eval_overall_accuracy": 0.9800938105179091,
+      "eval_runtime": 5.2596,
+      "eval_samples_per_second": 22.435,
+      "eval_steps_per_second": 1.521,
+      "step": 3320
+    },
+    {
+      "epoch": 195.35,
+      "learning_rate": 1.2229102167182663e-06,
+      "loss": 0.021,
+      "step": 3321
+    },
+    {
+      "epoch": 195.41,
+      "learning_rate": 1.2074303405572757e-06,
+      "loss": 0.0216,
+      "step": 3322
+    },
+    {
+      "epoch": 195.47,
+      "learning_rate": 1.1919504643962849e-06,
+      "loss": 0.0197,
+      "step": 3323
+    },
+    {
+      "epoch": 195.53,
+      "learning_rate": 1.1764705882352942e-06,
+      "loss": 0.018,
+      "step": 3324
+    },
+    {
+      "epoch": 195.59,
+      "learning_rate": 1.1609907120743036e-06,
+      "loss": 0.0206,
+      "step": 3325
+    },
+    {
+      "epoch": 195.65,
+      "learning_rate": 1.1455108359133127e-06,
+      "loss": 0.0201,
+      "step": 3326
+    },
+    {
+      "epoch": 195.71,
+      "learning_rate": 1.1300309597523221e-06,
+      "loss": 0.0236,
+      "step": 3327
+    },
+    {
+      "epoch": 195.76,
+      "learning_rate": 1.1145510835913315e-06,
+      "loss": 0.0222,
+      "step": 3328
+    },
+    {
+      "epoch": 195.82,
+      "learning_rate": 1.0990712074303406e-06,
+      "loss": 0.0189,
+      "step": 3329
+    },
+    {
+      "epoch": 195.88,
+      "learning_rate": 1.08359133126935e-06,
+      "loss": 0.0216,
+      "step": 3330
+    },
+    {
+      "epoch": 195.94,
+      "learning_rate": 1.0681114551083594e-06,
+      "loss": 0.0185,
+      "step": 3331
+    },
+    {
+      "epoch": 196.0,
+      "learning_rate": 1.0526315789473685e-06,
+      "loss": 0.0222,
+      "step": 3332
+    },
+    {
+      "epoch": 196.06,
+      "learning_rate": 1.037151702786378e-06,
+      "loss": 0.0244,
+      "step": 3333
+    },
+    {
+      "epoch": 196.12,
+      "learning_rate": 1.021671826625387e-06,
+      "loss": 0.0232,
+      "step": 3334
+    },
+    {
+      "epoch": 196.18,
+      "learning_rate": 1.0061919504643964e-06,
+      "loss": 0.0223,
+      "step": 3335
+    },
+    {
+      "epoch": 196.24,
+      "learning_rate": 9.907120743034056e-07,
+      "loss": 0.0202,
+      "step": 3336
+    },
+    {
+      "epoch": 196.29,
+      "learning_rate": 9.75232198142415e-07,
+      "loss": 0.019,
+      "step": 3337
+    },
+    {
+      "epoch": 196.35,
+      "learning_rate": 9.597523219814243e-07,
+      "loss": 0.0214,
+      "step": 3338
+    },
+    {
+      "epoch": 196.41,
+      "learning_rate": 9.442724458204336e-07,
+      "loss": 0.0227,
+      "step": 3339
+    },
+    {
+      "epoch": 196.47,
+      "learning_rate": 9.287925696594427e-07,
+      "loss": 0.0182,
+      "step": 3340
+    },
+    {
+      "epoch": 196.47,
+      "eval_accuracy_NAT": 0.9936138825798653,
+      "eval_accuracy_REST": 0.3544715447154472,
+      "eval_accuracy_concrete": 0.957865967292913,
+      "eval_accuracy_concrete tiles": 0.9313457459344803,
+      "eval_accuracy_dark bricks": 0.858875830254771,
+      "eval_accuracy_grass": 0.8976484347500255,
+      "eval_accuracy_red bricks": 0.9258567662565905,
+      "eval_accuracy_road": 0.9813162776206298,
+      "eval_accuracy_rough concrete": 0.6236063854509919,
+      "eval_accuracy_rough red sidewalk": 0.7861842847366878,
+      "eval_accuracy_speedway bricks": 0.9891476160022155,
+      "eval_accuracy_steel": 0.9238456271521888,
+      "eval_accuracy_tiles": 0.24505437910872793,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837998725836874,
+      "eval_iou_REST": 0.34824281150159747,
+      "eval_iou_concrete": 0.8920084609637854,
+      "eval_iou_concrete tiles": 0.8730337575114882,
+      "eval_iou_dark bricks": 0.8416933192020184,
+      "eval_iou_grass": 0.8409398144659802,
+      "eval_iou_red bricks": 0.9003418073061311,
+      "eval_iou_road": 0.960901245240602,
+      "eval_iou_rough concrete": 0.6171341970409072,
+      "eval_iou_rough red sidewalk": 0.6249536721542747,
+      "eval_iou_speedway bricks": 0.9795146041469359,
+      "eval_iou_steel": 0.8681885643968926,
+      "eval_iou_tiles": 0.24262147570485904,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08236201852560043,
+      "eval_mean_accuracy": 0.8052948262965797,
+      "eval_mean_iou": 0.7671825847860891,
+      "eval_overall_accuracy": 0.9800763211007846,
+      "eval_runtime": 5.3652,
+      "eval_samples_per_second": 21.993,
+      "eval_steps_per_second": 1.491,
+      "step": 3340
+    },
+    {
+      "epoch": 196.53,
+      "learning_rate": 9.133126934984521e-07,
+      "loss": 0.0205,
+      "step": 3341
+    },
+    {
+      "epoch": 196.59,
+      "learning_rate": 8.978328173374612e-07,
+      "loss": 0.0201,
+      "step": 3342
+    },
+    {
+      "epoch": 196.65,
+      "learning_rate": 8.823529411764706e-07,
+      "loss": 0.0221,
+      "step": 3343
+    },
+    {
+      "epoch": 196.71,
+      "learning_rate": 8.6687306501548e-07,
+      "loss": 0.0174,
+      "step": 3344
+    },
+    {
+      "epoch": 196.76,
+      "learning_rate": 8.513931888544891e-07,
+      "loss": 0.0189,
+      "step": 3345
+    },
+    {
+      "epoch": 196.82,
+      "learning_rate": 8.359133126934985e-07,
+      "loss": 0.0253,
+      "step": 3346
+    },
+    {
+      "epoch": 196.88,
+      "learning_rate": 8.204334365325079e-07,
+      "loss": 0.018,
+      "step": 3347
+    },
+    {
+      "epoch": 196.94,
+      "learning_rate": 8.04953560371517e-07,
+      "loss": 0.0197,
+      "step": 3348
+    },
+    {
+      "epoch": 197.0,
+      "learning_rate": 7.894736842105264e-07,
+      "loss": 0.0221,
+      "step": 3349
+    },
+    {
+      "epoch": 197.06,
+      "learning_rate": 7.739938080495357e-07,
+      "loss": 0.0195,
+      "step": 3350
+    },
+    {
+      "epoch": 197.12,
+      "learning_rate": 7.585139318885449e-07,
+      "loss": 0.0217,
+      "step": 3351
+    },
+    {
+      "epoch": 197.18,
+      "learning_rate": 7.430340557275542e-07,
+      "loss": 0.0207,
+      "step": 3352
+    },
+    {
+      "epoch": 197.24,
+      "learning_rate": 7.275541795665635e-07,
+      "loss": 0.02,
+      "step": 3353
+    },
+    {
+      "epoch": 197.29,
+      "learning_rate": 7.120743034055728e-07,
+      "loss": 0.0193,
+      "step": 3354
+    },
+    {
+      "epoch": 197.35,
+      "learning_rate": 6.965944272445821e-07,
+      "loss": 0.0188,
+      "step": 3355
+    },
+    {
+      "epoch": 197.41,
+      "learning_rate": 6.811145510835914e-07,
+      "loss": 0.0205,
+      "step": 3356
+    },
+    {
+      "epoch": 197.47,
+      "learning_rate": 6.656346749226007e-07,
+      "loss": 0.0227,
+      "step": 3357
+    },
+    {
+      "epoch": 197.53,
+      "learning_rate": 6.5015479876161e-07,
+      "loss": 0.0203,
+      "step": 3358
+    },
+    {
+      "epoch": 197.59,
+      "learning_rate": 6.346749226006192e-07,
+      "loss": 0.0245,
+      "step": 3359
+    },
+    {
+      "epoch": 197.65,
+      "learning_rate": 6.191950464396286e-07,
+      "loss": 0.0186,
+      "step": 3360
+    },
+    {
+      "epoch": 197.65,
+      "eval_accuracy_NAT": 0.9936343151365512,
+      "eval_accuracy_REST": 0.359349593495935,
+      "eval_accuracy_concrete": 0.9573469406475068,
+      "eval_accuracy_concrete tiles": 0.9342210699976432,
+      "eval_accuracy_dark bricks": 0.8621059280669181,
+      "eval_accuracy_grass": 0.8982227950054313,
+      "eval_accuracy_red bricks": 0.9247034270650264,
+      "eval_accuracy_road": 0.9810886163485376,
+      "eval_accuracy_rough concrete": 0.6238626138875812,
+      "eval_accuracy_rough red sidewalk": 0.7856052459410885,
+      "eval_accuracy_speedway bricks": 0.9894427068017573,
+      "eval_accuracy_steel": 0.9231584260846531,
+      "eval_accuracy_tiles": 0.24623587506437639,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838060293476084,
+      "eval_iou_REST": 0.35303514376996803,
+      "eval_iou_concrete": 0.8923098900018452,
+      "eval_iou_concrete tiles": 0.8728750110103056,
+      "eval_iou_dark bricks": 0.8439485075046377,
+      "eval_iou_grass": 0.8413473725260797,
+      "eval_iou_red bricks": 0.9003743315508022,
+      "eval_iou_road": 0.9610233108408759,
+      "eval_iou_rough concrete": 0.617336800457566,
+      "eval_iou_rough red sidewalk": 0.6247921725296939,
+      "eval_iou_speedway bricks": 0.9795697780926579,
+      "eval_iou_steel": 0.86840540191112,
+      "eval_iou_tiles": 0.24386438643864386,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08214017003774643,
+      "eval_mean_accuracy": 0.8060751964263851,
+      "eval_mean_iou": 0.7678990873832158,
+      "eval_overall_accuracy": 0.9801226470430019,
+      "eval_runtime": 5.4146,
+      "eval_samples_per_second": 21.793,
+      "eval_steps_per_second": 1.477,
+      "step": 3360
+    },
+    {
+      "epoch": 197.71,
+      "learning_rate": 1.7946336429308566e-05,
+      "loss": 0.0209,
+      "step": 3361
+    },
+    {
+      "epoch": 197.76,
+      "learning_rate": 1.7936016511867905e-05,
+      "loss": 0.0209,
+      "step": 3362
+    },
+    {
+      "epoch": 197.82,
+      "learning_rate": 1.7925696594427247e-05,
+      "loss": 0.0206,
+      "step": 3363
+    },
+    {
+      "epoch": 197.88,
+      "learning_rate": 1.7915376676986583e-05,
+      "loss": 0.0207,
+      "step": 3364
+    },
+    {
+      "epoch": 197.94,
+      "learning_rate": 1.7905056759545925e-05,
+      "loss": 0.0203,
+      "step": 3365
+    },
+    {
+      "epoch": 198.0,
+      "learning_rate": 1.7894736842105264e-05,
+      "loss": 0.0226,
+      "step": 3366
+    },
+    {
+      "epoch": 198.06,
+      "learning_rate": 1.7884416924664603e-05,
+      "loss": 0.0207,
+      "step": 3367
+    },
+    {
+      "epoch": 198.12,
+      "learning_rate": 1.7874097007223945e-05,
+      "loss": 0.0246,
+      "step": 3368
+    },
+    {
+      "epoch": 198.18,
+      "learning_rate": 1.786377708978328e-05,
+      "loss": 0.0245,
+      "step": 3369
+    },
+    {
+      "epoch": 198.24,
+      "learning_rate": 1.7853457172342623e-05,
+      "loss": 0.0192,
+      "step": 3370
+    },
+    {
+      "epoch": 198.29,
+      "learning_rate": 1.784313725490196e-05,
+      "loss": 0.0168,
+      "step": 3371
+    },
+    {
+      "epoch": 198.35,
+      "learning_rate": 1.7832817337461304e-05,
+      "loss": 0.021,
+      "step": 3372
+    },
+    {
+      "epoch": 198.41,
+      "learning_rate": 1.782249742002064e-05,
+      "loss": 0.0194,
+      "step": 3373
+    },
+    {
+      "epoch": 198.47,
+      "learning_rate": 1.7812177502579978e-05,
+      "loss": 0.0242,
+      "step": 3374
+    },
+    {
+      "epoch": 198.53,
+      "learning_rate": 1.780185758513932e-05,
+      "loss": 0.0215,
+      "step": 3375
+    },
+    {
+      "epoch": 198.59,
+      "learning_rate": 1.779153766769866e-05,
+      "loss": 0.0176,
+      "step": 3376
+    },
+    {
+      "epoch": 198.65,
+      "learning_rate": 1.7781217750257998e-05,
+      "loss": 0.0226,
+      "step": 3377
+    },
+    {
+      "epoch": 198.71,
+      "learning_rate": 1.7770897832817337e-05,
+      "loss": 0.022,
+      "step": 3378
+    },
+    {
+      "epoch": 198.76,
+      "learning_rate": 1.776057791537668e-05,
+      "loss": 0.0189,
+      "step": 3379
+    },
+    {
+      "epoch": 198.82,
+      "learning_rate": 1.7750257997936018e-05,
+      "loss": 0.0241,
+      "step": 3380
+    },
+    {
+      "epoch": 198.82,
+      "eval_accuracy_NAT": 0.9940302430951301,
+      "eval_accuracy_REST": 0.34308943089430893,
+      "eval_accuracy_concrete": 0.9557149496490646,
+      "eval_accuracy_concrete tiles": 0.9396417629036059,
+      "eval_accuracy_dark bricks": 0.8677239055950079,
+      "eval_accuracy_grass": 0.8903728894389626,
+      "eval_accuracy_red bricks": 0.9301955184534271,
+      "eval_accuracy_road": 0.9817208306323051,
+      "eval_accuracy_rough concrete": 0.6259899057913585,
+      "eval_accuracy_rough red sidewalk": 0.7847930876303777,
+      "eval_accuracy_speedway bricks": 0.9898779742993393,
+      "eval_accuracy_steel": 0.9173159675541139,
+      "eval_accuracy_tiles": 0.25593020085431245,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835996634931101,
+      "eval_iou_REST": 0.33438985736925514,
+      "eval_iou_concrete": 0.8929228343228841,
+      "eval_iou_concrete tiles": 0.8735921819536351,
+      "eval_iou_dark bricks": 0.8493049310551559,
+      "eval_iou_grass": 0.8387275905314452,
+      "eval_iou_red bricks": 0.9027770374713502,
+      "eval_iou_road": 0.9605544557354659,
+      "eval_iou_rough concrete": 0.6194893265715297,
+      "eval_iou_rough red sidewalk": 0.6250059888846301,
+      "eval_iou_speedway bricks": 0.9795639599215524,
+      "eval_iou_steel": 0.8669343220839336,
+      "eval_iou_tiles": 0.24934328974941708,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0823364406824112,
+      "eval_mean_accuracy": 0.805876666676255,
+      "eval_mean_iou": 0.7674004183956433,
+      "eval_overall_accuracy": 0.9800652972722458,
+      "eval_runtime": 5.3942,
+      "eval_samples_per_second": 21.875,
+      "eval_steps_per_second": 1.483,
+      "step": 3380
+    },
+    {
+      "epoch": 198.88,
+      "learning_rate": 1.7739938080495357e-05,
+      "loss": 0.0204,
+      "step": 3381
+    },
+    {
+      "epoch": 198.94,
+      "learning_rate": 1.7729618163054696e-05,
+      "loss": 0.02,
+      "step": 3382
+    },
+    {
+      "epoch": 199.0,
+      "learning_rate": 1.7719298245614035e-05,
+      "loss": 0.0196,
+      "step": 3383
+    },
+    {
+      "epoch": 199.06,
+      "learning_rate": 1.7708978328173377e-05,
+      "loss": 0.0206,
+      "step": 3384
+    },
+    {
+      "epoch": 199.12,
+      "learning_rate": 1.7698658410732716e-05,
+      "loss": 0.0212,
+      "step": 3385
+    },
+    {
+      "epoch": 199.18,
+      "learning_rate": 1.7688338493292055e-05,
+      "loss": 0.0187,
+      "step": 3386
+    },
+    {
+      "epoch": 199.24,
+      "learning_rate": 1.7678018575851394e-05,
+      "loss": 0.0174,
+      "step": 3387
+    },
+    {
+      "epoch": 199.29,
+      "learning_rate": 1.7667698658410732e-05,
+      "loss": 0.0221,
+      "step": 3388
+    },
+    {
+      "epoch": 199.35,
+      "learning_rate": 1.7657378740970075e-05,
+      "loss": 0.024,
+      "step": 3389
+    },
+    {
+      "epoch": 199.41,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.0222,
+      "step": 3390
+    },
+    {
+      "epoch": 199.47,
+      "learning_rate": 1.7636738906088752e-05,
+      "loss": 0.0238,
+      "step": 3391
+    },
+    {
+      "epoch": 199.53,
+      "learning_rate": 1.762641898864809e-05,
+      "loss": 0.0203,
+      "step": 3392
+    },
+    {
+      "epoch": 199.59,
+      "learning_rate": 1.761609907120743e-05,
+      "loss": 0.0224,
+      "step": 3393
+    },
+    {
+      "epoch": 199.65,
+      "learning_rate": 1.7605779153766772e-05,
+      "loss": 0.0213,
+      "step": 3394
+    },
+    {
+      "epoch": 199.71,
+      "learning_rate": 1.7595459236326108e-05,
+      "loss": 0.0191,
+      "step": 3395
+    },
+    {
+      "epoch": 199.76,
+      "learning_rate": 1.758513931888545e-05,
+      "loss": 0.025,
+      "step": 3396
+    },
+    {
+      "epoch": 199.82,
+      "learning_rate": 1.757481940144479e-05,
+      "loss": 0.0255,
+      "step": 3397
+    },
+    {
+      "epoch": 199.88,
+      "learning_rate": 1.7564499484004128e-05,
+      "loss": 0.0198,
+      "step": 3398
+    },
+    {
+      "epoch": 199.94,
+      "learning_rate": 1.755417956656347e-05,
+      "loss": 0.0194,
+      "step": 3399
+    },
+    {
+      "epoch": 200.0,
+      "learning_rate": 1.7543859649122806e-05,
+      "loss": 0.0212,
+      "step": 3400
+    },
+    {
+      "epoch": 200.0,
+      "eval_accuracy_NAT": 0.9944741525192776,
+      "eval_accuracy_REST": 0.3252032520325203,
+      "eval_accuracy_concrete": 0.9565779556618239,
+      "eval_accuracy_concrete tiles": 0.9140938015555031,
+      "eval_accuracy_dark bricks": 0.8484150986735065,
+      "eval_accuracy_grass": 0.8866315048427937,
+      "eval_accuracy_red bricks": 0.9209138840070299,
+      "eval_accuracy_road": 0.9818132846230844,
+      "eval_accuracy_rough concrete": 0.6251377972696774,
+      "eval_accuracy_rough red sidewalk": 0.7762879853209905,
+      "eval_accuracy_speedway bricks": 0.9894187156798433,
+      "eval_accuracy_steel": 0.9186628816464838,
+      "eval_accuracy_tiles": 0.2220606501257233,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835222196001595,
+      "eval_iou_REST": 0.31746031746031744,
+      "eval_iou_concrete": 0.8927768836305422,
+      "eval_iou_concrete tiles": 0.8665489968273828,
+      "eval_iou_dark bricks": 0.8352098211762046,
+      "eval_iou_grass": 0.8386372972919757,
+      "eval_iou_red bricks": 0.899860470108404,
+      "eval_iou_road": 0.9614694205654822,
+      "eval_iou_rough concrete": 0.6193765497697485,
+      "eval_iou_rough red sidewalk": 0.616486213712832,
+      "eval_iou_speedway bricks": 0.9793353489693541,
+      "eval_iou_steel": 0.8672650923588667,
+      "eval_iou_tiles": 0.22035834535834536,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08265627175569534,
+      "eval_mean_accuracy": 0.7968993049198659,
+      "eval_mean_iou": 0.7614082289868935,
+      "eval_overall_accuracy": 0.9799644017623643,
+      "eval_runtime": 5.2595,
+      "eval_samples_per_second": 22.436,
+      "eval_steps_per_second": 1.521,
+      "step": 3400
+    },
+    {
+      "epoch": 200.06,
+      "learning_rate": 1.7533539731682148e-05,
+      "loss": 0.0205,
+      "step": 3401
+    },
+    {
+      "epoch": 200.12,
+      "learning_rate": 1.7523219814241487e-05,
+      "loss": 0.0205,
+      "step": 3402
+    },
+    {
+      "epoch": 200.18,
+      "learning_rate": 1.7512899896800826e-05,
+      "loss": 0.0186,
+      "step": 3403
+    },
+    {
+      "epoch": 200.24,
+      "learning_rate": 1.7502579979360165e-05,
+      "loss": 0.0197,
+      "step": 3404
+    },
+    {
+      "epoch": 200.29,
+      "learning_rate": 1.7492260061919503e-05,
+      "loss": 0.0236,
+      "step": 3405
+    },
+    {
+      "epoch": 200.35,
+      "learning_rate": 1.7481940144478846e-05,
+      "loss": 0.0211,
+      "step": 3406
+    },
+    {
+      "epoch": 200.41,
+      "learning_rate": 1.7471620227038185e-05,
+      "loss": 0.0266,
+      "step": 3407
+    },
+    {
+      "epoch": 200.47,
+      "learning_rate": 1.7461300309597527e-05,
+      "loss": 0.02,
+      "step": 3408
+    },
+    {
+      "epoch": 200.53,
+      "learning_rate": 1.7450980392156862e-05,
+      "loss": 0.0198,
+      "step": 3409
+    },
+    {
+      "epoch": 200.59,
+      "learning_rate": 1.74406604747162e-05,
+      "loss": 0.0219,
+      "step": 3410
+    },
+    {
+      "epoch": 200.65,
+      "learning_rate": 1.7430340557275543e-05,
+      "loss": 0.0202,
+      "step": 3411
+    },
+    {
+      "epoch": 200.71,
+      "learning_rate": 1.7420020639834882e-05,
+      "loss": 0.02,
+      "step": 3412
+    },
+    {
+      "epoch": 200.76,
+      "learning_rate": 1.740970072239422e-05,
+      "loss": 0.0198,
+      "step": 3413
+    },
+    {
+      "epoch": 200.82,
+      "learning_rate": 1.739938080495356e-05,
+      "loss": 0.0234,
+      "step": 3414
+    },
+    {
+      "epoch": 200.88,
+      "learning_rate": 1.7389060887512902e-05,
+      "loss": 0.0192,
+      "step": 3415
+    },
+    {
+      "epoch": 200.94,
+      "learning_rate": 1.737874097007224e-05,
+      "loss": 0.0189,
+      "step": 3416
+    },
+    {
+      "epoch": 201.0,
+      "learning_rate": 1.736842105263158e-05,
+      "loss": 0.026,
+      "step": 3417
+    },
+    {
+      "epoch": 201.06,
+      "learning_rate": 1.735810113519092e-05,
+      "loss": 0.0229,
+      "step": 3418
+    },
+    {
+      "epoch": 201.12,
+      "learning_rate": 1.7347781217750258e-05,
+      "loss": 0.0224,
+      "step": 3419
+    },
+    {
+      "epoch": 201.18,
+      "learning_rate": 1.73374613003096e-05,
+      "loss": 0.0204,
+      "step": 3420
+    },
+    {
+      "epoch": 201.18,
+      "eval_accuracy_NAT": 0.9936125347858886,
+      "eval_accuracy_REST": 0.38699186991869916,
+      "eval_accuracy_concrete": 0.9568321181943682,
+      "eval_accuracy_concrete tiles": 0.9381333961819467,
+      "eval_accuracy_dark bricks": 0.8582728786631701,
+      "eval_accuracy_grass": 0.9025480021346899,
+      "eval_accuracy_red bricks": 0.9271748681898067,
+      "eval_accuracy_road": 0.981821835281191,
+      "eval_accuracy_rough concrete": 0.625310602494354,
+      "eval_accuracy_rough red sidewalk": 0.7837628497732725,
+      "eval_accuracy_speedway bricks": 0.9884844328464506,
+      "eval_accuracy_steel": 0.9280787857282082,
+      "eval_accuracy_tiles": 0.24305492441455362,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983867126749741,
+      "eval_iou_REST": 0.3801916932907348,
+      "eval_iou_concrete": 0.8918659434810876,
+      "eval_iou_concrete tiles": 0.8632617653437432,
+      "eval_iou_dark bricks": 0.8419157685230388,
+      "eval_iou_grass": 0.8437616932591098,
+      "eval_iou_red bricks": 0.9017199017199017,
+      "eval_iou_road": 0.9610321811181788,
+      "eval_iou_rough concrete": 0.6190470572272987,
+      "eval_iou_rough red sidewalk": 0.6216316159893118,
+      "eval_iou_speedway bricks": 0.9793822327218618,
+      "eval_iou_steel": 0.8688266875650638,
+      "eval_iou_tiles": 0.24031030971065717,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08170458674430847,
+      "eval_mean_accuracy": 0.8087753152774305,
+      "eval_mean_iou": 0.7689856905153639,
+      "eval_overall_accuracy": 0.9801190586413367,
+      "eval_runtime": 5.8331,
+      "eval_samples_per_second": 20.229,
+      "eval_steps_per_second": 1.371,
+      "step": 3420
+    },
+    {
+      "epoch": 201.24,
+      "learning_rate": 1.732714138286894e-05,
+      "loss": 0.0185,
+      "step": 3421
+    },
+    {
+      "epoch": 201.29,
+      "learning_rate": 1.7316821465428278e-05,
+      "loss": 0.0228,
+      "step": 3422
+    },
+    {
+      "epoch": 201.35,
+      "learning_rate": 1.7306501547987617e-05,
+      "loss": 0.0223,
+      "step": 3423
+    },
+    {
+      "epoch": 201.41,
+      "learning_rate": 1.7296181630546956e-05,
+      "loss": 0.0199,
+      "step": 3424
+    },
+    {
+      "epoch": 201.47,
+      "learning_rate": 1.7285861713106298e-05,
+      "loss": 0.0218,
+      "step": 3425
+    },
+    {
+      "epoch": 201.53,
+      "learning_rate": 1.7275541795665637e-05,
+      "loss": 0.0194,
+      "step": 3426
+    },
+    {
+      "epoch": 201.59,
+      "learning_rate": 1.7265221878224976e-05,
+      "loss": 0.0203,
+      "step": 3427
+    },
+    {
+      "epoch": 201.65,
+      "learning_rate": 1.7254901960784314e-05,
+      "loss": 0.0193,
+      "step": 3428
+    },
+    {
+      "epoch": 201.71,
+      "learning_rate": 1.7244582043343653e-05,
+      "loss": 0.0225,
+      "step": 3429
+    },
+    {
+      "epoch": 201.76,
+      "learning_rate": 1.7234262125902996e-05,
+      "loss": 0.0203,
+      "step": 3430
+    },
+    {
+      "epoch": 201.82,
+      "learning_rate": 1.722394220846233e-05,
+      "loss": 0.0237,
+      "step": 3431
+    },
+    {
+      "epoch": 201.88,
+      "learning_rate": 1.7213622291021673e-05,
+      "loss": 0.0218,
+      "step": 3432
+    },
+    {
+      "epoch": 201.94,
+      "learning_rate": 1.7203302373581012e-05,
+      "loss": 0.0188,
+      "step": 3433
+    },
+    {
+      "epoch": 202.0,
+      "learning_rate": 1.719298245614035e-05,
+      "loss": 0.0221,
+      "step": 3434
+    },
+    {
+      "epoch": 202.06,
+      "learning_rate": 1.7182662538699693e-05,
+      "loss": 0.0203,
+      "step": 3435
+    },
+    {
+      "epoch": 202.12,
+      "learning_rate": 1.717234262125903e-05,
+      "loss": 0.0207,
+      "step": 3436
+    },
+    {
+      "epoch": 202.18,
+      "learning_rate": 1.716202270381837e-05,
+      "loss": 0.0231,
+      "step": 3437
+    },
+    {
+      "epoch": 202.24,
+      "learning_rate": 1.715170278637771e-05,
+      "loss": 0.0231,
+      "step": 3438
+    },
+    {
+      "epoch": 202.29,
+      "learning_rate": 1.714138286893705e-05,
+      "loss": 0.022,
+      "step": 3439
+    },
+    {
+      "epoch": 202.35,
+      "learning_rate": 1.7131062951496388e-05,
+      "loss": 0.0216,
+      "step": 3440
+    },
+    {
+      "epoch": 202.35,
+      "eval_accuracy_NAT": 0.9938182620584836,
+      "eval_accuracy_REST": 0.39186991869918697,
+      "eval_accuracy_concrete": 0.9606078650500146,
+      "eval_accuracy_concrete tiles": 0.936153664859769,
+      "eval_accuracy_dark bricks": 0.8437589724939226,
+      "eval_accuracy_grass": 0.8911856990919999,
+      "eval_accuracy_red bricks": 0.9277240773286467,
+      "eval_accuracy_road": 0.980328142193169,
+      "eval_accuracy_rough concrete": 0.6227244829250561,
+      "eval_accuracy_rough red sidewalk": 0.7607817775739026,
+      "eval_accuracy_speedway bricks": 0.9890060683829229,
+      "eval_accuracy_steel": 0.9202771794269493,
+      "eval_accuracy_tiles": 0.23884395164955013,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836554784194569,
+      "eval_iou_REST": 0.3813291139240506,
+      "eval_iou_concrete": 0.8904380007092698,
+      "eval_iou_concrete tiles": 0.8700823622185228,
+      "eval_iou_dark bricks": 0.8295983325413219,
+      "eval_iou_grass": 0.8408160423090726,
+      "eval_iou_red bricks": 0.9007145142369628,
+      "eval_iou_road": 0.9610360421717928,
+      "eval_iou_rough concrete": 0.6166250685925689,
+      "eval_iou_rough red sidewalk": 0.6170737062971186,
+      "eval_iou_speedway bricks": 0.9790585979260981,
+      "eval_iou_steel": 0.8667708536823598,
+      "eval_iou_tiles": 0.23723407456443896,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08294358849525452,
+      "eval_mean_accuracy": 0.8043907739795055,
+      "eval_mean_iou": 0.7672640144302334,
+      "eval_overall_accuracy": 0.9798630536612818,
+      "eval_runtime": 5.6468,
+      "eval_samples_per_second": 20.897,
+      "eval_steps_per_second": 1.417,
+      "step": 3440
+    },
+    {
+      "epoch": 202.41,
+      "learning_rate": 1.7120743034055727e-05,
+      "loss": 0.0196,
+      "step": 3441
+    },
+    {
+      "epoch": 202.47,
+      "learning_rate": 1.711042311661507e-05,
+      "loss": 0.0226,
+      "step": 3442
+    },
+    {
+      "epoch": 202.53,
+      "learning_rate": 1.7100103199174408e-05,
+      "loss": 0.0166,
+      "step": 3443
+    },
+    {
+      "epoch": 202.59,
+      "learning_rate": 1.708978328173375e-05,
+      "loss": 0.0216,
+      "step": 3444
+    },
+    {
+      "epoch": 202.65,
+      "learning_rate": 1.7079463364293085e-05,
+      "loss": 0.0207,
+      "step": 3445
+    },
+    {
+      "epoch": 202.71,
+      "learning_rate": 1.7069143446852424e-05,
+      "loss": 0.0188,
+      "step": 3446
+    },
+    {
+      "epoch": 202.76,
+      "learning_rate": 1.7058823529411767e-05,
+      "loss": 0.0214,
+      "step": 3447
+    },
+    {
+      "epoch": 202.82,
+      "learning_rate": 1.7048503611971105e-05,
+      "loss": 0.0234,
+      "step": 3448
+    },
+    {
+      "epoch": 202.88,
+      "learning_rate": 1.7038183694530444e-05,
+      "loss": 0.0189,
+      "step": 3449
+    },
+    {
+      "epoch": 202.94,
+      "learning_rate": 1.7027863777089783e-05,
+      "loss": 0.0219,
+      "step": 3450
+    },
+    {
+      "epoch": 203.0,
+      "learning_rate": 1.7017543859649125e-05,
+      "loss": 0.0199,
+      "step": 3451
+    },
+    {
+      "epoch": 203.06,
+      "learning_rate": 1.7007223942208464e-05,
+      "loss": 0.0202,
+      "step": 3452
+    },
+    {
+      "epoch": 203.12,
+      "learning_rate": 1.69969040247678e-05,
+      "loss": 0.0214,
+      "step": 3453
+    },
+    {
+      "epoch": 203.18,
+      "learning_rate": 1.6986584107327142e-05,
+      "loss": 0.0214,
+      "step": 3454
+    },
+    {
+      "epoch": 203.24,
+      "learning_rate": 1.697626418988648e-05,
+      "loss": 0.0238,
+      "step": 3455
+    },
+    {
+      "epoch": 203.29,
+      "learning_rate": 1.6965944272445823e-05,
+      "loss": 0.0251,
+      "step": 3456
+    },
+    {
+      "epoch": 203.35,
+      "learning_rate": 1.6955624355005162e-05,
+      "loss": 0.0233,
+      "step": 3457
+    },
+    {
+      "epoch": 203.41,
+      "learning_rate": 1.6945304437564498e-05,
+      "loss": 0.0199,
+      "step": 3458
+    },
+    {
+      "epoch": 203.47,
+      "learning_rate": 1.693498452012384e-05,
+      "loss": 0.0202,
+      "step": 3459
+    },
+    {
+      "epoch": 203.53,
+      "learning_rate": 1.692466460268318e-05,
+      "loss": 0.0192,
+      "step": 3460
+    },
+    {
+      "epoch": 203.53,
+      "eval_accuracy_NAT": 0.9937634337995137,
+      "eval_accuracy_REST": 0.3951219512195122,
+      "eval_accuracy_concrete": 0.9580532449484719,
+      "eval_accuracy_concrete tiles": 0.9245816639170398,
+      "eval_accuracy_dark bricks": 0.85098960626304,
+      "eval_accuracy_grass": 0.8966889597928897,
+      "eval_accuracy_red bricks": 0.913938927943761,
+      "eval_accuracy_road": 0.9803463123416458,
+      "eval_accuracy_rough concrete": 0.6228496177429255,
+      "eval_accuracy_rough red sidewalk": 0.7711443160198227,
+      "eval_accuracy_speedway bricks": 0.9886127853486903,
+      "eval_accuracy_steel": 0.9359553594186529,
+      "eval_accuracy_tiles": 0.23166409161137871,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838696944893759,
+      "eval_iou_REST": 0.38571428571428573,
+      "eval_iou_concrete": 0.8909578627404147,
+      "eval_iou_concrete tiles": 0.8700377023730317,
+      "eval_iou_dark bricks": 0.8356499755648283,
+      "eval_iou_grass": 0.843578750965849,
+      "eval_iou_red bricks": 0.8956404736275565,
+      "eval_iou_road": 0.9600499277774289,
+      "eval_iou_rough concrete": 0.6175506176924123,
+      "eval_iou_rough red sidewalk": 0.6195571398362686,
+      "eval_iou_speedway bricks": 0.9792096415039288,
+      "eval_iou_steel": 0.8665950953602821,
+      "eval_iou_tiles": 0.2299019902591546,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08265773952007294,
+      "eval_mean_accuracy": 0.8049007900282572,
+      "eval_mean_iou": 0.7675625506080629,
+      "eval_overall_accuracy": 0.9799943374375165,
+      "eval_runtime": 5.7888,
+      "eval_samples_per_second": 20.384,
+      "eval_steps_per_second": 1.382,
+      "step": 3460
+    },
+    {
+      "epoch": 203.59,
+      "learning_rate": 1.691434468524252e-05,
+      "loss": 0.0203,
+      "step": 3461
+    },
+    {
+      "epoch": 203.65,
+      "learning_rate": 1.6904024767801856e-05,
+      "loss": 0.0195,
+      "step": 3462
+    },
+    {
+      "epoch": 203.71,
+      "learning_rate": 1.68937048503612e-05,
+      "loss": 0.0206,
+      "step": 3463
+    },
+    {
+      "epoch": 203.76,
+      "learning_rate": 1.6883384932920538e-05,
+      "loss": 0.0203,
+      "step": 3464
+    },
+    {
+      "epoch": 203.82,
+      "learning_rate": 1.6873065015479876e-05,
+      "loss": 0.0196,
+      "step": 3465
+    },
+    {
+      "epoch": 203.88,
+      "learning_rate": 1.686274509803922e-05,
+      "loss": 0.0212,
+      "step": 3466
+    },
+    {
+      "epoch": 203.94,
+      "learning_rate": 1.6852425180598554e-05,
+      "loss": 0.021,
+      "step": 3467
+    },
+    {
+      "epoch": 204.0,
+      "learning_rate": 1.6842105263157896e-05,
+      "loss": 0.0206,
+      "step": 3468
+    },
+    {
+      "epoch": 204.06,
+      "learning_rate": 1.6831785345717235e-05,
+      "loss": 0.0201,
+      "step": 3469
+    },
+    {
+      "epoch": 204.12,
+      "learning_rate": 1.6821465428276574e-05,
+      "loss": 0.0199,
+      "step": 3470
+    },
+    {
+      "epoch": 204.18,
+      "learning_rate": 1.6811145510835913e-05,
+      "loss": 0.019,
+      "step": 3471
+    },
+    {
+      "epoch": 204.24,
+      "learning_rate": 1.6800825593395252e-05,
+      "loss": 0.0173,
+      "step": 3472
+    },
+    {
+      "epoch": 204.29,
+      "learning_rate": 1.6790505675954594e-05,
+      "loss": 0.0201,
+      "step": 3473
+    },
+    {
+      "epoch": 204.35,
+      "learning_rate": 1.6780185758513933e-05,
+      "loss": 0.0245,
+      "step": 3474
+    },
+    {
+      "epoch": 204.41,
+      "learning_rate": 1.6769865841073272e-05,
+      "loss": 0.0221,
+      "step": 3475
+    },
+    {
+      "epoch": 204.47,
+      "learning_rate": 1.675954592363261e-05,
+      "loss": 0.0183,
+      "step": 3476
+    },
+    {
+      "epoch": 204.53,
+      "learning_rate": 1.674922600619195e-05,
+      "loss": 0.025,
+      "step": 3477
+    },
+    {
+      "epoch": 204.59,
+      "learning_rate": 1.6738906088751292e-05,
+      "loss": 0.023,
+      "step": 3478
+    },
+    {
+      "epoch": 204.65,
+      "learning_rate": 1.672858617131063e-05,
+      "loss": 0.0204,
+      "step": 3479
+    },
+    {
+      "epoch": 204.71,
+      "learning_rate": 1.671826625386997e-05,
+      "loss": 0.0189,
+      "step": 3480
+    },
+    {
+      "epoch": 204.71,
+      "eval_accuracy_NAT": 0.9939066773433515,
+      "eval_accuracy_REST": 0.35609756097560974,
+      "eval_accuracy_concrete": 0.9577987002166306,
+      "eval_accuracy_concrete tiles": 0.9363422106999765,
+      "eval_accuracy_dark bricks": 0.8616417510479872,
+      "eval_accuracy_grass": 0.8993260965833608,
+      "eval_accuracy_red bricks": 0.9234951669595782,
+      "eval_accuracy_road": 0.9806017632525849,
+      "eval_accuracy_rough concrete": 0.6255847073334962,
+      "eval_accuracy_rough red sidewalk": 0.7930199505185029,
+      "eval_accuracy_speedway bricks": 0.9867401069181485,
+      "eval_accuracy_steel": 0.935098232268963,
+      "eval_accuracy_tiles": 0.24014662667757278,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983746349131729,
+      "eval_iou_REST": 0.347068145800317,
+      "eval_iou_concrete": 0.8919445489467767,
+      "eval_iou_concrete tiles": 0.8699336530250279,
+      "eval_iou_dark bricks": 0.8441030593397527,
+      "eval_iou_grass": 0.8426668629038193,
+      "eval_iou_red bricks": 0.8996789727126806,
+      "eval_iou_road": 0.9607787798564044,
+      "eval_iou_rough concrete": 0.6185047720042418,
+      "eval_iou_rough red sidewalk": 0.6242378680431411,
+      "eval_iou_speedway bricks": 0.9790767716679359,
+      "eval_iou_steel": 0.8657333823804996,
+      "eval_iou_tiles": 0.2385925836744522,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0822020173072815,
+      "eval_mean_accuracy": 0.8069076577535201,
+      "eval_mean_iou": 0.7666204422682138,
+      "eval_overall_accuracy": 0.9800107600325245,
+      "eval_runtime": 5.5753,
+      "eval_samples_per_second": 21.165,
+      "eval_steps_per_second": 1.435,
+      "step": 3480
+    },
+    {
+      "epoch": 204.76,
+      "learning_rate": 1.670794633642931e-05,
+      "loss": 0.02,
+      "step": 3481
+    },
+    {
+      "epoch": 204.82,
+      "learning_rate": 1.6697626418988647e-05,
+      "loss": 0.0237,
+      "step": 3482
+    },
+    {
+      "epoch": 204.88,
+      "learning_rate": 1.668730650154799e-05,
+      "loss": 0.0221,
+      "step": 3483
+    },
+    {
+      "epoch": 204.94,
+      "learning_rate": 1.667698658410733e-05,
+      "loss": 0.021,
+      "step": 3484
+    },
+    {
+      "epoch": 205.0,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 0.0231,
+      "step": 3485
+    },
+    {
+      "epoch": 205.06,
+      "learning_rate": 1.6656346749226006e-05,
+      "loss": 0.0223,
+      "step": 3486
+    },
+    {
+      "epoch": 205.12,
+      "learning_rate": 1.6646026831785345e-05,
+      "loss": 0.0222,
+      "step": 3487
+    },
+    {
+      "epoch": 205.18,
+      "learning_rate": 1.6635706914344687e-05,
+      "loss": 0.0218,
+      "step": 3488
+    },
+    {
+      "epoch": 205.24,
+      "learning_rate": 1.6625386996904023e-05,
+      "loss": 0.0236,
+      "step": 3489
+    },
+    {
+      "epoch": 205.29,
+      "learning_rate": 1.6615067079463365e-05,
+      "loss": 0.0188,
+      "step": 3490
+    },
+    {
+      "epoch": 205.35,
+      "learning_rate": 1.6604747162022704e-05,
+      "loss": 0.0185,
+      "step": 3491
+    },
+    {
+      "epoch": 205.41,
+      "learning_rate": 1.6594427244582046e-05,
+      "loss": 0.0242,
+      "step": 3492
+    },
+    {
+      "epoch": 205.47,
+      "learning_rate": 1.6584107327141385e-05,
+      "loss": 0.0194,
+      "step": 3493
+    },
+    {
+      "epoch": 205.53,
+      "learning_rate": 1.657378740970072e-05,
+      "loss": 0.0196,
+      "step": 3494
+    },
+    {
+      "epoch": 205.59,
+      "learning_rate": 1.6563467492260063e-05,
+      "loss": 0.0185,
+      "step": 3495
+    },
+    {
+      "epoch": 205.65,
+      "learning_rate": 1.6553147574819402e-05,
+      "loss": 0.0212,
+      "step": 3496
+    },
+    {
+      "epoch": 205.71,
+      "learning_rate": 1.6542827657378744e-05,
+      "loss": 0.0198,
+      "step": 3497
+    },
+    {
+      "epoch": 205.76,
+      "learning_rate": 1.653250773993808e-05,
+      "loss": 0.0193,
+      "step": 3498
+    },
+    {
+      "epoch": 205.82,
+      "learning_rate": 1.6522187822497422e-05,
+      "loss": 0.0267,
+      "step": 3499
+    },
+    {
+      "epoch": 205.88,
+      "learning_rate": 1.651186790505676e-05,
+      "loss": 0.0226,
+      "step": 3500
+    },
+    {
+      "epoch": 205.88,
+      "eval_accuracy_NAT": 0.9928900093908893,
+      "eval_accuracy_REST": 0.3024390243902439,
+      "eval_accuracy_concrete": 0.9534959005303397,
+      "eval_accuracy_concrete tiles": 0.935234503888758,
+      "eval_accuracy_dark bricks": 0.8572153207128228,
+      "eval_accuracy_grass": 0.9105265946776581,
+      "eval_accuracy_red bricks": 0.9211884885764499,
+      "eval_accuracy_road": 0.9820826303534468,
+      "eval_accuracy_rough concrete": 0.6283674673308743,
+      "eval_accuracy_rough red sidewalk": 0.7859060453154257,
+      "eval_accuracy_speedway bricks": 0.991211880677756,
+      "eval_accuracy_steel": 0.919497518579418,
+      "eval_accuracy_tiles": 0.22027325880820384,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983984625069096,
+      "eval_iou_REST": 0.294770206022187,
+      "eval_iou_concrete": 0.8917134548324419,
+      "eval_iou_concrete tiles": 0.8776679273660231,
+      "eval_iou_dark bricks": 0.8367072407447196,
+      "eval_iou_grass": 0.8439809990913524,
+      "eval_iou_red bricks": 0.897383767588679,
+      "eval_iou_road": 0.9606030604660013,
+      "eval_iou_rough concrete": 0.6221577164973391,
+      "eval_iou_rough red sidewalk": 0.6266128645433615,
+      "eval_iou_speedway bricks": 0.9788154358219628,
+      "eval_iou_steel": 0.8677521065378728,
+      "eval_iou_tiles": 0.21900602409638553,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08071931451559067,
+      "eval_mean_accuracy": 0.8000252802486373,
+      "eval_mean_iou": 0.7616273406674939,
+      "eval_overall_accuracy": 0.980080653045137,
+      "eval_runtime": 5.3106,
+      "eval_samples_per_second": 22.22,
+      "eval_steps_per_second": 1.506,
+      "step": 3500
+    },
+    {
+      "epoch": 205.94,
+      "learning_rate": 1.65015479876161e-05,
+      "loss": 0.0201,
+      "step": 3501
+    },
+    {
+      "epoch": 206.0,
+      "learning_rate": 1.6491228070175442e-05,
+      "loss": 0.0189,
+      "step": 3502
+    },
+    {
+      "epoch": 206.06,
+      "learning_rate": 1.6480908152734777e-05,
+      "loss": 0.0193,
+      "step": 3503
+    },
+    {
+      "epoch": 206.12,
+      "learning_rate": 1.647058823529412e-05,
+      "loss": 0.0204,
+      "step": 3504
+    },
+    {
+      "epoch": 206.18,
+      "learning_rate": 1.646026831785346e-05,
+      "loss": 0.0228,
+      "step": 3505
+    },
+    {
+      "epoch": 206.24,
+      "learning_rate": 1.6449948400412797e-05,
+      "loss": 0.0212,
+      "step": 3506
+    },
+    {
+      "epoch": 206.29,
+      "learning_rate": 1.6439628482972136e-05,
+      "loss": 0.0231,
+      "step": 3507
+    },
+    {
+      "epoch": 206.35,
+      "learning_rate": 1.6429308565531475e-05,
+      "loss": 0.0213,
+      "step": 3508
+    },
+    {
+      "epoch": 206.41,
+      "learning_rate": 1.6418988648090817e-05,
+      "loss": 0.0226,
+      "step": 3509
+    },
+    {
+      "epoch": 206.47,
+      "learning_rate": 1.6408668730650156e-05,
+      "loss": 0.0208,
+      "step": 3510
+    },
+    {
+      "epoch": 206.53,
+      "learning_rate": 1.6398348813209495e-05,
+      "loss": 0.0197,
+      "step": 3511
+    },
+    {
+      "epoch": 206.59,
+      "learning_rate": 1.6388028895768834e-05,
+      "loss": 0.0183,
+      "step": 3512
+    },
+    {
+      "epoch": 206.65,
+      "learning_rate": 1.6377708978328173e-05,
+      "loss": 0.0214,
+      "step": 3513
+    },
+    {
+      "epoch": 206.71,
+      "learning_rate": 1.6367389060887515e-05,
+      "loss": 0.0192,
+      "step": 3514
+    },
+    {
+      "epoch": 206.76,
+      "learning_rate": 1.6357069143446854e-05,
+      "loss": 0.0217,
+      "step": 3515
+    },
+    {
+      "epoch": 206.82,
+      "learning_rate": 1.6346749226006193e-05,
+      "loss": 0.0223,
+      "step": 3516
+    },
+    {
+      "epoch": 206.88,
+      "learning_rate": 1.6336429308565532e-05,
+      "loss": 0.0228,
+      "step": 3517
+    },
+    {
+      "epoch": 206.94,
+      "learning_rate": 1.632610939112487e-05,
+      "loss": 0.0212,
+      "step": 3518
+    },
+    {
+      "epoch": 207.0,
+      "learning_rate": 1.6315789473684213e-05,
+      "loss": 0.0193,
+      "step": 3519
+    },
+    {
+      "epoch": 207.06,
+      "learning_rate": 1.6305469556243552e-05,
+      "loss": 0.019,
+      "step": 3520
+    },
+    {
+      "epoch": 207.06,
+      "eval_accuracy_NAT": 0.99323736285455,
+      "eval_accuracy_REST": 0.2926829268292683,
+      "eval_accuracy_concrete": 0.9541276759683783,
+      "eval_accuracy_concrete tiles": 0.9492576007541833,
+      "eval_accuracy_dark bricks": 0.853731600405796,
+      "eval_accuracy_grass": 0.9080834345797045,
+      "eval_accuracy_red bricks": 0.9246485061511424,
+      "eval_accuracy_road": 0.9807572783469013,
+      "eval_accuracy_rough concrete": 0.6291540290431954,
+      "eval_accuracy_rough red sidewalk": 0.7824318125418299,
+      "eval_accuracy_speedway bricks": 0.9918353071172062,
+      "eval_accuracy_steel": 0.9102090590593092,
+      "eval_accuracy_tiles": 0.23326971432033689,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9839699556638067,
+      "eval_iou_REST": 0.28938906752411575,
+      "eval_iou_concrete": 0.8919711257945927,
+      "eval_iou_concrete tiles": 0.8770550704440041,
+      "eval_iou_dark bricks": 0.8379920806775107,
+      "eval_iou_grass": 0.8451312977247128,
+      "eval_iou_red bricks": 0.9020574367766824,
+      "eval_iou_road": 0.9604076512586813,
+      "eval_iou_rough concrete": 0.6222353183840645,
+      "eval_iou_rough red sidewalk": 0.630549663656748,
+      "eval_iou_speedway bricks": 0.979126196428937,
+      "eval_iou_steel": 0.8667537603780724,
+      "eval_iou_tiles": 0.2318159922928709,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0815521776676178,
+      "eval_mean_accuracy": 0.8002635621516772,
+      "eval_mean_iou": 0.7629580474619077,
+      "eval_overall_accuracy": 0.9801769256591797,
+      "eval_runtime": 5.6078,
+      "eval_samples_per_second": 21.042,
+      "eval_steps_per_second": 1.427,
+      "step": 3520
+    },
+    {
+      "epoch": 207.12,
+      "learning_rate": 1.629514963880289e-05,
+      "loss": 0.02,
+      "step": 3521
+    },
+    {
+      "epoch": 207.18,
+      "learning_rate": 1.628482972136223e-05,
+      "loss": 0.0206,
+      "step": 3522
+    },
+    {
+      "epoch": 207.24,
+      "learning_rate": 1.627450980392157e-05,
+      "loss": 0.0189,
+      "step": 3523
+    },
+    {
+      "epoch": 207.29,
+      "learning_rate": 1.626418988648091e-05,
+      "loss": 0.0192,
+      "step": 3524
+    },
+    {
+      "epoch": 207.35,
+      "learning_rate": 1.6253869969040246e-05,
+      "loss": 0.0212,
+      "step": 3525
+    },
+    {
+      "epoch": 207.41,
+      "learning_rate": 1.624355005159959e-05,
+      "loss": 0.0193,
+      "step": 3526
+    },
+    {
+      "epoch": 207.47,
+      "learning_rate": 1.6233230134158927e-05,
+      "loss": 0.0217,
+      "step": 3527
+    },
+    {
+      "epoch": 207.53,
+      "learning_rate": 1.622291021671827e-05,
+      "loss": 0.0198,
+      "step": 3528
+    },
+    {
+      "epoch": 207.59,
+      "learning_rate": 1.6212590299277605e-05,
+      "loss": 0.0218,
+      "step": 3529
+    },
+    {
+      "epoch": 207.65,
+      "learning_rate": 1.6202270381836944e-05,
+      "loss": 0.0259,
+      "step": 3530
+    },
+    {
+      "epoch": 207.71,
+      "learning_rate": 1.6191950464396286e-05,
+      "loss": 0.0244,
+      "step": 3531
+    },
+    {
+      "epoch": 207.76,
+      "learning_rate": 1.6181630546955625e-05,
+      "loss": 0.0194,
+      "step": 3532
+    },
+    {
+      "epoch": 207.82,
+      "learning_rate": 1.6171310629514967e-05,
+      "loss": 0.0214,
+      "step": 3533
+    },
+    {
+      "epoch": 207.88,
+      "learning_rate": 1.6160990712074303e-05,
+      "loss": 0.021,
+      "step": 3534
+    },
+    {
+      "epoch": 207.94,
+      "learning_rate": 1.6150670794633645e-05,
+      "loss": 0.0222,
+      "step": 3535
+    },
+    {
+      "epoch": 208.0,
+      "learning_rate": 1.6140350877192984e-05,
+      "loss": 0.0205,
+      "step": 3536
+    },
+    {
+      "epoch": 208.06,
+      "learning_rate": 1.6130030959752323e-05,
+      "loss": 0.0243,
+      "step": 3537
+    },
+    {
+      "epoch": 208.12,
+      "learning_rate": 1.611971104231166e-05,
+      "loss": 0.0163,
+      "step": 3538
+    },
+    {
+      "epoch": 208.18,
+      "learning_rate": 1.6109391124871e-05,
+      "loss": 0.0217,
+      "step": 3539
+    },
+    {
+      "epoch": 208.24,
+      "learning_rate": 1.6099071207430343e-05,
+      "loss": 0.0192,
+      "step": 3540
+    },
+    {
+      "epoch": 208.24,
+      "eval_accuracy_NAT": 0.9939886232171314,
+      "eval_accuracy_REST": 0.36910569105691055,
+      "eval_accuracy_concrete": 0.9518008466478829,
+      "eval_accuracy_concrete tiles": 0.9420221541362244,
+      "eval_accuracy_dark bricks": 0.8676425549834428,
+      "eval_accuracy_grass": 0.8984309651144746,
+      "eval_accuracy_red bricks": 0.9314037785588752,
+      "eval_accuracy_road": 0.9819794880400342,
+      "eval_accuracy_rough concrete": 0.6263474338424135,
+      "eval_accuracy_rough red sidewalk": 0.7954639454349935,
+      "eval_accuracy_speedway bricks": 0.9911988569258599,
+      "eval_accuracy_steel": 0.9161064936752513,
+      "eval_accuracy_tiles": 0.25008331061225725,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838067211645181,
+      "eval_iou_REST": 0.358609794628752,
+      "eval_iou_concrete": 0.8936835252886507,
+      "eval_iou_concrete tiles": 0.871679678981114,
+      "eval_iou_dark bricks": 0.8468612797758057,
+      "eval_iou_grass": 0.8416251609923778,
+      "eval_iou_red bricks": 0.9064136825227151,
+      "eval_iou_road": 0.9613293111422228,
+      "eval_iou_rough concrete": 0.6196567844320908,
+      "eval_iou_rough red sidewalk": 0.6279272701369473,
+      "eval_iou_speedway bricks": 0.979761214103562,
+      "eval_iou_steel": 0.8692998451582918,
+      "eval_iou_tiles": 0.2475336591801853,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08186282217502594,
+      "eval_mean_accuracy": 0.8088903186342886,
+      "eval_mean_iou": 0.7698606098082486,
+      "eval_overall_accuracy": 0.980279275926493,
+      "eval_runtime": 5.3523,
+      "eval_samples_per_second": 22.047,
+      "eval_steps_per_second": 1.495,
+      "step": 3540
+    },
+    {
+      "epoch": 208.29,
+      "learning_rate": 1.608875128998968e-05,
+      "loss": 0.0203,
+      "step": 3541
+    },
+    {
+      "epoch": 208.35,
+      "learning_rate": 1.607843137254902e-05,
+      "loss": 0.0253,
+      "step": 3542
+    },
+    {
+      "epoch": 208.41,
+      "learning_rate": 1.606811145510836e-05,
+      "loss": 0.022,
+      "step": 3543
+    },
+    {
+      "epoch": 208.47,
+      "learning_rate": 1.6057791537667698e-05,
+      "loss": 0.0184,
+      "step": 3544
+    },
+    {
+      "epoch": 208.53,
+      "learning_rate": 1.604747162022704e-05,
+      "loss": 0.0209,
+      "step": 3545
+    },
+    {
+      "epoch": 208.59,
+      "learning_rate": 1.603715170278638e-05,
+      "loss": 0.0225,
+      "step": 3546
+    },
+    {
+      "epoch": 208.65,
+      "learning_rate": 1.6026831785345718e-05,
+      "loss": 0.0221,
+      "step": 3547
+    },
+    {
+      "epoch": 208.71,
+      "learning_rate": 1.6016511867905057e-05,
+      "loss": 0.0238,
+      "step": 3548
+    },
+    {
+      "epoch": 208.76,
+      "learning_rate": 1.6006191950464396e-05,
+      "loss": 0.021,
+      "step": 3549
+    },
+    {
+      "epoch": 208.82,
+      "learning_rate": 1.5995872033023738e-05,
+      "loss": 0.0193,
+      "step": 3550
+    },
+    {
+      "epoch": 208.88,
+      "learning_rate": 1.5985552115583077e-05,
+      "loss": 0.023,
+      "step": 3551
+    },
+    {
+      "epoch": 208.94,
+      "learning_rate": 1.5975232198142416e-05,
+      "loss": 0.0181,
+      "step": 3552
+    },
+    {
+      "epoch": 209.0,
+      "learning_rate": 1.5964912280701755e-05,
+      "loss": 0.023,
+      "step": 3553
+    },
+    {
+      "epoch": 209.06,
+      "learning_rate": 1.5954592363261094e-05,
+      "loss": 0.0206,
+      "step": 3554
+    },
+    {
+      "epoch": 209.12,
+      "learning_rate": 1.5944272445820436e-05,
+      "loss": 0.0226,
+      "step": 3555
+    },
+    {
+      "epoch": 209.18,
+      "learning_rate": 1.593395252837977e-05,
+      "loss": 0.0203,
+      "step": 3556
+    },
+    {
+      "epoch": 209.24,
+      "learning_rate": 1.5923632610939114e-05,
+      "loss": 0.022,
+      "step": 3557
+    },
+    {
+      "epoch": 209.29,
+      "learning_rate": 1.5913312693498453e-05,
+      "loss": 0.0193,
+      "step": 3558
+    },
+    {
+      "epoch": 209.35,
+      "learning_rate": 1.590299277605779e-05,
+      "loss": 0.0218,
+      "step": 3559
+    },
+    {
+      "epoch": 209.41,
+      "learning_rate": 1.5892672858617134e-05,
+      "loss": 0.0183,
+      "step": 3560
+    },
+    {
+      "epoch": 209.41,
+      "eval_accuracy_NAT": 0.9933793664279292,
+      "eval_accuracy_REST": 0.359349593495935,
+      "eval_accuracy_concrete": 0.957911066809966,
+      "eval_accuracy_concrete tiles": 0.9416686306858355,
+      "eval_accuracy_dark bricks": 0.8504919319334647,
+      "eval_accuracy_grass": 0.9045767144700936,
+      "eval_accuracy_red bricks": 0.9234951669595782,
+      "eval_accuracy_road": 0.9807733108308514,
+      "eval_accuracy_rough concrete": 0.6222001084501755,
+      "eval_accuracy_rough red sidewalk": 0.765203528376661,
+      "eval_accuracy_speedway bricks": 0.9897652160263436,
+      "eval_accuracy_steel": 0.9217615337328262,
+      "eval_accuracy_tiles": 0.24157048077796964,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838521064931751,
+      "eval_iou_REST": 0.3469387755102041,
+      "eval_iou_concrete": 0.8911068967184624,
+      "eval_iou_concrete tiles": 0.8719039825422804,
+      "eval_iou_dark bricks": 0.8346592403351242,
+      "eval_iou_grass": 0.8451640851545371,
+      "eval_iou_red bricks": 0.9016569253043059,
+      "eval_iou_road": 0.9597247207463498,
+      "eval_iou_rough concrete": 0.6159749874642362,
+      "eval_iou_rough red sidewalk": 0.6250675708880044,
+      "eval_iou_speedway bricks": 0.9796648583295905,
+      "eval_iou_steel": 0.8692593744697699,
+      "eval_iou_tiles": 0.23319198713262174,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08291895687580109,
+      "eval_mean_accuracy": 0.8040112806905869,
+      "eval_mean_iou": 0.7660127316222046,
+      "eval_overall_accuracy": 0.9800876035528668,
+      "eval_runtime": 5.3115,
+      "eval_samples_per_second": 22.216,
+      "eval_steps_per_second": 1.506,
+      "step": 3560
+    },
+    {
+      "epoch": 209.47,
+      "learning_rate": 1.588235294117647e-05,
+      "loss": 0.0188,
+      "step": 3561
+    },
+    {
+      "epoch": 209.53,
+      "learning_rate": 1.587203302373581e-05,
+      "loss": 0.0196,
+      "step": 3562
+    },
+    {
+      "epoch": 209.59,
+      "learning_rate": 1.586171310629515e-05,
+      "loss": 0.0218,
+      "step": 3563
+    },
+    {
+      "epoch": 209.65,
+      "learning_rate": 1.585139318885449e-05,
+      "loss": 0.0208,
+      "step": 3564
+    },
+    {
+      "epoch": 209.71,
+      "learning_rate": 1.5841073271413828e-05,
+      "loss": 0.0204,
+      "step": 3565
+    },
+    {
+      "epoch": 209.76,
+      "learning_rate": 1.5830753353973167e-05,
+      "loss": 0.0247,
+      "step": 3566
+    },
+    {
+      "epoch": 209.82,
+      "learning_rate": 1.582043343653251e-05,
+      "loss": 0.0183,
+      "step": 3567
+    },
+    {
+      "epoch": 209.88,
+      "learning_rate": 1.5810113519091848e-05,
+      "loss": 0.019,
+      "step": 3568
+    },
+    {
+      "epoch": 209.94,
+      "learning_rate": 1.579979360165119e-05,
+      "loss": 0.0221,
+      "step": 3569
+    },
+    {
+      "epoch": 210.0,
+      "learning_rate": 1.5789473684210526e-05,
+      "loss": 0.0224,
+      "step": 3570
+    },
+    {
+      "epoch": 210.06,
+      "learning_rate": 1.5779153766769865e-05,
+      "loss": 0.0198,
+      "step": 3571
+    },
+    {
+      "epoch": 210.12,
+      "learning_rate": 1.5768833849329207e-05,
+      "loss": 0.0193,
+      "step": 3572
+    },
+    {
+      "epoch": 210.18,
+      "learning_rate": 1.5758513931888546e-05,
+      "loss": 0.0218,
+      "step": 3573
+    },
+    {
+      "epoch": 210.24,
+      "learning_rate": 1.5748194014447885e-05,
+      "loss": 0.0191,
+      "step": 3574
+    },
+    {
+      "epoch": 210.29,
+      "learning_rate": 1.5737874097007224e-05,
+      "loss": 0.0174,
+      "step": 3575
+    },
+    {
+      "epoch": 210.35,
+      "learning_rate": 1.5727554179566566e-05,
+      "loss": 0.0183,
+      "step": 3576
+    },
+    {
+      "epoch": 210.41,
+      "learning_rate": 1.5717234262125905e-05,
+      "loss": 0.0216,
+      "step": 3577
+    },
+    {
+      "epoch": 210.47,
+      "learning_rate": 1.5706914344685244e-05,
+      "loss": 0.0238,
+      "step": 3578
+    },
+    {
+      "epoch": 210.53,
+      "learning_rate": 1.5696594427244582e-05,
+      "loss": 0.0205,
+      "step": 3579
+    },
+    {
+      "epoch": 210.59,
+      "learning_rate": 1.568627450980392e-05,
+      "loss": 0.0203,
+      "step": 3580
+    },
+    {
+      "epoch": 210.59,
+      "eval_accuracy_NAT": 0.9936146373444922,
+      "eval_accuracy_REST": 0.2731707317073171,
+      "eval_accuracy_concrete": 0.9575028779607069,
+      "eval_accuracy_concrete tiles": 0.936790007070469,
+      "eval_accuracy_dark bricks": 0.8534731925808242,
+      "eval_accuracy_grass": 0.8987053711673044,
+      "eval_accuracy_red bricks": 0.9253075571177505,
+      "eval_accuracy_road": 0.9808518700022072,
+      "eval_accuracy_rough concrete": 0.6302981188065714,
+      "eval_accuracy_rough red sidewalk": 0.8000962557997879,
+      "eval_accuracy_speedway bricks": 0.9898442153635032,
+      "eval_accuracy_steel": 0.9162664241055141,
+      "eval_accuracy_tiles": 0.2411766487927535,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838485155009997,
+      "eval_iou_REST": 0.27009646302250806,
+      "eval_iou_concrete": 0.8924069120041036,
+      "eval_iou_concrete tiles": 0.8761434522891088,
+      "eval_iou_dark bricks": 0.8376833512594465,
+      "eval_iou_grass": 0.8417710412405124,
+      "eval_iou_red bricks": 0.9034749034749034,
+      "eval_iou_road": 0.9604666034509105,
+      "eval_iou_rough concrete": 0.6225779870512066,
+      "eval_iou_rough red sidewalk": 0.6353744909049649,
+      "eval_iou_speedway bricks": 0.9794677677783086,
+      "eval_iou_steel": 0.8693835743110913,
+      "eval_iou_tiles": 0.23676540566262194,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08129559457302094,
+      "eval_mean_accuracy": 0.7997767621399388,
+      "eval_mean_iou": 0.7622661898423604,
+      "eval_overall_accuracy": 0.9801484447414592,
+      "eval_runtime": 5.3113,
+      "eval_samples_per_second": 22.217,
+      "eval_steps_per_second": 1.506,
+      "step": 3580
+    },
+    {
+      "epoch": 210.65,
+      "learning_rate": 1.5675954592363264e-05,
+      "loss": 0.0203,
+      "step": 3581
+    },
+    {
+      "epoch": 210.71,
+      "learning_rate": 1.5665634674922602e-05,
+      "loss": 0.0232,
+      "step": 3582
+    },
+    {
+      "epoch": 210.76,
+      "learning_rate": 1.565531475748194e-05,
+      "loss": 0.0176,
+      "step": 3583
+    },
+    {
+      "epoch": 210.82,
+      "learning_rate": 1.564499484004128e-05,
+      "loss": 0.0202,
+      "step": 3584
+    },
+    {
+      "epoch": 210.88,
+      "learning_rate": 1.563467492260062e-05,
+      "loss": 0.0242,
+      "step": 3585
+    },
+    {
+      "epoch": 210.94,
+      "learning_rate": 1.562435500515996e-05,
+      "loss": 0.0217,
+      "step": 3586
+    },
+    {
+      "epoch": 211.0,
+      "learning_rate": 1.56140350877193e-05,
+      "loss": 0.0181,
+      "step": 3587
+    },
+    {
+      "epoch": 211.06,
+      "learning_rate": 1.560371517027864e-05,
+      "loss": 0.0185,
+      "step": 3588
+    },
+    {
+      "epoch": 211.12,
+      "learning_rate": 1.5593395252837978e-05,
+      "loss": 0.0183,
+      "step": 3589
+    },
+    {
+      "epoch": 211.18,
+      "learning_rate": 1.5583075335397317e-05,
+      "loss": 0.0211,
+      "step": 3590
+    },
+    {
+      "epoch": 211.24,
+      "learning_rate": 1.557275541795666e-05,
+      "loss": 0.0217,
+      "step": 3591
+    },
+    {
+      "epoch": 211.29,
+      "learning_rate": 1.5562435500515995e-05,
+      "loss": 0.0218,
+      "step": 3592
+    },
+    {
+      "epoch": 211.35,
+      "learning_rate": 1.5552115583075337e-05,
+      "loss": 0.0215,
+      "step": 3593
+    },
+    {
+      "epoch": 211.41,
+      "learning_rate": 1.5541795665634676e-05,
+      "loss": 0.0188,
+      "step": 3594
+    },
+    {
+      "epoch": 211.47,
+      "learning_rate": 1.5531475748194015e-05,
+      "loss": 0.0197,
+      "step": 3595
+    },
+    {
+      "epoch": 211.53,
+      "learning_rate": 1.5521155830753353e-05,
+      "loss": 0.0245,
+      "step": 3596
+    },
+    {
+      "epoch": 211.59,
+      "learning_rate": 1.5510835913312692e-05,
+      "loss": 0.0219,
+      "step": 3597
+    },
+    {
+      "epoch": 211.65,
+      "learning_rate": 1.5500515995872035e-05,
+      "loss": 0.0191,
+      "step": 3598
+    },
+    {
+      "epoch": 211.71,
+      "learning_rate": 1.5490196078431373e-05,
+      "loss": 0.0203,
+      "step": 3599
+    },
+    {
+      "epoch": 211.76,
+      "learning_rate": 1.5479876160990712e-05,
+      "loss": 0.0202,
+      "step": 3600
+    },
+    {
+      "epoch": 211.76,
+      "eval_accuracy_NAT": 0.9940251214780189,
+      "eval_accuracy_REST": 0.4032520325203252,
+      "eval_accuracy_concrete": 0.9544223516264109,
+      "eval_accuracy_concrete tiles": 0.9563045015319349,
+      "eval_accuracy_dark bricks": 0.8577752043335949,
+      "eval_accuracy_grass": 0.8901571495077724,
+      "eval_accuracy_red bricks": 0.9382688927943761,
+      "eval_accuracy_road": 0.9820195692499095,
+      "eval_accuracy_rough concrete": 0.6282184973096014,
+      "eval_accuracy_rough red sidewalk": 0.7908241150858406,
+      "eval_accuracy_speedway bricks": 0.9893662779419456,
+      "eval_accuracy_steel": 0.9286835226676395,
+      "eval_accuracy_tiles": 0.2551122421157866,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836903118305419,
+      "eval_iou_REST": 0.389937106918239,
+      "eval_iou_concrete": 0.8925454210129015,
+      "eval_iou_concrete tiles": 0.8661571959185416,
+      "eval_iou_dark bricks": 0.8411551330120459,
+      "eval_iou_grass": 0.838484006071534,
+      "eval_iou_red bricks": 0.9048728813559322,
+      "eval_iou_road": 0.9599357241403809,
+      "eval_iou_rough concrete": 0.6210685062238219,
+      "eval_iou_rough red sidewalk": 0.6217842973789548,
+      "eval_iou_speedway bricks": 0.9795408096265609,
+      "eval_iou_steel": 0.8693399393203599,
+      "eval_iou_tiles": 0.2502079866888519,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08269161731004715,
+      "eval_mean_accuracy": 0.8129561137048582,
+      "eval_mean_iou": 0.7706707168845128,
+      "eval_overall_accuracy": 0.9800128613488149,
+      "eval_runtime": 5.648,
+      "eval_samples_per_second": 20.892,
+      "eval_steps_per_second": 1.416,
+      "step": 3600
+    },
+    {
+      "epoch": 211.82,
+      "learning_rate": 1.546955624355005e-05,
+      "loss": 0.0226,
+      "step": 3601
+    },
+    {
+      "epoch": 211.88,
+      "learning_rate": 1.545923632610939e-05,
+      "loss": 0.0214,
+      "step": 3602
+    },
+    {
+      "epoch": 211.94,
+      "learning_rate": 1.5448916408668732e-05,
+      "loss": 0.0203,
+      "step": 3603
+    },
+    {
+      "epoch": 212.0,
+      "learning_rate": 1.543859649122807e-05,
+      "loss": 0.0215,
+      "step": 3604
+    },
+    {
+      "epoch": 212.06,
+      "learning_rate": 1.542827657378741e-05,
+      "loss": 0.0222,
+      "step": 3605
+    },
+    {
+      "epoch": 212.12,
+      "learning_rate": 1.541795665634675e-05,
+      "loss": 0.0216,
+      "step": 3606
+    },
+    {
+      "epoch": 212.18,
+      "learning_rate": 1.5407636738906088e-05,
+      "loss": 0.019,
+      "step": 3607
+    },
+    {
+      "epoch": 212.24,
+      "learning_rate": 1.539731682146543e-05,
+      "loss": 0.021,
+      "step": 3608
+    },
+    {
+      "epoch": 212.29,
+      "learning_rate": 1.538699690402477e-05,
+      "loss": 0.0175,
+      "step": 3609
+    },
+    {
+      "epoch": 212.35,
+      "learning_rate": 1.5376676986584108e-05,
+      "loss": 0.019,
+      "step": 3610
+    },
+    {
+      "epoch": 212.41,
+      "learning_rate": 1.5366357069143447e-05,
+      "loss": 0.0204,
+      "step": 3611
+    },
+    {
+      "epoch": 212.47,
+      "learning_rate": 1.535603715170279e-05,
+      "loss": 0.0284,
+      "step": 3612
+    },
+    {
+      "epoch": 212.53,
+      "learning_rate": 1.5345717234262128e-05,
+      "loss": 0.0207,
+      "step": 3613
+    },
+    {
+      "epoch": 212.59,
+      "learning_rate": 1.5335397316821463e-05,
+      "loss": 0.0214,
+      "step": 3614
+    },
+    {
+      "epoch": 212.65,
+      "learning_rate": 1.5325077399380806e-05,
+      "loss": 0.0181,
+      "step": 3615
+    },
+    {
+      "epoch": 212.71,
+      "learning_rate": 1.5314757481940144e-05,
+      "loss": 0.0214,
+      "step": 3616
+    },
+    {
+      "epoch": 212.76,
+      "learning_rate": 1.5304437564499487e-05,
+      "loss": 0.0204,
+      "step": 3617
+    },
+    {
+      "epoch": 212.82,
+      "learning_rate": 1.5294117647058826e-05,
+      "loss": 0.0204,
+      "step": 3618
+    },
+    {
+      "epoch": 212.88,
+      "learning_rate": 1.5283797729618164e-05,
+      "loss": 0.0243,
+      "step": 3619
+    },
+    {
+      "epoch": 212.94,
+      "learning_rate": 1.5273477812177503e-05,
+      "loss": 0.0166,
+      "step": 3620
+    },
+    {
+      "epoch": 212.94,
+      "eval_accuracy_NAT": 0.9943533901789703,
+      "eval_accuracy_REST": 0.28780487804878047,
+      "eval_accuracy_concrete": 0.9547262000675728,
+      "eval_accuracy_concrete tiles": 0.9338439783172284,
+      "eval_accuracy_dark bricks": 0.8550092835403786,
+      "eval_accuracy_grass": 0.8885182830129406,
+      "eval_accuracy_red bricks": 0.9355777680140598,
+      "eval_accuracy_road": 0.982352510499941,
+      "eval_accuracy_rough concrete": 0.6238328198833267,
+      "eval_accuracy_rough red sidewalk": 0.7580294632987163,
+      "eval_accuracy_speedway bricks": 0.9892365545184536,
+      "eval_accuracy_steel": 0.9279113585590268,
+      "eval_accuracy_tiles": 0.2507800902784089,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835843441278214,
+      "eval_iou_REST": 0.2822966507177033,
+      "eval_iou_concrete": 0.8923195958595753,
+      "eval_iou_concrete tiles": 0.8737540795624945,
+      "eval_iou_dark bricks": 0.838415263622201,
+      "eval_iou_grass": 0.8380988677318758,
+      "eval_iou_red bricks": 0.9061170212765958,
+      "eval_iou_road": 0.9597132000446917,
+      "eval_iou_rough concrete": 0.6175658616580739,
+      "eval_iou_rough red sidewalk": 0.6063569976299611,
+      "eval_iou_speedway bricks": 0.9796562070910109,
+      "eval_iou_steel": 0.8700091610493736,
+      "eval_iou_tiles": 0.24832013438924885,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08346478641033173,
+      "eval_mean_accuracy": 0.798613582939831,
+      "eval_mean_iou": 0.7612467219046637,
+      "eval_overall_accuracy": 0.9799423217773438,
+      "eval_runtime": 5.238,
+      "eval_samples_per_second": 22.528,
+      "eval_steps_per_second": 1.527,
+      "step": 3620
+    },
+    {
+      "epoch": 213.0,
+      "learning_rate": 1.5263157894736842e-05,
+      "loss": 0.0181,
+      "step": 3621
+    },
+    {
+      "epoch": 213.06,
+      "learning_rate": 1.5252837977296183e-05,
+      "loss": 0.0205,
+      "step": 3622
+    },
+    {
+      "epoch": 213.12,
+      "learning_rate": 1.524251805985552e-05,
+      "loss": 0.0225,
+      "step": 3623
+    },
+    {
+      "epoch": 213.18,
+      "learning_rate": 1.523219814241486e-05,
+      "loss": 0.0193,
+      "step": 3624
+    },
+    {
+      "epoch": 213.24,
+      "learning_rate": 1.5221878224974201e-05,
+      "loss": 0.0215,
+      "step": 3625
+    },
+    {
+      "epoch": 213.29,
+      "learning_rate": 1.5211558307533542e-05,
+      "loss": 0.0212,
+      "step": 3626
+    },
+    {
+      "epoch": 213.35,
+      "learning_rate": 1.520123839009288e-05,
+      "loss": 0.02,
+      "step": 3627
+    },
+    {
+      "epoch": 213.41,
+      "learning_rate": 1.519091847265222e-05,
+      "loss": 0.0189,
+      "step": 3628
+    },
+    {
+      "epoch": 213.47,
+      "learning_rate": 1.5180598555211558e-05,
+      "loss": 0.0191,
+      "step": 3629
+    },
+    {
+      "epoch": 213.53,
+      "learning_rate": 1.5170278637770899e-05,
+      "loss": 0.0217,
+      "step": 3630
+    },
+    {
+      "epoch": 213.59,
+      "learning_rate": 1.515995872033024e-05,
+      "loss": 0.0222,
+      "step": 3631
+    },
+    {
+      "epoch": 213.65,
+      "learning_rate": 1.5149638802889577e-05,
+      "loss": 0.0205,
+      "step": 3632
+    },
+    {
+      "epoch": 213.71,
+      "learning_rate": 1.5139318885448917e-05,
+      "loss": 0.0229,
+      "step": 3633
+    },
+    {
+      "epoch": 213.76,
+      "learning_rate": 1.5128998968008256e-05,
+      "loss": 0.0188,
+      "step": 3634
+    },
+    {
+      "epoch": 213.82,
+      "learning_rate": 1.5118679050567597e-05,
+      "loss": 0.0198,
+      "step": 3635
+    },
+    {
+      "epoch": 213.88,
+      "learning_rate": 1.5108359133126937e-05,
+      "loss": 0.02,
+      "step": 3636
+    },
+    {
+      "epoch": 213.94,
+      "learning_rate": 1.5098039215686274e-05,
+      "loss": 0.0233,
+      "step": 3637
+    },
+    {
+      "epoch": 214.0,
+      "learning_rate": 1.5087719298245615e-05,
+      "loss": 0.0202,
+      "step": 3638
+    },
+    {
+      "epoch": 214.06,
+      "learning_rate": 1.5077399380804955e-05,
+      "loss": 0.0194,
+      "step": 3639
+    },
+    {
+      "epoch": 214.12,
+      "learning_rate": 1.5067079463364294e-05,
+      "loss": 0.0211,
+      "step": 3640
+    },
+    {
+      "epoch": 214.12,
+      "eval_accuracy_NAT": 0.9942175325461247,
+      "eval_accuracy_REST": 0.3252032520325203,
+      "eval_accuracy_concrete": 0.9541884456566108,
+      "eval_accuracy_concrete tiles": 0.9483384397831723,
+      "eval_accuracy_dark bricks": 0.8647617862680168,
+      "eval_accuracy_grass": 0.8890424931966223,
+      "eval_accuracy_red bricks": 0.9359072934973638,
+      "eval_accuracy_road": 0.9805670262040261,
+      "eval_accuracy_rough concrete": 0.6262640106305007,
+      "eval_accuracy_rough red sidewalk": 0.7818377337775138,
+      "eval_accuracy_speedway bricks": 0.9905999357037932,
+      "eval_accuracy_steel": 0.9217515380809348,
+      "eval_accuracy_tiles": 0.2551122421157866,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835946757499551,
+      "eval_iou_REST": 0.3179650238473768,
+      "eval_iou_concrete": 0.8936125661196693,
+      "eval_iou_concrete tiles": 0.8714428033092216,
+      "eval_iou_dark bricks": 0.8477040206025979,
+      "eval_iou_grass": 0.8377805300612665,
+      "eval_iou_red bricks": 0.9079332942618147,
+      "eval_iou_road": 0.961240513132962,
+      "eval_iou_rough concrete": 0.6199433728543621,
+      "eval_iou_rough red sidewalk": 0.6162052595081879,
+      "eval_iou_speedway bricks": 0.9796437913453447,
+      "eval_iou_steel": 0.8705382850777408,
+      "eval_iou_tiles": 0.25125313283208023,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08287326991558075,
+      "eval_mean_accuracy": 0.8052147484225375,
+      "eval_mean_iou": 0.7660659437463523,
+      "eval_overall_accuracy": 0.9801096512099444,
+      "eval_runtime": 5.7247,
+      "eval_samples_per_second": 20.613,
+      "eval_steps_per_second": 1.397,
+      "step": 3640
+    },
+    {
+      "epoch": 214.18,
+      "learning_rate": 1.5056759545923631e-05,
+      "loss": 0.0213,
+      "step": 3641
+    },
+    {
+      "epoch": 214.24,
+      "learning_rate": 1.5046439628482972e-05,
+      "loss": 0.0214,
+      "step": 3642
+    },
+    {
+      "epoch": 214.29,
+      "learning_rate": 1.5036119711042313e-05,
+      "loss": 0.0186,
+      "step": 3643
+    },
+    {
+      "epoch": 214.35,
+      "learning_rate": 1.5025799793601653e-05,
+      "loss": 0.0196,
+      "step": 3644
+    },
+    {
+      "epoch": 214.41,
+      "learning_rate": 1.5015479876160992e-05,
+      "loss": 0.0244,
+      "step": 3645
+    },
+    {
+      "epoch": 214.47,
+      "learning_rate": 1.500515995872033e-05,
+      "loss": 0.0207,
+      "step": 3646
+    },
+    {
+      "epoch": 214.53,
+      "learning_rate": 1.499484004127967e-05,
+      "loss": 0.0239,
+      "step": 3647
+    },
+    {
+      "epoch": 214.59,
+      "learning_rate": 1.498452012383901e-05,
+      "loss": 0.0179,
+      "step": 3648
+    },
+    {
+      "epoch": 214.65,
+      "learning_rate": 1.4974200206398351e-05,
+      "loss": 0.0228,
+      "step": 3649
+    },
+    {
+      "epoch": 214.71,
+      "learning_rate": 1.4963880288957688e-05,
+      "loss": 0.0195,
+      "step": 3650
+    },
+    {
+      "epoch": 214.76,
+      "learning_rate": 1.4953560371517029e-05,
+      "loss": 0.0174,
+      "step": 3651
+    },
+    {
+      "epoch": 214.82,
+      "learning_rate": 1.4943240454076368e-05,
+      "loss": 0.0231,
+      "step": 3652
+    },
+    {
+      "epoch": 214.88,
+      "learning_rate": 1.4932920536635708e-05,
+      "loss": 0.0227,
+      "step": 3653
+    },
+    {
+      "epoch": 214.94,
+      "learning_rate": 1.4922600619195049e-05,
+      "loss": 0.0165,
+      "step": 3654
+    },
+    {
+      "epoch": 215.0,
+      "learning_rate": 1.4912280701754386e-05,
+      "loss": 0.02,
+      "step": 3655
+    },
+    {
+      "epoch": 215.06,
+      "learning_rate": 1.4901960784313726e-05,
+      "loss": 0.0171,
+      "step": 3656
+    },
+    {
+      "epoch": 215.12,
+      "learning_rate": 1.4891640866873065e-05,
+      "loss": 0.0203,
+      "step": 3657
+    },
+    {
+      "epoch": 215.18,
+      "learning_rate": 1.4881320949432406e-05,
+      "loss": 0.0249,
+      "step": 3658
+    },
+    {
+      "epoch": 215.24,
+      "learning_rate": 1.4871001031991743e-05,
+      "loss": 0.0213,
+      "step": 3659
+    },
+    {
+      "epoch": 215.29,
+      "learning_rate": 1.4860681114551084e-05,
+      "loss": 0.0217,
+      "step": 3660
+    },
+    {
+      "epoch": 215.29,
+      "eval_accuracy_NAT": 0.9937655902698763,
+      "eval_accuracy_REST": 0.4682926829268293,
+      "eval_accuracy_concrete": 0.9548389488602053,
+      "eval_accuracy_concrete tiles": 0.9302851755833137,
+      "eval_accuracy_dark bricks": 0.8702122772428842,
+      "eval_accuracy_grass": 0.8986523824122752,
+      "eval_accuracy_red bricks": 0.9306898066783831,
+      "eval_accuracy_road": 0.982944643573833,
+      "eval_accuracy_rough concrete": 0.6274855648049387,
+      "eval_accuracy_rough red sidewalk": 0.7808526158265591,
+      "eval_accuracy_speedway bricks": 0.9910262922132358,
+      "eval_accuracy_steel": 0.9139899143872415,
+      "eval_accuracy_tiles": 0.24781120300524098,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840656975269255,
+      "eval_iou_REST": 0.4485981308411215,
+      "eval_iou_concrete": 0.8930460886885178,
+      "eval_iou_concrete tiles": 0.8733323008164259,
+      "eval_iou_dark bricks": 0.8507443135567053,
+      "eval_iou_grass": 0.8446424156399023,
+      "eval_iou_red bricks": 0.9019586970406642,
+      "eval_iou_road": 0.9605283909612663,
+      "eval_iou_rough concrete": 0.6204645352879482,
+      "eval_iou_rough red sidewalk": 0.6242004905261133,
+      "eval_iou_speedway bricks": 0.9797044823002276,
+      "eval_iou_steel": 0.8681498395457912,
+      "eval_iou_tiles": 0.24305452384489676,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08056919276714325,
+      "eval_mean_accuracy": 0.8146805459834474,
+      "eval_mean_iou": 0.7763453774289619,
+      "eval_overall_accuracy": 0.9803522724216267,
+      "eval_runtime": 5.4821,
+      "eval_samples_per_second": 21.524,
+      "eval_steps_per_second": 1.459,
+      "step": 3660
+    },
+    {
+      "epoch": 215.35,
+      "learning_rate": 1.4850361197110424e-05,
+      "loss": 0.0181,
+      "step": 3661
+    },
+    {
+      "epoch": 215.41,
+      "learning_rate": 1.4840041279669765e-05,
+      "loss": 0.0203,
+      "step": 3662
+    },
+    {
+      "epoch": 215.47,
+      "learning_rate": 1.4829721362229104e-05,
+      "loss": 0.019,
+      "step": 3663
+    },
+    {
+      "epoch": 215.53,
+      "learning_rate": 1.481940144478844e-05,
+      "loss": 0.0191,
+      "step": 3664
+    },
+    {
+      "epoch": 215.59,
+      "learning_rate": 1.4809081527347781e-05,
+      "loss": 0.0199,
+      "step": 3665
+    },
+    {
+      "epoch": 215.65,
+      "learning_rate": 1.4798761609907122e-05,
+      "loss": 0.0217,
+      "step": 3666
+    },
+    {
+      "epoch": 215.71,
+      "learning_rate": 1.4788441692466463e-05,
+      "loss": 0.0219,
+      "step": 3667
+    },
+    {
+      "epoch": 215.76,
+      "learning_rate": 1.47781217750258e-05,
+      "loss": 0.0243,
+      "step": 3668
+    },
+    {
+      "epoch": 215.82,
+      "learning_rate": 1.476780185758514e-05,
+      "loss": 0.0216,
+      "step": 3669
+    },
+    {
+      "epoch": 215.88,
+      "learning_rate": 1.4757481940144479e-05,
+      "loss": 0.0181,
+      "step": 3670
+    },
+    {
+      "epoch": 215.94,
+      "learning_rate": 1.474716202270382e-05,
+      "loss": 0.0203,
+      "step": 3671
+    },
+    {
+      "epoch": 216.0,
+      "learning_rate": 1.4736842105263157e-05,
+      "loss": 0.0188,
+      "step": 3672
+    },
+    {
+      "epoch": 216.06,
+      "learning_rate": 1.4726522187822497e-05,
+      "loss": 0.019,
+      "step": 3673
+    },
+    {
+      "epoch": 216.12,
+      "learning_rate": 1.4716202270381838e-05,
+      "loss": 0.0193,
+      "step": 3674
+    },
+    {
+      "epoch": 216.18,
+      "learning_rate": 1.4705882352941177e-05,
+      "loss": 0.0241,
+      "step": 3675
+    },
+    {
+      "epoch": 216.24,
+      "learning_rate": 1.4695562435500517e-05,
+      "loss": 0.0195,
+      "step": 3676
+    },
+    {
+      "epoch": 216.29,
+      "learning_rate": 1.4685242518059855e-05,
+      "loss": 0.0212,
+      "step": 3677
+    },
+    {
+      "epoch": 216.35,
+      "learning_rate": 1.4674922600619195e-05,
+      "loss": 0.0176,
+      "step": 3678
+    },
+    {
+      "epoch": 216.41,
+      "learning_rate": 1.4664602683178536e-05,
+      "loss": 0.0219,
+      "step": 3679
+    },
+    {
+      "epoch": 216.47,
+      "learning_rate": 1.4654282765737876e-05,
+      "loss": 0.0191,
+      "step": 3680
+    },
+    {
+      "epoch": 216.47,
+      "eval_accuracy_NAT": 0.9940533173280103,
+      "eval_accuracy_REST": 0.3772357723577236,
+      "eval_accuracy_concrete": 0.9545694983557786,
+      "eval_accuracy_concrete tiles": 0.9321942022154136,
+      "eval_accuracy_dark bricks": 0.8528271730183948,
+      "eval_accuracy_grass": 0.8987858005276166,
+      "eval_accuracy_red bricks": 0.929591388400703,
+      "eval_accuracy_road": 0.9824267943422433,
+      "eval_accuracy_rough concrete": 0.6285045197504454,
+      "eval_accuracy_rough red sidewalk": 0.7986148188811767,
+      "eval_accuracy_speedway bricks": 0.989320352079996,
+      "eval_accuracy_steel": 0.9253299814580658,
+      "eval_accuracy_tiles": 0.25344602987064135,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9839157986721624,
+      "eval_iou_REST": 0.36942675159235666,
+      "eval_iou_concrete": 0.8941230464988232,
+      "eval_iou_concrete tiles": 0.872576055064087,
+      "eval_iou_dark bricks": 0.8371908396946565,
+      "eval_iou_grass": 0.8426993007241169,
+      "eval_iou_red bricks": 0.9067338083248513,
+      "eval_iou_road": 0.9605485570489612,
+      "eval_iou_rough concrete": 0.6208173187284061,
+      "eval_iou_rough red sidewalk": 0.6239989188617494,
+      "eval_iou_speedway bricks": 0.979729715969953,
+      "eval_iou_steel": 0.8701638373470193,
+      "eval_iou_tiles": 0.25135955292491663,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08108249306678772,
+      "eval_mean_accuracy": 0.8089922806604777,
+      "eval_mean_iou": 0.7702525770347738,
+      "eval_overall_accuracy": 0.9802707413495597,
+      "eval_runtime": 5.451,
+      "eval_samples_per_second": 21.648,
+      "eval_steps_per_second": 1.468,
+      "step": 3680
+    },
+    {
+      "epoch": 216.53,
+      "learning_rate": 1.4643962848297213e-05,
+      "loss": 0.0179,
+      "step": 3681
+    },
+    {
+      "epoch": 216.59,
+      "learning_rate": 1.4633642930856552e-05,
+      "loss": 0.0211,
+      "step": 3682
+    },
+    {
+      "epoch": 216.65,
+      "learning_rate": 1.4623323013415893e-05,
+      "loss": 0.0183,
+      "step": 3683
+    },
+    {
+      "epoch": 216.71,
+      "learning_rate": 1.4613003095975234e-05,
+      "loss": 0.0232,
+      "step": 3684
+    },
+    {
+      "epoch": 216.76,
+      "learning_rate": 1.4602683178534574e-05,
+      "loss": 0.0198,
+      "step": 3685
+    },
+    {
+      "epoch": 216.82,
+      "learning_rate": 1.4592363261093911e-05,
+      "loss": 0.0201,
+      "step": 3686
+    },
+    {
+      "epoch": 216.88,
+      "learning_rate": 1.4582043343653252e-05,
+      "loss": 0.0202,
+      "step": 3687
+    },
+    {
+      "epoch": 216.94,
+      "learning_rate": 1.457172342621259e-05,
+      "loss": 0.0226,
+      "step": 3688
+    },
+    {
+      "epoch": 217.0,
+      "learning_rate": 1.4561403508771931e-05,
+      "loss": 0.0264,
+      "step": 3689
+    },
+    {
+      "epoch": 217.06,
+      "learning_rate": 1.4551083591331268e-05,
+      "loss": 0.0215,
+      "step": 3690
+    },
+    {
+      "epoch": 217.12,
+      "learning_rate": 1.4540763673890609e-05,
+      "loss": 0.0185,
+      "step": 3691
+    },
+    {
+      "epoch": 217.18,
+      "learning_rate": 1.453044375644995e-05,
+      "loss": 0.0236,
+      "step": 3692
+    },
+    {
+      "epoch": 217.24,
+      "learning_rate": 1.4520123839009288e-05,
+      "loss": 0.0186,
+      "step": 3693
+    },
+    {
+      "epoch": 217.29,
+      "learning_rate": 1.4509803921568629e-05,
+      "loss": 0.019,
+      "step": 3694
+    },
+    {
+      "epoch": 217.35,
+      "learning_rate": 1.4499484004127966e-05,
+      "loss": 0.0179,
+      "step": 3695
+    },
+    {
+      "epoch": 217.41,
+      "learning_rate": 1.4489164086687307e-05,
+      "loss": 0.0231,
+      "step": 3696
+    },
+    {
+      "epoch": 217.47,
+      "learning_rate": 1.4478844169246647e-05,
+      "loss": 0.02,
+      "step": 3697
+    },
+    {
+      "epoch": 217.53,
+      "learning_rate": 1.4468524251805988e-05,
+      "loss": 0.0201,
+      "step": 3698
+    },
+    {
+      "epoch": 217.59,
+      "learning_rate": 1.4458204334365325e-05,
+      "loss": 0.0206,
+      "step": 3699
+    },
+    {
+      "epoch": 217.65,
+      "learning_rate": 1.4447884416924664e-05,
+      "loss": 0.0202,
+      "step": 3700
+    },
+    {
+      "epoch": 217.65,
+      "eval_accuracy_NAT": 0.9935789477599907,
+      "eval_accuracy_REST": 0.3723577235772358,
+      "eval_accuracy_concrete": 0.9585409312515193,
+      "eval_accuracy_concrete tiles": 0.9363186424699506,
+      "eval_accuracy_dark bricks": 0.8553251153264552,
+      "eval_accuracy_grass": 0.8999780475157736,
+      "eval_accuracy_red bricks": 0.9306898066783831,
+      "eval_accuracy_road": 0.9813194841174198,
+      "eval_accuracy_rough concrete": 0.6279026808645028,
+      "eval_accuracy_rough red sidewalk": 0.7648425691274562,
+      "eval_accuracy_speedway bricks": 0.9895263329981431,
+      "eval_accuracy_steel": 0.9255223977569758,
+      "eval_accuracy_tiles": 0.23957102608379532,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840489855441429,
+      "eval_iou_REST": 0.365814696485623,
+      "eval_iou_concrete": 0.8922581526553631,
+      "eval_iou_concrete tiles": 0.8792882121198707,
+      "eval_iou_dark bricks": 0.8385872395527885,
+      "eval_iou_grass": 0.8432440962753074,
+      "eval_iou_red bricks": 0.9072705857158154,
+      "eval_iou_road": 0.9599368910366661,
+      "eval_iou_rough concrete": 0.6208257723941272,
+      "eval_iou_rough red sidewalk": 0.6135562955455818,
+      "eval_iou_speedway bricks": 0.9797576215179833,
+      "eval_iou_steel": 0.8707704026934,
+      "eval_iou_tiles": 0.23723525529489412,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08232484757900238,
+      "eval_mean_accuracy": 0.8058056696559693,
+      "eval_mean_iou": 0.7686610928331971,
+      "eval_overall_accuracy": 0.9801971306235103,
+      "eval_runtime": 5.2171,
+      "eval_samples_per_second": 22.618,
+      "eval_steps_per_second": 1.533,
+      "step": 3700
+    },
+    {
+      "epoch": 217.71,
+      "learning_rate": 1.4437564499484004e-05,
+      "loss": 0.021,
+      "step": 3701
+    },
+    {
+      "epoch": 217.76,
+      "learning_rate": 1.4427244582043345e-05,
+      "loss": 0.0183,
+      "step": 3702
+    },
+    {
+      "epoch": 217.82,
+      "learning_rate": 1.4416924664602686e-05,
+      "loss": 0.0191,
+      "step": 3703
+    },
+    {
+      "epoch": 217.88,
+      "learning_rate": 1.4406604747162023e-05,
+      "loss": 0.0204,
+      "step": 3704
+    },
+    {
+      "epoch": 217.94,
+      "learning_rate": 1.4396284829721363e-05,
+      "loss": 0.0223,
+      "step": 3705
+    },
+    {
+      "epoch": 218.0,
+      "learning_rate": 1.4385964912280702e-05,
+      "loss": 0.0239,
+      "step": 3706
+    },
+    {
+      "epoch": 218.06,
+      "learning_rate": 1.4375644994840043e-05,
+      "loss": 0.0216,
+      "step": 3707
+    },
+    {
+      "epoch": 218.12,
+      "learning_rate": 1.436532507739938e-05,
+      "loss": 0.0193,
+      "step": 3708
+    },
+    {
+      "epoch": 218.18,
+      "learning_rate": 1.435500515995872e-05,
+      "loss": 0.0184,
+      "step": 3709
+    },
+    {
+      "epoch": 218.24,
+      "learning_rate": 1.4344685242518061e-05,
+      "loss": 0.0208,
+      "step": 3710
+    },
+    {
+      "epoch": 218.29,
+      "learning_rate": 1.43343653250774e-05,
+      "loss": 0.0214,
+      "step": 3711
+    },
+    {
+      "epoch": 218.35,
+      "learning_rate": 1.432404540763674e-05,
+      "loss": 0.0198,
+      "step": 3712
+    },
+    {
+      "epoch": 218.41,
+      "learning_rate": 1.4313725490196078e-05,
+      "loss": 0.0235,
+      "step": 3713
+    },
+    {
+      "epoch": 218.47,
+      "learning_rate": 1.4303405572755418e-05,
+      "loss": 0.0235,
+      "step": 3714
+    },
+    {
+      "epoch": 218.53,
+      "learning_rate": 1.4293085655314759e-05,
+      "loss": 0.0207,
+      "step": 3715
+    },
+    {
+      "epoch": 218.59,
+      "learning_rate": 1.42827657378741e-05,
+      "loss": 0.0191,
+      "step": 3716
+    },
+    {
+      "epoch": 218.65,
+      "learning_rate": 1.4272445820433437e-05,
+      "loss": 0.0207,
+      "step": 3717
+    },
+    {
+      "epoch": 218.71,
+      "learning_rate": 1.4262125902992775e-05,
+      "loss": 0.0185,
+      "step": 3718
+    },
+    {
+      "epoch": 218.76,
+      "learning_rate": 1.4251805985552116e-05,
+      "loss": 0.0198,
+      "step": 3719
+    },
+    {
+      "epoch": 218.82,
+      "learning_rate": 1.4241486068111457e-05,
+      "loss": 0.0192,
+      "step": 3720
+    },
+    {
+      "epoch": 218.82,
+      "eval_accuracy_NAT": 0.993848075261247,
+      "eval_accuracy_REST": 0.3886178861788618,
+      "eval_accuracy_concrete": 0.9555666563218057,
+      "eval_accuracy_concrete tiles": 0.9488333726137167,
+      "eval_accuracy_dark bricks": 0.870470685067856,
+      "eval_accuracy_grass": 0.8987905316664585,
+      "eval_accuracy_red bricks": 0.9314586994727593,
+      "eval_accuracy_road": 0.9812692490010426,
+      "eval_accuracy_rough concrete": 0.6289752650176679,
+      "eval_accuracy_rough red sidewalk": 0.7773934230216801,
+      "eval_accuracy_speedway bricks": 0.9896910149135668,
+      "eval_accuracy_steel": 0.9304452563135036,
+      "eval_accuracy_tiles": 0.2486594565118604,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840034934660048,
+      "eval_iou_REST": 0.37936507936507935,
+      "eval_iou_concrete": 0.8944150640194353,
+      "eval_iou_concrete tiles": 0.8816740396828873,
+      "eval_iou_dark bricks": 0.8481045117794883,
+      "eval_iou_grass": 0.8428577766735288,
+      "eval_iou_red bricks": 0.9072429656574302,
+      "eval_iou_road": 0.9610050568966293,
+      "eval_iou_rough concrete": 0.6228256508296159,
+      "eval_iou_rough red sidewalk": 0.6139870523252361,
+      "eval_iou_speedway bricks": 0.9797042465135997,
+      "eval_iou_steel": 0.8704169510860926,
+      "eval_iou_tiles": 0.24512468269374346,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0816156342625618,
+      "eval_mean_accuracy": 0.8110784285663097,
+      "eval_mean_iou": 0.771594351614521,
+      "eval_overall_accuracy": 0.9803478111655025,
+      "eval_runtime": 5.6131,
+      "eval_samples_per_second": 21.022,
+      "eval_steps_per_second": 1.425,
+      "step": 3720
+    },
+    {
+      "epoch": 218.88,
+      "learning_rate": 1.4231166150670797e-05,
+      "loss": 0.0221,
+      "step": 3721
+    },
+    {
+      "epoch": 218.94,
+      "learning_rate": 1.4220846233230134e-05,
+      "loss": 0.0186,
+      "step": 3722
+    },
+    {
+      "epoch": 219.0,
+      "learning_rate": 1.4210526315789475e-05,
+      "loss": 0.0179,
+      "step": 3723
+    },
+    {
+      "epoch": 219.06,
+      "learning_rate": 1.4200206398348814e-05,
+      "loss": 0.0208,
+      "step": 3724
+    },
+    {
+      "epoch": 219.12,
+      "learning_rate": 1.4189886480908154e-05,
+      "loss": 0.0197,
+      "step": 3725
+    },
+    {
+      "epoch": 219.18,
+      "learning_rate": 1.4179566563467492e-05,
+      "loss": 0.0177,
+      "step": 3726
+    },
+    {
+      "epoch": 219.24,
+      "learning_rate": 1.4169246646026832e-05,
+      "loss": 0.0196,
+      "step": 3727
+    },
+    {
+      "epoch": 219.29,
+      "learning_rate": 1.4158926728586173e-05,
+      "loss": 0.0208,
+      "step": 3728
+    },
+    {
+      "epoch": 219.35,
+      "learning_rate": 1.4148606811145512e-05,
+      "loss": 0.0216,
+      "step": 3729
+    },
+    {
+      "epoch": 219.41,
+      "learning_rate": 1.4138286893704852e-05,
+      "loss": 0.0211,
+      "step": 3730
+    },
+    {
+      "epoch": 219.47,
+      "learning_rate": 1.412796697626419e-05,
+      "loss": 0.0178,
+      "step": 3731
+    },
+    {
+      "epoch": 219.53,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0202,
+      "step": 3732
+    },
+    {
+      "epoch": 219.59,
+      "learning_rate": 1.410732714138287e-05,
+      "loss": 0.0199,
+      "step": 3733
+    },
+    {
+      "epoch": 219.65,
+      "learning_rate": 1.4097007223942211e-05,
+      "loss": 0.0193,
+      "step": 3734
+    },
+    {
+      "epoch": 219.71,
+      "learning_rate": 1.4086687306501548e-05,
+      "loss": 0.0177,
+      "step": 3735
+    },
+    {
+      "epoch": 219.76,
+      "learning_rate": 1.4076367389060887e-05,
+      "loss": 0.0207,
+      "step": 3736
+    },
+    {
+      "epoch": 219.82,
+      "learning_rate": 1.4066047471620228e-05,
+      "loss": 0.021,
+      "step": 3737
+    },
+    {
+      "epoch": 219.88,
+      "learning_rate": 1.4055727554179568e-05,
+      "loss": 0.0208,
+      "step": 3738
+    },
+    {
+      "epoch": 219.94,
+      "learning_rate": 1.4045407636738909e-05,
+      "loss": 0.0204,
+      "step": 3739
+    },
+    {
+      "epoch": 220.0,
+      "learning_rate": 1.4035087719298246e-05,
+      "loss": 0.0211,
+      "step": 3740
+    },
+    {
+      "epoch": 220.0,
+      "eval_accuracy_NAT": 0.9943279977404503,
+      "eval_accuracy_REST": 0.34471544715447155,
+      "eval_accuracy_concrete": 0.9568344113901506,
+      "eval_accuracy_concrete tiles": 0.9371906669809097,
+      "eval_accuracy_dark bricks": 0.8627663036196237,
+      "eval_accuracy_grass": 0.8931519603946906,
+      "eval_accuracy_red bricks": 0.9250878734622144,
+      "eval_accuracy_road": 0.9792641196750108,
+      "eval_accuracy_rough concrete": 0.6242439771420399,
+      "eval_accuracy_rough red sidewalk": 0.7698734386632475,
+      "eval_accuracy_speedway bricks": 0.9896671951568095,
+      "eval_accuracy_steel": 0.9230509728268204,
+      "eval_accuracy_tiles": 0.2349056317973886,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838821169286156,
+      "eval_iou_REST": 0.33865814696485624,
+      "eval_iou_concrete": 0.8930426745591765,
+      "eval_iou_concrete tiles": 0.8822354846582211,
+      "eval_iou_dark bricks": 0.8454782058195972,
+      "eval_iou_grass": 0.8412629800189481,
+      "eval_iou_red bricks": 0.9036965502441118,
+      "eval_iou_road": 0.9599813913233902,
+      "eval_iou_rough concrete": 0.6189879700314339,
+      "eval_iou_rough red sidewalk": 0.6110746347053767,
+      "eval_iou_speedway bricks": 0.9799193727506095,
+      "eval_iou_steel": 0.8703194948400169,
+      "eval_iou_tiles": 0.2296868983085992,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08367869257926941,
+      "eval_mean_accuracy": 0.8026984612310635,
+      "eval_mean_iou": 0.7660173785502272,
+      "eval_overall_accuracy": 0.9801834235757084,
+      "eval_runtime": 5.7454,
+      "eval_samples_per_second": 20.538,
+      "eval_steps_per_second": 1.392,
+      "step": 3740
+    },
+    {
+      "epoch": 220.06,
+      "learning_rate": 1.4024767801857586e-05,
+      "loss": 0.022,
+      "step": 3741
+    },
+    {
+      "epoch": 220.12,
+      "learning_rate": 1.4014447884416925e-05,
+      "loss": 0.0224,
+      "step": 3742
+    },
+    {
+      "epoch": 220.18,
+      "learning_rate": 1.4004127966976266e-05,
+      "loss": 0.018,
+      "step": 3743
+    },
+    {
+      "epoch": 220.24,
+      "learning_rate": 1.3993808049535603e-05,
+      "loss": 0.0181,
+      "step": 3744
+    },
+    {
+      "epoch": 220.29,
+      "learning_rate": 1.3983488132094944e-05,
+      "loss": 0.0205,
+      "step": 3745
+    },
+    {
+      "epoch": 220.35,
+      "learning_rate": 1.3973168214654284e-05,
+      "loss": 0.0179,
+      "step": 3746
+    },
+    {
+      "epoch": 220.41,
+      "learning_rate": 1.3962848297213623e-05,
+      "loss": 0.0216,
+      "step": 3747
+    },
+    {
+      "epoch": 220.47,
+      "learning_rate": 1.395252837977296e-05,
+      "loss": 0.0189,
+      "step": 3748
+    },
+    {
+      "epoch": 220.53,
+      "learning_rate": 1.3942208462332301e-05,
+      "loss": 0.0187,
+      "step": 3749
+    },
+    {
+      "epoch": 220.59,
+      "learning_rate": 1.3931888544891641e-05,
+      "loss": 0.0206,
+      "step": 3750
+    },
+    {
+      "epoch": 220.65,
+      "learning_rate": 1.3921568627450982e-05,
+      "loss": 0.0197,
+      "step": 3751
+    },
+    {
+      "epoch": 220.71,
+      "learning_rate": 1.3911248710010321e-05,
+      "loss": 0.0193,
+      "step": 3752
+    },
+    {
+      "epoch": 220.76,
+      "learning_rate": 1.390092879256966e-05,
+      "loss": 0.02,
+      "step": 3753
+    },
+    {
+      "epoch": 220.82,
+      "learning_rate": 1.3890608875128999e-05,
+      "loss": 0.0205,
+      "step": 3754
+    },
+    {
+      "epoch": 220.88,
+      "learning_rate": 1.388028895768834e-05,
+      "loss": 0.0195,
+      "step": 3755
+    },
+    {
+      "epoch": 220.94,
+      "learning_rate": 1.386996904024768e-05,
+      "loss": 0.0188,
+      "step": 3756
+    },
+    {
+      "epoch": 221.0,
+      "learning_rate": 1.3859649122807017e-05,
+      "loss": 0.0197,
+      "step": 3757
+    },
+    {
+      "epoch": 221.06,
+      "learning_rate": 1.3849329205366357e-05,
+      "loss": 0.017,
+      "step": 3758
+    },
+    {
+      "epoch": 221.12,
+      "learning_rate": 1.3839009287925696e-05,
+      "loss": 0.0199,
+      "step": 3759
+    },
+    {
+      "epoch": 221.18,
+      "learning_rate": 1.3828689370485037e-05,
+      "loss": 0.0169,
+      "step": 3760
+    },
+    {
+      "epoch": 221.18,
+      "eval_accuracy_NAT": 0.9933691231937067,
+      "eval_accuracy_REST": 0.42601626016260163,
+      "eval_accuracy_concrete": 0.9574436370696627,
+      "eval_accuracy_concrete tiles": 0.9448503417393354,
+      "eval_accuracy_dark bricks": 0.863268763279291,
+      "eval_accuracy_grass": 0.9002770554905812,
+      "eval_accuracy_red bricks": 0.9379393673110721,
+      "eval_accuracy_road": 0.9820436179758347,
+      "eval_accuracy_rough concrete": 0.6276106996228079,
+      "eval_accuracy_rough red sidewalk": 0.791455793771949,
+      "eval_accuracy_speedway bricks": 0.9892874499699426,
+      "eval_accuracy_steel": 0.9291683117843738,
+      "eval_accuracy_tiles": 0.2544457572177285,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983955075792885,
+      "eval_iou_REST": 0.40307692307692305,
+      "eval_iou_concrete": 0.8934649931235983,
+      "eval_iou_concrete tiles": 0.879822674801387,
+      "eval_iou_dark bricks": 0.8463436718570403,
+      "eval_iou_grass": 0.839607339961207,
+      "eval_iou_red bricks": 0.9067643623234576,
+      "eval_iou_road": 0.9605192111578233,
+      "eval_iou_rough concrete": 0.6208626350630442,
+      "eval_iou_rough red sidewalk": 0.6207394825155853,
+      "eval_iou_speedway bricks": 0.9801219409914531,
+      "eval_iou_steel": 0.8713122091361137,
+      "eval_iou_tiles": 0.248527888740938,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08260580897331238,
+      "eval_mean_accuracy": 0.8151673983529915,
+      "eval_mean_iou": 0.7734706468108813,
+      "eval_overall_accuracy": 0.9802340168063923,
+      "eval_runtime": 5.4419,
+      "eval_samples_per_second": 21.684,
+      "eval_steps_per_second": 1.47,
+      "step": 3760
+    },
+    {
+      "epoch": 221.24,
+      "learning_rate": 1.3818369453044377e-05,
+      "loss": 0.0219,
+      "step": 3761
+    },
+    {
+      "epoch": 221.29,
+      "learning_rate": 1.3808049535603715e-05,
+      "loss": 0.022,
+      "step": 3762
+    },
+    {
+      "epoch": 221.35,
+      "learning_rate": 1.3797729618163055e-05,
+      "loss": 0.0206,
+      "step": 3763
+    },
+    {
+      "epoch": 221.41,
+      "learning_rate": 1.3787409700722396e-05,
+      "loss": 0.0252,
+      "step": 3764
+    },
+    {
+      "epoch": 221.47,
+      "learning_rate": 1.3777089783281735e-05,
+      "loss": 0.021,
+      "step": 3765
+    },
+    {
+      "epoch": 221.53,
+      "learning_rate": 1.3766769865841072e-05,
+      "loss": 0.0203,
+      "step": 3766
+    },
+    {
+      "epoch": 221.59,
+      "learning_rate": 1.3756449948400412e-05,
+      "loss": 0.0223,
+      "step": 3767
+    },
+    {
+      "epoch": 221.65,
+      "learning_rate": 1.3746130030959753e-05,
+      "loss": 0.0227,
+      "step": 3768
+    },
+    {
+      "epoch": 221.71,
+      "learning_rate": 1.3735810113519094e-05,
+      "loss": 0.0162,
+      "step": 3769
+    },
+    {
+      "epoch": 221.76,
+      "learning_rate": 1.3725490196078432e-05,
+      "loss": 0.0221,
+      "step": 3770
+    },
+    {
+      "epoch": 221.82,
+      "learning_rate": 1.3715170278637771e-05,
+      "loss": 0.0195,
+      "step": 3771
+    },
+    {
+      "epoch": 221.88,
+      "learning_rate": 1.370485036119711e-05,
+      "loss": 0.0209,
+      "step": 3772
+    },
+    {
+      "epoch": 221.94,
+      "learning_rate": 1.369453044375645e-05,
+      "loss": 0.0202,
+      "step": 3773
+    },
+    {
+      "epoch": 222.0,
+      "learning_rate": 1.3684210526315791e-05,
+      "loss": 0.0182,
+      "step": 3774
+    },
+    {
+      "epoch": 222.06,
+      "learning_rate": 1.3673890608875128e-05,
+      "loss": 0.0182,
+      "step": 3775
+    },
+    {
+      "epoch": 222.12,
+      "learning_rate": 1.3663570691434469e-05,
+      "loss": 0.0193,
+      "step": 3776
+    },
+    {
+      "epoch": 222.18,
+      "learning_rate": 1.3653250773993808e-05,
+      "loss": 0.0202,
+      "step": 3777
+    },
+    {
+      "epoch": 222.24,
+      "learning_rate": 1.3642930856553148e-05,
+      "loss": 0.0198,
+      "step": 3778
+    },
+    {
+      "epoch": 222.29,
+      "learning_rate": 1.3632610939112489e-05,
+      "loss": 0.0197,
+      "step": 3779
+    },
+    {
+      "epoch": 222.35,
+      "learning_rate": 1.3622291021671826e-05,
+      "loss": 0.0179,
+      "step": 3780
+    },
+    {
+      "epoch": 222.35,
+      "eval_accuracy_NAT": 0.9942545699246028,
+      "eval_accuracy_REST": 0.44552845528455287,
+      "eval_accuracy_concrete": 0.9545633831670257,
+      "eval_accuracy_concrete tiles": 0.9557152957812868,
+      "eval_accuracy_dark bricks": 0.8706620982715387,
+      "eval_accuracy_grass": 0.8969236242794476,
+      "eval_accuracy_red bricks": 0.9342596660808435,
+      "eval_accuracy_road": 0.9812174106362705,
+      "eval_accuracy_rough concrete": 0.6283853437334271,
+      "eval_accuracy_rough red sidewalk": 0.7858458854405583,
+      "eval_accuracy_speedway bricks": 0.9895167365493776,
+      "eval_accuracy_steel": 0.9230334804360103,
+      "eval_accuracy_tiles": 0.2579902450846739,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838151810449068,
+      "eval_iou_REST": 0.42946708463949845,
+      "eval_iou_concrete": 0.8943529679701437,
+      "eval_iou_concrete tiles": 0.8719707558327061,
+      "eval_iou_dark bricks": 0.849221462977484,
+      "eval_iou_grass": 0.8404425761913586,
+      "eval_iou_red bricks": 0.9089986106658117,
+      "eval_iou_road": 0.9618654811697945,
+      "eval_iou_rough concrete": 0.622373701605288,
+      "eval_iou_rough red sidewalk": 0.6242368851773532,
+      "eval_iou_speedway bricks": 0.9800931835155137,
+      "eval_iou_steel": 0.8724284695289878,
+      "eval_iou_tiles": 0.2549013738812895,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08182456344366074,
+      "eval_mean_accuracy": 0.8167612457438166,
+      "eval_mean_iou": 0.7764744410923181,
+      "eval_overall_accuracy": 0.9803676281945181,
+      "eval_runtime": 5.5495,
+      "eval_samples_per_second": 21.263,
+      "eval_steps_per_second": 1.442,
+      "step": 3780
+    },
+    {
+      "epoch": 222.41,
+      "learning_rate": 1.3611971104231167e-05,
+      "loss": 0.0238,
+      "step": 3781
+    },
+    {
+      "epoch": 222.47,
+      "learning_rate": 1.3601651186790507e-05,
+      "loss": 0.0213,
+      "step": 3782
+    },
+    {
+      "epoch": 222.53,
+      "learning_rate": 1.3591331269349846e-05,
+      "loss": 0.0233,
+      "step": 3783
+    },
+    {
+      "epoch": 222.59,
+      "learning_rate": 1.3581011351909183e-05,
+      "loss": 0.0215,
+      "step": 3784
+    },
+    {
+      "epoch": 222.65,
+      "learning_rate": 1.3570691434468524e-05,
+      "loss": 0.0178,
+      "step": 3785
+    },
+    {
+      "epoch": 222.71,
+      "learning_rate": 1.3560371517027865e-05,
+      "loss": 0.0197,
+      "step": 3786
+    },
+    {
+      "epoch": 222.76,
+      "learning_rate": 1.3550051599587205e-05,
+      "loss": 0.0209,
+      "step": 3787
+    },
+    {
+      "epoch": 222.82,
+      "learning_rate": 1.3539731682146544e-05,
+      "loss": 0.0206,
+      "step": 3788
+    },
+    {
+      "epoch": 222.88,
+      "learning_rate": 1.3529411764705883e-05,
+      "loss": 0.0184,
+      "step": 3789
+    },
+    {
+      "epoch": 222.94,
+      "learning_rate": 1.3519091847265222e-05,
+      "loss": 0.0221,
+      "step": 3790
+    },
+    {
+      "epoch": 223.0,
+      "learning_rate": 1.3508771929824562e-05,
+      "loss": 0.0192,
+      "step": 3791
+    },
+    {
+      "epoch": 223.06,
+      "learning_rate": 1.3498452012383903e-05,
+      "loss": 0.0208,
+      "step": 3792
+    },
+    {
+      "epoch": 223.12,
+      "learning_rate": 1.348813209494324e-05,
+      "loss": 0.0203,
+      "step": 3793
+    },
+    {
+      "epoch": 223.18,
+      "learning_rate": 1.347781217750258e-05,
+      "loss": 0.0205,
+      "step": 3794
+    },
+    {
+      "epoch": 223.24,
+      "learning_rate": 1.346749226006192e-05,
+      "loss": 0.0214,
+      "step": 3795
+    },
+    {
+      "epoch": 223.29,
+      "learning_rate": 1.345717234262126e-05,
+      "loss": 0.0204,
+      "step": 3796
+    },
+    {
+      "epoch": 223.35,
+      "learning_rate": 1.34468524251806e-05,
+      "loss": 0.0189,
+      "step": 3797
+    },
+    {
+      "epoch": 223.41,
+      "learning_rate": 1.3436532507739938e-05,
+      "loss": 0.0212,
+      "step": 3798
+    },
+    {
+      "epoch": 223.47,
+      "learning_rate": 1.3426212590299278e-05,
+      "loss": 0.0192,
+      "step": 3799
+    },
+    {
+      "epoch": 223.53,
+      "learning_rate": 1.3415892672858619e-05,
+      "loss": 0.0174,
+      "step": 3800
+    },
+    {
+      "epoch": 223.53,
+      "eval_accuracy_NAT": 0.9941067438812444,
+      "eval_accuracy_REST": 0.2682926829268293,
+      "eval_accuracy_concrete": 0.9577551294967658,
+      "eval_accuracy_concrete tiles": 0.9290596276219656,
+      "eval_accuracy_dark bricks": 0.8486639358382941,
+      "eval_accuracy_grass": 0.8959802351943741,
+      "eval_accuracy_red bricks": 0.9114674868189807,
+      "eval_accuracy_road": 0.9812542853493559,
+      "eval_accuracy_rough concrete": 0.6218306627974186,
+      "eval_accuracy_rough red sidewalk": 0.7663691259522181,
+      "eval_accuracy_speedway bricks": 0.9890788985744475,
+      "eval_accuracy_steel": 0.9271217020596041,
+      "eval_accuracy_tiles": 0.23242146081371748,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838887719764048,
+      "eval_iou_REST": 0.26570048309178745,
+      "eval_iou_concrete": 0.8924004202204376,
+      "eval_iou_concrete tiles": 0.8757470064203674,
+      "eval_iou_dark bricks": 0.8344602383674699,
+      "eval_iou_grass": 0.8414065013053438,
+      "eval_iou_red bricks": 0.8947595428078499,
+      "eval_iou_road": 0.9607213923449908,
+      "eval_iou_rough concrete": 0.6179642207865268,
+      "eval_iou_rough red sidewalk": 0.6166557750024204,
+      "eval_iou_speedway bricks": 0.9796615106147952,
+      "eval_iou_steel": 0.8718077656944666,
+      "eval_iou_tiles": 0.23089656002648448,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08449593931436539,
+      "eval_mean_accuracy": 0.7941078444096319,
+      "eval_mean_iou": 0.7589284760507188,
+      "eval_overall_accuracy": 0.9801405567233845,
+      "eval_runtime": 5.2993,
+      "eval_samples_per_second": 22.267,
+      "eval_steps_per_second": 1.51,
+      "step": 3800
+    },
+    {
+      "epoch": 223.59,
+      "learning_rate": 1.3405572755417958e-05,
+      "loss": 0.0194,
+      "step": 3801
+    },
+    {
+      "epoch": 223.65,
+      "learning_rate": 1.3395252837977295e-05,
+      "loss": 0.0207,
+      "step": 3802
+    },
+    {
+      "epoch": 223.71,
+      "learning_rate": 1.3384932920536636e-05,
+      "loss": 0.021,
+      "step": 3803
+    },
+    {
+      "epoch": 223.76,
+      "learning_rate": 1.3374613003095976e-05,
+      "loss": 0.019,
+      "step": 3804
+    },
+    {
+      "epoch": 223.82,
+      "learning_rate": 1.3364293085655317e-05,
+      "loss": 0.02,
+      "step": 3805
+    },
+    {
+      "epoch": 223.88,
+      "learning_rate": 1.3353973168214656e-05,
+      "loss": 0.0217,
+      "step": 3806
+    },
+    {
+      "epoch": 223.94,
+      "learning_rate": 1.3343653250773994e-05,
+      "loss": 0.024,
+      "step": 3807
+    },
+    {
+      "epoch": 224.0,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 0.0194,
+      "step": 3808
+    },
+    {
+      "epoch": 224.06,
+      "learning_rate": 1.3323013415892674e-05,
+      "loss": 0.0195,
+      "step": 3809
+    },
+    {
+      "epoch": 224.12,
+      "learning_rate": 1.3312693498452014e-05,
+      "loss": 0.0247,
+      "step": 3810
+    },
+    {
+      "epoch": 224.18,
+      "learning_rate": 1.3302373581011352e-05,
+      "loss": 0.0195,
+      "step": 3811
+    },
+    {
+      "epoch": 224.24,
+      "learning_rate": 1.3292053663570692e-05,
+      "loss": 0.0236,
+      "step": 3812
+    },
+    {
+      "epoch": 224.29,
+      "learning_rate": 1.3281733746130031e-05,
+      "loss": 0.0216,
+      "step": 3813
+    },
+    {
+      "epoch": 224.35,
+      "learning_rate": 1.3271413828689372e-05,
+      "loss": 0.0195,
+      "step": 3814
+    },
+    {
+      "epoch": 224.41,
+      "learning_rate": 1.3261093911248712e-05,
+      "loss": 0.0173,
+      "step": 3815
+    },
+    {
+      "epoch": 224.47,
+      "learning_rate": 1.325077399380805e-05,
+      "loss": 0.0191,
+      "step": 3816
+    },
+    {
+      "epoch": 224.53,
+      "learning_rate": 1.324045407636739e-05,
+      "loss": 0.0188,
+      "step": 3817
+    },
+    {
+      "epoch": 224.59,
+      "learning_rate": 1.323013415892673e-05,
+      "loss": 0.0217,
+      "step": 3818
+    },
+    {
+      "epoch": 224.65,
+      "learning_rate": 1.321981424148607e-05,
+      "loss": 0.0197,
+      "step": 3819
+    },
+    {
+      "epoch": 224.71,
+      "learning_rate": 1.3209494324045407e-05,
+      "loss": 0.0177,
+      "step": 3820
+    },
+    {
+      "epoch": 224.71,
+      "eval_accuracy_NAT": 0.993947326809687,
+      "eval_accuracy_REST": 0.35121951219512193,
+      "eval_accuracy_concrete": 0.9565546415047034,
+      "eval_accuracy_concrete tiles": 0.936790007070469,
+      "eval_accuracy_dark bricks": 0.8628045862603603,
+      "eval_accuracy_grass": 0.8966936909317316,
+      "eval_accuracy_red bricks": 0.9328866432337434,
+      "eval_accuracy_road": 0.9826320101368052,
+      "eval_accuracy_rough concrete": 0.6265619506730465,
+      "eval_accuracy_rough red sidewalk": 0.7955917851690868,
+      "eval_accuracy_speedway bricks": 0.9894190584101563,
+      "eval_accuracy_steel": 0.9307326313053822,
+      "eval_accuracy_tiles": 0.256566390984277,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840227644170095,
+      "eval_iou_REST": 0.3444976076555024,
+      "eval_iou_concrete": 0.8947834888120066,
+      "eval_iou_concrete tiles": 0.879516739318036,
+      "eval_iou_dark bricks": 0.8452144889626432,
+      "eval_iou_grass": 0.8425556620498713,
+      "eval_iou_red bricks": 0.9044728434504792,
+      "eval_iou_road": 0.9617738842608624,
+      "eval_iou_rough concrete": 0.6203078266307201,
+      "eval_iou_rough red sidewalk": 0.6298602718358745,
+      "eval_iou_speedway bricks": 0.9799495305574687,
+      "eval_iou_steel": 0.8718168986721034,
+      "eval_iou_tiles": 0.253146017037812,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08152302354574203,
+      "eval_mean_accuracy": 0.808646171898813,
+      "eval_mean_iou": 0.7701475402815684,
+      "eval_overall_accuracy": 0.9804594718803794,
+      "eval_runtime": 5.4472,
+      "eval_samples_per_second": 21.663,
+      "eval_steps_per_second": 1.469,
+      "step": 3820
+    },
+    {
+      "epoch": 224.76,
+      "learning_rate": 1.3199174406604747e-05,
+      "loss": 0.0188,
+      "step": 3821
+    },
+    {
+      "epoch": 224.82,
+      "learning_rate": 1.3188854489164088e-05,
+      "loss": 0.0179,
+      "step": 3822
+    },
+    {
+      "epoch": 224.88,
+      "learning_rate": 1.3178534571723428e-05,
+      "loss": 0.018,
+      "step": 3823
+    },
+    {
+      "epoch": 224.94,
+      "learning_rate": 1.3168214654282765e-05,
+      "loss": 0.0172,
+      "step": 3824
+    },
+    {
+      "epoch": 225.0,
+      "learning_rate": 1.3157894736842106e-05,
+      "loss": 0.0193,
+      "step": 3825
+    },
+    {
+      "epoch": 225.06,
+      "learning_rate": 1.3147574819401445e-05,
+      "loss": 0.0221,
+      "step": 3826
+    },
+    {
+      "epoch": 225.12,
+      "learning_rate": 1.3137254901960785e-05,
+      "loss": 0.0209,
+      "step": 3827
+    },
+    {
+      "epoch": 225.18,
+      "learning_rate": 1.3126934984520126e-05,
+      "loss": 0.0204,
+      "step": 3828
+    },
+    {
+      "epoch": 225.24,
+      "learning_rate": 1.3116615067079463e-05,
+      "loss": 0.0233,
+      "step": 3829
+    },
+    {
+      "epoch": 225.29,
+      "learning_rate": 1.3106295149638804e-05,
+      "loss": 0.0186,
+      "step": 3830
+    },
+    {
+      "epoch": 225.35,
+      "learning_rate": 1.3095975232198143e-05,
+      "loss": 0.0161,
+      "step": 3831
+    },
+    {
+      "epoch": 225.41,
+      "learning_rate": 1.3085655314757483e-05,
+      "loss": 0.0181,
+      "step": 3832
+    },
+    {
+      "epoch": 225.47,
+      "learning_rate": 1.307533539731682e-05,
+      "loss": 0.0192,
+      "step": 3833
+    },
+    {
+      "epoch": 225.53,
+      "learning_rate": 1.3065015479876161e-05,
+      "loss": 0.019,
+      "step": 3834
+    },
+    {
+      "epoch": 225.59,
+      "learning_rate": 1.3054695562435501e-05,
+      "loss": 0.019,
+      "step": 3835
+    },
+    {
+      "epoch": 225.65,
+      "learning_rate": 1.3044375644994842e-05,
+      "loss": 0.0209,
+      "step": 3836
+    },
+    {
+      "epoch": 225.71,
+      "learning_rate": 1.3034055727554181e-05,
+      "loss": 0.0214,
+      "step": 3837
+    },
+    {
+      "epoch": 225.76,
+      "learning_rate": 1.3023735810113518e-05,
+      "loss": 0.0192,
+      "step": 3838
+    },
+    {
+      "epoch": 225.82,
+      "learning_rate": 1.3013415892672859e-05,
+      "loss": 0.0189,
+      "step": 3839
+    },
+    {
+      "epoch": 225.88,
+      "learning_rate": 1.30030959752322e-05,
+      "loss": 0.0207,
+      "step": 3840
+    },
+    {
+      "epoch": 225.88,
+      "eval_accuracy_NAT": 0.9934408258332642,
+      "eval_accuracy_REST": 0.43414634146341463,
+      "eval_accuracy_concrete": 0.9566199975845004,
+      "eval_accuracy_concrete tiles": 0.9326891350459581,
+      "eval_accuracy_dark bricks": 0.8624983251344678,
+      "eval_accuracy_grass": 0.8967911523918746,
+      "eval_accuracy_red bricks": 0.937390158172232,
+      "eval_accuracy_road": 0.9824941307748339,
+      "eval_accuracy_rough concrete": 0.6269909843343126,
+      "eval_accuracy_rough red sidewalk": 0.7652486482828116,
+      "eval_accuracy_speedway bricks": 0.9907114944206933,
+      "eval_accuracy_steel": 0.9247277434316072,
+      "eval_accuracy_tiles": 0.25181011239358964,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840242071965829,
+      "eval_iou_REST": 0.4204724409448819,
+      "eval_iou_concrete": 0.8920270031391042,
+      "eval_iou_concrete tiles": 0.8748922247032034,
+      "eval_iou_dark bricks": 0.8435423179886553,
+      "eval_iou_grass": 0.8427161710126423,
+      "eval_iou_red bricks": 0.8982212398694874,
+      "eval_iou_road": 0.960068598560036,
+      "eval_iou_rough concrete": 0.6197600381677141,
+      "eval_iou_rough red sidewalk": 0.6150396480030945,
+      "eval_iou_speedway bricks": 0.9795909272827767,
+      "eval_iou_steel": 0.8708478748414455,
+      "eval_iou_tiles": 0.24677136834604993,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08247293531894684,
+      "eval_mean_accuracy": 0.8119660807125815,
+      "eval_mean_iou": 0.7729210815427442,
+      "eval_overall_accuracy": 0.9801858158434852,
+      "eval_runtime": 5.5857,
+      "eval_samples_per_second": 21.126,
+      "eval_steps_per_second": 1.432,
+      "step": 3840
+    },
+    {
+      "epoch": 225.94,
+      "learning_rate": 1.299277605779154e-05,
+      "loss": 0.0176,
+      "step": 3841
+    },
+    {
+      "epoch": 226.0,
+      "learning_rate": 1.2982456140350877e-05,
+      "loss": 0.0218,
+      "step": 3842
+    },
+    {
+      "epoch": 226.06,
+      "learning_rate": 1.2972136222910216e-05,
+      "loss": 0.0214,
+      "step": 3843
+    },
+    {
+      "epoch": 226.12,
+      "learning_rate": 1.2961816305469556e-05,
+      "loss": 0.0195,
+      "step": 3844
+    },
+    {
+      "epoch": 226.18,
+      "learning_rate": 1.2951496388028897e-05,
+      "loss": 0.0216,
+      "step": 3845
+    },
+    {
+      "epoch": 226.24,
+      "learning_rate": 1.2941176470588238e-05,
+      "loss": 0.0192,
+      "step": 3846
+    },
+    {
+      "epoch": 226.29,
+      "learning_rate": 1.2930856553147575e-05,
+      "loss": 0.021,
+      "step": 3847
+    },
+    {
+      "epoch": 226.35,
+      "learning_rate": 1.2920536635706915e-05,
+      "loss": 0.0228,
+      "step": 3848
+    },
+    {
+      "epoch": 226.41,
+      "learning_rate": 1.2910216718266254e-05,
+      "loss": 0.0214,
+      "step": 3849
+    },
+    {
+      "epoch": 226.47,
+      "learning_rate": 1.2899896800825595e-05,
+      "loss": 0.0177,
+      "step": 3850
+    },
+    {
+      "epoch": 226.53,
+      "learning_rate": 1.2889576883384932e-05,
+      "loss": 0.0217,
+      "step": 3851
+    },
+    {
+      "epoch": 226.59,
+      "learning_rate": 1.2879256965944272e-05,
+      "loss": 0.0188,
+      "step": 3852
+    },
+    {
+      "epoch": 226.65,
+      "learning_rate": 1.2868937048503613e-05,
+      "loss": 0.0215,
+      "step": 3853
+    },
+    {
+      "epoch": 226.71,
+      "learning_rate": 1.2858617131062952e-05,
+      "loss": 0.0185,
+      "step": 3854
+    },
+    {
+      "epoch": 226.76,
+      "learning_rate": 1.2848297213622292e-05,
+      "loss": 0.0206,
+      "step": 3855
+    },
+    {
+      "epoch": 226.82,
+      "learning_rate": 1.283797729618163e-05,
+      "loss": 0.0179,
+      "step": 3856
+    },
+    {
+      "epoch": 226.88,
+      "learning_rate": 1.282765737874097e-05,
+      "loss": 0.0167,
+      "step": 3857
+    },
+    {
+      "epoch": 226.94,
+      "learning_rate": 1.281733746130031e-05,
+      "loss": 0.019,
+      "step": 3858
+    },
+    {
+      "epoch": 227.0,
+      "learning_rate": 1.2807017543859651e-05,
+      "loss": 0.0182,
+      "step": 3859
+    },
+    {
+      "epoch": 227.06,
+      "learning_rate": 1.2796697626418989e-05,
+      "loss": 0.0186,
+      "step": 3860
+    },
+    {
+      "epoch": 227.06,
+      "eval_accuracy_NAT": 0.9943601291488535,
+      "eval_accuracy_REST": 0.359349593495935,
+      "eval_accuracy_concrete": 0.95683058939718,
+      "eval_accuracy_concrete tiles": 0.9338204100872024,
+      "eval_accuracy_dark bricks": 0.8585695691288785,
+      "eval_accuracy_grass": 0.8876486996938007,
+      "eval_accuracy_red bricks": 0.9340399824253075,
+      "eval_accuracy_road": 0.9812093943942954,
+      "eval_accuracy_rough concrete": 0.6258051829649801,
+      "eval_accuracy_rough red sidewalk": 0.7708435166454853,
+      "eval_accuracy_speedway bricks": 0.9902565199301104,
+      "eval_accuracy_steel": 0.9223487782814476,
+      "eval_accuracy_tiles": 0.24532703202156989,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838704007571296,
+      "eval_iou_REST": 0.34913112164297,
+      "eval_iou_concrete": 0.8930700090894952,
+      "eval_iou_concrete tiles": 0.8787314260368153,
+      "eval_iou_dark bricks": 0.8428833839923706,
+      "eval_iou_grass": 0.8408615027177421,
+      "eval_iou_red bricks": 0.8995081186862008,
+      "eval_iou_road": 0.960921292002755,
+      "eval_iou_rough concrete": 0.6192152353998998,
+      "eval_iou_rough red sidewalk": 0.6172345896081843,
+      "eval_iou_speedway bricks": 0.9797667637791925,
+      "eval_iou_steel": 0.8715672152825332,
+      "eval_iou_tiles": 0.2400972485768501,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08318942785263062,
+      "eval_mean_accuracy": 0.804646876739619,
+      "eval_mean_iou": 0.7674506390440107,
+      "eval_overall_accuracy": 0.9802304607326702,
+      "eval_runtime": 5.9202,
+      "eval_samples_per_second": 19.932,
+      "eval_steps_per_second": 1.351,
+      "step": 3860
+    },
+    {
+      "epoch": 227.12,
+      "learning_rate": 1.2786377708978327e-05,
+      "loss": 0.0191,
+      "step": 3861
+    },
+    {
+      "epoch": 227.18,
+      "learning_rate": 1.2776057791537668e-05,
+      "loss": 0.0204,
+      "step": 3862
+    },
+    {
+      "epoch": 227.24,
+      "learning_rate": 1.2765737874097009e-05,
+      "loss": 0.0191,
+      "step": 3863
+    },
+    {
+      "epoch": 227.29,
+      "learning_rate": 1.2755417956656349e-05,
+      "loss": 0.0235,
+      "step": 3864
+    },
+    {
+      "epoch": 227.35,
+      "learning_rate": 1.2745098039215686e-05,
+      "loss": 0.0213,
+      "step": 3865
+    },
+    {
+      "epoch": 227.41,
+      "learning_rate": 1.2734778121775027e-05,
+      "loss": 0.0192,
+      "step": 3866
+    },
+    {
+      "epoch": 227.47,
+      "learning_rate": 1.2724458204334366e-05,
+      "loss": 0.0179,
+      "step": 3867
+    },
+    {
+      "epoch": 227.53,
+      "learning_rate": 1.2714138286893706e-05,
+      "loss": 0.0197,
+      "step": 3868
+    },
+    {
+      "epoch": 227.59,
+      "learning_rate": 1.2703818369453043e-05,
+      "loss": 0.0195,
+      "step": 3869
+    },
+    {
+      "epoch": 227.65,
+      "learning_rate": 1.2693498452012384e-05,
+      "loss": 0.0204,
+      "step": 3870
+    },
+    {
+      "epoch": 227.71,
+      "learning_rate": 1.2683178534571725e-05,
+      "loss": 0.0174,
+      "step": 3871
+    },
+    {
+      "epoch": 227.76,
+      "learning_rate": 1.2672858617131063e-05,
+      "loss": 0.0211,
+      "step": 3872
+    },
+    {
+      "epoch": 227.82,
+      "learning_rate": 1.2662538699690404e-05,
+      "loss": 0.0194,
+      "step": 3873
+    },
+    {
+      "epoch": 227.88,
+      "learning_rate": 1.2652218782249741e-05,
+      "loss": 0.0213,
+      "step": 3874
+    },
+    {
+      "epoch": 227.94,
+      "learning_rate": 1.2641898864809082e-05,
+      "loss": 0.0201,
+      "step": 3875
+    },
+    {
+      "epoch": 228.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0176,
+      "step": 3876
+    },
+    {
+      "epoch": 228.06,
+      "learning_rate": 1.2621259029927763e-05,
+      "loss": 0.021,
+      "step": 3877
+    },
+    {
+      "epoch": 228.12,
+      "learning_rate": 1.26109391124871e-05,
+      "loss": 0.0205,
+      "step": 3878
+    },
+    {
+      "epoch": 228.18,
+      "learning_rate": 1.2600619195046439e-05,
+      "loss": 0.0185,
+      "step": 3879
+    },
+    {
+      "epoch": 228.24,
+      "learning_rate": 1.259029927760578e-05,
+      "loss": 0.0215,
+      "step": 3880
+    },
+    {
+      "epoch": 228.24,
+      "eval_accuracy_NAT": 0.9937780438862205,
+      "eval_accuracy_REST": 0.43414634146341463,
+      "eval_accuracy_concrete": 0.9570866629262096,
+      "eval_accuracy_concrete tiles": 0.9345274569879802,
+      "eval_accuracy_dark bricks": 0.8621202840571943,
+      "eval_accuracy_grass": 0.8988775846211493,
+      "eval_accuracy_red bricks": 0.9379393673110721,
+      "eval_accuracy_road": 0.9819062730299952,
+      "eval_accuracy_rough concrete": 0.6253046436935031,
+      "eval_accuracy_rough red sidewalk": 0.7557057881319607,
+      "eval_accuracy_speedway bricks": 0.9899655418943253,
+      "eval_accuracy_steel": 0.9258447575304742,
+      "eval_accuracy_tiles": 0.25453664152200917,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9839423490039392,
+      "eval_iou_REST": 0.41915227629513346,
+      "eval_iou_concrete": 0.8929739028231052,
+      "eval_iou_concrete tiles": 0.8791432942376338,
+      "eval_iou_dark bricks": 0.8442117101286286,
+      "eval_iou_grass": 0.8447474007334502,
+      "eval_iou_red bricks": 0.9060427608891718,
+      "eval_iou_road": 0.9598380126035473,
+      "eval_iou_rough concrete": 0.6205383541878563,
+      "eval_iou_rough red sidewalk": 0.60983572733286,
+      "eval_iou_speedway bricks": 0.9800127435300121,
+      "eval_iou_steel": 0.8715183278093899,
+      "eval_iou_tiles": 0.24997024872069498,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0826205387711525,
+      "eval_mean_accuracy": 0.8116722605427314,
+      "eval_mean_iou": 0.7739943929458017,
+      "eval_overall_accuracy": 0.9802899764756025,
+      "eval_runtime": 5.8458,
+      "eval_samples_per_second": 20.185,
+      "eval_steps_per_second": 1.369,
+      "step": 3880
+    },
+    {
+      "epoch": 228.29,
+      "learning_rate": 1.257997936016512e-05,
+      "loss": 0.0211,
+      "step": 3881
+    },
+    {
+      "epoch": 228.35,
+      "learning_rate": 1.256965944272446e-05,
+      "loss": 0.0211,
+      "step": 3882
+    },
+    {
+      "epoch": 228.41,
+      "learning_rate": 1.2559339525283798e-05,
+      "loss": 0.0181,
+      "step": 3883
+    },
+    {
+      "epoch": 228.47,
+      "learning_rate": 1.2549019607843138e-05,
+      "loss": 0.0209,
+      "step": 3884
+    },
+    {
+      "epoch": 228.53,
+      "learning_rate": 1.2538699690402477e-05,
+      "loss": 0.0209,
+      "step": 3885
+    },
+    {
+      "epoch": 228.59,
+      "learning_rate": 1.2528379772961818e-05,
+      "loss": 0.018,
+      "step": 3886
+    },
+    {
+      "epoch": 228.65,
+      "learning_rate": 1.2518059855521155e-05,
+      "loss": 0.0203,
+      "step": 3887
+    },
+    {
+      "epoch": 228.71,
+      "learning_rate": 1.2507739938080496e-05,
+      "loss": 0.0209,
+      "step": 3888
+    },
+    {
+      "epoch": 228.76,
+      "learning_rate": 1.2497420020639836e-05,
+      "loss": 0.0204,
+      "step": 3889
+    },
+    {
+      "epoch": 228.82,
+      "learning_rate": 1.2487100103199175e-05,
+      "loss": 0.0164,
+      "step": 3890
+    },
+    {
+      "epoch": 228.88,
+      "learning_rate": 1.2476780185758514e-05,
+      "loss": 0.0188,
+      "step": 3891
+    },
+    {
+      "epoch": 228.94,
+      "learning_rate": 1.2466460268317854e-05,
+      "loss": 0.0212,
+      "step": 3892
+    },
+    {
+      "epoch": 229.0,
+      "learning_rate": 1.2456140350877193e-05,
+      "loss": 0.0192,
+      "step": 3893
+    },
+    {
+      "epoch": 229.06,
+      "learning_rate": 1.2445820433436534e-05,
+      "loss": 0.0178,
+      "step": 3894
+    },
+    {
+      "epoch": 229.12,
+      "learning_rate": 1.2435500515995873e-05,
+      "loss": 0.017,
+      "step": 3895
+    },
+    {
+      "epoch": 229.18,
+      "learning_rate": 1.2425180598555212e-05,
+      "loss": 0.0186,
+      "step": 3896
+    },
+    {
+      "epoch": 229.24,
+      "learning_rate": 1.241486068111455e-05,
+      "loss": 0.0189,
+      "step": 3897
+    },
+    {
+      "epoch": 229.29,
+      "learning_rate": 1.2404540763673891e-05,
+      "loss": 0.0219,
+      "step": 3898
+    },
+    {
+      "epoch": 229.35,
+      "learning_rate": 1.239422084623323e-05,
+      "loss": 0.0237,
+      "step": 3899
+    },
+    {
+      "epoch": 229.41,
+      "learning_rate": 1.238390092879257e-05,
+      "loss": 0.0191,
+      "step": 3900
+    },
+    {
+      "epoch": 229.41,
+      "eval_accuracy_NAT": 0.9934891846811462,
+      "eval_accuracy_REST": 0.42276422764227645,
+      "eval_accuracy_concrete": 0.9609048339038295,
+      "eval_accuracy_concrete tiles": 0.9298373792128212,
+      "eval_accuracy_dark bricks": 0.8530042302318014,
+      "eval_accuracy_grass": 0.8970986764165976,
+      "eval_accuracy_red bricks": 0.9219573813708261,
+      "eval_accuracy_road": 0.9810346403192388,
+      "eval_accuracy_rough concrete": 0.6242141831377853,
+      "eval_accuracy_rough red sidewalk": 0.7613908963069357,
+      "eval_accuracy_speedway bricks": 0.9892141056829484,
+      "eval_accuracy_steel": 0.9238556228040802,
+      "eval_accuracy_tiles": 0.24496349480444726,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840060565190746,
+      "eval_iou_REST": 0.4107424960505529,
+      "eval_iou_concrete": 0.8909856904808988,
+      "eval_iou_concrete tiles": 0.87482815202448,
+      "eval_iou_dark bricks": 0.8368496662065857,
+      "eval_iou_grass": 0.8431423466751566,
+      "eval_iou_red bricks": 0.8969331053643941,
+      "eval_iou_road": 0.9596276522614631,
+      "eval_iou_rough concrete": 0.6181402978733448,
+      "eval_iou_rough red sidewalk": 0.6113553202023984,
+      "eval_iou_speedway bricks": 0.9797664943017047,
+      "eval_iou_steel": 0.871166439274513,
+      "eval_iou_tiles": 0.2416327994262491,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08389351516962051,
+      "eval_mean_accuracy": 0.8079791428088257,
+      "eval_mean_iou": 0.7707058858969856,
+      "eval_overall_accuracy": 0.9800893815897279,
+      "eval_runtime": 5.6728,
+      "eval_samples_per_second": 20.801,
+      "eval_steps_per_second": 1.41,
+      "step": 3900
+    },
+    {
+      "epoch": 229.47,
+      "learning_rate": 1.2373581011351911e-05,
+      "loss": 0.0169,
+      "step": 3901
+    },
+    {
+      "epoch": 229.53,
+      "learning_rate": 1.236326109391125e-05,
+      "loss": 0.021,
+      "step": 3902
+    },
+    {
+      "epoch": 229.59,
+      "learning_rate": 1.2352941176470589e-05,
+      "loss": 0.017,
+      "step": 3903
+    },
+    {
+      "epoch": 229.65,
+      "learning_rate": 1.2342621259029928e-05,
+      "loss": 0.0181,
+      "step": 3904
+    },
+    {
+      "epoch": 229.71,
+      "learning_rate": 1.2332301341589268e-05,
+      "loss": 0.0192,
+      "step": 3905
+    },
+    {
+      "epoch": 229.76,
+      "learning_rate": 1.2321981424148607e-05,
+      "loss": 0.021,
+      "step": 3906
+    },
+    {
+      "epoch": 229.82,
+      "learning_rate": 1.2311661506707948e-05,
+      "loss": 0.0258,
+      "step": 3907
+    },
+    {
+      "epoch": 229.88,
+      "learning_rate": 1.2301341589267287e-05,
+      "loss": 0.0159,
+      "step": 3908
+    },
+    {
+      "epoch": 229.94,
+      "learning_rate": 1.2291021671826625e-05,
+      "loss": 0.0206,
+      "step": 3909
+    },
+    {
+      "epoch": 230.0,
+      "learning_rate": 1.2280701754385964e-05,
+      "loss": 0.0229,
+      "step": 3910
+    },
+    {
+      "epoch": 230.06,
+      "learning_rate": 1.2270381836945305e-05,
+      "loss": 0.0196,
+      "step": 3911
+    },
+    {
+      "epoch": 230.12,
+      "learning_rate": 1.2260061919504645e-05,
+      "loss": 0.0167,
+      "step": 3912
+    },
+    {
+      "epoch": 230.18,
+      "learning_rate": 1.2249742002063984e-05,
+      "loss": 0.0183,
+      "step": 3913
+    },
+    {
+      "epoch": 230.24,
+      "learning_rate": 1.2239422084623323e-05,
+      "loss": 0.0177,
+      "step": 3914
+    },
+    {
+      "epoch": 230.29,
+      "learning_rate": 1.2229102167182662e-05,
+      "loss": 0.022,
+      "step": 3915
+    },
+    {
+      "epoch": 230.35,
+      "learning_rate": 1.2218782249742003e-05,
+      "loss": 0.0219,
+      "step": 3916
+    },
+    {
+      "epoch": 230.41,
+      "learning_rate": 1.2208462332301342e-05,
+      "loss": 0.0182,
+      "step": 3917
+    },
+    {
+      "epoch": 230.47,
+      "learning_rate": 1.2198142414860682e-05,
+      "loss": 0.0206,
+      "step": 3918
+    },
+    {
+      "epoch": 230.53,
+      "learning_rate": 1.2187822497420021e-05,
+      "loss": 0.0211,
+      "step": 3919
+    },
+    {
+      "epoch": 230.59,
+      "learning_rate": 1.2177502579979362e-05,
+      "loss": 0.0194,
+      "step": 3920
+    },
+    {
+      "epoch": 230.59,
+      "eval_accuracy_NAT": 0.9935296185004455,
+      "eval_accuracy_REST": 0.4113821138211382,
+      "eval_accuracy_concrete": 0.9572085845019714,
+      "eval_accuracy_concrete tiles": 0.9462880037709168,
+      "eval_accuracy_dark bricks": 0.8608186742721513,
+      "eval_accuracy_grass": 0.9016945046876124,
+      "eval_accuracy_red bricks": 0.9328866432337434,
+      "eval_accuracy_road": 0.9809347045026162,
+      "eval_accuracy_rough concrete": 0.6252152616807394,
+      "eval_accuracy_rough red sidewalk": 0.781529414418818,
+      "eval_accuracy_speedway bricks": 0.9901106881819048,
+      "eval_accuracy_steel": 0.9262745705618056,
+      "eval_accuracy_tiles": 0.24926535187373142,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840843588255395,
+      "eval_iou_REST": 0.40031645569620256,
+      "eval_iou_concrete": 0.8930179172310856,
+      "eval_iou_concrete tiles": 0.8789431054486548,
+      "eval_iou_dark bricks": 0.842084813758947,
+      "eval_iou_grass": 0.8442655194043497,
+      "eval_iou_red bricks": 0.9048100996111437,
+      "eval_iou_road": 0.9609952953520005,
+      "eval_iou_rough concrete": 0.618894256575416,
+      "eval_iou_rough red sidewalk": 0.6223881759002521,
+      "eval_iou_speedway bricks": 0.9796699822625258,
+      "eval_iou_steel": 0.8720849802371542,
+      "eval_iou_tiles": 0.24654660953465377,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08252972364425659,
+      "eval_mean_accuracy": 0.8120875487698151,
+      "eval_mean_iou": 0.7729308899875327,
+      "eval_overall_accuracy": 0.9803305480439785,
+      "eval_runtime": 5.8346,
+      "eval_samples_per_second": 20.224,
+      "eval_steps_per_second": 1.371,
+      "step": 3920
+    },
+    {
+      "epoch": 230.65,
+      "learning_rate": 1.21671826625387e-05,
+      "loss": 0.0199,
+      "step": 3921
+    },
+    {
+      "epoch": 230.71,
+      "learning_rate": 1.215686274509804e-05,
+      "loss": 0.0218,
+      "step": 3922
+    },
+    {
+      "epoch": 230.76,
+      "learning_rate": 1.214654282765738e-05,
+      "loss": 0.0186,
+      "step": 3923
+    },
+    {
+      "epoch": 230.82,
+      "learning_rate": 1.2136222910216719e-05,
+      "loss": 0.0201,
+      "step": 3924
+    },
+    {
+      "epoch": 230.88,
+      "learning_rate": 1.212590299277606e-05,
+      "loss": 0.0193,
+      "step": 3925
+    },
+    {
+      "epoch": 230.94,
+      "learning_rate": 1.2115583075335398e-05,
+      "loss": 0.019,
+      "step": 3926
+    },
+    {
+      "epoch": 231.0,
+      "learning_rate": 1.2105263157894737e-05,
+      "loss": 0.0173,
+      "step": 3927
+    },
+    {
+      "epoch": 231.06,
+      "learning_rate": 1.2094943240454076e-05,
+      "loss": 0.0182,
+      "step": 3928
+    },
+    {
+      "epoch": 231.12,
+      "learning_rate": 1.2084623323013416e-05,
+      "loss": 0.0199,
+      "step": 3929
+    },
+    {
+      "epoch": 231.18,
+      "learning_rate": 1.2074303405572757e-05,
+      "loss": 0.0206,
+      "step": 3930
+    },
+    {
+      "epoch": 231.24,
+      "learning_rate": 1.2063983488132096e-05,
+      "loss": 0.0225,
+      "step": 3931
+    },
+    {
+      "epoch": 231.29,
+      "learning_rate": 1.2053663570691435e-05,
+      "loss": 0.0212,
+      "step": 3932
+    },
+    {
+      "epoch": 231.35,
+      "learning_rate": 1.2043343653250774e-05,
+      "loss": 0.0229,
+      "step": 3933
+    },
+    {
+      "epoch": 231.41,
+      "learning_rate": 1.2033023735810114e-05,
+      "loss": 0.0169,
+      "step": 3934
+    },
+    {
+      "epoch": 231.47,
+      "learning_rate": 1.2022703818369453e-05,
+      "loss": 0.0194,
+      "step": 3935
+    },
+    {
+      "epoch": 231.53,
+      "learning_rate": 1.2012383900928794e-05,
+      "loss": 0.0182,
+      "step": 3936
+    },
+    {
+      "epoch": 231.59,
+      "learning_rate": 1.2002063983488133e-05,
+      "loss": 0.0198,
+      "step": 3937
+    },
+    {
+      "epoch": 231.65,
+      "learning_rate": 1.1991744066047471e-05,
+      "loss": 0.0182,
+      "step": 3938
+    },
+    {
+      "epoch": 231.71,
+      "learning_rate": 1.1981424148606812e-05,
+      "loss": 0.0206,
+      "step": 3939
+    },
+    {
+      "epoch": 231.76,
+      "learning_rate": 1.197110423116615e-05,
+      "loss": 0.0181,
+      "step": 3940
+    },
+    {
+      "epoch": 231.76,
+      "eval_accuracy_NAT": 0.9935437972930798,
+      "eval_accuracy_REST": 0.3983739837398374,
+      "eval_accuracy_concrete": 0.9567182228038447,
+      "eval_accuracy_concrete tiles": 0.9416686306858355,
+      "eval_accuracy_dark bricks": 0.8637712229389584,
+      "eval_accuracy_grass": 0.9067738553482686,
+      "eval_accuracy_red bricks": 0.9287675746924429,
+      "eval_accuracy_road": 0.9793084762139396,
+      "eval_accuracy_rough concrete": 0.6238626138875812,
+      "eval_accuracy_rough red sidewalk": 0.7886433196218952,
+      "eval_accuracy_speedway bricks": 0.9904936893067456,
+      "eval_accuracy_steel": 0.9218464967739034,
+      "eval_accuracy_tiles": 0.24802326638189584,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840796953439478,
+      "eval_iou_REST": 0.39074960127591707,
+      "eval_iou_concrete": 0.8936049141287628,
+      "eval_iou_concrete tiles": 0.881114100471927,
+      "eval_iou_dark bricks": 0.84539655479266,
+      "eval_iou_grass": 0.8442946693273776,
+      "eval_iou_red bricks": 0.9052028690718339,
+      "eval_iou_road": 0.9608589870643323,
+      "eval_iou_rough concrete": 0.6184051978735972,
+      "eval_iou_rough red sidewalk": 0.6327904857872094,
+      "eval_iou_speedway bricks": 0.9799560614840829,
+      "eval_iou_steel": 0.8720645457167308,
+      "eval_iou_tiles": 0.24386393423090671,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08125228434801102,
+      "eval_mean_accuracy": 0.8109073192067868,
+      "eval_mean_iou": 0.7732601243514835,
+      "eval_overall_accuracy": 0.9804205490370929,
+      "eval_runtime": 5.759,
+      "eval_samples_per_second": 20.49,
+      "eval_steps_per_second": 1.389,
+      "step": 3940
+    },
+    {
+      "epoch": 231.82,
+      "learning_rate": 1.1960784313725491e-05,
+      "loss": 0.0195,
+      "step": 3941
+    },
+    {
+      "epoch": 231.88,
+      "learning_rate": 1.195046439628483e-05,
+      "loss": 0.0187,
+      "step": 3942
+    },
+    {
+      "epoch": 231.94,
+      "learning_rate": 1.194014447884417e-05,
+      "loss": 0.02,
+      "step": 3943
+    },
+    {
+      "epoch": 232.0,
+      "learning_rate": 1.192982456140351e-05,
+      "loss": 0.0171,
+      "step": 3944
+    },
+    {
+      "epoch": 232.06,
+      "learning_rate": 1.1919504643962849e-05,
+      "loss": 0.0205,
+      "step": 3945
+    },
+    {
+      "epoch": 232.12,
+      "learning_rate": 1.1909184726522187e-05,
+      "loss": 0.0197,
+      "step": 3946
+    },
+    {
+      "epoch": 232.18,
+      "learning_rate": 1.1898864809081528e-05,
+      "loss": 0.0212,
+      "step": 3947
+    },
+    {
+      "epoch": 232.24,
+      "learning_rate": 1.1888544891640867e-05,
+      "loss": 0.0184,
+      "step": 3948
+    },
+    {
+      "epoch": 232.29,
+      "learning_rate": 1.1878224974200207e-05,
+      "loss": 0.0203,
+      "step": 3949
+    },
+    {
+      "epoch": 232.35,
+      "learning_rate": 1.1867905056759546e-05,
+      "loss": 0.0187,
+      "step": 3950
+    },
+    {
+      "epoch": 232.41,
+      "learning_rate": 1.1857585139318885e-05,
+      "loss": 0.0166,
+      "step": 3951
+    },
+    {
+      "epoch": 232.47,
+      "learning_rate": 1.1847265221878226e-05,
+      "loss": 0.0208,
+      "step": 3952
+    },
+    {
+      "epoch": 232.53,
+      "learning_rate": 1.1836945304437565e-05,
+      "loss": 0.019,
+      "step": 3953
+    },
+    {
+      "epoch": 232.59,
+      "learning_rate": 1.1826625386996905e-05,
+      "loss": 0.0205,
+      "step": 3954
+    },
+    {
+      "epoch": 232.65,
+      "learning_rate": 1.1816305469556244e-05,
+      "loss": 0.0158,
+      "step": 3955
+    },
+    {
+      "epoch": 232.71,
+      "learning_rate": 1.1805985552115583e-05,
+      "loss": 0.021,
+      "step": 3956
+    },
+    {
+      "epoch": 232.76,
+      "learning_rate": 1.1795665634674922e-05,
+      "loss": 0.0181,
+      "step": 3957
+    },
+    {
+      "epoch": 232.82,
+      "learning_rate": 1.1785345717234262e-05,
+      "loss": 0.0202,
+      "step": 3958
+    },
+    {
+      "epoch": 232.88,
+      "learning_rate": 1.1775025799793603e-05,
+      "loss": 0.0181,
+      "step": 3959
+    },
+    {
+      "epoch": 232.94,
+      "learning_rate": 1.1764705882352942e-05,
+      "loss": 0.0219,
+      "step": 3960
+    },
+    {
+      "epoch": 232.94,
+      "eval_accuracy_NAT": 0.9933269102863582,
+      "eval_accuracy_REST": 0.4796747967479675,
+      "eval_accuracy_concrete": 0.9603422365385585,
+      "eval_accuracy_concrete tiles": 0.9400424228140467,
+      "eval_accuracy_dark bricks": 0.8653647378596175,
+      "eval_accuracy_grass": 0.9075175903742142,
+      "eval_accuracy_red bricks": 0.9240992970123023,
+      "eval_accuracy_road": 0.9802244654636247,
+      "eval_accuracy_rough concrete": 0.6246372579982005,
+      "eval_accuracy_rough red sidewalk": 0.7859436452372179,
+      "eval_accuracy_speedway bricks": 0.9883819564828467,
+      "eval_accuracy_steel": 0.9301004063232494,
+      "eval_accuracy_tiles": 0.2488412251204217,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841071803786289,
+      "eval_iou_REST": 0.45314900153609833,
+      "eval_iou_concrete": 0.892382722393899,
+      "eval_iou_concrete tiles": 0.877656999515909,
+      "eval_iou_dark bricks": 0.848167760272784,
+      "eval_iou_grass": 0.8453237885637278,
+      "eval_iou_red bricks": 0.9004602376110457,
+      "eval_iou_road": 0.9611675516598273,
+      "eval_iou_rough concrete": 0.6176371810206162,
+      "eval_iou_rough red sidewalk": 0.6328390382135137,
+      "eval_iou_speedway bricks": 0.9800221570123376,
+      "eval_iou_steel": 0.8721535656876667,
+      "eval_iou_tiles": 0.24446428571428572,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08162851631641388,
+      "eval_mean_accuracy": 0.8175766883275869,
+      "eval_mean_iou": 0.7776562668907954,
+      "eval_overall_accuracy": 0.9803871219441042,
+      "eval_runtime": 6.0634,
+      "eval_samples_per_second": 19.461,
+      "eval_steps_per_second": 1.319,
+      "step": 3960
+    },
+    {
+      "epoch": 233.0,
+      "learning_rate": 1.1754385964912282e-05,
+      "loss": 0.0207,
+      "step": 3961
+    },
+    {
+      "epoch": 233.06,
+      "learning_rate": 1.1744066047471621e-05,
+      "loss": 0.0178,
+      "step": 3962
+    },
+    {
+      "epoch": 233.12,
+      "learning_rate": 1.173374613003096e-05,
+      "loss": 0.0215,
+      "step": 3963
+    },
+    {
+      "epoch": 233.18,
+      "learning_rate": 1.1723426212590299e-05,
+      "loss": 0.0187,
+      "step": 3964
+    },
+    {
+      "epoch": 233.24,
+      "learning_rate": 1.171310629514964e-05,
+      "loss": 0.0182,
+      "step": 3965
+    },
+    {
+      "epoch": 233.29,
+      "learning_rate": 1.1702786377708978e-05,
+      "loss": 0.0215,
+      "step": 3966
+    },
+    {
+      "epoch": 233.35,
+      "learning_rate": 1.1692466460268319e-05,
+      "loss": 0.019,
+      "step": 3967
+    },
+    {
+      "epoch": 233.41,
+      "learning_rate": 1.1682146542827658e-05,
+      "loss": 0.0191,
+      "step": 3968
+    },
+    {
+      "epoch": 233.47,
+      "learning_rate": 1.1671826625386997e-05,
+      "loss": 0.0212,
+      "step": 3969
+    },
+    {
+      "epoch": 233.53,
+      "learning_rate": 1.1661506707946337e-05,
+      "loss": 0.0205,
+      "step": 3970
+    },
+    {
+      "epoch": 233.59,
+      "learning_rate": 1.1651186790505676e-05,
+      "loss": 0.0236,
+      "step": 3971
+    },
+    {
+      "epoch": 233.65,
+      "learning_rate": 1.1640866873065017e-05,
+      "loss": 0.017,
+      "step": 3972
+    },
+    {
+      "epoch": 233.71,
+      "learning_rate": 1.1630546955624356e-05,
+      "loss": 0.0194,
+      "step": 3973
+    },
+    {
+      "epoch": 233.76,
+      "learning_rate": 1.1620227038183695e-05,
+      "loss": 0.0176,
+      "step": 3974
+    },
+    {
+      "epoch": 233.82,
+      "learning_rate": 1.1609907120743033e-05,
+      "loss": 0.0222,
+      "step": 3975
+    },
+    {
+      "epoch": 233.88,
+      "learning_rate": 1.1599587203302374e-05,
+      "loss": 0.0175,
+      "step": 3976
+    },
+    {
+      "epoch": 233.94,
+      "learning_rate": 1.1589267285861715e-05,
+      "loss": 0.0209,
+      "step": 3977
+    },
+    {
+      "epoch": 234.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 0.0195,
+      "step": 3978
+    },
+    {
+      "epoch": 234.06,
+      "learning_rate": 1.1568627450980394e-05,
+      "loss": 0.0206,
+      "step": 3979
+    },
+    {
+      "epoch": 234.12,
+      "learning_rate": 1.1558307533539733e-05,
+      "loss": 0.017,
+      "step": 3980
+    },
+    {
+      "epoch": 234.12,
+      "eval_accuracy_NAT": 0.9942759728929519,
+      "eval_accuracy_REST": 0.3983739837398374,
+      "eval_accuracy_concrete": 0.9591318113647725,
+      "eval_accuracy_concrete tiles": 0.9360122554796135,
+      "eval_accuracy_dark bricks": 0.855238979384798,
+      "eval_accuracy_grass": 0.8958931822396833,
+      "eval_accuracy_red bricks": 0.9249231107205624,
+      "eval_accuracy_road": 0.9809667694705165,
+      "eval_accuracy_rough concrete": 0.6253523141003104,
+      "eval_accuracy_rough red sidewalk": 0.7733100715150513,
+      "eval_accuracy_speedway bricks": 0.9885822823508283,
+      "eval_accuracy_steel": 0.9256398466667,
+      "eval_accuracy_tiles": 0.23923778363476628,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840191930816976,
+      "eval_iou_REST": 0.38341158059467917,
+      "eval_iou_concrete": 0.8924474863955038,
+      "eval_iou_concrete tiles": 0.8772142951804568,
+      "eval_iou_dark bricks": 0.8408896291486699,
+      "eval_iou_grass": 0.8454302983545985,
+      "eval_iou_red bricks": 0.9021803182086034,
+      "eval_iou_road": 0.9609441992681357,
+      "eval_iou_rough concrete": 0.6198855279712225,
+      "eval_iou_rough red sidewalk": 0.6260059657880319,
+      "eval_iou_speedway bricks": 0.9797955917160567,
+      "eval_iou_steel": 0.8721152533114843,
+      "eval_iou_tiles": 0.23573838025015673,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0821552649140358,
+      "eval_mean_accuracy": 0.8074567971969533,
+      "eval_mean_iou": 0.7707752091745613,
+      "eval_overall_accuracy": 0.9803459361448126,
+      "eval_runtime": 5.8399,
+      "eval_samples_per_second": 20.206,
+      "eval_steps_per_second": 1.37,
+      "step": 3980
+    },
+    {
+      "epoch": 234.18,
+      "learning_rate": 1.1547987616099072e-05,
+      "loss": 0.024,
+      "step": 3981
+    },
+    {
+      "epoch": 234.24,
+      "learning_rate": 1.153766769865841e-05,
+      "loss": 0.0213,
+      "step": 3982
+    },
+    {
+      "epoch": 234.29,
+      "learning_rate": 1.1527347781217751e-05,
+      "loss": 0.0202,
+      "step": 3983
+    },
+    {
+      "epoch": 234.35,
+      "learning_rate": 1.151702786377709e-05,
+      "loss": 0.0171,
+      "step": 3984
+    },
+    {
+      "epoch": 234.41,
+      "learning_rate": 1.150670794633643e-05,
+      "loss": 0.0195,
+      "step": 3985
+    },
+    {
+      "epoch": 234.47,
+      "learning_rate": 1.149638802889577e-05,
+      "loss": 0.0186,
+      "step": 3986
+    },
+    {
+      "epoch": 234.53,
+      "learning_rate": 1.1486068111455108e-05,
+      "loss": 0.0234,
+      "step": 3987
+    },
+    {
+      "epoch": 234.59,
+      "learning_rate": 1.1475748194014449e-05,
+      "loss": 0.019,
+      "step": 3988
+    },
+    {
+      "epoch": 234.65,
+      "learning_rate": 1.1465428276573788e-05,
+      "loss": 0.0206,
+      "step": 3989
+    },
+    {
+      "epoch": 234.71,
+      "learning_rate": 1.1455108359133128e-05,
+      "loss": 0.0194,
+      "step": 3990
+    },
+    {
+      "epoch": 234.76,
+      "learning_rate": 1.1444788441692467e-05,
+      "loss": 0.0211,
+      "step": 3991
+    },
+    {
+      "epoch": 234.82,
+      "learning_rate": 1.1434468524251806e-05,
+      "loss": 0.0193,
+      "step": 3992
+    },
+    {
+      "epoch": 234.88,
+      "learning_rate": 1.1424148606811145e-05,
+      "loss": 0.0176,
+      "step": 3993
+    },
+    {
+      "epoch": 234.94,
+      "learning_rate": 1.1413828689370486e-05,
+      "loss": 0.0184,
+      "step": 3994
+    },
+    {
+      "epoch": 235.0,
+      "learning_rate": 1.1403508771929824e-05,
+      "loss": 0.0185,
+      "step": 3995
+    },
+    {
+      "epoch": 235.06,
+      "learning_rate": 1.1393188854489165e-05,
+      "loss": 0.0196,
+      "step": 3996
+    },
+    {
+      "epoch": 235.12,
+      "learning_rate": 1.1382868937048506e-05,
+      "loss": 0.0219,
+      "step": 3997
+    },
+    {
+      "epoch": 235.18,
+      "learning_rate": 1.1372549019607843e-05,
+      "loss": 0.0202,
+      "step": 3998
+    },
+    {
+      "epoch": 235.24,
+      "learning_rate": 1.1362229102167183e-05,
+      "loss": 0.0182,
+      "step": 3999
+    },
+    {
+      "epoch": 235.29,
+      "learning_rate": 1.1351909184726522e-05,
+      "loss": 0.0178,
+      "step": 4000
+    },
+    {
+      "epoch": 235.29,
+      "eval_accuracy_NAT": 0.9938466735555113,
+      "eval_accuracy_REST": 0.4504065040650406,
+      "eval_accuracy_concrete": 0.9567686731110564,
+      "eval_accuracy_concrete tiles": 0.9284232854112656,
+      "eval_accuracy_dark bricks": 0.8600960894282488,
+      "eval_accuracy_grass": 0.8952497473571859,
+      "eval_accuracy_red bricks": 0.9262961335676626,
+      "eval_accuracy_road": 0.9841721974282827,
+      "eval_accuracy_rough concrete": 0.6244882879769276,
+      "eval_accuracy_rough red sidewalk": 0.785146526895224,
+      "eval_accuracy_speedway bricks": 0.9895040555277944,
+      "eval_accuracy_steel": 0.9195649892296851,
+      "eval_accuracy_tiles": 0.23560241146354025,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840878319603651,
+      "eval_iou_REST": 0.43213728549141966,
+      "eval_iou_concrete": 0.8925642229688413,
+      "eval_iou_concrete tiles": 0.8716615405040604,
+      "eval_iou_dark bricks": 0.8431580428765774,
+      "eval_iou_grass": 0.8438721657980808,
+      "eval_iou_red bricks": 0.9020698507781997,
+      "eval_iou_road": 0.958732895610906,
+      "eval_iou_rough concrete": 0.6172462129243528,
+      "eval_iou_rough red sidewalk": 0.6180072568854584,
+      "eval_iou_speedway bricks": 0.9800962606165554,
+      "eval_iou_steel": 0.8699473756377099,
+      "eval_iou_tiles": 0.22608215355097533,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08360882103443146,
+      "eval_mean_accuracy": 0.8115050442321097,
+      "eval_mean_iou": 0.7722817765848847,
+      "eval_overall_accuracy": 0.9802224110748808,
+      "eval_runtime": 5.8901,
+      "eval_samples_per_second": 20.034,
+      "eval_steps_per_second": 1.358,
+      "step": 4000
+    },
+    {
+      "epoch": 235.35,
+      "learning_rate": 1.1341589267285863e-05,
+      "loss": 0.0175,
+      "step": 4001
+    },
+    {
+      "epoch": 235.41,
+      "learning_rate": 1.1331269349845202e-05,
+      "loss": 0.0192,
+      "step": 4002
+    },
+    {
+      "epoch": 235.47,
+      "learning_rate": 1.1320949432404542e-05,
+      "loss": 0.0197,
+      "step": 4003
+    },
+    {
+      "epoch": 235.53,
+      "learning_rate": 1.1310629514963881e-05,
+      "loss": 0.0209,
+      "step": 4004
+    },
+    {
+      "epoch": 235.59,
+      "learning_rate": 1.130030959752322e-05,
+      "loss": 0.0199,
+      "step": 4005
+    },
+    {
+      "epoch": 235.65,
+      "learning_rate": 1.128998968008256e-05,
+      "loss": 0.0203,
+      "step": 4006
+    },
+    {
+      "epoch": 235.71,
+      "learning_rate": 1.12796697626419e-05,
+      "loss": 0.0196,
+      "step": 4007
+    },
+    {
+      "epoch": 235.76,
+      "learning_rate": 1.126934984520124e-05,
+      "loss": 0.022,
+      "step": 4008
+    },
+    {
+      "epoch": 235.82,
+      "learning_rate": 1.1259029927760579e-05,
+      "loss": 0.0196,
+      "step": 4009
+    },
+    {
+      "epoch": 235.88,
+      "learning_rate": 1.1248710010319918e-05,
+      "loss": 0.0234,
+      "step": 4010
+    },
+    {
+      "epoch": 235.94,
+      "learning_rate": 1.1238390092879257e-05,
+      "loss": 0.0184,
+      "step": 4011
+    },
+    {
+      "epoch": 236.0,
+      "learning_rate": 1.1228070175438597e-05,
+      "loss": 0.0185,
+      "step": 4012
+    },
+    {
+      "epoch": 236.06,
+      "learning_rate": 1.1217750257997936e-05,
+      "loss": 0.0219,
+      "step": 4013
+    },
+    {
+      "epoch": 236.12,
+      "learning_rate": 1.1207430340557277e-05,
+      "loss": 0.0196,
+      "step": 4014
+    },
+    {
+      "epoch": 236.18,
+      "learning_rate": 1.1197110423116617e-05,
+      "loss": 0.0193,
+      "step": 4015
+    },
+    {
+      "epoch": 236.24,
+      "learning_rate": 1.1186790505675954e-05,
+      "loss": 0.0164,
+      "step": 4016
+    },
+    {
+      "epoch": 236.29,
+      "learning_rate": 1.1176470588235295e-05,
+      "loss": 0.0181,
+      "step": 4017
+    },
+    {
+      "epoch": 236.35,
+      "learning_rate": 1.1166150670794634e-05,
+      "loss": 0.0196,
+      "step": 4018
+    },
+    {
+      "epoch": 236.41,
+      "learning_rate": 1.1155830753353974e-05,
+      "loss": 0.0203,
+      "step": 4019
+    },
+    {
+      "epoch": 236.47,
+      "learning_rate": 1.1145510835913313e-05,
+      "loss": 0.0198,
+      "step": 4020
+    },
+    {
+      "epoch": 236.47,
+      "eval_accuracy_NAT": 0.9939158423423926,
+      "eval_accuracy_REST": 0.3886178861788618,
+      "eval_accuracy_concrete": 0.9590912982392843,
+      "eval_accuracy_concrete tiles": 0.940560923874617,
+      "eval_accuracy_dark bricks": 0.8569473422276669,
+      "eval_accuracy_grass": 0.8932068416052564,
+      "eval_accuracy_red bricks": 0.9279986818980668,
+      "eval_accuracy_road": 0.9807375049500294,
+      "eval_accuracy_rough concrete": 0.6237136438663083,
+      "eval_accuracy_rough red sidewalk": 0.7594733002955354,
+      "eval_accuracy_speedway bricks": 0.9901866029462468,
+      "eval_accuracy_steel": 0.924925157556463,
+      "eval_accuracy_tiles": 0.24072222727135023,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841582654192389,
+      "eval_iou_REST": 0.37756714060031593,
+      "eval_iou_concrete": 0.8927869785644401,
+      "eval_iou_concrete tiles": 0.8755402470327549,
+      "eval_iou_dark bricks": 0.8402881073598761,
+      "eval_iou_grass": 0.8432453968219937,
+      "eval_iou_red bricks": 0.9005009592837349,
+      "eval_iou_road": 0.9603118993658297,
+      "eval_iou_rough concrete": 0.6186083071322191,
+      "eval_iou_rough red sidewalk": 0.6092564217028824,
+      "eval_iou_speedway bricks": 0.9797095473183068,
+      "eval_iou_steel": 0.8716553234580694,
+      "eval_iou_tiles": 0.2345613413626166,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08356532454490662,
+      "eval_mean_accuracy": 0.8061613271732369,
+      "eval_mean_iou": 0.7683223027247906,
+      "eval_overall_accuracy": 0.9802669266522941,
+      "eval_runtime": 5.5165,
+      "eval_samples_per_second": 21.39,
+      "eval_steps_per_second": 1.45,
+      "step": 4020
+    },
+    {
+      "epoch": 236.53,
+      "learning_rate": 1.1135190918472654e-05,
+      "loss": 0.0179,
+      "step": 4021
+    },
+    {
+      "epoch": 236.59,
+      "learning_rate": 1.1124871001031993e-05,
+      "loss": 0.021,
+      "step": 4022
+    },
+    {
+      "epoch": 236.65,
+      "learning_rate": 1.1114551083591331e-05,
+      "loss": 0.0192,
+      "step": 4023
+    },
+    {
+      "epoch": 236.71,
+      "learning_rate": 1.110423116615067e-05,
+      "loss": 0.0205,
+      "step": 4024
+    },
+    {
+      "epoch": 236.76,
+      "learning_rate": 1.1093911248710011e-05,
+      "loss": 0.0176,
+      "step": 4025
+    },
+    {
+      "epoch": 236.82,
+      "learning_rate": 1.1083591331269351e-05,
+      "loss": 0.0203,
+      "step": 4026
+    },
+    {
+      "epoch": 236.88,
+      "learning_rate": 1.107327141382869e-05,
+      "loss": 0.0212,
+      "step": 4027
+    },
+    {
+      "epoch": 236.94,
+      "learning_rate": 1.106295149638803e-05,
+      "loss": 0.0211,
+      "step": 4028
+    },
+    {
+      "epoch": 237.0,
+      "learning_rate": 1.1052631578947368e-05,
+      "loss": 0.0188,
+      "step": 4029
+    },
+    {
+      "epoch": 237.06,
+      "learning_rate": 1.1042311661506709e-05,
+      "loss": 0.0213,
+      "step": 4030
+    },
+    {
+      "epoch": 237.12,
+      "learning_rate": 1.1031991744066048e-05,
+      "loss": 0.0194,
+      "step": 4031
+    },
+    {
+      "epoch": 237.18,
+      "learning_rate": 1.1021671826625388e-05,
+      "loss": 0.0201,
+      "step": 4032
+    },
+    {
+      "epoch": 237.24,
+      "learning_rate": 1.1011351909184727e-05,
+      "loss": 0.0184,
+      "step": 4033
+    },
+    {
+      "epoch": 237.29,
+      "learning_rate": 1.1001031991744066e-05,
+      "loss": 0.0232,
+      "step": 4034
+    },
+    {
+      "epoch": 237.35,
+      "learning_rate": 1.0990712074303406e-05,
+      "loss": 0.0182,
+      "step": 4035
+    },
+    {
+      "epoch": 237.41,
+      "learning_rate": 1.0980392156862745e-05,
+      "loss": 0.0196,
+      "step": 4036
+    },
+    {
+      "epoch": 237.47,
+      "learning_rate": 1.0970072239422086e-05,
+      "loss": 0.0185,
+      "step": 4037
+    },
+    {
+      "epoch": 237.53,
+      "learning_rate": 1.0959752321981425e-05,
+      "loss": 0.0186,
+      "step": 4038
+    },
+    {
+      "epoch": 237.59,
+      "learning_rate": 1.0949432404540765e-05,
+      "loss": 0.0219,
+      "step": 4039
+    },
+    {
+      "epoch": 237.65,
+      "learning_rate": 1.0939112487100102e-05,
+      "loss": 0.017,
+      "step": 4040
+    },
+    {
+      "epoch": 237.65,
+      "eval_accuracy_NAT": 0.9940503521812617,
+      "eval_accuracy_REST": 0.4065040650406504,
+      "eval_accuracy_concrete": 0.9590855652498284,
+      "eval_accuracy_concrete tiles": 0.9321942022154136,
+      "eval_accuracy_dark bricks": 0.8549375035889976,
+      "eval_accuracy_grass": 0.894547646353049,
+      "eval_accuracy_red bricks": 0.9280536028119508,
+      "eval_accuracy_road": 0.9806381035495385,
+      "eval_accuracy_rough concrete": 0.626317639838159,
+      "eval_accuracy_rough red sidewalk": 0.7651659284548689,
+      "eval_accuracy_speedway bricks": 0.9906724231650048,
+      "eval_accuracy_steel": 0.9182805479616367,
+      "eval_accuracy_tiles": 0.24599351691962798,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841700932220923,
+      "eval_iou_REST": 0.3937007874015748,
+      "eval_iou_concrete": 0.8936566194135466,
+      "eval_iou_concrete tiles": 0.8820525400294367,
+      "eval_iou_dark bricks": 0.8394872614158577,
+      "eval_iou_grass": 0.8442872057020329,
+      "eval_iou_red bricks": 0.9028638597991023,
+      "eval_iou_road": 0.96114150845556,
+      "eval_iou_rough concrete": 0.620475917803529,
+      "eval_iou_rough red sidewalk": 0.614950864851144,
+      "eval_iou_speedway bricks": 0.9797976699331894,
+      "eval_iou_steel": 0.8722563370987612,
+      "eval_iou_tiles": 0.24209176828359322,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0825955718755722,
+      "eval_mean_accuracy": 0.8074185459484604,
+      "eval_mean_iou": 0.7716101871853399,
+      "eval_overall_accuracy": 0.9804121437719313,
+      "eval_runtime": 5.5028,
+      "eval_samples_per_second": 21.444,
+      "eval_steps_per_second": 1.454,
+      "step": 4040
+    },
+    {
+      "epoch": 237.71,
+      "learning_rate": 1.0928792569659443e-05,
+      "loss": 0.0198,
+      "step": 4041
+    },
+    {
+      "epoch": 237.76,
+      "learning_rate": 1.0918472652218782e-05,
+      "loss": 0.0184,
+      "step": 4042
+    },
+    {
+      "epoch": 237.82,
+      "learning_rate": 1.0908152734778122e-05,
+      "loss": 0.0212,
+      "step": 4043
+    },
+    {
+      "epoch": 237.88,
+      "learning_rate": 1.0897832817337463e-05,
+      "loss": 0.0198,
+      "step": 4044
+    },
+    {
+      "epoch": 237.94,
+      "learning_rate": 1.0887512899896802e-05,
+      "loss": 0.0169,
+      "step": 4045
+    },
+    {
+      "epoch": 238.0,
+      "learning_rate": 1.087719298245614e-05,
+      "loss": 0.0175,
+      "step": 4046
+    },
+    {
+      "epoch": 238.06,
+      "learning_rate": 1.086687306501548e-05,
+      "loss": 0.0199,
+      "step": 4047
+    },
+    {
+      "epoch": 238.12,
+      "learning_rate": 1.085655314757482e-05,
+      "loss": 0.0184,
+      "step": 4048
+    },
+    {
+      "epoch": 238.18,
+      "learning_rate": 1.0846233230134159e-05,
+      "loss": 0.0189,
+      "step": 4049
+    },
+    {
+      "epoch": 238.24,
+      "learning_rate": 1.08359133126935e-05,
+      "loss": 0.0183,
+      "step": 4050
+    },
+    {
+      "epoch": 238.29,
+      "learning_rate": 1.0825593395252839e-05,
+      "loss": 0.0192,
+      "step": 4051
+    },
+    {
+      "epoch": 238.35,
+      "learning_rate": 1.0815273477812177e-05,
+      "loss": 0.0197,
+      "step": 4052
+    },
+    {
+      "epoch": 238.41,
+      "learning_rate": 1.0804953560371518e-05,
+      "loss": 0.0197,
+      "step": 4053
+    },
+    {
+      "epoch": 238.47,
+      "learning_rate": 1.0794633642930857e-05,
+      "loss": 0.0209,
+      "step": 4054
+    },
+    {
+      "epoch": 238.53,
+      "learning_rate": 1.0784313725490197e-05,
+      "loss": 0.0203,
+      "step": 4055
+    },
+    {
+      "epoch": 238.59,
+      "learning_rate": 1.0773993808049536e-05,
+      "loss": 0.0162,
+      "step": 4056
+    },
+    {
+      "epoch": 238.65,
+      "learning_rate": 1.0763673890608877e-05,
+      "loss": 0.019,
+      "step": 4057
+    },
+    {
+      "epoch": 238.71,
+      "learning_rate": 1.0753353973168214e-05,
+      "loss": 0.0193,
+      "step": 4058
+    },
+    {
+      "epoch": 238.76,
+      "learning_rate": 1.0743034055727555e-05,
+      "loss": 0.0211,
+      "step": 4059
+    },
+    {
+      "epoch": 238.82,
+      "learning_rate": 1.0732714138286893e-05,
+      "loss": 0.0195,
+      "step": 4060
+    },
+    {
+      "epoch": 238.82,
+      "eval_accuracy_NAT": 0.9942276140450699,
+      "eval_accuracy_REST": 0.3804878048780488,
+      "eval_accuracy_concrete": 0.9595174504555051,
+      "eval_accuracy_concrete tiles": 0.9243695498468065,
+      "eval_accuracy_dark bricks": 0.8474771739754609,
+      "eval_accuracy_grass": 0.891308708701889,
+      "eval_accuracy_red bricks": 0.9234951669595782,
+      "eval_accuracy_road": 0.9799300021750736,
+      "eval_accuracy_rough concrete": 0.6238924078918359,
+      "eval_accuracy_rough red sidewalk": 0.769151520164838,
+      "eval_accuracy_speedway bricks": 0.9900565367924419,
+      "eval_accuracy_steel": 0.9209368924517835,
+      "eval_accuracy_tiles": 0.24387288315307945,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840174831072203,
+      "eval_iou_REST": 0.36850393700787404,
+      "eval_iou_concrete": 0.8931210727203324,
+      "eval_iou_concrete tiles": 0.8769956620902464,
+      "eval_iou_dark bricks": 0.834498779579874,
+      "eval_iou_grass": 0.841873829750153,
+      "eval_iou_red bricks": 0.9021406727828746,
+      "eval_iou_road": 0.9606885180797818,
+      "eval_iou_rough concrete": 0.6195215469547878,
+      "eval_iou_rough red sidewalk": 0.6139242025906052,
+      "eval_iou_speedway bricks": 0.9798476703405582,
+      "eval_iou_steel": 0.8724539075594443,
+      "eval_iou_tiles": 0.2412129565816678,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0847599133849144,
+      "eval_mean_accuracy": 0.8037479778070316,
+      "eval_mean_iou": 0.7683692491650324,
+      "eval_overall_accuracy": 0.9802570343017578,
+      "eval_runtime": 5.3637,
+      "eval_samples_per_second": 22.0,
+      "eval_steps_per_second": 1.492,
+      "step": 4060
+    },
+    {
+      "epoch": 238.88,
+      "learning_rate": 1.0722394220846234e-05,
+      "loss": 0.0194,
+      "step": 4061
+    },
+    {
+      "epoch": 238.94,
+      "learning_rate": 1.0712074303405573e-05,
+      "loss": 0.02,
+      "step": 4062
+    },
+    {
+      "epoch": 239.0,
+      "learning_rate": 1.0701754385964913e-05,
+      "loss": 0.021,
+      "step": 4063
+    },
+    {
+      "epoch": 239.06,
+      "learning_rate": 1.0691434468524252e-05,
+      "loss": 0.0208,
+      "step": 4064
+    },
+    {
+      "epoch": 239.12,
+      "learning_rate": 1.0681114551083591e-05,
+      "loss": 0.0194,
+      "step": 4065
+    },
+    {
+      "epoch": 239.18,
+      "learning_rate": 1.0670794633642932e-05,
+      "loss": 0.0219,
+      "step": 4066
+    },
+    {
+      "epoch": 239.24,
+      "learning_rate": 1.066047471620227e-05,
+      "loss": 0.0195,
+      "step": 4067
+    },
+    {
+      "epoch": 239.29,
+      "learning_rate": 1.0650154798761611e-05,
+      "loss": 0.0179,
+      "step": 4068
+    },
+    {
+      "epoch": 239.35,
+      "learning_rate": 1.063983488132095e-05,
+      "loss": 0.0183,
+      "step": 4069
+    },
+    {
+      "epoch": 239.41,
+      "learning_rate": 1.0629514963880289e-05,
+      "loss": 0.0231,
+      "step": 4070
+    },
+    {
+      "epoch": 239.47,
+      "learning_rate": 1.0619195046439628e-05,
+      "loss": 0.0189,
+      "step": 4071
+    },
+    {
+      "epoch": 239.53,
+      "learning_rate": 1.0608875128998968e-05,
+      "loss": 0.0158,
+      "step": 4072
+    },
+    {
+      "epoch": 239.59,
+      "learning_rate": 1.0598555211558309e-05,
+      "loss": 0.0186,
+      "step": 4073
+    },
+    {
+      "epoch": 239.65,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0205,
+      "step": 4074
+    },
+    {
+      "epoch": 239.71,
+      "learning_rate": 1.0577915376676988e-05,
+      "loss": 0.022,
+      "step": 4075
+    },
+    {
+      "epoch": 239.76,
+      "learning_rate": 1.0567595459236326e-05,
+      "loss": 0.0161,
+      "step": 4076
+    },
+    {
+      "epoch": 239.82,
+      "learning_rate": 1.0557275541795666e-05,
+      "loss": 0.0172,
+      "step": 4077
+    },
+    {
+      "epoch": 239.88,
+      "learning_rate": 1.0546955624355005e-05,
+      "loss": 0.0214,
+      "step": 4078
+    },
+    {
+      "epoch": 239.94,
+      "learning_rate": 1.0536635706914346e-05,
+      "loss": 0.0191,
+      "step": 4079
+    },
+    {
+      "epoch": 240.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0195,
+      "step": 4080
+    },
+    {
+      "epoch": 240.0,
+      "eval_accuracy_NAT": 0.9938835491987122,
+      "eval_accuracy_REST": 0.37398373983739835,
+      "eval_accuracy_concrete": 0.9575911659983275,
+      "eval_accuracy_concrete tiles": 0.9335847277869432,
+      "eval_accuracy_dark bricks": 0.8518270390291522,
+      "eval_accuracy_grass": 0.9005126662049076,
+      "eval_accuracy_red bricks": 0.929426625659051,
+      "eval_accuracy_road": 0.9802148459732546,
+      "eval_accuracy_rough concrete": 0.6219140860093315,
+      "eval_accuracy_rough red sidewalk": 0.7682416020574677,
+      "eval_accuracy_speedway bricks": 0.9899965589876569,
+      "eval_accuracy_steel": 0.9266594031596256,
+      "eval_accuracy_tiles": 0.24341846163167621,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840869946121068,
+      "eval_iou_REST": 0.3668261562998405,
+      "eval_iou_concrete": 0.8931821617104772,
+      "eval_iou_concrete tiles": 0.8809518514400089,
+      "eval_iou_dark bricks": 0.8361579797827966,
+      "eval_iou_grass": 0.8435565475135262,
+      "eval_iou_red bricks": 0.9046346287485968,
+      "eval_iou_road": 0.9608539974487992,
+      "eval_iou_rough concrete": 0.6178605256926356,
+      "eval_iou_rough red sidewalk": 0.6147624836019209,
+      "eval_iou_speedway bricks": 0.9798454769013358,
+      "eval_iou_steel": 0.8731231354968437,
+      "eval_iou_tiles": 0.24085009442163005,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08413463085889816,
+      "eval_mean_accuracy": 0.8054811131948849,
+      "eval_mean_iou": 0.7689763102823476,
+      "eval_overall_accuracy": 0.9803124443959381,
+      "eval_runtime": 5.5622,
+      "eval_samples_per_second": 21.215,
+      "eval_steps_per_second": 1.438,
+      "step": 4080
+    },
+    {
+      "epoch": 240.06,
+      "learning_rate": 1.0515995872033025e-05,
+      "loss": 0.019,
+      "step": 4081
+    },
+    {
+      "epoch": 240.12,
+      "learning_rate": 1.0505675954592364e-05,
+      "loss": 0.0207,
+      "step": 4082
+    },
+    {
+      "epoch": 240.18,
+      "learning_rate": 1.0495356037151703e-05,
+      "loss": 0.0207,
+      "step": 4083
+    },
+    {
+      "epoch": 240.24,
+      "learning_rate": 1.0485036119711043e-05,
+      "loss": 0.0195,
+      "step": 4084
+    },
+    {
+      "epoch": 240.29,
+      "learning_rate": 1.0474716202270382e-05,
+      "loss": 0.0201,
+      "step": 4085
+    },
+    {
+      "epoch": 240.35,
+      "learning_rate": 1.0464396284829723e-05,
+      "loss": 0.0185,
+      "step": 4086
+    },
+    {
+      "epoch": 240.41,
+      "learning_rate": 1.0454076367389062e-05,
+      "loss": 0.0194,
+      "step": 4087
+    },
+    {
+      "epoch": 240.47,
+      "learning_rate": 1.04437564499484e-05,
+      "loss": 0.0181,
+      "step": 4088
+    },
+    {
+      "epoch": 240.53,
+      "learning_rate": 1.043343653250774e-05,
+      "loss": 0.0207,
+      "step": 4089
+    },
+    {
+      "epoch": 240.59,
+      "learning_rate": 1.042311661506708e-05,
+      "loss": 0.0191,
+      "step": 4090
+    },
+    {
+      "epoch": 240.65,
+      "learning_rate": 1.0412796697626419e-05,
+      "loss": 0.0176,
+      "step": 4091
+    },
+    {
+      "epoch": 240.71,
+      "learning_rate": 1.040247678018576e-05,
+      "loss": 0.0187,
+      "step": 4092
+    },
+    {
+      "epoch": 240.76,
+      "learning_rate": 1.0392156862745098e-05,
+      "loss": 0.0182,
+      "step": 4093
+    },
+    {
+      "epoch": 240.82,
+      "learning_rate": 1.0381836945304437e-05,
+      "loss": 0.0197,
+      "step": 4094
+    },
+    {
+      "epoch": 240.88,
+      "learning_rate": 1.0371517027863778e-05,
+      "loss": 0.0219,
+      "step": 4095
+    },
+    {
+      "epoch": 240.94,
+      "learning_rate": 1.0361197110423117e-05,
+      "loss": 0.0184,
+      "step": 4096
+    },
+    {
+      "epoch": 241.0,
+      "learning_rate": 1.0350877192982457e-05,
+      "loss": 0.0267,
+      "step": 4097
+    },
+    {
+      "epoch": 241.06,
+      "learning_rate": 1.0340557275541796e-05,
+      "loss": 0.0175,
+      "step": 4098
+    },
+    {
+      "epoch": 241.12,
+      "learning_rate": 1.0330237358101137e-05,
+      "loss": 0.019,
+      "step": 4099
+    },
+    {
+      "epoch": 241.18,
+      "learning_rate": 1.0319917440660474e-05,
+      "loss": 0.019,
+      "step": 4100
+    },
+    {
+      "epoch": 241.18,
+      "eval_accuracy_NAT": 0.9933779647221935,
+      "eval_accuracy_REST": 0.41300813008130083,
+      "eval_accuracy_concrete": 0.9557313842188382,
+      "eval_accuracy_concrete tiles": 0.9382748055621023,
+      "eval_accuracy_dark bricks": 0.8568277089753651,
+      "eval_accuracy_grass": 0.9091242851249209,
+      "eval_accuracy_red bricks": 0.9265158172231986,
+      "eval_accuracy_road": 0.981566384370252,
+      "eval_accuracy_rough concrete": 0.6237076850654574,
+      "eval_accuracy_rough red sidewalk": 0.7526301145293618,
+      "eval_accuracy_speedway bricks": 0.9907488520248164,
+      "eval_accuracy_steel": 0.9207319815880092,
+      "eval_accuracy_tiles": 0.2527189554363961,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841524518863645,
+      "eval_iou_REST": 0.4031746031746032,
+      "eval_iou_concrete": 0.8923381836141016,
+      "eval_iou_concrete tiles": 0.8822773308512288,
+      "eval_iou_dark bricks": 0.8387853860315646,
+      "eval_iou_grass": 0.8452016087885175,
+      "eval_iou_red bricks": 0.9044121589020533,
+      "eval_iou_road": 0.9596335988898456,
+      "eval_iou_rough concrete": 0.6181384389154969,
+      "eval_iou_rough red sidewalk": 0.6075307000770916,
+      "eval_iou_speedway bricks": 0.980055641801327,
+      "eval_iou_steel": 0.8725779026093052,
+      "eval_iou_tiles": 0.247316928550252,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08385305851697922,
+      "eval_mean_accuracy": 0.8088433899170933,
+      "eval_mean_iou": 0.771968841083981,
+      "eval_overall_accuracy": 0.9802849979723914,
+      "eval_runtime": 5.7565,
+      "eval_samples_per_second": 20.499,
+      "eval_steps_per_second": 1.39,
+      "step": 4100
+    },
+    {
+      "epoch": 241.24,
+      "learning_rate": 1.0309597523219814e-05,
+      "loss": 0.0204,
+      "step": 4101
+    },
+    {
+      "epoch": 241.29,
+      "learning_rate": 1.0299277605779155e-05,
+      "loss": 0.023,
+      "step": 4102
+    },
+    {
+      "epoch": 241.35,
+      "learning_rate": 1.0288957688338494e-05,
+      "loss": 0.0226,
+      "step": 4103
+    },
+    {
+      "epoch": 241.41,
+      "learning_rate": 1.0278637770897834e-05,
+      "loss": 0.0183,
+      "step": 4104
+    },
+    {
+      "epoch": 241.47,
+      "learning_rate": 1.0268317853457173e-05,
+      "loss": 0.0183,
+      "step": 4105
+    },
+    {
+      "epoch": 241.53,
+      "learning_rate": 1.0257997936016512e-05,
+      "loss": 0.0172,
+      "step": 4106
+    },
+    {
+      "epoch": 241.59,
+      "learning_rate": 1.0247678018575851e-05,
+      "loss": 0.0195,
+      "step": 4107
+    },
+    {
+      "epoch": 241.65,
+      "learning_rate": 1.0237358101135192e-05,
+      "loss": 0.0171,
+      "step": 4108
+    },
+    {
+      "epoch": 241.71,
+      "learning_rate": 1.022703818369453e-05,
+      "loss": 0.0221,
+      "step": 4109
+    },
+    {
+      "epoch": 241.76,
+      "learning_rate": 1.0216718266253871e-05,
+      "loss": 0.0177,
+      "step": 4110
+    },
+    {
+      "epoch": 241.82,
+      "learning_rate": 1.020639834881321e-05,
+      "loss": 0.0232,
+      "step": 4111
+    },
+    {
+      "epoch": 241.88,
+      "learning_rate": 1.0196078431372549e-05,
+      "loss": 0.0192,
+      "step": 4112
+    },
+    {
+      "epoch": 241.94,
+      "learning_rate": 1.018575851393189e-05,
+      "loss": 0.0156,
+      "step": 4113
+    },
+    {
+      "epoch": 242.0,
+      "learning_rate": 1.0175438596491228e-05,
+      "loss": 0.0205,
+      "step": 4114
+    },
+    {
+      "epoch": 242.06,
+      "learning_rate": 1.0165118679050569e-05,
+      "loss": 0.0187,
+      "step": 4115
+    },
+    {
+      "epoch": 242.12,
+      "learning_rate": 1.0154798761609908e-05,
+      "loss": 0.0205,
+      "step": 4116
+    },
+    {
+      "epoch": 242.18,
+      "learning_rate": 1.0144478844169248e-05,
+      "loss": 0.0209,
+      "step": 4117
+    },
+    {
+      "epoch": 242.24,
+      "learning_rate": 1.0134158926728585e-05,
+      "loss": 0.0212,
+      "step": 4118
+    },
+    {
+      "epoch": 242.29,
+      "learning_rate": 1.0123839009287926e-05,
+      "loss": 0.0179,
+      "step": 4119
+    },
+    {
+      "epoch": 242.35,
+      "learning_rate": 1.0113519091847266e-05,
+      "loss": 0.0155,
+      "step": 4120
+    },
+    {
+      "epoch": 242.35,
+      "eval_accuracy_NAT": 0.9937249947152998,
+      "eval_accuracy_REST": 0.4146341463414634,
+      "eval_accuracy_concrete": 0.958778277014993,
+      "eval_accuracy_concrete tiles": 0.9400659910440726,
+      "eval_accuracy_dark bricks": 0.8506498478265031,
+      "eval_accuracy_grass": 0.898026925857377,
+      "eval_accuracy_red bricks": 0.9291520210896309,
+      "eval_accuracy_road": 0.9800064236819027,
+      "eval_accuracy_rough concrete": 0.626174628617737,
+      "eval_accuracy_rough red sidewalk": 0.7814391746065168,
+      "eval_accuracy_speedway bricks": 0.9902707432381023,
+      "eval_accuracy_steel": 0.9262945618655885,
+      "eval_accuracy_tiles": 0.2435396407040504,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984141556708404,
+      "eval_iou_REST": 0.3965785381026439,
+      "eval_iou_concrete": 0.8933200292291454,
+      "eval_iou_concrete tiles": 0.8802939683520558,
+      "eval_iou_dark bricks": 0.8336670903113554,
+      "eval_iou_grass": 0.8435871430869782,
+      "eval_iou_red bricks": 0.9036427732079906,
+      "eval_iou_road": 0.9614757257119488,
+      "eval_iou_rough concrete": 0.6203159311468442,
+      "eval_iou_rough red sidewalk": 0.6217912662605761,
+      "eval_iou_speedway bricks": 0.9796247068737626,
+      "eval_iou_steel": 0.8736227348038303,
+      "eval_iou_tiles": 0.24129547364629608,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08268430083990097,
+      "eval_mean_accuracy": 0.8102121058925568,
+      "eval_mean_iou": 0.7717966874955255,
+      "eval_overall_accuracy": 0.9803488133317333,
+      "eval_runtime": 5.7229,
+      "eval_samples_per_second": 20.619,
+      "eval_steps_per_second": 1.398,
+      "step": 4120
+    },
+    {
+      "epoch": 242.41,
+      "learning_rate": 1.0103199174406605e-05,
+      "loss": 0.0202,
+      "step": 4121
+    },
+    {
+      "epoch": 242.47,
+      "learning_rate": 1.0092879256965946e-05,
+      "loss": 0.018,
+      "step": 4122
+    },
+    {
+      "epoch": 242.53,
+      "learning_rate": 1.0082559339525285e-05,
+      "loss": 0.0167,
+      "step": 4123
+    },
+    {
+      "epoch": 242.59,
+      "learning_rate": 1.0072239422084624e-05,
+      "loss": 0.0215,
+      "step": 4124
+    },
+    {
+      "epoch": 242.65,
+      "learning_rate": 1.0061919504643963e-05,
+      "loss": 0.0183,
+      "step": 4125
+    },
+    {
+      "epoch": 242.71,
+      "learning_rate": 1.0051599587203303e-05,
+      "loss": 0.0197,
+      "step": 4126
+    },
+    {
+      "epoch": 242.76,
+      "learning_rate": 1.0041279669762642e-05,
+      "loss": 0.0195,
+      "step": 4127
+    },
+    {
+      "epoch": 242.82,
+      "learning_rate": 1.0030959752321983e-05,
+      "loss": 0.0178,
+      "step": 4128
+    },
+    {
+      "epoch": 242.88,
+      "learning_rate": 1.0020639834881321e-05,
+      "loss": 0.0207,
+      "step": 4129
+    },
+    {
+      "epoch": 242.94,
+      "learning_rate": 1.001031991744066e-05,
+      "loss": 0.0184,
+      "step": 4130
+    },
+    {
+      "epoch": 243.0,
+      "learning_rate": 1e-05,
+      "loss": 0.0183,
+      "step": 4131
+    },
+    {
+      "epoch": 243.06,
+      "learning_rate": 9.98968008255934e-06,
+      "loss": 0.021,
+      "step": 4132
+    },
+    {
+      "epoch": 243.12,
+      "learning_rate": 9.97936016511868e-06,
+      "loss": 0.021,
+      "step": 4133
+    },
+    {
+      "epoch": 243.18,
+      "learning_rate": 9.969040247678019e-06,
+      "loss": 0.0208,
+      "step": 4134
+    },
+    {
+      "epoch": 243.24,
+      "learning_rate": 9.958720330237358e-06,
+      "loss": 0.0172,
+      "step": 4135
+    },
+    {
+      "epoch": 243.29,
+      "learning_rate": 9.948400412796697e-06,
+      "loss": 0.0198,
+      "step": 4136
+    },
+    {
+      "epoch": 243.35,
+      "learning_rate": 9.938080495356037e-06,
+      "loss": 0.021,
+      "step": 4137
+    },
+    {
+      "epoch": 243.41,
+      "learning_rate": 9.927760577915376e-06,
+      "loss": 0.0225,
+      "step": 4138
+    },
+    {
+      "epoch": 243.47,
+      "learning_rate": 9.917440660474717e-06,
+      "loss": 0.0176,
+      "step": 4139
+    },
+    {
+      "epoch": 243.53,
+      "learning_rate": 9.907120743034057e-06,
+      "loss": 0.0185,
+      "step": 4140
+    },
+    {
+      "epoch": 243.53,
+      "eval_accuracy_NAT": 0.9932658821750958,
+      "eval_accuracy_REST": 0.44065040650406506,
+      "eval_accuracy_concrete": 0.9585818265763045,
+      "eval_accuracy_concrete tiles": 0.9357530049493283,
+      "eval_accuracy_dark bricks": 0.8510948835250656,
+      "eval_accuracy_grass": 0.9063518377635718,
+      "eval_accuracy_red bricks": 0.9301955184534271,
+      "eval_accuracy_road": 0.9809181376025344,
+      "eval_accuracy_rough concrete": 0.6268956435206979,
+      "eval_accuracy_rough red sidewalk": 0.7668428849667993,
+      "eval_accuracy_speedway bricks": 0.9902135072758218,
+      "eval_accuracy_steel": 0.9265444531628741,
+      "eval_accuracy_tiles": 0.23484504226120148,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842127289223486,
+      "eval_iou_REST": 0.4274447949526814,
+      "eval_iou_concrete": 0.8925083287843213,
+      "eval_iou_concrete tiles": 0.8810776024676564,
+      "eval_iou_dark bricks": 0.8336770758139666,
+      "eval_iou_grass": 0.8459774784720689,
+      "eval_iou_red bricks": 0.9037886872998933,
+      "eval_iou_road": 0.9600752998183933,
+      "eval_iou_rough concrete": 0.6209202403295678,
+      "eval_iou_rough red sidewalk": 0.6185453199968458,
+      "eval_iou_speedway bricks": 0.9796522854834242,
+      "eval_iou_steel": 0.8737039794898863,
+      "eval_iou_tiles": 0.2329326923076923,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08267069607973099,
+      "eval_mean_accuracy": 0.8109348483643684,
+      "eval_mean_iou": 0.7734243472414419,
+      "eval_overall_accuracy": 0.9803190069683527,
+      "eval_runtime": 5.188,
+      "eval_samples_per_second": 22.745,
+      "eval_steps_per_second": 1.542,
+      "step": 4140
+    },
+    {
+      "epoch": 243.59,
+      "learning_rate": 9.896800825593396e-06,
+      "loss": 0.0168,
+      "step": 4141
+    },
+    {
+      "epoch": 243.65,
+      "learning_rate": 9.886480908152735e-06,
+      "loss": 0.017,
+      "step": 4142
+    },
+    {
+      "epoch": 243.71,
+      "learning_rate": 9.876160990712074e-06,
+      "loss": 0.0178,
+      "step": 4143
+    },
+    {
+      "epoch": 243.76,
+      "learning_rate": 9.865841073271415e-06,
+      "loss": 0.025,
+      "step": 4144
+    },
+    {
+      "epoch": 243.82,
+      "learning_rate": 9.855521155830754e-06,
+      "loss": 0.0193,
+      "step": 4145
+    },
+    {
+      "epoch": 243.88,
+      "learning_rate": 9.845201238390094e-06,
+      "loss": 0.0159,
+      "step": 4146
+    },
+    {
+      "epoch": 243.94,
+      "learning_rate": 9.834881320949433e-06,
+      "loss": 0.02,
+      "step": 4147
+    },
+    {
+      "epoch": 244.0,
+      "learning_rate": 9.824561403508772e-06,
+      "loss": 0.0159,
+      "step": 4148
+    },
+    {
+      "epoch": 244.06,
+      "learning_rate": 9.814241486068112e-06,
+      "loss": 0.0182,
+      "step": 4149
+    },
+    {
+      "epoch": 244.12,
+      "learning_rate": 9.803921568627451e-06,
+      "loss": 0.0178,
+      "step": 4150
+    },
+    {
+      "epoch": 244.18,
+      "learning_rate": 9.793601651186792e-06,
+      "loss": 0.0181,
+      "step": 4151
+    },
+    {
+      "epoch": 244.24,
+      "learning_rate": 9.78328173374613e-06,
+      "loss": 0.0206,
+      "step": 4152
+    },
+    {
+      "epoch": 244.29,
+      "learning_rate": 9.77296181630547e-06,
+      "loss": 0.0182,
+      "step": 4153
+    },
+    {
+      "epoch": 244.35,
+      "learning_rate": 9.762641898864808e-06,
+      "loss": 0.0193,
+      "step": 4154
+    },
+    {
+      "epoch": 244.41,
+      "learning_rate": 9.752321981424149e-06,
+      "loss": 0.0216,
+      "step": 4155
+    },
+    {
+      "epoch": 244.47,
+      "learning_rate": 9.742002063983488e-06,
+      "loss": 0.0203,
+      "step": 4156
+    },
+    {
+      "epoch": 244.53,
+      "learning_rate": 9.731682146542828e-06,
+      "loss": 0.0163,
+      "step": 4157
+    },
+    {
+      "epoch": 244.59,
+      "learning_rate": 9.721362229102169e-06,
+      "loss": 0.0186,
+      "step": 4158
+    },
+    {
+      "epoch": 244.65,
+      "learning_rate": 9.711042311661508e-06,
+      "loss": 0.021,
+      "step": 4159
+    },
+    {
+      "epoch": 244.71,
+      "learning_rate": 9.700722394220847e-06,
+      "loss": 0.0197,
+      "step": 4160
+    },
+    {
+      "epoch": 244.71,
+      "eval_accuracy_NAT": 0.9940509991223705,
+      "eval_accuracy_REST": 0.4016260162601626,
+      "eval_accuracy_concrete": 0.9574088569336303,
+      "eval_accuracy_concrete tiles": 0.9367664388404431,
+      "eval_accuracy_dark bricks": 0.859421357885267,
+      "eval_accuracy_grass": 0.8953017898844466,
+      "eval_accuracy_red bricks": 0.9265158172231986,
+      "eval_accuracy_road": 0.9815535583830919,
+      "eval_accuracy_rough concrete": 0.6265321566687919,
+      "eval_accuracy_rough red sidewalk": 0.7805593364365803,
+      "eval_accuracy_speedway bricks": 0.9889479755948599,
+      "eval_accuracy_steel": 0.9314523182415649,
+      "eval_accuracy_tiles": 0.24599351691962798,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840251918380024,
+      "eval_iou_REST": 0.39144215530903326,
+      "eval_iou_concrete": 0.8931350156842415,
+      "eval_iou_concrete tiles": 0.8747524098771953,
+      "eval_iou_dark bricks": 0.8420818192474505,
+      "eval_iou_grass": 0.842924301802143,
+      "eval_iou_red bricks": 0.9028632592989029,
+      "eval_iou_road": 0.9600514345751159,
+      "eval_iou_rough concrete": 0.619824918206738,
+      "eval_iou_rough red sidewalk": 0.6194085071847998,
+      "eval_iou_speedway bricks": 0.9799513199546414,
+      "eval_iou_steel": 0.8737099205149301,
+      "eval_iou_tiles": 0.24382919944748063,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08382995426654816,
+      "eval_mean_accuracy": 0.8097023183380029,
+      "eval_mean_iou": 0.7713845733031287,
+      "eval_overall_accuracy": 0.9802998364981959,
+      "eval_runtime": 5.5563,
+      "eval_samples_per_second": 21.237,
+      "eval_steps_per_second": 1.44,
+      "step": 4160
+    },
+    {
+      "epoch": 244.76,
+      "learning_rate": 9.690402476780186e-06,
+      "loss": 0.0204,
+      "step": 4161
+    },
+    {
+      "epoch": 244.82,
+      "learning_rate": 9.680082559339526e-06,
+      "loss": 0.0193,
+      "step": 4162
+    },
+    {
+      "epoch": 244.88,
+      "learning_rate": 9.669762641898865e-06,
+      "loss": 0.0164,
+      "step": 4163
+    },
+    {
+      "epoch": 244.94,
+      "learning_rate": 9.659442724458206e-06,
+      "loss": 0.0176,
+      "step": 4164
+    },
+    {
+      "epoch": 245.0,
+      "learning_rate": 9.649122807017545e-06,
+      "loss": 0.017,
+      "step": 4165
+    },
+    {
+      "epoch": 245.06,
+      "learning_rate": 9.638802889576883e-06,
+      "loss": 0.0194,
+      "step": 4166
+    },
+    {
+      "epoch": 245.12,
+      "learning_rate": 9.628482972136222e-06,
+      "loss": 0.0198,
+      "step": 4167
+    },
+    {
+      "epoch": 245.18,
+      "learning_rate": 9.618163054695563e-06,
+      "loss": 0.0208,
+      "step": 4168
+    },
+    {
+      "epoch": 245.24,
+      "learning_rate": 9.607843137254903e-06,
+      "loss": 0.0181,
+      "step": 4169
+    },
+    {
+      "epoch": 245.29,
+      "learning_rate": 9.597523219814242e-06,
+      "loss": 0.0211,
+      "step": 4170
+    },
+    {
+      "epoch": 245.35,
+      "learning_rate": 9.587203302373581e-06,
+      "loss": 0.0181,
+      "step": 4171
+    },
+    {
+      "epoch": 245.41,
+      "learning_rate": 9.57688338493292e-06,
+      "loss": 0.0197,
+      "step": 4172
+    },
+    {
+      "epoch": 245.47,
+      "learning_rate": 9.56656346749226e-06,
+      "loss": 0.0181,
+      "step": 4173
+    },
+    {
+      "epoch": 245.53,
+      "learning_rate": 9.5562435500516e-06,
+      "loss": 0.0218,
+      "step": 4174
+    },
+    {
+      "epoch": 245.59,
+      "learning_rate": 9.54592363261094e-06,
+      "loss": 0.0186,
+      "step": 4175
+    },
+    {
+      "epoch": 245.65,
+      "learning_rate": 9.535603715170279e-06,
+      "loss": 0.0186,
+      "step": 4176
+    },
+    {
+      "epoch": 245.71,
+      "learning_rate": 9.52528379772962e-06,
+      "loss": 0.0184,
+      "step": 4177
+    },
+    {
+      "epoch": 245.76,
+      "learning_rate": 9.514963880288958e-06,
+      "loss": 0.0261,
+      "step": 4178
+    },
+    {
+      "epoch": 245.82,
+      "learning_rate": 9.504643962848297e-06,
+      "loss": 0.0178,
+      "step": 4179
+    },
+    {
+      "epoch": 245.88,
+      "learning_rate": 9.494324045407638e-06,
+      "loss": 0.0178,
+      "step": 4180
+    },
+    {
+      "epoch": 245.88,
+      "eval_accuracy_NAT": 0.9942444345138984,
+      "eval_accuracy_REST": 0.4211382113821138,
+      "eval_accuracy_concrete": 0.9548710536011582,
+      "eval_accuracy_concrete tiles": 0.947489983502239,
+      "eval_accuracy_dark bricks": 0.8525591945332389,
+      "eval_accuracy_grass": 0.8956604102086622,
+      "eval_accuracy_red bricks": 0.9281085237258347,
+      "eval_accuracy_road": 0.9833577472436152,
+      "eval_accuracy_rough concrete": 0.62567408934626,
+      "eval_accuracy_rough red sidewalk": 0.7644891298626099,
+      "eval_accuracy_speedway bricks": 0.9886806459506756,
+      "eval_accuracy_steel": 0.9299929530654165,
+      "eval_accuracy_tiles": 0.25050743736556697,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838478215871674,
+      "eval_iou_REST": 0.4098101265822785,
+      "eval_iou_concrete": 0.8933120990640164,
+      "eval_iou_concrete tiles": 0.8740135226210405,
+      "eval_iou_dark bricks": 0.8386770292471438,
+      "eval_iou_grass": 0.8413625935640885,
+      "eval_iou_red bricks": 0.9038831835686778,
+      "eval_iou_road": 0.9585448097272455,
+      "eval_iou_rough concrete": 0.6200286984711864,
+      "eval_iou_rough red sidewalk": 0.6099208658559266,
+      "eval_iou_speedway bricks": 0.9800612011862444,
+      "eval_iou_steel": 0.8722755963276949,
+      "eval_iou_tiles": 0.24763416387158602,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08508313447237015,
+      "eval_mean_accuracy": 0.8105210626385607,
+      "eval_mean_iou": 0.7717978239749459,
+      "eval_overall_accuracy": 0.9801540051476431,
+      "eval_runtime": 5.4987,
+      "eval_samples_per_second": 21.46,
+      "eval_steps_per_second": 1.455,
+      "step": 4180
+    },
+    {
+      "epoch": 245.94,
+      "learning_rate": 9.484004127966977e-06,
+      "loss": 0.0194,
+      "step": 4181
+    },
+    {
+      "epoch": 246.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0194,
+      "step": 4182
+    },
+    {
+      "epoch": 246.06,
+      "learning_rate": 9.463364293085656e-06,
+      "loss": 0.0188,
+      "step": 4183
+    },
+    {
+      "epoch": 246.12,
+      "learning_rate": 9.453044375644995e-06,
+      "loss": 0.0205,
+      "step": 4184
+    },
+    {
+      "epoch": 246.18,
+      "learning_rate": 9.442724458204334e-06,
+      "loss": 0.0173,
+      "step": 4185
+    },
+    {
+      "epoch": 246.24,
+      "learning_rate": 9.432404540763674e-06,
+      "loss": 0.0216,
+      "step": 4186
+    },
+    {
+      "epoch": 246.29,
+      "learning_rate": 9.422084623323015e-06,
+      "loss": 0.0182,
+      "step": 4187
+    },
+    {
+      "epoch": 246.35,
+      "learning_rate": 9.411764705882354e-06,
+      "loss": 0.0152,
+      "step": 4188
+    },
+    {
+      "epoch": 246.41,
+      "learning_rate": 9.401444788441693e-06,
+      "loss": 0.0218,
+      "step": 4189
+    },
+    {
+      "epoch": 246.47,
+      "learning_rate": 9.391124871001032e-06,
+      "loss": 0.0239,
+      "step": 4190
+    },
+    {
+      "epoch": 246.53,
+      "learning_rate": 9.380804953560372e-06,
+      "loss": 0.0204,
+      "step": 4191
+    },
+    {
+      "epoch": 246.59,
+      "learning_rate": 9.370485036119711e-06,
+      "loss": 0.02,
+      "step": 4192
+    },
+    {
+      "epoch": 246.65,
+      "learning_rate": 9.360165118679052e-06,
+      "loss": 0.0192,
+      "step": 4193
+    },
+    {
+      "epoch": 246.71,
+      "learning_rate": 9.34984520123839e-06,
+      "loss": 0.0203,
+      "step": 4194
+    },
+    {
+      "epoch": 246.76,
+      "learning_rate": 9.33952528379773e-06,
+      "loss": 0.017,
+      "step": 4195
+    },
+    {
+      "epoch": 246.82,
+      "learning_rate": 9.32920536635707e-06,
+      "loss": 0.0197,
+      "step": 4196
+    },
+    {
+      "epoch": 246.88,
+      "learning_rate": 9.318885448916409e-06,
+      "loss": 0.019,
+      "step": 4197
+    },
+    {
+      "epoch": 246.94,
+      "learning_rate": 9.30856553147575e-06,
+      "loss": 0.0183,
+      "step": 4198
+    },
+    {
+      "epoch": 247.0,
+      "learning_rate": 9.298245614035088e-06,
+      "loss": 0.0173,
+      "step": 4199
+    },
+    {
+      "epoch": 247.06,
+      "learning_rate": 9.287925696594429e-06,
+      "loss": 0.0181,
+      "step": 4200
+    },
+    {
+      "epoch": 247.06,
+      "eval_accuracy_NAT": 0.9934387232746607,
+      "eval_accuracy_REST": 0.4796747967479675,
+      "eval_accuracy_concrete": 0.9564648246698945,
+      "eval_accuracy_concrete tiles": 0.9472543012019797,
+      "eval_accuracy_dark bricks": 0.8639243535019046,
+      "eval_accuracy_grass": 0.9013765721574372,
+      "eval_accuracy_red bricks": 0.9277789982425307,
+      "eval_accuracy_road": 0.9819543704818456,
+      "eval_accuracy_rough concrete": 0.6289871826193697,
+      "eval_accuracy_rough red sidewalk": 0.7833868505553508,
+      "eval_accuracy_speedway bricks": 0.9902949057251728,
+      "eval_accuracy_steel": 0.9272316542304098,
+      "eval_accuracy_tiles": 0.2492047623375443,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840704706523947,
+      "eval_iou_REST": 0.4631083202511774,
+      "eval_iou_concrete": 0.8939710788971686,
+      "eval_iou_concrete tiles": 0.8791478005993394,
+      "eval_iou_dark bricks": 0.8450319222632042,
+      "eval_iou_grass": 0.8422619942316636,
+      "eval_iou_red bricks": 0.9041909757533587,
+      "eval_iou_road": 0.9605624628503826,
+      "eval_iou_rough concrete": 0.6226206977910166,
+      "eval_iou_rough red sidewalk": 0.6199541762132889,
+      "eval_iou_speedway bricks": 0.9799172230237201,
+      "eval_iou_steel": 0.8731319225637635,
+      "eval_iou_tiles": 0.24648668084978875,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08187825232744217,
+      "eval_mean_accuracy": 0.8177670996727746,
+      "eval_mean_iou": 0.7780350558415591,
+      "eval_overall_accuracy": 0.9803601281117584,
+      "eval_runtime": 5.3743,
+      "eval_samples_per_second": 21.956,
+      "eval_steps_per_second": 1.489,
+      "step": 4200
+    },
+    {
+      "epoch": 247.12,
+      "learning_rate": 9.277605779153768e-06,
+      "loss": 0.0231,
+      "step": 4201
+    },
+    {
+      "epoch": 247.18,
+      "learning_rate": 9.267285861713106e-06,
+      "loss": 0.0185,
+      "step": 4202
+    },
+    {
+      "epoch": 247.24,
+      "learning_rate": 9.256965944272445e-06,
+      "loss": 0.0174,
+      "step": 4203
+    },
+    {
+      "epoch": 247.29,
+      "learning_rate": 9.246646026831786e-06,
+      "loss": 0.018,
+      "step": 4204
+    },
+    {
+      "epoch": 247.35,
+      "learning_rate": 9.236326109391125e-06,
+      "loss": 0.0196,
+      "step": 4205
+    },
+    {
+      "epoch": 247.41,
+      "learning_rate": 9.226006191950465e-06,
+      "loss": 0.0209,
+      "step": 4206
+    },
+    {
+      "epoch": 247.47,
+      "learning_rate": 9.215686274509804e-06,
+      "loss": 0.0207,
+      "step": 4207
+    },
+    {
+      "epoch": 247.53,
+      "learning_rate": 9.205366357069143e-06,
+      "loss": 0.0196,
+      "step": 4208
+    },
+    {
+      "epoch": 247.59,
+      "learning_rate": 9.195046439628484e-06,
+      "loss": 0.0165,
+      "step": 4209
+    },
+    {
+      "epoch": 247.65,
+      "learning_rate": 9.184726522187823e-06,
+      "loss": 0.0178,
+      "step": 4210
+    },
+    {
+      "epoch": 247.71,
+      "learning_rate": 9.174406604747163e-06,
+      "loss": 0.0201,
+      "step": 4211
+    },
+    {
+      "epoch": 247.76,
+      "learning_rate": 9.164086687306502e-06,
+      "loss": 0.0204,
+      "step": 4212
+    },
+    {
+      "epoch": 247.82,
+      "learning_rate": 9.153766769865841e-06,
+      "loss": 0.0176,
+      "step": 4213
+    },
+    {
+      "epoch": 247.88,
+      "learning_rate": 9.14344685242518e-06,
+      "loss": 0.0198,
+      "step": 4214
+    },
+    {
+      "epoch": 247.94,
+      "learning_rate": 9.13312693498452e-06,
+      "loss": 0.0187,
+      "step": 4215
+    },
+    {
+      "epoch": 248.0,
+      "learning_rate": 9.122807017543861e-06,
+      "loss": 0.0181,
+      "step": 4216
+    },
+    {
+      "epoch": 248.06,
+      "learning_rate": 9.1124871001032e-06,
+      "loss": 0.0179,
+      "step": 4217
+    },
+    {
+      "epoch": 248.12,
+      "learning_rate": 9.10216718266254e-06,
+      "loss": 0.0165,
+      "step": 4218
+    },
+    {
+      "epoch": 248.18,
+      "learning_rate": 9.09184726522188e-06,
+      "loss": 0.0198,
+      "step": 4219
+    },
+    {
+      "epoch": 248.24,
+      "learning_rate": 9.081527347781218e-06,
+      "loss": 0.0222,
+      "step": 4220
+    },
+    {
+      "epoch": 248.24,
+      "eval_accuracy_NAT": 0.9938033824129814,
+      "eval_accuracy_REST": 0.40487804878048783,
+      "eval_accuracy_concrete": 0.9569647413504477,
+      "eval_accuracy_concrete tiles": 0.9410794249351874,
+      "eval_accuracy_dark bricks": 0.8531286488141953,
+      "eval_accuracy_grass": 0.9024732501409879,
+      "eval_accuracy_red bricks": 0.9344244288224957,
+      "eval_accuracy_road": 0.9806771159271506,
+      "eval_accuracy_rough concrete": 0.6241963067352326,
+      "eval_accuracy_rough red sidewalk": 0.7708209566924101,
+      "eval_accuracy_speedway bricks": 0.9899754810734039,
+      "eval_accuracy_steel": 0.9230384782619561,
+      "eval_accuracy_tiles": 0.24199460753127935,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841098455567124,
+      "eval_iou_REST": 0.3933649289099526,
+      "eval_iou_concrete": 0.8930063138657885,
+      "eval_iou_concrete tiles": 0.8778332270758679,
+      "eval_iou_dark bricks": 0.8377189684986092,
+      "eval_iou_grass": 0.8429968065858753,
+      "eval_iou_red bricks": 0.9059637912673056,
+      "eval_iou_road": 0.9607177934281149,
+      "eval_iou_rough concrete": 0.6193322572825579,
+      "eval_iou_rough red sidewalk": 0.6135626294429612,
+      "eval_iou_speedway bricks": 0.9800175952104113,
+      "eval_iou_steel": 0.8722972143241737,
+      "eval_iou_tiles": 0.23927630002396358,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08373229205608368,
+      "eval_mean_accuracy": 0.8090349901137088,
+      "eval_mean_iou": 0.7707844362670997,
+      "eval_overall_accuracy": 0.9803000951217393,
+      "eval_runtime": 5.2431,
+      "eval_samples_per_second": 22.506,
+      "eval_steps_per_second": 1.526,
+      "step": 4220
+    },
+    {
+      "epoch": 248.29,
+      "learning_rate": 9.071207430340557e-06,
+      "loss": 0.0199,
+      "step": 4221
+    },
+    {
+      "epoch": 248.35,
+      "learning_rate": 9.060887512899897e-06,
+      "loss": 0.0216,
+      "step": 4222
+    },
+    {
+      "epoch": 248.41,
+      "learning_rate": 9.050567595459236e-06,
+      "loss": 0.018,
+      "step": 4223
+    },
+    {
+      "epoch": 248.47,
+      "learning_rate": 9.040247678018577e-06,
+      "loss": 0.018,
+      "step": 4224
+    },
+    {
+      "epoch": 248.53,
+      "learning_rate": 9.029927760577916e-06,
+      "loss": 0.0177,
+      "step": 4225
+    },
+    {
+      "epoch": 248.59,
+      "learning_rate": 9.019607843137255e-06,
+      "loss": 0.0205,
+      "step": 4226
+    },
+    {
+      "epoch": 248.65,
+      "learning_rate": 9.009287925696595e-06,
+      "loss": 0.0183,
+      "step": 4227
+    },
+    {
+      "epoch": 248.71,
+      "learning_rate": 8.998968008255934e-06,
+      "loss": 0.0181,
+      "step": 4228
+    },
+    {
+      "epoch": 248.76,
+      "learning_rate": 8.988648090815275e-06,
+      "loss": 0.0181,
+      "step": 4229
+    },
+    {
+      "epoch": 248.82,
+      "learning_rate": 8.978328173374614e-06,
+      "loss": 0.0159,
+      "step": 4230
+    },
+    {
+      "epoch": 248.88,
+      "learning_rate": 8.968008255933952e-06,
+      "loss": 0.0197,
+      "step": 4231
+    },
+    {
+      "epoch": 248.94,
+      "learning_rate": 8.957688338493291e-06,
+      "loss": 0.0195,
+      "step": 4232
+    },
+    {
+      "epoch": 249.0,
+      "learning_rate": 8.947368421052632e-06,
+      "loss": 0.0185,
+      "step": 4233
+    },
+    {
+      "epoch": 249.06,
+      "learning_rate": 8.937048503611972e-06,
+      "loss": 0.02,
+      "step": 4234
+    },
+    {
+      "epoch": 249.12,
+      "learning_rate": 8.926728586171311e-06,
+      "loss": 0.0208,
+      "step": 4235
+    },
+    {
+      "epoch": 249.18,
+      "learning_rate": 8.916408668730652e-06,
+      "loss": 0.0243,
+      "step": 4236
+    },
+    {
+      "epoch": 249.24,
+      "learning_rate": 8.906088751289989e-06,
+      "loss": 0.0196,
+      "step": 4237
+    },
+    {
+      "epoch": 249.29,
+      "learning_rate": 8.89576883384933e-06,
+      "loss": 0.0208,
+      "step": 4238
+    },
+    {
+      "epoch": 249.35,
+      "learning_rate": 8.885448916408668e-06,
+      "loss": 0.0187,
+      "step": 4239
+    },
+    {
+      "epoch": 249.41,
+      "learning_rate": 8.875128998968009e-06,
+      "loss": 0.0197,
+      "step": 4240
+    },
+    {
+      "epoch": 249.41,
+      "eval_accuracy_NAT": 0.993612049580057,
+      "eval_accuracy_REST": 0.37398373983739835,
+      "eval_accuracy_concrete": 0.9589193085556077,
+      "eval_accuracy_concrete tiles": 0.9473721423521093,
+      "eval_accuracy_dark bricks": 0.8534971192312846,
+      "eval_accuracy_grass": 0.9008022119020314,
+      "eval_accuracy_red bricks": 0.929261862917399,
+      "eval_accuracy_road": 0.9819399412462905,
+      "eval_accuracy_rough concrete": 0.625245055684994,
+      "eval_accuracy_rough red sidewalk": 0.7759119861030689,
+      "eval_accuracy_speedway bricks": 0.9897144919400113,
+      "eval_accuracy_steel": 0.9224612293652261,
+      "eval_accuracy_tiles": 0.24029810051804054,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984085920916299,
+      "eval_iou_REST": 0.36334913112164297,
+      "eval_iou_concrete": 0.8931721723553364,
+      "eval_iou_concrete tiles": 0.875694399058885,
+      "eval_iou_dark bricks": 0.8384038282550039,
+      "eval_iou_grass": 0.8433433850209774,
+      "eval_iou_red bricks": 0.9036530655842768,
+      "eval_iou_road": 0.9607211184771577,
+      "eval_iou_rough concrete": 0.6192890405056866,
+      "eval_iou_rough red sidewalk": 0.6238738466375631,
+      "eval_iou_speedway bricks": 0.9799991278294848,
+      "eval_iou_steel": 0.8733005126555178,
+      "eval_iou_tiles": 0.23822681403171553,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08354980498552322,
+      "eval_mean_accuracy": 0.8071553260948859,
+      "eval_mean_iou": 0.7690086432653498,
+      "eval_overall_accuracy": 0.9803465180477854,
+      "eval_runtime": 5.4082,
+      "eval_samples_per_second": 21.819,
+      "eval_steps_per_second": 1.479,
+      "step": 4240
+    },
+    {
+      "epoch": 249.47,
+      "learning_rate": 8.864809081527348e-06,
+      "loss": 0.019,
+      "step": 4241
+    },
+    {
+      "epoch": 249.53,
+      "learning_rate": 8.854489164086688e-06,
+      "loss": 0.0196,
+      "step": 4242
+    },
+    {
+      "epoch": 249.59,
+      "learning_rate": 8.844169246646027e-06,
+      "loss": 0.0183,
+      "step": 4243
+    },
+    {
+      "epoch": 249.65,
+      "learning_rate": 8.833849329205366e-06,
+      "loss": 0.0174,
+      "step": 4244
+    },
+    {
+      "epoch": 249.71,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0174,
+      "step": 4245
+    },
+    {
+      "epoch": 249.76,
+      "learning_rate": 8.813209494324046e-06,
+      "loss": 0.0184,
+      "step": 4246
+    },
+    {
+      "epoch": 249.82,
+      "learning_rate": 8.802889576883386e-06,
+      "loss": 0.0172,
+      "step": 4247
+    },
+    {
+      "epoch": 249.88,
+      "learning_rate": 8.792569659442725e-06,
+      "loss": 0.0169,
+      "step": 4248
+    },
+    {
+      "epoch": 249.94,
+      "learning_rate": 8.782249742002064e-06,
+      "loss": 0.0195,
+      "step": 4249
+    },
+    {
+      "epoch": 250.0,
+      "learning_rate": 8.771929824561403e-06,
+      "loss": 0.0194,
+      "step": 4250
+    },
+    {
+      "epoch": 250.06,
+      "learning_rate": 8.761609907120743e-06,
+      "loss": 0.021,
+      "step": 4251
+    },
+    {
+      "epoch": 250.12,
+      "learning_rate": 8.751289989680082e-06,
+      "loss": 0.0233,
+      "step": 4252
+    },
+    {
+      "epoch": 250.18,
+      "learning_rate": 8.740970072239423e-06,
+      "loss": 0.022,
+      "step": 4253
+    },
+    {
+      "epoch": 250.24,
+      "learning_rate": 8.730650154798763e-06,
+      "loss": 0.0177,
+      "step": 4254
+    },
+    {
+      "epoch": 250.29,
+      "learning_rate": 8.7203302373581e-06,
+      "loss": 0.0215,
+      "step": 4255
+    },
+    {
+      "epoch": 250.35,
+      "learning_rate": 8.710010319917441e-06,
+      "loss": 0.0171,
+      "step": 4256
+    },
+    {
+      "epoch": 250.41,
+      "learning_rate": 8.69969040247678e-06,
+      "loss": 0.018,
+      "step": 4257
+    },
+    {
+      "epoch": 250.47,
+      "learning_rate": 8.68937048503612e-06,
+      "loss": 0.0173,
+      "step": 4258
+    },
+    {
+      "epoch": 250.53,
+      "learning_rate": 8.67905056759546e-06,
+      "loss": 0.0196,
+      "step": 4259
+    },
+    {
+      "epoch": 250.59,
+      "learning_rate": 8.6687306501548e-06,
+      "loss": 0.0198,
+      "step": 4260
+    },
+    {
+      "epoch": 250.59,
+      "eval_accuracy_NAT": 0.9939015018144811,
+      "eval_accuracy_REST": 0.3853658536585366,
+      "eval_accuracy_concrete": 0.959620644265711,
+      "eval_accuracy_concrete tiles": 0.9399481498939429,
+      "eval_accuracy_dark bricks": 0.8663840131692284,
+      "eval_accuracy_grass": 0.8952469086738807,
+      "eval_accuracy_red bricks": 0.9290421792618629,
+      "eval_accuracy_road": 0.9808091167116734,
+      "eval_accuracy_rough concrete": 0.6237791906756684,
+      "eval_accuracy_rough red sidewalk": 0.7739342302168012,
+      "eval_accuracy_speedway bricks": 0.9886417460601437,
+      "eval_accuracy_steel": 0.9304627487043137,
+      "eval_accuracy_tiles": 0.24366081977642462,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840321765879912,
+      "eval_iou_REST": 0.36517719568567025,
+      "eval_iou_concrete": 0.8926642414658149,
+      "eval_iou_concrete tiles": 0.874739543350953,
+      "eval_iou_dark bricks": 0.8456881813850509,
+      "eval_iou_grass": 0.8428298326851091,
+      "eval_iou_red bricks": 0.9029571901355824,
+      "eval_iou_road": 0.9608782972271408,
+      "eval_iou_rough concrete": 0.6167414896249426,
+      "eval_iou_rough red sidewalk": 0.6180161894696388,
+      "eval_iou_speedway bricks": 0.9798551925840124,
+      "eval_iou_steel": 0.8736048838742524,
+      "eval_iou_tiles": 0.24147351987510507,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08478410542011261,
+      "eval_mean_accuracy": 0.8085228540678977,
+      "eval_mean_iou": 0.7691275333808665,
+      "eval_overall_accuracy": 0.9802864527298232,
+      "eval_runtime": 5.6242,
+      "eval_samples_per_second": 20.981,
+      "eval_steps_per_second": 1.422,
+      "step": 4260
+    },
+    {
+      "epoch": 250.65,
+      "learning_rate": 8.658410732714139e-06,
+      "loss": 0.0141,
+      "step": 4261
+    },
+    {
+      "epoch": 250.71,
+      "learning_rate": 8.648090815273478e-06,
+      "loss": 0.016,
+      "step": 4262
+    },
+    {
+      "epoch": 250.76,
+      "learning_rate": 8.637770897832818e-06,
+      "loss": 0.0188,
+      "step": 4263
+    },
+    {
+      "epoch": 250.82,
+      "learning_rate": 8.627450980392157e-06,
+      "loss": 0.0181,
+      "step": 4264
+    },
+    {
+      "epoch": 250.88,
+      "learning_rate": 8.617131062951498e-06,
+      "loss": 0.0185,
+      "step": 4265
+    },
+    {
+      "epoch": 250.94,
+      "learning_rate": 8.606811145510837e-06,
+      "loss": 0.0205,
+      "step": 4266
+    },
+    {
+      "epoch": 251.0,
+      "learning_rate": 8.596491228070176e-06,
+      "loss": 0.0201,
+      "step": 4267
+    },
+    {
+      "epoch": 251.06,
+      "learning_rate": 8.586171310629514e-06,
+      "loss": 0.02,
+      "step": 4268
+    },
+    {
+      "epoch": 251.12,
+      "learning_rate": 8.575851393188855e-06,
+      "loss": 0.0196,
+      "step": 4269
+    },
+    {
+      "epoch": 251.18,
+      "learning_rate": 8.565531475748194e-06,
+      "loss": 0.0188,
+      "step": 4270
+    },
+    {
+      "epoch": 251.24,
+      "learning_rate": 8.555211558307534e-06,
+      "loss": 0.0226,
+      "step": 4271
+    },
+    {
+      "epoch": 251.29,
+      "learning_rate": 8.544891640866875e-06,
+      "loss": 0.0177,
+      "step": 4272
+    },
+    {
+      "epoch": 251.35,
+      "learning_rate": 8.534571723426212e-06,
+      "loss": 0.0187,
+      "step": 4273
+    },
+    {
+      "epoch": 251.41,
+      "learning_rate": 8.524251805985553e-06,
+      "loss": 0.0152,
+      "step": 4274
+    },
+    {
+      "epoch": 251.47,
+      "learning_rate": 8.513931888544892e-06,
+      "loss": 0.0207,
+      "step": 4275
+    },
+    {
+      "epoch": 251.53,
+      "learning_rate": 8.503611971104232e-06,
+      "loss": 0.0167,
+      "step": 4276
+    },
+    {
+      "epoch": 251.59,
+      "learning_rate": 8.493292053663571e-06,
+      "loss": 0.0167,
+      "step": 4277
+    },
+    {
+      "epoch": 251.65,
+      "learning_rate": 8.482972136222912e-06,
+      "loss": 0.0218,
+      "step": 4278
+    },
+    {
+      "epoch": 251.71,
+      "learning_rate": 8.472652218782249e-06,
+      "loss": 0.0184,
+      "step": 4279
+    },
+    {
+      "epoch": 251.76,
+      "learning_rate": 8.46233230134159e-06,
+      "loss": 0.0194,
+      "step": 4280
+    },
+    {
+      "epoch": 251.76,
+      "eval_accuracy_NAT": 0.9936363098716366,
+      "eval_accuracy_REST": 0.45528455284552843,
+      "eval_accuracy_concrete": 0.9587843922037459,
+      "eval_accuracy_concrete tiles": 0.932712703275984,
+      "eval_accuracy_dark bricks": 0.8496449285071684,
+      "eval_accuracy_grass": 0.902160994977423,
+      "eval_accuracy_red bricks": 0.929261862917399,
+      "eval_accuracy_road": 0.980904242783111,
+      "eval_accuracy_rough concrete": 0.6247206812101133,
+      "eval_accuracy_rough red sidewalk": 0.758006903345641,
+      "eval_accuracy_speedway bricks": 0.9896995831713933,
+      "eval_accuracy_steel": 0.9295756345989494,
+      "eval_accuracy_tiles": 0.2382380562876791,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842116890393903,
+      "eval_iou_REST": 0.4354587869362364,
+      "eval_iou_concrete": 0.8918181679211832,
+      "eval_iou_concrete tiles": 0.8755724684174429,
+      "eval_iou_dark bricks": 0.8360384794677289,
+      "eval_iou_grass": 0.8449635623870603,
+      "eval_iou_red bricks": 0.9026407041877834,
+      "eval_iou_road": 0.9606279802606065,
+      "eval_iou_rough concrete": 0.6184885847442628,
+      "eval_iou_rough red sidewalk": 0.6110770946875776,
+      "eval_iou_speedway bricks": 0.9799366434608212,
+      "eval_iou_steel": 0.8738134556072077,
+      "eval_iou_tiles": 0.23607108549471661,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08346042782068253,
+      "eval_mean_accuracy": 0.8109716035381364,
+      "eval_mean_iou": 0.7731322078932321,
+      "eval_overall_accuracy": 0.980296118784759,
+      "eval_runtime": 5.5195,
+      "eval_samples_per_second": 21.379,
+      "eval_steps_per_second": 1.449,
+      "step": 4280
+    },
+    {
+      "epoch": 251.82,
+      "learning_rate": 8.452012383900928e-06,
+      "loss": 0.0208,
+      "step": 4281
+    },
+    {
+      "epoch": 251.88,
+      "learning_rate": 8.441692466460269e-06,
+      "loss": 0.0196,
+      "step": 4282
+    },
+    {
+      "epoch": 251.94,
+      "learning_rate": 8.43137254901961e-06,
+      "loss": 0.0185,
+      "step": 4283
+    },
+    {
+      "epoch": 252.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0193,
+      "step": 4284
+    },
+    {
+      "epoch": 252.06,
+      "learning_rate": 8.410732714138287e-06,
+      "loss": 0.0222,
+      "step": 4285
+    },
+    {
+      "epoch": 252.12,
+      "learning_rate": 8.400412796697626e-06,
+      "loss": 0.0192,
+      "step": 4286
+    },
+    {
+      "epoch": 252.18,
+      "learning_rate": 8.390092879256967e-06,
+      "loss": 0.0194,
+      "step": 4287
+    },
+    {
+      "epoch": 252.24,
+      "learning_rate": 8.379772961816305e-06,
+      "loss": 0.0172,
+      "step": 4288
+    },
+    {
+      "epoch": 252.29,
+      "learning_rate": 8.369453044375646e-06,
+      "loss": 0.0201,
+      "step": 4289
+    },
+    {
+      "epoch": 252.35,
+      "learning_rate": 8.359133126934985e-06,
+      "loss": 0.0162,
+      "step": 4290
+    },
+    {
+      "epoch": 252.41,
+      "learning_rate": 8.348813209494324e-06,
+      "loss": 0.0207,
+      "step": 4291
+    },
+    {
+      "epoch": 252.47,
+      "learning_rate": 8.338493292053664e-06,
+      "loss": 0.0192,
+      "step": 4292
+    },
+    {
+      "epoch": 252.53,
+      "learning_rate": 8.328173374613003e-06,
+      "loss": 0.0186,
+      "step": 4293
+    },
+    {
+      "epoch": 252.59,
+      "learning_rate": 8.317853457172344e-06,
+      "loss": 0.0182,
+      "step": 4294
+    },
+    {
+      "epoch": 252.65,
+      "learning_rate": 8.307533539731683e-06,
+      "loss": 0.0169,
+      "step": 4295
+    },
+    {
+      "epoch": 252.71,
+      "learning_rate": 8.297213622291023e-06,
+      "loss": 0.0197,
+      "step": 4296
+    },
+    {
+      "epoch": 252.76,
+      "learning_rate": 8.28689370485036e-06,
+      "loss": 0.0177,
+      "step": 4297
+    },
+    {
+      "epoch": 252.82,
+      "learning_rate": 8.276573787409701e-06,
+      "loss": 0.0181,
+      "step": 4298
+    },
+    {
+      "epoch": 252.88,
+      "learning_rate": 8.26625386996904e-06,
+      "loss": 0.0195,
+      "step": 4299
+    },
+    {
+      "epoch": 252.94,
+      "learning_rate": 8.25593395252838e-06,
+      "loss": 0.0181,
+      "step": 4300
+    },
+    {
+      "epoch": 252.94,
+      "eval_accuracy_NAT": 0.9935118276199538,
+      "eval_accuracy_REST": 0.46016260162601624,
+      "eval_accuracy_concrete": 0.9566941442481299,
+      "eval_accuracy_concrete tiles": 0.9443318406787651,
+      "eval_accuracy_dark bricks": 0.8639674214727332,
+      "eval_accuracy_grass": 0.9062496451645868,
+      "eval_accuracy_red bricks": 0.9366212653778558,
+      "eval_accuracy_road": 0.9818624509071981,
+      "eval_accuracy_rough concrete": 0.6251258796679756,
+      "eval_accuracy_rough red sidewalk": 0.7717383947841389,
+      "eval_accuracy_speedway bricks": 0.9900116391214314,
+      "eval_accuracy_steel": 0.9281162694228011,
+      "eval_accuracy_tiles": 0.25111333272743797,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984221043397685,
+      "eval_iou_REST": 0.4387596899224806,
+      "eval_iou_concrete": 0.8939273924092583,
+      "eval_iou_concrete tiles": 0.8768766139974614,
+      "eval_iou_dark bricks": 0.8460085844954266,
+      "eval_iou_grass": 0.8454422914465339,
+      "eval_iou_red bricks": 0.9042897290418368,
+      "eval_iou_road": 0.9609834631896963,
+      "eval_iou_rough concrete": 0.6190102491783546,
+      "eval_iou_rough red sidewalk": 0.6160370732761467,
+      "eval_iou_speedway bricks": 0.9800422504174404,
+      "eval_iou_steel": 0.8731841382018061,
+      "eval_iou_tiles": 0.2486575671216439,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08339465409517288,
+      "eval_mean_accuracy": 0.816115900986079,
+      "eval_mean_iou": 0.7759569296996747,
+      "eval_overall_accuracy": 0.9804744073900126,
+      "eval_runtime": 5.3597,
+      "eval_samples_per_second": 22.016,
+      "eval_steps_per_second": 1.493,
+      "step": 4300
+    },
+    {
+      "epoch": 253.0,
+      "learning_rate": 8.245614035087721e-06,
+      "loss": 0.0263,
+      "step": 4301
+    },
+    {
+      "epoch": 253.06,
+      "learning_rate": 8.23529411764706e-06,
+      "loss": 0.0172,
+      "step": 4302
+    },
+    {
+      "epoch": 253.12,
+      "learning_rate": 8.224974200206399e-06,
+      "loss": 0.0204,
+      "step": 4303
+    },
+    {
+      "epoch": 253.18,
+      "learning_rate": 8.214654282765738e-06,
+      "loss": 0.02,
+      "step": 4304
+    },
+    {
+      "epoch": 253.24,
+      "learning_rate": 8.204334365325078e-06,
+      "loss": 0.0191,
+      "step": 4305
+    },
+    {
+      "epoch": 253.29,
+      "learning_rate": 8.194014447884417e-06,
+      "loss": 0.0197,
+      "step": 4306
+    },
+    {
+      "epoch": 253.35,
+      "learning_rate": 8.183694530443758e-06,
+      "loss": 0.0199,
+      "step": 4307
+    },
+    {
+      "epoch": 253.41,
+      "learning_rate": 8.173374613003096e-06,
+      "loss": 0.0196,
+      "step": 4308
+    },
+    {
+      "epoch": 253.47,
+      "learning_rate": 8.163054695562435e-06,
+      "loss": 0.0177,
+      "step": 4309
+    },
+    {
+      "epoch": 253.53,
+      "learning_rate": 8.152734778121776e-06,
+      "loss": 0.0183,
+      "step": 4310
+    },
+    {
+      "epoch": 253.59,
+      "learning_rate": 8.142414860681115e-06,
+      "loss": 0.0195,
+      "step": 4311
+    },
+    {
+      "epoch": 253.65,
+      "learning_rate": 8.132094943240455e-06,
+      "loss": 0.0189,
+      "step": 4312
+    },
+    {
+      "epoch": 253.71,
+      "learning_rate": 8.121775025799794e-06,
+      "loss": 0.0192,
+      "step": 4313
+    },
+    {
+      "epoch": 253.76,
+      "learning_rate": 8.111455108359135e-06,
+      "loss": 0.0194,
+      "step": 4314
+    },
+    {
+      "epoch": 253.82,
+      "learning_rate": 8.101135190918472e-06,
+      "loss": 0.0208,
+      "step": 4315
+    },
+    {
+      "epoch": 253.88,
+      "learning_rate": 8.090815273477812e-06,
+      "loss": 0.0173,
+      "step": 4316
+    },
+    {
+      "epoch": 253.94,
+      "learning_rate": 8.080495356037151e-06,
+      "loss": 0.0174,
+      "step": 4317
+    },
+    {
+      "epoch": 254.0,
+      "learning_rate": 8.070175438596492e-06,
+      "loss": 0.0187,
+      "step": 4318
+    },
+    {
+      "epoch": 254.06,
+      "learning_rate": 8.05985552115583e-06,
+      "loss": 0.0196,
+      "step": 4319
+    },
+    {
+      "epoch": 254.12,
+      "learning_rate": 8.049535603715171e-06,
+      "loss": 0.024,
+      "step": 4320
+    },
+    {
+      "epoch": 254.12,
+      "eval_accuracy_NAT": 0.9944025577032383,
+      "eval_accuracy_REST": 0.39186991869918697,
+      "eval_accuracy_concrete": 0.9566219085809857,
+      "eval_accuracy_concrete tiles": 0.9334904548668395,
+      "eval_accuracy_dark bricks": 0.8589523955362441,
+      "eval_accuracy_grass": 0.8939590926811175,
+      "eval_accuracy_red bricks": 0.9316783831282952,
+      "eval_accuracy_road": 0.980429146842055,
+      "eval_accuracy_rough concrete": 0.6241188423241707,
+      "eval_accuracy_rough red sidewalk": 0.7724227133607562,
+      "eval_accuracy_speedway bricks": 0.9890727294288125,
+      "eval_accuracy_steel": 0.9282836965919825,
+      "eval_accuracy_tiles": 0.25302190311733164,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9839375226180234,
+      "eval_iou_REST": 0.3813291139240506,
+      "eval_iou_concrete": 0.8933797297523998,
+      "eval_iou_concrete tiles": 0.8776812621875554,
+      "eval_iou_dark bricks": 0.8436159063030206,
+      "eval_iou_grass": 0.8412082628439141,
+      "eval_iou_red bricks": 0.9046501706484642,
+      "eval_iou_road": 0.9602471357409388,
+      "eval_iou_rough concrete": 0.6196400681527758,
+      "eval_iou_rough red sidewalk": 0.6129188173165856,
+      "eval_iou_speedway bricks": 0.9800481455376062,
+      "eval_iou_steel": 0.8729496639563848,
+      "eval_iou_tiles": 0.2506527415143603,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08468490839004517,
+      "eval_mean_accuracy": 0.8083325956046935,
+      "eval_mean_iou": 0.7709429646535446,
+      "eval_overall_accuracy": 0.9802673469155522,
+      "eval_runtime": 5.6041,
+      "eval_samples_per_second": 21.056,
+      "eval_steps_per_second": 1.428,
+      "step": 4320
+    },
+    {
+      "epoch": 254.18,
+      "learning_rate": 8.03921568627451e-06,
+      "loss": 0.0174,
+      "step": 4321
+    },
+    {
+      "epoch": 254.24,
+      "learning_rate": 8.028895768833849e-06,
+      "loss": 0.0185,
+      "step": 4322
+    },
+    {
+      "epoch": 254.29,
+      "learning_rate": 8.01857585139319e-06,
+      "loss": 0.0198,
+      "step": 4323
+    },
+    {
+      "epoch": 254.35,
+      "learning_rate": 8.008255933952529e-06,
+      "loss": 0.0166,
+      "step": 4324
+    },
+    {
+      "epoch": 254.41,
+      "learning_rate": 7.997936016511869e-06,
+      "loss": 0.0197,
+      "step": 4325
+    },
+    {
+      "epoch": 254.47,
+      "learning_rate": 7.987616099071208e-06,
+      "loss": 0.0178,
+      "step": 4326
+    },
+    {
+      "epoch": 254.53,
+      "learning_rate": 7.977296181630547e-06,
+      "loss": 0.0182,
+      "step": 4327
+    },
+    {
+      "epoch": 254.59,
+      "learning_rate": 7.966976264189886e-06,
+      "loss": 0.0189,
+      "step": 4328
+    },
+    {
+      "epoch": 254.65,
+      "learning_rate": 7.956656346749226e-06,
+      "loss": 0.0188,
+      "step": 4329
+    },
+    {
+      "epoch": 254.71,
+      "learning_rate": 7.946336429308567e-06,
+      "loss": 0.0228,
+      "step": 4330
+    },
+    {
+      "epoch": 254.76,
+      "learning_rate": 7.936016511867906e-06,
+      "loss": 0.0199,
+      "step": 4331
+    },
+    {
+      "epoch": 254.82,
+      "learning_rate": 7.925696594427245e-06,
+      "loss": 0.0177,
+      "step": 4332
+    },
+    {
+      "epoch": 254.88,
+      "learning_rate": 7.915376676986583e-06,
+      "loss": 0.0197,
+      "step": 4333
+    },
+    {
+      "epoch": 254.94,
+      "learning_rate": 7.905056759545924e-06,
+      "loss": 0.0185,
+      "step": 4334
+    },
+    {
+      "epoch": 255.0,
+      "learning_rate": 7.894736842105263e-06,
+      "loss": 0.018,
+      "step": 4335
+    },
+    {
+      "epoch": 255.06,
+      "learning_rate": 7.884416924664603e-06,
+      "loss": 0.0197,
+      "step": 4336
+    },
+    {
+      "epoch": 255.12,
+      "learning_rate": 7.874097007223942e-06,
+      "loss": 0.0199,
+      "step": 4337
+    },
+    {
+      "epoch": 255.18,
+      "learning_rate": 7.863777089783283e-06,
+      "loss": 0.0156,
+      "step": 4338
+    },
+    {
+      "epoch": 255.24,
+      "learning_rate": 7.853457172342622e-06,
+      "loss": 0.0194,
+      "step": 4339
+    },
+    {
+      "epoch": 255.29,
+      "learning_rate": 7.84313725490196e-06,
+      "loss": 0.0186,
+      "step": 4340
+    },
+    {
+      "epoch": 255.29,
+      "eval_accuracy_NAT": 0.9941673406984344,
+      "eval_accuracy_REST": 0.44552845528455287,
+      "eval_accuracy_concrete": 0.9557378816068881,
+      "eval_accuracy_concrete tiles": 0.9382983737921282,
+      "eval_accuracy_dark bricks": 0.86347453247325,
+      "eval_accuracy_grass": 0.9018506322693949,
+      "eval_accuracy_red bricks": 0.937335237258348,
+      "eval_accuracy_road": 0.981271921081701,
+      "eval_accuracy_rough concrete": 0.6262163402236934,
+      "eval_accuracy_rough red sidewalk": 0.7774911828183397,
+      "eval_accuracy_speedway bricks": 0.9894629278902276,
+      "eval_accuracy_steel": 0.9279188552979454,
+      "eval_accuracy_tiles": 0.2573540549547093,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841580075953654,
+      "eval_iou_REST": 0.428125,
+      "eval_iou_concrete": 0.8946662282708756,
+      "eval_iou_concrete tiles": 0.8749120956399438,
+      "eval_iou_dark bricks": 0.8462516121467933,
+      "eval_iou_grass": 0.8457146862801992,
+      "eval_iou_red bricks": 0.9051230377598642,
+      "eval_iou_road": 0.9604913989490946,
+      "eval_iou_rough concrete": 0.6209327196346168,
+      "eval_iou_rough red sidewalk": 0.6189016725130796,
+      "eval_iou_speedway bricks": 0.9801095909292211,
+      "eval_iou_steel": 0.8730948831653742,
+      "eval_iou_tiles": 0.2545928612101777,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08295485377311707,
+      "eval_mean_accuracy": 0.8150852104345854,
+      "eval_mean_iou": 0.7759287533918927,
+      "eval_overall_accuracy": 0.9805203453969147,
+      "eval_runtime": 5.3863,
+      "eval_samples_per_second": 21.908,
+      "eval_steps_per_second": 1.485,
+      "step": 4340
+    },
+    {
+      "epoch": 255.35,
+      "learning_rate": 7.832817337461301e-06,
+      "loss": 0.0233,
+      "step": 4341
+    },
+    {
+      "epoch": 255.41,
+      "learning_rate": 7.82249742002064e-06,
+      "loss": 0.0181,
+      "step": 4342
+    },
+    {
+      "epoch": 255.47,
+      "learning_rate": 7.81217750257998e-06,
+      "loss": 0.0174,
+      "step": 4343
+    },
+    {
+      "epoch": 255.53,
+      "learning_rate": 7.80185758513932e-06,
+      "loss": 0.0152,
+      "step": 4344
+    },
+    {
+      "epoch": 255.59,
+      "learning_rate": 7.791537667698658e-06,
+      "loss": 0.0192,
+      "step": 4345
+    },
+    {
+      "epoch": 255.65,
+      "learning_rate": 7.781217750257997e-06,
+      "loss": 0.0179,
+      "step": 4346
+    },
+    {
+      "epoch": 255.71,
+      "learning_rate": 7.770897832817338e-06,
+      "loss": 0.0171,
+      "step": 4347
+    },
+    {
+      "epoch": 255.76,
+      "learning_rate": 7.760577915376677e-06,
+      "loss": 0.0181,
+      "step": 4348
+    },
+    {
+      "epoch": 255.82,
+      "learning_rate": 7.750257997936017e-06,
+      "loss": 0.0206,
+      "step": 4349
+    },
+    {
+      "epoch": 255.88,
+      "learning_rate": 7.739938080495356e-06,
+      "loss": 0.0185,
+      "step": 4350
+    },
+    {
+      "epoch": 255.94,
+      "learning_rate": 7.729618163054695e-06,
+      "loss": 0.0189,
+      "step": 4351
+    },
+    {
+      "epoch": 256.0,
+      "learning_rate": 7.719298245614036e-06,
+      "loss": 0.017,
+      "step": 4352
+    },
+    {
+      "epoch": 256.06,
+      "learning_rate": 7.708978328173374e-06,
+      "loss": 0.0199,
+      "step": 4353
+    },
+    {
+      "epoch": 256.12,
+      "learning_rate": 7.698658410732715e-06,
+      "loss": 0.0174,
+      "step": 4354
+    },
+    {
+      "epoch": 256.18,
+      "learning_rate": 7.688338493292054e-06,
+      "loss": 0.0175,
+      "step": 4355
+    },
+    {
+      "epoch": 256.24,
+      "learning_rate": 7.678018575851394e-06,
+      "loss": 0.0171,
+      "step": 4356
+    },
+    {
+      "epoch": 256.29,
+      "learning_rate": 7.667698658410732e-06,
+      "loss": 0.0218,
+      "step": 4357
+    },
+    {
+      "epoch": 256.35,
+      "learning_rate": 7.657378740970072e-06,
+      "loss": 0.016,
+      "step": 4358
+    },
+    {
+      "epoch": 256.41,
+      "learning_rate": 7.647058823529413e-06,
+      "loss": 0.0194,
+      "step": 4359
+    },
+    {
+      "epoch": 256.47,
+      "learning_rate": 7.636738906088752e-06,
+      "loss": 0.0222,
+      "step": 4360
+    },
+    {
+      "epoch": 256.47,
+      "eval_accuracy_NAT": 0.9939905640404577,
+      "eval_accuracy_REST": 0.432520325203252,
+      "eval_accuracy_concrete": 0.9572292232640126,
+      "eval_accuracy_concrete tiles": 0.9444025453688428,
+      "eval_accuracy_dark bricks": 0.8710162126983519,
+      "eval_accuracy_grass": 0.8992087643400818,
+      "eval_accuracy_red bricks": 0.9340949033391915,
+      "eval_accuracy_road": 0.9806669620206487,
+      "eval_accuracy_rough concrete": 0.6243214415531019,
+      "eval_accuracy_rough red sidewalk": 0.7862444446115552,
+      "eval_accuracy_speedway bricks": 0.9897031818396803,
+      "eval_accuracy_steel": 0.9271916716228441,
+      "eval_accuracy_tiles": 0.25644521191190284,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841058566949593,
+      "eval_iou_REST": 0.41304347826086957,
+      "eval_iou_concrete": 0.8945113817174634,
+      "eval_iou_concrete tiles": 0.8742827220561604,
+      "eval_iou_dark bricks": 0.8524395155625076,
+      "eval_iou_grass": 0.8440093929904277,
+      "eval_iou_red bricks": 0.9021854445151708,
+      "eval_iou_road": 0.9602945840123335,
+      "eval_iou_rough concrete": 0.6192294280699059,
+      "eval_iou_rough red sidewalk": 0.6287298399220658,
+      "eval_iou_speedway bricks": 0.9804346135864618,
+      "eval_iou_steel": 0.8734868566639515,
+      "eval_iou_tiles": 0.25261869945387805,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08453436940908432,
+      "eval_mean_accuracy": 0.8151565732164556,
+      "eval_mean_iou": 0.7753362933466273,
+      "eval_overall_accuracy": 0.9805330826064288,
+      "eval_runtime": 5.3922,
+      "eval_samples_per_second": 21.883,
+      "eval_steps_per_second": 1.484,
+      "step": 4360
+    },
+    {
+      "epoch": 256.53,
+      "learning_rate": 7.626418988648091e-06,
+      "loss": 0.0193,
+      "step": 4361
+    },
+    {
+      "epoch": 256.59,
+      "learning_rate": 7.61609907120743e-06,
+      "loss": 0.0181,
+      "step": 4362
+    },
+    {
+      "epoch": 256.65,
+      "learning_rate": 7.605779153766771e-06,
+      "loss": 0.0162,
+      "step": 4363
+    },
+    {
+      "epoch": 256.71,
+      "learning_rate": 7.59545923632611e-06,
+      "loss": 0.0172,
+      "step": 4364
+    },
+    {
+      "epoch": 256.76,
+      "learning_rate": 7.585139318885449e-06,
+      "loss": 0.0236,
+      "step": 4365
+    },
+    {
+      "epoch": 256.82,
+      "learning_rate": 7.574819401444788e-06,
+      "loss": 0.021,
+      "step": 4366
+    },
+    {
+      "epoch": 256.88,
+      "learning_rate": 7.564499484004128e-06,
+      "loss": 0.019,
+      "step": 4367
+    },
+    {
+      "epoch": 256.94,
+      "learning_rate": 7.5541795665634686e-06,
+      "loss": 0.0195,
+      "step": 4368
+    },
+    {
+      "epoch": 257.0,
+      "learning_rate": 7.5438596491228074e-06,
+      "loss": 0.0193,
+      "step": 4369
+    },
+    {
+      "epoch": 257.06,
+      "learning_rate": 7.533539731682147e-06,
+      "loss": 0.0208,
+      "step": 4370
+    },
+    {
+      "epoch": 257.12,
+      "learning_rate": 7.523219814241486e-06,
+      "loss": 0.015,
+      "step": 4371
+    },
+    {
+      "epoch": 257.18,
+      "learning_rate": 7.512899896800827e-06,
+      "loss": 0.0168,
+      "step": 4372
+    },
+    {
+      "epoch": 257.24,
+      "learning_rate": 7.502579979360165e-06,
+      "loss": 0.0165,
+      "step": 4373
+    },
+    {
+      "epoch": 257.29,
+      "learning_rate": 7.492260061919505e-06,
+      "loss": 0.0215,
+      "step": 4374
+    },
+    {
+      "epoch": 257.35,
+      "learning_rate": 7.481940144478844e-06,
+      "loss": 0.0214,
+      "step": 4375
+    },
+    {
+      "epoch": 257.41,
+      "learning_rate": 7.471620227038184e-06,
+      "loss": 0.0206,
+      "step": 4376
+    },
+    {
+      "epoch": 257.47,
+      "learning_rate": 7.461300309597524e-06,
+      "loss": 0.0203,
+      "step": 4377
+    },
+    {
+      "epoch": 257.53,
+      "learning_rate": 7.450980392156863e-06,
+      "loss": 0.0213,
+      "step": 4378
+    },
+    {
+      "epoch": 257.59,
+      "learning_rate": 7.440660474716203e-06,
+      "loss": 0.0165,
+      "step": 4379
+    },
+    {
+      "epoch": 257.65,
+      "learning_rate": 7.430340557275542e-06,
+      "loss": 0.0218,
+      "step": 4380
+    },
+    {
+      "epoch": 257.65,
+      "eval_accuracy_NAT": 0.9937031065411192,
+      "eval_accuracy_REST": 0.43089430894308944,
+      "eval_accuracy_concrete": 0.9583364546275926,
+      "eval_accuracy_concrete tiles": 0.941503653075654,
+      "eval_accuracy_dark bricks": 0.8609909461554658,
+      "eval_accuracy_grass": 0.8996648461244403,
+      "eval_accuracy_red bricks": 0.9356326889279437,
+      "eval_accuracy_road": 0.9813397919304233,
+      "eval_accuracy_rough concrete": 0.6261269582109297,
+      "eval_accuracy_rough red sidewalk": 0.769324479805082,
+      "eval_accuracy_speedway bricks": 0.9899513185863335,
+      "eval_accuracy_steel": 0.9262670738228871,
+      "eval_accuracy_tiles": 0.2509315641188767,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841528257934773,
+      "eval_iou_REST": 0.4033485540334855,
+      "eval_iou_concrete": 0.8935810654295268,
+      "eval_iou_concrete tiles": 0.8816207627118644,
+      "eval_iou_dark bricks": 0.843698858174486,
+      "eval_iou_grass": 0.8441533305218298,
+      "eval_iou_red bricks": 0.901805092372029,
+      "eval_iou_road": 0.9600184863853731,
+      "eval_iou_rough concrete": 0.6197894240127407,
+      "eval_iou_rough red sidewalk": 0.6181884101758415,
+      "eval_iou_speedway bricks": 0.9800462128062404,
+      "eval_iou_steel": 0.8736853518377968,
+      "eval_iou_tiles": 0.24891813919942302,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08372850716114044,
+      "eval_mean_accuracy": 0.8126667069899877,
+      "eval_mean_iou": 0.7733081933426242,
+      "eval_overall_accuracy": 0.9804058398230601,
+      "eval_runtime": 5.3918,
+      "eval_samples_per_second": 21.885,
+      "eval_steps_per_second": 1.484,
+      "step": 4380
+    },
+    {
+      "epoch": 257.71,
+      "learning_rate": 7.420020639834882e-06,
+      "loss": 0.0166,
+      "step": 4381
+    },
+    {
+      "epoch": 257.76,
+      "learning_rate": 7.40970072239422e-06,
+      "loss": 0.0197,
+      "step": 4382
+    },
+    {
+      "epoch": 257.82,
+      "learning_rate": 7.399380804953561e-06,
+      "loss": 0.02,
+      "step": 4383
+    },
+    {
+      "epoch": 257.88,
+      "learning_rate": 7.3890608875129e-06,
+      "loss": 0.0183,
+      "step": 4384
+    },
+    {
+      "epoch": 257.94,
+      "learning_rate": 7.3787409700722396e-06,
+      "loss": 0.0194,
+      "step": 4385
+    },
+    {
+      "epoch": 258.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0174,
+      "step": 4386
+    },
+    {
+      "epoch": 258.06,
+      "learning_rate": 7.358101135190919e-06,
+      "loss": 0.021,
+      "step": 4387
+    },
+    {
+      "epoch": 258.12,
+      "learning_rate": 7.347781217750259e-06,
+      "loss": 0.0175,
+      "step": 4388
+    },
+    {
+      "epoch": 258.18,
+      "learning_rate": 7.337461300309598e-06,
+      "loss": 0.0164,
+      "step": 4389
+    },
+    {
+      "epoch": 258.24,
+      "learning_rate": 7.327141382868938e-06,
+      "loss": 0.0175,
+      "step": 4390
+    },
+    {
+      "epoch": 258.29,
+      "learning_rate": 7.316821465428276e-06,
+      "loss": 0.0217,
+      "step": 4391
+    },
+    {
+      "epoch": 258.35,
+      "learning_rate": 7.306501547987617e-06,
+      "loss": 0.0189,
+      "step": 4392
+    },
+    {
+      "epoch": 258.41,
+      "learning_rate": 7.296181630546956e-06,
+      "loss": 0.0194,
+      "step": 4393
+    },
+    {
+      "epoch": 258.47,
+      "learning_rate": 7.285861713106295e-06,
+      "loss": 0.0187,
+      "step": 4394
+    },
+    {
+      "epoch": 258.53,
+      "learning_rate": 7.275541795665634e-06,
+      "loss": 0.0204,
+      "step": 4395
+    },
+    {
+      "epoch": 258.59,
+      "learning_rate": 7.265221878224975e-06,
+      "loss": 0.0211,
+      "step": 4396
+    },
+    {
+      "epoch": 258.65,
+      "learning_rate": 7.2549019607843145e-06,
+      "loss": 0.0195,
+      "step": 4397
+    },
+    {
+      "epoch": 258.71,
+      "learning_rate": 7.244582043343653e-06,
+      "loss": 0.0181,
+      "step": 4398
+    },
+    {
+      "epoch": 258.76,
+      "learning_rate": 7.234262125902994e-06,
+      "loss": 0.0178,
+      "step": 4399
+    },
+    {
+      "epoch": 258.82,
+      "learning_rate": 7.223942208462332e-06,
+      "loss": 0.0192,
+      "step": 4400
+    },
+    {
+      "epoch": 258.82,
+      "eval_accuracy_NAT": 0.9938184237937607,
+      "eval_accuracy_REST": 0.46504065040650405,
+      "eval_accuracy_concrete": 0.9569184952355035,
+      "eval_accuracy_concrete tiles": 0.9477020975724723,
+      "eval_accuracy_dark bricks": 0.8656470723350497,
+      "eval_accuracy_grass": 0.8979521738636751,
+      "eval_accuracy_red bricks": 0.9341498242530756,
+      "eval_accuracy_road": 0.9821435537924573,
+      "eval_accuracy_rough concrete": 0.6272352951692002,
+      "eval_accuracy_rough red sidewalk": 0.7713097556757081,
+      "eval_accuracy_speedway bricks": 0.9900457407875806,
+      "eval_accuracy_steel": 0.9275615107428269,
+      "eval_accuracy_tiles": 0.2485988669756733,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984142603600966,
+      "eval_iou_REST": 0.42686567164179107,
+      "eval_iou_concrete": 0.8939074936439324,
+      "eval_iou_concrete tiles": 0.8792172296927955,
+      "eval_iou_dark bricks": 0.8464832033242398,
+      "eval_iou_grass": 0.8448995181572453,
+      "eval_iou_red bricks": 0.902717333616389,
+      "eval_iou_road": 0.9603333441326688,
+      "eval_iou_rough concrete": 0.6196620945428858,
+      "eval_iou_rough red sidewalk": 0.6161313381910363,
+      "eval_iou_speedway bricks": 0.9799368691366066,
+      "eval_iou_steel": 0.8731302382145318,
+      "eval_iou_tiles": 0.24678214844219898,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08336872607469559,
+      "eval_mean_accuracy": 0.816009496969499,
+      "eval_mean_iou": 0.7749391604874835,
+      "eval_overall_accuracy": 0.9804315405376887,
+      "eval_runtime": 5.4639,
+      "eval_samples_per_second": 21.596,
+      "eval_steps_per_second": 1.464,
+      "step": 4400
+    },
+    {
+      "epoch": 258.88,
+      "learning_rate": 7.2136222910216725e-06,
+      "loss": 0.0203,
+      "step": 4401
+    },
+    {
+      "epoch": 258.94,
+      "learning_rate": 7.203302373581011e-06,
+      "loss": 0.0188,
+      "step": 4402
+    },
+    {
+      "epoch": 259.0,
+      "learning_rate": 7.192982456140351e-06,
+      "loss": 0.0201,
+      "step": 4403
+    },
+    {
+      "epoch": 259.06,
+      "learning_rate": 7.18266253869969e-06,
+      "loss": 0.0203,
+      "step": 4404
+    },
+    {
+      "epoch": 259.12,
+      "learning_rate": 7.1723426212590306e-06,
+      "loss": 0.0142,
+      "step": 4405
+    },
+    {
+      "epoch": 259.18,
+      "learning_rate": 7.16202270381837e-06,
+      "loss": 0.0216,
+      "step": 4406
+    },
+    {
+      "epoch": 259.24,
+      "learning_rate": 7.151702786377709e-06,
+      "loss": 0.0197,
+      "step": 4407
+    },
+    {
+      "epoch": 259.29,
+      "learning_rate": 7.14138286893705e-06,
+      "loss": 0.0204,
+      "step": 4408
+    },
+    {
+      "epoch": 259.35,
+      "learning_rate": 7.131062951496388e-06,
+      "loss": 0.0166,
+      "step": 4409
+    },
+    {
+      "epoch": 259.41,
+      "learning_rate": 7.120743034055728e-06,
+      "loss": 0.0182,
+      "step": 4410
+    },
+    {
+      "epoch": 259.47,
+      "learning_rate": 7.110423116615067e-06,
+      "loss": 0.0212,
+      "step": 4411
+    },
+    {
+      "epoch": 259.53,
+      "learning_rate": 7.100103199174407e-06,
+      "loss": 0.0212,
+      "step": 4412
+    },
+    {
+      "epoch": 259.59,
+      "learning_rate": 7.089783281733746e-06,
+      "loss": 0.0198,
+      "step": 4413
+    },
+    {
+      "epoch": 259.65,
+      "learning_rate": 7.079463364293086e-06,
+      "loss": 0.0176,
+      "step": 4414
+    },
+    {
+      "epoch": 259.71,
+      "learning_rate": 7.069143446852426e-06,
+      "loss": 0.0202,
+      "step": 4415
+    },
+    {
+      "epoch": 259.76,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 0.0186,
+      "step": 4416
+    },
+    {
+      "epoch": 259.82,
+      "learning_rate": 7.0485036119711055e-06,
+      "loss": 0.0173,
+      "step": 4417
+    },
+    {
+      "epoch": 259.88,
+      "learning_rate": 7.0381836945304435e-06,
+      "loss": 0.016,
+      "step": 4418
+    },
+    {
+      "epoch": 259.94,
+      "learning_rate": 7.027863777089784e-06,
+      "loss": 0.0184,
+      "step": 4419
+    },
+    {
+      "epoch": 260.0,
+      "learning_rate": 7.017543859649123e-06,
+      "loss": 0.0184,
+      "step": 4420
+    },
+    {
+      "epoch": 260.0,
+      "eval_accuracy_NAT": 0.9941259364674718,
+      "eval_accuracy_REST": 0.44715447154471544,
+      "eval_accuracy_concrete": 0.9546719277673904,
+      "eval_accuracy_concrete tiles": 0.9485505538534056,
+      "eval_accuracy_dark bricks": 0.8584355798863006,
+      "eval_accuracy_grass": 0.8965810898272945,
+      "eval_accuracy_red bricks": 0.9336006151142355,
+      "eval_accuracy_road": 0.9813285691916582,
+      "eval_accuracy_rough concrete": 0.6281469916993904,
+      "eval_accuracy_rough red sidewalk": 0.7854849261913536,
+      "eval_accuracy_speedway bricks": 0.9906979565733275,
+      "eval_accuracy_steel": 0.9238056445446231,
+      "eval_accuracy_tiles": 0.2487806355842346,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840730618482472,
+      "eval_iou_REST": 0.42242703533026116,
+      "eval_iou_concrete": 0.8939498432265308,
+      "eval_iou_concrete tiles": 0.8807554271708683,
+      "eval_iou_dark bricks": 0.8416526304430442,
+      "eval_iou_grass": 0.8438506074183318,
+      "eval_iou_red bricks": 0.9043945520323473,
+      "eval_iou_road": 0.9600135302787038,
+      "eval_iou_rough concrete": 0.6213638587453065,
+      "eval_iou_rough red sidewalk": 0.6226446586431564,
+      "eval_iou_speedway bricks": 0.9801028214441624,
+      "eval_iou_steel": 0.8739922739029083,
+      "eval_iou_tiles": 0.24690318701142513,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08423759788274765,
+      "eval_mean_accuracy": 0.8147203767881077,
+      "eval_mean_iou": 0.7750864221150225,
+      "eval_overall_accuracy": 0.9804220684504105,
+      "eval_runtime": 5.4643,
+      "eval_samples_per_second": 21.595,
+      "eval_steps_per_second": 1.464,
+      "step": 4420
+    },
+    {
+      "epoch": 260.06,
+      "learning_rate": 7.007223942208463e-06,
+      "loss": 0.019,
+      "step": 4421
+    },
+    {
+      "epoch": 260.12,
+      "learning_rate": 6.9969040247678016e-06,
+      "loss": 0.0205,
+      "step": 4422
+    },
+    {
+      "epoch": 260.18,
+      "learning_rate": 6.986584107327142e-06,
+      "loss": 0.0218,
+      "step": 4423
+    },
+    {
+      "epoch": 260.24,
+      "learning_rate": 6.97626418988648e-06,
+      "loss": 0.0188,
+      "step": 4424
+    },
+    {
+      "epoch": 260.29,
+      "learning_rate": 6.965944272445821e-06,
+      "loss": 0.0183,
+      "step": 4425
+    },
+    {
+      "epoch": 260.35,
+      "learning_rate": 6.9556243550051604e-06,
+      "loss": 0.0211,
+      "step": 4426
+    },
+    {
+      "epoch": 260.41,
+      "learning_rate": 6.945304437564499e-06,
+      "loss": 0.0171,
+      "step": 4427
+    },
+    {
+      "epoch": 260.47,
+      "learning_rate": 6.93498452012384e-06,
+      "loss": 0.0186,
+      "step": 4428
+    },
+    {
+      "epoch": 260.53,
+      "learning_rate": 6.924664602683179e-06,
+      "loss": 0.0167,
+      "step": 4429
+    },
+    {
+      "epoch": 260.59,
+      "learning_rate": 6.9143446852425185e-06,
+      "loss": 0.0195,
+      "step": 4430
+    },
+    {
+      "epoch": 260.65,
+      "learning_rate": 6.904024767801857e-06,
+      "loss": 0.0191,
+      "step": 4431
+    },
+    {
+      "epoch": 260.71,
+      "learning_rate": 6.893704850361198e-06,
+      "loss": 0.0166,
+      "step": 4432
+    },
+    {
+      "epoch": 260.76,
+      "learning_rate": 6.883384932920536e-06,
+      "loss": 0.0184,
+      "step": 4433
+    },
+    {
+      "epoch": 260.82,
+      "learning_rate": 6.8730650154798765e-06,
+      "loss": 0.0183,
+      "step": 4434
+    },
+    {
+      "epoch": 260.88,
+      "learning_rate": 6.862745098039216e-06,
+      "loss": 0.0176,
+      "step": 4435
+    },
+    {
+      "epoch": 260.94,
+      "learning_rate": 6.852425180598555e-06,
+      "loss": 0.0169,
+      "step": 4436
+    },
+    {
+      "epoch": 261.0,
+      "learning_rate": 6.842105263157896e-06,
+      "loss": 0.0233,
+      "step": 4437
+    },
+    {
+      "epoch": 261.06,
+      "learning_rate": 6.8317853457172345e-06,
+      "loss": 0.0168,
+      "step": 4438
+    },
+    {
+      "epoch": 261.12,
+      "learning_rate": 6.821465428276574e-06,
+      "loss": 0.0173,
+      "step": 4439
+    },
+    {
+      "epoch": 261.18,
+      "learning_rate": 6.811145510835913e-06,
+      "loss": 0.022,
+      "step": 4440
+    },
+    {
+      "epoch": 261.18,
+      "eval_accuracy_NAT": 0.9940929424709236,
+      "eval_accuracy_REST": 0.3902439024390244,
+      "eval_accuracy_concrete": 0.9556511223664558,
+      "eval_accuracy_concrete tiles": 0.9395239217534763,
+      "eval_accuracy_dark bricks": 0.8583494439446433,
+      "eval_accuracy_grass": 0.8975850374895442,
+      "eval_accuracy_red bricks": 0.938213971880492,
+      "eval_accuracy_road": 0.9808443881763638,
+      "eval_accuracy_rough concrete": 0.6245717111888404,
+      "eval_accuracy_rough red sidewalk": 0.773565750983238,
+      "eval_accuracy_speedway bricks": 0.990993561468339,
+      "eval_accuracy_steel": 0.9206670098507149,
+      "eval_accuracy_tiles": 0.2464782332091248,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841198104009627,
+      "eval_iou_REST": 0.3755868544600939,
+      "eval_iou_concrete": 0.8935472977198534,
+      "eval_iou_concrete tiles": 0.8816348195329087,
+      "eval_iou_dark bricks": 0.8424891736259195,
+      "eval_iou_grass": 0.8450980654185208,
+      "eval_iou_red bricks": 0.9032889170896785,
+      "eval_iou_road": 0.9606352253730046,
+      "eval_iou_rough concrete": 0.6192105014414669,
+      "eval_iou_rough red sidewalk": 0.6169513482391326,
+      "eval_iou_speedway bricks": 0.9798342988895877,
+      "eval_iou_steel": 0.8738718652381506,
+      "eval_iou_tiles": 0.24447849995492654,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08534476906061172,
+      "eval_mean_accuracy": 0.8085216151708601,
+      "eval_mean_iou": 0.7708266674910929,
+      "eval_overall_accuracy": 0.9804208076606362,
+      "eval_runtime": 5.7079,
+      "eval_samples_per_second": 20.673,
+      "eval_steps_per_second": 1.402,
+      "step": 4440
+    },
+    {
+      "epoch": 261.24,
+      "learning_rate": 6.800825593395254e-06,
+      "loss": 0.0206,
+      "step": 4441
+    },
+    {
+      "epoch": 261.29,
+      "learning_rate": 6.790505675954592e-06,
+      "loss": 0.0189,
+      "step": 4442
+    },
+    {
+      "epoch": 261.35,
+      "learning_rate": 6.780185758513932e-06,
+      "loss": 0.0189,
+      "step": 4443
+    },
+    {
+      "epoch": 261.41,
+      "learning_rate": 6.769865841073272e-06,
+      "loss": 0.0194,
+      "step": 4444
+    },
+    {
+      "epoch": 261.47,
+      "learning_rate": 6.759545923632611e-06,
+      "loss": 0.0174,
+      "step": 4445
+    },
+    {
+      "epoch": 261.53,
+      "learning_rate": 6.7492260061919514e-06,
+      "loss": 0.0163,
+      "step": 4446
+    },
+    {
+      "epoch": 261.59,
+      "learning_rate": 6.73890608875129e-06,
+      "loss": 0.0174,
+      "step": 4447
+    },
+    {
+      "epoch": 261.65,
+      "learning_rate": 6.72858617131063e-06,
+      "loss": 0.0178,
+      "step": 4448
+    },
+    {
+      "epoch": 261.71,
+      "learning_rate": 6.718266253869969e-06,
+      "loss": 0.0238,
+      "step": 4449
+    },
+    {
+      "epoch": 261.76,
+      "learning_rate": 6.7079463364293095e-06,
+      "loss": 0.0224,
+      "step": 4450
+    },
+    {
+      "epoch": 261.82,
+      "learning_rate": 6.6976264189886475e-06,
+      "loss": 0.0161,
+      "step": 4451
+    },
+    {
+      "epoch": 261.88,
+      "learning_rate": 6.687306501547988e-06,
+      "loss": 0.0173,
+      "step": 4452
+    },
+    {
+      "epoch": 261.94,
+      "learning_rate": 6.676986584107328e-06,
+      "loss": 0.0185,
+      "step": 4453
+    },
+    {
+      "epoch": 262.0,
+      "learning_rate": 6.666666666666667e-06,
+      "loss": 0.0182,
+      "step": 4454
+    },
+    {
+      "epoch": 262.06,
+      "learning_rate": 6.656346749226007e-06,
+      "loss": 0.0164,
+      "step": 4455
+    },
+    {
+      "epoch": 262.12,
+      "learning_rate": 6.646026831785346e-06,
+      "loss": 0.023,
+      "step": 4456
+    },
+    {
+      "epoch": 262.18,
+      "learning_rate": 6.635706914344686e-06,
+      "loss": 0.0193,
+      "step": 4457
+    },
+    {
+      "epoch": 262.24,
+      "learning_rate": 6.625386996904025e-06,
+      "loss": 0.018,
+      "step": 4458
+    },
+    {
+      "epoch": 262.29,
+      "learning_rate": 6.615067079463365e-06,
+      "loss": 0.0188,
+      "step": 4459
+    },
+    {
+      "epoch": 262.35,
+      "learning_rate": 6.604747162022703e-06,
+      "loss": 0.0171,
+      "step": 4460
+    },
+    {
+      "epoch": 262.35,
+      "eval_accuracy_NAT": 0.9937673154461665,
+      "eval_accuracy_REST": 0.3902439024390244,
+      "eval_accuracy_concrete": 0.9570874273248037,
+      "eval_accuracy_concrete tiles": 0.9363657789300024,
+      "eval_accuracy_dark bricks": 0.8588806155848631,
+      "eval_accuracy_grass": 0.9031980606115659,
+      "eval_accuracy_red bricks": 0.9347539543057997,
+      "eval_accuracy_road": 0.9814584323116544,
+      "eval_accuracy_rough concrete": 0.6239698723028978,
+      "eval_accuracy_rough red sidewalk": 0.7597590597011559,
+      "eval_accuracy_speedway bricks": 0.9902267023928745,
+      "eval_accuracy_steel": 0.9245078390899959,
+      "eval_accuracy_tiles": 0.24662970704959253,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841828387321433,
+      "eval_iou_REST": 0.3785488958990536,
+      "eval_iou_concrete": 0.8932826458332813,
+      "eval_iou_concrete tiles": 0.8815567586757789,
+      "eval_iou_dark bricks": 0.8419072636442526,
+      "eval_iou_grass": 0.8455070163286204,
+      "eval_iou_red bricks": 0.9045493197278912,
+      "eval_iou_road": 0.9605257377169217,
+      "eval_iou_rough concrete": 0.6178983642929639,
+      "eval_iou_rough red sidewalk": 0.6088502401484883,
+      "eval_iou_speedway bricks": 0.9801292826357239,
+      "eval_iou_steel": 0.8744602046435992,
+      "eval_iou_tiles": 0.24455525849379675,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08464911580085754,
+      "eval_mean_accuracy": 0.8077575898069534,
+      "eval_mean_iou": 0.7704579866748088,
+      "eval_overall_accuracy": 0.9804157321735963,
+      "eval_runtime": 5.2174,
+      "eval_samples_per_second": 22.617,
+      "eval_steps_per_second": 1.533,
+      "step": 4460
+    },
+    {
+      "epoch": 262.41,
+      "learning_rate": 6.594427244582044e-06,
+      "loss": 0.0167,
+      "step": 4461
+    },
+    {
+      "epoch": 262.47,
+      "learning_rate": 6.584107327141383e-06,
+      "loss": 0.0206,
+      "step": 4462
+    },
+    {
+      "epoch": 262.53,
+      "learning_rate": 6.5737874097007224e-06,
+      "loss": 0.0189,
+      "step": 4463
+    },
+    {
+      "epoch": 262.59,
+      "learning_rate": 6.563467492260063e-06,
+      "loss": 0.0186,
+      "step": 4464
+    },
+    {
+      "epoch": 262.65,
+      "learning_rate": 6.553147574819402e-06,
+      "loss": 0.0188,
+      "step": 4465
+    },
+    {
+      "epoch": 262.71,
+      "learning_rate": 6.542827657378742e-06,
+      "loss": 0.0192,
+      "step": 4466
+    },
+    {
+      "epoch": 262.76,
+      "learning_rate": 6.5325077399380805e-06,
+      "loss": 0.0159,
+      "step": 4467
+    },
+    {
+      "epoch": 262.82,
+      "learning_rate": 6.522187822497421e-06,
+      "loss": 0.0184,
+      "step": 4468
+    },
+    {
+      "epoch": 262.88,
+      "learning_rate": 6.511867905056759e-06,
+      "loss": 0.0202,
+      "step": 4469
+    },
+    {
+      "epoch": 262.94,
+      "learning_rate": 6.5015479876161e-06,
+      "loss": 0.0184,
+      "step": 4470
+    },
+    {
+      "epoch": 263.0,
+      "learning_rate": 6.4912280701754385e-06,
+      "loss": 0.0216,
+      "step": 4471
+    },
+    {
+      "epoch": 263.06,
+      "learning_rate": 6.480908152734778e-06,
+      "loss": 0.0206,
+      "step": 4472
+    },
+    {
+      "epoch": 263.12,
+      "learning_rate": 6.470588235294119e-06,
+      "loss": 0.0185,
+      "step": 4473
+    },
+    {
+      "epoch": 263.18,
+      "learning_rate": 6.460268317853458e-06,
+      "loss": 0.0233,
+      "step": 4474
+    },
+    {
+      "epoch": 263.24,
+      "learning_rate": 6.449948400412797e-06,
+      "loss": 0.0175,
+      "step": 4475
+    },
+    {
+      "epoch": 263.29,
+      "learning_rate": 6.439628482972136e-06,
+      "loss": 0.0162,
+      "step": 4476
+    },
+    {
+      "epoch": 263.35,
+      "learning_rate": 6.429308565531476e-06,
+      "loss": 0.0185,
+      "step": 4477
+    },
+    {
+      "epoch": 263.41,
+      "learning_rate": 6.418988648090815e-06,
+      "loss": 0.017,
+      "step": 4478
+    },
+    {
+      "epoch": 263.47,
+      "learning_rate": 6.408668730650155e-06,
+      "loss": 0.0179,
+      "step": 4479
+    },
+    {
+      "epoch": 263.53,
+      "learning_rate": 6.398348813209494e-06,
+      "loss": 0.0186,
+      "step": 4480
+    },
+    {
+      "epoch": 263.53,
+      "eval_accuracy_NAT": 0.9937675310932027,
+      "eval_accuracy_REST": 0.40487804878048783,
+      "eval_accuracy_concrete": 0.9589269525415489,
+      "eval_accuracy_concrete tiles": 0.9366250294602876,
+      "eval_accuracy_dark bricks": 0.8536023964933102,
+      "eval_accuracy_grass": 0.8974364797299087,
+      "eval_accuracy_red bricks": 0.9305250439367311,
+      "eval_accuracy_road": 0.9810292961579221,
+      "eval_accuracy_rough concrete": 0.6223669548740012,
+      "eval_accuracy_rough red sidewalk": 0.7808601358109175,
+      "eval_accuracy_speedway bricks": 0.9894123751690517,
+      "eval_accuracy_steel": 0.9278538835606511,
+      "eval_accuracy_tiles": 0.24396376745736012,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9840769181859401,
+      "eval_iou_REST": 0.3915094339622642,
+      "eval_iou_concrete": 0.8926123996644393,
+      "eval_iou_concrete tiles": 0.8788562329994029,
+      "eval_iou_dark bricks": 0.8384205380811822,
+      "eval_iou_grass": 0.8423369255336345,
+      "eval_iou_red bricks": 0.903385763796321,
+      "eval_iou_road": 0.9600595584266701,
+      "eval_iou_rough concrete": 0.6177180303048225,
+      "eval_iou_rough red sidewalk": 0.6193182834819432,
+      "eval_iou_speedway bricks": 0.980190625650954,
+      "eval_iou_steel": 0.8739975472715161,
+      "eval_iou_tiles": 0.24170843713419576,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08595266938209534,
+      "eval_mean_accuracy": 0.8093267611588755,
+      "eval_mean_iou": 0.771091591884099,
+      "eval_overall_accuracy": 0.9802799871412374,
+      "eval_runtime": 5.4157,
+      "eval_samples_per_second": 21.789,
+      "eval_steps_per_second": 1.477,
+      "step": 4480
+    },
+    {
+      "epoch": 263.59,
+      "learning_rate": 6.388028895768834e-06,
+      "loss": 0.0169,
+      "step": 4481
+    },
+    {
+      "epoch": 263.65,
+      "learning_rate": 6.3777089783281746e-06,
+      "loss": 0.0168,
+      "step": 4482
+    },
+    {
+      "epoch": 263.71,
+      "learning_rate": 6.3673890608875134e-06,
+      "loss": 0.0188,
+      "step": 4483
+    },
+    {
+      "epoch": 263.76,
+      "learning_rate": 6.357069143446853e-06,
+      "loss": 0.0201,
+      "step": 4484
+    },
+    {
+      "epoch": 263.82,
+      "learning_rate": 6.346749226006192e-06,
+      "loss": 0.0183,
+      "step": 4485
+    },
+    {
+      "epoch": 263.88,
+      "learning_rate": 6.336429308565532e-06,
+      "loss": 0.0215,
+      "step": 4486
+    },
+    {
+      "epoch": 263.94,
+      "learning_rate": 6.326109391124871e-06,
+      "loss": 0.0197,
+      "step": 4487
+    },
+    {
+      "epoch": 264.0,
+      "learning_rate": 6.315789473684211e-06,
+      "loss": 0.0188,
+      "step": 4488
+    },
+    {
+      "epoch": 264.06,
+      "learning_rate": 6.30546955624355e-06,
+      "loss": 0.0208,
+      "step": 4489
+    },
+    {
+      "epoch": 264.12,
+      "learning_rate": 6.29514963880289e-06,
+      "loss": 0.0202,
+      "step": 4490
+    },
+    {
+      "epoch": 264.18,
+      "learning_rate": 6.28482972136223e-06,
+      "loss": 0.0199,
+      "step": 4491
+    },
+    {
+      "epoch": 264.24,
+      "learning_rate": 6.274509803921569e-06,
+      "loss": 0.0178,
+      "step": 4492
+    },
+    {
+      "epoch": 264.29,
+      "learning_rate": 6.264189886480909e-06,
+      "loss": 0.0184,
+      "step": 4493
+    },
+    {
+      "epoch": 264.35,
+      "learning_rate": 6.253869969040248e-06,
+      "loss": 0.0162,
+      "step": 4494
+    },
+    {
+      "epoch": 264.41,
+      "learning_rate": 6.2435500515995875e-06,
+      "loss": 0.0204,
+      "step": 4495
+    },
+    {
+      "epoch": 264.47,
+      "learning_rate": 6.233230134158927e-06,
+      "loss": 0.0169,
+      "step": 4496
+    },
+    {
+      "epoch": 264.53,
+      "learning_rate": 6.222910216718267e-06,
+      "loss": 0.0164,
+      "step": 4497
+    },
+    {
+      "epoch": 264.59,
+      "learning_rate": 6.212590299277606e-06,
+      "loss": 0.0165,
+      "step": 4498
+    },
+    {
+      "epoch": 264.65,
+      "learning_rate": 6.2022703818369455e-06,
+      "loss": 0.0177,
+      "step": 4499
+    },
+    {
+      "epoch": 264.71,
+      "learning_rate": 6.191950464396285e-06,
+      "loss": 0.0161,
+      "step": 4500
+    },
+    {
+      "epoch": 264.71,
+      "eval_accuracy_NAT": 0.9936167938148548,
+      "eval_accuracy_REST": 0.432520325203252,
+      "eval_accuracy_concrete": 0.9586571198378252,
+      "eval_accuracy_concrete tiles": 0.9336318642469951,
+      "eval_accuracy_dark bricks": 0.8584164385659323,
+      "eval_accuracy_grass": 0.9028791818536224,
+      "eval_accuracy_red bricks": 0.9324472759226714,
+      "eval_accuracy_road": 0.9816780773417714,
+      "eval_accuracy_rough concrete": 0.6249590332441499,
+      "eval_accuracy_rough red sidewalk": 0.783928289429158,
+      "eval_accuracy_speedway bricks": 0.9892925909246384,
+      "eval_accuracy_steel": 0.9286285465822367,
+      "eval_accuracy_tiles": 0.24420612560210853,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841894149495595,
+      "eval_iou_REST": 0.41112828438948995,
+      "eval_iou_concrete": 0.8937431052225,
+      "eval_iou_concrete tiles": 0.8808983766955748,
+      "eval_iou_dark bricks": 0.8413062441961899,
+      "eval_iou_grass": 0.844349151879457,
+      "eval_iou_red bricks": 0.9048659596013431,
+      "eval_iou_road": 0.960489521459048,
+      "eval_iou_rough concrete": 0.618852338102965,
+      "eval_iou_rough red sidewalk": 0.6238763809592206,
+      "eval_iou_speedway bricks": 0.9801180486088898,
+      "eval_iou_steel": 0.874391758981261,
+      "eval_iou_tiles": 0.24202119674542888,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08457301557064056,
+      "eval_mean_accuracy": 0.8126816663514782,
+      "eval_mean_iou": 0.7738638293685328,
+      "eval_overall_accuracy": 0.9804370039600437,
+      "eval_runtime": 5.3377,
+      "eval_samples_per_second": 22.107,
+      "eval_steps_per_second": 1.499,
+      "step": 4500
+    },
+    {
+      "epoch": 264.76,
+      "learning_rate": 6.181630546955625e-06,
+      "loss": 0.0188,
+      "step": 4501
+    },
+    {
+      "epoch": 264.82,
+      "learning_rate": 6.171310629514964e-06,
+      "loss": 0.0204,
+      "step": 4502
+    },
+    {
+      "epoch": 264.88,
+      "learning_rate": 6.160990712074304e-06,
+      "loss": 0.0185,
+      "step": 4503
+    },
+    {
+      "epoch": 264.94,
+      "learning_rate": 6.150670794633643e-06,
+      "loss": 0.0185,
+      "step": 4504
+    },
+    {
+      "epoch": 265.0,
+      "learning_rate": 6.140350877192982e-06,
+      "loss": 0.0173,
+      "step": 4505
+    },
+    {
+      "epoch": 265.06,
+      "learning_rate": 6.130030959752323e-06,
+      "loss": 0.017,
+      "step": 4506
+    },
+    {
+      "epoch": 265.12,
+      "learning_rate": 6.119711042311662e-06,
+      "loss": 0.0174,
+      "step": 4507
+    },
+    {
+      "epoch": 265.18,
+      "learning_rate": 6.109391124871001e-06,
+      "loss": 0.0188,
+      "step": 4508
+    },
+    {
+      "epoch": 265.24,
+      "learning_rate": 6.099071207430341e-06,
+      "loss": 0.0195,
+      "step": 4509
+    },
+    {
+      "epoch": 265.29,
+      "learning_rate": 6.088751289989681e-06,
+      "loss": 0.0191,
+      "step": 4510
+    },
+    {
+      "epoch": 265.35,
+      "learning_rate": 6.07843137254902e-06,
+      "loss": 0.0175,
+      "step": 4511
+    },
+    {
+      "epoch": 265.41,
+      "learning_rate": 6.068111455108359e-06,
+      "loss": 0.021,
+      "step": 4512
+    },
+    {
+      "epoch": 265.47,
+      "learning_rate": 6.057791537667699e-06,
+      "loss": 0.0193,
+      "step": 4513
+    },
+    {
+      "epoch": 265.53,
+      "learning_rate": 6.047471620227038e-06,
+      "loss": 0.0213,
+      "step": 4514
+    },
+    {
+      "epoch": 265.59,
+      "learning_rate": 6.0371517027863785e-06,
+      "loss": 0.0202,
+      "step": 4515
+    },
+    {
+      "epoch": 265.65,
+      "learning_rate": 6.026831785345717e-06,
+      "loss": 0.0172,
+      "step": 4516
+    },
+    {
+      "epoch": 265.71,
+      "learning_rate": 6.016511867905057e-06,
+      "loss": 0.0169,
+      "step": 4517
+    },
+    {
+      "epoch": 265.76,
+      "learning_rate": 6.006191950464397e-06,
+      "loss": 0.0177,
+      "step": 4518
+    },
+    {
+      "epoch": 265.82,
+      "learning_rate": 5.995872033023736e-06,
+      "loss": 0.0183,
+      "step": 4519
+    },
+    {
+      "epoch": 265.88,
+      "learning_rate": 5.985552115583075e-06,
+      "loss": 0.0159,
+      "step": 4520
+    },
+    {
+      "epoch": 265.88,
+      "eval_accuracy_NAT": 0.9937682858578296,
+      "eval_accuracy_REST": 0.416260162601626,
+      "eval_accuracy_concrete": 0.9579072448169953,
+      "eval_accuracy_concrete tiles": 0.9452274334197501,
+      "eval_accuracy_dark bricks": 0.8555595965009667,
+      "eval_accuracy_grass": 0.9034317788703554,
+      "eval_accuracy_red bricks": 0.9331612478031635,
+      "eval_accuracy_road": 0.9811848112522386,
+      "eval_accuracy_rough concrete": 0.6251318384688266,
+      "eval_accuracy_rough red sidewalk": 0.7862369246271967,
+      "eval_accuracy_speedway bricks": 0.9896526291185045,
+      "eval_accuracy_steel": 0.9289409107038439,
+      "eval_accuracy_tiles": 0.24623587506437639,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842334595330053,
+      "eval_iou_REST": 0.4,
+      "eval_iou_concrete": 0.8942366529193511,
+      "eval_iou_concrete tiles": 0.8813536973958905,
+      "eval_iou_dark bricks": 0.8398258230892446,
+      "eval_iou_grass": 0.8449496670280316,
+      "eval_iou_red bricks": 0.9062349991999573,
+      "eval_iou_road": 0.9611286073850097,
+      "eval_iou_rough concrete": 0.6193523629601204,
+      "eval_iou_rough red sidewalk": 0.6278102031993082,
+      "eval_iou_speedway bricks": 0.980192012138874,
+      "eval_iou_steel": 0.8750029422703028,
+      "eval_iou_tiles": 0.24434089884262739,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08410942554473877,
+      "eval_mean_accuracy": 0.8125152876235135,
+      "eval_mean_iou": 0.7737431789201327,
+      "eval_overall_accuracy": 0.9805351515947762,
+      "eval_runtime": 5.366,
+      "eval_samples_per_second": 21.99,
+      "eval_steps_per_second": 1.491,
+      "step": 4520
+    },
+    {
+      "epoch": 265.94,
+      "learning_rate": 5.975232198142415e-06,
+      "loss": 0.0192,
+      "step": 4521
+    },
+    {
+      "epoch": 266.0,
+      "learning_rate": 5.964912280701755e-06,
+      "loss": 0.0205,
+      "step": 4522
+    },
+    {
+      "epoch": 266.06,
+      "learning_rate": 5.954592363261094e-06,
+      "loss": 0.0191,
+      "step": 4523
+    },
+    {
+      "epoch": 266.12,
+      "learning_rate": 5.9442724458204335e-06,
+      "loss": 0.0199,
+      "step": 4524
+    },
+    {
+      "epoch": 266.18,
+      "learning_rate": 5.933952528379773e-06,
+      "loss": 0.0176,
+      "step": 4525
+    },
+    {
+      "epoch": 266.24,
+      "learning_rate": 5.923632610939113e-06,
+      "loss": 0.0179,
+      "step": 4526
+    },
+    {
+      "epoch": 266.29,
+      "learning_rate": 5.913312693498453e-06,
+      "loss": 0.0182,
+      "step": 4527
+    },
+    {
+      "epoch": 266.35,
+      "learning_rate": 5.9029927760577915e-06,
+      "loss": 0.0172,
+      "step": 4528
+    },
+    {
+      "epoch": 266.41,
+      "learning_rate": 5.892672858617131e-06,
+      "loss": 0.0181,
+      "step": 4529
+    },
+    {
+      "epoch": 266.47,
+      "learning_rate": 5.882352941176471e-06,
+      "loss": 0.0197,
+      "step": 4530
+    },
+    {
+      "epoch": 266.53,
+      "learning_rate": 5.872033023735811e-06,
+      "loss": 0.0174,
+      "step": 4531
+    },
+    {
+      "epoch": 266.59,
+      "learning_rate": 5.8617131062951495e-06,
+      "loss": 0.0194,
+      "step": 4532
+    },
+    {
+      "epoch": 266.65,
+      "learning_rate": 5.851393188854489e-06,
+      "loss": 0.0198,
+      "step": 4533
+    },
+    {
+      "epoch": 266.71,
+      "learning_rate": 5.841073271413829e-06,
+      "loss": 0.0172,
+      "step": 4534
+    },
+    {
+      "epoch": 266.76,
+      "learning_rate": 5.830753353973169e-06,
+      "loss": 0.0186,
+      "step": 4535
+    },
+    {
+      "epoch": 266.82,
+      "learning_rate": 5.820433436532508e-06,
+      "loss": 0.0183,
+      "step": 4536
+    },
+    {
+      "epoch": 266.88,
+      "learning_rate": 5.810113519091847e-06,
+      "loss": 0.0205,
+      "step": 4537
+    },
+    {
+      "epoch": 266.94,
+      "learning_rate": 5.799793601651187e-06,
+      "loss": 0.0208,
+      "step": 4538
+    },
+    {
+      "epoch": 267.0,
+      "learning_rate": 5.789473684210527e-06,
+      "loss": 0.017,
+      "step": 4539
+    },
+    {
+      "epoch": 267.06,
+      "learning_rate": 5.779153766769866e-06,
+      "loss": 0.0195,
+      "step": 4540
+    },
+    {
+      "epoch": 267.06,
+      "eval_accuracy_NAT": 0.993972719248207,
+      "eval_accuracy_REST": 0.44065040650406506,
+      "eval_accuracy_concrete": 0.9592621413250697,
+      "eval_accuracy_concrete tiles": 0.9412208343153429,
+      "eval_accuracy_dark bricks": 0.8514059299810501,
+      "eval_accuracy_grass": 0.8951655330858002,
+      "eval_accuracy_red bricks": 0.9284380492091389,
+      "eval_accuracy_road": 0.9813531523337151,
+      "eval_accuracy_rough concrete": 0.6238626138875812,
+      "eval_accuracy_rough red sidewalk": 0.7694523195391754,
+      "eval_accuracy_speedway bricks": 0.9897425958256819,
+      "eval_accuracy_steel": 0.9248851749488972,
+      "eval_accuracy_tiles": 0.23766245569390165,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841567935319594,
+      "eval_iou_REST": 0.4214618973561431,
+      "eval_iou_concrete": 0.8929247261561318,
+      "eval_iou_concrete tiles": 0.8807339449541285,
+      "eval_iou_dark bricks": 0.8358349188452775,
+      "eval_iou_grass": 0.8432693001933385,
+      "eval_iou_red bricks": 0.9018886043533931,
+      "eval_iou_road": 0.960329783256152,
+      "eval_iou_rough concrete": 0.6183138934002658,
+      "eval_iou_rough red sidewalk": 0.6176193637954971,
+      "eval_iou_speedway bricks": 0.9800119896162389,
+      "eval_iou_steel": 0.8746331226988936,
+      "eval_iou_tiles": 0.2359824329202262,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08584803342819214,
+      "eval_mean_accuracy": 0.8105441481459714,
+      "eval_mean_iou": 0.7728585208521266,
+      "eval_overall_accuracy": 0.9803405697062865,
+      "eval_runtime": 5.5877,
+      "eval_samples_per_second": 21.118,
+      "eval_steps_per_second": 1.432,
+      "step": 4540
+    },
+    {
+      "epoch": 267.12,
+      "learning_rate": 5.768833849329205e-06,
+      "loss": 0.0174,
+      "step": 4541
+    },
+    {
+      "epoch": 267.18,
+      "learning_rate": 5.758513931888545e-06,
+      "loss": 0.0194,
+      "step": 4542
+    },
+    {
+      "epoch": 267.24,
+      "learning_rate": 5.748194014447885e-06,
+      "loss": 0.0182,
+      "step": 4543
+    },
+    {
+      "epoch": 267.29,
+      "learning_rate": 5.7378740970072245e-06,
+      "loss": 0.0225,
+      "step": 4544
+    },
+    {
+      "epoch": 267.35,
+      "learning_rate": 5.727554179566564e-06,
+      "loss": 0.0185,
+      "step": 4545
+    },
+    {
+      "epoch": 267.41,
+      "learning_rate": 5.717234262125903e-06,
+      "loss": 0.0157,
+      "step": 4546
+    },
+    {
+      "epoch": 267.47,
+      "learning_rate": 5.706914344685243e-06,
+      "loss": 0.0209,
+      "step": 4547
+    },
+    {
+      "epoch": 267.53,
+      "learning_rate": 5.6965944272445825e-06,
+      "loss": 0.0173,
+      "step": 4548
+    },
+    {
+      "epoch": 267.59,
+      "learning_rate": 5.686274509803921e-06,
+      "loss": 0.021,
+      "step": 4549
+    },
+    {
+      "epoch": 267.65,
+      "learning_rate": 5.675954592363261e-06,
+      "loss": 0.0222,
+      "step": 4550
+    },
+    {
+      "epoch": 267.71,
+      "learning_rate": 5.665634674922601e-06,
+      "loss": 0.0171,
+      "step": 4551
+    },
+    {
+      "epoch": 267.76,
+      "learning_rate": 5.6553147574819405e-06,
+      "loss": 0.0173,
+      "step": 4552
+    },
+    {
+      "epoch": 267.82,
+      "learning_rate": 5.64499484004128e-06,
+      "loss": 0.0164,
+      "step": 4553
+    },
+    {
+      "epoch": 267.88,
+      "learning_rate": 5.63467492260062e-06,
+      "loss": 0.0182,
+      "step": 4554
+    },
+    {
+      "epoch": 267.94,
+      "learning_rate": 5.624355005159959e-06,
+      "loss": 0.0203,
+      "step": 4555
+    },
+    {
+      "epoch": 268.0,
+      "learning_rate": 5.6140350877192985e-06,
+      "loss": 0.0198,
+      "step": 4556
+    },
+    {
+      "epoch": 268.06,
+      "learning_rate": 5.603715170278638e-06,
+      "loss": 0.0189,
+      "step": 4557
+    },
+    {
+      "epoch": 268.12,
+      "learning_rate": 5.593395252837977e-06,
+      "loss": 0.0173,
+      "step": 4558
+    },
+    {
+      "epoch": 268.18,
+      "learning_rate": 5.583075335397317e-06,
+      "loss": 0.0204,
+      "step": 4559
+    },
+    {
+      "epoch": 268.24,
+      "learning_rate": 5.5727554179566566e-06,
+      "loss": 0.0172,
+      "step": 4560
+    },
+    {
+      "epoch": 268.24,
+      "eval_accuracy_NAT": 0.9933902026915015,
+      "eval_accuracy_REST": 0.4634146341463415,
+      "eval_accuracy_concrete": 0.9585676852023134,
+      "eval_accuracy_concrete tiles": 0.949375441904313,
+      "eval_accuracy_dark bricks": 0.8576172884405566,
+      "eval_accuracy_grass": 0.9058001869746071,
+      "eval_accuracy_red bricks": 0.9289323374340949,
+      "eval_accuracy_road": 0.9816567006965046,
+      "eval_accuracy_rough concrete": 0.6260435349990168,
+      "eval_accuracy_rough red sidewalk": 0.7861617247836125,
+      "eval_accuracy_speedway bricks": 0.9899208155884713,
+      "eval_accuracy_steel": 0.9261371303482985,
+      "eval_accuracy_tiles": 0.23717773940440487,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9843329190595584,
+      "eval_iou_REST": 0.42857142857142855,
+      "eval_iou_concrete": 0.8937243721111554,
+      "eval_iou_concrete tiles": 0.88127064691855,
+      "eval_iou_dark bricks": 0.8383526528015568,
+      "eval_iou_grass": 0.846097754993813,
+      "eval_iou_red bricks": 0.9010227999147666,
+      "eval_iou_road": 0.960949965524854,
+      "eval_iou_rough concrete": 0.619882350858768,
+      "eval_iou_rough red sidewalk": 0.6250814663342242,
+      "eval_iou_speedway bricks": 0.9800143524110694,
+      "eval_iou_steel": 0.8739562380472897,
+      "eval_iou_tiles": 0.23542324462486844,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08350671827793121,
+      "eval_mean_accuracy": 0.8157073402010796,
+      "eval_mean_iou": 0.7745138609363003,
+      "eval_overall_accuracy": 0.9805010779429291,
+      "eval_runtime": 5.1917,
+      "eval_samples_per_second": 22.729,
+      "eval_steps_per_second": 1.541,
+      "step": 4560
+    },
+    {
+      "epoch": 268.29,
+      "learning_rate": 5.562435500515996e-06,
+      "loss": 0.0181,
+      "step": 4561
+    },
+    {
+      "epoch": 268.35,
+      "learning_rate": 5.552115583075335e-06,
+      "loss": 0.0206,
+      "step": 4562
+    },
+    {
+      "epoch": 268.41,
+      "learning_rate": 5.541795665634676e-06,
+      "loss": 0.0205,
+      "step": 4563
+    },
+    {
+      "epoch": 268.47,
+      "learning_rate": 5.531475748194015e-06,
+      "loss": 0.0162,
+      "step": 4564
+    },
+    {
+      "epoch": 268.53,
+      "learning_rate": 5.521155830753354e-06,
+      "loss": 0.0183,
+      "step": 4565
+    },
+    {
+      "epoch": 268.59,
+      "learning_rate": 5.510835913312694e-06,
+      "loss": 0.0181,
+      "step": 4566
+    },
+    {
+      "epoch": 268.65,
+      "learning_rate": 5.500515995872033e-06,
+      "loss": 0.0166,
+      "step": 4567
+    },
+    {
+      "epoch": 268.71,
+      "learning_rate": 5.490196078431373e-06,
+      "loss": 0.0205,
+      "step": 4568
+    },
+    {
+      "epoch": 268.76,
+      "learning_rate": 5.479876160990712e-06,
+      "loss": 0.0193,
+      "step": 4569
+    },
+    {
+      "epoch": 268.82,
+      "learning_rate": 5.469556243550051e-06,
+      "loss": 0.0213,
+      "step": 4570
+    },
+    {
+      "epoch": 268.88,
+      "learning_rate": 5.459236326109391e-06,
+      "loss": 0.0172,
+      "step": 4571
+    },
+    {
+      "epoch": 268.94,
+      "learning_rate": 5.4489164086687315e-06,
+      "loss": 0.0181,
+      "step": 4572
+    },
+    {
+      "epoch": 269.0,
+      "learning_rate": 5.43859649122807e-06,
+      "loss": 0.0198,
+      "step": 4573
+    },
+    {
+      "epoch": 269.06,
+      "learning_rate": 5.42827657378741e-06,
+      "loss": 0.0172,
+      "step": 4574
+    },
+    {
+      "epoch": 269.12,
+      "learning_rate": 5.41795665634675e-06,
+      "loss": 0.0146,
+      "step": 4575
+    },
+    {
+      "epoch": 269.18,
+      "learning_rate": 5.407636738906089e-06,
+      "loss": 0.0182,
+      "step": 4576
+    },
+    {
+      "epoch": 269.24,
+      "learning_rate": 5.397316821465428e-06,
+      "loss": 0.0188,
+      "step": 4577
+    },
+    {
+      "epoch": 269.29,
+      "learning_rate": 5.386996904024768e-06,
+      "loss": 0.0178,
+      "step": 4578
+    },
+    {
+      "epoch": 269.35,
+      "learning_rate": 5.376676986584107e-06,
+      "loss": 0.0226,
+      "step": 4579
+    },
+    {
+      "epoch": 269.41,
+      "learning_rate": 5.366357069143447e-06,
+      "loss": 0.019,
+      "step": 4580
+    },
+    {
+      "epoch": 269.41,
+      "eval_accuracy_NAT": 0.9940149860673145,
+      "eval_accuracy_REST": 0.408130081300813,
+      "eval_accuracy_concrete": 0.9559477090209736,
+      "eval_accuracy_concrete tiles": 0.9444025453688428,
+      "eval_accuracy_dark bricks": 0.8569808395383114,
+      "eval_accuracy_grass": 0.90551253373302,
+      "eval_accuracy_red bricks": 0.9306348857644992,
+      "eval_accuracy_road": 0.9817870982326324,
+      "eval_accuracy_rough concrete": 0.6255549133292416,
+      "eval_accuracy_rough red sidewalk": 0.7708359966611269,
+      "eval_accuracy_speedway bricks": 0.9897096937156284,
+      "eval_accuracy_steel": 0.9284386291962996,
+      "eval_accuracy_tiles": 0.2432366930231149,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842737750050661,
+      "eval_iou_REST": 0.38854489164086686,
+      "eval_iou_concrete": 0.8943282980102685,
+      "eval_iou_concrete tiles": 0.8814174475386036,
+      "eval_iou_dark bricks": 0.840415597092322,
+      "eval_iou_grass": 0.8465982879972257,
+      "eval_iou_red bricks": 0.9032515991471215,
+      "eval_iou_road": 0.9612792701247069,
+      "eval_iou_rough concrete": 0.620328186582995,
+      "eval_iou_rough red sidewalk": 0.6222833345474855,
+      "eval_iou_speedway bricks": 0.980005684445406,
+      "eval_iou_steel": 0.8741759094994518,
+      "eval_iou_tiles": 0.24151000150398555,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08344563841819763,
+      "eval_mean_accuracy": 0.8103989696116785,
+      "eval_mean_iou": 0.7721855602411927,
+      "eval_overall_accuracy": 0.9805688373113083,
+      "eval_runtime": 5.1838,
+      "eval_samples_per_second": 22.763,
+      "eval_steps_per_second": 1.543,
+      "step": 4580
+    },
+    {
+      "epoch": 269.47,
+      "learning_rate": 5.3560371517027864e-06,
+      "loss": 0.0207,
+      "step": 4581
+    },
+    {
+      "epoch": 269.53,
+      "learning_rate": 5.345717234262126e-06,
+      "loss": 0.0211,
+      "step": 4582
+    },
+    {
+      "epoch": 269.59,
+      "learning_rate": 5.335397316821466e-06,
+      "loss": 0.0178,
+      "step": 4583
+    },
+    {
+      "epoch": 269.65,
+      "learning_rate": 5.325077399380806e-06,
+      "loss": 0.0176,
+      "step": 4584
+    },
+    {
+      "epoch": 269.71,
+      "learning_rate": 5.3147574819401445e-06,
+      "loss": 0.0185,
+      "step": 4585
+    },
+    {
+      "epoch": 269.76,
+      "learning_rate": 5.304437564499484e-06,
+      "loss": 0.0179,
+      "step": 4586
+    },
+    {
+      "epoch": 269.82,
+      "learning_rate": 5.294117647058824e-06,
+      "loss": 0.0192,
+      "step": 4587
+    },
+    {
+      "epoch": 269.88,
+      "learning_rate": 5.283797729618163e-06,
+      "loss": 0.0186,
+      "step": 4588
+    },
+    {
+      "epoch": 269.94,
+      "learning_rate": 5.2734778121775025e-06,
+      "loss": 0.0168,
+      "step": 4589
+    },
+    {
+      "epoch": 270.0,
+      "learning_rate": 5.263157894736842e-06,
+      "loss": 0.0165,
+      "step": 4590
+    },
+    {
+      "epoch": 270.06,
+      "learning_rate": 5.252837977296182e-06,
+      "loss": 0.0159,
+      "step": 4591
+    },
+    {
+      "epoch": 270.12,
+      "learning_rate": 5.242518059855522e-06,
+      "loss": 0.0169,
+      "step": 4592
+    },
+    {
+      "epoch": 270.18,
+      "learning_rate": 5.232198142414861e-06,
+      "loss": 0.0192,
+      "step": 4593
+    },
+    {
+      "epoch": 270.24,
+      "learning_rate": 5.2218782249742e-06,
+      "loss": 0.0192,
+      "step": 4594
+    },
+    {
+      "epoch": 270.29,
+      "learning_rate": 5.21155830753354e-06,
+      "loss": 0.0206,
+      "step": 4595
+    },
+    {
+      "epoch": 270.35,
+      "learning_rate": 5.20123839009288e-06,
+      "loss": 0.0179,
+      "step": 4596
+    },
+    {
+      "epoch": 270.41,
+      "learning_rate": 5.1909184726522186e-06,
+      "loss": 0.0209,
+      "step": 4597
+    },
+    {
+      "epoch": 270.47,
+      "learning_rate": 5.180598555211558e-06,
+      "loss": 0.0164,
+      "step": 4598
+    },
+    {
+      "epoch": 270.53,
+      "learning_rate": 5.170278637770898e-06,
+      "loss": 0.0209,
+      "step": 4599
+    },
+    {
+      "epoch": 270.59,
+      "learning_rate": 5.159958720330237e-06,
+      "loss": 0.018,
+      "step": 4600
+    },
+    {
+      "epoch": 270.59,
+      "eval_accuracy_NAT": 0.9938310930571412,
+      "eval_accuracy_REST": 0.4292682926829268,
+      "eval_accuracy_concrete": 0.9568267674042094,
+      "eval_accuracy_concrete tiles": 0.9399952863539948,
+      "eval_accuracy_dark bricks": 0.8524060639702926,
+      "eval_accuracy_grass": 0.9017843963256084,
+      "eval_accuracy_red bricks": 0.9287126537785588,
+      "eval_accuracy_road": 0.9820548407145999,
+      "eval_accuracy_rough concrete": 0.6257336773547691,
+      "eval_accuracy_rough red sidewalk": 0.786883643282022,
+      "eval_accuracy_speedway bricks": 0.9903672218212277,
+      "eval_accuracy_steel": 0.9226786347938647,
+      "eval_accuracy_tiles": 0.23935896270714047,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842745896280315,
+      "eval_iou_REST": 0.3922734026745914,
+      "eval_iou_concrete": 0.8937669200607347,
+      "eval_iou_concrete tiles": 0.8820966493420325,
+      "eval_iou_dark bricks": 0.8363806250469537,
+      "eval_iou_grass": 0.8462337606985597,
+      "eval_iou_red bricks": 0.9013378817760247,
+      "eval_iou_road": 0.961380445433576,
+      "eval_iou_rough concrete": 0.6199742587585164,
+      "eval_iou_rough red sidewalk": 0.6244122210287624,
+      "eval_iou_speedway bricks": 0.9798541971637655,
+      "eval_iou_steel": 0.873557997142019,
+      "eval_iou_tiles": 0.2377241545312312,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08349091559648514,
+      "eval_mean_accuracy": 0.8115308872497197,
+      "eval_mean_iou": 0.7717897771757539,
+      "eval_overall_accuracy": 0.9804994292178396,
+      "eval_runtime": 5.3134,
+      "eval_samples_per_second": 22.208,
+      "eval_steps_per_second": 1.506,
+      "step": 4600
+    },
+    {
+      "epoch": 270.65,
+      "learning_rate": 5.1496388028895774e-06,
+      "loss": 0.0165,
+      "step": 4601
+    },
+    {
+      "epoch": 270.71,
+      "learning_rate": 5.139318885448917e-06,
+      "loss": 0.0215,
+      "step": 4602
+    },
+    {
+      "epoch": 270.76,
+      "learning_rate": 5.128998968008256e-06,
+      "loss": 0.0164,
+      "step": 4603
+    },
+    {
+      "epoch": 270.82,
+      "learning_rate": 5.118679050567596e-06,
+      "loss": 0.019,
+      "step": 4604
+    },
+    {
+      "epoch": 270.88,
+      "learning_rate": 5.1083591331269355e-06,
+      "loss": 0.0192,
+      "step": 4605
+    },
+    {
+      "epoch": 270.94,
+      "learning_rate": 5.098039215686274e-06,
+      "loss": 0.0157,
+      "step": 4606
+    },
+    {
+      "epoch": 271.0,
+      "learning_rate": 5.087719298245614e-06,
+      "loss": 0.0204,
+      "step": 4607
+    },
+    {
+      "epoch": 271.06,
+      "learning_rate": 5.077399380804954e-06,
+      "loss": 0.0173,
+      "step": 4608
+    },
+    {
+      "epoch": 271.12,
+      "learning_rate": 5.067079463364293e-06,
+      "loss": 0.0172,
+      "step": 4609
+    },
+    {
+      "epoch": 271.18,
+      "learning_rate": 5.056759545923633e-06,
+      "loss": 0.0205,
+      "step": 4610
+    },
+    {
+      "epoch": 271.24,
+      "learning_rate": 5.046439628482973e-06,
+      "loss": 0.0183,
+      "step": 4611
+    },
+    {
+      "epoch": 271.29,
+      "learning_rate": 5.036119711042312e-06,
+      "loss": 0.021,
+      "step": 4612
+    },
+    {
+      "epoch": 271.35,
+      "learning_rate": 5.0257997936016515e-06,
+      "loss": 0.018,
+      "step": 4613
+    },
+    {
+      "epoch": 271.41,
+      "learning_rate": 5.015479876160991e-06,
+      "loss": 0.0177,
+      "step": 4614
+    },
+    {
+      "epoch": 271.47,
+      "learning_rate": 5.00515995872033e-06,
+      "loss": 0.0179,
+      "step": 4615
+    },
+    {
+      "epoch": 271.53,
+      "learning_rate": 4.99484004127967e-06,
+      "loss": 0.0204,
+      "step": 4616
+    },
+    {
+      "epoch": 271.59,
+      "learning_rate": 4.9845201238390096e-06,
+      "loss": 0.0163,
+      "step": 4617
+    },
+    {
+      "epoch": 271.65,
+      "learning_rate": 4.9742002063983484e-06,
+      "loss": 0.0188,
+      "step": 4618
+    },
+    {
+      "epoch": 271.71,
+      "learning_rate": 4.963880288957688e-06,
+      "loss": 0.0198,
+      "step": 4619
+    },
+    {
+      "epoch": 271.76,
+      "learning_rate": 4.953560371517029e-06,
+      "loss": 0.0223,
+      "step": 4620
+    },
+    {
+      "epoch": 271.76,
+      "eval_accuracy_NAT": 0.9936796549259255,
+      "eval_accuracy_REST": 0.408130081300813,
+      "eval_accuracy_concrete": 0.9571432284221743,
+      "eval_accuracy_concrete tiles": 0.9405844921046429,
+      "eval_accuracy_dark bricks": 0.8527888903776583,
+      "eval_accuracy_grass": 0.9022745423096284,
+      "eval_accuracy_red bricks": 0.9290971001757469,
+      "eval_accuracy_road": 0.9811329728874664,
+      "eval_accuracy_rough concrete": 0.6253642317020123,
+      "eval_accuracy_rough red sidewalk": 0.7920573925206236,
+      "eval_accuracy_speedway bricks": 0.9903002180450252,
+      "eval_accuracy_steel": 0.9256823281872385,
+      "eval_accuracy_tiles": 0.24466054712351176,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842608250382844,
+      "eval_iou_REST": 0.3768768768768769,
+      "eval_iou_concrete": 0.8937453627419806,
+      "eval_iou_concrete tiles": 0.8821618037135278,
+      "eval_iou_dark bricks": 0.8363871028300558,
+      "eval_iou_grass": 0.8445781489144606,
+      "eval_iou_red bricks": 0.9016629357211384,
+      "eval_iou_road": 0.9609701737066741,
+      "eval_iou_rough concrete": 0.6193705257815313,
+      "eval_iou_rough red sidewalk": 0.6266331120154208,
+      "eval_iou_speedway bricks": 0.9799449124383156,
+      "eval_iou_steel": 0.8734487921623183,
+      "eval_iou_tiles": 0.24250067561480948,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08391601592302322,
+      "eval_mean_accuracy": 0.8109919753909589,
+      "eval_mean_iou": 0.7709647113504149,
+      "eval_overall_accuracy": 0.9804516808461334,
+      "eval_runtime": 5.4813,
+      "eval_samples_per_second": 21.528,
+      "eval_steps_per_second": 1.46,
+      "step": 4620
+    },
+    {
+      "epoch": 271.82,
+      "learning_rate": 4.943240454076368e-06,
+      "loss": 0.0197,
+      "step": 4621
+    },
+    {
+      "epoch": 271.88,
+      "learning_rate": 4.932920536635707e-06,
+      "loss": 0.016,
+      "step": 4622
+    },
+    {
+      "epoch": 271.94,
+      "learning_rate": 4.922600619195047e-06,
+      "loss": 0.0175,
+      "step": 4623
+    },
+    {
+      "epoch": 272.0,
+      "learning_rate": 4.912280701754386e-06,
+      "loss": 0.0214,
+      "step": 4624
+    },
+    {
+      "epoch": 272.06,
+      "learning_rate": 4.901960784313726e-06,
+      "loss": 0.0176,
+      "step": 4625
+    },
+    {
+      "epoch": 272.12,
+      "learning_rate": 4.891640866873065e-06,
+      "loss": 0.0156,
+      "step": 4626
+    },
+    {
+      "epoch": 272.18,
+      "learning_rate": 4.881320949432404e-06,
+      "loss": 0.021,
+      "step": 4627
+    },
+    {
+      "epoch": 272.24,
+      "learning_rate": 4.871001031991744e-06,
+      "loss": 0.0187,
+      "step": 4628
+    },
+    {
+      "epoch": 272.29,
+      "learning_rate": 4.8606811145510845e-06,
+      "loss": 0.0177,
+      "step": 4629
+    },
+    {
+      "epoch": 272.35,
+      "learning_rate": 4.850361197110423e-06,
+      "loss": 0.0173,
+      "step": 4630
+    },
+    {
+      "epoch": 272.41,
+      "learning_rate": 4.840041279669763e-06,
+      "loss": 0.0182,
+      "step": 4631
+    },
+    {
+      "epoch": 272.47,
+      "learning_rate": 4.829721362229103e-06,
+      "loss": 0.0172,
+      "step": 4632
+    },
+    {
+      "epoch": 272.53,
+      "learning_rate": 4.819401444788442e-06,
+      "loss": 0.0176,
+      "step": 4633
+    },
+    {
+      "epoch": 272.59,
+      "learning_rate": 4.809081527347781e-06,
+      "loss": 0.0197,
+      "step": 4634
+    },
+    {
+      "epoch": 272.65,
+      "learning_rate": 4.798761609907121e-06,
+      "loss": 0.0202,
+      "step": 4635
+    },
+    {
+      "epoch": 272.71,
+      "learning_rate": 4.78844169246646e-06,
+      "loss": 0.018,
+      "step": 4636
+    },
+    {
+      "epoch": 272.76,
+      "learning_rate": 4.7781217750258e-06,
+      "loss": 0.0166,
+      "step": 4637
+    },
+    {
+      "epoch": 272.82,
+      "learning_rate": 4.7678018575851394e-06,
+      "loss": 0.0201,
+      "step": 4638
+    },
+    {
+      "epoch": 272.88,
+      "learning_rate": 4.757481940144479e-06,
+      "loss": 0.0175,
+      "step": 4639
+    },
+    {
+      "epoch": 272.94,
+      "learning_rate": 4.747162022703819e-06,
+      "loss": 0.0227,
+      "step": 4640
+    },
+    {
+      "epoch": 272.94,
+      "eval_accuracy_NAT": 0.9940208085372937,
+      "eval_accuracy_REST": 0.4,
+      "eval_accuracy_concrete": 0.9574050349406598,
+      "eval_accuracy_concrete tiles": 0.9357294367193024,
+      "eval_accuracy_dark bricks": 0.8552964033459028,
+      "eval_accuracy_grass": 0.8967902061641062,
+      "eval_accuracy_red bricks": 0.9275593145869947,
+      "eval_accuracy_road": 0.9807396426145562,
+      "eval_accuracy_rough concrete": 0.6246193815956477,
+      "eval_accuracy_rough red sidewalk": 0.7850638070672813,
+      "eval_accuracy_speedway bricks": 0.9906216990786724,
+      "eval_accuracy_steel": 0.9235732456381474,
+      "eval_accuracy_tiles": 0.24487261050016662,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841911689740295,
+      "eval_iou_REST": 0.3813953488372093,
+      "eval_iou_concrete": 0.8940609820080591,
+      "eval_iou_concrete tiles": 0.8829560112084687,
+      "eval_iou_dark bricks": 0.8385866301328729,
+      "eval_iou_grass": 0.8434690206890659,
+      "eval_iou_red bricks": 0.9025758871312527,
+      "eval_iou_road": 0.9611775932758039,
+      "eval_iou_rough concrete": 0.6195469079690058,
+      "eval_iou_rough red sidewalk": 0.6245632717526562,
+      "eval_iou_speedway bricks": 0.9800725734164556,
+      "eval_iou_steel": 0.8742788069177763,
+      "eval_iou_tiles": 0.24295891069764647,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08451928943395615,
+      "eval_mean_accuracy": 0.8089455069837487,
+      "eval_mean_iou": 0.7715256240777156,
+      "eval_overall_accuracy": 0.9804759268033303,
+      "eval_runtime": 5.3041,
+      "eval_samples_per_second": 22.247,
+      "eval_steps_per_second": 1.508,
+      "step": 4640
+    },
+    {
+      "epoch": 273.0,
+      "learning_rate": 4.736842105263159e-06,
+      "loss": 0.0159,
+      "step": 4641
+    },
+    {
+      "epoch": 273.06,
+      "learning_rate": 4.7265221878224975e-06,
+      "loss": 0.016,
+      "step": 4642
+    },
+    {
+      "epoch": 273.12,
+      "learning_rate": 4.716202270381837e-06,
+      "loss": 0.0193,
+      "step": 4643
+    },
+    {
+      "epoch": 273.18,
+      "learning_rate": 4.705882352941177e-06,
+      "loss": 0.0192,
+      "step": 4644
+    },
+    {
+      "epoch": 273.24,
+      "learning_rate": 4.695562435500516e-06,
+      "loss": 0.0176,
+      "step": 4645
+    },
+    {
+      "epoch": 273.29,
+      "learning_rate": 4.6852425180598555e-06,
+      "loss": 0.0191,
+      "step": 4646
+    },
+    {
+      "epoch": 273.35,
+      "learning_rate": 4.674922600619195e-06,
+      "loss": 0.0173,
+      "step": 4647
+    },
+    {
+      "epoch": 273.41,
+      "learning_rate": 4.664602683178535e-06,
+      "loss": 0.0214,
+      "step": 4648
+    },
+    {
+      "epoch": 273.47,
+      "learning_rate": 4.654282765737875e-06,
+      "loss": 0.0185,
+      "step": 4649
+    },
+    {
+      "epoch": 273.53,
+      "learning_rate": 4.643962848297214e-06,
+      "loss": 0.0193,
+      "step": 4650
+    },
+    {
+      "epoch": 273.59,
+      "learning_rate": 4.633642930856553e-06,
+      "loss": 0.0167,
+      "step": 4651
+    },
+    {
+      "epoch": 273.65,
+      "learning_rate": 4.623323013415893e-06,
+      "loss": 0.0163,
+      "step": 4652
+    },
+    {
+      "epoch": 273.71,
+      "learning_rate": 4.613003095975233e-06,
+      "loss": 0.0186,
+      "step": 4653
+    },
+    {
+      "epoch": 273.76,
+      "learning_rate": 4.6026831785345716e-06,
+      "loss": 0.0172,
+      "step": 4654
+    },
+    {
+      "epoch": 273.82,
+      "learning_rate": 4.592363261093911e-06,
+      "loss": 0.0162,
+      "step": 4655
+    },
+    {
+      "epoch": 273.88,
+      "learning_rate": 4.582043343653251e-06,
+      "loss": 0.0184,
+      "step": 4656
+    },
+    {
+      "epoch": 273.94,
+      "learning_rate": 4.57172342621259e-06,
+      "loss": 0.0205,
+      "step": 4657
+    },
+    {
+      "epoch": 274.0,
+      "learning_rate": 4.5614035087719304e-06,
+      "loss": 0.018,
+      "step": 4658
+    },
+    {
+      "epoch": 274.06,
+      "learning_rate": 4.55108359133127e-06,
+      "loss": 0.0167,
+      "step": 4659
+    },
+    {
+      "epoch": 274.12,
+      "learning_rate": 4.540763673890609e-06,
+      "loss": 0.0165,
+      "step": 4660
+    },
+    {
+      "epoch": 274.12,
+      "eval_accuracy_NAT": 0.9940100800972396,
+      "eval_accuracy_REST": 0.4032520325203252,
+      "eval_accuracy_concrete": 0.9575590612573746,
+      "eval_accuracy_concrete tiles": 0.9453452745698798,
+      "eval_accuracy_dark bricks": 0.8601248014088012,
+      "eval_accuracy_grass": 0.8980042163909359,
+      "eval_accuracy_red bricks": 0.9318431458699473,
+      "eval_accuracy_road": 0.9805007586036989,
+      "eval_accuracy_rough concrete": 0.6254238197105214,
+      "eval_accuracy_rough red sidewalk": 0.795697064950105,
+      "eval_accuracy_speedway bricks": 0.9899991294650048,
+      "eval_accuracy_steel": 0.9278089031271397,
+      "eval_accuracy_tiles": 0.25405192523251235,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842153255058979,
+      "eval_iou_REST": 0.37575757575757573,
+      "eval_iou_concrete": 0.8945796597629967,
+      "eval_iou_concrete tiles": 0.8774144153997594,
+      "eval_iou_dark bricks": 0.843265306122449,
+      "eval_iou_grass": 0.8437683094778526,
+      "eval_iou_red bricks": 0.9025,
+      "eval_iou_road": 0.9615105248097904,
+      "eval_iou_rough concrete": 0.619897823583262,
+      "eval_iou_rough red sidewalk": 0.6288018351022737,
+      "eval_iou_speedway bricks": 0.9802337338280824,
+      "eval_iou_steel": 0.8743462155269249,
+      "eval_iou_tiles": 0.25169578005882703,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08500112593173981,
+      "eval_mean_accuracy": 0.812586170246422,
+      "eval_mean_iou": 0.7721528080719762,
+      "eval_overall_accuracy": 0.9805551625914493,
+      "eval_runtime": 5.4024,
+      "eval_samples_per_second": 21.842,
+      "eval_steps_per_second": 1.481,
+      "step": 4660
+    },
+    {
+      "epoch": 274.18,
+      "learning_rate": 4.530443756449949e-06,
+      "loss": 0.0161,
+      "step": 4661
+    },
+    {
+      "epoch": 274.24,
+      "learning_rate": 4.5201238390092885e-06,
+      "loss": 0.018,
+      "step": 4662
+    },
+    {
+      "epoch": 274.29,
+      "learning_rate": 4.509803921568627e-06,
+      "loss": 0.0195,
+      "step": 4663
+    },
+    {
+      "epoch": 274.35,
+      "learning_rate": 4.499484004127967e-06,
+      "loss": 0.0222,
+      "step": 4664
+    },
+    {
+      "epoch": 274.41,
+      "learning_rate": 4.489164086687307e-06,
+      "loss": 0.0159,
+      "step": 4665
+    },
+    {
+      "epoch": 274.47,
+      "learning_rate": 4.478844169246646e-06,
+      "loss": 0.0177,
+      "step": 4666
+    },
+    {
+      "epoch": 274.53,
+      "learning_rate": 4.468524251805986e-06,
+      "loss": 0.0204,
+      "step": 4667
+    },
+    {
+      "epoch": 274.59,
+      "learning_rate": 4.458204334365326e-06,
+      "loss": 0.018,
+      "step": 4668
+    },
+    {
+      "epoch": 274.65,
+      "learning_rate": 4.447884416924665e-06,
+      "loss": 0.0203,
+      "step": 4669
+    },
+    {
+      "epoch": 274.71,
+      "learning_rate": 4.4375644994840045e-06,
+      "loss": 0.0199,
+      "step": 4670
+    },
+    {
+      "epoch": 274.76,
+      "learning_rate": 4.427244582043344e-06,
+      "loss": 0.0204,
+      "step": 4671
+    },
+    {
+      "epoch": 274.82,
+      "learning_rate": 4.416924664602683e-06,
+      "loss": 0.0176,
+      "step": 4672
+    },
+    {
+      "epoch": 274.88,
+      "learning_rate": 4.406604747162023e-06,
+      "loss": 0.0162,
+      "step": 4673
+    },
+    {
+      "epoch": 274.94,
+      "learning_rate": 4.3962848297213626e-06,
+      "loss": 0.0198,
+      "step": 4674
+    },
+    {
+      "epoch": 275.0,
+      "learning_rate": 4.3859649122807014e-06,
+      "loss": 0.0187,
+      "step": 4675
+    },
+    {
+      "epoch": 275.06,
+      "learning_rate": 4.375644994840041e-06,
+      "loss": 0.0204,
+      "step": 4676
+    },
+    {
+      "epoch": 275.12,
+      "learning_rate": 4.365325077399382e-06,
+      "loss": 0.02,
+      "step": 4677
+    },
+    {
+      "epoch": 275.18,
+      "learning_rate": 4.355005159958721e-06,
+      "loss": 0.0191,
+      "step": 4678
+    },
+    {
+      "epoch": 275.24,
+      "learning_rate": 4.34468524251806e-06,
+      "loss": 0.0189,
+      "step": 4679
+    },
+    {
+      "epoch": 275.29,
+      "learning_rate": 4.3343653250774e-06,
+      "loss": 0.0192,
+      "step": 4680
+    },
+    {
+      "epoch": 275.29,
+      "eval_accuracy_NAT": 0.9941409239364921,
+      "eval_accuracy_REST": 0.4032520325203252,
+      "eval_accuracy_concrete": 0.9570599089754154,
+      "eval_accuracy_concrete tiles": 0.9375913268913505,
+      "eval_accuracy_dark bricks": 0.8547460903853148,
+      "eval_accuracy_grass": 0.8986145333015401,
+      "eval_accuracy_red bricks": 0.929756151142355,
+      "eval_accuracy_road": 0.9805328235715992,
+      "eval_accuracy_rough concrete": 0.6271220779530328,
+      "eval_accuracy_rough red sidewalk": 0.7904029959617684,
+      "eval_accuracy_speedway bricks": 0.9903000466798686,
+      "eval_accuracy_steel": 0.9243004293132487,
+      "eval_accuracy_tiles": 0.2443878942106698,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842155841584211,
+      "eval_iou_REST": 0.3887147335423197,
+      "eval_iou_concrete": 0.8945748968184022,
+      "eval_iou_concrete tiles": 0.878752402200084,
+      "eval_iou_dark bricks": 0.8394215812056187,
+      "eval_iou_grass": 0.8444498784028311,
+      "eval_iou_red bricks": 0.9024949354941891,
+      "eval_iou_road": 0.9616604024376246,
+      "eval_iou_rough concrete": 0.6213865665296868,
+      "eval_iou_rough red sidewalk": 0.6275157166994035,
+      "eval_iou_speedway bricks": 0.9800544289261912,
+      "eval_iou_steel": 0.8747271510430339,
+      "eval_iou_tiles": 0.24263843354287604,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08358141034841537,
+      "eval_mean_accuracy": 0.8101697872956138,
+      "eval_mean_iou": 0.7723543623846679,
+      "eval_overall_accuracy": 0.9805535138663599,
+      "eval_runtime": 5.4124,
+      "eval_samples_per_second": 21.802,
+      "eval_steps_per_second": 1.478,
+      "step": 4680
+    },
+    {
+      "epoch": 275.35,
+      "learning_rate": 4.324045407636739e-06,
+      "loss": 0.0191,
+      "step": 4681
+    },
+    {
+      "epoch": 275.41,
+      "learning_rate": 4.313725490196079e-06,
+      "loss": 0.0153,
+      "step": 4682
+    },
+    {
+      "epoch": 275.47,
+      "learning_rate": 4.303405572755418e-06,
+      "loss": 0.0164,
+      "step": 4683
+    },
+    {
+      "epoch": 275.53,
+      "learning_rate": 4.293085655314757e-06,
+      "loss": 0.0202,
+      "step": 4684
+    },
+    {
+      "epoch": 275.59,
+      "learning_rate": 4.282765737874097e-06,
+      "loss": 0.0183,
+      "step": 4685
+    },
+    {
+      "epoch": 275.65,
+      "learning_rate": 4.2724458204334375e-06,
+      "loss": 0.0188,
+      "step": 4686
+    },
+    {
+      "epoch": 275.71,
+      "learning_rate": 4.262125902992776e-06,
+      "loss": 0.0189,
+      "step": 4687
+    },
+    {
+      "epoch": 275.76,
+      "learning_rate": 4.251805985552116e-06,
+      "loss": 0.0166,
+      "step": 4688
+    },
+    {
+      "epoch": 275.82,
+      "learning_rate": 4.241486068111456e-06,
+      "loss": 0.0202,
+      "step": 4689
+    },
+    {
+      "epoch": 275.88,
+      "learning_rate": 4.231166150670795e-06,
+      "loss": 0.0184,
+      "step": 4690
+    },
+    {
+      "epoch": 275.94,
+      "learning_rate": 4.220846233230134e-06,
+      "loss": 0.0176,
+      "step": 4691
+    },
+    {
+      "epoch": 276.0,
+      "learning_rate": 4.210526315789474e-06,
+      "loss": 0.019,
+      "step": 4692
+    },
+    {
+      "epoch": 276.06,
+      "learning_rate": 4.200206398348813e-06,
+      "loss": 0.0177,
+      "step": 4693
+    },
+    {
+      "epoch": 276.12,
+      "learning_rate": 4.189886480908153e-06,
+      "loss": 0.018,
+      "step": 4694
+    },
+    {
+      "epoch": 276.18,
+      "learning_rate": 4.1795665634674924e-06,
+      "loss": 0.0194,
+      "step": 4695
+    },
+    {
+      "epoch": 276.24,
+      "learning_rate": 4.169246646026832e-06,
+      "loss": 0.0205,
+      "step": 4696
+    },
+    {
+      "epoch": 276.29,
+      "learning_rate": 4.158926728586172e-06,
+      "loss": 0.0179,
+      "step": 4697
+    },
+    {
+      "epoch": 276.35,
+      "learning_rate": 4.148606811145512e-06,
+      "loss": 0.0182,
+      "step": 4698
+    },
+    {
+      "epoch": 276.41,
+      "learning_rate": 4.1382868937048505e-06,
+      "loss": 0.019,
+      "step": 4699
+    },
+    {
+      "epoch": 276.47,
+      "learning_rate": 4.12796697626419e-06,
+      "loss": 0.0172,
+      "step": 4700
+    },
+    {
+      "epoch": 276.47,
+      "eval_accuracy_NAT": 0.9939741748657017,
+      "eval_accuracy_REST": 0.40487804878048783,
+      "eval_accuracy_concrete": 0.9572521552218362,
+      "eval_accuracy_concrete tiles": 0.933349045486684,
+      "eval_accuracy_dark bricks": 0.8524682732614896,
+      "eval_accuracy_grass": 0.8995219657314152,
+      "eval_accuracy_red bricks": 0.9312390158172232,
+      "eval_accuracy_road": 0.9817582397615221,
+      "eval_accuracy_rough concrete": 0.6277596696440808,
+      "eval_accuracy_rough red sidewalk": 0.7866054038607599,
+      "eval_accuracy_speedway bricks": 0.989350512347545,
+      "eval_accuracy_steel": 0.9314898019361578,
+      "eval_accuracy_tiles": 0.23905601502620497,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984212925429753,
+      "eval_iou_REST": 0.3878504672897196,
+      "eval_iou_concrete": 0.8942779314902829,
+      "eval_iou_concrete tiles": 0.877898470405675,
+      "eval_iou_dark bricks": 0.8361040448320207,
+      "eval_iou_grass": 0.8443033680123203,
+      "eval_iou_red bricks": 0.9031158455392809,
+      "eval_iou_road": 0.9609191638582034,
+      "eval_iou_rough concrete": 0.6212040804292706,
+      "eval_iou_rough red sidewalk": 0.6244671832650771,
+      "eval_iou_speedway bricks": 0.9798422265402311,
+      "eval_iou_steel": 0.8740155080201365,
+      "eval_iou_tiles": 0.23740898971057223,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08429407328367233,
+      "eval_mean_accuracy": 0.8099001785954698,
+      "eval_mean_iou": 0.7712015542171188,
+      "eval_overall_accuracy": 0.9804499351372153,
+      "eval_runtime": 5.2965,
+      "eval_samples_per_second": 22.279,
+      "eval_steps_per_second": 1.51,
+      "step": 4700
+    },
+    {
+      "epoch": 276.53,
+      "learning_rate": 4.11764705882353e-06,
+      "loss": 0.0178,
+      "step": 4701
+    },
+    {
+      "epoch": 276.59,
+      "learning_rate": 4.107327141382869e-06,
+      "loss": 0.0189,
+      "step": 4702
+    },
+    {
+      "epoch": 276.65,
+      "learning_rate": 4.0970072239422085e-06,
+      "loss": 0.0199,
+      "step": 4703
+    },
+    {
+      "epoch": 276.71,
+      "learning_rate": 4.086687306501548e-06,
+      "loss": 0.0171,
+      "step": 4704
+    },
+    {
+      "epoch": 276.76,
+      "learning_rate": 4.076367389060888e-06,
+      "loss": 0.0187,
+      "step": 4705
+    },
+    {
+      "epoch": 276.82,
+      "learning_rate": 4.066047471620228e-06,
+      "loss": 0.0169,
+      "step": 4706
+    },
+    {
+      "epoch": 276.88,
+      "learning_rate": 4.055727554179567e-06,
+      "loss": 0.0177,
+      "step": 4707
+    },
+    {
+      "epoch": 276.94,
+      "learning_rate": 4.045407636738906e-06,
+      "loss": 0.0164,
+      "step": 4708
+    },
+    {
+      "epoch": 277.0,
+      "learning_rate": 4.035087719298246e-06,
+      "loss": 0.0195,
+      "step": 4709
+    },
+    {
+      "epoch": 277.06,
+      "learning_rate": 4.024767801857586e-06,
+      "loss": 0.0164,
+      "step": 4710
+    },
+    {
+      "epoch": 277.12,
+      "learning_rate": 4.0144478844169246e-06,
+      "loss": 0.0186,
+      "step": 4711
+    },
+    {
+      "epoch": 277.18,
+      "learning_rate": 4.004127966976264e-06,
+      "loss": 0.0188,
+      "step": 4712
+    },
+    {
+      "epoch": 277.24,
+      "learning_rate": 3.993808049535604e-06,
+      "loss": 0.018,
+      "step": 4713
+    },
+    {
+      "epoch": 277.29,
+      "learning_rate": 3.983488132094943e-06,
+      "loss": 0.0178,
+      "step": 4714
+    },
+    {
+      "epoch": 277.35,
+      "learning_rate": 3.9731682146542834e-06,
+      "loss": 0.0162,
+      "step": 4715
+    },
+    {
+      "epoch": 277.41,
+      "learning_rate": 3.962848297213622e-06,
+      "loss": 0.018,
+      "step": 4716
+    },
+    {
+      "epoch": 277.47,
+      "learning_rate": 3.952528379772962e-06,
+      "loss": 0.0184,
+      "step": 4717
+    },
+    {
+      "epoch": 277.53,
+      "learning_rate": 3.942208462332302e-06,
+      "loss": 0.0205,
+      "step": 4718
+    },
+    {
+      "epoch": 277.59,
+      "learning_rate": 3.9318885448916415e-06,
+      "loss": 0.0181,
+      "step": 4719
+    },
+    {
+      "epoch": 277.65,
+      "learning_rate": 3.92156862745098e-06,
+      "loss": 0.0173,
+      "step": 4720
+    },
+    {
+      "epoch": 277.65,
+      "eval_accuracy_NAT": 0.9941813577557915,
+      "eval_accuracy_REST": 0.3983739837398374,
+      "eval_accuracy_concrete": 0.9576263283336569,
+      "eval_accuracy_concrete tiles": 0.9423992458166391,
+      "eval_accuracy_dark bricks": 0.8548417969871561,
+      "eval_accuracy_grass": 0.8967779052031173,
+      "eval_accuracy_red bricks": 0.9320628295254832,
+      "eval_accuracy_road": 0.9812200827169288,
+      "eval_accuracy_rough concrete": 0.626812220308785,
+      "eval_accuracy_rough red sidewalk": 0.7831537310402394,
+      "eval_accuracy_speedway bricks": 0.9892422095686191,
+      "eval_accuracy_steel": 0.9319146171415434,
+      "eval_accuracy_tiles": 0.24826562452664425,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841675765591904,
+      "eval_iou_REST": 0.38341158059467917,
+      "eval_iou_concrete": 0.8948012937929153,
+      "eval_iou_concrete tiles": 0.87815698159617,
+      "eval_iou_dark bricks": 0.8391369907413931,
+      "eval_iou_grass": 0.8435565484799801,
+      "eval_iou_red bricks": 0.9036741214057508,
+      "eval_iou_road": 0.960887504474575,
+      "eval_iou_rough concrete": 0.6214684926327232,
+      "eval_iou_rough red sidewalk": 0.6229542518064795,
+      "eval_iou_speedway bricks": 0.9800258727380453,
+      "eval_iou_steel": 0.8737646612278178,
+      "eval_iou_tiles": 0.2464439298709891,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08508964627981186,
+      "eval_mean_accuracy": 0.810528610204957,
+      "eval_mean_iou": 0.7717269081477469,
+      "eval_overall_accuracy": 0.9804832329184322,
+      "eval_runtime": 5.4848,
+      "eval_samples_per_second": 21.514,
+      "eval_steps_per_second": 1.459,
+      "step": 4720
+    },
+    {
+      "epoch": 277.71,
+      "learning_rate": 3.91124871001032e-06,
+      "loss": 0.0174,
+      "step": 4721
+    },
+    {
+      "epoch": 277.76,
+      "learning_rate": 3.90092879256966e-06,
+      "loss": 0.0157,
+      "step": 4722
+    },
+    {
+      "epoch": 277.82,
+      "learning_rate": 3.890608875128999e-06,
+      "loss": 0.0226,
+      "step": 4723
+    },
+    {
+      "epoch": 277.88,
+      "learning_rate": 3.880288957688338e-06,
+      "loss": 0.018,
+      "step": 4724
+    },
+    {
+      "epoch": 277.94,
+      "learning_rate": 3.869969040247678e-06,
+      "loss": 0.0201,
+      "step": 4725
+    },
+    {
+      "epoch": 278.0,
+      "learning_rate": 3.859649122807018e-06,
+      "loss": 0.0201,
+      "step": 4726
+    },
+    {
+      "epoch": 278.06,
+      "learning_rate": 3.8493292053663575e-06,
+      "loss": 0.0187,
+      "step": 4727
+    },
+    {
+      "epoch": 278.12,
+      "learning_rate": 3.839009287925697e-06,
+      "loss": 0.0172,
+      "step": 4728
+    },
+    {
+      "epoch": 278.18,
+      "learning_rate": 3.828689370485036e-06,
+      "loss": 0.019,
+      "step": 4729
+    },
+    {
+      "epoch": 278.24,
+      "learning_rate": 3.818369453044376e-06,
+      "loss": 0.0174,
+      "step": 4730
+    },
+    {
+      "epoch": 278.29,
+      "learning_rate": 3.808049535603715e-06,
+      "loss": 0.0173,
+      "step": 4731
+    },
+    {
+      "epoch": 278.35,
+      "learning_rate": 3.797729618163055e-06,
+      "loss": 0.0163,
+      "step": 4732
+    },
+    {
+      "epoch": 278.41,
+      "learning_rate": 3.787409700722394e-06,
+      "loss": 0.0181,
+      "step": 4733
+    },
+    {
+      "epoch": 278.47,
+      "learning_rate": 3.7770897832817343e-06,
+      "loss": 0.0194,
+      "step": 4734
+    },
+    {
+      "epoch": 278.53,
+      "learning_rate": 3.7667698658410736e-06,
+      "loss": 0.0172,
+      "step": 4735
+    },
+    {
+      "epoch": 278.59,
+      "learning_rate": 3.7564499484004133e-06,
+      "loss": 0.0205,
+      "step": 4736
+    },
+    {
+      "epoch": 278.65,
+      "learning_rate": 3.7461300309597526e-06,
+      "loss": 0.016,
+      "step": 4737
+    },
+    {
+      "epoch": 278.71,
+      "learning_rate": 3.735810113519092e-06,
+      "loss": 0.0171,
+      "step": 4738
+    },
+    {
+      "epoch": 278.76,
+      "learning_rate": 3.7254901960784316e-06,
+      "loss": 0.0194,
+      "step": 4739
+    },
+    {
+      "epoch": 278.82,
+      "learning_rate": 3.715170278637771e-06,
+      "loss": 0.0183,
+      "step": 4740
+    },
+    {
+      "epoch": 278.82,
+      "eval_accuracy_NAT": 0.994098980587939,
+      "eval_accuracy_REST": 0.3951219512195122,
+      "eval_accuracy_concrete": 0.9570278042344624,
+      "eval_accuracy_concrete tiles": 0.9430120197973132,
+      "eval_accuracy_dark bricks": 0.854626457133013,
+      "eval_accuracy_grass": 0.8971932991934355,
+      "eval_accuracy_red bricks": 0.9328317223198594,
+      "eval_accuracy_road": 0.9813841484693521,
+      "eval_accuracy_rough concrete": 0.626812220308785,
+      "eval_accuracy_rough red sidewalk": 0.7811158152791042,
+      "eval_accuracy_speedway bricks": 0.9900926948404694,
+      "eval_accuracy_steel": 0.9249476477732187,
+      "eval_accuracy_tiles": 0.25053773213366054,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841713280739576,
+      "eval_iou_REST": 0.3767441860465116,
+      "eval_iou_concrete": 0.8946525970986696,
+      "eval_iou_concrete tiles": 0.8781301437506859,
+      "eval_iou_dark bricks": 0.8386262144356426,
+      "eval_iou_grass": 0.8437310074008292,
+      "eval_iou_red bricks": 0.9042751424160145,
+      "eval_iou_road": 0.9608998210969546,
+      "eval_iou_rough concrete": 0.6207716639520336,
+      "eval_iou_rough red sidewalk": 0.6214781885519065,
+      "eval_iou_speedway bricks": 0.9799436187240382,
+      "eval_iou_steel": 0.8736929068806798,
+      "eval_iou_tiles": 0.24846026738771218,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08438953757286072,
+      "eval_mean_accuracy": 0.8099078840992402,
+      "eval_mean_iou": 0.7711982373704337,
+      "eval_overall_accuracy": 0.9804710776118909,
+      "eval_runtime": 5.5715,
+      "eval_samples_per_second": 21.179,
+      "eval_steps_per_second": 1.436,
+      "step": 4740
+    },
+    {
+      "epoch": 278.88,
+      "learning_rate": 3.70485036119711e-06,
+      "loss": 0.02,
+      "step": 4741
+    },
+    {
+      "epoch": 278.94,
+      "learning_rate": 3.69453044375645e-06,
+      "loss": 0.0186,
+      "step": 4742
+    },
+    {
+      "epoch": 279.0,
+      "learning_rate": 3.6842105263157892e-06,
+      "loss": 0.0177,
+      "step": 4743
+    },
+    {
+      "epoch": 279.06,
+      "learning_rate": 3.6738906088751294e-06,
+      "loss": 0.0161,
+      "step": 4744
+    },
+    {
+      "epoch": 279.12,
+      "learning_rate": 3.663570691434469e-06,
+      "loss": 0.0146,
+      "step": 4745
+    },
+    {
+      "epoch": 279.18,
+      "learning_rate": 3.6532507739938084e-06,
+      "loss": 0.0178,
+      "step": 4746
+    },
+    {
+      "epoch": 279.24,
+      "learning_rate": 3.6429308565531477e-06,
+      "loss": 0.0191,
+      "step": 4747
+    },
+    {
+      "epoch": 279.29,
+      "learning_rate": 3.6326109391124874e-06,
+      "loss": 0.0194,
+      "step": 4748
+    },
+    {
+      "epoch": 279.35,
+      "learning_rate": 3.6222910216718267e-06,
+      "loss": 0.0194,
+      "step": 4749
+    },
+    {
+      "epoch": 279.41,
+      "learning_rate": 3.611971104231166e-06,
+      "loss": 0.0162,
+      "step": 4750
+    },
+    {
+      "epoch": 279.47,
+      "learning_rate": 3.6016511867905057e-06,
+      "loss": 0.0213,
+      "step": 4751
+    },
+    {
+      "epoch": 279.53,
+      "learning_rate": 3.591331269349845e-06,
+      "loss": 0.0182,
+      "step": 4752
+    },
+    {
+      "epoch": 279.59,
+      "learning_rate": 3.581011351909185e-06,
+      "loss": 0.0165,
+      "step": 4753
+    },
+    {
+      "epoch": 279.65,
+      "learning_rate": 3.570691434468525e-06,
+      "loss": 0.0193,
+      "step": 4754
+    },
+    {
+      "epoch": 279.71,
+      "learning_rate": 3.560371517027864e-06,
+      "loss": 0.0221,
+      "step": 4755
+    },
+    {
+      "epoch": 279.76,
+      "learning_rate": 3.5500515995872035e-06,
+      "loss": 0.0164,
+      "step": 4756
+    },
+    {
+      "epoch": 279.82,
+      "learning_rate": 3.539731682146543e-06,
+      "loss": 0.0178,
+      "step": 4757
+    },
+    {
+      "epoch": 279.88,
+      "learning_rate": 3.5294117647058825e-06,
+      "loss": 0.0186,
+      "step": 4758
+    },
+    {
+      "epoch": 279.94,
+      "learning_rate": 3.5190918472652218e-06,
+      "loss": 0.0194,
+      "step": 4759
+    },
+    {
+      "epoch": 280.0,
+      "learning_rate": 3.5087719298245615e-06,
+      "loss": 0.0193,
+      "step": 4760
+    },
+    {
+      "epoch": 280.0,
+      "eval_accuracy_NAT": 0.9940887373537165,
+      "eval_accuracy_REST": 0.41788617886178864,
+      "eval_accuracy_concrete": 0.9564954006136592,
+      "eval_accuracy_concrete tiles": 0.9445439547489983,
+      "eval_accuracy_dark bricks": 0.8497119231284574,
+      "eval_accuracy_grass": 0.8988104024495944,
+      "eval_accuracy_red bricks": 0.9307996485061512,
+      "eval_accuracy_road": 0.9817854949842374,
+      "eval_accuracy_rough concrete": 0.6264308570543264,
+      "eval_accuracy_rough red sidewalk": 0.7781003015513728,
+      "eval_accuracy_speedway bricks": 0.9901170286926964,
+      "eval_accuracy_steel": 0.9254424325418443,
+      "eval_accuracy_tiles": 0.252052470538338,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9841708523804727,
+      "eval_iou_REST": 0.3984496124031008,
+      "eval_iou_concrete": 0.8944831113574928,
+      "eval_iou_concrete tiles": 0.8805809456846546,
+      "eval_iou_dark bricks": 0.8352588104690764,
+      "eval_iou_grass": 0.8440275345359034,
+      "eval_iou_red bricks": 0.9032189298657003,
+      "eval_iou_road": 0.960672982193394,
+      "eval_iou_rough concrete": 0.6205001682180106,
+      "eval_iou_rough red sidewalk": 0.6243799707937581,
+      "eval_iou_speedway bricks": 0.9799369550860109,
+      "eval_iou_steel": 0.8736653377055368,
+      "eval_iou_tiles": 0.24998497686437113,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08378135412931442,
+      "eval_mean_accuracy": 0.8112511408480908,
+      "eval_mean_iou": 0.7730253990428833,
+      "eval_overall_accuracy": 0.9804651615983349,
+      "eval_runtime": 5.6969,
+      "eval_samples_per_second": 20.713,
+      "eval_steps_per_second": 1.404,
+      "step": 4760
+    },
+    {
+      "epoch": 280.06,
+      "learning_rate": 3.4984520123839008e-06,
+      "loss": 0.0218,
+      "step": 4761
+    },
+    {
+      "epoch": 280.12,
+      "learning_rate": 3.48813209494324e-06,
+      "loss": 0.0194,
+      "step": 4762
+    },
+    {
+      "epoch": 280.18,
+      "learning_rate": 3.4778121775025802e-06,
+      "loss": 0.0171,
+      "step": 4763
+    },
+    {
+      "epoch": 280.24,
+      "learning_rate": 3.46749226006192e-06,
+      "loss": 0.0169,
+      "step": 4764
+    },
+    {
+      "epoch": 280.29,
+      "learning_rate": 3.4571723426212592e-06,
+      "loss": 0.0173,
+      "step": 4765
+    },
+    {
+      "epoch": 280.35,
+      "learning_rate": 3.446852425180599e-06,
+      "loss": 0.0182,
+      "step": 4766
+    },
+    {
+      "epoch": 280.41,
+      "learning_rate": 3.4365325077399382e-06,
+      "loss": 0.0191,
+      "step": 4767
+    },
+    {
+      "epoch": 280.47,
+      "learning_rate": 3.4262125902992775e-06,
+      "loss": 0.0155,
+      "step": 4768
+    },
+    {
+      "epoch": 280.53,
+      "learning_rate": 3.4158926728586173e-06,
+      "loss": 0.019,
+      "step": 4769
+    },
+    {
+      "epoch": 280.59,
+      "learning_rate": 3.4055727554179566e-06,
+      "loss": 0.017,
+      "step": 4770
+    },
+    {
+      "epoch": 280.65,
+      "learning_rate": 3.395252837977296e-06,
+      "loss": 0.0195,
+      "step": 4771
+    },
+    {
+      "epoch": 280.71,
+      "learning_rate": 3.384932920536636e-06,
+      "loss": 0.0178,
+      "step": 4772
+    },
+    {
+      "epoch": 280.76,
+      "learning_rate": 3.3746130030959757e-06,
+      "loss": 0.0196,
+      "step": 4773
+    },
+    {
+      "epoch": 280.82,
+      "learning_rate": 3.364293085655315e-06,
+      "loss": 0.0192,
+      "step": 4774
+    },
+    {
+      "epoch": 280.88,
+      "learning_rate": 3.3539731682146547e-06,
+      "loss": 0.0192,
+      "step": 4775
+    },
+    {
+      "epoch": 280.94,
+      "learning_rate": 3.343653250773994e-06,
+      "loss": 0.0174,
+      "step": 4776
+    },
+    {
+      "epoch": 281.0,
+      "learning_rate": 3.3333333333333333e-06,
+      "loss": 0.019,
+      "step": 4777
+    },
+    {
+      "epoch": 281.06,
+      "learning_rate": 3.323013415892673e-06,
+      "loss": 0.0217,
+      "step": 4778
+    },
+    {
+      "epoch": 281.12,
+      "learning_rate": 3.3126934984520123e-06,
+      "loss": 0.0194,
+      "step": 4779
+    },
+    {
+      "epoch": 281.18,
+      "learning_rate": 3.3023735810113516e-06,
+      "loss": 0.0207,
+      "step": 4780
+    },
+    {
+      "epoch": 281.18,
+      "eval_accuracy_NAT": 0.9939328784582574,
+      "eval_accuracy_REST": 0.408130081300813,
+      "eval_accuracy_concrete": 0.958164464943916,
+      "eval_accuracy_concrete tiles": 0.9366014612302616,
+      "eval_accuracy_dark bricks": 0.8500803935455468,
+      "eval_accuracy_grass": 0.9014371307346134,
+      "eval_accuracy_red bricks": 0.9304152021089631,
+      "eval_accuracy_road": 0.9805301514909408,
+      "eval_accuracy_rough concrete": 0.626174628617737,
+      "eval_accuracy_rough red sidewalk": 0.7810481354198783,
+      "eval_accuracy_speedway bricks": 0.990468155898423,
+      "eval_accuracy_steel": 0.9216890652566134,
+      "eval_accuracy_tiles": 0.2483565088309249,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842949348691709,
+      "eval_iou_REST": 0.39280125195618154,
+      "eval_iou_concrete": 0.8940289208821408,
+      "eval_iou_concrete tiles": 0.8790285120219425,
+      "eval_iou_dark bricks": 0.8351142597911779,
+      "eval_iou_grass": 0.8461986683436725,
+      "eval_iou_red bricks": 0.9037610029341158,
+      "eval_iou_road": 0.9610281519799411,
+      "eval_iou_rough concrete": 0.6204294663258016,
+      "eval_iou_rough red sidewalk": 0.627677190098627,
+      "eval_iou_speedway bricks": 0.980000915592541,
+      "eval_iou_steel": 0.8739568277136697,
+      "eval_iou_tiles": 0.24634893923913698,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08420391380786896,
+      "eval_mean_accuracy": 0.8097714044489913,
+      "eval_mean_iou": 0.7726668493652399,
+      "eval_overall_accuracy": 0.9805428779731362,
+      "eval_runtime": 5.557,
+      "eval_samples_per_second": 21.234,
+      "eval_steps_per_second": 1.44,
+      "step": 4780
+    },
+    {
+      "epoch": 281.24,
+      "learning_rate": 3.2920536635706914e-06,
+      "loss": 0.0167,
+      "step": 4781
+    },
+    {
+      "epoch": 281.29,
+      "learning_rate": 3.2817337461300315e-06,
+      "loss": 0.0167,
+      "step": 4782
+    },
+    {
+      "epoch": 281.35,
+      "learning_rate": 3.271413828689371e-06,
+      "loss": 0.0198,
+      "step": 4783
+    },
+    {
+      "epoch": 281.41,
+      "learning_rate": 3.2610939112487105e-06,
+      "loss": 0.0141,
+      "step": 4784
+    },
+    {
+      "epoch": 281.47,
+      "learning_rate": 3.25077399380805e-06,
+      "loss": 0.0184,
+      "step": 4785
+    },
+    {
+      "epoch": 281.53,
+      "learning_rate": 3.240454076367389e-06,
+      "loss": 0.0185,
+      "step": 4786
+    },
+    {
+      "epoch": 281.59,
+      "learning_rate": 3.230134158926729e-06,
+      "loss": 0.0186,
+      "step": 4787
+    },
+    {
+      "epoch": 281.65,
+      "learning_rate": 3.219814241486068e-06,
+      "loss": 0.0166,
+      "step": 4788
+    },
+    {
+      "epoch": 281.71,
+      "learning_rate": 3.2094943240454074e-06,
+      "loss": 0.0167,
+      "step": 4789
+    },
+    {
+      "epoch": 281.76,
+      "learning_rate": 3.199174406604747e-06,
+      "loss": 0.0184,
+      "step": 4790
+    },
+    {
+      "epoch": 281.82,
+      "learning_rate": 3.1888544891640873e-06,
+      "loss": 0.0192,
+      "step": 4791
+    },
+    {
+      "epoch": 281.88,
+      "learning_rate": 3.1785345717234266e-06,
+      "loss": 0.0178,
+      "step": 4792
+    },
+    {
+      "epoch": 281.94,
+      "learning_rate": 3.168214654282766e-06,
+      "loss": 0.0193,
+      "step": 4793
+    },
+    {
+      "epoch": 282.0,
+      "learning_rate": 3.1578947368421056e-06,
+      "loss": 0.0205,
+      "step": 4794
+    },
+    {
+      "epoch": 282.06,
+      "learning_rate": 3.147574819401445e-06,
+      "loss": 0.0204,
+      "step": 4795
+    },
+    {
+      "epoch": 282.12,
+      "learning_rate": 3.1372549019607846e-06,
+      "loss": 0.0197,
+      "step": 4796
+    },
+    {
+      "epoch": 282.18,
+      "learning_rate": 3.126934984520124e-06,
+      "loss": 0.0174,
+      "step": 4797
+    },
+    {
+      "epoch": 282.24,
+      "learning_rate": 3.1166150670794636e-06,
+      "loss": 0.0171,
+      "step": 4798
+    },
+    {
+      "epoch": 282.29,
+      "learning_rate": 3.106295149638803e-06,
+      "loss": 0.0186,
+      "step": 4799
+    },
+    {
+      "epoch": 282.35,
+      "learning_rate": 3.0959752321981426e-06,
+      "loss": 0.0166,
+      "step": 4800
+    },
+    {
+      "epoch": 282.35,
+      "eval_accuracy_NAT": 0.9942094996940238,
+      "eval_accuracy_REST": 0.42276422764227645,
+      "eval_accuracy_concrete": 0.9570656419648713,
+      "eval_accuracy_concrete tiles": 0.9355880273391468,
+      "eval_accuracy_dark bricks": 0.8498124150603908,
+      "eval_accuracy_grass": 0.8965735200051475,
+      "eval_accuracy_red bricks": 0.9316783831282952,
+      "eval_accuracy_road": 0.9815038576828464,
+      "eval_accuracy_rough concrete": 0.6259184001811475,
+      "eval_accuracy_rough red sidewalk": 0.7806270162958061,
+      "eval_accuracy_speedway bricks": 0.9899633141472904,
+      "eval_accuracy_steel": 0.9290208759189752,
+      "eval_accuracy_tiles": 0.2539610409282317,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842022633732447,
+      "eval_iou_REST": 0.40752351097178685,
+      "eval_iou_concrete": 0.8946441771338315,
+      "eval_iou_concrete tiles": 0.8780773739742087,
+      "eval_iou_dark bricks": 0.835648474924005,
+      "eval_iou_grass": 0.844449633308706,
+      "eval_iou_red bricks": 0.9056644065986866,
+      "eval_iou_road": 0.9608863050271771,
+      "eval_iou_rough concrete": 0.6202598169471508,
+      "eval_iou_rough red sidewalk": 0.6242415495541003,
+      "eval_iou_speedway bricks": 0.980016855768164,
+      "eval_iou_steel": 0.874068083125457,
+      "eval_iou_tiles": 0.2517871087883703,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08464650064706802,
+      "eval_mean_accuracy": 0.8114374015375729,
+      "eval_mean_iou": 0.7739591968842223,
+      "eval_overall_accuracy": 0.9805087719933462,
+      "eval_runtime": 5.522,
+      "eval_samples_per_second": 21.369,
+      "eval_steps_per_second": 1.449,
+      "step": 4800
+    },
+    {
+      "epoch": 282.41,
+      "learning_rate": 3.085655314757482e-06,
+      "loss": 0.0172,
+      "step": 4801
+    },
+    {
+      "epoch": 282.47,
+      "learning_rate": 3.0753353973168216e-06,
+      "loss": 0.0179,
+      "step": 4802
+    },
+    {
+      "epoch": 282.53,
+      "learning_rate": 3.0650154798761614e-06,
+      "loss": 0.0191,
+      "step": 4803
+    },
+    {
+      "epoch": 282.59,
+      "learning_rate": 3.0546955624355007e-06,
+      "loss": 0.0158,
+      "step": 4804
+    },
+    {
+      "epoch": 282.65,
+      "learning_rate": 3.0443756449948404e-06,
+      "loss": 0.0191,
+      "step": 4805
+    },
+    {
+      "epoch": 282.71,
+      "learning_rate": 3.0340557275541797e-06,
+      "loss": 0.0175,
+      "step": 4806
+    },
+    {
+      "epoch": 282.76,
+      "learning_rate": 3.023735810113519e-06,
+      "loss": 0.0199,
+      "step": 4807
+    },
+    {
+      "epoch": 282.82,
+      "learning_rate": 3.0134158926728587e-06,
+      "loss": 0.0176,
+      "step": 4808
+    },
+    {
+      "epoch": 282.88,
+      "learning_rate": 3.0030959752321984e-06,
+      "loss": 0.0214,
+      "step": 4809
+    },
+    {
+      "epoch": 282.94,
+      "learning_rate": 2.9927760577915377e-06,
+      "loss": 0.0172,
+      "step": 4810
+    },
+    {
+      "epoch": 283.0,
+      "learning_rate": 2.9824561403508774e-06,
+      "loss": 0.0206,
+      "step": 4811
+    },
+    {
+      "epoch": 283.06,
+      "learning_rate": 2.9721362229102167e-06,
+      "loss": 0.0168,
+      "step": 4812
+    },
+    {
+      "epoch": 283.12,
+      "learning_rate": 2.9618163054695564e-06,
+      "loss": 0.0223,
+      "step": 4813
+    },
+    {
+      "epoch": 283.18,
+      "learning_rate": 2.9514963880288957e-06,
+      "loss": 0.0217,
+      "step": 4814
+    },
+    {
+      "epoch": 283.24,
+      "learning_rate": 2.9411764705882355e-06,
+      "loss": 0.0158,
+      "step": 4815
+    },
+    {
+      "epoch": 283.29,
+      "learning_rate": 2.9308565531475748e-06,
+      "loss": 0.0188,
+      "step": 4816
+    },
+    {
+      "epoch": 283.35,
+      "learning_rate": 2.9205366357069145e-06,
+      "loss": 0.0196,
+      "step": 4817
+    },
+    {
+      "epoch": 283.41,
+      "learning_rate": 2.910216718266254e-06,
+      "loss": 0.0181,
+      "step": 4818
+    },
+    {
+      "epoch": 283.47,
+      "learning_rate": 2.8998968008255935e-06,
+      "loss": 0.0206,
+      "step": 4819
+    },
+    {
+      "epoch": 283.53,
+      "learning_rate": 2.889576883384933e-06,
+      "loss": 0.0203,
+      "step": 4820
+    },
+    {
+      "epoch": 283.53,
+      "eval_accuracy_NAT": 0.994091594676947,
+      "eval_accuracy_REST": 0.424390243902439,
+      "eval_accuracy_concrete": 0.9570801655381596,
+      "eval_accuracy_concrete tiles": 0.942139995286354,
+      "eval_accuracy_dark bricks": 0.8524634879313975,
+      "eval_accuracy_grass": 0.8983628367151514,
+      "eval_accuracy_red bricks": 0.9326669595782073,
+      "eval_accuracy_road": 0.9812997107205479,
+      "eval_accuracy_rough concrete": 0.6270207783385672,
+      "eval_accuracy_rough red sidewalk": 0.791282834131705,
+      "eval_accuracy_speedway bricks": 0.9900704173701206,
+      "eval_accuracy_steel": 0.9257048184039942,
+      "eval_accuracy_tiles": 0.2532036717258929,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842212223979613,
+      "eval_iou_REST": 0.4078125,
+      "eval_iou_concrete": 0.894757490944835,
+      "eval_iou_concrete tiles": 0.8788225207202057,
+      "eval_iou_dark bricks": 0.836967501562199,
+      "eval_iou_grass": 0.844454386486097,
+      "eval_iou_red bricks": 0.9064801964342906,
+      "eval_iou_road": 0.961303056082717,
+      "eval_iou_rough concrete": 0.6203558500666188,
+      "eval_iou_rough red sidewalk": 0.62730416120186,
+      "eval_iou_speedway bricks": 0.9800894634854186,
+      "eval_iou_steel": 0.8745030736253671,
+      "eval_iou_tiles": 0.25102868298543324,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08432457596063614,
+      "eval_mean_accuracy": 0.8130598087938066,
+      "eval_mean_iou": 0.7744692389225389,
+      "eval_overall_accuracy": 0.9805438801393671,
+      "eval_runtime": 5.3544,
+      "eval_samples_per_second": 22.038,
+      "eval_steps_per_second": 1.494,
+      "step": 4820
+    },
+    {
+      "epoch": 283.59,
+      "learning_rate": 2.8792569659442725e-06,
+      "loss": 0.0202,
+      "step": 4821
+    },
+    {
+      "epoch": 283.65,
+      "learning_rate": 2.8689370485036122e-06,
+      "loss": 0.0187,
+      "step": 4822
+    },
+    {
+      "epoch": 283.71,
+      "learning_rate": 2.8586171310629515e-06,
+      "loss": 0.0168,
+      "step": 4823
+    },
+    {
+      "epoch": 283.76,
+      "learning_rate": 2.8482972136222912e-06,
+      "loss": 0.0177,
+      "step": 4824
+    },
+    {
+      "epoch": 283.82,
+      "learning_rate": 2.8379772961816305e-06,
+      "loss": 0.0177,
+      "step": 4825
+    },
+    {
+      "epoch": 283.88,
+      "learning_rate": 2.8276573787409703e-06,
+      "loss": 0.0149,
+      "step": 4826
+    },
+    {
+      "epoch": 283.94,
+      "learning_rate": 2.81733746130031e-06,
+      "loss": 0.0188,
+      "step": 4827
+    },
+    {
+      "epoch": 284.0,
+      "learning_rate": 2.8070175438596493e-06,
+      "loss": 0.0155,
+      "step": 4828
+    },
+    {
+      "epoch": 284.06,
+      "learning_rate": 2.7966976264189886e-06,
+      "loss": 0.0191,
+      "step": 4829
+    },
+    {
+      "epoch": 284.12,
+      "learning_rate": 2.7863777089783283e-06,
+      "loss": 0.0199,
+      "step": 4830
+    },
+    {
+      "epoch": 284.18,
+      "learning_rate": 2.7760577915376676e-06,
+      "loss": 0.0168,
+      "step": 4831
+    },
+    {
+      "epoch": 284.24,
+      "learning_rate": 2.7657378740970073e-06,
+      "loss": 0.0203,
+      "step": 4832
+    },
+    {
+      "epoch": 284.29,
+      "learning_rate": 2.755417956656347e-06,
+      "loss": 0.0187,
+      "step": 4833
+    },
+    {
+      "epoch": 284.35,
+      "learning_rate": 2.7450980392156863e-06,
+      "loss": 0.0175,
+      "step": 4834
+    },
+    {
+      "epoch": 284.41,
+      "learning_rate": 2.7347781217750256e-06,
+      "loss": 0.0189,
+      "step": 4835
+    },
+    {
+      "epoch": 284.47,
+      "learning_rate": 2.7244582043343658e-06,
+      "loss": 0.0198,
+      "step": 4836
+    },
+    {
+      "epoch": 284.53,
+      "learning_rate": 2.714138286893705e-06,
+      "loss": 0.0157,
+      "step": 4837
+    },
+    {
+      "epoch": 284.59,
+      "learning_rate": 2.7038183694530443e-06,
+      "loss": 0.0178,
+      "step": 4838
+    },
+    {
+      "epoch": 284.65,
+      "learning_rate": 2.693498452012384e-06,
+      "loss": 0.0173,
+      "step": 4839
+    },
+    {
+      "epoch": 284.71,
+      "learning_rate": 2.6831785345717234e-06,
+      "loss": 0.0153,
+      "step": 4840
+    },
+    {
+      "epoch": 284.71,
+      "eval_accuracy_NAT": 0.9938774032581787,
+      "eval_accuracy_REST": 0.4,
+      "eval_accuracy_concrete": 0.958636863275081,
+      "eval_accuracy_concrete tiles": 0.931133631864247,
+      "eval_accuracy_dark bricks": 0.8476877284995119,
+      "eval_accuracy_grass": 0.900676363608837,
+      "eval_accuracy_red bricks": 0.9293167838312829,
+      "eval_accuracy_road": 0.9804772442939054,
+      "eval_accuracy_rough concrete": 0.6264487334568791,
+      "eval_accuracy_rough red sidewalk": 0.7916287534121929,
+      "eval_accuracy_speedway bricks": 0.9900104395653357,
+      "eval_accuracy_steel": 0.9248651836451144,
+      "eval_accuracy_tiles": 0.24765972916477325,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842483527355015,
+      "eval_iou_REST": 0.38618524332810045,
+      "eval_iou_concrete": 0.8941603926843328,
+      "eval_iou_concrete tiles": 0.8751744456504884,
+      "eval_iou_dark bricks": 0.8333043245099468,
+      "eval_iou_grass": 0.8450626699029988,
+      "eval_iou_red bricks": 0.9043343487787933,
+      "eval_iou_road": 0.961446653170803,
+      "eval_iou_rough concrete": 0.6202103747927814,
+      "eval_iou_rough red sidewalk": 0.6282563156858181,
+      "eval_iou_speedway bricks": 0.9799283559021231,
+      "eval_iou_steel": 0.874246475679353,
+      "eval_iou_tiles": 0.2458203031031994,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08417260646820068,
+      "eval_mean_accuracy": 0.80941683522118,
+      "eval_mean_iou": 0.7717214043018648,
+      "eval_overall_accuracy": 0.9804968753103482,
+      "eval_runtime": 5.4847,
+      "eval_samples_per_second": 21.514,
+      "eval_steps_per_second": 1.459,
+      "step": 4840
+    },
+    {
+      "epoch": 284.76,
+      "learning_rate": 2.672858617131063e-06,
+      "loss": 0.0173,
+      "step": 4841
+    },
+    {
+      "epoch": 284.82,
+      "learning_rate": 2.662538699690403e-06,
+      "loss": 0.0178,
+      "step": 4842
+    },
+    {
+      "epoch": 284.88,
+      "learning_rate": 2.652218782249742e-06,
+      "loss": 0.0195,
+      "step": 4843
+    },
+    {
+      "epoch": 284.94,
+      "learning_rate": 2.6418988648090814e-06,
+      "loss": 0.0201,
+      "step": 4844
+    },
+    {
+      "epoch": 285.0,
+      "learning_rate": 2.631578947368421e-06,
+      "loss": 0.0169,
+      "step": 4845
+    },
+    {
+      "epoch": 285.06,
+      "learning_rate": 2.621259029927761e-06,
+      "loss": 0.017,
+      "step": 4846
+    },
+    {
+      "epoch": 285.12,
+      "learning_rate": 2.6109391124871e-06,
+      "loss": 0.0194,
+      "step": 4847
+    },
+    {
+      "epoch": 285.18,
+      "learning_rate": 2.60061919504644e-06,
+      "loss": 0.0177,
+      "step": 4848
+    },
+    {
+      "epoch": 285.24,
+      "learning_rate": 2.590299277605779e-06,
+      "loss": 0.0148,
+      "step": 4849
+    },
+    {
+      "epoch": 285.29,
+      "learning_rate": 2.5799793601651184e-06,
+      "loss": 0.0183,
+      "step": 4850
+    },
+    {
+      "epoch": 285.35,
+      "learning_rate": 2.5696594427244586e-06,
+      "loss": 0.0212,
+      "step": 4851
+    },
+    {
+      "epoch": 285.41,
+      "learning_rate": 2.559339525283798e-06,
+      "loss": 0.0198,
+      "step": 4852
+    },
+    {
+      "epoch": 285.47,
+      "learning_rate": 2.549019607843137e-06,
+      "loss": 0.0165,
+      "step": 4853
+    },
+    {
+      "epoch": 285.53,
+      "learning_rate": 2.538699690402477e-06,
+      "loss": 0.0158,
+      "step": 4854
+    },
+    {
+      "epoch": 285.59,
+      "learning_rate": 2.5283797729618166e-06,
+      "loss": 0.0195,
+      "step": 4855
+    },
+    {
+      "epoch": 285.65,
+      "learning_rate": 2.518059855521156e-06,
+      "loss": 0.0193,
+      "step": 4856
+    },
+    {
+      "epoch": 285.71,
+      "learning_rate": 2.5077399380804956e-06,
+      "loss": 0.0202,
+      "step": 4857
+    },
+    {
+      "epoch": 285.76,
+      "learning_rate": 2.497420020639835e-06,
+      "loss": 0.0185,
+      "step": 4858
+    },
+    {
+      "epoch": 285.82,
+      "learning_rate": 2.4871001031991742e-06,
+      "loss": 0.0157,
+      "step": 4859
+    },
+    {
+      "epoch": 285.88,
+      "learning_rate": 2.4767801857585144e-06,
+      "loss": 0.017,
+      "step": 4860
+    },
+    {
+      "epoch": 285.88,
+      "eval_accuracy_NAT": 0.9938280739986336,
+      "eval_accuracy_REST": 0.40487804878048783,
+      "eval_accuracy_concrete": 0.9582649833590426,
+      "eval_accuracy_concrete tiles": 0.9443789771388169,
+      "eval_accuracy_dark bricks": 0.8532195700859445,
+      "eval_accuracy_grass": 0.9027116995386193,
+      "eval_accuracy_red bricks": 0.9307996485061512,
+      "eval_accuracy_road": 0.9802052264828846,
+      "eval_accuracy_rough concrete": 0.6259660705879548,
+      "eval_accuracy_rough red sidewalk": 0.7857255656908234,
+      "eval_accuracy_speedway bricks": 0.9899019654212533,
+      "eval_accuracy_steel": 0.9276064911763383,
+      "eval_accuracy_tiles": 0.2542033990729801,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842912024689578,
+      "eval_iou_REST": 0.3921259842519685,
+      "eval_iou_concrete": 0.8940512923283305,
+      "eval_iou_concrete tiles": 0.8784968867841796,
+      "eval_iou_dark bricks": 0.8372456670063252,
+      "eval_iou_grass": 0.8459884294237413,
+      "eval_iou_red bricks": 0.9050034709243339,
+      "eval_iou_road": 0.9611174603573849,
+      "eval_iou_rough concrete": 0.6199446441112075,
+      "eval_iou_rough red sidewalk": 0.6282620198667532,
+      "eval_iou_speedway bricks": 0.9800048350772128,
+      "eval_iou_steel": 0.8743551935479312,
+      "eval_iou_tiles": 0.2516268329984706,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08425776660442352,
+      "eval_mean_accuracy": 0.8116684399876869,
+      "eval_mean_iou": 0.7732703014728306,
+      "eval_overall_accuracy": 0.9805394188832428,
+      "eval_runtime": 5.4173,
+      "eval_samples_per_second": 21.782,
+      "eval_steps_per_second": 1.477,
+      "step": 4860
+    },
+    {
+      "epoch": 285.94,
+      "learning_rate": 2.4664602683178537e-06,
+      "loss": 0.0192,
+      "step": 4861
+    },
+    {
+      "epoch": 286.0,
+      "learning_rate": 2.456140350877193e-06,
+      "loss": 0.0203,
+      "step": 4862
+    },
+    {
+      "epoch": 286.06,
+      "learning_rate": 2.4458204334365327e-06,
+      "loss": 0.0187,
+      "step": 4863
+    },
+    {
+      "epoch": 286.12,
+      "learning_rate": 2.435500515995872e-06,
+      "loss": 0.016,
+      "step": 4864
+    },
+    {
+      "epoch": 286.18,
+      "learning_rate": 2.4251805985552117e-06,
+      "loss": 0.0201,
+      "step": 4865
+    },
+    {
+      "epoch": 286.24,
+      "learning_rate": 2.4148606811145514e-06,
+      "loss": 0.0175,
+      "step": 4866
+    },
+    {
+      "epoch": 286.29,
+      "learning_rate": 2.4045407636738907e-06,
+      "loss": 0.019,
+      "step": 4867
+    },
+    {
+      "epoch": 286.35,
+      "learning_rate": 2.39422084623323e-06,
+      "loss": 0.0171,
+      "step": 4868
+    },
+    {
+      "epoch": 286.41,
+      "learning_rate": 2.3839009287925697e-06,
+      "loss": 0.0149,
+      "step": 4869
+    },
+    {
+      "epoch": 286.47,
+      "learning_rate": 2.3735810113519094e-06,
+      "loss": 0.019,
+      "step": 4870
+    },
+    {
+      "epoch": 286.53,
+      "learning_rate": 2.3632610939112487e-06,
+      "loss": 0.0192,
+      "step": 4871
+    },
+    {
+      "epoch": 286.59,
+      "learning_rate": 2.3529411764705885e-06,
+      "loss": 0.0187,
+      "step": 4872
+    },
+    {
+      "epoch": 286.65,
+      "learning_rate": 2.3426212590299277e-06,
+      "loss": 0.0201,
+      "step": 4873
+    },
+    {
+      "epoch": 286.71,
+      "learning_rate": 2.3323013415892675e-06,
+      "loss": 0.016,
+      "step": 4874
+    },
+    {
+      "epoch": 286.76,
+      "learning_rate": 2.321981424148607e-06,
+      "loss": 0.0183,
+      "step": 4875
+    },
+    {
+      "epoch": 286.82,
+      "learning_rate": 2.3116615067079465e-06,
+      "loss": 0.0181,
+      "step": 4876
+    },
+    {
+      "epoch": 286.88,
+      "learning_rate": 2.3013415892672858e-06,
+      "loss": 0.0174,
+      "step": 4877
+    },
+    {
+      "epoch": 286.94,
+      "learning_rate": 2.2910216718266255e-06,
+      "loss": 0.0187,
+      "step": 4878
+    },
+    {
+      "epoch": 287.0,
+      "learning_rate": 2.2807017543859652e-06,
+      "loss": 0.0188,
+      "step": 4879
+    },
+    {
+      "epoch": 287.06,
+      "learning_rate": 2.2703818369453045e-06,
+      "loss": 0.0192,
+      "step": 4880
+    },
+    {
+      "epoch": 287.06,
+      "eval_accuracy_NAT": 0.9941324058785598,
+      "eval_accuracy_REST": 0.4065040650406504,
+      "eval_accuracy_concrete": 0.9575208413276687,
+      "eval_accuracy_concrete tiles": 0.9453217063398539,
+      "eval_accuracy_dark bricks": 0.8525831211836993,
+      "eval_accuracy_grass": 0.8979455502692965,
+      "eval_accuracy_red bricks": 0.9343145869947276,
+      "eval_accuracy_road": 0.9812623015913309,
+      "eval_accuracy_rough concrete": 0.6260375761981659,
+      "eval_accuracy_rough red sidewalk": 0.7883049203257657,
+      "eval_accuracy_speedway bricks": 0.9895926513137195,
+      "eval_accuracy_steel": 0.9279238531238911,
+      "eval_accuracy_tiles": 0.25350661940682845,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984182987663816,
+      "eval_iou_REST": 0.3924646781789639,
+      "eval_iou_concrete": 0.8944938094318288,
+      "eval_iou_concrete tiles": 0.8776229131566855,
+      "eval_iou_dark bricks": 0.8372462406015038,
+      "eval_iou_grass": 0.844629336409328,
+      "eval_iou_red bricks": 0.9069197142552511,
+      "eval_iou_road": 0.9611617450988912,
+      "eval_iou_rough concrete": 0.6201874830285357,
+      "eval_iou_rough red sidewalk": 0.6266094420600858,
+      "eval_iou_speedway bricks": 0.9800840344549072,
+      "eval_iou_steel": 0.8743680610334719,
+      "eval_iou_tiles": 0.2512535655307011,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08481348305940628,
+      "eval_mean_accuracy": 0.8119192460764736,
+      "eval_mean_iou": 0.7731710777618438,
+      "eval_overall_accuracy": 0.98051565784519,
+      "eval_runtime": 5.3391,
+      "eval_samples_per_second": 22.101,
+      "eval_steps_per_second": 1.498,
+      "step": 4880
+    },
+    {
+      "epoch": 287.12,
+      "learning_rate": 2.2600619195046442e-06,
+      "loss": 0.0162,
+      "step": 4881
+    },
+    {
+      "epoch": 287.18,
+      "learning_rate": 2.2497420020639835e-06,
+      "loss": 0.017,
+      "step": 4882
+    },
+    {
+      "epoch": 287.24,
+      "learning_rate": 2.239422084623323e-06,
+      "loss": 0.0187,
+      "step": 4883
+    },
+    {
+      "epoch": 287.29,
+      "learning_rate": 2.229102167182663e-06,
+      "loss": 0.0182,
+      "step": 4884
+    },
+    {
+      "epoch": 287.35,
+      "learning_rate": 2.2187822497420023e-06,
+      "loss": 0.0206,
+      "step": 4885
+    },
+    {
+      "epoch": 287.41,
+      "learning_rate": 2.2084623323013416e-06,
+      "loss": 0.0176,
+      "step": 4886
+    },
+    {
+      "epoch": 287.47,
+      "learning_rate": 2.1981424148606813e-06,
+      "loss": 0.0201,
+      "step": 4887
+    },
+    {
+      "epoch": 287.53,
+      "learning_rate": 2.1878224974200206e-06,
+      "loss": 0.019,
+      "step": 4888
+    },
+    {
+      "epoch": 287.59,
+      "learning_rate": 2.1775025799793603e-06,
+      "loss": 0.0207,
+      "step": 4889
+    },
+    {
+      "epoch": 287.65,
+      "learning_rate": 2.1671826625387e-06,
+      "loss": 0.0166,
+      "step": 4890
+    },
+    {
+      "epoch": 287.71,
+      "learning_rate": 2.1568627450980393e-06,
+      "loss": 0.0171,
+      "step": 4891
+    },
+    {
+      "epoch": 287.76,
+      "learning_rate": 2.1465428276573786e-06,
+      "loss": 0.0199,
+      "step": 4892
+    },
+    {
+      "epoch": 287.82,
+      "learning_rate": 2.1362229102167187e-06,
+      "loss": 0.016,
+      "step": 4893
+    },
+    {
+      "epoch": 287.88,
+      "learning_rate": 2.125902992776058e-06,
+      "loss": 0.0192,
+      "step": 4894
+    },
+    {
+      "epoch": 287.94,
+      "learning_rate": 2.1155830753353973e-06,
+      "loss": 0.0168,
+      "step": 4895
+    },
+    {
+      "epoch": 288.0,
+      "learning_rate": 2.105263157894737e-06,
+      "loss": 0.0175,
+      "step": 4896
+    },
+    {
+      "epoch": 288.06,
+      "learning_rate": 2.0949432404540764e-06,
+      "loss": 0.0173,
+      "step": 4897
+    },
+    {
+      "epoch": 288.12,
+      "learning_rate": 2.084623323013416e-06,
+      "loss": 0.0165,
+      "step": 4898
+    },
+    {
+      "epoch": 288.18,
+      "learning_rate": 2.074303405572756e-06,
+      "loss": 0.0166,
+      "step": 4899
+    },
+    {
+      "epoch": 288.24,
+      "learning_rate": 2.063983488132095e-06,
+      "loss": 0.0173,
+      "step": 4900
+    },
+    {
+      "epoch": 288.24,
+      "eval_accuracy_NAT": 0.9940785480312531,
+      "eval_accuracy_REST": 0.4097560975609756,
+      "eval_accuracy_concrete": 0.9585630988107486,
+      "eval_accuracy_concrete tiles": 0.935540890879095,
+      "eval_accuracy_dark bricks": 0.8500756082154547,
+      "eval_accuracy_grass": 0.8982956545435965,
+      "eval_accuracy_red bricks": 0.929371704745167,
+      "eval_accuracy_road": 0.9804521267357168,
+      "eval_accuracy_rough concrete": 0.6254953253207325,
+      "eval_accuracy_rough red sidewalk": 0.78358237014867,
+      "eval_accuracy_speedway bricks": 0.9898699201369825,
+      "eval_accuracy_steel": 0.9256073607980528,
+      "eval_accuracy_tiles": 0.24796267684570875,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842174462176391,
+      "eval_iou_REST": 0.39622641509433965,
+      "eval_iou_concrete": 0.8939787706601812,
+      "eval_iou_concrete tiles": 0.8787912331193269,
+      "eval_iou_dark bricks": 0.8350388981596822,
+      "eval_iou_grass": 0.8451813187351268,
+      "eval_iou_red bricks": 0.9047746350852804,
+      "eval_iou_road": 0.9611434876089171,
+      "eval_iou_rough concrete": 0.6197344416958419,
+      "eval_iou_rough red sidewalk": 0.62562292109466,
+      "eval_iou_speedway bricks": 0.9800264807000352,
+      "eval_iou_steel": 0.8743655960682299,
+      "eval_iou_tiles": 0.24591395264992189,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08537660539150238,
+      "eval_mean_accuracy": 0.8098962602132427,
+      "eval_mean_iou": 0.7726935074530139,
+      "eval_overall_accuracy": 0.9804944507146286,
+      "eval_runtime": 5.4595,
+      "eval_samples_per_second": 21.614,
+      "eval_steps_per_second": 1.465,
+      "step": 4900
+    },
+    {
+      "epoch": 288.29,
+      "learning_rate": 2.0536635706914344e-06,
+      "loss": 0.0159,
+      "step": 4901
+    },
+    {
+      "epoch": 288.35,
+      "learning_rate": 2.043343653250774e-06,
+      "loss": 0.0196,
+      "step": 4902
+    },
+    {
+      "epoch": 288.41,
+      "learning_rate": 2.033023735810114e-06,
+      "loss": 0.0198,
+      "step": 4903
+    },
+    {
+      "epoch": 288.47,
+      "learning_rate": 2.022703818369453e-06,
+      "loss": 0.0204,
+      "step": 4904
+    },
+    {
+      "epoch": 288.53,
+      "learning_rate": 2.012383900928793e-06,
+      "loss": 0.0194,
+      "step": 4905
+    },
+    {
+      "epoch": 288.59,
+      "learning_rate": 2.002063983488132e-06,
+      "loss": 0.0179,
+      "step": 4906
+    },
+    {
+      "epoch": 288.65,
+      "learning_rate": 1.9917440660474714e-06,
+      "loss": 0.017,
+      "step": 4907
+    },
+    {
+      "epoch": 288.71,
+      "learning_rate": 1.981424148606811e-06,
+      "loss": 0.0174,
+      "step": 4908
+    },
+    {
+      "epoch": 288.76,
+      "learning_rate": 1.971104231166151e-06,
+      "loss": 0.0172,
+      "step": 4909
+    },
+    {
+      "epoch": 288.82,
+      "learning_rate": 1.96078431372549e-06,
+      "loss": 0.0178,
+      "step": 4910
+    },
+    {
+      "epoch": 288.88,
+      "learning_rate": 1.95046439628483e-06,
+      "loss": 0.0197,
+      "step": 4911
+    },
+    {
+      "epoch": 288.94,
+      "learning_rate": 1.940144478844169e-06,
+      "loss": 0.0195,
+      "step": 4912
+    },
+    {
+      "epoch": 289.0,
+      "learning_rate": 1.929824561403509e-06,
+      "loss": 0.0183,
+      "step": 4913
+    },
+    {
+      "epoch": 289.06,
+      "learning_rate": 1.9195046439628486e-06,
+      "loss": 0.0194,
+      "step": 4914
+    },
+    {
+      "epoch": 289.12,
+      "learning_rate": 1.909184726522188e-06,
+      "loss": 0.0186,
+      "step": 4915
+    },
+    {
+      "epoch": 289.18,
+      "learning_rate": 1.8988648090815274e-06,
+      "loss": 0.0224,
+      "step": 4916
+    },
+    {
+      "epoch": 289.24,
+      "learning_rate": 1.8885448916408671e-06,
+      "loss": 0.0171,
+      "step": 4917
+    },
+    {
+      "epoch": 289.29,
+      "learning_rate": 1.8782249742002066e-06,
+      "loss": 0.0149,
+      "step": 4918
+    },
+    {
+      "epoch": 289.35,
+      "learning_rate": 1.867905056759546e-06,
+      "loss": 0.0203,
+      "step": 4919
+    },
+    {
+      "epoch": 289.41,
+      "learning_rate": 1.8575851393188855e-06,
+      "loss": 0.0162,
+      "step": 4920
+    },
+    {
+      "epoch": 289.41,
+      "eval_accuracy_NAT": 0.9939764391595826,
+      "eval_accuracy_REST": 0.43902439024390244,
+      "eval_accuracy_concrete": 0.9584495856195221,
+      "eval_accuracy_concrete tiles": 0.9386047607824652,
+      "eval_accuracy_dark bricks": 0.8495635778956032,
+      "eval_accuracy_grass": 0.899998864526678,
+      "eval_accuracy_red bricks": 0.9310742530755711,
+      "eval_accuracy_road": 0.9803404337641974,
+      "eval_accuracy_rough concrete": 0.625310602494354,
+      "eval_accuracy_rough red sidewalk": 0.7781153415200897,
+      "eval_accuracy_speedway bricks": 0.9899703401187081,
+      "eval_accuracy_steel": 0.9282936922438739,
+      "eval_accuracy_tiles": 0.24520585294919567,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842704948926261,
+      "eval_iou_REST": 0.4225352112676056,
+      "eval_iou_concrete": 0.8938155577503434,
+      "eval_iou_concrete tiles": 0.8789256471938381,
+      "eval_iou_dark bricks": 0.8340889550808319,
+      "eval_iou_grass": 0.8458365088159103,
+      "eval_iou_red bricks": 0.9054638679698767,
+      "eval_iou_road": 0.9610506093226018,
+      "eval_iou_rough concrete": 0.6197563237126675,
+      "eval_iou_rough red sidewalk": 0.6239853822681606,
+      "eval_iou_speedway bricks": 0.9800209848279864,
+      "eval_iou_steel": 0.8745559288356095,
+      "eval_iou_tiles": 0.24330417530886464,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08482333272695541,
+      "eval_mean_accuracy": 0.8121483180302882,
+      "eval_mean_iou": 0.7744315113266864,
+      "eval_overall_accuracy": 0.9805036318504204,
+      "eval_runtime": 5.3249,
+      "eval_samples_per_second": 22.16,
+      "eval_steps_per_second": 1.502,
+      "step": 4920
+    },
+    {
+      "epoch": 289.47,
+      "learning_rate": 1.847265221878225e-06,
+      "loss": 0.0167,
+      "step": 4921
+    },
+    {
+      "epoch": 289.53,
+      "learning_rate": 1.8369453044375647e-06,
+      "loss": 0.0189,
+      "step": 4922
+    },
+    {
+      "epoch": 289.59,
+      "learning_rate": 1.8266253869969042e-06,
+      "loss": 0.0168,
+      "step": 4923
+    },
+    {
+      "epoch": 289.65,
+      "learning_rate": 1.8163054695562437e-06,
+      "loss": 0.02,
+      "step": 4924
+    },
+    {
+      "epoch": 289.71,
+      "learning_rate": 1.805985552115583e-06,
+      "loss": 0.0199,
+      "step": 4925
+    },
+    {
+      "epoch": 289.76,
+      "learning_rate": 1.7956656346749225e-06,
+      "loss": 0.0152,
+      "step": 4926
+    },
+    {
+      "epoch": 289.82,
+      "learning_rate": 1.7853457172342624e-06,
+      "loss": 0.0189,
+      "step": 4927
+    },
+    {
+      "epoch": 289.88,
+      "learning_rate": 1.7750257997936017e-06,
+      "loss": 0.0181,
+      "step": 4928
+    },
+    {
+      "epoch": 289.94,
+      "learning_rate": 1.7647058823529412e-06,
+      "loss": 0.0202,
+      "step": 4929
+    },
+    {
+      "epoch": 290.0,
+      "learning_rate": 1.7543859649122807e-06,
+      "loss": 0.0188,
+      "step": 4930
+    },
+    {
+      "epoch": 290.06,
+      "learning_rate": 1.74406604747162e-06,
+      "loss": 0.0163,
+      "step": 4931
+    },
+    {
+      "epoch": 290.12,
+      "learning_rate": 1.73374613003096e-06,
+      "loss": 0.0172,
+      "step": 4932
+    },
+    {
+      "epoch": 290.18,
+      "learning_rate": 1.7234262125902995e-06,
+      "loss": 0.0165,
+      "step": 4933
+    },
+    {
+      "epoch": 290.24,
+      "learning_rate": 1.7131062951496388e-06,
+      "loss": 0.0159,
+      "step": 4934
+    },
+    {
+      "epoch": 290.29,
+      "learning_rate": 1.7027863777089783e-06,
+      "loss": 0.0183,
+      "step": 4935
+    },
+    {
+      "epoch": 290.35,
+      "learning_rate": 1.692466460268318e-06,
+      "loss": 0.0171,
+      "step": 4936
+    },
+    {
+      "epoch": 290.41,
+      "learning_rate": 1.6821465428276575e-06,
+      "loss": 0.0171,
+      "step": 4937
+    },
+    {
+      "epoch": 290.47,
+      "learning_rate": 1.671826625386997e-06,
+      "loss": 0.0181,
+      "step": 4938
+    },
+    {
+      "epoch": 290.53,
+      "learning_rate": 1.6615067079463365e-06,
+      "loss": 0.0171,
+      "step": 4939
+    },
+    {
+      "epoch": 290.59,
+      "learning_rate": 1.6511867905056758e-06,
+      "loss": 0.0181,
+      "step": 4940
+    },
+    {
+      "epoch": 290.59,
+      "eval_accuracy_NAT": 0.9940049584801283,
+      "eval_accuracy_REST": 0.424390243902439,
+      "eval_accuracy_concrete": 0.957756276094657,
+      "eval_accuracy_concrete tiles": 0.9366721659203394,
+      "eval_accuracy_dark bricks": 0.8517983270485998,
+      "eval_accuracy_grass": 0.8985889851517939,
+      "eval_accuracy_red bricks": 0.9304152021089631,
+      "eval_accuracy_road": 0.9817261747936219,
+      "eval_accuracy_rough concrete": 0.6252629320875467,
+      "eval_accuracy_rough red sidewalk": 0.7813188548567819,
+      "eval_accuracy_speedway bricks": 0.9899840493312303,
+      "eval_accuracy_steel": 0.9261546227391085,
+      "eval_accuracy_tiles": 0.24996213153988306,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842341595718223,
+      "eval_iou_REST": 0.4084507042253521,
+      "eval_iou_concrete": 0.8941403823393298,
+      "eval_iou_concrete tiles": 0.8774838823633313,
+      "eval_iou_dark bricks": 0.8356352164646456,
+      "eval_iou_grass": 0.8451540787756998,
+      "eval_iou_red bricks": 0.9053548525010688,
+      "eval_iou_road": 0.9608279721742768,
+      "eval_iou_rough concrete": 0.6196761391821983,
+      "eval_iou_rough red sidewalk": 0.6246144967266037,
+      "eval_iou_speedway bricks": 0.9800862645781844,
+      "eval_iou_steel": 0.8742964308461701,
+      "eval_iou_tiles": 0.2475993278117873,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08557610958814621,
+      "eval_mean_accuracy": 0.8113873018503915,
+      "eval_mean_iou": 0.773657992889267,
+      "eval_overall_accuracy": 0.9805030176195048,
+      "eval_runtime": 5.3956,
+      "eval_samples_per_second": 21.87,
+      "eval_steps_per_second": 1.483,
+      "step": 4940
+    },
+    {
+      "epoch": 290.65,
+      "learning_rate": 1.6408668730650157e-06,
+      "loss": 0.0206,
+      "step": 4941
+    },
+    {
+      "epoch": 290.71,
+      "learning_rate": 1.6305469556243553e-06,
+      "loss": 0.0182,
+      "step": 4942
+    },
+    {
+      "epoch": 290.76,
+      "learning_rate": 1.6202270381836946e-06,
+      "loss": 0.0223,
+      "step": 4943
+    },
+    {
+      "epoch": 290.82,
+      "learning_rate": 1.609907120743034e-06,
+      "loss": 0.019,
+      "step": 4944
+    },
+    {
+      "epoch": 290.88,
+      "learning_rate": 1.5995872033023736e-06,
+      "loss": 0.0165,
+      "step": 4945
+    },
+    {
+      "epoch": 290.94,
+      "learning_rate": 1.5892672858617133e-06,
+      "loss": 0.0204,
+      "step": 4946
+    },
+    {
+      "epoch": 291.0,
+      "learning_rate": 1.5789473684210528e-06,
+      "loss": 0.018,
+      "step": 4947
+    },
+    {
+      "epoch": 291.06,
+      "learning_rate": 1.5686274509803923e-06,
+      "loss": 0.015,
+      "step": 4948
+    },
+    {
+      "epoch": 291.12,
+      "learning_rate": 1.5583075335397318e-06,
+      "loss": 0.0167,
+      "step": 4949
+    },
+    {
+      "epoch": 291.18,
+      "learning_rate": 1.5479876160990713e-06,
+      "loss": 0.0194,
+      "step": 4950
+    },
+    {
+      "epoch": 291.24,
+      "learning_rate": 1.5376676986584108e-06,
+      "loss": 0.0194,
+      "step": 4951
+    },
+    {
+      "epoch": 291.29,
+      "learning_rate": 1.5273477812177503e-06,
+      "loss": 0.0193,
+      "step": 4952
+    },
+    {
+      "epoch": 291.35,
+      "learning_rate": 1.5170278637770898e-06,
+      "loss": 0.0174,
+      "step": 4953
+    },
+    {
+      "epoch": 291.41,
+      "learning_rate": 1.5067079463364293e-06,
+      "loss": 0.0199,
+      "step": 4954
+    },
+    {
+      "epoch": 291.47,
+      "learning_rate": 1.4963880288957689e-06,
+      "loss": 0.0172,
+      "step": 4955
+    },
+    {
+      "epoch": 291.53,
+      "learning_rate": 1.4860681114551084e-06,
+      "loss": 0.017,
+      "step": 4956
+    },
+    {
+      "epoch": 291.59,
+      "learning_rate": 1.4757481940144479e-06,
+      "loss": 0.0188,
+      "step": 4957
+    },
+    {
+      "epoch": 291.65,
+      "learning_rate": 1.4654282765737874e-06,
+      "loss": 0.0196,
+      "step": 4958
+    },
+    {
+      "epoch": 291.71,
+      "learning_rate": 1.455108359133127e-06,
+      "loss": 0.0182,
+      "step": 4959
+    },
+    {
+      "epoch": 291.76,
+      "learning_rate": 1.4447884416924666e-06,
+      "loss": 0.02,
+      "step": 4960
+    },
+    {
+      "epoch": 291.76,
+      "eval_accuracy_NAT": 0.9937969130018935,
+      "eval_accuracy_REST": 0.4292682926829268,
+      "eval_accuracy_concrete": 0.9577474855108247,
+      "eval_accuracy_concrete tiles": 0.9439076125382984,
+      "eval_accuracy_dark bricks": 0.8526740424554485,
+      "eval_accuracy_grass": 0.9013093899858823,
+      "eval_accuracy_red bricks": 0.9345891915641477,
+      "eval_accuracy_road": 0.9819730750464541,
+      "eval_accuracy_rough concrete": 0.6250484152569137,
+      "eval_accuracy_rough red sidewalk": 0.7867558035479286,
+      "eval_accuracy_speedway bricks": 0.9899297265766108,
+      "eval_accuracy_steel": 0.9269442792385313,
+      "eval_accuracy_tiles": 0.24990154200369596,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842706411892898,
+      "eval_iou_REST": 0.4105754276827372,
+      "eval_iou_concrete": 0.8941175001382627,
+      "eval_iou_concrete tiles": 0.8790798744485173,
+      "eval_iou_dark bricks": 0.8362940886583906,
+      "eval_iou_grass": 0.8456100635630837,
+      "eval_iou_red bricks": 0.9059788106266304,
+      "eval_iou_road": 0.9609303950418816,
+      "eval_iou_rough concrete": 0.6197307085590722,
+      "eval_iou_rough red sidewalk": 0.6259243305334194,
+      "eval_iou_speedway bricks": 0.9801307570133052,
+      "eval_iou_steel": 0.8744972582006441,
+      "eval_iou_tiles": 0.2476136158972204,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08544110506772995,
+      "eval_mean_accuracy": 0.813372751493043,
+      "eval_mean_iou": 0.774211805504035,
+      "eval_overall_accuracy": 0.9805259058030985,
+      "eval_runtime": 5.4086,
+      "eval_samples_per_second": 21.817,
+      "eval_steps_per_second": 1.479,
+      "step": 4960
+    },
+    {
+      "epoch": 291.82,
+      "learning_rate": 1.4344685242518061e-06,
+      "loss": 0.0194,
+      "step": 4961
+    },
+    {
+      "epoch": 291.88,
+      "learning_rate": 1.4241486068111456e-06,
+      "loss": 0.0186,
+      "step": 4962
+    },
+    {
+      "epoch": 291.94,
+      "learning_rate": 1.4138286893704851e-06,
+      "loss": 0.0168,
+      "step": 4963
+    },
+    {
+      "epoch": 292.0,
+      "learning_rate": 1.4035087719298246e-06,
+      "loss": 0.0177,
+      "step": 4964
+    },
+    {
+      "epoch": 292.06,
+      "learning_rate": 1.3931888544891641e-06,
+      "loss": 0.0194,
+      "step": 4965
+    },
+    {
+      "epoch": 292.12,
+      "learning_rate": 1.3828689370485037e-06,
+      "loss": 0.0142,
+      "step": 4966
+    },
+    {
+      "epoch": 292.18,
+      "learning_rate": 1.3725490196078432e-06,
+      "loss": 0.017,
+      "step": 4967
+    },
+    {
+      "epoch": 292.24,
+      "learning_rate": 1.3622291021671829e-06,
+      "loss": 0.0197,
+      "step": 4968
+    },
+    {
+      "epoch": 292.29,
+      "learning_rate": 1.3519091847265222e-06,
+      "loss": 0.0192,
+      "step": 4969
+    },
+    {
+      "epoch": 292.35,
+      "learning_rate": 1.3415892672858617e-06,
+      "loss": 0.0204,
+      "step": 4970
+    },
+    {
+      "epoch": 292.41,
+      "learning_rate": 1.3312693498452014e-06,
+      "loss": 0.0166,
+      "step": 4971
+    },
+    {
+      "epoch": 292.47,
+      "learning_rate": 1.3209494324045407e-06,
+      "loss": 0.0187,
+      "step": 4972
+    },
+    {
+      "epoch": 292.53,
+      "learning_rate": 1.3106295149638804e-06,
+      "loss": 0.0168,
+      "step": 4973
+    },
+    {
+      "epoch": 292.59,
+      "learning_rate": 1.30030959752322e-06,
+      "loss": 0.0184,
+      "step": 4974
+    },
+    {
+      "epoch": 292.65,
+      "learning_rate": 1.2899896800825592e-06,
+      "loss": 0.0165,
+      "step": 4975
+    },
+    {
+      "epoch": 292.71,
+      "learning_rate": 1.279669762641899e-06,
+      "loss": 0.0182,
+      "step": 4976
+    },
+    {
+      "epoch": 292.76,
+      "learning_rate": 1.2693498452012384e-06,
+      "loss": 0.0217,
+      "step": 4977
+    },
+    {
+      "epoch": 292.82,
+      "learning_rate": 1.259029927760578e-06,
+      "loss": 0.0198,
+      "step": 4978
+    },
+    {
+      "epoch": 292.88,
+      "learning_rate": 1.2487100103199175e-06,
+      "loss": 0.0181,
+      "step": 4979
+    },
+    {
+      "epoch": 292.94,
+      "learning_rate": 1.2383900928792572e-06,
+      "loss": 0.0202,
+      "step": 4980
+    },
+    {
+      "epoch": 292.94,
+      "eval_accuracy_NAT": 0.9939174596951645,
+      "eval_accuracy_REST": 0.424390243902439,
+      "eval_accuracy_concrete": 0.9585283186747163,
+      "eval_accuracy_concrete tiles": 0.9413386754654726,
+      "eval_accuracy_dark bricks": 0.8516356258254695,
+      "eval_accuracy_grass": 0.8999619616437112,
+      "eval_accuracy_red bricks": 0.9309094903339191,
+      "eval_accuracy_road": 0.9803580694965426,
+      "eval_accuracy_rough concrete": 0.6255906661343471,
+      "eval_accuracy_rough red sidewalk": 0.7881018807480881,
+      "eval_accuracy_speedway bricks": 0.9901549003922892,
+      "eval_accuracy_steel": 0.9256848271002114,
+      "eval_accuracy_tiles": 0.2458723378472538,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842719408736538,
+      "eval_iou_REST": 0.4078125,
+      "eval_iou_concrete": 0.8940467184905434,
+      "eval_iou_concrete tiles": 0.8796608303050325,
+      "eval_iou_dark bricks": 0.8353305077188091,
+      "eval_iou_grass": 0.8455980412030172,
+      "eval_iou_red bricks": 0.905351992308514,
+      "eval_iou_road": 0.961456256538301,
+      "eval_iou_rough concrete": 0.6198106078496198,
+      "eval_iou_rough red sidewalk": 0.6271978646749734,
+      "eval_iou_speedway bricks": 0.9800760003005681,
+      "eval_iou_steel": 0.8748208132892186,
+      "eval_iou_tiles": 0.24403884896412786,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08454770594835281,
+      "eval_mean_accuracy": 0.8120341890199712,
+      "eval_mean_iou": 0.773805609424337,
+      "eval_overall_accuracy": 0.9805383197331833,
+      "eval_runtime": 5.4237,
+      "eval_samples_per_second": 21.756,
+      "eval_steps_per_second": 1.475,
+      "step": 4980
+    },
+    {
+      "epoch": 293.0,
+      "learning_rate": 1.2280701754385965e-06,
+      "loss": 0.0184,
+      "step": 4981
+    },
+    {
+      "epoch": 293.06,
+      "learning_rate": 1.217750257997936e-06,
+      "loss": 0.0209,
+      "step": 4982
+    },
+    {
+      "epoch": 293.12,
+      "learning_rate": 1.2074303405572757e-06,
+      "loss": 0.0197,
+      "step": 4983
+    },
+    {
+      "epoch": 293.18,
+      "learning_rate": 1.197110423116615e-06,
+      "loss": 0.0213,
+      "step": 4984
+    },
+    {
+      "epoch": 293.24,
+      "learning_rate": 1.1867905056759547e-06,
+      "loss": 0.0202,
+      "step": 4985
+    },
+    {
+      "epoch": 293.29,
+      "learning_rate": 1.1764705882352942e-06,
+      "loss": 0.0177,
+      "step": 4986
+    },
+    {
+      "epoch": 293.35,
+      "learning_rate": 1.1661506707946337e-06,
+      "loss": 0.0184,
+      "step": 4987
+    },
+    {
+      "epoch": 293.41,
+      "learning_rate": 1.1558307533539732e-06,
+      "loss": 0.0194,
+      "step": 4988
+    },
+    {
+      "epoch": 293.47,
+      "learning_rate": 1.1455108359133127e-06,
+      "loss": 0.0183,
+      "step": 4989
+    },
+    {
+      "epoch": 293.53,
+      "learning_rate": 1.1351909184726523e-06,
+      "loss": 0.0199,
+      "step": 4990
+    },
+    {
+      "epoch": 293.59,
+      "learning_rate": 1.1248710010319918e-06,
+      "loss": 0.0158,
+      "step": 4991
+    },
+    {
+      "epoch": 293.65,
+      "learning_rate": 1.1145510835913315e-06,
+      "loss": 0.017,
+      "step": 4992
+    },
+    {
+      "epoch": 293.71,
+      "learning_rate": 1.1042311661506708e-06,
+      "loss": 0.0169,
+      "step": 4993
+    },
+    {
+      "epoch": 293.76,
+      "learning_rate": 1.0939112487100103e-06,
+      "loss": 0.023,
+      "step": 4994
+    },
+    {
+      "epoch": 293.82,
+      "learning_rate": 1.08359133126935e-06,
+      "loss": 0.016,
+      "step": 4995
+    },
+    {
+      "epoch": 293.88,
+      "learning_rate": 1.0732714138286893e-06,
+      "loss": 0.0182,
+      "step": 4996
+    },
+    {
+      "epoch": 293.94,
+      "learning_rate": 1.062951496388029e-06,
+      "loss": 0.0166,
+      "step": 4997
+    },
+    {
+      "epoch": 294.0,
+      "learning_rate": 1.0526315789473685e-06,
+      "loss": 0.0159,
+      "step": 4998
+    },
+    {
+      "epoch": 294.06,
+      "learning_rate": 1.042311661506708e-06,
+      "loss": 0.0159,
+      "step": 4999
+    },
+    {
+      "epoch": 294.12,
+      "learning_rate": 1.0319917440660475e-06,
+      "loss": 0.0179,
+      "step": 5000
+    },
+    {
+      "epoch": 294.12,
+      "eval_accuracy_NAT": 0.9938741685526348,
+      "eval_accuracy_REST": 0.44227642276422763,
+      "eval_accuracy_concrete": 0.9585864129678693,
+      "eval_accuracy_concrete tiles": 0.9298373792128212,
+      "eval_accuracy_dark bricks": 0.8507790517389889,
+      "eval_accuracy_grass": 0.9008921035400274,
+      "eval_accuracy_red bricks": 0.9312939367311072,
+      "eval_accuracy_road": 0.9809801298738083,
+      "eval_accuracy_rough concrete": 0.6255608721300926,
+      "eval_accuracy_rough red sidewalk": 0.7895682776979824,
+      "eval_accuracy_speedway bricks": 0.9899268133689498,
+      "eval_accuracy_steel": 0.9268468216325898,
+      "eval_accuracy_tiles": 0.24826562452664425,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.984281198822677,
+      "eval_iou_REST": 0.4243369734789392,
+      "eval_iou_concrete": 0.8939970086987269,
+      "eval_iou_concrete tiles": 0.8799794798590356,
+      "eval_iou_dark bricks": 0.835294062373735,
+      "eval_iou_grass": 0.8459462749373373,
+      "eval_iou_red bricks": 0.9060162427869203,
+      "eval_iou_road": 0.96142637611051,
+      "eval_iou_rough concrete": 0.6198506184867003,
+      "eval_iou_rough red sidewalk": 0.6280303619387139,
+      "eval_iou_speedway bricks": 0.9800719999714973,
+      "eval_iou_steel": 0.8749395515610441,
+      "eval_iou_tiles": 0.2462809917355372,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08469025790691376,
+      "eval_mean_accuracy": 0.8129760011336724,
+      "eval_mean_iou": 0.7754193185201056,
+      "eval_overall_accuracy": 0.9805464017189155,
+      "eval_runtime": 5.6252,
+      "eval_samples_per_second": 20.977,
+      "eval_steps_per_second": 1.422,
+      "step": 5000
+    },
+    {
+      "epoch": 294.18,
+      "learning_rate": 1.021671826625387e-06,
+      "loss": 0.0192,
+      "step": 5001
+    },
+    {
+      "epoch": 294.24,
+      "learning_rate": 1.0113519091847266e-06,
+      "loss": 0.0174,
+      "step": 5002
+    },
+    {
+      "epoch": 294.29,
+      "learning_rate": 1.001031991744066e-06,
+      "loss": 0.0185,
+      "step": 5003
+    },
+    {
+      "epoch": 294.35,
+      "learning_rate": 9.907120743034056e-07,
+      "loss": 0.0191,
+      "step": 5004
+    },
+    {
+      "epoch": 294.41,
+      "learning_rate": 9.80392156862745e-07,
+      "loss": 0.0182,
+      "step": 5005
+    },
+    {
+      "epoch": 294.47,
+      "learning_rate": 9.700722394220846e-07,
+      "loss": 0.0182,
+      "step": 5006
+    },
+    {
+      "epoch": 294.53,
+      "learning_rate": 9.597523219814243e-07,
+      "loss": 0.0192,
+      "step": 5007
+    },
+    {
+      "epoch": 294.59,
+      "learning_rate": 9.494324045407637e-07,
+      "loss": 0.0194,
+      "step": 5008
+    },
+    {
+      "epoch": 294.65,
+      "learning_rate": 9.391124871001033e-07,
+      "loss": 0.0157,
+      "step": 5009
+    },
+    {
+      "epoch": 294.71,
+      "learning_rate": 9.287925696594427e-07,
+      "loss": 0.0157,
+      "step": 5010
+    },
+    {
+      "epoch": 294.76,
+      "learning_rate": 9.184726522187823e-07,
+      "loss": 0.0206,
+      "step": 5011
+    },
+    {
+      "epoch": 294.82,
+      "learning_rate": 9.081527347781218e-07,
+      "loss": 0.0162,
+      "step": 5012
+    },
+    {
+      "epoch": 294.88,
+      "learning_rate": 8.978328173374612e-07,
+      "loss": 0.017,
+      "step": 5013
+    },
+    {
+      "epoch": 294.94,
+      "learning_rate": 8.875128998968009e-07,
+      "loss": 0.0248,
+      "step": 5014
+    },
+    {
+      "epoch": 295.0,
+      "learning_rate": 8.771929824561404e-07,
+      "loss": 0.0207,
+      "step": 5015
+    },
+    {
+      "epoch": 295.06,
+      "learning_rate": 8.6687306501548e-07,
+      "loss": 0.0194,
+      "step": 5016
+    },
+    {
+      "epoch": 295.12,
+      "learning_rate": 8.565531475748194e-07,
+      "loss": 0.0189,
+      "step": 5017
+    },
+    {
+      "epoch": 295.18,
+      "learning_rate": 8.46233230134159e-07,
+      "loss": 0.0176,
+      "step": 5018
+    },
+    {
+      "epoch": 295.24,
+      "learning_rate": 8.359133126934985e-07,
+      "loss": 0.0196,
+      "step": 5019
+    },
+    {
+      "epoch": 295.29,
+      "learning_rate": 8.255933952528379e-07,
+      "loss": 0.0163,
+      "step": 5020
+    },
+    {
+      "epoch": 295.29,
+      "eval_accuracy_NAT": 0.9937811707682463,
+      "eval_accuracy_REST": 0.44715447154471544,
+      "eval_accuracy_concrete": 0.958370852564328,
+      "eval_accuracy_concrete tiles": 0.9438133396181947,
+      "eval_accuracy_dark bricks": 0.8519753842620064,
+      "eval_accuracy_grass": 0.9005032039272237,
+      "eval_accuracy_red bricks": 0.9350834797891037,
+      "eval_accuracy_road": 0.9813515490853201,
+      "eval_accuracy_rough concrete": 0.625745594956471,
+      "eval_accuracy_rough red sidewalk": 0.7885154798878018,
+      "eval_accuracy_speedway bricks": 0.990214535466761,
+      "eval_accuracy_steel": 0.92558736949427,
+      "eval_accuracy_tiles": 0.2510224484231573,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9843039803180609,
+      "eval_iou_REST": 0.4198473282442748,
+      "eval_iou_concrete": 0.8940445425658754,
+      "eval_iou_concrete tiles": 0.8803640520576855,
+      "eval_iou_dark bricks": 0.8358403049679352,
+      "eval_iou_grass": 0.8458644344383669,
+      "eval_iou_red bricks": 0.90607205577138,
+      "eval_iou_road": 0.9612612409824798,
+      "eval_iou_rough concrete": 0.6199677653601601,
+      "eval_iou_rough red sidewalk": 0.6273693279724296,
+      "eval_iou_speedway bricks": 0.9800625481369465,
+      "eval_iou_steel": 0.8746770697057142,
+      "eval_iou_tiles": 0.24879147275183908,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0845131203532219,
+      "eval_mean_accuracy": 0.8148552984452,
+      "eval_mean_iou": 0.7752666248671652,
+      "eval_overall_accuracy": 0.9805485676910918,
+      "eval_runtime": 5.2412,
+      "eval_samples_per_second": 22.514,
+      "eval_steps_per_second": 1.526,
+      "step": 5020
+    },
+    {
+      "epoch": 295.35,
+      "learning_rate": 8.152734778121776e-07,
+      "loss": 0.0175,
+      "step": 5021
+    },
+    {
+      "epoch": 295.41,
+      "learning_rate": 8.04953560371517e-07,
+      "loss": 0.0206,
+      "step": 5022
+    },
+    {
+      "epoch": 295.47,
+      "learning_rate": 7.946336429308566e-07,
+      "loss": 0.0183,
+      "step": 5023
+    },
+    {
+      "epoch": 295.53,
+      "learning_rate": 7.843137254901962e-07,
+      "loss": 0.0184,
+      "step": 5024
+    },
+    {
+      "epoch": 295.59,
+      "learning_rate": 7.739938080495357e-07,
+      "loss": 0.0181,
+      "step": 5025
+    },
+    {
+      "epoch": 295.65,
+      "learning_rate": 7.636738906088752e-07,
+      "loss": 0.0177,
+      "step": 5026
+    },
+    {
+      "epoch": 295.71,
+      "learning_rate": 7.533539731682147e-07,
+      "loss": 0.0182,
+      "step": 5027
+    },
+    {
+      "epoch": 295.76,
+      "learning_rate": 7.430340557275542e-07,
+      "loss": 0.014,
+      "step": 5028
+    },
+    {
+      "epoch": 295.82,
+      "learning_rate": 7.327141382868937e-07,
+      "loss": 0.0199,
+      "step": 5029
+    },
+    {
+      "epoch": 295.88,
+      "learning_rate": 7.223942208462333e-07,
+      "loss": 0.0206,
+      "step": 5030
+    },
+    {
+      "epoch": 295.94,
+      "learning_rate": 7.120743034055728e-07,
+      "loss": 0.0197,
+      "step": 5031
+    },
+    {
+      "epoch": 296.0,
+      "learning_rate": 7.017543859649123e-07,
+      "loss": 0.0163,
+      "step": 5032
+    },
+    {
+      "epoch": 296.06,
+      "learning_rate": 6.914344685242518e-07,
+      "loss": 0.0217,
+      "step": 5033
+    },
+    {
+      "epoch": 296.12,
+      "learning_rate": 6.811145510835914e-07,
+      "loss": 0.0188,
+      "step": 5034
+    },
+    {
+      "epoch": 296.18,
+      "learning_rate": 6.707946336429308e-07,
+      "loss": 0.02,
+      "step": 5035
+    },
+    {
+      "epoch": 296.24,
+      "learning_rate": 6.604747162022703e-07,
+      "loss": 0.0217,
+      "step": 5036
+    },
+    {
+      "epoch": 296.29,
+      "learning_rate": 6.5015479876161e-07,
+      "loss": 0.0188,
+      "step": 5037
+    },
+    {
+      "epoch": 296.35,
+      "learning_rate": 6.398348813209495e-07,
+      "loss": 0.019,
+      "step": 5038
+    },
+    {
+      "epoch": 296.41,
+      "learning_rate": 6.29514963880289e-07,
+      "loss": 0.0165,
+      "step": 5039
+    },
+    {
+      "epoch": 296.47,
+      "learning_rate": 6.191950464396286e-07,
+      "loss": 0.0172,
+      "step": 5040
+    },
+    {
+      "epoch": 296.47,
+      "eval_accuracy_NAT": 0.9939002079322635,
+      "eval_accuracy_REST": 0.44227642276422763,
+      "eval_accuracy_concrete": 0.957671810050007,
+      "eval_accuracy_concrete tiles": 0.9471128918218242,
+      "eval_accuracy_dark bricks": 0.8509417529621194,
+      "eval_accuracy_grass": 0.9005448379490324,
+      "eval_accuracy_red bricks": 0.9342047451669596,
+      "eval_accuracy_road": 0.9813579620789001,
+      "eval_accuracy_rough concrete": 0.6260375761981659,
+      "eval_accuracy_rough red sidewalk": 0.7869287631881726,
+      "eval_accuracy_speedway bricks": 0.9901818047218641,
+      "eval_accuracy_steel": 0.9258622499212842,
+      "eval_accuracy_tiles": 0.25483958920294464,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842675207524041,
+      "eval_iou_REST": 0.4146341463414634,
+      "eval_iou_concrete": 0.8943860803862974,
+      "eval_iou_concrete tiles": 0.8798248494800219,
+      "eval_iou_dark bricks": 0.8355166095005403,
+      "eval_iou_grass": 0.84561943788917,
+      "eval_iou_red bricks": 0.906571443798966,
+      "eval_iou_road": 0.9612997285688485,
+      "eval_iou_rough concrete": 0.6197667490576166,
+      "eval_iou_rough red sidewalk": 0.6271162063200033,
+      "eval_iou_speedway bricks": 0.980086338950007,
+      "eval_iou_steel": 0.8747137454140245,
+      "eval_iou_tiles": 0.2524003840614498,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08426297456026077,
+      "eval_mean_accuracy": 0.8147585087659819,
+      "eval_mean_iou": 0.7750925569631394,
+      "eval_overall_accuracy": 0.9805557768223649,
+      "eval_runtime": 5.2371,
+      "eval_samples_per_second": 22.531,
+      "eval_steps_per_second": 1.528,
+      "step": 5040
+    },
+    {
+      "epoch": 296.53,
+      "learning_rate": 2.141795665634675e-05,
+      "loss": 0.018,
+      "step": 5041
+    },
+    {
+      "epoch": 296.59,
+      "learning_rate": 2.1411764705882352e-05,
+      "loss": 0.0152,
+      "step": 5042
+    },
+    {
+      "epoch": 296.65,
+      "learning_rate": 2.140557275541796e-05,
+      "loss": 0.0188,
+      "step": 5043
+    },
+    {
+      "epoch": 296.71,
+      "learning_rate": 2.1399380804953562e-05,
+      "loss": 0.0144,
+      "step": 5044
+    },
+    {
+      "epoch": 296.76,
+      "learning_rate": 2.1393188854489166e-05,
+      "loss": 0.0161,
+      "step": 5045
+    },
+    {
+      "epoch": 296.82,
+      "learning_rate": 2.138699690402477e-05,
+      "loss": 0.0181,
+      "step": 5046
+    },
+    {
+      "epoch": 296.88,
+      "learning_rate": 2.1380804953560372e-05,
+      "loss": 0.016,
+      "step": 5047
+    },
+    {
+      "epoch": 296.94,
+      "learning_rate": 2.1374613003095976e-05,
+      "loss": 0.0202,
+      "step": 5048
+    },
+    {
+      "epoch": 297.0,
+      "learning_rate": 2.136842105263158e-05,
+      "loss": 0.019,
+      "step": 5049
+    },
+    {
+      "epoch": 297.06,
+      "learning_rate": 2.1362229102167182e-05,
+      "loss": 0.0208,
+      "step": 5050
+    },
+    {
+      "epoch": 297.12,
+      "learning_rate": 2.1356037151702786e-05,
+      "loss": 0.0179,
+      "step": 5051
+    },
+    {
+      "epoch": 297.18,
+      "learning_rate": 2.1349845201238392e-05,
+      "loss": 0.0187,
+      "step": 5052
+    },
+    {
+      "epoch": 297.24,
+      "learning_rate": 2.1343653250773996e-05,
+      "loss": 0.0166,
+      "step": 5053
+    },
+    {
+      "epoch": 297.29,
+      "learning_rate": 2.13374613003096e-05,
+      "loss": 0.0166,
+      "step": 5054
+    },
+    {
+      "epoch": 297.35,
+      "learning_rate": 2.1331269349845202e-05,
+      "loss": 0.02,
+      "step": 5055
+    },
+    {
+      "epoch": 297.41,
+      "learning_rate": 2.1325077399380806e-05,
+      "loss": 0.0192,
+      "step": 5056
+    },
+    {
+      "epoch": 297.47,
+      "learning_rate": 2.131888544891641e-05,
+      "loss": 0.0169,
+      "step": 5057
+    },
+    {
+      "epoch": 297.53,
+      "learning_rate": 2.1312693498452012e-05,
+      "loss": 0.0186,
+      "step": 5058
+    },
+    {
+      "epoch": 297.59,
+      "learning_rate": 2.1306501547987616e-05,
+      "loss": 0.0189,
+      "step": 5059
+    },
+    {
+      "epoch": 297.65,
+      "learning_rate": 2.1300309597523222e-05,
+      "loss": 0.0203,
+      "step": 5060
+    },
+    {
+      "epoch": 297.65,
+      "eval_accuracy_NAT": 0.9945095725449838,
+      "eval_accuracy_REST": 0.44065040650406506,
+      "eval_accuracy_concrete": 0.9548905457653082,
+      "eval_accuracy_concrete tiles": 0.9539948149893943,
+      "eval_accuracy_dark bricks": 0.8580623241391191,
+      "eval_accuracy_grass": 0.8916096091322334,
+      "eval_accuracy_red bricks": 0.9342596660808435,
+      "eval_accuracy_road": 0.9801357523857672,
+      "eval_accuracy_rough concrete": 0.624452535171822,
+      "eval_accuracy_rough red sidewalk": 0.7857631656126155,
+      "eval_accuracy_speedway bricks": 0.9913428036573437,
+      "eval_accuracy_steel": 0.9137450209159016,
+      "eval_accuracy_tiles": 0.250840679814596,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9838999157601487,
+      "eval_iou_REST": 0.4195046439628483,
+      "eval_iou_concrete": 0.8946985811793186,
+      "eval_iou_concrete tiles": 0.8846490077804003,
+      "eval_iou_dark bricks": 0.8415551623637192,
+      "eval_iou_grass": 0.8421189039239921,
+      "eval_iou_red bricks": 0.9053701633934749,
+      "eval_iou_road": 0.9614100512306523,
+      "eval_iou_rough concrete": 0.619733052626599,
+      "eval_iou_rough red sidewalk": 0.6280164201441271,
+      "eval_iou_speedway bricks": 0.9798651590844342,
+      "eval_iou_steel": 0.8705393598152535,
+      "eval_iou_tiles": 0.2471641791044776,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08535641431808472,
+      "eval_mean_accuracy": 0.8134043766703072,
+      "eval_mean_iou": 0.7752711231053421,
+      "eval_overall_accuracy": 0.9804086523540949,
+      "eval_runtime": 5.6966,
+      "eval_samples_per_second": 20.714,
+      "eval_steps_per_second": 1.404,
+      "step": 5060
+    },
+    {
+      "epoch": 297.71,
+      "learning_rate": 2.1294117647058826e-05,
+      "loss": 0.0196,
+      "step": 5061
+    },
+    {
+      "epoch": 297.76,
+      "learning_rate": 2.128792569659443e-05,
+      "loss": 0.0184,
+      "step": 5062
+    },
+    {
+      "epoch": 297.82,
+      "learning_rate": 2.1281733746130032e-05,
+      "loss": 0.0204,
+      "step": 5063
+    },
+    {
+      "epoch": 297.88,
+      "learning_rate": 2.1275541795665636e-05,
+      "loss": 0.0185,
+      "step": 5064
+    },
+    {
+      "epoch": 297.94,
+      "learning_rate": 2.126934984520124e-05,
+      "loss": 0.0199,
+      "step": 5065
+    },
+    {
+      "epoch": 298.0,
+      "learning_rate": 2.1263157894736842e-05,
+      "loss": 0.0165,
+      "step": 5066
+    },
+    {
+      "epoch": 298.06,
+      "learning_rate": 2.1256965944272446e-05,
+      "loss": 0.0181,
+      "step": 5067
+    },
+    {
+      "epoch": 298.12,
+      "learning_rate": 2.1250773993808052e-05,
+      "loss": 0.0176,
+      "step": 5068
+    },
+    {
+      "epoch": 298.18,
+      "learning_rate": 2.1244582043343656e-05,
+      "loss": 0.0197,
+      "step": 5069
+    },
+    {
+      "epoch": 298.24,
+      "learning_rate": 2.1238390092879256e-05,
+      "loss": 0.0182,
+      "step": 5070
+    },
+    {
+      "epoch": 298.29,
+      "learning_rate": 2.1232198142414862e-05,
+      "loss": 0.0187,
+      "step": 5071
+    },
+    {
+      "epoch": 298.35,
+      "learning_rate": 2.1226006191950466e-05,
+      "loss": 0.0226,
+      "step": 5072
+    },
+    {
+      "epoch": 298.41,
+      "learning_rate": 2.121981424148607e-05,
+      "loss": 0.0172,
+      "step": 5073
+    },
+    {
+      "epoch": 298.47,
+      "learning_rate": 2.1213622291021672e-05,
+      "loss": 0.0168,
+      "step": 5074
+    },
+    {
+      "epoch": 298.53,
+      "learning_rate": 2.1207430340557276e-05,
+      "loss": 0.0199,
+      "step": 5075
+    },
+    {
+      "epoch": 298.59,
+      "learning_rate": 2.1201238390092882e-05,
+      "loss": 0.0203,
+      "step": 5076
+    },
+    {
+      "epoch": 298.65,
+      "learning_rate": 2.1195046439628486e-05,
+      "loss": 0.0204,
+      "step": 5077
+    },
+    {
+      "epoch": 298.71,
+      "learning_rate": 2.1188854489164086e-05,
+      "loss": 0.0174,
+      "step": 5078
+    },
+    {
+      "epoch": 298.76,
+      "learning_rate": 2.1182662538699692e-05,
+      "loss": 0.0195,
+      "step": 5079
+    },
+    {
+      "epoch": 298.82,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0205,
+      "step": 5080
+    },
+    {
+      "epoch": 298.82,
+      "eval_accuracy_NAT": 0.9930572436675112,
+      "eval_accuracy_REST": 0.3105691056910569,
+      "eval_accuracy_concrete": 0.9610565670247619,
+      "eval_accuracy_concrete tiles": 0.9416450624558096,
+      "eval_accuracy_dark bricks": 0.8558132189958463,
+      "eval_accuracy_grass": 0.9023672726309295,
+      "eval_accuracy_red bricks": 0.9214630931458699,
+      "eval_accuracy_road": 0.9789653810574065,
+      "eval_accuracy_rough concrete": 0.6242558947437418,
+      "eval_accuracy_rough red sidewalk": 0.7903653960399762,
+      "eval_accuracy_speedway bricks": 0.9895333589695607,
+      "eval_accuracy_steel": 0.9241404988829859,
+      "eval_accuracy_tiles": 0.2323305765094368,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9839668319658884,
+      "eval_iou_REST": 0.2898330804248862,
+      "eval_iou_concrete": 0.892051705121733,
+      "eval_iou_concrete tiles": 0.8871766403908071,
+      "eval_iou_dark bricks": 0.8377333920424205,
+      "eval_iou_grass": 0.8392674373066616,
+      "eval_iou_red bricks": 0.9016552020636286,
+      "eval_iou_road": 0.9608568943794434,
+      "eval_iou_rough concrete": 0.6151000780896798,
+      "eval_iou_rough red sidewalk": 0.623584205998398,
+      "eval_iou_speedway bricks": 0.979964106763287,
+      "eval_iou_steel": 0.8718644499349314,
+      "eval_iou_tiles": 0.2302173390970221,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08709858357906342,
+      "eval_mean_accuracy": 0.8019663592165303,
+      "eval_mean_iou": 0.7625593356599066,
+      "eval_overall_accuracy": 0.9801059011685647,
+      "eval_runtime": 5.7661,
+      "eval_samples_per_second": 20.465,
+      "eval_steps_per_second": 1.387,
+      "step": 5080
+    },
+    {
+      "epoch": 298.88,
+      "learning_rate": 2.11702786377709e-05,
+      "loss": 0.0183,
+      "step": 5081
+    },
+    {
+      "epoch": 298.94,
+      "learning_rate": 2.1164086687306502e-05,
+      "loss": 0.0184,
+      "step": 5082
+    },
+    {
+      "epoch": 299.0,
+      "learning_rate": 2.1157894736842106e-05,
+      "loss": 0.0187,
+      "step": 5083
+    },
+    {
+      "epoch": 299.06,
+      "learning_rate": 2.1151702786377712e-05,
+      "loss": 0.025,
+      "step": 5084
+    },
+    {
+      "epoch": 299.12,
+      "learning_rate": 2.1145510835913312e-05,
+      "loss": 0.0211,
+      "step": 5085
+    },
+    {
+      "epoch": 299.18,
+      "learning_rate": 2.1139318885448916e-05,
+      "loss": 0.0166,
+      "step": 5086
+    },
+    {
+      "epoch": 299.24,
+      "learning_rate": 2.1133126934984522e-05,
+      "loss": 0.0192,
+      "step": 5087
+    },
+    {
+      "epoch": 299.29,
+      "learning_rate": 2.1126934984520126e-05,
+      "loss": 0.0181,
+      "step": 5088
+    },
+    {
+      "epoch": 299.35,
+      "learning_rate": 2.112074303405573e-05,
+      "loss": 0.0205,
+      "step": 5089
+    },
+    {
+      "epoch": 299.41,
+      "learning_rate": 2.1114551083591332e-05,
+      "loss": 0.0191,
+      "step": 5090
+    },
+    {
+      "epoch": 299.47,
+      "learning_rate": 2.1108359133126936e-05,
+      "loss": 0.0204,
+      "step": 5091
+    },
+    {
+      "epoch": 299.53,
+      "learning_rate": 2.110216718266254e-05,
+      "loss": 0.0191,
+      "step": 5092
+    },
+    {
+      "epoch": 299.59,
+      "learning_rate": 2.1095975232198142e-05,
+      "loss": 0.0198,
+      "step": 5093
+    },
+    {
+      "epoch": 299.65,
+      "learning_rate": 2.1089783281733746e-05,
+      "loss": 0.0163,
+      "step": 5094
+    },
+    {
+      "epoch": 299.71,
+      "learning_rate": 2.1083591331269352e-05,
+      "loss": 0.0168,
+      "step": 5095
+    },
+    {
+      "epoch": 299.76,
+      "learning_rate": 2.1077399380804956e-05,
+      "loss": 0.0152,
+      "step": 5096
+    },
+    {
+      "epoch": 299.82,
+      "learning_rate": 2.107120743034056e-05,
+      "loss": 0.0153,
+      "step": 5097
+    },
+    {
+      "epoch": 299.88,
+      "learning_rate": 2.1065015479876162e-05,
+      "loss": 0.0174,
+      "step": 5098
+    },
+    {
+      "epoch": 299.94,
+      "learning_rate": 2.1058823529411766e-05,
+      "loss": 0.0197,
+      "step": 5099
+    },
+    {
+      "epoch": 300.0,
+      "learning_rate": 2.105263157894737e-05,
+      "loss": 0.0188,
+      "step": 5100
+    },
+    {
+      "epoch": 300.0,
+      "eval_accuracy_NAT": 0.9939894318935174,
+      "eval_accuracy_REST": 0.3934959349593496,
+      "eval_accuracy_concrete": 0.9573702548046273,
+      "eval_accuracy_concrete tiles": 0.9434598161678058,
+      "eval_accuracy_dark bricks": 0.8422276668644603,
+      "eval_accuracy_grass": 0.8962480176528252,
+      "eval_accuracy_red bricks": 0.9153668717047452,
+      "eval_accuracy_road": 0.9826656783531005,
+      "eval_accuracy_rough concrete": 0.6268777671181451,
+      "eval_accuracy_rough red sidewalk": 0.7988028184901376,
+      "eval_accuracy_speedway bricks": 0.9902383552235184,
+      "eval_accuracy_steel": 0.9197374142248123,
+      "eval_accuracy_tiles": 0.23099760671332062,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9842334968567513,
+      "eval_iou_REST": 0.3717357910906298,
+      "eval_iou_concrete": 0.8939276820528413,
+      "eval_iou_concrete tiles": 0.8782965465794901,
+      "eval_iou_dark bricks": 0.82738811583302,
+      "eval_iou_grass": 0.8436821374682232,
+      "eval_iou_red bricks": 0.8987328120787275,
+      "eval_iou_road": 0.9615784871741773,
+      "eval_iou_rough concrete": 0.621137155340379,
+      "eval_iou_rough red sidewalk": 0.6333261787221864,
+      "eval_iou_speedway bricks": 0.9794269638487522,
+      "eval_iou_steel": 0.8726231684764569,
+      "eval_iou_tiles": 0.22949586155003762,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08451282232999802,
+      "eval_mean_accuracy": 0.807036741090028,
+      "eval_mean_iou": 0.7688911074670517,
+      "eval_overall_accuracy": 0.9804017341743081,
+      "eval_runtime": 5.6635,
+      "eval_samples_per_second": 20.835,
+      "eval_steps_per_second": 1.413,
+      "step": 5100
+    },
+    {
+      "epoch": 300.06,
+      "learning_rate": 2.1046439628482972e-05,
+      "loss": 0.0168,
+      "step": 5101
+    },
+    {
+      "epoch": 300.12,
+      "learning_rate": 2.1040247678018576e-05,
+      "loss": 0.0176,
+      "step": 5102
+    },
+    {
+      "epoch": 300.18,
+      "learning_rate": 2.103405572755418e-05,
+      "loss": 0.0196,
+      "step": 5103
+    },
+    {
+      "epoch": 300.24,
+      "learning_rate": 2.1027863777089786e-05,
+      "loss": 0.0193,
+      "step": 5104
+    },
+    {
+      "epoch": 300.29,
+      "learning_rate": 2.102167182662539e-05,
+      "loss": 0.0217,
+      "step": 5105
+    },
+    {
+      "epoch": 300.35,
+      "learning_rate": 2.1015479876160992e-05,
+      "loss": 0.0185,
+      "step": 5106
+    },
+    {
+      "epoch": 300.41,
+      "learning_rate": 2.1009287925696596e-05,
+      "loss": 0.0191,
+      "step": 5107
+    },
+    {
+      "epoch": 300.47,
+      "learning_rate": 2.10030959752322e-05,
+      "loss": 0.0204,
+      "step": 5108
+    },
+    {
+      "epoch": 300.53,
+      "learning_rate": 2.0996904024767802e-05,
+      "loss": 0.0196,
+      "step": 5109
+    },
+    {
+      "epoch": 300.59,
+      "learning_rate": 2.0990712074303406e-05,
+      "loss": 0.0195,
+      "step": 5110
+    },
+    {
+      "epoch": 300.65,
+      "learning_rate": 2.098452012383901e-05,
+      "loss": 0.0202,
+      "step": 5111
+    },
+    {
+      "epoch": 300.71,
+      "learning_rate": 2.0978328173374616e-05,
+      "loss": 0.0191,
+      "step": 5112
+    },
+    {
+      "epoch": 300.76,
+      "learning_rate": 2.097213622291022e-05,
+      "loss": 0.0194,
+      "step": 5113
+    },
+    {
+      "epoch": 300.82,
+      "learning_rate": 2.0965944272445822e-05,
+      "loss": 0.0167,
+      "step": 5114
+    },
+    {
+      "epoch": 300.88,
+      "learning_rate": 2.0959752321981426e-05,
+      "loss": 0.017,
+      "step": 5115
+    },
+    {
+      "epoch": 300.94,
+      "learning_rate": 2.095356037151703e-05,
+      "loss": 0.0178,
+      "step": 5116
+    },
+    {
+      "epoch": 301.0,
+      "learning_rate": 2.0947368421052632e-05,
+      "loss": 0.0171,
+      "step": 5117
+    },
+    {
+      "epoch": 301.06,
+      "learning_rate": 2.0941176470588235e-05,
+      "loss": 0.0181,
+      "step": 5118
+    },
+    {
+      "epoch": 301.12,
+      "learning_rate": 2.093498452012384e-05,
+      "loss": 0.0206,
+      "step": 5119
+    },
+    {
+      "epoch": 301.18,
+      "learning_rate": 2.0928792569659446e-05,
+      "loss": 0.0206,
+      "step": 5120
+    },
+    {
+      "epoch": 301.18,
+      "eval_accuracy_NAT": 0.9935564126447012,
+      "eval_accuracy_REST": 0.45528455284552843,
+      "eval_accuracy_concrete": 0.9555651275246174,
+      "eval_accuracy_concrete tiles": 0.9445439547489983,
+      "eval_accuracy_dark bricks": 0.8542005627548188,
+      "eval_accuracy_grass": 0.9060348514611649,
+      "eval_accuracy_red bricks": 0.9257469244288224,
+      "eval_accuracy_road": 0.981486756366633,
+      "eval_accuracy_rough concrete": 0.6285819841615073,
+      "eval_accuracy_rough red sidewalk": 0.785545086066221,
+      "eval_accuracy_speedway bricks": 0.9910777017601944,
+      "eval_accuracy_steel": 0.9237131847646274,
+      "eval_accuracy_tiles": 0.2609591323578418,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9843549519613541,
+      "eval_iou_REST": 0.417910447761194,
+      "eval_iou_concrete": 0.8934263432260868,
+      "eval_iou_concrete tiles": 0.8865219988054948,
+      "eval_iou_dark bricks": 0.8381256456005258,
+      "eval_iou_grass": 0.847430731392594,
+      "eval_iou_red bricks": 0.9043403616073824,
+      "eval_iou_road": 0.9607921506875472,
+      "eval_iou_rough concrete": 0.6225353941303873,
+      "eval_iou_rough red sidewalk": 0.6310203391263901,
+      "eval_iou_speedway bricks": 0.9795492034360116,
+      "eval_iou_steel": 0.8737498729485624,
+      "eval_iou_tiles": 0.2586553763925172,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0831647664308548,
+      "eval_mean_accuracy": 0.8158689409142827,
+      "eval_mean_iou": 0.7768009859289269,
+      "eval_overall_accuracy": 0.980536153761007,
+      "eval_runtime": 5.8026,
+      "eval_samples_per_second": 20.336,
+      "eval_steps_per_second": 1.379,
+      "step": 5120
+    }
+  ],
+  "max_steps": 8500,
+  "num_train_epochs": 500,
+  "total_flos": 3.2342981014365143e+19,
+  "trial_name": null,
+  "trial_params": null
+}