@@ -90,15 +90,22 @@ def __init__(
9090 image_name (str): If specified, the estimator will use this image for training and
9191 hosting, instead of selecting the appropriate SageMaker official image based on
9292 framework_version and py_version. It can be an ECR url or dockerhub image and tag.
93+
9394 Examples:
94- 123 .dkr.ecr.us-west-2.amazonaws.com/my-custom-image:1.0
95- custom-image:latest.
95+ * ``123412341234 .dkr.ecr.us-west-2.amazonaws.com/my-custom-image:1.0``
96+ * `` custom-image:latest``
9697
9798 distributions (dict): A dictionary with information on how to run distributed
9899 training (default: None). To have parameter servers launched for training,
99100 set this value to be ``{'parameter_server': {'enabled': True}}``.
100101 **kwargs: Additional kwargs passed to the
101102 :class:`~sagemaker.estimator.Framework` constructor.
103+
104+ .. tip::
105+
106+ You can find additional parameters for initializing this class at
107+ :class:`~sagemaker.estimator.Framework` and
108+ :class:`~sagemaker.estimator.EstimatorBase`.
102109 """
103110 if framework_version is None :
104111 logger .warning (empty_framework_version_warning (MXNET_VERSION , self .LATEST_VERSION ))
@@ -161,8 +168,10 @@ def create_model(
161168 role from the Estimator will be used.
162169 vpc_config_override (dict[str, list[str]]): Optional override for VpcConfig set on
163170 the model. Default: use subnets and security groups from this Estimator.
171+
164172 * 'Subnets' (list[str]): List of subnet ids.
165173 * 'SecurityGroupIds' (list[str]): List of security group ids.
174+
166175 entry_point (str): Path (absolute or relative) to the local Python source file which
167176 should be executed as the entry point to training. If not specified, the training
168177 entry point is used.
@@ -175,10 +184,13 @@ def create_model(
175184 image_name (str): If specified, the estimator will use this image for hosting, instead
176185 of selecting the appropriate SageMaker official image based on framework_version
177186 and py_version. It can be an ECR url or dockerhub image and tag.
187+
178188 Examples:
179- 123.dkr.ecr.us-west-2.amazonaws.com/my-custom-image:1.0
180- custom-image:latest.
181- **kwargs: Additional kwargs passed to the MXNetModel constructor.
189+ * ``123412341234.dkr.ecr.us-west-2.amazonaws.com/my-custom-image:1.0``
190+ * ``custom-image:latest``
191+
192+ **kwargs: Additional kwargs passed to the :class:`~sagemaker.mxnet.model.MXNetModel`
193+ constructor.
182194
183195 Returns:
184196 sagemaker.mxnet.model.MXNetModel: A SageMaker ``MXNetModel`` object.
0 commit comments