Upload folder using huggingface_hub

checkpoint-4688/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-4688/model.safetensors

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

checkpoint-4688/optimizer.pt

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

checkpoint-4688/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-4688/rng_state.pth

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

checkpoint-4688/scheduler.pt

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

checkpoint-4688/trainer_state.json

@@ -0,0 +1,116 @@
+{
+  "best_metric": 1.233594298362732,
+  "best_model_checkpoint": "fairface_age_image_detection/checkpoint-4688",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 4688,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.21331058020477817,
+      "grad_norm": 1.7164175510406494,
+      "learning_rate": 2.2574385510996123e-06,
+      "loss": 1.945,
+      "step": 500
+    },
+    {
+      "epoch": 0.42662116040955633,
+      "grad_norm": 1.979918360710144,
+      "learning_rate": 1.9879258300991807e-06,
+      "loss": 1.5803,
+      "step": 1000
+    },
+    {
+      "epoch": 0.6399317406143344,
+      "grad_norm": 2.0418295860290527,
+      "learning_rate": 1.7184131090987496e-06,
+      "loss": 1.4368,
+      "step": 1500
+    },
+    {
+      "epoch": 0.8532423208191127,
+      "grad_norm": 2.872443199157715,
+      "learning_rate": 1.4489003880983182e-06,
+      "loss": 1.3531,
+      "step": 2000
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.5275,
+      "eval_loss": 1.3043227195739746,
+      "eval_model_preparation_time": 0.0035,
+      "eval_runtime": 894.9363,
+      "eval_samples_per_second": 11.174,
+      "eval_steps_per_second": 1.397,
+      "step": 2344
+    },
+    {
+      "epoch": 1.0665529010238908,
+      "grad_norm": 4.807504177093506,
+      "learning_rate": 1.179387667097887e-06,
+      "loss": 1.3097,
+      "step": 2500
+    },
+    {
+      "epoch": 1.2798634812286689,
+      "grad_norm": 6.204404830932617,
+      "learning_rate": 9.098749460974558e-07,
+      "loss": 1.2671,
+      "step": 3000
+    },
+    {
+      "epoch": 1.493174061433447,
+      "grad_norm": 6.795459270477295,
+      "learning_rate": 6.403622250970247e-07,
+      "loss": 1.2428,
+      "step": 3500
+    },
+    {
+      "epoch": 1.7064846416382253,
+      "grad_norm": 3.4942402839660645,
+      "learning_rate": 3.708495040965934e-07,
+      "loss": 1.2307,
+      "step": 4000
+    },
+    {
+      "epoch": 1.9197952218430034,
+      "grad_norm": 3.384183406829834,
+      "learning_rate": 1.0133678309616215e-07,
+      "loss": 1.2214,
+      "step": 4500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.5385,
+      "eval_loss": 1.233594298362732,
+      "eval_model_preparation_time": 0.0035,
+      "eval_runtime": 903.6391,
+      "eval_samples_per_second": 11.066,
+      "eval_steps_per_second": 1.383,
+      "step": 4688
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 4688,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "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.16245277051904e+19,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-4688/training_args.bin

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

model.safetensors

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