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.2564
 - Map: 0.2948

all_results.json

@@ -28,7 +28,7 @@
     "test_steps_per_second": 0.337,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 0.1288,
-    "train_samples_per_second": 659976.194,
-    "train_steps_per_second": 83079.356
+    "train_runtime": 0.1313,
+    "train_samples_per_second": 647591.285,
+    "train_steps_per_second": 81520.315
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 100.0,
     "total_flos": 2.28455679528e+19,
     "train_loss": 0.0,
-    "train_runtime": 0.1288,
-    "train_samples_per_second": 659976.194,
-    "train_steps_per_second": 83079.356
+    "train_runtime": 0.1313,
+    "train_samples_per_second": 647591.285,
+    "train_steps_per_second": 81520.315
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10700,
       "total_flos": 2.28455679528e+19,
       "train_loss": 0.0,
-      "train_runtime": 0.1288,
-      "train_samples_per_second": 659976.194,
-      "train_steps_per_second": 83079.356
+      "train_runtime": 0.1313,
+      "train_samples_per_second": 647591.285,
+      "train_steps_per_second": 81520.315
     }
   ],
   "logging_steps": 500,