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 |
}
|