打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
batchsize = 1
corenum = 4
...
elif opt.cfg == 'mlu':
from models.yolo import Model
device = ct.mlu_device()
model = Model('./models/yolov5s.yaml').to(torch.device('cpu'))
model.float().fuse().eval()
quantized_net = torch_mlu.core.mlu_quantize.quantize_dynamic_mlu(model)
state_dict = torch.load("./yolov5s_int8.pt")
quantized_net.load_state_dict(state_dict, strict=False)
quantized_net.eval()
quantized_net.to(ct.mlu_device())
if opt.jit:
print("### jit")
ct.save_as_cambricon('yolov5s_int8_' + str(batchsize)+ '_' + str(corenum))
torch.set_grad_enabled(False)
ct.set_core_number(corenum)
trace_input = torch.randn(batchsize, 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)热门帖子
精华帖子