如何使用关注层保存Tensorflow 2模型(NotImplementedError)

时间:2020-02-11 15:41:53

标签: python tensorflow keras tensorflow2.0

我正在尝试保存一个包含关注层(tf.keras.layers.attention)的tensorflow 2模型。

我尝试使用model.save()和model.to_json(),都产生以下错误:

NotImplementedError:__init__中带有参数的图层必须覆盖get_config

从模型中删除关注层时没有错误。

我应该澄清一下,我正在Tensorflow 2中使用Keras来处理会话和模型层/编译/训练。我能够将训练有素的keras模型保存为递归网络而无需注意。但是,当我使用keras.layers.Attention时,该模型仍可以编译和训练,但使用上述命令将无法保存。

0 个答案:

没有答案