Upload ViTForImageClassification
Browse files- README.md +3 -1
- config.json +2 -2
README.md
CHANGED
@@ -1,8 +1,10 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
-
base_model: google/vit-base-patch16-224-in21k
|
4 |
tags:
|
5 |
- generated_from_trainer
|
|
|
|
|
|
|
6 |
metrics:
|
7 |
- accuracy
|
8 |
model-index:
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
3 |
tags:
|
4 |
- generated_from_trainer
|
5 |
+
- image-classification
|
6 |
+
- vision
|
7 |
+
base_model: google/vit-base-patch16-224-in21k
|
8 |
metrics:
|
9 |
- accuracy
|
10 |
model-index:
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"ViTForImageClassification"
|
5 |
],
|
@@ -38,5 +38,5 @@
|
|
38 |
"problem_type": "single_label_classification",
|
39 |
"qkv_bias": true,
|
40 |
"torch_dtype": "float32",
|
41 |
-
"transformers_version": "4.38.
|
42 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "tuphamdf/skincare-detection",
|
3 |
"architectures": [
|
4 |
"ViTForImageClassification"
|
5 |
],
|
|
|
38 |
"problem_type": "single_label_classification",
|
39 |
"qkv_bias": true,
|
40 |
"torch_dtype": "float32",
|
41 |
+
"transformers_version": "4.38.1"
|
42 |
}
|