keras - ValueError：未知优化器：来自 Keras 的 load_model 的 RAdam

Question

当我运行以下单元格时，我得到了“ValueError：未知优化器：RAdam” 。

from keras_bert import get_custom_objects
bert_model = load_model("model.h5", custom_objects=get_custom_objects())

如果我将其更改为：

from keras_bert import get_custom_objects
bert_model = load_model("model.h5", custom_objects={'RAdam': RAdam})

我有“ValueError：未知层：TokenEmbedding”

我在用

from keras_radam import RAdam
print(tf.__version__)
print(keras.__version__)
1.15.0
2.2.4

对此有何建议？

标签： kerasneural-networktransfer-learning

Answer 1

也许尝试将“TokenEmbedding”导入 custom_objects 字典？