Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

请问如何插入split算子 #130

Open
ahaha4321 opened this issue Jan 9, 2025 · 2 comments
Open

请问如何插入split算子 #130

ahaha4321 opened this issue Jan 9, 2025 · 2 comments

Comments

@ahaha4321
Copy link

image
这两个split算子和Add算子是我新增进去的,但是在download模型失败,提示错误:
image
其中split算子左右分别为如下
image
image
Add算子如下
image
如果我不增加Add算子是可以download模型成功的,但是增加了Add算子就失败了。
请问我应该怎么操作?

@ZhangGe6
Copy link
Owner

@ahaha4321 谢谢反馈!从这个case看,onnx-modifier对像是split这种多输出的算子支持还不完善,近期会增加支持。同时也需要您check一下逻辑上构图的正确性:两个split都有多个输出,在执行add操作前是否需要增加类似slice的操作?

@ahaha4321
Copy link
Author

@ahaha4321谢谢反馈!从这个案例来看,onnx-modifier对split这种多输出的算子支持还不够完善,近期会增加支持。同时还需要您检查一下逻辑上结构图的正确性:两个split都有多次输出,在执行添加操作前是否需要增加类似slice的操作?

感谢感谢作者后续完善支持。这个图我只做了简单用例,我需要做4个add来把这两个split的输出连接,另外我已经在我的python代码中实现该功能了,感谢回复!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants