pytoch报错

训练pytorch模型的时候,

1.报出错误:
boxes.append([x1/imgW,y1/imgH,x2/imgW,y2/imgH])

ZeroDivisionError: float division by zero

网上查是因为被除数不能为0。后面检查数据的时候,发现因为给图片和xml文件重命名,导致一些图片无法正常显示,而这部分图片无法正常显示的原因是,图片格式本来png格式的,但是可能数据标注者同意命名为jpg的后缀名了,导致图片无法正常显示,当前无较好解决方案。所以可能导致这部分图片无法被加载,进而无法知道标签信息而出错,所以,有必要对原无法正常显示的图片进行删除,以及重命名文件时注意检查一遍。

2.报出错误:

Expected object of scalar type Float but got scalar type double for argument

本机已经能够训练,但是移植到服务器报这个错误,后面排查这个错误是因为服务器上pytorch的版本和本机不一致导致的,解决方法是降低pytorch的版本。再本机复现这个错误时,发现本机报的错误又有不同,提示Nvidia驱动版本太低,以及建议更新nvidia驱动版本以及安装相应的CUDA版本,解决方法同样是降低pytorch的版本。

你可能感兴趣的:(pytoch报错)