Extend a training plan using the SageMaker API or AWS CLI - Amazon SageMaker AI

Extend a training plan using the SageMaker API or AWS CLI

SageMaker training plans support extending training plans programmatically through its API. You can interact with the training plans API using the AWS CLI or SageMaker SDKs.

The training plan extension involves the following API actions:

  • SearchTrainingPlanOfferings: Search for available extension offerings by specifying your training plan ARN and desired extension duration. The API returns extension offerings in the TrainingPlanExtensionOfferings field.

  • ExtendTrainingPlan: Purchase a specific extension offering to extend your training plan by providing the TrainingPlanExtensionOfferingId. This reserves the additional compute capacity and updates your training plan's end date.

  • DescribeTrainingPlanExtensionHistory: View the complete extension history for a training plan, including all past extensions with their status, dates, and payment information.