candylion

README.md

@@ -0,0 +1,70 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: google/vit-base-patch16-224-in21k
+tags:
+- HHD
+- 3_class
+- ViT
+- generated_from_trainer
+model-index:
+- name: ViT_face
+  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. -->
+
+# ViT_face
+
+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 face dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.5561
+
+## 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: 2e-05
+- train_batch_size: 32
+- eval_batch_size: 32
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 10
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| No log        | 1.0   | 38   | 1.0143          |
+| No log        | 2.0   | 76   | 0.8770          |
+| No log        | 3.0   | 114  | 0.7350          |
+| No log        | 4.0   | 152  | 0.6426          |
+| No log        | 5.0   | 190  | 0.5853          |
+| No log        | 6.0   | 228  | 0.5629          |
+| No log        | 7.0   | 266  | 0.5550          |
+| No log        | 8.0   | 304  | 0.5535          |
+| No log        | 9.0   | 342  | 0.5580          |
+| No log        | 10.0  | 380  | 0.5561          |
+
+
+### Framework versions
+
+- Transformers 4.44.2
+- Pytorch 2.4.1+cu121
+- Datasets 3.0.0
+- Tokenizers 0.19.1

config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "\u1105\u1172\u110c\u116e\u11ab\u110b\u1167\u11af",
+    "1": "\u1109\u1169\u11bc\u110c\u116e\u11bc\u1100\u1175",
+    "2": "\u1112\u1167\u11ab\u1107\u1175\u11ab"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "\u1105\u1172\u110c\u116e\u11ab\u110b\u1167\u11af": "0",
+    "\u1109\u1169\u11bc\u110c\u116e\u11bc\u1100\u1175": "1",
+    "\u1112\u1167\u11ab\u1107\u1175\u11ab": "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.44.2"
+}

model.safetensors

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

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/Sep03_01-50-14_481ed07ae1fe/events.out.tfevents.1725329107.481ed07ae1fe.258.0

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

runs/Sep03_02-21-04_481ed07ae1fe/events.out.tfevents.1725330069.481ed07ae1fe.258.1

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

runs/Sep03_02-21-04_481ed07ae1fe/events.out.tfevents.1725330098.481ed07ae1fe.258.2

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

runs/Sep03_02-22-39_481ed07ae1fe/events.out.tfevents.1725330163.481ed07ae1fe.258.3

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

runs/Sep03_02-25-04_481ed07ae1fe/events.out.tfevents.1725330310.481ed07ae1fe.258.4

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

runs/Sep05_11-32-32_5e43c2898260/events.out.tfevents.1725535955.5e43c2898260.578.0

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

runs/Sep08_09-40-59_0c5c756f8330/events.out.tfevents.1725792162.0c5c756f8330.4030.0

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

runs/Sep21_06-18-56_f739677fd135/events.out.tfevents.1726899539.f739677fd135.885.0

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

runs/Sep21_06-33-11_f739677fd135/events.out.tfevents.1726900395.f739677fd135.5929.0

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

training_args.bin

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