Upload folder using huggingface_hub

checkpoint-640/config.json

@@ -0,0 +1,34 @@
+{
+  "_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": "good_quality",
+    "1": "empty_background",
+    "2": "bad_quality"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "bad_quality": 2,
+    "empty_background": 1,
+    "good_quality": 0
+  },
+  "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.3"
+}

checkpoint-640/optimizer.pt

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

checkpoint-640/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-640/pytorch_model.bin

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

checkpoint-640/rng_state.pth

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

checkpoint-640/scheduler.pt

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

checkpoint-640/trainer_state.json

@@ -0,0 +1,115 @@
+{
+  "best_metric": 0.055376775562763214,
+  "best_model_checkpoint": "lemon_quality_image_detection/checkpoint-640",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 640,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9426877470355731,
+      "eval_loss": 0.5755197405815125,
+      "eval_runtime": 5.1227,
+      "eval_samples_per_second": 98.776,
+      "eval_steps_per_second": 12.493,
+      "step": 64
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9881422924901185,
+      "eval_loss": 0.20843671262264252,
+      "eval_runtime": 5.1165,
+      "eval_samples_per_second": 98.896,
+      "eval_steps_per_second": 12.509,
+      "step": 128
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.1252429187297821,
+      "eval_runtime": 5.0129,
+      "eval_samples_per_second": 100.94,
+      "eval_steps_per_second": 12.767,
+      "step": 192
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.09138158708810806,
+      "eval_runtime": 5.0548,
+      "eval_samples_per_second": 100.102,
+      "eval_steps_per_second": 12.661,
+      "step": 256
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.07500764727592468,
+      "eval_runtime": 5.0408,
+      "eval_samples_per_second": 100.382,
+      "eval_steps_per_second": 12.697,
+      "step": 320
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.06633897125720978,
+      "eval_runtime": 5.0106,
+      "eval_samples_per_second": 100.986,
+      "eval_steps_per_second": 12.773,
+      "step": 384
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.06040707230567932,
+      "eval_runtime": 5.0324,
+      "eval_samples_per_second": 100.548,
+      "eval_steps_per_second": 12.718,
+      "step": 448
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 2.372881355932204e-06,
+      "loss": 0.1834,
+      "step": 500
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.057528626173734665,
+      "eval_runtime": 4.9848,
+      "eval_samples_per_second": 101.508,
+      "eval_steps_per_second": 12.839,
+      "step": 512
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.05608272925019264,
+      "eval_runtime": 5.1403,
+      "eval_samples_per_second": 98.439,
+      "eval_steps_per_second": 12.451,
+      "step": 576
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.055376775562763214,
+      "eval_runtime": 5.0293,
+      "eval_samples_per_second": 100.61,
+      "eval_steps_per_second": 12.725,
+      "step": 640
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 640,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "total_flos": 1.5669020735288525e+18,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-640/training_args.bin

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

config.json

@@ -9,17 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "bad_quality",
-    "1": "good_quality",
-    "2": "empty_background"
+    "0": "good_quality",
+    "1": "empty_background",
+    "2": "bad_quality"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "bad_quality": 0,
-    "empty_background": 2,
-    "good_quality": 1
+    "bad_quality": 2,
+    "empty_background": 1,
+    "good_quality": 0
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ab5475aaa36f798fe2ba0b8b0ec41643f82c79fb2feda7319f2b43dbfef7ea38
+oid sha256:bbcabadaeed9ece55468a2d8f111cd80e8e4c587b5c4965d70350b14e9f0aab2
 size 343271789

training_args.bin

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