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])
相关推荐
cn分享汇2 分钟前
2026 企业 AI 平台:低代码融合赛道
人工智能·低代码·rxjava
2501_942389553 分钟前
为支撑持续攀升的AI资本开支
人工智能·oracle·hbase·database·storm
lkforce9 分钟前
Transformer架构下的详细计算流程模拟,精确到数字(单层单头)
人工智能·深度学习·ai·transformer
summer_du10 分钟前
Qdrant
人工智能·python
doiito(Do It Together)12 分钟前
【AI 应用】从“外国人味”到地道中文:kokoroi-rs v0.1.2 架构升级深度解析
人工智能
GuWenyue17 分钟前
AI对话打字机效果卡顿崩溃?1套Vue3流式代码搞定逐Token输出,彻底解决断包报错
人工智能
叶总没有会17 分钟前
3.2 构建AI智能体项目扩展知识
java·数据库·人工智能·spring·ai
易知微EasyV数据可视化18 分钟前
工业制造如何实现柔性规划?数字孪生技术助力产线布局编排与方案推演
人工智能·经验分享·制造·数字孪生·可视化·三维建模
pla8888888819 分钟前
热词驱动,智辨声纹——从ASR热词到说话人日志的尝试:海光DCU环境部署FunASR热词语音识别系统(说话人日志已集成,含完整代码)
人工智能·语音识别
编程牛马姐24 分钟前
并发、多线程和HTTP连接之间有什么关系?
人工智能