Model save
Browse files
README.md
CHANGED
@@ -2,7 +2,6 @@
|
|
2 |
license: apache-2.0
|
3 |
base_model: google/vit-base-patch16-224-in21k
|
4 |
tags:
|
5 |
-
- image-classification
|
6 |
- generated_from_trainer
|
7 |
datasets:
|
8 |
- imagefolder
|
@@ -15,7 +14,7 @@ model-index:
|
|
15 |
name: Image Classification
|
16 |
type: image-classification
|
17 |
dataset:
|
18 |
-
name:
|
19 |
type: imagefolder
|
20 |
config: default
|
21 |
split: train
|
@@ -23,7 +22,7 @@ model-index:
|
|
23 |
metrics:
|
24 |
- name: Accuracy
|
25 |
type: accuracy
|
26 |
-
value: 0.
|
27 |
---
|
28 |
|
29 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -31,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
31 |
|
32 |
# Action_model
|
33 |
|
34 |
-
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the
|
35 |
It achieves the following results on the evaluation set:
|
36 |
-
- Loss: 0.
|
37 |
-
- Accuracy: 0.
|
38 |
|
39 |
## Model description
|
40 |
|
@@ -54,80 +53,24 @@ More information needed
|
|
54 |
|
55 |
The following hyperparameters were used during training:
|
56 |
- learning_rate: 0.0001
|
57 |
-
- train_batch_size:
|
58 |
- eval_batch_size: 8
|
59 |
- seed: 42
|
60 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
61 |
- lr_scheduler_type: linear
|
62 |
-
- num_epochs:
|
63 |
- mixed_precision_training: Native AMP
|
64 |
|
65 |
### Training results
|
66 |
|
67 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
68 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
69 |
-
| 1.
|
70 |
-
| 0.
|
71 |
-
| 0.
|
72 |
-
| 0.
|
73 |
-
| 0.
|
74 |
-
| 0.
|
75 |
-
| 0.5381 | 1.11 | 700 | 0.7965 | 0.7524 |
|
76 |
-
| 0.4066 | 1.27 | 800 | 0.8102 | 0.7486 |
|
77 |
-
| 0.4825 | 1.43 | 900 | 0.7498 | 0.7743 |
|
78 |
-
| 0.4955 | 1.59 | 1000 | 0.9752 | 0.7019 |
|
79 |
-
| 0.3945 | 1.75 | 1100 | 0.8150 | 0.7381 |
|
80 |
-
| 0.4142 | 1.91 | 1200 | 0.7953 | 0.7610 |
|
81 |
-
| 0.3915 | 2.07 | 1300 | 0.8141 | 0.7638 |
|
82 |
-
| 0.3937 | 2.23 | 1400 | 0.7882 | 0.7705 |
|
83 |
-
| 0.3144 | 2.39 | 1500 | 0.8657 | 0.7514 |
|
84 |
-
| 0.3143 | 2.55 | 1600 | 1.0562 | 0.7086 |
|
85 |
-
| 0.3884 | 2.71 | 1700 | 1.0502 | 0.7162 |
|
86 |
-
| 0.3472 | 2.87 | 1800 | 0.8506 | 0.7571 |
|
87 |
-
| 0.2545 | 3.03 | 1900 | 1.0029 | 0.7210 |
|
88 |
-
| 0.2213 | 3.18 | 2000 | 0.8099 | 0.7933 |
|
89 |
-
| 0.3429 | 3.34 | 2100 | 0.9166 | 0.7467 |
|
90 |
-
| 0.3478 | 3.5 | 2200 | 0.9202 | 0.7562 |
|
91 |
-
| 0.2247 | 3.66 | 2300 | 0.9859 | 0.7638 |
|
92 |
-
| 0.2873 | 3.82 | 2400 | 1.0161 | 0.7390 |
|
93 |
-
| 0.2815 | 3.98 | 2500 | 0.9631 | 0.7590 |
|
94 |
-
| 0.1706 | 4.14 | 2600 | 0.9996 | 0.7419 |
|
95 |
-
| 0.1709 | 4.3 | 2700 | 1.1969 | 0.7029 |
|
96 |
-
| 0.2847 | 4.46 | 2800 | 1.0896 | 0.7276 |
|
97 |
-
| 0.286 | 4.62 | 2900 | 0.9894 | 0.7629 |
|
98 |
-
| 0.2066 | 4.78 | 3000 | 1.0704 | 0.7486 |
|
99 |
-
| 0.1579 | 4.94 | 3100 | 0.9728 | 0.7810 |
|
100 |
-
| 0.1716 | 5.1 | 3200 | 1.1834 | 0.7124 |
|
101 |
-
| 0.1584 | 5.25 | 3300 | 1.0278 | 0.7524 |
|
102 |
-
| 0.1419 | 5.41 | 3400 | 1.1342 | 0.7371 |
|
103 |
-
| 0.2002 | 5.57 | 3500 | 1.1519 | 0.7324 |
|
104 |
-
| 0.1987 | 5.73 | 3600 | 1.0742 | 0.7562 |
|
105 |
-
| 0.1207 | 5.89 | 3700 | 1.1579 | 0.7381 |
|
106 |
-
| 0.1403 | 6.05 | 3800 | 1.1007 | 0.7305 |
|
107 |
-
| 0.1569 | 6.21 | 3900 | 1.1128 | 0.7467 |
|
108 |
-
| 0.1763 | 6.37 | 4000 | 1.0720 | 0.7524 |
|
109 |
-
| 0.2426 | 6.53 | 4100 | 1.1484 | 0.7248 |
|
110 |
-
| 0.1434 | 6.69 | 4200 | 1.1790 | 0.7343 |
|
111 |
-
| 0.2191 | 6.85 | 4300 | 1.1169 | 0.7486 |
|
112 |
-
| 0.2062 | 7.01 | 4400 | 1.1300 | 0.7610 |
|
113 |
-
| 0.1495 | 7.17 | 4500 | 1.1477 | 0.7495 |
|
114 |
-
| 0.1261 | 7.32 | 4600 | 1.0891 | 0.7657 |
|
115 |
-
| 0.12 | 7.48 | 4700 | 1.1359 | 0.76 |
|
116 |
-
| 0.1396 | 7.64 | 4800 | 1.1230 | 0.7410 |
|
117 |
-
| 0.0728 | 7.8 | 4900 | 1.1210 | 0.7552 |
|
118 |
-
| 0.175 | 7.96 | 5000 | 1.1204 | 0.7505 |
|
119 |
-
| 0.1214 | 8.12 | 5100 | 1.1064 | 0.7543 |
|
120 |
-
| 0.1218 | 8.28 | 5200 | 1.0232 | 0.7771 |
|
121 |
-
| 0.1556 | 8.44 | 5300 | 1.0489 | 0.7771 |
|
122 |
-
| 0.1019 | 8.6 | 5400 | 1.0916 | 0.7752 |
|
123 |
-
| 0.1446 | 8.76 | 5500 | 1.1856 | 0.7505 |
|
124 |
-
| 0.1348 | 8.92 | 5600 | 1.1380 | 0.7638 |
|
125 |
-
| 0.1402 | 9.08 | 5700 | 1.1233 | 0.7695 |
|
126 |
-
| 0.1075 | 9.24 | 5800 | 1.1472 | 0.7629 |
|
127 |
-
| 0.0991 | 9.39 | 5900 | 1.1530 | 0.7648 |
|
128 |
-
| 0.081 | 9.55 | 6000 | 1.1586 | 0.7629 |
|
129 |
-
| 0.0724 | 9.71 | 6100 | 1.1591 | 0.7676 |
|
130 |
-
| 0.0399 | 9.87 | 6200 | 1.1418 | 0.7695 |
|
131 |
|
132 |
|
133 |
### Framework versions
|
|
|
2 |
license: apache-2.0
|
3 |
base_model: google/vit-base-patch16-224-in21k
|
4 |
tags:
|
|
|
5 |
- generated_from_trainer
|
6 |
datasets:
|
7 |
- imagefolder
|
|
|
14 |
name: Image Classification
|
15 |
type: image-classification
|
16 |
dataset:
|
17 |
+
name: imagefolder
|
18 |
type: imagefolder
|
19 |
config: default
|
20 |
split: train
|
|
|
22 |
metrics:
|
23 |
- name: Accuracy
|
24 |
type: accuracy
|
25 |
+
value: 0.8085714285714286
|
26 |
---
|
27 |
|
28 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
30 |
|
31 |
# Action_model
|
32 |
|
33 |
+
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
|
34 |
It achieves the following results on the evaluation set:
|
35 |
+
- Loss: 0.6364
|
36 |
+
- Accuracy: 0.8086
|
37 |
|
38 |
## Model description
|
39 |
|
|
|
53 |
|
54 |
The following hyperparameters were used during training:
|
55 |
- learning_rate: 0.0001
|
56 |
+
- train_batch_size: 32
|
57 |
- eval_batch_size: 8
|
58 |
- seed: 42
|
59 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
60 |
- lr_scheduler_type: linear
|
61 |
+
- num_epochs: 2
|
62 |
- mixed_precision_training: Native AMP
|
63 |
|
64 |
### Training results
|
65 |
|
66 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
67 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
68 |
+
| 1.1604 | 0.32 | 100 | 1.0592 | 0.7571 |
|
69 |
+
| 0.8165 | 0.64 | 200 | 0.7481 | 0.8029 |
|
70 |
+
| 0.6333 | 0.96 | 300 | 0.7136 | 0.7886 |
|
71 |
+
| 0.5122 | 1.27 | 400 | 0.6776 | 0.8010 |
|
72 |
+
| 0.4711 | 1.59 | 500 | 0.6334 | 0.8143 |
|
73 |
+
| 0.4339 | 1.91 | 600 | 0.6364 | 0.8086 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
74 |
|
75 |
|
76 |
### Framework versions
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 343248584
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d1f30d73d44700affdb2cca2a0a9a40deacab5f599ff8c778f9d002a659cee2d
|
3 |
size 343248584
|
runs/Apr17_11-33-15_a10a130002dd/events.out.tfevents.1713353798.a10a130002dd.35.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:db2e44a47b17041171993521d150a09e2bdd2aac754263e868acb2d6eab65246
|
3 |
+
size 14404
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4920
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0bbf47446cd1683709acbfaad1d518d37fb0f0d6f1a3e6386c1d4653b6f04301
|
3 |
size 4920
|