词表大小越界错误 indexSelectLargeIndex

报错信息

在pytorch的代码中,可能会出现如下错误,
it/s]/pytorch/aten/src/ATen/native/cuda/Indexing.cu:702: indexSelectLargeIndex: block: [394,0,0], thread: [64,0,0] Assertion srcIndex < srcSelectDimSize failed.

问题

仔细阅读这个报错信息,就知道大概率是下标越界错误。比如我一个词表的大小是200,但是我在访问Embedding的时候,id大小超过了200,就会报这种错误。

常见于 index_selct()
在这里插入图片描述
等语句下报错。

猜你喜欢

转载自blog.csdn.net/liu16659/article/details/127331329
今日推荐