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的版本。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容