Model save

README.md

@@ -2,32 +2,12 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - imagefolder
-metrics:
-- accuracy
-- f1
 model-index:
 - name: fine-tuned
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: custom_dataset
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.42857142857142855
-    - name: F1
-      type: f1
-      value: 0.23982683982683983
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,11 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # fine-tuned
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the custom_dataset dataset.
-It achieves the following results on the evaluation set:
-- Loss: 1.3223
-- Accuracy: 0.4286
-- F1: 0.2398
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
 
 ## Model description
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a86c9163e63b3ff2f6a72a24677b7ae6389c6035c65575c597befebe3540ef63
+oid sha256:dc37da75f14aaf46f9553f048b39092b29d885c927b5a7809bf917f777d54d20
 size 343296365

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a5f6b3765d3d98d244658f338cb0208b6495673a52a176903acaa78a98a36a2d
+oid sha256:80e7e1742be08c7d95390048243f80f191648bfb899c3023e71bcf0e5f232df8
 size 4027