Instructions to use anum231/class2_v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use anum231/class2_v1 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="anum231/class2_v1") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("anum231/class2_v1") model = AutoModelForImageClassification.from_pretrained("anum231/class2_v1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress epoch 3
Browse files- README.md +5 -4
- checkpoint/extra_data.pickle +1 -1
- checkpoint/weights.h5 +1 -1
- tf_model.h5 +1 -1
README.md
CHANGED
|
@@ -15,10 +15,10 @@ probably proofread and complete it, then remove this comment. -->
|
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
-
- Train Loss: 0.
|
| 19 |
-
- Validation Loss: 0.
|
| 20 |
-
- Train Accuracy: 0.
|
| 21 |
-
- Epoch:
|
| 22 |
|
| 23 |
## Model description
|
| 24 |
|
|
@@ -47,6 +47,7 @@ The following hyperparameters were used during training:
|
|
| 47 |
| 0.6922 | 0.6296 | 0.7241 | 0 |
|
| 48 |
| 0.6562 | 0.6218 | 0.6897 | 1 |
|
| 49 |
| 0.6217 | 0.5985 | 0.7586 | 2 |
|
|
|
|
| 50 |
|
| 51 |
|
| 52 |
### Framework versions
|
|
|
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
+
- Train Loss: 0.6168
|
| 19 |
+
- Validation Loss: 0.5412
|
| 20 |
+
- Train Accuracy: 0.8276
|
| 21 |
+
- Epoch: 3
|
| 22 |
|
| 23 |
## Model description
|
| 24 |
|
|
|
|
| 47 |
| 0.6922 | 0.6296 | 0.7241 | 0 |
|
| 48 |
| 0.6562 | 0.6218 | 0.6897 | 1 |
|
| 49 |
| 0.6217 | 0.5985 | 0.7586 | 2 |
|
| 50 |
+
| 0.6168 | 0.5412 | 0.8276 | 3 |
|
| 51 |
|
| 52 |
|
| 53 |
### Framework versions
|
checkpoint/extra_data.pickle
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 686417159
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:267a42422376bef6c0e374b7fb7dd4507dfd1943598cc28f23c9461507919216
|
| 3 |
size 686417159
|
checkpoint/weights.h5
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 343486136
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5b16d8af1cd3a111efb0af7c5ff2dc62c8d36f0b2506cfc3ba2622b3d064a87d
|
| 3 |
size 343486136
|
tf_model.h5
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 343486136
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5b16d8af1cd3a111efb0af7c5ff2dc62c8d36f0b2506cfc3ba2622b3d064a87d
|
| 3 |
size 343486136
|