Skip MNN export for Raspberry Pi and NVIDIA Jetson (#17292)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
1e70710f3e
commit
daaac35fff
2 changed files with 4 additions and 0 deletions
|
|
@ -77,6 +77,7 @@ from ultralytics.utils import (
|
|||
ARM64,
|
||||
DEFAULT_CFG,
|
||||
IS_JETSON,
|
||||
IS_RASPBERRYPI,
|
||||
LINUX,
|
||||
LOGGER,
|
||||
MACOS,
|
||||
|
|
@ -244,6 +245,8 @@ class Exporter:
|
|||
"WARNING ⚠️ INT8 export requires a missing 'data' arg for calibration. "
|
||||
f"Using default 'data={self.args.data}'."
|
||||
)
|
||||
if mnn and (IS_RASPBERRYPI or IS_JETSON):
|
||||
raise SystemError("MNN export not supported on Raspberry Pi and NVIDIA Jetson")
|
||||
# Input
|
||||
im = torch.zeros(self.args.batch, 3, *self.imgsz).to(self.device)
|
||||
file = Path(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue