sagemaker/aws.sdk.kotlin.services.sagemaker.model/CreateTrainingJobRequest/Builder

Builder

Properties

var algorithmSpecification: AlgorithmSpecification?

The registry path of the Docker image that contains the training algorithm and algorithm-specific metadata, including the input mode. For more information about algorithms provided by SageMaker, see Algorithms. For information about providing your own algorithms, see Using Your Own Algorithms with Amazon SageMaker.

checkpointConfig

var checkpointConfig: CheckpointConfig?

Contains information about the output location for managed spot training checkpoint data.

debugHookConfig

var debugHookConfig: DebugHookConfig?

Configuration information for the Amazon SageMaker Debugger hook parameters, metric and tensor collections, and storage paths. To learn more about how to configure the DebugHookConfig parameter, see Use the SageMaker and Debugger Configuration API Operations to Create, Update, and Debug Your Training Job.

debugRuleConfigurations

var debugRuleConfigurations: List<DebugRuleConfiguration>?

Configuration information for Amazon SageMaker Debugger rules for debugging output tensors.

enableInterContainerTrafficEncryption

var enableInterContainerTrafficEncryption: Boolean?

To encrypt all communications between ML compute instances in distributed training, choose True. Encryption provides greater security for distributed training, but training might take longer. How long it takes depends on the amount of communication between compute instances, especially if you use a deep learning algorithm in distributed training. For more information, see Protect Communications Between ML Compute Instances in a Distributed Training Job.

enableManagedSpotTraining

var enableManagedSpotTraining: Boolean?

To train models using managed spot training, choose True. Managed spot training provides a fully managed and scalable infrastructure for training machine learning models. this option is useful when training jobs can be interrupted and when there is flexibility when the training job is run.

enableNetworkIsolation

var enableNetworkIsolation: Boolean?

Isolates the training container. No inbound or outbound network calls can be made, except for calls between peers within a training cluster for distributed training. If you enable network isolation for training jobs that are configured to use a VPC, SageMaker downloads and uploads customer data and model artifacts through the specified VPC, but the training container does not have network access.

environment

var environment: Map<String, String>?

The environment variables to set in the Docker container.

experimentConfig

var experimentConfig: ExperimentConfig?

Associates a SageMaker job as a trial component with an experiment and trial. Specified when you call the following APIs:

hyperParameters

var hyperParameters: Map<String, String>?

Algorithm-specific parameters that influence the quality of the model. You set hyperparameters before you start the learning process. For a list of hyperparameters for each training algorithm provided by SageMaker, see Algorithms.

infraCheckConfig

var infraCheckConfig: InfraCheckConfig?

Contains information about the infrastructure health check configuration for the training job.

inputDataConfig

var inputDataConfig: List<Channel>?

An array of Channel objects. Each channel is a named input source. InputDataConfig describes the input data and its location.

outputDataConfig

var outputDataConfig: OutputDataConfig?

Specifies the path to the S3 location where you want to store model artifacts. SageMaker creates subfolders for the artifacts.

profilerConfig

var profilerConfig: ProfilerConfig?

Configuration information for Amazon SageMaker Debugger system monitoring, framework profiling, and storage paths.

profilerRuleConfigurations

var profilerRuleConfigurations: List<ProfilerRuleConfiguration>?

Configuration information for Amazon SageMaker Debugger rules for profiling system and framework metrics.

remoteDebugConfig

var remoteDebugConfig: RemoteDebugConfig?

Configuration for remote debugging. To learn more about the remote debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) for remote debugging.

resourceConfig

var resourceConfig: ResourceConfig?

The resources, including the ML compute instances and ML storage volumes, to use for model training.

retryStrategy

var retryStrategy: RetryStrategy?

The number of times to retry the job when the job fails due to an InternalServerError.

roleArn

var roleArn: String?

The Amazon Resource Name (ARN) of an IAM role that SageMaker can assume to perform tasks on your behalf.

sessionChainingConfig

var sessionChainingConfig: SessionChainingConfig?

Contains information about attribute-based access control (ABAC) for the training job.

stoppingCondition

var stoppingCondition: StoppingCondition?

Specifies a limit to how long a model training job can run. It also specifies how long a managed Spot training job has to complete. When the job reaches the time limit, SageMaker ends the training job. Use this API to cap model training costs.