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

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

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

猜你喜欢

转载自www.cnblogs.com/chuanyang/p/12803786.html