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

[fix] fix export model input parameter in speechcommand example (#101)

Browse Source 
Co-authored-by: jiqiang.fu <jiqiang.fu@rokid.com>
This commit is contained in:
xiaoqiang306 2022-11-02 21:13:26 +08:00 committed by GitHub
parent f3d6a0a40e
commit 80285fa696
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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