raffaelsiregar's picture
Update config.json
8591ec3 verified
raw
history blame
3.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': 'Afghan',
'1': 'African Wild Dog',
'2': 'Airedale',
'3': 'American Hairless',
'4': 'American Spaniel',
'5': 'Basenji',
'6': 'Basset',
'7': 'Beagle',
'8': 'Bearded Collie',
'9': 'Bermaise',
'10': 'Bichon Frise',
'11': 'Blenheim',
'12': 'Bloodhound',
'13': 'Bluetick',
'14': 'Border Collie',
'15': 'Borzoi',
'16': 'Boston Terrier',
'17': 'Boxer',
'18': 'Bull Mastiff',
'19': 'Bull Terrier',
'20': 'Bulldog',
'21': 'Cairn',
'22': 'Chihuahua',
'23': 'Chinese Crested',
'24': 'Chow',
'25': 'Clumber',
'26': 'Cockapoo',
'27': 'Cocker',
'28': 'Collie',
'29': 'Corgi',
'30': 'Coyote',
'31': 'Dalmation',
'32': 'Dhole',
'33': 'Dingo',
'34': 'Doberman',
'35': 'Elk Hound',
'36': 'French Bulldog',
'37': 'German Sheperd',
'38': 'Golden Retriever',
'39': 'Great Dane',
'40': 'Great Perenees',
'41': 'Greyhound',
'42': 'Groenendael',
'43': 'Irish Spaniel',
'44': 'Irish Wolfhound',
'45': 'Japanese Spaniel',
'46': 'Komondor',
'47': 'Labradoodle',
'48': 'Labrador',
'49': 'Lhasa',
'50': 'Malinois',
'51': 'Maltese',
'52': 'Mex Hairless',
'53': 'Newfoundland',
'54': 'Pekinese',
'55': 'Pit Bull',
'56': 'Pomeranian',
'57': 'Poodle',
'58': 'Pug',
'59': 'Rhodesian',
'60': 'Rottweiler',
'61': 'Saint Bernard',
'62': 'Schnauzer',
'63': 'Scotch Terrier',
'64': 'Shar_Pei',
'65': 'Shiba Inu',
'66': 'Shih-Tzu',
'67': 'Siberian Husky',
'68': 'Vizsla',
'69': 'Yorkie',
'70': 'American Spaniel'
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
'Afghan': 0,
'African Wild Dog': 1,
'Airedale': 2,
'American Hairless': 3,
'American Spaniel': 4,
'Basenji': 5,
'Basset': 6,
'Beagle': 7,
'Bearded Collie': 8,
'Bermaise': 9,
'Bichon Frise': 10,
'Blenheim': 11,
'Bloodhound': 12,
'Bluetick': 13,
'Border Collie': 14,
'Borzoi': 15,
'Boston Terrier': 16,
'Boxer': 17,
'Bull Mastiff': 18,
'Bull Terrier': 19,
'Bulldog': 20,
'Cairn': 21,
'Chihuahua': 22,
'Chinese Crested': 23,
'Chow': 24,
'Clumber': 25,
'Cockapoo': 26,
'Cocker': 27,
'Collie': 28,
'Corgi': 29,
'Coyote': 30,
'Dalmation': 31,
'Dhole': 32,
'Dingo': 33,
'Doberman': 34,
'Elk Hound': 35,
'French Bulldog': 36,
'German Sheperd': 37,
'Golden Retriever': 38,
'Great Dane': 39,
'Great Perenees': 40,
'Greyhound': 41,
'Groenendael': 42,
'Irish Spaniel': 43,
'Irish Wolfhound': 44,
'Japanese Spaniel': 45,
'Komondor': 46,
'Labradoodle': 47,
'Labrador': 48,
'Lhasa': 49,
'Malinois': 50,
'Maltese': 51,
'Mex Hairless': 52,
'Newfoundland': 53,
'Pekinese': 54,
'Pit Bull': 55,
'Pomeranian': 56,
'Poodle': 57,
'Pug': 58,
'Rhodesian': 59,
'Rottweiler': 60,
'Saint Bernard': 61,
'Schnauzer': 62,
'Scotch Terrier': 63,
'Shar_Pei': 64,
'Shiba Inu': 65,
'Shih-Tzu': 66,
'Siberian Husky': 67,
'Vizsla': 68,
'Yorkie': 69,
'American Spaniel': 70
},
"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.44.2"
}