使用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在对Yolov5模型进行格式转换时遇到的错误,通过设置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)
验证结果无误。
997

被折叠的 条评论
为什么被折叠?