Выполнение двух графиков тензорного потока за один раз

Question

Я подготовил 2 модели для немного разных задач OCR, и я хотел бы запускать их оба одновременно в производственной системе.

У меня есть класс, который запускает сеанс и загружает сохраненную модель. Проблема, с которой я сталкиваюсь, состоит в том, что у двух моделей есть переменные, сохраненные с тем же именем, поэтому они конфликтуют, когда я пытаюсь их загрузить.

Есть ли способ запускать их каждый в своей собственной сессии или на их собственном графике ?

Answer 1

Простейшим вариантом было бы построить две модели на разных графиках и начать отдельные сеансы для каждого графика. Однако каждый сеанс имеет свои устройства (процессор и, возможно, графический процессор), поэтому у вас будут отдельные пулы потоков для каждой модели, и это может привести к неоптимальному планированию. Для обеспечения хорошей производительности потребуется тщательное использование конфигурационных параметров tf.Session.

В качестве альтернативы вы можете объединить две модели в одном графике и использовать один сеанс. Как вы заметили, переменные для двух моделей обязательно будут иметь разные имена. Поэтому для выполнения этой работы вам нужно будет указать явное сопоставление «имя-к-Variable» при загрузке в модель construct the tf.Saver.