nqvii commited on
Commit
9d07c4b
·
verified ·
1 Parent(s): b97ffd4

End of training Fold 4

Browse files
Files changed (5) hide show
  1. README.md +117 -0
  2. config.json +37 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +23 -0
  5. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/vit-base-patch16-224
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - imagefolder
9
+ metrics:
10
+ - accuracy
11
+ - recall
12
+ model-index:
13
+ - name: fold_4
14
+ results:
15
+ - task:
16
+ name: Image Classification
17
+ type: image-classification
18
+ dataset:
19
+ name: imagefolder
20
+ type: imagefolder
21
+ config: default
22
+ split: None
23
+ args: default
24
+ metrics:
25
+ - name: Accuracy
26
+ type: accuracy
27
+ value: 0.9619377162629758
28
+ - name: Recall
29
+ type: recall
30
+ value: 0.9658753615233592
31
+ ---
32
+
33
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
34
+ should probably proofread and complete it, then remove this comment. -->
35
+
36
+ # fold_4
37
+
38
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
39
+ It achieves the following results on the evaluation set:
40
+ - Loss: 1.1305
41
+ - Accuracy: 0.9619
42
+ - F1 Score: 0.9652
43
+ - Recall: 0.9659
44
+
45
+ ## Model description
46
+
47
+ More information needed
48
+
49
+ ## Intended uses & limitations
50
+
51
+ More information needed
52
+
53
+ ## Training and evaluation data
54
+
55
+ More information needed
56
+
57
+ ## Training procedure
58
+
59
+ ### Training hyperparameters
60
+
61
+ The following hyperparameters were used during training:
62
+ - learning_rate: 1e-05
63
+ - train_batch_size: 64
64
+ - eval_batch_size: 64
65
+ - seed: 42
66
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
67
+ - lr_scheduler_type: cosine
68
+ - lr_scheduler_warmup_steps: 150
69
+ - num_epochs: 100
70
+ - label_smoothing_factor: 0.15
71
+
72
+ ### Training results
73
+
74
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Score | Recall |
75
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:------:|
76
+ | 3.1319 | 1.0 | 19 | 3.1085 | 0.1453 | 0.1309 | 0.1601 |
77
+ | 2.9606 | 2.0 | 38 | 2.8599 | 0.2526 | 0.2480 | 0.2557 |
78
+ | 2.6010 | 3.0 | 57 | 2.5562 | 0.4775 | 0.4447 | 0.4458 |
79
+ | 2.2535 | 4.0 | 76 | 2.2286 | 0.6125 | 0.6151 | 0.6017 |
80
+ | 1.8319 | 5.0 | 95 | 1.8960 | 0.7405 | 0.7493 | 0.7328 |
81
+ | 1.5134 | 6.0 | 114 | 1.5738 | 0.8547 | 0.8679 | 0.8626 |
82
+ | 1.3165 | 7.0 | 133 | 1.3901 | 0.8962 | 0.9047 | 0.8956 |
83
+ | 1.1793 | 8.0 | 152 | 1.2636 | 0.9446 | 0.9499 | 0.9499 |
84
+ | 1.1215 | 9.0 | 171 | 1.2150 | 0.9377 | 0.9435 | 0.9413 |
85
+ | 1.0953 | 10.0 | 190 | 1.1884 | 0.9481 | 0.9530 | 0.9538 |
86
+ | 1.0713 | 11.0 | 209 | 1.1682 | 0.9516 | 0.9555 | 0.9563 |
87
+ | 1.0654 | 12.0 | 228 | 1.1829 | 0.9446 | 0.9497 | 0.9524 |
88
+ | 1.0202 | 13.0 | 247 | 1.1502 | 0.9585 | 0.9615 | 0.9610 |
89
+ | 1.0255 | 14.0 | 266 | 1.1526 | 0.9550 | 0.9589 | 0.9573 |
90
+ | 1.0126 | 15.0 | 285 | 1.1521 | 0.9619 | 0.9651 | 0.9646 |
91
+ | 1.0030 | 16.0 | 304 | 1.1346 | 0.9585 | 0.9621 | 0.9621 |
92
+ | 0.9920 | 17.0 | 323 | 1.1365 | 0.9550 | 0.9589 | 0.9573 |
93
+ | 0.9882 | 18.0 | 342 | 1.1356 | 0.9585 | 0.9621 | 0.9621 |
94
+ | 0.9950 | 19.0 | 361 | 1.1334 | 0.9585 | 0.9621 | 0.9621 |
95
+ | 0.9819 | 20.0 | 380 | 1.1356 | 0.9585 | 0.9621 | 0.9621 |
96
+ | 0.9801 | 21.0 | 399 | 1.1312 | 0.9516 | 0.9545 | 0.9547 |
97
+ | 0.9718 | 22.0 | 418 | 1.1346 | 0.9516 | 0.9545 | 0.9547 |
98
+ | 0.9679 | 23.0 | 437 | 1.1564 | 0.9481 | 0.9534 | 0.9572 |
99
+ | 0.9785 | 24.0 | 456 | 1.1227 | 0.9516 | 0.9545 | 0.9547 |
100
+ | 0.9702 | 25.0 | 475 | 1.1275 | 0.9550 | 0.9589 | 0.9584 |
101
+ | 0.9742 | 26.0 | 494 | 1.1230 | 0.9585 | 0.9620 | 0.9609 |
102
+ | 0.9732 | 27.0 | 513 | 1.1303 | 0.9516 | 0.9554 | 0.9548 |
103
+ | 0.9651 | 28.0 | 532 | 1.1305 | 0.9619 | 0.9652 | 0.9659 |
104
+ | 0.9683 | 29.0 | 551 | 1.1283 | 0.9585 | 0.9620 | 0.9609 |
105
+ | 0.9692 | 30.0 | 570 | 1.1373 | 0.9516 | 0.9554 | 0.9548 |
106
+ | 0.9642 | 31.0 | 589 | 1.1312 | 0.9550 | 0.9589 | 0.9584 |
107
+ | 0.9642 | 32.0 | 608 | 1.1352 | 0.9550 | 0.9589 | 0.9584 |
108
+ | 0.9649 | 33.0 | 627 | 1.1363 | 0.9516 | 0.9554 | 0.9548 |
109
+ | 0.9621 | 34.0 | 646 | 1.1327 | 0.9550 | 0.9589 | 0.9584 |
110
+
111
+
112
+ ### Framework versions
113
+
114
+ - Transformers 5.0.0
115
+ - Pytorch 2.10.0+cu128
116
+ - Datasets 5.0.0
117
+ - Tokenizers 0.22.2
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "dtype": "float32",
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Bacterial Leaf Blight",
13
+ "1": "Brown Spot",
14
+ "2": "Healthy Leaf",
15
+ "3": "Tungro Virus"
16
+ },
17
+ "image_size": 224,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "Bacterial Leaf Blight": "0",
22
+ "Brown Spot": "1",
23
+ "Healthy Leaf": "2",
24
+ "Tungro Virus": "3"
25
+ },
26
+ "layer_norm_eps": 1e-12,
27
+ "model_type": "vit",
28
+ "num_attention_heads": 12,
29
+ "num_channels": 3,
30
+ "num_hidden_layers": 12,
31
+ "patch_size": 16,
32
+ "pooler_act": "tanh",
33
+ "pooler_output_size": 768,
34
+ "qkv_bias": true,
35
+ "transformers_version": "5.0.0",
36
+ "use_cache": false
37
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a879c46546f450b5f94b05400136909a0b6daadab7a9aeb393f040f45b0e28f
3
+ size 343230128
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:bb637894605e8d2f1a8416043f408e43d779452dda9650ffc7dc036e08e049b6
3
+ size 5201