marcoyang commited on
Commit
e1d475d
1 Parent(s): 4cbc7c6

Create train.sh

Browse files
Files changed (1) hide show
  1. train.sh +19 -0
train.sh ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ python ./conv_emformer_transducer_stateless2_small/train.py \
4
+ --world-size 4 \
5
+ --num-epochs 40 \
6
+ --start-epoch 1 \
7
+ --exp-dir conv_emformer_transducer_stateless2_small/exp_960h_small \
8
+ --full-libri 1 \
9
+ --max-duration 420 \
10
+ --master-port 12321 \
11
+ --num-encoder-layers 16 \
12
+ --chunk-length 32 \
13
+ --cnn-module-kernel 31 \
14
+ --left-context-length 32 \
15
+ --right-context-length 8 \
16
+ --memory-size 32 \
17
+ --encoder-dim 144 \
18
+ --dim-feedforward 576 \
19
+ --nhead 4