Skip to content

distributed.algorithms.join.rst

Latest commit

 

History

History
20 lines (15 loc) · 725 Bytes

distributed.algorithms.join.rst

File metadata and controls

20 lines (15 loc) · 725 Bytes
Raw

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: