EdgeTPU optimizations (#6808)

* removed transpose op for better edgetpu support

* fix for training case

* enabled experimental new quantizer flag

* precalculate add and mul ops at compile time

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
paradigmandGlenn Jocher authored and GitHub committed 2022-03-12 16:15:09 +01:00
1 parent 701e1177ac
commit c13d4ce7ef
2 files changed
+7 -5

No files matched your search

+1 -1
View File
@@ -331,7 +331,7 @@ def export_tflite(keras_model, im, file, int8, data, ncalib, prefix=colorstr('Te
converter.target_spec.supported_types = []
converter.inference_input_type = tf.uint8 # or tf.int8
converter.inference_output_type = tf.uint8 # or tf.int8
converter.experimental_new_quantizer = False
converter.experimental_new_quantizer = True
f = str(file).replace('.pt', '-int8.tflite')
tflite_model = converter.convert()