使用torch.jit.trace(model,img) 对yolov5模型进行格式转换,出现报错。
内容类似于:
xxxxxxx:the number of sizes provided (0) must be greater or equal to the number of d
解决方法:增加参数check_trace=False
torch.jit.trace(model,img,check_trace=False)
验证结果无误。
使用torch.jit.trace(model,img) 对yolov5模型进行格式转换,出现报错。
内容类似于:
xxxxxxx:the number of sizes provided (0) must be greater or equal to the number of d
解决方法:增加参数check_trace=False
torch.jit.trace(model,img,check_trace=False)
验证结果无误。