vit-base-pets / config.json
OmAlve's picture
added preprocessor
f7edb1e verified
raw
history blame
2.83 kB
{
"_name_or_path": "google/vit-base-patch16-224",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "pug Dog",
"1": "japanese chin Dog",
"10": "Ragdoll Cat",
"11": "scottish terrier Dog",
"12": "shiba inu Dog",
"13": "chihuahua Dog",
"14": "samoyed Dog",
"15": "Maine Coon Cat",
"16": "newfoundland Dog",
"17": "Abyssinian Cat",
"18": "saint bernard Dog",
"19": "Persian Cat",
"2": "american pit bull terrier Dog",
"20": "american bulldog Dog",
"21": "boxer Dog",
"22": "leonberger Dog",
"23": "german shorthaired Dog",
"24": "staffordshire bull terrier Dog",
"25": "Birman Cat",
"26": "english cocker spaniel Dog",
"27": "english setter Dog",
"28": "Siamese Cat",
"29": "Sphynx Cat",
"3": "beagle Dog",
"30": "Bombay Cat",
"31": "keeshond Dog",
"32": "havanese Dog",
"33": "Bengal Cat",
"34": "great pyrenees Dog",
"35": "Russian Blue Cat",
"36": "basset hound Dog",
"4": "miniature pinscher Dog",
"5": "pomeranian Dog",
"6": "yorkshire terrier Dog",
"7": "Egyptian Mau Cat",
"8": "British Shorthair Cat",
"9": "wheaten terrier Dog"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Abyssinian Cat": "17",
"Bengal Cat": "33",
"Birman Cat": "25",
"Bombay Cat": "30",
"British Shorthair Cat": "8",
"Egyptian Mau Cat": "7",
"Maine Coon Cat": "15",
"Persian Cat": "19",
"Ragdoll Cat": "10",
"Russian Blue Cat": "35",
"Siamese Cat": "28",
"Sphynx Cat": "29",
"american bulldog Dog": "20",
"american pit bull terrier Dog": "2",
"basset hound Dog": "36",
"beagle Dog": "3",
"boxer Dog": "21",
"chihuahua Dog": "13",
"english cocker spaniel Dog": "26",
"english setter Dog": "27",
"german shorthaired Dog": "23",
"great pyrenees Dog": "34",
"havanese Dog": "32",
"japanese chin Dog": "1",
"keeshond Dog": "31",
"leonberger Dog": "22",
"miniature pinscher Dog": "4",
"newfoundland Dog": "16",
"pomeranian Dog": "5",
"pug Dog": "0",
"saint bernard Dog": "18",
"samoyed Dog": "14",
"scottish terrier Dog": "11",
"shiba inu Dog": "12",
"staffordshire bull terrier Dog": "24",
"wheaten terrier Dog": "9",
"yorkshire terrier Dog": "6"
},
"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.39.2"
}