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

mobilenetv3当作Backbone无法导出onnx模型 #42

Closed
quhaohao opened this issue Aug 2, 2021 · 2 comments
Closed

mobilenetv3当作Backbone无法导出onnx模型 #42

quhaohao opened this issue Aug 2, 2021 · 2 comments

Comments

@quhaohao
Copy link

quhaohao commented Aug 2, 2021

大佬，您好。感谢您的开源项目，我尝试用mobilenetv3当作Backbone来训练，精度感觉还可以，训完执行deploy/export.py却无法生成onnx的文件，也不报错。请问有什么解决办法吗？
@Bobo-y
Copy link
Owner

Bobo-y commented Aug 2, 2021

@quhaohao 我这两天试一下看看，我并没有把每一个组合都完整测试了一遍

@Bobo-y
Copy link
Owner

Bobo-y commented Aug 2, 2021

@quhaohao 由于目前onnx 无法导出 F.threshold 函数，所以需要把 mobilenet SE 模块中的 HardSigmoid() 替换掉，便可导出

@Bobo-y Bobo-y closed this as completed Aug 2, 2021
@Bobo-y Bobo-y pinned this issue Sep 6, 2021
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
@Bobo-y @quhaohao