heado

README.md

@@ -19,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 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:
-- Loss: 0.6899
+- Loss: 0.7528
 
 ## Model description
 
@@ -42,21 +42,24 @@ The following hyperparameters were used during training:
 - train_batch_size: 64
 - eval_batch_size: 64
 - seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- 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: 2
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 17   | 0.7851          |
-| No log        | 2.0   | 34   | 0.6899          |
+| No log        | 1.0   | 17   | 0.7528          |
+| No log        | 2.0   | 34   | 0.5402          |
+| No log        | 3.0   | 51   | 0.4056          |
+| No log        | 4.0   | 68   | 0.3359          |
+| No log        | 5.0   | 85   | 0.3176          |
 
 
 ### Framework versions
 
-- Transformers 4.44.2
-- Pytorch 2.5.0+cu121
-- Datasets 3.0.2
-- Tokenizers 0.19.1
+- Transformers 4.50.3
+- Pytorch 2.6.0+cu124
+- Datasets 3.5.0
+- Tokenizers 0.21.1

config.json

@@ -1,5 +1,4 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -9,17 +8,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "angular_leaf_spot",
-    "1": "bean_rust",
-    "2": "healthy"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "angular_leaf_spot": "0",
-    "bean_rust": "1",
-    "healthy": "2"
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -27,8 +26,10 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
+  "pooler_act": "tanh",
+  "pooler_output_size": 768,
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.44.2"
+  "transformers_version": "4.50.3"
 }

model.safetensors

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

preprocessor_config.json

@@ -1,4 +1,5 @@
 {
+  "do_convert_rgb": null,
   "do_normalize": true,
   "do_rescale": true,
   "do_resize": true,

runs/Apr09_22-39-48_eeefc6bd4371/events.out.tfevents.1744238402.eeefc6bd4371.1050.0

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

runs/Apr09_22-39-48_eeefc6bd4371/events.out.tfevents.1744238914.eeefc6bd4371.1050.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dc8da59856d9d93a6ad45b0ce8d6d492115b78edacab7bb733c28c4e5f787bb5
-size 5176
+oid sha256:832d1d04904ff9213481fd5eb9501d58fc0e76d07c535bfa9e3dc022df6a0728
+size 5368