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])
相关推荐
盖伦发发5 分钟前
RAG 能跑≠能用:用 EDD 把 Eval 做成基础设施 (附源码)
人工智能·后端·python·功能测试
XTurnV00716 分钟前
一文讲明白DeepSeek Harness 是什么,怎么用
人工智能
Robot_Nav17 分钟前
前沿 | DeepSeek Harness 完整指南:架构原理、Windows/Linux 安装、插件生态与实战配置
人工智能·ai编辑器·deepseekharness
数据智研17 分钟前
【数据分享】全国农产品成本收益资料汇编(1953-2025)
大数据·人工智能·信息可视化·数据分析
daad77720 分钟前
802.11 前导码与 STF 深度解析(含检测算法与 5G 对比
人工智能·算法·5g·wifi·802.11·前导码
fail_to_code23 分钟前
感受deepseek-harness那极端的工程纪律性
人工智能
咖啡星人k24 分钟前
Vibe Coding 实战:用 MonkeyCode 一个下午做出可玩的小游戏
人工智能
正经教主28 分钟前
AI提示词工程(进阶)第11课:结构化输出与格式化控制
人工智能
GGBond今天继续上班29 分钟前
给 DeepSeek Harness 写了个生图插件,补上了原生对话生图能力
人工智能·github·deepseek
AI刀刀31 分钟前
Kimi 文档导出格式错乱、排版丢失、导出报错?AI 导出鸭一键智能适配,稳定输出规范 Word、PDF,高效解决各类导出难题
人工智能·pdf·word·ai导出鸭