End of training
Browse files
README.md
CHANGED
@@ -15,6 +15,32 @@ should probably proofread and complete it, then remove this comment. -->
|
|
15 |
# rtdetr-r50-cppe5-finetune
|
16 |
|
17 |
This model is a fine-tuned version of [hungnguyen2k4/rtdetr-r50-fruits-finetune](https://huggingface.co/hungnguyen2k4/rtdetr-r50-fruits-finetune) on an unknown dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
## Model description
|
20 |
|
@@ -40,7 +66,18 @@ The following hyperparameters were used during training:
|
|
40 |
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
41 |
- lr_scheduler_type: linear
|
42 |
- lr_scheduler_warmup_steps: 300
|
43 |
-
- num_epochs:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
44 |
|
45 |
### Framework versions
|
46 |
|
|
|
15 |
# rtdetr-r50-cppe5-finetune
|
16 |
|
17 |
This model is a fine-tuned version of [hungnguyen2k4/rtdetr-r50-fruits-finetune](https://huggingface.co/hungnguyen2k4/rtdetr-r50-fruits-finetune) on an unknown dataset.
|
18 |
+
It achieves the following results on the evaluation set:
|
19 |
+
- Loss: 10.2690
|
20 |
+
- Map: 0.5625
|
21 |
+
- Map 50: 0.6875
|
22 |
+
- Map 75: 0.5839
|
23 |
+
- Map Small: 0.2457
|
24 |
+
- Map Medium: 0.5338
|
25 |
+
- Map Large: 0.7163
|
26 |
+
- Mar 1: 0.2862
|
27 |
+
- Mar 10: 0.6012
|
28 |
+
- Mar 100: 0.6875
|
29 |
+
- Mar Small: 0.3606
|
30 |
+
- Mar Medium: 0.6517
|
31 |
+
- Mar Large: 0.8526
|
32 |
+
- Map Apple: 0.5267
|
33 |
+
- Mar 100 Apple: 0.6703
|
34 |
+
- Map Banana: 0.5716
|
35 |
+
- Mar 100 Banana: 0.728
|
36 |
+
- Map Grapes: 0.4784
|
37 |
+
- Mar 100 Grapes: 0.5725
|
38 |
+
- Map Orange: 0.4928
|
39 |
+
- Mar 100 Orange: 0.611
|
40 |
+
- Map Pineapple: 0.5979
|
41 |
+
- Mar 100 Pineapple: 0.7181
|
42 |
+
- Map Watermelon: 0.7077
|
43 |
+
- Mar 100 Watermelon: 0.8253
|
44 |
|
45 |
## Model description
|
46 |
|
|
|
66 |
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
67 |
- lr_scheduler_type: linear
|
68 |
- lr_scheduler_warmup_steps: 300
|
69 |
+
- num_epochs: 5
|
70 |
+
|
71 |
+
### Training results
|
72 |
+
|
73 |
+
| Training Loss | Epoch | Step | Validation Loss | Map | Map 50 | Map 75 | Map Small | Map Medium | Map Large | Mar 1 | Mar 10 | Mar 100 | Mar Small | Mar Medium | Mar Large | Map Apple | Mar 100 Apple | Map Banana | Mar 100 Banana | Map Grapes | Mar 100 Grapes | Map Orange | Mar 100 Orange | Map Pineapple | Mar 100 Pineapple | Map Watermelon | Mar 100 Watermelon |
|
74 |
+
|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:---------:|:-------------:|:----------:|:--------------:|:----------:|:--------------:|:----------:|:--------------:|:-------------:|:-----------------:|:--------------:|:------------------:|
|
75 |
+
| 8.3443 | 1.0 | 750 | 10.1464 | 0.5536 | 0.678 | 0.5795 | 0.252 | 0.5322 | 0.7006 | 0.2857 | 0.6109 | 0.7115 | 0.3705 | 0.6841 | 0.8718 | 0.5243 | 0.6875 | 0.5532 | 0.7416 | 0.4845 | 0.6026 | 0.5187 | 0.6506 | 0.5935 | 0.7558 | 0.6474 | 0.831 |
|
76 |
+
| 8.585 | 2.0 | 1500 | 10.7682 | 0.5491 | 0.6692 | 0.5745 | 0.2354 | 0.5262 | 0.7017 | 0.2807 | 0.5986 | 0.6917 | 0.3547 | 0.6558 | 0.8609 | 0.5213 | 0.6823 | 0.5466 | 0.7152 | 0.4704 | 0.5735 | 0.4877 | 0.6173 | 0.5865 | 0.7275 | 0.6821 | 0.8341 |
|
77 |
+
| 8.3452 | 3.0 | 2250 | 10.2876 | 0.5573 | 0.6809 | 0.5787 | 0.2559 | 0.5387 | 0.7033 | 0.2828 | 0.6037 | 0.6905 | 0.3755 | 0.656 | 0.8498 | 0.5161 | 0.6791 | 0.5681 | 0.736 | 0.4935 | 0.5882 | 0.5031 | 0.6283 | 0.5795 | 0.6877 | 0.6832 | 0.8236 |
|
78 |
+
| 8.0157 | 4.0 | 3000 | 10.3314 | 0.5574 | 0.6809 | 0.5796 | 0.2721 | 0.5321 | 0.7041 | 0.2828 | 0.6024 | 0.6864 | 0.3688 | 0.646 | 0.8524 | 0.5279 | 0.6705 | 0.562 | 0.7246 | 0.4804 | 0.5767 | 0.4949 | 0.6113 | 0.5887 | 0.7199 | 0.6903 | 0.8153 |
|
79 |
+
| 7.7933 | 5.0 | 3750 | 10.2690 | 0.5625 | 0.6875 | 0.5839 | 0.2457 | 0.5338 | 0.7163 | 0.2862 | 0.6012 | 0.6875 | 0.3606 | 0.6517 | 0.8526 | 0.5267 | 0.6703 | 0.5716 | 0.728 | 0.4784 | 0.5725 | 0.4928 | 0.611 | 0.5979 | 0.7181 | 0.7077 | 0.8253 |
|
80 |
+
|
81 |
|
82 |
### Framework versions
|
83 |
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 171567560
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d5a6730b8512d41a4041afda840b3b0cb65cd26ecdc379d9c5d5bfcdea655543
|
3 |
size 171567560
|
runs/Jun11_19-01-25_b3d85beef6c6/events.out.tfevents.1749668494.b3d85beef6c6.19.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec656032058671c9a515940324124f780cacc6419de40aac80fe88035616f56f
|
3 |
+
size 16700
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 5304
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1d22a606ae196c68b637c22c36eec45e827304a24d39946f5227003b6bae0ef4
|
3 |
size 5304
|