End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: SenseTime/deformable-detr
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: sensetime-deformable-detr-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/x3gitxj7)
 # sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on an unknown dataset.
+This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.0442
 - Map: 0.3514
@@ -59,7 +61,7 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
 - train_batch_size: 4
-- eval_batch_size: 8
+- eval_batch_size: 1
 - seed: 1337
 - gradient_accumulation_steps: 2
 - total_train_batch_size: 8

all_results.json

@@ -26,35 +26,35 @@
     "eval_runtime": 15.6632,
     "eval_samples_per_second": 9.577,
     "eval_steps_per_second": 1.213,
-    "test_loss": 2.403155565261841,
-    "test_map": 0.0814,
-    "test_map_50": 0.2065,
-    "test_map_75": 0.0234,
-    "test_map_Coverall": 0.0796,
-    "test_map_Face_Shield": 0.2432,
-    "test_map_Gloves": 0.0278,
-    "test_map_Goggles": 0.0116,
-    "test_map_Mask": 0.0447,
-    "test_map_large": 0.1559,
-    "test_map_medium": 0.0718,
-    "test_map_small": 0.0144,
-    "test_mar_1": 0.1018,
-    "test_mar_10": 0.1983,
-    "test_mar_100": 0.2149,
-    "test_mar_100_Coverall": 0.2822,
-    "test_mar_100_Face_Shield": 0.3941,
-    "test_mar_100_Gloves": 0.1246,
-    "test_mar_100_Goggles": 0.1219,
-    "test_mar_100_Mask": 0.1519,
-    "test_mar_large": 0.3267,
-    "test_mar_medium": 0.1528,
-    "test_mar_small": 0.0427,
-    "test_runtime": 4.5658,
-    "test_samples_per_second": 6.352,
-    "test_steps_per_second": 0.876,
+    "test_loss": 0.7111734747886658,
+    "test_map": 0.5281,
+    "test_map_50": 0.8343,
+    "test_map_75": 0.5628,
+    "test_map_Coverall": 0.6264,
+    "test_map_Face_Shield": 0.612,
+    "test_map_Gloves": 0.4787,
+    "test_map_Goggles": 0.4403,
+    "test_map_Mask": 0.483,
+    "test_map_large": 0.705,
+    "test_map_medium": 0.4585,
+    "test_map_small": 0.5477,
+    "test_mar_1": 0.3712,
+    "test_mar_10": 0.621,
+    "test_mar_100": 0.6349,
+    "test_mar_100_Coverall": 0.7467,
+    "test_mar_100_Face_Shield": 0.7471,
+    "test_mar_100_Gloves": 0.5738,
+    "test_mar_100_Goggles": 0.5281,
+    "test_mar_100_Mask": 0.5788,
+    "test_mar_large": 0.8137,
+    "test_mar_medium": 0.5441,
+    "test_mar_small": 0.5739,
+    "test_runtime": 5.5464,
+    "test_samples_per_second": 5.229,
+    "test_steps_per_second": 5.229,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.0991,
-    "train_samples_per_second": 27427.639,
-    "train_steps_per_second": 3420.388
+    "train_runtime": 3.224,
+    "train_samples_per_second": 26365.049,
+    "train_steps_per_second": 3287.877
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 99.53051643192488,
-    "test_loss": 2.403155565261841,
-    "test_map": 0.0814,
-    "test_map_50": 0.2065,
-    "test_map_75": 0.0234,
-    "test_map_Coverall": 0.0796,
-    "test_map_Face_Shield": 0.2432,
-    "test_map_Gloves": 0.0278,
-    "test_map_Goggles": 0.0116,
-    "test_map_Mask": 0.0447,
-    "test_map_large": 0.1559,
-    "test_map_medium": 0.0718,
-    "test_map_small": 0.0144,
-    "test_mar_1": 0.1018,
-    "test_mar_10": 0.1983,
-    "test_mar_100": 0.2149,
-    "test_mar_100_Coverall": 0.2822,
-    "test_mar_100_Face_Shield": 0.3941,
-    "test_mar_100_Gloves": 0.1246,
-    "test_mar_100_Goggles": 0.1219,
-    "test_mar_100_Mask": 0.1519,
-    "test_mar_large": 0.3267,
-    "test_mar_medium": 0.1528,
-    "test_mar_small": 0.0427,
-    "test_runtime": 4.5658,
-    "test_samples_per_second": 6.352,
-    "test_steps_per_second": 0.876
+    "test_loss": 0.7111734747886658,
+    "test_map": 0.5281,
+    "test_map_50": 0.8343,
+    "test_map_75": 0.5628,
+    "test_map_Coverall": 0.6264,
+    "test_map_Face_Shield": 0.612,
+    "test_map_Gloves": 0.4787,
+    "test_map_Goggles": 0.4403,
+    "test_map_Mask": 0.483,
+    "test_map_large": 0.705,
+    "test_map_medium": 0.4585,
+    "test_map_small": 0.5477,
+    "test_mar_1": 0.3712,
+    "test_mar_10": 0.621,
+    "test_mar_100": 0.6349,
+    "test_mar_100_Coverall": 0.7467,
+    "test_mar_100_Face_Shield": 0.7471,
+    "test_mar_100_Gloves": 0.5738,
+    "test_mar_100_Goggles": 0.5281,
+    "test_mar_100_Mask": 0.5788,
+    "test_mar_large": 0.8137,
+    "test_mar_medium": 0.5441,
+    "test_mar_small": 0.5739,
+    "test_runtime": 5.5464,
+    "test_samples_per_second": 5.229,
+    "test_steps_per_second": 5.229
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 99.53051643192488,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.0991,
-    "train_samples_per_second": 27427.639,
-    "train_steps_per_second": 3420.388
+    "train_runtime": 3.224,
+    "train_samples_per_second": 26365.049,
+    "train_steps_per_second": 3287.877
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10600,
       "total_flos": 2.1870706694064443e+19,
       "train_loss": 0.0,
-      "train_runtime": 3.0991,
-      "train_samples_per_second": 27427.639,
-      "train_steps_per_second": 3420.388
+      "train_runtime": 3.224,
+      "train_samples_per_second": 26365.049,
+      "train_steps_per_second": 3287.877
     }
   ],
   "logging_steps": 500,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:745796e4ed950de5bb93a3693a411905dc07c0f9e1cb8590fc1268dfe8b13783
+oid sha256:7e9345802b954b2baf811328649ad04bc8ff7cacc639a541e90a74424379fdcc
 size 4923