mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
16 lines
411 B
Bash
Executable File
16 lines
411 B
Bash
Executable File
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/test \
|
|
--max-duration 600 \
|
|
--use-fp16 1 \
|
|
--num-encoder-layers 12 \
|
|
--dim-feedforward 2048 \
|
|
--nhead 8 \
|
|
--encoder-dim 512 \
|
|
--decoder-dim 512 \
|
|
--joiner-dim 512
|