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
519 B
22 lines
519 B
2 years ago
|
train_output_path=$1
|
||
|
|
model_dir=$2
|
||
|
|
output_dir=$3
|
||
|
|
model=$4
|
||
|
|
|
||
|
|
enable_dev_version=True
|
||
|
|
|
||
|
|
model_name=${model%_*}
|
||
|
|
echo model_name: ${model_name}
|
||
|
|
|
||
|
|
if [ ${model_name} = 'mb_melgan' ] ;then
|
||
|
|
enable_dev_version=False
|
||
|
|
fi
|
||
|
|
|
||
|
|
mkdir -p ${train_output_path}/${output_dir}
|
||
|
|
|
||
|
|
paddle2onnx \
|
||
|
|
--model_dir ${train_output_path}/${model_dir} \
|
||
|
|
--model_filename ${model}.pdmodel \
|
||
|
|
--params_filename ${model}.pdiparams \
|
||
|
|
--save_file ${train_output_path}/${output_dir}/${model}.onnx \
|
||
|
|
--enable_dev_version ${enable_dev_version}
|