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-auto-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/jf9me88v)
 # sensetime-deformable-detr-finetuned-10k-cppe5-auto-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.1077
 - Map: 0.2966
@@ -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.523,
     "eval_samples_per_second": 9.663,
     "eval_steps_per_second": 1.224,
-    "test_loss": 0.9166296720504761,
-    "test_map": 0.4366,
-    "test_map_50": 0.7304,
-    "test_map_75": 0.4448,
-    "test_map_Coverall": 0.6296,
-    "test_map_Face_Shield": 0.4194,
-    "test_map_Gloves": 0.3438,
-    "test_map_Goggles": 0.301,
-    "test_map_Mask": 0.4891,
-    "test_map_large": 0.6324,
-    "test_map_medium": 0.2898,
-    "test_map_small": 0.4357,
-    "test_mar_1": 0.3442,
-    "test_mar_10": 0.5781,
-    "test_mar_100": 0.6031,
-    "test_mar_100_Coverall": 0.7222,
-    "test_mar_100_Face_Shield": 0.6824,
-    "test_mar_100_Gloves": 0.5016,
-    "test_mar_100_Goggles": 0.5344,
-    "test_mar_100_Mask": 0.575,
-    "test_mar_large": 0.7618,
-    "test_mar_medium": 0.4489,
-    "test_mar_small": 0.5004,
-    "test_runtime": 3.552,
-    "test_samples_per_second": 8.164,
-    "test_steps_per_second": 1.126,
+    "test_loss": 0.9160362482070923,
+    "test_map": 0.4177,
+    "test_map_50": 0.7034,
+    "test_map_75": 0.4325,
+    "test_map_Coverall": 0.585,
+    "test_map_Face_Shield": 0.3651,
+    "test_map_Gloves": 0.4048,
+    "test_map_Goggles": 0.2813,
+    "test_map_Mask": 0.4521,
+    "test_map_large": 0.6049,
+    "test_map_medium": 0.2966,
+    "test_map_small": 0.4271,
+    "test_mar_1": 0.3319,
+    "test_mar_10": 0.5697,
+    "test_mar_100": 0.59,
+    "test_mar_100_Coverall": 0.7244,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.5148,
+    "test_mar_100_Goggles": 0.4844,
+    "test_mar_100_Mask": 0.5615,
+    "test_mar_large": 0.7603,
+    "test_mar_medium": 0.4395,
+    "test_mar_small": 0.486,
+    "test_runtime": 5.3168,
+    "test_samples_per_second": 5.454,
+    "test_steps_per_second": 5.454,
     "total_flos": 2.0297599094333424e+19,
-    "train_loss": 1.1465214905648862,
-    "train_runtime": 16105.6449,
-    "train_samples_per_second": 5.278,
-    "train_steps_per_second": 0.658
+    "train_loss": 0.0,
+    "train_runtime": 3.268,
+    "train_samples_per_second": 26009.618,
+    "train_steps_per_second": 3243.552
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 99.53051643192488,
-    "test_loss": 0.9166296720504761,
-    "test_map": 0.4366,
-    "test_map_50": 0.7304,
-    "test_map_75": 0.4448,
-    "test_map_Coverall": 0.6296,
-    "test_map_Face_Shield": 0.4194,
-    "test_map_Gloves": 0.3438,
-    "test_map_Goggles": 0.301,
-    "test_map_Mask": 0.4891,
-    "test_map_large": 0.6324,
-    "test_map_medium": 0.2898,
-    "test_map_small": 0.4357,
-    "test_mar_1": 0.3442,
-    "test_mar_10": 0.5781,
-    "test_mar_100": 0.6031,
-    "test_mar_100_Coverall": 0.7222,
-    "test_mar_100_Face_Shield": 0.6824,
-    "test_mar_100_Gloves": 0.5016,
-    "test_mar_100_Goggles": 0.5344,
-    "test_mar_100_Mask": 0.575,
-    "test_mar_large": 0.7618,
-    "test_mar_medium": 0.4489,
-    "test_mar_small": 0.5004,
-    "test_runtime": 3.552,
-    "test_samples_per_second": 8.164,
-    "test_steps_per_second": 1.126
+    "test_loss": 0.9160362482070923,
+    "test_map": 0.4177,
+    "test_map_50": 0.7034,
+    "test_map_75": 0.4325,
+    "test_map_Coverall": 0.585,
+    "test_map_Face_Shield": 0.3651,
+    "test_map_Gloves": 0.4048,
+    "test_map_Goggles": 0.2813,
+    "test_map_Mask": 0.4521,
+    "test_map_large": 0.6049,
+    "test_map_medium": 0.2966,
+    "test_map_small": 0.4271,
+    "test_mar_1": 0.3319,
+    "test_mar_10": 0.5697,
+    "test_mar_100": 0.59,
+    "test_mar_100_Coverall": 0.7244,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.5148,
+    "test_mar_100_Goggles": 0.4844,
+    "test_mar_100_Mask": 0.5615,
+    "test_mar_large": 0.7603,
+    "test_mar_medium": 0.4395,
+    "test_mar_small": 0.486,
+    "test_runtime": 5.3168,
+    "test_samples_per_second": 5.454,
+    "test_steps_per_second": 5.454
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 99.53051643192488,
     "total_flos": 2.0297599094333424e+19,
-    "train_loss": 1.1465214905648862,
-    "train_runtime": 16105.6449,
-    "train_samples_per_second": 5.278,
-    "train_steps_per_second": 0.658
+    "train_loss": 0.0,
+    "train_runtime": 3.268,
+    "train_samples_per_second": 26009.618,
+    "train_steps_per_second": 3243.552
 }

trainer_state.json

@@ -3712,10 +3712,10 @@
       "epoch": 99.53051643192488,
       "step": 10600,
       "total_flos": 2.0297599094333424e+19,
-      "train_loss": 1.1465214905648862,
-      "train_runtime": 16105.6449,
-      "train_samples_per_second": 5.278,
-      "train_steps_per_second": 0.658
+      "train_loss": 0.0,
+      "train_runtime": 3.268,
+      "train_samples_per_second": 26009.618,
+      "train_steps_per_second": 3243.552
     }
   ],
   "logging_steps": 500,

training_args.bin

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