打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
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)
热门帖子
精华帖子