Toolkit for running MXNet training scripts on SageMaker. Dockerfiles used for building SageMaker MXNet Containers are at https://github.com/aws/deep-learning-containers.
Apache License 2.0
60
stars
55
forks
source link
Add functional test - MXNet distributed training + hosting + inference on SageMaker #8
MXNet test code based on mxnet_mnist example: https://github.com/awslabs/amazon-sagemaker-examples/pull/191/files
Test succeeds locally.