Python / Tensorflow:网络保护程序

时间:2017-02-06 21:16:16

标签: python tensorflow neural-network

根据有关此API的文档和大量SO帖子,必须使用

创建保护程序对象
saver = tf.train.Saver(...variables...)

我想知道是否有任何方法可以自动填充(...variables...),而无需明确列出我的网络中使用的所有变量和操作。

现在我的网络只有两个层次,所以这不是一个很大的麻烦,但感觉非常石头时代喜欢手动列出所有变量。

1 个答案:

答案 0 :(得分:1)

default initializer for tf.train.Saver将创建一个实例,用于保存/恢复图表中的所有可保存对象,通常包括所有模型变量。因此,您应该能够写下:

saver = tf.train.Saver()

...并且没有太多麻烦就能获得理想的效果。