End of training

README.md

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 3.2053
+- Loss: 2.3771
 
 ## Model description
 
@@ -34,7 +34,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
+- learning_rate: 2e-05
 - train_batch_size: 32
 - eval_batch_size: 8
 - seed: 42
@@ -47,9 +47,12 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 4.3863        | 0.32  | 100  | 4.0305          |
-| 3.5436        | 0.64  | 200  | 3.4199          |
-| 3.1853        | 0.96  | 300  | 3.2053          |
+| 4.3956        | 0.16  | 50   | 4.0045          |
+| 3.5668        | 0.32  | 100  | 3.3500          |
+| 2.9609        | 0.48  | 150  | 2.9119          |
+| 2.6387        | 0.64  | 200  | 2.5813          |
+| 2.411         | 0.8   | 250  | 2.4280          |
+| 2.3426        | 0.96  | 300  | 2.3771          |
 
 
 ### Framework versions

config.json

@@ -122,19 +122,19 @@
   "giou_cost": 2,
   "giou_loss_coefficient": 2,
   "id2label": {
-    "0": "Clothing",
-    "1": "Shoes",
-    "2": "Bags",
-    "3": "Accessories"
+    "0": "Accessories",
+    "1": "Bags",
+    "2": "Clothing",
+    "3": "Shoes"
   },
   "init_std": 0.02,
   "init_xavier_std": 1.0,
   "is_encoder_decoder": true,
   "label2id": {
-    "Accessories": 3,
-    "Bags": 2,
-    "Clothing": 0,
-    "Shoes": 1
+    "Accessories": 0,
+    "Bags": 1,
+    "Clothing": 2,
+    "Shoes": 3
   },
   "mask_loss_coefficient": 1,
   "model_type": "detr",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7150742a94cb6cccdce3868c1886246f72fbc212335b280808fdc82c4cf063b7
+oid sha256:a7433f57bc988d114e0fee32e66a94be8ea2cfa49099553c31c715856f3e8ba1
 size 166506812

runs/Apr21_20-30-33_493b988ffc43/events.out.tfevents.1713731438.493b988ffc43.13595.5

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:64e76d88a2e849843a9144515c21cb1ee3c84c43c68fe4ae62d316e741fb36a7
+oid sha256:41cc772c57a932956e2cb481e2ba87084c5abd3e804d0105c24d5d15a0bca46b
 size 4856