Skip to content

CreateTrainingSessionRequest

Source code in docs_src/capo_sagemaker/types/create_training_session_request.py
class CreateTrainingSessionRequest(TypedDict, closed=True):
    training_session_name: NotRequired[
        "capo_sagemaker.types.training_session_name.TrainingSessionName"
    ]
    """<p>The name of the training session.</p>"""
    model_package_group_arn: NotRequired[
        "capo_sagemaker.types.model_package_group_arn.ModelPackageGroupArn"
    ]
    """<p>The Amazon Resource Name (ARN) of the model package group where checkpoints and fine-tuned model artifacts are published.</p>"""
    lora_config: NotRequired["capo_sagemaker.types.lora_config.LoraConfig"]
    """<p>The Low-Rank Adaptation (LoRA) configuration for the training session, including the rank and alpha parameters.</p>"""
    role_arn: NotRequired["capo_sagemaker.types.role_arn.RoleArn"]
    """<p>The Amazon Resource Name (ARN) of the IAM role that Amazon SageMaker assumes to access resources on your behalf. The role must have permissions to access the output S3 path and model package group.</p>"""
    vpc_config: NotRequired["capo_sagemaker.types.vpc_config.VpcConfig"]
    """<p>The Virtual Private Cloud (VPC) configuration for the training session. Specify subnets and security groups to run the training session in your VPC.</p>"""
    model_config: NotRequired[
        "capo_sagemaker.types.training_session_model_config.TrainingSessionModelConfig"
    ]
    """<p>The configuration for the base model, including the model ARN and end-user license agreement (EULA) acceptance.</p>"""
    output_config: NotRequired[
        "capo_sagemaker.types.training_session_output_config.TrainingSessionOutputConfig"
    ]
    """<p>The output configuration for the training session, including the S3 location and optional encryption settings.</p>"""
    tags: NotRequired["capo_sagemaker.types.tag_list.TagList"]
    """<p>An array of key-value pairs to apply to the training session as tags. For more information, see <a href="https://docs.aws.amazon.com/general/latest/gr/aws_tagging.html">Tagging Amazon Web Services Resources</a>.</p>"""

lora_config instance-attribute

lora_config: NotRequired[LoraConfig]

The Low-Rank Adaptation (LoRA) configuration for the training session, including the rank and alpha parameters.

model_config instance-attribute

model_config: NotRequired[TrainingSessionModelConfig]

The configuration for the base model, including the model ARN and end-user license agreement (EULA) acceptance.

model_package_group_arn instance-attribute

model_package_group_arn: NotRequired[ModelPackageGroupArn]

The Amazon Resource Name (ARN) of the model package group where checkpoints and fine-tuned model artifacts are published.

output_config instance-attribute

output_config: NotRequired[TrainingSessionOutputConfig]

The output configuration for the training session, including the S3 location and optional encryption settings.

role_arn instance-attribute

role_arn: NotRequired[RoleArn]

The Amazon Resource Name (ARN) of the IAM role that Amazon SageMaker assumes to access resources on your behalf. The role must have permissions to access the output S3 path and model package group.

tags instance-attribute

tags: NotRequired[TagList]

An array of key-value pairs to apply to the training session as tags. For more information, see Tagging Amazon Web Services Resources.

training_session_name instance-attribute

training_session_name: NotRequired[TrainingSessionName]

The name of the training session.

vpc_config instance-attribute

vpc_config: NotRequired[VpcConfig]

The Virtual Private Cloud (VPC) configuration for the training session. Specify subnets and security groups to run the training session in your VPC.