emlababia commited on
Commit
cb6a3a8
1 Parent(s): 4253dc6

Add configuration file and model file

Browse files
Files changed (2) hide show
  1. config.json +147 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "custom/dog-breed-classifier",
3
+ "architectures": [
4
+ "ResNetForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "hidden_act": "relu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 2048,
10
+ "id2label": {
11
+ "0": " Afghan Hound",
12
+ "1": " Akita",
13
+ "2": " Alaskan Malamute",
14
+ "3": " Australian Cattle Dog",
15
+ "4": " Australian Shepherd",
16
+ "5": " Basenji",
17
+ "6": " Basset Hound",
18
+ "7": " Beagle",
19
+ "8": " Belgian Malinois",
20
+ "9": " Bernese Mountain Dog",
21
+ "10": " Bichon Frise",
22
+ "11": " Bloodhound",
23
+ "12": " Border Collie",
24
+ "13": " Boston Terrier",
25
+ "14": " Boxer",
26
+ "15": " Brittany",
27
+ "16": " Bull Terrier",
28
+ "17": " Bulldog",
29
+ "18": " Bullmastiff",
30
+ "19": " Cairn Terrier",
31
+ "20": " Cavalier King Charles Spaniel",
32
+ "21": " Chesapeake Bay Retriever",
33
+ "22": " Chihuahua",
34
+ "23": " Chinese Shar-Pei",
35
+ "24": " Collie",
36
+ "25": " Dachshund",
37
+ "26": " Dalmatian",
38
+ "27": " Doberman Pinscher",
39
+ "28": " English Mastiff",
40
+ "29": " English Springer Spaniel",
41
+ "30": " German Shepherd",
42
+ "31": " Golden Retriever",
43
+ "32": " Great Dane",
44
+ "33": " Havanese",
45
+ "34": " Irish Setter",
46
+ "35": " Italian Greyhound",
47
+ "36": " Labrador Retriever",
48
+ "37": " Lhasa Apso",
49
+ "38": " Maltese",
50
+ "39": " Miniature Schnauzer",
51
+ "40": " Newfoundland",
52
+ "41": " Papillon",
53
+ "42": " Pembroke Welsh Corgi",
54
+ "43": " Poodle",
55
+ "44": " Portuguese Water Dog",
56
+ "45": " Rhodesian Ridgeback",
57
+ "46": " Rottweiler",
58
+ "47": " Saint Bernard",
59
+ "48": " Samoyed",
60
+ "49": " Shetland Sheepdog",
61
+ "50": " Shiba Inu",
62
+ "51": " Shih Tzu",
63
+ "52": " Siberian Husky",
64
+ "53": " Staffordshire Bull Terrier",
65
+ "54": " Vizsla",
66
+ "55": " Weimaraner",
67
+ "56": " West Highland White Terrier",
68
+ "57": " Whippet",
69
+ "58": " Wire Fox Terrier",
70
+ "59": " Yorkshire Terrier"
71
+ },
72
+ "image_size": 224,
73
+ "initializer_range": 0.02,
74
+ "intermediate_size": 4096,
75
+ "label2id": {
76
+ " Afghan Hound": "0",
77
+ " Akita": "1",
78
+ " Alaskan Malamute": "2",
79
+ " Australian Cattle Dog": "3",
80
+ " Australian Shepherd": "4",
81
+ " Basenji": "5",
82
+ " Basset Hound": "6",
83
+ " Beagle": "7",
84
+ " Belgian Malinois": "8",
85
+ " Bernese Mountain Dog": "9",
86
+ " Bichon Frise": "10",
87
+ " Bloodhound": "11",
88
+ " Border Collie": "12",
89
+ " Boston Terrier": "13",
90
+ " Boxer": "14",
91
+ " Brittany": "15",
92
+ " Bull Terrier": "16",
93
+ " Bulldog": "17",
94
+ " Bullmastiff": "18",
95
+ " Cairn Terrier": "19",
96
+ " Cavalier King Charles Spaniel": "20",
97
+ " Chesapeake Bay Retriever": "21",
98
+ " Chihuahua": "22",
99
+ " Chinese Shar-Pei": "23",
100
+ " Collie": "24",
101
+ " Dachshund": "25",
102
+ " Dalmatian": "26",
103
+ " Doberman Pinscher": "27",
104
+ " English Mastiff": "28",
105
+ " English Springer Spaniel": "29",
106
+ " German Shepherd": "30",
107
+ " Golden Retriever": "31",
108
+ " Great Dane": "32",
109
+ " Havanese": "33",
110
+ " Irish Setter": "34",
111
+ " Italian Greyhound": "35",
112
+ " Labrador Retriever": "36",
113
+ " Lhasa Apso": "37",
114
+ " Maltese": "38",
115
+ " Miniature Schnauzer": "39",
116
+ " Newfoundland": "40",
117
+ " Papillon": "41",
118
+ " Pembroke Welsh Corgi": "42",
119
+ " Poodle": "43",
120
+ " Portuguese Water Dog": "44",
121
+ " Rhodesian Ridgeback": "45",
122
+ " Rottweiler": "46",
123
+ " Saint Bernard": "47",
124
+ " Samoyed": "48",
125
+ " Shetland Sheepdog": "49",
126
+ " Shiba Inu": "50",
127
+ " Shih Tzu": "51",
128
+ " Siberian Husky": "52",
129
+ " Staffordshire Bull Terrier": "53",
130
+ " Vizsla": "54",
131
+ " Weimaraner": "55",
132
+ " West Highland White Terrier": "56",
133
+ " Whippet": "57",
134
+ " Wire Fox Terrier": "58",
135
+ " Yorkshire Terrier": "59"
136
+ },
137
+ "layer_norm_eps": 1e-12,
138
+ "model_type": "resnet",
139
+ "num_attention_heads": 32,
140
+ "num_channels": 3,
141
+ "num_hidden_layers": 50,
142
+ "patch_size": 16,
143
+ "problem_type": "single_label_classification",
144
+ "qkv_bias": true,
145
+ "torch_dtype": "float32",
146
+ "transformers_version": "4.38.2"
147
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f1c0d64216216b0d95ae3cf755dd55193f2e9072e5aeefa3985a29f61d9bd94
3
+ size 44906673