使用LIME进行BERT变压器可视化会导致内存错误

Question

情况：我目前正在可视化我在LIME package之后使用this tutorial构建的拥抱变压器机器学习模型的结果。

并发症：在创建LIME explainer 对象之前，我的代码已设置并运行良好。这时我遇到了内存错误。

问题：我做错了什么？为什么我遇到内存错误？

代码：这是我的代码（您应该可以将其复制粘贴到google colab中并运行整个程序）

logicalSize

Answer 1

最终通过按照GitHub帖子的方式重新实现来解决此问题： https://github.com/marcotcr/lime/issues/409

我的代码现在与上述代码有很大不同-如果遇到类似问题，请向GitHub帖子寻求指导可能很有道理。