ValueError：未知优化器：来自 Keras 的 load_model 的 RAdam答案

【问题标题】：ValueError: Unknown optimizer: RAdam from load_model of KerasValueError：未知优化器：来自 Keras 的 load_model 的 RAdam
【发布时间】：2020-11-07 22:37:47
【问题描述】：

我有 "ValueError: Unknown optimizer: 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: Unknown layer: TokenEmbedding"

我正在使用

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

对此有何建议？

【问题讨论】：

标签： keras neural-network transfer-learning

【解决方案1】：

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

【讨论】：