运行环境:
tensorflow-gpu == 1.14.0
keras==2.2.4
keras-bert==0.69.0
使用keras-bert的加载bert预训练模型时报错:
原因分析:
这是keras-transformer版本问题,pip安装keras-bert==0.69.0时自动安装了keras-transformer == 0.38.0(pip list可以查看一下版本)
keras-transformer版本改成0.31.0就可以了
解决步骤:
1. 卸载原来的keras-transformer
pip uninstall keras-transformer
2.pip安装keras-transformer==0.31.0版本
pip install keras-transformer==0.31.0
3.重新运行代码就可以没有再报错了