Model save
Browse files- README.md +37 -21
- config.json +9 -9
- model.safetensors +1 -1
- preprocessor_config.json +14 -0
- runs/Apr30_14-52-00_e47bc2c7ac19/events.out.tfevents.1714488721.e47bc2c7ac19.34.0 +3 -0
- training_args.bin +2 -2
README.md
CHANGED
|
@@ -2,13 +2,27 @@
|
|
| 2 |
license: apache-2.0
|
| 3 |
base_model: google/vit-base-patch16-224-in21k
|
| 4 |
tags:
|
| 5 |
-
- image-classification
|
| 6 |
- generated_from_trainer
|
|
|
|
|
|
|
| 7 |
metrics:
|
| 8 |
- accuracy
|
| 9 |
model-index:
|
| 10 |
- name: Action_all_10_class
|
| 11 |
-
results:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
@@ -16,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 16 |
|
| 17 |
# Action_all_10_class
|
| 18 |
|
| 19 |
-
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
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
-
- Loss: 0.
|
| 22 |
-
- Accuracy: 0.
|
| 23 |
|
| 24 |
## Model description
|
| 25 |
|
|
@@ -51,23 +65,25 @@ The following hyperparameters were used during training:
|
|
| 51 |
|
| 52 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 53 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 54 |
-
| 1.
|
| 55 |
-
| 0.
|
| 56 |
-
| 0.
|
| 57 |
-
| 0.
|
| 58 |
-
| 0.
|
| 59 |
-
| 0.
|
| 60 |
-
| 0.
|
| 61 |
-
| 0.
|
| 62 |
-
| 0.
|
| 63 |
-
| 0.
|
| 64 |
-
| 0.
|
| 65 |
-
| 0.
|
|
|
|
|
|
|
| 66 |
|
| 67 |
|
| 68 |
### Framework versions
|
| 69 |
|
| 70 |
-
- Transformers 4.
|
| 71 |
-
- Pytorch 2.1.
|
| 72 |
-
- Datasets 2.
|
| 73 |
-
- Tokenizers 0.15.
|
|
|
|
| 2 |
license: apache-2.0
|
| 3 |
base_model: google/vit-base-patch16-224-in21k
|
| 4 |
tags:
|
|
|
|
| 5 |
- generated_from_trainer
|
| 6 |
+
datasets:
|
| 7 |
+
- imagefolder
|
| 8 |
metrics:
|
| 9 |
- accuracy
|
| 10 |
model-index:
|
| 11 |
- name: Action_all_10_class
|
| 12 |
+
results:
|
| 13 |
+
- task:
|
| 14 |
+
name: Image Classification
|
| 15 |
+
type: image-classification
|
| 16 |
+
dataset:
|
| 17 |
+
name: imagefolder
|
| 18 |
+
type: imagefolder
|
| 19 |
+
config: default
|
| 20 |
+
split: train
|
| 21 |
+
args: default
|
| 22 |
+
metrics:
|
| 23 |
+
- name: Accuracy
|
| 24 |
+
type: accuracy
|
| 25 |
+
value: 0.8785276073619632
|
| 26 |
---
|
| 27 |
|
| 28 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
|
| 30 |
|
| 31 |
# Action_all_10_class
|
| 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.4146
|
| 36 |
+
- Accuracy: 0.8785
|
| 37 |
|
| 38 |
## Model description
|
| 39 |
|
|
|
|
| 65 |
|
| 66 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 67 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 68 |
+
| 1.1239 | 0.35 | 100 | 0.9934 | 0.7117 |
|
| 69 |
+
| 0.8878 | 0.69 | 200 | 0.7667 | 0.7706 |
|
| 70 |
+
| 0.9037 | 1.04 | 300 | 0.6369 | 0.8098 |
|
| 71 |
+
| 0.7307 | 1.38 | 400 | 0.5772 | 0.8319 |
|
| 72 |
+
| 0.6624 | 1.73 | 500 | 0.6925 | 0.7718 |
|
| 73 |
+
| 0.5781 | 2.08 | 600 | 0.5439 | 0.8405 |
|
| 74 |
+
| 0.5537 | 2.42 | 700 | 0.5257 | 0.8331 |
|
| 75 |
+
| 0.4112 | 2.77 | 800 | 0.4500 | 0.8564 |
|
| 76 |
+
| 0.3263 | 3.11 | 900 | 0.4911 | 0.8417 |
|
| 77 |
+
| 0.4592 | 3.46 | 1000 | 0.4551 | 0.8712 |
|
| 78 |
+
| 0.3204 | 3.81 | 1100 | 0.4325 | 0.8724 |
|
| 79 |
+
| 0.3268 | 4.15 | 1200 | 0.4529 | 0.8540 |
|
| 80 |
+
| 0.4267 | 4.5 | 1300 | 0.4356 | 0.8724 |
|
| 81 |
+
| 0.2886 | 4.84 | 1400 | 0.4146 | 0.8785 |
|
| 82 |
|
| 83 |
|
| 84 |
### Framework versions
|
| 85 |
|
| 86 |
+
- Transformers 4.39.3
|
| 87 |
+
- Pytorch 2.1.2
|
| 88 |
+
- Datasets 2.18.0
|
| 89 |
+
- Tokenizers 0.15.2
|
config.json
CHANGED
|
@@ -11,11 +11,11 @@
|
|
| 11 |
"id2label": {
|
| 12 |
"0": "\u0995\u09a5\u09be_\u09ac\u09b2\u09be",
|
| 13 |
"1": "\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
|
| 14 |
-
"2": "\u0996\u09be\u0993\
|
| 15 |
"3": "\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
|
| 16 |
-
"4": "\u0998\u09c1\u09ae\u09be\u09a8\
|
| 17 |
-
"5": "\u09aa\
|
| 18 |
-
"6": "\u09aa\
|
| 19 |
"7": "\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be",
|
| 20 |
"8": "\u09b2\u09c7\u0996\u09be",
|
| 21 |
"9": "\u09b9\u09be\u0981\u099f\u09be"
|
|
@@ -26,11 +26,11 @@
|
|
| 26 |
"label2id": {
|
| 27 |
"\u0995\u09a5\u09be_\u09ac\u09b2\u09be": "0",
|
| 28 |
"\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be": "1",
|
| 29 |
-
"\u0996\u09be\u0993\
|
| 30 |
"\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be": "3",
|
| 31 |
-
"\u0998\u09c1\u09ae\u09be\u09a8\
|
| 32 |
-
"\u09aa\
|
| 33 |
-
"\u09aa\
|
| 34 |
"\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be": "7",
|
| 35 |
"\u09b2\u09c7\u0996\u09be": "8",
|
| 36 |
"\u09b9\u09be\u0981\u099f\u09be": "9"
|
|
@@ -44,5 +44,5 @@
|
|
| 44 |
"problem_type": "single_label_classification",
|
| 45 |
"qkv_bias": true,
|
| 46 |
"torch_dtype": "float32",
|
| 47 |
-
"transformers_version": "4.
|
| 48 |
}
|
|
|
|
| 11 |
"id2label": {
|
| 12 |
"0": "\u0995\u09a5\u09be_\u09ac\u09b2\u09be",
|
| 13 |
"1": "\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
|
| 14 |
+
"2": "\u0996\u09be\u0993\u09df\u09be",
|
| 15 |
"3": "\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
|
| 16 |
+
"4": "\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
|
| 17 |
+
"5": "\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
|
| 18 |
+
"6": "\u09aa\u09dc\u09be",
|
| 19 |
"7": "\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be",
|
| 20 |
"8": "\u09b2\u09c7\u0996\u09be",
|
| 21 |
"9": "\u09b9\u09be\u0981\u099f\u09be"
|
|
|
|
| 26 |
"label2id": {
|
| 27 |
"\u0995\u09a5\u09be_\u09ac\u09b2\u09be": "0",
|
| 28 |
"\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be": "1",
|
| 29 |
+
"\u0996\u09be\u0993\u09df\u09be": "2",
|
| 30 |
"\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be": "3",
|
| 31 |
+
"\u0998\u09c1\u09ae\u09be\u09a8\u09cb": "4",
|
| 32 |
+
"\u09aa\u09be\u09a8_\u0995\u09b0\u09be": "5",
|
| 33 |
+
"\u09aa\u09dc\u09be": "6",
|
| 34 |
"\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be": "7",
|
| 35 |
"\u09b2\u09c7\u0996\u09be": "8",
|
| 36 |
"\u09b9\u09be\u0981\u099f\u09be": "9"
|
|
|
|
| 44 |
"problem_type": "single_label_classification",
|
| 45 |
"qkv_bias": true,
|
| 46 |
"torch_dtype": "float32",
|
| 47 |
+
"transformers_version": "4.39.3"
|
| 48 |
}
|
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:378bc1b2a46162fca2d8cbae6c384887e379afe64609c177c0f1d7b05a3e6ea6
|
| 3 |
size 343248584
|
preprocessor_config.json
CHANGED
|
@@ -1,4 +1,18 @@
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
"do_normalize": true,
|
| 3 |
"do_rescale": true,
|
| 4 |
"do_resize": true,
|
|
|
|
| 1 |
{
|
| 2 |
+
"_valid_processor_keys": [
|
| 3 |
+
"images",
|
| 4 |
+
"do_resize",
|
| 5 |
+
"size",
|
| 6 |
+
"resample",
|
| 7 |
+
"do_rescale",
|
| 8 |
+
"rescale_factor",
|
| 9 |
+
"do_normalize",
|
| 10 |
+
"image_mean",
|
| 11 |
+
"image_std",
|
| 12 |
+
"return_tensors",
|
| 13 |
+
"data_format",
|
| 14 |
+
"input_data_format"
|
| 15 |
+
],
|
| 16 |
"do_normalize": true,
|
| 17 |
"do_rescale": true,
|
| 18 |
"do_resize": true,
|
runs/Apr30_14-52-00_e47bc2c7ac19/events.out.tfevents.1714488721.e47bc2c7ac19.34.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e0c4afa52d404198d85d29150d97d4e399c04df1c17f1cdae9b1ba15e86960f
|
| 3 |
+
size 40828
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:405d5f7c60f45612e847cd409de8106be3239247e18b525f29a24d38270667fb
|
| 3 |
+
size 4920
|