Package com.pulumi.aws.sagemaker

Class TrainingJobArgs.Builder

  • Enclosing class:
    TrainingJobArgs
    public static final class TrainingJobArgs.Builder
extends java.lang.Object

    • Constructor Detail

      • Builder

        public Builder()

    • Method Detail

      • algorithmSpecification

        public TrainingJobArgs.Builder algorithmSpecification​(@Nullable
                                                      com.pulumi.core.Output<TrainingJobAlgorithmSpecificationArgs> algorithmSpecification)
        Parameters:
        algorithmSpecification - Algorithm-related parameters of the training job. See `algorithmSpecification` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • algorithmSpecification

        public TrainingJobArgs.Builder algorithmSpecification​(TrainingJobAlgorithmSpecificationArgs algorithmSpecification)
        Parameters:
        algorithmSpecification - Algorithm-related parameters of the training job. See `algorithmSpecification` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • checkpointConfig

        public TrainingJobArgs.Builder checkpointConfig​(@Nullable
                                                com.pulumi.core.Output<TrainingJobCheckpointConfigArgs> checkpointConfig)
        Parameters:
        checkpointConfig - Location of checkpoints during training. See `checkpointConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • checkpointConfig

        public TrainingJobArgs.Builder checkpointConfig​(TrainingJobCheckpointConfigArgs checkpointConfig)
        Parameters:
        checkpointConfig - Location of checkpoints during training. See `checkpointConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • debugHookConfig

        public TrainingJobArgs.Builder debugHookConfig​(@Nullable
                                               com.pulumi.core.Output<TrainingJobDebugHookConfigArgs> debugHookConfig)
        Parameters:
        debugHookConfig - Configuration for debugging rules. See `debugHookConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • debugHookConfig

        public TrainingJobArgs.Builder debugHookConfig​(TrainingJobDebugHookConfigArgs debugHookConfig)
        Parameters:
        debugHookConfig - Configuration for debugging rules. See `debugHookConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • debugRuleConfigurations

        public TrainingJobArgs.Builder debugRuleConfigurations​(@Nullable
                                                       com.pulumi.core.Output<java.util.List<TrainingJobDebugRuleConfigurationArgs>> debugRuleConfigurations)
        Parameters:
        debugRuleConfigurations - List of debug rule configurations. Maximum of 20. See `debugRuleConfigurations` below.
        Returns:
        builder

      • debugRuleConfigurations

        public TrainingJobArgs.Builder debugRuleConfigurations​(java.util.List<TrainingJobDebugRuleConfigurationArgs> debugRuleConfigurations)
        Parameters:
        debugRuleConfigurations - List of debug rule configurations. Maximum of 20. See `debugRuleConfigurations` below.
        Returns:
        builder

      • deleteModelPackagesOnDestroy

        public TrainingJobArgs.Builder deleteModelPackagesOnDestroy​(@Nullable
                                                            com.pulumi.core.Output<java.lang.Boolean> deleteModelPackagesOnDestroy)
        Parameters:
        deleteModelPackagesOnDestroy - Whether to delete model packages in the configured model package group when the training job is destroyed. Default is `false`.
        Returns:
        builder

      • deleteModelPackagesOnDestroy

        public TrainingJobArgs.Builder deleteModelPackagesOnDestroy​(java.lang.Boolean deleteModelPackagesOnDestroy)
        Parameters:
        deleteModelPackagesOnDestroy - Whether to delete model packages in the configured model package group when the training job is destroyed. Default is `false`.
        Returns:
        builder

      • deleteVpcEnisOnDestroy

        public TrainingJobArgs.Builder deleteVpcEnisOnDestroy​(@Nullable
                                                      com.pulumi.core.Output<java.lang.Boolean> deleteVpcEnisOnDestroy)
        Parameters:
        deleteVpcEnisOnDestroy - Whether to delete detached VPC ENIs SageMaker may leave behind when the training job is destroyed. Default is `false`.
        Returns:
        builder

      • deleteVpcEnisOnDestroy

        public TrainingJobArgs.Builder deleteVpcEnisOnDestroy​(java.lang.Boolean deleteVpcEnisOnDestroy)
        Parameters:
        deleteVpcEnisOnDestroy - Whether to delete detached VPC ENIs SageMaker may leave behind when the training job is destroyed. Default is `false`.
        Returns:
        builder

      • enableInterContainerTrafficEncryption

        public TrainingJobArgs.Builder enableInterContainerTrafficEncryption​(@Nullable
                                                                     com.pulumi.core.Output<java.lang.Boolean> enableInterContainerTrafficEncryption)
        Parameters:
        enableInterContainerTrafficEncryption - Whether to encrypt inter-container traffic. When enabled, communications between containers are encrypted.
        Returns:
        builder

      • enableInterContainerTrafficEncryption

        public TrainingJobArgs.Builder enableInterContainerTrafficEncryption​(java.lang.Boolean enableInterContainerTrafficEncryption)
        Parameters:
        enableInterContainerTrafficEncryption - Whether to encrypt inter-container traffic. When enabled, communications between containers are encrypted.
        Returns:
        builder

      • enableManagedSpotTraining

        public TrainingJobArgs.Builder enableManagedSpotTraining​(@Nullable
                                                         com.pulumi.core.Output<java.lang.Boolean> enableManagedSpotTraining)
        Parameters:
        enableManagedSpotTraining - Whether to use managed spot training. Optimizes the cost of training by using Amazon EC2 Spot Instances. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • enableManagedSpotTraining

        public TrainingJobArgs.Builder enableManagedSpotTraining​(java.lang.Boolean enableManagedSpotTraining)
        Parameters:
        enableManagedSpotTraining - Whether to use managed spot training. Optimizes the cost of training by using Amazon EC2 Spot Instances. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • enableNetworkIsolation

        public TrainingJobArgs.Builder enableNetworkIsolation​(@Nullable
                                                      com.pulumi.core.Output<java.lang.Boolean> enableNetworkIsolation)
        Parameters:
        enableNetworkIsolation - Whether to isolate the training container from the network. No inbound or outbound network calls can be made.
        Returns:
        builder

      • enableNetworkIsolation

        public TrainingJobArgs.Builder enableNetworkIsolation​(java.lang.Boolean enableNetworkIsolation)
        Parameters:
        enableNetworkIsolation - Whether to isolate the training container from the network. No inbound or outbound network calls can be made.
        Returns:
        builder

      • environment

        public TrainingJobArgs.Builder environment​(@Nullable
                                           com.pulumi.core.Output<java.util.Map<java.lang.String,​java.lang.String>> environment)
        Parameters:
        environment - Map of environment variables to set in the training container. Maximum of 100 entries. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • environment

        public TrainingJobArgs.Builder environment​(java.util.Map<java.lang.String,​java.lang.String> environment)
        Parameters:
        environment - Map of environment variables to set in the training container. Maximum of 100 entries. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • experimentConfig

        public TrainingJobArgs.Builder experimentConfig​(@Nullable
                                                com.pulumi.core.Output<TrainingJobExperimentConfigArgs> experimentConfig)
        Parameters:
        experimentConfig - Associates a SageMaker AI Experiment or Trial to the training job. See `experimentConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • experimentConfig

        public TrainingJobArgs.Builder experimentConfig​(TrainingJobExperimentConfigArgs experimentConfig)
        Parameters:
        experimentConfig - Associates a SageMaker AI Experiment or Trial to the training job. See `experimentConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • hyperParameters

        public TrainingJobArgs.Builder hyperParameters​(@Nullable
                                               com.pulumi.core.Output<java.util.Map<java.lang.String,​java.lang.String>> hyperParameters)
        Parameters:
        hyperParameters - Map of hyperparameters for the training algorithm. Maximum of 100 entries.
        Returns:
        builder

      • hyperParameters

        public TrainingJobArgs.Builder hyperParameters​(java.util.Map<java.lang.String,​java.lang.String> hyperParameters)
        Parameters:
        hyperParameters - Map of hyperparameters for the training algorithm. Maximum of 100 entries.
        Returns:
        builder

      • infraCheckConfig

        public TrainingJobArgs.Builder infraCheckConfig​(@Nullable
                                                com.pulumi.core.Output<TrainingJobInfraCheckConfigArgs> infraCheckConfig)
        Parameters:
        infraCheckConfig - Infrastructure health check configuration. See `infraCheckConfig` below.
        Returns:
        builder

      • inputDataConfigs

        public TrainingJobArgs.Builder inputDataConfigs​(@Nullable
                                                com.pulumi.core.Output<java.util.List<TrainingJobInputDataConfigArgs>> inputDataConfigs)
        Parameters:
        inputDataConfigs - List of input data channel configurations for the training job. Maximum of 20. See `inputDataConfig` below.
        Returns:
        builder

      • inputDataConfigs

        public TrainingJobArgs.Builder inputDataConfigs​(java.util.List<TrainingJobInputDataConfigArgs> inputDataConfigs)
        Parameters:
        inputDataConfigs - List of input data channel configurations for the training job. Maximum of 20. See `inputDataConfig` below.
        Returns:
        builder

      • inputDataConfigs

        public TrainingJobArgs.Builder inputDataConfigs​(TrainingJobInputDataConfigArgs... inputDataConfigs)
        Parameters:
        inputDataConfigs - List of input data channel configurations for the training job. Maximum of 20. See `inputDataConfig` below.
        Returns:
        builder

      • modelPackageConfig

        public TrainingJobArgs.Builder modelPackageConfig​(@Nullable
                                                  com.pulumi.core.Output<TrainingJobModelPackageConfigArgs> modelPackageConfig)
        Parameters:
        modelPackageConfig - Model package configuration. Requires `serverlessJobConfig`. See `modelPackageConfig` below.
        Returns:
        builder

      • outputDataConfig

        public TrainingJobArgs.Builder outputDataConfig​(@Nullable
                                                com.pulumi.core.Output<TrainingJobOutputDataConfigArgs> outputDataConfig)
        Parameters:
        outputDataConfig - Location of the output data from the training job. See `outputDataConfig` below. The following arguments are optional:
        Returns:
        builder

      • outputDataConfig

        public TrainingJobArgs.Builder outputDataConfig​(TrainingJobOutputDataConfigArgs outputDataConfig)
        Parameters:
        outputDataConfig - Location of the output data from the training job. See `outputDataConfig` below. The following arguments are optional:
        Returns:
        builder

      • profilerConfig

        public TrainingJobArgs.Builder profilerConfig​(@Nullable
                                              com.pulumi.core.Output<TrainingJobProfilerConfigArgs> profilerConfig)
        Parameters:
        profilerConfig - Configuration for the profiler. See `profilerConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • profilerRuleConfigurations

        public TrainingJobArgs.Builder profilerRuleConfigurations​(@Nullable
                                                          com.pulumi.core.Output<java.util.List<TrainingJobProfilerRuleConfigurationArgs>> profilerRuleConfigurations)
        Parameters:
        profilerRuleConfigurations - List of profiler rule configurations. Maximum of 20. See `profilerRuleConfigurations` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • profilerRuleConfigurations

        public TrainingJobArgs.Builder profilerRuleConfigurations​(java.util.List<TrainingJobProfilerRuleConfigurationArgs> profilerRuleConfigurations)
        Parameters:
        profilerRuleConfigurations - List of profiler rule configurations. Maximum of 20. See `profilerRuleConfigurations` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • profilerRuleConfigurations

        public TrainingJobArgs.Builder profilerRuleConfigurations​(TrainingJobProfilerRuleConfigurationArgs... profilerRuleConfigurations)
        Parameters:
        profilerRuleConfigurations - List of profiler rule configurations. Maximum of 20. See `profilerRuleConfigurations` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • region

        public TrainingJobArgs.Builder region​(@Nullable
                                      com.pulumi.core.Output<java.lang.String> region)
        Parameters:
        region - Region where this resource will be [managed](https://docs.aws.amazon.com/general/latest/gr/rande.html#regional-endpoints). Defaults to the Region set in the provider configuration.
        Returns:
        builder

      • region

        public TrainingJobArgs.Builder region​(java.lang.String region)
        Parameters:
        region - Region where this resource will be [managed](https://docs.aws.amazon.com/general/latest/gr/rande.html#regional-endpoints). Defaults to the Region set in the provider configuration.
        Returns:
        builder

      • remoteDebugConfig

        public TrainingJobArgs.Builder remoteDebugConfig​(@Nullable
                                                 com.pulumi.core.Output<TrainingJobRemoteDebugConfigArgs> remoteDebugConfig)
        Parameters:
        remoteDebugConfig - Configuration for remote debugging. See `remoteDebugConfig` below.
        Returns:
        builder

      • resourceConfig

        public TrainingJobArgs.Builder resourceConfig​(@Nullable
                                              com.pulumi.core.Output<TrainingJobResourceConfigArgs> resourceConfig)
        Parameters:
        resourceConfig - Resources for the training job, including compute instances and storage volumes. See `resourceConfig` below.
        Returns:
        builder

      • resourceConfig

        public TrainingJobArgs.Builder resourceConfig​(TrainingJobResourceConfigArgs resourceConfig)
        Parameters:
        resourceConfig - Resources for the training job, including compute instances and storage volumes. See `resourceConfig` below.
        Returns:
        builder

      • retryStrategy

        public TrainingJobArgs.Builder retryStrategy​(@Nullable
                                             com.pulumi.core.Output<TrainingJobRetryStrategyArgs> retryStrategy)
        Parameters:
        retryStrategy - Number of times to retry the job if it fails. See `retryStrategy` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • retryStrategy

        public TrainingJobArgs.Builder retryStrategy​(TrainingJobRetryStrategyArgs retryStrategy)
        Parameters:
        retryStrategy - Number of times to retry the job if it fails. See `retryStrategy` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • roleArn

        public TrainingJobArgs.Builder roleArn​(com.pulumi.core.Output<java.lang.String> roleArn)
        Parameters:
        roleArn - ARN of the IAM role that SageMaker AI assumes to perform tasks on your behalf during training.
        Returns:
        builder

      • roleArn

        public TrainingJobArgs.Builder roleArn​(java.lang.String roleArn)
        Parameters:
        roleArn - ARN of the IAM role that SageMaker AI assumes to perform tasks on your behalf during training.
        Returns:
        builder

      • serverlessJobConfig

        public TrainingJobArgs.Builder serverlessJobConfig​(@Nullable
                                                   com.pulumi.core.Output<TrainingJobServerlessJobConfigArgs> serverlessJobConfig)
        Parameters:
        serverlessJobConfig - Configuration for serverless training jobs using foundation models. Conflicts with `algorithmSpecification`, `enableManagedSpotTraining`, `environment`, `retryStrategy`, `checkpointConfig`, `debugHookConfig`, `experimentConfig`, `profilerConfig`, `profilerRuleConfigurations`, and `tensorBoardOutputConfig`. See `serverlessJobConfig` below.
        Returns:
        builder

      • serverlessJobConfig

        public TrainingJobArgs.Builder serverlessJobConfig​(TrainingJobServerlessJobConfigArgs serverlessJobConfig)
        Parameters:
        serverlessJobConfig - Configuration for serverless training jobs using foundation models. Conflicts with `algorithmSpecification`, `enableManagedSpotTraining`, `environment`, `retryStrategy`, `checkpointConfig`, `debugHookConfig`, `experimentConfig`, `profilerConfig`, `profilerRuleConfigurations`, and `tensorBoardOutputConfig`. See `serverlessJobConfig` below.
        Returns:
        builder

      • sessionChainingConfig

        public TrainingJobArgs.Builder sessionChainingConfig​(@Nullable
                                                     com.pulumi.core.Output<TrainingJobSessionChainingConfigArgs> sessionChainingConfig)
        Parameters:
        sessionChainingConfig - Configuration for session tag chaining. See `sessionChainingConfig` below.
        Returns:
        builder

      • tags

        public TrainingJobArgs.Builder tags​(@Nullable
                                    com.pulumi.core.Output<java.util.Map<java.lang.String,​java.lang.String>> tags)
        Parameters:
        tags - Map of tags to assign to the resource. If configured with a provider `defaultTags` configuration block present, tags with matching keys will overwrite those defined at the provider-level.
        Returns:
        builder

      • tags

        public TrainingJobArgs.Builder tags​(java.util.Map<java.lang.String,​java.lang.String> tags)
        Parameters:
        tags - Map of tags to assign to the resource. If configured with a provider `defaultTags` configuration block present, tags with matching keys will overwrite those defined at the provider-level.
        Returns:
        builder

      • tensorBoardOutputConfig

        public TrainingJobArgs.Builder tensorBoardOutputConfig​(@Nullable
                                                       com.pulumi.core.Output<TrainingJobTensorBoardOutputConfigArgs> tensorBoardOutputConfig)
        Parameters:
        tensorBoardOutputConfig - Configuration for TensorBoard output. See `tensorBoardOutputConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • tensorBoardOutputConfig

        public TrainingJobArgs.Builder tensorBoardOutputConfig​(TrainingJobTensorBoardOutputConfigArgs tensorBoardOutputConfig)
        Parameters:
        tensorBoardOutputConfig - Configuration for TensorBoard output. See `tensorBoardOutputConfig` below. Conflicts with `serverlessJobConfig`.
        Returns:
        builder

      • trainingJobName

        public TrainingJobArgs.Builder trainingJobName​(com.pulumi.core.Output<java.lang.String> trainingJobName)
        Parameters:
        trainingJobName - Name of the training job. Must be between 1 and 63 characters, start with a letter or number, and contain only letters, numbers, and hyphens.
        Returns:
        builder

      • trainingJobName

        public TrainingJobArgs.Builder trainingJobName​(java.lang.String trainingJobName)
        Parameters:
        trainingJobName - Name of the training job. Must be between 1 and 63 characters, start with a letter or number, and contain only letters, numbers, and hyphens.
        Returns:
        builder