Как выполнить пример обслуживания TensorFlow mnist_export без использования bazel?

Я установил все необходимые компоненты для работы TensorFlow, как объяснено здесь

Когда я попытался запустить пример mnist по умолчанию со следующей строкой на моем компьютере с Ubuntu 16.04:

python  mnist_export.py --training_iteration=1000 --export_version=1 export_models

Я получаю следующую ошибку:

Traceback (most recent call last):
  File "mnist_export.py", line 40, in <module>
    from tensorflow_serving.example import mnist_input_data
ImportError: No module named tensorflow_serving.example

Этот код доступен здесь.

Однако, когда я использую bazel для сборки следующим образом, он отлично работает:

bazel-bin/tensorflow_serving/example/mnist_export --training_iteration=1000 --export_version=1 export_models

person Balaraman L    schedule 12.07.2017    source источник


Ответы (1)


Эта проблема была решена путем изменения следующей строки:

from tensorflow_serving.example import mnist_input_data

к этой строке:

from tensorflow.contrib.learn.python.learn.datasets import mnist as mnist_input_data

Эта проблема обсуждается здесь.

person Balaraman L    schedule 12.07.2017