Training with PyTorch on Amazon SageMaker – Julien Simon – Medium

PyTorch is a flexible open source framework for Deep Learning experimentation. In this post, you will learn how to train PyTorch jobs on Amazon SageMaker. As usual, you'll find my code on Github:) SageMaker provides a collection of built-in algorithms as well as environments for TensorFlow and MXNet… but not for PyTorch. Fortunately, developers have the option to build custom containers for training and prediction. This will require some changes in our PyTorch script, the well-known example of learning MNIST with a simple CNN.