Preparing a Training Script

The code samples for this tutorial show how to train a model using the SageMaker Python SDK. Refer to the training script train.py in the train directory here to see an example and a full explanation. This script will be wrapped in a custom image that will later be used when launching the SageMaker training job.

Among other details, this script is responsible for the following:

• Parsing job input arguments
• Loading the training data
• Configuring and training the ML model
• Retrieving the necessary credentials and other info from the AWS Secret
• Saving the model and related artifacts to Artifactory