Upload folder using huggingface_hub

checkpoint-9376/config.json

@@ -0,0 +1,46 @@
+{
+  "_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": "0-2",
+    "1": "3-9",
+    "2": "10-19",
+    "3": "20-29",
+    "4": "30-39",
+    "5": "40-49",
+    "6": "50-59",
+    "7": "60-69",
+    "8": "more than 70"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "0-2": 0,
+    "10-19": 2,
+    "20-29": 3,
+    "3-9": 1,
+    "30-39": 4,
+    "40-49": 5,
+    "50-59": 6,
+    "60-69": 7,
+    "more than 70": 8
+  },
+  "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"
+}

checkpoint-9376/model.safetensors

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

checkpoint-9376/optimizer.pt

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

checkpoint-9376/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": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-9376/rng_state.pth

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

checkpoint-9376/scheduler.pt

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

checkpoint-9376/trainer_state.json

@@ -0,0 +1,199 @@
+{
+  "best_metric": 1.0641573667526245,
+  "best_model_checkpoint": "fairface_age_image_detection/checkpoint-9376",
+  "epoch": 4.0,
+  "eval_steps": 500,
+  "global_step": 9376,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.21331058020477817,
+      "grad_norm": 1.7275338172912598,
+      "learning_rate": 2.855243405532919e-06,
+      "loss": 1.8728,
+      "step": 500
+    },
+    {
+      "epoch": 0.42662116040955633,
+      "grad_norm": 2.228030204772949,
+      "learning_rate": 2.6944027450139397e-06,
+      "loss": 1.4714,
+      "step": 1000
+    },
+    {
+      "epoch": 0.6399317406143344,
+      "grad_norm": 2.2512168884277344,
+      "learning_rate": 2.5335620844949606e-06,
+      "loss": 1.3358,
+      "step": 1500
+    },
+    {
+      "epoch": 0.8532423208191127,
+      "grad_norm": 3.124258279800415,
+      "learning_rate": 2.372721423975981e-06,
+      "loss": 1.264,
+      "step": 2000
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.5423,
+      "eval_loss": 1.2151408195495605,
+      "eval_model_preparation_time": 0.0043,
+      "eval_runtime": 907.0206,
+      "eval_samples_per_second": 11.025,
+      "eval_steps_per_second": 1.378,
+      "step": 2344
+    },
+    {
+      "epoch": 1.0665529010238908,
+      "grad_norm": 5.8204874992370605,
+      "learning_rate": 2.211880763457002e-06,
+      "loss": 1.2174,
+      "step": 2500
+    },
+    {
+      "epoch": 1.2798634812286689,
+      "grad_norm": 5.175824165344238,
+      "learning_rate": 2.051040102938023e-06,
+      "loss": 1.171,
+      "step": 3000
+    },
+    {
+      "epoch": 1.493174061433447,
+      "grad_norm": 3.312744617462158,
+      "learning_rate": 1.8901994424190437e-06,
+      "loss": 1.1425,
+      "step": 3500
+    },
+    {
+      "epoch": 1.7064846416382253,
+      "grad_norm": 4.630620002746582,
+      "learning_rate": 1.7293587819000642e-06,
+      "loss": 1.1267,
+      "step": 4000
+    },
+    {
+      "epoch": 1.9197952218430034,
+      "grad_norm": 3.5600452423095703,
+      "learning_rate": 1.5685181213810853e-06,
+      "loss": 1.1051,
+      "step": 4500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.5688,
+      "eval_loss": 1.1124917268753052,
+      "eval_model_preparation_time": 0.0043,
+      "eval_runtime": 915.0894,
+      "eval_samples_per_second": 10.928,
+      "eval_steps_per_second": 1.366,
+      "step": 4688
+    },
+    {
+      "epoch": 2.1331058020477816,
+      "grad_norm": 7.453662395477295,
+      "learning_rate": 1.407677460862106e-06,
+      "loss": 1.0871,
+      "step": 5000
+    },
+    {
+      "epoch": 2.34641638225256,
+      "grad_norm": 8.157800674438477,
+      "learning_rate": 1.2468368003431269e-06,
+      "loss": 1.0696,
+      "step": 5500
+    },
+    {
+      "epoch": 2.5597269624573378,
+      "grad_norm": 3.6688616275787354,
+      "learning_rate": 1.0859961398241475e-06,
+      "loss": 1.0571,
+      "step": 6000
+    },
+    {
+      "epoch": 2.773037542662116,
+      "grad_norm": 3.1099190711975098,
+      "learning_rate": 9.251554793051683e-07,
+      "loss": 1.0476,
+      "step": 6500
+    },
+    {
+      "epoch": 2.986348122866894,
+      "grad_norm": 5.784822463989258,
+      "learning_rate": 7.643148187861893e-07,
+      "loss": 1.0407,
+      "step": 7000
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.5756,
+      "eval_loss": 1.0761935710906982,
+      "eval_model_preparation_time": 0.0043,
+      "eval_runtime": 920.0161,
+      "eval_samples_per_second": 10.869,
+      "eval_steps_per_second": 1.359,
+      "step": 7032
+    },
+    {
+      "epoch": 3.1996587030716723,
+      "grad_norm": 5.856372833251953,
+      "learning_rate": 6.0347415826721e-07,
+      "loss": 1.0212,
+      "step": 7500
+    },
+    {
+      "epoch": 3.4129692832764507,
+      "grad_norm": 5.725022792816162,
+      "learning_rate": 4.426334977482307e-07,
+      "loss": 1.0253,
+      "step": 8000
+    },
+    {
+      "epoch": 3.6262798634812285,
+      "grad_norm": 4.74147891998291,
+      "learning_rate": 2.8179283722925154e-07,
+      "loss": 1.0232,
+      "step": 8500
+    },
+    {
+      "epoch": 3.839590443686007,
+      "grad_norm": 4.050413131713867,
+      "learning_rate": 1.2095217671027237e-07,
+      "loss": 1.0173,
+      "step": 9000
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.5791,
+      "eval_loss": 1.0641573667526245,
+      "eval_model_preparation_time": 0.0043,
+      "eval_runtime": 919.3696,
+      "eval_samples_per_second": 10.877,
+      "eval_steps_per_second": 1.36,
+      "step": 9376
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 9376,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 4,
+  "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": 2.32490554103808e+19,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-9376/training_args.bin

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1f6f3b28c3de06bcedbae8b985363f2df21de98524d8165e2c6cc7f998d98c57
+oid sha256:ac667f9891e096b94eec6a038cc2967f2cd9d88254165edf42d05ba621ba1c7a
 size 343245508

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28ebf2b6dbb08045128c07625eb89924fd949f0c3794edd35784a5c320305df8
+oid sha256:50f7f93c67bf88403da8f77fa2109e270c1cad6076487c78154dcb74bbd014af
 size 5304