sagemaker.core.constants

Constants used across the SageMaker Core package.

This module contains constant values that are shared across different components of the SageMaker SDK.

Module Attributes

SCRIPT_PARAM_NAME	Environment variable name for the entry point script name.
DIR_PARAM_NAME	Environment variable name for the directory containing the entry point script.
CONTAINER_LOG_LEVEL_PARAM_NAME	Environment variable name for the container log level.
JOB_NAME_PARAM_NAME	Environment variable name for the SageMaker job name.
MODEL_SERVER_WORKERS_PARAM_NAME	Environment variable name for the number of model server workers.
SAGEMAKER_REGION_PARAM_NAME	Environment variable name for the AWS region.
SAGEMAKER_OUTPUT_LOCATION	Environment variable name for the S3 output location.
NEO_ALLOWED_FRAMEWORKS	Set of frameworks allowed for Neo compilation.

sagemaker.core.constants.CONTAINER_LOG_LEVEL_PARAM_NAME = 'sagemaker_container_log_level'

Environment variable name for the container log level.

sagemaker.core.constants.DIR_PARAM_NAME = 'sagemaker_submit_directory'

Environment variable name for the directory containing the entry point script.

This constant specifies the S3 location or local path where the training/inference code and dependencies are located.

sagemaker.core.constants.JOB_NAME_PARAM_NAME = 'sagemaker_job_name'

Environment variable name for the SageMaker job name.

sagemaker.core.constants.MODEL_SERVER_WORKERS_PARAM_NAME = 'sagemaker_model_server_workers'

Environment variable name for the number of model server workers.

This constant specifies how many worker processes the model server should use for handling inference requests. More workers can improve throughput for CPU-bound models.

sagemaker.core.constants.NEO_ALLOWED_FRAMEWORKS = {'keras', 'mxnet', 'onnx', 'pytorch', 'tensorflow', 'tflite', 'xgboost'}

Set of frameworks allowed for Neo compilation.

Neo is SageMaker’s model compilation service that optimizes models for specific hardware targets.

sagemaker.core.constants.SAGEMAKER_OUTPUT_LOCATION = 'sagemaker_s3_output'

Environment variable name for the S3 output location.

sagemaker.core.constants.SAGEMAKER_REGION_PARAM_NAME = 'sagemaker_region'

Environment variable name for the AWS region.

sagemaker.core.constants.SCRIPT_PARAM_NAME = 'sagemaker_program'

Environment variable name for the entry point script name.