Python MXNet.hyperparameters Examples

Example #1

File: test_mxnet.py Project: zf109/sagemaker-python-sdk

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'

Example #2

File: test_mxnet.py Project: zf109/sagemaker-python-sdk

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'

Example #3

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"