Hugging Face's logo

kacper-cierzniewski
/
daigram_detr_r50_albumentations

Object Detection
Transformers
PyTorch
TensorBoard
detr
Generated from Trainer
Model card Files
xet
Metrics Community
1
kacper-cierzniewski commited on
Commit
43008ae
·
1 Parent(s): 78b7af2

End of training

Browse files
Files changed (2) hide show
  1. README.md +198 -0
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -16,6 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
  # daigram_detr_r50_albumentations
17
 
18
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the bpmn-shapes dataset.
 
 
19
 
20
  ## Model description
21
 
@@ -42,6 +44,202 @@ The following hyperparameters were used during training:
42
  - lr_scheduler_type: linear
43
  - num_epochs: 500
44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  ### Framework versions
46
 
47
  - Transformers 4.34.1
 
16
  # daigram_detr_r50_albumentations
17
 
18
  This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the bpmn-shapes dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 1.0088
21
 
22
  ## Model description
23
 
 
44
  - lr_scheduler_type: linear
45
  - num_epochs: 500
46
 
47
+ ### Training results
48
+
49
+ | Training Loss | Epoch | Step | Validation Loss |
50
+ |:-------------:|:------:|:----:|:---------------:|
51
+ | 3.8163 | 2.63 | 50 | 3.0660 |
52
+ | 2.9036 | 5.26 | 100 | 2.8878 |
53
+ | 2.7516 | 7.89 | 150 | 2.8043 |
54
+ | 2.6278 | 10.53 | 200 | 2.6820 |
55
+ | 2.4806 | 13.16 | 250 | 2.5676 |
56
+ | 2.3781 | 15.79 | 300 | 2.4282 |
57
+ | 2.253 | 18.42 | 350 | 2.3161 |
58
+ | 2.1405 | 21.05 | 400 | 2.1735 |
59
+ | 2.0263 | 23.68 | 450 | 2.0909 |
60
+ | 1.9732 | 26.32 | 500 | 2.0120 |
61
+ | 1.8647 | 28.95 | 550 | 1.9260 |
62
+ | 1.7793 | 31.58 | 600 | 1.8655 |
63
+ | 1.7706 | 34.21 | 650 | 1.8166 |
64
+ | 1.6792 | 36.84 | 700 | 1.7325 |
65
+ | 1.5654 | 39.47 | 750 | 1.7061 |
66
+ | 1.5802 | 42.11 | 800 | 1.6463 |
67
+ | 1.5053 | 44.74 | 850 | 1.5985 |
68
+ | 1.4858 | 47.37 | 900 | 1.6060 |
69
+ | 1.4186 | 50.0 | 950 | 1.5563 |
70
+ | 1.4391 | 52.63 | 1000 | 1.5219 |
71
+ | 1.3938 | 55.26 | 1050 | 1.4995 |
72
+ | 1.3734 | 57.89 | 1100 | 1.4661 |
73
+ | 1.3379 | 60.53 | 1150 | 1.4451 |
74
+ | 1.341 | 63.16 | 1200 | 1.4854 |
75
+ | 1.3647 | 65.79 | 1250 | 1.4509 |
76
+ | 1.3198 | 68.42 | 1300 | 1.4116 |
77
+ | 1.3054 | 71.05 | 1350 | 1.3821 |
78
+ | 1.2945 | 73.68 | 1400 | 1.3952 |
79
+ | 1.2899 | 76.32 | 1450 | 1.3868 |
80
+ | 1.2533 | 78.95 | 1500 | 1.3580 |
81
+ | 1.2655 | 81.58 | 1550 | 1.3374 |
82
+ | 1.2649 | 84.21 | 1600 | 1.3451 |
83
+ | 1.2286 | 86.84 | 1650 | 1.2973 |
84
+ | 1.2497 | 89.47 | 1700 | 1.3322 |
85
+ | 1.2456 | 92.11 | 1750 | 1.3289 |
86
+ | 1.2234 | 94.74 | 1800 | 1.3080 |
87
+ | 1.1695 | 97.37 | 1850 | 1.3218 |
88
+ | 1.2265 | 100.0 | 1900 | 1.3280 |
89
+ | 1.1899 | 102.63 | 1950 | 1.2834 |
90
+ | 1.1914 | 105.26 | 2000 | 1.2931 |
91
+ | 1.1698 | 107.89 | 2050 | 1.3176 |
92
+ | 1.177 | 110.53 | 2100 | 1.2896 |
93
+ | 1.1625 | 113.16 | 2150 | 1.2936 |
94
+ | 1.1626 | 115.79 | 2200 | 1.2614 |
95
+ | 1.1698 | 118.42 | 2250 | 1.2545 |
96
+ | 1.1703 | 121.05 | 2300 | 1.2398 |
97
+ | 1.1659 | 123.68 | 2350 | 1.2254 |
98
+ | 1.1734 | 126.32 | 2400 | 1.2489 |
99
+ | 1.1234 | 128.95 | 2450 | 1.2072 |
100
+ | 1.1464 | 131.58 | 2500 | 1.1707 |
101
+ | 1.1268 | 134.21 | 2550 | 1.1971 |
102
+ | 1.1511 | 136.84 | 2600 | 1.2247 |
103
+ | 1.1234 | 139.47 | 2650 | 1.1921 |
104
+ | 1.0923 | 142.11 | 2700 | 1.1751 |
105
+ | 1.1267 | 144.74 | 2750 | 1.1905 |
106
+ | 1.1021 | 147.37 | 2800 | 1.1885 |
107
+ | 1.1075 | 150.0 | 2850 | 1.1780 |
108
+ | 1.1116 | 152.63 | 2900 | 1.1666 |
109
+ | 1.0987 | 155.26 | 2950 | 1.1694 |
110
+ | 1.0974 | 157.89 | 3000 | 1.1931 |
111
+ | 1.0867 | 160.53 | 3050 | 1.1461 |
112
+ | 1.1076 | 163.16 | 3100 | 1.1501 |
113
+ | 1.0912 | 165.79 | 3150 | 1.1611 |
114
+ | 1.0671 | 168.42 | 3200 | 1.1718 |
115
+ | 1.0981 | 171.05 | 3250 | 1.1961 |
116
+ | 1.0602 | 173.68 | 3300 | 1.1786 |
117
+ | 1.0305 | 176.32 | 3350 | 1.1640 |
118
+ | 1.0647 | 178.95 | 3400 | 1.1416 |
119
+ | 1.0628 | 181.58 | 3450 | 1.1296 |
120
+ | 1.0856 | 184.21 | 3500 | 1.1140 |
121
+ | 1.0626 | 186.84 | 3550 | 1.1214 |
122
+ | 1.0782 | 189.47 | 3600 | 1.1449 |
123
+ | 1.0601 | 192.11 | 3650 | 1.1441 |
124
+ | 1.0906 | 194.74 | 3700 | 1.1396 |
125
+ | 1.0376 | 197.37 | 3750 | 1.1271 |
126
+ | 1.0625 | 200.0 | 3800 | 1.1397 |
127
+ | 1.057 | 202.63 | 3850 | 1.1121 |
128
+ | 1.0448 | 205.26 | 3900 | 1.1376 |
129
+ | 1.0747 | 207.89 | 3950 | 1.1475 |
130
+ | 1.0605 | 210.53 | 4000 | 1.0916 |
131
+ | 1.0344 | 213.16 | 4050 | 1.1001 |
132
+ | 1.0443 | 215.79 | 4100 | 1.0976 |
133
+ | 1.0202 | 218.42 | 4150 | 1.1240 |
134
+ | 1.078 | 221.05 | 4200 | 1.1024 |
135
+ | 1.0251 | 223.68 | 4250 | 1.0793 |
136
+ | 1.0353 | 226.32 | 4300 | 1.1153 |
137
+ | 1.0047 | 228.95 | 4350 | 1.0972 |
138
+ | 1.0143 | 231.58 | 4400 | 1.0948 |
139
+ | 1.0172 | 234.21 | 4450 | 1.1265 |
140
+ | 1.0299 | 236.84 | 4500 | 1.1038 |
141
+ | 0.9968 | 239.47 | 4550 | 1.0901 |
142
+ | 1.0233 | 242.11 | 4600 | 1.0945 |
143
+ | 0.9943 | 244.74 | 4650 | 1.0918 |
144
+ | 1.0321 | 247.37 | 4700 | 1.1270 |
145
+ | 1.0113 | 250.0 | 4750 | 1.1060 |
146
+ | 1.0229 | 252.63 | 4800 | 1.0859 |
147
+ | 0.9945 | 255.26 | 4850 | 1.0875 |
148
+ | 1.0073 | 257.89 | 4900 | 1.0976 |
149
+ | 1.0096 | 260.53 | 4950 | 1.0933 |
150
+ | 1.0 | 263.16 | 5000 | 1.0821 |
151
+ | 1.0326 | 265.79 | 5050 | 1.0747 |
152
+ | 0.997 | 268.42 | 5100 | 1.0931 |
153
+ | 1.0056 | 271.05 | 5150 | 1.0853 |
154
+ | 0.9858 | 273.68 | 5200 | 1.0945 |
155
+ | 1.0005 | 276.32 | 5250 | 1.0669 |
156
+ | 1.0217 | 278.95 | 5300 | 1.0497 |
157
+ | 0.9777 | 281.58 | 5350 | 1.0672 |
158
+ | 0.9888 | 284.21 | 5400 | 1.0844 |
159
+ | 0.9662 | 286.84 | 5450 | 1.0524 |
160
+ | 1.0029 | 289.47 | 5500 | 1.0519 |
161
+ | 0.984 | 292.11 | 5550 | 1.0538 |
162
+ | 0.9724 | 294.74 | 5600 | 1.0524 |
163
+ | 0.991 | 297.37 | 5650 | 1.0553 |
164
+ | 0.9936 | 300.0 | 5700 | 1.0601 |
165
+ | 0.9817 | 302.63 | 5750 | 1.0524 |
166
+ | 0.9868 | 305.26 | 5800 | 1.0644 |
167
+ | 0.9982 | 307.89 | 5850 | 1.0523 |
168
+ | 0.9814 | 310.53 | 5900 | 1.0611 |
169
+ | 0.9761 | 313.16 | 5950 | 1.0505 |
170
+ | 0.9507 | 315.79 | 6000 | 1.0361 |
171
+ | 0.9786 | 318.42 | 6050 | 1.0275 |
172
+ | 0.9684 | 321.05 | 6100 | 1.0292 |
173
+ | 0.9759 | 323.68 | 6150 | 1.0529 |
174
+ | 0.9442 | 326.32 | 6200 | 1.0689 |
175
+ | 0.9653 | 328.95 | 6250 | 1.0696 |
176
+ | 0.9579 | 331.58 | 6300 | 1.0572 |
177
+ | 1.0016 | 334.21 | 6350 | 1.0660 |
178
+ | 0.9462 | 336.84 | 6400 | 1.0525 |
179
+ | 0.9596 | 339.47 | 6450 | 1.0505 |
180
+ | 0.9655 | 342.11 | 6500 | 1.0514 |
181
+ | 0.9713 | 344.74 | 6550 | 1.0616 |
182
+ | 0.952 | 347.37 | 6600 | 1.0497 |
183
+ | 0.9433 | 350.0 | 6650 | 1.0389 |
184
+ | 0.9619 | 352.63 | 6700 | 1.0404 |
185
+ | 0.9594 | 355.26 | 6750 | 1.0332 |
186
+ | 0.9586 | 357.89 | 6800 | 1.0323 |
187
+ | 0.9582 | 360.53 | 6850 | 1.0294 |
188
+ | 0.9437 | 363.16 | 6900 | 1.0329 |
189
+ | 0.9585 | 365.79 | 6950 | 1.0361 |
190
+ | 0.9661 | 368.42 | 7000 | 1.0428 |
191
+ | 0.9603 | 371.05 | 7050 | 1.0299 |
192
+ | 0.9619 | 373.68 | 7100 | 1.0416 |
193
+ | 0.9766 | 376.32 | 7150 | 1.0471 |
194
+ | 0.9547 | 378.95 | 7200 | 1.0498 |
195
+ | 0.967 | 381.58 | 7250 | 1.0318 |
196
+ | 0.9463 | 384.21 | 7300 | 1.0238 |
197
+ | 0.9531 | 386.84 | 7350 | 1.0329 |
198
+ | 0.9342 | 389.47 | 7400 | 1.0354 |
199
+ | 0.939 | 392.11 | 7450 | 1.0312 |
200
+ | 0.9635 | 394.74 | 7500 | 1.0325 |
201
+ | 0.9261 | 397.37 | 7550 | 1.0245 |
202
+ | 0.962 | 400.0 | 7600 | 1.0381 |
203
+ | 0.9385 | 402.63 | 7650 | 1.0243 |
204
+ | 0.9422 | 405.26 | 7700 | 1.0235 |
205
+ | 0.9285 | 407.89 | 7750 | 1.0286 |
206
+ | 0.9598 | 410.53 | 7800 | 1.0353 |
207
+ | 0.9529 | 413.16 | 7850 | 1.0361 |
208
+ | 0.928 | 415.79 | 7900 | 1.0316 |
209
+ | 0.935 | 418.42 | 7950 | 1.0263 |
210
+ | 0.9456 | 421.05 | 8000 | 1.0368 |
211
+ | 0.9387 | 423.68 | 8050 | 1.0440 |
212
+ | 0.9321 | 426.32 | 8100 | 1.0440 |
213
+ | 0.9236 | 428.95 | 8150 | 1.0394 |
214
+ | 0.9448 | 431.58 | 8200 | 1.0467 |
215
+ | 0.9151 | 434.21 | 8250 | 1.0516 |
216
+ | 0.9373 | 436.84 | 8300 | 1.0383 |
217
+ | 0.9577 | 439.47 | 8350 | 1.0190 |
218
+ | 0.9199 | 442.11 | 8400 | 1.0215 |
219
+ | 0.9321 | 444.74 | 8450 | 1.0184 |
220
+ | 0.9387 | 447.37 | 8500 | 1.0236 |
221
+ | 0.9382 | 450.0 | 8550 | 1.0259 |
222
+ | 0.9391 | 452.63 | 8600 | 1.0282 |
223
+ | 0.9392 | 455.26 | 8650 | 1.0193 |
224
+ | 0.9438 | 457.89 | 8700 | 1.0124 |
225
+ | 0.9398 | 460.53 | 8750 | 1.0060 |
226
+ | 0.9246 | 463.16 | 8800 | 1.0140 |
227
+ | 0.9383 | 465.79 | 8850 | 1.0145 |
228
+ | 0.9267 | 468.42 | 8900 | 1.0122 |
229
+ | 0.9253 | 471.05 | 8950 | 1.0144 |
230
+ | 0.9238 | 473.68 | 9000 | 1.0065 |
231
+ | 0.9082 | 476.32 | 9050 | 1.0136 |
232
+ | 0.9287 | 478.95 | 9100 | 1.0120 |
233
+ | 0.9161 | 481.58 | 9150 | 1.0120 |
234
+ | 0.9093 | 484.21 | 9200 | 1.0128 |
235
+ | 0.9264 | 486.84 | 9250 | 1.0125 |
236
+ | 0.9487 | 489.47 | 9300 | 1.0131 |
237
+ | 0.9398 | 492.11 | 9350 | 1.0101 |
238
+ | 0.9039 | 494.74 | 9400 | 1.0090 |
239
+ | 0.908 | 497.37 | 9450 | 1.0097 |
240
+ | 0.944 | 500.0 | 9500 | 1.0088 |
241
+
242
+
243
  ### Framework versions
244
 
245
  - Transformers 4.34.1
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2b6ffc11af883763c85047e269d71b9d40a8a90bb728d2108b5252bd143276bd
3
  size 166620821
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e26014faeb565afc7b2a82b8b1f16de21dd649fb10d76e95368c678d1d9d78e7
3
  size 166620821