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) ]