heado

README.md

@@ -17,9 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ViT_dog_food
 
-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 dog_food 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 lewtun/dog_food dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4453
+- Loss: 0.4318
 
 ## Model description
 
@@ -44,20 +44,22 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 36   | 0.4453          |
-| No log        | 2.0   | 72   | 0.2022          |
-| No log        | 3.0   | 108  | 0.1640          |
+| No log        | 1.0   | 36   | 0.4318          |
+| No log        | 2.0   | 72   | 0.1543          |
+| No log        | 3.0   | 108  | 0.1014          |
+| No log        | 4.0   | 144  | 0.0853          |
+| No log        | 5.0   | 180  | 0.0807          |
 
 
 ### Framework versions
 
-- Transformers 4.50.3
+- Transformers 4.55.0
 - Pytorch 2.6.0+cu124
-- Datasets 3.5.0
-- Tokenizers 0.21.1
+- Datasets 4.0.0
+- Tokenizers 0.21.4

config.json

@@ -8,9 +8,9 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2"
+    "0": "chicken",
+    "1": "dog",
+    "2": "muffin"
   },
   "image_size": 224,
   "initializer_range": 0.02,
@@ -31,5 +31,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.50.3"
+  "transformers_version": "4.55.0"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:44e3d3758219c52439f0d955107f0784f8f2ff6de28331ec63c413b179c3ab02
+oid sha256:4a0e197af465baca08afc94cd732d59abf5fc4c1ce768f49ea61fd3d4003258a
 size 343227052

runs/Aug14_05-40-42_24fbe35260b1/events.out.tfevents.1755150540.24fbe35260b1.262.0

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

runs/Aug14_05-52-26_4e7432784e33/events.out.tfevents.1755150759.4e7432784e33.502.0

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

runs/Aug14_05-52-26_4e7432784e33/events.out.tfevents.1755151656.4e7432784e33.502.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6e5076e850430d8ffe69f8bd4455ae1ec50dc8c190ad5bd8c30a76473712ddb3
+oid sha256:c975fcc668bd2ec79afa497fd6c3dbf3628a75d6ecaf26171284fcc11e2cff09
 size 5368