amyeroberts HF staff commited on
Commit
ac112ba
1 Parent(s): 34adb59

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -10
README.md CHANGED
@@ -2,8 +2,6 @@
2
  license: apache-2.0
3
  base_model: facebook/detr-resnet-50
4
  tags:
5
- - object-detection
6
- - vision
7
  - generated_from_trainer
8
  model-index:
9
  - name: detr-resnet-50-base-coco
@@ -15,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # detr-resnet-50-base-coco
17
 
18
- 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.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 6719.0215
21
 
22
  ## Model description
23
 
@@ -36,7 +34,7 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - learning_rate: 2e-05
40
  - train_batch_size: 8
41
  - eval_batch_size: 8
42
  - seed: 1337
@@ -48,11 +46,11 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss |
50
  |:-------------:|:-----:|:----:|:---------------:|
51
- | 6457.3832 | 1.0 | 128 | 6720.3701 |
52
- | 6234.0246 | 2.0 | 256 | 6719.1660 |
53
- | 6335.5293 | 3.0 | 384 | 6719.0308 |
54
- | 6520.8047 | 4.0 | 512 | 6719.0605 |
55
- | 5738.273 | 5.0 | 640 | 6719.0215 |
56
 
57
 
58
  ### Framework versions
 
2
  license: apache-2.0
3
  base_model: facebook/detr-resnet-50
4
  tags:
 
 
5
  - generated_from_trainer
6
  model-index:
7
  - name: detr-resnet-50-base-coco
 
13
 
14
  # detr-resnet-50-base-coco
15
 
16
+ This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 6729.0347
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - learning_rate: 2e-07
38
  - train_batch_size: 8
39
  - eval_batch_size: 8
40
  - seed: 1337
 
46
 
47
  | Training Loss | Epoch | Step | Validation Loss |
48
  |:-------------:|:-----:|:----:|:---------------:|
49
+ | 6463.6664 | 1.0 | 128 | 6729.2021 |
50
+ | 6242.4801 | 2.0 | 256 | 6729.1201 |
51
+ | 6344.1949 | 3.0 | 384 | 6729.0586 |
52
+ | 6528.6664 | 4.0 | 512 | 6729.0420 |
53
+ | 5746.1234 | 5.0 | 640 | 6729.0347 |
54
 
55
 
56
  ### Framework versions