diff --git a/egs/librispeech/ASR/.gtrans.sh.swp b/egs/librispeech/ASR/.gtrans.sh.swp index ab6ed93bf..aea435f00 100644 Binary files a/egs/librispeech/ASR/.gtrans.sh.swp and b/egs/librispeech/ASR/.gtrans.sh.swp differ diff --git a/egs/librispeech/ASR/gtrans.sh b/egs/librispeech/ASR/gtrans.sh new file mode 100644 index 000000000..c03ea292d --- /dev/null +++ b/egs/librispeech/ASR/gtrans.sh @@ -0,0 +1,15 @@ +export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" +./pruned_transducer_stateless_gtrans/train.py \ + --world-size 8 \ + --num-epochs 30 \ + --start-epoch 1 \ + --full-libri 1 \ + --exp-dir pruned_transducer_stateless_gtrans/layer12_group6 \ + --max-duration 600 \ + --use-fp16 1 \ + --num-encoder-layers 12 \ + --dim-feedforward 2048 \ + --nhead 8 \ + --encoder-dim 512 \ + --decoder-dim 512 \ + --joiner-dim 512