为Bert导入BertModelLayer

Question

我想为Bert导入BertModelLayer，但我一直收到错误的ImportError: cannot import name 'BertModelLayer' from 'bert'，我已经安装了bert，但仍然有相同的错误。

from bert import BertModelLayer

BertModelLayer的正确安装是什么？

Answer 1

安装命令为：

pip install bert-for-tf2

根据PyPi documentation

然后，您可以使用：

from bert import BertModelLayer

我假设您使用的是tensorflow-2

Answer 2

也许你正在寻找一个不同的包？

请尝试安装bert-tensorflow，因为bert不是包的名称。