from local

This commit is contained in:
dohe0342 2023-01-13 18:19:01 +09:00
parent 2b4dcf3a45
commit 5034ed579a
3 changed files with 3 additions and 3 deletions

Binary file not shown.

View File

@ -1,7 +1,7 @@
git pull git pull
#for group_num in 12 6 4 3 2 #for group_num in 12 6 4 3 2
for group_num in 4 for group_num in 8
do do
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
./pruned_transducer_stateless_gtrans/train.py \ ./pruned_transducer_stateless_gtrans/train.py \
@ -11,9 +11,9 @@ do
--start-epoch 1 \ --start-epoch 1 \
--full-libri 1 \ --full-libri 1 \
--exp-dir pruned_transducer_stateless_gtrans/layer24_group$group_num \ --exp-dir pruned_transducer_stateless_gtrans/layer24_group$group_num \
--max-duration 600 \ --max-duration 450 \
--use-fp16 1 \ --use-fp16 1 \
--num-encoder-layers 12 \ --num-encoder-layers 24 \
--group-num $group_num \ --group-num $group_num \
--dim-feedforward 2048 \ --dim-feedforward 2048 \
--nhead 8 \ --nhead 8 \