Model save

README.md

@@ -2,7 +2,6 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - renovation
@@ -15,7 +14,7 @@ model-index:
       name: Image Classification
       type: image-classification
     dataset:
-      name: beans
+      name: renovation
       type: renovation
       config: default
       split: validation
@@ -23,7 +22,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.6575342465753424
+      value: 0.6986301369863014
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-beans-demo-v5
 
-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 beans 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 the renovation dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7925
-- Accuracy: 0.6575
+- Loss: 1.0644
+- Accuracy: 0.6986
 
 ## Model description
 
@@ -66,10 +65,10 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.1711        | 0.81  | 100  | 1.0255          | 0.5982   |
-| 0.7083        | 1.61  | 200  | 0.7925          | 0.6575   |
-| 0.2479        | 2.42  | 300  | 0.8712          | 0.6941   |
-| 0.127         | 3.23  | 400  | 0.8440          | 0.6941   |
+| 0.6438        | 0.81  | 100  | 0.9295          | 0.6347   |
+| 0.3105        | 1.61  | 200  | 0.9350          | 0.6575   |
+| 0.0634        | 2.42  | 300  | 1.0782          | 0.6895   |
+| 0.0257        | 3.23  | 400  | 1.0644          | 0.6986   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d8d9ca9cd14d0fb9a19ef6219bd49ca53f46a719bb98109e6f0a81e29d015c8c
+oid sha256:8bc0e701d02c681d64cf03f95b2e4e3c64596003ffa835c199c1a4c94f4e600d
 size 343239356

runs/Mar23_01-10-51_3ec072334407/events.out.tfevents.1711156269.3ec072334407.4424.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:09fd71f3395b493786d1bad7001481882b32342be777d2e6deeb00e11731d1cd
-size 14596
+oid sha256:e1435e2e352b1ac10d0bfe27a075d0bbe5ddc02652784d4b70492e7616fe6e16
+size 16849