问题预测
复现代码时,由于transformer或者其他包的版本和代码作者(年代久远)的版本不相同,出现报错。
问题解决
考虑到在遇到类似版本问题时,想要“完全复现源代码的环境”这一解决方案过于理想化。
1、将transformers更新至目前最新版。
2、更改bert返回数据类型
#源代码
self.bert = BertModel.from_pretrained('bert-base-chinese')
#改为
self.bert = BertModel.from_pretrained('bert-base-chinese', return_dict=False)