Training in progress, step 200

README.md

@@ -3,6 +3,8 @@ license: other
 base_model: nvidia/mit-b0
 tags:
 - generated_from_trainer
+datasets:
+- scene_parse_150
 model-index:
 - name: Segments
   results: []
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Segments
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the None dataset.
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the scene_parse_150 dataset.
 
 ## Model description
 

all_results.json

@@ -1,85 +1,8 @@
 {
-    "epoch": 19.81,
-    "eval_accuracy_construction-bridge": 0.0,
-    "eval_accuracy_construction-building": 0.8590146051251705,
-    "eval_accuracy_construction-door": 0.0,
-    "eval_accuracy_construction-fenceguardrail": 0.0,
-    "eval_accuracy_construction-stairs": 0.0,
-    "eval_accuracy_construction-tunnel": NaN,
-    "eval_accuracy_construction-wall": 0.001317080924782568,
-    "eval_accuracy_flat-crosswalk": 0.5587614814704475,
-    "eval_accuracy_flat-curb": 0.1276987072001905,
-    "eval_accuracy_flat-cyclinglane": 0.5549554426151009,
-    "eval_accuracy_flat-parkingdriveway": 0.11587664148520291,
-    "eval_accuracy_flat-railtrack": NaN,
-    "eval_accuracy_flat-road": 0.7259208131761419,
-    "eval_accuracy_flat-sidewalk": 0.9517671543102283,
-    "eval_accuracy_human-person": 0.09899381037890112,
-    "eval_accuracy_human-rider": 0.0,
-    "eval_accuracy_nature-terrain": 0.7756278458254022,
-    "eval_accuracy_nature-vegetation": 0.9205963171170912,
-    "eval_accuracy_object-pole": 0.008821574806525607,
-    "eval_accuracy_object-trafficlight": 0.0,
-    "eval_accuracy_object-trafficsign": 0.0,
-    "eval_accuracy_sky": 0.8390657685487607,
-    "eval_accuracy_unlabeled": NaN,
-    "eval_accuracy_vehicle-bicycle": 0.0,
-    "eval_accuracy_vehicle-bus": 0.0,
-    "eval_accuracy_vehicle-car": 0.9049240346951853,
-    "eval_accuracy_vehicle-caravan": 0.0,
-    "eval_accuracy_vehicle-cartrailer": 0.0,
-    "eval_accuracy_vehicle-motorcycle": 0.0,
-    "eval_accuracy_vehicle-tramtrain": NaN,
-    "eval_accuracy_vehicle-truck": 0.0,
-    "eval_accuracy_void-dynamic": 0.0,
-    "eval_accuracy_void-ground": 0.0,
-    "eval_accuracy_void-static": 0.0,
-    "eval_accuracy_void-unclear": 0.0,
-    "eval_iou_construction-bridge": 0.0,
-    "eval_iou_construction-building": 0.49966273102422204,
-    "eval_iou_construction-door": 0.0,
-    "eval_iou_construction-fenceguardrail": 0.0,
-    "eval_iou_construction-stairs": 0.0,
-    "eval_iou_construction-tunnel": NaN,
-    "eval_iou_construction-wall": 0.0013105616961982412,
-    "eval_iou_flat-crosswalk": 0.5497595515292523,
-    "eval_iou_flat-curb": 0.107839451025818,
-    "eval_iou_flat-cyclinglane": 0.4665512602202192,
-    "eval_iou_flat-parkingdriveway": 0.10007739238668793,
-    "eval_iou_flat-railtrack": NaN,
-    "eval_iou_flat-road": 0.5950889830014944,
-    "eval_iou_flat-sidewalk": 0.7821736374605205,
-    "eval_iou_human-person": 0.09788031780354503,
-    "eval_iou_human-rider": 0.0,
-    "eval_iou_nature-terrain": 0.603375343788221,
-    "eval_iou_nature-vegetation": 0.7404922773118722,
-    "eval_iou_object-pole": 0.008814276343092762,
-    "eval_iou_object-trafficlight": 0.0,
-    "eval_iou_object-trafficsign": 0.0,
-    "eval_iou_sky": 0.805219699788265,
-    "eval_iou_unlabeled": NaN,
-    "eval_iou_vehicle-bicycle": 0.0,
-    "eval_iou_vehicle-bus": 0.0,
-    "eval_iou_vehicle-car": 0.6265363619850278,
-    "eval_iou_vehicle-caravan": 0.0,
-    "eval_iou_vehicle-cartrailer": 0.0,
-    "eval_iou_vehicle-motorcycle": 0.0,
-    "eval_iou_vehicle-tramtrain": NaN,
-    "eval_iou_vehicle-truck": 0.0,
-    "eval_iou_void-dynamic": 0.0,
-    "eval_iou_void-ground": 0.0,
-    "eval_iou_void-static": 0.0,
-    "eval_iou_void-unclear": 0.0,
-    "eval_loss": 0.9830149412155151,
-    "eval_mean_accuracy": 0.24010778315093972,
-    "eval_mean_iou": 0.19305747888272376,
-    "eval_overall_accuracy": 0.7586429852069163,
-    "eval_runtime": 153.8717,
-    "eval_samples_per_second": 0.975,
-    "eval_steps_per_second": 0.123,
-    "total_flos": 2.9595008832425165e+17,
-    "train_loss": 1.403595557302799,
-    "train_runtime": 1785.2215,
-    "train_samples_per_second": 9.523,
-    "train_steps_per_second": 0.594
+    "epoch": 20.0,
+    "total_flos": 1.9265685892890624e+17,
+    "train_loss": 2.621326805563534,
+    "train_runtime": 1247.0654,
+    "train_samples_per_second": 8.724,
+    "train_steps_per_second": 0.545
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7affa7c930dc24183fa2ec9b2178e658932b30fc87ffdd44271567be7d7b6c78
-size 14918708
+oid sha256:4af59b6f0d82b58c95f0705c743eeca2e3529dec94a8fd70ef9b0a6cb5c05198
+size 15036944

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebf0adc6f392ca0ff5083b942a5ba3fc70868207e0ecf52e0c731c97e0128bfb
+oid sha256:5ab63c4274fa9bf67664b814d0a6c66ef06bde4fac2510d81d49a1cdd290f0c6
 size 14965709

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 19.81,
-    "total_flos": 2.9595008832425165e+17,
-    "train_loss": 1.403595557302799,
-    "train_runtime": 1785.2215,
-    "train_samples_per_second": 9.523,
-    "train_steps_per_second": 0.594
+    "epoch": 20.0,
+    "total_flos": 1.9265685892890624e+17,
+    "train_loss": 2.621326805563534,
+    "train_runtime": 1247.0654,
+    "train_samples_per_second": 8.724,
+    "train_steps_per_second": 0.545
 }

trainer_state.json

@@ -1,40 +1,34 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 19.813084112149532,
+  "epoch": 20.0,
   "eval_steps": 500,
-  "global_step": 1060,
+  "global_step": 680,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 9.35,
-      "learning_rate": 2.641509433962264e-05,
-      "loss": 1.7773,
+      "epoch": 14.71,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 2.8344,
       "step": 500
     },
     {
-      "epoch": 18.69,
-      "learning_rate": 2.830188679245283e-06,
-      "loss": 1.082,
-      "step": 1000
-    },
-    {
-      "epoch": 19.81,
-      "step": 1060,
-      "total_flos": 2.9595008832425165e+17,
-      "train_loss": 1.403595557302799,
-      "train_runtime": 1785.2215,
-      "train_samples_per_second": 9.523,
-      "train_steps_per_second": 0.594
+      "epoch": 20.0,
+      "step": 680,
+      "total_flos": 1.9265685892890624e+17,
+      "train_loss": 2.621326805563534,
+      "train_runtime": 1247.0654,
+      "train_samples_per_second": 8.724,
+      "train_steps_per_second": 0.545
     }
   ],
   "logging_steps": 500,
-  "max_steps": 1060,
+  "max_steps": 680,
   "num_train_epochs": 20,
   "save_steps": 200,
-  "total_flos": 2.9595008832425165e+17,
+  "total_flos": 1.9265685892890624e+17,
   "trial_name": null,
   "trial_params": null
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:04ea4bc39abf7080260339e20083b740070e21d79f50349cfe0a327b56db4893
+oid sha256:025b3ecf0049ec68046c1579f6adc8e4df0557674cbfe14991ccd8eb9a82af12
 size 4219