We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a573c54 commit b1dbf33Copy full SHA for b1dbf33
torchx/components/dist.py
@@ -35,7 +35,8 @@ def ddp(
35
Args:
36
image: container image.
37
entrypoint: script or binary to run within the image.
38
- resource: Registered named resource.
+ resource: Optional resource identifier. The resource parameter
39
+ gets ignored when running on the local scheduler.
40
nnodes: Number of nodes.
41
nproc_per_node: Number of processes per node.
42
name: Name of the application.
@@ -56,8 +57,8 @@ def ddp(
56
57
entrypoint=entrypoint,
58
resource=resource or specs.NULL_RESOURCE,
59
num_replicas=nnodes,
- script_args=list(script_args),
60
- script_envs=env,
+ args=list(script_args),
61
+ env=env,
62
nproc_per_node=nproc_per_node,
63
nnodes=nnodes,
64
max_restarts=0,
0 commit comments