Use export_beam_width in ONNX metadata

This commit is contained in:
Reuben Morais 2021-07-04 14:09:35 +02:00
parent df64a92e3c
commit 5b443a046f

View File

@ -1270,7 +1270,7 @@ def export_onnx():
meta = onnx_model.metadata_props.add()
meta.key = "beam_width"
meta.value = str(Config.beam_width)
meta.value = str(Config.export_beam_width)
meta = onnx_model.metadata_props.add()
meta.key = "alphabet"