export CUDA_VISIBLE_DEVICES="0,1,2,3" | |
# For non-streaming model training: | |
./zipformer/train.py \ | |
--world-size 4 \ | |
--num-epochs 50 \ | |
--start-epoch 1 \ | |
--use-fp16 1 \ | |
--exp-dir zipformer/exp \ | |
--full-libri 1 \ | |
--use-ctc 1 \ | |
--use-transducer 0 \ | |
--use-attention-decoder 1 \ | |
--ctc-loss-scale 0.1 \ | |
--attention-decoder-loss-scale 0.9 \ | |
--max-duration 1200 \ | |
--master-port 12345 | |