End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: microsoft/conditional-detr-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: microsoft-conditional-detr-resnet-50-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/zldi4iq1)
 # microsoft-conditional-detr-resnet-50-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.3721
 - Map: 0.3104
@@ -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": 6.5804,
     "eval_samples_per_second": 22.795,
     "eval_steps_per_second": 2.887,
-    "test_loss": 2.7462191581726074,
-    "test_map": 0.0872,
-    "test_map_50": 0.2398,
-    "test_map_75": 0.0428,
-    "test_map_Coverall": 0.1274,
-    "test_map_Face_Shield": 0.2348,
-    "test_map_Gloves": 0.0258,
-    "test_map_Goggles": 0.0081,
-    "test_map_Mask": 0.0399,
-    "test_map_large": 0.1959,
-    "test_map_medium": 0.0643,
-    "test_map_small": 0.0054,
-    "test_mar_1": 0.1195,
-    "test_mar_10": 0.2054,
-    "test_mar_100": 0.2171,
-    "test_mar_100_Coverall": 0.3044,
-    "test_mar_100_Face_Shield": 0.3824,
-    "test_mar_100_Gloves": 0.1246,
-    "test_mar_100_Goggles": 0.1125,
-    "test_mar_100_Mask": 0.1615,
-    "test_mar_large": 0.3461,
-    "test_mar_medium": 0.1574,
-    "test_mar_small": 0.0449,
-    "test_runtime": 3.0316,
-    "test_samples_per_second": 9.566,
-    "test_steps_per_second": 1.319,
+    "test_loss": 1.007723093032837,
+    "test_map": 0.4282,
+    "test_map_50": 0.8092,
+    "test_map_75": 0.3479,
+    "test_map_Coverall": 0.6154,
+    "test_map_Face_Shield": 0.5313,
+    "test_map_Gloves": 0.3247,
+    "test_map_Goggles": 0.247,
+    "test_map_Mask": 0.4228,
+    "test_map_large": 0.6798,
+    "test_map_medium": 0.2881,
+    "test_map_small": 0.4374,
+    "test_mar_1": 0.3426,
+    "test_mar_10": 0.5385,
+    "test_mar_100": 0.5519,
+    "test_mar_100_Coverall": 0.7067,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.4426,
+    "test_mar_100_Goggles": 0.4281,
+    "test_mar_100_Mask": 0.5173,
+    "test_mar_large": 0.7432,
+    "test_mar_medium": 0.4023,
+    "test_mar_small": 0.4868,
+    "test_runtime": 3.545,
+    "test_samples_per_second": 8.18,
+    "test_steps_per_second": 8.18,
     "total_flos": 2.386038004919996e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.301,
-    "train_samples_per_second": 25749.774,
-    "train_steps_per_second": 3241.442
+    "train_runtime": 3.3486,
+    "train_samples_per_second": 25383.788,
+    "train_steps_per_second": 3195.371
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 100.0,
-    "test_loss": 2.7462191581726074,
-    "test_map": 0.0872,
-    "test_map_50": 0.2398,
-    "test_map_75": 0.0428,
-    "test_map_Coverall": 0.1274,
-    "test_map_Face_Shield": 0.2348,
-    "test_map_Gloves": 0.0258,
-    "test_map_Goggles": 0.0081,
-    "test_map_Mask": 0.0399,
-    "test_map_large": 0.1959,
-    "test_map_medium": 0.0643,
-    "test_map_small": 0.0054,
-    "test_mar_1": 0.1195,
-    "test_mar_10": 0.2054,
-    "test_mar_100": 0.2171,
-    "test_mar_100_Coverall": 0.3044,
-    "test_mar_100_Face_Shield": 0.3824,
-    "test_mar_100_Gloves": 0.1246,
-    "test_mar_100_Goggles": 0.1125,
-    "test_mar_100_Mask": 0.1615,
-    "test_mar_large": 0.3461,
-    "test_mar_medium": 0.1574,
-    "test_mar_small": 0.0449,
-    "test_runtime": 3.0316,
-    "test_samples_per_second": 9.566,
-    "test_steps_per_second": 1.319
+    "test_loss": 1.007723093032837,
+    "test_map": 0.4282,
+    "test_map_50": 0.8092,
+    "test_map_75": 0.3479,
+    "test_map_Coverall": 0.6154,
+    "test_map_Face_Shield": 0.5313,
+    "test_map_Gloves": 0.3247,
+    "test_map_Goggles": 0.247,
+    "test_map_Mask": 0.4228,
+    "test_map_large": 0.6798,
+    "test_map_medium": 0.2881,
+    "test_map_small": 0.4374,
+    "test_mar_1": 0.3426,
+    "test_mar_10": 0.5385,
+    "test_mar_100": 0.5519,
+    "test_mar_100_Coverall": 0.7067,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.4426,
+    "test_mar_100_Goggles": 0.4281,
+    "test_mar_100_Mask": 0.5173,
+    "test_mar_large": 0.7432,
+    "test_mar_medium": 0.4023,
+    "test_mar_small": 0.4868,
+    "test_runtime": 3.545,
+    "test_samples_per_second": 8.18,
+    "test_steps_per_second": 8.18
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 100.0,
     "total_flos": 2.386038004919996e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.301,
-    "train_samples_per_second": 25749.774,
-    "train_steps_per_second": 3241.442
+    "train_runtime": 3.3486,
+    "train_samples_per_second": 25383.788,
+    "train_steps_per_second": 3195.371
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10700,
       "total_flos": 2.386038004919996e+19,
       "train_loss": 0.0,
-      "train_runtime": 3.301,
-      "train_samples_per_second": 25749.774,
-      "train_steps_per_second": 3241.442
+      "train_runtime": 3.3486,
+      "train_samples_per_second": 25383.788,
+      "train_steps_per_second": 3195.371
     }
   ],
   "logging_steps": 500,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4e71c86f93fd52f29fe23a74e572e40ff62d585a9b357d12b23eb3f90489054a
+oid sha256:0bdd18b2f3bd929647bf69b16367dcd71376063039b5a976dd6408d6dbf41ff3
 size 4987