根据有关此API的文档和大量SO帖子，必须使用以下命令创建保护对象
saver = tf.train.Saver(...variables...)
我想知道是否有任何方法可以自动填充，(...variables...)而不必明确列出网络中使用的所有变量和操作。
现在，我的网络只有两层，所以这不是一个很大的麻烦，但是感觉就像是必须手动列出所有变量的石器时代。
的默认初始化程序tf.train.Saver将创建一个实例，该实例保存/恢复图形中的所有可保存对象，该对象通常包括所有模型变量。因此，您应该能够编写：
saver = tf.train.Saver()
…并获得期望的效果而没有太多麻烦。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句