【debug】 indexSelectLargeIndex: block: [x,0,0], thread: [x,0,0] Assertion `srcIndex < 解决

用了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越界了。

你可能感兴趣的:(pytorch,pytorch,深度学习,自然语言处理)