mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

small fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2025-07-16 18:20:33 +08:00
parent 1e792e4fea
commit d14d2d1909
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
egs/fluent_speech_commands/SLU/transducer/export-onnx.py
View File

@ -276,20 +276,20 @@ def main():
logging.info("Exporting encoder")
encoder_filename = params.exp_dir / "encoder.onnx"
# export_encoder_model_onnx(
# encoder,
# encoder_filename,
# opset_version=opset_version,
# )
export_encoder_model_onnx(
encoder,
encoder_filename,
opset_version=opset_version,
)
logging.info(f"Exported encoder to {encoder_filename}")
logging.info("Exporting decoder")
decoder_filename = params.exp_dir / "decoder.onnx"
# export_decoder_model_onnx(
# model.decoder,
# decoder_filename,
# opset_version=opset_version,
# )
export_decoder_model_onnx(
model.decoder,
decoder_filename,
opset_version=opset_version,
)
logging.info(f"Exported decoder to {decoder_filename}")
logging.info("Exporting joiner")