你好,加完之后还是报同样的错误,加完之后的代码如下:[图片]
你好,加完之后还是报同样的错误,加完之后的代码如下:[图片]
您好,在执行pred=quantized_net(img)[0]之前,加一段
quantized_net.to(ct.mlu_device()) trace_input = torch.randn(batch_size, 3, 640, 640,dtype=torch.float) trace_input=trace_input.half() trace_input=trace_input.to(ct.mlu_device()) quantized_net = torch.jit.trace(quantized_net, trace_input, check_trace = False)
试试。
请登录后评论