RuntimeError: CUDA error: device-side assert triggered的解决方法(个人的方法)

RuntimeError: CUDA error: device-side assert triggered

说明:在用pytorch运行模型时,遇到错误,网上查了一堆解决方案,基本上是说标签溢出问题。由于我的数据输入是一个网络,因此输入中包含了网络的节点总数。假设网络的节点总数为14048个,那么我在输入时设置self.nodes = 14048时就遇到了这个问题
问题:因为需要获取每个节点的嵌入,因此嵌入查找表的维度应该是要等同于节点数的,但是输入设置为14048时会报错,那么将其增加1,将输入设置为self.nodes = 14049就能正常运行
或许原因就是嵌入矩阵的维度与输入节点的数量不匹配导致的,可以试试将节点数量+1再运行。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值