控制平面

此模組包含選用的輔助程式，可將額外的偵錯和控制處理程序添加到您的應用程式中。

torch.distributed.elastic.control_plane.worker_main()

這是一個 context manager，它包裝了您的主要入口函數。 它結合了現有的 errors.record 邏輯，以及一個新的 _WorkerServer，它通過 Torch_WORKER_SERVER_SOCKET 指定的 unix socket 公開 handlers。

範例

@worker_main()
def main():
    pass

if __name__=="__main__":
   main()

回傳型別

Generator[None, None, None]