Preparing the entrypoint PyTorch training script
PyTorch is a popular open source software library for machine learning. In this recipe, we will define a custom PyTorch neural network model and prepare the entrypoint training script. In one of the succeeding recipes, we will use the PyTorch estimator class from the SageMaker Python SDK with this script as the entrypoint argument when initializing the estimator object. Take note that we will have a separate entrypoint inference script to deploy a PyTorch model with SageMaker as we will see in the succeeding recipes.
If you are planning to migrate your custom PyTorch neural network code and perform training and deployment with the SageMaker platform, then this recipe and the next ones are for you!
Getting ready
This recipe continues from Generating a synthetic dataset for deep learning experiments.
How to do it
The instructions in this recipe focus on preparing the training entrypoint script. Let's start by creating...
