我尝试使用Tensorflow的Java API在Android中运行Tensorflow。我想知道如何在创建Session对象时设置Session配置。
在Python API中,您可以使用" config" Session构造函数的选项(link)。 但是在Java API中,我在Session构造函数(link)中看不到这样的选项。
答案 0 :(得分:2)
最近的一项改变使这成为可能(link)。您可以提供序列化的ConfigProto
协议缓冲区并将其提供给Session
构造函数。
但是,由于这是最近的更改,您必须从源代码构建Java API或等待下一个版本(TensorFlow 1.1)