Hugging Face's logo

AmirGenAI
/
segformer-b0-finetuned-wrinkle

Image Segmentation
Transformers
TensorBoard
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
AmirGenAI commited on
Commit
f0004a7
•
1 Parent(s): e801424

End of training

Browse files
Files changed (5) hide show
  1. README.md +67 -18
  2. config.json +2 -2
  3. model.safetensors +1 -1
  4. runs/Dec15_07-59-45_d3fe2635473c/events.out.tfevents.1734249596.d3fe2635473c.536.0 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -18,19 +18,14 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the face-wrinkles dataset.
20
  It achieves the following results on the evaluation set:
21
- - eval_loss: 0.0185
22
- - eval_mean_iou: 0.1947
23
- - eval_mean_accuracy: 0.3894
24
- - eval_overall_accuracy: 0.3894
25
- - eval_accuracy_unlabeled: nan
26
- - eval_accuracy_wrinkle: 0.3894
27
- - eval_iou_unlabeled: 0.0
28
- - eval_iou_wrinkle: 0.3894
29
- - eval_runtime: 13.3525
30
- - eval_samples_per_second: 9.811
31
- - eval_steps_per_second: 4.943
32
- - epoch: 8.9918
33
- - step: 3300
34
 
35
  ## Model description
36
 
@@ -50,16 +45,70 @@ More information needed
50
 
51
  The following hyperparameters were used during training:
52
  - learning_rate: 6e-05
53
- - train_batch_size: 2
54
- - eval_batch_size: 2
55
  - seed: 42
56
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
57
  - lr_scheduler_type: linear
58
- - num_epochs: 20
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
 
60
  ### Framework versions
61
 
62
- - Transformers 4.46.2
63
  - Pytorch 2.5.1+cu121
64
- - Datasets 3.1.0
65
  - Tokenizers 0.20.3
 
18
 
19
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the face-wrinkles dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0188
22
+ - Mean Iou: 0.2008
23
+ - Mean Accuracy: 0.4015
24
+ - Overall Accuracy: 0.4015
25
+ - Accuracy Unlabeled: nan
26
+ - Accuracy Wrinkle: 0.4015
27
+ - Iou Unlabeled: 0.0
28
+ - Iou Wrinkle: 0.4015
 
 
 
 
 
29
 
30
  ## Model description
31
 
 
45
 
46
  The following hyperparameters were used during training:
47
  - learning_rate: 6e-05
48
+ - train_batch_size: 8
49
+ - eval_batch_size: 8
50
  - seed: 42
51
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
  - lr_scheduler_type: linear
53
+ - lr_scheduler_warmup_steps: 100
54
+ - num_epochs: 10
55
+ - mixed_precision_training: Native AMP
56
+
57
+ ### Training results
58
+
59
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Unlabeled | Accuracy Wrinkle | Iou Unlabeled | Iou Wrinkle |
60
+ |:-------------:|:------:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:------------------:|:----------------:|:-------------:|:-----------:|
61
+ | 0.0127 | 0.2174 | 20 | 0.0185 | 0.1901 | 0.3802 | 0.3802 | nan | 0.3802 | 0.0 | 0.3802 |
62
+ | 0.011 | 0.4348 | 40 | 0.0185 | 0.1886 | 0.3772 | 0.3772 | nan | 0.3772 | 0.0 | 0.3772 |
63
+ | 0.0109 | 0.6522 | 60 | 0.0190 | 0.1380 | 0.2761 | 0.2761 | nan | 0.2761 | 0.0 | 0.2761 |
64
+ | 0.0157 | 0.8696 | 80 | 0.0190 | 0.1587 | 0.3174 | 0.3174 | nan | 0.3174 | 0.0 | 0.3174 |
65
+ | 0.0155 | 1.0870 | 100 | 0.0187 | 0.2034 | 0.4068 | 0.4068 | nan | 0.4068 | 0.0 | 0.4068 |
66
+ | 0.0185 | 1.3043 | 120 | 0.0184 | 0.1819 | 0.3639 | 0.3639 | nan | 0.3639 | 0.0 | 0.3639 |
67
+ | 0.0164 | 1.5217 | 140 | 0.0192 | 0.2445 | 0.4890 | 0.4890 | nan | 0.4890 | 0.0 | 0.4890 |
68
+ | 0.0202 | 1.7391 | 160 | 0.0187 | 0.1624 | 0.3249 | 0.3249 | nan | 0.3249 | 0.0 | 0.3249 |
69
+ | 0.008 | 1.9565 | 180 | 0.0185 | 0.1828 | 0.3656 | 0.3656 | nan | 0.3656 | 0.0 | 0.3656 |
70
+ | 0.015 | 2.1739 | 200 | 0.0190 | 0.2415 | 0.4831 | 0.4831 | nan | 0.4831 | 0.0 | 0.4831 |
71
+ | 0.0119 | 2.3913 | 220 | 0.0186 | 0.2115 | 0.4230 | 0.4230 | nan | 0.4230 | 0.0 | 0.4230 |
72
+ | 0.0113 | 2.6087 | 240 | 0.0185 | 0.1545 | 0.3090 | 0.3090 | nan | 0.3090 | 0.0 | 0.3090 |
73
+ | 0.0125 | 2.8261 | 260 | 0.0187 | 0.1798 | 0.3597 | 0.3597 | nan | 0.3597 | 0.0 | 0.3597 |
74
+ | 0.0197 | 3.0435 | 280 | 0.0195 | 0.1493 | 0.2987 | 0.2987 | nan | 0.2987 | 0.0 | 0.2987 |
75
+ | 0.0116 | 3.2609 | 300 | 0.0190 | 0.1612 | 0.3224 | 0.3224 | nan | 0.3224 | 0.0 | 0.3224 |
76
+ | 0.013 | 3.4783 | 320 | 0.0184 | 0.2097 | 0.4193 | 0.4193 | nan | 0.4193 | 0.0 | 0.4193 |
77
+ | 0.0176 | 3.6957 | 340 | 0.0185 | 0.2268 | 0.4537 | 0.4537 | nan | 0.4537 | 0.0 | 0.4537 |
78
+ | 0.0122 | 3.9130 | 360 | 0.0185 | 0.2039 | 0.4079 | 0.4079 | nan | 0.4079 | 0.0 | 0.4079 |
79
+ | 0.014 | 4.1304 | 380 | 0.0185 | 0.1942 | 0.3885 | 0.3885 | nan | 0.3885 | 0.0 | 0.3885 |
80
+ | 0.0105 | 4.3478 | 400 | 0.0185 | 0.2202 | 0.4403 | 0.4403 | nan | 0.4403 | 0.0 | 0.4403 |
81
+ | 0.0113 | 4.5652 | 420 | 0.0187 | 0.1648 | 0.3296 | 0.3296 | nan | 0.3296 | 0.0 | 0.3296 |
82
+ | 0.0092 | 4.7826 | 440 | 0.0185 | 0.2044 | 0.4087 | 0.4087 | nan | 0.4087 | 0.0 | 0.4087 |
83
+ | 0.0175 | 5.0 | 460 | 0.0185 | 0.2160 | 0.4320 | 0.4320 | nan | 0.4320 | 0.0 | 0.4320 |
84
+ | 0.0124 | 5.2174 | 480 | 0.0190 | 0.2009 | 0.4018 | 0.4018 | nan | 0.4018 | 0.0 | 0.4018 |
85
+ | 0.0162 | 5.4348 | 500 | 0.0186 | 0.2431 | 0.4863 | 0.4863 | nan | 0.4863 | 0.0 | 0.4863 |
86
+ | 0.0203 | 5.6522 | 520 | 0.0185 | 0.2091 | 0.4181 | 0.4181 | nan | 0.4181 | 0.0 | 0.4181 |
87
+ | 0.0172 | 5.8696 | 540 | 0.0190 | 0.1700 | 0.3401 | 0.3401 | nan | 0.3401 | 0.0 | 0.3401 |
88
+ | 0.014 | 6.0870 | 560 | 0.0189 | 0.1771 | 0.3541 | 0.3541 | nan | 0.3541 | 0.0 | 0.3541 |
89
+ | 0.0191 | 6.3043 | 580 | 0.0189 | 0.1788 | 0.3575 | 0.3575 | nan | 0.3575 | 0.0 | 0.3575 |
90
+ | 0.0157 | 6.5217 | 600 | 0.0188 | 0.1986 | 0.3971 | 0.3971 | nan | 0.3971 | 0.0 | 0.3971 |
91
+ | 0.0128 | 6.7391 | 620 | 0.0187 | 0.2218 | 0.4436 | 0.4436 | nan | 0.4436 | 0.0 | 0.4436 |
92
+ | 0.0155 | 6.9565 | 640 | 0.0185 | 0.2099 | 0.4198 | 0.4198 | nan | 0.4198 | 0.0 | 0.4198 |
93
+ | 0.0132 | 7.1739 | 660 | 0.0189 | 0.1909 | 0.3819 | 0.3819 | nan | 0.3819 | 0.0 | 0.3819 |
94
+ | 0.0142 | 7.3913 | 680 | 0.0185 | 0.1892 | 0.3784 | 0.3784 | nan | 0.3784 | 0.0 | 0.3784 |
95
+ | 0.0113 | 7.6087 | 700 | 0.0187 | 0.1914 | 0.3828 | 0.3828 | nan | 0.3828 | 0.0 | 0.3828 |
96
+ | 0.0111 | 7.8261 | 720 | 0.0187 | 0.2136 | 0.4271 | 0.4271 | nan | 0.4271 | 0.0 | 0.4271 |
97
+ | 0.0094 | 8.0435 | 740 | 0.0188 | 0.1922 | 0.3843 | 0.3843 | nan | 0.3843 | 0.0 | 0.3843 |
98
+ | 0.0198 | 8.2609 | 760 | 0.0188 | 0.1911 | 0.3822 | 0.3822 | nan | 0.3822 | 0.0 | 0.3822 |
99
+ | 0.0164 | 8.4783 | 780 | 0.0189 | 0.1896 | 0.3792 | 0.3792 | nan | 0.3792 | 0.0 | 0.3792 |
100
+ | 0.0222 | 8.6957 | 800 | 0.0186 | 0.2178 | 0.4355 | 0.4355 | nan | 0.4355 | 0.0 | 0.4355 |
101
+ | 0.0108 | 8.9130 | 820 | 0.0190 | 0.1855 | 0.3710 | 0.3710 | nan | 0.3710 | 0.0 | 0.3710 |
102
+ | 0.0128 | 9.1304 | 840 | 0.0187 | 0.2006 | 0.4011 | 0.4011 | nan | 0.4011 | 0.0 | 0.4011 |
103
+ | 0.0143 | 9.3478 | 860 | 0.0187 | 0.2013 | 0.4026 | 0.4026 | nan | 0.4026 | 0.0 | 0.4026 |
104
+ | 0.0088 | 9.5652 | 880 | 0.0187 | 0.2020 | 0.4040 | 0.4040 | nan | 0.4040 | 0.0 | 0.4040 |
105
+ | 0.0127 | 9.7826 | 900 | 0.0188 | 0.2023 | 0.4046 | 0.4046 | nan | 0.4046 | 0.0 | 0.4046 |
106
+ | 0.0109 | 10.0 | 920 | 0.0188 | 0.2008 | 0.4015 | 0.4015 | nan | 0.4015 | 0.0 | 0.4015 |
107
+
108
 
109
  ### Framework versions
110
 
111
+ - Transformers 4.46.3
112
  - Pytorch 2.5.1+cu121
113
+ - Datasets 3.2.0
114
  - Tokenizers 0.20.3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "nvidia/mit-b0",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
@@ -74,5 +74,5 @@
74
  2
75
  ],
76
  "torch_dtype": "float32",
77
- "transformers_version": "4.46.2"
78
  }
 
1
  {
2
+ "_name_or_path": "AmirGenAI/segformer-b0-finetuned-wrinkle",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
 
74
  2
75
  ],
76
  "torch_dtype": "float32",
77
+ "transformers_version": "4.46.3"
78
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f62fd4948e62e25b63f85f4f723ae07eedbfe5669abd4375eac9d44040c74f53
3
  size 14884776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ab732644ac79311ddb909d01fd469bbfd75d9c74ef0500b6d20a62f1b0295d1
3
  size 14884776
runs/Dec15_07-59-45_d3fe2635473c/events.out.tfevents.1734249596.d3fe2635473c.536.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66dacb636dadeb41d062a684ca46f1972f7d3b48eabf52306ee219333bb4f9e1
3
+ size 230527
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b0b4e6f49088e2730af09d71a3febd260c16b85d05603f855aca67c245afa7ec
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7352ac4aa3d701fe1abc2f94ef6fc859d5279d0ed1a50160fa3f9ae3f3eb90b4
3
  size 5368