控制平面 ¶

此模块包含可选的辅助程序，这些程序将额外的调试和控制处理程序添加到您的应用程序中。

torch.distributed.elastic.control_plane.worker_main()[源代码][源代码] ¶

这是一个上下文管理器，用于包装您的入口函数。它结合了现有的 errors.record 逻辑以及新的 _WorkerServer ，通过指定的 Torch_WORKER_SERVER_SOCKET Unix 套接字公开处理程序。

示例

@worker_main()
def main():
    pass


if __name__ == "__main__":
    main()

文档