End of training

README.md

@@ -2,6 +2,8 @@
 license: other
 base_model: nvidia/mit-b0
 tags:
+- image-segmentation
+- vision
 - generated_from_trainer
 model-index:
 - name: model1
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # model1
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the giuseppemartino/i-SAID_custom_or_1 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2328
 - Mean Iou: 0.1042

all_results.json

@@ -1,46 +1,46 @@
 {
-    "epoch": 1.9,
+    "epoch": 11.43,
     "eval_accuracy_background": NaN,
     "eval_accuracy_baseball-diamond": 0.0,
     "eval_accuracy_basketball-court": 0.0,
     "eval_accuracy_bridge": 0.0,
-    "eval_accuracy_ground-track-field": 0.0,
-    "eval_accuracy_harbor": 0.5786254048218784,
+    "eval_accuracy_ground-track-field": 0.009750103511372895,
+    "eval_accuracy_harbor": 0.37845062458232664,
     "eval_accuracy_helicopter": NaN,
-    "eval_accuracy_large-vehicle": 0.2588257343134778,
+    "eval_accuracy_large-vehicle": 0.21512753535397483,
     "eval_accuracy_plane": 0.0,
     "eval_accuracy_roundabout": 0.0,
-    "eval_accuracy_ship": 0.1352787805856398,
-    "eval_accuracy_small-vehicle": 0.00013013386665388158,
+    "eval_accuracy_ship": 0.5956177296430004,
+    "eval_accuracy_small-vehicle": 0.04763241977075892,
     "eval_accuracy_soccer-ball-field": 0.0,
     "eval_accuracy_storage-tank": 0.0,
     "eval_accuracy_swimming-pool": 0.0,
-    "eval_accuracy_tennis-court": 0.7306043520805181,
+    "eval_accuracy_tennis-court": 0.5923186244816179,
     "eval_iou_background": 0.0,
     "eval_iou_baseball-diamond": 0.0,
     "eval_iou_basketball-court": 0.0,
     "eval_iou_bridge": 0.0,
-    "eval_iou_ground-track-field": 0.0,
-    "eval_iou_harbor": 0.2867916836526702,
+    "eval_iou_ground-track-field": 0.009739046907519078,
+    "eval_iou_harbor": 0.29934130275676996,
     "eval_iou_helicopter": NaN,
-    "eval_iou_large-vehicle": 0.2563089613847982,
+    "eval_iou_large-vehicle": 0.2123670434628911,
     "eval_iou_plane": 0.0,
     "eval_iou_roundabout": 0.0,
-    "eval_iou_ship": 0.053192957552098734,
-    "eval_iou_small-vehicle": 0.00012985237834882448,
+    "eval_iou_ship": 0.4620560893072465,
+    "eval_iou_small-vehicle": 0.045786725832680446,
     "eval_iou_soccer-ball-field": 0.0,
     "eval_iou_storage-tank": 0.0,
     "eval_iou_swimming-pool": 0.0,
-    "eval_iou_tennis-court": 0.7061565483696631,
-    "eval_loss": 1.3138407468795776,
-    "eval_mean_accuracy": 0.1216760289762977,
-    "eval_mean_iou": 0.08683866688917194,
-    "eval_overall_accuracy": 0.22847871034583,
-    "eval_runtime": 132.83,
-    "eval_samples_per_second": 0.745,
-    "eval_steps_per_second": 0.098,
-    "train_loss": 1.7915446472167968,
-    "train_runtime": 623.7007,
-    "train_samples_per_second": 2.565,
-    "train_steps_per_second": 0.321
+    "eval_iou_tennis-court": 0.5337389825645475,
+    "eval_loss": 0.23282881081104279,
+    "eval_mean_accuracy": 0.13134978838164654,
+    "eval_mean_iou": 0.10420194605544363,
+    "eval_overall_accuracy": 0.20168839376559577,
+    "eval_runtime": 129.5622,
+    "eval_samples_per_second": 0.764,
+    "eval_steps_per_second": 0.1,
+    "train_loss": 0.5983023818333943,
+    "train_runtime": 3623.6202,
+    "train_samples_per_second": 2.649,
+    "train_steps_per_second": 0.331
 }

eval_results.json

@@ -1,42 +1,42 @@
 {
-    "epoch": 1.9,
+    "epoch": 11.43,
     "eval_accuracy_background": NaN,
     "eval_accuracy_baseball-diamond": 0.0,
     "eval_accuracy_basketball-court": 0.0,
     "eval_accuracy_bridge": 0.0,
-    "eval_accuracy_ground-track-field": 0.0,
-    "eval_accuracy_harbor": 0.5786254048218784,
+    "eval_accuracy_ground-track-field": 0.009750103511372895,
+    "eval_accuracy_harbor": 0.37845062458232664,
     "eval_accuracy_helicopter": NaN,
-    "eval_accuracy_large-vehicle": 0.2588257343134778,
+    "eval_accuracy_large-vehicle": 0.21512753535397483,
     "eval_accuracy_plane": 0.0,
     "eval_accuracy_roundabout": 0.0,
-    "eval_accuracy_ship": 0.1352787805856398,
-    "eval_accuracy_small-vehicle": 0.00013013386665388158,
+    "eval_accuracy_ship": 0.5956177296430004,
+    "eval_accuracy_small-vehicle": 0.04763241977075892,
     "eval_accuracy_soccer-ball-field": 0.0,
     "eval_accuracy_storage-tank": 0.0,
     "eval_accuracy_swimming-pool": 0.0,
-    "eval_accuracy_tennis-court": 0.7306043520805181,
+    "eval_accuracy_tennis-court": 0.5923186244816179,
     "eval_iou_background": 0.0,
     "eval_iou_baseball-diamond": 0.0,
     "eval_iou_basketball-court": 0.0,
     "eval_iou_bridge": 0.0,
-    "eval_iou_ground-track-field": 0.0,
-    "eval_iou_harbor": 0.2867916836526702,
+    "eval_iou_ground-track-field": 0.009739046907519078,
+    "eval_iou_harbor": 0.29934130275676996,
     "eval_iou_helicopter": NaN,
-    "eval_iou_large-vehicle": 0.2563089613847982,
+    "eval_iou_large-vehicle": 0.2123670434628911,
     "eval_iou_plane": 0.0,
     "eval_iou_roundabout": 0.0,
-    "eval_iou_ship": 0.053192957552098734,
-    "eval_iou_small-vehicle": 0.00012985237834882448,
+    "eval_iou_ship": 0.4620560893072465,
+    "eval_iou_small-vehicle": 0.045786725832680446,
     "eval_iou_soccer-ball-field": 0.0,
     "eval_iou_storage-tank": 0.0,
     "eval_iou_swimming-pool": 0.0,
-    "eval_iou_tennis-court": 0.7061565483696631,
-    "eval_loss": 1.3138407468795776,
-    "eval_mean_accuracy": 0.1216760289762977,
-    "eval_mean_iou": 0.08683866688917194,
-    "eval_overall_accuracy": 0.22847871034583,
-    "eval_runtime": 132.83,
-    "eval_samples_per_second": 0.745,
-    "eval_steps_per_second": 0.098
+    "eval_iou_tennis-court": 0.5337389825645475,
+    "eval_loss": 0.23282881081104279,
+    "eval_mean_accuracy": 0.13134978838164654,
+    "eval_mean_iou": 0.10420194605544363,
+    "eval_overall_accuracy": 0.20168839376559577,
+    "eval_runtime": 129.5622,
+    "eval_samples_per_second": 0.764,
+    "eval_steps_per_second": 0.1
 }

train_results.json

@@ -1,7 +1,7 @@
 {
-    "epoch": 1.9,
-    "train_loss": 1.7915446472167968,
-    "train_runtime": 623.7007,
-    "train_samples_per_second": 2.565,
-    "train_steps_per_second": 0.321
+    "epoch": 11.43,
+    "train_loss": 0.5983023818333943,
+    "train_runtime": 3623.6202,
+    "train_samples_per_second": 2.649,
+    "train_steps_per_second": 0.331
 }

trainer_state.json

@@ -1,17 +1,17 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 1.9047619047619047,
+  "epoch": 11.428571428571429,
   "eval_steps": 500,
-  "global_step": 200,
+  "global_step": 1200,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.95,
-      "learning_rate": 3.0050000000000002e-05,
-      "loss": 2.069,
+      "learning_rate": 5.500833333333333e-05,
+      "loss": 1.9822,
       "step": 100
     },
     {
@@ -21,106 +21,596 @@
       "eval_accuracy_basketball-court": 0.0,
       "eval_accuracy_bridge": 0.0,
       "eval_accuracy_ground-track-field": 0.0,
-      "eval_accuracy_harbor": 0.47334601346835964,
+      "eval_accuracy_harbor": 0.4963244743741325,
       "eval_accuracy_helicopter": NaN,
-      "eval_accuracy_large-vehicle": 0.35059128278327106,
+      "eval_accuracy_large-vehicle": 0.22959375768701085,
       "eval_accuracy_plane": 0.0,
       "eval_accuracy_roundabout": 0.0,
-      "eval_accuracy_ship": 0.4326614520657842,
-      "eval_accuracy_small-vehicle": 0.0001883516491043023,
-      "eval_accuracy_soccer-ball-field": 0.00015808777033008727,
+      "eval_accuracy_ship": 0.47352587244283995,
+      "eval_accuracy_small-vehicle": 0.0,
+      "eval_accuracy_soccer-ball-field": 0.0,
       "eval_accuracy_storage-tank": 0.0,
       "eval_accuracy_swimming-pool": 0.0,
-      "eval_accuracy_tennis-court": 0.8373677834210894,
+      "eval_accuracy_tennis-court": 0.8168771259493965,
       "eval_iou_background": 0.0,
       "eval_iou_baseball-diamond": 0.0,
       "eval_iou_basketball-court": 0.0,
       "eval_iou_bridge": 0.0,
       "eval_iou_ground-track-field": 0.0,
-      "eval_iou_harbor": 0.22125573683831126,
+      "eval_iou_harbor": 0.3354876819903402,
       "eval_iou_helicopter": NaN,
-      "eval_iou_large-vehicle": 0.34602565881179354,
+      "eval_iou_large-vehicle": 0.22687399857464824,
       "eval_iou_plane": 0.0,
       "eval_iou_roundabout": 0.0,
-      "eval_iou_ship": 0.07936099180546487,
-      "eval_iou_small-vehicle": 0.00018768512578315883,
-      "eval_iou_soccer-ball-field": 0.00015805841048559496,
+      "eval_iou_ship": 0.25260117153021105,
+      "eval_iou_small-vehicle": 0.0,
+      "eval_iou_soccer-ball-field": 0.0,
       "eval_iou_storage-tank": 0.0,
       "eval_iou_swimming-pool": 0.0,
-      "eval_iou_tennis-court": 0.7660002983738624,
-      "eval_loss": 1.4974907636642456,
-      "eval_mean_accuracy": 0.1495937836541385,
-      "eval_mean_iou": 0.09419922862438006,
-      "eval_overall_accuracy": 0.2837216122470019,
-      "eval_runtime": 131.7329,
-      "eval_samples_per_second": 0.752,
-      "eval_steps_per_second": 0.099,
+      "eval_iou_tennis-court": 0.6683037628857792,
+      "eval_loss": 1.289176106452942,
+      "eval_mean_accuracy": 0.14402294503238428,
+      "eval_mean_iou": 0.0988844409987319,
+      "eval_overall_accuracy": 0.23478969675121228,
+      "eval_runtime": 130.3608,
+      "eval_samples_per_second": 0.759,
+      "eval_steps_per_second": 0.1,
       "step": 105
     },
     {
       "epoch": 1.9,
-      "learning_rate": 1e-07,
-      "loss": 1.5141,
+      "learning_rate": 5.001666666666667e-05,
+      "loss": 1.2543,
       "step": 200
     },
     {
-      "epoch": 1.9,
+      "epoch": 2.0,
       "eval_accuracy_background": NaN,
       "eval_accuracy_baseball-diamond": 0.0,
       "eval_accuracy_basketball-court": 0.0,
       "eval_accuracy_bridge": 0.0,
       "eval_accuracy_ground-track-field": 0.0,
-      "eval_accuracy_harbor": 0.5786254048218784,
+      "eval_accuracy_harbor": 0.405490155760037,
       "eval_accuracy_helicopter": NaN,
-      "eval_accuracy_large-vehicle": 0.2588257343134778,
+      "eval_accuracy_large-vehicle": 0.2275305012223916,
       "eval_accuracy_plane": 0.0,
       "eval_accuracy_roundabout": 0.0,
-      "eval_accuracy_ship": 0.1352787805856398,
-      "eval_accuracy_small-vehicle": 0.00013013386665388158,
+      "eval_accuracy_ship": 0.1505214600882471,
+      "eval_accuracy_small-vehicle": 0.0,
       "eval_accuracy_soccer-ball-field": 0.0,
       "eval_accuracy_storage-tank": 0.0,
       "eval_accuracy_swimming-pool": 0.0,
-      "eval_accuracy_tennis-court": 0.7306043520805181,
+      "eval_accuracy_tennis-court": 0.8537929267042542,
       "eval_iou_background": 0.0,
       "eval_iou_baseball-diamond": 0.0,
       "eval_iou_basketball-court": 0.0,
       "eval_iou_bridge": 0.0,
       "eval_iou_ground-track-field": 0.0,
-      "eval_iou_harbor": 0.2867916836526702,
+      "eval_iou_harbor": 0.25194033664441534,
       "eval_iou_helicopter": NaN,
-      "eval_iou_large-vehicle": 0.2563089613847982,
+      "eval_iou_large-vehicle": 0.22483629869700378,
       "eval_iou_plane": 0.0,
       "eval_iou_roundabout": 0.0,
-      "eval_iou_ship": 0.053192957552098734,
-      "eval_iou_small-vehicle": 0.00012985237834882448,
+      "eval_iou_ship": 0.08610848177150561,
+      "eval_iou_small-vehicle": 0.0,
       "eval_iou_soccer-ball-field": 0.0,
       "eval_iou_storage-tank": 0.0,
       "eval_iou_swimming-pool": 0.0,
-      "eval_iou_tennis-court": 0.7061565483696631,
-      "eval_loss": 1.313840627670288,
-      "eval_mean_accuracy": 0.1216760289762977,
-      "eval_mean_iou": 0.08683866688917194,
-      "eval_overall_accuracy": 0.22847871034583,
-      "eval_runtime": 130.8677,
-      "eval_samples_per_second": 0.756,
-      "eval_steps_per_second": 0.099,
-      "step": 200
+      "eval_iou_tennis-court": 0.736334319001517,
+      "eval_loss": 0.8622854948043823,
+      "eval_mean_accuracy": 0.11695250312678071,
+      "eval_mean_iou": 0.08661462907429612,
+      "eval_overall_accuracy": 0.23483131750165362,
+      "eval_runtime": 129.8172,
+      "eval_samples_per_second": 0.763,
+      "eval_steps_per_second": 0.1,
+      "step": 210
     },
     {
-      "epoch": 1.9,
-      "step": 200,
-      "total_flos": 2.80718928248832e+16,
-      "train_loss": 1.7915446472167968,
-      "train_runtime": 623.7007,
-      "train_samples_per_second": 2.565,
-      "train_steps_per_second": 0.321
+      "epoch": 2.86,
+      "learning_rate": 4.5025000000000003e-05,
+      "loss": 0.8713,
+      "step": 300
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0,
+      "eval_accuracy_harbor": 0.26088521050737673,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.19634403672027564,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.00947653429602888,
+      "eval_accuracy_small-vehicle": 0.0,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5983372940061817,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.0,
+      "eval_iou_harbor": 0.18212158185602526,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.19528883474139022,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.009133523413714782,
+      "eval_iou_small-vehicle": 0.0,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5713994356127606,
+      "eval_loss": 0.5621587038040161,
+      "eval_mean_accuracy": 0.07607450539499021,
+      "eval_mean_iou": 0.0638628917082594,
+      "eval_overall_accuracy": 0.17716992984022115,
+      "eval_runtime": 129.7771,
+      "eval_samples_per_second": 0.763,
+      "eval_steps_per_second": 0.1,
+      "step": 315
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 4.003333333333334e-05,
+      "loss": 0.5934,
+      "step": 400
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0,
+      "eval_accuracy_harbor": 0.3259651467639953,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.2754489214339381,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.015593662254312074,
+      "eval_accuracy_small-vehicle": 0.0,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5852049454048429,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.0,
+      "eval_iou_harbor": 0.21494915254237287,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.2705921998931492,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.01374464135767004,
+      "eval_iou_small-vehicle": 0.0,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5480765733985992,
+      "eval_loss": 0.4177829325199127,
+      "eval_mean_accuracy": 0.08587233398979202,
+      "eval_mean_iou": 0.06982417114611943,
+      "eval_overall_accuracy": 0.2061629445976586,
+      "eval_runtime": 129.5042,
+      "eval_samples_per_second": 0.764,
+      "eval_steps_per_second": 0.1,
+      "step": 420
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 3.504166666666667e-05,
+      "loss": 0.4793,
+      "step": 500
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0,
+      "eval_accuracy_harbor": 0.2203773196936205,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.09540845951884123,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.1355796229442439,
+      "eval_accuracy_small-vehicle": 0.0004760159859181458,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.4300902412127425,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.0,
+      "eval_iou_harbor": 0.16730408991570403,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.09505470780546928,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.11770339093718714,
+      "eval_iou_small-vehicle": 0.0004760159859181458,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.3971643921084903,
+      "eval_loss": 0.32397353649139404,
+      "eval_mean_accuracy": 0.0629951185253833,
+      "eval_mean_iou": 0.05184683978351792,
+      "eval_overall_accuracy": 0.11199695720298312,
+      "eval_runtime": 130.4117,
+      "eval_samples_per_second": 0.759,
+      "eval_steps_per_second": 0.1,
+      "step": 525
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 3.0050000000000002e-05,
+      "loss": 0.3711,
+      "step": 600
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 2.6712612359925738e-05,
+      "eval_accuracy_harbor": 0.23220068883976763,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.10020933769969495,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.46074007220216606,
+      "eval_accuracy_small-vehicle": 0.00018150249822778222,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5083445940698632,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 2.6712612359925738e-05,
+      "eval_iou_harbor": 0.1977843944303354,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.09980816698638835,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.3787248073197873,
+      "eval_iou_small-vehicle": 0.00018150249822778222,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.4270271680806855,
+      "eval_loss": 0.2836246192455292,
+      "eval_mean_accuracy": 0.0929787791372914,
+      "eval_mean_iou": 0.07357018346185229,
+      "eval_overall_accuracy": 0.13098434355432245,
+      "eval_runtime": 130.0055,
+      "eval_samples_per_second": 0.762,
+      "eval_steps_per_second": 0.1,
+      "step": 630
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 2.505833333333333e-05,
+      "loss": 0.347,
+      "step": 700
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.00012020675561966583,
+      "eval_accuracy_harbor": 0.3911993008790418,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.22613993664648033,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.528830726032892,
+      "eval_accuracy_small-vehicle": 0.016006465598427434,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5768875324231552,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.00012020515012287637,
+      "eval_iou_harbor": 0.2954995534500835,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.2223430579594738,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.4019895567328582,
+      "eval_iou_small-vehicle": 0.0158949856319396,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5461379543583108,
+      "eval_loss": 0.26471075415611267,
+      "eval_mean_accuracy": 0.12422744059540117,
+      "eval_mean_iou": 0.09879902088551924,
+      "eval_overall_accuracy": 0.19628858163526008,
+      "eval_runtime": 130.1284,
+      "eval_samples_per_second": 0.761,
+      "eval_steps_per_second": 0.1,
+      "step": 735
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 2.006666666666667e-05,
+      "loss": 0.3004,
+      "step": 800
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0006277463904582549,
+      "eval_accuracy_harbor": 0.4247159821107284,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.36401834675091493,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.5257220216606499,
+      "eval_accuracy_small-vehicle": 0.06174851972726681,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.6456246214062719,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.0006277463904582549,
+      "eval_iou_harbor": 0.3104031258218432,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.35574267253855674,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.4010327022375215,
+      "eval_iou_small-vehicle": 0.0589859429539752,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5756658784263353,
+      "eval_loss": 0.2667113244533539,
+      "eval_mean_accuracy": 0.14446123128902072,
+      "eval_mean_iou": 0.11349720455791269,
+      "eval_overall_accuracy": 0.2693381212137123,
+      "eval_runtime": 133.0464,
+      "eval_samples_per_second": 0.744,
+      "eval_steps_per_second": 0.098,
+      "step": 840
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 1.5075000000000002e-05,
+      "loss": 0.2622,
+      "step": 900
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.00012020675561966583,
+      "eval_accuracy_harbor": 0.2554875854623965,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.13435430782180025,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.4917769755314882,
+      "eval_accuracy_small-vehicle": 0.020715256826034992,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5719871705263812,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.00012020675561966583,
+      "eval_iou_harbor": 0.22073192396518032,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.1333791829466018,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.40104677788681714,
+      "eval_iou_small-vehicle": 0.020298657718120807,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.507794186591655,
+      "eval_loss": 0.23994781076908112,
+      "eval_mean_accuracy": 0.1053172502088372,
+      "eval_mean_iou": 0.08555806239093297,
+      "eval_overall_accuracy": 0.15907514850603918,
+      "eval_runtime": 132.2048,
+      "eval_samples_per_second": 0.749,
+      "eval_steps_per_second": 0.098,
+      "step": 945
+    },
+    {
+      "epoch": 9.52,
+      "learning_rate": 1.0083333333333332e-05,
+      "loss": 0.2489,
+      "step": 1000
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0030185251966716085,
+      "eval_accuracy_harbor": 0.42620675474219916,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.18800901607934417,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.5400120336943441,
+      "eval_accuracy_small-vehicle": 0.039057282873355775,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5640697078421323,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.003018081782003686,
+      "eval_iou_harbor": 0.321157421753951,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.18596267637687755,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.42935735927284324,
+      "eval_iou_small-vehicle": 0.03790245393879776,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5256290571199063,
+      "eval_loss": 0.2446403056383133,
+      "eval_mean_accuracy": 0.12574095145914624,
+      "eval_mean_iou": 0.10020180334962529,
+      "eval_overall_accuracy": 0.18461172001913273,
+      "eval_runtime": 130.4966,
+      "eval_samples_per_second": 0.759,
+      "eval_steps_per_second": 0.1,
+      "step": 1050
+    },
+    {
+      "epoch": 10.48,
+      "learning_rate": 5.091666666666669e-06,
+      "loss": 0.242,
+      "step": 1100
+    },
+    {
+      "epoch": 11.0,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.0024041351123933164,
+      "eval_accuracy_harbor": 0.35285046008327764,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.18537179613754373,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.565734055354994,
+      "eval_accuracy_small-vehicle": 0.026071292811473697,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5443440038519485,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.0024041351123933164,
+      "eval_iou_harbor": 0.2829348722176422,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.1833538571542067,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.4500957395883198,
+      "eval_iou_small-vehicle": 0.025655455954707824,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.49170127392109547,
+      "eval_loss": 0.23460666835308075,
+      "eval_mean_accuracy": 0.11976969595368792,
+      "eval_mean_iou": 0.09574302226322434,
+      "eval_overall_accuracy": 0.17730695815705877,
+      "eval_runtime": 130.5312,
+      "eval_samples_per_second": 0.758,
+      "eval_steps_per_second": 0.1,
+      "step": 1155
+    },
+    {
+      "epoch": 11.43,
+      "learning_rate": 1e-07,
+      "loss": 0.2276,
+      "step": 1200
+    },
+    {
+      "epoch": 11.43,
+      "eval_accuracy_background": NaN,
+      "eval_accuracy_baseball-diamond": 0.0,
+      "eval_accuracy_basketball-court": 0.0,
+      "eval_accuracy_bridge": 0.0,
+      "eval_accuracy_ground-track-field": 0.009750103511372895,
+      "eval_accuracy_harbor": 0.37845062458232664,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.21512753535397483,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.5956177296430004,
+      "eval_accuracy_small-vehicle": 0.04763241977075892,
+      "eval_accuracy_soccer-ball-field": 0.0,
+      "eval_accuracy_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.5923186244816179,
+      "eval_iou_background": 0.0,
+      "eval_iou_baseball-diamond": 0.0,
+      "eval_iou_basketball-court": 0.0,
+      "eval_iou_bridge": 0.0,
+      "eval_iou_ground-track-field": 0.009739046907519078,
+      "eval_iou_harbor": 0.29934130275676996,
+      "eval_iou_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.2123670434628911,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.4620560893072465,
+      "eval_iou_small-vehicle": 0.045786725832680446,
+      "eval_iou_soccer-ball-field": 0.0,
+      "eval_iou_storage-tank": 0.0,
+      "eval_iou_swimming-pool": 0.0,
+      "eval_iou_tennis-court": 0.5337389825645475,
+      "eval_loss": 0.23282881081104279,
+      "eval_mean_accuracy": 0.13134978838164654,
+      "eval_mean_iou": 0.10420194605544363,
+      "eval_overall_accuracy": 0.20168839376559577,
+      "eval_runtime": 130.4137,
+      "eval_samples_per_second": 0.759,
+      "eval_steps_per_second": 0.1,
+      "step": 1200
+    },
+    {
+      "epoch": 11.43,
+      "step": 1200,
+      "total_flos": 1.684313569492992e+17,
+      "train_loss": 0.5983023818333943,
+      "train_runtime": 3623.6202,
+      "train_samples_per_second": 2.649,
+      "train_steps_per_second": 0.331
     }
   ],
   "logging_steps": 100,
-  "max_steps": 200,
-  "num_train_epochs": 2,
+  "max_steps": 1200,
+  "num_train_epochs": 12,
   "save_steps": 500,
-  "total_flos": 2.80718928248832e+16,
+  "total_flos": 1.684313569492992e+17,
   "trial_name": null,
   "trial_params": null
 }