@@ -558,19 +558,22 @@ def create_model(
558
558
model_server_workers (int): Optional. The number of worker processes used by the
559
559
inference server. If None, server will use one worker per vCPU.
560
560
vpc_config_override (dict[str, list[str]]): Optional override for VpcConfig set on the
561
- model.
562
- Default: use subnets and security groups from this Estimator.
561
+ model. Default: use subnets and security groups from this Estimator.
562
+
563
563
* 'Subnets' (list[str]): List of subnet ids.
564
564
* 'SecurityGroupIds' (list[str]): List of security group ids.
565
+
565
566
endpoint_type (str): Optional. Selects the software stack used by the inference server.
566
567
If not specified, the model will be configured to use the default
567
568
SageMaker model server. If 'tensorflow-serving', the model will be configured to
568
569
use the SageMaker Tensorflow Serving container.
569
570
entry_point (str): Path (absolute or relative) to the local Python source file which
570
- should be executed as the entry point to training. If not specified and
571
- ``endpoint_type`` is 'tensorflow-serving', no entry point is used. If
572
- ``endpoint_type`` is also ``None``, then the training entry point is used.
573
- source_dir (str): Path (absolute or relative or an S3 URI ) to a directory with any
571
+ should be executed as the entry point to training. If ``source_dir`` is specified,
572
+ then ``entry_point`` must point to a file located at the root of ``source_dir``.
573
+ If not specified and ``endpoint_type`` is 'tensorflow-serving',
574
+ no entry point is used. If ``endpoint_type`` is also ``None``,
575
+ then the training entry point is used.
576
+ source_dir (str): Path (absolute or relative or an S3 URI) to a directory with any
574
577
other serving source code dependencies aside from the entry point file. If
575
578
``source_dir`` is an S3 URI, it must point to a tar.gz file. If not specified
576
579
and ``endpoint_type`` is 'tensorflow-serving', no source_dir is used. If
@@ -828,9 +831,11 @@ def transformer(
828
831
If 'tensorflow-serving', the model will be configured to
829
832
use the SageMaker Tensorflow Serving container.
830
833
entry_point (str): Path (absolute or relative) to the local Python source file which
831
- should be executed as the entry point to training. If not specified and
832
- ``endpoint_type`` is 'tensorflow-serving', no entry point is used. If
833
- ``endpoint_type`` is also ``None``, then the training entry point is used.
834
+ should be executed as the entry point to training. If ``source_dir`` is specified,
835
+ then ``entry_point`` must point to a file located at the root of ``source_dir``.
836
+ If not specified and ``endpoint_type`` is 'tensorflow-serving',
837
+ no entry point is used. If ``endpoint_type`` is also ``None``,
838
+ then the training entry point is used.
834
839
vpc_config_override (dict[str, list[str]]): Optional override for
835
840
the VpcConfig set on the model.
836
841
Default: use subnets and security groups from this Estimator.
0 commit comments