Что такое формат экспорта SageMaker TensorFlow SavedModel? - PullRequest
Новая
       8

Что такое формат экспорта SageMaker TensorFlow SavedModel?

0 голосов
/ 21 декабря 2018

Я новичок в режиме сценариев AWS SageMaker TensorFlow 1.11.0.Некоторое время просматривал документацию и, похоже, не смог найти структуру папок для экспорта модели после обучения.Все, что я знаю, это то, что я предполагаю экспортировать в каталог, указанный в переменной env "SM_MODEL_DIR", и что формат - SavedModel.

Я использовал tf.saved_model.simple_save для экспорта в следующую структуру папок:

- model.tar.gz
-- saved_model.pb
-- variables
--- variables.index
--- variables.data-00000-of-00001

Но при развертывании я получаю ошибку: 

Traceback (most recent call last):
File "/sagemaker/serve.py", line 189, in <module>
ServiceManager().start()
File "/sagemaker/serve.py", line 163, in start
self._create_tfs_config()
File "/sagemaker/serve.py", line 53, in _create_tfs_config
raise ValueError('no SavedModel bundles found!')

любая помощь будет хорошей.Thx

1 Ответ

0 голосов
/ 03 января 2019

Вы видите эту ошибку, потому что служба TensorFlow Serving (которую SageMaker использует для размещения вашей модели) требует, чтобы SavedModel был вложен в подкаталог версии.

Добавьте каталог версии в вашу модель архива, чтобы сделать ееРабота.Пример: 

- model.tar.gz
-- 1
--- saved_model.pb
--- variables
---- variables.index
---- variables.data-00000-of-00001

См. Загрузка и подача SavedModel в сервировку TensorFlow раздела TensorFlow Руководство по сохранению и восстановлению для получения дополнительной информации.

...