Upload folder using huggingface_hub

checkpoint-1190/config.json

@@ -0,0 +1,44 @@
+{
+  "_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": "pear",
+    "1": "cushion",
+    "2": "princess",
+    "3": "round",
+    "4": "marquise",
+    "5": "oval",
+    "6": "emerald",
+    "7": "heart"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "cushion": 1,
+    "emerald": 6,
+    "heart": 7,
+    "marquise": 4,
+    "oval": 5,
+    "pear": 0,
+    "princess": 2,
+    "round": 3
+  },
+  "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.33.0"
+}

checkpoint-1190/optimizer.pt

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

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

checkpoint-1190/pytorch_model.bin

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

checkpoint-1190/rng_state.pth

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

checkpoint-1190/scheduler.pt

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

checkpoint-1190/trainer_state.json

@@ -0,0 +1,121 @@
+{
+  "best_metric": 0.14204449951648712,
+  "best_model_checkpoint": "diamond_types_image_detection/checkpoint-1190",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 1190,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9556025369978859,
+      "eval_loss": 0.8833887577056885,
+      "eval_runtime": 10.1705,
+      "eval_samples_per_second": 93.014,
+      "eval_steps_per_second": 11.7,
+      "step": 119
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9830866807610994,
+      "eval_loss": 0.40806570649147034,
+      "eval_runtime": 10.2561,
+      "eval_samples_per_second": 92.238,
+      "eval_steps_per_second": 11.603,
+      "step": 238
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9841437632135307,
+      "eval_loss": 0.2761622667312622,
+      "eval_runtime": 10.4086,
+      "eval_samples_per_second": 90.887,
+      "eval_steps_per_second": 11.433,
+      "step": 357
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9957716701902748,
+      "eval_loss": 0.21955327689647675,
+      "eval_runtime": 9.9686,
+      "eval_samples_per_second": 94.898,
+      "eval_steps_per_second": 11.938,
+      "step": 476
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 6.0526315789473685e-06,
+      "loss": 0.6062,
+      "step": 500
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.18926191329956055,
+      "eval_runtime": 10.0236,
+      "eval_samples_per_second": 94.377,
+      "eval_steps_per_second": 11.872,
+      "step": 595
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.16965460777282715,
+      "eval_runtime": 9.9648,
+      "eval_samples_per_second": 94.934,
+      "eval_steps_per_second": 11.942,
+      "step": 714
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.1565340757369995,
+      "eval_runtime": 10.0192,
+      "eval_samples_per_second": 94.418,
+      "eval_steps_per_second": 11.877,
+      "step": 833
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.14831364154815674,
+      "eval_runtime": 10.0259,
+      "eval_samples_per_second": 94.356,
+      "eval_steps_per_second": 11.869,
+      "step": 952
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 1.6666666666666667e-06,
+      "loss": 0.1525,
+      "step": 1000
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.14375551044940948,
+      "eval_runtime": 10.0924,
+      "eval_samples_per_second": 93.734,
+      "eval_steps_per_second": 11.791,
+      "step": 1071
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.14204449951648712,
+      "eval_runtime": 10.4032,
+      "eval_samples_per_second": 90.933,
+      "eval_steps_per_second": 11.439,
+      "step": 1190
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 1190,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "total_flos": 2.9316796129448755e+18,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-1190/training_args.bin

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

config.json

@@ -0,0 +1,44 @@
+{
+  "_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": "pear",
+    "1": "cushion",
+    "2": "princess",
+    "3": "round",
+    "4": "marquise",
+    "5": "oval",
+    "6": "emerald",
+    "7": "heart"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "cushion": 1,
+    "emerald": 6,
+    "heart": 7,
+    "marquise": 4,
+    "oval": 5,
+    "pear": 0,
+    "princess": 2,
+    "round": 3
+  },
+  "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.33.0"
+}

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

pytorch_model.bin

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

training_args.bin

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