mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 17:42:21 +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") logging.info("Exporting encoder")
encoder_filename = params.exp_dir / "encoder.onnx" encoder_filename = params.exp_dir / "encoder.onnx"
# export_encoder_model_onnx( export_encoder_model_onnx(
# encoder, encoder,
# encoder_filename, encoder_filename,
# opset_version=opset_version, opset_version=opset_version,
# ) )
logging.info(f"Exported encoder to {encoder_filename}") logging.info(f"Exported encoder to {encoder_filename}")
logging.info("Exporting decoder") logging.info("Exporting decoder")
decoder_filename = params.exp_dir / "decoder.onnx" decoder_filename = params.exp_dir / "decoder.onnx"
# export_decoder_model_onnx( export_decoder_model_onnx(
# model.decoder, model.decoder,
# decoder_filename, decoder_filename,
# opset_version=opset_version, opset_version=opset_version,
# ) )
logging.info(f"Exported decoder to {decoder_filename}") logging.info(f"Exported decoder to {decoder_filename}")
logging.info("Exporting joiner") logging.info("Exporting joiner")