Class: Aws::SageMaker::Types::CreateCompilationJobRequest

Inherits:
Struct
  • Object
show all
Includes:
Aws::Structure
Defined in:
lib/aws-sdk-sagemaker/types.rb

Overview

Note:

When making an API call, you may pass CreateCompilationJobRequest data as a hash:

{
  compilation_job_name: "EntityName", # required
  role_arn: "RoleArn", # required
  input_config: { # required
    s3_uri: "S3Uri", # required
    data_input_config: "DataInputConfig", # required
    framework: "TENSORFLOW", # required, accepts TENSORFLOW, MXNET, ONNX, PYTORCH, XGBOOST
  },
  output_config: { # required
    s3_output_location: "S3Uri", # required
    target_device: "ml_m4", # required, accepts ml_m4, ml_m5, ml_c4, ml_c5, ml_p2, ml_p3, jetson_tx1, jetson_tx2, rasp3b, deeplens, rk3399, rk3288
  },
  stopping_condition: { # required
    max_runtime_in_seconds: 1,
  },
}

Instance Attribute Summary collapse

Instance Attribute Details

#compilation_job_nameString

A name for the model compilation job. The name must be unique within the AWS Region and within your AWS account.

Returns:

  • (String)


1318
1319
1320
1321
1322
1323
1324
1325
# File 'lib/aws-sdk-sagemaker/types.rb', line 1318

class CreateCompilationJobRequest < Struct.new(
  :compilation_job_name,
  :role_arn,
  :input_config,
  :output_config,
  :stopping_condition)
  include Aws::Structure
end

#input_configTypes::InputConfig

Provides information about the location of input model artifacts, the name and shape of the expected data inputs, and the framework in which the model was trained.

Returns:



1318
1319
1320
1321
1322
1323
1324
1325
# File 'lib/aws-sdk-sagemaker/types.rb', line 1318

class CreateCompilationJobRequest < Struct.new(
  :compilation_job_name,
  :role_arn,
  :input_config,
  :output_config,
  :stopping_condition)
  include Aws::Structure
end

#output_configTypes::OutputConfig

Provides information about the output location for the compiled model and the target device the model runs on.

Returns:



1318
1319
1320
1321
1322
1323
1324
1325
# File 'lib/aws-sdk-sagemaker/types.rb', line 1318

class CreateCompilationJobRequest < Struct.new(
  :compilation_job_name,
  :role_arn,
  :input_config,
  :output_config,
  :stopping_condition)
  include Aws::Structure
end

#role_arnString

The Amazon Resource Name (ARN) of an IIAMAM role that enables Amazon SageMaker to perform tasks on your behalf.

During model compilation, Amazon SageMaker needs your permission to:

  • Read input data from an S3 bucket

  • Write model artifacts to an S3 bucket

  • Write logs to Amazon CloudWatch Logs

  • Publish metrics to Amazon CloudWatch

You grant permissions for all of these tasks to an IAM role. To pass this role to Amazon SageMaker, the caller of this API must have the ‘iam:PassRole` permission. For more information, see [Amazon SageMaker Roles.]

[1]: docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-roles.html

Returns:

  • (String)


1318
1319
1320
1321
1322
1323
1324
1325
# File 'lib/aws-sdk-sagemaker/types.rb', line 1318

class CreateCompilationJobRequest < Struct.new(
  :compilation_job_name,
  :role_arn,
  :input_config,
  :output_config,
  :stopping_condition)
  include Aws::Structure
end

#stopping_conditionTypes::StoppingCondition

The duration allowed for model compilation.



1318
1319
1320
1321
1322
1323
1324
1325
# File 'lib/aws-sdk-sagemaker/types.rb', line 1318

class CreateCompilationJobRequest < Struct.new(
  :compilation_job_name,
  :role_arn,
  :input_config,
  :output_config,
  :stopping_condition)
  include Aws::Structure
end