-
Book Overview & Buying
-
Table Of Contents
-
Feedback & Rating

Machine Learning with Amazon SageMaker Cookbook
By :

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!
This recipe continues from Generating a synthetic dataset for deep learning experiments.
The instructions in this recipe focus on preparing the training entrypoint
script. Let's start by creating...