Upload folder using huggingface_hub

checkpoint-2130/config.json

@@ -0,0 +1,70 @@
+{
+  "_name_or_path": "Robys01/facial_age_estimator",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "01",
+    "1": "02",
+    "2": "03",
+    "3": "04-05",
+    "4": "06-07",
+    "5": "08-09",
+    "6": "10-12",
+    "7": "13-15",
+    "8": "16-20",
+    "9": "21-25",
+    "10": "26-30",
+    "11": "31-35",
+    "12": "36-40",
+    "13": "41-45",
+    "14": "46-50",
+    "15": "51-55",
+    "16": "56-60",
+    "17": "61-70",
+    "18": "71-80",
+    "19": "81-90",
+    "20": "90+"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "01": 0,
+    "02": 1,
+    "03": 2,
+    "04-05": 3,
+    "06-07": 4,
+    "08-09": 5,
+    "10-12": 6,
+    "13-15": 7,
+    "16-20": 8,
+    "21-25": 9,
+    "26-30": 10,
+    "31-35": 11,
+    "36-40": 12,
+    "41-45": 13,
+    "46-50": 14,
+    "51-55": 15,
+    "56-60": 16,
+    "61-70": 17,
+    "71-80": 18,
+    "81-90": 19,
+    "90+": 20
+  },
+  "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.41.2"
+}

checkpoint-2130/model.safetensors

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

checkpoint-2130/optimizer.pt

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

checkpoint-2130/preprocessor_config.json

@@ -0,0 +1,36 @@
+{
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "return_tensors",
+    "data_format",
+    "input_data_format"
+  ],
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-2130/rng_state.pth

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

checkpoint-2130/scheduler.pt

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

checkpoint-2130/trainer_state.json

@@ -0,0 +1,151 @@
+{
+  "best_metric": 1.6845855712890625,
+  "best_model_checkpoint": "facial_age_estimator/checkpoint-2130",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 2130,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.38575197889182056,
+      "eval_loss": 1.7030466794967651,
+      "eval_runtime": 15.5991,
+      "eval_samples_per_second": 121.482,
+      "eval_steps_per_second": 0.962,
+      "step": 213
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.3862796833773087,
+      "eval_loss": 1.7021386623382568,
+      "eval_runtime": 15.4209,
+      "eval_samples_per_second": 122.886,
+      "eval_steps_per_second": 0.973,
+      "step": 426
+    },
+    {
+      "epoch": 2.347417840375587,
+      "grad_norm": 6.318545818328857,
+      "learning_rate": 2.3509615384615386e-06,
+      "loss": 1.244,
+      "step": 500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.39050131926121373,
+      "eval_loss": 1.6881287097930908,
+      "eval_runtime": 15.5567,
+      "eval_samples_per_second": 121.813,
+      "eval_steps_per_second": 0.964,
+      "step": 639
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.3883905013192612,
+      "eval_loss": 1.698473334312439,
+      "eval_runtime": 15.7552,
+      "eval_samples_per_second": 120.278,
+      "eval_steps_per_second": 0.952,
+      "step": 852
+    },
+    {
+      "epoch": 4.694835680751174,
+      "grad_norm": 6.993715286254883,
+      "learning_rate": 1.6298076923076921e-06,
+      "loss": 1.1736,
+      "step": 1000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.39472295514511874,
+      "eval_loss": 1.6962207555770874,
+      "eval_runtime": 15.3454,
+      "eval_samples_per_second": 123.49,
+      "eval_steps_per_second": 0.977,
+      "step": 1065
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.3899736147757256,
+      "eval_loss": 1.6851215362548828,
+      "eval_runtime": 15.5276,
+      "eval_samples_per_second": 122.041,
+      "eval_steps_per_second": 0.966,
+      "step": 1278
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.39366754617414246,
+      "eval_loss": 1.6846251487731934,
+      "eval_runtime": 15.4536,
+      "eval_samples_per_second": 122.625,
+      "eval_steps_per_second": 0.971,
+      "step": 1491
+    },
+    {
+      "epoch": 7.042253521126761,
+      "grad_norm": 7.442652225494385,
+      "learning_rate": 9.086538461538461e-07,
+      "loss": 1.1331,
+      "step": 1500
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.3941952506596306,
+      "eval_loss": 1.687286376953125,
+      "eval_runtime": 15.4771,
+      "eval_samples_per_second": 122.439,
+      "eval_steps_per_second": 0.969,
+      "step": 1704
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.38680738786279684,
+      "eval_loss": 1.6869077682495117,
+      "eval_runtime": 15.554,
+      "eval_samples_per_second": 121.834,
+      "eval_steps_per_second": 0.964,
+      "step": 1917
+    },
+    {
+      "epoch": 9.389671361502348,
+      "grad_norm": 11.96031665802002,
+      "learning_rate": 1.875e-07,
+      "loss": 1.1068,
+      "step": 2000
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.39261213720316623,
+      "eval_loss": 1.6845855712890625,
+      "eval_runtime": 15.5194,
+      "eval_samples_per_second": 122.105,
+      "eval_steps_per_second": 0.967,
+      "step": 2130
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 2130,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.0530629594950349e+19,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-2130/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "facial_age_estimator",
+  "_name_or_path": "Robys01/facial_age_estimator",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -66,5 +66,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.41.1"
+  "transformers_version": "4.41.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a2b1ff8101cfd15bbb2eb33d9b78cf3b1f588f3f6733c7fbc4acf4a82b98f3ad
+oid sha256:5c9d1b24ce857f8d1cd71fb076a42521cbf1b357dec4976d513e303348f17d87
 size 343282420

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5e98d6372a61974c4c1acff76262d92430dbfe0d3eca9f2ec10c966c0aa184c
+oid sha256:98df71fabde41428db4dd73f22b05c6db075f3cce18303b5eb362d94896ab363
 size 5048