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/3e2mix7l)
 # 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": 1.0454223155975342,
-    "test_map": 0.4602,
-    "test_map_50": 0.8147,
-    "test_map_75": 0.4515,
-    "test_map_Coverall": 0.6633,
-    "test_map_Face_Shield": 0.5231,
-    "test_map_Gloves": 0.3942,
-    "test_map_Goggles": 0.2844,
-    "test_map_Mask": 0.4361,
-    "test_map_large": 0.6894,
-    "test_map_medium": 0.3479,
-    "test_map_small": 0.3561,
-    "test_mar_1": 0.3531,
-    "test_mar_10": 0.5695,
-    "test_mar_100": 0.5755,
-    "test_mar_100_Coverall": 0.7533,
-    "test_mar_100_Face_Shield": 0.6765,
-    "test_mar_100_Gloves": 0.4721,
-    "test_mar_100_Goggles": 0.4563,
-    "test_mar_100_Mask": 0.5192,
-    "test_mar_large": 0.7381,
-    "test_mar_medium": 0.4441,
-    "test_mar_small": 0.4885,
-    "test_runtime": 1.9771,
-    "test_samples_per_second": 14.668,
-    "test_steps_per_second": 2.023,
+    "test_loss": 1.2813799381256104,
+    "test_map": 0.3549,
+    "test_map_50": 0.721,
+    "test_map_75": 0.2678,
+    "test_map_Coverall": 0.6333,
+    "test_map_Face_Shield": 0.4265,
+    "test_map_Gloves": 0.2742,
+    "test_map_Goggles": 0.1321,
+    "test_map_Mask": 0.3084,
+    "test_map_large": 0.6306,
+    "test_map_medium": 0.2621,
+    "test_map_small": 0.1455,
+    "test_mar_1": 0.2902,
+    "test_mar_10": 0.4824,
+    "test_mar_100": 0.4937,
+    "test_mar_100_Coverall": 0.7156,
+    "test_mar_100_Face_Shield": 0.5941,
+    "test_mar_100_Gloves": 0.3754,
+    "test_mar_100_Goggles": 0.3812,
+    "test_mar_100_Mask": 0.4019,
+    "test_mar_large": 0.7465,
+    "test_mar_medium": 0.3856,
+    "test_mar_small": 0.2586,
+    "test_runtime": 3.8191,
+    "test_samples_per_second": 7.593,
+    "test_steps_per_second": 7.593,
     "total_flos": 2.28455679528e+19,
-    "train_loss": 0.9875392821124781,
-    "train_runtime": 5219.9382,
-    "train_samples_per_second": 16.284,
-    "train_steps_per_second": 2.05
+    "train_loss": 0.0,
+    "train_runtime": 3.8694,
+    "train_samples_per_second": 21967.505,
+    "train_steps_per_second": 2765.321
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 100.0,
-    "test_loss": 1.0454223155975342,
-    "test_map": 0.4602,
-    "test_map_50": 0.8147,
-    "test_map_75": 0.4515,
-    "test_map_Coverall": 0.6633,
-    "test_map_Face_Shield": 0.5231,
-    "test_map_Gloves": 0.3942,
-    "test_map_Goggles": 0.2844,
-    "test_map_Mask": 0.4361,
-    "test_map_large": 0.6894,
-    "test_map_medium": 0.3479,
-    "test_map_small": 0.3561,
-    "test_mar_1": 0.3531,
-    "test_mar_10": 0.5695,
-    "test_mar_100": 0.5755,
-    "test_mar_100_Coverall": 0.7533,
-    "test_mar_100_Face_Shield": 0.6765,
-    "test_mar_100_Gloves": 0.4721,
-    "test_mar_100_Goggles": 0.4563,
-    "test_mar_100_Mask": 0.5192,
-    "test_mar_large": 0.7381,
-    "test_mar_medium": 0.4441,
-    "test_mar_small": 0.4885,
-    "test_runtime": 1.9771,
-    "test_samples_per_second": 14.668,
-    "test_steps_per_second": 2.023
+    "test_loss": 1.2813799381256104,
+    "test_map": 0.3549,
+    "test_map_50": 0.721,
+    "test_map_75": 0.2678,
+    "test_map_Coverall": 0.6333,
+    "test_map_Face_Shield": 0.4265,
+    "test_map_Gloves": 0.2742,
+    "test_map_Goggles": 0.1321,
+    "test_map_Mask": 0.3084,
+    "test_map_large": 0.6306,
+    "test_map_medium": 0.2621,
+    "test_map_small": 0.1455,
+    "test_mar_1": 0.2902,
+    "test_mar_10": 0.4824,
+    "test_mar_100": 0.4937,
+    "test_mar_100_Coverall": 0.7156,
+    "test_mar_100_Face_Shield": 0.5941,
+    "test_mar_100_Gloves": 0.3754,
+    "test_mar_100_Goggles": 0.3812,
+    "test_mar_100_Mask": 0.4019,
+    "test_mar_large": 0.7465,
+    "test_mar_medium": 0.3856,
+    "test_mar_small": 0.2586,
+    "test_runtime": 3.8191,
+    "test_samples_per_second": 7.593,
+    "test_steps_per_second": 7.593
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 100.0,
     "total_flos": 2.28455679528e+19,
-    "train_loss": 0.9875392821124781,
-    "train_runtime": 5219.9382,
-    "train_samples_per_second": 16.284,
-    "train_steps_per_second": 2.05
+    "train_loss": 0.0,
+    "train_runtime": 3.8694,
+    "train_samples_per_second": 21967.505,
+    "train_steps_per_second": 2765.321
 }

trainer_state.json

@@ -3712,10 +3712,10 @@
       "epoch": 100.0,
       "step": 10700,
       "total_flos": 2.28455679528e+19,
-      "train_loss": 0.9875392821124781,
-      "train_runtime": 5219.9382,
-      "train_samples_per_second": 16.284,
-      "train_steps_per_second": 2.05
+      "train_loss": 0.0,
+      "train_runtime": 3.8694,
+      "train_samples_per_second": 21967.505,
+      "train_steps_per_second": 2765.321
     }
   ],
   "logging_steps": 500,

training_args.bin

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