sagemaker.core.image_retriever.image_retriever_utils

Contents

sagemaker.core.image_retriever.image_retriever_utils#

Functions for generating ECR image URIs for pre-built SageMaker Docker images.

Functions

config_for_framework(framework)

Loads the JSON config for the given framework.

sagemaker.core.image_retriever.image_retriever_utils.config_for_framework(framework)[source]#: Loads the JSON config for the given framework.