from local

This commit is contained in:
dohe0342 2023-01-12 16:17:28 +09:00
parent 1634126990
commit 0baf5f1b5d
2 changed files with 3 additions and 2 deletions

Binary file not shown.

View File

@ -1,6 +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
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 \
@ -8,7 +9,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/nocombine_layer12_group$group_num \ --exp-dir pruned_transducer_stateless_gtrans/layer24_group$group_num \
--max-duration 900 \ --max-duration 900 \
--use-fp16 1 \ --use-fp16 1 \
--num-encoder-layers 12 \ --num-encoder-layers 12 \