End of training

README.md

@@ -12,11 +12,11 @@ probably proofread and complete it, then remove this comment. -->
 
 # shbA/food_classifier
 
-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 [facebook/dino-vitb16](https://huggingface.co/facebook/dino-vitb16) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.6782
-- Validation Loss: 0.7004
-- Train Accuracy: 0.7391
+- Train Loss: 0.4493
+- Validation Loss: 0.4618
+- Train Accuracy: 0.8043
 - Epoch: 6
 
 ## Model description
@@ -43,13 +43,13 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 1.0935     | 1.0228          | 0.6304         | 0     |
-| 1.0469     | 0.9620          | 0.7174         | 1     |
-| 0.9548     | 0.8669          | 0.8043         | 2     |
-| 0.8802     | 0.7712          | 0.7391         | 3     |
-| 0.8107     | 0.6756          | 0.8261         | 4     |
-| 0.7338     | 0.5727          | 0.8478         | 5     |
-| 0.6782     | 0.7004          | 0.7391         | 6     |
+| 1.2277     | 0.8208          | 0.6087         | 0     |
+| 0.7967     | 0.4934          | 0.8261         | 1     |
+| 0.5488     | 0.5541          | 0.7391         | 2     |
+| 0.6664     | 1.0464          | 0.6304         | 3     |
+| 0.7975     | 1.1285          | 0.5652         | 4     |
+| 0.6134     | 0.8258          | 0.6957         | 5     |
+| 0.4493     | 0.4618          | 0.8043         | 6     |
 
 
 ### Framework versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "facebook/dino-vitb16",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -28,5 +28,6 @@
   "num_hidden_layers": 12,
   "patch_size": 16,
   "qkv_bias": true,
+  "torch_dtype": "float32",
   "transformers_version": "4.29.1"
 }

preprocessor_config.json

@@ -3,15 +3,15 @@
   "do_rescale": true,
   "do_resize": true,
   "image_mean": [
-    0.5,
-    0.5,
-    0.5
+    0.485,
+    0.456,
+    0.406
   ],
   "image_processor_type": "ViTImageProcessor",
   "image_std": [
-    0.5,
-    0.5,
-    0.5
+    0.229,
+    0.224,
+    0.225
   ],
   "resample": 2,
   "rescale_factor": 0.00392156862745098,

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:50360fa6935fe9264a20fea00732028bed006766bf2cd8e20d73fc95e3edc056
+oid sha256:831d89bffc0c679feffdc68af98122c90fc885a2d208843c43849c10ee64a58b
 size 343472824