使用单个会话在tensorflow中执行多个模型

时间:2018-02-23 11:23:03

标签: tensorflow parallel-processing neural-network

我试图在张量流中并行运行几种神经网络模型,每种模型都独立于其余模型。有必要为我使用tensorflow启动的每个执行创建一个会话,或者我可以为每个模型重用相同的会话?谢谢

1 个答案:

答案 0 :(得分:0)

会话链接到特定的Tensorflow Graph实例。如果您想为所有人设置一个会话,则需要将所有模型放在同一个图表中。这可能会导致你为张量命名问题,而IMO通常是一个坏主意(你应该保持彼此不相关的东西)。

将所有内容放在同一个图表中也会提高模型的资源需求(即使只运行子图,也总是加载所有内容),这是在独立图中拆分内容的另一个原因。使用独立的图表,您将不得不使用多个会话。