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])
相关推荐
算家计算11 分钟前
字节跳动开源Seed-OSS-36B:512K上下文,代理与长上下文基准新SOTA
人工智能·开源·资讯
THMAIL13 分钟前
大模型“知识”的外挂:RAG检索增强生成详解
人工智能
汀丶人工智能14 分钟前
AI Compass前沿速览:DINOv3-Meta视觉基础模型、DeepSeek-V3.1、Qwen-Image、Seed-OSS、CombatVLA-3D动
人工智能
范男18 分钟前
基于Pytochvideo训练自己的的视频分类模型
人工智能·pytorch·python·深度学习·计算机视觉·3d·视频
二向箔reverse25 分钟前
机器学习算法核心总结
人工智能·算法·机器学习
dundunmm1 小时前
【论文阅读】SIMBA: single-cell embedding along with features(2)
论文阅读·人工智能·embedding·生物信息·单细胞·多组学·细胞类型识别
金井PRATHAMA1 小时前
意象框架:连接感知与认知的统一信息结构分析——基于上古汉语同源词意义系统的词源学与认知语言学探索
人工智能·自然语言处理
聚客AI1 小时前
🧠深度解析模型压缩革命:减枝、量化、知识蒸馏
人工智能·深度学习·llm
SHIPKING3931 小时前
【机器学习&深度学习】Ollama、vLLM、LMDeploy对比:选择适合你的 LLM 推理框架
人工智能·深度学习·机器学习
zzywxc7872 小时前
AI 行业应用:金融、医疗、教育、制造业领域的落地案例与技术实现
android·前端·人工智能·chrome·金融·rxjava