RuntimeError: CUDA error: device-side assert triggered

本文主要是介绍RuntimeError: CUDA error: device-side assert triggered，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

背景：

        使用SAGEConv卷积层的图神经网络，网络架构如下

原因：

        我在卷积层之前改变了特征矩阵的维度，原本为[172,1,32] 现在改为了 [172,2,32]。导致了特征矩阵x在进行 “x = x.squeeze(1)” 操作时并没有将第二向量值去除（本矩阵中代表着未编码特征，需要在编码后删除）。导致出错。

解决方法：

        使用reshape代替x = x.squeeze(1)。

x = x.reshape(x.shape[0], self.embed_dim*self.feature_dim)

        其中

        self.embed_dim为编码特征维度，如：你想将特征编码为128、64、32位都可以。

        feature_dim特征维度，如：一个节点有几个特征点，[123,1234]就是两个特征点。

         最后全连接层也需要改变shape，以链接下一次卷积操作。

x = torch.sigmoid(self.lin3(x))  # batch个结果
x = x.reshape(x.shape[0])

 

这篇关于RuntimeError: CUDA error: device-side assert triggered的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---