Tensorflow - сохранить .pb из оценщика с фиксированным размером партии 1 - PullRequest
0 голосов
/ 25 сентября 2018

Я хотел бы спросить, есть ли способ сохранить файл pb в фиксированном размере пакета 1 для производства через Estimator.

Я могу сделать это со стилем сеанса, поскольку он просто объявляет другой сеанс с размером пакета 1.

Как сохранить модель в файле `.pb` с помощью Estimator в TensorFlow?

Здесь приведены примеры того, как сохранить контрольную точку оценки в файл pb.

Я могу сохранить двоичный файл pb с фиксированным размером пакета 1 и загрузить его с другой платформы, такой как Java tenorflow.

Есть две проблемы, которые, по моему мнению, возникают.

1,сохранить двоичный файл с определенным именем слоя, но не знаю, как сохранить размер пакета 1.

2.Когда я использую оценщик, я не могу определить входное имя, так как я использую tf.data только для оценщика (по крайней мере, tenorflow предложил это сделать).

Например, я не уверен, как определить имя вГрафик с использованием tf.data, например:

предложение = tf.placeholder ([batch_size, None, embedding_size], name = "ать предложение_input ")

Спасибо вам, ребятапомощь.

...