marcoyang's picture
Create train.sh
e1d475d
raw
history blame
494 Bytes
#!/usr/bin/env bash
python ./conv_emformer_transducer_stateless2_small/train.py \
--world-size 4 \
--num-epochs 40 \
--start-epoch 1 \
--exp-dir conv_emformer_transducer_stateless2_small/exp_960h_small \
--full-libri 1 \
--max-duration 420 \
--master-port 12321 \
--num-encoder-layers 16 \
--chunk-length 32 \
--cnn-module-kernel 31 \
--left-context-length 32 \
--right-context-length 8 \
--memory-size 32 \
--encoder-dim 144 \
--dim-feedforward 576 \
--nhead 4