Hugging Face's logo

mrdbourke
/
detr_finetuned_trashify_box_detector

Object Detection
Transformers
Safetensors
conditional_detr
Generated from Trainer
Inference Endpoints
Model card Files Community
mrdbourke commited on
Commit
7913fe9
•
1 Parent(s): 3e55c42

upload trashify object detection model

Browse files
Files changed (3) hide show
  1. README.md +28 -29
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.0728
20
 
21
  ## Model description
22
 
@@ -36,12 +36,11 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
39
- - train_batch_size: 32
40
- - eval_batch_size: 32
41
  - seed: 42
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
- - lr_scheduler_warmup_ratio: 0.05
45
  - num_epochs: 25
46
  - mixed_precision_training: Native AMP
47
 
@@ -49,31 +48,31 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
- | 2.6374 | 1.0 | 25 | 2.2767 |
53
- | 2.3806 | 2.0 | 50 | 2.0779 |
54
- | 2.17 | 3.0 | 75 | 1.8047 |
55
- | 1.9959 | 4.0 | 100 | 1.7031 |
56
- | 1.9129 | 5.0 | 125 | 1.6561 |
57
- | 1.7917 | 6.0 | 150 | 1.5245 |
58
- | 1.6619 | 7.0 | 175 | 1.4002 |
59
- | 1.5913 | 8.0 | 200 | 1.3501 |
60
- | 1.4466 | 9.0 | 225 | 1.2675 |
61
- | 1.3785 | 10.0 | 250 | 1.2594 |
62
- | 1.3542 | 11.0 | 275 | 1.2620 |
63
- | 1.349 | 12.0 | 300 | 1.1957 |
64
- | 1.3055 | 13.0 | 325 | 1.1818 |
65
- | 1.1731 | 14.0 | 350 | 1.1466 |
66
- | 1.1585 | 15.0 | 375 | 1.1627 |
67
- | 1.0614 | 16.0 | 400 | 1.1615 |
68
- | 1.0144 | 17.0 | 425 | 1.1411 |
69
- | 0.967 | 18.0 | 450 | 1.1193 |
70
- | 0.9114 | 19.0 | 475 | 1.1043 |
71
- | 0.9031 | 20.0 | 500 | 1.1005 |
72
- | 0.8719 | 21.0 | 525 | 1.1000 |
73
- | 0.8505 | 22.0 | 550 | 1.0800 |
74
- | 0.8314 | 23.0 | 575 | 1.0739 |
75
- | 0.8178 | 24.0 | 600 | 1.0757 |
76
- | 0.804 | 25.0 | 625 | 1.0728 |
77
 
78
 
79
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.1024
20
 
21
  ## Model description
22
 
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
39
+ - train_batch_size: 16
40
+ - eval_batch_size: 16
41
  - seed: 42
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
 
44
  - num_epochs: 25
45
  - mixed_precision_training: Native AMP
46
 
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss |
50
  |:-------------:|:-----:|:----:|:---------------:|
51
+ | 47.984 | 1.0 | 50 | 3.3624 |
52
+ | 2.7348 | 2.0 | 100 | 2.3693 |
53
+ | 2.2463 | 3.0 | 150 | 1.9785 |
54
+ | 2.1473 | 4.0 | 200 | 1.9305 |
55
+ | 1.9375 | 5.0 | 250 | 1.7268 |
56
+ | 1.7557 | 6.0 | 300 | 1.5692 |
57
+ | 1.6032 | 7.0 | 350 | 1.4597 |
58
+ | 1.5135 | 8.0 | 400 | 1.3900 |
59
+ | 1.4173 | 9.0 | 450 | 1.3377 |
60
+ | 1.3089 | 10.0 | 500 | 1.2696 |
61
+ | 1.2517 | 11.0 | 550 | 1.2276 |
62
+ | 1.1837 | 12.0 | 600 | 1.2114 |
63
+ | 1.136 | 13.0 | 650 | 1.1715 |
64
+ | 1.0892 | 14.0 | 700 | 1.1834 |
65
+ | 1.046 | 15.0 | 750 | 1.1676 |
66
+ | 0.986 | 16.0 | 800 | 1.1197 |
67
+ | 0.912 | 17.0 | 850 | 1.1203 |
68
+ | 0.8765 | 18.0 | 900 | 1.1224 |
69
+ | 0.836 | 19.0 | 950 | 1.0934 |
70
+ | 0.8173 | 20.0 | 1000 | 1.1082 |
71
+ | 0.7742 | 21.0 | 1050 | 1.0997 |
72
+ | 0.7471 | 22.0 | 1100 | 1.0991 |
73
+ | 0.7351 | 23.0 | 1150 | 1.1083 |
74
+ | 0.7151 | 24.0 | 1200 | 1.0987 |
75
+ | 0.7052 | 25.0 | 1250 | 1.1024 |
76
 
77
 
78
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d50b5beb7b5f8a524a56542ed9793be6fa4dbe61cf73ba5eeac0a97e2a218c1a
3
  size 174081852
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b4234f8bddb6b3aa5a9fc5e8667b9883bb3955d0c50f06a57c7063fb639e7d0
3
  size 174081852
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a03ea92e513daa5b2ede961727b411077de93967808efeb6d3214da08fb083be
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff66dec1f16d86504daf9ec3819bf3beda20cd5534543cdb6bb247a59a4090ae
3
  size 5240