Abhiram4
/

VitTea

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

Abhiram4 commited on Jun 5

Commit

c99214a

•

1 Parent(s): c24b380

Model save

Files changed (2) hide show

README.md +6 -6
pytorch_model.bin +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.5056179775280899
 ---
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 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 image_folder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.9271
-- Accuracy: 0.5056
 ## Model description
@@ -67,9 +67,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 0.8   | 2    | 2.0343          | 0.2472   |
-| No log        | 2.0   | 5    | 1.9372          | 0.4831   |
-| No log        | 2.4   | 6    | 1.9271          | 0.5056   |
 ### Framework versions

     metrics:
     - name: Accuracy
       type: accuracy
+      value: 0.30337078651685395
 ---
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 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 image_folder dataset.
 It achieves the following results on the evaluation set:
+- Loss: 1.9377
+- Accuracy: 0.3034
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 0.8   | 2    | 2.0404          | 0.1573   |
+| No log        | 2.0   | 5    | 1.9487          | 0.3034   |
+| No log        | 2.4   | 6    | 1.9377          | 0.3034   |
 ### Framework versions

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:9f7b1ea05bce6add5f416590e8678b6fdbd476513d021305767a0f2337595292
 size 343287149