End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: facebook/detr-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: facebook-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/oaigsyzd)
 # facebook-detr-resnet-50-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.3496
 - Map: 0.2868
@@ -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.9883,
     "eval_samples_per_second": 21.464,
     "eval_steps_per_second": 2.719,
-    "test_loss": 3.211362838745117,
-    "test_map": 0.1054,
-    "test_map_50": 0.2494,
-    "test_map_75": 0.0627,
-    "test_map_Coverall": 0.1674,
-    "test_map_Face_Shield": 0.2585,
-    "test_map_Gloves": 0.0366,
-    "test_map_Goggles": 0.0144,
-    "test_map_Mask": 0.0503,
-    "test_map_large": 0.2175,
-    "test_map_medium": 0.0706,
-    "test_map_small": 0.0021,
-    "test_mar_1": 0.1248,
-    "test_mar_10": 0.225,
-    "test_mar_100": 0.2321,
-    "test_mar_100_Coverall": 0.3356,
-    "test_mar_100_Face_Shield": 0.4235,
-    "test_mar_100_Gloves": 0.1262,
-    "test_mar_100_Goggles": 0.1406,
-    "test_mar_100_Mask": 0.1346,
-    "test_mar_large": 0.3524,
-    "test_mar_medium": 0.1804,
-    "test_mar_small": 0.013,
-    "test_runtime": 2.9936,
-    "test_samples_per_second": 9.687,
-    "test_steps_per_second": 1.336,
+    "test_loss": 0.9010592699050903,
+    "test_map": 0.4627,
+    "test_map_50": 0.8215,
+    "test_map_75": 0.4515,
+    "test_map_Coverall": 0.6652,
+    "test_map_Face_Shield": 0.5306,
+    "test_map_Gloves": 0.3932,
+    "test_map_Goggles": 0.2849,
+    "test_map_Mask": 0.4396,
+    "test_map_large": 0.6956,
+    "test_map_medium": 0.3529,
+    "test_map_small": 0.3581,
+    "test_mar_1": 0.3546,
+    "test_mar_10": 0.5719,
+    "test_mar_100": 0.5782,
+    "test_mar_100_Coverall": 0.7556,
+    "test_mar_100_Face_Shield": 0.6824,
+    "test_mar_100_Gloves": 0.4738,
+    "test_mar_100_Goggles": 0.4563,
+    "test_mar_100_Mask": 0.5231,
+    "test_mar_large": 0.7451,
+    "test_mar_medium": 0.4453,
+    "test_mar_small": 0.4885,
+    "test_runtime": 3.4389,
+    "test_samples_per_second": 8.433,
+    "test_steps_per_second": 8.433,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.1271,
-    "train_samples_per_second": 27182.099,
-    "train_steps_per_second": 3421.747
+    "train_runtime": 3.2678,
+    "train_samples_per_second": 26011.548,
+    "train_steps_per_second": 3274.395
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 100.0,
-    "test_loss": 3.211362838745117,
-    "test_map": 0.1054,
-    "test_map_50": 0.2494,
-    "test_map_75": 0.0627,
-    "test_map_Coverall": 0.1674,
-    "test_map_Face_Shield": 0.2585,
-    "test_map_Gloves": 0.0366,
-    "test_map_Goggles": 0.0144,
-    "test_map_Mask": 0.0503,
-    "test_map_large": 0.2175,
-    "test_map_medium": 0.0706,
-    "test_map_small": 0.0021,
-    "test_mar_1": 0.1248,
-    "test_mar_10": 0.225,
-    "test_mar_100": 0.2321,
-    "test_mar_100_Coverall": 0.3356,
-    "test_mar_100_Face_Shield": 0.4235,
-    "test_mar_100_Gloves": 0.1262,
-    "test_mar_100_Goggles": 0.1406,
-    "test_mar_100_Mask": 0.1346,
-    "test_mar_large": 0.3524,
-    "test_mar_medium": 0.1804,
-    "test_mar_small": 0.013,
-    "test_runtime": 2.9936,
-    "test_samples_per_second": 9.687,
-    "test_steps_per_second": 1.336
+    "test_loss": 0.9010592699050903,
+    "test_map": 0.4627,
+    "test_map_50": 0.8215,
+    "test_map_75": 0.4515,
+    "test_map_Coverall": 0.6652,
+    "test_map_Face_Shield": 0.5306,
+    "test_map_Gloves": 0.3932,
+    "test_map_Goggles": 0.2849,
+    "test_map_Mask": 0.4396,
+    "test_map_large": 0.6956,
+    "test_map_medium": 0.3529,
+    "test_map_small": 0.3581,
+    "test_mar_1": 0.3546,
+    "test_mar_10": 0.5719,
+    "test_mar_100": 0.5782,
+    "test_mar_100_Coverall": 0.7556,
+    "test_mar_100_Face_Shield": 0.6824,
+    "test_mar_100_Gloves": 0.4738,
+    "test_mar_100_Goggles": 0.4563,
+    "test_mar_100_Mask": 0.5231,
+    "test_mar_large": 0.7451,
+    "test_mar_medium": 0.4453,
+    "test_mar_small": 0.4885,
+    "test_runtime": 3.4389,
+    "test_samples_per_second": 8.433,
+    "test_steps_per_second": 8.433
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 100.0,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.1271,
-    "train_samples_per_second": 27182.099,
-    "train_steps_per_second": 3421.747
+    "train_runtime": 3.2678,
+    "train_samples_per_second": 26011.548,
+    "train_steps_per_second": 3274.395
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10700,
       "total_flos": 2.28455679528e+19,
       "train_loss": 0.0,
-      "train_runtime": 3.1271,
-      "train_samples_per_second": 27182.099,
-      "train_steps_per_second": 3421.747
+      "train_runtime": 3.2678,
+      "train_samples_per_second": 26011.548,
+      "train_steps_per_second": 3274.395
     }
   ],
   "logging_steps": 500,

training_args.bin

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