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
 model-index:
 - name: phaseseg_models
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # phaseseg_models
 
-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 TommyClas/phase_seg dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.0372
 - Mean Iou: 0.9744

all_results.json

@@ -0,0 +1,25 @@
+{
+    "epoch": 200.0,
+    "eval_accuracy_\u5176\u4ed6\u6c34\u5316\u7269": 0.9886675467942931,
+    "eval_accuracy_\u5b54\u9699": 0.989316704492472,
+    "eval_accuracy_\u672a\u6c34\u5316\u6c34\u6ce5\u9897\u7c92": 0.9806369082847917,
+    "eval_accuracy_\u6c22\u6c27\u5316\u9499": 0.9900781566029906,
+    "eval_accuracy_\u80cc\u666f": NaN,
+    "eval_iou_\u5176\u4ed6\u6c34\u5316\u7269": 0.9782279382295338,
+    "eval_iou_\u5b54\u9699": 0.9695048668829725,
+    "eval_iou_\u672a\u6c34\u5316\u6c34\u6ce5\u9897\u7c92": 0.9730225146640659,
+    "eval_iou_\u6c22\u6c27\u5316\u9499": 0.9767085912455425,
+    "eval_iou_\u80cc\u666f": NaN,
+    "eval_loss": 0.037199750542640686,
+    "eval_mean_accuracy": 0.9871748290436368,
+    "eval_mean_iou": 0.9743659777555287,
+    "eval_overall_accuracy": 0.9869468669047343,
+    "eval_runtime": 12.9029,
+    "eval_samples_per_second": 7.75,
+    "eval_steps_per_second": 1.008,
+    "total_flos": 1.40252748447744e+18,
+    "train_loss": 0.0196058931350708,
+    "train_runtime": 6623.9714,
+    "train_samples_per_second": 12.077,
+    "train_steps_per_second": 1.51
+}

eval_results.json

@@ -0,0 +1,20 @@
+{
+    "epoch": 200.0,
+    "eval_accuracy_\u5176\u4ed6\u6c34\u5316\u7269": 0.9886675467942931,
+    "eval_accuracy_\u5b54\u9699": 0.989316704492472,
+    "eval_accuracy_\u672a\u6c34\u5316\u6c34\u6ce5\u9897\u7c92": 0.9806369082847917,
+    "eval_accuracy_\u6c22\u6c27\u5316\u9499": 0.9900781566029906,
+    "eval_accuracy_\u80cc\u666f": NaN,
+    "eval_iou_\u5176\u4ed6\u6c34\u5316\u7269": 0.9782279382295338,
+    "eval_iou_\u5b54\u9699": 0.9695048668829725,
+    "eval_iou_\u672a\u6c34\u5316\u6c34\u6ce5\u9897\u7c92": 0.9730225146640659,
+    "eval_iou_\u6c22\u6c27\u5316\u9499": 0.9767085912455425,
+    "eval_iou_\u80cc\u666f": NaN,
+    "eval_loss": 0.037199750542640686,
+    "eval_mean_accuracy": 0.9871748290436368,
+    "eval_mean_iou": 0.9743659777555287,
+    "eval_overall_accuracy": 0.9869468669047343,
+    "eval_runtime": 12.9029,
+    "eval_samples_per_second": 7.75,
+    "eval_steps_per_second": 1.008
+}

runs/Apr19_08-14-58_91e999812272/events.out.tfevents.1745057168.91e999812272.8247.1

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 200.0,
+    "total_flos": 1.40252748447744e+18,
+    "train_loss": 0.0196058931350708,
+    "train_runtime": 6623.9714,
+    "train_samples_per_second": 12.077,
+    "train_steps_per_second": 1.51
+}