mgeconvert .pkl转.onnx有问题

.pkl转.onnx有问题

Open nnzs9248 opened this issue 2 years ago • 3 comments

想将CVPR2021的NBNet训练的模型转为.onnx，报错，是否目前仅支持部分网络？ lALPJxf-v_HmYN3NASvNBU0_1357_299

Megengine 1.7.0+cu112

Mar 04 '22 06:03 nnzs9248

要先把 .pkl 转 .mge 格式，再转 .onnx

https://megengine.org.cn/doc/stable/zh/user-guide/model-development/jit/dump.html

Mar 04 '22 06:03 ChaiByte

要先把 .pkl 转 .mge 格式，再转 .onnx

https://megengine.org.cn/doc/stable/zh/user-guide/model-development/jit/dump.html

任意网络结构都支持吗

Mar 04 '22 07:03 nnzs9248

要先把 .pkl 转 .mge 格式，再转 .onnx https://megengine.org.cn/doc/stable/zh/user-guide/model-development/jit/dump.html

任意网络结构都支持吗

显然没有这种保证。取决于 1. ONXX Opset 是否对齐； 2. 相应的 OP 在 MegEngine 中是否已经实现。

从实践经验上看，走 Torch -> ONNX -> MegEngine 路线踩的坑，可能会略微比用 MegEngine 搭好网络结构训练麻烦些。

Mar 04 '22 07:03 ChaiByte