MegEngine/MegCC

支持torchscript模型编译

chenqy4933 opened this issue · 4 comments

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

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

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

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

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