File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
src/sagemaker/remote_function Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -208,7 +208,8 @@ def remote(
208
208
files are accepted and uploaded to S3.
209
209
210
210
instance_count (int): The number of instances to use. Defaults to 1.
211
- NOTE: Remote function does not support instance_count > 1 for non Spark jobs.
211
+ NOTE: Remote function supports instance_count > 1 for Spark jobs, torchrun and
212
+ mpirun utilities
212
213
213
214
instance_type (str): The Amazon Elastic Compute Cloud (EC2) instance type to use to run
214
215
the SageMaker job. e.g. ml.c4.xlarge. If not provided, a ValueError is thrown.
@@ -657,7 +658,8 @@ def __init__(
657
658
files are accepted and uploaded to S3.
658
659
659
660
instance_count (int): The number of instances to use. Defaults to 1.
660
- NOTE: Remote function does not support instance_count > 1 for non Spark jobs.
661
+ NOTE: Remote function supports instance_count > 1 for Spark jobs, torchrun and
662
+ mpirun utilities
661
663
662
664
instance_type (str): The Amazon Elastic Compute Cloud (EC2) instance type to use to run
663
665
the SageMaker job. e.g. ml.c4.xlarge. If not provided, a ValueError is thrown.
You can’t perform that action at this time.
0 commit comments