msluszniak commited on
Commit
cec202f
·
verified ·
1 Parent(s): da37526

Add backend config (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. n/coreml/config.json +139 -0
n/coreml/config.json ADDED
@@ -0,0 +1,139 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "yolo26",
4
+ "family": "yolo",
5
+ "capabilities": [
6
+ "object-detection"
7
+ ],
8
+ "backend": "coreml",
9
+ "license": "agpl-3.0",
10
+ "variants": [
11
+ {
12
+ "file": "yolo26n_384_coreml_fp16.pte",
13
+ "precision": "fp16",
14
+ "quantized": false,
15
+ "default": false,
16
+ "methods": {
17
+ "forward": {
18
+ "inputs": [
19
+ {
20
+ "shape": [
21
+ 1,
22
+ 3,
23
+ 384,
24
+ 384
25
+ ],
26
+ "dtype": "float32"
27
+ }
28
+ ],
29
+ "outputs": [
30
+ {
31
+ "shape": [
32
+ 300,
33
+ 4
34
+ ],
35
+ "dtype": "float32"
36
+ },
37
+ {
38
+ "shape": [
39
+ 300
40
+ ],
41
+ "dtype": "float32"
42
+ },
43
+ {
44
+ "shape": [
45
+ 300
46
+ ],
47
+ "dtype": "float32"
48
+ }
49
+ ]
50
+ }
51
+ }
52
+ },
53
+ {
54
+ "file": "yolo26n_512_coreml_fp16.pte",
55
+ "precision": "fp16",
56
+ "quantized": false,
57
+ "default": false,
58
+ "methods": {
59
+ "forward": {
60
+ "inputs": [
61
+ {
62
+ "shape": [
63
+ 1,
64
+ 3,
65
+ 512,
66
+ 512
67
+ ],
68
+ "dtype": "float32"
69
+ }
70
+ ],
71
+ "outputs": [
72
+ {
73
+ "shape": [
74
+ 300,
75
+ 4
76
+ ],
77
+ "dtype": "float32"
78
+ },
79
+ {
80
+ "shape": [
81
+ 300
82
+ ],
83
+ "dtype": "float32"
84
+ },
85
+ {
86
+ "shape": [
87
+ 300
88
+ ],
89
+ "dtype": "float32"
90
+ }
91
+ ]
92
+ }
93
+ }
94
+ },
95
+ {
96
+ "file": "yolo26n_640_coreml_fp16.pte",
97
+ "precision": "fp16",
98
+ "quantized": false,
99
+ "default": true,
100
+ "methods": {
101
+ "forward": {
102
+ "inputs": [
103
+ {
104
+ "shape": [
105
+ 1,
106
+ 3,
107
+ 640,
108
+ 640
109
+ ],
110
+ "dtype": "float32"
111
+ }
112
+ ],
113
+ "outputs": [
114
+ {
115
+ "shape": [
116
+ 300,
117
+ 4
118
+ ],
119
+ "dtype": "float32"
120
+ },
121
+ {
122
+ "shape": [
123
+ 300
124
+ ],
125
+ "dtype": "float32"
126
+ },
127
+ {
128
+ "shape": [
129
+ 300
130
+ ],
131
+ "dtype": "float32"
132
+ }
133
+ ]
134
+ }
135
+ }
136
+ }
137
+ ],
138
+ "size": "n"
139
+ }