保存编码器解码器模型张量流

时间:2020-03-03 10:33:25

标签: python tensorflow tensorflow2.0

我正在使用此笔记本进行使用tensorflow 2.0的图像字幕模型:https://www.tensorflow.org/tutorials/text/image_captioning

请注意使用 encoder-decoder 模型。在笔记本中,模型会按一定的时间间隔保存为检查点。

我的问题是,例如,如何保存此模型以通过API进行投放?

我设法使用以下方法保存编码器模型:

encoder.save('encoder_model')

但是,当我尝试使用以下方法保存解码器模型时:

decoder.save('decoder_model')

我收到以下错误:

TypeError: call() missing 2 required positional arguments: 'features' and 'hidden'

非常感谢您的帮助或解释。

谢谢!

0 个答案:

没有答案