ZhuiyiTechnology / simbert

a bert for retrieval and generation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

引用的bert4keras会报错

myinvoke opened this issue · comments

AttributeError: module 'keras.utils' has no attribute 'get_custom_objects'

什么版本tf和keras

commented

我也会遇到这个错,tf2.6 keras 2.6

暂不支持这么高的版本。

推荐使用tf 1.15,以及真正的keras(keras<=2.3.1),大于此版本的keras都是假的keras(披着keras皮的tf.keras)

commented

暂不支持这么高的版本。

推荐使用tf 1.15,以及真正的keras(keras<=2.3.1),大于此版本的keras都是假的keras(披着keras皮的tf.keras)

好吧,线上的cuda版本比较高

commented

cuda11也可以用tf 1.15。https://github.com/NVIDIA/tensorflow

好的,感谢,我试试