|
#!/usr/bin/env bash |
|
|
|
set -x |
|
|
|
K2_ROOT=/ceph-hw/kangwei/code/k2_release/k2 |
|
ICEFALL=/ceph-hw/kangwei/code2/icefall_streaming2 |
|
|
|
export PYTHONPATH=$K2_ROOT/k2/python:$PYTHONPATH |
|
export PYTHONPATH=$K2_ROOT/build/lib:$PYTHONPATH |
|
export PYTHONPATH=$ICEFALL:$PYTHONPATH |
|
|
|
export CUDA_VISIBLE_DEVICES="" |
|
|
|
./pruned_transducer_stateless5/export.py \ |
|
--streaming-model 1 \ |
|
--causal-convolution 1 \ |
|
--num-encoder-layers 18 \ |
|
--dim-feedforward 2048 \ |
|
--nhead 8 \ |
|
--encoder-dim 512 \ |
|
--decoder-dim 512 \ |
|
--joiner-dim 512 \ |
|
--epoch 25 \ |
|
--avg 5 \ |
|
--exp-dir ./pruned_transducer_stateless5/exp-L-nolinear |
|
|
|
./pruned_transducer_stateless5/export.py \ |
|
--streaming-model 1 \ |
|
--causal-convolution 1 \ |
|
--num-encoder-layers 18 \ |
|
--dim-feedforward 2048 \ |
|
--nhead 8 \ |
|
--encoder-dim 512 \ |
|
--decoder-dim 512 \ |
|
--joiner-dim 512 \ |
|
--epoch 25 \ |
|
--avg 5 \ |
|
--exp-dir ./pruned_transducer_stateless5/exp-L-nolinear \ |
|
--jit 1 |
|
|