From 7a10ed6971c7ac7556000bf92bbad1d0ea371559 Mon Sep 17 00:00:00 2001 From: nicolaspanel Date: Fri, 1 Mar 2019 15:10:05 +0100 Subject: [PATCH] allow custom export_batch_size (see #1920) --- DeepSpeech.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DeepSpeech.py b/DeepSpeech.py index 94bf1671..20a69326 100755 --- a/DeepSpeech.py +++ b/DeepSpeech.py @@ -749,7 +749,7 @@ def export(): tf.reset_default_graph() session = tf.Session(config=Config.session_config) - inputs, outputs, _ = create_inference_graph(batch_size=1, n_steps=FLAGS.n_steps, tflite=FLAGS.export_tflite) + inputs, outputs, _ = create_inference_graph(batch_size=FLAGS.export_batch_size, n_steps=FLAGS.n_steps, tflite=FLAGS.export_tflite) input_names = ",".join(tensor.op.name for tensor in inputs.values()) output_names_tensors = [ tensor.op.name for tensor in outputs.values() if isinstance(tensor, Tensor) ] output_names_ops = [ tensor.name for tensor in outputs.values() if isinstance(tensor, Operation) ]