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.

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

v8用ckpt转换为onnx时出错 #333

Open
wxqnl opened this issue Jul 31, 2024 · 2 comments
Open

v8用ckpt转换为onnx时出错 #333

wxqnl opened this issue Jul 31, 2024 · 2 comments
Assignees
Labels
bug Something isn't working rfc 需求单issue

Comments

@wxqnl
Copy link

wxqnl commented Jul 31, 2024

芯片310b4,mindspore2.2.14,cann7.0.0,python3.9.2

问题如下:

执行代码: python ./deploy/export.py --config ./configs/yolov8/yolov8n.yaml --weight yolov8-n_500e_mAP372-cc07f5bd.ckpt --per_batch_size 1 --file_format ONNX --device_target CPU

fd9f51884a1d4b60807a6c2308130bc7 20240728135518 49118590252441221578227590072401

网上查资料说是:脚本在导出过程中遇到了一个错误,提示 FillV2 操作符不支持导出。这通常是由于模型中使用了某些 MindSpore 特有的操作符,而这些操作符在 ONNX 中没有直接对应的实现。

如何解决此问题(v5没问题)

@wxqnl wxqnl added the rfc 需求单issue label Jul 31, 2024
@zhanghuiyao zhanghuiyao assigned yuedongli1 and unassigned zhanghuiyao Dec 2, 2024
@zhanghuiyao zhanghuiyao added the bug Something isn't working label Dec 2, 2024
@zhanghuiyao
Copy link
Collaborator

@yuedongli1

@zhanghuiyao
Copy link
Collaborator

看报错应该是fillv2算子在onnx上不支持,(不建议) 可以尝试替换v8里面的fill算子为别的等价实现,比如直接赋值等

⚠️ 当前仅支持导出模型为 mindir 在310机器上进行推理,流程可以参考 deploy 部分readme,暂不支持onnx模型导出,过程可能会遇到不可预期的问题

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working rfc 需求单issue
Projects
None yet
Development

No branches or pull requests

3 participants
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy