Upload folder using huggingface_hub

checkpoint-128/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/lemon_quality_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -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",
@@ -30,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.33.3"
+  "transformers_version": "4.39.2"
 }

checkpoint-128/model.safetensors

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

checkpoint-128/optimizer.pt

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

checkpoint-128/preprocessor_config.json

@@ -1,4 +1,18 @@
 {
+  "_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,

checkpoint-128/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c33047c27d8e9143c9dcf947d37b78ef88270f94db0df0e2fbdd5921c35f2d93
+oid sha256:9120c7be6dc7a83a29fbe261cc51f4ec3c951303d1c0e44912c1dfec99016f42
 size 14575

checkpoint-128/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5d5ed4e4e1b7febcb3a328700315229b1a315093c0872ab2c59a9e3fb7370a10
+oid sha256:0635e963639eec33518717a9f8e4bab758f1b7e0b7ee1a261608b542bfd2a157
 size 627

checkpoint-128/trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.3760019540786743,
+  "best_metric": 0.05322761833667755,
   "best_model_checkpoint": "lemon_quality_image_detection/checkpoint-128",
   "epoch": 2.0,
   "eval_steps": 500,
@@ -10,28 +10,30 @@
   "log_history": [
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.9209486166007905,
-      "eval_loss": 0.6092899441719055,
-      "eval_runtime": 5.2948,
-      "eval_samples_per_second": 95.566,
-      "eval_steps_per_second": 12.087,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.054433587938547134,
+      "eval_runtime": 5.0385,
+      "eval_samples_per_second": 100.427,
+      "eval_steps_per_second": 12.702,
       "step": 64
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.9723320158102767,
-      "eval_loss": 0.3760019540786743,
-      "eval_runtime": 4.9359,
-      "eval_samples_per_second": 102.515,
-      "eval_steps_per_second": 12.966,
+      "eval_accuracy": 1.0,
+      "eval_loss": 0.05322761833667755,
+      "eval_runtime": 5.1226,
+      "eval_samples_per_second": 98.779,
+      "eval_steps_per_second": 12.494,
       "step": 128
     }
   ],
   "logging_steps": 500,
   "max_steps": 128,
+  "num_input_tokens_seen": 0,
   "num_train_epochs": 2,
   "save_steps": 500,
   "total_flos": 3.133804147057705e+17,
+  "train_batch_size": 32,
   "trial_name": null,
   "trial_params": null
 }

checkpoint-128/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/lemon_quality_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -30,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.33.3"
+  "transformers_version": "4.39.2"
 }

model.safetensors

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

preprocessor_config.json

@@ -1,4 +1,18 @@
 {
+  "_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,

training_args.bin

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