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.
PaddleSpeech/speechx/examples/ds2_ol/onnx
Hui Zhang cd0b82ff1b
fix pretrain model error
2 years ago
..
local fix pretrain model error 2 years ago
.gitignore test with run.sh 2 years ago
README.md format 2 years ago
path.sh pdmodel prune and infershape 2 years ago
run.sh opt onnx script 2 years ago
utils pdmodel prune and infershape 2 years ago

README.md

DeepSpeech2 ONNX model

  1. convert deepspeech2 model to ONNX, using Paddle2ONNX.
  2. check paddleinference and onnxruntime output equal.
  3. optimize onnx model
  4. check paddleinference and optimized onnxruntime output equal.

Please make sure Paddle2ONNX and onnx-simplifier version is correct.

The example test with these packages installed:

paddle2onnx              0.9.8rc0    # develop af4354b4e9a61a93be6490640059a02a4499bc7a
paddleaudio              0.2.1
paddlefsl                1.1.0
paddlenlp                2.2.6
paddlepaddle-gpu         2.2.2
paddlespeech             0.0.0       # develop
paddlespeech-ctcdecoders 0.2.0
paddlespeech-feat        0.1.0
onnx                     1.11.0
onnx-simplifier          0.0.0       # https://github.com/zh794390558/onnx-simplifier/tree/dyn_time_shape
onnxoptimizer            0.2.7
onnxruntime              1.11.0

Using

bash run.sh

For more details please see run.sh.

Outputs

The optimized onnx model is exp/model.opt.onnx.

To show the graph, please using local/netron.sh.