SodaXII commited on
Commit
c343474
·
verified ·
1 Parent(s): e0714b6

Model save

Browse files
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: WinKawaks/vit-tiny-patch16-224
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: vit-tiny-patch16-224_rice-leaf-disease-augmented-v4_fft
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
+ # vit-tiny-patch16-224_rice-leaf-disease-augmented-v4_fft
18
+
19
+ This model is a fine-tuned version of [WinKawaks/vit-tiny-patch16-224](https://huggingface.co/WinKawaks/vit-tiny-patch16-224) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3269
22
+ - Accuracy: 0.9228
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: 5e-05
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: 15
49
+ - mixed_precision_training: Native AMP
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 2.0564 | 0.5 | 64 | 1.4541 | 0.4899 |
56
+ | 1.0767 | 1.0 | 128 | 0.6909 | 0.7651 |
57
+ | 0.4917 | 1.5 | 192 | 0.4307 | 0.8322 |
58
+ | 0.285 | 2.0 | 256 | 0.2932 | 0.9027 |
59
+ | 0.0902 | 2.5 | 320 | 0.3134 | 0.8993 |
60
+ | 0.0588 | 3.0 | 384 | 0.3076 | 0.9161 |
61
+ | 0.0155 | 3.5 | 448 | 0.2627 | 0.9396 |
62
+ | 0.0066 | 4.0 | 512 | 0.2992 | 0.9295 |
63
+ | 0.0017 | 4.5 | 576 | 0.2936 | 0.9228 |
64
+ | 0.0009 | 5.0 | 640 | 0.2961 | 0.9228 |
65
+ | 0.0006 | 5.5 | 704 | 0.3005 | 0.9228 |
66
+ | 0.0005 | 6.0 | 768 | 0.3004 | 0.9228 |
67
+ | 0.0005 | 6.5 | 832 | 0.2867 | 0.9262 |
68
+ | 0.0004 | 7.0 | 896 | 0.2977 | 0.9295 |
69
+ | 0.0003 | 7.5 | 960 | 0.2944 | 0.9295 |
70
+ | 0.0002 | 8.0 | 1024 | 0.3074 | 0.9295 |
71
+ | 0.0002 | 8.5 | 1088 | 0.3053 | 0.9329 |
72
+ | 0.0002 | 9.0 | 1152 | 0.3098 | 0.9295 |
73
+ | 0.0001 | 9.5 | 1216 | 0.3102 | 0.9295 |
74
+ | 0.0001 | 10.0 | 1280 | 0.3105 | 0.9262 |
75
+ | 0.0001 | 10.5 | 1344 | 0.3105 | 0.9262 |
76
+ | 0.0001 | 11.0 | 1408 | 0.3202 | 0.9262 |
77
+ | 0.0001 | 11.5 | 1472 | 0.3183 | 0.9295 |
78
+ | 0.0001 | 12.0 | 1536 | 0.3131 | 0.9329 |
79
+ | 0.0001 | 12.5 | 1600 | 0.3157 | 0.9295 |
80
+ | 0.0001 | 13.0 | 1664 | 0.3238 | 0.9228 |
81
+ | 0.0001 | 13.5 | 1728 | 0.3220 | 0.9228 |
82
+ | 0.0001 | 14.0 | 1792 | 0.3266 | 0.9228 |
83
+ | 0.0001 | 14.5 | 1856 | 0.3274 | 0.9228 |
84
+ | 0.0001 | 15.0 | 1920 | 0.3269 | 0.9228 |
85
+
86
+
87
+ ### Framework versions
88
+
89
+ - Transformers 4.48.3
90
+ - Pytorch 2.5.1+cu124
91
+ - Datasets 3.3.2
92
+ - Tokenizers 0.21.0
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "WinKawaks/vit-tiny-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 192,
11
+ "id2label": {
12
+ "0": "Bacterial Leaf Blight",
13
+ "1": "Brown Spot",
14
+ "2": "Healthy Rice Leaf",
15
+ "3": "Leaf Blast",
16
+ "4": "Leaf scald",
17
+ "5": "Narrow Brown Leaf Spot",
18
+ "6": "Rice Hispa",
19
+ "7": "Sheath Blight"
20
+ },
21
+ "image_size": 224,
22
+ "initializer_range": 0.02,
23
+ "intermediate_size": 768,
24
+ "label2id": {
25
+ "Bacterial Leaf Blight": 0,
26
+ "Brown Spot": 1,
27
+ "Healthy Rice Leaf": 2,
28
+ "Leaf Blast": 3,
29
+ "Leaf scald": 4,
30
+ "Narrow Brown Leaf Spot": 5,
31
+ "Rice Hispa": 6,
32
+ "Sheath Blight": 7
33
+ },
34
+ "layer_norm_eps": 1e-12,
35
+ "model_type": "vit",
36
+ "num_attention_heads": 3,
37
+ "num_channels": 3,
38
+ "num_hidden_layers": 12,
39
+ "patch_size": 16,
40
+ "problem_type": "single_label_classification",
41
+ "qkv_bias": true,
42
+ "torch_dtype": "float32",
43
+ "transformers_version": "4.48.3"
44
+ }
logs/events.out.tfevents.1741847730.b21d00ddba50.811.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b2e6c888877b8564ca4299300af8382794723d4e3bfa6d7fdfe31f04102c385
3
+ size 22073
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0d6bf661054beeeb2301f2c210e5f156652e24c8f8840984b5a302b8459286a
3
+ size 22126552
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.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
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:4c48cc43f351ec35233c274136f90dffa9129d37aa5ea7648a6e73cce86a2af3
3
+ size 5496
training_metrics.xlsx ADDED
Binary file (8.28 kB). View file