快捷方式

控制平面

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

torch.distributed.elastic.control_plane.worker_main()[source]

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

示例

@worker_main()
def main():
    pass

if __name__=="__main__":
   main()
返回类型

Generator[None, None, None]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源