We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5b28da0 commit bfb030dCopy full SHA for bfb030d
tools/modelarts_adapter/modelarts.py
@@ -67,7 +67,7 @@ def run_with_single_rank(local_rank: int = 0, signal: str = "/tmp/SUCCESS") -> C
67
def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
68
@functools.wraps(func)
69
def wrapper(*args: Any, **kwargs: Any) -> Any:
70
- if local_rank in [0, None]:
+ if local_rank is None or local_rank % 8 == 0:
71
result = func(*args, **kwargs)
72
with open(signal, "w") as f:
73
f.write("\n")
0 commit comments