控制平面
創建於: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]