Added Error for TFLite int8 end2end model export (#17360)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/17332/head^2
Francesco Mattioli 3 weeks ago committed by GitHub
parent d0abd95f95
commit da15e27a4d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      ultralytics/engine/exporter.py

@ -226,6 +226,8 @@ class Exporter:
if self.args.optimize:
assert not ncnn, "optimize=True not compatible with format='ncnn', i.e. use optimize=False"
assert self.device.type == "cpu", "optimize=True not compatible with cuda devices, i.e. use device='cpu'"
if self.args.int8 and tflite:
assert not model.end2end, "TFLite INT8 export not supported for end2end models, please use half precision."
if edgetpu:
if not LINUX:
raise SystemError("Edge TPU export only supported on Linux. See https://coral.ai/docs/edgetpu/compiler")

Loading…
Cancel
Save