search

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

MXNet 1.5 container #109

Closed eric-haibin-lin closed 4 years ago

eric-haibin-lin commented 5 years ago

Will MXNet 1.5 sagemaker container be released?

zeeshansayyed commented 5 years ago

Any update on this will be greatly appreciated. Can you point in the necessary changes required for supporting mxnet 1.5?

For instance, will just changing MX_VERSION=1.4.1 to MX_VERSION=1.5 be enough or are there other gotchas that we need to take care of.

JohnCalhoun commented 5 years ago

I updated MX_VERSION=1.5.1 and everything worked fine

laurenyu commented 4 years ago

The Dockerfiles have been moved to https://github.com/aws/deep-learning-containers