Merge pull request #1878 from lissyx/tflite-quantize

Enable TFLite post-training quantization
2019-02-14 19:48:40 +01:00 · 2019-02-14 19:48:40 +01:00 · 8cd2071b5d
commit 8cd2071b5d
parent e2eb9f0188 e6cca187d6
1 changed files with 1 additions and 1 deletions
--- a/DeepSpeech.py
+++ b/DeepSpeech.py
@ -808,6 +808,7 @@ def export():
                        self.output_arrays  = output_names
                        self.output_file    = output_tflite_path
                        self.output_format  = 'TFLITE'
+                        self.post_training_quantize = True

                        default_empty = [
                            'inference_input_type',
@ -818,7 +819,6 @@ def export():
                            'change_concat_input_ranges',
                            'allow_custom_ops',
                            'converter_mode',
-                            'post_training_quantize',
                            'dump_graphviz_dir',
                            'dump_graphviz_video'
                        ]