End of training

README.md

@@ -3,6 +3,8 @@ library_name: transformers
 license: other
 base_model: nvidia/mit-b1
 tags:
+- image-segmentation
+- vision
 - generated_from_trainer
 model-index:
 - name: segformer-finetuned-tt-2k-b1
@@ -14,16 +16,16 @@ should probably proofread and complete it, then remove this comment. -->
 
 # segformer-finetuned-tt-2k-b1
 
-This model is a fine-tuned version of [nvidia/mit-b1](https://huggingface.co/nvidia/mit-b1) on an unknown dataset.
+This model is a fine-tuned version of [nvidia/mit-b1](https://huggingface.co/nvidia/mit-b1) on the Saumya-Mundra/text255 dataset.
 It achieves the following results on the evaluation set:
-- Accuracy No Text: 0.9793
-- Accuracy Text: nan
-- Iou No Text: 0.9793
-- Iou Text: 0.0
 - Loss: 0.0929
-- Mean Accuracy: 0.9793
 - Mean Iou: 0.4897
+- Mean Accuracy: 0.9793
 - Overall Accuracy: 0.9793
+- Accuracy Text: nan
+- Accuracy No Text: 0.9793
+- Iou Text: 0.0
+- Iou No Text: 0.9793
 
 ## Model description
 

all_results.json

@@ -8,12 +8,12 @@
     "eval_mean_accuracy": 0.9793073981417643,
     "eval_mean_iou": 0.48965369907088213,
     "eval_overall_accuracy": 0.9793073981417643,
-    "eval_runtime": 10.2025,
-    "eval_samples_per_second": 24.896,
-    "eval_steps_per_second": 3.137,
+    "eval_runtime": 10.192,
+    "eval_samples_per_second": 24.922,
+    "eval_steps_per_second": 3.14,
     "total_flos": 1.032636453617664e+18,
-    "train_loss": 0.11973714399337769,
-    "train_runtime": 853.4119,
-    "train_samples_per_second": 18.748,
-    "train_steps_per_second": 2.344
+    "train_loss": 0.0,
+    "train_runtime": 0.0014,
+    "train_samples_per_second": 11544617.925,
+    "train_steps_per_second": 1443077.241
 }

eval_results.json

@@ -8,7 +8,7 @@
     "eval_mean_accuracy": 0.9793073981417643,
     "eval_mean_iou": 0.48965369907088213,
     "eval_overall_accuracy": 0.9793073981417643,
-    "eval_runtime": 10.2025,
-    "eval_samples_per_second": 24.896,
-    "eval_steps_per_second": 3.137
+    "eval_runtime": 10.192,
+    "eval_samples_per_second": 24.922,
+    "eval_steps_per_second": 3.14
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 16.0,
     "total_flos": 1.032636453617664e+18,
-    "train_loss": 0.11973714399337769,
-    "train_runtime": 853.4119,
-    "train_samples_per_second": 18.748,
-    "train_steps_per_second": 2.344
+    "train_loss": 0.0,
+    "train_runtime": 0.0014,
+    "train_samples_per_second": 11544617.925,
+    "train_steps_per_second": 1443077.241
 }

trainer_state.json

@@ -392,10 +392,10 @@
       "epoch": 16.0,
       "step": 2000,
       "total_flos": 1.032636453617664e+18,
-      "train_loss": 0.11973714399337769,
-      "train_runtime": 853.4119,
-      "train_samples_per_second": 18.748,
-      "train_steps_per_second": 2.344
+      "train_loss": 0.0,
+      "train_runtime": 0.0014,
+      "train_samples_per_second": 11544617.925,
+      "train_steps_per_second": 1443077.241
     }
   ],
   "logging_steps": 100,