×
分享到微信

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

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

MLU270生成MLU220 yolov5s离线模型问题 已解决 vicky_shan2022-07-05 10:14:49 回复 2 查看 使用求助
MLU270生成MLU220 yolov5s离线模型问题
分享到:

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

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

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

MLU220: v4.9.4


【使用AI框架】必填*:
Pytorch yolov5s


【操作步骤】选填:

1、系统上同时插上MLU270, MLU220

2、export CNRT_DEFAULT_DEVICE=1 (MLU270)

3、MLU270上生成270的离线模型并成功完成推理。

4、修改core_version:

            ct.set_core_version('MLU220')

            ct.save_as_cambricon('yolov5s_int8_1_4')
            torch.set_grad_enabled(False)
            ct.set_core_number(4)
            trace_input = torch.randn(1, 3, 640, 640, dtype=torch.float).to(ct.mlu_device())
            net = torch.jit.trace(quantized_net, trace_input, check_trace = False)
            net(trace_input)
            ct.save_as_cambricon("")

5、在MLU270上生成MLU220的模型报错。

【出错信息】必填*:

2022-07-04 20:05:50.382283: [cnrtWarning] [34391] [Card : 1] The device you run is not the same as the platform in kernel!
2022-07-04 20:05:50.382316: [cnrtWarning] [34391] [Card : 1] The device you run is MLU270, but the platform in kernel is MLU220!


【当前已做了哪些信息确认】选填:
设置core_version为MLU270可以生成离线模型并推理


请问还需要做什么修改吗?



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