msluszniak commited on
Commit
ee4f0c8
·
verified ·
1 Parent(s): faf5272

Add XNNPACK int8 export

Browse files
.gitattributes CHANGED
@@ -40,3 +40,4 @@ xnnpack/rfdetr_nano_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
40
  mlx/rfdetr_nano_mlx_fp16.pte filter=lfs diff=lfs merge=lfs -text
41
  coreml/rfdetr_nano_coreml_fp16.pte filter=lfs diff=lfs merge=lfs -text
42
  vulkan/rfdetr_nano_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
 
 
40
  mlx/rfdetr_nano_mlx_fp16.pte filter=lfs diff=lfs merge=lfs -text
41
  coreml/rfdetr_nano_coreml_fp16.pte filter=lfs diff=lfs merge=lfs -text
42
  vulkan/rfdetr_nano_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
43
+ xnnpack/rfdetr_nano_xnnpack_int8.pte filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -17,7 +17,15 @@ Upstream model: [RFDetr Nano](https://github.com/roboflow/rf-detr)
17
  | Path | Backend | Precision |
18
  | --- | --- | --- |
19
  | `coreml/rfdetr_nano_coreml_fp16.pte` | coreml | fp16 |
 
20
  | `xnnpack/rfdetr_nano_xnnpack_fp32.pte` | xnnpack | fp32 |
 
 
 
 
 
 
 
21
 
22
  ## Unshipped backends
23
 
@@ -33,8 +41,11 @@ config.json 42 B
33
  coreml/config.json 1.1 kB
34
  coreml/rfdetr_nano_coreml_fp16.pte 52.2 MB
35
  mlx/NOTES.md 1.9 kB
36
- xnnpack/config.json 1.1 kB
 
 
37
  xnnpack/rfdetr_nano_xnnpack_fp32.pte 106 MB
 
38
  ```
39
 
40
  ## Compatibility
 
17
  | Path | Backend | Precision |
18
  | --- | --- | --- |
19
  | `coreml/rfdetr_nano_coreml_fp16.pte` | coreml | fp16 |
20
+ | `vulkan/rfdetr_nano_vulkan_fp16.pte` | vulkan | fp16 |
21
  | `xnnpack/rfdetr_nano_xnnpack_fp32.pte` | xnnpack | fp32 |
22
+ | `xnnpack/rfdetr_nano_xnnpack_int8.pte` | xnnpack | int8 |
23
+
24
+ ## XNNPACK int8
25
+
26
+ `xnnpack/rfdetr_nano_xnnpack_int8.pte` is 8da8w (int8 dynamic activations, int8 per-channel weights) on the Linear
27
+ layers; convolutions stay fp32. COCO box mAP50-95 on the first
28
+ 500 val2017 images: fp32 50.15, int8 49.76.
29
 
30
  ## Unshipped backends
31
 
 
41
  coreml/config.json 1.1 kB
42
  coreml/rfdetr_nano_coreml_fp16.pte 52.2 MB
43
  mlx/NOTES.md 1.9 kB
44
+ vulkan/config.json 1.1 kB
45
+ vulkan/rfdetr_nano_vulkan_fp16.pte 51.5 MB
46
+ xnnpack/config.json 1.8 kB
47
  xnnpack/rfdetr_nano_xnnpack_fp32.pte 106 MB
48
+ xnnpack/rfdetr_nano_xnnpack_int8.pte 37.3 MB
49
  ```
50
 
51
  ## Compatibility
xnnpack/config.json CHANGED
@@ -48,6 +48,46 @@
48
  ]
49
  }
50
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  }
52
  ]
53
  }
 
48
  ]
49
  }
50
  }
51
+ },
52
+ {
53
+ "file": "rfdetr_nano_xnnpack_int8.pte",
54
+ "precision": "int8",
55
+ "methods": {
56
+ "forward": {
57
+ "inputs": [
58
+ {
59
+ "shape": [
60
+ 1,
61
+ 3,
62
+ 384,
63
+ 384
64
+ ],
65
+ "dtype": "float32"
66
+ }
67
+ ],
68
+ "outputs": [
69
+ {
70
+ "shape": [
71
+ 300,
72
+ 4
73
+ ],
74
+ "dtype": "float32"
75
+ },
76
+ {
77
+ "shape": [
78
+ 300
79
+ ],
80
+ "dtype": "float32"
81
+ },
82
+ {
83
+ "shape": [
84
+ 300
85
+ ],
86
+ "dtype": "float32"
87
+ }
88
+ ]
89
+ }
90
+ }
91
  }
92
  ]
93
  }
xnnpack/rfdetr_nano_xnnpack_int8.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ed3ac056b816b5324a4bb247759729e1567f68a35f0ab61f073db56077b411b
3
+ size 39082384