CreateTrainingSessionRequest
¶
Source code in docs_src/capo_sagemaker/types/create_training_session_request.py
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.