LH-Tech-AI commited on
Commit
a00bf00
·
verified ·
1 Parent(s): 164d0d2

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +51 -0
config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ResNetForImageClassification"
4
+ ],
5
+ "depths": [
6
+ 2,
7
+ 2,
8
+ 2,
9
+ 2
10
+ ],
11
+ "downsample_in_bottleneck": false,
12
+ "downsample_in_first_stage": false,
13
+ "dtype": "float32",
14
+ "embedding_size": 64,
15
+ "hidden_act": "relu",
16
+ "hidden_sizes": [
17
+ 64,
18
+ 128,
19
+ 256,
20
+ 512
21
+ ],
22
+ "id2label": {
23
+ "0": "LABEL_0",
24
+ "1": "LABEL_1",
25
+ "2": "LABEL_2",
26
+ "3": "LABEL_3"
27
+ },
28
+ "label2id": {
29
+ "LABEL_0": 0,
30
+ "LABEL_1": 1,
31
+ "LABEL_2": 2,
32
+ "LABEL_3": 3
33
+ },
34
+ "layer_type": "basic",
35
+ "model_type": "resnet",
36
+ "num_channels": 3,
37
+ "out_features": [
38
+ "stage4"
39
+ ],
40
+ "out_indices": [
41
+ 4
42
+ ],
43
+ "stage_names": [
44
+ "stem",
45
+ "stage1",
46
+ "stage2",
47
+ "stage3",
48
+ "stage4"
49
+ ],
50
+ "transformers_version": "5.0.0"
51
+ }