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.
22 lines
363 B
22 lines
363 B
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
if [ $# != 5 ]; then
|
|
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 |