tensorflow SavedModel - 如何迭代保存

时间:2018-01-31 11:11:08

标签: tensorflow

我正在采用新的SavedModel API,因为我已经了解它是"未来"应优先于tf.train.Saver()

我想要完成的是每个N个批次保存一个模型,我想保留最多20个已保存的模型。 显然我可以自己监控,但如果tf.train.Saver()可以达到这个目的,我相信SavedModel也有类似的方法。 我还希望能够在我想要时明确保存模型并将它们全部命名为model-001model-002 ....

tf.train.Saver()可以实现这一切 - 我可以定义max_to_keepglobal_step来保存每个纪元左右,我可以在需要时明确保存。

这可以通过SavedModel API实现吗?如果是这样的话?一个例子将不胜感激。

可以合理地假设signature def map将等于所有模型。

0 个答案:

没有答案