yolo模型格式转换报错

使用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)

验证结果无误。

你可能感兴趣的:(深度学习)