from local

This commit is contained in:
dohe0342 2023-01-18 21:52:20 +09:00
parent 1c6eb1f661
commit b8efe4ad77
2 changed files with 2 additions and 2 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 6
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 \
@ -10,7 +10,7 @@ do
--num-epochs 30 \ --num-epochs 30 \
--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/layer18_group$group_num \
--max-duration 600 \ --max-duration 600 \
--use-fp16 1 \ --use-fp16 1 \
--num-encoder-layers 18 \ --num-encoder-layers 18 \