tensorflow - 一次运行两个 tensorflow 图

Question

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

我有一个启动 session 并加载保存的模型的类。我遇到的问题是这 2 个模型的变量以相同的名称保存，因此当我尝试加载它们时它们会发生冲突。

有没有办法在各自的 session 或各自的图表上运行它们？

Answer 1

最简单的选择是在不同的图表中构建两个模型，并为每个图表启动单独的 session 。但是，每个 session 都拥有自己的设备(CPU 和可能的 GPU)，因此每个模型都有单独的线程池，这可能会导致次优调度行为。小心使用 tf.Session 配置选项对于获得良好的性能是必要的。

或者，您可以将两个模型组合在同一个图中并使用单个 session 。正如您所指出的，两个模型的变量必然具有不同的名称。因此，要实现此功能，您需要在 construct the tf.Saver 时提供显式名称到 Variable 的映射。用于加载模型。