Use export_beam_width in ONNX metadata
This commit is contained in:
parent
df64a92e3c
commit
5b443a046f
@ -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"
|
||||
|
Loading…
Reference in New Issue
Block a user