mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
13 lines
334 B
Bash
Executable File
13 lines
334 B
Bash
Executable File
export CUDA_VISIBLE_DEVICES="0,1,2,3"
|
|
|
|
./pruned_transducer_stateless_d2v_v2/train.py \
|
|
--world-size 4 \
|
|
--num-epochs 30 \
|
|
--full-libri 1 \
|
|
--use-fp16 1 \
|
|
--max-duration 750 \
|
|
--exp-dir pruned_transducer_stateless7_ctc/exp \
|
|
--feedforward-dims "1024,1024,2048,2048,1024" \
|
|
--ctc-loss-scale 0.2 \
|
|
--master-port 12535
|