当前位置:   article > 正文

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

runtimeerror: cuda error: device-side assert triggeredzenmejiejue

RuntimeError: CUDA error: device-side assert triggered

说明:在用pytorch运行模型时,遇到错误,网上查了一堆解决方案,基本上是说标签溢出问题。由于我的数据输入是一个网络,因此输入中包含了网络的节点总数。假设网络的节点总数为14048个,那么我在输入时设置self.nodes = 14048时就遇到了这个问题
问题:因为需要获取每个节点的嵌入,因此嵌入查找表的维度应该是要等同于节点数的,但是输入设置为14048时会报错,那么将其增加1,将输入设置为self.nodes = 14049就能正常运行
或许原因就是嵌入矩阵的维度与输入节点的数量不匹配导致的,可以试试将节点数量+1再运行。
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/879796
推荐阅读
相关标签
  

闽ICP备14008679号