Skip to content

Commit f3c3ac2

Browse files
add sbatch network arg (#230)
Signed-off-by: Youngeun Kwon <[email protected]>
1 parent 71e3e70 commit f3c3ac2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nemo_run/core/execution/slurm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,7 @@ class ResourceRequest:
340340
setup_lines: Optional[str] = None
341341
het_group_indices: Optional[list[int]] = None
342342
segment: Optional[int] = None
343+
network: Optional[str] = None
343344

344345
#: Set by the executor; cannot be initialized
345346
job_name: str = field(init=False, default="nemo-job")

0 commit comments

Comments
 (0)