使用pytorch写代码时候,报这个错误
…/aten/src/ATen/native/cuda/Indexing.cu:1146: indexSelectLargeIndex: block: [116,0,0], thread: [73,0,0] Assertion srcIndex < srcSelectDimSize` failed.
经查,我的是因为 nn.Embedding 的问题。因为我申请词表的范围没有包括填充的字符。所以遇到填充字符就会报错。
这个如果不处理,虽然能让程序继续往下运行,但是当经过cuda操作(例如 前向网络)就会报错。所以必须修复。