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-augmented
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: custom_dataset_augmented
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.4672131147540984
-    - name: F1
-      type: f1
-      value: 0.08120213120213123
+  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-augmented
 
-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_augmented dataset.
-It achieves the following results on the evaluation set:
-- Loss: 1.5780
-- Accuracy: 0.4672
-- F1: 0.0812
+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
 
@@ -64,7 +40,7 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 10
 
 ### Training results
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ae669a49423730714a1c328ed8f24560b788be49d6564885eeb48ae38c16d38b
+oid sha256:156a2a0c5abbe677a5d26824b6069f51313a41eeaa0067fbfd0662b23499b718
 size 343302829

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20481d13de648234ef28a37423c68c256157ffc5e4cde3b5b65668b2492cfa15
+oid sha256:339d3326c66569fc004b5fc4b75aceb0f60c6ddfd04d0c867c4de1279017c74f
 size 4091