Managed Spot Training in Amazon SageMaker AI
Amazon SageMaker AI makes it easy to train machine learning models using managed Amazon EC2 Spot instances. Managed spot training can optimize the cost of training models up to 90% over on-demand instances. SageMaker AI manages the Spot interruptions on your behalf.
Managed Spot Training uses Amazon EC2 Spot instance to run training jobs instead of on-demand instances. You can specify which training jobs use spot instances and a stopping condition that specifies how long SageMaker AI waits for a job to run using Amazon EC2 Spot instances. Metrics and logs generated during training runs are available in CloudWatch.
Amazon SageMaker AI automatic model tuning, also known as hyperparameter tuning, can use managed spot training. For more information on automatic model tuning, see Automatic model tuning with SageMaker AI.
Spot instances can be interrupted, causing jobs to take longer to start or finish. You can configure your managed spot training job to use checkpoints. SageMaker AI copies checkpoint data from a local path to Amazon S3. When the job is restarted, SageMaker AI copies the data from Amazon S3 back into the local path. The training job can then resume from the last checkpoint instead of restarting. For more information about checkpointing, see Checkpoints in Amazon SageMaker AI.
Note
Unless your training job will complete quickly, we recommend you use checkpointing with
managed spot training. SageMaker AI built-in algorithms and marketplace algorithms that do not checkpoint
are currently limited to a MaxWaitTimeInSeconds of 3600 seconds (60 minutes).
To use managed spot training, create a training job. Set
EnableManagedSpotTraining to True and specify the
MaxWaitTimeInSeconds. MaxWaitTimeInSeconds must be larger than
MaxRuntimeInSeconds. For more information about creating a training job, see DescribeTrainingJob.
You can calculate the savings from using managed spot training using the formula (1 -
(BillableTimeInSeconds / TrainingTimeInSeconds)) * 100. For example, if
BillableTimeInSeconds is 100 and TrainingTimeInSeconds is 500, this
means that your training job ran for 500 seconds, but you were billed for only 100 seconds. Your savings is (1 - (100 / 500)) * 100 = 80%.
To learn how to run training jobs on Amazon SageMaker AI spot instances and how managed spot training works and reduces the billable time, see the following example notebooks:
