diff --git a/egs/librispeech/ASR/.run_adapter.sh.swp b/egs/librispeech/ASR/.run_adapter.sh.swp index cca875387..fb5725da3 100644 Binary files a/egs/librispeech/ASR/.run_adapter.sh.swp and b/egs/librispeech/ASR/.run_adapter.sh.swp differ diff --git a/egs/librispeech/ASR/run_adapter.sh b/egs/librispeech/ASR/run_adapter.sh index c7ed19296..c51b8efbb 100755 --- a/egs/librispeech/ASR/run_adapter.sh +++ b/egs/librispeech/ASR/run_adapter.sh @@ -7,23 +7,45 @@ if [ $workstation = "whale" ]; then ./pruned_transducer_stateless_d2v_v2/train_adapter.py \ --add-adapter True \ --adapter-lr 0.001 \ - --gender female + --gender male \ --wandb False \ --input-strategy AudioSamples \ --enable-spec-aug False \ --multi-optim False \ - --world-size 8 \ + --world-size 4 \ --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 \ + --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 + +# ./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 \