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: 1.3138
 - Mean Iou: 0.0868

all_results.json

@@ -0,0 +1,46 @@
+{
+    "epoch": 1.9,
+    "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_helicopter": NaN,
+    "eval_accuracy_large-vehicle": 0.2588257343134778,
+    "eval_accuracy_plane": 0.0,
+    "eval_accuracy_roundabout": 0.0,
+    "eval_accuracy_ship": 0.1352787805856398,
+    "eval_accuracy_small-vehicle": 0.00013013386665388158,
+    "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_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_helicopter": NaN,
+    "eval_iou_large-vehicle": 0.2563089613847982,
+    "eval_iou_plane": 0.0,
+    "eval_iou_roundabout": 0.0,
+    "eval_iou_ship": 0.053192957552098734,
+    "eval_iou_small-vehicle": 0.00012985237834882448,
+    "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_results.json

@@ -0,0 +1,42 @@
+{
+    "epoch": 1.9,
+    "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_helicopter": NaN,
+    "eval_accuracy_large-vehicle": 0.2588257343134778,
+    "eval_accuracy_plane": 0.0,
+    "eval_accuracy_roundabout": 0.0,
+    "eval_accuracy_ship": 0.1352787805856398,
+    "eval_accuracy_small-vehicle": 0.00013013386665388158,
+    "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_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_helicopter": NaN,
+    "eval_iou_large-vehicle": 0.2563089613847982,
+    "eval_iou_plane": 0.0,
+    "eval_iou_roundabout": 0.0,
+    "eval_iou_ship": 0.053192957552098734,
+    "eval_iou_small-vehicle": 0.00012985237834882448,
+    "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_results.json

@@ -0,0 +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
+}

trainer_state.json

@@ -0,0 +1,126 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.9047619047619047,
+  "eval_steps": 500,
+  "global_step": 200,
+  "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,
+      "step": 100
+    },
+    {
+      "epoch": 1.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.47334601346835964,
+      "eval_accuracy_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.35059128278327106,
+      "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_storage-tank": 0.0,
+      "eval_accuracy_swimming-pool": 0.0,
+      "eval_accuracy_tennis-court": 0.8373677834210894,
+      "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_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.34602565881179354,
+      "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_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,
+      "step": 105
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 1e-07,
+      "loss": 1.5141,
+      "step": 200
+    },
+    {
+      "epoch": 1.9,
+      "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_helicopter": NaN,
+      "eval_accuracy_large-vehicle": 0.2588257343134778,
+      "eval_accuracy_plane": 0.0,
+      "eval_accuracy_roundabout": 0.0,
+      "eval_accuracy_ship": 0.1352787805856398,
+      "eval_accuracy_small-vehicle": 0.00013013386665388158,
+      "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_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_helicopter": NaN,
+      "eval_iou_large-vehicle": 0.2563089613847982,
+      "eval_iou_plane": 0.0,
+      "eval_iou_roundabout": 0.0,
+      "eval_iou_ship": 0.053192957552098734,
+      "eval_iou_small-vehicle": 0.00012985237834882448,
+      "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
+    },
+    {
+      "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
+    }
+  ],
+  "logging_steps": 100,
+  "max_steps": 200,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "total_flos": 2.80718928248832e+16,
+  "trial_name": null,
+  "trial_params": null
+}