diff --git "a/checkpoint-3120/trainer_state.json" "b/checkpoint-3120/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-3120/trainer_state.json"
+++ /dev/null
@@ -1,24820 +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": 183.52941176470588,
-  "global_step": 3120,
-  "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
-    }
-  ],
-  "max_steps": 3400,
-  "num_train_epochs": 200,
-  "total_flos": 1.9709509308666348e+19,
-  "trial_name": null,
-  "trial_params": null
-}