亚马逊SageMaker中的Tensorflow服务

Question

我在AWS SageMaker上面临服务tensorflow模型的问题。我已经在SageMaker环境之外训练了模型，现在我有一个savedmodel.pb文件，我需要在SageMaker端点上部署它。所以我只是将模型文件压缩并上传到S3存储桶。现在，在尝试创建端点时，我在Cloudwatch日志中收到以下错误：

tensorflow_serving / sources / storage_path / file_system_storage_path_source.cc：369] FileSystemStoragePathSource遇到文件系统访问错误：找不到基本路径/ opt / ml / model / export / Servo for servable generic_model

我相信SageMaker正在寻找tar.gz来遵循特定的目录结构。但是，我只有一个.pb文件。

Answer 1

TensorFlow服务需要以下文件夹结构：

export / Servo / {version_number}其中版本号是任何有效的正数。

SageMaker期望TFS的目录格式相同，有关于这个https://github.com/aws/sagemaker-python-sdk/issues/599的GH问题