New ONNX, TorchScript, CoreML export test matrices (#11652)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
2583f842b8
commit
972f615110
2 changed files with 74 additions and 17 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue