flytekitplugins.awssagemaker.SagemakerTrainingJobConfig#

class flytekitplugins.awssagemaker.SagemakerTrainingJobConfig(training_job_resource_config, algorithm_specification, should_persist_output=<function SagemakerTrainingJobConfig.<lambda>>)[source]#

Configuration for Running Training Jobs on Sagemaker. This config can be used to run either the built-in algorithms or custom algorithms.

Parameters
Return type

None

Methods

should_persist_output()#

Attributes

training_job_resource_config: flytekitplugins.awssagemaker.models.training_job.TrainingJobResourceConfig
algorithm_specification: flytekitplugins.awssagemaker.models.training_job.AlgorithmSpecification