Model save

README.md

@@ -2,11 +2,9 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
-- vision
 - generated_from_trainer
-metrics:
-- accuracy
+datasets:
+- imagefolder
 model-index:
 - name: vit-base-beans
   results: []
@@ -17,10 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-beans
 
-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 beans dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.1208
-- Accuracy: 0.9774
+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.
 
 ## Model description
 
@@ -53,7 +48,7 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.41.0
-- Pytorch 2.3.0+cu121
+- Transformers 4.39.2
+- Pytorch 1.13.1
 - Datasets 2.19.1
-- Tokenizers 0.19.1
+- Tokenizers 0.15.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4425516648950b55cd9c4254be563f05e1dac0be892a8646a3e3c4d0fc3bc782
+oid sha256:d779c9810e85809178c502f30a1114315a7bb0bec18da0f305c2b7c409c4b057
 size 343223968

runs/May22_15-17-42_DL-Station/events.out.tfevents.1716358722.DL-Station.2893395.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69d18bd0428db1134469d4ad03959ad76b15e6a7fc2938cd3af22dc00ba426da
-size 226198
+oid sha256:3390987b4f01d159bcd7c2fb7d0a8748f9173d611aa6dc0f476be08ad59f9070
+size 232460