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
 metrics:
 - accuracy
@@ -16,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-beans-tutorial
 
-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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3187
-- Accuracy: 0.9609
+- Loss: 0.1777
+- Accuracy: 0.9850
 
 ## Model description
 
@@ -39,9 +38,9 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0002
-- train_batch_size: 8
+- train_batch_size: 16
 - eval_batch_size: 8
-- seed: 1337
+- seed: 1787
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 5
@@ -51,12 +50,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.0           | 0.77  | 100  | 0.0780          | 0.9925   |
-| 0.0           | 1.54  | 200  | 0.1820          | 0.9774   |
-| 0.0           | 2.31  | 300  | 0.1846          | 0.9774   |
-| 0.0           | 3.08  | 400  | 0.1543          | 0.9774   |
-| 0.0           | 3.85  | 500  | 0.1396          | 0.9850   |
-| 0.0           | 4.62  | 600  | 0.1436          | 0.9850   |
+| 0.0           | 1.54  | 100  | 0.1658          | 0.9850   |
+| 0.0           | 3.08  | 200  | 0.1742          | 0.9850   |
+| 0.0           | 4.62  | 300  | 0.1777          | 0.9850   |
 
 
 ### Framework versions

model.safetensors

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

runs/Jan24_01-31-50_579caf070ae1/events.out.tfevents.1706059916.579caf070ae1.3225.11

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8965fb6285f2733062dd00a8e92dd5425b2a6746ab99c8ef2d35213b16408052
-size 10028
+oid sha256:59e7f58b61cf26ed8ead641ab9cbf26b757008128b09f331d29c110affc36e87
+size 10696