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 aishell rnnlm training command (#1015)

Browse Source
This commit is contained in:
Wei Kang 2023-04-20 16:14:16 +08:00 committed by GitHub
parent 81d386ef3e
commit 5c65516e05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/aishell/ASR/prepare.sh
View File

@ -230,7 +230,7 @@ if [ $stage -le 9 ] && [ $stop_stage -ge 9 ]; then
if [ ! -f $dl_dir/lm/aishell-train-word.txt ]; then if [ ! -f $dl_dir/lm/aishell-train-word.txt ]; then
cp $lang_phone_dir/transcript_words.txt $dl_dir/lm/aishell-train-word.txt cp $lang_phone_dir/transcript_words.txt $dl_dir/lm/aishell-train-word.txt
fi fi
# training words # training words
./local/prepare_char_lm_training_data.py \ ./local/prepare_char_lm_training_data.py \
--lang-char data/lang_char \ --lang-char data/lang_char \
@ -278,7 +278,7 @@ if [ $stage -le 10 ] && [ $stop_stage -ge 10 ]; then
out_dir=data/lm_training_char out_dir=data/lm_training_char
mkdir -p $out_dir mkdir -p $out_dir
ln -snf ../../../librispeech/ASR/local/sort_lm_training_data.py local/ ln -snf ../../../librispeech/ASR/local/sort_lm_training_data.py local/
./local/sort_lm_training_data.py \ ./local/sort_lm_training_data.py \
--in-lm-data $out_dir/lm_data.pt \ --in-lm-data $out_dir/lm_data.pt \
--out-lm-data $out_dir/sorted_lm_data.pt \ --out-lm-data $out_dir/sorted_lm_data.pt \
@ -306,9 +306,9 @@ if [ $stage -le 11 ] && [ $stop_stage -ge 11 ]; then
--hidden-dim 512 \ --hidden-dim 512 \
--num-layers 2 \ --num-layers 2 \
--batch-size 400 \ --batch-size 400 \
--exp-dir rnnlm_char/exp_aishell1_small \ --exp-dir rnnlm_char/exp \
--lm-data data/lm_char/sorted_lm_data_aishell1.pt \ --lm-data $out_dir/sorted_lm_data.pt \
--lm-data-valid data/lm_char/sorted_lm_data_valid.pt \ --lm-data-valid $out_dir/sorted_lm_data_valid.pt \
--vocab-size 4336 \ --vocab-size 4336 \
--master-port 12345 --master-port 12345
fi fi