0
Как мы знаем, демо-версия android demorflow является многопоточной. Итак, как я могу изменить его на однопоточный режим?Как установить номер потока при запуске образца тензорного потока на Android?
A
ответ
0
Невозможно сделать это с Java прямо сейчас. Значение для внутриоперационного параллелизма автоматически выбирается в local_device.cc, соответствующем количеству ядер, обнаруженных на вашем устройстве. Если вы просто хотите протестировать, вы можете перезаписать его там.
Интерпольный параллелизм в настоящее время всегда 1 на мобильном телефоне - только один запуск работает одновременно.
В общем случае внутренний код внутри и интерполятор может быть установлен при создании объекта Session. например:
tensorflow::SessionOptions options;
tensorflow::ConfigProto& config = options.config;
config.set_intra_op_parallelism_threads(num_threads);
Session session = tensorflow::NewSession(options);
Это будет отменять автоматически выбранное значение. См. Пример benchmark_model.cc.
Если вы хотите установить это на Java, вам нужно добавить параметр в TensorFlowInferenceInterface.initializeTensorFlow().
В настоящее время мы переходим к API Java, поэтому будет легче получить доступ к SessionOptions для установки значения.
+0
Я также нашел это, в любом случае, благодаря вам. –
Пожалуйста, избегайте задавать такие вопросы здесь ... быть конкретным программным обеспечением – Akshay
Мы действительно просим, чтобы разработчики задавали вопросы использования TensorFlow в stackoverflow с тегом TensorFlow, чтобы мы могли сохранить нашу очередь на выпуск github доступным элементам. –