From 4482404a542f15e6623b2cef81187112c25b86e5 Mon Sep 17 00:00:00 2001 From: "yang.liu636" Date: Wed, 9 Oct 2024 08:37:33 +0000 Subject: [PATCH] =?UTF-8?q?update=20ACL=5FPyTorch/built-in/nlp/Uie=5Ffor?= =?UTF-8?q?=5FPytorch/fix=5Fonnx.py.=20=E4=BC=A0=E5=8F=82=E5=92=8C?= =?UTF-8?q?=E7=AD=BE=E5=90=8D=E4=B8=8D=E5=8C=B9=E9=85=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: yang.liu636 --- ACL_PyTorch/built-in/nlp/Uie_for_Pytorch/fix_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ACL_PyTorch/built-in/nlp/Uie_for_Pytorch/fix_onnx.py b/ACL_PyTorch/built-in/nlp/Uie_for_Pytorch/fix_onnx.py index e7e5292219..356216382b 100644 --- a/ACL_PyTorch/built-in/nlp/Uie_for_Pytorch/fix_onnx.py +++ b/ACL_PyTorch/built-in/nlp/Uie_for_Pytorch/fix_onnx.py @@ -104,7 +104,7 @@ if __name__ == '__main__': seq_len_ = sys.argv[4] onnx_graph = OnnxGraph.parse(input_path) fix_mul(onnx_graph) - fix_add_shape(onnx_graph, bs_) + fix_add_shape(onnx_graph) fix_transpose(onnx_graph) fix_reshape(onnx_graph, bs_, seq_len_) onnx_graph.save(save_path) -- Gitee