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: segformer-finetuned-biofilm-randaug
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # segformer-finetuned-biofilm-randaug
 
-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 heroza/biofilm dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.4306
 - Mean Iou: 0.3057

all_results.json

@@ -0,0 +1,18 @@
+{
+    "epoch": 588.24,
+    "eval_accuracy_background": NaN,
+    "eval_accuracy_biofilm": 0.6114226490112141,
+    "eval_iou_background": 0.0,
+    "eval_iou_biofilm": 0.6114226490112141,
+    "eval_loss": 0.4306454062461853,
+    "eval_mean_accuracy": 0.6114226490112141,
+    "eval_mean_iou": 0.30571132450560706,
+    "eval_overall_accuracy": 0.6114226490112141,
+    "eval_runtime": 32.8324,
+    "eval_samples_per_second": 1.005,
+    "eval_steps_per_second": 0.152,
+    "train_loss": 0.03221149683594704,
+    "train_runtime": 26753.4257,
+    "train_samples_per_second": 2.99,
+    "train_steps_per_second": 0.374
+}

eval_results.json

@@ -0,0 +1,14 @@
+{
+    "epoch": 588.24,
+    "eval_accuracy_background": NaN,
+    "eval_accuracy_biofilm": 0.6114226490112141,
+    "eval_iou_background": 0.0,
+    "eval_iou_biofilm": 0.6114226490112141,
+    "eval_loss": 0.4306454062461853,
+    "eval_mean_accuracy": 0.6114226490112141,
+    "eval_mean_iou": 0.30571132450560706,
+    "eval_overall_accuracy": 0.6114226490112141,
+    "eval_runtime": 32.8324,
+    "eval_samples_per_second": 1.005,
+    "eval_steps_per_second": 0.152
+}

runs/Feb15_11-01-00_gpu-1-164.local/events.out.tfevents.1708021673.gpu-1-164.local.3266.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b9e95516abbdb94a13bd7856105b6100c24a7f3bf3494501beb38e721fd6cc5
+size 764

train_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 588.24,
+    "train_loss": 0.03221149683594704,
+    "train_runtime": 26753.4257,
+    "train_samples_per_second": 2.99,
+    "train_steps_per_second": 0.374
+}