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 torch.jit.script() export for pruned_transducer_stateless2 (#1410)

Browse Source
This commit is contained in:
Fangjun Kuang 2023-12-10 11:38:39 +08:00 committed by GitHub
parent df56aff31e
commit b0f70c9d04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/export.py
View File

@ -49,6 +49,7 @@ from pathlib import Path
import k2 import k2
import torch import torch
from scaling_converter import convert_scaled_to_non_scaled
from train import add_model_arguments, get_params, get_transducer_model from train import add_model_arguments, get_params, get_transducer_model
from icefall.checkpoint import average_checkpoints, find_checkpoints, load_checkpoint from icefall.checkpoint import average_checkpoints, find_checkpoints, load_checkpoint
@ -198,6 +199,7 @@ def main():
model.eval() model.eval()
if params.jit: if params.jit:
convert_scaled_to_non_scaled(model, inplace=True)
# We won't use the forward() method of the model in C++, so just ignore # We won't use the forward() method of the model in C++, so just ignore
# it here. # it here.
# Otherwise, one of its arguments is a ragged tensor and is not # Otherwise, one of its arguments is a ragged tensor and is not

1
egs/librispeech/ASR/pruned_transducer_stateless2/lstmp.py Symbolic link
View File

@ -0,0 +1 @@
../lstm_transducer_stateless2/lstmp.py

1
egs/librispeech/ASR/pruned_transducer_stateless2/scaling_converter.py Symbolic link
View File

@ -0,0 +1 @@
../pruned_transducer_stateless3/scaling_converter.py