Upload folder using huggingface_hub

checkpoint-1024/config.json

@@ -0,0 +1,126 @@
+{
+  "_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": "A10",
+    "1": "A400M",
+    "2": "AG600",
+    "3": "AV8B",
+    "4": "B1",
+    "5": "B2",
+    "6": "B52",
+    "7": "Be200",
+    "8": "C130",
+    "9": "C17",
+    "10": "C2",
+    "11": "C5",
+    "12": "E2",
+    "13": "E7",
+    "14": "EF2000",
+    "15": "F117",
+    "16": "F14",
+    "17": "F15",
+    "18": "F16",
+    "19": "F18",
+    "20": "F22",
+    "21": "F35",
+    "22": "F4",
+    "23": "H6",
+    "24": "J10",
+    "25": "J20",
+    "26": "JAS39",
+    "27": "JF17",
+    "28": "KC135",
+    "29": "MQ9",
+    "30": "Mig31",
+    "31": "Mirage2000",
+    "32": "P3",
+    "33": "RQ4",
+    "34": "Rafale",
+    "35": "SR71",
+    "36": "Su24",
+    "37": "Su25",
+    "38": "Su34",
+    "39": "Su57",
+    "40": "Tornado",
+    "41": "Tu160",
+    "42": "Tu95",
+    "43": "U2",
+    "44": "US2",
+    "45": "V22",
+    "46": "Vulcan",
+    "47": "XB70",
+    "48": "YF23"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "A10": 0,
+    "A400M": 1,
+    "AG600": 2,
+    "AV8B": 3,
+    "B1": 4,
+    "B2": 5,
+    "B52": 6,
+    "Be200": 7,
+    "C130": 8,
+    "C17": 9,
+    "C2": 10,
+    "C5": 11,
+    "E2": 12,
+    "E7": 13,
+    "EF2000": 14,
+    "F117": 15,
+    "F14": 16,
+    "F15": 17,
+    "F16": 18,
+    "F18": 19,
+    "F22": 20,
+    "F35": 21,
+    "F4": 22,
+    "H6": 23,
+    "J10": 24,
+    "J20": 25,
+    "JAS39": 26,
+    "JF17": 27,
+    "KC135": 28,
+    "MQ9": 29,
+    "Mig31": 30,
+    "Mirage2000": 31,
+    "P3": 32,
+    "RQ4": 33,
+    "Rafale": 34,
+    "SR71": 35,
+    "Su24": 36,
+    "Su25": 37,
+    "Su34": 38,
+    "Su57": 39,
+    "Tornado": 40,
+    "Tu160": 41,
+    "Tu95": 42,
+    "U2": 43,
+    "US2": 44,
+    "V22": 45,
+    "Vulcan": 46,
+    "XB70": 47,
+    "YF23": 48
+  },
+  "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-1024/model.safetensors

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

checkpoint-1024/optimizer.pt

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

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

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

checkpoint-1024/scheduler.pt

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

checkpoint-1024/trainer_state.json

@@ -0,0 +1,44 @@
+{
+  "best_metric": 3.8178980350494385,
+  "best_model_checkpoint": "military_aircraft_image_detection/checkpoint-1024",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 1024,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.48828125,
+      "grad_norm": 0.9782480597496033,
+      "learning_rate": 1.6139630390143737e-06,
+      "loss": 3.8644,
+      "step": 500
+    },
+    {
+      "epoch": 0.9765625,
+      "grad_norm": 1.0809119939804077,
+      "learning_rate": 7.392197125256674e-08,
+      "loss": 3.8062,
+      "step": 1000
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.07585543493197747,
+      "eval_loss": 3.8178980350494385,
+      "eval_runtime": 110.1308,
+      "eval_samples_per_second": 66.076,
+      "eval_steps_per_second": 2.07,
+      "step": 1024
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 1024,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "total_flos": 5.076933153829061e+18,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-1024/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "dima806/military_aircraft_image_detection",
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -32,29 +32,32 @@
     "20": "F22",
     "21": "F35",
     "22": "F4",
-    "23": "J10",
-    "24": "J20",
-    "25": "JAS39",
-    "26": "KC135",
-    "27": "MQ9",
-    "28": "Mig31",
-    "29": "Mirage2000",
-    "30": "P3",
-    "31": "RQ4",
-    "32": "Rafale",
-    "33": "SR71",
-    "34": "Su25",
-    "35": "Su34",
-    "36": "Su57",
-    "37": "Tornado",
-    "38": "Tu160",
-    "39": "Tu95",
-    "40": "U2",
-    "41": "US2",
-    "42": "V22",
-    "43": "Vulcan",
-    "44": "XB70",
-    "45": "YF23"
+    "23": "H6",
+    "24": "J10",
+    "25": "J20",
+    "26": "JAS39",
+    "27": "JF17",
+    "28": "KC135",
+    "29": "MQ9",
+    "30": "Mig31",
+    "31": "Mirage2000",
+    "32": "P3",
+    "33": "RQ4",
+    "34": "Rafale",
+    "35": "SR71",
+    "36": "Su24",
+    "37": "Su25",
+    "38": "Su34",
+    "39": "Su57",
+    "40": "Tornado",
+    "41": "Tu160",
+    "42": "Tu95",
+    "43": "U2",
+    "44": "US2",
+    "45": "V22",
+    "46": "Vulcan",
+    "47": "XB70",
+    "48": "YF23"
   },
   "image_size": 224,
   "initializer_range": 0.02,
@@ -83,29 +86,32 @@
     "F22": 20,
     "F35": 21,
     "F4": 22,
-    "J10": 23,
-    "J20": 24,
-    "JAS39": 25,
-    "KC135": 26,
-    "MQ9": 27,
-    "Mig31": 28,
-    "Mirage2000": 29,
-    "P3": 30,
-    "RQ4": 31,
-    "Rafale": 32,
-    "SR71": 33,
-    "Su25": 34,
-    "Su34": 35,
-    "Su57": 36,
-    "Tornado": 37,
-    "Tu160": 38,
-    "Tu95": 39,
-    "U2": 40,
-    "US2": 41,
-    "V22": 42,
-    "Vulcan": 43,
-    "XB70": 44,
-    "YF23": 45
+    "H6": 23,
+    "J10": 24,
+    "J20": 25,
+    "JAS39": 26,
+    "JF17": 27,
+    "KC135": 28,
+    "MQ9": 29,
+    "Mig31": 30,
+    "Mirage2000": 31,
+    "P3": 32,
+    "RQ4": 33,
+    "Rafale": 34,
+    "SR71": 35,
+    "Su24": 36,
+    "Su25": 37,
+    "Su34": 38,
+    "Su57": 39,
+    "Tornado": 40,
+    "Tu160": 41,
+    "Tu95": 42,
+    "U2": 43,
+    "US2": 44,
+    "V22": 45,
+    "Vulcan": 46,
+    "XB70": 47,
+    "YF23": 48
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -116,5 +122,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.38.2"
+  "transformers_version": "4.40.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:27884d30033f9de85b047f24f8b639c9ab29ca0a528ec3fb735401d0ecc2db37
-size 343359328
+oid sha256:1a62a071b765c50623d57a302bd49cd8f98992efd75de9e3907f77e2b4497993
+size 343368556

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:fc0ace95420731c461a6448ff578b4ad6b10c7b696d3812f6b927e7df9888d65
-size 4411
+oid sha256:23b52bd3299e4e908ee627cdb33a816274baae78414c93c7c1dd4adf24cee529
+size 4475