convert_onnx_batch_size.sh

input_dir=/home/sunzhq/workspace/yidong-infer/conformer/onnx_models_1
output_dir=/home/sunzhq/workspace/yidong-infer/conformer/onnx_models_batch24_1

rm -rf ${output_dir}
mkdir -p ${output_dir}
cp -r ${input_dir}/* ${output_dir}
rm -rf ${output_dir}/transformer_lm/full/*



python convert_onnx_batch_size.py \
  --input ${input_dir}/transformer_lm/full \
  --output ${output_dir}/transformer_lm/full/ \
  --batch_size 24 \
  --batch_mode