detorcla commited on
Commit
db8e8a0
·
verified ·
1 Parent(s): 9213542

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +34 -0
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ResNetForImageClassification"
4
+ ],
5
+ "model_type": "resnet",
6
+ "num_labels": 10,
7
+ "image_size": 224,
8
+ "torch_dtype": "float32",
9
+ "id2label": {
10
+ "0": "airplane",
11
+ "1": "automobile",
12
+ "2": "bird",
13
+ "3": "cat",
14
+ "4": "deer",
15
+ "5": "dog",
16
+ "6": "frog",
17
+ "7": "horse",
18
+ "8": "ship",
19
+ "9": "truck"
20
+ },
21
+ "label2id": {
22
+ "airplane": 0,
23
+ "automobile": 1,
24
+ "bird": 2,
25
+ "cat": 3,
26
+ "deer": 4,
27
+ "dog": 5,
28
+ "frog": 6,
29
+ "horse": 7,
30
+ "ship": 8,
31
+ "truck": 9
32
+ },
33
+ "num_channels": 3
34
+ }