diff --git a/src/sagemaker_training/params.py b/src/sagemaker_training/params.py index c2568539..60850b6c 100644 --- a/src/sagemaker_training/params.py +++ b/src/sagemaker_training/params.py @@ -35,7 +35,7 @@ DEFAULT_MODULE_NAME_PARAM = "default_user_module_name" # type: str MPI_ENABLED = "sagemaker_mpi_enabled" # type: str PARAMETER_SERVER_ENABLED = "sagemaker_parameter_server_enabled" # type: str -MULTI_WORKER_MIRRORED_ENABLED = "sagemaker_multi_worker_mirrored_enabled" # type: str +MULTI_WORKER_MIRRORED_ENABLED = "sagemaker_multi_worker_mirrored_strategy_enabled" # type: str REGION_NAME_PARAM = "sagemaker_region" # type: str REGION_NAME_ENV = REGION_NAME_PARAM.upper() # type: str DEFAULT_INVOCATIONS_ACCEPT_ENV = "SAGEMAKER_DEFAULT_INVOCATIONS_ACCEPT" # type: str