今天使用 stanfordnlp 时发生报错,“index_select_out_cuda_impl” not implemented for ‘float’。
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/06139fe200394730ff2192ceb6e90d2d.png)
解决方法
找到文件 beam.by,我的电脑中文件的位置在 “D:\softwares\anaconda3\Lib\site-packages\stanfordnlp\models\common\beam.by”,然后找到下面的代码:
prevK = bestScoresId / numWords
然后修改成:
prevK = bestScoresId // numWords
重新执行上面的代码,没报错。
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/1d032c4b4e8f64eb678d8afcfec410f7.png)
问题解决。
真开心!!