Hugging Face's logo

eleninaneversmiles
/
segformer-b0-finetuned-segments-sidewalk-2

Image Segmentation
Transformers
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Community
eleninaneversmiles commited on
Commit
df7e8d1
•
1 Parent(s): b5ffde8

End of training

Browse files
Files changed (4) hide show
  1. README.md +57 -80
  2. config.json +4 -2
  3. model.safetensors +2 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -1,10 +1,10 @@
1
  ---
2
  license: other
 
3
  tags:
4
  - vision
5
  - image-segmentation
6
  - generated_from_trainer
7
- base_model: nvidia/mit-b0
8
  model-index:
9
  - name: segformer-b0-finetuned-segments-sidewalk-2
10
  results: []
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the eleninaneversmiles/wheels dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.1170
21
 
22
  ## Model description
23
 
@@ -42,87 +42,64 @@ The following hyperparameters were used during training:
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - num_epochs: 300
46
 
47
  ### Training results
48
 
49
- | Training Loss | Epoch | Step | Validation Loss |
50
- |:-------------:|:-----:|:----:|:---------------:|
51
- | 2.9481 | 4.0 | 20 | 3.5435 |
52
- | 2.6731 | 8.0 | 40 | 3.0067 |
53
- | 2.6124 | 12.0 | 60 | 2.4952 |
54
- | 2.0959 | 16.0 | 80 | 2.0672 |
55
- | 2.0621 | 20.0 | 100 | 1.8896 |
56
- | 1.8677 | 24.0 | 120 | 1.6907 |
57
- | 1.7867 | 28.0 | 140 | 1.5101 |
58
- | 1.3991 | 32.0 | 160 | 1.3829 |
59
- | 1.1594 | 36.0 | 180 | 1.2332 |
60
- | 1.0339 | 40.0 | 200 | 1.0889 |
61
- | 0.8877 | 44.0 | 220 | 0.9699 |
62
- | 0.9484 | 48.0 | 240 | 0.8998 |
63
- | 0.8533 | 52.0 | 260 | 0.7505 |
64
- | 0.5803 | 56.0 | 280 | 0.6548 |
65
- | 0.5161 | 60.0 | 300 | 0.5719 |
66
- | 0.4307 | 64.0 | 320 | 0.5128 |
67
- | 0.3615 | 68.0 | 340 | 0.4228 |
68
- | 0.3185 | 72.0 | 360 | 0.3733 |
69
- | 0.2852 | 76.0 | 380 | 0.3421 |
70
- | 0.299 | 80.0 | 400 | 0.3033 |
71
- | 0.2014 | 84.0 | 420 | 0.2814 |
72
- | 0.2139 | 88.0 | 440 | 0.2346 |
73
- | 0.147 | 92.0 | 460 | 0.2176 |
74
- | 0.1386 | 96.0 | 480 | 0.2030 |
75
- | 0.1189 | 100.0 | 500 | 0.1950 |
76
- | 0.1527 | 104.0 | 520 | 0.1804 |
77
- | 0.093 | 108.0 | 540 | 0.1708 |
78
- | 0.0934 | 112.0 | 560 | 0.1712 |
79
- | 0.0878 | 116.0 | 580 | 0.1582 |
80
- | 0.1096 | 120.0 | 600 | 0.1553 |
81
- | 0.0743 | 124.0 | 620 | 0.1460 |
82
- | 0.0677 | 128.0 | 640 | 0.1413 |
83
- | 0.0661 | 132.0 | 660 | 0.1399 |
84
- | 0.0619 | 136.0 | 680 | 0.1359 |
85
- | 0.0617 | 140.0 | 700 | 0.1318 |
86
- | 0.064 | 144.0 | 720 | 0.1316 |
87
- | 0.0542 | 148.0 | 740 | 0.1309 |
88
- | 0.0584 | 152.0 | 760 | 0.1286 |
89
- | 0.0525 | 156.0 | 780 | 0.1298 |
90
- | 0.069 | 160.0 | 800 | 0.1283 |
91
- | 0.05 | 164.0 | 820 | 0.1270 |
92
- | 0.0497 | 168.0 | 840 | 0.1240 |
93
- | 0.0478 | 172.0 | 860 | 0.1231 |
94
- | 0.0472 | 176.0 | 880 | 0.1190 |
95
- | 0.0457 | 180.0 | 900 | 0.1207 |
96
- | 0.0435 | 184.0 | 920 | 0.1221 |
97
- | 0.0427 | 188.0 | 940 | 0.1212 |
98
- | 0.0425 | 192.0 | 960 | 0.1197 |
99
- | 0.0481 | 196.0 | 980 | 0.1199 |
100
- | 0.0484 | 200.0 | 1000 | 0.1210 |
101
- | 0.0553 | 204.0 | 1020 | 0.1197 |
102
- | 0.0389 | 208.0 | 1040 | 0.1177 |
103
- | 0.0391 | 212.0 | 1060 | 0.1181 |
104
- | 0.0491 | 216.0 | 1080 | 0.1226 |
105
- | 0.038 | 220.0 | 1100 | 0.1189 |
106
- | 0.0443 | 224.0 | 1120 | 0.1177 |
107
- | 0.038 | 228.0 | 1140 | 0.1178 |
108
- | 0.0365 | 232.0 | 1160 | 0.1186 |
109
- | 0.044 | 236.0 | 1180 | 0.1174 |
110
- | 0.0608 | 240.0 | 1200 | 0.1205 |
111
- | 0.0356 | 244.0 | 1220 | 0.1200 |
112
- | 0.0431 | 248.0 | 1240 | 0.1195 |
113
- | 0.0414 | 252.0 | 1260 | 0.1183 |
114
- | 0.0374 | 256.0 | 1280 | 0.1173 |
115
- | 0.0389 | 260.0 | 1300 | 0.1185 |
116
- | 0.0335 | 264.0 | 1320 | 0.1181 |
117
- | 0.0363 | 268.0 | 1340 | 0.1182 |
118
- | 0.0408 | 272.0 | 1360 | 0.1191 |
119
- | 0.0411 | 276.0 | 1380 | 0.1186 |
120
- | 0.037 | 280.0 | 1400 | 0.1176 |
121
- | 0.0424 | 284.0 | 1420 | 0.1185 |
122
- | 0.04 | 288.0 | 1440 | 0.1200 |
123
- | 0.0399 | 292.0 | 1460 | 0.1187 |
124
- | 0.0415 | 296.0 | 1480 | 0.1186 |
125
- | 0.038 | 300.0 | 1500 | 0.1170 |
126
 
127
 
128
  ### Framework versions
 
1
  ---
2
  license: other
3
+ base_model: nvidia/mit-b0
4
  tags:
5
  - vision
6
  - image-segmentation
7
  - generated_from_trainer
 
8
  model-index:
9
  - name: segformer-b0-finetuned-segments-sidewalk-2
10
  results: []
 
17
 
18
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the eleninaneversmiles/wheels dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.1287
21
 
22
  ## Model description
23
 
 
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 150
46
 
47
  ### Training results
48
 
49
+ | Training Loss | Epoch | Step | Validation Loss |
50
+ |:-------------:|:--------:|:----:|:---------------:|
51
+ | 2.9957 | 2.8571 | 20 | 3.4269 |
52
+ | 2.6593 | 5.7143 | 40 | 2.3621 |
53
+ | 1.9746 | 8.5714 | 60 | 1.2378 |
54
+ | 1.5998 | 11.4286 | 80 | 1.2329 |
55
+ | 1.3299 | 14.2857 | 100 | 0.8019 |
56
+ | 1.3781 | 17.1429 | 120 | 0.8478 |
57
+ | 2.1912 | 20.0 | 140 | 0.6386 |
58
+ | 1.0362 | 22.8571 | 160 | 0.6467 |
59
+ | 1.3817 | 25.7143 | 180 | 0.4496 |
60
+ | 0.8108 | 28.5714 | 200 | 0.4032 |
61
+ | 0.8187 | 31.4286 | 220 | 0.4650 |
62
+ | 0.6671 | 34.2857 | 240 | 0.3251 |
63
+ | 0.6062 | 37.1429 | 260 | 0.4035 |
64
+ | 1.4152 | 40.0 | 280 | 0.3076 |
65
+ | 1.3078 | 42.8571 | 300 | 0.2517 |
66
+ | 0.4267 | 45.7143 | 320 | 0.2405 |
67
+ | 0.5829 | 48.5714 | 340 | 0.2142 |
68
+ | 0.8742 | 51.4286 | 360 | 0.2055 |
69
+ | 0.3055 | 54.2857 | 380 | 0.2257 |
70
+ | 0.5966 | 57.1429 | 400 | 0.1559 |
71
+ | 0.5006 | 60.0 | 420 | 0.1927 |
72
+ | 0.4433 | 62.8571 | 440 | 0.1525 |
73
+ | 0.2377 | 65.7143 | 460 | 0.1597 |
74
+ | 0.2612 | 68.5714 | 480 | 0.1703 |
75
+ | 0.477 | 71.4286 | 500 | 0.1663 |
76
+ | 0.2006 | 74.2857 | 520 | 0.1427 |
77
+ | 0.2641 | 77.1429 | 540 | 0.1370 |
78
+ | 0.5154 | 80.0 | 560 | 0.1386 |
79
+ | 0.447 | 82.8571 | 580 | 0.1274 |
80
+ | 0.195 | 85.7143 | 600 | 0.1236 |
81
+ | 0.1643 | 88.5714 | 620 | 0.1420 |
82
+ | 0.4199 | 91.4286 | 640 | 0.1226 |
83
+ | 0.1644 | 94.2857 | 660 | 0.1419 |
84
+ | 0.312 | 97.1429 | 680 | 0.1365 |
85
+ | 0.3905 | 100.0 | 700 | 0.1181 |
86
+ | 0.4035 | 102.8571 | 720 | 0.1305 |
87
+ | 0.1411 | 105.7143 | 740 | 0.1262 |
88
+ | 0.3018 | 108.5714 | 760 | 0.1322 |
89
+ | 0.1332 | 111.4286 | 780 | 0.1317 |
90
+ | 0.303 | 114.2857 | 800 | 0.1205 |
91
+ | 0.2399 | 117.1429 | 820 | 0.1358 |
92
+ | 0.2488 | 120.0 | 840 | 0.1226 |
93
+ | 0.304 | 122.8571 | 860 | 0.1275 |
94
+ | 0.2278 | 125.7143 | 880 | 0.1280 |
95
+ | 0.2718 | 128.5714 | 900 | 0.1294 |
96
+ | 0.5304 | 131.4286 | 920 | 0.1320 |
97
+ | 0.1143 | 134.2857 | 940 | 0.1279 |
98
+ | 0.1075 | 137.1429 | 960 | 0.1258 |
99
+ | 0.2103 | 140.0 | 980 | 0.1349 |
100
+ | 0.1483 | 142.8571 | 1000 | 0.1230 |
101
+ | 0.287 | 145.7143 | 1020 | 0.1253 |
102
+ | 0.3606 | 148.5714 | 1040 | 0.1287 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
103
 
104
 
105
  ### Framework versions
config.json CHANGED
@@ -74,12 +74,14 @@
74
  "44": 1,
75
  "45": 1,
76
  "46": 1,
77
- "47": 1
 
 
78
  },
79
  "image_size": 224,
80
  "initializer_range": 0.02,
81
  "label2id": {
82
- "1": 47
83
  },
84
  "layer_norm_eps": 1e-06,
85
  "mlp_ratios": [
 
74
  "44": 1,
75
  "45": 1,
76
  "46": 1,
77
+ "47": 1,
78
+ "48": 1,
79
+ "49": 1
80
  },
81
  "image_size": 224,
82
  "initializer_range": 0.02,
83
  "label2id": {
84
+ "1": 49
85
  },
86
  "layer_norm_eps": 1e-06,
87
  "mlp_ratios": [
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3448663fb4c8597c2cb4b1a680bd82b46706e4657e59d7c26f6b34e41630d418
3
- size 14931044
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57223bc5c3c2e728b909cf86d98218b744db8a3cc34aac740231f6fab6ac8646
3
+ size 14933100
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:348a94c0bc6a5aa7d1e665ffb5e1eed4957cd69cf1218442e3d26e29962c302c
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6816dff685fdf020093a38412ab243a963c1c7abe818db1709603a79a3458a66
3
  size 5176