Upload folder using huggingface_hub

checkpoint-180/config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "dima806/skin_types_image_detection",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "dry",
+    "1": "normal",
+    "2": "oily"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "dry": 0,
+    "normal": 1,
+    "oily": 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.40.1"
+}

checkpoint-180/model.safetensors

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

checkpoint-180/optimizer.pt

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

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

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

checkpoint-180/scheduler.pt

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

checkpoint-180/trainer_state.json

@@ -0,0 +1,66 @@
+{
+  "best_metric": 0.9216932654380798,
+  "best_model_checkpoint": "skin_types_image_detection/checkpoint-180",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 180,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.6500981033355134,
+      "eval_loss": 0.9241806864738464,
+      "eval_runtime": 19.6412,
+      "eval_samples_per_second": 77.847,
+      "eval_steps_per_second": 2.444,
+      "step": 36
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.6507521255722695,
+      "eval_loss": 0.9233015179634094,
+      "eval_runtime": 19.4603,
+      "eval_samples_per_second": 78.57,
+      "eval_steps_per_second": 2.467,
+      "step": 72
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.6546762589928058,
+      "eval_loss": 0.9224075078964233,
+      "eval_runtime": 19.6334,
+      "eval_samples_per_second": 77.878,
+      "eval_steps_per_second": 2.445,
+      "step": 108
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.6533682145192936,
+      "eval_loss": 0.9218368530273438,
+      "eval_runtime": 19.439,
+      "eval_samples_per_second": 78.656,
+      "eval_steps_per_second": 2.469,
+      "step": 144
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.6533682145192936,
+      "eval_loss": 0.9216932654380798,
+      "eval_runtime": 19.7524,
+      "eval_samples_per_second": 77.408,
+      "eval_steps_per_second": 2.43,
+      "step": 180
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 180,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 8.884536237887386e+17,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-180/training_args.bin

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

config.json

@@ -30,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.3"
+  "transformers_version": "4.40.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:83ce1a71aa0ca417c480bd0936102a0ff117ee9bf38a31979b1474a2a43f8376
+oid sha256:572185743c6be523efcc72b71f12c089cb81baf54525a1831b333544ba47224a
 size 343227052

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9956e965656e23e62ab9efcf647d3f321fe84137baf350162bf67e10f66199bf
-size 4856
+oid sha256:feda3225a79e649d66ee4670b47cd0d12ba7eff7961c499a34768a1c99ec915b
+size 4920