Tensorflow: сохранение модели SavedModelBundle из Java

Question

Можно ли сохранить SavedModelBundle из Java API?

Мой пример использования:

1. Загрузить существующий SavedModelBundle
2. Назначитьновые значения для некоторых переменных (путем вызова assign-op, уже определенного на графике)
3. Сохраните модель еще раз

Шаги 1. и 2. Я могу сделать, но не могувыяснить, как сделать шаг 3.

Любой способ достичь этого

(Я хотел бы сделать это на Java вместо Python, потому что все мое приложение на Java (Scala), так чтопереход на Python добавит много сложностей.)

Answer 1

К сожалению, нет API для сохранения модели Tensorflow в средах JAVA.

По сути, в средах Java вы можете загрузить только существующую замороженную модель Tensorflow, в которой вы не можете добавлять новые переменные или изменятьзначение в переменных.

Шаг 2, вы сказали, что вы назначаете новые значения переменной java, а не переменной модели Tensorflow.

Во многих случаях строите модель с помощью Python и замораживайте модель (выходной файл - «***. pb» в общем случае), затем используется модель на Java.