End of training

README.md

@@ -1,6 +1,8 @@
 ---
 base_model: jozhang97/deta-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: jozhang97-deta-resnet-50-finetuned-10k-cppe5-auto-pad
@@ -13,7 +15,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/p99219yc)
 # jozhang97-deta-resnet-50-finetuned-10k-cppe5-auto-pad
 
-This model is a fine-tuned version of [jozhang97/deta-resnet-50](https://huggingface.co/jozhang97/deta-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [jozhang97/deta-resnet-50](https://huggingface.co/jozhang97/deta-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.6173
 - Map: 0.1965
@@ -58,7 +60,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": 21.4083,
     "eval_samples_per_second": 7.007,
     "eval_steps_per_second": 0.888,
-    "test_loss": 1.6902095079421997,
-    "test_map": 0.264,
-    "test_map_50": 0.559,
-    "test_map_75": 0.2055,
-    "test_map_Coverall": 0.4752,
-    "test_map_Face_Shield": 0.4395,
-    "test_map_Gloves": 0.0822,
-    "test_map_Goggles": 0.1197,
-    "test_map_Mask": 0.2037,
-    "test_map_large": 0.3582,
-    "test_map_medium": 0.1795,
-    "test_map_small": 0.2979,
-    "test_mar_1": 0.2365,
-    "test_mar_10": 0.3811,
-    "test_mar_100": 0.4012,
-    "test_mar_100_Coverall": 0.5689,
-    "test_mar_100_Face_Shield": 0.5882,
-    "test_mar_100_Gloves": 0.2295,
-    "test_mar_100_Goggles": 0.2656,
-    "test_mar_100_Mask": 0.3538,
-    "test_mar_large": 0.5382,
-    "test_mar_medium": 0.3294,
-    "test_mar_small": 0.3411,
-    "test_runtime": 4.6886,
-    "test_samples_per_second": 6.185,
-    "test_steps_per_second": 0.853,
+    "test_loss": 2.093271255493164,
+    "test_map": 0.2273,
+    "test_map_50": 0.4997,
+    "test_map_75": 0.1598,
+    "test_map_Coverall": 0.4647,
+    "test_map_Face_Shield": 0.2719,
+    "test_map_Gloves": 0.0883,
+    "test_map_Goggles": 0.0944,
+    "test_map_Mask": 0.2171,
+    "test_map_large": 0.3695,
+    "test_map_medium": 0.1365,
+    "test_map_small": 0.1844,
+    "test_mar_1": 0.1963,
+    "test_mar_10": 0.3698,
+    "test_mar_100": 0.4004,
+    "test_mar_100_Coverall": 0.5578,
+    "test_mar_100_Face_Shield": 0.5235,
+    "test_mar_100_Gloves": 0.2672,
+    "test_mar_100_Goggles": 0.2844,
+    "test_mar_100_Mask": 0.3692,
+    "test_mar_large": 0.5659,
+    "test_mar_medium": 0.2851,
+    "test_mar_small": 0.3137,
+    "test_runtime": 6.1322,
+    "test_samples_per_second": 4.729,
+    "test_steps_per_second": 4.729,
     "total_flos": 2.4566240955078824e+19,
-    "train_loss": 1.7565026848271208,
-    "train_runtime": 21477.6558,
-    "train_samples_per_second": 3.958,
-    "train_steps_per_second": 0.494
+    "train_loss": 0.0,
+    "train_runtime": 3.4196,
+    "train_samples_per_second": 24856.616,
+    "train_steps_per_second": 3099.766
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 99.53051643192488,
-    "test_loss": 1.6902095079421997,
-    "test_map": 0.264,
-    "test_map_50": 0.559,
-    "test_map_75": 0.2055,
-    "test_map_Coverall": 0.4752,
-    "test_map_Face_Shield": 0.4395,
-    "test_map_Gloves": 0.0822,
-    "test_map_Goggles": 0.1197,
-    "test_map_Mask": 0.2037,
-    "test_map_large": 0.3582,
-    "test_map_medium": 0.1795,
-    "test_map_small": 0.2979,
-    "test_mar_1": 0.2365,
-    "test_mar_10": 0.3811,
-    "test_mar_100": 0.4012,
-    "test_mar_100_Coverall": 0.5689,
-    "test_mar_100_Face_Shield": 0.5882,
-    "test_mar_100_Gloves": 0.2295,
-    "test_mar_100_Goggles": 0.2656,
-    "test_mar_100_Mask": 0.3538,
-    "test_mar_large": 0.5382,
-    "test_mar_medium": 0.3294,
-    "test_mar_small": 0.3411,
-    "test_runtime": 4.6886,
-    "test_samples_per_second": 6.185,
-    "test_steps_per_second": 0.853
+    "test_loss": 2.093271255493164,
+    "test_map": 0.2273,
+    "test_map_50": 0.4997,
+    "test_map_75": 0.1598,
+    "test_map_Coverall": 0.4647,
+    "test_map_Face_Shield": 0.2719,
+    "test_map_Gloves": 0.0883,
+    "test_map_Goggles": 0.0944,
+    "test_map_Mask": 0.2171,
+    "test_map_large": 0.3695,
+    "test_map_medium": 0.1365,
+    "test_map_small": 0.1844,
+    "test_mar_1": 0.1963,
+    "test_mar_10": 0.3698,
+    "test_mar_100": 0.4004,
+    "test_mar_100_Coverall": 0.5578,
+    "test_mar_100_Face_Shield": 0.5235,
+    "test_mar_100_Gloves": 0.2672,
+    "test_mar_100_Goggles": 0.2844,
+    "test_mar_100_Mask": 0.3692,
+    "test_mar_large": 0.5659,
+    "test_mar_medium": 0.2851,
+    "test_mar_small": 0.3137,
+    "test_runtime": 6.1322,
+    "test_samples_per_second": 4.729,
+    "test_steps_per_second": 4.729
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 99.53051643192488,
     "total_flos": 2.4566240955078824e+19,
-    "train_loss": 1.7565026848271208,
-    "train_runtime": 21477.6558,
-    "train_samples_per_second": 3.958,
-    "train_steps_per_second": 0.494
+    "train_loss": 0.0,
+    "train_runtime": 3.4196,
+    "train_samples_per_second": 24856.616,
+    "train_steps_per_second": 3099.766
 }

trainer_state.json

@@ -3712,10 +3712,10 @@
       "epoch": 99.53051643192488,
       "step": 10600,
       "total_flos": 2.4566240955078824e+19,
-      "train_loss": 1.7565026848271208,
-      "train_runtime": 21477.6558,
-      "train_samples_per_second": 3.958,
-      "train_steps_per_second": 0.494
+      "train_loss": 0.0,
+      "train_runtime": 3.4196,
+      "train_samples_per_second": 24856.616,
+      "train_steps_per_second": 3099.766
     }
   ],
   "logging_steps": 500,

training_args.bin

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