python - 为 Bert 导入 BertModelLayer

Question

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

from bert import BertModelLayer

BertModelLayer 的正确安装是什么？

标签： pythonnlpbert-language-model

Answer 1

安装命令为：

pip install bert-for-tf2

根据PyPi 文档

那么你可以使用：

from bert import BertModelLayer

我假设您正在使用 tensorflow-2