Python/Tensorflow:网络保护程序



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

saver = tf.train.Saver(...variables...)

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

现在我的网络只有两层,所以这不是一个很大的麻烦,但感觉就像必须手动列出所有变量一样彻头彻尾的石器时代。

tf.train.Saver的默认初始值设定项将创建一个实例,用于保存/还原图形中的所有可保存对象,其中通常包括所有模型变量。因此,您应该能够编写:

saver = tf.train.Saver()

。并获得所需的效果,而不会有太多麻烦。

最新更新