End of training

README.md

@@ -3,6 +3,8 @@ library_name: transformers
 license: other
 base_model: nvidia/mit-b0
 tags:
+- image-segmentation
+- vision
 - generated_from_trainer
 datasets:
 - generator
@@ -16,16 +18,16 @@ should probably proofread and complete it, then remove this comment. -->
 
 # autocrop-tekst
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the generator dataset.
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the /mnt/disk1/autocrop-data/datasets/tekst dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0201
-- Mean Iou: 0.4972
-- Mean Accuracy: 0.9944
-- Overall Accuracy: 0.9944
+- Loss: 0.0197
+- Mean Iou: 0.4970
+- Mean Accuracy: 0.9939
+- Overall Accuracy: 0.9939
 - Accuracy Background: nan
-- Accuracy Crop: 0.9944
+- Accuracy Crop: 0.9939
 - Iou Background: 0.0
-- Iou Crop: 0.9944
+- Iou Crop: 0.9939
 
 ## Model description
 

all_results.json

@@ -0,0 +1,19 @@
+{
+    "epoch": 50.0,
+    "eval_accuracy_background": NaN,
+    "eval_accuracy_crop": 0.9939011345509936,
+    "eval_iou_background": 0.0,
+    "eval_iou_crop": 0.9939011345509936,
+    "eval_loss": 0.019662605598568916,
+    "eval_mean_accuracy": 0.9939011345509936,
+    "eval_mean_iou": 0.4969505672754968,
+    "eval_overall_accuracy": 0.9939011345509936,
+    "eval_runtime": 35.7077,
+    "eval_samples_per_second": 24.673,
+    "eval_steps_per_second": 3.109,
+    "total_flos": 4.3705957092950016e+18,
+    "train_loss": 0.02810766745263185,
+    "train_runtime": 4914.0636,
+    "train_samples_per_second": 50.742,
+    "train_steps_per_second": 6.349
+}

eval_results.json

@@ -0,0 +1,14 @@
+{
+    "epoch": 50.0,
+    "eval_accuracy_background": NaN,
+    "eval_accuracy_crop": 0.9939011345509936,
+    "eval_iou_background": 0.0,
+    "eval_iou_crop": 0.9939011345509936,
+    "eval_loss": 0.019662605598568916,
+    "eval_mean_accuracy": 0.9939011345509936,
+    "eval_mean_iou": 0.4969505672754968,
+    "eval_overall_accuracy": 0.9939011345509936,
+    "eval_runtime": 35.7077,
+    "eval_samples_per_second": 24.673,
+    "eval_steps_per_second": 3.109
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 50.0,
+    "total_flos": 4.3705957092950016e+18,
+    "train_loss": 0.02810766745263185,
+    "train_runtime": 4914.0636,
+    "train_samples_per_second": 50.742,
+    "train_steps_per_second": 6.349
+}