training_plans
Creates, updates, deletes, gets or lists a training_plans resource.
Overview
| Name | training_plans |
| Type | Resource |
| Id | aws.sagemaker.training_plans |
Fields
The following fields are returned by SELECT queries:
- describe_training_plan
- list_training_plans
| Name | Datatype | Description |
|---|---|---|
available_instance_count | integer | The number of instances currently available for use in this training plan. |
available_spare_instance_count | integer | The number of available spare instances in the training plan. |
currency_code | string | The currency code for the upfront fee (e.g., USD). |
duration_hours | integer (int64) | The number of whole hours in the total duration for this training plan. |
duration_minutes | integer (int64) | The additional minutes beyond whole hours in the total duration for this training plan. |
end_time | string (date-time) | The end time of the training plan. |
in_use_instance_count | integer | The number of instances currently in use from this training plan. |
reserved_capacity_summaries | array | The list of Reserved Capacity providing the underlying compute resources of the plan. |
start_time | string (date-time) | The start time of the training plan. |
status | string | The current status of the training plan (e.g., Pending, Active, Expired). To see the complete list of status values available for a training plan, refer to the Status attribute within the TrainingPlanSummary object. (Pending, Active, Scheduled, Expired, Failed) |
status_message | string | A message providing additional information about the current status of the training plan. |
target_resources | array | The target resources (e.g., SageMaker Training Jobs, SageMaker HyperPod, SageMaker Endpoints, Studio apps) that can use this training plan. Training plans are specific to their target resource. A training plan designed for SageMaker training jobs can only be used to schedule and run training jobs. A training plan for HyperPod clusters can be used exclusively to provide compute resources to a cluster's instance group. A training plan for SageMaker endpoints can be used exclusively to provide compute resources to SageMaker endpoints for model deployment. A training plan for Studio apps can be used to launch JupyterLab and Code Editor apps on reserved training plan capacity. |
total_instance_count | integer | The total number of instances reserved in this training plan. |
total_ultra_server_count | integer | The total number of UltraServers reserved to this training plan. |
training_plan_arn | string | The Amazon Resource Name (ARN); of the training plan. (pattern: <code>arn:aws[a-z-]:sagemaker:[a-z0-9-]:[0-9]{12}:training-plan/.*</code>) |
training_plan_name | string | The name of the training plan. (pattern: <code>[a-zA-Z0-9](-*[a-zA-Z0-9]){0,63}</code>) |
unhealthy_instance_count | integer | The number of instances in the training plan that are currently in an unhealthy state. |
upfront_fee | string | The upfront fee for the training plan. |
| Name | Datatype | Description |
|---|---|---|
available_instance_count | integer | The number of instances currently available for use in this training plan. |
currency_code | string | The currency code for the upfront fee (e.g., USD). |
duration_hours | integer (int64) | The number of whole hours in the total duration for this training plan. |
duration_minutes | integer (int64) | The additional minutes beyond whole hours in the total duration for this training plan. |
end_time | string (date-time) | The end time of the training plan. |
in_use_instance_count | integer | The number of instances currently in use from this training plan. |
reserved_capacity_summaries | array | A list of reserved capacities associated with this training plan, including details such as instance types, counts, and availability zones. |
start_time | string (date-time) | The start time of the training plan. |
status | string | The current status of the training plan (e.g., Pending, Active, Expired). To see the complete list of status values available for a training plan, refer to the Status attribute within the TrainingPlanSummary object. (Pending, Active, Scheduled, Expired, Failed) |
status_message | string | A message providing additional information about the current status of the training plan. |
target_resources | array | The target resources (e.g., training jobs, HyperPod clusters, Endpoints, Studio apps) that can use this training plan. Training plans are specific to their target resource. A training plan designed for SageMaker training jobs can only be used to schedule and run training jobs. A training plan for HyperPod clusters can be used exclusively to provide compute resources to a cluster's instance group. A training plan for SageMaker endpoints can be used exclusively to provide compute resources to SageMaker endpoints for model deployment. A training plan for Studio apps can be used to launch JupyterLab and Code Editor apps on reserved training plan capacity. |
total_instance_count | integer | The total number of instances reserved in this training plan. |
total_ultra_server_count | integer | The total number of UltraServers allocated to this training plan. |
training_plan_arn | string | The Amazon Resource Name (ARN); of the training plan. (pattern: <code>arn:aws[a-z-]:sagemaker:[a-z0-9-]:[0-9]{12}:training-plan/.*</code>) |
training_plan_name | string | The name of the training plan. (pattern: <code>[a-zA-Z0-9](-*[a-zA-Z0-9]){0,63}</code>) |
upfront_fee | string | The upfront fee for the training plan. |
Methods
The following methods are available for this resource:
| Name | Accessible by | Required Params | Optional Params | Description |
|---|---|---|---|---|
describe_training_plan | select | region | Retrieves detailed information about a specific training plan. | |
list_training_plans | select | region | Retrieves a list of training plans for the current account. | |
create_training_plan | insert | region, TrainingPlanName, TrainingPlanOfferingId | Creates a new training plan in SageMaker to reserve compute capacity. Amazon SageMaker Training Plan is a capability within SageMaker that allows customers to reserve and manage GPU capacity for large-scale AI model training. It provides a way to secure predictable access to computational resources within specific timelines and budgets, without the need to manage underlying infrastructure. How it works Plans can be created for specific resources such as SageMaker Training Jobs or SageMaker HyperPod clusters, automatically provisioning resources, setting up infrastructure, executing workloads, and handling infrastructure failures. Plan creation workflow Users search for available plan offerings based on their requirements (e.g., instance type, count, start time, duration) using the SearchTrainingPlanOfferings API operation. They create a plan that best matches their needs using the ID of the plan offering they want to use. After successful upfront payment, the plan's status becomes Scheduled. The plan can be used to: Queue training jobs. Allocate to an instance group of a SageMaker HyperPod cluster. When the plan start date arrives, it becomes Active. Based on available reserved capacity: Training jobs are launched. Instance groups are provisioned. Plan composition A plan can consist of one or more Reserved Capacities, each defined by a specific instance type, quantity, Availability Zone, duration, and start and end times. For more information about Reserved Capacity, see ReservedCapacitySummary . |
Parameters
Parameters can be passed in the WHERE clause of a query. Check the Methods section to see which parameters are required or optional for each operation.
| Name | Datatype | Description |
|---|---|---|
region | string | AWS region (default: us-east-1) |
SELECT examples
- describe_training_plan
- list_training_plans
Retrieves detailed information about a specific training plan.
SELECT
available_instance_count,
available_spare_instance_count,
currency_code,
duration_hours,
duration_minutes,
end_time,
in_use_instance_count,
reserved_capacity_summaries,
start_time,
status,
status_message,
target_resources,
total_instance_count,
total_ultra_server_count,
training_plan_arn,
training_plan_name,
unhealthy_instance_count,
upfront_fee
FROM aws.sagemaker.training_plans
WHERE region = '{{ region }}' -- required
;
Retrieves a list of training plans for the current account.
SELECT
available_instance_count,
currency_code,
duration_hours,
duration_minutes,
end_time,
in_use_instance_count,
reserved_capacity_summaries,
start_time,
status,
status_message,
target_resources,
total_instance_count,
total_ultra_server_count,
training_plan_arn,
training_plan_name,
upfront_fee
FROM aws.sagemaker.training_plans
WHERE region = '{{ region }}' -- required
;
INSERT examples
- create_training_plan
- Manifest
Creates a new training plan in SageMaker to reserve compute capacity. Amazon SageMaker Training Plan is a capability within SageMaker that allows customers to reserve and manage GPU capacity for large-scale AI model training. It provides a way to secure predictable access to computational resources within specific timelines and budgets, without the need to manage underlying infrastructure. How it works Plans can be created for specific resources such as SageMaker Training Jobs or SageMaker HyperPod clusters, automatically provisioning resources, setting up infrastructure, executing workloads, and handling infrastructure failures. Plan creation workflow Users search for available plan offerings based on their requirements (e.g., instance type, count, start time, duration) using the SearchTrainingPlanOfferings API operation. They create a plan that best matches their needs using the ID of the plan offering they want to use. After successful upfront payment, the plan's status becomes Scheduled. The plan can be used to: Queue training jobs. Allocate to an instance group of a SageMaker HyperPod cluster. When the plan start date arrives, it becomes Active. Based on available reserved capacity: Training jobs are launched. Instance groups are provisioned. Plan composition A plan can consist of one or more Reserved Capacities, each defined by a specific instance type, quantity, Availability Zone, duration, and start and end times. For more information about Reserved Capacity, see ReservedCapacitySummary .
INSERT INTO aws.sagemaker.training_plans (
TrainingPlanName,
TrainingPlanOfferingId,
SpareInstanceCountPerUltraServer,
Tags,
region
)
SELECT
'{{ TrainingPlanName }}' /* required */,
'{{ TrainingPlanOfferingId }}' /* required */,
{{ SpareInstanceCountPerUltraServer }},
'{{ Tags }}',
'{{ region }}'
RETURNING
training_plan_arn
;
# Description fields are for documentation purposes
- name: training_plans
props:
- name: region
value: "{{ region }}"
description: Required parameter for the training_plans resource.
- name: TrainingPlanName
value: "{{ TrainingPlanName }}"
description: |
The name of the training plan to create.
- name: TrainingPlanOfferingId
value: "{{ TrainingPlanOfferingId }}"
description: |
The unique identifier of the training plan offering to use for creating this plan.
- name: SpareInstanceCountPerUltraServer
value: {{ SpareInstanceCountPerUltraServer }}
description: |
Number of spare instances to reserve per UltraServer for enhanced resiliency. Default is 1.
- name: Tags
description: |
An array of key-value pairs to apply to this training plan.
value:
- Key: "{{ Key }}"
Value: "{{ Value }}"