bojone/bert4keras

import build_transformer_model时报错

whats0mai0name opened this issue · 1 comments

whats0mai0name commented

大佬您好，我在尝试import build_transformer_model的时候报错了，报错提示‘ModuleNotFoundError: No module named 'keras.api'’，尝试使用pip install keras==2.9.0但是依然报错，请问这种情况应该怎么解决呢？系统的基本信息如下：

基本信息

  • 你使用的操作系统: 使用的是colab
  • 你使用的Python版本: 3.7
  • 你使用的Tensorflow版本: 2.9.0
  • 你使用的Keras版本: 2.3.1
  • 你使用的bert4keras版本: 0.11.4

核心代码

from bert4keras.models import build_transformer_model


### 输出信息
ModuleNotFoundError: No module named 'keras.api'
bojone commented

keras不超过2.3.1，tf不超过2.6，最好用tf 1.15