RuntimeError: CUDA error: device-side assert triggered的解决方法(个人的方法)
RuntimeError: CUDA error: device-side assert triggered说明:在用pytorch运行模型时,遇到错误,网上查了一堆解决方案,基本上是说标签溢出问题。由于我的数据输入是一个网络,因此输入中包含了网络的节点总数。假设网络的节点总数为14048个,那么我在输入时设置self.nodes = 14048时就遇到了这个问题问题:因为需要获取每个节点的嵌入,因此嵌入查找表的维度应该是要等同于节点数的,但是输入设置为14048时会报错,那么将其增加1,将输入设置为s
复制链接