diff --git "a/checkpoint-560/trainer_state.json" "b/checkpoint-560/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-560/trainer_state.json"
@@ -0,0 +1,4468 @@
+{
+  "best_metric": 0.14111237227916718,
+  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-userstudy-terrain/checkpoint-240",
+  "epoch": 560.0,
+  "global_step": 560,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "learning_rate": 1.6666666666666667e-06,
+      "loss": 2.8187,
+      "step": 1
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 3.3333333333333333e-06,
+      "loss": 2.8228,
+      "step": 2
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 5e-06,
+      "loss": 2.821,
+      "step": 3
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 6.666666666666667e-06,
+      "loss": 2.7963,
+      "step": 4
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 8.333333333333334e-06,
+      "loss": 2.7811,
+      "step": 5
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1e-05,
+      "loss": 2.7249,
+      "step": 6
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 1.1666666666666668e-05,
+      "loss": 2.6948,
+      "step": 7
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 2.6495,
+      "step": 8
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 1.5e-05,
+      "loss": 2.594,
+      "step": 9
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 2.538,
+      "step": 10
+    },
+    {
+      "epoch": 11.0,
+      "learning_rate": 1.8333333333333333e-05,
+      "loss": 2.4672,
+      "step": 11
+    },
+    {
+      "epoch": 12.0,
+      "learning_rate": 2e-05,
+      "loss": 2.4293,
+      "step": 12
+    },
+    {
+      "epoch": 13.0,
+      "learning_rate": 2.1666666666666667e-05,
+      "loss": 2.3298,
+      "step": 13
+    },
+    {
+      "epoch": 14.0,
+      "learning_rate": 2.3333333333333336e-05,
+      "loss": 2.258,
+      "step": 14
+    },
+    {
+      "epoch": 15.0,
+      "learning_rate": 2.5e-05,
+      "loss": 2.1281,
+      "step": 15
+    },
+    {
+      "epoch": 16.0,
+      "learning_rate": 2.6666666666666667e-05,
+      "loss": 2.0277,
+      "step": 16
+    },
+    {
+      "epoch": 17.0,
+      "learning_rate": 2.8333333333333335e-05,
+      "loss": 1.9414,
+      "step": 17
+    },
+    {
+      "epoch": 18.0,
+      "learning_rate": 3e-05,
+      "loss": 1.8256,
+      "step": 18
+    },
+    {
+      "epoch": 19.0,
+      "learning_rate": 3.1666666666666666e-05,
+      "loss": 1.678,
+      "step": 19
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 1.5957,
+      "step": 20
+    },
+    {
+      "epoch": 20.0,
+      "eval_accuracy_NAT": 0.8321817496578309,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.0071490677229253214,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.0,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.0,
+      "eval_accuracy_speedway-bricks": 0.9400317341795781,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.8227595719506057,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.006510645785676579,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.0,
+      "eval_iou_red-bricks": 0.0,
+      "eval_iou_road": 0.0,
+      "eval_iou_rough-red-sidewalk": 0.0,
+      "eval_iou_speedway-bricks": 0.6827730558356707,
+      "eval_iou_steel": 0.0,
+      "eval_iou_steel-sidewalk": 0.0,
+      "eval_iou_tiles-marble": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.5208492279052734,
+      "eval_mean_accuracy": 0.2965604252600557,
+      "eval_mean_iou": 0.13745847941563208,
+      "eval_overall_accuracy": 0.7877578735351562,
+      "eval_runtime": 0.7158,
+      "eval_samples_per_second": 1.397,
+      "eval_steps_per_second": 1.397,
+      "step": 20
+    },
+    {
+      "epoch": 21.0,
+      "learning_rate": 3.5e-05,
+      "loss": 1.4548,
+      "step": 21
+    },
+    {
+      "epoch": 22.0,
+      "learning_rate": 3.6666666666666666e-05,
+      "loss": 1.3462,
+      "step": 22
+    },
+    {
+      "epoch": 23.0,
+      "learning_rate": 3.8333333333333334e-05,
+      "loss": 1.2189,
+      "step": 23
+    },
+    {
+      "epoch": 24.0,
+      "learning_rate": 4e-05,
+      "loss": 1.1207,
+      "step": 24
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 4.166666666666667e-05,
+      "loss": 0.9904,
+      "step": 25
+    },
+    {
+      "epoch": 26.0,
+      "learning_rate": 4.3333333333333334e-05,
+      "loss": 0.9286,
+      "step": 26
+    },
+    {
+      "epoch": 27.0,
+      "learning_rate": 4.5e-05,
+      "loss": 0.8425,
+      "step": 27
+    },
+    {
+      "epoch": 28.0,
+      "learning_rate": 4.666666666666667e-05,
+      "loss": 0.7906,
+      "step": 28
+    },
+    {
+      "epoch": 29.0,
+      "learning_rate": 4.8333333333333334e-05,
+      "loss": 0.7611,
+      "step": 29
+    },
+    {
+      "epoch": 30.0,
+      "learning_rate": 5e-05,
+      "loss": 0.6952,
+      "step": 30
+    },
+    {
+      "epoch": 31.0,
+      "learning_rate": 4.991228070175439e-05,
+      "loss": 0.6077,
+      "step": 31
+    },
+    {
+      "epoch": 32.0,
+      "learning_rate": 4.9824561403508773e-05,
+      "loss": 0.5674,
+      "step": 32
+    },
+    {
+      "epoch": 33.0,
+      "learning_rate": 4.973684210526316e-05,
+      "loss": 0.5183,
+      "step": 33
+    },
+    {
+      "epoch": 34.0,
+      "learning_rate": 4.9649122807017544e-05,
+      "loss": 0.4915,
+      "step": 34
+    },
+    {
+      "epoch": 35.0,
+      "learning_rate": 4.956140350877193e-05,
+      "loss": 0.4782,
+      "step": 35
+    },
+    {
+      "epoch": 36.0,
+      "learning_rate": 4.9473684210526315e-05,
+      "loss": 0.4367,
+      "step": 36
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 4.93859649122807e-05,
+      "loss": 0.4252,
+      "step": 37
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 4.9298245614035086e-05,
+      "loss": 0.3915,
+      "step": 38
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 4.921052631578947e-05,
+      "loss": 0.3803,
+      "step": 39
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 4.912280701754386e-05,
+      "loss": 0.3589,
+      "step": 40
+    },
+    {
+      "epoch": 40.0,
+      "eval_accuracy_NAT": 0.962933331570143,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.44266254468167326,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.14368837885889768,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.0,
+      "eval_accuracy_speedway-bricks": 1.0,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9404239931293225,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.34041604754829125,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.11037993543580829,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.0,
+      "eval_iou_speedway-bricks": 0.836688077155854,
+      "eval_iou_steel": 0.0,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.37735000252723694,
+      "eval_mean_accuracy": 0.424880709185119,
+      "eval_mean_iou": 0.37131800887821265,
+      "eval_overall_accuracy": 0.9033889770507812,
+      "eval_runtime": 0.5957,
+      "eval_samples_per_second": 1.679,
+      "eval_steps_per_second": 1.679,
+      "step": 40
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 4.903508771929825e-05,
+      "loss": 0.3388,
+      "step": 41
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 4.8947368421052635e-05,
+      "loss": 0.3296,
+      "step": 42
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 4.885964912280702e-05,
+      "loss": 0.3273,
+      "step": 43
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 4.8771929824561406e-05,
+      "loss": 0.2959,
+      "step": 44
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 4.868421052631579e-05,
+      "loss": 0.2805,
+      "step": 45
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 4.859649122807018e-05,
+      "loss": 0.2713,
+      "step": 46
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 4.850877192982456e-05,
+      "loss": 0.2579,
+      "step": 47
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 4.842105263157895e-05,
+      "loss": 0.2477,
+      "step": 48
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 4.8333333333333334e-05,
+      "loss": 0.2412,
+      "step": 49
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 4.824561403508772e-05,
+      "loss": 0.2234,
+      "step": 50
+    },
+    {
+      "epoch": 51.0,
+      "learning_rate": 4.8157894736842105e-05,
+      "loss": 0.2311,
+      "step": 51
+    },
+    {
+      "epoch": 52.0,
+      "learning_rate": 4.807017543859649e-05,
+      "loss": 0.2132,
+      "step": 52
+    },
+    {
+      "epoch": 53.0,
+      "learning_rate": 4.7982456140350876e-05,
+      "loss": 0.214,
+      "step": 53
+    },
+    {
+      "epoch": 54.0,
+      "learning_rate": 4.789473684210526e-05,
+      "loss": 0.2097,
+      "step": 54
+    },
+    {
+      "epoch": 55.0,
+      "learning_rate": 4.780701754385965e-05,
+      "loss": 0.2033,
+      "step": 55
+    },
+    {
+      "epoch": 56.0,
+      "learning_rate": 4.771929824561404e-05,
+      "loss": 0.192,
+      "step": 56
+    },
+    {
+      "epoch": 57.0,
+      "learning_rate": 4.7631578947368424e-05,
+      "loss": 0.1857,
+      "step": 57
+    },
+    {
+      "epoch": 58.0,
+      "learning_rate": 4.754385964912281e-05,
+      "loss": 0.1799,
+      "step": 58
+    },
+    {
+      "epoch": 59.0,
+      "learning_rate": 4.7456140350877195e-05,
+      "loss": 0.1864,
+      "step": 59
+    },
+    {
+      "epoch": 60.0,
+      "learning_rate": 4.736842105263158e-05,
+      "loss": 0.173,
+      "step": 60
+    },
+    {
+      "epoch": 60.0,
+      "eval_accuracy_NAT": 0.9537294781177061,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.6674717418606898,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5639243575238403,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.5100017702248185,
+      "eval_accuracy_speedway-bricks": 1.0,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.940778617689453,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.635193527627103,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.29731572219855135,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.4458372021046116,
+      "eval_iou_speedway-bricks": 0.8960722611941079,
+      "eval_iou_steel": 0.0,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.25648337602615356,
+      "eval_mean_accuracy": 0.6158545579545092,
+      "eval_mean_iou": 0.5358662218023046,
+      "eval_overall_accuracy": 0.9278640747070312,
+      "eval_runtime": 0.5992,
+      "eval_samples_per_second": 1.669,
+      "eval_steps_per_second": 1.669,
+      "step": 60
+    },
+    {
+      "epoch": 61.0,
+      "learning_rate": 4.728070175438597e-05,
+      "loss": 0.1641,
+      "step": 61
+    },
+    {
+      "epoch": 62.0,
+      "learning_rate": 4.719298245614036e-05,
+      "loss": 0.1643,
+      "step": 62
+    },
+    {
+      "epoch": 63.0,
+      "learning_rate": 4.7105263157894744e-05,
+      "loss": 0.1636,
+      "step": 63
+    },
+    {
+      "epoch": 64.0,
+      "learning_rate": 4.701754385964913e-05,
+      "loss": 0.16,
+      "step": 64
+    },
+    {
+      "epoch": 65.0,
+      "learning_rate": 4.6929824561403515e-05,
+      "loss": 0.1544,
+      "step": 65
+    },
+    {
+      "epoch": 66.0,
+      "learning_rate": 4.68421052631579e-05,
+      "loss": 0.1483,
+      "step": 66
+    },
+    {
+      "epoch": 67.0,
+      "learning_rate": 4.6754385964912286e-05,
+      "loss": 0.1445,
+      "step": 67
+    },
+    {
+      "epoch": 68.0,
+      "learning_rate": 4.666666666666667e-05,
+      "loss": 0.1532,
+      "step": 68
+    },
+    {
+      "epoch": 69.0,
+      "learning_rate": 4.657894736842106e-05,
+      "loss": 0.1446,
+      "step": 69
+    },
+    {
+      "epoch": 70.0,
+      "learning_rate": 4.649122807017544e-05,
+      "loss": 0.1374,
+      "step": 70
+    },
+    {
+      "epoch": 71.0,
+      "learning_rate": 4.640350877192983e-05,
+      "loss": 0.1345,
+      "step": 71
+    },
+    {
+      "epoch": 72.0,
+      "learning_rate": 4.6315789473684214e-05,
+      "loss": 0.1318,
+      "step": 72
+    },
+    {
+      "epoch": 73.0,
+      "learning_rate": 4.62280701754386e-05,
+      "loss": 0.1328,
+      "step": 73
+    },
+    {
+      "epoch": 74.0,
+      "learning_rate": 4.6140350877192985e-05,
+      "loss": 0.124,
+      "step": 74
+    },
+    {
+      "epoch": 75.0,
+      "learning_rate": 4.605263157894737e-05,
+      "loss": 0.1365,
+      "step": 75
+    },
+    {
+      "epoch": 76.0,
+      "learning_rate": 4.5964912280701756e-05,
+      "loss": 0.1256,
+      "step": 76
+    },
+    {
+      "epoch": 77.0,
+      "learning_rate": 4.587719298245614e-05,
+      "loss": 0.1216,
+      "step": 77
+    },
+    {
+      "epoch": 78.0,
+      "learning_rate": 4.5789473684210527e-05,
+      "loss": 0.124,
+      "step": 78
+    },
+    {
+      "epoch": 79.0,
+      "learning_rate": 4.570175438596491e-05,
+      "loss": 0.1213,
+      "step": 79
+    },
+    {
+      "epoch": 80.0,
+      "learning_rate": 4.56140350877193e-05,
+      "loss": 0.1134,
+      "step": 80
+    },
+    {
+      "epoch": 80.0,
+      "eval_accuracy_NAT": 0.9792516579498945,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8183750362283837,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.49070631970260226,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9412285360240751,
+      "eval_accuracy_speedway-bricks": 0.9946098562628337,
+      "eval_accuracy_steel": 0.5342876165113183,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9568551695621555,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7767284063818082,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.365739067582219,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.7858409695536506,
+      "eval_iou_speedway-bricks": 0.9649559685780907,
+      "eval_iou_steel": 0.47122724603640637,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.16272249817848206,
+      "eval_mean_accuracy": 0.7930765037798513,
+      "eval_mean_iou": 0.7202244712823885,
+      "eval_overall_accuracy": 0.9604721069335938,
+      "eval_runtime": 0.6081,
+      "eval_samples_per_second": 1.644,
+      "eval_steps_per_second": 1.644,
+      "step": 80
+    },
+    {
+      "epoch": 81.0,
+      "learning_rate": 4.552631578947369e-05,
+      "loss": 0.1129,
+      "step": 81
+    },
+    {
+      "epoch": 82.0,
+      "learning_rate": 4.5438596491228075e-05,
+      "loss": 0.1262,
+      "step": 82
+    },
+    {
+      "epoch": 83.0,
+      "learning_rate": 4.535087719298246e-05,
+      "loss": 0.1174,
+      "step": 83
+    },
+    {
+      "epoch": 84.0,
+      "learning_rate": 4.5263157894736846e-05,
+      "loss": 0.1124,
+      "step": 84
+    },
+    {
+      "epoch": 85.0,
+      "learning_rate": 4.517543859649123e-05,
+      "loss": 0.111,
+      "step": 85
+    },
+    {
+      "epoch": 86.0,
+      "learning_rate": 4.508771929824562e-05,
+      "loss": 0.1085,
+      "step": 86
+    },
+    {
+      "epoch": 87.0,
+      "learning_rate": 4.5e-05,
+      "loss": 0.105,
+      "step": 87
+    },
+    {
+      "epoch": 88.0,
+      "learning_rate": 4.491228070175439e-05,
+      "loss": 0.1071,
+      "step": 88
+    },
+    {
+      "epoch": 89.0,
+      "learning_rate": 4.4824561403508774e-05,
+      "loss": 0.1021,
+      "step": 89
+    },
+    {
+      "epoch": 90.0,
+      "learning_rate": 4.473684210526316e-05,
+      "loss": 0.099,
+      "step": 90
+    },
+    {
+      "epoch": 91.0,
+      "learning_rate": 4.4649122807017545e-05,
+      "loss": 0.1008,
+      "step": 91
+    },
+    {
+      "epoch": 92.0,
+      "learning_rate": 4.456140350877193e-05,
+      "loss": 0.0986,
+      "step": 92
+    },
+    {
+      "epoch": 93.0,
+      "learning_rate": 4.4473684210526316e-05,
+      "loss": 0.1006,
+      "step": 93
+    },
+    {
+      "epoch": 94.0,
+      "learning_rate": 4.43859649122807e-05,
+      "loss": 0.0982,
+      "step": 94
+    },
+    {
+      "epoch": 95.0,
+      "learning_rate": 4.429824561403509e-05,
+      "loss": 0.0949,
+      "step": 95
+    },
+    {
+      "epoch": 96.0,
+      "learning_rate": 4.421052631578947e-05,
+      "loss": 0.0995,
+      "step": 96
+    },
+    {
+      "epoch": 97.0,
+      "learning_rate": 4.412280701754386e-05,
+      "loss": 0.0963,
+      "step": 97
+    },
+    {
+      "epoch": 98.0,
+      "learning_rate": 4.403508771929824e-05,
+      "loss": 0.0944,
+      "step": 98
+    },
+    {
+      "epoch": 99.0,
+      "learning_rate": 4.394736842105263e-05,
+      "loss": 0.0924,
+      "step": 99
+    },
+    {
+      "epoch": 100.0,
+      "learning_rate": 4.3859649122807014e-05,
+      "loss": 0.0925,
+      "step": 100
+    },
+    {
+      "epoch": 100.0,
+      "eval_accuracy_NAT": 0.9773738602627594,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7627282388175055,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5282042993373202,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.8376703841387856,
+      "eval_accuracy_speedway-bricks": 0.9968965839089042,
+      "eval_accuracy_steel": 0.5838881491344874,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9558231113927488,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7395784543325526,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.37933836331979104,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.7238794554076794,
+      "eval_iou_speedway-bricks": 0.9531277886846332,
+      "eval_iou_steel": 0.5400246305418719,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.16368532180786133,
+      "eval_mean_accuracy": 0.7811269192666271,
+      "eval_mean_iou": 0.7152953006132129,
+      "eval_overall_accuracy": 0.9571609497070312,
+      "eval_runtime": 0.6356,
+      "eval_samples_per_second": 1.573,
+      "eval_steps_per_second": 1.573,
+      "step": 100
+    },
+    {
+      "epoch": 101.0,
+      "learning_rate": 4.37719298245614e-05,
+      "loss": 0.1101,
+      "step": 101
+    },
+    {
+      "epoch": 102.0,
+      "learning_rate": 4.368421052631579e-05,
+      "loss": 0.0919,
+      "step": 102
+    },
+    {
+      "epoch": 103.0,
+      "learning_rate": 4.359649122807018e-05,
+      "loss": 0.0923,
+      "step": 103
+    },
+    {
+      "epoch": 104.0,
+      "learning_rate": 4.350877192982456e-05,
+      "loss": 0.0908,
+      "step": 104
+    },
+    {
+      "epoch": 105.0,
+      "learning_rate": 4.342105263157895e-05,
+      "loss": 0.0903,
+      "step": 105
+    },
+    {
+      "epoch": 106.0,
+      "learning_rate": 4.3333333333333334e-05,
+      "loss": 0.088,
+      "step": 106
+    },
+    {
+      "epoch": 107.0,
+      "learning_rate": 4.324561403508772e-05,
+      "loss": 0.086,
+      "step": 107
+    },
+    {
+      "epoch": 108.0,
+      "learning_rate": 4.3157894736842105e-05,
+      "loss": 0.0894,
+      "step": 108
+    },
+    {
+      "epoch": 109.0,
+      "learning_rate": 4.30701754385965e-05,
+      "loss": 0.0861,
+      "step": 109
+    },
+    {
+      "epoch": 110.0,
+      "learning_rate": 4.298245614035088e-05,
+      "loss": 0.085,
+      "step": 110
+    },
+    {
+      "epoch": 111.0,
+      "learning_rate": 4.289473684210527e-05,
+      "loss": 0.0877,
+      "step": 111
+    },
+    {
+      "epoch": 112.0,
+      "learning_rate": 4.2807017543859654e-05,
+      "loss": 0.0873,
+      "step": 112
+    },
+    {
+      "epoch": 113.0,
+      "learning_rate": 4.271929824561404e-05,
+      "loss": 0.0799,
+      "step": 113
+    },
+    {
+      "epoch": 114.0,
+      "learning_rate": 4.2631578947368425e-05,
+      "loss": 0.0818,
+      "step": 114
+    },
+    {
+      "epoch": 115.0,
+      "learning_rate": 4.254385964912281e-05,
+      "loss": 0.079,
+      "step": 115
+    },
+    {
+      "epoch": 116.0,
+      "learning_rate": 4.2456140350877196e-05,
+      "loss": 0.0845,
+      "step": 116
+    },
+    {
+      "epoch": 117.0,
+      "learning_rate": 4.236842105263158e-05,
+      "loss": 0.0779,
+      "step": 117
+    },
+    {
+      "epoch": 118.0,
+      "learning_rate": 4.228070175438597e-05,
+      "loss": 0.0784,
+      "step": 118
+    },
+    {
+      "epoch": 119.0,
+      "learning_rate": 4.219298245614035e-05,
+      "loss": 0.0745,
+      "step": 119
+    },
+    {
+      "epoch": 120.0,
+      "learning_rate": 4.210526315789474e-05,
+      "loss": 0.076,
+      "step": 120
+    },
+    {
+      "epoch": 120.0,
+      "eval_accuracy_NAT": 0.9815856811314392,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8138344121340934,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5185065459835139,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9499026376349796,
+      "eval_accuracy_speedway-bricks": 0.9870496546574575,
+      "eval_accuracy_steel": 0.7936085219707057,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9593775445580386,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7564655172413793,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.39468503937007876,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.8212427303336394,
+      "eval_iou_speedway-bricks": 0.9676762593219563,
+      "eval_iou_steel": 0.6144329896907217,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14839401841163635,
+      "eval_mean_accuracy": 0.8407479089186981,
+      "eval_mean_iou": 0.7523133467526356,
+      "eval_overall_accuracy": 0.962982177734375,
+      "eval_runtime": 0.6102,
+      "eval_samples_per_second": 1.639,
+      "eval_steps_per_second": 1.639,
+      "step": 120
+    },
+    {
+      "epoch": 121.0,
+      "learning_rate": 4.201754385964912e-05,
+      "loss": 0.0739,
+      "step": 121
+    },
+    {
+      "epoch": 122.0,
+      "learning_rate": 4.1929824561403516e-05,
+      "loss": 0.0753,
+      "step": 122
+    },
+    {
+      "epoch": 123.0,
+      "learning_rate": 4.18421052631579e-05,
+      "loss": 0.0706,
+      "step": 123
+    },
+    {
+      "epoch": 124.0,
+      "learning_rate": 4.1754385964912287e-05,
+      "loss": 0.0751,
+      "step": 124
+    },
+    {
+      "epoch": 125.0,
+      "learning_rate": 4.166666666666667e-05,
+      "loss": 0.0737,
+      "step": 125
+    },
+    {
+      "epoch": 126.0,
+      "learning_rate": 4.157894736842106e-05,
+      "loss": 0.0704,
+      "step": 126
+    },
+    {
+      "epoch": 127.0,
+      "learning_rate": 4.149122807017544e-05,
+      "loss": 0.071,
+      "step": 127
+    },
+    {
+      "epoch": 128.0,
+      "learning_rate": 4.140350877192983e-05,
+      "loss": 0.07,
+      "step": 128
+    },
+    {
+      "epoch": 129.0,
+      "learning_rate": 4.1315789473684214e-05,
+      "loss": 0.0735,
+      "step": 129
+    },
+    {
+      "epoch": 130.0,
+      "learning_rate": 4.12280701754386e-05,
+      "loss": 0.0708,
+      "step": 130
+    },
+    {
+      "epoch": 131.0,
+      "learning_rate": 4.1140350877192985e-05,
+      "loss": 0.0688,
+      "step": 131
+    },
+    {
+      "epoch": 132.0,
+      "learning_rate": 4.105263157894737e-05,
+      "loss": 0.0671,
+      "step": 132
+    },
+    {
+      "epoch": 133.0,
+      "learning_rate": 4.0964912280701756e-05,
+      "loss": 0.0676,
+      "step": 133
+    },
+    {
+      "epoch": 134.0,
+      "learning_rate": 4.087719298245614e-05,
+      "loss": 0.0695,
+      "step": 134
+    },
+    {
+      "epoch": 135.0,
+      "learning_rate": 4.078947368421053e-05,
+      "loss": 0.0646,
+      "step": 135
+    },
+    {
+      "epoch": 136.0,
+      "learning_rate": 4.070175438596491e-05,
+      "loss": 0.0668,
+      "step": 136
+    },
+    {
+      "epoch": 137.0,
+      "learning_rate": 4.06140350877193e-05,
+      "loss": 0.0645,
+      "step": 137
+    },
+    {
+      "epoch": 138.0,
+      "learning_rate": 4.0526315789473684e-05,
+      "loss": 0.0634,
+      "step": 138
+    },
+    {
+      "epoch": 139.0,
+      "learning_rate": 4.043859649122807e-05,
+      "loss": 0.0658,
+      "step": 139
+    },
+    {
+      "epoch": 140.0,
+      "learning_rate": 4.0350877192982455e-05,
+      "loss": 0.0648,
+      "step": 140
+    },
+    {
+      "epoch": 140.0,
+      "eval_accuracy_NAT": 0.9782201916147076,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8271664573471162,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5430741878131566,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.8104089219330854,
+      "eval_accuracy_speedway-bricks": 0.9863496359902931,
+      "eval_accuracy_steel": 0.8022636484687083,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9579143654457516,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7443922796035473,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.3991921112035167,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.682468694096601,
+      "eval_iou_speedway-bricks": 0.9595922907538988,
+      "eval_iou_steel": 0.64907083221115,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15345287322998047,
+      "eval_mean_accuracy": 0.8245805071945113,
+      "eval_mean_iou": 0.732105095552411,
+      "eval_overall_accuracy": 0.9590110778808594,
+      "eval_runtime": 0.6185,
+      "eval_samples_per_second": 1.617,
+      "eval_steps_per_second": 1.617,
+      "step": 140
+    },
+    {
+      "epoch": 141.0,
+      "learning_rate": 4.026315789473684e-05,
+      "loss": 0.0662,
+      "step": 141
+    },
+    {
+      "epoch": 142.0,
+      "learning_rate": 4.017543859649123e-05,
+      "loss": 0.0637,
+      "step": 142
+    },
+    {
+      "epoch": 143.0,
+      "learning_rate": 4.008771929824562e-05,
+      "loss": 0.0635,
+      "step": 143
+    },
+    {
+      "epoch": 144.0,
+      "learning_rate": 4e-05,
+      "loss": 0.0642,
+      "step": 144
+    },
+    {
+      "epoch": 145.0,
+      "learning_rate": 3.991228070175439e-05,
+      "loss": 0.066,
+      "step": 145
+    },
+    {
+      "epoch": 146.0,
+      "learning_rate": 3.9824561403508774e-05,
+      "loss": 0.0606,
+      "step": 146
+    },
+    {
+      "epoch": 147.0,
+      "learning_rate": 3.973684210526316e-05,
+      "loss": 0.0631,
+      "step": 147
+    },
+    {
+      "epoch": 148.0,
+      "learning_rate": 3.9649122807017545e-05,
+      "loss": 0.0606,
+      "step": 148
+    },
+    {
+      "epoch": 149.0,
+      "learning_rate": 3.956140350877193e-05,
+      "loss": 0.0654,
+      "step": 149
+    },
+    {
+      "epoch": 150.0,
+      "learning_rate": 3.9473684210526316e-05,
+      "loss": 0.0629,
+      "step": 150
+    },
+    {
+      "epoch": 151.0,
+      "learning_rate": 3.93859649122807e-05,
+      "loss": 0.0631,
+      "step": 151
+    },
+    {
+      "epoch": 152.0,
+      "learning_rate": 3.929824561403509e-05,
+      "loss": 0.0602,
+      "step": 152
+    },
+    {
+      "epoch": 153.0,
+      "learning_rate": 3.921052631578947e-05,
+      "loss": 0.0584,
+      "step": 153
+    },
+    {
+      "epoch": 154.0,
+      "learning_rate": 3.912280701754386e-05,
+      "loss": 0.0581,
+      "step": 154
+    },
+    {
+      "epoch": 155.0,
+      "learning_rate": 3.9035087719298244e-05,
+      "loss": 0.0593,
+      "step": 155
+    },
+    {
+      "epoch": 156.0,
+      "learning_rate": 3.894736842105263e-05,
+      "loss": 0.0618,
+      "step": 156
+    },
+    {
+      "epoch": 157.0,
+      "learning_rate": 3.8859649122807015e-05,
+      "loss": 0.0569,
+      "step": 157
+    },
+    {
+      "epoch": 158.0,
+      "learning_rate": 3.877192982456141e-05,
+      "loss": 0.0634,
+      "step": 158
+    },
+    {
+      "epoch": 159.0,
+      "learning_rate": 3.868421052631579e-05,
+      "loss": 0.0595,
+      "step": 159
+    },
+    {
+      "epoch": 160.0,
+      "learning_rate": 3.859649122807018e-05,
+      "loss": 0.0647,
+      "step": 160
+    },
+    {
+      "epoch": 160.0,
+      "eval_accuracy_NAT": 0.9807062899610556,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7850449231958265,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5291740746727008,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.8890069038767924,
+      "eval_accuracy_speedway-bricks": 0.9944115176404704,
+      "eval_accuracy_steel": 0.7689747003994674,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9588402611674963,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7218619525628498,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.3935569178987859,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.7595281306715064,
+      "eval_iou_speedway-bricks": 0.9702105862265225,
+      "eval_iou_steel": 0.7059902200488998,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14740905165672302,
+      "eval_mean_accuracy": 0.8245530682910521,
+      "eval_mean_iou": 0.75166467809601,
+      "eval_overall_accuracy": 0.96240234375,
+      "eval_runtime": 0.6115,
+      "eval_samples_per_second": 1.635,
+      "eval_steps_per_second": 1.635,
+      "step": 160
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 3.8508771929824563e-05,
+      "loss": 0.0573,
+      "step": 161
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 3.842105263157895e-05,
+      "loss": 0.0591,
+      "step": 162
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 3.8333333333333334e-05,
+      "loss": 0.0588,
+      "step": 163
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 3.824561403508773e-05,
+      "loss": 0.0568,
+      "step": 164
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 3.815789473684211e-05,
+      "loss": 0.059,
+      "step": 165
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 3.80701754385965e-05,
+      "loss": 0.0551,
+      "step": 166
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 3.798245614035088e-05,
+      "loss": 0.0563,
+      "step": 167
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 3.789473684210527e-05,
+      "loss": 0.0543,
+      "step": 168
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 3.7807017543859654e-05,
+      "loss": 0.0549,
+      "step": 169
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 3.771929824561404e-05,
+      "loss": 0.0566,
+      "step": 170
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 3.7631578947368425e-05,
+      "loss": 0.0541,
+      "step": 171
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 3.754385964912281e-05,
+      "loss": 0.0531,
+      "step": 172
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 3.7456140350877196e-05,
+      "loss": 0.0549,
+      "step": 173
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 3.736842105263158e-05,
+      "loss": 0.0535,
+      "step": 174
+    },
+    {
+      "epoch": 175.0,
+      "learning_rate": 3.728070175438597e-05,
+      "loss": 0.0538,
+      "step": 175
+    },
+    {
+      "epoch": 176.0,
+      "learning_rate": 3.719298245614035e-05,
+      "loss": 0.0539,
+      "step": 176
+    },
+    {
+      "epoch": 177.0,
+      "learning_rate": 3.710526315789474e-05,
+      "loss": 0.0557,
+      "step": 177
+    },
+    {
+      "epoch": 178.0,
+      "learning_rate": 3.7017543859649124e-05,
+      "loss": 0.053,
+      "step": 178
+    },
+    {
+      "epoch": 179.0,
+      "learning_rate": 3.692982456140351e-05,
+      "loss": 0.0528,
+      "step": 179
+    },
+    {
+      "epoch": 180.0,
+      "learning_rate": 3.6842105263157895e-05,
+      "loss": 0.0536,
+      "step": 180
+    },
+    {
+      "epoch": 180.0,
+      "eval_accuracy_NAT": 0.9788946119107914,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8163462467394454,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5758849199935349,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9368029739776952,
+      "eval_accuracy_speedway-bricks": 0.9898263953705432,
+      "eval_accuracy_steel": 0.8608521970705726,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9596310532354142,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7403189066059226,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4216069104248018,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.7822616407982261,
+      "eval_iou_speedway-bricks": 0.9738739152394509,
+      "eval_iou_steel": 0.7420373027259685,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14183500409126282,
+      "eval_mean_accuracy": 0.8597678908437638,
+      "eval_mean_iou": 0.7699549548382972,
+      "eval_overall_accuracy": 0.9642791748046875,
+      "eval_runtime": 0.6075,
+      "eval_samples_per_second": 1.646,
+      "eval_steps_per_second": 1.646,
+      "step": 180
+    },
+    {
+      "epoch": 181.0,
+      "learning_rate": 3.675438596491228e-05,
+      "loss": 0.0539,
+      "step": 181
+    },
+    {
+      "epoch": 182.0,
+      "learning_rate": 3.6666666666666666e-05,
+      "loss": 0.0511,
+      "step": 182
+    },
+    {
+      "epoch": 183.0,
+      "learning_rate": 3.657894736842106e-05,
+      "loss": 0.0507,
+      "step": 183
+    },
+    {
+      "epoch": 184.0,
+      "learning_rate": 3.6491228070175443e-05,
+      "loss": 0.0494,
+      "step": 184
+    },
+    {
+      "epoch": 185.0,
+      "learning_rate": 3.640350877192983e-05,
+      "loss": 0.0525,
+      "step": 185
+    },
+    {
+      "epoch": 186.0,
+      "learning_rate": 3.6315789473684214e-05,
+      "loss": 0.0494,
+      "step": 186
+    },
+    {
+      "epoch": 187.0,
+      "learning_rate": 3.62280701754386e-05,
+      "loss": 0.0547,
+      "step": 187
+    },
+    {
+      "epoch": 188.0,
+      "learning_rate": 3.6140350877192985e-05,
+      "loss": 0.0499,
+      "step": 188
+    },
+    {
+      "epoch": 189.0,
+      "learning_rate": 3.605263157894737e-05,
+      "loss": 0.0527,
+      "step": 189
+    },
+    {
+      "epoch": 190.0,
+      "learning_rate": 3.5964912280701756e-05,
+      "loss": 0.0531,
+      "step": 190
+    },
+    {
+      "epoch": 191.0,
+      "learning_rate": 3.587719298245614e-05,
+      "loss": 0.049,
+      "step": 191
+    },
+    {
+      "epoch": 192.0,
+      "learning_rate": 3.578947368421053e-05,
+      "loss": 0.049,
+      "step": 192
+    },
+    {
+      "epoch": 193.0,
+      "learning_rate": 3.570175438596491e-05,
+      "loss": 0.0519,
+      "step": 193
+    },
+    {
+      "epoch": 194.0,
+      "learning_rate": 3.56140350877193e-05,
+      "loss": 0.0507,
+      "step": 194
+    },
+    {
+      "epoch": 195.0,
+      "learning_rate": 3.5526315789473684e-05,
+      "loss": 0.0503,
+      "step": 195
+    },
+    {
+      "epoch": 196.0,
+      "learning_rate": 3.543859649122807e-05,
+      "loss": 0.0495,
+      "step": 196
+    },
+    {
+      "epoch": 197.0,
+      "learning_rate": 3.5350877192982455e-05,
+      "loss": 0.0486,
+      "step": 197
+    },
+    {
+      "epoch": 198.0,
+      "learning_rate": 3.526315789473684e-05,
+      "loss": 0.0506,
+      "step": 198
+    },
+    {
+      "epoch": 199.0,
+      "learning_rate": 3.5175438596491226e-05,
+      "loss": 0.0477,
+      "step": 199
+    },
+    {
+      "epoch": 200.0,
+      "learning_rate": 3.508771929824561e-05,
+      "loss": 0.0508,
+      "step": 200
+    },
+    {
+      "epoch": 200.0,
+      "eval_accuracy_NAT": 0.9823460569554552,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8078446526905613,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5829966057863262,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9417596034696406,
+      "eval_accuracy_speedway-bricks": 0.9883096882583535,
+      "eval_accuracy_steel": 0.9037949400798935,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9617055156744581,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7411806417301897,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.42646015606526366,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.8077740662010325,
+      "eval_iou_speedway-bricks": 0.9782094067923831,
+      "eval_iou_steel": 0.7556359588087949,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13466748595237732,
+      "eval_mean_accuracy": 0.8678419245400385,
+      "eval_mean_iou": 0.778494290878687,
+      "eval_overall_accuracy": 0.9662055969238281,
+      "eval_runtime": 0.8563,
+      "eval_samples_per_second": 1.168,
+      "eval_steps_per_second": 1.168,
+      "step": 200
+    },
+    {
+      "epoch": 201.0,
+      "learning_rate": 3.5e-05,
+      "loss": 0.0484,
+      "step": 201
+    },
+    {
+      "epoch": 202.0,
+      "learning_rate": 3.491228070175438e-05,
+      "loss": 0.0475,
+      "step": 202
+    },
+    {
+      "epoch": 203.0,
+      "learning_rate": 3.482456140350877e-05,
+      "loss": 0.0486,
+      "step": 203
+    },
+    {
+      "epoch": 204.0,
+      "learning_rate": 3.473684210526316e-05,
+      "loss": 0.0474,
+      "step": 204
+    },
+    {
+      "epoch": 205.0,
+      "learning_rate": 3.4649122807017546e-05,
+      "loss": 0.0498,
+      "step": 205
+    },
+    {
+      "epoch": 206.0,
+      "learning_rate": 3.456140350877193e-05,
+      "loss": 0.0498,
+      "step": 206
+    },
+    {
+      "epoch": 207.0,
+      "learning_rate": 3.447368421052632e-05,
+      "loss": 0.0476,
+      "step": 207
+    },
+    {
+      "epoch": 208.0,
+      "learning_rate": 3.43859649122807e-05,
+      "loss": 0.0461,
+      "step": 208
+    },
+    {
+      "epoch": 209.0,
+      "learning_rate": 3.429824561403509e-05,
+      "loss": 0.0478,
+      "step": 209
+    },
+    {
+      "epoch": 210.0,
+      "learning_rate": 3.421052631578947e-05,
+      "loss": 0.0484,
+      "step": 210
+    },
+    {
+      "epoch": 211.0,
+      "learning_rate": 3.412280701754386e-05,
+      "loss": 0.0487,
+      "step": 211
+    },
+    {
+      "epoch": 212.0,
+      "learning_rate": 3.403508771929825e-05,
+      "loss": 0.0471,
+      "step": 212
+    },
+    {
+      "epoch": 213.0,
+      "learning_rate": 3.3947368421052636e-05,
+      "loss": 0.0449,
+      "step": 213
+    },
+    {
+      "epoch": 214.0,
+      "learning_rate": 3.385964912280702e-05,
+      "loss": 0.0452,
+      "step": 214
+    },
+    {
+      "epoch": 215.0,
+      "learning_rate": 3.377192982456141e-05,
+      "loss": 0.0441,
+      "step": 215
+    },
+    {
+      "epoch": 216.0,
+      "learning_rate": 3.368421052631579e-05,
+      "loss": 0.0469,
+      "step": 216
+    },
+    {
+      "epoch": 217.0,
+      "learning_rate": 3.359649122807018e-05,
+      "loss": 0.0465,
+      "step": 217
+    },
+    {
+      "epoch": 218.0,
+      "learning_rate": 3.3508771929824564e-05,
+      "loss": 0.0472,
+      "step": 218
+    },
+    {
+      "epoch": 219.0,
+      "learning_rate": 3.342105263157895e-05,
+      "loss": 0.0463,
+      "step": 219
+    },
+    {
+      "epoch": 220.0,
+      "learning_rate": 3.3333333333333335e-05,
+      "loss": 0.0452,
+      "step": 220
+    },
+    {
+      "epoch": 220.0,
+      "eval_accuracy_NAT": 0.9795690322068751,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7840788329629987,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5747535154355907,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7316339175075235,
+      "eval_accuracy_speedway-bricks": 0.9934431584842263,
+      "eval_accuracy_steel": 0.8824900133155792,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9598131567695039,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7316325610745515,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.42730112953616917,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.6137511137511138,
+      "eval_iou_speedway-bricks": 0.9608548956769993,
+      "eval_iou_steel": 0.7776474039307715,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14316219091415405,
+      "eval_mean_accuracy": 0.824328078318799,
+      "eval_mean_iou": 0.7451667101231849,
+      "eval_overall_accuracy": 0.9603767395019531,
+      "eval_runtime": 0.5943,
+      "eval_samples_per_second": 1.683,
+      "eval_steps_per_second": 1.683,
+      "step": 220
+    },
+    {
+      "epoch": 221.0,
+      "learning_rate": 3.324561403508772e-05,
+      "loss": 0.0462,
+      "step": 221
+    },
+    {
+      "epoch": 222.0,
+      "learning_rate": 3.3157894736842106e-05,
+      "loss": 0.0429,
+      "step": 222
+    },
+    {
+      "epoch": 223.0,
+      "learning_rate": 3.307017543859649e-05,
+      "loss": 0.0508,
+      "step": 223
+    },
+    {
+      "epoch": 224.0,
+      "learning_rate": 3.2982456140350884e-05,
+      "loss": 0.0438,
+      "step": 224
+    },
+    {
+      "epoch": 225.0,
+      "learning_rate": 3.289473684210527e-05,
+      "loss": 0.0434,
+      "step": 225
+    },
+    {
+      "epoch": 226.0,
+      "learning_rate": 3.2807017543859655e-05,
+      "loss": 0.0477,
+      "step": 226
+    },
+    {
+      "epoch": 227.0,
+      "learning_rate": 3.271929824561404e-05,
+      "loss": 0.0455,
+      "step": 227
+    },
+    {
+      "epoch": 228.0,
+      "learning_rate": 3.2631578947368426e-05,
+      "loss": 0.043,
+      "step": 228
+    },
+    {
+      "epoch": 229.0,
+      "learning_rate": 3.254385964912281e-05,
+      "loss": 0.045,
+      "step": 229
+    },
+    {
+      "epoch": 230.0,
+      "learning_rate": 3.24561403508772e-05,
+      "loss": 0.0473,
+      "step": 230
+    },
+    {
+      "epoch": 231.0,
+      "learning_rate": 3.236842105263158e-05,
+      "loss": 0.0418,
+      "step": 231
+    },
+    {
+      "epoch": 232.0,
+      "learning_rate": 3.228070175438597e-05,
+      "loss": 0.0452,
+      "step": 232
+    },
+    {
+      "epoch": 233.0,
+      "learning_rate": 3.219298245614035e-05,
+      "loss": 0.0453,
+      "step": 233
+    },
+    {
+      "epoch": 234.0,
+      "learning_rate": 3.210526315789474e-05,
+      "loss": 0.0428,
+      "step": 234
+    },
+    {
+      "epoch": 235.0,
+      "learning_rate": 3.2017543859649124e-05,
+      "loss": 0.0432,
+      "step": 235
+    },
+    {
+      "epoch": 236.0,
+      "learning_rate": 3.192982456140351e-05,
+      "loss": 0.0415,
+      "step": 236
+    },
+    {
+      "epoch": 237.0,
+      "learning_rate": 3.1842105263157895e-05,
+      "loss": 0.048,
+      "step": 237
+    },
+    {
+      "epoch": 238.0,
+      "learning_rate": 3.175438596491228e-05,
+      "loss": 0.0424,
+      "step": 238
+    },
+    {
+      "epoch": 239.0,
+      "learning_rate": 3.1666666666666666e-05,
+      "loss": 0.0445,
+      "step": 239
+    },
+    {
+      "epoch": 240.0,
+      "learning_rate": 3.157894736842105e-05,
+      "loss": 0.0417,
+      "step": 240
+    },
+    {
+      "epoch": 240.0,
+      "eval_accuracy_NAT": 0.9801508850113395,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7948024345473867,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.542104412477776,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9183926358647548,
+      "eval_accuracy_speedway-bricks": 0.9923697965279075,
+      "eval_accuracy_steel": 0.8778295605858855,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9585141120558662,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7367902561346946,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4059549745824256,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.760815368822408,
+      "eval_iou_speedway-bricks": 0.9740952817224003,
+      "eval_iou_steel": 0.784121320249777,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14111237227916718,
+      "eval_mean_accuracy": 0.8509416208358416,
+      "eval_mean_iou": 0.770048552261262,
+      "eval_overall_accuracy": 0.9639854431152344,
+      "eval_runtime": 0.6048,
+      "eval_samples_per_second": 1.654,
+      "eval_steps_per_second": 1.654,
+      "step": 240
+    },
+    {
+      "epoch": 241.0,
+      "learning_rate": 3.149122807017544e-05,
+      "loss": 0.0427,
+      "step": 241
+    },
+    {
+      "epoch": 242.0,
+      "learning_rate": 3.140350877192982e-05,
+      "loss": 0.0416,
+      "step": 242
+    },
+    {
+      "epoch": 243.0,
+      "learning_rate": 3.131578947368421e-05,
+      "loss": 0.0433,
+      "step": 243
+    },
+    {
+      "epoch": 244.0,
+      "learning_rate": 3.12280701754386e-05,
+      "loss": 0.0436,
+      "step": 244
+    },
+    {
+      "epoch": 245.0,
+      "learning_rate": 3.1140350877192986e-05,
+      "loss": 0.0406,
+      "step": 245
+    },
+    {
+      "epoch": 246.0,
+      "learning_rate": 3.105263157894737e-05,
+      "loss": 0.0436,
+      "step": 246
+    },
+    {
+      "epoch": 247.0,
+      "learning_rate": 3.096491228070176e-05,
+      "loss": 0.0435,
+      "step": 247
+    },
+    {
+      "epoch": 248.0,
+      "learning_rate": 3.087719298245614e-05,
+      "loss": 0.0419,
+      "step": 248
+    },
+    {
+      "epoch": 249.0,
+      "learning_rate": 3.078947368421053e-05,
+      "loss": 0.0415,
+      "step": 249
+    },
+    {
+      "epoch": 250.0,
+      "learning_rate": 3.0701754385964913e-05,
+      "loss": 0.0423,
+      "step": 250
+    },
+    {
+      "epoch": 251.0,
+      "learning_rate": 3.06140350877193e-05,
+      "loss": 0.0421,
+      "step": 251
+    },
+    {
+      "epoch": 252.0,
+      "learning_rate": 3.0526315789473684e-05,
+      "loss": 0.0437,
+      "step": 252
+    },
+    {
+      "epoch": 253.0,
+      "learning_rate": 3.0438596491228073e-05,
+      "loss": 0.0407,
+      "step": 253
+    },
+    {
+      "epoch": 254.0,
+      "learning_rate": 3.035087719298246e-05,
+      "loss": 0.0402,
+      "step": 254
+    },
+    {
+      "epoch": 255.0,
+      "learning_rate": 3.0263157894736844e-05,
+      "loss": 0.047,
+      "step": 255
+    },
+    {
+      "epoch": 256.0,
+      "learning_rate": 3.017543859649123e-05,
+      "loss": 0.0403,
+      "step": 256
+    },
+    {
+      "epoch": 257.0,
+      "learning_rate": 3.0087719298245615e-05,
+      "loss": 0.0415,
+      "step": 257
+    },
+    {
+      "epoch": 258.0,
+      "learning_rate": 3e-05,
+      "loss": 0.0492,
+      "step": 258
+    },
+    {
+      "epoch": 259.0,
+      "learning_rate": 2.9912280701754386e-05,
+      "loss": 0.0396,
+      "step": 259
+    },
+    {
+      "epoch": 260.0,
+      "learning_rate": 2.9824561403508772e-05,
+      "loss": 0.0417,
+      "step": 260
+    },
+    {
+      "epoch": 260.0,
+      "eval_accuracy_NAT": 0.9801773328660879,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7820500434740605,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5682883465330532,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7863338644007789,
+      "eval_accuracy_speedway-bricks": 0.9955198805301475,
+      "eval_accuracy_steel": 0.8162450066577897,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9602785425101215,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7258787661406025,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.42904209884075656,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.6579765960598429,
+      "eval_iou_speedway-bricks": 0.9637879232837102,
+      "eval_iou_steel": 0.7650546021840874,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14068955183029175,
+      "eval_mean_accuracy": 0.8214357457436531,
+      "eval_mean_iou": 0.7503364215031869,
+      "eval_overall_accuracy": 0.9615936279296875,
+      "eval_runtime": 0.6119,
+      "eval_samples_per_second": 1.634,
+      "eval_steps_per_second": 1.634,
+      "step": 260
+    },
+    {
+      "epoch": 261.0,
+      "learning_rate": 2.9736842105263157e-05,
+      "loss": 0.0406,
+      "step": 261
+    },
+    {
+      "epoch": 262.0,
+      "learning_rate": 2.9649122807017543e-05,
+      "loss": 0.0391,
+      "step": 262
+    },
+    {
+      "epoch": 263.0,
+      "learning_rate": 2.9561403508771928e-05,
+      "loss": 0.0411,
+      "step": 263
+    },
+    {
+      "epoch": 264.0,
+      "learning_rate": 2.9473684210526314e-05,
+      "loss": 0.0428,
+      "step": 264
+    },
+    {
+      "epoch": 265.0,
+      "learning_rate": 2.9385964912280706e-05,
+      "loss": 0.0408,
+      "step": 265
+    },
+    {
+      "epoch": 266.0,
+      "learning_rate": 2.929824561403509e-05,
+      "loss": 0.0427,
+      "step": 266
+    },
+    {
+      "epoch": 267.0,
+      "learning_rate": 2.9210526315789477e-05,
+      "loss": 0.0427,
+      "step": 267
+    },
+    {
+      "epoch": 268.0,
+      "learning_rate": 2.9122807017543863e-05,
+      "loss": 0.0425,
+      "step": 268
+    },
+    {
+      "epoch": 269.0,
+      "learning_rate": 2.9035087719298248e-05,
+      "loss": 0.039,
+      "step": 269
+    },
+    {
+      "epoch": 270.0,
+      "learning_rate": 2.8947368421052634e-05,
+      "loss": 0.0409,
+      "step": 270
+    },
+    {
+      "epoch": 271.0,
+      "learning_rate": 2.885964912280702e-05,
+      "loss": 0.0408,
+      "step": 271
+    },
+    {
+      "epoch": 272.0,
+      "learning_rate": 2.8771929824561404e-05,
+      "loss": 0.0401,
+      "step": 272
+    },
+    {
+      "epoch": 273.0,
+      "learning_rate": 2.868421052631579e-05,
+      "loss": 0.041,
+      "step": 273
+    },
+    {
+      "epoch": 274.0,
+      "learning_rate": 2.8596491228070175e-05,
+      "loss": 0.0389,
+      "step": 274
+    },
+    {
+      "epoch": 275.0,
+      "learning_rate": 2.850877192982456e-05,
+      "loss": 0.0409,
+      "step": 275
+    },
+    {
+      "epoch": 276.0,
+      "learning_rate": 2.842105263157895e-05,
+      "loss": 0.04,
+      "step": 276
+    },
+    {
+      "epoch": 277.0,
+      "learning_rate": 2.8333333333333335e-05,
+      "loss": 0.0378,
+      "step": 277
+    },
+    {
+      "epoch": 278.0,
+      "learning_rate": 2.824561403508772e-05,
+      "loss": 0.0381,
+      "step": 278
+    },
+    {
+      "epoch": 279.0,
+      "learning_rate": 2.8157894736842106e-05,
+      "loss": 0.0425,
+      "step": 279
+    },
+    {
+      "epoch": 280.0,
+      "learning_rate": 2.8070175438596492e-05,
+      "loss": 0.0421,
+      "step": 280
+    },
+    {
+      "epoch": 280.0,
+      "eval_accuracy_NAT": 0.9818898314610456,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7966380059897594,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5474381768223695,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.9183926358647548,
+      "eval_accuracy_speedway-bricks": 0.9910397610602949,
+      "eval_accuracy_steel": 0.9081225033288948,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9603388624826203,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.733825754204859,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.41650270536153466,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.7788620327278186,
+      "eval_iou_speedway-bricks": 0.9745643120202843,
+      "eval_iou_steel": 0.7868474185174502,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13642747700214386,
+      "eval_mean_accuracy": 0.8572534857545198,
+      "eval_mean_iou": 0.7751568475524279,
+      "eval_overall_accuracy": 0.9650993347167969,
+      "eval_runtime": 0.615,
+      "eval_samples_per_second": 1.626,
+      "eval_steps_per_second": 1.626,
+      "step": 280
+    },
+    {
+      "epoch": 281.0,
+      "learning_rate": 2.7982456140350877e-05,
+      "loss": 0.0377,
+      "step": 281
+    },
+    {
+      "epoch": 282.0,
+      "learning_rate": 2.7894736842105263e-05,
+      "loss": 0.0431,
+      "step": 282
+    },
+    {
+      "epoch": 283.0,
+      "learning_rate": 2.780701754385965e-05,
+      "loss": 0.0393,
+      "step": 283
+    },
+    {
+      "epoch": 284.0,
+      "learning_rate": 2.7719298245614034e-05,
+      "loss": 0.0389,
+      "step": 284
+    },
+    {
+      "epoch": 285.0,
+      "learning_rate": 2.7631578947368426e-05,
+      "loss": 0.0479,
+      "step": 285
+    },
+    {
+      "epoch": 286.0,
+      "learning_rate": 2.754385964912281e-05,
+      "loss": 0.0425,
+      "step": 286
+    },
+    {
+      "epoch": 287.0,
+      "learning_rate": 2.7456140350877197e-05,
+      "loss": 0.0394,
+      "step": 287
+    },
+    {
+      "epoch": 288.0,
+      "learning_rate": 2.7368421052631583e-05,
+      "loss": 0.0397,
+      "step": 288
+    },
+    {
+      "epoch": 289.0,
+      "learning_rate": 2.7280701754385968e-05,
+      "loss": 0.0383,
+      "step": 289
+    },
+    {
+      "epoch": 290.0,
+      "learning_rate": 2.7192982456140354e-05,
+      "loss": 0.0376,
+      "step": 290
+    },
+    {
+      "epoch": 291.0,
+      "learning_rate": 2.710526315789474e-05,
+      "loss": 0.0398,
+      "step": 291
+    },
+    {
+      "epoch": 292.0,
+      "learning_rate": 2.7017543859649125e-05,
+      "loss": 0.0398,
+      "step": 292
+    },
+    {
+      "epoch": 293.0,
+      "learning_rate": 2.692982456140351e-05,
+      "loss": 0.0398,
+      "step": 293
+    },
+    {
+      "epoch": 294.0,
+      "learning_rate": 2.6842105263157896e-05,
+      "loss": 0.0396,
+      "step": 294
+    },
+    {
+      "epoch": 295.0,
+      "learning_rate": 2.675438596491228e-05,
+      "loss": 0.0376,
+      "step": 295
+    },
+    {
+      "epoch": 296.0,
+      "learning_rate": 2.6666666666666667e-05,
+      "loss": 0.0389,
+      "step": 296
+    },
+    {
+      "epoch": 297.0,
+      "learning_rate": 2.6578947368421052e-05,
+      "loss": 0.0449,
+      "step": 297
+    },
+    {
+      "epoch": 298.0,
+      "learning_rate": 2.6491228070175438e-05,
+      "loss": 0.0389,
+      "step": 298
+    },
+    {
+      "epoch": 299.0,
+      "learning_rate": 2.6403508771929823e-05,
+      "loss": 0.0375,
+      "step": 299
+    },
+    {
+      "epoch": 300.0,
+      "learning_rate": 2.6315789473684212e-05,
+      "loss": 0.0369,
+      "step": 300
+    },
+    {
+      "epoch": 300.0,
+      "eval_accuracy_NAT": 0.9779491011035367,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7464979229059994,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.587683853240666,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7797840325721367,
+      "eval_accuracy_speedway-bricks": 0.9962432331528841,
+      "eval_accuracy_steel": 0.8372170439414115,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9588719611021069,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.702710076391415,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4253129020938121,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.6560917485850462,
+      "eval_iou_speedway-bricks": 0.9597401429663265,
+      "eval_iou_steel": 0.7791201982651796,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15013903379440308,
+      "eval_mean_accuracy": 0.8208958644861056,
+      "eval_mean_iou": 0.7469745049006478,
+      "eval_overall_accuracy": 0.9596977233886719,
+      "eval_runtime": 0.6056,
+      "eval_samples_per_second": 1.651,
+      "eval_steps_per_second": 1.651,
+      "step": 300
+    },
+    {
+      "epoch": 301.0,
+      "learning_rate": 2.6228070175438597e-05,
+      "loss": 0.0385,
+      "step": 301
+    },
+    {
+      "epoch": 302.0,
+      "learning_rate": 2.6140350877192983e-05,
+      "loss": 0.0384,
+      "step": 302
+    },
+    {
+      "epoch": 303.0,
+      "learning_rate": 2.605263157894737e-05,
+      "loss": 0.0375,
+      "step": 303
+    },
+    {
+      "epoch": 304.0,
+      "learning_rate": 2.5964912280701754e-05,
+      "loss": 0.0377,
+      "step": 304
+    },
+    {
+      "epoch": 305.0,
+      "learning_rate": 2.5877192982456143e-05,
+      "loss": 0.0371,
+      "step": 305
+    },
+    {
+      "epoch": 306.0,
+      "learning_rate": 2.578947368421053e-05,
+      "loss": 0.0361,
+      "step": 306
+    },
+    {
+      "epoch": 307.0,
+      "learning_rate": 2.5701754385964917e-05,
+      "loss": 0.0375,
+      "step": 307
+    },
+    {
+      "epoch": 308.0,
+      "learning_rate": 2.5614035087719303e-05,
+      "loss": 0.0358,
+      "step": 308
+    },
+    {
+      "epoch": 309.0,
+      "learning_rate": 2.5526315789473688e-05,
+      "loss": 0.0385,
+      "step": 309
+    },
+    {
+      "epoch": 310.0,
+      "learning_rate": 2.5438596491228074e-05,
+      "loss": 0.0373,
+      "step": 310
+    },
+    {
+      "epoch": 311.0,
+      "learning_rate": 2.535087719298246e-05,
+      "loss": 0.0392,
+      "step": 311
+    },
+    {
+      "epoch": 312.0,
+      "learning_rate": 2.5263157894736845e-05,
+      "loss": 0.0376,
+      "step": 312
+    },
+    {
+      "epoch": 313.0,
+      "learning_rate": 2.517543859649123e-05,
+      "loss": 0.0377,
+      "step": 313
+    },
+    {
+      "epoch": 314.0,
+      "learning_rate": 2.5087719298245616e-05,
+      "loss": 0.0365,
+      "step": 314
+    },
+    {
+      "epoch": 315.0,
+      "learning_rate": 2.5e-05,
+      "loss": 0.038,
+      "step": 315
+    },
+    {
+      "epoch": 316.0,
+      "learning_rate": 2.4912280701754387e-05,
+      "loss": 0.0374,
+      "step": 316
+    },
+    {
+      "epoch": 317.0,
+      "learning_rate": 2.4824561403508772e-05,
+      "loss": 0.0427,
+      "step": 317
+    },
+    {
+      "epoch": 318.0,
+      "learning_rate": 2.4736842105263158e-05,
+      "loss": 0.0382,
+      "step": 318
+    },
+    {
+      "epoch": 319.0,
+      "learning_rate": 2.4649122807017543e-05,
+      "loss": 0.0348,
+      "step": 319
+    },
+    {
+      "epoch": 320.0,
+      "learning_rate": 2.456140350877193e-05,
+      "loss": 0.0369,
+      "step": 320
+    },
+    {
+      "epoch": 320.0,
+      "eval_accuracy_NAT": 0.975753929159421,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7764467201236596,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.580733796670438,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7157018941405558,
+      "eval_accuracy_speedway-bricks": 0.9951115363076348,
+      "eval_accuracy_steel": 0.8531957390146472,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9570050063552178,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7215189873417721,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.41662801484230055,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5929021850711248,
+      "eval_iou_speedway-bricks": 0.9568646368552133,
+      "eval_iou_steel": 0.7859552284575284,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15667399764060974,
+      "eval_mean_accuracy": 0.8161572692360594,
+      "eval_mean_iou": 0.7384790098205262,
+      "eval_overall_accuracy": 0.9578819274902344,
+      "eval_runtime": 0.6106,
+      "eval_samples_per_second": 1.638,
+      "eval_steps_per_second": 1.638,
+      "step": 320
+    },
+    {
+      "epoch": 321.0,
+      "learning_rate": 2.4473684210526318e-05,
+      "loss": 0.0382,
+      "step": 321
+    },
+    {
+      "epoch": 322.0,
+      "learning_rate": 2.4385964912280703e-05,
+      "loss": 0.0373,
+      "step": 322
+    },
+    {
+      "epoch": 323.0,
+      "learning_rate": 2.429824561403509e-05,
+      "loss": 0.0369,
+      "step": 323
+    },
+    {
+      "epoch": 324.0,
+      "learning_rate": 2.4210526315789474e-05,
+      "loss": 0.0355,
+      "step": 324
+    },
+    {
+      "epoch": 325.0,
+      "learning_rate": 2.412280701754386e-05,
+      "loss": 0.0342,
+      "step": 325
+    },
+    {
+      "epoch": 326.0,
+      "learning_rate": 2.4035087719298245e-05,
+      "loss": 0.0353,
+      "step": 326
+    },
+    {
+      "epoch": 327.0,
+      "learning_rate": 2.394736842105263e-05,
+      "loss": 0.0394,
+      "step": 327
+    },
+    {
+      "epoch": 328.0,
+      "learning_rate": 2.385964912280702e-05,
+      "loss": 0.0365,
+      "step": 328
+    },
+    {
+      "epoch": 329.0,
+      "learning_rate": 2.3771929824561405e-05,
+      "loss": 0.0362,
+      "step": 329
+    },
+    {
+      "epoch": 330.0,
+      "learning_rate": 2.368421052631579e-05,
+      "loss": 0.0369,
+      "step": 330
+    },
+    {
+      "epoch": 331.0,
+      "learning_rate": 2.359649122807018e-05,
+      "loss": 0.0374,
+      "step": 331
+    },
+    {
+      "epoch": 332.0,
+      "learning_rate": 2.3508771929824565e-05,
+      "loss": 0.0359,
+      "step": 332
+    },
+    {
+      "epoch": 333.0,
+      "learning_rate": 2.342105263157895e-05,
+      "loss": 0.0376,
+      "step": 333
+    },
+    {
+      "epoch": 334.0,
+      "learning_rate": 2.3333333333333336e-05,
+      "loss": 0.0372,
+      "step": 334
+    },
+    {
+      "epoch": 335.0,
+      "learning_rate": 2.324561403508772e-05,
+      "loss": 0.0368,
+      "step": 335
+    },
+    {
+      "epoch": 336.0,
+      "learning_rate": 2.3157894736842107e-05,
+      "loss": 0.0376,
+      "step": 336
+    },
+    {
+      "epoch": 337.0,
+      "learning_rate": 2.3070175438596492e-05,
+      "loss": 0.0372,
+      "step": 337
+    },
+    {
+      "epoch": 338.0,
+      "learning_rate": 2.2982456140350878e-05,
+      "loss": 0.0344,
+      "step": 338
+    },
+    {
+      "epoch": 339.0,
+      "learning_rate": 2.2894736842105263e-05,
+      "loss": 0.0365,
+      "step": 339
+    },
+    {
+      "epoch": 340.0,
+      "learning_rate": 2.280701754385965e-05,
+      "loss": 0.0347,
+      "step": 340
+    },
+    {
+      "epoch": 340.0,
+      "eval_accuracy_NAT": 0.9769903663689079,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7997294947348083,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5802489090027477,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7084439723844929,
+      "eval_accuracy_speedway-bricks": 0.9928014747059921,
+      "eval_accuracy_steel": 0.9101198402130493,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9581431239300721,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7343209438481327,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.43149038461538464,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.583126912428967,
+      "eval_iou_speedway-bricks": 0.9585468882005069,
+      "eval_iou_steel": 0.8086364980774918,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1487768143415451,
+      "eval_mean_accuracy": 0.8280556762349996,
+      "eval_mean_iou": 0.7457107918500925,
+      "eval_overall_accuracy": 0.9592437744140625,
+      "eval_runtime": 0.6021,
+      "eval_samples_per_second": 1.661,
+      "eval_steps_per_second": 1.661,
+      "step": 340
+    },
+    {
+      "epoch": 341.0,
+      "learning_rate": 2.2719298245614038e-05,
+      "loss": 0.0342,
+      "step": 341
+    },
+    {
+      "epoch": 342.0,
+      "learning_rate": 2.2631578947368423e-05,
+      "loss": 0.0358,
+      "step": 342
+    },
+    {
+      "epoch": 343.0,
+      "learning_rate": 2.254385964912281e-05,
+      "loss": 0.0357,
+      "step": 343
+    },
+    {
+      "epoch": 344.0,
+      "learning_rate": 2.2456140350877194e-05,
+      "loss": 0.0352,
+      "step": 344
+    },
+    {
+      "epoch": 345.0,
+      "learning_rate": 2.236842105263158e-05,
+      "loss": 0.0348,
+      "step": 345
+    },
+    {
+      "epoch": 346.0,
+      "learning_rate": 2.2280701754385965e-05,
+      "loss": 0.0344,
+      "step": 346
+    },
+    {
+      "epoch": 347.0,
+      "learning_rate": 2.219298245614035e-05,
+      "loss": 0.0375,
+      "step": 347
+    },
+    {
+      "epoch": 348.0,
+      "learning_rate": 2.2105263157894736e-05,
+      "loss": 0.0356,
+      "step": 348
+    },
+    {
+      "epoch": 349.0,
+      "learning_rate": 2.201754385964912e-05,
+      "loss": 0.034,
+      "step": 349
+    },
+    {
+      "epoch": 350.0,
+      "learning_rate": 2.1929824561403507e-05,
+      "loss": 0.0356,
+      "step": 350
+    },
+    {
+      "epoch": 351.0,
+      "learning_rate": 2.1842105263157896e-05,
+      "loss": 0.0365,
+      "step": 351
+    },
+    {
+      "epoch": 352.0,
+      "learning_rate": 2.175438596491228e-05,
+      "loss": 0.0335,
+      "step": 352
+    },
+    {
+      "epoch": 353.0,
+      "learning_rate": 2.1666666666666667e-05,
+      "loss": 0.0355,
+      "step": 353
+    },
+    {
+      "epoch": 354.0,
+      "learning_rate": 2.1578947368421053e-05,
+      "loss": 0.0341,
+      "step": 354
+    },
+    {
+      "epoch": 355.0,
+      "learning_rate": 2.149122807017544e-05,
+      "loss": 0.0337,
+      "step": 355
+    },
+    {
+      "epoch": 356.0,
+      "learning_rate": 2.1403508771929827e-05,
+      "loss": 0.0351,
+      "step": 356
+    },
+    {
+      "epoch": 357.0,
+      "learning_rate": 2.1315789473684212e-05,
+      "loss": 0.0424,
+      "step": 357
+    },
+    {
+      "epoch": 358.0,
+      "learning_rate": 2.1228070175438598e-05,
+      "loss": 0.0353,
+      "step": 358
+    },
+    {
+      "epoch": 359.0,
+      "learning_rate": 2.1140350877192983e-05,
+      "loss": 0.0346,
+      "step": 359
+    },
+    {
+      "epoch": 360.0,
+      "learning_rate": 2.105263157894737e-05,
+      "loss": 0.0341,
+      "step": 360
+    },
+    {
+      "epoch": 360.0,
+      "eval_accuracy_NAT": 0.9783259830337012,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7635977200270505,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5854210441247778,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.675163745795716,
+      "eval_accuracy_speedway-bricks": 0.9957065521747247,
+      "eval_accuracy_steel": 0.8528628495339547,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9590238843698351,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7182189913675602,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.433409118104583,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5656236096692867,
+      "eval_iou_speedway-bricks": 0.953979946568896,
+      "eval_iou_steel": 0.7936802973977695,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15374624729156494,
+      "eval_mean_accuracy": 0.8085129824483209,
+      "eval_mean_iou": 0.7373226412463217,
+      "eval_overall_accuracy": 0.9582862854003906,
+      "eval_runtime": 0.5974,
+      "eval_samples_per_second": 1.674,
+      "eval_steps_per_second": 1.674,
+      "step": 360
+    },
+    {
+      "epoch": 361.0,
+      "learning_rate": 2.0964912280701758e-05,
+      "loss": 0.034,
+      "step": 361
+    },
+    {
+      "epoch": 362.0,
+      "learning_rate": 2.0877192982456143e-05,
+      "loss": 0.0355,
+      "step": 362
+    },
+    {
+      "epoch": 363.0,
+      "learning_rate": 2.078947368421053e-05,
+      "loss": 0.0341,
+      "step": 363
+    },
+    {
+      "epoch": 364.0,
+      "learning_rate": 2.0701754385964914e-05,
+      "loss": 0.0339,
+      "step": 364
+    },
+    {
+      "epoch": 365.0,
+      "learning_rate": 2.06140350877193e-05,
+      "loss": 0.0346,
+      "step": 365
+    },
+    {
+      "epoch": 366.0,
+      "learning_rate": 2.0526315789473685e-05,
+      "loss": 0.0363,
+      "step": 366
+    },
+    {
+      "epoch": 367.0,
+      "learning_rate": 2.043859649122807e-05,
+      "loss": 0.0351,
+      "step": 367
+    },
+    {
+      "epoch": 368.0,
+      "learning_rate": 2.0350877192982456e-05,
+      "loss": 0.034,
+      "step": 368
+    },
+    {
+      "epoch": 369.0,
+      "learning_rate": 2.0263157894736842e-05,
+      "loss": 0.0338,
+      "step": 369
+    },
+    {
+      "epoch": 370.0,
+      "learning_rate": 2.0175438596491227e-05,
+      "loss": 0.0349,
+      "step": 370
+    },
+    {
+      "epoch": 371.0,
+      "learning_rate": 2.0087719298245616e-05,
+      "loss": 0.0326,
+      "step": 371
+    },
+    {
+      "epoch": 372.0,
+      "learning_rate": 2e-05,
+      "loss": 0.0364,
+      "step": 372
+    },
+    {
+      "epoch": 373.0,
+      "learning_rate": 1.9912280701754387e-05,
+      "loss": 0.0359,
+      "step": 373
+    },
+    {
+      "epoch": 374.0,
+      "learning_rate": 1.9824561403508773e-05,
+      "loss": 0.0331,
+      "step": 374
+    },
+    {
+      "epoch": 375.0,
+      "learning_rate": 1.9736842105263158e-05,
+      "loss": 0.0338,
+      "step": 375
+    },
+    {
+      "epoch": 376.0,
+      "learning_rate": 1.9649122807017544e-05,
+      "loss": 0.0338,
+      "step": 376
+    },
+    {
+      "epoch": 377.0,
+      "learning_rate": 1.956140350877193e-05,
+      "loss": 0.0339,
+      "step": 377
+    },
+    {
+      "epoch": 378.0,
+      "learning_rate": 1.9473684210526315e-05,
+      "loss": 0.0327,
+      "step": 378
+    },
+    {
+      "epoch": 379.0,
+      "learning_rate": 1.9385964912280704e-05,
+      "loss": 0.0333,
+      "step": 379
+    },
+    {
+      "epoch": 380.0,
+      "learning_rate": 1.929824561403509e-05,
+      "loss": 0.0319,
+      "step": 380
+    },
+    {
+      "epoch": 380.0,
+      "eval_accuracy_NAT": 0.9768911869136014,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7931600811515795,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5952804267011476,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.693220038944946,
+      "eval_accuracy_speedway-bricks": 0.9944115176404704,
+      "eval_accuracy_steel": 0.8794940079893475,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9584684848327581,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7336252345634885,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4352398960056724,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5835195946952764,
+      "eval_iou_speedway-bricks": 0.9570612079903882,
+      "eval_iou_steel": 0.8032836728488902,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1511119306087494,
+      "eval_mean_accuracy": 0.8220762098901822,
+      "eval_mean_iou": 0.7451996818227457,
+      "eval_overall_accuracy": 0.9591293334960938,
+      "eval_runtime": 0.6088,
+      "eval_samples_per_second": 1.643,
+      "eval_steps_per_second": 1.643,
+      "step": 380
+    },
+    {
+      "epoch": 381.0,
+      "learning_rate": 1.9210526315789474e-05,
+      "loss": 0.0328,
+      "step": 381
+    },
+    {
+      "epoch": 382.0,
+      "learning_rate": 1.9122807017543863e-05,
+      "loss": 0.0342,
+      "step": 382
+    },
+    {
+      "epoch": 383.0,
+      "learning_rate": 1.903508771929825e-05,
+      "loss": 0.0329,
+      "step": 383
+    },
+    {
+      "epoch": 384.0,
+      "learning_rate": 1.8947368421052634e-05,
+      "loss": 0.033,
+      "step": 384
+    },
+    {
+      "epoch": 385.0,
+      "learning_rate": 1.885964912280702e-05,
+      "loss": 0.0328,
+      "step": 385
+    },
+    {
+      "epoch": 386.0,
+      "learning_rate": 1.8771929824561405e-05,
+      "loss": 0.0324,
+      "step": 386
+    },
+    {
+      "epoch": 387.0,
+      "learning_rate": 1.868421052631579e-05,
+      "loss": 0.0335,
+      "step": 387
+    },
+    {
+      "epoch": 388.0,
+      "learning_rate": 1.8596491228070176e-05,
+      "loss": 0.0355,
+      "step": 388
+    },
+    {
+      "epoch": 389.0,
+      "learning_rate": 1.8508771929824562e-05,
+      "loss": 0.033,
+      "step": 389
+    },
+    {
+      "epoch": 390.0,
+      "learning_rate": 1.8421052631578947e-05,
+      "loss": 0.034,
+      "step": 390
+    },
+    {
+      "epoch": 391.0,
+      "learning_rate": 1.8333333333333333e-05,
+      "loss": 0.0361,
+      "step": 391
+    },
+    {
+      "epoch": 392.0,
+      "learning_rate": 1.8245614035087722e-05,
+      "loss": 0.0338,
+      "step": 392
+    },
+    {
+      "epoch": 393.0,
+      "learning_rate": 1.8157894736842107e-05,
+      "loss": 0.04,
+      "step": 393
+    },
+    {
+      "epoch": 394.0,
+      "learning_rate": 1.8070175438596493e-05,
+      "loss": 0.033,
+      "step": 394
+    },
+    {
+      "epoch": 395.0,
+      "learning_rate": 1.7982456140350878e-05,
+      "loss": 0.0326,
+      "step": 395
+    },
+    {
+      "epoch": 396.0,
+      "learning_rate": 1.7894736842105264e-05,
+      "loss": 0.0316,
+      "step": 396
+    },
+    {
+      "epoch": 397.0,
+      "learning_rate": 1.780701754385965e-05,
+      "loss": 0.0341,
+      "step": 397
+    },
+    {
+      "epoch": 398.0,
+      "learning_rate": 1.7719298245614035e-05,
+      "loss": 0.0361,
+      "step": 398
+    },
+    {
+      "epoch": 399.0,
+      "learning_rate": 1.763157894736842e-05,
+      "loss": 0.0331,
+      "step": 399
+    },
+    {
+      "epoch": 400.0,
+      "learning_rate": 1.7543859649122806e-05,
+      "loss": 0.0331,
+      "step": 400
+    },
+    {
+      "epoch": 400.0,
+      "eval_accuracy_NAT": 0.9766399322934919,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8053328180852092,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5676418296427994,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6640113294388388,
+      "eval_accuracy_speedway-bricks": 0.994819861862983,
+      "eval_accuracy_steel": 0.8808255659121171,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9572718258468836,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7440864054271177,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4272506082725061,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5538984051978736,
+      "eval_iou_speedway-bricks": 0.9548167474776883,
+      "eval_iou_steel": 0.8096695226438189,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1568463146686554,
+      "eval_mean_accuracy": 0.8148785562059064,
+      "eval_mean_iou": 0.7411655858109812,
+      "eval_overall_accuracy": 0.9583320617675781,
+      "eval_runtime": 0.7091,
+      "eval_samples_per_second": 1.41,
+      "eval_steps_per_second": 1.41,
+      "step": 400
+    },
+    {
+      "epoch": 401.0,
+      "learning_rate": 1.745614035087719e-05,
+      "loss": 0.0328,
+      "step": 401
+    },
+    {
+      "epoch": 402.0,
+      "learning_rate": 1.736842105263158e-05,
+      "loss": 0.0335,
+      "step": 402
+    },
+    {
+      "epoch": 403.0,
+      "learning_rate": 1.7280701754385966e-05,
+      "loss": 0.034,
+      "step": 403
+    },
+    {
+      "epoch": 404.0,
+      "learning_rate": 1.719298245614035e-05,
+      "loss": 0.0324,
+      "step": 404
+    },
+    {
+      "epoch": 405.0,
+      "learning_rate": 1.7105263157894737e-05,
+      "loss": 0.0336,
+      "step": 405
+    },
+    {
+      "epoch": 406.0,
+      "learning_rate": 1.7017543859649125e-05,
+      "loss": 0.0307,
+      "step": 406
+    },
+    {
+      "epoch": 407.0,
+      "learning_rate": 1.692982456140351e-05,
+      "loss": 0.0315,
+      "step": 407
+    },
+    {
+      "epoch": 408.0,
+      "learning_rate": 1.6842105263157896e-05,
+      "loss": 0.0307,
+      "step": 408
+    },
+    {
+      "epoch": 409.0,
+      "learning_rate": 1.6754385964912282e-05,
+      "loss": 0.0337,
+      "step": 409
+    },
+    {
+      "epoch": 410.0,
+      "learning_rate": 1.6666666666666667e-05,
+      "loss": 0.0343,
+      "step": 410
+    },
+    {
+      "epoch": 411.0,
+      "learning_rate": 1.6578947368421053e-05,
+      "loss": 0.032,
+      "step": 411
+    },
+    {
+      "epoch": 412.0,
+      "learning_rate": 1.6491228070175442e-05,
+      "loss": 0.0345,
+      "step": 412
+    },
+    {
+      "epoch": 413.0,
+      "learning_rate": 1.6403508771929827e-05,
+      "loss": 0.0333,
+      "step": 413
+    },
+    {
+      "epoch": 414.0,
+      "learning_rate": 1.6315789473684213e-05,
+      "loss": 0.0338,
+      "step": 414
+    },
+    {
+      "epoch": 415.0,
+      "learning_rate": 1.62280701754386e-05,
+      "loss": 0.033,
+      "step": 415
+    },
+    {
+      "epoch": 416.0,
+      "learning_rate": 1.6140350877192984e-05,
+      "loss": 0.0332,
+      "step": 416
+    },
+    {
+      "epoch": 417.0,
+      "learning_rate": 1.605263157894737e-05,
+      "loss": 0.033,
+      "step": 417
+    },
+    {
+      "epoch": 418.0,
+      "learning_rate": 1.5964912280701755e-05,
+      "loss": 0.0324,
+      "step": 418
+    },
+    {
+      "epoch": 419.0,
+      "learning_rate": 1.587719298245614e-05,
+      "loss": 0.0348,
+      "step": 419
+    },
+    {
+      "epoch": 420.0,
+      "learning_rate": 1.5789473684210526e-05,
+      "loss": 0.0322,
+      "step": 420
+    },
+    {
+      "epoch": 420.0,
+      "eval_accuracy_NAT": 0.9766928280029886,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7880398029175925,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5846128980119606,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6909187466808284,
+      "eval_accuracy_speedway-bricks": 0.9954615456412171,
+      "eval_accuracy_steel": 0.8665113182423435,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9580625494545407,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7342695112071294,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4321902258334329,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.57044723764981,
+      "eval_iou_speedway-bricks": 0.9562357107634375,
+      "eval_iou_steel": 0.8036430997221364,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1547592282295227,
+      "eval_mean_accuracy": 0.8170395232494884,
+      "eval_mean_iou": 0.7424747224384145,
+      "eval_overall_accuracy": 0.9587059020996094,
+      "eval_runtime": 0.59,
+      "eval_samples_per_second": 1.695,
+      "eval_steps_per_second": 1.695,
+      "step": 420
+    },
+    {
+      "epoch": 421.0,
+      "learning_rate": 1.570175438596491e-05,
+      "loss": 0.0352,
+      "step": 421
+    },
+    {
+      "epoch": 422.0,
+      "learning_rate": 1.56140350877193e-05,
+      "loss": 0.0334,
+      "step": 422
+    },
+    {
+      "epoch": 423.0,
+      "learning_rate": 1.5526315789473686e-05,
+      "loss": 0.0325,
+      "step": 423
+    },
+    {
+      "epoch": 424.0,
+      "learning_rate": 1.543859649122807e-05,
+      "loss": 0.0325,
+      "step": 424
+    },
+    {
+      "epoch": 425.0,
+      "learning_rate": 1.5350877192982457e-05,
+      "loss": 0.0304,
+      "step": 425
+    },
+    {
+      "epoch": 426.0,
+      "learning_rate": 1.5263157894736842e-05,
+      "loss": 0.032,
+      "step": 426
+    },
+    {
+      "epoch": 427.0,
+      "learning_rate": 1.517543859649123e-05,
+      "loss": 0.0324,
+      "step": 427
+    },
+    {
+      "epoch": 428.0,
+      "learning_rate": 1.5087719298245615e-05,
+      "loss": 0.0328,
+      "step": 428
+    },
+    {
+      "epoch": 429.0,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0305,
+      "step": 429
+    },
+    {
+      "epoch": 430.0,
+      "learning_rate": 1.4912280701754386e-05,
+      "loss": 0.0301,
+      "step": 430
+    },
+    {
+      "epoch": 431.0,
+      "learning_rate": 1.4824561403508771e-05,
+      "loss": 0.0338,
+      "step": 431
+    },
+    {
+      "epoch": 432.0,
+      "learning_rate": 1.4736842105263157e-05,
+      "loss": 0.031,
+      "step": 432
+    },
+    {
+      "epoch": 433.0,
+      "learning_rate": 1.4649122807017546e-05,
+      "loss": 0.0318,
+      "step": 433
+    },
+    {
+      "epoch": 434.0,
+      "learning_rate": 1.4561403508771931e-05,
+      "loss": 0.0326,
+      "step": 434
+    },
+    {
+      "epoch": 435.0,
+      "learning_rate": 1.4473684210526317e-05,
+      "loss": 0.0305,
+      "step": 435
+    },
+    {
+      "epoch": 436.0,
+      "learning_rate": 1.4385964912280702e-05,
+      "loss": 0.0316,
+      "step": 436
+    },
+    {
+      "epoch": 437.0,
+      "learning_rate": 1.4298245614035088e-05,
+      "loss": 0.0335,
+      "step": 437
+    },
+    {
+      "epoch": 438.0,
+      "learning_rate": 1.4210526315789475e-05,
+      "loss": 0.0311,
+      "step": 438
+    },
+    {
+      "epoch": 439.0,
+      "learning_rate": 1.412280701754386e-05,
+      "loss": 0.0321,
+      "step": 439
+    },
+    {
+      "epoch": 440.0,
+      "learning_rate": 1.4035087719298246e-05,
+      "loss": 0.0313,
+      "step": 440
+    },
+    {
+      "epoch": 440.0,
+      "eval_accuracy_NAT": 0.9766928280029886,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7942227804076901,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5855826733473412,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6657815542573907,
+      "eval_accuracy_speedway-bricks": 0.9953448758633564,
+      "eval_accuracy_steel": 0.8761651131824234,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9582427847657846,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7368468226225688,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4292654028436019,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5576809015421115,
+      "eval_iou_speedway-bricks": 0.9554704386878563,
+      "eval_iou_steel": 0.8125964803951837,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1580827534198761,
+      "eval_mean_accuracy": 0.8156316375101985,
+      "eval_mean_iou": 0.7416838051428511,
+      "eval_overall_accuracy": 0.9585037231445312,
+      "eval_runtime": 0.6236,
+      "eval_samples_per_second": 1.603,
+      "eval_steps_per_second": 1.603,
+      "step": 440
+    },
+    {
+      "epoch": 441.0,
+      "learning_rate": 1.3947368421052631e-05,
+      "loss": 0.0313,
+      "step": 441
+    },
+    {
+      "epoch": 442.0,
+      "learning_rate": 1.3859649122807017e-05,
+      "loss": 0.0332,
+      "step": 442
+    },
+    {
+      "epoch": 443.0,
+      "learning_rate": 1.3771929824561406e-05,
+      "loss": 0.0309,
+      "step": 443
+    },
+    {
+      "epoch": 444.0,
+      "learning_rate": 1.3684210526315791e-05,
+      "loss": 0.0313,
+      "step": 444
+    },
+    {
+      "epoch": 445.0,
+      "learning_rate": 1.3596491228070177e-05,
+      "loss": 0.0325,
+      "step": 445
+    },
+    {
+      "epoch": 446.0,
+      "learning_rate": 1.3508771929824562e-05,
+      "loss": 0.0296,
+      "step": 446
+    },
+    {
+      "epoch": 447.0,
+      "learning_rate": 1.3421052631578948e-05,
+      "loss": 0.0333,
+      "step": 447
+    },
+    {
+      "epoch": 448.0,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 0.0333,
+      "step": 448
+    },
+    {
+      "epoch": 449.0,
+      "learning_rate": 1.3245614035087719e-05,
+      "loss": 0.0318,
+      "step": 449
+    },
+    {
+      "epoch": 450.0,
+      "learning_rate": 1.3157894736842106e-05,
+      "loss": 0.0311,
+      "step": 450
+    },
+    {
+      "epoch": 451.0,
+      "learning_rate": 1.3070175438596491e-05,
+      "loss": 0.029,
+      "step": 451
+    },
+    {
+      "epoch": 452.0,
+      "learning_rate": 1.2982456140350877e-05,
+      "loss": 0.0323,
+      "step": 452
+    },
+    {
+      "epoch": 453.0,
+      "learning_rate": 1.2894736842105264e-05,
+      "loss": 0.0312,
+      "step": 453
+    },
+    {
+      "epoch": 454.0,
+      "learning_rate": 1.2807017543859651e-05,
+      "loss": 0.0319,
+      "step": 454
+    },
+    {
+      "epoch": 455.0,
+      "learning_rate": 1.2719298245614037e-05,
+      "loss": 0.0306,
+      "step": 455
+    },
+    {
+      "epoch": 456.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0347,
+      "step": 456
+    },
+    {
+      "epoch": 457.0,
+      "learning_rate": 1.2543859649122808e-05,
+      "loss": 0.0328,
+      "step": 457
+    },
+    {
+      "epoch": 458.0,
+      "learning_rate": 1.2456140350877193e-05,
+      "loss": 0.0382,
+      "step": 458
+    },
+    {
+      "epoch": 459.0,
+      "learning_rate": 1.2368421052631579e-05,
+      "loss": 0.0312,
+      "step": 459
+    },
+    {
+      "epoch": 460.0,
+      "learning_rate": 1.2280701754385964e-05,
+      "loss": 0.0303,
+      "step": 460
+    },
+    {
+      "epoch": 460.0,
+      "eval_accuracy_NAT": 0.9767060519303628,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7583808327697807,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5826733473411992,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6590546999468933,
+      "eval_accuracy_speedway-bricks": 0.9969549187978346,
+      "eval_accuracy_steel": 0.8428761651131824,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9578953511746893,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7142207260485852,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4316331417624521,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5483063328424154,
+      "eval_iou_speedway-bricks": 0.9507760778859528,
+      "eval_iou_steel": 0.793730407523511,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.16362285614013672,
+      "eval_mean_accuracy": 0.8027743359832088,
+      "eval_mean_iou": 0.7327603395396008,
+      "eval_overall_accuracy": 0.9570274353027344,
+      "eval_runtime": 0.6034,
+      "eval_samples_per_second": 1.657,
+      "eval_steps_per_second": 1.657,
+      "step": 460
+    },
+    {
+      "epoch": 461.0,
+      "learning_rate": 1.2192982456140352e-05,
+      "loss": 0.0313,
+      "step": 461
+    },
+    {
+      "epoch": 462.0,
+      "learning_rate": 1.2105263157894737e-05,
+      "loss": 0.0309,
+      "step": 462
+    },
+    {
+      "epoch": 463.0,
+      "learning_rate": 1.2017543859649123e-05,
+      "loss": 0.0316,
+      "step": 463
+    },
+    {
+      "epoch": 464.0,
+      "learning_rate": 1.192982456140351e-05,
+      "loss": 0.0321,
+      "step": 464
+    },
+    {
+      "epoch": 465.0,
+      "learning_rate": 1.1842105263157895e-05,
+      "loss": 0.0309,
+      "step": 465
+    },
+    {
+      "epoch": 466.0,
+      "learning_rate": 1.1754385964912282e-05,
+      "loss": 0.0313,
+      "step": 466
+    },
+    {
+      "epoch": 467.0,
+      "learning_rate": 1.1666666666666668e-05,
+      "loss": 0.0332,
+      "step": 467
+    },
+    {
+      "epoch": 468.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 0.031,
+      "step": 468
+    },
+    {
+      "epoch": 469.0,
+      "learning_rate": 1.1491228070175439e-05,
+      "loss": 0.0315,
+      "step": 469
+    },
+    {
+      "epoch": 470.0,
+      "learning_rate": 1.1403508771929824e-05,
+      "loss": 0.0319,
+      "step": 470
+    },
+    {
+      "epoch": 471.0,
+      "learning_rate": 1.1315789473684212e-05,
+      "loss": 0.032,
+      "step": 471
+    },
+    {
+      "epoch": 472.0,
+      "learning_rate": 1.1228070175438597e-05,
+      "loss": 0.0311,
+      "step": 472
+    },
+    {
+      "epoch": 473.0,
+      "learning_rate": 1.1140350877192983e-05,
+      "loss": 0.0317,
+      "step": 473
+    },
+    {
+      "epoch": 474.0,
+      "learning_rate": 1.1052631578947368e-05,
+      "loss": 0.0291,
+      "step": 474
+    },
+    {
+      "epoch": 475.0,
+      "learning_rate": 1.0964912280701754e-05,
+      "loss": 0.0306,
+      "step": 475
+    },
+    {
+      "epoch": 476.0,
+      "learning_rate": 1.087719298245614e-05,
+      "loss": 0.0302,
+      "step": 476
+    },
+    {
+      "epoch": 477.0,
+      "learning_rate": 1.0789473684210526e-05,
+      "loss": 0.0328,
+      "step": 477
+    },
+    {
+      "epoch": 478.0,
+      "learning_rate": 1.0701754385964913e-05,
+      "loss": 0.0291,
+      "step": 478
+    },
+    {
+      "epoch": 479.0,
+      "learning_rate": 1.0614035087719299e-05,
+      "loss": 0.0299,
+      "step": 479
+    },
+    {
+      "epoch": 480.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0321,
+      "step": 480
+    },
+    {
+      "epoch": 480.0,
+      "eval_accuracy_NAT": 0.9769837544052208,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7958651338034972,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5884919993534831,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7164099840679766,
+      "eval_accuracy_speedway-bricks": 0.9938398357289527,
+      "eval_accuracy_steel": 0.9151131824234354,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9586028376616215,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7349451333749666,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.43804138594802694,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.593575828688765,
+      "eval_iou_speedway-bricks": 0.9590740719891014,
+      "eval_iou_steel": 0.815727002967359,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.149010568857193,
+      "eval_mean_accuracy": 0.8311173149637611,
+      "eval_mean_iou": 0.74999437677164,
+      "eval_overall_accuracy": 0.9598503112792969,
+      "eval_runtime": 0.6087,
+      "eval_samples_per_second": 1.643,
+      "eval_steps_per_second": 1.643,
+      "step": 480
+    },
+    {
+      "epoch": 481.0,
+      "learning_rate": 1.0438596491228072e-05,
+      "loss": 0.0293,
+      "step": 481
+    },
+    {
+      "epoch": 482.0,
+      "learning_rate": 1.0350877192982457e-05,
+      "loss": 0.031,
+      "step": 482
+    },
+    {
+      "epoch": 483.0,
+      "learning_rate": 1.0263157894736843e-05,
+      "loss": 0.0298,
+      "step": 483
+    },
+    {
+      "epoch": 484.0,
+      "learning_rate": 1.0175438596491228e-05,
+      "loss": 0.031,
+      "step": 484
+    },
+    {
+      "epoch": 485.0,
+      "learning_rate": 1.0087719298245614e-05,
+      "loss": 0.0285,
+      "step": 485
+    },
+    {
+      "epoch": 486.0,
+      "learning_rate": 1e-05,
+      "loss": 0.0306,
+      "step": 486
+    },
+    {
+      "epoch": 487.0,
+      "learning_rate": 9.912280701754386e-06,
+      "loss": 0.0304,
+      "step": 487
+    },
+    {
+      "epoch": 488.0,
+      "learning_rate": 9.824561403508772e-06,
+      "loss": 0.0314,
+      "step": 488
+    },
+    {
+      "epoch": 489.0,
+      "learning_rate": 9.736842105263157e-06,
+      "loss": 0.0306,
+      "step": 489
+    },
+    {
+      "epoch": 490.0,
+      "learning_rate": 9.649122807017545e-06,
+      "loss": 0.0315,
+      "step": 490
+    },
+    {
+      "epoch": 491.0,
+      "learning_rate": 9.561403508771932e-06,
+      "loss": 0.0291,
+      "step": 491
+    },
+    {
+      "epoch": 492.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0316,
+      "step": 492
+    },
+    {
+      "epoch": 493.0,
+      "learning_rate": 9.385964912280703e-06,
+      "loss": 0.0317,
+      "step": 493
+    },
+    {
+      "epoch": 494.0,
+      "learning_rate": 9.298245614035088e-06,
+      "loss": 0.03,
+      "step": 494
+    },
+    {
+      "epoch": 495.0,
+      "learning_rate": 9.210526315789474e-06,
+      "loss": 0.0305,
+      "step": 495
+    },
+    {
+      "epoch": 496.0,
+      "learning_rate": 9.122807017543861e-06,
+      "loss": 0.03,
+      "step": 496
+    },
+    {
+      "epoch": 497.0,
+      "learning_rate": 9.035087719298246e-06,
+      "loss": 0.0316,
+      "step": 497
+    },
+    {
+      "epoch": 498.0,
+      "learning_rate": 8.947368421052632e-06,
+      "loss": 0.0299,
+      "step": 498
+    },
+    {
+      "epoch": 499.0,
+      "learning_rate": 8.859649122807017e-06,
+      "loss": 0.0296,
+      "step": 499
+    },
+    {
+      "epoch": 500.0,
+      "learning_rate": 8.771929824561403e-06,
+      "loss": 0.0304,
+      "step": 500
+    },
+    {
+      "epoch": 500.0,
+      "eval_accuracy_NAT": 0.9771027697515885,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.81441406627379,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5791175044448036,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6838378474066207,
+      "eval_accuracy_speedway-bricks": 0.9922064588389023,
+      "eval_accuracy_steel": 0.9267643142476698,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9581725875159666,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7458856839497434,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.43530555218078,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5703528716964418,
+      "eval_iou_speedway-bricks": 0.9570775844606001,
+      "eval_iou_steel": 0.8157046586580721,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15441998839378357,
+      "eval_mean_accuracy": 0.8289071601605625,
+      "eval_mean_iou": 0.7470831564102673,
+      "eval_overall_accuracy": 0.9593276977539062,
+      "eval_runtime": 0.6315,
+      "eval_samples_per_second": 1.583,
+      "eval_steps_per_second": 1.583,
+      "step": 500
+    },
+    {
+      "epoch": 501.0,
+      "learning_rate": 8.68421052631579e-06,
+      "loss": 0.0288,
+      "step": 501
+    },
+    {
+      "epoch": 502.0,
+      "learning_rate": 8.596491228070176e-06,
+      "loss": 0.0297,
+      "step": 502
+    },
+    {
+      "epoch": 503.0,
+      "learning_rate": 8.508771929824563e-06,
+      "loss": 0.031,
+      "step": 503
+    },
+    {
+      "epoch": 504.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0304,
+      "step": 504
+    },
+    {
+      "epoch": 505.0,
+      "learning_rate": 8.333333333333334e-06,
+      "loss": 0.0308,
+      "step": 505
+    },
+    {
+      "epoch": 506.0,
+      "learning_rate": 8.245614035087721e-06,
+      "loss": 0.0299,
+      "step": 506
+    },
+    {
+      "epoch": 507.0,
+      "learning_rate": 8.157894736842106e-06,
+      "loss": 0.0303,
+      "step": 507
+    },
+    {
+      "epoch": 508.0,
+      "learning_rate": 8.070175438596492e-06,
+      "loss": 0.0291,
+      "step": 508
+    },
+    {
+      "epoch": 509.0,
+      "learning_rate": 7.982456140350877e-06,
+      "loss": 0.0299,
+      "step": 509
+    },
+    {
+      "epoch": 510.0,
+      "learning_rate": 7.894736842105263e-06,
+      "loss": 0.0316,
+      "step": 510
+    },
+    {
+      "epoch": 511.0,
+      "learning_rate": 7.80701754385965e-06,
+      "loss": 0.0316,
+      "step": 511
+    },
+    {
+      "epoch": 512.0,
+      "learning_rate": 7.719298245614036e-06,
+      "loss": 0.0291,
+      "step": 512
+    },
+    {
+      "epoch": 513.0,
+      "learning_rate": 7.631578947368421e-06,
+      "loss": 0.0287,
+      "step": 513
+    },
+    {
+      "epoch": 514.0,
+      "learning_rate": 7.5438596491228074e-06,
+      "loss": 0.0299,
+      "step": 514
+    },
+    {
+      "epoch": 515.0,
+      "learning_rate": 7.456140350877193e-06,
+      "loss": 0.0302,
+      "step": 515
+    },
+    {
+      "epoch": 516.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0293,
+      "step": 516
+    },
+    {
+      "epoch": 517.0,
+      "learning_rate": 7.280701754385966e-06,
+      "loss": 0.0296,
+      "step": 517
+    },
+    {
+      "epoch": 518.0,
+      "learning_rate": 7.192982456140351e-06,
+      "loss": 0.0293,
+      "step": 518
+    },
+    {
+      "epoch": 519.0,
+      "learning_rate": 7.1052631578947375e-06,
+      "loss": 0.0309,
+      "step": 519
+    },
+    {
+      "epoch": 520.0,
+      "learning_rate": 7.017543859649123e-06,
+      "loss": 0.0303,
+      "step": 520
+    },
+    {
+      "epoch": 520.0,
+      "eval_accuracy_NAT": 0.9771292176063369,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.7962515698966284,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5768546953289154,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6530359355638166,
+      "eval_accuracy_speedway-bricks": 0.9944815195071869,
+      "eval_accuracy_steel": 0.9094540612516645,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9579749132985447,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7401221264367817,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4342905816500365,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5471670127558588,
+      "eval_iou_speedway-bricks": 0.95417147078907,
+      "eval_iou_steel": 0.8177192457348099,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.16023945808410645,
+      "eval_mean_accuracy": 0.8178678331924248,
+      "eval_mean_iou": 0.7419075584441837,
+      "eval_overall_accuracy": 0.9584541320800781,
+      "eval_runtime": 0.6058,
+      "eval_samples_per_second": 1.651,
+      "eval_steps_per_second": 1.651,
+      "step": 520
+    },
+    {
+      "epoch": 521.0,
+      "learning_rate": 6.9298245614035085e-06,
+      "loss": 0.0289,
+      "step": 521
+    },
+    {
+      "epoch": 522.0,
+      "learning_rate": 6.842105263157896e-06,
+      "loss": 0.0332,
+      "step": 522
+    },
+    {
+      "epoch": 523.0,
+      "learning_rate": 6.754385964912281e-06,
+      "loss": 0.0296,
+      "step": 523
+    },
+    {
+      "epoch": 524.0,
+      "learning_rate": 6.666666666666667e-06,
+      "loss": 0.0313,
+      "step": 524
+    },
+    {
+      "epoch": 525.0,
+      "learning_rate": 6.578947368421053e-06,
+      "loss": 0.0296,
+      "step": 525
+    },
+    {
+      "epoch": 526.0,
+      "learning_rate": 6.4912280701754385e-06,
+      "loss": 0.031,
+      "step": 526
+    },
+    {
+      "epoch": 527.0,
+      "learning_rate": 6.403508771929826e-06,
+      "loss": 0.03,
+      "step": 527
+    },
+    {
+      "epoch": 528.0,
+      "learning_rate": 6.315789473684211e-06,
+      "loss": 0.0311,
+      "step": 528
+    },
+    {
+      "epoch": 529.0,
+      "learning_rate": 6.228070175438597e-06,
+      "loss": 0.028,
+      "step": 529
+    },
+    {
+      "epoch": 530.0,
+      "learning_rate": 6.140350877192982e-06,
+      "loss": 0.0307,
+      "step": 530
+    },
+    {
+      "epoch": 531.0,
+      "learning_rate": 6.0526315789473685e-06,
+      "loss": 0.0304,
+      "step": 531
+    },
+    {
+      "epoch": 532.0,
+      "learning_rate": 5.964912280701755e-06,
+      "loss": 0.0292,
+      "step": 532
+    },
+    {
+      "epoch": 533.0,
+      "learning_rate": 5.877192982456141e-06,
+      "loss": 0.0311,
+      "step": 533
+    },
+    {
+      "epoch": 534.0,
+      "learning_rate": 5.789473684210527e-06,
+      "loss": 0.0289,
+      "step": 534
+    },
+    {
+      "epoch": 535.0,
+      "learning_rate": 5.701754385964912e-06,
+      "loss": 0.0293,
+      "step": 535
+    },
+    {
+      "epoch": 536.0,
+      "learning_rate": 5.6140350877192985e-06,
+      "loss": 0.0305,
+      "step": 536
+    },
+    {
+      "epoch": 537.0,
+      "learning_rate": 5.526315789473684e-06,
+      "loss": 0.0295,
+      "step": 537
+    },
+    {
+      "epoch": 538.0,
+      "learning_rate": 5.43859649122807e-06,
+      "loss": 0.0297,
+      "step": 538
+    },
+    {
+      "epoch": 539.0,
+      "learning_rate": 5.350877192982457e-06,
+      "loss": 0.0287,
+      "step": 539
+    },
+    {
+      "epoch": 540.0,
+      "learning_rate": 5.263157894736842e-06,
+      "loss": 0.0301,
+      "step": 540
+    },
+    {
+      "epoch": 540.0,
+      "eval_accuracy_NAT": 0.9771027697515885,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.803690464689402,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.5884919993534831,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.6891485218622765,
+      "eval_accuracy_speedway-bricks": 0.9934431584842263,
+      "eval_accuracy_steel": 0.9191078561917443,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9585019717725197,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7441631630736202,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.4393628574876312,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5714914856136231,
+      "eval_iou_speedway-bricks": 0.9576777299157604,
+      "eval_iou_steel": 0.817105652559929,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1527710258960724,
+      "eval_mean_accuracy": 0.82849746172212,
+      "eval_mean_iou": 0.7480504767371806,
+      "eval_overall_accuracy": 0.9595565795898438,
+      "eval_runtime": 0.5967,
+      "eval_samples_per_second": 1.676,
+      "eval_steps_per_second": 1.676,
+      "step": 540
+    },
+    {
+      "epoch": 541.0,
+      "learning_rate": 5.1754385964912286e-06,
+      "loss": 0.0295,
+      "step": 541
+    },
+    {
+      "epoch": 542.0,
+      "learning_rate": 5.087719298245614e-06,
+      "loss": 0.0297,
+      "step": 542
+    },
+    {
+      "epoch": 543.0,
+      "learning_rate": 5e-06,
+      "loss": 0.0294,
+      "step": 543
+    },
+    {
+      "epoch": 544.0,
+      "learning_rate": 4.912280701754386e-06,
+      "loss": 0.0309,
+      "step": 544
+    },
+    {
+      "epoch": 545.0,
+      "learning_rate": 4.824561403508772e-06,
+      "loss": 0.0317,
+      "step": 545
+    },
+    {
+      "epoch": 546.0,
+      "learning_rate": 4.736842105263159e-06,
+      "loss": 0.0287,
+      "step": 546
+    },
+    {
+      "epoch": 547.0,
+      "learning_rate": 4.649122807017544e-06,
+      "loss": 0.0315,
+      "step": 547
+    },
+    {
+      "epoch": 548.0,
+      "learning_rate": 4.5614035087719304e-06,
+      "loss": 0.03,
+      "step": 548
+    },
+    {
+      "epoch": 549.0,
+      "learning_rate": 4.473684210526316e-06,
+      "loss": 0.0303,
+      "step": 549
+    },
+    {
+      "epoch": 550.0,
+      "learning_rate": 4.3859649122807014e-06,
+      "loss": 0.0307,
+      "step": 550
+    },
+    {
+      "epoch": 551.0,
+      "learning_rate": 4.298245614035088e-06,
+      "loss": 0.0294,
+      "step": 551
+    },
+    {
+      "epoch": 552.0,
+      "learning_rate": 4.210526315789474e-06,
+      "loss": 0.0336,
+      "step": 552
+    },
+    {
+      "epoch": 553.0,
+      "learning_rate": 4.1228070175438605e-06,
+      "loss": 0.0301,
+      "step": 553
+    },
+    {
+      "epoch": 554.0,
+      "learning_rate": 4.035087719298246e-06,
+      "loss": 0.0322,
+      "step": 554
+    },
+    {
+      "epoch": 555.0,
+      "learning_rate": 3.9473684210526315e-06,
+      "loss": 0.0294,
+      "step": 555
+    },
+    {
+      "epoch": 556.0,
+      "learning_rate": 3.859649122807018e-06,
+      "loss": 0.0315,
+      "step": 556
+    },
+    {
+      "epoch": 557.0,
+      "learning_rate": 3.7719298245614037e-06,
+      "loss": 0.0317,
+      "step": 557
+    },
+    {
+      "epoch": 558.0,
+      "learning_rate": 3.6842105263157892e-06,
+      "loss": 0.0305,
+      "step": 558
+    },
+    {
+      "epoch": 559.0,
+      "learning_rate": 3.5964912280701756e-06,
+      "loss": 0.0288,
+      "step": 559
+    },
+    {
+      "epoch": 560.0,
+      "learning_rate": 3.5087719298245615e-06,
+      "loss": 0.0305,
+      "step": 560
+    },
+    {
+      "epoch": 560.0,
+      "eval_accuracy_NAT": 0.9780548925225303,
+      "eval_accuracy_REST": NaN,
+      "eval_accuracy_concrete-road-adj": NaN,
+      "eval_accuracy_concrete-speedway-side": 0.8006955849676359,
+      "eval_accuracy_dark-bricks": NaN,
+      "eval_accuracy_grass": 0.582188459673509,
+      "eval_accuracy_red-bricks": NaN,
+      "eval_accuracy_road": NaN,
+      "eval_accuracy_rough-red-sidewalk": 0.7091520623119136,
+      "eval_accuracy_speedway-bricks": 0.9940615083068882,
+      "eval_accuracy_steel": 0.9107856191744341,
+      "eval_accuracy_steel-sidewalk": NaN,
+      "eval_accuracy_tiles-marble": NaN,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.959149797045817,
+      "eval_iou_REST": NaN,
+      "eval_iou_concrete-road-adj": NaN,
+      "eval_iou_concrete-speedway-side": 0.7417882395059519,
+      "eval_iou_dark-bricks": NaN,
+      "eval_iou_grass": 0.43803964489845554,
+      "eval_iou_red-bricks": NaN,
+      "eval_iou_road": NaN,
+      "eval_iou_rough-red-sidewalk": 0.5944502151654548,
+      "eval_iou_speedway-bricks": 0.9590072598345433,
+      "eval_iou_steel": 0.8218684289576449,
+      "eval_iou_steel-sidewalk": NaN,
+      "eval_iou_tiles-marble": NaN,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15004150569438934,
+      "eval_mean_accuracy": 0.8291563544928184,
+      "eval_mean_iou": 0.7523839309013112,
+      "eval_overall_accuracy": 0.9603767395019531,
+      "eval_runtime": 0.6051,
+      "eval_samples_per_second": 1.653,
+      "eval_steps_per_second": 1.653,
+      "step": 560
+    }
+  ],
+  "max_steps": 600,
+  "num_train_epochs": 600,
+  "total_flos": 2.9062653737435136e+18,
+  "trial_name": null,
+  "trial_params": null
+}