End of training

README.md

@@ -2,12 +2,29 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 datasets:
 - imagefolder
+metrics:
+- accuracy
 model-index:
 - name: vit-base-beans
- results: []
+ results:
+ - task:
+ name: Image Classification
+ type: image-classification
+ dataset:
+ name: imagefolder
+ type: imagefolder
+ config: default
+ split: validation
+ args: default
+ metrics:
+ - name: Accuracy
+ type: accuracy
+ value: 0.9772129710780018
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +33,9 @@ should probably proofread and complete it, then remove this comment. -->
 # vit-base-beans
 
 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.
+It achieves the following results on the evaluation set:
+- Loss: 0.1584
+- Accuracy: 0.9772
 
 ## Model description
 

all_results.json

@@ -1,13 +1,12 @@
 {
  "epoch": 5.0,
- "eval_accuracy": 0.9774436090225563,
- "eval_loss": 0.12084060162305832,
- "eval_runtime": 1.7601,
- "eval_samples_per_second": 75.563,
- "eval_steps_per_second": 5.113,
- "total_flos": 4.006371770595533e+17,
- "train_loss": 0.29627929100623496,
- "train_runtime": 105.6493,
- "train_samples_per_second": 48.935,
- "train_steps_per_second": 3.076
+ "eval_accuracy": 0.9772129710780018,
+ "eval_loss": 0.1584448516368866,
+ "eval_runtime": 44.036,
+ "eval_samples_per_second": 25.911,
+ "eval_steps_per_second": 1.635,
+ "train_loss": 0.006059203078136595,
+ "train_runtime": 4479.7513,
+ "train_samples_per_second": 43.876,
+ "train_steps_per_second": 2.742
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
  "epoch": 5.0,
- "eval_accuracy": 0.9774436090225563,
- "eval_loss": 0.12084060162305832,
- "eval_runtime": 1.7601,
- "eval_samples_per_second": 75.563,
- "eval_steps_per_second": 5.113
+ "eval_accuracy": 0.9772129710780018,
+ "eval_loss": 0.1584448516368866,
+ "eval_runtime": 44.036,
+ "eval_samples_per_second": 25.911,
+ "eval_steps_per_second": 1.635
 }

runs/May22_15-17-42_DL-Station/events.out.tfevents.1716363273.DL-Station.2893395.1

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

train_results.json

@@ -1,8 +1,7 @@
 {
  "epoch": 5.0,
- "total_flos": 4.006371770595533e+17,
- "train_loss": 0.29627929100623496,
- "train_runtime": 105.6493,
- "train_samples_per_second": 48.935,
- "train_steps_per_second": 3.076
+ "train_loss": 0.006059203078136595,
+ "train_runtime": 4479.7513,
+ "train_samples_per_second": 43.876,
+ "train_steps_per_second": 2.742
 }