使用bert预训练模型做文本分类时，跑了一个数据集时报了一个KeyError:的错，KeyError后面的内容是自己数据集第一个样本的文本内容，错误代码如图所示：
https://github.com/terrifyzhao/bert-utils/issues/10
请问解决了吗？