Assertion `srcIndex < srcSelectDimSize` failed.

  这里具体问题具体分析,我是在使用BERT预训练模型的时候遇到的这个错误,这个错误实际上是因为输入的文本长度太长了,比如我用的BERT-base最长支持512位,而我输入的长度大于了512,这个时候只需要截断输入的文本,保持输入的长度小于等于512,就可以正常运行了。

参考网址:
https://www.cnblogs.com/laozhanghahaha/p/13037595.html

你可能感兴趣的:(深度学习,python,pytorch,Assertion,bert,srcIndex,GPT)