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-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/70ip6xxy)
 # microsoft-conditional-detr-resnet-50-finetuned-10k-cppe5-auto-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.3455
 - Map: 0.2991
@@ -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.2922,
     "eval_samples_per_second": 23.839,
     "eval_steps_per_second": 3.02,
-    "test_loss": 1.2219938039779663,
-    "test_map": 0.4162,
-    "test_map_50": 0.7489,
-    "test_map_75": 0.3848,
-    "test_map_Coverall": 0.6371,
-    "test_map_Face_Shield": 0.5697,
-    "test_map_Gloves": 0.2862,
-    "test_map_Goggles": 0.2241,
-    "test_map_Mask": 0.3638,
-    "test_map_large": 0.6303,
-    "test_map_medium": 0.2323,
-    "test_map_small": 0.4026,
-    "test_mar_1": 0.3173,
-    "test_mar_10": 0.5208,
-    "test_mar_100": 0.5371,
-    "test_mar_100_Coverall": 0.7156,
-    "test_mar_100_Face_Shield": 0.6941,
-    "test_mar_100_Gloves": 0.3967,
-    "test_mar_100_Goggles": 0.4062,
-    "test_mar_100_Mask": 0.4731,
-    "test_mar_large": 0.7364,
-    "test_mar_medium": 0.3612,
-    "test_mar_small": 0.47,
-    "test_runtime": 1.809,
-    "test_samples_per_second": 16.031,
-    "test_steps_per_second": 2.211,
+    "test_loss": 1.2524553537368774,
+    "test_map": 0.3993,
+    "test_map_50": 0.7379,
+    "test_map_75": 0.3616,
+    "test_map_Coverall": 0.6329,
+    "test_map_Face_Shield": 0.55,
+    "test_map_Gloves": 0.2937,
+    "test_map_Goggles": 0.1906,
+    "test_map_Mask": 0.3293,
+    "test_map_large": 0.6404,
+    "test_map_medium": 0.2283,
+    "test_map_small": 0.3665,
+    "test_mar_1": 0.3009,
+    "test_mar_10": 0.4921,
+    "test_mar_100": 0.5179,
+    "test_mar_100_Coverall": 0.7044,
+    "test_mar_100_Face_Shield": 0.6412,
+    "test_mar_100_Gloves": 0.3984,
+    "test_mar_100_Goggles": 0.3938,
+    "test_mar_100_Mask": 0.4519,
+    "test_mar_large": 0.7471,
+    "test_mar_medium": 0.3482,
+    "test_mar_small": 0.4078,
+    "test_runtime": 3.8706,
+    "test_samples_per_second": 7.492,
+    "test_steps_per_second": 7.492,
     "total_flos": 2.3505198523920773e+19,
-    "train_loss": 1.2746832000875028,
-    "train_runtime": 5432.0149,
-    "train_samples_per_second": 15.648,
-    "train_steps_per_second": 1.97
+    "train_loss": 0.0,
+    "train_runtime": 3.4151,
+    "train_samples_per_second": 24889.772,
+    "train_steps_per_second": 3133.183
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 100.0,
-    "test_loss": 1.2219938039779663,
-    "test_map": 0.4162,
-    "test_map_50": 0.7489,
-    "test_map_75": 0.3848,
-    "test_map_Coverall": 0.6371,
-    "test_map_Face_Shield": 0.5697,
-    "test_map_Gloves": 0.2862,
-    "test_map_Goggles": 0.2241,
-    "test_map_Mask": 0.3638,
-    "test_map_large": 0.6303,
-    "test_map_medium": 0.2323,
-    "test_map_small": 0.4026,
-    "test_mar_1": 0.3173,
-    "test_mar_10": 0.5208,
-    "test_mar_100": 0.5371,
-    "test_mar_100_Coverall": 0.7156,
-    "test_mar_100_Face_Shield": 0.6941,
-    "test_mar_100_Gloves": 0.3967,
-    "test_mar_100_Goggles": 0.4062,
-    "test_mar_100_Mask": 0.4731,
-    "test_mar_large": 0.7364,
-    "test_mar_medium": 0.3612,
-    "test_mar_small": 0.47,
-    "test_runtime": 1.809,
-    "test_samples_per_second": 16.031,
-    "test_steps_per_second": 2.211
+    "test_loss": 1.2524553537368774,
+    "test_map": 0.3993,
+    "test_map_50": 0.7379,
+    "test_map_75": 0.3616,
+    "test_map_Coverall": 0.6329,
+    "test_map_Face_Shield": 0.55,
+    "test_map_Gloves": 0.2937,
+    "test_map_Goggles": 0.1906,
+    "test_map_Mask": 0.3293,
+    "test_map_large": 0.6404,
+    "test_map_medium": 0.2283,
+    "test_map_small": 0.3665,
+    "test_mar_1": 0.3009,
+    "test_mar_10": 0.4921,
+    "test_mar_100": 0.5179,
+    "test_mar_100_Coverall": 0.7044,
+    "test_mar_100_Face_Shield": 0.6412,
+    "test_mar_100_Gloves": 0.3984,
+    "test_mar_100_Goggles": 0.3938,
+    "test_mar_100_Mask": 0.4519,
+    "test_mar_large": 0.7471,
+    "test_mar_medium": 0.3482,
+    "test_mar_small": 0.4078,
+    "test_runtime": 3.8706,
+    "test_samples_per_second": 7.492,
+    "test_steps_per_second": 7.492
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 100.0,
     "total_flos": 2.3505198523920773e+19,
-    "train_loss": 1.2746832000875028,
-    "train_runtime": 5432.0149,
-    "train_samples_per_second": 15.648,
-    "train_steps_per_second": 1.97
+    "train_loss": 0.0,
+    "train_runtime": 3.4151,
+    "train_samples_per_second": 24889.772,
+    "train_steps_per_second": 3133.183
 }

trainer_state.json

@@ -3712,10 +3712,10 @@
       "epoch": 100.0,
       "step": 10700,
       "total_flos": 2.3505198523920773e+19,
-      "train_loss": 1.2746832000875028,
-      "train_runtime": 5432.0149,
-      "train_samples_per_second": 15.648,
-      "train_steps_per_second": 1.97
+      "train_loss": 0.0,
+      "train_runtime": 3.4151,
+      "train_samples_per_second": 24889.772,
+      "train_steps_per_second": 3133.183
     }
   ],
   "logging_steps": 500,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2d560ee1c3acab83774c9afbb48ba01390a703bde4c4de3fffebaf92c590eb5f
+oid sha256:790ad8f7d751fed6952723a199b31931490af5ec87c63eeb6537b942c8698af1
 size 4987