frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix torchrun log message (#131652)

Browse Source 
fixes https://github.com/pytorch/pytorch/issues/131461

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131652
Approved by: https://github.com/awgu
This commit is contained in:
Howard Huang
2024-07-24 07:33:56 -07:00
committed by PyTorch MergeBot
parent d0e2ab617d
commit 1eedb0a962
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributed/run.py
View File

@ -708,10 +708,10 @@ def determine_local_world_size(nproc_per_node: str):
) from e
logger.info(
"Using nproc_per_node=%s," " setting to %s since the instance " "has %s %s",
"Using nproc_per_node=%s, setting nproc_per_node to %s since the instance has %s %s",
nproc_per_node,
num_proc,
os.cpu_count(),
num_proc,
device_type,
)
return num_proc