syne_tune.remote.estimators module

syne_tune.remote.estimators.instance_sagemaker_estimator(**kwargs)[source]

Returns SageMaker estimator to be used for simulator back-end experiments and for remote launching of SageMaker back-end experiments.

Parameters:

kwargs – Extra arguments to SageMaker estimator

Returns:

SageMaker estimator

syne_tune.remote.estimators.basic_cpu_instance_sagemaker_estimator(**kwargs)[source]

Returns SageMaker estimator to be used for simulator back-end experiments and for remote launching of SageMaker back-end experiments.

Parameters:

kwargs – Extra arguments to SageMaker estimator

Returns:

SageMaker estimator

syne_tune.remote.estimators.pytorch_estimator(**estimator_kwargs)[source]

Get the PyTorch sagemaker estimator with the most up-to-date framework version. List of available containers: https://github.com/aws/deep-learning-containers/blob/master/available_images.md

Parameters:

estimator_kwargs – Estimator parameters as discussed in https://sagemaker.readthedocs.io/en/stable/frameworks/pytorch/sagemaker.pytorch.html

Return type:

PyTorch

Returns:

PyTorch estimator

syne_tune.remote.estimators.huggingface_estimator(**estimator_kwargs)[source]

Get the Huggingface sagemaker estimator with the most up-to-date framework version. List of available containers: https://github.com/aws/deep-learning-containers/blob/master/available_images.md

Parameters:

estimator_kwargs – Estimator parameters as discussed in https://sagemaker.readthedocs.io/en/stable/frameworks/huggingface/sagemaker.huggingface.html

Return type:

HuggingFace

Returns:

PyTorch estimator

syne_tune.remote.estimators.sklearn_estimator(**estimator_kwargs)[source]

Get the Scikit-learn sagemaker estimator with the most up-to-date framework version. List of available containers: https://github.com/aws/deep-learning-containers/blob/master/available_images.md

Parameters:

estimator_kwargs – Estimator parameters as discussed in https://sagemaker.readthedocs.io/en/stable/frameworks/sklearn/sagemaker.sklearn.html

Return type:

SKLearn

Returns:

PyTorch estimator

syne_tune.remote.estimators.mxnet_estimator(**estimator_kwargs)[source]

Get the MXNet sagemaker estimator with the most up-to-date framework version. List of available containers: https://github.com/aws/deep-learning-containers/blob/master/available_images.md

Parameters:

estimator_kwargs – Estimator parameters as discussed in https://sagemaker.readthedocs.io/en/stable/frameworks/mxnet/sagemaker.mxnet.html

Return type:

MXNet

Returns:

PyTorch estimator