如何在Android上運行tensorflow示例時設置線程號？

Question

正如我們所知，tensorflow的android演示是多線程的。那麼我怎樣才能把它改成單線程模式呢？

Answer 1

現在沒有辦法從Java那裏做到這一點。在local_device.cc中自動選取符合設備上檢測到的內核數量的內部操作並行性值。如果你只是想測試，你可以覆蓋那裏的價值。

手機上的並行操作當前總是爲1 - 一次只能運行一個操作。

更一般地在本機代碼中創建Session對象時，可以設置intra和inter-parallism。例如：

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()。

不過，我們很快就會切換到Java API，因此訪問SessionOptions來設置值會更容易。