onnx推理正确,转om离线推理错误
收藏回复举报
onnx推理正确,转om离线推理错误
t('forum.solved') 已解决
发表于2024-07-10 19:33:39
0 查看

您好!我在910A卡上遇到一个问题:

用mindspore/samples里面的cycle_gan,https://gitee.com/mindspore/models/tree/master/official/cv/CycleGAN, 训练去雾数据,得到的ckpt转onnx推理正常;用atc将onnx转为om、相同输入情况下,离线推理失败。能否帮忙查看下om模型为何推理结果异常吗?

onnx及推理文件,转出的om及推理代码包以及测试用例如附件。

电话:13811240201,谢谢。亟待答复,拜托!

转换om的环境是cke_24247.png,python是3.9.18;转化指令是:

atc --model=CycleGAN_AtoB_640_15.onnx --framework=5 --output=**** --soc_version=Ascend910A --input_shape="x:1,3,640,640" --out_nodes="77"

我要发帖子