End of training

README.md

@@ -0,0 +1,92 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: google/vit-base-patch16-224
+tags:
+- generated_from_trainer
+metrics:
+- accuracy
+- f1
+model-index:
+- name: beans_ViT
+  results: []
+---
+
+<!-- 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. -->
+
+# beans_ViT
+
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.2997
+- Accuracy: 0.7969
+- F1: 0.7991
+
+## 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: 64
+- eval_batch_size: 64
+- seed: 42
+- optimizer: Use 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: 30
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| No log        | 1.0   | 17   | 0.8693          | 0.6090   | 0.5953 |
+| No log        | 2.0   | 34   | 0.9652          | 0.6015   | 0.5977 |
+| No log        | 3.0   | 51   | 0.7178          | 0.6992   | 0.6927 |
+| No log        | 4.0   | 68   | 0.7488          | 0.6992   | 0.6955 |
+| No log        | 5.0   | 85   | 0.6517          | 0.7068   | 0.7070 |
+| No log        | 6.0   | 102  | 0.7816          | 0.6842   | 0.6541 |
+| No log        | 7.0   | 119  | 0.5014          | 0.7744   | 0.7733 |
+| No log        | 8.0   | 136  | 0.5321          | 0.7669   | 0.7680 |
+| No log        | 9.0   | 153  | 0.5985          | 0.7444   | 0.7457 |
+| No log        | 10.0  | 170  | 0.4675          | 0.8271   | 0.8274 |
+| No log        | 11.0  | 187  | 0.5750          | 0.7744   | 0.7576 |
+| No log        | 12.0  | 204  | 0.6617          | 0.7293   | 0.7066 |
+| No log        | 13.0  | 221  | 0.6396          | 0.7594   | 0.7577 |
+| No log        | 14.0  | 238  | 0.4302          | 0.8346   | 0.8352 |
+| No log        | 15.0  | 255  | 0.4018          | 0.8421   | 0.8427 |
+| No log        | 16.0  | 272  | 0.5673          | 0.7895   | 0.7883 |
+| No log        | 17.0  | 289  | 0.5037          | 0.8120   | 0.8097 |
+| No log        | 18.0  | 306  | 0.5939          | 0.8496   | 0.8487 |
+| No log        | 19.0  | 323  | 0.6590          | 0.8120   | 0.8111 |
+| No log        | 20.0  | 340  | 0.6060          | 0.8571   | 0.8559 |
+| No log        | 21.0  | 357  | 0.5806          | 0.8421   | 0.8418 |
+| No log        | 22.0  | 374  | 0.6180          | 0.8421   | 0.8414 |
+| No log        | 23.0  | 391  | 0.7707          | 0.7669   | 0.7633 |
+| No log        | 24.0  | 408  | 0.5440          | 0.8421   | 0.8418 |
+| No log        | 25.0  | 425  | 0.6596          | 0.8496   | 0.8497 |
+| No log        | 26.0  | 442  | 0.5393          | 0.8346   | 0.8342 |
+| No log        | 27.0  | 459  | 0.6320          | 0.8797   | 0.8795 |
+| No log        | 28.0  | 476  | 0.5903          | 0.8496   | 0.8507 |
+| No log        | 29.0  | 493  | 0.6826          | 0.8647   | 0.8644 |
+| 0.3346        | 30.0  | 510  | 0.6493          | 0.8571   | 0.8567 |
+
+
+### Framework versions
+
+- Transformers 4.47.0
+- Pytorch 2.5.1+cu121
+- Datasets 3.3.1
+- Tokenizers 0.21.0

config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.47.0"
+}

model.safetensors

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

preprocessor_config.json

@@ -0,0 +1,23 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessorFast",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

training_args.bin

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