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

支持torchscript模型编译 #24

Open
chenqy4933 opened this issue Nov 23, 2022 · 4 comments
Open

支持torchscript模型编译 #24

chenqy4933 opened this issue Nov 23, 2022 · 4 comments

Comments

@chenqy4933
Copy link
Collaborator

目前MegEngine只支持了MegEngine的模型编译,用户只能使用MegEngine模型进行编译,onnx需要转换到MegEngine,考虑直接支持torchscript模型

@znsoftm
Copy link

znsoftm commented Dec 1, 2022

直接运行torchscript模型比较好。建议支持torchscript和onnx模型。 MegEninge支持的太少了,试了一下,从onnx转换到megEngine模型就是挑战。

@znsoftm
Copy link

znsoftm commented Dec 31, 2022

库主没有计划支持吗? 想法很好,但是通用性欠佳。

@chenqy4933
Copy link
Collaborator Author

有计划,但是短期内人力吃紧,请问你说的通用欠佳是指支持torchscript和onnx模型输入吗?

@znsoftm
Copy link

znsoftm commented Jan 1, 2023

是的,我测试过通过onnx 转到mengine类型,好多转换不了。
建议能不能直接读取pnnx格式进行编译?类似ncnn这样的,通用性会好很多,可以直接打通pytorch到megcc之间的通道。

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