Training in progress, step 5120

{checkpoint-4800 → checkpoint-5120}/optimizer.pt

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

{checkpoint-4800 → checkpoint-5120}/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6d11a8bdad6a6650151faa0b26a2700a75591851f28bf457825d455511eed1eb
+oid sha256:caa6c48f1f3d0e2e4eed5d4b0d97bca91e05d2f0e089c52c27a33f0b646addba
 size 14944077

{checkpoint-4800 → checkpoint-5120}/rng_state.pth

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

{checkpoint-4800 → checkpoint-5120}/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:26c7c473fc170adbeaed26d471423ac2439d0cd1fd3b935c9e2bb711e5ae0376
+oid sha256:2bc9e433b785327d7133c00a74a5ead91f06ca4e0d1c2f0f623210227fd1473d
 size 627

{checkpoint-4800 → checkpoint-5120}/trainer_state.json

@@ -1,8 +1,8 @@
 {
   "best_metric": 0.08047032356262207,
   "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-512sq/checkpoint-2560",
-  "epoch": 436.3636363636364,
-  "global_step": 4800,
+  "epoch": 465.45454545454544,
+  "global_step": 5120,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -38166,11 +38166,2555 @@
       "eval_samples_per_second": 9.017,
       "eval_steps_per_second": 0.501,
       "step": 4800
+    },
+    {
+      "epoch": 436.45,
+      "learning_rate": 1.3377990430622008e-05,
+      "loss": 0.0267,
+      "step": 4801
+    },
+    {
+      "epoch": 436.55,
+      "learning_rate": 1.3358851674641149e-05,
+      "loss": 0.0299,
+      "step": 4802
+    },
+    {
+      "epoch": 436.64,
+      "learning_rate": 1.3339712918660288e-05,
+      "loss": 0.0284,
+      "step": 4803
+    },
+    {
+      "epoch": 436.73,
+      "learning_rate": 1.3320574162679428e-05,
+      "loss": 0.0326,
+      "step": 4804
+    },
+    {
+      "epoch": 436.82,
+      "learning_rate": 1.3301435406698565e-05,
+      "loss": 0.0305,
+      "step": 4805
+    },
+    {
+      "epoch": 436.91,
+      "learning_rate": 1.3282296650717704e-05,
+      "loss": 0.0317,
+      "step": 4806
+    },
+    {
+      "epoch": 437.0,
+      "learning_rate": 1.3263157894736844e-05,
+      "loss": 0.0297,
+      "step": 4807
+    },
+    {
+      "epoch": 437.09,
+      "learning_rate": 1.3244019138755981e-05,
+      "loss": 0.0285,
+      "step": 4808
+    },
+    {
+      "epoch": 437.18,
+      "learning_rate": 1.322488038277512e-05,
+      "loss": 0.0286,
+      "step": 4809
+    },
+    {
+      "epoch": 437.27,
+      "learning_rate": 1.320574162679426e-05,
+      "loss": 0.0294,
+      "step": 4810
+    },
+    {
+      "epoch": 437.36,
+      "learning_rate": 1.3186602870813398e-05,
+      "loss": 0.0291,
+      "step": 4811
+    },
+    {
+      "epoch": 437.45,
+      "learning_rate": 1.3167464114832537e-05,
+      "loss": 0.027,
+      "step": 4812
+    },
+    {
+      "epoch": 437.55,
+      "learning_rate": 1.3148325358851677e-05,
+      "loss": 0.0282,
+      "step": 4813
+    },
+    {
+      "epoch": 437.64,
+      "learning_rate": 1.3129186602870812e-05,
+      "loss": 0.0302,
+      "step": 4814
+    },
+    {
+      "epoch": 437.73,
+      "learning_rate": 1.3110047846889953e-05,
+      "loss": 0.034,
+      "step": 4815
+    },
+    {
+      "epoch": 437.82,
+      "learning_rate": 1.3090909090909093e-05,
+      "loss": 0.0305,
+      "step": 4816
+    },
+    {
+      "epoch": 437.91,
+      "learning_rate": 1.3071770334928229e-05,
+      "loss": 0.0318,
+      "step": 4817
+    },
+    {
+      "epoch": 438.0,
+      "learning_rate": 1.305263157894737e-05,
+      "loss": 0.0273,
+      "step": 4818
+    },
+    {
+      "epoch": 438.09,
+      "learning_rate": 1.3033492822966508e-05,
+      "loss": 0.0283,
+      "step": 4819
+    },
+    {
+      "epoch": 438.18,
+      "learning_rate": 1.3014354066985645e-05,
+      "loss": 0.0327,
+      "step": 4820
+    },
+    {
+      "epoch": 438.18,
+      "eval_accuracy_NAT": 0.9935360662552768,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9500434568007211,
+      "eval_accuracy_concrete tiles": 0.9839478994679876,
+      "eval_accuracy_dark bricks": 0.8878796043262384,
+      "eval_accuracy_grass": 0.8986428628847639,
+      "eval_accuracy_red bricks": 0.8824573863636364,
+      "eval_accuracy_road": 0.981964649988452,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.438232293992777,
+      "eval_accuracy_speedway bricks": 0.9883916406296198,
+      "eval_accuracy_steel": 0.9060747971769673,
+      "eval_accuracy_tiles": 0.07745514313107793,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.980845442076097,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8862788081715797,
+      "eval_iou_concrete tiles": 0.9559186098321699,
+      "eval_iou_dark bricks": 0.8362212716595374,
+      "eval_iou_grass": 0.8372283095358265,
+      "eval_iou_red bricks": 0.8304093567251462,
+      "eval_iou_road": 0.97401926036273,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.28902869195243464,
+      "eval_iou_speedway bricks": 0.9772805376303028,
+      "eval_iou_steel": 0.8507989594946117,
+      "eval_iou_tiles": 0.07743393126112556,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08980442583560944,
+      "eval_mean_accuracy": 0.6914327539244245,
+      "eval_mean_iou": 0.6534971675924278,
+      "eval_overall_accuracy": 0.9783986409505209,
+      "eval_runtime": 4.1851,
+      "eval_samples_per_second": 8.602,
+      "eval_steps_per_second": 0.478,
+      "step": 4820
+    },
+    {
+      "epoch": 438.27,
+      "learning_rate": 1.2995215311004786e-05,
+      "loss": 0.0283,
+      "step": 4821
+    },
+    {
+      "epoch": 438.36,
+      "learning_rate": 1.2976076555023924e-05,
+      "loss": 0.0294,
+      "step": 4822
+    },
+    {
+      "epoch": 438.45,
+      "learning_rate": 1.2956937799043061e-05,
+      "loss": 0.0249,
+      "step": 4823
+    },
+    {
+      "epoch": 438.55,
+      "learning_rate": 1.2937799043062202e-05,
+      "loss": 0.0299,
+      "step": 4824
+    },
+    {
+      "epoch": 438.64,
+      "learning_rate": 1.291866028708134e-05,
+      "loss": 0.0312,
+      "step": 4825
+    },
+    {
+      "epoch": 438.73,
+      "learning_rate": 1.2899521531100478e-05,
+      "loss": 0.0289,
+      "step": 4826
+    },
+    {
+      "epoch": 438.82,
+      "learning_rate": 1.2880382775119618e-05,
+      "loss": 0.0283,
+      "step": 4827
+    },
+    {
+      "epoch": 438.91,
+      "learning_rate": 1.2861244019138757e-05,
+      "loss": 0.0317,
+      "step": 4828
+    },
+    {
+      "epoch": 439.0,
+      "learning_rate": 1.2842105263157894e-05,
+      "loss": 0.0354,
+      "step": 4829
+    },
+    {
+      "epoch": 439.09,
+      "learning_rate": 1.2822966507177035e-05,
+      "loss": 0.0277,
+      "step": 4830
+    },
+    {
+      "epoch": 439.18,
+      "learning_rate": 1.2803827751196173e-05,
+      "loss": 0.0319,
+      "step": 4831
+    },
+    {
+      "epoch": 439.27,
+      "learning_rate": 1.278468899521531e-05,
+      "loss": 0.03,
+      "step": 4832
+    },
+    {
+      "epoch": 439.36,
+      "learning_rate": 1.276555023923445e-05,
+      "loss": 0.0284,
+      "step": 4833
+    },
+    {
+      "epoch": 439.45,
+      "learning_rate": 1.274641148325359e-05,
+      "loss": 0.0267,
+      "step": 4834
+    },
+    {
+      "epoch": 439.55,
+      "learning_rate": 1.2727272727272727e-05,
+      "loss": 0.0285,
+      "step": 4835
+    },
+    {
+      "epoch": 439.64,
+      "learning_rate": 1.2708133971291866e-05,
+      "loss": 0.0302,
+      "step": 4836
+    },
+    {
+      "epoch": 439.73,
+      "learning_rate": 1.2688995215311006e-05,
+      "loss": 0.0294,
+      "step": 4837
+    },
+    {
+      "epoch": 439.82,
+      "learning_rate": 1.2669856459330145e-05,
+      "loss": 0.0349,
+      "step": 4838
+    },
+    {
+      "epoch": 439.91,
+      "learning_rate": 1.2650717703349282e-05,
+      "loss": 0.0311,
+      "step": 4839
+    },
+    {
+      "epoch": 440.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0291,
+      "step": 4840
+    },
+    {
+      "epoch": 440.0,
+      "eval_accuracy_NAT": 0.9935316549773684,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9529472831451994,
+      "eval_accuracy_concrete tiles": 0.98287775943252,
+      "eval_accuracy_dark bricks": 0.8857241830568492,
+      "eval_accuracy_grass": 0.8975069639343346,
+      "eval_accuracy_red bricks": 0.875,
+      "eval_accuracy_road": 0.9803684943052309,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.42808700686461004,
+      "eval_accuracy_speedway bricks": 0.9879698585732785,
+      "eval_accuracy_steel": 0.9070476881472198,
+      "eval_accuracy_tiles": 0.0777290782084646,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808598820870905,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8852350557004325,
+      "eval_iou_concrete tiles": 0.9562708234174203,
+      "eval_iou_dark bricks": 0.8361614766905272,
+      "eval_iou_grass": 0.8373643072127096,
+      "eval_iou_red bricks": 0.8232542599398597,
+      "eval_iou_road": 0.9735476673939487,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2839327835579367,
+      "eval_iou_speedway bricks": 0.9772873916134296,
+      "eval_iou_steel": 0.8519037626123789,
+      "eval_iou_tiles": 0.07770779131863617,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09085646271705627,
+      "eval_mean_accuracy": 0.6899069208188521,
+      "eval_mean_iou": 0.6525788616572593,
+      "eval_overall_accuracy": 0.9783368640475802,
+      "eval_runtime": 3.7187,
+      "eval_samples_per_second": 9.681,
+      "eval_steps_per_second": 0.538,
+      "step": 4840
+    },
+    {
+      "epoch": 440.09,
+      "learning_rate": 1.2612440191387561e-05,
+      "loss": 0.0351,
+      "step": 4841
+    },
+    {
+      "epoch": 440.18,
+      "learning_rate": 1.2593301435406698e-05,
+      "loss": 0.0292,
+      "step": 4842
+    },
+    {
+      "epoch": 440.27,
+      "learning_rate": 1.2574162679425839e-05,
+      "loss": 0.0284,
+      "step": 4843
+    },
+    {
+      "epoch": 440.36,
+      "learning_rate": 1.2555023923444977e-05,
+      "loss": 0.0315,
+      "step": 4844
+    },
+    {
+      "epoch": 440.45,
+      "learning_rate": 1.2535885167464115e-05,
+      "loss": 0.0324,
+      "step": 4845
+    },
+    {
+      "epoch": 440.55,
+      "learning_rate": 1.2516746411483255e-05,
+      "loss": 0.0277,
+      "step": 4846
+    },
+    {
+      "epoch": 440.64,
+      "learning_rate": 1.2497607655502392e-05,
+      "loss": 0.0433,
+      "step": 4847
+    },
+    {
+      "epoch": 440.73,
+      "learning_rate": 1.2478468899521533e-05,
+      "loss": 0.0279,
+      "step": 4848
+    },
+    {
+      "epoch": 440.82,
+      "learning_rate": 1.2459330143540671e-05,
+      "loss": 0.0301,
+      "step": 4849
+    },
+    {
+      "epoch": 440.91,
+      "learning_rate": 1.2440191387559808e-05,
+      "loss": 0.0325,
+      "step": 4850
+    },
+    {
+      "epoch": 441.0,
+      "learning_rate": 1.2421052631578949e-05,
+      "loss": 0.0263,
+      "step": 4851
+    },
+    {
+      "epoch": 441.09,
+      "learning_rate": 1.2401913875598086e-05,
+      "loss": 0.0283,
+      "step": 4852
+    },
+    {
+      "epoch": 441.18,
+      "learning_rate": 1.2382775119617225e-05,
+      "loss": 0.0307,
+      "step": 4853
+    },
+    {
+      "epoch": 441.27,
+      "learning_rate": 1.2363636363636365e-05,
+      "loss": 0.0285,
+      "step": 4854
+    },
+    {
+      "epoch": 441.36,
+      "learning_rate": 1.2344497607655502e-05,
+      "loss": 0.0258,
+      "step": 4855
+    },
+    {
+      "epoch": 441.45,
+      "learning_rate": 1.2325358851674641e-05,
+      "loss": 0.0313,
+      "step": 4856
+    },
+    {
+      "epoch": 441.55,
+      "learning_rate": 1.2306220095693782e-05,
+      "loss": 0.031,
+      "step": 4857
+    },
+    {
+      "epoch": 441.64,
+      "learning_rate": 1.2287081339712919e-05,
+      "loss": 0.0262,
+      "step": 4858
+    },
+    {
+      "epoch": 441.73,
+      "learning_rate": 1.2267942583732057e-05,
+      "loss": 0.0297,
+      "step": 4859
+    },
+    {
+      "epoch": 441.82,
+      "learning_rate": 1.2248803827751198e-05,
+      "loss": 0.03,
+      "step": 4860
+    },
+    {
+      "epoch": 441.82,
+      "eval_accuracy_NAT": 0.9931341106122642,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9522645821709086,
+      "eval_accuracy_concrete tiles": 0.983305815446707,
+      "eval_accuracy_dark bricks": 0.8856856933913244,
+      "eval_accuracy_grass": 0.9036303330455723,
+      "eval_accuracy_red bricks": 0.892578125,
+      "eval_accuracy_road": 0.98098156319265,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5448405149835966,
+      "eval_accuracy_speedway bricks": 0.987243355574945,
+      "eval_accuracy_steel": 0.9102054613811754,
+      "eval_accuracy_tiles": 0.07759211066977126,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.981022816125752,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8889325717490979,
+      "eval_iou_concrete tiles": 0.9564312267657993,
+      "eval_iou_dark bricks": 0.8351843786295006,
+      "eval_iou_grass": 0.8376010781671159,
+      "eval_iou_red bricks": 0.836996336996337,
+      "eval_iou_road": 0.9727804364483146,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35402335912869015,
+      "eval_iou_speedway bricks": 0.9772272792431193,
+      "eval_iou_steel": 0.8517695530402981,
+      "eval_iou_tiles": 0.07757086128988087,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08871608227491379,
+      "eval_mean_accuracy": 0.7008816665745319,
+      "eval_mean_iou": 0.6591953767372234,
+      "eval_overall_accuracy": 0.978656980726454,
+      "eval_runtime": 3.5134,
+      "eval_samples_per_second": 10.246,
+      "eval_steps_per_second": 0.569,
+      "step": 4860
+    },
+    {
+      "epoch": 441.91,
+      "learning_rate": 1.2229665071770335e-05,
+      "loss": 0.0298,
+      "step": 4861
+    },
+    {
+      "epoch": 442.0,
+      "learning_rate": 1.2210526315789474e-05,
+      "loss": 0.0349,
+      "step": 4862
+    },
+    {
+      "epoch": 442.09,
+      "learning_rate": 1.2191387559808613e-05,
+      "loss": 0.0276,
+      "step": 4863
+    },
+    {
+      "epoch": 442.18,
+      "learning_rate": 1.2172248803827751e-05,
+      "loss": 0.0289,
+      "step": 4864
+    },
+    {
+      "epoch": 442.27,
+      "learning_rate": 1.2153110047846892e-05,
+      "loss": 0.0302,
+      "step": 4865
+    },
+    {
+      "epoch": 442.36,
+      "learning_rate": 1.2133971291866029e-05,
+      "loss": 0.0311,
+      "step": 4866
+    },
+    {
+      "epoch": 442.45,
+      "learning_rate": 1.2114832535885168e-05,
+      "loss": 0.0296,
+      "step": 4867
+    },
+    {
+      "epoch": 442.55,
+      "learning_rate": 1.2095693779904308e-05,
+      "loss": 0.0313,
+      "step": 4868
+    },
+    {
+      "epoch": 442.64,
+      "learning_rate": 1.2076555023923445e-05,
+      "loss": 0.0277,
+      "step": 4869
+    },
+    {
+      "epoch": 442.73,
+      "learning_rate": 1.2057416267942584e-05,
+      "loss": 0.0321,
+      "step": 4870
+    },
+    {
+      "epoch": 442.82,
+      "learning_rate": 1.2038277511961724e-05,
+      "loss": 0.0291,
+      "step": 4871
+    },
+    {
+      "epoch": 442.91,
+      "learning_rate": 1.2019138755980862e-05,
+      "loss": 0.0302,
+      "step": 4872
+    },
+    {
+      "epoch": 443.0,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0252,
+      "step": 4873
+    },
+    {
+      "epoch": 443.09,
+      "learning_rate": 1.1980861244019139e-05,
+      "loss": 0.0303,
+      "step": 4874
+    },
+    {
+      "epoch": 443.18,
+      "learning_rate": 1.1961722488038278e-05,
+      "loss": 0.0272,
+      "step": 4875
+    },
+    {
+      "epoch": 443.27,
+      "learning_rate": 1.1942583732057417e-05,
+      "loss": 0.0317,
+      "step": 4876
+    },
+    {
+      "epoch": 443.36,
+      "learning_rate": 1.1923444976076555e-05,
+      "loss": 0.0285,
+      "step": 4877
+    },
+    {
+      "epoch": 443.45,
+      "learning_rate": 1.1904306220095694e-05,
+      "loss": 0.0313,
+      "step": 4878
+    },
+    {
+      "epoch": 443.55,
+      "learning_rate": 1.1885167464114833e-05,
+      "loss": 0.0293,
+      "step": 4879
+    },
+    {
+      "epoch": 443.64,
+      "learning_rate": 1.1866028708133972e-05,
+      "loss": 0.035,
+      "step": 4880
+    },
+    {
+      "epoch": 443.64,
+      "eval_accuracy_NAT": 0.9934324894499877,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9508549186660372,
+      "eval_accuracy_concrete tiles": 0.9852626429401333,
+      "eval_accuracy_dark bricks": 0.8869943420191678,
+      "eval_accuracy_grass": 0.9012362075654365,
+      "eval_accuracy_red bricks": 0.8831676136363636,
+      "eval_accuracy_road": 0.9800649828836487,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46753783806136795,
+      "eval_accuracy_speedway bricks": 0.9883162146538792,
+      "eval_accuracy_steel": 0.9056543104016885,
+      "eval_accuracy_tiles": 0.07944117244213121,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810547336156806,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8853818274464158,
+      "eval_iou_concrete tiles": 0.9550964758884377,
+      "eval_iou_dark bricks": 0.8379085917899866,
+      "eval_iou_grass": 0.8385631266239785,
+      "eval_iou_red bricks": 0.8330262937531402,
+      "eval_iou_road": 0.9723973855079975,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.30747892303508295,
+      "eval_iou_speedway bricks": 0.9772491053220065,
+      "eval_iou_steel": 0.8513796310649512,
+      "eval_iou_tiles": 0.0794194166780775,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08974248915910721,
+      "eval_mean_accuracy": 0.6939971332861418,
+      "eval_mean_iou": 0.6553042700558274,
+      "eval_overall_accuracy": 0.9784123102823893,
+      "eval_runtime": 3.9568,
+      "eval_samples_per_second": 9.098,
+      "eval_steps_per_second": 0.505,
+      "step": 4880
+    },
+    {
+      "epoch": 443.73,
+      "learning_rate": 1.184688995215311e-05,
+      "loss": 0.0285,
+      "step": 4881
+    },
+    {
+      "epoch": 443.82,
+      "learning_rate": 1.182775119617225e-05,
+      "loss": 0.0306,
+      "step": 4882
+    },
+    {
+      "epoch": 443.91,
+      "learning_rate": 1.1808612440191388e-05,
+      "loss": 0.0281,
+      "step": 4883
+    },
+    {
+      "epoch": 444.0,
+      "learning_rate": 1.1789473684210527e-05,
+      "loss": 0.0299,
+      "step": 4884
+    },
+    {
+      "epoch": 444.09,
+      "learning_rate": 1.1770334928229666e-05,
+      "loss": 0.0315,
+      "step": 4885
+    },
+    {
+      "epoch": 444.18,
+      "learning_rate": 1.1751196172248804e-05,
+      "loss": 0.0255,
+      "step": 4886
+    },
+    {
+      "epoch": 444.27,
+      "learning_rate": 1.1732057416267943e-05,
+      "loss": 0.0294,
+      "step": 4887
+    },
+    {
+      "epoch": 444.36,
+      "learning_rate": 1.1712918660287082e-05,
+      "loss": 0.0313,
+      "step": 4888
+    },
+    {
+      "epoch": 444.45,
+      "learning_rate": 1.169377990430622e-05,
+      "loss": 0.0281,
+      "step": 4889
+    },
+    {
+      "epoch": 444.55,
+      "learning_rate": 1.167464114832536e-05,
+      "loss": 0.029,
+      "step": 4890
+    },
+    {
+      "epoch": 444.64,
+      "learning_rate": 1.1655502392344498e-05,
+      "loss": 0.0244,
+      "step": 4891
+    },
+    {
+      "epoch": 444.73,
+      "learning_rate": 1.1636363636363637e-05,
+      "loss": 0.0306,
+      "step": 4892
+    },
+    {
+      "epoch": 444.82,
+      "learning_rate": 1.1617224880382776e-05,
+      "loss": 0.0326,
+      "step": 4893
+    },
+    {
+      "epoch": 444.91,
+      "learning_rate": 1.1598086124401915e-05,
+      "loss": 0.0294,
+      "step": 4894
+    },
+    {
+      "epoch": 445.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 0.0273,
+      "step": 4895
+    },
+    {
+      "epoch": 445.09,
+      "learning_rate": 1.1559808612440192e-05,
+      "loss": 0.0282,
+      "step": 4896
+    },
+    {
+      "epoch": 445.18,
+      "learning_rate": 1.1540669856459331e-05,
+      "loss": 0.026,
+      "step": 4897
+    },
+    {
+      "epoch": 445.27,
+      "learning_rate": 1.152153110047847e-05,
+      "loss": 0.0299,
+      "step": 4898
+    },
+    {
+      "epoch": 445.36,
+      "learning_rate": 1.1502392344497607e-05,
+      "loss": 0.0361,
+      "step": 4899
+    },
+    {
+      "epoch": 445.45,
+      "learning_rate": 1.1483253588516747e-05,
+      "loss": 0.0277,
+      "step": 4900
+    },
+    {
+      "epoch": 445.45,
+      "eval_accuracy_NAT": 0.9935106572945245,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9493956285677497,
+      "eval_accuracy_concrete tiles": 0.9871277441448052,
+      "eval_accuracy_dark bricks": 0.8859166313844733,
+      "eval_accuracy_grass": 0.9041755645417784,
+      "eval_accuracy_red bricks": 0.8776633522727273,
+      "eval_accuracy_road": 0.9804374192097336,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4598461665701762,
+      "eval_accuracy_speedway bricks": 0.9880404572865716,
+      "eval_accuracy_steel": 0.9065117736297078,
+      "eval_accuracy_tiles": 0.07978359128886454,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809280089087294,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8866451886052705,
+      "eval_iou_concrete tiles": 0.95452798391627,
+      "eval_iou_dark bricks": 0.839301341890315,
+      "eval_iou_grass": 0.8381254758395024,
+      "eval_iou_red bricks": 0.8299194089993284,
+      "eval_iou_road": 0.972432072631762,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3038362053262414,
+      "eval_iou_speedway bricks": 0.977217238627866,
+      "eval_iou_steel": 0.8502613833209601,
+      "eval_iou_tiles": 0.07976174174996577,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08970047533512115,
+      "eval_mean_accuracy": 0.6932622297070086,
+      "eval_mean_iou": 0.6548427730627855,
+      "eval_overall_accuracy": 0.9783994886610243,
+      "eval_runtime": 4.1267,
+      "eval_samples_per_second": 8.724,
+      "eval_steps_per_second": 0.485,
+      "step": 4900
+    },
+    {
+      "epoch": 445.55,
+      "learning_rate": 1.1464114832535886e-05,
+      "loss": 0.0328,
+      "step": 4901
+    },
+    {
+      "epoch": 445.64,
+      "learning_rate": 1.1444976076555025e-05,
+      "loss": 0.0289,
+      "step": 4902
+    },
+    {
+      "epoch": 445.73,
+      "learning_rate": 1.1425837320574164e-05,
+      "loss": 0.0298,
+      "step": 4903
+    },
+    {
+      "epoch": 445.82,
+      "learning_rate": 1.1406698564593302e-05,
+      "loss": 0.029,
+      "step": 4904
+    },
+    {
+      "epoch": 445.91,
+      "learning_rate": 1.1387559808612441e-05,
+      "loss": 0.0269,
+      "step": 4905
+    },
+    {
+      "epoch": 446.0,
+      "learning_rate": 1.136842105263158e-05,
+      "loss": 0.0324,
+      "step": 4906
+    },
+    {
+      "epoch": 446.09,
+      "learning_rate": 1.1349282296650717e-05,
+      "loss": 0.0291,
+      "step": 4907
+    },
+    {
+      "epoch": 446.18,
+      "learning_rate": 1.1330143540669858e-05,
+      "loss": 0.0315,
+      "step": 4908
+    },
+    {
+      "epoch": 446.27,
+      "learning_rate": 1.1311004784688996e-05,
+      "loss": 0.0318,
+      "step": 4909
+    },
+    {
+      "epoch": 446.36,
+      "learning_rate": 1.1291866028708133e-05,
+      "loss": 0.0295,
+      "step": 4910
+    },
+    {
+      "epoch": 446.45,
+      "learning_rate": 1.1272727272727274e-05,
+      "loss": 0.0279,
+      "step": 4911
+    },
+    {
+      "epoch": 446.55,
+      "learning_rate": 1.1253588516746413e-05,
+      "loss": 0.0304,
+      "step": 4912
+    },
+    {
+      "epoch": 446.64,
+      "learning_rate": 1.123444976076555e-05,
+      "loss": 0.0295,
+      "step": 4913
+    },
+    {
+      "epoch": 446.73,
+      "learning_rate": 1.121531100478469e-05,
+      "loss": 0.0317,
+      "step": 4914
+    },
+    {
+      "epoch": 446.82,
+      "learning_rate": 1.1196172248803829e-05,
+      "loss": 0.0258,
+      "step": 4915
+    },
+    {
+      "epoch": 446.91,
+      "learning_rate": 1.1177033492822966e-05,
+      "loss": 0.0305,
+      "step": 4916
+    },
+    {
+      "epoch": 447.0,
+      "learning_rate": 1.1157894736842106e-05,
+      "loss": 0.0259,
+      "step": 4917
+    },
+    {
+      "epoch": 447.09,
+      "learning_rate": 1.1138755980861244e-05,
+      "loss": 0.0282,
+      "step": 4918
+    },
+    {
+      "epoch": 447.18,
+      "learning_rate": 1.1119617224880384e-05,
+      "loss": 0.0289,
+      "step": 4919
+    },
+    {
+      "epoch": 447.27,
+      "learning_rate": 1.1100478468899523e-05,
+      "loss": 0.0266,
+      "step": 4920
+    },
+    {
+      "epoch": 447.27,
+      "eval_accuracy_NAT": 0.9937460430837162,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9500206553929353,
+      "eval_accuracy_concrete tiles": 0.9873723475814835,
+      "eval_accuracy_dark bricks": 0.8857626727223741,
+      "eval_accuracy_grass": 0.8989958653278204,
+      "eval_accuracy_red bricks": 0.8776633522727273,
+      "eval_accuracy_road": 0.9808957093642343,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4744851542469605,
+      "eval_accuracy_speedway bricks": 0.9888490237465108,
+      "eval_accuracy_steel": 0.8998911681287514,
+      "eval_accuracy_tiles": 0.06862073688535816,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810510278666376,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.887473468702466,
+      "eval_iou_concrete tiles": 0.9559232727488012,
+      "eval_iou_dark bricks": 0.8383301154784889,
+      "eval_iou_grass": 0.838354307020889,
+      "eval_iou_red bricks": 0.830337644884932,
+      "eval_iou_road": 0.9735065957487663,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.312807837007688,
+      "eval_iou_speedway bricks": 0.9772616679309012,
+      "eval_iou_steel": 0.8490548424737456,
+      "eval_iou_tiles": 0.0686113393590797,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09037618339061737,
+      "eval_mean_accuracy": 0.6927925175963748,
+      "eval_mean_iou": 0.6548240091709535,
+      "eval_overall_accuracy": 0.9785696665445963,
+      "eval_runtime": 3.9527,
+      "eval_samples_per_second": 9.108,
+      "eval_steps_per_second": 0.506,
+      "step": 4920
+    },
+    {
+      "epoch": 447.36,
+      "learning_rate": 1.108133971291866e-05,
+      "loss": 0.0299,
+      "step": 4921
+    },
+    {
+      "epoch": 447.45,
+      "learning_rate": 1.10622009569378e-05,
+      "loss": 0.0313,
+      "step": 4922
+    },
+    {
+      "epoch": 447.55,
+      "learning_rate": 1.104306220095694e-05,
+      "loss": 0.0294,
+      "step": 4923
+    },
+    {
+      "epoch": 447.64,
+      "learning_rate": 1.1023923444976076e-05,
+      "loss": 0.0261,
+      "step": 4924
+    },
+    {
+      "epoch": 447.73,
+      "learning_rate": 1.1004784688995217e-05,
+      "loss": 0.0316,
+      "step": 4925
+    },
+    {
+      "epoch": 447.82,
+      "learning_rate": 1.0985645933014354e-05,
+      "loss": 0.0278,
+      "step": 4926
+    },
+    {
+      "epoch": 447.91,
+      "learning_rate": 1.0966507177033493e-05,
+      "loss": 0.0287,
+      "step": 4927
+    },
+    {
+      "epoch": 448.0,
+      "learning_rate": 1.0947368421052633e-05,
+      "loss": 0.03,
+      "step": 4928
+    },
+    {
+      "epoch": 448.09,
+      "learning_rate": 1.092822966507177e-05,
+      "loss": 0.0311,
+      "step": 4929
+    },
+    {
+      "epoch": 448.18,
+      "learning_rate": 1.0909090909090909e-05,
+      "loss": 0.0309,
+      "step": 4930
+    },
+    {
+      "epoch": 448.27,
+      "learning_rate": 1.088995215311005e-05,
+      "loss": 0.0253,
+      "step": 4931
+    },
+    {
+      "epoch": 448.36,
+      "learning_rate": 1.0870813397129186e-05,
+      "loss": 0.0268,
+      "step": 4932
+    },
+    {
+      "epoch": 448.45,
+      "learning_rate": 1.0851674641148325e-05,
+      "loss": 0.0279,
+      "step": 4933
+    },
+    {
+      "epoch": 448.55,
+      "learning_rate": 1.0832535885167466e-05,
+      "loss": 0.0271,
+      "step": 4934
+    },
+    {
+      "epoch": 448.64,
+      "learning_rate": 1.0813397129186603e-05,
+      "loss": 0.0293,
+      "step": 4935
+    },
+    {
+      "epoch": 448.73,
+      "learning_rate": 1.0794258373205742e-05,
+      "loss": 0.0296,
+      "step": 4936
+    },
+    {
+      "epoch": 448.82,
+      "learning_rate": 1.077511961722488e-05,
+      "loss": 0.03,
+      "step": 4937
+    },
+    {
+      "epoch": 448.91,
+      "learning_rate": 1.0755980861244019e-05,
+      "loss": 0.0308,
+      "step": 4938
+    },
+    {
+      "epoch": 449.0,
+      "learning_rate": 1.073684210526316e-05,
+      "loss": 0.036,
+      "step": 4939
+    },
+    {
+      "epoch": 449.09,
+      "learning_rate": 1.0717703349282297e-05,
+      "loss": 0.0302,
+      "step": 4940
+    },
+    {
+      "epoch": 449.09,
+      "eval_accuracy_NAT": 0.9933292655469315,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9516905231984205,
+      "eval_accuracy_concrete tiles": 0.985018039503455,
+      "eval_accuracy_dark bricks": 0.8882645009814865,
+      "eval_accuracy_grass": 0.9024035621791086,
+      "eval_accuracy_red bricks": 0.8849431818181818,
+      "eval_accuracy_road": 0.9801181880730894,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4777382626195793,
+      "eval_accuracy_speedway bricks": 0.9883572463846821,
+      "eval_accuracy_steel": 0.9039228942681881,
+      "eval_accuracy_tiles": 0.07670182166826463,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809905736163723,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8869707614035527,
+      "eval_iou_concrete tiles": 0.9559643916913947,
+      "eval_iou_dark bricks": 0.8378594249201278,
+      "eval_iou_grass": 0.836859778430343,
+      "eval_iou_red bricks": 0.833863142044504,
+      "eval_iou_road": 0.9731799064214302,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31295035486608996,
+      "eval_iou_speedway bricks": 0.9773450752926589,
+      "eval_iou_steel": 0.8491727855748675,
+      "eval_iou_tiles": 0.0766860664156111,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08972227573394775,
+      "eval_mean_accuracy": 0.694806729710876,
+      "eval_mean_iou": 0.655526327744381,
+      "eval_overall_accuracy": 0.9784800211588541,
+      "eval_runtime": 4.1513,
+      "eval_samples_per_second": 8.672,
+      "eval_steps_per_second": 0.482,
+      "step": 4940
+    },
+    {
+      "epoch": 449.18,
+      "learning_rate": 1.0698564593301435e-05,
+      "loss": 0.0277,
+      "step": 4941
+    },
+    {
+      "epoch": 449.27,
+      "learning_rate": 1.0679425837320576e-05,
+      "loss": 0.032,
+      "step": 4942
+    },
+    {
+      "epoch": 449.36,
+      "learning_rate": 1.0660287081339713e-05,
+      "loss": 0.028,
+      "step": 4943
+    },
+    {
+      "epoch": 449.45,
+      "learning_rate": 1.0641148325358852e-05,
+      "loss": 0.0273,
+      "step": 4944
+    },
+    {
+      "epoch": 449.55,
+      "learning_rate": 1.0622009569377992e-05,
+      "loss": 0.028,
+      "step": 4945
+    },
+    {
+      "epoch": 449.64,
+      "learning_rate": 1.060287081339713e-05,
+      "loss": 0.037,
+      "step": 4946
+    },
+    {
+      "epoch": 449.73,
+      "learning_rate": 1.0583732057416268e-05,
+      "loss": 0.0311,
+      "step": 4947
+    },
+    {
+      "epoch": 449.82,
+      "learning_rate": 1.0564593301435407e-05,
+      "loss": 0.0273,
+      "step": 4948
+    },
+    {
+      "epoch": 449.91,
+      "learning_rate": 1.0545454545454546e-05,
+      "loss": 0.0295,
+      "step": 4949
+    },
+    {
+      "epoch": 450.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0275,
+      "step": 4950
+    },
+    {
+      "epoch": 450.09,
+      "learning_rate": 1.0507177033492823e-05,
+      "loss": 0.0268,
+      "step": 4951
+    },
+    {
+      "epoch": 450.18,
+      "learning_rate": 1.0488038277511962e-05,
+      "loss": 0.0329,
+      "step": 4952
+    },
+    {
+      "epoch": 450.27,
+      "learning_rate": 1.04688995215311e-05,
+      "loss": 0.0338,
+      "step": 4953
+    },
+    {
+      "epoch": 450.36,
+      "learning_rate": 1.044976076555024e-05,
+      "loss": 0.0331,
+      "step": 4954
+    },
+    {
+      "epoch": 450.45,
+      "learning_rate": 1.0430622009569378e-05,
+      "loss": 0.03,
+      "step": 4955
+    },
+    {
+      "epoch": 450.55,
+      "learning_rate": 1.0411483253588517e-05,
+      "loss": 0.0259,
+      "step": 4956
+    },
+    {
+      "epoch": 450.64,
+      "learning_rate": 1.0392344497607656e-05,
+      "loss": 0.0265,
+      "step": 4957
+    },
+    {
+      "epoch": 450.73,
+      "learning_rate": 1.0373205741626795e-05,
+      "loss": 0.0264,
+      "step": 4958
+    },
+    {
+      "epoch": 450.82,
+      "learning_rate": 1.0354066985645933e-05,
+      "loss": 0.0285,
+      "step": 4959
+    },
+    {
+      "epoch": 450.91,
+      "learning_rate": 1.0334928229665072e-05,
+      "loss": 0.0291,
+      "step": 4960
+    },
+    {
+      "epoch": 450.91,
+      "eval_accuracy_NAT": 0.9937225750852435,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9520003540924503,
+      "eval_accuracy_concrete tiles": 0.9839478994679876,
+      "eval_accuracy_dark bricks": 0.8856087140602749,
+      "eval_accuracy_grass": 0.8953574936127529,
+      "eval_accuracy_red bricks": 0.8758877840909091,
+      "eval_accuracy_road": 0.9823890822951268,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4838309486394839,
+      "eval_accuracy_speedway bricks": 0.9875975559570229,
+      "eval_accuracy_steel": 0.9049699887870193,
+      "eval_accuracy_tiles": 0.07903026982605123,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809683928307661,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8881399239467602,
+      "eval_iou_concrete tiles": 0.9562310572294527,
+      "eval_iou_dark bricks": 0.8390096266044341,
+      "eval_iou_grass": 0.8366055974657914,
+      "eval_iou_red bricks": 0.8278234603121329,
+      "eval_iou_road": 0.9744262334406804,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31682703590706407,
+      "eval_iou_speedway bricks": 0.9773414664003416,
+      "eval_iou_steel": 0.8498799845141308,
+      "eval_iou_tiles": 0.07900862659181158,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08985412120819092,
+      "eval_mean_accuracy": 0.6941802050703323,
+      "eval_mean_iou": 0.6558662619417974,
+      "eval_overall_accuracy": 0.9786167144775391,
+      "eval_runtime": 3.6961,
+      "eval_samples_per_second": 9.74,
+      "eval_steps_per_second": 0.541,
+      "step": 4960
+    },
+    {
+      "epoch": 451.0,
+      "learning_rate": 1.0315789473684211e-05,
+      "loss": 0.0326,
+      "step": 4961
+    },
+    {
+      "epoch": 451.09,
+      "learning_rate": 1.029665071770335e-05,
+      "loss": 0.0268,
+      "step": 4962
+    },
+    {
+      "epoch": 451.18,
+      "learning_rate": 1.0277511961722489e-05,
+      "loss": 0.0296,
+      "step": 4963
+    },
+    {
+      "epoch": 451.27,
+      "learning_rate": 1.0258373205741627e-05,
+      "loss": 0.0253,
+      "step": 4964
+    },
+    {
+      "epoch": 451.36,
+      "learning_rate": 1.0239234449760766e-05,
+      "loss": 0.0287,
+      "step": 4965
+    },
+    {
+      "epoch": 451.45,
+      "learning_rate": 1.0220095693779905e-05,
+      "loss": 0.0285,
+      "step": 4966
+    },
+    {
+      "epoch": 451.55,
+      "learning_rate": 1.0200956937799044e-05,
+      "loss": 0.0305,
+      "step": 4967
+    },
+    {
+      "epoch": 451.64,
+      "learning_rate": 1.0181818181818182e-05,
+      "loss": 0.028,
+      "step": 4968
+    },
+    {
+      "epoch": 451.73,
+      "learning_rate": 1.0162679425837321e-05,
+      "loss": 0.0285,
+      "step": 4969
+    },
+    {
+      "epoch": 451.82,
+      "learning_rate": 1.014354066985646e-05,
+      "loss": 0.0247,
+      "step": 4970
+    },
+    {
+      "epoch": 451.91,
+      "learning_rate": 1.0124401913875599e-05,
+      "loss": 0.0287,
+      "step": 4971
+    },
+    {
+      "epoch": 452.0,
+      "learning_rate": 1.0105263157894738e-05,
+      "loss": 0.0297,
+      "step": 4972
+    },
+    {
+      "epoch": 452.09,
+      "learning_rate": 1.0086124401913876e-05,
+      "loss": 0.0302,
+      "step": 4973
+    },
+    {
+      "epoch": 452.18,
+      "learning_rate": 1.0066985645933015e-05,
+      "loss": 0.0368,
+      "step": 4974
+    },
+    {
+      "epoch": 452.27,
+      "learning_rate": 1.0047846889952154e-05,
+      "loss": 0.0258,
+      "step": 4975
+    },
+    {
+      "epoch": 452.36,
+      "learning_rate": 1.0028708133971293e-05,
+      "loss": 0.0301,
+      "step": 4976
+    },
+    {
+      "epoch": 452.45,
+      "learning_rate": 1.0009569377990431e-05,
+      "loss": 0.028,
+      "step": 4977
+    },
+    {
+      "epoch": 452.55,
+      "learning_rate": 9.99043062200957e-06,
+      "loss": 0.033,
+      "step": 4978
+    },
+    {
+      "epoch": 452.64,
+      "learning_rate": 9.971291866028709e-06,
+      "loss": 0.0288,
+      "step": 4979
+    },
+    {
+      "epoch": 452.73,
+      "learning_rate": 9.952153110047848e-06,
+      "loss": 0.0311,
+      "step": 4980
+    },
+    {
+      "epoch": 452.73,
+      "eval_accuracy_NAT": 0.9936327614670287,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9498234902785527,
+      "eval_accuracy_concrete tiles": 0.9823885525591635,
+      "eval_accuracy_dark bricks": 0.8843000654324313,
+      "eval_accuracy_grass": 0.9015822198611058,
+      "eval_accuracy_red bricks": 0.8831676136363636,
+      "eval_accuracy_road": 0.9820964537532029,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48835221790312355,
+      "eval_accuracy_speedway bricks": 0.9881092457764471,
+      "eval_accuracy_steel": 0.904302156849812,
+      "eval_accuracy_tiles": 0.07923572113409122,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808785359580406,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8890247515554462,
+      "eval_iou_concrete tiles": 0.9569905283850598,
+      "eval_iou_dark bricks": 0.8408051235132662,
+      "eval_iou_grass": 0.8366078783672464,
+      "eval_iou_red bricks": 0.8319116909182137,
+      "eval_iou_road": 0.974495972700996,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.320076613121804,
+      "eval_iou_speedway bricks": 0.9773861848497943,
+      "eval_iou_steel": 0.8497528549514232,
+      "eval_iou_tiles": 0.07921402163494454,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08973326534032822,
+      "eval_mean_accuracy": 0.695153115280871,
+      "eval_mean_iou": 0.6567033966120182,
+      "eval_overall_accuracy": 0.9786481857299805,
+      "eval_runtime": 4.0546,
+      "eval_samples_per_second": 8.879,
+      "eval_steps_per_second": 0.493,
+      "step": 4980
+    },
+    {
+      "epoch": 452.82,
+      "learning_rate": 9.933014354066985e-06,
+      "loss": 0.0275,
+      "step": 4981
+    },
+    {
+      "epoch": 452.91,
+      "learning_rate": 9.913875598086125e-06,
+      "loss": 0.0286,
+      "step": 4982
+    },
+    {
+      "epoch": 453.0,
+      "learning_rate": 9.894736842105264e-06,
+      "loss": 0.0262,
+      "step": 4983
+    },
+    {
+      "epoch": 453.09,
+      "learning_rate": 9.875598086124401e-06,
+      "loss": 0.0293,
+      "step": 4984
+    },
+    {
+      "epoch": 453.18,
+      "learning_rate": 9.856459330143542e-06,
+      "loss": 0.0311,
+      "step": 4985
+    },
+    {
+      "epoch": 453.27,
+      "learning_rate": 9.83732057416268e-06,
+      "loss": 0.0274,
+      "step": 4986
+    },
+    {
+      "epoch": 453.36,
+      "learning_rate": 9.818181818181818e-06,
+      "loss": 0.0275,
+      "step": 4987
+    },
+    {
+      "epoch": 453.45,
+      "learning_rate": 9.799043062200958e-06,
+      "loss": 0.0289,
+      "step": 4988
+    },
+    {
+      "epoch": 453.55,
+      "learning_rate": 9.779904306220097e-06,
+      "loss": 0.0326,
+      "step": 4989
+    },
+    {
+      "epoch": 453.64,
+      "learning_rate": 9.760765550239234e-06,
+      "loss": 0.0313,
+      "step": 4990
+    },
+    {
+      "epoch": 453.73,
+      "learning_rate": 9.741626794258374e-06,
+      "loss": 0.0289,
+      "step": 4991
+    },
+    {
+      "epoch": 453.82,
+      "learning_rate": 9.722488038277511e-06,
+      "loss": 0.0285,
+      "step": 4992
+    },
+    {
+      "epoch": 453.91,
+      "learning_rate": 9.703349282296652e-06,
+      "loss": 0.029,
+      "step": 4993
+    },
+    {
+      "epoch": 454.0,
+      "learning_rate": 9.68421052631579e-06,
+      "loss": 0.0276,
+      "step": 4994
+    },
+    {
+      "epoch": 454.09,
+      "learning_rate": 9.665071770334928e-06,
+      "loss": 0.0283,
+      "step": 4995
+    },
+    {
+      "epoch": 454.18,
+      "learning_rate": 9.645933014354068e-06,
+      "loss": 0.0299,
+      "step": 4996
+    },
+    {
+      "epoch": 454.27,
+      "learning_rate": 9.626794258373207e-06,
+      "loss": 0.0251,
+      "step": 4997
+    },
+    {
+      "epoch": 454.36,
+      "learning_rate": 9.607655502392344e-06,
+      "loss": 0.0285,
+      "step": 4998
+    },
+    {
+      "epoch": 454.45,
+      "learning_rate": 9.588516746411485e-06,
+      "loss": 0.0305,
+      "step": 4999
+    },
+    {
+      "epoch": 454.55,
+      "learning_rate": 9.569377990430622e-06,
+      "loss": 0.032,
+      "step": 5000
+    },
+    {
+      "epoch": 454.55,
+      "eval_accuracy_NAT": 0.9932851527678475,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9509850208163441,
+      "eval_accuracy_concrete tiles": 0.984742860637192,
+      "eval_accuracy_dark bricks": 0.8827219891459144,
+      "eval_accuracy_grass": 0.9047767172170825,
+      "eval_accuracy_red bricks": 0.8908025568181818,
+      "eval_accuracy_road": 0.9811206222105063,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4747884101122047,
+      "eval_accuracy_speedway bricks": 0.9883053533133725,
+      "eval_accuracy_steel": 0.9044423191082382,
+      "eval_accuracy_tiles": 0.07909875359539789,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810135086268568,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8868043230560069,
+      "eval_iou_concrete tiles": 0.9564352319296787,
+      "eval_iou_dark bricks": 0.8394890003294411,
+      "eval_iou_grass": 0.8371449267863611,
+      "eval_iou_red bricks": 0.8357487922705314,
+      "eval_iou_road": 0.974120274212719,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3112822181252937,
+      "eval_iou_speedway bricks": 0.9773777392418946,
+      "eval_iou_steel": 0.8489834456818692,
+      "eval_iou_tiles": 0.07907709160618924,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08983661979436874,
+      "eval_mean_accuracy": 0.6950053658263294,
+      "eval_mean_iou": 0.6559597347589878,
+      "eval_overall_accuracy": 0.9785348044501411,
+      "eval_runtime": 4.291,
+      "eval_samples_per_second": 8.39,
+      "eval_steps_per_second": 0.466,
+      "step": 5000
+    },
+    {
+      "epoch": 454.64,
+      "learning_rate": 9.55023923444976e-06,
+      "loss": 0.0298,
+      "step": 5001
+    },
+    {
+      "epoch": 454.73,
+      "learning_rate": 9.531100478468901e-06,
+      "loss": 0.0308,
+      "step": 5002
+    },
+    {
+      "epoch": 454.82,
+      "learning_rate": 9.511961722488038e-06,
+      "loss": 0.0268,
+      "step": 5003
+    },
+    {
+      "epoch": 454.91,
+      "learning_rate": 9.492822966507177e-06,
+      "loss": 0.0283,
+      "step": 5004
+    },
+    {
+      "epoch": 455.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0295,
+      "step": 5005
+    },
+    {
+      "epoch": 455.09,
+      "learning_rate": 9.454545454545454e-06,
+      "loss": 0.0292,
+      "step": 5006
+    },
+    {
+      "epoch": 455.18,
+      "learning_rate": 9.435406698564593e-06,
+      "loss": 0.028,
+      "step": 5007
+    },
+    {
+      "epoch": 455.27,
+      "learning_rate": 9.416267942583734e-06,
+      "loss": 0.0299,
+      "step": 5008
+    },
+    {
+      "epoch": 455.36,
+      "learning_rate": 9.39712918660287e-06,
+      "loss": 0.0276,
+      "step": 5009
+    },
+    {
+      "epoch": 455.45,
+      "learning_rate": 9.377990430622011e-06,
+      "loss": 0.0325,
+      "step": 5010
+    },
+    {
+      "epoch": 455.55,
+      "learning_rate": 9.358851674641148e-06,
+      "loss": 0.0285,
+      "step": 5011
+    },
+    {
+      "epoch": 455.64,
+      "learning_rate": 9.339712918660287e-06,
+      "loss": 0.0276,
+      "step": 5012
+    },
+    {
+      "epoch": 455.73,
+      "learning_rate": 9.320574162679427e-06,
+      "loss": 0.0287,
+      "step": 5013
+    },
+    {
+      "epoch": 455.82,
+      "learning_rate": 9.301435406698564e-06,
+      "loss": 0.0287,
+      "step": 5014
+    },
+    {
+      "epoch": 455.91,
+      "learning_rate": 9.282296650717703e-06,
+      "loss": 0.0313,
+      "step": 5015
+    },
+    {
+      "epoch": 456.0,
+      "learning_rate": 9.263157894736844e-06,
+      "loss": 0.0321,
+      "step": 5016
+    },
+    {
+      "epoch": 456.09,
+      "learning_rate": 9.24401913875598e-06,
+      "loss": 0.0301,
+      "step": 5017
+    },
+    {
+      "epoch": 456.18,
+      "learning_rate": 9.22488038277512e-06,
+      "loss": 0.0271,
+      "step": 5018
+    },
+    {
+      "epoch": 456.27,
+      "learning_rate": 9.20574162679426e-06,
+      "loss": 0.0333,
+      "step": 5019
+    },
+    {
+      "epoch": 456.36,
+      "learning_rate": 9.186602870813397e-06,
+      "loss": 0.0301,
+      "step": 5020
+    },
+    {
+      "epoch": 456.36,
+      "eval_accuracy_NAT": 0.9933079149618548,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9509783145199365,
+      "eval_accuracy_concrete tiles": 0.9835198434538005,
+      "eval_accuracy_dark bricks": 0.8851468380739771,
+      "eval_accuracy_grass": 0.9015927050821867,
+      "eval_accuracy_red bricks": 0.890625,
+      "eval_accuracy_road": 0.9814616191064672,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5001791966476442,
+      "eval_accuracy_speedway bricks": 0.9884568086726597,
+      "eval_accuracy_steel": 0.9036095903964119,
+      "eval_accuracy_tiles": 0.07451034104917134,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809330787150664,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8879619877291701,
+      "eval_iou_concrete tiles": 0.9562696949878114,
+      "eval_iou_dark bricks": 0.8397049695110819,
+      "eval_iou_grass": 0.8364277191253146,
+      "eval_iou_red bricks": 0.8344701380801863,
+      "eval_iou_road": 0.9738952642870513,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.326759599452489,
+      "eval_iou_speedway bricks": 0.977438279266034,
+      "eval_iou_steel": 0.8497735942685234,
+      "eval_iou_tiles": 0.07449503594659364,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.089717335999012,
+      "eval_mean_accuracy": 0.69641447476647,
+      "eval_mean_iou": 0.656779181643794,
+      "eval_overall_accuracy": 0.9785900115966797,
+      "eval_runtime": 4.1561,
+      "eval_samples_per_second": 8.662,
+      "eval_steps_per_second": 0.481,
+      "step": 5020
+    },
+    {
+      "epoch": 456.45,
+      "learning_rate": 9.167464114832536e-06,
+      "loss": 0.0311,
+      "step": 5021
+    },
+    {
+      "epoch": 456.55,
+      "learning_rate": 9.148325358851675e-06,
+      "loss": 0.0324,
+      "step": 5022
+    },
+    {
+      "epoch": 456.64,
+      "learning_rate": 9.129186602870813e-06,
+      "loss": 0.0306,
+      "step": 5023
+    },
+    {
+      "epoch": 456.73,
+      "learning_rate": 9.110047846889952e-06,
+      "loss": 0.0251,
+      "step": 5024
+    },
+    {
+      "epoch": 456.82,
+      "learning_rate": 9.090909090909091e-06,
+      "loss": 0.025,
+      "step": 5025
+    },
+    {
+      "epoch": 456.91,
+      "learning_rate": 9.07177033492823e-06,
+      "loss": 0.0325,
+      "step": 5026
+    },
+    {
+      "epoch": 457.0,
+      "learning_rate": 9.05263157894737e-06,
+      "loss": 0.0254,
+      "step": 5027
+    },
+    {
+      "epoch": 457.09,
+      "learning_rate": 9.033492822966507e-06,
+      "loss": 0.0338,
+      "step": 5028
+    },
+    {
+      "epoch": 457.18,
+      "learning_rate": 9.014354066985646e-06,
+      "loss": 0.0357,
+      "step": 5029
+    },
+    {
+      "epoch": 457.27,
+      "learning_rate": 8.995215311004785e-06,
+      "loss": 0.029,
+      "step": 5030
+    },
+    {
+      "epoch": 457.36,
+      "learning_rate": 8.976076555023924e-06,
+      "loss": 0.0306,
+      "step": 5031
+    },
+    {
+      "epoch": 457.45,
+      "learning_rate": 8.956937799043062e-06,
+      "loss": 0.0315,
+      "step": 5032
+    },
+    {
+      "epoch": 457.55,
+      "learning_rate": 8.937799043062201e-06,
+      "loss": 0.0272,
+      "step": 5033
+    },
+    {
+      "epoch": 457.64,
+      "learning_rate": 8.91866028708134e-06,
+      "loss": 0.0258,
+      "step": 5034
+    },
+    {
+      "epoch": 457.73,
+      "learning_rate": 8.899521531100479e-06,
+      "loss": 0.0259,
+      "step": 5035
+    },
+    {
+      "epoch": 457.82,
+      "learning_rate": 8.880382775119618e-06,
+      "loss": 0.028,
+      "step": 5036
+    },
+    {
+      "epoch": 457.91,
+      "learning_rate": 8.861244019138756e-06,
+      "loss": 0.0309,
+      "step": 5037
+    },
+    {
+      "epoch": 458.0,
+      "learning_rate": 8.842105263157895e-06,
+      "loss": 0.0318,
+      "step": 5038
+    },
+    {
+      "epoch": 458.09,
+      "learning_rate": 8.822966507177034e-06,
+      "loss": 0.0253,
+      "step": 5039
+    },
+    {
+      "epoch": 458.18,
+      "learning_rate": 8.803827751196173e-06,
+      "loss": 0.0302,
+      "step": 5040
+    },
+    {
+      "epoch": 458.18,
+      "eval_accuracy_NAT": 0.9935646513361232,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9531337181853299,
+      "eval_accuracy_concrete tiles": 0.9770684278114108,
+      "eval_accuracy_dark bricks": 0.880989954197298,
+      "eval_accuracy_grass": 0.898198988525673,
+      "eval_accuracy_red bricks": 0.8870738636363636,
+      "eval_accuracy_road": 0.9801097236111329,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49190858214098643,
+      "eval_accuracy_speedway bricks": 0.9877671135504877,
+      "eval_accuracy_steel": 0.9011773629707803,
+      "eval_accuracy_tiles": 0.07594850020545131,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808637519344412,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8871679952060224,
+      "eval_iou_concrete tiles": 0.9555077143882311,
+      "eval_iou_dark bricks": 0.8370451636496617,
+      "eval_iou_grass": 0.8362075697625989,
+      "eval_iou_red bricks": 0.8294869666279263,
+      "eval_iou_road": 0.9728818157048531,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3229151585348197,
+      "eval_iou_speedway bricks": 0.977418796274182,
+      "eval_iou_steel": 0.8497329570632274,
+      "eval_iou_tiles": 0.0759277009448172,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09094765782356262,
+      "eval_mean_accuracy": 0.6943800681670028,
+      "eval_mean_iou": 0.6557811992377522,
+      "eval_overall_accuracy": 0.9784752527872721,
+      "eval_runtime": 4.4118,
+      "eval_samples_per_second": 8.16,
+      "eval_steps_per_second": 0.453,
+      "step": 5040
+    },
+    {
+      "epoch": 458.27,
+      "learning_rate": 8.784688995215311e-06,
+      "loss": 0.0299,
+      "step": 5041
+    },
+    {
+      "epoch": 458.36,
+      "learning_rate": 8.76555023923445e-06,
+      "loss": 0.0278,
+      "step": 5042
+    },
+    {
+      "epoch": 458.45,
+      "learning_rate": 8.746411483253589e-06,
+      "loss": 0.029,
+      "step": 5043
+    },
+    {
+      "epoch": 458.55,
+      "learning_rate": 8.727272727272728e-06,
+      "loss": 0.0317,
+      "step": 5044
+    },
+    {
+      "epoch": 458.64,
+      "learning_rate": 8.708133971291867e-06,
+      "loss": 0.03,
+      "step": 5045
+    },
+    {
+      "epoch": 458.73,
+      "learning_rate": 8.688995215311005e-06,
+      "loss": 0.0258,
+      "step": 5046
+    },
+    {
+      "epoch": 458.82,
+      "learning_rate": 8.669856459330144e-06,
+      "loss": 0.0279,
+      "step": 5047
+    },
+    {
+      "epoch": 458.91,
+      "learning_rate": 8.650717703349283e-06,
+      "loss": 0.0279,
+      "step": 5048
+    },
+    {
+      "epoch": 459.0,
+      "learning_rate": 8.631578947368422e-06,
+      "loss": 0.0341,
+      "step": 5049
+    },
+    {
+      "epoch": 459.09,
+      "learning_rate": 8.61244019138756e-06,
+      "loss": 0.0349,
+      "step": 5050
+    },
+    {
+      "epoch": 459.18,
+      "learning_rate": 8.5933014354067e-06,
+      "loss": 0.0273,
+      "step": 5051
+    },
+    {
+      "epoch": 459.27,
+      "learning_rate": 8.574162679425838e-06,
+      "loss": 0.0288,
+      "step": 5052
+    },
+    {
+      "epoch": 459.36,
+      "learning_rate": 8.555023923444977e-06,
+      "loss": 0.0308,
+      "step": 5053
+    },
+    {
+      "epoch": 459.45,
+      "learning_rate": 8.535885167464116e-06,
+      "loss": 0.0284,
+      "step": 5054
+    },
+    {
+      "epoch": 459.55,
+      "learning_rate": 8.516746411483253e-06,
+      "loss": 0.029,
+      "step": 5055
+    },
+    {
+      "epoch": 459.64,
+      "learning_rate": 8.497607655502393e-06,
+      "loss": 0.0273,
+      "step": 5056
+    },
+    {
+      "epoch": 459.73,
+      "learning_rate": 8.478468899521532e-06,
+      "loss": 0.0339,
+      "step": 5057
+    },
+    {
+      "epoch": 459.82,
+      "learning_rate": 8.459330143540669e-06,
+      "loss": 0.0298,
+      "step": 5058
+    },
+    {
+      "epoch": 459.91,
+      "learning_rate": 8.44019138755981e-06,
+      "loss": 0.0271,
+      "step": 5059
+    },
+    {
+      "epoch": 460.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0268,
+      "step": 5060
+    },
+    {
+      "epoch": 460.0,
+      "eval_accuracy_NAT": 0.9936998128912362,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9521774003176102,
+      "eval_accuracy_concrete tiles": 0.9728490185287103,
+      "eval_accuracy_dark bricks": 0.885377776067126,
+      "eval_accuracy_grass": 0.8971364861228099,
+      "eval_accuracy_red bricks": 0.8828125,
+      "eval_accuracy_road": 0.980980353983799,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48669809500179195,
+      "eval_accuracy_speedway bricks": 0.9873730882532189,
+      "eval_accuracy_steel": 0.9075506233098081,
+      "eval_accuracy_tiles": 0.07197644158334475,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.98086254655607,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.887925114975379,
+      "eval_iou_concrete tiles": 0.9538341627195875,
+      "eval_iou_dark bricks": 0.838301749271137,
+      "eval_iou_grass": 0.8361597748402186,
+      "eval_iou_red bricks": 0.8281145902731513,
+      "eval_iou_road": 0.9733908463167493,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3195062800883194,
+      "eval_iou_speedway bricks": 0.9773796304979276,
+      "eval_iou_steel": 0.8506437303905658,
+      "eval_iou_tiles": 0.0719616569667922,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09019830822944641,
+      "eval_mean_accuracy": 0.6937408920045735,
+      "eval_mean_iou": 0.6552369294535306,
+      "eval_overall_accuracy": 0.9785064061482748,
+      "eval_runtime": 3.7843,
+      "eval_samples_per_second": 9.513,
+      "eval_steps_per_second": 0.529,
+      "step": 5060
+    },
+    {
+      "epoch": 460.09,
+      "learning_rate": 8.401913875598085e-06,
+      "loss": 0.0276,
+      "step": 5061
+    },
+    {
+      "epoch": 460.18,
+      "learning_rate": 8.382775119617226e-06,
+      "loss": 0.0291,
+      "step": 5062
+    },
+    {
+      "epoch": 460.27,
+      "learning_rate": 8.363636363636365e-06,
+      "loss": 0.0306,
+      "step": 5063
+    },
+    {
+      "epoch": 460.36,
+      "learning_rate": 8.344497607655503e-06,
+      "loss": 0.0271,
+      "step": 5064
+    },
+    {
+      "epoch": 460.45,
+      "learning_rate": 8.325358851674642e-06,
+      "loss": 0.0283,
+      "step": 5065
+    },
+    {
+      "epoch": 460.55,
+      "learning_rate": 8.30622009569378e-06,
+      "loss": 0.0296,
+      "step": 5066
+    },
+    {
+      "epoch": 460.64,
+      "learning_rate": 8.28708133971292e-06,
+      "loss": 0.0243,
+      "step": 5067
+    },
+    {
+      "epoch": 460.73,
+      "learning_rate": 8.267942583732058e-06,
+      "loss": 0.0312,
+      "step": 5068
+    },
+    {
+      "epoch": 460.82,
+      "learning_rate": 8.248803827751196e-06,
+      "loss": 0.0282,
+      "step": 5069
+    },
+    {
+      "epoch": 460.91,
+      "learning_rate": 8.229665071770336e-06,
+      "loss": 0.0279,
+      "step": 5070
+    },
+    {
+      "epoch": 461.0,
+      "learning_rate": 8.210526315789475e-06,
+      "loss": 0.0263,
+      "step": 5071
+    },
+    {
+      "epoch": 461.09,
+      "learning_rate": 8.191387559808612e-06,
+      "loss": 0.0295,
+      "step": 5072
+    },
+    {
+      "epoch": 461.18,
+      "learning_rate": 8.172248803827752e-06,
+      "loss": 0.0294,
+      "step": 5073
+    },
+    {
+      "epoch": 461.27,
+      "learning_rate": 8.15311004784689e-06,
+      "loss": 0.0287,
+      "step": 5074
+    },
+    {
+      "epoch": 461.36,
+      "learning_rate": 8.133971291866028e-06,
+      "loss": 0.0306,
+      "step": 5075
+    },
+    {
+      "epoch": 461.45,
+      "learning_rate": 8.114832535885169e-06,
+      "loss": 0.0286,
+      "step": 5076
+    },
+    {
+      "epoch": 461.55,
+      "learning_rate": 8.095693779904306e-06,
+      "loss": 0.0349,
+      "step": 5077
+    },
+    {
+      "epoch": 461.64,
+      "learning_rate": 8.076555023923445e-06,
+      "loss": 0.0303,
+      "step": 5078
+    },
+    {
+      "epoch": 461.73,
+      "learning_rate": 8.057416267942585e-06,
+      "loss": 0.0289,
+      "step": 5079
+    },
+    {
+      "epoch": 461.82,
+      "learning_rate": 8.038277511961722e-06,
+      "loss": 0.0261,
+      "step": 5080
+    },
+    {
+      "epoch": 461.82,
+      "eval_accuracy_NAT": 0.9938192702969955,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9506604360702176,
+      "eval_accuracy_concrete tiles": 0.9776799364031065,
+      "eval_accuracy_dark bricks": 0.8868403833570686,
+      "eval_accuracy_grass": 0.8961438851938193,
+      "eval_accuracy_red bricks": 0.8872514204545454,
+      "eval_accuracy_road": 0.9812342878424933,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48190113858793043,
+      "eval_accuracy_speedway bricks": 0.9880489049958546,
+      "eval_accuracy_steel": 0.9024140887804235,
+      "eval_accuracy_tiles": 0.0775236269004246,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808489169894504,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8878675480428964,
+      "eval_iou_concrete tiles": 0.9558485038711028,
+      "eval_iou_dark bricks": 0.8393195395599592,
+      "eval_iou_grass": 0.8360216894418849,
+      "eval_iou_red bricks": 0.8308945793149318,
+      "eval_iou_road": 0.9737047356434384,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3158256093374528,
+      "eval_iou_speedway bricks": 0.9773463324026115,
+      "eval_iou_steel": 0.8503437827759003,
+      "eval_iou_tiles": 0.07748117727583846,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09070097655057907,
+      "eval_mean_accuracy": 0.6941167214525291,
+      "eval_mean_iou": 0.6558078780504205,
+      "eval_overall_accuracy": 0.9785166846381294,
+      "eval_runtime": 3.8092,
+      "eval_samples_per_second": 9.451,
+      "eval_steps_per_second": 0.525,
+      "step": 5080
+    },
+    {
+      "epoch": 461.91,
+      "learning_rate": 8.019138755980863e-06,
+      "loss": 0.0308,
+      "step": 5081
+    },
+    {
+      "epoch": 462.0,
+      "learning_rate": 8.000000000000001e-06,
+      "loss": 0.0244,
+      "step": 5082
+    },
+    {
+      "epoch": 462.09,
+      "learning_rate": 7.980861244019138e-06,
+      "loss": 0.0283,
+      "step": 5083
+    },
+    {
+      "epoch": 462.18,
+      "learning_rate": 7.961722488038279e-06,
+      "loss": 0.0265,
+      "step": 5084
+    },
+    {
+      "epoch": 462.27,
+      "learning_rate": 7.942583732057416e-06,
+      "loss": 0.027,
+      "step": 5085
+    },
+    {
+      "epoch": 462.36,
+      "learning_rate": 7.923444976076555e-06,
+      "loss": 0.03,
+      "step": 5086
+    },
+    {
+      "epoch": 462.45,
+      "learning_rate": 7.904306220095695e-06,
+      "loss": 0.0287,
+      "step": 5087
+    },
+    {
+      "epoch": 462.55,
+      "learning_rate": 7.885167464114832e-06,
+      "loss": 0.0277,
+      "step": 5088
+    },
+    {
+      "epoch": 462.64,
+      "learning_rate": 7.866028708133971e-06,
+      "loss": 0.028,
+      "step": 5089
+    },
+    {
+      "epoch": 462.73,
+      "learning_rate": 7.846889952153112e-06,
+      "loss": 0.0329,
+      "step": 5090
+    },
+    {
+      "epoch": 462.82,
+      "learning_rate": 7.827751196172249e-06,
+      "loss": 0.0282,
+      "step": 5091
+    },
+    {
+      "epoch": 462.91,
+      "learning_rate": 7.808612440191387e-06,
+      "loss": 0.0241,
+      "step": 5092
+    },
+    {
+      "epoch": 463.0,
+      "learning_rate": 7.789473684210528e-06,
+      "loss": 0.0266,
+      "step": 5093
+    },
+    {
+      "epoch": 463.09,
+      "learning_rate": 7.770334928229665e-06,
+      "loss": 0.045,
+      "step": 5094
+    },
+    {
+      "epoch": 463.18,
+      "learning_rate": 7.751196172248804e-06,
+      "loss": 0.0274,
+      "step": 5095
+    },
+    {
+      "epoch": 463.27,
+      "learning_rate": 7.732057416267943e-06,
+      "loss": 0.0268,
+      "step": 5096
+    },
+    {
+      "epoch": 463.36,
+      "learning_rate": 7.712918660287081e-06,
+      "loss": 0.0259,
+      "step": 5097
+    },
+    {
+      "epoch": 463.45,
+      "learning_rate": 7.693779904306222e-06,
+      "loss": 0.0293,
+      "step": 5098
+    },
+    {
+      "epoch": 463.55,
+      "learning_rate": 7.674641148325359e-06,
+      "loss": 0.0281,
+      "step": 5099
+    },
+    {
+      "epoch": 463.64,
+      "learning_rate": 7.655502392344498e-06,
+      "loss": 0.0347,
+      "step": 5100
+    },
+    {
+      "epoch": 463.64,
+      "eval_accuracy_NAT": 0.9935591813515168,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9510494012618568,
+      "eval_accuracy_concrete tiles": 0.9770072769522412,
+      "eval_accuracy_dark bricks": 0.8839921481082329,
+      "eval_accuracy_grass": 0.9008167987222011,
+      "eval_accuracy_red bricks": 0.884765625,
+      "eval_accuracy_road": 0.9812246141716859,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.481846001157886,
+      "eval_accuracy_speedway bricks": 0.9879318438815052,
+      "eval_accuracy_steel": 0.9042361981399644,
+      "eval_accuracy_tiles": 0.07670182166826463,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809032904078504,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.887834891999404,
+      "eval_iou_concrete tiles": 0.955733684273494,
+      "eval_iou_dark bricks": 0.8397133560016087,
+      "eval_iou_grass": 0.8366981233139529,
+      "eval_iou_red bricks": 0.8320253798630823,
+      "eval_iou_road": 0.9735129052304754,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3164527167713784,
+      "eval_iou_speedway bricks": 0.977378131736341,
+      "eval_iou_steel": 0.8497303746862119,
+      "eval_iou_tiles": 0.07667556650920791,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09045623987913132,
+      "eval_mean_accuracy": 0.6940869931088736,
+      "eval_mean_iou": 0.655896801599462,
+      "eval_overall_accuracy": 0.9785217709011502,
+      "eval_runtime": 3.9356,
+      "eval_samples_per_second": 9.147,
+      "eval_steps_per_second": 0.508,
+      "step": 5100
+    },
+    {
+      "epoch": 463.73,
+      "learning_rate": 7.636363636363638e-06,
+      "loss": 0.025,
+      "step": 5101
+    },
+    {
+      "epoch": 463.82,
+      "learning_rate": 7.617224880382775e-06,
+      "loss": 0.0295,
+      "step": 5102
+    },
+    {
+      "epoch": 463.91,
+      "learning_rate": 7.598086124401914e-06,
+      "loss": 0.0307,
+      "step": 5103
+    },
+    {
+      "epoch": 464.0,
+      "learning_rate": 7.578947368421054e-06,
+      "loss": 0.0285,
+      "step": 5104
+    },
+    {
+      "epoch": 464.09,
+      "learning_rate": 7.5598086124401915e-06,
+      "loss": 0.029,
+      "step": 5105
+    },
+    {
+      "epoch": 464.18,
+      "learning_rate": 7.54066985645933e-06,
+      "loss": 0.033,
+      "step": 5106
+    },
+    {
+      "epoch": 464.27,
+      "learning_rate": 7.52153110047847e-06,
+      "loss": 0.0297,
+      "step": 5107
+    },
+    {
+      "epoch": 464.36,
+      "learning_rate": 7.502392344497608e-06,
+      "loss": 0.0285,
+      "step": 5108
+    },
+    {
+      "epoch": 464.45,
+      "learning_rate": 7.483253588516747e-06,
+      "loss": 0.0288,
+      "step": 5109
+    },
+    {
+      "epoch": 464.55,
+      "learning_rate": 7.464114832535886e-06,
+      "loss": 0.0238,
+      "step": 5110
+    },
+    {
+      "epoch": 464.64,
+      "learning_rate": 7.444976076555024e-06,
+      "loss": 0.0299,
+      "step": 5111
+    },
+    {
+      "epoch": 464.73,
+      "learning_rate": 7.425837320574162e-06,
+      "loss": 0.0274,
+      "step": 5112
+    },
+    {
+      "epoch": 464.82,
+      "learning_rate": 7.406698564593302e-06,
+      "loss": 0.0327,
+      "step": 5113
+    },
+    {
+      "epoch": 464.91,
+      "learning_rate": 7.3875598086124405e-06,
+      "loss": 0.0298,
+      "step": 5114
+    },
+    {
+      "epoch": 465.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0313,
+      "step": 5115
+    },
+    {
+      "epoch": 465.09,
+      "learning_rate": 7.349282296650718e-06,
+      "loss": 0.0339,
+      "step": 5116
+    },
+    {
+      "epoch": 465.18,
+      "learning_rate": 7.330143540669857e-06,
+      "loss": 0.0277,
+      "step": 5117
+    },
+    {
+      "epoch": 465.27,
+      "learning_rate": 7.3110047846889965e-06,
+      "loss": 0.0291,
+      "step": 5118
+    },
+    {
+      "epoch": 465.36,
+      "learning_rate": 7.291866028708134e-06,
+      "loss": 0.0288,
+      "step": 5119
+    },
+    {
+      "epoch": 465.45,
+      "learning_rate": 7.272727272727272e-06,
+      "loss": 0.0281,
+      "step": 5120
+    },
+    {
+      "epoch": 465.45,
+      "eval_accuracy_NAT": 0.9936375256471698,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9514558028241555,
+      "eval_accuracy_concrete tiles": 0.9811961108053568,
+      "eval_accuracy_dark bricks": 0.886186059043147,
+      "eval_accuracy_grass": 0.8990482914332248,
+      "eval_accuracy_red bricks": 0.8831676136363636,
+      "eval_accuracy_road": 0.9810396052174943,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46977090397816557,
+      "eval_accuracy_speedway bricks": 0.9882106182878424,
+      "eval_accuracy_steel": 0.900690917485654,
+      "eval_accuracy_tiles": 0.07608546774414464,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808951990737362,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8878840978784655,
+      "eval_iou_concrete tiles": 0.9568549108473969,
+      "eval_iou_dark bricks": 0.8395259799453054,
+      "eval_iou_grass": 0.8367232759433886,
+      "eval_iou_red bricks": 0.8309388573337788,
+      "eval_iou_road": 0.9734017536197803,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3101565344011649,
+      "eval_iou_speedway bricks": 0.9773546216361056,
+      "eval_iou_steel": 0.8489376913632054,
+      "eval_iou_tiles": 0.07606463097357251,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09027755260467529,
+      "eval_mean_accuracy": 0.6931145320079014,
+      "eval_mean_iou": 0.6552875040781463,
+      "eval_overall_accuracy": 0.9785067240397135,
+      "eval_runtime": 4.1623,
+      "eval_samples_per_second": 8.649,
+      "eval_steps_per_second": 0.481,
+      "step": 5120
     }
   ],
   "max_steps": 5500,
   "num_train_epochs": 500,
-  "total_flos": 2.6868806431235113e+18,
+  "total_flos": 2.866007188832256e+18,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:26303ba4f9faefe661e6db82bb30ef1e40b93467ffeae6fc6e925fe7e5142b21
+oid sha256:caa6c48f1f3d0e2e4eed5d4b0d97bca91e05d2f0e089c52c27a33f0b646addba
 size 14944077

runs/May20_22-12-23_robolidar/events.out.tfevents.1716261149.robolidar.3147080.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:687c65142452e38f9586e3ed4d8f15bd7a00b6650c970e12ba54750bce58e6aa
-size 1326114
+oid sha256:d40dac99543fedfde1fb7332988e8068c18dde6b6894f19969fd15d4526522c0
+size 1347090