Model save

README.md

@@ -0,0 +1,54 @@
+---
+license: apache-2.0
+base_model: google/vit-base-patch16-224
+tags:
+- generated_from_trainer
+datasets:
+- imagefolder
+model-index:
+- name: Fine-Tuned_Model
+  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. -->
+
+# Fine-Tuned_Model
+
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
+
+## 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: 32
+- eval_batch_size: 8
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 50
+
+### Training results
+
+
+
+### Framework versions
+
+- Transformers 4.37.2
+- Pytorch 2.1.0+cu121
+- Datasets 2.16.1
+- Tokenizers 0.15.1

config.json

@@ -0,0 +1,120 @@
+{
+  "_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": "aeroplane",
+    "1": "blackboard",
+    "10": "clouds",
+    "11": "coins",
+    "12": "cycles",
+    "13": "deer",
+    "14": "desk",
+    "15": "dogs",
+    "16": "dogsledge",
+    "17": "door-frame",
+    "18": "factory",
+    "19": "fireman",
+    "2": "boat",
+    "20": "firetruck",
+    "21": "food",
+    "22": "helicopter",
+    "23": "horse",
+    "24": "horsepipe",
+    "25": "instrument",
+    "26": "ladder",
+    "27": "lake",
+    "28": "landscape",
+    "29": "machinaries",
+    "3": "books",
+    "30": "mountains",
+    "31": "painting",
+    "32": "people",
+    "33": "pole",
+    "34": "railway",
+    "35": "river",
+    "36": "road",
+    "37": "ship",
+    "38": "ski",
+    "39": "sky",
+    "4": "bridge",
+    "40": "snow",
+    "41": "stairs",
+    "42": "table",
+    "43": "telephone",
+    "44": "tree",
+    "45": "water",
+    "5": "building",
+    "6": "car",
+    "7": "chair",
+    "8": "children",
+    "9": "church"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "aeroplane": "0",
+    "blackboard": "1",
+    "boat": "2",
+    "books": "3",
+    "bridge": "4",
+    "building": "5",
+    "car": "6",
+    "chair": "7",
+    "children": "8",
+    "church": "9",
+    "clouds": "10",
+    "coins": "11",
+    "cycles": "12",
+    "deer": "13",
+    "desk": "14",
+    "dogs": "15",
+    "dogsledge": "16",
+    "door-frame": "17",
+    "factory": "18",
+    "fireman": "19",
+    "firetruck": "20",
+    "food": "21",
+    "helicopter": "22",
+    "horse": "23",
+    "horsepipe": "24",
+    "instrument": "25",
+    "ladder": "26",
+    "lake": "27",
+    "landscape": "28",
+    "machinaries": "29",
+    "mountains": "30",
+    "painting": "31",
+    "people": "32",
+    "pole": "33",
+    "railway": "34",
+    "river": "35",
+    "road": "36",
+    "ship": "37",
+    "ski": "38",
+    "sky": "39",
+    "snow": "40",
+    "stairs": "41",
+    "table": "42",
+    "telephone": "43",
+    "tree": "44",
+    "water": "45"
+  },
+  "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.37.2"
+}

model.safetensors

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

preprocessor_config.json

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

runs/Feb07_12-03-42_8977fbbee6f1/events.out.tfevents.1707307430.8977fbbee6f1.5750.0

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

runs/Feb07_12-24-50_8977fbbee6f1/events.out.tfevents.1707308701.8977fbbee6f1.9585.0

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

training_args.bin

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