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-rwymarkings-2-steps
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # segformer-finetuned-rwymarkings-2-steps
 
-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 Spatiallysaying/rwymarkings dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.2162
 - Mean Iou: 0.0387

all_results.json

@@ -0,0 +1,33 @@
+{
+    "epoch": 0.045454545454545456,
+    "eval_accuracy__backgound_": NaN,
+    "eval_accuracy_aim": 0.2144114770133681,
+    "eval_accuracy_arrow": 0.08341416100872939,
+    "eval_accuracy_chevron": 0.0022865095933989462,
+    "eval_accuracy_desig": 0.09223437061956827,
+    "eval_accuracy_disp": 0.27102803738317754,
+    "eval_accuracy_rwythr": 0.17653715579026782,
+    "eval_accuracy_tdz": 0.04933458099396964,
+    "eval_accuracy_thrbar": 0.01395139513951395,
+    "eval_iou__backgound_": 0.0,
+    "eval_iou_aim": 0.11579095647274264,
+    "eval_iou_arrow": 0.008531746031746031,
+    "eval_iou_chevron": 0.0019953153465776005,
+    "eval_iou_desig": 0.04427994616419919,
+    "eval_iou_disp": 0.02664951295717699,
+    "eval_iou_rwythr": 0.09797764099987438,
+    "eval_iou_tdz": 0.039930993856770174,
+    "eval_iou_thrbar": 0.0131286394917946,
+    "eval_loss": 2.2161576747894287,
+    "eval_mean_accuracy": 0.1128997109427492,
+    "eval_mean_iou": 0.038698305702320177,
+    "eval_overall_accuracy": 0.10500469317517107,
+    "eval_runtime": 134.5547,
+    "eval_samples_per_second": 0.647,
+    "eval_steps_per_second": 0.082,
+    "total_flos": 280583108296704.0,
+    "train_loss": 2.258466958999634,
+    "train_runtime": 267.1515,
+    "train_samples_per_second": 0.06,
+    "train_steps_per_second": 0.007
+}

eval_results.json

@@ -0,0 +1,28 @@
+{
+    "epoch": 0.045454545454545456,
+    "eval_accuracy__backgound_": NaN,
+    "eval_accuracy_aim": 0.2144114770133681,
+    "eval_accuracy_arrow": 0.08341416100872939,
+    "eval_accuracy_chevron": 0.0022865095933989462,
+    "eval_accuracy_desig": 0.09223437061956827,
+    "eval_accuracy_disp": 0.27102803738317754,
+    "eval_accuracy_rwythr": 0.17653715579026782,
+    "eval_accuracy_tdz": 0.04933458099396964,
+    "eval_accuracy_thrbar": 0.01395139513951395,
+    "eval_iou__backgound_": 0.0,
+    "eval_iou_aim": 0.11579095647274264,
+    "eval_iou_arrow": 0.008531746031746031,
+    "eval_iou_chevron": 0.0019953153465776005,
+    "eval_iou_desig": 0.04427994616419919,
+    "eval_iou_disp": 0.02664951295717699,
+    "eval_iou_rwythr": 0.09797764099987438,
+    "eval_iou_tdz": 0.039930993856770174,
+    "eval_iou_thrbar": 0.0131286394917946,
+    "eval_loss": 2.2161576747894287,
+    "eval_mean_accuracy": 0.1128997109427492,
+    "eval_mean_iou": 0.038698305702320177,
+    "eval_overall_accuracy": 0.10500469317517107,
+    "eval_runtime": 134.5547,
+    "eval_samples_per_second": 0.647,
+    "eval_steps_per_second": 0.082
+}

runs/Jul12_17-09-28_2ddd51c9d85f/events.out.tfevents.1720804576.2ddd51c9d85f.26734.1

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 0.045454545454545456,
+    "total_flos": 280583108296704.0,
+    "train_loss": 2.258466958999634,
+    "train_runtime": 267.1515,
+    "train_samples_per_second": 0.06,
+    "train_steps_per_second": 0.007
+}

trainer_state.json

@@ -0,0 +1,85 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.045454545454545456,
+  "eval_steps": 500,
+  "global_step": 2,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.022727272727272728,
+      "grad_norm": 5.735886096954346,
+      "learning_rate": 3.0050000000000002e-05,
+      "loss": 2.2694,
+      "step": 1
+    },
+    {
+      "epoch": 0.045454545454545456,
+      "grad_norm": 5.947845935821533,
+      "learning_rate": 1e-07,
+      "loss": 2.2475,
+      "step": 2
+    },
+    {
+      "epoch": 0.045454545454545456,
+      "eval_accuracy__backgound_": NaN,
+      "eval_accuracy_aim": 0.2144114770133681,
+      "eval_accuracy_arrow": 0.08341416100872939,
+      "eval_accuracy_chevron": 0.0022865095933989462,
+      "eval_accuracy_desig": 0.09223437061956827,
+      "eval_accuracy_disp": 0.27102803738317754,
+      "eval_accuracy_rwythr": 0.17653715579026782,
+      "eval_accuracy_tdz": 0.04933458099396964,
+      "eval_accuracy_thrbar": 0.01395139513951395,
+      "eval_iou__backgound_": 0.0,
+      "eval_iou_aim": 0.11579095647274264,
+      "eval_iou_arrow": 0.008531746031746031,
+      "eval_iou_chevron": 0.0019953153465776005,
+      "eval_iou_desig": 0.04427994616419919,
+      "eval_iou_disp": 0.02664951295717699,
+      "eval_iou_rwythr": 0.09797764099987438,
+      "eval_iou_tdz": 0.039930993856770174,
+      "eval_iou_thrbar": 0.0131286394917946,
+      "eval_loss": 2.2161576747894287,
+      "eval_mean_accuracy": 0.1128997109427492,
+      "eval_mean_iou": 0.038698305702320177,
+      "eval_overall_accuracy": 0.10500469317517107,
+      "eval_runtime": 192.9187,
+      "eval_samples_per_second": 0.451,
+      "eval_steps_per_second": 0.057,
+      "step": 2
+    },
+    {
+      "epoch": 0.045454545454545456,
+      "step": 2,
+      "total_flos": 280583108296704.0,
+      "train_loss": 2.258466958999634,
+      "train_runtime": 267.1515,
+      "train_samples_per_second": 0.06,
+      "train_steps_per_second": 0.007
+    }
+  ],
+  "logging_steps": 1.0,
+  "max_steps": 2,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 280583108296704.0,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}