End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 metrics:
 - accuracy
@@ -16,10 +18,10 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/aeroberts4444/huggingface/runs/120mmtvn)
 # vit-base-beans-2
 
-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.
+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.
 It achieves the following results on the evaluation set:
-- Accuracy: 0.0
-- Loss: 1.1626
+- Loss: 1.1599
+- Accuracy: 0.125
 
 ## Model description
 

all_results.json

@@ -2,12 +2,12 @@
     "epoch": 2.0,
     "eval_accuracy": 0.125,
     "eval_loss": 1.1599400043487549,
-    "eval_runtime": 0.178,
-    "eval_samples_per_second": 44.956,
-    "eval_steps_per_second": 5.619,
+    "eval_runtime": 0.3023,
+    "eval_samples_per_second": 26.461,
+    "eval_steps_per_second": 3.308,
     "total_flos": 1239882946412544.0,
-    "train_loss": 0.96820068359375,
-    "train_runtime": 21.5994,
-    "train_samples_per_second": 0.741,
-    "train_steps_per_second": 0.093
+    "train_loss": 0.0,
+    "train_runtime": 10.194,
+    "train_samples_per_second": 1.57,
+    "train_steps_per_second": 0.196
 }

config.json

@@ -28,6 +28,7 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.41.0.dev0"

eval_results.json

@@ -2,7 +2,7 @@
     "epoch": 2.0,
     "eval_accuracy": 0.125,
     "eval_loss": 1.1599400043487549,
-    "eval_runtime": 0.178,
-    "eval_samples_per_second": 44.956,
-    "eval_steps_per_second": 5.619
+    "eval_runtime": 0.3023,
+    "eval_samples_per_second": 26.461,
+    "eval_steps_per_second": 3.308
 }

runs/May16_18-16-28_amys-mbp-2.taildb5d.ts.net/events.out.tfevents.1715879844.amys-mbp-2.taildb5d.ts.net

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

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 2.0,
     "total_flos": 1239882946412544.0,
-    "train_loss": 0.96820068359375,
-    "train_runtime": 21.5994,
-    "train_samples_per_second": 0.741,
-    "train_steps_per_second": 0.093
+    "train_loss": 0.0,
+    "train_runtime": 10.194,
+    "train_samples_per_second": 1.57,
+    "train_steps_per_second": 0.196
 }

trainer_state.json

@@ -30,10 +30,10 @@
       "epoch": 2.0,
       "step": 2,
       "total_flos": 1239882946412544.0,
-      "train_loss": 0.96820068359375,
-      "train_runtime": 21.5994,
-      "train_samples_per_second": 0.741,
-      "train_steps_per_second": 0.093
+      "train_loss": 0.0,
+      "train_runtime": 10.194,
+      "train_samples_per_second": 1.57,
+      "train_steps_per_second": 0.196
     }
   ],
   "logging_steps": 10,