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