×
分享到微信

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

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

mlu270上生成的离线模型不能在mlu220芯片盒子上运行?是否需要设置生成离线模型参数 yinyin1234562022-04-19 18:44:28 回复 4 查看 技术答疑
mlu270上生成的离线模型不能在mlu220芯片盒子上运行?是否需要设置生成离线模型参数
分享到:

代码生成离线模型:

 if opt.jit:

            ct.save_as_cambricon('yolov5s_v4_head')

            torch.set_grad_enabled(False)

            ct.set_core_number(4)


            trace_input = torch.randn(1, 3, 640, 640, dtype=torch.float)


            input_mlu_data = trace_input.type(torch.HalfTensor).to(ct.mlu_device())


            quantized_net = torch.jit.trace(quantized_net, input_mlu_data, check_trace = False)

按照上面在MLU270上生成离线模型在MLU220上报错:

Dingtalk_20220419184240.jpg麻烦问一下 在mlu270上如何配置生成mlu220芯片边缘盒子上运行的离线模型

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