一次运行两个张量流图

Question

我已针对略有不同的OCR任务培训了2个模型，我想在生产系统中同时运行它们。

我有一个启动会话并加载已保存模型的类。我遇到的问题是2个模型的变量保存了相同的名称，所以当我尝试加载它们时它们是冲突的。

有没有办法在他们自己的会话或他们自己的图表中运行它们？

Answer 1

最简单的选择是在不同的图形中构建两个模型，并为每个图形启动单独的会话。但是，每个会话都拥有其设备（CPU和可能的GPU），因此每个模型都有单独的线程池，这可能会导致次优的调度行为。为了获得良好的性能，必须小心使用glOrtho配置选项。

或者，您可以将两个模型组合在同一个图表中并使用单个会话。正如您所指出的，两个模型的变量必然具有不同的名称。因此，为了使这项工作成为可能，您需要在construct the tf.Saver加载模型时提供明确的名称到tf.Session映射。