Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -9,29 +9,39 @@ tags:
|
|
| 9 |
---
|
| 10 |
# efficientnet_b1 — ExecuTorch XNNPACK
|
| 11 |
|
| 12 |
-
`efficientnet_b1_xnnpack_fp32.pte` (31.2 MB, fp32, XNNPACK-delegated)
|
| 13 |
-
|
| 14 |
- **Source**: torchvision efficientnet_b1 IMAGENET1K_V2
|
| 15 |
- **License**: BSD-3-Clause
|
| 16 |
- **Input**: [[1, 3, 240, 240]] — RGB, ImageNet norm, 240x240
|
| 17 |
- **Output**: ImageNet logits [1,1000]
|
| 18 |
|
| 19 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
| output | shape | max_abs_diff | corr |
|
| 24 |
|--------|-------|--------------|------|
|
| 25 |
-
| 0 | [1, 1000] |
|
| 26 |
|
| 27 |
-
|
| 28 |
-
ExecuTorch 9.2 ms vs torch eager 345.4 ms.
|
| 29 |
|
| 30 |
## Conversion
|
| 31 |
|
| 32 |
torch.export -> to_edge_transform_and_lower(XnnpackPartitioner) -> .pte
|
| 33 |
(conversion scripts: [executorch-models](https://github.com/john-rocky/executorch-models))
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
---
|
| 37 |
-
Part of [executorch-models](https://github.com/john-rocky/executorch-models) — a verified .pte zoo for ExecuTorch. Conversion scripts and all models are indexed there.
|
|
|
|
| 9 |
---
|
| 10 |
# efficientnet_b1 — ExecuTorch XNNPACK
|
| 11 |
|
|
|
|
|
|
|
| 12 |
- **Source**: torchvision efficientnet_b1 IMAGENET1K_V2
|
| 13 |
- **License**: BSD-3-Clause
|
| 14 |
- **Input**: [[1, 3, 240, 240]] — RGB, ImageNet norm, 240x240
|
| 15 |
- **Output**: ImageNet logits [1,1000]
|
| 16 |
|
| 17 |
+
## Variants
|
| 18 |
+
|
| 19 |
+
All variants take and return fp32 tensors — swap the `.pte` file, keep your app code.
|
| 20 |
+
|
| 21 |
+
| precision | file | size (MB) | parity vs fp32 eager (worst corr) | Mac median (ms)* |
|
| 22 |
+
|-----------|------|-----------|------------------------------------|------------------|
|
| 23 |
+
| fp32 | `efficientnet_b1_xnnpack_fp32.pte` | 31.2 | 1.000000 | 9.3 |
|
| 24 |
+
| fp16 | `efficientnet_b1_xnnpack_fp16.pte` | 28.8 | 0.999816 | 54.9 |
|
| 25 |
+
|
| 26 |
+
\*Mac arm64, single process, median of 10 — a reference point for relative cost
|
| 27 |
+
only, not a device number (torch eager fp32 on the same machine: 352.7 ms).
|
| 28 |
+
|
| 29 |
+
### Precisions that did not earn a slot
|
| 30 |
|
| 31 |
+
- **int8 is not shipped**: worst-output corr 0.077 against fp32 eager, below the 0.95 bar for this precision. The file converts and runs; the numbers do not hold up, so it is left out rather than shipped with a warning.
|
| 32 |
+
|
| 33 |
+
## Verification (executorch 1.4.0, torch 2.13.0)
|
| 34 |
+
|
| 35 |
+
Parity is measured against the fp32 eager model on real image input; `corr` is
|
| 36 |
+
the correlation over all elements of each output tensor.
|
| 37 |
|
| 38 |
| output | shape | max_abs_diff | corr |
|
| 39 |
|--------|-------|--------------|------|
|
| 40 |
+
| 0 | [1, 1000] | 1.490e-05 | 1.000000 |
|
| 41 |
|
| 42 |
+
XNNPACK delegate coverage (fp32): 100.0% (410/410 ops)
|
|
|
|
| 43 |
|
| 44 |
## Conversion
|
| 45 |
|
| 46 |
torch.export -> to_edge_transform_and_lower(XnnpackPartitioner) -> .pte
|
| 47 |
(conversion scripts: [executorch-models](https://github.com/john-rocky/executorch-models))
|
|
|
|
|
|
|
|
|
|
|
|