End of training

README.md

@@ -3,6 +3,8 @@ library_name: transformers
 license: apache-2.0
 base_model: facebook/detr-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: detr-finetuned-cppe-5-10k-steps
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # detr-finetuned-cppe-5-10k-steps
 
-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.2683
 - Map: 0.2876

all_results.json

@@ -23,12 +23,12 @@
     "test_mar_large": 0.7513,
     "test_mar_medium": 0.4402,
     "test_mar_small": 0.3399,
-    "test_runtime": 1.8236,
-    "test_samples_per_second": 15.903,
-    "test_steps_per_second": 2.193,
+    "test_runtime": 2.2584,
+    "test_samples_per_second": 12.841,
+    "test_steps_per_second": 1.771,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 0.0454,
-    "train_samples_per_second": 1871818.129,
-    "train_steps_per_second": 235628.87
+    "train_runtime": 0.0724,
+    "train_samples_per_second": 1174518.979,
+    "train_steps_per_second": 147851.213
 }

test_results.json

@@ -23,7 +23,7 @@
     "test_mar_large": 0.7513,
     "test_mar_medium": 0.4402,
     "test_mar_small": 0.3399,
-    "test_runtime": 1.8236,
-    "test_samples_per_second": 15.903,
-    "test_steps_per_second": 2.193
+    "test_runtime": 2.2584,
+    "test_samples_per_second": 12.841,
+    "test_steps_per_second": 1.771
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 100.0,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 0.0454,
-    "train_samples_per_second": 1871818.129,
-    "train_steps_per_second": 235628.87
+    "train_runtime": 0.0724,
+    "train_samples_per_second": 1174518.979,
+    "train_steps_per_second": 147851.213
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10700,
       "total_flos": 2.28455679528e+19,
       "train_loss": 0.0,
-      "train_runtime": 0.0454,
-      "train_samples_per_second": 1871818.129,
-      "train_steps_per_second": 235628.87
+      "train_runtime": 0.0724,
+      "train_samples_per_second": 1174518.979,
+      "train_steps_per_second": 147851.213
     }
   ],
   "logging_steps": 500,