convert_onnx_batch_size.sh 266 Bytes
Newer Older
sunzhq2's avatar
sunzhq2 committed
1
2
3
4
5
6
# 批量转换所有模型
python convert_onnx_batch_size.py \
  --input /home/sunzhq/workspace/yidong-infer/conformer/onnx_models/transformer_lm/full \
  --output /home/sunzhq/workspace/yidong-infer/conformer/onnx_models_batch24 \
  --batch_size 24 \
  --batch_mode