End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: hustvl/yolos-small
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: hustvl-yolos-small-finetuned-10k-cppe5-manual-pad
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/qubvel-hf-co/transformers-detection-model-finetuning-cppe5/runs/t4jytpie)
 # hustvl-yolos-small-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small) on an unknown dataset.
+This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.8651
 - Map: 0.2893
@@ -59,7 +61,7 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
 - train_batch_size: 8
-- eval_batch_size: 8
+- eval_batch_size: 1
 - seed: 1337
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear

all_results.json

@@ -26,35 +26,35 @@
     "eval_runtime": 7.7617,
     "eval_samples_per_second": 19.326,
     "eval_steps_per_second": 2.448,
-    "test_loss": 1.587841272354126,
-    "test_map": 0.3942,
-    "test_map_50": 0.7048,
-    "test_map_75": 0.3809,
-    "test_map_Coverall": 0.5521,
-    "test_map_Face_Shield": 0.575,
-    "test_map_Gloves": 0.3106,
-    "test_map_Goggles": 0.1884,
-    "test_map_Mask": 0.3449,
-    "test_map_large": 0.6081,
-    "test_map_medium": 0.2404,
-    "test_map_small": 0.422,
-    "test_mar_1": 0.3136,
-    "test_mar_10": 0.5138,
-    "test_mar_100": 0.5274,
-    "test_mar_100_Coverall": 0.6711,
-    "test_mar_100_Face_Shield": 0.7118,
-    "test_mar_100_Gloves": 0.4459,
-    "test_mar_100_Goggles": 0.3313,
-    "test_mar_100_Mask": 0.4769,
-    "test_mar_large": 0.7511,
-    "test_mar_medium": 0.4155,
-    "test_mar_small": 0.4541,
-    "test_runtime": 2.2158,
-    "test_samples_per_second": 13.088,
-    "test_steps_per_second": 1.805,
+    "test_loss": 1.8185374736785889,
+    "test_map": 0.3091,
+    "test_map_50": 0.6764,
+    "test_map_75": 0.246,
+    "test_map_Coverall": 0.5322,
+    "test_map_Face_Shield": 0.3981,
+    "test_map_Gloves": 0.2208,
+    "test_map_Goggles": 0.1491,
+    "test_map_Mask": 0.2453,
+    "test_map_large": 0.4896,
+    "test_map_medium": 0.199,
+    "test_map_small": 0.1521,
+    "test_mar_1": 0.2653,
+    "test_mar_10": 0.4428,
+    "test_mar_100": 0.4593,
+    "test_mar_100_Coverall": 0.6733,
+    "test_mar_100_Face_Shield": 0.5235,
+    "test_mar_100_Gloves": 0.3623,
+    "test_mar_100_Goggles": 0.3625,
+    "test_mar_100_Mask": 0.375,
+    "test_mar_large": 0.7234,
+    "test_mar_medium": 0.3232,
+    "test_mar_small": 0.2064,
+    "test_runtime": 4.1843,
+    "test_samples_per_second": 6.931,
+    "test_steps_per_second": 6.931,
     "total_flos": 1.643572308404736e+19,
-    "train_loss": 0.5523391162123635,
-    "train_runtime": 7636.311,
-    "train_samples_per_second": 11.131,
-    "train_steps_per_second": 1.401
+    "train_loss": 0.0,
+    "train_runtime": 3.4832,
+    "train_samples_per_second": 24402.891,
+    "train_steps_per_second": 3071.893
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 100.0,
-    "test_loss": 1.587841272354126,
-    "test_map": 0.3942,
-    "test_map_50": 0.7048,
-    "test_map_75": 0.3809,
-    "test_map_Coverall": 0.5521,
-    "test_map_Face_Shield": 0.575,
-    "test_map_Gloves": 0.3106,
-    "test_map_Goggles": 0.1884,
-    "test_map_Mask": 0.3449,
-    "test_map_large": 0.6081,
-    "test_map_medium": 0.2404,
-    "test_map_small": 0.422,
-    "test_mar_1": 0.3136,
-    "test_mar_10": 0.5138,
-    "test_mar_100": 0.5274,
-    "test_mar_100_Coverall": 0.6711,
-    "test_mar_100_Face_Shield": 0.7118,
-    "test_mar_100_Gloves": 0.4459,
-    "test_mar_100_Goggles": 0.3313,
-    "test_mar_100_Mask": 0.4769,
-    "test_mar_large": 0.7511,
-    "test_mar_medium": 0.4155,
-    "test_mar_small": 0.4541,
-    "test_runtime": 2.2158,
-    "test_samples_per_second": 13.088,
-    "test_steps_per_second": 1.805
+    "test_loss": 1.8185374736785889,
+    "test_map": 0.3091,
+    "test_map_50": 0.6764,
+    "test_map_75": 0.246,
+    "test_map_Coverall": 0.5322,
+    "test_map_Face_Shield": 0.3981,
+    "test_map_Gloves": 0.2208,
+    "test_map_Goggles": 0.1491,
+    "test_map_Mask": 0.2453,
+    "test_map_large": 0.4896,
+    "test_map_medium": 0.199,
+    "test_map_small": 0.1521,
+    "test_mar_1": 0.2653,
+    "test_mar_10": 0.4428,
+    "test_mar_100": 0.4593,
+    "test_mar_100_Coverall": 0.6733,
+    "test_mar_100_Face_Shield": 0.5235,
+    "test_mar_100_Gloves": 0.3623,
+    "test_mar_100_Goggles": 0.3625,
+    "test_mar_100_Mask": 0.375,
+    "test_mar_large": 0.7234,
+    "test_mar_medium": 0.3232,
+    "test_mar_small": 0.2064,
+    "test_runtime": 4.1843,
+    "test_samples_per_second": 6.931,
+    "test_steps_per_second": 6.931
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 100.0,
     "total_flos": 1.643572308404736e+19,
-    "train_loss": 0.5523391162123635,
-    "train_runtime": 7636.311,
-    "train_samples_per_second": 11.131,
-    "train_steps_per_second": 1.401
+    "train_loss": 0.0,
+    "train_runtime": 3.4832,
+    "train_samples_per_second": 24402.891,
+    "train_steps_per_second": 3071.893
 }

trainer_state.json

@@ -3712,10 +3712,10 @@
       "epoch": 100.0,
       "step": 10700,
       "total_flos": 1.643572308404736e+19,
-      "train_loss": 0.5523391162123635,
-      "train_runtime": 7636.311,
-      "train_samples_per_second": 11.131,
-      "train_steps_per_second": 1.401
+      "train_loss": 0.0,
+      "train_runtime": 3.4832,
+      "train_samples_per_second": 24402.891,
+      "train_steps_per_second": 3071.893
     }
   ],
   "logging_steps": 500,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f76ba4d51a61b4c21898db0083d54183015a61701bd55ed34cf6b9d1cb8af7b
+oid sha256:9e95b5421e7717b5f4cd2eb0fcb4ee4a3e4c80b020eb5c6f587ecb2af4700c16
 size 4923