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
Apply patch - Fix optimizer serialization for python3 #13
Applies this patch to all of the image builds for 0.12.1 and 1.0.0: https://github.com/apache/incubator-mxnet/pull/8334
Practically speaking, this fixes distributed training for our python 3 images for both 0.12.1 and 1.0.0.
I've built 1.0.0-cpu-py3 using this change, and all tests passed for it. Still waiting on the builds for the other versions.