Развернуть модель на мл-движке, экспортировать с помощью tf.train.Saver()

Я хочу развернуть модель в новой версии Google ML Engine. Раньше с помощью Google ML я мог экспортировать свою обученную модель, создавая tf.train.Saver(), сохраняя модель с помощью saver.save(session, output).

Пока мне не удалось выяснить, можно ли экспортированную таким образом модель развернуть на ml-engine, иначе я должен следовать процедуре обучения, описанной здесь и создайте новый пакет тренера и обязательно обучите мою модель с помощью ml-engine.

Могу ли я по-прежнему использовать tf.train.Saver() для получения модели, которую буду развертывать на ml-engine?


google-cloud-platform google-cloud-ml-engine
person EffePi    schedule 21.03.2017    source источник

Ответы (1)


arrow_upward
4
arrow_downward

tf.train.Saver() создает только контрольную точку.

Cloud ML Engine использует SavedModel, созданный из следующих API: https://www.tensorflow.org/versions/master/api_docs/python/tf/saved_model?hl=bn

Сохраненная модель — это контрольная точка + сериализованный protobuf, содержащий одно или несколько определений графа + набор подписей, объявляющих входные и выходные данные графа/модели + дополнительные файлы активов, если применимо, чтобы все это можно было использовать во время обслуживания.

Предлагаю рассмотреть пару примеров:

  1. Образец переписи — https://github.com/GoogleCloudPlatform/cloudml-samples/blob/master/census/tensorflowcore/trainer/task.py#L334

  2. И мой собственный код образца/библиотеки - https://github.com/TensorLab/tensorfx/blob/master/src/training/_hooks.py#L208, который обращается к https://github.com/TensorLab/tensorfx/blob/master/src./prediction/_model.py#L66, чтобы продемонстрировать, как использовать контрольную точку, загрузите ее в сеанс, а затем создайте сохраненную модель.

Надеемся, что эти указатели помогут адаптировать ваш существующий код для создания модели, которая теперь создает SavedModel.

Я думаю, вы также задали другой подобный вопрос, чтобы преобразовать ранее экспортированную модель, и я приведу ссылку на него здесь для полноты для всех остальных: Развертывание переобученной начальной модели SavedModel в движок google cloud ml

person Nikhil Kothari    schedule 27.03.2017