pytorch错误记录

最新推荐文章于 2024-09-17 20:45:29 发布

Memo56

最新推荐文章于 2024-09-17 20:45:29 发布

阅读量262

点赞数

分类专栏： NLP代码错误记录文章标签： pytorch nlp

本文链接：https://blog.csdn.net/Memo56/article/details/118992845

版权

NLP代码错误记录专栏收录该内容

1 篇文章 0 订阅

订阅专栏

博客介绍了在使用BERT等模型时遇到的CUDA错误和索引范围错误。错误原因是vocab.txt文件大小变化导致模型计算size不匹配。解决方案包括参考相关链接，检查并修正vocab.txt文件，确保模型前向传播计算尺寸正确。

摘要由CSDN通过智能技术生成

问题1：

Assertion srcIndex < srcSelectDimSize failed.
RuntimeError: CUDA error: device-side assert triggered

解决过程：

第一步：参考链接：https://www.pythonheidong.com/blog/article/586782/6183ad3e3b8b8e788993/
第二步：出现错误：RuntimeError: index out of range: Tried to access index 30522 out of table with 30521 rows.
bert等模型分词加入自定义字符造成的vocab.txt文件大小变化，导致模型前向传播计算size不匹配。
参考链接：https://blog.csdn.net/Mr_WangYC/article/details/106718148