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])
相关推荐
你住过的屋檐1 分钟前
【claude code】claude code在windows下安装使用教程,以及在idea中使用claude code自动AI写代码
人工智能·windows·intellij-idea
m0_图灵灵2 分钟前
吴恩达《深度学习》之深度剖析Batch Norm 作用机制的本质
人工智能·深度学习·batch
InsightCore2 分钟前
iOS 27改了一个15年的手势,苹果决定让你的肌肉记忆失效
人工智能
AI客栈2 分钟前
云原生存储与网络方案选型:从 CSI 到 CNI 的架构决策与落地实践
人工智能
柠檬味的Cat8 分钟前
GEO优化系统是什么?具体做什么,有什么用?
大数据·人工智能·aigc
用户276247978509 分钟前
上下文压缩,无脑调 LLM 总结是下策:我给 Agent 框架设计的三级压缩流水线
人工智能
2601_9494999410 分钟前
芯瑞科技800G硅光模块:为工业智能时代打造高速可靠的光互联底座
人工智能·科技·机器人
小博测试成长之路11 分钟前
行业日报 | 2026年6月12日:Claude新模型、鸿蒙开发者大会与AI工程化加速
人工智能·harmonyos
snpgroupcn13 分钟前
云转型过程中, 如何有效推进Rise with SAP项目实施
大数据·人工智能·云转型
小糖学代码14 分钟前
机器学习:9.贝叶斯分类器
人工智能·机器学习