Update config.json
Browse files- config.json +56 -0
config.json
CHANGED
@@ -2,5 +2,61 @@
|
|
2 |
"architectures": [
|
3 |
"ResNetForImageClassification"
|
4 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
"num_classes": 10
|
6 |
}
|
|
|
2 |
"architectures": [
|
3 |
"ResNetForImageClassification"
|
4 |
],
|
5 |
+
"id2label": {
|
6 |
+
"0": "tench, Tinca tinca",
|
7 |
+
"1": "goldfish, Carassius auratus",
|
8 |
+
"2": "great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias",
|
9 |
+
"3": "tiger shark, Galeocerdo cuvieri",
|
10 |
+
"4": "hammerhead, hammerhead shark",
|
11 |
+
"5": "electric ray, crampfish, numbfish, torpedo",
|
12 |
+
"6": "stingray",
|
13 |
+
"7": "cock",
|
14 |
+
"8": "hen",
|
15 |
+
"9": "ostrich, Struthio camelus",
|
16 |
+
"10": "brambling, Fringilla montifringilla",
|
17 |
+
"11": "goldfinch, Carduelis carduelis",
|
18 |
+
"12": "house finch, linnet, Carpodacus mexicanus",
|
19 |
+
"13": "junco, snowbird",
|
20 |
+
"14": "indigo bunting, indigo finch, indigo bird, Passerina cyanea",
|
21 |
+
"15": "robin, American robin, Turdus migratorius",
|
22 |
+
"16": "bulbul",
|
23 |
+
"17": "jay",
|
24 |
+
"18": "magpie",
|
25 |
+
"19": "chickadee",
|
26 |
+
"20": "water ouzel, dipper",
|
27 |
+
"21": "kite",
|
28 |
+
"22": "bald eagle, American eagle, Haliaeetus leucocephalus",
|
29 |
+
"23": "vulture",
|
30 |
+
"24": "great grey owl, great gray owl, Strix nebulosa",
|
31 |
+
"25": "European fire salamander, Salamandra salamandra",
|
32 |
+
"26": "common newt, Triturus vulgaris",
|
33 |
+
"27": "eft",
|
34 |
+
"28": "spotted salamander, Ambystoma maculatum",
|
35 |
+
"29": "axolotl, mud puppy, Ambystoma mexicanum",
|
36 |
+
"30": "bullfrog, Rana catesbeiana",
|
37 |
+
"31": "tree frog, tree-frog",
|
38 |
+
"32": "tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui",
|
39 |
+
"33": "loggerhead, loggerhead turtle, Caretta caretta",
|
40 |
+
"34": "leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea",
|
41 |
+
"35": "mud turtle",
|
42 |
+
"36": "terrapin",
|
43 |
+
"37": "box turtle, box tortoise",
|
44 |
+
"38": "banded gecko",
|
45 |
+
"39": "common iguana, iguana, Iguana iguana",
|
46 |
+
"40": "American chameleon, anole, Anolis carolinensis",
|
47 |
+
"41": "whiptail, whiptail lizard",
|
48 |
+
"42": "agama",
|
49 |
+
"43": "frilled lizard, Chlamydosaurus kingi",
|
50 |
+
"44": "alligator lizard",
|
51 |
+
"45": "Gila monster, Heloderma suspectum",
|
52 |
+
"46": "green lizard, Lacerta viridis",
|
53 |
+
"47": "African chameleon, Chamaeleo chamaeleon",
|
54 |
+
"48": "Komodo dragon, Komodo lizard, dragon lizard, giant lizard, Varanus komodoensis",
|
55 |
+
"49": "African crocodile, Nile crocodile, Crocodylus niloticus",
|
56 |
+
"50": "American alligator, Alligator mississipiensis"
|
57 |
+
},
|
58 |
+
"layer_type": "basic",
|
59 |
+
"model_type": "resnet",
|
60 |
+
"torch_dtype": "float32",
|
61 |
"num_classes": 10
|
62 |
}
|