我正在尝试使用 Tensorflow 的 Java API 在 Android 中运行 Tensorflow。我想知道如何在创建 Session 对象时设置 Session 配置。
在 Python API 中,您可以使用 session 构造函数 ( link ) 的“config”选项来执行此操作。 但在 Java API 中,我在 Session 构造函数 ( link ) 中没有看到这样的选项。
最佳答案
最近的一项更改使这成为可能 ( link )。您可以提供序列化的 ConfigProto Protocol Buffer 并将其提供给 Session 构造函数。
但是,由于这是最近的更改,因此您必须从源代码构建 Java API 或等待下一个版本 (TensorFlow 1.1)
关于java - 如何在 Tensorflow Java API 中设置 session 配置?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42870978/