New ONNX, TorchScript, CoreML export test matrices (#11652)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-05-05 16:57:52 +02:00 committed by GitHub
parent 2583f842b8
commit 972f615110
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 74 additions and 17 deletions

View file

@ -599,6 +599,7 @@ class Exporter:
LOGGER.info(f"\n{prefix} starting export with coremltools {ct.__version__}...")
assert not WINDOWS, "CoreML export is not supported on Windows, please run on macOS or Linux."
assert self.args.batch == 1, "CoreML batch sizes > 1 are not supported. Please retry at 'batch=1'."
f = self.file.with_suffix(".mlmodel" if mlmodel else ".mlpackage")
if f.is_dir():
shutil.rmtree(f)