我正在尝试保存一个包含关注层(tf.keras.layers.attention)的tensorflow 2模型。
我尝试使用model.save()和model.to_json(),都产生以下错误:
NotImplementedError:__init__
中带有参数的图层必须覆盖get_config
。
从模型中删除关注层时没有错误。
我应该澄清一下,我正在Tensorflow 2中使用Keras来处理会话和模型层/编译/训练。我能够将训练有素的keras模型保存为递归网络而无需注意。但是,当我使用keras.layers.Attention时,该模型仍可以编译和训练,但使用上述命令将无法保存。