File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -46,6 +46,8 @@ def retrieve_default(
46
46
retrieve the default hyperparameters. (Default: None).
47
47
model_version (str): The version of the model for which to retrieve the
48
48
default hyperparameters. (Default: None).
49
+ instance_type (str): An instance type to optionally supply in order to get hyperparameters
50
+ specific for the instance type.
49
51
include_container_hyperparameters (bool): ``True`` if the container hyperparameters
50
52
should be returned. Container hyperparameters are not used to tune
51
53
the specific algorithm. They are used by SageMaker Training jobs to set up
Original file line number Diff line number Diff line change @@ -64,6 +64,8 @@ def _retrieve_default_hyperparameters(
64
64
object, used for SageMaker interactions. If not
65
65
specified, one is created using the default AWS configuration
66
66
chain. (Default: sagemaker.jumpstart.constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION).
67
+ instance_type (str): An instance type to optionally supply in order to get hyperparameters
68
+ specific for the instance type.
67
69
Returns:
68
70
dict: the hyperparameters to use for the model.
69
71
"""
You can’t perform that action at this time.
0 commit comments