Hugging Face's logo

software-mansion
/
react-native-executorch-deeplab-v3

ExecuTorch
Model card Files
xet
 Community
msluszniak commited on
Commit
3062bc5
·
verified ·
1 Parent(s): 1dba9a8

Add backend config (ExecuTorch 1.4.1)

Browse files
Files changed (1) hide show
  1. mlx/config.json +111 -0
mlx/config.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "mlx",
9
+ "license": "bsd-3-clause",
10
+ "variants": [
11
+ {
12
+ "file": "deeplab_v3_mobilenet_v3_large_mlx_fp32.pte",
13
+ "precision": "fp32",
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_mlx_fp32.pte",
46
+ "precision": "fp32",
47
+ "quantized": false,
48
+ "default": true,
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_mlx_fp32.pte",
79
+ "precision": "fp32",
80
+ "quantized": false,
81
+ "default": true,
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
+ }