.pkl转.onnx有问题

[nnzs9248]

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

Megengine 1.7.0+cu112

[0xMashiro]

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

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

[nnzs9248]

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

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

任意网络结构都支持吗

[0xMashiro]

要先把 .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 搭好网络结构 训练麻烦些。