File size: 709 Bytes
00bc520 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
log "Stage 0: finetune"
base_lr=0.004
lr_epochs=100
lr_batches=100000
./pruned_transducer_stateless7_streaming/finetune.py \
--world-size 7 \
--master-port 18181 \
--num-epochs 30 \
--start-epoch 1 \
--init-modules "encoder" \
--exp-dir ./pruned_transducer_stateless7_streaming/exp1 \
--use-fp16 1 \
--base-lr $base_lr \
--lr-epochs $lr_epochs \
--lr-batches $lr_batches \
--bpe-model data/lang_bpe_500/bpe.model \
--do-finetune True \
--finetune-ckp /home/lishaojie/icefall/egs/commonvoice/ASR/pruned_transducer_stateless7_streaming/exp/english_pretrain/epoch-30.pt \
--max-duration 200
fi
|