Create train.sh

exp/train.sh

@@ -0,0 +1,22 @@
+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