控制平面

此模組包含可選的輔助工具，可為您的應用程式新增額外的除錯和控制處理程式。

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]