diff --git a/egs/LJSpeech/ASR/.run_ad.swp b/egs/LJSpeech/ASR/.run_ad.swp deleted file mode 100644 index 4b86d1d3e..000000000 Binary files a/egs/LJSpeech/ASR/.run_ad.swp and /dev/null differ diff --git a/egs/LJSpeech/ASR/run_adapter.sh b/egs/LJSpeech/ASR/run_adapter.sh new file mode 100755 index 000000000..ccfb8bd95 --- /dev/null +++ b/egs/LJSpeech/ASR/run_adapter.sh @@ -0,0 +1,75 @@ +git pull + +workstation="bear" + +if [ $workstation = "whale" ]; then + export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" + ./pruned_transducer_stateless_d2v_v2/train_adapter.py \ + --add-adapter True \ + --adapter-lr 0.001 \ + --gender male \ + --wandb False \ + --input-strategy AudioSamples \ + --enable-spec-aug False \ + --multi-optim False \ + --world-size 8 \ + --num-epochs 10 \ + --full-libri 1 \ + --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ + --max-duration 200 \ + --encoder-dim 768 \ + --decoder-dim 768 \ + --joiner-dim 768 \ + --use-fp16 0 \ + --accum-grads 1 \ + --encoder-type d2v \ + --additional-block True \ + --prune-range 10 + +# ./pruned_transducer_stateless_d2v_v2/train_adapter.py \ +# --add-adapter True \ +# --adapter-lr 0.001 \ +# --gender female +# --wandb False \ +# --input-strategy AudioSamples \ +# --enable-spec-aug False \ +# --multi-optim False \ +# --world-size 8 \ +# --num-epochs 10 \ +# --full-libri 1 \ +# --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ +# --max-duration 200 \ +# --use-fp16 0 \ +# --encoder-type d2v \ +# --additional-block True \ +# --encoder-dim 768 \ +# --decoder-dim 768 \ +# --joiner-dim 768 \ +# --prune-range 10 +else + export CUDA_VISIBLE_DEVICES="0,1,2,3" + ./pruned_transducer_stateless_d2v_v2/train_adapter.py \ + --add-adapter True \ + --adapter-lr 0.02472 \ + --gender male \ + --wandb True \ + --input-strategy AudioSamples \ + --enable-spec-aug False \ + --multi-optim False \ + --world-size 4 \ + --num-epochs 10 \ + --full-libri 1 \ + --exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \ + --max-duration 150 \ + --encoder-dim 768 \ + --decoder-dim 768 \ + --joiner-dim 768 \ + --use-fp16 0 \ + --accum-grads 4 \ + --encoder-type d2v \ + --additional-block True \ + --prune-range 10 \ + --ctc-loss-scale 0.1924 \ + --lm-scale 0.1254 \ + --simple-loss-scale 0.2869 +fi