search

ZhangGe6 / onnx-modifier

A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.
MIT License
1.28k stars 157 forks source link

添加Reshape算子失败 #88

Closed zerofix2 closed 10 months ago

zerofix2 commented 10 months ago

您好，很感谢您开发的工具。我在使用的时候遇到了一个问题，在模型的输出前添加Reshape算子后，使用onnxruntime推理报错，这是报错信息，似乎和Reshape的shape参数有关： Q1 这是我在onnx-modifier中设置的参数： image 这是在netron中查看的Reshape算子，看起来也是正常的，请问我应该怎么样设置参数才能得到正确的输出呢 image

ZhangGe6 commented 10 months ago

@zerofix2 Reshape op的shape参数数据类型为int64，请将下图里的float32修改为int64试一试。 image

zerofix2 commented 10 months ago

好的，已经修改成功了，感谢您的回复

发件人: @. @.> 代表 Zhang Ge @.> 发送时间: 2023年11月16日 21:15 收件人: ZhangGe6/onnx-modifier @.> 抄送: zerofix2 @.>; Author @.> 主题: Re: [ZhangGe6/onnx-modifier] 添加Reshape算子失败 (Issue #88)

Reshape op的shape参数数据类型为int64，请将下图里的float32修改为int64试一试。 [image]https://user-images.githubusercontent.com/63796752/283460811-1ce21a8b-d758-4d48-a25c-3b87a33327af.png

― Reply to this email directly, view it on GitHubhttps://github.com/ZhangGe6/onnx-modifier/issues/88#issuecomment-1814416498, or unsubscribehttps://github.com/notifications/unsubscribe-auth/ATTF5PLBWWUES42FSPWWL63YEYGWLAVCNFSM6AAAAAA7NT4K7WVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTQMJUGQYTMNBZHA. You are receiving this because you authored the thread.Message ID: @.***>