Hugging Face's logo

Augusto777
/
SWv2-DMAE-H-5-p-clean-fix-U-40-Cross-3

TensorBoard
Safetensors
swinv2
Generated from Trainer
Eval Results
Model card Files Metrics Community
Augusto777 commited on
Commit
fbb3ae5
·
verified ·
1 Parent(s): 752f321

End of training

Browse files
Files changed (7) hide show
  1. README.md +116 -0
  2. config.json +60 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +22 -0
  5. runs/Feb12_09-18-40_DESKTOP-SKBE9FB/events.out.tfevents.1739373520.DESKTOP-SKBE9FB.10128.4 +3 -0
  6. runs/Feb12_09-18-40_DESKTOP-SKBE9FB/events.out.tfevents.1739373970.DESKTOP-SKBE9FB.10128.5 +3 -0
  7. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: microsoft/swinv2-tiny-patch4-window8-256
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - imagefolder
8
+ metrics:
9
+ - accuracy
10
+ model-index:
11
+ - name: SWv2-DMAE-H-5-p-clean-fix-U-40-Cross-3
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.8390804597701149
26
+ ---
27
+
28
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
29
+ should probably proofread and complete it, then remove this comment. -->
30
+
31
+ # SWv2-DMAE-H-5-p-clean-fix-U-40-Cross-3
32
+
33
+ This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window8-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window8-256) on the imagefolder dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 0.4952
36
+ - Accuracy: 0.8391
37
+
38
+ ## Model description
39
+
40
+ More information needed
41
+
42
+ ## Intended uses & limitations
43
+
44
+ More information needed
45
+
46
+ ## Training and evaluation data
47
+
48
+ More information needed
49
+
50
+ ## Training procedure
51
+
52
+ ### Training hyperparameters
53
+
54
+ The following hyperparameters were used during training:
55
+ - learning_rate: 4e-05
56
+ - train_batch_size: 16
57
+ - eval_batch_size: 16
58
+ - seed: 42
59
+ - gradient_accumulation_steps: 4
60
+ - total_train_batch_size: 64
61
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
+ - lr_scheduler_type: linear
63
+ - lr_scheduler_warmup_ratio: 0.1
64
+ - num_epochs: 40
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
+ | 1.609 | 0.96 | 13 | 1.6081 | 0.1954 |
71
+ | 1.6048 | 2.0 | 27 | 1.5943 | 0.1954 |
72
+ | 1.5299 | 2.96 | 40 | 1.5695 | 0.1954 |
73
+ | 1.4527 | 4.0 | 54 | 1.4408 | 0.2184 |
74
+ | 1.3442 | 4.96 | 67 | 1.1869 | 0.5632 |
75
+ | 1.114 | 6.0 | 81 | 0.9027 | 0.6897 |
76
+ | 0.9651 | 6.96 | 94 | 0.7102 | 0.7586 |
77
+ | 0.8893 | 8.0 | 108 | 0.8029 | 0.6322 |
78
+ | 0.7704 | 8.96 | 121 | 0.5880 | 0.7816 |
79
+ | 0.6737 | 10.0 | 135 | 0.5514 | 0.8161 |
80
+ | 0.6713 | 10.96 | 148 | 0.4952 | 0.8391 |
81
+ | 0.6927 | 12.0 | 162 | 0.5375 | 0.8046 |
82
+ | 0.6031 | 12.96 | 175 | 0.5099 | 0.7931 |
83
+ | 0.542 | 14.0 | 189 | 0.5453 | 0.8046 |
84
+ | 0.511 | 14.96 | 202 | 0.5673 | 0.7701 |
85
+ | 0.4901 | 16.0 | 216 | 0.6610 | 0.7931 |
86
+ | 0.4824 | 16.96 | 229 | 0.5610 | 0.8046 |
87
+ | 0.4685 | 18.0 | 243 | 0.5992 | 0.7931 |
88
+ | 0.4442 | 18.96 | 256 | 0.5958 | 0.8161 |
89
+ | 0.4676 | 20.0 | 270 | 0.5621 | 0.8391 |
90
+ | 0.4231 | 20.96 | 283 | 0.5877 | 0.8161 |
91
+ | 0.3795 | 22.0 | 297 | 0.6075 | 0.8046 |
92
+ | 0.3645 | 22.96 | 310 | 0.6449 | 0.8046 |
93
+ | 0.366 | 24.0 | 324 | 0.6480 | 0.8161 |
94
+ | 0.344 | 24.96 | 337 | 0.6409 | 0.8276 |
95
+ | 0.2833 | 26.0 | 351 | 0.6246 | 0.8161 |
96
+ | 0.326 | 26.96 | 364 | 0.6491 | 0.8276 |
97
+ | 0.3416 | 28.0 | 378 | 0.6778 | 0.8161 |
98
+ | 0.2942 | 28.96 | 391 | 0.6374 | 0.8276 |
99
+ | 0.2767 | 30.0 | 405 | 0.6426 | 0.8276 |
100
+ | 0.2569 | 30.96 | 418 | 0.6631 | 0.8276 |
101
+ | 0.2889 | 32.0 | 432 | 0.6818 | 0.8161 |
102
+ | 0.2701 | 32.96 | 445 | 0.6889 | 0.8046 |
103
+ | 0.2553 | 34.0 | 459 | 0.6986 | 0.8161 |
104
+ | 0.2401 | 34.96 | 472 | 0.6985 | 0.8046 |
105
+ | 0.2536 | 36.0 | 486 | 0.7015 | 0.8046 |
106
+ | 0.2738 | 36.96 | 499 | 0.6940 | 0.8046 |
107
+ | 0.2304 | 38.0 | 513 | 0.6951 | 0.8046 |
108
+ | 0.3081 | 38.52 | 520 | 0.6951 | 0.8046 |
109
+
110
+
111
+ ### Framework versions
112
+
113
+ - Transformers 4.36.2
114
+ - Pytorch 2.1.2+cu118
115
+ - Datasets 2.16.1
116
+ - Tokenizers 0.15.0
config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/swinv2-tiny-patch4-window8-256",
3
+ "architectures": [
4
+ "Swinv2ForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 6,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 96,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 768,
19
+ "id2label": {
20
+ "0": "avanzada",
21
+ "1": "avanzada humeda",
22
+ "2": "leve",
23
+ "3": "moderada",
24
+ "4": "no dmae"
25
+ },
26
+ "image_size": 256,
27
+ "initializer_range": 0.02,
28
+ "label2id": {
29
+ "avanzada": 0,
30
+ "avanzada humeda": 1,
31
+ "leve": 2,
32
+ "moderada": 3,
33
+ "no dmae": 4
34
+ },
35
+ "layer_norm_eps": 1e-05,
36
+ "mlp_ratio": 4.0,
37
+ "model_type": "swinv2",
38
+ "num_channels": 3,
39
+ "num_heads": [
40
+ 3,
41
+ 6,
42
+ 12,
43
+ 24
44
+ ],
45
+ "num_layers": 4,
46
+ "patch_size": 4,
47
+ "path_norm": true,
48
+ "pretrained_window_sizes": [
49
+ 0,
50
+ 0,
51
+ 0,
52
+ 0
53
+ ],
54
+ "problem_type": "single_label_classification",
55
+ "qkv_bias": true,
56
+ "torch_dtype": "float32",
57
+ "transformers_version": "4.36.2",
58
+ "use_absolute_embeddings": false,
59
+ "window_size": 8
60
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a315bb8fc503de90d949deb2f5aa57fc8f8d7ad942c52defc81e9077d6d501d
3
+ size 110359372
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "resample": 3,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 256,
20
+ "width": 256
21
+ }
22
+ }
runs/Feb12_09-18-40_DESKTOP-SKBE9FB/events.out.tfevents.1739373520.DESKTOP-SKBE9FB.10128.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3127e9afd5a7dbe7eb06705a6c6dbc0f3e95875800ed7e54b335ecc9ff78d198
3
+ size 25907
runs/Feb12_09-18-40_DESKTOP-SKBE9FB/events.out.tfevents.1739373970.DESKTOP-SKBE9FB.10128.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f42de2ca97bad5b34ef555c9c46fd33b27e6eda2d50bcd6d2dfe87e4919b24b5
3
+ size 411
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01ea75ed273bf993fe63216530881fecdf858d8313e294a732ad6940662a0865
3
+ size 4792