簡體   English   中英

如何在Tensorflow Java API中設置會話配置?

[英]How to set Session config in Tensorflow Java API?

我正在嘗試使用Tensorflow的Java API在Android中運行Tensorflow。 我想知道在創建會話對象時如何設置會話配置。

在Python API中,您可以通過使用Session構造函數( link )的“ config”選項來實現。 但是在Java API中,我在Session構造函數( link )中看不到這樣的選項。

最近的更改使這成為可能( link )。 您可以提供序列化的ConfigProto協議緩沖區,並將其提供給Session構造函數。

但是,由於這是最近的更改,因此您必須從源代碼構建Java API或等待下一個版本(TensorFlow 1.1)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM