diff --git a/speechx/examples/ds2_ol/onnx/local/netron.sh b/speechx/examples/ds2_ol/onnx/local/netron.sh index 73e089edb363411a190753799188f7ebcb0e85b0..6dd9a39c9200af3934a6b9e52568e7dd943f94dc 100755 --- a/speechx/examples/ds2_ol/onnx/local/netron.sh +++ b/speechx/examples/ds2_ol/onnx/local/netron.sh @@ -1,5 +1,7 @@ #!/bin/bash +# show model + if [ $# != 1 ];then echo "usage: $0 model_path" exit 1 diff --git a/speechx/examples/ds2_ol/onnx/local/onnx_clone.sh b/speechx/examples/ds2_ol/onnx/local/onnx_clone.sh index 0a472af49cc06d885f957f7f1194501980b26d81..bce22dbc8acf2a3a3d62d4eabea9b99b78c85a90 100644 --- a/speechx/examples/ds2_ol/onnx/local/onnx_clone.sh +++ b/speechx/examples/ds2_ol/onnx/local/onnx_clone.sh @@ -1,6 +1,7 @@ #!/bin/bash +# clone onnx repos git clone https://github.com/onnx/onnx.git git clone https://github.com/microsoft/onnxruntime.git git clone https://github.com/PaddlePaddle/Paddle2ONNX.git \ No newline at end of file diff --git a/speechx/examples/ds2_ol/onnx/local/onnx_opt.sh b/speechx/examples/ds2_ol/onnx/local/onnx_opt.sh index dd8fbd20bc5d49b7a68ae70515d4e5839342221e..470a9a31755cf80d5ef5b4bc1051e68d04cd9721 100755 --- a/speechx/examples/ds2_ol/onnx/local/onnx_opt.sh +++ b/speechx/examples/ds2_ol/onnx/local/onnx_opt.sh @@ -1,4 +1,4 @@ #!/bin/bash - +# onnx optimizer onnx-simplifier \ No newline at end of file diff --git a/speechx/examples/ds2_ol/onnx/local/onnx_prune_model.py b/speechx/examples/ds2_ol/onnx/local/onnx_prune_model.py index a5148eddaa9ed455b4991fb2b96c2e0156fcd29b..f709b8f30f0b36ec23a2da0b25ea07b65faa85e1 100644 --- a/speechx/examples/ds2_ol/onnx/local/onnx_prune_model.py +++ b/speechx/examples/ds2_ol/onnx/local/onnx_prune_model.py @@ -1,11 +1,13 @@ #!/usr/bin/env python3 -W ignore::DeprecationWarning + +# prune model by output names + import argparse import copy import sys import onnx - def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument( diff --git a/speechx/examples/ds2_ol/onnx/local/onnx_rename_model.py b/speechx/examples/ds2_ol/onnx/local/onnx_rename_model.py index f508c0a359d52144e365e592019a76cc1fd5b75e..8724c84d7e48cca30880a2c979b8dcf557f2846d 100755 --- a/speechx/examples/ds2_ol/onnx/local/onnx_rename_model.py +++ b/speechx/examples/ds2_ol/onnx/local/onnx_rename_model.py @@ -1,4 +1,6 @@ #!/usr/bin/env python3 -W ignore::DeprecationWarning + +# rename node to new names import argparse import sys diff --git a/speechx/examples/ds2_ol/onnx/local/pd_infer_shape.py b/speechx/examples/ds2_ol/onnx/local/pd_infer_shape.py index 318131f9d2d78ff095ff656e8a2f1e57fe7d8b15..723137b8ef37fe1237b5e4aa806968782cf429d2 100755 --- a/speechx/examples/ds2_ol/onnx/local/pd_infer_shape.py +++ b/speechx/examples/ds2_ol/onnx/local/pd_infer_shape.py @@ -2,6 +2,7 @@ # https://github.com/jiangjiajun/PaddleUtils/blob/main/paddle/README.md#2-%E4%BF%AE%E6%94%B9paddle%E6%A8%A1%E5%9E%8B%E8%BE%93%E5%85%A5shape import argparse +# paddle inference shape def process_old_ops_desc(program): """set matmul op head_number attr to 1 is not exist. diff --git a/speechx/examples/ds2_ol/onnx/local/pd_prune_model.py b/speechx/examples/ds2_ol/onnx/local/pd_prune_model.py index d723c7cef9eac16282ec3e104f0645ee3ba6dd92..78346651f1ec63929237abb1b391a4c07b4878dc 100755 --- a/speechx/examples/ds2_ol/onnx/local/pd_prune_model.py +++ b/speechx/examples/ds2_ol/onnx/local/pd_prune_model.py @@ -4,6 +4,7 @@ import argparse import sys from typing import List +# paddle prune model. def prepend_feed_ops(program, feed_target_names: List[str],