update model card README.md
Browse files
README.md
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
tags:
|
4 |
-
- image-classification
|
5 |
- generated_from_trainer
|
6 |
datasets:
|
7 |
- imagefolder
|
@@ -15,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
# vit-base-beans-demo-v5
|
17 |
|
18 |
-
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
|
19 |
|
20 |
## Model description
|
21 |
|
@@ -49,7 +48,7 @@ The following hyperparameters were used during training:
|
|
49 |
|
50 |
### Framework versions
|
51 |
|
52 |
-
- Transformers 4.
|
53 |
-
- Pytorch 1.
|
54 |
-
- Datasets 2.
|
55 |
- Tokenizers 0.13.2
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
tags:
|
|
|
4 |
- generated_from_trainer
|
5 |
datasets:
|
6 |
- imagefolder
|
|
|
14 |
|
15 |
# vit-base-beans-demo-v5
|
16 |
|
17 |
+
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.
|
18 |
|
19 |
## Model description
|
20 |
|
|
|
48 |
|
49 |
### Framework versions
|
50 |
|
51 |
+
- Transformers 4.25.1
|
52 |
+
- Pytorch 1.13.0+cu116
|
53 |
+
- Datasets 2.8.0
|
54 |
- Tokenizers 0.13.2
|