Явный сеанс тензорного потока дает ошибку выборки в Tensorflow/nmt

Этот вид tf.session работает нормально:

with tf.Session(graph=self.infer_model.graph, config=utils.get_config_proto()) as sess:
          loaded_infer_model = model_helper.load_model(self.infer_model.model, self.ckpt, sess, "infer")

Но я должен сохранить постоянный сеанс для повторного использования. Поэтому вместо создания tf.session с помощью оператора with я создал under:

sess = tf.Session(
            graph=infer_model.graph, config=utils.get_config_proto())
loaded_infer_model = model_helper.load_model(
              infer_model.model, ckpt, sess, "infer")

Но это дает следующую ошибку (в model_helper.load_model): Может кто-нибудь предложить, как загрузить явный сеанс, который можно использовать повторно?

Файл "/home/pksingh/sans/app/nmt/model_helper.py", строка 444, в файле load_model session.run(tf.tables_initializer()) Файл "/usr/local/lib/python2.7/dist-packages/tenorflow/python/client/session.py", строка 889, в прогоне run_metadata_ptr) Файл"/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", строка 1103, в _run self._graph, fetches, feed_dict_tensor, feed_handles=feed_handles) Файл "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", строка 414, в init self._fetch_mapper = _FetchMapper.for_fetch(извлекает) файл "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", строка 242, в for_fetch возвращает _ElementFetchMapper(извлекает, сжатие_fn) файл "/ usr /local/lib/python2.7/dist-packages/tensorflow/python/client/session.py ", строка 278, в init 'Tensor. (%s)' % (fetch, str(e))) ValueError: Аргумент Fetch нельзя интерпретировать как Tensor. (Имя операции: "init_all_tables" op: "NoOp" не является элементом этого графа.)

1 ответ

Решение

Лучший вариант для достижения того же - использовать интерактивный сеанс. Вы можете инициализировать интерактивный сеанс следующим образом:

sess = tf.InteractiveSession()

Посетите эту ссылку для более подробной информации.

Другие вопросы по тегам