End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: SenseTime/deformable-detr
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: sensetime-deformable-detr-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/zv86s9w6)
 # sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on an unknown dataset.
+This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.0442
 - Map: 0.3514
@@ -59,7 +61,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

@@ -49,12 +49,12 @@
     "test_mar_large": 0.8137,
     "test_mar_medium": 0.5441,
     "test_mar_small": 0.5739,
-    "test_runtime": 5.5464,
-    "test_samples_per_second": 5.229,
-    "test_steps_per_second": 5.229,
+    "test_runtime": 5.4531,
+    "test_samples_per_second": 5.318,
+    "test_steps_per_second": 5.318,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.224,
-    "train_samples_per_second": 26365.049,
-    "train_steps_per_second": 3287.877
+    "train_runtime": 3.1369,
+    "train_samples_per_second": 27096.987,
+    "train_steps_per_second": 3379.154
 }

test_results.json

@@ -23,7 +23,7 @@
     "test_mar_large": 0.8137,
     "test_mar_medium": 0.5441,
     "test_mar_small": 0.5739,
-    "test_runtime": 5.5464,
-    "test_samples_per_second": 5.229,
-    "test_steps_per_second": 5.229
+    "test_runtime": 5.4531,
+    "test_samples_per_second": 5.318,
+    "test_steps_per_second": 5.318
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 99.53051643192488,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 3.224,
-    "train_samples_per_second": 26365.049,
-    "train_steps_per_second": 3287.877
+    "train_runtime": 3.1369,
+    "train_samples_per_second": 27096.987,
+    "train_steps_per_second": 3379.154
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10600,
       "total_flos": 2.1870706694064443e+19,
       "train_loss": 0.0,
-      "train_runtime": 3.224,
-      "train_samples_per_second": 26365.049,
-      "train_steps_per_second": 3287.877
+      "train_runtime": 3.1369,
+      "train_samples_per_second": 27096.987,
+      "train_steps_per_second": 3379.154
     }
   ],
   "logging_steps": 500,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98adaa035dc80492e4fa66bfeb25d486f379832ec881e30bfdac8f750bc376d8
+oid sha256:2c3222ccdb193d177cc05f5ceafdc6676c18c37cf5c8df097f28bd95c9df0e91
 size 4923