diff --git "a/checkpoint-4800/trainer_state.json" "b/checkpoint-4800/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-4800/trainer_state.json"
+++ /dev/null
@@ -1,38176 +0,0 @@
-{
-  "best_metric": 0.07582816481590271,
-  "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-1360",
-  "epoch": 282.3529411764706,
-  "global_step": 4800,
-  "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
-    }
-  ],
-  "max_steps": 5100,
-  "num_train_epochs": 300,
-  "total_flos": 3.0321769257872916e+19,
-  "trial_name": null,
-  "trial_params": null
-}