我正在使用此笔记本进行使用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'
非常感谢您的帮助或解释。
谢谢!