Tensorflow:从Java保存SavedModelBundle模型

Question

是否可以从Java API保存SavedModelBundle？

我的用例是我想

1. 加载现有SavedModelBundle
2. 为某些变量分配新值(通过调用图中已定义的赋值操作(
3. 再次保存模型

步骤 1. 和 2.我可以做，但我不知道如何做第 3 步。

任何实现这一目标的方法

(我想用Java而不是Python来做这件事，因为我的整个应用程序都是Java(Scala(的，所以避开Python会增加很多复杂性。

Answer 1

不幸的是，没有API可以在JAVA环境中保存Tensorflow模型。

基本上，在Java环境中，您只能加载现有的冻结的Tensorflow模型，您无法添加新变量或修改变量中的值。

Step2，你说的是你给java变量分配新值，而不是Tensorflow模型的变量。

在许多情况下，使用 Python 构建模型并冻结模型(输出文件通常为"***.pb"(，然后在 Java 中使用模型。