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:f169104f1e007c67ca84947bb692b27daeb9004635e327cea5e497631e99aca4
+oid sha256:9c9600ee8dc5c509c14cc7c0d3b7ceda2b2c10b9df35ffe22cacabcd893f4086
 size 29861573

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1dc6a11ed4dfb0ac041d51a80e3eabfc2c24ee6dd6fa6d4ee16341007b0c73a9
+oid sha256:82670e9b397007cd8e7075fc9fea31bd51f2f76e57dec7d59e6173c06fea6d8e
 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.08471038937568665,
   "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0/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": 10.152,
       "eval_steps_per_second": 0.564,
       "step": 4800
+    },
+    {
+      "epoch": 436.45,
+      "learning_rate": 1.3377990430622008e-05,
+      "loss": 0.0274,
+      "step": 4801
+    },
+    {
+      "epoch": 436.55,
+      "learning_rate": 1.3358851674641149e-05,
+      "loss": 0.0301,
+      "step": 4802
+    },
+    {
+      "epoch": 436.64,
+      "learning_rate": 1.3339712918660288e-05,
+      "loss": 0.0281,
+      "step": 4803
+    },
+    {
+      "epoch": 436.73,
+      "learning_rate": 1.3320574162679428e-05,
+      "loss": 0.0327,
+      "step": 4804
+    },
+    {
+      "epoch": 436.82,
+      "learning_rate": 1.3301435406698565e-05,
+      "loss": 0.0306,
+      "step": 4805
+    },
+    {
+      "epoch": 436.91,
+      "learning_rate": 1.3282296650717704e-05,
+      "loss": 0.0324,
+      "step": 4806
+    },
+    {
+      "epoch": 437.0,
+      "learning_rate": 1.3263157894736844e-05,
+      "loss": 0.0304,
+      "step": 4807
+    },
+    {
+      "epoch": 437.09,
+      "learning_rate": 1.3244019138755981e-05,
+      "loss": 0.0284,
+      "step": 4808
+    },
+    {
+      "epoch": 437.18,
+      "learning_rate": 1.322488038277512e-05,
+      "loss": 0.0291,
+      "step": 4809
+    },
+    {
+      "epoch": 437.27,
+      "learning_rate": 1.320574162679426e-05,
+      "loss": 0.0297,
+      "step": 4810
+    },
+    {
+      "epoch": 437.36,
+      "learning_rate": 1.3186602870813398e-05,
+      "loss": 0.0299,
+      "step": 4811
+    },
+    {
+      "epoch": 437.45,
+      "learning_rate": 1.3167464114832537e-05,
+      "loss": 0.0273,
+      "step": 4812
+    },
+    {
+      "epoch": 437.55,
+      "learning_rate": 1.3148325358851677e-05,
+      "loss": 0.0289,
+      "step": 4813
+    },
+    {
+      "epoch": 437.64,
+      "learning_rate": 1.3129186602870812e-05,
+      "loss": 0.0299,
+      "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.0309,
+      "step": 4816
+    },
+    {
+      "epoch": 437.91,
+      "learning_rate": 1.3071770334928229e-05,
+      "loss": 0.0321,
+      "step": 4817
+    },
+    {
+      "epoch": 438.0,
+      "learning_rate": 1.305263157894737e-05,
+      "loss": 0.0274,
+      "step": 4818
+    },
+    {
+      "epoch": 438.09,
+      "learning_rate": 1.3033492822966508e-05,
+      "loss": 0.0291,
+      "step": 4819
+    },
+    {
+      "epoch": 438.18,
+      "learning_rate": 1.3014354066985645e-05,
+      "loss": 0.03,
+      "step": 4820
+    },
+    {
+      "epoch": 438.18,
+      "eval_accuracy_NAT": 0.9933574977255452,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9493486844928967,
+      "eval_accuracy_concrete tiles": 0.9876781018773314,
+      "eval_accuracy_dark bricks": 0.8782571879450367,
+      "eval_accuracy_grass": 0.8991251830544847,
+      "eval_accuracy_red bricks": 0.9284446022727273,
+      "eval_accuracy_road": 0.922324051043124,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46607669616519176,
+      "eval_accuracy_speedway bricks": 0.9886257628583186,
+      "eval_accuracy_steel": 0.9095211397665062,
+      "eval_accuracy_tiles": 0.2963292699630188,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9805195302825362,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8899825349144538,
+      "eval_iou_concrete tiles": 0.9499206022466623,
+      "eval_iou_dark bricks": 0.8316810030616708,
+      "eval_iou_grass": 0.8352326593161128,
+      "eval_iou_red bricks": 0.8638691557905172,
+      "eval_iou_road": 0.9148338006951657,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31525752433521054,
+      "eval_iou_speedway bricks": 0.97729628874072,
+      "eval_iou_steel": 0.8528202115158637,
+      "eval_iou_tiles": 0.29511662801800576,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1112009584903717,
+      "eval_mean_accuracy": 0.7091606290126293,
+      "eval_mean_iou": 0.6697330722243784,
+      "eval_overall_accuracy": 0.9735697640313042,
+      "eval_runtime": 4.031,
+      "eval_samples_per_second": 8.931,
+      "eval_steps_per_second": 0.496,
+      "step": 4820
+    },
+    {
+      "epoch": 438.27,
+      "learning_rate": 1.2995215311004786e-05,
+      "loss": 0.029,
+      "step": 4821
+    },
+    {
+      "epoch": 438.36,
+      "learning_rate": 1.2976076555023924e-05,
+      "loss": 0.0293,
+      "step": 4822
+    },
+    {
+      "epoch": 438.45,
+      "learning_rate": 1.2956937799043061e-05,
+      "loss": 0.0261,
+      "step": 4823
+    },
+    {
+      "epoch": 438.55,
+      "learning_rate": 1.2937799043062202e-05,
+      "loss": 0.0297,
+      "step": 4824
+    },
+    {
+      "epoch": 438.64,
+      "learning_rate": 1.291866028708134e-05,
+      "loss": 0.0299,
+      "step": 4825
+    },
+    {
+      "epoch": 438.73,
+      "learning_rate": 1.2899521531100478e-05,
+      "loss": 0.0286,
+      "step": 4826
+    },
+    {
+      "epoch": 438.82,
+      "learning_rate": 1.2880382775119618e-05,
+      "loss": 0.0288,
+      "step": 4827
+    },
+    {
+      "epoch": 438.91,
+      "learning_rate": 1.2861244019138757e-05,
+      "loss": 0.0316,
+      "step": 4828
+    },
+    {
+      "epoch": 439.0,
+      "learning_rate": 1.2842105263157894e-05,
+      "loss": 0.0362,
+      "step": 4829
+    },
+    {
+      "epoch": 439.09,
+      "learning_rate": 1.2822966507177035e-05,
+      "loss": 0.0294,
+      "step": 4830
+    },
+    {
+      "epoch": 439.18,
+      "learning_rate": 1.2803827751196173e-05,
+      "loss": 0.0314,
+      "step": 4831
+    },
+    {
+      "epoch": 439.27,
+      "learning_rate": 1.278468899521531e-05,
+      "loss": 0.0303,
+      "step": 4832
+    },
+    {
+      "epoch": 439.36,
+      "learning_rate": 1.276555023923445e-05,
+      "loss": 0.0292,
+      "step": 4833
+    },
+    {
+      "epoch": 439.45,
+      "learning_rate": 1.274641148325359e-05,
+      "loss": 0.0262,
+      "step": 4834
+    },
+    {
+      "epoch": 439.55,
+      "learning_rate": 1.2727272727272727e-05,
+      "loss": 0.0281,
+      "step": 4835
+    },
+    {
+      "epoch": 439.64,
+      "learning_rate": 1.2708133971291866e-05,
+      "loss": 0.0307,
+      "step": 4836
+    },
+    {
+      "epoch": 439.73,
+      "learning_rate": 1.2688995215311006e-05,
+      "loss": 0.0299,
+      "step": 4837
+    },
+    {
+      "epoch": 439.82,
+      "learning_rate": 1.2669856459330145e-05,
+      "loss": 0.0331,
+      "step": 4838
+    },
+    {
+      "epoch": 439.91,
+      "learning_rate": 1.2650717703349282e-05,
+      "loss": 0.0316,
+      "step": 4839
+    },
+    {
+      "epoch": 440.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0296,
+      "step": 4840
+    },
+    {
+      "epoch": 440.0,
+      "eval_accuracy_NAT": 0.9930215348000421,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9513149705995966,
+      "eval_accuracy_concrete tiles": 0.987739252736501,
+      "eval_accuracy_dark bricks": 0.8774873946345406,
+      "eval_accuracy_grass": 0.8968673654484005,
+      "eval_accuracy_red bricks": 0.9185014204545454,
+      "eval_accuracy_road": 0.9200700857449996,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48454773523006095,
+      "eval_accuracy_speedway bricks": 0.9887373933024147,
+      "eval_accuracy_steel": 0.9088862871842227,
+      "eval_accuracy_tiles": 0.2524996575811533,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9805000848481428,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8887237697615643,
+      "eval_iou_concrete tiles": 0.9526126444916254,
+      "eval_iou_dark bricks": 0.8304374749571996,
+      "eval_iou_grass": 0.8308262643268794,
+      "eval_iou_red bricks": 0.8601596275357499,
+      "eval_iou_road": 0.912913482188921,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3264669279491799,
+      "eval_iou_speedway bricks": 0.9773943967355547,
+      "eval_iou_steel": 0.854378187342086,
+      "eval_iou_tiles": 0.2515178388703186,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11227044463157654,
+      "eval_mean_accuracy": 0.7061286998243445,
+      "eval_mean_iou": 0.6666100537697862,
+      "eval_overall_accuracy": 0.9732641643948026,
+      "eval_runtime": 3.3843,
+      "eval_samples_per_second": 10.637,
+      "eval_steps_per_second": 0.591,
+      "step": 4840
+    },
+    {
+      "epoch": 440.09,
+      "learning_rate": 1.2612440191387561e-05,
+      "loss": 0.0323,
+      "step": 4841
+    },
+    {
+      "epoch": 440.18,
+      "learning_rate": 1.2593301435406698e-05,
+      "loss": 0.0294,
+      "step": 4842
+    },
+    {
+      "epoch": 440.27,
+      "learning_rate": 1.2574162679425839e-05,
+      "loss": 0.0289,
+      "step": 4843
+    },
+    {
+      "epoch": 440.36,
+      "learning_rate": 1.2555023923444977e-05,
+      "loss": 0.0323,
+      "step": 4844
+    },
+    {
+      "epoch": 440.45,
+      "learning_rate": 1.2535885167464115e-05,
+      "loss": 0.0318,
+      "step": 4845
+    },
+    {
+      "epoch": 440.55,
+      "learning_rate": 1.2516746411483255e-05,
+      "loss": 0.0286,
+      "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.0284,
+      "step": 4848
+    },
+    {
+      "epoch": 440.82,
+      "learning_rate": 1.2459330143540671e-05,
+      "loss": 0.0304,
+      "step": 4849
+    },
+    {
+      "epoch": 440.91,
+      "learning_rate": 1.2440191387559808e-05,
+      "loss": 0.0322,
+      "step": 4850
+    },
+    {
+      "epoch": 441.0,
+      "learning_rate": 1.2421052631578949e-05,
+      "loss": 0.0264,
+      "step": 4851
+    },
+    {
+      "epoch": 441.09,
+      "learning_rate": 1.2401913875598086e-05,
+      "loss": 0.028,
+      "step": 4852
+    },
+    {
+      "epoch": 441.18,
+      "learning_rate": 1.2382775119617225e-05,
+      "loss": 0.0312,
+      "step": 4853
+    },
+    {
+      "epoch": 441.27,
+      "learning_rate": 1.2363636363636365e-05,
+      "loss": 0.0289,
+      "step": 4854
+    },
+    {
+      "epoch": 441.36,
+      "learning_rate": 1.2344497607655502e-05,
+      "loss": 0.027,
+      "step": 4855
+    },
+    {
+      "epoch": 441.45,
+      "learning_rate": 1.2325358851674641e-05,
+      "loss": 0.031,
+      "step": 4856
+    },
+    {
+      "epoch": 441.55,
+      "learning_rate": 1.2306220095693782e-05,
+      "loss": 0.0306,
+      "step": 4857
+    },
+    {
+      "epoch": 441.64,
+      "learning_rate": 1.2287081339712919e-05,
+      "loss": 0.026,
+      "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.0304,
+      "step": 4860
+    },
+    {
+      "epoch": 441.82,
+      "eval_accuracy_NAT": 0.9928404959546817,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9523920018026525,
+      "eval_accuracy_concrete tiles": 0.9884730630465358,
+      "eval_accuracy_dark bricks": 0.876024787344598,
+      "eval_accuracy_grass": 0.9014703775029097,
+      "eval_accuracy_red bricks": 0.9307528409090909,
+      "eval_accuracy_road": 0.9244462125764976,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5139635541587406,
+      "eval_accuracy_speedway bricks": 0.9875854878009044,
+      "eval_accuracy_steel": 0.910090033638942,
+      "eval_accuracy_tiles": 0.2457882481851801,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808025310056913,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8888044680989551,
+      "eval_iou_concrete tiles": 0.9510487453299209,
+      "eval_iou_dark bricks": 0.8286910613508102,
+      "eval_iou_grass": 0.8325081176690832,
+      "eval_iou_red bricks": 0.8606140206862584,
+      "eval_iou_road": 0.9161289162906908,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3432385160637025,
+      "eval_iou_speedway bricks": 0.9775052199548002,
+      "eval_iou_steel": 0.8535120004948658,
+      "eval_iou_tiles": 0.24506657562307954,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1113537847995758,
+      "eval_mean_accuracy": 0.7095251617631333,
+      "eval_mean_iou": 0.6675323209667583,
+      "eval_overall_accuracy": 0.9736852645874023,
+      "eval_runtime": 3.4315,
+      "eval_samples_per_second": 10.491,
+      "eval_steps_per_second": 0.583,
+      "step": 4860
+    },
+    {
+      "epoch": 441.91,
+      "learning_rate": 1.2229665071770335e-05,
+      "loss": 0.0297,
+      "step": 4861
+    },
+    {
+      "epoch": 442.0,
+      "learning_rate": 1.2210526315789474e-05,
+      "loss": 0.0346,
+      "step": 4862
+    },
+    {
+      "epoch": 442.09,
+      "learning_rate": 1.2191387559808613e-05,
+      "loss": 0.0281,
+      "step": 4863
+    },
+    {
+      "epoch": 442.18,
+      "learning_rate": 1.2172248803827751e-05,
+      "loss": 0.0283,
+      "step": 4864
+    },
+    {
+      "epoch": 442.27,
+      "learning_rate": 1.2153110047846892e-05,
+      "loss": 0.0304,
+      "step": 4865
+    },
+    {
+      "epoch": 442.36,
+      "learning_rate": 1.2133971291866029e-05,
+      "loss": 0.0319,
+      "step": 4866
+    },
+    {
+      "epoch": 442.45,
+      "learning_rate": 1.2114832535885168e-05,
+      "loss": 0.0297,
+      "step": 4867
+    },
+    {
+      "epoch": 442.55,
+      "learning_rate": 1.2095693779904308e-05,
+      "loss": 0.0311,
+      "step": 4868
+    },
+    {
+      "epoch": 442.64,
+      "learning_rate": 1.2076555023923445e-05,
+      "loss": 0.0282,
+      "step": 4869
+    },
+    {
+      "epoch": 442.73,
+      "learning_rate": 1.2057416267942584e-05,
+      "loss": 0.0327,
+      "step": 4870
+    },
+    {
+      "epoch": 442.82,
+      "learning_rate": 1.2038277511961724e-05,
+      "loss": 0.0289,
+      "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.0253,
+      "step": 4873
+    },
+    {
+      "epoch": 443.09,
+      "learning_rate": 1.1980861244019139e-05,
+      "loss": 0.0307,
+      "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.0322,
+      "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.0311,
+      "step": 4878
+    },
+    {
+      "epoch": 443.55,
+      "learning_rate": 1.1885167464114833e-05,
+      "loss": 0.0303,
+      "step": 4879
+    },
+    {
+      "epoch": 443.64,
+      "learning_rate": 1.1866028708133972e-05,
+      "loss": 0.0346,
+      "step": 4880
+    },
+    {
+      "epoch": 443.64,
+      "eval_accuracy_NAT": 0.992863258148689,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9517522211253702,
+      "eval_accuracy_concrete tiles": 0.9892680242157402,
+      "eval_accuracy_dark bricks": 0.8779492706208383,
+      "eval_accuracy_grass": 0.9035359660558443,
+      "eval_accuracy_red bricks": 0.9327059659090909,
+      "eval_accuracy_road": 0.9250508170019601,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49642985140462603,
+      "eval_accuracy_speedway bricks": 0.9887500648663391,
+      "eval_accuracy_steel": 0.9084328210540202,
+      "eval_accuracy_tiles": 0.2571565538967265,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809175203079192,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8892481324556094,
+      "eval_iou_concrete tiles": 0.9486600598135225,
+      "eval_iou_dark bricks": 0.8321173208813658,
+      "eval_iou_grass": 0.8343860645711021,
+      "eval_iou_red bricks": 0.8594568062827225,
+      "eval_iou_road": 0.917690314531801,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33380294744647326,
+      "eval_iou_speedway bricks": 0.9774145021038203,
+      "eval_iou_steel": 0.8545877608004343,
+      "eval_iou_tiles": 0.25584247462015397,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1097680851817131,
+      "eval_mean_accuracy": 0.7095303703307111,
+      "eval_mean_iou": 0.6680095310626866,
+      "eval_overall_accuracy": 0.9739066229926215,
+      "eval_runtime": 3.423,
+      "eval_samples_per_second": 10.517,
+      "eval_steps_per_second": 0.584,
+      "step": 4880
+    },
+    {
+      "epoch": 443.73,
+      "learning_rate": 1.184688995215311e-05,
+      "loss": 0.0288,
+      "step": 4881
+    },
+    {
+      "epoch": 443.82,
+      "learning_rate": 1.182775119617225e-05,
+      "loss": 0.0308,
+      "step": 4882
+    },
+    {
+      "epoch": 443.91,
+      "learning_rate": 1.1808612440191388e-05,
+      "loss": 0.0285,
+      "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.0323,
+      "step": 4885
+    },
+    {
+      "epoch": 444.18,
+      "learning_rate": 1.1751196172248804e-05,
+      "loss": 0.0257,
+      "step": 4886
+    },
+    {
+      "epoch": 444.27,
+      "learning_rate": 1.1732057416267943e-05,
+      "loss": 0.0306,
+      "step": 4887
+    },
+    {
+      "epoch": 444.36,
+      "learning_rate": 1.1712918660287082e-05,
+      "loss": 0.0306,
+      "step": 4888
+    },
+    {
+      "epoch": 444.45,
+      "learning_rate": 1.169377990430622e-05,
+      "loss": 0.028,
+      "step": 4889
+    },
+    {
+      "epoch": 444.55,
+      "learning_rate": 1.167464114832536e-05,
+      "loss": 0.0292,
+      "step": 4890
+    },
+    {
+      "epoch": 444.64,
+      "learning_rate": 1.1655502392344498e-05,
+      "loss": 0.025,
+      "step": 4891
+    },
+    {
+      "epoch": 444.73,
+      "learning_rate": 1.1636363636363637e-05,
+      "loss": 0.0314,
+      "step": 4892
+    },
+    {
+      "epoch": 444.82,
+      "learning_rate": 1.1617224880382776e-05,
+      "loss": 0.033,
+      "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.0277,
+      "step": 4895
+    },
+    {
+      "epoch": 445.09,
+      "learning_rate": 1.1559808612440192e-05,
+      "loss": 0.0287,
+      "step": 4896
+    },
+    {
+      "epoch": 445.18,
+      "learning_rate": 1.1540669856459331e-05,
+      "loss": 0.0265,
+      "step": 4897
+    },
+    {
+      "epoch": 445.27,
+      "learning_rate": 1.152153110047847e-05,
+      "loss": 0.0295,
+      "step": 4898
+    },
+    {
+      "epoch": 445.36,
+      "learning_rate": 1.1502392344497607e-05,
+      "loss": 0.0369,
+      "step": 4899
+    },
+    {
+      "epoch": 445.45,
+      "learning_rate": 1.1483253588516747e-05,
+      "loss": 0.0283,
+      "step": 4900
+    },
+    {
+      "epoch": 445.45,
+      "eval_accuracy_NAT": 0.9930799401195491,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.949193098416241,
+      "eval_accuracy_concrete tiles": 0.9894209013636641,
+      "eval_accuracy_dark bricks": 0.8812978715214965,
+      "eval_accuracy_grass": 0.9031759734654006,
+      "eval_accuracy_red bricks": 0.9348366477272727,
+      "eval_accuracy_road": 0.9236578084056944,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4617208391916853,
+      "eval_accuracy_speedway bricks": 0.9886993786106415,
+      "eval_accuracy_steel": 0.90791339621397,
+      "eval_accuracy_tiles": 0.26736063552937955,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806139629019868,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8884592496993223,
+      "eval_iou_concrete tiles": 0.9468910021945867,
+      "eval_iou_dark bricks": 0.831982849460412,
+      "eval_iou_grass": 0.8321922440277983,
+      "eval_iou_red bricks": 0.8614201570680629,
+      "eval_iou_road": 0.9161907914701095,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3134627262348163,
+      "eval_iou_speedway bricks": 0.9774938374421029,
+      "eval_iou_steel": 0.8543774439823723,
+      "eval_iou_tiles": 0.2659400544959128,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11146039515733719,
+      "eval_mean_accuracy": 0.7077197300434612,
+      "eval_mean_iou": 0.6668480245367294,
+      "eval_overall_accuracy": 0.9735794067382812,
+      "eval_runtime": 4.399,
+      "eval_samples_per_second": 8.184,
+      "eval_steps_per_second": 0.455,
+      "step": 4900
+    },
+    {
+      "epoch": 445.55,
+      "learning_rate": 1.1464114832535886e-05,
+      "loss": 0.0332,
+      "step": 4901
+    },
+    {
+      "epoch": 445.64,
+      "learning_rate": 1.1444976076555025e-05,
+      "loss": 0.0296,
+      "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.0293,
+      "step": 4904
+    },
+    {
+      "epoch": 445.91,
+      "learning_rate": 1.1387559808612441e-05,
+      "loss": 0.0275,
+      "step": 4905
+    },
+    {
+      "epoch": 446.0,
+      "learning_rate": 1.136842105263158e-05,
+      "loss": 0.0331,
+      "step": 4906
+    },
+    {
+      "epoch": 446.09,
+      "learning_rate": 1.1349282296650717e-05,
+      "loss": 0.0298,
+      "step": 4907
+    },
+    {
+      "epoch": 446.18,
+      "learning_rate": 1.1330143540669858e-05,
+      "loss": 0.0322,
+      "step": 4908
+    },
+    {
+      "epoch": 446.27,
+      "learning_rate": 1.1311004784688996e-05,
+      "loss": 0.0329,
+      "step": 4909
+    },
+    {
+      "epoch": 446.36,
+      "learning_rate": 1.1291866028708133e-05,
+      "loss": 0.0294,
+      "step": 4910
+    },
+    {
+      "epoch": 446.45,
+      "learning_rate": 1.1272727272727274e-05,
+      "loss": 0.0285,
+      "step": 4911
+    },
+    {
+      "epoch": 446.55,
+      "learning_rate": 1.1253588516746413e-05,
+      "loss": 0.03,
+      "step": 4912
+    },
+    {
+      "epoch": 446.64,
+      "learning_rate": 1.123444976076555e-05,
+      "loss": 0.0294,
+      "step": 4913
+    },
+    {
+      "epoch": 446.73,
+      "learning_rate": 1.121531100478469e-05,
+      "loss": 0.033,
+      "step": 4914
+    },
+    {
+      "epoch": 446.82,
+      "learning_rate": 1.1196172248803829e-05,
+      "loss": 0.0262,
+      "step": 4915
+    },
+    {
+      "epoch": 446.91,
+      "learning_rate": 1.1177033492822966e-05,
+      "loss": 0.0304,
+      "step": 4916
+    },
+    {
+      "epoch": 447.0,
+      "learning_rate": 1.1157894736842106e-05,
+      "loss": 0.0264,
+      "step": 4917
+    },
+    {
+      "epoch": 447.09,
+      "learning_rate": 1.1138755980861244e-05,
+      "loss": 0.0281,
+      "step": 4918
+    },
+    {
+      "epoch": 447.18,
+      "learning_rate": 1.1119617224880384e-05,
+      "loss": 0.0292,
+      "step": 4919
+    },
+    {
+      "epoch": 447.27,
+      "learning_rate": 1.1100478468899523e-05,
+      "loss": 0.027,
+      "step": 4920
+    },
+    {
+      "epoch": 447.27,
+      "eval_accuracy_NAT": 0.9928408488569144,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9508629662217263,
+      "eval_accuracy_concrete tiles": 0.9897572310890969,
+      "eval_accuracy_dark bricks": 0.8775643739655903,
+      "eval_accuracy_grass": 0.9054128206293229,
+      "eval_accuracy_red bricks": 0.9343039772727273,
+      "eval_accuracy_road": 0.9188512032232671,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48446502908499434,
+      "eval_accuracy_speedway bricks": 0.9890378903897652,
+      "eval_accuracy_steel": 0.9059841039509268,
+      "eval_accuracy_tiles": 0.2420901246404602,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806529658870141,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.88954094596375,
+      "eval_iou_concrete tiles": 0.9479340537058186,
+      "eval_iou_dark bricks": 0.8313885647607935,
+      "eval_iou_grass": 0.8342909039380628,
+      "eval_iou_red bricks": 0.8596634536840385,
+      "eval_iou_road": 0.9118561531972289,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.32741466686540466,
+      "eval_iou_speedway bricks": 0.9773964470931006,
+      "eval_iou_steel": 0.8545044519615849,
+      "eval_iou_tiles": 0.2413793103448276,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11351735144853592,
+      "eval_mean_accuracy": 0.7070131207172915,
+      "eval_mean_iou": 0.6658478398001251,
+      "eval_overall_accuracy": 0.9732879002888998,
+      "eval_runtime": 3.5854,
+      "eval_samples_per_second": 10.041,
+      "eval_steps_per_second": 0.558,
+      "step": 4920
+    },
+    {
+      "epoch": 447.36,
+      "learning_rate": 1.108133971291866e-05,
+      "loss": 0.0303,
+      "step": 4921
+    },
+    {
+      "epoch": 447.45,
+      "learning_rate": 1.10622009569378e-05,
+      "loss": 0.0316,
+      "step": 4922
+    },
+    {
+      "epoch": 447.55,
+      "learning_rate": 1.104306220095694e-05,
+      "loss": 0.0292,
+      "step": 4923
+    },
+    {
+      "epoch": 447.64,
+      "learning_rate": 1.1023923444976076e-05,
+      "loss": 0.026,
+      "step": 4924
+    },
+    {
+      "epoch": 447.73,
+      "learning_rate": 1.1004784688995217e-05,
+      "loss": 0.0318,
+      "step": 4925
+    },
+    {
+      "epoch": 447.82,
+      "learning_rate": 1.0985645933014354e-05,
+      "loss": 0.0281,
+      "step": 4926
+    },
+    {
+      "epoch": 447.91,
+      "learning_rate": 1.0966507177033493e-05,
+      "loss": 0.0291,
+      "step": 4927
+    },
+    {
+      "epoch": 448.0,
+      "learning_rate": 1.0947368421052633e-05,
+      "loss": 0.0305,
+      "step": 4928
+    },
+    {
+      "epoch": 448.09,
+      "learning_rate": 1.092822966507177e-05,
+      "loss": 0.0312,
+      "step": 4929
+    },
+    {
+      "epoch": 448.18,
+      "learning_rate": 1.0909090909090909e-05,
+      "loss": 0.0321,
+      "step": 4930
+    },
+    {
+      "epoch": 448.27,
+      "learning_rate": 1.088995215311005e-05,
+      "loss": 0.0257,
+      "step": 4931
+    },
+    {
+      "epoch": 448.36,
+      "learning_rate": 1.0870813397129186e-05,
+      "loss": 0.027,
+      "step": 4932
+    },
+    {
+      "epoch": 448.45,
+      "learning_rate": 1.0851674641148325e-05,
+      "loss": 0.0283,
+      "step": 4933
+    },
+    {
+      "epoch": 448.55,
+      "learning_rate": 1.0832535885167466e-05,
+      "loss": 0.0275,
+      "step": 4934
+    },
+    {
+      "epoch": 448.64,
+      "learning_rate": 1.0813397129186603e-05,
+      "loss": 0.0295,
+      "step": 4935
+    },
+    {
+      "epoch": 448.73,
+      "learning_rate": 1.0794258373205742e-05,
+      "loss": 0.0298,
+      "step": 4936
+    },
+    {
+      "epoch": 448.82,
+      "learning_rate": 1.077511961722488e-05,
+      "loss": 0.0299,
+      "step": 4937
+    },
+    {
+      "epoch": 448.91,
+      "learning_rate": 1.0755980861244019e-05,
+      "loss": 0.0321,
+      "step": 4938
+    },
+    {
+      "epoch": 449.0,
+      "learning_rate": 1.073684210526316e-05,
+      "loss": 0.0348,
+      "step": 4939
+    },
+    {
+      "epoch": 449.09,
+      "learning_rate": 1.0717703349282297e-05,
+      "loss": 0.0307,
+      "step": 4940
+    },
+    {
+      "epoch": 449.09,
+      "eval_accuracy_NAT": 0.9928232037452808,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9506228808103352,
+      "eval_accuracy_concrete tiles": 0.9882284596098575,
+      "eval_accuracy_dark bricks": 0.8797197952349795,
+      "eval_accuracy_grass": 0.9082787810580986,
+      "eval_accuracy_red bricks": 0.9330610795454546,
+      "eval_accuracy_road": 0.9208645359600575,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4991867229068453,
+      "eval_accuracy_speedway bricks": 0.9891525378728909,
+      "eval_accuracy_steel": 0.9067756084690983,
+      "eval_accuracy_tiles": 0.2577044240514998,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807382651174125,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8908454908703778,
+      "eval_iou_concrete tiles": 0.9498075171177525,
+      "eval_iou_dark bricks": 0.8318835304822566,
+      "eval_iou_grass": 0.8360136271951977,
+      "eval_iou_red bricks": 0.8623235969806367,
+      "eval_iou_road": 0.9140340411946224,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33620513582264144,
+      "eval_iou_speedway bricks": 0.97737802874878,
+      "eval_iou_steel": 0.8557633950108156,
+      "eval_iou_tiles": 0.2563176895306859,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11107033491134644,
+      "eval_mean_accuracy": 0.7097244637895691,
+      "eval_mean_iou": 0.6685623321593214,
+      "eval_overall_accuracy": 0.9736326005723741,
+      "eval_runtime": 3.4981,
+      "eval_samples_per_second": 10.291,
+      "eval_steps_per_second": 0.572,
+      "step": 4940
+    },
+    {
+      "epoch": 449.18,
+      "learning_rate": 1.0698564593301435e-05,
+      "loss": 0.0288,
+      "step": 4941
+    },
+    {
+      "epoch": 449.27,
+      "learning_rate": 1.0679425837320576e-05,
+      "loss": 0.0321,
+      "step": 4942
+    },
+    {
+      "epoch": 449.36,
+      "learning_rate": 1.0660287081339713e-05,
+      "loss": 0.0283,
+      "step": 4943
+    },
+    {
+      "epoch": 449.45,
+      "learning_rate": 1.0641148325358852e-05,
+      "loss": 0.0271,
+      "step": 4944
+    },
+    {
+      "epoch": 449.55,
+      "learning_rate": 1.0622009569377992e-05,
+      "loss": 0.0289,
+      "step": 4945
+    },
+    {
+      "epoch": 449.64,
+      "learning_rate": 1.060287081339713e-05,
+      "loss": 0.0361,
+      "step": 4946
+    },
+    {
+      "epoch": 449.73,
+      "learning_rate": 1.0583732057416268e-05,
+      "loss": 0.0325,
+      "step": 4947
+    },
+    {
+      "epoch": 449.82,
+      "learning_rate": 1.0564593301435407e-05,
+      "loss": 0.0277,
+      "step": 4948
+    },
+    {
+      "epoch": 449.91,
+      "learning_rate": 1.0545454545454546e-05,
+      "loss": 0.0299,
+      "step": 4949
+    },
+    {
+      "epoch": 450.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0279,
+      "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.0339,
+      "step": 4952
+    },
+    {
+      "epoch": 450.27,
+      "learning_rate": 1.04688995215311e-05,
+      "loss": 0.0346,
+      "step": 4953
+    },
+    {
+      "epoch": 450.36,
+      "learning_rate": 1.044976076555024e-05,
+      "loss": 0.0337,
+      "step": 4954
+    },
+    {
+      "epoch": 450.45,
+      "learning_rate": 1.0430622009569378e-05,
+      "loss": 0.0302,
+      "step": 4955
+    },
+    {
+      "epoch": 450.55,
+      "learning_rate": 1.0411483253588517e-05,
+      "loss": 0.0266,
+      "step": 4956
+    },
+    {
+      "epoch": 450.64,
+      "learning_rate": 1.0392344497607656e-05,
+      "loss": 0.0266,
+      "step": 4957
+    },
+    {
+      "epoch": 450.73,
+      "learning_rate": 1.0373205741626795e-05,
+      "loss": 0.03,
+      "step": 4958
+    },
+    {
+      "epoch": 450.82,
+      "learning_rate": 1.0354066985645933e-05,
+      "loss": 0.0283,
+      "step": 4959
+    },
+    {
+      "epoch": 450.91,
+      "learning_rate": 1.0334928229665072e-05,
+      "loss": 0.0292,
+      "step": 4960
+    },
+    {
+      "epoch": 450.91,
+      "eval_accuracy_NAT": 0.9932918579102683,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9518434267565131,
+      "eval_accuracy_concrete tiles": 0.9860576041093377,
+      "eval_accuracy_dark bricks": 0.8725992071128902,
+      "eval_accuracy_grass": 0.8997123554350144,
+      "eval_accuracy_red bricks": 0.9295099431818182,
+      "eval_accuracy_road": 0.92195282392589,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4959336145342266,
+      "eval_accuracy_speedway bricks": 0.9878763303633601,
+      "eval_accuracy_steel": 0.9132972759052833,
+      "eval_accuracy_tiles": 0.2733187234625394,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806421054153571,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8910127073370088,
+      "eval_iou_concrete tiles": 0.9521419503409997,
+      "eval_iou_dark bricks": 0.8330944769044207,
+      "eval_iou_grass": 0.8350135912756337,
+      "eval_iou_red bricks": 0.865146256817055,
+      "eval_iou_road": 0.9146663467594398,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33444262660816537,
+      "eval_iou_speedway bricks": 0.9775210322364924,
+      "eval_iou_steel": 0.8556200951615893,
+      "eval_iou_tiles": 0.2712935898307389,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11137256026268005,
+      "eval_mean_accuracy": 0.7096456278997801,
+      "eval_mean_iou": 0.6700457522066846,
+      "eval_overall_accuracy": 0.9736882315741645,
+      "eval_runtime": 3.4834,
+      "eval_samples_per_second": 10.335,
+      "eval_steps_per_second": 0.574,
+      "step": 4960
+    },
+    {
+      "epoch": 451.0,
+      "learning_rate": 1.0315789473684211e-05,
+      "loss": 0.0325,
+      "step": 4961
+    },
+    {
+      "epoch": 451.09,
+      "learning_rate": 1.029665071770335e-05,
+      "loss": 0.0272,
+      "step": 4962
+    },
+    {
+      "epoch": 451.18,
+      "learning_rate": 1.0277511961722489e-05,
+      "loss": 0.0297,
+      "step": 4963
+    },
+    {
+      "epoch": 451.27,
+      "learning_rate": 1.0258373205741627e-05,
+      "loss": 0.0252,
+      "step": 4964
+    },
+    {
+      "epoch": 451.36,
+      "learning_rate": 1.0239234449760766e-05,
+      "loss": 0.0295,
+      "step": 4965
+    },
+    {
+      "epoch": 451.45,
+      "learning_rate": 1.0220095693779905e-05,
+      "loss": 0.0295,
+      "step": 4966
+    },
+    {
+      "epoch": 451.55,
+      "learning_rate": 1.0200956937799044e-05,
+      "loss": 0.0303,
+      "step": 4967
+    },
+    {
+      "epoch": 451.64,
+      "learning_rate": 1.0181818181818182e-05,
+      "loss": 0.0286,
+      "step": 4968
+    },
+    {
+      "epoch": 451.73,
+      "learning_rate": 1.0162679425837321e-05,
+      "loss": 0.0292,
+      "step": 4969
+    },
+    {
+      "epoch": 451.82,
+      "learning_rate": 1.014354066985646e-05,
+      "loss": 0.0254,
+      "step": 4970
+    },
+    {
+      "epoch": 451.91,
+      "learning_rate": 1.0124401913875599e-05,
+      "loss": 0.029,
+      "step": 4971
+    },
+    {
+      "epoch": 452.0,
+      "learning_rate": 1.0105263157894738e-05,
+      "loss": 0.0303,
+      "step": 4972
+    },
+    {
+      "epoch": 452.09,
+      "learning_rate": 1.0086124401913876e-05,
+      "loss": 0.031,
+      "step": 4973
+    },
+    {
+      "epoch": 452.18,
+      "learning_rate": 1.0066985645933015e-05,
+      "loss": 0.0405,
+      "step": 4974
+    },
+    {
+      "epoch": 452.27,
+      "learning_rate": 1.0047846889952154e-05,
+      "loss": 0.0261,
+      "step": 4975
+    },
+    {
+      "epoch": 452.36,
+      "learning_rate": 1.0028708133971293e-05,
+      "loss": 0.0312,
+      "step": 4976
+    },
+    {
+      "epoch": 452.45,
+      "learning_rate": 1.0009569377990431e-05,
+      "loss": 0.0307,
+      "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.0284,
+      "step": 4979
+    },
+    {
+      "epoch": 452.73,
+      "learning_rate": 9.952153110047848e-06,
+      "loss": 0.0329,
+      "step": 4980
+    },
+    {
+      "epoch": 452.73,
+      "eval_accuracy_NAT": 0.9932819766477534,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9499012833168805,
+      "eval_accuracy_concrete tiles": 0.9850791903626246,
+      "eval_accuracy_dark bricks": 0.874023324737308,
+      "eval_accuracy_grass": 0.905224086649867,
+      "eval_accuracy_red bricks": 0.9351917613636364,
+      "eval_accuracy_road": 0.9335709025655784,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5081465552890579,
+      "eval_accuracy_speedway bricks": 0.9885117187829988,
+      "eval_accuracy_steel": 0.9105764791240684,
+      "eval_accuracy_tiles": 0.27708533077660596,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806673714049983,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.893616376960814,
+      "eval_iou_concrete tiles": 0.9537313874663272,
+      "eval_iou_dark bricks": 0.8323131620422974,
+      "eval_iou_grass": 0.8363634015125584,
+      "eval_iou_red bricks": 0.8648604269293925,
+      "eval_iou_road": 0.9257461536709046,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.34283801127169244,
+      "eval_iou_speedway bricks": 0.9776080747706384,
+      "eval_iou_steel": 0.8557746697144628,
+      "eval_iou_tiles": 0.2753879662401307,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10519985854625702,
+      "eval_mean_accuracy": 0.7123532776627984,
+      "eval_mean_iou": 0.6722236155372474,
+      "eval_overall_accuracy": 0.9748473697238498,
+      "eval_runtime": 4.0208,
+      "eval_samples_per_second": 8.953,
+      "eval_steps_per_second": 0.497,
+      "step": 4980
+    },
+    {
+      "epoch": 452.82,
+      "learning_rate": 9.933014354066985e-06,
+      "loss": 0.0273,
+      "step": 4981
+    },
+    {
+      "epoch": 452.91,
+      "learning_rate": 9.913875598086125e-06,
+      "loss": 0.0287,
+      "step": 4982
+    },
+    {
+      "epoch": 453.0,
+      "learning_rate": 9.894736842105264e-06,
+      "loss": 0.0261,
+      "step": 4983
+    },
+    {
+      "epoch": 453.09,
+      "learning_rate": 9.875598086124401e-06,
+      "loss": 0.0304,
+      "step": 4984
+    },
+    {
+      "epoch": 453.18,
+      "learning_rate": 9.856459330143542e-06,
+      "loss": 0.0309,
+      "step": 4985
+    },
+    {
+      "epoch": 453.27,
+      "learning_rate": 9.83732057416268e-06,
+      "loss": 0.028,
+      "step": 4986
+    },
+    {
+      "epoch": 453.36,
+      "learning_rate": 9.818181818181818e-06,
+      "loss": 0.0283,
+      "step": 4987
+    },
+    {
+      "epoch": 453.45,
+      "learning_rate": 9.799043062200958e-06,
+      "loss": 0.0293,
+      "step": 4988
+    },
+    {
+      "epoch": 453.55,
+      "learning_rate": 9.779904306220097e-06,
+      "loss": 0.0343,
+      "step": 4989
+    },
+    {
+      "epoch": 453.64,
+      "learning_rate": 9.760765550239234e-06,
+      "loss": 0.0315,
+      "step": 4990
+    },
+    {
+      "epoch": 453.73,
+      "learning_rate": 9.741626794258374e-06,
+      "loss": 0.0298,
+      "step": 4991
+    },
+    {
+      "epoch": 453.82,
+      "learning_rate": 9.722488038277511e-06,
+      "loss": 0.0292,
+      "step": 4992
+    },
+    {
+      "epoch": 453.91,
+      "learning_rate": 9.703349282296652e-06,
+      "loss": 0.0286,
+      "step": 4993
+    },
+    {
+      "epoch": 454.0,
+      "learning_rate": 9.68421052631579e-06,
+      "loss": 0.0283,
+      "step": 4994
+    },
+    {
+      "epoch": 454.09,
+      "learning_rate": 9.665071770334928e-06,
+      "loss": 0.028,
+      "step": 4995
+    },
+    {
+      "epoch": 454.18,
+      "learning_rate": 9.645933014354068e-06,
+      "loss": 0.0305,
+      "step": 4996
+    },
+    {
+      "epoch": 454.27,
+      "learning_rate": 9.626794258373207e-06,
+      "loss": 0.026,
+      "step": 4997
+    },
+    {
+      "epoch": 454.36,
+      "learning_rate": 9.607655502392344e-06,
+      "loss": 0.0299,
+      "step": 4998
+    },
+    {
+      "epoch": 454.45,
+      "learning_rate": 9.588516746411485e-06,
+      "loss": 0.031,
+      "step": 4999
+    },
+    {
+      "epoch": 454.55,
+      "learning_rate": 9.569377990430622e-06,
+      "loss": 0.0314,
+      "step": 5000
+    },
+    {
+      "epoch": 454.55,
+      "eval_accuracy_NAT": 0.993021005446693,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.951597976307996,
+      "eval_accuracy_concrete tiles": 0.9868525652785421,
+      "eval_accuracy_dark bricks": 0.8730225934336631,
+      "eval_accuracy_grass": 0.9056819413037324,
+      "eval_accuracy_red bricks": 0.939453125,
+      "eval_accuracy_road": 0.939891437229364,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49601632067929313,
+      "eval_accuracy_speedway bricks": 0.9890240120102289,
+      "eval_accuracy_steel": 0.9098426884770134,
+      "eval_accuracy_tiles": 0.22086015614299412,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808122783572064,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8909787088688349,
+      "eval_iou_concrete tiles": 0.9546570439823716,
+      "eval_iou_dark bricks": 0.8333455801307958,
+      "eval_iou_grass": 0.8362328521777855,
+      "eval_iou_red bricks": 0.8597660058498537,
+      "eval_iou_road": 0.9332289579308697,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33413189221312234,
+      "eval_iou_speedway bricks": 0.9774579214533796,
+      "eval_iou_steel": 0.8565784366995265,
+      "eval_iou_tiles": 0.2198364008179959,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10168782621622086,
+      "eval_mean_accuracy": 0.7080972170238092,
+      "eval_mean_iou": 0.6674635444985955,
+      "eval_overall_accuracy": 0.9753452936808268,
+      "eval_runtime": 3.8102,
+      "eval_samples_per_second": 9.448,
+      "eval_steps_per_second": 0.525,
+      "step": 5000
+    },
+    {
+      "epoch": 454.64,
+      "learning_rate": 9.55023923444976e-06,
+      "loss": 0.0305,
+      "step": 5001
+    },
+    {
+      "epoch": 454.73,
+      "learning_rate": 9.531100478468901e-06,
+      "loss": 0.0322,
+      "step": 5002
+    },
+    {
+      "epoch": 454.82,
+      "learning_rate": 9.511961722488038e-06,
+      "loss": 0.0267,
+      "step": 5003
+    },
+    {
+      "epoch": 454.91,
+      "learning_rate": 9.492822966507177e-06,
+      "loss": 0.0286,
+      "step": 5004
+    },
+    {
+      "epoch": 455.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0297,
+      "step": 5005
+    },
+    {
+      "epoch": 455.09,
+      "learning_rate": 9.454545454545454e-06,
+      "loss": 0.029,
+      "step": 5006
+    },
+    {
+      "epoch": 455.18,
+      "learning_rate": 9.435406698564593e-06,
+      "loss": 0.0282,
+      "step": 5007
+    },
+    {
+      "epoch": 455.27,
+      "learning_rate": 9.416267942583734e-06,
+      "loss": 0.0305,
+      "step": 5008
+    },
+    {
+      "epoch": 455.36,
+      "learning_rate": 9.39712918660287e-06,
+      "loss": 0.0277,
+      "step": 5009
+    },
+    {
+      "epoch": 455.45,
+      "learning_rate": 9.377990430622011e-06,
+      "loss": 0.0327,
+      "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.0284,
+      "step": 5012
+    },
+    {
+      "epoch": 455.73,
+      "learning_rate": 9.320574162679427e-06,
+      "loss": 0.0284,
+      "step": 5013
+    },
+    {
+      "epoch": 455.82,
+      "learning_rate": 9.301435406698564e-06,
+      "loss": 0.0288,
+      "step": 5014
+    },
+    {
+      "epoch": 455.91,
+      "learning_rate": 9.282296650717703e-06,
+      "loss": 0.0316,
+      "step": 5015
+    },
+    {
+      "epoch": 456.0,
+      "learning_rate": 9.263157894736844e-06,
+      "loss": 0.0317,
+      "step": 5016
+    },
+    {
+      "epoch": 456.09,
+      "learning_rate": 9.24401913875598e-06,
+      "loss": 0.0304,
+      "step": 5017
+    },
+    {
+      "epoch": 456.18,
+      "learning_rate": 9.22488038277512e-06,
+      "loss": 0.0267,
+      "step": 5018
+    },
+    {
+      "epoch": 456.27,
+      "learning_rate": 9.20574162679426e-06,
+      "loss": 0.0324,
+      "step": 5019
+    },
+    {
+      "epoch": 456.36,
+      "learning_rate": 9.186602870813397e-06,
+      "loss": 0.03,
+      "step": 5020
+    },
+    {
+      "epoch": 456.36,
+      "eval_accuracy_NAT": 0.9929534246691365,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9522578758745011,
+      "eval_accuracy_concrete tiles": 0.9881061578915183,
+      "eval_accuracy_dark bricks": 0.8754089526962011,
+      "eval_accuracy_grass": 0.9043118724158299,
+      "eval_accuracy_red bricks": 0.9403409090909091,
+      "eval_accuracy_road": 0.9388672373326303,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48951010393405564,
+      "eval_accuracy_speedway bricks": 0.9886722252593748,
+      "eval_accuracy_steel": 0.9099910955741706,
+      "eval_accuracy_tiles": 0.22277770168470074,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807709970934281,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8907654224333375,
+      "eval_iou_concrete tiles": 0.9533882054459096,
+      "eval_iou_dark bricks": 0.8322599531615925,
+      "eval_iou_grass": 0.8353263641822655,
+      "eval_iou_red bricks": 0.8601591684261816,
+      "eval_iou_road": 0.9313509927896351,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3303749185970788,
+      "eval_iou_speedway bricks": 0.9775521357241129,
+      "eval_iou_steel": 0.8571373100250841,
+      "eval_iou_tiles": 0.2216241994822183,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10293947160243988,
+      "eval_mean_accuracy": 0.7079382735710021,
+      "eval_mean_iou": 0.666977666720065,
+      "eval_overall_accuracy": 0.9751550886366103,
+      "eval_runtime": 3.3068,
+      "eval_samples_per_second": 10.887,
+      "eval_steps_per_second": 0.605,
+      "step": 5020
+    },
+    {
+      "epoch": 456.45,
+      "learning_rate": 9.167464114832536e-06,
+      "loss": 0.0315,
+      "step": 5021
+    },
+    {
+      "epoch": 456.55,
+      "learning_rate": 9.148325358851675e-06,
+      "loss": 0.0337,
+      "step": 5022
+    },
+    {
+      "epoch": 456.64,
+      "learning_rate": 9.129186602870813e-06,
+      "loss": 0.0304,
+      "step": 5023
+    },
+    {
+      "epoch": 456.73,
+      "learning_rate": 9.110047846889952e-06,
+      "loss": 0.0253,
+      "step": 5024
+    },
+    {
+      "epoch": 456.82,
+      "learning_rate": 9.090909090909091e-06,
+      "loss": 0.0257,
+      "step": 5025
+    },
+    {
+      "epoch": 456.91,
+      "learning_rate": 9.07177033492823e-06,
+      "loss": 0.0323,
+      "step": 5026
+    },
+    {
+      "epoch": 457.0,
+      "learning_rate": 9.05263157894737e-06,
+      "loss": 0.0255,
+      "step": 5027
+    },
+    {
+      "epoch": 457.09,
+      "learning_rate": 9.033492822966507e-06,
+      "loss": 0.0339,
+      "step": 5028
+    },
+    {
+      "epoch": 457.18,
+      "learning_rate": 9.014354066985646e-06,
+      "loss": 0.0348,
+      "step": 5029
+    },
+    {
+      "epoch": 457.27,
+      "learning_rate": 8.995215311004785e-06,
+      "loss": 0.0282,
+      "step": 5030
+    },
+    {
+      "epoch": 457.36,
+      "learning_rate": 8.976076555023924e-06,
+      "loss": 0.0313,
+      "step": 5031
+    },
+    {
+      "epoch": 457.45,
+      "learning_rate": 8.956937799043062e-06,
+      "loss": 0.0318,
+      "step": 5032
+    },
+    {
+      "epoch": 457.55,
+      "learning_rate": 8.937799043062201e-06,
+      "loss": 0.0273,
+      "step": 5033
+    },
+    {
+      "epoch": 457.64,
+      "learning_rate": 8.91866028708134e-06,
+      "loss": 0.0261,
+      "step": 5034
+    },
+    {
+      "epoch": 457.73,
+      "learning_rate": 8.899521531100479e-06,
+      "loss": 0.0261,
+      "step": 5035
+    },
+    {
+      "epoch": 457.82,
+      "learning_rate": 8.880382775119618e-06,
+      "loss": 0.0292,
+      "step": 5036
+    },
+    {
+      "epoch": 457.91,
+      "learning_rate": 8.861244019138756e-06,
+      "loss": 0.0315,
+      "step": 5037
+    },
+    {
+      "epoch": 458.0,
+      "learning_rate": 8.842105263157895e-06,
+      "loss": 0.0323,
+      "step": 5038
+    },
+    {
+      "epoch": 458.09,
+      "learning_rate": 8.822966507177034e-06,
+      "loss": 0.025,
+      "step": 5039
+    },
+    {
+      "epoch": 458.18,
+      "learning_rate": 8.803827751196173e-06,
+      "loss": 0.0311,
+      "step": 5040
+    },
+    {
+      "epoch": 458.18,
+      "eval_accuracy_NAT": 0.9930829397885269,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9527943795871068,
+      "eval_accuracy_concrete tiles": 0.9856906989543203,
+      "eval_accuracy_dark bricks": 0.8703668065124514,
+      "eval_accuracy_grass": 0.9032179143497241,
+      "eval_accuracy_red bricks": 0.9412286931818182,
+      "eval_accuracy_road": 0.9387100401820101,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48598130841121495,
+      "eval_accuracy_speedway bricks": 0.9883994849310969,
+      "eval_accuracy_steel": 0.9077897236330057,
+      "eval_accuracy_tiles": 0.23941925763594027,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807174002557003,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8906951171652149,
+      "eval_iou_concrete tiles": 0.9545212293480192,
+      "eval_iou_dark bricks": 0.8315437228800471,
+      "eval_iou_grass": 0.834547568300717,
+      "eval_iou_red bricks": 0.8613909652258693,
+      "eval_iou_road": 0.9311637792989952,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.32795668917787574,
+      "eval_iou_speedway bricks": 0.9776546015799764,
+      "eval_iou_steel": 0.8571005986252637,
+      "eval_iou_tiles": 0.23813091751243104,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10318261384963989,
+      "eval_mean_accuracy": 0.7082062497820935,
+      "eval_mean_iou": 0.6681094299515469,
+      "eval_overall_accuracy": 0.975142584906684,
+      "eval_runtime": 3.598,
+      "eval_samples_per_second": 10.005,
+      "eval_steps_per_second": 0.556,
+      "step": 5040
+    },
+    {
+      "epoch": 458.27,
+      "learning_rate": 8.784688995215311e-06,
+      "loss": 0.0293,
+      "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.0297,
+      "step": 5043
+    },
+    {
+      "epoch": 458.55,
+      "learning_rate": 8.727272727272728e-06,
+      "loss": 0.0324,
+      "step": 5044
+    },
+    {
+      "epoch": 458.64,
+      "learning_rate": 8.708133971291867e-06,
+      "loss": 0.0297,
+      "step": 5045
+    },
+    {
+      "epoch": 458.73,
+      "learning_rate": 8.688995215311005e-06,
+      "loss": 0.0262,
+      "step": 5046
+    },
+    {
+      "epoch": 458.82,
+      "learning_rate": 8.669856459330144e-06,
+      "loss": 0.0284,
+      "step": 5047
+    },
+    {
+      "epoch": 458.91,
+      "learning_rate": 8.650717703349283e-06,
+      "loss": 0.0286,
+      "step": 5048
+    },
+    {
+      "epoch": 459.0,
+      "learning_rate": 8.631578947368422e-06,
+      "loss": 0.0353,
+      "step": 5049
+    },
+    {
+      "epoch": 459.09,
+      "learning_rate": 8.61244019138756e-06,
+      "loss": 0.0366,
+      "step": 5050
+    },
+    {
+      "epoch": 459.18,
+      "learning_rate": 8.5933014354067e-06,
+      "loss": 0.029,
+      "step": 5051
+    },
+    {
+      "epoch": 459.27,
+      "learning_rate": 8.574162679425838e-06,
+      "loss": 0.0298,
+      "step": 5052
+    },
+    {
+      "epoch": 459.36,
+      "learning_rate": 8.555023923444977e-06,
+      "loss": 0.0316,
+      "step": 5053
+    },
+    {
+      "epoch": 459.45,
+      "learning_rate": 8.535885167464116e-06,
+      "loss": 0.0287,
+      "step": 5054
+    },
+    {
+      "epoch": 459.55,
+      "learning_rate": 8.516746411483253e-06,
+      "loss": 0.0293,
+      "step": 5055
+    },
+    {
+      "epoch": 459.64,
+      "learning_rate": 8.497607655502393e-06,
+      "loss": 0.0279,
+      "step": 5056
+    },
+    {
+      "epoch": 459.73,
+      "learning_rate": 8.478468899521532e-06,
+      "loss": 0.0347,
+      "step": 5057
+    },
+    {
+      "epoch": 459.82,
+      "learning_rate": 8.459330143540669e-06,
+      "loss": 0.03,
+      "step": 5058
+    },
+    {
+      "epoch": 459.91,
+      "learning_rate": 8.44019138755981e-06,
+      "loss": 0.0272,
+      "step": 5059
+    },
+    {
+      "epoch": 460.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0269,
+      "step": 5060
+    },
+    {
+      "epoch": 460.0,
+      "eval_accuracy_NAT": 0.993197633014145,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9534194064122924,
+      "eval_accuracy_concrete tiles": 0.9823885525591635,
+      "eval_accuracy_dark bricks": 0.8792964089142066,
+      "eval_accuracy_grass": 0.9008098085748139,
+      "eval_accuracy_red bricks": 0.9385653409090909,
+      "eval_accuracy_road": 0.9368962269056225,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4996002536321782,
+      "eval_accuracy_speedway bricks": 0.9877797851144121,
+      "eval_accuracy_steel": 0.9113927181584328,
+      "eval_accuracy_tiles": 0.24626763457060677,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807723872600012,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8912436479805813,
+      "eval_iou_concrete tiles": 0.9550561797752809,
+      "eval_iou_dark bricks": 0.8334853515268708,
+      "eval_iou_grass": 0.8345459549597681,
+      "eval_iou_red bricks": 0.8662733529990168,
+      "eval_iou_road": 0.9295529801324504,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3354435065896639,
+      "eval_iou_speedway bricks": 0.9775311427187753,
+      "eval_iou_steel": 0.8566878240450118,
+      "eval_iou_tiles": 0.24532678400873242,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10305383801460266,
+      "eval_mean_accuracy": 0.7099702899049973,
+      "eval_mean_iou": 0.6696860855381657,
+      "eval_overall_accuracy": 0.9750408596462674,
+      "eval_runtime": 3.7535,
+      "eval_samples_per_second": 9.591,
+      "eval_steps_per_second": 0.533,
+      "step": 5060
+    },
+    {
+      "epoch": 460.09,
+      "learning_rate": 8.401913875598085e-06,
+      "loss": 0.0277,
+      "step": 5061
+    },
+    {
+      "epoch": 460.18,
+      "learning_rate": 8.382775119617226e-06,
+      "loss": 0.0296,
+      "step": 5062
+    },
+    {
+      "epoch": 460.27,
+      "learning_rate": 8.363636363636365e-06,
+      "loss": 0.0308,
+      "step": 5063
+    },
+    {
+      "epoch": 460.36,
+      "learning_rate": 8.344497607655503e-06,
+      "loss": 0.0279,
+      "step": 5064
+    },
+    {
+      "epoch": 460.45,
+      "learning_rate": 8.325358851674642e-06,
+      "loss": 0.0286,
+      "step": 5065
+    },
+    {
+      "epoch": 460.55,
+      "learning_rate": 8.30622009569378e-06,
+      "loss": 0.0299,
+      "step": 5066
+    },
+    {
+      "epoch": 460.64,
+      "learning_rate": 8.28708133971292e-06,
+      "loss": 0.0246,
+      "step": 5067
+    },
+    {
+      "epoch": 460.73,
+      "learning_rate": 8.267942583732058e-06,
+      "loss": 0.0318,
+      "step": 5068
+    },
+    {
+      "epoch": 460.82,
+      "learning_rate": 8.248803827751196e-06,
+      "loss": 0.0287,
+      "step": 5069
+    },
+    {
+      "epoch": 460.91,
+      "learning_rate": 8.229665071770336e-06,
+      "loss": 0.028,
+      "step": 5070
+    },
+    {
+      "epoch": 461.0,
+      "learning_rate": 8.210526315789475e-06,
+      "loss": 0.0269,
+      "step": 5071
+    },
+    {
+      "epoch": 461.09,
+      "learning_rate": 8.191387559808612e-06,
+      "loss": 0.0291,
+      "step": 5072
+    },
+    {
+      "epoch": 461.18,
+      "learning_rate": 8.172248803827752e-06,
+      "loss": 0.0298,
+      "step": 5073
+    },
+    {
+      "epoch": 461.27,
+      "learning_rate": 8.15311004784689e-06,
+      "loss": 0.0292,
+      "step": 5074
+    },
+    {
+      "epoch": 461.36,
+      "learning_rate": 8.133971291866028e-06,
+      "loss": 0.0304,
+      "step": 5075
+    },
+    {
+      "epoch": 461.45,
+      "learning_rate": 8.114832535885169e-06,
+      "loss": 0.0285,
+      "step": 5076
+    },
+    {
+      "epoch": 461.55,
+      "learning_rate": 8.095693779904306e-06,
+      "loss": 0.036,
+      "step": 5077
+    },
+    {
+      "epoch": 461.64,
+      "learning_rate": 8.076555023923445e-06,
+      "loss": 0.0305,
+      "step": 5078
+    },
+    {
+      "epoch": 461.73,
+      "learning_rate": 8.057416267942585e-06,
+      "loss": 0.0293,
+      "step": 5079
+    },
+    {
+      "epoch": 461.82,
+      "learning_rate": 8.038277511961722e-06,
+      "loss": 0.0261,
+      "step": 5080
+    },
+    {
+      "epoch": 461.82,
+      "eval_accuracy_NAT": 0.9932726247385877,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9515751749002103,
+      "eval_accuracy_concrete tiles": 0.9827554577141809,
+      "eval_accuracy_dark bricks": 0.8798737538970787,
+      "eval_accuracy_grass": 0.9002366164890586,
+      "eval_accuracy_red bricks": 0.9389204545454546,
+      "eval_accuracy_road": 0.9369724070632307,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5139911228737628,
+      "eval_accuracy_speedway bricks": 0.9883578497924881,
+      "eval_accuracy_steel": 0.9087296352483345,
+      "eval_accuracy_tiles": 0.26085467744144636,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807680011094948,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8920069855423414,
+      "eval_iou_concrete tiles": 0.953825152828061,
+      "eval_iou_dark bricks": 0.8338196673475343,
+      "eval_iou_grass": 0.8339123199502708,
+      "eval_iou_red bricks": 0.867311792684927,
+      "eval_iou_road": 0.9292573151382798,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.344093165753096,
+      "eval_iou_speedway bricks": 0.9775813283374803,
+      "eval_iou_steel": 0.8565810743596121,
+      "eval_iou_tiles": 0.25961014176663033,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10353000462055206,
+      "eval_mean_accuracy": 0.7119645980541409,
+      "eval_mean_iou": 0.6714436111398252,
+      "eval_overall_accuracy": 0.9750777350531684,
+      "eval_runtime": 3.2724,
+      "eval_samples_per_second": 11.001,
+      "eval_steps_per_second": 0.611,
+      "step": 5080
+    },
+    {
+      "epoch": 461.91,
+      "learning_rate": 8.019138755980863e-06,
+      "loss": 0.0346,
+      "step": 5081
+    },
+    {
+      "epoch": 462.0,
+      "learning_rate": 8.000000000000001e-06,
+      "loss": 0.0245,
+      "step": 5082
+    },
+    {
+      "epoch": 462.09,
+      "learning_rate": 7.980861244019138e-06,
+      "loss": 0.0275,
+      "step": 5083
+    },
+    {
+      "epoch": 462.18,
+      "learning_rate": 7.961722488038279e-06,
+      "loss": 0.0264,
+      "step": 5084
+    },
+    {
+      "epoch": 462.27,
+      "learning_rate": 7.942583732057416e-06,
+      "loss": 0.0278,
+      "step": 5085
+    },
+    {
+      "epoch": 462.36,
+      "learning_rate": 7.923444976076555e-06,
+      "loss": 0.0306,
+      "step": 5086
+    },
+    {
+      "epoch": 462.45,
+      "learning_rate": 7.904306220095695e-06,
+      "loss": 0.0293,
+      "step": 5087
+    },
+    {
+      "epoch": 462.55,
+      "learning_rate": 7.885167464114832e-06,
+      "loss": 0.0284,
+      "step": 5088
+    },
+    {
+      "epoch": 462.64,
+      "learning_rate": 7.866028708133971e-06,
+      "loss": 0.0284,
+      "step": 5089
+    },
+    {
+      "epoch": 462.73,
+      "learning_rate": 7.846889952153112e-06,
+      "loss": 0.033,
+      "step": 5090
+    },
+    {
+      "epoch": 462.82,
+      "learning_rate": 7.827751196172249e-06,
+      "loss": 0.0283,
+      "step": 5091
+    },
+    {
+      "epoch": 462.91,
+      "learning_rate": 7.808612440191387e-06,
+      "loss": 0.0245,
+      "step": 5092
+    },
+    {
+      "epoch": 463.0,
+      "learning_rate": 7.789473684210528e-06,
+      "loss": 0.0273,
+      "step": 5093
+    },
+    {
+      "epoch": 463.09,
+      "learning_rate": 7.770334928229665e-06,
+      "loss": 0.0397,
+      "step": 5094
+    },
+    {
+      "epoch": 463.18,
+      "learning_rate": 7.751196172248804e-06,
+      "loss": 0.0277,
+      "step": 5095
+    },
+    {
+      "epoch": 463.27,
+      "learning_rate": 7.732057416267943e-06,
+      "loss": 0.0277,
+      "step": 5096
+    },
+    {
+      "epoch": 463.36,
+      "learning_rate": 7.712918660287081e-06,
+      "loss": 0.0265,
+      "step": 5097
+    },
+    {
+      "epoch": 463.45,
+      "learning_rate": 7.693779904306222e-06,
+      "loss": 0.0288,
+      "step": 5098
+    },
+    {
+      "epoch": 463.55,
+      "learning_rate": 7.674641148325359e-06,
+      "loss": 0.0292,
+      "step": 5099
+    },
+    {
+      "epoch": 463.64,
+      "learning_rate": 7.655502392344498e-06,
+      "loss": 0.0336,
+      "step": 5100
+    },
+    {
+      "epoch": 463.64,
+      "eval_accuracy_NAT": 0.9930603540456359,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.952349081505644,
+      "eval_accuracy_concrete tiles": 0.9831835137283679,
+      "eval_accuracy_dark bricks": 0.8741387937338825,
+      "eval_accuracy_grass": 0.9039274143095307,
+      "eval_accuracy_red bricks": 0.927734375,
+      "eval_accuracy_road": 0.9356035826439835,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5061340390924379,
+      "eval_accuracy_speedway bricks": 0.9884562052648538,
+      "eval_accuracy_steel": 0.9097932194446277,
+      "eval_accuracy_tiles": 0.2520202711957266,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.980851314128064,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8913360858545599,
+      "eval_iou_concrete tiles": 0.9549490689870223,
+      "eval_iou_dark bricks": 0.832697807435653,
+      "eval_iou_grass": 0.8354891230609199,
+      "eval_iou_red bricks": 0.8667883211678832,
+      "eval_iou_road": 0.9279687595572511,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3398619004424369,
+      "eval_iou_speedway bricks": 0.9775846758696747,
+      "eval_iou_steel": 0.8562925829931867,
+      "eval_iou_tiles": 0.25076660988074956,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1046426072716713,
+      "eval_mean_accuracy": 0.7097231423049762,
+      "eval_mean_iou": 0.6703527884136462,
+      "eval_overall_accuracy": 0.9749694400363498,
+      "eval_runtime": 3.3903,
+      "eval_samples_per_second": 10.618,
+      "eval_steps_per_second": 0.59,
+      "step": 5100
+    },
+    {
+      "epoch": 463.73,
+      "learning_rate": 7.636363636363638e-06,
+      "loss": 0.0264,
+      "step": 5101
+    },
+    {
+      "epoch": 463.82,
+      "learning_rate": 7.617224880382775e-06,
+      "loss": 0.0297,
+      "step": 5102
+    },
+    {
+      "epoch": 463.91,
+      "learning_rate": 7.598086124401914e-06,
+      "loss": 0.0318,
+      "step": 5103
+    },
+    {
+      "epoch": 464.0,
+      "learning_rate": 7.578947368421054e-06,
+      "loss": 0.0286,
+      "step": 5104
+    },
+    {
+      "epoch": 464.09,
+      "learning_rate": 7.5598086124401915e-06,
+      "loss": 0.0295,
+      "step": 5105
+    },
+    {
+      "epoch": 464.18,
+      "learning_rate": 7.54066985645933e-06,
+      "loss": 0.0337,
+      "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.0289,
+      "step": 5108
+    },
+    {
+      "epoch": 464.45,
+      "learning_rate": 7.483253588516747e-06,
+      "loss": 0.0289,
+      "step": 5109
+    },
+    {
+      "epoch": 464.55,
+      "learning_rate": 7.464114832535886e-06,
+      "loss": 0.0237,
+      "step": 5110
+    },
+    {
+      "epoch": 464.64,
+      "learning_rate": 7.444976076555024e-06,
+      "loss": 0.0294,
+      "step": 5111
+    },
+    {
+      "epoch": 464.73,
+      "learning_rate": 7.425837320574162e-06,
+      "loss": 0.028,
+      "step": 5112
+    },
+    {
+      "epoch": 464.82,
+      "learning_rate": 7.406698564593302e-06,
+      "loss": 0.0328,
+      "step": 5113
+    },
+    {
+      "epoch": 464.91,
+      "learning_rate": 7.3875598086124405e-06,
+      "loss": 0.0306,
+      "step": 5114
+    },
+    {
+      "epoch": 465.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0311,
+      "step": 5115
+    },
+    {
+      "epoch": 465.09,
+      "learning_rate": 7.349282296650718e-06,
+      "loss": 0.0343,
+      "step": 5116
+    },
+    {
+      "epoch": 465.18,
+      "learning_rate": 7.330143540669857e-06,
+      "loss": 0.0281,
+      "step": 5117
+    },
+    {
+      "epoch": 465.27,
+      "learning_rate": 7.3110047846889965e-06,
+      "loss": 0.0294,
+      "step": 5118
+    },
+    {
+      "epoch": 465.36,
+      "learning_rate": 7.291866028708134e-06,
+      "loss": 0.0297,
+      "step": 5119
+    },
+    {
+      "epoch": 465.45,
+      "learning_rate": 7.272727272727272e-06,
+      "loss": 0.029,
+      "step": 5120
+    },
+    {
+      "epoch": 465.45,
+      "eval_accuracy_NAT": 0.993097761682299,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9517790463110005,
+      "eval_accuracy_concrete tiles": 0.9864245092643552,
+      "eval_accuracy_dark bricks": 0.8792579192486817,
+      "eval_accuracy_grass": 0.902606276453339,
+      "eval_accuracy_red bricks": 0.9263139204545454,
+      "eval_accuracy_road": 0.9360280149506582,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.47415432966669424,
+      "eval_accuracy_speedway bricks": 0.9886692082203452,
+      "eval_accuracy_steel": 0.9101477475100587,
+      "eval_accuracy_tiles": 0.24318586495000685,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9806891444502527,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8907455432345787,
+      "eval_iou_concrete tiles": 0.9512044107674618,
+      "eval_iou_dark bricks": 0.8325072886297377,
+      "eval_iou_grass": 0.8349126295200685,
+      "eval_iou_red bricks": 0.8673316708229426,
+      "eval_iou_road": 0.9284888365387153,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3212004631531767,
+      "eval_iou_speedway bricks": 0.9775065789512933,
+      "eval_iou_steel": 0.856201039323664,
+      "eval_iou_tiles": 0.2421741799086135,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1045546680688858,
+      "eval_mean_accuracy": 0.7070511229778448,
+      "eval_mean_iou": 0.667920137330808,
+      "eval_overall_accuracy": 0.9748738606770834,
+      "eval_runtime": 3.5549,
+      "eval_samples_per_second": 10.127,
+      "eval_steps_per_second": 0.563,
+      "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:695ba03430453178feafe3eeb735745dbc23fe66a8c4c51290fd7b5f62c2abad
+oid sha256:82670e9b397007cd8e7075fc9fea31bd51f2f76e57dec7d59e6173c06fea6d8e
 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:ce702e76f8312c9c6c78e5634506a0176daa5208dcc8bf8c952dbe422607682c
-size 1326090
+oid sha256:a9437db7cc45d20540b96cad0d53c702c3719a21bb66638ccf9ee62ae66b3751
+size 1347066