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])
相关推荐
AC赳赳老秦几秒前
2026 AI原生工具链升级:DeepSeek与AI原生IDE深度联动,重塑开发效率新高度
大数据·ide·人工智能·web3·去中心化·ai-native·deepseek
virtaitech2 分钟前
GPU池化技术走向大众:趋动科技推出永久免费OrionX社区版
人工智能·科技·gpu算力·算力·云平台
LoserChaser2 分钟前
大语言模型入门-基本概念
人工智能·ai·语言模型·自然语言处理
格林威3 分钟前
工业相机图像高速存储(C#版):先存内存,后批量转存方法,附海康相机实战代码!
开发语言·人工智能·数码相机·计算机视觉·c#·视觉检测·海康相机
ZTLJQ3 分钟前
深入理解GAN:生成对抗网络的原理与实战应用
人工智能·神经网络·生成对抗网络
EQUINOX14 分钟前
计算机视觉,图像增广,微调,R-CNN,SSD,YOLO
人工智能·计算机视觉
xcbeyond7 分钟前
AI 时代生存指南:入门路径 × 工具选择 × 应用场景 × 高频 Prompt 模板
人工智能·prompt
夫唯不争,故无尤也11 分钟前
PostgreSQL + SQLAlchemy 快速搭一个能跑的 Agent 后端数据层
数据库·人工智能·postgresql·agent
Monly2115 分钟前
大模型:LangChain调用大语言模型
人工智能·语言模型·langchain