Class: Aws::SageMaker::Types::TrainingJobDefinition
- Inherits:
-
Struct
- Object
- Struct
- Aws::SageMaker::Types::TrainingJobDefinition
- Includes:
- Aws::Structure
- Defined in:
- lib/aws-sdk-sagemaker/types.rb
Overview
When making an API call, you may pass TrainingJobDefinition data as a hash:
{
training_input_mode: "Pipe", # required, accepts Pipe, File
hyper_parameters: {
"ParameterKey" => "ParameterValue",
},
input_data_config: [ # required
{
channel_name: "ChannelName", # required
data_source: { # required
s3_data_source: { # required
s3_data_type: "ManifestFile", # required, accepts ManifestFile, S3Prefix, AugmentedManifestFile
s3_uri: "S3Uri", # required
s3_data_distribution_type: "FullyReplicated", # accepts FullyReplicated, ShardedByS3Key
attribute_names: ["AttributeName"],
},
},
content_type: "ContentType",
compression_type: "None", # accepts None, Gzip
record_wrapper_type: "None", # accepts None, RecordIO
input_mode: "Pipe", # accepts Pipe, File
shuffle_config: {
seed: 1, # required
},
},
],
output_data_config: { # required
kms_key_id: "KmsKeyId",
s3_output_path: "S3Uri", # required
},
resource_config: { # required
instance_type: "ml.m4.xlarge", # required, accepts ml.m4.xlarge, ml.m4.2xlarge, ml.m4.4xlarge, ml.m4.10xlarge, ml.m4.16xlarge, ml.m5.large, ml.m5.xlarge, ml.m5.2xlarge, ml.m5.4xlarge, ml.m5.12xlarge, ml.m5.24xlarge, ml.c4.xlarge, ml.c4.2xlarge, ml.c4.4xlarge, ml.c4.8xlarge, ml.p2.xlarge, ml.p2.8xlarge, ml.p2.16xlarge, ml.p3.2xlarge, ml.p3.8xlarge, ml.p3.16xlarge, ml.c5.xlarge, ml.c5.2xlarge, ml.c5.4xlarge, ml.c5.9xlarge, ml.c5.18xlarge
instance_count: 1, # required
volume_size_in_gb: 1, # required
volume_kms_key_id: "KmsKeyId",
},
stopping_condition: { # required
max_runtime_in_seconds: 1,
},
}
Defines the input needed to run a training job using the algorithm.
Instance Attribute Summary collapse
-
#hyper_parameters ⇒ Hash<String,String>
The hyperparameters used for the training job.
-
#input_data_config ⇒ Array<Types::Channel>
An array of ‘Channel` objects, each of which specifies an input source.
-
#output_data_config ⇒ Types::OutputDataConfig
the path to the S3 bucket where you want to store model artifacts.
-
#resource_config ⇒ Types::ResourceConfig
The resources, including the ML compute instances and ML storage volumes, to use for model training.
-
#stopping_condition ⇒ Types::StoppingCondition
Sets a duration for training.
-
#training_input_mode ⇒ String
The input mode used by the algorithm for the training job.
Instance Attribute Details
#hyper_parameters ⇒ Hash<String,String>
The hyperparameters used for the training job.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |
#input_data_config ⇒ Array<Types::Channel>
An array of ‘Channel` objects, each of which specifies an input source.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |
#output_data_config ⇒ Types::OutputDataConfig
the path to the S3 bucket where you want to store model artifacts. Amazon SageMaker creates subfolders for the artifacts.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |
#resource_config ⇒ Types::ResourceConfig
The resources, including the ML compute instances and ML storage volumes, to use for model training.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |
#stopping_condition ⇒ Types::StoppingCondition
Sets a duration for training. Use this parameter to cap model training costs.
To stop a job, Amazon SageMaker sends the algorithm the SIGTERM signal, which delays job termination for 120 seconds. Algorithms might use this 120-second window to save the model artifacts.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |
#training_input_mode ⇒ String
The input mode used by the algorithm for the training job. For the input modes that Amazon SageMaker algorithms support, see [Algorithms].
If an algorithm supports the ‘File` input mode, Amazon SageMaker downloads the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker volume for training container. If an algorithm supports the `Pipe` input mode, Amazon SageMaker streams data directly from S3 to the container.
10465 10466 10467 10468 10469 10470 10471 10472 10473 |
# File 'lib/aws-sdk-sagemaker/types.rb', line 10465 class TrainingJobDefinition < Struct.new( :training_input_mode, :hyper_parameters, :input_data_config, :output_data_config, :resource_config, :stopping_condition) include Aws::Structure end |