評價此頁

控制平面#

創建於: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]