You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
570 B
23 lines
570 B
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
if [ $# != 5 ]; then
|
|
# local/prune.sh data/exp/deepspeech2_online/checkpoints avg_1.jit.pdmodel avg_1.jit.pdiparams softmax_0.tmp_0,tmp_5,concat_0.tmp_0,concat_1.tmp_0 $PWD
|
|
echo "usage: $0 model_dir model_filename param_filename outputs_names save_dir"
|
|
exit 1
|
|
fi
|
|
|
|
dir=$1
|
|
model=$2
|
|
param=$3
|
|
outputs=$4
|
|
save_dir=$5
|
|
|
|
|
|
python local/pd_prune_model.py \
|
|
--model_dir $dir \
|
|
--model_filename $model \
|
|
--params_filename $param \
|
|
--output_names $outputs \
|
|
--save_dir $save_dir |