End of training

README.md

@@ -3,26 +3,11 @@ license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
-datasets:
-- imagefolder
 metrics:
 - accuracy
 model-index:
 - name: image_classification
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: imagefolder
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.39375
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # image_classification
 
-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.
+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.
 It achieves the following results on the evaluation set:
-- Loss: 1.6084
-- Accuracy: 0.3937
+- Loss: 4.6845
+- Accuracy: 0.0626
 
 ## Model description
 
@@ -64,14 +49,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 40   | 1.8874          | 0.325    |
-| No log        | 2.0   | 80   | 1.6683          | 0.3812   |
-| No log        | 3.0   | 120  | 1.6096          | 0.4313   |
+| 0.6177        | 1.0   | 788  | 4.5441          | 0.0572   |
+| 0.6328        | 2.0   | 1576 | 4.6145          | 0.0628   |
+| 0.5851        | 3.0   | 2364 | 4.6799          | 0.0648   |
 
 
 ### Framework versions
 
-- Transformers 4.33.1
+- Transformers 4.33.2
 - Pytorch 2.0.1+cu118
 - Datasets 2.14.5
 - Tokenizers 0.13.3