Training in progress, step 2000

{checkpoint-1680 → checkpoint-2000}/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a7e4d829de8426a31521fb19bf569829ffb66aa822938b07c1b05d9b9b03b672
+oid sha256:2d68fbea77d605f010cd349b833d93cd4982d03b7d140d59d68451c4396af26e
 size 29861573

{checkpoint-1680 → checkpoint-2000}/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82a4e9a80105ce7c3324f100d4fa3bda5eedc2e63a18561b514dd5b1a2c9bc82
+oid sha256:7a89a602af7a559cba27b8f9e51253d3ee9a3753bb186dff447f69ecd6684727
 size 14944077

{checkpoint-1680 → checkpoint-2000}/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3d504f58c5f4a48a5f24477585ba7094733c7851c3d2df2002a57b9b5e367040
+oid sha256:b9c3f7553c7e7b6befb00a970be6886d2d0387893ba6e529fb89917f9170f5a9
 size 14575

{checkpoint-1680 → checkpoint-2000}/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac16f5862343fa2c55f6eef5881a26574de304da2e0acba52aad72ed542f94fc
+oid sha256:7d01f86ee3b0f90cca0cc7b4ddcfb9d971c729912b8909b6b7d8dfabd0da5b57
 size 627

{checkpoint-1680 → checkpoint-2000}/trainer_state.json

@@ -1,8 +1,8 @@
 {
   "best_metric": 0.09530654549598694,
   "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0/checkpoint-1440",
-  "epoch": 152.72727272727272,
-  "global_step": 1680,
+  "epoch": 181.8181818181818,
+  "global_step": 2000,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -13362,11 +13362,2555 @@
       "eval_samples_per_second": 10.559,
       "eval_steps_per_second": 0.587,
       "step": 1680
+    },
+    {
+      "epoch": 152.82,
+      "learning_rate": 7.30909090909091e-05,
+      "loss": 0.0403,
+      "step": 1681
+    },
+    {
+      "epoch": 152.91,
+      "learning_rate": 7.307177033492823e-05,
+      "loss": 0.0588,
+      "step": 1682
+    },
+    {
+      "epoch": 153.0,
+      "learning_rate": 7.305263157894738e-05,
+      "loss": 0.045,
+      "step": 1683
+    },
+    {
+      "epoch": 153.09,
+      "learning_rate": 7.303349282296651e-05,
+      "loss": 0.051,
+      "step": 1684
+    },
+    {
+      "epoch": 153.18,
+      "learning_rate": 7.301435406698564e-05,
+      "loss": 0.0449,
+      "step": 1685
+    },
+    {
+      "epoch": 153.27,
+      "learning_rate": 7.299521531100479e-05,
+      "loss": 0.0492,
+      "step": 1686
+    },
+    {
+      "epoch": 153.36,
+      "learning_rate": 7.297607655502393e-05,
+      "loss": 0.046,
+      "step": 1687
+    },
+    {
+      "epoch": 153.45,
+      "learning_rate": 7.295693779904306e-05,
+      "loss": 0.0436,
+      "step": 1688
+    },
+    {
+      "epoch": 153.55,
+      "learning_rate": 7.293779904306221e-05,
+      "loss": 0.0458,
+      "step": 1689
+    },
+    {
+      "epoch": 153.64,
+      "learning_rate": 7.291866028708134e-05,
+      "loss": 0.0456,
+      "step": 1690
+    },
+    {
+      "epoch": 153.73,
+      "learning_rate": 7.289952153110047e-05,
+      "loss": 0.0443,
+      "step": 1691
+    },
+    {
+      "epoch": 153.82,
+      "learning_rate": 7.288038277511962e-05,
+      "loss": 0.0478,
+      "step": 1692
+    },
+    {
+      "epoch": 153.91,
+      "learning_rate": 7.286124401913876e-05,
+      "loss": 0.044,
+      "step": 1693
+    },
+    {
+      "epoch": 154.0,
+      "learning_rate": 7.28421052631579e-05,
+      "loss": 0.0533,
+      "step": 1694
+    },
+    {
+      "epoch": 154.09,
+      "learning_rate": 7.282296650717704e-05,
+      "loss": 0.0448,
+      "step": 1695
+    },
+    {
+      "epoch": 154.18,
+      "learning_rate": 7.280382775119617e-05,
+      "loss": 0.0454,
+      "step": 1696
+    },
+    {
+      "epoch": 154.27,
+      "learning_rate": 7.27846889952153e-05,
+      "loss": 0.0466,
+      "step": 1697
+    },
+    {
+      "epoch": 154.36,
+      "learning_rate": 7.276555023923445e-05,
+      "loss": 0.0487,
+      "step": 1698
+    },
+    {
+      "epoch": 154.45,
+      "learning_rate": 7.27464114832536e-05,
+      "loss": 0.0538,
+      "step": 1699
+    },
+    {
+      "epoch": 154.55,
+      "learning_rate": 7.272727272727273e-05,
+      "loss": 0.0443,
+      "step": 1700
+    },
+    {
+      "epoch": 154.55,
+      "eval_accuracy_NAT": 0.9924235419667806,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9455247542812997,
+      "eval_accuracy_concrete tiles": 0.9844371063413441,
+      "eval_accuracy_dark bricks": 0.8664408606289211,
+      "eval_accuracy_grass": 0.887276883233083,
+      "eval_accuracy_red bricks": 0.9135298295454546,
+      "eval_accuracy_road": 0.8743208780790992,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.3527968461390015,
+      "eval_accuracy_speedway bricks": 0.987067160495615,
+      "eval_accuracy_steel": 0.898877052964844,
+      "eval_accuracy_tiles": 0.005341734009039857,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788732284057954,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.87227442469357,
+      "eval_iou_concrete tiles": 0.9309796437659033,
+      "eval_iou_dark bricks": 0.822139439757496,
+      "eval_iou_grass": 0.8204968261560937,
+      "eval_iou_red bricks": 0.8196590728054803,
+      "eval_iou_road": 0.8678564527028243,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.240477309029409,
+      "eval_iou_speedway bricks": 0.9737516637736648,
+      "eval_iou_steel": 0.8236854034451496,
+      "eval_iou_tiles": 0.005341734009039857,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12572631239891052,
+      "eval_mean_accuracy": 0.6698489728988063,
+      "eval_mean_iou": 0.6273488614264943,
+      "eval_overall_accuracy": 0.9667922125922309,
+      "eval_runtime": 3.275,
+      "eval_samples_per_second": 10.992,
+      "eval_steps_per_second": 0.611,
+      "step": 1700
+    },
+    {
+      "epoch": 154.64,
+      "learning_rate": 7.270813397129187e-05,
+      "loss": 0.0484,
+      "step": 1701
+    },
+    {
+      "epoch": 154.73,
+      "learning_rate": 7.2688995215311e-05,
+      "loss": 0.0539,
+      "step": 1702
+    },
+    {
+      "epoch": 154.82,
+      "learning_rate": 7.266985645933014e-05,
+      "loss": 0.0452,
+      "step": 1703
+    },
+    {
+      "epoch": 154.91,
+      "learning_rate": 7.265071770334928e-05,
+      "loss": 0.0453,
+      "step": 1704
+    },
+    {
+      "epoch": 155.0,
+      "learning_rate": 7.263157894736843e-05,
+      "loss": 0.0474,
+      "step": 1705
+    },
+    {
+      "epoch": 155.09,
+      "learning_rate": 7.261244019138756e-05,
+      "loss": 0.0457,
+      "step": 1706
+    },
+    {
+      "epoch": 155.18,
+      "learning_rate": 7.25933014354067e-05,
+      "loss": 0.0448,
+      "step": 1707
+    },
+    {
+      "epoch": 155.27,
+      "learning_rate": 7.257416267942584e-05,
+      "loss": 0.0492,
+      "step": 1708
+    },
+    {
+      "epoch": 155.36,
+      "learning_rate": 7.255502392344497e-05,
+      "loss": 0.0461,
+      "step": 1709
+    },
+    {
+      "epoch": 155.45,
+      "learning_rate": 7.253588516746413e-05,
+      "loss": 0.0505,
+      "step": 1710
+    },
+    {
+      "epoch": 155.55,
+      "learning_rate": 7.251674641148326e-05,
+      "loss": 0.0449,
+      "step": 1711
+    },
+    {
+      "epoch": 155.64,
+      "learning_rate": 7.24976076555024e-05,
+      "loss": 0.0411,
+      "step": 1712
+    },
+    {
+      "epoch": 155.73,
+      "learning_rate": 7.247846889952154e-05,
+      "loss": 0.0402,
+      "step": 1713
+    },
+    {
+      "epoch": 155.82,
+      "learning_rate": 7.245933014354067e-05,
+      "loss": 0.0483,
+      "step": 1714
+    },
+    {
+      "epoch": 155.91,
+      "learning_rate": 7.24401913875598e-05,
+      "loss": 0.0475,
+      "step": 1715
+    },
+    {
+      "epoch": 156.0,
+      "learning_rate": 7.242105263157896e-05,
+      "loss": 0.0419,
+      "step": 1716
+    },
+    {
+      "epoch": 156.09,
+      "learning_rate": 7.24019138755981e-05,
+      "loss": 0.0486,
+      "step": 1717
+    },
+    {
+      "epoch": 156.18,
+      "learning_rate": 7.238277511961723e-05,
+      "loss": 0.0486,
+      "step": 1718
+    },
+    {
+      "epoch": 156.27,
+      "learning_rate": 7.236363636363637e-05,
+      "loss": 0.0438,
+      "step": 1719
+    },
+    {
+      "epoch": 156.36,
+      "learning_rate": 7.23444976076555e-05,
+      "loss": 0.0417,
+      "step": 1720
+    },
+    {
+      "epoch": 156.36,
+      "eval_accuracy_NAT": 0.9929417788954583,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9489838619683248,
+      "eval_accuracy_concrete tiles": 0.9862410566868465,
+      "eval_accuracy_dark bricks": 0.8794888572418306,
+      "eval_accuracy_grass": 0.8583831090078534,
+      "eval_accuracy_red bricks": 0.900390625,
+      "eval_accuracy_road": 0.8707186449121933,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.40162104044330493,
+      "eval_accuracy_speedway bricks": 0.9863720347031897,
+      "eval_accuracy_steel": 0.8962551942484005,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9789846677739332,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.868326004543318,
+      "eval_iou_concrete tiles": 0.93206576704135,
+      "eval_iou_dark bricks": 0.8122134148508868,
+      "eval_iou_grass": 0.8055510000590392,
+      "eval_iou_red bricks": 0.8281887963416625,
+      "eval_iou_road": 0.8637479742965066,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.26952322806238554,
+      "eval_iou_speedway bricks": 0.9743610682113367,
+      "eval_iou_steel": 0.8308112074104645,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13718970119953156,
+      "eval_mean_accuracy": 0.6708766310082617,
+      "eval_mean_iou": 0.6279825483531448,
+      "eval_overall_accuracy": 0.9662430021497939,
+      "eval_runtime": 3.5774,
+      "eval_samples_per_second": 10.063,
+      "eval_steps_per_second": 0.559,
+      "step": 1720
+    },
+    {
+      "epoch": 156.45,
+      "learning_rate": 7.232535885167464e-05,
+      "loss": 0.0428,
+      "step": 1721
+    },
+    {
+      "epoch": 156.55,
+      "learning_rate": 7.23062200956938e-05,
+      "loss": 0.0391,
+      "step": 1722
+    },
+    {
+      "epoch": 156.64,
+      "learning_rate": 7.228708133971293e-05,
+      "loss": 0.0554,
+      "step": 1723
+    },
+    {
+      "epoch": 156.73,
+      "learning_rate": 7.226794258373206e-05,
+      "loss": 0.0633,
+      "step": 1724
+    },
+    {
+      "epoch": 156.82,
+      "learning_rate": 7.22488038277512e-05,
+      "loss": 0.0501,
+      "step": 1725
+    },
+    {
+      "epoch": 156.91,
+      "learning_rate": 7.222966507177034e-05,
+      "loss": 0.0458,
+      "step": 1726
+    },
+    {
+      "epoch": 157.0,
+      "learning_rate": 7.221052631578947e-05,
+      "loss": 0.044,
+      "step": 1727
+    },
+    {
+      "epoch": 157.09,
+      "learning_rate": 7.219138755980863e-05,
+      "loss": 0.0454,
+      "step": 1728
+    },
+    {
+      "epoch": 157.18,
+      "learning_rate": 7.217224880382776e-05,
+      "loss": 0.0393,
+      "step": 1729
+    },
+    {
+      "epoch": 157.27,
+      "learning_rate": 7.215311004784689e-05,
+      "loss": 0.0477,
+      "step": 1730
+    },
+    {
+      "epoch": 157.36,
+      "learning_rate": 7.213397129186604e-05,
+      "loss": 0.0432,
+      "step": 1731
+    },
+    {
+      "epoch": 157.45,
+      "learning_rate": 7.211483253588517e-05,
+      "loss": 0.0444,
+      "step": 1732
+    },
+    {
+      "epoch": 157.55,
+      "learning_rate": 7.20956937799043e-05,
+      "loss": 0.0561,
+      "step": 1733
+    },
+    {
+      "epoch": 157.64,
+      "learning_rate": 7.207655502392345e-05,
+      "loss": 0.0509,
+      "step": 1734
+    },
+    {
+      "epoch": 157.73,
+      "learning_rate": 7.205741626794259e-05,
+      "loss": 0.0474,
+      "step": 1735
+    },
+    {
+      "epoch": 157.82,
+      "learning_rate": 7.203827751196172e-05,
+      "loss": 0.0418,
+      "step": 1736
+    },
+    {
+      "epoch": 157.91,
+      "learning_rate": 7.201913875598087e-05,
+      "loss": 0.0436,
+      "step": 1737
+    },
+    {
+      "epoch": 158.0,
+      "learning_rate": 7.2e-05,
+      "loss": 0.0617,
+      "step": 1738
+    },
+    {
+      "epoch": 158.09,
+      "learning_rate": 7.198086124401913e-05,
+      "loss": 0.0506,
+      "step": 1739
+    },
+    {
+      "epoch": 158.18,
+      "learning_rate": 7.196172248803828e-05,
+      "loss": 0.0475,
+      "step": 1740
+    },
+    {
+      "epoch": 158.18,
+      "eval_accuracy_NAT": 0.9930121828908762,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9459190845100648,
+      "eval_accuracy_concrete tiles": 0.9752950528954932,
+      "eval_accuracy_dark bricks": 0.8883799699780609,
+      "eval_accuracy_grass": 0.8680469877707372,
+      "eval_accuracy_red bricks": 0.9161931818181818,
+      "eval_accuracy_road": 0.8860647144392838,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.44691643922476776,
+      "eval_accuracy_speedway bricks": 0.9888828145836426,
+      "eval_accuracy_steel": 0.8897829958446013,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791981999021095,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8702337455932854,
+      "eval_iou_concrete tiles": 0.9425844389941196,
+      "eval_iou_dark bricks": 0.818040049618997,
+      "eval_iou_grass": 0.8095168250739886,
+      "eval_iou_red bricks": 0.8168434383409846,
+      "eval_iou_road": 0.8800320900828194,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.29832535885167466,
+      "eval_iou_speedway bricks": 0.9739895815689456,
+      "eval_iou_steel": 0.8291397444663834,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11910075694322586,
+      "eval_mean_accuracy": 0.6768071864581314,
+      "eval_mean_iou": 0.6321464209610237,
+      "eval_overall_accuracy": 0.968208736843533,
+      "eval_runtime": 3.7624,
+      "eval_samples_per_second": 9.568,
+      "eval_steps_per_second": 0.532,
+      "step": 1740
+    },
+    {
+      "epoch": 158.27,
+      "learning_rate": 7.194258373205742e-05,
+      "loss": 0.0479,
+      "step": 1741
+    },
+    {
+      "epoch": 158.36,
+      "learning_rate": 7.192344497607656e-05,
+      "loss": 0.0411,
+      "step": 1742
+    },
+    {
+      "epoch": 158.45,
+      "learning_rate": 7.19043062200957e-05,
+      "loss": 0.0461,
+      "step": 1743
+    },
+    {
+      "epoch": 158.55,
+      "learning_rate": 7.188516746411483e-05,
+      "loss": 0.0429,
+      "step": 1744
+    },
+    {
+      "epoch": 158.64,
+      "learning_rate": 7.186602870813397e-05,
+      "loss": 0.0409,
+      "step": 1745
+    },
+    {
+      "epoch": 158.73,
+      "learning_rate": 7.184688995215311e-05,
+      "loss": 0.0403,
+      "step": 1746
+    },
+    {
+      "epoch": 158.82,
+      "learning_rate": 7.182775119617226e-05,
+      "loss": 0.0452,
+      "step": 1747
+    },
+    {
+      "epoch": 158.91,
+      "learning_rate": 7.180861244019139e-05,
+      "loss": 0.0459,
+      "step": 1748
+    },
+    {
+      "epoch": 159.0,
+      "learning_rate": 7.178947368421054e-05,
+      "loss": 0.0433,
+      "step": 1749
+    },
+    {
+      "epoch": 159.09,
+      "learning_rate": 7.177033492822967e-05,
+      "loss": 0.0504,
+      "step": 1750
+    },
+    {
+      "epoch": 159.18,
+      "learning_rate": 7.17511961722488e-05,
+      "loss": 0.0481,
+      "step": 1751
+    },
+    {
+      "epoch": 159.27,
+      "learning_rate": 7.173205741626794e-05,
+      "loss": 0.0506,
+      "step": 1752
+    },
+    {
+      "epoch": 159.36,
+      "learning_rate": 7.171291866028709e-05,
+      "loss": 0.0462,
+      "step": 1753
+    },
+    {
+      "epoch": 159.45,
+      "learning_rate": 7.169377990430622e-05,
+      "loss": 0.0411,
+      "step": 1754
+    },
+    {
+      "epoch": 159.55,
+      "learning_rate": 7.167464114832535e-05,
+      "loss": 0.045,
+      "step": 1755
+    },
+    {
+      "epoch": 159.64,
+      "learning_rate": 7.16555023923445e-05,
+      "loss": 0.0514,
+      "step": 1756
+    },
+    {
+      "epoch": 159.73,
+      "learning_rate": 7.163636363636363e-05,
+      "loss": 0.0479,
+      "step": 1757
+    },
+    {
+      "epoch": 159.82,
+      "learning_rate": 7.161722488038278e-05,
+      "loss": 0.0461,
+      "step": 1758
+    },
+    {
+      "epoch": 159.91,
+      "learning_rate": 7.159808612440192e-05,
+      "loss": 0.0491,
+      "step": 1759
+    },
+    {
+      "epoch": 160.0,
+      "learning_rate": 7.157894736842105e-05,
+      "loss": 0.0468,
+      "step": 1760
+    },
+    {
+      "epoch": 160.0,
+      "eval_accuracy_NAT": 0.9919785322513821,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9455019528735139,
+      "eval_accuracy_concrete tiles": 0.9856906989543203,
+      "eval_accuracy_dark bricks": 0.8582810515376621,
+      "eval_accuracy_grass": 0.8940992670830464,
+      "eval_accuracy_red bricks": 0.9371448863636364,
+      "eval_accuracy_road": 0.8714707728174688,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4859261709811706,
+      "eval_accuracy_speedway bricks": 0.9858832743803907,
+      "eval_accuracy_steel": 0.9074187058901128,
+      "eval_accuracy_tiles": 0.2383920010957403,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788417946247757,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8815325302530665,
+      "eval_iou_concrete tiles": 0.9147348409613256,
+      "eval_iou_dark bricks": 0.8043211657769441,
+      "eval_iou_grass": 0.8221080302855013,
+      "eval_iou_red bricks": 0.8352587434720684,
+      "eval_iou_road": 0.8638309195362345,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3264194970184081,
+      "eval_iou_speedway bricks": 0.9743969131495298,
+      "eval_iou_steel": 0.8223668479885229,
+      "eval_iou_tiles": 0.23835935360175295,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13528358936309814,
+      "eval_mean_accuracy": 0.7001374857098804,
+      "eval_mean_iou": 0.6509362028206254,
+      "eval_overall_accuracy": 0.9672504001193576,
+      "eval_runtime": 3.3305,
+      "eval_samples_per_second": 10.809,
+      "eval_steps_per_second": 0.601,
+      "step": 1760
+    },
+    {
+      "epoch": 160.09,
+      "learning_rate": 7.155980861244019e-05,
+      "loss": 0.0413,
+      "step": 1761
+    },
+    {
+      "epoch": 160.18,
+      "learning_rate": 7.154066985645933e-05,
+      "loss": 0.0436,
+      "step": 1762
+    },
+    {
+      "epoch": 160.27,
+      "learning_rate": 7.152153110047846e-05,
+      "loss": 0.0454,
+      "step": 1763
+    },
+    {
+      "epoch": 160.36,
+      "learning_rate": 7.150239234449761e-05,
+      "loss": 0.0448,
+      "step": 1764
+    },
+    {
+      "epoch": 160.45,
+      "learning_rate": 7.148325358851676e-05,
+      "loss": 0.0479,
+      "step": 1765
+    },
+    {
+      "epoch": 160.55,
+      "learning_rate": 7.146411483253589e-05,
+      "loss": 0.0449,
+      "step": 1766
+    },
+    {
+      "epoch": 160.64,
+      "learning_rate": 7.144497607655502e-05,
+      "loss": 0.0491,
+      "step": 1767
+    },
+    {
+      "epoch": 160.73,
+      "learning_rate": 7.142583732057417e-05,
+      "loss": 0.048,
+      "step": 1768
+    },
+    {
+      "epoch": 160.82,
+      "learning_rate": 7.14066985645933e-05,
+      "loss": 0.0407,
+      "step": 1769
+    },
+    {
+      "epoch": 160.91,
+      "learning_rate": 7.138755980861244e-05,
+      "loss": 0.0496,
+      "step": 1770
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 7.136842105263159e-05,
+      "loss": 0.0498,
+      "step": 1771
+    },
+    {
+      "epoch": 161.09,
+      "learning_rate": 7.134928229665072e-05,
+      "loss": 0.0423,
+      "step": 1772
+    },
+    {
+      "epoch": 161.18,
+      "learning_rate": 7.133014354066985e-05,
+      "loss": 0.0429,
+      "step": 1773
+    },
+    {
+      "epoch": 161.27,
+      "learning_rate": 7.1311004784689e-05,
+      "loss": 0.0469,
+      "step": 1774
+    },
+    {
+      "epoch": 161.36,
+      "learning_rate": 7.129186602870813e-05,
+      "loss": 0.039,
+      "step": 1775
+    },
+    {
+      "epoch": 161.45,
+      "learning_rate": 7.127272727272728e-05,
+      "loss": 0.047,
+      "step": 1776
+    },
+    {
+      "epoch": 161.55,
+      "learning_rate": 7.125358851674642e-05,
+      "loss": 0.041,
+      "step": 1777
+    },
+    {
+      "epoch": 161.64,
+      "learning_rate": 7.123444976076555e-05,
+      "loss": 0.0396,
+      "step": 1778
+    },
+    {
+      "epoch": 161.73,
+      "learning_rate": 7.121531100478468e-05,
+      "loss": 0.043,
+      "step": 1779
+    },
+    {
+      "epoch": 161.82,
+      "learning_rate": 7.119617224880383e-05,
+      "loss": 0.0569,
+      "step": 1780
+    },
+    {
+      "epoch": 161.82,
+      "eval_accuracy_NAT": 0.9930390034605593,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9492856453066655,
+      "eval_accuracy_concrete tiles": 0.9792698587415153,
+      "eval_accuracy_dark bricks": 0.8537777606712598,
+      "eval_accuracy_grass": 0.8787489034206286,
+      "eval_accuracy_red bricks": 0.8966619318181818,
+      "eval_accuracy_road": 0.8813221973259555,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.44752295095525596,
+      "eval_accuracy_speedway bricks": 0.9856455317048564,
+      "eval_accuracy_steel": 0.9042114636237715,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787156413862954,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8746314288327076,
+      "eval_iou_concrete tiles": 0.9403957954078337,
+      "eval_iou_dark bricks": 0.813839154681538,
+      "eval_iou_grass": 0.8278320135653491,
+      "eval_iou_red bricks": 0.8456128600133959,
+      "eval_iou_road": 0.8728029995485349,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3010180429098596,
+      "eval_iou_speedway bricks": 0.9744170522760448,
+      "eval_iou_steel": 0.8304181242711978,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12074758112430573,
+      "eval_mean_accuracy": 0.674575788232973,
+      "eval_mean_iou": 0.635360239453289,
+      "eval_overall_accuracy": 0.9679258134629991,
+      "eval_runtime": 3.3537,
+      "eval_samples_per_second": 10.734,
+      "eval_steps_per_second": 0.596,
+      "step": 1780
+    },
+    {
+      "epoch": 161.91,
+      "learning_rate": 7.117703349282298e-05,
+      "loss": 0.0444,
+      "step": 1781
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 7.115789473684211e-05,
+      "loss": 0.0471,
+      "step": 1782
+    },
+    {
+      "epoch": 162.09,
+      "learning_rate": 7.113875598086125e-05,
+      "loss": 0.0541,
+      "step": 1783
+    },
+    {
+      "epoch": 162.18,
+      "learning_rate": 7.111961722488039e-05,
+      "loss": 0.0454,
+      "step": 1784
+    },
+    {
+      "epoch": 162.27,
+      "learning_rate": 7.110047846889952e-05,
+      "loss": 0.0471,
+      "step": 1785
+    },
+    {
+      "epoch": 162.36,
+      "learning_rate": 7.108133971291866e-05,
+      "loss": 0.0406,
+      "step": 1786
+    },
+    {
+      "epoch": 162.45,
+      "learning_rate": 7.106220095693781e-05,
+      "loss": 0.0432,
+      "step": 1787
+    },
+    {
+      "epoch": 162.55,
+      "learning_rate": 7.104306220095694e-05,
+      "loss": 0.0484,
+      "step": 1788
+    },
+    {
+      "epoch": 162.64,
+      "learning_rate": 7.102392344497609e-05,
+      "loss": 0.0418,
+      "step": 1789
+    },
+    {
+      "epoch": 162.73,
+      "learning_rate": 7.100478468899522e-05,
+      "loss": 0.0493,
+      "step": 1790
+    },
+    {
+      "epoch": 162.82,
+      "learning_rate": 7.098564593301435e-05,
+      "loss": 0.0459,
+      "step": 1791
+    },
+    {
+      "epoch": 162.91,
+      "learning_rate": 7.09665071770335e-05,
+      "loss": 0.0475,
+      "step": 1792
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 7.094736842105264e-05,
+      "loss": 0.0628,
+      "step": 1793
+    },
+    {
+      "epoch": 163.09,
+      "learning_rate": 7.092822966507177e-05,
+      "loss": 0.0439,
+      "step": 1794
+    },
+    {
+      "epoch": 163.18,
+      "learning_rate": 7.090909090909092e-05,
+      "loss": 0.0467,
+      "step": 1795
+    },
+    {
+      "epoch": 163.27,
+      "learning_rate": 7.088995215311005e-05,
+      "loss": 0.0547,
+      "step": 1796
+    },
+    {
+      "epoch": 163.36,
+      "learning_rate": 7.087081339712918e-05,
+      "loss": 0.0465,
+      "step": 1797
+    },
+    {
+      "epoch": 163.45,
+      "learning_rate": 7.085167464114833e-05,
+      "loss": 0.0454,
+      "step": 1798
+    },
+    {
+      "epoch": 163.55,
+      "learning_rate": 7.083253588516747e-05,
+      "loss": 0.0468,
+      "step": 1799
+    },
+    {
+      "epoch": 163.64,
+      "learning_rate": 7.08133971291866e-05,
+      "loss": 0.0414,
+      "step": 1800
+    },
+    {
+      "epoch": 163.64,
+      "eval_accuracy_NAT": 0.9933093265707855,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9372478432550754,
+      "eval_accuracy_concrete tiles": 0.9910719745612426,
+      "eval_accuracy_dark bricks": 0.8719833724644933,
+      "eval_accuracy_grass": 0.8813142875117522,
+      "eval_accuracy_red bricks": 0.9128196022727273,
+      "eval_accuracy_road": 0.8683050640457468,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48851763019325667,
+      "eval_accuracy_speedway bricks": 0.9872536135076457,
+      "eval_accuracy_steel": 0.9058109623375766,
+      "eval_accuracy_tiles": 0.1494315847144227,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9779009757340675,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8811723374366657,
+      "eval_iou_concrete tiles": 0.9083622912229571,
+      "eval_iou_dark bricks": 0.8177814677110782,
+      "eval_iou_grass": 0.8256598658166423,
+      "eval_iou_red bricks": 0.8482098663586867,
+      "eval_iou_road": 0.8607135964980624,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3222697099208875,
+      "eval_iou_speedway bricks": 0.9744303817069401,
+      "eval_iou_steel": 0.8261010143543548,
+      "eval_iou_tiles": 0.14934976043805612,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14603829383850098,
+      "eval_mean_accuracy": 0.6913127124180558,
+      "eval_mean_iou": 0.6455347128614153,
+      "eval_overall_accuracy": 0.9668665991889106,
+      "eval_runtime": 3.9858,
+      "eval_samples_per_second": 9.032,
+      "eval_steps_per_second": 0.502,
+      "step": 1800
+    },
+    {
+      "epoch": 163.73,
+      "learning_rate": 7.079425837320575e-05,
+      "loss": 0.0432,
+      "step": 1801
+    },
+    {
+      "epoch": 163.82,
+      "learning_rate": 7.077511961722488e-05,
+      "loss": 0.0726,
+      "step": 1802
+    },
+    {
+      "epoch": 163.91,
+      "learning_rate": 7.075598086124402e-05,
+      "loss": 0.0472,
+      "step": 1803
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 7.073684210526316e-05,
+      "loss": 0.0452,
+      "step": 1804
+    },
+    {
+      "epoch": 164.09,
+      "learning_rate": 7.07177033492823e-05,
+      "loss": 0.0485,
+      "step": 1805
+    },
+    {
+      "epoch": 164.18,
+      "learning_rate": 7.069856459330144e-05,
+      "loss": 0.0502,
+      "step": 1806
+    },
+    {
+      "epoch": 164.27,
+      "learning_rate": 7.067942583732058e-05,
+      "loss": 0.0617,
+      "step": 1807
+    },
+    {
+      "epoch": 164.36,
+      "learning_rate": 7.066028708133972e-05,
+      "loss": 0.0436,
+      "step": 1808
+    },
+    {
+      "epoch": 164.45,
+      "learning_rate": 7.064114832535885e-05,
+      "loss": 0.0502,
+      "step": 1809
+    },
+    {
+      "epoch": 164.55,
+      "learning_rate": 7.0622009569378e-05,
+      "loss": 0.0512,
+      "step": 1810
+    },
+    {
+      "epoch": 164.64,
+      "learning_rate": 7.060287081339714e-05,
+      "loss": 0.0466,
+      "step": 1811
+    },
+    {
+      "epoch": 164.73,
+      "learning_rate": 7.058373205741627e-05,
+      "loss": 0.0485,
+      "step": 1812
+    },
+    {
+      "epoch": 164.82,
+      "learning_rate": 7.056459330143542e-05,
+      "loss": 0.0444,
+      "step": 1813
+    },
+    {
+      "epoch": 164.91,
+      "learning_rate": 7.054545454545455e-05,
+      "loss": 0.0472,
+      "step": 1814
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 7.052631578947368e-05,
+      "loss": 0.0425,
+      "step": 1815
+    },
+    {
+      "epoch": 165.09,
+      "learning_rate": 7.050717703349283e-05,
+      "loss": 0.0481,
+      "step": 1816
+    },
+    {
+      "epoch": 165.18,
+      "learning_rate": 7.048803827751197e-05,
+      "loss": 0.0431,
+      "step": 1817
+    },
+    {
+      "epoch": 165.27,
+      "learning_rate": 7.04688995215311e-05,
+      "loss": 0.05,
+      "step": 1818
+    },
+    {
+      "epoch": 165.36,
+      "learning_rate": 7.044976076555025e-05,
+      "loss": 0.0479,
+      "step": 1819
+    },
+    {
+      "epoch": 165.45,
+      "learning_rate": 7.043062200956938e-05,
+      "loss": 0.0447,
+      "step": 1820
+    },
+    {
+      "epoch": 165.45,
+      "eval_accuracy_NAT": 0.9922640301576132,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9439326795141423,
+      "eval_accuracy_concrete tiles": 0.985598972665566,
+      "eval_accuracy_dark bricks": 0.8512374427466225,
+      "eval_accuracy_grass": 0.8828311494947871,
+      "eval_accuracy_red bricks": 0.9451349431818182,
+      "eval_accuracy_road": 0.915026475627791,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5713340501199239,
+      "eval_accuracy_speedway bricks": 0.9875836775774867,
+      "eval_accuracy_steel": 0.8870539542246554,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9784177243160237,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8773594667271714,
+      "eval_iou_concrete tiles": 0.9301419667590027,
+      "eval_iou_dark bricks": 0.8094872076424728,
+      "eval_iou_grass": 0.8214994243490591,
+      "eval_iou_red bricks": 0.8043215472952554,
+      "eval_iou_road": 0.9049644694576061,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3752172653534183,
+      "eval_iou_speedway bricks": 0.9742733920870055,
+      "eval_iou_steel": 0.8290106333795654,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10441258549690247,
+      "eval_mean_accuracy": 0.6893844134854159,
+      "eval_mean_iou": 0.6388225459512754,
+      "eval_overall_accuracy": 0.97075441148546,
+      "eval_runtime": 3.4346,
+      "eval_samples_per_second": 10.482,
+      "eval_steps_per_second": 0.582,
+      "step": 1820
+    },
+    {
+      "epoch": 165.55,
+      "learning_rate": 7.041148325358851e-05,
+      "loss": 0.0471,
+      "step": 1821
+    },
+    {
+      "epoch": 165.64,
+      "learning_rate": 7.039234449760766e-05,
+      "loss": 0.042,
+      "step": 1822
+    },
+    {
+      "epoch": 165.73,
+      "learning_rate": 7.03732057416268e-05,
+      "loss": 0.0415,
+      "step": 1823
+    },
+    {
+      "epoch": 165.82,
+      "learning_rate": 7.035406698564594e-05,
+      "loss": 0.0439,
+      "step": 1824
+    },
+    {
+      "epoch": 165.91,
+      "learning_rate": 7.033492822966508e-05,
+      "loss": 0.0468,
+      "step": 1825
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 7.031578947368421e-05,
+      "loss": 0.0486,
+      "step": 1826
+    },
+    {
+      "epoch": 166.09,
+      "learning_rate": 7.029665071770335e-05,
+      "loss": 0.0461,
+      "step": 1827
+    },
+    {
+      "epoch": 166.18,
+      "learning_rate": 7.027751196172249e-05,
+      "loss": 0.0411,
+      "step": 1828
+    },
+    {
+      "epoch": 166.27,
+      "learning_rate": 7.025837320574164e-05,
+      "loss": 0.0433,
+      "step": 1829
+    },
+    {
+      "epoch": 166.36,
+      "learning_rate": 7.023923444976077e-05,
+      "loss": 0.0479,
+      "step": 1830
+    },
+    {
+      "epoch": 166.45,
+      "learning_rate": 7.022009569377991e-05,
+      "loss": 0.0495,
+      "step": 1831
+    },
+    {
+      "epoch": 166.55,
+      "learning_rate": 7.020095693779905e-05,
+      "loss": 0.0442,
+      "step": 1832
+    },
+    {
+      "epoch": 166.64,
+      "learning_rate": 7.018181818181818e-05,
+      "loss": 0.0466,
+      "step": 1833
+    },
+    {
+      "epoch": 166.73,
+      "learning_rate": 7.016267942583732e-05,
+      "loss": 0.0452,
+      "step": 1834
+    },
+    {
+      "epoch": 166.82,
+      "learning_rate": 7.014354066985647e-05,
+      "loss": 0.0464,
+      "step": 1835
+    },
+    {
+      "epoch": 166.91,
+      "learning_rate": 7.01244019138756e-05,
+      "loss": 0.0567,
+      "step": 1836
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 7.010526315789473e-05,
+      "loss": 0.0468,
+      "step": 1837
+    },
+    {
+      "epoch": 167.09,
+      "learning_rate": 7.008612440191388e-05,
+      "loss": 0.0462,
+      "step": 1838
+    },
+    {
+      "epoch": 167.18,
+      "learning_rate": 7.006698564593301e-05,
+      "loss": 0.0456,
+      "step": 1839
+    },
+    {
+      "epoch": 167.27,
+      "learning_rate": 7.004784688995216e-05,
+      "loss": 0.0485,
+      "step": 1840
+    },
+    {
+      "epoch": 167.27,
+      "eval_accuracy_NAT": 0.992405896855147,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.944899727456114,
+      "eval_accuracy_concrete tiles": 0.982449703418333,
+      "eval_accuracy_dark bricks": 0.8715599861437204,
+      "eval_accuracy_grass": 0.888444237846755,
+      "eval_accuracy_red bricks": 0.9263139204545454,
+      "eval_accuracy_road": 0.8881457628717259,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5142668100239848,
+      "eval_accuracy_speedway bricks": 0.9870575059707203,
+      "eval_accuracy_steel": 0.8977475100587032,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787805300386256,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8791844711567262,
+      "eval_iou_concrete tiles": 0.9453924914675768,
+      "eval_iou_dark bricks": 0.805463664496852,
+      "eval_iou_grass": 0.8205844185190685,
+      "eval_iou_red bricks": 0.845954272742014,
+      "eval_iou_road": 0.8793363120037545,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3444620895963364,
+      "eval_iou_speedway bricks": 0.9747215515412685,
+      "eval_iou_steel": 0.8303858091773624,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12035813927650452,
+      "eval_mean_accuracy": 0.6840993123922884,
+      "eval_mean_iou": 0.6387896623645835,
+      "eval_overall_accuracy": 0.9685901006062826,
+      "eval_runtime": 3.5728,
+      "eval_samples_per_second": 10.076,
+      "eval_steps_per_second": 0.56,
+      "step": 1840
+    },
+    {
+      "epoch": 167.36,
+      "learning_rate": 7.00287081339713e-05,
+      "loss": 0.0456,
+      "step": 1841
+    },
+    {
+      "epoch": 167.45,
+      "learning_rate": 7.000956937799043e-05,
+      "loss": 0.0413,
+      "step": 1842
+    },
+    {
+      "epoch": 167.55,
+      "learning_rate": 6.999043062200957e-05,
+      "loss": 0.0451,
+      "step": 1843
+    },
+    {
+      "epoch": 167.64,
+      "learning_rate": 6.997129186602871e-05,
+      "loss": 0.0407,
+      "step": 1844
+    },
+    {
+      "epoch": 167.73,
+      "learning_rate": 6.995215311004784e-05,
+      "loss": 0.0418,
+      "step": 1845
+    },
+    {
+      "epoch": 167.82,
+      "learning_rate": 6.993301435406699e-05,
+      "loss": 0.0421,
+      "step": 1846
+    },
+    {
+      "epoch": 167.91,
+      "learning_rate": 6.991387559808614e-05,
+      "loss": 0.0467,
+      "step": 1847
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 6.989473684210527e-05,
+      "loss": 0.047,
+      "step": 1848
+    },
+    {
+      "epoch": 168.09,
+      "learning_rate": 6.98755980861244e-05,
+      "loss": 0.0562,
+      "step": 1849
+    },
+    {
+      "epoch": 168.18,
+      "learning_rate": 6.985645933014354e-05,
+      "loss": 0.0459,
+      "step": 1850
+    },
+    {
+      "epoch": 168.27,
+      "learning_rate": 6.983732057416268e-05,
+      "loss": 0.042,
+      "step": 1851
+    },
+    {
+      "epoch": 168.36,
+      "learning_rate": 6.981818181818182e-05,
+      "loss": 0.0459,
+      "step": 1852
+    },
+    {
+      "epoch": 168.45,
+      "learning_rate": 6.979904306220097e-05,
+      "loss": 0.0459,
+      "step": 1853
+    },
+    {
+      "epoch": 168.55,
+      "learning_rate": 6.97799043062201e-05,
+      "loss": 0.0546,
+      "step": 1854
+    },
+    {
+      "epoch": 168.64,
+      "learning_rate": 6.976076555023923e-05,
+      "loss": 0.0454,
+      "step": 1855
+    },
+    {
+      "epoch": 168.73,
+      "learning_rate": 6.974162679425838e-05,
+      "loss": 0.043,
+      "step": 1856
+    },
+    {
+      "epoch": 168.82,
+      "learning_rate": 6.972248803827751e-05,
+      "loss": 0.0502,
+      "step": 1857
+    },
+    {
+      "epoch": 168.91,
+      "learning_rate": 6.970334928229666e-05,
+      "loss": 0.0527,
+      "step": 1858
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 6.96842105263158e-05,
+      "loss": 0.0396,
+      "step": 1859
+    },
+    {
+      "epoch": 169.09,
+      "learning_rate": 6.966507177033493e-05,
+      "loss": 0.0461,
+      "step": 1860
+    },
+    {
+      "epoch": 169.09,
+      "eval_accuracy_NAT": 0.9925281774787676,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9475098180179407,
+      "eval_accuracy_concrete tiles": 0.9845594080596832,
+      "eval_accuracy_dark bricks": 0.844771178938455,
+      "eval_accuracy_grass": 0.893718304050441,
+      "eval_accuracy_red bricks": 0.9366122159090909,
+      "eval_accuracy_road": 0.8708419842149877,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6013288120640697,
+      "eval_accuracy_speedway bricks": 0.9849799728949213,
+      "eval_accuracy_steel": 0.9035106523316404,
+      "eval_accuracy_tiles": 0.011505273250239693,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9782664537706919,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.885822733382822,
+      "eval_iou_concrete tiles": 0.9331188965197484,
+      "eval_iou_dark bricks": 0.8058452048758995,
+      "eval_iou_grass": 0.8221804946432935,
+      "eval_iou_red bricks": 0.8450817045818648,
+      "eval_iou_road": 0.8638630725900478,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.39869125737995575,
+      "eval_iou_speedway bricks": 0.9745276051937011,
+      "eval_iou_steel": 0.8296363030706801,
+      "eval_iou_tiles": 0.011505273250239693,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13130919635295868,
+      "eval_mean_accuracy": 0.690143522862326,
+      "eval_mean_iou": 0.6421953076353033,
+      "eval_overall_accuracy": 0.9675146738688151,
+      "eval_runtime": 3.8455,
+      "eval_samples_per_second": 9.362,
+      "eval_steps_per_second": 0.52,
+      "step": 1860
+    },
+    {
+      "epoch": 169.18,
+      "learning_rate": 6.964593301435406e-05,
+      "loss": 0.0413,
+      "step": 1861
+    },
+    {
+      "epoch": 169.27,
+      "learning_rate": 6.962679425837321e-05,
+      "loss": 0.0415,
+      "step": 1862
+    },
+    {
+      "epoch": 169.36,
+      "learning_rate": 6.960765550239234e-05,
+      "loss": 0.0516,
+      "step": 1863
+    },
+    {
+      "epoch": 169.45,
+      "learning_rate": 6.958851674641149e-05,
+      "loss": 0.0424,
+      "step": 1864
+    },
+    {
+      "epoch": 169.55,
+      "learning_rate": 6.956937799043063e-05,
+      "loss": 0.039,
+      "step": 1865
+    },
+    {
+      "epoch": 169.64,
+      "learning_rate": 6.955023923444977e-05,
+      "loss": 0.0479,
+      "step": 1866
+    },
+    {
+      "epoch": 169.73,
+      "learning_rate": 6.95311004784689e-05,
+      "loss": 0.0488,
+      "step": 1867
+    },
+    {
+      "epoch": 169.82,
+      "learning_rate": 6.951196172248804e-05,
+      "loss": 0.0445,
+      "step": 1868
+    },
+    {
+      "epoch": 169.91,
+      "learning_rate": 6.949282296650717e-05,
+      "loss": 0.0417,
+      "step": 1869
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 6.947368421052632e-05,
+      "loss": 0.0435,
+      "step": 1870
+    },
+    {
+      "epoch": 170.09,
+      "learning_rate": 6.945454545454547e-05,
+      "loss": 0.0461,
+      "step": 1871
+    },
+    {
+      "epoch": 170.18,
+      "learning_rate": 6.94354066985646e-05,
+      "loss": 0.0425,
+      "step": 1872
+    },
+    {
+      "epoch": 170.27,
+      "learning_rate": 6.941626794258373e-05,
+      "loss": 0.05,
+      "step": 1873
+    },
+    {
+      "epoch": 170.36,
+      "learning_rate": 6.939712918660288e-05,
+      "loss": 0.0407,
+      "step": 1874
+    },
+    {
+      "epoch": 170.45,
+      "learning_rate": 6.937799043062201e-05,
+      "loss": 0.044,
+      "step": 1875
+    },
+    {
+      "epoch": 170.55,
+      "learning_rate": 6.935885167464115e-05,
+      "loss": 0.038,
+      "step": 1876
+    },
+    {
+      "epoch": 170.64,
+      "learning_rate": 6.93397129186603e-05,
+      "loss": 0.0429,
+      "step": 1877
+    },
+    {
+      "epoch": 170.73,
+      "learning_rate": 6.932057416267943e-05,
+      "loss": 0.0451,
+      "step": 1878
+    },
+    {
+      "epoch": 170.82,
+      "learning_rate": 6.930143540669856e-05,
+      "loss": 0.0394,
+      "step": 1879
+    },
+    {
+      "epoch": 170.91,
+      "learning_rate": 6.928229665071771e-05,
+      "loss": 0.0443,
+      "step": 1880
+    },
+    {
+      "epoch": 170.91,
+      "eval_accuracy_NAT": 0.9920214098726517,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.943066226018284,
+      "eval_accuracy_concrete tiles": 0.9879838561731792,
+      "eval_accuracy_dark bricks": 0.8724837381163157,
+      "eval_accuracy_grass": 0.8903630333045572,
+      "eval_accuracy_red bricks": 0.9209872159090909,
+      "eval_accuracy_road": 0.9294063872829924,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46177597662172964,
+      "eval_accuracy_speedway bricks": 0.9887367898946088,
+      "eval_accuracy_steel": 0.8923966097223138,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9785171190347933,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8746502928894235,
+      "eval_iou_concrete tiles": 0.9393587022878572,
+      "eval_iou_dark bricks": 0.8114551637730446,
+      "eval_iou_grass": 0.8182545265328541,
+      "eval_iou_red bricks": 0.8130094043887147,
+      "eval_iou_road": 0.9230565069727,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3031564468254543,
+      "eval_iou_speedway bricks": 0.9741363884160983,
+      "eval_iou_steel": 0.8253231156353654,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1050824448466301,
+      "eval_mean_accuracy": 0.6830170186858248,
+      "eval_mean_iou": 0.6354552051351003,
+      "eval_overall_accuracy": 0.9719311396280924,
+      "eval_runtime": 3.224,
+      "eval_samples_per_second": 11.166,
+      "eval_steps_per_second": 0.62,
+      "step": 1880
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 6.926315789473684e-05,
+      "loss": 0.049,
+      "step": 1881
+    },
+    {
+      "epoch": 171.09,
+      "learning_rate": 6.924401913875599e-05,
+      "loss": 0.0426,
+      "step": 1882
+    },
+    {
+      "epoch": 171.18,
+      "learning_rate": 6.922488038277513e-05,
+      "loss": 0.0481,
+      "step": 1883
+    },
+    {
+      "epoch": 171.27,
+      "learning_rate": 6.920574162679426e-05,
+      "loss": 0.045,
+      "step": 1884
+    },
+    {
+      "epoch": 171.36,
+      "learning_rate": 6.91866028708134e-05,
+      "loss": 0.0374,
+      "step": 1885
+    },
+    {
+      "epoch": 171.45,
+      "learning_rate": 6.916746411483254e-05,
+      "loss": 0.0448,
+      "step": 1886
+    },
+    {
+      "epoch": 171.55,
+      "learning_rate": 6.914832535885167e-05,
+      "loss": 0.0444,
+      "step": 1887
+    },
+    {
+      "epoch": 171.64,
+      "learning_rate": 6.912918660287082e-05,
+      "loss": 0.0417,
+      "step": 1888
+    },
+    {
+      "epoch": 171.73,
+      "learning_rate": 6.911004784688996e-05,
+      "loss": 0.0443,
+      "step": 1889
+    },
+    {
+      "epoch": 171.82,
+      "learning_rate": 6.90909090909091e-05,
+      "loss": 0.0411,
+      "step": 1890
+    },
+    {
+      "epoch": 171.91,
+      "learning_rate": 6.907177033492823e-05,
+      "loss": 0.0447,
+      "step": 1891
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 6.905263157894737e-05,
+      "loss": 0.0451,
+      "step": 1892
+    },
+    {
+      "epoch": 172.09,
+      "learning_rate": 6.90334928229665e-05,
+      "loss": 0.0388,
+      "step": 1893
+    },
+    {
+      "epoch": 172.18,
+      "learning_rate": 6.901435406698565e-05,
+      "loss": 0.0456,
+      "step": 1894
+    },
+    {
+      "epoch": 172.27,
+      "learning_rate": 6.89952153110048e-05,
+      "loss": 0.0444,
+      "step": 1895
+    },
+    {
+      "epoch": 172.36,
+      "learning_rate": 6.897607655502393e-05,
+      "loss": 0.0425,
+      "step": 1896
+    },
+    {
+      "epoch": 172.45,
+      "learning_rate": 6.895693779904306e-05,
+      "loss": 0.0542,
+      "step": 1897
+    },
+    {
+      "epoch": 172.55,
+      "learning_rate": 6.89377990430622e-05,
+      "loss": 0.0445,
+      "step": 1898
+    },
+    {
+      "epoch": 172.64,
+      "learning_rate": 6.891866028708134e-05,
+      "loss": 0.0402,
+      "step": 1899
+    },
+    {
+      "epoch": 172.73,
+      "learning_rate": 6.889952153110048e-05,
+      "loss": 0.0435,
+      "step": 1900
+    },
+    {
+      "epoch": 172.73,
+      "eval_accuracy_NAT": 0.992198743244569,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9460116314004893,
+      "eval_accuracy_concrete tiles": 0.9903381642512077,
+      "eval_accuracy_dark bricks": 0.8757553596859243,
+      "eval_accuracy_grass": 0.9039029487936753,
+      "eval_accuracy_red bricks": 0.9556107954545454,
+      "eval_accuracy_road": 0.9051303103918199,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.47864803021531166,
+      "eval_accuracy_speedway bricks": 0.9865753831337863,
+      "eval_accuracy_steel": 0.8964778048941363,
+      "eval_accuracy_tiles": 0.0007533214628133132,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787434839117086,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8756522548806605,
+      "eval_iou_concrete tiles": 0.9229760920981392,
+      "eval_iou_dark bricks": 0.7989115168539326,
+      "eval_iou_grass": 0.8330659760473641,
+      "eval_iou_red bricks": 0.8367537313432836,
+      "eval_iou_road": 0.898967881093753,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31242352263729933,
+      "eval_iou_speedway bricks": 0.9745921480945452,
+      "eval_iou_steel": 0.8308461132888614,
+      "eval_iou_tiles": 0.0007533214628133132,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10976611822843552,
+      "eval_mean_accuracy": 0.6870309609944829,
+      "eval_mean_iou": 0.6356681570547971,
+      "eval_overall_accuracy": 0.9703302383422852,
+      "eval_runtime": 3.3141,
+      "eval_samples_per_second": 10.863,
+      "eval_steps_per_second": 0.603,
+      "step": 1900
+    },
+    {
+      "epoch": 172.82,
+      "learning_rate": 6.888038277511963e-05,
+      "loss": 0.0536,
+      "step": 1901
+    },
+    {
+      "epoch": 172.91,
+      "learning_rate": 6.886124401913876e-05,
+      "loss": 0.0433,
+      "step": 1902
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 6.88421052631579e-05,
+      "loss": 0.0414,
+      "step": 1903
+    },
+    {
+      "epoch": 173.09,
+      "learning_rate": 6.882296650717704e-05,
+      "loss": 0.0437,
+      "step": 1904
+    },
+    {
+      "epoch": 173.18,
+      "learning_rate": 6.880382775119617e-05,
+      "loss": 0.0441,
+      "step": 1905
+    },
+    {
+      "epoch": 173.27,
+      "learning_rate": 6.878468899521532e-05,
+      "loss": 0.044,
+      "step": 1906
+    },
+    {
+      "epoch": 173.36,
+      "learning_rate": 6.876555023923446e-05,
+      "loss": 0.0419,
+      "step": 1907
+    },
+    {
+      "epoch": 173.45,
+      "learning_rate": 6.87464114832536e-05,
+      "loss": 0.0438,
+      "step": 1908
+    },
+    {
+      "epoch": 173.55,
+      "learning_rate": 6.872727272727273e-05,
+      "loss": 0.0413,
+      "step": 1909
+    },
+    {
+      "epoch": 173.64,
+      "learning_rate": 6.870813397129187e-05,
+      "loss": 0.0425,
+      "step": 1910
+    },
+    {
+      "epoch": 173.73,
+      "learning_rate": 6.8688995215311e-05,
+      "loss": 0.0454,
+      "step": 1911
+    },
+    {
+      "epoch": 173.82,
+      "learning_rate": 6.866985645933015e-05,
+      "loss": 0.0394,
+      "step": 1912
+    },
+    {
+      "epoch": 173.91,
+      "learning_rate": 6.865071770334928e-05,
+      "loss": 0.047,
+      "step": 1913
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 6.863157894736843e-05,
+      "loss": 0.0558,
+      "step": 1914
+    },
+    {
+      "epoch": 174.09,
+      "learning_rate": 6.861244019138756e-05,
+      "loss": 0.045,
+      "step": 1915
+    },
+    {
+      "epoch": 174.18,
+      "learning_rate": 6.85933014354067e-05,
+      "loss": 0.047,
+      "step": 1916
+    },
+    {
+      "epoch": 174.27,
+      "learning_rate": 6.857416267942584e-05,
+      "loss": 0.0484,
+      "step": 1917
+    },
+    {
+      "epoch": 174.36,
+      "learning_rate": 6.855502392344498e-05,
+      "loss": 0.0379,
+      "step": 1918
+    },
+    {
+      "epoch": 174.45,
+      "learning_rate": 6.853588516746411e-05,
+      "loss": 0.052,
+      "step": 1919
+    },
+    {
+      "epoch": 174.55,
+      "learning_rate": 6.851674641148326e-05,
+      "loss": 0.0406,
+      "step": 1920
+    },
+    {
+      "epoch": 174.55,
+      "eval_accuracy_NAT": 0.9925837595804133,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9485492939611142,
+      "eval_accuracy_concrete tiles": 0.9919586620192014,
+      "eval_accuracy_dark bricks": 0.8798352642315538,
+      "eval_accuracy_grass": 0.8833868662120741,
+      "eval_accuracy_red bricks": 0.9554332386363636,
+      "eval_accuracy_road": 0.8674646638943538,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4360543655060238,
+      "eval_accuracy_speedway bricks": 0.983531794160702,
+      "eval_accuracy_steel": 0.9023398852318448,
+      "eval_accuracy_tiles": 0.015819750719079578,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.978735546489165,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8694489044723493,
+      "eval_iou_concrete tiles": 0.9024478442280945,
+      "eval_iou_dark bricks": 0.8014796115143228,
+      "eval_iou_grass": 0.8222491875168759,
+      "eval_iou_red bricks": 0.820649687357023,
+      "eval_iou_road": 0.8602888887556438,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2921823622861787,
+      "eval_iou_speedway bricks": 0.9738587513405806,
+      "eval_iou_steel": 0.8313431475320177,
+      "eval_iou_tiles": 0.015819750719079578,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14024387300014496,
+      "eval_mean_accuracy": 0.6813044264732865,
+      "eval_mean_iou": 0.6283464370931793,
+      "eval_overall_accuracy": 0.9662564595540365,
+      "eval_runtime": 3.5891,
+      "eval_samples_per_second": 10.03,
+      "eval_steps_per_second": 0.557,
+      "step": 1920
+    },
+    {
+      "epoch": 174.64,
+      "learning_rate": 6.849760765550239e-05,
+      "loss": 0.0548,
+      "step": 1921
+    },
+    {
+      "epoch": 174.73,
+      "learning_rate": 6.847846889952154e-05,
+      "loss": 0.0469,
+      "step": 1922
+    },
+    {
+      "epoch": 174.82,
+      "learning_rate": 6.845933014354068e-05,
+      "loss": 0.042,
+      "step": 1923
+    },
+    {
+      "epoch": 174.91,
+      "learning_rate": 6.844019138755981e-05,
+      "loss": 0.0409,
+      "step": 1924
+    },
+    {
+      "epoch": 175.0,
+      "learning_rate": 6.842105263157895e-05,
+      "loss": 0.0435,
+      "step": 1925
+    },
+    {
+      "epoch": 175.09,
+      "learning_rate": 6.840191387559809e-05,
+      "loss": 0.0472,
+      "step": 1926
+    },
+    {
+      "epoch": 175.18,
+      "learning_rate": 6.838277511961722e-05,
+      "loss": 0.0385,
+      "step": 1927
+    },
+    {
+      "epoch": 175.27,
+      "learning_rate": 6.836363636363637e-05,
+      "loss": 0.0423,
+      "step": 1928
+    },
+    {
+      "epoch": 175.36,
+      "learning_rate": 6.834449760765552e-05,
+      "loss": 0.0451,
+      "step": 1929
+    },
+    {
+      "epoch": 175.45,
+      "learning_rate": 6.832535885167465e-05,
+      "loss": 0.0493,
+      "step": 1930
+    },
+    {
+      "epoch": 175.55,
+      "learning_rate": 6.830622009569378e-05,
+      "loss": 0.0376,
+      "step": 1931
+    },
+    {
+      "epoch": 175.64,
+      "learning_rate": 6.828708133971292e-05,
+      "loss": 0.039,
+      "step": 1932
+    },
+    {
+      "epoch": 175.73,
+      "learning_rate": 6.826794258373206e-05,
+      "loss": 0.0433,
+      "step": 1933
+    },
+    {
+      "epoch": 175.82,
+      "learning_rate": 6.82488038277512e-05,
+      "loss": 0.0486,
+      "step": 1934
+    },
+    {
+      "epoch": 175.91,
+      "learning_rate": 6.822966507177035e-05,
+      "loss": 0.0433,
+      "step": 1935
+    },
+    {
+      "epoch": 176.0,
+      "learning_rate": 6.821052631578948e-05,
+      "loss": 0.0398,
+      "step": 1936
+    },
+    {
+      "epoch": 176.09,
+      "learning_rate": 6.819138755980861e-05,
+      "loss": 0.0389,
+      "step": 1937
+    },
+    {
+      "epoch": 176.18,
+      "learning_rate": 6.817224880382776e-05,
+      "loss": 0.0493,
+      "step": 1938
+    },
+    {
+      "epoch": 176.27,
+      "learning_rate": 6.815311004784689e-05,
+      "loss": 0.0439,
+      "step": 1939
+    },
+    {
+      "epoch": 176.36,
+      "learning_rate": 6.813397129186602e-05,
+      "loss": 0.0405,
+      "step": 1940
+    },
+    {
+      "epoch": 176.36,
+      "eval_accuracy_NAT": 0.9919912367317583,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9462087965148719,
+      "eval_accuracy_concrete tiles": 0.9792087078823457,
+      "eval_accuracy_dark bricks": 0.8778722912897887,
+      "eval_accuracy_grass": 0.9040637221835822,
+      "eval_accuracy_red bricks": 0.9209872159090909,
+      "eval_accuracy_road": 0.9058135133925926,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5158106580652276,
+      "eval_accuracy_speedway bricks": 0.9873181781428797,
+      "eval_accuracy_steel": 0.8948453268254073,
+      "eval_accuracy_tiles": 0.001712094233666621,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790039935755902,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8802783597389099,
+      "eval_iou_concrete tiles": 0.949565629910754,
+      "eval_iou_dark bricks": 0.8014618033593366,
+      "eval_iou_grass": 0.8317299035369775,
+      "eval_iou_red bricks": 0.8453389830508474,
+      "eval_iou_road": 0.8977649835031358,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33883264818269077,
+      "eval_iou_speedway bricks": 0.9749040281656406,
+      "eval_iou_steel": 0.8308695752026763,
+      "eval_iou_tiles": 0.001712094233666621,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11012466251850128,
+      "eval_mean_accuracy": 0.6866024416285548,
+      "eval_mean_iou": 0.6408816924969405,
+      "eval_overall_accuracy": 0.9704863230387369,
+      "eval_runtime": 3.4713,
+      "eval_samples_per_second": 10.371,
+      "eval_steps_per_second": 0.576,
+      "step": 1940
+    },
+    {
+      "epoch": 176.45,
+      "learning_rate": 6.811483253588518e-05,
+      "loss": 0.0423,
+      "step": 1941
+    },
+    {
+      "epoch": 176.55,
+      "learning_rate": 6.809569377990431e-05,
+      "loss": 0.047,
+      "step": 1942
+    },
+    {
+      "epoch": 176.64,
+      "learning_rate": 6.807655502392344e-05,
+      "loss": 0.0449,
+      "step": 1943
+    },
+    {
+      "epoch": 176.73,
+      "learning_rate": 6.805741626794259e-05,
+      "loss": 0.0478,
+      "step": 1944
+    },
+    {
+      "epoch": 176.82,
+      "learning_rate": 6.803827751196172e-05,
+      "loss": 0.0397,
+      "step": 1945
+    },
+    {
+      "epoch": 176.91,
+      "learning_rate": 6.801913875598085e-05,
+      "loss": 0.0389,
+      "step": 1946
+    },
+    {
+      "epoch": 177.0,
+      "learning_rate": 6.800000000000001e-05,
+      "loss": 0.0403,
+      "step": 1947
+    },
+    {
+      "epoch": 177.09,
+      "learning_rate": 6.798086124401915e-05,
+      "loss": 0.0403,
+      "step": 1948
+    },
+    {
+      "epoch": 177.18,
+      "learning_rate": 6.796172248803828e-05,
+      "loss": 0.047,
+      "step": 1949
+    },
+    {
+      "epoch": 177.27,
+      "learning_rate": 6.794258373205742e-05,
+      "loss": 0.0476,
+      "step": 1950
+    },
+    {
+      "epoch": 177.36,
+      "learning_rate": 6.792344497607655e-05,
+      "loss": 0.0412,
+      "step": 1951
+    },
+    {
+      "epoch": 177.45,
+      "learning_rate": 6.790430622009569e-05,
+      "loss": 0.0442,
+      "step": 1952
+    },
+    {
+      "epoch": 177.55,
+      "learning_rate": 6.788516746411485e-05,
+      "loss": 0.0509,
+      "step": 1953
+    },
+    {
+      "epoch": 177.64,
+      "learning_rate": 6.786602870813398e-05,
+      "loss": 0.0379,
+      "step": 1954
+    },
+    {
+      "epoch": 177.73,
+      "learning_rate": 6.784688995215311e-05,
+      "loss": 0.0379,
+      "step": 1955
+    },
+    {
+      "epoch": 177.82,
+      "learning_rate": 6.782775119617226e-05,
+      "loss": 0.0422,
+      "step": 1956
+    },
+    {
+      "epoch": 177.91,
+      "learning_rate": 6.780861244019139e-05,
+      "loss": 0.0448,
+      "step": 1957
+    },
+    {
+      "epoch": 178.0,
+      "learning_rate": 6.778947368421052e-05,
+      "loss": 0.0479,
+      "step": 1958
+    },
+    {
+      "epoch": 178.09,
+      "learning_rate": 6.777033492822968e-05,
+      "loss": 0.0409,
+      "step": 1959
+    },
+    {
+      "epoch": 178.18,
+      "learning_rate": 6.775119617224881e-05,
+      "loss": 0.0406,
+      "step": 1960
+    },
+    {
+      "epoch": 178.18,
+      "eval_accuracy_NAT": 0.9917911411658337,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9413748980642946,
+      "eval_accuracy_concrete tiles": 0.9896960802299273,
+      "eval_accuracy_dark bricks": 0.883183865132212,
+      "eval_accuracy_grass": 0.9071044362970393,
+      "eval_accuracy_red bricks": 0.9396306818181818,
+      "eval_accuracy_road": 0.9390425726160145,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5455848702891958,
+      "eval_accuracy_speedway bricks": 0.9902477230406443,
+      "eval_accuracy_steel": 0.8801612690455775,
+      "eval_accuracy_tiles": 0.019449390494452816,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787979952365885,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8794255771293391,
+      "eval_iou_concrete tiles": 0.9280369276642105,
+      "eval_iou_dark bricks": 0.8102115038310794,
+      "eval_iou_grass": 0.8313782521510165,
+      "eval_iou_red bricks": 0.8349637109498265,
+      "eval_iou_road": 0.9326055758711173,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35438005873504763,
+      "eval_iou_speedway bricks": 0.9742153138536812,
+      "eval_iou_steel": 0.8268760069401413,
+      "eval_iou_tiles": 0.019449390494452816,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09781524538993835,
+      "eval_mean_accuracy": 0.6944051483225674,
+      "eval_mean_iou": 0.6438723317581924,
+      "eval_overall_accuracy": 0.9735180536905924,
+      "eval_runtime": 3.4973,
+      "eval_samples_per_second": 10.294,
+      "eval_steps_per_second": 0.572,
+      "step": 1960
+    },
+    {
+      "epoch": 178.27,
+      "learning_rate": 6.773205741626794e-05,
+      "loss": 0.0449,
+      "step": 1961
+    },
+    {
+      "epoch": 178.36,
+      "learning_rate": 6.771291866028709e-05,
+      "loss": 0.0405,
+      "step": 1962
+    },
+    {
+      "epoch": 178.45,
+      "learning_rate": 6.769377990430622e-05,
+      "loss": 0.0373,
+      "step": 1963
+    },
+    {
+      "epoch": 178.55,
+      "learning_rate": 6.767464114832535e-05,
+      "loss": 0.0467,
+      "step": 1964
+    },
+    {
+      "epoch": 178.64,
+      "learning_rate": 6.765550239234451e-05,
+      "loss": 0.0463,
+      "step": 1965
+    },
+    {
+      "epoch": 178.73,
+      "learning_rate": 6.763636363636364e-05,
+      "loss": 0.0394,
+      "step": 1966
+    },
+    {
+      "epoch": 178.82,
+      "learning_rate": 6.761722488038278e-05,
+      "loss": 0.0481,
+      "step": 1967
+    },
+    {
+      "epoch": 178.91,
+      "learning_rate": 6.759808612440192e-05,
+      "loss": 0.0466,
+      "step": 1968
+    },
+    {
+      "epoch": 179.0,
+      "learning_rate": 6.757894736842105e-05,
+      "loss": 0.0438,
+      "step": 1969
+    },
+    {
+      "epoch": 179.09,
+      "learning_rate": 6.755980861244018e-05,
+      "loss": 0.0421,
+      "step": 1970
+    },
+    {
+      "epoch": 179.18,
+      "learning_rate": 6.754066985645934e-05,
+      "loss": 0.0429,
+      "step": 1971
+    },
+    {
+      "epoch": 179.27,
+      "learning_rate": 6.752153110047848e-05,
+      "loss": 0.0429,
+      "step": 1972
+    },
+    {
+      "epoch": 179.36,
+      "learning_rate": 6.750239234449761e-05,
+      "loss": 0.0399,
+      "step": 1973
+    },
+    {
+      "epoch": 179.45,
+      "learning_rate": 6.748325358851675e-05,
+      "loss": 0.0518,
+      "step": 1974
+    },
+    {
+      "epoch": 179.55,
+      "learning_rate": 6.746411483253589e-05,
+      "loss": 0.0502,
+      "step": 1975
+    },
+    {
+      "epoch": 179.64,
+      "learning_rate": 6.744497607655502e-05,
+      "loss": 0.044,
+      "step": 1976
+    },
+    {
+      "epoch": 179.73,
+      "learning_rate": 6.742583732057418e-05,
+      "loss": 0.0373,
+      "step": 1977
+    },
+    {
+      "epoch": 179.82,
+      "learning_rate": 6.740669856459331e-05,
+      "loss": 0.0407,
+      "step": 1978
+    },
+    {
+      "epoch": 179.91,
+      "learning_rate": 6.738755980861244e-05,
+      "loss": 0.0534,
+      "step": 1979
+    },
+    {
+      "epoch": 180.0,
+      "learning_rate": 6.736842105263159e-05,
+      "loss": 0.0456,
+      "step": 1980
+    },
+    {
+      "epoch": 180.0,
+      "eval_accuracy_NAT": 0.9922395034524425,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9444128503369243,
+      "eval_accuracy_concrete tiles": 0.9868219898489574,
+      "eval_accuracy_dark bricks": 0.8744082213925561,
+      "eval_accuracy_grass": 0.8798813072973644,
+      "eval_accuracy_red bricks": 0.9273792613636364,
+      "eval_accuracy_road": 0.9261717536067677,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46905411738758856,
+      "eval_accuracy_speedway bricks": 0.9875474731091312,
+      "eval_accuracy_steel": 0.8952823032781478,
+      "eval_accuracy_tiles": 0.13347486645664977,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979057033751834,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8721908076873053,
+      "eval_iou_concrete tiles": 0.929285076732602,
+      "eval_iou_dark bricks": 0.8041769911504425,
+      "eval_iou_grass": 0.812185246721404,
+      "eval_iou_red bricks": 0.8410628019323672,
+      "eval_iou_road": 0.9185334778812851,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31319490464619687,
+      "eval_iou_speedway bricks": 0.9746629568796816,
+      "eval_iou_steel": 0.8296937559216357,
+      "eval_iou_tiles": 0.1334657262206396,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1043887734413147,
+      "eval_mean_accuracy": 0.6935902805792435,
+      "eval_mean_iou": 0.6467314445788765,
+      "eval_overall_accuracy": 0.9716350767347548,
+      "eval_runtime": 3.5826,
+      "eval_samples_per_second": 10.049,
+      "eval_steps_per_second": 0.558,
+      "step": 1980
+    },
+    {
+      "epoch": 180.09,
+      "learning_rate": 6.734928229665072e-05,
+      "loss": 0.0367,
+      "step": 1981
+    },
+    {
+      "epoch": 180.18,
+      "learning_rate": 6.733014354066985e-05,
+      "loss": 0.0422,
+      "step": 1982
+    },
+    {
+      "epoch": 180.27,
+      "learning_rate": 6.731100478468901e-05,
+      "loss": 0.0443,
+      "step": 1983
+    },
+    {
+      "epoch": 180.36,
+      "learning_rate": 6.729186602870814e-05,
+      "loss": 0.0525,
+      "step": 1984
+    },
+    {
+      "epoch": 180.45,
+      "learning_rate": 6.727272727272727e-05,
+      "loss": 0.0429,
+      "step": 1985
+    },
+    {
+      "epoch": 180.55,
+      "learning_rate": 6.725358851674642e-05,
+      "loss": 0.0385,
+      "step": 1986
+    },
+    {
+      "epoch": 180.64,
+      "learning_rate": 6.723444976076555e-05,
+      "loss": 0.0416,
+      "step": 1987
+    },
+    {
+      "epoch": 180.73,
+      "learning_rate": 6.721531100478468e-05,
+      "loss": 0.0439,
+      "step": 1988
+    },
+    {
+      "epoch": 180.82,
+      "learning_rate": 6.719617224880383e-05,
+      "loss": 0.0423,
+      "step": 1989
+    },
+    {
+      "epoch": 180.91,
+      "learning_rate": 6.717703349282297e-05,
+      "loss": 0.0428,
+      "step": 1990
+    },
+    {
+      "epoch": 181.0,
+      "learning_rate": 6.71578947368421e-05,
+      "loss": 0.0434,
+      "step": 1991
+    },
+    {
+      "epoch": 181.09,
+      "learning_rate": 6.713875598086125e-05,
+      "loss": 0.0376,
+      "step": 1992
+    },
+    {
+      "epoch": 181.18,
+      "learning_rate": 6.711961722488038e-05,
+      "loss": 0.048,
+      "step": 1993
+    },
+    {
+      "epoch": 181.27,
+      "learning_rate": 6.710047846889953e-05,
+      "loss": 0.0425,
+      "step": 1994
+    },
+    {
+      "epoch": 181.36,
+      "learning_rate": 6.708133971291866e-05,
+      "loss": 0.0361,
+      "step": 1995
+    },
+    {
+      "epoch": 181.45,
+      "learning_rate": 6.70622009569378e-05,
+      "loss": 0.0485,
+      "step": 1996
+    },
+    {
+      "epoch": 181.55,
+      "learning_rate": 6.704306220095694e-05,
+      "loss": 0.0415,
+      "step": 1997
+    },
+    {
+      "epoch": 181.64,
+      "learning_rate": 6.702392344497608e-05,
+      "loss": 0.0435,
+      "step": 1998
+    },
+    {
+      "epoch": 181.73,
+      "learning_rate": 6.700478468899522e-05,
+      "loss": 0.0434,
+      "step": 1999
+    },
+    {
+      "epoch": 181.82,
+      "learning_rate": 6.698564593301436e-05,
+      "loss": 0.042,
+      "step": 2000
+    },
+    {
+      "epoch": 181.82,
+      "eval_accuracy_NAT": 0.9927947951155508,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9445469762650758,
+      "eval_accuracy_concrete tiles": 0.9852014920809637,
+      "eval_accuracy_dark bricks": 0.8683653439051615,
+      "eval_accuracy_grass": 0.8680225222548817,
+      "eval_accuracy_red bricks": 0.9385653409090909,
+      "eval_accuracy_road": 0.9425178388535733,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49596118324924876,
+      "eval_accuracy_speedway bricks": 0.9886082640319468,
+      "eval_accuracy_steel": 0.9001714926456038,
+      "eval_accuracy_tiles": 0.08033146144363786,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787256693140206,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8760236202351823,
+      "eval_iou_concrete tiles": 0.9470373853750294,
+      "eval_iou_dark bricks": 0.802996867881549,
+      "eval_iou_grass": 0.8114671075417078,
+      "eval_iou_red bricks": 0.8076394194041253,
+      "eval_iou_road": 0.9342271486205632,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3328030190913127,
+      "eval_iou_speedway bricks": 0.9749794694183597,
+      "eval_iou_steel": 0.8353289518985793,
+      "eval_iou_tiles": 0.08033146144363786,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09672261029481888,
+      "eval_mean_accuracy": 0.6926989777503644,
+      "eval_mean_iou": 0.6447353938633898,
+      "eval_overall_accuracy": 0.973306761847602,
+      "eval_runtime": 3.773,
+      "eval_samples_per_second": 9.541,
+      "eval_steps_per_second": 0.53,
+      "step": 2000
     }
   ],
   "max_steps": 5500,
   "num_train_epochs": 500,
-  "total_flos": 9.404187505978245e+17,
+  "total_flos": 1.1195452963065692e+18,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:22a8e67d23f94c536bd5f5106f8d0def6b5bf65b04b5f47f1af2e3f713990018
+oid sha256:7a89a602af7a559cba27b8f9e51253d3ee9a3753bb186dff447f69ecd6684727
 size 14944077

runs/May20_21-20-16_robolidar/events.out.tfevents.1716258023.robolidar.3008813.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e22bd96786bbec04ab664f0c81c7e023f79b7feb2f60c7268f93b0d32bb1269c
-size 508026
+oid sha256:d73c44d74aa0698e3509494fff6a546e2969f8bf5614cf0ca42e0173ff88e8dc
+size 529002