Training in progress, step 2240

{checkpoint-1920 → checkpoint-2240}/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3348d709bbceaf1320f1156a9b20b7801a47b5c5a7874c23f3332d3b812e5091
+oid sha256:86690679625bff114e6224583865f266e9d3ecb0ee7687b7d5d49cd320e6a2bf
 size 29861573

{checkpoint-1920 → checkpoint-2240}/pytorch_model.bin

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

{checkpoint-1920 → checkpoint-2240}/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:40df47160d89b3047cc1aebc3c618723a0a2442d6a30265d20e7249f6f665312
+oid sha256:8b41c2b1c27e22a85d4bf4be8821ef48cc2a6b9ab1d9ea23e85d7924cc349904
 size 14575

{checkpoint-1920 → checkpoint-2240}/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8a6af08fc0bc7ad4d0ba68c3a95fad7a6e6e0959e9bab6a7e9c3fee5a348c253
+oid sha256:0aa8b3ba62b2823286940cdf00d5e903441ea9d6c6042755e100b8f2dad56a36
 size 627

{checkpoint-1920 → checkpoint-2240}/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": 174.54545454545453,
-  "global_step": 1920,
+  "epoch": 203.63636363636363,
+  "global_step": 2240,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -15270,11 +15270,2555 @@
       "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
+    },
+    {
+      "epoch": 181.91,
+      "learning_rate": 6.69665071770335e-05,
+      "loss": 0.0504,
+      "step": 2001
+    },
+    {
+      "epoch": 182.0,
+      "learning_rate": 6.694736842105264e-05,
+      "loss": 0.0356,
+      "step": 2002
+    },
+    {
+      "epoch": 182.09,
+      "learning_rate": 6.692822966507177e-05,
+      "loss": 0.0462,
+      "step": 2003
+    },
+    {
+      "epoch": 182.18,
+      "learning_rate": 6.690909090909092e-05,
+      "loss": 0.0444,
+      "step": 2004
+    },
+    {
+      "epoch": 182.27,
+      "learning_rate": 6.688995215311005e-05,
+      "loss": 0.0435,
+      "step": 2005
+    },
+    {
+      "epoch": 182.36,
+      "learning_rate": 6.68708133971292e-05,
+      "loss": 0.0409,
+      "step": 2006
+    },
+    {
+      "epoch": 182.45,
+      "learning_rate": 6.685167464114833e-05,
+      "loss": 0.0397,
+      "step": 2007
+    },
+    {
+      "epoch": 182.55,
+      "learning_rate": 6.683253588516747e-05,
+      "loss": 0.0404,
+      "step": 2008
+    },
+    {
+      "epoch": 182.64,
+      "learning_rate": 6.68133971291866e-05,
+      "loss": 0.0423,
+      "step": 2009
+    },
+    {
+      "epoch": 182.73,
+      "learning_rate": 6.679425837320575e-05,
+      "loss": 0.0409,
+      "step": 2010
+    },
+    {
+      "epoch": 182.82,
+      "learning_rate": 6.677511961722488e-05,
+      "loss": 0.0417,
+      "step": 2011
+    },
+    {
+      "epoch": 182.91,
+      "learning_rate": 6.675598086124403e-05,
+      "loss": 0.0423,
+      "step": 2012
+    },
+    {
+      "epoch": 183.0,
+      "learning_rate": 6.673684210526316e-05,
+      "loss": 0.0433,
+      "step": 2013
+    },
+    {
+      "epoch": 183.09,
+      "learning_rate": 6.67177033492823e-05,
+      "loss": 0.0385,
+      "step": 2014
+    },
+    {
+      "epoch": 183.18,
+      "learning_rate": 6.669856459330144e-05,
+      "loss": 0.0465,
+      "step": 2015
+    },
+    {
+      "epoch": 183.27,
+      "learning_rate": 6.667942583732057e-05,
+      "loss": 0.0397,
+      "step": 2016
+    },
+    {
+      "epoch": 183.36,
+      "learning_rate": 6.666028708133971e-05,
+      "loss": 0.0424,
+      "step": 2017
+    },
+    {
+      "epoch": 183.45,
+      "learning_rate": 6.664114832535886e-05,
+      "loss": 0.048,
+      "step": 2018
+    },
+    {
+      "epoch": 183.55,
+      "learning_rate": 6.662200956937799e-05,
+      "loss": 0.0457,
+      "step": 2019
+    },
+    {
+      "epoch": 183.64,
+      "learning_rate": 6.660287081339714e-05,
+      "loss": 0.0374,
+      "step": 2020
+    },
+    {
+      "epoch": 183.64,
+      "eval_accuracy_NAT": 0.9924062497573797,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9454939053178248,
+      "eval_accuracy_concrete tiles": 0.9917752094416926,
+      "eval_accuracy_dark bricks": 0.8842230861013818,
+      "eval_accuracy_grass": 0.8587605769667653,
+      "eval_accuracy_red bricks": 0.9190340909090909,
+      "eval_accuracy_road": 0.919061605563328,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.45979102914013176,
+      "eval_accuracy_speedway bricks": 0.9891531412806969,
+      "eval_accuracy_steel": 0.8893872435855155,
+      "eval_accuracy_tiles": 0.03670730036981235,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790052617396954,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.865493816322747,
+      "eval_iou_concrete tiles": 0.9283097704767901,
+      "eval_iou_dark bricks": 0.8206401371722511,
+      "eval_iou_grass": 0.7978089201755981,
+      "eval_iou_red bricks": 0.8330919040721069,
+      "eval_iou_road": 0.9113890657582179,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.30605204242668915,
+      "eval_iou_speedway bricks": 0.9745489091243091,
+      "eval_iou_steel": 0.8341801028496307,
+      "eval_iou_tiles": 0.03670730036981235,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11285177618265152,
+      "eval_mean_accuracy": 0.6835225721872017,
+      "eval_mean_iou": 0.6374790177298344,
+      "eval_overall_accuracy": 0.9706172943115234,
+      "eval_runtime": 3.2685,
+      "eval_samples_per_second": 11.014,
+      "eval_steps_per_second": 0.612,
+      "step": 2020
+    },
+    {
+      "epoch": 183.73,
+      "learning_rate": 6.658373205741627e-05,
+      "loss": 0.0418,
+      "step": 2021
+    },
+    {
+      "epoch": 183.82,
+      "learning_rate": 6.65645933014354e-05,
+      "loss": 0.0404,
+      "step": 2022
+    },
+    {
+      "epoch": 183.91,
+      "learning_rate": 6.654545454545455e-05,
+      "loss": 0.0463,
+      "step": 2023
+    },
+    {
+      "epoch": 184.0,
+      "learning_rate": 6.652631578947369e-05,
+      "loss": 0.0434,
+      "step": 2024
+    },
+    {
+      "epoch": 184.09,
+      "learning_rate": 6.650717703349282e-05,
+      "loss": 0.0404,
+      "step": 2025
+    },
+    {
+      "epoch": 184.18,
+      "learning_rate": 6.648803827751197e-05,
+      "loss": 0.0402,
+      "step": 2026
+    },
+    {
+      "epoch": 184.27,
+      "learning_rate": 6.64688995215311e-05,
+      "loss": 0.0408,
+      "step": 2027
+    },
+    {
+      "epoch": 184.36,
+      "learning_rate": 6.644976076555023e-05,
+      "loss": 0.0458,
+      "step": 2028
+    },
+    {
+      "epoch": 184.45,
+      "learning_rate": 6.643062200956938e-05,
+      "loss": 0.0424,
+      "step": 2029
+    },
+    {
+      "epoch": 184.55,
+      "learning_rate": 6.641148325358852e-05,
+      "loss": 0.0433,
+      "step": 2030
+    },
+    {
+      "epoch": 184.64,
+      "learning_rate": 6.639234449760766e-05,
+      "loss": 0.0425,
+      "step": 2031
+    },
+    {
+      "epoch": 184.73,
+      "learning_rate": 6.63732057416268e-05,
+      "loss": 0.0435,
+      "step": 2032
+    },
+    {
+      "epoch": 184.82,
+      "learning_rate": 6.635406698564593e-05,
+      "loss": 0.0427,
+      "step": 2033
+    },
+    {
+      "epoch": 184.91,
+      "learning_rate": 6.633492822966507e-05,
+      "loss": 0.0498,
+      "step": 2034
+    },
+    {
+      "epoch": 185.0,
+      "learning_rate": 6.631578947368421e-05,
+      "loss": 0.0466,
+      "step": 2035
+    },
+    {
+      "epoch": 185.09,
+      "learning_rate": 6.629665071770336e-05,
+      "loss": 0.0417,
+      "step": 2036
+    },
+    {
+      "epoch": 185.18,
+      "learning_rate": 6.627751196172249e-05,
+      "loss": 0.0522,
+      "step": 2037
+    },
+    {
+      "epoch": 185.27,
+      "learning_rate": 6.625837320574163e-05,
+      "loss": 0.0455,
+      "step": 2038
+    },
+    {
+      "epoch": 185.36,
+      "learning_rate": 6.623923444976077e-05,
+      "loss": 0.0406,
+      "step": 2039
+    },
+    {
+      "epoch": 185.45,
+      "learning_rate": 6.62200956937799e-05,
+      "loss": 0.0442,
+      "step": 2040
+    },
+    {
+      "epoch": 185.45,
+      "eval_accuracy_NAT": 0.9925906411739505,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9447776728614962,
+      "eval_accuracy_concrete tiles": 0.9874640738702378,
+      "eval_accuracy_dark bricks": 0.8626303837419653,
+      "eval_accuracy_grass": 0.8853266321120381,
+      "eval_accuracy_red bricks": 0.9625355113636364,
+      "eval_accuracy_road": 0.9060456814919703,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5086152234444352,
+      "eval_accuracy_speedway bricks": 0.9888061817922902,
+      "eval_accuracy_steel": 0.8950349581162192,
+      "eval_accuracy_tiles": 0.08943980276674428,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791917652725288,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8785884630352135,
+      "eval_iou_concrete tiles": 0.9385643708224354,
+      "eval_iou_dark bricks": 0.8144487244712552,
+      "eval_iou_grass": 0.8235724434358246,
+      "eval_iou_red bricks": 0.8273809523809523,
+      "eval_iou_road": 0.8990046360730637,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3366482975073902,
+      "eval_iou_speedway bricks": 0.9749617292375536,
+      "eval_iou_steel": 0.8356966897613549,
+      "eval_iou_tiles": 0.08942143101677508,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11241282522678375,
+      "eval_mean_accuracy": 0.6940974432873064,
+      "eval_mean_iou": 0.6459599617703344,
+      "eval_overall_accuracy": 0.9705687628851997,
+      "eval_runtime": 3.8428,
+      "eval_samples_per_second": 9.368,
+      "eval_steps_per_second": 0.52,
+      "step": 2040
+    },
+    {
+      "epoch": 185.55,
+      "learning_rate": 6.620095693779904e-05,
+      "loss": 0.044,
+      "step": 2041
+    },
+    {
+      "epoch": 185.64,
+      "learning_rate": 6.618181818181819e-05,
+      "loss": 0.0445,
+      "step": 2042
+    },
+    {
+      "epoch": 185.73,
+      "learning_rate": 6.616267942583732e-05,
+      "loss": 0.0404,
+      "step": 2043
+    },
+    {
+      "epoch": 185.82,
+      "learning_rate": 6.614354066985647e-05,
+      "loss": 0.0393,
+      "step": 2044
+    },
+    {
+      "epoch": 185.91,
+      "learning_rate": 6.61244019138756e-05,
+      "loss": 0.0464,
+      "step": 2045
+    },
+    {
+      "epoch": 186.0,
+      "learning_rate": 6.610526315789473e-05,
+      "loss": 0.0439,
+      "step": 2046
+    },
+    {
+      "epoch": 186.09,
+      "learning_rate": 6.608612440191388e-05,
+      "loss": 0.0423,
+      "step": 2047
+    },
+    {
+      "epoch": 186.18,
+      "learning_rate": 6.606698564593302e-05,
+      "loss": 0.0464,
+      "step": 2048
+    },
+    {
+      "epoch": 186.27,
+      "learning_rate": 6.604784688995215e-05,
+      "loss": 0.0439,
+      "step": 2049
+    },
+    {
+      "epoch": 186.36,
+      "learning_rate": 6.60287081339713e-05,
+      "loss": 0.0355,
+      "step": 2050
+    },
+    {
+      "epoch": 186.45,
+      "learning_rate": 6.600956937799043e-05,
+      "loss": 0.0448,
+      "step": 2051
+    },
+    {
+      "epoch": 186.55,
+      "learning_rate": 6.599043062200956e-05,
+      "loss": 0.0427,
+      "step": 2052
+    },
+    {
+      "epoch": 186.64,
+      "learning_rate": 6.597129186602871e-05,
+      "loss": 0.0464,
+      "step": 2053
+    },
+    {
+      "epoch": 186.73,
+      "learning_rate": 6.595215311004786e-05,
+      "loss": 0.0476,
+      "step": 2054
+    },
+    {
+      "epoch": 186.82,
+      "learning_rate": 6.593301435406699e-05,
+      "loss": 0.0356,
+      "step": 2055
+    },
+    {
+      "epoch": 186.91,
+      "learning_rate": 6.591387559808613e-05,
+      "loss": 0.0429,
+      "step": 2056
+    },
+    {
+      "epoch": 187.0,
+      "learning_rate": 6.589473684210526e-05,
+      "loss": 0.0424,
+      "step": 2057
+    },
+    {
+      "epoch": 187.09,
+      "learning_rate": 6.58755980861244e-05,
+      "loss": 0.0397,
+      "step": 2058
+    },
+    {
+      "epoch": 187.18,
+      "learning_rate": 6.585645933014354e-05,
+      "loss": 0.0448,
+      "step": 2059
+    },
+    {
+      "epoch": 187.27,
+      "learning_rate": 6.583732057416269e-05,
+      "loss": 0.039,
+      "step": 2060
+    },
+    {
+      "epoch": 187.27,
+      "eval_accuracy_NAT": 0.9939258467712622,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9453208828705094,
+      "eval_accuracy_concrete tiles": 0.9836727206017245,
+      "eval_accuracy_dark bricks": 0.8662869019668219,
+      "eval_accuracy_grass": 0.8743346253455754,
+      "eval_accuracy_red bricks": 0.9382102272727273,
+      "eval_accuracy_road": 0.9282963335578431,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5814793372480909,
+      "eval_accuracy_speedway bricks": 0.9877212545572375,
+      "eval_accuracy_steel": 0.8960820526350505,
+      "eval_accuracy_tiles": 0.05307492124366525,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9789039772273064,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8853638805212686,
+      "eval_iou_concrete tiles": 0.9353141262319389,
+      "eval_iou_dark bricks": 0.8152937767152069,
+      "eval_iou_grass": 0.8229500430946569,
+      "eval_iou_red bricks": 0.83147128245476,
+      "eval_iou_road": 0.9221622413776535,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3904118463674225,
+      "eval_iou_speedway bricks": 0.9749494331027944,
+      "eval_iou_steel": 0.8369063019774533,
+      "eval_iou_tiles": 0.05307492124366525,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10347142815589905,
+      "eval_mean_accuracy": 0.6960311618515775,
+      "eval_mean_iou": 0.6497539869472404,
+      "eval_overall_accuracy": 0.9730592303805881,
+      "eval_runtime": 3.4146,
+      "eval_samples_per_second": 10.543,
+      "eval_steps_per_second": 0.586,
+      "step": 2060
+    },
+    {
+      "epoch": 187.36,
+      "learning_rate": 6.581818181818182e-05,
+      "loss": 0.0492,
+      "step": 2061
+    },
+    {
+      "epoch": 187.45,
+      "learning_rate": 6.579904306220097e-05,
+      "loss": 0.0399,
+      "step": 2062
+    },
+    {
+      "epoch": 187.55,
+      "learning_rate": 6.57799043062201e-05,
+      "loss": 0.0429,
+      "step": 2063
+    },
+    {
+      "epoch": 187.64,
+      "learning_rate": 6.576076555023923e-05,
+      "loss": 0.043,
+      "step": 2064
+    },
+    {
+      "epoch": 187.73,
+      "learning_rate": 6.574162679425838e-05,
+      "loss": 0.0406,
+      "step": 2065
+    },
+    {
+      "epoch": 187.82,
+      "learning_rate": 6.572248803827752e-05,
+      "loss": 0.0413,
+      "step": 2066
+    },
+    {
+      "epoch": 187.91,
+      "learning_rate": 6.570334928229665e-05,
+      "loss": 0.0372,
+      "step": 2067
+    },
+    {
+      "epoch": 188.0,
+      "learning_rate": 6.56842105263158e-05,
+      "loss": 0.0436,
+      "step": 2068
+    },
+    {
+      "epoch": 188.09,
+      "learning_rate": 6.566507177033493e-05,
+      "loss": 0.0425,
+      "step": 2069
+    },
+    {
+      "epoch": 188.18,
+      "learning_rate": 6.564593301435406e-05,
+      "loss": 0.0429,
+      "step": 2070
+    },
+    {
+      "epoch": 188.27,
+      "learning_rate": 6.562679425837321e-05,
+      "loss": 0.0461,
+      "step": 2071
+    },
+    {
+      "epoch": 188.36,
+      "learning_rate": 6.560765550239235e-05,
+      "loss": 0.0422,
+      "step": 2072
+    },
+    {
+      "epoch": 188.45,
+      "learning_rate": 6.558851674641149e-05,
+      "loss": 0.0402,
+      "step": 2073
+    },
+    {
+      "epoch": 188.55,
+      "learning_rate": 6.556937799043063e-05,
+      "loss": 0.0432,
+      "step": 2074
+    },
+    {
+      "epoch": 188.64,
+      "learning_rate": 6.555023923444976e-05,
+      "loss": 0.0455,
+      "step": 2075
+    },
+    {
+      "epoch": 188.73,
+      "learning_rate": 6.55311004784689e-05,
+      "loss": 0.0444,
+      "step": 2076
+    },
+    {
+      "epoch": 188.82,
+      "learning_rate": 6.551196172248804e-05,
+      "loss": 0.0409,
+      "step": 2077
+    },
+    {
+      "epoch": 188.91,
+      "learning_rate": 6.549282296650719e-05,
+      "loss": 0.0388,
+      "step": 2078
+    },
+    {
+      "epoch": 189.0,
+      "learning_rate": 6.547368421052632e-05,
+      "loss": 0.0407,
+      "step": 2079
+    },
+    {
+      "epoch": 189.09,
+      "learning_rate": 6.545454545454546e-05,
+      "loss": 0.0399,
+      "step": 2080
+    },
+    {
+      "epoch": 189.09,
+      "eval_accuracy_NAT": 0.9914188293103655,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9486512296665093,
+      "eval_accuracy_concrete tiles": 0.9881978841802728,
+      "eval_accuracy_dark bricks": 0.8916131018821446,
+      "eval_accuracy_grass": 0.9078523820674759,
+      "eval_accuracy_red bricks": 0.9398082386363636,
+      "eval_accuracy_road": 0.9178499782947012,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.539878146279602,
+      "eval_accuracy_speedway bricks": 0.9859828366683683,
+      "eval_accuracy_steel": 0.9133879691313238,
+      "eval_accuracy_tiles": 0.14361046431995617,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794614431605171,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8828215981017595,
+      "eval_iou_concrete tiles": 0.9358892685469392,
+      "eval_iou_dark bricks": 0.8145504412954042,
+      "eval_iou_grass": 0.8275756600641665,
+      "eval_iou_red bricks": 0.824840268038024,
+      "eval_iou_road": 0.9102355783932222,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35696968592209116,
+      "eval_iou_speedway bricks": 0.9751024319914832,
+      "eval_iou_steel": 0.8304697221847404,
+      "eval_iou_tiles": 0.14361046431995617,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10652013123035431,
+      "eval_mean_accuracy": 0.7052500815720835,
+      "eval_mean_iou": 0.6524251201552542,
+      "eval_overall_accuracy": 0.9719012578328451,
+      "eval_runtime": 3.3447,
+      "eval_samples_per_second": 10.763,
+      "eval_steps_per_second": 0.598,
+      "step": 2080
+    },
+    {
+      "epoch": 189.18,
+      "learning_rate": 6.54354066985646e-05,
+      "loss": 0.0437,
+      "step": 2081
+    },
+    {
+      "epoch": 189.27,
+      "learning_rate": 6.541626794258373e-05,
+      "loss": 0.0403,
+      "step": 2082
+    },
+    {
+      "epoch": 189.36,
+      "learning_rate": 6.539712918660287e-05,
+      "loss": 0.0399,
+      "step": 2083
+    },
+    {
+      "epoch": 189.45,
+      "learning_rate": 6.537799043062202e-05,
+      "loss": 0.0488,
+      "step": 2084
+    },
+    {
+      "epoch": 189.55,
+      "learning_rate": 6.535885167464115e-05,
+      "loss": 0.0405,
+      "step": 2085
+    },
+    {
+      "epoch": 189.64,
+      "learning_rate": 6.53397129186603e-05,
+      "loss": 0.0412,
+      "step": 2086
+    },
+    {
+      "epoch": 189.73,
+      "learning_rate": 6.532057416267943e-05,
+      "loss": 0.0477,
+      "step": 2087
+    },
+    {
+      "epoch": 189.82,
+      "learning_rate": 6.530143540669856e-05,
+      "loss": 0.0408,
+      "step": 2088
+    },
+    {
+      "epoch": 189.91,
+      "learning_rate": 6.52822966507177e-05,
+      "loss": 0.0367,
+      "step": 2089
+    },
+    {
+      "epoch": 190.0,
+      "learning_rate": 6.526315789473685e-05,
+      "loss": 0.0383,
+      "step": 2090
+    },
+    {
+      "epoch": 190.09,
+      "learning_rate": 6.524401913875598e-05,
+      "loss": 0.0543,
+      "step": 2091
+    },
+    {
+      "epoch": 190.18,
+      "learning_rate": 6.522488038277512e-05,
+      "loss": 0.0416,
+      "step": 2092
+    },
+    {
+      "epoch": 190.27,
+      "learning_rate": 6.520574162679426e-05,
+      "loss": 0.0413,
+      "step": 2093
+    },
+    {
+      "epoch": 190.36,
+      "learning_rate": 6.518660287081339e-05,
+      "loss": 0.0397,
+      "step": 2094
+    },
+    {
+      "epoch": 190.45,
+      "learning_rate": 6.516746411483254e-05,
+      "loss": 0.0483,
+      "step": 2095
+    },
+    {
+      "epoch": 190.55,
+      "learning_rate": 6.514832535885168e-05,
+      "loss": 0.0366,
+      "step": 2096
+    },
+    {
+      "epoch": 190.64,
+      "learning_rate": 6.512918660287082e-05,
+      "loss": 0.0432,
+      "step": 2097
+    },
+    {
+      "epoch": 190.73,
+      "learning_rate": 6.511004784688995e-05,
+      "loss": 0.0416,
+      "step": 2098
+    },
+    {
+      "epoch": 190.82,
+      "learning_rate": 6.50909090909091e-05,
+      "loss": 0.0394,
+      "step": 2099
+    },
+    {
+      "epoch": 190.91,
+      "learning_rate": 6.507177033492823e-05,
+      "loss": 0.0453,
+      "step": 2100
+    },
+    {
+      "epoch": 190.91,
+      "eval_accuracy_NAT": 0.9908030149143542,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9477874586892141,
+      "eval_accuracy_concrete tiles": 0.9827554577141809,
+      "eval_accuracy_dark bricks": 0.8860705900465725,
+      "eval_accuracy_grass": 0.8996424539611417,
+      "eval_accuracy_red bricks": 0.9151278409090909,
+      "eval_accuracy_road": 0.9166468154880306,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5587075786397596,
+      "eval_accuracy_speedway bricks": 0.9893957112186786,
+      "eval_accuracy_steel": 0.8935921113383022,
+      "eval_accuracy_tiles": 0.12977674291192987,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9793174076552572,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8821565046308723,
+      "eval_iou_concrete tiles": 0.9472753529221066,
+      "eval_iou_dark bricks": 0.8283616998308805,
+      "eval_iou_grass": 0.8144994399195002,
+      "eval_iou_red bricks": 0.8376401755241346,
+      "eval_iou_road": 0.9090402167148135,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.36637440115700987,
+      "eval_iou_speedway bricks": 0.9747450157921675,
+      "eval_iou_steel": 0.8313606971089309,
+      "eval_iou_tiles": 0.12977674291192987,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10846742987632751,
+      "eval_mean_accuracy": 0.7007927519870196,
+      "eval_mean_iou": 0.6538882810898156,
+      "eval_overall_accuracy": 0.9714558919270834,
+      "eval_runtime": 3.3367,
+      "eval_samples_per_second": 10.789,
+      "eval_steps_per_second": 0.599,
+      "step": 2100
+    },
+    {
+      "epoch": 191.0,
+      "learning_rate": 6.505263157894737e-05,
+      "loss": 0.0394,
+      "step": 2101
+    },
+    {
+      "epoch": 191.09,
+      "learning_rate": 6.503349282296652e-05,
+      "loss": 0.0398,
+      "step": 2102
+    },
+    {
+      "epoch": 191.18,
+      "learning_rate": 6.501435406698565e-05,
+      "loss": 0.0439,
+      "step": 2103
+    },
+    {
+      "epoch": 191.27,
+      "learning_rate": 6.499521531100478e-05,
+      "loss": 0.0495,
+      "step": 2104
+    },
+    {
+      "epoch": 191.36,
+      "learning_rate": 6.497607655502393e-05,
+      "loss": 0.0368,
+      "step": 2105
+    },
+    {
+      "epoch": 191.45,
+      "learning_rate": 6.495693779904306e-05,
+      "loss": 0.0466,
+      "step": 2106
+    },
+    {
+      "epoch": 191.55,
+      "learning_rate": 6.49377990430622e-05,
+      "loss": 0.0439,
+      "step": 2107
+    },
+    {
+      "epoch": 191.64,
+      "learning_rate": 6.491866028708135e-05,
+      "loss": 0.0404,
+      "step": 2108
+    },
+    {
+      "epoch": 191.73,
+      "learning_rate": 6.489952153110048e-05,
+      "loss": 0.0415,
+      "step": 2109
+    },
+    {
+      "epoch": 191.82,
+      "learning_rate": 6.488038277511961e-05,
+      "loss": 0.0392,
+      "step": 2110
+    },
+    {
+      "epoch": 191.91,
+      "learning_rate": 6.486124401913876e-05,
+      "loss": 0.0409,
+      "step": 2111
+    },
+    {
+      "epoch": 192.0,
+      "learning_rate": 6.484210526315789e-05,
+      "loss": 0.0432,
+      "step": 2112
+    },
+    {
+      "epoch": 192.09,
+      "learning_rate": 6.482296650717704e-05,
+      "loss": 0.0434,
+      "step": 2113
+    },
+    {
+      "epoch": 192.18,
+      "learning_rate": 6.480382775119618e-05,
+      "loss": 0.0467,
+      "step": 2114
+    },
+    {
+      "epoch": 192.27,
+      "learning_rate": 6.478468899521531e-05,
+      "loss": 0.0399,
+      "step": 2115
+    },
+    {
+      "epoch": 192.36,
+      "learning_rate": 6.476555023923445e-05,
+      "loss": 0.0452,
+      "step": 2116
+    },
+    {
+      "epoch": 192.45,
+      "learning_rate": 6.474641148325359e-05,
+      "loss": 0.042,
+      "step": 2117
+    },
+    {
+      "epoch": 192.55,
+      "learning_rate": 6.472727272727272e-05,
+      "loss": 0.0413,
+      "step": 2118
+    },
+    {
+      "epoch": 192.64,
+      "learning_rate": 6.470813397129187e-05,
+      "loss": 0.041,
+      "step": 2119
+    },
+    {
+      "epoch": 192.73,
+      "learning_rate": 6.468899521531101e-05,
+      "loss": 0.0424,
+      "step": 2120
+    },
+    {
+      "epoch": 192.73,
+      "eval_accuracy_NAT": 0.9935327136840664,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9427912678655737,
+      "eval_accuracy_concrete tiles": 0.9848651623555311,
+      "eval_accuracy_dark bricks": 0.8553173472922521,
+      "eval_accuracy_grass": 0.8620948772704873,
+      "eval_accuracy_red bricks": 0.8790838068181818,
+      "eval_accuracy_road": 0.8894638005192342,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.45764066936840075,
+      "eval_accuracy_speedway bricks": 0.9885358550952359,
+      "eval_accuracy_steel": 0.8939548842424642,
+      "eval_accuracy_tiles": 0.0784823996712779,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791588086365555,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8688431987184558,
+      "eval_iou_concrete tiles": 0.9333275382475661,
+      "eval_iou_dark bricks": 0.8213942485399571,
+      "eval_iou_grass": 0.8021202562518293,
+      "eval_iou_red bricks": 0.8395794471765304,
+      "eval_iou_road": 0.8821388396529848,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3031133022916096,
+      "eval_iou_speedway bricks": 0.9752040287872564,
+      "eval_iou_steel": 0.8347396298463339,
+      "eval_iou_tiles": 0.0784823996712779,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12418189644813538,
+      "eval_mean_accuracy": 0.6789048295525159,
+      "eval_mean_iou": 0.639853976755412,
+      "eval_overall_accuracy": 0.9684669706556532,
+      "eval_runtime": 3.2279,
+      "eval_samples_per_second": 11.153,
+      "eval_steps_per_second": 0.62,
+      "step": 2120
+    },
+    {
+      "epoch": 192.82,
+      "learning_rate": 6.466985645933015e-05,
+      "loss": 0.0422,
+      "step": 2121
+    },
+    {
+      "epoch": 192.91,
+      "learning_rate": 6.465071770334928e-05,
+      "loss": 0.039,
+      "step": 2122
+    },
+    {
+      "epoch": 193.0,
+      "learning_rate": 6.463157894736842e-05,
+      "loss": 0.0382,
+      "step": 2123
+    },
+    {
+      "epoch": 193.09,
+      "learning_rate": 6.461244019138756e-05,
+      "loss": 0.0354,
+      "step": 2124
+    },
+    {
+      "epoch": 193.18,
+      "learning_rate": 6.45933014354067e-05,
+      "loss": 0.0416,
+      "step": 2125
+    },
+    {
+      "epoch": 193.27,
+      "learning_rate": 6.457416267942585e-05,
+      "loss": 0.0411,
+      "step": 2126
+    },
+    {
+      "epoch": 193.36,
+      "learning_rate": 6.455502392344498e-05,
+      "loss": 0.0354,
+      "step": 2127
+    },
+    {
+      "epoch": 193.45,
+      "learning_rate": 6.453588516746411e-05,
+      "loss": 0.0452,
+      "step": 2128
+    },
+    {
+      "epoch": 193.55,
+      "learning_rate": 6.451674641148326e-05,
+      "loss": 0.0432,
+      "step": 2129
+    },
+    {
+      "epoch": 193.64,
+      "learning_rate": 6.449760765550239e-05,
+      "loss": 0.0388,
+      "step": 2130
+    },
+    {
+      "epoch": 193.73,
+      "learning_rate": 6.447846889952153e-05,
+      "loss": 0.04,
+      "step": 2131
+    },
+    {
+      "epoch": 193.82,
+      "learning_rate": 6.445933014354068e-05,
+      "loss": 0.0474,
+      "step": 2132
+    },
+    {
+      "epoch": 193.91,
+      "learning_rate": 6.444019138755981e-05,
+      "loss": 0.045,
+      "step": 2133
+    },
+    {
+      "epoch": 194.0,
+      "learning_rate": 6.442105263157894e-05,
+      "loss": 0.0518,
+      "step": 2134
+    },
+    {
+      "epoch": 194.09,
+      "learning_rate": 6.440191387559809e-05,
+      "loss": 0.0478,
+      "step": 2135
+    },
+    {
+      "epoch": 194.18,
+      "learning_rate": 6.438277511961724e-05,
+      "loss": 0.0416,
+      "step": 2136
+    },
+    {
+      "epoch": 194.27,
+      "learning_rate": 6.436363636363637e-05,
+      "loss": 0.0408,
+      "step": 2137
+    },
+    {
+      "epoch": 194.36,
+      "learning_rate": 6.434449760765551e-05,
+      "loss": 0.0398,
+      "step": 2138
+    },
+    {
+      "epoch": 194.45,
+      "learning_rate": 6.432535885167464e-05,
+      "loss": 0.0493,
+      "step": 2139
+    },
+    {
+      "epoch": 194.55,
+      "learning_rate": 6.430622009569378e-05,
+      "loss": 0.0447,
+      "step": 2140
+    },
+    {
+      "epoch": 194.55,
+      "eval_accuracy_NAT": 0.9928463188415207,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9496357139791407,
+      "eval_accuracy_concrete tiles": 0.983886748608818,
+      "eval_accuracy_dark bricks": 0.8604749624725762,
+      "eval_accuracy_grass": 0.8868644645372348,
+      "eval_accuracy_red bricks": 0.9291548295454546,
+      "eval_accuracy_road": 0.9397354492875946,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5942712210183884,
+      "eval_accuracy_speedway bricks": 0.9861083454920007,
+      "eval_accuracy_steel": 0.9056543104016885,
+      "eval_accuracy_tiles": 0.11847692096973017,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9795589160273369,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8869449531860307,
+      "eval_iou_concrete tiles": 0.9445797986321073,
+      "eval_iou_dark bricks": 0.8228798586572438,
+      "eval_iou_grass": 0.821935157862004,
+      "eval_iou_red bricks": 0.8297130172823847,
+      "eval_iou_road": 0.9313966713447293,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3982853553083774,
+      "eval_iou_speedway bricks": 0.9754710426508035,
+      "eval_iou_steel": 0.8337001730471477,
+      "eval_iou_tiles": 0.11847692096973017,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09687285125255585,
+      "eval_mean_accuracy": 0.7036237911657037,
+      "eval_mean_iou": 0.6571493742282997,
+      "eval_overall_accuracy": 0.9741036097208658,
+      "eval_runtime": 3.554,
+      "eval_samples_per_second": 10.13,
+      "eval_steps_per_second": 0.563,
+      "step": 2140
+    },
+    {
+      "epoch": 194.64,
+      "learning_rate": 6.428708133971292e-05,
+      "loss": 0.039,
+      "step": 2141
+    },
+    {
+      "epoch": 194.73,
+      "learning_rate": 6.426794258373207e-05,
+      "loss": 0.044,
+      "step": 2142
+    },
+    {
+      "epoch": 194.82,
+      "learning_rate": 6.42488038277512e-05,
+      "loss": 0.0382,
+      "step": 2143
+    },
+    {
+      "epoch": 194.91,
+      "learning_rate": 6.422966507177035e-05,
+      "loss": 0.0349,
+      "step": 2144
+    },
+    {
+      "epoch": 195.0,
+      "learning_rate": 6.421052631578948e-05,
+      "loss": 0.0426,
+      "step": 2145
+    },
+    {
+      "epoch": 195.09,
+      "learning_rate": 6.419138755980861e-05,
+      "loss": 0.0392,
+      "step": 2146
+    },
+    {
+      "epoch": 195.18,
+      "learning_rate": 6.417224880382775e-05,
+      "loss": 0.0404,
+      "step": 2147
+    },
+    {
+      "epoch": 195.27,
+      "learning_rate": 6.41531100478469e-05,
+      "loss": 0.0407,
+      "step": 2148
+    },
+    {
+      "epoch": 195.36,
+      "learning_rate": 6.413397129186603e-05,
+      "loss": 0.045,
+      "step": 2149
+    },
+    {
+      "epoch": 195.45,
+      "learning_rate": 6.411483253588518e-05,
+      "loss": 0.0457,
+      "step": 2150
+    },
+    {
+      "epoch": 195.55,
+      "learning_rate": 6.409569377990431e-05,
+      "loss": 0.0353,
+      "step": 2151
+    },
+    {
+      "epoch": 195.64,
+      "learning_rate": 6.407655502392344e-05,
+      "loss": 0.0434,
+      "step": 2152
+    },
+    {
+      "epoch": 195.73,
+      "learning_rate": 6.405741626794259e-05,
+      "loss": 0.0405,
+      "step": 2153
+    },
+    {
+      "epoch": 195.82,
+      "learning_rate": 6.403827751196173e-05,
+      "loss": 0.0424,
+      "step": 2154
+    },
+    {
+      "epoch": 195.91,
+      "learning_rate": 6.401913875598087e-05,
+      "loss": 0.0454,
+      "step": 2155
+    },
+    {
+      "epoch": 196.0,
+      "learning_rate": 6.400000000000001e-05,
+      "loss": 0.0399,
+      "step": 2156
+    },
+    {
+      "epoch": 196.09,
+      "learning_rate": 6.398086124401914e-05,
+      "loss": 0.0383,
+      "step": 2157
+    },
+    {
+      "epoch": 196.18,
+      "learning_rate": 6.396172248803827e-05,
+      "loss": 0.0358,
+      "step": 2158
+    },
+    {
+      "epoch": 196.27,
+      "learning_rate": 6.394258373205742e-05,
+      "loss": 0.0445,
+      "step": 2159
+    },
+    {
+      "epoch": 196.36,
+      "learning_rate": 6.392344497607657e-05,
+      "loss": 0.041,
+      "step": 2160
+    },
+    {
+      "epoch": 196.36,
+      "eval_accuracy_NAT": 0.9911310375396221,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9519936477960428,
+      "eval_accuracy_concrete tiles": 0.9870971687152205,
+      "eval_accuracy_dark bricks": 0.8729841037681383,
+      "eval_accuracy_grass": 0.8945431414421373,
+      "eval_accuracy_red bricks": 0.9222301136363636,
+      "eval_accuracy_road": 0.9329820178551779,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5270035563642379,
+      "eval_accuracy_speedway bricks": 0.9877224613728494,
+      "eval_accuracy_steel": 0.9020183365213377,
+      "eval_accuracy_tiles": 0.03328311190247911,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9801395814858749,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8793835929572597,
+      "eval_iou_concrete tiles": 0.9361208571345725,
+      "eval_iou_dark bricks": 0.815804618372779,
+      "eval_iou_grass": 0.8241609268686947,
+      "eval_iou_red bricks": 0.832505209168136,
+      "eval_iou_road": 0.9229113014063212,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3409492214671732,
+      "eval_iou_speedway bricks": 0.9750754877590396,
+      "eval_iou_steel": 0.8312931682965192,
+      "eval_iou_tiles": 0.03328311190247911,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10372213274240494,
+      "eval_mean_accuracy": 0.6925375920702774,
+      "eval_mean_iou": 0.64397131360145,
+      "eval_overall_accuracy": 0.9727885988023546,
+      "eval_runtime": 3.6334,
+      "eval_samples_per_second": 9.908,
+      "eval_steps_per_second": 0.55,
+      "step": 2160
+    },
+    {
+      "epoch": 196.45,
+      "learning_rate": 6.39043062200957e-05,
+      "loss": 0.0422,
+      "step": 2161
+    },
+    {
+      "epoch": 196.55,
+      "learning_rate": 6.388516746411484e-05,
+      "loss": 0.0416,
+      "step": 2162
+    },
+    {
+      "epoch": 196.64,
+      "learning_rate": 6.386602870813398e-05,
+      "loss": 0.0421,
+      "step": 2163
+    },
+    {
+      "epoch": 196.73,
+      "learning_rate": 6.384688995215311e-05,
+      "loss": 0.0404,
+      "step": 2164
+    },
+    {
+      "epoch": 196.82,
+      "learning_rate": 6.382775119617225e-05,
+      "loss": 0.0385,
+      "step": 2165
+    },
+    {
+      "epoch": 196.91,
+      "learning_rate": 6.38086124401914e-05,
+      "loss": 0.0401,
+      "step": 2166
+    },
+    {
+      "epoch": 197.0,
+      "learning_rate": 6.378947368421053e-05,
+      "loss": 0.0361,
+      "step": 2167
+    },
+    {
+      "epoch": 197.09,
+      "learning_rate": 6.377033492822966e-05,
+      "loss": 0.0384,
+      "step": 2168
+    },
+    {
+      "epoch": 197.18,
+      "learning_rate": 6.375119617224881e-05,
+      "loss": 0.0445,
+      "step": 2169
+    },
+    {
+      "epoch": 197.27,
+      "learning_rate": 6.373205741626794e-05,
+      "loss": 0.04,
+      "step": 2170
+    },
+    {
+      "epoch": 197.36,
+      "learning_rate": 6.371291866028709e-05,
+      "loss": 0.0351,
+      "step": 2171
+    },
+    {
+      "epoch": 197.45,
+      "learning_rate": 6.369377990430623e-05,
+      "loss": 0.0426,
+      "step": 2172
+    },
+    {
+      "epoch": 197.55,
+      "learning_rate": 6.367464114832536e-05,
+      "loss": 0.0389,
+      "step": 2173
+    },
+    {
+      "epoch": 197.64,
+      "learning_rate": 6.36555023923445e-05,
+      "loss": 0.0397,
+      "step": 2174
+    },
+    {
+      "epoch": 197.73,
+      "learning_rate": 6.363636363636364e-05,
+      "loss": 0.0374,
+      "step": 2175
+    },
+    {
+      "epoch": 197.82,
+      "learning_rate": 6.361722488038277e-05,
+      "loss": 0.0443,
+      "step": 2176
+    },
+    {
+      "epoch": 197.91,
+      "learning_rate": 6.359808612440192e-05,
+      "loss": 0.0404,
+      "step": 2177
+    },
+    {
+      "epoch": 198.0,
+      "learning_rate": 6.357894736842106e-05,
+      "loss": 0.0414,
+      "step": 2178
+    },
+    {
+      "epoch": 198.09,
+      "learning_rate": 6.35598086124402e-05,
+      "loss": 0.0383,
+      "step": 2179
+    },
+    {
+      "epoch": 198.18,
+      "learning_rate": 6.354066985645933e-05,
+      "loss": 0.0407,
+      "step": 2180
+    },
+    {
+      "epoch": 198.18,
+      "eval_accuracy_NAT": 0.9922264460698337,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9549095454740547,
+      "eval_accuracy_concrete tiles": 0.9853849446584725,
+      "eval_accuracy_dark bricks": 0.8594357415034063,
+      "eval_accuracy_grass": 0.8957594270875201,
+      "eval_accuracy_red bricks": 0.9149502840909091,
+      "eval_accuracy_road": 0.937661656108258,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5668127808562843,
+      "eval_accuracy_speedway bricks": 0.9863756551500252,
+      "eval_accuracy_steel": 0.9018534397467185,
+      "eval_accuracy_tiles": 0.08129023421449116,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9795040918657033,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8872049088806476,
+      "eval_iou_concrete tiles": 0.9427250921429825,
+      "eval_iou_dark bricks": 0.8159394869546153,
+      "eval_iou_grass": 0.8284132304599243,
+      "eval_iou_red bricks": 0.8389775317486161,
+      "eval_iou_road": 0.9311401944573531,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3776149282788767,
+      "eval_iou_speedway bricks": 0.9755072154145268,
+      "eval_iou_steel": 0.8361348713127097,
+      "eval_iou_tiles": 0.08129023421449116,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09696460515260696,
+      "eval_mean_accuracy": 0.6982046273046133,
+      "eval_mean_iou": 0.6534193681331112,
+      "eval_overall_accuracy": 0.974064826965332,
+      "eval_runtime": 3.3143,
+      "eval_samples_per_second": 10.862,
+      "eval_steps_per_second": 0.603,
+      "step": 2180
+    },
+    {
+      "epoch": 198.27,
+      "learning_rate": 6.352153110047847e-05,
+      "loss": 0.042,
+      "step": 2181
+    },
+    {
+      "epoch": 198.36,
+      "learning_rate": 6.35023923444976e-05,
+      "loss": 0.0377,
+      "step": 2182
+    },
+    {
+      "epoch": 198.45,
+      "learning_rate": 6.348325358851675e-05,
+      "loss": 0.0442,
+      "step": 2183
+    },
+    {
+      "epoch": 198.55,
+      "learning_rate": 6.34641148325359e-05,
+      "loss": 0.0423,
+      "step": 2184
+    },
+    {
+      "epoch": 198.64,
+      "learning_rate": 6.344497607655503e-05,
+      "loss": 0.0395,
+      "step": 2185
+    },
+    {
+      "epoch": 198.73,
+      "learning_rate": 6.342583732057416e-05,
+      "loss": 0.0366,
+      "step": 2186
+    },
+    {
+      "epoch": 198.82,
+      "learning_rate": 6.34066985645933e-05,
+      "loss": 0.0434,
+      "step": 2187
+    },
+    {
+      "epoch": 198.91,
+      "learning_rate": 6.338755980861244e-05,
+      "loss": 0.0401,
+      "step": 2188
+    },
+    {
+      "epoch": 199.0,
+      "learning_rate": 6.336842105263158e-05,
+      "loss": 0.0399,
+      "step": 2189
+    },
+    {
+      "epoch": 199.09,
+      "learning_rate": 6.334928229665073e-05,
+      "loss": 0.0417,
+      "step": 2190
+    },
+    {
+      "epoch": 199.18,
+      "learning_rate": 6.333014354066986e-05,
+      "loss": 0.0448,
+      "step": 2191
+    },
+    {
+      "epoch": 199.27,
+      "learning_rate": 6.331100478468899e-05,
+      "loss": 0.0405,
+      "step": 2192
+    },
+    {
+      "epoch": 199.36,
+      "learning_rate": 6.329186602870814e-05,
+      "loss": 0.0455,
+      "step": 2193
+    },
+    {
+      "epoch": 199.45,
+      "learning_rate": 6.327272727272727e-05,
+      "loss": 0.0366,
+      "step": 2194
+    },
+    {
+      "epoch": 199.55,
+      "learning_rate": 6.32535885167464e-05,
+      "loss": 0.0377,
+      "step": 2195
+    },
+    {
+      "epoch": 199.64,
+      "learning_rate": 6.323444976076556e-05,
+      "loss": 0.0341,
+      "step": 2196
+    },
+    {
+      "epoch": 199.73,
+      "learning_rate": 6.32153110047847e-05,
+      "loss": 0.0421,
+      "step": 2197
+    },
+    {
+      "epoch": 199.82,
+      "learning_rate": 6.319617224880383e-05,
+      "loss": 0.0362,
+      "step": 2198
+    },
+    {
+      "epoch": 199.91,
+      "learning_rate": 6.317703349282297e-05,
+      "loss": 0.0579,
+      "step": 2199
+    },
+    {
+      "epoch": 200.0,
+      "learning_rate": 6.31578947368421e-05,
+      "loss": 0.0423,
+      "step": 2200
+    },
+    {
+      "epoch": 200.0,
+      "eval_accuracy_NAT": 0.9931145245383509,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9535575561182883,
+      "eval_accuracy_concrete tiles": 0.9799730936219654,
+      "eval_accuracy_dark bricks": 0.8527770293676148,
+      "eval_accuracy_grass": 0.8819329155555244,
+      "eval_accuracy_red bricks": 0.9151278409090909,
+      "eval_accuracy_road": 0.9100554180416379,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49948997877208945,
+      "eval_accuracy_speedway bricks": 0.9866870135778825,
+      "eval_accuracy_steel": 0.8952740584394169,
+      "eval_accuracy_tiles": 0.045815641692918775,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979164865325187,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8851457683841473,
+      "eval_iou_concrete tiles": 0.9488720468944283,
+      "eval_iou_dark bricks": 0.818742840249806,
+      "eval_iou_grass": 0.825585237775851,
+      "eval_iou_red bricks": 0.8417442430181283,
+      "eval_iou_road": 0.902977795214939,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3363531726877808,
+      "eval_iou_speedway bricks": 0.9757475979124424,
+      "eval_iou_steel": 0.8350392580573222,
+      "eval_iou_tiles": 0.045815641692918775,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11419142782688141,
+      "eval_mean_accuracy": 0.6856773131257523,
+      "eval_mean_iou": 0.64578372824715,
+      "eval_overall_accuracy": 0.9712724685668945,
+      "eval_runtime": 3.3604,
+      "eval_samples_per_second": 10.713,
+      "eval_steps_per_second": 0.595,
+      "step": 2200
+    },
+    {
+      "epoch": 200.09,
+      "learning_rate": 6.313875598086124e-05,
+      "loss": 0.043,
+      "step": 2201
+    },
+    {
+      "epoch": 200.18,
+      "learning_rate": 6.31196172248804e-05,
+      "loss": 0.0445,
+      "step": 2202
+    },
+    {
+      "epoch": 200.27,
+      "learning_rate": 6.310047846889953e-05,
+      "loss": 0.041,
+      "step": 2203
+    },
+    {
+      "epoch": 200.36,
+      "learning_rate": 6.308133971291866e-05,
+      "loss": 0.0369,
+      "step": 2204
+    },
+    {
+      "epoch": 200.45,
+      "learning_rate": 6.30622009569378e-05,
+      "loss": 0.0367,
+      "step": 2205
+    },
+    {
+      "epoch": 200.55,
+      "learning_rate": 6.304306220095694e-05,
+      "loss": 0.0445,
+      "step": 2206
+    },
+    {
+      "epoch": 200.64,
+      "learning_rate": 6.302392344497607e-05,
+      "loss": 0.0416,
+      "step": 2207
+    },
+    {
+      "epoch": 200.73,
+      "learning_rate": 6.300478468899523e-05,
+      "loss": 0.0361,
+      "step": 2208
+    },
+    {
+      "epoch": 200.82,
+      "learning_rate": 6.298564593301436e-05,
+      "loss": 0.0404,
+      "step": 2209
+    },
+    {
+      "epoch": 200.91,
+      "learning_rate": 6.296650717703349e-05,
+      "loss": 0.0376,
+      "step": 2210
+    },
+    {
+      "epoch": 201.0,
+      "learning_rate": 6.294736842105264e-05,
+      "loss": 0.0444,
+      "step": 2211
+    },
+    {
+      "epoch": 201.09,
+      "learning_rate": 6.292822966507177e-05,
+      "loss": 0.0413,
+      "step": 2212
+    },
+    {
+      "epoch": 201.18,
+      "learning_rate": 6.290909090909091e-05,
+      "loss": 0.0366,
+      "step": 2213
+    },
+    {
+      "epoch": 201.27,
+      "learning_rate": 6.288995215311006e-05,
+      "loss": 0.0412,
+      "step": 2214
+    },
+    {
+      "epoch": 201.36,
+      "learning_rate": 6.287081339712919e-05,
+      "loss": 0.0392,
+      "step": 2215
+    },
+    {
+      "epoch": 201.45,
+      "learning_rate": 6.285167464114832e-05,
+      "loss": 0.0363,
+      "step": 2216
+    },
+    {
+      "epoch": 201.55,
+      "learning_rate": 6.283253588516747e-05,
+      "loss": 0.0405,
+      "step": 2217
+    },
+    {
+      "epoch": 201.64,
+      "learning_rate": 6.28133971291866e-05,
+      "loss": 0.0427,
+      "step": 2218
+    },
+    {
+      "epoch": 201.73,
+      "learning_rate": 6.279425837320575e-05,
+      "loss": 0.0383,
+      "step": 2219
+    },
+    {
+      "epoch": 201.82,
+      "learning_rate": 6.277511961722489e-05,
+      "loss": 0.0473,
+      "step": 2220
+    },
+    {
+      "epoch": 201.82,
+      "eval_accuracy_NAT": 0.9925163552539732,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9470497660843813,
+      "eval_accuracy_concrete tiles": 0.9885036384761205,
+      "eval_accuracy_dark bricks": 0.8598976174897041,
+      "eval_accuracy_grass": 0.9050283625230238,
+      "eval_accuracy_red bricks": 0.9083806818181818,
+      "eval_accuracy_road": 0.9181788831021528,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4843271855098834,
+      "eval_accuracy_speedway bricks": 0.9876198820458421,
+      "eval_accuracy_steel": 0.9042526878174263,
+      "eval_accuracy_tiles": 0.03314614436378578,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9799070348896157,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8820303599624248,
+      "eval_iou_concrete tiles": 0.9340960966166826,
+      "eval_iou_dark bricks": 0.8230244980659421,
+      "eval_iou_grass": 0.8343854018644008,
+      "eval_iou_red bricks": 0.8494105927278764,
+      "eval_iou_road": 0.9101699222308526,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.32553830189378496,
+      "eval_iou_speedway bricks": 0.9753367597765363,
+      "eval_iou_steel": 0.836702776930119,
+      "eval_iou_tiles": 0.03314614436378578,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10966138541698456,
+      "eval_mean_accuracy": 0.6868385541911135,
+      "eval_mean_iou": 0.6449036837940016,
+      "eval_overall_accuracy": 0.9720576604207357,
+      "eval_runtime": 3.2404,
+      "eval_samples_per_second": 11.11,
+      "eval_steps_per_second": 0.617,
+      "step": 2220
+    },
+    {
+      "epoch": 201.91,
+      "learning_rate": 6.275598086124402e-05,
+      "loss": 0.039,
+      "step": 2221
+    },
+    {
+      "epoch": 202.0,
+      "learning_rate": 6.273684210526316e-05,
+      "loss": 0.0474,
+      "step": 2222
+    },
+    {
+      "epoch": 202.09,
+      "learning_rate": 6.27177033492823e-05,
+      "loss": 0.0392,
+      "step": 2223
+    },
+    {
+      "epoch": 202.18,
+      "learning_rate": 6.269856459330143e-05,
+      "loss": 0.0374,
+      "step": 2224
+    },
+    {
+      "epoch": 202.27,
+      "learning_rate": 6.267942583732058e-05,
+      "loss": 0.0397,
+      "step": 2225
+    },
+    {
+      "epoch": 202.36,
+      "learning_rate": 6.266028708133973e-05,
+      "loss": 0.0421,
+      "step": 2226
+    },
+    {
+      "epoch": 202.45,
+      "learning_rate": 6.264114832535886e-05,
+      "loss": 0.0426,
+      "step": 2227
+    },
+    {
+      "epoch": 202.55,
+      "learning_rate": 6.262200956937799e-05,
+      "loss": 0.0401,
+      "step": 2228
+    },
+    {
+      "epoch": 202.64,
+      "learning_rate": 6.260287081339713e-05,
+      "loss": 0.0422,
+      "step": 2229
+    },
+    {
+      "epoch": 202.73,
+      "learning_rate": 6.258373205741627e-05,
+      "loss": 0.05,
+      "step": 2230
+    },
+    {
+      "epoch": 202.82,
+      "learning_rate": 6.256459330143541e-05,
+      "loss": 0.0402,
+      "step": 2231
+    },
+    {
+      "epoch": 202.91,
+      "learning_rate": 6.254545454545456e-05,
+      "loss": 0.0392,
+      "step": 2232
+    },
+    {
+      "epoch": 203.0,
+      "learning_rate": 6.252631578947369e-05,
+      "loss": 0.0361,
+      "step": 2233
+    },
+    {
+      "epoch": 203.09,
+      "learning_rate": 6.250717703349282e-05,
+      "loss": 0.0369,
+      "step": 2234
+    },
+    {
+      "epoch": 203.18,
+      "learning_rate": 6.248803827751197e-05,
+      "loss": 0.0499,
+      "step": 2235
+    },
+    {
+      "epoch": 203.27,
+      "learning_rate": 6.24688995215311e-05,
+      "loss": 0.0364,
+      "step": 2236
+    },
+    {
+      "epoch": 203.36,
+      "learning_rate": 6.244976076555024e-05,
+      "loss": 0.0417,
+      "step": 2237
+    },
+    {
+      "epoch": 203.45,
+      "learning_rate": 6.243062200956939e-05,
+      "loss": 0.0416,
+      "step": 2238
+    },
+    {
+      "epoch": 203.55,
+      "learning_rate": 6.241148325358852e-05,
+      "loss": 0.0359,
+      "step": 2239
+    },
+    {
+      "epoch": 203.64,
+      "learning_rate": 6.239234449760765e-05,
+      "loss": 0.0381,
+      "step": 2240
+    },
+    {
+      "epoch": 203.64,
+      "eval_accuracy_NAT": 0.9926790431832346,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9463563350358385,
+      "eval_accuracy_concrete tiles": 0.9801259707698893,
+      "eval_accuracy_dark bricks": 0.8731380624302375,
+      "eval_accuracy_grass": 0.9132347955556643,
+      "eval_accuracy_red bricks": 0.9332386363636364,
+      "eval_accuracy_road": 0.9424537507844742,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4723899319052739,
+      "eval_accuracy_speedway bricks": 0.9868414859761991,
+      "eval_accuracy_steel": 0.9039064045907262,
+      "eval_accuracy_tiles": 0.09485002054513081,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9797065095830372,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8858651640532895,
+      "eval_iou_concrete tiles": 0.94449027695934,
+      "eval_iou_dark bricks": 0.8194263834705967,
+      "eval_iou_grass": 0.8374797274341502,
+      "eval_iou_red bricks": 0.8610747051114024,
+      "eval_iou_road": 0.9350523614541956,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3155791296020038,
+      "eval_iou_speedway bricks": 0.9758083570804634,
+      "eval_iou_steel": 0.837596455038582,
+      "eval_iou_tiles": 0.09485002054513081,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09588578343391418,
+      "eval_mean_accuracy": 0.695324187472331,
+      "eval_mean_iou": 0.6528406992563224,
+      "eval_overall_accuracy": 0.9744073020087348,
+      "eval_runtime": 3.4227,
+      "eval_samples_per_second": 10.518,
+      "eval_steps_per_second": 0.584,
+      "step": 2240
     }
   ],
   "max_steps": 5500,
   "num_train_epochs": 500,
-  "total_flos": 1.0747592742524682e+18,
+  "total_flos": 1.253885819961213e+18,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:746b3a2ae86064feafd1a1638ea5f91a0b337a6ba88e4cc32e3e81d75d8c1b67
+oid sha256:c38aaf2bca6220cd321a7852dd42d1c619139cdccd245637facd0decff1e2e95
 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:bf6cdd304389b270480c5ab57d58505787417355a59586275eacbf6adb401f50
-size 570954
+oid sha256:b0de7575636d95f5e962b17b16aa1b5402648992cf35bd37d0a501476f78d916
+size 591930