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

Update pretrained.py

Browse Source
This commit is contained in:
jinzr 2023-07-13 14:19:22 +08:00
parent 9e79cf9f68
commit 54c023034e
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless7/pretrained.py
View File

@ -21,7 +21,7 @@ You can generate the checkpoint with the following command:
./pruned_transducer_stateless7/export.py \
--exp-dir ./pruned_transducer_stateless7/exp \
--bpe-model data/lang_bpe_500/bpe.model \
--tokens ./data/lang_bpe_500/tokens.txt \
--epoch 20 \
--avg 10
@ -87,9 +87,8 @@ from beam_search import (
)
from torch.nn.utils.rnn import pad_sequence
from train import add_model_arguments, get_params, get_transducer_model
from export import num_tokens
from icefall.utils import str2bool
from icefall.utils import num_tokens
def get_parser():