根据有关此API的文档和大量SO帖子,必须使用
创建保护程序对象saver = tf.train.Saver(...variables...)
我想知道是否有任何方法可以自动填充(...variables...)
,而无需明确列出我的网络中使用的所有变量和操作。
现在我的网络只有两个层次,所以这不是一个很大的麻烦,但感觉非常石头时代喜欢手动列出所有变量。
答案 0 :(得分:1)
default initializer for tf.train.Saver
将创建一个实例,用于保存/恢复图表中的所有可保存对象,通常包括所有模型变量。因此,您应该能够写下:
saver = tf.train.Saver()
...并且没有太多麻烦就能获得理想的效果。