Hugging Face's logo

depek
/
detr

Object Detection
Transformers
TensorBoard
Safetensors
detr
Generated from Trainer
Model card Files
xet
Metrics Community
depek commited on
Commit
196fc6a
·
verified ·
1 Parent(s): 2bc947d

End of training

Browse files
Files changed (3) hide show
  1. README.md +15 -11
  2. model.safetensors +1 -1
  3. runs/Apr21_20-56-20_black-ubuntu/events.out.tfevents.1713722180.black-ubuntu +2 -2
README.md CHANGED
@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 1.4507
19
 
20
  ## Model description
21
 
@@ -34,27 +34,31 @@ More information needed
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
- - learning_rate: 1e-05
38
  - train_batch_size: 16
39
  - eval_batch_size: 8
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
- - num_epochs: 4
44
  - mixed_precision_training: Native AMP
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 1.8557 | 0.48 | 300 | 1.8996 |
51
- | 1.7822 | 0.96 | 600 | 1.7404 |
52
- | 1.657 | 1.44 | 900 | 1.6266 |
53
- | 1.5922 | 1.92 | 1200 | 1.5711 |
54
- | 1.5762 | 2.4 | 1500 | 1.5155 |
55
- | 1.5673 | 2.88 | 1800 | 1.4816 |
56
- | 1.4927 | 3.36 | 2100 | 1.4529 |
57
- | 1.5326 | 3.84 | 2400 | 1.4507 |
 
 
 
 
58
 
59
 
60
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.8756
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - learning_rate: 5e-05
38
  - train_batch_size: 16
39
  - eval_batch_size: 8
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
+ - num_epochs: 2
44
  - mixed_precision_training: Native AMP
45
 
46
  ### Training results
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 1.7508 | 0.16 | 300 | 1.7472 |
51
+ | 1.412 | 0.32 | 600 | 1.3611 |
52
+ | 1.2549 | 0.48 | 900 | 1.1682 |
53
+ | 1.0701 | 0.64 | 1200 | 1.0696 |
54
+ | 0.9812 | 0.8 | 1500 | 1.0103 |
55
+ | 1.0346 | 0.96 | 1800 | 0.9782 |
56
+ | 1.014 | 1.12 | 2100 | 0.9429 |
57
+ | 0.9691 | 1.28 | 2400 | 0.9248 |
58
+ | 0.9749 | 1.44 | 2700 | 0.8947 |
59
+ | 0.9707 | 1.6 | 3000 | 0.8887 |
60
+ | 0.938 | 1.76 | 3300 | 0.8889 |
61
+ | 0.9477 | 1.92 | 3600 | 0.8756 |
62
 
63
 
64
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ac1ce898ee656829a0b4928bc5231f74465faa03bcdd64236b2e26a2182abb5
3
  size 166506812
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adb0dc2619846bf014a5ee6759bc2ac55b0d8f25fd8dfe34c1f81efa86b24460
3
  size 166506812
runs/Apr21_20-56-20_black-ubuntu/events.out.tfevents.1713722180.black-ubuntu CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1735a656cee4e4cb08fdcf8e6539f022b0532cfa293c35c7d964b53264be18dc
3
- size 79647
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe42dd2e947298e4bc418427e3b82a2ecaeaaa5d5b7774839a69517ff43421c9
3
+ size 90038