控制平面
创建于:2024年5月30日 | 最后更新于:2024年6月4日
此模块包含可选的辅助函数,可为您的应用程序添加额外的调试和控制处理程序。
-
torch.distributed.elastic.control_plane.worker_main()[源代码]
这是一个上下文管理器,可以包装您的主入口函数。它结合了现有的 errors.record
逻辑以及一个新的 _WorkerServer
,该服务器通过由 Torch_WORKER_SERVER_SOCKET
指定的 Unix 套接字公开处理程序。
示例
@worker_main()
def main():
pass
if __name__ == "__main__":
main()
- 返回类型
Generator[None, None, None]