| 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 | |