aws / sagemaker-mxnet-training-toolkit

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

Closed winstonaws closed 6 years ago

winstonaws commented 6 years ago

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.