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.

Sign up for GitHub

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

Jump to bottom

.pkl转.onnx有问题 #67

Open
nnzs9248 opened this issue Mar 4, 2022 · 3 comments
Open

.pkl转.onnx有问题 #67

nnzs9248 opened this issue Mar 4, 2022 · 3 comments

Comments

@nnzs9248
Copy link

nnzs9248 commented Mar 4, 2022

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

Megengine 1.7.0+cu112
@nnzs9248 nnzs9248 changed the title .mge转.onnx有问题 .pkl转.onnx有问题 Mar 4, 2022
@ChaiByte
Copy link

ChaiByte commented Mar 4, 2022

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

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

@nnzs9248
Copy link
Author

nnzs9248 commented Mar 4, 2022

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

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

任意网络结构都支持吗

@ChaiByte
Copy link

ChaiByte commented Mar 4, 2022 •
edited
Loading

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@ChaiByte @nnzs9248