Model save

README.md

@@ -2,32 +2,13 @@
 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: beans
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 1.0
-    - name: F1
-      type: f1
-      value: 1.0
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,11 +16,11 @@ 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 beans dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1679
-- Accuracy: 1.0
-- F1: 1.0
+- Loss: 7.3529
+- Accuracy: 0.0596
+- F1: 0.0075
 
 ## Model description
 
@@ -68,6 +49,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| 0.3773        | 2.54  | 1000 | 7.3529          | 0.0596   | 0.0075 |
 
 
 ### Framework versions