根据有关此 API 的文档和众多 SO 帖子,必须使用
saver = tf.train.Saver(...variables...)
我想知道是否有任何方法可以自动填充(...variables...)
,而无需明确列出我的网络中使用的所有变量和操作。
现在我的网络只有两层,所以这不是一个很大的麻烦,但感觉就像必须手动列出所有变量一样彻头彻尾的石器时代。
tf.train.Saver
的默认初始值设定项将创建一个实例,用于保存/还原图形中的所有可保存对象,其中通常包括所有模型变量。因此,您应该能够编写:
saver = tf.train.Saver()
。并获得所需的效果,而不会有太多麻烦。