mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
cd82096973
Fixes #155017 Pull Request resolved: https://github.com/pytorch/pytorch/pull/155298 Approved by: https://github.com/svekars Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
647 B
647 B
---
class: hidden-section
---
Generic Join Context Manager
The generic join context manager facilitates distributed training on uneven
inputs. This page outlines the API of the relevant classes: {class}Join,
{class}Joinable, and {class}JoinHook. For a tutorial, see
Distributed Training with Uneven Inputs Using the Join Context Manager.
.. autoclass:: torch.distributed.algorithms.Join
:members:
.. autoclass:: torch.distributed.algorithms.Joinable
:members:
.. autoclass:: torch.distributed.algorithms.JoinHook
:members: