msluszniak commited on
Commit
1dba9a8
·
verified ·
1 Parent(s): f395b80

Add backend config (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. coreml/config.json +210 -0
coreml/config.json ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "deeplab_v3",
4
+ "family": "deeplab",
5
+ "capabilities": [
6
+ "semantic-segmentation"
7
+ ],
8
+ "backend": "coreml",
9
+ "license": "bsd-3-clause",
10
+ "variants": [
11
+ {
12
+ "file": "deeplab_v3_mobilenet_v3_large_coreml_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": true,
16
+ "size": "mobilenet_v3_large",
17
+ "methods": {
18
+ "forward": {
19
+ "inputs": [
20
+ {
21
+ "shape": [
22
+ 1,
23
+ 3,
24
+ 520,
25
+ 520
26
+ ],
27
+ "dtype": "float32"
28
+ }
29
+ ],
30
+ "outputs": [
31
+ {
32
+ "shape": [
33
+ 1,
34
+ 21,
35
+ 520,
36
+ 520
37
+ ],
38
+ "dtype": "float32"
39
+ }
40
+ ]
41
+ }
42
+ }
43
+ },
44
+ {
45
+ "file": "deeplab_v3_resnet50_coreml_fp16.pte",
46
+ "precision": "fp16",
47
+ "quantized": false,
48
+ "default": false,
49
+ "size": "resnet50",
50
+ "methods": {
51
+ "forward": {
52
+ "inputs": [
53
+ {
54
+ "shape": [
55
+ 1,
56
+ 3,
57
+ 520,
58
+ 520
59
+ ],
60
+ "dtype": "float32"
61
+ }
62
+ ],
63
+ "outputs": [
64
+ {
65
+ "shape": [
66
+ 1,
67
+ 21,
68
+ 520,
69
+ 520
70
+ ],
71
+ "dtype": "float32"
72
+ }
73
+ ]
74
+ }
75
+ }
76
+ },
77
+ {
78
+ "file": "deeplab_v3_resnet101_coreml_fp16.pte",
79
+ "precision": "fp16",
80
+ "quantized": false,
81
+ "default": false,
82
+ "size": "resnet101",
83
+ "methods": {
84
+ "forward": {
85
+ "inputs": [
86
+ {
87
+ "shape": [
88
+ 1,
89
+ 3,
90
+ 520,
91
+ 520
92
+ ],
93
+ "dtype": "float32"
94
+ }
95
+ ],
96
+ "outputs": [
97
+ {
98
+ "shape": [
99
+ 1,
100
+ 21,
101
+ 520,
102
+ 520
103
+ ],
104
+ "dtype": "float32"
105
+ }
106
+ ]
107
+ }
108
+ }
109
+ },
110
+ {
111
+ "file": "deeplab_v3_mobilenet_v3_large_coreml_fp32.pte",
112
+ "precision": "fp32",
113
+ "quantized": false,
114
+ "default": false,
115
+ "size": "mobilenet_v3_large",
116
+ "methods": {
117
+ "forward": {
118
+ "inputs": [
119
+ {
120
+ "shape": [
121
+ 1,
122
+ 3,
123
+ 520,
124
+ 520
125
+ ],
126
+ "dtype": "float32"
127
+ }
128
+ ],
129
+ "outputs": [
130
+ {
131
+ "shape": [
132
+ 1,
133
+ 21,
134
+ 520,
135
+ 520
136
+ ],
137
+ "dtype": "float32"
138
+ }
139
+ ]
140
+ }
141
+ }
142
+ },
143
+ {
144
+ "file": "deeplab_v3_resnet50_coreml_fp32.pte",
145
+ "precision": "fp32",
146
+ "quantized": false,
147
+ "default": false,
148
+ "size": "resnet50",
149
+ "methods": {
150
+ "forward": {
151
+ "inputs": [
152
+ {
153
+ "shape": [
154
+ 1,
155
+ 3,
156
+ 520,
157
+ 520
158
+ ],
159
+ "dtype": "float32"
160
+ }
161
+ ],
162
+ "outputs": [
163
+ {
164
+ "shape": [
165
+ 1,
166
+ 21,
167
+ 520,
168
+ 520
169
+ ],
170
+ "dtype": "float32"
171
+ }
172
+ ]
173
+ }
174
+ }
175
+ },
176
+ {
177
+ "file": "deeplab_v3_resnet101_coreml_fp32.pte",
178
+ "precision": "fp32",
179
+ "quantized": false,
180
+ "default": false,
181
+ "size": "resnet101",
182
+ "methods": {
183
+ "forward": {
184
+ "inputs": [
185
+ {
186
+ "shape": [
187
+ 1,
188
+ 3,
189
+ 520,
190
+ 520
191
+ ],
192
+ "dtype": "float32"
193
+ }
194
+ ],
195
+ "outputs": [
196
+ {
197
+ "shape": [
198
+ 1,
199
+ 21,
200
+ 520,
201
+ 520
202
+ ],
203
+ "dtype": "float32"
204
+ }
205
+ ]
206
+ }
207
+ }
208
+ }
209
+ ]
210
+ }