diff --git a/egs/librispeech/ASR/.gtrans.sh.swp b/egs/librispeech/ASR/.gtrans.sh.swp index 76a06561d..df586c207 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 index e24ffb2db..6025ddb72 100755 --- a/egs/librispeech/ASR/gtrans.sh +++ b/egs/librispeech/ASR/gtrans.sh @@ -1,7 +1,7 @@ git pull #for group_num in 12 6 4 3 2 -for group_num in 4 +for group_num in 8 do export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" ./pruned_transducer_stateless_gtrans/train.py \ @@ -11,9 +11,9 @@ do --start-epoch 1 \ --full-libri 1 \ --exp-dir pruned_transducer_stateless_gtrans/layer24_group$group_num \ - --max-duration 600 \ + --max-duration 450 \ --use-fp16 1 \ - --num-encoder-layers 12 \ + --num-encoder-layers 24 \ --group-num $group_num \ --dim-feedforward 2048 \ --nhead 8 \ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp index 8455406da..e53c528cd 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp differ