asusevski commited on
Commit
e77f5b1
1 Parent(s): d34d99c

Upload ViTForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +101 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Abyssinian",
13
+ "1": "Bengal",
14
+ "10": "Siamese",
15
+ "11": "Sphynx",
16
+ "12": "american bulldog",
17
+ "13": "american pit bull terrier",
18
+ "14": "basset hound",
19
+ "15": "beagle",
20
+ "16": "boxer",
21
+ "17": "chihuahua",
22
+ "18": "english cocker spaniel",
23
+ "19": "english setter",
24
+ "2": "Birman",
25
+ "20": "german shorthaired",
26
+ "21": "great pyrenees",
27
+ "22": "havanese",
28
+ "23": "japanese chin",
29
+ "24": "keeshond",
30
+ "25": "leonberger",
31
+ "26": "miniature pinscher",
32
+ "27": "newfoundland",
33
+ "28": "pomeranian",
34
+ "29": "pug",
35
+ "3": "Bombay",
36
+ "30": "saint bernard",
37
+ "31": "samoyed",
38
+ "32": "scottish terrier",
39
+ "33": "shiba inu",
40
+ "34": "staffordshire bull terrier",
41
+ "35": "wheaten terrier",
42
+ "36": "yorkshire terrier",
43
+ "4": "British Shorthair",
44
+ "5": "Egyptian Mau",
45
+ "6": "Maine Coon",
46
+ "7": "Persian",
47
+ "8": "Ragdoll",
48
+ "9": "Russian Blue"
49
+ },
50
+ "image_size": 224,
51
+ "initializer_range": 0.02,
52
+ "intermediate_size": 3072,
53
+ "label2id": {
54
+ "Abyssinian": "0",
55
+ "Bengal": "1",
56
+ "Birman": "2",
57
+ "Bombay": "3",
58
+ "British Shorthair": "4",
59
+ "Egyptian Mau": "5",
60
+ "Maine Coon": "6",
61
+ "Persian": "7",
62
+ "Ragdoll": "8",
63
+ "Russian Blue": "9",
64
+ "Siamese": "10",
65
+ "Sphynx": "11",
66
+ "american bulldog": "12",
67
+ "american pit bull terrier": "13",
68
+ "basset hound": "14",
69
+ "beagle": "15",
70
+ "boxer": "16",
71
+ "chihuahua": "17",
72
+ "english cocker spaniel": "18",
73
+ "english setter": "19",
74
+ "german shorthaired": "20",
75
+ "great pyrenees": "21",
76
+ "havanese": "22",
77
+ "japanese chin": "23",
78
+ "keeshond": "24",
79
+ "leonberger": "25",
80
+ "miniature pinscher": "26",
81
+ "newfoundland": "27",
82
+ "pomeranian": "28",
83
+ "pug": "29",
84
+ "saint bernard": "30",
85
+ "samoyed": "31",
86
+ "scottish terrier": "32",
87
+ "shiba inu": "33",
88
+ "staffordshire bull terrier": "34",
89
+ "wheaten terrier": "35",
90
+ "yorkshire terrier": "36"
91
+ },
92
+ "layer_norm_eps": 1e-12,
93
+ "model_type": "vit",
94
+ "num_attention_heads": 12,
95
+ "num_channels": 3,
96
+ "num_hidden_layers": 12,
97
+ "patch_size": 16,
98
+ "qkv_bias": true,
99
+ "torch_dtype": "float32",
100
+ "transformers_version": "4.35.2"
101
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56108205279624a3c584a33108760cd47998b12f6df698e2ed916b06e5a15689
3
+ size 343331644