如何在Tensorflow Java API中设置Session配置?

时间:2017-03-18 05:47:57

标签: java tensorflow deep-learning

我尝试使用Tensorflow的Java API在Android中运行Tensorflow。我想知道如何在创建Session对象时设置Session配置。

在Python API中,您可以使用" config" Session构造函数的选项(link)。 但是在Java API中,我在Session构造函数(link)中看不到这样的选项。

1 个答案:

答案 0 :(得分:2)

最近的一项改变使这成为可能(link)。您可以提供序列化的ConfigProto协议缓冲区并将其提供给Session构造函数。

但是,由于这是最近的更改,您必须从源代码构建Java API或等待下一个版本(TensorFlow 1.1)