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: 2.0867
 - Map: 0.0493

all_results.json

@@ -0,0 +1,34 @@
+{
+    "epoch": 2.0,
+    "test_loss": 2.0999855995178223,
+    "test_map": 0.0537,
+    "test_map_50": 0.1084,
+    "test_map_75": 0.0431,
+    "test_map_Coverall": 0.2015,
+    "test_map_Face_Shield": 0.0,
+    "test_map_Gloves": 0.0431,
+    "test_map_Goggles": 0.0,
+    "test_map_Mask": 0.0239,
+    "test_map_large": 0.0752,
+    "test_map_medium": 0.014,
+    "test_map_small": 0.0456,
+    "test_mar_1": 0.09,
+    "test_mar_10": 0.1847,
+    "test_mar_100": 0.238,
+    "test_mar_100_Coverall": 0.6436,
+    "test_mar_100_Face_Shield": 0.0,
+    "test_mar_100_Gloves": 0.2525,
+    "test_mar_100_Goggles": 0.0,
+    "test_mar_100_Mask": 0.2941,
+    "test_mar_large": 0.2315,
+    "test_mar_medium": 0.1267,
+    "test_mar_small": 0.1328,
+    "test_runtime": 50.4569,
+    "test_samples_per_second": 0.575,
+    "test_steps_per_second": 0.079,
+    "total_flos": 4.5529873072992e+17,
+    "train_loss": 2.3308962253337735,
+    "train_runtime": 349.7464,
+    "train_samples_per_second": 4.861,
+    "train_steps_per_second": 1.218
+}

runs/May08_17-24-01_PRLNB25003/events.out.tfevents.1746696683.PRLNB25003.23296.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64268fa43000688394b7996c33f2b56f2768dd120dd67d42871e6513bbef8021
+size 1548

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 2.0,
+    "test_loss": 2.0999855995178223,
+    "test_map": 0.0537,
+    "test_map_50": 0.1084,
+    "test_map_75": 0.0431,
+    "test_map_Coverall": 0.2015,
+    "test_map_Face_Shield": 0.0,
+    "test_map_Gloves": 0.0431,
+    "test_map_Goggles": 0.0,
+    "test_map_Mask": 0.0239,
+    "test_map_large": 0.0752,
+    "test_map_medium": 0.014,
+    "test_map_small": 0.0456,
+    "test_mar_1": 0.09,
+    "test_mar_10": 0.1847,
+    "test_mar_100": 0.238,
+    "test_mar_100_Coverall": 0.6436,
+    "test_mar_100_Face_Shield": 0.0,
+    "test_mar_100_Gloves": 0.2525,
+    "test_mar_100_Goggles": 0.0,
+    "test_mar_100_Mask": 0.2941,
+    "test_mar_large": 0.2315,
+    "test_mar_medium": 0.1267,
+    "test_mar_small": 0.1328,
+    "test_runtime": 50.4569,
+    "test_samples_per_second": 0.575,
+    "test_steps_per_second": 0.079
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 2.0,
+    "total_flos": 4.5529873072992e+17,
+    "train_loss": 2.3308962253337735,
+    "train_runtime": 349.7464,
+    "train_samples_per_second": 4.861,
+    "train_steps_per_second": 1.218
+}

trainer_state.json

@@ -0,0 +1,117 @@
+{
+  "best_global_step": 426,
+  "best_metric": 0.0493,
+  "best_model_checkpoint": "detr-finetuned-cppe-5-10k-steps\\checkpoint-426",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 426,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "grad_norm": 75.64894104003906,
+      "learning_rate": 2.5821596244131456e-05,
+      "loss": 2.5499,
+      "step": 213
+    },
+    {
+      "epoch": 1.0,
+      "eval_loss": 2.3247692584991455,
+      "eval_map": 0.0367,
+      "eval_map_50": 0.0798,
+      "eval_map_75": 0.0325,
+      "eval_map_Coverall": 0.1697,
+      "eval_map_Face_Shield": 0.0,
+      "eval_map_Gloves": 0.0078,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.0058,
+      "eval_map_large": 0.0417,
+      "eval_map_medium": 0.0114,
+      "eval_map_small": 0.0018,
+      "eval_mar_1": 0.0555,
+      "eval_mar_10": 0.1278,
+      "eval_mar_100": 0.1648,
+      "eval_mar_100_Coverall": 0.5104,
+      "eval_mar_100_Face_Shield": 0.0,
+      "eval_mar_100_Gloves": 0.1813,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.1324,
+      "eval_mar_large": 0.2026,
+      "eval_mar_medium": 0.1044,
+      "eval_mar_small": 0.0416,
+      "eval_runtime": 58.8392,
+      "eval_samples_per_second": 2.549,
+      "eval_steps_per_second": 0.323,
+      "step": 213
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 47.80084228515625,
+      "learning_rate": 9.389671361502348e-07,
+      "loss": 2.1119,
+      "step": 426
+    },
+    {
+      "epoch": 2.0,
+      "eval_loss": 2.0866501331329346,
+      "eval_map": 0.0493,
+      "eval_map_50": 0.1064,
+      "eval_map_75": 0.0379,
+      "eval_map_Coverall": 0.1978,
+      "eval_map_Face_Shield": 0.0,
+      "eval_map_Gloves": 0.0253,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.0236,
+      "eval_map_large": 0.0574,
+      "eval_map_medium": 0.0304,
+      "eval_map_small": 0.0072,
+      "eval_mar_1": 0.0805,
+      "eval_mar_10": 0.1667,
+      "eval_mar_100": 0.2069,
+      "eval_mar_100_Coverall": 0.5874,
+      "eval_mar_100_Face_Shield": 0.0,
+      "eval_mar_100_Gloves": 0.204,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.2431,
+      "eval_mar_large": 0.2325,
+      "eval_mar_medium": 0.1572,
+      "eval_mar_small": 0.073,
+      "eval_runtime": 61.6528,
+      "eval_samples_per_second": 2.433,
+      "eval_steps_per_second": 0.308,
+      "step": 426
+    },
+    {
+      "epoch": 2.0,
+      "step": 426,
+      "total_flos": 4.5529873072992e+17,
+      "train_loss": 2.3308962253337735,
+      "train_runtime": 349.7464,
+      "train_samples_per_second": 4.861,
+      "train_steps_per_second": 1.218
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 426,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 4.5529873072992e+17,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}