update model card README.md

README.md

@@ -2,8 +2,6 @@
 license: apache-2.0
 base_model: facebook/detr-resnet-50
 tags:
-- object-detection
-- vision
 - generated_from_trainer
 model-index:
 - name: detr-resnet-50-base-coco
@@ -15,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # detr-resnet-50-base-coco
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the detection-datasets/coco dataset.
+This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 6395.8599
+- Loss: 6387.9546
 
 ## Model description
 
@@ -36,28 +34,38 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
+- learning_rate: 0.0002
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 1337
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 10.0
+- num_epochs: 20.0
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 4    | 6398.5435       |
-| No log        | 2.0   | 8    | 6397.9766       |
-| 6175.8828     | 3.0   | 12   | 6397.4253       |
-| 6175.8828     | 4.0   | 16   | 6396.9629       |
-| 6292.3258     | 5.0   | 20   | 6396.5718       |
-| 6292.3258     | 6.0   | 24   | 6396.2788       |
-| 6292.3258     | 7.0   | 28   | 6396.0767       |
-| 6230.4844     | 8.0   | 32   | 6395.9600       |
-| 6230.4844     | 9.0   | 36   | 6395.8672       |
-| 6089.8695     | 10.0  | 40   | 6395.8599       |
+| No log        | 1.0   | 4    | 6394.3193       |
+| No log        | 2.0   | 8    | 6393.3564       |
+| 6173.6062     | 3.0   | 12   | 6390.9482       |
+| 6173.6062     | 4.0   | 16   | 6389.5869       |
+| 6287.2859     | 5.0   | 20   | 6388.7280       |
+| 6287.2859     | 6.0   | 24   | 6388.6221       |
+| 6287.2859     | 7.0   | 28   | 6388.4399       |
+| 6224.3344     | 8.0   | 32   | 6388.2119       |
+| 6224.3344     | 9.0   | 36   | 6388.5449       |
+| 6083.1363     | 10.0  | 40   | 6388.3262       |
+| 6083.1363     | 11.0  | 44   | 6390.2056       |
+| 6083.1363     | 12.0  | 48   | 6388.6606       |
+| 6260.7426     | 13.0  | 52   | 6388.1484       |
+| 6260.7426     | 14.0  | 56   | 6388.0479       |
+| 6377.543      | 15.0  | 60   | 6388.0303       |
+| 6377.543      | 16.0  | 64   | 6388.1338       |
+| 6377.543      | 17.0  | 68   | 6388.0596       |
+| 6318.3168     | 18.0  | 72   | 6388.0195       |
+| 6318.3168     | 19.0  | 76   | 6387.9639       |
+| 6258.2133     | 20.0  | 80   | 6387.9546       |
 
 
 ### Framework versions