diff --git a/egs/librispeech/ASR/.run_v3.sh.swp b/egs/librispeech/ASR/.run_v3.sh.swp index 5e6826faa..778eb3f41 100644 Binary files a/egs/librispeech/ASR/.run_v3.sh.swp and b/egs/librispeech/ASR/.run_v3.sh.swp differ diff --git a/egs/librispeech/ASR/run_v3.sh b/egs/librispeech/ASR/run_v3.sh index e65486d58..cbfb1b20b 100755 --- a/egs/librispeech/ASR/run_v3.sh +++ b/egs/librispeech/ASR/run_v3.sh @@ -1,6 +1,7 @@ git pull -export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" +#export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" +export CUDA_VISIBLE_DEVICES="0,1,2,3" ./pruned_transducer_stateless_d2v_v2/train.py \ --wandb true \ @@ -21,8 +22,8 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" --encoder-type d2v \ --additional-block True \ --encoder-dim 768 \ - --decoder-dim 768 \ - --joiner-dim 768 \ + --decoder-dim 512 \ + --joiner-dim 512 \ --prune-range 20 \ --context-size 2 \ --ctc-loss-scale 0.2