flytekitplugins.awssagemaker.TrainingJobResourceConfig#

class flytekitplugins.awssagemaker.TrainingJobResourceConfig(instance_type, volume_size_in_gb, instance_count=1, distributed_protocol=0)[source]#

TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the number of instances to launch, and the size of the ML storage volume the user wants to provision Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html

Methods

Parameters
  • instance_type (str) –

  • volume_size_in_gb (int) –

  • instance_count (int) –

  • distributed_protocol (int) –

classmethod from_flyte_idl(pb2_object)[source]#
Parameters

pb2_object (flyteidl.plugins.sagemaker.training_job_pb2.TrainingJobResourceConfig) –

Return type

TrainingJobResourceConfig

serialize_to_string()#
Return type

str

short_string()#
Return type

Text

to_flyte_idl()[source]#
Return type

_training_job_pb2.TrainingJobResourceConfig

verbose_string()#
Return type

Text

Attributes

distributed_protocol

The distribution framework is used to determine through which mechanism the distributed training is done. enum value from DistributionFramework. :rtype: int

instance_count

The number of ML compute instances to use. For distributed training, provide a value greater than 1. :rtype: int

instance_type

The ML compute instance type. :rtype: str

is_empty
volume_size_in_gb

The size of the ML storage volume that you want to provision to store the data and intermediate artifacts, etc. :rtype: int