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

fix typo for export jit script (#544)

Browse Source
This commit is contained in:
Yuekai Zhang 2022-08-23 17:29:42 +08:00 committed by GitHub
parent dbd61a9db3
commit f9c3d7f92f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/librispeech/ASR/pruned_transducer_stateless3/export.py
View File

@ -652,13 +652,13 @@ def main():
# Also export encoder/decoder/joiner separately # Also export encoder/decoder/joiner separately
encoder_filename = params.exp_dir / "encoder_jit_script.pt" encoder_filename = params.exp_dir / "encoder_jit_script.pt"
export_encoder_model_jit_trace(model.encoder, encoder_filename) export_encoder_model_jit_script(model.encoder, encoder_filename)
decoder_filename = params.exp_dir / "decoder_jit_script.pt" decoder_filename = params.exp_dir / "decoder_jit_script.pt"
export_decoder_model_jit_trace(model.decoder, decoder_filename) export_decoder_model_jit_script(model.decoder, decoder_filename)
joiner_filename = params.exp_dir / "joiner_jit_script.pt" joiner_filename = params.exp_dir / "joiner_jit_script.pt"
export_joiner_model_jit_trace(model.joiner, joiner_filename) export_joiner_model_jit_script(model.joiner, joiner_filename)
elif params.jit_trace is True: elif params.jit_trace is True:
convert_scaled_to_non_scaled(model, inplace=True) convert_scaled_to_non_scaled(model, inplace=True)