报错:C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Indexing.cu:699: block: [9,0,0], CUDA

文章描述了在使用PyTorch进行CUDA操作时,遇到的关于Embedding词向量初始化错误,指出原因可能是词向量集合大小不足导致数据无法找到对应的向量,需检查word2index字典和词向量集合的大小关系。
摘要由CSDN通过智能技术生成

完整报错:

C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Indexing.cu:699: block: [9,0,0], thread: [0,0,0] Assertion `srcIndex < srcSelectDimSize` failed.


输出embedding后的词向量后报错:

 

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

 原因:初始化的Embedding词向量集合大小太小(必须要输入的要小),数据没有找到对应词向量,可查看word2index字典大小与词向量集合大小对比分析。
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值