sagemaker.core.training_compiler.config

sagemaker.core.training_compiler.config#

Configuration for the SageMaker Training Compiler.

Classes

TrainingCompilerConfig([enabled, debug])

The SageMaker Training Compiler configuration class.
class sagemaker.core.training_compiler.config.TrainingCompilerConfig(enabled=True, debug=False)[source]#

Bases: object

The SageMaker Training Compiler configuration class.

DEBUG_PATH = '/opt/ml/output/data/compiler/'#
HP_ENABLE_COMPILER = 'sagemaker_training_compiler_enabled'#
HP_ENABLE_DEBUG = 'sagemaker_training_compiler_debug_mode'#
SUPPORTED_INSTANCE_CLASS_PREFIXES = ['p3', 'p3dn', 'g4dn', 'p4d', 'g5']#
disclaimers_and_warnings()[source]#

Disclaimers and warnings.

Logs disclaimers and warnings about the requested configuration of SageMaker Training Compiler.

classmethod validate(estimator)[source]#

Checks if SageMaker Training Compiler is configured correctly.

Parameters:

estimator (sagemaker.estimator.Estimator) – An estimator object. When SageMaker Training Compiler is enabled, it validates if the estimator is configured to be compatible with Training Compiler.

Raises:

ValueError – Raised if the requested configuration is not compatible with SageMaker Training Compiler.

Contents