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 decoding finetune model (#1568)

Browse Source
This commit is contained in:
Wei Kang 2024-03-26 10:38:21 +08:00 committed by GitHub
parent b156b6c291
commit 42de459110
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/gigaspeech/KWS/run.sh
View File

@ -47,7 +47,9 @@ fi
if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then
log "Stage 1: Decode the model." log "Stage 1: Decode the model."
for t in small, large; do
export CUDA_VISIBLE_DEVICES="0"
for t in small large; do
python ./zipformer/decode.py \ python ./zipformer/decode.py \
--epoch 12 \ --epoch 12 \
--avg 2 \ --avg 2 \
@ -140,7 +142,8 @@ fi
if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
log "Stage 1: Decode the finetuned model." log "Stage 1: Decode the finetuned model."
for t in small, large; do export CUDA_VISIBLE_DEVICES="0"
for t in small large; do
python ./zipformer/decode.py \ python ./zipformer/decode.py \
--epoch 10 \ --epoch 10 \
--avg 2 \ --avg 2 \

6
egs/wenetspeech/KWS/run.sh
View File

@ -48,7 +48,8 @@ fi
if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then
log "Stage 1: Decode the model." log "Stage 1: Decode the model."
for t in small, large; do export CUDA_VISIBLE_DEVICES="0"
for t in small large; do
python ./zipformer/decode.py \ python ./zipformer/decode.py \
--epoch 18 \ --epoch 18 \
--avg 2 \ --avg 2 \
@ -143,7 +144,8 @@ fi
if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
log "Stage 1: Decode the finetuned model." log "Stage 1: Decode the finetuned model."
for t in small, large; do export CUDA_VISIBLE_DEVICES="0"
for t in small large; do
python ./zipformer/decode.py \ python ./zipformer/decode.py \
--epoch 10 \ --epoch 10 \
--avg 2 \ --avg 2 \