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.6238
 - Map: 0.1619

all_results.json

@@ -1,34 +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,
+    "epoch": 10.0,
+    "test_loss": 1.4663528203964233,
+    "test_map": 0.2212,
+    "test_map_50": 0.4088,
+    "test_map_75": 0.1654,
+    "test_map_Coverall": 0.4558,
+    "test_map_Face_Shield": 0.2018,
+    "test_map_Gloves": 0.1562,
+    "test_map_Goggles": 0.0026,
+    "test_map_Mask": 0.2894,
+    "test_map_large": 0.2247,
+    "test_map_medium": 0.1524,
+    "test_map_small": 0.082,
+    "test_mar_1": 0.2,
+    "test_mar_10": 0.3986,
+    "test_mar_100": 0.4165,
+    "test_mar_100_Coverall": 0.7103,
+    "test_mar_100_Face_Shield": 0.5588,
+    "test_mar_100_Gloves": 0.3458,
+    "test_mar_100_Goggles": 0.0517,
+    "test_mar_100_Mask": 0.4157,
+    "test_mar_large": 0.4387,
+    "test_mar_medium": 0.2702,
+    "test_mar_small": 0.135,
+    "test_runtime": 50.7469,
+    "test_samples_per_second": 0.571,
     "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
+    "total_flos": 2.2764936536496e+18,
+    "train_loss": 1.393765215806558,
+    "train_runtime": 1396.8073,
+    "train_samples_per_second": 6.085,
+    "train_steps_per_second": 1.525
 }

runs/May08_17-33-25_PRLNB25003/events.out.tfevents.1746698287.PRLNB25003.21208.1

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

test_results.json

@@ -1,29 +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,
+    "epoch": 10.0,
+    "test_loss": 1.4663528203964233,
+    "test_map": 0.2212,
+    "test_map_50": 0.4088,
+    "test_map_75": 0.1654,
+    "test_map_Coverall": 0.4558,
+    "test_map_Face_Shield": 0.2018,
+    "test_map_Gloves": 0.1562,
+    "test_map_Goggles": 0.0026,
+    "test_map_Mask": 0.2894,
+    "test_map_large": 0.2247,
+    "test_map_medium": 0.1524,
+    "test_map_small": 0.082,
+    "test_mar_1": 0.2,
+    "test_mar_10": 0.3986,
+    "test_mar_100": 0.4165,
+    "test_mar_100_Coverall": 0.7103,
+    "test_mar_100_Face_Shield": 0.5588,
+    "test_mar_100_Gloves": 0.3458,
+    "test_mar_100_Goggles": 0.0517,
+    "test_mar_100_Mask": 0.4157,
+    "test_mar_large": 0.4387,
+    "test_mar_medium": 0.2702,
+    "test_mar_small": 0.135,
+    "test_runtime": 50.7469,
+    "test_samples_per_second": 0.571,
     "test_steps_per_second": 0.079
 }

train_results.json

@@ -1,8 +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
+    "epoch": 10.0,
+    "total_flos": 2.2764936536496e+18,
+    "train_loss": 1.393765215806558,
+    "train_runtime": 1396.8073,
+    "train_samples_per_second": 6.085,
+    "train_steps_per_second": 1.525
 }

trainer_state.json

@@ -1,10 +1,10 @@
 {
-  "best_global_step": 426,
-  "best_metric": 0.0493,
-  "best_model_checkpoint": "detr-finetuned-cppe-5-10k-steps\\checkpoint-426",
-  "epoch": 2.0,
+  "best_global_step": 2130,
+  "best_metric": 0.1619,
+  "best_model_checkpoint": "detr-finetuned-cppe-5-10k-steps\\checkpoint-2130",
+  "epoch": 10.0,
   "eval_steps": 500,
-  "global_step": 426,
+  "global_step": 2130,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -84,19 +84,315 @@
       "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
+      "epoch": 3.0,
+      "grad_norm": 48.019927978515625,
+      "learning_rate": 3.5187793427230045e-05,
+      "loss": 2.0052,
+      "step": 639
+    },
+    {
+      "epoch": 3.0,
+      "eval_loss": 2.168910026550293,
+      "eval_map": 0.0563,
+      "eval_map_50": 0.1279,
+      "eval_map_75": 0.0441,
+      "eval_map_Coverall": 0.2039,
+      "eval_map_Face_Shield": 0.0,
+      "eval_map_Gloves": 0.0158,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.0618,
+      "eval_map_large": 0.0688,
+      "eval_map_medium": 0.0414,
+      "eval_map_small": 0.0154,
+      "eval_mar_1": 0.0818,
+      "eval_mar_10": 0.1611,
+      "eval_mar_100": 0.1793,
+      "eval_mar_100_Coverall": 0.4757,
+      "eval_mar_100_Face_Shield": 0.0,
+      "eval_mar_100_Gloves": 0.1942,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.2267,
+      "eval_mar_large": 0.2199,
+      "eval_mar_medium": 0.131,
+      "eval_mar_small": 0.0616,
+      "eval_runtime": 55.4656,
+      "eval_samples_per_second": 2.704,
+      "eval_steps_per_second": 0.343,
+      "step": 639
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 124.8763427734375,
+      "learning_rate": 3.018779342723005e-05,
+      "loss": 1.9373,
+      "step": 852
+    },
+    {
+      "epoch": 4.0,
+      "eval_loss": 1.9264371395111084,
+      "eval_map": 0.0813,
+      "eval_map_50": 0.1755,
+      "eval_map_75": 0.0679,
+      "eval_map_Coverall": 0.3464,
+      "eval_map_Face_Shield": 0.0,
+      "eval_map_Gloves": 0.0245,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.0357,
+      "eval_map_large": 0.0953,
+      "eval_map_medium": 0.056,
+      "eval_map_small": 0.0125,
+      "eval_mar_1": 0.0916,
+      "eval_mar_10": 0.1816,
+      "eval_mar_100": 0.206,
+      "eval_mar_100_Coverall": 0.6302,
+      "eval_mar_100_Face_Shield": 0.0,
+      "eval_mar_100_Gloves": 0.2004,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.1996,
+      "eval_mar_large": 0.2549,
+      "eval_mar_medium": 0.1446,
+      "eval_mar_small": 0.0662,
+      "eval_runtime": 63.4064,
+      "eval_samples_per_second": 2.366,
+      "eval_steps_per_second": 0.3,
+      "step": 852
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 49.01365661621094,
+      "learning_rate": 2.518779342723005e-05,
+      "loss": 1.8396,
+      "step": 1065
+    },
+    {
+      "epoch": 5.0,
+      "eval_loss": 1.841763973236084,
+      "eval_map": 0.0958,
+      "eval_map_50": 0.208,
+      "eval_map_75": 0.0731,
+      "eval_map_Coverall": 0.3278,
+      "eval_map_Face_Shield": 0.0228,
+      "eval_map_Gloves": 0.031,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.0975,
+      "eval_map_large": 0.1163,
+      "eval_map_medium": 0.0751,
+      "eval_map_small": 0.0194,
+      "eval_mar_1": 0.116,
+      "eval_mar_10": 0.2195,
+      "eval_mar_100": 0.2399,
+      "eval_mar_100_Coverall": 0.6054,
+      "eval_mar_100_Face_Shield": 0.1127,
+      "eval_mar_100_Gloves": 0.2071,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.2742,
+      "eval_mar_large": 0.2939,
+      "eval_mar_medium": 0.1802,
+      "eval_mar_small": 0.095,
+      "eval_runtime": 62.8825,
+      "eval_samples_per_second": 2.385,
+      "eval_steps_per_second": 0.302,
+      "step": 1065
+    },
+    {
+      "epoch": 6.0,
+      "grad_norm": 49.57404327392578,
+      "learning_rate": 2.0187793427230047e-05,
+      "loss": 1.7659,
+      "step": 1278
+    },
+    {
+      "epoch": 6.0,
+      "eval_loss": 1.8737202882766724,
+      "eval_map": 0.1004,
+      "eval_map_50": 0.2399,
+      "eval_map_75": 0.0791,
+      "eval_map_Coverall": 0.2906,
+      "eval_map_Face_Shield": 0.0439,
+      "eval_map_Gloves": 0.0497,
+      "eval_map_Goggles": 0.0,
+      "eval_map_Mask": 0.1178,
+      "eval_map_large": 0.1207,
+      "eval_map_medium": 0.0913,
+      "eval_map_small": 0.0377,
+      "eval_mar_1": 0.1226,
+      "eval_mar_10": 0.2204,
+      "eval_mar_100": 0.2382,
+      "eval_mar_100_Coverall": 0.5856,
+      "eval_mar_100_Face_Shield": 0.1165,
+      "eval_mar_100_Gloves": 0.2281,
+      "eval_mar_100_Goggles": 0.0,
+      "eval_mar_100_Mask": 0.2609,
+      "eval_mar_large": 0.295,
+      "eval_mar_medium": 0.1912,
+      "eval_mar_small": 0.0808,
+      "eval_runtime": 56.4295,
+      "eval_samples_per_second": 2.658,
+      "eval_steps_per_second": 0.337,
+      "step": 1278
+    },
+    {
+      "epoch": 7.0,
+      "grad_norm": 79.07571411132812,
+      "learning_rate": 1.5187793427230049e-05,
+      "loss": 1.6415,
+      "step": 1491
+    },
+    {
+      "epoch": 7.0,
+      "eval_loss": 1.720004677772522,
+      "eval_map": 0.1305,
+      "eval_map_50": 0.2822,
+      "eval_map_75": 0.105,
+      "eval_map_Coverall": 0.3916,
+      "eval_map_Face_Shield": 0.0516,
+      "eval_map_Gloves": 0.0526,
+      "eval_map_Goggles": 0.0059,
+      "eval_map_Mask": 0.1508,
+      "eval_map_large": 0.172,
+      "eval_map_medium": 0.1026,
+      "eval_map_small": 0.044,
+      "eval_mar_1": 0.1505,
+      "eval_mar_10": 0.2689,
+      "eval_mar_100": 0.2891,
+      "eval_mar_100_Coverall": 0.6257,
+      "eval_mar_100_Face_Shield": 0.1962,
+      "eval_mar_100_Gloves": 0.2554,
+      "eval_mar_100_Goggles": 0.0523,
+      "eval_mar_100_Mask": 0.316,
+      "eval_mar_large": 0.386,
+      "eval_mar_medium": 0.2239,
+      "eval_mar_small": 0.1047,
+      "eval_runtime": 63.9942,
+      "eval_samples_per_second": 2.344,
+      "eval_steps_per_second": 0.297,
+      "step": 1491
+    },
+    {
+      "epoch": 8.0,
+      "grad_norm": 68.87022399902344,
+      "learning_rate": 1.0187793427230047e-05,
+      "loss": 1.6405,
+      "step": 1704
+    },
+    {
+      "epoch": 8.0,
+      "eval_loss": 1.6820415258407593,
+      "eval_map": 0.1411,
+      "eval_map_50": 0.2866,
+      "eval_map_75": 0.1303,
+      "eval_map_Coverall": 0.4147,
+      "eval_map_Face_Shield": 0.0549,
+      "eval_map_Gloves": 0.0592,
+      "eval_map_Goggles": 0.0053,
+      "eval_map_Mask": 0.1713,
+      "eval_map_large": 0.1854,
+      "eval_map_medium": 0.1162,
+      "eval_map_small": 0.0542,
+      "eval_mar_1": 0.1572,
+      "eval_mar_10": 0.2867,
+      "eval_mar_100": 0.3088,
+      "eval_mar_100_Coverall": 0.6347,
+      "eval_mar_100_Face_Shield": 0.2468,
+      "eval_mar_100_Gloves": 0.271,
+      "eval_mar_100_Goggles": 0.0631,
+      "eval_mar_100_Mask": 0.3284,
+      "eval_mar_large": 0.4042,
+      "eval_mar_medium": 0.2432,
+      "eval_mar_small": 0.1204,
+      "eval_runtime": 61.79,
+      "eval_samples_per_second": 2.428,
+      "eval_steps_per_second": 0.307,
+      "step": 1704
+    },
+    {
+      "epoch": 9.0,
+      "grad_norm": 28.72616195678711,
+      "learning_rate": 5.211267605633803e-06,
+      "loss": 1.5513,
+      "step": 1917
+    },
+    {
+      "epoch": 9.0,
+      "eval_loss": 1.6379631757736206,
+      "eval_map": 0.1546,
+      "eval_map_50": 0.3144,
+      "eval_map_75": 0.1307,
+      "eval_map_Coverall": 0.4352,
+      "eval_map_Face_Shield": 0.0804,
+      "eval_map_Gloves": 0.0657,
+      "eval_map_Goggles": 0.0062,
+      "eval_map_Mask": 0.1853,
+      "eval_map_large": 0.2131,
+      "eval_map_medium": 0.1215,
+      "eval_map_small": 0.0569,
+      "eval_mar_1": 0.17,
+      "eval_mar_10": 0.3014,
+      "eval_mar_100": 0.323,
+      "eval_mar_100_Coverall": 0.6414,
+      "eval_mar_100_Face_Shield": 0.2544,
+      "eval_mar_100_Gloves": 0.2871,
+      "eval_mar_100_Goggles": 0.0846,
+      "eval_mar_100_Mask": 0.3476,
+      "eval_mar_large": 0.4431,
+      "eval_mar_medium": 0.2563,
+      "eval_mar_small": 0.112,
+      "eval_runtime": 61.6355,
+      "eval_samples_per_second": 2.434,
+      "eval_steps_per_second": 0.308,
+      "step": 1917
+    },
+    {
+      "epoch": 10.0,
+      "grad_norm": 53.006046295166016,
+      "learning_rate": 2.112676056338028e-07,
+      "loss": 1.5564,
+      "step": 2130
+    },
+    {
+      "epoch": 10.0,
+      "eval_loss": 1.623809814453125,
+      "eval_map": 0.1619,
+      "eval_map_50": 0.3226,
+      "eval_map_75": 0.1429,
+      "eval_map_Coverall": 0.4278,
+      "eval_map_Face_Shield": 0.1078,
+      "eval_map_Gloves": 0.0679,
+      "eval_map_Goggles": 0.0102,
+      "eval_map_Mask": 0.1959,
+      "eval_map_large": 0.227,
+      "eval_map_medium": 0.129,
+      "eval_map_small": 0.0501,
+      "eval_mar_1": 0.1773,
+      "eval_mar_10": 0.3167,
+      "eval_mar_100": 0.3392,
+      "eval_mar_100_Coverall": 0.6532,
+      "eval_mar_100_Face_Shield": 0.2937,
+      "eval_mar_100_Gloves": 0.2991,
+      "eval_mar_100_Goggles": 0.0985,
+      "eval_mar_100_Mask": 0.3516,
+      "eval_mar_large": 0.4711,
+      "eval_mar_medium": 0.2626,
+      "eval_mar_small": 0.128,
+      "eval_runtime": 65.5535,
+      "eval_samples_per_second": 2.288,
+      "eval_steps_per_second": 0.29,
+      "step": 2130
+    },
+    {
+      "epoch": 10.0,
+      "step": 2130,
+      "total_flos": 2.2764936536496e+18,
+      "train_loss": 1.393765215806558,
+      "train_runtime": 1396.8073,
+      "train_samples_per_second": 6.085,
+      "train_steps_per_second": 1.525
     }
   ],
   "logging_steps": 500,
-  "max_steps": 426,
+  "max_steps": 2130,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 2,
+  "num_train_epochs": 10,
   "save_steps": 500,
   "stateful_callbacks": {
     "TrainerControl": {
@@ -110,7 +406,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 4.5529873072992e+17,
+  "total_flos": 2.2764936536496e+18,
   "train_batch_size": 4,
   "trial_name": null,
   "trial_params": null