SodaXII commited on
Commit
f2edbd0
·
verified ·
1 Parent(s): b618b7a

Model save

Browse files
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: facebook/deit-tiny-patch16-224
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: deit-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
+ # deit-tiny-patch16-224_rice-leaf-disease-augmented-v4_fft
18
+
19
+ This model is a fine-tuned version of [facebook/deit-tiny-patch16-224](https://huggingface.co/facebook/deit-tiny-patch16-224) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3997
22
+ - Accuracy: 0.9195
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
+ | 1.9687 | 0.5 | 64 | 1.7115 | 0.4430 |
56
+ | 1.2893 | 1.0 | 128 | 0.8590 | 0.7215 |
57
+ | 0.6713 | 1.5 | 192 | 0.5154 | 0.8255 |
58
+ | 0.4203 | 2.0 | 256 | 0.3670 | 0.8725 |
59
+ | 0.195 | 2.5 | 320 | 0.3335 | 0.8826 |
60
+ | 0.138 | 3.0 | 384 | 0.3436 | 0.8926 |
61
+ | 0.0499 | 3.5 | 448 | 0.2509 | 0.9295 |
62
+ | 0.03 | 4.0 | 512 | 0.3101 | 0.9060 |
63
+ | 0.0095 | 4.5 | 576 | 0.3378 | 0.9128 |
64
+ | 0.0044 | 5.0 | 640 | 0.2718 | 0.9262 |
65
+ | 0.0021 | 5.5 | 704 | 0.3010 | 0.9094 |
66
+ | 0.0018 | 6.0 | 768 | 0.2916 | 0.9195 |
67
+ | 0.0017 | 6.5 | 832 | 0.3134 | 0.9295 |
68
+ | 0.0043 | 7.0 | 896 | 0.4023 | 0.9161 |
69
+ | 0.0057 | 7.5 | 960 | 0.4991 | 0.8826 |
70
+ | 0.0072 | 8.0 | 1024 | 0.3925 | 0.9060 |
71
+ | 0.0021 | 8.5 | 1088 | 0.4229 | 0.9128 |
72
+ | 0.0012 | 9.0 | 1152 | 0.4050 | 0.9195 |
73
+ | 0.0003 | 9.5 | 1216 | 0.3807 | 0.9161 |
74
+ | 0.0003 | 10.0 | 1280 | 0.3871 | 0.9195 |
75
+ | 0.0003 | 10.5 | 1344 | 0.3868 | 0.9195 |
76
+ | 0.0003 | 11.0 | 1408 | 0.4019 | 0.9195 |
77
+ | 0.0002 | 11.5 | 1472 | 0.3947 | 0.9161 |
78
+ | 0.0002 | 12.0 | 1536 | 0.3737 | 0.9195 |
79
+ | 0.0001 | 12.5 | 1600 | 0.3942 | 0.9195 |
80
+ | 0.0001 | 13.0 | 1664 | 0.3979 | 0.9195 |
81
+ | 0.0001 | 13.5 | 1728 | 0.3984 | 0.9195 |
82
+ | 0.0001 | 14.0 | 1792 | 0.4001 | 0.9195 |
83
+ | 0.0001 | 14.5 | 1856 | 0.4000 | 0.9195 |
84
+ | 0.0001 | 15.0 | 1920 | 0.3997 | 0.9195 |
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": "facebook/deit-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.1741847614.caa99427825e.5104.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b970aa648fb5aed07e7e37dcc73ee53617e0ca02616647882b834464bfd9c13d
3
+ size 22048
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0eb1ac5be4cf2fe51790758679f6ac2d808c2c9343ea07fcba1394e4c677cfeb
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:b4048f369f815ce45ff84723c183c8b24b0070bcf3dc45a29e131e29ee8d55be
3
+ size 5496
training_metrics.xlsx ADDED
Binary file (8.27 kB). View file