用了torch里面的nn.Embedding层,报错:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
/pytorch/aten/src/ATen/native/cuda/Indexing.cu:658: indexSelectLargeIndex: block: [234,0,0], thread: [102,0,0] Assertion srcIndex < srcSelectDimSize failed.
检查发现是输入nn.Embedding的id越界了。
【debug】 indexSelectLargeIndex: block: [x,0,0], thread: [x,0,0] Assertion `srcIndex < 解决
最新推荐文章于 2024-06-18 14:48:52 发布