Model save

README.md

@@ -4,6 +4,8 @@ license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: vit-base-beans-demo-v5
   results: []
@@ -17,7 +19,7 @@ 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 an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.4711
-- Roc Auc: nan
+- Accuracy: 0.5886
 
 ## Model description
 
@@ -47,9 +49,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Roc Auc |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 0.5163        | 1.0   | 28   | 0.4711          | nan     |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.5174        | 1.0   | 28   | 0.4711          | 0.5886   |
 
 
 ### Framework versions

runs/Jan15_19-02-27_a99b99d67614/events.out.tfevents.1736967750.a99b99d67614.505.14

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a909d5fd23deb2e0c3b778275d494a7c5135df082e0f585112e83425ea09d979
-size 17154
+oid sha256:7bea7c0ea5f65869caacfe9f5a47c1ed9065e84aab5ad972a7113ccb7eeb5107
+size 17819