update model card README.md

README.md

@@ -1,8 +1,6 @@
 ---
 license: apache-2.0
 tags:
-- image-segmentation
-- vision
 - generated_from_trainer
 model-index:
 - name: segformer-finetuned-sidewalk-50-epochs
@@ -14,82 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # segformer-finetuned-sidewalk-50-epochs
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the segments/sidewalk-semantic dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.7690
-- Mean Iou: 0.2117
-- Mean Accuracy: 0.2615
-- Overall Accuracy: 0.7840
-- Accuracy Unlabeled: nan
-- Accuracy Flat-road: 0.6563
-- Accuracy Flat-sidewalk: 0.9295
-- Accuracy Flat-crosswalk: 0.3228
-- Accuracy Flat-cyclinglane: 0.7303
-- Accuracy Flat-parkingdriveway: 0.2743
-- Accuracy Flat-railtrack: nan
-- Accuracy Flat-curb: 0.3172
-- Accuracy Human-person: 0.0109
-- Accuracy Human-rider: 0.0
-- Accuracy Vehicle-car: 0.9388
-- Accuracy Vehicle-truck: 0.0
-- Accuracy Vehicle-bus: 0.0
-- Accuracy Vehicle-tramtrain: 0.0
-- Accuracy Vehicle-motorcycle: 0.0
-- Accuracy Vehicle-bicycle: 0.0
-- Accuracy Vehicle-caravan: 0.0
-- Accuracy Vehicle-cartrailer: 0.0
-- Accuracy Construction-building: 0.9232
-- Accuracy Construction-door: 0.0
-- Accuracy Construction-wall: 0.2146
-- Accuracy Construction-fenceguardrail: 0.0943
-- Accuracy Construction-bridge: 0.0
-- Accuracy Construction-tunnel: nan
-- Accuracy Construction-stairs: 0.0
-- Accuracy Object-pole: 0.2119
-- Accuracy Object-trafficsign: 0.0
-- Accuracy Object-trafficlight: 0.0
-- Accuracy Nature-vegetation: 0.9083
-- Accuracy Nature-terrain: 0.7748
-- Accuracy Sky: 0.8846
-- Accuracy Void-ground: 0.0
-- Accuracy Void-dynamic: 0.0
-- Accuracy Void-static: 0.1757
-- Accuracy Void-unclear: 0.0
-- Iou Unlabeled: nan
-- Iou Flat-road: 0.5342
-- Iou Flat-sidewalk: 0.8036
-- Iou Flat-crosswalk: 0.2661
-- Iou Flat-cyclinglane: 0.6165
-- Iou Flat-parkingdriveway: 0.2254
-- Iou Flat-railtrack: nan
-- Iou Flat-curb: 0.1921
-- Iou Human-person: 0.0109
-- Iou Human-rider: 0.0
-- Iou Vehicle-car: 0.7190
-- Iou Vehicle-truck: 0.0
-- Iou Vehicle-bus: 0.0
-- Iou Vehicle-tramtrain: 0.0
-- Iou Vehicle-motorcycle: 0.0
-- Iou Vehicle-bicycle: 0.0
-- Iou Vehicle-caravan: 0.0
-- Iou Vehicle-cartrailer: 0.0
-- Iou Construction-building: 0.6152
-- Iou Construction-door: 0.0
-- Iou Construction-wall: 0.1796
-- Iou Construction-fenceguardrail: 0.0924
-- Iou Construction-bridge: 0.0
-- Iou Construction-tunnel: nan
-- Iou Construction-stairs: 0.0
-- Iou Object-pole: 0.1438
-- Iou Object-trafficsign: 0.0
-- Iou Object-trafficlight: 0.0
-- Iou Nature-vegetation: 0.7901
-- Iou Nature-terrain: 0.6467
-- Iou Sky: 0.8389
-- Iou Void-ground: 0.0
-- Iou Void-dynamic: 0.0
-- Iou Void-static: 0.0992
-- Iou Void-unclear: 0.0
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the None dataset.
 
 ## Model description
 
@@ -108,13 +31,13 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5e-05
+- learning_rate: 6e-05
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- num_epochs: 50.0
+- lr_scheduler_type: polynomial
+- training_steps: 1000
 
 ### Training results