This package makes it easier to build docker images which run on Amazon SageMaker. For example, the official SageMaker TensorFlow and MXNet images use this package. The requirements for SageMaker-compatible images are documented here: https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html
Features provided:
- Integration with SageMaker Python SDK Estimators, including:
- Downloading user-provided python code
- Deserializing hyperparameters (which preserves their python types)
bin/entry.py
, which acts as the docker entrypoint required by SageMaker- Reading in the metadata files provided to the container during training, as described here: https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms-training-algo.html
- nginx + Gunicorn HTTP server for serving inference requests which complies with the interface described here: https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms-inference-code.html
This library is licensed under the Apache 2.0 License.