def test_estimator_script_mode_launch_parameter_server(sagemaker_session): mx = MXNet(entry_point=SCRIPT_PATH, role=ROLE, sagemaker_session=sagemaker_session, train_instance_count=INSTANCE_COUNT, train_instance_type=INSTANCE_TYPE, distributions=LAUNCH_PS_DISTRIBUTIONS_DICT, framework_version='1.3.0') assert mx.hyperparameters().get(MXNet.LAUNCH_PS_ENV_NAME) == 'true'
def test_estimator_script_mode_dont_launch_parameter_server(sagemaker_session): mx = MXNet(entry_point=SCRIPT_PATH, role=ROLE, sagemaker_session=sagemaker_session, train_instance_count=INSTANCE_COUNT, train_instance_type=INSTANCE_TYPE, distributions={'parameter_server': { 'enabled': False }}, framework_version='1.3.0') assert mx.hyperparameters().get(MXNet.LAUNCH_PS_ENV_NAME) == 'false'
def test_estimator_script_mode_dont_launch_parameter_server(sagemaker_session): mx = MXNet( entry_point=SCRIPT_PATH, framework_version="1.3.0", py_version="py2", role=ROLE, sagemaker_session=sagemaker_session, instance_count=INSTANCE_COUNT, instance_type=INSTANCE_TYPE, distribution={"parameter_server": {"enabled": False}}, ) assert mx.hyperparameters().get(MXNet.LAUNCH_PS_ENV_NAME) == "false"