from local

This commit is contained in:
dohe0342 2023-01-08 18:36:07 +09:00
parent a1751ded51
commit 0b6c728fa4
2 changed files with 15 additions and 0 deletions

Binary file not shown.

View File

@ -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