我正在采用新的SavedModel
API,因为我已经了解它是"未来"应优先于tf.train.Saver()
。
我想要完成的是每个N
个批次保存一个模型,我想保留最多20个已保存的模型。
显然我可以自己监控,但如果tf.train.Saver()
可以达到这个目的,我相信SavedModel
也有类似的方法。
我还希望能够在我想要时明确保存模型并将它们全部命名为model-001
,model-002
....
tf.train.Saver()
可以实现这一切 - 我可以定义max_to_keep
和global_step
来保存每个纪元左右,我可以在需要时明确保存。
这可以通过SavedModel
API实现吗?如果是这样的话?一个例子将不胜感激。
可以合理地假设signature def map
将等于所有模型。