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

Merge 1559f9c0a9204e4cf5a6317873bf26b6950d920c into 34fc1fdf0d8ff520e2bb18267d046ca207c78ef9

Browse Source
This commit is contained in:
Anjos 2025-07-28 14:51:42 +08:00 committed by GitHub
commit 47edc42e38
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless7_streaming/jit_trace_export.py
View File

@ -132,7 +132,8 @@ def export_encoder_model_jit_trace(
states = encoder_model.get_init_state(device=x.device) states = encoder_model.get_init_state(device=x.device)
encoder_model.__class__.forward = encoder_model.__class__.streaming_forward encoder_model.__class__.forward = encoder_model.__class__.streaming_forward
traced_model = torch.jit.trace(encoder_model, (x, x_lens, states)) # traced_model = torch.jit.trace(encoder_model, (x, x_lens, states))
traced_model = torch.jit.script(encoder_model)
traced_model.save(encoder_filename) traced_model.save(encoder_filename)
logging.info(f"Saved to {encoder_filename}") logging.info(f"Saved to {encoder_filename}")