×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

【PyTorch MLU】在生成离线模型时,设置多个core后离线模型运行结果不正确 已完结 Xinqian2022-11-22 16:05:13 回复 11 查看 使用求助
【PyTorch MLU】在生成离线模型时,设置多个core后离线模型运行结果不正确
分享到:

【寒武纪硬件产品型号】必填*:
MLU270


【使用操作系统】必填*:
ubuntu


【使用驱动版本】必填*:
v4.20.6


【出错信息】必填*:

各位大佬,我在生成离线模型时,设置多个core导致离线模型运行结果不正确。

ct.set_core_number(16) # 设置4,8,16都能正确运行但是结果不对

将上面代码注释后,运行

traced_model_mlu = torch.jit.trace(quantized_net, trace_input, check_trace = False)
detect_out = traced_model_mlu(trace_input)
box_result = get_boxes_new(detect_out, batch_size=4)

结果是正确的,但是设置多个core后,运行结果就不正确了,请问这是什么原因导致的,应该如何解决呢?

【当前已做了哪些信息确认】选填:
如上


【相关日志文档】选填


【出错代码链接】选填:

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭