End of training

README.md

@@ -0,0 +1,80 @@
+---
+license: apache-2.0
+base_model: google/vit-base-patch16-224-in21k
+tags:
+- generated_from_trainer
+datasets:
+- food101
+metrics:
+- accuracy
+model-index:
+- name: my_awesome_food_model
+  results:
+  - task:
+      name: Image Classification
+      type: image-classification
+    dataset:
+      name: food101
+      type: food101
+      config: default
+      split: train[:1000]
+      args: default
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.845
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# my_awesome_food_model
+
+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 food101 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.4052
+- Accuracy: 0.845
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 16
+- eval_batch_size: 16
+- seed: 42
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 64
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 4.3315        | 0.96  | 12   | 3.4645          | 0.815    |
+| 3.2688        | 2.0   | 25   | 2.5844          | 0.785    |
+| 2.6586        | 2.88  | 36   | 2.4052          | 0.845    |
+
+
+### Framework versions
+
+- Transformers 4.35.2
+- Pytorch 2.1.0+cu118
+- Datasets 2.15.0
+- Tokenizers 0.15.0

config.json

@@ -226,5 +226,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.34.1"
+  "transformers_version": "4.35.2"
 }

model.safetensors

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

runs/Dec06_21-08-29_926b81d367df/events.out.tfevents.1701896914.926b81d367df.556.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28906ab3eb99d17fafae59fd850db2dd3fdad4f0bd2f4946193e6b7396ed7063
-size 4536
+oid sha256:6cb727a6b0c0e6b6d4d5f41f6b72ba4789dabd0312e8161f69eaf4def70f585c
+size 4600