AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix export model input parameter in speechcommand example

Browse Source
This commit is contained in:
jiqiang.fu 2022-11-02 20:45:59 +08:00
parent f3d6a0a40e
commit cc6dc28a80
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
examples/speechcommand_v1/s0/run.sh
View File

@ -101,8 +101,15 @@ fi
if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
python wekws/bin/export_jit.py --config $dir/config.yaml \
jit_model=$(basename $score_checkpoint | sed -e 's:.pt$:.zip:g')
onnx_model=$(basename $score_checkpoint | sed -e 's:.pt$:.onnx:g')
python wekws/bin/export_jit.py \
--config $dir/config.yaml \
--checkpoint $score_checkpoint \
--output_file $dir/final.zip \
--output_quant_file $dir/final.quant.zip
--jit_model $dir/$jit_model
python wekws/bin/export_onnx.py \
--config $dir/config.yaml \
--checkpoint $score_checkpoint \
--onnx_model $dir/$onnx_model
fi