RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26


在使用pytorch训练模型时遇到上面的问题,查了好久也不知道啥原因,最后在别人的帖子里找到类似的问题,就试试

解决办法:检查labels中是否有负值,如果有改成非负整数就可以了


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM