|
#!/usr/bin/env bash |
|
|
|
|
|
export CUDA_VISIBLE_DEVICES=0 |
|
|
|
. path.sh |
|
|
|
set -ex |
|
|
|
iter= |
|
epoch=39 |
|
avg=6 |
|
use_averaged_model=1 |
|
jit=0 |
|
|
|
|
|
./pruned_stateless_emformer_rnnt2/export.py \ |
|
--exp-dir ./pruned_stateless_emformer_rnnt2/exp-full \ |
|
--bpe-model data/lang_bpe_500/bpe.model \ |
|
--epoch $epoch \ |
|
--avg $avg \ |
|
--jit $jit \ |
|
--use-averaged-model $use_averaged_model |
|
|
|
if [ $jit -eq 1 ]; then |
|
mv -v ./pruned_stateless_emformer_rnnt2/exp-full/cpu_jit.pt ./pruned_stateless_emformer_rnnt2/exp-full/cpu_jit-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt |
|
else |
|
mv -v ./pruned_stateless_emformer_rnnt2/exp-full/pretrained.pt ./pruned_stateless_emformer_rnnt2/exp-full/pretrained-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt |
|
fi |
|
|
