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])
相关推荐
华玥作者6 小时前
[特殊字符] VitePress 对接 Algolia AI 问答(DocSearch + AI Search)完整实战(下)
前端·人工智能·ai
AAD555888996 小时前
YOLO11-EfficientRepBiPAN载重汽车轮胎热成像检测与分类_3
人工智能·分类·数据挖掘
王建文go6 小时前
RAG(宠物健康AI)
人工智能·宠物·rag
巫婆理发2226 小时前
循环序列模型
深度学习·神经网络
ALINX技术博客6 小时前
【202601芯动态】全球 FPGA 异构热潮,ALINX 高性能异构新品预告
人工智能·fpga开发·gpu算力·fpga
易营宝6 小时前
多语言网站建设避坑指南:既要“数据同步”,又能“按市场个性化”,别踩这 5 个坑
大数据·人工智能
春日见6 小时前
vscode代码无法跳转
大数据·人工智能·深度学习·elasticsearch·搜索引擎
Drgfd7 小时前
真智能 vs 伪智能:天选 WE H7 Lite 用 AI 人脸识别 + 呼吸灯带,重新定义智能化充电桩
人工智能·智能充电桩·家用充电桩·充电桩推荐
萤丰信息7 小时前
AI 筑基・生态共荣:智慧园区的价值重构与未来新途
大数据·运维·人工智能·科技·智慧城市·智慧园区
盖雅工场7 小时前
排班+成本双管控,餐饮零售精细化运营破局
人工智能·零售餐饮·ai智能排班