Upload folder using huggingface_hub

checkpoint-190/config.json

@@ -0,0 +1,48 @@
+{
+  "_name_or_path": "dima806/vehicle_10_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": "SUV",
+    "1": "bus",
+    "2": "family sedan",
+    "3": "fire engine",
+    "4": "heavy truck",
+    "5": "jeep",
+    "6": "minibus",
+    "7": "racing car",
+    "8": "taxi",
+    "9": "truck"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "SUV": 0,
+    "bus": 1,
+    "family sedan": 2,
+    "fire engine": 3,
+    "heavy truck": 4,
+    "jeep": 5,
+    "minibus": 6,
+    "racing car": 7,
+    "taxi": 8,
+    "truck": 9
+  },
+  "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.38.1"
+}

checkpoint-190/model.safetensors

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

checkpoint-190/optimizer.pt

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

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

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

checkpoint-190/scheduler.pt

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

checkpoint-190/trainer_state.json

@@ -0,0 +1,111 @@
+{
+  "best_metric": 1.6207549571990967,
+  "best_model_checkpoint": "vehicle_10_types_image_detection/checkpoint-190",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 190,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.93,
+      "eval_loss": 1.6812069416046143,
+      "eval_runtime": 10.8946,
+      "eval_samples_per_second": 36.715,
+      "eval_steps_per_second": 1.193,
+      "step": 19
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9325,
+      "eval_loss": 1.6739251613616943,
+      "eval_runtime": 10.9614,
+      "eval_samples_per_second": 36.492,
+      "eval_steps_per_second": 1.186,
+      "step": 38
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.935,
+      "eval_loss": 1.6619064807891846,
+      "eval_runtime": 11.065,
+      "eval_samples_per_second": 36.15,
+      "eval_steps_per_second": 1.175,
+      "step": 57
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.94,
+      "eval_loss": 1.6508053541183472,
+      "eval_runtime": 10.8878,
+      "eval_samples_per_second": 36.738,
+      "eval_steps_per_second": 1.194,
+      "step": 76
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.94,
+      "eval_loss": 1.6416151523590088,
+      "eval_runtime": 10.7705,
+      "eval_samples_per_second": 37.139,
+      "eval_steps_per_second": 1.207,
+      "step": 95
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9325,
+      "eval_loss": 1.6340376138687134,
+      "eval_runtime": 10.8454,
+      "eval_samples_per_second": 36.882,
+      "eval_steps_per_second": 1.199,
+      "step": 114
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9325,
+      "eval_loss": 1.6279455423355103,
+      "eval_runtime": 10.9246,
+      "eval_samples_per_second": 36.614,
+      "eval_steps_per_second": 1.19,
+      "step": 133
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.935,
+      "eval_loss": 1.6240490674972534,
+      "eval_runtime": 10.9299,
+      "eval_samples_per_second": 36.597,
+      "eval_steps_per_second": 1.189,
+      "step": 152
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.935,
+      "eval_loss": 1.621616244316101,
+      "eval_runtime": 10.8997,
+      "eval_samples_per_second": 36.698,
+      "eval_steps_per_second": 1.193,
+      "step": 171
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.935,
+      "eval_loss": 1.6207549571990967,
+      "eval_runtime": 10.9979,
+      "eval_samples_per_second": 36.371,
+      "eval_steps_per_second": 1.182,
+      "step": 190
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 190,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "total_flos": 9.29970550849536e+17,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-190/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/vehicle_10_types_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -44,5 +44,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.0"
+  "transformers_version": "4.38.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:972c0ebb2d90a3663c14b2dc35f3dcba5c73e5dcdcf72b291cb5c8219d9cd278
+oid sha256:822c878485c426d70844cb9828e8a7400d67ef9daf90ab9469387c25b33ac0d7
 size 343248584

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c5f0614e4fdaad6820df7757770fc519503676b10e7b93c7e551c574d4c5f1b3
-size 4091
+oid sha256:9fd99a9e041e2b40bb5a0dcc2294432e2baadba56dccdebd8441b69d841fd426
+size 4411