踩坑记:RuntimeError: CUDA error: device-side assert triggered

GAT分类的时候,遇到

RuntimeError: CUDA error: device-side assert triggered

原因是数据的类别和num_classes不一致造成的。
比如我的数据总共有10类,但是类别数是从1-10,而一般的分类程序是从0-9,所以会出现这个问题,将数据类别改成从0开始就好。

猜你喜欢

转载自blog.csdn.net/weixin_40313940/article/details/107003027