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

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

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

posted on 2020-04-29 18:03  川洋  阅读(4073)  评论(2编辑  收藏  举报