- 博客(1)
- 收藏
- 关注
原创 cpu报错IndexError: index out of range in self,gpu报错Assertion `srcIndex < srcSelectDimSize` failed.
经过仔细查找,发现是embedding层src_vocab_size指的输入序列中不同单词的个数,我忘记算上padding了,所以我要+1,就成功运行啦。其实这个只是gpu错误显示不全,所以显示出来这个。然后换cpu跑,就会显示出来详细错误。
2024-01-08 22:15:00 406
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人