experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use export_beam_width in ONNX metadata

Browse Source
This commit is contained in:
Reuben Morais 2021-07-04 14:09:35 +02:00
parent df64a92e3c
commit 5b443a046f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/coqui_stt_training/train.py
View File

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