SodaXII commited on
Commit
b6d4d89
·
verified ·
1 Parent(s): e8fc060

Model save

Browse files
README.md ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: microsoft/swin-base-patch4-window7-224
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: swin-base-patch4-window7-224_rice-leaf-disease-augmented-v4_v5_pft
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # swin-base-patch4-window7-224_rice-leaf-disease-augmented-v4_v5_pft
18
+
19
+ This model is a fine-tuned version of [microsoft/swin-base-patch4-window7-224](https://huggingface.co/microsoft/swin-base-patch4-window7-224) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.4024
22
+ - Accuracy: 0.8490
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 0.0003
42
+ - train_batch_size: 64
43
+ - eval_batch_size: 64
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: cosine_with_restarts
47
+ - lr_scheduler_warmup_steps: 256
48
+ - num_epochs: 30
49
+ - mixed_precision_training: Native AMP
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 2.0846 | 0.5 | 64 | 1.9602 | 0.2483 |
56
+ | 1.7504 | 1.0 | 128 | 1.5308 | 0.5034 |
57
+ | 1.3704 | 1.5 | 192 | 1.1825 | 0.6107 |
58
+ | 1.113 | 2.0 | 256 | 0.9313 | 0.7148 |
59
+ | 0.9305 | 2.5 | 320 | 0.8132 | 0.7617 |
60
+ | 0.8171 | 3.0 | 384 | 0.7214 | 0.7651 |
61
+ | 0.7497 | 3.5 | 448 | 0.6650 | 0.7785 |
62
+ | 0.7039 | 4.0 | 512 | 0.6244 | 0.8188 |
63
+ | 0.6696 | 4.5 | 576 | 0.6003 | 0.8188 |
64
+ | 0.649 | 5.0 | 640 | 0.5976 | 0.8121 |
65
+ | 0.6334 | 5.5 | 704 | 0.6032 | 0.8020 |
66
+ | 0.6256 | 6.0 | 768 | 0.5859 | 0.8188 |
67
+ | 0.6417 | 6.5 | 832 | 0.5851 | 0.8188 |
68
+ | 0.5991 | 7.0 | 896 | 0.5835 | 0.8154 |
69
+ | 0.6014 | 7.5 | 960 | 0.5394 | 0.8322 |
70
+ | 0.5614 | 8.0 | 1024 | 0.5211 | 0.8356 |
71
+ | 0.536 | 8.5 | 1088 | 0.5184 | 0.8121 |
72
+ | 0.5443 | 9.0 | 1152 | 0.5256 | 0.8154 |
73
+ | 0.5129 | 9.5 | 1216 | 0.5026 | 0.8221 |
74
+ | 0.5084 | 10.0 | 1280 | 0.5028 | 0.8188 |
75
+ | 0.5081 | 10.5 | 1344 | 0.4996 | 0.8188 |
76
+ | 0.4936 | 11.0 | 1408 | 0.5004 | 0.8188 |
77
+ | 0.5 | 11.5 | 1472 | 0.5091 | 0.8121 |
78
+ | 0.4934 | 12.0 | 1536 | 0.4892 | 0.8356 |
79
+ | 0.4831 | 12.5 | 1600 | 0.4736 | 0.8322 |
80
+ | 0.4638 | 13.0 | 1664 | 0.4727 | 0.8255 |
81
+ | 0.4549 | 13.5 | 1728 | 0.4552 | 0.8456 |
82
+ | 0.4454 | 14.0 | 1792 | 0.4646 | 0.8322 |
83
+ | 0.44 | 14.5 | 1856 | 0.4610 | 0.8322 |
84
+ | 0.4304 | 15.0 | 1920 | 0.4574 | 0.8356 |
85
+ | 0.4255 | 15.5 | 1984 | 0.4550 | 0.8356 |
86
+ | 0.4353 | 16.0 | 2048 | 0.4548 | 0.8356 |
87
+ | 0.4456 | 16.5 | 2112 | 0.4465 | 0.8322 |
88
+ | 0.4047 | 17.0 | 2176 | 0.4619 | 0.8255 |
89
+ | 0.4119 | 17.5 | 2240 | 0.4497 | 0.8389 |
90
+ | 0.4009 | 18.0 | 2304 | 0.4329 | 0.8423 |
91
+ | 0.3901 | 18.5 | 2368 | 0.4286 | 0.8456 |
92
+ | 0.3936 | 19.0 | 2432 | 0.4318 | 0.8456 |
93
+ | 0.3761 | 19.5 | 2496 | 0.4297 | 0.8456 |
94
+ | 0.3885 | 20.0 | 2560 | 0.4279 | 0.8456 |
95
+ | 0.3806 | 20.5 | 2624 | 0.4271 | 0.8456 |
96
+ | 0.3779 | 21.0 | 2688 | 0.4352 | 0.8523 |
97
+ | 0.3746 | 21.5 | 2752 | 0.4256 | 0.8490 |
98
+ | 0.3708 | 22.0 | 2816 | 0.4253 | 0.8557 |
99
+ | 0.362 | 22.5 | 2880 | 0.4205 | 0.8490 |
100
+ | 0.3558 | 23.0 | 2944 | 0.4122 | 0.8490 |
101
+ | 0.3507 | 23.5 | 3008 | 0.4147 | 0.8423 |
102
+ | 0.3481 | 24.0 | 3072 | 0.4134 | 0.8456 |
103
+ | 0.3452 | 24.5 | 3136 | 0.4120 | 0.8456 |
104
+ | 0.3437 | 25.0 | 3200 | 0.4117 | 0.8490 |
105
+ | 0.3499 | 25.5 | 3264 | 0.4164 | 0.8523 |
106
+ | 0.3373 | 26.0 | 3328 | 0.4109 | 0.8490 |
107
+ | 0.3468 | 26.5 | 3392 | 0.3999 | 0.8523 |
108
+ | 0.3297 | 27.0 | 3456 | 0.4079 | 0.8523 |
109
+ | 0.329 | 27.5 | 3520 | 0.3997 | 0.8423 |
110
+ | 0.3293 | 28.0 | 3584 | 0.4051 | 0.8423 |
111
+ | 0.3147 | 28.5 | 3648 | 0.3987 | 0.8523 |
112
+ | 0.3239 | 29.0 | 3712 | 0.4013 | 0.8523 |
113
+ | 0.3147 | 29.5 | 3776 | 0.4031 | 0.8490 |
114
+ | 0.3167 | 30.0 | 3840 | 0.4024 | 0.8490 |
115
+
116
+
117
+ ### Framework versions
118
+
119
+ - Transformers 4.48.3
120
+ - Pytorch 2.5.1+cu124
121
+ - Datasets 3.3.2
122
+ - Tokenizers 0.21.1
config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/swin-base-patch4-window7-224",
3
+ "architectures": [
4
+ "SwinForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 18,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 128,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "Bacterial Leaf Blight",
21
+ "1": "Brown Spot",
22
+ "2": "Healthy Rice Leaf",
23
+ "3": "Leaf Blast",
24
+ "4": "Leaf scald",
25
+ "5": "Narrow Brown Leaf Spot",
26
+ "6": "Rice Hispa",
27
+ "7": "Sheath Blight"
28
+ },
29
+ "image_size": 224,
30
+ "initializer_range": 0.02,
31
+ "label2id": {
32
+ "Bacterial Leaf Blight": 0,
33
+ "Brown Spot": 1,
34
+ "Healthy Rice Leaf": 2,
35
+ "Leaf Blast": 3,
36
+ "Leaf scald": 4,
37
+ "Narrow Brown Leaf Spot": 5,
38
+ "Rice Hispa": 6,
39
+ "Sheath Blight": 7
40
+ },
41
+ "layer_norm_eps": 1e-05,
42
+ "mlp_ratio": 4.0,
43
+ "model_type": "swin",
44
+ "num_channels": 3,
45
+ "num_heads": [
46
+ 4,
47
+ 8,
48
+ 16,
49
+ 32
50
+ ],
51
+ "num_layers": 4,
52
+ "out_features": [
53
+ "stage4"
54
+ ],
55
+ "out_indices": [
56
+ 4
57
+ ],
58
+ "patch_size": 4,
59
+ "path_norm": true,
60
+ "problem_type": "single_label_classification",
61
+ "qkv_bias": true,
62
+ "stage_names": [
63
+ "stem",
64
+ "stage1",
65
+ "stage2",
66
+ "stage3",
67
+ "stage4"
68
+ ],
69
+ "torch_dtype": "float32",
70
+ "transformers_version": "4.48.3",
71
+ "use_absolute_embeddings": false,
72
+ "window_size": 7
73
+ }
logs/events.out.tfevents.1745554344.bb3cd456e661.517.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b10a19b71fc17223b1c8375af19c958aab6958bf50b57eafdca519088cfbbad6
3
+ size 38471
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44a9ee7f1602d2cc02d2d093b704f925eb75555c65e0aad04abbf7ba361aa4dc
3
+ size 347523416
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 3,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d96df27a803347936b253b60f385452e3f0af9e3d76a7c4ccbaef013ae743b24
3
+ size 5560
training_metrics.xlsx ADDED
Binary file (11.1 kB). View file