天天看點

Assertion `srcIndex < srcSelectDimSize` failed.

  這裡具體問題具體分析,我是在使用BERT預訓練模型的時候遇到的這個錯誤,這個錯誤實際上是因為輸入的文本長度太長了,比如我用的BERT-base最長支援512位,而我輸入的長度大于了512,這個時候隻需要截斷輸入的文本,保持輸入的長度小于等于512,就可以正常運作了。

參考網址:

https://www.cnblogs.com/laozhanghahaha/p/13037595.html

繼續閱讀