打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
代码生成离线模型:
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上报错:
麻烦问一下 在mlu270上如何配置生成mlu220芯片边缘盒子上运行的离线模型
热门帖子
精华帖子