AWS Sagemaker API reference#
This package contains things that are useful when extending Flytekit.
The algorithm name is used for deciding which pre-built image to point to. |
|
|
Specifies the training algorithm to be used in the training job This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to CUSTOM. |
|
|
|
|
|
str(object='') -> str str(bytes_or_buffer[, encoding[, errors]]) -> str |
The distribution framework is used for determining which underlying distributed training mechanism to use. |
|
|
|
|
HPOJob Configuration should be used to configure the HPO Job. |
The specification of the hyperparameter tuning process https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config |
|
HyperparameterTuningObjective is a data structure that contains the target metric and the objective of the hyperparameter tuning. |
|
Specifies the type of content for input data. |
|
When using FILE input mode, different SageMaker built-in algorithms require different file types of input data See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html |
|
|
|
|
Initializes a new ParameterRangeOneOf. |
|
Allows a custom training algorithm to be executed on Amazon Sagemaker. |
|
|
|
Configuration for Running Training Jobs on Sagemaker. |
|
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 |