checkpoint-2154/config.json

{
  "_name_or_path": "dima806/military_aircraft_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": "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": "Tu22M",
    "43": "Tu95",
    "44": "U2",
    "45": "US2",
    "46": "V22",
    "47": "Vulcan",
    "48": "XB70",
    "49": "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,
    "Tu22M": 42,
    "Tu95": 43,
    "U2": 44,
    "US2": 45,
    "V22": 46,
    "Vulcan": 47,
    "XB70": 48,
    "YF23": 49
  },
  "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.41.2"
}