将变量从一个TensorFlow图复制到另一个

Question

我有两个张量流图。一个用于培训，另一个用于评估。他们共享许多变量名称。当我评估模型时，我想将所有变量值从列车图复制到测试图。显然，我可以通过tf.train.Saver来做到这一点，但这个解决方案对我来说似乎不太合适，尤其是我们必须使用磁盘这一事实。

Answer 1

当你谈到多个图表时，我认为你的意思是：

g1 = tf.Graph()
with g1.as_default():
  # add your stuff

g2 = tf.Graph()
with g2.as_default():
  # add other stuff

如果这是正确的，那么你确定你真的需要两张图吗？您是否有一个由两个连接组件组成的图表？

使用多个图表为discouraged（第47页），因为：

  

  
• 多个图表需要多个会话，默认情况下每个会尝试使用所有可用资源
  
• 不能在它们之间传递数据而不通过python / numpy传递数据，这在分布式传输中不起作用
  
• 最好在一个图表中包含断开连接的子图
  

这也为您提供了如何在非分布式设置中传递变量的解决方案。