控制面板
此模組包含可選的輔助程式,可將額外的偵錯和控制處理常式新增到您的應用程式中。
-
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]