从内存加载TensorFlow模型

时间:2018-11-13 09:20:02

标签: java python tensorflow

我正在用Java运行TensorFlow推理,并且我想从内存中加载模型(因为我已经在内存中保存了它)。

现在,我正在使用SavedModelBundle.load()方法(https://www.tensorflow.org/api_docs/java/reference/org/tensorflow/SavedModelBundle.html#load(java.lang.String,%20java.lang.String...)),该方法要求模型位于 exportDir 中的磁盘上。

是否具有从内存(InputStream或byte [])读取模型的功能?

我正在使用TensorFlow for Java 1.8.0,但可以使用最新的稳定器。使用Python解决方案也是可行的。

谢谢。

0 个答案:

没有答案