torch.distributed.elastic.multiprocessing.subprocess_handler.handlers 的源代码

# 版权所有（C）Facebook，Inc.及其关联公司。 # 版权所有。 # # 本源代码受存储在源代码树根目录中的 LICENSE 文件中规定的 BSD 风格许可协议的许可。从 torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler 导入( SubprocessHandler, ) __all__ = ["get_subprocess_handler"]
[文档]def get_subprocess_handler(
entrypoint: 字符串,
args: 元组,
env: 字典[str, str],
stdout: 字符串,
stderr: 字符串,
local_rank_id: 整数,
) -> 子进程处理器:
return 子进程处理器(
entrypoint=entrypoint,
args=args,
env=env,
stdout=stdout,
stderr=stderr,
local_rank_id=local_rank_id,
    )