diff --git "a/checkpoint-8240/trainer_state.json" "b/checkpoint-8240/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-8240/trainer_state.json"
+++ /dev/null
@@ -1,65524 +0,0 @@
-{
-  "best_metric": 0.07512204349040985,
-  "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-5520",
-  "epoch": 484.70588235294116,
-  "global_step": 8240,
-  "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
-    },
-    {
-      "epoch": 301.24,
-      "learning_rate": 2.092260061919505e-05,
-      "loss": 0.0195,
-      "step": 5121
-    },
-    {
-      "epoch": 301.29,
-      "learning_rate": 2.091640866873065e-05,
-      "loss": 0.0145,
-      "step": 5122
-    },
-    {
-      "epoch": 301.35,
-      "learning_rate": 2.0910216718266255e-05,
-      "loss": 0.0205,
-      "step": 5123
-    },
-    {
-      "epoch": 301.41,
-      "learning_rate": 2.090402476780186e-05,
-      "loss": 0.0178,
-      "step": 5124
-    },
-    {
-      "epoch": 301.47,
-      "learning_rate": 2.0897832817337462e-05,
-      "loss": 0.0166,
-      "step": 5125
-    },
-    {
-      "epoch": 301.53,
-      "learning_rate": 2.0891640866873065e-05,
-      "loss": 0.0178,
-      "step": 5126
-    },
-    {
-      "epoch": 301.59,
-      "learning_rate": 2.088544891640867e-05,
-      "loss": 0.0198,
-      "step": 5127
-    },
-    {
-      "epoch": 301.65,
-      "learning_rate": 2.0879256965944276e-05,
-      "loss": 0.0181,
-      "step": 5128
-    },
-    {
-      "epoch": 301.71,
-      "learning_rate": 2.087306501547988e-05,
-      "loss": 0.019,
-      "step": 5129
-    },
-    {
-      "epoch": 301.76,
-      "learning_rate": 2.086687306501548e-05,
-      "loss": 0.0205,
-      "step": 5130
-    },
-    {
-      "epoch": 301.82,
-      "learning_rate": 2.0860681114551085e-05,
-      "loss": 0.0172,
-      "step": 5131
-    },
-    {
-      "epoch": 301.88,
-      "learning_rate": 2.085448916408669e-05,
-      "loss": 0.0171,
-      "step": 5132
-    },
-    {
-      "epoch": 301.94,
-      "learning_rate": 2.0848297213622292e-05,
-      "loss": 0.0171,
-      "step": 5133
-    },
-    {
-      "epoch": 302.0,
-      "learning_rate": 2.0842105263157895e-05,
-      "loss": 0.0218,
-      "step": 5134
-    },
-    {
-      "epoch": 302.06,
-      "learning_rate": 2.08359133126935e-05,
-      "loss": 0.0206,
-      "step": 5135
-    },
-    {
-      "epoch": 302.12,
-      "learning_rate": 2.0829721362229105e-05,
-      "loss": 0.0211,
-      "step": 5136
-    },
-    {
-      "epoch": 302.18,
-      "learning_rate": 2.0823529411764705e-05,
-      "loss": 0.0219,
-      "step": 5137
-    },
-    {
-      "epoch": 302.24,
-      "learning_rate": 2.081733746130031e-05,
-      "loss": 0.0153,
-      "step": 5138
-    },
-    {
-      "epoch": 302.29,
-      "learning_rate": 2.0811145510835915e-05,
-      "loss": 0.019,
-      "step": 5139
-    },
-    {
-      "epoch": 302.35,
-      "learning_rate": 2.080495356037152e-05,
-      "loss": 0.0193,
-      "step": 5140
-    },
-    {
-      "epoch": 302.35,
-      "eval_accuracy_NAT": 0.9936958823654043,
-      "eval_accuracy_REST": 0.5626016260162602,
-      "eval_accuracy_concrete": 0.9592392093672462,
-      "eval_accuracy_concrete tiles": 0.9365071883101579,
-      "eval_accuracy_dark bricks": 0.8540952854927933,
-      "eval_accuracy_grass": 0.8986438663623598,
-      "eval_accuracy_red bricks": 0.9193211775043937,
-      "eval_accuracy_road": 0.9825384873137627,
-      "eval_accuracy_rough concrete": 0.6239758311037487,
-      "eval_accuracy_rough red sidewalk": 0.7898164371818107,
-      "eval_accuracy_speedway bricks": 0.987997070341284,
-      "eval_accuracy_steel": 0.9258572520953385,
-      "eval_accuracy_tiles": 0.2378745190705565,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841271748149145,
-      "eval_iou_REST": 0.49712643678160917,
-      "eval_iou_concrete": 0.8904722420614435,
-      "eval_iou_concrete tiles": 0.8768646835554771,
-      "eval_iou_dark bricks": 0.8375740283630698,
-      "eval_iou_grass": 0.8435923058816215,
-      "eval_iou_red bricks": 0.8946074501630057,
-      "eval_iou_road": 0.9585506572375377,
-      "eval_iou_rough concrete": 0.6173651070653712,
-      "eval_iou_rough red sidewalk": 0.6279948578432838,
-      "eval_iou_speedway bricks": 0.9800793351958773,
-      "eval_iou_steel": 0.8724700408090218,
-      "eval_iou_tiles": 0.23270701203248179,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08671916276216507,
-      "eval_mean_accuracy": 0.820935679425009,
-      "eval_mean_iou": 0.7779639486003627,
-      "eval_overall_accuracy": 0.9801435955500198,
-      "eval_runtime": 5.6852,
-      "eval_samples_per_second": 20.756,
-      "eval_steps_per_second": 1.407,
-      "step": 5140
-    },
-    {
-      "epoch": 302.41,
-      "learning_rate": 2.0798761609907122e-05,
-      "loss": 0.0198,
-      "step": 5141
-    },
-    {
-      "epoch": 302.47,
-      "learning_rate": 2.0792569659442725e-05,
-      "loss": 0.0212,
-      "step": 5142
-    },
-    {
-      "epoch": 302.53,
-      "learning_rate": 2.078637770897833e-05,
-      "loss": 0.0177,
-      "step": 5143
-    },
-    {
-      "epoch": 302.59,
-      "learning_rate": 2.0780185758513935e-05,
-      "loss": 0.0175,
-      "step": 5144
-    },
-    {
-      "epoch": 302.65,
-      "learning_rate": 2.0773993808049535e-05,
-      "loss": 0.0158,
-      "step": 5145
-    },
-    {
-      "epoch": 302.71,
-      "learning_rate": 2.076780185758514e-05,
-      "loss": 0.0155,
-      "step": 5146
-    },
-    {
-      "epoch": 302.76,
-      "learning_rate": 2.0761609907120742e-05,
-      "loss": 0.0174,
-      "step": 5147
-    },
-    {
-      "epoch": 302.82,
-      "learning_rate": 2.075541795665635e-05,
-      "loss": 0.021,
-      "step": 5148
-    },
-    {
-      "epoch": 302.88,
-      "learning_rate": 2.0749226006191952e-05,
-      "loss": 0.0176,
-      "step": 5149
-    },
-    {
-      "epoch": 302.94,
-      "learning_rate": 2.0743034055727555e-05,
-      "loss": 0.0193,
-      "step": 5150
-    },
-    {
-      "epoch": 303.0,
-      "learning_rate": 2.073684210526316e-05,
-      "loss": 0.0204,
-      "step": 5151
-    },
-    {
-      "epoch": 303.06,
-      "learning_rate": 2.0730650154798762e-05,
-      "loss": 0.022,
-      "step": 5152
-    },
-    {
-      "epoch": 303.12,
-      "learning_rate": 2.0724458204334365e-05,
-      "loss": 0.0203,
-      "step": 5153
-    },
-    {
-      "epoch": 303.18,
-      "learning_rate": 2.071826625386997e-05,
-      "loss": 0.0171,
-      "step": 5154
-    },
-    {
-      "epoch": 303.24,
-      "learning_rate": 2.0712074303405572e-05,
-      "loss": 0.0174,
-      "step": 5155
-    },
-    {
-      "epoch": 303.29,
-      "learning_rate": 2.070588235294118e-05,
-      "loss": 0.0199,
-      "step": 5156
-    },
-    {
-      "epoch": 303.35,
-      "learning_rate": 2.0699690402476782e-05,
-      "loss": 0.019,
-      "step": 5157
-    },
-    {
-      "epoch": 303.41,
-      "learning_rate": 2.0693498452012385e-05,
-      "loss": 0.0167,
-      "step": 5158
-    },
-    {
-      "epoch": 303.47,
-      "learning_rate": 2.068730650154799e-05,
-      "loss": 0.0213,
-      "step": 5159
-    },
-    {
-      "epoch": 303.53,
-      "learning_rate": 2.0681114551083592e-05,
-      "loss": 0.0208,
-      "step": 5160
-    },
-    {
-      "epoch": 303.53,
-      "eval_accuracy_NAT": 0.9938005250897509,
-      "eval_accuracy_REST": 0.4276422764227642,
-      "eval_accuracy_concrete": 0.9579485223410777,
-      "eval_accuracy_concrete tiles": 0.927386283290125,
-      "eval_accuracy_dark bricks": 0.85688513293647,
-      "eval_accuracy_grass": 0.899324204127824,
-      "eval_accuracy_red bricks": 0.9214630931458699,
-      "eval_accuracy_road": 0.9811131994905945,
-      "eval_accuracy_rough concrete": 0.6240890483199161,
-      "eval_accuracy_rough red sidewalk": 0.7872295625625099,
-      "eval_accuracy_speedway bricks": 0.9905428711066693,
-      "eval_accuracy_steel": 0.9237556662851659,
-      "eval_accuracy_tiles": 0.23884395164955013,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844290183461414,
-      "eval_iou_REST": 0.41029641185647425,
-      "eval_iou_concrete": 0.89288459880574,
-      "eval_iou_concrete tiles": 0.8791697388118059,
-      "eval_iou_dark bricks": 0.839250479230232,
-      "eval_iou_grass": 0.8451175247684983,
-      "eval_iou_red bricks": 0.8953997224890596,
-      "eval_iou_road": 0.9608984981413885,
-      "eval_iou_rough concrete": 0.6189513746070019,
-      "eval_iou_rough red sidewalk": 0.6293472967854803,
-      "eval_iou_speedway bricks": 0.9798234817393763,
-      "eval_iou_steel": 0.8744328745866312,
-      "eval_iou_tiles": 0.2325115017105108,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08478698134422302,
-      "eval_mean_accuracy": 0.810001872059099,
-      "eval_mean_iou": 0.7725009632214108,
-      "eval_overall_accuracy": 0.9804827156713454,
-      "eval_runtime": 5.5667,
-      "eval_samples_per_second": 21.197,
-      "eval_steps_per_second": 1.437,
-      "step": 5160
-    },
-    {
-      "epoch": 303.59,
-      "learning_rate": 2.0674922600619195e-05,
-      "loss": 0.0186,
-      "step": 5161
-    },
-    {
-      "epoch": 303.65,
-      "learning_rate": 2.06687306501548e-05,
-      "loss": 0.0167,
-      "step": 5162
-    },
-    {
-      "epoch": 303.71,
-      "learning_rate": 2.0662538699690402e-05,
-      "loss": 0.0191,
-      "step": 5163
-    },
-    {
-      "epoch": 303.76,
-      "learning_rate": 2.065634674922601e-05,
-      "loss": 0.0209,
-      "step": 5164
-    },
-    {
-      "epoch": 303.82,
-      "learning_rate": 2.0650154798761612e-05,
-      "loss": 0.0199,
-      "step": 5165
-    },
-    {
-      "epoch": 303.88,
-      "learning_rate": 2.0643962848297215e-05,
-      "loss": 0.0151,
-      "step": 5166
-    },
-    {
-      "epoch": 303.94,
-      "learning_rate": 2.063777089783282e-05,
-      "loss": 0.0202,
-      "step": 5167
-    },
-    {
-      "epoch": 304.0,
-      "learning_rate": 2.0631578947368422e-05,
-      "loss": 0.0209,
-      "step": 5168
-    },
-    {
-      "epoch": 304.06,
-      "learning_rate": 2.0625386996904025e-05,
-      "loss": 0.0191,
-      "step": 5169
-    },
-    {
-      "epoch": 304.12,
-      "learning_rate": 2.061919504643963e-05,
-      "loss": 0.0165,
-      "step": 5170
-    },
-    {
-      "epoch": 304.18,
-      "learning_rate": 2.0613003095975232e-05,
-      "loss": 0.0177,
-      "step": 5171
-    },
-    {
-      "epoch": 304.24,
-      "learning_rate": 2.060681114551084e-05,
-      "loss": 0.0171,
-      "step": 5172
-    },
-    {
-      "epoch": 304.29,
-      "learning_rate": 2.0600619195046442e-05,
-      "loss": 0.0193,
-      "step": 5173
-    },
-    {
-      "epoch": 304.35,
-      "learning_rate": 2.0594427244582042e-05,
-      "loss": 0.0189,
-      "step": 5174
-    },
-    {
-      "epoch": 304.41,
-      "learning_rate": 2.058823529411765e-05,
-      "loss": 0.015,
-      "step": 5175
-    },
-    {
-      "epoch": 304.47,
-      "learning_rate": 2.0582043343653252e-05,
-      "loss": 0.0207,
-      "step": 5176
-    },
-    {
-      "epoch": 304.53,
-      "learning_rate": 2.0575851393188855e-05,
-      "loss": 0.018,
-      "step": 5177
-    },
-    {
-      "epoch": 304.59,
-      "learning_rate": 2.056965944272446e-05,
-      "loss": 0.0214,
-      "step": 5178
-    },
-    {
-      "epoch": 304.65,
-      "learning_rate": 2.0563467492260062e-05,
-      "loss": 0.0187,
-      "step": 5179
-    },
-    {
-      "epoch": 304.71,
-      "learning_rate": 2.055727554179567e-05,
-      "loss": 0.0217,
-      "step": 5180
-    },
-    {
-      "epoch": 304.71,
-      "eval_accuracy_NAT": 0.9930087230843521,
-      "eval_accuracy_REST": 0.4585365853658537,
-      "eval_accuracy_concrete": 0.9544173830355491,
-      "eval_accuracy_concrete tiles": 0.9499646476549611,
-      "eval_accuracy_dark bricks": 0.8732701031717168,
-      "eval_accuracy_grass": 0.9173536280265095,
-      "eval_accuracy_red bricks": 0.9350834797891037,
-      "eval_accuracy_road": 0.9847386785278546,
-      "eval_accuracy_rough concrete": 0.6281171976951359,
-      "eval_accuracy_rough red sidewalk": 0.7900645966656389,
-      "eval_accuracy_speedway bricks": 0.9896802189087055,
-      "eval_accuracy_steel": 0.924347908659733,
-      "eval_accuracy_tiles": 0.23835923536005332,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844188477780315,
-      "eval_iou_REST": 0.44339622641509435,
-      "eval_iou_concrete": 0.8930620261634488,
-      "eval_iou_concrete tiles": 0.8820684523809523,
-      "eval_iou_dark bricks": 0.854005409806913,
-      "eval_iou_grass": 0.849703935605556,
-      "eval_iou_red bricks": 0.905975629223647,
-      "eval_iou_road": 0.9583916298510071,
-      "eval_iou_rough concrete": 0.6187812222998398,
-      "eval_iou_rough red sidewalk": 0.631040903357559,
-      "eval_iou_speedway bricks": 0.9803161748992104,
-      "eval_iou_steel": 0.8737306811981378,
-      "eval_iou_tiles": 0.2370379296839695,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08163490891456604,
-      "eval_mean_accuracy": 0.8182263373803974,
-      "eval_mean_iou": 0.7778406975894897,
-      "eval_overall_accuracy": 0.9805732662394896,
-      "eval_runtime": 6.298,
-      "eval_samples_per_second": 18.736,
-      "eval_steps_per_second": 1.27,
-      "step": 5180
-    },
-    {
-      "epoch": 304.76,
-      "learning_rate": 2.0551083591331272e-05,
-      "loss": 0.0188,
-      "step": 5181
-    },
-    {
-      "epoch": 304.82,
-      "learning_rate": 2.0544891640866872e-05,
-      "loss": 0.0176,
-      "step": 5182
-    },
-    {
-      "epoch": 304.88,
-      "learning_rate": 2.053869969040248e-05,
-      "loss": 0.0194,
-      "step": 5183
-    },
-    {
-      "epoch": 304.94,
-      "learning_rate": 2.0532507739938082e-05,
-      "loss": 0.0204,
-      "step": 5184
-    },
-    {
-      "epoch": 305.0,
-      "learning_rate": 2.0526315789473685e-05,
-      "loss": 0.0199,
-      "step": 5185
-    },
-    {
-      "epoch": 305.06,
-      "learning_rate": 2.052012383900929e-05,
-      "loss": 0.0176,
-      "step": 5186
-    },
-    {
-      "epoch": 305.12,
-      "learning_rate": 2.0513931888544892e-05,
-      "loss": 0.0217,
-      "step": 5187
-    },
-    {
-      "epoch": 305.18,
-      "learning_rate": 2.05077399380805e-05,
-      "loss": 0.0174,
-      "step": 5188
-    },
-    {
-      "epoch": 305.24,
-      "learning_rate": 2.05015479876161e-05,
-      "loss": 0.0175,
-      "step": 5189
-    },
-    {
-      "epoch": 305.29,
-      "learning_rate": 2.0495356037151702e-05,
-      "loss": 0.0202,
-      "step": 5190
-    },
-    {
-      "epoch": 305.35,
-      "learning_rate": 2.048916408668731e-05,
-      "loss": 0.0147,
-      "step": 5191
-    },
-    {
-      "epoch": 305.41,
-      "learning_rate": 2.0482972136222912e-05,
-      "loss": 0.0188,
-      "step": 5192
-    },
-    {
-      "epoch": 305.47,
-      "learning_rate": 2.0476780185758515e-05,
-      "loss": 0.019,
-      "step": 5193
-    },
-    {
-      "epoch": 305.53,
-      "learning_rate": 2.047058823529412e-05,
-      "loss": 0.0169,
-      "step": 5194
-    },
-    {
-      "epoch": 305.59,
-      "learning_rate": 2.0464396284829722e-05,
-      "loss": 0.0177,
-      "step": 5195
-    },
-    {
-      "epoch": 305.65,
-      "learning_rate": 2.045820433436533e-05,
-      "loss": 0.0199,
-      "step": 5196
-    },
-    {
-      "epoch": 305.71,
-      "learning_rate": 2.045201238390093e-05,
-      "loss": 0.0207,
-      "step": 5197
-    },
-    {
-      "epoch": 305.76,
-      "learning_rate": 2.0445820433436532e-05,
-      "loss": 0.0178,
-      "step": 5198
-    },
-    {
-      "epoch": 305.82,
-      "learning_rate": 2.0439628482972135e-05,
-      "loss": 0.0201,
-      "step": 5199
-    },
-    {
-      "epoch": 305.88,
-      "learning_rate": 2.0433436532507742e-05,
-      "loss": 0.02,
-      "step": 5200
-    },
-    {
-      "epoch": 305.88,
-      "eval_accuracy_NAT": 0.9934795883880325,
-      "eval_accuracy_REST": 0.4715447154471545,
-      "eval_accuracy_concrete": 0.9586869313829958,
-      "eval_accuracy_concrete tiles": 0.9494697148244167,
-      "eval_accuracy_dark bricks": 0.8488936316827135,
-      "eval_accuracy_grass": 0.9047858308069052,
-      "eval_accuracy_red bricks": 0.9413993848857645,
-      "eval_accuracy_road": 0.9818250417779811,
-      "eval_accuracy_rough concrete": 0.6308522872857066,
-      "eval_accuracy_rough red sidewalk": 0.788049240857579,
-      "eval_accuracy_speedway bricks": 0.9891287658349973,
-      "eval_accuracy_steel": 0.9333989714474203,
-      "eval_accuracy_tiles": 0.24629646460056348,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844328489185674,
-      "eval_iou_REST": 0.4461538461538462,
-      "eval_iou_concrete": 0.8930902143443858,
-      "eval_iou_concrete tiles": 0.8853871343486956,
-      "eval_iou_dark bricks": 0.8337445774525424,
-      "eval_iou_grass": 0.8467361090817963,
-      "eval_iou_red bricks": 0.904299657082564,
-      "eval_iou_road": 0.9603782146045872,
-      "eval_iou_rough concrete": 0.6252155528783692,
-      "eval_iou_rough red sidewalk": 0.6265447783949252,
-      "eval_iou_speedway bricks": 0.9797942213380995,
-      "eval_iou_steel": 0.8730413238593867,
-      "eval_iou_tiles": 0.24397563244605827,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08228321373462677,
-      "eval_mean_accuracy": 0.8182931207094023,
-      "eval_mean_iou": 0.7771380085310634,
-      "eval_overall_accuracy": 0.9804774462166479,
-      "eval_runtime": 5.7944,
-      "eval_samples_per_second": 20.364,
-      "eval_steps_per_second": 1.381,
-      "step": 5200
-    },
-    {
-      "epoch": 305.94,
-      "learning_rate": 2.0427244582043345e-05,
-      "loss": 0.0198,
-      "step": 5201
-    },
-    {
-      "epoch": 306.0,
-      "learning_rate": 2.042105263157895e-05,
-      "loss": 0.0241,
-      "step": 5202
-    },
-    {
-      "epoch": 306.06,
-      "learning_rate": 2.0414860681114552e-05,
-      "loss": 0.0199,
-      "step": 5203
-    },
-    {
-      "epoch": 306.12,
-      "learning_rate": 2.0408668730650155e-05,
-      "loss": 0.0201,
-      "step": 5204
-    },
-    {
-      "epoch": 306.18,
-      "learning_rate": 2.040247678018576e-05,
-      "loss": 0.0178,
-      "step": 5205
-    },
-    {
-      "epoch": 306.24,
-      "learning_rate": 2.0396284829721362e-05,
-      "loss": 0.0193,
-      "step": 5206
-    },
-    {
-      "epoch": 306.29,
-      "learning_rate": 2.0390092879256965e-05,
-      "loss": 0.0181,
-      "step": 5207
-    },
-    {
-      "epoch": 306.35,
-      "learning_rate": 2.0383900928792572e-05,
-      "loss": 0.0201,
-      "step": 5208
-    },
-    {
-      "epoch": 306.41,
-      "learning_rate": 2.0377708978328175e-05,
-      "loss": 0.0164,
-      "step": 5209
-    },
-    {
-      "epoch": 306.47,
-      "learning_rate": 2.037151702786378e-05,
-      "loss": 0.0181,
-      "step": 5210
-    },
-    {
-      "epoch": 306.53,
-      "learning_rate": 2.0365325077399382e-05,
-      "loss": 0.0186,
-      "step": 5211
-    },
-    {
-      "epoch": 306.59,
-      "learning_rate": 2.0359133126934985e-05,
-      "loss": 0.0188,
-      "step": 5212
-    },
-    {
-      "epoch": 306.65,
-      "learning_rate": 2.035294117647059e-05,
-      "loss": 0.0192,
-      "step": 5213
-    },
-    {
-      "epoch": 306.71,
-      "learning_rate": 2.0346749226006192e-05,
-      "loss": 0.0192,
-      "step": 5214
-    },
-    {
-      "epoch": 306.76,
-      "learning_rate": 2.0340557275541795e-05,
-      "loss": 0.0194,
-      "step": 5215
-    },
-    {
-      "epoch": 306.82,
-      "learning_rate": 2.0334365325077402e-05,
-      "loss": 0.0195,
-      "step": 5216
-    },
-    {
-      "epoch": 306.88,
-      "learning_rate": 2.0328173374613005e-05,
-      "loss": 0.0185,
-      "step": 5217
-    },
-    {
-      "epoch": 306.94,
-      "learning_rate": 2.032198142414861e-05,
-      "loss": 0.0173,
-      "step": 5218
-    },
-    {
-      "epoch": 307.0,
-      "learning_rate": 2.0315789473684212e-05,
-      "loss": 0.0173,
-      "step": 5219
-    },
-    {
-      "epoch": 307.06,
-      "learning_rate": 2.0309597523219815e-05,
-      "loss": 0.0216,
-      "step": 5220
-    },
-    {
-      "epoch": 307.06,
-      "eval_accuracy_NAT": 0.9935765217308327,
-      "eval_accuracy_REST": 0.42601626016260163,
-      "eval_accuracy_concrete": 0.9594868745117404,
-      "eval_accuracy_concrete tiles": 0.935234503888758,
-      "eval_accuracy_dark bricks": 0.8558658576268591,
-      "eval_accuracy_grass": 0.8956651413475041,
-      "eval_accuracy_red bricks": 0.9239894551845342,
-      "eval_accuracy_road": 0.9791465481260431,
-      "eval_accuracy_rough concrete": 0.6247206812101133,
-      "eval_accuracy_rough red sidewalk": 0.8019236119988871,
-      "eval_accuracy_speedway bricks": 0.9918054895799703,
-      "eval_accuracy_steel": 0.9096418058144707,
-      "eval_accuracy_tiles": 0.23205792359659486,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9842684160849654,
-      "eval_iou_REST": 0.4119496855345912,
-      "eval_iou_concrete": 0.8920786513260525,
-      "eval_iou_concrete tiles": 0.8747657782774508,
-      "eval_iou_dark bricks": 0.8409204178930442,
-      "eval_iou_grass": 0.841887871497374,
-      "eval_iou_red bricks": 0.8968973238085084,
-      "eval_iou_road": 0.9612883356077742,
-      "eval_iou_rough concrete": 0.6176250537564729,
-      "eval_iou_rough red sidewalk": 0.635425418446816,
-      "eval_iou_speedway bricks": 0.9799173321580885,
-      "eval_iou_steel": 0.8723456263495039,
-      "eval_iou_tiles": 0.22927267285243938,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08375765383243561,
-      "eval_mean_accuracy": 0.8099331288291468,
-      "eval_mean_iou": 0.7722032756610063,
-      "eval_overall_accuracy": 0.9803548263291181,
-      "eval_runtime": 5.7077,
-      "eval_samples_per_second": 20.674,
-      "eval_steps_per_second": 1.402,
-      "step": 5220
-    },
-    {
-      "epoch": 307.12,
-      "learning_rate": 2.030340557275542e-05,
-      "loss": 0.0176,
-      "step": 5221
-    },
-    {
-      "epoch": 307.18,
-      "learning_rate": 2.0297213622291022e-05,
-      "loss": 0.0174,
-      "step": 5222
-    },
-    {
-      "epoch": 307.24,
-      "learning_rate": 2.0291021671826625e-05,
-      "loss": 0.017,
-      "step": 5223
-    },
-    {
-      "epoch": 307.29,
-      "learning_rate": 2.0284829721362232e-05,
-      "loss": 0.019,
-      "step": 5224
-    },
-    {
-      "epoch": 307.35,
-      "learning_rate": 2.0278637770897835e-05,
-      "loss": 0.0235,
-      "step": 5225
-    },
-    {
-      "epoch": 307.41,
-      "learning_rate": 2.027244582043344e-05,
-      "loss": 0.0165,
-      "step": 5226
-    },
-    {
-      "epoch": 307.47,
-      "learning_rate": 2.0266253869969042e-05,
-      "loss": 0.0177,
-      "step": 5227
-    },
-    {
-      "epoch": 307.53,
-      "learning_rate": 2.0260061919504645e-05,
-      "loss": 0.0188,
-      "step": 5228
-    },
-    {
-      "epoch": 307.59,
-      "learning_rate": 2.025386996904025e-05,
-      "loss": 0.0189,
-      "step": 5229
-    },
-    {
-      "epoch": 307.65,
-      "learning_rate": 2.0247678018575852e-05,
-      "loss": 0.0187,
-      "step": 5230
-    },
-    {
-      "epoch": 307.71,
-      "learning_rate": 2.0241486068111455e-05,
-      "loss": 0.019,
-      "step": 5231
-    },
-    {
-      "epoch": 307.76,
-      "learning_rate": 2.0235294117647062e-05,
-      "loss": 0.0163,
-      "step": 5232
-    },
-    {
-      "epoch": 307.82,
-      "learning_rate": 2.0229102167182665e-05,
-      "loss": 0.0139,
-      "step": 5233
-    },
-    {
-      "epoch": 307.88,
-      "learning_rate": 2.0222910216718265e-05,
-      "loss": 0.0203,
-      "step": 5234
-    },
-    {
-      "epoch": 307.94,
-      "learning_rate": 2.0216718266253872e-05,
-      "loss": 0.0193,
-      "step": 5235
-    },
-    {
-      "epoch": 308.0,
-      "learning_rate": 2.0210526315789475e-05,
-      "loss": 0.0209,
-      "step": 5236
-    },
-    {
-      "epoch": 308.06,
-      "learning_rate": 2.020433436532508e-05,
-      "loss": 0.02,
-      "step": 5237
-    },
-    {
-      "epoch": 308.12,
-      "learning_rate": 2.0198142414860682e-05,
-      "loss": 0.0186,
-      "step": 5238
-    },
-    {
-      "epoch": 308.18,
-      "learning_rate": 2.0191950464396285e-05,
-      "loss": 0.0167,
-      "step": 5239
-    },
-    {
-      "epoch": 308.24,
-      "learning_rate": 2.0185758513931892e-05,
-      "loss": 0.0181,
-      "step": 5240
-    },
-    {
-      "epoch": 308.24,
-      "eval_accuracy_NAT": 0.9938104448534191,
-      "eval_accuracy_REST": 0.4097560975609756,
-      "eval_accuracy_concrete": 0.9602382783297585,
-      "eval_accuracy_concrete tiles": 0.9426820645769503,
-      "eval_accuracy_dark bricks": 0.860732538330494,
-      "eval_accuracy_grass": 0.8986996938006941,
-      "eval_accuracy_red bricks": 0.937719683655536,
-      "eval_accuracy_road": 0.9818384021812728,
-      "eval_accuracy_rough concrete": 0.6258349769692347,
-      "eval_accuracy_rough red sidewalk": 0.7931402702682379,
-      "eval_accuracy_speedway bricks": 0.9875049096117345,
-      "eval_accuracy_steel": 0.9376446245883041,
-      "eval_accuracy_tiles": 0.24090399587991154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843352353612732,
-      "eval_iou_REST": 0.38769230769230767,
-      "eval_iou_concrete": 0.8927239714175055,
-      "eval_iou_concrete tiles": 0.885048569468723,
-      "eval_iou_dark bricks": 0.8438690487360895,
-      "eval_iou_grass": 0.8436387183192796,
-      "eval_iou_red bricks": 0.9071299543087876,
-      "eval_iou_road": 0.9611479469856373,
-      "eval_iou_rough concrete": 0.6198111537326645,
-      "eval_iou_rough red sidewalk": 0.6237012507022264,
-      "eval_iou_speedway bricks": 0.9799434403760212,
-      "eval_iou_steel": 0.8725379853685987,
-      "eval_iou_tiles": 0.2392586352148273,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08506753295660019,
-      "eval_mean_accuracy": 0.8131158446620401,
-      "eval_mean_iou": 0.7723721705910724,
-      "eval_overall_accuracy": 0.9804047083450576,
-      "eval_runtime": 7.68,
-      "eval_samples_per_second": 15.365,
-      "eval_steps_per_second": 1.042,
-      "step": 5240
-    },
-    {
-      "epoch": 308.29,
-      "learning_rate": 2.0179566563467492e-05,
-      "loss": 0.0203,
-      "step": 5241
-    },
-    {
-      "epoch": 308.35,
-      "learning_rate": 2.0173374613003095e-05,
-      "loss": 0.021,
-      "step": 5242
-    },
-    {
-      "epoch": 308.41,
-      "learning_rate": 2.01671826625387e-05,
-      "loss": 0.0204,
-      "step": 5243
-    },
-    {
-      "epoch": 308.47,
-      "learning_rate": 2.0160990712074305e-05,
-      "loss": 0.0163,
-      "step": 5244
-    },
-    {
-      "epoch": 308.53,
-      "learning_rate": 2.015479876160991e-05,
-      "loss": 0.018,
-      "step": 5245
-    },
-    {
-      "epoch": 308.59,
-      "learning_rate": 2.0148606811145512e-05,
-      "loss": 0.0222,
-      "step": 5246
-    },
-    {
-      "epoch": 308.65,
-      "learning_rate": 2.0142414860681115e-05,
-      "loss": 0.0181,
-      "step": 5247
-    },
-    {
-      "epoch": 308.71,
-      "learning_rate": 2.0136222910216722e-05,
-      "loss": 0.0166,
-      "step": 5248
-    },
-    {
-      "epoch": 308.76,
-      "learning_rate": 2.013003095975232e-05,
-      "loss": 0.0181,
-      "step": 5249
-    },
-    {
-      "epoch": 308.82,
-      "learning_rate": 2.0123839009287925e-05,
-      "loss": 0.0176,
-      "step": 5250
-    },
-    {
-      "epoch": 308.88,
-      "learning_rate": 2.011764705882353e-05,
-      "loss": 0.0175,
-      "step": 5251
-    },
-    {
-      "epoch": 308.94,
-      "learning_rate": 2.0111455108359135e-05,
-      "loss": 0.0183,
-      "step": 5252
-    },
-    {
-      "epoch": 309.0,
-      "learning_rate": 2.010526315789474e-05,
-      "loss": 0.0163,
-      "step": 5253
-    },
-    {
-      "epoch": 309.06,
-      "learning_rate": 2.009907120743034e-05,
-      "loss": 0.0195,
-      "step": 5254
-    },
-    {
-      "epoch": 309.12,
-      "learning_rate": 2.0092879256965945e-05,
-      "loss": 0.016,
-      "step": 5255
-    },
-    {
-      "epoch": 309.18,
-      "learning_rate": 2.008668730650155e-05,
-      "loss": 0.0183,
-      "step": 5256
-    },
-    {
-      "epoch": 309.24,
-      "learning_rate": 2.008049535603715e-05,
-      "loss": 0.0184,
-      "step": 5257
-    },
-    {
-      "epoch": 309.29,
-      "learning_rate": 2.0074303405572755e-05,
-      "loss": 0.018,
-      "step": 5258
-    },
-    {
-      "epoch": 309.35,
-      "learning_rate": 2.0068111455108358e-05,
-      "loss": 0.0173,
-      "step": 5259
-    },
-    {
-      "epoch": 309.41,
-      "learning_rate": 2.0061919504643965e-05,
-      "loss": 0.0206,
-      "step": 5260
-    },
-    {
-      "epoch": 309.41,
-      "eval_accuracy_NAT": 0.9933379082852076,
-      "eval_accuracy_REST": 0.5642276422764227,
-      "eval_accuracy_concrete": 0.9566922332516447,
-      "eval_accuracy_concrete tiles": 0.947183596511902,
-      "eval_accuracy_dark bricks": 0.8751555232279923,
-      "eval_accuracy_grass": 0.9047792072125266,
-      "eval_accuracy_red bricks": 0.9462324253075571,
-      "eval_accuracy_road": 0.9834635616376862,
-      "eval_accuracy_rough concrete": 0.6272174187666474,
-      "eval_accuracy_rough red sidewalk": 0.8065709623323983,
-      "eval_accuracy_speedway bricks": 0.9903015889662774,
-      "eval_accuracy_steel": 0.9186303957778367,
-      "eval_accuracy_tiles": 0.25605137992668664,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9842703751431179,
-      "eval_iou_REST": 0.4880450070323488,
-      "eval_iou_concrete": 0.8949205140443363,
-      "eval_iou_concrete tiles": 0.8833329670088138,
-      "eval_iou_dark bricks": 0.8516960992511456,
-      "eval_iou_grass": 0.845656815074369,
-      "eval_iou_red bricks": 0.9010512002510329,
-      "eval_iou_road": 0.9598239595845588,
-      "eval_iou_rough concrete": 0.6185774815911802,
-      "eval_iou_rough red sidewalk": 0.6297085623033206,
-      "eval_iou_speedway bricks": 0.9805418612641733,
-      "eval_iou_steel": 0.8736254987654145,
-      "eval_iou_tiles": 0.2532889807905541,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08424735814332962,
-      "eval_mean_accuracy": 0.8284495264215989,
-      "eval_mean_iou": 0.7818876401618743,
-      "eval_overall_accuracy": 0.9806021350925251,
-      "eval_runtime": 5.6669,
-      "eval_samples_per_second": 20.823,
-      "eval_steps_per_second": 1.412,
-      "step": 5260
-    },
-    {
-      "epoch": 309.47,
-      "learning_rate": 2.005572755417957e-05,
-      "loss": 0.0167,
-      "step": 5261
-    },
-    {
-      "epoch": 309.53,
-      "learning_rate": 2.004953560371517e-05,
-      "loss": 0.0185,
-      "step": 5262
-    },
-    {
-      "epoch": 309.59,
-      "learning_rate": 2.0043343653250775e-05,
-      "loss": 0.0185,
-      "step": 5263
-    },
-    {
-      "epoch": 309.65,
-      "learning_rate": 2.003715170278638e-05,
-      "loss": 0.0169,
-      "step": 5264
-    },
-    {
-      "epoch": 309.71,
-      "learning_rate": 2.003095975232198e-05,
-      "loss": 0.0186,
-      "step": 5265
-    },
-    {
-      "epoch": 309.76,
-      "learning_rate": 2.0024767801857585e-05,
-      "loss": 0.0162,
-      "step": 5266
-    },
-    {
-      "epoch": 309.82,
-      "learning_rate": 2.0018575851393188e-05,
-      "loss": 0.0207,
-      "step": 5267
-    },
-    {
-      "epoch": 309.88,
-      "learning_rate": 2.0012383900928795e-05,
-      "loss": 0.0179,
-      "step": 5268
-    },
-    {
-      "epoch": 309.94,
-      "learning_rate": 2.00061919504644e-05,
-      "loss": 0.0171,
-      "step": 5269
-    },
-    {
-      "epoch": 310.0,
-      "learning_rate": 2e-05,
-      "loss": 0.0173,
-      "step": 5270
-    },
-    {
-      "epoch": 310.06,
-      "learning_rate": 1.9993808049535605e-05,
-      "loss": 0.0195,
-      "step": 5271
-    },
-    {
-      "epoch": 310.12,
-      "learning_rate": 1.9987616099071208e-05,
-      "loss": 0.0189,
-      "step": 5272
-    },
-    {
-      "epoch": 310.18,
-      "learning_rate": 1.998142414860681e-05,
-      "loss": 0.0212,
-      "step": 5273
-    },
-    {
-      "epoch": 310.24,
-      "learning_rate": 1.9975232198142415e-05,
-      "loss": 0.0173,
-      "step": 5274
-    },
-    {
-      "epoch": 310.29,
-      "learning_rate": 1.9969040247678018e-05,
-      "loss": 0.0166,
-      "step": 5275
-    },
-    {
-      "epoch": 310.35,
-      "learning_rate": 1.9962848297213625e-05,
-      "loss": 0.0191,
-      "step": 5276
-    },
-    {
-      "epoch": 310.41,
-      "learning_rate": 1.9956656346749228e-05,
-      "loss": 0.0165,
-      "step": 5277
-    },
-    {
-      "epoch": 310.47,
-      "learning_rate": 1.995046439628483e-05,
-      "loss": 0.0182,
-      "step": 5278
-    },
-    {
-      "epoch": 310.53,
-      "learning_rate": 1.9944272445820435e-05,
-      "loss": 0.0179,
-      "step": 5279
-    },
-    {
-      "epoch": 310.59,
-      "learning_rate": 1.9938080495356038e-05,
-      "loss": 0.0168,
-      "step": 5280
-    },
-    {
-      "epoch": 310.59,
-      "eval_accuracy_NAT": 0.9943529588848978,
-      "eval_accuracy_REST": 0.4764227642276423,
-      "eval_accuracy_concrete": 0.9577597158883305,
-      "eval_accuracy_concrete tiles": 0.9498468065048314,
-      "eval_accuracy_dark bricks": 0.877055299274544,
-      "eval_accuracy_grass": 0.8906529728584027,
-      "eval_accuracy_red bricks": 0.9370606326889279,
-      "eval_accuracy_road": 0.9826378887142536,
-      "eval_accuracy_rough concrete": 0.6255549133292416,
-      "eval_accuracy_rough red sidewalk": 0.7949450665142617,
-      "eval_accuracy_speedway bricks": 0.9884916301830249,
-      "eval_accuracy_steel": 0.9278638792125425,
-      "eval_accuracy_tiles": 0.24426671513829562,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9840971176339176,
-      "eval_iou_REST": 0.44801223241590216,
-      "eval_iou_concrete": 0.8934301473852041,
-      "eval_iou_concrete tiles": 0.882538431217974,
-      "eval_iou_dark bricks": 0.8586071525611115,
-      "eval_iou_grass": 0.8397211952953267,
-      "eval_iou_red bricks": 0.9080845175368567,
-      "eval_iou_road": 0.9609613203652119,
-      "eval_iou_rough concrete": 0.6197459148011712,
-      "eval_iou_rough red sidewalk": 0.6242566685760516,
-      "eval_iou_speedway bricks": 0.980767497810914,
-      "eval_iou_steel": 0.8769319062104408,
-      "eval_iou_tiles": 0.2428541308996717,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08514602482318878,
-      "eval_mean_accuracy": 0.8189931725707074,
-      "eval_mean_iou": 0.7784621717469041,
-      "eval_overall_accuracy": 0.9804842997405488,
-      "eval_runtime": 5.7286,
-      "eval_samples_per_second": 20.598,
-      "eval_steps_per_second": 1.396,
-      "step": 5280
-    },
-    {
-      "epoch": 310.65,
-      "learning_rate": 1.993188854489164e-05,
-      "loss": 0.0212,
-      "step": 5281
-    },
-    {
-      "epoch": 310.71,
-      "learning_rate": 1.9925696594427245e-05,
-      "loss": 0.019,
-      "step": 5282
-    },
-    {
-      "epoch": 310.76,
-      "learning_rate": 1.9919504643962848e-05,
-      "loss": 0.0205,
-      "step": 5283
-    },
-    {
-      "epoch": 310.82,
-      "learning_rate": 1.9913312693498455e-05,
-      "loss": 0.0179,
-      "step": 5284
-    },
-    {
-      "epoch": 310.88,
-      "learning_rate": 1.9907120743034058e-05,
-      "loss": 0.0173,
-      "step": 5285
-    },
-    {
-      "epoch": 310.94,
-      "learning_rate": 1.9900928792569658e-05,
-      "loss": 0.017,
-      "step": 5286
-    },
-    {
-      "epoch": 311.0,
-      "learning_rate": 1.9894736842105265e-05,
-      "loss": 0.0199,
-      "step": 5287
-    },
-    {
-      "epoch": 311.06,
-      "learning_rate": 1.9888544891640868e-05,
-      "loss": 0.018,
-      "step": 5288
-    },
-    {
-      "epoch": 311.12,
-      "learning_rate": 1.988235294117647e-05,
-      "loss": 0.0204,
-      "step": 5289
-    },
-    {
-      "epoch": 311.18,
-      "learning_rate": 1.9876160990712075e-05,
-      "loss": 0.0171,
-      "step": 5290
-    },
-    {
-      "epoch": 311.24,
-      "learning_rate": 1.9869969040247678e-05,
-      "loss": 0.02,
-      "step": 5291
-    },
-    {
-      "epoch": 311.29,
-      "learning_rate": 1.9863777089783285e-05,
-      "loss": 0.0157,
-      "step": 5292
-    },
-    {
-      "epoch": 311.35,
-      "learning_rate": 1.9857585139318888e-05,
-      "loss": 0.0173,
-      "step": 5293
-    },
-    {
-      "epoch": 311.41,
-      "learning_rate": 1.9851393188854488e-05,
-      "loss": 0.0179,
-      "step": 5294
-    },
-    {
-      "epoch": 311.47,
-      "learning_rate": 1.984520123839009e-05,
-      "loss": 0.018,
-      "step": 5295
-    },
-    {
-      "epoch": 311.53,
-      "learning_rate": 1.9839009287925698e-05,
-      "loss": 0.0261,
-      "step": 5296
-    },
-    {
-      "epoch": 311.59,
-      "learning_rate": 1.98328173374613e-05,
-      "loss": 0.0181,
-      "step": 5297
-    },
-    {
-      "epoch": 311.65,
-      "learning_rate": 1.9826625386996905e-05,
-      "loss": 0.0185,
-      "step": 5298
-    },
-    {
-      "epoch": 311.71,
-      "learning_rate": 1.9820433436532508e-05,
-      "loss": 0.0207,
-      "step": 5299
-    },
-    {
-      "epoch": 311.76,
-      "learning_rate": 1.9814241486068115e-05,
-      "loss": 0.0153,
-      "step": 5300
-    },
-    {
-      "epoch": 311.76,
-      "eval_accuracy_NAT": 0.9947765435758775,
-      "eval_accuracy_REST": 0.4211382113821138,
-      "eval_accuracy_concrete": 0.9401407104932052,
-      "eval_accuracy_concrete tiles": 0.9294602875324063,
-      "eval_accuracy_dark bricks": 0.8532339260762207,
-      "eval_accuracy_grass": 0.8909179166335487,
-      "eval_accuracy_red bricks": 0.9210237258347979,
-      "eval_accuracy_road": 0.9862099261383465,
-      "eval_accuracy_rough concrete": 0.8866457314130104,
-      "eval_accuracy_rough red sidewalk": 0.8041194474315494,
-      "eval_accuracy_speedway bricks": 0.9904515334782397,
-      "eval_accuracy_steel": 0.9114984981533033,
-      "eval_accuracy_tiles": 0.21921294192492957,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9840817906250918,
-      "eval_iou_REST": 0.39907550077041604,
-      "eval_iou_concrete": 0.8981820691337342,
-      "eval_iou_concrete tiles": 0.8796029887364782,
-      "eval_iou_dark bricks": 0.8366861406348983,
-      "eval_iou_grass": 0.8438820040278454,
-      "eval_iou_red bricks": 0.901273714193583,
-      "eval_iou_road": 0.9515259295620107,
-      "eval_iou_rough concrete": 0.8549086751432068,
-      "eval_iou_rough red sidewalk": 0.6266724490573335,
-      "eval_iou_speedway bricks": 0.9786919520069819,
-      "eval_iou_steel": 0.8682295646212841,
-      "eval_iou_tiles": 0.21795180722891566,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0752524733543396,
-      "eval_mean_accuracy": 0.8268330307744269,
-      "eval_mean_iou": 0.7877511219801367,
-      "eval_overall_accuracy": 0.980860920275672,
-      "eval_runtime": 8.4416,
-      "eval_samples_per_second": 13.978,
-      "eval_steps_per_second": 0.948,
-      "step": 5300
-    },
-    {
-      "epoch": 311.82,
-      "learning_rate": 1.9808049535603715e-05,
-      "loss": 0.0162,
-      "step": 5301
-    },
-    {
-      "epoch": 311.88,
-      "learning_rate": 1.9801857585139318e-05,
-      "loss": 0.02,
-      "step": 5302
-    },
-    {
-      "epoch": 311.94,
-      "learning_rate": 1.979566563467492e-05,
-      "loss": 0.0166,
-      "step": 5303
-    },
-    {
-      "epoch": 312.0,
-      "learning_rate": 1.9789473684210528e-05,
-      "loss": 0.0334,
-      "step": 5304
-    },
-    {
-      "epoch": 312.06,
-      "learning_rate": 1.978328173374613e-05,
-      "loss": 0.0165,
-      "step": 5305
-    },
-    {
-      "epoch": 312.12,
-      "learning_rate": 1.9777089783281735e-05,
-      "loss": 0.0199,
-      "step": 5306
-    },
-    {
-      "epoch": 312.18,
-      "learning_rate": 1.9770897832817338e-05,
-      "loss": 0.0202,
-      "step": 5307
-    },
-    {
-      "epoch": 312.24,
-      "learning_rate": 1.976470588235294e-05,
-      "loss": 0.0194,
-      "step": 5308
-    },
-    {
-      "epoch": 312.29,
-      "learning_rate": 1.9758513931888545e-05,
-      "loss": 0.045,
-      "step": 5309
-    },
-    {
-      "epoch": 312.35,
-      "learning_rate": 1.9752321981424148e-05,
-      "loss": 0.0218,
-      "step": 5310
-    },
-    {
-      "epoch": 312.41,
-      "learning_rate": 1.974613003095975e-05,
-      "loss": 0.0218,
-      "step": 5311
-    },
-    {
-      "epoch": 312.47,
-      "learning_rate": 1.9739938080495358e-05,
-      "loss": 0.0241,
-      "step": 5312
-    },
-    {
-      "epoch": 312.53,
-      "learning_rate": 1.973374613003096e-05,
-      "loss": 0.0215,
-      "step": 5313
-    },
-    {
-      "epoch": 312.59,
-      "learning_rate": 1.9727554179566565e-05,
-      "loss": 0.0225,
-      "step": 5314
-    },
-    {
-      "epoch": 312.65,
-      "learning_rate": 1.9721362229102168e-05,
-      "loss": 0.0225,
-      "step": 5315
-    },
-    {
-      "epoch": 312.71,
-      "learning_rate": 1.971517027863777e-05,
-      "loss": 0.0261,
-      "step": 5316
-    },
-    {
-      "epoch": 312.76,
-      "learning_rate": 1.9708978328173375e-05,
-      "loss": 0.0198,
-      "step": 5317
-    },
-    {
-      "epoch": 312.82,
-      "learning_rate": 1.9702786377708978e-05,
-      "loss": 0.0211,
-      "step": 5318
-    },
-    {
-      "epoch": 312.88,
-      "learning_rate": 1.969659442724458e-05,
-      "loss": 0.0194,
-      "step": 5319
-    },
-    {
-      "epoch": 312.94,
-      "learning_rate": 1.9690402476780188e-05,
-      "loss": 0.0199,
-      "step": 5320
-    },
-    {
-      "epoch": 312.94,
-      "eval_accuracy_NAT": 0.9941545096997767,
-      "eval_accuracy_REST": 0.4276422764227642,
-      "eval_accuracy_concrete": 0.9650390837001173,
-      "eval_accuracy_concrete tiles": 0.9373556445910912,
-      "eval_accuracy_dark bricks": 0.8544589705797906,
-      "eval_accuracy_grass": 0.889279050138717,
-      "eval_accuracy_red bricks": 0.9395869947275922,
-      "eval_accuracy_road": 0.9721863124271524,
-      "eval_accuracy_rough concrete": 0.6266870854909158,
-      "eval_accuracy_rough red sidewalk": 0.5984704351814948,
-      "eval_accuracy_speedway bricks": 0.9876605091738623,
-      "eval_accuracy_steel": 0.9382843463093554,
-      "eval_accuracy_tiles": 0.2031870096034415,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.983967023096148,
-      "eval_iou_REST": 0.41158059467918623,
-      "eval_iou_concrete": 0.8845961698001099,
-      "eval_iou_concrete tiles": 0.8863038730668092,
-      "eval_iou_dark bricks": 0.8338727227725085,
-      "eval_iou_grass": 0.8417879512849198,
-      "eval_iou_red bricks": 0.9138888888888889,
-      "eval_iou_road": 0.9520389365710697,
-      "eval_iou_rough concrete": 0.6225323933490786,
-      "eval_iou_rough red sidewalk": 0.5140753181319037,
-      "eval_iou_speedway bricks": 0.9790053884080432,
-      "eval_iou_steel": 0.8722631213367932,
-      "eval_iou_tiles": 0.20097084469481316,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08683325350284576,
-      "eval_mean_accuracy": 0.7949224790804672,
-      "eval_mean_iou": 0.7612987096984826,
-      "eval_overall_accuracy": 0.9792282621739274,
-      "eval_runtime": 5.6045,
-      "eval_samples_per_second": 21.054,
-      "eval_steps_per_second": 1.427,
-      "step": 5320
-    },
-    {
-      "epoch": 313.0,
-      "learning_rate": 1.968421052631579e-05,
-      "loss": 0.0226,
-      "step": 5321
-    },
-    {
-      "epoch": 313.06,
-      "learning_rate": 1.9678018575851395e-05,
-      "loss": 0.0246,
-      "step": 5322
-    },
-    {
-      "epoch": 313.12,
-      "learning_rate": 1.9671826625386998e-05,
-      "loss": 0.018,
-      "step": 5323
-    },
-    {
-      "epoch": 313.18,
-      "learning_rate": 1.96656346749226e-05,
-      "loss": 0.0207,
-      "step": 5324
-    },
-    {
-      "epoch": 313.24,
-      "learning_rate": 1.9659442724458205e-05,
-      "loss": 0.0219,
-      "step": 5325
-    },
-    {
-      "epoch": 313.29,
-      "learning_rate": 1.9653250773993808e-05,
-      "loss": 0.0181,
-      "step": 5326
-    },
-    {
-      "epoch": 313.35,
-      "learning_rate": 1.964705882352941e-05,
-      "loss": 0.0148,
-      "step": 5327
-    },
-    {
-      "epoch": 313.41,
-      "learning_rate": 1.9640866873065018e-05,
-      "loss": 0.0231,
-      "step": 5328
-    },
-    {
-      "epoch": 313.47,
-      "learning_rate": 1.963467492260062e-05,
-      "loss": 0.0265,
-      "step": 5329
-    },
-    {
-      "epoch": 313.53,
-      "learning_rate": 1.9628482972136225e-05,
-      "loss": 0.0187,
-      "step": 5330
-    },
-    {
-      "epoch": 313.59,
-      "learning_rate": 1.9622291021671828e-05,
-      "loss": 0.0167,
-      "step": 5331
-    },
-    {
-      "epoch": 313.65,
-      "learning_rate": 1.961609907120743e-05,
-      "loss": 0.0196,
-      "step": 5332
-    },
-    {
-      "epoch": 313.71,
-      "learning_rate": 1.9609907120743035e-05,
-      "loss": 0.0201,
-      "step": 5333
-    },
-    {
-      "epoch": 313.76,
-      "learning_rate": 1.9603715170278638e-05,
-      "loss": 0.0194,
-      "step": 5334
-    },
-    {
-      "epoch": 313.82,
-      "learning_rate": 1.959752321981424e-05,
-      "loss": 0.0234,
-      "step": 5335
-    },
-    {
-      "epoch": 313.88,
-      "learning_rate": 1.9591331269349848e-05,
-      "loss": 0.0192,
-      "step": 5336
-    },
-    {
-      "epoch": 313.94,
-      "learning_rate": 1.958513931888545e-05,
-      "loss": 0.0209,
-      "step": 5337
-    },
-    {
-      "epoch": 314.0,
-      "learning_rate": 1.957894736842105e-05,
-      "loss": 0.0191,
-      "step": 5338
-    },
-    {
-      "epoch": 314.06,
-      "learning_rate": 1.9572755417956655e-05,
-      "loss": 0.0225,
-      "step": 5339
-    },
-    {
-      "epoch": 314.12,
-      "learning_rate": 1.956656346749226e-05,
-      "loss": 0.0178,
-      "step": 5340
-    },
-    {
-      "epoch": 314.12,
-      "eval_accuracy_NAT": 0.9939240908415297,
-      "eval_accuracy_REST": 0.41788617886178864,
-      "eval_accuracy_concrete": 0.959872131403176,
-      "eval_accuracy_concrete tiles": 0.9347867075182654,
-      "eval_accuracy_dark bricks": 0.8498650536914036,
-      "eval_accuracy_grass": 0.9061881403596422,
-      "eval_accuracy_red bricks": 0.9080623901581723,
-      "eval_accuracy_road": 0.9757770544158538,
-      "eval_accuracy_rough concrete": 0.6306020176499681,
-      "eval_accuracy_rough red sidewalk": 0.788447800028576,
-      "eval_accuracy_speedway bricks": 0.9906755077378223,
-      "eval_accuracy_steel": 0.9161389795438984,
-      "eval_accuracy_tiles": 0.21630464418794873,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9842007409641246,
-      "eval_iou_REST": 0.4060031595576619,
-      "eval_iou_concrete": 0.8935389449707917,
-      "eval_iou_concrete tiles": 0.8810475809676129,
-      "eval_iou_dark bricks": 0.8355783481145169,
-      "eval_iou_grass": 0.8489460476009386,
-      "eval_iou_red bricks": 0.8895464572012697,
-      "eval_iou_road": 0.9646290784255922,
-      "eval_iou_rough concrete": 0.6261433964050316,
-      "eval_iou_rough red sidewalk": 0.6107366912677008,
-      "eval_iou_speedway bricks": 0.9798533646850487,
-      "eval_iou_steel": 0.8707781103035486,
-      "eval_iou_tiles": 0.21509911429776465,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0849614217877388,
-      "eval_mean_accuracy": 0.8068100535690803,
-      "eval_mean_iou": 0.7697000795970462,
-      "eval_overall_accuracy": 0.980528879973848,
-      "eval_runtime": 10.6904,
-      "eval_samples_per_second": 11.038,
-      "eval_steps_per_second": 0.748,
-      "step": 5340
-    },
-    {
-      "epoch": 314.18,
-      "learning_rate": 1.9560371517027865e-05,
-      "loss": 0.0212,
-      "step": 5341
-    },
-    {
-      "epoch": 314.24,
-      "learning_rate": 1.9554179566563468e-05,
-      "loss": 0.017,
-      "step": 5342
-    },
-    {
-      "epoch": 314.29,
-      "learning_rate": 1.954798761609907e-05,
-      "loss": 0.0224,
-      "step": 5343
-    },
-    {
-      "epoch": 314.35,
-      "learning_rate": 1.9541795665634678e-05,
-      "loss": 0.0181,
-      "step": 5344
-    },
-    {
-      "epoch": 314.41,
-      "learning_rate": 1.953560371517028e-05,
-      "loss": 0.0219,
-      "step": 5345
-    },
-    {
-      "epoch": 314.47,
-      "learning_rate": 1.952941176470588e-05,
-      "loss": 0.0194,
-      "step": 5346
-    },
-    {
-      "epoch": 314.53,
-      "learning_rate": 1.9523219814241485e-05,
-      "loss": 0.0183,
-      "step": 5347
-    },
-    {
-      "epoch": 314.59,
-      "learning_rate": 1.951702786377709e-05,
-      "loss": 0.0171,
-      "step": 5348
-    },
-    {
-      "epoch": 314.65,
-      "learning_rate": 1.9510835913312695e-05,
-      "loss": 0.0195,
-      "step": 5349
-    },
-    {
-      "epoch": 314.71,
-      "learning_rate": 1.9504643962848298e-05,
-      "loss": 0.0212,
-      "step": 5350
-    },
-    {
-      "epoch": 314.76,
-      "learning_rate": 1.94984520123839e-05,
-      "loss": 0.0195,
-      "step": 5351
-    },
-    {
-      "epoch": 314.82,
-      "learning_rate": 1.9492260061919508e-05,
-      "loss": 0.0208,
-      "step": 5352
-    },
-    {
-      "epoch": 314.88,
-      "learning_rate": 1.9486068111455108e-05,
-      "loss": 0.0198,
-      "step": 5353
-    },
-    {
-      "epoch": 314.94,
-      "learning_rate": 1.947987616099071e-05,
-      "loss": 0.0199,
-      "step": 5354
-    },
-    {
-      "epoch": 315.0,
-      "learning_rate": 1.9473684210526315e-05,
-      "loss": 0.0157,
-      "step": 5355
-    },
-    {
-      "epoch": 315.06,
-      "learning_rate": 1.946749226006192e-05,
-      "loss": 0.018,
-      "step": 5356
-    },
-    {
-      "epoch": 315.12,
-      "learning_rate": 1.9461300309597525e-05,
-      "loss": 0.0213,
-      "step": 5357
-    },
-    {
-      "epoch": 315.18,
-      "learning_rate": 1.9455108359133128e-05,
-      "loss": 0.019,
-      "step": 5358
-    },
-    {
-      "epoch": 315.24,
-      "learning_rate": 1.944891640866873e-05,
-      "loss": 0.0198,
-      "step": 5359
-    },
-    {
-      "epoch": 315.29,
-      "learning_rate": 1.9442724458204338e-05,
-      "loss": 0.017,
-      "step": 5360
-    },
-    {
-      "epoch": 315.29,
-      "eval_accuracy_NAT": 0.9942337060738443,
-      "eval_accuracy_REST": 0.45203252032520325,
-      "eval_accuracy_concrete": 0.9653085342045439,
-      "eval_accuracy_concrete tiles": 0.9470421871317464,
-      "eval_accuracy_dark bricks": 0.8803332503876118,
-      "eval_accuracy_grass": 0.8954219608110308,
-      "eval_accuracy_red bricks": 0.9409600175746925,
-      "eval_accuracy_road": 0.9738910998871848,
-      "eval_accuracy_rough concrete": 0.6238924078918359,
-      "eval_accuracy_rough red sidewalk": 0.7394626219177464,
-      "eval_accuracy_speedway bricks": 0.9886414033298306,
-      "eval_accuracy_steel": 0.9235607510732832,
-      "eval_accuracy_tiles": 0.22939198400436245,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9838101234247032,
-      "eval_iou_REST": 0.42967542503863987,
-      "eval_iou_concrete": 0.8924047994086599,
-      "eval_iou_concrete tiles": 0.8836866643209008,
-      "eval_iou_dark bricks": 0.8628509516617732,
-      "eval_iou_grass": 0.8438839599026549,
-      "eval_iou_red bricks": 0.9016893847692227,
-      "eval_iou_road": 0.9622644299058511,
-      "eval_iou_rough concrete": 0.6182192856594572,
-      "eval_iou_rough red sidewalk": 0.6110903961122091,
-      "eval_iou_speedway bricks": 0.9809556293914198,
-      "eval_iou_steel": 0.8726362191784703,
-      "eval_iou_tiles": 0.21774263120057513,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08720925450325012,
-      "eval_mean_accuracy": 0.811859418816378,
-      "eval_mean_iou": 0.7739161461518875,
-      "eval_overall_accuracy": 0.9804144067279363,
-      "eval_runtime": 6.3701,
-      "eval_samples_per_second": 18.524,
-      "eval_steps_per_second": 1.256,
-      "step": 5360
-    },
-    {
-      "epoch": 315.35,
-      "learning_rate": 1.9436532507739938e-05,
-      "loss": 0.0175,
-      "step": 5361
-    },
-    {
-      "epoch": 315.41,
-      "learning_rate": 1.943034055727554e-05,
-      "loss": 0.0209,
-      "step": 5362
-    },
-    {
-      "epoch": 315.47,
-      "learning_rate": 1.9424148606811145e-05,
-      "loss": 0.0211,
-      "step": 5363
-    },
-    {
-      "epoch": 315.53,
-      "learning_rate": 1.941795665634675e-05,
-      "loss": 0.021,
-      "step": 5364
-    },
-    {
-      "epoch": 315.59,
-      "learning_rate": 1.9411764705882355e-05,
-      "loss": 0.0201,
-      "step": 5365
-    },
-    {
-      "epoch": 315.65,
-      "learning_rate": 1.9405572755417958e-05,
-      "loss": 0.0184,
-      "step": 5366
-    },
-    {
-      "epoch": 315.71,
-      "learning_rate": 1.939938080495356e-05,
-      "loss": 0.0195,
-      "step": 5367
-    },
-    {
-      "epoch": 315.76,
-      "learning_rate": 1.9393188854489165e-05,
-      "loss": 0.0182,
-      "step": 5368
-    },
-    {
-      "epoch": 315.82,
-      "learning_rate": 1.9386996904024768e-05,
-      "loss": 0.0177,
-      "step": 5369
-    },
-    {
-      "epoch": 315.88,
-      "learning_rate": 1.938080495356037e-05,
-      "loss": 0.0196,
-      "step": 5370
-    },
-    {
-      "epoch": 315.94,
-      "learning_rate": 1.9374613003095975e-05,
-      "loss": 0.0173,
-      "step": 5371
-    },
-    {
-      "epoch": 316.0,
-      "learning_rate": 1.936842105263158e-05,
-      "loss": 0.0186,
-      "step": 5372
-    },
-    {
-      "epoch": 316.06,
-      "learning_rate": 1.9362229102167185e-05,
-      "loss": 0.0166,
-      "step": 5373
-    },
-    {
-      "epoch": 316.12,
-      "learning_rate": 1.9356037151702788e-05,
-      "loss": 0.0199,
-      "step": 5374
-    },
-    {
-      "epoch": 316.18,
-      "learning_rate": 1.934984520123839e-05,
-      "loss": 0.0179,
-      "step": 5375
-    },
-    {
-      "epoch": 316.24,
-      "learning_rate": 1.9343653250773995e-05,
-      "loss": 0.0175,
-      "step": 5376
-    },
-    {
-      "epoch": 316.29,
-      "learning_rate": 1.9337461300309598e-05,
-      "loss": 0.0225,
-      "step": 5377
-    },
-    {
-      "epoch": 316.35,
-      "learning_rate": 1.93312693498452e-05,
-      "loss": 0.0191,
-      "step": 5378
-    },
-    {
-      "epoch": 316.41,
-      "learning_rate": 1.9325077399380805e-05,
-      "loss": 0.0207,
-      "step": 5379
-    },
-    {
-      "epoch": 316.47,
-      "learning_rate": 1.931888544891641e-05,
-      "loss": 0.0214,
-      "step": 5380
-    },
-    {
-      "epoch": 316.47,
-      "eval_accuracy_NAT": 0.9944438541106827,
-      "eval_accuracy_REST": 0.21300813008130082,
-      "eval_accuracy_concrete": 0.955328546159738,
-      "eval_accuracy_concrete tiles": 0.93961819467358,
-      "eval_accuracy_dark bricks": 0.8456061099094615,
-      "eval_accuracy_grass": 0.8831768272604436,
-      "eval_accuracy_red bricks": 0.9237697715289982,
-      "eval_accuracy_road": 0.9837633690875539,
-      "eval_accuracy_rough concrete": 0.731979096526615,
-      "eval_accuracy_rough red sidewalk": 0.7827175719474504,
-      "eval_accuracy_speedway bricks": 0.9898481567621034,
-      "eval_accuracy_steel": 0.9252974955894187,
-      "eval_accuracy_tiles": 0.1308431033960435,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9835313750757679,
-      "eval_iou_REST": 0.21197411003236247,
-      "eval_iou_concrete": 0.899461558640737,
-      "eval_iou_concrete tiles": 0.8782465029188237,
-      "eval_iou_dark bricks": 0.8281958146837579,
-      "eval_iou_grass": 0.836784350289577,
-      "eval_iou_red bricks": 0.897976616304522,
-      "eval_iou_road": 0.9635212969088914,
-      "eval_iou_rough concrete": 0.7223035627135188,
-      "eval_iou_rough red sidewalk": 0.6153632407889136,
-      "eval_iou_speedway bricks": 0.9787299835880078,
-      "eval_iou_steel": 0.8745952150940187,
-      "eval_iou_tiles": 0.13014524196950522,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07800695300102234,
-      "eval_mean_accuracy": 0.7922615559256454,
-      "eval_mean_iou": 0.755448374539108,
-      "eval_overall_accuracy": 0.9805327593269995,
-      "eval_runtime": 5.8259,
-      "eval_samples_per_second": 20.254,
-      "eval_steps_per_second": 1.373,
-      "step": 5380
-    },
-    {
-      "epoch": 316.53,
-      "learning_rate": 1.9312693498452015e-05,
-      "loss": 0.0654,
-      "step": 5381
-    },
-    {
-      "epoch": 316.59,
-      "learning_rate": 1.9306501547987618e-05,
-      "loss": 0.0155,
-      "step": 5382
-    },
-    {
-      "epoch": 316.65,
-      "learning_rate": 1.930030959752322e-05,
-      "loss": 0.022,
-      "step": 5383
-    },
-    {
-      "epoch": 316.71,
-      "learning_rate": 1.9294117647058825e-05,
-      "loss": 0.0253,
-      "step": 5384
-    },
-    {
-      "epoch": 316.76,
-      "learning_rate": 1.9287925696594428e-05,
-      "loss": 0.0197,
-      "step": 5385
-    },
-    {
-      "epoch": 316.82,
-      "learning_rate": 1.928173374613003e-05,
-      "loss": 0.018,
-      "step": 5386
-    },
-    {
-      "epoch": 316.88,
-      "learning_rate": 1.9275541795665635e-05,
-      "loss": 0.0674,
-      "step": 5387
-    },
-    {
-      "epoch": 316.94,
-      "learning_rate": 1.926934984520124e-05,
-      "loss": 0.0164,
-      "step": 5388
-    },
-    {
-      "epoch": 317.0,
-      "learning_rate": 1.9263157894736845e-05,
-      "loss": 0.019,
-      "step": 5389
-    },
-    {
-      "epoch": 317.06,
-      "learning_rate": 1.9256965944272444e-05,
-      "loss": 0.0187,
-      "step": 5390
-    },
-    {
-      "epoch": 317.12,
-      "learning_rate": 1.9250773993808048e-05,
-      "loss": 0.0194,
-      "step": 5391
-    },
-    {
-      "epoch": 317.18,
-      "learning_rate": 1.9244582043343655e-05,
-      "loss": 0.021,
-      "step": 5392
-    },
-    {
-      "epoch": 317.24,
-      "learning_rate": 1.9238390092879258e-05,
-      "loss": 0.0204,
-      "step": 5393
-    },
-    {
-      "epoch": 317.29,
-      "learning_rate": 1.923219814241486e-05,
-      "loss": 0.0196,
-      "step": 5394
-    },
-    {
-      "epoch": 317.35,
-      "learning_rate": 1.9226006191950464e-05,
-      "loss": 0.0246,
-      "step": 5395
-    },
-    {
-      "epoch": 317.41,
-      "learning_rate": 1.921981424148607e-05,
-      "loss": 0.0211,
-      "step": 5396
-    },
-    {
-      "epoch": 317.47,
-      "learning_rate": 1.9213622291021675e-05,
-      "loss": 0.0212,
-      "step": 5397
-    },
-    {
-      "epoch": 317.53,
-      "learning_rate": 1.9207430340557274e-05,
-      "loss": 0.0205,
-      "step": 5398
-    },
-    {
-      "epoch": 317.59,
-      "learning_rate": 1.9201238390092878e-05,
-      "loss": 0.0221,
-      "step": 5399
-    },
-    {
-      "epoch": 317.65,
-      "learning_rate": 1.9195046439628485e-05,
-      "loss": 0.0192,
-      "step": 5400
-    },
-    {
-      "epoch": 317.65,
-      "eval_accuracy_NAT": 0.9931228542782944,
-      "eval_accuracy_REST": 0.35284552845528455,
-      "eval_accuracy_concrete": 0.9617223582002388,
-      "eval_accuracy_concrete tiles": 0.9215649304737215,
-      "eval_accuracy_dark bricks": 0.8046676109718048,
-      "eval_accuracy_grass": 0.9033759514320211,
-      "eval_accuracy_red bricks": 0.9130601933216169,
-      "eval_accuracy_road": 0.9712655134322823,
-      "eval_accuracy_rough concrete": 0.7786305483884423,
-      "eval_accuracy_rough red sidewalk": 0.7534497928244309,
-      "eval_accuracy_speedway bricks": 0.988837102338586,
-      "eval_accuracy_steel": 0.9377620734980283,
-      "eval_accuracy_tiles": 0.18155654518464662,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841012330176074,
-      "eval_iou_REST": 0.3505654281098546,
-      "eval_iou_concrete": 0.894002688810031,
-      "eval_iou_concrete tiles": 0.8681616341030195,
-      "eval_iou_dark bricks": 0.7948354344219288,
-      "eval_iou_grass": 0.8434173352721568,
-      "eval_iou_red bricks": 0.8917556187308909,
-      "eval_iou_road": 0.9530575001966491,
-      "eval_iou_rough concrete": 0.7684422855260992,
-      "eval_iou_rough red sidewalk": 0.5930158918055104,
-      "eval_iou_speedway bricks": 0.9781063284376185,
-      "eval_iou_steel": 0.8732158397594904,
-      "eval_iou_tiles": 0.1806099692604424,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0793975442647934,
-      "eval_mean_accuracy": 0.8047585386768769,
-      "eval_mean_iou": 0.7671759374962537,
-      "eval_overall_accuracy": 0.9800625817250397,
-      "eval_runtime": 6.8747,
-      "eval_samples_per_second": 17.164,
-      "eval_steps_per_second": 1.164,
-      "step": 5400
-    },
-    {
-      "epoch": 317.71,
-      "learning_rate": 1.9188854489164088e-05,
-      "loss": 0.0205,
-      "step": 5401
-    },
-    {
-      "epoch": 317.76,
-      "learning_rate": 1.918266253869969e-05,
-      "loss": 0.0215,
-      "step": 5402
-    },
-    {
-      "epoch": 317.82,
-      "learning_rate": 1.9176470588235294e-05,
-      "loss": 0.0189,
-      "step": 5403
-    },
-    {
-      "epoch": 317.88,
-      "learning_rate": 1.91702786377709e-05,
-      "loss": 0.0207,
-      "step": 5404
-    },
-    {
-      "epoch": 317.94,
-      "learning_rate": 1.91640866873065e-05,
-      "loss": 0.0196,
-      "step": 5405
-    },
-    {
-      "epoch": 318.0,
-      "learning_rate": 1.9157894736842104e-05,
-      "loss": 0.0162,
-      "step": 5406
-    },
-    {
-      "epoch": 318.06,
-      "learning_rate": 1.9151702786377708e-05,
-      "loss": 0.026,
-      "step": 5407
-    },
-    {
-      "epoch": 318.12,
-      "learning_rate": 1.9145510835913314e-05,
-      "loss": 0.0185,
-      "step": 5408
-    },
-    {
-      "epoch": 318.18,
-      "learning_rate": 1.9139318885448918e-05,
-      "loss": 0.0187,
-      "step": 5409
-    },
-    {
-      "epoch": 318.24,
-      "learning_rate": 1.913312693498452e-05,
-      "loss": 0.019,
-      "step": 5410
-    },
-    {
-      "epoch": 318.29,
-      "learning_rate": 1.9126934984520124e-05,
-      "loss": 0.0182,
-      "step": 5411
-    },
-    {
-      "epoch": 318.35,
-      "learning_rate": 1.912074303405573e-05,
-      "loss": 0.0173,
-      "step": 5412
-    },
-    {
-      "epoch": 318.41,
-      "learning_rate": 1.911455108359133e-05,
-      "loss": 0.0231,
-      "step": 5413
-    },
-    {
-      "epoch": 318.47,
-      "learning_rate": 1.9108359133126934e-05,
-      "loss": 0.0213,
-      "step": 5414
-    },
-    {
-      "epoch": 318.53,
-      "learning_rate": 1.9102167182662538e-05,
-      "loss": 0.0171,
-      "step": 5415
-    },
-    {
-      "epoch": 318.59,
-      "learning_rate": 1.9095975232198144e-05,
-      "loss": 0.0183,
-      "step": 5416
-    },
-    {
-      "epoch": 318.65,
-      "learning_rate": 1.9089783281733748e-05,
-      "loss": 0.0293,
-      "step": 5417
-    },
-    {
-      "epoch": 318.71,
-      "learning_rate": 1.908359133126935e-05,
-      "loss": 0.02,
-      "step": 5418
-    },
-    {
-      "epoch": 318.76,
-      "learning_rate": 1.9077399380804954e-05,
-      "loss": 0.0194,
-      "step": 5419
-    },
-    {
-      "epoch": 318.82,
-      "learning_rate": 1.9071207430340558e-05,
-      "loss": 0.0199,
-      "step": 5420
-    },
-    {
-      "epoch": 318.82,
-      "eval_accuracy_NAT": 0.9933527879307098,
-      "eval_accuracy_REST": 0.4373983739837398,
-      "eval_accuracy_concrete": 0.9597666443971876,
-      "eval_accuracy_concrete tiles": 0.9139995286353995,
-      "eval_accuracy_dark bricks": 0.8477307964703406,
-      "eval_accuracy_grass": 0.901931061629707,
-      "eval_accuracy_red bricks": 0.898451230228471,
-      "eval_accuracy_road": 0.9817972521391342,
-      "eval_accuracy_rough concrete": 0.7491940721849135,
-      "eval_accuracy_rough red sidewalk": 0.5848968634145241,
-      "eval_accuracy_speedway bricks": 0.9902697150471631,
-      "eval_accuracy_steel": 0.9157191621644585,
-      "eval_accuracy_tiles": 0.18134448180799176,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841385998960822,
-      "eval_iou_REST": 0.42902711323763953,
-      "eval_iou_concrete": 0.891705034760321,
-      "eval_iou_concrete tiles": 0.875575724735844,
-      "eval_iou_dark bricks": 0.8359617580716706,
-      "eval_iou_grass": 0.8419329764870069,
-      "eval_iou_red bricks": 0.8826481061832309,
-      "eval_iou_road": 0.9486512570349498,
-      "eval_iou_rough concrete": 0.744055439169596,
-      "eval_iou_rough red sidewalk": 0.5177397023191416,
-      "eval_iou_speedway bricks": 0.979950621856315,
-      "eval_iou_steel": 0.8735161188821191,
-      "eval_iou_tiles": 0.17940956091712873,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0805862545967102,
-      "eval_mean_accuracy": 0.796603997694903,
-      "eval_mean_iou": 0.7680240010423881,
-      "eval_overall_accuracy": 0.9799969560008938,
-      "eval_runtime": 8.6612,
-      "eval_samples_per_second": 13.624,
-      "eval_steps_per_second": 0.924,
-      "step": 5420
-    },
-    {
-      "epoch": 318.88,
-      "learning_rate": 1.906501547987616e-05,
-      "loss": 0.0183,
-      "step": 5421
-    },
-    {
-      "epoch": 318.94,
-      "learning_rate": 1.9058823529411764e-05,
-      "loss": 0.0209,
-      "step": 5422
-    },
-    {
-      "epoch": 319.0,
-      "learning_rate": 1.9052631578947368e-05,
-      "loss": 0.0186,
-      "step": 5423
-    },
-    {
-      "epoch": 319.06,
-      "learning_rate": 1.9046439628482974e-05,
-      "loss": 0.0205,
-      "step": 5424
-    },
-    {
-      "epoch": 319.12,
-      "learning_rate": 1.9040247678018578e-05,
-      "loss": 0.0195,
-      "step": 5425
-    },
-    {
-      "epoch": 319.18,
-      "learning_rate": 1.903405572755418e-05,
-      "loss": 0.0189,
-      "step": 5426
-    },
-    {
-      "epoch": 319.24,
-      "learning_rate": 1.9027863777089784e-05,
-      "loss": 0.0186,
-      "step": 5427
-    },
-    {
-      "epoch": 319.29,
-      "learning_rate": 1.9021671826625388e-05,
-      "loss": 0.021,
-      "step": 5428
-    },
-    {
-      "epoch": 319.35,
-      "learning_rate": 1.901547987616099e-05,
-      "loss": 0.0192,
-      "step": 5429
-    },
-    {
-      "epoch": 319.41,
-      "learning_rate": 1.9009287925696594e-05,
-      "loss": 0.0226,
-      "step": 5430
-    },
-    {
-      "epoch": 319.47,
-      "learning_rate": 1.9003095975232198e-05,
-      "loss": 0.0228,
-      "step": 5431
-    },
-    {
-      "epoch": 319.53,
-      "learning_rate": 1.8996904024767804e-05,
-      "loss": 0.0198,
-      "step": 5432
-    },
-    {
-      "epoch": 319.59,
-      "learning_rate": 1.8990712074303408e-05,
-      "loss": 0.0211,
-      "step": 5433
-    },
-    {
-      "epoch": 319.65,
-      "learning_rate": 1.898452012383901e-05,
-      "loss": 0.0194,
-      "step": 5434
-    },
-    {
-      "epoch": 319.71,
-      "learning_rate": 1.897832817337461e-05,
-      "loss": 0.0593,
-      "step": 5435
-    },
-    {
-      "epoch": 319.76,
-      "learning_rate": 1.8972136222910218e-05,
-      "loss": 0.018,
-      "step": 5436
-    },
-    {
-      "epoch": 319.82,
-      "learning_rate": 1.896594427244582e-05,
-      "loss": 0.0195,
-      "step": 5437
-    },
-    {
-      "epoch": 319.88,
-      "learning_rate": 1.8959752321981424e-05,
-      "loss": 0.0186,
-      "step": 5438
-    },
-    {
-      "epoch": 319.94,
-      "learning_rate": 1.8953560371517028e-05,
-      "loss": 0.0244,
-      "step": 5439
-    },
-    {
-      "epoch": 320.0,
-      "learning_rate": 1.8947368421052634e-05,
-      "loss": 0.0522,
-      "step": 5440
-    },
-    {
-      "epoch": 320.0,
-      "eval_accuracy_NAT": 0.9939061921375199,
-      "eval_accuracy_REST": 0.5154471544715448,
-      "eval_accuracy_concrete": 0.9493715879157755,
-      "eval_accuracy_concrete tiles": 0.951567287296724,
-      "eval_accuracy_dark bricks": 0.8744807916850105,
-      "eval_accuracy_grass": 0.9044631671378881,
-      "eval_accuracy_red bricks": 0.9384885764499121,
-      "eval_accuracy_road": 0.9510533609163313,
-      "eval_accuracy_rough concrete": 0.8251211126272949,
-      "eval_accuracy_rough red sidewalk": 0.8380496168567969,
-      "eval_accuracy_speedway bricks": 0.9896743924933836,
-      "eval_accuracy_steel": 0.9328067290728533,
-      "eval_accuracy_tiles": 0.2580205398527674,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9835432339681905,
-      "eval_iou_REST": 0.4899536321483771,
-      "eval_iou_concrete": 0.9003401326024204,
-      "eval_iou_concrete tiles": 0.8516495106311172,
-      "eval_iou_dark bricks": 0.8535835731101229,
-      "eval_iou_grass": 0.8501372333821314,
-      "eval_iou_red bricks": 0.916197522921023,
-      "eval_iou_road": 0.9432255807667234,
-      "eval_iou_rough concrete": 0.8062076434011038,
-      "eval_iou_rough red sidewalk": 0.4502147188455681,
-      "eval_iou_speedway bricks": 0.9806559750402307,
-      "eval_iou_steel": 0.8752347613230573,
-      "eval_iou_tiles": 0.25615037593984963,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07669045776128769,
-      "eval_mean_accuracy": 0.8401885006856771,
-      "eval_mean_iou": 0.7813149149292244,
-      "eval_overall_accuracy": 0.9796244411145226,
-      "eval_runtime": 5.3568,
-      "eval_samples_per_second": 22.028,
-      "eval_steps_per_second": 1.493,
-      "step": 5440
-    },
-    {
-      "epoch": 320.06,
-      "learning_rate": 1.8941176470588238e-05,
-      "loss": 0.0282,
-      "step": 5441
-    },
-    {
-      "epoch": 320.12,
-      "learning_rate": 1.893498452012384e-05,
-      "loss": 0.0255,
-      "step": 5442
-    },
-    {
-      "epoch": 320.18,
-      "learning_rate": 1.892879256965944e-05,
-      "loss": 0.0204,
-      "step": 5443
-    },
-    {
-      "epoch": 320.24,
-      "learning_rate": 1.8922600619195048e-05,
-      "loss": 0.0193,
-      "step": 5444
-    },
-    {
-      "epoch": 320.29,
-      "learning_rate": 1.891640866873065e-05,
-      "loss": 0.0209,
-      "step": 5445
-    },
-    {
-      "epoch": 320.35,
-      "learning_rate": 1.8910216718266254e-05,
-      "loss": 0.0195,
-      "step": 5446
-    },
-    {
-      "epoch": 320.41,
-      "learning_rate": 1.8904024767801858e-05,
-      "loss": 0.0226,
-      "step": 5447
-    },
-    {
-      "epoch": 320.47,
-      "learning_rate": 1.8897832817337464e-05,
-      "loss": 0.0208,
-      "step": 5448
-    },
-    {
-      "epoch": 320.53,
-      "learning_rate": 1.8891640866873068e-05,
-      "loss": 0.0217,
-      "step": 5449
-    },
-    {
-      "epoch": 320.59,
-      "learning_rate": 1.8885448916408668e-05,
-      "loss": 0.0182,
-      "step": 5450
-    },
-    {
-      "epoch": 320.65,
-      "learning_rate": 1.887925696594427e-05,
-      "loss": 0.0203,
-      "step": 5451
-    },
-    {
-      "epoch": 320.71,
-      "learning_rate": 1.8873065015479878e-05,
-      "loss": 0.0205,
-      "step": 5452
-    },
-    {
-      "epoch": 320.76,
-      "learning_rate": 1.886687306501548e-05,
-      "loss": 0.0226,
-      "step": 5453
-    },
-    {
-      "epoch": 320.82,
-      "learning_rate": 1.8860681114551084e-05,
-      "loss": 0.0179,
-      "step": 5454
-    },
-    {
-      "epoch": 320.88,
-      "learning_rate": 1.8854489164086688e-05,
-      "loss": 0.0227,
-      "step": 5455
-    },
-    {
-      "epoch": 320.94,
-      "learning_rate": 1.8848297213622294e-05,
-      "loss": 0.0186,
-      "step": 5456
-    },
-    {
-      "epoch": 321.0,
-      "learning_rate": 1.8842105263157894e-05,
-      "loss": 0.022,
-      "step": 5457
-    },
-    {
-      "epoch": 321.06,
-      "learning_rate": 1.8835913312693498e-05,
-      "loss": 0.0204,
-      "step": 5458
-    },
-    {
-      "epoch": 321.12,
-      "learning_rate": 1.88297213622291e-05,
-      "loss": 0.0192,
-      "step": 5459
-    },
-    {
-      "epoch": 321.18,
-      "learning_rate": 1.8823529411764708e-05,
-      "loss": 0.0183,
-      "step": 5460
-    },
-    {
-      "epoch": 321.18,
-      "eval_accuracy_NAT": 0.9938305539395506,
-      "eval_accuracy_REST": 0.5642276422764227,
-      "eval_accuracy_concrete": 0.9601270583343143,
-      "eval_accuracy_concrete tiles": 0.9131275041244402,
-      "eval_accuracy_dark bricks": 0.8574402312271501,
-      "eval_accuracy_grass": 0.9028470101094975,
-      "eval_accuracy_red bricks": 0.945573374340949,
-      "eval_accuracy_road": 0.9798070864647892,
-      "eval_accuracy_rough concrete": 0.8399108563392703,
-      "eval_accuracy_rough red sidewalk": 0.7226404169079328,
-      "eval_accuracy_speedway bricks": 0.989078727209291,
-      "eval_accuracy_steel": 0.9246252879997201,
-      "eval_accuracy_tiles": 0.17855736314338513,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841915007687464,
-      "eval_iou_REST": 0.528158295281583,
-      "eval_iou_concrete": 0.9065795205084705,
-      "eval_iou_concrete tiles": 0.848216826849399,
-      "eval_iou_dark bricks": 0.8383207478314572,
-      "eval_iou_grass": 0.8467136573102456,
-      "eval_iou_red bricks": 0.9117242109722516,
-      "eval_iou_road": 0.9568492618066831,
-      "eval_iou_rough concrete": 0.828565046409236,
-      "eval_iou_rough red sidewalk": 0.5554874735539961,
-      "eval_iou_speedway bricks": 0.980024634071631,
-      "eval_iou_steel": 0.8755791665680683,
-      "eval_iou_tiles": 0.1776050141625987,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07211485505104065,
-      "eval_mean_accuracy": 0.8285994701859012,
-      "eval_mean_iou": 0.7875396427764898,
-      "eval_overall_accuracy": 0.981291366835772,
-      "eval_runtime": 5.8157,
-      "eval_samples_per_second": 20.29,
-      "eval_steps_per_second": 1.376,
-      "step": 5460
-    },
-    {
-      "epoch": 321.24,
-      "learning_rate": 1.881733746130031e-05,
-      "loss": 0.0211,
-      "step": 5461
-    },
-    {
-      "epoch": 321.29,
-      "learning_rate": 1.8811145510835914e-05,
-      "loss": 0.0164,
-      "step": 5462
-    },
-    {
-      "epoch": 321.35,
-      "learning_rate": 1.8804953560371518e-05,
-      "loss": 0.0202,
-      "step": 5463
-    },
-    {
-      "epoch": 321.41,
-      "learning_rate": 1.8798761609907124e-05,
-      "loss": 0.0192,
-      "step": 5464
-    },
-    {
-      "epoch": 321.47,
-      "learning_rate": 1.8792569659442724e-05,
-      "loss": 0.0208,
-      "step": 5465
-    },
-    {
-      "epoch": 321.53,
-      "learning_rate": 1.8786377708978328e-05,
-      "loss": 0.0215,
-      "step": 5466
-    },
-    {
-      "epoch": 321.59,
-      "learning_rate": 1.878018575851393e-05,
-      "loss": 0.0195,
-      "step": 5467
-    },
-    {
-      "epoch": 321.65,
-      "learning_rate": 1.8773993808049538e-05,
-      "loss": 0.0202,
-      "step": 5468
-    },
-    {
-      "epoch": 321.71,
-      "learning_rate": 1.876780185758514e-05,
-      "loss": 0.0171,
-      "step": 5469
-    },
-    {
-      "epoch": 321.76,
-      "learning_rate": 1.8761609907120744e-05,
-      "loss": 0.0206,
-      "step": 5470
-    },
-    {
-      "epoch": 321.82,
-      "learning_rate": 1.8755417956656348e-05,
-      "loss": 0.0171,
-      "step": 5471
-    },
-    {
-      "epoch": 321.88,
-      "learning_rate": 1.874922600619195e-05,
-      "loss": 0.0186,
-      "step": 5472
-    },
-    {
-      "epoch": 321.94,
-      "learning_rate": 1.8743034055727554e-05,
-      "loss": 0.0189,
-      "step": 5473
-    },
-    {
-      "epoch": 322.0,
-      "learning_rate": 1.8736842105263158e-05,
-      "loss": 0.0194,
-      "step": 5474
-    },
-    {
-      "epoch": 322.06,
-      "learning_rate": 1.873065015479876e-05,
-      "loss": 0.0185,
-      "step": 5475
-    },
-    {
-      "epoch": 322.12,
-      "learning_rate": 1.8724458204334368e-05,
-      "loss": 0.0159,
-      "step": 5476
-    },
-    {
-      "epoch": 322.18,
-      "learning_rate": 1.871826625386997e-05,
-      "loss": 0.0193,
-      "step": 5477
-    },
-    {
-      "epoch": 322.24,
-      "learning_rate": 1.8712074303405574e-05,
-      "loss": 0.0205,
-      "step": 5478
-    },
-    {
-      "epoch": 322.29,
-      "learning_rate": 1.8705882352941178e-05,
-      "loss": 0.0201,
-      "step": 5479
-    },
-    {
-      "epoch": 322.35,
-      "learning_rate": 1.869969040247678e-05,
-      "loss": 0.0224,
-      "step": 5480
-    },
-    {
-      "epoch": 322.35,
-      "eval_accuracy_NAT": 0.9939929361578567,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9584564652068692,
-      "eval_accuracy_concrete tiles": 0.944049021918454,
-      "eval_accuracy_dark bricks": 0.8513772180004977,
-      "eval_accuracy_grass": 0.9074182364585344,
-      "eval_accuracy_red bricks": 0.9365663444639719,
-      "eval_accuracy_road": 0.9839835485338027,
-      "eval_accuracy_rough concrete": 0.7408279157902263,
-      "eval_accuracy_rough red sidewalk": 0.6870934508456222,
-      "eval_accuracy_speedway bricks": 0.9907181776617978,
-      "eval_accuracy_steel": 0.9263695292547742,
-      "eval_accuracy_tiles": 0.22021266927201671,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843188851927246,
-      "eval_iou_REST": 0.503793626707132,
-      "eval_iou_concrete": 0.9055249610291413,
-      "eval_iou_concrete tiles": 0.8846680507089536,
-      "eval_iou_dark bricks": 0.8349163741482553,
-      "eval_iou_grass": 0.8500803555678474,
-      "eval_iou_red bricks": 0.911632631241313,
-      "eval_iou_road": 0.9601751982557354,
-      "eval_iou_rough concrete": 0.7348117238892862,
-      "eval_iou_rough red sidewalk": 0.558009295167368,
-      "eval_iou_speedway bricks": 0.9803019823494611,
-      "eval_iou_steel": 0.8769922286227984,
-      "eval_iou_tiles": 0.21907775768535262,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07072684168815613,
-      "eval_mean_accuracy": 0.8216079163029544,
-      "eval_mean_iou": 0.7849463900434899,
-      "eval_overall_accuracy": 0.9813378544177039,
-      "eval_runtime": 5.8121,
-      "eval_samples_per_second": 20.302,
-      "eval_steps_per_second": 1.376,
-      "step": 5480
-    },
-    {
-      "epoch": 322.41,
-      "learning_rate": 1.8693498452012384e-05,
-      "loss": 0.019,
-      "step": 5481
-    },
-    {
-      "epoch": 322.47,
-      "learning_rate": 1.8687306501547988e-05,
-      "loss": 0.0179,
-      "step": 5482
-    },
-    {
-      "epoch": 322.53,
-      "learning_rate": 1.868111455108359e-05,
-      "loss": 0.0176,
-      "step": 5483
-    },
-    {
-      "epoch": 322.59,
-      "learning_rate": 1.8674922600619198e-05,
-      "loss": 0.0169,
-      "step": 5484
-    },
-    {
-      "epoch": 322.65,
-      "learning_rate": 1.86687306501548e-05,
-      "loss": 0.0194,
-      "step": 5485
-    },
-    {
-      "epoch": 322.71,
-      "learning_rate": 1.8662538699690404e-05,
-      "loss": 0.0171,
-      "step": 5486
-    },
-    {
-      "epoch": 322.76,
-      "learning_rate": 1.8656346749226004e-05,
-      "loss": 0.0185,
-      "step": 5487
-    },
-    {
-      "epoch": 322.82,
-      "learning_rate": 1.865015479876161e-05,
-      "loss": 0.0186,
-      "step": 5488
-    },
-    {
-      "epoch": 322.88,
-      "learning_rate": 1.8643962848297214e-05,
-      "loss": 0.0197,
-      "step": 5489
-    },
-    {
-      "epoch": 322.94,
-      "learning_rate": 1.8637770897832817e-05,
-      "loss": 0.0188,
-      "step": 5490
-    },
-    {
-      "epoch": 323.0,
-      "learning_rate": 1.863157894736842e-05,
-      "loss": 0.0174,
-      "step": 5491
-    },
-    {
-      "epoch": 323.06,
-      "learning_rate": 1.8625386996904028e-05,
-      "loss": 0.019,
-      "step": 5492
-    },
-    {
-      "epoch": 323.12,
-      "learning_rate": 1.861919504643963e-05,
-      "loss": 0.0196,
-      "step": 5493
-    },
-    {
-      "epoch": 323.18,
-      "learning_rate": 1.8613003095975234e-05,
-      "loss": 0.0178,
-      "step": 5494
-    },
-    {
-      "epoch": 323.24,
-      "learning_rate": 1.8606811145510834e-05,
-      "loss": 0.0162,
-      "step": 5495
-    },
-    {
-      "epoch": 323.29,
-      "learning_rate": 1.860061919504644e-05,
-      "loss": 0.019,
-      "step": 5496
-    },
-    {
-      "epoch": 323.35,
-      "learning_rate": 1.8594427244582044e-05,
-      "loss": 0.0184,
-      "step": 5497
-    },
-    {
-      "epoch": 323.41,
-      "learning_rate": 1.8588235294117647e-05,
-      "loss": 0.0199,
-      "step": 5498
-    },
-    {
-      "epoch": 323.47,
-      "learning_rate": 1.858204334365325e-05,
-      "loss": 0.0187,
-      "step": 5499
-    },
-    {
-      "epoch": 323.53,
-      "learning_rate": 1.8575851393188857e-05,
-      "loss": 0.0145,
-      "step": 5500
-    },
-    {
-      "epoch": 323.53,
-      "eval_accuracy_NAT": 0.9938795597285414,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9629461603494219,
-      "eval_accuracy_concrete tiles": 0.93019090266321,
-      "eval_accuracy_dark bricks": 0.855995061539345,
-      "eval_accuracy_grass": 0.8978925615142672,
-      "eval_accuracy_red bricks": 0.9306898066783831,
-      "eval_accuracy_road": 0.9834031726148073,
-      "eval_accuracy_rough concrete": 0.6621240741513178,
-      "eval_accuracy_rough red sidewalk": 0.8036757683544018,
-      "eval_accuracy_speedway bricks": 0.9902049390179954,
-      "eval_accuracy_steel": 0.9208644239755707,
-      "eval_accuracy_tiles": 0.18740343542670182,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844073708396988,
-      "eval_iou_REST": 0.5,
-      "eval_iou_concrete": 0.9002804301358056,
-      "eval_iou_concrete tiles": 0.8796469644289917,
-      "eval_iou_dark bricks": 0.8378642959521111,
-      "eval_iou_grass": 0.8474108869113267,
-      "eval_iou_red bricks": 0.9012391639631974,
-      "eval_iou_road": 0.9711983220704317,
-      "eval_iou_rough concrete": 0.6562428022182456,
-      "eval_iou_rough red sidewalk": 0.597267164053986,
-      "eval_iou_speedway bricks": 0.9802425833349026,
-      "eval_iou_steel": 0.8752758202162858,
-      "eval_iou_tiles": 0.1862075193401764,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07391217350959778,
-      "eval_mean_accuracy": 0.819931328029842,
-      "eval_mean_iou": 0.7782525633434738,
-      "eval_overall_accuracy": 0.9811689409158998,
-      "eval_runtime": 5.7174,
-      "eval_samples_per_second": 20.639,
-      "eval_steps_per_second": 1.399,
-      "step": 5500
-    },
-    {
-      "epoch": 323.59,
-      "learning_rate": 1.856965944272446e-05,
-      "loss": 0.0214,
-      "step": 5501
-    },
-    {
-      "epoch": 323.65,
-      "learning_rate": 1.856346749226006e-05,
-      "loss": 0.017,
-      "step": 5502
-    },
-    {
-      "epoch": 323.71,
-      "learning_rate": 1.8557275541795664e-05,
-      "loss": 0.0177,
-      "step": 5503
-    },
-    {
-      "epoch": 323.76,
-      "learning_rate": 1.855108359133127e-05,
-      "loss": 0.017,
-      "step": 5504
-    },
-    {
-      "epoch": 323.82,
-      "learning_rate": 1.8544891640866874e-05,
-      "loss": 0.0205,
-      "step": 5505
-    },
-    {
-      "epoch": 323.88,
-      "learning_rate": 1.8538699690402477e-05,
-      "loss": 0.0158,
-      "step": 5506
-    },
-    {
-      "epoch": 323.94,
-      "learning_rate": 1.853250773993808e-05,
-      "loss": 0.0179,
-      "step": 5507
-    },
-    {
-      "epoch": 324.0,
-      "learning_rate": 1.8526315789473687e-05,
-      "loss": 0.0199,
-      "step": 5508
-    },
-    {
-      "epoch": 324.06,
-      "learning_rate": 1.852012383900929e-05,
-      "loss": 0.0171,
-      "step": 5509
-    },
-    {
-      "epoch": 324.12,
-      "learning_rate": 1.851393188854489e-05,
-      "loss": 0.0181,
-      "step": 5510
-    },
-    {
-      "epoch": 324.18,
-      "learning_rate": 1.8507739938080494e-05,
-      "loss": 0.0206,
-      "step": 5511
-    },
-    {
-      "epoch": 324.24,
-      "learning_rate": 1.85015479876161e-05,
-      "loss": 0.0169,
-      "step": 5512
-    },
-    {
-      "epoch": 324.29,
-      "learning_rate": 1.8495356037151704e-05,
-      "loss": 0.0182,
-      "step": 5513
-    },
-    {
-      "epoch": 324.35,
-      "learning_rate": 1.8489164086687307e-05,
-      "loss": 0.0196,
-      "step": 5514
-    },
-    {
-      "epoch": 324.41,
-      "learning_rate": 1.848297213622291e-05,
-      "loss": 0.0185,
-      "step": 5515
-    },
-    {
-      "epoch": 324.47,
-      "learning_rate": 1.8476780185758517e-05,
-      "loss": 0.0178,
-      "step": 5516
-    },
-    {
-      "epoch": 324.53,
-      "learning_rate": 1.8470588235294117e-05,
-      "loss": 0.016,
-      "step": 5517
-    },
-    {
-      "epoch": 324.59,
-      "learning_rate": 1.846439628482972e-05,
-      "loss": 0.0176,
-      "step": 5518
-    },
-    {
-      "epoch": 324.65,
-      "learning_rate": 1.8458204334365324e-05,
-      "loss": 0.0171,
-      "step": 5519
-    },
-    {
-      "epoch": 324.71,
-      "learning_rate": 1.845201238390093e-05,
-      "loss": 0.0167,
-      "step": 5520
-    },
-    {
-      "epoch": 324.71,
-      "eval_accuracy_NAT": 0.9939524484267983,
-      "eval_accuracy_REST": 0.5056910569105691,
-      "eval_accuracy_concrete": 0.9594769373300168,
-      "eval_accuracy_concrete tiles": 0.9504124440254537,
-      "eval_accuracy_dark bricks": 0.8599908121662232,
-      "eval_accuracy_grass": 0.905172837964172,
-      "eval_accuracy_red bricks": 0.9443101933216169,
-      "eval_accuracy_road": 0.9835335701509351,
-      "eval_accuracy_rough concrete": 0.6530071088494152,
-      "eval_accuracy_rough red sidewalk": 0.7705351972867897,
-      "eval_accuracy_speedway bricks": 0.9896121869415638,
-      "eval_accuracy_steel": 0.928398646588734,
-      "eval_accuracy_tiles": 0.24072222727135023,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841853179343368,
-      "eval_iou_REST": 0.4726443768996961,
-      "eval_iou_concrete": 0.9006342873953318,
-      "eval_iou_concrete tiles": 0.8859157714361036,
-      "eval_iou_dark bricks": 0.840547037969,
-      "eval_iou_grass": 0.8473368923516537,
-      "eval_iou_red bricks": 0.9117132403626915,
-      "eval_iou_road": 0.9678621971402562,
-      "eval_iou_rough concrete": 0.6480296615180832,
-      "eval_iou_rough red sidewalk": 0.5969310178091848,
-      "eval_iou_speedway bricks": 0.9803517935481768,
-      "eval_iou_steel": 0.8766278046374882,
-      "eval_iou_tiles": 0.23929410347527555,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07512204349040985,
-      "eval_mean_accuracy": 0.8219088974795105,
-      "eval_mean_iou": 0.7809287309597907,
-      "eval_overall_accuracy": 0.9810885089938923,
-      "eval_runtime": 7.7714,
-      "eval_samples_per_second": 15.184,
-      "eval_steps_per_second": 1.029,
-      "step": 5520
-    },
-    {
-      "epoch": 324.76,
-      "learning_rate": 1.8445820433436534e-05,
-      "loss": 0.0176,
-      "step": 5521
-    },
-    {
-      "epoch": 324.82,
-      "learning_rate": 1.8439628482972137e-05,
-      "loss": 0.0183,
-      "step": 5522
-    },
-    {
-      "epoch": 324.88,
-      "learning_rate": 1.843343653250774e-05,
-      "loss": 0.0192,
-      "step": 5523
-    },
-    {
-      "epoch": 324.94,
-      "learning_rate": 1.8427244582043347e-05,
-      "loss": 0.0161,
-      "step": 5524
-    },
-    {
-      "epoch": 325.0,
-      "learning_rate": 1.8421052631578947e-05,
-      "loss": 0.0195,
-      "step": 5525
-    },
-    {
-      "epoch": 325.06,
-      "learning_rate": 1.841486068111455e-05,
-      "loss": 0.0185,
-      "step": 5526
-    },
-    {
-      "epoch": 325.12,
-      "learning_rate": 1.8408668730650154e-05,
-      "loss": 0.019,
-      "step": 5527
-    },
-    {
-      "epoch": 325.18,
-      "learning_rate": 1.840247678018576e-05,
-      "loss": 0.0189,
-      "step": 5528
-    },
-    {
-      "epoch": 325.24,
-      "learning_rate": 1.8396284829721364e-05,
-      "loss": 0.0169,
-      "step": 5529
-    },
-    {
-      "epoch": 325.29,
-      "learning_rate": 1.8390092879256967e-05,
-      "loss": 0.0174,
-      "step": 5530
-    },
-    {
-      "epoch": 325.35,
-      "learning_rate": 1.838390092879257e-05,
-      "loss": 0.0183,
-      "step": 5531
-    },
-    {
-      "epoch": 325.41,
-      "learning_rate": 1.8377708978328174e-05,
-      "loss": 0.0188,
-      "step": 5532
-    },
-    {
-      "epoch": 325.47,
-      "learning_rate": 1.8371517027863777e-05,
-      "loss": 0.0179,
-      "step": 5533
-    },
-    {
-      "epoch": 325.53,
-      "learning_rate": 1.836532507739938e-05,
-      "loss": 0.0175,
-      "step": 5534
-    },
-    {
-      "epoch": 325.59,
-      "learning_rate": 1.8359133126934984e-05,
-      "loss": 0.0169,
-      "step": 5535
-    },
-    {
-      "epoch": 325.65,
-      "learning_rate": 1.835294117647059e-05,
-      "loss": 0.0151,
-      "step": 5536
-    },
-    {
-      "epoch": 325.71,
-      "learning_rate": 1.8346749226006194e-05,
-      "loss": 0.0173,
-      "step": 5537
-    },
-    {
-      "epoch": 325.76,
-      "learning_rate": 1.8340557275541797e-05,
-      "loss": 0.0184,
-      "step": 5538
-    },
-    {
-      "epoch": 325.82,
-      "learning_rate": 1.8334365325077397e-05,
-      "loss": 0.0185,
-      "step": 5539
-    },
-    {
-      "epoch": 325.88,
-      "learning_rate": 1.8328173374613004e-05,
-      "loss": 0.0167,
-      "step": 5540
-    },
-    {
-      "epoch": 325.88,
-      "eval_accuracy_NAT": 0.9942657296587294,
-      "eval_accuracy_REST": 0.4813008130081301,
-      "eval_accuracy_concrete": 0.9591276071725049,
-      "eval_accuracy_concrete tiles": 0.9331369314164506,
-      "eval_accuracy_dark bricks": 0.8523294986888196,
-      "eval_accuracy_grass": 0.9005429454934957,
-      "eval_accuracy_red bricks": 0.9342596660808435,
-      "eval_accuracy_road": 0.982907234444616,
-      "eval_accuracy_rough concrete": 0.6656755194584641,
-      "eval_accuracy_rough red sidewalk": 0.7735732709675964,
-      "eval_accuracy_speedway bricks": 0.9897376262361426,
-      "eval_accuracy_steel": 0.924050538015963,
-      "eval_accuracy_tiles": 0.23205792359659486,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841065621476013,
-      "eval_iou_REST": 0.46394984326018807,
-      "eval_iou_concrete": 0.9011039124022905,
-      "eval_iou_concrete tiles": 0.8843842837677858,
-      "eval_iou_dark bricks": 0.8346595062700332,
-      "eval_iou_grass": 0.8447484977321748,
-      "eval_iou_red bricks": 0.9052738012878506,
-      "eval_iou_road": 0.9666510570151718,
-      "eval_iou_rough concrete": 0.6592760020773335,
-      "eval_iou_rough red sidewalk": 0.5966567870586802,
-      "eval_iou_speedway bricks": 0.9802464157295476,
-      "eval_iou_steel": 0.8778142299621129,
-      "eval_iou_tiles": 0.2305631640730819,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07574331015348434,
-      "eval_mean_accuracy": 0.8171511772491036,
-      "eval_mean_iou": 0.7791872355987579,
-      "eval_overall_accuracy": 0.9810088529425153,
-      "eval_runtime": 5.6996,
-      "eval_samples_per_second": 20.703,
-      "eval_steps_per_second": 1.404,
-      "step": 5540
-    },
-    {
-      "epoch": 325.94,
-      "learning_rate": 1.8321981424148607e-05,
-      "loss": 0.0163,
-      "step": 5541
-    },
-    {
-      "epoch": 326.0,
-      "learning_rate": 1.831578947368421e-05,
-      "loss": 0.0196,
-      "step": 5542
-    },
-    {
-      "epoch": 326.06,
-      "learning_rate": 1.8309597523219814e-05,
-      "loss": 0.0194,
-      "step": 5543
-    },
-    {
-      "epoch": 326.12,
-      "learning_rate": 1.830340557275542e-05,
-      "loss": 0.017,
-      "step": 5544
-    },
-    {
-      "epoch": 326.18,
-      "learning_rate": 1.8297213622291024e-05,
-      "loss": 0.0189,
-      "step": 5545
-    },
-    {
-      "epoch": 326.24,
-      "learning_rate": 1.8291021671826627e-05,
-      "loss": 0.0187,
-      "step": 5546
-    },
-    {
-      "epoch": 326.29,
-      "learning_rate": 1.8284829721362227e-05,
-      "loss": 0.0186,
-      "step": 5547
-    },
-    {
-      "epoch": 326.35,
-      "learning_rate": 1.8278637770897834e-05,
-      "loss": 0.0171,
-      "step": 5548
-    },
-    {
-      "epoch": 326.41,
-      "learning_rate": 1.8272445820433437e-05,
-      "loss": 0.0193,
-      "step": 5549
-    },
-    {
-      "epoch": 326.47,
-      "learning_rate": 1.826625386996904e-05,
-      "loss": 0.0187,
-      "step": 5550
-    },
-    {
-      "epoch": 326.53,
-      "learning_rate": 1.8260061919504644e-05,
-      "loss": 0.015,
-      "step": 5551
-    },
-    {
-      "epoch": 326.59,
-      "learning_rate": 1.825386996904025e-05,
-      "loss": 0.0178,
-      "step": 5552
-    },
-    {
-      "epoch": 326.65,
-      "learning_rate": 1.8247678018575854e-05,
-      "loss": 0.018,
-      "step": 5553
-    },
-    {
-      "epoch": 326.71,
-      "learning_rate": 1.8241486068111454e-05,
-      "loss": 0.0155,
-      "step": 5554
-    },
-    {
-      "epoch": 326.76,
-      "learning_rate": 1.8235294117647057e-05,
-      "loss": 0.0177,
-      "step": 5555
-    },
-    {
-      "epoch": 326.82,
-      "learning_rate": 1.8229102167182664e-05,
-      "loss": 0.0162,
-      "step": 5556
-    },
-    {
-      "epoch": 326.88,
-      "learning_rate": 1.8222910216718267e-05,
-      "loss": 0.018,
-      "step": 5557
-    },
-    {
-      "epoch": 326.94,
-      "learning_rate": 1.821671826625387e-05,
-      "loss": 0.0193,
-      "step": 5558
-    },
-    {
-      "epoch": 327.0,
-      "learning_rate": 1.8210526315789474e-05,
-      "loss": 0.0188,
-      "step": 5559
-    },
-    {
-      "epoch": 327.06,
-      "learning_rate": 1.820433436532508e-05,
-      "loss": 0.018,
-      "step": 5560
-    },
-    {
-      "epoch": 327.06,
-      "eval_accuracy_NAT": 0.9938241923519808,
-      "eval_accuracy_REST": 0.4764227642276423,
-      "eval_accuracy_concrete": 0.9587194183232458,
-      "eval_accuracy_concrete tiles": 0.9430355880273391,
-      "eval_accuracy_dark bricks": 0.8469603583255173,
-      "eval_accuracy_grass": 0.909821654990216,
-      "eval_accuracy_red bricks": 0.9276691564147628,
-      "eval_accuracy_road": 0.9847258525406944,
-      "eval_accuracy_rough concrete": 0.7120171136760438,
-      "eval_accuracy_rough red sidewalk": 0.7837778897419894,
-      "eval_accuracy_speedway bricks": 0.989622297485799,
-      "eval_accuracy_steel": 0.9323019486523363,
-      "eval_accuracy_tiles": 0.22469629494986215,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844606113220147,
-      "eval_iou_REST": 0.458528951486698,
-      "eval_iou_concrete": 0.9039477904265631,
-      "eval_iou_concrete tiles": 0.8877967605946305,
-      "eval_iou_dark bricks": 0.8317285325588936,
-      "eval_iou_grass": 0.8513809899492727,
-      "eval_iou_red bricks": 0.90374531835206,
-      "eval_iou_road": 0.9655141335775824,
-      "eval_iou_rough concrete": 0.7052178685883249,
-      "eval_iou_rough red sidewalk": 0.6138487905719384,
-      "eval_iou_speedway bricks": 0.9800145977201287,
-      "eval_iou_steel": 0.8768417217998284,
-      "eval_iou_tiles": 0.22368659147113817,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07328129559755325,
-      "eval_mean_accuracy": 0.8218149638236483,
-      "eval_mean_iou": 0.7835932814168518,
-      "eval_overall_accuracy": 0.9814823279946537,
-      "eval_runtime": 6.0618,
-      "eval_samples_per_second": 19.466,
-      "eval_steps_per_second": 1.32,
-      "step": 5560
-    },
-    {
-      "epoch": 327.12,
-      "learning_rate": 1.8198142414860684e-05,
-      "loss": 0.0187,
-      "step": 5561
-    },
-    {
-      "epoch": 327.18,
-      "learning_rate": 1.8191950464396284e-05,
-      "loss": 0.0174,
-      "step": 5562
-    },
-    {
-      "epoch": 327.24,
-      "learning_rate": 1.8185758513931887e-05,
-      "loss": 0.0171,
-      "step": 5563
-    },
-    {
-      "epoch": 327.29,
-      "learning_rate": 1.8179566563467494e-05,
-      "loss": 0.0206,
-      "step": 5564
-    },
-    {
-      "epoch": 327.35,
-      "learning_rate": 1.8173374613003097e-05,
-      "loss": 0.0172,
-      "step": 5565
-    },
-    {
-      "epoch": 327.41,
-      "learning_rate": 1.81671826625387e-05,
-      "loss": 0.0173,
-      "step": 5566
-    },
-    {
-      "epoch": 327.47,
-      "learning_rate": 1.8160990712074304e-05,
-      "loss": 0.017,
-      "step": 5567
-    },
-    {
-      "epoch": 327.53,
-      "learning_rate": 1.815479876160991e-05,
-      "loss": 0.0182,
-      "step": 5568
-    },
-    {
-      "epoch": 327.59,
-      "learning_rate": 1.814860681114551e-05,
-      "loss": 0.0164,
-      "step": 5569
-    },
-    {
-      "epoch": 327.65,
-      "learning_rate": 1.8142414860681114e-05,
-      "loss": 0.0166,
-      "step": 5570
-    },
-    {
-      "epoch": 327.71,
-      "learning_rate": 1.8136222910216717e-05,
-      "loss": 0.0181,
-      "step": 5571
-    },
-    {
-      "epoch": 327.76,
-      "learning_rate": 1.8130030959752324e-05,
-      "loss": 0.0147,
-      "step": 5572
-    },
-    {
-      "epoch": 327.82,
-      "learning_rate": 1.8123839009287927e-05,
-      "loss": 0.0189,
-      "step": 5573
-    },
-    {
-      "epoch": 327.88,
-      "learning_rate": 1.811764705882353e-05,
-      "loss": 0.0183,
-      "step": 5574
-    },
-    {
-      "epoch": 327.94,
-      "learning_rate": 1.8111455108359134e-05,
-      "loss": 0.0166,
-      "step": 5575
-    },
-    {
-      "epoch": 328.0,
-      "learning_rate": 1.810526315789474e-05,
-      "loss": 0.0177,
-      "step": 5576
-    },
-    {
-      "epoch": 328.06,
-      "learning_rate": 1.809907120743034e-05,
-      "loss": 0.0177,
-      "step": 5577
-    },
-    {
-      "epoch": 328.12,
-      "learning_rate": 1.8092879256965944e-05,
-      "loss": 0.0184,
-      "step": 5578
-    },
-    {
-      "epoch": 328.18,
-      "learning_rate": 1.8086687306501547e-05,
-      "loss": 0.0166,
-      "step": 5579
-    },
-    {
-      "epoch": 328.24,
-      "learning_rate": 1.8080495356037154e-05,
-      "loss": 0.0168,
-      "step": 5580
-    },
-    {
-      "epoch": 328.24,
-      "eval_accuracy_NAT": 0.9935682732316956,
-      "eval_accuracy_REST": 0.4943089430894309,
-      "eval_accuracy_concrete": 0.959296539261805,
-      "eval_accuracy_concrete tiles": 0.9365778930002356,
-      "eval_accuracy_dark bricks": 0.8504727906130964,
-      "eval_accuracy_grass": 0.9109997085618473,
-      "eval_accuracy_red bricks": 0.9332161687170475,
-      "eval_accuracy_road": 0.9825304710717876,
-      "eval_accuracy_rough concrete": 0.7000160887622975,
-      "eval_accuracy_rough red sidewalk": 0.7681363222764497,
-      "eval_accuracy_speedway bricks": 0.9904885483520498,
-      "eval_accuracy_steel": 0.9305727008751193,
-      "eval_accuracy_tiles": 0.2373898027810597,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844883819917025,
-      "eval_iou_REST": 0.4735202492211838,
-      "eval_iou_concrete": 0.9028498746590399,
-      "eval_iou_concrete tiles": 0.8832066497755257,
-      "eval_iou_dark bricks": 0.8345307188069345,
-      "eval_iou_grass": 0.8486809686970825,
-      "eval_iou_red bricks": 0.9034933801244217,
-      "eval_iou_road": 0.966393789889437,
-      "eval_iou_rough concrete": 0.6911979289244529,
-      "eval_iou_rough red sidewalk": 0.5945738284137674,
-      "eval_iou_speedway bricks": 0.9800825780634003,
-      "eval_iou_steel": 0.8781688181222203,
-      "eval_iou_tiles": 0.23621619992162302,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07447008043527603,
-      "eval_mean_accuracy": 0.8221210961995324,
-      "eval_mean_iou": 0.7828771820469839,
-      "eval_overall_accuracy": 0.9813258607508772,
-      "eval_runtime": 6.0694,
-      "eval_samples_per_second": 19.442,
-      "eval_steps_per_second": 1.318,
-      "step": 5580
-    },
-    {
-      "epoch": 328.29,
-      "learning_rate": 1.8074303405572757e-05,
-      "loss": 0.0139,
-      "step": 5581
-    },
-    {
-      "epoch": 328.35,
-      "learning_rate": 1.806811145510836e-05,
-      "loss": 0.0206,
-      "step": 5582
-    },
-    {
-      "epoch": 328.41,
-      "learning_rate": 1.8061919504643964e-05,
-      "loss": 0.0186,
-      "step": 5583
-    },
-    {
-      "epoch": 328.47,
-      "learning_rate": 1.8055727554179567e-05,
-      "loss": 0.0173,
-      "step": 5584
-    },
-    {
-      "epoch": 328.53,
-      "learning_rate": 1.804953560371517e-05,
-      "loss": 0.0164,
-      "step": 5585
-    },
-    {
-      "epoch": 328.59,
-      "learning_rate": 1.8043343653250774e-05,
-      "loss": 0.0182,
-      "step": 5586
-    },
-    {
-      "epoch": 328.65,
-      "learning_rate": 1.8037151702786377e-05,
-      "loss": 0.0177,
-      "step": 5587
-    },
-    {
-      "epoch": 328.71,
-      "learning_rate": 1.8030959752321984e-05,
-      "loss": 0.0155,
-      "step": 5588
-    },
-    {
-      "epoch": 328.76,
-      "learning_rate": 1.8024767801857587e-05,
-      "loss": 0.018,
-      "step": 5589
-    },
-    {
-      "epoch": 328.82,
-      "learning_rate": 1.801857585139319e-05,
-      "loss": 0.0211,
-      "step": 5590
-    },
-    {
-      "epoch": 328.88,
-      "learning_rate": 1.8012383900928794e-05,
-      "loss": 0.0171,
-      "step": 5591
-    },
-    {
-      "epoch": 328.94,
-      "learning_rate": 1.8006191950464397e-05,
-      "loss": 0.0166,
-      "step": 5592
-    },
-    {
-      "epoch": 329.0,
-      "learning_rate": 1.8e-05,
-      "loss": 0.0151,
-      "step": 5593
-    },
-    {
-      "epoch": 329.06,
-      "learning_rate": 1.7993808049535604e-05,
-      "loss": 0.0197,
-      "step": 5594
-    },
-    {
-      "epoch": 329.12,
-      "learning_rate": 1.7987616099071207e-05,
-      "loss": 0.0177,
-      "step": 5595
-    },
-    {
-      "epoch": 329.18,
-      "learning_rate": 1.7981424148606814e-05,
-      "loss": 0.0184,
-      "step": 5596
-    },
-    {
-      "epoch": 329.24,
-      "learning_rate": 1.7975232198142417e-05,
-      "loss": 0.0177,
-      "step": 5597
-    },
-    {
-      "epoch": 329.29,
-      "learning_rate": 1.796904024767802e-05,
-      "loss": 0.0187,
-      "step": 5598
-    },
-    {
-      "epoch": 329.35,
-      "learning_rate": 1.796284829721362e-05,
-      "loss": 0.016,
-      "step": 5599
-    },
-    {
-      "epoch": 329.41,
-      "learning_rate": 1.7956656346749227e-05,
-      "loss": 0.0175,
-      "step": 5600
-    },
-    {
-      "epoch": 329.41,
-      "eval_accuracy_NAT": 0.9937910366201553,
-      "eval_accuracy_REST": 0.5642276422764227,
-      "eval_accuracy_concrete": 0.961710892221327,
-      "eval_accuracy_concrete tiles": 0.9475842564223427,
-      "eval_accuracy_dark bricks": 0.8481566908485347,
-      "eval_accuracy_grass": 0.9038017539277915,
-      "eval_accuracy_red bricks": 0.9258567662565905,
-      "eval_accuracy_road": 0.9825048190974673,
-      "eval_accuracy_rough concrete": 0.6870259029072989,
-      "eval_accuracy_rough red sidewalk": 0.7651132885643598,
-      "eval_accuracy_speedway bricks": 0.9894337958136178,
-      "eval_accuracy_steel": 0.9297280682902938,
-      "eval_accuracy_tiles": 0.23166409161137871,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844833925045433,
-      "eval_iou_REST": 0.5133136094674556,
-      "eval_iou_concrete": 0.9004962963864269,
-      "eval_iou_concrete tiles": 0.8829885360390038,
-      "eval_iou_dark bricks": 0.8328485571840066,
-      "eval_iou_grass": 0.847592493830491,
-      "eval_iou_red bricks": 0.9002937249666222,
-      "eval_iou_road": 0.9647159450177231,
-      "eval_iou_rough concrete": 0.6804733378581757,
-      "eval_iou_rough red sidewalk": 0.5972469255378474,
-      "eval_iou_speedway bricks": 0.9799694493240663,
-      "eval_iou_steel": 0.8784696971127964,
-      "eval_iou_tiles": 0.23037989937637454,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07566554099321365,
-      "eval_mean_accuracy": 0.8254306926813523,
-      "eval_mean_iou": 0.7840978357388871,
-      "eval_overall_accuracy": 0.9811132075422901,
-      "eval_runtime": 5.76,
-      "eval_samples_per_second": 20.486,
-      "eval_steps_per_second": 1.389,
-      "step": 5600
-    },
-    {
-      "epoch": 329.47,
-      "learning_rate": 1.795046439628483e-05,
-      "loss": 0.0183,
-      "step": 5601
-    },
-    {
-      "epoch": 329.53,
-      "learning_rate": 1.7944272445820434e-05,
-      "loss": 0.0175,
-      "step": 5602
-    },
-    {
-      "epoch": 329.59,
-      "learning_rate": 1.7938080495356037e-05,
-      "loss": 0.0187,
-      "step": 5603
-    },
-    {
-      "epoch": 329.65,
-      "learning_rate": 1.7931888544891644e-05,
-      "loss": 0.0169,
-      "step": 5604
-    },
-    {
-      "epoch": 329.71,
-      "learning_rate": 1.7925696594427247e-05,
-      "loss": 0.0162,
-      "step": 5605
-    },
-    {
-      "epoch": 329.76,
-      "learning_rate": 1.7919504643962847e-05,
-      "loss": 0.016,
-      "step": 5606
-    },
-    {
-      "epoch": 329.82,
-      "learning_rate": 1.791331269349845e-05,
-      "loss": 0.0167,
-      "step": 5607
-    },
-    {
-      "epoch": 329.88,
-      "learning_rate": 1.7907120743034057e-05,
-      "loss": 0.0171,
-      "step": 5608
-    },
-    {
-      "epoch": 329.94,
-      "learning_rate": 1.790092879256966e-05,
-      "loss": 0.0169,
-      "step": 5609
-    },
-    {
-      "epoch": 330.0,
-      "learning_rate": 1.7894736842105264e-05,
-      "loss": 0.0168,
-      "step": 5610
-    },
-    {
-      "epoch": 330.06,
-      "learning_rate": 1.7888544891640867e-05,
-      "loss": 0.0191,
-      "step": 5611
-    },
-    {
-      "epoch": 330.12,
-      "learning_rate": 1.7882352941176474e-05,
-      "loss": 0.0181,
-      "step": 5612
-    },
-    {
-      "epoch": 330.18,
-      "learning_rate": 1.7876160990712077e-05,
-      "loss": 0.0198,
-      "step": 5613
-    },
-    {
-      "epoch": 330.24,
-      "learning_rate": 1.7869969040247677e-05,
-      "loss": 0.02,
-      "step": 5614
-    },
-    {
-      "epoch": 330.29,
-      "learning_rate": 1.786377708978328e-05,
-      "loss": 0.0151,
-      "step": 5615
-    },
-    {
-      "epoch": 330.35,
-      "learning_rate": 1.7857585139318887e-05,
-      "loss": 0.0168,
-      "step": 5616
-    },
-    {
-      "epoch": 330.41,
-      "learning_rate": 1.785139318885449e-05,
-      "loss": 0.0193,
-      "step": 5617
-    },
-    {
-      "epoch": 330.47,
-      "learning_rate": 1.7845201238390094e-05,
-      "loss": 0.0147,
-      "step": 5618
-    },
-    {
-      "epoch": 330.53,
-      "learning_rate": 1.7839009287925697e-05,
-      "loss": 0.016,
-      "step": 5619
-    },
-    {
-      "epoch": 330.59,
-      "learning_rate": 1.7832817337461304e-05,
-      "loss": 0.0164,
-      "step": 5620
-    },
-    {
-      "epoch": 330.59,
-      "eval_accuracy_NAT": 0.993953796220775,
-      "eval_accuracy_REST": 0.5544715447154471,
-      "eval_accuracy_concrete": 0.9603196867800321,
-      "eval_accuracy_concrete tiles": 0.9316757011548432,
-      "eval_accuracy_dark bricks": 0.8452759221331088,
-      "eval_accuracy_grass": 0.9011296067098904,
-      "eval_accuracy_red bricks": 0.9303602811950791,
-      "eval_accuracy_road": 0.981449347237416,
-      "eval_accuracy_rough concrete": 0.6607118383496505,
-      "eval_accuracy_rough red sidewalk": 0.7521337955617052,
-      "eval_accuracy_speedway bricks": 0.989459157856784,
-      "eval_accuracy_steel": 0.9350932344430173,
-      "eval_accuracy_tiles": 0.2425096185888697,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9842583193217644,
-      "eval_iou_REST": 0.4963609898107715,
-      "eval_iou_concrete": 0.8983037940470479,
-      "eval_iou_concrete tiles": 0.8818372445792808,
-      "eval_iou_dark bricks": 0.829906691348512,
-      "eval_iou_grass": 0.8475413585559192,
-      "eval_iou_red bricks": 0.9038040868590941,
-      "eval_iou_road": 0.963252471742152,
-      "eval_iou_rough concrete": 0.6549204680365971,
-      "eval_iou_rough red sidewalk": 0.5923482380811371,
-      "eval_iou_speedway bricks": 0.9797820444422725,
-      "eval_iou_steel": 0.8764964502127999,
-      "eval_iou_tiles": 0.24104911319220693,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07872229069471359,
-      "eval_mean_accuracy": 0.8214264254574322,
-      "eval_mean_iou": 0.7807585592484272,
-      "eval_overall_accuracy": 0.9807863720392777,
-      "eval_runtime": 5.7997,
-      "eval_samples_per_second": 20.346,
-      "eval_steps_per_second": 1.379,
-      "step": 5620
-    },
-    {
-      "epoch": 330.65,
-      "learning_rate": 1.7826625386996904e-05,
-      "loss": 0.016,
-      "step": 5621
-    },
-    {
-      "epoch": 330.71,
-      "learning_rate": 1.7820433436532507e-05,
-      "loss": 0.0165,
-      "step": 5622
-    },
-    {
-      "epoch": 330.76,
-      "learning_rate": 1.781424148606811e-05,
-      "loss": 0.0176,
-      "step": 5623
-    },
-    {
-      "epoch": 330.82,
-      "learning_rate": 1.7808049535603717e-05,
-      "loss": 0.023,
-      "step": 5624
-    },
-    {
-      "epoch": 330.88,
-      "learning_rate": 1.780185758513932e-05,
-      "loss": 0.019,
-      "step": 5625
-    },
-    {
-      "epoch": 330.94,
-      "learning_rate": 1.7795665634674924e-05,
-      "loss": 0.0188,
-      "step": 5626
-    },
-    {
-      "epoch": 331.0,
-      "learning_rate": 1.7789473684210527e-05,
-      "loss": 0.0144,
-      "step": 5627
-    },
-    {
-      "epoch": 331.06,
-      "learning_rate": 1.7783281733746134e-05,
-      "loss": 0.0187,
-      "step": 5628
-    },
-    {
-      "epoch": 331.12,
-      "learning_rate": 1.7777089783281734e-05,
-      "loss": 0.0169,
-      "step": 5629
-    },
-    {
-      "epoch": 331.18,
-      "learning_rate": 1.7770897832817337e-05,
-      "loss": 0.0171,
-      "step": 5630
-    },
-    {
-      "epoch": 331.24,
-      "learning_rate": 1.776470588235294e-05,
-      "loss": 0.0195,
-      "step": 5631
-    },
-    {
-      "epoch": 331.29,
-      "learning_rate": 1.7758513931888547e-05,
-      "loss": 0.0188,
-      "step": 5632
-    },
-    {
-      "epoch": 331.35,
-      "learning_rate": 1.775232198142415e-05,
-      "loss": 0.0149,
-      "step": 5633
-    },
-    {
-      "epoch": 331.41,
-      "learning_rate": 1.7746130030959754e-05,
-      "loss": 0.0171,
-      "step": 5634
-    },
-    {
-      "epoch": 331.47,
-      "learning_rate": 1.7739938080495357e-05,
-      "loss": 0.0176,
-      "step": 5635
-    },
-    {
-      "epoch": 331.53,
-      "learning_rate": 1.773374613003096e-05,
-      "loss": 0.0185,
-      "step": 5636
-    },
-    {
-      "epoch": 331.59,
-      "learning_rate": 1.7727554179566564e-05,
-      "loss": 0.0164,
-      "step": 5637
-    },
-    {
-      "epoch": 331.65,
-      "learning_rate": 1.7721362229102167e-05,
-      "loss": 0.0172,
-      "step": 5638
-    },
-    {
-      "epoch": 331.71,
-      "learning_rate": 1.771517027863777e-05,
-      "loss": 0.0158,
-      "step": 5639
-    },
-    {
-      "epoch": 331.76,
-      "learning_rate": 1.7708978328173377e-05,
-      "loss": 0.0179,
-      "step": 5640
-    },
-    {
-      "epoch": 331.76,
-      "eval_accuracy_NAT": 0.9938037597952949,
-      "eval_accuracy_REST": 0.5219512195121951,
-      "eval_accuracy_concrete": 0.9612010383590502,
-      "eval_accuracy_concrete tiles": 0.9530285175583314,
-      "eval_accuracy_dark bricks": 0.84879313975078,
-      "eval_accuracy_grass": 0.902578281423278,
-      "eval_accuracy_red bricks": 0.9211884885764499,
-      "eval_accuracy_road": 0.9824588593101435,
-      "eval_accuracy_rough concrete": 0.6635899391606433,
-      "eval_accuracy_rough red sidewalk": 0.7722723136735875,
-      "eval_accuracy_speedway bricks": 0.9894990859382551,
-      "eval_accuracy_steel": 0.9321894975685576,
-      "eval_accuracy_tiles": 0.25284013450877035,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844210248396555,
-      "eval_iou_REST": 0.4727540500736377,
-      "eval_iou_concrete": 0.8994581622580506,
-      "eval_iou_concrete tiles": 0.8859020703253369,
-      "eval_iou_dark bricks": 0.8323275740711196,
-      "eval_iou_grass": 0.8489106831369477,
-      "eval_iou_red bricks": 0.8999839029886785,
-      "eval_iou_road": 0.9636953419785975,
-      "eval_iou_rough concrete": 0.6562035967662102,
-      "eval_iou_rough red sidewalk": 0.611802833347234,
-      "eval_iou_speedway bricks": 0.9797477658710202,
-      "eval_iou_steel": 0.8778872606353562,
-      "eval_iou_tiles": 0.25071344888701974,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07775285094976425,
-      "eval_mean_accuracy": 0.8227226365488721,
-      "eval_mean_iou": 0.7818313627060665,
-      "eval_overall_accuracy": 0.981011697801493,
-      "eval_runtime": 5.7394,
-      "eval_samples_per_second": 20.56,
-      "eval_steps_per_second": 1.394,
-      "step": 5640
-    },
-    {
-      "epoch": 331.82,
-      "learning_rate": 1.770278637770898e-05,
-      "loss": 0.0175,
-      "step": 5641
-    },
-    {
-      "epoch": 331.88,
-      "learning_rate": 1.7696594427244584e-05,
-      "loss": 0.0194,
-      "step": 5642
-    },
-    {
-      "epoch": 331.94,
-      "learning_rate": 1.7690402476780187e-05,
-      "loss": 0.0148,
-      "step": 5643
-    },
-    {
-      "epoch": 332.0,
-      "learning_rate": 1.768421052631579e-05,
-      "loss": 0.0165,
-      "step": 5644
-    },
-    {
-      "epoch": 332.06,
-      "learning_rate": 1.7678018575851394e-05,
-      "loss": 0.0189,
-      "step": 5645
-    },
-    {
-      "epoch": 332.12,
-      "learning_rate": 1.7671826625386997e-05,
-      "loss": 0.0184,
-      "step": 5646
-    },
-    {
-      "epoch": 332.18,
-      "learning_rate": 1.76656346749226e-05,
-      "loss": 0.0176,
-      "step": 5647
-    },
-    {
-      "epoch": 332.24,
-      "learning_rate": 1.7659442724458207e-05,
-      "loss": 0.0178,
-      "step": 5648
-    },
-    {
-      "epoch": 332.29,
-      "learning_rate": 1.765325077399381e-05,
-      "loss": 0.0178,
-      "step": 5649
-    },
-    {
-      "epoch": 332.35,
-      "learning_rate": 1.7647058823529414e-05,
-      "loss": 0.015,
-      "step": 5650
-    },
-    {
-      "epoch": 332.41,
-      "learning_rate": 1.7640866873065014e-05,
-      "loss": 0.0175,
-      "step": 5651
-    },
-    {
-      "epoch": 332.47,
-      "learning_rate": 1.763467492260062e-05,
-      "loss": 0.0173,
-      "step": 5652
-    },
-    {
-      "epoch": 332.53,
-      "learning_rate": 1.7628482972136224e-05,
-      "loss": 0.0164,
-      "step": 5653
-    },
-    {
-      "epoch": 332.59,
-      "learning_rate": 1.7622291021671827e-05,
-      "loss": 0.0173,
-      "step": 5654
-    },
-    {
-      "epoch": 332.65,
-      "learning_rate": 1.761609907120743e-05,
-      "loss": 0.0177,
-      "step": 5655
-    },
-    {
-      "epoch": 332.71,
-      "learning_rate": 1.7609907120743037e-05,
-      "loss": 0.0186,
-      "step": 5656
-    },
-    {
-      "epoch": 332.76,
-      "learning_rate": 1.760371517027864e-05,
-      "loss": 0.0169,
-      "step": 5657
-    },
-    {
-      "epoch": 332.82,
-      "learning_rate": 1.7597523219814244e-05,
-      "loss": 0.0155,
-      "step": 5658
-    },
-    {
-      "epoch": 332.88,
-      "learning_rate": 1.7591331269349844e-05,
-      "loss": 0.0173,
-      "step": 5659
-    },
-    {
-      "epoch": 332.94,
-      "learning_rate": 1.758513931888545e-05,
-      "loss": 0.019,
-      "step": 5660
-    },
-    {
-      "epoch": 332.94,
-      "eval_accuracy_NAT": 0.9939676515428548,
-      "eval_accuracy_REST": 0.4666666666666667,
-      "eval_accuracy_concrete": 0.9582592503695867,
-      "eval_accuracy_concrete tiles": 0.9401602639641763,
-      "eval_accuracy_dark bricks": 0.839940279080451,
-      "eval_accuracy_grass": 0.8986145333015401,
-      "eval_accuracy_red bricks": 0.9242091388400703,
-      "eval_accuracy_road": 0.9823690774000228,
-      "eval_accuracy_rough concrete": 0.6476918584903973,
-      "eval_accuracy_rough red sidewalk": 0.7717835146902895,
-      "eval_accuracy_speedway bricks": 0.990631638257751,
-      "eval_accuracy_steel": 0.9297805454627237,
-      "eval_accuracy_tiles": 0.2279075403677785,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9841331268730128,
-      "eval_iou_REST": 0.44773790951638065,
-      "eval_iou_concrete": 0.8986720412513473,
-      "eval_iou_concrete tiles": 0.887018589344481,
-      "eval_iou_dark bricks": 0.8256805641143846,
-      "eval_iou_grass": 0.8443973201385276,
-      "eval_iou_red bricks": 0.8997005988023952,
-      "eval_iou_road": 0.9641217738653195,
-      "eval_iou_rough concrete": 0.6414084490419738,
-      "eval_iou_rough red sidewalk": 0.6015708801031623,
-      "eval_iou_speedway bricks": 0.9797387351170401,
-      "eval_iou_steel": 0.8781149771428707,
-      "eval_iou_tiles": 0.22667148754105276,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08040440827608109,
-      "eval_mean_accuracy": 0.8132293814180236,
-      "eval_mean_iou": 0.7753051117578421,
-      "eval_overall_accuracy": 0.980711047932253,
-      "eval_runtime": 5.7528,
-      "eval_samples_per_second": 20.512,
-      "eval_steps_per_second": 1.391,
-      "step": 5660
-    },
-    {
-      "epoch": 333.0,
-      "learning_rate": 1.7578947368421054e-05,
-      "loss": 0.0173,
-      "step": 5661
-    },
-    {
-      "epoch": 333.06,
-      "learning_rate": 1.7572755417956657e-05,
-      "loss": 0.0181,
-      "step": 5662
-    },
-    {
-      "epoch": 333.12,
-      "learning_rate": 1.756656346749226e-05,
-      "loss": 0.0181,
-      "step": 5663
-    },
-    {
-      "epoch": 333.18,
-      "learning_rate": 1.7560371517027867e-05,
-      "loss": 0.0187,
-      "step": 5664
-    },
-    {
-      "epoch": 333.24,
-      "learning_rate": 1.755417956656347e-05,
-      "loss": 0.0173,
-      "step": 5665
-    },
-    {
-      "epoch": 333.29,
-      "learning_rate": 1.754798761609907e-05,
-      "loss": 0.017,
-      "step": 5666
-    },
-    {
-      "epoch": 333.35,
-      "learning_rate": 1.7541795665634673e-05,
-      "loss": 0.0184,
-      "step": 5667
-    },
-    {
-      "epoch": 333.41,
-      "learning_rate": 1.753560371517028e-05,
-      "loss": 0.0162,
-      "step": 5668
-    },
-    {
-      "epoch": 333.47,
-      "learning_rate": 1.7529411764705884e-05,
-      "loss": 0.0187,
-      "step": 5669
-    },
-    {
-      "epoch": 333.53,
-      "learning_rate": 1.7523219814241487e-05,
-      "loss": 0.0155,
-      "step": 5670
-    },
-    {
-      "epoch": 333.59,
-      "learning_rate": 1.751702786377709e-05,
-      "loss": 0.0186,
-      "step": 5671
-    },
-    {
-      "epoch": 333.65,
-      "learning_rate": 1.7510835913312697e-05,
-      "loss": 0.0189,
-      "step": 5672
-    },
-    {
-      "epoch": 333.71,
-      "learning_rate": 1.75046439628483e-05,
-      "loss": 0.0161,
-      "step": 5673
-    },
-    {
-      "epoch": 333.76,
-      "learning_rate": 1.74984520123839e-05,
-      "loss": 0.018,
-      "step": 5674
-    },
-    {
-      "epoch": 333.82,
-      "learning_rate": 1.7492260061919503e-05,
-      "loss": 0.0185,
-      "step": 5675
-    },
-    {
-      "epoch": 333.88,
-      "learning_rate": 1.748606811145511e-05,
-      "loss": 0.0161,
-      "step": 5676
-    },
-    {
-      "epoch": 333.94,
-      "learning_rate": 1.7479876160990714e-05,
-      "loss": 0.0172,
-      "step": 5677
-    },
-    {
-      "epoch": 334.0,
-      "learning_rate": 1.7473684210526317e-05,
-      "loss": 0.0162,
-      "step": 5678
-    },
-    {
-      "epoch": 334.06,
-      "learning_rate": 1.746749226006192e-05,
-      "loss": 0.0147,
-      "step": 5679
-    },
-    {
-      "epoch": 334.12,
-      "learning_rate": 1.7461300309597527e-05,
-      "loss": 0.0126,
-      "step": 5680
-    },
-    {
-      "epoch": 334.12,
-      "eval_accuracy_NAT": 0.9940509991223705,
-      "eval_accuracy_REST": 0.424390243902439,
-      "eval_accuracy_concrete": 0.9574562496464657,
-      "eval_accuracy_concrete tiles": 0.9371906669809097,
-      "eval_accuracy_dark bricks": 0.8460846429186686,
-      "eval_accuracy_grass": 0.9043023084172638,
-      "eval_accuracy_red bricks": 0.9259116871704746,
-      "eval_accuracy_road": 0.981997658188511,
-      "eval_accuracy_rough concrete": 0.6521311651243303,
-      "eval_accuracy_rough red sidewalk": 0.7726107129697171,
-      "eval_accuracy_speedway bricks": 0.9905906819853407,
-      "eval_accuracy_steel": 0.9254349358029257,
-      "eval_accuracy_tiles": 0.22596867520979128,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843077536738949,
-      "eval_iou_REST": 0.41037735849056606,
-      "eval_iou_concrete": 0.8986704634974753,
-      "eval_iou_concrete tiles": 0.8842758344637417,
-      "eval_iou_dark bricks": 0.8309232323707029,
-      "eval_iou_grass": 0.8473678797271944,
-      "eval_iou_red bricks": 0.9020331728196896,
-      "eval_iou_road": 0.9636607165801259,
-      "eval_iou_rough concrete": 0.6472063206699153,
-      "eval_iou_rough red sidewalk": 0.6034217450547386,
-      "eval_iou_speedway bricks": 0.9799551605616372,
-      "eval_iou_steel": 0.8779767902229704,
-      "eval_iou_tiles": 0.2247837748244583,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07978561520576477,
-      "eval_mean_accuracy": 0.8106246636491699,
-      "eval_mean_iou": 0.7734584771505469,
-      "eval_overall_accuracy": 0.9808641853979079,
-      "eval_runtime": 6.3204,
-      "eval_samples_per_second": 18.67,
-      "eval_steps_per_second": 1.266,
-      "step": 5680
-    },
-    {
-      "epoch": 334.18,
-      "learning_rate": 1.7455108359133127e-05,
-      "loss": 0.0178,
-      "step": 5681
-    },
-    {
-      "epoch": 334.24,
-      "learning_rate": 1.744891640866873e-05,
-      "loss": 0.0164,
-      "step": 5682
-    },
-    {
-      "epoch": 334.29,
-      "learning_rate": 1.7442724458204333e-05,
-      "loss": 0.0187,
-      "step": 5683
-    },
-    {
-      "epoch": 334.35,
-      "learning_rate": 1.743653250773994e-05,
-      "loss": 0.0195,
-      "step": 5684
-    },
-    {
-      "epoch": 334.41,
-      "learning_rate": 1.7430340557275543e-05,
-      "loss": 0.0165,
-      "step": 5685
-    },
-    {
-      "epoch": 334.47,
-      "learning_rate": 1.7424148606811147e-05,
-      "loss": 0.0157,
-      "step": 5686
-    },
-    {
-      "epoch": 334.53,
-      "learning_rate": 1.741795665634675e-05,
-      "loss": 0.0162,
-      "step": 5687
-    },
-    {
-      "epoch": 334.59,
-      "learning_rate": 1.7411764705882353e-05,
-      "loss": 0.0196,
-      "step": 5688
-    },
-    {
-      "epoch": 334.65,
-      "learning_rate": 1.7405572755417957e-05,
-      "loss": 0.0179,
-      "step": 5689
-    },
-    {
-      "epoch": 334.71,
-      "learning_rate": 1.739938080495356e-05,
-      "loss": 0.0188,
-      "step": 5690
-    },
-    {
-      "epoch": 334.76,
-      "learning_rate": 1.7393188854489163e-05,
-      "loss": 0.0177,
-      "step": 5691
-    },
-    {
-      "epoch": 334.82,
-      "learning_rate": 1.738699690402477e-05,
-      "loss": 0.0179,
-      "step": 5692
-    },
-    {
-      "epoch": 334.88,
-      "learning_rate": 1.7380804953560373e-05,
-      "loss": 0.0173,
-      "step": 5693
-    },
-    {
-      "epoch": 334.94,
-      "learning_rate": 1.7374613003095977e-05,
-      "loss": 0.0149,
-      "step": 5694
-    },
-    {
-      "epoch": 335.0,
-      "learning_rate": 1.736842105263158e-05,
-      "loss": 0.0193,
-      "step": 5695
-    },
-    {
-      "epoch": 335.06,
-      "learning_rate": 1.7362229102167183e-05,
-      "loss": 0.0162,
-      "step": 5696
-    },
-    {
-      "epoch": 335.12,
-      "learning_rate": 1.7356037151702787e-05,
-      "loss": 0.0161,
-      "step": 5697
-    },
-    {
-      "epoch": 335.18,
-      "learning_rate": 1.734984520123839e-05,
-      "loss": 0.0165,
-      "step": 5698
-    },
-    {
-      "epoch": 335.24,
-      "learning_rate": 1.7343653250773993e-05,
-      "loss": 0.018,
-      "step": 5699
-    },
-    {
-      "epoch": 335.29,
-      "learning_rate": 1.73374613003096e-05,
-      "loss": 0.0167,
-      "step": 5700
-    },
-    {
-      "epoch": 335.29,
-      "eval_accuracy_NAT": 0.9940972014998898,
-      "eval_accuracy_REST": 0.45528455284552843,
-      "eval_accuracy_concrete": 0.9592759004997639,
-      "eval_accuracy_concrete tiles": 0.9583549375441904,
-      "eval_accuracy_dark bricks": 0.8482858947610206,
-      "eval_accuracy_grass": 0.9045464351815055,
-      "eval_accuracy_red bricks": 0.9399714411247804,
-      "eval_accuracy_road": 0.9816823526708248,
-      "eval_accuracy_rough concrete": 0.6608131379641161,
-      "eval_accuracy_rough red sidewalk": 0.7560968273185992,
-      "eval_accuracy_speedway bricks": 0.9895662610796142,
-      "eval_accuracy_steel": 0.9283711585460325,
-      "eval_accuracy_tiles": 0.25387015662395107,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984377413152405,
-      "eval_iou_REST": 0.42813455657492355,
-      "eval_iou_concrete": 0.8990344783120295,
-      "eval_iou_concrete tiles": 0.879523284234205,
-      "eval_iou_dark bricks": 0.8323574571185748,
-      "eval_iou_grass": 0.8485913179951603,
-      "eval_iou_red bricks": 0.9067549668874172,
-      "eval_iou_road": 0.9629003512082612,
-      "eval_iou_rough concrete": 0.6550557019150117,
-      "eval_iou_rough red sidewalk": 0.5951803092368527,
-      "eval_iou_speedway bricks": 0.980149556415996,
-      "eval_iou_steel": 0.878386366108203,
-      "eval_iou_tiles": 0.25232603655415375,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07881145924329758,
-      "eval_mean_accuracy": 0.8177089428969089,
-      "eval_mean_iou": 0.777136291977938,
-      "eval_overall_accuracy": 0.9809384750107587,
-      "eval_runtime": 6.0309,
-      "eval_samples_per_second": 19.566,
-      "eval_steps_per_second": 1.327,
-      "step": 5700
-    },
-    {
-      "epoch": 335.35,
-      "learning_rate": 1.7331269349845203e-05,
-      "loss": 0.0169,
-      "step": 5701
-    },
-    {
-      "epoch": 335.41,
-      "learning_rate": 1.7325077399380807e-05,
-      "loss": 0.0173,
-      "step": 5702
-    },
-    {
-      "epoch": 335.47,
-      "learning_rate": 1.7318885448916407e-05,
-      "loss": 0.0178,
-      "step": 5703
-    },
-    {
-      "epoch": 335.53,
-      "learning_rate": 1.7312693498452013e-05,
-      "loss": 0.0195,
-      "step": 5704
-    },
-    {
-      "epoch": 335.59,
-      "learning_rate": 1.7306501547987617e-05,
-      "loss": 0.0176,
-      "step": 5705
-    },
-    {
-      "epoch": 335.65,
-      "learning_rate": 1.730030959752322e-05,
-      "loss": 0.016,
-      "step": 5706
-    },
-    {
-      "epoch": 335.71,
-      "learning_rate": 1.7294117647058823e-05,
-      "loss": 0.018,
-      "step": 5707
-    },
-    {
-      "epoch": 335.76,
-      "learning_rate": 1.728792569659443e-05,
-      "loss": 0.0179,
-      "step": 5708
-    },
-    {
-      "epoch": 335.82,
-      "learning_rate": 1.7281733746130033e-05,
-      "loss": 0.0195,
-      "step": 5709
-    },
-    {
-      "epoch": 335.88,
-      "learning_rate": 1.7275541795665637e-05,
-      "loss": 0.0145,
-      "step": 5710
-    },
-    {
-      "epoch": 335.94,
-      "learning_rate": 1.7269349845201237e-05,
-      "loss": 0.0177,
-      "step": 5711
-    },
-    {
-      "epoch": 336.0,
-      "learning_rate": 1.7263157894736843e-05,
-      "loss": 0.0177,
-      "step": 5712
-    },
-    {
-      "epoch": 336.06,
-      "learning_rate": 1.7256965944272447e-05,
-      "loss": 0.0159,
-      "step": 5713
-    },
-    {
-      "epoch": 336.12,
-      "learning_rate": 1.725077399380805e-05,
-      "loss": 0.0182,
-      "step": 5714
-    },
-    {
-      "epoch": 336.18,
-      "learning_rate": 1.7244582043343653e-05,
-      "loss": 0.0146,
-      "step": 5715
-    },
-    {
-      "epoch": 336.24,
-      "learning_rate": 1.723839009287926e-05,
-      "loss": 0.0144,
-      "step": 5716
-    },
-    {
-      "epoch": 336.29,
-      "learning_rate": 1.7232198142414863e-05,
-      "loss": 0.0169,
-      "step": 5717
-    },
-    {
-      "epoch": 336.35,
-      "learning_rate": 1.7226006191950463e-05,
-      "loss": 0.0158,
-      "step": 5718
-    },
-    {
-      "epoch": 336.41,
-      "learning_rate": 1.7219814241486067e-05,
-      "loss": 0.0204,
-      "step": 5719
-    },
-    {
-      "epoch": 336.47,
-      "learning_rate": 1.7213622291021673e-05,
-      "loss": 0.0199,
-      "step": 5720
-    },
-    {
-      "epoch": 336.47,
-      "eval_accuracy_NAT": 0.9933210878163792,
-      "eval_accuracy_REST": 0.5317073170731708,
-      "eval_accuracy_concrete": 0.9609965617351237,
-      "eval_accuracy_concrete tiles": 0.9451095922696205,
-      "eval_accuracy_dark bricks": 0.8501473881668358,
-      "eval_accuracy_grass": 0.9132375372340626,
-      "eval_accuracy_red bricks": 0.930305360281195,
-      "eval_accuracy_road": 0.9813633062402168,
-      "eval_accuracy_rough concrete": 0.6460829822606499,
-      "eval_accuracy_rough red sidewalk": 0.7705502372555065,
-      "eval_accuracy_speedway bricks": 0.9895415844970741,
-      "eval_accuracy_steel": 0.9274515585720212,
-      "eval_accuracy_tiles": 0.2532036717258929,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843457291470721,
-      "eval_iou_REST": 0.49395770392749244,
-      "eval_iou_concrete": 0.8977263233360873,
-      "eval_iou_concrete tiles": 0.8929987084131297,
-      "eval_iou_dark bricks": 0.834481786796308,
-      "eval_iou_grass": 0.848771037123482,
-      "eval_iou_red bricks": 0.90278740073549,
-      "eval_iou_road": 0.9624718739533809,
-      "eval_iou_rough concrete": 0.640359321753613,
-      "eval_iou_rough red sidewalk": 0.6099335107174533,
-      "eval_iou_speedway bricks": 0.980181181517781,
-      "eval_iou_steel": 0.8785569719135036,
-      "eval_iou_tiles": 0.25110410094637226,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08000928908586502,
-      "eval_mean_accuracy": 0.8225398603944423,
-      "eval_mean_iou": 0.782898126944705,
-      "eval_overall_accuracy": 0.9808514481883938,
-      "eval_runtime": 5.4883,
-      "eval_samples_per_second": 21.5,
-      "eval_steps_per_second": 1.458,
-      "step": 5720
-    },
-    {
-      "epoch": 336.53,
-      "learning_rate": 1.7207430340557277e-05,
-      "loss": 0.0201,
-      "step": 5721
-    },
-    {
-      "epoch": 336.59,
-      "learning_rate": 1.720123839009288e-05,
-      "loss": 0.0144,
-      "step": 5722
-    },
-    {
-      "epoch": 336.65,
-      "learning_rate": 1.7195046439628483e-05,
-      "loss": 0.0159,
-      "step": 5723
-    },
-    {
-      "epoch": 336.71,
-      "learning_rate": 1.718885448916409e-05,
-      "loss": 0.0173,
-      "step": 5724
-    },
-    {
-      "epoch": 336.76,
-      "learning_rate": 1.7182662538699693e-05,
-      "loss": 0.0181,
-      "step": 5725
-    },
-    {
-      "epoch": 336.82,
-      "learning_rate": 1.7176470588235293e-05,
-      "loss": 0.0166,
-      "step": 5726
-    },
-    {
-      "epoch": 336.88,
-      "learning_rate": 1.7170278637770897e-05,
-      "loss": 0.0188,
-      "step": 5727
-    },
-    {
-      "epoch": 336.94,
-      "learning_rate": 1.7164086687306503e-05,
-      "loss": 0.0157,
-      "step": 5728
-    },
-    {
-      "epoch": 337.0,
-      "learning_rate": 1.7157894736842107e-05,
-      "loss": 0.0168,
-      "step": 5729
-    },
-    {
-      "epoch": 337.06,
-      "learning_rate": 1.715170278637771e-05,
-      "loss": 0.0171,
-      "step": 5730
-    },
-    {
-      "epoch": 337.12,
-      "learning_rate": 1.7145510835913313e-05,
-      "loss": 0.0173,
-      "step": 5731
-    },
-    {
-      "epoch": 337.18,
-      "learning_rate": 1.713931888544892e-05,
-      "loss": 0.0156,
-      "step": 5732
-    },
-    {
-      "epoch": 337.24,
-      "learning_rate": 1.713312693498452e-05,
-      "loss": 0.0144,
-      "step": 5733
-    },
-    {
-      "epoch": 337.29,
-      "learning_rate": 1.7126934984520123e-05,
-      "loss": 0.0186,
-      "step": 5734
-    },
-    {
-      "epoch": 337.35,
-      "learning_rate": 1.7120743034055727e-05,
-      "loss": 0.0195,
-      "step": 5735
-    },
-    {
-      "epoch": 337.41,
-      "learning_rate": 1.7114551083591333e-05,
-      "loss": 0.0207,
-      "step": 5736
-    },
-    {
-      "epoch": 337.47,
-      "learning_rate": 1.7108359133126937e-05,
-      "loss": 0.0155,
-      "step": 5737
-    },
-    {
-      "epoch": 337.53,
-      "learning_rate": 1.710216718266254e-05,
-      "loss": 0.0165,
-      "step": 5738
-    },
-    {
-      "epoch": 337.59,
-      "learning_rate": 1.7095975232198143e-05,
-      "loss": 0.0189,
-      "step": 5739
-    },
-    {
-      "epoch": 337.65,
-      "learning_rate": 1.708978328173375e-05,
-      "loss": 0.0148,
-      "step": 5740
-    },
-    {
-      "epoch": 337.65,
-      "eval_accuracy_NAT": 0.9945971791534657,
-      "eval_accuracy_REST": 0.4373983739837398,
-      "eval_accuracy_concrete": 0.9605566503442087,
-      "eval_accuracy_concrete tiles": 0.9406080603346688,
-      "eval_accuracy_dark bricks": 0.8454434086863312,
-      "eval_accuracy_grass": 0.8910589045710371,
-      "eval_accuracy_red bricks": 0.9276691564147628,
-      "eval_accuracy_road": 0.9810549481322424,
-      "eval_accuracy_rough concrete": 0.6395044661212378,
-      "eval_accuracy_rough red sidewalk": 0.7811834951383301,
-      "eval_accuracy_speedway bricks": 0.9899614291305686,
-      "eval_accuracy_steel": 0.9238481260651616,
-      "eval_accuracy_tiles": 0.2522645339149929,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9840941886816286,
-      "eval_iou_REST": 0.3853868194842407,
-      "eval_iou_concrete": 0.898105150310448,
-      "eval_iou_concrete tiles": 0.8880927479472174,
-      "eval_iou_dark bricks": 0.8310824477968605,
-      "eval_iou_grass": 0.8436201014820973,
-      "eval_iou_red bricks": 0.9010455563853622,
-      "eval_iou_road": 0.9651038205630746,
-      "eval_iou_rough concrete": 0.6346862377508236,
-      "eval_iou_rough red sidewalk": 0.6174682144831398,
-      "eval_iou_speedway bricks": 0.9800868162827525,
-      "eval_iou_steel": 0.8773407817516914,
-      "eval_iou_tiles": 0.25021033653846153,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0813649520277977,
-      "eval_mean_accuracy": 0.8127037486146728,
-      "eval_mean_iou": 0.7735633245736767,
-      "eval_overall_accuracy": 0.9808032795534296,
-      "eval_runtime": 5.8201,
-      "eval_samples_per_second": 20.275,
-      "eval_steps_per_second": 1.375,
-      "step": 5740
-    },
-    {
-      "epoch": 337.71,
-      "learning_rate": 1.708359133126935e-05,
-      "loss": 0.0202,
-      "step": 5741
-    },
-    {
-      "epoch": 337.76,
-      "learning_rate": 1.7077399380804953e-05,
-      "loss": 0.0161,
-      "step": 5742
-    },
-    {
-      "epoch": 337.82,
-      "learning_rate": 1.7071207430340557e-05,
-      "loss": 0.0173,
-      "step": 5743
-    },
-    {
-      "epoch": 337.88,
-      "learning_rate": 1.7065015479876163e-05,
-      "loss": 0.0209,
-      "step": 5744
-    },
-    {
-      "epoch": 337.94,
-      "learning_rate": 1.7058823529411767e-05,
-      "loss": 0.0154,
-      "step": 5745
-    },
-    {
-      "epoch": 338.0,
-      "learning_rate": 1.705263157894737e-05,
-      "loss": 0.0144,
-      "step": 5746
-    },
-    {
-      "epoch": 338.06,
-      "learning_rate": 1.7046439628482973e-05,
-      "loss": 0.0192,
-      "step": 5747
-    },
-    {
-      "epoch": 338.12,
-      "learning_rate": 1.7040247678018577e-05,
-      "loss": 0.0173,
-      "step": 5748
-    },
-    {
-      "epoch": 338.18,
-      "learning_rate": 1.703405572755418e-05,
-      "loss": 0.0149,
-      "step": 5749
-    },
-    {
-      "epoch": 338.24,
-      "learning_rate": 1.7027863777089783e-05,
-      "loss": 0.0165,
-      "step": 5750
-    },
-    {
-      "epoch": 338.29,
-      "learning_rate": 1.7021671826625387e-05,
-      "loss": 0.0179,
-      "step": 5751
-    },
-    {
-      "epoch": 338.35,
-      "learning_rate": 1.7015479876160993e-05,
-      "loss": 0.0163,
-      "step": 5752
-    },
-    {
-      "epoch": 338.41,
-      "learning_rate": 1.7009287925696597e-05,
-      "loss": 0.0159,
-      "step": 5753
-    },
-    {
-      "epoch": 338.47,
-      "learning_rate": 1.70030959752322e-05,
-      "loss": 0.0209,
-      "step": 5754
-    },
-    {
-      "epoch": 338.53,
-      "learning_rate": 1.69969040247678e-05,
-      "loss": 0.0166,
-      "step": 5755
-    },
-    {
-      "epoch": 338.59,
-      "learning_rate": 1.6990712074303407e-05,
-      "loss": 0.0184,
-      "step": 5756
-    },
-    {
-      "epoch": 338.65,
-      "learning_rate": 1.698452012383901e-05,
-      "loss": 0.0148,
-      "step": 5757
-    },
-    {
-      "epoch": 338.71,
-      "learning_rate": 1.6978328173374613e-05,
-      "loss": 0.0214,
-      "step": 5758
-    },
-    {
-      "epoch": 338.76,
-      "learning_rate": 1.6972136222910217e-05,
-      "loss": 0.0191,
-      "step": 5759
-    },
-    {
-      "epoch": 338.82,
-      "learning_rate": 1.6965944272445823e-05,
-      "loss": 0.016,
-      "step": 5760
-    },
-    {
-      "epoch": 338.82,
-      "eval_accuracy_NAT": 0.9930633896080447,
-      "eval_accuracy_REST": 0.5691056910569106,
-      "eval_accuracy_concrete": 0.9610328706683443,
-      "eval_accuracy_concrete tiles": 0.9573415036530757,
-      "eval_accuracy_dark bricks": 0.8547460903853148,
-      "eval_accuracy_grass": 0.9094015298610559,
-      "eval_accuracy_red bricks": 0.9471660808435852,
-      "eval_accuracy_road": 0.9813478081723984,
-      "eval_accuracy_rough concrete": 0.6475965176767827,
-      "eval_accuracy_rough red sidewalk": 0.8062626429737026,
-      "eval_accuracy_speedway bricks": 0.990664711732961,
-      "eval_accuracy_steel": 0.9274290683552655,
-      "eval_accuracy_tiles": 0.2461752855281893,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844789343948936,
-      "eval_iou_REST": 0.4820936639118457,
-      "eval_iou_concrete": 0.8976682582890803,
-      "eval_iou_concrete tiles": 0.8885680535503346,
-      "eval_iou_dark bricks": 0.8354404329259452,
-      "eval_iou_grass": 0.8480323513399282,
-      "eval_iou_red bricks": 0.9062532842879664,
-      "eval_iou_road": 0.9653487806044854,
-      "eval_iou_rough concrete": 0.640644895071917,
-      "eval_iou_rough red sidewalk": 0.6267925521031247,
-      "eval_iou_speedway bricks": 0.9801427128600099,
-      "eval_iou_steel": 0.8787852956846031,
-      "eval_iou_tiles": 0.2439361191162344,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07919301837682724,
-      "eval_mean_accuracy": 0.8301025531165869,
-      "eval_mean_iou": 0.782937333395413,
-      "eval_overall_accuracy": 0.9809923333636785,
-      "eval_runtime": 5.8812,
-      "eval_samples_per_second": 20.064,
-      "eval_steps_per_second": 1.36,
-      "step": 5760
-    },
-    {
-      "epoch": 338.88,
-      "learning_rate": 1.6959752321981427e-05,
-      "loss": 0.0177,
-      "step": 5761
-    },
-    {
-      "epoch": 338.94,
-      "learning_rate": 1.695356037151703e-05,
-      "loss": 0.0161,
-      "step": 5762
-    },
-    {
-      "epoch": 339.0,
-      "learning_rate": 1.694736842105263e-05,
-      "loss": 0.0161,
-      "step": 5763
-    },
-    {
-      "epoch": 339.06,
-      "learning_rate": 1.6941176470588237e-05,
-      "loss": 0.0179,
-      "step": 5764
-    },
-    {
-      "epoch": 339.12,
-      "learning_rate": 1.693498452012384e-05,
-      "loss": 0.0156,
-      "step": 5765
-    },
-    {
-      "epoch": 339.18,
-      "learning_rate": 1.6928792569659443e-05,
-      "loss": 0.0161,
-      "step": 5766
-    },
-    {
-      "epoch": 339.24,
-      "learning_rate": 1.6922600619195046e-05,
-      "loss": 0.0205,
-      "step": 5767
-    },
-    {
-      "epoch": 339.29,
-      "learning_rate": 1.6916408668730653e-05,
-      "loss": 0.0154,
-      "step": 5768
-    },
-    {
-      "epoch": 339.35,
-      "learning_rate": 1.6910216718266257e-05,
-      "loss": 0.0172,
-      "step": 5769
-    },
-    {
-      "epoch": 339.41,
-      "learning_rate": 1.6904024767801856e-05,
-      "loss": 0.0173,
-      "step": 5770
-    },
-    {
-      "epoch": 339.47,
-      "learning_rate": 1.689783281733746e-05,
-      "loss": 0.0147,
-      "step": 5771
-    },
-    {
-      "epoch": 339.53,
-      "learning_rate": 1.6891640866873066e-05,
-      "loss": 0.015,
-      "step": 5772
-    },
-    {
-      "epoch": 339.59,
-      "learning_rate": 1.688544891640867e-05,
-      "loss": 0.018,
-      "step": 5773
-    },
-    {
-      "epoch": 339.65,
-      "learning_rate": 1.6879256965944273e-05,
-      "loss": 0.015,
-      "step": 5774
-    },
-    {
-      "epoch": 339.71,
-      "learning_rate": 1.6873065015479876e-05,
-      "loss": 0.0194,
-      "step": 5775
-    },
-    {
-      "epoch": 339.76,
-      "learning_rate": 1.6866873065015483e-05,
-      "loss": 0.0154,
-      "step": 5776
-    },
-    {
-      "epoch": 339.82,
-      "learning_rate": 1.6860681114551087e-05,
-      "loss": 0.0199,
-      "step": 5777
-    },
-    {
-      "epoch": 339.88,
-      "learning_rate": 1.6854489164086686e-05,
-      "loss": 0.0166,
-      "step": 5778
-    },
-    {
-      "epoch": 339.94,
-      "learning_rate": 1.684829721362229e-05,
-      "loss": 0.0166,
-      "step": 5779
-    },
-    {
-      "epoch": 340.0,
-      "learning_rate": 1.6842105263157896e-05,
-      "loss": 0.0199,
-      "step": 5780
-    },
-    {
-      "epoch": 340.0,
-      "eval_accuracy_NAT": 0.994129332908293,
-      "eval_accuracy_REST": 0.44878048780487806,
-      "eval_accuracy_concrete": 0.9586796695963516,
-      "eval_accuracy_concrete tiles": 0.9332312043365544,
-      "eval_accuracy_dark bricks": 0.8489558409739104,
-      "eval_accuracy_grass": 0.908691859034772,
-      "eval_accuracy_red bricks": 0.9355777680140598,
-      "eval_accuracy_road": 0.9809063804476377,
-      "eval_accuracy_rough concrete": 0.6454453905696018,
-      "eval_accuracy_rough red sidewalk": 0.7833116507117666,
-      "eval_accuracy_speedway bricks": 0.9895669465402404,
-      "eval_accuracy_steel": 0.9298005367665065,
-      "eval_accuracy_tiles": 0.23354236723317884,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843545264613472,
-      "eval_iou_REST": 0.41566265060240964,
-      "eval_iou_concrete": 0.898511736636812,
-      "eval_iou_concrete tiles": 0.8860966276545752,
-      "eval_iou_dark bricks": 0.8329874447125994,
-      "eval_iou_grass": 0.8487583233609144,
-      "eval_iou_red bricks": 0.9081458577673526,
-      "eval_iou_road": 0.9649541173470862,
-      "eval_iou_rough concrete": 0.6408249471984098,
-      "eval_iou_rough red sidewalk": 0.6192276596756551,
-      "eval_iou_speedway bricks": 0.9801608828459751,
-      "eval_iou_steel": 0.8787830120499001,
-      "eval_iou_tiles": 0.23193332932186053,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07996427267789841,
-      "eval_mean_accuracy": 0.8146630334567502,
-      "eval_mean_iou": 0.7761847012026846,
-      "eval_overall_accuracy": 0.9809998981023239,
-      "eval_runtime": 5.7337,
-      "eval_samples_per_second": 20.58,
-      "eval_steps_per_second": 1.395,
-      "step": 5780
-    },
-    {
-      "epoch": 340.06,
-      "learning_rate": 1.68359133126935e-05,
-      "loss": 0.0208,
-      "step": 5781
-    },
-    {
-      "epoch": 340.12,
-      "learning_rate": 1.6829721362229103e-05,
-      "loss": 0.0154,
-      "step": 5782
-    },
-    {
-      "epoch": 340.18,
-      "learning_rate": 1.6823529411764706e-05,
-      "loss": 0.0169,
-      "step": 5783
-    },
-    {
-      "epoch": 340.24,
-      "learning_rate": 1.6817337461300313e-05,
-      "loss": 0.017,
-      "step": 5784
-    },
-    {
-      "epoch": 340.29,
-      "learning_rate": 1.6811145510835913e-05,
-      "loss": 0.0155,
-      "step": 5785
-    },
-    {
-      "epoch": 340.35,
-      "learning_rate": 1.6804953560371516e-05,
-      "loss": 0.0156,
-      "step": 5786
-    },
-    {
-      "epoch": 340.41,
-      "learning_rate": 1.679876160990712e-05,
-      "loss": 0.0182,
-      "step": 5787
-    },
-    {
-      "epoch": 340.47,
-      "learning_rate": 1.6792569659442726e-05,
-      "loss": 0.019,
-      "step": 5788
-    },
-    {
-      "epoch": 340.53,
-      "learning_rate": 1.678637770897833e-05,
-      "loss": 0.0159,
-      "step": 5789
-    },
-    {
-      "epoch": 340.59,
-      "learning_rate": 1.6780185758513933e-05,
-      "loss": 0.0156,
-      "step": 5790
-    },
-    {
-      "epoch": 340.65,
-      "learning_rate": 1.6773993808049536e-05,
-      "loss": 0.0172,
-      "step": 5791
-    },
-    {
-      "epoch": 340.71,
-      "learning_rate": 1.6767801857585143e-05,
-      "loss": 0.014,
-      "step": 5792
-    },
-    {
-      "epoch": 340.76,
-      "learning_rate": 1.6761609907120743e-05,
-      "loss": 0.0172,
-      "step": 5793
-    },
-    {
-      "epoch": 340.82,
-      "learning_rate": 1.6755417956656346e-05,
-      "loss": 0.017,
-      "step": 5794
-    },
-    {
-      "epoch": 340.88,
-      "learning_rate": 1.674922600619195e-05,
-      "loss": 0.0174,
-      "step": 5795
-    },
-    {
-      "epoch": 340.94,
-      "learning_rate": 1.6743034055727556e-05,
-      "loss": 0.021,
-      "step": 5796
-    },
-    {
-      "epoch": 341.0,
-      "learning_rate": 1.673684210526316e-05,
-      "loss": 0.0167,
-      "step": 5797
-    },
-    {
-      "epoch": 341.06,
-      "learning_rate": 1.6730650154798763e-05,
-      "loss": 0.0148,
-      "step": 5798
-    },
-    {
-      "epoch": 341.12,
-      "learning_rate": 1.6724458204334366e-05,
-      "loss": 0.0162,
-      "step": 5799
-    },
-    {
-      "epoch": 341.18,
-      "learning_rate": 1.671826625386997e-05,
-      "loss": 0.0174,
-      "step": 5800
-    },
-    {
-      "epoch": 341.18,
-      "eval_accuracy_NAT": 0.9939397252516587,
-      "eval_accuracy_REST": 0.46016260162601624,
-      "eval_accuracy_concrete": 0.9600441210868524,
-      "eval_accuracy_concrete tiles": 0.9448974781993872,
-      "eval_accuracy_dark bricks": 0.858191528051605,
-      "eval_accuracy_grass": 0.9025442172236163,
-      "eval_accuracy_red bricks": 0.9368958699472759,
-      "eval_accuracy_road": 0.9819030665332051,
-      "eval_accuracy_rough concrete": 0.6367217061238596,
-      "eval_accuracy_rough red sidewalk": 0.7819054136367396,
-      "eval_accuracy_speedway bricks": 0.9895575214566312,
-      "eval_accuracy_steel": 0.9298130313313708,
-      "eval_accuracy_tiles": 0.2367839074191887,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843053412943727,
-      "eval_iou_REST": 0.4255639097744361,
-      "eval_iou_concrete": 0.8980347867309761,
-      "eval_iou_concrete tiles": 0.886834188639179,
-      "eval_iou_dark bricks": 0.8407309480948095,
-      "eval_iou_grass": 0.8453929146580659,
-      "eval_iou_red bricks": 0.9070076563164611,
-      "eval_iou_road": 0.9638600608846052,
-      "eval_iou_rough concrete": 0.6314016769778944,
-      "eval_iou_rough red sidewalk": 0.6146738552122868,
-      "eval_iou_speedway bricks": 0.9804564270263478,
-      "eval_iou_steel": 0.8792911545560027,
-      "eval_iou_tiles": 0.23513838748495788,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0803859606385231,
-      "eval_mean_accuracy": 0.8164123221451853,
-      "eval_mean_iou": 0.7763608698192612,
-      "eval_overall_accuracy": 0.9808797351384567,
-      "eval_runtime": 6.2845,
-      "eval_samples_per_second": 18.776,
-      "eval_steps_per_second": 1.273,
-      "step": 5800
-    },
-    {
-      "epoch": 341.24,
-      "learning_rate": 1.6712074303405573e-05,
-      "loss": 0.017,
-      "step": 5801
-    },
-    {
-      "epoch": 341.29,
-      "learning_rate": 1.6705882352941176e-05,
-      "loss": 0.0182,
-      "step": 5802
-    },
-    {
-      "epoch": 341.35,
-      "learning_rate": 1.669969040247678e-05,
-      "loss": 0.0158,
-      "step": 5803
-    },
-    {
-      "epoch": 341.41,
-      "learning_rate": 1.6693498452012386e-05,
-      "loss": 0.0183,
-      "step": 5804
-    },
-    {
-      "epoch": 341.47,
-      "learning_rate": 1.668730650154799e-05,
-      "loss": 0.0171,
-      "step": 5805
-    },
-    {
-      "epoch": 341.53,
-      "learning_rate": 1.6681114551083593e-05,
-      "loss": 0.0152,
-      "step": 5806
-    },
-    {
-      "epoch": 341.59,
-      "learning_rate": 1.6674922600619196e-05,
-      "loss": 0.0162,
-      "step": 5807
-    },
-    {
-      "epoch": 341.65,
-      "learning_rate": 1.66687306501548e-05,
-      "loss": 0.0172,
-      "step": 5808
-    },
-    {
-      "epoch": 341.71,
-      "learning_rate": 1.6662538699690403e-05,
-      "loss": 0.0182,
-      "step": 5809
-    },
-    {
-      "epoch": 341.76,
-      "learning_rate": 1.6656346749226006e-05,
-      "loss": 0.015,
-      "step": 5810
-    },
-    {
-      "epoch": 341.82,
-      "learning_rate": 1.665015479876161e-05,
-      "loss": 0.0147,
-      "step": 5811
-    },
-    {
-      "epoch": 341.88,
-      "learning_rate": 1.6643962848297216e-05,
-      "loss": 0.0207,
-      "step": 5812
-    },
-    {
-      "epoch": 341.94,
-      "learning_rate": 1.663777089783282e-05,
-      "loss": 0.0167,
-      "step": 5813
-    },
-    {
-      "epoch": 342.0,
-      "learning_rate": 1.6631578947368423e-05,
-      "loss": 0.0213,
-      "step": 5814
-    },
-    {
-      "epoch": 342.06,
-      "learning_rate": 1.6625386996904023e-05,
-      "loss": 0.0175,
-      "step": 5815
-    },
-    {
-      "epoch": 342.12,
-      "learning_rate": 1.661919504643963e-05,
-      "loss": 0.0152,
-      "step": 5816
-    },
-    {
-      "epoch": 342.18,
-      "learning_rate": 1.6613003095975233e-05,
-      "loss": 0.016,
-      "step": 5817
-    },
-    {
-      "epoch": 342.24,
-      "learning_rate": 1.6606811145510836e-05,
-      "loss": 0.0192,
-      "step": 5818
-    },
-    {
-      "epoch": 342.29,
-      "learning_rate": 1.660061919504644e-05,
-      "loss": 0.0176,
-      "step": 5819
-    },
-    {
-      "epoch": 342.35,
-      "learning_rate": 1.6594427244582046e-05,
-      "loss": 0.0154,
-      "step": 5820
-    },
-    {
-      "epoch": 342.35,
-      "eval_accuracy_NAT": 0.9937419230076465,
-      "eval_accuracy_REST": 0.5365853658536586,
-      "eval_accuracy_concrete": 0.9583230776521956,
-      "eval_accuracy_concrete tiles": 0.9549846806504831,
-      "eval_accuracy_dark bricks": 0.8560811974810022,
-      "eval_accuracy_grass": 0.904865313939449,
-      "eval_accuracy_red bricks": 0.9394771528998243,
-      "eval_accuracy_road": 0.9834159986019674,
-      "eval_accuracy_rough concrete": 0.6410120427365197,
-      "eval_accuracy_rough red sidewalk": 0.7680836823859406,
-      "eval_accuracy_speedway bricks": 0.990976082222373,
-      "eval_accuracy_steel": 0.9229235282652046,
-      "eval_accuracy_tiles": 0.24868975127995396,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846046397715075,
-      "eval_iou_REST": 0.47277936962750716,
-      "eval_iou_concrete": 0.8977584958678041,
-      "eval_iou_concrete tiles": 0.8833467768306773,
-      "eval_iou_dark bricks": 0.8377563301068169,
-      "eval_iou_grass": 0.8499339188058991,
-      "eval_iou_red bricks": 0.9054626296845225,
-      "eval_iou_road": 0.9620852956676677,
-      "eval_iou_rough concrete": 0.6348271507311719,
-      "eval_iou_rough red sidewalk": 0.6047485123893543,
-      "eval_iou_speedway bricks": 0.9802515062712143,
-      "eval_iou_steel": 0.8778376476907829,
-      "eval_iou_tiles": 0.24648690847946192,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08056189864873886,
-      "eval_mean_accuracy": 0.8230122920750939,
-      "eval_mean_iou": 0.7798368601480298,
-      "eval_overall_accuracy": 0.9809440354169425,
-      "eval_runtime": 5.5462,
-      "eval_samples_per_second": 21.276,
-      "eval_steps_per_second": 1.442,
-      "step": 5820
-    },
-    {
-      "epoch": 342.41,
-      "learning_rate": 1.658823529411765e-05,
-      "loss": 0.0181,
-      "step": 5821
-    },
-    {
-      "epoch": 342.47,
-      "learning_rate": 1.658204334365325e-05,
-      "loss": 0.0171,
-      "step": 5822
-    },
-    {
-      "epoch": 342.53,
-      "learning_rate": 1.6575851393188853e-05,
-      "loss": 0.0156,
-      "step": 5823
-    },
-    {
-      "epoch": 342.59,
-      "learning_rate": 1.656965944272446e-05,
-      "loss": 0.0189,
-      "step": 5824
-    },
-    {
-      "epoch": 342.65,
-      "learning_rate": 1.6563467492260063e-05,
-      "loss": 0.0163,
-      "step": 5825
-    },
-    {
-      "epoch": 342.71,
-      "learning_rate": 1.6557275541795666e-05,
-      "loss": 0.0169,
-      "step": 5826
-    },
-    {
-      "epoch": 342.76,
-      "learning_rate": 1.655108359133127e-05,
-      "loss": 0.0176,
-      "step": 5827
-    },
-    {
-      "epoch": 342.82,
-      "learning_rate": 1.6544891640866876e-05,
-      "loss": 0.0157,
-      "step": 5828
-    },
-    {
-      "epoch": 342.88,
-      "learning_rate": 1.653869969040248e-05,
-      "loss": 0.0149,
-      "step": 5829
-    },
-    {
-      "epoch": 342.94,
-      "learning_rate": 1.653250773993808e-05,
-      "loss": 0.0184,
-      "step": 5830
-    },
-    {
-      "epoch": 343.0,
-      "learning_rate": 1.6526315789473683e-05,
-      "loss": 0.0167,
-      "step": 5831
-    },
-    {
-      "epoch": 343.06,
-      "learning_rate": 1.652012383900929e-05,
-      "loss": 0.0132,
-      "step": 5832
-    },
-    {
-      "epoch": 343.12,
-      "learning_rate": 1.6513931888544893e-05,
-      "loss": 0.019,
-      "step": 5833
-    },
-    {
-      "epoch": 343.18,
-      "learning_rate": 1.6507739938080496e-05,
-      "loss": 0.019,
-      "step": 5834
-    },
-    {
-      "epoch": 343.24,
-      "learning_rate": 1.65015479876161e-05,
-      "loss": 0.017,
-      "step": 5835
-    },
-    {
-      "epoch": 343.29,
-      "learning_rate": 1.6495356037151706e-05,
-      "loss": 0.015,
-      "step": 5836
-    },
-    {
-      "epoch": 343.35,
-      "learning_rate": 1.6489164086687306e-05,
-      "loss": 0.0183,
-      "step": 5837
-    },
-    {
-      "epoch": 343.41,
-      "learning_rate": 1.648297213622291e-05,
-      "loss": 0.0176,
-      "step": 5838
-    },
-    {
-      "epoch": 343.47,
-      "learning_rate": 1.6476780185758513e-05,
-      "loss": 0.0143,
-      "step": 5839
-    },
-    {
-      "epoch": 343.53,
-      "learning_rate": 1.647058823529412e-05,
-      "loss": 0.0183,
-      "step": 5840
-    },
-    {
-      "epoch": 343.53,
-      "eval_accuracy_NAT": 0.9940308900362389,
-      "eval_accuracy_REST": 0.5317073170731708,
-      "eval_accuracy_concrete": 0.9605856974907852,
-      "eval_accuracy_concrete tiles": 0.9375206222012726,
-      "eval_accuracy_dark bricks": 0.8518222536990602,
-      "eval_accuracy_grass": 0.8999629078714796,
-      "eval_accuracy_red bricks": 0.9337653778558875,
-      "eval_accuracy_road": 0.9822253194606031,
-      "eval_accuracy_rough concrete": 0.6386821516038113,
-      "eval_accuracy_rough red sidewalk": 0.7754758270102798,
-      "eval_accuracy_speedway bricks": 0.9898567250199297,
-      "eval_accuracy_steel": 0.9281562520303668,
-      "eval_accuracy_tiles": 0.24529673725347634,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844329911988858,
-      "eval_iou_REST": 0.461864406779661,
-      "eval_iou_concrete": 0.8974963201133569,
-      "eval_iou_concrete tiles": 0.8861636481097819,
-      "eval_iou_dark bricks": 0.8372466017590894,
-      "eval_iou_grass": 0.8468600664056622,
-      "eval_iou_red bricks": 0.9069184402837788,
-      "eval_iou_road": 0.9642547980823409,
-      "eval_iou_rough concrete": 0.6325197398704073,
-      "eval_iou_rough red sidewalk": 0.6080916129564874,
-      "eval_iou_speedway bricks": 0.9801644133336139,
-      "eval_iou_steel": 0.8788583624571424,
-      "eval_iou_tiles": 0.24341630591630592,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08126283437013626,
-      "eval_mean_accuracy": 0.82069908296972,
-      "eval_mean_iou": 0.7790990544051165,
-      "eval_overall_accuracy": 0.9808850369210971,
-      "eval_runtime": 5.7732,
-      "eval_samples_per_second": 20.439,
-      "eval_steps_per_second": 1.386,
-      "step": 5840
-    },
-    {
-      "epoch": 343.59,
-      "learning_rate": 1.6464396284829723e-05,
-      "loss": 0.0158,
-      "step": 5841
-    },
-    {
-      "epoch": 343.65,
-      "learning_rate": 1.6458204334365326e-05,
-      "loss": 0.0171,
-      "step": 5842
-    },
-    {
-      "epoch": 343.71,
-      "learning_rate": 1.645201238390093e-05,
-      "loss": 0.0171,
-      "step": 5843
-    },
-    {
-      "epoch": 343.76,
-      "learning_rate": 1.6445820433436533e-05,
-      "loss": 0.0177,
-      "step": 5844
-    },
-    {
-      "epoch": 343.82,
-      "learning_rate": 1.6439628482972136e-05,
-      "loss": 0.0165,
-      "step": 5845
-    },
-    {
-      "epoch": 343.88,
-      "learning_rate": 1.643343653250774e-05,
-      "loss": 0.0198,
-      "step": 5846
-    },
-    {
-      "epoch": 343.94,
-      "learning_rate": 1.6427244582043343e-05,
-      "loss": 0.017,
-      "step": 5847
-    },
-    {
-      "epoch": 344.0,
-      "learning_rate": 1.642105263157895e-05,
-      "loss": 0.0192,
-      "step": 5848
-    },
-    {
-      "epoch": 344.06,
-      "learning_rate": 1.6414860681114553e-05,
-      "loss": 0.0196,
-      "step": 5849
-    },
-    {
-      "epoch": 344.12,
-      "learning_rate": 1.6408668730650156e-05,
-      "loss": 0.0157,
-      "step": 5850
-    },
-    {
-      "epoch": 344.18,
-      "learning_rate": 1.640247678018576e-05,
-      "loss": 0.0175,
-      "step": 5851
-    },
-    {
-      "epoch": 344.24,
-      "learning_rate": 1.6396284829721363e-05,
-      "loss": 0.0201,
-      "step": 5852
-    },
-    {
-      "epoch": 344.29,
-      "learning_rate": 1.6390092879256966e-05,
-      "loss": 0.0161,
-      "step": 5853
-    },
-    {
-      "epoch": 344.35,
-      "learning_rate": 1.638390092879257e-05,
-      "loss": 0.0158,
-      "step": 5854
-    },
-    {
-      "epoch": 344.41,
-      "learning_rate": 1.6377708978328173e-05,
-      "loss": 0.0164,
-      "step": 5855
-    },
-    {
-      "epoch": 344.47,
-      "learning_rate": 1.637151702786378e-05,
-      "loss": 0.0184,
-      "step": 5856
-    },
-    {
-      "epoch": 344.53,
-      "learning_rate": 1.6365325077399383e-05,
-      "loss": 0.0166,
-      "step": 5857
-    },
-    {
-      "epoch": 344.59,
-      "learning_rate": 1.6359133126934986e-05,
-      "loss": 0.0177,
-      "step": 5858
-    },
-    {
-      "epoch": 344.65,
-      "learning_rate": 1.635294117647059e-05,
-      "loss": 0.0155,
-      "step": 5859
-    },
-    {
-      "epoch": 344.71,
-      "learning_rate": 1.6346749226006193e-05,
-      "loss": 0.016,
-      "step": 5860
-    },
-    {
-      "epoch": 344.71,
-      "eval_accuracy_NAT": 0.9938710416706089,
-      "eval_accuracy_REST": 0.5382113821138211,
-      "eval_accuracy_concrete": 0.9608918391277295,
-      "eval_accuracy_concrete tiles": 0.948432712703276,
-      "eval_accuracy_dark bricks": 0.8589954635070727,
-      "eval_accuracy_grass": 0.9019414701351591,
-      "eval_accuracy_red bricks": 0.9553492970123023,
-      "eval_accuracy_road": 0.9816657857707429,
-      "eval_accuracy_rough concrete": 0.6573391570680316,
-      "eval_accuracy_rough red sidewalk": 0.766925604794742,
-      "eval_accuracy_speedway bricks": 0.9901651823016808,
-      "eval_accuracy_steel": 0.9313698541134606,
-      "eval_accuracy_tiles": 0.2521736496107122,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844215899510004,
-      "eval_iou_REST": 0.45655172413793105,
-      "eval_iou_concrete": 0.899582253184984,
-      "eval_iou_concrete tiles": 0.8921651221566975,
-      "eval_iou_dark bricks": 0.8422529395756501,
-      "eval_iou_grass": 0.8477609683645063,
-      "eval_iou_red bricks": 0.8980382034073309,
-      "eval_iou_road": 0.9638693427784337,
-      "eval_iou_rough concrete": 0.6500721883379004,
-      "eval_iou_rough red sidewalk": 0.6110435402599115,
-      "eval_iou_speedway bricks": 0.9804039709569309,
-      "eval_iou_steel": 0.8786351463488232,
-      "eval_iou_tiles": 0.2502254554199483,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07841598987579346,
-      "eval_mean_accuracy": 0.8259486492253341,
-      "eval_mean_iou": 0.7811555726830807,
-      "eval_overall_accuracy": 0.9810966556355104,
-      "eval_runtime": 5.5342,
-      "eval_samples_per_second": 21.322,
-      "eval_steps_per_second": 1.446,
-      "step": 5860
-    },
-    {
-      "epoch": 344.76,
-      "learning_rate": 1.6340557275541796e-05,
-      "loss": 0.0151,
-      "step": 5861
-    },
-    {
-      "epoch": 344.82,
-      "learning_rate": 1.63343653250774e-05,
-      "loss": 0.0164,
-      "step": 5862
-    },
-    {
-      "epoch": 344.88,
-      "learning_rate": 1.6328173374613003e-05,
-      "loss": 0.0145,
-      "step": 5863
-    },
-    {
-      "epoch": 344.94,
-      "learning_rate": 1.632198142414861e-05,
-      "loss": 0.0162,
-      "step": 5864
-    },
-    {
-      "epoch": 345.0,
-      "learning_rate": 1.6315789473684213e-05,
-      "loss": 0.0162,
-      "step": 5865
-    },
-    {
-      "epoch": 345.06,
-      "learning_rate": 1.6309597523219816e-05,
-      "loss": 0.0166,
-      "step": 5866
-    },
-    {
-      "epoch": 345.12,
-      "learning_rate": 1.6303405572755416e-05,
-      "loss": 0.0175,
-      "step": 5867
-    },
-    {
-      "epoch": 345.18,
-      "learning_rate": 1.6297213622291023e-05,
-      "loss": 0.0144,
-      "step": 5868
-    },
-    {
-      "epoch": 345.24,
-      "learning_rate": 1.6291021671826626e-05,
-      "loss": 0.0182,
-      "step": 5869
-    },
-    {
-      "epoch": 345.29,
-      "learning_rate": 1.628482972136223e-05,
-      "loss": 0.0159,
-      "step": 5870
-    },
-    {
-      "epoch": 345.35,
-      "learning_rate": 1.6278637770897833e-05,
-      "loss": 0.018,
-      "step": 5871
-    },
-    {
-      "epoch": 345.41,
-      "learning_rate": 1.627244582043344e-05,
-      "loss": 0.0167,
-      "step": 5872
-    },
-    {
-      "epoch": 345.47,
-      "learning_rate": 1.6266253869969043e-05,
-      "loss": 0.016,
-      "step": 5873
-    },
-    {
-      "epoch": 345.53,
-      "learning_rate": 1.6260061919504646e-05,
-      "loss": 0.0158,
-      "step": 5874
-    },
-    {
-      "epoch": 345.59,
-      "learning_rate": 1.6253869969040246e-05,
-      "loss": 0.0176,
-      "step": 5875
-    },
-    {
-      "epoch": 345.65,
-      "learning_rate": 1.6247678018575853e-05,
-      "loss": 0.0174,
-      "step": 5876
-    },
-    {
-      "epoch": 345.71,
-      "learning_rate": 1.6241486068111456e-05,
-      "loss": 0.016,
-      "step": 5877
-    },
-    {
-      "epoch": 345.76,
-      "learning_rate": 1.623529411764706e-05,
-      "loss": 0.0159,
-      "step": 5878
-    },
-    {
-      "epoch": 345.82,
-      "learning_rate": 1.6229102167182663e-05,
-      "loss": 0.0178,
-      "step": 5879
-    },
-    {
-      "epoch": 345.88,
-      "learning_rate": 1.622291021671827e-05,
-      "loss": 0.0149,
-      "step": 5880
-    },
-    {
-      "epoch": 345.88,
-      "eval_accuracy_NAT": 0.9938042989128856,
-      "eval_accuracy_REST": 0.4585365853658537,
-      "eval_accuracy_concrete": 0.9576404697076482,
-      "eval_accuracy_concrete tiles": 0.9470657553617723,
-      "eval_accuracy_dark bricks": 0.849754991099286,
-      "eval_accuracy_grass": 0.9066991033545667,
-      "eval_accuracy_red bricks": 0.9416190685413005,
-      "eval_accuracy_road": 0.9834357719988392,
-      "eval_accuracy_rough concrete": 0.6571246402373986,
-      "eval_accuracy_rough red sidewalk": 0.7811759751539716,
-      "eval_accuracy_speedway bricks": 0.99071560718445,
-      "eval_accuracy_steel": 0.9308075986945679,
-      "eval_accuracy_tiles": 0.21833439365021662,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984587395208794,
-      "eval_iou_REST": 0.41654357459379615,
-      "eval_iou_concrete": 0.8996466552894529,
-      "eval_iou_concrete tiles": 0.8921054968475268,
-      "eval_iou_dark bricks": 0.8341906600711229,
-      "eval_iou_grass": 0.8499400835371203,
-      "eval_iou_red bricks": 0.9058966501109584,
-      "eval_iou_road": 0.9626748903379709,
-      "eval_iou_rough concrete": 0.6509033602285405,
-      "eval_iou_rough red sidewalk": 0.6134295483131869,
-      "eval_iou_speedway bricks": 0.9802415967188015,
-      "eval_iou_steel": 0.8787365500711275,
-      "eval_iou_tiles": 0.21739261583011582,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07938756048679352,
-      "eval_mean_accuracy": 0.8166703276355965,
-      "eval_mean_iou": 0.775868390550655,
-      "eval_overall_accuracy": 0.9810978194414559,
-      "eval_runtime": 5.5231,
-      "eval_samples_per_second": 21.365,
-      "eval_steps_per_second": 1.448,
-      "step": 5880
-    },
-    {
-      "epoch": 345.94,
-      "learning_rate": 1.6216718266253873e-05,
-      "loss": 0.0172,
-      "step": 5881
-    },
-    {
-      "epoch": 346.0,
-      "learning_rate": 1.6210526315789473e-05,
-      "loss": 0.0173,
-      "step": 5882
-    },
-    {
-      "epoch": 346.06,
-      "learning_rate": 1.6204334365325076e-05,
-      "loss": 0.0156,
-      "step": 5883
-    },
-    {
-      "epoch": 346.12,
-      "learning_rate": 1.6198142414860683e-05,
-      "loss": 0.0162,
-      "step": 5884
-    },
-    {
-      "epoch": 346.18,
-      "learning_rate": 1.6191950464396286e-05,
-      "loss": 0.0174,
-      "step": 5885
-    },
-    {
-      "epoch": 346.24,
-      "learning_rate": 1.618575851393189e-05,
-      "loss": 0.0164,
-      "step": 5886
-    },
-    {
-      "epoch": 346.29,
-      "learning_rate": 1.6179566563467493e-05,
-      "loss": 0.0185,
-      "step": 5887
-    },
-    {
-      "epoch": 346.35,
-      "learning_rate": 1.61733746130031e-05,
-      "loss": 0.0173,
-      "step": 5888
-    },
-    {
-      "epoch": 346.41,
-      "learning_rate": 1.6167182662538703e-05,
-      "loss": 0.0156,
-      "step": 5889
-    },
-    {
-      "epoch": 346.47,
-      "learning_rate": 1.6160990712074303e-05,
-      "loss": 0.0184,
-      "step": 5890
-    },
-    {
-      "epoch": 346.53,
-      "learning_rate": 1.6154798761609906e-05,
-      "loss": 0.0134,
-      "step": 5891
-    },
-    {
-      "epoch": 346.59,
-      "learning_rate": 1.6148606811145513e-05,
-      "loss": 0.0178,
-      "step": 5892
-    },
-    {
-      "epoch": 346.65,
-      "learning_rate": 1.6142414860681116e-05,
-      "loss": 0.0165,
-      "step": 5893
-    },
-    {
-      "epoch": 346.71,
-      "learning_rate": 1.613622291021672e-05,
-      "loss": 0.0146,
-      "step": 5894
-    },
-    {
-      "epoch": 346.76,
-      "learning_rate": 1.6130030959752323e-05,
-      "loss": 0.0178,
-      "step": 5895
-    },
-    {
-      "epoch": 346.82,
-      "learning_rate": 1.6123839009287926e-05,
-      "loss": 0.0191,
-      "step": 5896
-    },
-    {
-      "epoch": 346.88,
-      "learning_rate": 1.611764705882353e-05,
-      "loss": 0.0176,
-      "step": 5897
-    },
-    {
-      "epoch": 346.94,
-      "learning_rate": 1.6111455108359133e-05,
-      "loss": 0.018,
-      "step": 5898
-    },
-    {
-      "epoch": 347.0,
-      "learning_rate": 1.6105263157894736e-05,
-      "loss": 0.0154,
-      "step": 5899
-    },
-    {
-      "epoch": 347.06,
-      "learning_rate": 1.6099071207430343e-05,
-      "loss": 0.0163,
-      "step": 5900
-    },
-    {
-      "epoch": 347.06,
-      "eval_accuracy_NAT": 0.9938815005518677,
-      "eval_accuracy_REST": 0.5544715447154471,
-      "eval_accuracy_concrete": 0.957911449009263,
-      "eval_accuracy_concrete tiles": 0.9385576243224134,
-      "eval_accuracy_dark bricks": 0.8514872805926152,
-      "eval_accuracy_grass": 0.9033645966988005,
-      "eval_accuracy_red bricks": 0.9402460456942003,
-      "eval_accuracy_road": 0.9834277557568641,
-      "eval_accuracy_rough concrete": 0.6743932451033554,
-      "eval_accuracy_rough red sidewalk": 0.7659780867655795,
-      "eval_accuracy_speedway bricks": 0.9910924391636557,
-      "eval_accuracy_steel": 0.9261246357834342,
-      "eval_accuracy_tiles": 0.23084613287285285,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844573928938272,
-      "eval_iou_REST": 0.4703448275862069,
-      "eval_iou_concrete": 0.8999645589092331,
-      "eval_iou_concrete tiles": 0.8884104852203012,
-      "eval_iou_dark bricks": 0.8359933284784703,
-      "eval_iou_grass": 0.8485310926281772,
-      "eval_iou_red bricks": 0.9029059648752703,
-      "eval_iou_road": 0.9633710928624752,
-      "eval_iou_rough concrete": 0.6664742981986067,
-      "eval_iou_rough red sidewalk": 0.6085966767642367,
-      "eval_iou_speedway bricks": 0.9799978175247341,
-      "eval_iou_steel": 0.8791396737364224,
-      "eval_iou_tiles": 0.22944896115627822,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07891778647899628,
-      "eval_mean_accuracy": 0.8239832566946425,
-      "eval_mean_iou": 0.7813566285257109,
-      "eval_overall_accuracy": 0.981105966083074,
-      "eval_runtime": 6.5794,
-      "eval_samples_per_second": 17.935,
-      "eval_steps_per_second": 1.216,
-      "step": 5900
-    },
-    {
-      "epoch": 347.12,
-      "learning_rate": 1.6092879256965946e-05,
-      "loss": 0.0166,
-      "step": 5901
-    },
-    {
-      "epoch": 347.18,
-      "learning_rate": 1.608668730650155e-05,
-      "loss": 0.0151,
-      "step": 5902
-    },
-    {
-      "epoch": 347.24,
-      "learning_rate": 1.6080495356037153e-05,
-      "loss": 0.0215,
-      "step": 5903
-    },
-    {
-      "epoch": 347.29,
-      "learning_rate": 1.6074303405572756e-05,
-      "loss": 0.0165,
-      "step": 5904
-    },
-    {
-      "epoch": 347.35,
-      "learning_rate": 1.606811145510836e-05,
-      "loss": 0.0159,
-      "step": 5905
-    },
-    {
-      "epoch": 347.41,
-      "learning_rate": 1.6061919504643963e-05,
-      "loss": 0.018,
-      "step": 5906
-    },
-    {
-      "epoch": 347.47,
-      "learning_rate": 1.6055727554179566e-05,
-      "loss": 0.0169,
-      "step": 5907
-    },
-    {
-      "epoch": 347.53,
-      "learning_rate": 1.6049535603715173e-05,
-      "loss": 0.0154,
-      "step": 5908
-    },
-    {
-      "epoch": 347.59,
-      "learning_rate": 1.6043343653250776e-05,
-      "loss": 0.0164,
-      "step": 5909
-    },
-    {
-      "epoch": 347.65,
-      "learning_rate": 1.603715170278638e-05,
-      "loss": 0.0153,
-      "step": 5910
-    },
-    {
-      "epoch": 347.71,
-      "learning_rate": 1.6030959752321983e-05,
-      "loss": 0.0163,
-      "step": 5911
-    },
-    {
-      "epoch": 347.76,
-      "learning_rate": 1.6024767801857586e-05,
-      "loss": 0.0206,
-      "step": 5912
-    },
-    {
-      "epoch": 347.82,
-      "learning_rate": 1.601857585139319e-05,
-      "loss": 0.0171,
-      "step": 5913
-    },
-    {
-      "epoch": 347.88,
-      "learning_rate": 1.6012383900928793e-05,
-      "loss": 0.0159,
-      "step": 5914
-    },
-    {
-      "epoch": 347.94,
-      "learning_rate": 1.6006191950464396e-05,
-      "loss": 0.0167,
-      "step": 5915
-    },
-    {
-      "epoch": 348.0,
-      "learning_rate": 1.6000000000000003e-05,
-      "loss": 0.0181,
-      "step": 5916
-    },
-    {
-      "epoch": 348.06,
-      "learning_rate": 1.5993808049535606e-05,
-      "loss": 0.0151,
-      "step": 5917
-    },
-    {
-      "epoch": 348.12,
-      "learning_rate": 1.598761609907121e-05,
-      "loss": 0.0158,
-      "step": 5918
-    },
-    {
-      "epoch": 348.18,
-      "learning_rate": 1.598142414860681e-05,
-      "loss": 0.0141,
-      "step": 5919
-    },
-    {
-      "epoch": 348.24,
-      "learning_rate": 1.5975232198142416e-05,
-      "loss": 0.0145,
-      "step": 5920
-    },
-    {
-      "epoch": 348.24,
-      "eval_accuracy_NAT": 0.9945387927183975,
-      "eval_accuracy_REST": 0.4894308943089431,
-      "eval_accuracy_concrete": 0.9595828065353023,
-      "eval_accuracy_concrete tiles": 0.9455102521800612,
-      "eval_accuracy_dark bricks": 0.8508460463602779,
-      "eval_accuracy_grass": 0.8964353707509642,
-      "eval_accuracy_red bricks": 0.9311840949033392,
-      "eval_accuracy_road": 0.9812029814007154,
-      "eval_accuracy_rough concrete": 0.6515591202426424,
-      "eval_accuracy_rough red sidewalk": 0.7579693034238488,
-      "eval_accuracy_speedway bricks": 0.9898305061509809,
-      "eval_accuracy_steel": 0.9302753302313493,
-      "eval_accuracy_tiles": 0.2285740252658366,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9842732097888166,
-      "eval_iou_REST": 0.45606060606060606,
-      "eval_iou_concrete": 0.8992080896555257,
-      "eval_iou_concrete tiles": 0.886350581061376,
-      "eval_iou_dark bricks": 0.8359843337862681,
-      "eval_iou_grass": 0.8456942593397647,
-      "eval_iou_red bricks": 0.9037846481876333,
-      "eval_iou_road": 0.9636202176614826,
-      "eval_iou_rough concrete": 0.6468222043443282,
-      "eval_iou_rough red sidewalk": 0.6022550056464768,
-      "eval_iou_speedway bricks": 0.9801574116905212,
-      "eval_iou_steel": 0.8799612343548712,
-      "eval_iou_tiles": 0.2273275082856282,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08130298554897308,
-      "eval_mean_accuracy": 0.8159184249594352,
-      "eval_mean_iou": 0.7778076392202536,
-      "eval_overall_accuracy": 0.9809236364849543,
-      "eval_runtime": 5.5947,
-      "eval_samples_per_second": 21.091,
-      "eval_steps_per_second": 1.43,
-      "step": 5920
-    },
-    {
-      "epoch": 348.29,
-      "learning_rate": 1.596904024767802e-05,
-      "loss": 0.0165,
-      "step": 5921
-    },
-    {
-      "epoch": 348.35,
-      "learning_rate": 1.5962848297213623e-05,
-      "loss": 0.0178,
-      "step": 5922
-    },
-    {
-      "epoch": 348.41,
-      "learning_rate": 1.5956656346749226e-05,
-      "loss": 0.0156,
-      "step": 5923
-    },
-    {
-      "epoch": 348.47,
-      "learning_rate": 1.5950464396284833e-05,
-      "loss": 0.0205,
-      "step": 5924
-    },
-    {
-      "epoch": 348.53,
-      "learning_rate": 1.5944272445820436e-05,
-      "loss": 0.0157,
-      "step": 5925
-    },
-    {
-      "epoch": 348.59,
-      "learning_rate": 1.593808049535604e-05,
-      "loss": 0.0188,
-      "step": 5926
-    },
-    {
-      "epoch": 348.65,
-      "learning_rate": 1.593188854489164e-05,
-      "loss": 0.0189,
-      "step": 5927
-    },
-    {
-      "epoch": 348.71,
-      "learning_rate": 1.5925696594427246e-05,
-      "loss": 0.0168,
-      "step": 5928
-    },
-    {
-      "epoch": 348.76,
-      "learning_rate": 1.591950464396285e-05,
-      "loss": 0.0175,
-      "step": 5929
-    },
-    {
-      "epoch": 348.82,
-      "learning_rate": 1.5913312693498453e-05,
-      "loss": 0.0151,
-      "step": 5930
-    },
-    {
-      "epoch": 348.88,
-      "learning_rate": 1.5907120743034056e-05,
-      "loss": 0.0168,
-      "step": 5931
-    },
-    {
-      "epoch": 348.94,
-      "learning_rate": 1.5900928792569663e-05,
-      "loss": 0.0233,
-      "step": 5932
-    },
-    {
-      "epoch": 349.0,
-      "learning_rate": 1.5894736842105266e-05,
-      "loss": 0.0126,
-      "step": 5933
-    },
-    {
-      "epoch": 349.06,
-      "learning_rate": 1.5888544891640866e-05,
-      "loss": 0.019,
-      "step": 5934
-    },
-    {
-      "epoch": 349.12,
-      "learning_rate": 1.588235294117647e-05,
-      "loss": 0.0177,
-      "step": 5935
-    },
-    {
-      "epoch": 349.18,
-      "learning_rate": 1.5876160990712076e-05,
-      "loss": 0.0155,
-      "step": 5936
-    },
-    {
-      "epoch": 349.24,
-      "learning_rate": 1.586996904024768e-05,
-      "loss": 0.0169,
-      "step": 5937
-    },
-    {
-      "epoch": 349.29,
-      "learning_rate": 1.5863777089783283e-05,
-      "loss": 0.0207,
-      "step": 5938
-    },
-    {
-      "epoch": 349.35,
-      "learning_rate": 1.5857585139318886e-05,
-      "loss": 0.0177,
-      "step": 5939
-    },
-    {
-      "epoch": 349.41,
-      "learning_rate": 1.585139318885449e-05,
-      "loss": 0.0131,
-      "step": 5940
-    },
-    {
-      "epoch": 349.41,
-      "eval_accuracy_NAT": 0.9942268053686839,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9605172838166116,
-      "eval_accuracy_concrete tiles": 0.9415507895357059,
-      "eval_accuracy_dark bricks": 0.8539947935608598,
-      "eval_accuracy_grass": 0.8981764298447807,
-      "eval_accuracy_red bricks": 0.9347539543057997,
-      "eval_accuracy_road": 0.9812596295106726,
-      "eval_accuracy_rough concrete": 0.6388192040233823,
-      "eval_accuracy_rough red sidewalk": 0.7537881921205604,
-      "eval_accuracy_speedway bricks": 0.9905111685527116,
-      "eval_accuracy_steel": 0.9284911063687296,
-      "eval_accuracy_tiles": 0.24405465176164076,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844368517823776,
-      "eval_iou_REST": 0.46497764530551416,
-      "eval_iou_concrete": 0.8984205359638906,
-      "eval_iou_concrete tiles": 0.8903102157247281,
-      "eval_iou_dark bricks": 0.8375108524767112,
-      "eval_iou_grass": 0.8468318010892983,
-      "eval_iou_red bricks": 0.9083146547123493,
-      "eval_iou_road": 0.9627395596784181,
-      "eval_iou_rough concrete": 0.6338711870820498,
-      "eval_iou_rough red sidewalk": 0.6020481095528394,
-      "eval_iou_speedway bricks": 0.9804510931293856,
-      "eval_iou_steel": 0.8792690545605998,
-      "eval_iou_tiles": 0.24243153776707793,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08120648562908173,
-      "eval_mean_accuracy": 0.8174970063031439,
-      "eval_mean_iou": 0.7793548537557877,
-      "eval_overall_accuracy": 0.9809315568309719,
-      "eval_runtime": 8.3959,
-      "eval_samples_per_second": 14.055,
-      "eval_steps_per_second": 0.953,
-      "step": 5940
-    },
-    {
-      "epoch": 349.47,
-      "learning_rate": 1.5845201238390096e-05,
-      "loss": 0.0166,
-      "step": 5941
-    },
-    {
-      "epoch": 349.53,
-      "learning_rate": 1.5839009287925696e-05,
-      "loss": 0.014,
-      "step": 5942
-    },
-    {
-      "epoch": 349.59,
-      "learning_rate": 1.58328173374613e-05,
-      "loss": 0.0173,
-      "step": 5943
-    },
-    {
-      "epoch": 349.65,
-      "learning_rate": 1.5826625386996906e-05,
-      "loss": 0.0144,
-      "step": 5944
-    },
-    {
-      "epoch": 349.71,
-      "learning_rate": 1.582043343653251e-05,
-      "loss": 0.0162,
-      "step": 5945
-    },
-    {
-      "epoch": 349.76,
-      "learning_rate": 1.5814241486068113e-05,
-      "loss": 0.0174,
-      "step": 5946
-    },
-    {
-      "epoch": 349.82,
-      "learning_rate": 1.5808049535603716e-05,
-      "loss": 0.0178,
-      "step": 5947
-    },
-    {
-      "epoch": 349.88,
-      "learning_rate": 1.580185758513932e-05,
-      "loss": 0.0173,
-      "step": 5948
-    },
-    {
-      "epoch": 349.94,
-      "learning_rate": 1.5795665634674923e-05,
-      "loss": 0.0187,
-      "step": 5949
-    },
-    {
-      "epoch": 350.0,
-      "learning_rate": 1.5789473684210526e-05,
-      "loss": 0.0142,
-      "step": 5950
-    },
-    {
-      "epoch": 350.06,
-      "learning_rate": 1.578328173374613e-05,
-      "loss": 0.0135,
-      "step": 5951
-    },
-    {
-      "epoch": 350.12,
-      "learning_rate": 1.5777089783281736e-05,
-      "loss": 0.0169,
-      "step": 5952
-    },
-    {
-      "epoch": 350.18,
-      "learning_rate": 1.577089783281734e-05,
-      "loss": 0.0189,
-      "step": 5953
-    },
-    {
-      "epoch": 350.24,
-      "learning_rate": 1.5764705882352943e-05,
-      "loss": 0.0168,
-      "step": 5954
-    },
-    {
-      "epoch": 350.29,
-      "learning_rate": 1.5758513931888546e-05,
-      "loss": 0.0182,
-      "step": 5955
-    },
-    {
-      "epoch": 350.35,
-      "learning_rate": 1.575232198142415e-05,
-      "loss": 0.0168,
-      "step": 5956
-    },
-    {
-      "epoch": 350.41,
-      "learning_rate": 1.5746130030959752e-05,
-      "loss": 0.0164,
-      "step": 5957
-    },
-    {
-      "epoch": 350.47,
-      "learning_rate": 1.5739938080495356e-05,
-      "loss": 0.0168,
-      "step": 5958
-    },
-    {
-      "epoch": 350.53,
-      "learning_rate": 1.573374613003096e-05,
-      "loss": 0.0177,
-      "step": 5959
-    },
-    {
-      "epoch": 350.59,
-      "learning_rate": 1.5727554179566566e-05,
-      "loss": 0.0161,
-      "step": 5960
-    },
-    {
-      "epoch": 350.59,
-      "eval_accuracy_NAT": 0.9940582232980852,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9601549588829996,
-      "eval_accuracy_concrete tiles": 0.9368607117605467,
-      "eval_accuracy_dark bricks": 0.849405662002565,
-      "eval_accuracy_grass": 0.9036541423959245,
-      "eval_accuracy_red bricks": 0.9309644112478032,
-      "eval_accuracy_road": 0.9809721136318332,
-      "eval_accuracy_rough concrete": 0.6560163032791281,
-      "eval_accuracy_rough red sidewalk": 0.7679633626362057,
-      "eval_accuracy_speedway bricks": 0.9905084267102071,
-      "eval_accuracy_steel": 0.9308825660837535,
-      "eval_accuracy_tiles": 0.23772304523008878,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845931117202448,
-      "eval_iou_REST": 0.4777947932618683,
-      "eval_iou_concrete": 0.8992189681215861,
-      "eval_iou_concrete tiles": 0.8880325268637044,
-      "eval_iou_dark bricks": 0.8345494894025163,
-      "eval_iou_grass": 0.8493533844483201,
-      "eval_iou_red bricks": 0.9081217186328083,
-      "eval_iou_road": 0.9636974646973371,
-      "eval_iou_rough concrete": 0.649050819478835,
-      "eval_iou_rough red sidewalk": 0.6106228026117528,
-      "eval_iou_speedway bricks": 0.9802161015404209,
-      "eval_iou_steel": 0.879168210557618,
-      "eval_iou_tiles": 0.2360070979578333,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07980097830295563,
-      "eval_mean_accuracy": 0.8189600769484517,
-      "eval_mean_iou": 0.7815712684072958,
-      "eval_overall_accuracy": 0.9811078087758209,
-      "eval_runtime": 5.6494,
-      "eval_samples_per_second": 20.887,
-      "eval_steps_per_second": 1.416,
-      "step": 5960
-    },
-    {
-      "epoch": 350.65,
-      "learning_rate": 1.572136222910217e-05,
-      "loss": 0.0157,
-      "step": 5961
-    },
-    {
-      "epoch": 350.71,
-      "learning_rate": 1.5715170278637772e-05,
-      "loss": 0.0165,
-      "step": 5962
-    },
-    {
-      "epoch": 350.76,
-      "learning_rate": 1.5708978328173376e-05,
-      "loss": 0.019,
-      "step": 5963
-    },
-    {
-      "epoch": 350.82,
-      "learning_rate": 1.570278637770898e-05,
-      "loss": 0.0148,
-      "step": 5964
-    },
-    {
-      "epoch": 350.88,
-      "learning_rate": 1.5696594427244582e-05,
-      "loss": 0.0165,
-      "step": 5965
-    },
-    {
-      "epoch": 350.94,
-      "learning_rate": 1.5690402476780186e-05,
-      "loss": 0.0169,
-      "step": 5966
-    },
-    {
-      "epoch": 351.0,
-      "learning_rate": 1.568421052631579e-05,
-      "loss": 0.019,
-      "step": 5967
-    },
-    {
-      "epoch": 351.06,
-      "learning_rate": 1.5678018575851396e-05,
-      "loss": 0.0166,
-      "step": 5968
-    },
-    {
-      "epoch": 351.12,
-      "learning_rate": 1.5671826625387e-05,
-      "loss": 0.015,
-      "step": 5969
-    },
-    {
-      "epoch": 351.18,
-      "learning_rate": 1.5665634674922602e-05,
-      "loss": 0.0146,
-      "step": 5970
-    },
-    {
-      "epoch": 351.24,
-      "learning_rate": 1.5659442724458202e-05,
-      "loss": 0.0154,
-      "step": 5971
-    },
-    {
-      "epoch": 351.29,
-      "learning_rate": 1.565325077399381e-05,
-      "loss": 0.0166,
-      "step": 5972
-    },
-    {
-      "epoch": 351.35,
-      "learning_rate": 1.5647058823529412e-05,
-      "loss": 0.0171,
-      "step": 5973
-    },
-    {
-      "epoch": 351.41,
-      "learning_rate": 1.5640866873065016e-05,
-      "loss": 0.0184,
-      "step": 5974
-    },
-    {
-      "epoch": 351.47,
-      "learning_rate": 1.563467492260062e-05,
-      "loss": 0.0155,
-      "step": 5975
-    },
-    {
-      "epoch": 351.53,
-      "learning_rate": 1.5628482972136226e-05,
-      "loss": 0.0168,
-      "step": 5976
-    },
-    {
-      "epoch": 351.59,
-      "learning_rate": 1.562229102167183e-05,
-      "loss": 0.0149,
-      "step": 5977
-    },
-    {
-      "epoch": 351.65,
-      "learning_rate": 1.5616099071207432e-05,
-      "loss": 0.0173,
-      "step": 5978
-    },
-    {
-      "epoch": 351.71,
-      "learning_rate": 1.5609907120743032e-05,
-      "loss": 0.0155,
-      "step": 5979
-    },
-    {
-      "epoch": 351.76,
-      "learning_rate": 1.560371517027864e-05,
-      "loss": 0.0155,
-      "step": 5980
-    },
-    {
-      "epoch": 351.76,
-      "eval_accuracy_NAT": 0.9937214904509606,
-      "eval_accuracy_REST": 0.47479674796747967,
-      "eval_accuracy_concrete": 0.9589147221640429,
-      "eval_accuracy_concrete tiles": 0.9450860240395946,
-      "eval_accuracy_dark bricks": 0.8497980590701146,
-      "eval_accuracy_grass": 0.9104092624343791,
-      "eval_accuracy_red bricks": 0.9344244288224957,
-      "eval_accuracy_road": 0.9816930409934582,
-      "eval_accuracy_rough concrete": 0.6665276279801453,
-      "eval_accuracy_rough red sidewalk": 0.7652636882515285,
-      "eval_accuracy_speedway bricks": 0.9918200556182752,
-      "eval_accuracy_steel": 0.9207944544123307,
-      "eval_accuracy_tiles": 0.23578418007210156,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846967983251943,
-      "eval_iou_REST": 0.4485407066052227,
-      "eval_iou_concrete": 0.8998386061308151,
-      "eval_iou_concrete tiles": 0.8917452410603095,
-      "eval_iou_dark bricks": 0.8338686350744727,
-      "eval_iou_grass": 0.8525664825626988,
-      "eval_iou_red bricks": 0.911008781323624,
-      "eval_iou_road": 0.9647126230938216,
-      "eval_iou_rough concrete": 0.6596605451564582,
-      "eval_iou_rough red sidewalk": 0.6130952380952381,
-      "eval_iou_speedway bricks": 0.9797172446247026,
-      "eval_iou_steel": 0.8780581959771334,
-      "eval_iou_tiles": 0.23435006473758693,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0786697268486023,
-      "eval_mean_accuracy": 0.8176179832520698,
-      "eval_mean_iou": 0.7809122432897906,
-      "eval_overall_accuracy": 0.9812509892350536,
-      "eval_runtime": 6.1344,
-      "eval_samples_per_second": 19.236,
-      "eval_steps_per_second": 1.304,
-      "step": 5980
-    },
-    {
-      "epoch": 351.82,
-      "learning_rate": 1.5597523219814242e-05,
-      "loss": 0.0206,
-      "step": 5981
-    },
-    {
-      "epoch": 351.88,
-      "learning_rate": 1.5591331269349846e-05,
-      "loss": 0.0172,
-      "step": 5982
-    },
-    {
-      "epoch": 351.94,
-      "learning_rate": 1.558513931888545e-05,
-      "loss": 0.0169,
-      "step": 5983
-    },
-    {
-      "epoch": 352.0,
-      "learning_rate": 1.5578947368421056e-05,
-      "loss": 0.0182,
-      "step": 5984
-    },
-    {
-      "epoch": 352.06,
-      "learning_rate": 1.557275541795666e-05,
-      "loss": 0.0161,
-      "step": 5985
-    },
-    {
-      "epoch": 352.12,
-      "learning_rate": 1.556656346749226e-05,
-      "loss": 0.016,
-      "step": 5986
-    },
-    {
-      "epoch": 352.18,
-      "learning_rate": 1.5560371517027862e-05,
-      "loss": 0.0167,
-      "step": 5987
-    },
-    {
-      "epoch": 352.24,
-      "learning_rate": 1.555417956656347e-05,
-      "loss": 0.0159,
-      "step": 5988
-    },
-    {
-      "epoch": 352.29,
-      "learning_rate": 1.5547987616099072e-05,
-      "loss": 0.0182,
-      "step": 5989
-    },
-    {
-      "epoch": 352.35,
-      "learning_rate": 1.5541795665634676e-05,
-      "loss": 0.0169,
-      "step": 5990
-    },
-    {
-      "epoch": 352.41,
-      "learning_rate": 1.553560371517028e-05,
-      "loss": 0.0151,
-      "step": 5991
-    },
-    {
-      "epoch": 352.47,
-      "learning_rate": 1.5529411764705882e-05,
-      "loss": 0.0198,
-      "step": 5992
-    },
-    {
-      "epoch": 352.53,
-      "learning_rate": 1.552321981424149e-05,
-      "loss": 0.0138,
-      "step": 5993
-    },
-    {
-      "epoch": 352.59,
-      "learning_rate": 1.551702786377709e-05,
-      "loss": 0.0171,
-      "step": 5994
-    },
-    {
-      "epoch": 352.65,
-      "learning_rate": 1.5510835913312692e-05,
-      "loss": 0.0151,
-      "step": 5995
-    },
-    {
-      "epoch": 352.71,
-      "learning_rate": 1.55046439628483e-05,
-      "loss": 0.0165,
-      "step": 5996
-    },
-    {
-      "epoch": 352.76,
-      "learning_rate": 1.5498452012383902e-05,
-      "loss": 0.0166,
-      "step": 5997
-    },
-    {
-      "epoch": 352.82,
-      "learning_rate": 1.5492260061919506e-05,
-      "loss": 0.0165,
-      "step": 5998
-    },
-    {
-      "epoch": 352.88,
-      "learning_rate": 1.548606811145511e-05,
-      "loss": 0.0171,
-      "step": 5999
-    },
-    {
-      "epoch": 352.94,
-      "learning_rate": 1.5479876160990712e-05,
-      "loss": 0.0143,
-      "step": 6000
-    },
-    {
-      "epoch": 352.94,
-      "eval_accuracy_NAT": 0.9939089416372322,
-      "eval_accuracy_REST": 0.5138211382113821,
-      "eval_accuracy_concrete": 0.9601098593659466,
-      "eval_accuracy_concrete tiles": 0.9462880037709168,
-      "eval_accuracy_dark bricks": 0.8534492659303639,
-      "eval_accuracy_grass": 0.9068457686586654,
-      "eval_accuracy_red bricks": 0.9421682776801406,
-      "eval_accuracy_road": 0.982463134639197,
-      "eval_accuracy_rough concrete": 0.6399811701893111,
-      "eval_accuracy_rough red sidewalk": 0.7932004301431053,
-      "eval_accuracy_speedway bricks": 0.9891573838161375,
-      "eval_accuracy_steel": 0.9345509703279073,
-      "eval_accuracy_tiles": 0.2467811808900603,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845299455310536,
-      "eval_iou_REST": 0.4869029275808937,
-      "eval_iou_concrete": 0.898709641412826,
-      "eval_iou_concrete tiles": 0.8887486995595104,
-      "eval_iou_dark bricks": 0.8372626894259478,
-      "eval_iou_grass": 0.8502004896916362,
-      "eval_iou_red bricks": 0.9135204217476969,
-      "eval_iou_road": 0.9649605987168404,
-      "eval_iou_rough concrete": 0.635181888613554,
-      "eval_iou_rough red sidewalk": 0.6209797537957953,
-      "eval_iou_speedway bricks": 0.9801040684144976,
-      "eval_iou_steel": 0.8783809810129556,
-      "eval_iou_tiles": 0.24477163461538462,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08022807538509369,
-      "eval_mean_accuracy": 0.823286578866182,
-      "eval_mean_iou": 0.7834041338552763,
-      "eval_overall_accuracy": 0.9810844356730833,
-      "eval_runtime": 5.8576,
-      "eval_samples_per_second": 20.145,
-      "eval_steps_per_second": 1.366,
-      "step": 6000
-    },
-    {
-      "epoch": 353.0,
-      "learning_rate": 1.5473684210526316e-05,
-      "loss": 0.016,
-      "step": 6001
-    },
-    {
-      "epoch": 353.06,
-      "learning_rate": 1.546749226006192e-05,
-      "loss": 0.0155,
-      "step": 6002
-    },
-    {
-      "epoch": 353.12,
-      "learning_rate": 1.5461300309597522e-05,
-      "loss": 0.0154,
-      "step": 6003
-    },
-    {
-      "epoch": 353.18,
-      "learning_rate": 1.545510835913313e-05,
-      "loss": 0.0177,
-      "step": 6004
-    },
-    {
-      "epoch": 353.24,
-      "learning_rate": 1.5448916408668732e-05,
-      "loss": 0.0167,
-      "step": 6005
-    },
-    {
-      "epoch": 353.29,
-      "learning_rate": 1.5442724458204336e-05,
-      "loss": 0.0159,
-      "step": 6006
-    },
-    {
-      "epoch": 353.35,
-      "learning_rate": 1.543653250773994e-05,
-      "loss": 0.0183,
-      "step": 6007
-    },
-    {
-      "epoch": 353.41,
-      "learning_rate": 1.5430340557275542e-05,
-      "loss": 0.0167,
-      "step": 6008
-    },
-    {
-      "epoch": 353.47,
-      "learning_rate": 1.5424148606811146e-05,
-      "loss": 0.0168,
-      "step": 6009
-    },
-    {
-      "epoch": 353.53,
-      "learning_rate": 1.541795665634675e-05,
-      "loss": 0.0156,
-      "step": 6010
-    },
-    {
-      "epoch": 353.59,
-      "learning_rate": 1.5411764705882352e-05,
-      "loss": 0.0165,
-      "step": 6011
-    },
-    {
-      "epoch": 353.65,
-      "learning_rate": 1.540557275541796e-05,
-      "loss": 0.0165,
-      "step": 6012
-    },
-    {
-      "epoch": 353.71,
-      "learning_rate": 1.5399380804953562e-05,
-      "loss": 0.0158,
-      "step": 6013
-    },
-    {
-      "epoch": 353.76,
-      "learning_rate": 1.5393188854489166e-05,
-      "loss": 0.0143,
-      "step": 6014
-    },
-    {
-      "epoch": 353.82,
-      "learning_rate": 1.538699690402477e-05,
-      "loss": 0.017,
-      "step": 6015
-    },
-    {
-      "epoch": 353.88,
-      "learning_rate": 1.5380804953560372e-05,
-      "loss": 0.018,
-      "step": 6016
-    },
-    {
-      "epoch": 353.94,
-      "learning_rate": 1.5374613003095976e-05,
-      "loss": 0.015,
-      "step": 6017
-    },
-    {
-      "epoch": 354.0,
-      "learning_rate": 1.536842105263158e-05,
-      "loss": 0.0185,
-      "step": 6018
-    },
-    {
-      "epoch": 354.06,
-      "learning_rate": 1.5362229102167182e-05,
-      "loss": 0.0154,
-      "step": 6019
-    },
-    {
-      "epoch": 354.12,
-      "learning_rate": 1.535603715170279e-05,
-      "loss": 0.0142,
-      "step": 6020
-    },
-    {
-      "epoch": 354.12,
-      "eval_accuracy_NAT": 0.9942113866055912,
-      "eval_accuracy_REST": 0.5089430894308943,
-      "eval_accuracy_concrete": 0.9579806270820307,
-      "eval_accuracy_concrete tiles": 0.9447325005892058,
-      "eval_accuracy_dark bricks": 0.8528702409892235,
-      "eval_accuracy_grass": 0.899644029113536,
-      "eval_accuracy_red bricks": 0.9368409490333919,
-      "eval_accuracy_road": 0.9820174315853829,
-      "eval_accuracy_rough concrete": 0.6345288674107222,
-      "eval_accuracy_rough red sidewalk": 0.7824919724166973,
-      "eval_accuracy_speedway bricks": 0.9906496315991865,
-      "eval_accuracy_steel": 0.9291533183065367,
-      "eval_accuracy_tiles": 0.22699869732497197,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843982009569153,
-      "eval_iou_REST": 0.4807987711213518,
-      "eval_iou_concrete": 0.8980148333071195,
-      "eval_iou_concrete tiles": 0.8900262000976953,
-      "eval_iou_dark bricks": 0.8374455528354815,
-      "eval_iou_grass": 0.8472653681019318,
-      "eval_iou_red bricks": 0.9101483299541138,
-      "eval_iou_road": 0.9634305230270419,
-      "eval_iou_rough concrete": 0.6287256152283784,
-      "eval_iou_rough red sidewalk": 0.6151495087316883,
-      "eval_iou_speedway bricks": 0.9802810309612249,
-      "eval_iou_steel": 0.8796796631021103,
-      "eval_iou_tiles": 0.22552973753912833,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08295805007219315,
-      "eval_mean_accuracy": 0.8185432878067208,
-      "eval_mean_iou": 0.7800687180741678,
-      "eval_overall_accuracy": 0.9809183670302569,
-      "eval_runtime": 5.7997,
-      "eval_samples_per_second": 20.346,
-      "eval_steps_per_second": 1.379,
-      "step": 6020
-    },
-    {
-      "epoch": 354.18,
-      "learning_rate": 1.5349845201238392e-05,
-      "loss": 0.0183,
-      "step": 6021
-    },
-    {
-      "epoch": 354.24,
-      "learning_rate": 1.5343653250773996e-05,
-      "loss": 0.0184,
-      "step": 6022
-    },
-    {
-      "epoch": 354.29,
-      "learning_rate": 1.53374613003096e-05,
-      "loss": 0.0189,
-      "step": 6023
-    },
-    {
-      "epoch": 354.35,
-      "learning_rate": 1.5331269349845202e-05,
-      "loss": 0.0202,
-      "step": 6024
-    },
-    {
-      "epoch": 354.41,
-      "learning_rate": 1.5325077399380806e-05,
-      "loss": 0.0147,
-      "step": 6025
-    },
-    {
-      "epoch": 354.47,
-      "learning_rate": 1.531888544891641e-05,
-      "loss": 0.0195,
-      "step": 6026
-    },
-    {
-      "epoch": 354.53,
-      "learning_rate": 1.5312693498452012e-05,
-      "loss": 0.0156,
-      "step": 6027
-    },
-    {
-      "epoch": 354.59,
-      "learning_rate": 1.530650154798762e-05,
-      "loss": 0.0169,
-      "step": 6028
-    },
-    {
-      "epoch": 354.65,
-      "learning_rate": 1.5300309597523222e-05,
-      "loss": 0.0137,
-      "step": 6029
-    },
-    {
-      "epoch": 354.71,
-      "learning_rate": 1.5294117647058826e-05,
-      "loss": 0.0148,
-      "step": 6030
-    },
-    {
-      "epoch": 354.76,
-      "learning_rate": 1.5287925696594426e-05,
-      "loss": 0.0153,
-      "step": 6031
-    },
-    {
-      "epoch": 354.82,
-      "learning_rate": 1.5281733746130032e-05,
-      "loss": 0.0148,
-      "step": 6032
-    },
-    {
-      "epoch": 354.88,
-      "learning_rate": 1.5275541795665636e-05,
-      "loss": 0.0165,
-      "step": 6033
-    },
-    {
-      "epoch": 354.94,
-      "learning_rate": 1.526934984520124e-05,
-      "loss": 0.0183,
-      "step": 6034
-    },
-    {
-      "epoch": 355.0,
-      "learning_rate": 1.5263157894736842e-05,
-      "loss": 0.0152,
-      "step": 6035
-    },
-    {
-      "epoch": 355.06,
-      "learning_rate": 1.5256965944272447e-05,
-      "loss": 0.019,
-      "step": 6036
-    },
-    {
-      "epoch": 355.12,
-      "learning_rate": 1.525077399380805e-05,
-      "loss": 0.0188,
-      "step": 6037
-    },
-    {
-      "epoch": 355.18,
-      "learning_rate": 1.5244582043343656e-05,
-      "loss": 0.0167,
-      "step": 6038
-    },
-    {
-      "epoch": 355.24,
-      "learning_rate": 1.5238390092879257e-05,
-      "loss": 0.0163,
-      "step": 6039
-    },
-    {
-      "epoch": 355.29,
-      "learning_rate": 1.523219814241486e-05,
-      "loss": 0.0176,
-      "step": 6040
-    },
-    {
-      "epoch": 355.29,
-      "eval_accuracy_NAT": 0.9937903896790465,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9582313498209014,
-      "eval_accuracy_concrete tiles": 0.9403959462644356,
-      "eval_accuracy_dark bricks": 0.853669391114599,
-      "eval_accuracy_grass": 0.908601967396776,
-      "eval_accuracy_red bricks": 0.9336006151142355,
-      "eval_accuracy_road": 0.9827089660597659,
-      "eval_accuracy_rough concrete": 0.6323121934941812,
-      "eval_accuracy_rough red sidewalk": 0.7619548951338181,
-      "eval_accuracy_speedway bricks": 0.9904364533444652,
-      "eval_accuracy_steel": 0.9291158346119438,
-      "eval_accuracy_tiles": 0.24075252203944378,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845166656688147,
-      "eval_iou_REST": 0.4369747899159664,
-      "eval_iou_concrete": 0.8971521874953033,
-      "eval_iou_concrete tiles": 0.8906075621624034,
-      "eval_iou_dark bricks": 0.8372349512375982,
-      "eval_iou_grass": 0.850433924682472,
-      "eval_iou_red bricks": 0.9081632653061225,
-      "eval_iou_road": 0.9625162984563379,
-      "eval_iou_rough concrete": 0.6268401030221403,
-      "eval_iou_rough red sidewalk": 0.6081580716412177,
-      "eval_iou_speedway bricks": 0.9801979022618031,
-      "eval_iou_steel": 0.8788229037404716,
-      "eval_iou_tiles": 0.2392737783398067,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08224144577980042,
-      "eval_mean_accuracy": 0.8179144305572573,
-      "eval_mean_iou": 0.7769917233792659,
-      "eval_overall_accuracy": 0.9809061793957726,
-      "eval_runtime": 6.9577,
-      "eval_samples_per_second": 16.96,
-      "eval_steps_per_second": 1.15,
-      "step": 6040
-    },
-    {
-      "epoch": 355.35,
-      "learning_rate": 1.5226006191950466e-05,
-      "loss": 0.017,
-      "step": 6041
-    },
-    {
-      "epoch": 355.41,
-      "learning_rate": 1.5219814241486069e-05,
-      "loss": 0.0178,
-      "step": 6042
-    },
-    {
-      "epoch": 355.47,
-      "learning_rate": 1.5213622291021674e-05,
-      "loss": 0.0153,
-      "step": 6043
-    },
-    {
-      "epoch": 355.53,
-      "learning_rate": 1.5207430340557277e-05,
-      "loss": 0.0127,
-      "step": 6044
-    },
-    {
-      "epoch": 355.59,
-      "learning_rate": 1.520123839009288e-05,
-      "loss": 0.0185,
-      "step": 6045
-    },
-    {
-      "epoch": 355.65,
-      "learning_rate": 1.5195046439628482e-05,
-      "loss": 0.015,
-      "step": 6046
-    },
-    {
-      "epoch": 355.71,
-      "learning_rate": 1.5188854489164087e-05,
-      "loss": 0.019,
-      "step": 6047
-    },
-    {
-      "epoch": 355.76,
-      "learning_rate": 1.518266253869969e-05,
-      "loss": 0.015,
-      "step": 6048
-    },
-    {
-      "epoch": 355.82,
-      "learning_rate": 1.5176470588235295e-05,
-      "loss": 0.018,
-      "step": 6049
-    },
-    {
-      "epoch": 355.88,
-      "learning_rate": 1.5170278637770899e-05,
-      "loss": 0.0184,
-      "step": 6050
-    },
-    {
-      "epoch": 355.94,
-      "learning_rate": 1.5164086687306502e-05,
-      "loss": 0.016,
-      "step": 6051
-    },
-    {
-      "epoch": 356.0,
-      "learning_rate": 1.5157894736842107e-05,
-      "loss": 0.015,
-      "step": 6052
-    },
-    {
-      "epoch": 356.06,
-      "learning_rate": 1.5151702786377709e-05,
-      "loss": 0.0149,
-      "step": 6053
-    },
-    {
-      "epoch": 356.12,
-      "learning_rate": 1.5145510835913312e-05,
-      "loss": 0.0145,
-      "step": 6054
-    },
-    {
-      "epoch": 356.18,
-      "learning_rate": 1.5139318885448917e-05,
-      "loss": 0.0163,
-      "step": 6055
-    },
-    {
-      "epoch": 356.24,
-      "learning_rate": 1.513312693498452e-05,
-      "loss": 0.0168,
-      "step": 6056
-    },
-    {
-      "epoch": 356.29,
-      "learning_rate": 1.5126934984520125e-05,
-      "loss": 0.0171,
-      "step": 6057
-    },
-    {
-      "epoch": 356.35,
-      "learning_rate": 1.5120743034055729e-05,
-      "loss": 0.0163,
-      "step": 6058
-    },
-    {
-      "epoch": 356.41,
-      "learning_rate": 1.5114551083591332e-05,
-      "loss": 0.0176,
-      "step": 6059
-    },
-    {
-      "epoch": 356.47,
-      "learning_rate": 1.5108359133126937e-05,
-      "loss": 0.0155,
-      "step": 6060
-    },
-    {
-      "epoch": 356.47,
-      "eval_accuracy_NAT": 0.9940271701248634,
-      "eval_accuracy_REST": 0.591869918699187,
-      "eval_accuracy_concrete": 0.9605895194837558,
-      "eval_accuracy_concrete tiles": 0.9489040773037944,
-      "eval_accuracy_dark bricks": 0.8455582566085409,
-      "eval_accuracy_grass": 0.9041546968853966,
-      "eval_accuracy_red bricks": 0.9318431458699473,
-      "eval_accuracy_road": 0.9803008869704537,
-      "eval_accuracy_rough concrete": 0.6369302641536417,
-      "eval_accuracy_rough red sidewalk": 0.7700087983816993,
-      "eval_accuracy_speedway bricks": 0.9900627059380769,
-      "eval_accuracy_steel": 0.928448624848191,
-      "eval_accuracy_tiles": 0.2431155139507407,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844559557827639,
-      "eval_iou_REST": 0.5170454545454546,
-      "eval_iou_concrete": 0.8967329621982386,
-      "eval_iou_concrete tiles": 0.8927272727272727,
-      "eval_iou_dark bricks": 0.8330213985677716,
-      "eval_iou_grass": 0.8493170166319427,
-      "eval_iou_red bricks": 0.9054861778204718,
-      "eval_iou_road": 0.963149303001329,
-      "eval_iou_rough concrete": 0.6313212450534522,
-      "eval_iou_rough red sidewalk": 0.6089177504623601,
-      "eval_iou_speedway bricks": 0.980414460134273,
-      "eval_iou_steel": 0.8796993015271695,
-      "eval_iou_tiles": 0.2417241483177204,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08218328654766083,
-      "eval_mean_accuracy": 0.8250625830167915,
-      "eval_mean_iou": 0.7833855728284784,
-      "eval_overall_accuracy": 0.9808905973272809,
-      "eval_runtime": 5.4994,
-      "eval_samples_per_second": 21.457,
-      "eval_steps_per_second": 1.455,
-      "step": 6060
-    },
-    {
-      "epoch": 356.53,
-      "learning_rate": 1.5102167182662539e-05,
-      "loss": 0.0171,
-      "step": 6061
-    },
-    {
-      "epoch": 356.59,
-      "learning_rate": 1.5095975232198142e-05,
-      "loss": 0.0161,
-      "step": 6062
-    },
-    {
-      "epoch": 356.65,
-      "learning_rate": 1.5089783281733747e-05,
-      "loss": 0.0153,
-      "step": 6063
-    },
-    {
-      "epoch": 356.71,
-      "learning_rate": 1.508359133126935e-05,
-      "loss": 0.0156,
-      "step": 6064
-    },
-    {
-      "epoch": 356.76,
-      "learning_rate": 1.5077399380804955e-05,
-      "loss": 0.0174,
-      "step": 6065
-    },
-    {
-      "epoch": 356.82,
-      "learning_rate": 1.5071207430340559e-05,
-      "loss": 0.0176,
-      "step": 6066
-    },
-    {
-      "epoch": 356.88,
-      "learning_rate": 1.5065015479876162e-05,
-      "loss": 0.0168,
-      "step": 6067
-    },
-    {
-      "epoch": 356.94,
-      "learning_rate": 1.5058823529411764e-05,
-      "loss": 0.0187,
-      "step": 6068
-    },
-    {
-      "epoch": 357.0,
-      "learning_rate": 1.5052631578947369e-05,
-      "loss": 0.0193,
-      "step": 6069
-    },
-    {
-      "epoch": 357.06,
-      "learning_rate": 1.5046439628482972e-05,
-      "loss": 0.0192,
-      "step": 6070
-    },
-    {
-      "epoch": 357.12,
-      "learning_rate": 1.5040247678018577e-05,
-      "loss": 0.0169,
-      "step": 6071
-    },
-    {
-      "epoch": 357.18,
-      "learning_rate": 1.503405572755418e-05,
-      "loss": 0.0183,
-      "step": 6072
-    },
-    {
-      "epoch": 357.24,
-      "learning_rate": 1.5027863777089784e-05,
-      "loss": 0.0161,
-      "step": 6073
-    },
-    {
-      "epoch": 357.29,
-      "learning_rate": 1.5021671826625389e-05,
-      "loss": 0.0137,
-      "step": 6074
-    },
-    {
-      "epoch": 357.35,
-      "learning_rate": 1.5015479876160992e-05,
-      "loss": 0.0179,
-      "step": 6075
-    },
-    {
-      "epoch": 357.41,
-      "learning_rate": 1.5009287925696594e-05,
-      "loss": 0.0164,
-      "step": 6076
-    },
-    {
-      "epoch": 357.47,
-      "learning_rate": 1.5003095975232199e-05,
-      "loss": 0.0141,
-      "step": 6077
-    },
-    {
-      "epoch": 357.53,
-      "learning_rate": 1.4996904024767802e-05,
-      "loss": 0.0158,
-      "step": 6078
-    },
-    {
-      "epoch": 357.59,
-      "learning_rate": 1.4990712074303407e-05,
-      "loss": 0.0151,
-      "step": 6079
-    },
-    {
-      "epoch": 357.65,
-      "learning_rate": 1.498452012383901e-05,
-      "loss": 0.0157,
-      "step": 6080
-    },
-    {
-      "epoch": 357.65,
-      "eval_accuracy_NAT": 0.9940560668277226,
-      "eval_accuracy_REST": 0.5024390243902439,
-      "eval_accuracy_concrete": 0.9603766344752939,
-      "eval_accuracy_concrete tiles": 0.9453688427999057,
-      "eval_accuracy_dark bricks": 0.8563635319564343,
-      "eval_accuracy_grass": 0.8985804691018785,
-      "eval_accuracy_red bricks": 0.9379942882249561,
-      "eval_accuracy_road": 0.9814140757727257,
-      "eval_accuracy_rough concrete": 0.6333728600456444,
-      "eval_accuracy_rough red sidewalk": 0.7665646455455373,
-      "eval_accuracy_speedway bricks": 0.9901823188173337,
-      "eval_accuracy_steel": 0.9302703324054037,
-      "eval_accuracy_tiles": 0.226120149050259,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844918807318778,
-      "eval_iou_REST": 0.46396396396396394,
-      "eval_iou_concrete": 0.8965230334937926,
-      "eval_iou_concrete tiles": 0.8907443595665304,
-      "eval_iou_dark bricks": 0.8363328768979844,
-      "eval_iou_grass": 0.8474660217567845,
-      "eval_iou_red bricks": 0.9050874403815581,
-      "eval_iou_road": 0.9627583278286308,
-      "eval_iou_rough concrete": 0.6285339215071758,
-      "eval_iou_rough red sidewalk": 0.6066198130218221,
-      "eval_iou_speedway bricks": 0.9803078310603922,
-      "eval_iou_steel": 0.8801437471185559,
-      "eval_iou_tiles": 0.22498191463708705,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08230876922607422,
-      "eval_mean_accuracy": 0.81716178764718,
-      "eval_mean_iou": 0.7775350101512429,
-      "eval_overall_accuracy": 0.9808306936490334,
-      "eval_runtime": 6.7645,
-      "eval_samples_per_second": 17.444,
-      "eval_steps_per_second": 1.183,
-      "step": 6080
-    },
-    {
-      "epoch": 357.71,
-      "learning_rate": 1.4978328173374614e-05,
-      "loss": 0.0158,
-      "step": 6081
-    },
-    {
-      "epoch": 357.76,
-      "learning_rate": 1.4972136222910219e-05,
-      "loss": 0.0173,
-      "step": 6082
-    },
-    {
-      "epoch": 357.82,
-      "learning_rate": 1.496594427244582e-05,
-      "loss": 0.0169,
-      "step": 6083
-    },
-    {
-      "epoch": 357.88,
-      "learning_rate": 1.4959752321981424e-05,
-      "loss": 0.0155,
-      "step": 6084
-    },
-    {
-      "epoch": 357.94,
-      "learning_rate": 1.4953560371517029e-05,
-      "loss": 0.0158,
-      "step": 6085
-    },
-    {
-      "epoch": 358.0,
-      "learning_rate": 1.4947368421052632e-05,
-      "loss": 0.0175,
-      "step": 6086
-    },
-    {
-      "epoch": 358.06,
-      "learning_rate": 1.4941176470588237e-05,
-      "loss": 0.0184,
-      "step": 6087
-    },
-    {
-      "epoch": 358.12,
-      "learning_rate": 1.493498452012384e-05,
-      "loss": 0.0178,
-      "step": 6088
-    },
-    {
-      "epoch": 358.18,
-      "learning_rate": 1.4928792569659444e-05,
-      "loss": 0.0139,
-      "step": 6089
-    },
-    {
-      "epoch": 358.24,
-      "learning_rate": 1.4922600619195049e-05,
-      "loss": 0.0166,
-      "step": 6090
-    },
-    {
-      "epoch": 358.29,
-      "learning_rate": 1.491640866873065e-05,
-      "loss": 0.0173,
-      "step": 6091
-    },
-    {
-      "epoch": 358.35,
-      "learning_rate": 1.4910216718266254e-05,
-      "loss": 0.015,
-      "step": 6092
-    },
-    {
-      "epoch": 358.41,
-      "learning_rate": 1.4904024767801859e-05,
-      "loss": 0.0163,
-      "step": 6093
-    },
-    {
-      "epoch": 358.47,
-      "learning_rate": 1.4897832817337462e-05,
-      "loss": 0.0167,
-      "step": 6094
-    },
-    {
-      "epoch": 358.53,
-      "learning_rate": 1.4891640866873065e-05,
-      "loss": 0.0161,
-      "step": 6095
-    },
-    {
-      "epoch": 358.59,
-      "learning_rate": 1.488544891640867e-05,
-      "loss": 0.0169,
-      "step": 6096
-    },
-    {
-      "epoch": 358.65,
-      "learning_rate": 1.4879256965944274e-05,
-      "loss": 0.0179,
-      "step": 6097
-    },
-    {
-      "epoch": 358.71,
-      "learning_rate": 1.4873065015479875e-05,
-      "loss": 0.0136,
-      "step": 6098
-    },
-    {
-      "epoch": 358.76,
-      "learning_rate": 1.486687306501548e-05,
-      "loss": 0.0159,
-      "step": 6099
-    },
-    {
-      "epoch": 358.82,
-      "learning_rate": 1.4860681114551084e-05,
-      "loss": 0.0183,
-      "step": 6100
-    },
-    {
-      "epoch": 358.82,
-      "eval_accuracy_NAT": 0.9940608649742795,
-      "eval_accuracy_REST": 0.4894308943089431,
-      "eval_accuracy_concrete": 0.9596890579398847,
-      "eval_accuracy_concrete tiles": 0.943059156257365,
-      "eval_accuracy_dark bricks": 0.8428354037861532,
-      "eval_accuracy_grass": 0.9015298610559145,
-      "eval_accuracy_red bricks": 0.937829525483304,
-      "eval_accuracy_road": 0.9825598639590295,
-      "eval_accuracy_rough concrete": 0.6385272227816874,
-      "eval_accuracy_rough red sidewalk": 0.7568789056918762,
-      "eval_accuracy_speedway bricks": 0.9897790966040224,
-      "eval_accuracy_steel": 0.9286510367989924,
-      "eval_accuracy_tiles": 0.2303311218152625,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843867835331559,
-      "eval_iou_REST": 0.45884146341463417,
-      "eval_iou_concrete": 0.8967405473086337,
-      "eval_iou_concrete tiles": 0.8910414857371902,
-      "eval_iou_dark bricks": 0.8297599698490095,
-      "eval_iou_grass": 0.8469703246315692,
-      "eval_iou_red bricks": 0.9082495611935535,
-      "eval_iou_road": 0.9619206320140214,
-      "eval_iou_rough concrete": 0.6318070328529988,
-      "eval_iou_rough red sidewalk": 0.601568336809893,
-      "eval_iou_speedway bricks": 0.9802889858002014,
-      "eval_iou_steel": 0.8796182550221193,
-      "eval_iou_tiles": 0.22942063971031987,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08297911286354065,
-      "eval_mean_accuracy": 0.8150124624197473,
-      "eval_mean_iou": 0.7769703090674847,
-      "eval_overall_accuracy": 0.9807443780414129,
-      "eval_runtime": 6.0223,
-      "eval_samples_per_second": 19.594,
-      "eval_steps_per_second": 1.328,
-      "step": 6100
-    },
-    {
-      "epoch": 358.88,
-      "learning_rate": 1.4854489164086689e-05,
-      "loss": 0.0168,
-      "step": 6101
-    },
-    {
-      "epoch": 358.94,
-      "learning_rate": 1.4848297213622292e-05,
-      "loss": 0.0169,
-      "step": 6102
-    },
-    {
-      "epoch": 359.0,
-      "learning_rate": 1.4842105263157895e-05,
-      "loss": 0.0149,
-      "step": 6103
-    },
-    {
-      "epoch": 359.06,
-      "learning_rate": 1.48359133126935e-05,
-      "loss": 0.017,
-      "step": 6104
-    },
-    {
-      "epoch": 359.12,
-      "learning_rate": 1.4829721362229104e-05,
-      "loss": 0.016,
-      "step": 6105
-    },
-    {
-      "epoch": 359.18,
-      "learning_rate": 1.4823529411764705e-05,
-      "loss": 0.0187,
-      "step": 6106
-    },
-    {
-      "epoch": 359.24,
-      "learning_rate": 1.481733746130031e-05,
-      "loss": 0.0172,
-      "step": 6107
-    },
-    {
-      "epoch": 359.29,
-      "learning_rate": 1.4811145510835914e-05,
-      "loss": 0.0168,
-      "step": 6108
-    },
-    {
-      "epoch": 359.35,
-      "learning_rate": 1.4804953560371519e-05,
-      "loss": 0.0176,
-      "step": 6109
-    },
-    {
-      "epoch": 359.41,
-      "learning_rate": 1.4798761609907122e-05,
-      "loss": 0.0145,
-      "step": 6110
-    },
-    {
-      "epoch": 359.47,
-      "learning_rate": 1.4792569659442725e-05,
-      "loss": 0.0166,
-      "step": 6111
-    },
-    {
-      "epoch": 359.53,
-      "learning_rate": 1.478637770897833e-05,
-      "loss": 0.0165,
-      "step": 6112
-    },
-    {
-      "epoch": 359.59,
-      "learning_rate": 1.4780185758513932e-05,
-      "loss": 0.0167,
-      "step": 6113
-    },
-    {
-      "epoch": 359.65,
-      "learning_rate": 1.4773993808049535e-05,
-      "loss": 0.0149,
-      "step": 6114
-    },
-    {
-      "epoch": 359.71,
-      "learning_rate": 1.476780185758514e-05,
-      "loss": 0.0171,
-      "step": 6115
-    },
-    {
-      "epoch": 359.76,
-      "learning_rate": 1.4761609907120744e-05,
-      "loss": 0.0156,
-      "step": 6116
-    },
-    {
-      "epoch": 359.82,
-      "learning_rate": 1.4755417956656347e-05,
-      "loss": 0.0159,
-      "step": 6117
-    },
-    {
-      "epoch": 359.88,
-      "learning_rate": 1.4749226006191952e-05,
-      "loss": 0.0148,
-      "step": 6118
-    },
-    {
-      "epoch": 359.94,
-      "learning_rate": 1.4743034055727555e-05,
-      "loss": 0.0156,
-      "step": 6119
-    },
-    {
-      "epoch": 360.0,
-      "learning_rate": 1.4736842105263157e-05,
-      "loss": 0.0193,
-      "step": 6120
-    },
-    {
-      "epoch": 360.0,
-      "eval_accuracy_NAT": 0.9941726779625819,
-      "eval_accuracy_REST": 0.5056910569105691,
-      "eval_accuracy_concrete": 0.9596428118249405,
-      "eval_accuracy_concrete tiles": 0.9479377798727315,
-      "eval_accuracy_dark bricks": 0.8493913060122887,
-      "eval_accuracy_grass": 0.8990081640531856,
-      "eval_accuracy_red bricks": 0.9407403339191565,
-      "eval_accuracy_road": 0.9809405830800646,
-      "eval_accuracy_rough concrete": 0.6361377436404698,
-      "eval_accuracy_rough red sidewalk": 0.7655720076102241,
-      "eval_accuracy_speedway bricks": 0.9909162757827447,
-      "eval_accuracy_steel": 0.9198498653085908,
-      "eval_accuracy_tiles": 0.2428125662698052,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843636576185173,
-      "eval_iou_REST": 0.44812680115273773,
-      "eval_iou_concrete": 0.8968184006709232,
-      "eval_iou_concrete tiles": 0.8930458723744393,
-      "eval_iou_dark bricks": 0.8345118430826806,
-      "eval_iou_grass": 0.8472493958390925,
-      "eval_iou_red bricks": 0.9063442510185724,
-      "eval_iou_road": 0.9628718233870058,
-      "eval_iou_rough concrete": 0.6311060139397128,
-      "eval_iou_rough red sidewalk": 0.6083939904621894,
-      "eval_iou_speedway bricks": 0.9802368780966756,
-      "eval_iou_steel": 0.8780194590675009,
-      "eval_iou_tiles": 0.2412267501354361,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08270082622766495,
-      "eval_mean_accuracy": 0.8179087055574887,
-      "eval_mean_iou": 0.7778703951419602,
-      "eval_overall_accuracy": 0.9808148206290552,
-      "eval_runtime": 5.9063,
-      "eval_samples_per_second": 19.979,
-      "eval_steps_per_second": 1.354,
-      "step": 6120
-    },
-    {
-      "epoch": 360.06,
-      "learning_rate": 1.4730650154798762e-05,
-      "loss": 0.0172,
-      "step": 6121
-    },
-    {
-      "epoch": 360.12,
-      "learning_rate": 1.4724458204334365e-05,
-      "loss": 0.0154,
-      "step": 6122
-    },
-    {
-      "epoch": 360.18,
-      "learning_rate": 1.471826625386997e-05,
-      "loss": 0.0152,
-      "step": 6123
-    },
-    {
-      "epoch": 360.24,
-      "learning_rate": 1.4712074303405574e-05,
-      "loss": 0.0162,
-      "step": 6124
-    },
-    {
-      "epoch": 360.29,
-      "learning_rate": 1.4705882352941177e-05,
-      "loss": 0.0159,
-      "step": 6125
-    },
-    {
-      "epoch": 360.35,
-      "learning_rate": 1.4699690402476782e-05,
-      "loss": 0.0173,
-      "step": 6126
-    },
-    {
-      "epoch": 360.41,
-      "learning_rate": 1.4693498452012385e-05,
-      "loss": 0.0161,
-      "step": 6127
-    },
-    {
-      "epoch": 360.47,
-      "learning_rate": 1.4687306501547987e-05,
-      "loss": 0.0173,
-      "step": 6128
-    },
-    {
-      "epoch": 360.53,
-      "learning_rate": 1.4681114551083592e-05,
-      "loss": 0.0182,
-      "step": 6129
-    },
-    {
-      "epoch": 360.59,
-      "learning_rate": 1.4674922600619195e-05,
-      "loss": 0.0144,
-      "step": 6130
-    },
-    {
-      "epoch": 360.65,
-      "learning_rate": 1.46687306501548e-05,
-      "loss": 0.0165,
-      "step": 6131
-    },
-    {
-      "epoch": 360.71,
-      "learning_rate": 1.4662538699690404e-05,
-      "loss": 0.016,
-      "step": 6132
-    },
-    {
-      "epoch": 360.76,
-      "learning_rate": 1.4656346749226007e-05,
-      "loss": 0.0179,
-      "step": 6133
-    },
-    {
-      "epoch": 360.82,
-      "learning_rate": 1.4650154798761612e-05,
-      "loss": 0.0182,
-      "step": 6134
-    },
-    {
-      "epoch": 360.88,
-      "learning_rate": 1.4643962848297213e-05,
-      "loss": 0.0187,
-      "step": 6135
-    },
-    {
-      "epoch": 360.94,
-      "learning_rate": 1.4637770897832817e-05,
-      "loss": 0.0145,
-      "step": 6136
-    },
-    {
-      "epoch": 361.0,
-      "learning_rate": 1.4631578947368422e-05,
-      "loss": 0.0181,
-      "step": 6137
-    },
-    {
-      "epoch": 361.06,
-      "learning_rate": 1.4625386996904025e-05,
-      "loss": 0.0189,
-      "step": 6138
-    },
-    {
-      "epoch": 361.12,
-      "learning_rate": 1.461919504643963e-05,
-      "loss": 0.0147,
-      "step": 6139
-    },
-    {
-      "epoch": 361.18,
-      "learning_rate": 1.4613003095975234e-05,
-      "loss": 0.0175,
-      "step": 6140
-    },
-    {
-      "epoch": 361.18,
-      "eval_accuracy_NAT": 0.9940913251181517,
-      "eval_accuracy_REST": 0.4894308943089431,
-      "eval_accuracy_concrete": 0.9581491769720337,
-      "eval_accuracy_concrete tiles": 0.9371199622908319,
-      "eval_accuracy_dark bricks": 0.8485012346151638,
-      "eval_accuracy_grass": 0.9025044756573445,
-      "eval_accuracy_red bricks": 0.9352482425307557,
-      "eval_accuracy_road": 0.9820441523919664,
-      "eval_accuracy_rough concrete": 0.6345348262115732,
-      "eval_accuracy_rough red sidewalk": 0.7693395197737989,
-      "eval_accuracy_speedway bricks": 0.9905020861994156,
-      "eval_accuracy_steel": 0.9268468216325898,
-      "eval_accuracy_tiles": 0.23454209458026598,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844257765890144,
-      "eval_iou_REST": 0.43623188405797103,
-      "eval_iou_concrete": 0.8971311111683686,
-      "eval_iou_concrete tiles": 0.8897093374504934,
-      "eval_iou_dark bricks": 0.8344180443202086,
-      "eval_iou_grass": 0.8470869927804279,
-      "eval_iou_red bricks": 0.9054125903870693,
-      "eval_iou_road": 0.9623215090877384,
-      "eval_iou_rough concrete": 0.6291066776158377,
-      "eval_iou_rough red sidewalk": 0.6060458861790544,
-      "eval_iou_speedway bricks": 0.9803538018544754,
-      "eval_iou_steel": 0.879266242956269,
-      "eval_iou_tiles": 0.23343886627468716,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08237135410308838,
-      "eval_mean_accuracy": 0.815604216329449,
-      "eval_mean_iou": 0.7757652862093551,
-      "eval_overall_accuracy": 0.9808125576730502,
-      "eval_runtime": 6.1319,
-      "eval_samples_per_second": 19.244,
-      "eval_steps_per_second": 1.305,
-      "step": 6140
-    },
-    {
-      "epoch": 361.24,
-      "learning_rate": 1.4606811145510837e-05,
-      "loss": 0.0159,
-      "step": 6141
-    },
-    {
-      "epoch": 361.29,
-      "learning_rate": 1.4600619195046442e-05,
-      "loss": 0.0163,
-      "step": 6142
-    },
-    {
-      "epoch": 361.35,
-      "learning_rate": 1.4594427244582043e-05,
-      "loss": 0.0147,
-      "step": 6143
-    },
-    {
-      "epoch": 361.41,
-      "learning_rate": 1.4588235294117647e-05,
-      "loss": 0.0159,
-      "step": 6144
-    },
-    {
-      "epoch": 361.47,
-      "learning_rate": 1.4582043343653252e-05,
-      "loss": 0.0193,
-      "step": 6145
-    },
-    {
-      "epoch": 361.53,
-      "learning_rate": 1.4575851393188855e-05,
-      "loss": 0.0142,
-      "step": 6146
-    },
-    {
-      "epoch": 361.59,
-      "learning_rate": 1.4569659442724458e-05,
-      "loss": 0.0191,
-      "step": 6147
-    },
-    {
-      "epoch": 361.65,
-      "learning_rate": 1.4563467492260063e-05,
-      "loss": 0.0159,
-      "step": 6148
-    },
-    {
-      "epoch": 361.71,
-      "learning_rate": 1.4557275541795667e-05,
-      "loss": 0.0163,
-      "step": 6149
-    },
-    {
-      "epoch": 361.76,
-      "learning_rate": 1.4551083591331268e-05,
-      "loss": 0.0144,
-      "step": 6150
-    },
-    {
-      "epoch": 361.82,
-      "learning_rate": 1.4544891640866873e-05,
-      "loss": 0.0146,
-      "step": 6151
-    },
-    {
-      "epoch": 361.88,
-      "learning_rate": 1.4538699690402477e-05,
-      "loss": 0.0165,
-      "step": 6152
-    },
-    {
-      "epoch": 361.94,
-      "learning_rate": 1.4532507739938082e-05,
-      "loss": 0.0196,
-      "step": 6153
-    },
-    {
-      "epoch": 362.0,
-      "learning_rate": 1.4526315789473685e-05,
-      "loss": 0.018,
-      "step": 6154
-    },
-    {
-      "epoch": 362.06,
-      "learning_rate": 1.4520123839009288e-05,
-      "loss": 0.0166,
-      "step": 6155
-    },
-    {
-      "epoch": 362.12,
-      "learning_rate": 1.4513931888544893e-05,
-      "loss": 0.0161,
-      "step": 6156
-    },
-    {
-      "epoch": 362.18,
-      "learning_rate": 1.4507739938080497e-05,
-      "loss": 0.0169,
-      "step": 6157
-    },
-    {
-      "epoch": 362.24,
-      "learning_rate": 1.4501547987616098e-05,
-      "loss": 0.0154,
-      "step": 6158
-    },
-    {
-      "epoch": 362.29,
-      "learning_rate": 1.4495356037151703e-05,
-      "loss": 0.0186,
-      "step": 6159
-    },
-    {
-      "epoch": 362.35,
-      "learning_rate": 1.4489164086687307e-05,
-      "loss": 0.0188,
-      "step": 6160
-    },
-    {
-      "epoch": 362.35,
-      "eval_accuracy_NAT": 0.9939585404555727,
-      "eval_accuracy_REST": 0.48292682926829267,
-      "eval_accuracy_concrete": 0.9591084972076519,
-      "eval_accuracy_concrete tiles": 0.9467122319113834,
-      "eval_accuracy_dark bricks": 0.8462904121126276,
-      "eval_accuracy_grass": 0.9031166850234854,
-      "eval_accuracy_red bricks": 0.9363466608084359,
-      "eval_accuracy_road": 0.9817331222033335,
-      "eval_accuracy_rough concrete": 0.6369243053527908,
-      "eval_accuracy_rough red sidewalk": 0.7660081667030133,
-      "eval_accuracy_speedway bricks": 0.990882345481752,
-      "eval_accuracy_steel": 0.9237181825905731,
-      "eval_accuracy_tiles": 0.24114635402465995,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844337154627567,
-      "eval_iou_REST": 0.4267241379310345,
-      "eval_iou_concrete": 0.8974733944107609,
-      "eval_iou_concrete tiles": 0.8898168043772013,
-      "eval_iou_dark bricks": 0.8325259971660853,
-      "eval_iou_grass": 0.8468854009630887,
-      "eval_iou_red bricks": 0.9072477650063857,
-      "eval_iou_road": 0.9627664480855989,
-      "eval_iou_rough concrete": 0.6303176119543809,
-      "eval_iou_rough red sidewalk": 0.6088170076443552,
-      "eval_iou_speedway bricks": 0.9802991985577328,
-      "eval_iou_steel": 0.8797791317593298,
-      "eval_iou_tiles": 0.239629116743934,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08279965072870255,
-      "eval_mean_accuracy": 0.8160671025495057,
-      "eval_mean_iou": 0.7759012100048188,
-      "eval_overall_accuracy": 0.9808518684516518,
-      "eval_runtime": 6.6602,
-      "eval_samples_per_second": 17.717,
-      "eval_steps_per_second": 1.201,
-      "step": 6160
-    },
-    {
-      "epoch": 362.41,
-      "learning_rate": 1.4482972136222912e-05,
-      "loss": 0.0169,
-      "step": 6161
-    },
-    {
-      "epoch": 362.47,
-      "learning_rate": 1.4476780185758515e-05,
-      "loss": 0.0171,
-      "step": 6162
-    },
-    {
-      "epoch": 362.53,
-      "learning_rate": 1.4470588235294118e-05,
-      "loss": 0.0147,
-      "step": 6163
-    },
-    {
-      "epoch": 362.59,
-      "learning_rate": 1.4464396284829723e-05,
-      "loss": 0.0168,
-      "step": 6164
-    },
-    {
-      "epoch": 362.65,
-      "learning_rate": 1.4458204334365325e-05,
-      "loss": 0.0131,
-      "step": 6165
-    },
-    {
-      "epoch": 362.71,
-      "learning_rate": 1.4452012383900928e-05,
-      "loss": 0.0162,
-      "step": 6166
-    },
-    {
-      "epoch": 362.76,
-      "learning_rate": 1.4445820433436533e-05,
-      "loss": 0.0129,
-      "step": 6167
-    },
-    {
-      "epoch": 362.82,
-      "learning_rate": 1.4439628482972137e-05,
-      "loss": 0.0156,
-      "step": 6168
-    },
-    {
-      "epoch": 362.88,
-      "learning_rate": 1.443343653250774e-05,
-      "loss": 0.0167,
-      "step": 6169
-    },
-    {
-      "epoch": 362.94,
-      "learning_rate": 1.4427244582043345e-05,
-      "loss": 0.0146,
-      "step": 6170
-    },
-    {
-      "epoch": 363.0,
-      "learning_rate": 1.4421052631578948e-05,
-      "loss": 0.022,
-      "step": 6171
-    },
-    {
-      "epoch": 363.06,
-      "learning_rate": 1.4414860681114553e-05,
-      "loss": 0.0193,
-      "step": 6172
-    },
-    {
-      "epoch": 363.12,
-      "learning_rate": 1.4408668730650155e-05,
-      "loss": 0.0145,
-      "step": 6173
-    },
-    {
-      "epoch": 363.18,
-      "learning_rate": 1.4402476780185758e-05,
-      "loss": 0.0165,
-      "step": 6174
-    },
-    {
-      "epoch": 363.24,
-      "learning_rate": 1.4396284829721363e-05,
-      "loss": 0.0153,
-      "step": 6175
-    },
-    {
-      "epoch": 363.29,
-      "learning_rate": 1.4390092879256967e-05,
-      "loss": 0.0143,
-      "step": 6176
-    },
-    {
-      "epoch": 363.35,
-      "learning_rate": 1.438390092879257e-05,
-      "loss": 0.0138,
-      "step": 6177
-    },
-    {
-      "epoch": 363.41,
-      "learning_rate": 1.4377708978328175e-05,
-      "loss": 0.0156,
-      "step": 6178
-    },
-    {
-      "epoch": 363.47,
-      "learning_rate": 1.4371517027863778e-05,
-      "loss": 0.016,
-      "step": 6179
-    },
-    {
-      "epoch": 363.53,
-      "learning_rate": 1.436532507739938e-05,
-      "loss": 0.018,
-      "step": 6180
-    },
-    {
-      "epoch": 363.53,
-      "eval_accuracy_NAT": 0.994244865807971,
-      "eval_accuracy_REST": 0.49105691056910566,
-      "eval_accuracy_concrete": 0.9582940305056191,
-      "eval_accuracy_concrete tiles": 0.9477256658024983,
-      "eval_accuracy_dark bricks": 0.8545355358612637,
-      "eval_accuracy_grass": 0.9024921746963555,
-      "eval_accuracy_red bricks": 0.9499121265377856,
-      "eval_accuracy_road": 0.9818752768943583,
-      "eval_accuracy_rough concrete": 0.6378955898914902,
-      "eval_accuracy_rough red sidewalk": 0.7668052850450071,
-      "eval_accuracy_speedway bricks": 0.9904517048433963,
-      "eval_accuracy_steel": 0.924697756475933,
-      "eval_accuracy_tiles": 0.2542639886091672,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844545241798347,
-      "eval_iou_REST": 0.42715700141442714,
-      "eval_iou_concrete": 0.8979984706933632,
-      "eval_iou_concrete tiles": 0.88555132242507,
-      "eval_iou_dark bricks": 0.8376284065856747,
-      "eval_iou_grass": 0.848934713898786,
-      "eval_iou_red bricks": 0.9085465146819351,
-      "eval_iou_road": 0.9624034865326391,
-      "eval_iou_rough concrete": 0.6320130357005803,
-      "eval_iou_rough red sidewalk": 0.6120588235294118,
-      "eval_iou_speedway bricks": 0.9804249953054797,
-      "eval_iou_steel": 0.878483284507604,
-      "eval_iou_tiles": 0.2526261926978298,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08227313309907913,
-      "eval_mean_accuracy": 0.8195577624261502,
-      "eval_mean_iou": 0.7775600593963564,
-      "eval_overall_accuracy": 0.9809612985384666,
-      "eval_runtime": 5.6647,
-      "eval_samples_per_second": 20.831,
-      "eval_steps_per_second": 1.412,
-      "step": 6180
-    },
-    {
-      "epoch": 363.59,
-      "learning_rate": 1.4359133126934985e-05,
-      "loss": 0.0167,
-      "step": 6181
-    },
-    {
-      "epoch": 363.65,
-      "learning_rate": 1.4352941176470588e-05,
-      "loss": 0.0159,
-      "step": 6182
-    },
-    {
-      "epoch": 363.71,
-      "learning_rate": 1.4346749226006193e-05,
-      "loss": 0.0164,
-      "step": 6183
-    },
-    {
-      "epoch": 363.76,
-      "learning_rate": 1.4340557275541797e-05,
-      "loss": 0.0176,
-      "step": 6184
-    },
-    {
-      "epoch": 363.82,
-      "learning_rate": 1.43343653250774e-05,
-      "loss": 0.0166,
-      "step": 6185
-    },
-    {
-      "epoch": 363.88,
-      "learning_rate": 1.4328173374613005e-05,
-      "loss": 0.0164,
-      "step": 6186
-    },
-    {
-      "epoch": 363.94,
-      "learning_rate": 1.4321981424148608e-05,
-      "loss": 0.0171,
-      "step": 6187
-    },
-    {
-      "epoch": 364.0,
-      "learning_rate": 1.431578947368421e-05,
-      "loss": 0.0163,
-      "step": 6188
-    },
-    {
-      "epoch": 364.06,
-      "learning_rate": 1.4309597523219815e-05,
-      "loss": 0.0177,
-      "step": 6189
-    },
-    {
-      "epoch": 364.12,
-      "learning_rate": 1.4303405572755418e-05,
-      "loss": 0.0172,
-      "step": 6190
-    },
-    {
-      "epoch": 364.18,
-      "learning_rate": 1.4297213622291022e-05,
-      "loss": 0.0158,
-      "step": 6191
-    },
-    {
-      "epoch": 364.24,
-      "learning_rate": 1.4291021671826627e-05,
-      "loss": 0.0152,
-      "step": 6192
-    },
-    {
-      "epoch": 364.29,
-      "learning_rate": 1.428482972136223e-05,
-      "loss": 0.0157,
-      "step": 6193
-    },
-    {
-      "epoch": 364.35,
-      "learning_rate": 1.4278637770897835e-05,
-      "loss": 0.0157,
-      "step": 6194
-    },
-    {
-      "epoch": 364.41,
-      "learning_rate": 1.4272445820433437e-05,
-      "loss": 0.0165,
-      "step": 6195
-    },
-    {
-      "epoch": 364.47,
-      "learning_rate": 1.426625386996904e-05,
-      "loss": 0.0148,
-      "step": 6196
-    },
-    {
-      "epoch": 364.53,
-      "learning_rate": 1.4260061919504645e-05,
-      "loss": 0.0149,
-      "step": 6197
-    },
-    {
-      "epoch": 364.59,
-      "learning_rate": 1.4253869969040248e-05,
-      "loss": 0.0171,
-      "step": 6198
-    },
-    {
-      "epoch": 364.65,
-      "learning_rate": 1.4247678018575852e-05,
-      "loss": 0.0147,
-      "step": 6199
-    },
-    {
-      "epoch": 364.71,
-      "learning_rate": 1.4241486068111457e-05,
-      "loss": 0.0182,
-      "step": 6200
-    },
-    {
-      "epoch": 364.71,
-      "eval_accuracy_NAT": 0.9937946487080128,
-      "eval_accuracy_REST": 0.46991869918699186,
-      "eval_accuracy_concrete": 0.9591050574139784,
-      "eval_accuracy_concrete tiles": 0.9405137874145651,
-      "eval_accuracy_dark bricks": 0.8607564649809544,
-      "eval_accuracy_grass": 0.9102143395140931,
-      "eval_accuracy_red bricks": 0.9393673110720563,
-      "eval_accuracy_road": 0.9817956488907391,
-      "eval_accuracy_rough concrete": 0.6425196193518017,
-      "eval_accuracy_rough red sidewalk": 0.7789425397995172,
-      "eval_accuracy_speedway bricks": 0.9906025061811412,
-      "eval_accuracy_steel": 0.924435370613783,
-      "eval_accuracy_tiles": 0.24066163773516314,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846452043879482,
-      "eval_iou_REST": 0.4243759177679883,
-      "eval_iou_concrete": 0.8985846801699459,
-      "eval_iou_concrete tiles": 0.8896270370287803,
-      "eval_iou_dark bricks": 0.8443163522514445,
-      "eval_iou_grass": 0.849916725937772,
-      "eval_iou_red bricks": 0.9091585605698187,
-      "eval_iou_road": 0.962561406523762,
-      "eval_iou_rough concrete": 0.6368801866454032,
-      "eval_iou_rough red sidewalk": 0.6201720721098292,
-      "eval_iou_speedway bricks": 0.9806329898565147,
-      "eval_iou_steel": 0.8794219518515085,
-      "eval_iou_tiles": 0.23932035910104235,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08051618933677673,
-      "eval_mean_accuracy": 0.8178944331432921,
-      "eval_mean_iou": 0.7784318034001354,
-      "eval_overall_accuracy": 0.9811322487006753,
-      "eval_runtime": 5.7752,
-      "eval_samples_per_second": 20.432,
-      "eval_steps_per_second": 1.385,
-      "step": 6200
-    },
-    {
-      "epoch": 364.76,
-      "learning_rate": 1.423529411764706e-05,
-      "loss": 0.0154,
-      "step": 6201
-    },
-    {
-      "epoch": 364.82,
-      "learning_rate": 1.4229102167182662e-05,
-      "loss": 0.0151,
-      "step": 6202
-    },
-    {
-      "epoch": 364.88,
-      "learning_rate": 1.4222910216718267e-05,
-      "loss": 0.0176,
-      "step": 6203
-    },
-    {
-      "epoch": 364.94,
-      "learning_rate": 1.421671826625387e-05,
-      "loss": 0.0174,
-      "step": 6204
-    },
-    {
-      "epoch": 365.0,
-      "learning_rate": 1.4210526315789475e-05,
-      "loss": 0.0181,
-      "step": 6205
-    },
-    {
-      "epoch": 365.06,
-      "learning_rate": 1.4204334365325078e-05,
-      "loss": 0.0148,
-      "step": 6206
-    },
-    {
-      "epoch": 365.12,
-      "learning_rate": 1.4198142414860682e-05,
-      "loss": 0.0182,
-      "step": 6207
-    },
-    {
-      "epoch": 365.18,
-      "learning_rate": 1.4191950464396287e-05,
-      "loss": 0.0159,
-      "step": 6208
-    },
-    {
-      "epoch": 365.24,
-      "learning_rate": 1.418575851393189e-05,
-      "loss": 0.0153,
-      "step": 6209
-    },
-    {
-      "epoch": 365.29,
-      "learning_rate": 1.4179566563467492e-05,
-      "loss": 0.0157,
-      "step": 6210
-    },
-    {
-      "epoch": 365.35,
-      "learning_rate": 1.4173374613003097e-05,
-      "loss": 0.015,
-      "step": 6211
-    },
-    {
-      "epoch": 365.41,
-      "learning_rate": 1.41671826625387e-05,
-      "loss": 0.018,
-      "step": 6212
-    },
-    {
-      "epoch": 365.47,
-      "learning_rate": 1.4160990712074303e-05,
-      "loss": 0.0163,
-      "step": 6213
-    },
-    {
-      "epoch": 365.53,
-      "learning_rate": 1.4154798761609908e-05,
-      "loss": 0.0174,
-      "step": 6214
-    },
-    {
-      "epoch": 365.59,
-      "learning_rate": 1.4148606811145512e-05,
-      "loss": 0.0196,
-      "step": 6215
-    },
-    {
-      "epoch": 365.65,
-      "learning_rate": 1.4142414860681117e-05,
-      "loss": 0.017,
-      "step": 6216
-    },
-    {
-      "epoch": 365.71,
-      "learning_rate": 1.4136222910216718e-05,
-      "loss": 0.0169,
-      "step": 6217
-    },
-    {
-      "epoch": 365.76,
-      "learning_rate": 1.4130030959752322e-05,
-      "loss": 0.0164,
-      "step": 6218
-    },
-    {
-      "epoch": 365.82,
-      "learning_rate": 1.4123839009287927e-05,
-      "loss": 0.0192,
-      "step": 6219
-    },
-    {
-      "epoch": 365.88,
-      "learning_rate": 1.411764705882353e-05,
-      "loss": 0.0134,
-      "step": 6220
-    },
-    {
-      "epoch": 365.88,
-      "eval_accuracy_NAT": 0.9938807996989999,
-      "eval_accuracy_REST": 0.5495934959349593,
-      "eval_accuracy_concrete": 0.9608459752120824,
-      "eval_accuracy_concrete tiles": 0.944661795899128,
-      "eval_accuracy_dark bricks": 0.8563013226652374,
-      "eval_accuracy_grass": 0.9063016876918477,
-      "eval_accuracy_red bricks": 0.930140597539543,
-      "eval_accuracy_road": 0.9825609327912929,
-      "eval_accuracy_rough concrete": 0.635261799915385,
-      "eval_accuracy_rough red sidewalk": 0.762646733694794,
-      "eval_accuracy_speedway bricks": 0.9895028559716987,
-      "eval_accuracy_steel": 0.9315972551939906,
-      "eval_accuracy_tiles": 0.23317883001605622,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846314215957723,
-      "eval_iou_REST": 0.47875354107648727,
-      "eval_iou_concrete": 0.8968597323325384,
-      "eval_iou_concrete tiles": 0.8917415680341728,
-      "eval_iou_dark bricks": 0.8411386722697766,
-      "eval_iou_grass": 0.85053546691294,
-      "eval_iou_red bricks": 0.9017624194664821,
-      "eval_iou_road": 0.9619548952431978,
-      "eval_iou_rough concrete": 0.6308075548507728,
-      "eval_iou_rough red sidewalk": 0.6115365597752023,
-      "eval_iou_speedway bricks": 0.9804954916710533,
-      "eval_iou_steel": 0.8805393761603106,
-      "eval_iou_tiles": 0.2321170084439083,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08178497850894928,
-      "eval_mean_accuracy": 0.8212672370942318,
-      "eval_mean_iou": 0.7802210544486627,
-      "eval_overall_accuracy": 0.9809833138676013,
-      "eval_runtime": 6.1932,
-      "eval_samples_per_second": 19.053,
-      "eval_steps_per_second": 1.292,
-      "step": 6220
-    },
-    {
-      "epoch": 365.94,
-      "learning_rate": 1.4111455108359133e-05,
-      "loss": 0.0134,
-      "step": 6221
-    },
-    {
-      "epoch": 366.0,
-      "learning_rate": 1.4105263157894738e-05,
-      "loss": 0.0162,
-      "step": 6222
-    },
-    {
-      "epoch": 366.06,
-      "learning_rate": 1.4099071207430342e-05,
-      "loss": 0.0189,
-      "step": 6223
-    },
-    {
-      "epoch": 366.12,
-      "learning_rate": 1.4092879256965947e-05,
-      "loss": 0.0163,
-      "step": 6224
-    },
-    {
-      "epoch": 366.18,
-      "learning_rate": 1.4086687306501548e-05,
-      "loss": 0.0141,
-      "step": 6225
-    },
-    {
-      "epoch": 366.24,
-      "learning_rate": 1.4080495356037152e-05,
-      "loss": 0.0133,
-      "step": 6226
-    },
-    {
-      "epoch": 366.29,
-      "learning_rate": 1.4074303405572757e-05,
-      "loss": 0.0176,
-      "step": 6227
-    },
-    {
-      "epoch": 366.35,
-      "learning_rate": 1.406811145510836e-05,
-      "loss": 0.016,
-      "step": 6228
-    },
-    {
-      "epoch": 366.41,
-      "learning_rate": 1.4061919504643963e-05,
-      "loss": 0.0163,
-      "step": 6229
-    },
-    {
-      "epoch": 366.47,
-      "learning_rate": 1.4055727554179568e-05,
-      "loss": 0.0163,
-      "step": 6230
-    },
-    {
-      "epoch": 366.53,
-      "learning_rate": 1.4049535603715172e-05,
-      "loss": 0.016,
-      "step": 6231
-    },
-    {
-      "epoch": 366.59,
-      "learning_rate": 1.4043343653250773e-05,
-      "loss": 0.018,
-      "step": 6232
-    },
-    {
-      "epoch": 366.65,
-      "learning_rate": 1.4037151702786378e-05,
-      "loss": 0.019,
-      "step": 6233
-    },
-    {
-      "epoch": 366.71,
-      "learning_rate": 1.4030959752321981e-05,
-      "loss": 0.0167,
-      "step": 6234
-    },
-    {
-      "epoch": 366.76,
-      "learning_rate": 1.4024767801857586e-05,
-      "loss": 0.0159,
-      "step": 6235
-    },
-    {
-      "epoch": 366.82,
-      "learning_rate": 1.401857585139319e-05,
-      "loss": 0.0182,
-      "step": 6236
-    },
-    {
-      "epoch": 366.88,
-      "learning_rate": 1.4012383900928793e-05,
-      "loss": 0.0144,
-      "step": 6237
-    },
-    {
-      "epoch": 366.94,
-      "learning_rate": 1.4006191950464398e-05,
-      "loss": 0.0146,
-      "step": 6238
-    },
-    {
-      "epoch": 367.0,
-      "learning_rate": 1.4000000000000001e-05,
-      "loss": 0.0145,
-      "step": 6239
-    },
-    {
-      "epoch": 367.06,
-      "learning_rate": 1.3993808049535603e-05,
-      "loss": 0.0151,
-      "step": 6240
-    },
-    {
-      "epoch": 367.06,
-      "eval_accuracy_NAT": 0.9937379335374756,
-      "eval_accuracy_REST": 0.5365853658536586,
-      "eval_accuracy_concrete": 0.9594650891518081,
-      "eval_accuracy_concrete tiles": 0.9514965826066463,
-      "eval_accuracy_dark bricks": 0.8570669754799686,
-      "eval_accuracy_grass": 0.903095868012581,
-      "eval_accuracy_red bricks": 0.9307996485061512,
-      "eval_accuracy_road": 0.9827191199662677,
-      "eval_accuracy_rough concrete": 0.6398083649646346,
-      "eval_accuracy_rough red sidewalk": 0.7771151836004181,
-      "eval_accuracy_speedway bricks": 0.9900616777471377,
-      "eval_accuracy_steel": 0.9347683757565459,
-      "eval_accuracy_tiles": 0.2304523008876367,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984616938637193,
-      "eval_iou_REST": 0.47413793103448276,
-      "eval_iou_concrete": 0.8973289014599232,
-      "eval_iou_concrete tiles": 0.8930277826933286,
-      "eval_iou_dark bricks": 0.8408552072525481,
-      "eval_iou_grass": 0.8482153490116476,
-      "eval_iou_red bricks": 0.901681208767823,
-      "eval_iou_road": 0.9620131249359134,
-      "eval_iou_rough concrete": 0.6345900389482207,
-      "eval_iou_rough red sidewalk": 0.6127083320981139,
-      "eval_iou_speedway bricks": 0.9802241478153333,
-      "eval_iou_steel": 0.8796431292504209,
-      "eval_iou_tiles": 0.22920935277811258,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08293467015028,
-      "eval_mean_accuracy": 0.8220901912362254,
-      "eval_mean_iou": 0.7798654957448508,
-      "eval_overall_accuracy": 0.9809259640968452,
-      "eval_runtime": 5.7807,
-      "eval_samples_per_second": 20.413,
-      "eval_steps_per_second": 1.384,
-      "step": 6240
-    },
-    {
-      "epoch": 367.12,
-      "learning_rate": 1.3987616099071208e-05,
-      "loss": 0.0159,
-      "step": 6241
-    },
-    {
-      "epoch": 367.18,
-      "learning_rate": 1.3981424148606811e-05,
-      "loss": 0.0165,
-      "step": 6242
-    },
-    {
-      "epoch": 367.24,
-      "learning_rate": 1.3975232198142415e-05,
-      "loss": 0.0165,
-      "step": 6243
-    },
-    {
-      "epoch": 367.29,
-      "learning_rate": 1.396904024767802e-05,
-      "loss": 0.0178,
-      "step": 6244
-    },
-    {
-      "epoch": 367.35,
-      "learning_rate": 1.3962848297213623e-05,
-      "loss": 0.016,
-      "step": 6245
-    },
-    {
-      "epoch": 367.41,
-      "learning_rate": 1.3956656346749228e-05,
-      "loss": 0.0147,
-      "step": 6246
-    },
-    {
-      "epoch": 367.47,
-      "learning_rate": 1.395046439628483e-05,
-      "loss": 0.0175,
-      "step": 6247
-    },
-    {
-      "epoch": 367.53,
-      "learning_rate": 1.3944272445820433e-05,
-      "loss": 0.0134,
-      "step": 6248
-    },
-    {
-      "epoch": 367.59,
-      "learning_rate": 1.3938080495356038e-05,
-      "loss": 0.0161,
-      "step": 6249
-    },
-    {
-      "epoch": 367.65,
-      "learning_rate": 1.3931888544891641e-05,
-      "loss": 0.0178,
-      "step": 6250
-    },
-    {
-      "epoch": 367.71,
-      "learning_rate": 1.3925696594427245e-05,
-      "loss": 0.0137,
-      "step": 6251
-    },
-    {
-      "epoch": 367.76,
-      "learning_rate": 1.391950464396285e-05,
-      "loss": 0.0147,
-      "step": 6252
-    },
-    {
-      "epoch": 367.82,
-      "learning_rate": 1.3913312693498453e-05,
-      "loss": 0.0153,
-      "step": 6253
-    },
-    {
-      "epoch": 367.88,
-      "learning_rate": 1.3907120743034058e-05,
-      "loss": 0.0166,
-      "step": 6254
-    },
-    {
-      "epoch": 367.94,
-      "learning_rate": 1.390092879256966e-05,
-      "loss": 0.0161,
-      "step": 6255
-    },
-    {
-      "epoch": 368.0,
-      "learning_rate": 1.3894736842105263e-05,
-      "loss": 0.0182,
-      "step": 6256
-    },
-    {
-      "epoch": 368.06,
-      "learning_rate": 1.3888544891640868e-05,
-      "loss": 0.0176,
-      "step": 6257
-    },
-    {
-      "epoch": 368.12,
-      "learning_rate": 1.3882352941176471e-05,
-      "loss": 0.0154,
-      "step": 6258
-    },
-    {
-      "epoch": 368.18,
-      "learning_rate": 1.3876160990712075e-05,
-      "loss": 0.0174,
-      "step": 6259
-    },
-    {
-      "epoch": 368.24,
-      "learning_rate": 1.386996904024768e-05,
-      "loss": 0.0191,
-      "step": 6260
-    },
-    {
-      "epoch": 368.24,
-      "eval_accuracy_NAT": 0.9935834763477522,
-      "eval_accuracy_REST": 0.5626016260162602,
-      "eval_accuracy_concrete": 0.9590561359039549,
-      "eval_accuracy_concrete tiles": 0.9517558331369315,
-      "eval_accuracy_dark bricks": 0.8565597304902092,
-      "eval_accuracy_grass": 0.9047801534402949,
-      "eval_accuracy_red bricks": 0.9461225834797891,
-      "eval_accuracy_road": 0.9825320743201826,
-      "eval_accuracy_rough concrete": 0.6410954659484326,
-      "eval_accuracy_rough red sidewalk": 0.7688958406966514,
-      "eval_accuracy_speedway bricks": 0.9901444471177409,
-      "eval_accuracy_steel": 0.9323619225636848,
-      "eval_accuracy_tiles": 0.23875306734526947,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845259031542387,
-      "eval_iou_REST": 0.48055555555555557,
-      "eval_iou_concrete": 0.8968185268683623,
-      "eval_iou_concrete tiles": 0.890434820956077,
-      "eval_iou_dark bricks": 0.8402628799436686,
-      "eval_iou_grass": 0.8468582805410628,
-      "eval_iou_red bricks": 0.9041620742140345,
-      "eval_iou_road": 0.9610421720787249,
-      "eval_iou_rough concrete": 0.6340566471399441,
-      "eval_iou_rough red sidewalk": 0.6095673584241997,
-      "eval_iou_speedway bricks": 0.9803050968406684,
-      "eval_iou_steel": 0.8800897292554171,
-      "eval_iou_tiles": 0.2375226039783002,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08378235250711441,
-      "eval_mean_accuracy": 0.8252494120620889,
-      "eval_mean_iou": 0.7804770499192504,
-      "eval_overall_accuracy": 0.9808164370262017,
-      "eval_runtime": 5.7001,
-      "eval_samples_per_second": 20.701,
-      "eval_steps_per_second": 1.403,
-      "step": 6260
-    },
-    {
-      "epoch": 368.29,
-      "learning_rate": 1.3863777089783283e-05,
-      "loss": 0.0171,
-      "step": 6261
-    },
-    {
-      "epoch": 368.35,
-      "learning_rate": 1.3857585139318885e-05,
-      "loss": 0.0168,
-      "step": 6262
-    },
-    {
-      "epoch": 368.41,
-      "learning_rate": 1.385139318885449e-05,
-      "loss": 0.0138,
-      "step": 6263
-    },
-    {
-      "epoch": 368.47,
-      "learning_rate": 1.3845201238390093e-05,
-      "loss": 0.0158,
-      "step": 6264
-    },
-    {
-      "epoch": 368.53,
-      "learning_rate": 1.3839009287925696e-05,
-      "loss": 0.016,
-      "step": 6265
-    },
-    {
-      "epoch": 368.59,
-      "learning_rate": 1.3832817337461301e-05,
-      "loss": 0.0134,
-      "step": 6266
-    },
-    {
-      "epoch": 368.65,
-      "learning_rate": 1.3826625386996905e-05,
-      "loss": 0.0169,
-      "step": 6267
-    },
-    {
-      "epoch": 368.71,
-      "learning_rate": 1.382043343653251e-05,
-      "loss": 0.0158,
-      "step": 6268
-    },
-    {
-      "epoch": 368.76,
-      "learning_rate": 1.3814241486068111e-05,
-      "loss": 0.0181,
-      "step": 6269
-    },
-    {
-      "epoch": 368.82,
-      "learning_rate": 1.3808049535603715e-05,
-      "loss": 0.0143,
-      "step": 6270
-    },
-    {
-      "epoch": 368.88,
-      "learning_rate": 1.380185758513932e-05,
-      "loss": 0.0139,
-      "step": 6271
-    },
-    {
-      "epoch": 368.94,
-      "learning_rate": 1.3795665634674923e-05,
-      "loss": 0.0183,
-      "step": 6272
-    },
-    {
-      "epoch": 369.0,
-      "learning_rate": 1.3789473684210526e-05,
-      "loss": 0.0175,
-      "step": 6273
-    },
-    {
-      "epoch": 369.06,
-      "learning_rate": 1.3783281733746131e-05,
-      "loss": 0.0145,
-      "step": 6274
-    },
-    {
-      "epoch": 369.12,
-      "learning_rate": 1.3777089783281735e-05,
-      "loss": 0.0185,
-      "step": 6275
-    },
-    {
-      "epoch": 369.18,
-      "learning_rate": 1.377089783281734e-05,
-      "loss": 0.0157,
-      "step": 6276
-    },
-    {
-      "epoch": 369.24,
-      "learning_rate": 1.3764705882352941e-05,
-      "loss": 0.0166,
-      "step": 6277
-    },
-    {
-      "epoch": 369.29,
-      "learning_rate": 1.3758513931888545e-05,
-      "loss": 0.0173,
-      "step": 6278
-    },
-    {
-      "epoch": 369.35,
-      "learning_rate": 1.375232198142415e-05,
-      "loss": 0.0156,
-      "step": 6279
-    },
-    {
-      "epoch": 369.41,
-      "learning_rate": 1.3746130030959753e-05,
-      "loss": 0.0149,
-      "step": 6280
-    },
-    {
-      "epoch": 369.41,
-      "eval_accuracy_NAT": 0.9942148369581714,
-      "eval_accuracy_REST": 0.46016260162601624,
-      "eval_accuracy_concrete": 0.9583800253474574,
-      "eval_accuracy_concrete tiles": 0.9493047372142353,
-      "eval_accuracy_dark bricks": 0.8510853128648814,
-      "eval_accuracy_grass": 0.9036465725737773,
-      "eval_accuracy_red bricks": 0.9365663444639719,
-      "eval_accuracy_road": 0.9818624509071981,
-      "eval_accuracy_rough concrete": 0.6429546118139186,
-      "eval_accuracy_rough red sidewalk": 0.7671286443724197,
-      "eval_accuracy_speedway bricks": 0.989900251769688,
-      "eval_accuracy_steel": 0.9309350432561836,
-      "eval_accuracy_tiles": 0.23184586021994003,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845368392482176,
-      "eval_iou_REST": 0.4255639097744361,
-      "eval_iou_concrete": 0.8981347033713452,
-      "eval_iou_concrete tiles": 0.8882004013318927,
-      "eval_iou_dark bricks": 0.8358578619130648,
-      "eval_iou_grass": 0.8484543455567155,
-      "eval_iou_red bricks": 0.9092508664356171,
-      "eval_iou_road": 0.9617823258307211,
-      "eval_iou_rough concrete": 0.6381519026271276,
-      "eval_iou_rough red sidewalk": 0.6090377738108742,
-      "eval_iou_speedway bricks": 0.9802366222689517,
-      "eval_iou_steel": 0.8801629266310382,
-      "eval_iou_tiles": 0.23063014194015008,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08296280354261398,
-      "eval_mean_accuracy": 0.8152297917990661,
-      "eval_mean_iou": 0.7761538939030886,
-      "eval_overall_accuracy": 0.9809411905579648,
-      "eval_runtime": 6.4599,
-      "eval_samples_per_second": 18.267,
-      "eval_steps_per_second": 1.238,
-      "step": 6280
-    },
-    {
-      "epoch": 369.47,
-      "learning_rate": 1.3739938080495356e-05,
-      "loss": 0.0198,
-      "step": 6281
-    },
-    {
-      "epoch": 369.53,
-      "learning_rate": 1.3733746130030961e-05,
-      "loss": 0.0166,
-      "step": 6282
-    },
-    {
-      "epoch": 369.59,
-      "learning_rate": 1.3727554179566565e-05,
-      "loss": 0.0162,
-      "step": 6283
-    },
-    {
-      "epoch": 369.65,
-      "learning_rate": 1.3721362229102166e-05,
-      "loss": 0.0153,
-      "step": 6284
-    },
-    {
-      "epoch": 369.71,
-      "learning_rate": 1.3715170278637771e-05,
-      "loss": 0.0144,
-      "step": 6285
-    },
-    {
-      "epoch": 369.76,
-      "learning_rate": 1.3708978328173375e-05,
-      "loss": 0.0158,
-      "step": 6286
-    },
-    {
-      "epoch": 369.82,
-      "learning_rate": 1.3702786377708978e-05,
-      "loss": 0.0159,
-      "step": 6287
-    },
-    {
-      "epoch": 369.88,
-      "learning_rate": 1.3696594427244583e-05,
-      "loss": 0.0143,
-      "step": 6288
-    },
-    {
-      "epoch": 369.94,
-      "learning_rate": 1.3690402476780186e-05,
-      "loss": 0.0154,
-      "step": 6289
-    },
-    {
-      "epoch": 370.0,
-      "learning_rate": 1.3684210526315791e-05,
-      "loss": 0.0139,
-      "step": 6290
-    },
-    {
-      "epoch": 370.06,
-      "learning_rate": 1.3678018575851395e-05,
-      "loss": 0.0158,
-      "step": 6291
-    },
-    {
-      "epoch": 370.12,
-      "learning_rate": 1.3671826625386996e-05,
-      "loss": 0.0154,
-      "step": 6292
-    },
-    {
-      "epoch": 370.18,
-      "learning_rate": 1.3665634674922601e-05,
-      "loss": 0.0148,
-      "step": 6293
-    },
-    {
-      "epoch": 370.24,
-      "learning_rate": 1.3659442724458205e-05,
-      "loss": 0.0183,
-      "step": 6294
-    },
-    {
-      "epoch": 370.29,
-      "learning_rate": 1.3653250773993808e-05,
-      "loss": 0.0149,
-      "step": 6295
-    },
-    {
-      "epoch": 370.35,
-      "learning_rate": 1.3647058823529413e-05,
-      "loss": 0.0189,
-      "step": 6296
-    },
-    {
-      "epoch": 370.41,
-      "learning_rate": 1.3640866873065016e-05,
-      "loss": 0.0171,
-      "step": 6297
-    },
-    {
-      "epoch": 370.47,
-      "learning_rate": 1.3634674922600621e-05,
-      "loss": 0.0151,
-      "step": 6298
-    },
-    {
-      "epoch": 370.53,
-      "learning_rate": 1.3628482972136223e-05,
-      "loss": 0.0157,
-      "step": 6299
-    },
-    {
-      "epoch": 370.59,
-      "learning_rate": 1.3622291021671826e-05,
-      "loss": 0.0159,
-      "step": 6300
-    },
-    {
-      "epoch": 370.59,
-      "eval_accuracy_NAT": 0.9939523406032801,
-      "eval_accuracy_REST": 0.5430894308943089,
-      "eval_accuracy_concrete": 0.9600636132510025,
-      "eval_accuracy_concrete tiles": 0.9494461465943907,
-      "eval_accuracy_dark bricks": 0.8581006067798557,
-      "eval_accuracy_grass": 0.9057500369028829,
-      "eval_accuracy_red bricks": 0.9444749560632689,
-      "eval_accuracy_road": 0.9819100139429169,
-      "eval_accuracy_rough concrete": 0.6392482376846483,
-      "eval_accuracy_rough red sidewalk": 0.7696102392107025,
-      "eval_accuracy_speedway bricks": 0.9897833807329356,
-      "eval_accuracy_steel": 0.9323719182155762,
-      "eval_accuracy_tiles": 0.250840679814596,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847044004411252,
-      "eval_iou_REST": 0.4890190336749634,
-      "eval_iou_concrete": 0.8975389671572868,
-      "eval_iou_concrete tiles": 0.8869636055395319,
-      "eval_iou_dark bricks": 0.8428625146886016,
-      "eval_iou_grass": 0.8497166936376572,
-      "eval_iou_red bricks": 0.9053435114503817,
-      "eval_iou_road": 0.962460221632622,
-      "eval_iou_rough concrete": 0.6338097235597516,
-      "eval_iou_rough red sidewalk": 0.6131751595218837,
-      "eval_iou_speedway bricks": 0.980528043616802,
-      "eval_iou_steel": 0.8805748230523963,
-      "eval_iou_tiles": 0.24918743228602383,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08313237130641937,
-      "eval_mean_accuracy": 0.8245108923607973,
-      "eval_mean_iou": 0.7827603177122329,
-      "eval_overall_accuracy": 0.9810621940483482,
-      "eval_runtime": 5.695,
-      "eval_samples_per_second": 20.72,
-      "eval_steps_per_second": 1.405,
-      "step": 6300
-    },
-    {
-      "epoch": 370.65,
-      "learning_rate": 1.3616099071207431e-05,
-      "loss": 0.0158,
-      "step": 6301
-    },
-    {
-      "epoch": 370.71,
-      "learning_rate": 1.3609907120743035e-05,
-      "loss": 0.017,
-      "step": 6302
-    },
-    {
-      "epoch": 370.76,
-      "learning_rate": 1.3603715170278638e-05,
-      "loss": 0.0173,
-      "step": 6303
-    },
-    {
-      "epoch": 370.82,
-      "learning_rate": 1.3597523219814243e-05,
-      "loss": 0.0157,
-      "step": 6304
-    },
-    {
-      "epoch": 370.88,
-      "learning_rate": 1.3591331269349846e-05,
-      "loss": 0.0177,
-      "step": 6305
-    },
-    {
-      "epoch": 370.94,
-      "learning_rate": 1.3585139318885451e-05,
-      "loss": 0.0146,
-      "step": 6306
-    },
-    {
-      "epoch": 371.0,
-      "learning_rate": 1.3578947368421053e-05,
-      "loss": 0.0144,
-      "step": 6307
-    },
-    {
-      "epoch": 371.06,
-      "learning_rate": 1.3572755417956656e-05,
-      "loss": 0.0151,
-      "step": 6308
-    },
-    {
-      "epoch": 371.12,
-      "learning_rate": 1.356656346749226e-05,
-      "loss": 0.0164,
-      "step": 6309
-    },
-    {
-      "epoch": 371.18,
-      "learning_rate": 1.3560371517027865e-05,
-      "loss": 0.0206,
-      "step": 6310
-    },
-    {
-      "epoch": 371.24,
-      "learning_rate": 1.3554179566563468e-05,
-      "loss": 0.0174,
-      "step": 6311
-    },
-    {
-      "epoch": 371.29,
-      "learning_rate": 1.3547987616099073e-05,
-      "loss": 0.0146,
-      "step": 6312
-    },
-    {
-      "epoch": 371.35,
-      "learning_rate": 1.3541795665634676e-05,
-      "loss": 0.0152,
-      "step": 6313
-    },
-    {
-      "epoch": 371.41,
-      "learning_rate": 1.3535603715170278e-05,
-      "loss": 0.0128,
-      "step": 6314
-    },
-    {
-      "epoch": 371.47,
-      "learning_rate": 1.3529411764705883e-05,
-      "loss": 0.0169,
-      "step": 6315
-    },
-    {
-      "epoch": 371.53,
-      "learning_rate": 1.3523219814241486e-05,
-      "loss": 0.0147,
-      "step": 6316
-    },
-    {
-      "epoch": 371.59,
-      "learning_rate": 1.351702786377709e-05,
-      "loss": 0.0146,
-      "step": 6317
-    },
-    {
-      "epoch": 371.65,
-      "learning_rate": 1.3510835913312695e-05,
-      "loss": 0.0156,
-      "step": 6318
-    },
-    {
-      "epoch": 371.71,
-      "learning_rate": 1.3504643962848298e-05,
-      "loss": 0.0178,
-      "step": 6319
-    },
-    {
-      "epoch": 371.76,
-      "learning_rate": 1.3498452012383903e-05,
-      "loss": 0.0168,
-      "step": 6320
-    },
-    {
-      "epoch": 371.76,
-      "eval_accuracy_NAT": 0.9945830003608314,
-      "eval_accuracy_REST": 0.49105691056910566,
-      "eval_accuracy_concrete": 0.9581709623319661,
-      "eval_accuracy_concrete tiles": 0.9518029695969833,
-      "eval_accuracy_dark bricks": 0.856956912887851,
-      "eval_accuracy_grass": 0.8949365459658525,
-      "eval_accuracy_red bricks": 0.9435413005272407,
-      "eval_accuracy_road": 0.9819709373819274,
-      "eval_accuracy_rough concrete": 0.6373354626115041,
-      "eval_accuracy_rough red sidewalk": 0.7707833567706179,
-      "eval_accuracy_speedway bricks": 0.9903823019550022,
-      "eval_accuracy_steel": 0.9306951476107893,
-      "eval_accuracy_tiles": 0.24596322215153443,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844393969996192,
-      "eval_iou_REST": 0.45896656534954405,
-      "eval_iou_concrete": 0.8986494769752241,
-      "eval_iou_concrete tiles": 0.8869001866695948,
-      "eval_iou_dark bricks": 0.8416956115077481,
-      "eval_iou_grass": 0.8459953164944193,
-      "eval_iou_red bricks": 0.9056404849762784,
-      "eval_iou_road": 0.9623939055081405,
-      "eval_iou_rough concrete": 0.6324961709727208,
-      "eval_iou_rough red sidewalk": 0.6130776500424677,
-      "eval_iou_speedway bricks": 0.9805996496602818,
-      "eval_iou_steel": 0.8803874801733166,
-      "eval_iou_tiles": 0.24476199089566186,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08261697739362717,
-      "eval_mean_accuracy": 0.8190906946708618,
-      "eval_mean_iou": 0.7796926066326935,
-      "eval_overall_accuracy": 0.9809891975532143,
-      "eval_runtime": 7.1709,
-      "eval_samples_per_second": 16.455,
-      "eval_steps_per_second": 1.116,
-      "step": 6320
-    },
-    {
-      "epoch": 371.82,
-      "learning_rate": 1.3492260061919506e-05,
-      "loss": 0.0162,
-      "step": 6321
-    },
-    {
-      "epoch": 371.88,
-      "learning_rate": 1.3486068111455108e-05,
-      "loss": 0.017,
-      "step": 6322
-    },
-    {
-      "epoch": 371.94,
-      "learning_rate": 1.3479876160990713e-05,
-      "loss": 0.0171,
-      "step": 6323
-    },
-    {
-      "epoch": 372.0,
-      "learning_rate": 1.3473684210526316e-05,
-      "loss": 0.0155,
-      "step": 6324
-    },
-    {
-      "epoch": 372.06,
-      "learning_rate": 1.346749226006192e-05,
-      "loss": 0.0205,
-      "step": 6325
-    },
-    {
-      "epoch": 372.12,
-      "learning_rate": 1.3461300309597525e-05,
-      "loss": 0.013,
-      "step": 6326
-    },
-    {
-      "epoch": 372.18,
-      "learning_rate": 1.3455108359133128e-05,
-      "loss": 0.0181,
-      "step": 6327
-    },
-    {
-      "epoch": 372.24,
-      "learning_rate": 1.3448916408668733e-05,
-      "loss": 0.0149,
-      "step": 6328
-    },
-    {
-      "epoch": 372.29,
-      "learning_rate": 1.3442724458204334e-05,
-      "loss": 0.0131,
-      "step": 6329
-    },
-    {
-      "epoch": 372.35,
-      "learning_rate": 1.3436532507739938e-05,
-      "loss": 0.0162,
-      "step": 6330
-    },
-    {
-      "epoch": 372.41,
-      "learning_rate": 1.3430340557275543e-05,
-      "loss": 0.0152,
-      "step": 6331
-    },
-    {
-      "epoch": 372.47,
-      "learning_rate": 1.3424148606811146e-05,
-      "loss": 0.0148,
-      "step": 6332
-    },
-    {
-      "epoch": 372.53,
-      "learning_rate": 1.341795665634675e-05,
-      "loss": 0.0154,
-      "step": 6333
-    },
-    {
-      "epoch": 372.59,
-      "learning_rate": 1.3411764705882354e-05,
-      "loss": 0.0151,
-      "step": 6334
-    },
-    {
-      "epoch": 372.65,
-      "learning_rate": 1.3405572755417958e-05,
-      "loss": 0.0146,
-      "step": 6335
-    },
-    {
-      "epoch": 372.71,
-      "learning_rate": 1.339938080495356e-05,
-      "loss": 0.0184,
-      "step": 6336
-    },
-    {
-      "epoch": 372.76,
-      "learning_rate": 1.3393188854489164e-05,
-      "loss": 0.0147,
-      "step": 6337
-    },
-    {
-      "epoch": 372.82,
-      "learning_rate": 1.3386996904024768e-05,
-      "loss": 0.0169,
-      "step": 6338
-    },
-    {
-      "epoch": 372.88,
-      "learning_rate": 1.3380804953560371e-05,
-      "loss": 0.0165,
-      "step": 6339
-    },
-    {
-      "epoch": 372.94,
-      "learning_rate": 1.3374613003095976e-05,
-      "loss": 0.017,
-      "step": 6340
-    },
-    {
-      "epoch": 372.94,
-      "eval_accuracy_NAT": 0.9938217124110638,
-      "eval_accuracy_REST": 0.5300813008130081,
-      "eval_accuracy_concrete": 0.9586980151626106,
-      "eval_accuracy_concrete tiles": 0.940560923874617,
-      "eval_accuracy_dark bricks": 0.8545355358612637,
-      "eval_accuracy_grass": 0.9090798124198072,
-      "eval_accuracy_red bricks": 0.9388730228471002,
-      "eval_accuracy_road": 0.9818325236038244,
-      "eval_accuracy_rough concrete": 0.6387536572140222,
-      "eval_accuracy_rough red sidewalk": 0.7691364801961212,
-      "eval_accuracy_speedway bricks": 0.9896875876104363,
-      "eval_accuracy_steel": 0.9372997745980498,
-      "eval_accuracy_tiles": 0.24417583083401498,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846868514102283,
-      "eval_iou_REST": 0.4961948249619482,
-      "eval_iou_concrete": 0.8976266877656383,
-      "eval_iou_concrete tiles": 0.8898104793756968,
-      "eval_iou_dark bricks": 0.8399134565636611,
-      "eval_iou_grass": 0.8495914936590655,
-      "eval_iou_red bricks": 0.9089700643377466,
-      "eval_iou_road": 0.9615063760036593,
-      "eval_iou_rough concrete": 0.633390451429922,
-      "eval_iou_rough red sidewalk": 0.6070943118481893,
-      "eval_iou_speedway bricks": 0.980462940091454,
-      "eval_iou_steel": 0.8795556764514147,
-      "eval_iou_tiles": 0.24288081964743108,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08272599428892136,
-      "eval_mean_accuracy": 0.8220412444189182,
-      "eval_mean_iou": 0.7824372641189273,
-      "eval_overall_accuracy": 0.9809714495125463,
-      "eval_runtime": 5.4913,
-      "eval_samples_per_second": 21.489,
-      "eval_steps_per_second": 1.457,
-      "step": 6340
-    },
-    {
-      "epoch": 373.0,
-      "learning_rate": 1.336842105263158e-05,
-      "loss": 0.0168,
-      "step": 6341
-    },
-    {
-      "epoch": 373.06,
-      "learning_rate": 1.3362229102167184e-05,
-      "loss": 0.0187,
-      "step": 6342
-    },
-    {
-      "epoch": 373.12,
-      "learning_rate": 1.3356037151702788e-05,
-      "loss": 0.0167,
-      "step": 6343
-    },
-    {
-      "epoch": 373.18,
-      "learning_rate": 1.334984520123839e-05,
-      "loss": 0.0137,
-      "step": 6344
-    },
-    {
-      "epoch": 373.24,
-      "learning_rate": 1.3343653250773994e-05,
-      "loss": 0.0165,
-      "step": 6345
-    },
-    {
-      "epoch": 373.29,
-      "learning_rate": 1.3337461300309598e-05,
-      "loss": 0.0169,
-      "step": 6346
-    },
-    {
-      "epoch": 373.35,
-      "learning_rate": 1.3331269349845201e-05,
-      "loss": 0.0186,
-      "step": 6347
-    },
-    {
-      "epoch": 373.41,
-      "learning_rate": 1.3325077399380806e-05,
-      "loss": 0.0156,
-      "step": 6348
-    },
-    {
-      "epoch": 373.47,
-      "learning_rate": 1.331888544891641e-05,
-      "loss": 0.0155,
-      "step": 6349
-    },
-    {
-      "epoch": 373.53,
-      "learning_rate": 1.3312693498452014e-05,
-      "loss": 0.0159,
-      "step": 6350
-    },
-    {
-      "epoch": 373.59,
-      "learning_rate": 1.3306501547987616e-05,
-      "loss": 0.0134,
-      "step": 6351
-    },
-    {
-      "epoch": 373.65,
-      "learning_rate": 1.330030959752322e-05,
-      "loss": 0.0156,
-      "step": 6352
-    },
-    {
-      "epoch": 373.71,
-      "learning_rate": 1.3294117647058824e-05,
-      "loss": 0.0181,
-      "step": 6353
-    },
-    {
-      "epoch": 373.76,
-      "learning_rate": 1.3287925696594428e-05,
-      "loss": 0.0163,
-      "step": 6354
-    },
-    {
-      "epoch": 373.82,
-      "learning_rate": 1.3281733746130031e-05,
-      "loss": 0.0163,
-      "step": 6355
-    },
-    {
-      "epoch": 373.88,
-      "learning_rate": 1.3275541795665636e-05,
-      "loss": 0.0147,
-      "step": 6356
-    },
-    {
-      "epoch": 373.94,
-      "learning_rate": 1.326934984520124e-05,
-      "loss": 0.0145,
-      "step": 6357
-    },
-    {
-      "epoch": 374.0,
-      "learning_rate": 1.3263157894736844e-05,
-      "loss": 0.0147,
-      "step": 6358
-    },
-    {
-      "epoch": 374.06,
-      "learning_rate": 1.3256965944272446e-05,
-      "loss": 0.0166,
-      "step": 6359
-    },
-    {
-      "epoch": 374.12,
-      "learning_rate": 1.325077399380805e-05,
-      "loss": 0.0165,
-      "step": 6360
-    },
-    {
-      "epoch": 374.12,
-      "eval_accuracy_NAT": 0.9938348668802759,
-      "eval_accuracy_REST": 0.5365853658536586,
-      "eval_accuracy_concrete": 0.9597184872857581,
-      "eval_accuracy_concrete tiles": 0.9531934951685128,
-      "eval_accuracy_dark bricks": 0.8549853568899183,
-      "eval_accuracy_grass": 0.9054425128781599,
-      "eval_accuracy_red bricks": 0.9416190685413005,
-      "eval_accuracy_road": 0.9821307278052972,
-      "eval_accuracy_rough concrete": 0.6359470620132405,
-      "eval_accuracy_rough red sidewalk": 0.7719339143774581,
-      "eval_accuracy_speedway bricks": 0.9904541039555876,
-      "eval_accuracy_steel": 0.9282462128973896,
-      "eval_accuracy_tiles": 0.2533551455663607,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846675376785645,
-      "eval_iou_REST": 0.48672566371681414,
-      "eval_iou_concrete": 0.8975739100968405,
-      "eval_iou_concrete tiles": 0.8925670903954802,
-      "eval_iou_dark bricks": 0.8408420279828883,
-      "eval_iou_grass": 0.8495541358296518,
-      "eval_iou_red bricks": 0.9044629668706479,
-      "eval_iou_road": 0.9616589963030504,
-      "eval_iou_rough concrete": 0.6301159577734218,
-      "eval_iou_rough red sidewalk": 0.6117351894781381,
-      "eval_iou_speedway bricks": 0.9806268166866079,
-      "eval_iou_steel": 0.8803683962297689,
-      "eval_iou_tiles": 0.25171562725740426,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08339240401983261,
-      "eval_mean_accuracy": 0.8236497169317629,
-      "eval_mean_iou": 0.7825087935614831,
-      "eval_overall_accuracy": 0.9810156094825874,
-      "eval_runtime": 5.8011,
-      "eval_samples_per_second": 20.341,
-      "eval_steps_per_second": 1.379,
-      "step": 6360
-    },
-    {
-      "epoch": 374.18,
-      "learning_rate": 1.3244582043343653e-05,
-      "loss": 0.0133,
-      "step": 6361
-    },
-    {
-      "epoch": 374.24,
-      "learning_rate": 1.3238390092879258e-05,
-      "loss": 0.0184,
-      "step": 6362
-    },
-    {
-      "epoch": 374.29,
-      "learning_rate": 1.3232198142414861e-05,
-      "loss": 0.0139,
-      "step": 6363
-    },
-    {
-      "epoch": 374.35,
-      "learning_rate": 1.3226006191950466e-05,
-      "loss": 0.0167,
-      "step": 6364
-    },
-    {
-      "epoch": 374.41,
-      "learning_rate": 1.321981424148607e-05,
-      "loss": 0.0162,
-      "step": 6365
-    },
-    {
-      "epoch": 374.47,
-      "learning_rate": 1.3213622291021671e-05,
-      "loss": 0.0175,
-      "step": 6366
-    },
-    {
-      "epoch": 374.53,
-      "learning_rate": 1.3207430340557276e-05,
-      "loss": 0.0156,
-      "step": 6367
-    },
-    {
-      "epoch": 374.59,
-      "learning_rate": 1.320123839009288e-05,
-      "loss": 0.0148,
-      "step": 6368
-    },
-    {
-      "epoch": 374.65,
-      "learning_rate": 1.3195046439628483e-05,
-      "loss": 0.0174,
-      "step": 6369
-    },
-    {
-      "epoch": 374.71,
-      "learning_rate": 1.3188854489164088e-05,
-      "loss": 0.0162,
-      "step": 6370
-    },
-    {
-      "epoch": 374.76,
-      "learning_rate": 1.3182662538699691e-05,
-      "loss": 0.0178,
-      "step": 6371
-    },
-    {
-      "epoch": 374.82,
-      "learning_rate": 1.3176470588235296e-05,
-      "loss": 0.0178,
-      "step": 6372
-    },
-    {
-      "epoch": 374.88,
-      "learning_rate": 1.31702786377709e-05,
-      "loss": 0.0156,
-      "step": 6373
-    },
-    {
-      "epoch": 374.94,
-      "learning_rate": 1.3164086687306501e-05,
-      "loss": 0.019,
-      "step": 6374
-    },
-    {
-      "epoch": 375.0,
-      "learning_rate": 1.3157894736842106e-05,
-      "loss": 0.0129,
-      "step": 6375
-    },
-    {
-      "epoch": 375.06,
-      "learning_rate": 1.315170278637771e-05,
-      "loss": 0.0144,
-      "step": 6376
-    },
-    {
-      "epoch": 375.12,
-      "learning_rate": 1.3145510835913313e-05,
-      "loss": 0.0183,
-      "step": 6377
-    },
-    {
-      "epoch": 375.18,
-      "learning_rate": 1.3139318885448918e-05,
-      "loss": 0.0187,
-      "step": 6378
-    },
-    {
-      "epoch": 375.24,
-      "learning_rate": 1.3133126934984521e-05,
-      "loss": 0.0212,
-      "step": 6379
-    },
-    {
-      "epoch": 375.29,
-      "learning_rate": 1.3126934984520126e-05,
-      "loss": 0.0159,
-      "step": 6380
-    },
-    {
-      "epoch": 375.29,
-      "eval_accuracy_NAT": 0.9942228698102721,
-      "eval_accuracy_REST": 0.5463414634146342,
-      "eval_accuracy_concrete": 0.9585684496009075,
-      "eval_accuracy_concrete tiles": 0.9503417393353759,
-      "eval_accuracy_dark bricks": 0.8632496219589227,
-      "eval_accuracy_grass": 0.8997585226735098,
-      "eval_accuracy_red bricks": 0.9433216168717048,
-      "eval_accuracy_road": 0.9827159134694776,
-      "eval_accuracy_rough concrete": 0.6385093463791347,
-      "eval_accuracy_rough red sidewalk": 0.774400469247024,
-      "eval_accuracy_speedway bricks": 0.9900286042719277,
-      "eval_accuracy_steel": 0.9281612498563125,
-      "eval_accuracy_tiles": 0.2471447181071829,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845494193713165,
-      "eval_iou_REST": 0.47390691114245415,
-      "eval_iou_concrete": 0.8979629613053947,
-      "eval_iou_concrete tiles": 0.8897788958029922,
-      "eval_iou_dark bricks": 0.84607086743428,
-      "eval_iou_grass": 0.8473135449515746,
-      "eval_iou_red bricks": 0.9106139327748913,
-      "eval_iou_road": 0.9613169640756691,
-      "eval_iou_rough concrete": 0.6320129287972444,
-      "eval_iou_rough red sidewalk": 0.6087344608054667,
-      "eval_iou_speedway bricks": 0.9808553960421842,
-      "eval_iou_steel": 0.8801584837878763,
-      "eval_iou_tiles": 0.24570808987410397,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08336326479911804,
-      "eval_mean_accuracy": 0.8243665065381836,
-      "eval_mean_iou": 0.7814602197050344,
-      "eval_overall_accuracy": 0.9809818267822266,
-      "eval_runtime": 5.5756,
-      "eval_samples_per_second": 21.164,
-      "eval_steps_per_second": 1.435,
-      "step": 6380
-    },
-    {
-      "epoch": 375.35,
-      "learning_rate": 1.3120743034055728e-05,
-      "loss": 0.0135,
-      "step": 6381
-    },
-    {
-      "epoch": 375.41,
-      "learning_rate": 1.3114551083591331e-05,
-      "loss": 0.0152,
-      "step": 6382
-    },
-    {
-      "epoch": 375.47,
-      "learning_rate": 1.3108359133126934e-05,
-      "loss": 0.0181,
-      "step": 6383
-    },
-    {
-      "epoch": 375.53,
-      "learning_rate": 1.310216718266254e-05,
-      "loss": 0.0149,
-      "step": 6384
-    },
-    {
-      "epoch": 375.59,
-      "learning_rate": 1.3095975232198143e-05,
-      "loss": 0.0173,
-      "step": 6385
-    },
-    {
-      "epoch": 375.65,
-      "learning_rate": 1.3089783281733748e-05,
-      "loss": 0.0134,
-      "step": 6386
-    },
-    {
-      "epoch": 375.71,
-      "learning_rate": 1.3083591331269351e-05,
-      "loss": 0.0169,
-      "step": 6387
-    },
-    {
-      "epoch": 375.76,
-      "learning_rate": 1.3077399380804956e-05,
-      "loss": 0.0151,
-      "step": 6388
-    },
-    {
-      "epoch": 375.82,
-      "learning_rate": 1.3071207430340558e-05,
-      "loss": 0.0149,
-      "step": 6389
-    },
-    {
-      "epoch": 375.88,
-      "learning_rate": 1.3065015479876161e-05,
-      "loss": 0.0158,
-      "step": 6390
-    },
-    {
-      "epoch": 375.94,
-      "learning_rate": 1.3058823529411764e-05,
-      "loss": 0.0149,
-      "step": 6391
-    },
-    {
-      "epoch": 376.0,
-      "learning_rate": 1.305263157894737e-05,
-      "loss": 0.0186,
-      "step": 6392
-    },
-    {
-      "epoch": 376.06,
-      "learning_rate": 1.3046439628482973e-05,
-      "loss": 0.0159,
-      "step": 6393
-    },
-    {
-      "epoch": 376.12,
-      "learning_rate": 1.3040247678018578e-05,
-      "loss": 0.0163,
-      "step": 6394
-    },
-    {
-      "epoch": 376.18,
-      "learning_rate": 1.3034055727554181e-05,
-      "loss": 0.014,
-      "step": 6395
-    },
-    {
-      "epoch": 376.24,
-      "learning_rate": 1.3027863777089783e-05,
-      "loss": 0.0186,
-      "step": 6396
-    },
-    {
-      "epoch": 376.29,
-      "learning_rate": 1.3021671826625388e-05,
-      "loss": 0.016,
-      "step": 6397
-    },
-    {
-      "epoch": 376.35,
-      "learning_rate": 1.3015479876160991e-05,
-      "loss": 0.0178,
-      "step": 6398
-    },
-    {
-      "epoch": 376.41,
-      "learning_rate": 1.3009287925696594e-05,
-      "loss": 0.0183,
-      "step": 6399
-    },
-    {
-      "epoch": 376.47,
-      "learning_rate": 1.30030959752322e-05,
-      "loss": 0.0138,
-      "step": 6400
-    },
-    {
-      "epoch": 376.47,
-      "eval_accuracy_NAT": 0.9931617246565807,
-      "eval_accuracy_REST": 0.6211382113821138,
-      "eval_accuracy_concrete": 0.9580486585569071,
-      "eval_accuracy_concrete tiles": 0.9516379919868018,
-      "eval_accuracy_dark bricks": 0.859019390157533,
-      "eval_accuracy_grass": 0.91157596127279,
-      "eval_accuracy_red bricks": 0.9405206502636204,
-      "eval_accuracy_road": 0.9830745066938292,
-      "eval_accuracy_rough concrete": 0.6430261174241295,
-      "eval_accuracy_rough red sidewalk": 0.7722723136735875,
-      "eval_accuracy_speedway bricks": 0.9911644125293977,
-      "eval_accuracy_steel": 0.9285785683227796,
-      "eval_accuracy_tiles": 0.2504771425974734,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847398359564037,
-      "eval_iou_REST": 0.5568513119533528,
-      "eval_iou_concrete": 0.8970032889639888,
-      "eval_iou_concrete tiles": 0.8916024466182348,
-      "eval_iou_dark bricks": 0.842201318351357,
-      "eval_iou_grass": 0.8496566991079028,
-      "eval_iou_red bricks": 0.906420367331816,
-      "eval_iou_road": 0.960500799403502,
-      "eval_iou_rough concrete": 0.6351912130861915,
-      "eval_iou_rough red sidewalk": 0.6078952508923445,
-      "eval_iou_speedway bricks": 0.9804351393169962,
-      "eval_iou_steel": 0.8801767018807144,
-      "eval_iou_tiles": 0.2485644710338815,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08303279429674149,
-      "eval_mean_accuracy": 0.8310535115013494,
-      "eval_mean_iou": 0.7877876033766681,
-      "eval_overall_accuracy": 0.9809386043225304,
-      "eval_runtime": 5.5766,
-      "eval_samples_per_second": 21.16,
-      "eval_steps_per_second": 1.435,
-      "step": 6400
-    },
-    {
-      "epoch": 376.53,
-      "learning_rate": 1.2996904024767803e-05,
-      "loss": 0.0141,
-      "step": 6401
-    },
-    {
-      "epoch": 376.59,
-      "learning_rate": 1.2990712074303408e-05,
-      "loss": 0.016,
-      "step": 6402
-    },
-    {
-      "epoch": 376.65,
-      "learning_rate": 1.2984520123839011e-05,
-      "loss": 0.0163,
-      "step": 6403
-    },
-    {
-      "epoch": 376.71,
-      "learning_rate": 1.2978328173374613e-05,
-      "loss": 0.017,
-      "step": 6404
-    },
-    {
-      "epoch": 376.76,
-      "learning_rate": 1.2972136222910216e-05,
-      "loss": 0.0176,
-      "step": 6405
-    },
-    {
-      "epoch": 376.82,
-      "learning_rate": 1.2965944272445821e-05,
-      "loss": 0.0166,
-      "step": 6406
-    },
-    {
-      "epoch": 376.88,
-      "learning_rate": 1.2959752321981424e-05,
-      "loss": 0.0148,
-      "step": 6407
-    },
-    {
-      "epoch": 376.94,
-      "learning_rate": 1.295356037151703e-05,
-      "loss": 0.0157,
-      "step": 6408
-    },
-    {
-      "epoch": 377.0,
-      "learning_rate": 1.2947368421052633e-05,
-      "loss": 0.0159,
-      "step": 6409
-    },
-    {
-      "epoch": 377.06,
-      "learning_rate": 1.2941176470588238e-05,
-      "loss": 0.0132,
-      "step": 6410
-    },
-    {
-      "epoch": 377.12,
-      "learning_rate": 1.293498452012384e-05,
-      "loss": 0.016,
-      "step": 6411
-    },
-    {
-      "epoch": 377.18,
-      "learning_rate": 1.2928792569659443e-05,
-      "loss": 0.0161,
-      "step": 6412
-    },
-    {
-      "epoch": 377.24,
-      "learning_rate": 1.2922600619195046e-05,
-      "loss": 0.0167,
-      "step": 6413
-    },
-    {
-      "epoch": 377.29,
-      "learning_rate": 1.291640866873065e-05,
-      "loss": 0.0158,
-      "step": 6414
-    },
-    {
-      "epoch": 377.35,
-      "learning_rate": 1.2910216718266254e-05,
-      "loss": 0.0142,
-      "step": 6415
-    },
-    {
-      "epoch": 377.41,
-      "learning_rate": 1.290402476780186e-05,
-      "loss": 0.0147,
-      "step": 6416
-    },
-    {
-      "epoch": 377.47,
-      "learning_rate": 1.2897832817337463e-05,
-      "loss": 0.0167,
-      "step": 6417
-    },
-    {
-      "epoch": 377.53,
-      "learning_rate": 1.2891640866873064e-05,
-      "loss": 0.0181,
-      "step": 6418
-    },
-    {
-      "epoch": 377.59,
-      "learning_rate": 1.288544891640867e-05,
-      "loss": 0.0185,
-      "step": 6419
-    },
-    {
-      "epoch": 377.65,
-      "learning_rate": 1.2879256965944272e-05,
-      "loss": 0.0122,
-      "step": 6420
-    },
-    {
-      "epoch": 377.65,
-      "eval_accuracy_NAT": 0.994243302366958,
-      "eval_accuracy_REST": 0.5203252032520326,
-      "eval_accuracy_concrete": 0.9596745343665964,
-      "eval_accuracy_concrete tiles": 0.9450388875795428,
-      "eval_accuracy_dark bricks": 0.8539277989395708,
-      "eval_accuracy_grass": 0.8999070804331453,
-      "eval_accuracy_red bricks": 0.9339301405975395,
-      "eval_accuracy_road": 0.9814215575985691,
-      "eval_accuracy_rough concrete": 0.6351605003009194,
-      "eval_accuracy_rough red sidewalk": 0.7718135946277231,
-      "eval_accuracy_speedway bricks": 0.9895852826119889,
-      "eval_accuracy_steel": 0.9348683322754602,
-      "eval_accuracy_tiles": 0.24463025235541822,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984450296041599,
-      "eval_iou_REST": 0.4892966360856269,
-      "eval_iou_concrete": 0.8973470847165305,
-      "eval_iou_concrete tiles": 0.8900381781053005,
-      "eval_iou_dark bricks": 0.8394977512655012,
-      "eval_iou_grass": 0.8474806296532273,
-      "eval_iou_red bricks": 0.9069817056909701,
-      "eval_iou_road": 0.9616142268283079,
-      "eval_iou_rough concrete": 0.6299167927382754,
-      "eval_iou_rough red sidewalk": 0.6134971158731581,
-      "eval_iou_speedway bricks": 0.9806552660541983,
-      "eval_iou_steel": 0.8809013633473828,
-      "eval_iou_tiles": 0.24314233235976032,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08412186056375504,
-      "eval_mean_accuracy": 0.8203481897927278,
-      "eval_mean_iou": 0.7819091829815259,
-      "eval_overall_accuracy": 0.9809092828782938,
-      "eval_runtime": 9.5754,
-      "eval_samples_per_second": 12.323,
-      "eval_steps_per_second": 0.835,
-      "step": 6420
-    },
-    {
-      "epoch": 377.71,
-      "learning_rate": 1.2873065015479876e-05,
-      "loss": 0.0157,
-      "step": 6421
-    },
-    {
-      "epoch": 377.76,
-      "learning_rate": 1.286687306501548e-05,
-      "loss": 0.0173,
-      "step": 6422
-    },
-    {
-      "epoch": 377.82,
-      "learning_rate": 1.2860681114551084e-05,
-      "loss": 0.018,
-      "step": 6423
-    },
-    {
-      "epoch": 377.88,
-      "learning_rate": 1.285448916408669e-05,
-      "loss": 0.014,
-      "step": 6424
-    },
-    {
-      "epoch": 377.94,
-      "learning_rate": 1.2848297213622292e-05,
-      "loss": 0.0179,
-      "step": 6425
-    },
-    {
-      "epoch": 378.0,
-      "learning_rate": 1.2842105263157894e-05,
-      "loss": 0.0154,
-      "step": 6426
-    },
-    {
-      "epoch": 378.06,
-      "learning_rate": 1.2835913312693499e-05,
-      "loss": 0.0146,
-      "step": 6427
-    },
-    {
-      "epoch": 378.12,
-      "learning_rate": 1.2829721362229102e-05,
-      "loss": 0.0151,
-      "step": 6428
-    },
-    {
-      "epoch": 378.18,
-      "learning_rate": 1.2823529411764706e-05,
-      "loss": 0.013,
-      "step": 6429
-    },
-    {
-      "epoch": 378.24,
-      "learning_rate": 1.281733746130031e-05,
-      "loss": 0.0157,
-      "step": 6430
-    },
-    {
-      "epoch": 378.29,
-      "learning_rate": 1.2811145510835914e-05,
-      "loss": 0.0149,
-      "step": 6431
-    },
-    {
-      "epoch": 378.35,
-      "learning_rate": 1.2804953560371519e-05,
-      "loss": 0.0163,
-      "step": 6432
-    },
-    {
-      "epoch": 378.41,
-      "learning_rate": 1.279876160990712e-05,
-      "loss": 0.0167,
-      "step": 6433
-    },
-    {
-      "epoch": 378.47,
-      "learning_rate": 1.2792569659442724e-05,
-      "loss": 0.0164,
-      "step": 6434
-    },
-    {
-      "epoch": 378.53,
-      "learning_rate": 1.2786377708978327e-05,
-      "loss": 0.0158,
-      "step": 6435
-    },
-    {
-      "epoch": 378.59,
-      "learning_rate": 1.2780185758513932e-05,
-      "loss": 0.0165,
-      "step": 6436
-    },
-    {
-      "epoch": 378.65,
-      "learning_rate": 1.2773993808049536e-05,
-      "loss": 0.0171,
-      "step": 6437
-    },
-    {
-      "epoch": 378.71,
-      "learning_rate": 1.276780185758514e-05,
-      "loss": 0.0169,
-      "step": 6438
-    },
-    {
-      "epoch": 378.76,
-      "learning_rate": 1.2761609907120744e-05,
-      "loss": 0.0192,
-      "step": 6439
-    },
-    {
-      "epoch": 378.82,
-      "learning_rate": 1.2755417956656349e-05,
-      "loss": 0.0182,
-      "step": 6440
-    },
-    {
-      "epoch": 378.82,
-      "eval_accuracy_NAT": 0.9939855502468646,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9587110099387105,
-      "eval_accuracy_concrete tiles": 0.9533820410087203,
-      "eval_accuracy_dark bricks": 0.8534349099400876,
-      "eval_accuracy_grass": 0.904190653540595,
-      "eval_accuracy_red bricks": 0.937554920913884,
-      "eval_accuracy_road": 0.9814990479376614,
-      "eval_accuracy_rough concrete": 0.6341415453554127,
-      "eval_accuracy_rough red sidewalk": 0.7783334210664842,
-      "eval_accuracy_speedway bricks": 0.990516823602877,
-      "eval_accuracy_steel": 0.927646473783904,
-      "eval_accuracy_tiles": 0.23542064285497893,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845456823197026,
-      "eval_iou_REST": 0.46987951807228917,
-      "eval_iou_concrete": 0.8976166909642378,
-      "eval_iou_concrete tiles": 0.8939075862373765,
-      "eval_iou_dark bricks": 0.8387843214719078,
-      "eval_iou_grass": 0.847987306466656,
-      "eval_iou_red bricks": 0.9103076841038767,
-      "eval_iou_road": 0.9613594213989628,
-      "eval_iou_rough concrete": 0.6291962775958093,
-      "eval_iou_rough red sidewalk": 0.6157461390191086,
-      "eval_iou_speedway bricks": 0.9805801080509603,
-      "eval_iou_steel": 0.8794577601095472,
-      "eval_iou_tiles": 0.2341297339640264,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08375588059425354,
-      "eval_mean_accuracy": 0.8197026241046855,
-      "eval_mean_iou": 0.7802690945980356,
-      "eval_overall_accuracy": 0.980929229219081,
-      "eval_runtime": 5.899,
-      "eval_samples_per_second": 20.003,
-      "eval_steps_per_second": 1.356,
-      "step": 6440
-    },
-    {
-      "epoch": 378.88,
-      "learning_rate": 1.274922600619195e-05,
-      "loss": 0.0139,
-      "step": 6441
-    },
-    {
-      "epoch": 378.94,
-      "learning_rate": 1.2743034055727554e-05,
-      "loss": 0.0145,
-      "step": 6442
-    },
-    {
-      "epoch": 379.0,
-      "learning_rate": 1.2736842105263157e-05,
-      "loss": 0.014,
-      "step": 6443
-    },
-    {
-      "epoch": 379.06,
-      "learning_rate": 1.2730650154798762e-05,
-      "loss": 0.0131,
-      "step": 6444
-    },
-    {
-      "epoch": 379.12,
-      "learning_rate": 1.2724458204334366e-05,
-      "loss": 0.0181,
-      "step": 6445
-    },
-    {
-      "epoch": 379.18,
-      "learning_rate": 1.271826625386997e-05,
-      "loss": 0.0156,
-      "step": 6446
-    },
-    {
-      "epoch": 379.24,
-      "learning_rate": 1.2712074303405574e-05,
-      "loss": 0.0182,
-      "step": 6447
-    },
-    {
-      "epoch": 379.29,
-      "learning_rate": 1.2705882352941176e-05,
-      "loss": 0.0133,
-      "step": 6448
-    },
-    {
-      "epoch": 379.35,
-      "learning_rate": 1.269969040247678e-05,
-      "loss": 0.0188,
-      "step": 6449
-    },
-    {
-      "epoch": 379.41,
-      "learning_rate": 1.2693498452012384e-05,
-      "loss": 0.0189,
-      "step": 6450
-    },
-    {
-      "epoch": 379.47,
-      "learning_rate": 1.2687306501547987e-05,
-      "loss": 0.0169,
-      "step": 6451
-    },
-    {
-      "epoch": 379.53,
-      "learning_rate": 1.2681114551083592e-05,
-      "loss": 0.0142,
-      "step": 6452
-    },
-    {
-      "epoch": 379.59,
-      "learning_rate": 1.2674922600619196e-05,
-      "loss": 0.0133,
-      "step": 6453
-    },
-    {
-      "epoch": 379.65,
-      "learning_rate": 1.26687306501548e-05,
-      "loss": 0.0165,
-      "step": 6454
-    },
-    {
-      "epoch": 379.71,
-      "learning_rate": 1.2662538699690404e-05,
-      "loss": 0.0153,
-      "step": 6455
-    },
-    {
-      "epoch": 379.76,
-      "learning_rate": 1.2656346749226006e-05,
-      "loss": 0.0164,
-      "step": 6456
-    },
-    {
-      "epoch": 379.82,
-      "learning_rate": 1.2650154798761609e-05,
-      "loss": 0.014,
-      "step": 6457
-    },
-    {
-      "epoch": 379.88,
-      "learning_rate": 1.2643962848297214e-05,
-      "loss": 0.0178,
-      "step": 6458
-    },
-    {
-      "epoch": 379.94,
-      "learning_rate": 1.2637770897832817e-05,
-      "loss": 0.0164,
-      "step": 6459
-    },
-    {
-      "epoch": 380.0,
-      "learning_rate": 1.2631578947368422e-05,
-      "loss": 0.0177,
-      "step": 6460
-    },
-    {
-      "epoch": 380.0,
-      "eval_accuracy_NAT": 0.9931346609535298,
-      "eval_accuracy_REST": 0.6357723577235772,
-      "eval_accuracy_concrete": 0.9604068282197615,
-      "eval_accuracy_concrete tiles": 0.9514730143766204,
-      "eval_accuracy_dark bricks": 0.8592347300116762,
-      "eval_accuracy_grass": 0.9113791458969671,
-      "eval_accuracy_red bricks": 0.945628295254833,
-      "eval_accuracy_road": 0.9822931903093254,
-      "eval_accuracy_rough concrete": 0.6435147390939048,
-      "eval_accuracy_rough red sidewalk": 0.7762428654148399,
-      "eval_accuracy_speedway bricks": 0.990604733928176,
-      "eval_accuracy_steel": 0.9274465607460755,
-      "eval_accuracy_tiles": 0.24978036293132175,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846738936901978,
-      "eval_iou_REST": 0.5117801047120419,
-      "eval_iou_concrete": 0.8973288781902606,
-      "eval_iou_concrete tiles": 0.8956405990016639,
-      "eval_iou_dark bricks": 0.8424638252350656,
-      "eval_iou_grass": 0.8493361715936467,
-      "eval_iou_red bricks": 0.9113428253850633,
-      "eval_iou_road": 0.9612848733980268,
-      "eval_iou_rough concrete": 0.6359696130969907,
-      "eval_iou_rough red sidewalk": 0.6130637746920546,
-      "eval_iou_speedway bricks": 0.9805036256644564,
-      "eval_iou_steel": 0.8794474130066466,
-      "eval_iou_tiles": 0.24797738277842943,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08303236216306686,
-      "eval_mean_accuracy": 0.8328393449892777,
-      "eval_mean_iou": 0.785447152341888,
-      "eval_overall_accuracy": 0.9809711262331171,
-      "eval_runtime": 5.7793,
-      "eval_samples_per_second": 20.418,
-      "eval_steps_per_second": 1.384,
-      "step": 6460
-    },
-    {
-      "epoch": 380.06,
-      "learning_rate": 1.2625386996904026e-05,
-      "loss": 0.015,
-      "step": 6461
-    },
-    {
-      "epoch": 380.12,
-      "learning_rate": 1.261919504643963e-05,
-      "loss": 0.0174,
-      "step": 6462
-    },
-    {
-      "epoch": 380.18,
-      "learning_rate": 1.2613003095975232e-05,
-      "loss": 0.0196,
-      "step": 6463
-    },
-    {
-      "epoch": 380.24,
-      "learning_rate": 1.2606811145510836e-05,
-      "loss": 0.0181,
-      "step": 6464
-    },
-    {
-      "epoch": 380.29,
-      "learning_rate": 1.2600619195046439e-05,
-      "loss": 0.0154,
-      "step": 6465
-    },
-    {
-      "epoch": 380.35,
-      "learning_rate": 1.2594427244582044e-05,
-      "loss": 0.0134,
-      "step": 6466
-    },
-    {
-      "epoch": 380.41,
-      "learning_rate": 1.2588235294117647e-05,
-      "loss": 0.0159,
-      "step": 6467
-    },
-    {
-      "epoch": 380.47,
-      "learning_rate": 1.2582043343653252e-05,
-      "loss": 0.0161,
-      "step": 6468
-    },
-    {
-      "epoch": 380.53,
-      "learning_rate": 1.2575851393188856e-05,
-      "loss": 0.0142,
-      "step": 6469
-    },
-    {
-      "epoch": 380.59,
-      "learning_rate": 1.256965944272446e-05,
-      "loss": 0.0168,
-      "step": 6470
-    },
-    {
-      "epoch": 380.65,
-      "learning_rate": 1.2563467492260062e-05,
-      "loss": 0.0172,
-      "step": 6471
-    },
-    {
-      "epoch": 380.71,
-      "learning_rate": 1.2557275541795666e-05,
-      "loss": 0.0158,
-      "step": 6472
-    },
-    {
-      "epoch": 380.76,
-      "learning_rate": 1.2551083591331269e-05,
-      "loss": 0.0155,
-      "step": 6473
-    },
-    {
-      "epoch": 380.82,
-      "learning_rate": 1.2544891640866874e-05,
-      "loss": 0.016,
-      "step": 6474
-    },
-    {
-      "epoch": 380.88,
-      "learning_rate": 1.2538699690402477e-05,
-      "loss": 0.0141,
-      "step": 6475
-    },
-    {
-      "epoch": 380.94,
-      "learning_rate": 1.2532507739938082e-05,
-      "loss": 0.0143,
-      "step": 6476
-    },
-    {
-      "epoch": 381.0,
-      "learning_rate": 1.2526315789473686e-05,
-      "loss": 0.0179,
-      "step": 6477
-    },
-    {
-      "epoch": 381.06,
-      "learning_rate": 1.2520123839009287e-05,
-      "loss": 0.0173,
-      "step": 6478
-    },
-    {
-      "epoch": 381.12,
-      "learning_rate": 1.251393188854489e-05,
-      "loss": 0.0148,
-      "step": 6479
-    },
-    {
-      "epoch": 381.18,
-      "learning_rate": 1.2507739938080496e-05,
-      "loss": 0.0144,
-      "step": 6480
-    },
-    {
-      "epoch": 381.18,
-      "eval_accuracy_NAT": 0.994087497383258,
-      "eval_accuracy_REST": 0.5430894308943089,
-      "eval_accuracy_concrete": 0.959257554933505,
-      "eval_accuracy_concrete tiles": 0.9472778694320056,
-      "eval_accuracy_dark bricks": 0.8569951955285876,
-      "eval_accuracy_grass": 0.8994774930263013,
-      "eval_accuracy_red bricks": 0.9432666959578208,
-      "eval_accuracy_road": 0.9830787820228827,
-      "eval_accuracy_rough concrete": 0.6397666533586781,
-      "eval_accuracy_rough red sidewalk": 0.7834319704615015,
-      "eval_accuracy_speedway bricks": 0.9904222300364733,
-      "eval_accuracy_steel": 0.9250151184234858,
-      "eval_accuracy_tiles": 0.24090399587991154,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846244730496951,
-      "eval_iou_REST": 0.46005509641873277,
-      "eval_iou_concrete": 0.8979928012765926,
-      "eval_iou_concrete tiles": 0.8909207784723145,
-      "eval_iou_dark bricks": 0.8413852073535699,
-      "eval_iou_grass": 0.8473728218275433,
-      "eval_iou_red bricks": 0.9097409820435405,
-      "eval_iou_road": 0.9620979421100446,
-      "eval_iou_rough concrete": 0.6341214200917826,
-      "eval_iou_rough red sidewalk": 0.6104356487856327,
-      "eval_iou_speedway bricks": 0.980652494524595,
-      "eval_iou_steel": 0.8801031871515353,
-      "eval_iou_tiles": 0.2396841185158393,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08335646241903305,
-      "eval_mean_accuracy": 0.8235438836414399,
-      "eval_mean_iou": 0.7799374593554937,
-      "eval_overall_accuracy": 0.9809972795389467,
-      "eval_runtime": 5.5423,
-      "eval_samples_per_second": 21.291,
-      "eval_steps_per_second": 1.443,
-      "step": 6480
-    },
-    {
-      "epoch": 381.24,
-      "learning_rate": 1.2501547987616099e-05,
-      "loss": 0.0166,
-      "step": 6481
-    },
-    {
-      "epoch": 381.29,
-      "learning_rate": 1.2495356037151704e-05,
-      "loss": 0.0153,
-      "step": 6482
-    },
-    {
-      "epoch": 381.35,
-      "learning_rate": 1.2489164086687307e-05,
-      "loss": 0.0154,
-      "step": 6483
-    },
-    {
-      "epoch": 381.41,
-      "learning_rate": 1.248297213622291e-05,
-      "loss": 0.0151,
-      "step": 6484
-    },
-    {
-      "epoch": 381.47,
-      "learning_rate": 1.2476780185758514e-05,
-      "loss": 0.0159,
-      "step": 6485
-    },
-    {
-      "epoch": 381.53,
-      "learning_rate": 1.2470588235294119e-05,
-      "loss": 0.0143,
-      "step": 6486
-    },
-    {
-      "epoch": 381.59,
-      "learning_rate": 1.246439628482972e-05,
-      "loss": 0.0172,
-      "step": 6487
-    },
-    {
-      "epoch": 381.65,
-      "learning_rate": 1.2458204334365326e-05,
-      "loss": 0.0155,
-      "step": 6488
-    },
-    {
-      "epoch": 381.71,
-      "learning_rate": 1.2452012383900929e-05,
-      "loss": 0.0186,
-      "step": 6489
-    },
-    {
-      "epoch": 381.76,
-      "learning_rate": 1.2445820433436534e-05,
-      "loss": 0.0162,
-      "step": 6490
-    },
-    {
-      "epoch": 381.82,
-      "learning_rate": 1.2439628482972136e-05,
-      "loss": 0.0153,
-      "step": 6491
-    },
-    {
-      "epoch": 381.88,
-      "learning_rate": 1.243343653250774e-05,
-      "loss": 0.0154,
-      "step": 6492
-    },
-    {
-      "epoch": 381.94,
-      "learning_rate": 1.2427244582043344e-05,
-      "loss": 0.017,
-      "step": 6493
-    },
-    {
-      "epoch": 382.0,
-      "learning_rate": 1.2421052631578949e-05,
-      "loss": 0.015,
-      "step": 6494
-    },
-    {
-      "epoch": 382.06,
-      "learning_rate": 1.241486068111455e-05,
-      "loss": 0.0151,
-      "step": 6495
-    },
-    {
-      "epoch": 382.12,
-      "learning_rate": 1.2408668730650156e-05,
-      "loss": 0.0172,
-      "step": 6496
-    },
-    {
-      "epoch": 382.18,
-      "learning_rate": 1.2402476780185759e-05,
-      "loss": 0.0151,
-      "step": 6497
-    },
-    {
-      "epoch": 382.24,
-      "learning_rate": 1.2396284829721364e-05,
-      "loss": 0.0149,
-      "step": 6498
-    },
-    {
-      "epoch": 382.29,
-      "learning_rate": 1.2390092879256966e-05,
-      "loss": 0.0179,
-      "step": 6499
-    },
-    {
-      "epoch": 382.35,
-      "learning_rate": 1.238390092879257e-05,
-      "loss": 0.0158,
-      "step": 6500
-    },
-    {
-      "epoch": 382.35,
-      "eval_accuracy_NAT": 0.9941796864912604,
-      "eval_accuracy_REST": 0.43902439024390244,
-      "eval_accuracy_concrete": 0.9589762562508695,
-      "eval_accuracy_concrete tiles": 0.9543719066698091,
-      "eval_accuracy_dark bricks": 0.8505684972149379,
-      "eval_accuracy_grass": 0.8856758147967313,
-      "eval_accuracy_red bricks": 0.945353690685413,
-      "eval_accuracy_road": 0.9833764518082238,
-      "eval_accuracy_rough concrete": 0.6415125820079967,
-      "eval_accuracy_rough red sidewalk": 0.7825295723384895,
-      "eval_accuracy_speedway bricks": 0.9900325456705279,
-      "eval_accuracy_steel": 0.9321095323534263,
-      "eval_accuracy_tiles": 0.23072495380047867,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843421706796648,
-      "eval_iou_REST": 0.37344398340248963,
-      "eval_iou_concrete": 0.8984131769653034,
-      "eval_iou_concrete tiles": 0.8859280651089525,
-      "eval_iou_dark bricks": 0.8362502940484592,
-      "eval_iou_grass": 0.83646588656678,
-      "eval_iou_red bricks": 0.9056137212605883,
-      "eval_iou_road": 0.9624129165882132,
-      "eval_iou_rough concrete": 0.6353791039843247,
-      "eval_iou_rough red sidewalk": 0.5821473326172575,
-      "eval_iou_speedway bricks": 0.9803293631986827,
-      "eval_iou_steel": 0.8799365889677071,
-      "eval_iou_tiles": 0.22931470552812236,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08776503801345825,
-      "eval_mean_accuracy": 0.8144950677178513,
-      "eval_mean_iou": 0.7684597929935804,
-      "eval_overall_accuracy": 0.9805537078340175,
-      "eval_runtime": 8.4133,
-      "eval_samples_per_second": 14.025,
-      "eval_steps_per_second": 0.951,
-      "step": 6500
-    },
-    {
-      "epoch": 382.41,
-      "learning_rate": 1.2377708978328174e-05,
-      "loss": 0.0195,
-      "step": 6501
-    },
-    {
-      "epoch": 382.47,
-      "learning_rate": 1.2371517027863777e-05,
-      "loss": 0.0127,
-      "step": 6502
-    },
-    {
-      "epoch": 382.53,
-      "learning_rate": 1.236532507739938e-05,
-      "loss": 0.0168,
-      "step": 6503
-    },
-    {
-      "epoch": 382.59,
-      "learning_rate": 1.2359133126934986e-05,
-      "loss": 0.014,
-      "step": 6504
-    },
-    {
-      "epoch": 382.65,
-      "learning_rate": 1.2352941176470589e-05,
-      "loss": 0.0152,
-      "step": 6505
-    },
-    {
-      "epoch": 382.71,
-      "learning_rate": 1.2346749226006192e-05,
-      "loss": 0.0173,
-      "step": 6506
-    },
-    {
-      "epoch": 382.76,
-      "learning_rate": 1.2340557275541795e-05,
-      "loss": 0.0165,
-      "step": 6507
-    },
-    {
-      "epoch": 382.82,
-      "learning_rate": 1.23343653250774e-05,
-      "loss": 0.0166,
-      "step": 6508
-    },
-    {
-      "epoch": 382.88,
-      "learning_rate": 1.2328173374613004e-05,
-      "loss": 0.0152,
-      "step": 6509
-    },
-    {
-      "epoch": 382.94,
-      "learning_rate": 1.2321981424148607e-05,
-      "loss": 0.0162,
-      "step": 6510
-    },
-    {
-      "epoch": 383.0,
-      "learning_rate": 1.231578947368421e-05,
-      "loss": 0.0166,
-      "step": 6511
-    },
-    {
-      "epoch": 383.06,
-      "learning_rate": 1.2309597523219815e-05,
-      "loss": 0.0174,
-      "step": 6512
-    },
-    {
-      "epoch": 383.12,
-      "learning_rate": 1.2303405572755417e-05,
-      "loss": 0.0164,
-      "step": 6513
-    },
-    {
-      "epoch": 383.18,
-      "learning_rate": 1.2297213622291022e-05,
-      "loss": 0.0182,
-      "step": 6514
-    },
-    {
-      "epoch": 383.24,
-      "learning_rate": 1.2291021671826625e-05,
-      "loss": 0.0151,
-      "step": 6515
-    },
-    {
-      "epoch": 383.29,
-      "learning_rate": 1.228482972136223e-05,
-      "loss": 0.0181,
-      "step": 6516
-    },
-    {
-      "epoch": 383.35,
-      "learning_rate": 1.2278637770897832e-05,
-      "loss": 0.0143,
-      "step": 6517
-    },
-    {
-      "epoch": 383.41,
-      "learning_rate": 1.2272445820433437e-05,
-      "loss": 0.0151,
-      "step": 6518
-    },
-    {
-      "epoch": 383.47,
-      "learning_rate": 1.226625386996904e-05,
-      "loss": 0.0157,
-      "step": 6519
-    },
-    {
-      "epoch": 383.53,
-      "learning_rate": 1.2260061919504645e-05,
-      "loss": 0.0166,
-      "step": 6520
-    },
-    {
-      "epoch": 383.53,
-      "eval_accuracy_NAT": 0.994147339435821,
-      "eval_accuracy_REST": 0.46016260162601624,
-      "eval_accuracy_concrete": 0.9565175681728886,
-      "eval_accuracy_concrete tiles": 0.9532641998585906,
-      "eval_accuracy_dark bricks": 0.861756598970197,
-      "eval_accuracy_grass": 0.8991680765460416,
-      "eval_accuracy_red bricks": 0.9404657293497364,
-      "eval_accuracy_road": 0.9832727750786794,
-      "eval_accuracy_rough concrete": 0.6385212639808365,
-      "eval_accuracy_rough red sidewalk": 0.7698433587258139,
-      "eval_accuracy_speedway bricks": 0.9912619193034623,
-      "eval_accuracy_steel": 0.9225911728398147,
-      "eval_accuracy_tiles": 0.2446908418916053,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845243466009045,
-      "eval_iou_REST": 0.40256045519203415,
-      "eval_iou_concrete": 0.8982737425455433,
-      "eval_iou_concrete tiles": 0.8908442173424663,
-      "eval_iou_dark bricks": 0.8449087215385265,
-      "eval_iou_grass": 0.8446756331293939,
-      "eval_iou_red bricks": 0.9084832086582842,
-      "eval_iou_road": 0.9620824977842153,
-      "eval_iou_rough concrete": 0.6328832819495139,
-      "eval_iou_rough red sidewalk": 0.6013239664955418,
-      "eval_iou_speedway bricks": 0.9805184601701419,
-      "eval_iou_steel": 0.8796310856336874,
-      "eval_iou_tiles": 0.2432831325301205,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08408892154693604,
-      "eval_mean_accuracy": 0.8165894958291926,
-      "eval_mean_iou": 0.7749225191977211,
-      "eval_overall_accuracy": 0.9809055651648569,
-      "eval_runtime": 9.9132,
-      "eval_samples_per_second": 11.903,
-      "eval_steps_per_second": 0.807,
-      "step": 6520
-    },
-    {
-      "epoch": 383.59,
-      "learning_rate": 1.2253869969040247e-05,
-      "loss": 0.0162,
-      "step": 6521
-    },
-    {
-      "epoch": 383.65,
-      "learning_rate": 1.2247678018575852e-05,
-      "loss": 0.0134,
-      "step": 6522
-    },
-    {
-      "epoch": 383.71,
-      "learning_rate": 1.2241486068111455e-05,
-      "loss": 0.0153,
-      "step": 6523
-    },
-    {
-      "epoch": 383.76,
-      "learning_rate": 1.223529411764706e-05,
-      "loss": 0.0157,
-      "step": 6524
-    },
-    {
-      "epoch": 383.82,
-      "learning_rate": 1.2229102167182662e-05,
-      "loss": 0.0185,
-      "step": 6525
-    },
-    {
-      "epoch": 383.88,
-      "learning_rate": 1.2222910216718267e-05,
-      "loss": 0.0146,
-      "step": 6526
-    },
-    {
-      "epoch": 383.94,
-      "learning_rate": 1.221671826625387e-05,
-      "loss": 0.0146,
-      "step": 6527
-    },
-    {
-      "epoch": 384.0,
-      "learning_rate": 1.2210526315789474e-05,
-      "loss": 0.0183,
-      "step": 6528
-    },
-    {
-      "epoch": 384.06,
-      "learning_rate": 1.2204334365325077e-05,
-      "loss": 0.0163,
-      "step": 6529
-    },
-    {
-      "epoch": 384.12,
-      "learning_rate": 1.2198142414860682e-05,
-      "loss": 0.0164,
-      "step": 6530
-    },
-    {
-      "epoch": 384.18,
-      "learning_rate": 1.2191950464396285e-05,
-      "loss": 0.0156,
-      "step": 6531
-    },
-    {
-      "epoch": 384.24,
-      "learning_rate": 1.2185758513931889e-05,
-      "loss": 0.0166,
-      "step": 6532
-    },
-    {
-      "epoch": 384.29,
-      "learning_rate": 1.2179566563467492e-05,
-      "loss": 0.0161,
-      "step": 6533
-    },
-    {
-      "epoch": 384.35,
-      "learning_rate": 1.2173374613003097e-05,
-      "loss": 0.0142,
-      "step": 6534
-    },
-    {
-      "epoch": 384.41,
-      "learning_rate": 1.21671826625387e-05,
-      "loss": 0.0143,
-      "step": 6535
-    },
-    {
-      "epoch": 384.47,
-      "learning_rate": 1.2160990712074304e-05,
-      "loss": 0.0159,
-      "step": 6536
-    },
-    {
-      "epoch": 384.53,
-      "learning_rate": 1.2154798761609907e-05,
-      "loss": 0.0137,
-      "step": 6537
-    },
-    {
-      "epoch": 384.59,
-      "learning_rate": 1.2148606811145512e-05,
-      "loss": 0.0164,
-      "step": 6538
-    },
-    {
-      "epoch": 384.65,
-      "learning_rate": 1.2142414860681114e-05,
-      "loss": 0.0184,
-      "step": 6539
-    },
-    {
-      "epoch": 384.71,
-      "learning_rate": 1.2136222910216719e-05,
-      "loss": 0.0175,
-      "step": 6540
-    },
-    {
-      "epoch": 384.71,
-      "eval_accuracy_NAT": 0.9941391987602021,
-      "eval_accuracy_REST": 0.41300813008130083,
-      "eval_accuracy_concrete": 0.9587454078754458,
-      "eval_accuracy_concrete tiles": 0.9388875795427764,
-      "eval_accuracy_dark bricks": 0.8498554830312195,
-      "eval_accuracy_grass": 0.9004596774498783,
-      "eval_accuracy_red bricks": 0.9342047451669596,
-      "eval_accuracy_road": 0.9819607834754257,
-      "eval_accuracy_rough concrete": 0.637478473831926,
-      "eval_accuracy_rough red sidewalk": 0.7882372404665399,
-      "eval_accuracy_speedway bricks": 0.9905603503526352,
-      "eval_accuracy_steel": 0.9270817194520383,
-      "eval_accuracy_tiles": 0.22678663394831713,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845469826123922,
-      "eval_iou_REST": 0.3628571428571429,
-      "eval_iou_concrete": 0.898618414524362,
-      "eval_iou_concrete tiles": 0.8869617491205415,
-      "eval_iou_dark bricks": 0.8365568504055696,
-      "eval_iou_grass": 0.8455266999499773,
-      "eval_iou_red bricks": 0.9089937476620531,
-      "eval_iou_road": 0.9631093385301707,
-      "eval_iou_rough concrete": 0.6315177417164985,
-      "eval_iou_rough red sidewalk": 0.6050927101854203,
-      "eval_iou_speedway bricks": 0.980574291084527,
-      "eval_iou_steel": 0.8807627386229017,
-      "eval_iou_tiles": 0.22580158658341629,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08541996777057648,
-      "eval_mean_accuracy": 0.810877340264205,
-      "eval_mean_iou": 0.7700707687580748,
-      "eval_overall_accuracy": 0.9809291968911381,
-      "eval_runtime": 6.4767,
-      "eval_samples_per_second": 18.219,
-      "eval_steps_per_second": 1.235,
-      "step": 6540
-    },
-    {
-      "epoch": 384.76,
-      "learning_rate": 1.2130030959752322e-05,
-      "loss": 0.0166,
-      "step": 6541
-    },
-    {
-      "epoch": 384.82,
-      "learning_rate": 1.2123839009287927e-05,
-      "loss": 0.0143,
-      "step": 6542
-    },
-    {
-      "epoch": 384.88,
-      "learning_rate": 1.2117647058823529e-05,
-      "loss": 0.0161,
-      "step": 6543
-    },
-    {
-      "epoch": 384.94,
-      "learning_rate": 1.2111455108359134e-05,
-      "loss": 0.0164,
-      "step": 6544
-    },
-    {
-      "epoch": 385.0,
-      "learning_rate": 1.2105263157894737e-05,
-      "loss": 0.0139,
-      "step": 6545
-    },
-    {
-      "epoch": 385.06,
-      "learning_rate": 1.2099071207430342e-05,
-      "loss": 0.0168,
-      "step": 6546
-    },
-    {
-      "epoch": 385.12,
-      "learning_rate": 1.2092879256965944e-05,
-      "loss": 0.0159,
-      "step": 6547
-    },
-    {
-      "epoch": 385.18,
-      "learning_rate": 1.2086687306501549e-05,
-      "loss": 0.0163,
-      "step": 6548
-    },
-    {
-      "epoch": 385.24,
-      "learning_rate": 1.2080495356037152e-05,
-      "loss": 0.0172,
-      "step": 6549
-    },
-    {
-      "epoch": 385.29,
-      "learning_rate": 1.2074303405572757e-05,
-      "loss": 0.014,
-      "step": 6550
-    },
-    {
-      "epoch": 385.35,
-      "learning_rate": 1.2068111455108359e-05,
-      "loss": 0.0164,
-      "step": 6551
-    },
-    {
-      "epoch": 385.41,
-      "learning_rate": 1.2061919504643964e-05,
-      "loss": 0.0168,
-      "step": 6552
-    },
-    {
-      "epoch": 385.47,
-      "learning_rate": 1.2055727554179567e-05,
-      "loss": 0.0139,
-      "step": 6553
-    },
-    {
-      "epoch": 385.53,
-      "learning_rate": 1.204953560371517e-05,
-      "loss": 0.0157,
-      "step": 6554
-    },
-    {
-      "epoch": 385.59,
-      "learning_rate": 1.2043343653250774e-05,
-      "loss": 0.0158,
-      "step": 6555
-    },
-    {
-      "epoch": 385.65,
-      "learning_rate": 1.2037151702786379e-05,
-      "loss": 0.0153,
-      "step": 6556
-    },
-    {
-      "epoch": 385.71,
-      "learning_rate": 1.2030959752321982e-05,
-      "loss": 0.0148,
-      "step": 6557
-    },
-    {
-      "epoch": 385.76,
-      "learning_rate": 1.2024767801857585e-05,
-      "loss": 0.0154,
-      "step": 6558
-    },
-    {
-      "epoch": 385.82,
-      "learning_rate": 1.2018575851393189e-05,
-      "loss": 0.0157,
-      "step": 6559
-    },
-    {
-      "epoch": 385.88,
-      "learning_rate": 1.2012383900928794e-05,
-      "loss": 0.0183,
-      "step": 6560
-    },
-    {
-      "epoch": 385.88,
-      "eval_accuracy_NAT": 0.9936601388691437,
-      "eval_accuracy_REST": 0.4975609756097561,
-      "eval_accuracy_concrete": 0.9593274974048668,
-      "eval_accuracy_concrete tiles": 0.9504360122554796,
-      "eval_accuracy_dark bricks": 0.8468407250732155,
-      "eval_accuracy_grass": 0.9063102037417631,
-      "eval_accuracy_red bricks": 0.9361818980667839,
-      "eval_accuracy_road": 0.9820414803113081,
-      "eval_accuracy_rough concrete": 0.6389145448369971,
-      "eval_accuracy_rough red sidewalk": 0.7560667473811654,
-      "eval_accuracy_speedway bricks": 0.9908487579110724,
-      "eval_accuracy_steel": 0.9279613368184839,
-      "eval_accuracy_tiles": 0.23929837317095337,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847150175190623,
-      "eval_iou_REST": 0.42857142857142855,
-      "eval_iou_concrete": 0.8973522588616439,
-      "eval_iou_concrete tiles": 0.8918745576786978,
-      "eval_iou_dark bricks": 0.8340489308454734,
-      "eval_iou_grass": 0.846440239700454,
-      "eval_iou_red bricks": 0.9072812433468171,
-      "eval_iou_road": 0.9613563014298472,
-      "eval_iou_rough concrete": 0.6339845320593174,
-      "eval_iou_rough red sidewalk": 0.5944950331125828,
-      "eval_iou_speedway bricks": 0.9803982742410382,
-      "eval_iou_steel": 0.8798980174536114,
-      "eval_iou_tiles": 0.2378858605631682,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08495592325925827,
-      "eval_mean_accuracy": 0.8173422070346915,
-      "eval_mean_iou": 0.775253976567934,
-      "eval_overall_accuracy": 0.9808428812835176,
-      "eval_runtime": 5.7323,
-      "eval_samples_per_second": 20.585,
-      "eval_steps_per_second": 1.396,
-      "step": 6560
-    },
-    {
-      "epoch": 385.94,
-      "learning_rate": 1.2006191950464397e-05,
-      "loss": 0.0159,
-      "step": 6561
-    },
-    {
-      "epoch": 386.0,
-      "learning_rate": 1.2e-05,
-      "loss": 0.0151,
-      "step": 6562
-    },
-    {
-      "epoch": 386.06,
-      "learning_rate": 1.1993808049535604e-05,
-      "loss": 0.0184,
-      "step": 6563
-    },
-    {
-      "epoch": 386.12,
-      "learning_rate": 1.1987616099071209e-05,
-      "loss": 0.016,
-      "step": 6564
-    },
-    {
-      "epoch": 386.18,
-      "learning_rate": 1.1981424148606812e-05,
-      "loss": 0.0121,
-      "step": 6565
-    },
-    {
-      "epoch": 386.24,
-      "learning_rate": 1.1975232198142415e-05,
-      "loss": 0.0171,
-      "step": 6566
-    },
-    {
-      "epoch": 386.29,
-      "learning_rate": 1.1969040247678019e-05,
-      "loss": 0.0141,
-      "step": 6567
-    },
-    {
-      "epoch": 386.35,
-      "learning_rate": 1.1962848297213624e-05,
-      "loss": 0.0158,
-      "step": 6568
-    },
-    {
-      "epoch": 386.41,
-      "learning_rate": 1.1956656346749225e-05,
-      "loss": 0.0168,
-      "step": 6569
-    },
-    {
-      "epoch": 386.47,
-      "learning_rate": 1.195046439628483e-05,
-      "loss": 0.0156,
-      "step": 6570
-    },
-    {
-      "epoch": 386.53,
-      "learning_rate": 1.1944272445820434e-05,
-      "loss": 0.0148,
-      "step": 6571
-    },
-    {
-      "epoch": 386.59,
-      "learning_rate": 1.1938080495356039e-05,
-      "loss": 0.0147,
-      "step": 6572
-    },
-    {
-      "epoch": 386.65,
-      "learning_rate": 1.193188854489164e-05,
-      "loss": 0.0193,
-      "step": 6573
-    },
-    {
-      "epoch": 386.71,
-      "learning_rate": 1.1925696594427245e-05,
-      "loss": 0.0148,
-      "step": 6574
-    },
-    {
-      "epoch": 386.76,
-      "learning_rate": 1.1919504643962849e-05,
-      "loss": 0.0158,
-      "step": 6575
-    },
-    {
-      "epoch": 386.82,
-      "learning_rate": 1.1913312693498454e-05,
-      "loss": 0.0165,
-      "step": 6576
-    },
-    {
-      "epoch": 386.88,
-      "learning_rate": 1.1907120743034055e-05,
-      "loss": 0.0162,
-      "step": 6577
-    },
-    {
-      "epoch": 386.94,
-      "learning_rate": 1.190092879256966e-05,
-      "loss": 0.0155,
-      "step": 6578
-    },
-    {
-      "epoch": 387.0,
-      "learning_rate": 1.1894736842105264e-05,
-      "loss": 0.0171,
-      "step": 6579
-    },
-    {
-      "epoch": 387.06,
-      "learning_rate": 1.1888544891640867e-05,
-      "loss": 0.0132,
-      "step": 6580
-    },
-    {
-      "epoch": 387.06,
-      "eval_accuracy_NAT": 0.9944747994603864,
-      "eval_accuracy_REST": 0.4796747967479675,
-      "eval_accuracy_concrete": 0.9573327992735156,
-      "eval_accuracy_concrete tiles": 0.9480320527928353,
-      "eval_accuracy_dark bricks": 0.8506211358459507,
-      "eval_accuracy_grass": 0.8935739779793874,
-      "eval_accuracy_red bricks": 0.9468365553602812,
-      "eval_accuracy_road": 0.9823322026869374,
-      "eval_accuracy_rough concrete": 0.6399990465918639,
-      "eval_accuracy_rough red sidewalk": 0.7634363320524293,
-      "eval_accuracy_speedway bricks": 0.9905185372544423,
-      "eval_accuracy_steel": 0.9336038823111946,
-      "eval_accuracy_tiles": 0.2441455360659214,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844227574713155,
-      "eval_iou_REST": 0.4559505409582689,
-      "eval_iou_concrete": 0.8980729508587589,
-      "eval_iou_concrete tiles": 0.889382683293535,
-      "eval_iou_dark bricks": 0.8376498529744402,
-      "eval_iou_grass": 0.843347154636358,
-      "eval_iou_red bricks": 0.9079418580155888,
-      "eval_iou_road": 0.9619181835688928,
-      "eval_iou_rough concrete": 0.6353499322673576,
-      "eval_iou_rough red sidewalk": 0.595829469557358,
-      "eval_iou_speedway bricks": 0.9806370364287903,
-      "eval_iou_steel": 0.8809773650664146,
-      "eval_iou_tiles": 0.24270441198614667,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0858035758137703,
-      "eval_mean_accuracy": 0.817275511878701,
-      "eval_mean_iou": 0.778014169006402,
-      "eval_overall_accuracy": 0.9808267173120531,
-      "eval_runtime": 5.698,
-      "eval_samples_per_second": 20.709,
-      "eval_steps_per_second": 1.404,
-      "step": 6580
-    },
-    {
-      "epoch": 387.12,
-      "learning_rate": 1.188235294117647e-05,
-      "loss": 0.0169,
-      "step": 6581
-    },
-    {
-      "epoch": 387.18,
-      "learning_rate": 1.1876160990712075e-05,
-      "loss": 0.0191,
-      "step": 6582
-    },
-    {
-      "epoch": 387.24,
-      "learning_rate": 1.1869969040247679e-05,
-      "loss": 0.0185,
-      "step": 6583
-    },
-    {
-      "epoch": 387.29,
-      "learning_rate": 1.1863777089783282e-05,
-      "loss": 0.0161,
-      "step": 6584
-    },
-    {
-      "epoch": 387.35,
-      "learning_rate": 1.1857585139318885e-05,
-      "loss": 0.0155,
-      "step": 6585
-    },
-    {
-      "epoch": 387.41,
-      "learning_rate": 1.185139318885449e-05,
-      "loss": 0.0161,
-      "step": 6586
-    },
-    {
-      "epoch": 387.47,
-      "learning_rate": 1.1845201238390094e-05,
-      "loss": 0.0148,
-      "step": 6587
-    },
-    {
-      "epoch": 387.53,
-      "learning_rate": 1.1839009287925697e-05,
-      "loss": 0.0174,
-      "step": 6588
-    },
-    {
-      "epoch": 387.59,
-      "learning_rate": 1.18328173374613e-05,
-      "loss": 0.0161,
-      "step": 6589
-    },
-    {
-      "epoch": 387.65,
-      "learning_rate": 1.1826625386996905e-05,
-      "loss": 0.0157,
-      "step": 6590
-    },
-    {
-      "epoch": 387.71,
-      "learning_rate": 1.1820433436532509e-05,
-      "loss": 0.0168,
-      "step": 6591
-    },
-    {
-      "epoch": 387.76,
-      "learning_rate": 1.1814241486068112e-05,
-      "loss": 0.0124,
-      "step": 6592
-    },
-    {
-      "epoch": 387.82,
-      "learning_rate": 1.1808049535603715e-05,
-      "loss": 0.0184,
-      "step": 6593
-    },
-    {
-      "epoch": 387.88,
-      "learning_rate": 1.180185758513932e-05,
-      "loss": 0.0162,
-      "step": 6594
-    },
-    {
-      "epoch": 387.94,
-      "learning_rate": 1.1795665634674922e-05,
-      "loss": 0.0157,
-      "step": 6595
-    },
-    {
-      "epoch": 388.0,
-      "learning_rate": 1.1789473684210527e-05,
-      "loss": 0.0153,
-      "step": 6596
-    },
-    {
-      "epoch": 388.06,
-      "learning_rate": 1.178328173374613e-05,
-      "loss": 0.0134,
-      "step": 6597
-    },
-    {
-      "epoch": 388.12,
-      "learning_rate": 1.1777089783281735e-05,
-      "loss": 0.0152,
-      "step": 6598
-    },
-    {
-      "epoch": 388.18,
-      "learning_rate": 1.1770897832817337e-05,
-      "loss": 0.015,
-      "step": 6599
-    },
-    {
-      "epoch": 388.24,
-      "learning_rate": 1.1764705882352942e-05,
-      "loss": 0.0164,
-      "step": 6600
-    },
-    {
-      "epoch": 388.24,
-      "eval_accuracy_NAT": 0.9942207133399095,
-      "eval_accuracy_REST": 0.4861788617886179,
-      "eval_accuracy_concrete": 0.9571634849849184,
-      "eval_accuracy_concrete tiles": 0.948432712703276,
-      "eval_accuracy_dark bricks": 0.8549949275501024,
-      "eval_accuracy_grass": 0.8992655380061846,
-      "eval_accuracy_red bricks": 0.9417838312829525,
-      "eval_accuracy_road": 0.9830215994967938,
-      "eval_accuracy_rough concrete": 0.6454751845738563,
-      "eval_accuracy_rough red sidewalk": 0.7760924657276713,
-      "eval_accuracy_speedway bricks": 0.9910238931010444,
-      "eval_accuracy_steel": 0.927993822687131,
-      "eval_accuracy_tiles": 0.250840679814596,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845634733183392,
-      "eval_iou_REST": 0.41643454038997213,
-      "eval_iou_concrete": 0.8988093550418924,
-      "eval_iou_concrete tiles": 0.8884620479533714,
-      "eval_iou_dark bricks": 0.8416396593307205,
-      "eval_iou_grass": 0.8450789264017529,
-      "eval_iou_red bricks": 0.9116427432216906,
-      "eval_iou_road": 0.9633000089028075,
-      "eval_iou_rough concrete": 0.6401992884244867,
-      "eval_iou_rough red sidewalk": 0.6048302497172294,
-      "eval_iou_speedway bricks": 0.9807252138500885,
-      "eval_iou_steel": 0.8811251364305035,
-      "eval_iou_tiles": 0.24933000090336957,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08353140205144882,
-      "eval_mean_accuracy": 0.8197298242351582,
-      "eval_mean_iou": 0.7773954341450942,
-      "eval_overall_accuracy": 0.9810374631720075,
-      "eval_runtime": 5.4424,
-      "eval_samples_per_second": 21.682,
-      "eval_steps_per_second": 1.47,
-      "step": 6600
-    },
-    {
-      "epoch": 388.29,
-      "learning_rate": 1.1758513931888545e-05,
-      "loss": 0.0156,
-      "step": 6601
-    },
-    {
-      "epoch": 388.35,
-      "learning_rate": 1.175232198142415e-05,
-      "loss": 0.0155,
-      "step": 6602
-    },
-    {
-      "epoch": 388.41,
-      "learning_rate": 1.1746130030959752e-05,
-      "loss": 0.0156,
-      "step": 6603
-    },
-    {
-      "epoch": 388.47,
-      "learning_rate": 1.1739938080495357e-05,
-      "loss": 0.0182,
-      "step": 6604
-    },
-    {
-      "epoch": 388.53,
-      "learning_rate": 1.173374613003096e-05,
-      "loss": 0.0167,
-      "step": 6605
-    },
-    {
-      "epoch": 388.59,
-      "learning_rate": 1.1727554179566565e-05,
-      "loss": 0.0148,
-      "step": 6606
-    },
-    {
-      "epoch": 388.65,
-      "learning_rate": 1.1721362229102167e-05,
-      "loss": 0.0184,
-      "step": 6607
-    },
-    {
-      "epoch": 388.71,
-      "learning_rate": 1.1715170278637772e-05,
-      "loss": 0.0175,
-      "step": 6608
-    },
-    {
-      "epoch": 388.76,
-      "learning_rate": 1.1708978328173375e-05,
-      "loss": 0.0152,
-      "step": 6609
-    },
-    {
-      "epoch": 388.82,
-      "learning_rate": 1.1702786377708978e-05,
-      "loss": 0.014,
-      "step": 6610
-    },
-    {
-      "epoch": 388.88,
-      "learning_rate": 1.1696594427244582e-05,
-      "loss": 0.017,
-      "step": 6611
-    },
-    {
-      "epoch": 388.94,
-      "learning_rate": 1.1690402476780187e-05,
-      "loss": 0.0145,
-      "step": 6612
-    },
-    {
-      "epoch": 389.0,
-      "learning_rate": 1.168421052631579e-05,
-      "loss": 0.0132,
-      "step": 6613
-    },
-    {
-      "epoch": 389.06,
-      "learning_rate": 1.1678018575851393e-05,
-      "loss": 0.0156,
-      "step": 6614
-    },
-    {
-      "epoch": 389.12,
-      "learning_rate": 1.1671826625386997e-05,
-      "loss": 0.0158,
-      "step": 6615
-    },
-    {
-      "epoch": 389.18,
-      "learning_rate": 1.1665634674922602e-05,
-      "loss": 0.0136,
-      "step": 6616
-    },
-    {
-      "epoch": 389.24,
-      "learning_rate": 1.1659442724458205e-05,
-      "loss": 0.0155,
-      "step": 6617
-    },
-    {
-      "epoch": 389.29,
-      "learning_rate": 1.1653250773993808e-05,
-      "loss": 0.0156,
-      "step": 6618
-    },
-    {
-      "epoch": 389.35,
-      "learning_rate": 1.1647058823529412e-05,
-      "loss": 0.0136,
-      "step": 6619
-    },
-    {
-      "epoch": 389.41,
-      "learning_rate": 1.1640866873065017e-05,
-      "loss": 0.0169,
-      "step": 6620
-    },
-    {
-      "epoch": 389.41,
-      "eval_accuracy_NAT": 0.9944155504371731,
-      "eval_accuracy_REST": 0.43902439024390244,
-      "eval_accuracy_concrete": 0.9581919832933044,
-      "eval_accuracy_concrete tiles": 0.9385340560923875,
-      "eval_accuracy_dark bricks": 0.8513485060199453,
-      "eval_accuracy_grass": 0.8921527438712827,
-      "eval_accuracy_red bricks": 0.9359622144112478,
-      "eval_accuracy_road": 0.9818715359814365,
-      "eval_accuracy_rough concrete": 0.6969890179300318,
-      "eval_accuracy_rough red sidewalk": 0.7808826957639928,
-      "eval_accuracy_speedway bricks": 0.9909875636878605,
-      "eval_accuracy_steel": 0.9285435835411596,
-      "eval_accuracy_tiles": 0.23226998697324971,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984421476040555,
-      "eval_iou_REST": 0.4147465437788018,
-      "eval_iou_concrete": 0.9008757408186637,
-      "eval_iou_concrete tiles": 0.8857009408153733,
-      "eval_iou_dark bricks": 0.8371432066930801,
-      "eval_iou_grass": 0.8434384412456938,
-      "eval_iou_red bricks": 0.9040369211182431,
-      "eval_iou_road": 0.9622183699762231,
-      "eval_iou_rough concrete": 0.6911205123963036,
-      "eval_iou_rough red sidewalk": 0.6115128673223014,
-      "eval_iou_speedway bricks": 0.9805833342376854,
-      "eval_iou_steel": 0.8812247724480745,
-      "eval_iou_tiles": 0.2314216722004226,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08262140303850174,
-      "eval_mean_accuracy": 0.8170133714036135,
-      "eval_mean_iou": 0.7791111383916479,
-      "eval_overall_accuracy": 0.9811666456319518,
-      "eval_runtime": 8.2989,
-      "eval_samples_per_second": 14.219,
-      "eval_steps_per_second": 0.964,
-      "step": 6620
-    },
-    {
-      "epoch": 389.47,
-      "learning_rate": 1.1634674922600618e-05,
-      "loss": 0.0176,
-      "step": 6621
-    },
-    {
-      "epoch": 389.53,
-      "learning_rate": 1.1628482972136223e-05,
-      "loss": 0.0163,
-      "step": 6622
-    },
-    {
-      "epoch": 389.59,
-      "learning_rate": 1.1622291021671827e-05,
-      "loss": 0.0148,
-      "step": 6623
-    },
-    {
-      "epoch": 389.65,
-      "learning_rate": 1.1616099071207432e-05,
-      "loss": 0.0159,
-      "step": 6624
-    },
-    {
-      "epoch": 389.71,
-      "learning_rate": 1.1609907120743033e-05,
-      "loss": 0.0157,
-      "step": 6625
-    },
-    {
-      "epoch": 389.76,
-      "learning_rate": 1.1603715170278638e-05,
-      "loss": 0.0178,
-      "step": 6626
-    },
-    {
-      "epoch": 389.82,
-      "learning_rate": 1.1597523219814242e-05,
-      "loss": 0.0146,
-      "step": 6627
-    },
-    {
-      "epoch": 389.88,
-      "learning_rate": 1.1591331269349847e-05,
-      "loss": 0.0128,
-      "step": 6628
-    },
-    {
-      "epoch": 389.94,
-      "learning_rate": 1.1585139318885448e-05,
-      "loss": 0.015,
-      "step": 6629
-    },
-    {
-      "epoch": 390.0,
-      "learning_rate": 1.1578947368421053e-05,
-      "loss": 0.0193,
-      "step": 6630
-    },
-    {
-      "epoch": 390.06,
-      "learning_rate": 1.1572755417956657e-05,
-      "loss": 0.0144,
-      "step": 6631
-    },
-    {
-      "epoch": 390.12,
-      "learning_rate": 1.1566563467492262e-05,
-      "loss": 0.0161,
-      "step": 6632
-    },
-    {
-      "epoch": 390.18,
-      "learning_rate": 1.1560371517027863e-05,
-      "loss": 0.0173,
-      "step": 6633
-    },
-    {
-      "epoch": 390.24,
-      "learning_rate": 1.1554179566563468e-05,
-      "loss": 0.0137,
-      "step": 6634
-    },
-    {
-      "epoch": 390.29,
-      "learning_rate": 1.1547987616099072e-05,
-      "loss": 0.0166,
-      "step": 6635
-    },
-    {
-      "epoch": 390.35,
-      "learning_rate": 1.1541795665634675e-05,
-      "loss": 0.0146,
-      "step": 6636
-    },
-    {
-      "epoch": 390.41,
-      "learning_rate": 1.1535603715170278e-05,
-      "loss": 0.0152,
-      "step": 6637
-    },
-    {
-      "epoch": 390.47,
-      "learning_rate": 1.1529411764705883e-05,
-      "loss": 0.0178,
-      "step": 6638
-    },
-    {
-      "epoch": 390.53,
-      "learning_rate": 1.1523219814241487e-05,
-      "loss": 0.0166,
-      "step": 6639
-    },
-    {
-      "epoch": 390.59,
-      "learning_rate": 1.151702786377709e-05,
-      "loss": 0.0174,
-      "step": 6640
-    },
-    {
-      "epoch": 390.59,
-      "eval_accuracy_NAT": 0.9943003410080496,
-      "eval_accuracy_REST": 0.44715447154471544,
-      "eval_accuracy_concrete": 0.9610668864057825,
-      "eval_accuracy_concrete tiles": 0.9409144473250058,
-      "eval_accuracy_dark bricks": 0.8496975671381812,
-      "eval_accuracy_grass": 0.889196728322868,
-      "eval_accuracy_red bricks": 0.9328317223198594,
-      "eval_accuracy_road": 0.9805616820427094,
-      "eval_accuracy_rough concrete": 0.6636495271691525,
-      "eval_accuracy_rough red sidewalk": 0.7781454214575234,
-      "eval_accuracy_speedway bricks": 0.990157470869637,
-      "eval_accuracy_steel": 0.9333090105803975,
-      "eval_accuracy_tiles": 0.23863188827289528,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9843590140327054,
-      "eval_iou_REST": 0.42569659442724456,
-      "eval_iou_concrete": 0.8982708455013584,
-      "eval_iou_concrete tiles": 0.884759435321233,
-      "eval_iou_dark bricks": 0.8372730144808628,
-      "eval_iou_grass": 0.8407619149868928,
-      "eval_iou_red bricks": 0.9021138729551732,
-      "eval_iou_road": 0.9629236999049057,
-      "eval_iou_rough concrete": 0.6591210378050801,
-      "eval_iou_rough red sidewalk": 0.6061779454494329,
-      "eval_iou_speedway bricks": 0.9805091335118754,
-      "eval_iou_steel": 0.8816242322382056,
-      "eval_iou_tiles": 0.2376025579150579,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08503903448581696,
-      "eval_mean_accuracy": 0.8153551664966752,
-      "eval_mean_iou": 0.7770148691176945,
-      "eval_overall_accuracy": 0.9808701337394068,
-      "eval_runtime": 5.4539,
-      "eval_samples_per_second": 21.636,
-      "eval_steps_per_second": 1.467,
-      "step": 6640
-    },
-    {
-      "epoch": 390.65,
-      "learning_rate": 1.1510835913312693e-05,
-      "loss": 0.0155,
-      "step": 6641
-    },
-    {
-      "epoch": 390.71,
-      "learning_rate": 1.1504643962848298e-05,
-      "loss": 0.0169,
-      "step": 6642
-    },
-    {
-      "epoch": 390.76,
-      "learning_rate": 1.1498452012383902e-05,
-      "loss": 0.0161,
-      "step": 6643
-    },
-    {
-      "epoch": 390.82,
-      "learning_rate": 1.1492260061919505e-05,
-      "loss": 0.0154,
-      "step": 6644
-    },
-    {
-      "epoch": 390.88,
-      "learning_rate": 1.1486068111455108e-05,
-      "loss": 0.0146,
-      "step": 6645
-    },
-    {
-      "epoch": 390.94,
-      "learning_rate": 1.1479876160990713e-05,
-      "loss": 0.0131,
-      "step": 6646
-    },
-    {
-      "epoch": 391.0,
-      "learning_rate": 1.1473684210526315e-05,
-      "loss": 0.0141,
-      "step": 6647
-    },
-    {
-      "epoch": 391.06,
-      "learning_rate": 1.146749226006192e-05,
-      "loss": 0.0138,
-      "step": 6648
-    },
-    {
-      "epoch": 391.12,
-      "learning_rate": 1.1461300309597523e-05,
-      "loss": 0.0144,
-      "step": 6649
-    },
-    {
-      "epoch": 391.18,
-      "learning_rate": 1.1455108359133128e-05,
-      "loss": 0.015,
-      "step": 6650
-    },
-    {
-      "epoch": 391.24,
-      "learning_rate": 1.144891640866873e-05,
-      "loss": 0.0174,
-      "step": 6651
-    },
-    {
-      "epoch": 391.29,
-      "learning_rate": 1.1442724458204335e-05,
-      "loss": 0.0154,
-      "step": 6652
-    },
-    {
-      "epoch": 391.35,
-      "learning_rate": 1.1436532507739938e-05,
-      "loss": 0.0172,
-      "step": 6653
-    },
-    {
-      "epoch": 391.41,
-      "learning_rate": 1.1430340557275543e-05,
-      "loss": 0.0163,
-      "step": 6654
-    },
-    {
-      "epoch": 391.47,
-      "learning_rate": 1.1424148606811145e-05,
-      "loss": 0.0153,
-      "step": 6655
-    },
-    {
-      "epoch": 391.53,
-      "learning_rate": 1.141795665634675e-05,
-      "loss": 0.0154,
-      "step": 6656
-    },
-    {
-      "epoch": 391.59,
-      "learning_rate": 1.1411764705882353e-05,
-      "loss": 0.0178,
-      "step": 6657
-    },
-    {
-      "epoch": 391.65,
-      "learning_rate": 1.1405572755417958e-05,
-      "loss": 0.0168,
-      "step": 6658
-    },
-    {
-      "epoch": 391.71,
-      "learning_rate": 1.139938080495356e-05,
-      "loss": 0.0159,
-      "step": 6659
-    },
-    {
-      "epoch": 391.76,
-      "learning_rate": 1.1393188854489165e-05,
-      "loss": 0.0136,
-      "step": 6660
-    },
-    {
-      "epoch": 391.76,
-      "eval_accuracy_NAT": 0.9939184840185868,
-      "eval_accuracy_REST": 0.4926829268292683,
-      "eval_accuracy_concrete": 0.9601033619778967,
-      "eval_accuracy_concrete tiles": 0.9554089087909498,
-      "eval_accuracy_dark bricks": 0.8529946595716172,
-      "eval_accuracy_grass": 0.898451782125379,
-      "eval_accuracy_red bricks": 0.9436511423550088,
-      "eval_accuracy_road": 0.9808449225924954,
-      "eval_accuracy_rough concrete": 0.6811147724631895,
-      "eval_accuracy_rough red sidewalk": 0.7674369637311155,
-      "eval_accuracy_speedway bricks": 0.9904571885284051,
-      "eval_accuracy_steel": 0.9342685931619745,
-      "eval_accuracy_tiles": 0.2534157351025478,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984585405222643,
-      "eval_iou_REST": 0.425561797752809,
-      "eval_iou_concrete": 0.8998376956580454,
-      "eval_iou_concrete tiles": 0.8898107906405022,
-      "eval_iou_dark bricks": 0.8382057660385877,
-      "eval_iou_grass": 0.8439539869448355,
-      "eval_iou_red bricks": 0.9071326751491473,
-      "eval_iou_road": 0.9618651618029017,
-      "eval_iou_rough concrete": 0.6752442727347913,
-      "eval_iou_rough red sidewalk": 0.6015821646889608,
-      "eval_iou_speedway bricks": 0.9805941005943652,
-      "eval_iou_steel": 0.8804170954501461,
-      "eval_iou_tiles": 0.2517834030641423,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08315352350473404,
-      "eval_mean_accuracy": 0.823442264711418,
-      "eval_mean_iou": 0.7800441781339907,
-      "eval_overall_accuracy": 0.9810758687682071,
-      "eval_runtime": 7.344,
-      "eval_samples_per_second": 16.068,
-      "eval_steps_per_second": 1.089,
-      "step": 6660
-    },
-    {
-      "epoch": 391.82,
-      "learning_rate": 1.1386996904024768e-05,
-      "loss": 0.0139,
-      "step": 6661
-    },
-    {
-      "epoch": 391.88,
-      "learning_rate": 1.1380804953560372e-05,
-      "loss": 0.0149,
-      "step": 6662
-    },
-    {
-      "epoch": 391.94,
-      "learning_rate": 1.1374613003095975e-05,
-      "loss": 0.016,
-      "step": 6663
-    },
-    {
-      "epoch": 392.0,
-      "learning_rate": 1.136842105263158e-05,
-      "loss": 0.0147,
-      "step": 6664
-    },
-    {
-      "epoch": 392.06,
-      "learning_rate": 1.1362229102167183e-05,
-      "loss": 0.018,
-      "step": 6665
-    },
-    {
-      "epoch": 392.12,
-      "learning_rate": 1.1356037151702787e-05,
-      "loss": 0.0163,
-      "step": 6666
-    },
-    {
-      "epoch": 392.18,
-      "learning_rate": 1.134984520123839e-05,
-      "loss": 0.0152,
-      "step": 6667
-    },
-    {
-      "epoch": 392.24,
-      "learning_rate": 1.1343653250773995e-05,
-      "loss": 0.0152,
-      "step": 6668
-    },
-    {
-      "epoch": 392.29,
-      "learning_rate": 1.1337461300309598e-05,
-      "loss": 0.0185,
-      "step": 6669
-    },
-    {
-      "epoch": 392.35,
-      "learning_rate": 1.1331269349845202e-05,
-      "loss": 0.0172,
-      "step": 6670
-    },
-    {
-      "epoch": 392.41,
-      "learning_rate": 1.1325077399380805e-05,
-      "loss": 0.0142,
-      "step": 6671
-    },
-    {
-      "epoch": 392.47,
-      "learning_rate": 1.131888544891641e-05,
-      "loss": 0.0159,
-      "step": 6672
-    },
-    {
-      "epoch": 392.53,
-      "learning_rate": 1.1312693498452013e-05,
-      "loss": 0.0151,
-      "step": 6673
-    },
-    {
-      "epoch": 392.59,
-      "learning_rate": 1.1306501547987617e-05,
-      "loss": 0.0161,
-      "step": 6674
-    },
-    {
-      "epoch": 392.65,
-      "learning_rate": 1.130030959752322e-05,
-      "loss": 0.015,
-      "step": 6675
-    },
-    {
-      "epoch": 392.71,
-      "learning_rate": 1.1294117647058825e-05,
-      "loss": 0.0144,
-      "step": 6676
-    },
-    {
-      "epoch": 392.76,
-      "learning_rate": 1.1287925696594427e-05,
-      "loss": 0.0181,
-      "step": 6677
-    },
-    {
-      "epoch": 392.82,
-      "learning_rate": 1.1281733746130032e-05,
-      "loss": 0.0136,
-      "step": 6678
-    },
-    {
-      "epoch": 392.88,
-      "learning_rate": 1.1275541795665635e-05,
-      "loss": 0.0146,
-      "step": 6679
-    },
-    {
-      "epoch": 392.94,
-      "learning_rate": 1.126934984520124e-05,
-      "loss": 0.0154,
-      "step": 6680
-    },
-    {
-      "epoch": 392.94,
-      "eval_accuracy_NAT": 0.9942256732217435,
-      "eval_accuracy_REST": 0.46016260162601624,
-      "eval_accuracy_concrete": 0.9600265399191877,
-      "eval_accuracy_concrete tiles": 0.9464765496111242,
-      "eval_accuracy_dark bricks": 0.8510613862144211,
-      "eval_accuracy_grass": 0.8961013523487266,
-      "eval_accuracy_red bricks": 0.9382688927943761,
-      "eval_accuracy_road": 0.9815508863024336,
-      "eval_accuracy_rough concrete": 0.6740833874591077,
-      "eval_accuracy_rough red sidewalk": 0.7804014167650531,
-      "eval_accuracy_speedway bricks": 0.9901382779721059,
-      "eval_accuracy_steel": 0.9287959737514181,
-      "eval_accuracy_tiles": 0.2302705322790754,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845281750297746,
-      "eval_iou_REST": 0.4307458143074581,
-      "eval_iou_concrete": 0.8989484315227898,
-      "eval_iou_concrete tiles": 0.8899501385041552,
-      "eval_iou_dark bricks": 0.8380478472506915,
-      "eval_iou_grass": 0.8446794182836603,
-      "eval_iou_red bricks": 0.9087234042553192,
-      "eval_iou_road": 0.9617812348140825,
-      "eval_iou_rough concrete": 0.6691470924007878,
-      "eval_iou_rough red sidewalk": 0.6088484464470102,
-      "eval_iou_speedway bricks": 0.9807381033855312,
-      "eval_iou_steel": 0.8812553051246912,
-      "eval_iou_tiles": 0.2292841844891557,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08388333767652512,
-      "eval_mean_accuracy": 0.817812574635753,
-      "eval_mean_iou": 0.7789751996780853,
-      "eval_overall_accuracy": 0.9810487779520326,
-      "eval_runtime": 5.6145,
-      "eval_samples_per_second": 21.017,
-      "eval_steps_per_second": 1.425,
-      "step": 6680
-    },
-    {
-      "epoch": 393.0,
-      "learning_rate": 1.1263157894736842e-05,
-      "loss": 0.0134,
-      "step": 6681
-    },
-    {
-      "epoch": 393.06,
-      "learning_rate": 1.1256965944272447e-05,
-      "loss": 0.0146,
-      "step": 6682
-    },
-    {
-      "epoch": 393.12,
-      "learning_rate": 1.125077399380805e-05,
-      "loss": 0.0193,
-      "step": 6683
-    },
-    {
-      "epoch": 393.18,
-      "learning_rate": 1.1244582043343655e-05,
-      "loss": 0.0161,
-      "step": 6684
-    },
-    {
-      "epoch": 393.24,
-      "learning_rate": 1.1238390092879257e-05,
-      "loss": 0.0169,
-      "step": 6685
-    },
-    {
-      "epoch": 393.29,
-      "learning_rate": 1.1232198142414862e-05,
-      "loss": 0.0151,
-      "step": 6686
-    },
-    {
-      "epoch": 393.35,
-      "learning_rate": 1.1226006191950465e-05,
-      "loss": 0.0191,
-      "step": 6687
-    },
-    {
-      "epoch": 393.41,
-      "learning_rate": 1.1219814241486068e-05,
-      "loss": 0.0168,
-      "step": 6688
-    },
-    {
-      "epoch": 393.47,
-      "learning_rate": 1.1213622291021672e-05,
-      "loss": 0.0137,
-      "step": 6689
-    },
-    {
-      "epoch": 393.53,
-      "learning_rate": 1.1207430340557277e-05,
-      "loss": 0.0147,
-      "step": 6690
-    },
-    {
-      "epoch": 393.59,
-      "learning_rate": 1.120123839009288e-05,
-      "loss": 0.0153,
-      "step": 6691
-    },
-    {
-      "epoch": 393.65,
-      "learning_rate": 1.1195046439628483e-05,
-      "loss": 0.0182,
-      "step": 6692
-    },
-    {
-      "epoch": 393.71,
-      "learning_rate": 1.1188854489164086e-05,
-      "loss": 0.0162,
-      "step": 6693
-    },
-    {
-      "epoch": 393.76,
-      "learning_rate": 1.1182662538699692e-05,
-      "loss": 0.0144,
-      "step": 6694
-    },
-    {
-      "epoch": 393.82,
-      "learning_rate": 1.1176470588235295e-05,
-      "loss": 0.0138,
-      "step": 6695
-    },
-    {
-      "epoch": 393.88,
-      "learning_rate": 1.1170278637770898e-05,
-      "loss": 0.0143,
-      "step": 6696
-    },
-    {
-      "epoch": 393.94,
-      "learning_rate": 1.1164086687306501e-05,
-      "loss": 0.0156,
-      "step": 6697
-    },
-    {
-      "epoch": 394.0,
-      "learning_rate": 1.1157894736842106e-05,
-      "loss": 0.0134,
-      "step": 6698
-    },
-    {
-      "epoch": 394.06,
-      "learning_rate": 1.115170278637771e-05,
-      "loss": 0.0165,
-      "step": 6699
-    },
-    {
-      "epoch": 394.12,
-      "learning_rate": 1.1145510835913313e-05,
-      "loss": 0.017,
-      "step": 6700
-    },
-    {
-      "epoch": 394.12,
-      "eval_accuracy_NAT": 0.9941118654983557,
-      "eval_accuracy_REST": 0.45528455284552843,
-      "eval_accuracy_concrete": 0.9572674431937185,
-      "eval_accuracy_concrete tiles": 0.9472543012019797,
-      "eval_accuracy_dark bricks": 0.8558562869666749,
-      "eval_accuracy_grass": 0.8998626077280314,
-      "eval_accuracy_red bricks": 0.9406304920913884,
-      "eval_accuracy_road": 0.9826288036400151,
-      "eval_accuracy_rough concrete": 0.6706153653638742,
-      "eval_accuracy_rough red sidewalk": 0.7834620503989351,
-      "eval_accuracy_speedway bricks": 0.9905644631163919,
-      "eval_accuracy_steel": 0.9314173334599449,
-      "eval_accuracy_tiles": 0.23884395164955013,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845280081702599,
-      "eval_iou_REST": 0.40756914119359533,
-      "eval_iou_concrete": 0.8998428903857755,
-      "eval_iou_concrete tiles": 0.892818268654064,
-      "eval_iou_dark bricks": 0.8406699036884187,
-      "eval_iou_grass": 0.8445203417510077,
-      "eval_iou_red bricks": 0.9083532219570406,
-      "eval_iou_road": 0.9622176844974133,
-      "eval_iou_rough concrete": 0.665405331898565,
-      "eval_iou_rough red sidewalk": 0.6109101144019843,
-      "eval_iou_speedway bricks": 0.9806008191986646,
-      "eval_iou_steel": 0.8808811434729588,
-      "eval_iou_tiles": 0.23777067374389288,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08313030004501343,
-      "eval_mean_accuracy": 0.8190615013195683,
-      "eval_mean_iou": 0.7781605802318186,
-      "eval_overall_accuracy": 0.9810934228412176,
-      "eval_runtime": 5.4466,
-      "eval_samples_per_second": 21.665,
-      "eval_steps_per_second": 1.469,
-      "step": 6700
-    },
-    {
-      "epoch": 394.18,
-      "learning_rate": 1.1139318885448916e-05,
-      "loss": 0.014,
-      "step": 6701
-    },
-    {
-      "epoch": 394.24,
-      "learning_rate": 1.1133126934984521e-05,
-      "loss": 0.0164,
-      "step": 6702
-    },
-    {
-      "epoch": 394.29,
-      "learning_rate": 1.1126934984520123e-05,
-      "loss": 0.0132,
-      "step": 6703
-    },
-    {
-      "epoch": 394.35,
-      "learning_rate": 1.1120743034055728e-05,
-      "loss": 0.0157,
-      "step": 6704
-    },
-    {
-      "epoch": 394.41,
-      "learning_rate": 1.1114551083591331e-05,
-      "loss": 0.0139,
-      "step": 6705
-    },
-    {
-      "epoch": 394.47,
-      "learning_rate": 1.1108359133126936e-05,
-      "loss": 0.0158,
-      "step": 6706
-    },
-    {
-      "epoch": 394.53,
-      "learning_rate": 1.1102167182662538e-05,
-      "loss": 0.0179,
-      "step": 6707
-    },
-    {
-      "epoch": 394.59,
-      "learning_rate": 1.1095975232198143e-05,
-      "loss": 0.0164,
-      "step": 6708
-    },
-    {
-      "epoch": 394.65,
-      "learning_rate": 1.1089783281733746e-05,
-      "loss": 0.016,
-      "step": 6709
-    },
-    {
-      "epoch": 394.71,
-      "learning_rate": 1.1083591331269351e-05,
-      "loss": 0.0154,
-      "step": 6710
-    },
-    {
-      "epoch": 394.76,
-      "learning_rate": 1.1077399380804953e-05,
-      "loss": 0.0154,
-      "step": 6711
-    },
-    {
-      "epoch": 394.82,
-      "learning_rate": 1.1071207430340558e-05,
-      "loss": 0.0165,
-      "step": 6712
-    },
-    {
-      "epoch": 394.88,
-      "learning_rate": 1.1065015479876161e-05,
-      "loss": 0.0143,
-      "step": 6713
-    },
-    {
-      "epoch": 394.94,
-      "learning_rate": 1.1058823529411766e-05,
-      "loss": 0.0158,
-      "step": 6714
-    },
-    {
-      "epoch": 395.0,
-      "learning_rate": 1.1052631578947368e-05,
-      "loss": 0.016,
-      "step": 6715
-    },
-    {
-      "epoch": 395.06,
-      "learning_rate": 1.1046439628482973e-05,
-      "loss": 0.0155,
-      "step": 6716
-    },
-    {
-      "epoch": 395.12,
-      "learning_rate": 1.1040247678018576e-05,
-      "loss": 0.0138,
-      "step": 6717
-    },
-    {
-      "epoch": 395.18,
-      "learning_rate": 1.103405572755418e-05,
-      "loss": 0.0177,
-      "step": 6718
-    },
-    {
-      "epoch": 395.24,
-      "learning_rate": 1.1027863777089783e-05,
-      "loss": 0.0169,
-      "step": 6719
-    },
-    {
-      "epoch": 395.29,
-      "learning_rate": 1.1021671826625388e-05,
-      "loss": 0.0167,
-      "step": 6720
-    },
-    {
-      "epoch": 395.29,
-      "eval_accuracy_NAT": 0.9943915597043889,
-      "eval_accuracy_REST": 0.4682926829268293,
-      "eval_accuracy_concrete": 0.9576236529385775,
-      "eval_accuracy_concrete tiles": 0.948126325712939,
-      "eval_accuracy_dark bricks": 0.8525544092031468,
-      "eval_accuracy_grass": 0.89697472057894,
-      "eval_accuracy_red bricks": 0.9416190685413005,
-      "eval_accuracy_road": 0.9816930409934582,
-      "eval_accuracy_rough concrete": 0.6649068341486959,
-      "eval_accuracy_rough red sidewalk": 0.7849810872393386,
-      "eval_accuracy_speedway bricks": 0.9897023250138978,
-      "eval_accuracy_steel": 0.9378995137115355,
-      "eval_accuracy_tiles": 0.24750825532430548,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9844290921527471,
-      "eval_iou_REST": 0.4247787610619469,
-      "eval_iou_concrete": 0.8995917737198498,
-      "eval_iou_concrete tiles": 0.8902387749231008,
-      "eval_iou_dark bricks": 0.838980377013746,
-      "eval_iou_grass": 0.8437274148126787,
-      "eval_iou_red bricks": 0.9091150114003924,
-      "eval_iou_road": 0.962174968140422,
-      "eval_iou_rough concrete": 0.6594915985508017,
-      "eval_iou_rough red sidewalk": 0.61181476640663,
-      "eval_iou_speedway bricks": 0.9807029701440353,
-      "eval_iou_steel": 0.880025604291767,
-      "eval_iou_tiles": 0.24627720504009165,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0841011330485344,
-      "eval_mean_accuracy": 0.8204825750797965,
-      "eval_mean_iou": 0.7793344859737085,
-      "eval_overall_accuracy": 0.9810217517917439,
-      "eval_runtime": 7.9068,
-      "eval_samples_per_second": 14.924,
-      "eval_steps_per_second": 1.012,
-      "step": 6720
-    },
-    {
-      "epoch": 395.35,
-      "learning_rate": 1.1015479876160991e-05,
-      "loss": 0.0136,
-      "step": 6721
-    },
-    {
-      "epoch": 395.41,
-      "learning_rate": 1.1009287925696595e-05,
-      "loss": 0.0149,
-      "step": 6722
-    },
-    {
-      "epoch": 395.47,
-      "learning_rate": 1.1003095975232198e-05,
-      "loss": 0.0149,
-      "step": 6723
-    },
-    {
-      "epoch": 395.53,
-      "learning_rate": 1.0996904024767803e-05,
-      "loss": 0.0173,
-      "step": 6724
-    },
-    {
-      "epoch": 395.59,
-      "learning_rate": 1.0990712074303406e-05,
-      "loss": 0.015,
-      "step": 6725
-    },
-    {
-      "epoch": 395.65,
-      "learning_rate": 1.098452012383901e-05,
-      "loss": 0.0164,
-      "step": 6726
-    },
-    {
-      "epoch": 395.71,
-      "learning_rate": 1.0978328173374613e-05,
-      "loss": 0.0145,
-      "step": 6727
-    },
-    {
-      "epoch": 395.76,
-      "learning_rate": 1.0972136222910218e-05,
-      "loss": 0.0145,
-      "step": 6728
-    },
-    {
-      "epoch": 395.82,
-      "learning_rate": 1.096594427244582e-05,
-      "loss": 0.0138,
-      "step": 6729
-    },
-    {
-      "epoch": 395.88,
-      "learning_rate": 1.0959752321981425e-05,
-      "loss": 0.0143,
-      "step": 6730
-    },
-    {
-      "epoch": 395.94,
-      "learning_rate": 1.0953560371517028e-05,
-      "loss": 0.0194,
-      "step": 6731
-    },
-    {
-      "epoch": 396.0,
-      "learning_rate": 1.0947368421052633e-05,
-      "loss": 0.0168,
-      "step": 6732
-    },
-    {
-      "epoch": 396.06,
-      "learning_rate": 1.0941176470588235e-05,
-      "loss": 0.0146,
-      "step": 6733
-    },
-    {
-      "epoch": 396.12,
-      "learning_rate": 1.093498452012384e-05,
-      "loss": 0.018,
-      "step": 6734
-    },
-    {
-      "epoch": 396.18,
-      "learning_rate": 1.0928792569659443e-05,
-      "loss": 0.0165,
-      "step": 6735
-    },
-    {
-      "epoch": 396.24,
-      "learning_rate": 1.0922600619195048e-05,
-      "loss": 0.0164,
-      "step": 6736
-    },
-    {
-      "epoch": 396.29,
-      "learning_rate": 1.091640866873065e-05,
-      "loss": 0.0178,
-      "step": 6737
-    },
-    {
-      "epoch": 396.35,
-      "learning_rate": 1.0910216718266255e-05,
-      "loss": 0.0134,
-      "step": 6738
-    },
-    {
-      "epoch": 396.41,
-      "learning_rate": 1.0904024767801858e-05,
-      "loss": 0.0161,
-      "step": 6739
-    },
-    {
-      "epoch": 396.47,
-      "learning_rate": 1.0897832817337463e-05,
-      "loss": 0.0158,
-      "step": 6740
-    },
-    {
-      "epoch": 396.47,
-      "eval_accuracy_NAT": 0.9938397728503509,
-      "eval_accuracy_REST": 0.5024390243902439,
-      "eval_accuracy_concrete": 0.9575609722538598,
-      "eval_accuracy_concrete tiles": 0.9484562809333019,
-      "eval_accuracy_dark bricks": 0.8513867886606818,
-      "eval_accuracy_grass": 0.9064398369460309,
-      "eval_accuracy_red bricks": 0.9403009666080844,
-      "eval_accuracy_road": 0.9827752336600931,
-      "eval_accuracy_rough concrete": 0.6661879763316431,
-      "eval_accuracy_rough red sidewalk": 0.7836726099609713,
-      "eval_accuracy_speedway bricks": 0.9907190344875805,
-      "eval_accuracy_steel": 0.9314073378080535,
-      "eval_accuracy_tiles": 0.23790481383865006,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847088485994769,
-      "eval_iou_REST": 0.4433285509325681,
-      "eval_iou_concrete": 0.8993017068590233,
-      "eval_iou_concrete tiles": 0.8935541887781158,
-      "eval_iou_dark bricks": 0.8369169979067197,
-      "eval_iou_grass": 0.8480954944171568,
-      "eval_iou_red bricks": 0.9103041259038707,
-      "eval_iou_road": 0.9623137350940979,
-      "eval_iou_rough concrete": 0.6604656383475215,
-      "eval_iou_rough red sidewalk": 0.6129504696589164,
-      "eval_iou_speedway bricks": 0.9806413781721226,
-      "eval_iou_steel": 0.8806281881351167,
-      "eval_iou_tiles": 0.2366715891624725,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08329641073942184,
-      "eval_mean_accuracy": 0.8225454345176573,
-      "eval_mean_iou": 0.7807600701513214,
-      "eval_overall_accuracy": 0.9811657727774927,
-      "eval_runtime": 5.4003,
-      "eval_samples_per_second": 21.851,
-      "eval_steps_per_second": 1.481,
-      "step": 6740
-    },
-    {
-      "epoch": 396.53,
-      "learning_rate": 1.0891640866873065e-05,
-      "loss": 0.0152,
-      "step": 6741
-    },
-    {
-      "epoch": 396.59,
-      "learning_rate": 1.088544891640867e-05,
-      "loss": 0.0159,
-      "step": 6742
-    },
-    {
-      "epoch": 396.65,
-      "learning_rate": 1.0879256965944273e-05,
-      "loss": 0.0147,
-      "step": 6743
-    },
-    {
-      "epoch": 396.71,
-      "learning_rate": 1.0873065015479876e-05,
-      "loss": 0.0159,
-      "step": 6744
-    },
-    {
-      "epoch": 396.76,
-      "learning_rate": 1.086687306501548e-05,
-      "loss": 0.0148,
-      "step": 6745
-    },
-    {
-      "epoch": 396.82,
-      "learning_rate": 1.0860681114551085e-05,
-      "loss": 0.0163,
-      "step": 6746
-    },
-    {
-      "epoch": 396.88,
-      "learning_rate": 1.0854489164086688e-05,
-      "loss": 0.0149,
-      "step": 6747
-    },
-    {
-      "epoch": 396.94,
-      "learning_rate": 1.0848297213622291e-05,
-      "loss": 0.0173,
-      "step": 6748
-    },
-    {
-      "epoch": 397.0,
-      "learning_rate": 1.0842105263157895e-05,
-      "loss": 0.0149,
-      "step": 6749
-    },
-    {
-      "epoch": 397.06,
-      "learning_rate": 1.08359133126935e-05,
-      "loss": 0.0166,
-      "step": 6750
-    },
-    {
-      "epoch": 397.12,
-      "learning_rate": 1.0829721362229103e-05,
-      "loss": 0.0135,
-      "step": 6751
-    },
-    {
-      "epoch": 397.18,
-      "learning_rate": 1.0823529411764706e-05,
-      "loss": 0.0149,
-      "step": 6752
-    },
-    {
-      "epoch": 397.24,
-      "learning_rate": 1.081733746130031e-05,
-      "loss": 0.0158,
-      "step": 6753
-    },
-    {
-      "epoch": 397.29,
-      "learning_rate": 1.0811145510835915e-05,
-      "loss": 0.0144,
-      "step": 6754
-    },
-    {
-      "epoch": 397.35,
-      "learning_rate": 1.0804953560371518e-05,
-      "loss": 0.0152,
-      "step": 6755
-    },
-    {
-      "epoch": 397.41,
-      "learning_rate": 1.0798761609907121e-05,
-      "loss": 0.0145,
-      "step": 6756
-    },
-    {
-      "epoch": 397.47,
-      "learning_rate": 1.0792569659442725e-05,
-      "loss": 0.0157,
-      "step": 6757
-    },
-    {
-      "epoch": 397.53,
-      "learning_rate": 1.078637770897833e-05,
-      "loss": 0.0165,
-      "step": 6758
-    },
-    {
-      "epoch": 397.59,
-      "learning_rate": 1.0780185758513931e-05,
-      "loss": 0.0153,
-      "step": 6759
-    },
-    {
-      "epoch": 397.65,
-      "learning_rate": 1.0773993808049536e-05,
-      "loss": 0.0183,
-      "step": 6760
-    },
-    {
-      "epoch": 397.65,
-      "eval_accuracy_NAT": 0.9938975662560693,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9579768050890601,
-      "eval_accuracy_concrete tiles": 0.9481498939429649,
-      "eval_accuracy_dark bricks": 0.8494630859636698,
-      "eval_accuracy_grass": 0.9013737334741321,
-      "eval_accuracy_red bricks": 0.9401362038664324,
-      "eval_accuracy_road": 0.9831990256525087,
-      "eval_accuracy_rough concrete": 0.67936288501302,
-      "eval_accuracy_rough red sidewalk": 0.7842817286940044,
-      "eval_accuracy_speedway bricks": 0.9904679845332665,
-      "eval_accuracy_steel": 0.9318321530134391,
-      "eval_accuracy_tiles": 0.20600442303614167,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846244197338738,
-      "eval_iou_REST": 0.42448979591836733,
-      "eval_iou_concrete": 0.9002810229753523,
-      "eval_iou_concrete tiles": 0.8935631469059571,
-      "eval_iou_dark bricks": 0.8335790828066155,
-      "eval_iou_grass": 0.846594105461109,
-      "eval_iou_red bricks": 0.9016592046352383,
-      "eval_iou_road": 0.9628204014431606,
-      "eval_iou_rough concrete": 0.6723675286763188,
-      "eval_iou_rough red sidewalk": 0.6113664341403365,
-      "eval_iou_speedway bricks": 0.979738610617558,
-      "eval_iou_steel": 0.8786259416079376,
-      "eval_iou_tiles": 0.20509727039662193,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08303012698888779,
-      "eval_mean_accuracy": 0.8210355816696493,
-      "eval_mean_iou": 0.7765236127168035,
-      "eval_overall_accuracy": 0.9810729269254006,
-      "eval_runtime": 8.1254,
-      "eval_samples_per_second": 14.522,
-      "eval_steps_per_second": 0.985,
-      "step": 6760
-    },
-    {
-      "epoch": 397.71,
-      "learning_rate": 1.076780185758514e-05,
-      "loss": 0.0174,
-      "step": 6761
-    },
-    {
-      "epoch": 397.76,
-      "learning_rate": 1.0761609907120745e-05,
-      "loss": 0.0166,
-      "step": 6762
-    },
-    {
-      "epoch": 397.82,
-      "learning_rate": 1.0755417956656346e-05,
-      "loss": 0.0137,
-      "step": 6763
-    },
-    {
-      "epoch": 397.88,
-      "learning_rate": 1.0749226006191951e-05,
-      "loss": 0.0167,
-      "step": 6764
-    },
-    {
-      "epoch": 397.94,
-      "learning_rate": 1.0743034055727555e-05,
-      "loss": 0.018,
-      "step": 6765
-    },
-    {
-      "epoch": 398.0,
-      "learning_rate": 1.073684210526316e-05,
-      "loss": 0.0146,
-      "step": 6766
-    },
-    {
-      "epoch": 398.06,
-      "learning_rate": 1.0730650154798761e-05,
-      "loss": 0.014,
-      "step": 6767
-    },
-    {
-      "epoch": 398.12,
-      "learning_rate": 1.0724458204334366e-05,
-      "loss": 0.0137,
-      "step": 6768
-    },
-    {
-      "epoch": 398.18,
-      "learning_rate": 1.071826625386997e-05,
-      "loss": 0.0157,
-      "step": 6769
-    },
-    {
-      "epoch": 398.24,
-      "learning_rate": 1.0712074303405573e-05,
-      "loss": 0.014,
-      "step": 6770
-    },
-    {
-      "epoch": 398.29,
-      "learning_rate": 1.0705882352941176e-05,
-      "loss": 0.0125,
-      "step": 6771
-    },
-    {
-      "epoch": 398.35,
-      "learning_rate": 1.0699690402476781e-05,
-      "loss": 0.0178,
-      "step": 6772
-    },
-    {
-      "epoch": 398.41,
-      "learning_rate": 1.0693498452012385e-05,
-      "loss": 0.0167,
-      "step": 6773
-    },
-    {
-      "epoch": 398.47,
-      "learning_rate": 1.0687306501547988e-05,
-      "loss": 0.0159,
-      "step": 6774
-    },
-    {
-      "epoch": 398.53,
-      "learning_rate": 1.0681114551083591e-05,
-      "loss": 0.0148,
-      "step": 6775
-    },
-    {
-      "epoch": 398.59,
-      "learning_rate": 1.0674922600619196e-05,
-      "loss": 0.0191,
-      "step": 6776
-    },
-    {
-      "epoch": 398.65,
-      "learning_rate": 1.06687306501548e-05,
-      "loss": 0.0146,
-      "step": 6777
-    },
-    {
-      "epoch": 398.71,
-      "learning_rate": 1.0662538699690403e-05,
-      "loss": 0.0141,
-      "step": 6778
-    },
-    {
-      "epoch": 398.76,
-      "learning_rate": 1.0656346749226006e-05,
-      "loss": 0.0142,
-      "step": 6779
-    },
-    {
-      "epoch": 398.82,
-      "learning_rate": 1.0650154798761611e-05,
-      "loss": 0.0155,
-      "step": 6780
-    },
-    {
-      "epoch": 398.82,
-      "eval_accuracy_NAT": 0.994436791670245,
-      "eval_accuracy_REST": 0.4666666666666667,
-      "eval_accuracy_concrete": 0.9597735239845346,
-      "eval_accuracy_concrete tiles": 0.9470893235917982,
-      "eval_accuracy_dark bricks": 0.8478025764217215,
-      "eval_accuracy_grass": 0.8960814815655906,
-      "eval_accuracy_red bricks": 0.9411247803163445,
-      "eval_accuracy_road": 0.9828815824702958,
-      "eval_accuracy_rough concrete": 0.6737079830054999,
-      "eval_accuracy_rough red sidewalk": 0.7680310424954316,
-      "eval_accuracy_speedway bricks": 0.9898027449956234,
-      "eval_accuracy_steel": 0.9287210063622324,
-      "eval_accuracy_tiles": 0.23445121027598534,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846048254803605,
-      "eval_iou_REST": 0.4065155807365439,
-      "eval_iou_concrete": 0.9004236087854182,
-      "eval_iou_concrete tiles": 0.8902895628863239,
-      "eval_iou_dark bricks": 0.8337898382937068,
-      "eval_iou_grass": 0.8455883600387523,
-      "eval_iou_red bricks": 0.9042266898844388,
-      "eval_iou_road": 0.9628421073572386,
-      "eval_iou_rough concrete": 0.6688852208792574,
-      "eval_iou_rough red sidewalk": 0.601686078365529,
-      "eval_iou_speedway bricks": 0.9801851427601714,
-      "eval_iou_steel": 0.8793140569584464,
-      "eval_iou_tiles": 0.233397671753423,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08423535525798798,
-      "eval_mean_accuracy": 0.8177362087555362,
-      "eval_mean_iou": 0.776288364936893,
-      "eval_overall_accuracy": 0.9810994035106594,
-      "eval_runtime": 6.2112,
-      "eval_samples_per_second": 18.998,
-      "eval_steps_per_second": 1.288,
-      "step": 6780
-    },
-    {
-      "epoch": 398.88,
-      "learning_rate": 1.0643962848297215e-05,
-      "loss": 0.0152,
-      "step": 6781
-    },
-    {
-      "epoch": 398.94,
-      "learning_rate": 1.0637770897832818e-05,
-      "loss": 0.0188,
-      "step": 6782
-    },
-    {
-      "epoch": 399.0,
-      "learning_rate": 1.0631578947368421e-05,
-      "loss": 0.0171,
-      "step": 6783
-    },
-    {
-      "epoch": 399.06,
-      "learning_rate": 1.0625386996904026e-05,
-      "loss": 0.0132,
-      "step": 6784
-    },
-    {
-      "epoch": 399.12,
-      "learning_rate": 1.0619195046439628e-05,
-      "loss": 0.0153,
-      "step": 6785
-    },
-    {
-      "epoch": 399.18,
-      "learning_rate": 1.0613003095975233e-05,
-      "loss": 0.0143,
-      "step": 6786
-    },
-    {
-      "epoch": 399.24,
-      "learning_rate": 1.0606811145510836e-05,
-      "loss": 0.016,
-      "step": 6787
-    },
-    {
-      "epoch": 399.29,
-      "learning_rate": 1.0600619195046441e-05,
-      "loss": 0.0171,
-      "step": 6788
-    },
-    {
-      "epoch": 399.35,
-      "learning_rate": 1.0594427244582043e-05,
-      "loss": 0.016,
-      "step": 6789
-    },
-    {
-      "epoch": 399.41,
-      "learning_rate": 1.0588235294117648e-05,
-      "loss": 0.0142,
-      "step": 6790
-    },
-    {
-      "epoch": 399.47,
-      "learning_rate": 1.0582043343653251e-05,
-      "loss": 0.0162,
-      "step": 6791
-    },
-    {
-      "epoch": 399.53,
-      "learning_rate": 1.0575851393188856e-05,
-      "loss": 0.0176,
-      "step": 6792
-    },
-    {
-      "epoch": 399.59,
-      "learning_rate": 1.0569659442724458e-05,
-      "loss": 0.0148,
-      "step": 6793
-    },
-    {
-      "epoch": 399.65,
-      "learning_rate": 1.0563467492260063e-05,
-      "loss": 0.016,
-      "step": 6794
-    },
-    {
-      "epoch": 399.71,
-      "learning_rate": 1.0557275541795666e-05,
-      "loss": 0.0129,
-      "step": 6795
-    },
-    {
-      "epoch": 399.76,
-      "learning_rate": 1.055108359133127e-05,
-      "loss": 0.0154,
-      "step": 6796
-    },
-    {
-      "epoch": 399.82,
-      "learning_rate": 1.0544891640866873e-05,
-      "loss": 0.0152,
-      "step": 6797
-    },
-    {
-      "epoch": 399.88,
-      "learning_rate": 1.0538699690402478e-05,
-      "loss": 0.0155,
-      "step": 6798
-    },
-    {
-      "epoch": 399.94,
-      "learning_rate": 1.0532507739938081e-05,
-      "loss": 0.0163,
-      "step": 6799
-    },
-    {
-      "epoch": 400.0,
-      "learning_rate": 1.0526315789473684e-05,
-      "loss": 0.0149,
-      "step": 6800
-    },
-    {
-      "epoch": 400.0,
-      "eval_accuracy_NAT": 0.993901393990963,
-      "eval_accuracy_REST": 0.5723577235772358,
-      "eval_accuracy_concrete": 0.9578487683245454,
-      "eval_accuracy_concrete tiles": 0.9539948149893943,
-      "eval_accuracy_dark bricks": 0.8526788277855406,
-      "eval_accuracy_grass": 0.9038670436438095,
-      "eval_accuracy_red bricks": 0.944694639718805,
-      "eval_accuracy_road": 0.9838600984073865,
-      "eval_accuracy_rough concrete": 0.6656040138482532,
-      "eval_accuracy_rough red sidewalk": 0.7759871859466533,
-      "eval_accuracy_speedway bricks": 0.9903403174916529,
-      "eval_accuracy_steel": 0.927733935737954,
-      "eval_accuracy_tiles": 0.25702081250568026,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846527708567203,
-      "eval_iou_REST": 0.4631578947368421,
-      "eval_iou_concrete": 0.8995839404084642,
-      "eval_iou_concrete tiles": 0.8909579151259025,
-      "eval_iou_dark bricks": 0.8381831352958332,
-      "eval_iou_grass": 0.846895441351441,
-      "eval_iou_red bricks": 0.9095283417935702,
-      "eval_iou_road": 0.9619248110243585,
-      "eval_iou_rough concrete": 0.6588280397534578,
-      "eval_iou_rough red sidewalk": 0.6028474449527081,
-      "eval_iou_speedway bricks": 0.9805344554069072,
-      "eval_iou_steel": 0.8804856183356734,
-      "eval_iou_tiles": 0.25496619083395944,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08449146896600723,
-      "eval_mean_accuracy": 0.8292222750744518,
-      "eval_mean_iou": 0.7825035384519876,
-      "eval_overall_accuracy": 0.9810903193586964,
-      "eval_runtime": 6.6309,
-      "eval_samples_per_second": 17.796,
-      "eval_steps_per_second": 1.206,
-      "step": 6800
-    },
-    {
-      "epoch": 400.06,
-      "learning_rate": 1.0520123839009288e-05,
-      "loss": 0.0144,
-      "step": 6801
-    },
-    {
-      "epoch": 400.12,
-      "learning_rate": 1.0513931888544893e-05,
-      "loss": 0.0159,
-      "step": 6802
-    },
-    {
-      "epoch": 400.18,
-      "learning_rate": 1.0507739938080496e-05,
-      "loss": 0.0163,
-      "step": 6803
-    },
-    {
-      "epoch": 400.24,
-      "learning_rate": 1.05015479876161e-05,
-      "loss": 0.0145,
-      "step": 6804
-    },
-    {
-      "epoch": 400.29,
-      "learning_rate": 1.0495356037151703e-05,
-      "loss": 0.0132,
-      "step": 6805
-    },
-    {
-      "epoch": 400.35,
-      "learning_rate": 1.0489164086687308e-05,
-      "loss": 0.017,
-      "step": 6806
-    },
-    {
-      "epoch": 400.41,
-      "learning_rate": 1.0482972136222911e-05,
-      "loss": 0.0183,
-      "step": 6807
-    },
-    {
-      "epoch": 400.47,
-      "learning_rate": 1.0476780185758514e-05,
-      "loss": 0.0147,
-      "step": 6808
-    },
-    {
-      "epoch": 400.53,
-      "learning_rate": 1.0470588235294118e-05,
-      "loss": 0.0162,
-      "step": 6809
-    },
-    {
-      "epoch": 400.59,
-      "learning_rate": 1.0464396284829723e-05,
-      "loss": 0.0164,
-      "step": 6810
-    },
-    {
-      "epoch": 400.65,
-      "learning_rate": 1.0458204334365324e-05,
-      "loss": 0.0158,
-      "step": 6811
-    },
-    {
-      "epoch": 400.71,
-      "learning_rate": 1.045201238390093e-05,
-      "loss": 0.0151,
-      "step": 6812
-    },
-    {
-      "epoch": 400.76,
-      "learning_rate": 1.0445820433436533e-05,
-      "loss": 0.0183,
-      "step": 6813
-    },
-    {
-      "epoch": 400.82,
-      "learning_rate": 1.0439628482972138e-05,
-      "loss": 0.0157,
-      "step": 6814
-    },
-    {
-      "epoch": 400.88,
-      "learning_rate": 1.043343653250774e-05,
-      "loss": 0.0165,
-      "step": 6815
-    },
-    {
-      "epoch": 400.94,
-      "learning_rate": 1.0427244582043344e-05,
-      "loss": 0.0152,
-      "step": 6816
-    },
-    {
-      "epoch": 401.0,
-      "learning_rate": 1.0421052631578948e-05,
-      "loss": 0.0179,
-      "step": 6817
-    },
-    {
-      "epoch": 401.06,
-      "learning_rate": 1.0414860681114553e-05,
-      "loss": 0.0127,
-      "step": 6818
-    },
-    {
-      "epoch": 401.12,
-      "learning_rate": 1.0408668730650154e-05,
-      "loss": 0.0172,
-      "step": 6819
-    },
-    {
-      "epoch": 401.18,
-      "learning_rate": 1.040247678018576e-05,
-      "loss": 0.0184,
-      "step": 6820
-    },
-    {
-      "epoch": 401.18,
-      "eval_accuracy_NAT": 0.9940933198532371,
-      "eval_accuracy_REST": 0.510569105691057,
-      "eval_accuracy_concrete": 0.9596405186291581,
-      "eval_accuracy_concrete tiles": 0.9530049493283055,
-      "eval_accuracy_dark bricks": 0.8510948835250656,
-      "eval_accuracy_grass": 0.8991008943744867,
-      "eval_accuracy_red bricks": 0.9412895430579965,
-      "eval_accuracy_road": 0.9818351956844829,
-      "eval_accuracy_rough concrete": 0.662827212651726,
-      "eval_accuracy_rough red sidewalk": 0.7708359966611269,
-      "eval_accuracy_speedway bricks": 0.9899610864002555,
-      "eval_accuracy_steel": 0.934033695342526,
-      "eval_accuracy_tiles": 0.25053773213366054,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845024925347825,
-      "eval_iou_REST": 0.4248985115020298,
-      "eval_iou_concrete": 0.8991449514462628,
-      "eval_iou_concrete tiles": 0.8904254382101647,
-      "eval_iou_dark bricks": 0.8381598231831742,
-      "eval_iou_grass": 0.8444721532464624,
-      "eval_iou_red bricks": 0.9130087364159386,
-      "eval_iou_road": 0.9632047704879013,
-      "eval_iou_rough concrete": 0.6572929469603858,
-      "eval_iou_rough red sidewalk": 0.6001744812403391,
-      "eval_iou_speedway bricks": 0.9806767844708082,
-      "eval_iou_steel": 0.8808429109744803,
-      "eval_iou_tiles": 0.2489239382355597,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08453638106584549,
-      "eval_mean_accuracy": 0.8229864717948525,
-      "eval_mean_iou": 0.7789021491467915,
-      "eval_overall_accuracy": 0.981022172055002,
-      "eval_runtime": 5.7212,
-      "eval_samples_per_second": 20.625,
-      "eval_steps_per_second": 1.398,
-      "step": 6820
-    },
-    {
-      "epoch": 401.24,
-      "learning_rate": 1.0396284829721363e-05,
-      "loss": 0.013,
-      "step": 6821
-    },
-    {
-      "epoch": 401.29,
-      "learning_rate": 1.0390092879256968e-05,
-      "loss": 0.0172,
-      "step": 6822
-    },
-    {
-      "epoch": 401.35,
-      "learning_rate": 1.038390092879257e-05,
-      "loss": 0.0166,
-      "step": 6823
-    },
-    {
-      "epoch": 401.41,
-      "learning_rate": 1.0377708978328174e-05,
-      "loss": 0.0154,
-      "step": 6824
-    },
-    {
-      "epoch": 401.47,
-      "learning_rate": 1.0371517027863778e-05,
-      "loss": 0.0127,
-      "step": 6825
-    },
-    {
-      "epoch": 401.53,
-      "learning_rate": 1.0365325077399381e-05,
-      "loss": 0.0155,
-      "step": 6826
-    },
-    {
-      "epoch": 401.59,
-      "learning_rate": 1.0359133126934984e-05,
-      "loss": 0.0159,
-      "step": 6827
-    },
-    {
-      "epoch": 401.65,
-      "learning_rate": 1.035294117647059e-05,
-      "loss": 0.0152,
-      "step": 6828
-    },
-    {
-      "epoch": 401.71,
-      "learning_rate": 1.0346749226006193e-05,
-      "loss": 0.0165,
-      "step": 6829
-    },
-    {
-      "epoch": 401.76,
-      "learning_rate": 1.0340557275541796e-05,
-      "loss": 0.0139,
-      "step": 6830
-    },
-    {
-      "epoch": 401.82,
-      "learning_rate": 1.03343653250774e-05,
-      "loss": 0.0149,
-      "step": 6831
-    },
-    {
-      "epoch": 401.88,
-      "learning_rate": 1.0328173374613004e-05,
-      "loss": 0.0195,
-      "step": 6832
-    },
-    {
-      "epoch": 401.94,
-      "learning_rate": 1.0321981424148608e-05,
-      "loss": 0.0133,
-      "step": 6833
-    },
-    {
-      "epoch": 402.0,
-      "learning_rate": 1.0315789473684211e-05,
-      "loss": 0.0158,
-      "step": 6834
-    },
-    {
-      "epoch": 402.06,
-      "learning_rate": 1.0309597523219814e-05,
-      "loss": 0.0162,
-      "step": 6835
-    },
-    {
-      "epoch": 402.12,
-      "learning_rate": 1.030340557275542e-05,
-      "loss": 0.0138,
-      "step": 6836
-    },
-    {
-      "epoch": 402.18,
-      "learning_rate": 1.0297213622291021e-05,
-      "loss": 0.0178,
-      "step": 6837
-    },
-    {
-      "epoch": 402.24,
-      "learning_rate": 1.0291021671826626e-05,
-      "loss": 0.0165,
-      "step": 6838
-    },
-    {
-      "epoch": 402.29,
-      "learning_rate": 1.028482972136223e-05,
-      "loss": 0.0149,
-      "step": 6839
-    },
-    {
-      "epoch": 402.35,
-      "learning_rate": 1.0278637770897834e-05,
-      "loss": 0.0162,
-      "step": 6840
-    },
-    {
-      "epoch": 402.35,
-      "eval_accuracy_NAT": 0.9938259175282709,
-      "eval_accuracy_REST": 0.48292682926829267,
-      "eval_accuracy_concrete": 0.9589372719225695,
-      "eval_accuracy_concrete tiles": 0.9418807447560689,
-      "eval_accuracy_dark bricks": 0.8546838810941179,
-      "eval_accuracy_grass": 0.9035245091916565,
-      "eval_accuracy_red bricks": 0.9432117750439367,
-      "eval_accuracy_road": 0.9837815392360307,
-      "eval_accuracy_rough concrete": 0.6466073567355305,
-      "eval_accuracy_rough red sidewalk": 0.7756262266974485,
-      "eval_accuracy_speedway bricks": 0.9899470344574202,
-      "eval_accuracy_steel": 0.9319471030101906,
-      "eval_accuracy_tiles": 0.2384804144324275,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846487305192063,
-      "eval_iou_REST": 0.4119278779472954,
-      "eval_iou_concrete": 0.8981676148923398,
-      "eval_iou_concrete tiles": 0.8876352086711237,
-      "eval_iou_dark bricks": 0.8400948255182242,
-      "eval_iou_grass": 0.8459384994285816,
-      "eval_iou_red bricks": 0.9112325569056083,
-      "eval_iou_road": 0.9626502659657915,
-      "eval_iou_rough concrete": 0.6418380987306732,
-      "eval_iou_rough red sidewalk": 0.600360884749709,
-      "eval_iou_speedway bricks": 0.9806322337997345,
-      "eval_iou_steel": 0.8811239533521085,
-      "eval_iou_tiles": 0.23735150455285534,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0856589674949646,
-      "eval_mean_accuracy": 0.8188754310287661,
-      "eval_mean_iou": 0.7756617119256348,
-      "eval_overall_accuracy": 0.9809708676095736,
-      "eval_runtime": 5.8446,
-      "eval_samples_per_second": 20.19,
-      "eval_steps_per_second": 1.369,
-      "step": 6840
-    },
-    {
-      "epoch": 402.41,
-      "learning_rate": 1.0272445820433436e-05,
-      "loss": 0.0165,
-      "step": 6841
-    },
-    {
-      "epoch": 402.47,
-      "learning_rate": 1.0266253869969041e-05,
-      "loss": 0.0141,
-      "step": 6842
-    },
-    {
-      "epoch": 402.53,
-      "learning_rate": 1.0260061919504644e-05,
-      "loss": 0.0148,
-      "step": 6843
-    },
-    {
-      "epoch": 402.59,
-      "learning_rate": 1.025386996904025e-05,
-      "loss": 0.0139,
-      "step": 6844
-    },
-    {
-      "epoch": 402.65,
-      "learning_rate": 1.0247678018575851e-05,
-      "loss": 0.0166,
-      "step": 6845
-    },
-    {
-      "epoch": 402.71,
-      "learning_rate": 1.0241486068111456e-05,
-      "loss": 0.0155,
-      "step": 6846
-    },
-    {
-      "epoch": 402.76,
-      "learning_rate": 1.023529411764706e-05,
-      "loss": 0.0155,
-      "step": 6847
-    },
-    {
-      "epoch": 402.82,
-      "learning_rate": 1.0229102167182664e-05,
-      "loss": 0.0155,
-      "step": 6848
-    },
-    {
-      "epoch": 402.88,
-      "learning_rate": 1.0222910216718266e-05,
-      "loss": 0.0142,
-      "step": 6849
-    },
-    {
-      "epoch": 402.94,
-      "learning_rate": 1.0216718266253871e-05,
-      "loss": 0.0129,
-      "step": 6850
-    },
-    {
-      "epoch": 403.0,
-      "learning_rate": 1.0210526315789474e-05,
-      "loss": 0.0149,
-      "step": 6851
-    },
-    {
-      "epoch": 403.06,
-      "learning_rate": 1.0204334365325078e-05,
-      "loss": 0.0146,
-      "step": 6852
-    },
-    {
-      "epoch": 403.12,
-      "learning_rate": 1.0198142414860681e-05,
-      "loss": 0.0134,
-      "step": 6853
-    },
-    {
-      "epoch": 403.18,
-      "learning_rate": 1.0191950464396286e-05,
-      "loss": 0.0182,
-      "step": 6854
-    },
-    {
-      "epoch": 403.24,
-      "learning_rate": 1.018575851393189e-05,
-      "loss": 0.0145,
-      "step": 6855
-    },
-    {
-      "epoch": 403.29,
-      "learning_rate": 1.0179566563467493e-05,
-      "loss": 0.0159,
-      "step": 6856
-    },
-    {
-      "epoch": 403.35,
-      "learning_rate": 1.0173374613003096e-05,
-      "loss": 0.0155,
-      "step": 6857
-    },
-    {
-      "epoch": 403.41,
-      "learning_rate": 1.0167182662538701e-05,
-      "loss": 0.0139,
-      "step": 6858
-    },
-    {
-      "epoch": 403.47,
-      "learning_rate": 1.0160990712074304e-05,
-      "loss": 0.0173,
-      "step": 6859
-    },
-    {
-      "epoch": 403.53,
-      "learning_rate": 1.0154798761609908e-05,
-      "loss": 0.0157,
-      "step": 6860
-    },
-    {
-      "epoch": 403.53,
-      "eval_accuracy_NAT": 0.9942116022526274,
-      "eval_accuracy_REST": 0.49105691056910566,
-      "eval_accuracy_concrete": 0.9586078161285045,
-      "eval_accuracy_concrete tiles": 0.9476549611124204,
-      "eval_accuracy_dark bricks": 0.8567607143540762,
-      "eval_accuracy_grass": 0.8979938078854838,
-      "eval_accuracy_red bricks": 0.9431568541300527,
-      "eval_accuracy_road": 0.9830365631484805,
-      "eval_accuracy_rough concrete": 0.6455705253874711,
-      "eval_accuracy_rough red sidewalk": 0.7648726490648899,
-      "eval_accuracy_speedway bricks": 0.9900565367924419,
-      "eval_accuracy_steel": 0.9346509268468216,
-      "eval_accuracy_tiles": 0.2415401860098761,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846503884388991,
-      "eval_iou_REST": 0.42595204513399154,
-      "eval_iou_concrete": 0.8979596512897624,
-      "eval_iou_concrete tiles": 0.8896977474886046,
-      "eval_iou_dark bricks": 0.8418550631959073,
-      "eval_iou_grass": 0.8453180057790082,
-      "eval_iou_red bricks": 0.9091534755677908,
-      "eval_iou_road": 0.9627580453194627,
-      "eval_iou_rough concrete": 0.6403279095470853,
-      "eval_iou_rough red sidewalk": 0.5963939136298337,
-      "eval_iou_speedway bricks": 0.98070258256833,
-      "eval_iou_steel": 0.8806052729473035,
-      "eval_iou_tiles": 0.2402374352175485,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08549767732620239,
-      "eval_mean_accuracy": 0.8191669272063269,
-      "eval_mean_iou": 0.776585502778733,
-      "eval_overall_accuracy": 0.9809693805241989,
-      "eval_runtime": 7.5729,
-      "eval_samples_per_second": 15.582,
-      "eval_steps_per_second": 1.056,
-      "step": 6860
-    },
-    {
-      "epoch": 403.59,
-      "learning_rate": 1.0148606811145511e-05,
-      "loss": 0.0159,
-      "step": 6861
-    },
-    {
-      "epoch": 403.65,
-      "learning_rate": 1.0142414860681116e-05,
-      "loss": 0.0145,
-      "step": 6862
-    },
-    {
-      "epoch": 403.71,
-      "learning_rate": 1.013622291021672e-05,
-      "loss": 0.0146,
-      "step": 6863
-    },
-    {
-      "epoch": 403.76,
-      "learning_rate": 1.0130030959752323e-05,
-      "loss": 0.015,
-      "step": 6864
-    },
-    {
-      "epoch": 403.82,
-      "learning_rate": 1.0123839009287926e-05,
-      "loss": 0.0156,
-      "step": 6865
-    },
-    {
-      "epoch": 403.88,
-      "learning_rate": 1.0117647058823531e-05,
-      "loss": 0.0149,
-      "step": 6866
-    },
-    {
-      "epoch": 403.94,
-      "learning_rate": 1.0111455108359133e-05,
-      "loss": 0.0171,
-      "step": 6867
-    },
-    {
-      "epoch": 404.0,
-      "learning_rate": 1.0105263157894738e-05,
-      "loss": 0.0145,
-      "step": 6868
-    },
-    {
-      "epoch": 404.06,
-      "learning_rate": 1.0099071207430341e-05,
-      "loss": 0.0158,
-      "step": 6869
-    },
-    {
-      "epoch": 404.12,
-      "learning_rate": 1.0092879256965946e-05,
-      "loss": 0.0163,
-      "step": 6870
-    },
-    {
-      "epoch": 404.18,
-      "learning_rate": 1.0086687306501548e-05,
-      "loss": 0.0171,
-      "step": 6871
-    },
-    {
-      "epoch": 404.24,
-      "learning_rate": 1.0080495356037153e-05,
-      "loss": 0.0174,
-      "step": 6872
-    },
-    {
-      "epoch": 404.29,
-      "learning_rate": 1.0074303405572756e-05,
-      "loss": 0.0173,
-      "step": 6873
-    },
-    {
-      "epoch": 404.35,
-      "learning_rate": 1.0068111455108361e-05,
-      "loss": 0.0131,
-      "step": 6874
-    },
-    {
-      "epoch": 404.41,
-      "learning_rate": 1.0061919504643963e-05,
-      "loss": 0.0131,
-      "step": 6875
-    },
-    {
-      "epoch": 404.47,
-      "learning_rate": 1.0055727554179568e-05,
-      "loss": 0.0145,
-      "step": 6876
-    },
-    {
-      "epoch": 404.53,
-      "learning_rate": 1.004953560371517e-05,
-      "loss": 0.0144,
-      "step": 6877
-    },
-    {
-      "epoch": 404.59,
-      "learning_rate": 1.0043343653250774e-05,
-      "loss": 0.0138,
-      "step": 6878
-    },
-    {
-      "epoch": 404.65,
-      "learning_rate": 1.0037151702786377e-05,
-      "loss": 0.0144,
-      "step": 6879
-    },
-    {
-      "epoch": 404.71,
-      "learning_rate": 1.0030959752321983e-05,
-      "loss": 0.0167,
-      "step": 6880
-    },
-    {
-      "epoch": 404.71,
-      "eval_accuracy_NAT": 0.9941699284628696,
-      "eval_accuracy_REST": 0.5008130081300813,
-      "eval_accuracy_concrete": 0.9585879417650575,
-      "eval_accuracy_concrete tiles": 0.9479142116427056,
-      "eval_accuracy_dark bricks": 0.8534684072507321,
-      "eval_accuracy_grass": 0.9024902822408187,
-      "eval_accuracy_red bricks": 0.9388181019332161,
-      "eval_accuracy_road": 0.9831290171392597,
-      "eval_accuracy_rough concrete": 0.6442059599926111,
-      "eval_accuracy_rough red sidewalk": 0.7756187067130901,
-      "eval_accuracy_speedway bricks": 0.9901864315810903,
-      "eval_accuracy_steel": 0.9307001454367351,
-      "eval_accuracy_tiles": 0.23899542549001787,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846823530039968,
-      "eval_iou_REST": 0.4368794326241135,
-      "eval_iou_concrete": 0.8988256526570912,
-      "eval_iou_concrete tiles": 0.8913019390581718,
-      "eval_iou_dark bricks": 0.8400815818974856,
-      "eval_iou_grass": 0.8470714139808343,
-      "eval_iou_red bricks": 0.9088202456270934,
-      "eval_iou_road": 0.9631424830277624,
-      "eval_iou_rough concrete": 0.6388762491209616,
-      "eval_iou_rough red sidewalk": 0.6058706736530464,
-      "eval_iou_speedway bricks": 0.9807245401738555,
-      "eval_iou_steel": 0.8810878457181925,
-      "eval_iou_tiles": 0.23779238003375935,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08468903601169586,
-      "eval_mean_accuracy": 0.8199305821367913,
-      "eval_mean_iou": 0.7780889838904896,
-      "eval_overall_accuracy": 0.9810871512202893,
-      "eval_runtime": 5.7479,
-      "eval_samples_per_second": 20.529,
-      "eval_steps_per_second": 1.392,
-      "step": 6880
-    },
-    {
-      "epoch": 404.76,
-      "learning_rate": 1.0024767801857586e-05,
-      "loss": 0.0161,
-      "step": 6881
-    },
-    {
-      "epoch": 404.82,
-      "learning_rate": 1.001857585139319e-05,
-      "loss": 0.016,
-      "step": 6882
-    },
-    {
-      "epoch": 404.88,
-      "learning_rate": 1.0012383900928792e-05,
-      "loss": 0.0159,
-      "step": 6883
-    },
-    {
-      "epoch": 404.94,
-      "learning_rate": 1.0006191950464397e-05,
-      "loss": 0.0134,
-      "step": 6884
-    },
-    {
-      "epoch": 405.0,
-      "learning_rate": 1e-05,
-      "loss": 0.0154,
-      "step": 6885
-    },
-    {
-      "epoch": 405.06,
-      "learning_rate": 9.993808049535604e-06,
-      "loss": 0.0156,
-      "step": 6886
-    },
-    {
-      "epoch": 405.12,
-      "learning_rate": 9.987616099071207e-06,
-      "loss": 0.016,
-      "step": 6887
-    },
-    {
-      "epoch": 405.18,
-      "learning_rate": 9.981424148606812e-06,
-      "loss": 0.0153,
-      "step": 6888
-    },
-    {
-      "epoch": 405.24,
-      "learning_rate": 9.975232198142416e-06,
-      "loss": 0.0154,
-      "step": 6889
-    },
-    {
-      "epoch": 405.29,
-      "learning_rate": 9.969040247678019e-06,
-      "loss": 0.0137,
-      "step": 6890
-    },
-    {
-      "epoch": 405.35,
-      "learning_rate": 9.962848297213622e-06,
-      "loss": 0.0146,
-      "step": 6891
-    },
-    {
-      "epoch": 405.41,
-      "learning_rate": 9.956656346749227e-06,
-      "loss": 0.0145,
-      "step": 6892
-    },
-    {
-      "epoch": 405.47,
-      "learning_rate": 9.950464396284829e-06,
-      "loss": 0.0203,
-      "step": 6893
-    },
-    {
-      "epoch": 405.53,
-      "learning_rate": 9.944272445820434e-06,
-      "loss": 0.0156,
-      "step": 6894
-    },
-    {
-      "epoch": 405.59,
-      "learning_rate": 9.938080495356037e-06,
-      "loss": 0.0153,
-      "step": 6895
-    },
-    {
-      "epoch": 405.65,
-      "learning_rate": 9.931888544891642e-06,
-      "loss": 0.0164,
-      "step": 6896
-    },
-    {
-      "epoch": 405.71,
-      "learning_rate": 9.925696594427244e-06,
-      "loss": 0.0135,
-      "step": 6897
-    },
-    {
-      "epoch": 405.76,
-      "learning_rate": 9.919504643962849e-06,
-      "loss": 0.0164,
-      "step": 6898
-    },
-    {
-      "epoch": 405.82,
-      "learning_rate": 9.913312693498452e-06,
-      "loss": 0.0126,
-      "step": 6899
-    },
-    {
-      "epoch": 405.88,
-      "learning_rate": 9.907120743034057e-06,
-      "loss": 0.0145,
-      "step": 6900
-    },
-    {
-      "epoch": 405.88,
-      "eval_accuracy_NAT": 0.9942037850475628,
-      "eval_accuracy_REST": 0.5235772357723577,
-      "eval_accuracy_concrete": 0.9569039716622153,
-      "eval_accuracy_concrete tiles": 0.9531699269384869,
-      "eval_accuracy_dark bricks": 0.8571913940623624,
-      "eval_accuracy_grass": 0.9015894734053224,
-      "eval_accuracy_red bricks": 0.9433765377855887,
-      "eval_accuracy_road": 0.9826181153173817,
-      "eval_accuracy_rough concrete": 0.6595200781794671,
-      "eval_accuracy_rough red sidewalk": 0.7626016137886433,
-      "eval_accuracy_speedway bricks": 0.990836933715272,
-      "eval_accuracy_steel": 0.9292032965659938,
-      "eval_accuracy_tiles": 0.25184040716168316,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846324998654503,
-      "eval_iou_REST": 0.44660194174757284,
-      "eval_iou_concrete": 0.8989779991360946,
-      "eval_iou_concrete tiles": 0.8928602967149417,
-      "eval_iou_dark bricks": 0.8422742896371425,
-      "eval_iou_grass": 0.846711702585211,
-      "eval_iou_red bricks": 0.9122145512480085,
-      "eval_iou_road": 0.9625950985381054,
-      "eval_iou_rough concrete": 0.6513155184690437,
-      "eval_iou_rough red sidewalk": 0.5981161788037677,
-      "eval_iou_speedway bricks": 0.9806364853790379,
-      "eval_iou_steel": 0.8808815377399165,
-      "eval_iou_tiles": 0.2502408187838651,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0844673290848732,
-      "eval_mean_accuracy": 0.8235871361078723,
-      "eval_mean_iou": 0.780619916819089,
-      "eval_overall_accuracy": 0.9810830132435944,
-      "eval_runtime": 5.4674,
-      "eval_samples_per_second": 21.582,
-      "eval_steps_per_second": 1.463,
-      "step": 6900
-    },
-    {
-      "epoch": 405.94,
-      "learning_rate": 9.900928792569659e-06,
-      "loss": 0.0161,
-      "step": 6901
-    },
-    {
-      "epoch": 406.0,
-      "learning_rate": 9.894736842105264e-06,
-      "loss": 0.0141,
-      "step": 6902
-    },
-    {
-      "epoch": 406.06,
-      "learning_rate": 9.888544891640867e-06,
-      "loss": 0.0147,
-      "step": 6903
-    },
-    {
-      "epoch": 406.12,
-      "learning_rate": 9.88235294117647e-06,
-      "loss": 0.0199,
-      "step": 6904
-    },
-    {
-      "epoch": 406.18,
-      "learning_rate": 9.876160990712074e-06,
-      "loss": 0.0158,
-      "step": 6905
-    },
-    {
-      "epoch": 406.24,
-      "learning_rate": 9.869969040247679e-06,
-      "loss": 0.0137,
-      "step": 6906
-    },
-    {
-      "epoch": 406.29,
-      "learning_rate": 9.863777089783282e-06,
-      "loss": 0.0138,
-      "step": 6907
-    },
-    {
-      "epoch": 406.35,
-      "learning_rate": 9.857585139318886e-06,
-      "loss": 0.0151,
-      "step": 6908
-    },
-    {
-      "epoch": 406.41,
-      "learning_rate": 9.851393188854489e-06,
-      "loss": 0.0141,
-      "step": 6909
-    },
-    {
-      "epoch": 406.47,
-      "learning_rate": 9.845201238390094e-06,
-      "loss": 0.0158,
-      "step": 6910
-    },
-    {
-      "epoch": 406.53,
-      "learning_rate": 9.839009287925697e-06,
-      "loss": 0.0148,
-      "step": 6911
-    },
-    {
-      "epoch": 406.59,
-      "learning_rate": 9.8328173374613e-06,
-      "loss": 0.0167,
-      "step": 6912
-    },
-    {
-      "epoch": 406.65,
-      "learning_rate": 9.826625386996904e-06,
-      "loss": 0.0139,
-      "step": 6913
-    },
-    {
-      "epoch": 406.71,
-      "learning_rate": 9.820433436532509e-06,
-      "loss": 0.0178,
-      "step": 6914
-    },
-    {
-      "epoch": 406.76,
-      "learning_rate": 9.814241486068112e-06,
-      "loss": 0.0137,
-      "step": 6915
-    },
-    {
-      "epoch": 406.82,
-      "learning_rate": 9.808049535603716e-06,
-      "loss": 0.0154,
-      "step": 6916
-    },
-    {
-      "epoch": 406.88,
-      "learning_rate": 9.801857585139319e-06,
-      "loss": 0.0154,
-      "step": 6917
-    },
-    {
-      "epoch": 406.94,
-      "learning_rate": 9.795665634674924e-06,
-      "loss": 0.0173,
-      "step": 6918
-    },
-    {
-      "epoch": 407.0,
-      "learning_rate": 9.789473684210526e-06,
-      "loss": 0.0173,
-      "step": 6919
-    },
-    {
-      "epoch": 407.06,
-      "learning_rate": 9.78328173374613e-06,
-      "loss": 0.0154,
-      "step": 6920
-    },
-    {
-      "epoch": 407.06,
-      "eval_accuracy_NAT": 0.9940379524766766,
-      "eval_accuracy_REST": 0.5024390243902439,
-      "eval_accuracy_concrete": 0.958015407218063,
-      "eval_accuracy_concrete tiles": 0.9445439547489983,
-      "eval_accuracy_dark bricks": 0.85496621556955,
-      "eval_accuracy_grass": 0.901561086572271,
-      "eval_accuracy_red bricks": 0.9383238137082601,
-      "eval_accuracy_road": 0.9826977433210008,
-      "eval_accuracy_rough concrete": 0.6557183632365823,
-      "eval_accuracy_rough red sidewalk": 0.7667225652170644,
-      "eval_accuracy_speedway bricks": 0.9908763477012735,
-      "eval_accuracy_steel": 0.9291957998270752,
-      "eval_accuracy_tiles": 0.2376927504619952,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846041590894127,
-      "eval_iou_REST": 0.43156424581005587,
-      "eval_iou_concrete": 0.8985451345262143,
-      "eval_iou_concrete tiles": 0.8915509877202349,
-      "eval_iou_dark bricks": 0.8409063097154368,
-      "eval_iou_grass": 0.8463804343674188,
-      "eval_iou_red bricks": 0.9098897587474037,
-      "eval_iou_road": 0.9625294897584759,
-      "eval_iou_rough concrete": 0.649633097390062,
-      "eval_iou_rough red sidewalk": 0.5998482111864826,
-      "eval_iou_speedway bricks": 0.9807001101077265,
-      "eval_iou_steel": 0.8818270211302678,
-      "eval_iou_tiles": 0.23644637314287437,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08464083075523376,
-      "eval_mean_accuracy": 0.8197531557268504,
-      "eval_mean_iou": 0.7780327178993898,
-      "eval_overall_accuracy": 0.9810405019986428,
-      "eval_runtime": 5.8302,
-      "eval_samples_per_second": 20.24,
-      "eval_steps_per_second": 1.372,
-      "step": 6920
-    },
-    {
-      "epoch": 407.12,
-      "learning_rate": 9.777089783281734e-06,
-      "loss": 0.0157,
-      "step": 6921
-    },
-    {
-      "epoch": 407.18,
-      "learning_rate": 9.770897832817339e-06,
-      "loss": 0.0149,
-      "step": 6922
-    },
-    {
-      "epoch": 407.24,
-      "learning_rate": 9.76470588235294e-06,
-      "loss": 0.0152,
-      "step": 6923
-    },
-    {
-      "epoch": 407.29,
-      "learning_rate": 9.758513931888546e-06,
-      "loss": 0.0154,
-      "step": 6924
-    },
-    {
-      "epoch": 407.35,
-      "learning_rate": 9.752321981424149e-06,
-      "loss": 0.0149,
-      "step": 6925
-    },
-    {
-      "epoch": 407.41,
-      "learning_rate": 9.746130030959754e-06,
-      "loss": 0.0158,
-      "step": 6926
-    },
-    {
-      "epoch": 407.47,
-      "learning_rate": 9.739938080495356e-06,
-      "loss": 0.0138,
-      "step": 6927
-    },
-    {
-      "epoch": 407.53,
-      "learning_rate": 9.73374613003096e-06,
-      "loss": 0.0124,
-      "step": 6928
-    },
-    {
-      "epoch": 407.59,
-      "learning_rate": 9.727554179566564e-06,
-      "loss": 0.0186,
-      "step": 6929
-    },
-    {
-      "epoch": 407.65,
-      "learning_rate": 9.721362229102169e-06,
-      "loss": 0.0144,
-      "step": 6930
-    },
-    {
-      "epoch": 407.71,
-      "learning_rate": 9.71517027863777e-06,
-      "loss": 0.0163,
-      "step": 6931
-    },
-    {
-      "epoch": 407.76,
-      "learning_rate": 9.708978328173376e-06,
-      "loss": 0.0161,
-      "step": 6932
-    },
-    {
-      "epoch": 407.82,
-      "learning_rate": 9.702786377708979e-06,
-      "loss": 0.0155,
-      "step": 6933
-    },
-    {
-      "epoch": 407.88,
-      "learning_rate": 9.696594427244582e-06,
-      "loss": 0.0137,
-      "step": 6934
-    },
-    {
-      "epoch": 407.94,
-      "learning_rate": 9.690402476780186e-06,
-      "loss": 0.0171,
-      "step": 6935
-    },
-    {
-      "epoch": 408.0,
-      "learning_rate": 9.68421052631579e-06,
-      "loss": 0.0152,
-      "step": 6936
-    },
-    {
-      "epoch": 408.06,
-      "learning_rate": 9.678018575851394e-06,
-      "loss": 0.0123,
-      "step": 6937
-    },
-    {
-      "epoch": 408.12,
-      "learning_rate": 9.671826625386997e-06,
-      "loss": 0.0143,
-      "step": 6938
-    },
-    {
-      "epoch": 408.18,
-      "learning_rate": 9.6656346749226e-06,
-      "loss": 0.0151,
-      "step": 6939
-    },
-    {
-      "epoch": 408.24,
-      "learning_rate": 9.659442724458206e-06,
-      "loss": 0.0159,
-      "step": 6940
-    },
-    {
-      "epoch": 408.24,
-      "eval_accuracy_NAT": 0.9942183412225106,
-      "eval_accuracy_REST": 0.5252032520325203,
-      "eval_accuracy_concrete": 0.9588940834020018,
-      "eval_accuracy_concrete tiles": 0.9548668395003536,
-      "eval_accuracy_dark bricks": 0.8535114752215608,
-      "eval_accuracy_grass": 0.9005902568819145,
-      "eval_accuracy_red bricks": 0.945738137082601,
-      "eval_accuracy_road": 0.9823033442158272,
-      "eval_accuracy_rough concrete": 0.648722731037606,
-      "eval_accuracy_rough red sidewalk": 0.7608644974018454,
-      "eval_accuracy_speedway bricks": 0.9905829705532969,
-      "eval_accuracy_steel": 0.9285685726708882,
-      "eval_accuracy_tiles": 0.2400254476051986,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847057486118258,
-      "eval_iou_REST": 0.4511173184357542,
-      "eval_iou_concrete": 0.898588767734421,
-      "eval_iou_concrete tiles": 0.8927154944473823,
-      "eval_iou_dark bricks": 0.8395308116656939,
-      "eval_iou_grass": 0.8474647353077881,
-      "eval_iou_red bricks": 0.9077490774907749,
-      "eval_iou_road": 0.9620530433562862,
-      "eval_iou_rough concrete": 0.6431428149460936,
-      "eval_iou_rough red sidewalk": 0.5972504250094447,
-      "eval_iou_speedway bricks": 0.9807472495404236,
-      "eval_iou_steel": 0.8805259615931451,
-      "eval_iou_tiles": 0.2386302029998193,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08592057973146439,
-      "eval_mean_accuracy": 0.8218530729867789,
-      "eval_mean_iou": 0.7787862808568348,
-      "eval_overall_accuracy": 0.9810510085800946,
-      "eval_runtime": 5.5537,
-      "eval_samples_per_second": 21.247,
-      "eval_steps_per_second": 1.44,
-      "step": 6940
-    },
-    {
-      "epoch": 408.29,
-      "learning_rate": 9.653250773993809e-06,
-      "loss": 0.0176,
-      "step": 6941
-    },
-    {
-      "epoch": 408.35,
-      "learning_rate": 9.647058823529412e-06,
-      "loss": 0.0159,
-      "step": 6942
-    },
-    {
-      "epoch": 408.41,
-      "learning_rate": 9.640866873065016e-06,
-      "loss": 0.0148,
-      "step": 6943
-    },
-    {
-      "epoch": 408.47,
-      "learning_rate": 9.63467492260062e-06,
-      "loss": 0.0155,
-      "step": 6944
-    },
-    {
-      "epoch": 408.53,
-      "learning_rate": 9.628482972136222e-06,
-      "loss": 0.0147,
-      "step": 6945
-    },
-    {
-      "epoch": 408.59,
-      "learning_rate": 9.622291021671827e-06,
-      "loss": 0.0165,
-      "step": 6946
-    },
-    {
-      "epoch": 408.65,
-      "learning_rate": 9.61609907120743e-06,
-      "loss": 0.016,
-      "step": 6947
-    },
-    {
-      "epoch": 408.71,
-      "learning_rate": 9.609907120743036e-06,
-      "loss": 0.0169,
-      "step": 6948
-    },
-    {
-      "epoch": 408.76,
-      "learning_rate": 9.603715170278637e-06,
-      "loss": 0.0151,
-      "step": 6949
-    },
-    {
-      "epoch": 408.82,
-      "learning_rate": 9.597523219814242e-06,
-      "loss": 0.0153,
-      "step": 6950
-    },
-    {
-      "epoch": 408.88,
-      "learning_rate": 9.591331269349846e-06,
-      "loss": 0.0136,
-      "step": 6951
-    },
-    {
-      "epoch": 408.94,
-      "learning_rate": 9.58513931888545e-06,
-      "loss": 0.0156,
-      "step": 6952
-    },
-    {
-      "epoch": 409.0,
-      "learning_rate": 9.578947368421052e-06,
-      "loss": 0.0145,
-      "step": 6953
-    },
-    {
-      "epoch": 409.06,
-      "learning_rate": 9.572755417956657e-06,
-      "loss": 0.0138,
-      "step": 6954
-    },
-    {
-      "epoch": 409.12,
-      "learning_rate": 9.56656346749226e-06,
-      "loss": 0.0144,
-      "step": 6955
-    },
-    {
-      "epoch": 409.18,
-      "learning_rate": 9.560371517027866e-06,
-      "loss": 0.0136,
-      "step": 6956
-    },
-    {
-      "epoch": 409.24,
-      "learning_rate": 9.554179566563467e-06,
-      "loss": 0.0161,
-      "step": 6957
-    },
-    {
-      "epoch": 409.29,
-      "learning_rate": 9.547987616099072e-06,
-      "loss": 0.0154,
-      "step": 6958
-    },
-    {
-      "epoch": 409.35,
-      "learning_rate": 9.541795665634676e-06,
-      "loss": 0.0148,
-      "step": 6959
-    },
-    {
-      "epoch": 409.41,
-      "learning_rate": 9.535603715170279e-06,
-      "loss": 0.0152,
-      "step": 6960
-    },
-    {
-      "epoch": 409.41,
-      "eval_accuracy_NAT": 0.994548227276234,
-      "eval_accuracy_REST": 0.5056910569105691,
-      "eval_accuracy_concrete": 0.9576679880570363,
-      "eval_accuracy_concrete tiles": 0.9441904312986095,
-      "eval_accuracy_dark bricks": 0.8519753842620064,
-      "eval_accuracy_grass": 0.8985019321971031,
-      "eval_accuracy_red bricks": 0.9381590509666081,
-      "eval_accuracy_road": 0.9815460765572486,
-      "eval_accuracy_rough concrete": 0.6384437995697746,
-      "eval_accuracy_rough red sidewalk": 0.7575331443310598,
-      "eval_accuracy_speedway bricks": 0.9902733137154502,
-      "eval_accuracy_steel": 0.9332015573225647,
-      "eval_accuracy_tiles": 0.24069193250325668,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9845465140289181,
-      "eval_iou_REST": 0.43194444444444446,
-      "eval_iou_concrete": 0.8981767815389049,
-      "eval_iou_concrete tiles": 0.8938220923227951,
-      "eval_iou_dark bricks": 0.8390071724111932,
-      "eval_iou_grass": 0.8462131426722905,
-      "eval_iou_red bricks": 0.9079891564343805,
-      "eval_iou_road": 0.961533327016193,
-      "eval_iou_rough concrete": 0.6337347544745839,
-      "eval_iou_rough red sidewalk": 0.5975808700088389,
-      "eval_iou_speedway bricks": 0.9807331893364462,
-      "eval_iou_steel": 0.8808428133719847,
-      "eval_iou_tiles": 0.23939375677955888,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.085886649787426,
-      "eval_mean_accuracy": 0.8178787611513478,
-      "eval_mean_iou": 0.7765783088338871,
-      "eval_overall_accuracy": 0.9809302313853119,
-      "eval_runtime": 7.0517,
-      "eval_samples_per_second": 16.734,
-      "eval_steps_per_second": 1.134,
-      "step": 6960
-    },
-    {
-      "epoch": 409.47,
-      "learning_rate": 9.529411764705882e-06,
-      "loss": 0.0137,
-      "step": 6961
-    },
-    {
-      "epoch": 409.53,
-      "learning_rate": 9.523219814241487e-06,
-      "loss": 0.0166,
-      "step": 6962
-    },
-    {
-      "epoch": 409.59,
-      "learning_rate": 9.51702786377709e-06,
-      "loss": 0.0154,
-      "step": 6963
-    },
-    {
-      "epoch": 409.65,
-      "learning_rate": 9.510835913312694e-06,
-      "loss": 0.0156,
-      "step": 6964
-    },
-    {
-      "epoch": 409.71,
-      "learning_rate": 9.504643962848297e-06,
-      "loss": 0.015,
-      "step": 6965
-    },
-    {
-      "epoch": 409.76,
-      "learning_rate": 9.498452012383902e-06,
-      "loss": 0.0153,
-      "step": 6966
-    },
-    {
-      "epoch": 409.82,
-      "learning_rate": 9.492260061919506e-06,
-      "loss": 0.017,
-      "step": 6967
-    },
-    {
-      "epoch": 409.88,
-      "learning_rate": 9.486068111455109e-06,
-      "loss": 0.0156,
-      "step": 6968
-    },
-    {
-      "epoch": 409.94,
-      "learning_rate": 9.479876160990712e-06,
-      "loss": 0.0152,
-      "step": 6969
-    },
-    {
-      "epoch": 410.0,
-      "learning_rate": 9.473684210526317e-06,
-      "loss": 0.0161,
-      "step": 6970
-    },
-    {
-      "epoch": 410.06,
-      "learning_rate": 9.46749226006192e-06,
-      "loss": 0.0141,
-      "step": 6971
-    },
-    {
-      "epoch": 410.12,
-      "learning_rate": 9.461300309597524e-06,
-      "loss": 0.0132,
-      "step": 6972
-    },
-    {
-      "epoch": 410.18,
-      "learning_rate": 9.455108359133127e-06,
-      "loss": 0.0159,
-      "step": 6973
-    },
-    {
-      "epoch": 410.24,
-      "learning_rate": 9.448916408668732e-06,
-      "loss": 0.0134,
-      "step": 6974
-    },
-    {
-      "epoch": 410.29,
-      "learning_rate": 9.442724458204334e-06,
-      "loss": 0.0166,
-      "step": 6975
-    },
-    {
-      "epoch": 410.35,
-      "learning_rate": 9.436532507739939e-06,
-      "loss": 0.0158,
-      "step": 6976
-    },
-    {
-      "epoch": 410.41,
-      "learning_rate": 9.430340557275542e-06,
-      "loss": 0.0148,
-      "step": 6977
-    },
-    {
-      "epoch": 410.47,
-      "learning_rate": 9.424148606811147e-06,
-      "loss": 0.0159,
-      "step": 6978
-    },
-    {
-      "epoch": 410.53,
-      "learning_rate": 9.417956656346749e-06,
-      "loss": 0.0145,
-      "step": 6979
-    },
-    {
-      "epoch": 410.59,
-      "learning_rate": 9.411764705882354e-06,
-      "loss": 0.014,
-      "step": 6980
-    },
-    {
-      "epoch": 410.59,
-      "eval_accuracy_NAT": 0.9942878334799463,
-      "eval_accuracy_REST": 0.5186991869918699,
-      "eval_accuracy_concrete": 0.9577428991192599,
-      "eval_accuracy_concrete tiles": 0.9504124440254537,
-      "eval_accuracy_dark bricks": 0.8494295886530253,
-      "eval_accuracy_grass": 0.9011930039703717,
-      "eval_accuracy_red bricks": 0.9382688927943761,
-      "eval_accuracy_road": 0.9816780773417714,
-      "eval_accuracy_rough concrete": 0.6393316608965611,
-      "eval_accuracy_rough red sidewalk": 0.7703697576309041,
-      "eval_accuracy_speedway bricks": 0.9907488520248164,
-      "eval_accuracy_steel": 0.9287509933179067,
-      "eval_accuracy_tiles": 0.23729891847677906,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846612201606467,
-      "eval_iou_REST": 0.45376955903271693,
-      "eval_iou_concrete": 0.8976125376964433,
-      "eval_iou_concrete tiles": 0.894007581971756,
-      "eval_iou_dark bricks": 0.8362912534451485,
-      "eval_iou_grass": 0.8475896709336599,
-      "eval_iou_red bricks": 0.907805940804506,
-      "eval_iou_road": 0.9616792505578201,
-      "eval_iou_rough concrete": 0.6342671687583871,
-      "eval_iou_rough red sidewalk": 0.6033298782067893,
-      "eval_iou_speedway bricks": 0.9806642387135546,
-      "eval_iou_steel": 0.8812335211214173,
-      "eval_iou_tiles": 0.23601904302760035,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08623054623603821,
-      "eval_mean_accuracy": 0.8198624699017724,
-      "eval_mean_iou": 0.7783792972638804,
-      "eval_overall_accuracy": 0.9809605226678363,
-      "eval_runtime": 5.6021,
-      "eval_samples_per_second": 21.064,
-      "eval_steps_per_second": 1.428,
-      "step": 6980
-    },
-    {
-      "epoch": 410.65,
-      "learning_rate": 9.405572755417957e-06,
-      "loss": 0.0188,
-      "step": 6981
-    },
-    {
-      "epoch": 410.71,
-      "learning_rate": 9.399380804953562e-06,
-      "loss": 0.0164,
-      "step": 6982
-    },
-    {
-      "epoch": 410.76,
-      "learning_rate": 9.393188854489164e-06,
-      "loss": 0.0154,
-      "step": 6983
-    },
-    {
-      "epoch": 410.82,
-      "learning_rate": 9.386996904024769e-06,
-      "loss": 0.0163,
-      "step": 6984
-    },
-    {
-      "epoch": 410.88,
-      "learning_rate": 9.380804953560372e-06,
-      "loss": 0.0143,
-      "step": 6985
-    },
-    {
-      "epoch": 410.94,
-      "learning_rate": 9.374613003095975e-06,
-      "loss": 0.0149,
-      "step": 6986
-    },
-    {
-      "epoch": 411.0,
-      "learning_rate": 9.368421052631579e-06,
-      "loss": 0.0152,
-      "step": 6987
-    },
-    {
-      "epoch": 411.06,
-      "learning_rate": 9.362229102167184e-06,
-      "loss": 0.0137,
-      "step": 6988
-    },
-    {
-      "epoch": 411.12,
-      "learning_rate": 9.356037151702787e-06,
-      "loss": 0.0134,
-      "step": 6989
-    },
-    {
-      "epoch": 411.18,
-      "learning_rate": 9.34984520123839e-06,
-      "loss": 0.014,
-      "step": 6990
-    },
-    {
-      "epoch": 411.24,
-      "learning_rate": 9.343653250773994e-06,
-      "loss": 0.0184,
-      "step": 6991
-    },
-    {
-      "epoch": 411.29,
-      "learning_rate": 9.337461300309599e-06,
-      "loss": 0.0141,
-      "step": 6992
-    },
-    {
-      "epoch": 411.35,
-      "learning_rate": 9.331269349845202e-06,
-      "loss": 0.0128,
-      "step": 6993
-    },
-    {
-      "epoch": 411.41,
-      "learning_rate": 9.325077399380805e-06,
-      "loss": 0.0162,
-      "step": 6994
-    },
-    {
-      "epoch": 411.47,
-      "learning_rate": 9.318885448916409e-06,
-      "loss": 0.0162,
-      "step": 6995
-    },
-    {
-      "epoch": 411.53,
-      "learning_rate": 9.312693498452014e-06,
-      "loss": 0.0156,
-      "step": 6996
-    },
-    {
-      "epoch": 411.59,
-      "learning_rate": 9.306501547987617e-06,
-      "loss": 0.0134,
-      "step": 6997
-    },
-    {
-      "epoch": 411.65,
-      "learning_rate": 9.30030959752322e-06,
-      "loss": 0.0165,
-      "step": 6998
-    },
-    {
-      "epoch": 411.71,
-      "learning_rate": 9.294117647058824e-06,
-      "loss": 0.0152,
-      "step": 6999
-    },
-    {
-      "epoch": 411.76,
-      "learning_rate": 9.287925696594429e-06,
-      "loss": 0.0155,
-      "step": 7000
-    },
-    {
-      "epoch": 411.76,
-      "eval_accuracy_NAT": 0.9941385518190933,
-      "eval_accuracy_REST": 0.5138211382113821,
-      "eval_accuracy_concrete": 0.9595208902491786,
-      "eval_accuracy_concrete tiles": 0.9521329248173462,
-      "eval_accuracy_dark bricks": 0.8580049001780142,
-      "eval_accuracy_grass": 0.8998919407888512,
-      "eval_accuracy_red bricks": 0.9423879613356766,
-      "eval_accuracy_road": 0.9816021902510741,
-      "eval_accuracy_rough concrete": 0.6356729571740983,
-      "eval_accuracy_rough red sidewalk": 0.7542093112446326,
-      "eval_accuracy_speedway bricks": 0.9902919925175118,
-      "eval_accuracy_steel": 0.9334764377495789,
-      "eval_accuracy_tiles": 0.25044684782937987,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846909209669396,
-      "eval_iou_REST": 0.4334705075445816,
-      "eval_iou_concrete": 0.8972454250052537,
-      "eval_iou_concrete tiles": 0.8948126162842208,
-      "eval_iou_dark bricks": 0.8432441330009877,
-      "eval_iou_grass": 0.8464789935861947,
-      "eval_iou_red bricks": 0.9088934795275173,
-      "eval_iou_road": 0.9614201868534678,
-      "eval_iou_rough concrete": 0.6305554406227649,
-      "eval_iou_rough red sidewalk": 0.597802957602923,
-      "eval_iou_speedway bricks": 0.9807595112869365,
-      "eval_iou_steel": 0.8807345722382809,
-      "eval_iou_tiles": 0.24879619597929456,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08661714196205139,
-      "eval_mean_accuracy": 0.8204306187819861,
-      "eval_mean_iou": 0.777608072346105,
-      "eval_overall_accuracy": 0.9809346279855502,
-      "eval_runtime": 5.787,
-      "eval_samples_per_second": 20.391,
-      "eval_steps_per_second": 1.382,
-      "step": 7000
-    },
-    {
-      "epoch": 411.82,
-      "learning_rate": 9.28173374613003e-06,
-      "loss": 0.0144,
-      "step": 7001
-    },
-    {
-      "epoch": 411.88,
-      "learning_rate": 9.275541795665635e-06,
-      "loss": 0.0161,
-      "step": 7002
-    },
-    {
-      "epoch": 411.94,
-      "learning_rate": 9.269349845201239e-06,
-      "loss": 0.0167,
-      "step": 7003
-    },
-    {
-      "epoch": 412.0,
-      "learning_rate": 9.263157894736844e-06,
-      "loss": 0.0172,
-      "step": 7004
-    },
-    {
-      "epoch": 412.06,
-      "learning_rate": 9.256965944272445e-06,
-      "loss": 0.0133,
-      "step": 7005
-    },
-    {
-      "epoch": 412.12,
-      "learning_rate": 9.25077399380805e-06,
-      "loss": 0.0147,
-      "step": 7006
-    },
-    {
-      "epoch": 412.18,
-      "learning_rate": 9.244582043343654e-06,
-      "loss": 0.0132,
-      "step": 7007
-    },
-    {
-      "epoch": 412.24,
-      "learning_rate": 9.238390092879259e-06,
-      "loss": 0.0171,
-      "step": 7008
-    },
-    {
-      "epoch": 412.29,
-      "learning_rate": 9.23219814241486e-06,
-      "loss": 0.0136,
-      "step": 7009
-    },
-    {
-      "epoch": 412.35,
-      "learning_rate": 9.226006191950465e-06,
-      "loss": 0.0141,
-      "step": 7010
-    },
-    {
-      "epoch": 412.41,
-      "learning_rate": 9.219814241486069e-06,
-      "loss": 0.0145,
-      "step": 7011
-    },
-    {
-      "epoch": 412.47,
-      "learning_rate": 9.213622291021674e-06,
-      "loss": 0.0159,
-      "step": 7012
-    },
-    {
-      "epoch": 412.53,
-      "learning_rate": 9.207430340557275e-06,
-      "loss": 0.0148,
-      "step": 7013
-    },
-    {
-      "epoch": 412.59,
-      "learning_rate": 9.20123839009288e-06,
-      "loss": 0.0153,
-      "step": 7014
-    },
-    {
-      "epoch": 412.65,
-      "learning_rate": 9.195046439628484e-06,
-      "loss": 0.0151,
-      "step": 7015
-    },
-    {
-      "epoch": 412.71,
-      "learning_rate": 9.188854489164087e-06,
-      "loss": 0.0172,
-      "step": 7016
-    },
-    {
-      "epoch": 412.76,
-      "learning_rate": 9.18266253869969e-06,
-      "loss": 0.0178,
-      "step": 7017
-    },
-    {
-      "epoch": 412.82,
-      "learning_rate": 9.176470588235295e-06,
-      "loss": 0.0159,
-      "step": 7018
-    },
-    {
-      "epoch": 412.88,
-      "learning_rate": 9.170278637770899e-06,
-      "loss": 0.0139,
-      "step": 7019
-    },
-    {
-      "epoch": 412.94,
-      "learning_rate": 9.164086687306502e-06,
-      "loss": 0.0165,
-      "step": 7020
-    },
-    {
-      "epoch": 412.94,
-      "eval_accuracy_NAT": 0.9939278646646643,
-      "eval_accuracy_REST": 0.5447154471544715,
-      "eval_accuracy_concrete": 0.9579817736799219,
-      "eval_accuracy_concrete tiles": 0.9501296252651426,
-      "eval_accuracy_dark bricks": 0.8594787818463717,
-      "eval_accuracy_grass": 0.9050498283542828,
-      "eval_accuracy_red bricks": 0.9466717926186292,
-      "eval_accuracy_road": 0.9819297873397888,
-      "eval_accuracy_rough concrete": 0.6353809759324034,
-      "eval_accuracy_rough red sidewalk": 0.7652486482828116,
-      "eval_accuracy_speedway bricks": 0.990879089543778,
-      "eval_accuracy_steel": 0.9307376291313278,
-      "eval_accuracy_tiles": 0.23926807840285982,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.98478613635146,
-      "eval_iou_REST": 0.4472630173564753,
-      "eval_iou_concrete": 0.8973211728585994,
-      "eval_iou_concrete tiles": 0.8921987385194201,
-      "eval_iou_dark bricks": 0.8441686015359886,
-      "eval_iou_grass": 0.8482292825077907,
-      "eval_iou_red bricks": 0.9081186449607502,
-      "eval_iou_road": 0.9619958376418539,
-      "eval_iou_rough concrete": 0.6300423656206239,
-      "eval_iou_rough red sidewalk": 0.5963968399090419,
-      "eval_iou_speedway bricks": 0.9807726883532599,
-      "eval_iou_steel": 0.8812671837364364,
-      "eval_iou_tiles": 0.23804930978359154,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08609697222709656,
-      "eval_mean_accuracy": 0.8231845632474196,
-      "eval_mean_iou": 0.7777392168565609,
-      "eval_overall_accuracy": 0.9809937557931674,
-      "eval_runtime": 5.9158,
-      "eval_samples_per_second": 19.947,
-      "eval_steps_per_second": 1.352,
-      "step": 7020
-    },
-    {
-      "epoch": 413.0,
-      "learning_rate": 9.157894736842105e-06,
-      "loss": 0.0146,
-      "step": 7021
-    },
-    {
-      "epoch": 413.06,
-      "learning_rate": 9.15170278637771e-06,
-      "loss": 0.0136,
-      "step": 7022
-    },
-    {
-      "epoch": 413.12,
-      "learning_rate": 9.145510835913314e-06,
-      "loss": 0.014,
-      "step": 7023
-    },
-    {
-      "epoch": 413.18,
-      "learning_rate": 9.139318885448917e-06,
-      "loss": 0.0156,
-      "step": 7024
-    },
-    {
-      "epoch": 413.24,
-      "learning_rate": 9.13312693498452e-06,
-      "loss": 0.0155,
-      "step": 7025
-    },
-    {
-      "epoch": 413.29,
-      "learning_rate": 9.126934984520125e-06,
-      "loss": 0.0158,
-      "step": 7026
-    },
-    {
-      "epoch": 413.35,
-      "learning_rate": 9.120743034055727e-06,
-      "loss": 0.0184,
-      "step": 7027
-    },
-    {
-      "epoch": 413.41,
-      "learning_rate": 9.114551083591332e-06,
-      "loss": 0.0162,
-      "step": 7028
-    },
-    {
-      "epoch": 413.47,
-      "learning_rate": 9.108359133126935e-06,
-      "loss": 0.0125,
-      "step": 7029
-    },
-    {
-      "epoch": 413.53,
-      "learning_rate": 9.10216718266254e-06,
-      "loss": 0.0144,
-      "step": 7030
-    },
-    {
-      "epoch": 413.59,
-      "learning_rate": 9.095975232198142e-06,
-      "loss": 0.0153,
-      "step": 7031
-    },
-    {
-      "epoch": 413.65,
-      "learning_rate": 9.089783281733747e-06,
-      "loss": 0.0147,
-      "step": 7032
-    },
-    {
-      "epoch": 413.71,
-      "learning_rate": 9.08359133126935e-06,
-      "loss": 0.0157,
-      "step": 7033
-    },
-    {
-      "epoch": 413.76,
-      "learning_rate": 9.077399380804955e-06,
-      "loss": 0.0135,
-      "step": 7034
-    },
-    {
-      "epoch": 413.82,
-      "learning_rate": 9.071207430340557e-06,
-      "loss": 0.0154,
-      "step": 7035
-    },
-    {
-      "epoch": 413.88,
-      "learning_rate": 9.065015479876162e-06,
-      "loss": 0.0152,
-      "step": 7036
-    },
-    {
-      "epoch": 413.94,
-      "learning_rate": 9.058823529411765e-06,
-      "loss": 0.0177,
-      "step": 7037
-    },
-    {
-      "epoch": 414.0,
-      "learning_rate": 9.05263157894737e-06,
-      "loss": 0.0175,
-      "step": 7038
-    },
-    {
-      "epoch": 414.06,
-      "learning_rate": 9.046439628482972e-06,
-      "loss": 0.0156,
-      "step": 7039
-    },
-    {
-      "epoch": 414.12,
-      "learning_rate": 9.040247678018577e-06,
-      "loss": 0.0147,
-      "step": 7040
-    },
-    {
-      "epoch": 414.12,
-      "eval_accuracy_NAT": 0.9942964593613969,
-      "eval_accuracy_REST": 0.5073170731707317,
-      "eval_accuracy_concrete": 0.9575819932151981,
-      "eval_accuracy_concrete tiles": 0.9442139995286354,
-      "eval_accuracy_dark bricks": 0.8571866087322704,
-      "eval_accuracy_grass": 0.9003300442456105,
-      "eval_accuracy_red bricks": 0.9396419156414763,
-      "eval_accuracy_road": 0.9813307068561848,
-      "eval_accuracy_rough concrete": 0.6356729571740983,
-      "eval_accuracy_rough red sidewalk": 0.7539611517608044,
-      "eval_accuracy_speedway bricks": 0.9907731858770434,
-      "eval_accuracy_steel": 0.9307451258702465,
-      "eval_accuracy_tiles": 0.2317852706837529,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846226083186919,
-      "eval_iou_REST": 0.4400564174894217,
-      "eval_iou_concrete": 0.8968979016733346,
-      "eval_iou_concrete tiles": 0.8921723638793008,
-      "eval_iou_dark bricks": 0.8419370455496177,
-      "eval_iou_grass": 0.8462200008004234,
-      "eval_iou_red bricks": 0.9079764368731094,
-      "eval_iou_road": 0.9614386182428966,
-      "eval_iou_rough concrete": 0.6304622767514154,
-      "eval_iou_rough red sidewalk": 0.5952327238185704,
-      "eval_iou_speedway bricks": 0.9807405600534809,
-      "eval_iou_steel": 0.8818961026660984,
-      "eval_iou_tiles": 0.23056987011421512,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08679600059986115,
-      "eval_mean_accuracy": 0.8172951147782653,
-      "eval_mean_iou": 0.7761709943254289,
-      "eval_overall_accuracy": 0.9808801877296577,
-      "eval_runtime": 5.7179,
-      "eval_samples_per_second": 20.637,
-      "eval_steps_per_second": 1.399,
-      "step": 7040
-    },
-    {
-      "epoch": 414.18,
-      "learning_rate": 9.03405572755418e-06,
-      "loss": 0.0145,
-      "step": 7041
-    },
-    {
-      "epoch": 414.24,
-      "learning_rate": 9.027863777089784e-06,
-      "loss": 0.017,
-      "step": 7042
-    },
-    {
-      "epoch": 414.29,
-      "learning_rate": 9.021671826625387e-06,
-      "loss": 0.016,
-      "step": 7043
-    },
-    {
-      "epoch": 414.35,
-      "learning_rate": 9.015479876160992e-06,
-      "loss": 0.0143,
-      "step": 7044
-    },
-    {
-      "epoch": 414.41,
-      "learning_rate": 9.009287925696595e-06,
-      "loss": 0.0144,
-      "step": 7045
-    },
-    {
-      "epoch": 414.47,
-      "learning_rate": 9.003095975232199e-06,
-      "loss": 0.0146,
-      "step": 7046
-    },
-    {
-      "epoch": 414.53,
-      "learning_rate": 8.996904024767802e-06,
-      "loss": 0.0143,
-      "step": 7047
-    },
-    {
-      "epoch": 414.59,
-      "learning_rate": 8.990712074303407e-06,
-      "loss": 0.0175,
-      "step": 7048
-    },
-    {
-      "epoch": 414.65,
-      "learning_rate": 8.98452012383901e-06,
-      "loss": 0.0153,
-      "step": 7049
-    },
-    {
-      "epoch": 414.71,
-      "learning_rate": 8.978328173374614e-06,
-      "loss": 0.0158,
-      "step": 7050
-    },
-    {
-      "epoch": 414.76,
-      "learning_rate": 8.972136222910217e-06,
-      "loss": 0.0176,
-      "step": 7051
-    },
-    {
-      "epoch": 414.82,
-      "learning_rate": 8.965944272445822e-06,
-      "loss": 0.0165,
-      "step": 7052
-    },
-    {
-      "epoch": 414.88,
-      "learning_rate": 8.959752321981424e-06,
-      "loss": 0.0158,
-      "step": 7053
-    },
-    {
-      "epoch": 414.94,
-      "learning_rate": 8.953560371517029e-06,
-      "loss": 0.0145,
-      "step": 7054
-    },
-    {
-      "epoch": 415.0,
-      "learning_rate": 8.947368421052632e-06,
-      "loss": 0.0152,
-      "step": 7055
-    },
-    {
-      "epoch": 415.06,
-      "learning_rate": 8.941176470588237e-06,
-      "loss": 0.015,
-      "step": 7056
-    },
-    {
-      "epoch": 415.12,
-      "learning_rate": 8.934984520123839e-06,
-      "loss": 0.0157,
-      "step": 7057
-    },
-    {
-      "epoch": 415.18,
-      "learning_rate": 8.928792569659444e-06,
-      "loss": 0.016,
-      "step": 7058
-    },
-    {
-      "epoch": 415.24,
-      "learning_rate": 8.922600619195047e-06,
-      "loss": 0.0145,
-      "step": 7059
-    },
-    {
-      "epoch": 415.29,
-      "learning_rate": 8.916408668730652e-06,
-      "loss": 0.0147,
-      "step": 7060
-    },
-    {
-      "epoch": 415.29,
-      "eval_accuracy_NAT": 0.9941638364340951,
-      "eval_accuracy_REST": 0.5089430894308943,
-      "eval_accuracy_concrete": 0.9590393191348843,
-      "eval_accuracy_concrete tiles": 0.9514494461465944,
-      "eval_accuracy_dark bricks": 0.8597276190111593,
-      "eval_accuracy_grass": 0.9003234206512318,
-      "eval_accuracy_red bricks": 0.9428822495606327,
-      "eval_accuracy_road": 0.9812569574300142,
-      "eval_accuracy_rough concrete": 0.6385808519893457,
-      "eval_accuracy_rough red sidewalk": 0.7823039728077366,
-      "eval_accuracy_speedway bricks": 0.9901465034996192,
-      "eval_accuracy_steel": 0.9343585540289974,
-      "eval_accuracy_tiles": 0.24217637613984064,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846568126809163,
-      "eval_iou_REST": 0.443342776203966,
-      "eval_iou_concrete": 0.8982914932653152,
-      "eval_iou_concrete tiles": 0.8922927303671286,
-      "eval_iou_dark bricks": 0.8452512573452959,
-      "eval_iou_grass": 0.8453409218117818,
-      "eval_iou_red bricks": 0.9091294217326837,
-      "eval_iou_road": 0.962724318406562,
-      "eval_iou_rough concrete": 0.6342834820900116,
-      "eval_iou_rough red sidewalk": 0.6062566290196627,
-      "eval_iou_speedway bricks": 0.9808566329545537,
-      "eval_iou_steel": 0.8819244939452692,
-      "eval_iou_tiles": 0.24089196926322134,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08587287366390228,
-      "eval_mean_accuracy": 0.821950168943465,
-      "eval_mean_iou": 0.7788648414681821,
-      "eval_overall_accuracy": 0.9810255018331236,
-      "eval_runtime": 5.9462,
-      "eval_samples_per_second": 19.845,
-      "eval_steps_per_second": 1.345,
-      "step": 7060
-    },
-    {
-      "epoch": 415.35,
-      "learning_rate": 8.910216718266253e-06,
-      "loss": 0.0141,
-      "step": 7061
-    },
-    {
-      "epoch": 415.41,
-      "learning_rate": 8.904024767801859e-06,
-      "loss": 0.015,
-      "step": 7062
-    },
-    {
-      "epoch": 415.47,
-      "learning_rate": 8.897832817337462e-06,
-      "loss": 0.0164,
-      "step": 7063
-    },
-    {
-      "epoch": 415.53,
-      "learning_rate": 8.891640866873067e-06,
-      "loss": 0.0147,
-      "step": 7064
-    },
-    {
-      "epoch": 415.59,
-      "learning_rate": 8.885448916408668e-06,
-      "loss": 0.0134,
-      "step": 7065
-    },
-    {
-      "epoch": 415.65,
-      "learning_rate": 8.879256965944274e-06,
-      "loss": 0.0159,
-      "step": 7066
-    },
-    {
-      "epoch": 415.71,
-      "learning_rate": 8.873065015479877e-06,
-      "loss": 0.0161,
-      "step": 7067
-    },
-    {
-      "epoch": 415.76,
-      "learning_rate": 8.86687306501548e-06,
-      "loss": 0.0158,
-      "step": 7068
-    },
-    {
-      "epoch": 415.82,
-      "learning_rate": 8.860681114551083e-06,
-      "loss": 0.0155,
-      "step": 7069
-    },
-    {
-      "epoch": 415.88,
-      "learning_rate": 8.854489164086688e-06,
-      "loss": 0.0128,
-      "step": 7070
-    },
-    {
-      "epoch": 415.94,
-      "learning_rate": 8.848297213622292e-06,
-      "loss": 0.0169,
-      "step": 7071
-    },
-    {
-      "epoch": 416.0,
-      "learning_rate": 8.842105263157895e-06,
-      "loss": 0.0144,
-      "step": 7072
-    },
-    {
-      "epoch": 416.06,
-      "learning_rate": 8.835913312693498e-06,
-      "loss": 0.0151,
-      "step": 7073
-    },
-    {
-      "epoch": 416.12,
-      "learning_rate": 8.829721362229103e-06,
-      "loss": 0.0141,
-      "step": 7074
-    },
-    {
-      "epoch": 416.18,
-      "learning_rate": 8.823529411764707e-06,
-      "loss": 0.0148,
-      "step": 7075
-    },
-    {
-      "epoch": 416.24,
-      "learning_rate": 8.81733746130031e-06,
-      "loss": 0.015,
-      "step": 7076
-    },
-    {
-      "epoch": 416.29,
-      "learning_rate": 8.811145510835913e-06,
-      "loss": 0.0163,
-      "step": 7077
-    },
-    {
-      "epoch": 416.35,
-      "learning_rate": 8.804953560371518e-06,
-      "loss": 0.0177,
-      "step": 7078
-    },
-    {
-      "epoch": 416.41,
-      "learning_rate": 8.798761609907122e-06,
-      "loss": 0.0158,
-      "step": 7079
-    },
-    {
-      "epoch": 416.47,
-      "learning_rate": 8.792569659442725e-06,
-      "loss": 0.0158,
-      "step": 7080
-    },
-    {
-      "epoch": 416.47,
-      "eval_accuracy_NAT": 0.9939149797542476,
-      "eval_accuracy_REST": 0.5593495934959349,
-      "eval_accuracy_concrete": 0.9586410674673487,
-      "eval_accuracy_concrete tiles": 0.9570822531227905,
-      "eval_accuracy_dark bricks": 0.8539756522404915,
-      "eval_accuracy_grass": 0.9062401828869031,
-      "eval_accuracy_red bricks": 0.9404108084358523,
-      "eval_accuracy_road": 0.9813675815692702,
-      "eval_accuracy_rough concrete": 0.6393137844940084,
-      "eval_accuracy_rough red sidewalk": 0.7754833469946383,
-      "eval_accuracy_speedway bricks": 0.9911308249587182,
-      "eval_accuracy_steel": 0.9282312194195524,
-      "eval_accuracy_tiles": 0.2555363688690963,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848523853315644,
-      "eval_iou_REST": 0.4642375168690958,
-      "eval_iou_concrete": 0.8978802173908375,
-      "eval_iou_concrete tiles": 0.893526667840169,
-      "eval_iou_dark bricks": 0.8402484156206153,
-      "eval_iou_grass": 0.8486261903264022,
-      "eval_iou_red bricks": 0.9084301554459122,
-      "eval_iou_road": 0.9629010108086312,
-      "eval_iou_rough concrete": 0.6344744794469512,
-      "eval_iou_rough red sidewalk": 0.6056688769726835,
-      "eval_iou_speedway bricks": 0.9807307455826949,
-      "eval_iou_steel": 0.8821228665537852,
-      "eval_iou_tiles": 0.2538216177178623,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08579849451780319,
-      "eval_mean_accuracy": 0.8262059741314501,
-      "eval_mean_iou": 0.7813477804544002,
-      "eval_overall_accuracy": 0.9811151795468088,
-      "eval_runtime": 5.4249,
-      "eval_samples_per_second": 21.752,
-      "eval_steps_per_second": 1.475,
-      "step": 7080
-    },
-    {
-      "epoch": 416.53,
-      "learning_rate": 8.786377708978328e-06,
-      "loss": 0.0141,
-      "step": 7081
-    },
-    {
-      "epoch": 416.59,
-      "learning_rate": 8.780185758513933e-06,
-      "loss": 0.0156,
-      "step": 7082
-    },
-    {
-      "epoch": 416.65,
-      "learning_rate": 8.773993808049535e-06,
-      "loss": 0.0142,
-      "step": 7083
-    },
-    {
-      "epoch": 416.71,
-      "learning_rate": 8.76780185758514e-06,
-      "loss": 0.015,
-      "step": 7084
-    },
-    {
-      "epoch": 416.76,
-      "learning_rate": 8.761609907120743e-06,
-      "loss": 0.0153,
-      "step": 7085
-    },
-    {
-      "epoch": 416.82,
-      "learning_rate": 8.755417956656348e-06,
-      "loss": 0.0157,
-      "step": 7086
-    },
-    {
-      "epoch": 416.88,
-      "learning_rate": 8.74922600619195e-06,
-      "loss": 0.0144,
-      "step": 7087
-    },
-    {
-      "epoch": 416.94,
-      "learning_rate": 8.743034055727555e-06,
-      "loss": 0.0138,
-      "step": 7088
-    },
-    {
-      "epoch": 417.0,
-      "learning_rate": 8.736842105263158e-06,
-      "loss": 0.0148,
-      "step": 7089
-    },
-    {
-      "epoch": 417.06,
-      "learning_rate": 8.730650154798763e-06,
-      "loss": 0.014,
-      "step": 7090
-    },
-    {
-      "epoch": 417.12,
-      "learning_rate": 8.724458204334365e-06,
-      "loss": 0.0147,
-      "step": 7091
-    },
-    {
-      "epoch": 417.18,
-      "learning_rate": 8.71826625386997e-06,
-      "loss": 0.016,
-      "step": 7092
-    },
-    {
-      "epoch": 417.24,
-      "learning_rate": 8.712074303405573e-06,
-      "loss": 0.0157,
-      "step": 7093
-    },
-    {
-      "epoch": 417.29,
-      "learning_rate": 8.705882352941177e-06,
-      "loss": 0.0143,
-      "step": 7094
-    },
-    {
-      "epoch": 417.35,
-      "learning_rate": 8.69969040247678e-06,
-      "loss": 0.0172,
-      "step": 7095
-    },
-    {
-      "epoch": 417.41,
-      "learning_rate": 8.693498452012385e-06,
-      "loss": 0.0163,
-      "step": 7096
-    },
-    {
-      "epoch": 417.47,
-      "learning_rate": 8.687306501547988e-06,
-      "loss": 0.0138,
-      "step": 7097
-    },
-    {
-      "epoch": 417.53,
-      "learning_rate": 8.681114551083592e-06,
-      "loss": 0.0114,
-      "step": 7098
-    },
-    {
-      "epoch": 417.59,
-      "learning_rate": 8.674922600619195e-06,
-      "loss": 0.0157,
-      "step": 7099
-    },
-    {
-      "epoch": 417.65,
-      "learning_rate": 8.6687306501548e-06,
-      "loss": 0.0151,
-      "step": 7100
-    },
-    {
-      "epoch": 417.65,
-      "eval_accuracy_NAT": 0.9942923620677079,
-      "eval_accuracy_REST": 0.5024390243902439,
-      "eval_accuracy_concrete": 0.9590790678617784,
-      "eval_accuracy_concrete tiles": 0.9516144237567759,
-      "eval_accuracy_dark bricks": 0.8518701069999809,
-      "eval_accuracy_grass": 0.9029463640251773,
-      "eval_accuracy_red bricks": 0.9344244288224957,
-      "eval_accuracy_road": 0.9821611895248025,
-      "eval_accuracy_rough concrete": 0.6451355329253541,
-      "eval_accuracy_rough red sidewalk": 0.7760473458215207,
-      "eval_accuracy_speedway bricks": 0.9899310974978631,
-      "eval_accuracy_steel": 0.932496863864219,
-      "eval_accuracy_tiles": 0.25105274319125087,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847558200063337,
-      "eval_iou_REST": 0.43216783216783217,
-      "eval_iou_concrete": 0.8983386936947257,
-      "eval_iou_concrete tiles": 0.8937314622161229,
-      "eval_iou_dark bricks": 0.8388995443042747,
-      "eval_iou_grass": 0.8483380554150438,
-      "eval_iou_red bricks": 0.9069296375266525,
-      "eval_iou_road": 0.9632887301538214,
-      "eval_iou_rough concrete": 0.6401840146170994,
-      "eval_iou_rough red sidewalk": 0.6065653745555001,
-      "eval_iou_speedway bricks": 0.9808648714731307,
-      "eval_iou_steel": 0.8823024542488296,
-      "eval_iou_tiles": 0.24951073374883329,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08537375926971436,
-      "eval_mean_accuracy": 0.8210377346730131,
-      "eval_mean_iou": 0.7789136326252462,
-      "eval_overall_accuracy": 0.9811456971249338,
-      "eval_runtime": 5.8952,
-      "eval_samples_per_second": 20.016,
-      "eval_steps_per_second": 1.357,
-      "step": 7100
-    },
-    {
-      "epoch": 417.71,
-      "learning_rate": 8.662538699690403e-06,
-      "loss": 0.0159,
-      "step": 7101
-    },
-    {
-      "epoch": 417.76,
-      "learning_rate": 8.656346749226007e-06,
-      "loss": 0.0159,
-      "step": 7102
-    },
-    {
-      "epoch": 417.82,
-      "learning_rate": 8.65015479876161e-06,
-      "loss": 0.0163,
-      "step": 7103
-    },
-    {
-      "epoch": 417.88,
-      "learning_rate": 8.643962848297215e-06,
-      "loss": 0.0152,
-      "step": 7104
-    },
-    {
-      "epoch": 417.94,
-      "learning_rate": 8.637770897832818e-06,
-      "loss": 0.016,
-      "step": 7105
-    },
-    {
-      "epoch": 418.0,
-      "learning_rate": 8.631578947368422e-06,
-      "loss": 0.0141,
-      "step": 7106
-    },
-    {
-      "epoch": 418.06,
-      "learning_rate": 8.625386996904025e-06,
-      "loss": 0.0135,
-      "step": 7107
-    },
-    {
-      "epoch": 418.12,
-      "learning_rate": 8.61919504643963e-06,
-      "loss": 0.0149,
-      "step": 7108
-    },
-    {
-      "epoch": 418.18,
-      "learning_rate": 8.613003095975232e-06,
-      "loss": 0.0152,
-      "step": 7109
-    },
-    {
-      "epoch": 418.24,
-      "learning_rate": 8.606811145510837e-06,
-      "loss": 0.0138,
-      "step": 7110
-    },
-    {
-      "epoch": 418.29,
-      "learning_rate": 8.60061919504644e-06,
-      "loss": 0.0147,
-      "step": 7111
-    },
-    {
-      "epoch": 418.35,
-      "learning_rate": 8.594427244582045e-06,
-      "loss": 0.0173,
-      "step": 7112
-    },
-    {
-      "epoch": 418.41,
-      "learning_rate": 8.588235294117647e-06,
-      "loss": 0.0142,
-      "step": 7113
-    },
-    {
-      "epoch": 418.47,
-      "learning_rate": 8.582043343653252e-06,
-      "loss": 0.0171,
-      "step": 7114
-    },
-    {
-      "epoch": 418.53,
-      "learning_rate": 8.575851393188855e-06,
-      "loss": 0.015,
-      "step": 7115
-    },
-    {
-      "epoch": 418.59,
-      "learning_rate": 8.56965944272446e-06,
-      "loss": 0.0134,
-      "step": 7116
-    },
-    {
-      "epoch": 418.65,
-      "learning_rate": 8.563467492260062e-06,
-      "loss": 0.0136,
-      "step": 7117
-    },
-    {
-      "epoch": 418.71,
-      "learning_rate": 8.557275541795667e-06,
-      "loss": 0.0138,
-      "step": 7118
-    },
-    {
-      "epoch": 418.76,
-      "learning_rate": 8.55108359133127e-06,
-      "loss": 0.0131,
-      "step": 7119
-    },
-    {
-      "epoch": 418.82,
-      "learning_rate": 8.544891640866875e-06,
-      "loss": 0.0163,
-      "step": 7120
-    },
-    {
-      "epoch": 418.82,
-      "eval_accuracy_NAT": 0.9940192990080398,
-      "eval_accuracy_REST": 0.5056910569105691,
-      "eval_accuracy_concrete": 0.9587457900747429,
-      "eval_accuracy_concrete tiles": 0.9567287296724016,
-      "eval_accuracy_dark bricks": 0.8553299006565472,
-      "eval_accuracy_grass": 0.9058493908185627,
-      "eval_accuracy_red bricks": 0.938213971880492,
-      "eval_accuracy_road": 0.983044044974324,
-      "eval_accuracy_rough concrete": 0.6370434813698092,
-      "eval_accuracy_rough red sidewalk": 0.7597515397167974,
-      "eval_accuracy_speedway bricks": 0.9902596045029279,
-      "eval_accuracy_steel": 0.9305152258767436,
-      "eval_accuracy_tiles": 0.2475688448604926,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984840852317944,
-      "eval_iou_REST": 0.43194444444444446,
-      "eval_iou_concrete": 0.8978275660772888,
-      "eval_iou_concrete tiles": 0.8932555836725712,
-      "eval_iou_dark bricks": 0.8413630137308712,
-      "eval_iou_grass": 0.8481971114444461,
-      "eval_iou_red bricks": 0.9091538052155402,
-      "eval_iou_road": 0.9617591222061128,
-      "eval_iou_rough concrete": 0.6311054439839904,
-      "eval_iou_rough red sidewalk": 0.5988619187338846,
-      "eval_iou_speedway bricks": 0.9809060627855031,
-      "eval_iou_steel": 0.8820479293543015,
-      "eval_iou_tiles": 0.24615199252989547,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08606668561697006,
-      "eval_mean_accuracy": 0.8202123754094193,
-      "eval_mean_iou": 0.7774934497305227,
-      "eval_overall_accuracy": 0.981057700464281,
-      "eval_runtime": 5.4396,
-      "eval_samples_per_second": 21.693,
-      "eval_steps_per_second": 1.471,
-      "step": 7120
-    },
-    {
-      "epoch": 418.88,
-      "learning_rate": 8.538699690402477e-06,
-      "loss": 0.0146,
-      "step": 7121
-    },
-    {
-      "epoch": 418.94,
-      "learning_rate": 8.532507739938082e-06,
-      "loss": 0.0178,
-      "step": 7122
-    },
-    {
-      "epoch": 419.0,
-      "learning_rate": 8.526315789473685e-06,
-      "loss": 0.0159,
-      "step": 7123
-    },
-    {
-      "epoch": 419.06,
-      "learning_rate": 8.520123839009288e-06,
-      "loss": 0.0149,
-      "step": 7124
-    },
-    {
-      "epoch": 419.12,
-      "learning_rate": 8.513931888544892e-06,
-      "loss": 0.0133,
-      "step": 7125
-    },
-    {
-      "epoch": 419.18,
-      "learning_rate": 8.507739938080497e-06,
-      "loss": 0.015,
-      "step": 7126
-    },
-    {
-      "epoch": 419.24,
-      "learning_rate": 8.5015479876161e-06,
-      "loss": 0.0154,
-      "step": 7127
-    },
-    {
-      "epoch": 419.29,
-      "learning_rate": 8.495356037151703e-06,
-      "loss": 0.0176,
-      "step": 7128
-    },
-    {
-      "epoch": 419.35,
-      "learning_rate": 8.489164086687307e-06,
-      "loss": 0.0136,
-      "step": 7129
-    },
-    {
-      "epoch": 419.41,
-      "learning_rate": 8.482972136222912e-06,
-      "loss": 0.014,
-      "step": 7130
-    },
-    {
-      "epoch": 419.47,
-      "learning_rate": 8.476780185758515e-06,
-      "loss": 0.0147,
-      "step": 7131
-    },
-    {
-      "epoch": 419.53,
-      "learning_rate": 8.470588235294118e-06,
-      "loss": 0.0168,
-      "step": 7132
-    },
-    {
-      "epoch": 419.59,
-      "learning_rate": 8.464396284829722e-06,
-      "loss": 0.0156,
-      "step": 7133
-    },
-    {
-      "epoch": 419.65,
-      "learning_rate": 8.458204334365327e-06,
-      "loss": 0.0156,
-      "step": 7134
-    },
-    {
-      "epoch": 419.71,
-      "learning_rate": 8.452012383900928e-06,
-      "loss": 0.0133,
-      "step": 7135
-    },
-    {
-      "epoch": 419.76,
-      "learning_rate": 8.445820433436533e-06,
-      "loss": 0.0131,
-      "step": 7136
-    },
-    {
-      "epoch": 419.82,
-      "learning_rate": 8.439628482972137e-06,
-      "loss": 0.015,
-      "step": 7137
-    },
-    {
-      "epoch": 419.88,
-      "learning_rate": 8.433436532507742e-06,
-      "loss": 0.0171,
-      "step": 7138
-    },
-    {
-      "epoch": 419.94,
-      "learning_rate": 8.427244582043343e-06,
-      "loss": 0.0156,
-      "step": 7139
-    },
-    {
-      "epoch": 420.0,
-      "learning_rate": 8.421052631578948e-06,
-      "loss": 0.0171,
-      "step": 7140
-    },
-    {
-      "epoch": 420.0,
-      "eval_accuracy_NAT": 0.9938926063742353,
-      "eval_accuracy_REST": 0.49105691056910566,
-      "eval_accuracy_concrete": 0.957975658491169,
-      "eval_accuracy_concrete tiles": 0.9507423992458166,
-      "eval_accuracy_dark bricks": 0.8616513217081714,
-      "eval_accuracy_grass": 0.9082774112722222,
-      "eval_accuracy_red bricks": 0.937335237258348,
-      "eval_accuracy_road": 0.9826923991596841,
-      "eval_accuracy_rough concrete": 0.6387119456080659,
-      "eval_accuracy_rough red sidewalk": 0.7643763300972334,
-      "eval_accuracy_speedway bricks": 0.9902913070568857,
-      "eval_accuracy_steel": 0.9341636388171145,
-      "eval_accuracy_tiles": 0.249022993728983,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847991771413676,
-      "eval_iou_REST": 0.41770401106500693,
-      "eval_iou_concrete": 0.8984994789633468,
-      "eval_iou_concrete tiles": 0.8942386557602359,
-      "eval_iou_dark bricks": 0.8452972795343051,
-      "eval_iou_grass": 0.8481493262646345,
-      "eval_iou_red bricks": 0.9087859424920128,
-      "eval_iou_road": 0.9620740405890641,
-      "eval_iou_rough concrete": 0.6331395898308289,
-      "eval_iou_rough red sidewalk": 0.6017582807921144,
-      "eval_iou_speedway bricks": 0.980830411183038,
-      "eval_iou_steel": 0.8813538542786549,
-      "eval_iou_tiles": 0.24759036144578314,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08561260253190994,
-      "eval_mean_accuracy": 0.8200146276451564,
-      "eval_mean_iou": 0.7772477237954148,
-      "eval_overall_accuracy": 0.9810956211413367,
-      "eval_runtime": 10.6904,
-      "eval_samples_per_second": 11.038,
-      "eval_steps_per_second": 0.748,
-      "step": 7140
-    },
-    {
-      "epoch": 420.06,
-      "learning_rate": 8.414860681114552e-06,
-      "loss": 0.0142,
-      "step": 7141
-    },
-    {
-      "epoch": 420.12,
-      "learning_rate": 8.408668730650157e-06,
-      "loss": 0.0145,
-      "step": 7142
-    },
-    {
-      "epoch": 420.18,
-      "learning_rate": 8.402476780185758e-06,
-      "loss": 0.0145,
-      "step": 7143
-    },
-    {
-      "epoch": 420.24,
-      "learning_rate": 8.396284829721363e-06,
-      "loss": 0.0165,
-      "step": 7144
-    },
-    {
-      "epoch": 420.29,
-      "learning_rate": 8.390092879256967e-06,
-      "loss": 0.0138,
-      "step": 7145
-    },
-    {
-      "epoch": 420.35,
-      "learning_rate": 8.383900928792572e-06,
-      "loss": 0.0141,
-      "step": 7146
-    },
-    {
-      "epoch": 420.41,
-      "learning_rate": 8.377708978328173e-06,
-      "loss": 0.0167,
-      "step": 7147
-    },
-    {
-      "epoch": 420.47,
-      "learning_rate": 8.371517027863778e-06,
-      "loss": 0.0146,
-      "step": 7148
-    },
-    {
-      "epoch": 420.53,
-      "learning_rate": 8.365325077399382e-06,
-      "loss": 0.0142,
-      "step": 7149
-    },
-    {
-      "epoch": 420.59,
-      "learning_rate": 8.359133126934985e-06,
-      "loss": 0.0142,
-      "step": 7150
-    },
-    {
-      "epoch": 420.65,
-      "learning_rate": 8.352941176470588e-06,
-      "loss": 0.018,
-      "step": 7151
-    },
-    {
-      "epoch": 420.71,
-      "learning_rate": 8.346749226006193e-06,
-      "loss": 0.0169,
-      "step": 7152
-    },
-    {
-      "epoch": 420.76,
-      "learning_rate": 8.340557275541797e-06,
-      "loss": 0.0147,
-      "step": 7153
-    },
-    {
-      "epoch": 420.82,
-      "learning_rate": 8.3343653250774e-06,
-      "loss": 0.0168,
-      "step": 7154
-    },
-    {
-      "epoch": 420.88,
-      "learning_rate": 8.328173374613003e-06,
-      "loss": 0.0164,
-      "step": 7155
-    },
-    {
-      "epoch": 420.94,
-      "learning_rate": 8.321981424148608e-06,
-      "loss": 0.0134,
-      "step": 7156
-    },
-    {
-      "epoch": 421.0,
-      "learning_rate": 8.315789473684212e-06,
-      "loss": 0.0144,
-      "step": 7157
-    },
-    {
-      "epoch": 421.06,
-      "learning_rate": 8.309597523219815e-06,
-      "loss": 0.0156,
-      "step": 7158
-    },
-    {
-      "epoch": 421.12,
-      "learning_rate": 8.303405572755418e-06,
-      "loss": 0.0129,
-      "step": 7159
-    },
-    {
-      "epoch": 421.18,
-      "learning_rate": 8.297213622291023e-06,
-      "loss": 0.0154,
-      "step": 7160
-    },
-    {
-      "epoch": 421.18,
-      "eval_accuracy_NAT": 0.9939507232505082,
-      "eval_accuracy_REST": 0.5382113821138211,
-      "eval_accuracy_concrete": 0.9592372983707609,
-      "eval_accuracy_concrete tiles": 0.9494932830544426,
-      "eval_accuracy_dark bricks": 0.858057538809027,
-      "eval_accuracy_grass": 0.9079907042584034,
-      "eval_accuracy_red bricks": 0.9423879613356766,
-      "eval_accuracy_road": 0.9822376110316315,
-      "eval_accuracy_rough concrete": 0.6397428181552745,
-      "eval_accuracy_rough red sidewalk": 0.7639326510200859,
-      "eval_accuracy_speedway bricks": 0.9899071063759491,
-      "eval_accuracy_steel": 0.9359403659408158,
-      "eval_accuracy_tiles": 0.24102517495228573,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848166228189846,
-      "eval_iou_REST": 0.45095367847411444,
-      "eval_iou_concrete": 0.8983299640064256,
-      "eval_iou_concrete tiles": 0.8954656590353411,
-      "eval_iou_dark bricks": 0.8421631065772417,
-      "eval_iou_grass": 0.849692474706421,
-      "eval_iou_red bricks": 0.912033592005953,
-      "eval_iou_road": 0.9622646646143537,
-      "eval_iou_rough concrete": 0.6348664758615796,
-      "eval_iou_rough red sidewalk": 0.6026648790066622,
-      "eval_iou_speedway bricks": 0.9808052947164048,
-      "eval_iou_steel": 0.8814860141917414,
-      "eval_iou_tiles": 0.23994933196610066,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08540049940347672,
-      "eval_mean_accuracy": 0.8232395860514372,
-      "eval_mean_iou": 0.7796532121524097,
-      "eval_overall_accuracy": 0.9811234555001986,
-      "eval_runtime": 5.7311,
-      "eval_samples_per_second": 20.59,
-      "eval_steps_per_second": 1.396,
-      "step": 7160
-    },
-    {
-      "epoch": 421.24,
-      "learning_rate": 8.291021671826625e-06,
-      "loss": 0.0142,
-      "step": 7161
-    },
-    {
-      "epoch": 421.29,
-      "learning_rate": 8.28482972136223e-06,
-      "loss": 0.0165,
-      "step": 7162
-    },
-    {
-      "epoch": 421.35,
-      "learning_rate": 8.278637770897833e-06,
-      "loss": 0.0161,
-      "step": 7163
-    },
-    {
-      "epoch": 421.41,
-      "learning_rate": 8.272445820433438e-06,
-      "loss": 0.016,
-      "step": 7164
-    },
-    {
-      "epoch": 421.47,
-      "learning_rate": 8.26625386996904e-06,
-      "loss": 0.0142,
-      "step": 7165
-    },
-    {
-      "epoch": 421.53,
-      "learning_rate": 8.260061919504645e-06,
-      "loss": 0.0158,
-      "step": 7166
-    },
-    {
-      "epoch": 421.59,
-      "learning_rate": 8.253869969040248e-06,
-      "loss": 0.0144,
-      "step": 7167
-    },
-    {
-      "epoch": 421.65,
-      "learning_rate": 8.247678018575853e-06,
-      "loss": 0.016,
-      "step": 7168
-    },
-    {
-      "epoch": 421.71,
-      "learning_rate": 8.241486068111455e-06,
-      "loss": 0.0141,
-      "step": 7169
-    },
-    {
-      "epoch": 421.76,
-      "learning_rate": 8.23529411764706e-06,
-      "loss": 0.0147,
-      "step": 7170
-    },
-    {
-      "epoch": 421.82,
-      "learning_rate": 8.229102167182663e-06,
-      "loss": 0.0151,
-      "step": 7171
-    },
-    {
-      "epoch": 421.88,
-      "learning_rate": 8.222910216718266e-06,
-      "loss": 0.0171,
-      "step": 7172
-    },
-    {
-      "epoch": 421.94,
-      "learning_rate": 8.21671826625387e-06,
-      "loss": 0.0153,
-      "step": 7173
-    },
-    {
-      "epoch": 422.0,
-      "learning_rate": 8.210526315789475e-06,
-      "loss": 0.0126,
-      "step": 7174
-    },
-    {
-      "epoch": 422.06,
-      "learning_rate": 8.204334365325078e-06,
-      "loss": 0.0182,
-      "step": 7175
-    },
-    {
-      "epoch": 422.12,
-      "learning_rate": 8.198142414860681e-06,
-      "loss": 0.0141,
-      "step": 7176
-    },
-    {
-      "epoch": 422.18,
-      "learning_rate": 8.191950464396285e-06,
-      "loss": 0.0133,
-      "step": 7177
-    },
-    {
-      "epoch": 422.24,
-      "learning_rate": 8.18575851393189e-06,
-      "loss": 0.0173,
-      "step": 7178
-    },
-    {
-      "epoch": 422.29,
-      "learning_rate": 8.179566563467493e-06,
-      "loss": 0.0155,
-      "step": 7179
-    },
-    {
-      "epoch": 422.35,
-      "learning_rate": 8.173374613003096e-06,
-      "loss": 0.0132,
-      "step": 7180
-    },
-    {
-      "epoch": 422.35,
-      "eval_accuracy_NAT": 0.9941788239031154,
-      "eval_accuracy_REST": 0.5430894308943089,
-      "eval_accuracy_concrete": 0.9590179159742489,
-      "eval_accuracy_concrete tiles": 0.9540655196794721,
-      "eval_accuracy_dark bricks": 0.8513724326704056,
-      "eval_accuracy_grass": 0.904245534751161,
-      "eval_accuracy_red bricks": 0.9383238137082601,
-      "eval_accuracy_road": 0.9817967177230025,
-      "eval_accuracy_rough concrete": 0.639343578498263,
-      "eval_accuracy_rough red sidewalk": 0.7734905511396536,
-      "eval_accuracy_speedway bricks": 0.9901581563302632,
-      "eval_accuracy_steel": 0.9313623573745421,
-      "eval_accuracy_tiles": 0.24887151988851525,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847885510062162,
-      "eval_iou_REST": 0.46453407510431155,
-      "eval_iou_concrete": 0.8983434837572333,
-      "eval_iou_concrete tiles": 0.8953002322238195,
-      "eval_iou_dark bricks": 0.8374543999623432,
-      "eval_iou_grass": 0.8484777455184721,
-      "eval_iou_red bricks": 0.9103260869565217,
-      "eval_iou_road": 0.962535221303479,
-      "eval_iou_rough concrete": 0.6338331029430877,
-      "eval_iou_rough red sidewalk": 0.6083896229874723,
-      "eval_iou_speedway bricks": 0.9805846925836119,
-      "eval_iou_steel": 0.8806064658502643,
-      "eval_iou_tiles": 0.2472535741158766,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08567735552787781,
-      "eval_mean_accuracy": 0.8237935655796318,
-      "eval_mean_iou": 0.7809559426394391,
-      "eval_overall_accuracy": 0.9810812998626192,
-      "eval_runtime": 5.5333,
-      "eval_samples_per_second": 21.326,
-      "eval_steps_per_second": 1.446,
-      "step": 7180
-    },
-    {
-      "epoch": 422.41,
-      "learning_rate": 8.1671826625387e-06,
-      "loss": 0.015,
-      "step": 7181
-    },
-    {
-      "epoch": 422.47,
-      "learning_rate": 8.160990712074305e-06,
-      "loss": 0.0149,
-      "step": 7182
-    },
-    {
-      "epoch": 422.53,
-      "learning_rate": 8.154798761609908e-06,
-      "loss": 0.0143,
-      "step": 7183
-    },
-    {
-      "epoch": 422.59,
-      "learning_rate": 8.148606811145511e-06,
-      "loss": 0.0182,
-      "step": 7184
-    },
-    {
-      "epoch": 422.65,
-      "learning_rate": 8.142414860681115e-06,
-      "loss": 0.0134,
-      "step": 7185
-    },
-    {
-      "epoch": 422.71,
-      "learning_rate": 8.13622291021672e-06,
-      "loss": 0.0147,
-      "step": 7186
-    },
-    {
-      "epoch": 422.76,
-      "learning_rate": 8.130030959752323e-06,
-      "loss": 0.0157,
-      "step": 7187
-    },
-    {
-      "epoch": 422.82,
-      "learning_rate": 8.123839009287926e-06,
-      "loss": 0.0146,
-      "step": 7188
-    },
-    {
-      "epoch": 422.88,
-      "learning_rate": 8.11764705882353e-06,
-      "loss": 0.0149,
-      "step": 7189
-    },
-    {
-      "epoch": 422.94,
-      "learning_rate": 8.111455108359135e-06,
-      "loss": 0.014,
-      "step": 7190
-    },
-    {
-      "epoch": 423.0,
-      "learning_rate": 8.105263157894736e-06,
-      "loss": 0.0162,
-      "step": 7191
-    },
-    {
-      "epoch": 423.06,
-      "learning_rate": 8.099071207430341e-06,
-      "loss": 0.017,
-      "step": 7192
-    },
-    {
-      "epoch": 423.12,
-      "learning_rate": 8.092879256965945e-06,
-      "loss": 0.0135,
-      "step": 7193
-    },
-    {
-      "epoch": 423.18,
-      "learning_rate": 8.08668730650155e-06,
-      "loss": 0.0161,
-      "step": 7194
-    },
-    {
-      "epoch": 423.24,
-      "learning_rate": 8.080495356037151e-06,
-      "loss": 0.0163,
-      "step": 7195
-    },
-    {
-      "epoch": 423.29,
-      "learning_rate": 8.074303405572756e-06,
-      "loss": 0.0135,
-      "step": 7196
-    },
-    {
-      "epoch": 423.35,
-      "learning_rate": 8.06811145510836e-06,
-      "loss": 0.0125,
-      "step": 7197
-    },
-    {
-      "epoch": 423.41,
-      "learning_rate": 8.061919504643963e-06,
-      "loss": 0.015,
-      "step": 7198
-    },
-    {
-      "epoch": 423.47,
-      "learning_rate": 8.055727554179566e-06,
-      "loss": 0.015,
-      "step": 7199
-    },
-    {
-      "epoch": 423.53,
-      "learning_rate": 8.049535603715171e-06,
-      "loss": 0.0146,
-      "step": 7200
-    },
-    {
-      "epoch": 423.53,
-      "eval_accuracy_NAT": 0.9940199459491486,
-      "eval_accuracy_REST": 0.5284552845528455,
-      "eval_accuracy_concrete": 0.9590676018828667,
-      "eval_accuracy_concrete tiles": 0.9538769738392646,
-      "eval_accuracy_dark bricks": 0.8489462703137263,
-      "eval_accuracy_grass": 0.9052182568970542,
-      "eval_accuracy_red bricks": 0.945738137082601,
-      "eval_accuracy_road": 0.9821590518602759,
-      "eval_accuracy_rough concrete": 0.6385272227816874,
-      "eval_accuracy_rough red sidewalk": 0.7739191902480843,
-      "eval_accuracy_speedway bricks": 0.9900040990545441,
-      "eval_accuracy_steel": 0.937069874604547,
-      "eval_accuracy_tiles": 0.24881093035232815,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848358120559624,
-      "eval_iou_REST": 0.44398907103825136,
-      "eval_iou_concrete": 0.898341782715146,
-      "eval_iou_concrete tiles": 0.8948066591496983,
-      "eval_iou_dark bricks": 0.8356225035797724,
-      "eval_iou_grass": 0.8486656044965988,
-      "eval_iou_red bricks": 0.9115451802445609,
-      "eval_iou_road": 0.9627250030644734,
-      "eval_iou_rough concrete": 0.6331023236852833,
-      "eval_iou_rough red sidewalk": 0.6083309689317634,
-      "eval_iou_speedway bricks": 0.980490530307209,
-      "eval_iou_steel": 0.8799775659001683,
-      "eval_iou_tiles": 0.24718593872268704,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08576717227697372,
-      "eval_mean_accuracy": 0.8235240645706904,
-      "eval_mean_iou": 0.7792014572224287,
-      "eval_overall_accuracy": 0.9810749312578622,
-      "eval_runtime": 5.6667,
-      "eval_samples_per_second": 20.823,
-      "eval_steps_per_second": 1.412,
-      "step": 7200
-    },
-    {
-      "epoch": 423.59,
-      "learning_rate": 8.043343653250775e-06,
-      "loss": 0.0167,
-      "step": 7201
-    },
-    {
-      "epoch": 423.65,
-      "learning_rate": 8.037151702786378e-06,
-      "loss": 0.018,
-      "step": 7202
-    },
-    {
-      "epoch": 423.71,
-      "learning_rate": 8.030959752321981e-06,
-      "loss": 0.0144,
-      "step": 7203
-    },
-    {
-      "epoch": 423.76,
-      "learning_rate": 8.024767801857586e-06,
-      "loss": 0.014,
-      "step": 7204
-    },
-    {
-      "epoch": 423.82,
-      "learning_rate": 8.01857585139319e-06,
-      "loss": 0.015,
-      "step": 7205
-    },
-    {
-      "epoch": 423.88,
-      "learning_rate": 8.012383900928793e-06,
-      "loss": 0.017,
-      "step": 7206
-    },
-    {
-      "epoch": 423.94,
-      "learning_rate": 8.006191950464396e-06,
-      "loss": 0.0154,
-      "step": 7207
-    },
-    {
-      "epoch": 424.0,
-      "learning_rate": 8.000000000000001e-06,
-      "loss": 0.0143,
-      "step": 7208
-    },
-    {
-      "epoch": 424.06,
-      "learning_rate": 7.993808049535605e-06,
-      "loss": 0.017,
-      "step": 7209
-    },
-    {
-      "epoch": 424.12,
-      "learning_rate": 7.987616099071208e-06,
-      "loss": 0.0127,
-      "step": 7210
-    },
-    {
-      "epoch": 424.18,
-      "learning_rate": 7.981424148606811e-06,
-      "loss": 0.0131,
-      "step": 7211
-    },
-    {
-      "epoch": 424.24,
-      "learning_rate": 7.975232198142416e-06,
-      "loss": 0.0126,
-      "step": 7212
-    },
-    {
-      "epoch": 424.29,
-      "learning_rate": 7.96904024767802e-06,
-      "loss": 0.0138,
-      "step": 7213
-    },
-    {
-      "epoch": 424.35,
-      "learning_rate": 7.962848297213623e-06,
-      "loss": 0.0129,
-      "step": 7214
-    },
-    {
-      "epoch": 424.41,
-      "learning_rate": 7.956656346749226e-06,
-      "loss": 0.0145,
-      "step": 7215
-    },
-    {
-      "epoch": 424.47,
-      "learning_rate": 7.950464396284831e-06,
-      "loss": 0.017,
-      "step": 7216
-    },
-    {
-      "epoch": 424.53,
-      "learning_rate": 7.944272445820433e-06,
-      "loss": 0.0187,
-      "step": 7217
-    },
-    {
-      "epoch": 424.59,
-      "learning_rate": 7.938080495356038e-06,
-      "loss": 0.0138,
-      "step": 7218
-    },
-    {
-      "epoch": 424.65,
-      "learning_rate": 7.931888544891641e-06,
-      "loss": 0.0155,
-      "step": 7219
-    },
-    {
-      "epoch": 424.71,
-      "learning_rate": 7.925696594427245e-06,
-      "loss": 0.0172,
-      "step": 7220
-    },
-    {
-      "epoch": 424.71,
-      "eval_accuracy_NAT": 0.9940274935954178,
-      "eval_accuracy_REST": 0.5284552845528455,
-      "eval_accuracy_concrete": 0.9597131364955993,
-      "eval_accuracy_concrete tiles": 0.9489983502238982,
-      "eval_accuracy_dark bricks": 0.8525831211836993,
-      "eval_accuracy_grass": 0.9055428130216081,
-      "eval_accuracy_red bricks": 0.9421682776801406,
-      "eval_accuracy_road": 0.9825219204136808,
-      "eval_accuracy_rough concrete": 0.6389502976421025,
-      "eval_accuracy_rough red sidewalk": 0.7775513426932072,
-      "eval_accuracy_speedway bricks": 0.9892598601797414,
-      "eval_accuracy_steel": 0.9387916256428454,
-      "eval_accuracy_tiles": 0.24144930170559545,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848158314064452,
-      "eval_iou_REST": 0.4433833560709413,
-      "eval_iou_concrete": 0.8982614845438861,
-      "eval_iou_concrete tiles": 0.8940648799875658,
-      "eval_iou_dark bricks": 0.8386255589550482,
-      "eval_iou_grass": 0.8488298247885915,
-      "eval_iou_red bricks": 0.9114334289661035,
-      "eval_iou_road": 0.962853719236796,
-      "eval_iou_rough concrete": 0.6333010465638215,
-      "eval_iou_rough red sidewalk": 0.6116089649175731,
-      "eval_iou_speedway bricks": 0.9804331772878797,
-      "eval_iou_steel": 0.8795531061092043,
-      "eval_iou_tiles": 0.23997350355293268,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08535595238208771,
-      "eval_mean_accuracy": 0.8230779096177216,
-      "eval_mean_iou": 0.7790106063374453,
-      "eval_overall_accuracy": 0.9810748019460904,
-      "eval_runtime": 5.5843,
-      "eval_samples_per_second": 21.131,
-      "eval_steps_per_second": 1.433,
-      "step": 7220
-    },
-    {
-      "epoch": 424.76,
-      "learning_rate": 7.919504643962848e-06,
-      "loss": 0.015,
-      "step": 7221
-    },
-    {
-      "epoch": 424.82,
-      "learning_rate": 7.913312693498453e-06,
-      "loss": 0.0147,
-      "step": 7222
-    },
-    {
-      "epoch": 424.88,
-      "learning_rate": 7.907120743034056e-06,
-      "loss": 0.0137,
-      "step": 7223
-    },
-    {
-      "epoch": 424.94,
-      "learning_rate": 7.90092879256966e-06,
-      "loss": 0.0186,
-      "step": 7224
-    },
-    {
-      "epoch": 425.0,
-      "learning_rate": 7.894736842105263e-06,
-      "loss": 0.0153,
-      "step": 7225
-    },
-    {
-      "epoch": 425.06,
-      "learning_rate": 7.888544891640868e-06,
-      "loss": 0.0139,
-      "step": 7226
-    },
-    {
-      "epoch": 425.12,
-      "learning_rate": 7.882352941176471e-06,
-      "loss": 0.013,
-      "step": 7227
-    },
-    {
-      "epoch": 425.18,
-      "learning_rate": 7.876160990712075e-06,
-      "loss": 0.0134,
-      "step": 7228
-    },
-    {
-      "epoch": 425.24,
-      "learning_rate": 7.869969040247678e-06,
-      "loss": 0.0144,
-      "step": 7229
-    },
-    {
-      "epoch": 425.29,
-      "learning_rate": 7.863777089783283e-06,
-      "loss": 0.0138,
-      "step": 7230
-    },
-    {
-      "epoch": 425.35,
-      "learning_rate": 7.857585139318886e-06,
-      "loss": 0.016,
-      "step": 7231
-    },
-    {
-      "epoch": 425.41,
-      "learning_rate": 7.85139318885449e-06,
-      "loss": 0.0158,
-      "step": 7232
-    },
-    {
-      "epoch": 425.47,
-      "learning_rate": 7.845201238390093e-06,
-      "loss": 0.0144,
-      "step": 7233
-    },
-    {
-      "epoch": 425.53,
-      "learning_rate": 7.839009287925698e-06,
-      "loss": 0.0167,
-      "step": 7234
-    },
-    {
-      "epoch": 425.59,
-      "learning_rate": 7.832817337461301e-06,
-      "loss": 0.0139,
-      "step": 7235
-    },
-    {
-      "epoch": 425.65,
-      "learning_rate": 7.826625386996905e-06,
-      "loss": 0.0167,
-      "step": 7236
-    },
-    {
-      "epoch": 425.71,
-      "learning_rate": 7.820433436532508e-06,
-      "loss": 0.017,
-      "step": 7237
-    },
-    {
-      "epoch": 425.76,
-      "learning_rate": 7.814241486068113e-06,
-      "loss": 0.0153,
-      "step": 7238
-    },
-    {
-      "epoch": 425.82,
-      "learning_rate": 7.808049535603716e-06,
-      "loss": 0.014,
-      "step": 7239
-    },
-    {
-      "epoch": 425.88,
-      "learning_rate": 7.80185758513932e-06,
-      "loss": 0.0169,
-      "step": 7240
-    },
-    {
-      "epoch": 425.88,
-      "eval_accuracy_NAT": 0.994186748931698,
-      "eval_accuracy_REST": 0.5300813008130081,
-      "eval_accuracy_concrete": 0.9600062833564437,
-      "eval_accuracy_concrete tiles": 0.9582842328541127,
-      "eval_accuracy_dark bricks": 0.8559902762092529,
-      "eval_accuracy_grass": 0.9020739420227322,
-      "eval_accuracy_red bricks": 0.945628295254833,
-      "eval_accuracy_road": 0.9823562514128626,
-      "eval_accuracy_rough concrete": 0.6327710211597019,
-      "eval_accuracy_rough red sidewalk": 0.7743102294347228,
-      "eval_accuracy_speedway bricks": 0.9893731325482067,
-      "eval_accuracy_steel": 0.9340711790371189,
-      "eval_accuracy_tiles": 0.23999515283710504,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847386600462781,
-      "eval_iou_REST": 0.42782152230971127,
-      "eval_iou_concrete": 0.8978052018332162,
-      "eval_iou_concrete tiles": 0.8896571341049822,
-      "eval_iou_dark bricks": 0.8418620193054372,
-      "eval_iou_grass": 0.8481170539373682,
-      "eval_iou_red bricks": 0.9108125264494287,
-      "eval_iou_road": 0.9622761023699541,
-      "eval_iou_rough concrete": 0.627892125849234,
-      "eval_iou_rough red sidewalk": 0.6092541640779858,
-      "eval_iou_speedway bricks": 0.9806280458433178,
-      "eval_iou_steel": 0.8800714810419845,
-      "eval_iou_tiles": 0.2387223142986289,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08692407608032227,
-      "eval_mean_accuracy": 0.823009849682446,
-      "eval_mean_iou": 0.776896796266733,
-      "eval_overall_accuracy": 0.9810156741384732,
-      "eval_runtime": 9.1494,
-      "eval_samples_per_second": 12.897,
-      "eval_steps_per_second": 0.874,
-      "step": 7240
-    },
-    {
-      "epoch": 425.94,
-      "learning_rate": 7.795665634674923e-06,
-      "loss": 0.0152,
-      "step": 7241
-    },
-    {
-      "epoch": 426.0,
-      "learning_rate": 7.789473684210528e-06,
-      "loss": 0.0142,
-      "step": 7242
-    },
-    {
-      "epoch": 426.06,
-      "learning_rate": 7.78328173374613e-06,
-      "loss": 0.015,
-      "step": 7243
-    },
-    {
-      "epoch": 426.12,
-      "learning_rate": 7.777089783281735e-06,
-      "loss": 0.0154,
-      "step": 7244
-    },
-    {
-      "epoch": 426.18,
-      "learning_rate": 7.770897832817338e-06,
-      "loss": 0.0139,
-      "step": 7245
-    },
-    {
-      "epoch": 426.24,
-      "learning_rate": 7.764705882352941e-06,
-      "loss": 0.0152,
-      "step": 7246
-    },
-    {
-      "epoch": 426.29,
-      "learning_rate": 7.758513931888544e-06,
-      "loss": 0.0141,
-      "step": 7247
-    },
-    {
-      "epoch": 426.35,
-      "learning_rate": 7.75232198142415e-06,
-      "loss": 0.0141,
-      "step": 7248
-    },
-    {
-      "epoch": 426.41,
-      "learning_rate": 7.746130030959753e-06,
-      "loss": 0.0152,
-      "step": 7249
-    },
-    {
-      "epoch": 426.47,
-      "learning_rate": 7.739938080495356e-06,
-      "loss": 0.0154,
-      "step": 7250
-    },
-    {
-      "epoch": 426.53,
-      "learning_rate": 7.73374613003096e-06,
-      "loss": 0.0163,
-      "step": 7251
-    },
-    {
-      "epoch": 426.59,
-      "learning_rate": 7.727554179566565e-06,
-      "loss": 0.0152,
-      "step": 7252
-    },
-    {
-      "epoch": 426.65,
-      "learning_rate": 7.721362229102168e-06,
-      "loss": 0.0145,
-      "step": 7253
-    },
-    {
-      "epoch": 426.71,
-      "learning_rate": 7.715170278637771e-06,
-      "loss": 0.0156,
-      "step": 7254
-    },
-    {
-      "epoch": 426.76,
-      "learning_rate": 7.708978328173374e-06,
-      "loss": 0.0136,
-      "step": 7255
-    },
-    {
-      "epoch": 426.82,
-      "learning_rate": 7.70278637770898e-06,
-      "loss": 0.0165,
-      "step": 7256
-    },
-    {
-      "epoch": 426.88,
-      "learning_rate": 7.696594427244583e-06,
-      "loss": 0.015,
-      "step": 7257
-    },
-    {
-      "epoch": 426.94,
-      "learning_rate": 7.690402476780186e-06,
-      "loss": 0.0155,
-      "step": 7258
-    },
-    {
-      "epoch": 427.0,
-      "learning_rate": 7.68421052631579e-06,
-      "loss": 0.0168,
-      "step": 7259
-    },
-    {
-      "epoch": 427.06,
-      "learning_rate": 7.678018575851394e-06,
-      "loss": 0.0159,
-      "step": 7260
-    },
-    {
-      "epoch": 427.06,
-      "eval_accuracy_NAT": 0.9942667539821517,
-      "eval_accuracy_REST": 0.5056910569105691,
-      "eval_accuracy_concrete": 0.9567300709820534,
-      "eval_accuracy_concrete tiles": 0.9463115720009427,
-      "eval_accuracy_dark bricks": 0.8580910361196715,
-      "eval_accuracy_grass": 0.9059553683286211,
-      "eval_accuracy_red bricks": 0.9405755711775043,
-      "eval_accuracy_road": 0.9829809838707867,
-      "eval_accuracy_rough concrete": 0.6364595188864193,
-      "eval_accuracy_rough red sidewalk": 0.7612705765572008,
-      "eval_accuracy_speedway bricks": 0.9903589962937144,
-      "eval_accuracy_steel": 0.9334314573160675,
-      "eval_accuracy_tiles": 0.2368444969553758,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847495967956165,
-      "eval_iou_REST": 0.43496503496503497,
-      "eval_iou_concrete": 0.8981296671762671,
-      "eval_iou_concrete tiles": 0.8946125395481485,
-      "eval_iou_dark bricks": 0.8430789917862835,
-      "eval_iou_grass": 0.8492534520028135,
-      "eval_iou_red bricks": 0.9098443393720448,
-      "eval_iou_road": 0.9620173255222212,
-      "eval_iou_rough concrete": 0.6315335130788515,
-      "eval_iou_rough red sidewalk": 0.6033243539620482,
-      "eval_iou_speedway bricks": 0.9807830913605456,
-      "eval_iou_steel": 0.8816420844928141,
-      "eval_iou_tiles": 0.23579442634817227,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08696701377630234,
-      "eval_mean_accuracy": 0.8191513430293137,
-      "eval_mean_iou": 0.7776714166469894,
-      "eval_overall_accuracy": 0.9810895434880661,
-      "eval_runtime": 5.8775,
-      "eval_samples_per_second": 20.077,
-      "eval_steps_per_second": 1.361,
-      "step": 7260
-    },
-    {
-      "epoch": 427.12,
-      "learning_rate": 7.671826625386998e-06,
-      "loss": 0.0141,
-      "step": 7261
-    },
-    {
-      "epoch": 427.18,
-      "learning_rate": 7.665634674922601e-06,
-      "loss": 0.014,
-      "step": 7262
-    },
-    {
-      "epoch": 427.24,
-      "learning_rate": 7.659442724458204e-06,
-      "loss": 0.0169,
-      "step": 7263
-    },
-    {
-      "epoch": 427.29,
-      "learning_rate": 7.65325077399381e-06,
-      "loss": 0.0143,
-      "step": 7264
-    },
-    {
-      "epoch": 427.35,
-      "learning_rate": 7.647058823529413e-06,
-      "loss": 0.0169,
-      "step": 7265
-    },
-    {
-      "epoch": 427.41,
-      "learning_rate": 7.640866873065016e-06,
-      "loss": 0.0144,
-      "step": 7266
-    },
-    {
-      "epoch": 427.47,
-      "learning_rate": 7.63467492260062e-06,
-      "loss": 0.0146,
-      "step": 7267
-    },
-    {
-      "epoch": 427.53,
-      "learning_rate": 7.628482972136224e-06,
-      "loss": 0.0161,
-      "step": 7268
-    },
-    {
-      "epoch": 427.59,
-      "learning_rate": 7.622291021671828e-06,
-      "loss": 0.0138,
-      "step": 7269
-    },
-    {
-      "epoch": 427.65,
-      "learning_rate": 7.61609907120743e-06,
-      "loss": 0.0156,
-      "step": 7270
-    },
-    {
-      "epoch": 427.71,
-      "learning_rate": 7.609907120743034e-06,
-      "loss": 0.0159,
-      "step": 7271
-    },
-    {
-      "epoch": 427.76,
-      "learning_rate": 7.603715170278639e-06,
-      "loss": 0.0155,
-      "step": 7272
-    },
-    {
-      "epoch": 427.82,
-      "learning_rate": 7.597523219814241e-06,
-      "loss": 0.0131,
-      "step": 7273
-    },
-    {
-      "epoch": 427.88,
-      "learning_rate": 7.591331269349845e-06,
-      "loss": 0.0165,
-      "step": 7274
-    },
-    {
-      "epoch": 427.94,
-      "learning_rate": 7.585139318885449e-06,
-      "loss": 0.014,
-      "step": 7275
-    },
-    {
-      "epoch": 428.0,
-      "learning_rate": 7.578947368421054e-06,
-      "loss": 0.0148,
-      "step": 7276
-    },
-    {
-      "epoch": 428.06,
-      "learning_rate": 7.572755417956656e-06,
-      "loss": 0.0157,
-      "step": 7277
-    },
-    {
-      "epoch": 428.12,
-      "learning_rate": 7.56656346749226e-06,
-      "loss": 0.0163,
-      "step": 7278
-    },
-    {
-      "epoch": 428.18,
-      "learning_rate": 7.560371517027864e-06,
-      "loss": 0.0157,
-      "step": 7279
-    },
-    {
-      "epoch": 428.24,
-      "learning_rate": 7.5541795665634686e-06,
-      "loss": 0.0171,
-      "step": 7280
-    },
-    {
-      "epoch": 428.24,
-      "eval_accuracy_NAT": 0.9939458711921922,
-      "eval_accuracy_REST": 0.5821138211382114,
-      "eval_accuracy_concrete": 0.9589193085556077,
-      "eval_accuracy_concrete tiles": 0.9557152957812868,
-      "eval_accuracy_dark bricks": 0.8552198380644297,
-      "eval_accuracy_grass": 0.9037449802616887,
-      "eval_accuracy_red bricks": 0.9461775043936731,
-      "eval_accuracy_road": 0.9821948577410978,
-      "eval_accuracy_rough concrete": 0.6338614817154196,
-      "eval_accuracy_rough red sidewalk": 0.7752126275577347,
-      "eval_accuracy_speedway bricks": 0.9911681825628413,
-      "eval_accuracy_steel": 0.92588474013804,
-      "eval_accuracy_tiles": 0.24753855009239903,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848456941540871,
-      "eval_iou_REST": 0.4710526315789474,
-      "eval_iou_concrete": 0.8976870778480565,
-      "eval_iou_concrete tiles": 0.8969674179919928,
-      "eval_iou_dark bricks": 0.8404042209013618,
-      "eval_iou_grass": 0.848272419731442,
-      "eval_iou_red bricks": 0.9050220634587098,
-      "eval_iou_road": 0.9624989264160311,
-      "eval_iou_rough concrete": 0.6286024949327219,
-      "eval_iou_rough red sidewalk": 0.6074983352092922,
-      "eval_iou_speedway bricks": 0.9806867269337137,
-      "eval_iou_steel": 0.8814901695818503,
-      "eval_iou_tiles": 0.2458626707588614,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08651664853096008,
-      "eval_mean_accuracy": 0.8270536199380477,
-      "eval_mean_iou": 0.7808377576536206,
-      "eval_overall_accuracy": 0.9810698234428794,
-      "eval_runtime": 5.6552,
-      "eval_samples_per_second": 20.866,
-      "eval_steps_per_second": 1.415,
-      "step": 7280
-    },
-    {
-      "epoch": 428.29,
-      "learning_rate": 7.547987616099071e-06,
-      "loss": 0.0174,
-      "step": 7281
-    },
-    {
-      "epoch": 428.35,
-      "learning_rate": 7.541795665634675e-06,
-      "loss": 0.0154,
-      "step": 7282
-    },
-    {
-      "epoch": 428.41,
-      "learning_rate": 7.535603715170279e-06,
-      "loss": 0.0167,
-      "step": 7283
-    },
-    {
-      "epoch": 428.47,
-      "learning_rate": 7.529411764705882e-06,
-      "loss": 0.0117,
-      "step": 7284
-    },
-    {
-      "epoch": 428.53,
-      "learning_rate": 7.523219814241486e-06,
-      "loss": 0.0163,
-      "step": 7285
-    },
-    {
-      "epoch": 428.59,
-      "learning_rate": 7.51702786377709e-06,
-      "loss": 0.0134,
-      "step": 7286
-    },
-    {
-      "epoch": 428.65,
-      "learning_rate": 7.510835913312694e-06,
-      "loss": 0.0143,
-      "step": 7287
-    },
-    {
-      "epoch": 428.71,
-      "learning_rate": 7.504643962848297e-06,
-      "loss": 0.0145,
-      "step": 7288
-    },
-    {
-      "epoch": 428.76,
-      "learning_rate": 7.498452012383901e-06,
-      "loss": 0.0157,
-      "step": 7289
-    },
-    {
-      "epoch": 428.82,
-      "learning_rate": 7.492260061919505e-06,
-      "loss": 0.0143,
-      "step": 7290
-    },
-    {
-      "epoch": 428.88,
-      "learning_rate": 7.486068111455109e-06,
-      "loss": 0.0151,
-      "step": 7291
-    },
-    {
-      "epoch": 428.94,
-      "learning_rate": 7.479876160990712e-06,
-      "loss": 0.0141,
-      "step": 7292
-    },
-    {
-      "epoch": 429.0,
-      "learning_rate": 7.473684210526316e-06,
-      "loss": 0.015,
-      "step": 7293
-    },
-    {
-      "epoch": 429.06,
-      "learning_rate": 7.46749226006192e-06,
-      "loss": 0.0143,
-      "step": 7294
-    },
-    {
-      "epoch": 429.12,
-      "learning_rate": 7.461300309597524e-06,
-      "loss": 0.0141,
-      "step": 7295
-    },
-    {
-      "epoch": 429.18,
-      "learning_rate": 7.455108359133127e-06,
-      "loss": 0.0139,
-      "step": 7296
-    },
-    {
-      "epoch": 429.24,
-      "learning_rate": 7.448916408668731e-06,
-      "loss": 0.0148,
-      "step": 7297
-    },
-    {
-      "epoch": 429.29,
-      "learning_rate": 7.442724458204335e-06,
-      "loss": 0.0166,
-      "step": 7298
-    },
-    {
-      "epoch": 429.35,
-      "learning_rate": 7.436532507739938e-06,
-      "loss": 0.0143,
-      "step": 7299
-    },
-    {
-      "epoch": 429.41,
-      "learning_rate": 7.430340557275542e-06,
-      "loss": 0.0131,
-      "step": 7300
-    },
-    {
-      "epoch": 429.41,
-      "eval_accuracy_NAT": 0.9938198254994965,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9620904161233067,
-      "eval_accuracy_concrete tiles": 0.9501531934951685,
-      "eval_accuracy_dark bricks": 0.8604693451754302,
-      "eval_accuracy_grass": 0.9061597535265908,
-      "eval_accuracy_red bricks": 0.9388181019332161,
-      "eval_accuracy_road": 0.981784426151974,
-      "eval_accuracy_rough concrete": 0.635982814818346,
-      "eval_accuracy_rough red sidewalk": 0.770332157709112,
-      "eval_accuracy_speedway bricks": 0.9895916231227804,
-      "eval_accuracy_steel": 0.9332915181895876,
-      "eval_accuracy_tiles": 0.24593292738344089,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848936829304024,
-      "eval_iou_REST": 0.43626806833114323,
-      "eval_iou_concrete": 0.8969797174988241,
-      "eval_iou_concrete tiles": 0.8959287079425752,
-      "eval_iou_dark bricks": 0.846904893108954,
-      "eval_iou_grass": 0.8497307033655425,
-      "eval_iou_red bricks": 0.9078549046683307,
-      "eval_iou_road": 0.9627284632585331,
-      "eval_iou_rough concrete": 0.6298873367445099,
-      "eval_iou_rough red sidewalk": 0.6099860064906065,
-      "eval_iou_speedway bricks": 0.9809052501983976,
-      "eval_iou_steel": 0.8819116480278072,
-      "eval_iou_tiles": 0.24445180523352103,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08648530393838882,
-      "eval_mean_accuracy": 0.8237125770386486,
-      "eval_mean_iou": 0.7791100913691652,
-      "eval_overall_accuracy": 0.9811300180726132,
-      "eval_runtime": 5.5315,
-      "eval_samples_per_second": 21.332,
-      "eval_steps_per_second": 1.446,
-      "step": 7300
-    },
-    {
-      "epoch": 429.47,
-      "learning_rate": 7.424148606811146e-06,
-      "loss": 0.0153,
-      "step": 7301
-    },
-    {
-      "epoch": 429.53,
-      "learning_rate": 7.41795665634675e-06,
-      "loss": 0.0157,
-      "step": 7302
-    },
-    {
-      "epoch": 429.59,
-      "learning_rate": 7.411764705882353e-06,
-      "loss": 0.0147,
-      "step": 7303
-    },
-    {
-      "epoch": 429.65,
-      "learning_rate": 7.405572755417957e-06,
-      "loss": 0.0166,
-      "step": 7304
-    },
-    {
-      "epoch": 429.71,
-      "learning_rate": 7.399380804953561e-06,
-      "loss": 0.0167,
-      "step": 7305
-    },
-    {
-      "epoch": 429.76,
-      "learning_rate": 7.393188854489165e-06,
-      "loss": 0.0161,
-      "step": 7306
-    },
-    {
-      "epoch": 429.82,
-      "learning_rate": 7.386996904024768e-06,
-      "loss": 0.0148,
-      "step": 7307
-    },
-    {
-      "epoch": 429.88,
-      "learning_rate": 7.380804953560372e-06,
-      "loss": 0.0129,
-      "step": 7308
-    },
-    {
-      "epoch": 429.94,
-      "learning_rate": 7.374613003095976e-06,
-      "loss": 0.014,
-      "step": 7309
-    },
-    {
-      "epoch": 430.0,
-      "learning_rate": 7.3684210526315784e-06,
-      "loss": 0.0176,
-      "step": 7310
-    },
-    {
-      "epoch": 430.06,
-      "learning_rate": 7.362229102167183e-06,
-      "loss": 0.0145,
-      "step": 7311
-    },
-    {
-      "epoch": 430.12,
-      "learning_rate": 7.356037151702787e-06,
-      "loss": 0.0149,
-      "step": 7312
-    },
-    {
-      "epoch": 430.18,
-      "learning_rate": 7.349845201238391e-06,
-      "loss": 0.0157,
-      "step": 7313
-    },
-    {
-      "epoch": 430.24,
-      "learning_rate": 7.343653250773993e-06,
-      "loss": 0.0145,
-      "step": 7314
-    },
-    {
-      "epoch": 430.29,
-      "learning_rate": 7.337461300309598e-06,
-      "loss": 0.0145,
-      "step": 7315
-    },
-    {
-      "epoch": 430.35,
-      "learning_rate": 7.331269349845202e-06,
-      "loss": 0.0158,
-      "step": 7316
-    },
-    {
-      "epoch": 430.41,
-      "learning_rate": 7.325077399380806e-06,
-      "loss": 0.0153,
-      "step": 7317
-    },
-    {
-      "epoch": 430.47,
-      "learning_rate": 7.318885448916408e-06,
-      "loss": 0.0131,
-      "step": 7318
-    },
-    {
-      "epoch": 430.53,
-      "learning_rate": 7.312693498452013e-06,
-      "loss": 0.0166,
-      "step": 7319
-    },
-    {
-      "epoch": 430.59,
-      "learning_rate": 7.306501547987617e-06,
-      "loss": 0.0164,
-      "step": 7320
-    },
-    {
-      "epoch": 430.59,
-      "eval_accuracy_NAT": 0.993843977967558,
-      "eval_accuracy_REST": 0.583739837398374,
-      "eval_accuracy_concrete": 0.9597674087957817,
-      "eval_accuracy_concrete tiles": 0.9560452510016497,
-      "eval_accuracy_dark bricks": 0.858052753478935,
-      "eval_accuracy_grass": 0.9064256435295053,
-      "eval_accuracy_red bricks": 0.9421682776801406,
-      "eval_accuracy_road": 0.9821831005862011,
-      "eval_accuracy_rough concrete": 0.6358219271953712,
-      "eval_accuracy_rough red sidewalk": 0.7537881921205604,
-      "eval_accuracy_speedway bricks": 0.9904462211583873,
-      "eval_accuracy_steel": 0.9270992118428484,
-      "eval_accuracy_tiles": 0.2503862582931928,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984891781878804,
-      "eval_iou_REST": 0.45558375634517767,
-      "eval_iou_concrete": 0.8967449042376274,
-      "eval_iou_concrete tiles": 0.8956525579033362,
-      "eval_iou_dark bricks": 0.8425976833251099,
-      "eval_iou_grass": 0.8488668023053856,
-      "eval_iou_red bricks": 0.905133751912626,
-      "eval_iou_road": 0.9610474635174573,
-      "eval_iou_rough concrete": 0.6312853128235468,
-      "eval_iou_rough red sidewalk": 0.5967329054995297,
-      "eval_iou_speedway bricks": 0.9807326604889053,
-      "eval_iou_steel": 0.8809759572573463,
-      "eval_iou_tiles": 0.24896828026628912,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0864110216498375,
-      "eval_mean_accuracy": 0.8261360046960389,
-      "eval_mean_iou": 0.7791702936739339,
-      "eval_overall_accuracy": 0.9809896824723583,
-      "eval_runtime": 5.6022,
-      "eval_samples_per_second": 21.063,
-      "eval_steps_per_second": 1.428,
-      "step": 7320
-    },
-    {
-      "epoch": 430.65,
-      "learning_rate": 7.300309597523221e-06,
-      "loss": 0.0143,
-      "step": 7321
-    },
-    {
-      "epoch": 430.71,
-      "learning_rate": 7.294117647058823e-06,
-      "loss": 0.0154,
-      "step": 7322
-    },
-    {
-      "epoch": 430.76,
-      "learning_rate": 7.2879256965944276e-06,
-      "loss": 0.0192,
-      "step": 7323
-    },
-    {
-      "epoch": 430.82,
-      "learning_rate": 7.281733746130032e-06,
-      "loss": 0.0169,
-      "step": 7324
-    },
-    {
-      "epoch": 430.88,
-      "learning_rate": 7.275541795665634e-06,
-      "loss": 0.0134,
-      "step": 7325
-    },
-    {
-      "epoch": 430.94,
-      "learning_rate": 7.269349845201238e-06,
-      "loss": 0.0125,
-      "step": 7326
-    },
-    {
-      "epoch": 431.0,
-      "learning_rate": 7.2631578947368426e-06,
-      "loss": 0.015,
-      "step": 7327
-    },
-    {
-      "epoch": 431.06,
-      "learning_rate": 7.256965944272447e-06,
-      "loss": 0.0155,
-      "step": 7328
-    },
-    {
-      "epoch": 431.12,
-      "learning_rate": 7.250773993808049e-06,
-      "loss": 0.0143,
-      "step": 7329
-    },
-    {
-      "epoch": 431.18,
-      "learning_rate": 7.244582043343653e-06,
-      "loss": 0.0148,
-      "step": 7330
-    },
-    {
-      "epoch": 431.24,
-      "learning_rate": 7.2383900928792575e-06,
-      "loss": 0.0138,
-      "step": 7331
-    },
-    {
-      "epoch": 431.29,
-      "learning_rate": 7.232198142414862e-06,
-      "loss": 0.0155,
-      "step": 7332
-    },
-    {
-      "epoch": 431.35,
-      "learning_rate": 7.226006191950464e-06,
-      "loss": 0.0152,
-      "step": 7333
-    },
-    {
-      "epoch": 431.41,
-      "learning_rate": 7.219814241486068e-06,
-      "loss": 0.0136,
-      "step": 7334
-    },
-    {
-      "epoch": 431.47,
-      "learning_rate": 7.2136222910216725e-06,
-      "loss": 0.0157,
-      "step": 7335
-    },
-    {
-      "epoch": 431.53,
-      "learning_rate": 7.207430340557277e-06,
-      "loss": 0.0148,
-      "step": 7336
-    },
-    {
-      "epoch": 431.59,
-      "learning_rate": 7.201238390092879e-06,
-      "loss": 0.0148,
-      "step": 7337
-    },
-    {
-      "epoch": 431.65,
-      "learning_rate": 7.195046439628483e-06,
-      "loss": 0.0162,
-      "step": 7338
-    },
-    {
-      "epoch": 431.71,
-      "learning_rate": 7.1888544891640875e-06,
-      "loss": 0.0165,
-      "step": 7339
-    },
-    {
-      "epoch": 431.76,
-      "learning_rate": 7.18266253869969e-06,
-      "loss": 0.0132,
-      "step": 7340
-    },
-    {
-      "epoch": 431.76,
-      "eval_accuracy_NAT": 0.9938640870536894,
-      "eval_accuracy_REST": 0.45691056910569106,
-      "eval_accuracy_concrete": 0.9581820461115808,
-      "eval_accuracy_concrete tiles": 0.9508838086259722,
-      "eval_accuracy_dark bricks": 0.8752512298298336,
-      "eval_accuracy_grass": 0.9055333507439243,
-      "eval_accuracy_red bricks": 0.9367311072056239,
-      "eval_accuracy_road": 0.9826929335758158,
-      "eval_accuracy_rough concrete": 0.6360543204285569,
-      "eval_accuracy_rough red sidewalk": 0.7511185976733168,
-      "eval_accuracy_speedway bricks": 0.9910377736787233,
-      "eval_accuracy_steel": 0.9262021020855927,
-      "eval_accuracy_tiles": 0.24650852797721834,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847950124273177,
-      "eval_iou_REST": 0.3844049247606019,
-      "eval_iou_concrete": 0.8974885219492549,
-      "eval_iou_concrete tiles": 0.8956820956820957,
-      "eval_iou_dark bricks": 0.8593813871099605,
-      "eval_iou_grass": 0.8470823685198168,
-      "eval_iou_red bricks": 0.9084420772303595,
-      "eval_iou_road": 0.9604690106753833,
-      "eval_iou_rough concrete": 0.631344228164499,
-      "eval_iou_rough red sidewalk": 0.5933056133056133,
-      "eval_iou_speedway bricks": 0.9813044244781384,
-      "eval_iou_steel": 0.8817524568987051,
-      "eval_iou_tiles": 0.2453860072376357,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08728453516960144,
-      "eval_mean_accuracy": 0.8162284964688877,
-      "eval_mean_iou": 0.7746798560337986,
-      "eval_overall_accuracy": 0.9810570215774794,
-      "eval_runtime": 7.1737,
-      "eval_samples_per_second": 16.449,
-      "eval_steps_per_second": 1.115,
-      "step": 7340
-    },
-    {
-      "epoch": 431.82,
-      "learning_rate": 7.176470588235294e-06,
-      "loss": 0.0147,
-      "step": 7341
-    },
-    {
-      "epoch": 431.88,
-      "learning_rate": 7.170278637770898e-06,
-      "loss": 0.0156,
-      "step": 7342
-    },
-    {
-      "epoch": 431.94,
-      "learning_rate": 7.1640866873065025e-06,
-      "loss": 0.0154,
-      "step": 7343
-    },
-    {
-      "epoch": 432.0,
-      "learning_rate": 7.157894736842105e-06,
-      "loss": 0.0147,
-      "step": 7344
-    },
-    {
-      "epoch": 432.06,
-      "learning_rate": 7.151702786377709e-06,
-      "loss": 0.0155,
-      "step": 7345
-    },
-    {
-      "epoch": 432.12,
-      "learning_rate": 7.145510835913313e-06,
-      "loss": 0.0165,
-      "step": 7346
-    },
-    {
-      "epoch": 432.18,
-      "learning_rate": 7.1393188854489175e-06,
-      "loss": 0.0142,
-      "step": 7347
-    },
-    {
-      "epoch": 432.24,
-      "learning_rate": 7.13312693498452e-06,
-      "loss": 0.015,
-      "step": 7348
-    },
-    {
-      "epoch": 432.29,
-      "learning_rate": 7.126934984520124e-06,
-      "loss": 0.0163,
-      "step": 7349
-    },
-    {
-      "epoch": 432.35,
-      "learning_rate": 7.120743034055728e-06,
-      "loss": 0.0159,
-      "step": 7350
-    },
-    {
-      "epoch": 432.41,
-      "learning_rate": 7.114551083591331e-06,
-      "loss": 0.016,
-      "step": 7351
-    },
-    {
-      "epoch": 432.47,
-      "learning_rate": 7.108359133126935e-06,
-      "loss": 0.0154,
-      "step": 7352
-    },
-    {
-      "epoch": 432.53,
-      "learning_rate": 7.102167182662539e-06,
-      "loss": 0.0138,
-      "step": 7353
-    },
-    {
-      "epoch": 432.59,
-      "learning_rate": 7.095975232198143e-06,
-      "loss": 0.0162,
-      "step": 7354
-    },
-    {
-      "epoch": 432.65,
-      "learning_rate": 7.089783281733746e-06,
-      "loss": 0.0144,
-      "step": 7355
-    },
-    {
-      "epoch": 432.71,
-      "learning_rate": 7.08359133126935e-06,
-      "loss": 0.0154,
-      "step": 7356
-    },
-    {
-      "epoch": 432.76,
-      "learning_rate": 7.077399380804954e-06,
-      "loss": 0.0137,
-      "step": 7357
-    },
-    {
-      "epoch": 432.82,
-      "learning_rate": 7.071207430340558e-06,
-      "loss": 0.0146,
-      "step": 7358
-    },
-    {
-      "epoch": 432.88,
-      "learning_rate": 7.065015479876161e-06,
-      "loss": 0.0153,
-      "step": 7359
-    },
-    {
-      "epoch": 432.94,
-      "learning_rate": 7.058823529411765e-06,
-      "loss": 0.0127,
-      "step": 7360
-    },
-    {
-      "epoch": 432.94,
-      "eval_accuracy_NAT": 0.994130141584679,
-      "eval_accuracy_REST": 0.5154471544715448,
-      "eval_accuracy_concrete": 0.9588455440912753,
-      "eval_accuracy_concrete tiles": 0.9555974546311572,
-      "eval_accuracy_dark bricks": 0.8660394694025993,
-      "eval_accuracy_grass": 0.9027183231329979,
-      "eval_accuracy_red bricks": 0.9387631810193322,
-      "eval_accuracy_road": 0.9819741438787175,
-      "eval_accuracy_rough concrete": 0.6431095406360424,
-      "eval_accuracy_rough red sidewalk": 0.7603832184029057,
-      "eval_accuracy_speedway bricks": 0.9908787468134649,
-      "eval_accuracy_steel": 0.9249351532083544,
-      "eval_accuracy_tiles": 0.24965918385894756,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848221090813075,
-      "eval_iou_REST": 0.4318801089918256,
-      "eval_iou_concrete": 0.8984396655147099,
-      "eval_iou_concrete tiles": 0.8971941936625951,
-      "eval_iou_dark bricks": 0.8503206709422793,
-      "eval_iou_grass": 0.8481464260726316,
-      "eval_iou_red bricks": 0.906790450928382,
-      "eval_iou_road": 0.9608736531868564,
-      "eval_iou_rough concrete": 0.6372014760147602,
-      "eval_iou_rough red sidewalk": 0.6001281982800064,
-      "eval_iou_speedway bricks": 0.9809274174121396,
-      "eval_iou_steel": 0.8808082394543837,
-      "eval_iou_tiles": 0.24844739222188725,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08625509589910507,
-      "eval_mean_accuracy": 0.8217293273178475,
-      "eval_mean_iou": 0.7789215385972125,
-      "eval_overall_accuracy": 0.9811146946276649,
-      "eval_runtime": 5.7516,
-      "eval_samples_per_second": 20.516,
-      "eval_steps_per_second": 1.391,
-      "step": 7360
-    },
-    {
-      "epoch": 433.0,
-      "learning_rate": 7.052631578947369e-06,
-      "loss": 0.0123,
-      "step": 7361
-    },
-    {
-      "epoch": 433.06,
-      "learning_rate": 7.046439628482973e-06,
-      "loss": 0.0159,
-      "step": 7362
-    },
-    {
-      "epoch": 433.12,
-      "learning_rate": 7.040247678018576e-06,
-      "loss": 0.0128,
-      "step": 7363
-    },
-    {
-      "epoch": 433.18,
-      "learning_rate": 7.03405572755418e-06,
-      "loss": 0.0145,
-      "step": 7364
-    },
-    {
-      "epoch": 433.24,
-      "learning_rate": 7.027863777089784e-06,
-      "loss": 0.0148,
-      "step": 7365
-    },
-    {
-      "epoch": 433.29,
-      "learning_rate": 7.0216718266253866e-06,
-      "loss": 0.0139,
-      "step": 7366
-    },
-    {
-      "epoch": 433.35,
-      "learning_rate": 7.015479876160991e-06,
-      "loss": 0.0148,
-      "step": 7367
-    },
-    {
-      "epoch": 433.41,
-      "learning_rate": 7.009287925696595e-06,
-      "loss": 0.0167,
-      "step": 7368
-    },
-    {
-      "epoch": 433.47,
-      "learning_rate": 7.003095975232199e-06,
-      "loss": 0.0152,
-      "step": 7369
-    },
-    {
-      "epoch": 433.53,
-      "learning_rate": 6.9969040247678016e-06,
-      "loss": 0.0151,
-      "step": 7370
-    },
-    {
-      "epoch": 433.59,
-      "learning_rate": 6.990712074303406e-06,
-      "loss": 0.0175,
-      "step": 7371
-    },
-    {
-      "epoch": 433.65,
-      "learning_rate": 6.98452012383901e-06,
-      "loss": 0.0121,
-      "step": 7372
-    },
-    {
-      "epoch": 433.71,
-      "learning_rate": 6.978328173374614e-06,
-      "loss": 0.0162,
-      "step": 7373
-    },
-    {
-      "epoch": 433.76,
-      "learning_rate": 6.9721362229102165e-06,
-      "loss": 0.0147,
-      "step": 7374
-    },
-    {
-      "epoch": 433.82,
-      "learning_rate": 6.965944272445821e-06,
-      "loss": 0.0173,
-      "step": 7375
-    },
-    {
-      "epoch": 433.88,
-      "learning_rate": 6.959752321981425e-06,
-      "loss": 0.0147,
-      "step": 7376
-    },
-    {
-      "epoch": 433.94,
-      "learning_rate": 6.953560371517029e-06,
-      "loss": 0.0147,
-      "step": 7377
-    },
-    {
-      "epoch": 434.0,
-      "learning_rate": 6.9473684210526315e-06,
-      "loss": 0.0138,
-      "step": 7378
-    },
-    {
-      "epoch": 434.06,
-      "learning_rate": 6.941176470588236e-06,
-      "loss": 0.0179,
-      "step": 7379
-    },
-    {
-      "epoch": 434.12,
-      "learning_rate": 6.93498452012384e-06,
-      "loss": 0.0152,
-      "step": 7380
-    },
-    {
-      "epoch": 434.12,
-      "eval_accuracy_NAT": 0.9944320474354472,
-      "eval_accuracy_REST": 0.5349593495934959,
-      "eval_accuracy_concrete": 0.9600487074784172,
-      "eval_accuracy_concrete tiles": 0.9524628800377092,
-      "eval_accuracy_dark bricks": 0.8667524835863177,
-      "eval_accuracy_grass": 0.8962782969414134,
-      "eval_accuracy_red bricks": 0.9396968365553603,
-      "eval_accuracy_road": 0.9812932977269678,
-      "eval_accuracy_rough concrete": 0.6386166047944511,
-      "eval_accuracy_rough red sidewalk": 0.7583227426886953,
-      "eval_accuracy_speedway bricks": 0.9905041425812939,
-      "eval_accuracy_steel": 0.9251050792905087,
-      "eval_accuracy_tiles": 0.2580811293889545,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847768029487006,
-      "eval_iou_REST": 0.4392523364485981,
-      "eval_iou_concrete": 0.8973550969140476,
-      "eval_iou_concrete tiles": 0.8945085105912038,
-      "eval_iou_dark bricks": 0.8512047145294165,
-      "eval_iou_grass": 0.8471060854571829,
-      "eval_iou_red bricks": 0.9068744368474055,
-      "eval_iou_road": 0.9610673402488562,
-      "eval_iou_rough concrete": 0.6328133303416431,
-      "eval_iou_rough red sidewalk": 0.5992773561695123,
-      "eval_iou_speedway bricks": 0.9810164039654925,
-      "eval_iou_steel": 0.8809029777229961,
-      "eval_iou_tiles": 0.25620282096779046,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08766279369592667,
-      "eval_mean_accuracy": 0.8228118152383872,
-      "eval_mean_iou": 0.7794121702425266,
-      "eval_overall_accuracy": 0.9810450279106528,
-      "eval_runtime": 5.9706,
-      "eval_samples_per_second": 19.764,
-      "eval_steps_per_second": 1.34,
-      "step": 7380
-    },
-    {
-      "epoch": 434.18,
-      "learning_rate": 6.928792569659442e-06,
-      "loss": 0.0175,
-      "step": 7381
-    },
-    {
-      "epoch": 434.24,
-      "learning_rate": 6.9226006191950465e-06,
-      "loss": 0.0131,
-      "step": 7382
-    },
-    {
-      "epoch": 434.29,
-      "learning_rate": 6.916408668730651e-06,
-      "loss": 0.0129,
-      "step": 7383
-    },
-    {
-      "epoch": 434.35,
-      "learning_rate": 6.910216718266255e-06,
-      "loss": 0.015,
-      "step": 7384
-    },
-    {
-      "epoch": 434.41,
-      "learning_rate": 6.904024767801857e-06,
-      "loss": 0.0144,
-      "step": 7385
-    },
-    {
-      "epoch": 434.47,
-      "learning_rate": 6.8978328173374615e-06,
-      "loss": 0.0156,
-      "step": 7386
-    },
-    {
-      "epoch": 434.53,
-      "learning_rate": 6.891640866873066e-06,
-      "loss": 0.0162,
-      "step": 7387
-    },
-    {
-      "epoch": 434.59,
-      "learning_rate": 6.88544891640867e-06,
-      "loss": 0.013,
-      "step": 7388
-    },
-    {
-      "epoch": 434.65,
-      "learning_rate": 6.879256965944272e-06,
-      "loss": 0.0133,
-      "step": 7389
-    },
-    {
-      "epoch": 434.71,
-      "learning_rate": 6.8730650154798765e-06,
-      "loss": 0.0163,
-      "step": 7390
-    },
-    {
-      "epoch": 434.76,
-      "learning_rate": 6.866873065015481e-06,
-      "loss": 0.0154,
-      "step": 7391
-    },
-    {
-      "epoch": 434.82,
-      "learning_rate": 6.860681114551083e-06,
-      "loss": 0.0128,
-      "step": 7392
-    },
-    {
-      "epoch": 434.88,
-      "learning_rate": 6.854489164086687e-06,
-      "loss": 0.0149,
-      "step": 7393
-    },
-    {
-      "epoch": 434.94,
-      "learning_rate": 6.8482972136222915e-06,
-      "loss": 0.0142,
-      "step": 7394
-    },
-    {
-      "epoch": 435.0,
-      "learning_rate": 6.842105263157896e-06,
-      "loss": 0.0143,
-      "step": 7395
-    },
-    {
-      "epoch": 435.06,
-      "learning_rate": 6.835913312693498e-06,
-      "loss": 0.0138,
-      "step": 7396
-    },
-    {
-      "epoch": 435.12,
-      "learning_rate": 6.829721362229102e-06,
-      "loss": 0.0164,
-      "step": 7397
-    },
-    {
-      "epoch": 435.18,
-      "learning_rate": 6.8235294117647065e-06,
-      "loss": 0.0132,
-      "step": 7398
-    },
-    {
-      "epoch": 435.24,
-      "learning_rate": 6.817337461300311e-06,
-      "loss": 0.0143,
-      "step": 7399
-    },
-    {
-      "epoch": 435.29,
-      "learning_rate": 6.811145510835913e-06,
-      "loss": 0.018,
-      "step": 7400
-    },
-    {
-      "epoch": 435.29,
-      "eval_accuracy_NAT": 0.9940060367153096,
-      "eval_accuracy_REST": 0.4894308943089431,
-      "eval_accuracy_concrete": 0.9589678478663342,
-      "eval_accuracy_concrete tiles": 0.9510723544661795,
-      "eval_accuracy_dark bricks": 0.8596797657102387,
-      "eval_accuracy_grass": 0.9032056304337129,
-      "eval_accuracy_red bricks": 0.9357974516695958,
-      "eval_accuracy_road": 0.981497979105398,
-      "eval_accuracy_rough concrete": 0.6346480434277406,
-      "eval_accuracy_rough red sidewalk": 0.7510885177358831,
-      "eval_accuracy_speedway bricks": 0.991171952596285,
-      "eval_accuracy_steel": 0.9264594901217971,
-      "eval_accuracy_tiles": 0.24541791632585053,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848273968636938,
-      "eval_iou_REST": 0.4078590785907859,
-      "eval_iou_concrete": 0.8968112898204539,
-      "eval_iou_concrete tiles": 0.895144295823074,
-      "eval_iou_dark bricks": 0.8454587810077793,
-      "eval_iou_grass": 0.8483787593567298,
-      "eval_iou_red bricks": 0.9066191337660956,
-      "eval_iou_road": 0.9607116646187372,
-      "eval_iou_rough concrete": 0.6295722696427305,
-      "eval_iou_rough red sidewalk": 0.5961857577747269,
-      "eval_iou_speedway bricks": 0.980773769958083,
-      "eval_iou_steel": 0.8807277772102824,
-      "eval_iou_tiles": 0.24411631761337954,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08813048154115677,
-      "eval_mean_accuracy": 0.817111067729482,
-      "eval_mean_iou": 0.7751681763112732,
-      "eval_overall_accuracy": 0.9809716758081468,
-      "eval_runtime": 5.6061,
-      "eval_samples_per_second": 21.049,
-      "eval_steps_per_second": 1.427,
-      "step": 7400
-    },
-    {
-      "epoch": 435.35,
-      "learning_rate": 6.804953560371517e-06,
-      "loss": 0.017,
-      "step": 7401
-    },
-    {
-      "epoch": 435.41,
-      "learning_rate": 6.7987616099071215e-06,
-      "loss": 0.0155,
-      "step": 7402
-    },
-    {
-      "epoch": 435.47,
-      "learning_rate": 6.792569659442726e-06,
-      "loss": 0.0133,
-      "step": 7403
-    },
-    {
-      "epoch": 435.53,
-      "learning_rate": 6.786377708978328e-06,
-      "loss": 0.0159,
-      "step": 7404
-    },
-    {
-      "epoch": 435.59,
-      "learning_rate": 6.780185758513932e-06,
-      "loss": 0.0153,
-      "step": 7405
-    },
-    {
-      "epoch": 435.65,
-      "learning_rate": 6.7739938080495364e-06,
-      "loss": 0.0153,
-      "step": 7406
-    },
-    {
-      "epoch": 435.71,
-      "learning_rate": 6.767801857585139e-06,
-      "loss": 0.0155,
-      "step": 7407
-    },
-    {
-      "epoch": 435.76,
-      "learning_rate": 6.761609907120743e-06,
-      "loss": 0.0164,
-      "step": 7408
-    },
-    {
-      "epoch": 435.82,
-      "learning_rate": 6.755417956656347e-06,
-      "loss": 0.014,
-      "step": 7409
-    },
-    {
-      "epoch": 435.88,
-      "learning_rate": 6.7492260061919514e-06,
-      "loss": 0.014,
-      "step": 7410
-    },
-    {
-      "epoch": 435.94,
-      "learning_rate": 6.743034055727554e-06,
-      "loss": 0.0138,
-      "step": 7411
-    },
-    {
-      "epoch": 436.0,
-      "learning_rate": 6.736842105263158e-06,
-      "loss": 0.0173,
-      "step": 7412
-    },
-    {
-      "epoch": 436.06,
-      "learning_rate": 6.730650154798762e-06,
-      "loss": 0.0133,
-      "step": 7413
-    },
-    {
-      "epoch": 436.12,
-      "learning_rate": 6.724458204334366e-06,
-      "loss": 0.0142,
-      "step": 7414
-    },
-    {
-      "epoch": 436.18,
-      "learning_rate": 6.718266253869969e-06,
-      "loss": 0.0141,
-      "step": 7415
-    },
-    {
-      "epoch": 436.24,
-      "learning_rate": 6.712074303405573e-06,
-      "loss": 0.0133,
-      "step": 7416
-    },
-    {
-      "epoch": 436.29,
-      "learning_rate": 6.705882352941177e-06,
-      "loss": 0.0165,
-      "step": 7417
-    },
-    {
-      "epoch": 436.35,
-      "learning_rate": 6.69969040247678e-06,
-      "loss": 0.0171,
-      "step": 7418
-    },
-    {
-      "epoch": 436.41,
-      "learning_rate": 6.693498452012384e-06,
-      "loss": 0.0133,
-      "step": 7419
-    },
-    {
-      "epoch": 436.47,
-      "learning_rate": 6.687306501547988e-06,
-      "loss": 0.0148,
-      "step": 7420
-    },
-    {
-      "epoch": 436.47,
-      "eval_accuracy_NAT": 0.9942423319552949,
-      "eval_accuracy_REST": 0.5154471544715448,
-      "eval_accuracy_concrete": 0.9594322200122609,
-      "eval_accuracy_concrete tiles": 0.9505302851755834,
-      "eval_accuracy_dark bricks": 0.8614072698734758,
-      "eval_accuracy_grass": 0.8996241583304,
-      "eval_accuracy_red bricks": 0.9395869947275922,
-      "eval_accuracy_road": 0.9816492188706611,
-      "eval_accuracy_rough concrete": 0.6353452231272979,
-      "eval_accuracy_rough red sidewalk": 0.7677828830116034,
-      "eval_accuracy_speedway bricks": 0.9905968511309757,
-      "eval_accuracy_steel": 0.9274290683552655,
-      "eval_accuracy_tiles": 0.24750825532430548,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848034962013947,
-      "eval_iou_REST": 0.4421199442119944,
-      "eval_iou_concrete": 0.8971904062582985,
-      "eval_iou_concrete tiles": 0.8943166951238442,
-      "eval_iou_dark bricks": 0.8471098687523235,
-      "eval_iou_grass": 0.847449045988302,
-      "eval_iou_red bricks": 0.9093228446901244,
-      "eval_iou_road": 0.9616927683372678,
-      "eval_iou_rough concrete": 0.6302526969114822,
-      "eval_iou_rough red sidewalk": 0.6016298967614199,
-      "eval_iou_speedway bricks": 0.9809635422765213,
-      "eval_iou_steel": 0.8811994263571782,
-      "eval_iou_tiles": 0.24621782894340305,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08788444846868515,
-      "eval_mean_accuracy": 0.8208139934127895,
-      "eval_mean_iou": 0.7787898816010426,
-      "eval_overall_accuracy": 0.981034941592459,
-      "eval_runtime": 7.9971,
-      "eval_samples_per_second": 14.755,
-      "eval_steps_per_second": 1.0,
-      "step": 7420
-    },
-    {
-      "epoch": 436.53,
-      "learning_rate": 6.681114551083592e-06,
-      "loss": 0.0153,
-      "step": 7421
-    },
-    {
-      "epoch": 436.59,
-      "learning_rate": 6.674922600619195e-06,
-      "loss": 0.0156,
-      "step": 7422
-    },
-    {
-      "epoch": 436.65,
-      "learning_rate": 6.668730650154799e-06,
-      "loss": 0.0173,
-      "step": 7423
-    },
-    {
-      "epoch": 436.71,
-      "learning_rate": 6.662538699690403e-06,
-      "loss": 0.0184,
-      "step": 7424
-    },
-    {
-      "epoch": 436.76,
-      "learning_rate": 6.656346749226007e-06,
-      "loss": 0.0169,
-      "step": 7425
-    },
-    {
-      "epoch": 436.82,
-      "learning_rate": 6.65015479876161e-06,
-      "loss": 0.0159,
-      "step": 7426
-    },
-    {
-      "epoch": 436.88,
-      "learning_rate": 6.643962848297214e-06,
-      "loss": 0.016,
-      "step": 7427
-    },
-    {
-      "epoch": 436.94,
-      "learning_rate": 6.637770897832818e-06,
-      "loss": 0.0142,
-      "step": 7428
-    },
-    {
-      "epoch": 437.0,
-      "learning_rate": 6.631578947368422e-06,
-      "loss": 0.0138,
-      "step": 7429
-    },
-    {
-      "epoch": 437.06,
-      "learning_rate": 6.625386996904025e-06,
-      "loss": 0.0162,
-      "step": 7430
-    },
-    {
-      "epoch": 437.12,
-      "learning_rate": 6.619195046439629e-06,
-      "loss": 0.0141,
-      "step": 7431
-    },
-    {
-      "epoch": 437.18,
-      "learning_rate": 6.613003095975233e-06,
-      "loss": 0.0132,
-      "step": 7432
-    },
-    {
-      "epoch": 437.24,
-      "learning_rate": 6.6068111455108355e-06,
-      "loss": 0.0146,
-      "step": 7433
-    },
-    {
-      "epoch": 437.29,
-      "learning_rate": 6.60061919504644e-06,
-      "loss": 0.0133,
-      "step": 7434
-    },
-    {
-      "epoch": 437.35,
-      "learning_rate": 6.594427244582044e-06,
-      "loss": 0.0154,
-      "step": 7435
-    },
-    {
-      "epoch": 437.41,
-      "learning_rate": 6.588235294117648e-06,
-      "loss": 0.0142,
-      "step": 7436
-    },
-    {
-      "epoch": 437.47,
-      "learning_rate": 6.5820433436532505e-06,
-      "loss": 0.014,
-      "step": 7437
-    },
-    {
-      "epoch": 437.53,
-      "learning_rate": 6.575851393188855e-06,
-      "loss": 0.0154,
-      "step": 7438
-    },
-    {
-      "epoch": 437.59,
-      "learning_rate": 6.569659442724459e-06,
-      "loss": 0.0128,
-      "step": 7439
-    },
-    {
-      "epoch": 437.65,
-      "learning_rate": 6.563467492260063e-06,
-      "loss": 0.0144,
-      "step": 7440
-    },
-    {
-      "epoch": 437.65,
-      "eval_accuracy_NAT": 0.9945067152217534,
-      "eval_accuracy_REST": 0.4845528455284553,
-      "eval_accuracy_concrete": 0.9592655811187432,
-      "eval_accuracy_concrete tiles": 0.9494932830544426,
-      "eval_accuracy_dark bricks": 0.8569856248684035,
-      "eval_accuracy_grass": 0.8978897228309621,
-      "eval_accuracy_red bricks": 0.9385434973637962,
-      "eval_accuracy_road": 0.9816112753253124,
-      "eval_accuracy_rough concrete": 0.6345348262115732,
-      "eval_accuracy_rough red sidewalk": 0.7622782544612308,
-      "eval_accuracy_speedway bricks": 0.9900973216996956,
-      "eval_accuracy_steel": 0.9301603802345979,
-      "eval_accuracy_tiles": 0.2467811808900603,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847266737276271,
-      "eval_iou_REST": 0.4179523141654979,
-      "eval_iou_concrete": 0.8974188263452842,
-      "eval_iou_concrete tiles": 0.8923516512725098,
-      "eval_iou_dark bricks": 0.8430671022775419,
-      "eval_iou_grass": 0.8472235986746772,
-      "eval_iou_red bricks": 0.9073965910901078,
-      "eval_iou_road": 0.9615483676430865,
-      "eval_iou_rough concrete": 0.6298024603737875,
-      "eval_iou_rough red sidewalk": 0.6031380529077862,
-      "eval_iou_speedway bricks": 0.9807889320347147,
-      "eval_iou_steel": 0.8812001666635102,
-      "eval_iou_tiles": 0.2455463451394122,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08794984966516495,
-      "eval_mean_accuracy": 0.8174385006776175,
-      "eval_mean_iou": 0.7763200832550419,
-      "eval_overall_accuracy": 0.980997570490433,
-      "eval_runtime": 5.5075,
-      "eval_samples_per_second": 21.425,
-      "eval_steps_per_second": 1.453,
-      "step": 7440
-    },
-    {
-      "epoch": 437.71,
-      "learning_rate": 6.5572755417956655e-06,
-      "loss": 0.0155,
-      "step": 7441
-    },
-    {
-      "epoch": 437.76,
-      "learning_rate": 6.55108359133127e-06,
-      "loss": 0.0172,
-      "step": 7442
-    },
-    {
-      "epoch": 437.82,
-      "learning_rate": 6.544891640866874e-06,
-      "loss": 0.0183,
-      "step": 7443
-    },
-    {
-      "epoch": 437.88,
-      "learning_rate": 6.538699690402478e-06,
-      "loss": 0.0188,
-      "step": 7444
-    },
-    {
-      "epoch": 437.94,
-      "learning_rate": 6.5325077399380805e-06,
-      "loss": 0.014,
-      "step": 7445
-    },
-    {
-      "epoch": 438.0,
-      "learning_rate": 6.526315789473685e-06,
-      "loss": 0.0147,
-      "step": 7446
-    },
-    {
-      "epoch": 438.06,
-      "learning_rate": 6.520123839009289e-06,
-      "loss": 0.0146,
-      "step": 7447
-    },
-    {
-      "epoch": 438.12,
-      "learning_rate": 6.513931888544891e-06,
-      "loss": 0.0139,
-      "step": 7448
-    },
-    {
-      "epoch": 438.18,
-      "learning_rate": 6.5077399380804954e-06,
-      "loss": 0.0134,
-      "step": 7449
-    },
-    {
-      "epoch": 438.24,
-      "learning_rate": 6.5015479876161e-06,
-      "loss": 0.0179,
-      "step": 7450
-    },
-    {
-      "epoch": 438.29,
-      "learning_rate": 6.495356037151704e-06,
-      "loss": 0.0124,
-      "step": 7451
-    },
-    {
-      "epoch": 438.35,
-      "learning_rate": 6.489164086687306e-06,
-      "loss": 0.0159,
-      "step": 7452
-    },
-    {
-      "epoch": 438.41,
-      "learning_rate": 6.4829721362229104e-06,
-      "loss": 0.0161,
-      "step": 7453
-    },
-    {
-      "epoch": 438.47,
-      "learning_rate": 6.476780185758515e-06,
-      "loss": 0.0154,
-      "step": 7454
-    },
-    {
-      "epoch": 438.53,
-      "learning_rate": 6.470588235294119e-06,
-      "loss": 0.0159,
-      "step": 7455
-    },
-    {
-      "epoch": 438.59,
-      "learning_rate": 6.464396284829721e-06,
-      "loss": 0.0172,
-      "step": 7456
-    },
-    {
-      "epoch": 438.65,
-      "learning_rate": 6.458204334365325e-06,
-      "loss": 0.0191,
-      "step": 7457
-    },
-    {
-      "epoch": 438.71,
-      "learning_rate": 6.45201238390093e-06,
-      "loss": 0.0134,
-      "step": 7458
-    },
-    {
-      "epoch": 438.76,
-      "learning_rate": 6.445820433436532e-06,
-      "loss": 0.0142,
-      "step": 7459
-    },
-    {
-      "epoch": 438.82,
-      "learning_rate": 6.439628482972136e-06,
-      "loss": 0.0135,
-      "step": 7460
-    },
-    {
-      "epoch": 438.82,
-      "eval_accuracy_NAT": 0.9937867236794301,
-      "eval_accuracy_REST": 0.5349593495934959,
-      "eval_accuracy_concrete": 0.958222177037772,
-      "eval_accuracy_concrete tiles": 0.9475135517322649,
-      "eval_accuracy_dark bricks": 0.8619240855234194,
-      "eval_accuracy_grass": 0.9079528551476683,
-      "eval_accuracy_red bricks": 0.9383238137082601,
-      "eval_accuracy_road": 0.9822675383350051,
-      "eval_accuracy_rough concrete": 0.6337661409018049,
-      "eval_accuracy_rough red sidewalk": 0.7745433489498342,
-      "eval_accuracy_speedway bricks": 0.9907680449223476,
-      "eval_accuracy_steel": 0.928833457446011,
-      "eval_accuracy_tiles": 0.2426307976612439,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848720337093143,
-      "eval_iou_REST": 0.44519621109607577,
-      "eval_iou_concrete": 0.8974451492629474,
-      "eval_iou_concrete tiles": 0.8956889829564443,
-      "eval_iou_dark bricks": 0.8460931408010072,
-      "eval_iou_grass": 0.8482479884442732,
-      "eval_iou_red bricks": 0.9073769185830368,
-      "eval_iou_road": 0.9612909400622269,
-      "eval_iou_rough concrete": 0.6276252353048785,
-      "eval_iou_rough red sidewalk": 0.6064163629619599,
-      "eval_iou_speedway bricks": 0.9809120130787838,
-      "eval_iou_steel": 0.881334752845941,
-      "eval_iou_tiles": 0.24135852695657414,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08697782456874847,
-      "eval_mean_accuracy": 0.8227301449721967,
-      "eval_mean_iou": 0.7787583273894971,
-      "eval_overall_accuracy": 0.9810460947327695,
-      "eval_runtime": 5.5265,
-      "eval_samples_per_second": 21.351,
-      "eval_steps_per_second": 1.448,
-      "step": 7460
-    },
-    {
-      "epoch": 438.88,
-      "learning_rate": 6.43343653250774e-06,
-      "loss": 0.015,
-      "step": 7461
-    },
-    {
-      "epoch": 438.94,
-      "learning_rate": 6.427244582043345e-06,
-      "loss": 0.0148,
-      "step": 7462
-    },
-    {
-      "epoch": 439.0,
-      "learning_rate": 6.421052631578947e-06,
-      "loss": 0.015,
-      "step": 7463
-    },
-    {
-      "epoch": 439.06,
-      "learning_rate": 6.414860681114551e-06,
-      "loss": 0.0146,
-      "step": 7464
-    },
-    {
-      "epoch": 439.12,
-      "learning_rate": 6.408668730650155e-06,
-      "loss": 0.0144,
-      "step": 7465
-    },
-    {
-      "epoch": 439.18,
-      "learning_rate": 6.4024767801857596e-06,
-      "loss": 0.0148,
-      "step": 7466
-    },
-    {
-      "epoch": 439.24,
-      "learning_rate": 6.396284829721362e-06,
-      "loss": 0.0157,
-      "step": 7467
-    },
-    {
-      "epoch": 439.29,
-      "learning_rate": 6.390092879256966e-06,
-      "loss": 0.016,
-      "step": 7468
-    },
-    {
-      "epoch": 439.35,
-      "learning_rate": 6.38390092879257e-06,
-      "loss": 0.0142,
-      "step": 7469
-    },
-    {
-      "epoch": 439.41,
-      "learning_rate": 6.3777089783281746e-06,
-      "loss": 0.0129,
-      "step": 7470
-    },
-    {
-      "epoch": 439.47,
-      "learning_rate": 6.371517027863777e-06,
-      "loss": 0.0159,
-      "step": 7471
-    },
-    {
-      "epoch": 439.53,
-      "learning_rate": 6.365325077399381e-06,
-      "loss": 0.0176,
-      "step": 7472
-    },
-    {
-      "epoch": 439.59,
-      "learning_rate": 6.359133126934985e-06,
-      "loss": 0.0163,
-      "step": 7473
-    },
-    {
-      "epoch": 439.65,
-      "learning_rate": 6.352941176470588e-06,
-      "loss": 0.0144,
-      "step": 7474
-    },
-    {
-      "epoch": 439.71,
-      "learning_rate": 6.346749226006192e-06,
-      "loss": 0.0163,
-      "step": 7475
-    },
-    {
-      "epoch": 439.76,
-      "learning_rate": 6.340557275541796e-06,
-      "loss": 0.0144,
-      "step": 7476
-    },
-    {
-      "epoch": 439.82,
-      "learning_rate": 6.3343653250774e-06,
-      "loss": 0.0167,
-      "step": 7477
-    },
-    {
-      "epoch": 439.88,
-      "learning_rate": 6.328173374613003e-06,
-      "loss": 0.0122,
-      "step": 7478
-    },
-    {
-      "epoch": 439.94,
-      "learning_rate": 6.321981424148607e-06,
-      "loss": 0.0139,
-      "step": 7479
-    },
-    {
-      "epoch": 440.0,
-      "learning_rate": 6.315789473684211e-06,
-      "loss": 0.0163,
-      "step": 7480
-    },
-    {
-      "epoch": 440.0,
-      "eval_accuracy_NAT": 0.9937722214162414,
-      "eval_accuracy_REST": 0.5495934959349593,
-      "eval_accuracy_concrete": 0.958869240447693,
-      "eval_accuracy_concrete tiles": 0.9500824888050907,
-      "eval_accuracy_dark bricks": 0.8584164385659323,
-      "eval_accuracy_grass": 0.9073728175256522,
-      "eval_accuracy_red bricks": 0.9421133567662566,
-      "eval_accuracy_road": 0.9818234385295861,
-      "eval_accuracy_rough concrete": 0.6371686161876784,
-      "eval_accuracy_rough red sidewalk": 0.7663240060460674,
-      "eval_accuracy_speedway bricks": 0.9912636329550276,
-      "eval_accuracy_steel": 0.9244428673527015,
-      "eval_accuracy_tiles": 0.24344875639976976,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848395608928107,
-      "eval_iou_REST": 0.4435695538057743,
-      "eval_iou_concrete": 0.8973098444706332,
-      "eval_iou_concrete tiles": 0.8945102738206187,
-      "eval_iou_dark bricks": 0.8438192363597021,
-      "eval_iou_grass": 0.8491879494173072,
-      "eval_iou_red bricks": 0.9107512609503584,
-      "eval_iou_road": 0.9616283293675262,
-      "eval_iou_rough concrete": 0.6311139179243223,
-      "eval_iou_rough red sidewalk": 0.6054601627948428,
-      "eval_iou_speedway bricks": 0.9807336100643134,
-      "eval_iou_steel": 0.880452580872413,
-      "eval_iou_tiles": 0.24225250211021343,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08660035580396652,
-      "eval_mean_accuracy": 0.8234377982255889,
-      "eval_mean_iou": 0.7788945217577565,
-      "eval_overall_accuracy": 0.9810484870005463,
-      "eval_runtime": 5.7348,
-      "eval_samples_per_second": 20.576,
-      "eval_steps_per_second": 1.395,
-      "step": 7480
-    },
-    {
-      "epoch": 440.06,
-      "learning_rate": 6.309597523219815e-06,
-      "loss": 0.0151,
-      "step": 7481
-    },
-    {
-      "epoch": 440.12,
-      "learning_rate": 6.303405572755418e-06,
-      "loss": 0.0165,
-      "step": 7482
-    },
-    {
-      "epoch": 440.18,
-      "learning_rate": 6.297213622291022e-06,
-      "loss": 0.0149,
-      "step": 7483
-    },
-    {
-      "epoch": 440.24,
-      "learning_rate": 6.291021671826626e-06,
-      "loss": 0.0154,
-      "step": 7484
-    },
-    {
-      "epoch": 440.29,
-      "learning_rate": 6.28482972136223e-06,
-      "loss": 0.0169,
-      "step": 7485
-    },
-    {
-      "epoch": 440.35,
-      "learning_rate": 6.278637770897833e-06,
-      "loss": 0.0168,
-      "step": 7486
-    },
-    {
-      "epoch": 440.41,
-      "learning_rate": 6.272445820433437e-06,
-      "loss": 0.014,
-      "step": 7487
-    },
-    {
-      "epoch": 440.47,
-      "learning_rate": 6.266253869969041e-06,
-      "loss": 0.0131,
-      "step": 7488
-    },
-    {
-      "epoch": 440.53,
-      "learning_rate": 6.260061919504644e-06,
-      "loss": 0.013,
-      "step": 7489
-    },
-    {
-      "epoch": 440.59,
-      "learning_rate": 6.253869969040248e-06,
-      "loss": 0.0146,
-      "step": 7490
-    },
-    {
-      "epoch": 440.65,
-      "learning_rate": 6.247678018575852e-06,
-      "loss": 0.0161,
-      "step": 7491
-    },
-    {
-      "epoch": 440.71,
-      "learning_rate": 6.241486068111455e-06,
-      "loss": 0.0166,
-      "step": 7492
-    },
-    {
-      "epoch": 440.76,
-      "learning_rate": 6.2352941176470595e-06,
-      "loss": 0.0128,
-      "step": 7493
-    },
-    {
-      "epoch": 440.82,
-      "learning_rate": 6.229102167182663e-06,
-      "loss": 0.0128,
-      "step": 7494
-    },
-    {
-      "epoch": 440.88,
-      "learning_rate": 6.222910216718267e-06,
-      "loss": 0.0137,
-      "step": 7495
-    },
-    {
-      "epoch": 440.94,
-      "learning_rate": 6.21671826625387e-06,
-      "loss": 0.0158,
-      "step": 7496
-    },
-    {
-      "epoch": 441.0,
-      "learning_rate": 6.2105263157894745e-06,
-      "loss": 0.0133,
-      "step": 7497
-    },
-    {
-      "epoch": 441.06,
-      "learning_rate": 6.204334365325078e-06,
-      "loss": 0.015,
-      "step": 7498
-    },
-    {
-      "epoch": 441.12,
-      "learning_rate": 6.198142414860682e-06,
-      "loss": 0.0149,
-      "step": 7499
-    },
-    {
-      "epoch": 441.18,
-      "learning_rate": 6.191950464396285e-06,
-      "loss": 0.0163,
-      "step": 7500
-    },
-    {
-      "epoch": 441.18,
-      "eval_accuracy_NAT": 0.9941660468162168,
-      "eval_accuracy_REST": 0.5560975609756098,
-      "eval_accuracy_concrete": 0.959646633817911,
-      "eval_accuracy_concrete tiles": 0.9450624558095687,
-      "eval_accuracy_dark bricks": 0.8556696590930842,
-      "eval_accuracy_grass": 0.9017333000261158,
-      "eval_accuracy_red bricks": 0.9386533391915641,
-      "eval_accuracy_road": 0.9819666620528741,
-      "eval_accuracy_rough concrete": 0.6380803127178687,
-      "eval_accuracy_rough red sidewalk": 0.7674444837154739,
-      "eval_accuracy_speedway bricks": 0.9901267965066185,
-      "eval_accuracy_steel": 0.9297355650292123,
-      "eval_accuracy_tiles": 0.23838953012814687,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847178917531083,
-      "eval_iou_REST": 0.44940867279894875,
-      "eval_iou_concrete": 0.897590128865832,
-      "eval_iou_concrete tiles": 0.8932525450535742,
-      "eval_iou_dark bricks": 0.8424824375833361,
-      "eval_iou_grass": 0.8473211380162071,
-      "eval_iou_red bricks": 0.9087573775721806,
-      "eval_iou_road": 0.9618637613077257,
-      "eval_iou_rough concrete": 0.6324163428260947,
-      "eval_iou_rough red sidewalk": 0.6034948700511517,
-      "eval_iou_speedway bricks": 0.9808177967389236,
-      "eval_iou_steel": 0.8806288447746569,
-      "eval_iou_tiles": 0.23721813577716147,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08795193582773209,
-      "eval_mean_accuracy": 0.8228286419907895,
-      "eval_mean_iou": 0.7784592263937616,
-      "eval_overall_accuracy": 0.9809974088507184,
-      "eval_runtime": 5.4621,
-      "eval_samples_per_second": 21.603,
-      "eval_steps_per_second": 1.465,
-      "step": 7500
-    },
-    {
-      "epoch": 441.24,
-      "learning_rate": 6.185758513931889e-06,
-      "loss": 0.0142,
-      "step": 7501
-    },
-    {
-      "epoch": 441.29,
-      "learning_rate": 6.179566563467493e-06,
-      "loss": 0.0142,
-      "step": 7502
-    },
-    {
-      "epoch": 441.35,
-      "learning_rate": 6.173374613003096e-06,
-      "loss": 0.0144,
-      "step": 7503
-    },
-    {
-      "epoch": 441.41,
-      "learning_rate": 6.1671826625387e-06,
-      "loss": 0.013,
-      "step": 7504
-    },
-    {
-      "epoch": 441.47,
-      "learning_rate": 6.160990712074304e-06,
-      "loss": 0.0129,
-      "step": 7505
-    },
-    {
-      "epoch": 441.53,
-      "learning_rate": 6.154798761609908e-06,
-      "loss": 0.0156,
-      "step": 7506
-    },
-    {
-      "epoch": 441.59,
-      "learning_rate": 6.148606811145511e-06,
-      "loss": 0.0128,
-      "step": 7507
-    },
-    {
-      "epoch": 441.65,
-      "learning_rate": 6.142414860681115e-06,
-      "loss": 0.0159,
-      "step": 7508
-    },
-    {
-      "epoch": 441.71,
-      "learning_rate": 6.1362229102167186e-06,
-      "loss": 0.0148,
-      "step": 7509
-    },
-    {
-      "epoch": 441.76,
-      "learning_rate": 6.130030959752323e-06,
-      "loss": 0.0138,
-      "step": 7510
-    },
-    {
-      "epoch": 441.82,
-      "learning_rate": 6.123839009287926e-06,
-      "loss": 0.0129,
-      "step": 7511
-    },
-    {
-      "epoch": 441.88,
-      "learning_rate": 6.11764705882353e-06,
-      "loss": 0.0149,
-      "step": 7512
-    },
-    {
-      "epoch": 441.94,
-      "learning_rate": 6.1114551083591336e-06,
-      "loss": 0.0158,
-      "step": 7513
-    },
-    {
-      "epoch": 442.0,
-      "learning_rate": 6.105263157894737e-06,
-      "loss": 0.0166,
-      "step": 7514
-    },
-    {
-      "epoch": 442.06,
-      "learning_rate": 6.099071207430341e-06,
-      "loss": 0.0145,
-      "step": 7515
-    },
-    {
-      "epoch": 442.12,
-      "learning_rate": 6.092879256965944e-06,
-      "loss": 0.0152,
-      "step": 7516
-    },
-    {
-      "epoch": 442.18,
-      "learning_rate": 6.0866873065015485e-06,
-      "loss": 0.013,
-      "step": 7517
-    },
-    {
-      "epoch": 442.24,
-      "learning_rate": 6.080495356037152e-06,
-      "loss": 0.0162,
-      "step": 7518
-    },
-    {
-      "epoch": 442.29,
-      "learning_rate": 6.074303405572756e-06,
-      "loss": 0.0139,
-      "step": 7519
-    },
-    {
-      "epoch": 442.35,
-      "learning_rate": 6.068111455108359e-06,
-      "loss": 0.0161,
-      "step": 7520
-    },
-    {
-      "epoch": 442.35,
-      "eval_accuracy_NAT": 0.9941044795873637,
-      "eval_accuracy_REST": 0.5365853658536586,
-      "eval_accuracy_concrete": 0.9603873360556116,
-      "eval_accuracy_concrete tiles": 0.9515201508366722,
-      "eval_accuracy_dark bricks": 0.8547317343950386,
-      "eval_accuracy_grass": 0.9009744253558762,
-      "eval_accuracy_red bricks": 0.9428822495606327,
-      "eval_accuracy_road": 0.9819415444946855,
-      "eval_accuracy_rough concrete": 0.632687597947789,
-      "eval_accuracy_rough red sidewalk": 0.7661886463276156,
-      "eval_accuracy_speedway bricks": 0.9900351161478758,
-      "eval_accuracy_steel": 0.9293857172130123,
-      "eval_accuracy_tiles": 0.2358750643763822,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847447262184708,
-      "eval_iou_REST": 0.4230769230769231,
-      "eval_iou_concrete": 0.897021413919869,
-      "eval_iou_concrete tiles": 0.8917874182717794,
-      "eval_iou_dark bricks": 0.8416343109167672,
-      "eval_iou_grass": 0.8474632840820998,
-      "eval_iou_red bricks": 0.9081195450938905,
-      "eval_iou_road": 0.9614365271682568,
-      "eval_iou_rough concrete": 0.6275681490413032,
-      "eval_iou_rough red sidewalk": 0.6021725896725157,
-      "eval_iou_speedway bricks": 0.9807750940992648,
-      "eval_iou_steel": 0.8809814265173713,
-      "eval_iou_tiles": 0.23479388438225626,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08832405507564545,
-      "eval_mean_accuracy": 0.821330725242478,
-      "eval_mean_iou": 0.7755057917277514,
-      "eval_overall_accuracy": 0.9809411582300218,
-      "eval_runtime": 5.4171,
-      "eval_samples_per_second": 21.783,
-      "eval_steps_per_second": 1.477,
-      "step": 7520
-    },
-    {
-      "epoch": 442.41,
-      "learning_rate": 6.0619195046439635e-06,
-      "loss": 0.0153,
-      "step": 7521
-    },
-    {
-      "epoch": 442.47,
-      "learning_rate": 6.055727554179567e-06,
-      "loss": 0.0134,
-      "step": 7522
-    },
-    {
-      "epoch": 442.53,
-      "learning_rate": 6.049535603715171e-06,
-      "loss": 0.015,
-      "step": 7523
-    },
-    {
-      "epoch": 442.59,
-      "learning_rate": 6.043343653250774e-06,
-      "loss": 0.0136,
-      "step": 7524
-    },
-    {
-      "epoch": 442.65,
-      "learning_rate": 6.0371517027863785e-06,
-      "loss": 0.0135,
-      "step": 7525
-    },
-    {
-      "epoch": 442.71,
-      "learning_rate": 6.030959752321982e-06,
-      "loss": 0.0148,
-      "step": 7526
-    },
-    {
-      "epoch": 442.76,
-      "learning_rate": 6.024767801857585e-06,
-      "loss": 0.0176,
-      "step": 7527
-    },
-    {
-      "epoch": 442.82,
-      "learning_rate": 6.018575851393189e-06,
-      "loss": 0.0161,
-      "step": 7528
-    },
-    {
-      "epoch": 442.88,
-      "learning_rate": 6.012383900928793e-06,
-      "loss": 0.0134,
-      "step": 7529
-    },
-    {
-      "epoch": 442.94,
-      "learning_rate": 6.006191950464397e-06,
-      "loss": 0.0154,
-      "step": 7530
-    },
-    {
-      "epoch": 443.0,
-      "learning_rate": 6e-06,
-      "loss": 0.0139,
-      "step": 7531
-    },
-    {
-      "epoch": 443.06,
-      "learning_rate": 5.993808049535604e-06,
-      "loss": 0.0155,
-      "step": 7532
-    },
-    {
-      "epoch": 443.12,
-      "learning_rate": 5.987616099071208e-06,
-      "loss": 0.0185,
-      "step": 7533
-    },
-    {
-      "epoch": 443.18,
-      "learning_rate": 5.981424148606812e-06,
-      "loss": 0.0169,
-      "step": 7534
-    },
-    {
-      "epoch": 443.24,
-      "learning_rate": 5.975232198142415e-06,
-      "loss": 0.0127,
-      "step": 7535
-    },
-    {
-      "epoch": 443.29,
-      "learning_rate": 5.969040247678019e-06,
-      "loss": 0.0144,
-      "step": 7536
-    },
-    {
-      "epoch": 443.35,
-      "learning_rate": 5.962848297213623e-06,
-      "loss": 0.0147,
-      "step": 7537
-    },
-    {
-      "epoch": 443.41,
-      "learning_rate": 5.956656346749227e-06,
-      "loss": 0.0168,
-      "step": 7538
-    },
-    {
-      "epoch": 443.47,
-      "learning_rate": 5.95046439628483e-06,
-      "loss": 0.0142,
-      "step": 7539
-    },
-    {
-      "epoch": 443.53,
-      "learning_rate": 5.9442724458204335e-06,
-      "loss": 0.0139,
-      "step": 7540
-    },
-    {
-      "epoch": 443.53,
-      "eval_accuracy_NAT": 0.994148148112207,
-      "eval_accuracy_REST": 0.5300813008130081,
-      "eval_accuracy_concrete": 0.9591012354210078,
-      "eval_accuracy_concrete tiles": 0.945604525100165,
-      "eval_accuracy_dark bricks": 0.854281913366384,
-      "eval_accuracy_grass": 0.9029728584026918,
-      "eval_accuracy_red bricks": 0.9431568541300527,
-      "eval_accuracy_road": 0.9826512491175453,
-      "eval_accuracy_rough concrete": 0.6362509608566372,
-      "eval_accuracy_rough red sidewalk": 0.7530361936847171,
-      "eval_accuracy_speedway bricks": 0.9902587476771453,
-      "eval_accuracy_steel": 0.9309000584745636,
-      "eval_accuracy_tiles": 0.23645066497015965,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847560088282764,
-      "eval_iou_REST": 0.43007915567282323,
-      "eval_iou_concrete": 0.8972496772202794,
-      "eval_iou_concrete tiles": 0.8932077739931877,
-      "eval_iou_dark bricks": 0.8415482666616384,
-      "eval_iou_grass": 0.8486028497010775,
-      "eval_iou_red bricks": 0.9103583545377438,
-      "eval_iou_road": 0.9611974284958541,
-      "eval_iou_rough concrete": 0.6311026786769747,
-      "eval_iou_rough red sidewalk": 0.5958821779232372,
-      "eval_iou_speedway bricks": 0.9808545990700747,
-      "eval_iou_steel": 0.8813313113198843,
-      "eval_iou_tiles": 0.23550888627379982,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08803031593561172,
-      "eval_mean_accuracy": 0.8199149777020219,
-      "eval_mean_iou": 0.7762830129519116,
-      "eval_overall_accuracy": 0.9809838311146881,
-      "eval_runtime": 5.5496,
-      "eval_samples_per_second": 21.263,
-      "eval_steps_per_second": 1.442,
-      "step": 7540
-    },
-    {
-      "epoch": 443.59,
-      "learning_rate": 5.938080495356038e-06,
-      "loss": 0.0149,
-      "step": 7541
-    },
-    {
-      "epoch": 443.65,
-      "learning_rate": 5.931888544891641e-06,
-      "loss": 0.0128,
-      "step": 7542
-    },
-    {
-      "epoch": 443.71,
-      "learning_rate": 5.925696594427245e-06,
-      "loss": 0.016,
-      "step": 7543
-    },
-    {
-      "epoch": 443.76,
-      "learning_rate": 5.9195046439628484e-06,
-      "loss": 0.0137,
-      "step": 7544
-    },
-    {
-      "epoch": 443.82,
-      "learning_rate": 5.913312693498453e-06,
-      "loss": 0.0161,
-      "step": 7545
-    },
-    {
-      "epoch": 443.88,
-      "learning_rate": 5.907120743034056e-06,
-      "loss": 0.0122,
-      "step": 7546
-    },
-    {
-      "epoch": 443.94,
-      "learning_rate": 5.90092879256966e-06,
-      "loss": 0.0158,
-      "step": 7547
-    },
-    {
-      "epoch": 444.0,
-      "learning_rate": 5.8947368421052634e-06,
-      "loss": 0.0138,
-      "step": 7548
-    },
-    {
-      "epoch": 444.06,
-      "learning_rate": 5.888544891640868e-06,
-      "loss": 0.0146,
-      "step": 7549
-    },
-    {
-      "epoch": 444.12,
-      "learning_rate": 5.882352941176471e-06,
-      "loss": 0.0144,
-      "step": 7550
-    },
-    {
-      "epoch": 444.18,
-      "learning_rate": 5.876160990712075e-06,
-      "loss": 0.0156,
-      "step": 7551
-    },
-    {
-      "epoch": 444.24,
-      "learning_rate": 5.869969040247678e-06,
-      "loss": 0.0171,
-      "step": 7552
-    },
-    {
-      "epoch": 444.29,
-      "learning_rate": 5.863777089783283e-06,
-      "loss": 0.0129,
-      "step": 7553
-    },
-    {
-      "epoch": 444.35,
-      "learning_rate": 5.857585139318886e-06,
-      "loss": 0.0164,
-      "step": 7554
-    },
-    {
-      "epoch": 444.41,
-      "learning_rate": 5.851393188854489e-06,
-      "loss": 0.0143,
-      "step": 7555
-    },
-    {
-      "epoch": 444.47,
-      "learning_rate": 5.845201238390093e-06,
-      "loss": 0.015,
-      "step": 7556
-    },
-    {
-      "epoch": 444.53,
-      "learning_rate": 5.839009287925697e-06,
-      "loss": 0.0157,
-      "step": 7557
-    },
-    {
-      "epoch": 444.59,
-      "learning_rate": 5.832817337461301e-06,
-      "loss": 0.0145,
-      "step": 7558
-    },
-    {
-      "epoch": 444.65,
-      "learning_rate": 5.826625386996904e-06,
-      "loss": 0.0137,
-      "step": 7559
-    },
-    {
-      "epoch": 444.71,
-      "learning_rate": 5.820433436532508e-06,
-      "loss": 0.0131,
-      "step": 7560
-    },
-    {
-      "epoch": 444.71,
-      "eval_accuracy_NAT": 0.9942333286915309,
-      "eval_accuracy_REST": 0.5203252032520326,
-      "eval_accuracy_concrete": 0.9586846381872135,
-      "eval_accuracy_concrete tiles": 0.947183596511902,
-      "eval_accuracy_dark bricks": 0.8559328522481481,
-      "eval_accuracy_grass": 0.8992475596785854,
-      "eval_accuracy_red bricks": 0.9400263620386643,
-      "eval_accuracy_road": 0.981997658188511,
-      "eval_accuracy_rough concrete": 0.6355716575596327,
-      "eval_accuracy_rough red sidewalk": 0.7613532963851435,
-      "eval_accuracy_speedway bricks": 0.9909702558070511,
-      "eval_accuracy_steel": 0.9279838270352396,
-      "eval_accuracy_tiles": 0.2425096185888697,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847758048593774,
-      "eval_iou_REST": 0.41939711664482304,
-      "eval_iou_concrete": 0.8972176397618615,
-      "eval_iou_concrete tiles": 0.8954768270944742,
-      "eval_iou_dark bricks": 0.8410376493490006,
-      "eval_iou_grass": 0.847756992088434,
-      "eval_iou_red bricks": 0.9091681716774673,
-      "eval_iou_road": 0.9610167709786287,
-      "eval_iou_rough concrete": 0.6302537891098177,
-      "eval_iou_rough red sidewalk": 0.6025531914893617,
-      "eval_iou_speedway bricks": 0.9807160408696199,
-      "eval_iou_steel": 0.8809358883727991,
-      "eval_iou_tiles": 0.24120890710217857,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08763953298330307,
-      "eval_mean_accuracy": 0.8196938349363478,
-      "eval_mean_iou": 0.7762703684152188,
-      "eval_overall_accuracy": 0.9809791758909063,
-      "eval_runtime": 5.5276,
-      "eval_samples_per_second": 21.347,
-      "eval_steps_per_second": 1.447,
-      "step": 7560
-    },
-    {
-      "epoch": 444.76,
-      "learning_rate": 5.814241486068112e-06,
-      "loss": 0.012,
-      "step": 7561
-    },
-    {
-      "epoch": 444.82,
-      "learning_rate": 5.808049535603716e-06,
-      "loss": 0.0143,
-      "step": 7562
-    },
-    {
-      "epoch": 444.88,
-      "learning_rate": 5.801857585139319e-06,
-      "loss": 0.0162,
-      "step": 7563
-    },
-    {
-      "epoch": 444.94,
-      "learning_rate": 5.795665634674923e-06,
-      "loss": 0.0146,
-      "step": 7564
-    },
-    {
-      "epoch": 445.0,
-      "learning_rate": 5.789473684210527e-06,
-      "loss": 0.0149,
-      "step": 7565
-    },
-    {
-      "epoch": 445.06,
-      "learning_rate": 5.783281733746131e-06,
-      "loss": 0.0145,
-      "step": 7566
-    },
-    {
-      "epoch": 445.12,
-      "learning_rate": 5.777089783281734e-06,
-      "loss": 0.0135,
-      "step": 7567
-    },
-    {
-      "epoch": 445.18,
-      "learning_rate": 5.7708978328173375e-06,
-      "loss": 0.0149,
-      "step": 7568
-    },
-    {
-      "epoch": 445.24,
-      "learning_rate": 5.764705882352942e-06,
-      "loss": 0.0163,
-      "step": 7569
-    },
-    {
-      "epoch": 445.29,
-      "learning_rate": 5.758513931888545e-06,
-      "loss": 0.0141,
-      "step": 7570
-    },
-    {
-      "epoch": 445.35,
-      "learning_rate": 5.752321981424149e-06,
-      "loss": 0.0154,
-      "step": 7571
-    },
-    {
-      "epoch": 445.41,
-      "learning_rate": 5.7461300309597525e-06,
-      "loss": 0.0136,
-      "step": 7572
-    },
-    {
-      "epoch": 445.47,
-      "learning_rate": 5.739938080495357e-06,
-      "loss": 0.0162,
-      "step": 7573
-    },
-    {
-      "epoch": 445.53,
-      "learning_rate": 5.73374613003096e-06,
-      "loss": 0.0151,
-      "step": 7574
-    },
-    {
-      "epoch": 445.59,
-      "learning_rate": 5.727554179566564e-06,
-      "loss": 0.016,
-      "step": 7575
-    },
-    {
-      "epoch": 445.65,
-      "learning_rate": 5.7213622291021675e-06,
-      "loss": 0.0133,
-      "step": 7576
-    },
-    {
-      "epoch": 445.71,
-      "learning_rate": 5.715170278637772e-06,
-      "loss": 0.0149,
-      "step": 7577
-    },
-    {
-      "epoch": 445.76,
-      "learning_rate": 5.708978328173375e-06,
-      "loss": 0.0135,
-      "step": 7578
-    },
-    {
-      "epoch": 445.82,
-      "learning_rate": 5.702786377708979e-06,
-      "loss": 0.0154,
-      "step": 7579
-    },
-    {
-      "epoch": 445.88,
-      "learning_rate": 5.6965944272445825e-06,
-      "loss": 0.0159,
-      "step": 7580
-    },
-    {
-      "epoch": 445.88,
-      "eval_accuracy_NAT": 0.9939442538394203,
-      "eval_accuracy_REST": 0.5235772357723577,
-      "eval_accuracy_concrete": 0.9603758700766998,
-      "eval_accuracy_concrete tiles": 0.9463115720009427,
-      "eval_accuracy_dark bricks": 0.8545355358612637,
-      "eval_accuracy_grass": 0.904635380591733,
-      "eval_accuracy_red bricks": 0.937335237258348,
-      "eval_accuracy_road": 0.9825229892459442,
-      "eval_accuracy_rough concrete": 0.6338912757196742,
-      "eval_accuracy_rough red sidewalk": 0.7557133081163191,
-      "eval_accuracy_speedway bricks": 0.9901336511128797,
-      "eval_accuracy_steel": 0.929723070464348,
-      "eval_accuracy_tiles": 0.24253991335696326,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848389886049497,
-      "eval_iou_REST": 0.41494845360824745,
-      "eval_iou_concrete": 0.8969402623325797,
-      "eval_iou_concrete tiles": 0.8945527459062047,
-      "eval_iou_dark bricks": 0.8403719634436737,
-      "eval_iou_grass": 0.8489845067458305,
-      "eval_iou_red bricks": 0.9098517965667982,
-      "eval_iou_road": 0.9609976195877507,
-      "eval_iou_rough concrete": 0.6285354714060349,
-      "eval_iou_rough red sidewalk": 0.5992951153041083,
-      "eval_iou_speedway bricks": 0.9808312469009073,
-      "eval_iou_steel": 0.8810069571065257,
-      "eval_iou_tiles": 0.24108648518429293,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08724670857191086,
-      "eval_mean_accuracy": 0.8196337918012995,
-      "eval_mean_iou": 0.775557047130608,
-      "eval_overall_accuracy": 0.980983928098517,
-      "eval_runtime": 5.7243,
-      "eval_samples_per_second": 20.614,
-      "eval_steps_per_second": 1.398,
-      "step": 7580
-    },
-    {
-      "epoch": 445.94,
-      "learning_rate": 5.690402476780186e-06,
-      "loss": 0.0138,
-      "step": 7581
-    },
-    {
-      "epoch": 446.0,
-      "learning_rate": 5.68421052631579e-06,
-      "loss": 0.0152,
-      "step": 7582
-    },
-    {
-      "epoch": 446.06,
-      "learning_rate": 5.678018575851393e-06,
-      "loss": 0.0138,
-      "step": 7583
-    },
-    {
-      "epoch": 446.12,
-      "learning_rate": 5.6718266253869975e-06,
-      "loss": 0.0137,
-      "step": 7584
-    },
-    {
-      "epoch": 446.18,
-      "learning_rate": 5.665634674922601e-06,
-      "loss": 0.0154,
-      "step": 7585
-    },
-    {
-      "epoch": 446.24,
-      "learning_rate": 5.659442724458205e-06,
-      "loss": 0.0154,
-      "step": 7586
-    },
-    {
-      "epoch": 446.29,
-      "learning_rate": 5.653250773993808e-06,
-      "loss": 0.019,
-      "step": 7587
-    },
-    {
-      "epoch": 446.35,
-      "learning_rate": 5.6470588235294125e-06,
-      "loss": 0.0167,
-      "step": 7588
-    },
-    {
-      "epoch": 446.41,
-      "learning_rate": 5.640866873065016e-06,
-      "loss": 0.0154,
-      "step": 7589
-    },
-    {
-      "epoch": 446.47,
-      "learning_rate": 5.63467492260062e-06,
-      "loss": 0.0149,
-      "step": 7590
-    },
-    {
-      "epoch": 446.53,
-      "learning_rate": 5.628482972136223e-06,
-      "loss": 0.0141,
-      "step": 7591
-    },
-    {
-      "epoch": 446.59,
-      "learning_rate": 5.6222910216718274e-06,
-      "loss": 0.0119,
-      "step": 7592
-    },
-    {
-      "epoch": 446.65,
-      "learning_rate": 5.616099071207431e-06,
-      "loss": 0.0171,
-      "step": 7593
-    },
-    {
-      "epoch": 446.71,
-      "learning_rate": 5.609907120743034e-06,
-      "loss": 0.0126,
-      "step": 7594
-    },
-    {
-      "epoch": 446.76,
-      "learning_rate": 5.603715170278638e-06,
-      "loss": 0.0149,
-      "step": 7595
-    },
-    {
-      "epoch": 446.82,
-      "learning_rate": 5.597523219814242e-06,
-      "loss": 0.0134,
-      "step": 7596
-    },
-    {
-      "epoch": 446.88,
-      "learning_rate": 5.591331269349846e-06,
-      "loss": 0.0135,
-      "step": 7597
-    },
-    {
-      "epoch": 446.94,
-      "learning_rate": 5.585139318885449e-06,
-      "loss": 0.015,
-      "step": 7598
-    },
-    {
-      "epoch": 447.0,
-      "learning_rate": 5.578947368421053e-06,
-      "loss": 0.015,
-      "step": 7599
-    },
-    {
-      "epoch": 447.06,
-      "learning_rate": 5.5727554179566566e-06,
-      "loss": 0.0148,
-      "step": 7600
-    },
-    {
-      "epoch": 447.06,
-      "eval_accuracy_NAT": 0.9941225400266508,
-      "eval_accuracy_REST": 0.5495934959349593,
-      "eval_accuracy_concrete": 0.9594008796699021,
-      "eval_accuracy_concrete tiles": 0.9502003299552204,
-      "eval_accuracy_dark bricks": 0.8524395612809371,
-      "eval_accuracy_grass": 0.9045048011596968,
-      "eval_accuracy_red bricks": 0.9340949033391915,
-      "eval_accuracy_road": 0.9818560379136181,
-      "eval_accuracy_rough concrete": 0.6363045900642955,
-      "eval_accuracy_rough red sidewalk": 0.7584129825009964,
-      "eval_accuracy_speedway bricks": 0.9905870833170537,
-      "eval_accuracy_steel": 0.9246852619110687,
-      "eval_accuracy_tiles": 0.24705383380290225,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848459268962232,
-      "eval_iou_REST": 0.4338896020539153,
-      "eval_iou_concrete": 0.8968988844015954,
-      "eval_iou_concrete tiles": 0.8954756457810452,
-      "eval_iou_dark bricks": 0.8392586241013126,
-      "eval_iou_grass": 0.8490745864573498,
-      "eval_iou_red bricks": 0.9099566636349045,
-      "eval_iou_road": 0.9613276551035261,
-      "eval_iou_rough concrete": 0.6306303114056069,
-      "eval_iou_rough red sidewalk": 0.6004119709715252,
-      "eval_iou_speedway bricks": 0.9808143903016745,
-      "eval_iou_steel": 0.8802141810500675,
-      "eval_iou_tiles": 0.2452779114533205,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08790773898363113,
-      "eval_mean_accuracy": 0.8217889462212686,
-      "eval_mean_iou": 0.7775443348932358,
-      "eval_overall_accuracy": 0.9810032278804456,
-      "eval_runtime": 5.6739,
-      "eval_samples_per_second": 20.797,
-      "eval_steps_per_second": 1.41,
-      "step": 7600
-    },
-    {
-      "epoch": 447.12,
-      "learning_rate": 5.566563467492261e-06,
-      "loss": 0.0167,
-      "step": 7601
-    },
-    {
-      "epoch": 447.18,
-      "learning_rate": 5.560371517027864e-06,
-      "loss": 0.0141,
-      "step": 7602
-    },
-    {
-      "epoch": 447.24,
-      "learning_rate": 5.554179566563468e-06,
-      "loss": 0.0148,
-      "step": 7603
-    },
-    {
-      "epoch": 447.29,
-      "learning_rate": 5.5479876160990716e-06,
-      "loss": 0.0143,
-      "step": 7604
-    },
-    {
-      "epoch": 447.35,
-      "learning_rate": 5.541795665634676e-06,
-      "loss": 0.015,
-      "step": 7605
-    },
-    {
-      "epoch": 447.41,
-      "learning_rate": 5.535603715170279e-06,
-      "loss": 0.0132,
-      "step": 7606
-    },
-    {
-      "epoch": 447.47,
-      "learning_rate": 5.529411764705883e-06,
-      "loss": 0.0164,
-      "step": 7607
-    },
-    {
-      "epoch": 447.53,
-      "learning_rate": 5.5232198142414865e-06,
-      "loss": 0.0143,
-      "step": 7608
-    },
-    {
-      "epoch": 447.59,
-      "learning_rate": 5.51702786377709e-06,
-      "loss": 0.0172,
-      "step": 7609
-    },
-    {
-      "epoch": 447.65,
-      "learning_rate": 5.510835913312694e-06,
-      "loss": 0.0138,
-      "step": 7610
-    },
-    {
-      "epoch": 447.71,
-      "learning_rate": 5.504643962848297e-06,
-      "loss": 0.0168,
-      "step": 7611
-    },
-    {
-      "epoch": 447.76,
-      "learning_rate": 5.4984520123839015e-06,
-      "loss": 0.0134,
-      "step": 7612
-    },
-    {
-      "epoch": 447.82,
-      "learning_rate": 5.492260061919505e-06,
-      "loss": 0.0143,
-      "step": 7613
-    },
-    {
-      "epoch": 447.88,
-      "learning_rate": 5.486068111455109e-06,
-      "loss": 0.0152,
-      "step": 7614
-    },
-    {
-      "epoch": 447.94,
-      "learning_rate": 5.479876160990712e-06,
-      "loss": 0.0168,
-      "step": 7615
-    },
-    {
-      "epoch": 448.0,
-      "learning_rate": 5.4736842105263165e-06,
-      "loss": 0.0151,
-      "step": 7616
-    },
-    {
-      "epoch": 448.06,
-      "learning_rate": 5.46749226006192e-06,
-      "loss": 0.0142,
-      "step": 7617
-    },
-    {
-      "epoch": 448.12,
-      "learning_rate": 5.461300309597524e-06,
-      "loss": 0.0162,
-      "step": 7618
-    },
-    {
-      "epoch": 448.18,
-      "learning_rate": 5.455108359133127e-06,
-      "loss": 0.0127,
-      "step": 7619
-    },
-    {
-      "epoch": 448.24,
-      "learning_rate": 5.4489164086687315e-06,
-      "loss": 0.0134,
-      "step": 7620
-    },
-    {
-      "epoch": 448.24,
-      "eval_accuracy_NAT": 0.9940998970878431,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9591520679275166,
-      "eval_accuracy_concrete tiles": 0.9566580249823238,
-      "eval_accuracy_dark bricks": 0.8574832991979787,
-      "eval_accuracy_grass": 0.9036229168795679,
-      "eval_accuracy_red bricks": 0.9385434973637962,
-      "eval_accuracy_road": 0.9804499890711901,
-      "eval_accuracy_rough concrete": 0.6343560621860457,
-      "eval_accuracy_rough red sidewalk": 0.7697531189135126,
-      "eval_accuracy_speedway bricks": 0.9908598966462467,
-      "eval_accuracy_steel": 0.9285110976725125,
-      "eval_accuracy_tiles": 0.24284286103789876,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848198659916974,
-      "eval_iou_REST": 0.42346938775510207,
-      "eval_iou_concrete": 0.8967242743299652,
-      "eval_iou_concrete tiles": 0.8956531332744925,
-      "eval_iou_dark bricks": 0.843044728088788,
-      "eval_iou_grass": 0.8489812392263125,
-      "eval_iou_red bricks": 0.9080716297359052,
-      "eval_iou_road": 0.9617495419601223,
-      "eval_iou_rough concrete": 0.6283763044812769,
-      "eval_iou_rough red sidewalk": 0.6081297045526108,
-      "eval_iou_speedway bricks": 0.9808616660800088,
-      "eval_iou_steel": 0.8811751341685816,
-      "eval_iou_tiles": 0.24148942579984334,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08790744841098785,
-      "eval_mean_accuracy": 0.8227823174877243,
-      "eval_mean_iou": 0.7771189258034389,
-      "eval_overall_accuracy": 0.9810334221791412,
-      "eval_runtime": 5.5973,
-      "eval_samples_per_second": 21.082,
-      "eval_steps_per_second": 1.429,
-      "step": 7620
-    },
-    {
-      "epoch": 448.29,
-      "learning_rate": 5.442724458204335e-06,
-      "loss": 0.0142,
-      "step": 7621
-    },
-    {
-      "epoch": 448.35,
-      "learning_rate": 5.436532507739938e-06,
-      "loss": 0.0137,
-      "step": 7622
-    },
-    {
-      "epoch": 448.41,
-      "learning_rate": 5.430340557275542e-06,
-      "loss": 0.0145,
-      "step": 7623
-    },
-    {
-      "epoch": 448.47,
-      "learning_rate": 5.424148606811146e-06,
-      "loss": 0.0172,
-      "step": 7624
-    },
-    {
-      "epoch": 448.53,
-      "learning_rate": 5.41795665634675e-06,
-      "loss": 0.0159,
-      "step": 7625
-    },
-    {
-      "epoch": 448.59,
-      "learning_rate": 5.411764705882353e-06,
-      "loss": 0.0145,
-      "step": 7626
-    },
-    {
-      "epoch": 448.65,
-      "learning_rate": 5.405572755417957e-06,
-      "loss": 0.0159,
-      "step": 7627
-    },
-    {
-      "epoch": 448.71,
-      "learning_rate": 5.399380804953561e-06,
-      "loss": 0.0161,
-      "step": 7628
-    },
-    {
-      "epoch": 448.76,
-      "learning_rate": 5.393188854489165e-06,
-      "loss": 0.0151,
-      "step": 7629
-    },
-    {
-      "epoch": 448.82,
-      "learning_rate": 5.386996904024768e-06,
-      "loss": 0.0178,
-      "step": 7630
-    },
-    {
-      "epoch": 448.88,
-      "learning_rate": 5.380804953560372e-06,
-      "loss": 0.0148,
-      "step": 7631
-    },
-    {
-      "epoch": 448.94,
-      "learning_rate": 5.374613003095976e-06,
-      "loss": 0.0119,
-      "step": 7632
-    },
-    {
-      "epoch": 449.0,
-      "learning_rate": 5.36842105263158e-06,
-      "loss": 0.0142,
-      "step": 7633
-    },
-    {
-      "epoch": 449.06,
-      "learning_rate": 5.362229102167183e-06,
-      "loss": 0.0135,
-      "step": 7634
-    },
-    {
-      "epoch": 449.12,
-      "learning_rate": 5.3560371517027864e-06,
-      "loss": 0.0133,
-      "step": 7635
-    },
-    {
-      "epoch": 449.18,
-      "learning_rate": 5.349845201238391e-06,
-      "loss": 0.0125,
-      "step": 7636
-    },
-    {
-      "epoch": 449.24,
-      "learning_rate": 5.343653250773994e-06,
-      "loss": 0.0133,
-      "step": 7637
-    },
-    {
-      "epoch": 449.29,
-      "learning_rate": 5.337461300309598e-06,
-      "loss": 0.0151,
-      "step": 7638
-    },
-    {
-      "epoch": 449.35,
-      "learning_rate": 5.3312693498452014e-06,
-      "loss": 0.0132,
-      "step": 7639
-    },
-    {
-      "epoch": 449.41,
-      "learning_rate": 5.325077399380806e-06,
-      "loss": 0.0147,
-      "step": 7640
-    },
-    {
-      "epoch": 449.41,
-      "eval_accuracy_NAT": 0.9945943218302352,
-      "eval_accuracy_REST": 0.47804878048780486,
-      "eval_accuracy_concrete": 0.9583020566908573,
-      "eval_accuracy_concrete tiles": 0.9493990101343389,
-      "eval_accuracy_dark bricks": 0.8589188982255996,
-      "eval_accuracy_grass": 0.8969453875181203,
-      "eval_accuracy_red bricks": 0.9395869947275922,
-      "eval_accuracy_road": 0.9820591160436533,
-      "eval_accuracy_rough concrete": 0.6332417664269242,
-      "eval_accuracy_rough red sidewalk": 0.7670534445288354,
-      "eval_accuracy_speedway bricks": 0.9904513621130832,
-      "eval_accuracy_steel": 0.9286735270157481,
-      "eval_accuracy_tiles": 0.24463025235541822,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846997736294751,
-      "eval_iou_REST": 0.3941018766756032,
-      "eval_iou_concrete": 0.8973082927087636,
-      "eval_iou_concrete tiles": 0.8919864484843116,
-      "eval_iou_dark bricks": 0.8446111278421924,
-      "eval_iou_grass": 0.8472607306744178,
-      "eval_iou_red bricks": 0.9078751857355126,
-      "eval_iou_road": 0.9619855442777818,
-      "eval_iou_rough concrete": 0.6283443802586223,
-      "eval_iou_rough red sidewalk": 0.6066023205057299,
-      "eval_iou_speedway bricks": 0.9809939321933212,
-      "eval_iou_steel": 0.88144216196216,
-      "eval_iou_tiles": 0.24352362857746013,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08875033259391785,
-      "eval_mean_accuracy": 0.8170696090844778,
-      "eval_mean_iou": 0.7746719541173347,
-      "eval_overall_accuracy": 0.9810356204792604,
-      "eval_runtime": 5.5379,
-      "eval_samples_per_second": 21.308,
-      "eval_steps_per_second": 1.445,
-      "step": 7640
-    },
-    {
-      "epoch": 449.47,
-      "learning_rate": 5.318885448916409e-06,
-      "loss": 0.0209,
-      "step": 7641
-    },
-    {
-      "epoch": 449.53,
-      "learning_rate": 5.312693498452013e-06,
-      "loss": 0.0152,
-      "step": 7642
-    },
-    {
-      "epoch": 449.59,
-      "learning_rate": 5.306501547987616e-06,
-      "loss": 0.0121,
-      "step": 7643
-    },
-    {
-      "epoch": 449.65,
-      "learning_rate": 5.300309597523221e-06,
-      "loss": 0.0159,
-      "step": 7644
-    },
-    {
-      "epoch": 449.71,
-      "learning_rate": 5.294117647058824e-06,
-      "loss": 0.0145,
-      "step": 7645
-    },
-    {
-      "epoch": 449.76,
-      "learning_rate": 5.287925696594428e-06,
-      "loss": 0.0152,
-      "step": 7646
-    },
-    {
-      "epoch": 449.82,
-      "learning_rate": 5.281733746130031e-06,
-      "loss": 0.0166,
-      "step": 7647
-    },
-    {
-      "epoch": 449.88,
-      "learning_rate": 5.275541795665635e-06,
-      "loss": 0.0124,
-      "step": 7648
-    },
-    {
-      "epoch": 449.94,
-      "learning_rate": 5.269349845201239e-06,
-      "loss": 0.0146,
-      "step": 7649
-    },
-    {
-      "epoch": 450.0,
-      "learning_rate": 5.263157894736842e-06,
-      "loss": 0.0154,
-      "step": 7650
-    },
-    {
-      "epoch": 450.06,
-      "learning_rate": 5.256965944272446e-06,
-      "loss": 0.0177,
-      "step": 7651
-    },
-    {
-      "epoch": 450.12,
-      "learning_rate": 5.25077399380805e-06,
-      "loss": 0.013,
-      "step": 7652
-    },
-    {
-      "epoch": 450.18,
-      "learning_rate": 5.244582043343654e-06,
-      "loss": 0.0143,
-      "step": 7653
-    },
-    {
-      "epoch": 450.24,
-      "learning_rate": 5.238390092879257e-06,
-      "loss": 0.0137,
-      "step": 7654
-    },
-    {
-      "epoch": 450.29,
-      "learning_rate": 5.232198142414861e-06,
-      "loss": 0.0178,
-      "step": 7655
-    },
-    {
-      "epoch": 450.35,
-      "learning_rate": 5.226006191950465e-06,
-      "loss": 0.0123,
-      "step": 7656
-    },
-    {
-      "epoch": 450.41,
-      "learning_rate": 5.219814241486069e-06,
-      "loss": 0.013,
-      "step": 7657
-    },
-    {
-      "epoch": 450.47,
-      "learning_rate": 5.213622291021672e-06,
-      "loss": 0.0145,
-      "step": 7658
-    },
-    {
-      "epoch": 450.53,
-      "learning_rate": 5.207430340557276e-06,
-      "loss": 0.0133,
-      "step": 7659
-    },
-    {
-      "epoch": 450.59,
-      "learning_rate": 5.20123839009288e-06,
-      "loss": 0.0166,
-      "step": 7660
-    },
-    {
-      "epoch": 450.59,
-      "eval_accuracy_NAT": 0.9941980164893428,
-      "eval_accuracy_REST": 0.5252032520325203,
-      "eval_accuracy_concrete": 0.9586555910406369,
-      "eval_accuracy_concrete tiles": 0.9500824888050907,
-      "eval_accuracy_dark bricks": 0.8584116532358402,
-      "eval_accuracy_grass": 0.9036418414349355,
-      "eval_accuracy_red bricks": 0.9328317223198594,
-      "eval_accuracy_road": 0.9817400696130453,
-      "eval_accuracy_rough concrete": 0.6334264892533027,
-      "eval_accuracy_rough red sidewalk": 0.7694297595861,
-      "eval_accuracy_speedway bricks": 0.9910482269532714,
-      "eval_accuracy_steel": 0.9256123586239986,
-      "eval_accuracy_tiles": 0.24350934593595686,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848393810380541,
-      "eval_iou_REST": 0.4157014157014157,
-      "eval_iou_concrete": 0.8971448866989576,
-      "eval_iou_concrete tiles": 0.8955635038766578,
-      "eval_iou_dark bricks": 0.8440170135882862,
-      "eval_iou_grass": 0.8498383962069382,
-      "eval_iou_red bricks": 0.9076577780152835,
-      "eval_iou_road": 0.9621615931056625,
-      "eval_iou_rough concrete": 0.6286317482658088,
-      "eval_iou_rough red sidewalk": 0.6109595094075989,
-      "eval_iou_speedway bricks": 0.9808832004797857,
-      "eval_iou_steel": 0.8808913410544841,
-      "eval_iou_tiles": 0.24233471012089602,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08644358813762665,
-      "eval_mean_accuracy": 0.8205992934864539,
-      "eval_mean_iou": 0.7769711136584483,
-      "eval_overall_accuracy": 0.9811148562673795,
-      "eval_runtime": 5.4208,
-      "eval_samples_per_second": 21.768,
-      "eval_steps_per_second": 1.476,
-      "step": 7660
-    },
-    {
-      "epoch": 450.65,
-      "learning_rate": 5.195046439628484e-06,
-      "loss": 0.0151,
-      "step": 7661
-    },
-    {
-      "epoch": 450.71,
-      "learning_rate": 5.188854489164087e-06,
-      "loss": 0.0137,
-      "step": 7662
-    },
-    {
-      "epoch": 450.76,
-      "learning_rate": 5.1826625386996905e-06,
-      "loss": 0.0157,
-      "step": 7663
-    },
-    {
-      "epoch": 450.82,
-      "learning_rate": 5.176470588235295e-06,
-      "loss": 0.016,
-      "step": 7664
-    },
-    {
-      "epoch": 450.88,
-      "learning_rate": 5.170278637770898e-06,
-      "loss": 0.0149,
-      "step": 7665
-    },
-    {
-      "epoch": 450.94,
-      "learning_rate": 5.164086687306502e-06,
-      "loss": 0.0143,
-      "step": 7666
-    },
-    {
-      "epoch": 451.0,
-      "learning_rate": 5.1578947368421055e-06,
-      "loss": 0.0145,
-      "step": 7667
-    },
-    {
-      "epoch": 451.06,
-      "learning_rate": 5.15170278637771e-06,
-      "loss": 0.0134,
-      "step": 7668
-    },
-    {
-      "epoch": 451.12,
-      "learning_rate": 5.145510835913313e-06,
-      "loss": 0.0176,
-      "step": 7669
-    },
-    {
-      "epoch": 451.18,
-      "learning_rate": 5.139318885448917e-06,
-      "loss": 0.0158,
-      "step": 7670
-    },
-    {
-      "epoch": 451.24,
-      "learning_rate": 5.1331269349845205e-06,
-      "loss": 0.0133,
-      "step": 7671
-    },
-    {
-      "epoch": 451.29,
-      "learning_rate": 5.126934984520125e-06,
-      "loss": 0.0149,
-      "step": 7672
-    },
-    {
-      "epoch": 451.35,
-      "learning_rate": 5.120743034055728e-06,
-      "loss": 0.015,
-      "step": 7673
-    },
-    {
-      "epoch": 451.41,
-      "learning_rate": 5.114551083591332e-06,
-      "loss": 0.0165,
-      "step": 7674
-    },
-    {
-      "epoch": 451.47,
-      "learning_rate": 5.1083591331269355e-06,
-      "loss": 0.0145,
-      "step": 7675
-    },
-    {
-      "epoch": 451.53,
-      "learning_rate": 5.102167182662539e-06,
-      "loss": 0.0147,
-      "step": 7676
-    },
-    {
-      "epoch": 451.59,
-      "learning_rate": 5.095975232198143e-06,
-      "loss": 0.0136,
-      "step": 7677
-    },
-    {
-      "epoch": 451.65,
-      "learning_rate": 5.089783281733746e-06,
-      "loss": 0.0141,
-      "step": 7678
-    },
-    {
-      "epoch": 451.71,
-      "learning_rate": 5.0835913312693505e-06,
-      "loss": 0.0134,
-      "step": 7679
-    },
-    {
-      "epoch": 451.76,
-      "learning_rate": 5.077399380804954e-06,
-      "loss": 0.0154,
-      "step": 7680
-    },
-    {
-      "epoch": 451.76,
-      "eval_accuracy_NAT": 0.9941634051400225,
-      "eval_accuracy_REST": 0.5170731707317073,
-      "eval_accuracy_concrete": 0.9593553979535521,
-      "eval_accuracy_concrete tiles": 0.9468300730615131,
-      "eval_accuracy_dark bricks": 0.8584260092261164,
-      "eval_accuracy_grass": 0.9022016827714633,
-      "eval_accuracy_red bricks": 0.9340949033391915,
-      "eval_accuracy_road": 0.9812799373236761,
-      "eval_accuracy_rough concrete": 0.6352141295085777,
-      "eval_accuracy_rough red sidewalk": 0.7750171079644155,
-      "eval_accuracy_speedway bricks": 0.9906746509120397,
-      "eval_accuracy_steel": 0.9295481465562481,
-      "eval_accuracy_tiles": 0.2382380562876791,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847912637815804,
-      "eval_iou_REST": 0.4228723404255319,
-      "eval_iou_concrete": 0.8972824420058639,
-      "eval_iou_concrete tiles": 0.8967410714285714,
-      "eval_iou_dark bricks": 0.8438881884726116,
-      "eval_iou_grass": 0.8488616817837035,
-      "eval_iou_red bricks": 0.9070933333333333,
-      "eval_iou_road": 0.9619136726677612,
-      "eval_iou_rough concrete": 0.6298508697296276,
-      "eval_iou_rough red sidewalk": 0.6129038013226129,
-      "eval_iou_speedway bricks": 0.9808669762214488,
-      "eval_iou_steel": 0.8814130815964818,
-      "eval_iou_tiles": 0.23711029367424472,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08689098805189133,
-      "eval_mean_accuracy": 0.8201628208289388,
-      "eval_mean_iou": 0.7773530012648748,
-      "eval_overall_accuracy": 0.9810810089111328,
-      "eval_runtime": 5.564,
-      "eval_samples_per_second": 21.208,
-      "eval_steps_per_second": 1.438,
-      "step": 7680
-    },
-    {
-      "epoch": 451.82,
-      "learning_rate": 5.071207430340558e-06,
-      "loss": 0.0147,
-      "step": 7681
-    },
-    {
-      "epoch": 451.88,
-      "learning_rate": 5.065015479876161e-06,
-      "loss": 0.0116,
-      "step": 7682
-    },
-    {
-      "epoch": 451.94,
-      "learning_rate": 5.0588235294117654e-06,
-      "loss": 0.0126,
-      "step": 7683
-    },
-    {
-      "epoch": 452.0,
-      "learning_rate": 5.052631578947369e-06,
-      "loss": 0.0162,
-      "step": 7684
-    },
-    {
-      "epoch": 452.06,
-      "learning_rate": 5.046439628482973e-06,
-      "loss": 0.0172,
-      "step": 7685
-    },
-    {
-      "epoch": 452.12,
-      "learning_rate": 5.040247678018576e-06,
-      "loss": 0.0154,
-      "step": 7686
-    },
-    {
-      "epoch": 452.18,
-      "learning_rate": 5.0340557275541804e-06,
-      "loss": 0.0138,
-      "step": 7687
-    },
-    {
-      "epoch": 452.24,
-      "learning_rate": 5.027863777089784e-06,
-      "loss": 0.0165,
-      "step": 7688
-    },
-    {
-      "epoch": 452.29,
-      "learning_rate": 5.021671826625387e-06,
-      "loss": 0.0167,
-      "step": 7689
-    },
-    {
-      "epoch": 452.35,
-      "learning_rate": 5.015479876160991e-06,
-      "loss": 0.0157,
-      "step": 7690
-    },
-    {
-      "epoch": 452.41,
-      "learning_rate": 5.009287925696595e-06,
-      "loss": 0.0147,
-      "step": 7691
-    },
-    {
-      "epoch": 452.47,
-      "learning_rate": 5.003095975232199e-06,
-      "loss": 0.014,
-      "step": 7692
-    },
-    {
-      "epoch": 452.53,
-      "learning_rate": 4.996904024767802e-06,
-      "loss": 0.0129,
-      "step": 7693
-    },
-    {
-      "epoch": 452.59,
-      "learning_rate": 4.990712074303406e-06,
-      "loss": 0.0153,
-      "step": 7694
-    },
-    {
-      "epoch": 452.65,
-      "learning_rate": 4.9845201238390096e-06,
-      "loss": 0.0158,
-      "step": 7695
-    },
-    {
-      "epoch": 452.71,
-      "learning_rate": 4.978328173374614e-06,
-      "loss": 0.0155,
-      "step": 7696
-    },
-    {
-      "epoch": 452.76,
-      "learning_rate": 4.972136222910217e-06,
-      "loss": 0.0135,
-      "step": 7697
-    },
-    {
-      "epoch": 452.82,
-      "learning_rate": 4.965944272445821e-06,
-      "loss": 0.0142,
-      "step": 7698
-    },
-    {
-      "epoch": 452.88,
-      "learning_rate": 4.9597523219814246e-06,
-      "loss": 0.0143,
-      "step": 7699
-    },
-    {
-      "epoch": 452.94,
-      "learning_rate": 4.953560371517029e-06,
-      "loss": 0.0117,
-      "step": 7700
-    },
-    {
-      "epoch": 452.94,
-      "eval_accuracy_NAT": 0.9942273983980336,
-      "eval_accuracy_REST": 0.5154471544715448,
-      "eval_accuracy_concrete": 0.9585225856852604,
-      "eval_accuracy_concrete tiles": 0.9512844685364129,
-      "eval_accuracy_dark bricks": 0.8610675114369389,
-      "eval_accuracy_grass": 0.902160994977423,
-      "eval_accuracy_red bricks": 0.9394222319859402,
-      "eval_accuracy_road": 0.9818816898879383,
-      "eval_accuracy_rough concrete": 0.635982814818346,
-      "eval_accuracy_rough red sidewalk": 0.7770851036629843,
-      "eval_accuracy_speedway bricks": 0.9905810855365752,
-      "eval_accuracy_steel": 0.9303203106648608,
-      "eval_accuracy_tiles": 0.24887151988851525,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847644620987447,
-      "eval_iou_REST": 0.422103861517976,
-      "eval_iou_concrete": 0.8980881660962104,
-      "eval_iou_concrete tiles": 0.893401801722039,
-      "eval_iou_dark bricks": 0.846003394563996,
-      "eval_iou_grass": 0.8487410368141577,
-      "eval_iou_red bricks": 0.9078123341471181,
-      "eval_iou_road": 0.9621225054068065,
-      "eval_iou_rough concrete": 0.6306354214674844,
-      "eval_iou_rough red sidewalk": 0.6114918042487721,
-      "eval_iou_speedway bricks": 0.9809076469869541,
-      "eval_iou_steel": 0.8815916910770484,
-      "eval_iou_tiles": 0.2475217692608997,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08692168444395065,
-      "eval_mean_accuracy": 0.8220657592269826,
-      "eval_mean_iou": 0.7780912227237083,
-      "eval_overall_accuracy": 0.9811277551166082,
-      "eval_runtime": 5.539,
-      "eval_samples_per_second": 21.303,
-      "eval_steps_per_second": 1.444,
-      "step": 7700
-    },
-    {
-      "epoch": 453.0,
-      "learning_rate": 4.947368421052632e-06,
-      "loss": 0.0161,
-      "step": 7701
-    },
-    {
-      "epoch": 453.06,
-      "learning_rate": 4.941176470588235e-06,
-      "loss": 0.0141,
-      "step": 7702
-    },
-    {
-      "epoch": 453.12,
-      "learning_rate": 4.9349845201238395e-06,
-      "loss": 0.0122,
-      "step": 7703
-    },
-    {
-      "epoch": 453.18,
-      "learning_rate": 4.928792569659443e-06,
-      "loss": 0.0166,
-      "step": 7704
-    },
-    {
-      "epoch": 453.24,
-      "learning_rate": 4.922600619195047e-06,
-      "loss": 0.0156,
-      "step": 7705
-    },
-    {
-      "epoch": 453.29,
-      "learning_rate": 4.91640866873065e-06,
-      "loss": 0.0156,
-      "step": 7706
-    },
-    {
-      "epoch": 453.35,
-      "learning_rate": 4.9102167182662545e-06,
-      "loss": 0.0139,
-      "step": 7707
-    },
-    {
-      "epoch": 453.41,
-      "learning_rate": 4.904024767801858e-06,
-      "loss": 0.0138,
-      "step": 7708
-    },
-    {
-      "epoch": 453.47,
-      "learning_rate": 4.897832817337462e-06,
-      "loss": 0.0153,
-      "step": 7709
-    },
-    {
-      "epoch": 453.53,
-      "learning_rate": 4.891640866873065e-06,
-      "loss": 0.0149,
-      "step": 7710
-    },
-    {
-      "epoch": 453.59,
-      "learning_rate": 4.8854489164086695e-06,
-      "loss": 0.014,
-      "step": 7711
-    },
-    {
-      "epoch": 453.65,
-      "learning_rate": 4.879256965944273e-06,
-      "loss": 0.0149,
-      "step": 7712
-    },
-    {
-      "epoch": 453.71,
-      "learning_rate": 4.873065015479877e-06,
-      "loss": 0.0142,
-      "step": 7713
-    },
-    {
-      "epoch": 453.76,
-      "learning_rate": 4.86687306501548e-06,
-      "loss": 0.0161,
-      "step": 7714
-    },
-    {
-      "epoch": 453.82,
-      "learning_rate": 4.8606811145510845e-06,
-      "loss": 0.0132,
-      "step": 7715
-    },
-    {
-      "epoch": 453.88,
-      "learning_rate": 4.854489164086688e-06,
-      "loss": 0.0146,
-      "step": 7716
-    },
-    {
-      "epoch": 453.94,
-      "learning_rate": 4.848297213622291e-06,
-      "loss": 0.0153,
-      "step": 7717
-    },
-    {
-      "epoch": 454.0,
-      "learning_rate": 4.842105263157895e-06,
-      "loss": 0.0142,
-      "step": 7718
-    },
-    {
-      "epoch": 454.06,
-      "learning_rate": 4.835913312693499e-06,
-      "loss": 0.0152,
-      "step": 7719
-    },
-    {
-      "epoch": 454.12,
-      "learning_rate": 4.829721362229103e-06,
-      "loss": 0.0128,
-      "step": 7720
-    },
-    {
-      "epoch": 454.12,
-      "eval_accuracy_NAT": 0.9940269544778272,
-      "eval_accuracy_REST": 0.5430894308943089,
-      "eval_accuracy_concrete": 0.9583444808128309,
-      "eval_accuracy_concrete tiles": 0.9505774216356352,
-      "eval_accuracy_dark bricks": 0.8599238175449343,
-      "eval_accuracy_grass": 0.9036664433569134,
-      "eval_accuracy_red bricks": 0.9433216168717048,
-      "eval_accuracy_road": 0.983297892636868,
-      "eval_accuracy_rough concrete": 0.6335218300669173,
-      "eval_accuracy_rough red sidewalk": 0.7809052557170681,
-      "eval_accuracy_speedway bricks": 0.9907003556855188,
-      "eval_accuracy_steel": 0.9269792640201512,
-      "eval_accuracy_tiles": 0.24393347268926657,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848223458841231,
-      "eval_iou_REST": 0.4507422402159244,
-      "eval_iou_concrete": 0.8975960027406342,
-      "eval_iou_concrete tiles": 0.8949961167202929,
-      "eval_iou_dark bricks": 0.8451538866731884,
-      "eval_iou_grass": 0.8489157217676988,
-      "eval_iou_red bricks": 0.9089273429644917,
-      "eval_iou_road": 0.9619878576970614,
-      "eval_iou_rough concrete": 0.6280095928927534,
-      "eval_iou_rough red sidewalk": 0.6109764420700854,
-      "eval_iou_speedway bricks": 0.9809639952143164,
-      "eval_iou_steel": 0.8809540207228538,
-      "eval_iou_tiles": 0.2427055702917772,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08839111775159836,
-      "eval_mean_accuracy": 0.8240221720315343,
-      "eval_mean_iou": 0.7797500873734768,
-      "eval_overall_accuracy": 0.9811008259401484,
-      "eval_runtime": 5.3724,
-      "eval_samples_per_second": 21.964,
-      "eval_steps_per_second": 1.489,
-      "step": 7720
-    },
-    {
-      "epoch": 454.18,
-      "learning_rate": 4.823529411764706e-06,
-      "loss": 0.0146,
-      "step": 7721
-    },
-    {
-      "epoch": 454.24,
-      "learning_rate": 4.81733746130031e-06,
-      "loss": 0.0139,
-      "step": 7722
-    },
-    {
-      "epoch": 454.29,
-      "learning_rate": 4.811145510835914e-06,
-      "loss": 0.0145,
-      "step": 7723
-    },
-    {
-      "epoch": 454.35,
-      "learning_rate": 4.804953560371518e-06,
-      "loss": 0.015,
-      "step": 7724
-    },
-    {
-      "epoch": 454.41,
-      "learning_rate": 4.798761609907121e-06,
-      "loss": 0.0152,
-      "step": 7725
-    },
-    {
-      "epoch": 454.47,
-      "learning_rate": 4.792569659442725e-06,
-      "loss": 0.0147,
-      "step": 7726
-    },
-    {
-      "epoch": 454.53,
-      "learning_rate": 4.786377708978329e-06,
-      "loss": 0.0142,
-      "step": 7727
-    },
-    {
-      "epoch": 454.59,
-      "learning_rate": 4.780185758513933e-06,
-      "loss": 0.0162,
-      "step": 7728
-    },
-    {
-      "epoch": 454.65,
-      "learning_rate": 4.773993808049536e-06,
-      "loss": 0.0143,
-      "step": 7729
-    },
-    {
-      "epoch": 454.71,
-      "learning_rate": 4.7678018575851394e-06,
-      "loss": 0.0171,
-      "step": 7730
-    },
-    {
-      "epoch": 454.76,
-      "learning_rate": 4.761609907120744e-06,
-      "loss": 0.0138,
-      "step": 7731
-    },
-    {
-      "epoch": 454.82,
-      "learning_rate": 4.755417956656347e-06,
-      "loss": 0.015,
-      "step": 7732
-    },
-    {
-      "epoch": 454.88,
-      "learning_rate": 4.749226006191951e-06,
-      "loss": 0.0159,
-      "step": 7733
-    },
-    {
-      "epoch": 454.94,
-      "learning_rate": 4.7430340557275544e-06,
-      "loss": 0.0121,
-      "step": 7734
-    },
-    {
-      "epoch": 455.0,
-      "learning_rate": 4.736842105263159e-06,
-      "loss": 0.0148,
-      "step": 7735
-    },
-    {
-      "epoch": 455.06,
-      "learning_rate": 4.730650154798762e-06,
-      "loss": 0.0146,
-      "step": 7736
-    },
-    {
-      "epoch": 455.12,
-      "learning_rate": 4.724458204334366e-06,
-      "loss": 0.0145,
-      "step": 7737
-    },
-    {
-      "epoch": 455.18,
-      "learning_rate": 4.718266253869969e-06,
-      "loss": 0.0172,
-      "step": 7738
-    },
-    {
-      "epoch": 455.24,
-      "learning_rate": 4.712074303405574e-06,
-      "loss": 0.013,
-      "step": 7739
-    },
-    {
-      "epoch": 455.29,
-      "learning_rate": 4.705882352941177e-06,
-      "loss": 0.0139,
-      "step": 7740
-    },
-    {
-      "epoch": 455.29,
-      "eval_accuracy_NAT": 0.9938417675854363,
-      "eval_accuracy_REST": 0.551219512195122,
-      "eval_accuracy_concrete": 0.9599562152485289,
-      "eval_accuracy_concrete tiles": 0.9537119962290832,
-      "eval_accuracy_dark bricks": 0.8583542292747354,
-      "eval_accuracy_grass": 0.904565359736873,
-      "eval_accuracy_red bricks": 0.9439257469244289,
-      "eval_accuracy_road": 0.9814092660275406,
-      "eval_accuracy_rough concrete": 0.6308105756797502,
-      "eval_accuracy_rough red sidewalk": 0.7676400033087931,
-      "eval_accuracy_speedway bricks": 0.9906143303769417,
-      "eval_accuracy_steel": 0.930612683482685,
-      "eval_accuracy_tiles": 0.2412372383289406,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984825032310012,
-      "eval_iou_REST": 0.4442988204456094,
-      "eval_iou_concrete": 0.8964699272984127,
-      "eval_iou_concrete tiles": 0.8955626867323226,
-      "eval_iou_dark bricks": 0.8437818996053269,
-      "eval_iou_grass": 0.8490907072277117,
-      "eval_iou_red bricks": 0.9088841882601798,
-      "eval_iou_road": 0.9618147174756275,
-      "eval_iou_rough concrete": 0.625181598062954,
-      "eval_iou_rough red sidewalk": 0.6057333420363986,
-      "eval_iou_speedway bricks": 0.9807340369419103,
-      "eval_iou_steel": 0.8812844171302946,
-      "eval_iou_tiles": 0.23976995573755683,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08830472081899643,
-      "eval_mean_accuracy": 0.823684532646066,
-      "eval_mean_iou": 0.7782639484049475,
-      "eval_overall_accuracy": 0.9809929475945941,
-      "eval_runtime": 5.8582,
-      "eval_samples_per_second": 20.143,
-      "eval_steps_per_second": 1.366,
-      "step": 7740
-    },
-    {
-      "epoch": 455.35,
-      "learning_rate": 4.699690402476781e-06,
-      "loss": 0.0135,
-      "step": 7741
-    },
-    {
-      "epoch": 455.41,
-      "learning_rate": 4.693498452012384e-06,
-      "loss": 0.0132,
-      "step": 7742
-    },
-    {
-      "epoch": 455.47,
-      "learning_rate": 4.687306501547988e-06,
-      "loss": 0.0143,
-      "step": 7743
-    },
-    {
-      "epoch": 455.53,
-      "learning_rate": 4.681114551083592e-06,
-      "loss": 0.0142,
-      "step": 7744
-    },
-    {
-      "epoch": 455.59,
-      "learning_rate": 4.674922600619195e-06,
-      "loss": 0.0142,
-      "step": 7745
-    },
-    {
-      "epoch": 455.65,
-      "learning_rate": 4.668730650154799e-06,
-      "loss": 0.0166,
-      "step": 7746
-    },
-    {
-      "epoch": 455.71,
-      "learning_rate": 4.662538699690403e-06,
-      "loss": 0.0129,
-      "step": 7747
-    },
-    {
-      "epoch": 455.76,
-      "learning_rate": 4.656346749226007e-06,
-      "loss": 0.0139,
-      "step": 7748
-    },
-    {
-      "epoch": 455.82,
-      "learning_rate": 4.65015479876161e-06,
-      "loss": 0.0152,
-      "step": 7749
-    },
-    {
-      "epoch": 455.88,
-      "learning_rate": 4.643962848297214e-06,
-      "loss": 0.0163,
-      "step": 7750
-    },
-    {
-      "epoch": 455.94,
-      "learning_rate": 4.637770897832818e-06,
-      "loss": 0.0183,
-      "step": 7751
-    },
-    {
-      "epoch": 456.0,
-      "learning_rate": 4.631578947368422e-06,
-      "loss": 0.0139,
-      "step": 7752
-    },
-    {
-      "epoch": 456.06,
-      "learning_rate": 4.625386996904025e-06,
-      "loss": 0.0146,
-      "step": 7753
-    },
-    {
-      "epoch": 456.12,
-      "learning_rate": 4.619195046439629e-06,
-      "loss": 0.0141,
-      "step": 7754
-    },
-    {
-      "epoch": 456.18,
-      "learning_rate": 4.613003095975233e-06,
-      "loss": 0.0144,
-      "step": 7755
-    },
-    {
-      "epoch": 456.24,
-      "learning_rate": 4.606811145510837e-06,
-      "loss": 0.0161,
-      "step": 7756
-    },
-    {
-      "epoch": 456.29,
-      "learning_rate": 4.60061919504644e-06,
-      "loss": 0.0165,
-      "step": 7757
-    },
-    {
-      "epoch": 456.35,
-      "learning_rate": 4.5944272445820435e-06,
-      "loss": 0.0142,
-      "step": 7758
-    },
-    {
-      "epoch": 456.41,
-      "learning_rate": 4.588235294117648e-06,
-      "loss": 0.015,
-      "step": 7759
-    },
-    {
-      "epoch": 456.47,
-      "learning_rate": 4.582043343653251e-06,
-      "loss": 0.0134,
-      "step": 7760
-    },
-    {
-      "epoch": 456.47,
-      "eval_accuracy_NAT": 0.9940947215589728,
-      "eval_accuracy_REST": 0.5365853658536586,
-      "eval_accuracy_concrete": 0.9589066959788047,
-      "eval_accuracy_concrete tiles": 0.9466415272213057,
-      "eval_accuracy_dark bricks": 0.8550140688704707,
-      "eval_accuracy_grass": 0.9037137547453322,
-      "eval_accuracy_red bricks": 0.9393673110720563,
-      "eval_accuracy_road": 0.982284105235087,
-      "eval_accuracy_rough concrete": 0.6325743807316215,
-      "eval_accuracy_rough red sidewalk": 0.7642635303318569,
-      "eval_accuracy_speedway bricks": 0.9906408919762035,
-      "eval_accuracy_steel": 0.9296456041621894,
-      "eval_accuracy_tiles": 0.23705656033203065,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848524526036272,
-      "eval_iou_REST": 0.4411764705882353,
-      "eval_iou_concrete": 0.8969079793815243,
-      "eval_iou_concrete tiles": 0.895343394039366,
-      "eval_iou_dark bricks": 0.840992958542004,
-      "eval_iou_grass": 0.8495922700775963,
-      "eval_iou_red bricks": 0.907325871306562,
-      "eval_iou_road": 0.9618167261462465,
-      "eval_iou_rough concrete": 0.6270555713070598,
-      "eval_iou_rough red sidewalk": 0.6052814951133664,
-      "eval_iou_speedway bricks": 0.9806641685577212,
-      "eval_iou_steel": 0.8809064324023537,
-      "eval_iou_tiles": 0.2356430873008703,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08787562698125839,
-      "eval_mean_accuracy": 0.8208298860053531,
-      "eval_mean_iou": 0.7775045290281948,
-      "eval_overall_accuracy": 0.9810275384935282,
-      "eval_runtime": 5.4009,
-      "eval_samples_per_second": 21.848,
-      "eval_steps_per_second": 1.481,
-      "step": 7760
-    },
-    {
-      "epoch": 456.53,
-      "learning_rate": 4.575851393188855e-06,
-      "loss": 0.0148,
-      "step": 7761
-    },
-    {
-      "epoch": 456.59,
-      "learning_rate": 4.5696594427244585e-06,
-      "loss": 0.0139,
-      "step": 7762
-    },
-    {
-      "epoch": 456.65,
-      "learning_rate": 4.563467492260063e-06,
-      "loss": 0.0147,
-      "step": 7763
-    },
-    {
-      "epoch": 456.71,
-      "learning_rate": 4.557275541795666e-06,
-      "loss": 0.0158,
-      "step": 7764
-    },
-    {
-      "epoch": 456.76,
-      "learning_rate": 4.55108359133127e-06,
-      "loss": 0.0162,
-      "step": 7765
-    },
-    {
-      "epoch": 456.82,
-      "learning_rate": 4.5448916408668735e-06,
-      "loss": 0.0161,
-      "step": 7766
-    },
-    {
-      "epoch": 456.88,
-      "learning_rate": 4.538699690402478e-06,
-      "loss": 0.0132,
-      "step": 7767
-    },
-    {
-      "epoch": 456.94,
-      "learning_rate": 4.532507739938081e-06,
-      "loss": 0.0109,
-      "step": 7768
-    },
-    {
-      "epoch": 457.0,
-      "learning_rate": 4.526315789473685e-06,
-      "loss": 0.0158,
-      "step": 7769
-    },
-    {
-      "epoch": 457.06,
-      "learning_rate": 4.5201238390092885e-06,
-      "loss": 0.0129,
-      "step": 7770
-    },
-    {
-      "epoch": 457.12,
-      "learning_rate": 4.513931888544892e-06,
-      "loss": 0.0158,
-      "step": 7771
-    },
-    {
-      "epoch": 457.18,
-      "learning_rate": 4.507739938080496e-06,
-      "loss": 0.0148,
-      "step": 7772
-    },
-    {
-      "epoch": 457.24,
-      "learning_rate": 4.501547987616099e-06,
-      "loss": 0.0156,
-      "step": 7773
-    },
-    {
-      "epoch": 457.29,
-      "learning_rate": 4.4953560371517035e-06,
-      "loss": 0.0144,
-      "step": 7774
-    },
-    {
-      "epoch": 457.35,
-      "learning_rate": 4.489164086687307e-06,
-      "loss": 0.0151,
-      "step": 7775
-    },
-    {
-      "epoch": 457.41,
-      "learning_rate": 4.482972136222911e-06,
-      "loss": 0.0136,
-      "step": 7776
-    },
-    {
-      "epoch": 457.47,
-      "learning_rate": 4.476780185758514e-06,
-      "loss": 0.0141,
-      "step": 7777
-    },
-    {
-      "epoch": 457.53,
-      "learning_rate": 4.4705882352941184e-06,
-      "loss": 0.0127,
-      "step": 7778
-    },
-    {
-      "epoch": 457.59,
-      "learning_rate": 4.464396284829722e-06,
-      "loss": 0.0128,
-      "step": 7779
-    },
-    {
-      "epoch": 457.65,
-      "learning_rate": 4.458204334365326e-06,
-      "loss": 0.0115,
-      "step": 7780
-    },
-    {
-      "epoch": 457.65,
-      "eval_accuracy_NAT": 0.9941195748799021,
-      "eval_accuracy_REST": 0.5447154471544715,
-      "eval_accuracy_concrete": 0.9590798322603725,
-      "eval_accuracy_concrete tiles": 0.9501531934951685,
-      "eval_accuracy_dark bricks": 0.8578996229159888,
-      "eval_accuracy_grass": 0.9026596570113585,
-      "eval_accuracy_red bricks": 0.9396968365553603,
-      "eval_accuracy_road": 0.9817988553875292,
-      "eval_accuracy_rough concrete": 0.6308403696840048,
-      "eval_accuracy_rough red sidewalk": 0.7713849555192925,
-      "eval_accuracy_speedway bricks": 0.9908203112950887,
-      "eval_accuracy_steel": 0.9295406498173294,
-      "eval_accuracy_tiles": 0.23814717198339846,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984884123685791,
-      "eval_iou_REST": 0.45209176788124156,
-      "eval_iou_concrete": 0.8969737242354778,
-      "eval_iou_concrete tiles": 0.895451112789303,
-      "eval_iou_dark bricks": 0.8434937259164115,
-      "eval_iou_grass": 0.8495149348764228,
-      "eval_iou_red bricks": 0.9081740976645435,
-      "eval_iou_road": 0.9618373485302625,
-      "eval_iou_rough concrete": 0.6262577862961188,
-      "eval_iou_rough red sidewalk": 0.6084754006952107,
-      "eval_iou_speedway bricks": 0.9807855445712841,
-      "eval_iou_steel": 0.8814164156929836,
-      "eval_iou_tiles": 0.2368984118373866,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08860909938812256,
-      "eval_mean_accuracy": 0.8223735752276358,
-      "eval_mean_iou": 0.7789426457440337,
-      "eval_overall_accuracy": 0.9810712458723683,
-      "eval_runtime": 5.4111,
-      "eval_samples_per_second": 21.807,
-      "eval_steps_per_second": 1.478,
-      "step": 7780
-    },
-    {
-      "epoch": 457.71,
-      "learning_rate": 4.452012383900929e-06,
-      "loss": 0.0179,
-      "step": 7781
-    },
-    {
-      "epoch": 457.76,
-      "learning_rate": 4.4458204334365334e-06,
-      "loss": 0.0165,
-      "step": 7782
-    },
-    {
-      "epoch": 457.82,
-      "learning_rate": 4.439628482972137e-06,
-      "loss": 0.0133,
-      "step": 7783
-    },
-    {
-      "epoch": 457.88,
-      "learning_rate": 4.43343653250774e-06,
-      "loss": 0.0145,
-      "step": 7784
-    },
-    {
-      "epoch": 457.94,
-      "learning_rate": 4.427244582043344e-06,
-      "loss": 0.0168,
-      "step": 7785
-    },
-    {
-      "epoch": 458.0,
-      "learning_rate": 4.4210526315789476e-06,
-      "loss": 0.0149,
-      "step": 7786
-    },
-    {
-      "epoch": 458.06,
-      "learning_rate": 4.414860681114552e-06,
-      "loss": 0.0141,
-      "step": 7787
-    },
-    {
-      "epoch": 458.12,
-      "learning_rate": 4.408668730650155e-06,
-      "loss": 0.0141,
-      "step": 7788
-    },
-    {
-      "epoch": 458.18,
-      "learning_rate": 4.402476780185759e-06,
-      "loss": 0.0153,
-      "step": 7789
-    },
-    {
-      "epoch": 458.24,
-      "learning_rate": 4.3962848297213626e-06,
-      "loss": 0.0142,
-      "step": 7790
-    },
-    {
-      "epoch": 458.29,
-      "learning_rate": 4.390092879256967e-06,
-      "loss": 0.0141,
-      "step": 7791
-    },
-    {
-      "epoch": 458.35,
-      "learning_rate": 4.38390092879257e-06,
-      "loss": 0.0142,
-      "step": 7792
-    },
-    {
-      "epoch": 458.41,
-      "learning_rate": 4.377708978328174e-06,
-      "loss": 0.0148,
-      "step": 7793
-    },
-    {
-      "epoch": 458.47,
-      "learning_rate": 4.3715170278637775e-06,
-      "loss": 0.0192,
-      "step": 7794
-    },
-    {
-      "epoch": 458.53,
-      "learning_rate": 4.365325077399382e-06,
-      "loss": 0.0158,
-      "step": 7795
-    },
-    {
-      "epoch": 458.59,
-      "learning_rate": 4.359133126934985e-06,
-      "loss": 0.0144,
-      "step": 7796
-    },
-    {
-      "epoch": 458.65,
-      "learning_rate": 4.352941176470588e-06,
-      "loss": 0.0139,
-      "step": 7797
-    },
-    {
-      "epoch": 458.71,
-      "learning_rate": 4.3467492260061925e-06,
-      "loss": 0.0138,
-      "step": 7798
-    },
-    {
-      "epoch": 458.76,
-      "learning_rate": 4.340557275541796e-06,
-      "loss": 0.0149,
-      "step": 7799
-    },
-    {
-      "epoch": 458.82,
-      "learning_rate": 4.3343653250774e-06,
-      "loss": 0.0123,
-      "step": 7800
-    },
-    {
-      "epoch": 458.82,
-      "eval_accuracy_NAT": 0.9939714253659894,
-      "eval_accuracy_REST": 0.5219512195121951,
-      "eval_accuracy_concrete": 0.9594008796699021,
-      "eval_accuracy_concrete tiles": 0.9529578128682535,
-      "eval_accuracy_dark bricks": 0.8554303925884807,
-      "eval_accuracy_grass": 0.9044442425825205,
-      "eval_accuracy_red bricks": 0.9354130052724078,
-      "eval_accuracy_road": 0.9819116171913119,
-      "eval_accuracy_rough concrete": 0.629463886687443,
-      "eval_accuracy_rough red sidewalk": 0.7669556847321758,
-      "eval_accuracy_speedway bricks": 0.9907591339342081,
-      "eval_accuracy_steel": 0.9285810672357524,
-      "eval_accuracy_tiles": 0.23675361265109515,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848496551665195,
-      "eval_iou_REST": 0.44275862068965516,
-      "eval_iou_concrete": 0.8964687101578153,
-      "eval_iou_concrete tiles": 0.8941421021206961,
-      "eval_iou_dark bricks": 0.8415093984343004,
-      "eval_iou_grass": 0.8496441304863772,
-      "eval_iou_red bricks": 0.9061984570364459,
-      "eval_iou_road": 0.9615330068832918,
-      "eval_iou_rough concrete": 0.6248543390690714,
-      "eval_iou_rough red sidewalk": 0.6062401921156498,
-      "eval_iou_speedway bricks": 0.9808219842724223,
-      "eval_iou_steel": 0.8814254844241822,
-      "eval_iou_tiles": 0.23556185194116228,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08878256380558014,
-      "eval_mean_accuracy": 0.8198456907916719,
-      "eval_mean_iou": 0.7773852255998146,
-      "eval_overall_accuracy": 0.9810095964852026,
-      "eval_runtime": 5.2661,
-      "eval_samples_per_second": 22.407,
-      "eval_steps_per_second": 1.519,
-      "step": 7800
-    },
-    {
-      "epoch": 458.88,
-      "learning_rate": 4.328173374613003e-06,
-      "loss": 0.0152,
-      "step": 7801
-    },
-    {
-      "epoch": 458.94,
-      "learning_rate": 4.3219814241486075e-06,
-      "loss": 0.0138,
-      "step": 7802
-    },
-    {
-      "epoch": 459.0,
-      "learning_rate": 4.315789473684211e-06,
-      "loss": 0.0161,
-      "step": 7803
-    },
-    {
-      "epoch": 459.06,
-      "learning_rate": 4.309597523219815e-06,
-      "loss": 0.017,
-      "step": 7804
-    },
-    {
-      "epoch": 459.12,
-      "learning_rate": 4.303405572755418e-06,
-      "loss": 0.0139,
-      "step": 7805
-    },
-    {
-      "epoch": 459.18,
-      "learning_rate": 4.2972136222910225e-06,
-      "loss": 0.016,
-      "step": 7806
-    },
-    {
-      "epoch": 459.24,
-      "learning_rate": 4.291021671826626e-06,
-      "loss": 0.0133,
-      "step": 7807
-    },
-    {
-      "epoch": 459.29,
-      "learning_rate": 4.28482972136223e-06,
-      "loss": 0.0142,
-      "step": 7808
-    },
-    {
-      "epoch": 459.35,
-      "learning_rate": 4.278637770897833e-06,
-      "loss": 0.0144,
-      "step": 7809
-    },
-    {
-      "epoch": 459.41,
-      "learning_rate": 4.2724458204334375e-06,
-      "loss": 0.0148,
-      "step": 7810
-    },
-    {
-      "epoch": 459.47,
-      "learning_rate": 4.266253869969041e-06,
-      "loss": 0.0132,
-      "step": 7811
-    },
-    {
-      "epoch": 459.53,
-      "learning_rate": 4.260061919504644e-06,
-      "loss": 0.0146,
-      "step": 7812
-    },
-    {
-      "epoch": 459.59,
-      "learning_rate": 4.253869969040248e-06,
-      "loss": 0.0147,
-      "step": 7813
-    },
-    {
-      "epoch": 459.65,
-      "learning_rate": 4.247678018575852e-06,
-      "loss": 0.0152,
-      "step": 7814
-    },
-    {
-      "epoch": 459.71,
-      "learning_rate": 4.241486068111456e-06,
-      "loss": 0.0126,
-      "step": 7815
-    },
-    {
-      "epoch": 459.76,
-      "learning_rate": 4.235294117647059e-06,
-      "loss": 0.0158,
-      "step": 7816
-    },
-    {
-      "epoch": 459.82,
-      "learning_rate": 4.229102167182663e-06,
-      "loss": 0.016,
-      "step": 7817
-    },
-    {
-      "epoch": 459.88,
-      "learning_rate": 4.222910216718267e-06,
-      "loss": 0.0149,
-      "step": 7818
-    },
-    {
-      "epoch": 459.94,
-      "learning_rate": 4.216718266253871e-06,
-      "loss": 0.0161,
-      "step": 7819
-    },
-    {
-      "epoch": 460.0,
-      "learning_rate": 4.210526315789474e-06,
-      "loss": 0.0149,
-      "step": 7820
-    },
-    {
-      "epoch": 460.0,
-      "eval_accuracy_NAT": 0.9941125663512236,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9587859210009341,
-      "eval_accuracy_concrete tiles": 0.9520622201272685,
-      "eval_accuracy_dark bricks": 0.8598951055643819,
-      "eval_accuracy_grass": 0.9047309495963393,
-      "eval_accuracy_red bricks": 0.9421682776801406,
-      "eval_accuracy_road": 0.9815124083409532,
-      "eval_accuracy_rough concrete": 0.6342190097664746,
-      "eval_accuracy_rough red sidewalk": 0.7670609645131938,
-      "eval_accuracy_speedway bricks": 0.9906576857615433,
-      "eval_accuracy_steel": 0.9300054476302808,
-      "eval_accuracy_tiles": 0.2414190069375019,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984863082813559,
-      "eval_iou_REST": 0.4598337950138504,
-      "eval_iou_concrete": 0.8970762617918625,
-      "eval_iou_concrete tiles": 0.8937761355842202,
-      "eval_iou_dark bricks": 0.8448117798056445,
-      "eval_iou_grass": 0.8500698800836427,
-      "eval_iou_red bricks": 0.9084410082609616,
-      "eval_iou_road": 0.9618629044210646,
-      "eval_iou_rough concrete": 0.6289236078283067,
-      "eval_iou_rough red sidewalk": 0.6067489917555944,
-      "eval_iou_speedway bricks": 0.9808448536302021,
-      "eval_iou_steel": 0.8812661908663386,
-      "eval_iou_tiles": 0.24010967489228358,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08788131177425385,
-      "eval_mean_accuracy": 0.8228051508957092,
-      "eval_mean_iou": 0.7798944743651948,
-      "eval_overall_accuracy": 0.9810916448043565,
-      "eval_runtime": 5.4736,
-      "eval_samples_per_second": 21.558,
-      "eval_steps_per_second": 1.462,
-      "step": 7820
-    },
-    {
-      "epoch": 460.06,
-      "learning_rate": 4.204334365325078e-06,
-      "loss": 0.0139,
-      "step": 7821
-    },
-    {
-      "epoch": 460.12,
-      "learning_rate": 4.198142414860682e-06,
-      "loss": 0.0133,
-      "step": 7822
-    },
-    {
-      "epoch": 460.18,
-      "learning_rate": 4.191950464396286e-06,
-      "loss": 0.0149,
-      "step": 7823
-    },
-    {
-      "epoch": 460.24,
-      "learning_rate": 4.185758513931889e-06,
-      "loss": 0.0153,
-      "step": 7824
-    },
-    {
-      "epoch": 460.29,
-      "learning_rate": 4.1795665634674924e-06,
-      "loss": 0.0143,
-      "step": 7825
-    },
-    {
-      "epoch": 460.35,
-      "learning_rate": 4.173374613003097e-06,
-      "loss": 0.0162,
-      "step": 7826
-    },
-    {
-      "epoch": 460.41,
-      "learning_rate": 4.1671826625387e-06,
-      "loss": 0.0163,
-      "step": 7827
-    },
-    {
-      "epoch": 460.47,
-      "learning_rate": 4.160990712074304e-06,
-      "loss": 0.0185,
-      "step": 7828
-    },
-    {
-      "epoch": 460.53,
-      "learning_rate": 4.154798761609907e-06,
-      "loss": 0.0124,
-      "step": 7829
-    },
-    {
-      "epoch": 460.59,
-      "learning_rate": 4.148606811145512e-06,
-      "loss": 0.0149,
-      "step": 7830
-    },
-    {
-      "epoch": 460.65,
-      "learning_rate": 4.142414860681115e-06,
-      "loss": 0.0158,
-      "step": 7831
-    },
-    {
-      "epoch": 460.71,
-      "learning_rate": 4.136222910216719e-06,
-      "loss": 0.0154,
-      "step": 7832
-    },
-    {
-      "epoch": 460.76,
-      "learning_rate": 4.130030959752322e-06,
-      "loss": 0.0164,
-      "step": 7833
-    },
-    {
-      "epoch": 460.82,
-      "learning_rate": 4.1238390092879266e-06,
-      "loss": 0.0128,
-      "step": 7834
-    },
-    {
-      "epoch": 460.88,
-      "learning_rate": 4.11764705882353e-06,
-      "loss": 0.012,
-      "step": 7835
-    },
-    {
-      "epoch": 460.94,
-      "learning_rate": 4.111455108359133e-06,
-      "loss": 0.0136,
-      "step": 7836
-    },
-    {
-      "epoch": 461.0,
-      "learning_rate": 4.105263157894737e-06,
-      "loss": 0.014,
-      "step": 7837
-    },
-    {
-      "epoch": 461.06,
-      "learning_rate": 4.099071207430341e-06,
-      "loss": 0.0136,
-      "step": 7838
-    },
-    {
-      "epoch": 461.12,
-      "learning_rate": 4.092879256965945e-06,
-      "loss": 0.0135,
-      "step": 7839
-    },
-    {
-      "epoch": 461.18,
-      "learning_rate": 4.086687306501548e-06,
-      "loss": 0.0139,
-      "step": 7840
-    },
-    {
-      "epoch": 461.18,
-      "eval_accuracy_NAT": 0.9940619971212199,
-      "eval_accuracy_REST": 0.5203252032520326,
-      "eval_accuracy_concrete": 0.9592025182347285,
-      "eval_accuracy_concrete tiles": 0.9530756540183832,
-      "eval_accuracy_dark bricks": 0.8575455084891755,
-      "eval_accuracy_grass": 0.9046912080300673,
-      "eval_accuracy_red bricks": 0.9443101933216169,
-      "eval_accuracy_road": 0.9818720703975682,
-      "eval_accuracy_rough concrete": 0.6346420846268896,
-      "eval_accuracy_rough red sidewalk": 0.7639401710044443,
-      "eval_accuracy_speedway bricks": 0.9904479348099526,
-      "eval_accuracy_steel": 0.9328492105933919,
-      "eval_accuracy_tiles": 0.24644793844103124,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848214526062027,
-      "eval_iou_REST": 0.43478260869565216,
-      "eval_iou_concrete": 0.8974901165628705,
-      "eval_iou_concrete tiles": 0.8956390777612899,
-      "eval_iou_dark bricks": 0.8430384629859622,
-      "eval_iou_grass": 0.8496306815152366,
-      "eval_iou_red bricks": 0.9100725136293866,
-      "eval_iou_road": 0.9619055497208961,
-      "eval_iou_rough concrete": 0.629369183040331,
-      "eval_iou_rough red sidewalk": 0.607013749051429,
-      "eval_iou_speedway bricks": 0.9808546904693134,
-      "eval_iou_steel": 0.8814187625730773,
-      "eval_iou_tiles": 0.24509656232111116,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08776962012052536,
-      "eval_mean_accuracy": 0.8218008994108076,
-      "eval_mean_iou": 0.7785487239179045,
-      "eval_overall_accuracy": 0.9810948129427635,
-      "eval_runtime": 5.5333,
-      "eval_samples_per_second": 21.326,
-      "eval_steps_per_second": 1.446,
-      "step": 7840
-    },
-    {
-      "epoch": 461.24,
-      "learning_rate": 4.080495356037152e-06,
-      "loss": 0.0129,
-      "step": 7841
-    },
-    {
-      "epoch": 461.29,
-      "learning_rate": 4.074303405572756e-06,
-      "loss": 0.0136,
-      "step": 7842
-    },
-    {
-      "epoch": 461.35,
-      "learning_rate": 4.06811145510836e-06,
-      "loss": 0.0145,
-      "step": 7843
-    },
-    {
-      "epoch": 461.41,
-      "learning_rate": 4.061919504643963e-06,
-      "loss": 0.0146,
-      "step": 7844
-    },
-    {
-      "epoch": 461.47,
-      "learning_rate": 4.055727554179567e-06,
-      "loss": 0.015,
-      "step": 7845
-    },
-    {
-      "epoch": 461.53,
-      "learning_rate": 4.049535603715171e-06,
-      "loss": 0.0149,
-      "step": 7846
-    },
-    {
-      "epoch": 461.59,
-      "learning_rate": 4.043343653250775e-06,
-      "loss": 0.0191,
-      "step": 7847
-    },
-    {
-      "epoch": 461.65,
-      "learning_rate": 4.037151702786378e-06,
-      "loss": 0.013,
-      "step": 7848
-    },
-    {
-      "epoch": 461.71,
-      "learning_rate": 4.0309597523219815e-06,
-      "loss": 0.0151,
-      "step": 7849
-    },
-    {
-      "epoch": 461.76,
-      "learning_rate": 4.024767801857586e-06,
-      "loss": 0.0147,
-      "step": 7850
-    },
-    {
-      "epoch": 461.82,
-      "learning_rate": 4.018575851393189e-06,
-      "loss": 0.0141,
-      "step": 7851
-    },
-    {
-      "epoch": 461.88,
-      "learning_rate": 4.012383900928793e-06,
-      "loss": 0.0148,
-      "step": 7852
-    },
-    {
-      "epoch": 461.94,
-      "learning_rate": 4.0061919504643965e-06,
-      "loss": 0.0141,
-      "step": 7853
-    },
-    {
-      "epoch": 462.0,
-      "learning_rate": 4.000000000000001e-06,
-      "loss": 0.0155,
-      "step": 7854
-    },
-    {
-      "epoch": 462.06,
-      "learning_rate": 3.993808049535604e-06,
-      "loss": 0.0162,
-      "step": 7855
-    },
-    {
-      "epoch": 462.12,
-      "learning_rate": 3.987616099071208e-06,
-      "loss": 0.0144,
-      "step": 7856
-    },
-    {
-      "epoch": 462.18,
-      "learning_rate": 3.9814241486068115e-06,
-      "loss": 0.0169,
-      "step": 7857
-    },
-    {
-      "epoch": 462.24,
-      "learning_rate": 3.975232198142416e-06,
-      "loss": 0.0132,
-      "step": 7858
-    },
-    {
-      "epoch": 462.29,
-      "learning_rate": 3.969040247678019e-06,
-      "loss": 0.0144,
-      "step": 7859
-    },
-    {
-      "epoch": 462.35,
-      "learning_rate": 3.962848297213622e-06,
-      "loss": 0.0143,
-      "step": 7860
-    },
-    {
-      "epoch": 462.35,
-      "eval_accuracy_NAT": 0.9941663163750122,
-      "eval_accuracy_REST": 0.5317073170731708,
-      "eval_accuracy_concrete": 0.9595935081156198,
-      "eval_accuracy_concrete tiles": 0.9504595804855055,
-      "eval_accuracy_dark bricks": 0.8588710449246789,
-      "eval_accuracy_grass": 0.9033030918938559,
-      "eval_accuracy_red bricks": 0.9386533391915641,
-      "eval_accuracy_road": 0.9812815405720711,
-      "eval_accuracy_rough concrete": 0.6324134931086468,
-      "eval_accuracy_rough red sidewalk": 0.7688131208687086,
-      "eval_accuracy_speedway bricks": 0.9901365643205405,
-      "eval_accuracy_steel": 0.9312998845502206,
-      "eval_accuracy_tiles": 0.24935623617801206,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847783177914665,
-      "eval_iou_REST": 0.44794520547945205,
-      "eval_iou_concrete": 0.896973202429212,
-      "eval_iou_concrete tiles": 0.8942105146455576,
-      "eval_iou_dark bricks": 0.8451685816537955,
-      "eval_iou_grass": 0.8489902057651962,
-      "eval_iou_red bricks": 0.9062997136493796,
-      "eval_iou_road": 0.9616623441375611,
-      "eval_iou_rough concrete": 0.6275002512815359,
-      "eval_iou_rough red sidewalk": 0.6075133998074707,
-      "eval_iou_speedway bricks": 0.9809125612907585,
-      "eval_iou_steel": 0.8814741931058624,
-      "eval_iou_tiles": 0.24789182026261897,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08878564834594727,
-      "eval_mean_accuracy": 0.822311925973662,
-      "eval_mean_iou": 0.7793323316384514,
-      "eval_overall_accuracy": 0.9810427326267048,
-      "eval_runtime": 5.4673,
-      "eval_samples_per_second": 21.583,
-      "eval_steps_per_second": 1.463,
-      "step": 7860
-    },
-    {
-      "epoch": 462.41,
-      "learning_rate": 3.9566563467492265e-06,
-      "loss": 0.014,
-      "step": 7861
-    },
-    {
-      "epoch": 462.47,
-      "learning_rate": 3.95046439628483e-06,
-      "loss": 0.0164,
-      "step": 7862
-    },
-    {
-      "epoch": 462.53,
-      "learning_rate": 3.944272445820434e-06,
-      "loss": 0.0152,
-      "step": 7863
-    },
-    {
-      "epoch": 462.59,
-      "learning_rate": 3.938080495356037e-06,
-      "loss": 0.0123,
-      "step": 7864
-    },
-    {
-      "epoch": 462.65,
-      "learning_rate": 3.9318885448916415e-06,
-      "loss": 0.0132,
-      "step": 7865
-    },
-    {
-      "epoch": 462.71,
-      "learning_rate": 3.925696594427245e-06,
-      "loss": 0.0131,
-      "step": 7866
-    },
-    {
-      "epoch": 462.76,
-      "learning_rate": 3.919504643962849e-06,
-      "loss": 0.0133,
-      "step": 7867
-    },
-    {
-      "epoch": 462.82,
-      "learning_rate": 3.913312693498452e-06,
-      "loss": 0.0176,
-      "step": 7868
-    },
-    {
-      "epoch": 462.88,
-      "learning_rate": 3.9071207430340564e-06,
-      "loss": 0.0156,
-      "step": 7869
-    },
-    {
-      "epoch": 462.94,
-      "learning_rate": 3.90092879256966e-06,
-      "loss": 0.0162,
-      "step": 7870
-    },
-    {
-      "epoch": 463.0,
-      "learning_rate": 3.894736842105264e-06,
-      "loss": 0.0149,
-      "step": 7871
-    },
-    {
-      "epoch": 463.06,
-      "learning_rate": 3.888544891640867e-06,
-      "loss": 0.0141,
-      "step": 7872
-    },
-    {
-      "epoch": 463.12,
-      "learning_rate": 3.882352941176471e-06,
-      "loss": 0.0159,
-      "step": 7873
-    },
-    {
-      "epoch": 463.18,
-      "learning_rate": 3.876160990712075e-06,
-      "loss": 0.0147,
-      "step": 7874
-    },
-    {
-      "epoch": 463.24,
-      "learning_rate": 3.869969040247678e-06,
-      "loss": 0.0166,
-      "step": 7875
-    },
-    {
-      "epoch": 463.29,
-      "learning_rate": 3.863777089783282e-06,
-      "loss": 0.0164,
-      "step": 7876
-    },
-    {
-      "epoch": 463.35,
-      "learning_rate": 3.8575851393188856e-06,
-      "loss": 0.0152,
-      "step": 7877
-    },
-    {
-      "epoch": 463.41,
-      "learning_rate": 3.85139318885449e-06,
-      "loss": 0.0128,
-      "step": 7878
-    },
-    {
-      "epoch": 463.47,
-      "learning_rate": 3.845201238390093e-06,
-      "loss": 0.0139,
-      "step": 7879
-    },
-    {
-      "epoch": 463.53,
-      "learning_rate": 3.839009287925697e-06,
-      "loss": 0.0155,
-      "step": 7880
-    },
-    {
-      "epoch": 463.53,
-      "eval_accuracy_NAT": 0.9940707308261886,
-      "eval_accuracy_REST": 0.526829268292683,
-      "eval_accuracy_concrete": 0.9583609153826045,
-      "eval_accuracy_concrete tiles": 0.9567758661324535,
-      "eval_accuracy_dark bricks": 0.8593352219436097,
-      "eval_accuracy_grass": 0.9065656852392253,
-      "eval_accuracy_red bricks": 0.9435962214411248,
-      "eval_accuracy_road": 0.981732587787202,
-      "eval_accuracy_rough concrete": 0.6382650355442471,
-      "eval_accuracy_rough red sidewalk": 0.7720617541115514,
-      "eval_accuracy_speedway bricks": 0.9905389297080692,
-      "eval_accuracy_steel": 0.9314873030231849,
-      "eval_accuracy_tiles": 0.255627253173377,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848352599887175,
-      "eval_iou_REST": 0.44505494505494503,
-      "eval_iou_concrete": 0.8978720615833478,
-      "eval_iou_concrete tiles": 0.8965151715913608,
-      "eval_iou_dark bricks": 0.8451756223143646,
-      "eval_iou_grass": 0.8498821091550358,
-      "eval_iou_red bricks": 0.9083747488632758,
-      "eval_iou_road": 0.9620449187062122,
-      "eval_iou_rough concrete": 0.632707790635171,
-      "eval_iou_rough red sidewalk": 0.6103415886905965,
-      "eval_iou_speedway bricks": 0.9809599531605989,
-      "eval_iou_steel": 0.8817871496869153,
-      "eval_iou_tiles": 0.25404949719997594,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0877520740032196,
-      "eval_mean_accuracy": 0.8242497517388861,
-      "eval_mean_iou": 0.7807385243561936,
-      "eval_overall_accuracy": 0.9811652555304059,
-      "eval_runtime": 5.4061,
-      "eval_samples_per_second": 21.827,
-      "eval_steps_per_second": 1.48,
-      "step": 7880
-    },
-    {
-      "epoch": 463.59,
-      "learning_rate": 3.8328173374613006e-06,
-      "loss": 0.0161,
-      "step": 7881
-    },
-    {
-      "epoch": 463.65,
-      "learning_rate": 3.826625386996905e-06,
-      "loss": 0.0132,
-      "step": 7882
-    },
-    {
-      "epoch": 463.71,
-      "learning_rate": 3.820433436532508e-06,
-      "loss": 0.0138,
-      "step": 7883
-    },
-    {
-      "epoch": 463.76,
-      "learning_rate": 3.814241486068112e-06,
-      "loss": 0.0132,
-      "step": 7884
-    },
-    {
-      "epoch": 463.82,
-      "learning_rate": 3.808049535603715e-06,
-      "loss": 0.0157,
-      "step": 7885
-    },
-    {
-      "epoch": 463.88,
-      "learning_rate": 3.8018575851393193e-06,
-      "loss": 0.0126,
-      "step": 7886
-    },
-    {
-      "epoch": 463.94,
-      "learning_rate": 3.7956656346749226e-06,
-      "loss": 0.0121,
-      "step": 7887
-    },
-    {
-      "epoch": 464.0,
-      "learning_rate": 3.789473684210527e-06,
-      "loss": 0.0151,
-      "step": 7888
-    },
-    {
-      "epoch": 464.06,
-      "learning_rate": 3.78328173374613e-06,
-      "loss": 0.0162,
-      "step": 7889
-    },
-    {
-      "epoch": 464.12,
-      "learning_rate": 3.7770897832817343e-06,
-      "loss": 0.018,
-      "step": 7890
-    },
-    {
-      "epoch": 464.18,
-      "learning_rate": 3.7708978328173376e-06,
-      "loss": 0.0135,
-      "step": 7891
-    },
-    {
-      "epoch": 464.24,
-      "learning_rate": 3.764705882352941e-06,
-      "loss": 0.0129,
-      "step": 7892
-    },
-    {
-      "epoch": 464.29,
-      "learning_rate": 3.758513931888545e-06,
-      "loss": 0.0144,
-      "step": 7893
-    },
-    {
-      "epoch": 464.35,
-      "learning_rate": 3.7523219814241484e-06,
-      "loss": 0.0145,
-      "step": 7894
-    },
-    {
-      "epoch": 464.41,
-      "learning_rate": 3.7461300309597526e-06,
-      "loss": 0.014,
-      "step": 7895
-    },
-    {
-      "epoch": 464.47,
-      "learning_rate": 3.739938080495356e-06,
-      "loss": 0.0135,
-      "step": 7896
-    },
-    {
-      "epoch": 464.53,
-      "learning_rate": 3.73374613003096e-06,
-      "loss": 0.017,
-      "step": 7897
-    },
-    {
-      "epoch": 464.59,
-      "learning_rate": 3.7275541795665634e-06,
-      "loss": 0.0118,
-      "step": 7898
-    },
-    {
-      "epoch": 464.65,
-      "learning_rate": 3.7213622291021676e-06,
-      "loss": 0.0148,
-      "step": 7899
-    },
-    {
-      "epoch": 464.71,
-      "learning_rate": 3.715170278637771e-06,
-      "loss": 0.0152,
-      "step": 7900
-    },
-    {
-      "epoch": 464.71,
-      "eval_accuracy_NAT": 0.9942177481931609,
-      "eval_accuracy_REST": 0.5203252032520326,
-      "eval_accuracy_concrete": 0.9595461154027846,
-      "eval_accuracy_concrete tiles": 0.955338204100872,
-      "eval_accuracy_dark bricks": 0.8621394253775625,
-      "eval_accuracy_grass": 0.9024751425965247,
-      "eval_accuracy_red bricks": 0.9365663444639719,
-      "eval_accuracy_road": 0.9824556528133536,
-      "eval_accuracy_rough concrete": 0.6366859533187541,
-      "eval_accuracy_rough red sidewalk": 0.7642635303318569,
-      "eval_accuracy_speedway bricks": 0.9901192564397312,
-      "eval_accuracy_steel": 0.9319371073582992,
-      "eval_accuracy_tiles": 0.24590263261534734,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847437033581592,
-      "eval_iou_REST": 0.450070323488045,
-      "eval_iou_concrete": 0.8978308720319996,
-      "eval_iou_concrete tiles": 0.8949704142011834,
-      "eval_iou_dark bricks": 0.8486445747662451,
-      "eval_iou_grass": 0.8482225995021446,
-      "eval_iou_red bricks": 0.9102700971495676,
-      "eval_iou_road": 0.9622812862647224,
-      "eval_iou_rough concrete": 0.6313513002475818,
-      "eval_iou_rough red sidewalk": 0.6088824184909594,
-      "eval_iou_speedway bricks": 0.9812049164236509,
-      "eval_iou_steel": 0.8817457258844363,
-      "eval_iou_tiles": 0.24462794973026733,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08762776851654053,
-      "eval_mean_accuracy": 0.8216901781741731,
-      "eval_mean_iou": 0.7803727831953048,
-      "eval_overall_accuracy": 0.9811445333189883,
-      "eval_runtime": 5.4595,
-      "eval_samples_per_second": 21.614,
-      "eval_steps_per_second": 1.465,
-      "step": 7900
-    },
-    {
-      "epoch": 464.76,
-      "learning_rate": 3.708978328173375e-06,
-      "loss": 0.0171,
-      "step": 7901
-    },
-    {
-      "epoch": 464.82,
-      "learning_rate": 3.7027863777089784e-06,
-      "loss": 0.0155,
-      "step": 7902
-    },
-    {
-      "epoch": 464.88,
-      "learning_rate": 3.6965944272445826e-06,
-      "loss": 0.0141,
-      "step": 7903
-    },
-    {
-      "epoch": 464.94,
-      "learning_rate": 3.690402476780186e-06,
-      "loss": 0.0144,
-      "step": 7904
-    },
-    {
-      "epoch": 465.0,
-      "learning_rate": 3.6842105263157892e-06,
-      "loss": 0.0131,
-      "step": 7905
-    },
-    {
-      "epoch": 465.06,
-      "learning_rate": 3.6780185758513934e-06,
-      "loss": 0.0144,
-      "step": 7906
-    },
-    {
-      "epoch": 465.12,
-      "learning_rate": 3.6718266253869967e-06,
-      "loss": 0.0157,
-      "step": 7907
-    },
-    {
-      "epoch": 465.18,
-      "learning_rate": 3.665634674922601e-06,
-      "loss": 0.0139,
-      "step": 7908
-    },
-    {
-      "epoch": 465.24,
-      "learning_rate": 3.659442724458204e-06,
-      "loss": 0.0163,
-      "step": 7909
-    },
-    {
-      "epoch": 465.29,
-      "learning_rate": 3.6532507739938084e-06,
-      "loss": 0.0162,
-      "step": 7910
-    },
-    {
-      "epoch": 465.35,
-      "learning_rate": 3.6470588235294117e-06,
-      "loss": 0.0144,
-      "step": 7911
-    },
-    {
-      "epoch": 465.41,
-      "learning_rate": 3.640866873065016e-06,
-      "loss": 0.0148,
-      "step": 7912
-    },
-    {
-      "epoch": 465.47,
-      "learning_rate": 3.634674922600619e-06,
-      "loss": 0.0146,
-      "step": 7913
-    },
-    {
-      "epoch": 465.53,
-      "learning_rate": 3.6284829721362234e-06,
-      "loss": 0.0148,
-      "step": 7914
-    },
-    {
-      "epoch": 465.59,
-      "learning_rate": 3.6222910216718267e-06,
-      "loss": 0.0151,
-      "step": 7915
-    },
-    {
-      "epoch": 465.65,
-      "learning_rate": 3.616099071207431e-06,
-      "loss": 0.0131,
-      "step": 7916
-    },
-    {
-      "epoch": 465.71,
-      "learning_rate": 3.609907120743034e-06,
-      "loss": 0.0131,
-      "step": 7917
-    },
-    {
-      "epoch": 465.76,
-      "learning_rate": 3.6037151702786384e-06,
-      "loss": 0.0154,
-      "step": 7918
-    },
-    {
-      "epoch": 465.82,
-      "learning_rate": 3.5975232198142417e-06,
-      "loss": 0.0131,
-      "step": 7919
-    },
-    {
-      "epoch": 465.88,
-      "learning_rate": 3.591331269349845e-06,
-      "loss": 0.0137,
-      "step": 7920
-    },
-    {
-      "epoch": 465.88,
-      "eval_accuracy_NAT": 0.994165399875108,
-      "eval_accuracy_REST": 0.5284552845528455,
-      "eval_accuracy_concrete": 0.9587461722740399,
-      "eval_accuracy_concrete tiles": 0.9500824888050907,
-      "eval_accuracy_dark bricks": 0.8634123231820531,
-      "eval_accuracy_grass": 0.9029104073699789,
-      "eval_accuracy_red bricks": 0.9413444639718805,
-      "eval_accuracy_road": 0.9827062939791076,
-      "eval_accuracy_rough concrete": 0.6363165076659973,
-      "eval_accuracy_rough red sidewalk": 0.7727611126568856,
-      "eval_accuracy_speedway bricks": 0.9906266686682117,
-      "eval_accuracy_steel": 0.9292057954789666,
-      "eval_accuracy_tiles": 0.2493259414099185,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847658929280532,
-      "eval_iou_REST": 0.4421768707482993,
-      "eval_iou_concrete": 0.8980776790374863,
-      "eval_iou_concrete tiles": 0.8958222222222222,
-      "eval_iou_dark bricks": 0.8485585288999671,
-      "eval_iou_grass": 0.8488143813140411,
-      "eval_iou_red bricks": 0.9109752856763221,
-      "eval_iou_road": 0.9623882481135245,
-      "eval_iou_rough concrete": 0.6312460468058191,
-      "eval_iou_rough red sidewalk": 0.611134238884792,
-      "eval_iou_speedway bricks": 0.981145916074326,
-      "eval_iou_steel": 0.8817988655119425,
-      "eval_iou_tiles": 0.24798867026245217,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08796266466379166,
-      "eval_mean_accuracy": 0.8230814507607758,
-      "eval_mean_iou": 0.7803763728060958,
-      "eval_overall_accuracy": 0.9811784453311209,
-      "eval_runtime": 5.5953,
-      "eval_samples_per_second": 21.089,
-      "eval_steps_per_second": 1.43,
-      "step": 7920
-    },
-    {
-      "epoch": 465.94,
-      "learning_rate": 3.585139318885449e-06,
-      "loss": 0.0146,
-      "step": 7921
-    },
-    {
-      "epoch": 466.0,
-      "learning_rate": 3.5789473684210525e-06,
-      "loss": 0.0151,
-      "step": 7922
-    },
-    {
-      "epoch": 466.06,
-      "learning_rate": 3.5727554179566567e-06,
-      "loss": 0.014,
-      "step": 7923
-    },
-    {
-      "epoch": 466.12,
-      "learning_rate": 3.56656346749226e-06,
-      "loss": 0.0149,
-      "step": 7924
-    },
-    {
-      "epoch": 466.18,
-      "learning_rate": 3.560371517027864e-06,
-      "loss": 0.014,
-      "step": 7925
-    },
-    {
-      "epoch": 466.24,
-      "learning_rate": 3.5541795665634675e-06,
-      "loss": 0.0128,
-      "step": 7926
-    },
-    {
-      "epoch": 466.29,
-      "learning_rate": 3.5479876160990717e-06,
-      "loss": 0.0167,
-      "step": 7927
-    },
-    {
-      "epoch": 466.35,
-      "learning_rate": 3.541795665634675e-06,
-      "loss": 0.0146,
-      "step": 7928
-    },
-    {
-      "epoch": 466.41,
-      "learning_rate": 3.535603715170279e-06,
-      "loss": 0.014,
-      "step": 7929
-    },
-    {
-      "epoch": 466.47,
-      "learning_rate": 3.5294117647058825e-06,
-      "loss": 0.0142,
-      "step": 7930
-    },
-    {
-      "epoch": 466.53,
-      "learning_rate": 3.5232198142414866e-06,
-      "loss": 0.0169,
-      "step": 7931
-    },
-    {
-      "epoch": 466.59,
-      "learning_rate": 3.51702786377709e-06,
-      "loss": 0.013,
-      "step": 7932
-    },
-    {
-      "epoch": 466.65,
-      "learning_rate": 3.5108359133126933e-06,
-      "loss": 0.0133,
-      "step": 7933
-    },
-    {
-      "epoch": 466.71,
-      "learning_rate": 3.5046439628482975e-06,
-      "loss": 0.0141,
-      "step": 7934
-    },
-    {
-      "epoch": 466.76,
-      "learning_rate": 3.4984520123839008e-06,
-      "loss": 0.0166,
-      "step": 7935
-    },
-    {
-      "epoch": 466.82,
-      "learning_rate": 3.492260061919505e-06,
-      "loss": 0.0139,
-      "step": 7936
-    },
-    {
-      "epoch": 466.88,
-      "learning_rate": 3.4860681114551083e-06,
-      "loss": 0.016,
-      "step": 7937
-    },
-    {
-      "epoch": 466.94,
-      "learning_rate": 3.4798761609907124e-06,
-      "loss": 0.0168,
-      "step": 7938
-    },
-    {
-      "epoch": 467.0,
-      "learning_rate": 3.4736842105263158e-06,
-      "loss": 0.0119,
-      "step": 7939
-    },
-    {
-      "epoch": 467.06,
-      "learning_rate": 3.46749226006192e-06,
-      "loss": 0.0164,
-      "step": 7940
-    },
-    {
-      "epoch": 467.06,
-      "eval_accuracy_NAT": 0.9942077745177337,
-      "eval_accuracy_REST": 0.5235772357723577,
-      "eval_accuracy_concrete": 0.9598572256305906,
-      "eval_accuracy_concrete tiles": 0.9508131039358945,
-      "eval_accuracy_dark bricks": 0.8595074938269242,
-      "eval_accuracy_grass": 0.9015355384225248,
-      "eval_accuracy_red bricks": 0.9410149384885764,
-      "eval_accuracy_road": 0.9821136264890837,
-      "eval_accuracy_rough concrete": 0.6310727629171905,
-      "eval_accuracy_rough red sidewalk": 0.7624060941953241,
-      "eval_accuracy_speedway bricks": 0.9901900302493775,
-      "eval_accuracy_steel": 0.9317796758410092,
-      "eval_accuracy_tiles": 0.25020448968463144,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847703185684099,
-      "eval_iou_REST": 0.42933333333333334,
-      "eval_iou_concrete": 0.8972091328377031,
-      "eval_iou_concrete tiles": 0.8948010468881693,
-      "eval_iou_dark bricks": 0.8456236493834833,
-      "eval_iou_grass": 0.8484313110876813,
-      "eval_iou_red bricks": 0.9096894080169896,
-      "eval_iou_road": 0.9619674317638973,
-      "eval_iou_rough concrete": 0.626221772834513,
-      "eval_iou_rough red sidewalk": 0.6046422863140819,
-      "eval_iou_speedway bricks": 0.9810901148427623,
-      "eval_iou_steel": 0.8818511414725978,
-      "eval_iou_tiles": 0.24872760126486976,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08872515708208084,
-      "eval_mean_accuracy": 0.8214061530747091,
-      "eval_mean_iou": 0.7780275806621917,
-      "eval_overall_accuracy": 0.9810682393736758,
-      "eval_runtime": 5.9227,
-      "eval_samples_per_second": 19.923,
-      "eval_steps_per_second": 1.351,
-      "step": 7940
-    },
-    {
-      "epoch": 467.12,
-      "learning_rate": 3.4613003095975233e-06,
-      "loss": 0.0139,
-      "step": 7941
-    },
-    {
-      "epoch": 467.18,
-      "learning_rate": 3.4551083591331274e-06,
-      "loss": 0.0158,
-      "step": 7942
-    },
-    {
-      "epoch": 467.24,
-      "learning_rate": 3.4489164086687308e-06,
-      "loss": 0.0144,
-      "step": 7943
-    },
-    {
-      "epoch": 467.29,
-      "learning_rate": 3.442724458204335e-06,
-      "loss": 0.0132,
-      "step": 7944
-    },
-    {
-      "epoch": 467.35,
-      "learning_rate": 3.4365325077399382e-06,
-      "loss": 0.0143,
-      "step": 7945
-    },
-    {
-      "epoch": 467.41,
-      "learning_rate": 3.4303405572755416e-06,
-      "loss": 0.0127,
-      "step": 7946
-    },
-    {
-      "epoch": 467.47,
-      "learning_rate": 3.4241486068111457e-06,
-      "loss": 0.0151,
-      "step": 7947
-    },
-    {
-      "epoch": 467.53,
-      "learning_rate": 3.417956656346749e-06,
-      "loss": 0.0142,
-      "step": 7948
-    },
-    {
-      "epoch": 467.59,
-      "learning_rate": 3.4117647058823532e-06,
-      "loss": 0.0149,
-      "step": 7949
-    },
-    {
-      "epoch": 467.65,
-      "learning_rate": 3.4055727554179566e-06,
-      "loss": 0.0121,
-      "step": 7950
-    },
-    {
-      "epoch": 467.71,
-      "learning_rate": 3.3993808049535607e-06,
-      "loss": 0.015,
-      "step": 7951
-    },
-    {
-      "epoch": 467.76,
-      "learning_rate": 3.393188854489164e-06,
-      "loss": 0.0147,
-      "step": 7952
-    },
-    {
-      "epoch": 467.82,
-      "learning_rate": 3.3869969040247682e-06,
-      "loss": 0.0145,
-      "step": 7953
-    },
-    {
-      "epoch": 467.88,
-      "learning_rate": 3.3808049535603715e-06,
-      "loss": 0.0113,
-      "step": 7954
-    },
-    {
-      "epoch": 467.94,
-      "learning_rate": 3.3746130030959757e-06,
-      "loss": 0.0153,
-      "step": 7955
-    },
-    {
-      "epoch": 468.0,
-      "learning_rate": 3.368421052631579e-06,
-      "loss": 0.0161,
-      "step": 7956
-    },
-    {
-      "epoch": 468.06,
-      "learning_rate": 3.362229102167183e-06,
-      "loss": 0.014,
-      "step": 7957
-    },
-    {
-      "epoch": 468.12,
-      "learning_rate": 3.3560371517027865e-06,
-      "loss": 0.0161,
-      "step": 7958
-    },
-    {
-      "epoch": 468.18,
-      "learning_rate": 3.34984520123839e-06,
-      "loss": 0.0157,
-      "step": 7959
-    },
-    {
-      "epoch": 468.24,
-      "learning_rate": 3.343653250773994e-06,
-      "loss": 0.0142,
-      "step": 7960
-    },
-    {
-      "epoch": 468.24,
-      "eval_accuracy_NAT": 0.9941275538202439,
-      "eval_accuracy_REST": 0.4943089430894309,
-      "eval_accuracy_concrete": 0.959490314305414,
-      "eval_accuracy_concrete tiles": 0.9503888757954277,
-      "eval_accuracy_dark bricks": 0.8614024845433838,
-      "eval_accuracy_grass": 0.901463625112128,
-      "eval_accuracy_red bricks": 0.9389828646748682,
-      "eval_accuracy_road": 0.9816118097414441,
-      "eval_accuracy_rough concrete": 0.6286713661742711,
-      "eval_accuracy_rough red sidewalk": 0.7749719880582648,
-      "eval_accuracy_speedway bricks": 0.9905905106201842,
-      "eval_accuracy_steel": 0.9294506889503066,
-      "eval_accuracy_tiles": 0.24611469599200217,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846980106324454,
-      "eval_iou_REST": 0.41643835616438357,
-      "eval_iou_concrete": 0.8974182528969675,
-      "eval_iou_concrete tiles": 0.8934110244594116,
-      "eval_iou_dark bricks": 0.8472367343480839,
-      "eval_iou_grass": 0.8471649037627883,
-      "eval_iou_red bricks": 0.9106258322237017,
-      "eval_iou_road": 0.9622944481350235,
-      "eval_iou_rough concrete": 0.6238535907518552,
-      "eval_iou_rough red sidewalk": 0.6097856830096685,
-      "eval_iou_speedway bricks": 0.9811006125829248,
-      "eval_iou_steel": 0.8820396315724571,
-      "eval_iou_tiles": 0.24489057695785857,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08869675546884537,
-      "eval_mean_accuracy": 0.8193519785290284,
-      "eval_mean_iou": 0.7769967428844282,
-      "eval_overall_accuracy": 0.9810487779520326,
-      "eval_runtime": 5.8041,
-      "eval_samples_per_second": 20.331,
-      "eval_steps_per_second": 1.378,
-      "step": 7960
-    },
-    {
-      "epoch": 468.29,
-      "learning_rate": 3.3374613003095974e-06,
-      "loss": 0.0129,
-      "step": 7961
-    },
-    {
-      "epoch": 468.35,
-      "learning_rate": 3.3312693498452015e-06,
-      "loss": 0.0149,
-      "step": 7962
-    },
-    {
-      "epoch": 468.41,
-      "learning_rate": 3.325077399380805e-06,
-      "loss": 0.0145,
-      "step": 7963
-    },
-    {
-      "epoch": 468.47,
-      "learning_rate": 3.318885448916409e-06,
-      "loss": 0.0138,
-      "step": 7964
-    },
-    {
-      "epoch": 468.53,
-      "learning_rate": 3.3126934984520123e-06,
-      "loss": 0.0177,
-      "step": 7965
-    },
-    {
-      "epoch": 468.59,
-      "learning_rate": 3.3065015479876165e-06,
-      "loss": 0.0148,
-      "step": 7966
-    },
-    {
-      "epoch": 468.65,
-      "learning_rate": 3.30030959752322e-06,
-      "loss": 0.0151,
-      "step": 7967
-    },
-    {
-      "epoch": 468.71,
-      "learning_rate": 3.294117647058824e-06,
-      "loss": 0.0139,
-      "step": 7968
-    },
-    {
-      "epoch": 468.76,
-      "learning_rate": 3.2879256965944273e-06,
-      "loss": 0.0182,
-      "step": 7969
-    },
-    {
-      "epoch": 468.82,
-      "learning_rate": 3.2817337461300315e-06,
-      "loss": 0.0141,
-      "step": 7970
-    },
-    {
-      "epoch": 468.88,
-      "learning_rate": 3.275541795665635e-06,
-      "loss": 0.0129,
-      "step": 7971
-    },
-    {
-      "epoch": 468.94,
-      "learning_rate": 3.269349845201239e-06,
-      "loss": 0.0139,
-      "step": 7972
-    },
-    {
-      "epoch": 469.0,
-      "learning_rate": 3.2631578947368423e-06,
-      "loss": 0.0141,
-      "step": 7973
-    },
-    {
-      "epoch": 469.06,
-      "learning_rate": 3.2569659442724456e-06,
-      "loss": 0.0149,
-      "step": 7974
-    },
-    {
-      "epoch": 469.12,
-      "learning_rate": 3.25077399380805e-06,
-      "loss": 0.0121,
-      "step": 7975
-    },
-    {
-      "epoch": 469.18,
-      "learning_rate": 3.244582043343653e-06,
-      "loss": 0.0154,
-      "step": 7976
-    },
-    {
-      "epoch": 469.24,
-      "learning_rate": 3.2383900928792573e-06,
-      "loss": 0.0144,
-      "step": 7977
-    },
-    {
-      "epoch": 469.29,
-      "learning_rate": 3.2321981424148606e-06,
-      "loss": 0.0135,
-      "step": 7978
-    },
-    {
-      "epoch": 469.35,
-      "learning_rate": 3.226006191950465e-06,
-      "loss": 0.0135,
-      "step": 7979
-    },
-    {
-      "epoch": 469.41,
-      "learning_rate": 3.219814241486068e-06,
-      "loss": 0.0152,
-      "step": 7980
-    },
-    {
-      "epoch": 469.41,
-      "eval_accuracy_NAT": 0.994294087243998,
-      "eval_accuracy_REST": 0.5138211382113821,
-      "eval_accuracy_concrete": 0.9590649264877872,
-      "eval_accuracy_concrete tiles": 0.9521329248173462,
-      "eval_accuracy_dark bricks": 0.8576364297609249,
-      "eval_accuracy_grass": 0.9029198696476626,
-      "eval_accuracy_red bricks": 0.9437060632688928,
-      "eval_accuracy_road": 0.9813718568983236,
-      "eval_accuracy_rough concrete": 0.6300180551665783,
-      "eval_accuracy_rough red sidewalk": 0.7729415922814881,
-      "eval_accuracy_speedway bricks": 0.9903394606658702,
-      "eval_accuracy_steel": 0.9306426704383594,
-      "eval_accuracy_tiles": 0.2470235390348087,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847307657314595,
-      "eval_iou_REST": 0.42473118279569894,
-      "eval_iou_concrete": 0.8978163301523331,
-      "eval_iou_concrete tiles": 0.8944164009918526,
-      "eval_iou_dark bricks": 0.8437470576050318,
-      "eval_iou_grass": 0.8485356965645291,
-      "eval_iou_red bricks": 0.9109367544929227,
-      "eval_iou_road": 0.9623813953488372,
-      "eval_iou_rough concrete": 0.6248980460300011,
-      "eval_iou_rough red sidewalk": 0.6100687911396537,
-      "eval_iou_speedway bricks": 0.9810086443786415,
-      "eval_iou_steel": 0.8822397892587337,
-      "eval_iou_tiles": 0.2456690066584315,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08870723843574524,
-      "eval_mean_accuracy": 0.8212240472248785,
-      "eval_mean_iou": 0.7777830662421634,
-      "eval_overall_accuracy": 0.9810956211413367,
-      "eval_runtime": 5.6587,
-      "eval_samples_per_second": 20.853,
-      "eval_steps_per_second": 1.414,
-      "step": 7980
-    },
-    {
-      "epoch": 469.47,
-      "learning_rate": 3.2136222910216723e-06,
-      "loss": 0.0133,
-      "step": 7981
-    },
-    {
-      "epoch": 469.53,
-      "learning_rate": 3.2074303405572756e-06,
-      "loss": 0.016,
-      "step": 7982
-    },
-    {
-      "epoch": 469.59,
-      "learning_rate": 3.2012383900928798e-06,
-      "loss": 0.0172,
-      "step": 7983
-    },
-    {
-      "epoch": 469.65,
-      "learning_rate": 3.195046439628483e-06,
-      "loss": 0.0133,
-      "step": 7984
-    },
-    {
-      "epoch": 469.71,
-      "learning_rate": 3.1888544891640873e-06,
-      "loss": 0.0146,
-      "step": 7985
-    },
-    {
-      "epoch": 469.76,
-      "learning_rate": 3.1826625386996906e-06,
-      "loss": 0.0131,
-      "step": 7986
-    },
-    {
-      "epoch": 469.82,
-      "learning_rate": 3.176470588235294e-06,
-      "loss": 0.017,
-      "step": 7987
-    },
-    {
-      "epoch": 469.88,
-      "learning_rate": 3.170278637770898e-06,
-      "loss": 0.0137,
-      "step": 7988
-    },
-    {
-      "epoch": 469.94,
-      "learning_rate": 3.1640866873065014e-06,
-      "loss": 0.0152,
-      "step": 7989
-    },
-    {
-      "epoch": 470.0,
-      "learning_rate": 3.1578947368421056e-06,
-      "loss": 0.015,
-      "step": 7990
-    },
-    {
-      "epoch": 470.06,
-      "learning_rate": 3.151702786377709e-06,
-      "loss": 0.012,
-      "step": 7991
-    },
-    {
-      "epoch": 470.12,
-      "learning_rate": 3.145510835913313e-06,
-      "loss": 0.0158,
-      "step": 7992
-    },
-    {
-      "epoch": 470.18,
-      "learning_rate": 3.1393188854489164e-06,
-      "loss": 0.016,
-      "step": 7993
-    },
-    {
-      "epoch": 470.24,
-      "learning_rate": 3.1331269349845206e-06,
-      "loss": 0.016,
-      "step": 7994
-    },
-    {
-      "epoch": 470.29,
-      "learning_rate": 3.126934984520124e-06,
-      "loss": 0.0143,
-      "step": 7995
-    },
-    {
-      "epoch": 470.35,
-      "learning_rate": 3.1207430340557276e-06,
-      "loss": 0.0167,
-      "step": 7996
-    },
-    {
-      "epoch": 470.41,
-      "learning_rate": 3.1145510835913314e-06,
-      "loss": 0.015,
-      "step": 7997
-    },
-    {
-      "epoch": 470.47,
-      "learning_rate": 3.108359133126935e-06,
-      "loss": 0.015,
-      "step": 7998
-    },
-    {
-      "epoch": 470.53,
-      "learning_rate": 3.102167182662539e-06,
-      "loss": 0.0135,
-      "step": 7999
-    },
-    {
-      "epoch": 470.59,
-      "learning_rate": 3.0959752321981426e-06,
-      "loss": 0.0134,
-      "step": 8000
-    },
-    {
-      "epoch": 470.59,
-      "eval_accuracy_NAT": 0.9943522041202709,
-      "eval_accuracy_REST": 0.49105691056910566,
-      "eval_accuracy_concrete": 0.9594241938270227,
-      "eval_accuracy_concrete tiles": 0.9536177233089795,
-      "eval_accuracy_dark bricks": 0.8549231475987213,
-      "eval_accuracy_grass": 0.8982947083158281,
-      "eval_accuracy_red bricks": 0.9365114235500879,
-      "eval_accuracy_road": 0.9819335282527104,
-      "eval_accuracy_rough concrete": 0.6323360286975849,
-      "eval_accuracy_rough red sidewalk": 0.7691891200866302,
-      "eval_accuracy_speedway bricks": 0.9903725341410802,
-      "eval_accuracy_steel": 0.9307251345664636,
-      "eval_accuracy_tiles": 0.2428125662698052,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9846502652150159,
-      "eval_iou_REST": 0.40700808625336926,
-      "eval_iou_concrete": 0.8979232677145357,
-      "eval_iou_concrete tiles": 0.8936546149259006,
-      "eval_iou_dark bricks": 0.8410895960152348,
-      "eval_iou_grass": 0.8465219093273697,
-      "eval_iou_red bricks": 0.9091006024417551,
-      "eval_iou_road": 0.962065488071277,
-      "eval_iou_rough concrete": 0.6273492045662802,
-      "eval_iou_rough red sidewalk": 0.609908948344434,
-      "eval_iou_speedway bricks": 0.9808436056986679,
-      "eval_iou_steel": 0.8819251888983868,
-      "eval_iou_tiles": 0.24168501040316015,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08850747346878052,
-      "eval_mean_accuracy": 0.818119171023407,
-      "eval_mean_iou": 0.7756712144519529,
-      "eval_overall_accuracy": 0.9810150922355005,
-      "eval_runtime": 5.4821,
-      "eval_samples_per_second": 21.525,
-      "eval_steps_per_second": 1.459,
-      "step": 8000
-    },
-    {
-      "epoch": 470.65,
-      "learning_rate": 3.0897832817337464e-06,
-      "loss": 0.0138,
-      "step": 8001
-    },
-    {
-      "epoch": 470.71,
-      "learning_rate": 3.08359133126935e-06,
-      "loss": 0.0139,
-      "step": 8002
-    },
-    {
-      "epoch": 470.76,
-      "learning_rate": 3.077399380804954e-06,
-      "loss": 0.0139,
-      "step": 8003
-    },
-    {
-      "epoch": 470.82,
-      "learning_rate": 3.0712074303405576e-06,
-      "loss": 0.0131,
-      "step": 8004
-    },
-    {
-      "epoch": 470.88,
-      "learning_rate": 3.0650154798761614e-06,
-      "loss": 0.015,
-      "step": 8005
-    },
-    {
-      "epoch": 470.94,
-      "learning_rate": 3.058823529411765e-06,
-      "loss": 0.0135,
-      "step": 8006
-    },
-    {
-      "epoch": 471.0,
-      "learning_rate": 3.0526315789473684e-06,
-      "loss": 0.015,
-      "step": 8007
-    },
-    {
-      "epoch": 471.06,
-      "learning_rate": 3.046439628482972e-06,
-      "loss": 0.0137,
-      "step": 8008
-    },
-    {
-      "epoch": 471.12,
-      "learning_rate": 3.040247678018576e-06,
-      "loss": 0.0165,
-      "step": 8009
-    },
-    {
-      "epoch": 471.18,
-      "learning_rate": 3.0340557275541797e-06,
-      "loss": 0.0153,
-      "step": 8010
-    },
-    {
-      "epoch": 471.24,
-      "learning_rate": 3.0278637770897834e-06,
-      "loss": 0.0164,
-      "step": 8011
-    },
-    {
-      "epoch": 471.29,
-      "learning_rate": 3.021671826625387e-06,
-      "loss": 0.0161,
-      "step": 8012
-    },
-    {
-      "epoch": 471.35,
-      "learning_rate": 3.015479876160991e-06,
-      "loss": 0.0141,
-      "step": 8013
-    },
-    {
-      "epoch": 471.41,
-      "learning_rate": 3.0092879256965947e-06,
-      "loss": 0.0143,
-      "step": 8014
-    },
-    {
-      "epoch": 471.47,
-      "learning_rate": 3.0030959752321984e-06,
-      "loss": 0.0158,
-      "step": 8015
-    },
-    {
-      "epoch": 471.53,
-      "learning_rate": 2.996904024767802e-06,
-      "loss": 0.015,
-      "step": 8016
-    },
-    {
-      "epoch": 471.59,
-      "learning_rate": 2.990712074303406e-06,
-      "loss": 0.0129,
-      "step": 8017
-    },
-    {
-      "epoch": 471.65,
-      "learning_rate": 2.9845201238390097e-06,
-      "loss": 0.0154,
-      "step": 8018
-    },
-    {
-      "epoch": 471.71,
-      "learning_rate": 2.9783281733746134e-06,
-      "loss": 0.0138,
-      "step": 8019
-    },
-    {
-      "epoch": 471.76,
-      "learning_rate": 2.9721362229102167e-06,
-      "loss": 0.0132,
-      "step": 8020
-    },
-    {
-      "epoch": 471.76,
-      "eval_accuracy_NAT": 0.9942180177519562,
-      "eval_accuracy_REST": 0.5463414634146342,
-      "eval_accuracy_concrete": 0.9598285606833112,
-      "eval_accuracy_concrete tiles": 0.9569408437426349,
-      "eval_accuracy_dark bricks": 0.8547078077445782,
-      "eval_accuracy_grass": 0.903380682570863,
-      "eval_accuracy_red bricks": 0.9407952548330404,
-      "eval_accuracy_road": 0.9807930842277233,
-      "eval_accuracy_rough concrete": 0.634046204541798,
-      "eval_accuracy_rough red sidewalk": 0.7710991961136721,
-      "eval_accuracy_speedway bricks": 0.990407835363325,
-      "eval_accuracy_steel": 0.9288184639681738,
-      "eval_accuracy_tiles": 0.2508103850465025,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847631110721892,
-      "eval_iou_REST": 0.44503311258278144,
-      "eval_iou_concrete": 0.8977911020608945,
-      "eval_iou_concrete tiles": 0.893217766240623,
-      "eval_iou_dark bricks": 0.8408223215000259,
-      "eval_iou_grass": 0.8489362931873858,
-      "eval_iou_red bricks": 0.9090426661006156,
-      "eval_iou_road": 0.9620129682921588,
-      "eval_iou_rough concrete": 0.6286259496886557,
-      "eval_iou_rough red sidewalk": 0.6114198487847926,
-      "eval_iou_speedway bricks": 0.9808254658504204,
-      "eval_iou_steel": 0.8821948956026612,
-      "eval_iou_tiles": 0.24921733895243828,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08830041438341141,
-      "eval_mean_accuracy": 0.8240144461540163,
-      "eval_mean_iou": 0.7795309876858186,
-      "eval_overall_accuracy": 0.9810883473541777,
-      "eval_runtime": 5.553,
-      "eval_samples_per_second": 21.25,
-      "eval_steps_per_second": 1.441,
-      "step": 8020
-    },
-    {
-      "epoch": 471.82,
-      "learning_rate": 2.9659442724458205e-06,
-      "loss": 0.0138,
-      "step": 8021
-    },
-    {
-      "epoch": 471.88,
-      "learning_rate": 2.9597523219814242e-06,
-      "loss": 0.0162,
-      "step": 8022
-    },
-    {
-      "epoch": 471.94,
-      "learning_rate": 2.953560371517028e-06,
-      "loss": 0.0134,
-      "step": 8023
-    },
-    {
-      "epoch": 472.0,
-      "learning_rate": 2.9473684210526317e-06,
-      "loss": 0.0135,
-      "step": 8024
-    },
-    {
-      "epoch": 472.06,
-      "learning_rate": 2.9411764705882355e-06,
-      "loss": 0.0124,
-      "step": 8025
-    },
-    {
-      "epoch": 472.12,
-      "learning_rate": 2.934984520123839e-06,
-      "loss": 0.0179,
-      "step": 8026
-    },
-    {
-      "epoch": 472.18,
-      "learning_rate": 2.928792569659443e-06,
-      "loss": 0.0153,
-      "step": 8027
-    },
-    {
-      "epoch": 472.24,
-      "learning_rate": 2.9226006191950467e-06,
-      "loss": 0.0155,
-      "step": 8028
-    },
-    {
-      "epoch": 472.29,
-      "learning_rate": 2.9164086687306504e-06,
-      "loss": 0.0146,
-      "step": 8029
-    },
-    {
-      "epoch": 472.35,
-      "learning_rate": 2.910216718266254e-06,
-      "loss": 0.0128,
-      "step": 8030
-    },
-    {
-      "epoch": 472.41,
-      "learning_rate": 2.904024767801858e-06,
-      "loss": 0.0145,
-      "step": 8031
-    },
-    {
-      "epoch": 472.47,
-      "learning_rate": 2.8978328173374617e-06,
-      "loss": 0.0146,
-      "step": 8032
-    },
-    {
-      "epoch": 472.53,
-      "learning_rate": 2.8916408668730654e-06,
-      "loss": 0.0153,
-      "step": 8033
-    },
-    {
-      "epoch": 472.59,
-      "learning_rate": 2.8854489164086688e-06,
-      "loss": 0.016,
-      "step": 8034
-    },
-    {
-      "epoch": 472.65,
-      "learning_rate": 2.8792569659442725e-06,
-      "loss": 0.0151,
-      "step": 8035
-    },
-    {
-      "epoch": 472.71,
-      "learning_rate": 2.8730650154798763e-06,
-      "loss": 0.013,
-      "step": 8036
-    },
-    {
-      "epoch": 472.76,
-      "learning_rate": 2.86687306501548e-06,
-      "loss": 0.0156,
-      "step": 8037
-    },
-    {
-      "epoch": 472.82,
-      "learning_rate": 2.8606811145510837e-06,
-      "loss": 0.0126,
-      "step": 8038
-    },
-    {
-      "epoch": 472.88,
-      "learning_rate": 2.8544891640866875e-06,
-      "loss": 0.0139,
-      "step": 8039
-    },
-    {
-      "epoch": 472.94,
-      "learning_rate": 2.8482972136222912e-06,
-      "loss": 0.0136,
-      "step": 8040
-    },
-    {
-      "epoch": 472.94,
-      "eval_accuracy_NAT": 0.994086688706872,
-      "eval_accuracy_REST": 0.5495934959349593,
-      "eval_accuracy_concrete": 0.958830256119393,
-      "eval_accuracy_concrete tiles": 0.9491397596040537,
-      "eval_accuracy_dark bricks": 0.8532913500373256,
-      "eval_accuracy_grass": 0.9047716373903795,
-      "eval_accuracy_red bricks": 0.9404108084358523,
-      "eval_accuracy_road": 0.9820243789950946,
-      "eval_accuracy_rough concrete": 0.6301372311835967,
-      "eval_accuracy_rough red sidewalk": 0.7678956827769798,
-      "eval_accuracy_speedway bricks": 0.9905689186104616,
-      "eval_accuracy_steel": 0.931207424770225,
-      "eval_accuracy_tiles": 0.2476900239328668,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848027309569752,
-      "eval_iou_REST": 0.43896103896103894,
-      "eval_iou_concrete": 0.8974081698720987,
-      "eval_iou_concrete tiles": 0.8951521483029184,
-      "eval_iou_dark bricks": 0.8390812710871437,
-      "eval_iou_grass": 0.8493907086398741,
-      "eval_iou_red bricks": 0.9087194183516425,
-      "eval_iou_road": 0.9616620491076615,
-      "eval_iou_rough concrete": 0.624909143556136,
-      "eval_iou_rough red sidewalk": 0.6096540195229708,
-      "eval_iou_speedway bricks": 0.980767449239117,
-      "eval_iou_steel": 0.8818705894046317,
-      "eval_iou_tiles": 0.24616125730113808,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08815133571624756,
-      "eval_mean_accuracy": 0.8230498197306199,
-      "eval_mean_iou": 0.7783492303310268,
-      "eval_overall_accuracy": 0.9810497154623775,
-      "eval_runtime": 5.5628,
-      "eval_samples_per_second": 21.212,
-      "eval_steps_per_second": 1.438,
-      "step": 8040
-    },
-    {
-      "epoch": 473.0,
-      "learning_rate": 2.842105263157895e-06,
-      "loss": 0.0135,
-      "step": 8041
-    },
-    {
-      "epoch": 473.06,
-      "learning_rate": 2.8359133126934987e-06,
-      "loss": 0.0163,
-      "step": 8042
-    },
-    {
-      "epoch": 473.12,
-      "learning_rate": 2.8297213622291025e-06,
-      "loss": 0.0134,
-      "step": 8043
-    },
-    {
-      "epoch": 473.18,
-      "learning_rate": 2.8235294117647062e-06,
-      "loss": 0.0152,
-      "step": 8044
-    },
-    {
-      "epoch": 473.24,
-      "learning_rate": 2.81733746130031e-06,
-      "loss": 0.0158,
-      "step": 8045
-    },
-    {
-      "epoch": 473.29,
-      "learning_rate": 2.8111455108359137e-06,
-      "loss": 0.0132,
-      "step": 8046
-    },
-    {
-      "epoch": 473.35,
-      "learning_rate": 2.804953560371517e-06,
-      "loss": 0.0147,
-      "step": 8047
-    },
-    {
-      "epoch": 473.41,
-      "learning_rate": 2.798761609907121e-06,
-      "loss": 0.0151,
-      "step": 8048
-    },
-    {
-      "epoch": 473.47,
-      "learning_rate": 2.7925696594427245e-06,
-      "loss": 0.0146,
-      "step": 8049
-    },
-    {
-      "epoch": 473.53,
-      "learning_rate": 2.7863777089783283e-06,
-      "loss": 0.016,
-      "step": 8050
-    },
-    {
-      "epoch": 473.59,
-      "learning_rate": 2.780185758513932e-06,
-      "loss": 0.0127,
-      "step": 8051
-    },
-    {
-      "epoch": 473.65,
-      "learning_rate": 2.7739938080495358e-06,
-      "loss": 0.014,
-      "step": 8052
-    },
-    {
-      "epoch": 473.71,
-      "learning_rate": 2.7678018575851395e-06,
-      "loss": 0.0137,
-      "step": 8053
-    },
-    {
-      "epoch": 473.76,
-      "learning_rate": 2.7616099071207433e-06,
-      "loss": 0.0139,
-      "step": 8054
-    },
-    {
-      "epoch": 473.82,
-      "learning_rate": 2.755417956656347e-06,
-      "loss": 0.013,
-      "step": 8055
-    },
-    {
-      "epoch": 473.88,
-      "learning_rate": 2.7492260061919508e-06,
-      "loss": 0.0165,
-      "step": 8056
-    },
-    {
-      "epoch": 473.94,
-      "learning_rate": 2.7430340557275545e-06,
-      "loss": 0.0142,
-      "step": 8057
-    },
-    {
-      "epoch": 474.0,
-      "learning_rate": 2.7368421052631583e-06,
-      "loss": 0.0141,
-      "step": 8058
-    },
-    {
-      "epoch": 474.06,
-      "learning_rate": 2.730650154798762e-06,
-      "loss": 0.0157,
-      "step": 8059
-    },
-    {
-      "epoch": 474.12,
-      "learning_rate": 2.7244582043343658e-06,
-      "loss": 0.0149,
-      "step": 8060
-    },
-    {
-      "epoch": 474.12,
-      "eval_accuracy_NAT": 0.9942098770763372,
-      "eval_accuracy_REST": 0.5349593495934959,
-      "eval_accuracy_concrete": 0.9595789845423316,
-      "eval_accuracy_concrete tiles": 0.9463587084609946,
-      "eval_accuracy_dark bricks": 0.8544637559098827,
-      "eval_accuracy_grass": 0.9020427165063757,
-      "eval_accuracy_red bricks": 0.9379942882249561,
-      "eval_accuracy_road": 0.9815017200183198,
-      "eval_accuracy_rough concrete": 0.6300895607767892,
-      "eval_accuracy_rough red sidewalk": 0.7731747117965995,
-      "eval_accuracy_speedway bricks": 0.9904217159410038,
-      "eval_accuracy_steel": 0.9312624008556278,
-      "eval_accuracy_tiles": 0.24581174831106667,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847851273510457,
-      "eval_iou_REST": 0.4306282722513089,
-      "eval_iou_concrete": 0.8973651741777204,
-      "eval_iou_concrete tiles": 0.8937410968660968,
-      "eval_iou_dark bricks": 0.8401116014717092,
-      "eval_iou_grass": 0.8488156431445492,
-      "eval_iou_red bricks": 0.9085541015001596,
-      "eval_iou_road": 0.962126542066748,
-      "eval_iou_rough concrete": 0.6249505020715252,
-      "eval_iou_rough red sidewalk": 0.6110289004974238,
-      "eval_iou_speedway bricks": 0.9808170740565003,
-      "eval_iou_steel": 0.8821898805493876,
-      "eval_iou_tiles": 0.2443975903614458,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08821345865726471,
-      "eval_mean_accuracy": 0.8216822721549063,
-      "eval_mean_iou": 0.777654731258894,
-      "eval_overall_accuracy": 0.9810578297760527,
-      "eval_runtime": 5.3181,
-      "eval_samples_per_second": 22.188,
-      "eval_steps_per_second": 1.504,
-      "step": 8060
-    },
-    {
-      "epoch": 474.18,
-      "learning_rate": 2.718266253869969e-06,
-      "loss": 0.0126,
-      "step": 8061
-    },
-    {
-      "epoch": 474.24,
-      "learning_rate": 2.712074303405573e-06,
-      "loss": 0.0159,
-      "step": 8062
-    },
-    {
-      "epoch": 474.29,
-      "learning_rate": 2.7058823529411766e-06,
-      "loss": 0.0144,
-      "step": 8063
-    },
-    {
-      "epoch": 474.35,
-      "learning_rate": 2.6996904024767803e-06,
-      "loss": 0.0154,
-      "step": 8064
-    },
-    {
-      "epoch": 474.41,
-      "learning_rate": 2.693498452012384e-06,
-      "loss": 0.0149,
-      "step": 8065
-    },
-    {
-      "epoch": 474.47,
-      "learning_rate": 2.687306501547988e-06,
-      "loss": 0.0131,
-      "step": 8066
-    },
-    {
-      "epoch": 474.53,
-      "learning_rate": 2.6811145510835916e-06,
-      "loss": 0.0145,
-      "step": 8067
-    },
-    {
-      "epoch": 474.59,
-      "learning_rate": 2.6749226006191953e-06,
-      "loss": 0.0136,
-      "step": 8068
-    },
-    {
-      "epoch": 474.65,
-      "learning_rate": 2.668730650154799e-06,
-      "loss": 0.0153,
-      "step": 8069
-    },
-    {
-      "epoch": 474.71,
-      "learning_rate": 2.662538699690403e-06,
-      "loss": 0.0146,
-      "step": 8070
-    },
-    {
-      "epoch": 474.76,
-      "learning_rate": 2.6563467492260065e-06,
-      "loss": 0.0135,
-      "step": 8071
-    },
-    {
-      "epoch": 474.82,
-      "learning_rate": 2.6501547987616103e-06,
-      "loss": 0.0162,
-      "step": 8072
-    },
-    {
-      "epoch": 474.88,
-      "learning_rate": 2.643962848297214e-06,
-      "loss": 0.0136,
-      "step": 8073
-    },
-    {
-      "epoch": 474.94,
-      "learning_rate": 2.6377708978328174e-06,
-      "loss": 0.0142,
-      "step": 8074
-    },
-    {
-      "epoch": 475.0,
-      "learning_rate": 2.631578947368421e-06,
-      "loss": 0.0131,
-      "step": 8075
-    },
-    {
-      "epoch": 475.06,
-      "learning_rate": 2.625386996904025e-06,
-      "loss": 0.0154,
-      "step": 8076
-    },
-    {
-      "epoch": 475.12,
-      "learning_rate": 2.6191950464396286e-06,
-      "loss": 0.016,
-      "step": 8077
-    },
-    {
-      "epoch": 475.18,
-      "learning_rate": 2.6130030959752324e-06,
-      "loss": 0.0169,
-      "step": 8078
-    },
-    {
-      "epoch": 475.24,
-      "learning_rate": 2.606811145510836e-06,
-      "loss": 0.015,
-      "step": 8079
-    },
-    {
-      "epoch": 475.29,
-      "learning_rate": 2.60061919504644e-06,
-      "loss": 0.0139,
-      "step": 8080
-    },
-    {
-      "epoch": 475.29,
-      "eval_accuracy_NAT": 0.9939909414227712,
-      "eval_accuracy_REST": 0.5398373983739837,
-      "eval_accuracy_concrete": 0.9599019429483465,
-      "eval_accuracy_concrete tiles": 0.9457695027103464,
-      "eval_accuracy_dark bricks": 0.8554399632486649,
-      "eval_accuracy_grass": 0.9041802450351429,
-      "eval_accuracy_red bricks": 0.9404657293497364,
-      "eval_accuracy_road": 0.9821702745990409,
-      "eval_accuracy_rough concrete": 0.6311978977350597,
-      "eval_accuracy_rough red sidewalk": 0.7698884786319644,
-      "eval_accuracy_speedway bricks": 0.9908009470324011,
-      "eval_accuracy_steel": 0.9270617281482555,
-      "eval_accuracy_tiles": 0.2503256687570057,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848573657255217,
-      "eval_iou_REST": 0.43741765480895917,
-      "eval_iou_concrete": 0.8973426171662301,
-      "eval_iou_concrete tiles": 0.8943790674868504,
-      "eval_iou_dark bricks": 0.8412693183743388,
-      "eval_iou_grass": 0.8494958034740354,
-      "eval_iou_red bricks": 0.9094964945825367,
-      "eval_iou_road": 0.9619811229895021,
-      "eval_iou_rough concrete": 0.6256947931126141,
-      "eval_iou_rough red sidewalk": 0.6087067679007795,
-      "eval_iou_speedway bricks": 0.9808730292832912,
-      "eval_iou_steel": 0.8817192034281396,
-      "eval_iou_tiles": 0.24880310740417333,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08831731230020523,
-      "eval_mean_accuracy": 0.8223869783071323,
-      "eval_mean_iou": 0.7786181804413055,
-      "eval_overall_accuracy": 0.9810885736497782,
-      "eval_runtime": 5.7219,
-      "eval_samples_per_second": 20.622,
-      "eval_steps_per_second": 1.398,
-      "step": 8080
-    },
-    {
-      "epoch": 475.35,
-      "learning_rate": 2.5944272445820436e-06,
-      "loss": 0.0129,
-      "step": 8081
-    },
-    {
-      "epoch": 475.41,
-      "learning_rate": 2.5882352941176473e-06,
-      "loss": 0.0164,
-      "step": 8082
-    },
-    {
-      "epoch": 475.47,
-      "learning_rate": 2.582043343653251e-06,
-      "loss": 0.013,
-      "step": 8083
-    },
-    {
-      "epoch": 475.53,
-      "learning_rate": 2.575851393188855e-06,
-      "loss": 0.0123,
-      "step": 8084
-    },
-    {
-      "epoch": 475.59,
-      "learning_rate": 2.5696594427244586e-06,
-      "loss": 0.0126,
-      "step": 8085
-    },
-    {
-      "epoch": 475.65,
-      "learning_rate": 2.5634674922600623e-06,
-      "loss": 0.014,
-      "step": 8086
-    },
-    {
-      "epoch": 475.71,
-      "learning_rate": 2.557275541795666e-06,
-      "loss": 0.0119,
-      "step": 8087
-    },
-    {
-      "epoch": 475.76,
-      "learning_rate": 2.5510835913312694e-06,
-      "loss": 0.0134,
-      "step": 8088
-    },
-    {
-      "epoch": 475.82,
-      "learning_rate": 2.544891640866873e-06,
-      "loss": 0.0169,
-      "step": 8089
-    },
-    {
-      "epoch": 475.88,
-      "learning_rate": 2.538699690402477e-06,
-      "loss": 0.0163,
-      "step": 8090
-    },
-    {
-      "epoch": 475.94,
-      "learning_rate": 2.5325077399380806e-06,
-      "loss": 0.0152,
-      "step": 8091
-    },
-    {
-      "epoch": 476.0,
-      "learning_rate": 2.5263157894736844e-06,
-      "loss": 0.0147,
-      "step": 8092
-    },
-    {
-      "epoch": 476.06,
-      "learning_rate": 2.520123839009288e-06,
-      "loss": 0.0152,
-      "step": 8093
-    },
-    {
-      "epoch": 476.12,
-      "learning_rate": 2.513931888544892e-06,
-      "loss": 0.0144,
-      "step": 8094
-    },
-    {
-      "epoch": 476.18,
-      "learning_rate": 2.5077399380804956e-06,
-      "loss": 0.0129,
-      "step": 8095
-    },
-    {
-      "epoch": 476.24,
-      "learning_rate": 2.5015479876160994e-06,
-      "loss": 0.0147,
-      "step": 8096
-    },
-    {
-      "epoch": 476.29,
-      "learning_rate": 2.495356037151703e-06,
-      "loss": 0.0137,
-      "step": 8097
-    },
-    {
-      "epoch": 476.35,
-      "learning_rate": 2.489164086687307e-06,
-      "loss": 0.0138,
-      "step": 8098
-    },
-    {
-      "epoch": 476.41,
-      "learning_rate": 2.4829721362229106e-06,
-      "loss": 0.0145,
-      "step": 8099
-    },
-    {
-      "epoch": 476.47,
-      "learning_rate": 2.4767801857585144e-06,
-      "loss": 0.0164,
-      "step": 8100
-    },
-    {
-      "epoch": 476.47,
-      "eval_accuracy_NAT": 0.9943045461252568,
-      "eval_accuracy_REST": 0.5170731707317073,
-      "eval_accuracy_concrete": 0.9591046752146813,
-      "eval_accuracy_concrete tiles": 0.9484798491633278,
-      "eval_accuracy_dark bricks": 0.8523821373198324,
-      "eval_accuracy_grass": 0.9029965140969013,
-      "eval_accuracy_red bricks": 0.9393123901581723,
-      "eval_accuracy_road": 0.9817673248357606,
-      "eval_accuracy_rough concrete": 0.6292970402636173,
-      "eval_accuracy_rough red sidewalk": 0.7669331247791005,
-      "eval_accuracy_speedway bricks": 0.9903058730951906,
-      "eval_accuracy_steel": 0.929897994372448,
-      "eval_accuracy_tiles": 0.25044684782937987,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.984765678790844,
-      "eval_iou_REST": 0.42513368983957217,
-      "eval_iou_concrete": 0.8973969638993688,
-      "eval_iou_concrete tiles": 0.8938343994314144,
-      "eval_iou_dark bricks": 0.8394947685927043,
-      "eval_iou_grass": 0.8490282516536848,
-      "eval_iou_red bricks": 0.9095888953890336,
-      "eval_iou_road": 0.9618084732656171,
-      "eval_iou_rough concrete": 0.6246480705988123,
-      "eval_iou_rough red sidewalk": 0.6073125707139879,
-      "eval_iou_speedway bricks": 0.9808877909647117,
-      "eval_iou_steel": 0.8819684253687302,
-      "eval_iou_tiles": 0.249073544033021,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08860953897237778,
-      "eval_mean_accuracy": 0.8201770375373365,
-      "eval_mean_iou": 0.7773031940416542,
-      "eval_overall_accuracy": 0.9810467089636851,
-      "eval_runtime": 5.6303,
-      "eval_samples_per_second": 20.958,
-      "eval_steps_per_second": 1.421,
-      "step": 8100
-    },
-    {
-      "epoch": 476.53,
-      "learning_rate": 2.4705882352941177e-06,
-      "loss": 0.0155,
-      "step": 8101
-    },
-    {
-      "epoch": 476.59,
-      "learning_rate": 2.4643962848297214e-06,
-      "loss": 0.0153,
-      "step": 8102
-    },
-    {
-      "epoch": 476.65,
-      "learning_rate": 2.458204334365325e-06,
-      "loss": 0.0152,
-      "step": 8103
-    },
-    {
-      "epoch": 476.71,
-      "learning_rate": 2.452012383900929e-06,
-      "loss": 0.015,
-      "step": 8104
-    },
-    {
-      "epoch": 476.76,
-      "learning_rate": 2.4458204334365327e-06,
-      "loss": 0.0132,
-      "step": 8105
-    },
-    {
-      "epoch": 476.82,
-      "learning_rate": 2.4396284829721364e-06,
-      "loss": 0.014,
-      "step": 8106
-    },
-    {
-      "epoch": 476.88,
-      "learning_rate": 2.43343653250774e-06,
-      "loss": 0.0138,
-      "step": 8107
-    },
-    {
-      "epoch": 476.94,
-      "learning_rate": 2.427244582043344e-06,
-      "loss": 0.0128,
-      "step": 8108
-    },
-    {
-      "epoch": 477.0,
-      "learning_rate": 2.4210526315789477e-06,
-      "loss": 0.0159,
-      "step": 8109
-    },
-    {
-      "epoch": 477.06,
-      "learning_rate": 2.4148606811145514e-06,
-      "loss": 0.0134,
-      "step": 8110
-    },
-    {
-      "epoch": 477.12,
-      "learning_rate": 2.408668730650155e-06,
-      "loss": 0.0172,
-      "step": 8111
-    },
-    {
-      "epoch": 477.18,
-      "learning_rate": 2.402476780185759e-06,
-      "loss": 0.0141,
-      "step": 8112
-    },
-    {
-      "epoch": 477.24,
-      "learning_rate": 2.3962848297213626e-06,
-      "loss": 0.0141,
-      "step": 8113
-    },
-    {
-      "epoch": 477.29,
-      "learning_rate": 2.3900928792569664e-06,
-      "loss": 0.0135,
-      "step": 8114
-    },
-    {
-      "epoch": 477.35,
-      "learning_rate": 2.3839009287925697e-06,
-      "loss": 0.0141,
-      "step": 8115
-    },
-    {
-      "epoch": 477.41,
-      "learning_rate": 2.3777089783281735e-06,
-      "loss": 0.0137,
-      "step": 8116
-    },
-    {
-      "epoch": 477.47,
-      "learning_rate": 2.3715170278637772e-06,
-      "loss": 0.0147,
-      "step": 8117
-    },
-    {
-      "epoch": 477.53,
-      "learning_rate": 2.365325077399381e-06,
-      "loss": 0.0148,
-      "step": 8118
-    },
-    {
-      "epoch": 477.59,
-      "learning_rate": 2.3591331269349847e-06,
-      "loss": 0.015,
-      "step": 8119
-    },
-    {
-      "epoch": 477.65,
-      "learning_rate": 2.3529411764705885e-06,
-      "loss": 0.0168,
-      "step": 8120
-    },
-    {
-      "epoch": 477.65,
-      "eval_accuracy_NAT": 0.9942390433379918,
-      "eval_accuracy_REST": 0.5089430894308943,
-      "eval_accuracy_concrete": 0.9591765286825285,
-      "eval_accuracy_concrete tiles": 0.9501296252651426,
-      "eval_accuracy_dark bricks": 0.8556265911222556,
-      "eval_accuracy_grass": 0.9030116537411953,
-      "eval_accuracy_red bricks": 0.9400812829525483,
-      "eval_accuracy_road": 0.981890240546045,
-      "eval_accuracy_rough concrete": 0.6308463284848557,
-      "eval_accuracy_rough red sidewalk": 0.7575933042059273,
-      "eval_accuracy_speedway bricks": 0.9902542921830756,
-      "eval_accuracy_steel": 0.9320845432236977,
-      "eval_accuracy_tiles": 0.24929564664182496,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9847758910321971,
-      "eval_iou_REST": 0.42469470827679784,
-      "eval_iou_concrete": 0.8973759442184777,
-      "eval_iou_concrete tiles": 0.8945149551788408,
-      "eval_iou_dark bricks": 0.8412074108226615,
-      "eval_iou_grass": 0.8489488306057401,
-      "eval_iou_red bricks": 0.9095594877517402,
-      "eval_iou_road": 0.9614099385577234,
-      "eval_iou_rough concrete": 0.625804659190996,
-      "eval_iou_rough red sidewalk": 0.6026295955112637,
-      "eval_iou_speedway bricks": 0.9808630055790136,
-      "eval_iou_steel": 0.8822648667954992,
-      "eval_iou_tiles": 0.2478838449257463,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0889807716012001,
-      "eval_mean_accuracy": 0.819474782293691,
-      "eval_mean_iou": 0.7770717798805151,
-      "eval_overall_accuracy": 0.9810315148305084,
-      "eval_runtime": 5.2947,
-      "eval_samples_per_second": 22.287,
-      "eval_steps_per_second": 1.511,
-      "step": 8120
-    },
-    {
-      "epoch": 477.71,
-      "learning_rate": 2.346749226006192e-06,
-      "loss": 0.0149,
-      "step": 8121
-    },
-    {
-      "epoch": 477.76,
-      "learning_rate": 2.340557275541796e-06,
-      "loss": 0.0155,
-      "step": 8122
-    },
-    {
-      "epoch": 477.82,
-      "learning_rate": 2.3343653250773997e-06,
-      "loss": 0.0134,
-      "step": 8123
-    },
-    {
-      "epoch": 477.88,
-      "learning_rate": 2.3281733746130034e-06,
-      "loss": 0.0145,
-      "step": 8124
-    },
-    {
-      "epoch": 477.94,
-      "learning_rate": 2.321981424148607e-06,
-      "loss": 0.0155,
-      "step": 8125
-    },
-    {
-      "epoch": 478.0,
-      "learning_rate": 2.315789473684211e-06,
-      "loss": 0.014,
-      "step": 8126
-    },
-    {
-      "epoch": 478.06,
-      "learning_rate": 2.3095975232198147e-06,
-      "loss": 0.0142,
-      "step": 8127
-    },
-    {
-      "epoch": 478.12,
-      "learning_rate": 2.3034055727554184e-06,
-      "loss": 0.0162,
-      "step": 8128
-    },
-    {
-      "epoch": 478.18,
-      "learning_rate": 2.2972136222910218e-06,
-      "loss": 0.0146,
-      "step": 8129
-    },
-    {
-      "epoch": 478.24,
-      "learning_rate": 2.2910216718266255e-06,
-      "loss": 0.0153,
-      "step": 8130
-    },
-    {
-      "epoch": 478.29,
-      "learning_rate": 2.2848297213622292e-06,
-      "loss": 0.016,
-      "step": 8131
-    },
-    {
-      "epoch": 478.35,
-      "learning_rate": 2.278637770897833e-06,
-      "loss": 0.015,
-      "step": 8132
-    },
-    {
-      "epoch": 478.41,
-      "learning_rate": 2.2724458204334367e-06,
-      "loss": 0.0141,
-      "step": 8133
-    },
-    {
-      "epoch": 478.47,
-      "learning_rate": 2.2662538699690405e-06,
-      "loss": 0.0139,
-      "step": 8134
-    },
-    {
-      "epoch": 478.53,
-      "learning_rate": 2.2600619195046442e-06,
-      "loss": 0.0125,
-      "step": 8135
-    },
-    {
-      "epoch": 478.59,
-      "learning_rate": 2.253869969040248e-06,
-      "loss": 0.0138,
-      "step": 8136
-    },
-    {
-      "epoch": 478.65,
-      "learning_rate": 2.2476780185758517e-06,
-      "loss": 0.0139,
-      "step": 8137
-    },
-    {
-      "epoch": 478.71,
-      "learning_rate": 2.2414860681114555e-06,
-      "loss": 0.0134,
-      "step": 8138
-    },
-    {
-      "epoch": 478.76,
-      "learning_rate": 2.2352941176470592e-06,
-      "loss": 0.0134,
-      "step": 8139
-    },
-    {
-      "epoch": 478.82,
-      "learning_rate": 2.229102167182663e-06,
-      "loss": 0.0148,
-      "step": 8140
-    },
-    {
-      "epoch": 478.82,
-      "eval_accuracy_NAT": 0.9941614643166962,
-      "eval_accuracy_REST": 0.5252032520325203,
-      "eval_accuracy_concrete": 0.9588929368041106,
-      "eval_accuracy_concrete tiles": 0.9535941550789536,
-      "eval_accuracy_dark bricks": 0.8582537373428019,
-      "eval_accuracy_grass": 0.9035803366299909,
-      "eval_accuracy_red bricks": 0.9442003514938488,
-      "eval_accuracy_road": 0.9820858368502369,
-      "eval_accuracy_rough concrete": 0.6332834780328807,
-      "eval_accuracy_rough red sidewalk": 0.7735883109363133,
-      "eval_accuracy_speedway bricks": 0.9903906988476722,
-      "eval_accuracy_steel": 0.9303078160999965,
-      "eval_accuracy_tiles": 0.2513859856402799,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848018055319168,
-      "eval_iou_REST": 0.42838196286472147,
-      "eval_iou_concrete": 0.8977969838471247,
-      "eval_iou_concrete tiles": 0.8932971254470791,
-      "eval_iou_dark bricks": 0.8434331558849533,
-      "eval_iou_grass": 0.8491979151752723,
-      "eval_iou_red bricks": 0.9104485516072658,
-      "eval_iou_road": 0.9617619952029132,
-      "eval_iou_rough concrete": 0.6278326520002836,
-      "eval_iou_rough red sidewalk": 0.6089347445186343,
-      "eval_iou_speedway bricks": 0.9809368450567677,
-      "eval_iou_steel": 0.8823864026906404,
-      "eval_iou_tiles": 0.24990212317422075,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08883687108755112,
-      "eval_mean_accuracy": 0.8229944892389462,
-      "eval_mean_iou": 0.7783932510001379,
-      "eval_overall_accuracy": 0.9811044789976993,
-      "eval_runtime": 5.4231,
-      "eval_samples_per_second": 21.759,
-      "eval_steps_per_second": 1.475,
-      "step": 8140
-    },
-    {
-      "epoch": 478.88,
-      "learning_rate": 2.2229102167182667e-06,
-      "loss": 0.0181,
-      "step": 8141
-    },
-    {
-      "epoch": 478.94,
-      "learning_rate": 2.21671826625387e-06,
-      "loss": 0.0139,
-      "step": 8142
-    },
-    {
-      "epoch": 479.0,
-      "learning_rate": 2.2105263157894738e-06,
-      "loss": 0.0154,
-      "step": 8143
-    },
-    {
-      "epoch": 479.06,
-      "learning_rate": 2.2043343653250775e-06,
-      "loss": 0.0134,
-      "step": 8144
-    },
-    {
-      "epoch": 479.12,
-      "learning_rate": 2.1981424148606813e-06,
-      "loss": 0.014,
-      "step": 8145
-    },
-    {
-      "epoch": 479.18,
-      "learning_rate": 2.191950464396285e-06,
-      "loss": 0.0155,
-      "step": 8146
-    },
-    {
-      "epoch": 479.24,
-      "learning_rate": 2.1857585139318888e-06,
-      "loss": 0.0136,
-      "step": 8147
-    },
-    {
-      "epoch": 479.29,
-      "learning_rate": 2.1795665634674925e-06,
-      "loss": 0.0146,
-      "step": 8148
-    },
-    {
-      "epoch": 479.35,
-      "learning_rate": 2.1733746130030963e-06,
-      "loss": 0.0161,
-      "step": 8149
-    },
-    {
-      "epoch": 479.41,
-      "learning_rate": 2.1671826625387e-06,
-      "loss": 0.0151,
-      "step": 8150
-    },
-    {
-      "epoch": 479.47,
-      "learning_rate": 2.1609907120743038e-06,
-      "loss": 0.013,
-      "step": 8151
-    },
-    {
-      "epoch": 479.53,
-      "learning_rate": 2.1547987616099075e-06,
-      "loss": 0.0146,
-      "step": 8152
-    },
-    {
-      "epoch": 479.59,
-      "learning_rate": 2.1486068111455113e-06,
-      "loss": 0.0113,
-      "step": 8153
-    },
-    {
-      "epoch": 479.65,
-      "learning_rate": 2.142414860681115e-06,
-      "loss": 0.0161,
-      "step": 8154
-    },
-    {
-      "epoch": 479.71,
-      "learning_rate": 2.1362229102167187e-06,
-      "loss": 0.0161,
-      "step": 8155
-    },
-    {
-      "epoch": 479.76,
-      "learning_rate": 2.130030959752322e-06,
-      "loss": 0.0154,
-      "step": 8156
-    },
-    {
-      "epoch": 479.82,
-      "learning_rate": 2.123839009287926e-06,
-      "loss": 0.014,
-      "step": 8157
-    },
-    {
-      "epoch": 479.88,
-      "learning_rate": 2.1176470588235296e-06,
-      "loss": 0.013,
-      "step": 8158
-    },
-    {
-      "epoch": 479.94,
-      "learning_rate": 2.1114551083591333e-06,
-      "loss": 0.0159,
-      "step": 8159
-    },
-    {
-      "epoch": 480.0,
-      "learning_rate": 2.105263157894737e-06,
-      "loss": 0.0141,
-      "step": 8160
-    },
-    {
-      "epoch": 480.0,
-      "eval_accuracy_NAT": 0.994172731874341,
-      "eval_accuracy_REST": 0.5235772357723577,
-      "eval_accuracy_concrete": 0.9586078161285045,
-      "eval_accuracy_concrete tiles": 0.9552203629507424,
-      "eval_accuracy_dark bricks": 0.8557940776754781,
-      "eval_accuracy_grass": 0.9027523873326596,
-      "eval_accuracy_red bricks": 0.9437060632688928,
-      "eval_accuracy_road": 0.981633186386711,
-      "eval_accuracy_rough concrete": 0.6325028751214106,
-      "eval_accuracy_rough red sidewalk": 0.7710390362388047,
-      "eval_accuracy_speedway bricks": 0.9907354855426072,
-      "eval_accuracy_steel": 0.9298405193740723,
-      "eval_accuracy_tiles": 0.2473870762519313,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848199619481691,
-      "eval_iou_REST": 0.42990654205607476,
-      "eval_iou_concrete": 0.897311681198297,
-      "eval_iou_concrete tiles": 0.8943466172381835,
-      "eval_iou_dark bricks": 0.8415146011161407,
-      "eval_iou_grass": 0.8489520395941634,
-      "eval_iou_red bricks": 0.908624610015335,
-      "eval_iou_road": 0.961601544565672,
-      "eval_iou_rough concrete": 0.627062472308374,
-      "eval_iou_rough red sidewalk": 0.6069460312198471,
-      "eval_iou_speedway bricks": 0.9808634690503788,
-      "eval_iou_steel": 0.8823488960501573,
-      "eval_iou_tiles": 0.24594163178026082,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08873803168535233,
-      "eval_mean_accuracy": 0.8220745272245009,
-      "eval_mean_iou": 0.777710776780081,
-      "eval_overall_accuracy": 0.9810562780347921,
-      "eval_runtime": 5.5544,
-      "eval_samples_per_second": 21.244,
-      "eval_steps_per_second": 1.44,
-      "step": 8160
-    },
-    {
-      "epoch": 480.06,
-      "learning_rate": 2.099071207430341e-06,
-      "loss": 0.0141,
-      "step": 8161
-    },
-    {
-      "epoch": 480.12,
-      "learning_rate": 2.0928792569659446e-06,
-      "loss": 0.0129,
-      "step": 8162
-    },
-    {
-      "epoch": 480.18,
-      "learning_rate": 2.0866873065015483e-06,
-      "loss": 0.0168,
-      "step": 8163
-    },
-    {
-      "epoch": 480.24,
-      "learning_rate": 2.080495356037152e-06,
-      "loss": 0.0137,
-      "step": 8164
-    },
-    {
-      "epoch": 480.29,
-      "learning_rate": 2.074303405572756e-06,
-      "loss": 0.0138,
-      "step": 8165
-    },
-    {
-      "epoch": 480.35,
-      "learning_rate": 2.0681114551083595e-06,
-      "loss": 0.013,
-      "step": 8166
-    },
-    {
-      "epoch": 480.41,
-      "learning_rate": 2.0619195046439633e-06,
-      "loss": 0.0152,
-      "step": 8167
-    },
-    {
-      "epoch": 480.47,
-      "learning_rate": 2.0557275541795666e-06,
-      "loss": 0.0147,
-      "step": 8168
-    },
-    {
-      "epoch": 480.53,
-      "learning_rate": 2.0495356037151704e-06,
-      "loss": 0.0144,
-      "step": 8169
-    },
-    {
-      "epoch": 480.59,
-      "learning_rate": 2.043343653250774e-06,
-      "loss": 0.0164,
-      "step": 8170
-    },
-    {
-      "epoch": 480.65,
-      "learning_rate": 2.037151702786378e-06,
-      "loss": 0.0164,
-      "step": 8171
-    },
-    {
-      "epoch": 480.71,
-      "learning_rate": 2.0309597523219816e-06,
-      "loss": 0.0137,
-      "step": 8172
-    },
-    {
-      "epoch": 480.76,
-      "learning_rate": 2.0247678018575853e-06,
-      "loss": 0.0133,
-      "step": 8173
-    },
-    {
-      "epoch": 480.82,
-      "learning_rate": 2.018575851393189e-06,
-      "loss": 0.0156,
-      "step": 8174
-    },
-    {
-      "epoch": 480.88,
-      "learning_rate": 2.012383900928793e-06,
-      "loss": 0.0155,
-      "step": 8175
-    },
-    {
-      "epoch": 480.94,
-      "learning_rate": 2.0061919504643966e-06,
-      "loss": 0.0161,
-      "step": 8176
-    },
-    {
-      "epoch": 481.0,
-      "learning_rate": 2.0000000000000003e-06,
-      "loss": 0.0139,
-      "step": 8177
-    },
-    {
-      "epoch": 481.06,
-      "learning_rate": 1.993808049535604e-06,
-      "loss": 0.0145,
-      "step": 8178
-    },
-    {
-      "epoch": 481.12,
-      "learning_rate": 1.987616099071208e-06,
-      "loss": 0.0137,
-      "step": 8179
-    },
-    {
-      "epoch": 481.18,
-      "learning_rate": 1.981424148606811e-06,
-      "loss": 0.0175,
-      "step": 8180
-    },
-    {
-      "epoch": 481.18,
-      "eval_accuracy_NAT": 0.9941835142261541,
-      "eval_accuracy_REST": 0.526829268292683,
-      "eval_accuracy_concrete": 0.9592873664786755,
-      "eval_accuracy_concrete tiles": 0.9523921753476314,
-      "eval_accuracy_dark bricks": 0.8569377715674827,
-      "eval_accuracy_grass": 0.9033153928548449,
-      "eval_accuracy_red bricks": 0.9432666959578208,
-      "eval_accuracy_road": 0.9818437463425896,
-      "eval_accuracy_rough concrete": 0.6337244292958485,
-      "eval_accuracy_rough red sidewalk": 0.7734529512178615,
-      "eval_accuracy_speedway bricks": 0.990416917716621,
-      "eval_accuracy_steel": 0.9291708106973466,
-      "eval_accuracy_tiles": 0.24475143142779243,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848184410990569,
-      "eval_iou_REST": 0.4268774703557312,
-      "eval_iou_concrete": 0.8975887650936868,
-      "eval_iou_concrete tiles": 0.8949373255968464,
-      "eval_iou_dark bricks": 0.8423894891829468,
-      "eval_iou_grass": 0.8490954031484115,
-      "eval_iou_red bricks": 0.9089706271500397,
-      "eval_iou_road": 0.9619321924213088,
-      "eval_iou_rough concrete": 0.6289162753840877,
-      "eval_iou_rough red sidewalk": 0.6098425780439358,
-      "eval_iou_speedway bricks": 0.9809141985843913,
-      "eval_iou_steel": 0.8822008161715859,
-      "eval_iou_tiles": 0.243446031459049,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08796320110559464,
-      "eval_mean_accuracy": 0.822274805494104,
-      "eval_mean_iou": 0.7778407395146983,
-      "eval_overall_accuracy": 0.9811012785313493,
-      "eval_runtime": 5.4584,
-      "eval_samples_per_second": 21.618,
-      "eval_steps_per_second": 1.466,
-      "step": 8180
-    },
-    {
-      "epoch": 481.24,
-      "learning_rate": 1.975232198142415e-06,
-      "loss": 0.0156,
-      "step": 8181
-    },
-    {
-      "epoch": 481.29,
-      "learning_rate": 1.9690402476780186e-06,
-      "loss": 0.0159,
-      "step": 8182
-    },
-    {
-      "epoch": 481.35,
-      "learning_rate": 1.9628482972136224e-06,
-      "loss": 0.0126,
-      "step": 8183
-    },
-    {
-      "epoch": 481.41,
-      "learning_rate": 1.956656346749226e-06,
-      "loss": 0.0142,
-      "step": 8184
-    },
-    {
-      "epoch": 481.47,
-      "learning_rate": 1.95046439628483e-06,
-      "loss": 0.0142,
-      "step": 8185
-    },
-    {
-      "epoch": 481.53,
-      "learning_rate": 1.9442724458204336e-06,
-      "loss": 0.0152,
-      "step": 8186
-    },
-    {
-      "epoch": 481.59,
-      "learning_rate": 1.9380804953560374e-06,
-      "loss": 0.0163,
-      "step": 8187
-    },
-    {
-      "epoch": 481.65,
-      "learning_rate": 1.931888544891641e-06,
-      "loss": 0.0131,
-      "step": 8188
-    },
-    {
-      "epoch": 481.71,
-      "learning_rate": 1.925696594427245e-06,
-      "loss": 0.0124,
-      "step": 8189
-    },
-    {
-      "epoch": 481.76,
-      "learning_rate": 1.9195046439628486e-06,
-      "loss": 0.014,
-      "step": 8190
-    },
-    {
-      "epoch": 481.82,
-      "learning_rate": 1.9133126934984524e-06,
-      "loss": 0.0139,
-      "step": 8191
-    },
-    {
-      "epoch": 481.88,
-      "learning_rate": 1.907120743034056e-06,
-      "loss": 0.0171,
-      "step": 8192
-    },
-    {
-      "epoch": 481.94,
-      "learning_rate": 1.9009287925696596e-06,
-      "loss": 0.0133,
-      "step": 8193
-    },
-    {
-      "epoch": 482.0,
-      "learning_rate": 1.8947368421052634e-06,
-      "loss": 0.015,
-      "step": 8194
-    },
-    {
-      "epoch": 482.06,
-      "learning_rate": 1.8885448916408671e-06,
-      "loss": 0.0146,
-      "step": 8195
-    },
-    {
-      "epoch": 482.12,
-      "learning_rate": 1.8823529411764705e-06,
-      "loss": 0.0158,
-      "step": 8196
-    },
-    {
-      "epoch": 482.18,
-      "learning_rate": 1.8761609907120742e-06,
-      "loss": 0.0155,
-      "step": 8197
-    },
-    {
-      "epoch": 482.24,
-      "learning_rate": 1.869969040247678e-06,
-      "loss": 0.0145,
-      "step": 8198
-    },
-    {
-      "epoch": 482.29,
-      "learning_rate": 1.8637770897832817e-06,
-      "loss": 0.0134,
-      "step": 8199
-    },
-    {
-      "epoch": 482.35,
-      "learning_rate": 1.8575851393188855e-06,
-      "loss": 0.0141,
-      "step": 8200
-    },
-    {
-      "epoch": 482.35,
-      "eval_accuracy_NAT": 0.9942189881636194,
-      "eval_accuracy_REST": 0.5089430894308943,
-      "eval_accuracy_concrete": 0.9591188165886725,
-      "eval_accuracy_concrete tiles": 0.9524157435776573,
-      "eval_accuracy_dark bricks": 0.8563826732768026,
-      "eval_accuracy_grass": 0.9027325165495237,
-      "eval_accuracy_red bricks": 0.9422231985940246,
-      "eval_accuracy_road": 0.9816353240512377,
-      "eval_accuracy_rough concrete": 0.633760182100954,
-      "eval_accuracy_rough red sidewalk": 0.766752645154498,
-      "eval_accuracy_speedway bricks": 0.9904794659987538,
-      "eval_accuracy_steel": 0.9316172464977734,
-      "eval_accuracy_tiles": 0.24132812263322123,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848080882691804,
-      "eval_iou_REST": 0.4184491978609626,
-      "eval_iou_concrete": 0.8976493460270675,
-      "eval_iou_concrete tiles": 0.8959714431412544,
-      "eval_iou_dark bricks": 0.8417448237585017,
-      "eval_iou_grass": 0.8489137127745527,
-      "eval_iou_red bricks": 0.9085420748821691,
-      "eval_iou_road": 0.9619052756048182,
-      "eval_iou_rough concrete": 0.6284723248104662,
-      "eval_iou_rough red sidewalk": 0.6073721958135269,
-      "eval_iou_speedway bricks": 0.9808334836559571,
-      "eval_iou_steel": 0.8822273810453479,
-      "eval_iou_tiles": 0.24013504958852078,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08833372592926025,
-      "eval_mean_accuracy": 0.8201236932782796,
-      "eval_mean_iou": 0.7766941844024865,
-      "eval_overall_accuracy": 0.9810822697009071,
-      "eval_runtime": 6.5453,
-      "eval_samples_per_second": 18.028,
-      "eval_steps_per_second": 1.222,
-      "step": 8200
-    },
-    {
-      "epoch": 482.41,
-      "learning_rate": 1.8513931888544892e-06,
-      "loss": 0.0148,
-      "step": 8201
-    },
-    {
-      "epoch": 482.47,
-      "learning_rate": 1.845201238390093e-06,
-      "loss": 0.0135,
-      "step": 8202
-    },
-    {
-      "epoch": 482.53,
-      "learning_rate": 1.8390092879256967e-06,
-      "loss": 0.0134,
-      "step": 8203
-    },
-    {
-      "epoch": 482.59,
-      "learning_rate": 1.8328173374613004e-06,
-      "loss": 0.0147,
-      "step": 8204
-    },
-    {
-      "epoch": 482.65,
-      "learning_rate": 1.8266253869969042e-06,
-      "loss": 0.0132,
-      "step": 8205
-    },
-    {
-      "epoch": 482.71,
-      "learning_rate": 1.820433436532508e-06,
-      "loss": 0.0146,
-      "step": 8206
-    },
-    {
-      "epoch": 482.76,
-      "learning_rate": 1.8142414860681117e-06,
-      "loss": 0.0147,
-      "step": 8207
-    },
-    {
-      "epoch": 482.82,
-      "learning_rate": 1.8080495356037154e-06,
-      "loss": 0.016,
-      "step": 8208
-    },
-    {
-      "epoch": 482.88,
-      "learning_rate": 1.8018575851393192e-06,
-      "loss": 0.0138,
-      "step": 8209
-    },
-    {
-      "epoch": 482.94,
-      "learning_rate": 1.7956656346749225e-06,
-      "loss": 0.0143,
-      "step": 8210
-    },
-    {
-      "epoch": 483.0,
-      "learning_rate": 1.7894736842105262e-06,
-      "loss": 0.0161,
-      "step": 8211
-    },
-    {
-      "epoch": 483.06,
-      "learning_rate": 1.78328173374613e-06,
-      "loss": 0.0161,
-      "step": 8212
-    },
-    {
-      "epoch": 483.12,
-      "learning_rate": 1.7770897832817337e-06,
-      "loss": 0.0146,
-      "step": 8213
-    },
-    {
-      "epoch": 483.18,
-      "learning_rate": 1.7708978328173375e-06,
-      "loss": 0.0176,
-      "step": 8214
-    },
-    {
-      "epoch": 483.24,
-      "learning_rate": 1.7647058823529412e-06,
-      "loss": 0.0134,
-      "step": 8215
-    },
-    {
-      "epoch": 483.29,
-      "learning_rate": 1.758513931888545e-06,
-      "loss": 0.0145,
-      "step": 8216
-    },
-    {
-      "epoch": 483.35,
-      "learning_rate": 1.7523219814241487e-06,
-      "loss": 0.0178,
-      "step": 8217
-    },
-    {
-      "epoch": 483.41,
-      "learning_rate": 1.7461300309597525e-06,
-      "loss": 0.013,
-      "step": 8218
-    },
-    {
-      "epoch": 483.47,
-      "learning_rate": 1.7399380804953562e-06,
-      "loss": 0.0141,
-      "step": 8219
-    },
-    {
-      "epoch": 483.53,
-      "learning_rate": 1.73374613003096e-06,
-      "loss": 0.0141,
-      "step": 8220
-    },
-    {
-      "epoch": 483.53,
-      "eval_accuracy_NAT": 0.9941529462587638,
-      "eval_accuracy_REST": 0.5138211382113821,
-      "eval_accuracy_concrete": 0.9593515759605815,
-      "eval_accuracy_concrete tiles": 0.9526985623379685,
-      "eval_accuracy_dark bricks": 0.8545977451524606,
-      "eval_accuracy_grass": 0.9032728126052678,
-      "eval_accuracy_red bricks": 0.9412895430579965,
-      "eval_accuracy_road": 0.9821216427310588,
-      "eval_accuracy_rough concrete": 0.6310489277137868,
-      "eval_accuracy_rough red sidewalk": 0.7660457666248054,
-      "eval_accuracy_speedway bricks": 0.9904864919701715,
-      "eval_accuracy_steel": 0.9297030791605652,
-      "eval_accuracy_tiles": 0.2429943348783665,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848154042521375,
-      "eval_iou_REST": 0.42133333333333334,
-      "eval_iou_concrete": 0.8974200196067355,
-      "eval_iou_concrete tiles": 0.895621925820889,
-      "eval_iou_dark bricks": 0.84042523153377,
-      "eval_iou_grass": 0.8490206419835657,
-      "eval_iou_red bricks": 0.909086087094892,
-      "eval_iou_road": 0.9616541262891628,
-      "eval_iou_rough concrete": 0.626168500594228,
-      "eval_iou_rough red sidewalk": 0.6067797215918229,
-      "eval_iou_speedway bricks": 0.9808993662651015,
-      "eval_iou_steel": 0.8819445102929044,
-      "eval_iou_tiles": 0.24156000602318928,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08903730660676956,
-      "eval_mean_accuracy": 0.820121889743321,
-      "eval_mean_iou": 0.7766714518985947,
-      "eval_overall_accuracy": 0.9810587026305118,
-      "eval_runtime": 5.365,
-      "eval_samples_per_second": 21.994,
-      "eval_steps_per_second": 1.491,
-      "step": 8220
-    },
-    {
-      "epoch": 483.59,
-      "learning_rate": 1.7275541795665637e-06,
-      "loss": 0.0151,
-      "step": 8221
-    },
-    {
-      "epoch": 483.65,
-      "learning_rate": 1.7213622291021675e-06,
-      "loss": 0.0132,
-      "step": 8222
-    },
-    {
-      "epoch": 483.71,
-      "learning_rate": 1.7151702786377708e-06,
-      "loss": 0.0144,
-      "step": 8223
-    },
-    {
-      "epoch": 483.76,
-      "learning_rate": 1.7089783281733745e-06,
-      "loss": 0.0127,
-      "step": 8224
-    },
-    {
-      "epoch": 483.82,
-      "learning_rate": 1.7027863777089783e-06,
-      "loss": 0.0136,
-      "step": 8225
-    },
-    {
-      "epoch": 483.88,
-      "learning_rate": 1.696594427244582e-06,
-      "loss": 0.0144,
-      "step": 8226
-    },
-    {
-      "epoch": 483.94,
-      "learning_rate": 1.6904024767801858e-06,
-      "loss": 0.0125,
-      "step": 8227
-    },
-    {
-      "epoch": 484.0,
-      "learning_rate": 1.6842105263157895e-06,
-      "loss": 0.0157,
-      "step": 8228
-    },
-    {
-      "epoch": 484.06,
-      "learning_rate": 1.6780185758513933e-06,
-      "loss": 0.0149,
-      "step": 8229
-    },
-    {
-      "epoch": 484.12,
-      "learning_rate": 1.671826625386997e-06,
-      "loss": 0.0152,
-      "step": 8230
-    },
-    {
-      "epoch": 484.18,
-      "learning_rate": 1.6656346749226008e-06,
-      "loss": 0.0148,
-      "step": 8231
-    },
-    {
-      "epoch": 484.24,
-      "learning_rate": 1.6594427244582045e-06,
-      "loss": 0.0152,
-      "step": 8232
-    },
-    {
-      "epoch": 484.29,
-      "learning_rate": 1.6532507739938083e-06,
-      "loss": 0.0144,
-      "step": 8233
-    },
-    {
-      "epoch": 484.35,
-      "learning_rate": 1.647058823529412e-06,
-      "loss": 0.0138,
-      "step": 8234
-    },
-    {
-      "epoch": 484.41,
-      "learning_rate": 1.6408668730650157e-06,
-      "loss": 0.0153,
-      "step": 8235
-    },
-    {
-      "epoch": 484.47,
-      "learning_rate": 1.6346749226006195e-06,
-      "loss": 0.013,
-      "step": 8236
-    },
-    {
-      "epoch": 484.53,
-      "learning_rate": 1.6284829721362228e-06,
-      "loss": 0.0166,
-      "step": 8237
-    },
-    {
-      "epoch": 484.59,
-      "learning_rate": 1.6222910216718266e-06,
-      "loss": 0.0152,
-      "step": 8238
-    },
-    {
-      "epoch": 484.65,
-      "learning_rate": 1.6160990712074303e-06,
-      "loss": 0.0149,
-      "step": 8239
-    },
-    {
-      "epoch": 484.71,
-      "learning_rate": 1.609907120743034e-06,
-      "loss": 0.0132,
-      "step": 8240
-    },
-    {
-      "epoch": 484.71,
-      "eval_accuracy_NAT": 0.9940273318601406,
-      "eval_accuracy_REST": 0.5235772357723577,
-      "eval_accuracy_concrete": 0.95982091669737,
-      "eval_accuracy_concrete tiles": 0.955644591091209,
-      "eval_accuracy_dark bricks": 0.8543297666673048,
-      "eval_accuracy_grass": 0.903255780505437,
-      "eval_accuracy_red bricks": 0.9427174868189807,
-      "eval_accuracy_road": 0.9817737378293406,
-      "eval_accuracy_rough concrete": 0.6318056954218533,
-      "eval_accuracy_rough red sidewalk": 0.7625564938824927,
-      "eval_accuracy_speedway bricks": 0.9905999357037932,
-      "eval_accuracy_steel": 0.9322494714799062,
-      "eval_accuracy_tiles": 0.24157048077796964,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9848482323462975,
-      "eval_iou_REST": 0.42480211081794195,
-      "eval_iou_concrete": 0.8972809775618122,
-      "eval_iou_concrete tiles": 0.8956331588363926,
-      "eval_iou_dark bricks": 0.840280327205294,
-      "eval_iou_grass": 0.8491216896577489,
-      "eval_iou_red bricks": 0.9091149833165616,
-      "eval_iou_road": 0.9616229378793685,
-      "eval_iou_rough concrete": 0.6267600638411066,
-      "eval_iou_rough red sidewalk": 0.6044695601375799,
-      "eval_iou_speedway bricks": 0.9808186284748409,
-      "eval_iou_steel": 0.8820329253328353,
-      "eval_iou_tiles": 0.24017348875033884,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08873334527015686,
-      "eval_mean_accuracy": 0.8210714557313966,
-      "eval_mean_iou": 0.7766891603198551,
-      "eval_overall_accuracy": 0.9810466443077993,
-      "eval_runtime": 5.6522,
-      "eval_samples_per_second": 20.877,
-      "eval_steps_per_second": 1.415,
-      "step": 8240
-    }
-  ],
-  "max_steps": 8500,
-  "num_train_epochs": 500,
-  "total_flos": 5.205249032303149e+19,
-  "trial_name": null,
-  "trial_params": null
-}