RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26
在使用pytorch训练模型时遇到上面的问题,查了好久也不知道啥原因,最后在别人的帖子里找到类似的问题,就试试
解决办法:检查labels中是否有负值,如果有改成非负整数就可以了
在使用pytorch训练模型时遇到上面的问题,查了好久也不知道啥原因,最后在别人的帖子里找到类似的问题,就试试
解决办法:检查labels中是否有负值,如果有改成非负整数就可以了